dnet-ffi 0.1.3

data/samples/eth_send_raw.rb

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+
+begin ; require 'rubygems'; rescue LoadError ; end
+require 'dnet'
+include Dnet
+
+unless dev=ARGV.shift
+  STDERR.puts "Usage: #{File.basename $0} interface"
+  exit 1
+end
+
+STDERR.puts "Input data via stdin:"
+data = STDIN.read
+
+begin
+  sent = Eth::Handle.open(dev) {|h| h.eth_send(data) }
+rescue Exception => e
+  STDERR.puts "Error: <#{e.class}> - #{e}"
+  STDERR.puts " ** try running as root?" if e.is_a? Dnet::HandleError
+  exit 1
+end
+
+if sent == data.size
+  puts "Sent: #{sent} bytes"
+  exit 0
+else
+  STDERR.puts "Error: expected #{data.size} sent bytes - got #{sent}"
+  exit 1
+end

data/samples/ifconfig-alike.rb

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+
+begin; require 'rubygems' ; rescue LoadError; end
+require 'dnet'
+
+@arg_if = ARGV.shift
+
+module Dumper
+  def self.dump_if_inet(a)
+    return "#{a.lookup_atype} address: #{a.string} net: #{a.net.string} "+
+           "bcast: #{a.bcast.string}"
+  end
+
+  def self.dump_if_inet6(a)
+    return "#{a.lookup_atype} address: #{a.string} net: #{a.net.string} "
+  end
+
+  def self.dump_if_link(a)
+    return "#{a.lookup_atype} address: #{a.string}"
+  end
+end
+
+Dnet::IntfHandle.each_entry do |entry|
+  next if @arg_if and entry.if_name != @arg_if
+
+  puts("#{entry.if_name}: "+
+       "type=#{entry.lookup_itype.downcase} "+
+       "flags=#{entry.flags}<#{entry.lookup_flags.join(',')}> "+
+       "mtu #{entry.mtu}\n" )
+
+  [:if_addr, :link_addr, :dst_addr].each do |addr_field|
+    addr = entry.send(addr_field)
+    if kind=addr.lookup_atype
+      puts "    " + Dumper.__send__(:"dump_if_#{kind}", addr)
+    end
+  end
+
+  entry.aliases.each do |alias_addr|
+    if kind = alias_addr.lookup_atype
+      puts "    alias: " + Dumper.__send__(:"dump_if_#{kind}", alias_addr)
+    end
+  end
+end
+

data/samples/udp_send_raw.rb

