From c6cb4774747e63392eb777727d14eeb3c7ed956b Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Mon, 15 Aug 2022 17:54:40 +0200 Subject: [PATCH] Agent: Add event_queue to the exploit_host in HostExploiter --- monkey/infection_monkey/exploit/HostExploiter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/monkey/infection_monkey/exploit/HostExploiter.py b/monkey/infection_monkey/exploit/HostExploiter.py index 602dd338a..4b0c78c21 100644 --- a/monkey/infection_monkey/exploit/HostExploiter.py +++ b/monkey/infection_monkey/exploit/HostExploiter.py @@ -4,6 +4,7 @@ from abc import abstractmethod from datetime import datetime from typing import Dict +from common.event_queue import IEventQueue from common.utils.exceptions import FailedExploitationError from infection_monkey.i_puppet import ExploiterResultData from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger @@ -31,6 +32,7 @@ class HostExploiter: self.exploit_attempts = [] self.host = None self.telemetry_messenger = None + self.event_queue = None self.options = {} self.exploit_result = {} @@ -58,6 +60,7 @@ class HostExploiter: host, current_depth: int, telemetry_messenger: ITelemetryMessenger, + event_queue: IEventQueue, agent_repository: IAgentRepository, options: Dict, interrupt: threading.Event, @@ -65,6 +68,7 @@ class HostExploiter: self.host = host self.current_depth = current_depth self.telemetry_messenger = telemetry_messenger + self.event_queue = event_queue self.agent_repository = agent_repository self.options = options self.interrupt = interrupt