net-dns 0.8.0 → 0.20.0

data/lib/net/dns/packet.rb

@@ -6,7 +6,6 @@ require 'net/dns/rr'
 
 module Net
   module DNS
-
     #
     # = Net::DNS::Packet
     #
@@ -91,9 +90,13 @@ module Net
       class PacketError < Error
       end
 
-
-      attr_reader :header, :question, :answer, :authority, :additional
-      attr_reader :answerfrom, :answersize
+      attr_reader :header
+      attr_reader :question
+      attr_reader :answer
+      attr_reader :authority
+      attr_reader :additional
+      attr_reader :answerfrom
+      attr_reader :answersize
 
       # Creates a new instance of <tt>Net::DNS::Packet</tt> class. Arguments are the
       # canonical name of the resource, an optional type field and an optional
@@ -107,7 +110,7 @@ module Net
       # This class no longer instantiate object from binary data coming from
       # network streams. Please use <tt>Net::DNS::Packet.parse</tt> instead.
       def initialize(name, type = Net::DNS::A, cls = Net::DNS::IN)
-        @header = Net::DNS::Header.new(:qdCount => 1)
+        @header = Net::DNS::Header.new(qdCount: 1)
         @question = [Net::DNS::Question.new(name, type, cls)]
         @answer = []
         @authority = []
@@ -116,7 +119,6 @@ module Net
         @logger.level = $DEBUG ? Logger::DEBUG : Logger::WARN
       end
 
-
       # Checks if the packet is a QUERY packet
       def query?
         @header.opCode == Net::DNS::Header::QUERY
@@ -129,7 +131,7 @@ module Net
       #   puts "Packet is #{packet_data.size} bytes long"
       #
       def data
-        qdcount=ancount=nscount=arcount=0
+        qdcount = ancount = nscount = arcount = 0
         data = @header.data
         headerlength = data.length
 
@@ -138,15 +140,15 @@ module Net
           qdcount += 1
         end
         @answer.each do |rr|
-          data += rr.data#(data.length)
+          data += rr.data # (data.length)
           ancount += 1
         end
         @authority.each do |rr|
-          data += rr.data#(data.length)
+          data += rr.data # (data.length)
           nscount += 1
         end
         @additional.each do |rr|
-          data += rr.data#(data.length)
+          data += rr.data # (data.length)
           arcount += 1
         end
 
@@ -168,33 +170,33 @@ module Net
       def data_comp
         offset = 0
         compnames = {}
-        qdcount=ancount=nscount=arcount=0
+        qdcount = ancount = nscount = arcount = 0
         data = @header.data
         headerlength = data.length
 
         @question.each do |question|
-          str,offset,names = question.data
+          str, offset, names = question.data
           data += str
           compnames.update(names)
           qdcount += 1
         end
 
         @answer.each do |rr|
-          str,offset,names = rr.data(offset,compnames)
+          str, offset, names = rr.data(offset, compnames)
           data += str
           compnames.update(names)
           ancount += 1
         end
 
         @authority.each do |rr|
-          str,offset,names = rr.data(offset,compnames)
+          str, offset, names = rr.data(offset, compnames)
           data += str
           compnames.update(names)
           nscount += 1
         end
 
         @additional.each do |rr|
-          str,offset,names = rr.data(offset,compnames)
+          str, offset, names = rr.data(offset, compnames)
           data += str
           compnames.update(names)
           arcount += 1
@@ -212,39 +214,39 @@ module Net
       # of this <tt>Net::DNS::Packet</tt> instance.
       def inspect
         retval = ""
-        if @answerfrom != "0.0.0.0:0" and @answerfrom
-          retval += ";; Answer received from #@answerfrom (#{@answersize} bytes)\n;;\n"
+        if (@answerfrom != "0.0.0.0:0") && @answerfrom
+          retval += ";; Answer received from #{@answerfrom} (#{@answersize} bytes)\n;;\n"
         end
 
         retval += ";; HEADER SECTION\n"
         retval += @header.inspect
 
         retval += "\n"
-        section = (@header.opCode == "UPDATE") ? "ZONE" : "QUESTION"
+        section = @header.opCode == "UPDATE" ? "ZONE" : "QUESTION"
         retval += ";; #{section} SECTION (#{@header.qdCount} record#{@header.qdCount == 1 ? '' : 's'}):\n"
         @question.each do |qr|
           retval += ";; " + qr.inspect + "\n"
         end
 
-        unless @answer.size == 0
+        unless @answer.empty?
           retval += "\n"
