DIY-pcap 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/DIY-pcap.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "diy/version"
+
+Gem::Specification.new do |s|
+  s.name        = "DIY-pcap"
+  s.version     = DIY::PCAP::VERSION
+  s.authors     = ["yafei Lee"]
+  s.email       = ["lyfi2003@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{DIY pcap send and recv}
+  s.description = %q{DIY pcap send and recv}
+  
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+  s.add_dependency "ffi-pcap", ">=0.2.0"
+end

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://ruby.taobao.org"
+
+# Specify your gem's dependencies in DIY-pcap.gemspec
+gem "ffi-pcap", ">=0.2.0"

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/pcap

@@ -0,0 +1,10 @@
+$LOAD_PATH.unshift File.join( File.dirname(__FILE__), '..', 'lib' )
+require 'diy/pcap'
+require 'diy/task'
+
+if ARGV[0].nil?
+  puts "Usage: #{File.basename(__FILE__)} file"
+  exit 0
+end
+
+require File.join( Dir.pwd, ARGV[0] )

data/bin/rpcap

@@ -0,0 +1,12 @@
+$LOAD_PATH.unshift File.join( File.dirname(__FILE__), '..', 'lib' )
+require 'diy/pcap'
+require 'diy/task'
+
+$SERVER = true
+
+if ARGV[0].nil?
+  puts "Usage: #{File.basename(__FILE__)} file"
+  exit 0
+end
+
+require File.join( Dir.pwd, ARGV[0] )

data/lib/DIY-pcap.rb

@@ -0,0 +1,2 @@
+require "diy/version"
+require 'diy/pcap'

data/lib/diy/pcap.rb

@@ -0,0 +1,110 @@
+require 'ffi/pcap'
+require 'logger'
+require 'diy/version'
+module DIY
+  class PCAP
+    def initialize
+      @timeout = 10
+      @dir = Dir.pwd
+      @pkt_stack = []
+      @device_name = FFI::PCap.dump_devices[0][0]
+      yield self
+      @driver = FFI::PCap::Live.new(:dev=>@device_name, :handler => FFI::PCap::Handler, :promisc => true)
+      #~ pause_for_user
+      run
+    end
+    
+    attr_accessor :timeout, :dir, :device_name
+    
+    def send(pkt_dir)
+      @pkt_stack << Packet.new( pkt_dir2pkt(pkt_dir), Packet::SEND)
+    end
+    
+    def recv(pkt_dir)
+      @pkt_stack << Packet.new( pkt_dir2pkt(pkt_dir), Packet::RECV)
+    end
+    
+    def pkt_dir2pkt(dir)
+      File.read( File.join( @dir, dir ) )
+    end
+    
+    def run
+      if $SERVER
+        run_server
+      else
+        run_client
+      end
+    end
+    
+    def pause_for_user
+      puts "Input ENTER for going..."
+      gets
+    end
+    
+    def run_client
+      @pkt_stack.each do |pkt|
+        if pkt.to_outer?
+          send_pkt(pkt.pkt)
+        else
+          recv_pkt(pkt.pkt)
+        end
+      end
+    end
+    
+    def run_server
+      @pkt_stack.each do |pkt|
+        if pkt.to_inner?
+          send_pkt(pkt.pkt)
+        else
+          recv_pkt(pkt.pkt)
+        end
+      end
+    end
+    
+    def send_pkt(pkt)
+      sleep 1
+      logger.info("send pkt: [ #{Time.now} ]#{pkt[0..10].dump}...")
+      @driver.inject(pkt)
+    end
+    
+    def recv_pkt(pkt)
+      logger.info("I hope pkt: #{pkt[0..10].dump}")
+      @driver.loop do |this, new_pkt|
+        #~ logger.info("recv pkt: [ #{new_pkt.time} ]: #{new_pkt.body[0..10].dump}..." )
+        if new_pkt.body == pkt
+          logger.info("recv pkt: [ #{new_pkt.time} ]: #{new_pkt.body[0..10].dump}..." )
+          logger.info "got the same pkt,stop"
+          return true
+        end
+      end
+    end
+    
+    def logger
+      @@logger ||= Logger.new(STDOUT)
+    end
+    
+    def logger=(logger)
+      @@logger = logger
+    end
+    
+  end
+  
+  class Packet
+    SEND = 1
+    RECV = 0
+    def initialize( pkt, pos )
+      @pkt = pkt
+      @pos = pos
+    end
+    
+    def to_outer?
+      @pos == SEND
+    end
+    
+    def to_inner?
+      @pos == RECV
+    end
+    
+    attr_reader :pkt, :pos
+  end
+end

data/lib/diy/sender.rb

@@ -0,0 +1,10 @@
+require 'eventmachine'
+
+
+module DIY
+  class Sender < EM::Connection
+    def notify_readable(*arg)
+    end
+    
+  end
+end

data/lib/diy/task.rb

@@ -0,0 +1,3 @@
+def pcap(&block)
+  DIY::PCAP.new(&block)
+end

data/lib/diy/version.rb

@@ -0,0 +1,5 @@
+module DIY
+  class PCAP
+    VERSION = "0.0.1"
+  end
+end

data/simple/cmd-pcap.rb

@@ -0,0 +1,8 @@
+pcap do |s|  
+  s.dir = File.join( File.dirname(__FILE__), 'pcaps')
+  s.send("r1.dat")
+  s.recv("s1.dat")
+  s.recv("s2.dat")
+  s.recv("s3.dat")
+  s.send("r3.dat")
+end

data/simple/diy-pcap.rb

@@ -0,0 +1,13 @@
+$LOAD_PATH.unshift File.join( File.dirname(__FILE__), '..', 'lib' )
+require 'rubygems'
+require 'diy/pcap'
+
+# client and server
+DIY::PCAP.new do |s|
+  s.dir = File.join( File.dirname(__FILE__), 'pcaps')
+  s.send("r1.dat")
+  s.recv("s1.dat")
+  s.recv("s2.dat")
+  s.recv("s3.dat")
+  s.send("r3.dat")
+end

data/spec/spec_helper.rb

@@ -0,0 +1,17 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification 
+name: DIY-pcap
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- yafei Lee
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-08-30 00:00:00 +08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ffi-pcap
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 2
+        - 0
+        version: 0.2.0
+  type: :runtime
+  version_requirements: *id001
+description: DIY pcap send and recv
+email: 
+- lyfi2003@gmail.com
+executables: 
+- pcap
+- rpcap
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .rspec
+- DIY-pcap.gemspec
+- Gemfile
+- Rakefile
+- bin/pcap
+- bin/rpcap
+- lib/DIY-pcap.rb
+- lib/diy/pcap.rb
+- lib/diy/sender.rb
+- lib/diy/task.rb
+- lib/diy/version.rb
+- simple/cmd-pcap.rb
+- simple/diy-pcap.rb
+- simple/pcaps/r1.dat
+- simple/pcaps/r3-2.dat
+- simple/pcaps/r3.dat
+- simple/pcaps/s1.dat
+- simple/pcaps/s2.dat
+- simple/pcaps/s3.dat
+- simple/pcaps/s4.dat
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: DIY pcap send and recv
+test_files: []
+