code.delx.au - refind/blob - install.sh

   1 #!/bin/bash
   2 #
   3 # Linux/MacOS X script to install rEFInd
   4 #
   5 # Usage:
   6 #
   7 # ./install.sh [options]
   8 #
   9 # options include:
  10 #    "--notesp" to install to the OS X root filesystem rather than to the ESP.
  11 #           This option may not be used under Linux.
  12 #    "--usedefault {devicefile}" to install as default
  13 #           (/EFI/BOOT/BOOTX64.EFI and similar) to the specified device
  14 #           (/dev/sdd1 or whatever) without registering with the NVRAM.
  15 #    "--ownhfs {devicefile}" to install to an HFS+ volume that's NOT currently
  16 #           an OS X boot volume.
  17 #    "--root {dir}" to specify installation using the specified directory
  18 #           as the system's root
  19 #    "--alldrivers" to install all drivers along with regular files
  20 #    "--nodrivers" to suppress driver installation (default in Linux is
  21 #           driver used on /boot; --nodrivers is OS X default)
  22 #    "--shim {shimfile}" to install a shim.efi file for Secure Boot
  23 #    "--preloader" is synonymous with "--shim"
  24 #    "--localkeys" to re-sign x86-64 binaries with a locally-generated key
  25 #    "--yes" to assume a "yes" response to all prompts
  26 #
  27 # The "esp" option is valid only on Mac OS X; it causes
  28 # installation to the EFI System Partition (ESP) rather than
  29 # to the current OS X boot partition. Under Linux, this script
  30 # installs to the ESP by default.
  31 #
  32 # This program is copyright (c) 2012 by Roderick W. Smith
  33 # It is released under the terms of the GNU GPL, version 3,
  34 # a copy of which should be included in the file COPYING.txt.
  35 #
  36 # Revision history:
  37 #
  38 # 0.8.4   -- OS X default changed to install to ESP under /EFI/BOOT
  39 # 0.7.9   -- Fixed bug that caused errors if dmraid utility not installed
  40 # 0.7.7   -- Fixed bug that created mangled refind_linux.conf file; added ability
  41 #            to locate and mount ESP on Linux, if it's not mounted
  42 # 0.7.6   -- Added --ownhfs {device-filename} option
  43 # 0.7.5   -- Fixed bug when installing to ESP on recent versions of OS X
  44 # 0.7.2   -- Fixed code that could be confused by use of autofs to mount the ESP
  45 # 0.7.0   -- Added support for the new Btrfs driver
  46 # 0.6.12  -- Added support for PreLoader as well as for shim
  47 # 0.6.11  -- Improvements in script's ability to handle directories with spaces
  48 #            in their names
  49 # 0.6.9   -- Install gptsync on Macs
  50 # 0.6.8   -- Bug fix: ESP scan now uses "uniq".
  51 # 0.6.6   -- Bug fix: Upgrade drivers when installed to EFI/BOOT. Also enable
  52 #            copying shim.efi and MokManager.efi over themselves.
  53 # 0.6.4   -- Copies ext2 driver rather than ext4 driver for ext2/3fs
  54 # 0.6.3   -- Support for detecting rEFInd in EFI/BOOT and EFI/Microsoft/Boot
  55 #            directories & for installing to EFI/BOOT in BIOS mode
  56 # 0.6.2-1 -- Added --yes option & tweaked key-copying for use with RPM install script
  57 # 0.6.1   -- Added --root option; minor bug fixes
  58 # 0.6.0   -- Changed --drivers to --alldrivers and added --nodrivers option;
  59 #            changed default driver installation behavior in Linux to install
  60 #            the driver needed to read /boot (if available)
  61 # 0.5.1.2 -- Fixed bug that caused failure to generate refind_linux.conf file
  62 # 0.5.1.1 -- Fixed bug that caused script failure under OS X
  63 # 0.5.1   -- Added --shim & --localkeys options & create sample refind_linux.conf
  64 #            in /boot
  65 # 0.5.0   -- Added --usedefault & --drivers options & changed "esp" option to "--esp"
  66 # 0.4.5   -- Fixed check for rEFItBlesser in OS X
  67 # 0.4.2   -- Added notice about BIOS-based OSes & made NVRAM changes in Linux smarter
  68 # 0.4.1   -- Added check for rEFItBlesser in OS X
  69 # 0.3.3.1 -- Fixed OS X 10.7 bug; also works as make target
  70 # 0.3.2.1 -- Check for presence of source files; aborts if not present
  71 # 0.3.2   -- Initial version
  72 #
  73 # Note: install.sh version numbers match those of the rEFInd package
  74 # with which they first appeared.
  75
  76 RootDir="/"
  77 TargetDir=/EFI/refind
  78 LocalKeysBase="refind_local"
  79 ShimSource="none"
  80 ShimType="none"
  81 TargetShim="default"
  82 TargetX64="refind_x64.efi"
  83 TargetIA32="refind_ia32.efi"
  84 LocalKeys=0
  85 DeleteRefindDir=0
  86 AlwaysYes=0
  87
  88 #
  89 # Functions used by both OS X and Linux....
  90 #
  91
  92 GetParams() {
  93    InstallToEspOnMac=1
  94    # Install the driver required to read Linux /boot, if it's available
  95    # Note: Under OS X, this will be installed only if a Linux partition
  96    # is detected, in which case the ext4fs driver will be installed.
  97    InstallDrivers="boot"
  98    while [[ $# -gt 0 ]]; do
  99       case $1 in
 100          --notesp) InstallToEspOnMac=0
 101               ;;
 102          --ownhfs) OwnHfs=1
 103               InstallToEspOnMac=0
 104               TargetPart="$2"
 105               TargetDir=/System/Library/CoreServices
 106               shift
 107               ;;
 108          --usedefault) TargetDir=/EFI/BOOT
 109               TargetPart="$2"
 110               TargetX64="bootx64.efi"
 111               TargetIA32="bootia32.efi"
 112               shift
 113               ;;
 114          --root) RootDir="$2"
 115               InstallToEspOnMac=0
 116               shift
 117               ;;
 118          --localkeys) LocalKeys=1
 119               ;;
 120          --shim | --preloader) ShimSource="$2"
 121               ShimType=`basename $ShimSource`
 122               shift
 123               ;;
 124          --drivers | --alldrivers) InstallDrivers="all"
 125               ;;
 126          --nodrivers) InstallDrivers="none"
 127               ;;
 128          --yes) AlwaysYes=1
 129               ;;
 130          * ) echo "Usage: $0 [--notesp | --usedefault {device-file} | --root {dir} |"
 131              echo "                     --ownhfs {device-file} ]"
 132              echo "                  [--nodrivers | --alldrivers] [--shim {shim-filename}]"
 133              echo "                  [--localkeys] [--yes]"
 134              exit 1
 135       esac
 136       shift
 137    done
 138
 139    if [[ "$InstallToEspOnMac" == 0 && "$RootDir" == '/' && "$TargetDir" == '/EFI/BOOT' ]] ; then
 140       echo "You may use --notesp OR --usedefault, but not both! Aborting!"
 141       exit 1
 142    fi
 143    if [[ "$RootDir" != '/' && "$TargetDir" == '/EFI/BOOT' ]] ; then
 144       echo "You may use --root OR --usedefault, but not both! Aborting!"
 145       exit 1
 146    fi
 147    if [[ "$TargetDir" != '/System/Library/CoreServices' && "$OwnHfs" == '1' ]] ; then
 148       echo "If you use --ownhfs, you may NOT use --usedefault! Aborting!"
 149       exit 1
 150    fi
 151    exit
 152    RLConfFile="$RootDir/boot/refind_linux.conf"
 153    EtcKeysDir="$RootDir/etc/refind.d/keys"
 154 } # GetParams()
 155
 156 # Get a yes/no response from the user and place it in the YesNo variable.
 157 # If the AlwaysYes variable is set to 1, skip the user input and set "Y"
 158 # in the YesNo variable.
 159 ReadYesNo() {
 160    if [[ $AlwaysYes == 1 ]] ; then
 161       YesNo="Y"
 162       echo "Y"
 163    else
 164       read YesNo
 165    fi
 166 }
 167
 168 # Abort if the rEFInd files can't be found.
 169 # Also sets $ConfFile to point to the configuration file,
 170 # $IconsDir to point to the icons directory, and
 171 # $ShimSource to the source of the shim.efi file (if necessary).
 172 CheckForFiles() {
 173    # Note: This check is satisfied if EITHER the 32- or the 64-bit version
 174    # is found, even on the wrong platform. This is because the platform
 175    # hasn't yet been determined. This could obviously be improved, but it
 176    # would mean restructuring lots more code....
 177    if [[ ! -f "$RefindDir/refind_ia32.efi" && ! -f "$RefindDir/refind_x64.efi" ]] ; then
 178       echo "The rEFInd binary file is missing! Aborting installation!"
 179       exit 1
 180    fi
 181
 182    if [[ -f "$RefindDir/refind.conf-sample" ]] ; then
 183       ConfFile="$RefindDir/refind.conf-sample"
 184    elif [[ -f "$ThisDir/refind.conf-sample" ]] ; then
 185       ConfFile="$ThisDir/refind.conf-sample"
 186    else
 187       echo "The sample configuration file is missing! Aborting installation!"
 188       exit 1
 189    fi
 190
 191    if [[ -d "$RefindDir/icons" ]] ; then
 192       IconsDir="$RefindDir/icons"
 193    elif [[ -d "$ThisDir/icons" ]] ; then
 194       IconsDir="$ThisDir/icons"
 195    else
 196       echo "The icons directory is missing! Aborting installation!"
 197       exit 1
 198    fi
 199
 200    if [[ "$ShimSource" != "none" ]] ; then
 201       if [[ -f "$ShimSource" ]] ; then
 202          if [[ $ShimType == "shimx64.efi" || $ShimType == "shim.efi" ]] ; then
 203             TargetX64="grubx64.efi"
 204             MokManagerSource=`dirname "$ShimSource"`/MokManager.efi
 205          elif [[ $ShimType == "preloader.efi" || $ShimType == "PreLoader.efi" ]] ; then
 206             TargetX64="loader.efi"
 207             MokManagerSource=`dirname "$ShimSource"`/HashTool.efi
 208          else
 209             echo "Unknown shim/PreBootloader filename: $ShimType!"
 210             echo "Known filenames are shimx64.efi, shim.efi, and PreLoader.efi. Aborting!"
 211             exit 1
 212          fi
 213       else
 214          echo "The specified shim/PreBootloader file, $ShimSource, doesn't exist!"
 215          echo "Aborting installation!"
 216          exit 1
 217       fi
 218    fi
 219 } # CheckForFiles()
 220
 221 # Helper for CopyRefindFiles; copies shim files (including MokManager, if it's
 222 # available) to target.
 223 CopyShimFiles() {
 224    cp -fb "$ShimSource" "$InstallDir/$TargetDir/$TargetShim"
 225    if [[ $? != 0 ]] ; then
 226       Problems=1
 227    fi
 228    if [[ -f "$MokManagerSource" ]] ; then
 229       cp -fb "$MokManagerSource" "$InstallDir/$TargetDir/"
 230    fi
 231    if [[ $? != 0 ]] ; then
 232       Problems=1
 233    fi
 234 } # CopyShimFiles()
 235
 236 # Copy the public keys to the installation medium
 237 CopyKeys() {
 238    if [[ $LocalKeys == 1 ]] ; then
 239       mkdir -p "$InstallDir/$TargetDir/keys/"
 240       cp "$EtcKeysDir/$LocalKeysBase.cer" "$InstallDir/$TargetDir/keys/"
 241       cp "$EtcKeysDir/$LocalKeysBase.crt" "$InstallDir/$TargetDir/keys/"
 242    fi
 243 } # CopyKeys()
 244
 245 # Determine (or guess) the filesystem used on the Linux /boot filesystem.
 246 # Store the result in the BootFS global variable.
 247 SetBootFS() {
 248    local Blkid
 249
 250    Blkid=`which blkid 2> /dev/null`
 251    BootFS=""
 252    if [[ $OSName == 'Linux' && -x "$Blkid" ]] ; then
 253       BootPart=`df /boot | grep dev | cut -f 1 -d " "`
 254       BootFS=`$Blkid -o export $BootPart 2> /dev/null | grep TYPE= | cut -f 2 -d =`
 255    fi
 256    if [[ $OSName == 'Darwin' ]] ; then
 257       # 0FC63DAF-8483-4772-8E79-3D69D8477DE4 = Linux filesystem
 258       # BC13C2FF-59E6-4262-A352-B275FD6F7172 = Freedesktop $boot partition
 259       # 933AC7E1-2EB4-4F13-B844-0E14E2AEF915 = Freedesktop Linux /home
 260       # E6D6D379-F507-44C2-A23C-238F2A3DF928 = Linux LVM
 261       # A19D880F-05FC-4D3B-A006-743F0F84911E = Linux RAID
 262       # 0657FD6D-A4AB-43C4-84E5-0933C84B4F4F = Linux swap
 263       Temp=$(diskutil list | grep -i '0FC63DAF-8483-4772-8E79-3D69D8477DE4\|BC13C2FF-59E6-4262-A352-B275FD6F7172\|933AC7E1-2EB4-4F13-B844-0E14E2AEF915\|E6D6D379-F507-44C2-A23C-238F2A3DF928\|A19D880F-05FC-4D3B-A006-743F0F84911E\|0657FD6D-A4AB-43C4-84E5-0933C84B4F4F\|Linux')
 264       BootFS=""
 265       if [[ -n $Temp ]] ; then
 266          echo "Found suspected Linux partition(s); installing ext4fs driver."
 267          BootFS="ext4"
 268       fi
 269    fi
 270 } # SetBootFS()
 271
 272 # Copy drivers from $RefindDir/drivers_$1 to $InstallDir/$TargetDir/drivers_$1,
 273 # honoring the $InstallDrivers condition. Must be passed a suitable
 274 # architecture code (ia32 or x64).
 275 CopyDrivers() {
 276    if [[ $InstallDrivers == "all" ]] ; then
 277       mkdir -p "$InstallDir/$TargetDir/drivers_$1"
 278       cp "$ThisDir"/drivers_$1/*_$1.efi "$InstallDir/$TargetDir/drivers_$1/" 2> /dev/null
 279       cp "$RefindDir"/drivers_$1/*_$1.efi "$InstallDir/$TargetDir/drivers_$1/" 2> /dev/null
 280    elif [[ "$InstallDrivers" == "boot" ]] ; then
 281       SetBootFS
 282       DriverType=""
 283       case $BootFS in
 284          ext2 | ext3) DriverType="ext2"
 285               # Could use ext4, but that can create unwanted entries from symbolic
 286               # links in / to /boot/vmlinuz if a separate /boot partition is used.
 287               ;;
 288          ext4) DriverType="ext4"
 289               ;;
 290          reiserfs) DriverType="reiserfs"
 291               ;;
 292          btrfs) DriverType="btrfs"
 293               ;;
 294          hfsplus) DriverType="hfs"
 295               ;;
 296          *) BootFS=""
 297       esac
 298       if [[ -n $BootFS ]] ; then
 299          echo "Installing driver for $BootFS (${DriverType}_$1.efi)"
 300          mkdir -p "$InstallDir/$TargetDir/drivers_$1"
 301          cp "$ThisDir/drivers_$1/${DriverType}_$1.efi" "$InstallDir/$TargetDir/drivers_$1/" 2> /dev/null
 302          cp "$RefindDir/drivers_$1/${DriverType}_$1.efi" "$InstallDir/$TargetDir/drivers_$1"/ 2> /dev/null
 303       fi
 304    fi
 305 } # CopyDrivers()
 306
 307 # Copy tools (currently only gptsync, and that only on Macs) to the EFI/tools
 308 # directory on the ESP. Must be passed a suitable architecture code (ia32
 309 # or x64).
 310 CopyTools() {
 311    mkdir -p $InstallDir/EFI/tools
 312    if [[ $OSName == 'Darwin' ]] ; then
 313       cp -f "$RefindDir/tools_$1/gptsync_$1.efi" "$InstallDir/EFI/tools/"
 314       if [[ -f "$InstallDir/EFI/tools/gptsync.efi" ]] ; then
 315          mv "$InstallDir/EFI/tools/gptsync.efi" "$InstallDir/EFI/tools/gptsync.efi-disabled"
 316          echo "Found old gptsync.efi; disabling it by renaming it to gptsync.efi-disabled"
 317       fi
 318    fi
 319 } # CopyTools()
 320
 321 # Copy the rEFInd files to the ESP or OS X root partition.
 322 # Sets Problems=1 if any critical commands fail.
 323 CopyRefindFiles() {
 324    mkdir -p "$InstallDir/$TargetDir"
 325    if [[ "$TargetDir" == '/EFI/BOOT' ]] ; then
 326       cp "$RefindDir/refind_ia32.efi" "$InstallDir/$TargetDir/$TargetIA32" 2> /dev/null
 327       if [[ $? != 0 ]] ; then
 328          echo "Note: IA32 (x86) binary not installed!"
 329       fi
 330       cp "$RefindDir/refind_x64.efi" "$InstallDir/$TargetDir/$TargetX64" 2> /dev/null
 331       if [[ $? != 0 ]] ; then
 332          Problems=1
 333       fi
 334       if [[ "$ShimSource" != "none" ]] ; then
 335          TargetShim="bootx64.efi"
 336          CopyShimFiles
 337       fi
 338       if [[ $InstallDrivers == "all" ]] ; then
 339          cp -r "$RefindDir"/drivers_* "$InstallDir/$TargetDir/" 2> /dev/null
 340          cp -r "$ThisDir"/drivers_* "$InstallDir/$TargetDir/" 2> /dev/null
 341       elif [[ $Upgrade == 1 || $InstallToEspOnMac == 1 ]] ; then
 342          if [[ $Platform == 'EFI64' ]] ; then
 343             CopyDrivers x64
 344             CopyTools x64
 345          else
 346             CopyDrivers ia32
 347             CopyTools ia32
 348          fi
 349       fi
 350       Refind=""
 351       if [[ $Platform == 'EFI64' ]] ; then
 352          Refind='bootx64.efi'
 353       elif [[ $Platform == 'EFI32' ]] ; then
 354          Refind='bootia32.efi'
 355       fi
 356       CopyKeys
 357    elif [[ $Platform == 'EFI64' || $TargetDir == "/EFI/Microsoft/Boot" ]] ; then
 358       cp "$RefindDir/refind_x64.efi" "$InstallDir/$TargetDir/$TargetX64"
 359       if [[ $? != 0 ]] ; then
 360          Problems=1
 361       fi
 362       CopyDrivers x64
 363       CopyTools x64
 364       Refind="refind_x64.efi"
 365       CopyKeys
 366       if [[ "$ShimSource" != "none" ]] ; then
 367          if [[ "$TargetShim" == "default" ]] ; then
 368             TargetShim=`basename "$ShimSource"`
 369          fi
 370          CopyShimFiles
 371          Refind="$TargetShim"
 372          if [[ $LocalKeys == 0 ]] ; then
 373             echo "Storing copies of rEFInd Secure Boot public keys in $EtcKeysDir"
 374             mkdir -p "$EtcKeysDir"
 375             cp "$ThisDir/keys/refind.cer" "$EtcKeysDir" 2> /dev/null
 376             cp "$ThisDir/keys/refind.crt" "$EtcKeysDir" 2> /dev/null
 377          fi
 378       fi
 379       if [[ "$TargetDir" == '/System/Library/CoreServices' ]] ; then
 380          SetupMacHfs $TargetX64
 381       fi
 382    elif [[ $Platform == 'EFI32' ]] ; then
 383       cp "$RefindDir/refind_ia32.efi" "$InstallDir/$TargetDir/$TargetIA32"
 384       if [[ $? != 0 ]] ; then
 385          Problems=1
 386       fi
 387       CopyDrivers ia32
 388       CopyTools ia32
 389       Refind="refind_ia32.efi"
 390       if [[ "$TargetDir" == '/System/Library/CoreServices' ]] ; then
 391          SetupMacHfs $TargetIA32
 392       fi
 393    else
 394       echo "Unknown platform! Aborting!"
 395       exit 1
 396    fi
 397    echo "Copied rEFInd binary files"
 398    echo ""
 399    if [[ -d "$InstallDir/$TargetDir/icons" ]] ; then
 400       rm -rf "$InstallDir/$TargetDir/icons-backup" &> /dev/null
 401       mv -f "$InstallDir/$TargetDir/icons" "$InstallDir/$TargetDir/icons-backup"
 402       echo "Notice: Backed up existing icons directory as icons-backup."
 403    fi
 404    cp -r "$IconsDir" "$InstallDir/$TargetDir"
 405    if [[ $? != 0 ]] ; then
 406       Problems=1
 407    fi
 408    mkdir -p "$InstallDir/$TargetDir/keys"
 409    cp -rf "$ThisDir"/keys/*.[cd]er "$InstallDir/$TargetDir/keys/" 2> /dev/null
 410    cp -rf "$EtcKeysDir"/*.[cd]er "$InstallDir/$TargetDir/keys/" 2> /dev/null
 411    if [[ -f "$InstallDir/$TargetDir/refind.conf" ]] ; then
 412       echo "Existing refind.conf file found; copying sample file as refind.conf-sample"
 413       echo "to avoid overwriting your customizations."
 414       echo ""
 415       cp -f "$ConfFile" "$InstallDir/$TargetDir"
 416       if [[ $? != 0 ]] ; then
 417          Problems=1
 418       fi
 419    else
 420       echo "Copying sample configuration file as refind.conf; edit this file to configure"
 421       echo "rEFInd."
 422       echo ""
 423       cp -f "$ConfFile" "$InstallDir/$TargetDir/refind.conf"
 424       if [[ $? != 0 ]] ; then
 425          Problems=1
 426       fi
 427    fi
 428    if [[ $DeleteRefindDir == 1 ]] ; then
 429       echo "Deleting the temporary directory $RefindDir"
 430       rm -r "$RefindDir"
 431    fi
 432 } # CopyRefindFiles()
 433
 434 # Mount the partition the user specified with the --usedefault or --ownhfs option
 435 MountDefaultTarget() {
 436    InstallDir=/tmp/refind_install
 437    mkdir -p "$InstallDir"
 438    UnmountEsp=1
 439    if [[ $OSName == 'Darwin' ]] ; then
 440       if [[ $OwnHfs == '1' ]] ; then
 441          Temp=`diskutil info "$TargetPart" | grep "Mount Point"`
 442          InstallDir=`echo $Temp | cut -f 3-30 -d ' '`
 443          if [[ $InstallDir == '' ]] ; then
 444             InstallDir=/tmp/refind_install
 445             mount -t hfs "$TargetPart" "$InstallDir"
 446          else
 447             UnmountEsp=0
 448          fi
 449       else
 450          mount -t msdos "$TargetPart" "$InstallDir"
 451       fi
 452    elif [[ $OSName == 'Linux' ]] ; then
 453       mount -t vfat "$TargetPart" "$InstallDir"
 454    fi
 455    if [[ $? != 0 ]] ; then
 456       echo "Couldn't mount $TargetPart ! Aborting!"
 457       rmdir "$InstallDir"
 458       exit 1
 459    fi
 460 } # MountDefaultTarget()
 461
 462 #
 463 # A series of OS X support functions....
 464 #
 465
 466 # Mount the ESP at /Volumes/ESP or determine its current mount
 467 # point.
 468 # Sets InstallDir to the ESP mount point
 469 # Sets UnmountEsp if we mounted it
 470 MountOSXESP() {
 471    # Identify the ESP. Note: This returns the FIRST ESP found;
 472    # if the system has multiple disks, this could be wrong!
 473    Temp=$(mount | sed -n -E "/^(\/dev\/disk[0-9]+s[0-9]+) on \/ \(.*$/s//\1/p")
 474    if [ $Temp ]; then
 475       Temp=$(diskutil list | grep " EFI " | grep -o 'disk.*' | head -n 1)
 476       if [ -z $Temp ]; then
 477          echo "Warning: root device doesn't have an EFI partition"
 478       fi
 479    else
 480       echo "Warning: root device could not be found"
 481    fi
 482    if [ -z $Temp ]; then
 483       Temp=$(diskutil list | sed -n -E '/^ *[0-9]+:[ ]+EFI EFI[ ]+[0-9.]+ [A-Z]+[ ]+(disk[0-9]+s[0-9]+)$/ { s//\1/p
 484              q
 485          }' )
 486
 487       if [ -z $Temp ]; then
 488          echo "Could not find an EFI partition. Aborting!"
 489          exit 1
 490       fi
 491    fi
 492    Esp=/dev/`echo $Temp`
 493    # If the ESP is mounted, use its current mount point....
 494    Temp=`df -P | grep "$Esp"`
 495    InstallDir=`echo $Temp | cut -f 6 -d ' '`
 496    if [[ "$InstallDir" == '' ]] ; then
 497       mkdir /Volumes/ESP &> /dev/null
 498       mount -t msdos "$Esp" /Volumes/ESP
 499       if [[ $? != 0 ]] ; then
 500          echo "Unable to mount ESP! Aborting!\n"
 501          exit 1
 502       fi
 503       UnmountEsp=1
 504       InstallDir="/Volumes/ESP"
 505    fi
 506 } # MountOSXESP()
 507
 508 # Set up for booting from Mac HFS+ volume that boots rEFInd in MJG's way
 509 # (http://mjg59.dreamwidth.org/7468.html)
 510 # Must be passed the original rEFInd binary filename (without a path).
 511 SetupMacHfs() {
 512    if [[ -s "$InstallDir/mach_kernel" ]] ; then
 513       echo "Attempt to install rEFInd to a partition with a /mach_kernel file! Aborting!"
 514       exit 1
 515    fi
 516    cp -n "$InstallDir/$TargetDir/boot.efi" "$InstallDir/$TargetDir/boot.efi-backup" &> /dev/null
 517    ln -f "$InstallDir/$TargetDir/$1" "$InstallDir/$TargetDir/boot.efi"
 518    touch "$InstallDir/mach_kernel"
 519    rm "$InstallDir/$TargetDir/SystemVersion.plist" &> /dev/null
 520    cat - << ENDOFHERE >> "$InstallDir/$TargetDir/SystemVersion.plist"
 521 <xml version="1.0" encoding="UTF-8"?>
 522 <plist version="1.0">
 523 <dict>
 524         <key>ProductBuildVersion</key>
 525         <string></string>
 526         <key>ProductName</key>
 527         <string>rEFInd</string>
 528         <key>ProductVersion</key>
 529         <string>0.7.6</string>
 530 </dict>
 531 </plist>
 532 ENDOFHERE
 533 } # SetupMacHfs()
 534
 535 # Control the OS X installation.
 536 # Sets Problems=1 if problems found during the installation.
 537 InstallOnOSX() {
 538    echo "Installing rEFInd on OS X...."
 539    if [[ "$InstallToEspOnMac" == "1" ]] ; then
 540       MountOSXESP
 541    elif [[ "$TargetDir" == "/EFI/BOOT" || "$OwnHfs" == '1' ]] ; then
 542       MountDefaultTarget
 543    else
 544       InstallDir="$RootDir/"
 545    fi
 546    echo "Installing rEFInd to the partition mounted at $InstallDir"
 547    Platform=`ioreg -l -p IODeviceTree | grep firmware-abi | cut -d "\"" -f 4`
 548    CopyRefindFiles
 549    if [[ $InstallToEspOnMac == "1" ]] ; then
 550       bless --mount "$InstallDir" --setBoot --file "$InstallDir/$TargetDir/$Refind"
 551    elif [[ "$TargetDir" != "/EFI/BOOT" ]] ; then
 552       bless --setBoot --folder "$InstallDir/$TargetDir" --file "$InstallDir/$TargetDir/$Refind"
 553    fi
 554    if [[ $? != 0 ]] ; then
 555       Problems=1
 556    fi
 557    if [[ -f /Library/StartupItems/rEFItBlesser || -d /Library/StartupItems/rEFItBlesser ]] ; then
 558       echo
 559       echo "/Library/StartupItems/rEFItBlesser found!"
 560       echo "This program is part of rEFIt, and will cause rEFInd to fail to work after"
 561       echo -n "its first boot. Do you want to remove rEFItBlesser (Y/N)? "
 562       ReadYesNo
 563       if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
 564          echo "Deleting /Library/StartupItems/rEFItBlesser..."
 565          rm -r /Library/StartupItems/rEFItBlesser
 566       else
 567          echo "Not deleting rEFItBlesser."
 568       fi
 569    fi
 570    echo
 571    echo "WARNING: If you have an Advanced Format disk, *DO NOT* attempt to check the"
 572    echo "bless status with 'bless --info', since this is known to cause disk corruption"
 573    echo "on some systems!!"
 574    echo
 575 } # InstallOnOSX()
 576
 577
 578 #
 579 # Now a series of Linux support functions....
 580 #
 581
 582 # Check for evidence that we're running in Secure Boot mode. If so, and if
 583 # appropriate options haven't been set, warn the user and offer to abort.
 584 # If we're NOT in Secure Boot mode but the user HAS specified the --shim
 585 # or --localkeys option, warn the user and offer to abort.
 586 #
 587 # FIXME: Although I checked the presence (and lack thereof) of the
 588 # /sys/firmware/efi/vars/SecureBoot* files on my Secure Boot test system
 589 # before releasing this script, I've since found that they are at least
 590 # sometimes present when Secure Boot is absent. This means that the first
 591 # test can produce false alarms. A better test is highly desirable.
 592 CheckSecureBoot() {
 593    VarFile=`ls -d /sys/firmware/efi/vars/SecureBoot* 2> /dev/null`
 594    if [[ -n "$VarFile" && "$TargetDir" != '/EFI/BOOT' && "$ShimSource" == "none" ]] ; then
 595       echo ""
 596       echo "CAUTION: Your computer appears to support Secure Boot, but you haven't"
 597       echo "specified a valid shim.efi file source. If you've disabled Secure Boot and"
 598       echo "intend to leave it disabled, this is fine; but if Secure Boot is active, the"
 599       echo "resulting installation won't boot. You can read more about this topic at"
 600       echo "http://www.rodsbooks.com/refind/secureboot.html."
 601       echo ""
 602       echo -n "Do you want to proceed with installation (Y/N)? "
 603       ReadYesNo
 604       if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
 605          echo "OK; continuing with the installation..."
 606       else
 607          exit 0
 608       fi
 609    fi
 610
 611    if [[ "$ShimSource" != "none" && ! -n "$VarFile" ]] ; then
 612       echo ""
 613       echo "You've specified installing using a shim.efi file, but your computer does not"
 614       echo "appear to be running in Secure Boot mode. Although installing in this way"
 615       echo "should work, it's unnecessarily complex. You may continue, but unless you"
 616       echo "plan to enable Secure Boot, you should consider stopping and omitting the"
 617       echo "--shim option. You can read more about this topic at"
 618       echo "http://www.rodsbooks.com/refind/secureboot.html."
 619       echo ""
 620       echo -n "Do you want to proceed with installation (Y/N)? "
 621       ReadYesNo
 622       if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
 623          echo "OK; continuing with the installation..."
 624       else
 625          exit 0
 626       fi
 627    fi
 628
 629    if [[ $LocalKeys != 0 && ! -n "$VarFile" ]] ; then
 630       echo ""
 631       echo "You've specified re-signing your rEFInd binaries with locally-generated keys,"
 632       echo "but your computer does not appear to be running in Secure Boot mode. The"
 633       echo "keys you generate will be useless unless you enable Secure Boot. You may"
 634       echo "proceed with this installation, but before you do so, you may want to read"
 635       echo "more about it at http://www.rodsbooks.com/refind/secureboot.html."
 636       echo ""
 637       echo -n "Do you want to proceed with installation (Y/N)? "
 638       ReadYesNo
 639       if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
 640          echo "OK; continuing with the installation..."
 641       else
 642          exit 0
 643       fi
 644    fi
 645
 646 } # CheckSecureBoot()
 647
 648 # Check for the presence of locally-generated keys from a previous installation in
 649 # $EtcKeysDir (/etc/refind.d/keys). If they're not present, generate them using
 650 # openssl.
 651 GenerateKeys() {
 652    PrivateKey="$EtcKeysDir/$LocalKeysBase.key"
 653    CertKey="$EtcKeysDir/$LocalKeysBase.crt"
 654    DerKey="$EtcKeysDir/$LocalKeysBase.cer"
 655    OpenSSL=`which openssl 2> /dev/null`
 656
 657    # Do the work only if one or more of the necessary keys is missing
 658    # TODO: Technically, we don't need the DerKey; but if it's missing and openssl
 659    # is also missing, this will fail. This could be improved.
 660    if [[ ! -f "$PrivateKey" || ! -f "$CertKey" || ! -f "$DerKey" ]] ; then
 661       echo "Generating a fresh set of local keys...."
 662       mkdir -p "$EtcKeysDir"
 663       chmod 0700 "$EtcKeysDir"
 664       if [[ ! -x "$OpenSSL" ]] ; then
 665          echo "Can't find openssl, which is required to create your private signing keys!"
 666          echo "Aborting!"
 667          exit 1
 668       fi
 669       if [[ -f "$PrivateKey" ]] ; then
 670          echo "Backing up existing $PrivateKey"
 671          cp -f "$PrivateKey" "$PrivateKey.backup" 2> /dev/null
 672       fi
 673       if [[ -f "$CertKey" ]] ; then
 674          echo "Backing up existing $CertKey"
 675          cp -f "$CertKey" "$CertKey.backup" 2> /dev/null
 676       fi
 677       if [[ -f "$DerKey" ]] ; then
 678          echo "Backing up existing $DerKey"
 679          cp -f "$DerKey" "$DerKey.backup" 2> /dev/null
 680       fi
 681       "$OpenSSL" req -new -x509 -newkey rsa:2048 -keyout "$PrivateKey" -out "$CertKey" \
 682                      -nodes -days 3650 -subj "/CN=Locally-generated rEFInd key/"
 683       "$OpenSSL" x509 -in "$CertKey" -out "$DerKey" -outform DER
 684       chmod 0600 "$PrivateKey"
 685    else
 686       echo "Using existing local keys...."
 687    fi
 688 }
 689
 690 # Sign a single binary. Requires parameters:
 691 #   $1 = source file
 692 #   $2 = destination file
 693 # Also assumes that the SBSign, PESign, UseSBSign, UsePESign, and various key variables are set
 694 # appropriately.
 695 # Aborts script on error
 696 SignOneBinary() {
 697    $SBSign --key "$PrivateKey" --cert "$CertKey" --output "$2" "$1"
 698    if [[ $? != 0 ]] ; then
 699       echo "Problem signing the binary $1! Aborting!"
 700       exit 1
 701    fi
 702 }
 703
 704 # Re-sign the x86-64 binaries with a locally-generated key, First look for appropriate
 705 # key files in $EtcKeysDir. If they're present, use them to re-sign the binaries. If
 706 # not, try to generate new keys and store them in $EtcKeysDir.
 707 ReSignBinaries() {
 708    SBSign=`which sbsign 2> /dev/null`
 709    echo "Found sbsign at $SBSign"
 710    TempDir="/tmp/refind_local"
 711    if [[ ! -x "$SBSign" ]] ; then
 712       echo "Can't find sbsign, which is required to sign rEFInd with your own keys!"
 713       echo "Aborting!"
 714       exit 1
 715    fi
 716    GenerateKeys
 717    mkdir -p "$TempDir/drivers_x64"
 718    cp "$RefindDir/refind.conf-sample $TempDir" 2> /dev/null
 719    cp "$ThisDir/refind.conf-sample $TempDir" 2> /dev/null
 720    cp "$RefindDir/refind_ia32.efi $TempDir" 2> /dev/null
 721    cp -a "$RefindDir/drivers_ia32 $TempDir" 2> /dev/null
 722    cp -a "$ThisDir/drivers_ia32 $TempDir" 2> /dev/null
 723    SignOneBinary "$RefindDir/refind_x64.efi" "$TempDir/refind_x64.efi"
 724    SaveIFS=$IFS
 725    IFS=$(echo -en "\n\b")
 726    for Driver in `ls "$RefindDir"/drivers_x64/*.efi "$ThisDir"/drivers_x64/*.efi 2> /dev/null` ; do
 727       TempName=`basename "$Driver"`
 728       SignOneBinary "$Driver" "$TempDir/drivers_x64/$TempName"
 729    done
 730    IFS=$SaveIFS
 731    RefindDir="$TempDir"
 732    DeleteRefindDir=1
 733 } # ReSignBinaries()
 734
 735 # Locate and mount an ESP, if possible, based on parted output.
 736 # Should be called only if /boot/efi is NOT an acceptable ESP.
 737 # Sets InstallDir to the mounted ESP's path ($RootDir/boot/efi)
 738 # and EspFilesystem the filesystem (always "vfat")
 739 FindLinuxESP() {
 740    echo "The ESP doesn't seem to be mounted! Trying to find it...."
 741    local Drive
 742    local PartNum
 743    local TableType
 744    local DmStatus
 745    local SkipIt
 746    local Dmraid
 747    for Drive in `ls /dev/[sh]d?` ; do
 748       SkipIt=0
 749       Dmraid=`which dmraid 2> /dev/null`
 750       if [ -x "$Dmraid" ] ; then
 751          DmStatus=`dmraid -r | grep $Drive`
 752          if [ -n "$DmStatus" ] ; then
 753             echo "$Drive seems to be part of a RAID array; skipping!"
 754             SkipIt=1
 755          fi
 756       fi
 757       TableType=`parted $Drive print -m -s 2>/dev/null | awk -F: '$1 == "'$Drive'" { print $6 }'`
 758       if [[ $TableType == 'gpt' && $SkipIt == 0 ]] ; then # read only GPT disks that aren't part of dmraid array
 759          PartNum=`LANG=C parted $Drive print -m -s 2>/dev/null | awk -F: '$7 ~ "(^boot| boot)" { print $1 }' | head -n 1`
 760          if [ "$PartNum" -eq "$PartNum" ] 2> /dev/null ; then
 761             InstallDir="$RootDir/boot/efi"
 762             mkdir -p $InstallDir
 763             mount $Drive$PartNum $InstallDir
 764             EspFilesystem=`grep "$Drive$PartNum.*/boot/efi" /etc/mtab | uniq | grep -v autofs | cut -d " " -f 3`
 765             if [[ $EspFilesystem != 'vfat' ]] ; then
 766                umount $InstallDir
 767             else
 768                echo "Mounting ESP at $InstallDir"
 769                break;
 770             fi
 771          fi # $PartNum -eq $PartNum
 772       fi # TableType
 773    done
 774 } # FindLinuxESP()
 775
 776 # Identifies the ESP's location (/boot or /boot/efi, or these locations under
 777 # the directory specified by --root); aborts if the ESP isn't mounted at
 778 # either location.
 779 # Sets InstallDir to the ESP mount point.
 780 FindMountedESP() {
 781    mount /boot &> /dev/null
 782    mount /boot/efi &> /dev/null
 783    EspLine=`df "$RootDir/boot/efi" 2> /dev/null | grep boot/efi`
 784    if [[ ! -n "$EspLine" ]] ; then
 785       EspLine=`df "$RootDir"/boot | grep boot`
 786    fi
 787    InstallDir=`echo $EspLine | cut -d " " -f 6`
 788
 789    if [[ -n "$InstallDir" ]] ; then
 790       EspFilesystem=`grep -w "$InstallDir" /etc/mtab | uniq | grep -v autofs | cut -d " " -f 3`
 791    fi
 792    if [[ $EspFilesystem != 'vfat' ]] ; then
 793       FindLinuxESP
 794    fi
 795    if [[ $EspFilesystem != 'vfat' ]] ; then
 796       echo "$RootDir/$InstallDir doesn't seem to be on a VFAT filesystem. The ESP must be"
 797       echo "mounted at $RootDir/boot or $RootDir/boot/efi and it must be VFAT! Aborting!"
 798       exit 1
 799    fi
 800    echo "ESP was found at $InstallDir using $EspFilesystem"
 801 } # FindMountedESP
 802
 803 # Uses efibootmgr to add an entry for rEFInd to the EFI's NVRAM.
 804 # If this fails, sets Problems=1
 805 AddBootEntry() {
 806    local PartNum
 807    InstallIt="0"
 808    Efibootmgr=`which efibootmgr 2> /dev/null`
 809    if [[ "$Efibootmgr" ]] ; then
 810       InstallDisk=`grep "$InstallDir" /etc/mtab | cut -d " " -f 1 | cut -c 1-8`
 811       PartNum=`grep "$InstallDir" /etc/mtab | cut -d " " -f 1 | cut -c 9-10`
 812       EntryFilename="$TargetDir/$Refind"
 813       EfiEntryFilename=`echo ${EntryFilename//\//\\\}`
 814       EfiEntryFilename2=`echo ${EfiEntryFilename} | sed s/\\\\\\\\/\\\\\\\\\\\\\\\\/g`
 815       ExistingEntry=`"$Efibootmgr" -v | grep -i "$EfiEntryFilename2"`
 816
 817       if [[ "$ExistingEntry" ]] ; then
 818          ExistingEntryBootNum=`echo "$ExistingEntry" | cut -c 5-8`
 819          FirstBoot=`"$Efibootmgr" | grep BootOrder | cut -c 12-15`
 820          if [[ "$ExistingEntryBootNum" != "$FirstBoot" ]] ; then
 821             echo "An existing rEFInd boot entry exists, but isn't set as the default boot"
 822             echo "manager. The boot order is being adjusted to make rEFInd the default boot"
 823             echo "manager. If this is NOT what you want, you should use efibootmgr to"
 824             echo "manually adjust your EFI's boot order."
 825             "$Efibootmgr" -b $ExistingEntryBootNum -B &> /dev/null
 826             InstallIt="1"
 827          fi
 828       else
 829          InstallIt="1"
 830       fi
 831
 832       if [[ $InstallIt == "1" ]] ; then
 833          echo "Installing it!"
 834          "$Efibootmgr" -c -l "$EfiEntryFilename" -L "rEFInd Boot Manager" -d $InstallDisk -p $PartNum &> /dev/null
 835          if [[ $? != 0 ]] ; then
 836             EfibootmgrProblems=1
 837             Problems=1
 838          fi
 839       fi
 840
 841    else # efibootmgr not found
 842       EfibootmgrProblems=1
 843       Problems=1
 844    fi
 845
 846    if [[ $EfibootmgrProblems ]] ; then
 847       echo
 848       echo "ALERT: There were problems running the efibootmgr program! You may need to"
 849       echo "rename the $Refind binary to the default name (EFI/boot/bootx64.efi"
 850       echo "on x86-64 systems or EFI/boot/bootia32.efi on x86 systems) to have it run!"
 851       echo
 852    else
 853       echo "rEFInd has been set as the default boot manager."
 854    fi
 855 } # AddBootEntry()
 856
 857 # Create a minimal/sample refind_linux.conf file in /boot.
 858 GenerateRefindLinuxConf() {
 859    if [[ -f "$RLConfFile" ]] ; then
 860       echo "Existing $RLConfFile found; not overwriting."
 861    else
 862       echo "Creating $RLConfFile; edit it to adjust kernel options."
 863       if [[ -f "$RootDir/etc/default/grub" ]] ; then
 864          # We want the default options used by the distribution, stored here....
 865          source "$RootDir/etc/default/grub"
 866          echo "Setting default boot options based on $RootDir/etc/default/grub"
 867       fi
 868       RootFS=`df "$RootDir" | grep dev | cut -f 1 -d " "`
 869       StartOfDevname=`echo "$RootFS" | cut -b 1-7`
 870       if [[ "$StartOfDevname" == "/dev/sd" || "$StartOfDevName" == "/dev/hd" ]] ; then
 871          # Identify root filesystem by UUID rather than by device node, if possible
 872          Uuid=`blkid -o export -s UUID "$RootFS" 2> /dev/null | grep UUID=`
 873          if [[ -n $Uuid ]] ; then
 874             RootFS="$Uuid"
 875          fi
 876       fi
 877       DefaultOptions="$GRUB_CMDLINE_LINUX $GRUB_CMDLINE_LINUX_DEFAULT"
 878       echo "\"Boot with standard options\"        \"ro root=$RootFS $DefaultOptions \"" > $RLConfFile
 879       echo "\"Boot to single-user mode\"          \"ro root=$RootFS $DefaultOptions single\"" >> $RLConfFile
 880       echo "\"Boot with minimal options\"         \"ro root=$RootFS\"" >> $RLConfFile
 881    fi
 882 }
 883
 884 # Set varaibles for installation in EFI/BOOT directory
 885 SetVarsForBoot() {
 886    TargetDir="/EFI/BOOT"
 887    if [[ $ShimSource == "none" ]] ; then
 888       TargetX64="bootx64.efi"
 889       TargetIA32="bootia32.efi"
 890    else
 891       if [[ $ShimType == "shim.efi" || $ShimType == "shimx64.efi" ]] ; then
 892          TargetX64="grubx64.efi"
 893       elif [[ $ShimType == "preloader.efi" || $ShimType == "PreLoader.efi" ]] ; then
 894          TargetX64="loader.efi"
 895       else
 896          echo "Unknown shim/PreBootloader type: $ShimType"
 897          echo "Aborting!"
 898          exit 1
 899       fi
 900       TargetIA32="bootia32.efi"
 901       TargetShim="bootx64.efi"
 902    fi
 903 } # SetFilenamesForBoot()
 904
 905 # Set variables for installation in EFI/Microsoft/Boot directory
 906 SetVarsForMsBoot() {
 907    TargetDir="/EFI/Microsoft/Boot"
 908    if [[ $ShimSource == "none" ]] ; then
 909       TargetX64="bootmgfw.efi"
 910    else
 911       if [[ $ShimType == "shim.efi" || $ShimType == "shimx64.efi" ]] ; then
 912          TargetX64="grubx64.efi"
 913       elif [[ $ShimType == "preloader.efi" || $ShimType == "PreLoader.efi" ]] ; then
 914          TargetX64="loader.efi"
 915       else
 916          echo "Unknown shim/PreBootloader type: $ShimType"
 917          echo "Aborting!"
 918          exit 1
 919       fi
 920       TargetShim="bootmgfw.efi"
 921    fi
 922 }
 923
 924 # TargetDir defaults to /EFI/refind; however, this function adjusts it as follows:
 925 # - If an existing refind.conf is available in /EFI/BOOT or /EFI/Microsoft/Boot,
 926 #   install to that directory under the suitable name; but DO NOT do this if
 927 #   refind.conf is also in /EFI/refind.
 928 # - If booted in BIOS mode and the ESP lacks any other EFI files, install to
 929 #   /EFI/BOOT
 930 # - If booted in BIOS mode and there's no refind.conf file and there is a
 931 #   /EFI/Microsoft/Boot/bootmgfw.efi file, move it down one level and
 932 #   install under that name, "hijacking" the Windows boot loader filename
 933 DetermineTargetDir() {
 934    Upgrade=0
 935
 936    if [[ -f $InstallDir/EFI/BOOT/refind.conf ]] ; then
 937       SetVarsForBoot
 938       Upgrade=1
 939    fi
 940    if [[ -f $InstallDir/EFI/Microsoft/Boot/refind.conf ]] ; then
 941       SetVarsForMsBoot
 942       Upgrade=1
 943    fi
 944    if [[ -f $InstallDir/EFI/refind/refind.conf ]] ; then
 945       TargetDir="/EFI/refind"
 946       Upgrade=1
 947    fi
 948    if [[ $Upgrade == 1 ]] ; then
 949       echo "Found rEFInd installation in $InstallDir$TargetDir; upgrading it."
 950    fi
 951
 952    if [[ ! -d /sys/firmware/efi && $Upgrade == 0 ]] ; then     # BIOS-mode
 953       FoundEfiFiles=`find "$InstallDir/EFI/BOOT" -name "*.efi" 2> /dev/null`
 954       FoundConfFiles=`find "$InstallDir" -name "refind\.conf" 2> /dev/null`
 955       if [[ ! -n "$FoundConfFiles" && -f "$InstallDir/EFI/Microsoft/Boot/bootmgfw.efi" ]] ; then
 956          mv -n "$InstallDir/EFI/Microsoft/Boot/bootmgfw.efi" "$InstallDir/EFI/Microsoft" &> /dev/null
 957          SetVarsForMsBoot
 958          echo "Running in BIOS mode with a suspected Windows installation; moving boot loader"
 959          echo "files so as to install to $InstallDir$TargetDir."
 960       elif [[ ! -n "$FoundEfiFiles" ]] ; then  # In BIOS mode and no default loader; install as default loader
 961          SetVarsForBoot
 962          echo "Running in BIOS mode with no existing default boot loader; installing to"
 963          echo $InstallDir$TargetDir
 964       else
 965          echo "Running in BIOS mode with an existing default boot loader; backing it up and"
 966          echo "installing rEFInd in its place."
 967          if [[ -d "$InstallDir/EFI/BOOT-rEFIndBackup" ]] ; then
 968             echo ""
 969             echo "Caution: An existing backup of a default boot loader exists! If the current"
 970             echo "default boot loader and the backup are different boot loaders, the current"
 971             echo "one will become inaccessible."
 972             echo ""
 973             echo -n "Do you want to proceed with installation (Y/N)? "
 974             ReadYesNo
 975             if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
 976                echo "OK; continuing with the installation..."
 977             else
 978                exit 0
 979             fi
 980          fi
 981          mv -n "$InstallDir/EFI/BOOT" "$InstallDir/EFI/BOOT-rEFIndBackup"
 982          SetVarsForBoot
 983       fi
 984    fi # BIOS-mode
 985 } # DetermineTargetDir()
 986
 987 # Controls rEFInd installation under Linux.
 988 # Sets Problems=1 if something goes wrong.
 989 InstallOnLinux() {
 990    if [[ "$TargetDir" == "/System/Library/CoreServices" ]] ; then
 991       echo "You may not use the --ownhfs option under Linux! Aborting!"
 992       exit 1
 993    fi
 994    echo "Installing rEFInd on Linux...."
 995    modprobe efivars &> /dev/null
 996    if [[ $TargetDir == "/EFI/BOOT" ]] ; then
 997       MountDefaultTarget
 998    else
 999       FindMountedESP
1000       DetermineTargetDir
1001    fi
1002    CpuType=`uname -m`
1003    if [[ $CpuType == 'x86_64' ]] ; then
1004       Platform="EFI64"
1005    elif [[ ($CpuType == 'i386' || $CpuType == 'i486' || $CpuType == 'i586' || $CpuType == 'i686') ]] ; then
1006       Platform="EFI32"
1007       # If we're in EFI mode, do some sanity checks, and alert the user or even
1008       # abort. Not in BIOS mode, though, since that could be used on an emergency
1009       # disc to try to recover a troubled Linux installation.
1010       if [[ -d /sys/firmware/efi ]] ; then
1011          if [[ "$ShimSource" != "none" && "$TargetDir" != "/BOOT/EFI" ]] ; then
1012             echo ""
1013             echo "CAUTION: shim does not currently supports 32-bit systems, so you should not"
1014             echo "use the --shim option to install on such systems. Aborting!"
1015             echo ""
1016             exit 1
1017          fi
1018          echo
1019          echo "CAUTION: This Linux installation uses a 32-bit kernel. 32-bit EFI-based"
1020          echo "computers are VERY RARE. If you've installed a 32-bit version of Linux"
1021          echo "on a 64-bit computer, you should manually install the 64-bit version of"
1022          echo "rEFInd. If you're installing on a Mac, you should do so from OS X. If"
1023          echo "you're positive you want to continue with this installation, answer 'Y'"
1024          echo "to the following question..."
1025          echo
1026          echo -n "Are you sure you want to continue (Y/N)? "
1027          ReadYesNo
1028          if [[ $YesNo == "Y" || $YesNo == "y" ]] ; then
1029             echo "OK; continuing with the installation..."
1030          else
1031             exit 0
1032          fi
1033       fi # in EFI mode
1034    else
1035       echo "Unknown CPU type '$CpuType'; aborting!"
1036       exit 1
1037    fi
1038
1039    if [[ $LocalKeys == 1 ]] ; then
1040       ReSignBinaries
1041    fi
1042
1043    CheckSecureBoot
1044    CopyRefindFiles
1045    if [[ "$TargetDir" != "/EFI/BOOT" && "$TargetDir" != "/EFI/Microsoft/Boot" ]] ; then
1046       AddBootEntry
1047       GenerateRefindLinuxConf
1048    fi
1049 } # InstallOnLinux()
1050
1051 #
1052 # The main part of the script. Sets a few environment variables,
1053 # performs a few startup checks, and then calls functions to
1054 # install under OS X or Linux, depending on the detected platform.
1055 #
1056
1057 OSName=`uname -s`
1058 GetParams "$@"
1059 ThisDir="$( cd -P "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
1060 RefindDir="$ThisDir/refind"
1061 ThisScript="$ThisDir/`basename $0`"
1062 if [[ `whoami` != "root" ]] ; then
1063    echo "Not running as root; attempting to elevate privileges via sudo...."
1064    sudo "$ThisScript" "$@"
1065    if [[ $? != 0 ]] ; then
1066       echo "This script must be run as root (or using sudo). Exiting!"
1067       exit 1
1068    else
1069       exit 0
1070    fi
1071 fi
1072 CheckForFiles
1073 if [[ $OSName == 'Darwin' ]] ; then
1074    if [[ "$ShimSource" != "none" ]] ; then
1075       echo "The --shim option is not supported on OS X! Exiting!"
1076       exit 1
1077    fi
1078    if [[ "$LocalKeys" != 0 ]] ; then
1079       echo "The --localkeys option is not supported on OS X! Exiting!"
1080       exit 1
1081    fi
1082    if [[ $InstallToEspOnMac == 1 ]] ; then
1083       TargetDir=/EFI/BOOT
1084       TargetX64="bootx64.efi"
1085       TargetIA32="bootia32.efi"
1086    fi
1087    InstallOnOSX $1
1088 elif [[ $OSName == 'Linux' ]] ; then
1089    InstallOnLinux
1090 else
1091    echo "Running on unknown OS; aborting!"
1092    if [[ "$InstallToEspOnMac" == 0 ]] ; then
1093       echo "The --notesp option is not supported on Linux! Exiting!"
1094       exit 1
1095    fi
1096 fi
1097
1098 if [[ $Problems ]] ; then
1099    echo
1100    echo "ALERT:"
1101    echo "Installation has completed, but problems were detected. Review the output for"
1102    echo "error messages and take corrective measures as necessary. You may need to"
1103    echo "re-run this script or install manually before rEFInd will work."
1104    echo
1105 else
1106    echo
1107    echo "Installation has completed successfully."
1108    echo
1109 fi
1110
1111 if [[ $UnmountEsp == '1' ]] ; then
1112    echo "Unmounting install dir"
1113    umount $InstallDir
1114 fi
1115
1116 if [[ "$InstallDir" == /tmp/refind_install ]] ; then
1117 #   sleep 5
1118    rmdir "$InstallDir"
1119 fi