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

Agent: Make agent target paths a Path objects

This commit is contained in:
Ilija Lazoroski 2022-06-01 17:09:56 +02:00
parent e4a2a54c71
commit 09958c21c6
3 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/dropper.py
View File

@ -20,8 +20,11 @@ from infection_monkey.utils.file_utils import mark_file_for_deletion_on_windows
if "win32" == sys.platform: if "win32" == sys.platform:
from win32process import DETACHED_PROCESS from win32process import DETACHED_PROCESS
DATE_REFERENCE_PATH_WINDOWS = os.path.expandvars(WindowsPath(r"%windir%\system32\kernel32.dll"))
else: else:
DETACHED_PROCESS = 0 DETACHED_PROCESS = 0
DATE_REFERENCE_PATH_LINUX = PosixPath("/bin/sh")
# Linux doesn't have WindowsError # Linux doesn't have WindowsError
try: try:
@ -34,8 +37,6 @@ except NameError:
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
MOVEFILE_DELAY_UNTIL_REBOOT = 4 MOVEFILE_DELAY_UNTIL_REBOOT = 4
DATE_REFERENCE_PATH_WINDOWS = WindowsPath(r"%windir%\system32\kernel32.dll")
DATE_REFERENCE_PATH_LINUX = PosixPath("/bin/sh")
class MonkeyDrops(object): class MonkeyDrops(object):
@ -112,7 +113,7 @@ class MonkeyDrops(object):
return False return False
if sys.platform == "win32": if sys.platform == "win32":
dropper_date_reference_path = os.path.expandvars(DATE_REFERENCE_PATH_WINDOWS) dropper_date_reference_path = DATE_REFERENCE_PATH_WINDOWS
else: else:
dropper_date_reference_path = DATE_REFERENCE_PATH_LINUX dropper_date_reference_path = DATE_REFERENCE_PATH_LINUX

4
monkey/infection_monkey/exploit/tools/helpers.py
View File

@ -10,8 +10,8 @@ logger = logging.getLogger(__name__)
RAND_SUFFIX_LEN = 8 RAND_SUFFIX_LEN = 8
# Where to upload agent binaries on victims # Where to upload agent binaries on victims
AGENT_BINARY_PATH_LINUX = "/tmp/monkey" AGENT_BINARY_PATH_LINUX = PurePosixPath("/tmp/monkey")
AGENT_BINARY_PATH_WIN64 = r"C:\Windows\temp\monkey64.exe" AGENT_BINARY_PATH_WIN64 = PureWindowsPath(r"C:\Windows\temp\monkey64.exe")
def get_agent_dst_path(host: VictimHost) -> PurePath: def get_agent_dst_path(host: VictimHost) -> PurePath:

7
monkey/tests/unit_tests/infection_monkey/exploit/test_powershell.py
View File

@ -5,7 +5,7 @@ from unittest.mock import MagicMock
import pytest import pytest
from infection_monkey.exploit import powershell from infection_monkey.exploit import powershell
from infection_monkey.exploit.tools.helpers import AGENT_BINARY_PATH_WIN64, RAND_SUFFIX_LEN from infection_monkey.exploit.tools.helpers import AGENT_BINARY_PATH_WIN64
from infection_monkey.model.host import VictimHost from infection_monkey.model.host import VictimHost
# Use the path_win32api_get_user_name fixture for all tests in this module # Use the path_win32api_get_user_name fixture for all tests in this module
@ -113,10 +113,7 @@ def test_successful_copy(monkeypatch, powershell_exploiter, powershell_arguments
exploit_result = powershell_exploiter.exploit_host(**powershell_arguments) exploit_result = powershell_exploiter.exploit_host(**powershell_arguments)
# Check if the copied agent name has randomness of 8 plus dash assert AGENT_BINARY_PATH_WIN64.stem in str(mock_client.return_value.copy_file.call_args[0][1])
assert len(str(AGENT_BINARY_PATH_WIN64)) + RAND_SUFFIX_LEN + 1 == len(
str(mock_client.return_value.copy_file.call_args[0][1])
)
assert exploit_result.exploitation_success assert exploit_result.exploitation_success