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

Agent: Remove unused return value from RansomwarePayload._encrypt_files

This commit is contained in:
Mike Salvatore 2021-07-14 11:49:37 -04:00
parent 0be919b805
commit fd3cc46e55
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import logging
from pathlib import Path
from pprint import pformat
from typing import Callable, List, Optional, Tuple
from typing import Callable, List
from common.utils.file_utils import InvalidPath, expand_path
from infection_monkey.telemetry.file_encryption_telem import FileEncryptionTelem
@ -66,10 +66,9 @@ class RansomwarePayload:
LOG.info(f"Collecting files in {self._target_dir}")
return sorted(self._select_files(self._target_dir))
def _encrypt_files(self, file_list: List[Path]) -> List[Tuple[Path, Optional[Exception]]]:
def _encrypt_files(self, file_list: List[Path]):
LOG.info(f"Encrypting files in {self._target_dir}")
results = []
for filepath in file_list:
try:
LOG.debug(f"Encrypting {filepath}")
@ -79,8 +78,6 @@ class RansomwarePayload:
LOG.warning(f"Error encrypting {filepath}: {ex}")
self._send_telemetry(filepath, False, str(ex))
return results
def _send_telemetry(self, filepath: Path, success: bool, error: str):
encryption_attempt = FileEncryptionTelem(str(filepath), success, error)
self._telemetry_messenger.send_telemetry(encryption_attempt)