From bbda9340820d2e641076f8e53bcf3a70c47a1093 Mon Sep 17 00:00:00 2001 From: VakarisZ Date: Tue, 5 Oct 2021 16:04:02 +0300 Subject: [PATCH] Monkey: include credential key into info dict of InfoCollector class This change cleans up the code because the info collectors can just add credentials to the info dictionary without explicitly checking if the key already exists --- monkey/infection_monkey/system_info/__init__.py | 4 +--- monkey/infection_monkey/system_info/windows_info_collector.py | 3 +-- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/monkey/infection_monkey/system_info/__init__.py b/monkey/infection_monkey/system_info/__init__.py index 51da9b869..7d0659926 100644 --- a/monkey/infection_monkey/system_info/__init__.py +++ b/monkey/infection_monkey/system_info/__init__.py @@ -59,7 +59,7 @@ class InfoCollector(object): """ def __init__(self): - self.info = {} + self.info = {"credentials": {}} def get_info(self): # Collect all hardcoded @@ -96,8 +96,6 @@ class InfoCollector(object): return logger.debug("Harvesting creds if on an Azure machine") azure_collector = AzureCollector() - if "credentials" not in self.info: - self.info["credentials"] = {} azure_creds = azure_collector.extract_stored_credentials() for cred in azure_creds: username = cred[0] diff --git a/monkey/infection_monkey/system_info/windows_info_collector.py b/monkey/infection_monkey/system_info/windows_info_collector.py index ffc720dff..f3242922e 100644 --- a/monkey/infection_monkey/system_info/windows_info_collector.py +++ b/monkey/infection_monkey/system_info/windows_info_collector.py @@ -45,8 +45,7 @@ class WindowsInfoCollector(InfoCollector): try: credentials = MimikatzCredentialCollector.get_creds() if credentials: - if "credentials" in self.info: - self.info["credentials"].update(credentials) + self.info["credentials"].update(credentials) logger.info("Mimikatz info gathered successfully") else: logger.info("No mimikatz info was gathered")