-          section = (@header.opCode == "UPDATE") ? "PREREQUISITE" : "ANSWER"
+          section = @header.opCode == "UPDATE" ? "PREREQUISITE" : "ANSWER"
           retval += ";; #{section} SECTION (#{@header.anCount} record#{@header.anCount == 1 ? '' : 's'}):\n"
           @answer.each do |rr|
             retval += rr.inspect + "\n"
           end
         end
 
-        unless @authority.size == 0
+        unless @authority.empty?
           retval += "\n"
-          section = (@header.opCode == "UPDATE") ? "UPDATE" : "AUTHORITY"
+          section = @header.opCode == "UPDATE" ? "UPDATE" : "AUTHORITY"
           retval += ";; #{section} SECTION (#{@header.nsCount} record#{@header.nsCount == 1 ? '' : 's'}):\n"
           @authority.each do |rr|
             retval += rr.inspect + "\n"
           end
         end
 
-        unless @additional.size == 0
+        unless @additional.empty?
           retval += "\n"
           retval += ";; ADDITIONAL SECTION (#{@header.arCount} record#{@header.arCount == 1 ? '' : 's'}):\n"
           @additional.each do |rr|
@@ -254,7 +256,7 @@ module Net
 
         retval
       end
-      alias_method :to_s, :inspect
+      alias to_s inspect
 
       # Delegates to <tt>Net::DNS::Header#truncated?</tt>.
       def truncated?
@@ -264,11 +266,9 @@ module Net
       # Assigns a <tt>Net::DNS::Header</tt> <tt>object</tt>
       # to this <tt>Net::DNS::Packet</tt> instance.
       def header=(object)
-        if object.kind_of? Net::DNS::Header
-          @header = object
-        else
-          raise ArgumentError, "Argument must be a Net::DNS::Header object"
-        end
+        raise ArgumentError, "Argument must be a Net::DNS::Header object" unless object.is_a? Net::DNS::Header
+
+        @header = object
       end
 
       # Assigns a <tt>Net::DNS::Question</tt> <tt>object</tt>
@@ -276,11 +276,10 @@ module Net
       def question=(object)
         case object
         when Array
-          if object.all? {|x| x.kind_of? Net::DNS::Question}
-            @question = object
-          else
-            raise ArgumentError, "Some of the elements is not an Net::DNS::Question object"
-          end
+          raise ArgumentError, "Some of the elements is not an Net::DNS::Question object" unless object.all? { |x| x.is_a? Net::DNS::Question }
+
+          @question = object
+
         when Net::DNS::Question
           @question = [object]
         else
@@ -292,16 +291,15 @@ module Net
       # to the answer section of this <tt>Net::DNS::Packet</tt> instance.
       def answer=(object)
         case object
-          when Array
-            if object.all? {|x| x.kind_of? Net::DNS::RR}
-              @answer = object
-            else
-              raise ArgumentError, "Some of the elements is not an Net::DNS::RR object"
-            end
-          when Net::DNS::RR
-            @answer = [object]
-          else
-            raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
+        when Array
+          raise ArgumentError, "Some of the elements is not an Net::DNS::RR object" unless object.all? { |x| x.is_a? Net::DNS::RR }
+
+          @answer = object
+
+        when Net::DNS::RR
+          @answer = [object]
+        else
+          raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
         end
       end
 
@@ -309,16 +307,15 @@ module Net
       # to the additional section of this <tt>Net::DNS::Packet</tt> instance.
       def additional=(object)
         case object
-          when Array
-            if object.all? {|x| x.kind_of? Net::DNS::RR}
-              @additional = object
-            else
-              raise ArgumentError, "Some of the elements is not an Net::DNS::RR object"
-            end
-          when Net::DNS::RR
-            @additional = [object]
-          else
-            raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
+        when Array
+          raise ArgumentError, "Some of the elements is not an Net::DNS::RR object" unless object.all? { |x| x.is_a? Net::DNS::RR }
+
+          @additional = object
+
+        when Net::DNS::RR
+          @additional = [object]
+        else
+          raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
         end
       end
 
@@ -326,16 +323,15 @@ module Net
       # to the authority section of this <tt>Net::DNS::Packet</tt> instance.
       def authority=(object)
         case object
-          when Array
-            if object.all? {|x| x.kind_of? Net::DNS::RR}
-              @authority = object
-            else
-              raise ArgumentError, "Some of the elements is not an Net::DNS::RR object"
-            end
-          when Net::DNS::RR
-            @authority = [object]
-          else
-            raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
+        when Array
+          raise ArgumentError, "Some of the elements is not an Net::DNS::RR object" unless object.all? { |x| x.is_a? Net::DNS::RR }
+
+          @authority = object
+
+        when Net::DNS::RR
+          @authority = [object]
+        else
+          raise ArgumentError, "Invalid argument, not a RR object nor an array of objects"
         end
       end
 
@@ -348,9 +344,10 @@ module Net
       #
       # As you can see in the documentation for the <tt>Net::DNS::RR::A</tt> class,
       # the address returned is an instance of <tt>IPAddr</tt> class.
-      def each_address(&block)
+      def each_address
         @answer.each do |elem|
           next unless elem.class == Net::DNS::RR::A
+
           yield elem.address
         end
       end
@@ -362,9 +359,10 @@ module Net
       #     puts "Nameserver found: #{ns}"
       #   end
       #
-      def each_nameserver(&block)
+      def each_nameserver
         @answer.each do |elem|
           next unless elem.class == Net::DNS::RR::NS
+
           yield elem.nsdname
         end
       end
@@ -376,9 +374,10 @@ module Net
       #     puts "Mail exchange #{name} has preference #{pref}"
       #   end
       #
-      def each_mx(&block)
+      def each_mx
         @answer.each do |elem|
           next unless elem.class == Net::DNS::RR::MX
+
           yield elem.preference, elem.exchange
         end
       end
@@ -390,9 +389,10 @@ module Net
       #     puts "Canonical name: #{cname}"
       #   end
       #
-      def each_cname(&block)
+      def each_cname
         @answer.each do |elem|
           next unless elem.class == Net::DNS::RR::CNAME
+
           yield elem.cname
         end
       end
@@ -404,9 +404,10 @@ module Net
       #     puts "Pointer for resource: #{ptr}"
       #   end
       #
-      def each_ptr(&block)
+      def each_ptr
         @answer.each do |elem|
           next unless elem.class == Net::DNS::RR::PTR
+
           yield elem.ptrdname
         end
       end
@@ -436,7 +437,6 @@ module Net
         header.rCode.code == Net::DNS::Header::RCode::NAME
       end
 
-
       # Creates a new instance of <tt>Net::DNS::Packet</tt> class from binary data,
       # taken out from a network stream. For example:
       #
@@ -459,105 +459,95 @@ module Net
 
       private
 
-        # New packet from binary data
-        def new_from_data(data, from = nil)
-          unless from
-            if data.kind_of? Array
-              data, from = data
-            else
-              from = [0, 0, "0.0.0.0", "unknown"]
-            end
+      # New packet from binary data
+      def new_from_data(data, from = nil)
+        unless from
+          if data.is_a? Array
+            data, from = data
+          else
+            from = [0, 0, "0.0.0.0", "unknown"]
           end
+        end
 
-          @answerfrom = from[2] + ":" + from[1].to_s
-          @answersize = data.size
-          @logger = Logger.new $stdout
-          @logger.level = $DEBUG ? Logger::DEBUG : Logger::WARN
-
-          #------------------------------------------------------------
-          # Header section
-          #------------------------------------------------------------
-          offset = Net::DNS::HFIXEDSZ
-          @header = Net::DNS::Header.parse(data[0..offset-1])
-
-          @logger.debug ";; HEADER SECTION"
-          @logger.debug @header.inspect
-
-          #------------------------------------------------------------
-          # Question section
-          #------------------------------------------------------------
-          section = @header.opCode == "UPDATE" ? "ZONE" : "QUESTION"
-          @logger.debug ";; #{section} SECTION (#{@header.qdCount} record#{@header.qdCount == 1 ? '': 's'})"
-
-          @question = []
-          @header.qdCount.times do
-            qobj,offset = parse_question(data,offset)
-            @question << qobj
-            @logger.debug ";; #{qobj.inspect}"
-          end
+        @answerfrom = from[2] + ":" + from[1].to_s
+        @answersize = data.size
+        @logger = Logger.new $stdout
+        @logger.level = $DEBUG ? Logger::DEBUG : Logger::WARN
 
