ip_frag 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+spec/helper/*.dat

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gem 'ip_frag'

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 yafei Lee
+
+MIT License
+
+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.md

@@ -0,0 +1,22 @@
+# IpFrag
+
+Help you quickly create fragged packets, used by network test.
+
+
+## Installation
+
+    $ gem install ip_frag
+
+## Usage
+
+    $ ip_gen --size 200 --mtu 1480
+    
+It will generate **2** UDP packets fragged.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/bin/ipgen

@@ -0,0 +1,29 @@
+$LOAD_PATH.unshift File.join( File.dirname(__FILE__), '..', 'lib')
+require 'ip_frag'
+require 'optparse'
+
+options = {
+  :mtu => 1480
+}
+
+OptionParser.new do |opts|
+  opts.on('-s size', '--size size', 'total size, should be bigger than mtu') do |v|
+    options[:size] = v.to_i
+  end
+  
+  opts.on('-m mtu', '--mtu mtu', 'ip mtu, offen less than 1480') do |v|
+    options[:mtu] = v.to_i
+  end
+  
+  opts.on('-h', '--help', 'Show this help and exit') do
+    puts opts
+    exit 0
+  end
+end.parse!
+
+if options[:size].nil?
+  $stderr.puts "less -s argument, please see help for more."
+  exit 1
+end
+
+IPFrag::Generator.new( options[:size], options[:mtu] ).write_dat

data/ip_frag.gemspec

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/ip_frag/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["yafei Lee"]
+  gem.email         = ["lyfi2003@gmail.com"]
+  gem.description   = %q{ip frag tool}
+  gem.summary       = %q{ip frag tool}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "ip_frag"
+  gem.require_paths = ["lib"]
+  gem.version       = IPFrag::VERSION
+  gem.add_dependency "DIY-pcap", ">= 0.3.7"
+end

data/lib/ip_frag/generator.rb

@@ -0,0 +1,131 @@
+module IPFrag
+  class Generator
+    def initialize(size, mtu)
+      @size = size
+      @mtu = mtu
+      check_size_mtu
+    end
+    
+    def split_to_ip_packet
+      ret = []
+      
+      offset = 0
+      ret << make_ip_packet(packet_offset) do |p|
+        p.offset = make_offset(offset)
+      end
+      offset += packet_offset
+      
+      (packet_count - 2).times do
+        ret << make_ip_packet(packet_offset) do |p|
+          p.offset = make_offset(offset)
+        end
+        offset += packet_offset
+      end
+      
+      ret << make_ip_packet(packet_left) do |p|
+        p.offset = make_offset(offset, true)
+      end
+      
+      ret
+    end
+    
+    def split_to_ethernet_packet
+      ip_packets = split_to_ip_packet
+      ip_packets.map do |ip_packet|
+        make_ether_packet do |e|
+          e.payload = ip_packet
+          e.type
+        end
+      end
+    end
+    
+    def write_dat(path = '.')
+      split_to_ethernet_packet.each_with_index do |p, i|
+        file_name = "#{i+1}.dat"
+        file = File.open( File.join(path, file_name), 'w') do |f|
+          p.write(f)
+        end
+      end
+    end
+    
+    # Get total packet size, it depends on @size and @mtu
+    # 
+    #
+    def packet_count
+      count, left = @size.divmod(packet_offset)
+      if left != 0
+        count += 1
+      end
+      count
+    end
+    
+    def packet_offset
+      @mtu & (~7)
+    end
+    
+    def packet_left
+      left = @size % packet_offset
+      left = packet_offset if left == 0
+      left
+    end
+    
+    def make_ip_packet(size)
+      ip = Mu::Pcap::IPv4.new(src_ip, dst_ip)
+      ip.payload = contant(size)
+      ip.proto = Mu::Pcap::IP::IPPROTO_UDP
+      yield ip if block_given?
+      ip
+    end
+    
+    def make_ether_packet
+      packet = Mu::Pcap::Ethernet.new(src_mac, dst_mac)
+      packet.type = Mu::Pcap::Ethernet::ETHERTYPE_IP
+      yield packet
+      packet
+    end
+    
+    def make_offset(offset, last = false)
+      offset = offset >> 3
+      
+      flag = 1
+      if last
+        flag = 0
+      end
+      ( flag << 13) | offset
+    end
+    
+    def check_size_mtu
+      
+      if @size <= @mtu
+        raise "Size#{@size} MUST be big than Mtu#{@mtu}."
+      end
+      
+      if @mtu <= 0
+        raise "Mtu MUST be an integer big than ZERO"
+      end
+    end
+    
+    def src_ip
+      @src_ip ||= '200.200.200.1'
+    end
+    
+    def dst_ip
+      @dst_ip ||= '200.200.200.2'
+    end
+    
+    def src_mac
+      @src_mac ||= '00:00:00:00:00:01'
+    end
+    
+    def dst_mac
+      @dst_mac ||= '00:00:00:00:00:02'
+    end
+    
+    attr_writer :src_ip, :dst_ip, :src_mac, :dst_mac
+    
+    def contant(size)
+      'a' * size
+    end
+  end
+  
+end

data/lib/ip_frag/version.rb

@@ -0,0 +1,3 @@
+module IPFrag
+  VERSION = "0.0.1"
+end

data/lib/ip_frag.rb

@@ -0,0 +1,5 @@
+require "ip_frag/version"
+
+require 'rubygems'
+require 'diy-pcap'
+require 'ip_frag/generator'

data/spec/generator_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe IPFrag do
+  it "show version" do
+    lambda { IPFrag::VERSION }.should_not raise_error
+  end
+  
+  it "#check_size_mtu" do
+    lambda { IPFrag::Generator.new(100, 10) }.should_not raise_error
+    lambda { IPFrag::Generator.new(100, 0) }.should raise_error
+    lambda { IPFrag::Generator.new(10, 10) }.should raise_error
+    lambda { IPFrag::Generator.new(10, -1) }.should raise_error
+  end
+  
+  it "#split" do
+    g = IPFrag::Generator.new(100, 8)
+    g.split_to_ip_packet.size.should == 13
+    g.split_to_ethernet_packet.size.should == 13
+  end
+  
+  it "#write_dat" do
+    g = IPFrag::Generator.new(2000, 1460)
+    g.write_dat('helper')
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,5 @@
+$LOAD_PATH.unshift File.join( File.dirname(__FILE__), '..', 'lib') unless $LOADED
+
+$LOADED = true
+
+require 'ip_frag'

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: ip_frag
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- yafei Lee
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-10-27 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: DIY-pcap
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 29
+        segments: 
+        - 0
+        - 3
+        - 7
+        version: 0.3.7
+  type: :runtime
+  version_requirements: *id001
+description: ip frag tool
+email: 
+- lyfi2003@gmail.com
+executables: 
+- ipgen
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- bin/ipgen
+- ip_frag.gemspec
+- lib/ip_frag.rb
+- lib/ip_frag/generator.rb
+- lib/ip_frag/ip_wrapper.rb
+- lib/ip_frag/version.rb
+- spec/generator_spec.rb
+- spec/spec_helper.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: ip frag tool
+test_files: 
+- spec/generator_spec.rb
+- spec/spec_helper.rb