bin/vzmigrate.in - vzctl - Rivoreo Source Code Repositories

 #!/bin/sh
 # Copyright (C) 2000-2010, Parallels, Inc. All rights reserved.
 #
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU General Public License
 # as published by the Free Software Foundation; either version 2
 # of the License, or (at your option) any later version.
 #
 # This program is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU General Public License for more details.
 #
 # You should have received a copy of the GNU General Public License
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 #
 #
 # vzmigrate is used for container migration to another node


 ACT_SCRIPTS_SFX="start stop mount umount premount postumount"
 # blowfish is a fast block cipher, much faster then 3des
 SSH_OPTIONS="-c blowfish"
 SCP_OPTIONS=$SSH_OPTIONS
 RSYNC_OPTIONS="-aH --delete --numeric-ids"
 VZCTL=vzctl

 online=0
 verbose=0
 remove_area=1
 keep_dst=0
 debug=0
 confdir="@PKGCONFDIR@/conf"
 vzconf="@PKGCONFDIR@/vz.conf"
 tmpdir="/var/tmp"
 act_scripts=
 PLOOP=no

 # Errors:
 MIG_ERR_USAGE=1
 MIG_ERR_VPS_IS_STOPPED=2
 MIG_ERR_CANT_CONNECT=4
 MIG_ERR_COPY=6
 MIG_ERR_START_VPS=7
 MIG_ERR_STOP_SOURCE=8
 MIG_ERR_EXISTS=9
 MIG_ERR_NOEXIST=10
 MIG_ERR_IP_IN_USE=12
 MIG_ERR_QUOTA=13
 MIG_ERR_CHECKPOINT=$MIG_ERR_STOP_SOURCE
 MIG_ERR_MOUNT_VPS=$MIG_ERR_START_VPS
 MIG_ERR_RESTORE_VPS=$MIG_ERR_START_VPS
 MIG_ERR_OVZ_NOT_RUNNING=14
 MIG_ERR_APPLY_CONFIG=15
 MIG_ERR_UNSUP_PLOOP=16

 # For local vzctl to work, make sure @SBINDIR@ is in $PATH
 if ! echo ":${PATH}:" | fgrep -q ':@SBINDIR@:'; then
 	PATH="@SBINDIR@:$PATH"
 fi

 usage() {
 	cat >&2 <<EOF
 This program is used for container migration to another node.
 Usage:
 vzmigrate [-r yes|no] [--ssh=<options>] [--keep-dst] [--online] [-v]
 	destination_address <CTID>
 Options:
 -r, --remove-area yes|no
 	Whether to remove container on source host after successful migration.
 --ssh=<ssh options>
 	Additional options that will be passed to ssh while establishing
 	connection to destination host. Please be careful with options
 	passed, DO NOT pass destination hostname.
 --rsync=<rsync options>
 	Additional options that will be passed to rsync.
 --keep-dst
 	Do not clean synced destination container private area in case of some
 	error. It makes sense to use this option on big container migration to
 	avoid re-syncing container private area in case some error
 	(on container stop for example) occurs during first migration attempt.
 --online
 	Perform online (zero-downtime) migration: during the migration the
 	container freezes for some time and after the migration it
 	keeps working as though nothing has happened.
 -v
 	Verbose mode. Causes vzmigrate to print debugging messages about
 	its progress (including some time statistics). Multiple -v options
 	increase the verbosity. The maximum is 3.
 Examples:
 	Online migration of CT #101 to foo.com:
 		vzmigrate --online foo.com 101
 	Migration of CT #102 to foo.com with downtime:
 		vzmigrate foo.com 102
 Notes:
 	This program uses ssh as a transport layer. You need to put ssh
 	public key to destination node and be able to connect without
 	entering a password.
 EOF
 	exit $MIG_ERR_USAGE
 }

 # Logs message
 # There are 3 types of messages:
 # 0 - error messages (print to stderr)
 # 1 - normal messages (print to stdout)
 # 2 - debug messages (print to stdout if in verbose mode)
 log () {
 	if [ $1 -eq 0 ]; then
 		shift
 		echo "Error: $*" >&2
 	elif [ $1 -eq 1 ]; then
 		shift
 		echo "$*"
 	elif [ $verbose -gt 0 ]; then
 		shift
 		echo "   $@"
 	fi
 }

 # Executes command and returns result of execution
 # There are 2 types of execution:
 # 1 - normal execution (all output will be printed)
 # 2 - debug execution (output will be printed if verbose mode is set,
 #     in other case stdout and stderr redirected to /dev/null)
 logexec () {
 	if [ $1 -eq 1 -o $verbose -gt 0 ]; then
 		shift
 		$@
 	else
 		shift
 		$@ >/dev/null 2>&1
 	fi
 }

 undo_conf () {
 	$SSH "root@$host" "$VZCTL set $VEID --name '' --save > /dev/null"
 	$SSH "root@$host" "rm -f $vpsconf"
 }

 undo_act_scripts () {
 	if [ -n "$act_scripts" ] ; then
 		$SSH "root@$host" "rm -f $act_scripts"
 	fi
 	undo_conf
 }

 undo_private () {
 	if [ $keep_dst -eq 0 ]; then
 		$SSH "root@$host" "rm -rf $VE_PRIVATE"
 	fi
 	undo_act_scripts
 }

 undo_root () {
 	$SSH "root@$host" "rm -rf $VE_ROOT"
 	undo_private
 }

 undo_quota_init () {
 	[ "${DISK_QUOTA}" = 'no' ] || $SSH "root@$host" "vzquota drop $VEID"
 	undo_root
 }

 undo_quota_on () {
 	[ "${DISK_QUOTA}" = 'no' ] || $SSH "root@$host" "vzquota off $VEID"
 	undo_quota_init
 }

 undo_sync () {
 	# Root will be destroyed in undo_root
 	undo_quota_on
 }

 undo_suspend () {
 	logexec 2 $VZCTL chkpnt $VEID --resume
 	undo_sync
 }

 undo_dump () {
 	if [ $debug -eq 0 ]; then
 		rm -f "$VE_DUMPFILE"
 	fi
 	undo_suspend
 }

 undo_copy_dump () {
 	$SSH "root@$host" "rm -f $VE_DUMPFILE"
 	undo_suspend
 }

 undo_stop () {
 	if [ "$state" = "running" ]; then
 		$VZCTL start $VEID
 	elif [ "$mounted" = "mounted" ]; then
 		$VZCTL mount $VEID
 	fi
 	undo_sync
 }

 undo_source_stage() {
 	if [ $online -eq 1 ]; then
 		undo_copy_dump
 	else
 		undo_stop
 	fi
 }

 undo_quota_dump () {
 	rm -f "$VE_QUOTADUMP"
 	undo_source_stage
 }

 undo_copy_quota () {
 	$SSH "root@$host" "rm -f $VE_QUOTADUMP"
 	undo_quota_dump
 }

 undo_undump () {
 	logexec 2 $SSH root@$host $VZCTL restore $VEID --kill
 	undo_copy_quota
 }

 get_status() {
 	exist=$3
 	mounted=$4
 	state=$5
 }

 get_time () {
 	awk -v t2=$2 -v t1=$1 'BEGIN{print t2-t1}'
 }

 get_ploop_info() {
 	local top img

 	PLOOP_DEV=$(awk '$2=="'$VE_ROOT'" {print $1}' /proc/mounts | \
 		sed -e 's|^/dev/||' -e 's|p1$||')
 	top=$(cat /sys/block/${dev}/pstate/top)
 	img=$(cat /sys/block/${dev}/pdelta/${top}/image)
 	TOP_DELTA=$(basename $img)
 }

 if [ $# -lt 2 ]; then
 	usage
 fi

 while [ ! -z "$1" ]; do
 	case "$1" in
 	--online)
 		online=1
 		;;
 	-v)
 		verbose=$((verbose+1)) # can just be 'let verbose++' in bash
 		;;
 	-vv)
 		verbose=$((verbose+2))
 		;;
 	-vvv)
 		verbose=$((verbose+3))
 		;;
 	-vvvv)
 		verbose=$((verbose+4))
 		;;
 	--remove-area|-r)
 		shift
 		if [ "$1" = "yes" ]; then
 			remove_area=1
 		elif [ "$1" = "no" ]; then
 			remove_area=0
 		else
 			usage
 		fi
 		;;
 	--keep-dst)
 		keep_dst=1
 		;;
 	--ssh=*)
 		SSH_OPTIONS="$SSH_OPTIONS $(echo $1 | cut -c7-)"
 		SCP_OPTIONS="`echo $SSH_OPTIONS | sed 's/-p/-P/1'`"
 		;;
 	--rsync=*)
 		RSYNC_OPTIONS="$RSYNC_OPTIONS $(echo $1 | cut -c9-)"
 		;;
 	*)
 		break
 		;;
 	esac
 	shift
 done

 if [ $verbose -gt 1 ]; then
 	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
 	VZCTL="$VZCTL --verbose"
 fi
 if [ $verbose -gt 2 ]; then
 	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
 	VZCTL="$VZCTL --verbose"
 fi
 if [ $verbose -gt 3 ]; then
 	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
 	VZCTL="$VZCTL --verbose"
 	SSH_OPTIONS="$SSH_OPTIONS -v"
 	SCP_OPTIONS="$SCP_OPTIONS -v"
 fi

 RSYNC="rsync $RSYNC_OPTIONS"
 SSH="ssh $SSH_OPTIONS"
 SCP="scp $SCP_OPTIONS"
 export RSYNC_RSH="$SSH"

 host=$1
 shift
 VEID=$1
 shift

 if [ -z "$host" -o -z "$VEID" -o $# -ne 0 ]; then
 	usage
 fi

 # Support CT names as well
 if echo $VEID | egrep -qv '^[[:digit:]]+$'; then
 	VEID=$(vzlist -o ctid -H $VEID | tr -d ' ')
 	if [ -z "$VEID" ]; then
 		# Error message is printed by vzlist to stderr
 		exit $MIG_ERR_NOEXIST
 	fi
 fi

 vpsconf="$confdir/$VEID.conf"

 if [ ! -r "$vzconf" ]; then
 	log 0 "Can't read global config file $vzconf"
 	exit $MIG_ERR_NOEXIST
 fi

 get_status $($VZCTL status $VEID)
 if [ "$exist" = "deleted" ]; then
 	log 0 "CT #$VEID doesn't exist"
 	exit $MIG_ERR_NOEXIST
 fi

 if [ $online -eq 1 ]; then
 	log 1 "Starting online migration of CT $VEID to $host"
 else
 	log 1 "Starting migration of CT $VEID to $host"
 fi

 # Try to connect to destination
 if ! logexec 2 $SSH -o BatchMode=yes root@$host /bin/true; then
 	log 0 "Can't connect to destination address using public key"
 	log 0 "Please put your public key to destination node"
 	exit $MIG_ERR_CANT_CONNECT
 fi

 # Check if OpenVZ is running
 if ! logexec 2 $SSH -o BatchMode=yes root@$host /etc/init.d/vz status ; then
 	log 0 "OpenVZ is not running on the target machine"
 	log 0 "Can't continue migration"
 	exit $MIG_ERR_OVZ_NOT_RUNNING
 fi

 # Check if CPT modules are loaded for online migration
 if [ $online -eq 1 ]; then
 	if [ ! -f /proc/cpt ]; then
 		log 0 "vzcpt module is not loaded on the source node"
 		log 0 "Can't continue online migration"
 		exit $MIG_ERR_OVZ_NOT_RUNNING
 	fi
 	if ! logexec 2 $SSH -o BatchMode=yes root@$host "test -f /proc/rst";
 	then
 		log 0 "vzrst module is not loaded on the destination node"
 		log 0 "Can't continue online migration"
 		exit $MIG_ERR_OVZ_NOT_RUNNING
 	fi
 fi

 dst_exist=$($SSH "root@$host" "$VZCTL status $VEID" | awk '{print $3}')
 if [ "$dst_exist" = "exist" ]; then
 	log 0 "CT #$VEID already exists on destination node"
 	exit $MIG_ERR_EXISTS
 fi

 if [ $online -eq 1 -a "$state" != "running" ]; then
 	log 0 "Can't perform online migration of a stopped container"
 	exit $MIG_ERR_VPS_IS_STOPPED
 fi

 log 2 "Loading $vzconf and $vpsconf files"

 . "$vzconf"
 . "$vpsconf"
 VE_DUMPFILE="$tmpdir/dump.$VEID"
 VE_QUOTADUMP="$tmpdir/quotadump.$VEID"

 DDXML=$VE_PRIVATE/root.hdd/DiskDescriptor.xml
 if [ -f $DDXML ]; then
 	PLOOP=yes
 	# Disable vzquota operations for ploop CT
 	DISK_QUOTA=no
 fi

 if [ "$PLOOP" = "yes" ]; then
 	log 2 "Check if ploop is supported on destination node"
 	if ! logexec 2 $SSH "root@$host" ploop getdev ; then
 		log 0 "Destination node don't support ploop, can't migrate"
 		exit $MIG_ERR_UNSUP_PLOOP
 	fi
 fi

 log 2 "Check IPs on destination node: $IP_ADDRESS"
 for IP in $IP_ADDRESS; do
 	if [ $($SSH "root@$host" "grep -c \" $IP \" /proc/vz/veip") -gt 0 ];
 	then
 		log 0 "IP address $IP already in use on destination node"
 		exit $MIG_ERR_IP_IN_USE
 	fi
 done

 log 1 "Preparing remote node"

 log 2 "Copying config file"
 if ! logexec 2 $SCP $vpsconf root@$host:$vpsconf ; then
 	log 0 "Failed to copy config file"
 	exit $MIG_ERR_COPY
 fi

 logexec 2 $SSH root@$host $VZCTL set $VEID --applyconfig_map name --save
 # vzctl return code 20 or 21 in case of unrecognized option
 if [ $? != 20 ] && [ $? != 21 ] && [ $? != 0 ]; then
 	log 0 "Failed to apply config on destination node"
 	undo_conf
 	exit $MIG_ERR_APPLY_CONFIG
 fi

 for sfx in $ACT_SCRIPTS_SFX; do
 	file="$confdir/$VEID.$sfx"
 	if [ -f "$file" ]; then
 		act_scripts="$act_scripts $file"
 	fi
 done
 if [ -n "$act_scripts" ]; then
 	log 2 "Copying action scripts"
 	if ! logexec 2 $SCP $act_scripts root@$host:$confdir ; then
 		log 0 "Failed to copy action scripts"
 		undo_conf
 		exit $MIG_ERR_COPY
 	fi
 fi

 log 2 "Creating remote container root dir"
 if ! $SSH "root@$host" "mkdir -p $VE_ROOT"; then
 	log 0 "Failed to make container root directory"
 	undo_act_scripts
 	exit $MIG_ERR_COPY
 fi

 log 2 "Creating remote container private dir"
 if ! $SSH "root@$host" "mkdir -p $VE_PRIVATE"; then
 	log 0 "Failed to make container private area directory"
 	undo_private
 	exit $MIG_ERR_COPY
 fi

 if [ "$PLOOP" != "yes" ]; then
 	RSYNC1_OPTIONS="--sparse"
 else
 	get_ploop_info
 	RSYNC1_OPTIONS="--exclude $TOP_DELTA"
 fi

 if [ "${DISK_QUOTA}" != "no" ]; then
 	log 1 "Initializing remote quota"

 	log 2 "Quota init"
 	if ! $SSH "root@$host" "$VZCTL quotainit $VEID"; then
 		log 0 "Failed to initialize quota"
 		undo_root
 		exit $MIG_ERR_QUOTA
 	fi

 	log 2 "Turning remote quota on"
 	if ! $SSH "root@$host" "$VZCTL quotaon $VEID"; then
 		log 0 "Failed to turn quota on"
 		undo_quota_init
 		exit $MIG_ERR_QUOTA
 	fi
 fi

 log 1 "Syncing private"
 $RSYNC $RSYNC1_OPTIONS "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}"
 # Ignore rsync error 24 "Partial transfer due to vanished source files"
 if [ $? != 24 ] && [ $? != 0 ]; then
 	log 0 "Failed to sync container private areas"
 	undo_quota_on
 	exit $MIG_ERR_COPY
 fi

 if [ $online -eq 1 ]; then
 	log 1 "Live migrating container..."

 	time_suspend=$(date +%s.%N)
 	if [ "$PLOOP" != "yes" ]; then
 		log 2 "Suspending container"
 		if ! logexec 2 $VZCTL chkpnt $VEID --suspend ; then
 			log 0 "Failed to suspend container"
 			undo_sync
 			exit $MIG_ERR_CHECKPOINT
 		fi
 	else
 		log 2 "Migrating top ploop delta with ploop copy"
 		if ! ploop copy -s $PLOOP_DEV -F "$VZCTL chkpnt $VEID --suspend" | \
 				$SSH "root@$host" ploop copy -d $VE_PRIVATE/root.hdd/$TOP_DELTA; then
 			log 0 "Failed to migrate top ploop delta"
 			undo_suspend
 			exit $MIG_ERR_COPY
 		fi
 	fi

 	log 2 "Dumping container"
 	if ! logexec 2 $VZCTL chkpnt $VEID --dump --dumpfile $VE_DUMPFILE ; then
 		log 0 "Failed to dump container"
 		undo_suspend
 		exit $MIG_ERR_CHECKPOINT
 	fi

 	log 2 "Copying dumpfile"
 	time_copy_dump=$(date +%s.%N)
 	if ! logexec 2 $SCP $VE_DUMPFILE root@$host:$VE_DUMPFILE ; then
 		log 0 "Failed to copy dump"
 		undo_dump
 		exit $MIG_ERR_COPY
 	fi
 else
 	if [ "$state" = "running" ]; then
 		log 1 "Stopping container"
 		if ! logexec 2 $VZCTL stop $VEID ; then
 			log 0 "Failed to stop container"
 			undo_sync
 			exit $MIG_ERR_STOP_SOURCE
 		fi
 	elif [ "$mounted" = "mounted" ]; then
 		log 1 "Unmounting container"
 		if ! logexec 2 $VZCTL umount $VEID ; then
 			log 0 "Failed to umount container"
 			undo_sync
 			exit $MIG_ERR_STOP_SOURCE
 		fi
 	fi
 fi

 if [ "$state" = "running" ]; then
 	time_rsync2=$(date +%s.%N)
 	if [ "$PLOOP" != "yes" ]; then
 		log 2 "Syncing private (2nd pass)"
 		if ! $RSYNC "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}"; then
 			log 0 "Failed to sync container private areas"
 			undo_source_stage
 			exit $MIG_ERR_COPY
 		fi
 	elif [ $online -eq 1 ]; then
 		# Do nothing! We have already used ploop copy
 		true
 	else
 		log 2 "Copying top delta"
 		if ! $SCP "$VE_PRIVATE/root.hdd/$TOP_DELTA" \
 				"root@$host:$VE_PRIVATE/root.hdd/$TOP_DELTA"; then
 			log 0 "Failed to scp top ploop delta"
 			undo_source_stage
 			exit $MIG_ERR_COPY
 		fi
 	fi
 fi

 if [ "${DISK_QUOTA}" != "no" ]; then
 	log 1 "Syncing 2nd level quota"

 	log 2 "Dumping 2nd level quota"
 	time_quota=$(date +%s.%N)
 	if ! vzdqdump $VEID -U -G -T > "$VE_QUOTADUMP"; then
 		log 0 "Failed to dump 2nd level quota"
 		undo_quota_dump
 		exit $MIG_ERR_QUOTA
 	fi

 	log 2 "Copying 2nd level quota"
 	if ! logexec 2 $SCP $VE_QUOTADUMP root@$host:$VE_QUOTADUMP ; then
 		log 0 "Failed to copy 2nd level quota dump"
 		undo_quota_dump
 		exit $MIG_ERR_COPY
 	fi

 	log 2 "Load 2nd level quota"
 	if ! $SSH "root@$host" "(vzdqload $VEID -U -G -T < $VE_QUOTADUMP &&
 			vzquota reload2 $VEID)"; then
 		log 0 "Failed to load 2nd level quota"
 		undo_copy_quota
 		exit $MIG_ERR_QUOTA
 	fi
 fi

 if [ $online -eq 1 ]; then
 	log 2 "Undumping container"
 	time_undump=$(date +%s.%N)
 	if ! logexec 2 $SSH root@$host $VZCTL restore $VEID --undump \
 			--dumpfile $VE_DUMPFILE --skip_arpdetect ; then
 		log 0 "Failed to undump container"
 		undo_copy_quota
 		exit $MIG_ERR_RESTORE_VPS
 	fi

 	log 2 "Resuming container"
 	if ! logexec 2 $SSH root@$host $VZCTL restore $VEID --resume ; then
 		log 0 "Failed to resume container"
 		undo_undump
 		exit $MIG_ERR_RESTORE_VPS
 	fi
 	time_finish=$(date +%s.%N)
 	log 2 "Times:"
 	log 2 "  Suspend + Dump: " $(get_time $time_suspend $time_copy_dump)
 	log 2 "  Copy dump file: " $(get_time $time_copy_dump $time_rsync2)
 	log 2 "  Second rsync:   " $(get_time $time_rsync2 $time_quota)
 	log 2 "  2nd level quota:" $(get_time $time_quota $time_undump)
 	log 2 "  Undump + Resume:" $(get_time $time_undump $time_finish)
 	log 2 "Total time: " $(get_time $time_suspend $time_finish)

 	log 1 "Cleanup"

 	log 2 "Killing container"
 	logexec 2 $VZCTL chkpnt $VEID --kill
 	logexec 2 $VZCTL umount $VEID

 	log 2 "Removing dumpfiles"
 	rm -f "$VE_DUMPFILE"
 	$SSH "root@$host" "rm -f $VE_DUMPFILE"
 else
 	if [ "$state" = "running" ]; then
 		log 1 "Starting container"
 		if ! logexec 2 $SSH root@$host $VZCTL start $VEID ; then
 			log 0 "Failed to start container"
 			undo_copy_quota
 			exit $MIG_ERR_START_VPS
 		fi
 	elif [ "$mounted" = "mounted" ]; then
 		log 1 "Mounting container"
 		if ! logexec 2 $SSH root@$host $VZCTL mount $VEID ; then
 			log 0 "Failed to mount container"
 			undo_copy_quota
 			exit $MIG_ERR_MOUNT_VPS
 		fi
 	elif [ "${DISK_QUOTA}" != "no" ]; then
 		log 1 "Turning quota off"
 		if ! logexec 2 $SSH root@$host vzquota off $VEID ; then
 			log 0 "failed to turn quota off"
 			undo_copy_quota
 			exit $MIG_ERR_QUOTA
 		fi
 	fi

 	log 1 "Cleanup"
 fi

 if [ $remove_area -eq 1 ]; then
 	log 2 "Destroying container"
 	logexec 2 $VZCTL destroy $VEID
 else
 	# Move config as veid.migrated to allow backward migration
 	mv -f $vpsconf $vpsconf.migrated
 fi
	#!/bin/sh
	# Copyright (C) 2000-2010, Parallels, Inc. All rights reserved.
	#
	# This program is free software; you can redistribute it and/or
	# modify it under the terms of the GNU General Public License
	# as published by the Free Software Foundation; either version 2
	# of the License, or (at your option) any later version.
	#
	# This program is distributed in the hope that it will be useful,
	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	# GNU General Public License for more details.
	#
	# You should have received a copy of the GNU General Public License
	# along with this program; if not, write to the Free Software
	# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	#
	#
	# vzmigrate is used for container migration to another node


	ACT_SCRIPTS_SFX="start stop mount umount premount postumount"
	# blowfish is a fast block cipher, much faster then 3des
	SSH_OPTIONS="-c blowfish"
	SCP_OPTIONS=$SSH_OPTIONS
	RSYNC_OPTIONS="-aH --delete --numeric-ids"
	VZCTL=vzctl

	online=0
	verbose=0
	remove_area=1
	keep_dst=0
	debug=0
	confdir="@PKGCONFDIR@/conf"
	vzconf="@PKGCONFDIR@/vz.conf"
	tmpdir="/var/tmp"
	act_scripts=
	PLOOP=no

	# Errors:
	MIG_ERR_USAGE=1
	MIG_ERR_VPS_IS_STOPPED=2
	MIG_ERR_CANT_CONNECT=4
	MIG_ERR_COPY=6
	MIG_ERR_START_VPS=7
	MIG_ERR_STOP_SOURCE=8
	MIG_ERR_EXISTS=9
	MIG_ERR_NOEXIST=10
	MIG_ERR_IP_IN_USE=12
	MIG_ERR_QUOTA=13
	MIG_ERR_CHECKPOINT=$MIG_ERR_STOP_SOURCE
	MIG_ERR_MOUNT_VPS=$MIG_ERR_START_VPS
	MIG_ERR_RESTORE_VPS=$MIG_ERR_START_VPS
	MIG_ERR_OVZ_NOT_RUNNING=14
	MIG_ERR_APPLY_CONFIG=15
	MIG_ERR_UNSUP_PLOOP=16

	# For local vzctl to work, make sure @SBINDIR@ is in $PATH
	if ! echo ":${PATH}:" \| fgrep -q ':@SBINDIR@:'; then
	PATH="@SBINDIR@:$PATH"
	fi

	usage() {
	cat >&2 <<EOF
	This program is used for container migration to another node.
	Usage:
	vzmigrate [-r yes\|no] [--ssh=<options>] [--keep-dst] [--online] [-v]
	destination_address <CTID>
	Options:
	-r, --remove-area yes\|no
	Whether to remove container on source host after successful migration.
	--ssh=<ssh options>
	Additional options that will be passed to ssh while establishing
	connection to destination host. Please be careful with options
	passed, DO NOT pass destination hostname.
	--rsync=<rsync options>
	Additional options that will be passed to rsync.
	--keep-dst
	Do not clean synced destination container private area in case of some
	error. It makes sense to use this option on big container migration to
	avoid re-syncing container private area in case some error
	(on container stop for example) occurs during first migration attempt.
	--online
	Perform online (zero-downtime) migration: during the migration the
	container freezes for some time and after the migration it
	keeps working as though nothing has happened.
	-v
	Verbose mode. Causes vzmigrate to print debugging messages about
	its progress (including some time statistics). Multiple -v options
	increase the verbosity. The maximum is 3.
	Examples:
	Online migration of CT #101 to foo.com:
	vzmigrate --online foo.com 101
	Migration of CT #102 to foo.com with downtime:
	vzmigrate foo.com 102
	Notes:
	This program uses ssh as a transport layer. You need to put ssh
	public key to destination node and be able to connect without
	entering a password.
	EOF
	exit $MIG_ERR_USAGE
	}

	# Logs message
	# There are 3 types of messages:
	# 0 - error messages (print to stderr)
	# 1 - normal messages (print to stdout)
	# 2 - debug messages (print to stdout if in verbose mode)
	log () {
	if [ $1 -eq 0 ]; then
	shift
	echo "Error: $*" >&2
	elif [ $1 -eq 1 ]; then
	shift
	echo "$*"
	elif [ $verbose -gt 0 ]; then
	shift
	echo " $@"
	fi
	}

	# Executes command and returns result of execution
	# There are 2 types of execution:
	# 1 - normal execution (all output will be printed)
	# 2 - debug execution (output will be printed if verbose mode is set,
	# in other case stdout and stderr redirected to /dev/null)
	logexec () {
	if [ $1 -eq 1 -o $verbose -gt 0 ]; then
	shift
	$@
	else
	shift
	$@ >/dev/null 2>&1
	fi
	}

	undo_conf () {
	$SSH "root@$host" "$VZCTL set $VEID --name '' --save > /dev/null"
	$SSH "root@$host" "rm -f $vpsconf"
	}

	undo_act_scripts () {
	if [ -n "$act_scripts" ] ; then
	$SSH "root@$host" "rm -f $act_scripts"
	fi
	undo_conf
	}

	undo_private () {
	if [ $keep_dst -eq 0 ]; then
	$SSH "root@$host" "rm -rf $VE_PRIVATE"
	fi
	undo_act_scripts
	}

	undo_root () {
	$SSH "root@$host" "rm -rf $VE_ROOT"
	undo_private
	}

	undo_quota_init () {
	[ "${DISK_QUOTA}" = 'no' ] \|\| $SSH "root@$host" "vzquota drop $VEID"
	undo_root
	}

	undo_quota_on () {
	[ "${DISK_QUOTA}" = 'no' ] \|\| $SSH "root@$host" "vzquota off $VEID"
	undo_quota_init
	}

	undo_sync () {
	# Root will be destroyed in undo_root
	undo_quota_on
	}

	undo_suspend () {
	logexec 2 $VZCTL chkpnt $VEID --resume
	undo_sync
	}

	undo_dump () {
	if [ $debug -eq 0 ]; then
	rm -f "$VE_DUMPFILE"
	fi
	undo_suspend
	}

	undo_copy_dump () {
	$SSH "root@$host" "rm -f $VE_DUMPFILE"
	undo_suspend
	}

	undo_stop () {
	if [ "$state" = "running" ]; then
	$VZCTL start $VEID
	elif [ "$mounted" = "mounted" ]; then
	$VZCTL mount $VEID
	fi
	undo_sync
	}

	undo_source_stage() {
	if [ $online -eq 1 ]; then
	undo_copy_dump
	else
	undo_stop
	fi
	}

	undo_quota_dump () {
	rm -f "$VE_QUOTADUMP"
	undo_source_stage
	}

	undo_copy_quota () {
	$SSH "root@$host" "rm -f $VE_QUOTADUMP"
	undo_quota_dump
	}

	undo_undump () {
	logexec 2 $SSH root@$host $VZCTL restore $VEID --kill
	undo_copy_quota
	}

	get_status() {
	exist=$3
	mounted=$4
	state=$5
	}

	get_time () {
	awk -v t2=$2 -v t1=$1 'BEGIN{print t2-t1}'
	}

	get_ploop_info() {
	local top img

	PLOOP_DEV=$(awk '$2=="'$VE_ROOT'" {print $1}' /proc/mounts \| \
	sed -e 's\|^/dev/\|\|' -e 's\|p1$\|\|')
	top=$(cat /sys/block/${dev}/pstate/top)
	img=$(cat /sys/block/${dev}/pdelta/${top}/image)
	TOP_DELTA=$(basename $img)
	}

	if [ $# -lt 2 ]; then
	usage
	fi

	while [ ! -z "$1" ]; do
	case "$1" in
	--online)
	online=1
	;;
	-v)
	verbose=$((verbose+1)) # can just be 'let verbose++' in bash
	;;
	-vv)
	verbose=$((verbose+2))
	;;
	-vvv)
	verbose=$((verbose+3))
	;;
	-vvvv)
	verbose=$((verbose+4))
	;;
	--remove-area\|-r)
	shift
	if [ "$1" = "yes" ]; then
	remove_area=1
	elif [ "$1" = "no" ]; then
	remove_area=0
	else
	usage
	fi
	;;
	--keep-dst)
	keep_dst=1
	;;
	--ssh=*)
	SSH_OPTIONS="$SSH_OPTIONS $(echo $1 \| cut -c7-)"
	SCP_OPTIONS="`echo $SSH_OPTIONS \| sed 's/-p/-P/1'`"
	;;
	--rsync=*)
	RSYNC_OPTIONS="$RSYNC_OPTIONS $(echo $1 \| cut -c9-)"
	;;
	*)
	break
	;;
	esac
	shift
	done

	if [ $verbose -gt 1 ]; then
	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
	VZCTL="$VZCTL --verbose"
	fi
	if [ $verbose -gt 2 ]; then
	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
	VZCTL="$VZCTL --verbose"
	fi
	if [ $verbose -gt 3 ]; then
	RSYNC_OPTIONS="$RSYNC_OPTIONS -v"
	VZCTL="$VZCTL --verbose"
	SSH_OPTIONS="$SSH_OPTIONS -v"
	SCP_OPTIONS="$SCP_OPTIONS -v"
	fi

	RSYNC="rsync $RSYNC_OPTIONS"
	SSH="ssh $SSH_OPTIONS"
	SCP="scp $SCP_OPTIONS"
	export RSYNC_RSH="$SSH"

	host=$1
	shift
	VEID=$1
	shift

	if [ -z "$host" -o -z "$VEID" -o $# -ne 0 ]; then
	usage
	fi

	# Support CT names as well
	if echo $VEID \| egrep -qv '^[[:digit:]]+$'; then
	VEID=$(vzlist -o ctid -H $VEID \| tr -d ' ')
	if [ -z "$VEID" ]; then
	# Error message is printed by vzlist to stderr
	exit $MIG_ERR_NOEXIST
	fi
	fi

	vpsconf="$confdir/$VEID.conf"

	if [ ! -r "$vzconf" ]; then
	log 0 "Can't read global config file $vzconf"
	exit $MIG_ERR_NOEXIST
	fi

	get_status $($VZCTL status $VEID)
	if [ "$exist" = "deleted" ]; then
	log 0 "CT #$VEID doesn't exist"
	exit $MIG_ERR_NOEXIST
	fi

	if [ $online -eq 1 ]; then
	log 1 "Starting online migration of CT $VEID to $host"
	else
	log 1 "Starting migration of CT $VEID to $host"
	fi

	# Try to connect to destination
	if ! logexec 2 $SSH -o BatchMode=yes root@$host /bin/true; then
	log 0 "Can't connect to destination address using public key"
	log 0 "Please put your public key to destination node"
	exit $MIG_ERR_CANT_CONNECT
	fi

	# Check if OpenVZ is running
	if ! logexec 2 $SSH -o BatchMode=yes root@$host /etc/init.d/vz status ; then
	log 0 "OpenVZ is not running on the target machine"
	log 0 "Can't continue migration"
	exit $MIG_ERR_OVZ_NOT_RUNNING
	fi

	# Check if CPT modules are loaded for online migration
	if [ $online -eq 1 ]; then
	if [ ! -f /proc/cpt ]; then
	log 0 "vzcpt module is not loaded on the source node"
	log 0 "Can't continue online migration"
	exit $MIG_ERR_OVZ_NOT_RUNNING
	fi
	if ! logexec 2 $SSH -o BatchMode=yes root@$host "test -f /proc/rst";
	then
	log 0 "vzrst module is not loaded on the destination node"
	log 0 "Can't continue online migration"
	exit $MIG_ERR_OVZ_NOT_RUNNING
	fi
	fi

	dst_exist=$($SSH "root@$host" "$VZCTL status $VEID" \| awk '{print $3}')
	if [ "$dst_exist" = "exist" ]; then
	log 0 "CT #$VEID already exists on destination node"
	exit $MIG_ERR_EXISTS
	fi

	if [ $online -eq 1 -a "$state" != "running" ]; then
	log 0 "Can't perform online migration of a stopped container"
	exit $MIG_ERR_VPS_IS_STOPPED
	fi

	log 2 "Loading $vzconf and $vpsconf files"

	. "$vzconf"
	. "$vpsconf"
	VE_DUMPFILE="$tmpdir/dump.$VEID"
	VE_QUOTADUMP="$tmpdir/quotadump.$VEID"

	DDXML=$VE_PRIVATE/root.hdd/DiskDescriptor.xml
	if [ -f $DDXML ]; then
	PLOOP=yes
	# Disable vzquota operations for ploop CT
	DISK_QUOTA=no
	fi

	if [ "$PLOOP" = "yes" ]; then
	log 2 "Check if ploop is supported on destination node"
	if ! logexec 2 $SSH "root@$host" ploop getdev ; then
	log 0 "Destination node don't support ploop, can't migrate"
	exit $MIG_ERR_UNSUP_PLOOP
	fi
	fi

	log 2 "Check IPs on destination node: $IP_ADDRESS"
	for IP in $IP_ADDRESS; do
	if [ $($SSH "root@$host" "grep -c \" $IP \" /proc/vz/veip") -gt 0 ];
	then
	log 0 "IP address $IP already in use on destination node"
	exit $MIG_ERR_IP_IN_USE
	fi
	done

	log 1 "Preparing remote node"

	log 2 "Copying config file"
	if ! logexec 2 $SCP $vpsconf root@$host:$vpsconf ; then
	log 0 "Failed to copy config file"
	exit $MIG_ERR_COPY
	fi

	logexec 2 $SSH root@$host $VZCTL set $VEID --applyconfig_map name --save
	# vzctl return code 20 or 21 in case of unrecognized option
	if [ $? != 20 ] && [ $? != 21 ] && [ $? != 0 ]; then
	log 0 "Failed to apply config on destination node"
	undo_conf
	exit $MIG_ERR_APPLY_CONFIG
	fi

	for sfx in $ACT_SCRIPTS_SFX; do
	file="$confdir/$VEID.$sfx"
	if [ -f "$file" ]; then
	act_scripts="$act_scripts $file"
	fi
	done
	if [ -n "$act_scripts" ]; then
	log 2 "Copying action scripts"
	if ! logexec 2 $SCP $act_scripts root@$host:$confdir ; then
	log 0 "Failed to copy action scripts"
	undo_conf
	exit $MIG_ERR_COPY
	fi
	fi

	log 2 "Creating remote container root dir"
	if ! $SSH "root@$host" "mkdir -p $VE_ROOT"; then
	log 0 "Failed to make container root directory"
	undo_act_scripts
	exit $MIG_ERR_COPY
	fi

	log 2 "Creating remote container private dir"
	if ! $SSH "root@$host" "mkdir -p $VE_PRIVATE"; then
	log 0 "Failed to make container private area directory"
	undo_private
	exit $MIG_ERR_COPY
	fi

	if [ "$PLOOP" != "yes" ]; then
	RSYNC1_OPTIONS="--sparse"
	else
	get_ploop_info
	RSYNC1_OPTIONS="--exclude $TOP_DELTA"
	fi

	if [ "${DISK_QUOTA}" != "no" ]; then
	log 1 "Initializing remote quota"

	log 2 "Quota init"
	if ! $SSH "root@$host" "$VZCTL quotainit $VEID"; then
	log 0 "Failed to initialize quota"
	undo_root
	exit $MIG_ERR_QUOTA
	fi

	log 2 "Turning remote quota on"
	if ! $SSH "root@$host" "$VZCTL quotaon $VEID"; then
	log 0 "Failed to turn quota on"
	undo_quota_init
	exit $MIG_ERR_QUOTA
	fi
	fi

	log 1 "Syncing private"
	$RSYNC $RSYNC1_OPTIONS "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}"
	# Ignore rsync error 24 "Partial transfer due to vanished source files"
	if [ $? != 24 ] && [ $? != 0 ]; then
	log 0 "Failed to sync container private areas"
	undo_quota_on
	exit $MIG_ERR_COPY
	fi

	if [ $online -eq 1 ]; then
	log 1 "Live migrating container..."

	time_suspend=$(date +%s.%N)
	if [ "$PLOOP" != "yes" ]; then
	log 2 "Suspending container"
	if ! logexec 2 $VZCTL chkpnt $VEID --suspend ; then
	log 0 "Failed to suspend container"
	undo_sync
	exit $MIG_ERR_CHECKPOINT
	fi
	else
	log 2 "Migrating top ploop delta with ploop copy"
	if ! ploop copy -s $PLOOP_DEV -F "$VZCTL chkpnt $VEID --suspend" \| \
	$SSH "root@$host" ploop copy -d $VE_PRIVATE/root.hdd/$TOP_DELTA; then
	log 0 "Failed to migrate top ploop delta"
	undo_suspend
	exit $MIG_ERR_COPY
	fi
	fi

	log 2 "Dumping container"
	if ! logexec 2 $VZCTL chkpnt $VEID --dump --dumpfile $VE_DUMPFILE ; then
	log 0 "Failed to dump container"
	undo_suspend
	exit $MIG_ERR_CHECKPOINT
	fi

	log 2 "Copying dumpfile"
	time_copy_dump=$(date +%s.%N)
	if ! logexec 2 $SCP $VE_DUMPFILE root@$host:$VE_DUMPFILE ; then
	log 0 "Failed to copy dump"
	undo_dump
	exit $MIG_ERR_COPY
	fi
	else
	if [ "$state" = "running" ]; then
	log 1 "Stopping container"
	if ! logexec 2 $VZCTL stop $VEID ; then
	log 0 "Failed to stop container"
	undo_sync
	exit $MIG_ERR_STOP_SOURCE
	fi
	elif [ "$mounted" = "mounted" ]; then
	log 1 "Unmounting container"
	if ! logexec 2 $VZCTL umount $VEID ; then
	log 0 "Failed to umount container"
	undo_sync
	exit $MIG_ERR_STOP_SOURCE
	fi
	fi
	fi

	if [ "$state" = "running" ]; then
	time_rsync2=$(date +%s.%N)
	if [ "$PLOOP" != "yes" ]; then
	log 2 "Syncing private (2nd pass)"
	if ! $RSYNC "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}"; then
	log 0 "Failed to sync container private areas"
	undo_source_stage
	exit $MIG_ERR_COPY
	fi
	elif [ $online -eq 1 ]; then
	# Do nothing! We have already used ploop copy
	true
	else
	log 2 "Copying top delta"
	if ! $SCP "$VE_PRIVATE/root.hdd/$TOP_DELTA" \
	"root@$host:$VE_PRIVATE/root.hdd/$TOP_DELTA"; then
	log 0 "Failed to scp top ploop delta"
	undo_source_stage
	exit $MIG_ERR_COPY
	fi
	fi
	fi

	if [ "${DISK_QUOTA}" != "no" ]; then
	log 1 "Syncing 2nd level quota"

	log 2 "Dumping 2nd level quota"
	time_quota=$(date +%s.%N)
	if ! vzdqdump $VEID -U -G -T > "$VE_QUOTADUMP"; then
	log 0 "Failed to dump 2nd level quota"
	undo_quota_dump
	exit $MIG_ERR_QUOTA
	fi

	log 2 "Copying 2nd level quota"
	if ! logexec 2 $SCP $VE_QUOTADUMP root@$host:$VE_QUOTADUMP ; then
	log 0 "Failed to copy 2nd level quota dump"
	undo_quota_dump
	exit $MIG_ERR_COPY
	fi

	log 2 "Load 2nd level quota"
	if ! $SSH "root@$host" "(vzdqload $VEID -U -G -T < $VE_QUOTADUMP &&
	vzquota reload2 $VEID)"; then
	log 0 "Failed to load 2nd level quota"
	undo_copy_quota
	exit $MIG_ERR_QUOTA
	fi
	fi

	if [ $online -eq 1 ]; then
	log 2 "Undumping container"
	time_undump=$(date +%s.%N)
	if ! logexec 2 $SSH root@$host $VZCTL restore $VEID --undump \
	--dumpfile $VE_DUMPFILE --skip_arpdetect ; then
	log 0 "Failed to undump container"
	undo_copy_quota
	exit $MIG_ERR_RESTORE_VPS
	fi

	log 2 "Resuming container"
	if ! logexec 2 $SSH root@$host $VZCTL restore $VEID --resume ; then
	log 0 "Failed to resume container"
	undo_undump
	exit $MIG_ERR_RESTORE_VPS
	fi
	time_finish=$(date +%s.%N)
	log 2 "Times:"
	log 2 " Suspend + Dump: " $(get_time $time_suspend $time_copy_dump)
	log 2 " Copy dump file: " $(get_time $time_copy_dump $time_rsync2)
	log 2 " Second rsync: " $(get_time $time_rsync2 $time_quota)
	log 2 " 2nd level quota:" $(get_time $time_quota $time_undump)
	log 2 " Undump + Resume:" $(get_time $time_undump $time_finish)
	log 2 "Total time: " $(get_time $time_suspend $time_finish)

	log 1 "Cleanup"

	log 2 "Killing container"
	logexec 2 $VZCTL chkpnt $VEID --kill
	logexec 2 $VZCTL umount $VEID

	log 2 "Removing dumpfiles"
	rm -f "$VE_DUMPFILE"
	$SSH "root@$host" "rm -f $VE_DUMPFILE"
	else
	if [ "$state" = "running" ]; then
	log 1 "Starting container"
	if ! logexec 2 $SSH root@$host $VZCTL start $VEID ; then
	log 0 "Failed to start container"
	undo_copy_quota
	exit $MIG_ERR_START_VPS
	fi
	elif [ "$mounted" = "mounted" ]; then
	log 1 "Mounting container"
	if ! logexec 2 $SSH root@$host $VZCTL mount $VEID ; then
	log 0 "Failed to mount container"
	undo_copy_quota
	exit $MIG_ERR_MOUNT_VPS
	fi
	elif [ "${DISK_QUOTA}" != "no" ]; then
	log 1 "Turning quota off"
	if ! logexec 2 $SSH root@$host vzquota off $VEID ; then
	log 0 "failed to turn quota off"
	undo_copy_quota
	exit $MIG_ERR_QUOTA
	fi
	fi

	log 1 "Cleanup"
	fi

	if [ $remove_area -eq 1 ]; then
	log 2 "Destroying container"
	logexec 2 $VZCTL destroy $VEID
	else
	# Move config as veid.migrated to allow backward migration
	mv -f $vpsconf $vpsconf.migrated
	fi