| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204 |
- /*
- 3APA3A simpliest proxy server
- (c) 2002-2016 by Vladimir Dubrovin <3proxy@3proxy.ru>
- please read License Agreement
- */
- #include "proxy.h"
- pthread_mutex_t log_mutex;
- #define param ((struct clientparam *) p)
- #ifdef _WIN32
- DWORD WINAPI threadfunc(LPVOID p) {
- #else
- void * threadfunc (void *p) {
- #endif
- int i = 0;
- if(param->srv->cbsock != INVALID_SOCKET){
- SASIZETYPE size = sizeof(param->sinsr);
- for(i=0; i<3; i++){
- param->remsock = so._accept(param->srv->cbsock, (struct sockaddr*)¶m->sinsr, &size);
- if(param->remsock == INVALID_SOCKET) {
- param->res = 13;
- param->srv->logfunc(param, (unsigned char *)"Connect back accept() failed");
- continue;
- }
- #ifndef WITHMAIN
- param->req = param->sinsr;
- if(param->srv->acl) param->res = checkACL(param);
- if(param->res){
- param->srv->logfunc(param, (unsigned char *)"Connect back ACL failed");
- so._closesocket(param->remsock);
- param->remsock = INVALID_SOCKET;
- continue;
- }
- #endif
- if(so._sendto(param->remsock, "C", 1, 0, (struct sockaddr*)¶m->sinsr, size) != 1){
- param->srv->logfunc(param, (unsigned char *)"Connect back sending command failed");
- so._closesocket(param->remsock);
- param->remsock = INVALID_SOCKET;
- continue;
- }
-
- break;
- }
- }
- if(i == 3){
- freeparam(param);
- }
- else {
- ((struct clientparam *) p)->srv->pf((struct clientparam *)p);
- }
- #ifdef _WIN32
- return 0;
- #else
- return NULL;
- #endif
- }
- #undef param
- struct socketoptions sockopts[] = {
- #ifdef TCP_NODELAY
- {TCP_NODELAY, "TCP_NODELAY"},
- #endif
- #ifdef TCP_CORK
- {TCP_CORK, "TCP_CORK"},
- #endif
- #ifdef TCP_DEFER_ACCEPT
- {TCP_DEFER_ACCEPT, "TCP_DEFER_ACCEPT"},
- #endif
- #ifdef TCP_QUICKACK
- {TCP_QUICKACK, "TCP_QUICKACK"},
- #endif
- #ifdef TCP_TIMESTAMPS
- {TCP_TIMESTAMPS, "TCP_TIMESTAMPS"},
- #endif
- #ifdef USE_TCP_FASTOPEN
- {USE_TCP_FASTOPEN, "USE_TCP_FASTOPEN"},
- #endif
- #ifdef SO_REUSEADDR
- {SO_REUSEADDR, "SO_REUSEADDR"},
- #endif
- #ifdef SO_REUSEPORT
- {SO_REUSEPORT, "SO_REUSEPORT"},
- #endif
- #ifdef SO_PORT_SCALABILITY
- {SO_PORT_SCALABILITY, "SO_PORT_SCALABILITY"},
- #endif
- #ifdef SO_REUSE_UNICASTPORT
- {SO_REUSE_UNICASTPORT, "SO_REUSE_UNICASTPORT"},
- #endif
- #ifdef SO_KEEPALIVE
- {SO_KEEPALIVE, "SO_KEEPALIVE"},
- #endif
- #ifdef SO_DONTROUTE
- {SO_DONTROUTE, "SO_DONTROUTE"},
- #endif
- {0, NULL}
- };
- int getopts(const char *s){
- int i=0, ret=0;
- for(; sockopts[i].optname; i++)if(strstr(s,sockopts[i].optname)) ret |= (1<<i);
- return ret;
- }
- void setopts(SOCKET s, int opts){
- int i, opt, set;
- for(i = 0; opts >= (opt = (1<<i)); i++){
- set = 1;
- if(opts & opt) setsockopt(s, *sockopts[i].optname == 'T'? IPPROTO_TCP:SOL_SOCKET, sockopts[i].opt, (char *)&set, sizeof(set));
- }
- }
- #ifndef MODULEMAINFUNC
- #define MODULEMAINFUNC main
- #define STDMAIN
- #ifndef _WINCE
- int main (int argc, char** argv){
- #else
- int WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow){
- int argc;
- char ** argv;
- WNDCLASS wc;
- HWND hwnd = 0;
- #endif
- #else
- extern int linenum;
- extern int haveerror;
- int MODULEMAINFUNC (int argc, char** argv){
- #endif
- SOCKET sock = INVALID_SOCKET, new_sock = INVALID_SOCKET;
- int i=0;
- SASIZETYPE size;
- pthread_t thread;
- struct clientparam defparam;
- struct srvparam srv;
- struct clientparam * newparam;
- int error = 0;
- unsigned sleeptime;
- unsigned char buf[256];
- char *hostname=NULL;
- int opt = 1, isudp = 0, iscbl = 0, iscbc = 0;
- unsigned char *cbc_string = NULL, *cbl_string = NULL;
- #ifndef NOIPV6
- struct sockaddr_in6 cbsa;
- #else
- struct sockaddr_in cbsa;
- #endif
- FILE *fp = NULL;
- struct linger lg;
- int nlog = 5000;
- char loghelp[] =
- #ifdef STDMAIN
- #ifndef _WIN32
- " -I inetd mode (requires real socket, doesn't work with TTY)\n"
- " -l@IDENT log to syslog IDENT\n"
- #endif
- " -d go to background (daemon)\n"
- #else
- " -u never ask for username\n"
- " -u2 always ask for username\n"
- #endif
- #ifdef WITHSLICE
- " -s Use slice() - faster proxing, but no filtering for data\n"
- #endif
- " -fFORMAT logging format (see documentation)\n"
- " -l log to stderr\n"
- " -lFILENAME log to FILENAME\n"
- " -b(BUFSIZE) size of network buffer (default 4096 for TCP, 16384 for UDP)\n"
- " -S(STACKSIZE) value to add to default client thread stack size\n"
- " -t be silent (do not log service start/stop)\n"
- " -ocOPTIONS, -osOPTIONS, -olOPTIONS - options for client (oc), server (os) or listening (ol) socket,"
- " where possible options are: "
- #ifdef TCP_NODELAY
- "TCP_NODELAY "
- #endif
- #ifdef TCP_CORK
- "TCP_CORK "
- #endif
- #ifdef TCP_DEFER_ACCEPT
- "TCP_DEFER_ACCEPT "
- #endif
- #ifdef TCP_QUICKACK
- "TCP_QUICKACK "
- #endif
- #ifdef TCP_TIMESTAMPS
- "TCP_TIMESTAMPS "
- #endif
- #ifdef USE_TCP_FASTOPEN
- "USE_TCP_FASTOPEN "
- #endif
- #ifdef SO_REUSEADDR
- "SO_REUSEADDR "
- #endif
- #ifdef SO_REUSEPORT
- "SO_REUSEPORT "
- #endif
- #ifdef SO_PORT_SCALABILITY
- "SO_PORT_SCALABILITY "
- #endif
- #ifdef SO_REUSE_UNICASTPORT
- "SO_REUSE_UNICASTPORT "
- #endif
- #ifdef SO_KEEPALIVE
- "SO_KEEPALIVE "
- #endif
- #ifdef SO_DONTROUTE
- "SO_DONTROUTE "
- #endif
- "\n"
- " -iIP ip address or internal interface (clients are expected to connect)\n"
- " -eIP ip address or external interface (outgoing connection will have this)\n"
- " -rHOST:PORT Use IP:port for connect back proxy instead of listen port\n"
- " -RHOST:PORT Use PORT to listen connect back proxy connection to pass data to\n"
- " -4 Use IPv4 for outgoing connections\n"
- " -6 Use IPv6 for outgoing connections\n"
- " -46 Prefer IPv4 for outgoing connections, use both IPv4 and IPv6\n"
- " -64 Prefer IPv6 for outgoing connections, use both IPv4 and IPv6\n";
- #ifdef _WIN32
- unsigned long ul = 1;
- #else
- pthread_attr_t pa;
- #ifdef STDMAIN
- int inetd = 0;
- #endif
- #endif
- #ifdef _WIN32
- HANDLE h;
- #endif
- #ifdef STDMAIN
- #ifdef _WINCE
- argc = ceparseargs((char *)lpCmdLine);
- argv = ceargv;
- if(FindWindow(lpCmdLine, lpCmdLine)) return 0;
- ZeroMemory(&wc,sizeof(wc));
- wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH);
- wc.hInstance=hInstance;
- wc.hCursor=LoadCursor(NULL,IDC_ARROW);
- wc.lpfnWndProc=DefWindowProc;
- wc.style=CS_HREDRAW|CS_VREDRAW;
- wc.lpszClassName=lpCmdLine;
- RegisterClass(&wc);
- hwnd = CreateWindowEx(WS_EX_TOOLWINDOW,lpCmdLine,lpCmdLine,WS_VISIBLE|WS_POPUP,0,0,0,0,0,0,hInstance,0);
- #endif
- #ifdef _WIN32
- WSADATA wd;
- WSAStartup(MAKEWORD( 1, 1 ), &wd);
- #else
- signal(SIGPIPE, SIG_IGN);
- #endif
- #endif
- srvinit(&srv, &defparam);
- srv.pf = childdef.pf;
- isudp = childdef.isudp;
- srv.service = defparam.service = childdef.service;
-
- #ifndef STDMAIN
- srv.acl = copyacl(conf.acl);
- srv.authfuncs = copyauth(conf.authfuncs);
- if(!conf.services){
- conf.services = &srv;
- }
- else {
- srv.next = conf.services;
- conf.services = conf.services->prev = &srv;
- }
- #else
- srv.needuser = 0;
- pthread_mutex_init(&log_mutex, NULL);
- #endif
- for (i=1; i<argc; i++) {
- if(*argv[i]=='-') {
- switch(argv[i][1]) {
- case 'd':
- if(!conf.demon)daemonize();
- conf.demon = 1;
- break;
- case 'l':
- srv.logfunc = logstdout;
- if(srv.logtarget) myfree(srv.logtarget);
- srv.logtarget = (unsigned char *)mystrdup(argv[i] + 2);
- if(argv[i][2]) {
- if(argv[i][2]=='@'){
- #ifdef STDMAIN
- #ifndef _WIN32
- openlog(argv[i]+3, LOG_PID, LOG_DAEMON);
- srv.logfunc = logsyslog;
- #endif
- #endif
- }
- else {
- fp = fopen(argv[i] + 2, "a");
- if (fp) {
- srv.stdlog = fp;
- fseek(fp, 0L, SEEK_END);
- }
- }
- }
- break;
- case 'i':
- getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&srv.intsa);
- break;
- case 'e':
- {
- #ifndef NOIPV6
- struct sockaddr_in6 sa6;
- memset(&sa6, 0, sizeof(sa6));
- error = !getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&sa6);
- if(!error) {
- if (*SAFAMILY(&sa6)==AF_INET) srv.extsa = sa6;
- else srv.extsa6 = sa6;
- }
- #else
- error = !getip46(46, (unsigned char *)argv[i]+2, (struct sockaddr *)&srv.extsa);
- #endif
- }
- break;
- case 'p':
- *SAPORT(&srv.intsa) = htons(atoi(argv[i]+2));
- break;
- case '4':
- case '6':
- srv.family = atoi(argv[i]+1);
- break;
- case 'b':
- srv.bufsize = atoi(argv[i]+2);
- break;
- case 'n':
- srv.usentlm = atoi(argv[i]+2);
- break;
- #ifdef STDMAIN
- #ifndef _WIN32
- case 'I':
- size = sizeof(defparam.sincl);
- if(so._getsockname(0, (struct sockaddr*)&defparam.sincl, &size) ||
- *SAFAMILY(&defparam.sincl) != AF_INET) error = 1;
- else inetd = 1;
- break;
- #endif
- #endif
- case 'f':
- if(srv.logformat)myfree(srv.logformat);
- srv.logformat = (unsigned char *)mystrdup(argv[i] + 2);
- break;
- case 't':
- srv.silent = 1;
- break;
- case 'h':
- hostname = argv[i] + 2;
- break;
- case 'r':
- cbc_string = (unsigned char *)mystrdup(argv[i] + 2);
- iscbc = 1;
- break;
- case 'R':
- cbl_string = (unsigned char *)mystrdup(argv[i] + 2);
- iscbl = 1;
- break;
- case 'u':
- srv.needuser = 0;
- if(*(argv[i] + 2)) srv.needuser = atoi(argv[i] + 2);
- break;
- case 'T':
- srv.transparent = 1;
- break;
- case 'S':
- srv.stacksize = atoi(argv[i]+2);
- break;
- case 'a':
- srv.anonymous = 1 + atoi(argv[i]+2);
- break;
- case 's':
- if(isudp)
- srv.singlepacket = 1 + atoi(argv[i]+2);
- #ifdef WITHSPLICE
- else
- srv.usesplice = 1 + atoi(argv[i]+2);
- #endif
- break;
- case 'o':
- if(argv[i][2] == 's'){
- srv.srvsockopts = getopts(argv[i]+3);
- break;
- }
- else if(argv[i][2] == 'c'){
- srv.clisockopts = getopts(argv[i]+3);
- break;
- }
- else if(argv[i][2] == 'l'){
- srv.lissockopts = getopts(argv[i]+3);
- break;
- }
- default:
- error = 1;
- break;
- }
- }
- else break;
- }
- #ifndef STDMAIN
- if(childdef.port) {
- #endif
- #ifndef PORTMAP
- if (error || i!=argc) {
- #ifndef STDMAIN
- haveerror = 1;
- conf.threadinit = 0;
- #endif
- fprintf(stderr, "%s of " VERSION " (" BUILDDATE ")\n"
- "Usage: %s options\n"
- "Available options are:\n"
- "%s"
- " -pPORT - service port to accept connections\n"
- " -RIP:PORT - connect back IP:PORT to listen and accept connections\n"
- " -rIP:PORT - connect back IP:PORT to establish connect back connection\n"
- "%s"
- "\tExample: %s -i127.0.0.1\n\n"
- "%s",
- argv[0], argv[0], loghelp, childdef.helpmessage, argv[0],
- #ifdef STDMAIN
- copyright
- #else
- ""
- #endif
- );
- return (1);
- }
- #endif
- #ifndef STDMAIN
- }
- else {
- #endif
- #ifndef NOPORTMAP
- 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) {
- #ifndef STDMAIN
- haveerror = 1;
- conf.threadinit = 0;
- #endif
- fprintf(stderr, "%s of " VERSION " (" BUILDDATE ")\n"
- "Usage: %s options"
- " [-e<external_ip>] <port_to_bind>"
- " <target_hostname> <target_port>\n"
- "Available options are:\n"
- " -RIP:PORT - connect back IP:PORT to listen and accept connections\n"
- " -rIP:PORT - connect back IP:PORT to establish connect back connection\n"
- "%s"
- "%s"
- "\tExample: %s -d -i127.0.0.1 6666 serv.somehost.ru 6666\n\n"
- "%s",
- argv[0], argv[0], loghelp, childdef.helpmessage, argv[0],
- #ifdef STDMAIN
- copyright
- #else
- ""
- #endif
- );
- return (1);
- }
- srv.target = (unsigned char *)mystrdup(argv[i+1]);
- #endif
- #ifndef STDMAIN
- }
- #else
- #ifndef _WIN32
- if(inetd) {
- fcntl(0,F_SETFL,O_NONBLOCK);
- if(!isudp){
- so._setsockopt(0, SOL_SOCKET, SO_LINGER, (unsigned char *)&lg, sizeof(lg));
- so._setsockopt(0, SOL_SOCKET, SO_OOBINLINE, (unsigned char *)&opt, sizeof(int));
- }
- defparam.clisock = 0;
- if(! (newparam = myalloc (sizeof(defparam)))){
- return 2;
- };
- *newparam = defparam;
- return((*srv.pf)((void *)newparam)? 1:0);
-
- }
- #endif
- #endif
-
- srvinit2(&srv, &defparam);
- if(!*SAFAMILY(&srv.intsa)) *SAFAMILY(&srv.intsa) = AF_INET;
- if(!*SAPORT(&srv.intsa)) *SAPORT(&srv.intsa) = htons(childdef.port);
- *SAFAMILY(&srv.extsa) = AF_INET;
- #ifndef NOIPV6
- *SAFAMILY(&srv.extsa6) = AF_INET6;
- #endif
- if(hostname)parsehostname(hostname, &defparam, childdef.port);
- #ifndef STDMAIN
- copyfilter(conf.filters, &srv);
- conf.threadinit = 0;
- #endif
- if (!iscbc) {
- if(srv.srvsock == INVALID_SOCKET){
- if(!isudp){
- lg.l_onoff = 1;
- lg.l_linger = conf.timeouts[STRING_L];
- sock=so._socket(SASOCK(&srv.intsa), SOCK_STREAM, IPPROTO_TCP);
- }
- else {
- sock=so._socket(SASOCK(&srv.intsa), SOCK_DGRAM, IPPROTO_UDP);
- }
- if( sock == INVALID_SOCKET) {
- perror("socket()");
- return -2;
- }
- setopts(sock, srv.lissockopts);
- #ifdef _WIN32
- ioctlsocket(sock, FIONBIO, &ul);
- #else
- fcntl(sock,F_SETFL,O_NONBLOCK);
- #endif
- srv.srvsock = sock;
- opt = 1;
- if(so._setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(int)))perror("setsockopt()");
- #ifdef SO_REUSEPORT
- opt = 1;
- so._setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, (char *)&opt, sizeof(int));
- #endif
- }
- size = sizeof(srv.intsa);
- for(sleeptime = SLEEPTIME * 100; so._bind(sock, (struct sockaddr*)&srv.intsa, SASIZE(&srv.intsa))==-1; usleep(sleeptime)) {
- sprintf((char *)buf, "bind(): %s", strerror(errno));
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- sleeptime = (sleeptime<<1);
- if(!sleeptime) {
- so._closesocket(sock);
- return -3;
- }
- }
- if(!isudp){
- if(so._listen (sock, 1 + (srv.maxchild>>4))==-1) {
- sprintf((char *)buf, "listen(): %s", strerror(errno));
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- return -4;
- }
- }
- else
- defparam.clisock = sock;
- if(!srv.silent && !iscbc){
- sprintf((char *)buf, "Accepting connections [%u/%u]", (unsigned)getpid(), (unsigned)pthread_self());
- (*srv.logfunc)(&defparam, buf);
- }
- }
- if(iscbl){
- parsehost(srv.family, cbl_string, (struct sockaddr *)&cbsa);
- if((srv.cbsock=so._socket(SASOCK(&cbsa), SOCK_STREAM, IPPROTO_TCP))==INVALID_SOCKET) {
- (*srv.logfunc)(&defparam, (unsigned char *)"Failed to allocate connect back socket");
- return -6;
- }
- opt = 1;
- so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(int));
- #ifdef SO_REUSEPORT
- opt = 1;
- so._setsockopt(srv.cbsock, SOL_SOCKET, SO_REUSEPORT, (char *)&opt, sizeof(int));
- #endif
- if(so._bind(srv.cbsock, (struct sockaddr*)&cbsa, SASIZE(&cbsa))==-1) {
- (*srv.logfunc)(&defparam, (unsigned char *)"Failed to bind connect back socket");
- return -7;
- }
- if(so._listen(srv.cbsock, 1 + (srv.maxchild>>4))==-1) {
- (*srv.logfunc)(&defparam, (unsigned char *)"Failed to listen connect back socket");
- return -8;
- }
- }
- srv.fds.fd = sock;
- srv.fds.events = POLLIN;
-
- #ifndef _WIN32
- pthread_attr_init(&pa);
- pthread_attr_setstacksize(&pa,PTHREAD_STACK_MIN + (16384 + srv.stacksize));
- pthread_attr_setdetachstate(&pa,PTHREAD_CREATE_DETACHED);
- #endif
- for (;;) {
- for(;;){
- while((conf.paused == srv.paused && srv.childcount >= srv.maxchild)){
- nlog++;
- if(!srv.silent && nlog > 5000) {
- sprintf((char *)buf, "Warning: too many connected clients (%d/%d)", srv.childcount, srv.maxchild);
- (*srv.logfunc)(&defparam, buf);
- nlog = 0;
- }
- usleep(SLEEPTIME);
- }
- if (iscbc) break;
- if (conf.paused != srv.paused) break;
- if (srv.fds.events & POLLIN) {
- error = so._poll(&srv.fds, 1, 1000);
- }
- else {
- usleep(SLEEPTIME);
- continue;
- }
- if (error >= 1) break;
- if (error == 0) continue;
- if (errno != EAGAIN && errno != EINTR) {
- sprintf((char *)buf, "poll(): %s/%d", strerror(errno), errno);
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- break;
- }
- }
- if((conf.paused != srv.paused) || (error < 0)) break;
- error = 0;
- if(!isudp){
- size = sizeof(defparam.sincr);
- if(iscbc){
- new_sock=so._socket(SASOCK(&defparam.sincr), SOCK_STREAM, IPPROTO_TCP);
- if(new_sock != INVALID_SOCKET){
- parsehost(srv.family, cbc_string, (struct sockaddr *)&defparam.sincr);
- if(connectwithpoll(new_sock,(struct sockaddr *)&defparam.sincr,SASIZE(&defparam.sincr))) {
- so._closesocket(new_sock);
- new_sock = INVALID_SOCKET;
- usleep(SLEEPTIME);
- continue;
- }
- if(so._recvfrom(new_sock,(char *)buf,1,0,(struct sockaddr*)&defparam.sincr, &size) != 1) {
- so._closesocket(new_sock);
- new_sock = INVALID_SOCKET;
- usleep(SLEEPTIME);
- continue;
- }
- }
- else {
- usleep(SLEEPTIME);
- continue;
- }
- }
- else {
- new_sock = so._accept(sock, (struct sockaddr*)&defparam.sincr, &size);
- if(new_sock == INVALID_SOCKET){
- #ifdef _WIN32
- switch(WSAGetLastError()){
- case WSAEMFILE:
- case WSAENOBUFS:
- case WSAENETDOWN:
- usleep(SLEEPTIME * 10);
- break;
- case WSAEINTR:
- error = 1;
- break;
- default:
- break;
- }
- #else
- switch (errno){
- #ifdef EMFILE
- case EMFILE:
- #endif
- #ifdef ENFILE
- case ENFILE:
- #endif
- #ifdef ENOBUFS
- case ENOBUFS:
- #endif
- #ifdef ENOMEM
- case ENOMEM:
- #endif
- usleep(SLEEPTIME * 10);
- break;
- default:
- break;
- }
- #endif
- nlog++;
- if(!srv.silent && (error || nlog > 5000)) {
- sprintf((char *)buf, "accept(): %s", strerror(errno));
- (*srv.logfunc)(&defparam, buf);
- nlog = 0;
- }
- continue;
- }
- }
- setopts(new_sock, srv.clisockopts);
- size = sizeof(defparam.sincl);
- if(so._getsockname(new_sock, (struct sockaddr *)&defparam.sincl, &size)){
- sprintf((char *)buf, "getsockname(): %s", strerror(errno));
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- continue;
- }
- #ifdef _WIN32
- ioctlsocket(new_sock, FIONBIO, &ul);
- #else
- fcntl(new_sock,F_SETFL,O_NONBLOCK);
- #endif
- so._setsockopt(new_sock, SOL_SOCKET, SO_LINGER, (char *)&lg, sizeof(lg));
- so._setsockopt(new_sock, SOL_SOCKET, SO_OOBINLINE, (char *)&opt, sizeof(int));
- }
- else {
- srv.fds.events = 0;
- }
- if(! (newparam = myalloc (sizeof(defparam)))){
- if(!isudp) so._closesocket(new_sock);
- defparam.res = 21;
- if(!srv.silent)(*srv.logfunc)(&defparam, (unsigned char *)"Memory Allocation Failed");
- usleep(SLEEPTIME);
- continue;
- };
- *newparam = defparam;
- if(defparam.hostname)newparam->hostname=(unsigned char *)strdup((char *)defparam.hostname);
- clearstat(newparam);
- if(!isudp) newparam->clisock = new_sock;
- #ifndef STDMAIN
- if(makefilters(&srv, newparam) > CONTINUE){
- freeparam(newparam);
- continue;
- }
- #endif
- newparam->prev = newparam->next = NULL;
- pthread_mutex_lock(&srv.counter_mutex);
- if(!srv.child){
- srv.child = newparam;
- }
- else {
- newparam->next = srv.child;
- srv.child = srv.child->prev = newparam;
- }
- #ifdef _WIN32
- #ifndef _WINCE
- h = (HANDLE)_beginthreadex((LPSECURITY_ATTRIBUTES )NULL, (unsigned)(16384 + srv.stacksize), (void *)threadfunc, (void *) newparam, 0, &thread);
- #else
- h = (HANDLE)CreateThread((LPSECURITY_ATTRIBUTES )NULL, (unsigned)(16384 + srv.stacksize), (void *)threadfunc, (void *) newparam, 0, &thread);
- #endif
- srv.childcount++;
- if (h) {
- newparam->threadid = (unsigned)thread;
- CloseHandle(h);
- }
- else {
- sprintf((char *)buf, "_beginthreadex(): %s", _strerror(NULL));
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- freeparam(newparam);
- }
- #else
- error = pthread_create(&thread, &pa, threadfunc, (void *)newparam);
- srv.childcount++;
- if(error){
- sprintf((char *)buf, "pthread_create(): %s", strerror(error));
- if(!srv.silent)(*srv.logfunc)(&defparam, buf);
- freeparam(newparam);
- }
- else {
- newparam->threadid = (unsigned)thread;
- }
- #endif
- pthread_mutex_unlock(&srv.counter_mutex);
- memset(&defparam.sincl, 0, sizeof(defparam.sincl));
- memset(&defparam.sincr, 0, sizeof(defparam.sincr));
- if(isudp) while(!srv.fds.events)usleep(SLEEPTIME);
- }
- if(!srv.silent) srv.logfunc(&defparam, (unsigned char *)"Exiting thread");
- if(fp) fclose(fp);
- srvfree(&srv);
- #ifndef STDMAIN
- pthread_mutex_lock(&config_mutex);
- if(srv.next)srv.next->prev = srv.prev;
- if(srv.prev)srv.prev->next = srv.next;
- else conf.services = srv.next;
- pthread_mutex_unlock(&config_mutex);
- #endif
- if(defparam.hostname)myfree(defparam.hostname);
- if(cbc_string)myfree(cbc_string);
- if(cbl_string)myfree(cbl_string);
- return 0;
- }
- void srvinit(struct srvparam * srv, struct clientparam *param){
- memset(srv, 0, sizeof(struct srvparam));
- srv->version = conf.version + 1;
- srv->paused = conf.paused;
- srv->logfunc = conf.logfunc;
- srv->noforce = conf.noforce;
- if(srv->logformat)myfree(srv->logformat);
- srv->logformat = conf.logformat? (unsigned char *)mystrdup((char *)conf.logformat) : NULL;
- srv->authfunc = conf.authfunc;
- srv->usentlm = 0;
- srv->maxchild = conf.maxchild;
- srv->stacksize = conf.stacksize;
- srv->time_start = time(NULL);
- if(conf.logtarget){
- if(srv->logtarget) myfree(srv->logtarget);
- srv->logtarget = (unsigned char *)mystrdup((char *)conf.logtarget);
- }
- srv->srvsock = INVALID_SOCKET;
- srv->logdumpsrv = conf.logdumpsrv;
- srv->logdumpcli = conf.logdumpcli;
- srv->cbsock = INVALID_SOCKET;
- srv->needuser = 1;
- memset(param, 0, sizeof(struct clientparam));
- param->srv = srv;
- param->version = srv->version;
- param->paused = srv->paused;
- param->remsock = param->clisock = param->ctrlsock = param->ctrlsocksrv = INVALID_SOCKET;
- *SAFAMILY(¶m->req) = *SAFAMILY(¶m->sinsl) = *SAFAMILY(¶m->sinsr) = *SAFAMILY(¶m->sincr) = *SAFAMILY(¶m->sincl) = AF_INET;
- pthread_mutex_init(&srv->counter_mutex, NULL);
- srv->intsa = conf.intsa;
- srv->extsa = conf.extsa;
- #ifndef NOIPV6
- srv->extsa6 = conf.extsa6;
- #endif
- }
- void srvinit2(struct srvparam * srv, struct clientparam *param){
- if(srv->logformat){
- char *s;
- if(*srv->logformat == '-' && (s = strchr((char *)srv->logformat + 1, '+')) && s[1]){
- unsigned char* logformat = srv->logformat;
- *s = 0;
- srv->nonprintable = (unsigned char *)mystrdup((char *)srv->logformat + 1);
- srv->replace = s[1];
- srv->logformat = (unsigned char *)mystrdup(s + 2);
- *s = '+';
- myfree(logformat);
- }
- }
- memset(¶m->sinsl, 0, sizeof(param->sinsl));
- memset(¶m->sinsr, 0, sizeof(param->sinsr));
- memset(¶m->req, 0, sizeof(param->req));
- *SAFAMILY(¶m->sinsl) = AF_INET;
- *SAFAMILY(¶m->sinsr) = AF_INET;
- *SAFAMILY(¶m->req) = AF_INET;
- param->sincr = param->sincl = srv->intsa;
- #ifndef NOIPV6
- if (srv->family == 6 || srv->family == 64) param->sinsr = srv->extsa6;
- else
- #endif
- param->sinsr = srv->extsa;
- }
- void srvfree(struct srvparam * srv){
- if(srv->srvsock != INVALID_SOCKET) so._closesocket(srv->srvsock);
- srv->srvsock = INVALID_SOCKET;
- if(srv->cbsock != INVALID_SOCKET) so._closesocket(srv->cbsock);
- srv->cbsock = INVALID_SOCKET;
- srv->service = S_ZOMBIE;
- while(srv->child) usleep(SLEEPTIME * 100);
- #ifndef STDMAIN
- if(srv->filter){
- while(srv->nfilters){
- srv->nfilters--;
- if(srv->filter[srv->nfilters].filter_close){
- (*srv->filter[srv->nfilters].filter_close)(srv->filter[srv->nfilters].data);
- }
- }
- myfree(srv->filter);
- }
- if(srv->acl)freeacl(srv->acl);
- if(srv->authfuncs)freeauth(srv->authfuncs);
- #endif
- pthread_mutex_destroy(&srv->counter_mutex);
- if(srv->target) myfree(srv->target);
- if(srv->logtarget) myfree(srv->logtarget);
- if(srv->logformat) myfree(srv->logformat);
- if(srv->nonprintable) myfree(srv->nonprintable);
- }
- void freeparam(struct clientparam * param) {
- if(param->res == 2) return;
- if(param->datfilterssrv) myfree(param->datfilterssrv);
- #ifndef STDMAIN
- if(param->reqfilters) myfree(param->reqfilters);
- if(param->hdrfilterscli) myfree(param->hdrfilterscli);
- if(param->hdrfilterssrv) myfree(param->hdrfilterssrv);
- if(param->predatfilters) myfree(param->predatfilters);
- if(param->datfilterscli) myfree(param->datfilterscli);
- if(param->filters){
- if(param->nfilters)while(param->nfilters--){
- if(param->filters[param->nfilters].filter->filter_clear)
- (*param->filters[param->nfilters].filter->filter_clear)(param->filters[param->nfilters].data);
- }
- myfree(param->filters);
- }
- #endif
- if(param->clibuf) myfree(param->clibuf);
- if(param->srvbuf) myfree(param->srvbuf);
- if(param->srv){
- pthread_mutex_lock(¶m->srv->counter_mutex);
- if(param->prev){
- param->prev->next = param->next;
- }
- else
- param->srv->child = param->next;
- if(param->next){
- param->next->prev = param->prev;
- }
- (param->srv->childcount)--;
- pthread_mutex_unlock(¶m->srv->counter_mutex);
- }
- if(param->hostname) myfree(param->hostname);
- if(param->username) myfree(param->username);
- if(param->password) myfree(param->password);
- if(param->extusername) myfree(param->extusername);
- if(param->extpassword) myfree(param->extpassword);
- if(param->ctrlsocksrv != INVALID_SOCKET && param->ctrlsocksrv != param->remsock) {
- so._shutdown(param->ctrlsocksrv, SHUT_RDWR);
- so._closesocket(param->ctrlsocksrv);
- }
- if(param->ctrlsock != INVALID_SOCKET && param->ctrlsock != param->clisock) {
- so._shutdown(param->ctrlsock, SHUT_RDWR);
- so._closesocket(param->ctrlsock);
- }
- if(param->remsock != INVALID_SOCKET) {
- so._shutdown(param->remsock, SHUT_RDWR);
- so._closesocket(param->remsock);
- }
- if(param->clisock != INVALID_SOCKET) {
- so._shutdown(param->clisock, SHUT_RDWR);
- so._closesocket(param->clisock);
- }
- myfree(param);
- }
- #ifndef STDMAIN
- static void * itcopy (void * from, size_t size){
- void * ret;
- if(!from) return NULL;
- ret = myalloc(size);
- if(ret) memcpy(ret, from, size);
- return ret;
- }
- struct auth * copyauth (struct auth * authfuncs){
- struct auth * newauth = NULL;
- newauth = authfuncs = itcopy(authfuncs, sizeof(struct auth));
- for( ; authfuncs; authfuncs = authfuncs->next = itcopy(authfuncs->next, sizeof(struct auth)));
- return newauth;
- }
- struct ace * copyacl (struct ace *ac){
- struct ace * ret = NULL;
- struct iplist *ipl;
- struct portlist *pl;
- struct userlist *ul;
- struct chain *ch;
- struct period *pel;
- struct hostname *hst;
- ret = ac = itcopy(ac, sizeof(struct ace));
- for( ; ac; ac = ac->next = itcopy(ac->next, sizeof(struct ace))){
- ac->src = itcopy(ac->src, sizeof(struct iplist));
- for(ipl = ac->src; ipl; ipl = ipl->next = itcopy(ipl->next, sizeof(struct iplist)));
- ac->dst = itcopy(ac->dst, sizeof(struct iplist));
- for(ipl = ac->dst; ipl; ipl = ipl->next = itcopy(ipl->next, sizeof(struct iplist)));
- ac->ports = itcopy(ac->ports, sizeof(struct portlist));
- for(pl = ac->ports; pl; pl = pl->next = itcopy(pl->next, sizeof(struct portlist)));
- ac->periods = itcopy(ac->periods, sizeof(struct period));
- for(pel = ac->periods; pel; pel = pel->next = itcopy(pel->next, sizeof(struct period)));
- ac->users = itcopy(ac->users, sizeof(struct userlist));
- for(ul = ac->users; ul; ul = ul->next = itcopy(ul->next, sizeof(struct userlist))){
- if(ul->user) ul->user = (unsigned char*)mystrdup((char *)ul->user);
- }
- ac->dstnames = itcopy(ac->dstnames, sizeof(struct hostname));
- for(hst = ac->dstnames; hst; hst = hst->next = itcopy(hst->next, sizeof(struct hostname))){
- if(hst->name) hst->name = (unsigned char*)mystrdup((char *)hst->name);
- }
- ac->chains = itcopy(ac->chains, sizeof(struct chain));
- for(ch = ac->chains; ch; ch = ch->next = itcopy(ch->next, sizeof(struct chain))){
- if(ch->extuser)ch->extuser = (unsigned char*)mystrdup((char *)ch->extuser);
- if(ch->extpass)ch->extpass = (unsigned char*)mystrdup((char *)ch->extpass);
- if(ch->exthost)ch->exthost = (unsigned char*)mystrdup((char *)ch->exthost);
- }
- }
- return ret;
- }
- void copyfilter (struct filter *filter, struct srvparam *srv){
- int nfilters = 0;
- if(!filter) return;
- for(srv->filter = filter; srv->filter; srv->filter = srv->filter->next) nfilters++;
- srv->filter = myalloc(sizeof(struct filter) * nfilters);
- if(!srv->filter) return;
- for(; filter; filter = filter->next){
- void *data = NULL;
- if(!filter->filter_open || !(data = (*filter->filter_open)(filter->data, srv))) continue;
- srv->filter[srv->nfilters] = *filter;
- srv->filter[srv->nfilters].data = data;
- if(srv->nfilters>0)srv->filter[srv->nfilters - 1].next = srv->filter + srv->nfilters;
- srv->nfilters++;
- if(filter->filter_request)srv->nreqfilters++;
- if(filter->filter_header_srv)srv->nhdrfilterssrv++;
- if(filter->filter_header_cli)srv->nhdrfilterscli++;
- if(filter->filter_predata)srv->npredatfilters++;
- if(filter->filter_data_srv)srv->ndatfilterssrv++;
- if(filter->filter_data_cli)srv->ndatfilterscli++;
- }
- }
- FILTER_ACTION makefilters (struct srvparam *srv, struct clientparam *param){
- FILTER_ACTION res=PASS;
- FILTER_ACTION action;
- int i;
- if(!srv->nfilters) return PASS;
- if(!(param->filters = myalloc(sizeof(struct filterp) * srv->nfilters)) ||
- (srv->nreqfilters && !(param->reqfilters = myalloc(sizeof(struct filterp *) * srv->nreqfilters))) ||
- (srv->nhdrfilterssrv && !(param->hdrfilterssrv = myalloc(sizeof(struct filterp *) * srv->nhdrfilterssrv))) ||
- (srv->nhdrfilterscli && !(param->hdrfilterscli = myalloc(sizeof(struct filterp *) * srv->nhdrfilterscli))) ||
- (srv->npredatfilters && !(param->predatfilters = myalloc(sizeof(struct filterp *) * srv->npredatfilters))) ||
- (srv->ndatfilterssrv && !(param->datfilterssrv = myalloc(sizeof(struct filterp *) * srv->ndatfilterssrv))) ||
- (srv->ndatfilterscli && !(param->datfilterscli = myalloc(sizeof(struct filterp *) * srv->ndatfilterscli)))
- ){
- param->res = 21;
- return REJECT;
- }
-
- for(i=0; i<srv->nfilters; i++){
- if(!srv->filter[i].filter_client)continue;
- action = (*srv->filter[i].filter_client)(srv->filter[i].data, param, ¶m->filters[param->nfilters].data);
- if(action == PASS) continue;
- if(action > CONTINUE) return action;
- param->filters[param->nfilters].filter = srv->filter + i;
- if(srv->filter[i].filter_request)param->reqfilters[param->nreqfilters++] = param->filters + param->nfilters;
- if(srv->filter[i].filter_header_cli)param->hdrfilterscli[param->nhdrfilterscli++] = param->filters + param->nfilters;
- if(srv->filter[i].filter_header_srv)param->hdrfilterssrv[param->nhdrfilterssrv++] = param->filters + param->nfilters;
- if(srv->filter[i].filter_predata)param->predatfilters[param->npredatfilters++] = param->filters + param->nfilters;
- if(srv->filter[i].filter_data_cli)param->datfilterscli[param->ndatfilterscli++] = param->filters + param->nfilters;
- if(srv->filter[i].filter_data_srv)param->datfilterssrv[param->ndatfilterssrv++] = param->filters + param->nfilters;
- param->nfilters++;
- }
- return res;
- }
- void * itfree(void *data, void * retval){
- myfree(data);
- return retval;
- }
- void freeauth(struct auth * authfuncs){
- for(; authfuncs; authfuncs = (struct auth *)itfree(authfuncs, authfuncs->next));
- }
- void freeacl(struct ace *ac){
- struct iplist *ipl;
- struct portlist *pl;
- struct userlist *ul;
- struct chain *ch;
- struct period *pel;
- struct hostname *hst;
- for(; ac; ac = (struct ace *) itfree(ac, ac->next)){
- for(ipl = ac->src; ipl; ipl = (struct iplist *)itfree(ipl, ipl->next));
- for(ipl = ac->dst; ipl; ipl = (struct iplist *)itfree(ipl,ipl->next));
- for(pl = ac->ports; pl; pl = (struct portlist *)itfree(pl, pl->next));
- for(pel = ac->periods; pel; pel = (struct period *)itfree(pel, pel->next));
- for(ul = ac->users; ul; ul = (struct userlist *)itfree(ul, ul->next)){
- if(ul->user)myfree(ul->user);
- }
- for(hst = ac->dstnames; hst; hst = (struct hostname *)itfree(hst, hst->next)){
- if(hst->name)myfree(hst->name);
- }
- for(ch = ac->chains; ch; ch = (struct chain *) itfree(ch, ch->next)){
- if(ch->extuser) myfree(ch->extuser);
- if(ch->extpass) myfree(ch->extpass);
- if(ch->exthost) myfree(ch->exthost);
- }
- }
- }
- FILTER_ACTION handlereqfilters(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
- FILTER_ACTION action;
- int i;
- for(i=0; i<param->nreqfilters; i++){
- action = (*param->reqfilters[i]->filter->filter_request)(param->reqfilters[i]->data, param, buf_p, bufsize_p, offset, length_p);
- if(action!=CONTINUE) return action;
- }
- return PASS;
- }
- FILTER_ACTION handlehdrfilterssrv(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
- FILTER_ACTION action;
- int i;
- for(i=0; i<param->nhdrfilterssrv; i++){
- action = (*param->hdrfilterssrv[i]->filter->filter_header_srv)(param->hdrfilterssrv[i]->data, param, buf_p, bufsize_p, offset, length_p);
- if(action!=CONTINUE) return action;
- }
- return PASS;
- }
- FILTER_ACTION handlehdrfilterscli(struct clientparam *param, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
- FILTER_ACTION action;
- int i;
- for(i = 0; i < param->nhdrfilterscli; i++){
- action = (*param->hdrfilterscli[i]->filter->filter_header_cli)(param->hdrfilterscli[i]->data, param, buf_p, bufsize_p, offset, length_p);
- if(action!=CONTINUE) return action;
- }
- return PASS;
- }
- #endif
- FILTER_ACTION handlepredatflt(struct clientparam *cparam){
- #ifndef STDMAIN
- FILTER_ACTION action;
- int i;
- for(i=0; i<cparam->npredatfilters ;i++){
- action = (*cparam->predatfilters[i]->filter->filter_predata)(cparam->predatfilters[i]->data, cparam);
- if(action!=CONTINUE) return action;
- }
- #endif
- return PASS;
- }
- FILTER_ACTION handledatfltcli(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
- #ifndef STDMAIN
- FILTER_ACTION action;
- int i;
- for(i=0; i<cparam->ndatfilterscli ;i++){
- action = (*cparam->datfilterscli[i]->filter->filter_data_cli)(cparam->datfilterscli[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
- if(action!=CONTINUE) return action;
- }
- #endif
- return PASS;
- }
- FILTER_ACTION handledatfltsrv(struct clientparam *cparam, unsigned char ** buf_p, int * bufsize_p, int offset, int * length_p){
- FILTER_ACTION action;
- int i;
- for(i=0; i<cparam->ndatfilterssrv; i++){
- action = (*cparam->datfilterssrv[i]->filter->filter_data_srv)(cparam->datfilterssrv[i]->data, cparam, buf_p, bufsize_p, offset, length_p);
- if(action!=CONTINUE) return action;
- }
- return PASS;
- }
|