lili/amodem
1
0
Fork 0
mirror of https://github.com/romanz/amodem.git synced 2026-02-07 01:18:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

protocol: use Handler class and fix pylint warnings

Browse Source
This commit is contained in:
Roman Zeyde
2015-06-16 10:34:02 +03:00
parent 7ef71df81b
commit 8596537a54
2 changed files with 70 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sshagent/server.py
View File

@@ -31,12 +31,12 @@ def unix_domain_socket_server(sock_path):
os.remove(sock_path)
def handle_connection(conn, keys, signer):
def handle_connection(conn, handler):
try:
log.debug('welcome agent')
while True:
msg = util.read_frame(conn)
reply = protocol.handle_message(msg=msg, keys=keys, signer=signer)
reply = handler.handle(msg=msg)
util.send(conn, reply)
except EOFError:
log.debug('goodbye agent')
@@ -47,6 +47,7 @@ def handle_connection(conn, keys, signer):
def server_thread(server, keys, signer):
log.debug('server thread started')
handler = protocol.Handler(keys=keys, signer=signer)
while True:
log.debug('waiting for connection on %s', server.getsockname())
try:
@@ -55,7 +56,7 @@ def server_thread(server, keys, signer):
log.debug('server error: %s', e, exc_info=True)
break
with contextlib.closing(conn):
handle_connection(conn, keys, signer)
handle_connection(conn, handler)
log.debug('server thread stopped')