ipc: socket close() accounts for references

lux-operating-system · Nov 28, 2024 · 160d5ac · 160d5ac
1 parent e7249b1
commit 160d5ac
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/ipc/sockinit.c b/src/ipc/sockinit.c
@@ -225,6 +225,12 @@ int closeSocket(Thread *t, int sd) {
         return -EBADF;
     }
 
+    sock->refCount--;
+    if(sock->refCount) {
+        releaseLock(&lock);
+        return 0;
+    }
+
     if(sock->peer) {
         // disconnect the socket from its peer
         // TODO: for future TCP sockets, terminate the connection here
@@ -234,6 +240,7 @@ int closeSocket(Thread *t, int sd) {
 
     // and delete the socket
     socketUnregister(sock->globalIndex);
+    free(sock);
     closeIO(p, &p->io[sd]);
     releaseLock(&lock);
     return 0;