tools/livecd-iso-to-disk.sh

   1 #!/bin/bash
   2 # Transfer a Live image so that it's bootable off of a USB/SD device.
   3 # Copyright 2007-2012  Red Hat, Inc.
   4 #
   5 # Jeremy Katz <katzj@redhat.com>
   6 # Brian C. Lane <bcl@redhat.com>
   7 #
   8 # overlay/persistence enhancements by Douglas McClendon <dmc@viros.org>
   9 # GPT+MBR hybrid enhancements by Stewart Adam <s.adam@diffingo.com>
  10 #
  11 # This program is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; version 2 of the License.
  14 #
  15 # This program is distributed in the hope that it will be useful,
  16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 # GNU Library General Public License for more details.
  19 #
  20 # You should have received a copy of the GNU General Public License
  21 # along with this program; if not, write to the Free Software
  22 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  23
  24
  25 export PATH=/sbin:/usr/sbin:$PATH
  26
  27 shortusage() {
  28     echo "
  29     SYNTAX
  30
  31     livecd-iso-to-disk [--help] [--noverify] [--format] [--msdos] [--reset-mbr]
  32                        [--efi] [--skipcopy] [--force] [--xo] [--xo-no-home]
  33                        [--timeout <time>] [--totaltimeout <time>]
  34                        [--extra-kernel-args <args>] [--multi] [--livedir <dir>]
  35                        [--compress] [--skipcompress] [--swap-size-mb <size>]
  36                        [--overlay-size-mb <size>] [--home-size-mb <size>]
  37                        [--delete-home] [--crypted-home] [--unencrypted-home]
  38                        <source> <target device>
  39
  40     (Enter livecd-iso-to-disk --help on the command line for more information.)"
  41 }
  42
  43 usage() {
  44     echo "
  45     "
  46     shortusage
  47     echo "
  48     livecd-iso-to-disk  -  Transfer a LiveOS image so that it's bootable off of
  49                            a USB/SD device.
  50
  51     The script may be run in simplest form with just the two arguments:
  52
  53              <source>
  54                  This may be the filesystem path to a LiveOS .iso image file,
  55                  such as from a CD-ROM, DVD, or download.  It could also be the
  56                  device node reference for the mount point of another LiveOS
  57                  filesystem, including the currently-running one (such as a
  58                  booted Live CD/DVD/USB, where /dev/live references the running
  59                  image device).
  60
  61              <target device>
  62                  This should be the device partition name for the attached,
  63                  target device, such as /dev/sdb1 or /dev/sdc1.  (Issue the
  64                  df -Th command to get a listing of the mounted partitions,
  65                  where you can confirm the filesystem types, available space,
  66                  and device names.)  Be careful to specify the correct device,
  67                  or you may overwrite important data on another disk!
  68
  69     To execute the script to completion, you will need to run it with root user
  70     permissions.
  71     SYSLINUX must be installed on the computer running the installation script.
  72
  73     DESCRIPTION
  74
  75     livecd-iso-to-disk installs a Live CD/DVD/USB image (LiveOS) onto a USB/SD
  76     storage device (or any storage partition that will boot with a SYSLINUX
  77     bootloader).  The target storage device can then boot the installed
  78     operating system on systems that support booting via the USB or the SD
  79     interface.  The script requires a LiveOS source image and a target storage
  80     device.  The source image may be either a LiveOS .iso file, the currently-
  81     running LiveOS image, the device node reference for an attached device with
  82     an installed LiveOS image, or a file backed by a block device with an
  83     installed LiveOS image.  If the operating system supports persistent
  84     overlays for saving system changes, a pre-sized overlay may be included with
  85     the installation.
  86
  87     Unless you request the --format option, the installation does not destroy
  88     data outside of the LiveOS, syslinux, & EFI folders on your target device.
  89     This allows one to maintain other files on the target disk outside of the
  90     LiveOS filesystem.
  91
  92     LiveOS images provide embedded filesystems through the Device-mapper
  93     component of the Linux kernel.  The embedded filesystems exist within files
  94     such as /LiveOS/squashfs.img (the default compressed storage) or
  95     /LiveOS/ext3fs.img (an uncompressed version) on the primary volume partition
  96     of the storage device.  In use, these are read-only filesystems. Optionally,
  97     one may specify a persistent LiveOS overlay to hold image-change snapshots
  98     (that use write-once, difference-tracking storage) in the
  99     /LiveOS/overlay-<device_id> file, which, *one should note*, always grows in
 100     size due to the storage mechanism.  (The fraction of allocated space that
 101     has been consumed by system activity and changes may be displayed by issuing
 102     the 'dmsetup status' command in a terminal session of a running LiveOS
 103     image.)  One way to conserve the unrecoverable, overlay file space, is to
 104     specify a persistent home folder for user files, which will be saved in a
 105     /LiveOS/home.img filesystem image file.  This file space is encrypted by
 106     default, but is not compressed.  (One may bypass encryption with the
 107     --unencrypted-home installation option.)  Files in this home folder may be
 108     erased to recover and reuse their storage space.  The home.img file is also
 109     convenient for backing up or swapping user account files.
 110
 111     OPTIONS
 112
 113     --help
 114         Displays usage information and exits.
 115
 116     --noverify
 117         Disables the image validation process that occurs before the image is
 118         installed from the original Live CD .iso image.  When this option is
 119         specified, the image is not verified before loading onto the target
 120         storage device.
 121
 122     --format
 123         Formats the target device and creates an MS-DOS partition table (or GPT
 124         partition table, if the --efi option is passed).
 125
 126     --msdos
 127         Forces format to use msdos instead of ext4.
 128
 129     --reset-mbr
 130         Sets the Master Boot Record (MBR) of the target storage device to the
 131         mbr.bin file from the installation system's syslinux directory.  This
 132         may be helpful in recovering a damaged or corrupted device.
 133
 134     --efi
 135         Creates a GUID partition table when --format is passed, and installs a
 136         hybrid Extensible Firmware Interface (EFI)/MBR bootloader on the disk.
 137         This is necessary for most Intel Macs.
 138
 139     --skipcopy
 140         Skips the copying of the live image to the target device, bypassing the
 141         actions of the --format, --overlay-size-mb, --home-size-mb, &
 142         --swap-size-mb options, if present on the command line. (The --skipcopy
 143         option may be used while testing the script, in order to avoid repeated
 144         and lengthy copy commands, or to repair boot configuration files on a
 145         previously installed device.)
 146
 147     --force
 148         This option allows the installation script to bypass a delete
 149         confirmation dialog in the event that a pre-existing LiveOS directory
 150         is found on the target device.
 151
 152     --xo
 153         Used to prepare an image for the OLPC XO-1 laptop with its compressed,
 154         JFFS2 filesystem.  Do not use the following options with --xo:
 155             --overlay-size-mb <size>, home-size-mb <size>, --delete-home,
 156             --compress
 157
 158     --xo-no-home
 159         Used together with the --xo option to prepare an image for an OLPC XO
 160         laptop with the home folder on an SD card instead of the internal flash
 161         storage.
 162
 163     --timeout
 164         Modifies the bootloader's timeout value, which indicates how long to
 165         pause at the boot: prompt before booting automatically.  This overrides
 166         the value set during iso creation.  Units are 1/10 s.  The timeout is
 167         canceled when any key is pressed, the assumption being that the user
 168         will complete the command line.  A timeout of zero will disable the
 169         timeout completely.
 170
 171     --totaltimeout
 172         Adds a bootloader totaltimeout, which indicates how long to wait before
 173         booting automatically.  This is used to force an automatic boot.  This
 174         timeout cannot be canceled by the user.  Units are 1/10 s.
 175
 176     --extra-kernel-args <args>
 177         Specifies additional kernel arguments, <args>, that will be inserted
 178         into the syslinux and EFI boot configurations.  Multiple arguments
 179         should be specified in one string, i.e.,
 180             --extra-kernel-args \"arg1 arg2 ...\"
 181
 182     --multi
 183         Used when installing multiple image copies to signal configuration of
 184         the boot files for the image in the --livedir <dir> parameter.
 185
 186
 187     --livedir <dir>
 188         Used with multiple image installations to designate the directory <dir>
 189         for the particular image.
 190
 191     --compress   (default state for the operating system files)
 192         The default, compressed SquashFS filesystem image is copied on
 193         installation.  This option has no effect when the source filesystem is
 194         already expanded.
 195
 196     --skipcompress   (default option when  --xo is specified)
 197         Expands the source SquashFS image on installation into the read-only
 198         /LiveOS/ext3fs.img filesystem image file.
 199
 200     --swap-size-mb <size>
 201         Sets up a swap file of <size> mebibytes (integer values only) on the
 202         target device.
 203
 204     --overlay-size-mb <size>
 205         This option sets the overlay size in mebibytes (integer values only).
 206         The overlay makes persistent storage available to the live operating
 207         system, if the operating system supports it.  The persistent LiveOS
 208         overlay holds image-change snapshots (using write-once, difference-
 209         tracking  storage) in the /LiveOS/overlay-<device_id> file, which, *one
 210         should note*, always grows in size due to the storage mechanism.  (The
 211         fraction of allocated space that has been consumed may be displayed by
 212         issuing the 'dmsetup status' command in a terminal session of a running
 213         LiveOS installation.)  One way to conserve the unrecoverable, overlay
 214         file space, is to specify a persistent home folder for user files, see
 215         --home-size-mb below.  The target storage device must have enough free
 216         space for the image and the overlay.  A maximum <size> of 2047 MiB is
 217         permitted for vfat-formatted devices.  If there is insufficient room on
 218         your device, you will be given information to help in adjusting your
 219         settings.
 220
 221     --home-size-mb <size>
 222         Sets the home directory size in mebibytes (integer values only).  A
 223         persistent home directory will be made in the /LiveOS/home.img
 224         filesystem image file.  This file space is encrypted by default, but not
 225         compressed  (one may bypass encryption with the --unencrypted-home
 226         installation option).  Files in this home folder may be erased to
 227         recover and reuse their storage space.  The target storage device must
 228         have enough free space for the image, any overlay, and the home
 229         directory.  Note that the --delete-home option must also be selected to
 230         replace an existing persistent home with a new, empty one.  A maximum
 231         <size> of 2047 MiB is permitted for vfat-formatted devices.  If there is
 232         insufficient room on your device, you will be given information to help
 233         in adjusting your settings.
 234
 235     --delete-home
 236         To prevent unwitting deletion of user files, this option must be
 237         explicitly selected when the option --home-size-mb <size> is selected
 238         and there is an existing persistent home directory on the target device.
 239
 240     --crypted-home   (default that only applies to new home-size-mb requests)
 241         Specifies the default option to encrypt a new persistent home directory
 242         if --home-size-mb <size> is specified.
 243
 244     --unencrypted-home
 245         Prevents the default option to encrypt a new persistent home directory.
 246
 247     CONTRIBUTORS
 248
 249     livecd-iso-to-disk: David Zeuthen, Jeremy Katz, Douglas McClendon,
 250                         Chris Curran and other contributors.
 251                         (See the AUTHORS file in the source distribution for
 252                         the complete list of credits.)
 253
 254     BUGS
 255
 256     Report bugs to the mailing list
 257     http://admin.fedoraproject.org/mailman/listinfo/livecd or directly to
 258     Bugzilla http://bugzilla.redhat.com/bugzilla/ against the Fedora product,
 259     and the livecd-tools component.
 260
 261     COPYRIGHT
 262
 263     Copyright (C) Fedora Project 2008, 2009, 2010 and various contributors.
 264     This is free software. You may redistribute copies of it under the terms of
 265     the GNU General Public License http://www.gnu.org/licenses/gpl.html.
 266     There is NO WARRANTY, to the extent permitted by law.
 267
 268     SEE ALSO
 269
 270     livecd-creator, project website http://fedoraproject.org/wiki/FedoraLiveCD
 271     "
 272     exit 1
 273 }
 274
 275 cleanup() {
 276     sleep 2
 277     [ -d "$SRCMNT" ] && umount $SRCMNT && rmdir $SRCMNT
 278     [ -d "$TGTMNT" ] && umount $TGTMNT && rmdir $TGTMNT
 279 }
 280
 281 exitclean() {
 282     RETVAL=$?
 283     if [ -d "$SRCMNT" ] || [ -d "$TGTMNT" ];
 284     then
 285         [ "$RETVAL" = 0 ] || echo "Cleaning up to exit..."
 286         cleanup
 287     fi
 288     exit $RETVAL
 289 }
 290
 291 isdevloop() {
 292     [ x"${1#/dev/loop}" != x"$1" ]
 293 }
 294
 295 getdisk() {
 296     DEV=$1
 297
 298     if isdevloop "$DEV"; then
 299         device="$DEV"
 300         return
 301     fi
 302
 303     p=$(udevadm info -q path -n $DEV)
 304     if [ $? -gt 0 ]; then
 305         echo "Error getting udev path to $DEV"
 306         exitclean
 307     fi
 308     if [ -e /sys/$p/device ]; then
 309         device=$(basename /sys/$p)
 310     else
 311         device=$(basename $(readlink -f /sys/$p/../))
 312     fi
 313     if [ ! -e /sys/block/$device -o ! -e /dev/$device ]; then
 314         echo "Error finding block device of $DEV.  Aborting!"
 315         exitclean
 316     fi
 317
 318     device="/dev/$device"
 319     # FIXME: weird dev names could mess this up I guess
 320     p=/dev/$(basename $p)
 321     partnum=${p##$device}
 322 }
 323
 324 resetMBR() {
 325     if isdevloop "$DEV"; then
 326         return
 327     fi
 328     getdisk $1
 329     # if efi, we need to use the hybrid MBR
 330     if [ -n "$efi" ]; then
 331         if [ -f /usr/lib/syslinux/gptmbr.bin ]; then
 332             cat /usr/lib/syslinux/gptmbr.bin > $device
 333         elif [ -f /usr/share/syslinux/gptmbr.bin ]; then
 334             cat /usr/share/syslinux/gptmbr.bin > $device
 335         else
 336             echo "Could not find gptmbr.bin (syslinux)"
 337             exitclean
 338         fi
 339         # Make it bootable on EFI and BIOS
 340         parted -s $device set $partnum legacy_boot on
 341     else
 342         if [ -f /usr/lib/syslinux/mbr.bin ]; then
 343             cat /usr/lib/syslinux/mbr.bin > $device
 344         elif [ -f /usr/share/syslinux/mbr.bin ]; then
 345             cat /usr/share/syslinux/mbr.bin > $device
 346         else
 347             echo "Could not find mbr.bin (syslinux)"
 348             exitclean
 349         fi
 350     fi
 351 }
 352
 353 checkMBR() {
 354     if isdevloop "$DEV"; then
 355         return 0
 356     fi
 357     getdisk $1
 358
 359     bs=$(mktemp /tmp/bs.XXXXXX)
 360     dd if=$device of=$bs bs=512 count=1 2>/dev/null || exit 2
 361
 362     mbrword=$(hexdump -n 2 $bs |head -n 1|awk {'print $2;'})
 363     rm -f $bs
 364     if [ "$mbrword" = "0000" ]; then
 365         if [ -z "$format" ]; then
 366             echo "MBR appears to be blank."
 367             echo "Press Enter to replace the MBR and continue or ctrl-c to abort"
 368             read
 369         fi
 370         resetMBR $1
 371     fi
 372
 373     return 0
 374 }
 375
 376 checkPartActive() {
 377     dev=$1
 378     getdisk $dev
 379
 380     # if we're installing to whole-disk and not a partition, then we
 381     # don't need to worry about being active
 382     if [ "$dev" = "$device" ]; then
 383         return
 384     fi
 385     if isdevloop "$DEV"; then
 386         return
 387     fi
 388
 389     if [ "$(/sbin/fdisk -l $device 2>/dev/null |grep -m1 $dev |awk {'print $2;'})" != "*" ]; then
 390         echo "Partition isn't marked bootable!"
 391         echo "You can mark the partition as bootable with "
 392         echo "    # /sbin/parted $device"
 393         echo "    (parted) toggle N boot"
 394         echo "    (parted) quit"
 395         exitclean
 396     fi
 397 }
 398
 399 checkLVM() {
 400     dev=$1
 401
 402     if [ -x /sbin/pvs -a \
 403         "$(/sbin/pvs -o vg_name --noheadings $dev* 2>/dev/null || :)" ]; then
 404         echo "Device, $dev, contains a volume group and cannot be formated!"
 405         echo "You can remove the volume group using vgremove."
 406         exitclean
 407     fi
 408     return 0
 409 }
 410
 411 createGPTLayout() {
 412     dev=$1
 413     getdisk $dev
 414
 415     echo "WARNING: THIS WILL DESTROY ANY DATA ON $device!!!"
 416     echo "Press Enter to continue or ctrl-c to abort"
 417     read
 418     umount ${device}* &> /dev/null || :
 419     wipefs -a ${device}
 420     /sbin/parted --script $device mklabel gpt
 421     partinfo=$(LC_ALL=C /sbin/parted --script -m $device "unit MB print" |grep ^$device:)
 422     dev_size=$(echo $partinfo |cut -d : -f 2 |sed -e 's/MB$//')
 423     p1_size=$(($dev_size - 3))
 424
 425     if [ $p1_size -le 0 ]; then
 426         echo "Your device isn't big enough to hold $SRC"
 427         echo "It is $(($p1_size * -1)) MB too small"
 428         exitclean
 429     fi
 430     p1_start=1
 431     p1_end=$(($p1_size + 1))
 432     /sbin/parted -s $device u MB mkpart '"EFI System Partition"' fat32 $p1_start $p1_end set 1 boot on
 433     # Sometimes automount can be _really_ annoying.
 434     echo "Waiting for devices to settle..."
 435     /sbin/udevadm settle
 436     sleep 5
 437     TGTDEV=${device}1
 438     umount $TGTDEV &> /dev/null || :
 439     /sbin/mkdosfs -n LIVE $TGTDEV
 440     TGTLABEL="UUID=$(/sbin/blkid -s UUID -o value $TGTDEV)"
 441 }
 442
 443 createMSDOSLayout() {
 444     dev=$1
 445     getdisk $dev
 446
 447     echo "WARNING: THIS WILL DESTROY ANY DATA ON $device!!!"
 448     echo "Press Enter to continue or ctrl-c to abort"
 449     read
 450     umount ${device}* &> /dev/null || :
 451     wipefs -a ${device}
 452     /sbin/parted --script $device mklabel msdos
 453     partinfo=$(LC_ALL=C /sbin/parted --script -m $device "unit MB print" |grep ^$device:)
 454     dev_size=$(echo $partinfo |cut -d : -f 2 |sed -e 's/MB$//')
 455     p1_size=$(($dev_size - 3))
 456
 457     if [ $p1_size -le 0 ]; then
 458         echo "Your device isn't big enough to hold $SRC"
 459         echo "It is $(($p1_size * -1)) MB too small"
 460         exitclean
 461     fi
 462     p1_start=1
 463     p1_end=$(($p1_size + 1))
 464     /sbin/parted -s $device u MB mkpart primary fat32 $p1_start $p1_end set 1 boot on
 465     # Sometimes automount can be _really_ annoying.
 466     echo "Waiting for devices to settle..."
 467     /sbin/udevadm settle
 468     sleep 5
 469     if ! isdevloop "$DEV"; then
 470         TGTDEV=${device}1
 471     else
 472         TGTDEV=${device}
 473     fi
 474     umount $TGTDEV &> /dev/null || :
 475     /sbin/mkdosfs -n LIVE $TGTDEV
 476     TGTLABEL="UUID=$(/sbin/blkid -s UUID -o value $TGTDEV)"
 477 }
 478
 479 createEXTFSLayout() {
 480     dev=$1
 481     getdisk $dev
 482
 483     echo "WARNING: THIS WILL DESTROY ANY DATA ON $device!!!"
 484     echo "Press Enter to continue or ctrl-c to abort"
 485     read
 486     umount ${device}* &> /dev/null || :
 487     wipefs -a ${device}
 488     /sbin/parted -s $device mklabel msdos
 489     partinfo=$(LC_ALL=C /sbin/parted -s -m $device "u MB print" |grep ^$device:)
 490     dev_size=$(echo $partinfo |cut -d : -f 2 |sed -e 's/MB$//')
 491     p1_size=$(($dev_size - 3))
 492
 493     if [ $p1_size -le 0 ]; then
 494         echo "Your device isn't big enough to hold $SRC"
 495         echo "It is $(($p1_size * -1)) MB too small"
 496         exitclean
 497     fi
 498     p1_start=1
 499     p1_end=$(($p1_size + 1))
 500     /sbin/parted -s $device u MB mkpart primary ext2 $p1_start $p1_end set 1 boot on
 501     # Sometimes automount can be _really_ annoying.
 502     echo "Waiting for devices to settle..."
 503     /sbin/udevadm settle
 504     sleep 5
 505     TGTDEV=${device}1
 506     umount $TGTDEV &> /dev/null || :
 507
 508     # Check extlinux version
 509     if extlinux -v 2>&1 | grep -q 'extlinux 3'; then
 510         mkfs=/sbin/mkfs.ext3
 511     else
 512         mkfs=/sbin/mkfs.ext4
 513     fi
 514     $mkfs -L LIVE $TGTDEV
 515     TGTLABEL="UUID=$(/sbin/blkid -s UUID -o value $TGTDEV)"
 516 }
 517
 518 checkGPT() {
 519     dev=$1
 520     getdisk $dev
 521
 522     if [ "$(/sbin/fdisk -l $device 2>/dev/null |grep -c GPT)" -eq "0" ]; then
 523         echo "EFI boot requires a GPT partition table."
 524         echo "This can be done manually or you can run with --format"
 525         exitclean
 526     fi
 527
 528     partinfo=$(LC_ALL=C /sbin/parted --script -m $device "print" |grep ^$partnum:)
 529     volname=$(echo $partinfo |cut -d : -f 6)
 530     flags=$(echo $partinfo |cut -d : -f 7)
 531     if [ "$volname" != "EFI System Partition" ]; then
 532         echo "Partition name must be 'EFI System Partition'"
 533         echo "This can be set in parted or you can run with --reset-mbr"
 534         exitclean
 535     fi
 536     if [ "$(echo $flags |grep -c boot)" = "0" ]; then
 537         echo "Partition isn't marked bootable!"
 538         echo "You can mark the partition as bootable with "
 539         echo "    # /sbin/parted $device"
 540         echo "    (parted) toggle N boot"
 541         echo "    (parted) quit"
 542         exitclean
 543     fi
 544 }
 545
 546 checkFilesystem() {
 547     dev=$1
 548
 549     TGTFS=$(/sbin/blkid -s TYPE -o value $dev || :)
 550     if [ "$TGTFS" != "vfat" ] && [ "$TGTFS" != "msdos" ]; then
 551         if [ "$TGTFS" != "ext2" ] && [ "$TGTFS" != "ext3" ] && [ "$TGTFS" != "ext4" ] && [ "$TGTFS" != "btrfs" ]; then
 552             echo "Target filesystem ($dev:$TGTFS) must be vfat, ext[234] or btrfs"
 553             exitclean
 554         fi
 555     fi
 556
 557     TGTLABEL=$(/sbin/blkid -s LABEL -o value $dev)
 558     if [ "$TGTLABEL" != "LIVE" ]; then
 559         if [ "$TGTFS" = "vfat" -o "$TGTFS" = "msdos" ]; then
 560             /sbin/dosfslabel $dev LIVE
 561             if [ $? -gt 0 ]; then
 562                 echo "dosfslabel failed on $dev, device not setup"
 563                 exitclean
 564             fi
 565         elif [ "$TGTFS" = "ext2" -o "$TGTFS" = "ext3" -o "$TGTFS" = "ext4" ]; then
 566             /sbin/e2label $dev LIVE
 567             if [ $? -gt 0 ]; then
 568                 echo "e2label failed on $dev, device not setup"
 569                 exitclean
 570             fi
 571         else
 572             echo "Unknown filesystem type. Try setting its label to LIVE and re-running"
 573             exitclean
 574         fi
 575     fi
 576
 577     # Use UUID if available
 578     TGTUUID=$(/sbin/blkid -s UUID -o value $dev)
 579     if [ -n "$TGTUUID" ]; then
 580         TGTLABEL="UUID=$TGTUUID"
 581     else
 582         TGTLABEL="LABEL=LIVE"
 583     fi
 584
 585     if [ "$TGTFS" = "vfat" -o "$TGTFS" = "msdos" ]; then
 586         mountopts="-o shortname=winnt,umask=0077"
 587     fi
 588 }
 589
 590 checkSyslinuxVersion() {
 591     if [ ! -x /usr/bin/syslinux ]; then
 592         echo "You need to have syslinux installed to run this script"
 593         exit 1
 594     fi
 595     check=($(syslinux --version 2>&1)) || :
 596     if [[ 'syslinux' != $check ]]; then
 597         SYSLINUXPATH=""
 598     elif [ -n "$multi" ]; then
 599         SYSLINUXPATH="$LIVEOS/syslinux"
 600     else
 601         SYSLINUXPATH="syslinux"
 602     fi
 603 }
 604
 605 checkMounted() {
 606     dev=$1
 607     if grep -q "^$dev " /proc/mounts ; then
 608         echo "$dev is mounted, please unmount for safety"
 609         exitclean
 610     fi
 611     if grep -q "^$dev " /proc/swaps; then
 612         echo "$dev is in use as a swap device, please disable swap"
 613         exitclean
 614     fi
 615 }
 616
 617 checkint() {
 618     if ! test $1 -gt 0 2>/dev/null ; then
 619         shortusage
 620         exit 1
 621     fi
 622 }
 623
 624 if [ $(id -u) != 0 ]; then
 625     echo "You need to be root to run this script"
 626     exit 1
 627 fi
 628
 629 detectsrctype() {
 630     if [[ -e "$SRCMNT/Packages" ]]; then
 631         echo "/Packages found, will copy source packages to target"
 632         packages=1
 633     fi
 634     if [[ -e "$SRCMNT/LiveOS/squashfs.img" ]]; then
 635         # LiveOS style boot image
 636         srctype=live
 637         return
 638     fi
 639     if [ -e $SRCMNT/images/install.img -o -e $SRCMNT/isolinux/initrd.img ]; then
 640         if [ -n "$packages" ]; then
 641             srctype=installer
 642         else
 643             srctype=netinst
 644         fi
 645         imgtype=install
 646         if [ ! -e $SRCMNT/images/install.img ]; then
 647             echo "$SRC uses initrd.img w/o install.img"
 648             imgtype=initrd
 649         fi
 650         return
 651     fi
 652     echo "ERROR: $SRC does not appear to be a Live image or DVD installer."
 653     exitclean
 654 }
 655
 656 cp_p() {
 657     strace -q -ewrite cp -- "${1}" "${2}" 2>&1 \
 658         | awk '{
 659         count += $NF
 660         if (count % 10 == 0) {
 661             percent = count / total_size * 100
 662             printf "%3d%% [", percent
 663             for (i=0;i<=percent;i++)
 664                 printf "="
 665                 printf ">"
 666             for (i=percent;i<100;i++)
 667                 printf " "
 668                 printf "]\r"
 669             }
 670         }
 671         END { print "" }' total_size=$(stat -c '%s' "${1}") count=0
 672 }
 673
 674 copyFile() {
 675     if [ -x /usr/bin/rsync ]; then
 676         rsync --inplace -P "$1" "$2"
 677         return
 678     fi
 679     if [ -x /usr/bin/gvfs-copy ]; then
 680         gvfs-copy -p "$1" "$2"
 681         return
 682     fi
 683     if [ -x /usr/bin/strace -a -x /bin/awk ]; then
 684         cp_p "$1" "$2"
 685         return
 686     fi
 687     cp "$1" "$2"
 688 }
 689
 690 set -e
 691 set -o pipefail
 692 trap exitclean EXIT
 693 shopt -s extglob
 694
 695 cryptedhome=1
 696 keephome=1
 697 homesizemb=0
 698 swapsizemb=0
 699 overlaysizemb=0
 700 srctype=
 701 imgtype=
 702 packages=
 703 LIVEOS=LiveOS
 704 HOMEFILE="home.img"
 705
 706 if [[ "$*" =~ "--help" ]]; then
 707     usage
 708 fi
 709 while [ $# -gt 2 ]; do
 710     case $1 in
 711         --help)
 712             usage
 713             ;;
 714         --noverify)
 715             noverify=1
 716             ;;
 717         --format)
 718             format=1
 719             ;;
 720         --msdos)
 721             usemsdos=1
 722             ;;
 723         --reset-mbr|--resetmbr)
 724             resetmbr=1
 725             ;;
 726         --efi|--mactel)
 727             efi=1
 728             ;;
 729         --skipcopy)
 730             skipcopy=1
 731             ;;
 732         --force)
 733             force=1
 734             ;;
 735         --xo)
 736             xo=1
 737             skipcompress=1
 738             ;;
 739         --xo-no-home)
 740             xonohome=1
 741             ;;
 742         --timeout)
 743             checkint $2
 744             timeout=$2
 745             shift
 746             ;;
 747         --totaltimeout)
 748             checkint $2
 749             totaltimeout=$2
 750             shift
 751             ;;
 752         --extra-kernel-args)
 753             kernelargs=$2
 754             shift
 755             ;;
 756         --multi)
 757             multi=1
 758             ;;
 759         --livedir)
 760             LIVEOS=$2
 761             shift
 762             ;;
 763         --compress)
 764             skipcompress=""
 765             ;;
 766         --skipcompress)
 767             skipcompress=1
 768             ;;
 769         --swap-size-mb)
 770             checkint $2
 771             swapsizemb=$2
 772             shift
 773             ;;
 774         --overlay-size-mb)
 775             checkint $2
 776             overlaysizemb=$2
 777             shift
 778             ;;
 779         --home-size-mb)
 780             checkint $2
 781             homesizemb=$2
 782             shift
 783             ;;
 784         --crypted-home)
 785             cryptedhome=1
 786             ;;
 787         --unencrypted-home)
 788             cryptedhome=""
 789             ;;
 790         --delete-home)
 791             keephome=""
 792             ;;
 793         *)
 794             echo "invalid arg -- $1"
 795             shortusage
 796             exit 1
 797             ;;
 798     esac
 799     shift
 800 done
 801
 802 SRC=$(readlink -f "$1")
 803 TGTDEV=$(readlink -f "$2")
 804
 805 if [ -z "$SRC" ]; then
 806     echo "Missing source"
 807     shortusage
 808     exit 1
 809 fi
 810
 811 if [ ! -b "$SRC" -a ! -f "$SRC" ]; then
 812     echo "$SRC is not a file or block device"
 813     shortusage
 814     exit 1
 815 fi
 816
 817 if [ -z "$TGTDEV" ]; then
 818     echo "Missing target device"
 819     shortusage
 820     exit 1
 821 fi
 822
 823 if [ ! -b "$TGTDEV" ]; then
 824     echo "$TGTDEV is not a block device"
 825     shortusage
 826     exit 1
 827 fi
 828
 829 if [ -z "$noverify" ]; then
 830     # verify the image
 831     echo "Verifying image..."
 832     if !  checkisomd5 --verbose "$SRC"; then
 833         echo "Are you SURE you want to continue?"
 834         echo "Press Enter to continue or ctrl-c to abort"
 835         read
 836     fi
 837 fi
 838
 839 # do some basic sanity checks.
 840 checkMounted $TGTDEV
 841
 842 # FIXME: would be better if we had better mountpoints
 843 SRCMNT=$(mktemp -d /media/srctmp.XXXXXX)
 844 if [ -b "$SRC" ]; then
 845     mount -o ro "$SRC" $SRCMNT || exitclean
 846 elif [ -f "$SRC" ]; then
 847     mount -o loop,ro "$SRC" $SRCMNT || exitclean
 848 else
 849     echo "$SRC is not a file or block device."
 850     exitclean
 851 fi
 852 # Figure out what needs to be done based on the source image
 853 detectsrctype
 854
 855 # Format the device
 856 if [ -n "$format" -a -z "$skipcopy" ]; then
 857     checkLVM $TGTDEV
 858
 859     if [ -n "$efi" ]; then
 860         createGPTLayout $TGTDEV
 861     elif [ -n "$usemsdos" -o ! -x /sbin/extlinux ]; then
 862         createMSDOSLayout $TGTDEV
 863     else
 864         createEXTFSLayout $TGTDEV
 865     fi
 866 fi
 867
 868 checkFilesystem $TGTDEV
 869 if [ -n "$efi" ]; then
 870     checkGPT $TGTDEV
 871 fi
 872
 873 checkSyslinuxVersion
 874 # Because we can't set boot flag for EFI Protective on msdos partition tables
 875 [ -z "$efi" ] && checkPartActive $TGTDEV
 876 [ -n "$resetmbr" ] && resetMBR $TGTDEV
 877 checkMBR $TGTDEV
 878
 879
 880 if [ "$overlaysizemb" -gt 0 ]; then
 881     if [ "$TGTFS" = "vfat" -a "$overlaysizemb" -gt 2047 ]; then
 882         echo "Can't have an overlay of 2048MB or greater on VFAT"
 883         exitclean
 884     fi
 885     LABEL=$(/sbin/blkid -s LABEL -o value $TGTDEV)
 886     if [[ "$LABEL" =~ ( ) ]]; then
 887         echo "The LABEL($LABEL) on $TGTDEV has spaces in it, which do not work with the overlay"
 888         echo "You can re-format or use dosfslabel/e2fslabel to change it"
 889         exitclean
 890     fi
 891 fi
 892
 893 if [ "$homesizemb" -gt 0 -a "$TGTFS" = "vfat" ]; then
 894     if [ "$homesizemb" -gt 2047 ]; then
 895         echo "Can't have a home overlay greater than 2048MB on VFAT"
 896         exitclean
 897     fi
 898 fi
 899
 900 if [ "$swapsizemb" -gt 0 -a "$TGTFS" = "vfat" ]; then
 901     if [ "$swapsizemb" -gt 2047 ]; then
 902         echo "Can't have a swap file greater than 2048MB on VFAT"
 903         exitclean
 904     fi
 905 fi
 906
 907 TGTMNT=$(mktemp -d /media/tgttmp.XXXXXX)
 908 mount $mountopts $TGTDEV $TGTMNT || exitclean
 909
 910 trap exitclean SIGINT SIGTERM
 911
 912 if [ -f "$TGTMNT/$LIVEOS/$HOMEFILE" -a -n "$keephome" -a "$homesizemb" -gt 0 ]; then
 913     echo "ERROR: Requested keeping existing /home and specified a size for /home"
 914     echo "Please either don't specify a size or specify --delete-home"
 915     exitclean
 916 fi
 917
 918 if [ -n "$efi" ]; then
 919     if [ -d $SRCMNT/EFI/BOOT ]; then
 920         EFI_BOOT="/EFI/BOOT"
 921     elif [ -d $SRCMNT/EFI/boot ]; then
 922         EFI_BOOT="/EFI/boot"
 923     else
 924         echo "ERROR: This live image does not support EFI booting"
 925         exitclean
 926     fi
 927 fi
 928
 929 # let's try to make sure there's enough room on the target device
 930 if [[ -d $TGTMNT/$LIVEOS ]]; then
 931     tbd=($(du -B 1M $TGTMNT/$LIVEOS))
 932     if [[ -s $TGTMNT/$LIVEOS/$HOMEFILE ]] && [[ -n $keephome ]]; then
 933         homesize=($(du -B 1M $TGTMNT/$LIVEOS/$HOMEFILE))
 934         tbd=$((tbd - homesize))
 935     fi
 936 else
 937     tbd=0
 938 fi
 939
 940 if [[ live == $srctype ]]; then
 941    targets="$TGTMNT/$SYSLINUXPATH"
 942    [[ -n $efi ]] && targets+=" $TGTMNT$EFI_BOOT"
 943    [[ -n $xo ]] && targets+=" $TGTMNT/boot/olpc.fth"
 944    duTable=($(du -c -B 1M $targets 2> /dev/null || :))
 945    tbd=$((tbd + ${duTable[*]: -2:1}))
 946 fi
 947
 948 if [[ -n $skipcompress ]] && [[ -s $SRCMNT/LiveOS/squashfs.img ]]; then
 949     if mount -o loop $SRCMNT/LiveOS/squashfs.img $SRCMNT; then
 950         livesize=($(du -B 1M --apparent-size $SRCMNT/LiveOS/ext3fs.img))
 951         umount $SRCMNT
 952         if ((livesize > 2048)) &&  [[ vfat == $TGTFS ]]; then
 953             echo "
 954             An uncompressed image size greater than 2048 MB is not suitable
 955             for a VFAT-formatted device.  The compressed SquashFS will be
 956             copied to the target device.
 957             "
 958             skipcompress=""
 959             livesize=0
 960         fi
 961     else
 962         echo "WARNING: --skipcompress or --xo was specified but the
 963         currently-running kernel can not mount the SquashFS from the source
 964         file to extract it. Instead, the compressed SquashFS will be copied
 965         to the target device."
 966         skipcompress=""
 967     fi
 968 fi
 969 if [[ live == $srctype ]]; then
 970     thisScriptpath=$(readlink -f "$0")
 971     sources="$SRCMNT/LiveOS/ext3fs.img $SRCMNT/LiveOS/osmin.img"
 972     [[ -z $skipcompress ]] && sources+=" $SRCMNT/LiveOS/squashfs.img"
 973     sources+=" $SRCMNT/isolinux $SRCMNT/syslinux"
 974     [[ -n $efi ]] && sources+=" $SRCMNT$EFI_BOOT"
 975     [[ -n $xo ]] && sources+=" $SRCMNT/boot/olpc.fth"
 976     duTable=($(du -c -B 1M "$thisScriptpath" $sources 2> /dev/null || :))
 977     livesize=$((livesize + ${duTable[*]: -2:1}))
 978 fi
 979
 980 freespace=($(df -B 1M --total $TGTDEV))
 981 freespace=${freespace[*]: -2:1}
 982
 983 if [[ live == $srctype ]]; then
 984     tba=$((overlaysizemb + homesizemb + livesize + swapsizemb))
 985     if ((tba > freespace + tbd)); then
 986         needed=$((tba - freespace - tbd))
 987         printf "\n  The live image + overlay, home, & swap space, if requested,
 988         \r  will NOT fit in the space available on the target device.\n
 989         \r  + Size of live image: %10s  MiB\n" $livesize
 990         (($overlaysizemb > 0)) && \
 991             printf "  + Overlay size: %16s\n" $overlaysizemb
 992         (($homesizemb > 0)) && \
 993             printf "  + Home directory size: %9s\n" $homesizemb
 994         (($swapsizemb > 0)) && \
 995             printf "  + Swap overlay size: %11s\n" $swapsizemb
 996         printf "  = Total requested space:  %6s  MiB\n" $tba
 997         printf "  - Space available:  %12s\n" $((freespace + tbd))
 998         printf "    ==============================\n"
 999         printf "    Space needed:  %15s  MiB\n\n" $needed
