Give up after timeout on SSH#expect

1 files changed, 6 insertions, 3 deletions

diff --git a/lib/oxidized/input/ssh.rb b/lib/oxidized/input/ssh.rb
index 24401dd..a0a5916 100644
--- a/lib/oxidized/input/ssh.rb
+++ b/lib/oxidized/input/ssh.rb
@@ -1,5 +1,6 @@
 module Oxidized
   require 'net/ssh'
+  require 'timeout'
   require 'oxidized/input/cli'
   class SSH < Input
     RescueFail = {
@@ -83,9 +84,11 @@ module Oxidized
     end
 
     def expect regexp
-      @ssh.loop(0.1) do
-        sleep 0.1
-        not @output.match regexp
+      Timeout::timeout(CFG.timeout) do
+        @ssh.loop(0.1) do
+          sleep 0.1
+          not @output.match regexp
+        end
       end
     end