3proxy.c 51 KB

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