pNet-DNS 0.0.1

data/test/tc_tcp.rb

@@ -0,0 +1,34 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require 'test/unit'
+require 'Net/DNS'
+class TestTcp < Test::Unit::TestCase
+  def test_TCP
+    res = Net::DNS::Resolver.new(:config_file=>"/etc/resolv.conf")
+    res.debug=true
+    res.usevc = true
+    ret=res.query("example.com")
+    assert(ret.is_a?(Net::DNS::Packet))
+  end
+  def test_TCP_port
+    res = Net::DNS::Resolver.new(:config_file=>"/etc/resolv.conf")
+    res.debug=true
+    res.usevc = true
+    res.srcport=rand(60000) + 1025
+    ret=res.query("example.com")
+    assert(ret.is_a?(Net::DNS::Packet))
+  end
+end

data/test/tc_tkey.rb

@@ -0,0 +1,115 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require 'test/unit'
+require 'Net/DNS'
+require "digest/md5"
+class TestTKey < Test::Unit::TestCase
+  def is_empty(string)
+    return (string == "; no data" || string == "; rdlength = 0")
+  end
+  
+  def test_tkey
+    
+    
+    #------------------------------------------------------------------------------
+    # Canned data.
+    #------------------------------------------------------------------------------
+    
+    zone	= "example.com"
+    name	= "123456789-test"
+    klass	= "IN"
+    type	= "TKEY"
+    algorithm   = "fake.algorithm.example.com"
+    key         = "fake key"
+    inception   = 100000 # use a strange fixed inception time to give a fixed
+    # checksum
+    expiration  = inception + 24*60*60
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Packet creation.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS::RR.create(
+	:name       => name,
+	:type       => "TKEY",
+	:ttl        => 0,
+	:rrclass      => "ANY",
+	:algorithm  => algorithm,
+	:inception  => inception,
+	:expiration => expiration,
+	:mode       => 3, # GSSAPI
+	:key        => "fake key",
+	:other_data => ""
+    )
+    
+    packet = Net::DNS::Packet.new_from_values(name, "TKEY", "IN")
+    packet.push("answer", rr)
+    
+    z = (packet.zone)[0]
+    
+    assert(packet,                                'new() returned packet')  #2
+    assert_equal('QUERY',       packet.header.opcode, 'header opcode correct')  #3 
+    assert_equal(name,                      z.zname,  'zname correct')          #4
+    assert_equal("IN",                       z.zclass, 'zclass correct')         #5
+    assert_equal('TKEY',                     z.ztype,  'ztype correct')          #6       
+    
+    
+    #------------------------------------------------------------------------------
+    # create a signed TKEY query packet using an external signing function
+    # and compare it to a known good result. This effectively tests the
+    # sign_func and sig_data methods of TSIG as well.
+    #------------------------------------------------------------------------------
+    
+    
+    tsig = Net::DNS::RR.create({
+                               :name        => name,
+                               :type        => "TSIG",
+    :ttl         => 0,
+    :rrclass       => "ANY",
+    :algorithm   => algorithm,
+    :time_signed => inception + 1,
+    :fudge       => 36000,
+    :mac_size    => 0,
+    :mac         => "",
+    :key         => key,
+    :sign_func   => Proc.new { |key,data|     
+   #         OpenSSL::HMAC.hexdigest(OpenSSL::Digest::Digest.new("md5"), key, data) 
+        hmac = Digest::MD5.new(key)
+        hmac.update(data)
+        return hmac.hexdigest
+        },
+    :other_len   => 0,
+    :other_data  => nil,
+    :error       => 0
+    })
+    
+    packet.push(:additional, tsig) 
+    
+    # use a fixed packet id so we get a known checksum
+    packet.header.id=(1234)
+    
+    # create the packet - this will fill in the 'mac' field
+    raw_packet = packet.data
+    
+    assert_equal(
+   "6365643161343964663364643264656131306638303633626465366236643465",
+     (packet.additional)[0].mac, 
+   'MAC correct')
+    
+  end
+end

data/test/tc_update.rb

