154 lines
4.4 KiB
Ruby
154 lines
4.4 KiB
Ruby
require "timeout"
|
|
|
|
require "log4r"
|
|
|
|
require "vagrant/util/retryable"
|
|
require "vagrant/util/silence_warnings"
|
|
|
|
Vagrant::Util::SilenceWarnings.silence! do
|
|
require "winrm"
|
|
end
|
|
|
|
require_relative "file_manager"
|
|
|
|
module VagrantPlugins
|
|
module CommunicatorWinRM
|
|
class WinRMShell
|
|
include Vagrant::Util::Retryable
|
|
|
|
# These are the exceptions that we retry because they represent
|
|
# errors that are generally fixed from a retry and don't
|
|
# necessarily represent immediate failure cases.
|
|
@@exceptions_to_retry_on = [
|
|
HTTPClient::KeepAliveDisconnected,
|
|
WinRM::WinRMHTTPTransportError,
|
|
Errno::EACCES,
|
|
Errno::EADDRINUSE,
|
|
Errno::ECONNREFUSED,
|
|
Errno::ECONNRESET,
|
|
Errno::ENETUNREACH,
|
|
Errno::EHOSTUNREACH,
|
|
Timeout::Error
|
|
]
|
|
|
|
attr_reader :logger
|
|
attr_reader :host
|
|
attr_reader :port
|
|
attr_reader :username
|
|
attr_reader :password
|
|
attr_reader :config
|
|
|
|
def initialize(host, port, config)
|
|
@logger = Log4r::Logger.new("vagrant::communication::winrmshell")
|
|
@logger.debug("initializing WinRMShell")
|
|
|
|
@host = host
|
|
@port = port
|
|
@username = config.username
|
|
@password = config.password
|
|
@config = config
|
|
end
|
|
|
|
def powershell(command, &block)
|
|
# ensure an exit code
|
|
command << "\r\n"
|
|
command << "if ($?) { exit 0 } else { if($LASTEXITCODE) { exit $LASTEXITCODE } else { exit 1 } }"
|
|
execute_shell(command, :powershell, &block)
|
|
end
|
|
|
|
def cmd(command, &block)
|
|
execute_shell(command, :cmd, &block)
|
|
end
|
|
|
|
def wql(query, &block)
|
|
execute_shell(query, :wql, &block)
|
|
end
|
|
|
|
def upload(from, to)
|
|
FileManager.new(self).upload(from, to)
|
|
end
|
|
|
|
def download(from, to)
|
|
FileManager.new(self).download(from, to)
|
|
end
|
|
|
|
protected
|
|
|
|
def execute_shell(command, shell=:powershell, &block)
|
|
raise Errors::WinRMInvalidShell, shell: shell unless [:powershell, :cmd, :wql].include?(shell)
|
|
|
|
begin
|
|
execute_shell_with_retry(command, shell, &block)
|
|
rescue => e
|
|
raise_winrm_exception(e, shell, command)
|
|
end
|
|
end
|
|
|
|
def execute_shell_with_retry(command, shell, &block)
|
|
retryable(tries: @config.max_tries, on: @@exceptions_to_retry_on, sleep: 10) do
|
|
@logger.debug("#{shell} executing:\n#{command}")
|
|
output = session.send(shell, command) do |out, err|
|
|
block.call(:stdout, out) if block_given? && out
|
|
block.call(:stderr, err) if block_given? && err
|
|
end
|
|
@logger.debug("Output: #{output.inspect}")
|
|
return output
|
|
end
|
|
end
|
|
|
|
def raise_winrm_exception(winrm_exception, shell, command)
|
|
# If the error is a 401, we can return a more specific error message
|
|
if winrm_exception.message.include?("401")
|
|
raise Errors::AuthError,
|
|
user: @config.username,
|
|
password: @config.password,
|
|
endpoint: endpoint,
|
|
message: winrm_exception.message
|
|
end
|
|
|
|
raise Errors::ExecutionError,
|
|
shell: shell,
|
|
command: command,
|
|
message: winrm_exception.message
|
|
end
|
|
|
|
def new_session
|
|
@logger.info("Attempting to connect to WinRM...")
|
|
@logger.info(" - Host: #{@host}")
|
|
@logger.info(" - Port: #{@port}")
|
|
@logger.info(" - Username: #{@config.username}")
|
|
@logger.info(" - Transport: #{@config.transport}")
|
|
|
|
client = ::WinRM::WinRMWebService.new(endpoint, @config.transport.to_sym, endpoint_options)
|
|
client.set_timeout(@config.timeout)
|
|
client.toggle_nori_type_casting(:off) #we don't want coersion of types
|
|
client
|
|
end
|
|
|
|
def session
|
|
@session ||= new_session
|
|
end
|
|
|
|
def endpoint
|
|
case @config.transport.to_sym
|
|
when :ssl
|
|
"https://#{@host}:#{@port}/wsman"
|
|
when :plaintext
|
|
"http://#{@host}:#{@port}/wsman"
|
|
else
|
|
raise Errors::WinRMInvalidTransport, transport: @config.transport
|
|
end
|
|
end
|
|
|
|
def endpoint_options
|
|
{ user: @username,
|
|
pass: @password,
|
|
host: @host,
|
|
port: @port,
|
|
basic_auth_only: true,
|
|
no_ssl_peer_verification: !@config.ssl_peer_verification }
|
|
end
|
|
end #WinShell class
|
|
end
|
|
end
|