blob: 4a88bca3de425c334248fcc26d623fcc47acda19 [file] [log] [blame] [raw]
#!/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=
# 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
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}'
}
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))
;;
--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
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"
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 [ "${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
else
log 2 "VZ disk quota disabled -- skipping quota migration"
fi
log 1 "Syncing private"
$RSYNC "$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..."
log 2 "Suspending container"
time_suspend=$(date +%s.%N)
if ! logexec 2 $VZCTL chkpnt $VEID --suspend ; then
log 0 "Failed to suspend container"
undo_sync
exit $MIG_ERR_CHECKPOINT
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
log 2 "Syncing private (2nd pass)"
time_rsync2=$(date +%s.%N)
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
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
else
log 2 "VZ disk quota disabled -- skipping quota migration"
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