From b3e9bbe05f26aa6898c4ea3a3e5859d6e6535b18 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Wed, 14 Sep 2022 13:54:38 -0400
Subject: [PATCH] Island: Subscribe save_stolen_credentials_to_repository()
---
.../monkey_island/cc/setup/agent_event_handlers.py | 13 +++++++++++--
1 file changed, 11 insertions(+), 2 deletions(-)
diff --git a/monkey/monkey_island/cc/setup/agent_event_handlers.py b/monkey/monkey_island/cc/setup/agent_event_handlers.py
index 9fd0a2ea7..af37e6740 100644
--- a/monkey/monkey_island/cc/setup/agent_event_handlers.py
+++ b/monkey/monkey_island/cc/setup/agent_event_handlers.py
@@ -1,7 +1,11 @@
from common import DIContainer
from common.event_queue import IAgentEventQueue
-from monkey_island.cc.agent_event_subscribers import save_event_to_event_repository
-from monkey_island.cc.repository import IEventRepository
+from common.events import CredentialsStolenEvent
+from monkey_island.cc.agent_event_subscribers import (
+ save_event_to_event_repository,
+ save_stolen_credentials_to_repository,
+)
+from monkey_island.cc.repository import ICredentialsRepository, IEventRepository
def setup_agent_event_handlers(container: DIContainer):
@@ -14,3 +18,8 @@ def _subscribe_and_store_to_event_repository(container: DIContainer):
event_repository = container.resolve(IEventRepository)
save_event_subscriber = save_event_to_event_repository(event_repository)
agent_event_queue.subscribe_all_events(save_event_subscriber)
+
+ save_stolen_credentials_subscriber = save_stolen_credentials_to_repository(
+ container.resolve(ICredentialsRepository)
+ )
+ agent_event_queue.subscribe_type(CredentialsStolenEvent, save_stolen_credentials_subscriber)