feat: update

parent 0c20d632
...@@ -148,11 +148,16 @@ class SSHws(Namespace, BaseWebSocketHandler): ...@@ -148,11 +148,16 @@ class SSHws(Namespace, BaseWebSocketHandler):
def on_logout(self, connection): def on_logout(self, connection):
logger.debug("{} logout".format(connection)) logger.debug("{} logout".format(connection))
if connection: if connection:
self.clients[request.sid]["proxy"][connection].close() if connection in self.clients[request.sid]["proxy"].keys():
proxy = self.clients[request.sid]["proxy"][connection]
del self.clients[request.sid]["proxy"][connection] del self.clients[request.sid]["proxy"][connection]
proxy.close()
if connection in self.clients[request.sid]["forwarder"].keys():
del self.clients[request.sid]["forwarder"][connection] del self.clients[request.sid]["forwarder"][connection]
self.clients[request.sid]["client"][connection].close() if connection in self.clients[request.sid]["client"].keys():
client = self.clients[request.sid]["client"][connection]
del self.clients[request.sid]["client"][connection] del self.clients[request.sid]["client"][connection]
client.close()
class HttpServer: class HttpServer:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment