Agent: Make credential collector names consistent

This commit is contained in:
Mike Salvatore 2022-02-16 14:52:17 -05:00
parent c96f272919
commit dd1df14b8e
2 changed files with 3 additions and 3 deletions

View File

@ -34,11 +34,11 @@ class MockMaster(IMaster):
def _run_credential_collectors(self): def _run_credential_collectors(self):
logger.info("Running credential collectors") logger.info("Running credential collectors")
windows_credentials = self._puppet.run_credential_collector("MimikatzCredentialCollector") windows_credentials = self._puppet.run_credential_collector("MimikatzCollector")
if windows_credentials: if windows_credentials:
self._telemetry_messenger.send_telemetry(CredentialsTelem(windows_credentials)) self._telemetry_messenger.send_telemetry(CredentialsTelem(windows_credentials))
ssh_credentials = self._puppet.run_sys_info_collector("SSHCredentialCollector") ssh_credentials = self._puppet.run_sys_info_collector("SSHCollector")
if ssh_credentials: if ssh_credentials:
self._telemetry_messenger.send_telemetry(CredentialsTelem(ssh_credentials)) self._telemetry_messenger.send_telemetry(CredentialsTelem(ssh_credentials))

View File

@ -30,7 +30,7 @@ class MockPuppet(IPuppet):
def run_credential_collector(self, name: str, options: Dict) -> Sequence[Credentials]: def run_credential_collector(self, name: str, options: Dict) -> Sequence[Credentials]:
logger.debug(f"run_credential_collector({name})") logger.debug(f"run_credential_collector({name})")
if name == "SSHCredentialCollector": if name == "SSHCollector":
# TODO: Replace Passwords with SSHKeypair after it is implemented # TODO: Replace Passwords with SSHKeypair after it is implemented
ssh_credentials = Credentials( ssh_credentials = Credentials(
[Username("m0nk3y")], [Username("m0nk3y")],