Rooba/pupy
1
1
Fork 0
mirror of https://github.com/n1nj4sec/pupy.git synced 2025-02-26 21:06:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ssh-like connection close sequence (RET ~.)

Browse Source
This commit is contained in:
Oleksii Shevchuk 2016-10-20 08:59:42 +03:00
parent 1b8b2948ae
commit 4baa96747c
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pupy/modules/interactive_shell.py
View File

@ -60,6 +60,8 @@ class InteractiveShell(PupyModule):
return b''.join(buf) return b''.join(buf)
def _read_loop(self, write_cb, complete): def _read_loop(self, write_cb, complete):
lastbuf = b''
while not complete.is_set(): while not complete.is_set():
r, _, x = select.select([sys.stdin], [], [sys.stdin], None) r, _, x = select.select([sys.stdin], [], [sys.stdin], None)
if x: if x:
@ -68,11 +70,28 @@ class InteractiveShell(PupyModule):
if r: if r:
if not complete.is_set(): if not complete.is_set():
buf = self._read_stdin_non_block() buf = self._read_stdin_non_block()
if lastbuf.startswith(b'\r'):
vbuf = lastbuf + buf
if vbuf.startswith(b'\r~'):
if len(vbuf) < 3:
lastbuf = vbuf
continue
elif vbuf.startswith(b'\r~.'):
sys.stdout.write('\r\n')
break
elif vbuf.startswith(b'\r~,'):
self.client.conn._conn.ping(timeout=1)
buf = buf[3:]
if not buf:
continue
try: try:
write_cb(buf) write_cb(buf)
except: except:
break break
lastbuf = buf
complete.set() complete.set()
def _remote_read(self, data): def _remote_read(self, data):