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

Agent: Extract method _copy_file() from _leave_readme() 

Reworks the logic in _leave_readme() to reduce indenting and improve
clarity and extracts the logic to copy the file into _copy_readme_file()
This commit is contained in:
Mike Salvatore 2021-07-08 12:04:50 -04:00
parent 7454ee72b2
commit 92c5c3b682
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -97,15 +97,21 @@ class RansomwarePayload:
self._telemetry_messenger.send_telemetry(encryption_attempt) self._telemetry_messenger.send_telemetry(encryption_attempt)
def _leave_readme(self): def _leave_readme(self):
if self._readme_enabled: if not self._readme_enabled:
readme_dest_path = self._target_dir / README_DEST return
if readme_dest_path.exists():
LOG.warning(f"{readme_dest_path} already exists, not leaving a new README.txt")
return
LOG.info(f"Leaving a ransomware README file at {readme_dest_path}") readme_dest_path = self._target_dir / README_DEST
try: if readme_dest_path.exists():
self._copy_file(README_SRC, readme_dest_path) LOG.warning(f"{readme_dest_path} already exists, not leaving a new README.txt")
except Exception as ex: return
LOG.warning(f"An error occurred while attempting to leave a README.txt file: {ex}")
self._copy_readme_file(readme_dest_path)
def _copy_readme_file(self, dest: Path):
LOG.info(f"Leaving a ransomware README file at {dest}")
try:
self._copy_file(README_SRC, dest)
except Exception as ex:
LOG.warning(f"An error occurred while attempting to leave a README.txt file: {ex}")