dwelch-r7
71 lines
2.1 KiB
Ruby
71 lines
2.1 KiB
Ruby
require 'spec_helper'
|
|
|
|
RSpec.describe ::Msf::Ui::Console::CommandDispatcher::Common do
|
|
let(:subject) do
|
|
dummy_class = Class.new
|
|
dummy_class.extend(described_class)
|
|
end
|
|
|
|
def with_path_extensions(original_paths)
|
|
extensions = ['rb', 'br', 'r', 'b']
|
|
extensions += ['py', 'yp', 'p', 'y']
|
|
extensions += ['go', 'og', 'g', 'o']
|
|
extensions += ['']
|
|
|
|
paths_with_extensions = extensions.flat_map do |extension|
|
|
original_paths.map { |path| "#{path}.#{extension}" }
|
|
end
|
|
original_paths + paths_with_extensions
|
|
end
|
|
|
|
describe 'Trimming a path of errors' do
|
|
context 'when a user is inputting a Module path' do
|
|
let(:valid_path) { 'windows/smb/ms08_067_netapi' }
|
|
let(:all_paths) do
|
|
[
|
|
'./exploits/windows/smb/ms08_067_netapi',
|
|
'/exploits/windows/smb/ms08_067_netapi',
|
|
'.exploits/windows/smb/ms08_067_netapi',
|
|
'exploits/windows/smb/ms08_067_netapi',
|
|
|
|
'./windows/smb/ms08_067_netapi',
|
|
'/windows/smb/ms08_067_netapi',
|
|
'.windows/smb/ms08_067_netapi',
|
|
|
|
valid_path
|
|
]
|
|
end
|
|
|
|
it 'corrects the module paths to be valid' do
|
|
with_path_extensions(all_paths).each do |path|
|
|
expect(subject.trim_path(path, 'exploits')).to eql(valid_path)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when a user is inputting a Payload path' do
|
|
let(:valid_path) { 'windows/x64/vncinject/reverse_winhttps' }
|
|
let(:all_paths) do
|
|
[
|
|
'./payload/windows/x64/vncinject/reverse_winhttps',
|
|
'/payload/windows/x64/vncinject/reverse_winhttps',
|
|
'.payload/windows/x64/vncinject/reverse_winhttps',
|
|
'payload/windows/x64/vncinject/reverse_winhttps',
|
|
|
|
'./windows/x64/vncinject/reverse_winhttps',
|
|
'/windows/x64/vncinject/reverse_winhttps',
|
|
'.windows/x64/vncinject/reverse_winhttps',
|
|
|
|
valid_path
|
|
]
|
|
end
|
|
|
|
it "when a user is inputting a Modules path" do
|
|
with_path_extensions(all_paths).each do |path|
|
|
expect(subject.trim_path(path, 'payload')).to eql(valid_path)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|