dnsruby 1.55 → 1.56.0

data/lib/dnsruby/resource/RRSet.rb

@@ -0,0 +1,190 @@
+module Dnsruby
+
+# RFC2181, section 5
+# "It is however possible for most record types to exist
+# with the same label, class and type, but with different data.  Such a
+# group of records is hereby defined to be a Resource Record Set
+# (RRSet)."
+# This class also stores the RRSIG records which cover the RRSet
+class RRSet
+  include Comparable
+  #  The number of RRSIGs stored in this RRSet
+  attr_reader :num_sigs
+  def initialize(rrs = [])
+    if (!rrs.instance_of?Array)
+      rrs = [rrs]
+    end
+    @rrs = []
+    @num_sigs = 0
+    rrs.each {|rr| add(rr)}
+  end
+  def self.new_from_string(string)
+    rr_strings = string.split("\n")
+    rrs = rr_strings.map { |s| Dnsruby::RR.new_from_string(s) }
+
+    Dnsruby::RRSet.new(rrs)
+  end    # The RRSIGs stored with this RRSet
+  def sigs
+    return @rrs[@rrs.length-@num_sigs, @num_sigs]
+  end
+  #  The RRs (not RRSIGs) stored in this RRSet
+  def rrs
+    return @rrs[0, @rrs.length-@num_sigs]
+  end
+  def privateAdd(r) #:nodoc:
+    if @rrs.include?r
+      return true
+    end
+    new_pos = @rrs.length - @num_sigs
+    if ((@num_sigs == @rrs.length)  && @num_sigs > 0) # if we added RRSIG first
+      if (((r.type != @rrs.last.type_covered) && (r.type != Types.RRSIG))||
+          ((r.type == Types.RRSIG) && (r.type_covered != @rrs.last.type_covered)))
+        return false
+      end
+    end
+    if (r.type == Types::RRSIG)
+      new_pos = @rrs.length
+      @num_sigs += 1
+    end
+    @rrs.insert(new_pos, r)
+    return true
+  end
+
+  # Add the RR to this RRSet
+  # Takes a copy of the RR by default. To suppress this, pass false
+  # as the second parameter.
+  def add(rin, do_clone = true)
+    if (rin.instance_of?RRSet)
+      ret = false
+      [rin.rrs, rin.sigs].each {|rr| ret = add(rr)}
+      return ret
+    end
+    #       r = RR.create(r.to_s) # clone the record
+    r = nil
+    if do_clone
+      r = rin.clone
+    else
+      r = rin
+    end
+    if (@rrs.size() == 0) #  && !(r.type == Types.RRSIG))
+      return privateAdd(r)
+    end
+    #  Check the type, klass and ttl are correct
+    first = @rrs[0]
+    if (!r.sameRRset(first))
+      return false
+      #         raise ArgumentError.new("record does not match rrset")
+    end
+
+    if (!(r.type == Types::RRSIG) && (!(first.type == Types::RRSIG)))
+      if (r.ttl != first.ttl) # RFC2181, section 5.2
+        if (r.ttl > first.ttl)
+          r.ttl=(first.ttl)
+        else
+          @rrs.each do |rr|
+            rr.ttl = r.ttl
+          end
+        end
+      end
+    end
+
+    return privateAdd(r)
+    #       return true
+  end
+
+  def <=>(other)
+    #       return 1 if ((!other) || !(other.name) || !(other.type))
+    #       return -1 if (!@name)
+    if (name.canonical == other.name.canonical)
+      return type.code <=> other.type.code
+    else
+      return name <=> other.name
+    end
+  end
+
+  def sort_canonical
+    # Make a list, for all the RRs, where each RR contributes
+    # the canonical RDATA encoding
+    canonical_rrs = {}
+    self.rrs.each do |rr|
+      data = MessageEncoder.new {|msg|
+        rr.encode_rdata(msg, true)
+      }.to_s
+      canonical_rrs[data] = rr
+    end
+
+    return_rrs = RRSet.new
+    canonical_rrs.keys.sort.each { |rdata|
+      return_rrs.add(canonical_rrs[rdata], false)
+    }
+    return return_rrs
+  end
+
+  def ==(other)
+    return false unless other.instance_of?RRSet
+    return false if (other.sigs.length != self.sigs.length)
+    return false if (other.rrs.length != self.rrs.length)
+    return false if (other.ttl != self.ttl)
+    otherrrs = other.rrs
+    self.rrs.each {|rr|
+      return false if (!otherrrs.include?rr)
+    }
+    othersigs= other.sigs
+    self.sigs.each {|sig|
+      return false if (!othersigs.include?sig)
+    }
+    return true
+  end
+  # Delete the RR from this RRSet
+  def delete(rr)
+    @rrs.delete(rr)
+  end
+  def each
+    @rrs.each do |rr|
+      yield rr
+    end
+  end
+  def [](index)
+    return @rrs[index]
+  end
+  # Return the type of this RRSet
+  def type
+    if (@rrs[0])
+      return @rrs[0].type
+    end
+    return nil
+  end
+  # Return the klass of this RRSet
+  def klass
+    return @rrs[0].klass
+  end
+  # Return the ttl of this RRSet
+  def ttl
+    return @rrs[0].ttl
+  end
+  def ttl=(ttl)
+    [rrs, sigs].each {|rrs|
+      rrs.each {|rr|
+        rr.ttl = ttl
+      }
+    }
+  end
+  def name
+    if (@rrs[0])
+      return @rrs[0].name
+    else
+      return nil
+    end
+  end
+  def to_s
+    ret = ""
+    each {|rec|
+      ret += rec.to_s + "\n"
+    }
+    return ret
+  end
+  def length
+    return @rrs.length
+  end
+end
+end

