128 lines
4.1 KiB
Ruby
128 lines
4.1 KiB
Ruby
require "vagrant/util"
|
|
require "vagrant/util/retryable"
|
|
|
|
module VagrantPlugins
|
|
module HostLinux
|
|
module Cap
|
|
class NFS
|
|
extend Vagrant::Util::Retryable
|
|
|
|
def self.nfs_apply_command(env)
|
|
"/usr/bin/exportfs -ar"
|
|
end
|
|
|
|
def self.nfs_check_command(env)
|
|
"/etc/init.d/nfs-kernel-server status"
|
|
end
|
|
|
|
def self.nfs_start_command(env)
|
|
"/etc/init.d/nfs-kernel-server start"
|
|
end
|
|
|
|
def self.nfs_export(env, ui, id, ips, folders)
|
|
# Get some values we need before we do anything
|
|
nfs_apply_command = env.host.capability(:nfs_apply_command)
|
|
nfs_check_command = env.host.capability(:nfs_check_command)
|
|
nfs_start_command = env.host.capability(:nfs_start_command)
|
|
|
|
nfs_opts_setup(folders)
|
|
output = Vagrant::Util::TemplateRenderer.render('nfs/exports_linux',
|
|
:uuid => id,
|
|
:ips => ips,
|
|
:folders => folders,
|
|
:user => Process.uid)
|
|
|
|
ui.info I18n.t("vagrant.hosts.linux.nfs_export")
|
|
sleep 0.5
|
|
|
|
nfs_cleanup(id)
|
|
|
|
output.split("\n").each do |line|
|
|
# This should only ask for administrative permission once, even
|
|
# though its executed in multiple subshells.
|
|
system(%Q[sudo su root -c "echo '#{line}' >> /etc/exports"])
|
|
end
|
|
|
|
if nfs_running?(nfs_check_command)
|
|
system("sudo #{nfs_apply_command}")
|
|
else
|
|
system("sudo #{nfs_start_command}")
|
|
end
|
|
end
|
|
|
|
def self.nfs_installed(environment)
|
|
retryable(:tries => 10, :on => TypeError) do
|
|
# Check procfs to see if NFSd is a supported filesystem
|
|
system("cat /proc/filesystems | grep nfsd > /dev/null 2>&1")
|
|
end
|
|
end
|
|
|
|
def self.nfs_prune(environment, ui, valid_ids)
|
|
return if !File.exist?("/etc/exports")
|
|
|
|
logger = Log4r::Logger.new("vagrant::hosts::linux")
|
|
logger.info("Pruning invalid NFS entries...")
|
|
|
|
output = false
|
|
user = Process.uid
|
|
|
|
File.read("/etc/exports").lines.each do |line|
|
|
if id = line[/^# VAGRANT-BEGIN:( #{user})? ([A-Za-z0-9-]+?)$/, 2]
|
|
if valid_ids.include?(id)
|
|
logger.debug("Valid ID: #{id}")
|
|
else
|
|
if !output
|
|
# We want to warn the user but we only want to output once
|
|
ui.info I18n.t("vagrant.hosts.linux.nfs_prune")
|
|
output = true
|
|
end
|
|
|
|
logger.info("Invalid ID, pruning: #{id}")
|
|
nfs_cleanup(id)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def self.nfs_cleanup(id)
|
|
return if !File.exist?("/etc/exports")
|
|
|
|
user = Regexp.escape(Process.uid.to_s)
|
|
id = Regexp.escape(id.to_s)
|
|
|
|
# Use sed to just strip out the block of code which was inserted
|
|
# by Vagrant
|
|
system("sudo sed -r -e '/^# VAGRANT-BEGIN:( #{user})? #{id}/,/^# VAGRANT-END:( #{user})? #{id}/ d' -ibak /etc/exports")
|
|
end
|
|
|
|
def self.nfs_opts_setup(folders)
|
|
folders.each do |k, opts|
|
|
if !opts[:linux__nfs_options]
|
|
opts[:linux__nfs_options] ||= ["rw", "no_subtree_check", "all_squash"]
|
|
end
|
|
|
|
# Only automatically set anonuid/anongid if they weren't
|
|
# explicitly set by the user.
|
|
hasgid = false
|
|
hasuid = false
|
|
opts[:linux__nfs_options].each do |opt|
|
|
hasgid = !!(opt =~ /^anongid=/) if !hasgid
|
|
hasuid = !!(opt =~ /^anonuid=/) if !hasuid
|
|
end
|
|
|
|
opts[:linux__nfs_options] << "anonuid=#{opts[:map_uid]}" if !hasuid
|
|
opts[:linux__nfs_options] << "anongid=#{opts[:map_gid]}" if !hasgid
|
|
opts[:linux__nfs_options] << "fsid=#{opts[:uuid]}"
|
|
end
|
|
end
|
|
|
|
def self.nfs_running?(check_command)
|
|
system(check_command)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|