lib/packetgen/header/header_methods.rb in packetgen-0.1.0 vs lib/packetgen/header/header_methods.rb in packetgen-0.2.0

@@ -40,10 +40,10 @@
       # @return [Header]
       # @raise FormatError no IP or IPv6 header previous +header+ in packet
       # @raise FormatError +header+ not in a packet
       def ip_header(header)
         hid = header_id(header)
-        iph = packet.headers[0...hid].reverse.find { |h| h.is_a? IP }
+        iph = packet.headers[0...hid].reverse.find { |h| h.is_a? IP or h.is_a? IPv6 }
         raise FormatError, 'no IP or IPv6 header in packet' if iph.nil?
         iph
       end
     end
   end