78e8e17869
Force string type prior to empty? check to prevent errors if host ip value is nil. Add coverage for proper handling of nil value.
84 lines
2.6 KiB
Ruby
84 lines
2.6 KiB
Ruby
module VagrantPlugins
|
|
module CommunicatorWinRM
|
|
# This is a helper module that provides some functions to the
|
|
# communicator. This is extracted into a module so that we can
|
|
# easily unit test these methods.
|
|
module Helper
|
|
# Returns the host and port to access WinRM.
|
|
#
|
|
# This asks the provider via the `winrm_info` capability if it
|
|
# exists, otherwise defaulting to its own heuristics.
|
|
#
|
|
# @param [Vagrant::Machine] machine
|
|
# @return [Hash]
|
|
def self.winrm_info(machine)
|
|
info = {}
|
|
if machine.provider.capability?(:winrm_info)
|
|
info = machine.provider.capability(:winrm_info)
|
|
raise Errors::WinRMNotReady if !info
|
|
end
|
|
|
|
info[:host] ||= winrm_address(machine)
|
|
info[:port] ||= winrm_port(machine, info[:host] == "127.0.0.1")
|
|
return info
|
|
end
|
|
|
|
# Returns the address to access WinRM. This does not contain
|
|
# the port.
|
|
#
|
|
# @param [Vagrant::Machine] machine
|
|
# @return [String]
|
|
def self.winrm_address(machine)
|
|
addr = machine.config.winrm.host
|
|
return addr if addr
|
|
|
|
ssh_info = machine.ssh_info
|
|
raise Errors::WinRMNotReady if !ssh_info || ssh_info[:host].to_s.empty?
|
|
return ssh_info[:host]
|
|
end
|
|
|
|
# Returns the port to access WinRM.
|
|
#
|
|
# @param [Vagrant::Machine] machine
|
|
# @return [Integer]
|
|
def self.winrm_port(machine, local=true)
|
|
host_port = machine.config.winrm.port
|
|
if machine.config.winrm.guest_port
|
|
# If we're not requesting a local port, return
|
|
# the guest port directly.
|
|
return machine.config.winrm.guest_port if !local
|
|
|
|
# Search by guest port if we can. We use a provider capability
|
|
# if we have it. Otherwise, we just scan the Vagrantfile defined
|
|
# ports.
|
|
port = nil
|
|
if machine.provider.capability?(:forwarded_ports)
|
|
machine.provider.capability(:forwarded_ports).each do |host, guest|
|
|
if guest == machine.config.winrm.guest_port
|
|
port = host
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
if !port
|
|
machine.config.vm.networks.each do |type, netopts|
|
|
next if type != :forwarded_port
|
|
next if !netopts[:host]
|
|
if netopts[:guest] == machine.config.winrm.guest_port
|
|
port = netopts[:host]
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
# Set it if we found it
|
|
host_port = port if port
|
|
end
|
|
|
|
host_port
|
|
end
|
|
end
|
|
end
|
|
end
|