metasploit-framework/lib/msf/core/db_manager.rb

122 lines
2.2 KiB
Ruby
Raw Normal View History

require 'msf/core'
require 'msf/core/db'
module Msf
###
#
# The db module provides persistent storage and events
#
###
class DBManager
# Provides :framework and other accessors
include Framework::Offspring
# Returns true if we are ready to load/store data
attr_accessor :active
# Returns true if the prerequisites have been installed
attr_accessor :usable
# Returns the list of usable database drivers
attr_accessor :drivers
# Returns the active driver
attr_accessor :driver
# Stores the error message for why the db was not loaded
attr_accessor :error
def initialize(framework)
self.framework = framework
@usable = false
@active = false
#
# Prefer our local copy of active_record
#
dir_ar = File.join(Msf::Config.data_directory, 'msfweb', 'vendor', 'rails', 'activerecord', 'lib')
if(File.directory?(dir_ar) and not $:.include?(dir_ar))
$:.unshift(dir_ar)
end
# Load ActiveRecord if it is available
begin
require 'rubygems'
require 'active_record'
require 'msf/core/db_objects'
@usable = true
rescue ::Exception => e
self.error = e
elog("DB is not enabled due to load error: #{e}")
return
end
#
# Determine what drivers are available
#
initialize_drivers
end
#
#
#
def initialize_drivers
self.drivers = []
tdrivers = %W{ sqlite3 mysql postgresql }
tdrivers.each do |driver|
begin
ActiveRecord::Base.establish_connection(:adapter => driver)
ActiveRecord::Base.remove_connection
self.drivers << driver
rescue ::Exception
end
end
if(not self.drivers.empty?)
self.driver = self.drivers[0]
end
end
#
# Connects this instance to a database
#
def connect(opts={})
return false if not @usable
nopts = opts.dup
if (nopts['port'])
nopts['port'] = nopts['port'].to_i
end
begin
ActiveRecord::Base.establish_connection(nopts)
rescue ::Exception => e
elog("DB.connect threw an exception: #{e}")
return false
end
@active = true
end
#
# Disconnects a database session
#
def disconnect
begin
ActiveRecord::Base.remove_connection
rescue ::Exception => e
elog("DB.disconnect threw an exception: #{e}")
end
@active = false
end
end
end