proxymain.c 29 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028
  1. /*
  2. 3APA3A simpliest proxy server
  3. (c) 2002-2008 by ZARAZA <3APA3A@security.nnov.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 = 0;
  14. if(param->srv->cbsock != INVALID_SOCKET){
  15. SASIZETYPE size = sizeof(param->sinsr);
  16. for(i=0; i<3; i++){
  17. param->remsock = so._accept(param->srv->cbsock, (struct sockaddr*)&param->sinsr, &size);
  18. if(param->remsock == INVALID_SOCKET) {
  19. param->res = 13;
  20. param->srv->logfunc(param, "Connect back accept() failed");
  21. continue;
  22. }
  23. #ifndef WITHMAIN
  24. memcpy(&param->req, &param->sinsr, size);
  25. if(param->srv->acl) param->res = checkACL(param);
  26. if(param->res){
  27. param->srv->logfunc(param, "Connect back ACL failed");
  28. so._closesocket(param->remsock);
  29. param->remsock = INVALID_SOCKET;
  30. continue;
  31. }
  32. #endif
  33. if(so._sendto(param->remsock, "C", 1, 0, (struct sockaddr*)&param->sinsr, size) != 1){
  34. param->srv->logfunc(param, "Connect back sending command failed");
  35. so._closesocket(param->remsock);
  36. param->remsock = INVALID_SOCKET;
  37. continue;
  38. }
  39. break;
  40. }
  41. }
  42. if(i == 3){
  43. freeparam(param);
  44. }
  45. else {
  46. ((struct clientparam *) p)->srv->pf((struct clientparam *)p);
  47. }
  48. #ifdef _WIN32
  49. return 0;
  50. #else
  51. return NULL;
  52. #endif
  53. }
  54. #undef param
  55. #ifndef MODULEMAINFUNC
  56. #define MODULEMAINFUNC main
  57. #define STDMAIN
  58. #ifndef _WINCE
  59. int main (int argc, char** argv){
  60. #else
  61. int WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow){
  62. int argc;
  63. char ** argv;
  64. WNDCLASS wc;
  65. HWND hwnd = 0;
  66. #endif
  67. #else
  68. extern int linenum;
  69. extern int haveerror;
  70. int MODULEMAINFUNC (int argc, char** argv){
  71. #endif
  72. SOCKET sock = INVALID_SOCKET, new_sock = INVALID_SOCKET;
  73. int i=0;
  74. SASIZETYPE size;
  75. pthread_t thread;
  76. struct clientparam defparam;
  77. struct srvparam srv;
  78. struct clientparam * newparam;
  79. int error = 0;
  80. unsigned sleeptime;
  81. unsigned char buf[256];
  82. char *hostname=NULL;
  83. int opt = 1, isudp = 0, iscbl = 0, iscbc = 0;
  84. unsigned char *cbc_string = NULL, *cbl_string = NULL;
  85. #ifndef NOIPV6
  86. struct sockaddr_in6 cbsa;
  87. #else
  88. struct sockaddr_in cbsa;
  89. #endif
  90. FILE *fp = NULL;
  91. struct linger lg;
  92. int nlog = 5000;
  93. char loghelp[] =
  94. #ifdef STDMAIN
  95. #ifndef _WIN32
  96. " -I inetd mode (requires real socket, doesn't work with TTY)\n"
  97. " -l@IDENT log to syslog IDENT\n"
  98. #endif
  99. " -d go to background (daemon)\n"
  100. #else
  101. " -u never ask for username\n"
  102. " -u2 always ask for username\n"
  103. #endif
  104. " -fFORMAT logging format (see documentation)\n"
  105. " -l log to stderr\n"
  106. " -lFILENAME log to FILENAME\n"
  107. " -bBUFSIZE size of network buffer (default 4096 for TCP, 16384 for UDP)\n"
  108. " -t be silent (do not log service start/stop)\n"
  109. " -iIP ip address or internal interface (clients are expected to connect)\n"
  110. " -eIP ip address or external interface (outgoing connection will have this)\n"
  111. " -rHOST:PORT Use IP:port for connect back proxy instead of listen port\n"
  112. " -RHOST:PORT Use PORT to listen connect back proxy connection to pass data to\n"
  113. " -4 Use IPv4 for outgoing connections\n"
  114. " -6 Use IPv6 for outgoing connections\n"
  115. " -46 Prefer IPv4 for outgoing connections, use both IPv4 and IPv6\n"
  116. " -64 Prefer IPv6 for outgoing connections, use both IPv4 and IPv6\n";
  117. #ifdef _WIN32
  118. unsigned long ul = 1;
  119. #else
  120. #ifdef STDMAIN
  121. int inetd = 0;
  122. #endif
  123. #endif
  124. #ifdef _WIN32
  125. HANDLE h;
  126. #endif
  127. #ifdef STDMAIN
  128. #ifdef _WINCE
  129. argc = ceparseargs((char *)lpCmdLine);
  130. argv = ceargv;
  131. if(FindWindow(lpCmdLine, lpCmdLine)) return 0;
  132. ZeroMemory(&wc,sizeof(wc));
  133. wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH);
  134. wc.hInstance=hInstance;
  135. wc.hCursor=LoadCursor(NULL,IDC_ARROW);
  136. wc.lpfnWndProc=DefWindowProc;
  137. wc.style=CS_HREDRAW|CS_VREDRAW;
  138. wc.lpszClassName=lpCmdLine;
  139. RegisterClass(&wc);
  140. hwnd = CreateWindowEx(WS_EX_TOOLWINDOW,lpCmdLine,lpCmdLine,WS_VISIBLE|WS_POPUP,0,0,0,0,0,0,hInstance,0);
  141. #endif
  142. #ifdef _WIN32
  143. WSADATA wd;
  144. WSAStartup(MAKEWORD( 1, 1 ), &wd);
  145. #else
  146. signal(SIGPIPE, SIG_IGN);
  147. pthread_attr_init(&pa);
  148. pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + 8192);
  149. pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
  150. #endif
  151. #endif
  152. srvinit(&srv, &defparam);
  153. srv.pf = childdef.pf;
  154. isudp = childdef.isudp;
  155. srv.service = defparam.service = childdef.service;
  156. #ifndef STDMAIN
  157. srv.acl = copyacl(conf.acl);
  158. srv.authfuncs = copyauth(conf.authfuncs);
  159. if(!conf.services){
  160. conf.services = &srv;
  161. }
  162. else {
  163. srv.next = conf.services;
  164. conf.services = conf.services->prev = &srv;
  165. }
  166. #else
  167. srv.needuser = 0;
  168. #endif
  169. for (i=1; i<argc; i++) {
  170. if(*argv[i]=='-') {
  171. switch(argv[i][1]) {
  172. case 'd':
  173. if(!conf.demon)daemonize();
  174. conf.demon = 1;
  175. break;
  176. case 'l':
  177. srv.logfunc = logstdout;
  178. if(srv.logtarget) myfree(srv.logtarget);
  179. srv.logtarget = mystrdup((unsigned char*)argv[i] + 2);
  180. if(argv[i][2]) {
  181. if(argv[i][2]=='@'){
  182. #ifdef STDMAIN
  183. #ifndef _WIN32
  184. openlog(argv[i]+3, LOG_PID, LOG_DAEMON);
  185. srv.logfunc = logsyslog;
  186. #endif
  187. #endif
  188. }
  189. else {
  190. fp = fopen(argv[i] + 2, "a");
  191. if (fp) {
  192. srv.stdlog = fp;
  193. fseek(fp, 0L, SEEK_END);
  194. }
  195. }
  196. }
  197. break;
  198. case 'i':
  199. getip46(46, argv[i]+2, (struct sockaddr *)&srv.intsa);
  200. break;
  201. case 'e':
  202. {
  203. #ifndef NOIPV6
  204. struct sockaddr_in6 sa6;
  205. error = !getip46(46, argv[i]+2, (struct sockaddr *)&sa6);
  206. if(!error) memcpy((*SAFAMILY(&sa6)==AF_INET)?(void *)&srv.extsa:(void *)&srv.extsa6, &sa6, SASIZE(&sa6));
  207. #else
  208. error = !getip46(46, argv[i]+2, (struct sockaddr *)&srv.extsa);
  209. #endif
  210. }
  211. break;
  212. case 'p':
  213. *SAPORT(&srv.intsa) = htons(atoi(argv[i]+2));
  214. break;
  215. case '4':
  216. case '6':
  217. srv.family = atoi(argv[i]+1);
  218. break;
  219. case 'b':
  220. srv.bufsize = atoi(argv[i]+2);
  221. break;
  222. case 'n':
  223. srv.usentlm = atoi(argv[i]+2);
  224. break;
  225. #ifdef STDMAIN
  226. #ifndef _WIN32
  227. case 'I':
  228. size = sizeof(defparam.sincl);
  229. if(so._getsockname(0, (struct sockaddr*)&defparam.sincl, &size) ||
  230. SAFAMILY(&defparam.sincl) != AF_INET) error = 1;
  231. else inetd = 1;
  232. break;
  233. #endif
  234. #endif
  235. case 'f':
  236. if(srv.logformat)myfree(srv.logformat);
  237. srv.logformat = mystrdup((unsigned char *)argv[i] + 2);
  238. break;
  239. case 't':
  240. srv.silent = 1;
  241. break;
  242. case 'h':
  243. hostname = argv[i] + 2;
  244. break;
  245. case 'r':
  246. cbc_string = mystrdup(argv[i] + 2);
  247. iscbc = 1;
  248. break;
  249. case 'R':
  250. cbl_string = mystrdup(argv[i] + 2);
  251. iscbl = 1;
  252. break;
  253. case 'u':
  254. srv.needuser = 0;
  255. if(*(argv[i] + 2)) srv.needuser = atoi(argv[i] + 2);
  256. break;
  257. case 'T':
  258. srv.transparent = 1;
  259. break;
  260. case 's':
  261. case 'a':
  262. srv.singlepacket = 1 + atoi(argv[i]+2);
  263. break;
  264. default:
  265. error = 1;
  266. break;
  267. }
  268. }
  269. else break;
  270. }
  271. #ifndef STDMAIN
  272. if(childdef.port) {
  273. #endif
  274. #ifndef PORTMAP
  275. if (error || i!=argc) {
  276. #ifndef STDMAIN
  277. haveerror = 1;
  278. conf.threadinit = 0;
  279. #endif
  280. fprintf(stderr, "%s of " VERSION " (" BUILDDATE ")\n"
  281. "Usage: %s options\n"
  282. "Available options are:\n"
  283. "%s"
  284. " -pPORT - service port to accept connections\n"
  285. " -RIP:PORT - connect back IP:PORT to listen and accept connections\n"
  286. " -rIP:PORT - connect back IP:PORT to establish connect back connection\n"
  287. "%s"
  288. "\tExample: %s -i127.0.0.1\n\n"
  289. "%s",
  290. argv[0], argv[0], loghelp, childdef.helpmessage, argv[0],
  291. #ifdef STDMAIN
  292. copyright
  293. #else
  294. ""
  295. #endif
  296. );
  297. return (1);
  298. }
  299. #endif
  300. #ifndef STDMAIN
  301. }
  302. else {
  303. #endif
  304. #ifndef NOPORTMAP
  305. 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) {
  306. #ifndef STDMAIN
  307. haveerror = 1;
  308. conf.threadinit = 0;
  309. #endif
  310. fprintf(stderr, "%s of " VERSION " (" BUILDDATE ")\n"
  311. "Usage: %s options"
  312. " [-e<external_ip>] <port_to_bind>"
  313. " <target_hostname> <target_port>\n"
  314. "Available options are:\n"
  315. " -RIP:PORT - connect back IP:PORT to listen and accept connections\n"
  316. " -rIP:PORT - connect back IP:PORT to establish connect back connection\n"
  317. "%s"
  318. "%s"
  319. "\tExample: %s -d -i127.0.0.1 6666 serv.somehost.ru 6666\n\n"
  320. "%s",
  321. argv[0], argv[0], loghelp, childdef.helpmessage, argv[0],
  322. #ifdef STDMAIN
  323. copyright
  324. #else
  325. ""
  326. #endif
  327. );
  328. return (1);
  329. }
  330. srv.target = (unsigned char *)mystrdup(argv[i+1]);
  331. #endif
  332. #ifndef STDMAIN
  333. }
  334. #else
  335. #ifndef _WIN32
  336. if(inetd) {
  337. fcntl(0,F_SETFL,O_NONBLOCK);
  338. if(!isudp){
  339. so._setsockopt(0, SOL_SOCKET, SO_LINGER, (unsigned char *)&lg, sizeof(lg));
  340. so._setsockopt(0, SOL_SOCKET, SO_OOBINLINE, (unsigned char *)&opt, sizeof(int));
  341. }
  342. defparam.clisock = 0;
  343. if(! (newparam = myalloc (sizeof(defparam)))){
  344. return 2;
  345. };
  346. memcpy(newparam, &defparam, sizeof(defparam));
  347. return((*srv.pf)((void *)newparam)? 1:0);
  348. }
  349. #endif
  350. #endif
  351. srvinit2(&srv, &defparam);
  352. if(!*SAFAMILY(&srv.intsa)) *SAFAMILY(&srv.intsa) = AF_INET;
  353. if(!*SAPORT(&srv.intsa)) *SAPORT(&srv.intsa) = htons(childdef.port);
  354. *SAFAMILY(&srv.extsa) = AF_INET;
  355. #ifndef NOIPV6
  356. *SAFAMILY(&srv.extsa6) = AF_INET6;
  357. #endif
  358. if(hostname)parsehostname(hostname, &defparam, childdef.port);
  359. #ifndef STDMAIN
  360. copyfilter(conf.filters, &srv);
  361. conf.threadinit = 0;
  362. #endif
  363. if (!iscbc) {
  364. if(srv.srvsock == INVALID_SOCKET){
  365. if(!isudp){
  366. lg.l_onoff = 1;
  367. lg.l_linger = conf.timeouts[STRING_L];
  368. sock=so._socket(SASOCK(&srv.intsa), SOCK_STREAM, IPPROTO_TCP);
  369. }
  370. else {
  371. sock=so._socket(SASOCK(&srv.intsa), SOCK_DGRAM, IPPROTO_UDP);
  372. }
  373. if( sock == INVALID_SOCKET) {
  374. perror("socket()");
  375. return -2;
  376. }
  377. #ifdef _WIN32
  378. ioctlsocket(sock, FIONBIO, &ul);
  379. #else
  380. fcntl(sock,F_SETFL,O_NONBLOCK);
  381. #endif
  382. srv.srvsock = sock;
  383. opt = 1;
  384. if(so._setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (unsigned char *)&opt, sizeof(int)))perror("setsockopt()");
  385. #ifdef SO_REUSEPORT
  386. opt = 1;
  387. so._setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (unsigned char *)&opt, sizeof(int));
  388. #endif
  389. }
  390. size = sizeof(srv.intsa);
  391. for(sleeptime = SLEEPTIME * 100; so._bind(sock, (struct sockaddr*)&srv.intsa, size)==-1; usleep(sleeptime)) {
  392. sprintf((char *)buf, "bind(): %s", strerror(errno));
  393. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  394. sleeptime = (sleeptime<<1);
  395. if(!sleeptime) {
  396. so._closesocket(sock);
  397. return -3;
  398. }
  399. }
  400. if(!isudp){
  401. if(so._listen (sock, 1 + (srv.maxchild>>4))==-1) {
  402. sprintf((char *)buf, "listen(): %s", strerror(errno));
  403. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  404. return -4;
  405. }
  406. }
  407. else
  408. defparam.clisock = sock;
  409. if(!srv.silent && !iscbc){
  410. sprintf((char *)buf, "Accepting connections [%u/%u]", (unsigned)getpid(), (unsigned)pthread_self());
  411. (*srv.logfunc)(&defparam, buf);
  412. }
  413. }
  414. if(iscbl){
  415. parsehost(srv.family, cbl_string, (struct sockaddr *)&cbsa);
  416. if((srv.cbsock=so._socket(SASOCK(&cbsa), SOCK_STREAM, IPPROTO_TCP))==INVALID_SOCKET) {
  417. (*srv.logfunc)(&defparam, "Failed to allocate connect back socket");
  418. return -6;
  419. }
  420. opt = 1;
  421. so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEADDR, (unsigned char *)&opt, sizeof(int));
  422. #ifdef SO_REUSEPORT
  423. opt = 1;
  424. so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEPORT, (unsigned char *)&opt, sizeof(int));
  425. #endif
  426. if(so._bind(srv.cbsock, (struct sockaddr*)&cbsa, sizeof(cbsa))==-1) {
  427. (*srv.logfunc)(&defparam, "Failed to bind connect back socket");
  428. return -7;
  429. }
  430. if(so._listen(srv.cbsock, 1 + (srv.maxchild>>4))==-1) {
  431. (*srv.logfunc)(&defparam, "Failed to listen connect back socket");
  432. return -8;
  433. }
  434. }
  435. srv.fds.fd = sock;
  436. srv.fds.events = POLLIN;
  437. for (;;) {
  438. for(;;){
  439. while((conf.paused == srv.version && srv.childcount >= srv.maxchild)){
  440. nlog++;
  441. if(nlog > 5000) {
  442. sprintf((char *)buf, "Warning: too many connected clients (%d/%d)", srv.childcount, srv.maxchild);
  443. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  444. nlog = 0;
  445. }
  446. usleep(SLEEPTIME);
  447. }
  448. if (iscbc) break;
  449. if (conf.paused != srv.version) break;
  450. if (srv.fds.events & POLLIN) {
  451. error = so._poll(&srv.fds, 1, 1000);
  452. }
  453. else {
  454. usleep(SLEEPTIME);
  455. continue;
  456. }
  457. if (error >= 1) break;
  458. if (error == 0) continue;
  459. if (errno != EAGAIN && errno != EINTR) {
  460. sprintf((char *)buf, "poll(): %s/%d", strerror(errno), errno);
  461. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  462. break;
  463. }
  464. }
  465. if((conf.paused != srv.version) || (error < 0)) break;
  466. if(!isudp){
  467. size = sizeof(defparam.sincr);
  468. if(iscbc){
  469. new_sock=so._socket(SASOCK(&defparam.sincr), SOCK_STREAM, IPPROTO_TCP);
  470. if(new_sock != INVALID_SOCKET){
  471. parsehost(srv.family, cbc_string, (struct sockaddr *)&defparam.sincr);
  472. if(so._connect(new_sock,(struct sockaddr *)&defparam.sincr,sizeof(defparam.sincr))) {
  473. so._closesocket(new_sock);
  474. new_sock = INVALID_SOCKET;
  475. usleep(SLEEPTIME);
  476. continue;
  477. }
  478. if(so._recvfrom(new_sock,buf,1,0,(struct sockaddr*)&defparam.sincr, &size) != 1) {
  479. so._closesocket(new_sock);
  480. new_sock = INVALID_SOCKET;
  481. usleep(SLEEPTIME);
  482. continue;
  483. }
  484. }
  485. else {
  486. usleep(SLEEPTIME);
  487. continue;
  488. }
  489. }
  490. else {
  491. new_sock = so._accept(sock, (struct sockaddr*)&defparam.sincr, &size);
  492. if(new_sock == INVALID_SOCKET){
  493. sprintf((char *)buf, "accept(): %s", strerror(errno));
  494. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  495. continue;
  496. }
  497. }
  498. size = sizeof(defparam.sincl);
  499. if(so._getsockname(new_sock, (struct sockaddr *)&defparam.sincl, &size)){
  500. sprintf((char *)buf, "getsockname(): %s", strerror(errno));
  501. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  502. continue;
  503. }
  504. #ifdef _WIN32
  505. ioctlsocket(new_sock, FIONBIO, &ul);
  506. #else
  507. fcntl(new_sock,F_SETFL,O_NONBLOCK);
  508. #endif
  509. so._setsockopt(new_sock, SOL_SOCKET, SO_LINGER, (unsigned char *)&lg, sizeof(lg));
  510. so._setsockopt(new_sock, SOL_SOCKET, SO_OOBINLINE, (unsigned char *)&opt, sizeof(int));
  511. }
  512. else {
  513. srv.fds.events = 0;
  514. }
  515. if(! (newparam = myalloc (sizeof(defparam)))){
  516. if(!isudp) so._closesocket(new_sock);
  517. defparam.res = 21;
  518. if(!srv.silent)(*srv.logfunc)(&defparam, (unsigned char *)"Memory Allocation Failed");
  519. usleep(SLEEPTIME);
  520. continue;
  521. };
  522. memcpy(newparam, &defparam, sizeof(defparam));
  523. if(defparam.hostname)newparam->hostname=strdup(defparam.hostname);
  524. clearstat(newparam);
  525. if(!isudp) newparam->clisock = new_sock;
  526. #ifndef STDMAIN
  527. if(makefilters(&srv, newparam) > CONTINUE){
  528. freeparam(newparam);
  529. continue;
  530. }
  531. #endif
  532. newparam->prev = newparam->next = NULL;
  533. pthread_mutex_lock(&srv.counter_mutex);
  534. if(!srv.child){
  535. srv.child = newparam;
  536. }
  537. else {
  538. newparam->next = srv.child;
  539. srv.child = srv.child->prev = newparam;
  540. }
  541. #ifdef _WIN32
  542. #ifndef _WINCE
  543. h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, (unsigned)16384, threadfunc, (void *) newparam, 0, &thread);
  544. #else
  545. h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, (unsigned)16384, threadfunc, (void *) newparam, 0, &thread);
  546. #endif
  547. srv.childcount++;
  548. if (h) {
  549. newparam->threadid = (unsigned)thread;
  550. CloseHandle(h);
  551. }
  552. else {
  553. sprintf((char *)buf, "_beginthreadex(): %s", _strerror(NULL));
  554. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  555. freeparam(newparam);
  556. }
  557. #else
  558. error = pthread_create(&thread, &pa, threadfunc, (void *)newparam);
  559. srv.childcount++;
  560. if(error){
  561. sprintf((char *)buf, "pthread_create(): %s", strerror(error));
  562. if(!srv.silent)(*srv.logfunc)(&defparam, buf);
  563. freeparam(newparam);
  564. }
  565. else {
  566. newparam->threadid = (unsigned)thread;
  567. }
  568. #endif
  569. pthread_mutex_unlock(&srv.counter_mutex);
  570. memset(&defparam.sincl, 0, sizeof(defparam.sincl));
  571. memset(&defparam.sincr, 0, sizeof(defparam.sincr));
  572. if(isudp) while(!srv.fds.events)usleep(SLEEPTIME);
  573. }
  574. if(!srv.silent) srv.logfunc(&defparam, (unsigned char *)"Exiting thread");
  575. if(fp) fclose(fp);
  576. srvfree(&srv);
  577. #ifndef STDMAIN
  578. pthread_mutex_lock(&config_mutex);
  579. if(srv.next)srv.next->prev = srv.prev;
  580. if(srv.prev)srv.prev->next = srv.next;
  581. else conf.services = srv.next;
  582. pthread_mutex_unlock(&config_mutex);
  583. #endif
  584. if(defparam.hostname)myfree(defparam.hostname);
  585. if(cbc_string)myfree(cbc_string);
  586. if(cbl_string)myfree(cbl_string);
  587. return 0;
  588. }
  589. void srvinit(struct srvparam * srv, struct clientparam *param){
  590. memset(srv, 0, sizeof(struct srvparam));
  591. srv->version = conf.paused;
  592. srv->logfunc = conf.logfunc;
  593. if(srv->logformat)myfree(srv->logformat);
  594. srv->logformat = conf.logformat? mystrdup(conf.logformat) : NULL;
  595. srv->authfunc = conf.authfunc;
  596. srv->usentlm = 0;
  597. srv->maxchild = conf.maxchild;
  598. srv->time_start = time(NULL);
  599. if(conf.logtarget){
  600. if(srv->logtarget) myfree(srv->logtarget);
  601. srv->logtarget = mystrdup(conf.logtarget);
  602. }
  603. srv->srvsock = INVALID_SOCKET;
  604. srv->logdumpsrv = conf.logdumpsrv;
  605. srv->logdumpcli = conf.logdumpcli;
  606. srv->cbsock = INVALID_SOCKET;
  607. srv->needuser = 1;
  608. memset(param, 0, sizeof(struct clientparam));
  609. param->srv = srv;
  610. param->remsock = param->clisock = param->ctrlsock = param->ctrlsocksrv = INVALID_SOCKET;
  611. *SAFAMILY(&param->req) = *SAFAMILY(&param->sinsl) = *SAFAMILY(&param->sinsr) = *SAFAMILY(&param->sincr) = *SAFAMILY(&param->sincl) = AF_INET;
  612. pthread_mutex_init(&srv->counter_mutex, NULL);
  613. memcpy(&srv->intsa, &conf.intsa, sizeof(srv->intsa));
  614. memcpy(&srv->extsa, &conf.extsa, sizeof(srv->extsa));
  615. #ifndef NOIPV6
  616. memcpy(&srv->extsa6, &conf.extsa6, sizeof(srv->extsa6));
  617. #endif
  618. }
  619. void srvinit2(struct srvparam * srv, struct clientparam *param){
  620. if(srv->logformat){
  621. char *s;
  622. if(*srv->logformat == '-' && (s = strchr((char *)srv->logformat + 1, '+')) && s[1]){
  623. char* logformat = srv->logformat;
  624. *s = 0;
  625. srv->nonprintable = (unsigned char *)mystrdup((char *)srv->logformat + 1);
  626. srv->replace = s[1];
  627. srv->logformat = (unsigned char *)mystrdup(s + 2);
  628. *s = '+';
  629. myfree(logformat);
  630. }
  631. }
  632. memset(&param->sinsl, 0, sizeof(param->sinsl));
  633. memset(&param->sinsr, 0, sizeof(param->sinsr));
  634. memset(&param->req, 0, sizeof(param->req));
  635. *SAFAMILY(&param->sinsl) = AF_INET;
  636. *SAFAMILY(&param->sinsr) = AF_INET;
  637. *SAFAMILY(&param->req) = AF_INET;
  638. memcpy(&param->sincr, &srv->intsa, sizeof(param->sincr));
  639. memcpy(&param->sincl, &srv->intsa, sizeof(param->sincl));
  640. #ifndef NOIPV6
  641. memcpy(&param->sinsr, (srv->family == 6 || srv->family == 64)? (void *)&srv->extsa6: (void *)&srv->extsa, sizeof(param->sinsl));
  642. #else
  643. memcpy(&param->sinsr, &srv->extsa, sizeof(param->sinsl));
  644. #endif
  645. }
  646. void srvfree(struct srvparam * srv){
  647. if(srv->srvsock != INVALID_SOCKET) so._closesocket(srv->srvsock);
  648. srv->srvsock = INVALID_SOCKET;
  649. if(srv->cbsock != INVALID_SOCKET) so._closesocket(srv->cbsock);
  650. srv->cbsock = INVALID_SOCKET;
  651. srv->service = S_ZOMBIE;
  652. while(srv->child) usleep(SLEEPTIME * 100);
  653. #ifndef STDMAIN
  654. if(srv->filter){
  655. while(srv->nfilters){
  656. srv->nfilters--;
  657. if(srv->filter[srv->nfilters].filter_close){
  658. (*srv->filter[srv->nfilters].filter_close)(srv->filter[srv->nfilters].data);
  659. }
  660. }
  661. myfree(srv->filter);
  662. }
  663. if(srv->acl)freeacl(srv->acl);
  664. if(srv->authfuncs)freeauth(srv->authfuncs);
  665. #endif
  666. pthread_mutex_destroy(&srv->counter_mutex);
  667. if(srv->target) myfree(srv->target);
  668. if(srv->logtarget) myfree(srv->logtarget);
  669. if(srv->logformat) myfree(srv->logformat);
  670. if(srv->nonprintable) myfree(srv->nonprintable);
  671. }
  672. void freeparam(struct clientparam * param) {
  673. if(param->res == 2) return;
  674. if(param->datfilterssrv) myfree(param->datfilterssrv);
  675. #ifndef STDMAIN
  676. if(param->reqfilters) myfree(param->reqfilters);
  677. if(param->hdrfilterscli) myfree(param->hdrfilterscli);
  678. if(param->hdrfilterssrv) myfree(param->hdrfilterssrv);
  679. if(param->predatfilters) myfree(param->predatfilters);
  680. if(param->datfilterscli) myfree(param->datfilterscli);
  681. if(param->filters){
  682. if(param->nfilters)while(param->nfilters--){
  683. if(param->filters[param->nfilters].filter->filter_clear)
  684. (*param->filters[param->nfilters].filter->filter_clear)(param->filters[param->nfilters].data);
  685. }
  686. myfree(param->filters);
  687. }
  688. #endif
  689. if(param->clibuf) myfree(param->clibuf);
  690. if(param->srvbuf) myfree(param->srvbuf);
  691. if(param->srv){
  692. pthread_mutex_lock(&param->srv->counter_mutex);
  693. if(param->prev){
  694. param->prev->next = param->next;
  695. }
  696. else
  697. param->srv->child = param->next;
  698. if(param->next){
  699. param->next->prev = param->prev;
  700. }
  701. (param->srv->childcount)--;
  702. pthread_mutex_unlock(&param->srv->counter_mutex);
  703. }
  704. if(param->hostname) myfree(param->hostname);
  705. if(param->username) myfree(param->username);
  706. if(param->password) myfree(param->password);
  707. if(param->extusername) myfree(param->extusername);
  708. if(param->extpassword) myfree(param->extpassword);
  709. if(param->ctrlsocksrv != INVALID_SOCKET && param->ctrlsocksrv != param->remsock) {
  710. so._shutdown(param->ctrlsocksrv, SHUT_RDWR);
  711. so._closesocket(param->ctrlsocksrv);
  712. }
  713. if(param->ctrlsock != INVALID_SOCKET && param->ctrlsock != param->clisock) {
  714. so._shutdown(param->ctrlsock, SHUT_RDWR);
  715. so._closesocket(param->ctrlsock);
  716. }
  717. if(param->remsock != INVALID_SOCKET) {
  718. so._shutdown(param->remsock, SHUT_RDWR);
  719. so._closesocket(param->remsock);
  720. }
  721. if(param->clisock != INVALID_SOCKET) {
  722. so._shutdown(param->clisock, SHUT_RDWR);
  723. so._closesocket(param->clisock);
  724. }
  725. myfree(param);
  726. }
  727. #ifndef STDMAIN
  728. static void * itcopy (void * from, size_t size){
  729. void * ret;
  730. if(!from) return NULL;
  731. ret = myalloc(size);
  732. if(ret) memcpy(ret, from, size);
  733. return ret;
  734. }
  735. struct auth * copyauth (struct auth * authfuncs){
  736. struct auth * newauth = NULL;
  737. newauth = authfuncs = itcopy(authfuncs, sizeof(struct auth));
  738. for( ; authfuncs; authfuncs = authfuncs->next = itcopy(authfuncs->next, sizeof(struct auth)));
  739. return newauth;
  740. }
  741. struct ace * copyacl (struct ace *ac){
  742. struct ace * ret = NULL;
  743. struct iplist *ipl;
  744. struct portlist *pl;
  745. struct userlist *ul;
  746. struct chain *ch;
  747. struct period *pel;
  748. struct hostname *hst;
  749. ret = ac = itcopy(ac, sizeof(struct ace));
  750. for( ; ac; ac = ac->next = itcopy(ac->next, sizeof(struct ace))){
  751. ac->src = itcopy(ac->src, sizeof(struct iplist));
  752. for(ipl = ac->src; ipl; ipl = ipl->next = itcopy(ipl->next, sizeof(struct iplist)));
  753. ac->dst = itcopy(ac->dst, sizeof(struct iplist));
  754. for(ipl = ac->dst; ipl; ipl = ipl->next = itcopy(ipl->next, sizeof(struct iplist)));
  755. ac->ports = itcopy(ac->ports, sizeof(struct portlist));
  756. for(pl = ac->ports; pl; pl = pl->next = itcopy(pl->next, sizeof(struct portlist)));
  757. ac->periods = itcopy(ac->periods, sizeof(struct period));
  758. for(pel = ac->periods; pel; pel = pel->next = itcopy(pel->next, sizeof(struct period)));
  759. ac->users = itcopy(ac->users, sizeof(struct userlist));
  760. for(ul = ac->users; ul; ul = ul->next = itcopy(ul->next, sizeof(struct userlist))){
  761. if(ul->user) ul->user = (unsigned char*)mystrdup((char *)ul->user);
  762. }
  763. ac->dstnames = itcopy(ac->dstnames, sizeof(struct hostname));
  764. for(hst = ac->dstnames; hst; hst = hst->next = itcopy(hst->next, sizeof(struct hostname))){
  765. if(hst->name) hst->name = (unsigned char*)mystrdup((char *)hst->name);
  766. }
  767. ac->chains = itcopy(ac->chains, sizeof(struct chain));
  768. for(ch = ac->chains; ch; ch = ch->next = itcopy(ch->next, sizeof(struct chain))){
  769. if(ch->extuser)ch->extuser = (unsigned char*)mystrdup((char *)ch->extuser);
  770. if(ch->extpass)ch->extpass = (unsigned char*)mystrdup((char *)ch->extpass);
  771. }
  772. }
  773. return ret;
  774. }
  775. void copyfilter (struct filter *filter, struct srvparam *srv){
  776. int nfilters = 0;
  777. if(!filter) return;
  778. for(srv->filter = filter; srv->filter; srv->filter = srv->filter->next) nfilters++;
  779. srv->filter = myalloc(sizeof(struct filter) * nfilters);
  780. if(!srv->filter) return;
  781. for(; filter; filter = filter->next){
  782. void *data = NULL;
  783. if(!filter->filter_open || !(data = (*filter->filter_open)(filter->data, srv))) continue;
  784. memcpy(srv->filter + srv->nfilters, filter, sizeof(struct filter));
  785. srv->filter[srv->nfilters].data = data;
  786. if(srv->nfilters>0)srv->filter[srv->nfilters - 1].next = srv->filter + srv->nfilters;
  787. srv->nfilters++;
  788. if(filter->filter_request)srv->nreqfilters++;
  789. if(filter->filter_header_srv)srv->nhdrfilterssrv++;
  790. if(filter->filter_header_cli)srv->nhdrfilterscli++;
  791. if(filter->filter_predata)srv->npredatfilters++;
  792. if(filter->filter_data_srv)srv->ndatfilterssrv++;
  793. if(filter->filter_data_cli)srv->ndatfilterscli++;
  794. }
  795. }
  796. FILTER_ACTION makefilters (struct srvparam *srv, struct clientparam *param){
  797. FILTER_ACTION res=PASS;
  798. FILTER_ACTION action;
  799. int i;
  800. if(!srv->nfilters) return PASS;
  801. if(!(param->filters = myalloc(sizeof(struct filterp) * srv->nfilters)) ||
  802. (srv->nreqfilters && !(param->reqfilters = myalloc(sizeof(struct filterp *) * srv->nreqfilters))) ||
  803. (srv->nhdrfilterssrv && !(param->hdrfilterssrv = myalloc(sizeof(struct filterp *) * srv->nhdrfilterssrv))) ||
  804. (srv->nhdrfilterscli && !(param->hdrfilterscli = myalloc(sizeof(struct filterp *) * srv->nhdrfilterscli))) ||
  805. (srv->npredatfilters && !(param->predatfilters = myalloc(sizeof(struct filterp *) * srv->npredatfilters))) ||
  806. (srv->ndatfilterssrv && !(param->datfilterssrv = myalloc(sizeof(struct filterp *) * srv->ndatfilterssrv))) ||
  807. (srv->ndatfilterscli && !(param->datfilterscli = myalloc(sizeof(struct filterp *) * srv->ndatfilterscli)))
  808. ){
  809. param->res = 21;
  810. return REJECT;
  811. }
  812. for(i=0; i<srv->nfilters; i++){
  813. if(!srv->filter[i].filter_client)continue;
  814. action = (*srv->filter[i].filter_client)(srv->filter[i].data, param, &param->filters[param->nfilters].data);
  815. if(action == PASS) continue;
  816. if(action > CONTINUE) return action;
  817. param->filters[param->nfilters].filter = srv->filter + i;
  818. if(srv->filter[i].filter_request)param->reqfilters[param->nreqfilters++] = param->filters + param->nfilters;
  819. if(srv->filter[i].filter_header_cli)param->hdrfilterscli[param->nhdrfilterscli++] = param->filters + param->nfilters;
  820. if(srv->filter[i].filter_header_srv)param->hdrfilterssrv[param->nhdrfilterssrv++] = param->filters + param->nfilters;
  821. if(srv->filter[i].filter_predata)param->predatfilters[param->npredatfilters++] = param->filters + param->nfilters;
  822. if(srv->filter[i].filter_data_cli)param->datfilterscli[param->ndatfilterscli++] = param->filters + param->nfilters;
  823. if(srv->filter[i].filter_data_srv)param->datfilterssrv[param->ndatfilterssrv++] = param->filters + param->nfilters;
  824. param->nfilters++;
  825. }
  826. return res;
  827. }
  828. void * itfree(void *data, void * retval){
  829. myfree(data);
  830. return retval;
  831. }
  832. void freeauth(struct auth * authfuncs){
  833. for(; authfuncs; authfuncs = (struct auth *)itfree(authfuncs, authfuncs->next));
  834. }
  835. void freeacl(struct ace *ac){
  836. struct iplist *ipl;
  837. struct portlist *pl;
  838. struct userlist *ul;
  839. struct chain *ch;
  840. struct period *pel;
  841. struct hostname *hst;
  842. for(; ac; ac = (struct ace *) itfree(ac, ac->next)){
  843. for(ipl = ac->src; ipl; ipl = (struct iplist *)itfree(ipl, ipl->next));
  844. for(ipl = ac->dst; ipl; ipl = (struct iplist *)itfree(ipl,ipl->next));
  845. for(pl = ac->ports; pl; pl = (struct portlist *)itfree(pl, pl->next));
  846. for(pel = ac->periods; pel; pel = (struct period *)itfree(pel, pel->next));
  847. for(ul = ac->users; ul; ul = (struct userlist *)itfree(ul, ul->next)){
  848. if(ul->user)myfree(ul->user);
  849. }
  850. for(hst = ac->dstnames; hst; hst = (struct hostname *)itfree(hst, hst->next)){
  851. if(hst->name)myfree(hst->name);
  852. }
  853. for(ch = ac->chains; ch; ch = (struct chain *) itfree(ch, ch->next)){
  854. if(ch->extuser) myfree(ch->extuser);
  855. if(ch->extpass) myfree(ch->extpass);
  856. }
  857. }
  858. }
  859. FILTER_ACTION handlereqfilters(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  860. FILTER_ACTION action;
  861. int i;
  862. for(i=0; i<param->nreqfilters; i++){
  863. action = (*param->reqfilters[i]->filter->filter_request)(param->reqfilters[i]->data, param, buf_p, bufsize_p, offset, length_p);
  864. if(action!=CONTINUE) return action;
  865. }
  866. return PASS;
  867. }
  868. FILTER_ACTION handlehdrfilterssrv(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  869. FILTER_ACTION action;
  870. int i;
  871. for(i=0; i<param->nhdrfilterssrv; i++){
  872. action = (*param->hdrfilterssrv[i]->filter->filter_header_srv)(param->hdrfilterssrv[i]->data, param, buf_p, bufsize_p, offset, length_p);
  873. if(action!=CONTINUE) return action;
  874. }
  875. return PASS;
  876. }
  877. FILTER_ACTION handlehdrfilterscli(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  878. FILTER_ACTION action;
  879. int i;
  880. for(i = 0; i < param->nhdrfilterscli; i++){
  881. action = (*param->hdrfilterscli[i]->filter->filter_header_cli)(param->hdrfilterscli[i]->data, param, buf_p, bufsize_p, offset, length_p);
  882. if(action!=CONTINUE) return action;
  883. }
  884. return PASS;
  885. }
  886. #endif
  887. FILTER_ACTION handlepredatflt(struct clientparam *cparam){
  888. #ifndef STDMAIN
  889. FILTER_ACTION action;
  890. int i;
  891. for(i=0; i<cparam->npredatfilters ;i++){
  892. action = (*cparam->predatfilters[i]->filter->filter_predata)(cparam->predatfilters[i]->data, cparam);
  893. if(action!=CONTINUE) return action;
  894. }
  895. #endif
  896. return PASS;
  897. }
  898. FILTER_ACTION handledatfltcli(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  899. #ifndef STDMAIN
  900. FILTER_ACTION action;
  901. int i;
  902. for(i=0; i<cparam->ndatfilterscli ;i++){
  903. action = (*cparam->datfilterscli[i]->filter->filter_data_cli)(cparam->datfilterscli[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
  904. if(action!=CONTINUE) return action;
  905. }
  906. #endif
  907. return PASS;
  908. }
  909. FILTER_ACTION handledatfltsrv(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
  910. FILTER_ACTION action;
  911. int i;
  912. for(i=0; i<cparam->ndatfilterssrv; i++){
  913. action = (*cparam->datfilterssrv[i]->filter->filter_data_srv)(cparam->datfilterssrv[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
  914. if(action!=CONTINUE) return action;
  915. }
  916. return PASS;
  917. }