adam/metasploit-gs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Retab all the things (except external/)

Browse Source
This commit is contained in:
Tab Assassin
2013-09-30 13:47:53 -05:00
parent 0ecba377f5
commit 2e8d19edcf
293 changed files with 32962 additions and 32962 deletions
Download Patch File Download Diff File Expand all files Collapse all files
spec/support/shared/examples/msf/module_manager/loading.rb
+126 -126
View File
@@ -1,162 +1,162 @@
shared_examples_for 'Msf::ModuleManager::Loading' do
context '#file_changed?' do
let(:module_basename) do
[basename_prefix, '.rb']
end
context '#file_changed?' do
let(:module_basename) do
[basename_prefix, '.rb']
end
it 'should return true if module info is not cached' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
it 'should return true if module info is not cached' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
subject.send(:module_info_by_path)[module_path].should be_nil
subject.file_changed?(module_path).should be_true
end
end
subject.send(:module_info_by_path)[module_path].should be_nil
subject.file_changed?(module_path).should be_true
end
end
it 'should return true if the cached type is Msf::MODULE_PAYLOAD' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path)
it 'should return true if the cached type is Msf::MODULE_PAYLOAD' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path)
subject.send(:module_info_by_path)[module_path] = {
# :modification_time must match so that it is the :type that is causing the `true` and not the
# :modification_time causing the `true`.
:modification_time => modification_time,
:type => Msf::MODULE_PAYLOAD
}
subject.send(:module_info_by_path)[module_path] = {
# :modification_time must match so that it is the :type that is causing the `true` and not the
# :modification_time causing the `true`.
:modification_time => modification_time,
:type => Msf::MODULE_PAYLOAD
}
subject.file_changed?(module_path).should be_true
end
end
subject.file_changed?(module_path).should be_true
end
end
context 'with cache module info and not a payload module' do
it 'should return true if the file does not exist on the file system' do
tempfile = Tempfile.new(module_basename)
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
context 'with cache module info and not a payload module' do
it 'should return true if the file does not exist on the file system' do
tempfile = Tempfile.new(module_basename)
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
subject.send(:module_info_by_path)[module_path] = {
:modification_time => modification_time
}
subject.send(:module_info_by_path)[module_path] = {
:modification_time => modification_time
}
tempfile.unlink
tempfile.unlink
File.exist?(module_path).should be_false
subject.file_changed?(module_path).should be_true
end
File.exist?(module_path).should be_false
subject.file_changed?(module_path).should be_true
end
it 'should return true if modification time does not match the cached modification time' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
cached_modification_time = (modification_time * rand).to_i
it 'should return true if modification time does not match the cached modification time' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
cached_modification_time = (modification_time * rand).to_i
subject.send(:module_info_by_path)[module_path] = {
:modification_time => cached_modification_time
}
subject.send(:module_info_by_path)[module_path] = {
:modification_time => cached_modification_time
}
cached_modification_time.should_not == modification_time
subject.file_changed?(module_path).should be_true
end
end
cached_modification_time.should_not == modification_time
subject.file_changed?(module_path).should be_true
end
end
it 'should return false if modification time does match the cached modification time' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
cached_modification_time = modification_time
it 'should return false if modification time does match the cached modification time' do
Tempfile.open(module_basename) do |tempfile|
module_path = tempfile.path
modification_time = File.mtime(module_path).to_i
cached_modification_time = modification_time
subject.send(:module_info_by_path)[module_path] = {
:modification_time => cached_modification_time
}
subject.send(:module_info_by_path)[module_path] = {
:modification_time => cached_modification_time
}
cached_modification_time.should == modification_time
subject.file_changed?(module_path).should be_false
end
end
end
end
cached_modification_time.should == modification_time
subject.file_changed?(module_path).should be_false
end
end
end
end
context '#on_module_load' do
def on_module_load
module_manager.on_module_load(klass, type, reference_name, options)
end
context '#on_module_load' do
def on_module_load
module_manager.on_module_load(klass, type, reference_name, options)
end
let(:klass) do
Class.new(Msf::Auxiliary)
end
let(:klass) do
Class.new(Msf::Auxiliary)
end
let(:module_set) do
module_manager.module_set(type)
end
let(:module_set) do
module_manager.module_set(type)
end
let(:namespace_module) do
double('Namespace Module', :parent_path => parent_path)
end
let(:namespace_module) do
double('Namespace Module', :parent_path => parent_path)
end
let(:options) do
{
'files' => [
path
],
'paths' => [
reference_name
],
'type' => type
}
end
let(:options) do
{
'files' => [
path
],
'paths' => [
reference_name
],
'type' => type
}
end
let(:parent_path) do
Metasploit::Framework.root.join('modules')
end
let(:parent_path) do
Metasploit::Framework.root.join('modules')
end
let(:path) do
type_directory = Mdm::Module::Detail::DIRECTORY_BY_TYPE[type]
let(:path) do
type_directory = Mdm::Module::Detail::DIRECTORY_BY_TYPE[type]
File.join(parent_path, type_directory, "#{reference_name}.rb")
end
File.join(parent_path, type_directory, "#{reference_name}.rb")
end
let(:reference_name) do
'admin/2wire/xslt_password_reset'
end
let(:reference_name) do
'admin/2wire/xslt_password_reset'
end
let(:type) do
klass.type
end
let(:type) do
klass.type
end
before(:each) do
klass.stub(:parent => namespace_module)
end
before(:each) do
klass.stub(:parent => namespace_module)
end
it "should add module to type's module_set" do
module_set.should_receive(:add_module).with(
klass,
reference_name,
options
)
it "should add module to type's module_set" do
module_set.should_receive(:add_module).with(
klass,
reference_name,
options
)
on_module_load
end
on_module_load
end
it 'should call cache_in_memory' do
module_manager.should_receive(:cache_in_memory)
it 'should call cache_in_memory' do
module_manager.should_receive(:cache_in_memory)
on_module_load
end
on_module_load
end
it 'should pass class to #auto_subscribe_module' do
module_manager.should_receive(:auto_subscribe_module).with(klass)
it 'should pass class to #auto_subscribe_module' do
module_manager.should_receive(:auto_subscribe_module).with(klass)
on_module_load
end
on_module_load
end
it 'should fire on_module_load event with class' do
framework.events.should_receive(:on_module_load).with(
reference_name,
klass
)
it 'should fire on_module_load event with class' do
framework.events.should_receive(:on_module_load).with(
reference_name,
klass
)
on_module_load
end
end
on_module_load
end
end
end