Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

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)
def _read_loop(self, write_cb, complete):
lastbuf = b''
while not complete.is_set():
r, _, x = select.select([sys.stdin], [], [sys.stdin], None)
if x:
@ -68,11 +70,28 @@ class InteractiveShell(PupyModule):
if r:
if not complete.is_set():
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:
write_cb(buf)
except:
break
lastbuf = buf
complete.set()
def _remote_read(self, data):