p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Subscribe ping scan events

This commit is contained in:
vakarisz 2022-09-28 12:10:39 +03:00 committed by Kekoa Kaaikala
parent 799f08e383
commit ecb7ca0d8d
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
monkey/monkey_island/cc/setup/agent_event_handlers.py
View File

@ -1,11 +1,18 @@
from common import DIContainer
from common.agent_events import CredentialsStolenEvent
from common.agent_events import CredentialsStolenEvent, PingScanEvent
from common.event_queue import IAgentEventQueue
from monkey_island.cc.agent_event_handlers import (
handle_scan_data,
save_event_to_event_repository,
save_stolen_credentials_to_repository,
)
from monkey_island.cc.repository import IAgentEventRepository, ICredentialsRepository
from monkey_island.cc.repository import (
IAgentEventRepository,
IAgentRepository,
ICredentialsRepository,
IMachineRepository,
INodeRepository,
)
def setup_agent_event_handlers(container: DIContainer):
@ -22,3 +29,16 @@ def _subscribe_and_store_to_event_repository(container: DIContainer):
container.resolve(ICredentialsRepository)
)
agent_event_queue.subscribe_type(CredentialsStolenEvent, save_stolen_credentials_subscriber)
def _subscribe_ping_scan_event(container: DIContainer):
# Mypy don't handle cases where abstract class is passed as Type[...]
# https://github.com/python/mypy/issues/4717
agent_event_queue = container.resolve(IAgentEventQueue) # type: ignore
agent_repository = container.resolve(IAgentRepository) # type: ignore
machine_repository = container.resolve(IMachineRepository) # type: ignore
node_repository = container.resolve(INodeRepository) # type: ignore
handler = handle_scan_data(agent_repository, machine_repository, node_repository)
agent_event_queue.subscribe_type(PingScanEvent, handler)