gitlab-net-dns 0.9.1

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

@@ -0,0 +1,70 @@
+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.
+        attr_reader :newname
+
+        # 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.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,82 @@
+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.
+        attr_reader :preference
+
+        # Gets the exchange value.
+        #
+        # Returns a String.
+        attr_reader :exchange
+
+        # 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.key?(:preference) && options.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
+
+          [Regexp.last_match(1).to_i, Regexp.last_match(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,70 @@
+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.
+        attr_reader :nsdname
+
+        # 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.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*$/ && 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,50 @@
+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.to_s
+        end
+
+        def subclass_new_from_hash(args)
+          if args.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]
+          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,77 @@
+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 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.key?(:ptrdname) || args.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 StandardError
+          raise ArgumentError, "Invalid PTR Section `#{input}'"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,75 @@
+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.key? :rdata
+            subclass_new_from_string(args[:rdata])
+          else
+            %i[mname rname serial refresh retry expire minimum].each do |key|
+              raise ArgumentError, "Missing field :#{key}" unless args.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.is_a?(Integer) && (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")
+          offset + 5 * Net::DNS::INT32SZ
+        end
+
+        private
+
+        def set_type
+          @type = Net::DNS::RR::Types.new("SOA")
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,41 @@
+module Net # :nodoc:
+  module DNS
+    class RR
+      #------------------------------------------------------------
+      # RR type SRV
+      #------------------------------------------------------------
+      class SRV < RR
+        attr_reader :priority, :weight, :port, :host
+
+        private
+
+        def build_pack
+          str = ""
+        end
+
+        def subclass_new_from_binary(data, offset)
+          off_end = offset + @rdlength
+          @priority, @weight, @port = data.unpack("@#{offset} n n n")
+          offset += 6
+
+          @host = []
+          while offset < off_end
+            len = data.unpack("@#{offset} C")[0]
+            offset += 1
+            str = data[offset..offset + len - 1]
+            offset += len
+            @host << str
+          end
+          @host = @host.join(".")
+          offset
+        end
+
+        private
+
+        def set_type
+          @type = Net::DNS::RR::Types.new("SRV")
+        end
+      end
+    end
+  end
+end