conf.c 44 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2016 by Vladimir Dubrovin <3proxy@3proxy.ru>
  4. please read License Agreement
  5. */
  6. #include "proxy.h"
  7. #ifndef _WIN32
  8. #include <sys/resource.h>
  9. #ifndef NOPLUGINS
  10. #include <dlfcn.h>
  11. #endif
  12. #endif
  13. #ifndef DEFAULTCONFIG
  14. #define DEFAULTCONFIG conf.stringtable[25]
  15. #endif
  16. pthread_mutex_t bandlim_mutex;
  17. pthread_mutex_t tc_mutex;
  18. pthread_mutex_t pwl_mutex;
  19. pthread_mutex_t hash_mutex;
  20. pthread_mutex_t config_mutex;
  21. int haveerror = 0;
  22. int linenum = 0;
  23. FILE *writable;
  24. struct counter_header cheader = {"3CF", (time_t)0};
  25. struct counter_record crecord;
  26. int mainfunc (int argc, char** argv);
  27. struct proxydef childdef = {NULL, 0, 0, S_NOSERVICE, ""};
  28. #define STRINGBUF 65535
  29. #define NPARAMS 4096
  30. #ifndef _WIN32
  31. char *chrootp = NULL;
  32. static pthread_attr_t pa;
  33. #endif
  34. char * curconf = NULL;
  35. FILE * confopen(){
  36. curconf = conf.conffile;
  37. #ifndef _WIN32
  38. if(chrootp){
  39. if(strstr(curconf, chrootp) == curconf)
  40. curconf += strlen(chrootp);
  41. }
  42. #endif
  43. if(writable) {
  44. rewind(writable);
  45. return writable;
  46. }
  47. return fopen(curconf, "r");
  48. }
  49. #ifdef _WIN32
  50. DWORD WINAPI startsrv(LPVOID data) {
  51. #else
  52. void * startsrv(void * data) {
  53. #endif
  54. struct child *d = (struct child *)data;
  55. mainfunc(d->argc, (char **)d->argv);
  56. return 0;
  57. }
  58. int included =0;
  59. int getrotate(char c){
  60. switch(c){
  61. case 'c':
  62. case 'C':
  63. return MINUTELY;
  64. case 'h':
  65. case 'H':
  66. return HOURLY;
  67. case 'd':
  68. case 'D':
  69. return DAILY;
  70. case 'w':
  71. case 'W':
  72. return WEEKLY;
  73. case 'y':
  74. case 'Y':
  75. return ANNUALLY;
  76. case 'm':
  77. case 'M':
  78. return MONTHLY;
  79. default:
  80. return NEVER;
  81. }
  82. }
  83. unsigned char * dologname (unsigned char *buf, unsigned char *name, const unsigned char *ext, ROTATION lt, time_t t) {
  84. struct tm *ts;
  85. ts = localtime(&t);
  86. if(strchr((char *)name, '%')){
  87. struct clientparam fakecli;
  88. memset(&fakecli, 0, sizeof(fakecli));
  89. dobuf2(&fakecli, buf, NULL, NULL, ts, (char *)name);
  90. }
  91. else switch(lt){
  92. case NONE:
  93. sprintf((char *)buf, "%s", name);
  94. break;
  95. case ANNUALLY:
  96. sprintf((char *)buf, "%s.%04d", name, ts->tm_year+1900);
  97. break;
  98. case MONTHLY:
  99. sprintf((char *)buf, "%s.%04d.%02d", name, ts->tm_year+1900, ts->tm_mon+1);
  100. break;
  101. case WEEKLY:
  102. t = t - (ts->tm_wday * (60*60*24));
  103. ts = localtime(&t);
  104. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  105. break;
  106. case DAILY:
  107. sprintf((char *)buf, "%s.%04d.%02d.%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday);
  108. break;
  109. case HOURLY:
  110. sprintf((char *)buf, "%s.%04d.%02d.%02d-%02d", name, ts->tm_year+1900, ts->tm_mon+1, ts->tm_mday, ts->tm_hour);
  111. break;
  112. case MINUTELY:
  113. 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);
  114. break;
  115. default:
  116. break;
  117. }
  118. if(ext){
  119. strcat((char *)buf, ".");
  120. strcat((char *)buf, (char *)ext);
  121. }
  122. return buf;
  123. }
  124. int start_proxy_thread(struct child * chp){
  125. pthread_t thread;
  126. #ifdef _WIN32
  127. HANDLE h;
  128. #endif
  129. conf.threadinit = 1;
  130. #ifdef _WIN32
  131. #ifndef _WINCE
  132. h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, 16384+conf.stacksize, (void *)startsrv, (void *) chp, (DWORD)0, &thread);
  133. #else
  134. h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, 16384+conf.stacksize, (void *)startsrv, (void *) chp, (DWORD)0, &thread);
  135. #endif
  136. if(h)CloseHandle(h);
  137. #else
  138. pthread_attr_init(&pa);
  139. pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + (16384+conf.stacksize));
  140. pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
  141. pthread_create(&thread, &pa, startsrv, (void *)chp);
  142. #endif
  143. while(conf.threadinit)usleep(SLEEPTIME);
  144. if(haveerror) {
  145. fprintf(stderr, "Service not started on line: %d\n", linenum);
  146. return(40);
  147. }
  148. return 0;
  149. }
  150. static int h_proxy(int argc, unsigned char ** argv){
  151. struct child ch;
  152. ch.argc = argc;
  153. ch.argv = argv;
  154. if(!strcmp((char *)argv[0], "proxy")) {
  155. childdef.pf = proxychild;
  156. childdef.port = 3128;
  157. childdef.isudp = 0;
  158. childdef.service = S_PROXY;
  159. childdef.helpmessage = " -n - no NTLM support\n";
  160. #ifdef NOIPV6
  161. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize)){
  162. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, proxy may run very slow\n", linenum);
  163. }
  164. #endif
  165. }
  166. else if(!strcmp((char *)argv[0], "pop3p")) {
  167. childdef.pf = pop3pchild;
  168. childdef.port = 110;
  169. childdef.isudp = 0;
  170. childdef.service = S_POP3P;
  171. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  172. }
  173. else if(!strcmp((char *)argv[0], "smtpp")) {
  174. childdef.pf = smtppchild;
  175. childdef.port = 25;
  176. childdef.isudp = 0;
  177. childdef.service = S_SMTPP;
  178. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  179. }
  180. else if(!strcmp((char *)argv[0], "ftppr")) {
  181. childdef.pf = ftpprchild;
  182. childdef.port = 21;
  183. childdef.isudp = 0;
  184. childdef.service = S_FTPPR;
  185. childdef.helpmessage = " -hdefault_host[:port] - use this host and port as default if no host specified\n";
  186. }
  187. else if(!strcmp((char *)argv[0], "socks")) {
  188. childdef.pf = sockschild;
  189. childdef.port = 1080;
  190. childdef.isudp = 0;
  191. childdef.service = S_SOCKS;
  192. childdef.helpmessage = " -n - no NTLM support\n";
  193. #ifdef NOIPV6
  194. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize)){
  195. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, socks may run very slow\n", linenum);
  196. }
  197. #endif
  198. }
  199. else if(!strcmp((char *)argv[0], "tcppm")) {
  200. childdef.pf = tcppmchild;
  201. childdef.port = 0;
  202. childdef.isudp = 0;
  203. childdef.service = S_TCPPM;
  204. childdef.helpmessage = "";
  205. }
  206. else if(!strcmp((char *)argv[0], "icqpr")) {
  207. childdef.pf = icqprchild;
  208. childdef.port = 0;
  209. childdef.isudp = 0;
  210. childdef.service = S_ICQPR;
  211. childdef.helpmessage = "";
  212. }
  213. /*
  214. else if(!strcmp((char *)argv[0], "msnpr")) {
  215. childdef.pf = msnprchild;
  216. childdef.port = 0;
  217. childdef.isudp = 0;
  218. childdef.service = S_MSNPR;
  219. childdef.helpmessage = "";
  220. }
  221. */
  222. else if(!strcmp((char *)argv[0], "udppm")) {
  223. childdef.pf = udppmchild;
  224. childdef.port = 0;
  225. childdef.isudp = 1;
  226. childdef.service = S_UDPPM;
  227. childdef.helpmessage = " -s single packet UDP service for request/reply (DNS-like) services\n";
  228. }
  229. else if(!strcmp((char *)argv[0], "admin")) {
  230. childdef.pf = adminchild;
  231. childdef.port = 80;
  232. childdef.isudp = 0;
  233. childdef.service = S_ADMIN;
  234. }
  235. else if(!strcmp((char *)argv[0], "dnspr")) {
  236. childdef.pf = dnsprchild;
  237. childdef.port = 53;
  238. childdef.isudp = 1;
  239. childdef.service = S_DNSPR;
  240. childdef.helpmessage = " -s - simple DNS forwarding - do not use 3proxy resolver / name cache\n";
  241. #ifndef NOIPV6
  242. if(!resolvfunc || (resolvfunc == myresolver && !dns_table.hashsize) || resolvfunc == fakeresolver){
  243. fprintf(stderr, "[line %d] Warning: no nserver/nscache configured, dnspr will not work as expected\n", linenum);
  244. }
  245. #endif
  246. }
  247. return start_proxy_thread(&ch);
  248. }
  249. static int h_internal(int argc, unsigned char ** argv){
  250. getip46(46, argv[1], (struct sockaddr *)&conf.intsa);
  251. return 0;
  252. }
  253. static int h_external(int argc, unsigned char ** argv){
  254. int res;
  255. #ifndef NOIPV6
  256. struct sockaddr_in6 sa6;
  257. memset(&sa6, 0, sizeof(sa6));
  258. res = getip46(46, argv[1], (struct sockaddr *)&sa6);
  259. if(!res) return 1;
  260. if (*SAFAMILY(&sa6)==AF_INET) conf.extsa = sa6;
  261. else conf.extsa6 = sa6;
  262. #else
  263. res = getip46(46, argv[1], (struct sockaddr *)&conf.extsa);
  264. if(!res) return 1;
  265. #endif
  266. return 0;
  267. }
  268. static int h_log(int argc, unsigned char ** argv){
  269. unsigned char tmpbuf[8192];
  270. conf.logfunc = logstdout;
  271. if(conf.logtarget){
  272. myfree(conf.logtarget);
  273. conf.logtarget = NULL;
  274. }
  275. if(argc > 1) {
  276. conf.logtarget = (unsigned char *)mystrdup((char *)argv[1]);
  277. if(*argv[1]=='@'){
  278. #ifndef _WIN32
  279. openlog((char *)conf.logtarget+1, LOG_PID, LOG_DAEMON);
  280. conf.logfunc = logsyslog;
  281. #endif
  282. }
  283. #ifndef NOODBC
  284. else if(*argv[1]=='&'){
  285. pthread_mutex_lock(&log_mutex);
  286. close_sql();
  287. init_sql((char *)argv[1]+1);
  288. pthread_mutex_unlock(&log_mutex);
  289. conf.logfunc = logsql;
  290. }
  291. #endif
  292. else {
  293. FILE *fp;
  294. if(argc > 2) {
  295. conf.logtype = getrotate(*argv[2]);
  296. }
  297. conf.logtime = time(0);
  298. if(conf.logname)myfree(conf.logname);
  299. conf.logname = (unsigned char *)mystrdup((char *)argv[1]);
  300. fp = fopen((char *)dologname (tmpbuf, conf.logname, NULL, conf.logtype, conf.logtime), "a");
  301. if(!fp){
  302. perror((char *)tmpbuf);
  303. return 1;
  304. }
  305. else {
  306. if(conf.stdlog)fclose(conf.stdlog);
  307. conf.stdlog = fp;
  308. #ifdef _WINCE
  309. freopen(tmpbuf, "w", stdout);
  310. freopen(tmpbuf, "w", stderr);
  311. #endif
  312. }
  313. }
  314. }
  315. return 0;
  316. }
  317. static int h_stacksize(int argc, unsigned char **argv){
  318. conf.stacksize = atoi((char *)argv[1]);
  319. return 0;
  320. }
  321. static int h_force(int argc, unsigned char **argv){
  322. conf.noforce = 0;
  323. return 0;
  324. }
  325. static int h_noforce(int argc, unsigned char **argv){
  326. conf.noforce = 1;
  327. return 0;
  328. }
  329. static int h_service(int argc, unsigned char **argv){
  330. return 0;
  331. }
  332. static int h_daemon(int argc, unsigned char **argv){
  333. if(!conf.demon)daemonize();
  334. conf.demon = 1;
  335. return 0;
  336. }
  337. static int h_config(int argc, unsigned char **argv){
  338. if(conf.conffile)myfree(conf.conffile);
  339. conf.conffile = mystrdup((char *)argv[1]);
  340. return 0;
  341. }
  342. static int h_include(int argc, unsigned char **argv){
  343. int res;
  344. FILE *fp1;
  345. fp1 = fopen((char *)argv[1], "r");
  346. if(!fp1){
  347. fprintf(stderr, "Unable to open included file: %s\n", argv[1]);
  348. return 1;
  349. }
  350. res = readconfig(fp1);
  351. fclose(fp1);
  352. return res;
  353. }
  354. static int h_archiver(int argc, unsigned char **argv){
  355. int j;
  356. conf.archiver = myalloc(argc * sizeof(char *));
  357. if(conf.archiver) {
  358. conf.archiverc = argc;
  359. for(j = 0; j < conf.archiverc; j++) conf.archiver[j] = (unsigned char *)mystrdup((char *)argv[j]);
  360. }
  361. return 0;
  362. }
  363. static int h_counter(int argc, unsigned char **argv){
  364. struct counter_header ch1;
  365. if(conf.counterd >=0)close(conf.counterd);
  366. if(!conf.trafcountfunc) conf.trafcountfunc = trafcountfunc;
  367. conf.counterd = open((char *)argv[1], O_BINARY|O_RDWR|O_CREAT, 0660);
  368. if(conf.counterd<0){
  369. fprintf(stderr, "Unable to open counter file %s, line %d\n", argv[1], linenum);
  370. return 1;
  371. }
  372. if(read(conf.counterd, &ch1, sizeof(ch1))==sizeof(ch1)){
  373. if(memcmp(&ch1, &cheader, 4)){
  374. fprintf(stderr, "Not a counter file %s, line %d\n", argv[1], linenum);
  375. return 2;
  376. }
  377. #ifdef _TIME64_T_DEFINED
  378. #ifdef _MAX__TIME64_T
  379. #define MAX_COUNTER_TIME (_MAX__TIME64_T)
  380. #elif defined (MAX__TIME64_T)
  381. #define MAX_COUNTER_TIME (MAX__TIME64_T)
  382. #else
  383. #define MAX_COUNTER_TIME (0x793406fff)
  384. #endif
  385. #else
  386. #define MAX_COUNTER_TIME ((sizeof(time_t)>4)?(time_t)0x793406fff:(time_t)0x7fffffff)
  387. #endif
  388. if(ch1.updated < 0 || ch1.updated >= MAX_COUNTER_TIME){
  389. fprintf(stderr, "Invalid or corrupted counter file %s. Use countersutil utility to convert from older version\n", argv[1]);
  390. return 3;
  391. }
  392. cheader.updated = ch1.updated;
  393. }
  394. if(argc >=4) {
  395. conf.countertype = getrotate(*argv[2]);
  396. if(conf.counterfile) myfree(conf.counterfile);
  397. conf.counterfile = mystrdup((char *)argv[3]);
  398. }
  399. return 0;
  400. }
  401. static int h_rotate(int argc, unsigned char **argv){
  402. conf.rotate = atoi((char *)argv[1]);
  403. return 0;
  404. }
  405. static int h_logformat(int argc, unsigned char **argv){
  406. if(conf.logformat) myfree(conf.logformat);
  407. conf.logformat = (unsigned char *)mystrdup((char *)argv[1]);
  408. return 0;
  409. }
  410. static int h_timeouts(int argc, unsigned char **argv){
  411. int j;
  412. for(j = 0; conf.timeouts[j] && j + 1 < argc; j++) {
  413. if((conf.timeouts[j] = atoi((char *)argv[j + 1])) <= 0 || conf.timeouts[j] > 2000000){
  414. fprintf(stderr, "Invalid timeout: %s, line %d\n", argv[j + 1], linenum);
  415. return(1);
  416. }
  417. }
  418. return 0;
  419. }
  420. static int h_noop(int argc, unsigned char **argv){
  421. return 0;
  422. }
  423. static int h_auth(int argc, unsigned char **argv){
  424. struct auth *au, * newau;
  425. freeauth(conf.authfuncs);
  426. conf.authfuncs = NULL;
  427. if(!conf.bandlimfunc)conf.bandlimfunc = bandlimitfunc;
  428. for(argc--; argc; argc--){
  429. for(au = authfuncs; au; au=au->next){
  430. if(!strcmp((char *)argv[argc], au->desc)){
  431. newau = myalloc(sizeof(struct auth));
  432. newau->next = conf.authfuncs;
  433. conf.authfuncs = newau;
  434. conf.authfuncs->desc = au->desc;
  435. conf.authfuncs->authenticate = au->authenticate;
  436. conf.authfuncs->authorize = au->authorize;
  437. break;
  438. }
  439. }
  440. if(!au) return 1;
  441. }
  442. conf.authfunc = doauth;
  443. return 0;
  444. }
  445. static int h_users(int argc, unsigned char **argv){
  446. int j;
  447. unsigned char *arg;
  448. struct passwords *pwl = NULL;
  449. for (j = 1; j<argc; j++) {
  450. if(!(pwl = myalloc(sizeof(struct passwords)))) {
  451. fprintf(stderr, "No memory for PWL entry, line %d\n", linenum);
  452. return(1);
  453. }
  454. memset(pwl, 0, sizeof(struct passwords));
  455. arg = (unsigned char *)strchr((char *)argv[j], ':');
  456. if(!arg||!arg[1]||!arg[2]||arg[3]!=':') {
  457. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  458. pwl->pwtype = SYS;
  459. }
  460. else {
  461. *arg = 0;
  462. pwl->user = (unsigned char *)mystrdup((char *)argv[j]);
  463. if((arg[1] == 'C' && arg[2] == 'L' && (pwl->pwtype = CL)) ||
  464. (arg[1] == 'C' && arg[2] == 'R' && (pwl->pwtype = CR)) ||
  465. (arg[1] == 'N' && arg[2] == 'T' && (pwl->pwtype = NT)) ||
  466. (arg[1] == 'L' && arg[2] == 'M' && (pwl->pwtype = LM))){
  467. pwl->password = (unsigned char *)mystrdup((char *)arg+4);
  468. }
  469. else {
  470. pwl->password = (unsigned char *) mystrdup((char *)arg + 1);
  471. pwl->pwtype = UN;
  472. }
  473. }
  474. pthread_mutex_lock(&pwl_mutex);
  475. pwl->next = conf.pwl;
  476. conf.pwl = pwl;
  477. pthread_mutex_unlock(&pwl_mutex);
  478. }
  479. return 0;
  480. }
  481. static int h_maxconn(int argc, unsigned char **argv){
  482. conf.maxchild = atoi((char *)argv[1]);
  483. if(!conf.maxchild) {
  484. return(1);
  485. }
  486. #ifndef _WIN32
  487. {
  488. struct rlimit rl;
  489. if(!getrlimit(RLIMIT_NOFILE, &rl)){
  490. if((conf.maxchild<<1) > rl.rlim_cur)
  491. fprintf(stderr, "[line %d] Warning: current open file ulimits are too low (cur: %d/max: %d),"
  492. " maxconn requires at least %d for every running service."
  493. " Configure ulimits according to system documentation\n",
  494. linenum, (int)rl.rlim_cur, (int)rl.rlim_max, (conf.maxchild<<1));
  495. }
  496. }
  497. #endif
  498. return 0;
  499. }
  500. static int h_flush(int argc, unsigned char **argv){
  501. freeacl(conf.acl);
  502. conf.acl = NULL;
  503. return 0;
  504. }
  505. /*
  506. static int h_flushusers(int argc, unsigned char **argv){
  507. freepwl(conf.pwl);
  508. conf.pwl = NULL;
  509. return 0;
  510. }
  511. */
  512. static int h_nserver(int argc, unsigned char **argv){
  513. char *str;
  514. if(numservers < MAXNSERVERS) {
  515. if((str = strchr((char *)argv[1], '/')))
  516. *str = 0;
  517. *SAPORT(&nservers[numservers].addr) = htons(53);
  518. if(parsehost(46, argv[1], (struct sockaddr *)&nservers[numservers].addr)) return 1;
  519. if(str) {
  520. nservers[numservers].usetcp = strstr(str + 1, "tcp")? 1:0;
  521. *str = '/';
  522. }
  523. numservers++;
  524. }
  525. resolvfunc = myresolver;
  526. return 0;
  527. }
  528. static int h_authnserver(int argc, unsigned char **argv){
  529. char *str;
  530. if((str = strchr((char *)argv[1], '/')))
  531. *str = 0;
  532. if(parsehost(46, argv[1], (struct sockaddr *)&authnserver.addr)) return 1;
  533. *SAPORT(&authnserver.addr) = htons(53);
  534. if(str) {
  535. authnserver.usetcp = strstr(str + 1, "tcp")? 1:0;
  536. *str = '/';
  537. }
  538. return 0;
  539. }
  540. static int h_fakeresolve(int argc, unsigned char **argv){
  541. resolvfunc = fakeresolver;
  542. return 0;
  543. }
  544. static int h_nscache(int argc, unsigned char **argv){
  545. int res;
  546. res = atoi((char *)argv[1]);
  547. if(res < 256) {
  548. fprintf(stderr, "Invalid NS cache size: %d\n", res);
  549. return 1;
  550. }
  551. if(inithashtable(&dns_table, (unsigned)res)){
  552. fprintf(stderr, "Failed to initialize NS cache\n");
  553. return 2;
  554. }
  555. return 0;
  556. }
  557. static int h_nscache6(int argc, unsigned char **argv){
  558. int res;
  559. res = atoi((char *)argv[1]);
  560. if(res < 256) {
  561. fprintf(stderr, "Invalid NS cache size: %d\n", res);
  562. return 1;
  563. }
  564. if(inithashtable(&dns6_table, (unsigned)res)){
  565. fprintf(stderr, "Failed to initialize NS cache\n");
  566. return 2;
  567. }
  568. return 0;
  569. }
  570. static int h_nsrecord(int argc, unsigned char **argv){
  571. #ifndef NOIPV6
  572. struct sockaddr_in6 sa;
  573. #else
  574. struct sockaddr_in sa;
  575. #endif
  576. memset(&sa, 0, sizeof(sa));
  577. if(!getip46(46, argv[2], (struct sockaddr *)&sa)) return 1;
  578. hashadd(*SAFAMILY(&sa)==AF_INET6?&dns6_table:&dns_table, argv[1], SAADDR(&sa), (time_t)0xffffffff);
  579. return 0;
  580. }
  581. static int h_dialer(int argc, unsigned char **argv){
  582. if(conf.demanddialprog) myfree(conf.demanddialprog);
  583. conf.demanddialprog = mystrdup((char *)argv[1]);
  584. return 0;
  585. }
  586. static int h_system(int argc, unsigned char **argv){
  587. int res;
  588. if((res = system((char *)argv[1])) == -1){
  589. fprintf(stderr, "Failed to start %s\n", argv[1]);
  590. return(1);
  591. }
  592. return 0;
  593. }
  594. static int h_pidfile(int argc, unsigned char **argv){
  595. FILE *pidf;
  596. if(!(pidf = fopen((char *)argv[1], "w"))){
  597. fprintf(stderr, "Failed to open pid file %s\n", argv[1]);
  598. return(1);
  599. }
  600. fprintf(pidf,"%u", (unsigned)getpid());
  601. fclose(pidf);
  602. return 0;
  603. }
  604. static int h_monitor(int argc, unsigned char **argv){
  605. struct filemon * fm;
  606. fm = myalloc(sizeof (struct filemon));
  607. if(stat((char *)argv[1], &fm->sb)){
  608. myfree(fm);
  609. fprintf(stderr, "Warning: file %s doesn't exist on line %d\n", argv[1], linenum);
  610. }
  611. else {
  612. fm->path = mystrdup((char *)argv[1]);
  613. fm->next = conf.fmon;
  614. conf.fmon = fm;
  615. }
  616. return 0;
  617. }
  618. static int h_parent(int argc, unsigned char **argv){
  619. struct ace *acl = NULL;
  620. struct chain *chains;
  621. acl = conf.acl;
  622. while(acl && acl->next) acl = acl->next;
  623. if(!acl || (acl->action && acl->action != 2)) {
  624. fprintf(stderr, "Chaining error: last ACL entry was not \"allow\" or \"redirect\" on line %d\n", linenum);
  625. return(1);
  626. }
  627. acl->action = 2;
  628. chains = NULL;
  629. if(!acl->chains) {
  630. chains = acl->chains = myalloc(sizeof(struct chain));
  631. }
  632. else {
  633. chains = acl->chains;
  634. while(chains->next)chains = chains->next;
  635. chains->next = myalloc(sizeof(struct chain));
  636. chains = chains->next;
  637. }
  638. memset(chains, 0, sizeof(struct chain));
  639. if(!chains){
  640. fprintf(stderr, "Chainig error: unable to allocate memory for chain\n");
  641. return(2);
  642. }
  643. chains->weight = (unsigned)atoi((char *)argv[1]);
  644. if(chains->weight == 0 || chains->weight >1000) {
  645. fprintf(stderr, "Chaining error: bad chain weight %u line %d\n", chains->weight, linenum);
  646. return(3);
  647. }
  648. if(!strcmp((char *)argv[2], "tcp"))chains->type = R_TCP;
  649. else if(!strcmp((char *)argv[2], "http"))chains->type = R_HTTP;
  650. else if(!strcmp((char *)argv[2], "connect"))chains->type = R_CONNECT;
  651. else if(!strcmp((char *)argv[2], "socks4"))chains->type = R_SOCKS4;
  652. else if(!strcmp((char *)argv[2], "socks5"))chains->type = R_SOCKS5;
  653. else if(!strcmp((char *)argv[2], "connect+"))chains->type = R_CONNECTP;
  654. else if(!strcmp((char *)argv[2], "socks4+"))chains->type = R_SOCKS4P;
  655. else if(!strcmp((char *)argv[2], "socks5+"))chains->type = R_SOCKS5P;
  656. else if(!strcmp((char *)argv[2], "socks4b"))chains->type = R_SOCKS4B;
  657. else if(!strcmp((char *)argv[2], "socks5b"))chains->type = R_SOCKS5B;
  658. else if(!strcmp((char *)argv[2], "pop3"))chains->type = R_POP3;
  659. else if(!strcmp((char *)argv[2], "ftp"))chains->type = R_FTP;
  660. else if(!strcmp((char *)argv[2], "admin"))chains->type = R_ADMIN;
  661. else if(!strcmp((char *)argv[2], "icq"))chains->type = R_ICQ;
  662. else if(!strcmp((char *)argv[2], "extip"))chains->type = R_EXTIP;
  663. else if(!strcmp((char *)argv[2], "smtp"))chains->type = R_SMTP;
  664. else {
  665. fprintf(stderr, "Chaining error: bad chain type (%s)\n", argv[2]);
  666. return(4);
  667. }
  668. if(!getip46(46, argv[3], (struct sockaddr *)&chains->addr)) return 5;
  669. *SAPORT(&chains->addr) = htons((unsigned short)atoi((char *)argv[4]));
  670. if(argc > 5) chains->extuser = (unsigned char *)mystrdup((char *)argv[5]);
  671. if(argc > 6) chains->extpass = (unsigned char *)mystrdup((char *)argv[6]);
  672. return 0;
  673. }
  674. static int h_nolog(int argc, unsigned char **argv){
  675. struct ace *acl = NULL;
  676. acl = conf.acl;
  677. if(!acl) {
  678. fprintf(stderr, "Chaining error: last ACL entry was not \"allow/deny\" on line %d\n", linenum);
  679. return(1);
  680. }
  681. while(acl->next) acl = acl->next;
  682. if(!strcmp((char *)argv[0],"nolog")) acl->nolog = 1;
  683. else acl->weight = atoi((char*)argv[1]);
  684. return 0;
  685. }
  686. int scanipl(unsigned char *arg, struct iplist *dst){
  687. #ifndef NOIPV6
  688. struct sockaddr_in6 sa;
  689. #else
  690. struct sockaddr_in sa;
  691. #endif
  692. char * slash, *dash;
  693. int masklen, addrlen;
  694. if((slash = strchr((char *)arg, '/'))) *slash = 0;
  695. if((dash = strchr((char *)arg,'-'))) *dash = 0;
  696. if(!getip46(46, arg, (struct sockaddr *)&sa)) return 1;
  697. memcpy(&dst->ip_from, SAADDR(&sa), SAADDRLEN(&sa));
  698. dst->family = *SAFAMILY(&sa);
  699. if(dash){
  700. if(!getip46(46, (unsigned char *)dash+1, (struct sockaddr *)&sa)) return 2;
  701. memcpy(&dst->ip_to, SAADDR(&sa), SAADDRLEN(&sa));
  702. if(*SAFAMILY(&sa) != dst->family || memcmp(&dst->ip_to, &dst->ip_from, SAADDRLEN(&sa)) < 0) return 3;
  703. return 0;
  704. }
  705. memcpy(&dst->ip_to, &dst->ip_from, SAADDRLEN(&sa));
  706. if(slash){
  707. addrlen = SAADDRLEN(&sa);
  708. masklen = atoi(slash+1);
  709. if(masklen < 0 || masklen > (addrlen*8)) return 4;
  710. else {
  711. int i, nbytes = masklen / 8, nbits = (8 - (masklen % 8)) % 8;
  712. for(i = addrlen; i>(nbytes + (nbits > 0)); i--){
  713. ((unsigned char *)&dst->ip_from)[i-1] = 0x00;
  714. ((unsigned char *)&dst->ip_to)[i-1] = 0xff;
  715. }
  716. for(;nbits;nbits--){
  717. ((unsigned char *)&dst->ip_from)[nbytes] &= ~(0x01<<(nbits-1));
  718. ((unsigned char *)&dst->ip_to)[nbytes] |= (0x01<<(nbits-1));
  719. }
  720. return 0;
  721. }
  722. }
  723. return 0;
  724. }
  725. struct ace * make_ace (int argc, unsigned char ** argv){
  726. struct ace * acl;
  727. unsigned char *arg;
  728. struct iplist *ipl=NULL;
  729. struct portlist *portl=NULL;
  730. struct userlist *userl=NULL;
  731. struct hostname *hostnamel=NULL;
  732. int res;
  733. acl = myalloc(sizeof(struct ace));
  734. if(!acl) return acl;
  735. memset(acl, 0, sizeof(struct ace));
  736. if(argc > 0 && strcmp("*", (char *)argv[0])) {
  737. arg = argv[0];
  738. arg = (unsigned char *)strtok((char *)arg, ",");
  739. do {
  740. if(!acl->users) {
  741. acl->users = userl = myalloc(sizeof(struct userlist));
  742. }
  743. else {
  744. userl->next = myalloc(sizeof(struct userlist));
  745. userl = userl -> next;
  746. }
  747. if(!userl) {
  748. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  749. return(NULL);
  750. }
  751. memset(userl, 0, sizeof(struct userlist));
  752. userl->user=(unsigned char*)mystrdup((char *)arg);
  753. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  754. }
  755. if(argc > 1 && strcmp("*", (char *)argv[1])) {
  756. arg = (unsigned char *)strtok((char *)argv[1], ",");
  757. do {
  758. if(!acl->src) {
  759. acl->src = ipl = myalloc(sizeof(struct iplist));
  760. }
  761. else {
  762. ipl->next = myalloc(sizeof(struct iplist));
  763. ipl = ipl -> next;
  764. }
  765. if(!ipl) {
  766. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  767. return(NULL);
  768. }
  769. memset(ipl, 0, sizeof(struct iplist));
  770. if (scanipl(arg, ipl)) {
  771. fprintf(stderr, "Invalid IP, IP range or CIDR, line %d\n", linenum);
  772. return(NULL);
  773. }
  774. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  775. }
  776. if(argc > 2 && strcmp("*", (char *)argv[2])) {
  777. arg = (unsigned char *)strtok((char *)argv[2], ",");
  778. do {
  779. int arglen;
  780. unsigned char *pattern;
  781. arglen = (int)strlen((char *)arg);
  782. if(arglen > 0 && (arg[arglen-1] < '0' || arg[arglen-1] > '9')){
  783. if(!acl->dstnames) {
  784. acl->dstnames = hostnamel = myalloc(sizeof(struct hostname));
  785. }
  786. else {
  787. hostnamel->next = myalloc(sizeof(struct hostname));
  788. hostnamel = hostnamel -> next;
  789. }
  790. if(!hostnamel){
  791. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  792. return(NULL);
  793. }
  794. memset(hostnamel, 0, sizeof(struct hostname));
  795. hostnamel->matchtype = 3;
  796. pattern = arg;
  797. if(pattern[arglen-1] == '*'){
  798. arglen --;
  799. pattern[arglen] = 0;
  800. hostnamel->matchtype ^= MATCHEND;
  801. }
  802. if(pattern[0] == '*'){
  803. pattern++;
  804. arglen--;
  805. hostnamel->matchtype ^= MATCHBEGIN;
  806. }
  807. hostnamel->name = (unsigned char *) mystrdup( (char *)pattern);
  808. if(!hostnamel->name) {
  809. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  810. return(NULL);
  811. }
  812. }
  813. else {
  814. if(!acl->dst) {
  815. acl->dst = ipl = myalloc(sizeof(struct iplist));
  816. }
  817. else {
  818. ipl->next = myalloc(sizeof(struct iplist));
  819. ipl = ipl -> next;
  820. }
  821. if(!ipl) {
  822. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  823. return(NULL);
  824. }
  825. memset(ipl, 0, sizeof(struct iplist));
  826. if (scanipl(arg, ipl)) {
  827. fprintf(stderr, "Invalid IP, IP range or CIDR, line %d\n", linenum);
  828. return(NULL);
  829. }
  830. }
  831. }while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  832. }
  833. if(argc > 3 && strcmp("*", (char *)argv[3])) {
  834. arg = (unsigned char *)strtok((char *)argv[3], ",");
  835. do {
  836. if(!acl->ports) {
  837. acl->ports = portl = myalloc(sizeof(struct portlist));
  838. }
  839. else {
  840. portl->next = myalloc(sizeof(struct portlist));
  841. portl = portl -> next;
  842. }
  843. if(!portl) {
  844. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  845. return(NULL);
  846. }
  847. memset(portl, 0, sizeof(struct portlist));
  848. res = sscanf((char *)arg, "%hu-%hu", &portl->startport, &portl->endport);
  849. if(res < 1) {
  850. fprintf(stderr, "Invalid port or port range, line %d\n", linenum);
  851. return(NULL);
  852. }
  853. if (res == 1) portl->endport = portl->startport;
  854. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  855. }
  856. if(argc > 4 && strcmp("*", (char *)argv[4])) {
  857. arg = (unsigned char *)strtok((char *)argv[4], ",");
  858. do {
  859. if(!strcmp((char *)arg, "CONNECT")){
  860. acl->operation |= CONNECT;
  861. }
  862. else if(!strcmp((char *)arg, "BIND")){
  863. acl->operation |= BIND;
  864. }
  865. else if(!strcmp((char *)arg, "UDPASSOC")){
  866. acl->operation |= UDPASSOC;
  867. }
  868. else if(!strcmp((char *)arg, "ICMPASSOC")){
  869. acl->operation |= ICMPASSOC;
  870. }
  871. else if(!strcmp((char *)arg, "HTTP_GET")){
  872. acl->operation |= HTTP_GET;
  873. }
  874. else if(!strcmp((char *)arg, "HTTP_PUT")){
  875. acl->operation |= HTTP_PUT;
  876. }
  877. else if(!strcmp((char *)arg, "HTTP_POST")){
  878. acl->operation |= HTTP_POST;
  879. }
  880. else if(!strcmp((char *)arg, "HTTP_HEAD")){
  881. acl->operation |= HTTP_HEAD;
  882. }
  883. else if(!strcmp((char *)arg, "HTTP_OTHER")){
  884. acl->operation |= HTTP_OTHER;
  885. }
  886. else if(!strcmp((char *)arg, "HTTP_CONNECT")){
  887. acl->operation |= HTTP_CONNECT;
  888. }
  889. else if(!strcmp((char *)arg, "HTTP")){
  890. acl->operation |= HTTP;
  891. }
  892. else if(!strcmp((char *)arg, "HTTPS")){
  893. acl->operation |= HTTPS;
  894. }
  895. else if(!strcmp((char *)arg, "FTP_GET")){
  896. acl->operation |= FTP_GET;
  897. }
  898. else if(!strcmp((char *)arg, "FTP_PUT")){
  899. acl->operation |= FTP_PUT;
  900. }
  901. else if(!strcmp((char *)arg, "FTP_LIST")){
  902. acl->operation |= FTP_LIST;
  903. }
  904. else if(!strcmp((char *)arg, "FTP_DATA")){
  905. acl->operation |= FTP_DATA;
  906. }
  907. else if(!strcmp((char *)arg, "FTP")){
  908. acl->operation |= FTP;
  909. }
  910. else if(!strcmp((char *)arg, "ADMIN")){
  911. acl->operation |= ADMIN;
  912. }
  913. else if(!strcmp((char *)arg, "DNSRESOLVE")){
  914. acl->operation |= DNSRESOLVE;
  915. }
  916. else if(!strcmp((char *)arg, "ICQ")){
  917. acl->operation |= IM_ICQ;
  918. }
  919. else {
  920. fprintf(stderr, "Unknown operation type: %s line %d\n", arg, linenum);
  921. return(NULL);
  922. }
  923. } while((arg = (unsigned char *)strtok((char *)NULL, ",")));
  924. }
  925. if(argc > 5){
  926. for(arg = argv[5]; *arg;){
  927. int val, val1;
  928. if(!isnumber(*arg)){
  929. arg++;
  930. continue;
  931. }
  932. val1 = val = (*arg - '0');
  933. arg++;
  934. if(*arg == '-' && isnumber(*(arg+1)) && (*(arg+1) - '0') > val) {
  935. val1 = (*(arg+1) - '0');
  936. arg+=2;
  937. }
  938. for(; val<=val1; val++) acl->wdays |= (1 << (val % 7));
  939. }
  940. }
  941. if(argc > 6){
  942. for(arg = argv[6]; strlen((char *)arg) >= 17 &&
  943. isdigit(arg[0]) &&
  944. isdigit(arg[1]) &&
  945. isdigit(arg[3]) &&
  946. isdigit(arg[4]) &&
  947. isdigit(arg[6]) &&
  948. isdigit(arg[7]) &&
  949. isdigit(arg[9]) &&
  950. isdigit(arg[10]) &&
  951. isdigit(arg[12]) &&
  952. isdigit(arg[13]) &&
  953. isdigit(arg[15]) &&
  954. isdigit(arg[16])
  955. ; arg+=18){
  956. int t1, t2;
  957. struct period *sp;
  958. t1 = (arg[0] - '0') * 10 + (arg[1] - '0');
  959. t1 = (t1 * 60) + (arg[3] - '0') * 10 + (arg[4] - '0');
  960. t1 = (t1 * 60) + (arg[6] - '0') * 10 + (arg[7] - '0');
  961. t2 = (arg[9] - '0') * 10 + (arg[10] - '0');
  962. t2 = (t2 * 60) + (arg[12] - '0') * 10 + (arg[13] - '0');
  963. t2 = (t2 * 60) + (arg[15] - '0') * 10 + (arg[16] - '0');
  964. if(t2 < t1) break;
  965. sp = myalloc(sizeof(struct period));
  966. if(sp){
  967. sp->fromtime = t1;
  968. sp->totime = t2;
  969. sp->next = acl->periods;
  970. acl->periods = sp;
  971. }
  972. if(arg[17]!=',') break;
  973. }
  974. }
  975. if (argc > 7){
  976. acl->weight = atoi((char *)argv[7]);
  977. }
  978. return acl;
  979. }
  980. static int h_ace(int argc, unsigned char **argv){
  981. int res = 0;
  982. int offset = 0;
  983. struct ace *acl = NULL;
  984. struct bandlim * nbl;
  985. struct trafcount * tl;
  986. if(!strcmp((char *)argv[0], "allow")){
  987. res = ALLOW;
  988. }
  989. else if(!strcmp((char *)argv[0], "deny")){
  990. res = DENY;
  991. }
  992. else if(!strcmp((char *)argv[0], "redirect")){
  993. res = REDIRECT;
  994. offset = 2;
  995. }
  996. else if(!strcmp((char *)argv[0], "bandlimin")||!strcmp((char *)argv[0], "bandlimout")){
  997. res = BANDLIM;
  998. offset = 1;
  999. }
  1000. else if(!strcmp((char *)argv[0], "nobandlimin")||!strcmp((char *)argv[0], "nobandlimout")){
  1001. res = NOBANDLIM;
  1002. }
  1003. else if(!strcmp((char *)argv[0], "countin")){
  1004. res = COUNTIN;
  1005. offset = 3;
  1006. }
  1007. else if(!strcmp((char *)argv[0], "nocountin")){
  1008. res = NOCOUNTIN;
  1009. }
  1010. else if(!strcmp((char *)argv[0], "countout")){
  1011. res = COUNTOUT;
  1012. offset = 3;
  1013. }
  1014. else if(!strcmp((char *)argv[0], "nocountout")){
  1015. res = NOCOUNTOUT;
  1016. }
  1017. acl = make_ace(argc - (offset+1), argv + (offset + 1));
  1018. if(!acl) {
  1019. fprintf(stderr, "Unable to parse ACL entry, line %d\n", linenum);
  1020. return(1);
  1021. }
  1022. acl->action = res;
  1023. switch(acl->action){
  1024. case REDIRECT:
  1025. acl->chains = myalloc(sizeof(struct chain));
  1026. memset(acl->chains, 0, sizeof(struct chain));
  1027. if(!acl->chains) {
  1028. fprintf(stderr, "No memory for ACL entry, line %d\n", linenum);
  1029. return(2);
  1030. }
  1031. acl->chains->type = R_HTTP;
  1032. if(!getip46(46, argv[1], (struct sockaddr *)&acl->chains->addr)) return 5;
  1033. *SAPORT(&acl->chains->addr) = htons((unsigned short)atoi((char *)argv[2]));
  1034. acl->chains->weight = 1000;
  1035. acl->chains->extuser = NULL;
  1036. acl->chains->extpass = NULL;
  1037. acl->chains->next = NULL;
  1038. case ALLOW:
  1039. case DENY:
  1040. if(!conf.acl){
  1041. conf.acl = acl;
  1042. }
  1043. else {
  1044. struct ace * acei;
  1045. for(acei = conf.acl; acei->next; acei = acei->next);
  1046. acei->next = acl;
  1047. }
  1048. break;
  1049. case BANDLIM:
  1050. case NOBANDLIM:
  1051. nbl = myalloc(sizeof(struct bandlim));
  1052. if(!nbl) {
  1053. fprintf(stderr, "No memory to create band limit filter\n");
  1054. return(3);
  1055. }
  1056. memset(nbl, 0, sizeof(struct bandlim));
  1057. nbl->ace = acl;
  1058. if(acl->action == BANDLIM) {
  1059. sscanf((char *)argv[1], "%u", &nbl->rate);
  1060. if(nbl->rate < 300) {
  1061. fprintf(stderr, "Wrong bandwidth specified, line %d\n", linenum);
  1062. return(4);
  1063. }
  1064. }
  1065. pthread_mutex_lock(&bandlim_mutex);
  1066. if(!strcmp((char *)argv[0], "bandlimin") || !strcmp((char *)argv[0], "nobandlimin")){
  1067. if(!conf.bandlimiter){
  1068. conf.bandlimiter = nbl;
  1069. }
  1070. else {
  1071. struct bandlim * bli;
  1072. for(bli = conf.bandlimiter; bli->next; bli = bli->next);
  1073. bli->next = nbl;
  1074. }
  1075. }
  1076. else {
  1077. if(!conf.bandlimiterout){
  1078. conf.bandlimiterout = nbl;
  1079. }
  1080. else {
  1081. struct bandlim * bli;
  1082. for(bli = conf.bandlimiterout; bli->next; bli = bli->next);
  1083. bli->next = nbl;
  1084. }
  1085. }
  1086. pthread_mutex_unlock(&bandlim_mutex);
  1087. break;
  1088. case COUNTIN:
  1089. case NOCOUNTIN:
  1090. case COUNTOUT:
  1091. case NOCOUNTOUT:
  1092. tl = myalloc(sizeof(struct trafcount));
  1093. if(!tl) {
  1094. fprintf(stderr, "No memory to create traffic limit filter\n");
  1095. return(5);
  1096. }
  1097. memset(tl, 0, sizeof(struct trafcount));
  1098. tl->ace = acl;
  1099. if((acl->action == COUNTIN)||(acl->action == COUNTOUT)) {
  1100. unsigned long lim;
  1101. tl->comment = ( char *)argv[1];
  1102. while(isdigit(*tl->comment))tl->comment++;
  1103. if(*tl->comment== '/')tl->comment++;
  1104. tl->comment = mystrdup(tl->comment);
  1105. sscanf((char *)argv[1], "%u", &tl->number);
  1106. sscanf((char *)argv[3], "%lu", &lim);
  1107. tl->type = getrotate(*argv[2]);
  1108. tl->traflim64 = ((uint64_t)lim)*(1024*1024);
  1109. if(!tl->traflim64) {
  1110. fprintf(stderr, "Wrong traffic limit specified, line %d\n", linenum);
  1111. return(6);
  1112. }
  1113. if(tl->number != 0 && conf.counterd >= 0) {
  1114. lseek(conf.counterd,
  1115. sizeof(struct counter_header) + (tl->number - 1) * sizeof(struct counter_record),
  1116. SEEK_SET);
  1117. memset(&crecord, 0, sizeof(struct counter_record));
  1118. read(conf.counterd, &crecord, sizeof(struct counter_record));
  1119. tl->traf64 = crecord.traf64;
  1120. tl->cleared = crecord.cleared;
  1121. tl->updated = crecord.updated;
  1122. if(tl->cleared < 0 || tl->cleared >= MAX_COUNTER_TIME || tl->updated < 0 || tl->updated >= MAX_COUNTER_TIME){
  1123. fprintf(stderr, "Invalid, incompatible or corrupted counter file.\n");
  1124. return(6);
  1125. }
  1126. }
  1127. }
  1128. pthread_mutex_lock(&tc_mutex);
  1129. if(!conf.trafcounter){
  1130. conf.trafcounter = tl;
  1131. }
  1132. else {
  1133. struct trafcount * ntl;
  1134. for(ntl = conf.trafcounter; ntl->next; ntl = ntl->next);
  1135. ntl->next = tl;
  1136. }
  1137. pthread_mutex_unlock(&tc_mutex);
  1138. }
  1139. return 0;
  1140. }
  1141. static int h_logdump(int argc, unsigned char **argv){
  1142. conf.logdumpsrv = (unsigned) atoi((char *) *(argv + 1));
  1143. if(argc > 2) conf.logdumpcli = (unsigned) atoi((char *) *(argv + 2));
  1144. return 0;
  1145. }
  1146. static int h_filtermaxsize(int argc, unsigned char **argv){
  1147. conf.filtermaxsize = atoi((char *) *(argv + 1));
  1148. return 0;
  1149. }
  1150. static int h_delimchar(int argc, unsigned char **argv){
  1151. conf.delimchar = *argv[1];
  1152. return 0;
  1153. }
  1154. static int h_authcache(int argc, unsigned char **argv){
  1155. conf.authcachetype = 0;
  1156. if(strstr((char *) *(argv + 1), "ip")) conf.authcachetype |= 1;
  1157. if(strstr((char *) *(argv + 1), "user")) conf.authcachetype |= 2;
  1158. if(strstr((char *) *(argv + 1), "pass")) conf.authcachetype |= 4;
  1159. if(argc > 2) conf.authcachetime = (unsigned) atoi((char *) *(argv + 2));
  1160. if(!conf.authcachetype) conf.authcachetype = 6;
  1161. if(!conf.authcachetime) conf.authcachetime = 600;
  1162. return 0;
  1163. }
  1164. static int h_plugin(int argc, unsigned char **argv){
  1165. #ifdef NOPLUGINS
  1166. return 999;
  1167. #else
  1168. #ifdef _WIN32
  1169. HINSTANCE hi;
  1170. FARPROC fp;
  1171. #ifdef _WINCE
  1172. hi = LoadLibraryW((LPCWSTR)CEToUnicode(argv[1]));
  1173. #else
  1174. hi = LoadLibrary((char *)argv[1]);
  1175. #endif
  1176. if(!hi) {
  1177. fprintf(stderr, "Failed to load %s, code %d\n", argv[1], (int)GetLastError());
  1178. return 1;
  1179. }
  1180. #ifdef _WINCE
  1181. fp = GetProcAddressW(hi, (LPCWSTR)CEToUnicode(argv[2]));
  1182. #else
  1183. fp = GetProcAddress(hi, (char *)argv[2]);
  1184. #endif
  1185. if(!fp) {
  1186. printf("%s not found in %s, code: %d\n", argv[2], argv[1], (int)GetLastError());
  1187. return 2;
  1188. }
  1189. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1190. #else
  1191. void *hi, *fp;
  1192. hi = dlopen((char *)argv[1], RTLD_LAZY);
  1193. if(!hi) return 1;
  1194. fp = dlsym(hi, (char *)argv[2]);
  1195. if(!fp) return 2;
  1196. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1197. #endif
  1198. #endif
  1199. }
  1200. #ifndef _WIN32
  1201. static int h_setuid(int argc, unsigned char **argv){
  1202. int res;
  1203. res = atoi((char *)argv[1]);
  1204. if(!res || setreuid(res,res)) {
  1205. fprintf(stderr, "Unable to set uid %d", res);
  1206. return(1);
  1207. }
  1208. return 0;
  1209. }
  1210. static int h_setgid(int argc, unsigned char **argv){
  1211. int res;
  1212. res = atoi((char *)argv[1]);
  1213. if(!res || setregid(res,res)) {
  1214. fprintf(stderr, "Unable to set gid %d", res);
  1215. return(1);
  1216. }
  1217. return 0;
  1218. }
  1219. static int h_chroot(int argc, unsigned char **argv){
  1220. if(!chrootp){
  1221. char *p;
  1222. if(chroot((char *)argv[1])) {
  1223. fprintf(stderr, "Unable to chroot %s", argv[1]);
  1224. return(1);
  1225. }
  1226. p = (char *)argv[1] + strlen((char *)argv[1]) ;
  1227. while (p > (char *)argv[1] && p[-1] == '/'){
  1228. p--;
  1229. *p = 0;
  1230. }
  1231. chrootp = mystrdup((char *)argv[1]);
  1232. }
  1233. return 0;
  1234. }
  1235. #endif
  1236. struct commands specificcommands[]={
  1237. #ifndef _WIN32
  1238. {specificcommands+1, "setuid", h_setuid, 2, 2},
  1239. {specificcommands+2, "setgid", h_setgid, 2, 2},
  1240. {specificcommands+3, "chroot", h_chroot, 2, 2},
  1241. #endif
  1242. {NULL, "", h_noop, 1, 0}
  1243. };
  1244. struct commands commandhandlers[]={
  1245. {commandhandlers+1, "", h_noop, 1, 0},
  1246. {commandhandlers+2, "proxy", h_proxy, 1, 0},
  1247. {commandhandlers+3, "pop3p", h_proxy, 1, 0},
  1248. {commandhandlers+4, "ftppr", h_proxy, 1, 0},
  1249. {commandhandlers+5, "socks", h_proxy, 1, 0},
  1250. {commandhandlers+6, "tcppm", h_proxy, 4, 0},
  1251. {commandhandlers+7, "udppm", h_proxy, 4, 0},
  1252. {commandhandlers+8, "admin", h_proxy, 1, 0},
  1253. {commandhandlers+9, "dnspr", h_proxy, 1, 0},
  1254. {commandhandlers+10, "internal", h_internal, 2, 2},
  1255. {commandhandlers+11, "external", h_external, 2, 2},
  1256. {commandhandlers+12, "log", h_log, 1, 0},
  1257. {commandhandlers+13, "service", h_service, 1, 1},
  1258. {commandhandlers+14, "daemon", h_daemon, 1, 1},
  1259. {commandhandlers+15, "config", h_config, 2, 2},
  1260. {commandhandlers+16, "include", h_include, 2, 2},
  1261. {commandhandlers+17, "archiver", h_archiver, 3, 0},
  1262. {commandhandlers+18, "counter", h_counter, 2, 4},
  1263. {commandhandlers+19, "rotate", h_rotate, 2, 2},
  1264. {commandhandlers+20, "logformat", h_logformat, 2, 2},
  1265. {commandhandlers+21, "timeouts", h_timeouts, 2, 0},
  1266. {commandhandlers+22, "auth", h_auth, 2, 0},
  1267. {commandhandlers+23, "users", h_users, 2, 0},
  1268. {commandhandlers+24, "maxconn", h_maxconn, 2, 2},
  1269. {commandhandlers+25, "flush", h_flush, 1, 1},
  1270. {commandhandlers+26, "nserver", h_nserver, 2, 2},
  1271. {commandhandlers+27, "fakeresolve", h_fakeresolve, 1, 1},
  1272. {commandhandlers+28, "nscache", h_nscache, 2, 2},
  1273. {commandhandlers+29, "nscache6", h_nscache6, 2, 2},
  1274. {commandhandlers+30, "nsrecord", h_nsrecord, 3, 3},
  1275. {commandhandlers+31, "dialer", h_dialer, 2, 2},
  1276. {commandhandlers+32, "system", h_system, 2, 2},
  1277. {commandhandlers+33, "pidfile", h_pidfile, 2, 2},
  1278. {commandhandlers+34, "monitor", h_monitor, 2, 2},
  1279. {commandhandlers+35, "parent", h_parent, 5, 0},
  1280. {commandhandlers+36, "allow", h_ace, 1, 0},
  1281. {commandhandlers+37, "deny", h_ace, 1, 0},
  1282. {commandhandlers+38, "redirect", h_ace, 3, 0},
  1283. {commandhandlers+39, "bandlimin", h_ace, 2, 0},
  1284. {commandhandlers+40, "bandlimout", h_ace, 2, 0},
  1285. {commandhandlers+41, "nobandlimin", h_ace, 1, 0},
  1286. {commandhandlers+42, "nobandlimout", h_ace, 1, 0},
  1287. {commandhandlers+43, "countin", h_ace, 4, 0},
  1288. {commandhandlers+44, "nocountin", h_ace, 1, 0},
  1289. {commandhandlers+45, "countout", h_ace, 4, 0},
  1290. {commandhandlers+46, "nocountout", h_ace, 1, 0},
  1291. {commandhandlers+47, "plugin", h_plugin, 3, 0},
  1292. {commandhandlers+48, "logdump", h_logdump, 2, 3},
  1293. {commandhandlers+49, "filtermaxsize", h_filtermaxsize, 2, 2},
  1294. {commandhandlers+50, "nolog", h_nolog, 1, 1},
  1295. {commandhandlers+51, "weight", h_nolog, 2, 2},
  1296. {commandhandlers+52, "authcache", h_authcache, 2, 3},
  1297. {commandhandlers+53, "smtpp", h_proxy, 1, 0},
  1298. {commandhandlers+54, "icqpr", h_proxy, 4, 0},
  1299. {commandhandlers+55, "msnpr", h_proxy, 4, 0},
  1300. {commandhandlers+56, "delimchar",h_delimchar, 2, 2},
  1301. {commandhandlers+57, "authnserver", h_authnserver, 2, 2},
  1302. {commandhandlers+58, "stacksize", h_stacksize, 2, 2},
  1303. {commandhandlers+59, "force", h_force, 1, 1},
  1304. {commandhandlers+60, "noforce", h_noforce, 1, 1},
  1305. {specificcommands, "", h_noop, 1, 0}
  1306. };
  1307. int parsestr (unsigned char *str, unsigned char **argm, int nitems, unsigned char ** buff, int *inbuf, int *bufsize){
  1308. #define buf (*buff)
  1309. int argc = 0;
  1310. int space = 1;
  1311. int comment = 0;
  1312. unsigned char * incbegin = 0;
  1313. int fd;
  1314. int res, len;
  1315. int i = 1;
  1316. unsigned char *str1;
  1317. for(;;str++){
  1318. if(*str == '\"'){
  1319. str1 = str;
  1320. do {
  1321. *str1 = *(str1 + 1);
  1322. }while(*(str1++));
  1323. if(!comment || *str != '\"'){
  1324. comment = !comment;
  1325. }
  1326. }
  1327. switch(*str){
  1328. case '\0':
  1329. if(comment) return -1;
  1330. argm[argc] = 0;
  1331. return argc;
  1332. case '$':
  1333. if(!comment && !included){
  1334. incbegin = str;
  1335. *str = 0;
  1336. }
  1337. break;
  1338. case '\r':
  1339. case '\n':
  1340. case '\t':
  1341. case ' ':
  1342. if(!comment){
  1343. *str = 0;
  1344. space = 1;
  1345. i = 0;
  1346. if(incbegin){
  1347. argc--;
  1348. if((fd = open((char *)incbegin+1, O_RDONLY)) <= 0){
  1349. fprintf(stderr, "Failed to open %s\n", incbegin+1);
  1350. break;
  1351. }
  1352. if((*bufsize - *inbuf) <STRINGBUF){
  1353. *bufsize += STRINGBUF;
  1354. if(!(buf = myrealloc(buf, *bufsize))){
  1355. fprintf(stderr, "Failed to allocate memory for %s\n", incbegin+1);
  1356. close(fd);
  1357. break;
  1358. }
  1359. }
  1360. len = 0;
  1361. if(argm[argc]!=(incbegin+1)) {
  1362. len = (int)strlen((char *)argm[argc]);
  1363. memmove(buf+*inbuf, argm[argc], len);
  1364. }
  1365. if((res = read(fd, buf+*inbuf+len, STRINGBUF-(1+len))) <= 0) {
  1366. perror((char *)incbegin+1);
  1367. close(fd);
  1368. break;
  1369. }
  1370. close(fd);
  1371. buf[*inbuf+res+len] = 0;
  1372. incbegin = buf + *inbuf;
  1373. (*inbuf) += (res + len + 1);
  1374. included++;
  1375. argc+=parsestr(incbegin, argm + argc, nitems - argc, buff, inbuf, bufsize);
  1376. included--;
  1377. incbegin = NULL;
  1378. }
  1379. break;
  1380. }
  1381. default:
  1382. i++;
  1383. if(space) {
  1384. if(comment && *str == '\"' && str[1] != '\"'){
  1385. str++;
  1386. comment = 0;
  1387. }
  1388. argm[argc++] = str;
  1389. if(argc >= nitems) return argc;
  1390. space = 0;
  1391. }
  1392. }
  1393. }
  1394. #undef buf
  1395. }
  1396. int readconfig(FILE * fp){
  1397. unsigned char ** argv = NULL;
  1398. unsigned char * buf = NULL;
  1399. int bufsize = STRINGBUF*2;
  1400. int inbuf = 0;
  1401. int argc;
  1402. struct commands * cm;
  1403. int res = 0;
  1404. if( !(buf = myalloc(bufsize)) || ! (argv = myalloc((NPARAMS + 1) * sizeof(unsigned char *))) ) {
  1405. fprintf(stderr, "No memory for configuration");
  1406. return(10);
  1407. }
  1408. for (linenum = 1; fgets((char *)buf, STRINGBUF, fp); linenum++){
  1409. if(!*buf || isspace(*buf) || (*buf) == '#')continue;
  1410. inbuf = (int)(strlen((char *)buf) + 1);
  1411. argc = parsestr (buf, argv, NPARAMS-1, &buf, &inbuf, &bufsize);
  1412. if(argc < 1) {
  1413. fprintf(stderr, "Parse error line %d\n", linenum);
  1414. return(21);
  1415. }
  1416. argv[argc] = NULL;
  1417. if(!strcmp((char *)argv[0], "end") && argc == 1) {
  1418. break;
  1419. }
  1420. else if(!strcmp((char *)argv[0], "writable") && argc == 1) {
  1421. if(!writable){
  1422. writable = freopen(curconf, "r+", fp);
  1423. if(!writable){
  1424. fprintf(stderr, "Unable to reopen config for writing: %s\n", curconf);
  1425. return 1;
  1426. }
  1427. }
  1428. continue;
  1429. }
  1430. res = 1;
  1431. for(cm = commandhandlers; cm; cm = cm->next){
  1432. if(!strcmp((char *)argv[0], (char *)cm->command) && argc >= cm->minargs && (!cm->maxargs || argc <= cm->maxargs)){
  1433. res = (*cm->handler)(argc, argv);
  1434. if(res > 0){
  1435. fprintf(stderr, "Command: '%s' failed with code %d, line %d\n", argv[0], res, linenum);
  1436. return(linenum);
  1437. }
  1438. if(!res) break;
  1439. }
  1440. }
  1441. if(res != 1)continue;
  1442. fprintf(stderr, "Unknown command: '%s' line %d\n", argv[0], linenum);
  1443. return(linenum);
  1444. }
  1445. myfree(buf);
  1446. myfree(argv);
  1447. return 0;
  1448. }
  1449. void freepwl(struct passwords *pwl){
  1450. for(; pwl; pwl = (struct passwords *)itfree(pwl, pwl->next)){
  1451. if(pwl->user)myfree(pwl->user);
  1452. if(pwl->password)myfree(pwl->password);
  1453. }
  1454. }
  1455. void freeconf(struct extparam *confp){
  1456. struct bandlim * bl;
  1457. struct bandlim * blout;
  1458. struct trafcount * tc;
  1459. struct passwords *pw;
  1460. struct ace *acl;
  1461. struct filemon *fm;
  1462. int counterd, archiverc;
  1463. unsigned char *logname, *logtarget;
  1464. unsigned char **archiver;
  1465. unsigned char * logformat;
  1466. int i;
  1467. pthread_mutex_lock(&tc_mutex);
  1468. confp->trafcountfunc = NULL;
  1469. tc = confp->trafcounter;
  1470. confp->trafcounter = NULL;
  1471. counterd = confp->counterd;
  1472. confp->counterd = -1;
  1473. confp->countertype = NONE;
  1474. pthread_mutex_unlock(&tc_mutex);
  1475. pthread_mutex_lock(&bandlim_mutex);
  1476. bl = confp->bandlimiter;
  1477. blout = confp->bandlimiterout;
  1478. confp->bandlimiter = NULL;
  1479. confp->bandlimiterout = NULL;
  1480. confp->bandlimfunc = NULL;
  1481. pthread_mutex_unlock(&bandlim_mutex);
  1482. pthread_mutex_lock(&pwl_mutex);
  1483. pw = confp->pwl;
  1484. confp->pwl = NULL;
  1485. pthread_mutex_unlock(&pwl_mutex);
  1486. logtarget = confp->logtarget;
  1487. confp->logtarget = NULL;
  1488. logformat = confp->logformat;
  1489. confp->logformat = NULL;
  1490. logname = confp->logname;
  1491. confp->logname = NULL;
  1492. confp->rotate = 0;
  1493. confp->logtype = NONE;
  1494. archiverc = confp->archiverc;
  1495. confp->archiverc = 0;
  1496. archiver = confp->archiver;
  1497. confp->archiver = NULL;
  1498. fm = confp->fmon;
  1499. confp->fmon = NULL;
  1500. confp->bandlimfunc = NULL;
  1501. memset(&confp->intsa, 0, sizeof(confp->intsa));
  1502. memset(&confp->extsa, 0, sizeof(confp->extsa));
  1503. #ifndef NOIPV6
  1504. memset(&confp->extsa6, 0, sizeof(confp->extsa6));
  1505. *SAFAMILY(&confp->extsa6) = AF_INET6;
  1506. #endif
  1507. *SAFAMILY(&confp->intsa) = AF_INET;
  1508. *SAFAMILY(&confp->extsa) = AF_INET;
  1509. confp->maxchild = 100;
  1510. resolvfunc = NULL;
  1511. numservers = 0;
  1512. acl = confp->acl;
  1513. confp->acl = NULL;
  1514. confp->logtime = confp->time = 0;
  1515. usleep(SLEEPTIME);
  1516. {
  1517. char * args[] = {"auth", "iponly", NULL};
  1518. h_auth(2, (unsigned char **)args);
  1519. }
  1520. if(tc)dumpcounters(tc,counterd);
  1521. for(; tc; tc = (struct trafcount *) itfree(tc, tc->next)){
  1522. if(tc->comment)myfree(tc->comment);
  1523. freeacl(tc->ace);
  1524. }
  1525. freeacl(acl);
  1526. freepwl(pw);
  1527. for(; bl; bl = (struct bandlim *) itfree(bl, bl->next)) freeacl(bl->ace);
  1528. for(; blout; blout = (struct bandlim *) itfree(blout, blout->next))freeacl(blout->ace);
  1529. if(counterd != -1) {
  1530. close(counterd);
  1531. }
  1532. for(; fm; fm = (struct filemon *)itfree(fm, fm->next)){
  1533. if(fm->path) myfree(fm->path);
  1534. }
  1535. if(logtarget) {
  1536. myfree(logtarget);
  1537. }
  1538. if(logname) {
  1539. myfree(logname);
  1540. }
  1541. if(logformat) {
  1542. myfree(logformat);
  1543. }
  1544. if(archiver) {
  1545. for(i = 0; i < archiverc; i++) myfree(archiver[i]);
  1546. myfree(archiver);
  1547. }
  1548. }
  1549. int reload (void){
  1550. FILE *fp;
  1551. int error = -2;
  1552. conf.paused++;
  1553. freeconf(&conf);
  1554. conf.paused++;
  1555. fp = confopen();
  1556. if(fp){
  1557. error = readconfig(fp);
  1558. conf.version++;
  1559. if(error) {
  1560. freeconf(&conf);
  1561. }
  1562. if(!writable)fclose(fp);
  1563. }
  1564. return error;
  1565. }