external/uhub
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Ensure we set a max recv and send buffer per user, and that it does not

Browse Source 
exceed the one used internally by the application.
This commit is contained in:
Jan Vidar Krey 2009-06-22 19:50:10 +02:00
parent 57fd872f14
commit 1d9acece34
2 changed files with 74 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/netevent.c
View File

@ -160,8 +160,6 @@ void net_on_read(int fd, short ev, void *arg)
buf_size = hub_recvq_get(q, buf, MAX_RECV_BUF); buf_size = hub_recvq_get(q, buf, MAX_RECV_BUF);
for (;;)
{
int size = net_user_recv(user, &buf[buf_size], MAX_RECV_BUF - buf_size); int size = net_user_recv(user, &buf[buf_size], MAX_RECV_BUF - buf_size);
if (size > 0) if (size > 0)
{ {
@ -172,12 +170,10 @@ void net_on_read(int fd, short ev, void *arg)
{ {
if (net_error() != EWOULDBLOCK) if (net_error() != EWOULDBLOCK)
flag_close = quit_socket_error; flag_close = quit_socket_error;
break;
} }
else if (size == 0) else if (size == 0)
{ {
flag_close = quit_disconnected; flag_close = quit_disconnected;
break;
} }
else else
{ {
@ -214,8 +210,6 @@ void net_on_read(int fd, short ev, void *arg)
{ {
hub_recvq_set(q, 0, 0); hub_recvq_set(q, 0, 0);
} }
}
} }
if (flag_close) if (flag_close)
@ -257,19 +251,41 @@ void net_on_write(int fd, short ev, void *arg)
break; break;
} }
#if 0
if (close_flag)
{
hub_disconnect_user(g_hub, user, close_flag);
}
else
#endif
if (hub_sendq_get_bytes(user->net.send_queue)) if (hub_sendq_get_bytes(user->net.send_queue))
{ {
user_net_io_want_write(user); user_net_io_want_write(user);
} }
} }
static void prepare_user_net(struct hub_info* hub, struct user* user)
{
int fd = user->net.sd;
struct timeval timeout = { TIMEOUT_CONNECTED, 0 };
size_t sendbuf = 0;
size_t recvbuf = 0;
net_set_nonblocking(fd, 1);
net_set_nosigpipe(fd, 1);
if (net_get_recvbuf_size(fd, &recvbuf) != -1)
{
if (recvbuf > MAX_RECV_BUF || !recvbuf) recvbuf = MAX_RECV_BUF;
net_set_recvbuf_size(fd, recvbuf);
}
if (net_get_sendbuf_size(fd, &sendbuf) != -1)
{
if (sendbuf > MAX_SEND_BUF || !sendbuf) sendbuf = MAX_SEND_BUF;
net_set_sendbuf_size(fd, sendbuf);
}
event_set(user->net.ev_read, fd, EV_READ | EV_PERSIST, net_on_read, user);
event_set(user->net.ev_write, fd, EV_WRITE, net_on_write, user);
event_base_set(hub->evbase, user->net.ev_read);
event_base_set(hub->evbase, user->net.ev_write);
event_add(user->net.ev_read, &timeout);
}
void net_on_accept(int server_fd, short ev, void *arg) void net_on_accept(int server_fd, short ev, void *arg)
{ {
@ -277,7 +293,6 @@ void net_on_accept(int server_fd, short ev, void *arg)
struct user* user = 0; struct user* user = 0;
struct ip_addr_encap ipaddr; struct ip_addr_encap ipaddr;
const char* addr; const char* addr;
struct timeval timeout = { TIMEOUT_CONNECTED, 0 };
for (;;) for (;;)
{ {
@ -319,14 +334,7 @@ void net_on_accept(int server_fd, short ev, void *arg)
/* Store IP address in user object */ /* Store IP address in user object */
memcpy(&user->net.ipaddr, &ipaddr, sizeof(ipaddr)); memcpy(&user->net.ipaddr, &ipaddr, sizeof(ipaddr));
net_set_nonblocking(fd, 1); prepare_user_net(hub, user);
net_set_nosigpipe(fd, 1);
event_set(user->net.ev_read, fd, EV_READ | EV_PERSIST, net_on_read, user);
event_set(user->net.ev_write, fd, EV_WRITE, net_on_write, user);
event_base_set(hub->evbase, user->net.ev_read);
event_base_set(hub->evbase, user->net.ev_write);
event_add(user->net.ev_read, &timeout);
} }
} }