[mirror_edk2.git] / UefiPayloadPkg / Library / PlatformBootManagerLib / PlatformBootManager.c

/** @file\r
  This file include all platform action which can be customized\r
  by IBV/OEM.\r
\r
Copyright (c) 2015 - 2021, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "PlatformBootManager.h"\r
#include "PlatformConsole.h"\r
#include <Protocol/PlatformBootManagerOverride.h>\r
#include <Guid/BootManagerMenu.h>\r
#include <Library/HobLib.h>\r
\r
UNIVERSAL_PAYLOAD_PLATFORM_BOOT_MANAGER_OVERRIDE_PROTOCOL  *mUniversalPayloadPlatformBootManagerOverrideInstance = NULL;\r
\r
/**\r
  Signal EndOfDxe event and install SMM Ready to lock protocol.\r
\r
**/\r
VOID\r
InstallReadyToLock (\r
  VOID\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_HANDLE                Handle;\r
  EFI_SMM_ACCESS2_PROTOCOL  *SmmAccess;\r
\r
  DEBUG ((DEBUG_INFO, "InstallReadyToLock  entering......\n"));\r
  //\r
  // Inform the SMM infrastructure that we're entering BDS and may run 3rd party code hereafter\r
  // Since PI1.2.1, we need signal EndOfDxe as ExitPmAuth\r
  //\r
  EfiEventGroupSignal (&gEfiEndOfDxeEventGroupGuid);\r
  DEBUG ((DEBUG_INFO, "All EndOfDxe callbacks have returned successfully\n"));\r
\r
  //\r
  // Install DxeSmmReadyToLock protocol in order to lock SMM\r
  //\r
  Status = gBS->LocateProtocol (&gEfiSmmAccess2ProtocolGuid, NULL, (VOID **)&SmmAccess);\r
  if (!EFI_ERROR (Status)) {\r
    Handle = NULL;\r
    Status = gBS->InstallProtocolInterface (\r
                    &Handle,\r
                    &gEfiDxeSmmReadyToLockProtocolGuid,\r
                    EFI_NATIVE_INTERFACE,\r
                    NULL\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "InstallReadyToLock  end\n"));\r
  return;\r
}\r
\r
/**\r
  Return the index of the load option in the load option array.\r
\r
  The function consider two load options are equal when the\r
  OptionType, Attributes, Description, FilePath and OptionalData are equal.\r
\r
  @param Key    Pointer to the load option to be found.\r
  @param Array  Pointer to the array of load options to be found.\r
  @param Count  Number of entries in the Array.\r
\r
  @retval -1          Key wasn't found in the Array.\r
  @retval 0 ~ Count-1 The index of the Key in the Array.\r
**/\r
INTN\r
PlatformFindLoadOption (\r
  IN CONST EFI_BOOT_MANAGER_LOAD_OPTION  *Key,\r
  IN CONST EFI_BOOT_MANAGER_LOAD_OPTION  *Array,\r
  IN UINTN                               Count\r
  )\r
{\r
  UINTN  Index;\r
\r
  for (Index = 0; Index < Count; Index++) {\r
    if ((Key->OptionType == Array[Index].OptionType) &&\r
        (Key->Attributes == Array[Index].Attributes) &&\r
        (StrCmp (Key->Description, Array[Index].Description) == 0) &&\r
        (CompareMem (Key->FilePath, Array[Index].FilePath, GetDevicePathSize (Key->FilePath)) == 0) &&\r
        (Key->OptionalDataSize == Array[Index].OptionalDataSize) &&\r
        (CompareMem (Key->OptionalData, Array[Index].OptionalData, Key->OptionalDataSize) == 0))\r
    {\r
      return (INTN)Index;\r
    }\r
  }\r
\r
  return -1;\r
}\r
\r
/**\r
  Register a boot option using a file GUID in the FV.\r
\r
  @param FileGuid     The file GUID name in FV.\r
  @param Description  The boot option description.\r
  @param Attributes   The attributes used for the boot option loading.\r
**/\r
VOID\r
PlatformRegisterFvBootOption (\r
  EFI_GUID  *FileGuid,\r
  CHAR16    *Description,\r
  UINT32    Attributes\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  UINTN                              OptionIndex;\r
  EFI_BOOT_MANAGER_LOAD_OPTION       NewOption;\r
  EFI_BOOT_MANAGER_LOAD_OPTION       *BootOptions;\r
  UINTN                              BootOptionCount;\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  FileNode;\r
  EFI_LOADED_IMAGE_PROTOCOL          *LoadedImage;\r
  EFI_DEVICE_PATH_PROTOCOL           *DevicePath;\r
\r
  Status = gBS->HandleProtocol (gImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **)&LoadedImage);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  EfiInitializeFwVolDevicepathNode (&FileNode, FileGuid);\r
  DevicePath = AppendDevicePathNode (\r
                 DevicePathFromHandle (LoadedImage->DeviceHandle),\r
                 (EFI_DEVICE_PATH_PROTOCOL *)&FileNode\r
                 );\r