data/lib/{Dnsruby → dnsruby}/resource/RT.rb

@@ -1,67 +1,67 @@
-#--
-#Copyright 2007 Nominet UK
-#
-#Licensed under the Apache License, Version 2.0 (the "License");
-#you may not use this file except in compliance with the License. 
-#You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0 
-#
-#Unless required by applicable law or agreed to in writing, software 
-#distributed under the License is distributed on an "AS IS" BASIS, 
-#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
-#See the License for the specific language governing permissions and 
-#limitations under the License.
-#++
-module Dnsruby
-  class RR   
-    #Class for DNS Route Through (RT) resource records.
-    #RFC 1183 Section 3.3
-    class RT < RR
-      ClassValue = nil #:nodoc: all
-      TypeValue = Types::RT #:nodoc: all
-      
-      #The preference for this route.
-      attr_accessor :preference
-      
-      #The domain name of the intermediate host.
-      attr_accessor :intermediate
-      
-      def from_hash(hash) #:nodoc: all
-        @preference = hash[:preference]
-        @intermediate = Name.create(hash[:intermediate])
-      end
-      
-      def from_data(data) #:nodoc: all
-        @preference, @intermediate = data
-      end
-      
-      def from_string(input) #:nodoc: all
-        if (input.length > 0)
-          names = input.split(" ")
-          @preference = names[0].to_i
-          @intermediate = Name.create(names[1])
-        end
-      end
-      
-      def rdata_to_string #:nodoc: all
-        if (@preference!=nil)
-          return "#{@preference} #{@intermediate.to_s(true)}"
-        else
-          return ""
-        end
-      end
-      
-      def encode_rdata(msg, canonical = false) #:nodoc: all
-        msg.put_pack('n', @preference)
-        msg.put_name(@intermediate, canonical)
-      end
-      
-      def self.decode_rdata(msg) #:nodoc: all
-        preference, = msg.get_unpack('n')
-        intermediate = msg.get_name
-        return self.new([preference, intermediate])
-      end
-    end
-  end
-end
+# --
+# Copyright 2007 Nominet UK
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ++
+module Dnsruby
+  class RR
+    # Class for DNS Route Through (RT) resource records.
+    # RFC 1183 Section 3.3
+    class RT < RR
+      ClassValue = nil #:nodoc: all
+      TypeValue = Types::RT #:nodoc: all
+
+      # The preference for this route.
+      attr_accessor :preference
+
+      # The domain name of the intermediate host.
+      attr_accessor :intermediate
+
+      def from_hash(hash) #:nodoc: all
+        @preference = hash[:preference]
+        @intermediate = Name.create(hash[:intermediate])
+      end
+
+      def from_data(data) #:nodoc: all
+        @preference, @intermediate = data
+      end
+
+      def from_string(input) #:nodoc: all
+        if (input.length > 0)
+          names = input.split(" ")
+          @preference = names[0].to_i
+          @intermediate = Name.create(names[1])
+        end
+      end
+
+      def rdata_to_string #:nodoc: all
+        if (@preference!=nil)
+          return "#{@preference} #{@intermediate.to_s(true)}"
+        else
+          return ""
+        end
+      end
+
+      def encode_rdata(msg, canonical = false) #:nodoc: all
+        msg.put_pack('n', @preference)
+        msg.put_name(@intermediate, canonical)
+      end
+
+      def self.decode_rdata(msg) #:nodoc: all
+        preference, = msg.get_unpack('n')
+        intermediate = msg.get_name
+        return self.new([preference, intermediate])
+      end
+    end
+  end
+end

