From 9479b56722c79a03ce42c240e68b333b77635ff9 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 12 Sep 2022 13:51:45 -0400 Subject: [PATCH] Island: Subscribe ICredentialsRepository to CLEAR_SIMULATION_DATA --- monkey/monkey_island/cc/setup/island_event_handlers.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/monkey/monkey_island/cc/setup/island_event_handlers.py b/monkey/monkey_island/cc/setup/island_event_handlers.py index 9f179f315..3bf01e56f 100644 --- a/monkey/monkey_island/cc/setup/island_event_handlers.py +++ b/monkey/monkey_island/cc/setup/island_event_handlers.py @@ -1,12 +1,14 @@ from common import DIContainer from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic from monkey_island.cc.island_event_handlers import reset_agent_configuration +from monkey_island.cc.repository import ICredentialsRepository def setup_island_event_handlers(container: DIContainer): event_queue = container.resolve(IIslandEventQueue) _handle_reset_agent_configuration_events(event_queue, container) + _handle_clear_simulation_data_events(event_queue, container) def _handle_reset_agent_configuration_events( @@ -15,3 +17,11 @@ def _handle_reset_agent_configuration_events( event_queue.subscribe( IslandEventTopic.RESET_AGENT_CONFIGURATION, container.resolve(reset_agent_configuration) ) + + +def _handle_clear_simulation_data_events(event_queue: IIslandEventQueue, container: DIContainer): + credentials_repository = container.resolve(ICredentialsRepository) + + event_queue.subscribe( + IslandEventTopic.CLEAR_SIMULATION_DATA, credentials_repository.remove_stolen_credentials + )