conf.c 43 KB

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