blob: 7a6abe37b6bb9089f706572b01d3c782feec3613 (
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
|
require 'custodian/settings'
require 'custodian/testfactory'
require 'custodian/util/tftp'
#
# 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
#
# Extract host, port and file from URL
#
if line =~ /^tftp:\/\/([^\/]+)\/([^\s]*)/
@host = $1.split(':')[0]
@file = $2.dup
p = $1.split(':')[1]
@port = (p && p.to_i > 0) ? p.to_i : 69
end
#
# 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
|
