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

Agent: Handle unexpected errors in mssqlexec.py

This commit is contained in:
vakarisz 2022-03-15 14:10:35 +02:00
parent 1f327a1305
commit 43c8528409
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
monkey/infection_monkey/exploit/mssqlexec.py
View File

@ -73,20 +73,27 @@ class MSSQLExploiter(HostExploiter):
)
return self.exploit_result
# Create dir for payload
self.create_temp_dir()
self.create_empty_payload_file()
try:
# Create dir for payload
self.create_temp_dir()
self.create_empty_payload_file()
http_thread = self.start_monkey_server()
self.upload_monkey()
MSSQLExploiter._stop_monkey_server(http_thread)
http_thread = self.start_monkey_server()
self.upload_monkey()
MSSQLExploiter._stop_monkey_server(http_thread)
# Clear payload to pass in another command
self.create_empty_payload_file()
# Clear payload to pass in another command
self.create_empty_payload_file()
self.run_monkey()
self.run_monkey()
self.remove_temp_dir()
self.remove_temp_dir()
except Exception as e:
logger.error(
f"Unhandled exception occurred when trying "
f"to exploit MSSQL on host {self.host}: {e}"
)
return self.exploit_result
self.exploit_result.propagation_success = True
return self.exploit_result