jvazquez-r7
74 lines
1.7 KiB
Ruby
74 lines
1.7 KiB
Ruby
# -*- coding:binary -*-
|
|
require 'spec_helper'
|
|
|
|
require 'rex/java/serialization'
|
|
require 'msf/java/rmi/util'
|
|
|
|
describe Msf::Java::Rmi::Util do
|
|
subject(:mod) do
|
|
mod = ::Msf::Exploit.new
|
|
mod.extend ::Msf::Java::Rmi::Util
|
|
mod.send(:initialize)
|
|
mod
|
|
end
|
|
|
|
let(:interface_methods) do
|
|
[
|
|
{name: 'sayHello', descriptor: '()Ljava/lang/String;'},
|
|
{name: 'sayHelloTwo', descriptor: '(Ljava/lang/String;)Ljava/lang/String;'}
|
|
]
|
|
end
|
|
|
|
let(:interface_exceptions) do
|
|
['java.rmi.RemoteException']
|
|
end
|
|
|
|
let(:interface_hash) do
|
|
0x3e664fcbd9e953bb
|
|
end
|
|
|
|
let(:method_signature) do
|
|
'sayHello()Ljava/lang/String;'
|
|
end
|
|
|
|
let(:method_hash) do
|
|
0x53e0822d3e3724df
|
|
end
|
|
|
|
let(:dgc_methods) do
|
|
[
|
|
{name: 'clean', descriptor: '([Ljava/rmi/server/ObjID;JLjava/rmi/dgc/VMID;Z)V'},
|
|
{name: 'dirty', descriptor: '([Ljava/rmi/server/ObjID;JLjava/rmi/dgc/Lease;)Ljava/rmi/dgc/Lease;'}
|
|
]
|
|
end
|
|
|
|
let(:dgc_exceptions) do
|
|
['java.rmi.RemoteException']
|
|
end
|
|
|
|
let(:dgc_hash) do
|
|
0xf6b6898d8bf28643
|
|
end
|
|
|
|
describe "#calculate_interface_hash" do
|
|
context "when an example interface is provided" do
|
|
it "generates a correct interface hash" do
|
|
expect(mod.calculate_interface_hash(interface_methods, interface_exceptions)).to eq(interface_hash)
|
|
end
|
|
end
|
|
|
|
context "when a DGC interface is provided" do
|
|
it "generates a correct interface hash" do
|
|
expect(mod.calculate_interface_hash(dgc_methods, dgc_exceptions)).to eq(dgc_hash)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe "#calculate_method_hash" do
|
|
it "generates a correct interface hash" do
|
|
expect(mod.calculate_method_hash(method_signature)).to eq(method_hash)
|
|
end
|
|
end
|
|
end
|
|
|