1 # Copyright (C) 2011 GRNET S.A.
2 # Copyright (C) 2007, 2008, 2009 Google Inc.
4 # This program is free software; you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation; either version 2 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful, but
10 # WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 # General Public License for more details.
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
24 INSTALL_MBR="install-mbr"
28 network_backend_support="@network_backend_support@"
30 MSG_TYPE_ERROR="error"
37 for arg; do cmd+=$(printf "%q " "$arg"); done
55 local type="$MSG_TYPE_ERROR"
58 report="\"type\":\"$type\","
59 report+="\"timestamp\":$(date +%s.%N),"
60 report+="\"location\":\"$location\","
62 if [ ${#ERROR_MSGS[@]} -gt 0 ]; then
64 for err in "${ERROR_MSGS[@]}"; do
65 msg+="\"$(sed 's/"/\\"/g' <<< "$err")\","
67 if [ ${#msg} -gt 1 ]; then
68 # remove last comma (,)
72 report+="\"messages\":$msg}"
74 local stderr="$(cat "$error_file" | sed 's/"/\\"/g')"
75 report+="\"stderr\":\"$stderr\"}"
78 eval "echo $(printf "%q" "$report") >&${monitor_fd}"
81 get_api5_arguments() {
83 # Note the quotes around `$TEMP': they are essential!
84 eval set -- "$GETOPT_RESULT"
87 -i|-n) instance=$2; shift 2;;
89 -o) old_name=$2; shift 2;;
91 -b) blockdev=$2; shift 2;;
93 -s) swapdev=$2; shift 2;;
97 *) log_error "Internal error!" >&2; exit 1;;
100 if [ -z "$instance" -o -z "$blockdev" ]; then
101 log_error "Missing OS API Argument (-i, -n, or -b)"
104 if [ "$SCRIPT_NAME" != "export" -a -z "$swapdev" ]; then
105 log_error "Missing OS API Argument -s (swapdev)"
108 if [ "$SCRIPT_NAME" = "rename" -a -z "$old_name" ]; then
109 log_error "Missing OS API Argument -o (old_name)"
114 get_api10_arguments() {
115 if [ -z "$INSTANCE_NAME" -o -z "$HYPERVISOR" -o -z "$DISK_COUNT" ]; then
116 log_error "Missing OS API Variable:"
117 log_error "(INSTANCE_NAME HYPERVISOR or DISK_COUNT)"
120 instance=$INSTANCE_NAME
121 if [ $DISK_COUNT -lt 1 -o -z "$DISK_0_PATH" ]; then
122 log_error "At least one disk is needed"
125 if [ "$SCRIPT_NAME" = "export" ]; then
126 if [ -z "$EXPORT_DEVICE" ]; then
127 log_error "Missing OS API Variable EXPORT_DEVICE"
129 blockdev=$EXPORT_DEVICE
130 elif [ "$SCRIPT_NAME" = "import" ]; then
131 if [ -z "$IMPORT_DEVICE" ]; then
132 log_error "Missing OS API Variable IMPORT_DEVICE"
134 blockdev=$IMPORT_DEVICE
136 blockdev=$DISK_0_PATH
138 if [ "$SCRIPT_NAME" = "rename" -a -z "$OLD_INSTANCE_NAME" ]; then
139 log_error "Missing OS API Variable OLD_INSTANCE_NAME"
141 old_name=$OLD_INSTANCE_NAME
144 get_api20_arguments() {
146 if [ -z "$OSP_IMG_ID" ]; then
147 log_error "Missing OS API Parameter: OSP_IMG_ID"
150 if [ -z "$OSP_IMG_FORMAT" ]; then
151 log_error "Missing OS API Parameter: OSP_IMG_FORMAT"
154 if [ -z "$OSP_IMG_PASSWD" ]; then
155 log_error "Missing OS API Parameter: OSP_IMG_PASSWD"
160 IMG_FORMAT=$OSP_IMG_FORMAT
161 IMG_PASSWD=$OSP_IMG_PASSWD
162 if [ -n "$OSP_IMG_PROPERTIES" ]; then
163 IMG_PROPERTIES="$OSP_IMG_PROPERTIES"
165 if [ -n "$OSP_IMG_PERSONALITY" ]; then
166 IMG_PERSONALITY="$OSP_IMG_PERSONALITY"
172 filesystem_dev_base=$($KPARTX -l -p- $blockdev | \
173 grep -m 1 -- "-1.*$blockdev" | \
175 if [ -z "$filesystem_dev_base" ]; then
176 log_error "Cannot interpret kpartx output and get partition mapping"
179 $KPARTX -a -p- "$blockdev" > /dev/null
180 filesystem_dev="/dev/mapper/${filesystem_dev_base/%-1/}"
181 if [ ! -b "/dev/mapper/$filesystem_dev_base" ]; then
182 log_error "Can't find kpartx mapped partition:" \
183 "/dev/mapper/$filesystem_dev_base"
186 echo "$filesystem_dev"
195 local image_type="$2"
197 declare -A part_id=( ['extdump']="83" ["ntfsdump"]="7" )
199 # The -f is needed, because we use an optimal alignment and sfdisk complains
200 # about partitions not ending on clylinder boundary.
201 local sfdisk_cmd="$SFDISK -uS -H 255 -S 63 -f --quiet --Linux --DOS $device"
203 $sfdisk_cmd > /dev/null <<EOF
204 2048,,${part_id["$image_type"]},*
211 local target=$(mktemp -d)
212 add_cleanup rmdir "$target"
214 dd bs=512 count=2880 if=/dev/zero of="$img"
215 mkfs.ext2 -F "$img" > /dev/null
216 mount "$img" "$target" -o loop
217 set | egrep ^snf_export_\\w+= | sed -e 's/^snf_export_/export SNF_IMAGE_/' \
219 if [ -n "$UNATTEND" ]; then
220 if [ -f "$UNATTEND" ]; then
221 cat "$UNATTEND" > "$target/unattend.xml"
223 log_error "Unattend file: \`"$UNATTEND"' does not exist"
232 if [[ "$id" =~ ^pithos: ]]; then
234 elif [[ "$id" =~ ^(http|ftp)s?: ]]; then
235 if [ "$network_backend_support" = "yes" ]; then
238 log_error "Retrieving images from the network is not supported."
246 # this one is only to be called by create
248 if [ -z "$OS_API_VERSION" -o "$OS_API_VERSION" = "5" ]; then
250 GETOPT_RESULT=`getopt -o o:n:i:b:s: -n '$0' -- "$@"`
251 if [ $? != 0 ] ; then log_error "Terminating..."; exit 1 ; fi
252 get_api5_arguments $GETOPT_RESULT
253 elif [ "$OS_API_VERSION" = "10" -o "$OS_API_VERSION" = "15" ]; then
255 elif [ "$OS_API_VERSION" = "20" ]; then
258 IMAGE_TYPE="$IMG_FORMAT"
259 BACKEND_TYPE=$(get_backend_type $IMG_ID)
261 log_error "Unknown OS API VERSION $OS_API_VERSION"
265 if [ -n "$OS_VARIANT" ]; then
266 if [ ! -d "$VARIANTS_DIR" ]; then
267 log_error "OS Variants directory $VARIANTS_DIR doesn't exist"
270 VARIANT_CONFIG="$VARIANTS_DIR/$OS_VARIANT.conf"
271 if [ -f "$VARIANT_CONFIG" ]; then
274 if grep -qxF "$OS_VARIANT" variants.list; then
275 log_error "ERROR: instance-image configuration error"
276 log_error " Published variant $OS_VARIANT is missing its" \
278 log_error " Please create $VARIANT_CONFIG or unpublish the" \
280 log_error " (by removing $OS_VARIANT from variants.list)"
282 log_error "Unofficial variant $OS_VARIANT is unsupported"
283 log_error "Most probably this is a user error, forcing a" \
285 log_error "To support this variant please create file" \
297 echo "Debootstraping to create a new root filesystem:"
299 # Create a policy-rc.d file to deny init script execution
300 mkdir -p "$target/usr/sbin"
301 cat > "$target/usr/sbin/policy-rc.d" <<EOF
305 chmod +x "$target/usr/sbin/policy-rc.d"
307 debootstrap --arch $(dpkg --print-architecture) \
308 --include "$HELPER_EXTRA_PKGS" --variant=minbase stable "$target" \
309 "$HELPER_MIRROR" 2>&1 | sed -e 's/^/DEBOOTSTRAP: /g'
311 # Save the package list
312 chroot "$target" dpkg-query -W -f "\${Package}\n" > "$HELPER_CACHE_PKGS"
314 rm "$target/usr/sbin/policy-rc.d"
316 # remove the downloaded debs, as they are no longer needed
317 find "$target/var/cache/apt/archives" -type f -name '*.deb' -print0 | \
320 local tmp_cache=$(mktemp "$CACHE_FILE.XXXXXX")
321 tar cf "$tmp_cache" --one-file-system -C "$target" . || \
322 { rm "$tmp_cache"; false; }
323 # Overwrite the default cache file. Not the user specified if present.
324 mv -f "$tmp_cache" "$HELPER_CACHE_FILE"
327 report_and_cleanup(){
332 report_error "$err_file" "$fd"
337 # if something fails here, it souldn't call cleanup again...
340 if [ ${#CLEANUP[*]} -gt 0 ]; then
341 LAST_ELEMENT=$((${#CLEANUP[*]}-1))
342 REVERSE_INDEXES=$(seq ${LAST_ELEMENT} -1 0)
343 for i in $REVERSE_INDEXES; do
344 # If something fails here, it's better to retry it for a few times
345 # before we give up with an error. This is needed for kpartx when
346 # dealing with ntfs partitions mounted through fuse. umount is not
347 # synchronous and may return while the partition is still busy. A
348 # premature attempt to delete partition mappings through kpartx on
349 # a device that hosts previously mounted ntfs partition may fail
350 # with errors like this one:
351 # `device-mapper: remove ioctl failed: Device or resource busy'
352 # A sensible workaround for this is to wait for a while and then
354 local cmd=${CLEANUP[$i]}
355 $cmd || for interval in 0.25 0.5 1 2 4; do
356 echo "Command $cmd failed!"
357 echo "I'll wait for $interval secs and will retry..."
361 if [ "$?" != "0" ]; then
371 DEFAULT_FILE="@sysconfdir@/default/snf-image"
372 if [ -f "$DEFAULT_FILE" ]; then
376 : ${VARIANTS_DIR:="@sysconfdir@/ganeti/snf-image/variants"}
377 : ${IMAGE_DIR:="@localstatedir@/lib/snf-image"}
378 : ${HELPER_DIR:="@HELPER_DIR@"}
379 : ${HELPER_IMG:="@HELPER_IMG@"}
380 : ${HELPER_KERNEL:="@HELPER_KERNEL@"}
381 : ${HELPER_INITRD:="@HELPER_INITRD@"}
382 : ${HELPER_PKG:="@HELPER_DIR@/snf-image-helper.deb"}
383 : ${HELPER_SOFT_TIMEOUT:=20}
384 : ${HELPER_HARD_TIMEOUT:=5}
385 : ${HELPER_USER:="nobody"}
386 : ${HELPER_CACHE_FILE:="@HELPER_DIR@/cache.tar"}
387 : ${HELPER_CACHE_PKGS:="@HELPER_DIR@/packages"}
388 : ${HELPER_EXTRA_PKGS:="linux-image-amd64,e2fsprogs,ntfs-3g,ntfsprogs,xmlstarlet,python,parted,reglookup,chntpw,util-linux"}
389 : ${HELPER_MIRROR:=""}
390 : ${PITHOS_DB:="sqlite:////@localstatedir@/lib/pithos/backend.db"}
391 : ${PITHOS_DATA:="@localstatedir@/lib/pithos/data/"}
392 : ${PROGRESS_MONITOR:="@PROGRESS_MONITOR@"}
393 : ${UNATTEND:="@UNATTEND@"}
395 SCRIPT_NAME=$(basename $0)
397 # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :