Synnefo » snf-image

#!/bin/bash

set -e

set -x

. @osdir@/common.sh

CACHE_FILE="$HELPER_DIR/cache.tar"

ARCH=amd64

EXTRA_PKGS="linux-image-amd64,e2fsprogs,ntfs-3g,xmlstarlet"

helper_img=$(mktemp $HELPER_DIR/image.XXXXXXXX)

dd if=/dev/zero of=$helper_img bs=1k count=400000

blockdev=$(losetup -sf $helper_img)

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

format_disk0 $blockdev "extdump"

root_dev=$(map_disk0 $blockdev)-1

CLEANUP+=("unmap_disk0 $blockdev")

mkfs.ext3 $root_dev

TMPDIR=$(mktemp -d)

CLEANUP+=("rmdir $TMPDIR")

mount $root_dev $TMPDIR

CLEANUP+=("umount $root_dev")

if [  -f "$CACHE_FILE" ]; then

    tar xf "$CACHE_FILE" -C $TMPDIR

else

    debootstrap --arch "$ARCH" --include $EXTRA_PKGS --variant=minbase \

	    squeeze $TMPDIR

    # remove the downloaded debs, as they are no longer needed

    find "$TMPDIR/var/cache/apt/archives" -type f -name '*.deb' -print0 | \

        xargs -r0 rm -f

    TMP_CACHE=$(mktemp "${CACHE_FILE}.XXXXXX")

    tar cf "$TMP_CACHE" -C $TMPDIR .

    mv -f "$TMP_CACHE" "$CACHE_FILE"

fi

echo helper > $TMPDIR/etc/hostname

cat > $TMPDIR/etc/fstab <<EOF

# /etc/fstab: static file system information.

#

# <file system>   <mount point>   <type>  <options>       <dump>  <pass>

/dev/sda1         /               ext3    defaults        0       1

proc              /proc           proc    defaults        0       0

EOF

if [ ! -L "$TMPDIR/vmlinuz" -o ! -L "$TMPDIR/vmlinuz" ]; then

	log_error "vmlinuz or initrd.img link in root is missing."

	log_error "I don't know how to find a usable kernel/initrd pair."

	exit 1

fi

kernel=$(readlink -en $TMPDIR/vmlinuz)

initrd=$(readlink -en $TMPDIR/initrd.img)

echo "Moving $(basename $kernel) and $(basename $initrd) to $HELPER_DIR"

mv $kernel $initrd $HELPER_DIR

kernel=$(basename $kernel)

initrd=$(basename $initrd)

(cd $HELPER_DIR; ln -fs $kernel kernel; ln -fs $initrd initrd)

rm $TMPDIR/vmlinuz $TMPDIR/initrd.img

cleanup

mv $helper_img $HELPER_DIR/image

trap - EXIT

exit 0

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

dist_bin_SCRIPTS = snf-image-update-helper

	   -e 's|@HELPER_DIR[@]|$(HELPER_DIR)|g' \

	   -e 's|@HELPER_IMG[@]|$(HELPER_IMG)|g' \

	   -e 's|@HELPER_PKG[@]|$(HELPER_PKG)|g' \

%:%.in Makefile

CLEANFILES = $(os_DATA) $(dist_bin_SCRIPTS)

# this one is only to be called by create

