From 984425021e8957aa1d8a5dfea5204ccb9ec5dfa7 Mon Sep 17 00:00:00 2001 From: Akemi Izuko Date: Sat, 14 Sep 2024 13:41:37 -0600 Subject: [PATCH] Ssh: reformat --- ssh_daemon/daemon.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/ssh_daemon/daemon.py b/ssh_daemon/daemon.py index 68a37db..ed7a965 100644 --- a/ssh_daemon/daemon.py +++ b/ssh_daemon/daemon.py @@ -3,7 +3,6 @@ import socket import threading import select import sys -import os class CustomServer(paramiko.ServerInterface): @@ -27,7 +26,9 @@ class CustomServer(paramiko.ServerInterface): return paramiko.OPEN_SUCCEEDED return paramiko.OPEN_FAILED_ADMINISTRATIVELY_PROHIBITED - def check_channel_pty_request(self, channel, term, width, height, pixelwidth, pixelheight, modes): + def check_channel_pty_request( + self, channel, term, width, height, pixelwidth, pixelheight, modes + ): return True def check_channel_shell_request(self, channel): @@ -40,11 +41,7 @@ class CustomServer(paramiko.ServerInterface): def get_vm_port(username): - user_map = { - 'user0': 9000, - 'user1': 9001, - 'emiliko': 9002 - } + user_map = {"user0": 9000, "user1": 9001, "emiliko": 9002} return user_map.get(username) @@ -73,7 +70,9 @@ 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( + paramiko.Ed25519Key.from_private_key_file("daemon.key") + ) server = CustomServer(None) transport.start_server(server=server) @@ -98,7 +97,7 @@ def handle_client(client_sock): vm_transport = paramiko.Transport(("localhost", vm_port)) vm_transport.start_client() - vm_transport.auth_password('root', '') + vm_transport.auth_password("root", "") vm_channel = vm_transport.open_session() vm_channel.get_pty() @@ -129,7 +128,8 @@ def handle_client(client_sock): except: pass -def start_server(port=22, bind_address=''): + +def start_server(port=22, bind_address=""): try: sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) @@ -148,5 +148,6 @@ def start_server(port=22, bind_address=''): except Exception as e: print(f"Error: {str(e)}") -if __name__ == '__main__': + +if __name__ == "__main__": start_server(port=2222)