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

Agent: Improve whitespace and formatting in _self_delete_windows()

This commit is contained in:
Mike Salvatore 2022-04-20 11:08:40 -04:00
parent 838848bc3a
commit 6d51f17f29
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/infection_monkey/monkey.py
View File

@ -429,8 +429,8 @@ class InfectionMonkey:
@staticmethod @staticmethod
def _self_delete_windows(): def _self_delete_windows():
delay_delete_cmd = InfectionMonkey._build_windows_delete_command() delay_delete_cmd = InfectionMonkey._build_windows_delete_command()
startupinfo = InfectionMonkey._get_startup_info() startupinfo = InfectionMonkey._get_startup_info()
subprocess.Popen( subprocess.Popen(
delay_delete_cmd, delay_delete_cmd,
stdin=None, stdin=None,
@ -442,14 +442,15 @@ class InfectionMonkey:
@staticmethod @staticmethod
def _build_windows_delete_command() -> str: def _build_windows_delete_command() -> str:
# Time for delay deleting monkey executable
delay_seconds = 5
monkey_pid = os.getpid() monkey_pid = os.getpid()
monkey_file_path = sys.executable monkey_file_path = sys.executable
# Time for delay deleting monkey executable
delay_seconds = 5
# Command that returns 1 if the process is running and 0 otherwise # Command that returns 1 if the process is running and 0 otherwise
check_running_monkey_cmd = f'tasklist /fi "PID eq {monkey_pid}" ^| find /C "{monkey_pid}"' check_running_monkey_cmd = f'tasklist /fi "PID eq {monkey_pid}" ^| find /C "{monkey_pid}"'
delete_file_and_exit_cmd = f"del /f /q {monkey_file_path} & exit" delete_file_and_exit_cmd = f"del /f /q {monkey_file_path} & exit"
# Command that checks for running monkey process 20 times # Command that checks for running monkey process 20 times
# If the monkey is running it sleeps for 'delay_seconds' # If the monkey is running it sleeps for 'delay_seconds'
# If the monkey is not running it deletes the executable and exits the loop # If the monkey is not running it deletes the executable and exits the loop