@@ -0,0 +1,226 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require 'test/unit'
+require 'Net/DNS'
+class TestUpdate < Test::Unit::TestCase
+  def is_empty(string)
+    return true if string == nil || string.length == 0
+    
+    return (string == "; no data" || string == "; rdlength = 0");
+  end
+  
+  def test_update
+    #------------------------------------------------------------------------------
+    # Canned data.
+    #------------------------------------------------------------------------------
+    
+    zone	= "example.com";
+    name	= "foo.example.com";
+    klass	= "HS";
+    klass2  = "CH";
+    type	= "A";
+    ttl	    = 43200;
+    rdata	= "10.1.2.3";
+    rr      = nil;
+    
+    #------------------------------------------------------------------------------
+    # Packet creation.
+    #------------------------------------------------------------------------------
+    
+    packet = Net::DNS::Update.new_from_values(zone, klass);
+    z = (packet.zone)[0];
+    
+    assert(packet,                                'new() returned packet');  #2
+    assert_equal(packet.header.opcode, 'UPDATE',      'header opcode correct');  #3 
+    assert_equal(z.zname,  zone,                      'zname correct');          #4
+    assert_equal(z.zclass, klass,                     'zclass correct');         #5
+    assert_equal(z.ztype,  'SOA',                      'ztype correct');          #6       
+    
+    #------------------------------------------------------------------------------
+    # RRset exists (value-independent).
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.yxrrset("#{name} #{klass} #{type}");
+    
+    assert(rr,                                    'yxrrset() returned RR');  #7
+    assert_equal(name,                      rr.name,  'yxrrset - right name');   #8
+    assert_equal(0,                          rr.ttl,   'yxrrset - right TTL');    #9
+    assert_equal('ANY',                      rr.rrclass, 'yxrrset - right class');  #10
+    assert_equal(type,                      rr.type,  'yxrrset - right type');   #11
+    assert(is_empty(rr.rdatastr),                "yxrrset - data empty (#{rr.rdatastr})");   #12
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # RRset exists (value-dependent).
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.yxrrset("#{name} #{klass} #{type} #{rdata}");
+    
+    assert(rr,                                    'yxrrset() returned RR');  #13
+    assert_equal(name,                   rr.name,     'yxrrset - right name');   #14
+    assert_equal(0,                       rr.ttl,      'yxrrset - right TTL');    #15
+    assert_equal(klass,                  rr.rrclass,    'yxrrset - right class');  #16
+    assert_equal(type,                   rr.type,     'yxrrset - right type');   #17
+    assert_equal(rdata,                  rr.rdatastr, 'yxrrset - right data');   #18
+    
+    rr=nil
+    
+    #------------------------------------------------------------------------------
+    # RRset does not exist.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.nxrrset("#{name} #{klass} #{type}");
+    
+    assert(rr,                                    'nxrrset() returned RR');  #19
+    assert_equal(name,                      rr.name,  'nxrrset - right name');   #20
+    assert_equal(0,                          rr.ttl,   'nxrrset - right ttl');    #21
+    assert_equal('NONE',                     rr.rrclass, 'nxrrset - right class');  #22
+    assert_equal(type,                      rr.type,  'nxrrset - right type');   #23
+    assert(is_empty(rr.rdatastr),                'nxrrset - data empty');   #24
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Name is in use.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.yxdomain("#{name} #{klass}");
+    
+    assert(rr,                                    'yxdomain() returned RR'); #25
+    assert_equal(rr.name,  name,                      'yxdomain - right name');  #26
+    assert_equal(rr.ttl,   0,                          'yxdomain - right ttl');   #27
+    assert_equal(rr.rrclass, 'ANY',                      'yxdomain - right class'); #28
+    assert_equal(rr.type,  'ANY',                      'yxdomain - right type');  #29
+    assert(is_empty(rr.rdatastr),                'yxdomain - data empty');  #30
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Name is not in use.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.nxdomain("#{name} #{klass}");
+    
+    assert(rr,                                    'nxdomain() returned RR'); #31
+    assert_equal(rr.name,  name,                      'nxdomain - right name');  #32
+    assert_equal(rr.ttl,   0,                          'nxdomain - right ttl');   #33
+    assert_equal(rr.rrclass, 'NONE',                     'nxdomain - right class'); #34
+    assert_equal(rr.type,  'ANY',                      'nxdomain - right type');  #35
+    assert(is_empty(rr.rdatastr),                'nxdomain - data empty');  #36
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Name is not in use. (No Class)
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.nxdomain("#{name}");
+    
+    assert(rr,                                    'nxdomain() returned RR'); #31
+    assert_equal(rr.name,  name,                      'nxdomain - right name');  #32
+    assert_equal(rr.ttl,   0,                          'nxdomain - right ttl');   #33
+    assert_equal(rr.rrclass, 'NONE',                     'nxdomain - right class'); #34
+    assert_equal(rr.type,  'ANY',                      'nxdomain - right type');  #35
+    assert(is_empty(rr.rdatastr),                'nxdomain - data empty');  #36
+    
+    rr = nil
+    
+    
+    
+    #------------------------------------------------------------------------------
+    # Add to an RRset.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.rr_add("#{name} #{ttl} #{klass} #{type} #{rdata}");
+    
+    assert(rr,                                    'rr_add() returned RR');   #37
+    assert_equal(rr.name,     name,                   'rr_add - right name');    #38
+    assert_equal(rr.ttl,      ttl,                    'rr_add - right ttl');     #39
+    assert_equal(rr.rrclass,    klass,                  'rr_add - right class');   #40
+    assert_equal(rr.type,     type,                   'rr_add - right type');    #41
+    assert_equal(rr.rdatastr, rdata,                  'rr_add - right data');    #42
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Delete an RRset.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.rr_del("#{name} #{klass} #{type}");
+    
+    assert(rr,                                    'rr_del() returned RR');   #43
+    assert_equal(name,                      rr.name,  'rr_del - right name');    #44
+    assert_equal(0,                          rr.ttl,   'rr_del - right ttl');     #45
+    assert_equal('ANY',                      rr.rrclass, 'rr_del - right class');   #46
+    assert_equal(type,                      rr.type,  'rr_del - right type');    #47
+    assert(is_empty(rr.rdatastr),                'rr_del - data empty');    #48
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Delete All RRsets From A Name.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.rr_del("#{name} #{klass}");
+    
+    assert(rr,                                    'rr_del() returned RR');   #49
+    assert_equal(name,                      rr.name,  'rr_del - right name');    #50
+    assert_equal(0,                          rr.ttl,   'rr_del - right ttl');     #51
+    assert_equal('ANY',                      rr.rrclass, 'rr_del - right class');   #52
+    assert_equal('ANY',                      rr.type,  'rr_del - right type');    #53
+    assert(is_empty(rr.rdatastr),                'rr_del - data empty');    #54
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Delete An RR From An RRset.
+    #------------------------------------------------------------------------------
+    
+    rr = Net::DNS.rr_del("#{name} #{klass} #{type} #{rdata}");
+    
+    assert(rr,                                    'rr_del() returned RR');   #55
+    assert_equal(name,                   rr.name,     'rr_del - right name');    #56
+    assert_equal(0,                       rr.ttl,      'rr_del - right ttl');     #57
+    assert_equal('NONE',                  rr.rrclass,    'rr_del - right class');   #58
+    assert_equal(type,                   rr.type,     'rr_del - right type');    #59
+    assert_equal(rdata,                  rr.rdatastr, 'rr_del - right data');    #60
+    
+    rr = nil
+    
+    #------------------------------------------------------------------------------
+    # Make sure RRs in an update packet have the same class as the zone, unless
+    # the class is NONE or ANY.
+    #------------------------------------------------------------------------------
+    
+    packet = Net::DNS::Update.new_from_values(zone, klass);
+    assert(packet,                               'packet created');          #61
+    
+    
+    packet.push("pre", Net::DNS.yxrrset("#{name} #{klass} #{type} #{rdata}"));
+    packet.push("pre", Net::DNS.yxrrset("#{name} #{klass2} #{type} #{rdata}"));
+    packet.push("pre", Net::DNS.yxrrset("#{name} #{klass2} #{type}"));
+    packet.push("pre", Net::DNS.nxrrset("#{name} #{klass2} #{type}"));
+    
+    pre = packet.pre;
+    
+    assert_equal(4,                     pre.size, 'pushed inserted correctly'); #62
+    assert_equal(klass,              pre[0].rrclass, 'first class right');         #63
+    assert_equal(klass,              pre[1].rrclass, 'second class right');        #64
+    assert_equal('ANY',               pre[2].rrclass, 'third class right');         #65
+    assert_equal('NONE',              pre[3].rrclass, 'forth class right');         #66
+  end
+end

