racket2 1.0.9

data/lib/racket/racketpart.rb

@@ -0,0 +1,66 @@
+# $Id: racketpart.rb 14 2008-03-02 05:42:30Z warchild $
+#
+# Copyright (c) 2008, Jon Hart 
+# All rights reserved.
+# 
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#     * Redistributions of source code must retain the above copyright
+#       notice, this list of conditions and the following disclaimer.
+#     * Redistributions in binary form must reproduce the above copyright
+#       notice, this list of conditions and the following disclaimer in the
+#       documentation and/or other materials provided with the distribution.
+#     * Neither the name of the <organization> nor the
+#       names of its contributors may be used to endorse or promote products
+#       derived from this software without specific prior written permission.
+# 
+# THIS SOFTWARE IS PROVIDED BY Jon Hart ``AS IS'' AND ANY
+# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL Jon Hart BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+require 'rubygems'
+require 'bit-struct'
+
+module Racket
+# Every Racket object is made up of numerous sub-parts, namely layers of the OSI stack.
+class RacketPart < BitStruct
+
+  # Boolean indicating whether or not this instance should be
+  # automatically "fixed" prior to be packed and sent.
+  attr_accessor :autofix
+
+  # Should this instance be automatically fixed 
+  # prior to being packed and sent?
+  def autofix?
+    @autofix
+  end
+
+  def initialize(*args)
+    @autofix = true
+    super(*args)
+  end
+
+
+  # Print out all of the fields and all of their values
+  def pretty
+    s  = ""
+    self.fields.each do |f|
+      unless (f.name == "payload")
+        s += "#{f.name}=#{self.send(f.name)} "
+      end
+    end
+    s.gsub(/ $/, '')
+  end
+ 
+  def fix!
+  end
+end
+end
+# vim: set ts=2 et sw=2:

data/test/l2/ts_ethernet.rb

@@ -0,0 +1,22 @@
+# $Id: ts_ethernet.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestEthernet <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L2::Ethernet.new }
+    assert_nothing_raised() { Racket::L2::Ethernet.new(Racket::Misc.randstring(30)) }
+  end
+
+  def test_attrs
+    binary = "\x00\x00\x24\xc1\x8c\x09\x00\x30\x1b\xa0\x63\x16\x08\x00\x45\x10\x00\x3c\x2f\xdf\x40\x00\x40\x06\x89\x17\xc0\xa8\x00\x64\xc0\xa8\x00\x01\x99\xb7\x00\x35\x29\x39\x28\x66\x00\x00\x00\x00\xa0\x02\x16\xd0\xbc\x04\x00\x00\x02\x04\x05\xb4\x04\x02\x08\x0a\x00\x31\x07\xb9\x00\x00\x00\x00\x01\x03\x03\x07"
+    e = Racket::L2::Ethernet.new(binary)
+    assert_equal(e.src_mac, "00:30:1b:a0:63:16")
+    assert_equal(e.dst_mac, "00:00:24:c1:8c:09")
+    assert_equal(e.ethertype, 2048)
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l2/ts_misc.rb

@@ -0,0 +1,23 @@
+# $Id: ts_misc.rb 142 2009-12-13 01:53:14Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestL2Misc <  Test::Unit::TestCase
+  def test_convert
+    (0..rand(1024)).each {
+      len = rand(512)+1
+      mac = Racket::L2::Misc.randommac(len)
+      assert_equal(mac.length, (len*2) + (len-1))
+      long = Racket::L2::Misc.mac2long(mac)
+      assert_equal(mac, Racket::L2::Misc.long2mac(long, len))
+      assert_equal(long, Racket::L2::Misc.mac2long(mac))
+      mac = "00:11:22:33:44:55"
+      string = Racket::L2::Misc.mac2string(mac)
+      assert_equal(mac, Racket::L2::Misc.string2mac(string))
+    }
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l2/ts_vlan.rb

@@ -0,0 +1,15 @@
+# $Id: ts_vlan.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestVLAN <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L2::VLAN.new }
+    assert_nothing_raised() { Racket::L2::VLAN.new(Racket::Misc.randstring(30)) }
+  end
+
+end
+# vim: set ts=2 et sw=2:

data/test/l3/ts_ipv4.rb

