blob: 596a2044b59b7a0ee66d7943bd305e50782fdf4b (
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
|
require 'custodian/settings'
require 'custodian/testfactory'
require 'custodian/util/tftp'
require 'uri'
#
# The TFTP-protocol test
#
# This object is instantiated if the parser sees a line such as:
#
###
### tftp://foo.vm.bytemark.co.uk/pxelinux.0 must run tftp otherwise 'tftp fail'
###
#
#
module Custodian
module ProtocolTest
class TFTPTest < TestFactory
#
# The line from which we were constructed.
#
attr_reader :line
#
# The TFTP server
#
attr_reader :host
#
# The TFTP port
#
attr_reader :port
#
# The file on the TFTP server to test
#
attr_reader :file
#
# Constructor
#
def initialize(line)
#
# Save the line
#
@line = line
#
# Save the URL
#
url = line.split(/\s+/)[0]
#
# Ensure we've got a TFTP url.
#
if url !~ /^tftp:/
raise ArgumentError, "The target wasn't a TFTP URL: #{line}"
end
#
# Parse the URL which should have a host + path, and optional port
#
u = URI.parse(url)
#
# Record the values.
#
@host = u.host
@file = u.path
# Port might not be specified, if it is missing then default to 69.
@port = u.port || "69"
@port = @port.to_i
#
# Ensure there is a file to fetch
#
if @file.nil? || @file.empty?
raise ArgumentError, "Missing file name"
end
end
#
# Allow this test to be serialized.
#
def to_s
@line
end
#
# Run the test.
#
def run_test
tftp = Custodian::Util::Tftp.new(@host, @port)
if tftp.test(@file)
Custodian::TestResult::TEST_PASSED
else
@error = "TFTP failed for #{@file} on #{@host}"
Custodian::TestResult::TEST_FAILED
end
end
#
# If the test fails then report the error.
#
def error
@error
end
register_test_type 'tftp'
end
end
end
|
