With more workers blockingly waiting on udp sockets, we need to keep track of all sockets to shut them down on exit

1 files changed, 7 insertions, 0 deletions

diff --git a/ot_socket.c b/ot_socket.c
index 158459f..c73738c 100644
--- a/ot_socket.c
+++ b/ot_socket.c
@@ -9,6 +9,7 @@
 #include <netinet/in.h>
 #include <string.h>
 #include <sys/socket.h>
+#include <sys/time.h>
 #include <unistd.h>
 
 /* Opentracker */
@@ -48,6 +49,12 @@ int socket_set_blocking(int s, int blocking) {
   return fcntl(s, F_SETFL, blocking ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK));
 }
 
+int socket_set_timeout(int s, int seconds) {
+  struct timeval patience = {seconds, 0};
+
+  return setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &patience, sizeof(patience));
+}
+
 static int set_nonblocking(int s) { return socket_set_blocking(s, 0); }
 
 /* Close without letting the failure we are about to report be overwritten */