Class: Fog::SSH::Real

Inherits:
Object
  • Object
show all
Defined in:
lib/fog/core/ssh.rb

Instance Method Summary collapse

Constructor Details

#initialize(address, username, options) ⇒ Real 



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/fog/core/ssh.rb', line 40

def initialize(address, username, options)
  require 'net/ssh'

  key_manager = Net::SSH::Authentication::KeyManager.new(nil, options)

  unless options[:key_data] || options[:keys] || options[:password] || key_manager.agent
    raise ArgumentError.new(':key_data, :keys, :password or a loaded ssh-agent is required to initialize SSH')
  end

  options[:timeout] ||= 30
  if options[:key_data] || options[:keys]
    options[:keys_only] = true
    #Explicitly set these so net-ssh doesn't add the default keys
    #as seen at https://github.com/net-ssh/net-ssh/blob/master/lib/net/ssh/authentication/session.rb#L131-146
    options[:keys] = [] unless options[:keys]
    options[:key_data] = [] unless options[:key_data]
  end

  @address  = address
  @username = username
  @options  = { :paranoid => false }.merge(options)
end

Instance Method Details

#run(commands, &blk) ⇒ Object 



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
 
# File 'lib/fog/core/ssh.rb', line 63

def run(commands, &blk)
  commands = [*commands]
  results  = []
  begin
    Net::SSH.start(@address, @username, @options) do |ssh|
      commands.each do |command|
        result = Result.new(command)
        ssh.open_channel do |ssh_channel|
          ssh_channel.request_pty
          ssh_channel.exec(command) do |channel, success|
            unless success
              raise "Could not execute command: #{command.inspect}"
            end

            channel.on_data do |ch, data|
              result.stdout << data
              yield [data, ''] if blk
            end

            channel.on_extended_data do |ch, type, data|
              next unless type == 1
              result.stderr << data
              yield ['', data] if blk
            end

            channel.on_request('exit-status') do |ch, data|
              result.status = data.read_long
            end

            channel.on_request('exit-signal') do |ch, data|
              result.status = 255
            end
          end
        end
        ssh.loop
        results << result
      end
    end
  rescue Net::SSH::HostKeyMismatch => exception
    exception.remember_host!
    sleep 0.2
    retry
  end
  results
end