set -e
-TEMP=`getopt -o i:b:s: -n '$0' -- "$@"`
+. common.sh
-if [ $? != 0 ] ; then echo "Terminating..." >&2 ; exit 1 ; fi
-
-# Note the quotes around `$TEMP': they are essential!
-eval set -- "$TEMP"
-
-while true; do
- case "$1" in
- -i) instance=$2; shift 2;;
-
- -b) blockdev=$2; shift 2;;
-
- -s) swapdev=$2; shift 2;;
-
- --) shift; break;;
-
- *) echo "Internal error!"; exit 1;;
- esac
-done
-if [ -z "$instance" -o -z "$blockdev" -o -z "$swapdev" ]; then
- echo "Missing -i or -b or -s argument!"
- exit 1
+mke2fs -Fjq $blockdev
+root_uuid=$($VOL_ID -u $blockdev )
+if [ -n "$swapdev" ]; then
+ mkswap $swapdev
+ swap_uuid=$($VOL_ID -u $swapdev )
fi
-mkswap $swapdev
-mke2fs -Fjq $blockdev
TMPDIR=`mktemp -d` || exit 1
trap "umount $TMPDIR; rmdir $TMPDIR" EXIT
-mount $blockdev $TMPDIR
+
+# If it's not a block device try to mount it via loopback device.
+# This is needed for file disks.
+MOUNT_OPTIONS=""
+if [ ! -b $blockdev ]; then
+ MOUNT_OPTIONS="$MOUNT_OPTIONS -o loop"
+fi
+mount $MOUNT_OPTIONS $blockdev $TMPDIR
+
( cd $TMPDIR; restore -r -y -f - )
+rm -f $TMPDIR/etc/udev/rules.d/z25_persistent-net.rules
+
+# Fix /etc/fstab with the new volumes' UUIDs
+if [ -e $TMPDIR/etc/fstab ]; then
+ ROOT_LINE="UUID=$root_uuid / ext3 defaults 0 1"
+ if [ -n "$swapdev" ]; then
+ SWAP_LINE="UUID=$swap_uuid swap swap defaults 0 0"
+ cat $TMPDIR/etc/fstab | \
+ sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
+ -re "s#^(/dev/sdb|UUID=[a-f0-9-]+)\s+swap\s+.*\$#$SWAP_LINE#" \
+ > $TMPDIR/etc/fstab.new
+ else
+ cat $TMPDIR/etc/fstab | \
+ sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
+ > $TMPDIR/etc/fstab.new
+ fi
+ mv $TMPDIR/etc/fstab.new $TMPDIR/etc/fstab
+fi
umount $TMPDIR
rmdir $TMPDIR