CorebootPayloadPkg: Use EfiEventGroupSignal from UefiLib

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

diff --git a/CorebootPayloadPkg/Library/PlatformBootManagerLib/PlatformBootManager.c b/CorebootPayloadPkg/Library/PlatformBootManagerLib/PlatformBootManager.c
index 200ea9580877a41648a4f4ebe4e28930491b05d2..7e92441da11f8de28e2e72f24b2461cb6d431cb4 100644 (file)
--- a/CorebootPayloadPkg/Library/PlatformBootManagerLib/PlatformBootManager.c
+++ b/CorebootPayloadPkg/Library/PlatformBootManagerLib/PlatformBootManager.c
@@ -2,7 +2,7 @@
   This file include all platform action which can be customized\r
   by IBV/OEM.\r
 \r
-Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2015 - 2017, Intel Corporation. All rights reserved.<BR>\r
 This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
@@ -16,6 +16,42 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include "PlatformBootManager.h"\r
 #include "PlatformConsole.h"\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
@@ -124,6 +160,7 @@ PlatformBootManagerBeforeConsole (
 {\r
   EFI_INPUT_KEY                Enter;\r
   EFI_INPUT_KEY                F2;\r
+  EFI_INPUT_KEY                Down;\r
   EFI_BOOT_MANAGER_LOAD_OPTION BootOption;\r
 \r
   PlatformConsoleInit ();\r
@@ -144,9 +181,23 @@ PlatformBootManagerBeforeConsole (
   EfiBootManagerAddKeyOptionVariable (NULL, (UINT16) BootOption.OptionNumber, 0, &F2, NULL);\r
 \r
   //\r
-  // Register UEFI Shell\r
+  // Also add Down key to Boot Manager Menu since some serial terminals don't support F2 key.\r
   //\r
-  PlatformRegisterFvBootOption (PcdGetPtr (PcdShellFile), L"UEFI Shell", LOAD_OPTION_ACTIVE);\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
 \r
 /**\r
@@ -174,10 +225,15 @@ PlatformBootManagerAfterConsole (
   EfiBootManagerConnectAll ();\r
   EfiBootManagerRefreshAllBootOption ();\r
 \r
+  //\r
+  // Register UEFI Shell\r
+  //\r
+  PlatformRegisterFvBootOption (PcdGetPtr (PcdShellFile), L"UEFI Shell", LOAD_OPTION_ACTIVE);\r
+  \r
   Print (\r
     L"\n"\r
-    L"F2      to enter Boot Manager Menu.\n"\r
-    L"ENTER   to boot directly.\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