Agent: Add event_queue to the exploit_host in HostExploiter

This commit is contained in:
Ilija Lazoroski 2022-08-15 17:54:40 +02:00
parent fb0f7c86af
commit c6cb477474
1 changed files with 4 additions and 0 deletions

View File

@ -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