Class: Bolt::Transport::SSH::ExecConnection

Inherits:
Object
  • Object
show all
Defined in:
lib/bolt/transport/ssh/exec_connection.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(target) ⇒ ExecConnection

Returns a new instance of ExecConnection.

Raises:



11
12
13
14
15
16
17
18
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 11

def initialize(target)
  raise Bolt::ValidationError, "Target #{target.safe_name} does not have a host" unless target.host

  @target = target
  ssh_config = Net::SSH::Config.for(target.host)
  @user = @target.user || ssh_config[:user] || Etc.getlogin
  @logger = Logging.logger[self]
end

Instance Attribute Details

#target ⇒ Object (readonly)

Returns the value of attribute target.



9
10
11
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 9

def target
  @target
end

#user ⇒ Object (readonly)

Returns the value of attribute user.



9
10
11
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 9

def user
  @user
end

Instance Method Details

#build_ssh_command(command) ⇒ Object 



60
61
62
63
64
65
66
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 60

def build_ssh_command(command)
  ssh_conf = @target.transport_config['ssh-command']
  ssh_cmd = Array(ssh_conf)
  ssh_cmd += ssh_opts
  ssh_cmd << userhost
  ssh_cmd << command
end

#connect ⇒ Object

This is used to verify we can connect to targets with ‘connected?`



21
22
23
24
25
26
27
28
29
30
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 21

def connect
  cmd = build_ssh_command('exit')
  _, err, stat = Open3.capture3(*cmd)
  unless stat.success?
    raise Bolt::Node::ConnectError.new(
      "Failed to connect to #{@target.safe_name}: #{err}",
      'CONNECT_ERROR'
    )
  end
end

#copy_file(source, dest) ⇒ Object 



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
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 68

def copy_file(source, dest)
  @logger.debug { "Uploading #{source}, to #{userhost}:#{dest}" } unless source.is_a?(StringIO)

  cp_conf = @target.transport_config['copy-command'] || ["scp", "-r"]
  cp_cmd = Array(cp_conf)
  cp_cmd += ssh_opts

  _, err, stat = if source.is_a?(StringIO)
                   Tempfile.create(File.basename(dest)) do |f|
                     f.write(source.read)
                     f.close
                     cp_cmd << f.path
                     cp_cmd << "#{userhost}:#{Shellwords.escape(dest)}"
                     Open3.capture3(*cp_cmd)
                   end
                 else
                   cp_cmd << source
                   cp_cmd << "#{userhost}:#{Shellwords.escape(dest)}"
                   Open3.capture3(*cp_cmd)
                 end

  if stat.success?
    @logger.debug "Successfully uploaded #{source} to #{dest}"
  else
    message = "Could not copy file to #{dest}: #{err}"
    raise Bolt::Node::FileError.new(message, 'COPY_ERROR')
  end
end

#disconnect ⇒ Object 



32
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 32

def disconnect; end

#execute(command) ⇒ Object 



97
98
99
100
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 97

def execute(command)
  cmd_array = build_ssh_command(command)
  Open3.popen3(*cmd_array)
end

#reset_cwd? ⇒ Boolean

This is used by the Bash shell to decide whether to ‘cd` before executing commands as a run-as user

Returns:

  • (Boolean) 


104
105
106
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 104

def reset_cwd?
  true
end

#shell ⇒ Object 



34
35
36
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 34

def shell
  Bolt::Shell::Bash.new(@target, self)
end

#ssh_opts ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 42

def ssh_opts
  cmd = []
  # BatchMode is SSH's noninteractive option: if key authentication
  # fails it will error out instead of falling back to password prompt
  cmd += %w[-o BatchMode=yes]
  cmd += %W[-o Port=#{@target.port}] if @target.port

  if @target.transport_config.key?('host-key-check')
    hkc = @target.transport_config['host-key-check'] ? 'yes' : 'no'
    cmd += %W[-o StrictHostKeyChecking=#{hkc}]
  end

  if (key = target.transport_config['private-key'])
    cmd += ['-i', key]
  end
  cmd
end

#userhost ⇒ Object 



38
39
40
 
# File 'lib/bolt/transport/ssh/exec_connection.rb', line 38

def userhost
  "#{@user}@#{@target.host}"
end