Randomise debugger IPC authkey

Author: ObaraEmmanuel

studio/debugtools/debugger.py

@@ -81,18 +81,19 @@ def __init__(self):
 
     def start_server_client(self):
         logging.debug("Starting server client...")
-        self._server_client = Client(("localhost", 6999), authkey=b'studio-debugger')
+        self._server_client = Client(("localhost", 6999), authkey=self.pref.get("IPC::authkey"))
         self._server_client.send("SERVER")
         self._server_client.send(Message("HOOK", payload={"set": "styles", "value": self.style}))
 
     def stream_client(self):
         logging.debug("Starting stream client...")
-        self._stream_client = Client(("localhost", 6999), authkey=b'studio-debugger')
+        self._stream_client = Client(("localhost", 6999), authkey=self.pref.get("IPC::authkey"))
         self._stream_client.send("STREAM")
         while True:
             try:
                 msg = self._stream_client.recv()
-            except ConnectionAbortedError:
+            except (ConnectionAbortedError, ConnectionResetError):
+                self.exit()
                 break
             if msg == "TERMINATE":
                 self.exit()
@@ -114,9 +115,17 @@ def transmit(self, msg, response=False):
             return
         if isinstance(msg, Message):
             msg.payload = marshal(msg.payload)
-        self._server_client.send(msg)
+        try:
+            self._server_client.send(msg)
+        except (ConnectionResetError, ConnectionAbortedError):
+            self._server_client = None
+            return
         if response:
-            result = self._server_client.recv()
+            try:
+                result = self._server_client.recv()
+            except (ConnectionResetError, ConnectionAbortedError, EOFError):
+                self._server_client = None
+                return
             logger.debug("Received response: %s", result)
             if isinstance(result, Exception):
                 raise result

studio/debugtools/hook.py

@@ -1,6 +1,7 @@
 # ======================================================================= #
 # Copyright (C) 2024 Hoverset Group.                                      #
 # ======================================================================= #
+from studio.debugtools.preferences import Preferences
 
 _global_freeze = dict(globals())
 
@@ -88,11 +89,12 @@ def root(self) -> tkinter.Tk:
 class DebuggerHook:
 
     def __init__(self, path=None):
+        pref = Preferences.acquire()
         self.path = path
         self.root = None
         self.active_widget = None
         self.enable_hooks = True
-        self.listener = Listener(('localhost', 6999), authkey=b'studio-debugger')
+        self.listener = Listener(('localhost', 6999), authkey=pref.get("IPC::authkey"))
         self._handle_map = {}
         self.styles = None
         self._allow_hover = False

studio/debugtools/preferences.py

@@ -1,3 +1,5 @@
+import uuid
+
 from hoverset.data.preferences import *
 import studio.preferences
 
@@ -12,6 +14,9 @@
     },
     "locale": {
         "language": "en"
+    },
+    "IPC": {
+        "authkey": bytes(str(uuid.uuid4()), encoding='utf-8')
     }
 }