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 errtrace

. @osdir@/common.sh

DEFAULT_FILE="@sysconfdir@/default/snf-image-update-helper"

if [ -f "$DEFAULT_FILE" ]; then

    . "$DEFAULT_FILE"

fi

: ${HELPER_URL:="@HELPER_URL@"}

usage() {

    local rc="$1"

    cat <<EOF

Usage: $0 [options]

This script will download a suitable fixed snf-image-helper image and will

install it under $IMAGE_DIR.

OPTIONS:

    -h Print this message

    -u URL

       Download URL to use for the helper image instead of $HELPER_URL

    -y Assume Yes to all queries and do not prompt

    -f Force the installation even if the snf-image and helper versions don't match

EOF

    exit "$rc"

}

while getopts "hu:yf" opt; do

    case $opt in

        h) usage 0

            ;;

        u) HELPER_URL="$OPTARG"

            ;;

        y) NO_PROMPT="yes"

            ;;

        f) FORCE="yes"

            ;;

        ?) log_error "User \`-h' for help"; exit 1

            ;;

    esac

done

cat >&1 <<EOF

$(basename $0) will download a snf-image-helper image from:

\`$HELPER_URL' and will install it under $HELPER_DIR.

The following files will be overwritten if present:

  \`$HELPER_DIR/initrd'

  \`$HELPER_DIR/kernel'

  \`$HELPER_DIR/image'

  \`$HELPER_DIR/packages'

  \`$HELPER_DIR/version'

EOF

while [[ 1 ]]; do

    echo -n "Do you want to continue [y/N]? "

    if [ "x$NO_PROMPT" = "xyes" ]; then

        echo "y";

        break;

    fi

    read answer

    [ "$(echo -n "$answer" | tr [A-Z] [a-z])" = "y" ] && break

    if [ -z "$answer" -o "$(echo -n "$answer" | tr [A-Z] [a-z])" = "n" ]; then

        log_error "Abort."

        exit 1

    fi

done

IMAGE_DIR=$(mktemp -d)

add_cleanup rmdir "$IMAGE_DIR"

cd "$IMAGE_DIR"

IMAGE=$(basename "$HELPER_URL")

echo >&2

echo "Downloading helper image from $HELPER_URL ..." >&2

$CURL -O -L -f "$HELPER_URL"

add_cleanup rm -f "$IMAGE_DIR/$IMAGE"

cd "$HELPER_DIR"

echo >&2

echo "Extracting helper image under \`$HELPER_DIR':" >&2

tar -xvf "$IMAGE_DIR/$IMAGE"

if [ "x$FORCE" != "xyes" ]; then

    echo >&2

    echo "Checking helper image version ..." >&2

    if [ ! -f "$HELPER_DIR/version" ]; then

        log_error "File: \`$HELPER_DIR/version' is missing!"

        exit 1

    fi

    helper_version="$(cat $HELPER_DIR/version)"

    if [ "x$SNF_IMAGE_VERSION" != "x$helper_version" ]; then

        log_error "snf-image version (=$SNF_IMAGE_VERSION) and " \

            "helper image version (=$helper_version) don't match!"

        log_error "Use \`-f' to bypass the version check."

        exit 1

    fi

fi

echo >&2

echo "Helper image was installed successfully!" >&2

exit 0

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