From d478e992273aa96dc9e8079b94773bc98e3e5885 Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Thu, 1 Sep 2022 11:29:03 +0200 Subject: [PATCH] Agent: Make TCPRelay non-optional in ExploitInterceptingTelemetryMessenger --- .../messengers/exploit_intercepting_telemetry_messenger.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 5eb8f0310..0d1f4127c 100644 --- a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py +++ b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py @@ -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()