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

Agent: Remove `current_server` option from custom_pba

This commit is contained in:
Ilija Lazoroski 2022-06-13 16:28:09 +02:00 committed by vakarisz
parent df116e4fb7
commit 02a30e6950
2 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/post_breach/custom_pba/custom_pba.py
View File

@ -37,9 +37,6 @@ class CustomPBA(PBA):
return super().run(options) return super().run(options)
def _set_options(self, options: Dict): def _set_options(self, options: Dict):
# Required for attack telemetry
self.current_server = options["current_server"]
if is_windows_os(): if is_windows_os():
# Add windows commands to PBA's # Add windows commands to PBA's
if options["windows_filename"]: if options["windows_filename"]:
@ -89,8 +86,8 @@ class CustomPBA(PBA):
self.telemetry_messenger.send_telemetry( self.telemetry_messenger.send_telemetry(
T1105Telem( T1105Telem(
status, status,
self.current_server.split(":")[0], self.cc_client.server_address.split(":")[0],
get_interface_to_target(self.current_server.split(":")[0]), get_interface_to_target(self.cc_client.server_address.split(":")[0]),
filename, filename,
) )
) )

8
monkey/tests/unit_tests/infection_monkey/post_breach/actions/test_users_custom_pba.py
View File

@ -44,8 +44,6 @@ def fake_custom_pba_linux_options():
"linux_filename": CUSTOM_LINUX_FILENAME, "linux_filename": CUSTOM_LINUX_FILENAME,
"windows_command": "", "windows_command": "",
"windows_filename": "", "windows_filename": "",
# Current server is used for attack telemetry
"current_server": CUSTOM_SERVER,
} }
@ -71,8 +69,6 @@ def fake_custom_pba_windows_options():
"linux_filename": "", "linux_filename": "",
"windows_command": CUSTOM_WINDOWS_CMD, "windows_command": CUSTOM_WINDOWS_CMD,
"windows_filename": CUSTOM_WINDOWS_FILENAME, "windows_filename": CUSTOM_WINDOWS_FILENAME,
# Current server is used for attack telemetry
"current_server": CUSTOM_SERVER,
} }
@ -94,8 +90,6 @@ def fake_options_files_only():
"linux_filename": CUSTOM_LINUX_FILENAME, "linux_filename": CUSTOM_LINUX_FILENAME,
"windows_command": "", "windows_command": "",
"windows_filename": CUSTOM_WINDOWS_FILENAME, "windows_filename": CUSTOM_WINDOWS_FILENAME,
# Current server is used for attack telemetry
"current_server": CUSTOM_SERVER,
} }
@ -113,8 +107,6 @@ def fake_options_commands_only():
"linux_filename": "", "linux_filename": "",
"windows_command": CUSTOM_WINDOWS_CMD, "windows_command": CUSTOM_WINDOWS_CMD,
"windows_filename": "", "windows_filename": "",
# Current server is used for attack telemetry
"current_server": CUSTOM_SERVER,
} }