From 2a46b52eacdc773a35e91f225810c32a2877b5f6 Mon Sep 17 00:00:00 2001
From: Sheng <webmaster0115@gmail.com>
Date: Mon, 21 Nov 2022 14:52:38 +0800
Subject: [PATCH] Close websocket if there is no corresponding ssh connection

---
 webssh/handler.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/webssh/handler.py b/webssh/handler.py
index 0fe6afc..6cfc822 100644
--- 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: