summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/custodian/protocoltest/tcp.rb7
1 files changed, 5 insertions, 2 deletions
diff --git a/lib/custodian/protocoltest/tcp.rb b/lib/custodian/protocoltest/tcp.rb
index 684452b..c50c144 100644
--- a/lib/custodian/protocoltest/tcp.rb
+++ b/lib/custodian/protocoltest/tcp.rb
@@ -68,10 +68,13 @@ module Custodian
#
# Save the port
#
+ @port = nil
+
if line =~ /on\s+([0-9]+)/
@port = $1.dup
- else
- @port = nil
+ end
+ if line =~ /on\s+port\s+([0-9]+)/
+ @port = $1.dup
end
#