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

agent: Fix typo RansomewarePayload -> RansomwarePayload

This commit is contained in:
Mike Salvatore 2021-06-29 12:00:49 -04:00
parent 32a0a41c21
commit d87b8ae4a7
3 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/monkey.py
View File

@ -19,7 +19,7 @@ from infection_monkey.network.HostFinger import HostFinger
from infection_monkey.network.network_scanner import NetworkScanner from infection_monkey.network.network_scanner import NetworkScanner
from infection_monkey.network.tools import get_interface_to_target, is_running_on_island from infection_monkey.network.tools import get_interface_to_target, is_running_on_island
from infection_monkey.post_breach.post_breach_handler import PostBreach from infection_monkey.post_breach.post_breach_handler import PostBreach
from infection_monkey.ransomware.ransomware_payload import RansomewarePayload from infection_monkey.ransomware.ransomware_payload import RansomwarePayload
from infection_monkey.system_info import SystemInfoCollector from infection_monkey.system_info import SystemInfoCollector
from infection_monkey.system_singleton import SystemSingleton from infection_monkey.system_singleton import SystemSingleton
from infection_monkey.telemetry.attack.t1106_telem import T1106Telem from infection_monkey.telemetry.attack.t1106_telem import T1106Telem
@ -476,7 +476,7 @@ class InfectionMonkey(object):
batching_telemetry_messenger = BatchingTelemetryMessenger(telemetry_messenger) batching_telemetry_messenger = BatchingTelemetryMessenger(telemetry_messenger)
try: try:
RansomewarePayload( RansomwarePayload(
WormConfiguration.ransomware, batching_telemetry_messenger WormConfiguration.ransomware, batching_telemetry_messenger
).run_payload() ).run_payload()
except Exception as ex: except Exception as ex:

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

@ -19,7 +19,7 @@ README_SRC = Path(__file__).parent / "ransomware_readme.txt"
README_DEST = "README.txt" README_DEST = "README.txt"
class RansomewarePayload: class RansomwarePayload:
def __init__(self, config: dict, telemetry_messenger: ITelemetryMessenger): def __init__(self, config: dict, telemetry_messenger: ITelemetryMessenger):
target_directories = config["directories"] target_directories = config["directories"]
LOG.info( LOG.info(

12
monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py
View File

@ -22,11 +22,7 @@ from tests.unit_tests.infection_monkey.ransomware.ransomware_target_files import
from tests.utils import hash_file, is_user_admin from tests.utils import hash_file, is_user_admin
from infection_monkey.ransomware import ransomware_payload as ransomware_payload_module from infection_monkey.ransomware import ransomware_payload as ransomware_payload_module
from infection_monkey.ransomware.ransomware_payload import ( from infection_monkey.ransomware.ransomware_payload import EXTENSION, README_DEST, RansomwarePayload
EXTENSION,
README_DEST,
RansomewarePayload,
)
def with_extension(filename): def with_extension(filename):
@ -43,7 +39,7 @@ def ransomware_payload_config(ransomware_target):
@pytest.fixture @pytest.fixture
def ransomware_payload(ransomware_payload_config, telemetry_messenger_spy): def ransomware_payload(ransomware_payload_config, telemetry_messenger_spy):
return RansomewarePayload(ransomware_payload_config, telemetry_messenger_spy) return RansomwarePayload(ransomware_payload_config, telemetry_messenger_spy)
def test_file_with_excluded_extension_not_encrypted(ransomware_target, ransomware_payload): def test_file_with_excluded_extension_not_encrypted(ransomware_target, ransomware_payload):
@ -163,7 +159,7 @@ def test_telemetry_failure(monkeypatch, ransomware_payload, telemetry_messenger_
def test_readme_false(ransomware_payload_config, ransomware_target, telemetry_messenger_spy): def test_readme_false(ransomware_payload_config, ransomware_target, telemetry_messenger_spy):
ransomware_payload_config["other_behaviors"]["readme"] = False ransomware_payload_config["other_behaviors"]["readme"] = False
ransomware_payload = RansomewarePayload(ransomware_payload_config, telemetry_messenger_spy) ransomware_payload = RansomwarePayload(ransomware_payload_config, telemetry_messenger_spy)
ransomware_payload.run_payload() ransomware_payload.run_payload()
assert not Path(ransomware_target / README_DEST).exists() assert not Path(ransomware_target / README_DEST).exists()
@ -171,7 +167,7 @@ def test_readme_false(ransomware_payload_config, ransomware_target, telemetry_me
def test_readme_true(ransomware_payload_config, ransomware_target, telemetry_messenger_spy): def test_readme_true(ransomware_payload_config, ransomware_target, telemetry_messenger_spy):
ransomware_payload_config["other_behaviors"]["readme"] = True ransomware_payload_config["other_behaviors"]["readme"] = True
ransomware_payload = RansomewarePayload(ransomware_payload_config, telemetry_messenger_spy) ransomware_payload = RansomwarePayload(ransomware_payload_config, telemetry_messenger_spy)
ransomware_payload.run_payload() ransomware_payload.run_payload()
assert Path(ransomware_target / README_DEST).exists() assert Path(ransomware_target / README_DEST).exists()