metasploit-framework/spec/lib/rex/payloads/meterpreter/uri_checksum_spec.rb

119 lines
3.7 KiB
Ruby
Raw Normal View History

require 'spec_helper'
require 'rex/payloads/meterpreter/uri_checksum'
RSpec.describe Rex::Payloads::Meterpreter::UriChecksum do
class DummyClass
include Rex::Payloads::Meterpreter::UriChecksum
end
subject(:dummy_object) { DummyClass.new }
it { is_expected.to respond_to :generate_uri_checksum}
it { is_expected.to respond_to :process_uri_resource}
it { is_expected.to respond_to :uri_checksum_lookup}
describe '#process_uri_resource' do
context 'when passed a value for INITW' do
let(:uri) { "/7E37v"}
it 'returns a static value of /INITM' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq :init_native
end
context 'with junk appended at the end' do
let(:uri) { "/7E37v_foobar"}
it 'returns a static value of /INITM' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq nil
end
end
end
context 'when passed a value for INITJ' do
let(:uri) { "/a6BF9"}
it 'returns a static value of /INITJM' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq :init_java
end
context 'with junk appended at the end' do
let(:uri) { "/a6BF9_foobar"}
it 'returns a static value of /INITJM' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq nil
end
end
end
context 'when passed a value for CONN' do
let(:uri) { "/39ab3"}
it 'returns /CONN plus random junk' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq :connect
end
context 'with junk appended at the end' do
let(:uri) { "/39ab3_foobar"}
it 'returns /CONN plus the junk' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq nil
end
end
end
context 'when passed a junk value' do
let(:uri) { "/lolz"}
it 'returns the original uri string' do
2015-04-06 03:56:59 +00:00
expect(dummy_object.process_uri_resource(uri)[:mode]).to eq nil
end
end
end
describe '#generate_uri_checksum' do
[0, 80, 88, 90, 92, 98, 255, 127].each do |checksum_value|
[5,30,50,100,127].each do |uri_length|
["", "/boom", "/___AAAAAAAAAAAAA"].each do |prefix|
it "generates a #{uri_length} byte string that checksums back to the original value (#{checksum_value}) with prefix #{prefix}" do
uri_string = dummy_object.generate_uri_checksum(checksum_value, uri_length + prefix.to_s.length, prefix)
expect(Rex::Text.checksum8(uri_string)).to eq checksum_value
end
end
end
end
end
describe '#uri_checksum_lookup' do
context 'when passed a value for :connect' do
let(:mode) { :connect }
it 'returns a URI_CHECKSUM_CONN' do
expect(dummy_object.uri_checksum_lookup(mode)).to eq Rex::Payloads::Meterpreter::UriChecksum::URI_CHECKSUM_CONN
end
end
2015-04-06 03:56:59 +00:00
context 'when passed a value for :init_native' do
let(:mode) { :init_native }
it 'returns a URI_CHECKSUM_INITN' do
expect(dummy_object.uri_checksum_lookup(mode)).to eq Rex::Payloads::Meterpreter::UriChecksum::URI_CHECKSUM_INITN
end
end
context 'when passed a value for :init_java' do
let(:mode) { :init_java }
it 'returns a URI_CHECKSUM_INITJ' do
expect(dummy_object.uri_checksum_lookup(mode)).to eq Rex::Payloads::Meterpreter::UriChecksum::URI_CHECKSUM_INITJ
end
end
context 'when passed a value for :init_python' do
let(:mode) { :init_python }
it 'returns a URI_CHECKSUM_INITP' do
expect(dummy_object.uri_checksum_lookup(mode)).to eq Rex::Payloads::Meterpreter::UriChecksum::URI_CHECKSUM_INITP
end
end
end
end