dany/antd-tunnel-plugin
1
0
Fork 0
mirror of https://github.com/lxsang/antd-tunnel-plugin synced 2024-12-26 01:48:21 +01:00
Code Issues Projects Releases Wiki Activity

use poll instead of select on client sockets whose fid may be greater than 1024

Browse Source
This commit is contained in:
DanyLE 2022-08-30 17:48:15 +02:00
parent 4f8f020cd8
commit 1cb8bfa32b
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tunnel.c
View File

@ -19,6 +19,7 @@
#include <fcntl.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <poll.h>
#define MAX_CHANNEL_NAME 64
#define HOT_LINE_SOCKET "antd_hotline.sock"
@ -1037,7 +1038,7 @@ void *handle(void *rq_data)
uint8_t *buffer;
struct timeval timeout;
int status;
fd_set fd_in;
struct pollfd pfd;
int offset;
bst_node_t *node = NULL;
antd_tunnel_key_t *key_p = NULL;
@ -1099,13 +1100,13 @@ void *handle(void *rq_data)
// session is valid, continue
timeout.tv_sec = 0;
timeout.tv_usec = PROCESS_TIMEOUT;
FD_ZERO(&fd_in);
FD_SET(client->sock, &fd_in);
status = select(client->sock + 1, &fd_in, NULL, NULL, &timeout);
pfd.fd = client->sock;
pfd.events = POLLIN;
status = poll(&pfd, 1, PROCESS_TIMEOUT);
switch (status)
{
case -1:
LOG("Error %d on select()\n", errno);
ERROR("Error on poll(): %s", strerror(errno));
pthread_mutex_lock(&g_tunnel.lock);
bst_for_each(g_tunnel.channels, unsubscribe_notify, argv, 1);
pthread_mutex_unlock(&g_tunnel.lock);
@ -1117,6 +1118,15 @@ void *handle(void *rq_data)
select(0, NULL, NULL, NULL, &timeout);
break;
default:
if(pfd.revents & (POLLERR | POLLHUP))
{
ERROR("POLLHUP or POLLERR found");
pthread_mutex_lock(&g_tunnel.lock);
bst_for_each(g_tunnel.channels, unsubscribe_notify, argv, 1);
pthread_mutex_unlock(&g_tunnel.lock);
return task;
break;
}
pthread_mutex_lock(&g_tunnel.lock);
h = ws_read_header(rq->client);
pthread_mutex_unlock(&g_tunnel.lock);