From efac1dae980674f420f117a2a52325d6875cf57f Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Tue, 6 Sep 2022 19:33:03 +0000 Subject: [PATCH] Agent: Pass correct type to add_potential_user() --- .../messengers/exploit_intercepting_telemetry_messenger.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py index 8f3d15ca3..1e5c6591a 100644 --- a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py +++ b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py @@ -1,4 +1,5 @@ from functools import singledispatch +from ipaddress import IPv4Address from infection_monkey.network.relay.tcp_relay import TCPRelay from infection_monkey.telemetry.exploit_telem import ExploitTelem @@ -41,6 +42,6 @@ def _( if telemetry.propagation_result is True: tunnel.set_wait_for_exploited_machines() if relay: - relay.add_potential_user(str(telemetry.host["ip_addr"])) + relay.add_potential_user(IPv4Address(telemetry.host["ip_addr"])) telemetry_messenger.send_telemetry(telemetry)