1000         printf "  To fit the installation on this device,
1001         \r  free space on the target, or decrease the
1002         \r  requested size total by:  %6s  MiB\n\n" $needed
1003         exitclean
1004     fi
1005 fi
1006
1007 # Verify available space for DVD installer
1008 if [ "$srctype" = "installer" ]; then
1009     if [ "$imgtype" = "install" ]; then
1010         imgpath=images/install.img
1011     else
1012         imgpath=isolinux/initrd.img
1013     fi
1014     installimgsize=$(du -s -B 1M $SRCMNT/$imgpath | awk {'print $1;'})
1015
1016     tbd=0
1017     if [ -e $TGTMNT/$imgpath ]; then
1018         tbd=$(du -s -B 1M $TGTMNT/$imgpath | awk {'print $1;'})
1019     fi
1020     if [ -e "$TGTMNT/$(basename "$SRC")" ]; then
1021         tbd=$(($tbd + $(du -s -B 1M "$TGTMNT/$(basename "$SRC")" | awk {'print $1;'})))
1022     fi
1023     echo "Size of $imgpath: $installimgsize"
1024     echo "Available space: $((freespace + tbd))"
1025     if (( installimgsize > ((freespace + tbd)) )); then
1026         echo "ERROR: Unable to fit DVD image + install.img on available space on the target device."
1027         exitclean
1028     fi
1029 fi
1030
1031 if [ -z "$skipcopy" ] && [ "$srctype" = "live" ]; then
1032     if [ -d $TGTMNT/$LIVEOS -a -z "$force" ]; then
1033         echo "Already set up as live image."
1034         if [ -z "$keephome" -a -e $TGTMNT/$LIVEOS/$HOMEFILE ]; then
1035             echo "WARNING: Persistent /home will be deleted!!!"
1036             echo "Press Enter to continue or ctrl-c to abort"
1037             read
1038         else
1039             echo "Deleting old OS in fifteen seconds..."
1040             sleep 15
1041
1042             [ -e "$TGTMNT/$LIVEOS/$HOMEFILE" -a -n "$keephome" ] && mv $TGTMNT/$LIVEOS/$HOMEFILE $TGTMNT/$HOMEFILE
1043         fi
1044
1045         rm -rf $TGTMNT/$LIVEOS
1046     fi
1047 fi
1048
1049 # Bootloader is always reconfigured, so keep these out of the if skipcopy stuff.
1050 [ ! -d $TGTMNT/$SYSLINUXPATH ] && mkdir -p $TGTMNT/$SYSLINUXPATH
1051 [ -n "$efi" -a ! -d $TGTMNT$EFI_BOOT ] && mkdir -p $TGTMNT$EFI_BOOT
1052
1053 # Live image copy
1054 if [ "$srctype" = "live" -a -z "$skipcopy" ]; then
1055     echo "Copying live image to target device."
1056     [ ! -d $TGTMNT/$LIVEOS ] && mkdir $TGTMNT/$LIVEOS
1057     [ -n "$keephome" -a -f "$TGTMNT/$HOMEFILE" ] && mv $TGTMNT/$HOMEFILE $TGTMNT/$LIVEOS/$HOMEFILE
1058     if [ -n "$skipcompress" -a -f $SRCMNT/LiveOS/squashfs.img ]; then
1059         mount -o loop $SRCMNT/LiveOS/squashfs.img $SRCMNT || exitclean
1060         copyFile $SRCMNT/LiveOS/ext3fs.img $TGTMNT/$LIVEOS/ext3fs.img || {
1061             umount $SRCMNT ; exitclean ; }
1062         umount $SRCMNT
1063     elif [ -f $SRCMNT/LiveOS/squashfs.img ]; then
1064         copyFile $SRCMNT/LiveOS/squashfs.img $TGTMNT/$LIVEOS/squashfs.img || exitclean
1065     elif [ -f $SRCMNT/LiveOS/ext3fs.img ]; then
1066         copyFile $SRCMNT/LiveOS/ext3fs.img $TGTMNT/$LIVEOS/ext3fs.img || exitclean
1067     fi
1068     if [ -f $SRCMNT/LiveOS/osmin.img ]; then
1069         copyFile $SRCMNT/LiveOS/osmin.img $TGTMNT/$LIVEOS/osmin.img || exitclean
1070     fi
1071     sync
1072 fi
1073
1074 # Adjust syslinux sources for replication of installed images
1075 # between filesystem types.
1076 if [[ -d $SRCMNT/isolinux/ ]]; then
1077     cp $SRCMNT/isolinux/* $TGTMNT/$SYSLINUXPATH
1078 elif [[ -d $SRCMNT/syslinux/ ]]; then
1079     cp $SRCMNT/syslinux/* $TGTMNT/$SYSLINUXPATH
1080     if [[ -f $SRCMNT/syslinux/extlinux.conf ]]; then
1081         mv $TGTMNT/$SYSLINUXPATH/extlinux.conf \
1082             $TGTMNT/$SYSLINUXPATH/isolinux.cfg
1083     elif [[ -f $SRCMNT/syslinux/syslinux.cfg ]]; then
1084         mv $TGTMNT/$SYSLINUXPATH/syslinux.cfg $TGTMNT/$SYSLINUXPATH/isolinux.cfg
1085     fi
1086 fi
1087 BOOTCONFIG=$TGTMNT/$SYSLINUXPATH/isolinux.cfg
1088 # Set this to nothing so sed doesn't care
1089 BOOTCONFIG_EFI=
1090 if [ -n "$efi" ]; then
1091     echo "Setting up $EFI_BOOT"
1092     cp $SRCMNT$EFI_BOOT/* $TGTMNT$EFI_BOOT
1093
1094     # FIXME
1095     # There is a problem here. On older LiveCD's the files are boot?*.conf
1096     # They really should be renamed to BOOT?*.conf
1097
1098     # this is a little ugly, but it gets the "interesting" named config file
1099     BOOTCONFIG_EFI=$TGTMNT$EFI_BOOT/+(BOOT|boot)?*.conf
1100     rm -f $TGTMNT$EFI_BOOT/grub.conf
1101
1102     # On some images (RHEL) the BOOT*.efi file isn't in $EFI_BOOT, but is in
1103     # the eltorito image, so try to extract it if it is missing
1104
1105     # test for presence of *.efi grub binary
1106     if [ ! -f $TGTMNT$EFI_BOOT/+(BOOT|boot)?*.efi ]; then
1107         if [ ! -x /usr/bin/dumpet ]; then
1108             echo "No /usr/bin/dumpet tool found. EFI image will not boot."
1109             echo "Source media is missing grub binary in /EFI/BOOT/*efi"
1110             exitclean
1111         else
1112             # dump the eltorito image with dumpet, output is $SRC.1
1113             dumpet -i "$SRC" -d
1114             EFIMNT=$(mktemp -d /media/srctmp.XXXXXX)
1115             mount -o loop "$SRC".1 $EFIMNT
1116
1117             if [ -f $EFIMNT$EFI_BOOT/+(BOOT|boot)?*.efi ]; then
1118                 cp $EFIMNT$EFI_BOOT/+(BOOT|boot)?*.efi $TGTMNT$EFI_BOOT
1119             else
1120                 echo "No BOOT*.efi found in eltorito image. EFI will not boot"
1121                 umount $EFIMNT
1122                 rm "$SRC".1
1123                 exitclean
1124             fi
1125             umount $EFIMNT
1126             rm "$SRC".1
1127         fi
1128     fi
1129 fi
1130
1131 # DVD installer copy
1132 if [ -z "$skipcopy" -a \( "$srctype" = "installer" -o "$srctype" = "netinst" \) ]; then
1133     echo "Copying DVD image to target device."
1134     mkdir -p $TGTMNT/images/
1135     if [ "$imgtype" = "install" ]; then
1136         for img in install.img updates.img product.img; do
1137             if [ -e $SRCMNT/images/$img ]; then
1138                 copyFile $SRCMNT/images/$img $TGTMNT/images/$img || exitclean
1139             fi
1140         done
1141     fi
1142 fi
1143
1144 # Copy packages over.
1145 # Before Fedora17 we could copy the .iso and setup a repo=
1146 # F17 and later look for repodata on the source media.
1147 # The presence of packages and LiveOS indicates F17 or later.
1148 if [ -n "$packages" -a -z "$skipcopy" ]; then
1149     if [ "$srctype" != "live" ]; then
1150         echo "Copying $SRC to device"
1151         copyFile "$SRC" "$TGTMNT/"
1152
1153         # Setup a repo= to point to the .iso
1154         sed -i -e "s;initrd.img;initrd.img repo=hd:$TGTLABEL:/;g" $BOOTCONFIG
1155         if [ -n "$efi" ]; then
1156             sed -i -e "s;vmlinuz;vmlinuz repo=hd:$TGTLABEL:/;g" $BOOTCONFIG_EFI
1157         fi
1158     else
1159         echo "Copying package data from $SRC to device"
1160         rsync --inplace -rLDP --exclude EFI/ --exclude images/ --exclude isolinux/ \
1161             --exclude TRANS.TBL --exclude LiveOS/ "$SRCMNT/" "$TGTMNT/"
1162     fi
1163     echo "Waiting for device to finish writing"
1164     sync
1165 fi
1166
1167 if [ "$srctype" = "live" ]; then
1168     # Copy this installer script.
1169     cp -fT "$thisScriptpath" $TGTMNT/$LIVEOS/livecd-iso-to-disk
1170     chmod +x $TGTMNT/$LIVEOS/livecd-iso-to-disk &> /dev/null || :
1171
1172     # When the source is an installed Live USB/SD image, restore the boot config
1173     # file to a base state before updating.
1174     if [[ -d $SRCMNT/syslinux/ ]]; then
1175         echo "Preparing boot config file."
1176         sed -i -e "s/root=live:[^ ]*/root=live:CDLABEL=name/"\
1177                -e "s/\(r*d*.*live.*ima*ge*\) .* quiet/\1 quiet/"\
1178                     $BOOTCONFIG $BOOTCONFIG_EFI
1179         sed -i -e "s/^timeout.*$/timeout\ 100/"\
1180                -e "/^totaltimeout.*$/d" $BOOTCONFIG
1181     fi
1182 fi
1183
1184
1185 echo "Updating boot config file"
1186 # adjust label and fstype
1187 sed -i -e "s/CDLABEL=[^ ]*/$TGTLABEL/" -e "s/rootfstype=[^ ]*/rootfstype=$TGTFS/" -e "s/LABEL=[^ ]*/$TGTLABEL/" $BOOTCONFIG  $BOOTCONFIG_EFI
1188 if [ -n "$kernelargs" ]; then
1189     sed -i -e "s;initrd.img;initrd.img ${kernelargs};" $BOOTCONFIG
1190     if [ -n "$efi" ]; then
1191         sed -i -e "s;vmlinuz;vmlinuz ${kernelargs};" $BOOTCONFIG_EFI
1192     fi
1193 fi
1194 if [ "$LIVEOS" != "LiveOS" ]; then
1195     sed -i -e "s;r*d*.*live.*ima*ge*;& live_dir=$LIVEOS;"\
1196               $BOOTCONFIG $BOOTCONFIG_EFI
1197 fi
1198
1199 # EFI images are in $SYSLINUXPATH now
1200 if [ -n "$efi" ]; then
1201     sed -i -e "s;/isolinux/;/$SYSLINUXPATH/;g" $BOOTCONFIG_EFI
1202     sed -i -e "s;/images/pxeboot/;/$SYSLINUXPATH/;g" $BOOTCONFIG_EFI
1203     sed -i -e "s;findiso;;g" $BOOTCONFIG_EFI
1204 fi
1205
1206 # DVD Installer for netinst
1207 if [ "$srctype" != "live" ]; then
1208     if [ "$imgtype" = "install" ]; then
1209         sed -i -e "s;initrd.img;initrd.img stage2=hd:$TGTLABEL:/images/install.img;g" $BOOTCONFIG
1210         if [ -n "$efi" ]; then
1211             sed -i -e "s;vmlinuz;vmlinuz stage2=hd:$TGTLABEL:/images/install.img;g" $BOOTCONFIG_EFI
1212         fi
1213     else
1214         # The initrd has everything, so no stage2
1215         sed -i -e "s;stage2=\S*;;g" $BOOTCONFIG $BOOTCONFIG_EFI
1216     fi
1217 fi
1218
1219 # Adjust the boot timeouts
1220 if [ -n "$timeout" ]; then
1221     sed -i -e "s/^timeout.*$/timeout\ $timeout/" $BOOTCONFIG
1222 fi
1223 if [ -n "$totaltimeout" ]; then
1224     sed -i -e "/^timeout.*$/a\totaltimeout\ $totaltimeout" $BOOTCONFIG
1225 fi
1226
1227 if [ "$overlaysizemb" -gt 0 ]; then
1228     echo "Initializing persistent overlay file"
1229     OVERFILE="overlay-$( /sbin/blkid -s LABEL -o value $TGTDEV )-$( /sbin/blkid -s UUID -o value $TGTDEV )"
1230     if [ -z "$skipcopy" ]; then
1231         if [ "$TGTFS" = "vfat" ]; then
1232             # vfat can't handle sparse files
1233             dd if=/dev/zero of=$TGTMNT/$LIVEOS/$OVERFILE count=$overlaysizemb bs=1M
1234         else
1235             dd if=/dev/null of=$TGTMNT/$LIVEOS/$OVERFILE count=1 bs=1M seek=$overlaysizemb
1236         fi
1237     fi
1238     sed -i -e "s/r*d*.*live.*ima*ge*/& overlay=${TGTLABEL}/"\
1239               $BOOTCONFIG $BOOTCONFIG_EFI
1240     sed -i -e "s/\ ro\ /\ rw\ /" $BOOTCONFIG  $BOOTCONFIG_EFI
1241 fi
1242
1243 if [ "$swapsizemb" -gt 0 -a -z "$skipcopy" ]; then
1244     echo "Initializing swap file"
1245     dd if=/dev/zero of=$TGTMNT/$LIVEOS/swap.img count=$swapsizemb bs=1M
1246     mkswap -f $TGTMNT/$LIVEOS/swap.img
1247 fi
1248
1249 if [ "$homesizemb" -gt 0 -a -z "$skipcopy" ]; then
1250     echo "Initializing persistent /home"
1251     homesource=/dev/zero
1252     [ -n "$cryptedhome" ] && homesource=/dev/urandom
1253     if [ "$TGTFS" = "vfat" ]; then
1254         # vfat can't handle sparse files
1255         dd if=${homesource} of=$TGTMNT/$LIVEOS/$HOMEFILE count=$homesizemb bs=1M
1256     else
1257         dd if=/dev/null of=$TGTMNT/$LIVEOS/$HOMEFILE count=1 bs=1M seek=$homesizemb
1258     fi
1259     if [ -n "$cryptedhome" ]; then
1260         loop=$(losetup -f)
1261         losetup $loop $TGTMNT/$LIVEOS/$HOMEFILE
1262
1263         echo "Encrypting persistent /home"
1264         while ! cryptsetup luksFormat -y -q $loop; do :; done;
1265
1266         echo "Please enter the password again to unlock the device"
1267         while ! cryptsetup luksOpen $loop EncHomeFoo; do :; done;
1268
1269         mkfs.ext4 -j /dev/mapper/EncHomeFoo
1270         tune2fs -c0 -i0 -ouser_xattr,acl /dev/mapper/EncHomeFoo
1271         sleep 2
1272         cryptsetup luksClose EncHomeFoo
1273         losetup -d $loop
1274     else
1275         echo "Formatting unencrypted /home"
1276         mkfs.ext4 -F -j $TGTMNT/$LIVEOS/$HOMEFILE
1277         tune2fs -c0 -i0 -ouser_xattr,acl $TGTMNT/$LIVEOS/$HOMEFILE
1278     fi
1279 fi
1280
1281 # create the forth files for booting on the XO if requested
1282 # we'd do this unconditionally, but you have to have a kernel that will
1283 # boot on the XO anyway.
1284 if [ -n "$xo" ]; then
1285     echo "Setting up /boot/olpc.fth file"
1286     args=$(grep "^ *append" $TGTMNT/$SYSLINUXPATH/isolinux.cfg |head -n1 |sed -e 's/.*initrd=[^ ]*//')
1287     if [ -z "$xonohome" -a ! -f $TGTMNT/$LIVEOS/$HOMEFILE ]; then
1288         args="$args persistenthome=mtd0"
1289     fi
1290     args="$args reset_overlay"
1291     xosyspath=$(echo $SYSLINUXPATH | sed -e 's;/;\\;')
1292     if [ ! -d $TGTMNT/boot ]; then
1293         mkdir -p $TGTMNT/boot
1294     fi
1295     cat > $TGTMNT/boot/olpc.fth <<EOF
1296 \ Boot script for USB boot
1297 hex  rom-pa fffc7 + 4 \$number drop  h# 2e19 < [if]
1298   patch 2drop erase claim-params
1299   : high-ramdisk  ( -- )
1300      cv-load-ramdisk
1301      h# 22c +lp l@ 1+   memory-limit  umin  /ramdisk - ffff.f000 and ( new-ramdisk-adr )
1302      ramdisk-adr over  /ramdisk move                    ( new-ramdisk-adr )
1303      to ramdisk-adr
1304   ;
1305   ' high-ramdisk to load-ramdisk
1306 [then]
1307
1308 : set-bootpath-dev  ( -- )
1309    " /chosen" find-package  if                       ( phandle )
1310       " bootpath" rot  get-package-property  0=  if  ( propval$ )
1311          get-encoded-string                          ( bootpath$ )
1312          [char] \ left-parse-string  2nip            ( dn$ )
1313          dn-buf place                                ( )
1314       then
1315    then
1316
1317    " /sd"  dn-buf  count  sindex  0>=   if
1318           " sd:"
1319    else
1320           " u:"
1321    then
1322    " BOOTPATHDEV" \$set-macro
1323 ;
1324
1325 set-bootpath-dev
1326 " $args" to boot-file
1327 " \${BOOTPATHDEV}$xosyspath\initrd0.img" expand$ to ramdisk
1328 " \${BOOTPATHDEV}$xosyspath\vmlinuz0" expand$ to boot-device
1329 unfreeze
1330 boot
1331 EOF
1332
1333 fi
1334
1335 if [ -z "$multi" ]; then
1336     echo "Installing boot loader"
1337     if [ -n "$efi" ]; then
1338         # replace the ia32 hack
1339         if [ -f "$TGTMNT$EFI_BOOT/BOOT.conf" ]; then
1340             cp -f $TGTMNT$EFI_BOOT/BOOTia32.conf $TGTMNT$EFI_BOOT/BOOT.conf
1341         fi
1342     fi
1343
1344     # this is a bit of a kludge, but syslinux doesn't guarantee the API for its com32 modules :/
1345     if [ -f $TGTMNT/$SYSLINUXPATH/vesamenu.c32 -a -f /usr/share/syslinux/vesamenu.c32 ]; then
1346         cp /usr/share/syslinux/vesamenu.c32 $TGTMNT/$SYSLINUXPATH/vesamenu.c32
1347     elif [ -f $TGTMNT/$SYSLINUXPATH/vesamenu.c32 -a -f /usr/lib/syslinux/vesamenu.c32 ]; then
1348         cp /usr/lib/syslinux/vesamenu.c32 $TGTMNT/$SYSLINUXPATH/vesamenu.c32
1349     elif [ -f $TGTMNT/$SYSLINUXPATH/menu.c32 -a -f /usr/share/syslinux/menu.c32 ]; then
1350         cp /usr/share/syslinux/menu.c32 $TGTMNT/$SYSLINUXPATH/menu.c32
1351     elif [ -f $TGTMNT/$SYSLINUXPATH/menu.c32 -a -f /usr/lib/syslinux/menu.c32 ]; then
1352         cp /usr/lib/syslinux/menu.c32 $TGTMNT/$SYSLINUXPATH/menu.c32
1353     fi
1354
1355     if [ "$TGTFS" == "vfat" -o "$TGTFS" == "msdos" ]; then
1356         # syslinux expects the config to be named syslinux.cfg
1357         # and has to run with the file system unmounted
1358         mv $TGTMNT/$SYSLINUXPATH/isolinux.cfg $TGTMNT/$SYSLINUXPATH/syslinux.cfg
1359         # deal with mtools complaining about ldlinux.sys
1360         if [ -f $TGTMNT/$SYSLINUXPATH/ldlinux.sys ]; then
1361             rm -f $TGTMNT/$SYSLINUXPATH/ldlinux.sys
1362         fi
1363         cleanup
1364         if [ -n "$SYSLINUXPATH" ]; then
1365             syslinux -d $SYSLINUXPATH $TGTDEV
1366         else
1367             syslinux $TGTDEV
1368         fi
1369     elif [ "$TGTFS" == "ext2" -o "$TGTFS" == "ext3" -o "$TGTFS" == "ext4" -o "$TGTFS" == "btrfs" ]; then
1370         # extlinux expects the config to be named extlinux.conf
1371         # and has to be run with the file system mounted
1372         mv $TGTMNT/$SYSLINUXPATH/isolinux.cfg $TGTMNT/$SYSLINUXPATH/extlinux.conf
1373         extlinux -i $TGTMNT/$SYSLINUXPATH
1374         # Starting with syslinux 4 ldlinux.sys is used on all file systems.
1375         if [ -f "$TGTMNT/$SYSLINUXPATH/extlinux.sys" ]; then
1376             chattr -i $TGTMNT/$SYSLINUXPATH/extlinux.sys
1377         elif [ -f "$TGTMNT/$SYSLINUXPATH/ldlinux.sys" ]; then
1378             chattr -i $TGTMNT/$SYSLINUXPATH/ldlinux.sys
1379         fi
1380         cleanup
1381     fi
1382 else
1383     # we need to do some more config file tweaks for multi-image mode
1384     sed -i -e "s;kernel vm;kernel /$LIVEOS/syslinux/vm;" $TGTMNT/$SYSLINUXPATH/isolinux.cfg
1385     sed -i -e "s;initrd=i;initrd=/$LIVEOS/syslinux/i;" $TGTMNT/$SYSLINUXPATH/isolinux.cfg
1386     mv $TGTMNT/$SYSLINUXPATH/isolinux.cfg $TGTMNT/$SYSLINUXPATH/syslinux.cfg
1387     cleanup
1388 fi
1389
1390 echo "Target device is now set up with a Live image!"