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

agent: ransomware: Incorporate changes into stub

This commit is contained in:
Shreya 2021-06-22 19:42:50 +05:30
parent 48967a2e5b
commit c6da73ad79
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,8 @@
import logging import logging
from infection_monkey.ransomware.utils import get_files_to_encrypt
from infection_monkey.utils.environment import is_windows_os
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -17,7 +20,12 @@ class RansomewarePayload:
self._encrypt_files(file_list) self._encrypt_files(file_list)
def _find_files(self): def _find_files(self):
return [] dir_path = (
self.config["windows_dir_ransom"]
if is_windows_os()
else self.config["linux_dir_ransom"]
)
return get_files_to_encrypt(dir_path)
def _encrypt_files(self, file_list): def _encrypt_files(self, file_list):
for file in file_list: for file in file_list: