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

Agent: Rename readme_utils to readme_dropper

This commit is contained in:
Mike Salvatore 2021-07-15 11:35:05 -04:00
parent 8ae41907ba
commit 7966703f63
3 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/infection_monkey/ransomware/ransomware_payload_builder.py
View File

@ -1,7 +1,7 @@
import logging import logging
from pprint import pformat from pprint import pformat
from infection_monkey.ransomware import ransomware_payload, readme_utils from infection_monkey.ransomware import ransomware_payload, readme_dropper
from infection_monkey.ransomware.file_selectors import ProductionSafeTargetFileSelector from infection_monkey.ransomware.file_selectors import ProductionSafeTargetFileSelector
from infection_monkey.ransomware.in_place_file_encryptor import InPlaceFileEncryptor from infection_monkey.ransomware.in_place_file_encryptor import InPlaceFileEncryptor
from infection_monkey.ransomware.ransomware_config import RansomwareConfig from infection_monkey.ransomware.ransomware_config import RansomwareConfig
@ -27,13 +27,14 @@ def build_ransomware_payload(config: dict):
file_encryptor = _build_file_encryptor() file_encryptor = _build_file_encryptor()
file_selector = _build_file_selector() file_selector = _build_file_selector()
leave_readme = _build_leave_readme()
telemetry_messenger = _build_telemetry_messenger() telemetry_messenger = _build_telemetry_messenger()
return RansomwarePayload( return RansomwarePayload(
ransomware_config, ransomware_config,
file_encryptor, file_encryptor,
file_selector, file_selector,
readme_utils.leave_readme, leave_readme,
telemetry_messenger, telemetry_messenger,
) )
@ -51,6 +52,10 @@ def _build_file_selector():
return ProductionSafeTargetFileSelector(targeted_file_extensions) return ProductionSafeTargetFileSelector(targeted_file_extensions)
def _build_leave_readme():
return readme_dropper.leave_readme
def _build_telemetry_messenger(): def _build_telemetry_messenger():
telemetry_messenger = LegacyTelemetryMessengerAdapter() telemetry_messenger = LegacyTelemetryMessengerAdapter()

0
monkey/infection_monkey/ransomware/readme_utils.py → monkey/infection_monkey/ransomware/readme_dropper.py
View File

2
monkey/tests/unit_tests/infection_monkey/ransomware/test_readme_utils.py → monkey/tests/unit_tests/infection_monkey/ransomware/test_readme_dropper.py
View File

@ -1,7 +1,7 @@
import pytest import pytest
from tests.utils import hash_file from tests.utils import hash_file
from infection_monkey.ransomware.readme_utils import leave_readme from infection_monkey.ransomware.readme_dropper import leave_readme
DEST_FILE = "README.TXT" DEST_FILE = "README.TXT"
README_HASH = "c98c24b677eff44860afea6f493bbaec5bb1c4cbb209c6fc2bbb47f66ff2ad31" README_HASH = "c98c24b677eff44860afea6f493bbaec5bb1c4cbb209c6fc2bbb47f66ff2ad31"