Close websocket if there is no corresponding ssh connection

changsongyang · Nov 21, 2022 · 2a46b52 · 2a46b52
1 parent 9a7cfe7
commit 2a46b52
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/webssh/handler.py b/webssh/handler.py
@@ -566,7 +566,13 @@ def on_message(self, message):
                     *self.src_addr
                 )
             )
+            self.close(reason='No worker found')
             return
+
+        if worker.closed:
+            self.close(reason='Worker closed')
+            return
+
         try:
             msg = json.loads(message)
         except JSONDecodeError: