require "vagrant/util/retryable" require "vagrant/util/subprocess" require "vagrant-lxc/errors" require "vagrant-lxc/driver/cli" require "etc" require "tempfile" module Vagrant module LXC class Driver # This is raised if the container can't be found when initializing it with # a name. class ContainerNotFound < StandardError; end # Default root folder where container configs are stored DEFAULT_CONTAINERS_PATH = '/var/lib/lxc' attr_reader :container_name, :customizations def initialize(container_name, sudo_wrapper, cli = nil) @container_name = container_name @sudo_wrapper = sudo_wrapper @cli = cli || CLI.new(sudo_wrapper, container_name) @logger = Log4r::Logger.new("vagrant::provider::lxc::driver") @customizations = [] end def validate! raise ContainerNotFound if @container_name && ! @cli.list.include?(@container_name) end # Root folder where container configs are stored def containers_path @containers_path ||= @cli.support_config_command? ? @cli.config('lxc.lxcpath') : DEFAULT_CONTAINERS_PATH end def all_containers @cli.list end def base_path Pathname.new("#{containers_path}/#{@container_name}") end def rootfs_path config_entry = config_string.match(/^lxc\.rootfs\s+=\s+(.+)$/)[1] case config_entry when /^overlayfs:/ # Split on colon (:), ignoring any colon escaped by an escape character ( \ ) # Pays attention to when the escape character is itself escaped. fs_type, master_path, overlay_path = config_entry.split(/(?/dev/null | tail -n +2 | grep -q veth` $?.to_i == 0 end def remove_bridge(bridge_name) if ['lxcbr0', 'virbr0'].include? bridge_name @logger.info "Skipping removal of system bridge #{bridge_name}" return end return unless bridge_exists?(bridge_name) @logger.info "Removing bridge #{bridge_name}" @sudo_wrapper.run('ip', 'link', 'set', bridge_name, 'down') @sudo_wrapper.run('brctl', 'delbr', bridge_name) end def version @version ||= @cli.version end # TODO: This needs to be reviewed and specs needs to be written def compress_rootfs # TODO: Pass in tmpdir so we can clean up from outside target_path = "#{Dir.mktmpdir}/rootfs.tar.gz" @logger.info "Compressing '#{rootfs_path}' rootfs to #{target_path}" @sudo_wrapper.run('tar', '--numeric-owner', '-cvzf', target_path, '-C', rootfs_path.parent.to_s, "./#{rootfs_path.basename.to_s}") @logger.info "Changing rootfs tarball owner" user_details = Etc.getpwnam(Etc.getlogin) @sudo_wrapper.run('chown', "#{user_details.uid}:#{user_details.gid}", target_path) target_path end def state if @container_name @cli.state end end def prune_customizations # Use sed to just strip out the block of code which was inserted by Vagrant @logger.debug 'Prunning vagrant-lxc customizations' contents = config_string contents.gsub! /^# VAGRANT-BEGIN(.|\s)*# VAGRANT-END\n/, '' write_config(contents) end protected def write_customizations(customizations) customizations = customizations.map do |key, value| "lxc.#{key}=#{value}" end customizations.unshift '# VAGRANT-BEGIN' customizations << "# VAGRANT-END\n" contents = config_string contents << customizations.join("\n") write_config(contents) end def write_config(contents) Tempfile.new('lxc-config').tap do |file| file.chmod 0644 file.write contents file.close @sudo_wrapper.run 'cp', '-f', file.path, base_path.join('config').to_s @sudo_wrapper.run 'chown', 'root:root', base_path.join('config').to_s end end end end end