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

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

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

elif [ "$IMAGE_TYPE" = "qemu" ] ; then

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

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

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

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

    # If the image is tarball based, then we need to manually create the

    # volumes, filesystems, etc

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

        # Create 3 partitions, /boot, swap, & /

        format_disk0 $blockdev

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

        # need a recent version of qemu for this

        ${QEMU_IMG} convert ${IMAGE_FILE} -O host_device ${blockdev}

    fi

    filesystem_base_dev=$(map_disk0 $blockdev)

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

        boot_dev="${filesystem_base_dev}-1"

        swap_dev="${filesystem_base_dev}-2"

        root_dev="${filesystem_base_dev}-3"

    else

        boot_dev="${filesystem_base_dev}-1"

        root_dev="${filesystem_base_dev}-2"

    fi

    CLEANUP+=("unmap_disk0 $blockdev")

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

        mkfs_disk0 $boot_dev $root_dev $swap_dev

    fi

    TARGET=`mktemp -d` || exit 1

    CLEANUP+=("rmdir $TARGET")

    # mount filesystems

    mount_disk0 $TARGET $root_dev $boot_dev

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

        # unpack image

        tar pzxf $IMAGE_FILE -C $TARGET

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

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

        root_dump="${IMAGE_FILE}"

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

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

          exit 1

        fi

        ( cd ${TARGET}; restore -r -y -f ${root_dump} )

        ( cd ${TARGET}/boot; restore -r -y -f ${boot_dump} )

    fi

    RUN_PARTS=`which run-parts`

    if [ -n "$RUN_PARTS" -a -n "$CUSTOMIZE_DIR" -a -d "$CUSTOMIZE_DIR" ]; then

      TARGET=$TARGET

      BLOCKDEV=$blockdev

      ROOT_DEV=$root_dev

      BOOT_DEV=$boot_dev

      export TARGET SUITE BLOCKDEV ROOT_DEV BOOT_DEV

      $RUN_PARTS $CUSTOMIZE_DIR

    fi

fi

# execute cleanups

cleanup

trap - EXIT

exit 0