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

agent: Use larger chunk size in RansomwarePayload 

The larger chunk size improves efficiency by reducing the number of
reads.
This commit is contained in:
Mike Salvatore 2021-06-23 09:40:14 -04:00
parent ae0dfec3cc
commit 9165737389
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,6 +15,7 @@ from infection_monkey.utils.environment import is_windows_os
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
EXTENSION = ".m0nk3y" EXTENSION = ".m0nk3y"
CHUNK_SIZE = 4096 * 24
class RansomewarePayload: class RansomewarePayload:
@ -25,7 +26,7 @@ class RansomewarePayload:
self._target_dir = config["windows_dir"] if is_windows_os() else config["linux_dir"] self._target_dir = config["windows_dir"] if is_windows_os() else config["linux_dir"]
self._valid_file_extensions_for_encryption = VALID_FILE_EXTENSIONS_FOR_ENCRYPTION.copy() self._valid_file_extensions_for_encryption = VALID_FILE_EXTENSIONS_FOR_ENCRYPTION.copy()
self._valid_file_extensions_for_encryption.discard(EXTENSION) self._valid_file_extensions_for_encryption.discard(EXTENSION)
self._encryptor = RansomwareBitflipEncryptor(EXTENSION) self._encryptor = RansomwareBitflipEncryptor(EXTENSION, chunk_size=CHUNK_SIZE)
def run_payload(self): def run_payload(self):
file_list = self._find_files() file_list = self._find_files()