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 # If the target device is not a real block device we'll first losetup it.
  25 # This is needed for file disks.
  26 if [ ! -b $blockdev ]; then
  27   ORIGINAL_BLOCKDEV=$blockdev
  28   blockdev=$(losetup -sf $blockdev)
  29   CLEANUP+=("losetup -d $blockdev")
  30 fi
  31
  32 if [ "$PARTITION_STYLE" = "none" ]; then
  33   filesystem_dev=$blockdev
  34 elif [ "$PARTITION_STYLE" = "msdos" ]; then
  35   # Create one big partition, and make it bootable
  36   format_disk0 $blockdev
  37   filesystem_dev=$(map_disk0 $blockdev)
  38   CLEANUP+=("unmap_disk0 $blockdev")
  39 else
  40   echo "Unknown partition style $PARTITION_STYLE"
  41   exit 1
  42 fi
  43
  44 mke2fs -Fjq $filesystem_dev
  45 root_uuid=$($VOL_ID $filesystem_dev )
  46
  47 if [ -n "$swapdev" ]; then
  48   mkswap $swapdev
  49   swap_uuid=$($VOL_ID $swapdev || true )
  50 fi
  51
  52 TMPDIR=`mktemp -d` || exit 1
  53 CLEANUP+=("rmdir $TMPDIR")
  54
  55 mount $filesystem_dev $TMPDIR
  56 CLEANUP+=("umount $TMPDIR")
  57
  58 ( cd $TMPDIR; restore -r -y -f - )
  59 rm -f $TMPDIR/etc/udev/rules.d/z*_persistent-net.rules
  60
  61 # Fix /etc/fstab with the new volumes' UUIDs
  62 if [ -e $TMPDIR/etc/fstab ]; then
  63   ROOT_LINE="UUID=$root_uuid  /     ext3  defaults  0  1"
  64   if [ -n "$swapdev" -a -n "$swap_uuid" ]; then
  65     SWAP_LINE="UUID=$swap_uuid  swap  swap  defaults  0  0"
  66     cat $TMPDIR/etc/fstab | \
  67       sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
  68           -e "s#^(/dev/sdb|UUID=[a-f0-9-]+)\s+swap\s+.*\$#$SWAP_LINE#" \
  69       > $TMPDIR/etc/fstab.new
  70   else
  71     cat $TMPDIR/etc/fstab | \
  72       sed -re "s#^(/dev/sda|UUID=[a-f0-9-]+)\s+/\s+.*\$#$ROOT_LINE#" \
  73       > $TMPDIR/etc/fstab.new
  74   fi
  75   mv $TMPDIR/etc/fstab.new  $TMPDIR/etc/fstab
  76 fi
  77
  78 # execute cleanups
  79 cleanup
  80 trap - EXIT
  81
  82 exit 0