p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Agent: Rename InPlaceEncryptor -> InPlaceFileEncryptor

This commit is contained in:
Mike Salvatore 2021-07-14 08:38:51 -04:00
parent 39171f0950
commit 0cb975a592
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/infection_monkey/ransomware/in_place_encryptor.py → monkey/infection_monkey/ransomware/in_place_file_encryptor.py
View File

@ -5,7 +5,7 @@ from typing import Callable
FILE_EXTENSION_REGEX = re.compile(r"^\.[^\\/]+$")
class InPlaceEncryptor:
class InPlaceFileEncryptor:
def __init__(
self,
encrypt_bytes: Callable[[bytes], bytes],

18
monkey/tests/unit_tests/infection_monkey/ransomware/test_in_place_encryptor.py → monkey/tests/unit_tests/infection_monkey/ransomware/test_in_place_file_encryptor.py
View File

@ -11,7 +11,7 @@ from tests.unit_tests.infection_monkey.ransomware.ransomware_target_files import
)
from tests.utils import hash_file
from infection_monkey.ransomware.in_place_encryptor import InPlaceEncryptor
from infection_monkey.ransomware.in_place_file_encryptor import InPlaceFileEncryptor
from infection_monkey.utils.bit_manipulators import flip_bits
EXTENSION = ".m0nk3y"
@ -22,14 +22,14 @@ def with_extension(filename):
@pytest.fixture(scope="module")
def in_place_bitflip_encryptor():
return InPlaceEncryptor(encrypt_bytes=flip_bits, chunk_size=64)
def in_place_bitflip_file_encryptor():
return InPlaceFileEncryptor(encrypt_bytes=flip_bits, chunk_size=64)
@pytest.mark.parametrize("invalid_extension", ["no_dot", ".has/slash", ".has\\slash"])
def test_invalid_file_extension(invalid_extension):
with pytest.raises(ValueError):
InPlaceEncryptor(encrypt_bytes=None, new_file_extension=invalid_extension)
InPlaceFileEncryptor(encrypt_bytes=None, new_file_extension=invalid_extension)
@pytest.mark.parametrize(
@ -40,22 +40,22 @@ def test_invalid_file_extension(invalid_extension):
],
)
def test_file_encrypted(
in_place_bitflip_encryptor, ransomware_target, file_name, cleartext_hash, encrypted_hash
in_place_bitflip_file_encryptor, ransomware_target, file_name, cleartext_hash, encrypted_hash
):
test_keyboard = ransomware_target / file_name
assert hash_file(test_keyboard) == cleartext_hash
in_place_bitflip_encryptor(test_keyboard)
in_place_bitflip_file_encryptor(test_keyboard)
assert hash_file(test_keyboard) == encrypted_hash
def test_file_encrypted_in_place(in_place_bitflip_encryptor, ransomware_target):
def test_file_encrypted_in_place(in_place_bitflip_file_encryptor, ransomware_target):
test_keyboard = ransomware_target / TEST_KEYBOARD_TXT
expected_inode = os.stat(test_keyboard).st_ino
in_place_bitflip_encryptor(test_keyboard)
in_place_bitflip_file_encryptor(test_keyboard)
actual_inode = os.stat(test_keyboard).st_ino
assert expected_inode == actual_inode
@ -64,7 +64,7 @@ def test_file_encrypted_in_place(in_place_bitflip_encryptor, ransomware_target):
def test_encrypted_file_has_new_extension(ransomware_target):
test_keyboard = ransomware_target / TEST_KEYBOARD_TXT
encrypted_test_keyboard = ransomware_target / with_extension(TEST_KEYBOARD_TXT)
encryptor = InPlaceEncryptor(encrypt_bytes=flip_bits, new_file_extension=EXTENSION)
encryptor = InPlaceFileEncryptor(encrypt_bytes=flip_bits, new_file_extension=EXTENSION)
encryptor(test_keyboard)