forked from p15670423/monkey
Agent: Add event_queue to the exploit_host in HostExploiter
This commit is contained in:
parent
fb0f7c86af
commit
c6cb477474
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue