net-dns 0.1

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

@@ -0,0 +1,68 @@
+##
+#
+# Net::DNS::RR::MR
+#
+#       $Id: MR.rb,v 1.4 2006/07/28 07:33:36 bluemonk Exp $
+#
+##
+
+module Net
+  module DNS
+
+    class RR
+      
+      #------------------------------------------------------------
+      # RR type MR
+      #------------------------------------------------------------
+      class MR < RR
+        attr_reader :newname
+
+        private
+        
+        def check_name(name)
+          unless name =~ /(\w\.?)+\s*$/
+            raise RRArgumentError, "Name not valid: #{name.inspect}"
+          end
+          name
+        end
+
+        def build_pack
+          @newname_pack = pack_name(@newname)
+          @rdlength = @newname_pack.size
+        end
+
+        def set_type
+          @type = Net::DNS::RR::Types.new("MR")
+        end
+
+        def get_data
+          @newname_pack
+        end
+
+        def get_inspect
+          "#@newname"
+        end
+          
+        def subclass_new_from_hash(args)
+          if args.has_key? :newname
+            @newname = check_name args[:newname]
+          else
+            raise RRArgumentError, ":newname field is mandatory but missing"
+          end
+        end
+
+        def subclass_new_from_string(str)
+          @newname = check_name(str)
+        end
+
+        def subclass_new_from_array(data,offset)
+          @newname = dn_expand(data,offset)
+          return offset
+        end
+        
+      end # class MR
+
+    end # class RR
+  end # module DNS
+end # module Net
+

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

@@ -0,0 +1,74 @@
+##
+#
+# Net::DNS::RR::MX
+#
+#       $Id: MX.rb,v 1.8 2006/07/28 07:33:36 bluemonk Exp $     
+#
+##
+
+
+module Net
+  module DNS
+    class RR
+      
+      #------------------------------------------------------------
+      # RR type MX
+      #------------------------------------------------------------
+      class MX < RR
+        attr_reader :preference, :exchange
+
+        private
+        
+        def check_mx(str)
+          if str.strip =~ /^(\d+)\s+(\S+)$/
+            return $1.to_i,$2
+          else
+            raise RRArgumentError, "MX section not valid"
+          end
+        end
+        
+        def build_pack
+          @mx_pack = [@preference].pack("n") + pack_name(@exchange)
+          @rdlength = @mx_pack.size
+        end
+
+        def set_type
+          @type = Net::DNS::RR::Types.new("MX")
+        end
+
+        def get_data
+          @mx_pack
+        end
+
+        def get_inspect
+          "#@preference #@exchange"
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key? :preference and args.has_key? :exchange
+            @preference = args[0][:preference].to_i
+            @exchange =  args[0][:exchange]
+          else
+            raise RRArgumentError, ":preference and :exchange fields are mandatory but missing"
+          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)
+          return offset
+        end
+
+      end # class MX
+      
+    end # class RR
+  end # module DNS
+end # module Net
+
+
+

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

@@ -0,0 +1,70 @@
+##
+#
+# Net::DNS::RR::NS
+#
+#       $Id: NS.rb,v 1.8 2006/07/28 07:33:36 bluemonk Exp $     
+#
+##
+
+module Net
+  module DNS
+    
+    class RR
+      
+      #------------------------------------------------------------
+      # RR type NS
+      #------------------------------------------------------------
+      class NS < RR
+        attr_reader :nsdname
+
+        private
+        
+        def check_name(name)
+          unless name =~ /(\w\.?)+\s*$/ and name =~ /[a-zA-Z]/
+            raise RRArgumentError, "NS Domain Name not valid: #{name}"
+          end
+          name
+        end
+
+        def build_pack
+          @nsdname_pack = pack_name(@nsdname)
+          @rdlength = @nsdname_pack.size
+        end
+
+        def set_type
+          @type = Net::DNS::RR::Types.new("NS")
+        end
+
+        def get_data
+          @nsdname_pack
+        end
+
+        def get_inspect
+          "#@nsdname"
+        end
+          
+        def subclass_new_from_hash(args)
+          if args.has_key? :nsdname
+            @nsdname = check_name args[:nsdname]
+          else
+            raise RRArgumentError, ":nsdname field is mandatory but missing"
+          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)
+          return offset
+        end
+        
+      end # class NS
+      
+    end # class RR
+  end # module DNS
+end # module Net
+
+
+

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

