Check for partly received data

diff --git a/dragonnet/recv.c b/dragonnet/recv.c
index ffad3968476bae19e74b132512d7673d5c52f493..089aa3544aa53dfb4a2dda2aa0cc6e52dac4d424 100644 (file)
--- 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 48f02ef1a0854d8e38e97621f80266e2648a4e6e..348c3361e191219f19a7794c35b3f75a2f556daa 100644 (file)
--- 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);