racket 1.0.10

data/lib/racket/misc/vt.rb

@@ -0,0 +1,114 @@
+# $Id: vt.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.
+#
+module Racket
+module Misc
+# Simple class to represent a datastructure that is made up of a 
+# null terminted string followed by an arbitrary number of
+# arbitrarily sized values, followed by a "rest" field.
+class VT 
+  # the value for this VT object
+  attr_accessor :value
+  # the array of types for this VT object
+  attr_accessor :types
+  # everything else
+  attr_accessor :rest
+
+  # Create a new VT which consists of a null terminated string
+  # followed by some number of arbitrarily sized values, as 
+  # specified by +args+
+  def initialize(*args)
+    @lengths = args
+    @types = []
+    @value = ""
+  end 
+
+  # Given +data+, return the value and an array
+  # of the types as dictated by this instance
+  def decode(data)
+    null = data.index(0x00)
+    value = data.unpack("a#{null}")[0]
+    data = data.slice(null+1, data.length)
+  
+    n = 0
+    types = []
+    @lengths.each do |l|
+      types[n] = data.unpack("#{punpack_string(l)}")[0]
+      data = data.slice(l, data.length)
+      n += 1
+    end
+
+    [value, types, data]
+  end
+
+  # Given +data+, set the +value+ and +types+ array
+  # accordingly
+  def decode!(data)
+    @value, @types, @rest = self.decode(data)
+  end
+  
+  # Return a string suitable for use elsewhere
+  def encode
+    s = "#{@value}\000"
+
+    n = 0
+    @lengths.each do |l|
+      s << [@types[n]].pack("#{punpack_string(l)}")
+      n += 1
+    end
+    s
+  end
+
+  def to_s
+    encode
+  end
+
+  def to_str
+    encode
+  end
+
+private
+  # XXX: make this handle arbitrarily sized fields
+  def punpack_string(size)
+    s = ""
+    case size
+        when 1
+          s << "C"
+        when 2
+          s << "n"
+        when 4
+          s << "N"
+        else
+          raise ArgumentError, "Size #{s} not supported"
+      end
+    s
+  end
+
+end
+end
+end
+# vim: set ts=2 et sw=2:

data/lib/racket/racket.rb

@@ -0,0 +1,164 @@
+# $Id: racket.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 'socket'
+require 'racket/racketpart'
+require 'racket/misc'
+require 'racket/l2'
+require 'racket/l3'
+require 'racket/l4'
+require 'racket/l5'
+
+module Racket
+class Racket
+
+  attr_accessor :iface, :mtu, :timeout
+  attr_accessor :layers, :payload
+
+  @@loaded_pcaprub = false
+  begin
+    require 'pcaprub'
+    @@loaded_pcaprub = true
+  rescue ::LoadError
+  end
+
+
+  def initialize(payload="")
+    @layers = []
+    @mtu = 1500
+    @timeout = 10
+    @payload = payload
+    1.upto(7) do |l|
+      self.class.send(:define_method, "layer#{l}", lambda { @layers[l] })
+      self.class.send(:define_method, "l#{l}", lambda { @layers[l] })
+      self.class.send(:define_method, "layer#{l}=", lambda { |x| @layers[l] = x; })
+      self.class.send(:define_method, "l#{l}=", lambda { |x| @layers[l] = x; })
+    end
+  end
+
+  # Assemble all the pieces of this Racket as a string, ready for sending.
+  def pack
+    last_payload = ""
+    orig_payload = ""
+    @layers.compact.reverse.each do |l|
+      # save the original payload
+      orig_payload = l.payload
+      # tack on the last payload in
+      # case fix needs it...
+      l.payload += last_payload
+      if (l.autofix?)
+        l.fix!
+      end
+
+      if (l.payload == orig_payload + last_payload)
+        # payload was not modified by fix, so reset it to what
+        # it used to be
+        l.payload = orig_payload
+      else
+        # payload was modified by fix.  chop off what we added.
+        # XXX: this assumes that what we added is still at the end.  
+        # XXX: this is not always true
+        l.payload = l.payload.slice(0, l.payload.length - last_payload.length)
+      end
+
+      # save this layer for the next guy
+      last_payload += l
+      
+    end
+    
+    payload = ""
+    @layers.compact.each do |l|
+      payload += l
+    end
+    payload
+  end
+
+  # return a pretty interpretation of this packet
+  def pretty
+    s = ""
+    @layers.compact.each do |l|
+      s << "#{l.class}: "
+      s << l.pretty
+      s << "\n"
+    end
+    s
+  end
+      
+
+  # Attempt to figure out which of send2 or send3 needs to be called.
+  def sendpacket
+    if (@layers[2])
+      send2
+    else
+      send3
+    end
+  end
+
+  # Write raw layer2 frames
+  def send2
+    if(not @@loaded_pcaprub)
+      raise RuntimeError, "Could not initialize the pcaprub library (You need pcaprub from SVN (http://rubyforge.org/projects/pcaprub/))" 
+    end
+    begin
+
+      p = Pcap::open_live(@iface, @mtu, false, @timeout)
+    rescue Exception => e
+      puts "Pcap: can't open device '#{@iface}' (#{e})"
+      return
+    end
+
+    begin
+      b = p.inject(pack)
+      #p.pcap_close
+      return b
+    rescue Exception => e
+      puts "Pcap: error while sending packet on '#{@iface}' (#{e})"
+    end
+  end
+
+  # Write raw layer3 frames
+  def send3
+    begin
+      s = Socket.open(Socket::PF_INET, Socket::SOCK_RAW, Socket::IPPROTO_RAW)
+
+      if (Socket.const_defined?('SOL_IP'))
+        s.setsockopt(Socket::SOL_IP, Socket::IP_HDRINCL, true)
+      else
+        # BSD
+        s.setsockopt(Socket::IPPROTO_IP, Socket::IP_HDRINCL, true)
+      end
+    rescue Errno::EPERM
+      raise ArgumentError, "Must run #{$0} as root."
+    end
+
+    return s.send(pack, 0, Socket.pack_sockaddr_in(1024, @layers[3].dst_ip))
+  end
+end
+end
+
+# vim: set ts=2 et sw=2:

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: