Agent: Construct AutomatedMaster with list of servers

This commit is contained in:
Ilija Lazoroski 2022-09-05 12:40:56 +02:00 committed by Mike Salvatore
parent e6b0e43580
commit a79b579cdc
1 changed files with 3 additions and 1 deletions

View File

@ -154,7 +154,8 @@ class InfectionMonkey:
logger.debug(f"Default server set to: {self._control_client.server_address}") logger.debug(f"Default server set to: {self._control_client.server_address}")
else: else:
raise Exception( raise Exception(
f"Failed to connect to the island via any known server address: {self._opts.servers}" f"Failed to connect to the island via "
f"any known server address: {self._opts.servers}"
) )
self._control_client.wakeup(parent=self._opts.parent) self._control_client.wakeup(parent=self._opts.parent)
@ -219,6 +220,7 @@ class InfectionMonkey:
self._master = AutomatedMaster( self._master = AutomatedMaster(
self._current_depth, self._current_depth,
self._opts.servers,
puppet, puppet,
telemetry_messenger, telemetry_messenger,
victim_host_factory, victim_host_factory,