Merge pull request #2 from invariantlabs-ai/playground

Error handling + cleanup

Author: adrgs

server/ipc/controller.py

@@ -23,8 +23,12 @@ def _init(self):
         self.start_process()
 
     def exists(self):
+        if not os.path.exists(self.socket_path):
+            self.close()
+            return False
+
         if settings.production:
-            return os.path.exists(self.socket_path)
+            return True
 
         process = False
         for proc in psutil.process_iter():
@@ -92,7 +96,8 @@ def close(self):
                         break
             except (psutil.AccessDenied, psutil.ZombieProcess, psutil.NoSuchProcess):
                 continue
-        os.remove(self.socket_path)
+        if os.path.exists(self.socket_path):
+            os.remove(self.socket_path)
 
 
 class IpcControllerSingleton:

server/ipc/invariant-ipc.py

@@ -48,6 +48,8 @@ def worker(client_socket):
         data = client_socket.recv(10 * 1024 * 1024)
         response = handle_request(data)
         client_socket.sendall(response)
+    except Exception as e:
+        client_socket.sendall(json.dumps(str(e)).encode())
     finally:
         client_socket.close()