1 files changed, 4 insertions, 2 deletions

diff --git a/ot_socket.c b/ot_socket.c
index 84020ca..576087f 100644
--- a/ot_socket.c
+++ b/ot_socket.c
@@ -40,14 +40,16 @@ static void ip_from_sockaddr(ot_ip6 ip, uint16_t *port, const struct sockaddr_st
   }
 }
 
-static int set_nonblocking(int s) {
+int socket_set_blocking(int s, int blocking) {
   int flags = fcntl(s, F_GETFL, 0);
 
   if (flags == -1)
     return -1;
-  return fcntl(s, F_SETFL, flags | O_NONBLOCK);
+  return fcntl(s, F_SETFL, blocking ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK));
 }
 
+static int set_nonblocking(int s) { return socket_set_blocking(s, 0); }
+
 /* Close without letting the failure we are about to report be overwritten */
 static int close_and_fail(int s) {
   int saved = errno;