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

Agent: Create credential attribute even if we don't have credentials

This commit is contained in:
Ilija Lazoroski 2022-03-29 13:39:58 +02:00
parent 0a5fc84b4e
commit 638658178b
1 changed files with 18 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
monkey/infection_monkey/credential_store/aggregating_credentials_store.py
View File

@ -48,22 +48,27 @@ class AggregatingCredentialsStore(ICredentialsStore):
def _aggregate_credentials(self, credentials_to_aggr: Mapping): def _aggregate_credentials(self, credentials_to_aggr: Mapping):
for cred_attr, credentials_values in credentials_to_aggr.items(): for cred_attr, credentials_values in credentials_to_aggr.items():
if credentials_values: self._set_attribute(cred_attr, credentials_values)
self._set_attribute(cred_attr, credentials_values)
def _set_attribute(self, attribute_to_be_set, credentials_values): def _set_attribute(self, attribute_to_be_set, credentials_values):
if attribute_to_be_set not in self.stored_credentials: if attribute_to_be_set not in self.stored_credentials:
self.stored_credentials[attribute_to_be_set] = [] self.stored_credentials[attribute_to_be_set] = []
if isinstance(credentials_values[0], dict): if credentials_values:
self.stored_credentials.setdefault(attribute_to_be_set, []).extend(credentials_values) if isinstance(credentials_values[0], dict):
self.stored_credentials[attribute_to_be_set] = [ self.stored_credentials.setdefault(attribute_to_be_set, []).extend(
dict(s_c) credentials_values
for s_c in set( )
frozenset(d_c.items()) for d_c in self.stored_credentials[attribute_to_be_set] self.stored_credentials[attribute_to_be_set] = [
dict(s_c)
for s_c in set(
frozenset(d_c.items())
for d_c in self.stored_credentials[attribute_to_be_set]
)
]
else:
self.stored_credentials[attribute_to_be_set] = sorted(
list(
set(self.stored_credentials[attribute_to_be_set]).union(credentials_values)
)
) )
]
else:
self.stored_credentials[attribute_to_be_set] = sorted(
list(set(self.stored_credentials[attribute_to_be_set]).union(credentials_values))
)