import

   1 #!/bin/bash
   2
   3 # Copyright (C) 2007, 2008, 2009 Google Inc.
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18 # 02110-1301, USA.
  19
  20 set -e
  21
  22 . common.sh
  23
  24 CLEANUP=( )
  25
  26 trap cleanup EXIT
  27
  28 # If the target device is not a real block device we'll first losetup it.
  29 # This is needed for file disks.
  30 if [ ! -b $blockdev ]; then
  31   ORIGINAL_BLOCKDEV=$blockdev
  32   blockdev=$(losetup -sf $blockdev)
  33   CLEANUP+=("losetup -d $blockdev")
  34 fi
  35
  36 if [ "$PARTITION_STYLE" = "none" ]; then
  37   filesystem_dev=$blockdev
  38 elif [ "$PARTITION_STYLE" = "msdos" ]; then
  39   # Create one big partition, and make it bootable
  40   format_disk0 $blockdev
  41   filesystem_dev=$(map_disk0 $blockdev)
  42   CLEANUP+=("unmap_disk0 $blockdev")
  43 else
  44   echo "Unknown partition style $PARTITION_STYLE"
  45   exit 1
  46 fi
  47
  48 mke2fs -Fjq $filesystem_dev
  49 root_uuid=$($VOL_ID -u $filesystem_dev )
  50
  51 if [ -n "$swapdev" ]; then
  52   mkswap $swapdev
  53   swap_uuid=$($VOL_ID -u $swapdev || true )
  54 fi
  55
  56 TMPDIR=`mktemp -d` || exit 1
  57 CLEANUP+=("rmdir $TMPDIR")
  58
  59 mount $filesystem_dev $TMPDIR
  60 CLEANUP+=("umount $TMPDIR")
  61
  62 ( cd $TMPDIR; restore -r -y -f - )
  63 rm -f $TMPDIR/etc/udev/rules.d/z*_persistent-net.rules
  64
  65 # Fix /etc/fstab with the new volumes' UUIDs
  66 if [ -e $TMPDIR/etc/fstab ]; then
  67   ROOT_LINE="UUID=$root_uuid  /     ext3  defaults  0  1"
  68   if [ -n "$swapdev" -a -n "$swap_uuid" ]; then
  69     SWAP_LINE="UUID=$swap_uuid  swap  swap  defaults  0  0"
  70     cat $TMPDIR/etc/fstab | \
  71       sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
  72           -e "s#^(/dev/sdb|UUID=[a-f0-9-]+)\s+swap\s+.*\$#$SWAP_LINE#" \
  73       > $TMPDIR/etc/fstab.new
  74   else
  75     cat $TMPDIR/etc/fstab | \
  76       sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
  77       > $TMPDIR/etc/fstab.new
  78   fi
  79   mv $TMPDIR/etc/fstab.new  $TMPDIR/etc/fstab
  80 fi
  81
  82 # execute cleanups
  83 cleanup
  84 trap - EXIT
  85
  86 exit 0