Agent: replace get_monkey_depth with self.current_depth

This commit is contained in:
vakarisz 2022-03-09 12:56:41 +02:00
parent dc1a2ab1c1
commit 4e7e4a9eee
1 changed files with 2 additions and 3 deletions

View File

@ -7,7 +7,6 @@ from impacket.dcerpc.v5.rpcrt import DCERPCException
from common.utils.exploit_enum import ExploitType
from infection_monkey.exploit.HostExploiter import HostExploiter
from infection_monkey.exploit.tools.helpers import get_monkey_depth
from infection_monkey.exploit.tools.smb_tools import SmbTools
from infection_monkey.exploit.tools.wmi_tools import AccessDeniedException, WmiTools
from infection_monkey.i_puppet import ExploiterResultData
@ -97,13 +96,13 @@ class WmiExploiter(HostExploiter):
"dropper_path": remote_full_path
} + build_monkey_commandline(
self.host,
get_monkey_depth() - 1,
self.current_depth,
self.options["dropper_target_path_win_64"],
)
else:
cmdline = MONKEY_CMDLINE_WINDOWS % {
"monkey_path": remote_full_path
} + build_monkey_commandline(self.host, get_monkey_depth() - 1)
} + build_monkey_commandline(self.host, self.current_depth)
# execute the remote monkey
result = WmiTools.get_object(wmi_connection, "Win32_Process").Create(