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