implement a some server

server.py

@@ -61,7 +61,16 @@ class Server(paramiko.ServerInterface):
         return True
 
 
-def handle_ssh_request(client, addr):
+class SSHServer:
+    def __init__(self, host, port):
+        self.host = host
+        self.port = port
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        self.sock.bind((self.host, self.port))
+
+    @staticmethod
+    def handle_ssh_request(client, addr):
         print('Got a connection!')
 
         try:
@@ -74,6 +83,7 @@ def handle_ssh_request(client, addr):
                 raise
             t.add_server_key(host_key)
             server = Server()
+            server.add_prompt(">>")
             try:
                 t.start_server(server=server)
             except paramiko.SSHException:
@@ -111,30 +121,23 @@ def handle_ssh_request(client, addr):
                 pass
             sys.exit(1)
 
-
-def run_server():
+    def listen(self):
+        self.sock.listen(5)
+        while True:
             try:
-        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        sock.bind(('', 2200))
+                client, addr = self.sock.accept()
+                print('Listening for connection ...')
+                threading.Thread(target=self.handle_ssh_request, args=(client, addr)).start()
             except Exception as e:
                 print('*** Bind failed: ' + str(e))
                 traceback.print_exc()
                 sys.exit(1)
 
-    try:
-        sock.listen(100)
-        print('Listening for connection ...')
-        client, addr = sock.accept()
-
-        t = threading.Thread(target=handle_ssh_request, args=(client, addr))
-        t.start()
 
-    except Exception as e:
-        print('*** Listen/accept failed: ' + str(e))
-        traceback.print_exc()
+if __name__ == '__main__':
+    server = SSHServer('', 2200)
+    try:
+        server.listen()
+    except KeyboardInterrupt:
         sys.exit(1)
 
-
-if __name__ == '__main__':
-    run_server()
\ No newline at end of file

test_server.py

+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# 
+
+import socket
+import sys
+import threading
+
+
+class ThreadSocket:
+    def __init__(self, host, port):
+        self.host = host
+        self.port = port
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        self.sock.bind((self.host, self.port))
+
+    def listen(self):
+        self.sock.listen(5)
+        while True:
+            client, address = self.sock.accept()
+            client.settimeout(60)
+            threading.Thread(target=self.handle_client_request, args=(client, address)).start()
+
+    def handle_client_request(self, client, address):
+        print("Get client: %s" % str(address))
+        while True:
+            try:
+                data = client.recv(1024)
+                print("sleep : %s" % str(address))
+                if data:
+                    client.send(data)
+                else:
+                    raise IndexError('Client has disconnected')
+            except:
+                client.close()
+
+
+if __name__ == '__main__':
+    server = ThreadSocket('', 9000)
+    try:
+        server.listen()
+    except KeyboardInterrupt:
+        sys.exit(1)