@@ -0,0 +1,61 @@
+##
+#
+# Net::DNS::RR::NULL
+#
+#       $Id: NULL.rb,v 1.5 2006/07/28 07:33:36 bluemonk Exp $
+#
+##
+
+
+module Net
+  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 set_type
+          @type = Net::DNS::RR::RRTypes.new("NULL")
+        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 RRArgumentError, ":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
+        
+      end # class NULL
+      
+    end # class RR
+  end # module DNS
+end # module Net
+

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

@@ -0,0 +1,71 @@
+##
+#
+# Net::DNS::RR::PTR
+#
+#       $Id: PTR.rb,v 1.5 2006/07/28 07:33:36 bluemonk Exp $    
+#
+##
+
+module Net
+  module DNS
+    class RR
+      
+      #------------------------------------------------------------
+      # RR type PTR
+      #------------------------------------------------------------
+      class PTR < RR
+
+        # Getter for PTR resource
+        def ptr
+          @ptrdname.to_s
+        end
+        alias_method :ptrdname, :ptr
+        
+        private
+        
+        def check_ptr(str)
+          IPAddr.new str
+        rescue
+          raise RRArgumentError, "PTR section not valid"
+        end
+        
+        def build_pack
+          @ptrdname_pack = pack_name(@ptrdname)
+          @rdlength = @ptrdname_pack.size
+        end
+
+        def set_type
+          @type = Net::DNS::RR::RRTypes.new("PTR")
+        end
+
+        def get_data
+          @ptrdname_pack
+        end
+
+        def get_inspect
+          "#@ptrdname"
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key? :ptrdname or args.has_key? :ptr
+            @ptrdname = args[0][:ptrdname]
+          else
+            raise RRArgumentError, ":ptrdname or :ptr field is mandatory but missing"
+          end
+        end
+
+        def subclass_new_from_string(str)
+          @ptrdname = check_ptr(str)
+        end
+
+        def subclass_new_from_binary(data,offset)
+          @ptrdname,offset = dn_expand(data,offset)
+          return offset
+        end
+
+      end # class PTR
+      
+    end # class RR
+  end # module DNS
+end # module Net
+

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

@@ -0,0 +1,85 @@
+##
+#
+# Net::DNS::RR::SOA
+#
+#       $Id: SOA.rb,v 1.4 2006/07/28 07:33:36 bluemonk Exp $    
+#
+##
+
+module Net
+  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 set_type
+          @type = Net::DNS::RR::Types.new("SOA")
+        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 RRArgumentError, "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 RRArgumentError, "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
+
+      end # class SOA
+      
+    end # class RR
+  end # module DNS
+end # module Net
+

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

@@ -0,0 +1,72 @@
+##
+#
+# Net::DNS::RR::TXT
+#
+#       $Id: TXT.rb,v 1.4 2006/07/28 07:33:36 bluemonk Exp $
+#
+##
+
+
+module Net
+  module DNS
+    class RR
+      
+      #------------------------------------------------------------
+      # RR type TXT
+      #------------------------------------------------------------
+      class TXT < RR
+        attr_reader :txt
+
+        private
+        
+        def build_pack
+          str = ""
+          @txt.split(" ").each do |txt|
+            str += [txt.length,txt].pack("C a*")
+          end
+          @txt_pack = str
+          @rdlength = @txt_pack.size
+        end
+        
+        def set_type
+          @type = Net::DNS::RR::Types.new("TXT")
+        end
+
+        def get_data
+          @txt_pack
+        end
+
+        def subclass_new_from_hash(args)
+          if args.has_key? :txt
+            @txt = args[:txt].strip
+          else
+            raise RRArgumentError, ":txt field is mandatory but missing"
+          end
+        end
+        
+        def subclass_new_from_string(str)
+          @txt = str.strip
+        end
+        
+        def subclass_new_from_binary(data,offset)
+          off_end = offset + @rdlength
+          @txt = ""
+          while offset < off_end
+            len = data.unpack("@#{offset} C")[0]
+            offset += 1
+            str = data[offset..offset+len-1]
+            offset += len
+            @txt << str << " "
+          end
+          return offset
+        end 
+
+      end # class TXT
+      
+    end # class RR
+  end # module DNS
+end # module Net
+
+
+
+