net-dns2 0.8.1

data/lib/net/dns/rr/mr.rb

@@ -0,0 +1,79 @@
+module Net # :nodoc:
+  module DNS
+
+    class RR
+
+      #
+      # = Mail Rename Record (MR)
+      #
+      # Class for DNS MR resource records.
+      #
+      class MR < RR
+
+        # Gets the newname value.
+        #
+        # Returns a String.
+        def newname
+          @newname
+        end
+
+        # Gets the standardized value for this record,
+        # represented by the value of <tt>newname</tt>.
+        #
+        # Returns a String.
+        def value
+          newname.to_s
+        end
+
+
+        private
+
+          def subclass_new_from_hash(options)
+            if options.has_key?(:newname)
+              @newname = check_name(options[:newname])
+            else
+              raise ArgumentError, ":newname field is mandatory"
+            end
+          end
+
+          def subclass_new_from_string(str)
+            @newname = check_name(str)
+          end
+
+          def subclass_new_from_binary(data, offset)
+            @newname = dn_expand(data,offset)
+            offset
+          end
+
+
+          def set_type
+            @type = Net::DNS::RR::Types.new("MR")
+          end
+
+          def get_inspect
+            value
+          end
+
+
+          def check_name(input)
+            name = input.to_s
+            unless name =~ /(\w\.?)+\s*$/
+              raise ArgumentError, "Invalid Domain Name `#{name}'"
+            end
+            name
+          end
+
+          def build_pack
+            @newname_pack = pack_name(@newname)
+            @rdlength = @newname_pack.size
+          end
+
+          def get_data
+            @newname_pack
+          end
+
+      end
+
+    end
+  end
+end

data/lib/net/dns/rr/mx.rb

@@ -0,0 +1,92 @@
+module Net # :nodoc:
+  module DNS
+    class RR
+
+      #
+      # = Mail Exchange Record (MX)
+      #
+      # Class for DNS MX resource records.
+      #
+      # A MX record specifies the name and relative preference of mail servers
+      # (mail exchangers in the DNS jargon) for the zone.
+      # The MX RR is used by SMTP (Mail) Agents to route mail for the domain.
+      #
+      class MX < RR
+
+        # Gets the preference value.
+        #
+        # Returns an Integer.
+        def preference
+          @preference
+        end
+
+        # Gets the exchange value.
+        #
+        # Returns a String.
+        def exchange
+          @exchange
+        end
+
+        # Gets the standardized value for this record,
+        # represented by the value of <tt>preference</tt> and <tt>exchange</tt>.
+        #
+        # Returns a String.
+        def value
+          "#{preference} #{exchange}"
+        end
+
+
+        private
+
+          def subclass_new_from_hash(options)
+            if options.has_key?(:preference) && options.has_key?(:exchange)
+              @preference = options[:preference].to_i
+              @exchange = options[:exchange]
+            else
+              raise ArgumentError, ":preference and :exchange fields are mandatory"
+            end
+          end
+
+          def subclass_new_from_string(str)
+            @preference, @exchange = check_mx(str)
+          end
+
+          def subclass_new_from_binary(data, offset)
+            @preference = data.unpack("@#{offset} n")[0]
+            offset += 2
+            @exchange, offset = dn_expand(data, offset)
+            offset
+          end
+
+
+          def set_type
+            @type = Net::DNS::RR::Types.new("MX")
+          end
+
+          def get_inspect
+            value
+          end
+
+
+          def check_mx(input)
+            str = input.to_s
+            unless str.strip =~ /^(\d+)\s+(\S+)$/
+              raise ArgumentError, "Invalid MX section `#{str}'"
+            end
+            [$1.to_i, $2]
+          end
+
+          def build_pack
+            @mx_pack = [@preference].pack("n") + pack_name(@exchange)
+            @rdlength = @mx_pack.size
+          end
+
+          def get_data
+            @mx_pack
+          end
+
+      end
+
+    end
+  end
+end

data/lib/net/dns/rr/ns.rb

@@ -0,0 +1,78 @@
+module Net # :nodoc:
+  module DNS
+    class RR
+
+      #
+      # = Name Server Record (NS)
+      #
+      # Class for DNS NS resource records.
+      #
+      class NS < RR
+
+        # Gets the name server value.
+        #
+        # Returns a String.
+        def nsdname
+          @nsdname
+        end
+
+        # Gets the standardized value for this record,
+        # represented by the value of <tt>nsdname</tt>.
+        #
+        # Returns a String.
+        def value
+          nsdname.to_s
+        end
+
+
+        private
+
+          def subclass_new_from_hash(options)
+            if options.has_key?(:nsdname)
+              @nsdname = check_name(options[:nsdname])
+            else
+              raise ArgumentError, ":nsdname field is mandatory"
+            end
+          end
+
+          def subclass_new_from_string(str)
+            @nsdname = check_name(str)
+          end
+
+          def subclass_new_from_binary(data, offset)
+            @nsdname, offset = dn_expand(data, offset)
+            offset
+          end
+
+
+          def set_type
+            @type = Net::DNS::RR::Types.new("NS")
+          end
+
+          def get_inspect
+            value
+          end
+
+
+          def check_name(input)
+            name = input.to_s
+            unless name =~ /(\w\.?)+\s*$/ and name =~ /[a-zA-Z]/
+              raise ArgumentError, "Invalid Name Server `#{name}'"
+            end
+            name
+          end
+
+          def build_pack
+            @nsdname_pack = pack_name(@nsdname)
+            @rdlength = @nsdname_pack.size
+          end
+
+          def get_data
+            @nsdname_pack
+          end
+
+      end
+
+    end
+  end
+end