\r
  Status = EfiBootManagerInitializeLoadOption (\r
             &NewOption,\r
             LoadOptionNumberUnassigned,\r
             LoadOptionTypeBoot,\r
             Attributes,\r
             Description,\r
             DevicePath,\r
             NULL,\r
             0\r
             );\r
  if (!EFI_ERROR (Status)) {\r
    BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount, LoadOptionTypeBoot);\r
\r
    OptionIndex = PlatformFindLoadOption (&NewOption, BootOptions, BootOptionCount);\r
\r
    if (OptionIndex == -1) {\r
      Status = EfiBootManagerAddLoadOptionVariable (&NewOption, (UINTN)-1);\r
      ASSERT_EFI_ERROR (Status);\r
    }\r
\r
    EfiBootManagerFreeLoadOption (&NewOption);\r
    EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
  }\r
}\r
\r
/**\r
  Do the platform specific action before the console is connected.\r
\r
  Such as:\r
    Update console variable;\r
    Register new Driver#### or Boot####;\r
    Signal ReadyToLock event.\r
**/\r
VOID\r
EFIAPI\r
PlatformBootManagerBeforeConsole (\r
  VOID\r
  )\r
{\r
  EFI_INPUT_KEY                 Enter;\r
  EFI_INPUT_KEY                 CustomKey;\r
  EFI_INPUT_KEY                 Down;\r
  EFI_BOOT_MANAGER_LOAD_OPTION  BootOption;\r
  EFI_STATUS                    Status;\r
\r
  Status = gBS->LocateProtocol (&gUniversalPayloadPlatformBootManagerOverrideProtocolGuid, NULL, (VOID **)&mUniversalPayloadPlatformBootManagerOverrideInstance);\r
  if (EFI_ERROR (Status)) {\r
    mUniversalPayloadPlatformBootManagerOverrideInstance = NULL;\r
  }\r
\r
  if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
    mUniversalPayloadPlatformBootManagerOverrideInstance->BeforeConsole ();\r
    return;\r
  }\r
\r
  //\r
  // Register ENTER as CONTINUE key\r
  //\r
  Enter.ScanCode    = SCAN_NULL;\r
  Enter.UnicodeChar = CHAR_CARRIAGE_RETURN;\r
  EfiBootManagerRegisterContinueKeyOption (0, &Enter, NULL);\r
\r
  if (FixedPcdGetBool (PcdBootManagerEscape)) {\r
    //\r
    // Map Esc to Boot Manager Menu\r
    //\r
    CustomKey.ScanCode    = SCAN_ESC;\r
    CustomKey.UnicodeChar = CHAR_NULL;\r
  } else {\r
    //\r
    // Map Esc to Boot Manager Menu\r
    //\r
    CustomKey.ScanCode    = SCAN_F2;\r
    CustomKey.UnicodeChar = CHAR_NULL;\r
  }\r
\r
  EfiBootManagerGetBootManagerMenu (&BootOption);\r
  EfiBootManagerAddKeyOptionVariable (NULL, (UINT16)BootOption.OptionNumber, 0, &CustomKey, NULL);\r
\r
  //\r
  // Also add Down key to Boot Manager Menu since some serial terminals don't support F2 key.\r
  //\r
  Down.ScanCode    = SCAN_DOWN;\r
  Down.UnicodeChar = CHAR_NULL;\r
  EfiBootManagerGetBootManagerMenu (&BootOption);\r
  EfiBootManagerAddKeyOptionVariable (NULL, (UINT16)BootOption.OptionNumber, 0, &Down, NULL);\r