@@ -0,0 +1,44 @@
+# $Id: ts_ipv4.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestIPv4 <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L3::IPv4.new }
+    assert_nothing_raised() { Racket::L3::IPv4.new(Racket::Misc.randstring(30)) }
+  end
+
+  def test_attrs
+    binary = "\x45\x10\x00\x3c\x2f\xdf\x40\x00\x40\x06\x89\x17\xc0\xa8\x00\x64\xc0\xa8\x00\x01\x99\xb7\x00\x35\x29\x39\x28\x66\x00\x00\x00\x00\xa0\x02\x16\xd0\xbc\x04\x00\x00\x02\x04\x05\xb4\x04\x02\x08\x0a\x00\x31\x07\xb9\x00\x00\x00\x00\x01\x03\x03\x07"
+    i = Racket::L3::IPv4.new(binary)
+    assert_equal(i.version, 4)
+    assert_equal(i.hlen, 5)
+    assert_equal(i.tos, 16)
+    assert_equal(i.len, 60)
+    assert_equal(i.id, 12255)
+    assert_equal(i.flags, 2)
+    assert_equal(i.foffset, 0)
+    assert_equal(i.ttl, 64)
+    assert_equal(i.protocol, 6)
+    assert_equal(i.checksum, 35095)
+    assert_equal(i.src_ip, "192.168.0.100")
+    assert_equal(i.dst_ip, "192.168.0.1")
+
+    i.src_ip = "1.2.3.4"
+    i.fix!
+    assert_equal(i.checksum, 17950)
+  end
+
+  def test_convert
+    (0..512).each {
+      long = rand(2**32)
+      ipv4 = Racket::L3::Misc.long2ipv4(long)
+      assert_equal(long, Racket::L3::Misc.ipv42long(ipv4))
+      assert_equal(ipv4, Racket::L3::Misc.long2ipv4(long))
+    }
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l3/ts_ipv6.rb

@@ -0,0 +1,26 @@
+# $Id: ts_ipv6.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestIPV6 <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L3::IPv6.new }
+    assert_nothing_raised() { Racket::L3::IPv6.new(Racket::Misc.randstring(30)) }
+  end
+
+  def test_convert
+    long = rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF)
+    ipv6 = Racket::L3::Misc.long2ipv6(long)
+
+    assert_equal(long, Racket::L3::Misc.ipv62long(ipv6))
+    assert_equal(ipv6, Racket::L3::Misc.long2ipv6(long))
+    assert_nothing_raised { Racket::L3::IPv6.new("a:b:c::1") }
+    assert_nothing_raised { Racket::L3::Misc.soll_mcast_addr6(ipv6) }
+    mac = Racket::L2::Misc.randommac
+    assert_nothing_raised { Racket::L3::Misc.soll_mcast_mac(mac) }
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l3/ts_misc.rb

@@ -0,0 +1,31 @@
+# $Id: ts_misc.rb 135 2009-12-06 20:39:40Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestL3Misc <  Test::Unit::TestCase
+  def test_misc 
+    assert_nothing_raised {
+      Racket::L3::Misc.checksum(Racket::Misc.randstring(rand(2048)))
+    }
+
+    mac = "6e:f1:eb:b7:c8:72"
+    ll = "fe80::6cf1:ebff:feb7:c872"
+    assert_equal(ll, Racket::L3::Misc.linklocaladdr(mac))
+
+    ipv4 = Racket::L3::Misc.randomipv4
+    lipv4 = Racket::L3::Misc.ipv42long(ipv4)
+    ipv6 = Racket::L3::Misc.randomipv6
+    lipv6 = Racket::L3::Misc.ipv62long(ipv6)
+
+    assert_equal(ipv4, Racket::L3::Misc.long2ipv4(lipv4))
+    assert_equal(ipv6, Racket::L3::Misc.long2ipv6(lipv6))
+
+    ipv6 = "a:0:0:0f:0:1"
+    zipv6 = "a::f:0:1"
+    assert_equal(zipv6, Racket::L3::Misc.compressipv6(ipv6))
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l4/ts_icmp.rb

