Agent: Rename credentials_store -> propagation_credentials_repository

This commit is contained in:
Mike Salvatore 2022-08-16 08:17:04 -04:00
parent c3557caf1c
commit 1d79d98689
1 changed files with 16 additions and 9 deletions

View File

@ -203,12 +203,14 @@ class InfectionMonkey:
control_channel = ControlChannel(
self._control_client.server_address, GUID, self._control_client.proxies
)
credentials_store = AggregatingPropagationCredentialsRepository(control_channel)
propagation_credentials_repository = AggregatingPropagationCredentialsRepository(
control_channel
)
event_queue = PyPubSubEventQueue(Publisher())
InfectionMonkey._subscribe_events(event_queue, credentials_store)
InfectionMonkey._subscribe_events(event_queue, propagation_credentials_repository)
puppet = self._build_puppet(credentials_store, event_queue)
puppet = self._build_puppet(propagation_credentials_repository, event_queue)
victim_host_factory = self._build_victim_host_factory(local_network_interfaces)
@ -216,7 +218,7 @@ class InfectionMonkey:
ExploitInterceptingTelemetryMessenger(
self._telemetry_messenger, self._monkey_inbound_tunnel
),
credentials_store,
propagation_credentials_repository,
)
self._master = AutomatedMaster(
@ -226,16 +228,19 @@ class InfectionMonkey:
victim_host_factory,
control_channel,
local_network_interfaces,
credentials_store,
propagation_credentials_repository,
)
@staticmethod
def _subscribe_events(
event_queue: IEventQueue, credentials_store: IPropagationCredentialsRepository
event_queue: IEventQueue,
propagation_credentials_repository: IPropagationCredentialsRepository,
):
event_queue.subscribe_type(
CredentialsStolenEvent,
add_credentials_from_event_to_propagation_credentials_repository(credentials_store),
add_credentials_from_event_to_propagation_credentials_repository(
propagation_credentials_repository
),
)
@staticmethod
@ -247,7 +252,9 @@ class InfectionMonkey:
return local_network_interfaces
def _build_puppet(
self, credentials_store: IPropagationCredentialsRepository, event_queue: IEventQueue
self,
propagation_credentials_repository: IPropagationCredentialsRepository,
event_queue: IEventQueue,
) -> IPuppet:
puppet = Puppet()
@ -290,7 +297,7 @@ class InfectionMonkey:
)
zerologon_telemetry_messenger = CredentialsInterceptingTelemetryMessenger(
self._telemetry_messenger, credentials_store
self._telemetry_messenger, propagation_credentials_repository
)
zerologon_wrapper = ExploiterWrapper(zerologon_telemetry_messenger, agent_repository)
puppet.load_plugin(