3proxy.c 50 KB

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