#!/bin/bash
-# Copyright (C) 2011 GRNET S.A.
+# 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
PS4='$(date "+%s.%N ($LINENO) + ")'
set -x
elif [ "$IMAGE_DEBUG" != "no" ]; then
- echo "Warning: Unsupported IMAGE_DEBUG value: \`$IMAGE_DEBUG'"
+ log_warning "Unsupported IMAGE_DEBUG value: \`$IMAGE_DEBUG'"
fi
monitor_pipe=$(mktemp -u)
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.
+# Ignore sigpipe signals. If progress monitor is dead and snf-image 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
+trap report_and_cleanup EXIT
+
+echo "Processing image with ID: \`$IMG_ID' and type: \`$IMAGE_TYPE'" >&2
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."
+ if [[ "$IMAGE_NAME" =~ ^local:// ]]; then
+ IMAGE_NAME="${IMAGE_NAME:8}"
+ elif [[ "$IMAGE_NAME" =~ ^file:// ]]; then
+ IMAGE_NAME="${IMAGE_NAME:7}"
+ log_warning "The file:// back-end identifier is deprecated and" \
+ "will be removed in the future. Use local:// instead."
+ fi
+
+ canonical_image_dir="$(canonicalize "$IMAGE_DIR")"
+ if [ ! -d "$canonical_image_dir" ]; then
+ log_error "The IMAGE_DIR directory: \`$IMAGE_DIR' does not exist."
report_error "Unable to retrieve image file."
+ fi
+
+ image_file="$IMAGE_DIR/$IMAGE_NAME"
+ if [ ! -e "$image_file" ]; then
+ if [ -e "$image_file.$IMAGE_TYPE" ] ; then
+ image_file="$image_file.$IMAGE_TYPE"
+ log_warning "The \`.$IMAGE_TYPE' extension is missing from" \
+ "the local back-end id. This id form is deprecated and" \
+ " will be remove in the future."
+ else
+ log_error "Image file \`$image_file' does not exist."
+ report_error "Unable to retrieve image file."
+ exit 1
+ fi
+ fi
+
+ canonical_image_file="$(canonicalize "$image_file")"
+
+ if [[ "$canonical_image_file" != "$canonical_image_dir"* ]]; then
+ log_error "Image ID points to a file outside the image directory: \`$IMAGE_DIR'"
+ report_error "Invalid image ID"
exit 1
fi
+
image_size="$(stat -L -c %s "$image_file")"
;;
null)
BACKEND_TYPE="local"
;;
network)
- image_cmd="$CURL $(printf "%q" "$IMAGE_NAME") 2> /dev/null"
+ image_cmd="$CURL $(printf "%q" "$IMAGE_NAME")"
image_size=$($CURL -sI "$IMAGE_NAME" | grep ^Content-Length: | cut -d" " -f2)
;;
pithos)
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"
snf_export_PERSONALITY="$IMG_PERSONALITY"
fi
-create_floppy "$floppy"
+snf_export_DEV=$(get_img_dev)
-# 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"
-
-report_info "Starting customization VM..."
-echo "$($DATE +%Y:%m:%d-%H:%M:%S.%N) VM START" >&2
-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
-echo "$($DATE +%Y:%m:%d-%H:%M:%S.%N) VM STOP" >&2
-if [ $rc -ne 0 ]; then
- if [ $rc -eq 124 ]; then
- log_error "Customization VM was terminated. Did not finish on time."
- report_error "Image customization failed. Did not finish on time."
- elif [ $rc -eq 137 ]; then # (128 + SIGKILL)
- log_error "Customization VM was killed. Did not finish on time."
- report_error "Image customization failed. Did not finish on time."
- elif [ $rc -eq 141 ]; then # (128 + SIGPIPE)
- log_error "Customization VM 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 "Customization VM died unexpectedly (return code $rc)."
- fi
- exit 1
-else
- report_info "Customization VM exited normally."
-fi
-
-report_info "Checking customization status..."
-# Read the first line. This will remove \r and \n chars
-result=$(sed 's|\r||g' "$result_file" | head -1)
+create_floppy "$floppy"
-if [ "x$result" != "xSUCCESS" ]; then
- log_error "Image customization failed."
- report_error "Image customization failed."
- exit 1
-fi
+launch_helper "$blockdev" "$floppy"
report_info "Image customization finished successfully."
projects / snf-image / blobdiff