packetgen 3.1.3 → 3.1.4

data/lib/packetgen/header/ip/addr.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class IP
@@ -60,6 +60,11 @@ module PacketGen
         def mcast?
           self.a1 >= 224 && self.a1 <= 239
         end
+
+        def ==(other)
+          other.is_a?(self.class) &&
+            fields.all? { |attr| self[attr].value == other[attr].value }
+        end
       end
     end
   end

data/lib/packetgen/header/ip/option.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class IP
@@ -83,7 +83,7 @@ module PacketGen
 
         # Factory to build an option from its type
         # @return [Option]
-        def Option.build(options={})
+        def self.build(options={})
           type = options.delete(:type)
           klass = case type
                   when String
@@ -99,9 +99,7 @@ module PacketGen
         def initialize(options={})
           unless options[:type]
             opt_name = self.class.to_s.gsub(/.*::/, '')
-            if Option.const_defined? "#{opt_name}_TYPE"
-              options[:type] = Option.const_get("#{opt_name}_TYPE")
-            end
+            options[:type] = Option.const_get("#{opt_name}_TYPE") if Option.const_defined? "#{opt_name}_TYPE"
           end
           super
           self.length = sz if respond_to?(:length) && options[:length].nil?

data/lib/packetgen/header/ip/options.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class IP
@@ -19,9 +19,7 @@ module PacketGen
         # @return [String]
         def to_s
           str = super
-          if str.length % 4 != 0
-            str += ([0] * (4 - (str.length % 4))).pack('C*')
-          end
+          str += ([0] * (4 - (str.length % 4))).pack('C*') if str.length % 4 != 0
           str
         end
 

data/lib/packetgen/header/ipv6.rb

@@ -1,11 +1,11 @@
 # coding: utf-8
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 require 'ipaddr'
 
 module PacketGen

data/lib/packetgen/header/ipv6/addr.rb

@@ -1,11 +1,11 @@
 # coding: utf-8
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 require 'ipaddr'
 
 module PacketGen
@@ -85,6 +85,11 @@ module PacketGen
         def mcast?
           self.a1 & 0xff00 == 0xff00
         end
+
+        def ==(other)
+          other.is_a?(self.class) &&
+            fields.all? { |attr| self[attr].value == other[attr].value }
+        end
       end
 
       # Class to handle series of IPv6 addresses

data/lib/packetgen/header/ipv6/extension.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class IPv6

data/lib/packetgen/header/ipv6/hop_by_hop.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class IPv6
@@ -106,7 +106,7 @@ module PacketGen
         # @!attribute options
         #  Specific options of extension header
         #  @return [Options]
-        define_field_before :body, :options, Options, builder: ->(h, t) { t.new(length_from: -> { h.real_length - 2} ) }
+        define_field_before :body, :options, Options, builder: ->(h, t) { t.new(length_from: -> { h.real_length - 2 }) }
       end
     end
 

data/lib/packetgen/header/llc.rb

@@ -1,11 +1,11 @@
 # coding: utf-8
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # Logical-Link Control header

data/lib/packetgen/header/mdns.rb

@@ -1,11 +1,11 @@
 # coding: utf-8
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # Multicast DNS.

data/lib/packetgen/header/mld.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # This class supports MLDv1 (RFC 2710).

data/lib/packetgen/header/mldv2.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # This module contains all MLDv2 specific classes.

data/lib/packetgen/header/mldv2/mcast_address_record.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     module MLDv2

data/lib/packetgen/header/mldv2/mlq.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     module MLDv2

data/lib/packetgen/header/mldv2/mlr.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 require_relative 'mcast_address_record'
 
 module PacketGen

data/lib/packetgen/header/ospfv2.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # This class supports OSPFv2 (RFC 2328).
@@ -75,18 +75,18 @@ module PacketGen
 
       # OSPF packet types
       TYPES = {
-        'HELLO'          => 1,
+        'HELLO' => 1,
         'DB_DESCRIPTION' => 2,
-        'LS_REQUEST'     => 3,
-        'LS_UPDATE'      => 4,
-        'LS_ACK'         => 5
+        'LS_REQUEST' => 3,
+        'LS_UPDATE' => 4,
+        'LS_ACK' => 5
       }.freeze
 
       # Authentication types
       AU_TYPES = {
-        'NO_AUTH'         => 0,
-        'PASSWORD'        => 1,
-        'CRYPTO'          => 2,
+        'NO_AUTH' => 0,
+        'PASSWORD' => 1,
+        'CRYPTO' => 2,
         'CRYPTO_WITH_ESN' => 3
       }.freeze
 

data/lib/packetgen/header/ospfv2/db_description.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2

data/lib/packetgen/header/ospfv2/hello.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2

data/lib/packetgen/header/ospfv2/ls_ack.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2

data/lib/packetgen/header/ospfv2/ls_request.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2

data/lib/packetgen/header/ospfv2/ls_update.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2

data/lib/packetgen/header/ospfv2/lsa.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2
@@ -204,9 +204,7 @@ module PacketGen
         private
 
         def record_from_hash(hsh)
-          unless hsh.key? :type
-            raise ArgumentError, 'hash should have :type key'
-          end
+          raise ArgumentError, 'hash should have :type key' unless hsh.key? :type
 
           klass = if @only_headers
                     LSAHeader

data/lib/packetgen/header/ospfv2/lsa_header.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv2
@@ -31,11 +31,11 @@ module PacketGen
       class LSAHeader < Types::Fields
         # LSA Types
         TYPES = {
-          'Router'       => 1,
-          'Network'      => 2,
-          'Summary-IP'   => 3,
+          'Router' => 1,
+          'Network' => 2,
+          'Summary-IP' => 3,
           'Summary-ABSR' => 4,
-          'AS-External'  => 5
+          'AS-External' => 5
         }.freeze
 
         # @!attribute age

data/lib/packetgen/header/ospfv3.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     # This class supports OSPFv3 (RFC 5340).

data/lib/packetgen/header/ospfv3/db_description.rb

@@ -1,10 +1,10 @@
+# frozen_string_literal: true
+
 # This file is part of PacketGen
 # See https://github.com/sdaubert/packetgen for more informations
 # Copyright (C) 2016 Sylvain Daubert <sylvain.daubert@laposte.net>
 # This program is published under MIT license.
 
-# frozen_string_literal: true
-
 module PacketGen
   module Header
     class OSPFv3