bughz/3proxy
1
0
Fork 0
mirror of https://github.com/3proxy/3proxy.git synced 2025-02-23 10:35:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Call poll() only on EAGAIN/EINTR

Browse Source
This commit is contained in:
z3APA3A 2020-11-16 14:11:18 +03:00
parent 4ec9229acb
commit 2b1025a79b
3 changed files with 105 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/3proxy.c
View File

@ -265,9 +265,7 @@ void cyclestep(void){
if(conf.timetoexit){
conf.paused++;
doschedule();
usleep(SLEEPTIME*999);
usleep(SLEEPTIME*999);
usleep(SLEEPTIME*999);
usleep(3*SLEEPTIME*999);
return;
}

38
src/sockgetchar.c
View File

@ -10,19 +10,20 @@
int socksend(SOCKET sock, char * buf, int bufsize, int to){
int sent = 0;
int res;
struct pollfd fds;
struct pollfd fds={0};
fds.fd = sock;
fds.events = POLLOUT;
do {
if(conf.timetoexit) return 0;
res = so._poll(&fds, 1, to*1000);
if(res < 0 && (errno == EAGAIN || errno == EINTR)) continue;
if(res < 1) break;
res = so._send(sock, (char *)buf + sent, bufsize - sent, 0);
if(res < 0) {
if(errno == EAGAIN || errno == EINTR) continue;
break;
fds.events = POLLOUT;
if(conf.timetoexit) return sent;
res = so._poll(&fds, 1, to*1000);
if(res < 0 && (errno == EAGAIN || errno == EINTR)) continue;
if(res < 1) break;
res = 0;
}
sent += res;
} while (sent < bufsize);
@ -33,19 +34,20 @@ int socksend(SOCKET sock, char * buf, int bufsize, int to){
int socksendto(SOCKET sock, struct sockaddr * sin, char * buf, int bufsize, int to){
int sent = 0;
int res;
struct pollfd fds;
struct pollfd fds={0};
fds.fd = sock;
do {
if(conf.timetoexit) return 0;
fds.events = POLLOUT;
res = so._poll(&fds, 1, to);
if(res < 0 && (errno == EAGAIN || errno == EINTR)) continue;
if(res < 1) break;
res = so._sendto(sock, (char *)buf + sent, bufsize - sent, 0, sin, SASIZE(sin));
if(res < 0) {
if(errno != EAGAIN && errno != EINTR) break;
continue;
fds.events = POLLOUT;
if(conf.timetoexit) return sent;
res = so._poll(&fds, 1, to);
if(res < 0 && (errno == EAGAIN || errno == EINTR)) continue;
if(res < 1) break;
res = 0;
}
sent += res;
} while (sent < bufsize);
@ -53,18 +55,18 @@ int socksendto(SOCKET sock, struct sockaddr * sin, char * buf, int bufsize, int
}
int sockrecvfrom(SOCKET sock, struct sockaddr * sin, char * buf, int bufsize, int to){
struct pollfd fds;
struct pollfd fds={0};
SASIZETYPE sasize;
int res;
fds.fd = sock;
fds.events = POLLIN;
if(conf.timetoexit) return EOF;
if (so._poll(&fds, 1, to)<1) return 0;
sasize = SASIZE(sin);
do {
sasize = SASIZE(sin);
res = so._recvfrom(sock, (char *)buf, bufsize, 0, (struct sockaddr *)sin, &sasize);
} while (res < 0 && (errno == EAGAIN || errno == EINTR));
if ((res >= 0) || (errno != EAGAIN && errno != EINTR) || conf.timetoexit) break;
fds.events = POLLIN;
res = so._poll(&fds, 1, to);
} while (res == 1 || (res < 0 && (errno == EAGAIN || errno == EINTR)));
return res;
}

53
src/socks.c
View File

@ -7,7 +7,6 @@
*/
#include "proxy.h"
#define RETURN(xxx) { param->res = xxx; goto CLEANRET; }
char * commands[] = {"UNKNOWN", "CONNECT", "BIND", "UDPMAP"};
@ -333,22 +332,19 @@ fflush(stderr);
if(!(buf = myalloc(LARGEBUFSIZE))) {RETURN(21);}
sin = param->sincr;
for(;;){
fds[2].events = fds[1].events = fds[0].events = 0;
fds[0].fd = param->remsock;
fds[1].fd = param->clisock;
fds[2].fd = param->ctrlsock;
fds[2].events = fds[1].events = fds[0].events = POLLIN;
for(;;){
if(!fds[1].events || fds[1].revents){
sasize = sizeof(sin);
len = so._recvfrom(param->clisock, (char *)buf, 65535, 0, (struct sockaddr *)&sin, &sasize);
if(len >= 0) {
fds[1].events = fds[1].revents = 0;
if(len < 10) continue;
res = so._poll(fds, 3, conf.timeouts[CONNECTION_L]*1000);
if(res <= 0) {
param->res = 463;
break;
}
if (fds[2].revents) {
param->res = 0;
break;
}
if (fds[1].revents) {
sasize = sizeof(sin);
if((len = so._recvfrom(param->clisock, (char *)buf, 65535, 0, (struct sockaddr *)&sin, &sasize)) <= 10) {
param->res = 464;
@ -405,16 +401,25 @@ fprintf(stderr, "client address is assumed to be %s:%hu\n",
fflush(stderr);
#endif
}
continue;
}
if (fds[0].revents) {
else if(len ==0 || (len <0 && errno != EINTR && errno != EAGAIN)){
param->res = 464;
break;
}
fds[1].events = POLLIN;
}
if (!fds[0].events || fds[0].revents) {
sasize = sizeof(param->sinsr);
buf[0]=buf[1]=buf[2]=0;
buf[3]=(*SAFAMILY(&param->sinsl) == AF_INET)?1:4;
if((len = so._recvfrom(param->remsock, (char *)buf+6+SAADDRLEN(&param->sinsl), 65535 - (6+SAADDRLEN(&param->sinsl)), 0, (struct sockaddr *)&param->sinsr, &sasize)) <= 0) {
len = so._recvfrom(param->remsock, (char *)buf+6+SAADDRLEN(&param->sinsl), 65535 - (6+SAADDRLEN(&param->sinsl)), 0, (struct sockaddr *)&param->sinsr, &sasize);
if(len == 0 || (len < 0 && errno != EAGAIN && errno !=EINTR)) {
param->res = 468;
break;
}
else if(len > 0){
fds[0].events = fds[0].revents = 0;
param->statssrv64+=len;
param->nreads++;
memcpy(buf+4, SAADDR(&param->sinsr), SAADDRLEN(&param->sinsr));
@ -429,6 +434,22 @@ fprintf(stderr, "UDP packet relayed to client from %hu size %d\n",
fflush(stderr);
#endif
continue;
}
fds[0].events = POLLIN;
}
if (fds[2].revents) {
param->res = 0;
break;
}
fds[0].revents = fds[1].revents = fds[2].revents = 0;
fds[2].events = POLLIN;
res = so._poll(fds, 3, conf.timeouts[CONNECTION_L]*1000);
if(res == 0 || (res < 0 && errno != EAGAIN && errno != EINTR)) {
param->res = 463;
break;
}
}
break;