| #!/bin/bash |
| # |
| # vzmigrate is used for VPS migration to another node |
| # Copyright (C) 2000-2006 SWsoft. 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 |
| # |
| # |
| # Usage: |
| # vzmigrate [-r yes|no] [--ssh=<options>] [--keep-dst] [--online] [-v] |
| # destination_address VPSID |
| # Options: |
| # -r, --remove-area yes|no |
| # Whether to remove VPS on source HN for successfully migrated VPS. |
| # --ssh=<ssh options> |
| # Additional options that will be passed to ssh while establishing |
| # connection to destination HN. Please be careful with options |
| # passed, DO NOT pass destination hostname. |
| # --keep-dst |
| # Do not clean synced destination VPS private area in case of some |
| # error. It makes sense to use this option on big VPS migration to |
| # avoid syncing VPS private area again in case some error |
| # (on VPS stop for example) occurs during first migration attempt. |
| # --online |
| # Perform online (zero-downtime) migration: during the migration the |
| # VPS hangs for a while and after the migration it continues working |
| # as though nothing has happened. |
| # -v |
| # Verbose mode. Causes vzmigrate to print debugging messages about |
| # its progress (including some time statistics). |
| # |
| # Examples: |
| # Online migration of VPS #101 to foo.com: |
| # vzmigrate --online foo.com 101 |
| # Migration of VPS #102 to foo.com with downtime: |
| # vzmigrate foo.com 102 |
| # NOTE: |
| # 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. |
| |
| |
| ACT_SCRIPTS_SFX="start stop mount umount" |
| SSH_OPTIONS="" |
| SSH="ssh $SSH_OPTIONS" |
| SCP="scp $SSH_OPTIONS" |
| |
| online=0 |
| verbose=0 |
| remove_area=1 |
| keep_dst=0 |
| debug=0 |
| out="/dev/null" |
| confdir="/etc/vz/conf" |
| vzconf="/etc/vz/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 |
| |
| usage() { |
| cat <<EOF |
| This program is used for VPS migration to another node |
| Usage: |
| vzmigrate [-r yes|no] [--ssh=<options>] [--keep-dst] [--online] [-v] |
| destination_address VPSID |
| Options: |
| -r, --remove-area yes|no |
| Whether to remove VPS on source HN for successfully migrated VPS. |
| --ssh=<ssh options> |
| Additional options that will be passed to ssh while establishing |
| connection to destination HN. Please be careful with options |
| passed, DO NOT pass destination hostname. |
| --keep-dst |
| Do not clean synced destination VPS private area in case of some |
| error. It makes sense to use this option on big VPS migration to |
| avoid syncing VPS private area again in case some error |
| (on VPS stop for example) occurs during first migration attempt. |
| --online |
| Perform online (zero-downtime) migration: during the migration the |
| VPS hangs for a while and after the migration it continues working |
| as though nothing has happened. |
| -v |
| Verbose mode. Causes vzmigrate to print debugging messages about |
| its progress (including some time statistics). |
| EOF |
| exit $MIG_ERR_USAGE |
| } |
| |
| log () { |
| if [ $1 -eq 1 ]; then |
| shift |
| echo -e $@ |
| elif [ $verbose -eq 1 ]; then |
| shift |
| echo -e " " $@ |
| fi |
| } |
| |
| undo_conf () { |
| $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 () { |
| $SSH "root@$host" "vzquota drop $VEID" |
| undo_root |
| } |
| |
| undo_quota_on () { |
| $SSH "root@$host" "vzquota off $VEID" |
| undo_quota_init |
| } |
| |
| undo_sync () { |
| # Root will be destroed in undo_root |
| undo_quota_on |
| } |
| |
| undo_suspend () { |
| vzctl chkpnt $VEID --resume &> $out |
| 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 () { |
| $SSH "root@$host" "vzctl restore $VEID --kill" &> $out |
| 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 |
| echo "OPT:$1" |
| case "$1" in |
| --online) |
| online=1 |
| ;; |
| -v) |
| verbose=1 |
| out="/dev/stdout" |
| ;; |
| --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-)" |
| SSH="ssh $SSH_OPTIONS" |
| SCP="scp $SSH_OPTIONS" |
| ;; |
| *) |
| break |
| ;; |
| esac |
| shift |
| done |
| |
| host=$1 |
| shift |
| VEID=$1 |
| shift |
| |
| if [ -z "$host" -o -z "$VEID" -o $# -ne 0 ]; then |
| usage |
| fi |
| |
| vpsconf="$confdir/$VEID.conf" |
| |
| if [ ! -r "$vpsconf" -o ! -r "$vpsconf" ]; then |
| echo "Can't read VPS #$VEID config file" |
| exit $MIG_ERR_NOEXIST |
| fi |
| |
| get_status $(vzctl status $VEID) |
| if [ "$exist" = "deleted" ]; then |
| echo "VPS #$VEID doesn't exist" |
| exit $MIG_ERR_NOEXIST |
| fi |
| |
| if [ $online -eq 1 ]; then |
| log 1 "Starting online migration of VPS $VEID on $host" |
| else |
| log 1 "Starting migration of VPS $VEID on $host" |
| fi |
| |
| # Try to connect to destination |
| if ! $SSH "-o BatchMode=yes" "root@$host" "/bin/true"; then |
| echo "Can't connect to destination address using public key" |
| echo "Please put your public key to destination node" |
| exit $MIG_ERR_CANT_CONNECT |
| fi |
| |
| dst_exist=$($SSH "root@$host" "vzctl status $VEID" | awk '{print $3}') |
| if [ "$dst_exist" = "exist" ]; then |
| echo "VPS #$VEID already exists on destination node" |
| exit $MIG_ERR_EXISTS |
| fi |
| |
| if [ $online -eq 1 -a "$state" != "running" ]; then |
| echo "Can't perform online migration of stopped VPS" |
| 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 |
| echo "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 ! $SCP "$vpsconf" "root@$host:$vpsconf" &> $out; then |
| log 1 "Failed to copy config file" |
| exit $MIG_ERR_COPY |
| 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 ! $SCP $act_scripts "root@$host:$confdir" &> $out; then |
| log 1 "Failed to copy action scripts" |
| undo_conf |
| exit $MIG_ERR_COPY |
| fi |
| fi |
| |
| log 2 "Creating remote VPS root dir" |
| if ! $SSH "root@$host" "mkdir -p $VE_ROOT"; then |
| log 1 "Failed to make VPS root" |
| undo_act_scripts |
| exit $MIG_ERR_COPY |
| fi |
| |
| log 2 "Creating remote VPS private dir" |
| if ! $SSH "root@$host" "mkdir -p $VE_PRIVATE"; then |
| log 1 "Failed to make VPS private area" |
| undo_private |
| exit $MIG_ERR_COPY |
| fi |
| |
| log 1 "Initializing remote quota" |
| |
| log 2 "Quota init" |
| if ! $SSH "root@$host" "vzctl quotainit $VEID"; then |
| log 1 "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 1 "Failed to turn quota on" |
| undo_quota_init |
| exit $MIG_ERR_QUOTA |
| fi |
| |
| log 1 "Syncing private" |
| if ! rsync -aH --progress "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}" | \ |
| grep "% of" | awk --assign ORS="\r" '{print $10}'; then |
| log 1 "Failed to sync VPS private areas" |
| undo_quota_on |
| exit $MIG_ERR_COPY |
| fi |
| |
| if [ $online -eq 1 ]; then |
| log 1 "Live migrating VPS" |
| |
| log 2 "Suspending VPS" |
| time_suspend=$(date +%s.%N) |
| if ! vzctl chkpnt $VEID --suspend &> $out; then |
| log 1 "Failed to suspend VPS" |
| undo_sync |
| exit $MIG_ERR_CHECKPOINT |
| fi |
| |
| log 2 "Dumping VPS" |
| if ! vzctl chkpnt $VEID --dump --dumpfile $VE_DUMPFILE &> $out; then |
| log 1 "Failed to dump VPS" |
| undo_suspend |
| exit $MIG_ERR_CHECKPOINT |
| fi |
| |
| log 2 "Copying dumpfile" |
| time_copy_dump=$(date +%s.%N) |
| if ! $SCP "$VE_DUMPFILE" "root@$host:$VE_DUMPFILE" &> $out; then |
| log 1 "Failed to copy dump" |
| undo_dump |
| exit $MIG_ERR_COPY |
| fi |
| else |
| if [ "$state" = "running" ]; then |
| log 1 "Stopping VPS" |
| if ! vzctl stop $VEID &> $out; then |
| log 1 "Failed to stop VPS" |
| undo_sync |
| exit $MIG_ERR_STOP_SOURCE |
| fi |
| elif [ "$mounted" = "mounted" ]; then |
| log 1 "Unmounting VPS" |
| if ! vzctl umount $VEID &> $out; then |
| log 1 "Failed to umount VPS" |
| 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 -aH "$VE_PRIVATE" "root@$host:${VE_PRIVATE%/*}"; then |
| log 1 "Failed to sync VPS private areas" |
| undo_source_stage |
| exit $MIG_ERR_COPY |
| fi |
| fi |
| |
| 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 2 "Failed to dump 2nd level quota" |
| undo_quota_dump |
| exit $MIG_ERR_QUOTA |
| fi |
| |
| log 2 "Copying 2nd level quota" |
| if ! $SCP "$VE_QUOTADUMP" "root@$host:$VE_QUOTADUMP" &> $out; then |
| log 2 "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 2 "Failed to load 2nd level quota" |
| undo_copy_quota |
| exit $MIG_ERR_QUOTA |
| fi |
| |
| if [ $online -eq 1 ]; then |
| log 2 "Undumping VPS" |
| time_undump=$(date +%s.%N) |
| if ! $SSH "root@$host" "vzctl restore $VEID --undump \ |
| --dumpfile $VE_DUMPFILE --skip_arpdetect" &> $out; then |
| log 1 "Failed to undump VPS" |
| undo_copy_quota |
| exit $MIG_ERR_RESTORE_VPS |
| fi |
| |
| log 2 "Resuming VPS" |
| if ! $SSH "root@$host" "vzctl restore $VEID --resume" &> $out; then |
| log 1 "Failed to resume VPS" |
| undo_undump |
| exit $MIG_ERR_RESTORE_VPS |
| fi |
| time_finish=$(date +%s.%N) |
| log 2 "Times:" |
| log 2 "\tSuspend + Dump:\t" $(get_time $time_suspend $time_copy_dump) |
| log 2 "\tCopy dump file:\t" $(get_time $time_copy_dump $time_rsync2) |
| log 2 "\tSecond rsync:\t" $(get_time $time_rsync2 $time_quota) |
| log 2 "\t2nd level quota:\t" $(get_time $time_quota $time_undump) |
| log 2 "\tUndump + Resume:\t" $(get_time $time_undump $time_finish) |
| log 2 "Total time: " $(get_time $time_suspend $time_finish) |
| |
| log 1 "Cleanup" |
| |
| log 2 "Killing VPS" |
| vzctl chkpnt $VEID --kill &> $out |
| vzctl umount $VEID &> $out |
| |
| log 2 "Removing dumpfiles" |
| rm -f "$VE_DUMPFILE" |
| $SSH "root@$host" "rm -f $VE_DUMPFILE" |
| else |
| if [ "$state" = "running" ]; then |
| log 1 "Staring VPS" |
| if ! $SSH "root@$host" "vzctl start $VEID" &> $out; then |
| log 1 "Failed to start VPS" |
| undo_copy_quota |
| exit $MIG_ERR_START_VPS |
| fi |
| elif [ "$mounted" = "mounted" ]; then |
| log 1 "Mounting VPS" |
| if ! $SSH "root@$host" "vzctl mount $VEID" &> $out; then |
| log 1 "Failed to mount VPS" |
| undo_copy_quota |
| exit $MIG_ERR_MOUNT_VPS |
| fi |
| else |
| log 1 "Turning quota off" |
| if ! $SSH "root@$host" "vzquota off $VEID" &> $out; then |
| log 1 "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 VPS" |
| vzctl destroy $VEID &> $out |
| else |
| # Move config as veid.migrated to allow backward migration |
| mv -f $vpsconf $vpsconf.migrated |
| fi |