\r
  //\r
  // Install ready to lock.\r
  // This needs to be done before option rom dispatched.\r
  //\r
  InstallReadyToLock ();\r
\r
  //\r
  // Dispatch deferred images after EndOfDxe event and ReadyToLock installation.\r
  //\r
  EfiBootManagerDispatchDeferredImages ();\r
\r
  PlatformConsoleInit ();\r
}\r
\r
/**\r
  Do the platform specific action after the console is connected.\r
\r
  Such as:\r
    Dynamically switch output mode;\r
    Signal console ready platform customized event;\r
    Run diagnostics like memory testing;\r
    Connect certain devices;\r
    Dispatch additional option roms.\r
**/\r
VOID\r
EFIAPI\r
PlatformBootManagerAfterConsole (\r
  VOID\r
  )\r
{\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL  Black;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL  White;\r
  EDKII_PLATFORM_LOGO_PROTOCOL   *PlatformLogo;\r
  EFI_STATUS                     Status;\r
\r
  if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
    mUniversalPayloadPlatformBootManagerOverrideInstance->AfterConsole ();\r
    return;\r
  }\r
\r
  Black.Blue = Black.Green = Black.Red = Black.Reserved = 0;\r
  White.Blue = White.Green = White.Red = White.Reserved = 0xFF;\r
\r
  Status = gBS->LocateProtocol (&gEdkiiPlatformLogoProtocolGuid, NULL, (VOID **)&PlatformLogo);\r
\r
  if (!EFI_ERROR (Status)) {\r
    gST->ConOut->ClearScreen (gST->ConOut);\r
    BootLogoEnableLogo ();\r
  }\r
\r
  EfiBootManagerConnectAll ();\r
  EfiBootManagerRefreshAllBootOption ();\r
\r
  //\r
  // Register UEFI Shell\r
  //\r
  PlatformRegisterFvBootOption (PcdGetPtr (PcdShellFile), L"UEFI Shell", LOAD_OPTION_ACTIVE);\r
\r
  if (FixedPcdGetBool (PcdBootManagerEscape)) {\r
    Print (\r
      L"\n"\r
      L"    Esc or Down      to enter Boot Manager Menu.\n"\r
      L"    ENTER            to boot directly.\n"\r
      L"\n"\r
      );\r
  } else {\r
    Print (\r
      L"\n"\r
      L"    F2 or Down      to enter Boot Manager Menu.\n"\r
      L"    ENTER           to boot directly.\n"\r
      L"\n"\r
      );\r
  }\r
}\r
\r
/**\r
  This function is called each second during the boot manager waits the timeout.\r
\r
  @param TimeoutRemain  The remaining timeout.\r
**/\r
VOID\r
EFIAPI\r
PlatformBootManagerWaitCallback (\r
  UINT16  TimeoutRemain\r
  )\r
{\r
  if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
    mUniversalPayloadPlatformBootManagerOverrideInstance->WaitCallback (TimeoutRemain);\r
  }\r
\r
  return;\r
}\r
\r
/**\r
  The function is called when no boot option could be launched,\r
  including platform recovery options and options pointing to applications\r
  built into firmware volumes.\r
\r
  If this function returns, BDS attempts to enter an infinite loop.\r
**/\r
VOID\r
EFIAPI\r
PlatformBootManagerUnableToBoot (\r
  VOID\r
  )\r
{\r
  if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
    mUniversalPayloadPlatformBootManagerOverrideInstance->UnableToBoot ();\r
  }\r
