Synnefo » snf-image

#!/bin/bash

# Copyright (C) 2011 GRNET S.A. 

#

# 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

set -o pipefail

. common.sh

ganeti_os_main

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

    PS4='$(date "+%s.%N ($LINENO) + ")'

    set -x

elif [ "$IMAGE_DEBUG" != "no" ]; then

    echo "Warning: Unsupported IMAGE_DEBUG value: \`$IMAGE_DEBUG'"

fi

# Save stderr to a file

stderr=$(mktemp)

add_cleanup rm "$stderr"

exec 2> >(tee -a "$stderr" >&2)

monitor_pipe=$(mktemp -u)

mkfifo -m 600 "$monitor_pipe"

add_cleanup rm -f "$monitor_pipe"

if [ -n "$PROGRESS_MONITOR" ]; then

    { sleep 1; $PROGRESS_MONITOR "$instance" < "$monitor_pipe" ; } &

    monitor_pid="$!"

else

    sed -u 's|^|MONITOR MSG: |g' < "$monitor_pipe" &

    monitor_pid="$!"

fi

# Create file descriptor to monitor_pipe

exec {MONITOR_FD}>${monitor_pipe}

add_cleanup  close_fd ${MONITOR_FD}

# Ignore sigpipe signals. If progress monitor is dead and snf-image-host tries

# to output something to the opened pipe, then a sigpipe will be raised. If we

# do not catch this, the program will terminate.

trap "" SIGPIPE

trap "report_and_cleanup $(printf "%q" "$stderr")" EXIT

case $BACKEND_TYPE in

    local)

        image_file="$IMAGE_DIR/$(echo "$IMAGE_NAME" | sed 's/^file://').$IMAGE_TYPE"

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

            log_error "Image file \`$image_file' does not exit."

            exit 1

        fi

        image_size="$(stat -L -c %s "$image_file")"

        ;;

    null)

        image_file=/dev/null

        image_size=0

        # Treat it as local file from now on...

        BACKEND_TYPE="local"

        ;;

    network)

        image_cmd="$CURL $(printf "%q" "$IMAGE_NAME") 2> /dev/null"

        image_size=$($CURL -sI "$IMAGE_NAME" | grep ^Content-Length: | cut -d" " -f2)

        ;;

    pithos)

        # For security reasons pass the database url to pithcat as an

        # environmental variable.

        export PITHCAT_INPUT_DB="$PITHOS_DB"

        export PITHCAT_INPUT_DATA="$PITHOS_DATA"

        cmd_args="$(printf "%q" "${IMAGE_NAME}")"

        image_cmd="./pithcat $cmd_args"

        image_size=$(./pithcat -s  $cmd_args)

        ;;

esac

# 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")

    add_cleanup $LOSETUP -d "$blockdev"

fi

case "$IMAGE_TYPE" in

    ntfsdump|extdump)

        # Create partitions

        format_disk0 "$blockdev" "$IMAGE_TYPE"

        # Install a new MBR

        $INSTALL_MBR -p 1 -i n "$blockdev"

        target="$(map_disk0 "$blockdev")-1" #the root device

        add_cleanup unmap_disk0 "$blockdev"

        snf_export_PROPERTY_ROOT_PARTITION=1

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

            snf_export_PROPERTY_OSFAMILY="windows"

        else

            snf_export_PROPERTY_OSFAMILY="linux"

        fi

        ;;

    diskdump)

        target="$blockdev"

        ;;

    *)

        log_error "Unknown Image format: \`$IMAGE_TYPE'"

        exit 1

        ;;

esac

log_info "Starting image copy..."

monitor="./copy-monitor.py -o $MONITOR_FD -r $image_size"

if [ "$BACKEND_TYPE" = "local" ]; then

    # 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="$target" oflag=direct

else

    $image_cmd | $monitor dd bs=4M of="$target" oflag=direct

fi

log_info "Image copy finished."

# Create a floppy image

floppy=$(mktemp --tmpdir floppy.XXXXXX)

add_cleanup rm "$floppy"

snf_export_DEV=/dev/vda

snf_export_TYPE="$IMG_FORMAT"

snf_export_PASSWORD="$IMG_PASSWD"

snf_export_HOSTNAME="$instance"

if [ -n "$IMG_PROPERTIES" ]; then

    snf_export_PROPERTIES="$IMG_PROPERTIES"

fi

if [ -n "$IMG_PERSONALITY" ]; then

    snf_export_PERSONALITY="$IMG_PERSONALITY"

fi

create_floppy "$floppy"

# Invoke the helper vm to do the dirty job...

jail=$(mktemp -d --tmpdir tmpfsXXXXXXX)

add_cleanup rmdir "$jail"

mount tmpfs -t tmpfs "$jail" -o size=50M

add_cleanup umount "$jail"

result_file=$(mktemp --tmpdir="$jail" result.XXXXXX)

add_cleanup rm "$result_file"

snapshot=$(mktemp --tmpdir="$jail" helperXXXXXX.img)

add_cleanup rm "$snapshot"

"$QEMU_IMG" create -f qcow2 -b "$HELPER_IMG" "$snapshot"

echo -n "$($DATE +%Y:%m:%d-%H:%M:%S.%N) " >&2

log_info "Starting customization VM..."

set +e

$TIMEOUT -k "$HELPER_HARD_TIMEOUT" "$HELPER_SOFT_TIMEOUT" \

    kvm -runas "$HELPER_USER" -drive file="$snapshot" \

    -drive file="$blockdev",format=raw,if=virtio,cache=none \

    -boot c -serial stdio -serial "file:$(printf "%q" "$result_file")" \

    -serial file:>(./helper-monitor.py ${MONITOR_FD}) \

    -fda "$floppy" -vga none -nographic -parallel none -monitor null \

    -kernel "$HELPER_KERNEL" -initrd "$HELPER_INITRD" \

    -append "quiet ro root=/dev/sda1 console=ttyS0,9600n8 snf_image_activate_helper init=/usr/bin/snf-image-helper" \

    2>&1 | sed -u 's|^|HELPER: |g'

rc=$?

set -e

if [ $rc -ne 0 ]; then

    if [ $rc -eq 124 ];  then

        log_error "Image customization was terminated. Did not finish on time."

    elif [ $rc -eq 137 ]; then # (128 + SIGKILL)

        log_error "Image customization was killed. Did not finish on time."

    elif [ $rc -eq 141 ]; then # (128 + SIGPIPE)

        log_error "Image customization was terminated by a SIGPIPE."

        log_error "Maybe progress monitor has died unexpectedly."

    elif [ $rc -eq 125 ]; then

        log_error "Internal Error. Image customization could not start."

        log_error "timeout did not manage to run."

    else

        log_error "Image customization died unexpectedly (return code $rc)."

    fi

    exit 1

else

    echo -n "$($DATE +%Y:%m:%d-%H:%M:%S.%N)" >&2

    log_info "Customization VM finished."

fi

# Read the first line. This will remove \r and \n chars

result=$(sed 's|\r||g' "$result_file" | head -1)

if [ "x$result" != "xSUCCESS" ]; then

    log_error "Image customization failed."

    exit 1

fi

log_info "Image customization finished successfully."

# Execute cleanups

cleanup

trap - EXIT

exit 0

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