Agent: Add TODO to rethink on_user_disconnected

monkey/infection_monkey/tcp_relay.py

@@ -44,8 +44,7 @@ class TCPRelay(Thread):
             local_port=self._local_port,
             dest_host=self._target_addr,
             dest_port=self._target_port,
-            client_connected=self.on_user_connected,
-            client_disconnected=self.on_user_disconnected,
+            client_connected=self.add_relay_user,
             client_data_received=self.on_user_data_received,
         )
         proxy.start()

monkey/infection_monkey/transport/tcp.py

@@ -80,6 +80,7 @@ class TcpProxy(TransportProxyBase):
     ):
         super().__init__(local_port, dest_host, dest_port, local_host)
         self._client_connected = client_connected
+        # TODO: Rethink client_disconnected
         self._client_disconnected = client_disconnected
         self._client_data_received = client_data_received