Agent: Handle unexpected errors in mssqlexec.py

monkey/infection_monkey/exploit/mssqlexec.py

@@ -73,6 +73,7 @@ class MSSQLExploiter(HostExploiter):
             )
             return self.exploit_result
 
+        try:
             # Create dir for payload
             self.create_temp_dir()
             self.create_empty_payload_file()
@@ -87,6 +88,12 @@ class MSSQLExploiter(HostExploiter):
             self.run_monkey()
 
             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