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

Island: Register reset_machine_repository for CLEAR_SIMULATION_DATA

This commit is contained in:
Mike Salvatore 2022-09-20 14:34:11 -04:00
parent 35025ead18
commit aed9022a7a
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monkey/monkey_island/cc/setup/island_event_handlers.py
View File

@ -2,12 +2,14 @@ from functools import partial
from common import DIContainer from common import DIContainer
from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic
from monkey_island.cc.island_event_handlers import reset_agent_configuration from monkey_island.cc.island_event_handlers import (
reset_agent_configuration,
reset_machine_repository,
)
from monkey_island.cc.repository import ( from monkey_island.cc.repository import (
IAgentEventRepository, IAgentEventRepository,
IAgentRepository, IAgentRepository,
ICredentialsRepository, ICredentialsRepository,
IMachineRepository,
INodeRepository, INodeRepository,
) )
from monkey_island.cc.services.database import Database from monkey_island.cc.services.database import Database
@ -39,11 +41,14 @@ def _subscribe_clear_simulation_data_events(
IslandEventTopic.CLEAR_SIMULATION_DATA, credentials_repository.remove_stolen_credentials IslandEventTopic.CLEAR_SIMULATION_DATA, credentials_repository.remove_stolen_credentials
) )
island_event_queue.subscribe(
IslandEventTopic.CLEAR_SIMULATION_DATA, container.resolve(reset_machine_repository)
)
for i_repository in [ for i_repository in [
INodeRepository, INodeRepository,
IAgentEventRepository, IAgentEventRepository,
IAgentRepository, IAgentRepository,
IMachineRepository,
]: ]:
repository = container.resolve(i_repository) repository = container.resolve(i_repository)
island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, repository.reset) island_event_queue.subscribe(IslandEventTopic.CLEAR_SIMULATION_DATA, repository.reset)