From c73397fd755ff7fc70a9e790e99e024eb9c2f632 Mon Sep 17 00:00:00 2001 From: Elias Fleckenstein Date: Tue, 26 Apr 2022 11:29:57 +0200 Subject: [PATCH] Check for partly received data --- dragonnet/recv.c | 2 +- dragonnet/recv_thread.c | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dragonnet/recv.c b/dragonnet/recv.c index ffad396..089aa35 100644 --- a/dragonnet/recv.c +++ b/dragonnet/recv.c @@ -16,5 +16,5 @@ bool dragonnet_recv_raw(DragonnetPeer *p, void *buf, size_t n) abort(); } - return len != 0; + return len == (ssize_t) n; } diff --git a/dragonnet/recv_thread.c b/dragonnet/recv_thread.c index 48f02ef..348c336 100644 --- a/dragonnet/recv_thread.c +++ b/dragonnet/recv_thread.c @@ -37,7 +37,7 @@ void *dragonnet_peer_recv_thread(void *g_peer) } // Connection closed - if (len == 0 || reset) { + if (len != sizeof type_id || reset) { if (p->on_disconnect) p->on_disconnect(p); -- 2.44.0