Add input method fallback, e.g. if ssh does not work, try telnet

3 files changed, 27 insertions, 18 deletions

diff --git a/lib/oxidized/config/bootstrap.rb b/lib/oxidized/config/bootstrap.rb
index c3ea09d..6d865b5 100644
--- a/lib/oxidized/config/bootstrap.rb
+++ b/lib/oxidized/config/bootstrap.rb
@@ -12,7 +12,7 @@ module Oxidized
   CFG.prompt   = /^([\w\.\-@]{3,30}[#>]\s?)$/ 
   CFG.rest     = 8888
   CFG.input    = {
-    :default => 'ssh',
+    :default => 'ssh, telnet',
   }
   CFG.output    = {
     :default => 'git',
diff --git a/lib/oxidized/input/ssh.rb b/lib/oxidized/input/ssh.rb
index 45c89ba..3471eea 100644
--- a/lib/oxidized/input/ssh.rb
+++ b/lib/oxidized/input/ssh.rb
@@ -12,7 +12,7 @@ module Oxidized
       begin
         @ssh = Net::SSH.start @node.ip, @node.auth[:username],
                               :password => @node.auth[:password], :timeout => CFG.timeout
-      rescue Timeout::Error, Net::SSH::Disconnect
+      rescue Timeout::Error, Net::SSH::Disconnect, Errno::ECONNREFUSED
         return false
       end
       @ses = open_shell @ssh unless @exec
diff --git a/lib/oxidized/node.rb b/lib/oxidized/node.rb
index 35a5948..a41fcdc 100644
--- a/lib/oxidized/node.rb
+++ b/lib/oxidized/node.rb
@@ -10,7 +10,8 @@ module Oxidized
       @name           = opt[:name]
       @ip             = Resolv.getaddress @name
       @group          = opt[:group]
-      @input, @output = resolve_io opt
+      @input          = resolve_input opt
+      @output         = resolve_output opt
       @model          = resolve_model opt
       @auth           = resolve_auth opt
       @prompt         = resolve_prompt opt
@@ -18,12 +19,15 @@ module Oxidized
 
     def run
       status, config = :fail, nil
-      @model.input = input = @input.new
-      if input.connect self
-        config = input.get
-        status = :success if config
-      else
-        status = :no_cconnection
+      @input.each do |input|
+        @model.input = input = input.new
+        if input.connect self
+          config = input.get
+          status = :success if config
+          break
+        else
+          status = :no_cconnection
+        end
       end
       [status, config]
     end
@@ -66,17 +70,22 @@ module Oxidized
       auth
     end
 
-    def resolve_io opt
-      input  = (opt[:input]  or CFG.input[:default])
-      output = (opt[:output] or CFG.output[:default])
-      mgr = Oxidized.mgr
-      if not mgr.input[input]
-        mgr.input = input or raise MethodNotFound, "#{input} not found"
+    def resolve_input opt
+      inputs = (opt[:input]  or CFG.input[:default])
+      inputs.split(/\s*,\s*/).map do |input|
+        if not Oxidized.mgr.input[input]
+          Oxidized.mgr.input = input or raise MethodNotFound, "#{input} not found"
+        end
+        Oxidized.mgr.input[input]
       end
-      if not mgr.output[output]
-        mgr.output = output or raise MethodNotFound, "#{output} not found"
+    end
+
+    def resolve_output opt
+      output = (opt[:output] or CFG.output[:default])
+      if not Oxidized.mgr.output[output]
+        Oxidized.mgr.output = output or raise MethodNotFound, "#{output} not found"
       end
-      [ mgr.input[input], mgr.output[output] ]
+      Oxidized.mgr.output[output]
     end
 
     def resolve_model opt