From 644b1a142967c07be3c6eaae4bc72402609ad663 Mon Sep 17 00:00:00 2001
From: Steve Kemp <steve@steve.org.uk>
Date: Thu, 15 Nov 2012 13:38:39 +0000
Subject:   DNS protocol-tester catches raised exceptions and runs under a
 timeout.

---
 lib/custodian/protocol-tests/dns.rb | 50 ++++++++++++++++++++++++-------------
 1 file changed, 33 insertions(+), 17 deletions(-)

(limited to 'lib/custodian/protocol-tests')

diff --git a/lib/custodian/protocol-tests/dns.rb b/lib/custodian/protocol-tests/dns.rb
index 4efabd3..a3396de 100755
--- a/lib/custodian/protocol-tests/dns.rb
+++ b/lib/custodian/protocol-tests/dns.rb
@@ -91,9 +91,13 @@ class DNSTest
     expected = @test_data["resolve_expected"]
 
     #
-    # Do the lookup
+    # Do the lookup abort if that fails.
     #
-    results = resolve_via( nameserver, record, lookup )
+    timeout = @test_data["timeout"].to_i
+
+    results = resolve_via( nameserver, record, lookup, timeout )
+    return false if ( results.nil? )
+
 
     #
     # Show the results if we found something.
@@ -122,26 +126,38 @@ class DNSTest
   #
   # Resolve an IP
   #
-  def resolve_via( server, ltype, name )
+  def resolve_via( server, ltype, name, period )
     puts "Resolving #{name} [#{ltype}] via server #{server}"
 
     results = Array.new()
 
-    Resolv::DNS.open(:nameserver=>[server]) do |dns|
-      case ltype
-      when /^A$/ then
-        dns.getresources(name, Resolv::DNS::Resource::IN::A).map{ |r| results.push r.address.to_s() }
-      when /^AAAA$/ then
-        dns.getresources(name, Resolv::DNS::Resource::IN::AAAA).map{ |r| results.push r.address.to_s() }
-
-      when /^NS$/ then
-        dns.getresources(name, Resolv::DNS::Resource::IN::MX).map{ |r| results.pushResolv.getaddresses(r.name.to_s()) }
-
-      when /^MX$/ then
-        dns.getresources(name, Resolv::DNS::Resource::IN::MX).map{ |r| results.push Resolv.getaddresses(r.exchange.to_s()) }
+    begin
+      timeout( period ) do
+
+        begin
+          Resolv::DNS.open(:nameserver=>[server]) do |dns|
+            case ltype
+            when /^A$/ then
+              dns.getresources(name, Resolv::DNS::Resource::IN::A).map{ |r| results.push r.address.to_s() }
+            when /^AAAA$/ then
+              dns.getresources(name, Resolv::DNS::Resource::IN::AAAA).map{ |r| results.push r.address.to_s() }
+
+            when /^NS$/ then
+              dns.getresources(name, Resolv::DNS::Resource::IN::MX).map{ |r| results.pushResolv.getaddresses(r.name.to_s()) }
+
+            when /^MX$/ then
+              dns.getresources(name, Resolv::DNS::Resource::IN::MX).map{ |r| results.push Resolv.getaddresses(r.exchange.to_s()) }
+            end
+          end
+        rescue Exception => x
+          @error = "Exception was received when resolving : #{x}"
+          return nil
+        end
       end
+    rescue Timeout::Error => e
+      @error = "Timed-out connecting #{e}"
+      return nil
     end
-
     results.flatten!
     return results
   end
@@ -173,7 +189,7 @@ if __FILE__ == $0 then
     "target_host"      => "a.ns.bytemark.co.uk",
     "test_type"        => "dns",
     "verbose"          => 1,
-    "timeout"          => 4,
+    "timeout"          => "4",
     "test_alert"       => "DNS failure",
     "resolve_name"     => "support.bytemark.co.uk",
     "resolve_type"     => "MX",
-- 
cgit v1.2.1