\r
  return;\r
}\r
\r
/**\r
  Get/update PcdBootManagerMenuFile from GUID HOB which will be assigned in bootloader.\r
\r
  @param  ImageHandle   The firmware allocated handle for the EFI image.\r
  @param  SystemTable   A pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS       The entry point is executed successfully.\r
  @retval other             Some error occurs.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PlatformBootManagerLibConstructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS                           Status;\r
  UINTN                                Size;\r
  VOID                                 *GuidHob;\r
  UNIVERSAL_PAYLOAD_GENERIC_HEADER     *GenericHeader;\r
  UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU  *BootManagerMenuFile;\r
\r
  GuidHob = GetFirstGuidHob (&gEdkiiBootManagerMenuFileGuid);\r
\r
  if (GuidHob == NULL) {\r
    //\r
    // If the HOB is not create, the default value of PcdBootManagerMenuFile will be used.\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  GenericHeader = (UNIVERSAL_PAYLOAD_GENERIC_HEADER *)GET_GUID_HOB_DATA (GuidHob);\r
  if ((sizeof (UNIVERSAL_PAYLOAD_GENERIC_HEADER) > GET_GUID_HOB_DATA_SIZE (GuidHob)) || (GenericHeader->Length > GET_GUID_HOB_DATA_SIZE (GuidHob))) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  if (GenericHeader->Revision == UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU_REVISION) {\r
    BootManagerMenuFile = (UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU *)GET_GUID_HOB_DATA (GuidHob);\r
    if (BootManagerMenuFile->Header.Length < UNIVERSAL_PAYLOAD_SIZEOF_THROUGH_FIELD (UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU, FileName)) {\r
      return EFI_NOT_FOUND;\r
    }\r
\r
    Size   = sizeof (BootManagerMenuFile->FileName);\r
    Status = PcdSetPtrS (PcdBootManagerMenuFile, &Size, &BootManagerMenuFile->FileName);\r
    ASSERT_EFI_ERROR (Status);\r
  } else {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
Commit	Line	Data
04af8bf2 DG	1	/** @file\r
	2	This file include all platform action which can be customized\r
	3	by IBV/OEM.\r
	4	\r
d58016b7	5	Copyright (c) 2015 - 2021, Intel Corporation. All rights reserved.<BR>\r
04af8bf2 DG	6	SPDX-License-Identifier: BSD-2-Clause-Patent\r
	7	\r
	8	**/\r
	9	\r
	10	#include "PlatformBootManager.h"\r
	11	#include "PlatformConsole.h"\r
d58016b7	12	#include <Protocol/PlatformBootManagerOverride.h>\r
19a541d7	13	#include <Guid/BootManagerMenu.h>\r
19a541d7	14	#include <Library/HobLib.h>\r
d58016b7 ZL	15	\r
d58016b7 ZL	16	UNIVERSAL_PAYLOAD_PLATFORM_BOOT_MANAGER_OVERRIDE_PROTOCOL *mUniversalPayloadPlatformBootManagerOverrideInstance = NULL;\r
04af8bf2	17	\r
6ef57974 GD	18	/**\r
	19	Signal EndOfDxe event and install SMM Ready to lock protocol.\r
	20	\r
	21	**/\r
04af8bf2 DG	22	VOID\r
	23	InstallReadyToLock (\r
	24	VOID\r
	25	)\r
	26	{\r
e5efcf8b MK	27	EFI_STATUS Status;\r
	28	EFI_HANDLE Handle;\r
	29	EFI_SMM_ACCESS2_PROTOCOL *SmmAccess;\r
04af8bf2	30	\r
e5efcf8b	31	DEBUG ((DEBUG_INFO, "InstallReadyToLock entering......\n"));\r
04af8bf2 DG	32	//\r
	33	// Inform the SMM infrastructure that we're entering BDS and may run 3rd party code hereafter\r
	34	// Since PI1.2.1, we need signal EndOfDxe as ExitPmAuth\r
	35	//\r
	36	EfiEventGroupSignal (&gEfiEndOfDxeEventGroupGuid);\r
e5efcf8b	37	DEBUG ((DEBUG_INFO, "All EndOfDxe callbacks have returned successfully\n"));\r
04af8bf2 DG	38	\r
	39	//\r
	40	// Install DxeSmmReadyToLock protocol in order to lock SMM\r
	41	//\r
e5efcf8b	42	Status = gBS->LocateProtocol (&gEfiSmmAccess2ProtocolGuid, NULL, (VOID **)&SmmAccess);\r
04af8bf2 DG	43	if (!EFI_ERROR (Status)) {\r
	44	Handle = NULL;\r
	45	Status = gBS->InstallProtocolInterface (\r
	46	&Handle,\r
	47	&gEfiDxeSmmReadyToLockProtocolGuid,\r
	48	EFI_NATIVE_INTERFACE,\r
	49	NULL\r
	50	);\r
	51	ASSERT_EFI_ERROR (Status);\r
	52	}\r
	53	\r
e5efcf8b	54	DEBUG ((DEBUG_INFO, "InstallReadyToLock end\n"));\r
04af8bf2 DG	55	return;\r
	56	}\r
	57	\r
	58	/**\r
	59	Return the index of the load option in the load option array.\r
	60	\r
	61	The function consider two load options are equal when the\r
	62	OptionType, Attributes, Description, FilePath and OptionalData are equal.\r
	63	\r
	64	@param Key Pointer to the load option to be found.\r
	65	@param Array Pointer to the array of load options to be found.\r
	66	@param Count Number of entries in the Array.\r
	67	\r
	68	@retval -1 Key wasn't found in the Array.\r
	69	@retval 0 ~ Count-1 The index of the Key in the Array.\r
	70	**/\r
	71	INTN\r
	72	PlatformFindLoadOption (\r
e5efcf8b MK	73	IN CONST EFI_BOOT_MANAGER_LOAD_OPTION *Key,\r
	74	IN CONST EFI_BOOT_MANAGER_LOAD_OPTION *Array,\r
	75	IN UINTN Count\r
	76	)\r
04af8bf2	77	{\r
e5efcf8b	78	UINTN Index;\r
04af8bf2 DG	79	\r
	80	for (Index = 0; Index < Count; Index++) {\r
	81	if ((Key->OptionType == Array[Index].OptionType) &&\r
	82	(Key->Attributes == Array[Index].Attributes) &&\r
	83	(StrCmp (Key->Description, Array[Index].Description) == 0) &&\r
	84	(CompareMem (Key->FilePath, Array[Index].FilePath, GetDevicePathSize (Key->FilePath)) == 0) &&\r
	85	(Key->OptionalDataSize == Array[Index].OptionalDataSize) &&\r
e5efcf8b MK	86	(CompareMem (Key->OptionalData, Array[Index].OptionalData, Key->OptionalDataSize) == 0))\r
	87	{\r
	88	return (INTN)Index;\r
04af8bf2 DG	89	}\r
	90	}\r
	91	\r
	92	return -1;\r
	93	}\r
	94	\r
	95	/**\r
	96	Register a boot option using a file GUID in the FV.\r
	97	\r
	98	@param FileGuid The file GUID name in FV.\r
	99	@param Description The boot option description.\r
	100	@param Attributes The attributes used for the boot option loading.\r
	101	**/\r
	102	VOID\r
	103	PlatformRegisterFvBootOption (\r
e5efcf8b MK	104	EFI_GUID *FileGuid,\r
	105	CHAR16 *Description,\r
	106	UINT32 Attributes\r
	107	)\r
04af8bf2	108	{\r
e5efcf8b MK	109	EFI_STATUS Status;\r
	110	UINTN OptionIndex;\r
	111	EFI_BOOT_MANAGER_LOAD_OPTION NewOption;\r
	112	EFI_BOOT_MANAGER_LOAD_OPTION *BootOptions;\r
	113	UINTN BootOptionCount;\r
	114	MEDIA_FW_VOL_FILEPATH_DEVICE_PATH FileNode;\r
	115	EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;\r
	116	EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r
	117	\r
	118	Status = gBS->HandleProtocol (gImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **)&LoadedImage);\r
