RPJosh
/
neatvnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Stop reading from a client as soon as an error occurs on the socket

tight-png
Andri Yngvason 2019-12-27 13:57:24 +00:00
parent 95f5059200
commit 875cd5af7b
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/server.c
View File

@ -718,19 +718,21 @@ static void on_client_read(uv_stream_t* stream, ssize_t n_read,
struct nvnc_client* client = container_of( struct nvnc_client* client = container_of(
(uv_tcp_t*)stream, struct nvnc_client, stream_handle); (uv_tcp_t*)stream, struct nvnc_client, stream_handle);
if (n_read == UV_EOF) { if (n_read == 0)
return;
if (n_read < 0) {
uv_read_stop(stream);
client_unref(client); client_unref(client);
return; return;
} }
if (n_read < 0)
return;
assert(client->buffer_index == 0); assert(client->buffer_index == 0);
if (n_read > MSG_BUFFER_SIZE - client->buffer_len) { if ((size_t)n_read > MSG_BUFFER_SIZE - client->buffer_len) {
/* Can't handle this. Let's just give up */ /* Can't handle this. Let's just give up */
client->state = VNC_CLIENT_STATE_ERROR; client->state = VNC_CLIENT_STATE_ERROR;
uv_read_stop(stream);
client_unref(client); client_unref(client);
return; return;
} }