ip_frag 0.0.2 → 0.0.3

data/.gitignore

@@ -1,18 +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
+*.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/.travis.yml

@@ -0,0 +1,11 @@
+language: ruby
+rvm:
+  #- "1.8.7"
+  #- "1.9.2"
+  - "1.9.3"
+  #- jruby-18mode # JRuby in 1.8 mode
+  #- jruby-19mode # JRuby in 1.9 mode
+  #- rbx-18mode
+  #- rbx-19mode
+# uncomment this line if your project needs to run something other than `rake`:
+＃ script: rake

data/Gemfile

@@ -1,3 +1,5 @@
-source 'https://rubygems.org'
-
-gem 'ip_frag'
+source 'https://rubygems.org'
+
+gemspec
+gem 'rake'
+gem 'rspec'

data/LICENSE

@@ -1,22 +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
+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

@@ -1,22 +1,22 @@
-# IpFrag
-
+# IpFrag
+
 Help you quickly create fragged packets, used by network test.
 
-
-## Installation
-
-    $ gem install ip_frag
-
-## Usage
-
+
+## Installation
+
+    $ gem install ip_frag
+
+## Usage
+
     $ ipgen --size 2000 --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
+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/Rakefile

@@ -0,0 +1,5 @@
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/ip_frag.gemspec

@@ -1,18 +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"]
+# -*- 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
+  gem.add_dependency "DIY-pcap", ">= 0.3.7"
+end

data/lib/ip_frag.rb

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

data/lib/ip_frag/generator.rb

@@ -8,22 +8,25 @@ module IPFrag
     
     def split_to_ip_packet
       ret = []
-      
+
       offset = 0
       ret << make_ip_packet(packet_offset) do |p|
         p.offset = make_offset(offset)
+        p.payload = udp_offset( contant(udp_size_from(@size), p), 1)
       end
       offset += packet_offset
       
-      (packet_count - 2).times do
+      (packet_count - 2).times do |i|
         ret << make_ip_packet(packet_offset) do |p|
           p.offset = make_offset(offset)
+          p.payload = udp_offset( contant(udp_size_from(@size), p), i+2)
         end
         offset += packet_offset
       end
       
       ret << make_ip_packet(packet_left) do |p|
         p.offset = make_offset(offset, true)
+        p.payload = udp_offset( contant(udp_size_from(@size), p), packet_count)
       end
       
       ret
@@ -71,7 +74,6 @@ module IPFrag
     
     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
@@ -97,7 +99,7 @@ module IPFrag
     def check_size_mtu
       
       if @size <= @mtu
-        raise "Size#{@size} MUST be big than Mtu#{@mtu}."
+        raise "Size #{@size} MUST be big than Mtu #{@mtu}."
       end
       
       if @mtu <= 0
@@ -121,10 +123,36 @@ module IPFrag
       @dst_mac ||= '00:00:00:00:00:02'
     end
     
-    attr_writer :src_ip, :dst_ip, :src_mac, :dst_mac
+    def src_port
+      @src_port ||= 1000
+    end
+    
+    def dst_port
+      @dst_port ||= 2000
+    end
+    
+    
+    attr_writer :src_ip, :dst_ip, :src_mac, :dst_mac, :src_port, :dst_port
+    
+    def udp_offset( udp, packet_num)
+      size = @mtu
+      udp[(packet_num-1)*size..packet_num*size -1 ]
+    end
+    
+    def udp_size_from(ip_size)
+      ip_size - 8
+    end
+    
     
-    def contant(size)
-      'a' * size
+    # 生成特定长度的 UDP 报文
+    def contant(size, ip)
+      return @udp_packet if @udp_packet
+      udp = Mu::Pcap::UDP.new(src_port, dst_port)
+      udp.payload = 'a' * size
+      udp_packet_str = StringIO.new
+      udp.write(udp_packet_str, ip)
+      udp_packet_str.rewind
+      @udp_packet ||= udp_packet_str.read
     end
   end
   

data/lib/ip_frag/version.rb

@@ -1,3 +1,3 @@
-module IPFrag
-  VERSION = "0.0.2"
-end
+module IPFrag
+  VERSION = "0.0.3"
+end

data/spec/generator_spec.rb

@@ -1,4 +1,5 @@
 require 'spec_helper'
+require 'fileutils'
 
 describe IPFrag do
   it "show version" do
@@ -13,13 +14,30 @@ describe IPFrag do
   end
   
   it "#split" do
-    g = IPFrag::Generator.new(100, 8)
+    g = IPFrag::Generator.new(1000, 80)
     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')
+    FileUtils.mkdir('tmp') rescue nil
+    g.write_dat('tmp')
+    Dir["tmp/*"].size.should == 2
+    FileUtils.rm_rf('tmp')
   end
+  
+  it "#65535 split" do
+    g = IPFrag::Generator.new(65535, 696)
+    g.split_to_ip_packet.size.should == 95
+    g.split_to_ethernet_packet.size.should == 95
+  end
+  
+  it "#contant" do
+    g = IPFrag::Generator.new(2000, 1500)
+    ip = mock
+    ip.stub(:pseudo_header).and_return("123")
+    g.contant(g.udp_size_from(1000), ip).size.should == g.udp_size_from(1000) + 8
+  end
+  
 end

metadata

@@ -1,12 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: ip_frag
 version: !ruby/object:Gem::Version 
-  prerelease: false
+  hash: 25
+  prerelease: 
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - yafei Lee
@@ -14,8 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-10-29 00:00:00 +08:00
-default_executable: 
+date: 2012-12-11 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: DIY-pcap
@@ -25,6 +25,7 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
+        hash: 29
         segments: 
         - 0
         - 3
@@ -43,9 +44,11 @@ extra_rdoc_files: []
 
 files: 
 - .gitignore
+- .travis.yml
 - Gemfile
 - LICENSE
 - README.md
+- Rakefile
 - bin/ipgen
 - ip_frag.gemspec
 - lib/ip_frag.rb
@@ -54,7 +57,6 @@ files:
 - lib/ip_frag/version.rb
 - spec/generator_spec.rb
 - spec/spec_helper.rb
-has_rdoc: true
 homepage: ""
 licenses: []
 
@@ -68,6 +70,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      hash: 3
       segments: 
       - 0
       version: "0"
@@ -76,13 +79,14 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      hash: 3
       segments: 
       - 0
       version: "0"
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: ip frag tool