racket2 1.0.9

data/lib/racket/misc/orderedhash.rb

@@ -0,0 +1,63 @@
+# $Id: orderedhash.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 ordered hash. 
+#
+# XXX: todo -- add a method for moving/shifting
+# members around
+class OrderedHash < Hash
+  def initialize
+    @keys = []
+  end
+
+  def []=(key, val)
+    @keys << key unless (self[key])
+    super
+  end
+
+  def delete(key)
+    @keys.delete(key)
+    super
+  end
+
+  def each
+    @keys.each { |k| yield k, self[k] }
+  end
+
+  def each_key
+    @keys.each { |k| yield k }
+  end
+
+  def each_value
+    @keys.each { |k| yield self[k] }
+  end
+end
+end
+end
+# vim: set ts=2 et sw=2:

data/lib/racket/misc/raw.rb

@@ -0,0 +1,35 @@
+# $Id:$
+#
+# 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
+# A raw part of a packet that only has a payload
+class Raw < RacketPart
+  rest :payload
+end
+end
+# vim: set ts=2 et sw=2:
+

data/lib/racket/misc/tlv.rb

@@ -0,0 +1,103 @@
+# $Id: tlv.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 for your average type, length, value datastructure.
+# Everything after the TLV is stuff into +rest+
+class TLV
+  attr_accessor :type, :length, :value, :rest, :lbytes, :tlinclude
+
+  # Create a new TLV which requires +ts+ bytes for the type field
+  # and +ls+ bytes for the length field, where (optionally) the value
+  # in +length+ is a multiple of +lbytes+ and (optionally) whether or not
+  # the length field indicates the total length of the TLV of just that of 
+  # the value
+  def initialize(ts, ls, lbytes=1, tlinclude=false)
+    @ts = ts
+    @ls = ls
+    @lbytes = lbytes
+    @tlinclude = tlinclude
+  end 
+
+  # Given +data+, return the type, length, value and rest 
+  # values as dictated by this instance.
+  def decode(data)
+    s = "#{punpack_string(@ts)}#{punpack_string(@ls)}"
+    type, length, tmp = data.unpack("#{s}a*")
+    if (type.nil? or length.nil?)
+      nil
+    else
+      elength = (length * lbytes) - (@tlinclude ? (@ls + @ts) : 0) 
+      value, rest = tmp.unpack("a#{elength}a*")
+      if (value.empty? and length > 0)
+        nil
+      else
+        [type, length, value, rest]
+      end
+    end
+  end
+
+  def decode!(data)
+    @type, @length, @value, @rest = self.decode(data)
+  end
+
+  # Return a string suitable for use elswhere.
+  def encode
+    s = "#{punpack_string(@ts)}#{punpack_string(@ls)}"
+    [@type, @length, @value].pack("#{s}a*")
+  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} is not a supported conversion size"
+      end
+    s
+  end
+  
+end
+end
+end
+# vim: set ts=2 et sw=2:

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: