X-Git-Url: https://code.delx.au/refind/blobdiff_plain/88322d0912852afc0a33f718625199f35f7b7196..76f5e60d25a0c4768bc6c3d5fb1344def1c0dcdc:/refind/main.c diff --git a/refind/main.c b/refind/main.c index dfe419d..d183e83 100644 --- a/refind/main.c +++ b/refind/main.c @@ -34,27 +34,44 @@ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ /* - * Modifications copyright (c) 2012-2014 Roderick W. Smith + * Modifications copyright (c) 2012-2017 Roderick W. Smith * * Modifications distributed under the terms of the GNU General Public - * License (GPL) version 3 (GPLv3), a copy of which must be distributed - * with this source code or binaries made from it. + * License (GPL) version 3 (GPLv3), or (at your option) any later version. * */ +/* + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ #include "global.h" #include "config.h" #include "screen.h" +#include "legacy.h" #include "lib.h" #include "icns.h" #include "menu.h" #include "mok.h" #include "gpt.h" +#include "apple.h" +#include "mystrings.h" #include "security_policy.h" +#include "driver_support.h" #include "../include/Handle.h" #include "../include/refit_call_wrapper.h" -#include "driver_support.h" -#include "../include/syslinux_mbr.h" +#include "../EfiLib/BdsHelper.h" +#include "../EfiLib/legacy.h" #ifdef __MAKEWITH_GNUEFI #ifndef EFI_SECURITY_VIOLATION @@ -62,9 +79,6 @@ #endif #endif -#include "../EfiLib/BdsHelper.h" -#include "../EfiLib/legacy.h" - #ifndef EFI_OS_INDICATIONS_BOOT_TO_FW_UI #define EFI_OS_INDICATIONS_BOOT_TO_FW_UI 0x0000000000000001ULL #endif @@ -76,74 +90,84 @@ // // constants -#define MACOSX_LOADER_PATH L"System\\Library\\CoreServices\\boot.efi" +#define MACOSX_LOADER_DIR L"System\\Library\\CoreServices" +#define MACOSX_LOADER_PATH ( MACOSX_LOADER_DIR L"\\boot.efi" ) #if defined (EFIX64) #define SHELL_NAMES L"\\EFI\\tools\\shell.efi,\\EFI\\tools\\shellx64.efi,\\shell.efi,\\shellx64.efi" #define GPTSYNC_NAMES L"\\EFI\\tools\\gptsync.efi,\\EFI\\tools\\gptsync_x64.efi" #define GDISK_NAMES L"\\EFI\\tools\\gdisk.efi,\\EFI\\tools\\gdisk_x64.efi" +#define NETBOOT_NAMES L"\\EFI\\tools\\ipxe.efi" #define MEMTEST_NAMES L"memtest86.efi,memtest86_x64.efi,memtest86x64.efi,bootx64.efi" -#define DRIVER_DIRS L"drivers,drivers_x64" #define FALLBACK_FULLNAME L"EFI\\BOOT\\bootx64.efi" #define FALLBACK_BASENAME L"bootx64.efi" #define EFI_STUB_ARCH 0x8664 +EFI_GUID gFreedesktopRootGuid = { 0x4f68bce3, 0xe8cd, 0x4db1, { 0x96, 0xe7, 0xfb, 0xca, 0xf9, 0x84, 0xb7, 0x09 }}; #elif defined (EFI32) #define SHELL_NAMES L"\\EFI\\tools\\shell.efi,\\EFI\\tools\\shellia32.efi,\\shell.efi,\\shellia32.efi" #define GPTSYNC_NAMES L"\\EFI\\tools\\gptsync.efi,\\EFI\\tools\\gptsync_ia32.efi" #define GDISK_NAMES L"\\EFI\\tools\\gdisk.efi,\\EFI\\tools\\gdisk_ia32.efi" +#define NETBOOT_NAMES L"\\EFI\\tools\\ipxe.efi" #define MEMTEST_NAMES L"memtest86.efi,memtest86_ia32.efi,memtest86ia32.efi,bootia32.efi" -#define DRIVER_DIRS L"drivers,drivers_ia32" #define FALLBACK_FULLNAME L"EFI\\BOOT\\bootia32.efi" #define FALLBACK_BASENAME L"bootia32.efi" #define EFI_STUB_ARCH 0x014c +EFI_GUID gFreedesktopRootGuid = { 0x44479540, 0xf297, 0x41b2, { 0x9a, 0xf7, 0xd1, 0x31, 0xd5, 0xf0, 0x45, 0x8a }}; +#elif defined (EFIAARCH64) +#define SHELL_NAMES L"\\EFI\\tools\\shell.efi,\\EFI\\tools\\shellaa64.efi,\\shell.efi,\\shellaa64.efi" +#define GPTSYNC_NAMES L"\\EFI\\tools\\gptsync.efi,\\EFI\\tools\\gptsync_aa64.efi" +#define GDISK_NAMES L"\\EFI\\tools\\gdisk.efi,\\EFI\\tools\\gdisk_aa64.efi" +#define NETBOOT_NAMES L"\\EFI\\tools\\ipxe.efi" +#define MEMTEST_NAMES L"memtest86.efi,memtest86_aa64.efi,memtest86aa64.efi,bootaa64.efi" +#define FALLBACK_FULLNAME L"EFI\\BOOT\\bootaa64.efi" +#define FALLBACK_BASENAME L"bootaa64.efi" +#define EFI_STUB_ARCH 0xaa64 +EFI_GUID gFreedesktopRootGuid = { 0xb921b045, 0x1df0, 0x41c3, { 0xaf, 0x44, 0x4c, 0x6f, 0x28, 0x0d, 0x3f, 0xae }}; #else #define SHELL_NAMES L"\\EFI\\tools\\shell.efi,\\shell.efi" #define GPTSYNC_NAMES L"\\EFI\\tools\\gptsync.efi" #define GDISK_NAMES L"\\EFI\\tools\\gdisk.efi" +#define NETBOOT_NAMES L"\\EFI\\tools\\ipxe.efi" #define MEMTEST_NAMES L"memtest86.efi" #define DRIVER_DIRS L"drivers" #define FALLBACK_FULLNAME L"EFI\\BOOT\\boot.efi" /* Not really correct */ #define FALLBACK_BASENAME L"boot.efi" /* Not really correct */ +// Below is GUID for ARM32 +EFI_GUID gFreedesktopRootGuid = { 0x69dad710, 0x2ce4, 0x4e3c, { 0xb1, 0x6c, 0x21, 0xa1, 0xd4, 0x9a, 0xbe, 0xd3 }}; #endif #define FAT_ARCH 0x0ef1fab9 /* ID for Apple "fat" binary */ -// Filename patterns that identify EFI boot loaders. Note that a single case (either L"*.efi" or -// L"*.EFI") is fine for most systems; but Gigabyte's buggy Hybrid EFI does a case-sensitive -// comparison when it should do a case-insensitive comparison, so I'm doubling this up. It does -// no harm on other computers, AFAIK. In theory, every case variation should be done for -// completeness, but that's ridiculous.... -#define LOADER_MATCH_PATTERNS L"*.efi,*.EFI" +#define IPXE_DISCOVER_NAME L"\\efi\\tools\\ipxe_discover.efi" +#define IPXE_NAME L"\\efi\\tools\\ipxe.efi" // Patterns that identify Linux kernels. Added to the loader match pattern when the // scan_all_linux_kernels option is set in the configuration file. Causes kernels WITHOUT // a ".efi" extension to be found when scanning for boot loaders. -#define LINUX_MATCH_PATTERNS L"vmlinuz*,bzImage*" +#define LINUX_MATCH_PATTERNS L"vmlinuz*,bzImage*,kernel*" -// Default hint text for program-launch submenus -#define SUBSCREEN_HINT1 L"Use arrow keys to move cursor; Enter to boot;" -#define SUBSCREEN_HINT2 L"Insert or F2 to edit options; Esc to return to main menu" -#define SUBSCREEN_HINT2_NO_EDITOR L"Esc to return to main menu" - -// Load types -#define TYPE_EFI 1 -#define TYPE_LEGACY 2 +// Maximum length of a text string in certain menus +#define MAX_LINE_LENGTH 65 static REFIT_MENU_ENTRY MenuEntryAbout = { L"About rEFInd", TAG_ABOUT, 1, 0, 'A', NULL, NULL, NULL }; static REFIT_MENU_ENTRY MenuEntryReset = { L"Reboot Computer", TAG_REBOOT, 1, 0, 'R', NULL, NULL, NULL }; static REFIT_MENU_ENTRY MenuEntryShutdown = { L"Shut Down Computer", TAG_SHUTDOWN, 1, 0, 'U', NULL, NULL, NULL }; -static REFIT_MENU_ENTRY MenuEntryReturn = { L"Return to Main Menu", TAG_RETURN, 1, 0, 0, NULL, NULL, NULL }; +REFIT_MENU_ENTRY MenuEntryReturn = { L"Return to Main Menu", TAG_RETURN, 1, 0, 0, NULL, NULL, NULL }; static REFIT_MENU_ENTRY MenuEntryExit = { L"Exit rEFInd", TAG_EXIT, 1, 0, 0, NULL, NULL, NULL }; static REFIT_MENU_ENTRY MenuEntryFirmware = { L"Reboot to Computer Setup Utility", TAG_FIRMWARE, 1, 0, 0, NULL, NULL, NULL }; +static REFIT_MENU_ENTRY MenuEntryRotateCsr = { L"Change SIP Policy", TAG_CSR_ROTATE, 1, 0, 0, NULL, NULL, NULL }; -static REFIT_MENU_SCREEN MainMenu = { L"Main Menu", NULL, 0, NULL, 0, NULL, 0, L"Automatic boot", - L"Use arrow keys to move cursor; Enter to boot;", - L"Insert or F2 for more options; Esc to refresh" }; +REFIT_MENU_SCREEN MainMenu = { L"Main Menu", NULL, 0, NULL, 0, NULL, 0, L"Automatic boot", + L"Use arrow keys to move cursor; Enter to boot;", + L"Insert, Tab, or F2 for more options; Esc or Backspace to refresh" }; static REFIT_MENU_SCREEN AboutMenu = { L"About", NULL, 0, NULL, 0, NULL, 0, NULL, L"Press Enter to return to main menu", L"" }; -REFIT_CONFIG GlobalConfig = { FALSE, TRUE, FALSE, 0, 0, 0, DONT_CHANGE_TEXT_MODE, 20, 0, 0, GRAPHICS_FOR_OSX, LEGACY_TYPE_MAC, 0, 0, - { DEFAULT_BIG_ICON_SIZE / 4, DEFAULT_SMALL_ICON_SIZE, DEFAULT_BIG_ICON_SIZE }, BANNER_NOSCALE, - NULL, NULL, CONFIG_FILE_NAME, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, - { TAG_SHELL, TAG_MEMTEST, TAG_GDISK, TAG_APPLE_RECOVERY, TAG_WINDOWS_RECOVERY, TAG_MOK_TOOL, - TAG_ABOUT, TAG_SHUTDOWN, TAG_REBOOT, TAG_FIRMWARE, 0, 0, 0, 0, 0, 0 } +REFIT_CONFIG GlobalConfig = { FALSE, TRUE, FALSE, FALSE, TRUE, FALSE, 0, 0, 0, DONT_CHANGE_TEXT_MODE, + 20, 0, 0, GRAPHICS_FOR_OSX, LEGACY_TYPE_MAC, + 0, 0, { DEFAULT_BIG_ICON_SIZE / 4, DEFAULT_SMALL_ICON_SIZE, DEFAULT_BIG_ICON_SIZE }, + BANNER_NOSCALE, NULL, NULL, NULL, NULL, CONFIG_FILE_NAME, NULL, NULL, NULL, NULL, + NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, + { TAG_SHELL, TAG_MEMTEST, TAG_GDISK, TAG_APPLE_RECOVERY, TAG_WINDOWS_RECOVERY, + TAG_MOK_TOOL, TAG_ABOUT, TAG_SHUTDOWN, TAG_REBOOT, TAG_FIRMWARE, TAG_FWUPDATE_TOOL, + 0, 0, 0, 0, 0, 0, 0, 0 } }; EFI_GUID GlobalGuid = EFI_GLOBAL_VARIABLE; @@ -154,9 +178,9 @@ GPT_DATA *gPartitions = NULL; // Structure used to hold boot loader filenames and time stamps in // a linked list; used to sort entries within a directory. struct LOADER_LIST { - CHAR16 *FileName; - EFI_TIME TimeStamp; - struct LOADER_LIST *NextEntry; + CHAR16 *FileName; + EFI_TIME TimeStamp; + struct LOADER_LIST *NextEntry; }; // @@ -165,29 +189,38 @@ struct LOADER_LIST { static VOID AboutrEFInd(VOID) { - CHAR16 *FirmwareVendor; + CHAR16 *FirmwareVendor; + UINT32 CsrStatus; if (AboutMenu.EntryCount == 0) { AboutMenu.TitleImage = BuiltinIcon(BUILTIN_ICON_FUNC_ABOUT); - AddMenuInfoLine(&AboutMenu, L"rEFInd Version 0.8.2.4"); + AddMenuInfoLine(&AboutMenu, PoolPrint(L"rEFInd Version %s", REFIND_VERSION)); AddMenuInfoLine(&AboutMenu, L""); AddMenuInfoLine(&AboutMenu, L"Copyright (c) 2006-2010 Christoph Pfisterer"); - AddMenuInfoLine(&AboutMenu, L"Copyright (c) 2012-2014 Roderick W. Smith"); + AddMenuInfoLine(&AboutMenu, L"Copyright (c) 2012-2016 Roderick W. Smith"); AddMenuInfoLine(&AboutMenu, L"Portions Copyright (c) Intel Corporation and others"); AddMenuInfoLine(&AboutMenu, L"Distributed under the terms of the GNU GPLv3 license"); AddMenuInfoLine(&AboutMenu, L""); AddMenuInfoLine(&AboutMenu, L"Running on:"); AddMenuInfoLine(&AboutMenu, PoolPrint(L" EFI Revision %d.%02d", ST->Hdr.Revision >> 16, ST->Hdr.Revision & ((1 << 16) - 1))); #if defined(EFI32) - AddMenuInfoLine(&AboutMenu, L" Platform: x86 (32 bit)"); + AddMenuInfoLine(&AboutMenu, PoolPrint(L" Platform: x86 (32 bit); Secure Boot %s", + secure_mode() ? L"active" : L"inactive")); #elif defined(EFIX64) AddMenuInfoLine(&AboutMenu, PoolPrint(L" Platform: x86_64 (64 bit); Secure Boot %s", secure_mode() ? L"active" : L"inactive")); +#elif defined(EFIAARCH64) + AddMenuInfoLine(&AboutMenu, PoolPrint(L" Platform: ARM (64 bit); Secure Boot %s", + secure_mode() ? L"active" : L"inactive")); #else AddMenuInfoLine(&AboutMenu, L" Platform: unknown"); #endif + if (GetCsrStatus(&CsrStatus) == EFI_SUCCESS) { + RecordgCsrStatus(CsrStatus, FALSE); + AddMenuInfoLine(&AboutMenu, gCsrStatus); + } FirmwareVendor = StrDuplicate(ST->FirmwareVendor); - LimitStringLength(FirmwareVendor, 65); // More than ~65 causes empty info page on 800x600 display + LimitStringLength(FirmwareVendor, MAX_LINE_LENGTH); // More than ~65 causes empty info page on 800x600 display AddMenuInfoLine(&AboutMenu, PoolPrint(L" Firmware: %s %d.%02d", FirmwareVendor, ST->FirmwareRevision >> 16, ST->FirmwareRevision & ((1 << 16) - 1))); AddMenuInfoLine(&AboutMenu, PoolPrint(L" Screen Output: %s", egScreenDescription())); @@ -207,45 +240,45 @@ static VOID AboutrEFInd(VOID) } /* VOID AboutrEFInd() */ static VOID WarnSecureBootError(CHAR16 *Name, BOOLEAN Verbose) { - if (Name == NULL) - Name = L"the loader"; - - refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR); - Print(L"Secure Boot validation failure loading %s!\n", Name); - refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC); - if (Verbose && secure_mode()) { - Print(L"\nThis computer is configured with Secure Boot active, but\n%s has failed validation.\n", Name); - Print(L"\nYou can:\n * Launch another boot loader\n"); - Print(L" * Disable Secure Boot in your firmware\n"); - Print(L" * Sign %s with a machine owner key (MOK)\n", Name); - Print(L" * Use a MOK utility (often present on the second row) to add a MOK with which\n"); - Print(L" %s has already been signed.\n", Name); - Print(L" * Use a MOK utility to register %s (\"enroll its hash\") without\n", Name); - Print(L" signing it.\n"); - Print(L"\nSee http://www.rodsbooks.com/refind/secureboot.html for more information\n"); - PauseForKey(); - } // if + if (Name == NULL) + Name = L"the loader"; + + refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR); + Print(L"Secure Boot validation failure loading %s!\n", Name); + refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC); + if (Verbose && secure_mode()) { + Print(L"\nThis computer is configured with Secure Boot active, but\n%s has failed validation.\n", Name); + Print(L"\nYou can:\n * Launch another boot loader\n"); + Print(L" * Disable Secure Boot in your firmware\n"); + Print(L" * Sign %s with a machine owner key (MOK)\n", Name); + Print(L" * Use a MOK utility (often present on the second row) to add a MOK with which\n"); + Print(L" %s has already been signed.\n", Name); + Print(L" * Use a MOK utility to register %s (\"enroll its hash\") without\n", Name); + Print(L" signing it.\n"); + Print(L"\nSee http://www.rodsbooks.com/refind/secureboot.html for more information\n"); + PauseForKey(); + } // if } // VOID WarnSecureBootError() // Returns TRUE if this file is a valid EFI loader file, and is proper ARCH static BOOLEAN IsValidLoader(EFI_FILE *RootDir, CHAR16 *FileName) { BOOLEAN IsValid = TRUE; -#if defined (EFIX64) | defined (EFI32) +#if defined (EFIX64) | defined (EFI32) | defined (EFIAARCH64) EFI_STATUS Status; EFI_FILE_HANDLE FileHandle; CHAR8 Header[512]; UINTN Size = sizeof(Header); if ((RootDir == NULL) || (FileName == NULL)) { - // Assume valid here, because Macs produce NULL RootDir (& maybe FileName) - // when launching from a Firewire drive. This should be handled better, but - // fix would have to be in StartEFIImageList() and/or in FindVolumeAndFilename(). - return TRUE; + // Assume valid here, because Macs produce NULL RootDir (& maybe FileName) + // when launching from a Firewire drive. This should be handled better, but + // fix would have to be in StartEFIImageList() and/or in FindVolumeAndFilename(). + return TRUE; } // if Status = refit_call5_wrapper(RootDir->Open, RootDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0); if (EFI_ERROR(Status)) - return FALSE; + return FALSE; Status = refit_call3_wrapper(FileHandle->Read, FileHandle, &Size, Header); refit_call1_wrapper(FileHandle->Close, FileHandle); @@ -263,15 +296,15 @@ static BOOLEAN IsValidLoader(EFI_FILE *RootDir, CHAR16 *FileName) { } // BOOLEAN IsValidLoader() // Launch an EFI binary. -static EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, - IN CHAR16 *LoadOptions, IN UINTN LoaderType, - IN CHAR16 *ImageTitle, IN CHAR8 OSType, - OUT UINTN *ErrorInStep, - IN BOOLEAN Verbose, - IN BOOLEAN IsDriver) +EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, + IN CHAR16 *LoadOptions, IN UINTN LoaderType, + IN CHAR16 *ImageTitle, IN CHAR8 OSType, + OUT UINTN *ErrorInStep, + IN BOOLEAN Verbose, + IN BOOLEAN IsDriver) { EFI_STATUS Status, ReturnStatus; - EFI_HANDLE ChildImageHandle; + EFI_HANDLE ChildImageHandle, ChildImageHandle2; EFI_LOADED_IMAGE *ChildLoadedImage = NULL; REFIT_VOLUME *Volume = NULL; UINTN DevicePathIndex; @@ -285,52 +318,68 @@ static EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, // set load options if (LoadOptions != NULL) { - FullLoadOptions = StrDuplicate(LoadOptions); - if ((LoaderType == TYPE_EFI) && (OSType == 'M')) { - MergeStrings(&FullLoadOptions, L" ", 0); - // NOTE: That last space is also added by the EFI shell and seems to be significant - // when passing options to Apple's boot.efi... + FullLoadOptions = StrDuplicate(LoadOptions); + if ((LoaderType == TYPE_EFI) && (OSType == 'M')) { + MergeStrings(&FullLoadOptions, L" ", 0); + // NOTE: That last space is also added by the EFI shell and seems to be significant + // when passing options to Apple's boot.efi... } // if } // if (LoadOptions != NULL) if (Verbose) - Print(L"Starting %s\nUsing load options '%s'\n", ImageTitle, FullLoadOptions ? FullLoadOptions : L""); + Print(L"Starting %s\nUsing load options '%s'\n", ImageTitle, FullLoadOptions ? FullLoadOptions : L""); - // load the image into memory (and execute it, in the case of a shim/MOK image). + // load the image into memory ReturnStatus = Status = EFI_NOT_FOUND; // in case the list is empty for (DevicePathIndex = 0; DevicePaths[DevicePathIndex] != NULL; DevicePathIndex++) { - FindVolumeAndFilename(DevicePaths[DevicePathIndex], &Volume, &Filename); - // Some EFIs crash if attempting to load driver for invalid architecture, so - // protect for this condition; but sometimes Volume comes back NULL, so provide - // an exception. (TODO: Handle this special condition better.) - if ((LoaderType == TYPE_LEGACY) || (Volume == NULL) || IsValidLoader(Volume->RootDir, Filename)) { - if (Filename && (LoaderType != TYPE_LEGACY)) { - Temp = PoolPrint(L"\\%s %s", Filename, FullLoadOptions ? FullLoadOptions : L""); - if (Temp != NULL) { - MyFreePool(FullLoadOptions); - FullLoadOptions = Temp; - } - } // if (Filename) - - // NOTE: Below commented-out line could be more efficient if file were read ahead of - // time and passed as a pre-loaded image to LoadImage(), but it doesn't work on my - // 32-bit Mac Mini or my 64-bit Intel box when launching a Linux kernel; the - // kernel returns a "Failed to handle fs_proto" error message. - // TODO: Track down the cause of this error and fix it, if possible. - // ReturnStatus = Status = refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, DevicePaths[DevicePathIndex], - // ImageData, ImageSize, &ChildImageHandle); - ReturnStatus = Status = refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, DevicePaths[DevicePathIndex], - NULL, 0, &ChildImageHandle); - } else { - Print(L"Invalid loader file!\n"); - ReturnStatus = EFI_LOAD_ERROR; - } - if (ReturnStatus != EFI_NOT_FOUND) { - break; - } - } + FindVolumeAndFilename(DevicePaths[DevicePathIndex], &Volume, &Filename); + // Some EFIs crash if attempting to load driver for invalid architecture, so + // protect for this condition; but sometimes Volume comes back NULL, so provide + // an exception. (TODO: Handle this special condition better.) + if ((LoaderType == TYPE_LEGACY) || (Volume == NULL) || IsValidLoader(Volume->RootDir, Filename)) { + if (Filename && (LoaderType != TYPE_LEGACY)) { + Temp = PoolPrint(L"\\%s %s", Filename, FullLoadOptions ? FullLoadOptions : L""); + if (Temp != NULL) { + MyFreePool(FullLoadOptions); + FullLoadOptions = Temp; + } + } // if (Filename) + + // NOTE: Below commented-out line could be more efficient if file were read ahead of + // time and passed as a pre-loaded image to LoadImage(), but it doesn't work on my + // 32-bit Mac Mini or my 64-bit Intel box when launching a Linux kernel; the + // kernel returns a "Failed to handle fs_proto" error message. + // TODO: Track down the cause of this error and fix it, if possible. + // ReturnStatus = Status = refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, DevicePaths[DevicePathIndex], + // ImageData, ImageSize, &ChildImageHandle); + ReturnStatus = Status = refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, DevicePaths[DevicePathIndex], + NULL, 0, &ChildImageHandle); + if (secure_mode() && ShimLoaded()) { + // Load ourself into memory. This is a trick to work around a bug in Shim 0.8, + // which ties itself into the BS->LoadImage() and BS->StartImage() functions and + // then unregisters itself from the EFI system table when its replacement + // StartImage() function is called *IF* the previous LoadImage() was for the same + // program. The result is that rEFInd can validate only the first program it + // launches (often a filesystem driver). Loading a second program (rEFInd itself, + // here, to keep it smaller than a kernel) works around this problem. See the + // replacements.c file in Shim, and especially its start_image() function, for + // the source of the problem. + // NOTE: This doesn't check the return status or handle errors. It could + // conceivably do weird things if, say, rEFInd were on a USB drive that the + // user pulls before launching a program. + refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, GlobalConfig.SelfDevicePath, + NULL, 0, &ChildImageHandle2); + } + } else { + Print(L"Invalid loader file!\n"); + ReturnStatus = EFI_LOAD_ERROR; + } + if (ReturnStatus != EFI_NOT_FOUND) { + break; + } + } // for if ((Status == EFI_ACCESS_DENIED) || (Status == EFI_SECURITY_VIOLATION)) { - WarnSecureBootError(ImageTitle, Verbose); - goto bailout; + WarnSecureBootError(ImageTitle, Verbose); + goto bailout; } SPrint(ErrorInfo, 255, L"while loading %s", ImageTitle); if (CheckError(Status, ErrorInfo)) { @@ -342,9 +391,9 @@ static EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, ReturnStatus = Status = refit_call3_wrapper(BS->HandleProtocol, ChildImageHandle, &LoadedImageProtocol, (VOID **) &ChildLoadedImage); if (CheckError(Status, L"while getting a LoadedImageProtocol handle")) { - if (ErrorInStep != NULL) - *ErrorInStep = 2; - goto bailout_unload; + if (ErrorInStep != NULL) + *ErrorInStep = 2; + goto bailout_unload; } ChildLoadedImage->LoadOptions = (VOID *)FullLoadOptions; ChildLoadedImage->LoadOptionsSize = FullLoadOptions ? ((UINT32)StrLen(FullLoadOptions) + 1) * sizeof(CHAR16) : 0; @@ -361,6 +410,12 @@ static EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, if (ErrorInStep != NULL) *ErrorInStep = 3; } + if (IsDriver) { + // Below should have no effect on most systems, but works + // around bug with some EFIs that prevents filesystem drivers + // from binding to partitions. + ConnectFilesystemDriver(ChildImageHandle); + } // re-open file handles ReinitRefitLib(); @@ -368,20 +423,19 @@ static EFI_STATUS StartEFIImageList(IN EFI_DEVICE_PATH **DevicePaths, bailout_unload: // unload the image, we don't care if it works or not... if (!IsDriver) - Status = refit_call1_wrapper(BS->UnloadImage, ChildImageHandle); + Status = refit_call1_wrapper(BS->UnloadImage, ChildImageHandle); bailout: MyFreePool(FullLoadOptions); return ReturnStatus; -} /* static EFI_STATUS StartEFIImageList() */ - -static EFI_STATUS StartEFIImage(IN EFI_DEVICE_PATH *DevicePath, - IN CHAR16 *LoadOptions, IN UINTN LoaderType, - IN CHAR16 *ImageTitle, IN CHAR8 OSType, - OUT UINTN *ErrorInStep, - IN BOOLEAN Verbose, - IN BOOLEAN IsDriver - ) +} /* EFI_STATUS StartEFIImageList() */ + +EFI_STATUS StartEFIImage(IN EFI_DEVICE_PATH *DevicePath, + IN CHAR16 *LoadOptions, IN UINTN LoaderType, + IN CHAR16 *ImageTitle, IN CHAR8 OSType, + OUT UINTN *ErrorInStep, + IN BOOLEAN Verbose, + IN BOOLEAN IsDriver) { EFI_DEVICE_PATH *DevicePaths[2]; @@ -392,52 +446,77 @@ static EFI_STATUS StartEFIImage(IN EFI_DEVICE_PATH *DevicePath, // From gummiboot: Reboot the computer into its built-in user interface static EFI_STATUS RebootIntoFirmware(VOID) { - CHAR8 *b; - UINTN size; - UINT64 osind; - EFI_STATUS err; - - osind = EFI_OS_INDICATIONS_BOOT_TO_FW_UI; - - err = EfivarGetRaw(&GlobalGuid, L"OsIndications", &b, &size); - if (err == EFI_SUCCESS) - osind |= (UINT64)*b; - MyFreePool(b); - - err = EfivarSetRaw(&GlobalGuid, L"OsIndications", (CHAR8 *)&osind, sizeof(UINT64), TRUE); - if (err != EFI_SUCCESS) - return err; - - refit_call4_wrapper(RT->ResetSystem, EfiResetCold, EFI_SUCCESS, 0, NULL); - Print(L"Error calling ResetSystem: %r", err); - PauseForKey(); - return err; + CHAR8 *b; + UINTN size; + UINT64 osind; + EFI_STATUS err; + + osind = EFI_OS_INDICATIONS_BOOT_TO_FW_UI; + + err = EfivarGetRaw(&GlobalGuid, L"OsIndications", &b, &size); + if (err == EFI_SUCCESS) + osind |= (UINT64)*b; + MyFreePool(b); + + err = EfivarSetRaw(&GlobalGuid, L"OsIndications", (CHAR8 *)&osind, sizeof(UINT64), TRUE); + if (err != EFI_SUCCESS) + return err; + + refit_call4_wrapper(RT->ResetSystem, EfiResetCold, EFI_SUCCESS, 0, NULL); + Print(L"Error calling ResetSystem: %r", err); + PauseForKey(); + return err; } // Record the value of the loader's name/description in rEFInd's "PreviousBoot" EFI variable, // if it's different from what's already stored there. -static VOID StoreLoaderName(IN CHAR16 *Name) { - EFI_STATUS Status; - CHAR16 *OldName = NULL; - UINTN Length; - - if (Name) { - Status = EfivarGetRaw(&RefindGuid, L"PreviousBoot", (CHAR8**) &OldName, &Length); - if ((Status != EFI_SUCCESS) || (StrCmp(OldName, Name) != 0)) { - EfivarSetRaw(&RefindGuid, L"PreviousBoot", (CHAR8*) Name, StrLen(Name) * 2 + 2, TRUE); - } // if - MyFreePool(OldName); - } // if +VOID StoreLoaderName(IN CHAR16 *Name) { + EFI_STATUS Status; + CHAR16 *OldName = NULL; + UINTN Length; + + if (Name) { + Status = EfivarGetRaw(&RefindGuid, L"PreviousBoot", (CHAR8**) &OldName, &Length); + if ((Status != EFI_SUCCESS) || (StrCmp(OldName, Name) != 0)) { + EfivarSetRaw(&RefindGuid, L"PreviousBoot", (CHAR8*) Name, StrLen(Name) * 2 + 2, TRUE); + } // if + MyFreePool(OldName); + } // if } // VOID StoreLoaderName() // // EFI OS loader functions // +// See http://www.thomas-krenn.com/en/wiki/Activating_the_Intel_VT_Virtualization_Feature +// for information on Intel VMX features +static VOID DoEnableAndLockVMX(VOID) +{ +#if defined (EFIX64) | defined (EFI32) + UINT32 msr = 0x3a; + UINT32 low_bits = 0, high_bits = 0; + + // is VMX active ? + __asm__ volatile ("rdmsr" : "=a" (low_bits), "=d" (high_bits) : "c" (msr)); + + // enable and lock vmx if not locked + if ((low_bits & 1) == 0) { + high_bits = 0; + low_bits = 0x05; + msr = 0x3a; + __asm__ volatile ("wrmsr" : : "c" (msr), "a" (low_bits), "d" (high_bits)); + } +#endif +} // VOID DoEnableAndLockVMX() + static VOID StartLoader(LOADER_ENTRY *Entry, CHAR16 *SelectionName) { UINTN ErrorInStep = 0; + if (GlobalConfig.EnableAndLockVMX) { + DoEnableAndLockVMX(); + } + BeginExternalScreen(Entry->UseGraphicsMode, L"Booting OS"); StoreLoaderName(SelectionName); StartEFIImage(Entry->DevicePath, Entry->LoadOptions, TYPE_EFI, @@ -453,84 +532,84 @@ static VOID StartLoader(LOADER_ENTRY *Entry, CHAR16 *SelectionName) // '\EFI\kernels\initramfs-3.3.0.img'. If the directory ALSO contains the file // initramfs-3.3.0-rc7.img or initramfs-13.3.0.img, those files will NOT match; // however, initmine-3.3.0.img might match. (FindInitrd() returns the first match it -// finds). Thus, care should be taken to avoid placing duplicate matching files in +// finds.) Thus, care should be taken to avoid placing duplicate matching files in // the kernel's directory. // If no matching init file can be found, returns NULL. static CHAR16 * FindInitrd(IN CHAR16 *LoaderPath, IN REFIT_VOLUME *Volume) { - CHAR16 *InitrdName = NULL, *FileName, *KernelVersion, *InitrdVersion, *Path; - REFIT_DIR_ITER DirIter; - EFI_FILE_INFO *DirEntry; - - FileName = Basename(LoaderPath); - KernelVersion = FindNumbers(FileName); - Path = FindPath(LoaderPath); - - // Add trailing backslash for root directory; necessary on some systems, but must - // NOT be added to all directories, since on other systems, a trailing backslash on - // anything but the root directory causes them to flake out! - if (StrLen(Path) == 0) { - MergeStrings(&Path, L"\\", 0); - } // if - DirIterOpen(Volume->RootDir, Path, &DirIter); - // Now add a trailing backslash if it was NOT added earlier, for consistency in - // building the InitrdName later.... - if ((StrLen(Path) > 0) && (Path[StrLen(Path) - 1] != L'\\')) - MergeStrings(&Path, L"\\", 0); - while ((DirIterNext(&DirIter, 2, L"init*", &DirEntry)) && (InitrdName == NULL)) { - InitrdVersion = FindNumbers(DirEntry->FileName); - if (KernelVersion != NULL) { - if (StriCmp(InitrdVersion, KernelVersion) == 0) { - InitrdName = PoolPrint(L"%s%s", Path, DirEntry->FileName); - } // if - } else { - if (InitrdVersion == NULL) { - InitrdName = PoolPrint(L"%s%s", Path, DirEntry->FileName); - } // if - } // if/else - MyFreePool(InitrdVersion); - } // while - DirIterClose(&DirIter); - - // Note: Don't FreePool(FileName), since Basename returns a pointer WITHIN the string it's passed. - MyFreePool(KernelVersion); - MyFreePool(Path); - return (InitrdName); + CHAR16 *InitrdName = NULL, *FileName, *KernelVersion, *InitrdVersion, *Path; + REFIT_DIR_ITER DirIter; + EFI_FILE_INFO *DirEntry; + + FileName = Basename(LoaderPath); + KernelVersion = FindNumbers(FileName); + Path = FindPath(LoaderPath); + + // Add trailing backslash for root directory; necessary on some systems, but must + // NOT be added to all directories, since on other systems, a trailing backslash on + // anything but the root directory causes them to flake out! + if (StrLen(Path) == 0) { + MergeStrings(&Path, L"\\", 0); + } // if + DirIterOpen(Volume->RootDir, Path, &DirIter); + // Now add a trailing backslash if it was NOT added earlier, for consistency in + // building the InitrdName later.... + if ((StrLen(Path) > 0) && (Path[StrLen(Path) - 1] != L'\\')) + MergeStrings(&Path, L"\\", 0); + while ((DirIterNext(&DirIter, 2, L"init*", &DirEntry)) && (InitrdName == NULL)) { + InitrdVersion = FindNumbers(DirEntry->FileName); + if (KernelVersion != NULL) { + if (MyStriCmp(InitrdVersion, KernelVersion)) { + InitrdName = PoolPrint(L"%s%s", Path, DirEntry->FileName); + } // if + } else { + if (InitrdVersion == NULL) { + InitrdName = PoolPrint(L"%s%s", Path, DirEntry->FileName); + } // if + } // if/else + MyFreePool(InitrdVersion); + } // while + DirIterClose(&DirIter); + + // Note: Don't FreePool(FileName), since Basename returns a pointer WITHIN the string it's passed. + MyFreePool(KernelVersion); + MyFreePool(Path); + return (InitrdName); } // static CHAR16 * FindInitrd() LOADER_ENTRY * AddPreparedLoaderEntry(LOADER_ENTRY *Entry) { - AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); + AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); - return(Entry); + return(Entry); } // LOADER_ENTRY * AddPreparedLoaderEntry() // Creates a copy of a menu screen. // Returns a pointer to the copy of the menu screen. static REFIT_MENU_SCREEN* CopyMenuScreen(REFIT_MENU_SCREEN *Entry) { - REFIT_MENU_SCREEN *NewEntry; - UINTN i; - - NewEntry = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); - if ((Entry != NULL) && (NewEntry != NULL)) { - CopyMem(NewEntry, Entry, sizeof(REFIT_MENU_SCREEN)); - NewEntry->Title = (Entry->Title) ? StrDuplicate(Entry->Title) : NULL; - NewEntry->TimeoutText = (Entry->TimeoutText) ? StrDuplicate(Entry->TimeoutText) : NULL; - if (Entry->TitleImage != NULL) { - NewEntry->TitleImage = AllocatePool(sizeof(EG_IMAGE)); - if (NewEntry->TitleImage != NULL) - CopyMem(NewEntry->TitleImage, Entry->TitleImage, sizeof(EG_IMAGE)); - } // if - NewEntry->InfoLines = (CHAR16**) AllocateZeroPool(Entry->InfoLineCount * (sizeof(CHAR16*))); - for (i = 0; i < Entry->InfoLineCount && NewEntry->InfoLines; i++) { - NewEntry->InfoLines[i] = (Entry->InfoLines[i]) ? StrDuplicate(Entry->InfoLines[i]) : NULL; - } // for - NewEntry->Entries = (REFIT_MENU_ENTRY**) AllocateZeroPool(Entry->EntryCount * (sizeof (REFIT_MENU_ENTRY*))); - for (i = 0; i < Entry->EntryCount && NewEntry->Entries; i++) { - AddMenuEntry(NewEntry, Entry->Entries[i]); - } // for - NewEntry->Hint1 = (Entry->Hint1) ? StrDuplicate(Entry->Hint1) : NULL; - NewEntry->Hint2 = (Entry->Hint2) ? StrDuplicate(Entry->Hint2) : NULL; - } // if - return (NewEntry); + REFIT_MENU_SCREEN *NewEntry; + UINTN i; + + NewEntry = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); + if ((Entry != NULL) && (NewEntry != NULL)) { + CopyMem(NewEntry, Entry, sizeof(REFIT_MENU_SCREEN)); + NewEntry->Title = (Entry->Title) ? StrDuplicate(Entry->Title) : NULL; + NewEntry->TimeoutText = (Entry->TimeoutText) ? StrDuplicate(Entry->TimeoutText) : NULL; + if (Entry->TitleImage != NULL) { + NewEntry->TitleImage = AllocatePool(sizeof(EG_IMAGE)); + if (NewEntry->TitleImage != NULL) + CopyMem(NewEntry->TitleImage, Entry->TitleImage, sizeof(EG_IMAGE)); + } // if + NewEntry->InfoLines = (CHAR16**) AllocateZeroPool(Entry->InfoLineCount * (sizeof(CHAR16*))); + for (i = 0; i < Entry->InfoLineCount && NewEntry->InfoLines; i++) { + NewEntry->InfoLines[i] = (Entry->InfoLines[i]) ? StrDuplicate(Entry->InfoLines[i]) : NULL; + } // for + NewEntry->Entries = (REFIT_MENU_ENTRY**) AllocateZeroPool(Entry->EntryCount * (sizeof (REFIT_MENU_ENTRY*))); + for (i = 0; i < Entry->EntryCount && NewEntry->Entries; i++) { + AddMenuEntry(NewEntry, Entry->Entries[i]); + } // for + NewEntry->Hint1 = (Entry->Hint1) ? StrDuplicate(Entry->Hint1) : NULL; + NewEntry->Hint2 = (Entry->Hint2) ? StrDuplicate(Entry->Hint2) : NULL; + } // if + return (NewEntry); } // static REFIT_MENU_SCREEN* CopyMenuScreen() // Creates a copy of a menu entry. Intended to enable moving a stack-based @@ -539,27 +618,27 @@ static REFIT_MENU_SCREEN* CopyMenuScreen(REFIT_MENU_SCREEN *Entry) { // entries when re-scanning. // Returns a pointer to the copy of the menu entry. static REFIT_MENU_ENTRY* CopyMenuEntry(REFIT_MENU_ENTRY *Entry) { - REFIT_MENU_ENTRY *NewEntry; - - NewEntry = AllocateZeroPool(sizeof(REFIT_MENU_ENTRY)); - if ((Entry != NULL) && (NewEntry != NULL)) { - CopyMem(NewEntry, Entry, sizeof(REFIT_MENU_ENTRY)); - NewEntry->Title = (Entry->Title) ? StrDuplicate(Entry->Title) : NULL; - if (Entry->BadgeImage != NULL) { - NewEntry->BadgeImage = AllocatePool(sizeof(EG_IMAGE)); - if (NewEntry->BadgeImage != NULL) - CopyMem(NewEntry->BadgeImage, Entry->BadgeImage, sizeof(EG_IMAGE)); - } - if (Entry->Image != NULL) { - NewEntry->Image = AllocatePool(sizeof(EG_IMAGE)); - if (NewEntry->Image != NULL) - CopyMem(NewEntry->Image, Entry->Image, sizeof(EG_IMAGE)); - } - if (Entry->SubScreen != NULL) { - NewEntry->SubScreen = CopyMenuScreen(Entry->SubScreen); - } - } // if - return (NewEntry); + REFIT_MENU_ENTRY *NewEntry; + + NewEntry = AllocateZeroPool(sizeof(REFIT_MENU_ENTRY)); + if ((Entry != NULL) && (NewEntry != NULL)) { + CopyMem(NewEntry, Entry, sizeof(REFIT_MENU_ENTRY)); + NewEntry->Title = (Entry->Title) ? StrDuplicate(Entry->Title) : NULL; + if (Entry->BadgeImage != NULL) { + NewEntry->BadgeImage = AllocatePool(sizeof(EG_IMAGE)); + if (NewEntry->BadgeImage != NULL) + CopyMem(NewEntry->BadgeImage, Entry->BadgeImage, sizeof(EG_IMAGE)); + } + if (Entry->Image != NULL) { + NewEntry->Image = AllocatePool(sizeof(EG_IMAGE)); + if (NewEntry->Image != NULL) + CopyMem(NewEntry->Image, Entry->Image, sizeof(EG_IMAGE)); + } + if (Entry->SubScreen != NULL) { + NewEntry->SubScreen = CopyMenuScreen(Entry->SubScreen); + } + } // if + return (NewEntry); } // REFIT_MENU_ENTRY* CopyMenuEntry() // Creates a new LOADER_ENTRY data structure and populates it with @@ -568,25 +647,25 @@ static REFIT_MENU_ENTRY* CopyMenuEntry(REFIT_MENU_ENTRY *Entry) { // Returns a pointer to the new data structure, or NULL if it // couldn't be allocated LOADER_ENTRY *InitializeLoaderEntry(IN LOADER_ENTRY *Entry) { - LOADER_ENTRY *NewEntry = NULL; - - NewEntry = AllocateZeroPool(sizeof(LOADER_ENTRY)); - if (NewEntry != NULL) { - NewEntry->me.Title = NULL; - NewEntry->me.Tag = TAG_LOADER; - NewEntry->Enabled = TRUE; - NewEntry->UseGraphicsMode = FALSE; - NewEntry->OSType = 0; - if (Entry != NULL) { - NewEntry->LoaderPath = (Entry->LoaderPath) ? StrDuplicate(Entry->LoaderPath) : NULL; - NewEntry->VolName = (Entry->VolName) ? StrDuplicate(Entry->VolName) : NULL; - NewEntry->DevicePath = Entry->DevicePath; - NewEntry->UseGraphicsMode = Entry->UseGraphicsMode; - NewEntry->LoadOptions = (Entry->LoadOptions) ? StrDuplicate(Entry->LoadOptions) : NULL; - NewEntry->InitrdPath = (Entry->InitrdPath) ? StrDuplicate(Entry->InitrdPath) : NULL; - } - } // if - return (NewEntry); + LOADER_ENTRY *NewEntry = NULL; + + NewEntry = AllocateZeroPool(sizeof(LOADER_ENTRY)); + if (NewEntry != NULL) { + NewEntry->me.Title = NULL; + NewEntry->me.Tag = TAG_LOADER; + NewEntry->Enabled = TRUE; + NewEntry->UseGraphicsMode = FALSE; + NewEntry->OSType = 0; + if (Entry != NULL) { + NewEntry->LoaderPath = (Entry->LoaderPath) ? StrDuplicate(Entry->LoaderPath) : NULL; + NewEntry->VolName = (Entry->VolName) ? StrDuplicate(Entry->VolName) : NULL; + NewEntry->DevicePath = Entry->DevicePath; + NewEntry->UseGraphicsMode = Entry->UseGraphicsMode; + NewEntry->LoadOptions = (Entry->LoadOptions) ? StrDuplicate(Entry->LoadOptions) : NULL; + NewEntry->InitrdPath = (Entry->InitrdPath) ? StrDuplicate(Entry->InitrdPath) : NULL; + } + } // if + return (NewEntry); } // LOADER_ENTRY *InitializeLoaderEntry() // Adds InitrdPath to Options, but only if Options doesn't already include an @@ -595,15 +674,15 @@ LOADER_ENTRY *InitializeLoaderEntry(IN LOADER_ENTRY *Entry) { // Returns a pointer to a new string. The calling function is responsible for // freeing its memory. static CHAR16 *AddInitrdToOptions(CHAR16 *Options, CHAR16 *InitrdPath) { - CHAR16 *NewOptions = NULL; - - if (Options != NULL) - NewOptions = StrDuplicate(Options); - if ((InitrdPath != NULL) && !StriSubCmp(L"initrd=", Options)) { - MergeStrings(&NewOptions, L"initrd=", L' '); - MergeStrings(&NewOptions, InitrdPath, 0); - } - return NewOptions; + CHAR16 *NewOptions = NULL; + + if (Options != NULL) + NewOptions = StrDuplicate(Options); + if ((InitrdPath != NULL) && !StriSubCmp(L"initrd=", Options)) { + MergeStrings(&NewOptions, L"initrd=", L' '); + MergeStrings(&NewOptions, InitrdPath, 0); + } + return NewOptions; } // CHAR16 *AddInitrdToOptions() // Prepare a REFIT_MENU_SCREEN data structure for a subscreen entry. This sets up @@ -614,471 +693,524 @@ static CHAR16 *AddInitrdToOptions(CHAR16 *Options, CHAR16 *InitrdPath) { // Returns a pointer to the new subscreen data structure, or NULL if there // were problems allocating memory. REFIT_MENU_SCREEN *InitializeSubScreen(IN LOADER_ENTRY *Entry) { - CHAR16 *FileName, *MainOptions = NULL; - REFIT_MENU_SCREEN *SubScreen = NULL; - LOADER_ENTRY *SubEntry; - - FileName = Basename(Entry->LoaderPath); - if (Entry->me.SubScreen == NULL) { // No subscreen yet; initialize default entry.... - SubScreen = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); - if (SubScreen != NULL) { - SubScreen->Title = AllocateZeroPool(sizeof(CHAR16) * 256); - SPrint(SubScreen->Title, 255, L"Boot Options for %s on %s", - (Entry->Title != NULL) ? Entry->Title : FileName, Entry->VolName); - SubScreen->TitleImage = Entry->me.Image; - // default entry - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = StrDuplicate(L"Boot using default options"); - MainOptions = SubEntry->LoadOptions; - SubEntry->LoadOptions = AddInitrdToOptions(MainOptions, SubEntry->InitrdPath); - MyFreePool(MainOptions); - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if (SubEntry != NULL) - SubScreen->Hint1 = StrDuplicate(SUBSCREEN_HINT1); - if (GlobalConfig.HideUIFlags & HIDEUI_FLAG_EDITOR) { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2_NO_EDITOR); - } else { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2); - } // if/else - } // if (SubScreen != NULL) - } else { // existing subscreen; less initialization, and just add new entry later.... - SubScreen = Entry->me.SubScreen; - } // if/else - return SubScreen; + CHAR16 *FileName, *MainOptions = NULL; + REFIT_MENU_SCREEN *SubScreen = NULL; + LOADER_ENTRY *SubEntry; + + FileName = Basename(Entry->LoaderPath); + if (Entry->me.SubScreen == NULL) { // No subscreen yet; initialize default entry.... + SubScreen = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); + if (SubScreen != NULL) { + SubScreen->Title = AllocateZeroPool(sizeof(CHAR16) * 256); + SPrint(SubScreen->Title, 255, L"Boot Options for %s on %s", + (Entry->Title != NULL) ? Entry->Title : FileName, Entry->VolName); + SubScreen->TitleImage = Entry->me.Image; + // default entry + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = StrDuplicate(L"Boot using default options"); + MainOptions = SubEntry->LoadOptions; + SubEntry->LoadOptions = AddInitrdToOptions(MainOptions, SubEntry->InitrdPath); + MyFreePool(MainOptions); + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // if (SubEntry != NULL) + SubScreen->Hint1 = StrDuplicate(SUBSCREEN_HINT1); + if (GlobalConfig.HideUIFlags & HIDEUI_FLAG_EDITOR) { + SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2_NO_EDITOR); + } else { + SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2); + } // if/else + } // if (SubScreen != NULL) + } else { // existing subscreen; less initialization, and just add new entry later.... + SubScreen = Entry->me.SubScreen; + } // if/else + return SubScreen; } // REFIT_MENU_SCREEN *InitializeSubScreen() -VOID GenerateSubScreen(LOADER_ENTRY *Entry, IN REFIT_VOLUME *Volume) { - REFIT_MENU_SCREEN *SubScreen; - LOADER_ENTRY *SubEntry; - CHAR16 *InitrdName; - CHAR16 DiagsFileName[256]; - REFIT_FILE *File; - UINTN TokenCount; - CHAR16 **TokenList; - - // create the submenu - if (StrLen(Entry->Title) == 0) { - MyFreePool(Entry->Title); - Entry->Title = NULL; - } - SubScreen = InitializeSubScreen(Entry); - - // loader-specific submenu entries - if (Entry->OSType == 'M') { // entries for Mac OS X -#if defined(EFIX64) - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X with a 64-bit kernel"; - SubEntry->LoadOptions = L"arch=x86_64"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if - - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X with a 32-bit kernel"; - SubEntry->LoadOptions = L"arch=i386"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if -#endif +VOID GenerateSubScreen(LOADER_ENTRY *Entry, IN REFIT_VOLUME *Volume, IN BOOLEAN GenerateReturn) { + REFIT_MENU_SCREEN *SubScreen; + LOADER_ENTRY *SubEntry; + CHAR16 *InitrdName; + CHAR16 DiagsFileName[256]; + REFIT_FILE *File; + UINTN TokenCount; + CHAR16 **TokenList; - if (!(GlobalConfig.HideUIFlags & HIDEUI_FLAG_SINGLEUSER)) { - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X in verbose mode"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v"; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if + // create the submenu + if (StrLen(Entry->Title) == 0) { + MyFreePool(Entry->Title); + Entry->Title = NULL; + } + SubScreen = InitializeSubScreen(Entry); + // loader-specific submenu entries + if (Entry->OSType == 'M') { // entries for Mac OS X #if defined(EFIX64) - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X in verbose mode (64-bit)"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v arch=x86_64"; + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X with a 64-bit kernel"; + SubEntry->LoadOptions = L"arch=x86_64"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } + } // if - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X in verbose mode (32-bit)"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v arch=i386"; + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X with a 32-bit kernel"; + SubEntry->LoadOptions = L"arch=i386"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } + } // if #endif - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X in single user mode"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v -s"; + if (!(GlobalConfig.HideUIFlags & HIDEUI_FLAG_SINGLEUSER)) { + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X in verbose mode"; + SubEntry->UseGraphicsMode = FALSE; + SubEntry->LoadOptions = L"-v"; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // if + +#if defined(EFIX64) + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X in verbose mode (64-bit)"; + SubEntry->UseGraphicsMode = FALSE; + SubEntry->LoadOptions = L"-v arch=x86_64"; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } + + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X in verbose mode (32-bit)"; + SubEntry->UseGraphicsMode = FALSE; + SubEntry->LoadOptions = L"-v arch=i386"; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } +#endif + + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X in single user mode"; + SubEntry->UseGraphicsMode = FALSE; + SubEntry->LoadOptions = L"-v -s"; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // if + } // single-user mode allowed + + if (!(GlobalConfig.HideUIFlags & HIDEUI_FLAG_SAFEMODE)) { + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Mac OS X in safe mode"; + SubEntry->UseGraphicsMode = FALSE; + SubEntry->LoadOptions = L"-v -x"; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // if + } // safe mode allowed + + // check for Apple hardware diagnostics + StrCpy(DiagsFileName, L"System\\Library\\CoreServices\\.diagnostics\\diags.efi"); + if (FileExists(Volume->RootDir, DiagsFileName) && !(GlobalConfig.HideUIFlags & HIDEUI_FLAG_HWTEST)) { + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Run Apple Hardware Test"; + MyFreePool(SubEntry->LoaderPath); + SubEntry->LoaderPath = StrDuplicate(DiagsFileName); + SubEntry->DevicePath = FileDevicePath(Volume->DeviceHandle, SubEntry->LoaderPath); + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // if + } // if diagnostics entry found + + } else if (Entry->OSType == 'L') { // entries for Linux kernels with EFI stub loaders + File = ReadLinuxOptionsFile(Entry->LoaderPath, Volume); + if (File != NULL) { + InitrdName = FindInitrd(Entry->LoaderPath, Volume); + TokenCount = ReadTokenLine(File, &TokenList); + // first entry requires special processing, since it was initially set + // up with a default title but correct options by InitializeSubScreen(), + // earlier.... + if ((TokenCount > 1) && (SubScreen->Entries != NULL) && (SubScreen->Entries[0] != NULL)) { + MyFreePool(SubScreen->Entries[0]->Title); + SubScreen->Entries[0]->Title = TokenList[0] ? StrDuplicate(TokenList[0]) : StrDuplicate(L"Boot Linux"); + } // if + FreeTokenLine(&TokenList, &TokenCount); + while ((TokenCount = ReadTokenLine(File, &TokenList)) > 1) { + SubEntry = InitializeLoaderEntry(Entry); + SubEntry->me.Title = TokenList[0] ? StrDuplicate(TokenList[0]) : StrDuplicate(L"Boot Linux"); + MyFreePool(SubEntry->LoadOptions); + SubEntry->LoadOptions = AddInitrdToOptions(TokenList[1], InitrdName); + FreeTokenLine(&TokenList, &TokenCount); + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_LINUX; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // while + MyFreePool(InitrdName); + MyFreePool(File); + } // if + + } else if (Entry->OSType == 'E') { // entries for ELILO + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Run ELILO in interactive mode"; + SubEntry->LoadOptions = L"-p"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if - } // single-user mode allowed - - if (!(GlobalConfig.HideUIFlags & HIDEUI_FLAG_SAFEMODE)) { - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Mac OS X in safe mode"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v -x"; + } + + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Linux for a 17\" iMac or a 15\" MacBook Pro (*)"; + SubEntry->LoadOptions = L"-d 0 i17"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if - } // safe mode allowed - - // check for Apple hardware diagnostics - StrCpy(DiagsFileName, L"System\\Library\\CoreServices\\.diagnostics\\diags.efi"); - if (FileExists(Volume->RootDir, DiagsFileName) && !(GlobalConfig.HideUIFlags & HIDEUI_FLAG_HWTEST)) { - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Run Apple Hardware Test"; - MyFreePool(SubEntry->LoaderPath); - SubEntry->LoaderPath = StrDuplicate(DiagsFileName); - SubEntry->DevicePath = FileDevicePath(Volume->DeviceHandle, SubEntry->LoaderPath); - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; + } + + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Linux for a 20\" iMac (*)"; + SubEntry->LoadOptions = L"-d 0 i20"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // if - } // if diagnostics entry found - - } else if (Entry->OSType == 'L') { // entries for Linux kernels with EFI stub loaders - File = ReadLinuxOptionsFile(Entry->LoaderPath, Volume); - if (File != NULL) { - InitrdName = FindInitrd(Entry->LoaderPath, Volume); - TokenCount = ReadTokenLine(File, &TokenList); - // first entry requires special processing, since it was initially set - // up with a default title but correct options by InitializeSubScreen(), - // earlier.... - if ((SubScreen->Entries != NULL) && (SubScreen->Entries[0] != NULL)) { - MyFreePool(SubScreen->Entries[0]->Title); - SubScreen->Entries[0]->Title = TokenList[0] ? StrDuplicate(TokenList[0]) : StrDuplicate(L"Boot Linux"); - } // if - FreeTokenLine(&TokenList, &TokenCount); - while ((TokenCount = ReadTokenLine(File, &TokenList)) > 1) { - SubEntry = InitializeLoaderEntry(Entry); - SubEntry->me.Title = TokenList[0] ? StrDuplicate(TokenList[0]) : StrDuplicate(L"Boot Linux"); - MyFreePool(SubEntry->LoadOptions); - SubEntry->LoadOptions = AddInitrdToOptions(TokenList[1], InitrdName); - FreeTokenLine(&TokenList, &TokenCount); - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_LINUX; + } + + SubEntry = InitializeLoaderEntry(Entry); + if (SubEntry != NULL) { + SubEntry->me.Title = L"Boot Linux for a Mac Mini (*)"; + SubEntry->LoadOptions = L"-d 0 mini"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } // while - MyFreePool(InitrdName); - MyFreePool(File); - } // if - - } else if (Entry->OSType == 'E') { // entries for ELILO - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Run ELILO in interactive mode"; - SubEntry->LoadOptions = L"-p"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } - - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Linux for a 17\" iMac or a 15\" MacBook Pro (*)"; - SubEntry->UseGraphicsMode = TRUE; - SubEntry->LoadOptions = L"-d 0 i17"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } - - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Linux for a 20\" iMac (*)"; - SubEntry->UseGraphicsMode = TRUE; - SubEntry->LoadOptions = L"-d 0 i20"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } - - SubEntry = InitializeLoaderEntry(Entry); - if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Linux for a Mac Mini (*)"; - SubEntry->UseGraphicsMode = TRUE; - SubEntry->LoadOptions = L"-d 0 mini"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - } - - AddMenuInfoLine(SubScreen, L"NOTE: This is an example. Entries"); - AddMenuInfoLine(SubScreen, L"marked with (*) may not work."); - - } else if (Entry->OSType == 'X') { // entries for xom.efi + } + + AddMenuInfoLine(SubScreen, L"NOTE: This is an example. Entries"); + AddMenuInfoLine(SubScreen, L"marked with (*) may not work."); + + } else if (Entry->OSType == 'X') { // entries for xom.efi // by default, skip the built-in selection and boot from hard disk only Entry->LoadOptions = L"-s -h"; SubEntry = InitializeLoaderEntry(Entry); if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Windows from Hard Disk"; - SubEntry->LoadOptions = L"-s -h"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + SubEntry->me.Title = L"Boot Windows from Hard Disk"; + SubEntry->LoadOptions = L"-s -h"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); } SubEntry = InitializeLoaderEntry(Entry); if (SubEntry != NULL) { - SubEntry->me.Title = L"Boot Windows from CD-ROM"; - SubEntry->LoadOptions = L"-s -c"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + SubEntry->me.Title = L"Boot Windows from CD-ROM"; + SubEntry->LoadOptions = L"-s -c"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); } SubEntry = InitializeLoaderEntry(Entry); if (SubEntry != NULL) { - SubEntry->me.Title = L"Run XOM in text mode"; - SubEntry->UseGraphicsMode = FALSE; - SubEntry->LoadOptions = L"-v"; - SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + SubEntry->me.Title = L"Run XOM in text mode"; + SubEntry->LoadOptions = L"-v"; + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); } - } // entries for xom.efi - AddMenuEntry(SubScreen, &MenuEntryReturn); - Entry->me.SubScreen = SubScreen; + } // entries for xom.efi + if (GenerateReturn) + AddMenuEntry(SubScreen, &MenuEntryReturn); + Entry->me.SubScreen = SubScreen; } // VOID GenerateSubScreen() // Returns options for a Linux kernel. Reads them from an options file in the // kernel's directory; and if present, adds an initrd= option for an initial // RAM disk file with the same version number as the kernel file. static CHAR16 * GetMainLinuxOptions(IN CHAR16 * LoaderPath, IN REFIT_VOLUME *Volume) { - CHAR16 *Options = NULL, *InitrdName, *FullOptions = NULL; + CHAR16 *Options = NULL, *InitrdName, *FullOptions = NULL; - Options = GetFirstOptionsFromFile(LoaderPath, Volume); - InitrdName = FindInitrd(LoaderPath, Volume); - FullOptions = AddInitrdToOptions(Options, InitrdName); + Options = GetFirstOptionsFromFile(LoaderPath, Volume); + InitrdName = FindInitrd(LoaderPath, Volume); + FullOptions = AddInitrdToOptions(Options, InitrdName); - MyFreePool(Options); - MyFreePool(InitrdName); - return (FullOptions); + MyFreePool(Options); + MyFreePool(InitrdName); + return (FullOptions); } // static CHAR16 * GetMainLinuxOptions() +// Read the specified file and add values of "ID", "NAME", or "DISTRIB_ID" tokens to +// OSIconName list. Intended for adding Linux distribution clues gleaned from +// /etc/lsb-release and /etc/os-release files. +static VOID ParseReleaseFile(CHAR16 **OSIconName, REFIT_VOLUME *Volume, CHAR16 *FileName) { + UINTN FileSize = 0; + REFIT_FILE File; + CHAR16 **TokenList; + UINTN TokenCount = 0; + + if ((Volume == NULL) || (FileName == NULL) || (OSIconName == NULL) || (*OSIconName == NULL)) + return; + + if (FileExists(Volume->RootDir, FileName) && + (ReadFile(Volume->RootDir, FileName, &File, &FileSize) == EFI_SUCCESS)) { + do { + TokenCount = ReadTokenLine(&File, &TokenList); + if ((TokenCount > 1) && (MyStriCmp(TokenList[0], L"ID") || + MyStriCmp(TokenList[0], L"NAME") || + MyStriCmp(TokenList[0], L"DISTRIB_ID"))) { + MergeWords(OSIconName, TokenList[1], L','); + } // if + FreeTokenLine(&TokenList, &TokenCount); + } while (TokenCount > 0); + MyFreePool(File.Buffer); + } // if +} // VOID ParseReleaseFile() + // Try to guess the name of the Linux distribution & add that name to // OSIconName list. static VOID GuessLinuxDistribution(CHAR16 **OSIconName, REFIT_VOLUME *Volume, CHAR16 *LoaderPath) { - UINTN FileSize = 0; - REFIT_FILE File; - CHAR16** TokenList; - UINTN TokenCount = 0; - - // If on Linux root fs, /etc/os-release file probably has clues.... - if (FileExists(Volume->RootDir, L"etc\\os-release") && - (ReadFile(Volume->RootDir, L"etc\\os-release", &File, &FileSize) == EFI_SUCCESS)) { - do { - TokenCount = ReadTokenLine(&File, &TokenList); - if ((TokenCount > 1) && ((StriCmp(TokenList[0], L"ID") == 0) || (StriCmp(TokenList[0], L"NAME") == 0))) { - MergeStrings(OSIconName, TokenList[1], L','); - } // if - FreeTokenLine(&TokenList, &TokenCount); - } while (TokenCount > 0); - MyFreePool(File.Buffer); - } // if - - // Search for clues in the kernel's filename.... - if (StriSubCmp(L".fc", LoaderPath)) - MergeStrings(OSIconName, L"fedora", L','); - if (StriSubCmp(L".el", LoaderPath)) - MergeStrings(OSIconName, L"redhat", L','); + // If on Linux root fs, /etc/os-release or /etc/lsb-release file probably has clues.... + ParseReleaseFile(OSIconName, Volume, L"etc\\lsb-release"); + ParseReleaseFile(OSIconName, Volume, L"etc\\os-release"); + + // Search for clues in the kernel's filename.... + if (StriSubCmp(L".fc", LoaderPath)) + MergeStrings(OSIconName, L"fedora", L','); + if (StriSubCmp(L".el", LoaderPath)) + MergeStrings(OSIconName, L"redhat", L','); } // VOID GuessLinuxDistribution() // Sets a few defaults for a loader entry -- mainly the icon, but also the OS type // code and shortcut letter. For Linux EFI stub loaders, also sets kernel options // that will (with luck) work fairly automatically. VOID SetLoaderDefaults(LOADER_ENTRY *Entry, CHAR16 *LoaderPath, REFIT_VOLUME *Volume) { - CHAR16 *FileName, *PathOnly, *NoExtension, *OSIconName = NULL, *Temp, *SubString; - CHAR16 ShortcutLetter = 0; - UINTN i = 0, Length; - - FileName = Basename(LoaderPath); - PathOnly = FindPath(LoaderPath); - NoExtension = StripEfiExtension(FileName); - - // locate a custom icon for the loader - // Anything found here takes precedence over the "hints" in the OSIconName variable - if (!Entry->me.Image) { - Entry->me.Image = egLoadIconAnyType(Volume->RootDir, PathOnly, NoExtension, GlobalConfig.IconSizes[ICON_SIZE_BIG]); - } - if (!Entry->me.Image) { - Entry->me.Image = egCopyImage(Volume->VolIconImage); - } - - // Begin creating icon "hints" by using last part of directory path leading - // to the loader - Temp = FindLastDirName(LoaderPath); - MergeStrings(&OSIconName, Temp, L','); - MyFreePool(Temp); - Temp = NULL; - if (OSIconName != NULL) { - ShortcutLetter = OSIconName[0]; - } - - // Add every "word" in the volume label, delimited by spaces, dashes (-), or - // underscores (_), to the list of hints to be used in searching for OS - // icons. - if ((Volume->VolName) && (StrLen(Volume->VolName) > 0)) { - Temp = SubString = StrDuplicate(Volume->VolName); - if (Temp != NULL) { - Length = StrLen(Temp); - for (i = 0; i < Length; i++) { - if ((Temp[i] == L' ') || (Temp[i] == L'_') || (Temp[i] == L'-')) { - Temp[i] = 0; - if (StrLen(SubString) > 0) - MergeStrings(&OSIconName, SubString, L','); - SubString = Temp + i + 1; - } // if - } // for - MergeStrings(&OSIconName, SubString, L','); - MyFreePool(Temp); - } // if - } // if - - // detect specific loaders - if (StriSubCmp(L"bzImage", FileName) || StriSubCmp(L"vmlinuz", FileName)) { - GuessLinuxDistribution(&OSIconName, Volume, LoaderPath); - MergeStrings(&OSIconName, L"linux", L','); - Entry->OSType = 'L'; - if (ShortcutLetter == 0) - ShortcutLetter = 'L'; - Entry->LoadOptions = GetMainLinuxOptions(LoaderPath, Volume); - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_LINUX; - } else if (StriSubCmp(L"refit", LoaderPath)) { - MergeStrings(&OSIconName, L"refit", L','); - Entry->OSType = 'R'; - ShortcutLetter = 'R'; - } else if (StriSubCmp(L"refind", LoaderPath)) { - MergeStrings(&OSIconName, L"refind", L','); - Entry->OSType = 'R'; - ShortcutLetter = 'R'; - } else if (StriCmp(LoaderPath, MACOSX_LOADER_PATH) == 0) { - MergeStrings(&OSIconName, L"mac", L','); - Entry->OSType = 'M'; - ShortcutLetter = 'M'; - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; - } else if (StriCmp(FileName, L"diags.efi") == 0) { - MergeStrings(&OSIconName, L"hwtest", L','); - } else if (StriCmp(FileName, L"e.efi") == 0 || StriCmp(FileName, L"elilo.efi") == 0 || StriSubCmp(L"elilo", FileName)) { - MergeStrings(&OSIconName, L"elilo,linux", L','); - Entry->OSType = 'E'; - if (ShortcutLetter == 0) - ShortcutLetter = 'L'; - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; - } else if (StriSubCmp(L"grub", FileName)) { - Entry->OSType = 'G'; - ShortcutLetter = 'G'; - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_GRUB; - } else if (StriCmp(FileName, L"cdboot.efi") == 0 || - StriCmp(FileName, L"bootmgr.efi") == 0 || - StriCmp(FileName, L"bootmgfw.efi") == 0 || - StriCmp(FileName, L"bkpbootmgfw.efi") == 0) { - MergeStrings(&OSIconName, L"win", L','); - Entry->OSType = 'W'; - ShortcutLetter = 'W'; - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; - } else if (StriCmp(FileName, L"xom.efi") == 0) { - MergeStrings(&OSIconName, L"xom,win", L','); - Entry->UseGraphicsMode = TRUE; - Entry->OSType = 'X'; - ShortcutLetter = 'W'; - Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; - } - - if ((ShortcutLetter >= 'a') && (ShortcutLetter <= 'z')) - ShortcutLetter = ShortcutLetter - 'a' + 'A'; // convert lowercase to uppercase - Entry->me.ShortcutLetter = ShortcutLetter; - if (Entry->me.Image == NULL) - Entry->me.Image = LoadOSIcon(OSIconName, L"unknown", FALSE); - MyFreePool(PathOnly); + CHAR16 *NameClues, *PathOnly, *NoExtension, *OSIconName = NULL, *Temp; + CHAR16 ShortcutLetter = 0; + + NameClues = Basename(LoaderPath); + PathOnly = FindPath(LoaderPath); + NoExtension = StripEfiExtension(NameClues); + + if (Volume->DiskKind == DISK_KIND_NET) { + MergeStrings(&NameClues, Entry->me.Title, L' '); + } else { + // locate a custom icon for the loader + // Anything found here takes precedence over the "hints" in the OSIconName variable + if (!Entry->me.Image) { + Entry->me.Image = egLoadIconAnyType(Volume->RootDir, PathOnly, NoExtension, GlobalConfig.IconSizes[ICON_SIZE_BIG]); + } + if (!Entry->me.Image) { + Entry->me.Image = egCopyImage(Volume->VolIconImage); + } + + // Begin creating icon "hints" by using last part of directory path leading + // to the loader + Temp = FindLastDirName(LoaderPath); + MergeStrings(&OSIconName, Temp, L','); + MyFreePool(Temp); + Temp = NULL; + if (OSIconName != NULL) { + ShortcutLetter = OSIconName[0]; + } + + // Add every "word" in the volume label, delimited by spaces, dashes (-), or + // underscores (_), to the list of hints to be used in searching for OS + // icons. + MergeWords(&OSIconName, Volume->VolName, L','); + } // if/else network boot + + // detect specific loaders + if (StriSubCmp(L"bzImage", NameClues) || StriSubCmp(L"vmlinuz", NameClues) || StriSubCmp(L"kernel", NameClues)) { + if (Volume->DiskKind != DISK_KIND_NET) { + GuessLinuxDistribution(&OSIconName, Volume, LoaderPath); + Entry->LoadOptions = GetMainLinuxOptions(LoaderPath, Volume); + } + MergeStrings(&OSIconName, L"linux", L','); + Entry->OSType = 'L'; + if (ShortcutLetter == 0) + ShortcutLetter = 'L'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_LINUX; + } else if (StriSubCmp(L"refit", LoaderPath)) { + MergeStrings(&OSIconName, L"refit", L','); + Entry->OSType = 'R'; + ShortcutLetter = 'R'; + } else if (StriSubCmp(L"refind", LoaderPath)) { + MergeStrings(&OSIconName, L"refind", L','); + Entry->OSType = 'R'; + ShortcutLetter = 'R'; + } else if (MyStriCmp(LoaderPath, MACOSX_LOADER_PATH)) { + MergeStrings(&OSIconName, L"mac", L','); + Entry->OSType = 'M'; + ShortcutLetter = 'M'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_OSX; + } else if (MyStriCmp(NameClues, L"diags.efi")) { + MergeStrings(&OSIconName, L"hwtest", L','); + } else if (MyStriCmp(NameClues, L"e.efi") || MyStriCmp(NameClues, L"elilo.efi") || StriSubCmp(L"elilo", NameClues)) { + MergeStrings(&OSIconName, L"elilo,linux", L','); + Entry->OSType = 'E'; + if (ShortcutLetter == 0) + ShortcutLetter = 'L'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_ELILO; + } else if (StriSubCmp(L"grub", NameClues)) { + MergeStrings(&OSIconName, L"grub,linux", L','); + Entry->OSType = 'G'; + ShortcutLetter = 'G'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_GRUB; + } else if (MyStriCmp(NameClues, L"cdboot.efi") || + MyStriCmp(NameClues, L"bootmgr.efi") || + MyStriCmp(NameClues, L"bootmgfw.efi") || + MyStriCmp(NameClues, L"bkpbootmgfw.efi")) { + MergeStrings(&OSIconName, L"win8", L','); + Entry->OSType = 'W'; + ShortcutLetter = 'W'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; + } else if (MyStriCmp(NameClues, L"xom.efi")) { + MergeStrings(&OSIconName, L"xom,win,win8", L','); + Entry->OSType = 'X'; + ShortcutLetter = 'W'; + Entry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_WINDOWS; + } + else if (StriSubCmp(L"ipxe", NameClues)) { + Entry->OSType = 'N'; + ShortcutLetter = 'N'; + MergeStrings(&OSIconName, L"network", L','); + } + + if ((ShortcutLetter >= 'a') && (ShortcutLetter <= 'z')) + ShortcutLetter = ShortcutLetter - 'a' + 'A'; // convert lowercase to uppercase + Entry->me.ShortcutLetter = ShortcutLetter; + if (Entry->me.Image == NULL) + Entry->me.Image = LoadOSIcon(OSIconName, L"unknown", FALSE); + MyFreePool(PathOnly); } // VOID SetLoaderDefaults() // Add a specified EFI boot loader to the list, using automatic settings // for icons, options, etc. -LOADER_ENTRY * AddLoaderEntry(IN CHAR16 *LoaderPath, IN CHAR16 *LoaderTitle, IN REFIT_VOLUME *Volume) { - LOADER_ENTRY *Entry; - - CleanUpPathNameSlashes(LoaderPath); - Entry = InitializeLoaderEntry(NULL); - if (Entry != NULL) { - Entry->Title = StrDuplicate((LoaderTitle != NULL) ? LoaderTitle : LoaderPath); - Entry->me.Title = AllocateZeroPool(sizeof(CHAR16) * 256); - // Extra space at end of Entry->me.Title enables searching on Volume->VolName even if another volume - // name is identical except for something added to the end (e.g., VolB1 vs. VolB12). - SPrint(Entry->me.Title, 255, L"Boot %s from %s ", (LoaderTitle != NULL) ? LoaderTitle : LoaderPath, Volume->VolName); - Entry->me.Row = 0; - Entry->me.BadgeImage = Volume->VolBadgeImage; - if ((LoaderPath != NULL) && (LoaderPath[0] != L'\\')) { - Entry->LoaderPath = StrDuplicate(L"\\"); - } else { - Entry->LoaderPath = NULL; - } - MergeStrings(&(Entry->LoaderPath), LoaderPath, 0); - Entry->VolName = Volume->VolName; - Entry->DevicePath = FileDevicePath(Volume->DeviceHandle, Entry->LoaderPath); - SetLoaderDefaults(Entry, LoaderPath, Volume); - GenerateSubScreen(Entry, Volume); - AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); - } - - return(Entry); +static LOADER_ENTRY * AddLoaderEntry(IN CHAR16 *LoaderPath, IN CHAR16 *LoaderTitle, IN REFIT_VOLUME *Volume, IN BOOLEAN SubScreenReturn) { + LOADER_ENTRY *Entry; + + CleanUpPathNameSlashes(LoaderPath); + Entry = InitializeLoaderEntry(NULL); + if (Entry != NULL) { + Entry->Title = StrDuplicate((LoaderTitle != NULL) ? LoaderTitle : LoaderPath); + Entry->me.Title = AllocateZeroPool(sizeof(CHAR16) * 256); + // Extra space at end of Entry->me.Title enables searching on Volume->VolName even if another volume + // name is identical except for something added to the end (e.g., VolB1 vs. VolB12). + // Note: Volume->VolName will be NULL for network boot programs. + if (Volume->VolName) + SPrint(Entry->me.Title, 255, L"Boot %s from %s ", (LoaderTitle != NULL) ? LoaderTitle : LoaderPath, Volume->VolName); + else + SPrint(Entry->me.Title, 255, L"Boot %s ", (LoaderTitle != NULL) ? LoaderTitle : LoaderPath); + Entry->me.Row = 0; + Entry->me.BadgeImage = Volume->VolBadgeImage; + if ((LoaderPath != NULL) && (LoaderPath[0] != L'\\')) { + Entry->LoaderPath = StrDuplicate(L"\\"); + } else { + Entry->LoaderPath = NULL; + } + MergeStrings(&(Entry->LoaderPath), LoaderPath, 0); + Entry->VolName = Volume->VolName; + Entry->DevicePath = FileDevicePath(Volume->DeviceHandle, Entry->LoaderPath); + SetLoaderDefaults(Entry, LoaderPath, Volume); + GenerateSubScreen(Entry, Volume, SubScreenReturn); + AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); + } + + return(Entry); } // LOADER_ENTRY * AddLoaderEntry() +// Add a Linux kernel as a submenu entry for another (pre-existing) Linux kernel entry. +static VOID AddKernelToSubmenu(LOADER_ENTRY * TargetLoader, CHAR16 *FileName, REFIT_VOLUME *Volume) { + REFIT_FILE *File; + CHAR16 **TokenList = NULL, *InitrdName, *SubmenuName = NULL, *VolName = NULL, *Path = NULL, *Title; + REFIT_MENU_SCREEN *SubScreen; + LOADER_ENTRY *SubEntry; + UINTN TokenCount; + + File = ReadLinuxOptionsFile(TargetLoader->LoaderPath, Volume); + if (File != NULL) { + SubScreen = TargetLoader->me.SubScreen; + InitrdName = FindInitrd(FileName, Volume); + while ((TokenCount = ReadTokenLine(File, &TokenList)) > 1) { + SubEntry = InitializeLoaderEntry(TargetLoader); + SplitPathName(FileName, &VolName, &Path, &SubmenuName); + MergeStrings(&SubmenuName, L": ", '\0'); + MergeStrings(&SubmenuName, TokenList[0] ? StrDuplicate(TokenList[0]) : StrDuplicate(L"Boot Linux"), '\0'); + Title = StrDuplicate(SubmenuName); + LimitStringLength(Title, MAX_LINE_LENGTH); + SubEntry->me.Title = Title; + MyFreePool(SubEntry->LoadOptions); + SubEntry->LoadOptions = AddInitrdToOptions(TokenList[1], InitrdName); + MyFreePool(SubEntry->LoaderPath); + SubEntry->LoaderPath = StrDuplicate(FileName); + CleanUpPathNameSlashes(SubEntry->LoaderPath); + SubEntry->DevicePath = FileDevicePath(Volume->DeviceHandle, SubEntry->LoaderPath); + FreeTokenLine(&TokenList, &TokenCount); + SubEntry->UseGraphicsMode = GlobalConfig.GraphicsFor & GRAPHICS_FOR_LINUX; + AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); + } // while + MyFreePool(VolName); + MyFreePool(Path); + MyFreePool(SubmenuName); + MyFreePool(InitrdName); + MyFreePool(File); + } // if +} // static VOID AddKernelToSubmenu() + // Returns -1 if (Time1 < Time2), +1 if (Time1 > Time2), or 0 if // (Time1 == Time2). Precision is only to the nearest second; since // this is used for sorting boot loader entries, differences smaller // than this are likely to be meaningless (and unlikely!). INTN TimeComp(IN EFI_TIME *Time1, IN EFI_TIME *Time2) { - INT64 Time1InSeconds, Time2InSeconds; - - // Following values are overestimates; I'm assuming 31 days in every month. - // This is fine for the purpose of this function, which is limited - Time1InSeconds = Time1->Second + (Time1->Minute * 60) + (Time1->Hour * 3600) + (Time1->Day * 86400) + - (Time1->Month * 2678400) + ((Time1->Year - 1998) * 32140800); - Time2InSeconds = Time2->Second + (Time2->Minute * 60) + (Time2->Hour * 3600) + (Time2->Day * 86400) + - (Time2->Month * 2678400) + ((Time2->Year - 1998) * 32140800); - if (Time1InSeconds < Time2InSeconds) - return (-1); - else if (Time1InSeconds > Time2InSeconds) - return (1); - - return 0; + INT64 Time1InSeconds, Time2InSeconds; + + // Following values are overestimates; I'm assuming 31 days in every month. + // This is fine for the purpose of this function, which is limited + Time1InSeconds = Time1->Second + (Time1->Minute * 60) + (Time1->Hour * 3600) + (Time1->Day * 86400) + + (Time1->Month * 2678400) + ((Time1->Year - 1998) * 32140800); + Time2InSeconds = Time2->Second + (Time2->Minute * 60) + (Time2->Hour * 3600) + (Time2->Day * 86400) + + (Time2->Month * 2678400) + ((Time2->Year - 1998) * 32140800); + if (Time1InSeconds < Time2InSeconds) + return (-1); + else if (Time1InSeconds > Time2InSeconds) + return (1); + + return 0; } // INTN TimeComp() -// Adds a loader list element, keeping it sorted by date. Returns the new -// first element (the one with the most recent date). +// Adds a loader list element, keeping it sorted by date. EXCEPTION: Fedora's rescue +// kernel, which begins with "vmlinuz-0-rescue," should not be at the top of the list, +// since that will make it the default if kernel folding is enabled, so float it to +// the end. +// Returns the new first element (the one with the most recent date). static struct LOADER_LIST * AddLoaderListEntry(struct LOADER_LIST *LoaderList, struct LOADER_LIST *NewEntry) { - struct LOADER_LIST *LatestEntry, *CurrentEntry, *PrevEntry = NULL; - - LatestEntry = CurrentEntry = LoaderList; - if (LoaderList == NULL) { - LatestEntry = NewEntry; - } else { - while ((CurrentEntry != NULL) && (TimeComp(&(NewEntry->TimeStamp), &(CurrentEntry->TimeStamp)) < 0)) { - PrevEntry = CurrentEntry; - CurrentEntry = CurrentEntry->NextEntry; - } // while - NewEntry->NextEntry = CurrentEntry; - if (PrevEntry == NULL) { - LatestEntry = NewEntry; - } else { - PrevEntry->NextEntry = NewEntry; - } // if/else - } // if/else - return (LatestEntry); + struct LOADER_LIST *LatestEntry, *CurrentEntry, *PrevEntry = NULL; + BOOLEAN LinuxRescue = FALSE; + + LatestEntry = CurrentEntry = LoaderList; + if (LoaderList == NULL) { + LatestEntry = NewEntry; + } else { + if (StriSubCmp(L"vmlinuz-0-rescue", NewEntry->FileName)) + LinuxRescue = TRUE; + while ((CurrentEntry != NULL) && (LinuxRescue || (TimeComp(&(NewEntry->TimeStamp), &(CurrentEntry->TimeStamp)) < 0))) { + PrevEntry = CurrentEntry; + CurrentEntry = CurrentEntry->NextEntry; + } // while + NewEntry->NextEntry = CurrentEntry; + if (PrevEntry == NULL) { + LatestEntry = NewEntry; + } else { + PrevEntry->NextEntry = NewEntry; + } // if/else + } // if/else + return (LatestEntry); } // static VOID AddLoaderListEntry() // Delete the LOADER_LIST linked list static VOID CleanUpLoaderList(struct LOADER_LIST *LoaderList) { - struct LOADER_LIST *Temp; - - while (LoaderList != NULL) { - Temp = LoaderList; - LoaderList = LoaderList->NextEntry; - MyFreePool(Temp->FileName); - MyFreePool(Temp); - } // while + struct LOADER_LIST *Temp; + + while (LoaderList != NULL) { + Temp = LoaderList; + LoaderList = LoaderList->NextEntry; + MyFreePool(Temp->FileName); + MyFreePool(Temp); + } // while } // static VOID CleanUpLoaderList() // Returns FALSE if the specified file/volume matches the GlobalConfig.DontScanDirs @@ -1087,47 +1219,47 @@ static VOID CleanUpLoaderList(struct LOADER_LIST *LoaderList) { // Returns TRUE if none of these conditions is met -- that is, if the path is // eligible for scanning. static BOOLEAN ShouldScan(REFIT_VOLUME *Volume, CHAR16 *Path) { - CHAR16 *VolName = NULL, *DontScanDir, *PathCopy = NULL; - UINTN i = 0; - BOOLEAN ScanIt = TRUE; - - if ((IsIn(Volume->VolName, GlobalConfig.DontScanVolumes)) || (IsIn(Volume->PartName, GlobalConfig.DontScanVolumes))) - return FALSE; - - if ((StriCmp(Path, SelfDirPath) == 0) && (Volume->DeviceHandle == SelfVolume->DeviceHandle)) - return FALSE; - - // See if Path includes an explicit volume declaration that's NOT Volume.... - PathCopy = StrDuplicate(Path); - if (SplitVolumeAndFilename(&PathCopy, &VolName)) { - VolumeNumberToName(Volume, &VolName); - if (VolName && StriCmp(VolName, Volume->VolName) != 0) { - ScanIt = FALSE; - } // if - } // if Path includes volume specification - MyFreePool(PathCopy); - MyFreePool(VolName); - VolName = NULL; - - // See if Volume is in GlobalConfig.DontScanDirs.... - while (ScanIt && (DontScanDir = FindCommaDelimited(GlobalConfig.DontScanDirs, i++))) { - SplitVolumeAndFilename(&DontScanDir, &VolName); - CleanUpPathNameSlashes(DontScanDir); - VolumeNumberToName(Volume, &VolName); - if (VolName != NULL) { - if ((StriCmp(VolName, Volume->VolName) == 0) && (StriCmp(DontScanDir, Path) == 0)) - ScanIt = FALSE; - } else { - if (StriCmp(DontScanDir, Path) == 0) + CHAR16 *VolName = NULL, *DontScanDir, *PathCopy = NULL; + UINTN i = 0; + BOOLEAN ScanIt = TRUE; + + if ((IsIn(Volume->VolName, GlobalConfig.DontScanVolumes)) || (IsIn(Volume->PartName, GlobalConfig.DontScanVolumes))) + return FALSE; + + if (MyStriCmp(Path, SelfDirPath) && (Volume->DeviceHandle == SelfVolume->DeviceHandle)) + return FALSE; + + // See if Path includes an explicit volume declaration that's NOT Volume.... + PathCopy = StrDuplicate(Path); + if (SplitVolumeAndFilename(&PathCopy, &VolName)) { + VolumeNumberToName(Volume, &VolName); + if (VolName && !MyStriCmp(VolName, Volume->VolName)) { ScanIt = FALSE; - } - MyFreePool(DontScanDir); - MyFreePool(VolName); - DontScanDir = NULL; - VolName = NULL; - } // while() - - return ScanIt; + } // if + } // if Path includes volume specification + MyFreePool(PathCopy); + MyFreePool(VolName); + VolName = NULL; + + // See if Volume is in GlobalConfig.DontScanDirs.... + while (ScanIt && (DontScanDir = FindCommaDelimited(GlobalConfig.DontScanDirs, i++))) { + SplitVolumeAndFilename(&DontScanDir, &VolName); + CleanUpPathNameSlashes(DontScanDir); + VolumeNumberToName(Volume, &VolName); + if (VolName != NULL) { + if (MyStriCmp(VolName, Volume->VolName) && MyStriCmp(DontScanDir, Path)) + ScanIt = FALSE; + } else { + if (MyStriCmp(DontScanDir, Path)) + ScanIt = FALSE; + } + MyFreePool(DontScanDir); + MyFreePool(VolName); + DontScanDir = NULL; + VolName = NULL; + } // while() + + return ScanIt; } // BOOLEAN ShouldScan() // Returns TRUE if the file is byte-for-byte identical with the fallback file @@ -1136,61 +1268,61 @@ static BOOLEAN ShouldScan(REFIT_VOLUME *Volume, CHAR16 *Path) { // IS the fallback file. Intended for use in excluding the fallback boot // loader when it's a duplicate of another boot loader. static BOOLEAN DuplicatesFallback(IN REFIT_VOLUME *Volume, IN CHAR16 *FileName) { - CHAR8 *FileContents, *FallbackContents; - EFI_FILE_HANDLE FileHandle, FallbackHandle; - EFI_FILE_INFO *FileInfo, *FallbackInfo; - UINTN FileSize = 0, FallbackSize = 0; - EFI_STATUS Status; - BOOLEAN AreIdentical = FALSE; - - if (!FileExists(Volume->RootDir, FileName) || !FileExists(Volume->RootDir, FALLBACK_FULLNAME)) - return FALSE; - - CleanUpPathNameSlashes(FileName); - - if (StriCmp(FileName, FALLBACK_FULLNAME) == 0) - return FALSE; // identical filenames, so not a duplicate.... - - Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0); - if (Status == EFI_SUCCESS) { - FileInfo = LibFileInfo(FileHandle); - FileSize = FileInfo->FileSize; - } else { - return FALSE; - } - - Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FallbackHandle, FALLBACK_FULLNAME, EFI_FILE_MODE_READ, 0); - if (Status == EFI_SUCCESS) { - FallbackInfo = LibFileInfo(FallbackHandle); - FallbackSize = FallbackInfo->FileSize; - } else { - refit_call1_wrapper(FileHandle->Close, FileHandle); - return FALSE; - } - - if (FallbackSize != FileSize) { // not same size, so can't be identical - AreIdentical = FALSE; - } else { // could be identical; do full check.... - FileContents = AllocatePool(FileSize); - FallbackContents = AllocatePool(FallbackSize); - if (FileContents && FallbackContents) { - Status = refit_call3_wrapper(FileHandle->Read, FileHandle, &FileSize, FileContents); - if (Status == EFI_SUCCESS) { - Status = refit_call3_wrapper(FallbackHandle->Read, FallbackHandle, &FallbackSize, FallbackContents); - } - if (Status == EFI_SUCCESS) { - AreIdentical = (CompareMem(FileContents, FallbackContents, FileSize) == 0); - } // if - } // if - MyFreePool(FileContents); - MyFreePool(FallbackContents); - } // if/else - - // BUG ALERT: Some systems (e.g., DUET, some Macs with large displays) crash if the - // following two calls are reversed. Go figure.... - refit_call1_wrapper(FileHandle->Close, FallbackHandle); - refit_call1_wrapper(FileHandle->Close, FileHandle); - return AreIdentical; + CHAR8 *FileContents, *FallbackContents; + EFI_FILE_HANDLE FileHandle, FallbackHandle; + EFI_FILE_INFO *FileInfo, *FallbackInfo; + UINTN FileSize = 0, FallbackSize = 0; + EFI_STATUS Status; + BOOLEAN AreIdentical = FALSE; + + if (!FileExists(Volume->RootDir, FileName) || !FileExists(Volume->RootDir, FALLBACK_FULLNAME)) + return FALSE; + + CleanUpPathNameSlashes(FileName); + + if (MyStriCmp(FileName, FALLBACK_FULLNAME)) + return FALSE; // identical filenames, so not a duplicate.... + + Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0); + if (Status == EFI_SUCCESS) { + FileInfo = LibFileInfo(FileHandle); + FileSize = FileInfo->FileSize; + } else { + return FALSE; + } + + Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FallbackHandle, FALLBACK_FULLNAME, EFI_FILE_MODE_READ, 0); + if (Status == EFI_SUCCESS) { + FallbackInfo = LibFileInfo(FallbackHandle); + FallbackSize = FallbackInfo->FileSize; + } else { + refit_call1_wrapper(FileHandle->Close, FileHandle); + return FALSE; + } + + if (FallbackSize != FileSize) { // not same size, so can't be identical + AreIdentical = FALSE; + } else { // could be identical; do full check.... + FileContents = AllocatePool(FileSize); + FallbackContents = AllocatePool(FallbackSize); + if (FileContents && FallbackContents) { + Status = refit_call3_wrapper(FileHandle->Read, FileHandle, &FileSize, FileContents); + if (Status == EFI_SUCCESS) { + Status = refit_call3_wrapper(FallbackHandle->Read, FallbackHandle, &FallbackSize, FallbackContents); + } + if (Status == EFI_SUCCESS) { + AreIdentical = (CompareMem(FileContents, FallbackContents, FileSize) == 0); + } // if + } // if + MyFreePool(FileContents); + MyFreePool(FallbackContents); + } // if/else + + // BUG ALERT: Some systems (e.g., DUET, some Macs with large displays) crash if the + // following two calls are reversed. Go figure.... + refit_call1_wrapper(FileHandle->Close, FallbackHandle); + refit_call1_wrapper(FileHandle->Close, FileHandle); + return AreIdentical; } // BOOLEAN DuplicatesFallback() // Returns FALSE if two measures of file size are identical for a single file, @@ -1202,27 +1334,27 @@ static BOOLEAN DuplicatesFallback(IN REFIT_VOLUME *Volume, IN CHAR16 *FileName) // this function to exclude symbolic links from the list of boot loaders, // that would be fine, since such boot loaders wouldn't work.) static BOOLEAN IsSymbolicLink(REFIT_VOLUME *Volume, CHAR16 *Path, EFI_FILE_INFO *DirEntry) { - EFI_FILE_HANDLE FileHandle; - EFI_FILE_INFO *FileInfo = NULL; - EFI_STATUS Status; - UINTN FileSize2 = 0; - CHAR16 *FileName; - - FileName = StrDuplicate(Path); - MergeStrings(&FileName, DirEntry->FileName, L'\\'); - CleanUpPathNameSlashes(FileName); - - Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0); - if (Status == EFI_SUCCESS) { - FileInfo = LibFileInfo(FileHandle); - if (FileInfo != NULL) - FileSize2 = FileInfo->FileSize; - } - - MyFreePool(FileName); - MyFreePool(FileInfo); - - return (DirEntry->FileSize != FileSize2); + EFI_FILE_HANDLE FileHandle; + EFI_FILE_INFO *FileInfo = NULL; + EFI_STATUS Status; + UINTN FileSize2 = 0; + CHAR16 *FileName; + + FileName = StrDuplicate(Path); + MergeStrings(&FileName, DirEntry->FileName, L'\\'); + CleanUpPathNameSlashes(FileName); + + Status = refit_call5_wrapper(Volume->RootDir->Open, Volume->RootDir, &FileHandle, FileName, EFI_FILE_MODE_READ, 0); + if (Status == EFI_SUCCESS) { + FileInfo = LibFileInfo(FileHandle); + if (FileInfo != NULL) + FileSize2 = FileInfo->FileSize; + } + + MyFreePool(FileName); + MyFreePool(FileInfo); + + return (DirEntry->FileSize != FileSize2); } // BOOLEAN IsSymbolicLink() // Returns TRUE if a file with the same name as the original but with @@ -1232,20 +1364,20 @@ static BOOLEAN IsSymbolicLink(REFIT_VOLUME *Volume, CHAR16 *Path, EFI_FILE_INFO // behave identically on non-SB systems, or when one will fail when SB // is active. static BOOLEAN HasSignedCounterpart(IN REFIT_VOLUME *Volume, IN CHAR16 *Path, IN CHAR16 *Filename) { - CHAR16 *NewFile = NULL; - BOOLEAN retval = FALSE; - - MergeStrings(&NewFile, Path, 0); - MergeStrings(&NewFile, Filename, L'\\'); - MergeStrings(&NewFile, L".efi.signed", 0); - if (NewFile != NULL) { - CleanUpPathNameSlashes(NewFile); - if (FileExists(Volume->RootDir, NewFile)) - retval = TRUE; - MyFreePool(NewFile); - } // if - - return retval; + CHAR16 *NewFile = NULL; + BOOLEAN retval = FALSE; + + MergeStrings(&NewFile, Path, 0); + MergeStrings(&NewFile, Filename, L'\\'); + MergeStrings(&NewFile, L".efi.signed", 0); + if (NewFile != NULL) { + CleanUpPathNameSlashes(NewFile); + if (FileExists(Volume->RootDir, NewFile)) + retval = TRUE; + MyFreePool(NewFile); + } // if + + return retval; } // BOOLEAN HasSignedCounterpart() // Scan an individual directory for EFI boot loader files and, if found, @@ -1260,23 +1392,26 @@ static BOOLEAN ScanLoaderDir(IN REFIT_VOLUME *Volume, IN CHAR16 *Path, IN CHAR16 EFI_FILE_INFO *DirEntry; CHAR16 FileName[256], *Extension; struct LOADER_LIST *LoaderList = NULL, *NewLoader; - BOOLEAN FoundFallbackDuplicate = FALSE; + LOADER_ENTRY *FirstKernel = NULL, *LatestEntry = NULL; + BOOLEAN FoundFallbackDuplicate = FALSE, IsLinux = FALSE, InSelfPath; - if ((!SelfDirPath || !Path || ((StriCmp(Path, SelfDirPath) == 0) && (Volume->DeviceHandle != SelfVolume->DeviceHandle)) || - (StriCmp(Path, SelfDirPath) != 0)) && (ShouldScan(Volume, Path))) { + InSelfPath = MyStriCmp(Path, SelfDirPath); + if ((!SelfDirPath || !Path || (InSelfPath && (Volume->DeviceHandle != SelfVolume->DeviceHandle)) || + (!InSelfPath)) && (ShouldScan(Volume, Path))) { // look through contents of the directory DirIterOpen(Volume->RootDir, Path, &DirIter); while (DirIterNext(&DirIter, 2, Pattern, &DirEntry)) { Extension = FindExtension(DirEntry->FileName); if (DirEntry->FileName[0] == '.' || - StriCmp(Extension, L".icns") == 0 || - StriCmp(Extension, L".png") == 0 || - (StriCmp(DirEntry->FileName, FALLBACK_BASENAME) == 0 && (StriCmp(Path, L"EFI\\BOOT") == 0)) || - StriSubCmp(L"shell", DirEntry->FileName) || + MyStriCmp(Extension, L".icns") || + MyStriCmp(Extension, L".png") || + (MyStriCmp(DirEntry->FileName, FALLBACK_BASENAME) && (MyStriCmp(Path, L"EFI\\BOOT"))) || + FilenameIn(Volume, Path, DirEntry->FileName, SHELL_NAMES) || IsSymbolicLink(Volume, Path, DirEntry) || /* is symbolic link */ HasSignedCounterpart(Volume, Path, DirEntry->FileName) || /* a file with same name plus ".efi.signed" is present */ - FilenameIn(Volume, Path, DirEntry->FileName, GlobalConfig.DontScanFiles)) + FilenameIn(Volume, Path, DirEntry->FileName, GlobalConfig.DontScanFiles)) { continue; // skip this + } if (Path) SPrint(FileName, 255, L"\\%s\\%s", Path, DirEntry->FileName); @@ -1300,9 +1435,20 @@ static BOOLEAN ScanLoaderDir(IN REFIT_VOLUME *Volume, IN CHAR16 *Path, IN CHAR16 NewLoader = LoaderList; while (NewLoader != NULL) { - AddLoaderEntry(NewLoader->FileName, NULL, Volume); - NewLoader = NewLoader->NextEntry; + IsLinux = (StriSubCmp(L"bzImage", NewLoader->FileName) || + StriSubCmp(L"vmlinuz", NewLoader->FileName) || + StriSubCmp(L"kernel", NewLoader->FileName)); + if ((FirstKernel != NULL) && IsLinux && GlobalConfig.FoldLinuxKernels) { + AddKernelToSubmenu(FirstKernel, NewLoader->FileName, Volume); + } else { + LatestEntry = AddLoaderEntry(NewLoader->FileName, NULL, Volume, !(IsLinux && GlobalConfig.FoldLinuxKernels)); + if (IsLinux && (FirstKernel == NULL)) + FirstKernel = LatestEntry; + } + NewLoader = NewLoader->NextEntry; } // while + if ((NewLoader != NULL) && (FirstKernel != NULL) && IsLinux && GlobalConfig.FoldLinuxKernels) + AddMenuEntry(FirstKernel->me.SubScreen, &MenuEntryReturn); CleanUpLoaderList(LoaderList); Status = DirIterClose(&DirIter); @@ -1322,652 +1468,201 @@ static BOOLEAN ScanLoaderDir(IN REFIT_VOLUME *Volume, IN CHAR16 *Path, IN CHAR16 return FoundFallbackDuplicate; } /* static VOID ScanLoaderDir() */ +// Run the IPXE_DISCOVER_NAME program, which obtains the IP address of the boot +// server and the name of the boot file it delivers. +CHAR16* RuniPXEDiscover(EFI_HANDLE Volume) +{ + EFI_STATUS Status; + EFI_DEVICE_PATH *FilePath; + EFI_HANDLE iPXEHandle; + CHAR16 *boot_info = NULL; + UINTN boot_info_size = 0; + + FilePath = FileDevicePath (Volume, IPXE_DISCOVER_NAME); + Status = refit_call6_wrapper(BS->LoadImage, FALSE, SelfImageHandle, FilePath, NULL, 0, &iPXEHandle); + if (Status != 0) + return NULL; + + Status = refit_call3_wrapper(BS->StartImage, iPXEHandle, &boot_info_size, &boot_info); + + return boot_info; +} // RuniPXEDiscover() + +// Scan for network (PXE) boot servers. This function relies on the presence +// of the IPXE_DISCOVER_NAME and IPXE_NAME program files on the volume from +// which rEFInd launched. As of December 6, 2014, these tools aren't entirely +// reliable. See BUILDING.txt for information on building them. +static VOID ScanNetboot() { + CHAR16 *iPXEFileName = IPXE_NAME; + CHAR16 *Location; + REFIT_VOLUME *NetVolume; + + if (FileExists(SelfVolume->RootDir, IPXE_DISCOVER_NAME) && + FileExists(SelfVolume->RootDir, IPXE_NAME) && + IsValidLoader(SelfVolume->RootDir, IPXE_DISCOVER_NAME) && + IsValidLoader(SelfVolume->RootDir, IPXE_NAME)) { + Location = RuniPXEDiscover(SelfVolume->DeviceHandle); + if (Location != NULL && FileExists(SelfVolume->RootDir, iPXEFileName)) { + NetVolume = AllocatePool(sizeof(REFIT_VOLUME)); + CopyMem(NetVolume, SelfVolume, sizeof(REFIT_VOLUME)); + NetVolume->DiskKind = DISK_KIND_NET; + NetVolume->VolBadgeImage = BuiltinIcon(BUILTIN_ICON_VOL_NET); + NetVolume->PartName = NetVolume->VolName = NULL; + AddLoaderEntry(iPXEFileName, Location, NetVolume, TRUE); + MyFreePool(NetVolume); + } // if support files exist and are valid + } +} // VOID ScanNetBoot() + static VOID ScanEfiFiles(REFIT_VOLUME *Volume) { - EFI_STATUS Status; - REFIT_DIR_ITER EfiDirIter; - EFI_FILE_INFO *EfiDirEntry; - CHAR16 FileName[256], *Directory = NULL, *MatchPatterns, *VolName = NULL, *SelfPath; - UINTN i, Length; - BOOLEAN ScanFallbackLoader = TRUE; - BOOLEAN FoundBRBackup = FALSE; - - if ((Volume->RootDir != NULL) && (Volume->VolName != NULL) && (Volume->IsReadable)) { - MatchPatterns = StrDuplicate(LOADER_MATCH_PATTERNS); - if (GlobalConfig.ScanAllLinux) - MergeStrings(&MatchPatterns, LINUX_MATCH_PATTERNS, L','); - - // check for Mac OS X boot loader - if (ShouldScan(Volume, L"System\\Library\\CoreServices")) { - StrCpy(FileName, MACOSX_LOADER_PATH); - if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, Directory, L"boot.efi", GlobalConfig.DontScanFiles)) { - AddLoaderEntry(FileName, L"Mac OS X", Volume); - if (DuplicatesFallback(Volume, FileName)) - ScanFallbackLoader = FALSE; - } - - // check for XOM - StrCpy(FileName, L"System\\Library\\CoreServices\\xom.efi"); - if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, Directory, L"boot.efi", GlobalConfig.DontScanFiles)) { - AddLoaderEntry(FileName, L"Windows XP (XoM)", Volume); - if (DuplicatesFallback(Volume, FileName)) - ScanFallbackLoader = FALSE; - } - } // if should scan Mac directory - - // check for Microsoft boot loader/menu - if (ShouldScan(Volume, L"EFI\\Microsoft\\Boot")) { - StrCpy(FileName, L"EFI\\Microsoft\\Boot\\bkpbootmgfw.efi"); - if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, Directory, L"bkpbootmgfw.efi", - GlobalConfig.DontScanFiles)) { - AddLoaderEntry(FileName, L"Microsoft EFI boot (Boot Repair backup)", Volume); - FoundBRBackup = TRUE; - if (DuplicatesFallback(Volume, FileName)) - ScanFallbackLoader = FALSE; - } - StrCpy(FileName, L"EFI\\Microsoft\\Boot\\bootmgfw.efi"); - if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, Directory, L"bootmgfw.efi", GlobalConfig.DontScanFiles)) { - if (FoundBRBackup) - AddLoaderEntry(FileName, L"Supposed Microsoft EFI boot (probably GRUB)", Volume); - else - AddLoaderEntry(FileName, L"Microsoft EFI boot", Volume); - if (DuplicatesFallback(Volume, FileName)) - ScanFallbackLoader = FALSE; - } - } // if - - // scan the root directory for EFI executables - if (ScanLoaderDir(Volume, L"\\", MatchPatterns)) - ScanFallbackLoader = FALSE; - - // scan subdirectories of the EFI directory (as per the standard) - DirIterOpen(Volume->RootDir, L"EFI", &EfiDirIter); - while (DirIterNext(&EfiDirIter, 1, NULL, &EfiDirEntry)) { - if (StriCmp(EfiDirEntry->FileName, L"tools") == 0 || EfiDirEntry->FileName[0] == '.') - continue; // skip this, doesn't contain boot loaders or is scanned later - SPrint(FileName, 255, L"EFI\\%s", EfiDirEntry->FileName); - if (ScanLoaderDir(Volume, FileName, MatchPatterns)) + EFI_STATUS Status; + REFIT_DIR_ITER EfiDirIter; + EFI_FILE_INFO *EfiDirEntry; + CHAR16 FileName[256], *Directory = NULL, *MatchPatterns, *VolName = NULL, *SelfPath; + UINTN i, Length; + BOOLEAN ScanFallbackLoader = TRUE; + BOOLEAN FoundBRBackup = FALSE; + + if (Volume && (Volume->RootDir != NULL) && (Volume->VolName != NULL) && (Volume->IsReadable)) { + MatchPatterns = StrDuplicate(LOADER_MATCH_PATTERNS); + if (GlobalConfig.ScanAllLinux) + MergeStrings(&MatchPatterns, LINUX_MATCH_PATTERNS, L','); + + // check for Mac OS X boot loader + if (ShouldScan(Volume, MACOSX_LOADER_DIR)) { + StrCpy(FileName, MACOSX_LOADER_PATH); + if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, MACOSX_LOADER_DIR, L"boot.efi", GlobalConfig.DontScanFiles)) { + AddLoaderEntry(FileName, L"Mac OS X", Volume, TRUE); + if (DuplicatesFallback(Volume, FileName)) + ScanFallbackLoader = FALSE; + } + + // check for XOM + StrCpy(FileName, L"System\\Library\\CoreServices\\xom.efi"); + if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, MACOSX_LOADER_DIR, L"xom.efi", GlobalConfig.DontScanFiles)) { + AddLoaderEntry(FileName, L"Windows XP (XoM)", Volume, TRUE); + if (DuplicatesFallback(Volume, FileName)) + ScanFallbackLoader = FALSE; + } + } // if should scan Mac directory + + // check for Microsoft boot loader/menu + if (ShouldScan(Volume, L"EFI\\Microsoft\\Boot")) { + StrCpy(FileName, L"EFI\\Microsoft\\Boot\\bkpbootmgfw.efi"); + if (FileExists(Volume->RootDir, FileName) && !FilenameIn(Volume, L"EFI\\Microsoft\\Boot", L"bkpbootmgfw.efi", + GlobalConfig.DontScanFiles)) { + AddLoaderEntry(FileName, L"Microsoft EFI boot (Boot Repair backup)", Volume, TRUE); + FoundBRBackup = TRUE; + if (DuplicatesFallback(Volume, FileName)) + ScanFallbackLoader = FALSE; + } + StrCpy(FileName, L"EFI\\Microsoft\\Boot\\bootmgfw.efi"); + if (FileExists(Volume->RootDir, FileName) && + !FilenameIn(Volume, L"EFI\\Microsoft\\Boot", L"bootmgfw.efi", GlobalConfig.DontScanFiles)) { + if (FoundBRBackup) + AddLoaderEntry(FileName, L"Supposed Microsoft EFI boot (probably GRUB)", Volume, TRUE); + else + AddLoaderEntry(FileName, L"Microsoft EFI boot", Volume, TRUE); + if (DuplicatesFallback(Volume, FileName)) + ScanFallbackLoader = FALSE; + } + } // if + + // scan the root directory for EFI executables + if (ScanLoaderDir(Volume, L"\\", MatchPatterns)) + ScanFallbackLoader = FALSE; + + // scan subdirectories of the EFI directory (as per the standard) + DirIterOpen(Volume->RootDir, L"EFI", &EfiDirIter); + while (DirIterNext(&EfiDirIter, 1, NULL, &EfiDirEntry)) { + if (MyStriCmp(EfiDirEntry->FileName, L"tools") || EfiDirEntry->FileName[0] == '.') + continue; // skip this, doesn't contain boot loaders or is scanned later + SPrint(FileName, 255, L"EFI\\%s", EfiDirEntry->FileName); + if (ScanLoaderDir(Volume, FileName, MatchPatterns)) + ScanFallbackLoader = FALSE; + } // while() + Status = DirIterClose(&EfiDirIter); + if ((Status != EFI_NOT_FOUND) && (Status != EFI_INVALID_PARAMETER)) + CheckError(Status, L"while scanning the EFI directory"); + + // Scan user-specified (or additional default) directories.... + i = 0; + while ((Directory = FindCommaDelimited(GlobalConfig.AlsoScan, i++)) != NULL) { + if (ShouldScan(Volume, Directory)) { + SplitVolumeAndFilename(&Directory, &VolName); + CleanUpPathNameSlashes(Directory); + Length = StrLen(Directory); + if ((Length > 0) && ScanLoaderDir(Volume, Directory, MatchPatterns)) + ScanFallbackLoader = FALSE; + MyFreePool(VolName); + } // if should scan dir + MyFreePool(Directory); + } // while + + // Don't scan the fallback loader if it's on the same volume and a duplicate of rEFInd itself.... + SelfPath = DevicePathToStr(SelfLoadedImage->FilePath); + CleanUpPathNameSlashes(SelfPath); + if ((Volume->DeviceHandle == SelfLoadedImage->DeviceHandle) && DuplicatesFallback(Volume, SelfPath)) ScanFallbackLoader = FALSE; - } // while() - Status = DirIterClose(&EfiDirIter); - if (Status != EFI_NOT_FOUND) - CheckError(Status, L"while scanning the EFI directory"); - - // Scan user-specified (or additional default) directories.... - i = 0; - while ((Directory = FindCommaDelimited(GlobalConfig.AlsoScan, i++)) != NULL) { - if (ShouldScan(Volume, Directory)) { - SplitVolumeAndFilename(&Directory, &VolName); - CleanUpPathNameSlashes(Directory); - Length = StrLen(Directory); - if ((Length > 0) && ScanLoaderDir(Volume, Directory, MatchPatterns)) - ScanFallbackLoader = FALSE; - MyFreePool(VolName); - } // if should scan dir - MyFreePool(Directory); - } // while - - // Don't scan the fallback loader if it's on the same volume and a duplicate of rEFInd itself.... - SelfPath = DevicePathToStr(SelfLoadedImage->FilePath); - CleanUpPathNameSlashes(SelfPath); - if ((Volume->DeviceHandle == SelfLoadedImage->DeviceHandle) && DuplicatesFallback(Volume, SelfPath)) - ScanFallbackLoader = FALSE; - - // If not a duplicate & if it exists & if it's not us, create an entry - // for the fallback boot loader - if (ScanFallbackLoader && FileExists(Volume->RootDir, FALLBACK_FULLNAME) && ShouldScan(Volume, L"EFI\\BOOT")) - AddLoaderEntry(FALLBACK_FULLNAME, L"Fallback boot loader", Volume); - } // if + + // If not a duplicate & if it exists & if it's not us, create an entry + // for the fallback boot loader + if (ScanFallbackLoader && FileExists(Volume->RootDir, FALLBACK_FULLNAME) && ShouldScan(Volume, L"EFI\\BOOT")) { + AddLoaderEntry(FALLBACK_FULLNAME, L"Fallback boot loader", Volume, TRUE); + } + } // if } // static VOID ScanEfiFiles() // Scan internal disks for valid EFI boot loaders.... static VOID ScanInternal(VOID) { - UINTN VolumeIndex; + UINTN VolumeIndex; - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_INTERNAL) { - ScanEfiFiles(Volumes[VolumeIndex]); - } - } // for + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_INTERNAL) { + ScanEfiFiles(Volumes[VolumeIndex]); + } + } // for } // static VOID ScanInternal() // Scan external disks for valid EFI boot loaders.... static VOID ScanExternal(VOID) { - UINTN VolumeIndex; + UINTN VolumeIndex; - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_EXTERNAL) { - ScanEfiFiles(Volumes[VolumeIndex]); - } - } // for + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_EXTERNAL) { + ScanEfiFiles(Volumes[VolumeIndex]); + } + } // for } // static VOID ScanExternal() // Scan internal disks for valid EFI boot loaders.... static VOID ScanOptical(VOID) { - UINTN VolumeIndex; - - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_OPTICAL) { - ScanEfiFiles(Volumes[VolumeIndex]); - } - } // for -} // static VOID ScanOptical() - -// -// legacy boot functions -// - -static EFI_STATUS ActivateMbrPartition(IN EFI_BLOCK_IO *BlockIO, IN UINTN PartitionIndex) -{ - EFI_STATUS Status; - UINT8 SectorBuffer[512]; - MBR_PARTITION_INFO *MbrTable, *EMbrTable; - UINT32 ExtBase, ExtCurrent, NextExtCurrent; - UINTN LogicalPartitionIndex = 4; - UINTN i; - BOOLEAN HaveBootCode; - - // read MBR - Status = refit_call5_wrapper(BlockIO->ReadBlocks, BlockIO, BlockIO->Media->MediaId, 0, 512, SectorBuffer); - if (EFI_ERROR(Status)) - return Status; - if (*((UINT16 *)(SectorBuffer + 510)) != 0xaa55) - return EFI_NOT_FOUND; // safety measure #1 - - // add boot code if necessary - HaveBootCode = FALSE; - for (i = 0; i < MBR_BOOTCODE_SIZE; i++) { - if (SectorBuffer[i] != 0) { - HaveBootCode = TRUE; - break; - } - } - if (!HaveBootCode) { - // no boot code found in the MBR, add the syslinux MBR code - SetMem(SectorBuffer, MBR_BOOTCODE_SIZE, 0); - CopyMem(SectorBuffer, syslinux_mbr, SYSLINUX_MBR_SIZE); - } - - // set the partition active - MbrTable = (MBR_PARTITION_INFO *)(SectorBuffer + 446); - ExtBase = 0; - for (i = 0; i < 4; i++) { - if (MbrTable[i].Flags != 0x00 && MbrTable[i].Flags != 0x80) - return EFI_NOT_FOUND; // safety measure #2 - if (i == PartitionIndex) - MbrTable[i].Flags = 0x80; - else if (PartitionIndex >= 4 && IS_EXTENDED_PART_TYPE(MbrTable[i].Type)) { - MbrTable[i].Flags = 0x80; - ExtBase = MbrTable[i].StartLBA; - } else - MbrTable[i].Flags = 0x00; - } - - // write MBR - Status = refit_call5_wrapper(BlockIO->WriteBlocks, BlockIO, BlockIO->Media->MediaId, 0, 512, SectorBuffer); - if (EFI_ERROR(Status)) - return Status; - - if (PartitionIndex >= 4) { - // we have to activate a logical partition, so walk the EMBR chain - - // NOTE: ExtBase was set above while looking at the MBR table - for (ExtCurrent = ExtBase; ExtCurrent; ExtCurrent = NextExtCurrent) { - // read current EMBR - Status = refit_call5_wrapper(BlockIO->ReadBlocks, BlockIO, BlockIO->Media->MediaId, ExtCurrent, 512, SectorBuffer); - if (EFI_ERROR(Status)) - return Status; - if (*((UINT16 *)(SectorBuffer + 510)) != 0xaa55) - return EFI_NOT_FOUND; // safety measure #3 - - // scan EMBR, set appropriate partition active - EMbrTable = (MBR_PARTITION_INFO *)(SectorBuffer + 446); - NextExtCurrent = 0; - for (i = 0; i < 4; i++) { - if (EMbrTable[i].Flags != 0x00 && EMbrTable[i].Flags != 0x80) - return EFI_NOT_FOUND; // safety measure #4 - if (EMbrTable[i].StartLBA == 0 || EMbrTable[i].Size == 0) - break; - if (IS_EXTENDED_PART_TYPE(EMbrTable[i].Type)) { - // link to next EMBR - NextExtCurrent = ExtBase + EMbrTable[i].StartLBA; - EMbrTable[i].Flags = (PartitionIndex >= LogicalPartitionIndex) ? 0x80 : 0x00; - break; - } else { - // logical partition - EMbrTable[i].Flags = (PartitionIndex == LogicalPartitionIndex) ? 0x80 : 0x00; - LogicalPartitionIndex++; - } - } - - // write current EMBR - Status = refit_call5_wrapper(BlockIO->WriteBlocks, BlockIO, BlockIO->Media->MediaId, ExtCurrent, 512, SectorBuffer); - if (EFI_ERROR(Status)) - return Status; - - if (PartitionIndex < LogicalPartitionIndex) - break; // stop the loop, no need to touch further EMBRs - } - - } - - return EFI_SUCCESS; -} /* static EFI_STATUS ActivateMbrPartition() */ - -// early 2006 Core Duo / Core Solo models -static UINT8 LegacyLoaderDevicePath1Data[] = { - 0x01, 0x03, 0x18, 0x00, 0x0B, 0x00, 0x00, 0x00, - 0x00, 0x00, 0xE0, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xFF, 0xF9, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0x04, 0x06, 0x14, 0x00, 0xEB, 0x85, 0x05, 0x2B, - 0xB8, 0xD8, 0xA9, 0x49, 0x8B, 0x8C, 0xE2, 0x1B, - 0x01, 0xAE, 0xF2, 0xB7, 0x7F, 0xFF, 0x04, 0x00, -}; -// mid-2006 Mac Pro (and probably other Core 2 models) -static UINT8 LegacyLoaderDevicePath2Data[] = { - 0x01, 0x03, 0x18, 0x00, 0x0B, 0x00, 0x00, 0x00, - 0x00, 0x00, 0xE0, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xFF, 0xF7, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0x04, 0x06, 0x14, 0x00, 0xEB, 0x85, 0x05, 0x2B, - 0xB8, 0xD8, 0xA9, 0x49, 0x8B, 0x8C, 0xE2, 0x1B, - 0x01, 0xAE, 0xF2, 0xB7, 0x7F, 0xFF, 0x04, 0x00, -}; -// mid-2007 MBP ("Santa Rosa" based models) -static UINT8 LegacyLoaderDevicePath3Data[] = { - 0x01, 0x03, 0x18, 0x00, 0x0B, 0x00, 0x00, 0x00, - 0x00, 0x00, 0xE0, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xFF, 0xF8, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0x04, 0x06, 0x14, 0x00, 0xEB, 0x85, 0x05, 0x2B, - 0xB8, 0xD8, 0xA9, 0x49, 0x8B, 0x8C, 0xE2, 0x1B, - 0x01, 0xAE, 0xF2, 0xB7, 0x7F, 0xFF, 0x04, 0x00, -}; -// early-2008 MBA -static UINT8 LegacyLoaderDevicePath4Data[] = { - 0x01, 0x03, 0x18, 0x00, 0x0B, 0x00, 0x00, 0x00, - 0x00, 0x00, 0xC0, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xFF, 0xF8, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0x04, 0x06, 0x14, 0x00, 0xEB, 0x85, 0x05, 0x2B, - 0xB8, 0xD8, 0xA9, 0x49, 0x8B, 0x8C, 0xE2, 0x1B, - 0x01, 0xAE, 0xF2, 0xB7, 0x7F, 0xFF, 0x04, 0x00, -}; -// late-2008 MB/MBP (NVidia chipset) -static UINT8 LegacyLoaderDevicePath5Data[] = { - 0x01, 0x03, 0x18, 0x00, 0x0B, 0x00, 0x00, 0x00, - 0x00, 0x40, 0xCB, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xBF, 0xFF, 0xFF, 0x00, 0x00, 0x00, 0x00, - 0x04, 0x06, 0x14, 0x00, 0xEB, 0x85, 0x05, 0x2B, - 0xB8, 0xD8, 0xA9, 0x49, 0x8B, 0x8C, 0xE2, 0x1B, - 0x01, 0xAE, 0xF2, 0xB7, 0x7F, 0xFF, 0x04, 0x00, -}; - -static EFI_DEVICE_PATH *LegacyLoaderList[] = { - (EFI_DEVICE_PATH *)LegacyLoaderDevicePath1Data, - (EFI_DEVICE_PATH *)LegacyLoaderDevicePath2Data, - (EFI_DEVICE_PATH *)LegacyLoaderDevicePath3Data, - (EFI_DEVICE_PATH *)LegacyLoaderDevicePath4Data, - (EFI_DEVICE_PATH *)LegacyLoaderDevicePath5Data, - NULL -}; - -#define MAX_DISCOVERED_PATHS (16) - -static VOID StartLegacy(IN LEGACY_ENTRY *Entry, IN CHAR16 *SelectionName) -{ - EFI_STATUS Status; - EG_IMAGE *BootLogoImage; - UINTN ErrorInStep = 0; - EFI_DEVICE_PATH *DiscoveredPathList[MAX_DISCOVERED_PATHS]; - - BeginExternalScreen(TRUE, L"Booting Legacy OS (Mac mode)"); - - BootLogoImage = LoadOSIcon(Entry->Volume->OSIconName, L"legacy", TRUE); - if (BootLogoImage != NULL) - BltImageAlpha(BootLogoImage, - (UGAWidth - BootLogoImage->Width ) >> 1, - (UGAHeight - BootLogoImage->Height) >> 1, - &StdBackgroundPixel); - - if (Entry->Volume->IsMbrPartition) { - ActivateMbrPartition(Entry->Volume->WholeDiskBlockIO, Entry->Volume->MbrPartitionIndex); - } - - ExtractLegacyLoaderPaths(DiscoveredPathList, MAX_DISCOVERED_PATHS, LegacyLoaderList); + UINTN VolumeIndex; - StoreLoaderName(SelectionName); - Status = StartEFIImageList(DiscoveredPathList, Entry->LoadOptions, TYPE_LEGACY, L"legacy loader", 0, &ErrorInStep, TRUE, FALSE); - if (Status == EFI_NOT_FOUND) { - if (ErrorInStep == 1) { - Print(L"\nPlease make sure that you have the latest firmware update installed.\n"); - } else if (ErrorInStep == 3) { - Print(L"\nThe firmware refused to boot from the selected volume. Note that external\n" - L"hard drives are not well-supported by Apple's firmware for legacy OS booting.\n"); + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if (Volumes[VolumeIndex]->DiskKind == DISK_KIND_OPTICAL) { + ScanEfiFiles(Volumes[VolumeIndex]); } - } - FinishExternalScreen(); -} /* static VOID StartLegacy() */ - -// Start a device on a non-Mac using the EFI_LEGACY_BIOS_PROTOCOL -static VOID StartLegacyUEFI(LEGACY_ENTRY *Entry, CHAR16 *SelectionName) -{ - BeginExternalScreen(TRUE, L"Booting Legacy OS (UEFI mode)"); - StoreLoaderName(SelectionName); - - BdsLibConnectDevicePath (Entry->BdsOption->DevicePath); - BdsLibDoLegacyBoot(Entry->BdsOption); - - // If we get here, it means that there was a failure.... - Print(L"Failure booting legacy (BIOS) OS."); - PauseForKey(); - FinishExternalScreen(); -} // static VOID StartLegacyUEFI() - -static LEGACY_ENTRY * AddLegacyEntry(IN CHAR16 *LoaderTitle, IN REFIT_VOLUME *Volume) -{ - LEGACY_ENTRY *Entry, *SubEntry; - REFIT_MENU_SCREEN *SubScreen; - CHAR16 *VolDesc, *LegacyTitle; - CHAR16 ShortcutLetter = 0; - - if (LoaderTitle == NULL) { - if (Volume->OSName != NULL) { - LoaderTitle = Volume->OSName; - if (LoaderTitle[0] == 'W' || LoaderTitle[0] == 'L') - ShortcutLetter = LoaderTitle[0]; - } else - LoaderTitle = L"Legacy OS"; - } - if (Volume->VolName != NULL) - VolDesc = Volume->VolName; - else - VolDesc = (Volume->DiskKind == DISK_KIND_OPTICAL) ? L"CD" : L"HD"; - - LegacyTitle = AllocateZeroPool(256 * sizeof(CHAR16)); - if (LegacyTitle != NULL) - SPrint(LegacyTitle, 255, L"Boot %s from %s", LoaderTitle, VolDesc); - if (IsInSubstring(LegacyTitle, GlobalConfig.DontScanVolumes)) { - MyFreePool(LegacyTitle); - return NULL; - } // if - - // prepare the menu entry - Entry = AllocateZeroPool(sizeof(LEGACY_ENTRY)); - Entry->me.Title = LegacyTitle; - Entry->me.Tag = TAG_LEGACY; - Entry->me.Row = 0; - Entry->me.ShortcutLetter = ShortcutLetter; - Entry->me.Image = LoadOSIcon(Volume->OSIconName, L"legacy", FALSE); - Entry->me.BadgeImage = Volume->VolBadgeImage; - Entry->Volume = Volume; - Entry->LoadOptions = (Volume->DiskKind == DISK_KIND_OPTICAL) ? L"CD" : - ((Volume->DiskKind == DISK_KIND_EXTERNAL) ? L"USB" : L"HD"); - Entry->Enabled = TRUE; - - // create the submenu - SubScreen = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); - SubScreen->Title = AllocateZeroPool(256 * sizeof(CHAR16)); - SPrint(SubScreen->Title, 255, L"Boot Options for %s on %s", LoaderTitle, VolDesc); - SubScreen->TitleImage = Entry->me.Image; - SubScreen->Hint1 = StrDuplicate(SUBSCREEN_HINT1); - if (GlobalConfig.HideUIFlags & HIDEUI_FLAG_EDITOR) { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2_NO_EDITOR); - } else { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2); - } // if/else - - // default entry - SubEntry = AllocateZeroPool(sizeof(LEGACY_ENTRY)); - SubEntry->me.Title = AllocateZeroPool(256 * sizeof(CHAR16)); - SPrint(SubEntry->me.Title, 255, L"Boot %s", LoaderTitle); - SubEntry->me.Tag = TAG_LEGACY; - SubEntry->Volume = Entry->Volume; - SubEntry->LoadOptions = Entry->LoadOptions; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - - AddMenuEntry(SubScreen, &MenuEntryReturn); - Entry->me.SubScreen = SubScreen; - AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); - return Entry; -} /* static LEGACY_ENTRY * AddLegacyEntry() */ - + } // for +} // static VOID ScanOptical() // default volume badge icon based on disk kind -static EG_IMAGE * GetDiskBadge(IN UINTN DiskType) { - EG_IMAGE * Badge = NULL; - - switch (DiskType) { - case BBS_HARDDISK: - Badge = BuiltinIcon(BUILTIN_ICON_VOL_INTERNAL); - break; - case BBS_USB: - Badge = BuiltinIcon(BUILTIN_ICON_VOL_EXTERNAL); - break; - case BBS_CDROM: - Badge = BuiltinIcon(BUILTIN_ICON_VOL_OPTICAL); - break; - } // switch() - return Badge; -} // static EG_IMAGE * GetDiskBadge() - -/** - Create a rEFInd boot option from a Legacy BIOS protocol option. -*/ -static LEGACY_ENTRY * AddLegacyEntryUEFI(BDS_COMMON_OPTION *BdsOption, IN UINT16 DiskType) -{ - LEGACY_ENTRY *Entry, *SubEntry; - REFIT_MENU_SCREEN *SubScreen; - CHAR16 ShortcutLetter = 0; - CHAR16 *LegacyDescription = StrDuplicate(BdsOption->Description); - - if (IsInSubstring(LegacyDescription, GlobalConfig.DontScanVolumes)) - return NULL; - - // Remove stray spaces, since many EFIs produce descriptions with lots of - // extra spaces, especially at the end; this throws off centering of the - // description on the screen.... - LimitStringLength(LegacyDescription, 100); - - // prepare the menu entry - Entry = AllocateZeroPool(sizeof(LEGACY_ENTRY)); - Entry->me.Title = AllocateZeroPool(256 * sizeof(CHAR16)); - SPrint(Entry->me.Title, 255, L"Boot legacy target %s", LegacyDescription); - Entry->me.Tag = TAG_LEGACY_UEFI; - Entry->me.Row = 0; - Entry->me.ShortcutLetter = ShortcutLetter; - Entry->me.Image = LoadOSIcon(L"legacy", L"legacy", TRUE); - Entry->LoadOptions = (DiskType == BBS_CDROM) ? L"CD" : - ((DiskType == BBS_USB) ? L"USB" : L"HD"); - Entry->me.BadgeImage = GetDiskBadge(DiskType); - Entry->BdsOption = BdsOption; - Entry->Enabled = TRUE; +EG_IMAGE * GetDiskBadge(IN UINTN DiskType) { + EG_IMAGE * Badge = NULL; - // create the submenu - SubScreen = AllocateZeroPool(sizeof(REFIT_MENU_SCREEN)); - SubScreen->Title = AllocateZeroPool(256 * sizeof(CHAR16)); - SPrint(SubScreen->Title, 255, L"No boot options for legacy target"); - SubScreen->TitleImage = Entry->me.Image; - SubScreen->Hint1 = StrDuplicate(SUBSCREEN_HINT1); - if (GlobalConfig.HideUIFlags & HIDEUI_FLAG_EDITOR) { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2_NO_EDITOR); - } else { - SubScreen->Hint2 = StrDuplicate(SUBSCREEN_HINT2); - } // if/else - - // default entry - SubEntry = AllocateZeroPool(sizeof(LEGACY_ENTRY)); - SubEntry->me.Title = AllocateZeroPool(256 * sizeof(CHAR16)); - SPrint(SubEntry->me.Title, 255, L"Boot %s", LegacyDescription); - SubEntry->me.Tag = TAG_LEGACY_UEFI; - Entry->BdsOption = BdsOption; - AddMenuEntry(SubScreen, (REFIT_MENU_ENTRY *)SubEntry); - - AddMenuEntry(SubScreen, &MenuEntryReturn); - Entry->me.SubScreen = SubScreen; - AddMenuEntry(&MainMenu, (REFIT_MENU_ENTRY *)Entry); - return Entry; -} /* static LEGACY_ENTRY * AddLegacyEntryUEFI() */ - -/** - Scan for legacy BIOS targets on machines that implement EFI_LEGACY_BIOS_PROTOCOL. - In testing, protocol has not been implemented on Macs but has been - implemented on several Dell PCs and an ASUS motherboard. - Restricts output to disks of the specified DiskType. -*/ -static VOID ScanLegacyUEFI(IN UINTN DiskType) -{ - EFI_STATUS Status; - EFI_LEGACY_BIOS_PROTOCOL *LegacyBios; - UINT16 *BootOrder = NULL; - UINTN Index = 0; - CHAR16 BootOption[10]; - UINTN BootOrderSize = 0; - CHAR16 Buffer[20]; - BDS_COMMON_OPTION *BdsOption; - LIST_ENTRY TempList; - BBS_BBS_DEVICE_PATH *BbsDevicePath = NULL; - BOOLEAN SearchingForUsb = FALSE; - - InitializeListHead (&TempList); - ZeroMem (Buffer, sizeof (Buffer)); - - // If LegacyBios protocol is not implemented on this platform, then - //we do not support this type of legacy boot on this machine. - Status = refit_call3_wrapper(gBS->LocateProtocol, &gEfiLegacyBootProtocolGuid, NULL, (VOID **) &LegacyBios); - if (EFI_ERROR (Status)) - return; - - // EFI calls USB drives BBS_HARDDRIVE, but we want to distinguish them, - // so we set DiskType inappropriately elsewhere in the program and - // "translate" it here. - if (DiskType == BBS_USB) { - DiskType = BBS_HARDDISK; - SearchingForUsb = TRUE; - } // if - - // Grab the boot order - BootOrder = BdsLibGetVariableAndSize(L"BootOrder", &gEfiGlobalVariableGuid, &BootOrderSize); - if (BootOrder == NULL) { - BootOrderSize = 0; - } - - Index = 0; - while (Index < BootOrderSize / sizeof (UINT16)) - { - // Grab each boot option variable from the boot order, and convert - // the variable into a BDS boot option - UnicodeSPrint (BootOption, sizeof (BootOption), L"Boot%04x", BootOrder[Index]); - BdsOption = BdsLibVariableToOption (&TempList, BootOption); - - if (BdsOption != NULL) { - BbsDevicePath = (BBS_BBS_DEVICE_PATH *)BdsOption->DevicePath; - // Only add the entry if it is of a requested type (e.g. USB, HD) - // Two checks necessary because some systems return EFI boot loaders - // with a DeviceType value that would inappropriately include them - // as legacy loaders.... - if ((BbsDevicePath->DeviceType == DiskType) && (BdsOption->DevicePath->Type == DEVICE_TYPE_BIOS)) { - // USB flash drives appear as hard disks with certain media flags set. - // Look for this, and if present, pass it on with the (technically - // incorrect, but internally useful) BBS_TYPE_USB flag set. - if (DiskType == BBS_HARDDISK) { - if (SearchingForUsb && (BbsDevicePath->StatusFlag & (BBS_MEDIA_PRESENT | BBS_MEDIA_MAYBE_PRESENT))) { - AddLegacyEntryUEFI(BdsOption, BBS_USB); - } else if (!SearchingForUsb && !(BbsDevicePath->StatusFlag & (BBS_MEDIA_PRESENT | BBS_MEDIA_MAYBE_PRESENT))) { - AddLegacyEntryUEFI(BdsOption, DiskType); - } - } else { - AddLegacyEntryUEFI(BdsOption, DiskType); - } // if/else - } // if - } // if (BdsOption != NULL) - Index++; - } // while -} /* static VOID ScanLegacyUEFI() */ - -static VOID ScanLegacyVolume(REFIT_VOLUME *Volume, UINTN VolumeIndex) { - UINTN VolumeIndex2; - BOOLEAN ShowVolume, HideIfOthersFound; - - ShowVolume = FALSE; - HideIfOthersFound = FALSE; - if (Volume->IsAppleLegacy) { - ShowVolume = TRUE; - HideIfOthersFound = TRUE; - } else if (Volume->HasBootCode) { - ShowVolume = TRUE; - if (Volume->BlockIO == Volume->WholeDiskBlockIO && - Volume->BlockIOOffset == 0 && - Volume->OSName == NULL) - // this is a whole disk (MBR) entry; hide if we have entries for partitions - HideIfOthersFound = TRUE; - } - if (HideIfOthersFound) { - // check for other bootable entries on the same disk - for (VolumeIndex2 = 0; VolumeIndex2 < VolumesCount; VolumeIndex2++) { - if (VolumeIndex2 != VolumeIndex && Volumes[VolumeIndex2]->HasBootCode && - Volumes[VolumeIndex2]->WholeDiskBlockIO == Volume->WholeDiskBlockIO) - ShowVolume = FALSE; - } - } - - if (ShowVolume) - AddLegacyEntry(NULL, Volume); -} // static VOID ScanLegacyVolume() - -// Scan attached optical discs for legacy (BIOS) boot code -// and add anything found to the list.... -static VOID ScanLegacyDisc(VOID) -{ - UINTN VolumeIndex; - REFIT_VOLUME *Volume; - - if (GlobalConfig.LegacyType == LEGACY_TYPE_MAC) { - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - Volume = Volumes[VolumeIndex]; - if (Volume->DiskKind == DISK_KIND_OPTICAL) - ScanLegacyVolume(Volume, VolumeIndex); - } // for - } else if (GlobalConfig.LegacyType == LEGACY_TYPE_UEFI) { - ScanLegacyUEFI(BBS_CDROM); - } -} /* static VOID ScanLegacyDisc() */ - -// Scan internal hard disks for legacy (BIOS) boot code -// and add anything found to the list.... -static VOID ScanLegacyInternal(VOID) -{ - UINTN VolumeIndex; - REFIT_VOLUME *Volume; - - if (GlobalConfig.LegacyType == LEGACY_TYPE_MAC) { - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - Volume = Volumes[VolumeIndex]; - if (Volume->DiskKind == DISK_KIND_INTERNAL) - ScanLegacyVolume(Volume, VolumeIndex); - } // for - } else if (GlobalConfig.LegacyType == LEGACY_TYPE_UEFI) { - // TODO: This actually picks up USB flash drives, too; try to find - // a way to differentiate the two.... - ScanLegacyUEFI(BBS_HARDDISK); - } -} /* static VOID ScanLegacyInternal() */ - -// Scan external disks for legacy (BIOS) boot code -// and add anything found to the list.... -static VOID ScanLegacyExternal(VOID) -{ - UINTN VolumeIndex; - REFIT_VOLUME *Volume; - - if (GlobalConfig.LegacyType == LEGACY_TYPE_MAC) { - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - Volume = Volumes[VolumeIndex]; - if (Volume->DiskKind == DISK_KIND_EXTERNAL) - ScanLegacyVolume(Volume, VolumeIndex); - } // for - } else if (GlobalConfig.LegacyType == LEGACY_TYPE_UEFI) { - // TODO: This actually doesn't do anything useful; leaving in hopes of - // fixing it later.... - ScanLegacyUEFI(BBS_USB); - } -} /* static VOID ScanLegacyExternal() */ + switch (DiskType) { + case BBS_HARDDISK: + Badge = BuiltinIcon(BUILTIN_ICON_VOL_INTERNAL); + break; + case BBS_USB: + Badge = BuiltinIcon(BUILTIN_ICON_VOL_EXTERNAL); + break; + case BBS_CDROM: + Badge = BuiltinIcon(BUILTIN_ICON_VOL_OPTICAL); + break; + } // switch() + return Badge; +} // EG_IMAGE * GetDiskBadge() // // pre-boot tool functions @@ -1975,11 +1670,11 @@ static VOID ScanLegacyExternal(VOID) static VOID StartTool(IN LOADER_ENTRY *Entry) { - BeginExternalScreen(Entry->UseGraphicsMode, Entry->me.Title + 6); // assumes "Start " as assigned below - StoreLoaderName(Entry->me.Title); - StartEFIImage(Entry->DevicePath, Entry->LoadOptions, TYPE_EFI, - Basename(Entry->LoaderPath), Entry->OSType, NULL, TRUE, FALSE); - FinishExternalScreen(); + BeginExternalScreen(Entry->UseGraphicsMode, Entry->me.Title + 6); // assumes "Start <title>" as assigned below + StoreLoaderName(Entry->me.Title); + StartEFIImage(Entry->DevicePath, Entry->LoadOptions, TYPE_EFI, + Basename(Entry->LoaderPath), Entry->OSType, NULL, TRUE, FALSE); + FinishExternalScreen(); } /* static VOID StartTool() */ static LOADER_ENTRY * AddToolEntry(EFI_HANDLE DeviceHandle, IN CHAR16 *LoaderPath, IN CHAR16 *LoaderTitle, IN EG_IMAGE *Image, @@ -2004,433 +1699,280 @@ static LOADER_ENTRY * AddToolEntry(EFI_HANDLE DeviceHandle, IN CHAR16 *LoaderPat return Entry; } /* static LOADER_ENTRY * AddToolEntry() */ -// -// pre-boot driver functions -// - -static UINTN ScanDriverDir(IN CHAR16 *Path) -{ - EFI_STATUS Status; - REFIT_DIR_ITER DirIter; - UINTN NumFound = 0; - EFI_FILE_INFO *DirEntry; - CHAR16 FileName[256]; - - CleanUpPathNameSlashes(Path); - // look through contents of the directory - DirIterOpen(SelfRootDir, Path, &DirIter); - while (DirIterNext(&DirIter, 2, LOADER_MATCH_PATTERNS, &DirEntry)) { - if (DirEntry->FileName[0] == '.') - continue; // skip this - - SPrint(FileName, 255, L"%s\\%s", Path, DirEntry->FileName); - NumFound++; - Status = StartEFIImage(FileDevicePath(SelfLoadedImage->DeviceHandle, FileName), - L"", TYPE_EFI, DirEntry->FileName, 0, NULL, FALSE, TRUE); - } - Status = DirIterClose(&DirIter); - if (Status != EFI_NOT_FOUND) { - SPrint(FileName, 255, L"while scanning the %s directory", Path); - CheckError(Status, FileName); - } - return (NumFound); -} - -#ifdef __MAKEWITH_GNUEFI -static EFI_STATUS ConnectAllDriversToAllControllers(VOID) -{ - EFI_STATUS Status; - UINTN AllHandleCount; - EFI_HANDLE *AllHandleBuffer; - UINTN Index; - UINTN HandleCount; - EFI_HANDLE *HandleBuffer; - UINT32 *HandleType; - UINTN HandleIndex; - BOOLEAN Parent; - BOOLEAN Device; - - Status = LibLocateHandle(AllHandles, - NULL, - NULL, - &AllHandleCount, - &AllHandleBuffer); - if (EFI_ERROR(Status)) - return Status; - - for (Index = 0; Index < AllHandleCount; Index++) { - // - // Scan the handle database - // - Status = LibScanHandleDatabase(NULL, - NULL, - AllHandleBuffer[Index], - NULL, - &HandleCount, - &HandleBuffer, - &HandleType); - if (EFI_ERROR (Status)) - goto Done; - - Device = TRUE; - if (HandleType[Index] & EFI_HANDLE_TYPE_DRIVER_BINDING_HANDLE) - Device = FALSE; - if (HandleType[Index] & EFI_HANDLE_TYPE_IMAGE_HANDLE) - Device = FALSE; - - if (Device) { - Parent = FALSE; - for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) { - if (HandleType[HandleIndex] & EFI_HANDLE_TYPE_PARENT_HANDLE) - Parent = TRUE; - } // for - - if (!Parent) { - if (HandleType[Index] & EFI_HANDLE_TYPE_DEVICE_HANDLE) { - Status = refit_call4_wrapper(BS->ConnectController, - AllHandleBuffer[Index], - NULL, - NULL, - TRUE); - } - } - } - - MyFreePool (HandleBuffer); - MyFreePool (HandleType); - } - -Done: - MyFreePool (AllHandleBuffer); - return Status; -} /* EFI_STATUS ConnectAllDriversToAllControllers() */ -#else -static EFI_STATUS ConnectAllDriversToAllControllers(VOID) { - BdsLibConnectAllDriversToAllControllers(); - return 0; -} -#endif - -// Load all EFI drivers from rEFInd's "drivers" subdirectory and from the -// directories specified by the user in the "scan_driver_dirs" configuration -// file line. -static VOID LoadDrivers(VOID) -{ - CHAR16 *Directory, *SelfDirectory; - UINTN i = 0, Length, NumFound = 0; - - // load drivers from the subdirectories of rEFInd's home directory specified - // in the DRIVER_DIRS constant. - while ((Directory = FindCommaDelimited(DRIVER_DIRS, i++)) != NULL) { - SelfDirectory = SelfDirPath ? StrDuplicate(SelfDirPath) : NULL; - CleanUpPathNameSlashes(SelfDirectory); - MergeStrings(&SelfDirectory, Directory, L'\\'); - NumFound += ScanDriverDir(SelfDirectory); - MyFreePool(Directory); - MyFreePool(SelfDirectory); - } - - // Scan additional user-specified driver directories.... - i = 0; - while ((Directory = FindCommaDelimited(GlobalConfig.DriverDirs, i++)) != NULL) { - CleanUpPathNameSlashes(Directory); - Length = StrLen(Directory); - if (Length > 0) { - NumFound += ScanDriverDir(Directory); - } // if - MyFreePool(Directory); - } // while - - // connect all devices - if (NumFound > 0) - ConnectAllDriversToAllControllers(); -} /* static VOID LoadDrivers() */ - -// Determine what (if any) type of legacy (BIOS) boot support is available -static VOID FindLegacyBootType(VOID) { - EFI_STATUS Status; - EFI_LEGACY_BIOS_PROTOCOL *LegacyBios; - - GlobalConfig.LegacyType = LEGACY_TYPE_NONE; - - // UEFI-style legacy BIOS support is available only with some EFI implementations.... - Status = refit_call3_wrapper(gBS->LocateProtocol, &gEfiLegacyBootProtocolGuid, NULL, (VOID **) &LegacyBios); - if (!EFI_ERROR (Status)) - GlobalConfig.LegacyType = LEGACY_TYPE_UEFI; - - // Macs have their own system. If the firmware vendor code contains the - // string "Apple", assume it's available. Note that this overrides the - // UEFI type, and might yield false positives if the vendor string - // contains "Apple" as part of something bigger, so this isn't 100% - // perfect. - if (StriSubCmp(L"Apple", ST->FirmwareVendor)) - GlobalConfig.LegacyType = LEGACY_TYPE_MAC; -} // static VOID FindLegacyBootType - -// Warn the user if legacy OS scans are enabled but the firmware can't support them.... -static VOID WarnIfLegacyProblems(VOID) { - BOOLEAN found = FALSE; - UINTN i = 0; - - if (GlobalConfig.LegacyType == LEGACY_TYPE_NONE) { - do { - if (GlobalConfig.ScanFor[i] == 'h' || GlobalConfig.ScanFor[i] == 'b' || GlobalConfig.ScanFor[i] == 'c' || - GlobalConfig.ScanFor[i] == 'H' || GlobalConfig.ScanFor[i] == 'B' || GlobalConfig.ScanFor[i] == 'C') - found = TRUE; - i++; - } while ((i < NUM_SCAN_OPTIONS) && (!found)); - - if (found) { - Print(L"NOTE: refind.conf's 'scanfor' line specifies scanning for one or more legacy\n"); - Print(L"(BIOS) boot options; however, this is not possible because your computer lacks\n"); - Print(L"the necessary Compatibility Support Module (CSM) support or that support is\n"); - Print(L"disabled in your firmware.\n"); - PauseForKey(); - } // if (found) - } // if no legacy support -} // static VOID WarnIfLegacyProblems() - // Locates boot loaders. NOTE: This assumes that GlobalConfig.LegacyType is set correctly. static VOID ScanForBootloaders(VOID) { - UINTN i; - CHAR8 s; - BOOLEAN ScanForLegacy = FALSE; - - // Determine up-front if we'll be scanning for legacy loaders.... - for (i = 0; i < NUM_SCAN_OPTIONS; i++) { - s = GlobalConfig.ScanFor[i]; - if ((s == 'c') || (s == 'C') || (s == 'h') || (s == 'H') || (s == 'b') || (s == 'B')) - ScanForLegacy = TRUE; - } // for - - // If UEFI & scanning for legacy loaders & deep legacy scan, update NVRAM boot manager list - if ((GlobalConfig.LegacyType == LEGACY_TYPE_UEFI) && ScanForLegacy && GlobalConfig.DeepLegacyScan) { - BdsDeleteAllInvalidLegacyBootOptions(); - BdsAddNonExistingLegacyBootOptions(); - } // if - - // scan for loaders and tools, add them to the menu - for (i = 0; i < NUM_SCAN_OPTIONS; i++) { - switch(GlobalConfig.ScanFor[i]) { - case 'c': case 'C': - ScanLegacyDisc(); - break; - case 'h': case 'H': - ScanLegacyInternal(); - break; - case 'b': case 'B': - ScanLegacyExternal(); - break; - case 'm': case 'M': - ScanUserConfigured(GlobalConfig.ConfigFilename); - break; - case 'e': case 'E': - ScanExternal(); - break; - case 'i': case 'I': - ScanInternal(); - break; - case 'o': case 'O': - ScanOptical(); - break; - } // switch() - } // for + UINTN i; + CHAR8 s; + BOOLEAN ScanForLegacy = FALSE; + + // Determine up-front if we'll be scanning for legacy loaders.... + for (i = 0; i < NUM_SCAN_OPTIONS; i++) { + s = GlobalConfig.ScanFor[i]; + if ((s == 'c') || (s == 'C') || (s == 'h') || (s == 'H') || (s == 'b') || (s == 'B')) + ScanForLegacy = TRUE; + } // for + + // If UEFI & scanning for legacy loaders & deep legacy scan, update NVRAM boot manager list + if ((GlobalConfig.LegacyType == LEGACY_TYPE_UEFI) && ScanForLegacy && GlobalConfig.DeepLegacyScan) { + BdsDeleteAllInvalidLegacyBootOptions(); + BdsAddNonExistingLegacyBootOptions(); + } // if - // assign shortcut keys - for (i = 0; i < MainMenu.EntryCount && MainMenu.Entries[i]->Row == 0 && i < 9; i++) - MainMenu.Entries[i]->ShortcutDigit = (CHAR16)('1' + i); + // scan for loaders and tools, add them to the menu + for (i = 0; i < NUM_SCAN_OPTIONS; i++) { + switch(GlobalConfig.ScanFor[i]) { + case 'c': case 'C': + ScanLegacyDisc(); + break; + case 'h': case 'H': + ScanLegacyInternal(); + break; + case 'b': case 'B': + ScanLegacyExternal(); + break; + case 'm': case 'M': + ScanUserConfigured(GlobalConfig.ConfigFilename); + break; + case 'e': case 'E': + ScanExternal(); + break; + case 'i': case 'I': + ScanInternal(); + break; + case 'o': case 'O': + ScanOptical(); + break; + case 'n': case 'N': + ScanNetboot(); + break; + } // switch() + } // for + + // assign shortcut keys + for (i = 0; i < MainMenu.EntryCount && MainMenu.Entries[i]->Row == 0 && i < 9; i++) + MainMenu.Entries[i]->ShortcutDigit = (CHAR16)('1' + i); - // wait for user ACK when there were errors - FinishTextScreen(FALSE); + // wait for user ACK when there were errors + FinishTextScreen(FALSE); } // static VOID ScanForBootloaders() // Locate a single tool from the specified Locations using one of the // specified Names and add it to the menu. static VOID FindTool(CHAR16 *Locations, CHAR16 *Names, CHAR16 *Description, UINTN Icon) { - UINTN j = 0, k, VolumeIndex; - CHAR16 *DirName, *FileName, *PathName, FullDescription[256]; - - while ((DirName = FindCommaDelimited(Locations, j++)) != NULL) { - k = 0; - while ((FileName = FindCommaDelimited(Names, k++)) != NULL) { - PathName = StrDuplicate(DirName); - MergeStrings(&PathName, FileName, (StriCmp(PathName, L"\\") == 0) ? 0 : L'\\'); - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if ((Volumes[VolumeIndex]->RootDir != NULL) && (FileExists(Volumes[VolumeIndex]->RootDir, PathName))) { - SPrint(FullDescription, 255, L"%s at %s on %s", Description, PathName, Volumes[VolumeIndex]->VolName); - AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, PathName, FullDescription, BuiltinIcon(Icon), 'S', FALSE); - } // if - } // for - MyFreePool(PathName); - MyFreePool(FileName); - } // while Names - MyFreePool(DirName); - } // while Locations + UINTN j = 0, k, VolumeIndex; + CHAR16 *DirName, *FileName, *PathName, FullDescription[256]; + + while ((DirName = FindCommaDelimited(Locations, j++)) != NULL) { + k = 0; + while ((FileName = FindCommaDelimited(Names, k++)) != NULL) { + PathName = StrDuplicate(DirName); + MergeStrings(&PathName, FileName, MyStriCmp(PathName, L"\\") ? 0 : L'\\'); + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if ((Volumes[VolumeIndex]->RootDir != NULL) && + (FileExists(Volumes[VolumeIndex]->RootDir, PathName)) && + IsValidLoader(Volumes[VolumeIndex]->RootDir, PathName)) { + SPrint(FullDescription, 255, L"%s at %s on %s", Description, PathName, Volumes[VolumeIndex]->VolName); + AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, PathName, FullDescription, BuiltinIcon(Icon), 'S', FALSE); + } // if + } // for + MyFreePool(PathName); + MyFreePool(FileName); + } // while Names + MyFreePool(DirName); + } // while Locations } // VOID FindTool() // Add the second-row tags containing built-in and external tools (EFI shell, // reboot, etc.) static VOID ScanForTools(VOID) { - CHAR16 *FileName = NULL, *VolName = NULL, *MokLocations, Description[256]; - REFIT_MENU_ENTRY *TempMenuEntry; - UINTN i, j, VolumeIndex; - UINT64 osind; - CHAR8 *b = 0; - - MokLocations = StrDuplicate(MOK_LOCATIONS); - if (MokLocations != NULL) - MergeStrings(&MokLocations, SelfDirPath, L','); - - for (i = 0; i < NUM_TOOLS; i++) { - switch(GlobalConfig.ShowTools[i]) { - // NOTE: Be sure that FileName is NULL at the end of each case. - case TAG_SHUTDOWN: - TempMenuEntry = CopyMenuEntry(&MenuEntryShutdown); - TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_SHUTDOWN); - AddMenuEntry(&MainMenu, TempMenuEntry); - break; + CHAR16 *FileName = NULL, *VolName = NULL, *MokLocations, Description[256]; + REFIT_MENU_ENTRY *TempMenuEntry; + UINTN i, j, VolumeIndex; + UINT64 osind; + CHAR8 *b = 0; + UINT32 CsrValue; + + MokLocations = StrDuplicate(MOK_LOCATIONS); + if (MokLocations != NULL) + MergeStrings(&MokLocations, SelfDirPath, L','); + + for (i = 0; i < NUM_TOOLS; i++) { + switch(GlobalConfig.ShowTools[i]) { + // NOTE: Be sure that FileName is NULL at the end of each case. + case TAG_SHUTDOWN: + TempMenuEntry = CopyMenuEntry(&MenuEntryShutdown); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_SHUTDOWN); + AddMenuEntry(&MainMenu, TempMenuEntry); + break; - case TAG_REBOOT: - TempMenuEntry = CopyMenuEntry(&MenuEntryReset); - TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_RESET); - AddMenuEntry(&MainMenu, TempMenuEntry); - break; + case TAG_REBOOT: + TempMenuEntry = CopyMenuEntry(&MenuEntryReset); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_RESET); + AddMenuEntry(&MainMenu, TempMenuEntry); + break; - case TAG_ABOUT: - TempMenuEntry = CopyMenuEntry(&MenuEntryAbout); - TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_ABOUT); - AddMenuEntry(&MainMenu, TempMenuEntry); - break; + case TAG_ABOUT: + TempMenuEntry = CopyMenuEntry(&MenuEntryAbout); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_ABOUT); + AddMenuEntry(&MainMenu, TempMenuEntry); + break; - case TAG_EXIT: - TempMenuEntry = CopyMenuEntry(&MenuEntryExit); - TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_EXIT); - AddMenuEntry(&MainMenu, TempMenuEntry); - break; + case TAG_EXIT: + TempMenuEntry = CopyMenuEntry(&MenuEntryExit); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_EXIT); + AddMenuEntry(&MainMenu, TempMenuEntry); + break; - case TAG_FIRMWARE: - if (EfivarGetRaw(&GlobalGuid, L"OsIndicationsSupported", &b, &j) == EFI_SUCCESS) { - osind = (UINT64)*b; - if (osind & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) { - TempMenuEntry = CopyMenuEntry(&MenuEntryFirmware); - TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_FIRMWARE); - AddMenuEntry(&MainMenu, TempMenuEntry); - } // if - } // if - break; + case TAG_FIRMWARE: + if (EfivarGetRaw(&GlobalGuid, L"OsIndicationsSupported", &b, &j) == EFI_SUCCESS) { + osind = (UINT64)*b; + if (osind & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) { + TempMenuEntry = CopyMenuEntry(&MenuEntryFirmware); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_FIRMWARE); + AddMenuEntry(&MainMenu, TempMenuEntry); + } // if + } // if + break; - case TAG_SHELL: - j = 0; - while ((FileName = FindCommaDelimited(SHELL_NAMES, j++)) != NULL) { - if (FileExists(SelfRootDir, FileName)) { - AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"EFI Shell", BuiltinIcon(BUILTIN_ICON_TOOL_SHELL), - 'S', FALSE); - } - MyFreePool(FileName); - } // while - break; + case TAG_SHELL: + j = 0; + while ((FileName = FindCommaDelimited(SHELL_NAMES, j++)) != NULL) { + if (FileExists(SelfRootDir, FileName) && IsValidLoader(SelfRootDir, FileName)) { + AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"EFI Shell", BuiltinIcon(BUILTIN_ICON_TOOL_SHELL), + 'S', FALSE); + } + MyFreePool(FileName); + } // while + break; - case TAG_GPTSYNC: - j = 0; - while ((FileName = FindCommaDelimited(GPTSYNC_NAMES, j++)) != NULL) { - if (FileExists(SelfRootDir, FileName)) { - AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"Hybrid MBR tool", BuiltinIcon(BUILTIN_ICON_TOOL_PART), - 'P', FALSE); - } // if - MyFreePool(FileName); - } // while - FileName = NULL; - break; + case TAG_GPTSYNC: + j = 0; + while ((FileName = FindCommaDelimited(GPTSYNC_NAMES, j++)) != NULL) { + if (FileExists(SelfRootDir, FileName) && IsValidLoader(SelfRootDir, FileName)) { + AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"Hybrid MBR tool", BuiltinIcon(BUILTIN_ICON_TOOL_PART), + 'P', FALSE); + } // if + MyFreePool(FileName); + } // while + FileName = NULL; + break; - case TAG_GDISK: - j = 0; - while ((FileName = FindCommaDelimited(GDISK_NAMES, j++)) != NULL) { - if (FileExists(SelfRootDir, FileName)) { - AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"disk partitioning tool", - BuiltinIcon(BUILTIN_ICON_TOOL_PART), 'G', FALSE); - } // if - MyFreePool(FileName); - } // while - FileName = NULL; - break; + case TAG_GDISK: + j = 0; + while ((FileName = FindCommaDelimited(GDISK_NAMES, j++)) != NULL) { + if (FileExists(SelfRootDir, FileName) && IsValidLoader(SelfRootDir, FileName)) { + AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"disk partitioning tool", + BuiltinIcon(BUILTIN_ICON_TOOL_PART), 'G', FALSE); + } // if + MyFreePool(FileName); + } // while + FileName = NULL; + break; + + case TAG_NETBOOT: + j = 0; + while ((FileName = FindCommaDelimited(NETBOOT_NAMES, j++)) != NULL) { + if (FileExists(SelfRootDir, FileName) && IsValidLoader(SelfRootDir, FileName)) { + AddToolEntry(SelfLoadedImage->DeviceHandle, FileName, L"Netboot", + BuiltinIcon(BUILTIN_ICON_TOOL_NETBOOT), 'N', FALSE); + } // if + MyFreePool(FileName); + } // while + FileName = NULL; + break; - case TAG_APPLE_RECOVERY: - FileName = StrDuplicate(L"\\com.apple.recovery.boot\\boot.efi"); - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if ((Volumes[VolumeIndex]->RootDir != NULL) && (FileExists(Volumes[VolumeIndex]->RootDir, FileName))) { - SPrint(Description, 255, L"Apple Recovery on %s", Volumes[VolumeIndex]->VolName); - AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, FileName, Description, - BuiltinIcon(BUILTIN_ICON_TOOL_APPLE_RESCUE), 'R', TRUE); - } // if - } // for - MyFreePool(FileName); - FileName = NULL; - break; + case TAG_APPLE_RECOVERY: + FileName = StrDuplicate(L"\\com.apple.recovery.boot\\boot.efi"); + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if ((Volumes[VolumeIndex]->RootDir != NULL) && + (FileExists(Volumes[VolumeIndex]->RootDir, FileName)) && + IsValidLoader(Volumes[VolumeIndex]->RootDir, FileName)) { + SPrint(Description, 255, L"Apple Recovery on %s", Volumes[VolumeIndex]->VolName); + AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, FileName, Description, + BuiltinIcon(BUILTIN_ICON_TOOL_APPLE_RESCUE), 'R', TRUE); + } // if + } // for + MyFreePool(FileName); + FileName = NULL; + break; - case TAG_WINDOWS_RECOVERY: - j = 0; - while ((FileName = FindCommaDelimited(GlobalConfig.WindowsRecoveryFiles, j++)) != NULL) { - SplitVolumeAndFilename(&FileName, &VolName); - for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { - if ((Volumes[VolumeIndex]->RootDir != NULL) && (FileExists(Volumes[VolumeIndex]->RootDir, FileName)) && - ((VolName == NULL) || (StriCmp(VolName, Volumes[VolumeIndex]->VolName) == 0))) { - SPrint(Description, 255, L"Microsoft Recovery on %s", Volumes[VolumeIndex]->VolName); - AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, FileName, Description, - BuiltinIcon(BUILTIN_ICON_TOOL_WINDOWS_RESCUE), 'R', TRUE); - } // if - } // for - } // while - MyFreePool(FileName); - FileName = NULL; - MyFreePool(VolName); - VolName = NULL; - break; + case TAG_WINDOWS_RECOVERY: + j = 0; + while ((FileName = FindCommaDelimited(GlobalConfig.WindowsRecoveryFiles, j++)) != NULL) { + SplitVolumeAndFilename(&FileName, &VolName); + for (VolumeIndex = 0; VolumeIndex < VolumesCount; VolumeIndex++) { + if ((Volumes[VolumeIndex]->RootDir != NULL) && + (FileExists(Volumes[VolumeIndex]->RootDir, FileName)) && + IsValidLoader(Volumes[VolumeIndex]->RootDir, FileName) && + ((VolName == NULL) || MyStriCmp(VolName, Volumes[VolumeIndex]->VolName))) { + SPrint(Description, 255, L"Microsoft Recovery on %s", Volumes[VolumeIndex]->VolName); + AddToolEntry(Volumes[VolumeIndex]->DeviceHandle, FileName, Description, + BuiltinIcon(BUILTIN_ICON_TOOL_WINDOWS_RESCUE), 'R', TRUE); + } // if + } // for + } // while + MyFreePool(FileName); + FileName = NULL; + MyFreePool(VolName); + VolName = NULL; + break; - case TAG_MOK_TOOL: - FindTool(MokLocations, MOK_NAMES, L"MOK utility", BUILTIN_ICON_TOOL_MOK_TOOL); - break; + case TAG_MOK_TOOL: + FindTool(MokLocations, MOK_NAMES, L"MOK utility", BUILTIN_ICON_TOOL_MOK_TOOL); + break; - case TAG_MEMTEST: - FindTool(MEMTEST_LOCATIONS, MEMTEST_NAMES, L"Memory test utility", BUILTIN_ICON_TOOL_MEMTEST); - break; + case TAG_FWUPDATE_TOOL: + FindTool(MokLocations, FWUPDATE_NAMES, L"firmware update utility", BUILTIN_ICON_TOOL_FWUPDATE); + break; - } // switch() - } // for + case TAG_CSR_ROTATE: + if ((GetCsrStatus(&CsrValue) == EFI_SUCCESS) && (GlobalConfig.CsrValues)) { + TempMenuEntry = CopyMenuEntry(&MenuEntryRotateCsr); + TempMenuEntry->Image = BuiltinIcon(BUILTIN_ICON_FUNC_CSR_ROTATE); + AddMenuEntry(&MainMenu, TempMenuEntry); + } // if + break; + + case TAG_MEMTEST: + FindTool(MEMTEST_LOCATIONS, MEMTEST_NAMES, L"Memory test utility", BUILTIN_ICON_TOOL_MEMTEST); + break; + + } // switch() + } // for } // static VOID ScanForTools // Rescan for boot loaders static VOID RescanAll(BOOLEAN DisplayMessage) { - EG_PIXEL BGColor; - - BGColor.b = 255; - BGColor.g = 175; - BGColor.r = 100; - BGColor.a = 0; - if (DisplayMessage) - egDisplayMessage(L"Scanning for new boot loaders; please wait....", &BGColor); - FreeList((VOID ***) &(MainMenu.Entries), &MainMenu.EntryCount); - MainMenu.Entries = NULL; - MainMenu.EntryCount = 0; - ReadConfig(GlobalConfig.ConfigFilename); - ConnectAllDriversToAllControllers(); - ScanVolumes(); - ScanForBootloaders(); - ScanForTools(); - SetupScreen(); + EG_PIXEL BGColor; + + BGColor.b = 255; + BGColor.g = 175; + BGColor.r = 100; + BGColor.a = 0; + if (DisplayMessage) + egDisplayMessage(L"Scanning for new boot loaders; please wait....", &BGColor); + FreeList((VOID ***) &(MainMenu.Entries), &MainMenu.EntryCount); + MainMenu.Entries = NULL; + MainMenu.EntryCount = 0; + ConnectAllDriversToAllControllers(); + ScanVolumes(); + ReadConfig(GlobalConfig.ConfigFilename); + SetVolumeIcons(); + ScanForBootloaders(); + ScanForTools(); + SetupScreen(); } // VOID RescanAll() #ifdef __MAKEWITH_TIANO // Minimal initialization function static VOID InitializeLib(IN EFI_HANDLE ImageHandle, IN EFI_SYSTEM_TABLE *SystemTable) { - gST = SystemTable; - // gImageHandle = ImageHandle; - gBS = SystemTable->BootServices; - // gRS = SystemTable->RuntimeServices; - gRT = SystemTable->RuntimeServices; // Some BDS functions need gRT to be set - EfiGetSystemConfigurationTable (&gEfiDxeServicesTableGuid, (VOID **) &gDS); - -// InitializeConsoleSim(); + gST = SystemTable; + // gImageHandle = ImageHandle; + gBS = SystemTable->BootServices; + // gRS = SystemTable->RuntimeServices; + gRT = SystemTable->RuntimeServices; // Some BDS functions need gRT to be set + EfiGetSystemConfigurationTable (&gEfiDxeServicesTableGuid, (VOID **) &gDS); } #endif @@ -2438,37 +1980,38 @@ static VOID InitializeLib(IN EFI_HANDLE ImageHandle, IN EFI_SYSTEM_TABLE *System // Set up our own Secure Boot extensions.... // Returns TRUE on success, FALSE otherwise static BOOLEAN SecureBootSetup(VOID) { - EFI_STATUS Status; - BOOLEAN Success = FALSE; - - if (secure_mode() && ShimLoaded()) { - Status = security_policy_install(); - if (Status == EFI_SUCCESS) { - Success = TRUE; - } else { - Print(L"Failed to install MOK Secure Boot extensions"); - } - } - return Success; + EFI_STATUS Status; + BOOLEAN Success = FALSE; + + if (secure_mode() && ShimLoaded()) { + Status = security_policy_install(); + if (Status == EFI_SUCCESS) { + Success = TRUE; + } else { + Print(L"Failed to install MOK Secure Boot extensions"); + PauseForKey(); + } + } + return Success; } // VOID SecureBootSetup() // Remove our own Secure Boot extensions.... // Returns TRUE on success, FALSE otherwise static BOOLEAN SecureBootUninstall(VOID) { - EFI_STATUS Status; - BOOLEAN Success = TRUE; - - if (secure_mode()) { - Status = security_policy_uninstall(); - if (Status != EFI_SUCCESS) { - Success = FALSE; - BeginTextScreen(L"Secure Boot Policy Failure"); - Print(L"Failed to uninstall MOK Secure Boot extensions; forcing a reboot."); - PauseForKey(); - refit_call4_wrapper(RT->ResetSystem, EfiResetCold, EFI_SUCCESS, 0, NULL); - } - } - return Success; + EFI_STATUS Status; + BOOLEAN Success = TRUE; + + if (secure_mode()) { + Status = security_policy_uninstall(); + if (Status != EFI_SUCCESS) { + Success = FALSE; + BeginTextScreen(L"Secure Boot Policy Failure"); + Print(L"Failed to uninstall MOK Secure Boot extensions; forcing a reboot."); + PauseForKey(); + refit_call4_wrapper(RT->ResetSystem, EfiResetCold, EFI_SUCCESS, 0, NULL); + } + } + return Success; } // VOID SecureBootUninstall // Sets the global configuration filename; will be CONFIG_FILE_NAME unless the @@ -2476,29 +2019,28 @@ static BOOLEAN SecureBootUninstall(VOID) { // If an error is encountered, leaves the value alone (it should be set to // CONFIG_FILE_NAME when GlobalConfig is initialized). static VOID SetConfigFilename(EFI_HANDLE ImageHandle) { - EFI_LOADED_IMAGE *Info; - CHAR16 *Options, *FileName; - EFI_STATUS Status; - INTN Where; - - Status = refit_call3_wrapper(BS->HandleProtocol, ImageHandle, &LoadedImageProtocol, (VOID **) &Info); - if ((Status == EFI_SUCCESS) && (Info->LoadOptionsSize > 0)) { - Options = (CHAR16 *) Info->LoadOptions; - Where = FindSubString(L" -c ", Options); - if (Where >= 0) { - FileName = StrDuplicate(&Options[Where + 4]); - Where = FindSubString(L" ", FileName); - if (Where > 0) - FileName[Where] = L'\0'; - - if (FileExists(SelfDir, FileName)) { - GlobalConfig.ConfigFilename = FileName; - } else { - Print(L"Specified configuration file (%s) doesn't exist; using\n'refind.conf' default\n", FileName); - MyFreePool(FileName); - } // if/else - } // if - } // if + EFI_LOADED_IMAGE *Info; + CHAR16 *Options, *FileName, *SubString; + EFI_STATUS Status; + + Status = refit_call3_wrapper(BS->HandleProtocol, ImageHandle, &LoadedImageProtocol, (VOID **) &Info); + if ((Status == EFI_SUCCESS) && (Info->LoadOptionsSize > 0)) { + Options = (CHAR16 *) Info->LoadOptions; + SubString = MyStrStr(Options, L" -c "); + if (SubString) { + FileName = StrDuplicate(&SubString[4]); + if (FileName) { + LimitStringLength(FileName, 256); + } + + if (FileExists(SelfDir, FileName)) { + GlobalConfig.ConfigFilename = FileName; + } else { + Print(L"Specified configuration file (%s) doesn't exist; using\n'refind.conf' default\n", FileName); + MyFreePool(FileName); + } // if/else + } // if + } // if } // VOID SetConfigFilename() // @@ -2528,7 +2070,14 @@ efi_main (EFI_HANDLE ImageHandle, EFI_SYSTEM_TABLE *SystemTable) if (GlobalConfig.LegacyType == LEGACY_TYPE_MAC) CopyMem(GlobalConfig.ScanFor, "ihebocm ", NUM_SCAN_OPTIONS); SetConfigFilename(ImageHandle); + MokProtocol = SecureBootSetup(); + LoadDrivers(); + ScanVolumes(); // Do before ReadConfig() because it needs SelfVolume->VolName ReadConfig(GlobalConfig.ConfigFilename); + SetVolumeIcons(); + + if (GlobalConfig.SpoofOSXVersion && GlobalConfig.SpoofOSXVersion[0] != L'\0') + SetAppleOSInfo(); InitScreen(); WarnIfLegacyProblems(); @@ -2538,9 +2087,6 @@ efi_main (EFI_HANDLE ImageHandle, EFI_SYSTEM_TABLE *SystemTable) refit_call4_wrapper(BS->SetWatchdogTimer, 0x0000, 0x0000, 0x0000, NULL); // further bootstrap (now with config available) - MokProtocol = SecureBootSetup(); - LoadDrivers(); - ScanVolumes(); ScanForBootloaders(); ScanForTools(); SetupScreen(); @@ -2617,6 +2163,10 @@ efi_main (EFI_HANDLE ImageHandle, EFI_SYSTEM_TABLE *SystemTable) RebootIntoFirmware(); break; + case TAG_CSR_ROTATE: + RotateCsrValue(); + break; + } // switch() } // while()