data/test/ts_netdns.rb

@@ -0,0 +1,17 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require "test/ts_online.rb"
+require "test/ts_offline.rb"

data/test/ts_offline.rb

@@ -0,0 +1,32 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require "test/unit"
+require "test/tc_escapedchars.rb"
+require "test/tc_header.rb"
+require "test/tc_misc.rb"
+require "test/tc_packet.rb"
+require "test/tc_packet_unique_push.rb"
+require "test/tc_question.rb"
+require "test/tc_res_file.rb"
+require "test/tc_res_opt.rb"
+require "test/tc_rr-opt.rb"
+require "test/tc_rr-rrsort.rb"
+require "test/tc_rr-txt.rb"
+require "test/tc_rr-unknown.rb"
+require "test/tc_rr.rb"
+require "test/tc_tkey.rb"
+require "test/tc_update.rb"
+require "test/tc_res_env.rb"

data/test/ts_online.rb

@@ -0,0 +1,33 @@
+# The contents of this file are subject to the Mozilla
+# Public Licence Version 1.1 (the "Licence"); you may
+# not use this file except in compliance with the
+# Licence. You may obtain a copy of the Licence at
+# http://www.mozilla.org/MPL
+# Software distributed under the Licence is distributed
+# on an "AS IS" basis,  WITHOUT WARRANTY OF ANY KIND,
+# either express or implied. See the Licence of the
+# specific language governing rights and limitations
+# under the Licence.
+# The Original Code is pNet::DNS.
+# The Initial Developer of the Original Code is
+# Nominet UK (www.nominet.org.uk). Portions created by
+# Nominet UK are Copyright (c) Nominet UK 2006.
+# All rights reserved.
+require "test/unit"
+# Disable these tests if we're not online
+require 'socket'
+sock = UDPSocket.new()
+begin
+  sock.connect('193.0.14.129', # k.root-servers.net.
+          '25')
+  # OK - online and ready to go
+  require "test/tc_inet6.rb"
+  require "test/tc_online.rb"
+  require "test/tc_recurse.rb"
+  require "test/tc_resolver.rb"
+  require "test/tc_tcp.rb"
+rescue Error
+  puts "Cannot bind to socket:\n\t"+$!+"\n"
+  puts "This is an indication you have network problems\n"
+  puts "\n\nNo online tests will be run!!\n\n"
+end