Use non-blocking sockets for client communication

This approach is better than using blocking sockets and select() since
there's no guarantee that send() doesn't block. Plus we're using poll()
anyway so send() and recv() will only be called if the socket is actually
ready for writing/reading.

2 files changed, 31 insertions, 1 deletions

diff --git a/src/client.c b/src/client.c
index 268c8b9..4ec4025 100644
--- a/src/client.c
+++ b/src/client.c
@@ -33,6 +33,7 @@
 #include <sys/un.h>
 #include <arpa/inet.h>
 #include <pthread.h>
+#include <fcntl.h>
 
 #include <plist/plist.h>
 
@@ -100,12 +101,23 @@ int client_read(struct mux_client *client, void *buffer, uint32_t len)
  */
 int client_write(struct mux_client *client, void *buffer, uint32_t len)
 {
+        int sret = -1;
+
         usbmuxd_log(LL_SPEW, "client_write fd %d buf %p len %d", client->fd, buffer, len);
         if(client->state != CLIENT_CONNECTED) {
                 usbmuxd_log(LL_ERROR, "Attempted to write to client %d not in CONNECTED state", client->fd);
                 return -1;
         }
-        return send(client->fd, buffer, len, 0);
+
+        sret = send(client->fd, buffer, len, 0);
+        if (sret < 0) {
+                if ((errno == EAGAIN) || (errno == EWOULDBLOCK)) {
+                        usbmuxd_log(LL_ERROR, "ERROR: client_write: fd %d not ready for writing", client->fd);
+                } else {
+                        usbmuxd_log(LL_ERROR, "ERROR: client_write: sending to fd %d failed: %s", client->fd, strerror(errno));
+                }
+        }
+        return sret;
 }
 
 /**
@@ -150,6 +162,15 @@ int client_accept(int listenfd)
                 return cfd;
         }
 
+        int flags = fcntl(cfd, F_GETFL, 0);
+        if (flags < 0) {
+                usbmuxd_log(LL_ERROR, "ERROR: Could not get socket flags!");
+        } else {
+                if (fcntl(cfd, F_SETFL, flags | O_NONBLOCK) < 0) {
+                        usbmuxd_log(LL_ERROR, "ERROR: Could not set socket to non-blocking mode");
+                }
+        }
+
         struct mux_client *client;
         client = malloc(sizeof(struct mux_client));
         memset(client, 0, sizeof(struct mux_client));
diff --git a/src/main.c b/src/main.c
index b1f4eeb..2e4439c 100644
--- a/src/main.c
+++ b/src/main.c
@@ -82,6 +82,15 @@ static int create_socket(void) {
                 return -1;
         }
 
+        int flags = fcntl(listenfd, F_GETFL, 0);
+        if (flags < 0) {
+                usbmuxd_log(LL_FATAL, "ERROR: Could not get flags for socket");
+        } else {
+                if (fcntl(listenfd, F_SETFL, flags | O_NONBLOCK) < 0) {
+                        usbmuxd_log(LL_FATAL, "ERROR: Could not set socket to non-blocking");
+                }
+        }
+
         bzero(&bind_addr, sizeof(bind_addr));
         bind_addr.sun_family = AF_UNIX;
         strcpy(bind_addr.sun_path, socket_path);