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

Agent: Rename function for returning ExploiterResultData

This commit is contained in:
Shreya Malviya 2022-02-23 17:11:53 +05:30
parent 6cdb86aa4b
commit 4ecc5283e5
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/infection_monkey/exploit/HostExploiter.py
View File

@ -127,7 +127,7 @@ class HostExploiter:
powershell = True if "powershell" in cmd.lower() else False powershell = True if "powershell" in cmd.lower() else False
self.exploit_info["executed_cmds"].append({"cmd": cmd, "powershell": powershell}) self.exploit_info["executed_cmds"].append({"cmd": cmd, "powershell": powershell})
def get_exploit_result_data(self) -> ExploiterResultData: def return_exploit_result_data(self) -> ExploiterResultData:
return ExploiterResultData( return ExploiterResultData(
self.exploit_result["exploitation_success"], self.exploit_result["exploitation_success"],
self.exploit_result["propagation_success"], self.exploit_result["propagation_success"],

16
monkey/infection_monkey/exploit/sshexec.py
View File

@ -114,7 +114,7 @@ class SSHExploiter(HostExploiter):
self.exploit_result["error_message"] = f"SSH port is closed on {self.host}, skipping" self.exploit_result["error_message"] = f"SSH port is closed on {self.host}, skipping"
logger.info(self.exploit_result["error_message"]) logger.info(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()
try: try:
ssh = self.exploit_with_ssh_keys(port) ssh = self.exploit_with_ssh_keys(port)
@ -129,7 +129,7 @@ class SSHExploiter(HostExploiter):
self.exploit_result["propagation_success"] = False self.exploit_result["propagation_success"] = False
logger.debug(self.exploit_result["error_message"]) logger.debug(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()
if not self.host.os.get("type"): if not self.host.os.get("type"):
try: try:
@ -143,7 +143,7 @@ class SSHExploiter(HostExploiter):
if not uname_os: if not uname_os:
logger.error(self.exploit_result["error_message"]) logger.error(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()
except Exception as exc: except Exception as exc:
self.exploit_result["propagation_success"] = False self.exploit_result["propagation_success"] = False
self.exploit_result[ self.exploit_result[
@ -151,7 +151,7 @@ class SSHExploiter(HostExploiter):
] = f"Error running uname os command on victim {self.host}: ({exc})" ] = f"Error running uname os command on victim {self.host}: ({exc})"
logger.debug(self.exploit_result["error_message"]) logger.debug(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()
if not self.host.os.get("machine"): if not self.host.os.get("machine"):
try: try:
@ -174,7 +174,7 @@ class SSHExploiter(HostExploiter):
] = f"Can't find suitable monkey executable for host {self.host}" ] = f"Can't find suitable monkey executable for host {self.host}"
logger.info(self.exploit_result["error_message"]) logger.info(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()
try: try:
ftp = ssh.open_sftp() ftp = ssh.open_sftp()
@ -211,7 +211,7 @@ class SSHExploiter(HostExploiter):
) )
) )
if status == ScanStatus.SCANNED: if status == ScanStatus.SCANNED:
return self.get_exploit_result_data() return self.return_exploit_result_data()
try: try:
cmdline = "%s %s" % (self.options["dropper_target_path_linux"], MONKEY_ARG) cmdline = "%s %s" % (self.options["dropper_target_path_linux"], MONKEY_ARG)
@ -230,7 +230,7 @@ class SSHExploiter(HostExploiter):
ssh.close() ssh.close()
self.add_executed_cmd(cmdline) self.add_executed_cmd(cmdline)
return self.get_exploit_result_data() return self.return_exploit_result_data()
except Exception as exc: except Exception as exc:
self.exploit_result["propagation_success"] = False self.exploit_result["propagation_success"] = False
@ -239,4 +239,4 @@ class SSHExploiter(HostExploiter):
] = f"Error running monkey on victim {self.host}: ({exc})" ] = f"Error running monkey on victim {self.host}: ({exc})"
logger.error(self.exploit_result["error_message"]) logger.error(self.exploit_result["error_message"])
return self.get_exploit_result_data() return self.return_exploit_result_data()