diff --git a/monkey/infection_monkey/exploit/HostExploiter.py b/monkey/infection_monkey/exploit/HostExploiter.py index bb5d7f87d..db8feb1fd 100644 --- a/monkey/infection_monkey/exploit/HostExploiter.py +++ b/monkey/infection_monkey/exploit/HostExploiter.py @@ -2,7 +2,7 @@ import logging import threading from abc import abstractmethod from datetime import datetime -from typing import Dict +from typing import Dict, List from common.event_queue import IAgentEventQueue from common.utils.exceptions import FailedExploitationError @@ -35,6 +35,7 @@ class HostExploiter: self.event_queue = None self.options = {} self.exploit_result = {} + self.servers = [] def set_start_time(self): self.exploit_info["started"] = datetime.now().isoformat() @@ -58,6 +59,7 @@ class HostExploiter: def exploit_host( self, host, + servers: List[str], current_depth: int, telemetry_messenger: ITelemetryMessenger, event_queue: IAgentEventQueue, @@ -66,6 +68,7 @@ class HostExploiter: interrupt: threading.Event, ): self.host = host + self.servers = servers self.current_depth = current_depth self.telemetry_messenger = telemetry_messenger self.event_queue = event_queue