##
# This module requires Metasploit: https://metasploit.com/download
# Current source: https://github.com/rapid7/metasploit-framework
##

class MetasploitModule < Msf::Auxiliary
  include Msf::Exploit::Remote::MSSQL
  include Msf::OptionalSession::MSSQL

  def initialize(info = {})
    super(
      update_info(
        info,
        'Name' => 'Microsoft SQL Server Generic Query',
        'Description' => %q{
          This module will allow for simple SQL statements to be executed against a
          MSSQL/MSDE instance given the appropriate credentials.
        },
        'Author' => [ 'tebo <tebo[at]attackresearch.com>' ],
        'License' => MSF_LICENSE,
        'References' => [
          [ 'URL', 'http://www.attackresearch.com' ],
          [ 'URL', 'http://msdn.microsoft.com/en-us/library/cc448435(PROT.10).aspx'],
        ],
        'Notes' => {
          'Stability' => [CRASH_SAFE],
          'SideEffects' => [IOC_IN_LOGS],
          'Reliability' => []
        }
      )
    )

    register_options(
      [
        OptString.new('SQL', [ false, 'The SQL query to execute', 'select @@version']),
      ]
    )
  end

  def auxiliary_commands
    { 'select' => 'Run a select query (a LIMIT clause is probably a really good idea)' }
  end

  def cmd_select(*args)
    datastore['SQL'] = "select #{args.join(' ')}"
    run
  end

  def run
    if session
      set_mssql_session(session.client)
    else
      unless mssql_login_datastore
        print_error('Error with mssql_login call')
        info = mssql_client.initial_connection_info
        if info[:errors] && !info[:errors].empty?
          info[:errors].each do |err|
            print_error(err)
          end
        end
        return
      end
    end

    mssql_query(datastore['SQL'], true)
  end
end