packetfu 1.1.9 → 1.1.10

data/test/test_capture.rb

@@ -6,51 +6,51 @@ require 'packetfu'
 
 class CaptureTest < Test::Unit::TestCase
 
-  def test_cap
-    assert_nothing_raised { PacketFu::Capture }
-  end
-
-  def test_whoami
-    assert_nothing_raised { PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo')) }
-  end
-  
-  def test_new
-    cap = PacketFu::Capture.new
-    assert_kind_of PacketFu::Capture, cap
-    cap = PacketFu::Capture.new(
-      :filter => 'tcp and dst host 1.2.3.4'
-    )
-  end
-  
-  def test_filter
-    daddr = PacketFu::Utils.rand_routable_daddr.to_s
-    cap = PacketFu::Capture.new( :filter => "icmp and dst host #{daddr}")
-    cap.start
-    %x{ping -c 1 #{daddr}}
-    sleep 1
-    cap.save
-    assert cap.array.size == 1
-    pkt = PacketFu::Packet.parse(cap.array.first)
-    assert pkt.ip_daddr == daddr
-  end
-  
-  def test_no_filter
-    daddr = PacketFu::Utils.rand_routable_daddr.to_s
-    daddr2 = PacketFu::Utils.rand_routable_daddr.to_s
-    cap = PacketFu::Capture.new
-    cap.start
-    %x{ping -c 1 #{daddr}}
-    %x{ping -c 1 #{daddr2}}
-    sleep 1
-    cap.save
-    assert cap.array.size > 1
-  end
-
-  def test_bpf_alias
-    daddr = PacketFu::Utils.rand_routable_daddr.to_s
-    cap = PacketFu::Capture.new( :filter => "icmp and dst host #{daddr}")
-    assert cap.filter.object_id == cap.bpf.object_id
-  end
+	def test_cap
+		assert_nothing_raised { PacketFu::Capture }
+	end
+
+	def test_whoami
+		assert_nothing_raised { PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo')) }
+	end
+	
+	def test_new
+		cap = PacketFu::Capture.new
+		assert_kind_of PacketFu::Capture, cap
+		cap = PacketFu::Capture.new(
+			:filter => 'tcp and dst host 1.2.3.4'
+		)
+	end
+	
+	def test_filter
+		daddr = PacketFu::Utils.rand_routable_daddr.to_s
+		cap = PacketFu::Capture.new( :filter => "icmp and dst host #{daddr}")
+		cap.start
+		%x{ping -c 1 #{daddr}}
+		sleep 1
+		cap.save
+		assert cap.array.size == 1
+		pkt = PacketFu::Packet.parse(cap.array.first)
+		assert pkt.ip_daddr == daddr
+	end
+	
+	def test_no_filter
+		daddr = PacketFu::Utils.rand_routable_daddr.to_s
+		daddr2 = PacketFu::Utils.rand_routable_daddr.to_s
+		cap = PacketFu::Capture.new
+		cap.start
+		%x{ping -c 1 #{daddr}}
+		%x{ping -c 1 #{daddr2}}
+		sleep 1
+		cap.save
+		assert cap.array.size > 1
+	end
+
+	def test_bpf_alias
+		daddr = PacketFu::Utils.rand_routable_daddr.to_s
+		cap = PacketFu::Capture.new( :filter => "icmp and dst host #{daddr}")
+		assert cap.filter.object_id == cap.bpf.object_id
+	end
 
 end
 

data/test/test_eth.rb

@@ -6,85 +6,85 @@ puts "Testing #{PacketFu.version}: #{$0}"
 
 class EthTest < Test::Unit::TestCase
 
-  def test_ethmac
-    dst = "\x00\x03\x2f\x1a\x74\xde"
-    e = PacketFu::EthMac.new
-    e.read dst
-    assert_equal(dst, e.to_s)
-    assert_equal(0x32f, e.oui.oui)
-    assert_equal("\x1a\x74\xde", e.nic.to_s)
-    assert_equal(222, e.nic.n2)
-  end
+	def test_ethmac
+		dst = "\x00\x03\x2f\x1a\x74\xde"
+		e = PacketFu::EthMac.new
+		e.read dst
+		assert_equal(dst, e.to_s)
+		assert_equal(0x32f, e.oui.oui)
+		assert_equal("\x1a\x74\xde", e.nic.to_s)
+		assert_equal(222, e.nic.n2)
+	end
 
-  def test_ethmac_ipad
-    dst = "\x7c\x6d\x62\x01\x02\x03"
-    e = PacketFu::EthMac.new
-    e.read dst
-    assert_equal(dst, e.to_s)
-    assert_equal(0x6d62, e.oui.oui)
-  end
+	def test_ethmac_ipad
+		dst = "\x7c\x6d\x62\x01\x02\x03"
+		e = PacketFu::EthMac.new
+		e.read dst
+		assert_equal(dst, e.to_s)
+		assert_equal(0x6d62, e.oui.oui)
+	end
 
-  def test_ethmac_class
-    src = "\x00\x1b\x11\x51\xb7\xce"
-    e = PacketFu::EthMac.new
-    e.read src
-    assert_instance_of(PacketFu::EthMac, e)
-  end
+	def test_ethmac_class
+		src = "\x00\x1b\x11\x51\xb7\xce"
+		e = PacketFu::EthMac.new
+		e.read src
+		assert_instance_of(PacketFu::EthMac, e)
+	end
 
-  def test_eth
-    header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
-    src = "\x00\x1b\x11\x51\xb7\xce"
-    dst = "\x00\x03\x2f\x1a\x74\xde"
-    e = PacketFu::EthHeader.new
-    e.eth_dst = dst
-    e.eth_src = src
-    e.eth_proto = "\x08\x00"
-    assert_equal(header, e.to_s)
-    assert_equal(header, PacketFu::EthHeader.new.read(header).to_s)
-  end
+	def test_eth
+		header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
+		src = "\x00\x1b\x11\x51\xb7\xce"
+		dst = "\x00\x03\x2f\x1a\x74\xde"
+		e = PacketFu::EthHeader.new
+		e.eth_dst = dst
+		e.eth_src = src
+		e.eth_proto = "\x08\x00"
+		assert_equal(header, e.to_s)
+		assert_equal(header, PacketFu::EthHeader.new.read(header).to_s)
+	end
 
-  def test_macaddr
-    dst = "\x00\x03\x2f\x1a\x74\xde"
-    dstmac = "00:03:2f:1a:74:de"
-    assert_equal(dstmac,PacketFu::EthHeader.str2mac(dst))
-    assert_equal(dst, PacketFu::EthHeader.mac2str(dstmac))
-  end
+	def test_macaddr
+		dst = "\x00\x03\x2f\x1a\x74\xde"
+		dstmac = "00:03:2f:1a:74:de"
+		assert_equal(dstmac,PacketFu::EthHeader.str2mac(dst))
+		assert_equal(dst, PacketFu::EthHeader.mac2str(dstmac))
+	end
 
 end
 
 class EthPacketTest < Test::Unit::TestCase
-  include PacketFu
+	include PacketFu
 
-  def test_eth_create
-    sample_packet = PcapFile.new.file_to_array(:f => 'sample.pcap')[0]
-    e = EthPacket.new
-    header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
-    assert_kind_of EthPacket, e
-    assert_kind_of EthHeader, e.headers[0]
-    assert e.is_eth?
-    assert !e.is_tcp?
-    e.eth_dst = "\x00\x03\x2f\x1a\x74\xde"
-    e.eth_src = "\x00\x1b\x11\x51\xb7\xce"
-    e.eth_proto = 0x0800
-    assert_equal header, e.to_s[0,14]
-  end
+	def test_eth_create
+		sample_packet = PcapFile.new.file_to_array(:f => 'sample.pcap')[0]
+		e = EthPacket.new
+		header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
+		assert_kind_of EthPacket, e
+		assert_kind_of EthHeader, e.headers[0]
+		assert e.is_eth?
+		assert !e.is_tcp?
+		e.eth_dst = "\x00\x03\x2f\x1a\x74\xde"
+		e.eth_src = "\x00\x1b\x11\x51\xb7\xce"
+		e.eth_proto = 0x0800
+		assert_equal header, e.to_s[0,14]
+	end
 
-  def test_eth_new
-    p = EthPacket.new(
-    :eth_dst => "\x00\x03\x2f\x1a\x74\xde",
-    :eth_src => "\x00\x1b\x11\x51\xb7\xce",
-    :eth_proto => 0x0800)
-    header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
-    assert_equal header, p.to_s[0,14]
-  end
+	def test_eth_new
+		p = EthPacket.new(
+		:eth_dst => "\x00\x03\x2f\x1a\x74\xde",
+		:eth_src => "\x00\x1b\x11\x51\xb7\xce",
+		:eth_proto => 0x0800)
+		header = "00032f1a74de001b1151b7ce0800".scan(/../).map { |x| x.to_i(16) }.pack("C*")
+		assert_equal header, p.to_s[0,14]
+	end
 
-  def test_eth_write
-    p = EthPacket.new(
-    :eth_dst => "\x00\x03\x2f\x1a\x74\xde",
-    :eth_src => "\x00\x1b\x11\x51\xb7\xce",
-    :eth_proto => 0x0800)
-    p.to_f('eth_test.pcap')
-  end
+	def test_eth_write
+		p = EthPacket.new(
+		:eth_dst => "\x00\x03\x2f\x1a\x74\xde",
+		:eth_src => "\x00\x1b\x11\x51\xb7\xce",
+		:eth_proto => 0x0800)
+		p.to_f('eth_test.pcap')
+	end
 
 end
 

data/test/test_hsrp.rb

@@ -4,16 +4,16 @@ $:.unshift File.join(File.expand_path(File.dirname(__FILE__)), "..", "lib")
 require 'packetfu'
 
 class HSRPTest < Test::Unit::TestCase
-  include PacketFu
+	include PacketFu
 
-  def test_hsrp_read
-    sample_packet = PcapFile.new.file_to_array(:f => 'sample_hsrp_pcapr.cap')[0]
-    pkt = Packet.parse(sample_packet)
-    assert pkt.is_hsrp?
-    assert pkt.is_udp?
-    assert_equal(0x2d8d, pkt.udp_sum.to_i)
-    # pkt.to_f('udp_test.pcap','a')
-  end
+	def test_hsrp_read
+		sample_packet = PcapFile.new.file_to_array(:f => 'sample_hsrp_pcapr.cap')[0]
+		pkt = Packet.parse(sample_packet)
+		assert pkt.is_hsrp?
+		assert pkt.is_udp?
+		assert_equal(0x2d8d, pkt.udp_sum.to_i)
+		# pkt.to_f('udp_test.pcap','a')
+	end
 
 end
 

data/test/test_icmp.rb

@@ -4,58 +4,58 @@ $:.unshift File.join(File.expand_path(File.dirname(__FILE__)), "..", "lib")
 require 'packetfu'
 
 class ICMPTest < Test::Unit::TestCase
-  include PacketFu
-
-  def test_icmp_header_new
-    i = ICMPHeader.new
-    assert_kind_of ICMPHeader, i
-    assert_equal("\x00\x00\xff\xff", i.to_s)
-    i.icmp_type = 1
-    i.icmp_recalc :icmp_sum
-    assert_equal("\x01\x00\xfe\xff", i.to_s)
-  end
-
-  def test_icmp_peek
-    i = ICMPPacket.new
-    i.ip_saddr = "10.20.30.40"
-    i.ip_daddr = "50.60.70.80"
-    i.payload = "abcdefghijklmnopqrstuvwxyz"
-    i.recalc
-    puts "\n"
-    puts "ICMP Peek format: "
-    puts i.peek
-    assert (i.peek.size <= 80)
-  end
-
-  def test_icmp_pcap
-    i = ICMPPacket.new
-    assert_kind_of ICMPPacket, i
-    i.recalc
-    i.to_f('icmp_test.pcap')
-    i.ip_saddr = "10.20.30.40"
-    i.ip_daddr = "50.60.70.80"
-    i.payload = "\x00\x01\x00\01abcdefghijklmnopqrstuvwxyz"
-    i.icmp_code = 8
-    i.recalc
-    i.to_f('icmp_test.pcap','a')
-    assert File.exists?('icmp_test.pcap')
-  end
-
-  def test_icmp_read
-    sample_packet = PcapFile.new.file_to_array(:f => 'sample.pcap')[2]
-    pkt = Packet.parse(sample_packet)
-    assert pkt.is_icmp?
-    assert_kind_of ICMPPacket, pkt
-    assert_equal(0x4d58, pkt.icmp_sum.to_i)
-    assert_equal(8, pkt.icmp_type.to_i)
-  end
-
-  def test_icmp_reread
-    sample_packet = PacketFu::ICMPPacket.new
-    pkt = Packet.parse(sample_packet.to_s)
-    assert sample_packet.is_icmp?
-    assert pkt.is_icmp?
-  end
+	include PacketFu
+
+	def test_icmp_header_new
+		i = ICMPHeader.new
+		assert_kind_of ICMPHeader, i
+		assert_equal("\x00\x00\xff\xff", i.to_s)
+		i.icmp_type = 1
+		i.icmp_recalc :icmp_sum
+		assert_equal("\x01\x00\xfe\xff", i.to_s)
+	end
+
+	def test_icmp_peek
+		i = ICMPPacket.new
+		i.ip_saddr = "10.20.30.40"
+		i.ip_daddr = "50.60.70.80"
+		i.payload = "abcdefghijklmnopqrstuvwxyz"
+		i.recalc
+		puts "\n"
+		puts "ICMP Peek format: "
+		puts i.peek
+		assert (i.peek.size <= 80)
+	end
+
+	def test_icmp_pcap
+		i = ICMPPacket.new
+		assert_kind_of ICMPPacket, i
+		i.recalc
+		i.to_f('icmp_test.pcap')
+		i.ip_saddr = "10.20.30.40"
+		i.ip_daddr = "50.60.70.80"
+		i.payload = "\x00\x01\x00\01abcdefghijklmnopqrstuvwxyz"
+		i.icmp_code = 8
+		i.recalc
+		i.to_f('icmp_test.pcap','a')
+		assert File.exists?('icmp_test.pcap')
+	end
+
+	def test_icmp_read
+		sample_packet = PcapFile.new.file_to_array(:f => 'sample.pcap')[2]
+		pkt = Packet.parse(sample_packet)
+		assert pkt.is_icmp?
+		assert_kind_of ICMPPacket, pkt
+		assert_equal(0x4d58, pkt.icmp_sum.to_i)
+		assert_equal(8, pkt.icmp_type.to_i)
+	end
+
+	def test_icmp_reread
+		sample_packet = PacketFu::ICMPPacket.new
+		pkt = Packet.parse(sample_packet.to_s)
+		assert sample_packet.is_icmp?
+		assert pkt.is_icmp?
+	end
 
 end
 

data/test/test_inject.rb

@@ -6,24 +6,24 @@ require 'packetfu'
 
 class InjectTest < Test::Unit::TestCase
 
-  def test_cap
-    assert_nothing_raised { PacketFu::Capture }
-  end
-
-  def test_whoami
-    assert_nothing_raised { PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo')) }
-  end
-
-  def test_to_w
-    assert_equal(Process.euid, 0, "TEST FAIL: This test must be run as root")
-    conf = PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo'))
-    p = PacketFu::UDPPacket.new(:config => conf)
-    p.udp_dport = 12345
-    p.udp_sport = 12345
-    p.payload = "PacketFu test packet"
-    p.recalc
-    assert p.to_w
-  end
+	def test_cap
+		assert_nothing_raised { PacketFu::Capture }
+	end
+
+	def test_whoami
+		assert_nothing_raised { PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo')) }
+	end
+
+	def test_to_w
+		assert_equal(Process.euid, 0, "TEST FAIL: This test must be run as root")
+		conf = PacketFu::Utils.whoami?(:iface => (ENV['IFACE'] || 'lo'))
+		p = PacketFu::UDPPacket.new(:config => conf)
+		p.udp_dport = 12345
+		p.udp_sport = 12345
+		p.payload = "PacketFu test packet"
+		p.recalc
+		assert p.to_w
+	end
 
 end
 

data/test/test_invalid.rb

@@ -4,24 +4,24 @@ $:.unshift File.join(File.expand_path(File.dirname(__FILE__)), "..", "lib")
 require 'packetfu'
 
 class InvalidTest < Test::Unit::TestCase
-  include PacketFu
+	include PacketFu
 
-  def test_create_invalid
-    p = InvalidPacket.new
-    assert_kind_of InvalidPacket, p
-    assert_kind_of Packet, p
-    assert p.is_invalid?
-    assert_equal false, p.is_eth?
-    assert_not_equal EthPacket, p.class
-  end
+	def test_create_invalid
+		p = InvalidPacket.new
+		assert_kind_of InvalidPacket, p
+		assert_kind_of Packet, p
+		assert p.is_invalid?
+		assert_equal false, p.is_eth?
+		assert_not_equal EthPacket, p.class
+	end
 
-  # Sadly, the only way to generate an "InvalidPacket" is
-  # to read a packet that's less than 14 bytes. Otherwise,
-  # it's presumed to be an EthPacket. TODO: Fix this assumption!
-  def test_parse_invalid
-    p = Packet.parse("A" * 13)
-    assert_kind_of InvalidPacket, p
-  end
+	# Sadly, the only way to generate an "InvalidPacket" is
+	# to read a packet that's less than 14 bytes. Otherwise,
+	# it's presumed to be an EthPacket. TODO: Fix this assumption!
+	def test_parse_invalid
+		p = Packet.parse("A" * 13)
+		assert_kind_of InvalidPacket, p
+	end
 
 end