ipc: fix potential memory leak in send()

lux-operating-system · Sep 7, 2024 · dc9e8bc · dc9e8bc
1 parent f5e408a
commit dc9e8bc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/ipc/sockio.c b/src/ipc/sockio.c
@@ -52,9 +52,16 @@ ssize_t send(Thread *t, int sd, const void *buffer, size_t len, int flags) {
 
         size_t *newlen = realloc(peer->inboundLen, (peer->inboundCount+1) * sizeof(size_t));
 
+        if(!newlen) {
+            free(newlist);
+            socketRelease();
+            return -ENOBUFS;
+        }
+
         void *message = malloc(len);
         if(!message) {
             free(newlist);
+            free(newlen);
             socketRelease();
             return -ENOBUFS;
         }