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

agent: Modify ransomware payload to work with modified ransomware config schema

This commit is contained in:
Shreya 2021-06-29 16:17:28 +05:30 committed by Mike Salvatore
parent 13a94804b4
commit 4035d9d213
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -21,7 +21,10 @@ README_DEST = "README.txt"
class RansomwarePayload: class RansomwarePayload:
def __init__(self, config: dict, telemetry_messenger: ITelemetryMessenger): def __init__(self, config: dict, telemetry_messenger: ITelemetryMessenger):
target_directories = config["directories"] self.should_encrypt = config["encryption"]["should_encrypt"]
LOG.info(f"Encryption routine for ransomware simulation enabled: {self.should_encrypt}")
target_directories = config["encryption"]["directories"]
LOG.info( LOG.info(
f"Windows dir configured for encryption is \"{target_directories['windows_dir']}\"" f"Windows dir configured for encryption is \"{target_directories['windows_dir']}\""
) )
@ -44,9 +47,11 @@ class RansomwarePayload:
self._telemetry_messenger = telemetry_messenger self._telemetry_messenger = telemetry_messenger
def run_payload(self): def run_payload(self):
LOG.info("Running ransomware payload") if self.should_encrypt:
file_list = self._find_files() LOG.info("Running ransomware payload")
self._encrypt_files(file_list) file_list = self._find_files()
self._encrypt_files(file_list)
self._leave_readme() self._leave_readme()
def _find_files(self) -> List[Path]: def _find_files(self) -> List[Path]: