Agent: Pass correct type to add_potential_user()

This commit is contained in:
Kekoa Kaaikala 2022-09-06 19:33:03 +00:00
parent bdc8de026e
commit efac1dae98
1 changed files with 2 additions and 1 deletions

View File

@ -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)