Provide more visiblity into inv bundles

Band1to · Apr 15, 2015 · 6912320 · 6912320
1 parent ddfb821
commit 6912320
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/bitcoin/protocol/parser.rb b/lib/bitcoin/protocol/parser.rb
@@ -23,7 +23,7 @@ def log
       #
       def parse_inv(payload, type=:put)
         count, payload = Protocol.unpack_var_int(payload)
-        payload.each_byte.each_slice(36){|i|
+        payload.each_byte.each_slice(36).with_index{|i, idx|
           hash = i[4..-1].reverse.pack("C32")
           case i[0]
           when 1
@@ -34,7 +34,11 @@ def parse_inv(payload, type=:put)
             end
           when 2
             if type == :put
-              @h.on_inv_block(hash)
+              if @h.respond_to?(:on_inv_block_v2)
+                @h.on_inv_block_v2(hash, idx, count)
+              else
+                @h.on_inv_block(hash)
+              end
             else
               @h.on_get_block(hash)
             end