summaryrefslogtreecommitdiff
path: root/worker/tests/ftp.rb
blob: a25fe276af13a67c7e1453a4c6af6b66582601ec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/ruby 


require 'timeout'
require 'socket'

#
# This class is responsible for performing tests against a remote FTP server
#
#
class FTPTest

  #
  # Data passed from the JSON hash.
  #
  attr_reader :test_data

  #
  # The error text we return on failure.
  #
  attr_reader :error




  #
  # Save the data away.
  #
  def initialize( data )
    @test_data = data
  end


  #
  # Run the test.
  #
  #  Return "true" on success
  #
  #  Return "false" on failure.
  #
  # If the test fails the details should be retrieved from "get_details".
  #
  def run_test

    #
    #  Get the hostname & port to test against.
    #
    host = @test_data["target_host"]
    port = @test_data["test_port"]

    puts "FTP testing host #{host}:#{port}"

    begin
      timeout(3) do

        begin
          socket = TCPSocket.new( host, port )
          socket.puts( "QUIT")

          banner = socket.gets(nil)
          banner = banner[0,20]

          socket.close()

          if ( banner =~ /^220/ )
            return true
          else
            @error = "Banner didn't report OK: #{banner}"
          end
        rescue
          @error = "FTP exception on host #{host}:#{port} - #{$!}"
          return false
        end
      end
    rescue Timeout::Error => e
      @error = "Timed-out connecting #{e}"
      return false
    end
    @error = "Misc. failure."
    return false
  end


  #
  #  Return the error.
  #
  def error
    return @error
  end

end





#
# Sample test, for testing.
#
if __FILE__ == $0 then

  #
  #  Sample data.
  #
  test = {
    "target_host" => "mirror.bytemark.co.uk",
    "test_type"   => "ftp",
    "test_port"   => 21,
    "test_alert"  => "The FTP server no worky",
  }


  #
  #  Run the test.
  #
  tst = FTPTest.new( test )
  if ( tst.run_test )
    puts "TEST OK"
  else
    puts "TEST FAILED"
    puts tst.get_details()
  end

end