Properly handle socket_recv errors

1 files changed, 3 insertions, 2 deletions

diff --git a/ot_udp.c b/ot_udp.c
index 8e056db..02467e1 100644
--- a/ot_udp.c
+++ b/ot_udp.c
@@ -81,9 +81,10 @@ int handle_udp6(int64_t serversocket, struct ot_workstruct *ws) {
   uint16_t  port, remoteport;
   size_t    byte_count, scrape_count;
 
-  byte_count = socket_recv6(serversocket, ws->inbuf, G_INBUF_SIZE, remoteip, &remoteport, &scopeid);
-  if (!byte_count)
+  ssize_t   result = socket_recv6(serversocket, ws->inbuf, G_INBUF_SIZE, remoteip, &remoteport, &scopeid);
+  if (result <= 0)
     return 0;
+  byte_count = (size_t)result;
 
   stats_issue_event(EVENT_ACCEPT, FLAG_UDP, (uintptr_t)remoteip);
   stats_issue_event(EVENT_READ, FLAG_UDP, byte_count);