From 84cb14e1c50c7e9d876dd17eab9b2cbac53be9f0 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 2 Mar 2022 09:44:12 -0500 Subject: [PATCH] Agent: Pass ExploitInterceptingTelemetryMessenger to Master --- monkey/infection_monkey/monkey.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index db0ba58c7..bd11a38c4 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -36,6 +36,9 @@ from infection_monkey.puppet.puppet import Puppet from infection_monkey.system_singleton import SystemSingleton from infection_monkey.telemetry.attack.t1106_telem import T1106Telem from infection_monkey.telemetry.attack.t1107_telem import T1107Telem +from infection_monkey.telemetry.messengers.exploit_intercepting_telemetry_messenger import ( + ExploitInterceptingTelemetryMessenger, +) from infection_monkey.telemetry.messengers.legacy_telemetry_messenger_adapter import ( LegacyTelemetryMessengerAdapter, ) @@ -164,9 +167,13 @@ class InfectionMonkey: victim_host_factory = self._build_victim_host_factory(local_network_interfaces) + telemetry_messenger = ExploitInterceptingTelemetryMessenger( + self.telemetry_messenger, self._monkey_inbound_tunnel + ) + self._master = AutomatedMaster( puppet, - self.telemetry_messenger, + telemetry_messenger, victim_host_factory, ControlChannel(self._default_server, GUID), local_network_interfaces,