Close websocket if there is no corresponding ssh connection

2022-11-21 14:52:38 +08:00 · 2022-11-21 14:52:38 +08:00 · 2a46b52eac
parent 9a7cfe767c
commit 2a46b52eac
1 changed files with 6 additions and 0 deletions
--- a/webssh/handler.py
+++ b/webssh/handler.py
@ -566,7 +566,13 @@ class WsockHandler(MixinHandler, tornado.websocket.WebSocketHandler):
                    *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: