packet.rb in packetgen-1.0.1

- old
+ new

@@ -167,11 +167,11 @@
       if prev_header
         binding = prev_header.class.known_headers[klass]
         if binding.nil?
           msg = "#{prev_header.class} knowns no layer association with #{protocol}. "
           msg << "Try #{prev_header.class}.bind_layer(PacketGen::Header::#{protocol}, "
-          msg << "#{prev_header.class.to_s.gsub(/(.*)::/, '').downcase}_proto_field: "
+          msg << "#{prev_header.protocol_name.downcase}_proto_field: "
           msg << "value_for_#{protocol.downcase})"
           raise ArgumentError, msg
         end
         prev_header[binding.key].read binding.value
         prev_header.body = header
@@ -250,10 +250,10 @@
     def to_w(iface=nil)
       iface ||= PacketGen.default_iface
       if @headers.first.respond_to? :to_w
         @headers.first.to_w(iface)
       else
-        type = @headers.first.class.to_s.gsub(/.*::/, '')
+        type = @headers.first.protocol_name
         raise WireError, "don't known how to send a #{type} packet on wire"
       end
     end
 
     # @return [String]