conf.c 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730
  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 = myalloc(sizeof(struct chain));
  629. if(!chains){
  630. fprintf(stderr, "Chainig error: unable to allocate memory for chain\n");
  631. return(2);
  632. }
  633. memset(chains, 0, sizeof(struct chain));
  634. chains->weight = (unsigned)atoi((char *)argv[1]);
  635. if(chains->weight == 0 || chains->weight >1000) {
  636. fprintf(stderr, "Chaining error: bad chain weight %u line %d\n", chains->weight, linenum);
  637. return(3);
  638. }
  639. if(!strcmp((char *)argv[2], "tcp"))chains->type = R_TCP;
  640. else if(!strcmp((char *)argv[2], "http"))chains->type = R_HTTP;
  641. else if(!strcmp((char *)argv[2], "connect"))chains->type = R_CONNECT;
  642. else if(!strcmp((char *)argv[2], "socks4"))chains->type = R_SOCKS4;
  643. else if(!strcmp((char *)argv[2], "socks5"))chains->type = R_SOCKS5;
  644. else if(!strcmp((char *)argv[2], "connect+"))chains->type = R_CONNECTP;
  645. else if(!strcmp((char *)argv[2], "socks4+"))chains->type = R_SOCKS4P;
  646. else if(!strcmp((char *)argv[2], "socks5+"))chains->type = R_SOCKS5P;
  647. else if(!strcmp((char *)argv[2], "socks4b"))chains->type = R_SOCKS4B;
  648. else if(!strcmp((char *)argv[2], "socks5b"))chains->type = R_SOCKS5B;
  649. else if(!strcmp((char *)argv[2], "pop3"))chains->type = R_POP3;
  650. else if(!strcmp((char *)argv[2], "ftp"))chains->type = R_FTP;
  651. else if(!strcmp((char *)argv[2], "admin"))chains->type = R_ADMIN;
  652. else if(!strcmp((char *)argv[2], "icq"))chains->type = R_ICQ;
  653. else if(!strcmp((char *)argv[2], "extip"))chains->type = R_EXTIP;
  654. else if(!strcmp((char *)argv[2], "smtp"))chains->type = R_SMTP;
  655. else {
  656. fprintf(stderr, "Chaining error: bad chain type (%s)\n", argv[2]);
  657. return(4);
  658. }
  659. if(!getip46(46, argv[3], (struct sockaddr *)&chains->addr)) return 5;
  660. chains->exthost = (unsigned char *)mystrdup((char *)argv[3]);
  661. *SAPORT(&chains->addr) = htons((unsigned short)atoi((char *)argv[4]));
  662. if(argc > 5) chains->extuser = (unsigned char *)mystrdup((char *)argv[5]);
  663. if(argc > 6) chains->extpass = (unsigned char *)mystrdup((char *)argv[6]);
  664. if(!acl->chains) {
  665. acl->chains = chains;
  666. }
  667. else {
  668. struct chain *tmpchain;
  669. for(tmpchain = acl->chains; tmpchain->next; tmpchain = tmpchain->next);
  670. tmpchain->next = chains;
  671. }
  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. case ALLOW:
  1036. case DENY:
  1037. if(!conf.acl){
  1038. conf.acl = acl;
  1039. }
  1040. else {
  1041. struct ace * acei;
  1042. for(acei = conf.acl; acei->next; acei = acei->next);
  1043. acei->next = acl;
  1044. }
  1045. break;
  1046. case BANDLIM:
  1047. case NOBANDLIM:
  1048. nbl = myalloc(sizeof(struct bandlim));
  1049. if(!nbl) {
  1050. fprintf(stderr, "No memory to create band limit filter\n");
  1051. return(3);
  1052. }
  1053. memset(nbl, 0, sizeof(struct bandlim));
  1054. nbl->ace = acl;
  1055. if(acl->action == BANDLIM) {
  1056. sscanf((char *)argv[1], "%u", &nbl->rate);
  1057. if(nbl->rate < 300) {
  1058. fprintf(stderr, "Wrong bandwidth specified, line %d\n", linenum);
  1059. return(4);
  1060. }
  1061. }
  1062. pthread_mutex_lock(&bandlim_mutex);
  1063. if(!strcmp((char *)argv[0], "bandlimin") || !strcmp((char *)argv[0], "nobandlimin")){
  1064. if(!conf.bandlimiter){
  1065. conf.bandlimiter = nbl;
  1066. }
  1067. else {
  1068. struct bandlim * bli;
  1069. for(bli = conf.bandlimiter; bli->next; bli = bli->next);
  1070. bli->next = nbl;
  1071. }
  1072. }
  1073. else {
  1074. if(!conf.bandlimiterout){
  1075. conf.bandlimiterout = nbl;
  1076. }
  1077. else {
  1078. struct bandlim * bli;
  1079. for(bli = conf.bandlimiterout; bli->next; bli = bli->next);
  1080. bli->next = nbl;
  1081. }
  1082. }
  1083. pthread_mutex_unlock(&bandlim_mutex);
  1084. break;
  1085. case COUNTIN:
  1086. case NOCOUNTIN:
  1087. case COUNTOUT:
  1088. case NOCOUNTOUT:
  1089. tl = myalloc(sizeof(struct trafcount));
  1090. if(!tl) {
  1091. fprintf(stderr, "No memory to create traffic limit filter\n");
  1092. return(5);
  1093. }
  1094. memset(tl, 0, sizeof(struct trafcount));
  1095. tl->ace = acl;
  1096. if((acl->action == COUNTIN)||(acl->action == COUNTOUT)) {
  1097. unsigned long lim;
  1098. tl->comment = ( char *)argv[1];
  1099. while(isdigit(*tl->comment))tl->comment++;
  1100. if(*tl->comment== '/')tl->comment++;
  1101. tl->comment = mystrdup(tl->comment);
  1102. sscanf((char *)argv[1], "%u", &tl->number);
  1103. sscanf((char *)argv[3], "%lu", &lim);
  1104. tl->type = getrotate(*argv[2]);
  1105. tl->traflim64 = ((uint64_t)lim)*(1024*1024);
  1106. if(!tl->traflim64) {
  1107. fprintf(stderr, "Wrong traffic limit specified, line %d\n", linenum);
  1108. return(6);
  1109. }
  1110. if(tl->number != 0 && conf.counterd >= 0) {
  1111. lseek(conf.counterd,
  1112. sizeof(struct counter_header) + (tl->number - 1) * sizeof(struct counter_record),
  1113. SEEK_SET);
  1114. memset(&crecord, 0, sizeof(struct counter_record));
  1115. read(conf.counterd, &crecord, sizeof(struct counter_record));
  1116. tl->traf64 = crecord.traf64;
  1117. tl->cleared = crecord.cleared;
  1118. tl->updated = crecord.updated;
  1119. if(tl->cleared < 0 || tl->cleared >= MAX_COUNTER_TIME || tl->updated < 0 || tl->updated >= MAX_COUNTER_TIME){
  1120. fprintf(stderr, "Invalid, incompatible or corrupted counter file.\n");
  1121. return(6);
  1122. }
  1123. }
  1124. }
  1125. pthread_mutex_lock(&tc_mutex);
  1126. if(!conf.trafcounter){
  1127. conf.trafcounter = tl;
  1128. }
  1129. else {
  1130. struct trafcount * ntl;
  1131. for(ntl = conf.trafcounter; ntl->next; ntl = ntl->next);
  1132. ntl->next = tl;
  1133. }
  1134. pthread_mutex_unlock(&tc_mutex);
  1135. }
  1136. return 0;
  1137. }
  1138. static int h_logdump(int argc, unsigned char **argv){
  1139. conf.logdumpsrv = (unsigned) atoi((char *) *(argv + 1));
  1140. if(argc > 2) conf.logdumpcli = (unsigned) atoi((char *) *(argv + 2));
  1141. return 0;
  1142. }
  1143. static int h_filtermaxsize(int argc, unsigned char **argv){
  1144. conf.filtermaxsize = atoi((char *) *(argv + 1));
  1145. return 0;
  1146. }
  1147. static int h_delimchar(int argc, unsigned char **argv){
  1148. conf.delimchar = *argv[1];
  1149. return 0;
  1150. }
  1151. static int h_radius(int argc, unsigned char **argv){
  1152. char * rs = radiussecret;
  1153. radiussecret = mystrdup(argv[1]);
  1154. nradservers = 0;
  1155. memset(radiuslist, 0, sizeof(radiuslist));
  1156. for( ; nradservers < MAXRADIUS && nradservers < argc -2; nradservers++){
  1157. if( !getip46(46, argv[nradservers + 2], (struct sockaddr *)&radiuslist[nradservers])) return 1;
  1158. }
  1159. return 0;
  1160. }
  1161. static int h_authcache(int argc, unsigned char **argv){
  1162. conf.authcachetype = 0;
  1163. if(strstr((char *) *(argv + 1), "ip")) conf.authcachetype |= 1;
  1164. if(strstr((char *) *(argv + 1), "user")) conf.authcachetype |= 2;
  1165. if(strstr((char *) *(argv + 1), "pass")) conf.authcachetype |= 4;
  1166. if(argc > 2) conf.authcachetime = (unsigned) atoi((char *) *(argv + 2));
  1167. if(!conf.authcachetype) conf.authcachetype = 6;
  1168. if(!conf.authcachetime) conf.authcachetime = 600;
  1169. return 0;
  1170. }
  1171. static int h_plugin(int argc, unsigned char **argv){
  1172. #ifdef NOPLUGINS
  1173. return 999;
  1174. #else
  1175. #ifdef _WIN32
  1176. HINSTANCE hi;
  1177. FARPROC fp;
  1178. #ifdef _WINCE
  1179. hi = LoadLibraryW((LPCWSTR)CEToUnicode(argv[1]));
  1180. #else
  1181. hi = LoadLibrary((char *)argv[1]);
  1182. #endif
  1183. if(!hi) {
  1184. fprintf(stderr, "Failed to load %s, code %d\n", argv[1], (int)GetLastError());
  1185. return 1;
  1186. }
  1187. #ifdef _WINCE
  1188. fp = GetProcAddressW(hi, (LPCWSTR)CEToUnicode(argv[2]));
  1189. #else
  1190. fp = GetProcAddress(hi, (char *)argv[2]);
  1191. #endif
  1192. if(!fp) {
  1193. printf("%s not found in %s, code: %d\n", argv[2], argv[1], (int)GetLastError());
  1194. return 2;
  1195. }
  1196. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1197. #else
  1198. void *hi, *fp;
  1199. hi = dlopen((char *)argv[1], RTLD_LAZY);
  1200. if(!hi) return 1;
  1201. fp = dlsym(hi, (char *)argv[2]);
  1202. if(!fp) return 2;
  1203. return (*(PLUGINFUNC)fp)(&pluginlink, argc - 2, (char **)argv + 2);
  1204. #endif
  1205. #endif
  1206. }
  1207. #ifndef _WIN32
  1208. static int h_setuid(int argc, unsigned char **argv){
  1209. int res;
  1210. res = atoi((char *)argv[1]);
  1211. if(!res || setreuid(res,res)) {
  1212. fprintf(stderr, "Unable to set uid %d", res);
  1213. return(1);
  1214. }
  1215. return 0;
  1216. }
  1217. static int h_setgid(int argc, unsigned char **argv){
  1218. int res;
  1219. res = atoi((char *)argv[1]);
  1220. if(!res || setregid(res,res)) {
  1221. fprintf(stderr, "Unable to set gid %d", res);
  1222. return(1);
  1223. }
  1224. return 0;
  1225. }
  1226. static int h_chroot(int argc, unsigned char **argv){
  1227. if(!chrootp){
  1228. char *p;
  1229. if(chroot((char *)argv[1])) {
  1230. fprintf(stderr, "Unable to chroot %s", argv[1]);
  1231. return(1);
  1232. }
  1233. p = (char *)argv[1] + strlen((char *)argv[1]) ;
  1234. while (p > (char *)argv[1] && p[-1] == '/'){
  1235. p--;
  1236. *p = 0;
  1237. }
  1238. chrootp = mystrdup((char *)argv[1]);
  1239. }
  1240. return 0;
  1241. }
  1242. #endif
  1243. struct commands specificcommands[]={
  1244. #ifndef _WIN32
  1245. {specificcommands+1, "setuid", h_setuid, 2, 2},
  1246. {specificcommands+2, "setgid", h_setgid, 2, 2},
  1247. {specificcommands+3, "chroot", h_chroot, 2, 2},
  1248. #endif
  1249. {NULL, "", h_noop, 1, 0}
  1250. };
  1251. struct commands commandhandlers[]={
  1252. {commandhandlers+1, "", h_noop, 1, 0},
  1253. {commandhandlers+2, "proxy", h_proxy, 1, 0},
  1254. {commandhandlers+3, "pop3p", h_proxy, 1, 0},
  1255. {commandhandlers+4, "ftppr", h_proxy, 1, 0},
  1256. {commandhandlers+5, "socks", h_proxy, 1, 0},
  1257. {commandhandlers+6, "tcppm", h_proxy, 4, 0},
  1258. {commandhandlers+7, "udppm", h_proxy, 4, 0},
  1259. {commandhandlers+8, "admin", h_proxy, 1, 0},
  1260. {commandhandlers+9, "dnspr", h_proxy, 1, 0},
  1261. {commandhandlers+10, "internal", h_internal, 2, 2},
  1262. {commandhandlers+11, "external", h_external, 2, 2},
  1263. {commandhandlers+12, "log", h_log, 1, 0},
  1264. {commandhandlers+13, "service", h_service, 1, 1},
  1265. {commandhandlers+14, "daemon", h_daemon, 1, 1},
  1266. {commandhandlers+15, "config", h_config, 2, 2},
  1267. {commandhandlers+16, "include", h_include, 2, 2},
  1268. {commandhandlers+17, "archiver", h_archiver, 3, 0},
  1269. {commandhandlers+18, "counter", h_counter, 2, 4},
  1270. {commandhandlers+19, "rotate", h_rotate, 2, 2},
  1271. {commandhandlers+20, "logformat", h_logformat, 2, 2},
  1272. {commandhandlers+21, "timeouts", h_timeouts, 2, 0},
  1273. {commandhandlers+22, "auth", h_auth, 2, 0},
  1274. {commandhandlers+23, "users", h_users, 2, 0},
  1275. {commandhandlers+24, "maxconn", h_maxconn, 2, 2},
  1276. {commandhandlers+25, "flush", h_flush, 1, 1},
  1277. {commandhandlers+26, "nserver", h_nserver, 2, 2},
  1278. {commandhandlers+27, "fakeresolve", h_fakeresolve, 1, 1},
  1279. {commandhandlers+28, "nscache", h_nscache, 2, 2},
  1280. {commandhandlers+29, "nscache6", h_nscache6, 2, 2},
  1281. {commandhandlers+30, "nsrecord", h_nsrecord, 3, 3},
  1282. {commandhandlers+31, "dialer", h_dialer, 2, 2},
  1283. {commandhandlers+32, "system", h_system, 2, 2},
  1284. {commandhandlers+33, "pidfile", h_pidfile, 2, 2},
  1285. {commandhandlers+34, "monitor", h_monitor, 2, 2},
  1286. {commandhandlers+35, "parent", h_parent, 5, 0},
  1287. {commandhandlers+36, "allow", h_ace, 1, 0},
  1288. {commandhandlers+37, "deny", h_ace, 1, 0},
  1289. {commandhandlers+38, "redirect", h_ace, 3, 0},
  1290. {commandhandlers+39, "bandlimin", h_ace, 2, 0},
  1291. {commandhandlers+40, "bandlimout", h_ace, 2, 0},
  1292. {commandhandlers+41, "nobandlimin", h_ace, 1, 0},
  1293. {commandhandlers+42, "nobandlimout", h_ace, 1, 0},
  1294. {commandhandlers+43, "countin", h_ace, 4, 0},
  1295. {commandhandlers+44, "nocountin", h_ace, 1, 0},
  1296. {commandhandlers+45, "countout", h_ace, 4, 0},
  1297. {commandhandlers+46, "nocountout", h_ace, 1, 0},
  1298. {commandhandlers+47, "plugin", h_plugin, 3, 0},
  1299. {commandhandlers+48, "logdump", h_logdump, 2, 3},
  1300. {commandhandlers+49, "filtermaxsize", h_filtermaxsize, 2, 2},
  1301. {commandhandlers+50, "nolog", h_nolog, 1, 1},
  1302. {commandhandlers+51, "weight", h_nolog, 2, 2},
  1303. {commandhandlers+52, "authcache", h_authcache, 2, 3},
  1304. {commandhandlers+53, "smtpp", h_proxy, 1, 0},
  1305. {commandhandlers+54, "icqpr", h_proxy, 4, 0},
  1306. {commandhandlers+55, "msnpr", h_proxy, 4, 0},
  1307. {commandhandlers+56, "delimchar",h_delimchar, 2, 2},
  1308. {commandhandlers+57, "authnserver", h_authnserver, 2, 2},
  1309. {commandhandlers+58, "stacksize", h_stacksize, 2, 2},
  1310. {commandhandlers+59, "force", h_force, 1, 1},
  1311. {commandhandlers+60, "noforce", h_noforce, 1, 1},
  1312. #ifndef NORADIUS
  1313. {commandhandlers+61, "radius", h_radius, 3, 0},
  1314. #endif
  1315. {specificcommands, "", h_noop, 1, 0}
  1316. };
  1317. int parsestr (unsigned char *str, unsigned char **argm, int nitems, unsigned char ** buff, int *inbuf, int *bufsize){
  1318. #define buf (*buff)
  1319. int argc = 0;
  1320. int space = 1;
  1321. int comment = 0;
  1322. unsigned char * incbegin = 0;
  1323. int fd;
  1324. int res, len;
  1325. int i = 1;
  1326. unsigned char *str1;
  1327. for(;;str++){
  1328. if(*str == '\"'){
  1329. str1 = str;
  1330. do {
  1331. *str1 = *(str1 + 1);
  1332. }while(*(str1++));
  1333. if(!comment || *str != '\"'){
  1334. comment = !comment;
  1335. }
  1336. }
  1337. switch(*str){
  1338. case '\0':
  1339. if(comment) return -1;
  1340. argm[argc] = 0;
  1341. return argc;
  1342. case '$':
  1343. if(!comment && !included){
  1344. incbegin = str;
  1345. *str = 0;
  1346. }
  1347. break;
  1348. case '\r':
  1349. case '\n':
  1350. case '\t':
  1351. case ' ':
  1352. if(!comment){
  1353. *str = 0;
  1354. space = 1;
  1355. i = 0;
  1356. if(incbegin){
  1357. argc--;
  1358. if((fd = open((char *)incbegin+1, O_RDONLY)) <= 0){
  1359. fprintf(stderr, "Failed to open %s\n", incbegin+1);
  1360. break;
  1361. }
  1362. if((*bufsize - *inbuf) <STRINGBUF){
  1363. *bufsize += STRINGBUF;
  1364. if(!(buf = myrealloc(buf, *bufsize))){
  1365. fprintf(stderr, "Failed to allocate memory for %s\n", incbegin+1);
  1366. close(fd);
  1367. break;
  1368. }
  1369. }
  1370. len = 0;
  1371. if(argm[argc]!=(incbegin+1)) {
  1372. len = (int)strlen((char *)argm[argc]);
  1373. memmove(buf+*inbuf, argm[argc], len);
  1374. }
  1375. if((res = read(fd, buf+*inbuf+len, STRINGBUF-(1+len))) <= 0) {
  1376. perror((char *)incbegin+1);
  1377. close(fd);
  1378. break;
  1379. }
  1380. close(fd);
  1381. buf[*inbuf+res+len] = 0;
  1382. incbegin = buf + *inbuf;
  1383. (*inbuf) += (res + len + 1);
  1384. included++;
  1385. argc+=parsestr(incbegin, argm + argc, nitems - argc, buff, inbuf, bufsize);
  1386. included--;
  1387. incbegin = NULL;
  1388. }
  1389. break;
  1390. }
  1391. default:
  1392. i++;
  1393. if(space) {
  1394. if(comment && *str == '\"' && str[1] != '\"'){
  1395. str++;
  1396. comment = 0;
  1397. }
  1398. argm[argc++] = str;
  1399. if(argc >= nitems) return argc;
  1400. space = 0;
  1401. }
  1402. }
  1403. }
  1404. #undef buf
  1405. }
  1406. int readconfig(FILE * fp){
  1407. unsigned char ** argv = NULL;
  1408. unsigned char * buf = NULL;
  1409. int bufsize = STRINGBUF*2;
  1410. int inbuf = 0;
  1411. int argc;
  1412. struct commands * cm;
  1413. int res = 0;
  1414. if( !(buf = myalloc(bufsize)) || ! (argv = myalloc((NPARAMS + 1) * sizeof(unsigned char *))) ) {
  1415. fprintf(stderr, "No memory for configuration");
  1416. return(10);
  1417. }
  1418. for (linenum = 1; fgets((char *)buf, STRINGBUF, fp); linenum++){
  1419. if(!*buf || isspace(*buf) || (*buf) == '#')continue;
  1420. inbuf = (int)(strlen((char *)buf) + 1);
  1421. argc = parsestr (buf, argv, NPARAMS-1, &buf, &inbuf, &bufsize);
  1422. if(argc < 1) {
  1423. fprintf(stderr, "Parse error line %d\n", linenum);
  1424. return(21);
  1425. }
  1426. argv[argc] = NULL;
  1427. if(!strcmp((char *)argv[0], "end") && argc == 1) {
  1428. break;
  1429. }
  1430. else if(!strcmp((char *)argv[0], "writable") && argc == 1) {
  1431. if(!writable){
  1432. writable = freopen(curconf, "r+", fp);
  1433. if(!writable){
  1434. fprintf(stderr, "Unable to reopen config for writing: %s\n", curconf);
  1435. return 1;
  1436. }
  1437. }
  1438. continue;
  1439. }
  1440. res = 1;
  1441. for(cm = commandhandlers; cm; cm = cm->next){
  1442. if(!strcmp((char *)argv[0], (char *)cm->command) && argc >= cm->minargs && (!cm->maxargs || argc <= cm->maxargs)){
  1443. res = (*cm->handler)(argc, argv);
  1444. if(res > 0){
  1445. fprintf(stderr, "Command: '%s' failed with code %d, line %d\n", argv[0], res, linenum);
  1446. return(linenum);
  1447. }
  1448. if(!res) break;
  1449. }
  1450. }
  1451. if(res != 1)continue;
  1452. fprintf(stderr, "Unknown command: '%s' line %d\n", argv[0], linenum);
  1453. return(linenum);
  1454. }
  1455. myfree(buf);
  1456. myfree(argv);
  1457. return 0;
  1458. }
  1459. void freepwl(struct passwords *pwl){
  1460. for(; pwl; pwl = (struct passwords *)itfree(pwl, pwl->next)){
  1461. if(pwl->user)myfree(pwl->user);
  1462. if(pwl->password)myfree(pwl->password);
  1463. }
  1464. }
  1465. void freeconf(struct extparam *confp){
  1466. struct bandlim * bl;
  1467. struct bandlim * blout;
  1468. struct trafcount * tc;
  1469. struct passwords *pw;
  1470. struct ace *acl;
  1471. struct filemon *fm;
  1472. int counterd, archiverc;
  1473. unsigned char *logname, *logtarget;
  1474. unsigned char **archiver;
  1475. unsigned char * logformat;
  1476. int i;
  1477. pthread_mutex_lock(&tc_mutex);
  1478. confp->trafcountfunc = NULL;
  1479. tc = confp->trafcounter;
  1480. confp->trafcounter = NULL;
  1481. counterd = confp->counterd;
  1482. confp->counterd = -1;
  1483. confp->countertype = NONE;
  1484. pthread_mutex_unlock(&tc_mutex);
  1485. pthread_mutex_lock(&bandlim_mutex);
  1486. bl = confp->bandlimiter;
  1487. blout = confp->bandlimiterout;
  1488. confp->bandlimiter = NULL;
  1489. confp->bandlimiterout = NULL;
  1490. confp->bandlimfunc = NULL;
  1491. pthread_mutex_unlock(&bandlim_mutex);
  1492. pthread_mutex_lock(&pwl_mutex);
  1493. pw = confp->pwl;
  1494. confp->pwl = NULL;
  1495. pthread_mutex_unlock(&pwl_mutex);
  1496. logtarget = confp->logtarget;
  1497. confp->logtarget = NULL;
  1498. logformat = confp->logformat;
  1499. confp->logformat = NULL;
  1500. logname = confp->logname;
  1501. confp->logname = NULL;
  1502. confp->rotate = 0;
  1503. confp->logtype = NONE;
  1504. archiverc = confp->archiverc;
  1505. confp->archiverc = 0;
  1506. archiver = confp->archiver;
  1507. confp->archiver = NULL;
  1508. fm = confp->fmon;
  1509. confp->fmon = NULL;
  1510. confp->bandlimfunc = NULL;
  1511. memset(&confp->intsa, 0, sizeof(confp->intsa));
  1512. memset(&confp->extsa, 0, sizeof(confp->extsa));
  1513. #ifndef NOIPV6
  1514. memset(&confp->extsa6, 0, sizeof(confp->extsa6));
  1515. *SAFAMILY(&confp->extsa6) = AF_INET6;
  1516. #endif
  1517. *SAFAMILY(&confp->intsa) = AF_INET;
  1518. *SAFAMILY(&confp->extsa) = AF_INET;
  1519. confp->maxchild = 100;
  1520. resolvfunc = NULL;
  1521. numservers = 0;
  1522. acl = confp->acl;
  1523. confp->acl = NULL;
  1524. confp->logtime = confp->time = 0;
  1525. usleep(SLEEPTIME);
  1526. {
  1527. char * args[] = {"auth", "iponly", NULL};
  1528. h_auth(2, (unsigned char **)args);
  1529. }
  1530. if(tc)dumpcounters(tc,counterd);
  1531. for(; tc; tc = (struct trafcount *) itfree(tc, tc->next)){
  1532. if(tc->comment)myfree(tc->comment);
  1533. freeacl(tc->ace);
  1534. }
  1535. freeacl(acl);
  1536. freepwl(pw);
  1537. for(; bl; bl = (struct bandlim *) itfree(bl, bl->next)) freeacl(bl->ace);
  1538. for(; blout; blout = (struct bandlim *) itfree(blout, blout->next))freeacl(blout->ace);
  1539. if(counterd != -1) {
  1540. close(counterd);
  1541. }
  1542. for(; fm; fm = (struct filemon *)itfree(fm, fm->next)){
  1543. if(fm->path) myfree(fm->path);
  1544. }
  1545. if(logtarget) {
  1546. myfree(logtarget);
  1547. }
  1548. if(logname) {
  1549. myfree(logname);
  1550. }
  1551. if(logformat) {
  1552. myfree(logformat);
  1553. }
  1554. if(archiver) {
  1555. for(i = 0; i < archiverc; i++) myfree(archiver[i]);
  1556. myfree(archiver);
  1557. }
  1558. }
  1559. int reload (void){
  1560. FILE *fp;
  1561. int error = -2;
  1562. conf.paused++;
  1563. freeconf(&conf);
  1564. conf.paused++;
  1565. fp = confopen();
  1566. if(fp){
  1567. error = readconfig(fp);
  1568. conf.version++;
  1569. if(error) {
  1570. freeconf(&conf);
  1571. }
  1572. if(!writable)fclose(fp);
  1573. }
  1574. return error;
  1575. }