UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm

[mirror_edk2.git] / UefiCpuPkg / CpuFeatures / CpuFeaturesDxe.c

/** @file\r
  CPU Features DXE driver to initialize CPU features.\r
\r
  Copyright (c) 2017 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <PiDxe.h>\r
\r
#include <Library/BaseLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/UefiLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/RegisterCpuFeaturesLib.h>\r
#include <Library/HobLib.h>\r
\r
#include <Protocol/SmmConfiguration.h>\r
#include <Guid/CpuFeaturesInitDone.h>\r
\r
/**\r
  Worker function to perform CPU feature initialization.\r
\r
**/\r
VOID\r
CpuFeaturesInitializeWorker (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
  EFI_HANDLE  Handle;\r
\r
  CpuFeaturesDetect ();\r
\r
  CpuFeaturesInitialize ();\r
\r
  //\r
  // Install CPU Features Init Done Protocol\r
  //\r
  Handle = NULL;\r
  Status = gBS->InstallProtocolInterface (\r
                  &Handle,\r
                  &gEdkiiCpuFeaturesInitDoneGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
/**\r
  Event notification that initialize CPU features when gEfiSmmConfigurationProtocol installs.\r
\r
  @param[in]  Event                 The Event that is being processed, not used.\r
  @param[in]  Context               Event Context, not used.\r
**/\r
VOID\r
EFIAPI\r
SmmConfigurationEventNotify (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  EFI_STATUS                      Status;\r
  EFI_SMM_CONFIGURATION_PROTOCOL  *SmmConfiguration;\r
\r
  //\r
  // Make sure this notification is for this handler\r
  //\r
  Status = gBS->LocateProtocol (&gEfiSmmConfigurationProtocolGuid, NULL, (VOID **)&SmmConfiguration);\r
  if (EFI_ERROR (Status)) {\r
    return;\r
  }\r
\r
  CpuFeaturesInitializeWorker ();\r
}\r
\r
/**\r
  CPU Features driver entry point function.\r
\r
  If PcdCpuFeaturesInitAfterSmmRelocation is TRUE, it will register one\r
  SMM Configuration Protocol notify function to perform CPU features\r
  initialization. Otherwise, it will perform CPU features initialization\r
  directly.\r
\r
  @param ImageHandle     Image handle this driver.\r
  @param SystemTable     Pointer to the System Table.\r
\r
  @retval EFI_SUCCESS   CPU Features is initialized successfully.\r
**/\r
EFI_STATUS\r
EFIAPI\r
CpuFeaturesDxeInitialize (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  VOID        *Registration;\r
  EFI_STATUS  Status;\r
  EFI_HANDLE  Handle;\r
\r
  if (GetFirstGuidHob (&gEdkiiCpuFeaturesInitDoneGuid) != NULL) {\r
    //\r
    // Try to find HOB first. This HOB exist means CPU features have\r
    // been initialized by CpuFeaturesPei driver, just install\r
    // gEdkiiCpuFeaturesInitDoneGuid.\r
    //\r
    Handle = NULL;\r
    Status = gBS->InstallProtocolInterface (\r
                    &Handle,\r
                    &gEdkiiCpuFeaturesInitDoneGuid,\r
                    EFI_NATIVE_INTERFACE,\r
                    NULL\r
                    );\r
    ASSERT_EFI_ERROR (Status);\r
    return Status;\r
  }\r
\r
  if (PcdGetBool (PcdCpuFeaturesInitAfterSmmRelocation)) {\r
    //\r
    // Install notification callback on SMM Configuration Protocol\r
    //\r
    EfiCreateProtocolNotifyEvent (\r
      &gEfiSmmConfigurationProtocolGuid,\r
      TPL_CALLBACK,\r
      SmmConfigurationEventNotify,\r
      NULL,\r
      &Registration\r
      );\r
  } else {\r
    CpuFeaturesInitializeWorker ();\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r