@@ -0,0 +1,38 @@
+# $Id: ts_icmp.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestICMP <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L4::ICMP.new }
+    assert_nothing_raised() { Racket::L4::ICMP.new(Racket::Misc.randstring(20)) }
+  end
+
+  def test_raw
+    binary="\x9e\x96\x00\x35\x00\x0c\x69\x4f\x66\x6f\x6f\x0a"
+    binary="\x08\x00\x05\x7e\xab\x58\x00\x01\x78\xf1\x09\x4b\x00\x00\x00\x00\xce\xe6\x05\x00\x00\x00\x00\x00\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b"
+    assert_nothing_raised() { i = Racket::L4::ICMP.new(binary) }
+    i = Racket::L4::ICMP.new(binary)
+    assert_equal(i.type, 8)
+    assert_equal(i.code, 0)
+    assert_equal(i.checksum, 0x057e)
+    assert_equal(i.payload,
+    "\xab\x58\x00\x01\x78\xf1\x09\x4b\x00\x00\x00\x00\xce\xe6\x05\x00\x00\x00\x00\x00\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b\x0c\x0d\x0e\x0f\x0a\x0b")
+  end
+
+  def test_build
+    i = Racket::L4::ICMP.new
+    assert_nothing_raised() {
+      i.type = 8
+      i.code = 0 
+      i.payload = "thisisatest"
+      i.fix!
+    }
+    assert_equal(i.checksum, 30152)
+  end
+  
+end
+# vim: set ts=2 et sw=2:

data/test/l4/ts_tcp.rb

@@ -0,0 +1,55 @@
+# $Id: ts_tcp.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestTCP <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L4::TCP.new }
+    assert_nothing_raised() { Racket::L4::TCP.new(Racket::Misc.randstring(20)) }
+  end
+
+  def test_attrs
+    binary="\x90\xef\x08\xae\x47\x87\xa0\x34\x00\x00\x00\x00\xa0\x02\x16\xd0\xc2\x17\x00\x00\x02\x04\x05\xb4\x04\x02\x08\x0a\x03\x0f\xc8\x6e\x00\x00\x00\x00\x01\x03\x03\x07"
+   
+    assert_nothing_raised() { i = Racket::L4::TCP.new(binary) }
+    i = Racket::L4::TCP.new(binary)
+    assert_equal(i.dst_port, 2222)
+    assert_equal(i.src_port, 37103)
+    assert_equal(i.flag_syn, 1)
+    assert_equal(i.flag_ack, 0)
+    assert_equal(i.seq, 0x4787a034)
+    assert_equal(i.ack, 0)
+    assert_equal(i.window, 5840)
+    assert_equal(i.checksum, 0xc217)
+    assert_equal(i.payload,
+    "\x02\x04\x05\xb4\x04\x02\x08\x0a\x03\x0f\xc8\x6e\x00\x00\x00\x00\x01\x03\x03\x07")
+
+  end
+
+  def test_build_no_options
+    i = Racket::L4::TCP.new
+    assert_nothing_raised() {
+      i.src_port = 44781 
+      i.dst_port = 5190 
+      i.seq = 0x4e3e537a
+      i.ack = 0xbdca7ef8
+      i.flag_ack = 1
+      i.flag_psh = 1
+      i.offset = 5
+      i.window = 61200
+      i.payload = "\x2a\x05\x0b\xca\x00\x00"
+    }
+
+    i.checksum!("192.168.1.10", "64.12.26.128")
+    assert_equal(i.checksum, 0xccf8)
+  end
+  
+  def test_build_with_options
+    i = Racket::L4::TCP.new
+    i.add_option(1, "blah")
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/l4/ts_udp.rb

@@ -0,0 +1,40 @@
+# $Id: ts_udp.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestUDP <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::L4::UDP.new }
+    assert_nothing_raised() { Racket::L4::UDP.new(Racket::Misc.randstring(20)) }
+  end
+
+  def test_raw
+    binary="\x9e\x96\x00\x35\x00\x0c\x69\x4f\x66\x6f\x6f\x0a"
+    assert_nothing_raised() { i = Racket::L4::UDP.new(binary) }
+    i = Racket::L4::UDP.new(binary)
+    assert_equal(i.dst_port, 53)
+    assert_equal(i.src_port, 40598)
+    assert_equal(i.checksum, 0x694f)
+    assert_equal(i.len, 12)
+    assert_equal(i.payload, "foo\n")
+  end
+
+  def test_build
+    i = Racket::L4::UDP.new
+    assert_nothing_raised() {
+      i.src_port = 50243 
+      i.dst_port = 53
+      i.payload = "hhahahahahah"
+      i.fix!("192.168.1.10", "1.2.3.4")
+    }
+
+    assert_equal(i.checksum, 0x2623)
+    assert_equal(i.len, 20)
+
+  end
+  
+end
+# vim: set ts=2 et sw=2:

