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

Island: Register a thread-safe IAgentEventQueue

This commit is contained in:
Mike Salvatore 2022-09-29 12:21:28 -04:00
parent 5d893d64cd
commit 91375cdff2
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/monkey_island/cc/services/initialize.py
View File

@ -15,7 +15,11 @@ from common.agent_event_serializers import (
register_common_agent_event_serializers, register_common_agent_event_serializers,
) )
from common.aws import AWSInstance from common.aws import AWSInstance
from common.event_queue import IAgentEventQueue, PyPubSubAgentEventQueue from common.event_queue import (
IAgentEventQueue,
LockingAgentEventQueueDecorator,
PyPubSubAgentEventQueue,
)
from common.utils.file_utils import get_binary_io_sha256_hash from common.utils.file_utils import get_binary_io_sha256_hash
from monkey_island.cc.event_queue import ( from monkey_island.cc.event_queue import (
IIslandEventQueue, IIslandEventQueue,
@ -104,11 +108,17 @@ def _register_conventions(container: DIContainer):
def _register_event_queues(container: DIContainer): def _register_event_queues(container: DIContainer):
container.register_instance(IAgentEventQueue, container.resolve(PyPubSubAgentEventQueue)) agent_event_queue = container.resolve(PyPubSubAgentEventQueue)
container.register_instance(IAgentEventQueue, _decorate_agent_event_queue(agent_event_queue))
island_event_queue = container.resolve(PyPubSubIslandEventQueue) island_event_queue = container.resolve(PyPubSubIslandEventQueue)
container.register_instance(IIslandEventQueue, _decorate_island_event_queue(island_event_queue)) container.register_instance(IIslandEventQueue, _decorate_island_event_queue(island_event_queue))
def _decorate_agent_event_queue(agent_event_queue: IAgentEventQueue):
return LockingAgentEventQueueDecorator(agent_event_queue)
def _decorate_island_event_queue(island_event_queue: IIslandEventQueue): def _decorate_island_event_queue(island_event_queue: IIslandEventQueue):
return LockingIslandEventQueueDecorator(island_event_queue) return LockingIslandEventQueueDecorator(island_event_queue)