04af8bf2 DG	119	ASSERT_EFI_ERROR (Status);\r
	120	\r
	121	EfiInitializeFwVolDevicepathNode (&FileNode, FileGuid);\r
	122	DevicePath = AppendDevicePathNode (\r
	123	DevicePathFromHandle (LoadedImage->DeviceHandle),\r
e5efcf8b MK	124	(EFI_DEVICE_PATH_PROTOCOL *)&FileNode\r
e5efcf8b MK	125	);\r
04af8bf2 DG	126	\r
	127	Status = EfiBootManagerInitializeLoadOption (\r
	128	&NewOption,\r
	129	LoadOptionNumberUnassigned,\r
	130	LoadOptionTypeBoot,\r
	131	Attributes,\r
	132	Description,\r
	133	DevicePath,\r
	134	NULL,\r
	135	0\r
e5efcf8b	136	);\r
04af8bf2 DG	137	if (!EFI_ERROR (Status)) {\r
	138	BootOptions = EfiBootManagerGetLoadOptions (&BootOptionCount, LoadOptionTypeBoot);\r
	139	\r
	140	OptionIndex = PlatformFindLoadOption (&NewOption, BootOptions, BootOptionCount);\r
	141	\r
	142	if (OptionIndex == -1) {\r
e5efcf8b	143	Status = EfiBootManagerAddLoadOptionVariable (&NewOption, (UINTN)-1);\r
04af8bf2 DG	144	ASSERT_EFI_ERROR (Status);\r
04af8bf2 DG	145	}\r
e5efcf8b	146	\r
04af8bf2 DG	147	EfiBootManagerFreeLoadOption (&NewOption);\r
	148	EfiBootManagerFreeLoadOptions (BootOptions, BootOptionCount);\r
	149	}\r
	150	}\r
	151	\r
	152	/**\r
	153	Do the platform specific action before the console is connected.\r
	154	\r
	155	Such as:\r
	156	Update console variable;\r
	157	Register new Driver#### or Boot####;\r
	158	Signal ReadyToLock event.\r
	159	**/\r
	160	VOID\r
	161	EFIAPI\r
	162	PlatformBootManagerBeforeConsole (\r
	163	VOID\r
e5efcf8b	164	)\r
04af8bf2	165	{\r
e5efcf8b	166	EFI_INPUT_KEY Enter;\r
af74efe4	167	EFI_INPUT_KEY CustomKey;\r
e5efcf8b MK	168	EFI_INPUT_KEY Down;\r
	169	EFI_BOOT_MANAGER_LOAD_OPTION BootOption;\r
	170	EFI_STATUS Status;\r
d58016b7	171	\r
e5efcf8b	172	Status = gBS->LocateProtocol (&gUniversalPayloadPlatformBootManagerOverrideProtocolGuid, NULL, (VOID **)&mUniversalPayloadPlatformBootManagerOverrideInstance);\r
d58016b7 ZL	173	if (EFI_ERROR (Status)) {\r
	174	mUniversalPayloadPlatformBootManagerOverrideInstance = NULL;\r
	175	}\r
e5efcf8b MK	176	\r
	177	if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
	178	mUniversalPayloadPlatformBootManagerOverrideInstance->BeforeConsole ();\r
d58016b7 ZL	179	return;\r
d58016b7 ZL	180	}\r
04af8bf2	181	\r
04af8bf2 DG	182	//\r
	183	// Register ENTER as CONTINUE key\r
	184	//\r
	185	Enter.ScanCode = SCAN_NULL;\r
	186	Enter.UnicodeChar = CHAR_CARRIAGE_RETURN;\r
	187	EfiBootManagerRegisterContinueKeyOption (0, &Enter, NULL);\r
	188	\r
af74efe4 SR	189	if (FixedPcdGetBool (PcdBootManagerEscape)) {\r
	190	//\r
	191	// Map Esc to Boot Manager Menu\r
	192	//\r
	193	CustomKey.ScanCode = SCAN_ESC;\r
	194	CustomKey.UnicodeChar = CHAR_NULL;\r
	195	} else {\r
	196	//\r
	197	// Map Esc to Boot Manager Menu\r
	198	//\r
	199	CustomKey.ScanCode = SCAN_F2;\r
	200	CustomKey.UnicodeChar = CHAR_NULL;\r
	201	}\r
	202	\r
04af8bf2	203	EfiBootManagerGetBootManagerMenu (&BootOption);\r
af74efe4	204	EfiBootManagerAddKeyOptionVariable (NULL, (UINT16)BootOption.OptionNumber, 0, &CustomKey, NULL);\r
04af8bf2 DG	205	\r
	206	//\r
	207	// Also add Down key to Boot Manager Menu since some serial terminals don't support F2 key.\r
	208	//\r
	209	Down.ScanCode = SCAN_DOWN;\r
	210	Down.UnicodeChar = CHAR_NULL;\r
	211	EfiBootManagerGetBootManagerMenu (&BootOption);\r
