[Update] 添加ssh连接资产的超时时间为可配置选项SSH_TIMEOUT，默认15s (#88)

coco/app.py

@@ -54,6 +54,7 @@ class Coco:
         'HEARTBEAT_INTERVAL': 5,
         'MAX_CONNECTIONS': 500,
         'ADMINS': '',
+        'SSH_TIMEOUT': 15,
         'COMMAND_STORAGE': {'TYPE': 'server'},   # server
         'REPLAY_STORAGE': {'TYPE': 'server'},
     }

coco/connection.py

@@ -10,11 +10,11 @@ import telnetlib
 import paramiko
 from paramiko.ssh_exception import SSHException
 
-from .ctx import app_service
+from .ctx import app_service, current_app
 from .utils import get_logger, get_private_key_fingerprint, net_input
 
 logger = get_logger(__file__)
-TIMEOUT = 10
+
 BUF_SIZE = 1024
 MANUAL_LOGIN = 'manual'
 AUTO_LOGIN = 'auto'
@@ -46,7 +46,8 @@ class SSHConnection:
             ssh.connect(
                 asset.ip, port=asset.port, username=system_user.username,
                 password=system_user.password, pkey=system_user.private_key,
-                timeout=TIMEOUT, compress=True, auth_timeout=TIMEOUT,
+                timeout=current_app.config['SSH_TIMEOUT'],
+                compress=True, auth_timeout=current_app.config['SSH_TIMEOUT'],
                 look_for_keys=False, sock=sock
             )
         except (paramiko.AuthenticationException,
@@ -111,7 +112,7 @@ class SSHConnection:
                             username=gateway.username,
                             password=gateway.password,
                             pkey=gateway.private_key_obj,
-                            timeout=TIMEOUT)
+                            timeout=current_app.config['SSH_TIMEOUT'])
             except(paramiko.AuthenticationException,
                    paramiko.BadAuthenticationType,
                    SSHException):

coco/proxy.py

@@ -16,7 +16,6 @@ from .utils import wrap_with_line_feed as wr, wrap_with_warning as warning, \
 
 
 logger = get_logger(__file__)
-TIMEOUT = 10
 BUF_SIZE = 4096
 MANUAL_LOGIN = 'manual'
 AUTO_LOGIN = 'auto'
@@ -156,8 +155,11 @@ class ProxyServer:
             self.client.send('Connecting to {}@{} {:.1f}'.format(
                 system_user, asset, delay)
             )
-            while self.connecting and delay < TIMEOUT:
-                self.client.send('\x08\x08\x08{:.1f}'.format(delay).encode())
+            while self.connecting and delay < current_app.config['SSH_TIMEOUT']:
+                if 0 <= delay < 10:
+                    self.client.send('\x08\x08\x08{:.1f}'.format(delay).encode())
+                else:
+                    self.client.send('\x08\x08\x08\x08{:.1f}'.format(delay).encode())
                 time.sleep(0.1)
                 delay += 0.1
         thread = threading.Thread(target=func)

conf_example.py

@@ -66,5 +66,8 @@ class Config:
         "TYPE": "server"
     }
 
+    # SSH connection timeout (default 15 seconds)
+    # SSH_TIMEOUT = 15
+
 
 config = Config()