From 45378a6df7579909efab70cc6676791ded0ee317 Mon Sep 17 00:00:00 2001
From: Tristan Russell <tristanr@trussnexus.com>
Date: Fri, 17 Jan 2025 17:59:13 -0500
Subject: [PATCH] fix: No longer hangs when socket closes/disconnects

---
 main.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.c b/main.c
index 8608e15..bae3479 100644
--- a/main.c
+++ b/main.c
@@ -49,8 +49,9 @@ int StartEnv(int port) {
         bytes_received = recv(client_sock, buffer, sizeof(buffer), 0);
         if(bytes_received < 0) {
             printf("Receive Error: %s (%d)\n", strerror(errno), errno);
+            break;
         } else if(bytes_received == 0) {
-            continue;
+            break;
         }
 
         if(buffer[0] == 0x0) { // Step
@@ -80,6 +81,7 @@ int StartEnv(int port) {
     }
     shutdown(client_sock, SHUT_RDWR);
     close(client_sock);
+    return 0;
 }
 
 int main(int argc, char* argv[]) {