Luke Imhoff
0bb79ba890
[#46491831] Move Msf::DBManager#import_msf_xml into Msf::DBManager::ImportMsfXml#import_msf_xml and include Msf::DBManager::ImportMsfXml to cut down size of the infamous db.rb. Break up #import_msf_xml to have separate methods for parsing web_forms, web_pages, and web_vulns. The method for web_vulns, #import_msf_web_vuln_element is needed so that it can be overridden in Pro to handle the Pro-only changes to Mdm::WebVuln.
40 lines
848 B
Ruby
40 lines
848 B
Ruby
require 'msf/base/simple/framework'
|
|
require 'metasploit/framework'
|
|
|
|
shared_context 'Msf::Simple::Framework' do
|
|
let(:dummy_pathname) do
|
|
Metasploit::Framework.root.join('spec', 'dummy')
|
|
end
|
|
|
|
let(:framework) do
|
|
Msf::Simple::Framework.create(
|
|
'ConfigDirectory' => framework_config_pathname.to_s,
|
|
# don't load any module paths so we can just load the module under test and save time
|
|
'DeferModuleLoads' => true
|
|
)
|
|
end
|
|
|
|
let(:framework_config_pathname) do
|
|
dummy_pathname.join('framework', 'config')
|
|
end
|
|
|
|
before(:each) do
|
|
framework_config_pathname.mkpath
|
|
end
|
|
|
|
after(:each) do
|
|
dummy_pathname.rmtree
|
|
end
|
|
|
|
after(:each) do
|
|
# explicitly kill threads so that they don't exhaust connection pool
|
|
thread_manager = framework.threads
|
|
|
|
thread_manager.each do |thread|
|
|
thread.kill
|
|
end
|
|
|
|
thread_manager.monitor.kill
|
|
end
|
|
end
|