3proxy.c 53 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2008 by ZARAZA <3APA3A@security.nnov.ru>
  4. please read License Agreement
  5. $Id: 3proxy.c,v 1.132 2011-08-15 19:52:26 vlad Exp $
  6. */
  7. #include "proxy.h"
  8. #ifndef _WIN32
  9. #include <sys/resource.h>
  10. #ifndef NOPLUGINS
  11. #include <dlfcn.h>
  12. #endif
  13. #endif
  14. #ifndef DEFAULTCONFIG
  15. #define DEFAULTCONFIG conf.stringtable[25]
  16. #endif
  17. typedef int (*MAINFUNC)(int, char**);
  18. pthread_mutex_t bandlim_mutex;
  19. pthread_mutex_t tc_mutex;
  20. pthread_mutex_t pwl_mutex;
  21. pthread_mutex_t hash_mutex;
  22. #ifndef NOODBC
  23. pthread_mutex_t odbc_mutex;
  24. #endif
  25. int readconfig(FILE * fp);
  26. int haveerror = 0;
  27. int linenum = 0;
  28. time_t basetime = 0;
  29. void doschedule(void);
  30. struct counter_header {
  31. unsigned char sig[4];
  32. time_t updated;
  33. } cheader = {"3CF", (time_t)0};
  34. struct counter_record {
  35. uint64_t traf64;
  36. time_t cleared;
  37. time_t updated;
  38. } crecord;
  39. int mainfunc (int argc, char** argv);
  40. struct proxydef childdef = {NULL, 0, 0, S_NOSERVICE, ""};
  41. #define STRINGBUF 65535
  42. #define NPARAMS 4096
  43. unsigned char tmpbuf[1024];
  44. FILE *writable;
  45. extern unsigned char *strings[];
  46. #ifndef _WIN32
  47. char *chrootp = NULL;
  48. #endif
  49. char * curconf = NULL;
  50. FILE * confopen(){
  51. curconf = conf.conffile;
  52. #ifndef _WIN32
  53. if(chrootp){
  54. if(strstr(curconf, chrootp) == curconf)
  55. curconf += strlen(chrootp);
  56. }
  57. #endif
  58. if(writable) {
  59. rewind(writable);
  60. return writable;
  61. }
  62. return fopen(curconf, "r");
  63. }
  64. void clearall(){
  65. freeconf(&conf);
  66. }
  67. #ifdef _WIN32
  68. OSVERSIONINFO osv;
  69. int service = 0;
  70. void cyclestep(void);
  71. #ifndef _WINCE
  72. SERVICE_STATUS_HANDLE hSrv;
  73. DWORD dwCurrState;
  74. int SetStatus( DWORD dwState, DWORD dwExitCode, DWORD dwProgress )
  75. {
  76. SERVICE_STATUS srvStatus;
  77. srvStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
  78. srvStatus.dwCurrentState = dwCurrState = dwState;
  79. srvStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE | SERVICE_ACCEPT_SHUTDOWN;
  80. srvStatus.dwWin32ExitCode = dwExitCode;
  81. srvStatus.dwServiceSpecificExitCode = 0;
  82. srvStatus.dwCheckPoint = dwProgress;
  83. srvStatus.dwWaitHint = 3000;
  84. return SetServiceStatus( hSrv, &srvStatus );
  85. }
  86. void __stdcall CommandHandler( DWORD dwCommand )
  87. {
  88. FILE *fp;
  89. int error;
  90. switch( dwCommand )
  91. {
  92. case SERVICE_CONTROL_STOP:
  93. case SERVICE_CONTROL_SHUTDOWN:
  94. SetStatus( SERVICE_STOP_PENDING, 0, 1 );
  95. conf.paused++;
  96. conf.timetoexit = 1;
  97. Sleep(2000);
  98. SetStatus( SERVICE_STOPPED, 0, 0 );
  99. #ifndef NOODBC
  100. pthread_mutex_lock(&odbc_mutex);
  101. close_sql();
  102. pthread_mutex_unlock(&odbc_mutex);
  103. #endif
  104. break;
  105. case SERVICE_CONTROL_PAUSE:
  106. SetStatus( SERVICE_PAUSE_PENDING, 0, 1 );
  107. conf.paused++;
  108. SetStatus( SERVICE_PAUSED, 0, 0 );
  109. break;
  110. case SERVICE_CONTROL_CONTINUE:
  111. SetStatus( SERVICE_CONTINUE_PENDING, 0, 1 );
  112. clearall();
  113. fp = confopen();
  114. if(fp){
  115. error = readconfig(fp);
  116. if(error) {
  117. clearall();
  118. }
  119. if(!writable)fclose(fp);
  120. }
  121. SetStatus( SERVICE_RUNNING, 0, 0 );
  122. break;
  123. default: ;
  124. }
  125. }
  126. void __stdcall ServiceMain(int argc, unsigned char* argv[] )
  127. {
  128. hSrv = RegisterServiceCtrlHandler(conf.stringtable[1], (LPHANDLER_FUNCTION)CommandHandler);
  129. if( hSrv == 0 ) return;
  130. SetStatus( SERVICE_START_PENDING, 0, 1 );
  131. SetStatus( SERVICE_RUNNING, 0, 0 );
  132. cyclestep();
  133. }
  134. #endif
  135. #else
  136. void mysigusr1 (int sig){
  137. conf.needreload = 1;
  138. }
  139. int even = 0;
  140. void mysigpause (int sig){
  141. conf.paused++;
  142. even = !even;
  143. if(!even){
  144. conf.needreload = 1;
  145. }
  146. }
  147. void mysigterm (int sig){
  148. conf.paused++;
  149. usleep(999*SLEEPTIME);
  150. usleep(999*SLEEPTIME);
  151. #ifndef NOODBC
  152. pthread_mutex_lock(&odbc_mutex);
  153. close_sql();
  154. pthread_mutex_unlock(&odbc_mutex);
  155. #endif
  156. conf.timetoexit = 1;
  157. }
  158. #endif
  159. void dumpmem(void);
  160. int reload (void){
  161. FILE *fp;
  162. int error = -2;
  163. conf.paused++;
  164. clearall();
  165. conf.paused++;
  166. fp = confopen();
  167. if(fp){
  168. error = readconfig(fp);
  169. if(error) {
  170. clearall();
  171. }
  172. if(!writable)fclose(fp);
  173. }
  174. return error;
  175. }
  176. struct schedule *schedule;
  177. #ifdef _WIN32
  178. DWORD WINAPI startsrv(LPVOID data) {
  179. #else
  180. void * startsrv(void * data) {
  181. #endif
  182. struct child *d = (struct child *)data;
  183. mainfunc(d->argc, (char **)d->argv);
  184. return 0;
  185. }
  186. int included =0;
  187. int getrotate(char c){
  188. switch(c){
  189. case 'c':
  190. case 'C':
  191. return MINUTELY;
  192. case 'h':
  193. case 'H':
  194. return HOURLY;
  195. case 'd':
  196. case 'D':
  197. return DAILY;
  198. case 'w':
  199. case 'W':
  200. return WEEKLY;
  201. case 'y':
  202. case 'Y':
  203. return ANNUALLY;
  204. case 'm':
  205. case 'M':
  206. return MONTHLY;
  207. default:
  208. return NEVER;
  209. }
  210. }
  211. int parsestr (unsigned char *str, unsigned char **argm, int nitems, unsigned char ** buff, int *inbuf, int *bufsize){
  212. #define buf (*buff)
  213. int argc = 0;
  214. int space = 1;
  215. int comment = 0;
  216. unsigned char * incbegin = 0;
  217. int fd;
  218. int res, len;
  219. int i = 1;
  220. unsigned char *str1;
  221. for(;;str++){
  222. if(*str == '\"'){
  223. str1 = str;
  224. do {
  225. *str1 = *(str1 + 1);
  226. }while(*(str1++));
  227. if(!comment || *str != '\"'){
  228. comment = !comment;
  229. }
  230. }
  231. switch(*str){
  232. case '\0':
  233. if(comment) return -1;
  234. argm[argc] = 0;
  235. return argc;
  236. case '$':
  237. if(!comment && !included){
  238. incbegin = str;
  239. *str = 0;
  240. }
  241. break;
  242. case '\r':
  243. case '\n':
  244. case '\t':
  245. case ' ':
  246. if(!comment){
  247. *str = 0;
  248. space = 1;
  249. i = 0;
  250. if(incbegin){
  251. argc--;
  252. if((fd = open((char *)incbegin+1, O_RDONLY)) <= 0){
  253. fprintf(stderr, "Failed to open %s\n", incbegin+1);
  254. break;
  255. }
  256. if((*bufsize - *inbuf) <STRINGBUF){
  257. *bufsize += STRINGBUF;
  258. if(!(buf = myrealloc(buf, *bufsize))){
  259. fprintf(stderr, "Failed to allocate memory for %s\n", incbegin+1);
  260. close(fd);
  261. break;
  262. }
  263. }
  264. len = 0;
  265. if(argm[argc]!=(incbegin+1)) {
  266. len = (int)strlen((char *)argm[argc]);
  267. memcpy(buf+*inbuf, argm[argc], len);
  268. }
  269. if((res = read(fd, buf+*inbuf+len, STRINGBUF-(1+len))) <= 0) {
  270. perror((char *)incbegin+1);
  271. close(fd);
  272. break;
  273. }
  274. close(fd);
  275. buf[*inbuf+res+len] = 0;
  276. incbegin = buf + *inbuf;
  277. (*inbuf) += (res + len + 1);
  278. included++;
  279. argc+=parsestr(incbegin, argm + argc, nitems - argc, buff, inbuf, bufsize);
  280. included--;
  281. incbegin = NULL;
  282. }
  283. break;
  284. }
  285. default:
  286. i++;
  287. if(space) {
  288. if(comment && *str == '\"' && str[1] != '\"'){
  289. str++;
  290. comment = 0;
  291. }
  292. argm[argc++] = str;
  293. if(argc >= nitems) return argc;
  294. space = 0;
  295. }
  296. }
  297. }
  298. #undef buf
  299. }
  300. unsigned char * dologname (unsigned char *buf, unsigned char *name, const unsigned char *ext, ROTATION lt, time_t t) {
  301. struct tm *ts;
  302. ts = localtime(&t);
  303. if(strchr((char *)name, '%')){
  304. struct clientparam fakecli;
  305. memset(&fakecli, 0, sizeof(fakecli));
  306. dobuf2(&fakecli, buf, NULL, NULL, ts, (char *)name);
  307. return buf;
  308. }
  309. switch(lt){
  310. case NONE:
  311. sprintf((char *)buf, "%s", name);
  312. break;
  313. case ANNUALLY:
  314. sprintf((char *)buf, "%s.%04d", name, ts->tm_year+1900);
  315. break;
  316. case MONTHLY:
  317. sprintf((char *)buf, "%s.%04d.%02d", name, ts->tm_year+1900, ts->tm_mon+1);
  318. break;
  319. case WEEKLY:
  320. t = t - (ts->tm_wday * (60*60*24));
  321. ts = localtime(&t);
  322. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  323. break;
  324. case DAILY:
  325. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  326. break;
  327. case HOURLY:
  328. sprintf((char *)buf, "%s.%04d.%02d.%02d-%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday, ts->tm_hour);
  329. break;
  330. case MINUTELY:
  331. sprintf((char *)buf, "%s.%04d.%02d.%02d-%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday, ts->tm_hour, ts->tm_min);
  332. break;
  333. default:
  334. break;
  335. }
  336. if(ext){
  337. strcat((char *)buf, ".");
  338. strcat((char *)buf, (char *)ext);
  339. }
  340. return buf;
  341. }
  342. int wday = 0;
  343. int timechanged (time_t oldtime, time_t newtime, ROTATION lt){
  344. struct tm tmold;
  345. struct tm *tm;
  346. tm = localtime(&oldtime);
  347. memcpy(&tmold, tm, sizeof(tmold));
  348. tm = localtime(&newtime);
  349. switch(lt){
  350. case MINUTELY:
  351. if(tm->tm_min != tmold.tm_min)return 1;
  352. break;
  353. case HOURLY:
  354. if(tm->tm_hour != tmold.tm_hour)return 1;
  355. break;
  356. case DAILY:
  357. if(tm->tm_yday != tmold.tm_yday)return 1;
  358. break;
  359. case MONTHLY:
  360. if(tm->tm_mon != tmold.tm_mon)return 1;
  361. break;
  362. case ANNUALLY:
  363. if(tm->tm_year != tmold.tm_year)return 1;
  364. break;
  365. case WEEKLY:
  366. if(((newtime - oldtime) > (60*60*24*7))
  367. || tm->tm_wday < tmold.tm_wday
  368. || (tm->tm_wday == tmold.tm_wday && (newtime - oldtime) > (60*60*24*6))
  369. )return 1;
  370. break;
  371. default:
  372. break;
  373. }
  374. return 0;
  375. }
  376. void doschedule(void){
  377. struct schedule *sched, *prevsched = NULL, *nextsched;
  378. int res;
  379. conf.time = time(0);
  380. for(sched=schedule; sched; sched=sched->next){
  381. if(conf.needreload || conf.timetoexit || (conf.time > sched->start_time && timechanged(sched->start_time, conf.time, sched->type))){
  382. sched->start_time = conf.time;
  383. nextsched = sched->next;
  384. res = (*sched->function)(sched->data);
  385. switch(res){
  386. case 1:
  387. if(prevsched) prevsched->next = nextsched;
  388. else schedule = nextsched;
  389. break;
  390. }
  391. }
  392. prevsched = sched;
  393. }
  394. }
  395. void dumpcounters(struct trafcount *tlin, int counterd){
  396. struct trafcount *tl;
  397. if(counterd >= 0 && tlin) {
  398. conf.time = time(0);
  399. if(cheader.updated && conf.countertype && timechanged(cheader.updated, conf.time, conf.countertype)){
  400. FILE * cfp;
  401. cfp = fopen((char *)dologname(tmpbuf, (unsigned char *)conf.counterfile, NULL, conf.countertype, cheader.updated), "w");
  402. if(cfp){
  403. for(tl = tlin; cfp && tl; tl = tl->next){
  404. if(tl->type >= conf.countertype)
  405. fprintf(cfp, "%05d %020"PRINTF_INT64_MODIFIER"u%s%s\n", tl->number, tl->traf64, tl->comment?" #" : "", tl->comment? tl->comment : "");
  406. }
  407. fclose(cfp);
  408. }
  409. }
  410. cheader.updated = conf.time;
  411. lseek(counterd, 0, SEEK_SET);
  412. write(counterd, &cheader, sizeof(struct counter_header));
  413. for(tl=tlin; tl; tl = tl->next){
  414. if(tl->number){
  415. lseek(counterd,
  416. sizeof(struct counter_header) + (tl->number - 1) * sizeof(struct counter_record),
  417. SEEK_SET);
  418. crecord.traf64 = tl->traf64;
  419. crecord.cleared = tl->cleared;
  420. crecord.updated = tl->updated;
  421. write(counterd, &crecord, sizeof(struct counter_record));
  422. }
  423. if(tl->type!=NEVER && timechanged(tl->cleared, conf.time, tl->type)){
  424. tl->cleared = conf.time;
  425. tl->traf64 = 0;
  426. }
  427. }
  428. }
  429. }
  430. void cyclestep(void){
  431. struct tm *tm;
  432. time_t minutecounter;
  433. minutecounter = time(0);
  434. for(;;){
  435. usleep(SLEEPTIME*999);
  436. conf.time = time(0);
  437. if(conf.needreload) {
  438. doschedule();
  439. reload();
  440. conf.needreload = 0;
  441. }
  442. doschedule();
  443. if(conf.stdlog)fflush(conf.stdlog);
  444. if(timechanged(minutecounter, conf.time, MINUTELY)) {
  445. struct filemon *fm;
  446. struct stat sb;
  447. for(fm=conf.fmon; fm; fm=fm->next){
  448. if(!stat(fm->path, &sb)){
  449. if(fm->sb.st_mtime != sb.st_mtime || fm->sb.st_size != sb.st_size){
  450. stat(fm->path, &fm->sb);
  451. conf.needreload = 1;
  452. }
  453. }
  454. }
  455. }
  456. if(timechanged(basetime, conf.time, DAILY)) {
  457. tm = localtime(&conf.time);
  458. wday = (1 << tm->tm_wday);
  459. tm->tm_hour = tm->tm_min = tm->tm_sec = 0;
  460. basetime = mktime(tm);
  461. }
  462. if(conf.logname) {
  463. if(timechanged(conf.logtime, conf.time, conf.logtype)) {
  464. FILE *fp, *fp1;
  465. fp = fopen((char *)dologname (tmpbuf, conf.logname, NULL, conf.logtype, conf.time), "a");
  466. if (fp) {
  467. fp1 = conf.stdlog;
  468. conf.stdlog = fp;
  469. if(fp1) fclose(fp1);
  470. }
  471. fseek(stdout, 0L, SEEK_END);
  472. usleep(SLEEPTIME);
  473. conf.logtime = conf.time;
  474. if(conf.logtype != NONE && conf.rotate) {
  475. int t;
  476. t = 1;
  477. switch(conf.logtype){
  478. case ANNUALLY:
  479. t = t * 12;
  480. case MONTHLY:
  481. t = t * 4;
  482. case WEEKLY:
  483. t = t * 7;
  484. case DAILY:
  485. t = t * 24;
  486. case HOURLY:
  487. t = t * 60;
  488. case MINUTELY:
  489. t = t * 60;
  490. default:
  491. break;
  492. }
  493. dologname (tmpbuf, conf.logname, (conf.archiver)?conf.archiver[1]:NULL, conf.logtype, (conf.logtime - t * conf.rotate));
  494. remove ((char *) tmpbuf);
  495. if(conf.archiver) {
  496. int i;
  497. *tmpbuf = 0;
  498. for(i = 2; i < conf.archiverc && strlen((char *)tmpbuf) < 512; i++){
  499. strcat((char *)tmpbuf, " ");
  500. if(!strcmp((char *)conf.archiver[i], "%A")){
  501. strcat((char *)tmpbuf, "\"");
  502. dologname (tmpbuf + strlen((char *)tmpbuf), conf.logname, conf.archiver[1], conf.logtype, (conf.logtime - t));
  503. strcat((char *)tmpbuf, "\"");
  504. }
  505. else if(!strcmp((char *)conf.archiver[i], "%F")){
  506. strcat((char *)tmpbuf, "\"");
  507. dologname (tmpbuf+strlen((char *)tmpbuf), conf.logname, NULL, conf.logtype, (conf.logtime-t));
  508. strcat((char *)tmpbuf, "\"");
  509. }
  510. else
  511. strcat((char *)tmpbuf, (char *)conf.archiver[i]);
  512. }
  513. system((char *)tmpbuf+1);
  514. }
  515. }
  516. }
  517. }
  518. if(conf.counterd >= 0 && conf.trafcounter) {
  519. if(timechanged(cheader.updated, conf.time, MINUTELY)){
  520. dumpcounters(conf.trafcounter, conf.counterd);
  521. }
  522. }
  523. if(conf.timetoexit){
  524. conf.paused++;
  525. doschedule();
  526. usleep(SLEEPTIME*999);
  527. usleep(SLEEPTIME*999);
  528. usleep(SLEEPTIME*999);
  529. return;
  530. }
  531. }
  532. }
  533. #define RETURN(x) {res = x; goto CLEARRETURN;}
  534. int start_proxy_thread(struct child * chp){
  535. pthread_t thread;
  536. #ifdef _WIN32
  537. HANDLE h;
  538. #endif
  539. conf.threadinit = 1;
  540. #ifdef _WIN32
  541. #ifndef _WINCE
  542. h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, 16384, startsrv, (void *) chp, (DWORD)0, &thread);
  543. #else
  544. h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, 16384, startsrv, (void *) chp, (DWORD)0, &thread);
  545. #endif
  546. if(h)CloseHandle(h);
  547. #else
  548. pthread_attr_init(&pa);
  549. pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + 16384);
  550. pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
  551. pthread_create(&thread, &pa, startsrv, (void *)chp);
  552. #endif
  553. while(conf.threadinit)usleep(SLEEPTIME);
  554. if(haveerror) {
  555. fprintf(stderr, "Service not started on line: %d\n", linenum);
  556. return(40);
  557. }
  558. return 0;
  559. }
  560. static int h_proxy(int argc, unsigned char ** argv){
  561. struct child ch;
  562. ch.argc = argc;
  563. ch.argv = argv;
  564. if(!strcmp((char *)argv[0], "proxy")) {
  565. childdef.pf = proxychild;
  566. childdef.port = 3128;
  567. childdef.isudp = 0;
  568. childdef.service = S_PROXY;
  569. childdef.helpmessage = " -n - no NTLM support\n";
  570. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize)){
  571. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, proxy may run very slow\n", linenum);
  572. }
  573. }
  574. else if(!strcmp((char *)argv[0], "pop3p")) {
  575. childdef.pf = pop3pchild;
  576. childdef.port = 110;
  577. childdef.isudp = 0;
  578. childdef.service = S_POP3P;
  579. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  580. }
  581. else if(!strcmp((char *)argv[0], "smtpp")) {
  582. childdef.pf = smtppchild;
  583. childdef.port = 25;
  584. childdef.isudp = 0;
  585. childdef.service = S_SMTPP;
  586. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  587. }
  588. else if(!strcmp((char *)argv[0], "ftppr")) {
  589. childdef.pf = ftpprchild;
  590. childdef.port = 21;
  591. childdef.isudp = 0;
  592. childdef.service = S_FTPPR;
  593. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  594. }
  595. else if(!strcmp((char *)argv[0], "socks")) {
  596. childdef.pf = sockschild;
  597. childdef.port = 1080;
  598. childdef.isudp = 0;
  599. childdef.service = S_SOCKS;
  600. childdef.helpmessage = " -n - no NTLM support\n";
  601. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize)){
  602. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, socks may run very slow\n", linenum);
  603. }
  604. }
  605. else if(!strcmp((char *)argv[0], "tcppm")) {
  606. childdef.pf = tcppmchild;
  607. childdef.port = 0;
  608. childdef.isudp = 0;
  609. childdef.service = S_TCPPM;
  610. childdef.helpmessage = "";
  611. }
  612. else if(!strcmp((char *)argv[0], "icqpr")) {
  613. childdef.pf = icqprchild;
  614. childdef.port = 0;
  615. childdef.isudp = 0;
  616. childdef.service = S_ICQPR;
  617. childdef.helpmessage = "";
  618. }
  619. /*
  620. else if(!strcmp((char *)argv[0], "msnpr")) {
  621. childdef.pf = msnprchild;
  622. childdef.port = 0;
  623. childdef.isudp = 0;
  624. childdef.service = S_MSNPR;
  625. childdef.helpmessage = "";
  626. }
  627. */
  628. else if(!strcmp((char *)argv[0], "udppm")) {
  629. childdef.pf = udppmchild;
  630. childdef.port = 0;
  631. childdef.isudp = 1;
  632. childdef.service = S_UDPPM;
  633. childdef.helpmessage = " -s single packet UDP service for request/reply (DNS-like) services\n";
  634. }
  635. else if(!strcmp((char *)argv[0], "admin")) {
  636. childdef.pf = adminchild;
  637. childdef.port = 80;
  638. childdef.isudp = 0;
  639. childdef.service = S_ADMIN;
  640. }
  641. else if(!strcmp((char *)argv[0], "dnspr")) {
  642. childdef.pf = dnsprchild;
  643. childdef.port = 53;
  644. childdef.isudp = 1;
  645. childdef.service = S_DNSPR;
  646. childdef.helpmessage = " -s - simple DNS forwarding - do not use 3proxy resolver / name cache\n";
  647. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize) || resolvfunc == fakeresolver){
  648. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, dnspr will not work as expected\n", linenum);
  649. }
  650. }
  651. return start_proxy_thread(&ch);
  652. }
  653. static int h_internal(int argc, unsigned char ** argv){
  654. getip46(46, argv[1], (struct sockaddr *)&conf.intsa);
  655. return 0;
  656. }
  657. static int h_external(int argc, unsigned char ** argv){
  658. int res;
  659. #ifndef NOIPV6
  660. struct sockaddr_in6 sa6;
  661. res = getip46(46, argv[1], (struct sockaddr *)&sa6);
  662. if(!res) return 1;
  663. memcpy((*SAFAMILY(&sa6)==AF_INET)?(void *)&conf.extsa:(void *)&conf.extsa6, &sa6, SASIZE(&sa6));
  664. #else
  665. res = getip46(46, argv[1], (struct sockaddr *)&conf.extsa);
  666. if(!res) return 1;
  667. #endif
  668. return 0;
  669. }
  670. static int h_log(int argc, unsigned char ** argv){
  671. conf.logfunc = logstdout;
  672. if(conf.logtarget){
  673. myfree(conf.logtarget);
  674. conf.logtarget = NULL;
  675. }
  676. if(argc > 1) {
  677. conf.logtarget = (unsigned char *)mystrdup((char *)argv[1]);
  678. if(*argv[1]=='@'){
  679. #ifndef _WIN32
  680. openlog((char *)conf.logtarget+1, LOG_PID, LOG_DAEMON);
  681. conf.logfunc = logsyslog;
  682. #endif
  683. }
  684. #ifndef NOODBC
  685. else if(*argv[1]=='&'){
  686. pthread_mutex_lock(&odbc_mutex);
  687. close_sql();
  688. init_sql((char *)argv[1]+1);
  689. pthread_mutex_unlock(&odbc_mutex);
  690. conf.logfunc = logsql;
  691. }
  692. #endif
  693. else {
  694. FILE *fp, *fp1;
  695. if(argc > 2) {
  696. conf.logtype = getrotate(*argv[2]);
  697. }
  698. conf.logtime = time(0);
  699. if(conf.logname)myfree(conf.logname);
  700. conf.logname = (unsigned char *)mystrdup((char *)argv[1]);
  701. fp = fopen((char *)dologname (tmpbuf, conf.logname, NULL, conf.logtype, conf.logtime), "a");
  702. if(!fp){
  703. perror("fopen()");
  704. return 1;
  705. }
  706. else {
  707. fp1 = conf.stdlog;
  708. conf.stdlog = fp;
  709. if(fp1) fclose(fp1);
  710. #ifdef _WINCE
  711. freopen(tmpbuf, "w", stdout);
  712. freopen(tmpbuf, "w", stderr);
  713. #endif
  714. }
  715. }
  716. }
  717. return 0;
  718. }
  719. static int h_service(int argc, unsigned char **argv){
  720. #ifdef _WIN32
  721. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT) service = 1;
  722. else {
  723. if(!conf.demon)daemonize();
  724. conf.demon = 1;
  725. }
  726. #endif
  727. return 0;
  728. }
  729. static int h_daemon(int argc, unsigned char **argv){
  730. if(!conf.demon)daemonize();
  731. conf.demon = 1;
  732. return 0;
  733. }
  734. static int h_config(int argc, unsigned char **argv){
  735. if(conf.conffile)myfree(conf.conffile);
  736. conf.conffile = mystrdup((char *)argv[1]);
  737. return 0;
  738. }
  739. static int h_include(int argc, unsigned char **argv){
  740. int res;
  741. FILE *fp1;
  742. fp1 = fopen((char *)argv[1], "r");
  743. if(!fp1){
  744. fprintf(stderr, "Unable to open included file: %s\n", argv[1]);
  745. return 1;
  746. }
  747. res = readconfig(fp1);
  748. fclose(fp1);
  749. return res;
  750. }
  751. static int h_archiver(int argc, unsigned char **argv){
  752. int j;
  753. conf.archiver = myalloc(argc * sizeof(char *));
  754. if(conf.archiver) {
  755. conf.archiverc = argc;
  756. for(j = 0; j < conf.archiverc; j++) conf.archiver[j] = (unsigned char *)mystrdup((char *)argv[j]);
  757. }
  758. return 0;
  759. }
  760. static int h_counter(int argc, unsigned char **argv){
  761. struct counter_header ch1;
  762. if(conf.counterd >=0)close(conf.counterd);
  763. if(!conf.trafcountfunc) conf.trafcountfunc = trafcountfunc;
  764. conf.counterd = open((char *)argv[1], O_BINARY|O_RDWR|O_CREAT, 0660);
  765. if(conf.counterd<0){
  766. fprintf(stderr, "Unable to open counter file %s, line %d\n", argv[1], linenum);
  767. return 1;
  768. }
  769. if(read(conf.counterd, &ch1, sizeof(ch1))==sizeof(ch1)){
  770. if(memcmp(&ch1, &cheader, 4)){
  771. fprintf(stderr, "Not a counter file %s, line %d\n", argv[1], linenum);
  772. return 2;
  773. }
  774. #ifdef _MSC_VER
  775. #ifdef _TIME64_T_DEFINED
  776. #ifndef _MAX__TIME64_T
  777. #define _MAX__TIME64_T 0x793406fffi64
  778. #endif
  779. #endif
  780. if(ch1.updated >= _MAX__TIME64_T){
  781. fprintf(stderr, "Invalid or corrupted counter file %s. Use countersutil utility to convert from older version\n", argv[1]);
  782. return 3;
  783. }
  784. #endif
  785. cheader.updated = ch1.updated;
  786. }
  787. if(argc >=4) {
  788. conf.countertype = getrotate(*argv[2]);
  789. if(conf.counterfile) myfree(conf.counterfile);
  790. conf.counterfile = mystrdup((char *)argv[3]);
  791. }
  792. return 0;
  793. }
  794. static int h_rotate(int argc, unsigned char **argv){
  795. conf.rotate = atoi((char *)argv[1]);
  796. return 0;
  797. }
  798. static int h_logformat(int argc, unsigned char **argv){
  799. if(conf.logformat) myfree(conf.logformat);
  800. conf.logformat = (unsigned char *)mystrdup((char *)argv[1]);
  801. return 0;
  802. }
  803. static int h_timeouts(int argc, unsigned char **argv){
  804. int j;
  805. for(j = 0; conf.timeouts[j] && j + 1 < argc; j++) {
  806. if((conf.timeouts[j] = atoi((char *)argv[j + 1])) <= 0 || conf.timeouts[j] > 2000000){
  807. fprintf(stderr, "Invalid timeout: %s, line %d\n", argv[j + 1], linenum);
  808. return(1);
  809. }
  810. }
  811. return 0;
  812. }
  813. static int h_noop(int argc, unsigned char **argv){
  814. return 0;
  815. }
  816. static int h_auth(int argc, unsigned char **argv){
  817. struct auth *au, * newau;
  818. freeauth(conf.authfuncs);
  819. conf.authfuncs = NULL;
  820. if(!conf.bandlimfunc)conf.bandlimfunc = bandlimitfunc;
  821. for(argc--; argc; argc--){
  822. for(au = authfuncs; au; au=au->next){
  823. if(!strcmp((char *)argv[argc], au->desc)){
  824. newau = myalloc(sizeof(struct auth));
  825. newau->next = conf.authfuncs;
  826. conf.authfuncs = newau;
  827. conf.authfuncs->desc = au->desc;
  828. conf.authfuncs->authenticate = au->authenticate;
  829. conf.authfuncs->authorize = au->authorize;
  830. break;
  831. }
  832. }
  833. if(!au) return 1;
  834. }
  835. conf.authfunc = doauth;
  836. return 0;
  837. }
  838. static int h_users(int argc, unsigned char **argv){
  839. int j;
  840. unsigned char *arg;
  841. struct passwords *pwl = NULL;
  842. for (j = 1; j<argc; j++) {
  843. if(!(pwl = myalloc(sizeof(struct passwords)))) {
  844. fprintf(stderr, "No memory for PWL entry, line %d\n", linenum);
  845. return(1);
  846. }
  847. memset(pwl, 0, sizeof(struct passwords));
  848. pwl->next = conf.pwl;
  849. conf.pwl = pwl;
  850. arg = (unsigned char *)strchr((char *)argv[j], ':');
  851. if(!arg||!arg[1]||!arg[2]||arg[3]!=':') {
  852. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  853. pwl->pwtype = SYS;
  854. continue;
  855. }
  856. *arg = 0;
  857. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  858. if((arg[1] == 'C' && arg[2] == 'L' && (pwl->pwtype = CL)) ||
  859. (arg[1] == 'C' && arg[2] == 'R' && (pwl->pwtype = CR)) ||
  860. (arg[1] == 'N' && arg[2] == 'T' && (pwl->pwtype = NT)) ||
  861. (arg[1] == 'L' && arg[2] == 'M' && (pwl->pwtype = LM))){
  862. pwl->password = (unsigned char *)mystrdup((char *)arg+4);
  863. }
  864. else {
  865. pwl->password = (unsigned char *) mystrdup((char *)arg + 1);
  866. pwl->pwtype = UN;
  867. }
  868. }
  869. return 0;
  870. }
  871. static int h_maxconn(int argc, unsigned char **argv){
  872. conf.maxchild = atoi((char *)argv[1]);
  873. if(!conf.maxchild) {
  874. return(1);
  875. }
  876. #ifndef _WIN32
  877. {
  878. struct rlimit rl;
  879. if(!getrlimit(RLIMIT_NOFILE, &rl)){
  880. if((conf.maxchild<<1) > rl.rlim_cur)
  881. fprintf(stderr, "[line %d] Warning: current open file ulimits are too low (cur: %d/max: %d),"
  882. " maxconn requires at least %d for every running service."
  883. " Configure ulimits according to system documentation\n",
  884. linenum, (int)rl.rlim_cur, (int)rl.rlim_max, (conf.maxchild<<1));
  885. }
  886. }
  887. #endif
  888. return 0;
  889. }
  890. static int h_flush(int argc, unsigned char **argv){
  891. freeacl(conf.acl);
  892. conf.acl = NULL;
  893. return 0;
  894. }
  895. /*
  896. static int h_flushusers(int argc, unsigned char **argv){
  897. freepwl(conf.pwl);
  898. conf.pwl = NULL;
  899. return 0;
  900. }
  901. */
  902. static int h_nserver(int argc, unsigned char **argv){
  903. char *str;
  904. if(numservers < MAXNSERVERS) {
  905. if((str = strchr((char *)argv[1], '/')))
  906. *str = 0;
  907. if(!getip46(46, argv[1], (struct sockaddr *)&nservers[numservers].addr)) return 1;
  908. *SAPORT(&nservers[numservers].addr) = htons(53);
  909. if(str) {
  910. nservers[numservers].usetcp = strstr(str + 1, "tcp")? 1:0;
  911. *str = '/';
  912. }
  913. numservers++;
  914. }
  915. resolvfunc = myresolver;
  916. return 0;
  917. }
  918. static int h_authnserver(int argc, unsigned char **argv){
  919. authnserver = getip(argv[1]);
  920. return 0;
  921. }
  922. static int h_fakeresolve(int argc, unsigned char **argv){
  923. resolvfunc = fakeresolver;
  924. return 0;
  925. }
  926. static int h_nscache(int argc, unsigned char **argv){
  927. int res;
  928. res = atoi((char *)argv[1]);
  929. if(res < 256) {
  930. fprintf(stderr, "Invalid NS cache size: %d\n", res);
  931. return 1;
  932. }
  933. if(initdnshashtable((unsigned)res)){
  934. fprintf(stderr, "Failed to initialize NS cache\n");
  935. return 2;
  936. }
  937. return 0;
  938. }
  939. static int h_nsrecord(int argc, unsigned char **argv){
  940. hashadd(&dns_table, argv[1], getip(argv[2]), (time_t)0xffffffff);
  941. return 0;
  942. }
  943. static int h_dialer(int argc, unsigned char **argv){
  944. if(conf.demanddialprog) myfree(conf.demanddialprog);
  945. conf.demanddialprog = mystrdup((char *)argv[1]);
  946. return 0;
  947. }
  948. static int h_system(int argc, unsigned char **argv){
  949. int res;
  950. if((res = system((char *)argv[1])) == -1){
  951. fprintf(stderr, "Failed to start %s\n", argv[1]);
  952. return(1);
  953. }
  954. return 0;
  955. }
  956. static int h_pidfile(int argc, unsigned char **argv){
  957. FILE *pidf;
  958. if(!(pidf = fopen((char *)argv[1], "w"))){
  959. fprintf(stderr, "Failed to open pid file %s\n", argv[1]);
  960. return(1);
  961. }
  962. fprintf(pidf,"%u", (unsigned)getpid());
  963. fclose(pidf);
  964. return 0;
  965. }
  966. static int h_monitor(int argc, unsigned char **argv){
  967. struct filemon * fm;
  968. fm = myalloc(sizeof (struct filemon));
  969. if(stat((char *)argv[1], &fm->sb)){
  970. myfree(fm);
  971. fprintf(stderr, "Warning: file %s doesn't exist on line %d\n", argv[1], linenum);
  972. }
  973. else {
  974. fm->path = mystrdup((char *)argv[1]);
  975. fm->next = conf.fmon;
  976. conf.fmon = fm;
  977. }
  978. return 0;
  979. }
  980. static int h_parent(int argc, unsigned char **argv){
  981. struct ace *acl = NULL;
  982. struct chain *chains;
  983. acl = conf.acl;
  984. while(acl && acl->next) acl = acl->next;
  985. if(!acl || (acl->action && acl->action != 2)) {
  986. fprintf(stderr, "Chaining error: last ACL entry was not \"allow\" or \"redirect\" on line %d\n", linenum);
  987. return(1);
  988. }
  989. acl->action = 2;
  990. chains = NULL;
  991. if(!acl->chains) {
  992. chains = acl->chains = myalloc(sizeof(struct chain));
  993. }
  994. else {
  995. chains = acl->chains;
  996. while(chains->next)chains = chains->next;
  997. chains->next = myalloc(sizeof(struct chain));
  998. chains = chains->next;
  999. }
  1000. memset(chains, 0, sizeof(struct chain));
  1001. if(!chains){
  1002. fprintf(stderr, "Chainig error: unable to allocate memory for chain\n");
  1003. return(2);
  1004. }
  1005. chains->weight = (unsigned)atoi((char *)argv[1]);
  1006. if(chains->weight == 0 || chains->weight >1000) {
  1007. fprintf(stderr, "Chaining error: bad chain weight %u line %d\n", chains->weight, linenum);
  1008. return(3);
  1009. }
  1010. if(!strcmp((char *)argv[2], "tcp"))chains->type = R_TCP;
  1011. else if(!strcmp((char *)argv[2], "http"))chains->type = R_HTTP;
  1012. else if(!strcmp((char *)argv[2], "connect"))chains->type = R_CONNECT;
  1013. else if(!strcmp((char *)argv[2], "socks4"))chains->type = R_SOCKS4;
  1014. else if(!strcmp((char *)argv[2], "socks5"))chains->type = R_SOCKS5;
  1015. else if(!strcmp((char *)argv[2], "connect+"))chains->type = R_CONNECTP;
  1016. else if(!strcmp((char *)argv[2], "socks4+"))chains->type = R_SOCKS4P;
  1017. else if(!strcmp((char *)argv[2], "socks5+"))chains->type = R_SOCKS5P;
  1018. else if(!strcmp((char *)argv[2], "socks4b"))chains->type = R_SOCKS4B;
  1019. else if(!strcmp((char *)argv[2], "socks5b"))chains->type = R_SOCKS5B;
  1020. else if(!strcmp((char *)argv[2], "pop3"))chains->type = R_POP3;
  1021. else if(!strcmp((char *)argv[2], "ftp"))chains->type = R_FTP;
  1022. else if(!strcmp((char *)argv[2], "admin"))chains->type = R_ADMIN;
  1023. else if(!strcmp((char *)argv[2], "icq"))chains->type = R_ICQ;
  1024. else if(!strcmp((char *)argv[2], "msn"))chains->type = R_MSN;
  1025. else {
  1026. fprintf(stderr, "Chaining error: bad chain type (%s)\n", argv[2]);
  1027. return(4);
  1028. }
  1029. chains->redirip = getip(argv[3]);
  1030. chains->redirport = htons((unsigned short)atoi((char *)argv[4]));
  1031. if(argc > 5) chains->extuser = (unsigned char *)mystrdup((char *)argv[5]);
  1032. if(argc > 6) chains->extpass = (unsigned char *)mystrdup((char *)argv[6]);
  1033. return 0;
  1034. }
  1035. static int h_nolog(int argc, unsigned char **argv){
  1036. struct ace *acl = NULL;
  1037. acl = conf.acl;
  1038. if(!acl) {
  1039. fprintf(stderr, "Chaining error: last ACL entry was not \"allow/deny\" on line %d\n", linenum);
  1040. return(1);
  1041. }
  1042. while(acl->next) acl = acl->next;
  1043. if(!strcmp(argv[0],"nolog")) acl->nolog = 1;
  1044. else acl->weight = atoi((char*)argv[1]);
  1045. return 0;
  1046. }
  1047. int scanipl(unsigned char *arg, struct iplist *dst){
  1048. #ifndef NOIPV6
  1049. struct sockaddr_in6 sa;
  1050. #else
  1051. struct sockaddr_in sa;
  1052. #endif
  1053. char * slash, *dash;
  1054. int masklen, addrlen;
  1055. if((slash = strchr(arg, '/'))) *slash = 0;
  1056. if((dash = strchr(arg,'-'))) *dash = 0;
  1057. if(!getip46(46, arg, (struct sockaddr *)&sa)) return 1;
  1058. memcpy(&dst->ip_from, SAADDR(&sa), SAADDRLEN(&sa));
  1059. dst->family = *SAFAMILY(&sa);
  1060. if(dash){
  1061. if(!getip46(46, dash+1, (struct sockaddr *)&sa)) return 2;
  1062. memcpy(&dst->ip_to, SAADDR(&sa), SAADDRLEN(&sa));
  1063. return 0;
  1064. }
  1065. if(slash){
  1066. addrlen = SAADDRLEN(&sa);
  1067. masklen = atoi(slash+1);
  1068. if(masklen >= 0 && masklen<(addrlen*8)){
  1069. int i, nbytes = masklen / 8, nbits = masklen % 8;
  1070. for(i = addrlen; i; i--){
  1071. ((unsigned char *)&dst->ip_from)[i-1] = 0x00;
  1072. ((unsigned char *)&dst->ip_to)[i-1] = 0xff;
  1073. }
  1074. memcpy(&dst->ip_to, &dst->ip_from, addrlen - i);
  1075. for(;nbits;nbits--){
  1076. ((unsigned char *)&dst->ip_from)[i-1] &= (0x01<<(nbits-1));
  1077. ((unsigned char *)&dst->ip_to)[i-1] |= (0x01<<(nbits-1));
  1078. }
  1079. return 0;
  1080. }
  1081. }
  1082. memcpy(&dst->ip_to, &dst->ip_from, SAADDRLEN(&sa));
  1083. return 0;
  1084. }
  1085. struct ace * make_ace (int argc, unsigned char ** argv){
  1086. struct ace * acl;
  1087. unsigned char *arg;
  1088. struct iplist *ipl=NULL;
  1089. struct portlist *portl=NULL;
  1090. struct userlist *userl=NULL;
  1091. struct hostname *hostnamel=NULL;
  1092. int res;
  1093. acl = myalloc(sizeof(struct ace));
  1094. if(!acl) return acl;
  1095. memset(acl, 0, sizeof(struct ace));
  1096. if(argc > 0 && strcmp("*", (char *)argv[0])) {
  1097. arg = argv[0];
  1098. arg = (unsigned char *)strtok((char *)arg, ",");
  1099. do {
  1100. if(!acl->users) {
  1101. acl->users = userl = myalloc(sizeof(struct userlist));
  1102. }
  1103. else {
  1104. userl->next = myalloc(sizeof(struct userlist));
  1105. userl = userl -> next;
  1106. }
  1107. if(!userl) {
  1108. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1109. return(NULL);
  1110. }
  1111. memset(userl, 0, sizeof(struct userlist));
  1112. userl->user=(unsigned char*)mystrdup((char *)arg);
  1113. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1114. }
  1115. if(argc > 1 && strcmp("*", (char *)argv[1])) {
  1116. arg = (unsigned char *)strtok((char *)argv[1], ",");
  1117. do {
  1118. if(!acl->src) {
  1119. acl->src = ipl = myalloc(sizeof(struct iplist));
  1120. }
  1121. else {
  1122. ipl->next = myalloc(sizeof(struct iplist));
  1123. ipl = ipl -> next;
  1124. }
  1125. if(!ipl) {
  1126. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1127. return(NULL);
  1128. }
  1129. memset(ipl, 0, sizeof(struct iplist));
  1130. if (scanipl(arg, ipl)) {
  1131. fprintf(stderr, "Invalid IP or CIDR, line %d\n", linenum);
  1132. return(NULL);
  1133. }
  1134. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1135. }
  1136. if(argc > 2 && strcmp("*", (char *)argv[2])) {
  1137. arg = (unsigned char *)strtok((char *)argv[2], ",");
  1138. do {
  1139. int arglen;
  1140. unsigned char *pattern;
  1141. arglen = (int)strlen((char *)arg);
  1142. if(arglen > 0 && (arg[arglen-1] < '0' || arg[arglen-1] > '9')){
  1143. if(!acl->dstnames) {
  1144. acl->dstnames = hostnamel = myalloc(sizeof(struct hostname));
  1145. }
  1146. else {
  1147. hostnamel->next = myalloc(sizeof(struct hostname));
  1148. hostnamel = hostnamel -> next;
  1149. }
  1150. if(!hostnamel){
  1151. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1152. return(NULL);
  1153. }
  1154. memset(hostnamel, 0, sizeof(struct hostname));
  1155. hostnamel->matchtype = 3;
  1156. pattern = arg;
  1157. if(pattern[arglen-1] == '*'){
  1158. arglen --;
  1159. pattern[arglen] = 0;
  1160. hostnamel->matchtype ^= MATCHEND;
  1161. }
  1162. if(pattern[0] == '*'){
  1163. pattern++;
  1164. arglen--;
  1165. hostnamel->matchtype ^= MATCHBEGIN;
  1166. }
  1167. hostnamel->name = (unsigned char *) mystrdup( (char *)pattern);
  1168. if(!hostnamel->name) {
  1169. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1170. return(NULL);
  1171. }
  1172. }
  1173. else {
  1174. if(!acl->dst) {
  1175. acl->dst = ipl = myalloc(sizeof(struct iplist));
  1176. }
  1177. else {
  1178. ipl->next = myalloc(sizeof(struct iplist));
  1179. ipl = ipl -> next;
  1180. }
  1181. if(!ipl) {
  1182. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1183. return(NULL);
  1184. }
  1185. memset(ipl, 0, sizeof(struct iplist));
  1186. if (scanipl(arg, ipl)) {
  1187. fprintf(stderr, "Invalid IP or CIDR, line %d\n", linenum);
  1188. return(NULL);
  1189. }
  1190. }
  1191. }while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1192. }
  1193. if(argc > 3 && strcmp("*", (char *)argv[3])) {
  1194. arg = (unsigned char *)strtok((char *)argv[3], ",");
  1195. do {
  1196. if(!acl->ports) {
  1197. acl->ports = portl = myalloc(sizeof(struct portlist));
  1198. }
  1199. else {
  1200. portl->next = myalloc(sizeof(struct portlist));
  1201. portl = portl -> next;
  1202. }
  1203. if(!portl) {
  1204. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1205. return(NULL);
  1206. }
  1207. memset(portl, 0, sizeof(struct portlist));
  1208. res = sscanf((char *)arg, "%hu-%hu", &portl->startport, &portl->endport);
  1209. if(res < 1) {
  1210. fprintf(stderr, "Invalid port or port range, line %d\n", linenum);
  1211. return(NULL);
  1212. }
  1213. if (res == 1) portl->endport = portl->startport;
  1214. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1215. }
  1216. if(argc > 4 && strcmp("*", (char *)argv[4])) {
  1217. arg = (unsigned char *)strtok((char *)argv[4], ",");
  1218. do {
  1219. if(!strcmp((char *)arg, "CONNECT")){
  1220. acl->operation |= CONNECT;
  1221. }
  1222. else if(!strcmp((char *)arg, "BIND")){
  1223. acl->operation |= BIND;
  1224. }
  1225. else if(!strcmp((char *)arg, "UDPASSOC")){
  1226. acl->operation |= UDPASSOC;
  1227. }
  1228. else if(!strcmp((char *)arg, "ICMPASSOC")){
  1229. acl->operation |= ICMPASSOC;
  1230. }
  1231. else if(!strcmp((char *)arg, "HTTP_GET")){
  1232. acl->operation |= HTTP_GET;
  1233. }
  1234. else if(!strcmp((char *)arg, "HTTP_PUT")){
  1235. acl->operation |= HTTP_PUT;
  1236. }
  1237. else if(!strcmp((char *)arg, "HTTP_POST")){
  1238. acl->operation |= HTTP_POST;
  1239. }
  1240. else if(!strcmp((char *)arg, "HTTP_HEAD")){
  1241. acl->operation |= HTTP_HEAD;
  1242. }
  1243. else if(!strcmp((char *)arg, "HTTP_OTHER")){
  1244. acl->operation |= HTTP_OTHER;
  1245. }
  1246. else if(!strcmp((char *)arg, "HTTP_CONNECT")){
  1247. acl->operation |= HTTP_CONNECT;
  1248. }
  1249. else if(!strcmp((char *)arg, "HTTP")){
  1250. acl->operation |= HTTP;
  1251. }
  1252. else if(!strcmp((char *)arg, "HTTPS")){
  1253. acl->operation |= HTTPS;
  1254. }
  1255. else if(!strcmp((char *)arg, "FTP_GET")){
  1256. acl->operation |= FTP_GET;
  1257. }
  1258. else if(!strcmp((char *)arg, "FTP_PUT")){
  1259. acl->operation |= FTP_PUT;
  1260. }
  1261. else if(!strcmp((char *)arg, "FTP_LIST")){
  1262. acl->operation |= FTP_LIST;
  1263. }
  1264. else if(!strcmp((char *)arg, "FTP_DATA")){
  1265. acl->operation |= FTP_DATA;
  1266. }
  1267. else if(!strcmp((char *)arg, "FTP")){
  1268. acl->operation |= FTP;
  1269. }
  1270. else if(!strcmp((char *)arg, "ADMIN")){
  1271. acl->operation |= ADMIN;
  1272. }
  1273. else if(!strcmp((char *)arg, "DNSRESOLVE")){
  1274. acl->operation |= DNSRESOLVE;
  1275. }
  1276. else if(!strcmp((char *)arg, "ICQ")){
  1277. acl->operation |= IM_ICQ;
  1278. }
  1279. else {
  1280. fprintf(stderr, "Unknown operation type: %s line %d\n", arg, linenum);
  1281. return(NULL);
  1282. }
  1283. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1284. }
  1285. if(argc > 5){
  1286. for(arg = argv[5]; *arg;){
  1287. int val, val1;
  1288. if(!isnumber(*arg)){
  1289. arg++;
  1290. continue;
  1291. }
  1292. val1 = val = (*arg - '0');
  1293. arg++;
  1294. if(*arg == '-' && isnumber(*(arg+1)) && (*(arg+1) - '0') > val) {
  1295. val1 = (*(arg+1) - '0');
  1296. arg+=2;
  1297. }
  1298. for(; val<=val1; val++) acl->wdays |= (1 << (val % 7));
  1299. }
  1300. }
  1301. if(argc > 6){
  1302. for(arg = argv[6]; strlen((char *)arg) >= 17 &&
  1303. isdigit(arg[0]) &&
  1304. isdigit(arg[1]) &&
  1305. isdigit(arg[3]) &&
  1306. isdigit(arg[4]) &&
  1307. isdigit(arg[6]) &&
  1308. isdigit(arg[7]) &&
  1309. isdigit(arg[9]) &&
  1310. isdigit(arg[10]) &&
  1311. isdigit(arg[12]) &&
  1312. isdigit(arg[13]) &&
  1313. isdigit(arg[15]) &&
  1314. isdigit(arg[16])
  1315. ; arg+=18){
  1316. int t1, t2;
  1317. struct period *sp;
  1318. t1 = (arg[0] - '0') * 10 + (arg[1] - '0');
  1319. t1 = (t1 * 60) + (arg[3] - '0') * 10 + (arg[4] - '0');
  1320. t1 = (t1 * 60) + (arg[6] - '0') * 10 + (arg[7] - '0');
  1321. t2 = (arg[9] - '0') * 10 + (arg[10] - '0');
  1322. t2 = (t2 * 60) + (arg[12] - '0') * 10 + (arg[13] - '0');
  1323. t2 = (t2 * 60) + (arg[15] - '0') * 10 + (arg[16] - '0');
  1324. if(t2 < t1) break;
  1325. sp = myalloc(sizeof(struct period));
  1326. if(sp){
  1327. sp->fromtime = t1;
  1328. sp->totime = t2;
  1329. sp->next = acl->periods;
  1330. acl->periods = sp;
  1331. }
  1332. if(arg[17]!=',') break;
  1333. }
  1334. }
  1335. if (argc > 7){
  1336. acl->weight = atoi((char *)argv[7]);
  1337. }
  1338. return acl;
  1339. }
  1340. static int h_ace(int argc, unsigned char **argv){
  1341. int res = 0;
  1342. int offset = 0;
  1343. struct ace *acl = NULL;
  1344. struct bandlim * nbl;
  1345. struct trafcount * tl;
  1346. if(!strcmp((char *)argv[0], "allow")){
  1347. res = ALLOW;
  1348. }
  1349. else if(!strcmp((char *)argv[0], "deny")){
  1350. res = DENY;
  1351. }
  1352. else if(!strcmp((char *)argv[0], "redirect")){
  1353. res = REDIRECT;
  1354. offset = 2;
  1355. }
  1356. else if(!strcmp((char *)argv[0], "bandlimin")||!strcmp((char *)argv[0], "bandlimout")){
  1357. res = BANDLIM;
  1358. offset = 1;
  1359. }
  1360. else if(!strcmp((char *)argv[0], "nobandlimin")||!strcmp((char *)argv[0], "nobandlimout")){
  1361. res = NOBANDLIM;
  1362. }
  1363. else if(!strcmp((char *)argv[0], "countin")){
  1364. res = COUNTIN;
  1365. offset = 3;
  1366. }
  1367. else if(!strcmp((char *)argv[0], "nocountin")){
  1368. res = NOCOUNTIN;
  1369. }
  1370. else if(!strcmp((char *)argv[0], "countout")){
  1371. res = COUNTOUT;
  1372. offset = 3;
  1373. }
  1374. else if(!strcmp((char *)argv[0], "nocountout")){
  1375. res = NOCOUNTOUT;
  1376. }
  1377. acl = make_ace(argc - (offset+1), argv + (offset + 1));
  1378. if(!acl) {
  1379. fprintf(stderr, "Unable to parse ACL entry, line %d\n", linenum);
  1380. return(1);
  1381. }
  1382. acl->action = res;
  1383. switch(acl->action){
  1384. case REDIRECT:
  1385. acl->chains = myalloc(sizeof(struct chain));
  1386. if(!acl->chains) {
  1387. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1388. return(2);
  1389. }
  1390. acl->chains->type = R_HTTP;
  1391. acl->chains->redirip = getip(argv[1]);
  1392. acl->chains->redirport = htons((unsigned short)atoi((char *)argv[2]));
  1393. acl->chains->weight = 1000;
  1394. acl->chains->extuser = NULL;
  1395. acl->chains->extpass = NULL;
  1396. acl->chains->next = NULL;
  1397. case ALLOW:
  1398. case DENY:
  1399. if(!conf.acl){
  1400. conf.acl = acl;
  1401. }
  1402. else {
  1403. struct ace * acei;
  1404. for(acei = conf.acl; acei->next; acei = acei->next);
  1405. acei->next = acl;
  1406. }
  1407. break;
  1408. case BANDLIM:
  1409. case NOBANDLIM:
  1410. nbl = myalloc(sizeof(struct bandlim));
  1411. if(!nbl) {
  1412. fprintf(stderr, "No memory to create band limit filter\n");
  1413. return(3);
  1414. }
  1415. memset(nbl, 0, sizeof(struct bandlim));
  1416. nbl->ace = acl;
  1417. if(acl->action == BANDLIM) {
  1418. sscanf((char *)argv[1], "%u", &nbl->rate);
  1419. if(nbl->rate < 300) {
  1420. fprintf(stderr, "Wrong bandwidth specified, line %d\n", linenum);
  1421. return(4);
  1422. }
  1423. }
  1424. pthread_mutex_lock(&bandlim_mutex);
  1425. if(!strcmp((char *)argv[0], "bandlimin") || !strcmp((char *)argv[0], "nobandlimin")){
  1426. if(!conf.bandlimiter){
  1427. conf.bandlimiter = nbl;
  1428. }
  1429. else {
  1430. struct bandlim * bli;
  1431. for(bli = conf.bandlimiter; bli->next; bli = bli->next);
  1432. bli->next = nbl;
  1433. }
  1434. }
  1435. else {
  1436. if(!conf.bandlimiterout){
  1437. conf.bandlimiterout = nbl;
  1438. }
  1439. else {
  1440. struct bandlim * bli;
  1441. for(bli = conf.bandlimiterout; bli->next; bli = bli->next);
  1442. bli->next = nbl;
  1443. }
  1444. }
  1445. pthread_mutex_unlock(&bandlim_mutex);
  1446. break;
  1447. case COUNTIN:
  1448. case NOCOUNTIN:
  1449. case COUNTOUT:
  1450. case NOCOUNTOUT:
  1451. tl = myalloc(sizeof(struct trafcount));
  1452. if(!tl) {
  1453. fprintf(stderr, "No memory to create traffic limit filter\n");
  1454. return(5);
  1455. }
  1456. memset(tl, 0, sizeof(struct trafcount));
  1457. tl->ace = acl;
  1458. if((acl->action == COUNTIN)||(acl->action == COUNTOUT)) {
  1459. unsigned long lim;
  1460. tl->comment = ( char *)argv[1];
  1461. while(isdigit(*tl->comment))tl->comment++;
  1462. if(*tl->comment== '/')tl->comment++;
  1463. tl->comment = mystrdup(tl->comment);
  1464. sscanf((char *)argv[1], "%u", &tl->number);
  1465. sscanf((char *)argv[3], "%lu", &lim);
  1466. tl->type = getrotate(*argv[2]);
  1467. tl->traflim64 = ((uint64_t)lim)*(1024*1024);
  1468. if(!tl->traflim64) {
  1469. fprintf(stderr, "Wrong traffic limit specified, line %d\n", linenum);
  1470. return(6);
  1471. }
  1472. if(tl->number != 0 && conf.counterd >= 0) {
  1473. lseek(conf.counterd,
  1474. sizeof(struct counter_header) + (tl->number - 1) * sizeof(struct counter_record),
  1475. SEEK_SET);
  1476. memset(&crecord, 0, sizeof(struct counter_record));
  1477. read(conf.counterd, &crecord, sizeof(struct counter_record));
  1478. tl->traf64 = crecord.traf64;
  1479. tl->cleared = crecord.cleared;
  1480. tl->updated = crecord.updated;
  1481. #ifdef _MAX__TIME64_T
  1482. if(tl->cleared >= _MAX__TIME64_T || tl->updated >= _MAX__TIME64_T){
  1483. fprintf(stderr, "Invalid or corrupted counter file. Use countersutil utility to convert from older version\n");
  1484. return(6);
  1485. }
  1486. #endif
  1487. }
  1488. }
  1489. pthread_mutex_lock(&tc_mutex);
  1490. if(!conf.trafcounter){
  1491. conf.trafcounter = tl;
  1492. }
  1493. else {
  1494. struct trafcount * ntl;
  1495. for(ntl = conf.trafcounter; ntl->next; ntl = ntl->next);
  1496. ntl->next = tl;
  1497. }
  1498. pthread_mutex_unlock(&tc_mutex);
  1499. }
  1500. return 0;
  1501. }
  1502. static int h_logdump(int argc, unsigned char **argv){
  1503. conf.logdumpsrv = (unsigned) atoi((char *) *(argv + 1));
  1504. if(argc > 2) conf.logdumpcli = (unsigned) atoi((char *) *(argv + 2));
  1505. return 0;
  1506. }
  1507. static int h_filtermaxsize(int argc, unsigned char **argv){
  1508. conf.filtermaxsize = atoi((char *) *(argv + 1));
  1509. return 0;
  1510. }
  1511. static int h_delimchar(int argc, unsigned char **argv){
  1512. conf.delimchar = *argv[1];
  1513. return 0;
  1514. }
  1515. static int h_authcache(int argc, unsigned char **argv){
  1516. conf.authcachetype = 0;
  1517. if(strstr((char *) *(argv + 1), "ip")) conf.authcachetype |= 1;
  1518. if(strstr((char *) *(argv + 1), "user")) conf.authcachetype |= 2;
  1519. if(strstr((char *) *(argv + 1), "pass")) conf.authcachetype |= 4;
  1520. if(argc > 2) conf.authcachetime = (unsigned) atoi((char *) *(argv + 2));
  1521. if(!conf.authcachetype) conf.authcachetype = 6;
  1522. if(!conf.authcachetime) conf.authcachetime = 600;
  1523. return 0;
  1524. }
  1525. static int h_plugin(int argc, unsigned char **argv){
  1526. #ifdef NOPLUGINS
  1527. return 999;
  1528. #else
  1529. #ifdef _WIN32
  1530. HINSTANCE hi;
  1531. FARPROC fp;
  1532. #ifdef _WINCE
  1533. hi = LoadLibraryW((LPCWSTR)CEToUnicode(argv[1]));
  1534. #else
  1535. hi = LoadLibrary(argv[1]);
  1536. #endif
  1537. if(!hi) {
  1538. fprintf(stderr, "Failed to load %s, code %d\n", argv[1], (int)GetLastError());
  1539. return 1;
  1540. }
  1541. #ifdef _WINCE
  1542. fp = GetProcAddressW(hi, (LPCWSTR)CEToUnicode(argv[2]));
  1543. #else
  1544. fp = GetProcAddress(hi, argv[2]);
  1545. #endif
  1546. if(!fp) {
  1547. printf("%s not found in %s, code: %d\n", argv[2], argv[1], (int)GetLastError());
  1548. return 2;
  1549. }
  1550. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1551. #else
  1552. void *hi, *fp;
  1553. hi = dlopen((char *)argv[1], RTLD_LAZY);
  1554. if(!hi) return 1;
  1555. fp = dlsym(hi, (char *)argv[2]);
  1556. if(!fp) return 2;
  1557. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1558. #endif
  1559. #endif
  1560. }
  1561. #ifndef _WIN32
  1562. static int h_setuid(int argc, unsigned char **argv){
  1563. int res;
  1564. res = atoi((char *)argv[1]);
  1565. if(!res || setuid(res)) {
  1566. fprintf(stderr, "Unable to set uid %d", res);
  1567. return(1);
  1568. }
  1569. return 0;
  1570. }
  1571. static int h_setgid(int argc, unsigned char **argv){
  1572. int res;
  1573. res = atoi((char *)argv[1]);
  1574. if(!res || setgid(res)) {
  1575. fprintf(stderr, "Unable to set gid %d", res);
  1576. return(1);
  1577. }
  1578. return 0;
  1579. }
  1580. static int h_chroot(int argc, unsigned char **argv){
  1581. if(!chrootp){
  1582. char *p;
  1583. if(chroot((char *)argv[1])) {
  1584. fprintf(stderr, "Unable to chroot %s", argv[1]);
  1585. return(1);
  1586. }
  1587. p = (char *)argv[1] + strlen((char *)argv[1]) ;
  1588. while (p > (char *)argv[1] && p[-1] == '/'){
  1589. p--;
  1590. *p = 0;
  1591. }
  1592. chrootp = mystrdup((char *)argv[1]);
  1593. }
  1594. return 0;
  1595. }
  1596. #endif
  1597. struct commands specificcommands[]={
  1598. #ifndef _WIN32
  1599. {specificcommands+1, "setuid", h_setuid, 2, 2},
  1600. {specificcommands+2, "setgid", h_setgid, 2, 2},
  1601. {specificcommands+3, "chroot", h_chroot, 2, 2},
  1602. #endif
  1603. {NULL, "", h_noop, 1, 0}
  1604. };
  1605. struct commands commandhandlers[]={
  1606. {commandhandlers+1, "", h_noop, 1, 0},
  1607. {commandhandlers+2, "proxy", h_proxy, 1, 0},
  1608. {commandhandlers+3, "pop3p", h_proxy, 1, 0},
  1609. {commandhandlers+4, "ftppr", h_proxy, 1, 0},
  1610. {commandhandlers+5, "socks", h_proxy, 1, 0},
  1611. {commandhandlers+6, "tcppm", h_proxy, 4, 0},
  1612. {commandhandlers+7, "udppm", h_proxy, 4, 0},
  1613. {commandhandlers+8, "admin", h_proxy, 1, 0},
  1614. {commandhandlers+9, "dnspr", h_proxy, 1, 0},
  1615. {commandhandlers+10, "internal", h_internal, 2, 2},
  1616. {commandhandlers+11, "external", h_external, 2, 2},
  1617. {commandhandlers+12, "log", h_log, 1, 0},
  1618. {commandhandlers+13, "service", h_service, 1, 1},
  1619. {commandhandlers+14, "daemon", h_daemon, 1, 1},
  1620. {commandhandlers+15, "config", h_config, 2, 2},
  1621. {commandhandlers+16, "include", h_include, 2, 2},
  1622. {commandhandlers+17, "archiver", h_archiver, 3, 0},
  1623. {commandhandlers+18, "counter", h_counter, 2, 4},
  1624. {commandhandlers+19, "rotate", h_rotate, 2, 2},
  1625. {commandhandlers+20, "logformat", h_logformat, 2, 2},
  1626. {commandhandlers+21, "timeouts", h_timeouts, 2, 0},
  1627. {commandhandlers+22, "auth", h_auth, 2, 0},
  1628. {commandhandlers+23, "users", h_users, 2, 0},
  1629. {commandhandlers+24, "maxconn", h_maxconn, 2, 2},
  1630. {commandhandlers+25, "flush", h_flush, 1, 1},
  1631. {commandhandlers+26, "nserver", h_nserver, 2, 2},
  1632. {commandhandlers+27, "fakeresolve", h_fakeresolve, 1, 1},
  1633. {commandhandlers+28, "nscache", h_nscache, 2, 2},
  1634. {commandhandlers+29, "nsrecord", h_nsrecord, 3, 3},
  1635. {commandhandlers+30, "dialer", h_dialer, 2, 2},
  1636. {commandhandlers+31, "system", h_system, 2, 2},
  1637. {commandhandlers+32, "pidfile", h_pidfile, 2, 2},
  1638. {commandhandlers+33, "monitor", h_monitor, 2, 2},
  1639. {commandhandlers+34, "parent", h_parent, 5, 0},
  1640. {commandhandlers+35, "allow", h_ace, 1, 0},
  1641. {commandhandlers+36, "deny", h_ace, 1, 0},
  1642. {commandhandlers+37, "redirect", h_ace, 3, 0},
  1643. {commandhandlers+38, "bandlimin", h_ace, 2, 0},
  1644. {commandhandlers+39, "bandlimout", h_ace, 2, 0},
  1645. {commandhandlers+40, "nobandlimin", h_ace, 1, 0},
  1646. {commandhandlers+41, "nobandlimout", h_ace, 1, 0},
  1647. {commandhandlers+42, "countin", h_ace, 4, 0},
  1648. {commandhandlers+43, "nocountin", h_ace, 1, 0},
  1649. {commandhandlers+44, "countout", h_ace, 4, 0},
  1650. {commandhandlers+45, "nocountout", h_ace, 1, 0},
  1651. {commandhandlers+46, "plugin", h_plugin, 3, 0},
  1652. {commandhandlers+47, "logdump", h_logdump, 2, 3},
  1653. {commandhandlers+48, "filtermaxsize", h_filtermaxsize, 2, 2},
  1654. {commandhandlers+49, "nolog", h_nolog, 1, 1},
  1655. {commandhandlers+50, "weight", h_nolog, 2, 2},
  1656. {commandhandlers+51, "authcache", h_authcache, 2, 3},
  1657. {commandhandlers+52, "smtpp", h_proxy, 1, 0},
  1658. {commandhandlers+53, "icqpr", h_proxy, 4, 0},
  1659. {commandhandlers+54, "msnpr", h_proxy, 4, 0},
  1660. {commandhandlers+55, "delimchar",h_delimchar, 2, 2},
  1661. {commandhandlers+56, "authnserver", h_authnserver, 2, 2},
  1662. {specificcommands, "", h_noop, 1, 0}
  1663. };
  1664. int readconfig(FILE * fp){
  1665. unsigned char ** argv = NULL;
  1666. unsigned char * buf = NULL;
  1667. int bufsize = STRINGBUF*2;
  1668. int inbuf = 0;
  1669. int argc;
  1670. struct commands * cm;
  1671. int res = 0;
  1672. if( !(buf = myalloc(bufsize)) || ! (argv = myalloc(NPARAMS * sizeof(unsigned char *) + 1)) ) {
  1673. fprintf(stderr, "No memory for configuration");
  1674. return(10);
  1675. }
  1676. for (linenum = 1; fgets((char *)buf, STRINGBUF, fp); linenum++){
  1677. if(!*buf || isspace(*buf) || (*buf) == '#')continue;
  1678. inbuf = (int)(strlen((char *)buf) + 1);
  1679. argc = parsestr (buf, argv, NPARAMS-1, &buf, &inbuf, &bufsize);
  1680. if(argc < 1) {
  1681. fprintf(stderr, "Parse error line %d\n", linenum);
  1682. return(21);
  1683. }
  1684. argv[argc] = NULL;
  1685. if(!strcmp((char *)argv[0], "end") && argc == 1) {
  1686. break;
  1687. }
  1688. else if(!strcmp((char *)argv[0], "writable") && argc == 1) {
  1689. if(!writable){
  1690. writable = freopen(curconf, "r+", fp);
  1691. if(!writable){
  1692. fprintf(stderr, "Unable to reopen config for writing: %s\n", curconf);
  1693. return 1;
  1694. }
  1695. }
  1696. continue;
  1697. }
  1698. res = 1;
  1699. for(cm = commandhandlers; cm; cm = cm->next){
  1700. if(!strcmp((char *)argv[0], (char *)cm->command) && argc >= cm->minargs && (!cm->maxargs || argc <= cm->maxargs)){
  1701. res = (*cm->handler)(argc, argv);
  1702. if(res > 0){
  1703. fprintf(stderr, "Command: '%s' failed with code %d, line %d\n", argv[0], res, linenum);
  1704. return(linenum);
  1705. }
  1706. if(!res) break;
  1707. }
  1708. }
  1709. if(res != 1)continue;
  1710. fprintf(stderr, "Unknown command: '%s' line %d\n", argv[0], linenum);
  1711. return(linenum);
  1712. }
  1713. myfree(buf);
  1714. myfree(argv);
  1715. return 0;
  1716. }
  1717. #ifndef _WINCE
  1718. int main(int argc, char * argv[]) {
  1719. #else
  1720. int WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow){
  1721. int argc;
  1722. char ** argv;
  1723. WNDCLASS wc;
  1724. HWND hwnd = 0;
  1725. #endif
  1726. int res = 0;
  1727. FILE * fp = NULL;
  1728. #ifdef _WIN32
  1729. unsigned char * arg;
  1730. WSADATA wd;
  1731. WSAStartup(MAKEWORD( 1, 1 ), &wd);
  1732. osv.dwOSVersionInfoSize = sizeof(osv);
  1733. GetVersionEx(&osv);
  1734. #endif
  1735. #ifdef _WINCE
  1736. argc = ceparseargs((char *)lpCmdLine);
  1737. argv = ceargv;
  1738. if(FindWindow(L"3proxy", L"3proxy")) return 0;
  1739. ZeroMemory(&wc,sizeof(wc));
  1740. wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH);
  1741. wc.hInstance=hInstance;
  1742. wc.hCursor=LoadCursor(NULL,IDC_ARROW);
  1743. wc.lpfnWndProc=DefWindowProc;
  1744. wc.style=CS_HREDRAW|CS_VREDRAW;
  1745. wc.lpszClassName=L"3proxy";
  1746. RegisterClass(&wc);
  1747. hwnd = CreateWindowEx(0,L"3proxy",L"3proxy",WS_VISIBLE|WS_POPUP,0,0,0,0,0,0,hInstance,0);
  1748. #endif
  1749. conf.stringtable = strings;
  1750. #ifdef _WIN32
  1751. #ifndef _WINCE
  1752. if((argc == 2 || argc == 3)&& !strcmp((char *)argv[1], "--install")) {
  1753. sprintf((char *)tmpbuf, "%s will be installed and started.\n"
  1754. "By clicking Yes you confirm you read and accepted License Agreement.\n"
  1755. "You can use Administration/Services to control %s service.",
  1756. conf.stringtable[1], conf.stringtable[2]);
  1757. if(MessageBox(NULL, (char *)tmpbuf, conf.stringtable[2], MB_YESNO|MB_ICONASTERISK) != IDYES) return 1;
  1758. *tmpbuf = '\"';
  1759. if (!(res = SearchPath(NULL, argv[0], ".exe", 256, (char *)tmpbuf+1, (LPTSTR*)&arg))) {
  1760. perror("Failed to find executable filename");
  1761. RETURN(102);
  1762. }
  1763. strcat((char *)tmpbuf, "\" \"");
  1764. if(!(res = GetFullPathName ((argc == 3)?argv[2]:(char*)DEFAULTCONFIG, 256, (char *)tmpbuf+res+4, (char **)&arg))){
  1765. perror("Failed to find config filename");
  1766. RETURN(103);
  1767. }
  1768. strcat((char *)tmpbuf, "\" --service");
  1769. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT){
  1770. SC_HANDLE sch;
  1771. if(!(sch = OpenSCManager(NULL, NULL, GENERIC_WRITE|SERVICE_START ))){
  1772. perror("Failed to open Service Manager");
  1773. RETURN(101);
  1774. }
  1775. if (!(sch = CreateService(sch, conf.stringtable[1], conf.stringtable[2], GENERIC_EXECUTE, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START, SERVICE_ERROR_IGNORE, (char *)tmpbuf, NULL, NULL, NULL, NULL, NULL))){
  1776. perror("Failed to create service");
  1777. RETURN(103);
  1778. }
  1779. if (!StartService(sch, 0, NULL)) {
  1780. perror("Failed to start service");
  1781. RETURN(103);
  1782. }
  1783. }
  1784. else {
  1785. HKEY runsrv;
  1786. if(RegOpenKeyEx( HKEY_LOCAL_MACHINE,
  1787. "Software\\Microsoft\\Windows\\CurrentVersion\\RunServices",
  1788. 0,
  1789. KEY_ALL_ACCESS,
  1790. &runsrv) != ERROR_SUCCESS){
  1791. perror("Failed to open registry");
  1792. RETURN(104);
  1793. }
  1794. if(RegSetValueEx( runsrv,
  1795. conf.stringtable[1],
  1796. 0,
  1797. REG_EXPAND_SZ,
  1798. (char *)tmpbuf,
  1799. (int)strlen((char *)tmpbuf)+1)!=ERROR_SUCCESS){
  1800. perror("Failed to set registry value");
  1801. RETURN(105);
  1802. }
  1803. }
  1804. return 0;
  1805. }
  1806. if((argc == 2 || argc == 3)&& !strcmp((char *)argv[1], "--remove")) {
  1807. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT){
  1808. SC_HANDLE sch;
  1809. if(!(sch = OpenSCManager(NULL, NULL, GENERIC_WRITE))){
  1810. perror("Failed to open Service Manager\n");
  1811. RETURN(106);
  1812. }
  1813. if (!(sch = OpenService(sch, conf.stringtable[1], DELETE))){
  1814. perror("Failed to open service");
  1815. RETURN(107);
  1816. }
  1817. if (!DeleteService(sch)){
  1818. perror("Failed to delete service");
  1819. RETURN(108);
  1820. }
  1821. }
  1822. else {
  1823. HKEY runsrv;
  1824. if(RegOpenKeyEx( HKEY_LOCAL_MACHINE,
  1825. "Software\\Microsoft\\Windows\\CurrentVersion\\RunServices",
  1826. 0,
  1827. KEY_ALL_ACCESS,
  1828. &runsrv) != ERROR_SUCCESS){
  1829. perror("Failed to open registry");
  1830. RETURN(109);
  1831. }
  1832. if(RegDeleteValue(runsrv, conf.stringtable[1]) != ERROR_SUCCESS){
  1833. perror("Failed to clear registry");
  1834. RETURN(110);
  1835. }
  1836. }
  1837. RETURN(0);
  1838. }
  1839. if(argc==3 && !strcmp(argv[2], "--service")){
  1840. service = 1;
  1841. argc = 2;
  1842. }
  1843. #endif
  1844. #endif
  1845. conf.conffile = mystrdup((argc==2)?argv[1]:(char*)DEFAULTCONFIG);
  1846. if(conf.conffile && *conf.conffile != '-') {
  1847. fp = confopen();
  1848. #ifndef _WIN32
  1849. if(!fp) fp = stdin;
  1850. #endif
  1851. }
  1852. if(argc > 2 || !(fp)) {
  1853. fprintf(stderr, "Usage: %s [conffile]\n", argv[0]);
  1854. #ifdef _WIN32
  1855. fprintf(stderr, "\n\t%s --install [conffile]\n\tto install as service\n"
  1856. "\n\t%s --remove\n\tto remove service\n", argv[0], argv[0]);
  1857. #else
  1858. fprintf(stderr, "\n if conffile is missing, configuration is expected from stdin\n");
  1859. #endif
  1860. fprintf(stderr, "\n%s %s\n%s\n", conf.stringtable[2], conf.stringtable[3], copyright);
  1861. return 1;
  1862. }
  1863. pthread_mutex_init(&bandlim_mutex, NULL);
  1864. pthread_mutex_init(&hash_mutex, NULL);
  1865. pthread_mutex_init(&tc_mutex, NULL);
  1866. pthread_mutex_init(&pwl_mutex, NULL);
  1867. #ifndef NOODBC
  1868. pthread_mutex_init(&odbc_mutex, NULL);
  1869. #endif
  1870. {
  1871. char * args[] = {"auth", "iponly", NULL};
  1872. h_auth(2, (unsigned char **)args);
  1873. }
  1874. res = readconfig(fp);
  1875. if(res) RETURN(res);
  1876. if(!writable)fclose(fp);
  1877. #ifdef _WIN32
  1878. #ifndef _WINCE
  1879. if(service){
  1880. SERVICE_TABLE_ENTRY ste[] =
  1881. {
  1882. { conf.stringtable[1], (LPSERVICE_MAIN_FUNCTION)ServiceMain},
  1883. { NULL, NULL }
  1884. };
  1885. if(!StartServiceCtrlDispatcher( ste ))cyclestep();
  1886. }
  1887. else
  1888. #endif
  1889. {
  1890. cyclestep();
  1891. }
  1892. #else
  1893. signal(SIGCONT, mysigpause);
  1894. signal(SIGTERM, mysigterm);
  1895. signal(SIGUSR1, mysigusr1);
  1896. signal(SIGPIPE, SIG_IGN);
  1897. cyclestep();
  1898. #endif
  1899. CLEARRETURN:
  1900. return 0;
  1901. }