Updated sshserver.py

pull/12/merge v0.2.5
Sheng 2018-05-30 22:02:17 +08:00
parent 5f52977785
commit cbeca1c377
1 changed files with 3 additions and 2 deletions

View File

@ -38,6 +38,7 @@ host_key = paramiko.RSAKey(filename='tests/test_rsa.key')
print('Read key: ' + u(hexlify(host_key.get_fingerprint()))) print('Read key: ' + u(hexlify(host_key.get_fingerprint())))
banner = u'\r\n\u6b22\u8fce\r\n' banner = u'\r\n\u6b22\u8fce\r\n'
event_timeout = 5
class Server(paramiko.ServerInterface): class Server(paramiko.ServerInterface):
@ -131,12 +132,12 @@ def run_ssh_server(port=2200, running=True):
username = t.get_username() username = t.get_username()
print('{} Authenticated!'.format(username)) print('{} Authenticated!'.format(username))
server.shell_event.wait(2) server.shell_event.wait(timeout=event_timeout)
if not server.shell_event.is_set(): if not server.shell_event.is_set():
print('*** Client never asked for a shell.') print('*** Client never asked for a shell.')
continue continue
server.exec_event.wait(2) server.exec_event.wait(timeout=event_timeout)
if not server.exec_event.is_set(): if not server.exec_event.is_set():
print('*** Client never asked for a command.') print('*** Client never asked for a command.')
continue continue