ganeti_os_main() {

    if [ -z "$OS_API_VERSION" -o "$OS_API_VERSION" = "5" ]; then

        OS_API_VERSION=5

        GETOPT_RESULT=`getopt -o o:n:i:b:s: -n '$0' -- "$@"`

        if [ $? != 0 ] ; then log_error "Terminating..."; exit 1 ; fi

        get_api5_arguments $GETOPT_RESULT

    elif [ "$OS_API_VERSION" = "10" -o "$OS_API_VERSION" = "15" ]; then

        get_api10_arguments

    elif [ "$OS_API_VERSION" = "20" ]; then

        get_api20_arguments

        IMAGE_NAME=$IMG_ID

        IMAGE_TYPE=$IMG_FORMAT

    else

        log_error "Unknown OS API VERSION $OS_API_VERSION"

        exit 1

    fi

    if [ -n "$OS_VARIANT" ]; then

        if [ ! -d "$VARIANTS_DIR" ]; then

            log_error "OS Variants directory $VARIANTS_DIR doesn't exist"

            exit 1

        fi

        VARIANT_CONFIG="$VARIANTS_DIR/$OS_VARIANT.conf"

        if [ -f "$VARIANT_CONFIG" ]; then

            . "$VARIANT_CONFIG"

        else

            if grep -qxF "$OS_VARIANT" variants.list; then

                log_error "ERROR: instance-image configuration error"

                log_error "  Published variant $OS_VARIANT is missing its config" \

                          "file"

                log_error "  Please create $VARIANT_CONFIG or unpublish the variant"

                log_error "  (by removing $OS_VARIANT from variants.list)"

            else

                log_error "Unofficial variant $OS_VARIANT is unsupported"

                log_error "Most probably this is a user error, forcing a wrong name"

                log_error "To support this variant please create file" \

                            "$VARIANT_CONFIG"

            fi

            exit 1

        fi

    fi

}

# if something fails here, it souldn't call cleanup again...

    trap - EXIT

            # If something fails here, it's better to retry it for a few times

            # before we give up with an error. This is needed for kpartx when

            # dealing with ntfs partitions mounted through fuse. umount is not

            # synchronous and may return while the partition is still busy. A

            # premature attempt to delete partition mappings through kpartx on a

            # device that hosts previously mounted ntfs partition may fail with

            # an  `device-mapper: remove ioctl failed: Device or resource busy'

            # error. A sensible workaround for this is to wait for a while and

            # then try again.

            local cmd=${CLEANUP[$i]}

            $cmd || for interval in 0.25 0.5 1 2 4; do

            echo "Command $cmd failed!"

            echo "I'll wait for $interval secs and will retry..."

            sleep $interval

            $cmd && break

        test $? -eq 1 && { echo "Giving Up..."; exit 1; }

    done

  fi

: ${HELPER_DIR:="@HELPER_DIR@"}

: ${HELPER_IMG:="@HELPER_IMG@"}

: ${HELPER_USER="nobody"}

# --with-helper-dir

AC_ARG_WITH([helper-dir],

    [AS_HELP_STRING([--with-helper-dir=DIR],

        [top-level directory to host the helper VM

        [LOCALSTATEDIR/lib/snf-image/helper]]

    [helper_dir="$withval"],

    [helper_dir="$localstatedir/lib/snf-image/helper"])

AC_SUBST(HELPER_DIR, $helper_dir)

# --with-helper-img

AC_ARG_WITH([helper-img],

    [AS_HELP_STRING([--with-helper-img=IMG_PATH],

        [Path to helper VM image [HELPERDIR/image]]

    )],

    [helper_img="$withval"],

    [helper_img="$helper_dir/image"])

AC_SUBST(HELPER_IMG, $helper_img)

AC_ARG_WITH([helper-kernel],

        [Path to the helper VM kernel [HELPERDIR/kernel]]

    [helper_kernel="$helper_dir/kernel"])

AC_ARG_WITH([helper-initrd],

        [Path to the helper VM initial ramdist [HELPERDIR/initrd]]

    [helper_initrd="$helper_dir/initrd"])

ganeti_os_main

    kvm -runas $HELPER_USER -drive file=${HELPER_IMG},snapshot=on \

# HELPER_DIR: Directory hosting the helper files

# HELPER_DIR: /var/lib/snf-image/helper/

# HELPER_IMG: Path to the helper VM image

# HELPER_IMG="$(HELPER_DIR)/image"

# HELPER_KERNEL: Path to the helper VM kernel

# HELPER_KERNEL="$(HELPER_DIR)/kernel"

# HELPER_PKG: Path to the snf-image-helper package

# HELPER_PKG="$(HELPER_DIR)/snf-image-helper.deb"

# HELPER_INITRD: Path to the helper VM initial ramdisk

# HELPER_INITRD="$(HELPER_DIR)/initrd"

# HELPER_USER="nobody"