bughz/3proxy
1
0
Fork 0
mirror of https://github.com/3proxy/3proxy.git synced 2025-11-04 07:42:39 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

make spliced sockmap more reliable

Browse Source
This commit is contained in:
z3APA3A 2019-01-11 23:10:39 +03:00
parent 953959efab
commit 603ab8187f
1 changed files with 19 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/sockmap.c
View File

@ -41,6 +41,7 @@ int splicemap(struct clientparam * param, int timeo){
SASIZETYPE sasize; SASIZETYPE sasize;
int res = 0, stop = 0; int res = 0, stop = 0;
int srvstate = 0, clistate = 0; int srvstate = 0, clistate = 0;
int srvsockstate = 0, clisockstate = 0;
int insrvpipe = 0, inclipipe = 0; int insrvpipe = 0, inclipipe = 0;
int rfromserver = 0, rfromclient = 0; int rfromserver = 0, rfromclient = 0;
int sleeptime = 0; int sleeptime = 0;
@ -105,7 +106,7 @@ int splicemap(struct clientparam * param, int timeo){
fds[0].fd = param->clisock; fds[0].fd = param->clisock;
fds[1].fd = param->remsock; fds[1].fd = param->remsock;
while((!stop || (inclipipe && param->remsock != INVALID_SOCKET) || (insrvpipe && param->clisock != INVALID_SOCKET)) && !conf.timetoexit){ while((!stop || (inclipipe && !srvsockstate) || (insrvpipe && !clisockstate)) && !conf.timetoexit){
param->cycles++; param->cycles++;
#ifdef NOIPV6 #ifdef NOIPV6
@ -116,29 +117,29 @@ int splicemap(struct clientparam * param, int timeo){
fds[0].events = fds[1].events = 0; fds[0].events = fds[1].events = 0;
fds[0].revents = fds[1].revents = 0; fds[0].revents = fds[1].revents = 0;
if(srvstate && !param->waitclient64 && param->clisock != INVALID_SOCKET){ if(srvstate && !clisockstate && !param->waitclient64 && param->clisock != INVALID_SOCKET){
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: will send to client"); (*param->srv->logfunc)(param, "splice: will send to client");
#endif #endif
fds[0].events |= POLLOUT; fds[0].events |= POLLOUT;
} }
rfromserver = MAXSPLICE; rfromserver = MAXSPLICE - insrvpipe;
if(param->waitserver64) rfromserver = MIN(MAXSPLICE, param->waitserver64 - (received + insrvpipe)); if(param->waitserver64) rfromserver = MIN(MAXSPLICE, param->waitserver64 - (received + insrvpipe));
if(srvstate < 2 && rfromserver > 0 && param->remsock != INVALID_SOCKET) { if(srvstate < 2 && !srvsockstate && rfromserver > 0 && param->remsock != INVALID_SOCKET) {
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: will recv from server"); (*param->srv->logfunc)(param, "splice: will recv from server");
#endif #endif
fds[1].events |= (POLLIN|POLLRDHUP); fds[1].events |= (POLLIN|POLLRDHUP);
} }
if(clistate && !param->waitserver64 && param->remsock != INVALID_SOCKET){ if(clistate && !srvsockstate && !param->waitserver64 && param->remsock != INVALID_SOCKET){
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: will send to server"); (*param->srv->logfunc)(param, "splice: will send to server");
#endif #endif
fds[1].events |= POLLOUT; fds[1].events |= POLLOUT;
} }
rfromclient = MAXSPLICE; rfromclient = MAXSPLICE - inclipipe;
if(param->waitclient64) rfromclient = MIN(MAXSPLICE, param->waitclient64 - (sent + inclipipe)); if(param->waitclient64) rfromclient = MIN(MAXSPLICE, param->waitclient64 - (sent + inclipipe));
if(clistate < 2 && rfromclient > 0 && param->clisock != INVALID_SOCKET) { if(clistate < 2 && !clisockstate && rfromclient > 0 && param->clisock != INVALID_SOCKET) {
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: will recv from client"); (*param->srv->logfunc)(param, "splice: will recv from client");
#endif #endif
@ -166,21 +167,14 @@ int splicemap(struct clientparam * param, int timeo){
if(res < 1){ if(res < 1){
RETURN(92); RETURN(92);
} }
if( (fds[0].revents & (POLLERR|POLLNVAL|POLLHUP|POLLRDHUP)) && !(fds[0].revents & POLLIN)) { if( (fds[0].revents & (POLLERR|POLLNVAL)) || (fds[1].revents & (POLLERR|POLLNVAL))) {
fds[0].revents = 0; RETURN(90);
stop = 1; }
param->res = 90; if( (fds[0].revents & (POLLHUP|POLLRDHUP))) {
so._shutdown(param->clisock, SHUT_RDWR); stop = clisockstate = 1;
so._closesocket(param->clisock);
fds[0].fd = param->clisock = INVALID_SOCKET;
} }
if( (fds[1].revents & (POLLERR|POLLNVAL|POLLHUP|POLLRDHUP)) && !(fds[1].revents & POLLIN)){ if( (fds[1].revents & (POLLERR|POLLNVAL|POLLHUP|POLLRDHUP)) && !(fds[1].revents & POLLIN)){
fds[1].revents = 0; stop = srvsockstate = 1;
stop = 1;
param->res = 90;
so._shutdown(param->remsock, SHUT_RDWR);
so._closesocket(param->remsock);
fds[1].fd = param->remsock = INVALID_SOCKET;
} }
if((fds[0].revents & POLLOUT)){ if((fds[0].revents & POLLOUT)){
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
@ -238,7 +232,7 @@ int splicemap(struct clientparam * param, int timeo){
RETURN (99); RETURN (99);
} }
} }
if ((fds[0].revents & POLLIN)) { if ((fds[0].revents & POLLIN) || clisockstate == 1) {
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: recv from client"); (*param->srv->logfunc)(param, "splice: recv from client");
#endif #endif
@ -249,9 +243,7 @@ int splicemap(struct clientparam * param, int timeo){
continue; continue;
} }
if (res==0) { if (res==0) {
so._shutdown(param->clisock, SHUT_RDWR); clisockstate = 2;
so._closesocket(param->clisock);
fds[0].fd = param->clisock = INVALID_SOCKET;
stop = 1; stop = 1;
} }
else { else {
@ -260,7 +252,7 @@ int splicemap(struct clientparam * param, int timeo){
if(insrvpipe >= MAXSPLICE) clistate = 2; if(insrvpipe >= MAXSPLICE) clistate = 2;
} }
} }
if ((fds[1].revents & POLLIN)) { if ((fds[1].revents & POLLIN) || srvsockstate == 1) {
#if DEBUGLEVEL > 2 #if DEBUGLEVEL > 2
(*param->srv->logfunc)(param, "splice: recv from server"); (*param->srv->logfunc)(param, "splice: recv from server");
#endif #endif
@ -271,10 +263,8 @@ int splicemap(struct clientparam * param, int timeo){
continue; continue;
} }
if (res==0) { if (res==0) {
so._shutdown(param->remsock, SHUT_RDWR); srvsockstate = 2;
so._closesocket(param->remsock); stop = 1;
fds[1].fd = param->remsock = INVALID_SOCKET;
stop = 2;
} }
else { else {
insrvpipe += res; insrvpipe += res;