4e81be879c
With this change, any caller of machine.ssh_info is assured that best efforts will be done to fix possible wrong permissions on the private key files. Fix #4652
61 lines
1.8 KiB
Ruby
61 lines
1.8 KiB
Ruby
require "log4r"
|
|
|
|
require "vagrant/util/platform"
|
|
require "vagrant/util/ssh"
|
|
require "vagrant/util/shell_quote"
|
|
|
|
module Vagrant
|
|
module Action
|
|
module Builtin
|
|
# This class will run a single command on the remote machine and will
|
|
# mirror the output to the UI. The resulting exit status of the command
|
|
# will exist in the `:ssh_run_exit_status` key in the environment.
|
|
class SSHRun
|
|
# For quick access to the `SSH` class.
|
|
include Vagrant::Util
|
|
|
|
def initialize(app, env)
|
|
@app = app
|
|
@logger = Log4r::Logger.new("vagrant::action::builtin::ssh_run")
|
|
end
|
|
|
|
def call(env)
|
|
# Grab the SSH info from the machine
|
|
info = env[:machine].ssh_info
|
|
|
|
# If the result is nil, then the machine is telling us that it is
|
|
# not yet ready for SSH, so we raise this exception.
|
|
raise Errors::SSHNotReady if info.nil?
|
|
|
|
info[:private_key_path] ||= []
|
|
|
|
if info[:private_key_path].empty?
|
|
raise Errors::SSHRunRequiresKeys
|
|
end
|
|
|
|
# Get the command and wrap it in a login shell
|
|
command = ShellQuote.escape(env[:ssh_run_command], "'")
|
|
command = "#{env[:machine].config.ssh.shell} -c '#{command}'"
|
|
|
|
# Execute!
|
|
opts = env[:ssh_opts] || {}
|
|
opts[:extra_args] ||= []
|
|
|
|
# Allow the user to specify a tty or non-tty manually, but if they
|
|
# don't then we default to a TTY
|
|
if !opts[:extra_args].include?("-t") && !opts[:extra_args].include?("-T")
|
|
opts[:extra_args] << "-t"
|
|
end
|
|
|
|
opts[:extra_args] << command
|
|
opts[:subprocess] = true
|
|
env[:ssh_run_exit_status] = Util::SSH.exec(info, opts)
|
|
|
|
# Call the next middleware
|
|
@app.call(env)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|