126 lines
2.6 KiB
Ruby
126 lines
2.6 KiB
Ruby
#!/usr/bin/ruby
|
|
|
|
base = File.symlink?(__FILE__) ? File.readlink(__FILE__) : __FILE__
|
|
$:.unshift(File.join(File.dirname(base)))
|
|
|
|
require 'test/unit'
|
|
require 'pcaprub'
|
|
|
|
#
|
|
# Simple unit test, requires r00t.
|
|
#
|
|
|
|
class Pcap::UnitTest < Test::Unit::TestCase
|
|
|
|
def test_version
|
|
assert_equal(String, Pcap.version.class)
|
|
puts "Pcaprub version: #{Pcap.version}"
|
|
end
|
|
|
|
def test_lookupdev
|
|
assert_equal(String, Pcap.lookupdev.class)
|
|
puts "Pcaprub default device: #{Pcap.lookupdev}"
|
|
end
|
|
|
|
def test_lookupnet
|
|
dev = Pcap.lookupdev
|
|
assert_equal(Array, Pcap.lookupnet(dev).class)
|
|
net = Pcap.lookupnet(dev)
|
|
puts "Pcaprub net (#{dev}): #{net[0]} #{[net[1]].pack("N").unpack("H*")[0]}"
|
|
end
|
|
|
|
def test_pcap_new
|
|
o = Pcap.new
|
|
assert_equal(Pcap, o.class)
|
|
end
|
|
|
|
def test_pcap_setfilter_bad
|
|
e = nil
|
|
o = Pcap.new
|
|
begin
|
|
o.setfilter("not ip")
|
|
rescue ::Exception => e
|
|
end
|
|
|
|
assert_equal(e.class, ArgumentError)
|
|
end
|
|
|
|
def test_pcap_setfilter
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 65535, true, 1)
|
|
r = o.setfilter("not ip")
|
|
assert_equal(Pcap, r.class)
|
|
end
|
|
|
|
def test_pcap_inject
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 65535, true, 1)
|
|
r = o.inject("X" * 512)
|
|
assert_equal(512, r)
|
|
end
|
|
|
|
def test_pcap_datalink
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 65535, true, 1)
|
|
r = o.datalink
|
|
assert_equal(Fixnum, r.class)
|
|
end
|
|
|
|
def test_pcap_snapshot
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 1344, true, 1)
|
|
r = o.snapshot
|
|
assert_equal(1344, r)
|
|
end
|
|
|
|
def test_pcap_stats
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 1344, true, 1)
|
|
r = o.stats
|
|
assert_equal(Hash, r.class)
|
|
end
|
|
|
|
def test_pcap_next
|
|
d = Pcap.lookupdev
|
|
o = Pcap.open_live(d, 1344, true, 1)
|
|
|
|
@c = 0
|
|
t = Thread.new { while(true); @c += 1; select(nil, nil, nil, 0.10); end; }
|
|
|
|
require 'timeout'
|
|
begin
|
|
Timeout.timeout(10) do
|
|
o.each do |pkt|
|
|
end
|
|
end
|
|
rescue ::Timeout::Error
|
|
end
|
|
|
|
t.kill
|
|
puts "Background thread ticked #{@c} times while capture was running"
|
|
true
|
|
end
|
|
|
|
def test_netifaces_constants
|
|
puts "AF_LINK Value is #{Pcap::AF_LINK}"
|
|
puts "AF_INET Value is #{Pcap::AF_INET}"
|
|
puts "AF_INET6 Value is #{Pcap::AF_INET6}" if Pcap.const_defined?(:AF_INET6)
|
|
end
|
|
|
|
def test_netifaces_functions
|
|
Pcap.interfaces.sort.each do |iface|
|
|
puts "#{iface} :"
|
|
Pcap.addresses(iface).sort.each do |family,values|
|
|
puts "\t#{family} :"
|
|
values.each do |val|
|
|
puts "\t\taddr : #{val['addr']}" if val.has_key?("addr")
|
|
puts "\t\tnetmask : #{val['netmask']}" if val.has_key?("netmask")
|
|
puts "\t\tbroadcast : #{val['broadcast']}" if val.has_key?("broadcast")
|
|
puts "\n"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|