clean up code duplication

2024-02-14 08:42:45 -06:00
parent d18520adc6
commit 7ce91df66e
1 changed files with 5 additions and 15 deletions
@@ -70,15 +70,6 @@ class MetasploitModule < Msf::Exploit::Remote
  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 @@ class MetasploitModule < Msf::Exploit::Remote
  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