data/test/misc/ts_lv.rb

@@ -0,0 +1,59 @@
+# $Id: ts_lv.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestLV <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::Misc::LV.new(2) }
+    assert_nothing_raised() { Racket::Misc::LV.new(2,4,1,1,6) }
+  end
+
+  def test_decode
+    t1 = Racket::Misc::LV.new(2)
+    s1 = "\x00\x0b\x73\x70\x6f\x6f\x66\x65\x64\x2e\x6f\x72\x67"
+    t1.decode!(s1)
+    assert_equal(t1.lengths[0], 11)
+    assert_equal(t1.values[0], "spoofed.org")
+
+    t2 = Racket::Misc::LV.new(1,2,2,4)
+    s2 = "\x02\xab\xcd\x00\x01\xff\x00\x04\xde\xad\xba\xbe\x00\x00\x00\x0a\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffwtf?"
+    t2.decode!(s2)
+    assert_equal(t2.lengths[0], 2)
+    assert_equal(t2.values[0], "\xab\xcd")
+    assert_equal(t2.lengths[1], 1)
+    assert_equal(t2.values[1], "\xff")
+    assert_equal(t2.lengths[2], 4)
+    assert_equal(t2.values[2], "\xde\xad\xba\xbe")
+    assert_equal(t2.lengths[3], 10)
+    assert_equal(t2.values[3], "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff")
+    assert_equal(t2.rest, "wtf?")
+  end
+
+  def test_encode
+    t1 = Racket::Misc::LV.new(2)
+    s1 = "\x00\x0bspoofed.org"
+    t1.decode!(s1)
+    assert_equal(t1.encode, s1)
+    assert_equal("#{t1}", s1)
+
+    t2 = Racket::Misc::LV.new(1,2,2)
+    s2 = "\x01\xff\x00\x02\xff\xff\x00\x0bspoofed.org\xff\xff\xffwtf?"
+    t2.decode!(s2)
+    assert_equal(t2.encode, s2.slice(0, s2.size - 7))
+    assert_equal("#{t2}", s2.slice(0, s2.size - 7))
+
+    t3 = Racket::Misc::LV.new(1,1,1)
+    t3.lengths = [2,2,2]
+    t3.values = ["ww", "tt", "ff"]
+    assert_equal(t3.encode, "\x02ww\x02tt\x02ff")
+
+    t3 = Racket::Misc::LV.new(1,1,1)
+    t3.lengths = [2,2,4]
+    t3.values = ["ww", "tt", "ff"]
+    assert_equal(t3.encode, "\x02ww\x02tt\x04ff\00\00")
+  end
+end
+# vim: set ts=2 et sw=2:

data/test/misc/ts_orderedhash.rb

@@ -0,0 +1,33 @@
+# $Id: ts_orderedhash.rb 129 2009-11-29 01:37:06Z jhart $
+
+$:.unshift File.join(File.dirname(__FILE__), "../..", "lib")
+
+require 'test/unit'
+require 'racket'
+
+class TestBitField <  Test::Unit::TestCase
+  def test_init
+    assert_nothing_raised() { Racket::Misc::OrderedHash.new }
+  end
+  def test_order
+    oh = Racket::Misc::OrderedHash.new
+    ("a".."z").each { |c|
+        oh[c] = c.unpack("c")[0]
+    } 
+
+    97.upto(122) { |v| 
+      assert_equal(oh[sprintf("%c", v)], v)      
+    } 
+
+    assert_nil(oh['kajf'])
+    
+    copy = []
+    oh.each_key { |k| copy << k }
+    0.upto(25) { |v|
+      assert_equal(copy[v], sprintf("%c", 97 + v))
+    }
+
+    assert_nil(copy[205])
+  end
+end
+# vim: set ts=2 et sw=2: