Agent: Accept list of servers in HostExploiter

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