diff --git a/ssh_daemon/daemon.py b/ssh_daemon/daemon.py index ed7a965..82db52e 100644 --- a/ssh_daemon/daemon.py +++ b/ssh_daemon/daemon.py @@ -4,6 +4,8 @@ import threading import select import sys +SERVER_KEY = paramiko.Ed25519Key.from_private_key_file("daemon.key") + class CustomServer(paramiko.ServerInterface): def __init__(self, username): @@ -70,9 +72,7 @@ def handle_tcp_forwarding(channel, origin, destination): def handle_client(client_sock): try: transport = paramiko.Transport(client_sock) - transport.add_server_key( - paramiko.Ed25519Key.from_private_key_file("daemon.key") - ) + transport.add_server_key(SERVER_KEY) server = CustomServer(None) transport.start_server(server=server)