Samuel Huckins
149c3ecc63
* --ask option ported to new location * --version option now works * MSF version updated * All specs passing
59 lines
1.6 KiB
Ruby
59 lines
1.6 KiB
Ruby
require 'rails'
|
|
require File.expand_path('../boot', __FILE__)
|
|
|
|
all_environments = [
|
|
:development,
|
|
:production,
|
|
:test
|
|
]
|
|
|
|
Bundler.require(
|
|
*Rails.groups(
|
|
db: all_environments,
|
|
pcap: all_environments
|
|
)
|
|
)
|
|
|
|
#
|
|
# Railties
|
|
#
|
|
|
|
# For compatibility with jquery-rails (and other engines that need action_view) in pro
|
|
require 'action_view/railtie'
|
|
|
|
#
|
|
# Project
|
|
#
|
|
|
|
require 'metasploit/framework/common_engine'
|
|
require 'msf/base/config'
|
|
|
|
module Metasploit
|
|
module Framework
|
|
class Application < Rails::Application
|
|
include Metasploit::Framework::CommonEngine
|
|
|
|
environment_database_yaml = ENV['MSF_DATABASE_CONFIG']
|
|
|
|
if environment_database_yaml
|
|
# DO NOT check if the path exists: if the environment variable is set, then the user meant to use this path
|
|
# and if it doesn't exist then an error should occur so the user knows the environment variable points to a
|
|
# non-existent file.
|
|
config.paths['config/database'] = environment_database_yaml
|
|
else
|
|
user_config_root = Pathname.new(Msf::Config.get_config_root)
|
|
user_database_yaml = user_config_root.join('database.yml')
|
|
|
|
# DO check if the path exists as in test environments there may be no config root, in which case the normal
|
|
# rails location, `config/database.yml`, should contain the database config.
|
|
if user_database_yaml.exist?
|
|
config.paths['config/database'] = [user_database_yaml.to_path]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
# Silence warnings about this defaulting to true
|
|
I18n.enforce_available_locales = true
|