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