Agent: explicitly specify some timeouts in zerologon exploiter

monkey/infection_monkey/exploit/zerologon_utils/dump_secrets.py

@@ -56,6 +56,7 @@ from impacket.examples.secretsdump import (
 )
 from impacket.smbconnection import SMBConnection
 
+from common.common_consts.timeouts import LONG_REQUEST_TIMEOUT
 from infection_monkey.utils.capture_output import StdoutCapture
 
 logger = logging.getLogger(__name__)
@@ -96,7 +97,9 @@ class DumpSecrets:
             self.__lmhash, self.__nthash = options.hashes.split(":")
 
     def connect(self):
-        self.__smb_connection = SMBConnection(self.__remote_name, self.__remote_host)
+        self.__smb_connection = SMBConnection(
+            self.__remote_name, self.__remote_host, timeout=LONG_REQUEST_TIMEOUT
+        )
         self.__smb_connection.login(
             self.__username,
             self.__password,

monkey/infection_monkey/exploit/zerologon_utils/remote_shell.py

@@ -71,6 +71,7 @@ class RemoteShell(cmd.Cmd):
         self.__secrets_dir = secrets_dir
 
         # We don't wanna deal with timeouts from now on.
+        # TODO are we sure we don't need timeout anymore?
         if self.__transferClient is not None:
             self.__transferClient.setTimeout(100000)
             self.do_cd("\\")

monkey/infection_monkey/exploit/zerologon_utils/wmiexec.py

@@ -51,6 +51,7 @@ from impacket.dcerpc.v5.dcomrt import DCOMConnection
 from impacket.dcerpc.v5.dtypes import NULL
 from impacket.smbconnection import SMBConnection
 
+from common.common_consts.timeouts import LONG_REQUEST_TIMEOUT
 from infection_monkey.exploit.zerologon_utils.remote_shell import RemoteShell
 
 logger = logging.getLogger(__name__)
@@ -74,7 +75,7 @@ class Wmiexec:
         self.shell = None
 
     def connect(self):
-        self.smbConnection = SMBConnection(self.__ip, self.__ip)
+        self.smbConnection = SMBConnection(self.__ip, self.__ip, timeout=LONG_REQUEST_TIMEOUT)
         self.smbConnection.login(
             user=self.__username,
             password=self.__password,