p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 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:
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(
f"Windows dir configured for encryption is \"{target_directories['windows_dir']}\""
)
@ -44,9 +47,11 @@ class RansomwarePayload:
self._telemetry_messenger = telemetry_messenger
def run_payload(self):
LOG.info("Running ransomware payload")
file_list = self._find_files()
self._encrypt_files(file_list)
if self.should_encrypt:
LOG.info("Running ransomware payload")
file_list = self._find_files()
self._encrypt_files(file_list)
self._leave_readme()
def _find_files(self) -> List[Path]: