p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 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): if not isinstance(event_data, dict):
raise TypeError("Event encryption only supported for dict") raise TypeError("Event encryption only supported for dict")
data = event_data.copy()
for field in fields: for field in fields:
event_data[ENCRYPTED_PREFIX + field] = str( data[ENCRYPTED_PREFIX + field] = str(
encrypt(json.dumps(event_data[field]).encode()), "utf-8" encrypt(json.dumps(event_data[field]).encode()), "utf-8"
) )
del event_data[field] del data[field]
return event_data return data
def decrypt_event( def decrypt_event(
@ -46,14 +47,15 @@ def decrypt_event(
if not isinstance(event_data, dict): if not isinstance(event_data, dict):
raise TypeError("Event decryption only supported for dict") raise TypeError("Event decryption only supported for dict")
data = event_data.copy()
for field in event_data.keys(): for field in event_data.keys():
if field.startswith("encrypted_"): 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") str(decrypt(event_data[field].encode()), "utf-8")
) )
del event_data[field] del data[field]
return event_data return data
class MongoAgentEventRepository(IAgentEventRepository): class MongoAgentEventRepository(IAgentEventRepository):