119 lines
3.7 KiB
Ruby
119 lines
3.7 KiB
Ruby
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
|
|
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
|