summaryrefslogtreecommitdiff
path: root/lib/custodian/protocoltest/tcp.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/custodian/protocoltest/tcp.rb')
-rw-r--r--lib/custodian/protocoltest/tcp.rb105
1 files changed, 102 insertions, 3 deletions
diff --git a/lib/custodian/protocoltest/tcp.rb b/lib/custodian/protocoltest/tcp.rb
index fef6dea..115bf6a 100644
--- a/lib/custodian/protocoltest/tcp.rb
+++ b/lib/custodian/protocoltest/tcp.rb
@@ -1,3 +1,7 @@
+require 'socket'
+require 'timeout'
+
+
#
# The TCP-protocol test.
#
@@ -13,21 +17,70 @@ class TCPTest < ProtocolTest
#
+ # The host to test against.
+ #
+ attr_reader :host
+
+
+ #
+ # The port to connect to.
+ #
+ attr_reader :port
+
+
+ #
+ # The banner to look for, may be nil.
+ #
+ attr_reader :banner
+
+
+
+
+ #
# Constructor
#
# Ensure we received a port to run the TCP-test against.
#
def initialize( line )
- raise ArgumentError, "Missing port" unless ( line =~ /on\s+([0-9]+)/ );
+
+ #
+ # Save the host
+ #
+ @host = line.split( /\s+/)[0]
+
+ #
+ # Save the port
+ #
+ if ( line =~ /on\s+([0-9]+)/ )
+ @port = $1.dup
+ else
+ @port = nil
+ end
+
+ #
+ # Save the optional banner.
+ #
+ if ( line =~ /with\s+banner\s+'([^']+)'/ )
+ @banner = $1.dup
+ else
+ @banner = nil
+ end
+
@error = nil
+
+ if ( @port.nil? )
+ raise ArgumentError, "Missing port to test against"
+ end
end
+
+
#
# Helper for development.
#
- def display
- puts "I'm a TCP-test!"
+ def to_s
+ "tcp-test of #{@host}:#{@port} looking for banner '#{@banner}'."
end
@@ -39,6 +92,52 @@ class TCPTest < ProtocolTest
# reset the error, in case we were previously executed.
@error = nil
+ return( run_test_internal( @host, @port, @banner ) )
+ end
+
+
+ #
+ #
+ #
+ def run_test_internal( host, port, banner )
+ begin
+ timeout(30) do
+ begin
+ socket = TCPSocket.new( host, port )
+ socket.puts( "QUIT")
+
+ # read a banner from the remote server
+ read = socket.gets(nil)
+
+ # trim to a sane length & strip newlines.
+ read = read[0,255] unless ( read.nil? )
+ read.gsub!(/[\n\r]/, "") unless ( read.nil? )
+
+ socket.close()
+
+ if ( banner.nil? )
+ @error = nil
+ return true
+ else
+ # test for banner
+ if ( ( !read.nil? ) && ( read =~ /#{banner}/i ) )
+ return true
+ end
+
+ @error = "We expected a banner matching '#{banner}' but we got '#{read}'"
+ return false
+ end
+ rescue
+ @error = "Exception connecting to host #{host}:#{port} - #{$!}"
+ return false
+ end
+ end
+ rescue Timeout::Error => e
+ @error = "TIMEOUT: #{e}"
+ return false
+ end
+ @error = "Misc failure"
+ return false
end