Synnefo » Ganeti Instance Image

#!/bin/bash

# Copyright (C) 2010 Oregon State University

#

# 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., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

set -e

. common.sh

debug set -x

OS_FAMILY="linux"

case "$IMAGE_TYPE" in

tarball)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}.tar.gz"

    ;;

qemu)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}.img"

    ;;

dump)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}-root.dump"

    # Workaround for restore's non-unique /tmp/rstdir* and /tmp/rstmode* 

    export TMPDIR=$(mktemp -d /tmp/${INSTANCE_NAME}_XXXXXXXX)

    CLEANUP+=("rmdir --ignore-fail-on-non-empty $TMPDIR")

    ;;

extdump)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}-root.extdump"

    ;;

ntfsclone)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}.img"

    OS_FAMILY="windows"

    ;;

ntfsdump)

    IMAGE_FILE="${IMAGE_DIR}/${IMAGE_NAME}-${ARCH}.ntfsdump"

    OS_FAMILY="windows"

    ;;

*)

    log_error "Unknown image type.";

    exit 1

esac

MONITOR="" #Empty if progress monitor support is disabled

if [ "$progress_monitor_support" = "yes" ]; then

    IMAGE_SIZE="$(stat -L -c %s ${IMAGE_FILE})"

    MONITOR="$PROGRESS_MONITOR -i ${INSTANCE_NAME} -r ${IMAGE_SIZE}"

fi

if [ "$OS_FAMILY" = "linux" ]; then

    . common_linux.sh

elif [ "$OS_FAMILY" = "windows" ]; then

    . common_windows.sh

    SWAP="no"

fi

if [ "$CDINSTALL" = "no" ] ; then

    # If the target device is not a real block device we'll first losetup it.

    # This is needed for file disks.

    if [ ! -b $blockdev ]; then

      ORIGINAL_BLOCKDEV=$blockdev

      blockdev=$($LOSETUP -sf $blockdev)

      CLEANUP+=("$LOSETUP -d $blockdev")

    fi

    if [ ! -f "$IMAGE_FILE" ] ; then

      log_error "Can't find image file: $IMAGE_FILE"

      exit 1

    fi

    case "$IMAGE_TYPE" in

    tarball|dump|ntfsclone|ntfsdump|extdump)

        #manually create volumes, filesystems, etc

        ${OS_FAMILY}_format_disk0 $blockdev

        ;;

    qemu)

        # need a recent version of qemu for this

        ${QEMU_IMG} convert ${IMAGE_FILE} -O host_device ${blockdev} > /dev/null

        ;;

    esac

    # deploying something like a windows image, skip the rest

    if [ "${NOMOUNT}" = "yes" ] ; then

      cleanup

      exit 0

    fi

    filesystem_dev=$(map_disk0 $blockdev)

    CLEANUP+=("unmap_disk0 $blockdev")

    root_dev=$(map_partition $filesystem_dev root)

    boot_dev=$(map_partition $filesystem_dev boot)

    swap_dev=$(map_partition $filesystem_dev swap)

    # Check partition existence

    if [ ! -e "$root_dev" ] ; then

        log_error "Root partition: $root_dev does not exist!"

        exit 1;

    fi

    if [ -n "$boot_dev" -a ! -e "$boot_dev" ] ; then

        log_error "Boot partition: $boot_dev does not exist!"

        exit 1;

    fi

    if [ -n "$swap_dev" -a ! -e "$swap_dev" ] ; then

        log_error "Swap partiion: $swap_dev does not exist!"

        exit 1;

    fi

    # Before mounting

    case "${IMAGE_TYPE}" in

    tarball|dump)

        ${OS_FAMILY}_mkfs_disk0

        root_uuid="$($VOL_ID $root_dev)"

        [ -n "$boot_dev" ] && sleep 1 && boot_uuid="$($VOL_ID $boot_dev)"

        [ -n "$swap_dev" ] && sleep 1 && swap_uuid="$($VOL_ID $swap_dev)"

        ;;

    ntfsclone)

        # Restore the ntfs image

        $MONITOR $NTFSCLONE --restore-image --overwrite $root_dev $IMAGE_FILE \

            > /dev/null

        ;;

    ntfsdump|extdump)

        # dd the dump to its new home :-)

        # Deploying an image file on a target block device is a streaming

        # copy operation. Enable the direct I/O flag on the output fd to 

        # avoid polluting the host cache with useless data.

        $MONITOR dd bs=4M if=$IMAGE_FILE of=$root_dev oflag=direct

        # resize the partition

        [ "${IMAGE_TYPE}" = "extdump" ] && $RESIZE2FS $root_dev

        ;;

    esac

    TARGET=`mktemp -d` || exit 1

    CLEANUP+=("rmdir $TARGET")

    # mount filesystems

    mount_disk0 $TARGET

    if [ "${IMAGE_TYPE}" = "tarball" ] ; then

        # unpack image

        tar pzxf $IMAGE_FILE -C $TARGET

    elif [ "${IMAGE_TYPE}" = "dump" ] ; then

        root_dump="${IMAGE_FILE}"

        ( cd ${TARGET}; $MONITOR $RESTORE -r -y -f ${root_dump} > /dev/null )

        if [ -n "${boot_dev}" ] ; then

            boot_dump="${IMAGE_FILE/root.dump/boot.dump}"

            if [ ! -f "$boot_dump" ] ; then

              log_error "Can't find image file: $boot_dump"

              exit 1

            fi

            ( cd ${TARGET}/boot; $RESTORE -r -y -f ${boot_dump} > /dev/null )

        fi

    fi 

    if [ "${IMAGE_TYPE}" = "tarball" -o "${IMAGE_TYPE}" = "dump" ] ; then

        ${OS_FAMILY}_setup_fstab $TARGET

    fi

    if [ "${INSTANCE_HV_serial_console}" = "True" ] ; then

        ${OS_FAMILY}_setup_console $TARGET

    fi

    ${OS_FAMILY}_epilogue $TARGET

    RUN_PARTS=`which run-parts`

    if [ -n "$RUN_PARTS" -a -n "$CUSTOMIZE_DIR" -a -d "${CUSTOMIZE_DIR}/$OS_FAMILY" ]; then

      TARGET=$TARGET

      BLOCKDEV=$blockdev

      ROOT_DEV=$root_dev

      BOOT_DEV=$boot_dev

      SWAP_DEV=$swap_dev

      IMG_PASSWD=$IMG_PASSWD

      HOSTNAME=$instance

      export TARGET SUITE BLOCKDEV ROOT_DEV BOOT_DEV SWAP_DEV IMAGE_TYPE IMG_PASSWD HOSTNAME

      $RUN_PARTS $CUSTOMIZE_DIR/$OS_FAMILY

    fi

fi

# execute cleanups

cleanup

trap - EXIT

exit 0