Luke Imhoff
49 lines
1010 B
Ruby
49 lines
1010 B
Ruby
module Msf::DBManager::Adapter
|
|
#
|
|
# CONSTANTS
|
|
#
|
|
|
|
# The adapter to use to establish database connection.
|
|
ADAPTER = 'postgresql'
|
|
|
|
#
|
|
# Attributes
|
|
#
|
|
|
|
# Returns the list of usable database drivers
|
|
def drivers
|
|
@drivers ||= []
|
|
end
|
|
attr_writer :drivers
|
|
|
|
# Returns the active driver
|
|
attr_accessor :driver
|
|
|
|
#
|
|
# Instance Methods
|
|
#
|
|
|
|
#
|
|
# Scan through available drivers
|
|
#
|
|
def initialize_adapter
|
|
ActiveRecord::Base.default_timezone = :utc
|
|
|
|
if connection_established? && ActiveRecord::Base.connection_config[:adapter] == ADAPTER
|
|
dlog("Already established connection to #{ADAPTER}, so reusing active connection.")
|
|
self.drivers << ADAPTER
|
|
self.driver = ADAPTER
|
|
else
|
|
begin
|
|
ActiveRecord::Base.establish_connection(adapter: ADAPTER)
|
|
ActiveRecord::Base.remove_connection
|
|
rescue Exception => error
|
|
@adapter_error = error
|
|
else
|
|
self.drivers << ADAPTER
|
|
self.driver = ADAPTER
|
|
end
|
|
end
|
|
end
|
|
end
|