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

Island: Add subscribe_to_topics on server setup

This commit is contained in:
Ilija Lazoroski 2022-09-12 15:59:58 +02:00
parent b834bfe7f7
commit 0ca13eb93d
3 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
monkey/monkey_island/cc/__init__.py
View File

@ -1 +1,2 @@
from .version import Version from .version import Version
from .subscribe_to_topics import subscribe_to_topics

4
monkey/monkey_island/cc/server_setup.py
View File

@ -10,7 +10,7 @@ import gevent.hub
import requests import requests
from gevent.pywsgi import WSGIServer from gevent.pywsgi import WSGIServer
from monkey_island.cc import Version from monkey_island.cc import Version, subscribe_to_topics
from monkey_island.cc.deployment import Deployment from monkey_island.cc.deployment import Deployment
from monkey_island.cc.server_utils.consts import ISLAND_PORT from monkey_island.cc.server_utils.consts import ISLAND_PORT
from monkey_island.cc.server_utils.island_logger import get_log_file_path from monkey_island.cc.server_utils.island_logger import get_log_file_path
@ -62,6 +62,8 @@ def run_monkey_island():
_initialize_mongodb_connection(config_options.start_mongodb, config_options.data_dir) _initialize_mongodb_connection(config_options.start_mongodb, config_options.data_dir)
_start_island_server(ip_addresses, island_args.setup_only, config_options, container) _start_island_server(ip_addresses, island_args.setup_only, config_options, container)
subscribe_to_topics(container)
def _extract_config(island_args: IslandCmdArgs) -> IslandConfigOptions: def _extract_config(island_args: IslandCmdArgs) -> IslandConfigOptions:
try: try:

11
monkey/monkey_island/cc/subscribe_to_topics.py Normal file
View File

@ -0,0 +1,11 @@
from common import DIContainer
from monkey_island.cc.event_queue import IslandEventTopic, PyPubSubIslandEventQueue
from monkey_island.cc.services.reset_agent_configuration import reset_agent_configuration
def subscribe_to_topics(container: DIContainer):
event_queue = container.resolve(PyPubSubIslandEventQueue)
event_queue.subscribe(
IslandEventTopic.RESET_AGENT_CONFIGURATION, container.resolve(reset_agent_configuration)
)