proxymain.c 35 KB

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