[mssql] less code

Signed-off-by: XiaoliChan <30458572+XiaoliChan@users.noreply.github.com>

nxc/protocols/mssql.py

@@ -94,9 +94,6 @@ def enum_host_info(self):
         if self.domain is None:
             self.domain = ""
 
-        with contextlib.suppress(Exception):
-            self.conn.disconnect()
-
     def print_host_info(self):
         self.logger.display(f"{self.server_os} (name:{self.hostname}) (domain:{self.domain})")
         return True
@@ -116,6 +113,17 @@ def create_conn_obj(self):
             self.conn.socket = sock
             return True
 
+    def reconnect_mssql(func):
+        def wrapper(self, *args, **kwargs):
+            with contextlib.suppress(Exception):
+                self.conn.disconnect()
+            # When using ccache file, we must need to set target host to hostname when creating connection object.
+            if self.kerberos:
+                self.host = self.hostname
+            self.create_conn_obj()
+            return func(self, *args, **kwargs)
+        return wrapper
+
     def check_if_admin(self):
         self.admin_privs = False
         try:
@@ -127,6 +135,7 @@ def check_if_admin(self):
             if is_admin:
                 self.admin_privs = True
 
+    @reconnect_mssql
     def kerberos_login(
         self,
         domain,
@@ -137,12 +146,6 @@ def kerberos_login(
         kdcHost="",
         useCache=False,
     ):
-        with contextlib.suppress(Exception):
-            self.conn.disconnect()
-        # When using ccache file, we must need to set target host to hostname when creating connection object.
-        self.host = self.hostname
-        self.create_conn_obj()
-
         kerb_pass = next(s for s in [self.nthash, password, aesKey] if s) if not all(s == "" for s in [self.nthash, password, aesKey]) else ""
 
         if useCache and kerb_pass == "":
@@ -197,11 +200,8 @@ def kerberos_login(
                 add_user_bh(f"{self.hostname}$", self.domain, self.logger, self.config)
             return True
 
+    @reconnect_mssql
     def plaintext_login(self, domain, username, password):
-        with contextlib.suppress(Exception):
-            self.conn.disconnect()
-        self.create_conn_obj()
-
         self.password = password
         self.username = username
         self.domain = domain
@@ -235,11 +235,8 @@ def plaintext_login(self, domain, username, password):
                 add_user_bh(f"{self.hostname}$", self.domain, self.logger, self.config)
             return True
 
+    @reconnect_mssql
     def hash_login(self, domain, username, ntlm_hash):
-        with contextlib.suppress(Exception):
-            self.conn.disconnect()
-        self.create_conn_obj()
-
         self.username = username
         self.domain = domain
         self.nthash = f':{ntlm_hash.split(":")[1]}' if ntlm_hash.find(":") != -1 else f":{ntlm_hash}"