-          #------------------------------------------------------------
-          # Answer/prerequisite section
-          #------------------------------------------------------------
-          section = @header.opCode == "UPDATE" ? "PREREQUISITE" : "ANSWER"
-          @logger.debug ";; #{section} SECTION (#{@header.qdCount} record#{@header.qdCount == 1 ? '': 's'})"
-
-          @answer = []
-          @header.anCount.times do
-            begin
-              rrobj,offset = Net::DNS::RR.parse_packet(data,offset)
-              @answer << rrobj
-              @logger.debug rrobj.inspect
-            rescue NameError => e
-              warn "Net::DNS unsupported record type: #{e.message}"
-            end
-          end
+        #------------------------------------------------------------
+        # Header section
+        #------------------------------------------------------------
+        offset = Net::DNS::HFIXEDSZ
+        @header = Net::DNS::Header.parse(data[0..offset - 1])
+
+        @logger.debug ";; HEADER SECTION"
+        @logger.debug @header.inspect
+
+        #------------------------------------------------------------
+        # Question section
+        #------------------------------------------------------------
+        section = @header.opCode == "UPDATE" ? "ZONE" : "QUESTION"
+        @logger.debug ";; #{section} SECTION (#{@header.qdCount} record#{@header.qdCount == 1 ? '' : 's'})"
+
+        @question = []
+        @header.qdCount.times do
+          qobj, offset = parse_question(data, offset)
+          @question << qobj
+          @logger.debug ";; #{qobj.inspect}"
+        end
 
-          #------------------------------------------------------------
-          # Authority/update section
-          #------------------------------------------------------------
-          section = @header.opCode == "UPDATE" ? "UPDATE" : "AUTHORITY"
-          @logger.debug ";; #{section} SECTION (#{@header.nsCount} record#{@header.nsCount == 1 ? '': 's'})"
-
-          @authority = []
-          @header.nsCount.times do
-            begin
-              rrobj,offset = Net::DNS::RR.parse_packet(data,offset)
-              @authority << rrobj
-              @logger.debug rrobj.inspect
-            rescue NameError => e
-              warn "Net::DNS unsupported record type: #{e.message}"
-            end
-          end
+        #------------------------------------------------------------
+        # Answer/prerequisite section
+        #------------------------------------------------------------
+        section = @header.opCode == "UPDATE" ? "PREREQUISITE" : "ANSWER"
+        @logger.debug ";; #{section} SECTION (#{@header.qdCount} record#{@header.qdCount == 1 ? '' : 's'})"
 
-          #------------------------------------------------------------
-          # Additional section
-          #------------------------------------------------------------
-          @logger.debug ";; ADDITIONAL SECTION (#{@header.arCount} record#{@header.arCount == 1 ? '': 's'})"
-
-          @additional = []
-          @header.arCount.times do
-            begin
-              rrobj,offset = Net::DNS::RR.parse_packet(data,offset)
-              @additional << rrobj
-              @logger.debug rrobj.inspect
-            rescue NameError => e
-              warn "Net::DNS supported record type: #{e.message}"
-            end
-          end
+        @answer = []
+        @header.anCount.times do
+          rrobj, offset = Net::DNS::RR.parse_packet(data, offset)
+          @answer << rrobj
+          @logger.debug rrobj.inspect
+        rescue NameError => e
+          warn "Net::DNS unsupported record type: #{e.message}"
+        end
 
+        #------------------------------------------------------------
+        # Authority/update section
+        #------------------------------------------------------------
+        section = @header.opCode == "UPDATE" ? "UPDATE" : "AUTHORITY"
+        @logger.debug ";; #{section} SECTION (#{@header.nsCount} record#{@header.nsCount == 1 ? '' : 's'})"
+
+        @authority = []
+        @header.nsCount.times do
+          rrobj, offset = Net::DNS::RR.parse_packet(data, offset)
+          @authority << rrobj
+          @logger.debug rrobj.inspect
+        rescue NameError => e
+          warn "Net::DNS unsupported record type: #{e.message}"
         end
 
+        #------------------------------------------------------------
+        # Additional section
+        #------------------------------------------------------------
+        @logger.debug ";; ADDITIONAL SECTION (#{@header.arCount} record#{@header.arCount == 1 ? '' : 's'})"
 
-        # Parse question section
-        def parse_question(data,offset)
-          size = (dn_expand(data, offset)[1] - offset) + (2 * Net::DNS::INT16SZ)
-          return [Net::DNS::Question.parse(data[offset, size]), offset + size]
-        rescue StandardError => e
-          raise PacketError, "Caught exception, maybe packet malformed => #{e.message}"
+        @additional = []
+        @header.arCount.times do
+          rrobj, offset = Net::DNS::RR.parse_packet(data, offset)
+          @additional << rrobj
+          @logger.debug rrobj.inspect
+        rescue NameError => e
+          warn "Net::DNS unsupported record type: #{e.message}"
         end
+      end
 
+      # Parse question section
+      def parse_question(data, offset)
+        size = (dn_expand(data, offset)[1] - offset) + (2 * Net::DNS::INT16SZ)
+        [Net::DNS::Question.parse(data[offset, size]), offset + size]
+      rescue StandardError => e
+        raise PacketError, "Caught exception, maybe packet malformed => #{e.message}"
+      end
     end
-
   end
 end