135 lines
4.7 KiB
Ruby
135 lines
4.7 KiB
Ruby
module VagrantPlugins
|
|
module HostWindows
|
|
module Cap
|
|
class SMB
|
|
|
|
# Number of seconds to display UAC warning to user
|
|
UAC_PROMPT_WAIT = 4
|
|
|
|
@@logger = Log4r::Logger.new("vagrant::host::windows::smb")
|
|
|
|
def self.smb_installed(env)
|
|
psv = Vagrant::Util::PowerShell.version.to_i
|
|
if psv < 3
|
|
return false
|
|
end
|
|
|
|
true
|
|
end
|
|
|
|
def self.smb_cleanup(env, machine, opts)
|
|
script_path = File.expand_path("../../scripts/unset_share.ps1", __FILE__)
|
|
|
|
m_id = machine_id(machine)
|
|
prune_shares = existing_shares.map do |share_name, share_info|
|
|
if share_info["Description"].start_with?("vgt-#{m_id}-")
|
|
@@logger.info("removing smb share name=#{share_name} id=#{m_id}")
|
|
share_name
|
|
else
|
|
@@logger.info("skipping smb share removal, not owned name=#{share_name}")
|
|
@@logger.debug("smb share ID not present name=#{share_name} id=#{m_id} description=#{share_info["Description"]}")
|
|
nil
|
|
end
|
|
end.compact
|
|
|
|
@@logger.debug("shares to be removed: #{prune_shares}")
|
|
|
|
if prune_shares.size > 0
|
|
machine.env.ui.warn("\n" + I18n.t("vagrant_sf_smb.uac.prune_warning") + "\n")
|
|
sleep UAC_PROMPT_WAIT
|
|
@@logger.info("remove shares: #{prune_shares}")
|
|
result = Vagrant::Util::PowerShell.execute(script_path, *prune_shares, sudo: true)
|
|
if result.exit_code != 0
|
|
failed_name = result.stdout.to_s.sub("share name: ", "")
|
|
raise SyncedFolderSMB::Errors::PruneShareFailed,
|
|
name: failed_name,
|
|
stderr: result.stderr,
|
|
stdout: result.stdout
|
|
end
|
|
end
|
|
end
|
|
|
|
def self.smb_prepare(env, machine, folders, opts)
|
|
script_path = File.expand_path("../../scripts/set_share.ps1", __FILE__)
|
|
|
|
shares = []
|
|
current_shares = existing_shares
|
|
folders.each do |id, data|
|
|
hostpath = data[:hostpath].to_s
|
|
|
|
chksum_id = Digest::MD5.hexdigest(id)
|
|
name = "vgt-#{machine_id(machine)}-#{chksum_id}"
|
|
data[:smb_id] ||= name
|
|
|
|
# Check if this name is already in use
|
|
if share_info = current_shares[data[:smb_id]]
|
|
if !hostpath.empty? && share_info["Path"].downcase != hostpath.downcase
|
|
raise SyncedFolderSMB::Errors::SMBNameError,
|
|
path: hostpath,
|
|
existing_path: share_info["Path"],
|
|
name: data[:smb_id]
|
|
end
|
|
@@logger.info("skip creation of existing share name=#{name} id=#{data[:smb_id]}")
|
|
next
|
|
end
|
|
|
|
@@logger.info("creating new share name=#{name} id=#{data[:smb_id]}")
|
|
|
|
shares << [
|
|
"\"#{hostpath.gsub("/", "\\")}\"",
|
|
name,
|
|
data[:smb_id]
|
|
]
|
|
end
|
|
if !shares.empty?
|
|
machine.env.ui.warn("\n" + I18n.t("vagrant_sf_smb.uac.create_warning") + "\n")
|
|
sleep(UAC_PROMPT_WAIT)
|
|
result = Vagrant::Util::PowerShell.execute(script_path, *shares, sudo: true)
|
|
if result.exit_code != 0
|
|
share_path = result.stdout.to_s.sub("share path: ", "")
|
|
raise SyncedFolderSMB::Errors::DefineShareFailed,
|
|
host: share_path,
|
|
stderr: result.stderr,
|
|
stdout: result.stdout
|
|
end
|
|
end
|
|
end
|
|
|
|
# Generate a list of existing local smb shares
|
|
#
|
|
# @return [Hash]
|
|
def self.existing_shares
|
|
result = Vagrant::Util::PowerShell.execute_cmd("Get-SmbShare|Format-List")
|
|
if result.nil?
|
|
raise SyncedFolderSMB::Errors::SMBListFailed
|
|
end
|
|
shares = {}
|
|
name = nil
|
|
result.lines.each do |line|
|
|
key, value = line.split(":", 2).map(&:strip)
|
|
if key == "Name"
|
|
name = value
|
|
shares[name] = {}
|
|
end
|
|
next if name.nil? || key.to_s.empty?
|
|
shares[name][key] = value
|
|
end
|
|
@@logger.debug("local share listing: #{shares}")
|
|
shares
|
|
end
|
|
|
|
# Generates a unique identifier for the given machine
|
|
# based on the name, provider name, and working directory
|
|
# of the environment.
|
|
#
|
|
# @param [Vagrant::Machine] machine
|
|
# @return [String]
|
|
def self.machine_id(machine)
|
|
@@logger.debug("generating machine ID name=#{machine.name} cwd=#{machine.env.cwd}")
|
|
Digest::MD5.hexdigest("#{machine.name}-#{machine.provider_name}-#{machine.env.cwd}")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|