forked from p15670423/monkey
Agent: Make TCPRelay non-optional in ExploitInterceptingTelemetryMessenger
This commit is contained in:
parent
ecdf04a85e
commit
d478e99227
|
@ -9,7 +9,7 @@ from infection_monkey.tunnel import MonkeyTunnel
|
|||
|
||||
class ExploitInterceptingTelemetryMessenger(ITelemetryMessenger):
|
||||
def __init__(
|
||||
self, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, relay: TCPRelay = None
|
||||
self, telemetry_messenger: ITelemetryMessenger, tunnel: MonkeyTunnel, relay: TCPRelay
|
||||
):
|
||||
self._telemetry_messenger = telemetry_messenger
|
||||
self._tunnel = tunnel
|
||||
|
@ -26,7 +26,7 @@ def _send_telemetry(
|
|||
telemetry: ITelem,
|
||||
telemetry_messenger: ITelemetryMessenger,
|
||||
tunnel: MonkeyTunnel,
|
||||
relay: TCPRelay = None,
|
||||
relay: TCPRelay,
|
||||
):
|
||||
telemetry_messenger.send_telemetry(telemetry)
|
||||
|
||||
|
@ -36,7 +36,7 @@ def _(
|
|||
telemetry: ExploitTelem,
|
||||
telemetry_messenger: ITelemetryMessenger,
|
||||
tunnel: MonkeyTunnel,
|
||||
relay: TCPRelay = None,
|
||||
relay: TCPRelay,
|
||||
):
|
||||
if telemetry.propagation_result is True:
|
||||
tunnel.set_wait_for_exploited_machines()
|
||||
|
|
Loading…
Reference in New Issue