p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Make TCPRelay non-optional in ExploitInterceptingTelemetryMessenger

This commit is contained in:
Ilija Lazoroski 2022-09-01 11:29:03 +02:00
parent ecdf04a85e
commit d478e99227
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py
View File

@ -9,7 +9,7 @@ from infection_monkey.tunnel import MonkeyTunnel
class ExploitInterceptingTelemetryMessenger(ITelemetryMessenger): class ExploitInterceptingTelemetryMessenger(ITelemetryMessenger):
def __init__( 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._telemetry_messenger = telemetry_messenger
self._tunnel = tunnel self._tunnel = tunnel
@ -26,7 +26,7 @@ def _send_telemetry(
telemetry: ITelem, telemetry: ITelem,
telemetry_messenger: ITelemetryMessenger, telemetry_messenger: ITelemetryMessenger,
tunnel: MonkeyTunnel, tunnel: MonkeyTunnel,
relay: TCPRelay = None, relay: TCPRelay,
): ):
telemetry_messenger.send_telemetry(telemetry) telemetry_messenger.send_telemetry(telemetry)
@ -36,7 +36,7 @@ def _(
telemetry: ExploitTelem, telemetry: ExploitTelem,
telemetry_messenger: ITelemetryMessenger, telemetry_messenger: ITelemetryMessenger,
tunnel: MonkeyTunnel, tunnel: MonkeyTunnel,
relay: TCPRelay = None, relay: TCPRelay,
): ):
if telemetry.propagation_result is True: if telemetry.propagation_result is True:
tunnel.set_wait_for_exploited_machines() tunnel.set_wait_for_exploited_machines()