@@ -0,0 +1,74 @@
+#!/usr/bin/env ruby
+# A basic IP/UDP spoofing example using dnet-ffi libdnet ruby bindings.
+#
+# Usage: udp_send_raw.rb src-addr:sport dst-addr:dport [msg]
+#
+# Specify the contents for the udp message as 'msg', or via STDIN if no
+# msg is supplied on the commandline.
+
+begin ; require 'rubygems' ; rescue LoadError; end
+require 'dnet'
+
+include Dnet
+
+arg_rx = %r{^(#{Util::RX_IP4_ADDR}):(\d+)$}
+begin
+  src_arg = ARGV.shift or raise "wrong arguments"
+  dst_arg = ARGV.shift or raise "wrong arguments"
+  raise "Invalid src addr" unless src_arg and sm=arg_rx.match(src_arg)
+  raise "Invalid dst addr" unless dst_arg and dm=arg_rx.match(dst_arg)
+  src = sm[1]
+  sport = sm[2].to_i
+  dst = dm[1]
+  dport = dm[2].to_i
+rescue
+  STDERR.puts "Error: #{$!}"
+  STDERR.puts "Usage: #{File.basename $0} src-addr:sport dst-addr:dport [msg]"
+  exit 1
+end
+
+data=(ARGV.shift || STDIN.read)
+
+blob = Blob.new
+
+udp_sz = Udp::Hdr.size + data.size
+tot_sz = Ip::Hdr.size + udp_sz
+
+
+ip_hdr = Ip::Hdr.new :tos   => 0,
+                     :len   => tot_sz,
+                     :off   => 0, 
+                     :ttl   => 128, 
+                     :proto => Ip::Proto::UDP, 
+                     :src   => src, 
+                     :dst   => dst 
+
+blob.write(ip_hdr.to_ptr,  Ip::Hdr.size)
+
+udp_hdr = Udp::Hdr.new :sport => sport,
+                       :dport => dport,
+                       :len   => udp_sz
+
+blob.write(udp_hdr.to_ptr, Udp::Hdr.size)
+
+blob.write(data)
+
+begin
+  blob.rewind
+  sent = Ip::Handle.open { |h| h.ip_send(blob.read) }
+
+  if sent != tot_sz
+    raise "expected #{tot_sz} bytes sent, but got #{sent}"
+  else
+    STDERR.puts "Sent: #{sent} bytes"
+  end
+rescue Exception => e
+  STDERR.puts "Error: <#{e.class}> - #{e}"
+  STDERR.puts " ** try running as root?" if e.is_a? Dnet::HandleError
+  exit 1
+ensure
+  blob.release
+  blob=nil
+end
+
+exit 0

data/spec/addr_spec.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Addr" do
+  context "dnet(3) addr_* function bindings" do
+    funcs = %w{addr_cmp addr_bcast addr_net addr_ntop addr_pton addr_ntoa addr_aton addr_ntos addr_ston addr_btos addr_stob addr_btom addr_mtob}
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        ::Dnet.respond_to?(func).should == true
+      end
+    end
+
+  end
+
+end

data/spec/arp_spec.rb

@@ -0,0 +1,95 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Arp" do
+  context "dnet(3) arp_* function bindings" do
+    funcs = %w{ arp_open arp_add arp_delete arp_get arp_loop arp_close }
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        ::Dnet.respond_to?(func).should == true
+      end
+    end
+
+  end
+
+  context "the Dnet::Arp module" do
+    it "should be a module" do 
+      Arp.kind_of?(Module).should == true
+    end
+
+    [:open, :each_entry, :entries].each do |meth|
+      it "should provide a method called #{meth}()" do
+        Arp.respond_to?(meth).should == true
+      end
+    end
+
+    context "the each_entry() method" do
+      it "should fire a block for each entry" do
+        i=0
+        Arp.each_entry {|e| i+=1 }
+        i.should_not == 0
+      end
+
+      it "should yield arp entries to a block as Arp::Entry objects" do
+        Arp.each_entry {|e| e.kind_of?(Arp::Entry).should == true }
+      end
+
+    end
+
+    context "the entries() method" do
+      before(:all) do
+        @entries = Arp.entries()
+      end
+
+      it "should return an array" do
+        @entries.kind_of?(Array).should == true
+      end
+
+      it "should produce a non-emtpy list" do
+        @entries.empty?.should == false
+      end
+
+      it "should produce arp entries as Arp::Entry objects" do
+        @entries.each {|e| e.kind_of?(Arp::Entry).should == true }
+      end
+
+    end
+
+  end
+
+  context "the Dnet::Arp::Handle class" do
+    context "instance" do
+      before(:all) do
+        @h = Arp::Handle.new
+      end
+
+      it "should have opened a handle" do
+        @h.handle.kind_of?(::FFI::Pointer).should == true
+      end
+
+      it "should provide a way to get a list of entries" do
+        @h.respond_to?(:entries).should == true
+      end
+
+      it "should provide that list as an array" do
+        @h.entries().kind_of?(Array).should == true
+      end
+
+      it "should provide arp entries in that array" do
+        @h.entries().each{|e| e.kind_of?(Arp::Entry).should == true}
+      end
+
+      it "should provide an iterator for entries" do
+        @h.respond_to?(:loop).should == true
+      end
+
+      it "should provide arp entries through the iterator" do
+        @h.loop {|x| x.kind_of?(Arp::Entry).should == true }
+      end
+
+    end
+
+  end
+
+
+end

data/spec/blob_spec.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Blob" do
+  context "dnet(3) blob_* function bindings" do
+    funcs = %w{ blob_new blob_read blob_write blob_seek blob_index blob_rindex blob_pack blob_unpack blob_print blob_free }
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        ::Dnet.respond_to?(func).should == true
+      end
+    end
+
+  end
+
+end

data/spec/bsd_spec.rb

@@ -0,0 +1,60 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::AF" do
+  context "address family lookups" do
+    it "should map constants for address families" do
+      Dnet::AF.const_get("LINK").should == Socket::AF_LINK
+      Dnet::AF.const_get("INET").should == Socket::AF_INET
+      Dnet::AF.const_get("INET6").should == Socket::AF_INET6
+    end
+
+    it "should identify socket addr family names by number" do
+      Dnet::AF[ Socket::AF_LINK ].should == "LINK"
+      Dnet::AF[ Socket::AF_INET ].should == "INET"
+      Dnet::AF[ Socket::AF_INET6 ].should == "INET6"
+    end
+
+    it "should identify socket addr numbers by string name" do
+      Dnet::AF[ "LINK" ].should == Socket::AF_LINK
+      Dnet::AF[ "INET" ].should == Socket::AF_INET
+      Dnet::AF[ "INET6" ].should == Socket::AF_INET6
+    end
+
+    it "should identify socket addr numbers by symbol name" do
+      Dnet::AF[ :link ].should == Socket::AF_LINK
+      Dnet::AF[ :inet ].should == Socket::AF_INET
+      Dnet::AF[ :inet6 ].should == Socket::AF_INET6
+    end
+
+  end
+end
+
+describe "SockAddrIn" do
+  context "from Socket::pack_sockaddr_in / to unpack_sockaddr_in" do
+    before(:all) do
+      @raw= ::Socket.pack_sockaddr_in(9999, "127.0.0.1")
+      @sa = ::Dnet::SockAddrIn.new :raw => @raw
+    end
+
+    it "should capture a packed sockaddr_in" do
+      @sa.to_ptr.read_string(@sa.size).should == @raw
+    end
+
+    it "should return the correct family value family()" do
+      @sa.family.should == ::Socket::AF_INET
+    end
+
+    it "should return the correct family name with lookup_family()" do
+      @sa.lookup_family.should == "INET"
+    end
+
+    it "should return 8 bytes of null data starting at :_sa_zero" do
+      (@sa.to_ptr + @sa.offset_of(:_sa_zero)).read_string(8).should == "\x00"*8
+    end
+
+    it "should unpack to the correct address" do
+      (::Socket.unpack_sockaddr_in(@sa.to_ptr.read_string(@sa.size))).should == [9999, "127.0.0.1"]
+    end
+  end
+
+end

data/spec/dnet-ffi_spec.rb

@@ -0,0 +1,31 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet" do
+  context "assert..." do
+    # ...
+  end
+  context "Utilities" do
+    [:htonl, :htons, :ntohl, :ntohs].each do |func|
+      it "should have bound the utility C function #{func}()" do
+        Dnet::Util.respond_to?(func).should == true
+      end
+    end
+
+    context "long functions" do
+      [:htonl, :ntohl].each do |func|
+        it "should accept and return 32-bit numbers with #{func}()" do
+          Dnet.__send__(func, 0xffffffff).should == 0xffffffff
+        end
+      end
+    end
+
+    context "short functions" do
+      [:htons, :ntohs].each do |func|
+        it "should accept and return 16-bit numbers with #{func}()" do
+          Dnet.__send__(func, 0xffff).should == 0xffff
+        end
+      end
+    end
+  end
+end
+

data/spec/eth_spec.rb

@@ -0,0 +1,47 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Eth" do
+
+  context "dnet(3) function bindings" do
+    funcs = %w{ eth_open eth_get eth_set eth_send eth_close }
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        ::Dnet.respond_to?(func).should == true
+      end
+    end
+
+  end
+
+  context "the Dnet::Eth module" do
+    it "should be a module" do 
+      Eth.kind_of?(Module).should == true
+    end
+
+    [:open, :eth_send].each do |meth|
+      it "should provide a method called #{meth}()" do
+        Eth.respond_to?(meth).should == true
+      end
+    end
+  end
+
+  context "the Eth::Handle class" do
+    context "instance" do
+      before(:all) do
+        raise "!!! need an env. variable DNET_TEST_INTERFACE" unless NET_DEV
+        begin
+          @h = Eth::Handle.new(NET_DEV)
+        rescue HandleError
+          raise "!!! You may need to be root to run this test -- #{$!}"
+        end
+      end
+
+      it "should have opened a handle" do
+        @h.handle.kind_of?(::FFI::Pointer).should == true
+      end
+
+    end
+
+  end
+
+end

data/spec/fw_spec.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Fw" do
+  context "dnet(3) fw_* function bindings" do
+    funcs = %w{ fw_open fw_add fw_delete fw_loop fw_close }
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        ::Dnet.respond_to?(func).should == true
+      end
+    end
+
+  end
+
+end

data/spec/intf_spec.rb

@@ -0,0 +1,98 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Dnet::Intf" do
+
+  context "dnet(3) intf_* function bindings" do
+    funcs = %w{ intf_open intf_get intf_get_src intf_get_dst intf_set intf_loop intf_close }
+
+    funcs.each do |func|
+      it "should have bound: #{func}()" do
+        respond_to?(func).should == true
+      end
+    end
+
+  end
+
+  context "the Dnet::Intf module" do
+    it "should be a module" do 
+      Intf.kind_of?(Module).should == true
+    end
+
+    [:open, :each_entry, :entries].each do |meth|
+      it "should provide a method called #{meth}()" do
+        Intf.respond_to?(meth).should == true
+      end
+    end
+
+    context "the each_entry() method" do
+      it "should fire a block for each entry" do
+        i=0
+        Intf.each_entry {|e| i+=1 }
+        i.should_not == 0
+      end
+
+      it "should yield interface entries to a block as Intf::Entry objects" do
+        Intf.each_entry {|e| e.kind_of?(Intf::Entry).should == true }
+      end
+
+    end
+
+    context "the entries() method" do
+      before(:all) do
+        @entries = Intf.entries()
+      end
+
+      it "should return an array of entries" do
+        @entries.kind_of?(Array).should == true
+      end
+
+      it "should produce a non-emtpy entries list" do
+        @entries.empty?.should == false
+      end
+
+      it "should produce entries containing interface entries" do
+        @entries.each {|e| e.kind_of?(Intf::Entry).should == true }
+      end
+
+      it "should produce entries with interface name strings" do
+        @entries.each {|e| e.if_name.kind_of?(String).should == true }
+      end
+    end
+
+  end
+
+  context "the Intf::Handle class" do
+    context "instance" do
+      before(:all) do
+        @intf_h = Intf::Handle.new
+      end
+
+      it "should have opened a handle" do
+        @intf_h.handle.kind_of?(::FFI::Pointer).should == true
+      end
+
+      it "should provide a way to get a list entries" do
+        @intf_h.respond_to?(:entries).should == true
+      end
+
+      it "should provide that list as an array" do
+        @intf_h.entries().kind_of?(Array).should == true
+      end
+
+      it "should provide interface entries in that array" do
+        @intf_h.entries().each{|e| e.kind_of?(Intf::Entry).should == true}
+      end
+
+      it "should provide an iterator for entries" do
+        @intf_h.respond_to?(:loop).should == true
+      end
+
+      it "should provide an interface entry through the iterator" do
+        @intf_h.loop {|x| x.kind_of?(Intf::Entry).should == true }
+      end
+
+    end
+
+  end
+
+end