From eee06467f6bc120d86e38052d1d228c59ba058d7 Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Thu, 1 Sep 2022 13:44:39 +0200 Subject: [PATCH] Agent: Add TODO to rethink on_user_disconnected --- monkey/infection_monkey/tcp_relay.py | 3 +-- monkey/infection_monkey/transport/tcp.py | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/tcp_relay.py b/monkey/infection_monkey/tcp_relay.py index 8cdb287c6..833d9b48f 100644 --- a/monkey/infection_monkey/tcp_relay.py +++ b/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() diff --git a/monkey/infection_monkey/transport/tcp.py b/monkey/infection_monkey/transport/tcp.py index b174e0578..dc87c67d0 100644 --- a/monkey/infection_monkey/transport/tcp.py +++ b/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