Skip to content

C
coco
Commit 91576bf7 authored by liuzheng712's avatar liuzheng712
Browse files
Options

feat: update

parent 560b6de4
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
coco/httpd.py
View file @ 91576bf7
...@@ -47,12 +47,17 @@ class BaseWebSocketHandler: ...@@ -47,12 +47,17 @@ class BaseWebSocketHandler:
return True return True
def close(self): def close(self):
try:
self.ssh.close()
except:
pass
pass pass
class SSHws(Namespace, BaseWebSocketHandler): class SSHws(Namespace, BaseWebSocketHandler):
def ssh_with_password(self): def ssh_with_password(self):
self.ssh = paramiko.SSHClient() self.ssh = paramiko.SSHClient()
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh.connect("127.0.0.1", 22, "liuzheng", "liuzheng") self.ssh.connect("127.0.0.1", 22, "liuzheng", "liuzheng")
self.chan = self.ssh.invoke_shell(term='xterm', width=self.cols, height=self.rows) self.chan = self.ssh.invoke_shell(term='xterm', width=self.cols, height=self.rows)
...@@ -95,6 +100,10 @@ class SSHws(Namespace, BaseWebSocketHandler): ...@@ -95,6 +100,10 @@ class SSHws(Namespace, BaseWebSocketHandler):
system_user = username system_user = username
if system_user: if system_user:
self.forwarder.proxy(self.asset, system_user) self.forwarder.proxy(self.asset, system_user)
else:
self.close()
else:
self.close()
def on_resize(self, message): def on_resize(self, message):
self.request.meta['width'] = message.get('cols', 80) self.request.meta['width'] = message.get('cols', 80)
......
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