From 821fca673fc3f797980ebc0d9033c3970815f09a Mon Sep 17 00:00:00 2001
From: "B. Blechschmidt" <github@blechschmidt.pw>
Date: Wed, 4 Apr 2018 19:42:22 +0200
Subject: [PATCH] Make sockets non-blocking when making use of busy-wait
 polling

---
 main.c | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/main.c b/main.c
index fa7798c..665bfab 100644
--- a/main.c
+++ b/main.c
@@ -1437,6 +1437,18 @@ void read_control_message(socket_info_t *socket_info)
 
 }
 
+void make_query_sockets_nonblocking()
+{
+    for(size_t i = 0; i < context.sockets.interfaces4.len; i++)
+    {
+        socket_noblock(((socket_info_t*)context.sockets.interfaces4.data) + i);
+    }
+    for(size_t i = 0; i < context.sockets.interfaces6.len; i++)
+    {
+        socket_noblock(((socket_info_t*)context.sockets.interfaces6.data) + i);
+    }
+}
+
 void run()
 {
     static char multiproc_outfile_name[8192];
@@ -1558,6 +1570,10 @@ void run()
         add_sockets(context.epollfd, socket_events, EPOLL_CTL_ADD, &context.sockets.interfaces6);
     }
 #endif
+    if(context.cmd_args.busypoll)
+    {
+        make_query_sockets_nonblocking();
+    }
 
 
     clock_gettime(CLOCK_MONOTONIC, &context.stats.start_time);
@@ -1619,7 +1635,7 @@ void run()
             }
             for(size_t i = 0; i < context.sockets.interfaces6.len; i++)
             {
-                can_read(((socket_info_t*)context.sockets.interfaces4.data) + i);
+                can_read(((socket_info_t*)context.sockets.interfaces6.data) + i);
             }
             timed_ring_handle(&context.ring, ring_timeout);