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

Island, BB: Fix credential conversion to dict bugs

This commit is contained in:
vakaris_zilius 2022-09-06 11:51:16 +00:00
parent e5574240e9
commit 472ca382f1
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
View File

@ -1,7 +1,7 @@
import json import json
import logging import logging
import time import time
from typing import Sequence, Union from typing import List, Sequence, Union
from bson import json_util from bson import json_util
@ -59,9 +59,9 @@ class MonkeyIslandClient(object):
assert False assert False
@avoid_race_condition @avoid_race_condition
def _import_credentials(self, propagation_credentials: Credentials): def _import_credentials(self, propagation_credentials: List[Credentials]):
serialized_propagation_credentials = [ serialized_propagation_credentials = [
Credentials.dict(credentials) for credentials in propagation_credentials credentials.dict() for credentials in propagation_credentials
] ]
response = self.requests.put_json( response = self.requests.put_json(
"/api/propagation-credentials/configured-credentials", "/api/propagation-credentials/configured-credentials",

2
monkey/monkey_island/cc/repository/mongo_credentials_repository.py
View File

@ -68,7 +68,7 @@ class MongoCredentialsRepository(ICredentialsRepository):
def _save_credentials_to_collection(self, credentials: Sequence[Credentials], collection): def _save_credentials_to_collection(self, credentials: Sequence[Credentials], collection):
try: try:
for c in credentials: for c in credentials:
encrypted_credentials = self._encrypt_credentials_mapping(c.dict()) encrypted_credentials = self._encrypt_credentials_mapping(c.dict(simplify=True))
collection.insert_one(encrypted_credentials) collection.insert_one(encrypted_credentials)
except Exception as err: except Exception as err:
raise StorageError(err) raise StorageError(err)