require 'net/smtp' # # The open SMTP-relay test. # # This object is instantiated if the parser sees a line such as: # ### ### mail.bytemark.co.uk must not run smtprelay otherwise 'smtp fail'. ### # # The specification of the port is optional and defaults to 25 # module Custodian module ProtocolTest class SMTPRelayTest < TestFactory # # Save away state from the configuration line. # def initialize(line) @line = line @host = line.split(/\s+/)[0] # # Save the port # if line =~ /on\s+([0-9]+)/ @port = $1.dup else @port = 25 end # # Is this test inverted? # 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 # # Read the hostname for usage in the SMTP-transaction. # def get_hostname hostname = 'localhost.localdomain' if File.exist?('/etc/hostname') File.readlines('/etc/hostname').each do |line| hostname = line if !line.nil? hostname.chomp! end end hostname end # # run the test for open relays of SMTP protocol - return true on success. # false on fail. # def run_test # for if we've run the test before @error = nil message = "Subject: SMTP Relay check\nThis is a test for OPEN SMTP relays." begin Net::SMTP.start(@host, @port, get_hostname) do |smtp| sent = smtp.send_message message, 'noreply@bytemark.co.uk', 'noreply@bytemark.co.uk' @status = sent.status.to_s if @status === '250' @error = "NOT OK: message sent on #{@host} with status #{@status}" else @error = "OK: message not sent on #{@host} with status #{@status}" end # # give the parser an appropriate response depending on the smtp code # and whether or not we're inverting the test. (eg, 'must not') # return @inverted if @status == '250' and @inverted return !@inverted if @status == '250' and !@inverted return @inverted if @status != '250' and !@inverted return !@inverted if @status != '250' and @inverted end # Net SMTP rescue StandardError => ex # # for if we fail to send a message; this is a good thing # @error = "OK: Timed out or connection refused on #{@host} with status #{@status}" return !@inverted if @inverted return @inverted if !@inverted end end # # If the test failed here we will return a suitable error message. # def error @error end # register ourselves with the class-factory register_test_type 'smtprelay' end end end