ip_frag 0.0.3 → 0.0.4

data/lib/ip_frag/generator.rb

@@ -12,21 +12,21 @@ module IPFrag
       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)
+        p.payload = udp_offset( contant(udp_size_from(@size), p), offset, packet_offset)
       end
       offset += packet_offset
       
       (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)
+          p.payload = udp_offset( contant(udp_size_from(@size), p), offset, packet_offset)
         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)
+        p.payload = udp_offset( contant(udp_size_from(@size), p), offset, packet_left)
       end
       
       ret
@@ -134,9 +134,8 @@ module IPFrag
     
     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 ]
+    def udp_offset( udp, start, endt)
+      udp[start..(start + endt-1)]
     end
     
     def udp_size_from(ip_size)

data/lib/ip_frag/version.rb

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

data/spec/generator_spec.rb

@@ -21,9 +21,18 @@ describe IPFrag do
   
   it "#write_dat" do
     g = IPFrag::Generator.new(2000, 1460)
+    FileUtils.rm_rf('tmp')
     FileUtils.mkdir('tmp') rescue nil
     g.write_dat('tmp')
     Dir["tmp/*"].size.should == 2
+    
+  end
+
+  it "#write_dat2" do
+    g = IPFrag::Generator.new(60000, 700)
+    FileUtils.mkdir('tmp') rescue nil
+    g.write_dat('tmp')
+    Dir["tmp/*"].size.should == 87
     FileUtils.rm_rf('tmp')
   end
   

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: ip_frag
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 23
   prerelease: 
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - yafei Lee