From: Elias Fleckenstein <eliasfleckenstein@web.de>
Date: Tue, 26 Apr 2022 09:29:57 +0000 (+0200)
Subject: Check for partly received data
X-Git-Url: https://git.lizzy.rs/?a=commitdiff_plain;h=c73397fd755ff7fc70a9e790e99e024eb9c2f632;p=dragonnet.git

Check for partly received data
---

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);