From 4cde2aef1f9f77f559a19d35b796fb3a19184eae Mon Sep 17 00:00:00 2001 From: Jesse Luehrs Date: Mon, 15 Sep 2014 02:01:32 -0400 Subject: make the ssh private key configurable --- server.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'server.py') diff --git a/server.py b/server.py index 8eb5a7e..0137b2b 100644 --- a/server.py +++ b/server.py @@ -1,4 +1,5 @@ import socket +import sys import threading import uuid @@ -7,8 +8,9 @@ import ssh import termcast class Server(object): - def __init__(self): + def __init__(self, rsa_keyfile): self.publisher = pubsub.Publisher() + self.rsa_keyfile = rsa_keyfile def listen(self): ssh_sock = self._open_socket(2200) @@ -36,7 +38,7 @@ class Server(object): def handle_ssh_connection(self, client): self._handle_connection( client, - lambda client, connection_id: ssh.Connection(client, connection_id, self.publisher) + lambda client, connection_id: ssh.Connection(client, connection_id, self.publisher, self.rsa_keyfile) ) def handle_termcast_connection(self, client): @@ -71,5 +73,5 @@ class Server(object): return sock if __name__ == '__main__': - server = Server() + server = Server(sys.argv[1]) server.listen() -- cgit v1.2.3-54-g00ecf