proxymain.c 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2017 by Vladimir Dubrovin <3proxy@3proxy.ru>
  4. please read License Agreement
  5. */
  6. #include "proxy.h"
  7. #define param ((struct clientparam *) p)
  8. #ifdef _WIN32
  9. DWORD WINAPI threadfunc(LPVOID p) {
  10. #else
  11. void * threadfunc (void *p) {
  12. #endif
  13. int i = -1;
  14. if(param->srv->cbsock != INVALID_SOCKET){
  15. SASIZETYPE size = sizeof(param->sinsr);
  16. struct pollfd fds;
  17. fds.fd = param->srv->cbsock;
  18. fds.events = POLLIN;
  19. fds.revents = 0;
  20. for(i=5+(param->srv->maxchild>>10); i; i--){
  21. if(so._poll(&fds, 1, 1000*CONNBACK_TO)!=1){
  22. dolog(param, (unsigned char *)"Connect back not received, check connback client");
  23. i = 0;
  24. break;
  25. }
  26. param->remsock = so._accept(param->srv->cbsock, (struct sockaddr*)&param->sinsr, &size);
  27. if(param->remsock == INVALID_SOCKET) {
  28. dolog(param, (unsigned char *)"Connect back accept() failed");
  29. continue;
  30. }
  31. #ifndef WITHMAIN
  32. param->req = param->sinsr;
  33. if(param->srv->acl) param->res = checkACL(param);
  34. if(param->res){
  35. dolog(param, (unsigned char *)"Connect back ACL failed");
  36. so._closesocket(param->remsock);
  37. param->remsock = INVALID_SOCKET;
  38. continue;
  39. }
  40. #endif
  41. if(socksendto(param->remsock, (struct sockaddr*)&param->sinsr, (unsigned char *)"C", 1, CONNBACK_TO) != 1){
  42. dolog(param, (unsigned char *)"Connect back sending command failed");
  43. so._closesocket(param->remsock);
  44. param->remsock = INVALID_SOCKET;
  45. continue;
  46. }
  47. break;
  48. }
  49. }
  50. if(!i){
  51. param->res = 13;
  52. freeparam(param);
  53. }
  54. else {
  55. #ifndef _WIN32
  56. sigset_t mask;
  57. sigfillset(&mask);
  58. if(param->srv->service != S_UDPPM)pthread_sigmask(SIG_SETMASK, &mask, NULL);
  59. #endif
  60. ((struct clientparam *) p)->srv->pf((struct clientparam *)p);
  61. }
  62. #ifdef _WIN32
  63. return 0;
  64. #else
  65. return NULL;
  66. #endif
  67. }
  68. #undef param
  69. struct socketoptions sockopts[] = {
  70. #ifdef TCP_NODELAY
  71. {TCP_NODELAY, "TCP_NODELAY"},
  72. #endif
  73. #ifdef TCP_CORK
  74. {TCP_CORK, "TCP_CORK"},
  75. #endif
  76. #ifdef TCP_DEFER_ACCEPT
  77. {TCP_DEFER_ACCEPT, "TCP_DEFER_ACCEPT"},
  78. #endif
  79. #ifdef TCP_QUICKACK
  80. {TCP_QUICKACK, "TCP_QUICKACK"},
  81. #endif
  82. #ifdef TCP_TIMESTAMPS
  83. {TCP_TIMESTAMPS, "TCP_TIMESTAMPS"},
  84. #endif
  85. #ifdef USE_TCP_FASTOPEN
  86. {USE_TCP_FASTOPEN, "USE_TCP_FASTOPEN"},
  87. #endif
  88. #ifdef SO_REUSEADDR
  89. {SO_REUSEADDR, "SO_REUSEADDR"},
  90. #endif
  91. #ifdef SO_REUSEPORT
  92. {SO_REUSEPORT, "SO_REUSEPORT"},
  93. #endif
  94. #ifdef SO_PORT_SCALABILITY
  95. {SO_PORT_SCALABILITY, "SO_PORT_SCALABILITY"},
  96. #endif
  97. #ifdef SO_REUSE_UNICASTPORT
  98. {SO_REUSE_UNICASTPORT, "SO_REUSE_UNICASTPORT"},
  99. #endif
  100. #ifdef SO_KEEPALIVE
  101. {SO_KEEPALIVE, "SO_KEEPALIVE"},
  102. #endif
  103. #ifdef SO_DONTROUTE
  104. {SO_DONTROUTE, "SO_DONTROUTE"},
  105. #endif
  106. #ifdef IP_TRANSPARENT
  107. {IP_TRANSPARENT, "IP_TRANSPARENT"},
  108. #endif
  109. {0, NULL}
  110. };
  111. char optsbuf[1024];
  112. char * printopts(char *sep){
  113. int i=0, pos=0;
  114. for(; sockopts[i].optname; i++)pos += sprintf(optsbuf+pos,"%s%s",i?sep:"",sockopts[i].optname);
  115. return optsbuf;
  116. }
  117. int getopts(const char *s){
  118. int i=0, ret=0;
  119. for(; sockopts[i].optname; i++)if(strstr(s,sockopts[i].optname)) ret |= (1<<i);
  120. return ret;
  121. }
  122. void setopts(SOCKET s, int opts){
  123. int i, opt, set;
  124. for(i = 0; opts >= (opt = (1<<i)); i++){
  125. set = 1;
  126. if(opts & opt) setsockopt(s, *sockopts[i].optname == 'T'? IPPROTO_TCP:
  127. #ifdef SOL_IP
  128. *sockopts[i].optname == 'I'? SOL_IP:
  129. #endif
  130. SOL_SOCKET, sockopts[i].opt, (char *)&set, sizeof(set));
  131. }
  132. }
  133. #ifndef MODULEMAINFUNC
  134. #define MODULEMAINFUNC main
  135. #define STDMAIN
  136. #ifndef _WINCE
  137. int main (int argc, char** argv){
  138. #else
  139. int WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow){
  140. int argc;
  141. char ** argv;
  142. WNDCLASS wc;
  143. HWND hwnd = 0;
  144. #endif
  145. #else
  146. extern int linenum;
  147. extern int haveerror;
  148. int MODULEMAINFUNC (int argc, char** argv){
  149. #endif
  150. SOCKET sock = INVALID_SOCKET, new_sock = INVALID_SOCKET;
  151. int i=0;
  152. SASIZETYPE size;
  153. pthread_t thread;
  154. struct clientparam defparam;
  155. struct srvparam srv;
  156. struct clientparam * newparam;
  157. int error = 0;
  158. unsigned sleeptime;
  159. unsigned char buf[256];
  160. char *hostname=NULL;
  161. int opt = 1, isudp = 0, iscbl = 0, iscbc = 0;
  162. unsigned char *cbc_string = NULL, *cbl_string = NULL;
  163. #ifndef NOIPV6
  164. struct sockaddr_in6 cbsa;
  165. #else
  166. struct sockaddr_in cbsa;
  167. #endif
  168. FILE *fp = NULL;
  169. struct linger lg;
  170. int nlog = 5000;
  171. char loghelp[] =
  172. #ifdef STDMAIN
  173. #ifndef _WIN32
  174. " -I inetd mode (requires real socket, doesn't work with TTY)\n"
  175. " -l@IDENT log to syslog IDENT\n"
  176. #endif
  177. " -d go to background (daemon)\n"
  178. #else
  179. " -u never ask for username\n"
  180. " -u2 always ask for username\n"
  181. #endif
  182. #ifdef SO_BINDTODEVICE
  183. " -Di(DEVICENAME) bind internal interface to device, e.g. eth1\n"
  184. " -De(DEVICENAME) bind external interface to device, e.g. eth1\n"
  185. #endif
  186. #ifdef WITHSLICE
  187. " -s Use slice() - faster proxing, but no filtering for data\n"
  188. #endif
  189. " -fFORMAT logging format (see documentation)\n"
  190. " -l log to stderr\n"
  191. " -lFILENAME log to FILENAME\n"
  192. " -b(BUFSIZE) size of network buffer (default 4096 for TCP, 16384 for UDP)\n"
  193. " -S(STACKSIZE) value to add to default client thread stack size\n"
  194. " -t be silent (do not log service start/stop)\n"
  195. "\n"
  196. " -iIP ip address or internal interface (clients are expected to connect)\n"
  197. " -eIP ip address or external interface (outgoing connection will have this)\n"
  198. " -rHOST:PORT Use IP:port for connect back proxy instead of listen port\n"
  199. " -RHOST:PORT Use PORT to listen connect back proxy connection to pass data to\n"
  200. " -4 Use IPv4 for outgoing connections\n"
  201. " -6 Use IPv6 for outgoing connections\n"
  202. " -46 Prefer IPv4 for outgoing connections, use both IPv4 and IPv6\n"
  203. " -64 Prefer IPv6 for outgoing connections, use both IPv4 and IPv6\n"
  204. " -ocOPTIONS, -osOPTIONS, -olOPTIONS, -orOPTIONS -oROPTIONS - options for\n"
  205. " to-client (oc), to-server (os), listening (ol) socket, connect back client\n"
  206. " (or) socket, connect back server (oR) listening socket\n"
  207. " where possible options are: ";
  208. #ifdef _WIN32
  209. unsigned long ul = 1;
  210. #else
  211. pthread_attr_t pa;
  212. #ifdef STDMAIN
  213. int inetd = 0;
  214. #endif
  215. #endif
  216. #ifdef _WIN32
  217. HANDLE h;
  218. #endif
  219. #ifdef STDMAIN
  220. #ifdef _WINCE
  221. argc = ceparseargs((char *)lpCmdLine);
  222. argv = ceargv;
  223. if(FindWindow(lpCmdLine, lpCmdLine)) return 0;
  224. ZeroMemory(&wc,sizeof(wc));
  225. wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH);
  226. wc.hInstance=hInstance;
  227. wc.hCursor=LoadCursor(NULL,IDC_ARROW);
  228. wc.lpfnWndProc=DefWindowProc;
  229. wc.style=CS_HREDRAW|CS_VREDRAW;
  230. wc.lpszClassName=lpCmdLine;
  231. RegisterClass(&wc);
  232. hwnd = CreateWindowEx(WS_EX_TOOLWINDOW,lpCmdLine,lpCmdLine,WS_VISIBLE|WS_POPUP,0,0,0,0,0,0,hInstance,0);
  233. #endif
  234. #ifdef _WIN32
  235. WSADATA wd;
  236. WSAStartup(MAKEWORD( 1, 1 ), &wd);
  237. #endif
  238. #endif
  239. srvinit(&srv, &defparam);
  240. srv.pf = childdef.pf;
  241. isudp = childdef.isudp;
  242. srv.service = defparam.service = childdef.service;
  243. #ifndef STDMAIN
  244. if(conf.acl){
  245. srv.acl = copyacl(conf.acl);
  246. if(!srv.acl) haveerror = 2;
  247. }
  248. if(conf.authfuncs){
  249. srv.authfuncs = copyauth(conf.authfuncs);
  250. if(!srv.authfuncs) haveerror = 2;
  251. }
  252. if(!conf.services){
  253. conf.services = &srv;
  254. }
  255. else {
  256. srv.next = conf.services;
  257. conf.services = conf.services->prev = &srv;
  258. }
  259. #ifndef _WIN32
  260. {
  261. sigset_t mask;
  262. sigfillset(&mask);
  263. pthread_sigmask(SIG_SETMASK, &mask, NULL);
  264. }
  265. #endif
  266. #else
  267. srv.needuser = 0;
  268. pthread_mutex_init(&log_mutex, NULL);
  269. #endif
  270. for (i=1; i<argc; i++) {
  271. if(*argv[i]=='-') {
  272. switch(argv[i][1]) {
  273. case 'd':
  274. if(!conf.demon)daemonize();
  275. conf.demon = 1;
  276. break;
  277. #ifdef SO_BINDTODEVICE
  278. case 'D':
  279. if(argv[i][2] == 'i') srv.ibindtodevice = mystrdup(argv[i] + 3);
  280. else srv.obindtodevice = mystrdup(argv[i] + 3);
  281. break;
  282. #endif
  283. case 'l':
  284. srv.logfunc = logstdout;
  285. if(srv.logtarget) myfree(srv.logtarget);
  286. srv.logtarget = (unsigned char *)mystrdup(argv[i] + 2);
  287. if(argv[i][2]) {
  288. if(argv[i][2]=='@'){
  289. #ifdef STDMAIN
  290. #ifndef _WIN32
  291. openlog(argv[i]+3, LOG_PID, LOG_DAEMON);
  292. srv.logfunc = logsyslog;
  293. #endif
  294. #endif
  295. }
  296. else {
  297. fp = fopen(argv[i] + 2, "a");
  298. if (fp) {
  299. srv.stdlog = fp;
  300. fseek(fp, 0L, SEEK_END);
  301. }
  302. }
  303. }
  304. break;
  305. case 'i':
  306. getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&srv.intsa);
  307. break;
  308. case 'e':
  309. {
  310. #ifndef NOIPV6
  311. struct sockaddr_in6 sa6;
  312. memset(&sa6, 0, sizeof(sa6));
  313. error = !getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&sa6);
  314. if(!error) {
  315. if (*SAFAMILY(&sa6)==AF_INET) srv.extsa = sa6;
  316. else srv.extsa6 = sa6;
  317. }
  318. #else
  319. error = !getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&srv.extsa);
  320. #endif
  321. }
  322. break;
  323. case 'N':
  324. getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&srv.extNat);
  325. break;
  326. case 'p':
  327. *SAPORT(&srv.intsa) = htons(atoi(argv[i]+2));
  328. break;
  329. case '4':
  330. case '6':
  331. srv.family = atoi(argv[i]+1);
  332. break;
  333. case 'b':
  334. srv.bufsize = atoi(argv[i]+2);
  335. break;
  336. case 'n':
  337. srv.usentlm = atoi(argv[i]+2);
  338. break;
  339. #ifdef STDMAIN
  340. #ifndef _WIN32
  341. case 'I':
  342. size = sizeof(defparam.sincl);
  343. if(so._getsockname(0, (struct sockaddr*)&defparam.sincl, &size) ||
  344. *SAFAMILY(&defparam.sincl) != AF_INET) error = 1;
  345. else inetd = 1;
  346. break;
  347. #endif
  348. #endif
  349. case 'f':
  350. if(srv.logformat)myfree(srv.logformat);
  351. srv.logformat = (unsigned char *)mystrdup(argv[i] + 2);
  352. break;
  353. case 't':
  354. srv.silent = 1;
  355. break;
  356. case 'h':
  357. hostname = argv[i] + 2;
  358. break;
  359. case 'r':
  360. cbc_string = (unsigned char *)mystrdup(argv[i] + 2);
  361. iscbc = 1;
  362. break;
  363. case 'R':
  364. cbl_string = (unsigned char *)mystrdup(argv[i] + 2);
  365. iscbl = 1;
  366. break;
  367. case 'u':
  368. srv.needuser = 0;
  369. if(*(argv[i] + 2)) srv.needuser = atoi(argv[i] + 2);
  370. break;
  371. case 'T':
  372. srv.transparent = 1;
  373. break;
  374. case 'S':
  375. srv.stacksize = atoi(argv[i]+2);
  376. break;
  377. case 'a':
  378. srv.anonymous = 1 + atoi(argv[i]+2);
  379. break;
  380. case 's':
  381. #ifdef WITHSPLICE
  382. if(isudp || srv.service == S_ADMIN)
  383. #endif
  384. srv.singlepacket = 1 + atoi(argv[i]+2);
  385. #ifdef WITHSPLICE
  386. else
  387. if(*(argv[i]+2)) srv.usesplice = atoi(argv[i]+2);
  388. #endif
  389. break;
  390. case 'o':
  391. switch(argv[i][2]){
  392. case 's':
  393. srv.srvsockopts = getopts(argv[i]+3);
  394. break;
  395. case 'c':
  396. srv.clisockopts = getopts(argv[i]+3);
  397. break;
  398. case 'l':
  399. srv.lissockopts = getopts(argv[i]+3);
  400. break;
  401. case 'r':
  402. srv.cbcsockopts = getopts(argv[i]+3);
  403. break;
  404. case 'R':
  405. srv.cbcsockopts = getopts(argv[i]+3);
  406. break;
  407. default:
  408. error = 1;
  409. }
  410. if(!error) break;
  411. default:
  412. error = 1;
  413. break;
  414. }
  415. }
  416. else break;
  417. }
  418. #ifndef STDMAIN
  419. if(childdef.port) {
  420. #endif
  421. #ifndef PORTMAP
  422. if (error || i!=argc) {
  423. #ifndef STDMAIN
  424. haveerror = 1;
  425. conf.threadinit = 0;
  426. #endif
  427. fprintf(stderr, "%s of %s\n"
  428. "Usage: %s options\n"
  429. "Available options are:\n"
  430. "%s\n"
  431. "\t%s\n"
  432. " -pPORT - service port to accept connections\n"
  433. "%s"
  434. "\tExample: %s -i127.0.0.1\n\n"
  435. "%s",
  436. argv[0],
  437. conf.stringtable?(char *)conf.stringtable[3]: VERSION " (" BUILDDATE ")",
  438. argv[0], loghelp, printopts("\n\t"), childdef.helpmessage, argv[0],
  439. #ifdef STDMAIN
  440. copyright
  441. #else
  442. ""
  443. #endif
  444. );
  445. return (1);
  446. }
  447. #endif
  448. #ifndef STDMAIN
  449. }
  450. else {
  451. #endif
  452. #ifndef NOPORTMAP
  453. if (error || argc != i+3 || *argv[i]=='-'|| (*SAPORT(&srv.intsa) = htons((unsigned short)atoi(argv[i])))==0 || (srv.targetport = htons((unsigned short)atoi(argv[i+2])))==0) {
  454. #ifndef STDMAIN
  455. haveerror = 1;
  456. conf.threadinit = 0;
  457. #endif
  458. fprintf(stderr, "%s of %s\n"
  459. "Usage: %s options"
  460. " [-e<external_ip>] <port_to_bind>"
  461. " <target_hostname> <target_port>\n"
  462. "Available options are:\n"
  463. "%s\n"
  464. "\t%s\n"
  465. "%s"
  466. "\tExample: %s -d -i127.0.0.1 6666 serv.somehost.ru 6666\n\n"
  467. "%s",
  468. argv[0],
  469. conf.stringtable?(char *)conf.stringtable[3]: VERSION " (" BUILDDATE ")",
  470. argv[0], loghelp, printopts("\n\t"), childdef.helpmessage, argv[0],
  471. #ifdef STDMAIN
  472. copyright
  473. #else
  474. ""
  475. #endif
  476. );
  477. return (1);
  478. }
  479. srv.target = (unsigned char *)mystrdup(argv[i+1]);
  480. #endif
  481. #ifndef STDMAIN
  482. }
  483. #else
  484. #ifndef _WIN32
  485. if(inetd) {
  486. fcntl(0,F_SETFL,O_NONBLOCK | fcntl(0,F_GETFL));
  487. if(!isudp){
  488. so._setsockopt(0, SOL_SOCKET, SO_LINGER, (unsigned char *)&lg, sizeof(lg));
  489. so._setsockopt(0, SOL_SOCKET, SO_OOBINLINE, (unsigned char *)&opt, sizeof(int));
  490. }
  491. defparam.clisock = 0;
  492. if(! (newparam = myalloc (sizeof(defparam)))){
  493. return 2;
  494. };
  495. *newparam = defparam;
  496. return((*srv.pf)((void *)newparam)? 1:0);
  497. }
  498. #endif
  499. #endif
  500. srvinit2(&srv, &defparam);
  501. if(!*SAFAMILY(&srv.intsa)) *SAFAMILY(&srv.intsa) = AF_INET;
  502. if(!*SAPORT(&srv.intsa)) *SAPORT(&srv.intsa) = htons(childdef.port);
  503. *SAFAMILY(&srv.extsa) = AF_INET;
  504. #ifndef NOIPV6
  505. *SAFAMILY(&srv.extsa6) = AF_INET6;
  506. #endif
  507. if(hostname)parsehostname(hostname, &defparam, childdef.port);
  508. #ifndef STDMAIN
  509. copyfilter(conf.filters, &srv);
  510. conf.threadinit = 0;
  511. #endif
  512. if (!iscbc) {
  513. if(srv.srvsock == INVALID_SOCKET){
  514. if(!isudp){
  515. lg.l_onoff = 1;
  516. lg.l_linger = conf.timeouts[STRING_L];
  517. sock=so._socket(SASOCK(&srv.intsa), SOCK_STREAM, IPPROTO_TCP);
  518. }
  519. else {
  520. sock=so._socket(SASOCK(&srv.intsa), SOCK_DGRAM, IPPROTO_UDP);
  521. }
  522. if( sock == INVALID_SOCKET) {
  523. perror("socket()");
  524. return -2;
  525. }
  526. setopts(sock, srv.lissockopts);
  527. #ifdef _WIN32
  528. ioctlsocket(sock, FIONBIO, &ul);
  529. #else
  530. fcntl(sock,F_SETFL,O_NONBLOCK | fcntl(sock,F_GETFL));
  531. #endif
  532. srv.srvsock = sock;
  533. opt = 1;
  534. if(so._setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(int)))perror("setsockopt()");
  535. #ifdef SO_REUSEPORT
  536. opt = 1;
  537. so._setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (char *)&opt, sizeof(int));
  538. #endif
  539. #ifdef SO_BINDTODEVICE
  540. if(srv.ibindtodevice) so._setsockopt(sock, SOL_SOCKET, SO_BINDTODEVICE, srv.ibindtodevice, strlen(srv.ibindtodevice) + 1);
  541. #endif
  542. }
  543. size = sizeof(srv.intsa);
  544. for(sleeptime = SLEEPTIME * 100; so._bind(sock, (struct sockaddr*)&srv.intsa, SASIZE(&srv.intsa))==-1; usleep(sleeptime)) {
  545. sprintf((char *)buf, "bind(): %s", strerror(errno));
  546. if(!srv.silent)dolog(&defparam, buf);
  547. sleeptime = (sleeptime<<1);
  548. if(!sleeptime) {
  549. so._closesocket(sock);
  550. return -3;
  551. }
  552. }
  553. if(!isudp){
  554. if(so._listen (sock, 1 + (srv.maxchild>>4))==-1) {
  555. sprintf((char *)buf, "listen(): %s", strerror(errno));
  556. if(!srv.silent)dolog(&defparam, buf);
  557. return -4;
  558. }
  559. }
  560. else
  561. defparam.clisock = sock;
  562. if(!srv.silent && !iscbc){
  563. sprintf((char *)buf, "Accepting connections [%u/%u]", (unsigned)getpid(), (unsigned)pthread_self());
  564. dolog(&defparam, buf);
  565. }
  566. }
  567. if(iscbl){
  568. parsehost(srv.family, cbl_string, (struct sockaddr *)&cbsa);
  569. if((srv.cbsock=so._socket(SASOCK(&cbsa), SOCK_STREAM, IPPROTO_TCP))==INVALID_SOCKET) {
  570. dolog(&defparam, (unsigned char *)"Failed to allocate connect back socket");
  571. return -6;
  572. }
  573. opt = 1;
  574. so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(int));
  575. #ifdef SO_REUSEPORT
  576. opt = 1;
  577. so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEPORT, (char *)&opt, sizeof(int));
  578. #endif
  579. setopts(srv.cbsock, srv.cbssockopts);
  580. if(so._bind(srv.cbsock, (struct sockaddr*)&cbsa, SASIZE(&cbsa))==-1) {
  581. dolog(&defparam, (unsigned char *)"Failed to bind connect back socket");
  582. return -7;
  583. }
  584. if(so._listen(srv.cbsock, 1 + (srv.maxchild>>4))==-1) {
  585. dolog(&defparam, (unsigned char *)"Failed to listen connect back socket");
  586. return -8;
  587. }
  588. }
  589. srv.fds.fd = sock;
  590. srv.fds.events = POLLIN;
  591. #ifndef _WIN32
  592. pthread_attr_init(&pa);
  593. pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + (32768 + srv.stacksize));
  594. pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
  595. #endif
  596. for (;;) {
  597. for(;;){
  598. while((conf.paused == srv.paused && srv.childcount >= srv.maxchild)){
  599. nlog++;
  600. if(!srv.silent && nlog > 5000) {
  601. sprintf((char *)buf, "Warning: too many connected clients (%d/%d)", srv.childcount, srv.maxchild);
  602. dolog(&defparam, buf);
  603. nlog = 0;
  604. }
  605. usleep(SLEEPTIME);
  606. }
  607. if (iscbc) break;
  608. if (conf.paused != srv.paused) break;
  609. if (srv.fds.events & POLLIN) {
  610. error = so._poll(&srv.fds, 1, 1000);
  611. }
  612. else {
  613. usleep(SLEEPTIME);
  614. continue;
  615. }
  616. if (error >= 1) break;
  617. if (error == 0) continue;
  618. if (errno != EAGAIN && errno != EINTR) {
  619. sprintf((char *)buf, "poll(): %s/%d", strerror(errno), errno);
  620. if(!srv.silent)dolog(&defparam, buf);
  621. break;
  622. }
  623. }
  624. if((conf.paused != srv.paused) || (error < 0)) break;
  625. error = 0;
  626. if(!isudp){
  627. size = sizeof(defparam.sincr);
  628. if(iscbc){
  629. new_sock=so._socket(SASOCK(&defparam.sincr), SOCK_STREAM, IPPROTO_TCP);
  630. if(new_sock != INVALID_SOCKET){
  631. setopts(new_sock, srv.cbcsockopts);
  632. parsehost(srv.family, cbc_string, (struct sockaddr *)&defparam.sincr);
  633. if(connectwithpoll(new_sock,(struct sockaddr *)&defparam.sincr,SASIZE(&defparam.sincr),CONNBACK_TO)) {
  634. so._closesocket(new_sock);
  635. new_sock = INVALID_SOCKET;
  636. usleep(SLEEPTIME);
  637. continue;
  638. }
  639. if(sockrecvfrom(new_sock,(struct sockaddr*)&defparam.sincr,buf,1,60) != 1 || *buf!='C') {
  640. so._closesocket(new_sock);
  641. new_sock = INVALID_SOCKET;
  642. usleep(SLEEPTIME);
  643. continue;
  644. }
  645. }
  646. else {
  647. usleep(SLEEPTIME);
  648. continue;
  649. }
  650. }
  651. else {
  652. new_sock = so._accept(sock, (struct sockaddr*)&defparam.sincr, &size);
  653. if(new_sock == INVALID_SOCKET){
  654. #ifdef _WIN32
  655. switch(WSAGetLastError()){
  656. case WSAEMFILE:
  657. case WSAENOBUFS:
  658. case WSAENETDOWN:
  659. usleep(SLEEPTIME * 10);
  660. break;
  661. case WSAEINTR:
  662. error = 1;
  663. break;
  664. default:
  665. break;
  666. }
  667. #else
  668. switch (errno){
  669. #ifdef EMFILE
  670. case EMFILE:
  671. #endif
  672. #ifdef ENFILE
  673. case ENFILE:
  674. #endif
  675. #ifdef ENOBUFS
  676. case ENOBUFS:
  677. #endif
  678. #ifdef ENOMEM
  679. case ENOMEM:
  680. #endif
  681. usleep(SLEEPTIME * 10);
  682. break;
  683. default:
  684. break;
  685. }
  686. #endif
  687. nlog++;
  688. if(!srv.silent && (error || nlog > 5000)) {
  689. sprintf((char *)buf, "accept(): %s", strerror(errno));
  690. dolog(&defparam, buf);
  691. nlog = 0;
  692. }
  693. continue;
  694. }
  695. }
  696. setopts(new_sock, srv.clisockopts);
  697. size = sizeof(defparam.sincl);
  698. if(so._getsockname(new_sock, (struct sockaddr *)&defparam.sincl, &size)){
  699. sprintf((char *)buf, "getsockname(): %s", strerror(errno));
  700. if(!srv.silent)dolog(&defparam, buf);
  701. continue;
  702. }
  703. #ifdef _WIN32
  704. ioctlsocket(new_sock, FIONBIO, &ul);
  705. #else
  706. fcntl(new_sock,F_SETFL,O_NONBLOCK | fcntl(new_sock,F_GETFL));
  707. #endif
  708. so._setsockopt(new_sock, SOL_SOCKET, SO_LINGER, (char *)&lg, sizeof(lg));
  709. so._setsockopt(new_sock, SOL_SOCKET, SO_OOBINLINE, (char *)&opt, sizeof(int));
  710. }
  711. else {
  712. srv.fds.events = 0;
  713. }
  714. if(! (newparam = myalloc (sizeof(defparam)))){
  715. if(!isudp) so._closesocket(new_sock);
  716. defparam.res = 21;
  717. if(!srv.silent)dolog(&defparam, (unsigned char *)"Memory Allocation Failed");
  718. usleep(SLEEPTIME);
  719. continue;
  720. };
  721. *newparam = defparam;
  722. if(defparam.hostname)newparam->hostname=(unsigned char *)mystrdup((char *)defparam.hostname);
  723. clearstat(newparam);
  724. if(!isudp) newparam->clisock = new_sock;
  725. #ifndef STDMAIN
  726. if(makefilters(&srv, newparam) > CONTINUE){
  727. freeparam(newparam);
  728. continue;
  729. }
  730. #endif
  731. newparam->prev = newparam->next = NULL;
  732. error = 0;
  733. pthread_mutex_lock(&srv.counter_mutex);
  734. if(!srv.child){
  735. srv.child = newparam;
  736. }
  737. else {
  738. newparam->next = srv.child;
  739. srv.child = srv.child->prev = newparam;
  740. }
  741. #ifdef _WIN32
  742. #ifndef _WINCE
  743. h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, (unsigned)(16384 + srv.stacksize), (void *)threadfunc, (void *) newparam, 0, &thread);
  744. #else
  745. h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, (unsigned)(16384 + srv.stacksize), (void *)threadfunc, (void *) newparam, 0, &thread);
  746. #endif
  747. srv.childcount++;
  748. if (h) {
  749. newparam->threadid = (unsigned)thread;
  750. CloseHandle(h);
  751. }
  752. else {
  753. sprintf((char *)buf, "_beginthreadex(): %s", _strerror(NULL));
  754. if(!srv.silent)dolog(&defparam, buf);
  755. error = 1;
  756. }
  757. #else
  758. error = pthread_create(&thread, &pa, threadfunc, (void *)newparam);
  759. srv.childcount++;
  760. if(error){
  761. sprintf((char *)buf, "pthread_create(): %s", strerror(error));
  762. if(!srv.silent)dolog(&defparam, buf);
  763. }
  764. else {
  765. newparam->threadid = (unsigned)thread;
  766. }
  767. #endif
  768. pthread_mutex_unlock(&srv.counter_mutex);
  769. if(error) freeparam(newparam);
  770. memset(&defparam.sincl, 0, sizeof(defparam.sincl));
  771. memset(&defparam.sincr, 0, sizeof(defparam.sincr));
  772. if(isudp) while(!srv.fds.events)usleep(SLEEPTIME);
  773. }
  774. if(!srv.silent) srv.logfunc(&defparam, (unsigned char *)"Exiting thread");
  775. srvfree(&srv);
  776. #ifndef STDMAIN
  777. pthread_mutex_lock(&config_mutex);
  778. if(srv.next)srv.next->prev = srv.prev;
  779. if(srv.prev)srv.prev->next = srv.next;
  780. else conf.services = srv.next;
  781. pthread_mutex_unlock(&config_mutex);
  782. #endif
  783. #ifndef _WIN32
  784. pthread_attr_destroy(&pa);
  785. #endif
  786. if(defparam.hostname)myfree(defparam.hostname);
  787. if(cbc_string)myfree(cbc_string);
  788. if(cbl_string)myfree(cbl_string);
  789. if(fp) fclose(fp);
  790. return 0;
  791. }
  792. void srvinit(struct srvparam * srv, struct clientparam *param){
  793. memset(srv, 0, sizeof(struct srvparam));
  794. srv->version = conf.version + 1;
  795. srv->paused = conf.paused;
  796. srv->logfunc = havelog?conf.logfunc:lognone;
  797. srv->noforce = conf.noforce;
  798. srv->logformat = conf.logformat? (unsigned char *)mystrdup((char *)conf.logformat) : NULL;
  799. srv->authfunc = conf.authfunc;
  800. srv->usentlm = 0;
  801. srv->maxchild = conf.maxchild;
  802. srv->stacksize = conf.stacksize;
  803. srv->time_start = time(NULL);
  804. if(havelog && conf.logtarget){
  805. srv->logtarget = (unsigned char *)mystrdup((char *)conf.logtarget);
  806. }
  807. srv->srvsock = INVALID_SOCKET;
  808. srv->logdumpsrv = conf.logdumpsrv;
  809. srv->logdumpcli = conf.logdumpcli;
  810. srv->cbsock = INVALID_SOCKET;
  811. srv->needuser = 1;
  812. #ifdef WITHSPLICE
  813. srv->usesplice = 1;
  814. #endif
  815. memset(param, 0, sizeof(struct clientparam));
  816. param->srv = srv;
  817. param->version = srv->version;
  818. param->paused = srv->paused;
  819. param->remsock = param->clisock = param->ctrlsock = param->ctrlsocksrv = INVALID_SOCKET;
  820. *SAFAMILY(&param->req) = *SAFAMILY(&param->sinsl) = *SAFAMILY(&param->sinsr) = *SAFAMILY(&param->sincr) = *SAFAMILY(&param->sincl) = AF_INET;
  821. pthread_mutex_init(&srv->counter_mutex, NULL);
  822. srv->intsa = conf.intsa;
  823. srv->extsa = conf.extsa;
  824. #ifndef NOIPV6
  825. srv->extsa6 = conf.extsa6;
  826. #endif
  827. }
  828. void srvinit2(struct srvparam * srv, struct clientparam *param){
  829. if(srv->logformat){
  830. char *s;
  831. if(*srv->logformat == '-' && (s = strchr((char *)srv->logformat + 1, '+')) && s[1]){
  832. unsigned char* logformat = srv->logformat;
  833. *s = 0;
  834. srv->nonprintable = (unsigned char *)mystrdup((char *)srv->logformat + 1);
  835. srv->replace = s[1];
  836. srv->logformat = (unsigned char *)mystrdup(s + 2);
  837. *s = '+';
  838. myfree(logformat);
  839. }
  840. }
  841. memset(&param->sinsl, 0, sizeof(param->sinsl));
  842. memset(&param->sinsr, 0, sizeof(param->sinsr));
  843. memset(&param->req, 0, sizeof(param->req));
  844. *SAFAMILY(&param->sinsl) = AF_INET;
  845. *SAFAMILY(&param->sinsr) = AF_INET;
  846. *SAFAMILY(&param->req) = AF_INET;
  847. param->sincr = param->sincl = srv->intsa;
  848. #ifndef NOIPV6
  849. if (srv->family == 6 || srv->family == 64) param->sinsr = srv->extsa6;
  850. else
  851. #endif
  852. param->sinsr = srv->extsa;
  853. }
  854. void srvfree(struct srvparam * srv){
  855. if(srv->srvsock != INVALID_SOCKET) so._closesocket(srv->srvsock);
  856. srv->srvsock = INVALID_SOCKET;
  857. if(srv->cbsock != INVALID_SOCKET) so._closesocket(srv->cbsock);
  858. srv->cbsock = INVALID_SOCKET;
  859. srv->service = S_ZOMBIE;
  860. while(srv->child) usleep(SLEEPTIME * 100);
  861. #ifndef STDMAIN
  862. if(srv->filter){
  863. while(srv->nfilters){
  864. srv->nfilters--;
  865. if(srv->filter[srv->nfilters].filter_close){
  866. (*srv->filter[srv->nfilters].filter_close)(srv->filter[srv->nfilters].data);
  867. }
  868. }
  869. myfree(srv->filter);
  870. }
  871. if(srv->acl)freeacl(srv->acl);
  872. if(srv->authfuncs)freeauth(srv->authfuncs);
  873. #endif
  874. pthread_mutex_destroy(&srv->counter_mutex);
  875. if(srv->target) myfree(srv->target);
  876. if(srv->logtarget) myfree(srv->logtarget);
  877. if(srv->logformat) myfree(srv->logformat);
  878. if(srv->nonprintable) myfree(srv->nonprintable);
  879. #ifdef SO_BINDTODEVICE
  880. if(srv->ibindtodevice) myfree(srv->ibindtodevice);
  881. if(srv->obindtodevice) myfree(srv->obindtodevice);
  882. #endif
  883. }
  884. void freeparam(struct clientparam * param) {
  885. if(param->res == 2) return;
  886. if(param->datfilterssrv) myfree(param->datfilterssrv);
  887. #ifndef STDMAIN
  888. if(param->reqfilters) myfree(param->reqfilters);
  889. if(param->hdrfilterscli) myfree(param->hdrfilterscli);
  890. if(param->hdrfilterssrv) myfree(param->hdrfilterssrv);
  891. if(param->predatfilters) myfree(param->predatfilters);
  892. if(param->datfilterscli) myfree(param->datfilterscli);
  893. if(param->filters){
  894. if(param->nfilters)while(param->nfilters--){
  895. if(param->filters[param->nfilters].filter->filter_clear)
  896. (*param->filters[param->nfilters].filter->filter_clear)(param->filters[param->nfilters].data);
  897. }
  898. myfree(param->filters);
  899. }
  900. if(conf.connlimiter && (param->res != 95 || param->remsock != INVALID_SOCKET)) stopconnlims(param);
  901. #endif
  902. if(param->clibuf) myfree(param->clibuf);
  903. if(param->srvbuf) myfree(param->srvbuf);
  904. if(param->srv){
  905. pthread_mutex_lock(&param->srv->counter_mutex);
  906. if(param->prev){
  907. param->prev->next = param->next;
  908. }
  909. else
  910. param->srv->child = param->next;
  911. if(param->next){
  912. param->next->prev = param->prev;
  913. }
  914. (param->srv->childcount)--;
  915. pthread_mutex_unlock(&param->srv->counter_mutex);
  916. }
  917. if(param->hostname) myfree(param->hostname);
  918. if(param->username) myfree(param->username);
  919. if(param->password) myfree(param->password);
  920. if(param->extusername) myfree(param->extusername);
  921. if(param->extpassword) myfree(param->extpassword);
  922. if(param->ctrlsocksrv != INVALID_SOCKET && param->ctrlsocksrv != param->remsock) {
  923. so._shutdown(param->ctrlsocksrv, SHUT_RDWR);
  924. so._closesocket(param->ctrlsocksrv);
  925. }
  926. if(param->ctrlsock != INVALID_SOCKET && param->ctrlsock != param->clisock) {
  927. so._shutdown(param->ctrlsock, SHUT_RDWR);
  928. so._closesocket(param->ctrlsock);
  929. }
  930. if(param->remsock != INVALID_SOCKET) {
  931. so._shutdown(param->remsock, SHUT_RDWR);
  932. so._closesocket(param->remsock);
  933. }
  934. if(param->clisock != INVALID_SOCKET) {
  935. so._shutdown(param->clisock, SHUT_RDWR);
  936. so._closesocket(param->clisock);
  937. }
  938. myfree(param);
  939. }
  940. #ifndef STDMAIN
  941. static void * itcopy (void * from, size_t size){
  942. void * ret;
  943. if(!from) return NULL;
  944. ret = myalloc(size);
  945. if(ret) memcpy(ret, from, size);
  946. return ret;
  947. }
  948. struct auth * copyauth (struct auth * authfuncs){
  949. struct auth * newauth = NULL;
  950. newauth = authfuncs = itcopy(authfuncs, sizeof(struct auth));
  951. for( ; authfuncs->next; authfuncs = authfuncs->next){
  952. authfuncs->next = itcopy(authfuncs->next, sizeof(struct auth));
  953. if(!authfuncs->next)break;
  954. }
  955. if(authfuncs){
  956. freeauth(newauth);
  957. return NULL;
  958. }
  959. return newauth;
  960. }
  961. struct ace * copyacl (struct ace *ac){
  962. struct ace * ret = NULL;
  963. struct iplist *ipl;
  964. struct portlist *pl;
  965. struct userlist *ul;
  966. struct chain *ch;
  967. struct period *pel;
  968. struct hostname *hst;
  969. ret = ac = itcopy(ac, sizeof(struct ace));
  970. for( ; ac->next; ac = ac->next){
  971. if(ac->src){
  972. ac->src = itcopy(ac->src, sizeof(struct iplist));
  973. if(!ac->src) goto ERRORSRC;
  974. for(ipl = ac->src; ipl->next; ipl = ipl->next){
  975. ipl->next = itcopy(ipl->next, sizeof(struct iplist));
  976. if(!ipl->next) goto ERRORSRC;
  977. }
  978. }
  979. if(ac->dst){
  980. ac->dst = itcopy(ac->dst, sizeof(struct iplist));
  981. if(!ac->dst) goto ERRORDST;
  982. for(ipl = ac->dst; ipl->next; ipl = ipl->next){
  983. ipl->next = itcopy(ipl->next, sizeof(struct iplist));
  984. if(!ipl->next) goto ERRORDST;
  985. }
  986. }
  987. if(ac->ports){
  988. ac->ports = itcopy(ac->ports, sizeof(struct portlist));
  989. if(!ac->ports) goto ERRORPORTS;
  990. for(pl = ac->ports; pl->next; pl = pl->next){
  991. pl->next = itcopy(pl->next, sizeof(struct portlist));
  992. if(!pl->next) goto ERRORPORTS;
  993. }
  994. }
  995. if(ac->periods){
  996. ac->periods = itcopy(ac->periods, sizeof(struct period));
  997. if(!ac->periods) goto ERRORPERIODS;
  998. for(pel = ac->periods; pel->next; pel = pel->next){
  999. pel->next = itcopy(pel->next, sizeof(struct period));
  1000. if(!pel->next) goto ERRORPERIODS;
  1001. }
  1002. }
  1003. if(ac->users){
  1004. ac->users = itcopy(ac->users, sizeof(struct userlist));
  1005. if(!ac->users) goto ERRORUSERS;
  1006. for(ul = ac->users; ul; ul = ul->next){
  1007. if(ul->user) {
  1008. ul->user = (unsigned char*)mystrdup((char *)ul->user);
  1009. if(!ul->user) {
  1010. ul->next = NULL;
  1011. goto ERRORUSERS;
  1012. }
  1013. }
  1014. if(ul->next){
  1015. ul->next = itcopy(ul->next, sizeof(struct userlist));
  1016. if(!ul->next) goto ERRORUSERS;
  1017. }
  1018. }
  1019. }
  1020. if(ac->dstnames){
  1021. ac->dstnames = itcopy(ac->dstnames, sizeof(struct hostname));
  1022. if(!ac->dstnames) goto ERRORDSTNAMES;
  1023. for(hst = ac->dstnames; hst; hst = hst->next){
  1024. if(hst->name) {
  1025. hst->name = (unsigned char*)mystrdup((char *)hst->name);
  1026. if(!hst->name) {
  1027. hst->next = NULL;
  1028. goto ERRORDSTNAMES;
  1029. }
  1030. }
  1031. if(hst->next){
  1032. hst->next = itcopy(hst->next, sizeof(struct hostname));
  1033. if(!hst->next) goto ERRORDSTNAMES;
  1034. }
  1035. }
  1036. }
  1037. if(ac->chains){
  1038. ac->chains = itcopy(ac->chains, sizeof(struct chain));
  1039. if(!ac->chains) goto ERRORCHAINS;
  1040. for(ch = ac->chains; ch; ch = ch->next){
  1041. if(ch->extuser){
  1042. ch->extuser = (unsigned char*)mystrdup((char *)ch->extuser);
  1043. if(!ch->extuser){
  1044. ch->extpass = NULL;
  1045. ch->exthost = NULL;
  1046. ch->next = NULL;
  1047. goto ERRORCHAINS;
  1048. }
  1049. }
  1050. if(ch->extpass){
  1051. ch->extpass = (unsigned char*)mystrdup((char *)ch->extpass);
  1052. if(!ch->extpass){
  1053. ch->exthost = NULL;
  1054. ch->next = NULL;
  1055. goto ERRORCHAINS;
  1056. }
  1057. }
  1058. if(ch->exthost){
  1059. ch->exthost = (unsigned char*)mystrdup((char *)ch->exthost);
  1060. if(!ch->exthost){
  1061. ch->next = NULL;
  1062. goto ERRORCHAINS;
  1063. }
  1064. }
  1065. if(ch->next){
  1066. ch->next = itcopy(ch->next, sizeof(struct chain));
  1067. if(!ch->next) goto ERRORCHAINS;
  1068. }
  1069. }
  1070. }
  1071. }
  1072. if(!ac) return ret;
  1073. ERRORSRC:
  1074. ac->dst = NULL;
  1075. ERRORDST:
  1076. ac->ports = NULL;
  1077. ERRORPORTS:
  1078. ac->periods = NULL;
  1079. ERRORPERIODS:
  1080. ac->users = NULL;
  1081. ERRORUSERS:
  1082. ac->dstnames = NULL;
  1083. ERRORDSTNAMES:
  1084. ac->chains = NULL;
  1085. ERRORCHAINS:
  1086. ac->next = NULL;
  1087. freeacl(ret);
  1088. return NULL;
  1089. }
  1090. void copyfilter (struct filter *filter, struct srvparam *srv){
  1091. int nfilters = 0;
  1092. if(!filter) return;
  1093. for(srv->filter = filter; srv->filter; srv->filter = srv->filter->next) nfilters++;
  1094. srv->filter = myalloc(sizeof(struct filter) * nfilters);
  1095. if(!srv->filter) return;
  1096. for(; filter; filter = filter->next){
  1097. void *data = NULL;
  1098. if(!filter->filter_open || !(data = (*filter->filter_open)(filter->data, srv))) continue;
  1099. srv->filter[srv->nfilters] = *filter;
  1100. srv->filter[srv->nfilters].data = data;
  1101. if(srv->nfilters>0)srv->filter[srv->nfilters - 1].next = srv->filter + srv->nfilters;
  1102. srv->nfilters++;
  1103. if(filter->filter_request)srv->nreqfilters++;
  1104. if(filter->filter_header_srv)srv->nhdrfilterssrv++;
  1105. if(filter->filter_header_cli)srv->nhdrfilterscli++;
  1106. if(filter->filter_predata)srv->npredatfilters++;
  1107. if(filter->filter_data_srv)srv->ndatfilterssrv++;
  1108. if(filter->filter_data_cli)srv->ndatfilterscli++;
  1109. }
  1110. }
  1111. FILTER_ACTION makefilters (struct srvparam *srv, struct clientparam *param){
  1112. FILTER_ACTION res=PASS;
  1113. FILTER_ACTION action;
  1114. int i;
  1115. if(!srv->nfilters) return PASS;
  1116. if(!(param->filters = myalloc(sizeof(struct filterp) * srv->nfilters)) ||
  1117. (srv->nreqfilters && !(param->reqfilters = myalloc(sizeof(struct filterp *) * srv->nreqfilters))) ||
  1118. (srv->nhdrfilterssrv && !(param->hdrfilterssrv = myalloc(sizeof(struct filterp *) * srv->nhdrfilterssrv))) ||
  1119. (srv->nhdrfilterscli && !(param->hdrfilterscli = myalloc(sizeof(struct filterp *) * srv->nhdrfilterscli))) ||
  1120. (srv->npredatfilters && !(param->predatfilters = myalloc(sizeof(struct filterp *) * srv->npredatfilters))) ||
  1121. (srv->ndatfilterssrv && !(param->datfilterssrv = myalloc(sizeof(struct filterp *) * srv->ndatfilterssrv))) ||
  1122. (srv->ndatfilterscli && !(param->datfilterscli = myalloc(sizeof(struct filterp *) * srv->ndatfilterscli)))
  1123. ){
  1124. param->res = 21;
  1125. return REJECT;
  1126. }
  1127. for(i=0; i<srv->nfilters; i++){
  1128. if(!srv->filter[i].filter_client)continue;
  1129. action = (*srv->filter[i].filter_client)(srv->filter[i].data, param, &param->filters[param->nfilters].data);
  1130. if(action == PASS) continue;
  1131. if(action > CONTINUE) return action;
  1132. param->filters[param->nfilters].filter = srv->filter + i;
  1133. if(srv->filter[i].filter_request)param->reqfilters[param->nreqfilters++] = param->filters + param->nfilters;
  1134. if(srv->filter[i].filter_header_cli)param->hdrfilterscli[param->nhdrfilterscli++] = param->filters + param->nfilters;
  1135. if(srv->filter[i].filter_header_srv)param->hdrfilterssrv[param->nhdrfilterssrv++] = param->filters + param->nfilters;
  1136. if(srv->filter[i].filter_predata)param->predatfilters[param->npredatfilters++] = param->filters + param->nfilters;
  1137. if(srv->filter[i].filter_data_cli)param->datfilterscli[param->ndatfilterscli++] = param->filters + param->nfilters;
  1138. if(srv->filter[i].filter_data_srv)param->datfilterssrv[param->ndatfilterssrv++] = param->filters + param->nfilters;
  1139. param->nfilters++;
  1140. }
  1141. return res;
  1142. }
  1143. void * itfree(void *data, void * retval){
  1144. myfree(data);
  1145. return retval;
  1146. }
  1147. void freeauth(struct auth * authfuncs){
  1148. for(; authfuncs; authfuncs = (struct auth *)itfree(authfuncs, authfuncs->next));
  1149. }
  1150. void freeacl(struct ace *ac){
  1151. struct iplist *ipl;
  1152. struct portlist *pl;
  1153. struct userlist *ul;
  1154. struct chain *ch;
  1155. struct period *pel;
  1156. struct hostname *hst;
  1157. for(; ac; ac = (struct ace *) itfree(ac, ac->next)){
  1158. for(ipl = ac->src; ipl; ipl = (struct iplist *)itfree(ipl, ipl->next));
  1159. for(ipl = ac->dst; ipl; ipl = (struct iplist *)itfree(ipl,ipl->next));
  1160. for(pl = ac->ports; pl; pl = (struct portlist *)itfree(pl, pl->next));
  1161. for(pel = ac->periods; pel; pel = (struct period *)itfree(pel, pel->next));
  1162. for(ul = ac->users; ul; ul = (struct userlist *)itfree(ul, ul->next)){
  1163. if(ul->user)myfree(ul->user);
  1164. }
  1165. for(hst = ac->dstnames; hst; hst = (struct hostname *)itfree(hst, hst->next)){
  1166. if(hst->name)myfree(hst->name);
  1167. }
  1168. for(ch = ac->chains; ch; ch = (struct chain *) itfree(ch, ch->next)){
  1169. if(ch->extuser) myfree(ch->extuser);
  1170. if(ch->extpass) myfree(ch->extpass);
  1171. if(ch->exthost) myfree(ch->exthost);
  1172. }
  1173. }
  1174. }
  1175. FILTER_ACTION handlereqfilters(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  1176. FILTER_ACTION action;
  1177. int i;
  1178. for(i=0; i<param->nreqfilters; i++){
  1179. action = (*param->reqfilters[i]->filter->filter_request)(param->reqfilters[i]->data, param, buf_p, bufsize_p, offset, length_p);
  1180. if(action!=CONTINUE) return action;
  1181. }
  1182. return PASS;
  1183. }
  1184. FILTER_ACTION handlehdrfilterssrv(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  1185. FILTER_ACTION action;
  1186. int i;
  1187. for(i=0; i<param->nhdrfilterssrv; i++){
  1188. action = (*param->hdrfilterssrv[i]->filter->filter_header_srv)(param->hdrfilterssrv[i]->data, param, buf_p, bufsize_p, offset, length_p);
  1189. if(action!=CONTINUE) return action;
  1190. }
  1191. return PASS;
  1192. }
  1193. FILTER_ACTION handlehdrfilterscli(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  1194. FILTER_ACTION action;
  1195. int i;
  1196. for(i = 0; i < param->nhdrfilterscli; i++){
  1197. action = (*param->hdrfilterscli[i]->filter->filter_header_cli)(param->hdrfilterscli[i]->data, param, buf_p, bufsize_p, offset, length_p);
  1198. if(action!=CONTINUE) return action;
  1199. }
  1200. return PASS;
  1201. }
  1202. #endif
  1203. FILTER_ACTION handlepredatflt(struct clientparam *cparam){
  1204. #ifndef STDMAIN
  1205. FILTER_ACTION action;
  1206. int i;
  1207. for(i=0; i<cparam->npredatfilters ;i++){
  1208. action = (*cparam->predatfilters[i]->filter->filter_predata)(cparam->predatfilters[i]->data, cparam);
  1209. if(action!=CONTINUE) return action;
  1210. }
  1211. #endif
  1212. return PASS;
  1213. }
  1214. FILTER_ACTION handledatfltcli(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  1215. #ifndef STDMAIN
  1216. FILTER_ACTION action;
  1217. int i;
  1218. for(i=0; i<cparam->ndatfilterscli ;i++){
  1219. action = (*cparam->datfilterscli[i]->filter->filter_data_cli)(cparam->datfilterscli[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
  1220. if(action!=CONTINUE) return action;
  1221. }
  1222. #endif
  1223. return PASS;
  1224. }
  1225. FILTER_ACTION handledatfltsrv(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  1226. FILTER_ACTION action;
  1227. int i;
  1228. for(i=0; i<cparam->ndatfilterssrv; i++){
  1229. action = (*cparam->datfilterssrv[i]->filter->filter_data_srv)(cparam->datfilterssrv[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
  1230. if(action!=CONTINUE) return action;
  1231. }
  1232. return PASS;
  1233. }