fix: update

parent 66e74a7d
...@@ -146,23 +146,19 @@ class SSHws(Namespace, BaseWebSocketHandler): ...@@ -146,23 +146,19 @@ class SSHws(Namespace, BaseWebSocketHandler):
pass pass
def on_logout(self, connection): def on_logout(self, connection):
if connection and (request.sid in self.clients.keys()): if connection:
if connection in self.clients[request.sid]["proxy"].keys(): if connection in self.clients[request.sid]["proxy"].keys():
proxy = self.clients[request.sid]["proxy"][connection] self.clients[request.sid]["proxy"][connection].close()
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]
if connection in self.clients[request.sid]["client"].keys(): if connection in self.clients[request.sid]["client"].keys():
client = self.clients[request.sid]["client"][connection] self.clients[request.sid]["client"][connection].close()
del self.clients[request.sid]["client"][connection]
client.close()
def logout(self, connection): def logout(self, connection):
if connection and (request.sid in self.clients.keys()): if connection and (request.sid in self.clients.keys()):
if connection in self.clients[request.sid]["proxy"].keys(): if connection in self.clients[request.sid]["proxy"].keys():
del self.clients[request.sid]["proxy"][connection] del self.clients[request.sid]["proxy"][connection]
if connection in self.clients[request.sid]["forwarder"].keys(): if connection in self.clients[request.sid]["forwarder"].keys():
del self.clients[request.sid]["forwarder"][connection]
if connection in self.clients[request.sid]["client"].keys():
del self.clients[request.sid]["client"][connection] del self.clients[request.sid]["client"][connection]
......
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