require 'custodian/protocoltest/tcp' # # The MX (DNS + smtp) test. # # This object is instantiated if the parser sees a line such as: # ### ### bytemark.co.uk must run mx otherwise 'mail fail'. ### # # module Custodian module ProtocolTest class MXTest < TestFactory # # Constructor # def initialize( line ) # Save the line away @line = line # The main domain we're querying @host = line.split(/\s+/)[0] if ( line =~ /must\s+not\s+run\s+/ ) @inverted = true else @inverted = false end end # # Allow this test to be serialized. # def to_s @line end # # Run the test. # def run_test # reset the error, in case we were previously executed. @error = nil # # Get the timeout period. # settings = Custodian::Settings.instance() period = settings.timeout() # # The MX-hosts # mx = [] # # Lookup the MX record # begin timeout( period ) do Resolv::DNS.open do |dns| ress = dns.getresources(@host, Resolv::DNS::Resource::IN::MX) ress.map { |r| mx.push( IPSocket.getaddress(r.exchange.to_s) ) } end end rescue Timeout::Error => e @error = "Timed-out performing DNS lookups: #{e}" return nil end # # At this point we should have an array of IPv4 or IPv6 addresses. # # If that array is empty then there will be no incoming mail because # there are now working MX records in DNS - or because the domain # has expired, etc. # # So on that basis we must alert. # if ( mx.empty? ) then @error = "Failed to perform DNS lookup of MX record(s) for host #{@host}" return false end # # For each host we must make a connection. # # We'll keep count of failures. # failed = 0 passed = 0 error = "" mx.each do |backend| begin timeout(period) do begin socket = TCPSocket.new( backend, 25 ) read = socket.sysread(1024) # trim to a sane length & strip newlines. if ( ! read.nil? ) read = read[0,255] read.gsub!(/[\n\r]/, "") end if ( read =~ /^220/ ) passed += 1 else failed += 1 end rescue # Failure to connect. failed +=1 error += "Error connecting to #{backend}:25. " end end rescue Timeout::Error => ex # Timeout failed +=1 error += "Timeout connecting to #{backend}:25. " end end # # At this point we should have tested the things # if ( failed > 0 ) @error = "There are #{mx.size} hosts running as MX-servers for domain #{@host} - #{passed}:OK #{failed}:FAILED - #{error}" return false else return true; end end # # If the test fails then report the error. # def error @error end register_test_type "mx" end end end