EdkCompatibilityPkg: Add DxeSmmReadyToLockOnExitPmAuthThunk driver

[mirror_edk2.git] / EdkCompatibilityPkg / Compatibility / DxeSmmReadyToLockOnExitPmAuthThunk / DxeSmmReadyToLockOnExitPmAuthThunk.c

/** @file\r
  DxeSmmReadyToLock Protocol on ExitPmAuth Protocol Thunk driver.\r
  R8 platform uses ExitPmAuth point to lock SMRAM and SMM API.\r
  But R9 uses DxeSmmReadyToLock. We need a thunk driver to convert this event.\r
\r
Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions\r
of the BSD License which accompanies this distribution.  The\r
full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include  "DxeSmmReadyToLockOnExitPmAuthThunk.h"\r
\r
/**\r
  ExitPmAuth Protocol notification event handler.\r
\r
  @param[in] Event    Event whose notification function is being invoked.\r
  @param[in] Context  Pointer to the notification function's context.\r
**/\r
VOID\r
EFIAPI\r
ExitPmAuthProtocolNotification (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  EFI_STATUS Status;\r
  EFI_HANDLE ImageHandle;\r
  VOID       *ExitPmAuth;\r
\r
  //\r
  // Add more check to locate protocol after got event, because\r
  // R8 ECP will signal this event immediately once it is register\r
  // just in case it is already installed.\r
  //\r
  Status = gBS->LocateProtocol (\r
                  &gExitPmAuthProtocolGuid,\r
                  NULL,\r
                  &ExitPmAuth\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return ;\r
  }\r
\r
  //\r
  // Install DxeSmmReadyToLock protocol to let PI SMM lock\r
  //\r
  ImageHandle = NULL;\r
  Status = gBS->InstallProtocolInterface (\r
                  &ImageHandle,\r
                  &gEfiDxeSmmReadyToLockProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
/**\r
  Entry Point for DxeSmmReadyToLock Protocol on ExitPmAuth Protocol Thunk driver.\r
\r
  @param[in] ImageHandle  Image handle of this driver.\r
  @param[in] SystemTable  A Pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS  The entry point is executed successfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
DxeSmmReadyToLockMain (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  VOID *Registration;\r
\r
  //\r
  // Install notifications for required protocols\r
  //\r
  EfiCreateProtocolNotifyEvent (\r
    &gExitPmAuthProtocolGuid,\r
    TPL_CALLBACK,\r
    ExitPmAuthProtocolNotification,\r
    NULL,\r
    &Registration\r
    );\r
\r
  return EFI_SUCCESS;\r
}\r