e5efcf8b	212	EfiBootManagerAddKeyOptionVariable (NULL, (UINT16)BootOption.OptionNumber, 0, &Down, NULL);\r
04af8bf2 DG	213	\r
	214	//\r
	215	// Install ready to lock.\r
	216	// This needs to be done before option rom dispatched.\r
	217	//\r
	218	InstallReadyToLock ();\r
	219	\r
	220	//\r
	221	// Dispatch deferred images after EndOfDxe event and ReadyToLock installation.\r
	222	//\r
	223	EfiBootManagerDispatchDeferredImages ();\r
2e1e8c35 GD	224	\r
2e1e8c35 GD	225	PlatformConsoleInit ();\r
04af8bf2 DG	226	}\r
	227	\r
	228	/**\r
	229	Do the platform specific action after the console is connected.\r
	230	\r
	231	Such as:\r
	232	Dynamically switch output mode;\r
	233	Signal console ready platform customized event;\r
	234	Run diagnostics like memory testing;\r
	235	Connect certain devices;\r
	236	Dispatch additional option roms.\r
	237	**/\r
	238	VOID\r
	239	EFIAPI\r
	240	PlatformBootManagerAfterConsole (\r
	241	VOID\r
e5efcf8b	242	)\r
04af8bf2 DG	243	{\r
	244	EFI_GRAPHICS_OUTPUT_BLT_PIXEL Black;\r
	245	EFI_GRAPHICS_OUTPUT_BLT_PIXEL White;\r
79aab22f SR	246	EDKII_PLATFORM_LOGO_PROTOCOL *PlatformLogo;\r
79aab22f SR	247	EFI_STATUS Status;\r
04af8bf2	248	\r
e5efcf8b MK	249	if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
e5efcf8b MK	250	mUniversalPayloadPlatformBootManagerOverrideInstance->AfterConsole ();\r
d58016b7 ZL	251	return;\r
d58016b7 ZL	252	}\r
e5efcf8b	253	\r
04af8bf2 DG	254	Black.Blue = Black.Green = Black.Red = Black.Reserved = 0;\r
	255	White.Blue = White.Green = White.Red = White.Reserved = 0xFF;\r
	256	\r
79aab22f SR	257	Status = gBS->LocateProtocol (&gEdkiiPlatformLogoProtocolGuid, NULL, (VOID **)&PlatformLogo);\r
	258	\r
	259	if (!EFI_ERROR (Status)) {\r
	260	gST->ConOut->ClearScreen (gST->ConOut);\r
	261	BootLogoEnableLogo ();\r
	262	}\r
	263	\r
04af8bf2 DG	264	EfiBootManagerConnectAll ();\r
	265	EfiBootManagerRefreshAllBootOption ();\r
	266	\r
	267	//\r
	268	// Register UEFI Shell\r
	269	//\r
	270	PlatformRegisterFvBootOption (PcdGetPtr (PcdShellFile), L"UEFI Shell", LOAD_OPTION_ACTIVE);\r
	271	\r
af74efe4 SR	272	if (FixedPcdGetBool (PcdBootManagerEscape)) {\r
	273	Print (\r
	274	L"\n"\r
07c8e5e5	275	L" Esc or Down to enter Boot Manager Menu.\n"\r
d2191197	276	L" ENTER to boot directly.\n"\r
af74efe4 SR	277	L"\n"\r
	278	);\r
	279	} else {\r
	280	Print (\r
	281	L"\n"\r
07c8e5e5 SR	282	L" F2 or Down to enter Boot Manager Menu.\n"\r
07c8e5e5 SR	283	L" ENTER to boot directly.\n"\r
af74efe4 SR	284	L"\n"\r
	285	);\r
	286	}\r
