Fix partial socket writes

Signed-off-by: falkTX <[email protected]>
mod-audio · Oct 27, 2023 · c69480e · melonlm · Feb 5, 2024 · falkTX
1 parent 1714266
commit c69480e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/src/monitor.c b/src/monitor.c
@@ -176,15 +176,21 @@ int monitor_stop(void)
 
 int monitor_send(int instance, const char *symbol, float value)
 {
-    int ret;
+    int size, ret = -1;
 
     char msg[255];
-    sprintf(msg, "monitor %d %s %f", instance, symbol, value);
+    char *buffer = msg;
+    size = sprintf(msg, "monitor %d %s %f", instance, symbol, value);
 
-    ret = send(g_sockfd, msg, strlen(msg) + 1, 0);
-    if (ret < 0)
+    while (size > 0)
     {
-        perror("send error");
+        ret = send(g_sockfd, buffer, size + 1, 0);
+        if (ret < 0)
+        {
+            perror("send error");
+        }
+        size -= ret;
+        buffer += ret;
     }
 
     return ret;

diff --git a/src/socket.c b/src/socket.c
@@ -255,12 +255,17 @@ void socket_set_receive_cb(void (*receive_cb)(msg_t *msg))
 
 int socket_send(int destination, const char *buffer, int size)
 {
-    int ret;
+    int ret = -1;
 
-    ret = send(destination, buffer, size, 0);
-    if (ret < 0)
+    while (size > 0)
     {
-        perror("send error");
+        ret = send(destination, buffer, size, 0);
+        if (ret < 0)
+        {
+            perror("send error");
+        }
+        size -= ret;
+        buffer += ret;
     }
 
     return ret;