Fix Unix remote watchdog socket leak

src/fenrirscreenreader/remoteDriver/unixDriver.py

@@ -329,48 +329,54 @@ class driver(remoteDriver):
     def watch_dog(self, active, event_queue):
         # echo "command say this is a test" | socat -
         # UNIX-CLIENT:/tmp/fenrirscreenreader-daemon.sock
-        for socket_file, optional in self._get_socket_candidates():
-            fenrir_sock = self._bind_socket(socket_file, optional)
-            if fenrir_sock is None:
-                continue
-            self.fenrirSocks.append(fenrir_sock)
-            self.bound_sockets.append((fenrir_sock, socket_file))
+        try:
+            for socket_file, optional in self._get_socket_candidates():
+                fenrir_sock = self._bind_socket(socket_file, optional)
+                if fenrir_sock is None:
+                    continue
+                self.fenrirSocks.append(fenrir_sock)
+                self.bound_sockets.append((fenrir_sock, socket_file))
 
-        if not self.fenrirSocks:
-            return
+            if not self.fenrirSocks:
+                return
 
-        self._try_register_instance()
-        last_register = time.time()
-        while active.value:
-            if time.time() - last_register > 10.0:
-                self._try_register_instance()
-                last_register = time.time()
+            self._try_register_instance()
+            last_register = time.time()
+            while active.value:
+                if time.time() - last_register > 10.0:
+                    self._try_register_instance()
+                    last_register = time.time()
 
-            # Check if the client is still connected and if data is available:
-            try:
-                r, _, _ = select.select(self.fenrirSocks, [], [], 0.8)
-            except select.error:
-                break
-            if r == []:
-                continue
-            for fenrir_sock in r:
-                client_sock, client_addr = fenrir_sock.accept()
-                socket_file = self._socket_file_for_socket(fenrir_sock)
-                # Ensure client socket is always closed to prevent resource
-                # leaks
+                # Check if the client is still connected and data is available.
                 try:
-                    self._handle_client(client_sock, event_queue, socket_file)
-                finally:
-                    # Always close client socket, even if data processing fails
+                    r, _, _ = select.select(self.fenrirSocks, [], [], 0.8)
+                except select.error:
+                    break
+                if r == []:
+                    continue
+                for fenrir_sock in r:
+                    client_sock, _client_addr = fenrir_sock.accept()
+                    socket_file = self._socket_file_for_socket(fenrir_sock)
                     try:
-                        client_sock.close()
-                    except Exception as e:
-                        self.env["runtime"]["DebugManager"].write_debug_out(
-                            "unixDriver watch_dog: Error closing client socket: "
-                            + str(e),
-                            debug.DebugLevel.ERROR,
+                        self._handle_client(
+                            client_sock, event_queue, socket_file
                         )
-        self._cleanup()
+                    finally:
+                        try:
+                            client_sock.close()
+                        except Exception as e:
+                            self.env["runtime"][
+                                "DebugManager"
+                            ].write_debug_out(
+                                "unixDriver watch_dog: Error closing client "
+                                "socket: "
+                                + str(e),
+                                debug.DebugLevel.ERROR,
+                            )
+        finally:
+            # ProcessManager restarts a failed watchdog. Always release the
+            # old listeners first so each restart cannot leak another socket.
+            self._cleanup()
 
     def shutdown(self):
         self._cleanup()