3proxy.c 52 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093
  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. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize) || resolvfunc == fakeresolver){
  647. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, dnspr will not work as expected\n", linenum);
  648. }
  649. }
  650. return start_proxy_thread(&ch);
  651. }
  652. static int h_internal(int argc, unsigned char ** argv){
  653. getip46(46, argv[1], (struct sockaddr *)&conf.intsa);
  654. return 0;
  655. }
  656. static int h_external(int argc, unsigned char ** argv){
  657. int res;
  658. #ifndef NOIPV6
  659. struct sockaddr_in6 sa6;
  660. res = getip46(46, argv[1], (struct sockaddr *)&sa6);
  661. if(!res) return 1;
  662. memcpy((*SAFAMILY(&sa6)==AF_INET)?(void *)&conf.extsa:(void *)&conf.extsa6, &sa6, SASIZE(&sa6));
  663. #else
  664. res = getip46(46, argv[1], (struct sockaddr *)&conf.extsa);
  665. if(!res) return 1;
  666. #endif
  667. return 0;
  668. }
  669. static int h_log(int argc, unsigned char ** argv){
  670. conf.logfunc = logstdout;
  671. if(conf.logtarget){
  672. myfree(conf.logtarget);
  673. conf.logtarget = NULL;
  674. }
  675. if(argc > 1) {
  676. conf.logtarget = (unsigned char *)mystrdup((char *)argv[1]);
  677. if(*argv[1]=='@'){
  678. #ifndef _WIN32
  679. openlog((char *)conf.logtarget+1, LOG_PID, LOG_DAEMON);
  680. conf.logfunc = logsyslog;
  681. #endif
  682. }
  683. #ifndef NOODBC
  684. else if(*argv[1]=='&'){
  685. pthread_mutex_lock(&odbc_mutex);
  686. close_sql();
  687. init_sql((char *)argv[1]+1);
  688. pthread_mutex_unlock(&odbc_mutex);
  689. conf.logfunc = logsql;
  690. }
  691. #endif
  692. else {
  693. FILE *fp, *fp1;
  694. if(argc > 2) {
  695. conf.logtype = getrotate(*argv[2]);
  696. }
  697. conf.logtime = time(0);
  698. if(conf.logname)myfree(conf.logname);
  699. conf.logname = (unsigned char *)mystrdup((char *)argv[1]);
  700. fp = fopen((char *)dologname (tmpbuf, conf.logname, NULL, conf.logtype, conf.logtime), "a");
  701. if(!fp){
  702. perror("fopen()");
  703. return 1;
  704. }
  705. else {
  706. fp1 = conf.stdlog;
  707. conf.stdlog = fp;
  708. if(fp1) fclose(fp1);
  709. #ifdef _WINCE
  710. freopen(tmpbuf, "w", stdout);
  711. freopen(tmpbuf, "w", stderr);
  712. #endif
  713. }
  714. }
  715. }
  716. return 0;
  717. }
  718. static int h_service(int argc, unsigned char **argv){
  719. #ifdef _WIN32
  720. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT) service = 1;
  721. else {
  722. if(!conf.demon)daemonize();
  723. conf.demon = 1;
  724. }
  725. #endif
  726. return 0;
  727. }
  728. static int h_daemon(int argc, unsigned char **argv){
  729. if(!conf.demon)daemonize();
  730. conf.demon = 1;
  731. return 0;
  732. }
  733. static int h_config(int argc, unsigned char **argv){
  734. if(conf.conffile)myfree(conf.conffile);
  735. conf.conffile = mystrdup((char *)argv[1]);
  736. return 0;
  737. }
  738. static int h_include(int argc, unsigned char **argv){
  739. int res;
  740. FILE *fp1;
  741. fp1 = fopen((char *)argv[1], "r");
  742. if(!fp1){
  743. fprintf(stderr, "Unable to open included file: %s\n", argv[1]);
  744. return 1;
  745. }
  746. res = readconfig(fp1);
  747. fclose(fp1);
  748. return res;
  749. }
  750. static int h_archiver(int argc, unsigned char **argv){
  751. int j;
  752. conf.archiver = myalloc(argc * sizeof(char *));
  753. if(conf.archiver) {
  754. conf.archiverc = argc;
  755. for(j = 0; j < conf.archiverc; j++) conf.archiver[j] = (unsigned char *)mystrdup((char *)argv[j]);
  756. }
  757. return 0;
  758. }
  759. static int h_counter(int argc, unsigned char **argv){
  760. struct counter_header ch1;
  761. if(conf.counterd >=0)close(conf.counterd);
  762. if(!conf.trafcountfunc) conf.trafcountfunc = trafcountfunc;
  763. conf.counterd = open((char *)argv[1], O_BINARY|O_RDWR|O_CREAT, 0660);
  764. if(conf.counterd<0){
  765. fprintf(stderr, "Unable to open counter file %s, line %d\n", argv[1], linenum);
  766. return 1;
  767. }
  768. if(read(conf.counterd, &ch1, sizeof(ch1))==sizeof(ch1)){
  769. if(memcmp(&ch1, &cheader, 4)){
  770. fprintf(stderr, "Not a counter file %s, line %d\n", argv[1], linenum);
  771. return 2;
  772. }
  773. #ifdef _MSC_VER
  774. #ifdef _TIME64_T_DEFINED
  775. #ifndef _MAX__TIME64_T
  776. #define _MAX__TIME64_T 0x793406fffi64
  777. #endif
  778. #endif
  779. if(ch1.updated >= _MAX__TIME64_T){
  780. fprintf(stderr, "Invalid or corrupted counter file %s. Use countersutil utility to convert from older version\n", argv[1]);
  781. return 3;
  782. }
  783. #endif
  784. cheader.updated = ch1.updated;
  785. }
  786. if(argc >=4) {
  787. conf.countertype = getrotate(*argv[2]);
  788. if(conf.counterfile) myfree(conf.counterfile);
  789. conf.counterfile = mystrdup((char *)argv[3]);
  790. }
  791. return 0;
  792. }
  793. static int h_rotate(int argc, unsigned char **argv){
  794. conf.rotate = atoi((char *)argv[1]);
  795. return 0;
  796. }
  797. static int h_logformat(int argc, unsigned char **argv){
  798. if(conf.logformat) myfree(conf.logformat);
  799. conf.logformat = (unsigned char *)mystrdup((char *)argv[1]);
  800. return 0;
  801. }
  802. static int h_timeouts(int argc, unsigned char **argv){
  803. int j;
  804. for(j = 0; conf.timeouts[j] && j + 1 < argc; j++) {
  805. if((conf.timeouts[j] = atoi((char *)argv[j + 1])) <= 0 || conf.timeouts[j] > 2000000){
  806. fprintf(stderr, "Invalid timeout: %s, line %d\n", argv[j + 1], linenum);
  807. return(1);
  808. }
  809. }
  810. return 0;
  811. }
  812. static int h_noop(int argc, unsigned char **argv){
  813. return 0;
  814. }
  815. static int h_auth(int argc, unsigned char **argv){
  816. struct auth *au, * newau;
  817. freeauth(conf.authfuncs);
  818. conf.authfuncs = NULL;
  819. if(!conf.bandlimfunc)conf.bandlimfunc = bandlimitfunc;
  820. for(argc--; argc; argc--){
  821. for(au = authfuncs; au; au=au->next){
  822. if(!strcmp((char *)argv[argc], au->desc)){
  823. newau = myalloc(sizeof(struct auth));
  824. newau->next = conf.authfuncs;
  825. conf.authfuncs = newau;
  826. conf.authfuncs->desc = au->desc;
  827. conf.authfuncs->authenticate = au->authenticate;
  828. conf.authfuncs->authorize = au->authorize;
  829. break;
  830. }
  831. }
  832. if(!au) return 1;
  833. }
  834. conf.authfunc = doauth;
  835. return 0;
  836. }
  837. static int h_users(int argc, unsigned char **argv){
  838. int j;
  839. unsigned char *arg;
  840. struct passwords *pwl = NULL;
  841. for (j = 1; j<argc; j++) {
  842. if(!(pwl = myalloc(sizeof(struct passwords)))) {
  843. fprintf(stderr, "No memory for PWL entry, line %d\n", linenum);
  844. return(1);
  845. }
  846. memset(pwl, 0, sizeof(struct passwords));
  847. pwl->next = conf.pwl;
  848. conf.pwl = pwl;
  849. arg = (unsigned char *)strchr((char *)argv[j], ':');
  850. if(!arg||!arg[1]||!arg[2]||arg[3]!=':') {
  851. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  852. pwl->pwtype = SYS;
  853. continue;
  854. }
  855. *arg = 0;
  856. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  857. if((arg[1] == 'C' && arg[2] == 'L' && (pwl->pwtype = CL)) ||
  858. (arg[1] == 'C' && arg[2] == 'R' && (pwl->pwtype = CR)) ||
  859. (arg[1] == 'N' && arg[2] == 'T' && (pwl->pwtype = NT)) ||
  860. (arg[1] == 'L' && arg[2] == 'M' && (pwl->pwtype = LM))){
  861. pwl->password = (unsigned char *)mystrdup((char *)arg+4);
  862. }
  863. else {
  864. pwl->password = (unsigned char *) mystrdup((char *)arg + 1);
  865. pwl->pwtype = UN;
  866. }
  867. }
  868. return 0;
  869. }
  870. static int h_maxconn(int argc, unsigned char **argv){
  871. conf.maxchild = atoi((char *)argv[1]);
  872. if(!conf.maxchild) {
  873. return(1);
  874. }
  875. #ifndef _WIN32
  876. {
  877. struct rlimit rl;
  878. if(!getrlimit(RLIMIT_NOFILE, &rl)){
  879. if((conf.maxchild<<1) > rl.rlim_cur)
  880. fprintf(stderr, "[line %d] Warning: current open file ulimits are too low (cur: %d/max: %d),"
  881. " maxconn requires at least %d for every running service."
  882. " Configure ulimits according to system documentation\n",
  883. linenum, (int)rl.rlim_cur, (int)rl.rlim_max, (conf.maxchild<<1));
  884. }
  885. }
  886. #endif
  887. return 0;
  888. }
  889. static int h_flush(int argc, unsigned char **argv){
  890. freeacl(conf.acl);
  891. conf.acl = NULL;
  892. return 0;
  893. }
  894. /*
  895. static int h_flushusers(int argc, unsigned char **argv){
  896. freepwl(conf.pwl);
  897. conf.pwl = NULL;
  898. return 0;
  899. }
  900. */
  901. static int h_nserver(int argc, unsigned char **argv){
  902. int res;
  903. for(res = 0; nservers[res] && res < MAXNSERVERS; res++);
  904. if(res < MAXNSERVERS) {
  905. nservers[res] = getip(argv[1]);
  906. }
  907. resolvfunc = myresolver;
  908. return 0;
  909. }
  910. static int h_authnserver(int argc, unsigned char **argv){
  911. authnserver = getip(argv[1]);
  912. return 0;
  913. }
  914. static int h_fakeresolve(int argc, unsigned char **argv){
  915. resolvfunc = fakeresolver;
  916. return 0;
  917. }
  918. static int h_nscache(int argc, unsigned char **argv){
  919. int res;
  920. res = atoi((char *)argv[1]);
  921. if(res < 256) {
  922. fprintf(stderr, "Invalid NS cache size: %d\n", res);
  923. return 1;
  924. }
  925. if(initdnshashtable((unsigned)res)){
  926. fprintf(stderr, "Failed to initialize NS cache\n");
  927. return 2;
  928. }
  929. return 0;
  930. }
  931. static int h_nsrecord(int argc, unsigned char **argv){
  932. hashadd(&dns_table, argv[1], getip(argv[2]), (time_t)0xffffffff);
  933. return 0;
  934. }
  935. static int h_dialer(int argc, unsigned char **argv){
  936. if(conf.demanddialprog) myfree(conf.demanddialprog);
  937. conf.demanddialprog = mystrdup((char *)argv[1]);
  938. return 0;
  939. }
  940. static int h_system(int argc, unsigned char **argv){
  941. int res;
  942. if((res = system((char *)argv[1])) == -1){
  943. fprintf(stderr, "Failed to start %s\n", argv[1]);
  944. return(1);
  945. }
  946. return 0;
  947. }
  948. static int h_pidfile(int argc, unsigned char **argv){
  949. FILE *pidf;
  950. if(!(pidf = fopen((char *)argv[1], "w"))){
  951. fprintf(stderr, "Failed to open pid file %s\n", argv[1]);
  952. return(1);
  953. }
  954. fprintf(pidf,"%u", (unsigned)getpid());
  955. fclose(pidf);
  956. return 0;
  957. }
  958. static int h_monitor(int argc, unsigned char **argv){
  959. struct filemon * fm;
  960. fm = myalloc(sizeof (struct filemon));
  961. if(stat((char *)argv[1], &fm->sb)){
  962. myfree(fm);
  963. fprintf(stderr, "Warning: file %s doesn't exist on line %d\n", argv[1], linenum);
  964. }
  965. else {
  966. fm->path = mystrdup((char *)argv[1]);
  967. fm->next = conf.fmon;
  968. conf.fmon = fm;
  969. }
  970. return 0;
  971. }
  972. static int h_parent(int argc, unsigned char **argv){
  973. struct ace *acl = NULL;
  974. struct chain *chains;
  975. acl = conf.acl;
  976. while(acl && acl->next) acl = acl->next;
  977. if(!acl || (acl->action && acl->action != 2)) {
  978. fprintf(stderr, "Chaining error: last ACL entry was not \"allow\" or \"redirect\" on line %d\n", linenum);
  979. return(1);
  980. }
  981. acl->action = 2;
  982. chains = NULL;
  983. if(!acl->chains) {
  984. chains = acl->chains = myalloc(sizeof(struct chain));
  985. }
  986. else {
  987. chains = acl->chains;
  988. while(chains->next)chains = chains->next;
  989. chains->next = myalloc(sizeof(struct chain));
  990. chains = chains->next;
  991. }
  992. memset(chains, 0, sizeof(struct chain));
  993. if(!chains){
  994. fprintf(stderr, "Chainig error: unable to allocate memory for chain\n");
  995. return(2);
  996. }
  997. chains->weight = (unsigned)atoi((char *)argv[1]);
  998. if(chains->weight == 0 || chains->weight >1000) {
  999. fprintf(stderr, "Chaining error: bad chain weight %u line %d\n", chains->weight, linenum);
  1000. return(3);
  1001. }
  1002. if(!strcmp((char *)argv[2], "tcp"))chains->type = R_TCP;
  1003. else if(!strcmp((char *)argv[2], "http"))chains->type = R_HTTP;
  1004. else if(!strcmp((char *)argv[2], "connect"))chains->type = R_CONNECT;
  1005. else if(!strcmp((char *)argv[2], "socks4"))chains->type = R_SOCKS4;
  1006. else if(!strcmp((char *)argv[2], "socks5"))chains->type = R_SOCKS5;
  1007. else if(!strcmp((char *)argv[2], "connect+"))chains->type = R_CONNECTP;
  1008. else if(!strcmp((char *)argv[2], "socks4+"))chains->type = R_SOCKS4P;
  1009. else if(!strcmp((char *)argv[2], "socks5+"))chains->type = R_SOCKS5P;
  1010. else if(!strcmp((char *)argv[2], "socks4b"))chains->type = R_SOCKS4B;
  1011. else if(!strcmp((char *)argv[2], "socks5b"))chains->type = R_SOCKS5B;
  1012. else if(!strcmp((char *)argv[2], "pop3"))chains->type = R_POP3;
  1013. else if(!strcmp((char *)argv[2], "ftp"))chains->type = R_FTP;
  1014. else if(!strcmp((char *)argv[2], "admin"))chains->type = R_ADMIN;
  1015. else if(!strcmp((char *)argv[2], "icq"))chains->type = R_ICQ;
  1016. else if(!strcmp((char *)argv[2], "msn"))chains->type = R_MSN;
  1017. else {
  1018. fprintf(stderr, "Chaining error: bad chain type (%s)\n", argv[2]);
  1019. return(4);
  1020. }
  1021. chains->redirip = getip(argv[3]);
  1022. chains->redirport = htons((unsigned short)atoi((char *)argv[4]));
  1023. if(argc > 5) chains->extuser = (unsigned char *)mystrdup((char *)argv[5]);
  1024. if(argc > 6) chains->extpass = (unsigned char *)mystrdup((char *)argv[6]);
  1025. return 0;
  1026. }
  1027. static int h_nolog(int argc, unsigned char **argv){
  1028. struct ace *acl = NULL;
  1029. acl = conf.acl;
  1030. if(!acl) {
  1031. fprintf(stderr, "Chaining error: last ACL entry was not \"allow/deny\" on line %d\n", linenum);
  1032. return(1);
  1033. }
  1034. while(acl->next) acl = acl->next;
  1035. if(!strcmp(argv[0],"nolog")) acl->nolog = 1;
  1036. else acl->weight = atoi((char*)argv[1]);
  1037. return 0;
  1038. }
  1039. int scanipl(unsigned char *arg, struct iplist *dst){
  1040. #ifndef NOIPV6
  1041. struct sockaddr_in6 sa;
  1042. #else
  1043. struct sockaddr_in sa;
  1044. #endif
  1045. char * slash, *dash;
  1046. int masklen, addrlen;
  1047. if((slash = strchr(arg, '/'))) *slash = 0;
  1048. if((dash = strchr(arg,'-'))) *dash = 0;
  1049. if(!getip46(46, arg, (struct sockaddr *)&sa)) return 1;
  1050. memcpy(&dst->ip_from, SAADDR(&sa), SAADDRLEN(&sa));
  1051. dst->family = *SAFAMILY(&sa);
  1052. if(dash){
  1053. if(!getip46(46, dash+1, (struct sockaddr *)&sa)) return 2;
  1054. memcpy(&dst->ip_to, SAADDR(&sa), SAADDRLEN(&sa));
  1055. return 0;
  1056. }
  1057. if(slash){
  1058. addrlen = SAADDRLEN(&sa);
  1059. masklen = atoi(slash+1);
  1060. if(masklen >= 0 && masklen<(addrlen*8)){
  1061. int i, nbytes = masklen / 8, nbits = masklen % 8;
  1062. for(i = addrlen; i; i--){
  1063. ((unsigned char *)&dst->ip_from)[i-1] = 0x00;
  1064. ((unsigned char *)&dst->ip_to)[i-1] = 0xff;
  1065. }
  1066. memcpy(&dst->ip_to, &dst->ip_from, addrlen - i);
  1067. for(;nbits;nbits--){
  1068. ((unsigned char *)&dst->ip_from)[i-1] &= (0x01<<(nbits-1));
  1069. ((unsigned char *)&dst->ip_to)[i-1] |= (0x01<<(nbits-1));
  1070. }
  1071. return 0;
  1072. }
  1073. }
  1074. memcpy(&dst->ip_to, &dst->ip_from, SAADDRLEN(&sa));
  1075. return 0;
  1076. }
  1077. struct ace * make_ace (int argc, unsigned char ** argv){
  1078. struct ace * acl;
  1079. unsigned char *arg;
  1080. struct iplist *ipl=NULL;
  1081. struct portlist *portl=NULL;
  1082. struct userlist *userl=NULL;
  1083. struct hostname *hostnamel=NULL;
  1084. int res;
  1085. acl = myalloc(sizeof(struct ace));
  1086. if(!acl) return acl;
  1087. memset(acl, 0, sizeof(struct ace));
  1088. if(argc > 0 && strcmp("*", (char *)argv[0])) {
  1089. arg = argv[0];
  1090. arg = (unsigned char *)strtok((char *)arg, ",");
  1091. do {
  1092. if(!acl->users) {
  1093. acl->users = userl = myalloc(sizeof(struct userlist));
  1094. }
  1095. else {
  1096. userl->next = myalloc(sizeof(struct userlist));
  1097. userl = userl -> next;
  1098. }
  1099. if(!userl) {
  1100. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1101. return(NULL);
  1102. }
  1103. memset(userl, 0, sizeof(struct userlist));
  1104. userl->user=(unsigned char*)mystrdup((char *)arg);
  1105. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1106. }
  1107. if(argc > 1 && strcmp("*", (char *)argv[1])) {
  1108. arg = (unsigned char *)strtok((char *)argv[1], ",");
  1109. do {
  1110. if(!acl->src) {
  1111. acl->src = ipl = myalloc(sizeof(struct iplist));
  1112. }
  1113. else {
  1114. ipl->next = myalloc(sizeof(struct iplist));
  1115. ipl = ipl -> next;
  1116. }
  1117. if(!ipl) {
  1118. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1119. return(NULL);
  1120. }
  1121. memset(ipl, 0, sizeof(struct iplist));
  1122. if (scanipl(arg, ipl)) {
  1123. fprintf(stderr, "Invalid IP or CIDR, line %d\n", linenum);
  1124. return(NULL);
  1125. }
  1126. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1127. }
  1128. if(argc > 2 && strcmp("*", (char *)argv[2])) {
  1129. arg = (unsigned char *)strtok((char *)argv[2], ",");
  1130. do {
  1131. int arglen;
  1132. unsigned char *pattern;
  1133. arglen = (int)strlen((char *)arg);
  1134. if(arglen > 0 && (arg[arglen-1] < '0' || arg[arglen-1] > '9')){
  1135. if(!acl->dstnames) {
  1136. acl->dstnames = hostnamel = myalloc(sizeof(struct hostname));
  1137. }
  1138. else {
  1139. hostnamel->next = myalloc(sizeof(struct hostname));
  1140. hostnamel = hostnamel -> next;
  1141. }
  1142. if(!hostnamel){
  1143. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1144. return(NULL);
  1145. }
  1146. memset(hostnamel, 0, sizeof(struct hostname));
  1147. hostnamel->matchtype = 3;
  1148. pattern = arg;
  1149. if(pattern[arglen-1] == '*'){
  1150. arglen --;
  1151. pattern[arglen] = 0;
  1152. hostnamel->matchtype ^= MATCHEND;
  1153. }
  1154. if(pattern[0] == '*'){
  1155. pattern++;
  1156. arglen--;
  1157. hostnamel->matchtype ^= MATCHBEGIN;
  1158. }
  1159. hostnamel->name = (unsigned char *) mystrdup( (char *)pattern);
  1160. if(!hostnamel->name) {
  1161. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1162. return(NULL);
  1163. }
  1164. }
  1165. else {
  1166. if(!acl->dst) {
  1167. acl->dst = ipl = myalloc(sizeof(struct iplist));
  1168. }
  1169. else {
  1170. ipl->next = myalloc(sizeof(struct iplist));
  1171. ipl = ipl -> next;
  1172. }
  1173. if(!ipl) {
  1174. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1175. return(NULL);
  1176. }
  1177. memset(ipl, 0, sizeof(struct iplist));
  1178. if (scanipl(arg, ipl)) {
  1179. fprintf(stderr, "Invalid IP or CIDR, line %d\n", linenum);
  1180. return(NULL);
  1181. }
  1182. }
  1183. }while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1184. }
  1185. if(argc > 3 && strcmp("*", (char *)argv[3])) {
  1186. arg = (unsigned char *)strtok((char *)argv[3], ",");
  1187. do {
  1188. if(!acl->ports) {
  1189. acl->ports = portl = myalloc(sizeof(struct portlist));
  1190. }
  1191. else {
  1192. portl->next = myalloc(sizeof(struct portlist));
  1193. portl = portl -> next;
  1194. }
  1195. if(!portl) {
  1196. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1197. return(NULL);
  1198. }
  1199. memset(portl, 0, sizeof(struct portlist));
  1200. res = sscanf((char *)arg, "%hu-%hu", &portl->startport, &portl->endport);
  1201. if(res < 1) {
  1202. fprintf(stderr, "Invalid port or port range, line %d\n", linenum);
  1203. return(NULL);
  1204. }
  1205. if (res == 1) portl->endport = portl->startport;
  1206. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1207. }
  1208. if(argc > 4 && strcmp("*", (char *)argv[4])) {
  1209. arg = (unsigned char *)strtok((char *)argv[4], ",");
  1210. do {
  1211. if(!strcmp((char *)arg, "CONNECT")){
  1212. acl->operation |= CONNECT;
  1213. }
  1214. else if(!strcmp((char *)arg, "BIND")){
  1215. acl->operation |= BIND;
  1216. }
  1217. else if(!strcmp((char *)arg, "UDPASSOC")){
  1218. acl->operation |= UDPASSOC;
  1219. }
  1220. else if(!strcmp((char *)arg, "ICMPASSOC")){
  1221. acl->operation |= ICMPASSOC;
  1222. }
  1223. else if(!strcmp((char *)arg, "HTTP_GET")){
  1224. acl->operation |= HTTP_GET;
  1225. }
  1226. else if(!strcmp((char *)arg, "HTTP_PUT")){
  1227. acl->operation |= HTTP_PUT;
  1228. }
  1229. else if(!strcmp((char *)arg, "HTTP_POST")){
  1230. acl->operation |= HTTP_POST;
  1231. }
  1232. else if(!strcmp((char *)arg, "HTTP_HEAD")){
  1233. acl->operation |= HTTP_HEAD;
  1234. }
  1235. else if(!strcmp((char *)arg, "HTTP_OTHER")){
  1236. acl->operation |= HTTP_OTHER;
  1237. }
  1238. else if(!strcmp((char *)arg, "HTTP_CONNECT")){
  1239. acl->operation |= HTTP_CONNECT;
  1240. }
  1241. else if(!strcmp((char *)arg, "HTTP")){
  1242. acl->operation |= HTTP;
  1243. }
  1244. else if(!strcmp((char *)arg, "HTTPS")){
  1245. acl->operation |= HTTPS;
  1246. }
  1247. else if(!strcmp((char *)arg, "FTP_GET")){
  1248. acl->operation |= FTP_GET;
  1249. }
  1250. else if(!strcmp((char *)arg, "FTP_PUT")){
  1251. acl->operation |= FTP_PUT;
  1252. }
  1253. else if(!strcmp((char *)arg, "FTP_LIST")){
  1254. acl->operation |= FTP_LIST;
  1255. }
  1256. else if(!strcmp((char *)arg, "FTP_DATA")){
  1257. acl->operation |= FTP_DATA;
  1258. }
  1259. else if(!strcmp((char *)arg, "FTP")){
  1260. acl->operation |= FTP;
  1261. }
  1262. else if(!strcmp((char *)arg, "ADMIN")){
  1263. acl->operation |= ADMIN;
  1264. }
  1265. else if(!strcmp((char *)arg, "DNSRESOLVE")){
  1266. acl->operation |= DNSRESOLVE;
  1267. }
  1268. else if(!strcmp((char *)arg, "ICQ")){
  1269. acl->operation |= IM_ICQ;
  1270. }
  1271. else {
  1272. fprintf(stderr, "Unknown operation type: %s line %d\n", arg, linenum);
  1273. return(NULL);
  1274. }
  1275. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  1276. }
  1277. if(argc > 5){
  1278. for(arg = argv[5]; *arg;){
  1279. int val, val1;
  1280. if(!isnumber(*arg)){
  1281. arg++;
  1282. continue;
  1283. }
  1284. val1 = val = (*arg - '0');
  1285. arg++;
  1286. if(*arg == '-' && isnumber(*(arg+1)) && (*(arg+1) - '0') > val) {
  1287. val1 = (*(arg+1) - '0');
  1288. arg+=2;
  1289. }
  1290. for(; val<=val1; val++) acl->wdays |= (1 << (val % 7));
  1291. }
  1292. }
  1293. if(argc > 6){
  1294. for(arg = argv[6]; strlen((char *)arg) >= 17 &&
  1295. isdigit(arg[0]) &&
  1296. isdigit(arg[1]) &&
  1297. isdigit(arg[3]) &&
  1298. isdigit(arg[4]) &&
  1299. isdigit(arg[6]) &&
  1300. isdigit(arg[7]) &&
  1301. isdigit(arg[9]) &&
  1302. isdigit(arg[10]) &&
  1303. isdigit(arg[12]) &&
  1304. isdigit(arg[13]) &&
  1305. isdigit(arg[15]) &&
  1306. isdigit(arg[16])
  1307. ; arg+=18){
  1308. int t1, t2;
  1309. struct period *sp;
  1310. t1 = (arg[0] - '0') * 10 + (arg[1] - '0');
  1311. t1 = (t1 * 60) + (arg[3] - '0') * 10 + (arg[4] - '0');
  1312. t1 = (t1 * 60) + (arg[6] - '0') * 10 + (arg[7] - '0');
  1313. t2 = (arg[9] - '0') * 10 + (arg[10] - '0');
  1314. t2 = (t2 * 60) + (arg[12] - '0') * 10 + (arg[13] - '0');
  1315. t2 = (t2 * 60) + (arg[15] - '0') * 10 + (arg[16] - '0');
  1316. if(t2 < t1) break;
  1317. sp = myalloc(sizeof(struct period));
  1318. if(sp){
  1319. sp->fromtime = t1;
  1320. sp->totime = t2;
  1321. sp->next = acl->periods;
  1322. acl->periods = sp;
  1323. }
  1324. if(arg[17]!=',') break;
  1325. }
  1326. }
  1327. if (argc > 7){
  1328. acl->weight = atoi((char *)argv[7]);
  1329. }
  1330. return acl;
  1331. }
  1332. static int h_ace(int argc, unsigned char **argv){
  1333. int res = 0;
  1334. int offset = 0;
  1335. struct ace *acl = NULL;
  1336. struct bandlim * nbl;
  1337. struct trafcount * tl;
  1338. if(!strcmp((char *)argv[0], "allow")){
  1339. res = ALLOW;
  1340. }
  1341. else if(!strcmp((char *)argv[0], "deny")){
  1342. res = DENY;
  1343. }
  1344. else if(!strcmp((char *)argv[0], "redirect")){
  1345. res = REDIRECT;
  1346. offset = 2;
  1347. }
  1348. else if(!strcmp((char *)argv[0], "bandlimin")||!strcmp((char *)argv[0], "bandlimout")){
  1349. res = BANDLIM;
  1350. offset = 1;
  1351. }
  1352. else if(!strcmp((char *)argv[0], "nobandlimin")||!strcmp((char *)argv[0], "nobandlimout")){
  1353. res = NOBANDLIM;
  1354. }
  1355. else if(!strcmp((char *)argv[0], "countin")){
  1356. res = COUNTIN;
  1357. offset = 3;
  1358. }
  1359. else if(!strcmp((char *)argv[0], "nocountin")){
  1360. res = NOCOUNTIN;
  1361. }
  1362. else if(!strcmp((char *)argv[0], "countout")){
  1363. res = COUNTOUT;
  1364. offset = 3;
  1365. }
  1366. else if(!strcmp((char *)argv[0], "nocountout")){
  1367. res = NOCOUNTOUT;
  1368. }
  1369. acl = make_ace(argc - (offset+1), argv + (offset + 1));
  1370. if(!acl) {
  1371. fprintf(stderr, "Unable to parse ACL entry, line %d\n", linenum);
  1372. return(1);
  1373. }
  1374. acl->action = res;
  1375. switch(acl->action){
  1376. case REDIRECT:
  1377. acl->chains = myalloc(sizeof(struct chain));
  1378. if(!acl->chains) {
  1379. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1380. return(2);
  1381. }
  1382. acl->chains->type = R_HTTP;
  1383. acl->chains->redirip = getip(argv[1]);
  1384. acl->chains->redirport = htons((unsigned short)atoi((char *)argv[2]));
  1385. acl->chains->weight = 1000;
  1386. acl->chains->extuser = NULL;
  1387. acl->chains->extpass = NULL;
  1388. acl->chains->next = NULL;
  1389. case ALLOW:
  1390. case DENY:
  1391. if(!conf.acl){
  1392. conf.acl = acl;
  1393. }
  1394. else {
  1395. struct ace * acei;
  1396. for(acei = conf.acl; acei->next; acei = acei->next);
  1397. acei->next = acl;
  1398. }
  1399. break;
  1400. case BANDLIM:
  1401. case NOBANDLIM:
  1402. nbl = myalloc(sizeof(struct bandlim));
  1403. if(!nbl) {
  1404. fprintf(stderr, "No memory to create band limit filter\n");
  1405. return(3);
  1406. }
  1407. memset(nbl, 0, sizeof(struct bandlim));
  1408. nbl->ace = acl;
  1409. if(acl->action == BANDLIM) {
  1410. sscanf((char *)argv[1], "%u", &nbl->rate);
  1411. if(nbl->rate < 300) {
  1412. fprintf(stderr, "Wrong bandwidth specified, line %d\n", linenum);
  1413. return(4);
  1414. }
  1415. }
  1416. pthread_mutex_lock(&bandlim_mutex);
  1417. if(!strcmp((char *)argv[0], "bandlimin") || !strcmp((char *)argv[0], "nobandlimin")){
  1418. if(!conf.bandlimiter){
  1419. conf.bandlimiter = nbl;
  1420. }
  1421. else {
  1422. struct bandlim * bli;
  1423. for(bli = conf.bandlimiter; bli->next; bli = bli->next);
  1424. bli->next = nbl;
  1425. }
  1426. }
  1427. else {
  1428. if(!conf.bandlimiterout){
  1429. conf.bandlimiterout = nbl;
  1430. }
  1431. else {
  1432. struct bandlim * bli;
  1433. for(bli = conf.bandlimiterout; bli->next; bli = bli->next);
  1434. bli->next = nbl;
  1435. }
  1436. }
  1437. pthread_mutex_unlock(&bandlim_mutex);
  1438. break;
  1439. case COUNTIN:
  1440. case NOCOUNTIN:
  1441. case COUNTOUT:
  1442. case NOCOUNTOUT:
  1443. tl = myalloc(sizeof(struct trafcount));
  1444. if(!tl) {
  1445. fprintf(stderr, "No memory to create traffic limit filter\n");
  1446. return(5);
  1447. }
  1448. memset(tl, 0, sizeof(struct trafcount));
  1449. tl->ace = acl;
  1450. if((acl->action == COUNTIN)||(acl->action == COUNTOUT)) {
  1451. unsigned long lim;
  1452. tl->comment = ( char *)argv[1];
  1453. while(isdigit(*tl->comment))tl->comment++;
  1454. if(*tl->comment== '/')tl->comment++;
  1455. tl->comment = mystrdup(tl->comment);
  1456. sscanf((char *)argv[1], "%u", &tl->number);
  1457. sscanf((char *)argv[3], "%lu", &lim);
  1458. tl->type = getrotate(*argv[2]);
  1459. tl->traflim64 = ((uint64_t)lim)*(1024*1024);
  1460. if(!tl->traflim64) {
  1461. fprintf(stderr, "Wrong traffic limit specified, line %d\n", linenum);
  1462. return(6);
  1463. }
  1464. if(tl->number != 0 && conf.counterd >= 0) {
  1465. lseek(conf.counterd,
  1466. sizeof(struct counter_header) + (tl->number - 1) * sizeof(struct counter_record),
  1467. SEEK_SET);
  1468. memset(&crecord, 0, sizeof(struct counter_record));
  1469. read(conf.counterd, &crecord, sizeof(struct counter_record));
  1470. tl->traf64 = crecord.traf64;
  1471. tl->cleared = crecord.cleared;
  1472. tl->updated = crecord.updated;
  1473. #ifdef _MAX__TIME64_T
  1474. if(tl->cleared >= _MAX__TIME64_T || tl->updated >= _MAX__TIME64_T){
  1475. fprintf(stderr, "Invalid or corrupted counter file. Use countersutil utility to convert from older version\n");
  1476. return(6);
  1477. }
  1478. #endif
  1479. }
  1480. }
  1481. pthread_mutex_lock(&tc_mutex);
  1482. if(!conf.trafcounter){
  1483. conf.trafcounter = tl;
  1484. }
  1485. else {
  1486. struct trafcount * ntl;
  1487. for(ntl = conf.trafcounter; ntl->next; ntl = ntl->next);
  1488. ntl->next = tl;
  1489. }
  1490. pthread_mutex_unlock(&tc_mutex);
  1491. }
  1492. return 0;
  1493. }
  1494. static int h_logdump(int argc, unsigned char **argv){
  1495. conf.logdumpsrv = (unsigned) atoi((char *) *(argv + 1));
  1496. if(argc > 2) conf.logdumpcli = (unsigned) atoi((char *) *(argv + 2));
  1497. return 0;
  1498. }
  1499. static int h_filtermaxsize(int argc, unsigned char **argv){
  1500. conf.filtermaxsize = atoi((char *) *(argv + 1));
  1501. return 0;
  1502. }
  1503. static int h_delimchar(int argc, unsigned char **argv){
  1504. conf.delimchar = *argv[1];
  1505. return 0;
  1506. }
  1507. static int h_authcache(int argc, unsigned char **argv){
  1508. conf.authcachetype = 0;
  1509. if(strstr((char *) *(argv + 1), "ip")) conf.authcachetype |= 1;
  1510. if(strstr((char *) *(argv + 1), "user")) conf.authcachetype |= 2;
  1511. if(strstr((char *) *(argv + 1), "pass")) conf.authcachetype |= 4;
  1512. if(argc > 2) conf.authcachetime = (unsigned) atoi((char *) *(argv + 2));
  1513. if(!conf.authcachetype) conf.authcachetype = 6;
  1514. if(!conf.authcachetime) conf.authcachetime = 600;
  1515. return 0;
  1516. }
  1517. static int h_plugin(int argc, unsigned char **argv){
  1518. #ifdef NOPLUGINS
  1519. return 999;
  1520. #else
  1521. #ifdef _WIN32
  1522. HINSTANCE hi;
  1523. FARPROC fp;
  1524. #ifdef _WINCE
  1525. hi = LoadLibraryW((LPCWSTR)CEToUnicode(argv[1]));
  1526. #else
  1527. hi = LoadLibrary(argv[1]);
  1528. #endif
  1529. if(!hi) {
  1530. fprintf(stderr, "Failed to load %s, code %d\n", argv[1], (int)GetLastError());
  1531. return 1;
  1532. }
  1533. #ifdef _WINCE
  1534. fp = GetProcAddressW(hi, (LPCWSTR)CEToUnicode(argv[2]));
  1535. #else
  1536. fp = GetProcAddress(hi, argv[2]);
  1537. #endif
  1538. if(!fp) {
  1539. printf("%s not found in %s, code: %d\n", argv[2], argv[1], (int)GetLastError());
  1540. return 2;
  1541. }
  1542. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1543. #else
  1544. void *hi, *fp;
  1545. hi = dlopen((char *)argv[1], RTLD_LAZY);
  1546. if(!hi) return 1;
  1547. fp = dlsym(hi, (char *)argv[2]);
  1548. if(!fp) return 2;
  1549. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1550. #endif
  1551. #endif
  1552. }
  1553. #ifndef _WIN32
  1554. static int h_setuid(int argc, unsigned char **argv){
  1555. int res;
  1556. res = atoi((char *)argv[1]);
  1557. if(!res || setuid(res)) {
  1558. fprintf(stderr, "Unable to set uid %d", res);
  1559. return(1);
  1560. }
  1561. return 0;
  1562. }
  1563. static int h_setgid(int argc, unsigned char **argv){
  1564. int res;
  1565. res = atoi((char *)argv[1]);
  1566. if(!res || setgid(res)) {
  1567. fprintf(stderr, "Unable to set gid %d", res);
  1568. return(1);
  1569. }
  1570. return 0;
  1571. }
  1572. static int h_chroot(int argc, unsigned char **argv){
  1573. if(!chrootp){
  1574. char *p;
  1575. if(chroot((char *)argv[1])) {
  1576. fprintf(stderr, "Unable to chroot %s", argv[1]);
  1577. return(1);
  1578. }
  1579. p = (char *)argv[1] + strlen((char *)argv[1]) ;
  1580. while (p > (char *)argv[1] && p[-1] == '/'){
  1581. p--;
  1582. *p = 0;
  1583. }
  1584. chrootp = mystrdup((char *)argv[1]);
  1585. }
  1586. return 0;
  1587. }
  1588. #endif
  1589. struct commands specificcommands[]={
  1590. #ifndef _WIN32
  1591. {specificcommands+1, "setuid", h_setuid, 2, 2},
  1592. {specificcommands+2, "setgid", h_setgid, 2, 2},
  1593. {specificcommands+3, "chroot", h_chroot, 2, 2},
  1594. #endif
  1595. {NULL, "", h_noop, 1, 0}
  1596. };
  1597. struct commands commandhandlers[]={
  1598. {commandhandlers+1, "", h_noop, 1, 0},
  1599. {commandhandlers+2, "proxy", h_proxy, 1, 0},
  1600. {commandhandlers+3, "pop3p", h_proxy, 1, 0},
  1601. {commandhandlers+4, "ftppr", h_proxy, 1, 0},
  1602. {commandhandlers+5, "socks", h_proxy, 1, 0},
  1603. {commandhandlers+6, "tcppm", h_proxy, 4, 0},
  1604. {commandhandlers+7, "udppm", h_proxy, 4, 0},
  1605. {commandhandlers+8, "admin", h_proxy, 1, 0},
  1606. {commandhandlers+9, "dnspr", h_proxy, 1, 0},
  1607. {commandhandlers+10, "internal", h_internal, 2, 2},
  1608. {commandhandlers+11, "external", h_external, 2, 2},
  1609. {commandhandlers+12, "log", h_log, 1, 0},
  1610. {commandhandlers+13, "service", h_service, 1, 1},
  1611. {commandhandlers+14, "daemon", h_daemon, 1, 1},
  1612. {commandhandlers+15, "config", h_config, 2, 2},
  1613. {commandhandlers+16, "include", h_include, 2, 2},
  1614. {commandhandlers+17, "archiver", h_archiver, 3, 0},
  1615. {commandhandlers+18, "counter", h_counter, 2, 4},
  1616. {commandhandlers+19, "rotate", h_rotate, 2, 2},
  1617. {commandhandlers+20, "logformat", h_logformat, 2, 2},
  1618. {commandhandlers+21, "timeouts", h_timeouts, 2, 0},
  1619. {commandhandlers+22, "auth", h_auth, 2, 0},
  1620. {commandhandlers+23, "users", h_users, 2, 0},
  1621. {commandhandlers+24, "maxconn", h_maxconn, 2, 2},
  1622. {commandhandlers+25, "flush", h_flush, 1, 1},
  1623. {commandhandlers+26, "nserver", h_nserver, 2, 2},
  1624. {commandhandlers+27, "fakeresolve", h_fakeresolve, 1, 1},
  1625. {commandhandlers+28, "nscache", h_nscache, 2, 2},
  1626. {commandhandlers+29, "nsrecord", h_nsrecord, 3, 3},
  1627. {commandhandlers+30, "dialer", h_dialer, 2, 2},
  1628. {commandhandlers+31, "system", h_system, 2, 2},
  1629. {commandhandlers+32, "pidfile", h_pidfile, 2, 2},
  1630. {commandhandlers+33, "monitor", h_monitor, 2, 2},
  1631. {commandhandlers+34, "parent", h_parent, 5, 0},
  1632. {commandhandlers+35, "allow", h_ace, 1, 0},
  1633. {commandhandlers+36, "deny", h_ace, 1, 0},
  1634. {commandhandlers+37, "redirect", h_ace, 3, 0},
  1635. {commandhandlers+38, "bandlimin", h_ace, 2, 0},
  1636. {commandhandlers+39, "bandlimout", h_ace, 2, 0},
  1637. {commandhandlers+40, "nobandlimin", h_ace, 1, 0},
  1638. {commandhandlers+41, "nobandlimout", h_ace, 1, 0},
  1639. {commandhandlers+42, "countin", h_ace, 4, 0},
  1640. {commandhandlers+43, "nocountin", h_ace, 1, 0},
  1641. {commandhandlers+44, "countout", h_ace, 4, 0},
  1642. {commandhandlers+45, "nocountout", h_ace, 1, 0},
  1643. {commandhandlers+46, "plugin", h_plugin, 3, 0},
  1644. {commandhandlers+47, "logdump", h_logdump, 2, 3},
  1645. {commandhandlers+48, "filtermaxsize", h_filtermaxsize, 2, 2},
  1646. {commandhandlers+49, "nolog", h_nolog, 1, 1},
  1647. {commandhandlers+50, "weight", h_nolog, 2, 2},
  1648. {commandhandlers+51, "authcache", h_authcache, 2, 3},
  1649. {commandhandlers+52, "smtpp", h_proxy, 1, 0},
  1650. {commandhandlers+53, "icqpr", h_proxy, 4, 0},
  1651. {commandhandlers+54, "msnpr", h_proxy, 4, 0},
  1652. {commandhandlers+55, "delimchar",h_delimchar, 2, 2},
  1653. {commandhandlers+56, "authnserver", h_authnserver, 2, 2},
  1654. {specificcommands, "", h_noop, 1, 0}
  1655. };
  1656. int readconfig(FILE * fp){
  1657. unsigned char ** argv = NULL;
  1658. unsigned char * buf = NULL;
  1659. int bufsize = STRINGBUF*2;
  1660. int inbuf = 0;
  1661. int argc;
  1662. struct commands * cm;
  1663. int res = 0;
  1664. if( !(buf = myalloc(bufsize)) || ! (argv = myalloc(NPARAMS * sizeof(unsigned char *) + 1)) ) {
  1665. fprintf(stderr, "No memory for configuration");
  1666. return(10);
  1667. }
  1668. for (linenum = 1; fgets((char *)buf, STRINGBUF, fp); linenum++){
  1669. if(!*buf || isspace(*buf) || (*buf) == '#')continue;
  1670. inbuf = (int)(strlen((char *)buf) + 1);
  1671. argc = parsestr (buf, argv, NPARAMS-1, &buf, &inbuf, &bufsize);
  1672. if(argc < 1) {
  1673. fprintf(stderr, "Parse error line %d\n", linenum);
  1674. return(21);
  1675. }
  1676. argv[argc] = NULL;
  1677. if(!strcmp((char *)argv[0], "end") && argc == 1) {
  1678. break;
  1679. }
  1680. else if(!strcmp((char *)argv[0], "writable") && argc == 1) {
  1681. if(!writable){
  1682. writable = freopen(curconf, "r+", fp);
  1683. if(!writable){
  1684. fprintf(stderr, "Unable to reopen config for writing: %s\n", curconf);
  1685. return 1;
  1686. }
  1687. }
  1688. continue;
  1689. }
  1690. res = 1;
  1691. for(cm = commandhandlers; cm; cm = cm->next){
  1692. if(!strcmp((char *)argv[0], (char *)cm->command) && argc >= cm->minargs && (!cm->maxargs || argc <= cm->maxargs)){
  1693. res = (*cm->handler)(argc, argv);
  1694. if(res > 0){
  1695. fprintf(stderr, "Command: '%s' failed with code %d, line %d\n", argv[0], res, linenum);
  1696. return(linenum);
  1697. }
  1698. if(!res) break;
  1699. }
  1700. }
  1701. if(res != 1)continue;
  1702. fprintf(stderr, "Unknown command: '%s' line %d\n", argv[0], linenum);
  1703. return(linenum);
  1704. }
  1705. myfree(buf);
  1706. myfree(argv);
  1707. return 0;
  1708. }
  1709. #ifndef _WINCE
  1710. int main(int argc, char * argv[]) {
  1711. #else
  1712. int WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow){
  1713. int argc;
  1714. char ** argv;
  1715. WNDCLASS wc;
  1716. HWND hwnd = 0;
  1717. #endif
  1718. int res = 0;
  1719. FILE * fp = NULL;
  1720. #ifdef _WIN32
  1721. unsigned char * arg;
  1722. WSADATA wd;
  1723. WSAStartup(MAKEWORD( 1, 1 ), &wd);
  1724. osv.dwOSVersionInfoSize = sizeof(osv);
  1725. GetVersionEx(&osv);
  1726. #endif
  1727. #ifdef _WINCE
  1728. argc = ceparseargs((char *)lpCmdLine);
  1729. argv = ceargv;
  1730. if(FindWindow(L"3proxy", L"3proxy")) return 0;
  1731. ZeroMemory(&wc,sizeof(wc));
  1732. wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH);
  1733. wc.hInstance=hInstance;
  1734. wc.hCursor=LoadCursor(NULL,IDC_ARROW);
  1735. wc.lpfnWndProc=DefWindowProc;
  1736. wc.style=CS_HREDRAW|CS_VREDRAW;
  1737. wc.lpszClassName=L"3proxy";
  1738. RegisterClass(&wc);
  1739. hwnd = CreateWindowEx(0,L"3proxy",L"3proxy",WS_VISIBLE|WS_POPUP,0,0,0,0,0,0,hInstance,0);
  1740. #endif
  1741. conf.stringtable = strings;
  1742. #ifdef _WIN32
  1743. #ifndef _WINCE
  1744. if((argc == 2 || argc == 3)&& !strcmp((char *)argv[1], "--install")) {
  1745. sprintf((char *)tmpbuf, "%s will be installed and started.\n"
  1746. "By clicking Yes you confirm you read and accepted License Agreement.\n"
  1747. "You can use Administration/Services to control %s service.",
  1748. conf.stringtable[1], conf.stringtable[2]);
  1749. if(MessageBox(NULL, (char *)tmpbuf, conf.stringtable[2], MB_YESNO|MB_ICONASTERISK) != IDYES) return 1;
  1750. *tmpbuf = '\"';
  1751. if (!(res = SearchPath(NULL, argv[0], ".exe", 256, (char *)tmpbuf+1, (LPTSTR*)&arg))) {
  1752. perror("Failed to find executable filename");
  1753. RETURN(102);
  1754. }
  1755. strcat((char *)tmpbuf, "\" \"");
  1756. if(!(res = GetFullPathName ((argc == 3)?argv[2]:(char*)DEFAULTCONFIG, 256, (char *)tmpbuf+res+4, (char **)&arg))){
  1757. perror("Failed to find config filename");
  1758. RETURN(103);
  1759. }
  1760. strcat((char *)tmpbuf, "\" --service");
  1761. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT){
  1762. SC_HANDLE sch;
  1763. if(!(sch = OpenSCManager(NULL, NULL, GENERIC_WRITE|SERVICE_START ))){
  1764. perror("Failed to open Service Manager");
  1765. RETURN(101);
  1766. }
  1767. 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))){
  1768. perror("Failed to create service");
  1769. RETURN(103);
  1770. }
  1771. if (!StartService(sch, 0, NULL)) {
  1772. perror("Failed to start service");
  1773. RETURN(103);
  1774. }
  1775. }
  1776. else {
  1777. HKEY runsrv;
  1778. if(RegOpenKeyEx( HKEY_LOCAL_MACHINE,
  1779. "Software\\Microsoft\\Windows\\CurrentVersion\\RunServices",
  1780. 0,
  1781. KEY_ALL_ACCESS,
  1782. &runsrv) != ERROR_SUCCESS){
  1783. perror("Failed to open registry");
  1784. RETURN(104);
  1785. }
  1786. if(RegSetValueEx( runsrv,
  1787. conf.stringtable[1],
  1788. 0,
  1789. REG_EXPAND_SZ,
  1790. (char *)tmpbuf,
  1791. (int)strlen((char *)tmpbuf)+1)!=ERROR_SUCCESS){
  1792. perror("Failed to set registry value");
  1793. RETURN(105);
  1794. }
  1795. }
  1796. return 0;
  1797. }
  1798. if((argc == 2 || argc == 3)&& !strcmp((char *)argv[1], "--remove")) {
  1799. if(osv.dwPlatformId == VER_PLATFORM_WIN32_NT){
  1800. SC_HANDLE sch;
  1801. if(!(sch = OpenSCManager(NULL, NULL, GENERIC_WRITE))){
  1802. perror("Failed to open Service Manager\n");
  1803. RETURN(106);
  1804. }
  1805. if (!(sch = OpenService(sch, conf.stringtable[1], DELETE))){
  1806. perror("Failed to open service");
  1807. RETURN(107);
  1808. }
  1809. if (!DeleteService(sch)){
  1810. perror("Failed to delete service");
  1811. RETURN(108);
  1812. }
  1813. }
  1814. else {
  1815. HKEY runsrv;
  1816. if(RegOpenKeyEx( HKEY_LOCAL_MACHINE,
  1817. "Software\\Microsoft\\Windows\\CurrentVersion\\RunServices",
  1818. 0,
  1819. KEY_ALL_ACCESS,
  1820. &runsrv) != ERROR_SUCCESS){
  1821. perror("Failed to open registry");
  1822. RETURN(109);
  1823. }
  1824. if(RegDeleteValue(runsrv, conf.stringtable[1]) != ERROR_SUCCESS){
  1825. perror("Failed to clear registry");
  1826. RETURN(110);
  1827. }
  1828. }
  1829. RETURN(0);
  1830. }
  1831. if(argc==3 && !strcmp(argv[2], "--service")){
  1832. service = 1;
  1833. argc = 2;
  1834. }
  1835. #endif
  1836. #endif
  1837. conf.conffile = mystrdup((argc==2)?argv[1]:(char*)DEFAULTCONFIG);
  1838. if(conf.conffile && *conf.conffile != '-') {
  1839. fp = confopen();
  1840. #ifndef _WIN32
  1841. if(!fp) fp = stdin;
  1842. #endif
  1843. }
  1844. if(argc > 2 || !(fp)) {
  1845. fprintf(stderr, "Usage: %s [conffile]\n", argv[0]);
  1846. #ifdef _WIN32
  1847. fprintf(stderr, "\n\t%s --install [conffile]\n\tto install as service\n"
  1848. "\n\t%s --remove\n\tto remove service\n", argv[0], argv[0]);
  1849. #else
  1850. fprintf(stderr, "\n if conffile is missing, configuration is expected from stdin\n");
  1851. #endif
  1852. fprintf(stderr, "\n%s %s\n%s\n", conf.stringtable[2], conf.stringtable[3], copyright);
  1853. return 1;
  1854. }
  1855. pthread_mutex_init(&bandlim_mutex, NULL);
  1856. pthread_mutex_init(&hash_mutex, NULL);
  1857. pthread_mutex_init(&tc_mutex, NULL);
  1858. pthread_mutex_init(&pwl_mutex, NULL);
  1859. #ifndef NOODBC
  1860. pthread_mutex_init(&odbc_mutex, NULL);
  1861. #endif
  1862. {
  1863. char * args[] = {"auth", "iponly", NULL};
  1864. h_auth(2, (unsigned char **)args);
  1865. }
  1866. res = readconfig(fp);
  1867. if(res) RETURN(res);
  1868. if(!writable)fclose(fp);
  1869. #ifdef _WIN32
  1870. #ifndef _WINCE
  1871. if(service){
  1872. SERVICE_TABLE_ENTRY ste[] =
  1873. {
  1874. { conf.stringtable[1], (LPSERVICE_MAIN_FUNCTION)ServiceMain},
  1875. { NULL, NULL }
  1876. };
  1877. if(!StartServiceCtrlDispatcher( ste ))cyclestep();
  1878. }
  1879. else
  1880. #endif
  1881. {
  1882. cyclestep();
  1883. }
  1884. #else
  1885. signal(SIGCONT, mysigpause);
  1886. signal(SIGTERM, mysigterm);
  1887. signal(SIGUSR1, mysigusr1);
  1888. signal(SIGPIPE, SIG_IGN);
  1889. cyclestep();
  1890. #endif
  1891. CLEARRETURN:
  1892. return 0;
  1893. }