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

Agent: Rename event_queue to agent_event_queue

This commit is contained in:
Ilija Lazoroski 2022-09-27 13:10:07 +02:00 committed by Shreya Malviya
parent 56ead43c11
commit 89397d8cbd
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
monkey/infection_monkey/monkey.py
View File

@ -272,14 +272,14 @@ class InfectionMonkey:
self._control_channel self._control_channel
) )
event_queue = PyPubSubAgentEventQueue(Publisher()) agent_event_queue = PyPubSubAgentEventQueue(Publisher())
self._subscribe_events( self._subscribe_events(
event_queue, agent_event_queue,
propagation_credentials_repository, propagation_credentials_repository,
self._agent_event_serializer_registry, self._agent_event_serializer_registry,
) )
puppet = self._build_puppet(event_queue) puppet = self._build_puppet(agent_event_queue)
victim_host_factory = self._build_victim_host_factory(local_network_interfaces) victim_host_factory = self._build_victim_host_factory(local_network_interfaces)
@ -300,34 +300,34 @@ class InfectionMonkey:
def _subscribe_events( def _subscribe_events(
self, self,
event_queue: IAgentEventQueue, agent_event_queue: IAgentEventQueue,
propagation_credentials_repository: IPropagationCredentialsRepository, propagation_credentials_repository: IPropagationCredentialsRepository,
agent_event_serializer_registry: AgentEventSerializerRegistry, agent_event_serializer_registry: AgentEventSerializerRegistry,
): ):
event_queue.subscribe_type( agent_event_queue.subscribe_type(
CredentialsStolenEvent, CredentialsStolenEvent,
add_credentials_from_event_to_propagation_credentials_repository( add_credentials_from_event_to_propagation_credentials_repository(
propagation_credentials_repository propagation_credentials_repository
), ),
) )
event_queue.subscribe_all_events( agent_event_queue.subscribe_all_events(
AgentEventForwarder(self._island_api_client, agent_event_serializer_registry).send_event AgentEventForwarder(self._island_api_client, agent_event_serializer_registry).send_event
) )
def _build_puppet( def _build_puppet(
self, self,
event_queue: IAgentEventQueue, agent_event_queue: IAgentEventQueue,
) -> IPuppet: ) -> IPuppet:
puppet = Puppet() puppet = Puppet()
puppet.load_plugin( puppet.load_plugin(
"MimikatzCollector", "MimikatzCollector",
MimikatzCredentialCollector(event_queue), MimikatzCredentialCollector(agent_event_queue),
PluginType.CREDENTIAL_COLLECTOR, PluginType.CREDENTIAL_COLLECTOR,
) )
puppet.load_plugin( puppet.load_plugin(
"SSHCollector", "SSHCollector",
SSHCredentialCollector(self._telemetry_messenger, event_queue), SSHCredentialCollector(self._telemetry_messenger, agent_event_queue),
PluginType.CREDENTIAL_COLLECTOR, PluginType.CREDENTIAL_COLLECTOR,
) )
@ -341,7 +341,7 @@ class InfectionMonkey:
island_api_client=self._island_api_client, island_api_client=self._island_api_client,
) )
exploit_wrapper = ExploiterWrapper( exploit_wrapper = ExploiterWrapper(
self._telemetry_messenger, event_queue, agent_binary_repository self._telemetry_messenger, agent_event_queue, agent_binary_repository
) )
puppet.load_plugin( puppet.load_plugin(