ffi-pcap 0.2.0

data/.gitignore

@@ -0,0 +1,10 @@
+doc
+rdoc
+pkg
+tmp/*
+ref/*
+.yardoc
+.DS_Store
+*.swp
+*~
+*.gemspec

data/ChangeLog.rdoc

@@ -0,0 +1,27 @@
+=== 0.2.0 / 2010-04-22
+* emonti fork merged back into sophsec/ffi-pcap 
+* ... ignore that whole "caper" thing.  It will return as another lib entirely.
+
+=== 0.1.4 / 2010-04-20 (emonti/ffi-pcap)
+* Fixes and example for pcap dumper
+
+=== 0.1.3 / 2010-03-05 (emonti/ffi-pcap)
+* Minor fixes for ruby 1.9 compatability
+
+=== 0.1.2 / 2010-01-03 (emonti/ffi-pcap)
+* Branched from sophsec/ffi-pcap by emonti
+* Using ffi_dry for common struct interface
+* Redesigned  the pcap-specific pcap_pkthdr struct.
+* Dismantled all other network packet parsing code. 
+* 'Handlers' have been split out into type-specific 'Wrappers' by features.
+* The namespace 'Handler' has been reused instead for pcap_handler abstraction
+* Added filtering support and interfaces for compiling filters into bpf code.
+* Added packet injection on Live pcap interfaces.
+* Tackled some minor Jruby compatability issues.
+* Lots of documentation added throughout with yardoc tags.
+* Lots of misc namespace mods and such, and some general refactoring.
+* Lots of other stuff I'm probably forgetting.
+* specs all pass on OS X, Linux, and Win32(except 1 which is a known issue)
+
+=== 0.1.0 / 2009-04-29 (sophsec/ffi-pcap)
+* Initial release. 

data/LICENSE.txt

@@ -0,0 +1,23 @@
+
+The MIT License
+
+Copyright (c) 2009-2010 Hal Brodigan
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+'Software'), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,30 @@
+= ffi-pcap
+
+* [github.com/sophsec/ffi-pcap](http://github.com/sophsec/ffi-pcap/)
+* [github.com/sophsec/ffi-pcap/issues](http://github.com/sophsec/ffi-pcap/issues)
+* Postmodern (postmodern.mod3 at gmail.com)
+* Eric Monti (esmonti at gmail.com)
+
+== Description
+
+Ruby FFI bindings for libpcap.
+
+== Features
+
+== Examples
+
+== Requirements
+
+* [libpcap](http://www.tcpdump.org/) or [winpcap](http://winpcap.org/)
+* [ffi](http://github.com/ffi/ffi) >= 0.5.0
+* [ffi_dry](http://github.com/emonti/ffi_dry) >= 0.1.9
+
+== Install
+
+    $ sudo gem install ffi-pcap
+
+== License
+
+See {file:LICENSE.txt} for license information.
+
+

data/Rakefile

@@ -0,0 +1,26 @@
+# -*- ruby -*-
+
+require 'rubygems'
+Dir["tasks/*.rb"].each {|rt| require rt }
+require 'rake/clean'
+require './lib/ffi/pcap/version.rb'
+
+# Generate a gem using jeweler
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.rubyforge_project = 'ffi-pcap'
+    gemspec.name = "ffi-pcap"
+    gemspec.summary = "FFI bindings for libpcap"
+    gemspec.email = "postmodern.mod3@gmail.com"
+    gemspec.homepage = "http://github.com/sophsec/ffi-pcap"
+    gemspec.description = "Bindings to libpcap via FFI interface in Ruby."
+    gemspec.authors = ["Postmodern", "Dakrone", "Eric Monti"]
+    gemspec.add_dependency "ffi", ">= 0.5.0"
+    gemspec.add_dependency "ffi_dry", ">= 0.1.9"
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end
+
+# vim: syntax=Ruby

data/VERSION

@@ -0,0 +1 @@
+0.2.0

data/examples/ipfw_divert.rb

@@ -0,0 +1,49 @@
+#!/usr/bin/env ruby
+#
+# Thanks to justfalter(Mike Ryan) for turning me onto Divert Sockets for
+# this example.
+#
+# ipfw add tee 6666 tcp from 192.168.63.128 to any
+# ipfw add tee 6666 tcp from any to 192.168.63.128
+
+require 'ffi/pcap'
+require "socket"
+require 'pp'
+
+unless Process::Sys.getuid == 0  
+  $stderr.puts "Must run #{$0} as root."
+  exit!
+end
+
+IPPROTO_DIVERT = 254
+
+outfile = ARGV.shift
+#outfile = "test_#{$$}.pcap"
+
+# create a dummy pcap handle for dumping
+pcap        = FFI::PCap.open_dead(:datalink => :raw)
+pcap_dumper = pcap.open_dump(outfile)
+
+begin 
+  divert_sock = Socket.open(Socket::PF_INET, Socket::SOCK_RAW, IPPROTO_DIVERT)
+  sockaddr = Socket.pack_sockaddr_in( 6666, '0.0.0.0' )
+  divert_sock.bind(sockaddr)
+
+  puts "ready and waiting...."
+
+  while IO.select([divert_sock], nil, nil)
+    data = divert_sock.recv(65535) # or MTU?
+    pp data
+    pcap_dumper.write_pkt( FFI::PCap::Packet.from_string(data) )
+    pcap_dumper.flush 
+  end
+rescue Errno::EPERM
+  $stderr.puts "Must run #{$0} as root."
+  exit!
+ensure 
+  puts "Closing socket."
+  divert_sock.close
+  puts "Closing pcap dumper."
+  pcap_dumper.close
+  pcap.close
+end

data/examples/print_bytes.rb

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ffi/pcap'
+
+pcap =
+  FFI::PCap::Live.new(:dev => 'en0', 
+                 :promisc => true, 
+                 :handler => FFI::PCap::Handler)
+
+pcap.loop() do |this,pkt|
+  puts "#{pkt.time}:"
+
+  pkt.captured.times {|i| print ' %.2x' % pkt.body_ptr.get_uchar(i) }
+  putc "\n"
+end
+

data/lib/ffi-pcap.rb

@@ -0,0 +1 @@
+require 'ffi/pcap'

data/lib/ffi/pcap.rb

@@ -0,0 +1,42 @@
+begin; require 'rubygems'; rescue LoadError; end
+
+require 'ffi_dry'
+
+module FFI
+module PCap
+  extend FFI::Library
+
+  begin
+    ffi_lib "wpcap"
+  rescue LoadError
+    ffi_lib "pcap"
+  end
+end
+end
+
+require 'ffi/pcap/crt'
+
+require 'ffi/pcap/version'
+require 'ffi/pcap/exceptions'
+
+# FFI typedefs, pointer wrappers, and struct
+require 'ffi/pcap/typedefs'
+require 'ffi/pcap/bsd'
+require 'ffi/pcap/addr'
+require 'ffi/pcap/interface'
+require 'ffi/pcap/file_header'
+require 'ffi/pcap/time_val'
+require 'ffi/pcap/packet_header'
+require 'ffi/pcap/stat'
+require 'ffi/pcap/bpf'
+require 'ffi/pcap/dumper'
+
+# Ruby FFI function bindings, sugar, and misc wrappers
+require 'ffi/pcap/error_buffer'
+require 'ffi/pcap/pcap'
+require 'ffi/pcap/data_link'
+require 'ffi/pcap/packet'
+require 'ffi/pcap/live'
+require 'ffi/pcap/offline'
+require 'ffi/pcap/dead'
+

data/lib/ffi/pcap/addr.rb

@@ -0,0 +1,21 @@
+
+module FFI
+module PCap
+
+  # Representation of an interface address.
+  #
+  # See pcap_addr struct in pcap.h
+  class Addr < FFI::Struct
+    include FFI::DRY::StructHelper
+
+    dsl_layout do
+      p_struct :next,       ::FFI::PCap::Addr
+      p_struct :addr,       ::FFI::PCap::SockAddr, :desc => 'address'
+      p_struct :netmask,    ::FFI::PCap::SockAddr, :desc => 'netmask of the address'
+      p_struct :broadcast,  ::FFI::PCap::SockAddr, :desc => 'broadcast for the address'
+      p_struct :dest_addr,  ::FFI::PCap::SockAddr, :desc => 'p2p destination for address'
+    end
+
+  end
+end
+end

data/lib/ffi/pcap/bpf.rb

@@ -0,0 +1,106 @@
+
+module FFI
+module PCap
+
+  # Includes structures defined in pcap-bpf.h
+
+  # Berkeley Packet Filter instruction data structure.
+  #
+  # See bpf_insn struct in pcap-bpf.h
+  class BPFInstruction < FFI::Struct
+    include FFI::DRY::StructHelper
+
+    dsl_layout do
+      field :code,  :ushort
+      field :jt,    :uchar
+      field :jf,    :uchar
+      field :k,     :bpf_int32
+    end
+
+  end
+
+  # Structure for pcap_compile(), pcap_setfilter(), etc.
+  #
+  # See bpf_program struct in pcap-bpf.h
+  class BPFProgram < FFI::Struct
+    include FFI::DRY::StructHelper
+
+    dsl_layout do
+      field    :bf_len,  :uint
+      field    :bf_insn, :pointer
+    end
+
+    def instructions
+      i = 0
+      sz = BPFInstruction.size()
+      Array.new(self.bf_len) do 
+        ins = BPFInstruction.new( self[:bf_insn] + i )
+        i += sz
+        ins
+      end
+    end
+
+    def free!
+      unless @closed
+        @freed = true
+        FFI::PCap.pcap_freecode(self)
+      end
+    end
+
+    def freed?
+      return @freed == true
+    end
+
+    # Compiles a bpf filter without a pcap device being open. Downside is
+    # no error messages are available, whereas they are when you use 
+    # open_dead() and use compile() on the resulting Dead.
+    #
+    # @param [Hash] opts
+    #   Additional options for compile
+    #
+    # @option opts [optional, DataLink, Integer, String, Symbol] :datalink
+    #   DataLink layer type. The argument type will be resolved to a DataLink
+    #   value if possible. Defaults to data-link layer type NULL.
+    #
+    # @option opts [optional, Integer] :snaplen
+    #   The snapshot length for the filter. Defaults to SNAPLEN
+    #
+    # @option opts [optional, Integer] :optimize
+    #   Optimization flag. 0 means don't optimize. Defaults to 1.
+    #
+    # @option opts [optional, Integer] :netmask
+    #   A 32-bit number representing the IPv4 netmask of the network on which
+    #   packets are being captured. It is only used when checking for IPv4
+    #   broadcast addresses in the filter program. Default: 0 (unspecified 
+    #   netmask)
+    #
+    # @return [BPFProgram]
+    #   If no errors occur, a compiled BPFProgram is returned.
+    #
+    def self.compile(expr, opts={})
+      datalink = (opts[:datalink] || 1)
+      dl = datalink.kind_of?(DataLink) ? datalink : DataLink.new(datalink)
+      slen     = (opts[:snaplen] || DEFAULT_SNAPLEN)
+      optimize = (opts[:optimize] || 1)
+      mask     = (opts[:netmask] || 0)
+      code = BPFProgram.new()
+      r = FFI::PCap.pcap_compile_nopcap(slen, dl.value, code, expr, optimize, mask)
+      raise(LibError, "pcap_compile_nopcap(): unspecified error") if r < 0
+      return code
+    end
+
+  end
+
+
+  attach_function :pcap_compile_nopcap, [:int, :int, BPFProgram, :string, :int, :bpf_uint32], :int
+
+  attach_function :bpf_filter, [BPFInstruction, :pointer, :uint, :uint], :uint
+  attach_function :bpf_validate, [BPFInstruction, :int], :int
+  attach_function :bpf_image, [BPFInstruction, :int], :string
+  attach_function :bpf_dump, [BPFProgram, :int], :void
+  attach_function :pcap_freecode, [BPFProgram], :void
+
+end
+end
+
+

data/lib/ffi/pcap/bsd.rb

@@ -0,0 +1,98 @@
+# Here's where various BSD sockets typedefs and structures go
+# ... good to have around
+# cribbed from dnet-ffi - EM
+
+require 'socket'
+
+module FFI
+module PCap
+  typedef :uint8, :sa_family_t
+  typedef :uint32, :in_addr_t
+  typedef :uint16, :in_port_t
+
+  # contains AF_* constants culled from Ruby's ::Socket
+  module AF
+    include ::FFI::DRY::ConstMap
+    slurp_constants(::Socket, "AF_")
+    def self.list; @@list ||= super() ; end
+  end
+
+  # Common abstract superclass for all sockaddr struct classes
+  #
+  class SockAddrFamily < ::FFI::Struct
+    include ::FFI::DRY::StructHelper
+    
+    # returns an address family name for the :family struct member value
+    def lookup_family
+      AF[ self[:family] ]
+    end
+  end
+
+  # generic sockaddr, always good to have around
+  #
+  class SockAddr < SockAddrFamily
+    dsl_layout do
+      field :len,    :uint8,        :desc => 'total length of struct'
+      field :family, :sa_family_t,  :desc => 'address family (AF_*)'
+      field :data,   :char,         :desc => 'variable length bound by :len'
+    end
+  end
+
+
+  # Used to represent a 32-bit IPv4 address in a sock_addr_in structure
+  # 
+  class InAddr < ::FFI::Struct
+    include ::FFI::DRY::StructHelper
+    dsl_layout { field :in_addr,  :in_addr_t, :desc => 'inet address' }
+  end
+
+  # sockaddr inet, always good to have around
+  #
+  class SockAddrIn < SockAddrFamily
+    dsl_layout do
+      field :len,    :uint8,        :desc => 'length of structure (16)'
+      field :family, :sa_family_t,  :desc => 'address family (AF_INET)'
+      field :port,   :in_port_t,    :desc => '16-bit TCP or UDP port number'
+      field :addr,   :in_addr_t,    :desc => '32-bit IPv4 address'
+      array :_sa_zero, [:uint8,8],  :desc => 'unused'
+    end
+  end
+
+  # Used to represent an IPv6 address in a sock_addr_in6 structure
+  #
+  class In6Addr < ::FFI::Struct
+    include ::FFI::DRY::StructHelper
+    dsl_layout { array :s6_addr, [:uint8, 16], :desc => 'IPv6 address' }
+  end
+
+  # IPv6 socket address
+  #
+  class SockAddrIn6 < SockAddrFamily
+    dsl_layout do
+      field :len,      :uint8,       :desc => 'length of structure(24)'
+      field :family,   :sa_family_t, :desc => 'address family (AF_INET6)'
+      field :port,     :in_port_t,   :desc => 'transport layer port'
+      field :flowinfo, :uint32,      :desc => 'priority & flow label'
+      struct :addr,    ::FFI::PCap::In6Addr,     :desc => 'IPv6 address'
+    end
+  end
+
+
+  # data-link socket address
+  #
+  class SockAddrDl < SockAddrFamily
+    dsl_layout do
+      field :len,       :uint8,   :desc => 'length of structure(variable)'
+      field :family, :sa_family_t, :desc => 'address family (AF_LINK)'
+      field :sdl_index, :uint16,  :desc => 'system assigned index, if > 0'
+      field :dltype,    :uint8,   :desc => 'IFT_ETHER, etc. from net/if_types.h'
+      field :nlen,      :uint8,   :desc => 'name length, from :_data'
+      field :alen,      :uint8,   :desc => 'link-layer addres-length'
+      field :slen,      :uint8,   :desc => 'link-layer selector length'
+      field :_data,     :char,    :desc => 'minimum work area=12, can be larger'
+    end
+  end
+
+end
+end
+