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

Island: Copy dict when encrypting event data

This commit is contained in:
Kekoa Kaaikala 2022-09-19 13:19:48 +00:00
parent 7634e00737
commit 2e63f47606
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/monkey_island/cc/repository/mongo_agent_event_repository.py
View File

@ -31,13 +31,14 @@ def encrypt_event(
if not isinstance(event_data, dict):
raise TypeError("Event encryption only supported for dict")
data = event_data.copy()
for field in fields:
event_data[ENCRYPTED_PREFIX + field] = str(
data[ENCRYPTED_PREFIX + field] = str(
encrypt(json.dumps(event_data[field]).encode()), "utf-8"
)
del event_data[field]
del data[field]
return event_data
return data
def decrypt_event(
@ -46,14 +47,15 @@ def decrypt_event(
if not isinstance(event_data, dict):
raise TypeError("Event decryption only supported for dict")
data = event_data.copy()
for field in event_data.keys():
if field.startswith("encrypted_"):
event_data[field[len(ENCRYPTED_PREFIX) :]] = json.loads(
data[field[len(ENCRYPTED_PREFIX) :]] = json.loads(
str(decrypt(event_data[field].encode()), "utf-8")
)
del event_data[field]
del data[field]
return event_data
return data
class MongoAgentEventRepository(IAgentEventRepository):