mirrors/byedpi
1
0
Fork 0
mirror of https://github.com/hufrea/byedpi.git synced 2024-12-22 14:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix POLLOUT handling

Browse Source
This commit is contained in:
ruti 2024-01-23 03:08:35 +03:00
parent 86d725ed66
commit 676227271d
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
proxy.c
View File

@ -451,8 +451,10 @@ static inline int on_connect(struct poolhd *pool, struct eval *val,
if (e) { if (e) {
return -1; return -1;
} }
if (mod_etype(pool, val, POLLOUT, 0)) {
return -1;
}
val->type = EV_TUNNEL; val->type = EV_TUNNEL;
mod_etype(pool, val, POLLOUT, 0);
val->pair->type = EV_CONNECT; val->pair->type = EV_CONNECT;
} }
else { else {
@ -495,8 +497,10 @@ static inline int on_tunnel(struct poolhd *pool, struct eval *val,
free(val->tmpbuf); free(val->tmpbuf);
val->tmpbuf = 0; val->tmpbuf = 0;
mod_etype(pool, val, POLLOUT, 0); if (mod_etype(pool, val, POLLIN, 1) ||
mod_etype(pool, val->pair, POLLIN, 1); mod_etype(pool, pair, POLLOUT, 0)) {
return -1;
}
} }
do { do {
n = recv(val->fd, buffer, bfsize, 0); n = recv(val->fd, buffer, bfsize, 0);
@ -524,8 +528,10 @@ static inline int on_tunnel(struct poolhd *pool, struct eval *val,
} }
memcpy(val->tmpbuf, buffer + sn, val->size); memcpy(val->tmpbuf, buffer + sn, val->size);
mod_etype(pool, val, POLLIN, 0); if (mod_etype(pool, val, POLLIN, 0) ||
mod_etype(pool, pair, POLLOUT, 1); mod_etype(pool, pair, POLLOUT, 1)) {
return -1;
}
break; break;
} }
} while (n == bfsize); } while (n == bfsize);