p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Integrate VictimHostFactory with monkey.py

This commit is contained in:
Mike Salvatore 2021-12-16 15:20:38 -05:00
parent 18fb4e7533
commit 637053e6cd
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
monkey/infection_monkey/monkey.py
View File

@ -168,14 +168,30 @@ class InfectionMonkey:
return local_network_interfaces return local_network_interfaces
def _build_master(self, local_network_interfaces: List[NetworkInterface]): def _build_master(self, local_network_interfaces: List[NetworkInterface]):
victim_host_factory = self._build_victim_host_factory(local_network_interfaces)
self._master = AutomatedMaster( self._master = AutomatedMaster(
MockPuppet(), MockPuppet(),
LegacyTelemetryMessengerAdapter(), LegacyTelemetryMessengerAdapter(),
VictimHostFactory(), victim_host_factory,
ControlChannel(self._default_server, GUID), ControlChannel(self._default_server, GUID),
local_network_interfaces, local_network_interfaces,
) )
def _build_victim_host_factory(
self, local_network_interfaces: List[NetworkInterface]
) -> VictimHostFactory:
on_island = self._running_on_island(local_network_interfaces)
logger.debug(f"This agent is running on the island: {on_island}")
return VictimHostFactory(
self._monkey_inbound_tunnel, self._default_server, self._default_server_port, on_island
)
def _running_on_island(self, local_network_interfaces: List[NetworkInterface]) -> bool:
server_ip = self._default_server.split(":")[0]
return server_ip in {interface.address for interface in local_network_interfaces}
def _is_another_monkey_running(self): def _is_another_monkey_running(self):
return not self._singleton.try_lock() return not self._singleton.try_lock()