[Update] 增加异常捕捉

coco/app.py

@@ -105,7 +105,10 @@ class Coco:
     def keep_heartbeat(self):
         def func():
             while not self.stop_evt.is_set():
+                try:
                     self.heartbeat()
+                except Exception as e:
+                    logger.error("Unexpected error occur: {}".format(e))
                 time.sleep(config["HEARTBEAT_INTERVAL"])
         thread = threading.Thread(target=func)
         thread.start()
@@ -151,6 +154,7 @@ class Coco:
 
         def func():
             while not self.stop_evt.is_set():
+                try:
                     sessions_copy = [s for s in Session.sessions.values()]
                     for s in sessions_copy:
                         # Session 没有正常关闭,
@@ -162,6 +166,8 @@ class Coco:
                             Session.remove_session(s)
                         else:
                             check_session_idle_too_long(s)
+                except Exception as e:
+                    logger.error("Unexpected error occur: {}".format(e))
                 time.sleep(interval)
         thread = threading.Thread(target=func)
         thread.start()

coco/httpd.py

@@ -103,7 +103,10 @@ class ProxyNamespace(BaseNamespace):
         forwarder = ProxyServer(client, asset, system_user)
 
         def proxy():
+            try:
                 forwarder.proxy()
+            except Exception as e:
+                logger.error("Unexpected error occur: {}".format(e))
             self.logout(client_id, connection)
         self.socketio.start_background_task(proxy)
 

coco/sshd.py

@@ -110,7 +110,10 @@ class SSHServer:
         kind = client.request.kind
         if kind == 'session' and chan_type in supported:
             logger.info("Request type `{}:{}`, dispatch to interactive mode".format(kind, chan_type))
+            try:
                 InteractiveServer(client).interact()
+            except Exception as e:
+                logger.error("Unexpected error occur: {}".format(e))
             connection = Connection.get_connection(client.connection_id)
             connection.remove_client(client.id)
         elif chan_type == 'subsystem':