mirror of
https://github.com/rapid7/metasploit-framework
synced 2024-11-12 11:52:01 +01:00
132 lines
4.1 KiB
Ruby
132 lines
4.1 KiB
Ruby
$:.push("../../lib")
|
|
require 'rex'
|
|
require 'rex/socket'
|
|
require 'rex/socket/range_walker'
|
|
|
|
describe Rex::Socket::RangeWalker do
|
|
it "should have a num_ips attribute" do
|
|
walker = Rex::Socket::RangeWalker.new("")
|
|
walker.should respond_to("num_ips")
|
|
walker.should respond_to("length")
|
|
walker.num_ips.should == walker.length
|
|
end
|
|
it "should handle single ipv6 addresses" do
|
|
walker = Rex::Socket::RangeWalker.new("::1")
|
|
walker.should be_valid
|
|
walker.length.should == 1
|
|
end
|
|
it "should handle ranges" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1-2")
|
|
walker.should be_valid
|
|
walker.length.should == 2
|
|
walker.next_ip.should == "10.1.1.1"
|
|
walker = Rex::Socket::RangeWalker.new("10.1-2.1.1-2")
|
|
walker.should be_valid
|
|
walker.length.should == 4
|
|
walker = Rex::Socket::RangeWalker.new("10.1-2.3-4.5-6")
|
|
walker.should be_valid
|
|
walker.length.should == 8
|
|
walker.should include("10.1.3.5")
|
|
# Slow test on a 3.06 GHz proc
|
|
# ruby 1.9, ~ 11 seconds
|
|
# ruby 1.8.7, ~ 24 seconds
|
|
# ruby 1.8.6, ~ 23 seconds
|
|
#walker = Rex::Socket::RangeWalker.new("10.0-255.0-255.0-255")
|
|
#walker.should be_valid
|
|
#walker.length.should == 256 * 256 * 256
|
|
end
|
|
it "should default the lower bound of a range to 0" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.3.-17")
|
|
walker.should be_valid
|
|
walker.length.should == 18
|
|
walker = Rex::Socket::RangeWalker.new("10.1.3.-255")
|
|
walker.should be_valid
|
|
walker.length.should == 256
|
|
end
|
|
it "should default the upper bound of a range to 255" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.3.254-")
|
|
walker.should be_valid
|
|
walker.length.should == 2
|
|
end
|
|
it "should handle lists" do
|
|
#walker = Rex::Socket::RangeWalker.new("10.1.1.1,2")
|
|
#walker.should be_valid
|
|
#walker.length.should == 2
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1")
|
|
walker.should be_valid
|
|
walker.length.should == 1
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1,3")
|
|
walker.should be_valid
|
|
walker.length.should == 2
|
|
walker.should_not include("10.1.1.2")
|
|
end
|
|
it "should handle ranges and lists together" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1-2,3")
|
|
walker.should be_valid
|
|
walker.length.should == 3
|
|
walker = Rex::Socket::RangeWalker.new("10.1-2.1.1,2")
|
|
walker.should be_valid
|
|
walker.length.should == 4
|
|
walker = Rex::Socket::RangeWalker.new("10.1,2.3,4.5,6")
|
|
walker.length.should == 8
|
|
end
|
|
it "should handle cidr" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/31")
|
|
walker.should be_valid
|
|
walker.length.should == 2
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/30")
|
|
walker.should be_valid
|
|
walker.length.should == 4
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/29")
|
|
walker.should be_valid
|
|
walker.length.should == 8
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/28")
|
|
walker.should be_valid
|
|
walker.length.should == 16
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/27")
|
|
walker.should be_valid
|
|
walker.length.should == 32
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/26")
|
|
walker.should be_valid
|
|
walker.length.should == 64
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/25")
|
|
walker.should be_valid
|
|
walker.length.should == 128
|
|
pending("Decide whether cidr_crack should include 0")
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/24")
|
|
walker.should be_valid
|
|
walker.length.should == 256
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1/23")
|
|
walker.should be_valid
|
|
walker.length.should == 512
|
|
end
|
|
|
|
it "should handle ipv6 cidr" do
|
|
walker = Rex::Socket::RangeWalker.new("::1/127")
|
|
walker.should be_valid
|
|
walker.length.should == 2
|
|
walker = Rex::Socket::RangeWalker.new("::1/122")
|
|
walker.should be_valid
|
|
walker.length.should == 2 ** 6
|
|
walker = Rex::Socket::RangeWalker.new("::1/116")
|
|
walker.should be_valid
|
|
walker.length.should == 2 ** 12
|
|
end
|
|
|
|
#it "should handle ipv6 ranges" do
|
|
# pending("Need to define how this should be handled")
|
|
# walker = Rex::Socket::RangeWalker.new("::1-::1:1")
|
|
# walker.should be_valid
|
|
# walker.length.should == 2 ** 16
|
|
#end
|
|
|
|
it "should yield all ips" do
|
|
walker = Rex::Socket::RangeWalker.new("10.1.1.1,2,3")
|
|
got = []
|
|
walker.each { |ip|
|
|
got.push ip
|
|
}
|
|
got.should == ["10.1.1.1", "10.1.1.2", "10.1.1.3"]
|
|
end
|
|
end
|