Handle locally-defined hostnames, without full DNS info

purcell · Nov 6, 2012 · 271aee6 · 271aee6
1 parent aac460c
commit 271aee6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/postfix-policy-whois b/bin/postfix-policy-whois
@@ -34,7 +34,8 @@ require 'resolv'
 def nameserver_for(domain)
   Timeout::timeout(5) do
     begin
-      Resolv::DNS.new.getresource(domain, Resolv::DNS::Resource::IN::ANY).name.to_s
+      record = Resolv::DNS.new.getresource(domain, Resolv::DNS::Resource::IN::ANY)
+      record.name.to_s if record.respond_to?(:name)
     rescue Resolv::ResolvError
       nil
     end