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

Agent: Rename on_potential_new_user to add_potential_user

This commit is contained in:
Ilija Lazoroski 2022-09-01 13:38:31 +02:00
parent a754963222
commit 43a1ddb714
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/infection_monkey/tcp_relay.py
View File

@ -79,7 +79,7 @@ class TCPRelay(Thread):
with self._lock: with self._lock:
return self._relay_users.copy() return self._relay_users.copy()
def on_potential_new_user(self, user_address: IPv4Address): def add_potential_user(self, user_address: IPv4Address):
""" """
Notify TCPRelay that a new user may try and connect. Notify TCPRelay that a new user may try and connect.

2
monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py
View File

@ -41,6 +41,6 @@ def _(
if telemetry.propagation_result is True: if telemetry.propagation_result is True:
tunnel.set_wait_for_exploited_machines() tunnel.set_wait_for_exploited_machines()
if relay: if relay:
relay.on_potential_new_user(str(telemetry.host["ip_addr"])) relay.add_potential_user(str(telemetry.host["ip_addr"]))
telemetry_messenger.send_telemetry(telemetry) telemetry_messenger.send_telemetry(telemetry)

2
monkey/tests/unit_tests/infection_monkey/test_tcp_relay.py
View File

@ -57,7 +57,7 @@ def test_user_removed_on_request(tcp_relay):
# new_user = "0.0.0.1" # new_user = "0.0.0.1"
# relay.start() # relay.start()
# relay.on_potential_new_user(new_user) # relay.add_potential_user(new_user)
# relay.stop() # relay.stop()
# assert not join_or_kill_thread(relay, 0.1) # Should be waiting # assert not join_or_kill_thread(relay, 0.1) # Should be waiting