From bb554d923d62e56ee24b019f5ee36338066b9259 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 8 Jul 2021 12:15:37 -0400 Subject: [PATCH] Agent: Rename _valid_file_extensions... -> _targeted_file_extensions --- monkey/infection_monkey/ransomware/ransomware_payload.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/monkey/infection_monkey/ransomware/ransomware_payload.py b/monkey/infection_monkey/ransomware/ransomware_payload.py index 0e5cf813e..6daa45b87 100644 --- a/monkey/infection_monkey/ransomware/ransomware_payload.py +++ b/monkey/infection_monkey/ransomware/ransomware_payload.py @@ -34,8 +34,8 @@ class RansomwarePayload: self._target_dir = RansomwarePayload.get_target_dir(config) self._new_file_extension = EXTENSION - self._valid_file_extensions_for_encryption = TARGETED_FILE_EXTENSIONS.copy() - self._valid_file_extensions_for_encryption.discard(self._new_file_extension) + self._targeted_file_extensions = TARGETED_FILE_EXTENSIONS.copy() + self._targeted_file_extensions.discard(self._new_file_extension) self._encryptor = BitflipEncryptor(chunk_size=CHUNK_SIZE) self._copy_file = copy_file @@ -68,9 +68,7 @@ class RansomwarePayload: if not self._target_dir: return [] - return select_production_safe_target_files( - self._target_dir, self._valid_file_extensions_for_encryption - ) + return select_production_safe_target_files(self._target_dir, self._targeted_file_extensions) def _encrypt_files(self, file_list: List[Path]) -> List[Tuple[Path, Optional[Exception]]]: LOG.info(f"Encrypting files in {self._target_dir}")