clean up code duplication

rapid7 · Mar 11, 2024 · 7ce91df · 7ce91df
1 parent d18520a
commit 7ce91df
Showing 1 changed file with 5 additions and 15 deletions.
diff --git a/modules/exploits/windows/mssql/mssql_payload.rb b/modules/exploits/windows/mssql/mssql_payload.rb
@@ -70,15 +70,6 @@ def initialize(info = {})
   end
 
   def check
-    if session
-      set_session(session.client)
-    end
-
-    unless session || mssql_login_datastore
-      vprint_status("Invalid SQL Server credentials")
-      return Exploit::CheckCode::Detected
-    end
-
     if session
       set_session(session.client)
     else
@@ -105,14 +96,13 @@ def execute_command(cmd, opts)
   end
 
   def exploit
-
     if session
       set_session(session.client)
-    end
-
-    unless session || mssql_login_datastore
-      print_status("Invalid SQL Server credentials")
-      return
+    else
+      unless mssql_login_datastore
+        print_status("Invalid SQL Server credentials")
+        return
+      end
     end
 
     method = datastore['METHOD'].downcase