04af8bf2 DG	287	}\r
	288	\r
	289	/**\r
	290	This function is called each second during the boot manager waits the timeout.\r
	291	\r
	292	@param TimeoutRemain The remaining timeout.\r
	293	**/\r
	294	VOID\r
	295	EFIAPI\r
	296	PlatformBootManagerWaitCallback (\r
e5efcf8b MK	297	UINT16 TimeoutRemain\r
e5efcf8b MK	298	)\r
04af8bf2	299	{\r
e5efcf8b	300	if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
d58016b7 ZL	301	mUniversalPayloadPlatformBootManagerOverrideInstance->WaitCallback (TimeoutRemain);\r
d58016b7 ZL	302	}\r
e5efcf8b	303	\r
04af8bf2 DG	304	return;\r
	305	}\r
	306	\r
	307	/**\r
	308	The function is called when no boot option could be launched,\r
	309	including platform recovery options and options pointing to applications\r
	310	built into firmware volumes.\r
	311	\r
	312	If this function returns, BDS attempts to enter an infinite loop.\r
	313	**/\r
	314	VOID\r
	315	EFIAPI\r
	316	PlatformBootManagerUnableToBoot (\r
	317	VOID\r
	318	)\r
	319	{\r
e5efcf8b MK	320	if (mUniversalPayloadPlatformBootManagerOverrideInstance != NULL) {\r
e5efcf8b MK	321	mUniversalPayloadPlatformBootManagerOverrideInstance->UnableToBoot ();\r
d58016b7	322	}\r
e5efcf8b	323	\r
04af8bf2 DG	324	return;\r
	325	}\r
	326	\r
19a541d7	327	/**\r
	328	Get/update PcdBootManagerMenuFile from GUID HOB which will be assigned in bootloader.\r
	329	\r
	330	@param ImageHandle The firmware allocated handle for the EFI image.\r
	331	@param SystemTable A pointer to the EFI System Table.\r
	332	\r
	333	@retval EFI_SUCCESS The entry point is executed successfully.\r
	334	@retval other Some error occurs.\r
	335	\r
	336	**/\r
	337	EFI_STATUS\r
	338	EFIAPI\r
	339	PlatformBootManagerLibConstructor (\r
	340	IN EFI_HANDLE ImageHandle,\r
	341	IN EFI_SYSTEM_TABLE *SystemTable\r
e5efcf8b	342	)\r
19a541d7	343	{\r
	344	EFI_STATUS Status;\r
	345	UINTN Size;\r
	346	VOID *GuidHob;\r
	347	UNIVERSAL_PAYLOAD_GENERIC_HEADER *GenericHeader;\r
	348	UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU *BootManagerMenuFile;\r
	349	\r
	350	GuidHob = GetFirstGuidHob (&gEdkiiBootManagerMenuFileGuid);\r
	351	\r
	352	if (GuidHob == NULL) {\r
	353	//\r
	354	// If the HOB is not create, the default value of PcdBootManagerMenuFile will be used.\r
	355	//\r
	356	return EFI_SUCCESS;\r
	357	}\r
	358	\r
e5efcf8b	359	GenericHeader = (UNIVERSAL_PAYLOAD_GENERIC_HEADER *)GET_GUID_HOB_DATA (GuidHob);\r
19a541d7	360	if ((sizeof (UNIVERSAL_PAYLOAD_GENERIC_HEADER) > GET_GUID_HOB_DATA_SIZE (GuidHob)) \|\| (GenericHeader->Length > GET_GUID_HOB_DATA_SIZE (GuidHob))) {\r
	361	return EFI_NOT_FOUND;\r
	362	}\r
e5efcf8b	363	\r
19a541d7	364	if (GenericHeader->Revision == UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU_REVISION) {\r
e5efcf8b	365	BootManagerMenuFile = (UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU *)GET_GUID_HOB_DATA (GuidHob);\r
19a541d7	366	if (BootManagerMenuFile->Header.Length < UNIVERSAL_PAYLOAD_SIZEOF_THROUGH_FIELD (UNIVERSAL_PAYLOAD_BOOT_MANAGER_MENU, FileName)) {\r
	367	return EFI_NOT_FOUND;\r
	368	}\r
e5efcf8b MK	369	\r
e5efcf8b MK	370	Size = sizeof (BootManagerMenuFile->FileName);\r
19a541d7	371	Status = PcdSetPtrS (PcdBootManagerMenuFile, &Size, &BootManagerMenuFile->FileName);\r
	372	ASSERT_EFI_ERROR (Status);\r
	373	} else {\r
	374	return EFI_NOT_FOUND;\r
	375	}\r
	376	\r
	377	return EFI_SUCCESS;\r
	378	}\r