data/lib/net/dns/rr/null.rb

@@ -0,0 +1,53 @@
+module Net # :nodoc:
+  module DNS
+    class RR
+
+      #------------------------------------------------------------
+      # RR type NULL
+      #------------------------------------------------------------
+      class NULL < RR
+        attr_reader :null
+
+        private
+
+        def build_pack
+          @null_pack = @null
+          @rdlength = @null_pack.size
+        end
+
+        def get_data
+          @null_pack
+        end
+
+        def get_inspect
+          "#@null"
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key? :null
+            @null = args[:null]
+          else
+            raise ArgumentError, ":null field is mandatory but missing"
+          end
+        end
+
+        def subclass_new_from_string(str)
+          @null = str.strip
+        end
+
+        def subclass_new_from_binary(data,offset)
+          @null = data[offset..offset+@rdlength]
+          return offset + @rdlength
+        end
+
+        private
+
+          def set_type
+            @type = Net::DNS::RR::Types.new("NULL")
+          end
+
+      end
+
+    end
+  end
+end

data/lib/net/dns/rr/ptr.rb

@@ -0,0 +1,83 @@
+module Net
+  module DNS
+    class RR
+
+      #
+      # = Pointer Record (PTR)
+      #
+      # Class for DNS Pointer (PTR) resource records.
+      #
+      # Pointer records are the opposite of A and AAAA RRs 
+      # and are used in Reverse Map zone files to map
+      # an IP address (IPv4 or IPv6) to a host name.
+      #
+      class PTR < RR
+
+        # Gets the PTR value.
+        #
+        # Returns a String.
+        def ptrdname
+          @ptrdname.to_s
+        end
+
+        alias_method :ptr, :ptrdname
+
+        # Gets the standardized value for this record,
+        # represented by the value of <tt>ptrdname</tt>.
+        #
+        # Returns a String.
+        def value
+          ptrdname.to_s
+        end
+
+
+        private
+
+        def build_pack
+          @ptrdname_pack = pack_name(@ptrdname)
+          @rdlength = @ptrdname_pack.size
+        end
+
+        def get_data
+          @ptrdname_pack
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key?(:ptrdname) or args.has_key?(:ptr)
+            @ptrdname = args[:ptrdname]
+          else
+            raise ArgumentError, ":ptrdname or :ptr field is mandatory"
+          end
+        end
+
+        def subclass_new_from_string(str)
+          @ptrdname = check_name(str)
+        end
+
+        def subclass_new_from_binary(data, offset)
+          @ptrdname, offset = dn_expand(data, offset)
+          offset
+        end
+
+        private
+
+          def set_type
+            @type = Net::DNS::RR::Types.new("PTR")
+          end
+
+          def get_inspect
+            value
+          end
+
+
+          def check_name(input)
+            IPAddr.new(str)
+          rescue
+            raise ArgumentError, "Invalid PTR Section `#{input}'"
+          end
+
+      end
+
+    end
+  end
+end

data/lib/net/dns/rr/soa.rb

@@ -0,0 +1,78 @@
+module Net # :nodoc:
+  module DNS
+    class RR
+
+      #------------------------------------------------------------
+      # RR type SOA
+      #------------------------------------------------------------
+      class SOA < RR
+        attr_reader :mname, :rname, :serial, :refresh, :retry, :expire, :minimum
+        
+        private
+        
+        def build_pack
+          @soa_pack = pack_name(@mname)
+          @soa_pack += pack_name(@rname)
+          @soa_pack += [@serial,@refresh,@retry,@expire,@minimum].pack("N5")
+        end
+
+        def get_data
+          @soa_pack
+        end
+
+        def get_inspect
+          "#@mname #@rname #@serial #@refresh #@retry #@expire #@minimum"
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key? :rdata
+            subclass_new_from_string(args[:rdata])
+          else
+            [:mname,:rname,:serial,:refresh,:retry,:expire,:minimum].each do |key|
+              raise ArgumentError, "Missing field :#{key}" unless args.has_key? key
+            end
+            @mname = args[:mname] if valid? args[:mname] 
+            @rname = args[:rname] if valid? args[:rname]
+            @serial = args[:serial] if number? args[:serial] 
+            @refresh = args[:refresh] if number? args[:refresh] 
+            @retry = args[:retry] if number? args[:retry] 
+            @expire = args[:expire] if number? args[:expire] 
+            @minimum = args[:minimum] if number? args[:minimum] 
+          end
+        end
+        
+        def number?(num)
+          if num.kind_of? Integer and num > 0
+            true
+          else
+            raise ArgumentError, "Wrong format field: #{num} not a number or less than zero"
+          end
+        end
+
+        def subclass_new_from_string(str)
+          mname,rname,serial,refresh,ret,expire,minimum = str.strip.split(" ")
+          @mname = mname if valid? mname
+          @rname = rname if valid? rname
+          @serial,@refresh,@retry,@expire,@minimum = [serial,refresh,ret,expire,minimum].collect do |i| 
+            i.to_i if valid? i.to_i
+          end
+        end
+
+        def subclass_new_from_binary(data,offset)
+          @mname,offset = dn_expand(data,offset)
+          @rname,offset = dn_expand(data,offset)
+          @serial,@refresh,@retry,@expire,@minimum = data.unpack("@#{offset} N5")
+          return offset + 5*Net::DNS::INT32SZ
+        end
+        
+        private
+        
+          def set_type
+            @type = Net::DNS::RR::Types.new("SOA")
+          end
+        
+      end
+      
+    end
+  end
+end