215 lines
7.5 KiB
Ruby
215 lines
7.5 KiB
Ruby
require "ipaddr"
|
|
require "shellwords"
|
|
|
|
require "vagrant/util/platform"
|
|
require "vagrant/util/subprocess"
|
|
|
|
module VagrantPlugins
|
|
module SyncedFolderRSync
|
|
# This is a helper that abstracts out the functionality of rsyncing
|
|
# folders so that it can be called from anywhere.
|
|
class RsyncHelper
|
|
# This converts an rsync exclude pattern to a regular expression
|
|
# we can send to Listen.
|
|
def self.exclude_to_regexp(path, exclude)
|
|
start_anchor = false
|
|
|
|
if exclude.start_with?("/")
|
|
start_anchor = true
|
|
exclude = exclude[1..-1]
|
|
end
|
|
|
|
path = "#{path}/" if !path.end_with?("/")
|
|
regexp = "^#{Regexp.escape(path)}"
|
|
regexp += ".*" if !start_anchor
|
|
|
|
# This is REALLY ghetto, but its a start. We can improve and
|
|
# keep unit tests passing in the future.
|
|
exclude = exclude.gsub("**", "|||GLOBAL|||")
|
|
exclude = exclude.gsub("*", "|||PATH|||")
|
|
exclude = exclude.gsub("|||PATH|||", "[^/]*")
|
|
exclude = exclude.gsub("|||GLOBAL|||", ".*")
|
|
regexp += exclude
|
|
|
|
Regexp.new(regexp)
|
|
end
|
|
|
|
def self.rsync_single(machine, ssh_info, opts)
|
|
# Folder info
|
|
guestpath = opts[:guestpath]
|
|
hostpath = opts[:hostpath]
|
|
hostpath = File.expand_path(hostpath, machine.env.root_path)
|
|
hostpath = Vagrant::Util::Platform.fs_real_path(hostpath).to_s
|
|
|
|
# if the guest has a guest path scrubber capability, use it
|
|
if machine.guest.capability?(:rsync_scrub_guestpath)
|
|
guestpath = machine.guest.capability(:rsync_scrub_guestpath, opts)
|
|
end
|
|
|
|
# Shellescape
|
|
guestpath = Shellwords.escape(guestpath)
|
|
|
|
if Vagrant::Util::Platform.windows?
|
|
# rsync for Windows expects cygwin style paths, always.
|
|
hostpath = Vagrant::Util::Platform.cygwin_path(hostpath)
|
|
end
|
|
|
|
# Make sure the host path ends with a "/" to avoid creating
|
|
# a nested directory...
|
|
if !hostpath.end_with?("/")
|
|
hostpath += "/"
|
|
end
|
|
|
|
# Folder options
|
|
opts[:owner] ||= ssh_info[:username]
|
|
opts[:group] ||= ssh_info[:username]
|
|
|
|
# set log level
|
|
log_level = ssh_info[:log_level] || "FATAL"
|
|
|
|
# Connection information
|
|
# make it better match lib/vagrant/util/ssh.rb command_options style and logic
|
|
username = ssh_info[:username]
|
|
host = ssh_info[:host]
|
|
proxy_command = ""
|
|
if ssh_info[:proxy_command]
|
|
proxy_command = "-o ProxyCommand='#{ssh_info[:proxy_command]}' "
|
|
end
|
|
|
|
# Create the path for the control sockets. We used to do this
|
|
# in the machine data dir but this can result in paths that are
|
|
# too long for unix domain sockets.
|
|
control_options = ""
|
|
unless Vagrant::Util::Platform.windows?
|
|
controlpath = File.join(Dir.tmpdir, "ssh.#{rand(1000)}")
|
|
control_options = "-o ControlMaster=auto -o ControlPath=#{controlpath} -o ControlPersist=10m "
|
|
end
|
|
|
|
# rsh cmd option
|
|
rsh = [
|
|
"ssh", "-p", "#{ssh_info[:port]}",
|
|
"-o", "LogLevel=#{log_level}",
|
|
proxy_command,
|
|
control_options,
|
|
]
|
|
|
|
# Solaris/OpenSolaris/Illumos uses SunSSH which doesn't support the
|
|
# IdentitiesOnly option. Also, we don't enable it if keys_only is false
|
|
# so that SSH properly searches our identities and tries to do it itself.
|
|
if !Vagrant::Util::Platform.solaris? && ssh_info[:keys_only]
|
|
rsh += ["-o", "IdentitiesOnly=yes"]
|
|
end
|
|
|
|
# no strict hostkey checking unless paranoid
|
|
if ssh_info[:verify_host_key] == :never || !ssh_info[:verify_host_key]
|
|
rsh += [
|
|
"-o", "StrictHostKeyChecking=no",
|
|
"-o", "UserKnownHostsFile=/dev/null"]
|
|
end
|
|
|
|
# If specified, attach the private key paths.
|
|
if ssh_info[:private_key_path]
|
|
rsh += ssh_info[:private_key_path].map { |p| "-i '#{p}'" }
|
|
end
|
|
|
|
# Exclude some files by default, and any that might be configured
|
|
# by the user.
|
|
excludes = ['.vagrant/']
|
|
excludes += Array(opts[:exclude]).map(&:to_s) if opts[:exclude]
|
|
excludes.uniq!
|
|
|
|
# Get the command-line arguments
|
|
args = nil
|
|
args = Array(opts[:args]).dup if opts[:args]
|
|
args ||= ["--verbose", "--archive", "--delete", "-z", "--copy-links"]
|
|
|
|
# On Windows, we have to set a default chmod flag to avoid permission issues
|
|
if Vagrant::Util::Platform.windows? && !args.any? { |arg| arg.start_with?("--chmod=") }
|
|
# Ensures that all non-masked bits get enabled
|
|
args << "--chmod=ugo=rwX"
|
|
|
|
# Remove the -p option if --archive is enabled (--archive equals -rlptgoD)
|
|
# otherwise new files will not have the destination-default permissions
|
|
args << "--no-perms" if args.include?("--archive") || args.include?("-a")
|
|
end
|
|
|
|
# Disable rsync's owner/group preservation (implied by --archive) unless
|
|
# specifically requested, since we adjust owner/group to match shared
|
|
# folder setting ourselves.
|
|
args << "--no-owner" unless args.include?("--owner") || args.include?("-o")
|
|
args << "--no-group" unless args.include?("--group") || args.include?("-g")
|
|
|
|
# Tell local rsync how to invoke remote rsync with sudo
|
|
rsync_path = opts[:rsync_path]
|
|
if !rsync_path && machine.guest.capability?(:rsync_command)
|
|
rsync_path = machine.guest.capability(:rsync_command)
|
|
end
|
|
if rsync_path
|
|
args << "--rsync-path"<< rsync_path
|
|
end
|
|
|
|
# If the remote host is an IPv6 address reformat
|
|
begin
|
|
if IPAddr.new(host).ipv6?
|
|
host = "[#{host}]"
|
|
end
|
|
rescue IPAddr::Error
|
|
# Ignore
|
|
end
|
|
|
|
# Build up the actual command to execute
|
|
command = [
|
|
"rsync",
|
|
args,
|
|
"-e", rsh.flatten.join(" "),
|
|
excludes.map { |e| ["--exclude", e] },
|
|
hostpath,
|
|
"#{username}@#{host}:#{guestpath}",
|
|
].flatten
|
|
|
|
# The working directory should be the root path
|
|
command_opts = {}
|
|
command_opts[:workdir] = machine.env.root_path.to_s
|
|
|
|
machine.ui.info(I18n.t(
|
|
"vagrant.rsync_folder", guestpath: guestpath, hostpath: hostpath))
|
|
if excludes.length > 1
|
|
machine.ui.info(I18n.t(
|
|
"vagrant.rsync_folder_excludes", excludes: excludes.inspect))
|
|
end
|
|
if opts.include?(:verbose)
|
|
machine.ui.info(I18n.t("vagrant.rsync_showing_output"));
|
|
end
|
|
|
|
# If we have tasks to do before rsyncing, do those.
|
|
if machine.guest.capability?(:rsync_pre)
|
|
machine.guest.capability(:rsync_pre, opts)
|
|
end
|
|
|
|
if opts.include?(:verbose)
|
|
command_opts[:notify] = [:stdout, :stderr]
|
|
r = Vagrant::Util::Subprocess.execute(*(command + [command_opts])) {
|
|
|io_name,data| data.each_line { |line|
|
|
machine.ui.info("rsync[#{io_name}] -> #{line}") }
|
|
}
|
|
else
|
|
r = Vagrant::Util::Subprocess.execute(*(command + [command_opts]))
|
|
end
|
|
|
|
if r.exit_code != 0
|
|
raise Vagrant::Errors::RSyncError,
|
|
command: command.map(&:inspect).join(" "),
|
|
guestpath: guestpath,
|
|
hostpath: hostpath,
|
|
stderr: r.stderr
|
|
end
|
|
|
|
# If we have tasks to do after rsyncing, do those.
|
|
if machine.guest.capability?(:rsync_post)
|
|
machine.guest.capability(:rsync_post, opts)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|