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.

#

# This is an example script that configures, grub after installation. This

# script assumes that grub has been installed onto the image already and a

# working grub.conf exists. This is only enabled if using the tarball image

# type. 

set -e

. common.sh

debug set -x

CLEANUP=( )

trap cleanup EXIT

if [ -z "${TARGET}" -o ! -d "${TARGET}" ] ; then

    log_error "Missing target directory"

    exit 1

fi

# Set disk based on type of hypervisor

disk=""

if [ "${HYPERVISOR}" = "kvm" ] ; then

  disk="vda"

else

  disk="xda"

fi

boot_dir="${TARGET}/boot/grub"

setup_disk_devs() {

    root_part="${ROOT_DEV/*-/}"

    boot_part="${BOOT_DEV/*-/}"

    swap_part="${SWAP_DEV/*-/}"

    mknod ${TARGET}/dev/${disk} b $(stat -L -c "0x%t 0x%T" $BLOCKDEV)

    CLEANUP+=("rm -f ${TARGET}/dev/$disk")

    mknod ${TARGET}/dev/${disk}${root_part} b $(stat -L -c "0x%t 0x%T" $ROOT_DEV)

    CLEANUP+=("rm -f ${TARGET}/dev/${disk}${root_part}")

    if [ -n "$BOOT_DEV" ] ; then

        mknod ${TARGET}/dev/${disk}${boot_part} b $(stat -L -c "0x%t 0x%T" $BOOT_DEV)

        CLEANUP+=("rm -f ${TARGET}/dev/${disk}${boot_part}")

    fi

    if [ -n "$BOOT_DEV" ]; then

        grub1_root=$(($boot_part-1))

        grub2_root=$boot_part

    else

        grub1_root=$(($root_part-1))

        grub2_root=$root_part

    fi

    # setup minimal proc environment for grub2

    $MKDIR_P ${TARGET}/proc/

    cat > ${TARGET}/proc/mounts <<EOF

/dev/${disk}${boot_part} /boot ${FILESYSTEM} rw,relatime,barrier=1,data=ordered 0 0

/dev/${disk}${root_part} / ${FILESYSTEM} rw,relatime,barrier=1,data=ordered 0 0

EOF

    CLEANUP+=("rm -r ${TARGET}/proc/mounts")

    cat > ${TARGET}/proc/devices <<EOF

Block devices:

251 virtblk

252 device-mapper

EOF

    CLEANUP+=("rm -r ${TARGET}/proc/devices")

    cat > ${TARGET}/proc/misc << EOF

 60 device-mapper

EOF

    CLEANUP+=("rm -r ${TARGET}/proc/misc")

}

add_devicemap() {

    cat > "${TARGET}/boot/grub/device.map" <<EOF

(hd0) /dev/${disk}

EOF

}

if [ "${IMAGE_TYPE}" != "qemu" ] && \

   [ -z "${INSTANCE_HV_kernel_path}" -o "${IMPORT_SCRIPT}" = "1" ] ; then

    setup_disk_devs

    add_devicemap

    if [ -e "${boot_dir}/menu.lst" ] ; then

        # install grub to the block device

        chroot ${TARGET} grub --batch --no-floppy \

            --device-map=/boot/grub/device.map > /dev/null <<EOF

root (hd0,${grub1_root})

setup (hd0)

quit

EOF

        # check to see if grub is using UUID's and replace if so

        if [ -n "$(grep 'root=UUID' ${boot_dir}/menu.lst)" ] ; then

            root_uuid="$($VOL_ID $ROOT_DEV)"

            sed --follow-symlinks -ie "s/\(root=UUID=\)\([a-z0-9-]*\)/\1${root_uuid}/g" \

                ${boot_dir}/menu.lst

        fi

        # additional kernel arguments for the instance

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

            sed --follow-symlinks -ie "s/\(.*kernel.*\)/\1 ${KERNEL_ARGS}/g" \

                ${boot_dir}/menu.lst

        fi

    elif [ -e "${boot_dir}/grub.cfg" -a -e "${TARGET}/etc/default/grub" ] ; then

        echo "grub2 support is partially supported"

        echo "please run update-grub after the instance is online"

        # check to see if grub is using UUID's and replace if so

        if [ -n "$(grep 'root=UUID' ${boot_dir}/grub.cfg)" ] ; then

            root_uuid="$($VOL_ID $ROOT_DEV)"

            sed --follow-symlinks -ie "s/\(root=UUID=\)\([a-z0-9-]*\)/\1${root_uuid}/g" \

                ${boot_dir}/grub.cfg

        fi

        # additional kernel arguments for the instance

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

            sed -ie "s/.*GRUB_CMDLINE_LINUX=\"\(.*\)\"/GRUB_CMDLINE_LINUX=\"\1 ${KERNEL_ARGS}\"/" \

                ${TARGET}/etc/default/grub

        fi

        # show the menu countdown in case we want to boot to a different kernel

        sed -ie 's/.*GRUB_HIDDEN_TIMEOUT=.*/GRUB_HIDDEN_TIMEOUT=5/' ${TARGET}/etc/default/grub

        sed -ie 's/.*GRUB_HIDDEN_TIMEOUT_QUIET=.*/GRUB_HIDDEN_TIMEOUT_QUIET=false/' ${TARGET}/etc/default/grub

        # install grub2 MBR

        chroot ${TARGET} grub-setup --force --device-map=/boot/grub/device.map \

            --directory=/boot/grub --root-device="(hd0,${grub2_root})" "(hd0)" > /dev/null

    fi

fi

# setup serial console

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

    if [ ! -e ${TARGET}/dev/${disk} ] ; then

        setup_disk_devs

    fi

    if [ -e "${boot_dir}/menu.lst" ] ; then

        # grub 0.x

        sed --follow-symlinks -ie 's/^default.*/default 0\n\nserial --unit=0\nterminal --timeout=3 console serial/' \

            ${boot_dir}/menu.lst

        sed --follow-symlinks -ie 's/\(.*kernel.*\)/\1 console=ttyS0,115200n8/g' \

            ${boot_dir}/menu.lst

    elif [ -e "${boot_dir}/grub.cfg" -a -e "${TARGET}/etc/default/grub" ] ; then

        # grub 2.x

        add_devicemap

        sed -ie 's/.*GRUB_TERMINAL.*/GRUB_TERMINAL=serial/' ${TARGET}/etc/default/grub

        sed -ie 's/.*GRUB_CMDLINE_LINUX=\"\(.*\)\"/GRUB_CMDLINE_LINUX=\"\1 console=ttyS0,115200n8\"/' \

            ${TARGET}/etc/default/grub

        echo "GRUB_SERIAL_COMMAND=\"serial --speed=115200 --unit=0 --word=8 --parity=no --stop=1\"" \

            >> ${TARGET}/etc/default/grub

    else

        echo "No grub bootloader found, skipping..."

    fi

fi

# execute cleanups

cleanup

trap - EXIT

exit 0