p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Agent: Add TODO to rethink on_user_disconnected

This commit is contained in:
Ilija Lazoroski 2022-09-01 13:44:39 +02:00
parent 508b811939
commit eee06467f6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/infection_monkey/tcp_relay.py
View File

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

1
monkey/infection_monkey/transport/tcp.py
View File

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