p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Merge pull request #1271 from guardicore/ransomware-logging 

Ransomware logging
This commit is contained in:
Mike Salvatore 2021-06-29 06:19:49 -04:00 committed by GitHub
parent 04b2ac6bd9 2ec020f276
commit 60c12127e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/monkey.py
View File

@ -469,5 +469,8 @@ class InfectionMonkey(object):
@staticmethod
def run_ransomware():
telemetry_messenger = TelemetryMessengerWrapper()
RansomewarePayload(WormConfiguration.ransomware, telemetry_messenger).run_payload()
try:
telemetry_messenger = TelemetryMessengerWrapper()
RansomewarePayload(WormConfiguration.ransomware, telemetry_messenger).run_payload()
except Exception as ex:
LOG.error(f"An unexpected error occurred while running the ransomware payload: {ex}")

3
monkey/infection_monkey/ransomware/ransomware_payload.py
View File

@ -37,6 +37,7 @@ class RansomewarePayload:
self._telemetry_messenger = telemetry_messenger
def run_payload(self):
LOG.info("Running ransomware payload")
file_list = self._find_files()
self._encrypt_files(file_list)
@ -52,10 +53,12 @@ class RansomewarePayload:
results = []
for filepath in file_list:
try:
LOG.debug(f"Encrypting {filepath}")
self._encryptor.encrypt_file_in_place(filepath)
self._add_extension(filepath)
self._send_telemetry(filepath, "")
except Exception as ex:
LOG.warning(f"Error encrypting {filepath}: {ex}")
self._send_telemetry(filepath, str(ex))
return results