data/lib/{Dnsruby → dnsruby}/resource/SOA.rb

@@ -1,95 +1,95 @@
-#--
-#Copyright 2007 Nominet UK
-#
-#Licensed under the Apache License, Version 2.0 (the "License");
-#you may not use this file except in compliance with the License. 
-#You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0 
-#
-#Unless required by applicable law or agreed to in writing, software 
-#distributed under the License is distributed on an "AS IS" BASIS, 
-#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
-#See the License for the specific language governing permissions and 
-#limitations under the License.
-#++
-module Dnsruby
-  class RR
-    class SOA < RR
-      ClassValue = nil #:nodoc: all
-      TypeValue = Types::SOA #:nodoc: all
-      
-      #The domain name of the original or primary nameserver for
-      #this zone.
-      attr_accessor :mname
-      #A domain name that specifies the mailbox for the person
-      #responsible for this zone.
-      attr_accessor :rname
-      #The zone's serial number.
-      attr_accessor :serial
-      #The zone's refresh interval.
-      #How often, in seconds, a secondary nameserver is to check for
-      #updates from the primary nameserver.
-      attr_accessor :refresh
-      #The zone's retry interval.
-      #How often, in seconds, a secondary nameserver is to retry, after a 
-      #failure to check for a refresh
-      attr_accessor :retry
-      #The zone's expire interval.
-      #How often, in seconds, a secondary nameserver is to use the data
-      #before refreshing from the primary nameserver
-      attr_accessor :expire
-      #The minimum (default) TTL for records in this zone.
-      attr_accessor :minimum
-      
-      def from_data(data) #:nodoc: all
-        @mname, @rname, @serial, @refresh, @retry, @expire, @minimum = data
-      end
-      
-      def from_hash(hash)
-        @mname = Name.create(hash[:mname])
-        @rname = Name.create(hash[:rname])
-        @serial = hash[:serial].to_i
-        @refresh = hash[:refresh].to_i
-        @retry = hash[:retry].to_i
-        @expire = hash[:expire].to_i
-        @minimum = hash[:minimum].to_i
-      end
-      
-      def from_string(input)
-        if (input.length > 0)
-          names = input.split(" ")
-          @mname = Name.create(names[0])
-          @rname = Name.create(names[1])
-          @serial = names[2].to_i
-          @refresh = names[3].to_i
-          @retry = names[4].to_i
-          @expire = names[5].to_i
-          @minimum = names[6].to_i
-        end
-      end
-      
-      def rdata_to_string #:nodoc: all
-        if (@mname!=nil)
-          return "#{@mname.to_s(true)} #{@rname.to_s(true)} #{@serial} #{@refresh} #{@retry} #{@expire} #{@minimum}"
-        else
-          return ""
-        end
-      end
-      
-      def encode_rdata(msg, canonical=false) #:nodoc: all
-        msg.put_name(@mname, canonical)
-        msg.put_name(@rname, canonical)
-        msg.put_pack('NNNNN', @serial, @refresh, @retry, @expire, @minimum)
-      end
-      
-      def self.decode_rdata(msg) #:nodoc: all
-        mname = msg.get_name
-        rname = msg.get_name
-        serial, refresh, retry_, expire, minimum = msg.get_unpack('NNNNN')
-        return self.new(
-                        [mname, rname, serial, refresh, retry_, expire, minimum])
-      end
-    end 
-  end
+# --
+# Copyright 2007 Nominet UK
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ++
+module Dnsruby
+  class RR
+    class SOA < RR
+      ClassValue = nil #:nodoc: all
+      TypeValue = Types::SOA #:nodoc: all
+
+      # The domain name of the original or primary nameserver for
+      # this zone.
+      attr_accessor :mname
+      # A domain name that specifies the mailbox for the person
+      # responsible for this zone.
+      attr_accessor :rname
+      # The zone's serial number.
+      attr_accessor :serial
+      # The zone's refresh interval.
+      # How often, in seconds, a secondary nameserver is to check for
+      # updates from the primary nameserver.
+      attr_accessor :refresh
+      # The zone's retry interval.
+      # How often, in seconds, a secondary nameserver is to retry, after a
+      # failure to check for a refresh
+      attr_accessor :retry
+      # The zone's expire interval.
+      # How often, in seconds, a secondary nameserver is to use the data
+      # before refreshing from the primary nameserver
+      attr_accessor :expire
+      # The minimum (default) TTL for records in this zone.
+      attr_accessor :minimum
+
+      def from_data(data) #:nodoc: all
+        @mname, @rname, @serial, @refresh, @retry, @expire, @minimum = data
+      end
+
+      def from_hash(hash)
+        @mname = Name.create(hash[:mname])
+        @rname = Name.create(hash[:rname])
+        @serial = hash[:serial].to_i
+        @refresh = hash[:refresh].to_i
+        @retry = hash[:retry].to_i
+        @expire = hash[:expire].to_i
+        @minimum = hash[:minimum].to_i
+      end
+
+      def from_string(input)
+        if (input.length > 0)
+          names = input.split(" ")
+          @mname = Name.create(names[0])
+          @rname = Name.create(names[1])
+          @serial = names[2].to_i
+          @refresh = names[3].to_i
+          @retry = names[4].to_i
+          @expire = names[5].to_i
+          @minimum = names[6].to_i
+        end
+      end
+
+      def rdata_to_string #:nodoc: all
+        if (@mname!=nil)
+          return "#{@mname.to_s(true)} #{@rname.to_s(true)} #{@serial} #{@refresh} #{@retry} #{@expire} #{@minimum}"
+        else
+          return ""
+        end
+      end
+
+      def encode_rdata(msg, canonical=false) #:nodoc: all
+        msg.put_name(@mname, canonical)
+        msg.put_name(@rname, canonical)
+        msg.put_pack('NNNNN', @serial, @refresh, @retry, @expire, @minimum)
+      end
+
+      def self.decode_rdata(msg) #:nodoc: all
+        mname = msg.get_name
+        rname = msg.get_name
+        serial, refresh, retry_, expire, minimum = msg.get_unpack('NNNNN')
+        return self.new(
+                        [mname, rname, serial, refresh, retry_, expire, minimum])
+      end
+    end
+  end
 end