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 functools import singledispatch
|
||||||
|
from ipaddress import IPv4Address
|
||||||
|
|
||||||
from infection_monkey.network.relay.tcp_relay import TCPRelay
|
from infection_monkey.network.relay.tcp_relay import TCPRelay
|
||||||
from infection_monkey.telemetry.exploit_telem import ExploitTelem
|
from infection_monkey.telemetry.exploit_telem import ExploitTelem
|
||||||
|
@ -41,6 +42,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.add_potential_user(str(telemetry.host["ip_addr"]))
|
relay.add_potential_user(IPv4Address(telemetry.host["ip_addr"]))
|
||||||
|
|
||||||
telemetry_messenger.send_telemetry(telemetry)
|
telemetry_messenger.send_telemetry(telemetry)
|
||||||
|
|
Loading…
Reference in New Issue