forked from p15670423/monkey
Agent: Pass correct type to add_potential_user()
This commit is contained in:
parent
bdc8de026e
commit
efac1dae98
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue