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

[mirror_edk2.git] / UefiCpuPkg / Library / SmmCpuRendezvousLib / SmmCpuRendezvousLib.c

/** @file\r
  SMM CPU Rendezvous sevice implement.\r
\r
  Copyright (c) 2022, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <Base.h>\r
#include <Uefi.h>\r
#include <Library/BaseLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/MmServicesTableLib.h>\r
#include <Protocol/SmmCpuService.h>\r
#include <Library/SmmCpuRendezvousLib.h>\r
\r
STATIC EDKII_SMM_CPU_RENDEZVOUS_PROTOCOL  *mSmmCpuRendezvous = NULL;\r
STATIC VOID                               *mRegistration     = NULL;\r
\r
/**\r
  Callback function to wait Smm cpu rendezvous service located.\r
\r
  SmmCpuRendezvousLib need to support MM_STANDALONE and DXE_SMM_DRIVER driver.\r
  So do not use library constructor to locate the protocol.\r
\r
  @param[in] Protocol   Points to the protocol's unique identifier.\r
  @param[in] Interface  Points to the interface instance.\r
  @param[in] Handle     The handle on which the interface was installed.\r
\r
  @retval EFI_SUCCESS  Notification runs successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmmCpuRendezvousProtocolNotify (\r
  IN CONST EFI_GUID    *Protocol,\r
  IN       VOID        *Interface,\r
  IN       EFI_HANDLE  Handle\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  Status = gMmst->MmLocateProtocol (\r
                    &gEdkiiSmmCpuRendezvousProtocolGuid,\r
                    NULL,\r
                    (VOID **)&mSmmCpuRendezvous\r
                    );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This routine wait for all AP processors to arrive in SMM.\r
\r
  @param[in] BlockingMode  Blocking mode or non-blocking mode.\r
\r
  @retval EFI_SUCCESS  All avaiable APs arrived.\r
  @retval EFI_TIMEOUT  Wait for all APs until timeout.\r
  @retval OTHER        Fail to register SMM CPU Rendezvous service Protocol.\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmmWaitForAllProcessor (\r
  IN BOOLEAN  BlockingMode\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  if ((mRegistration == NULL) && (mSmmCpuRendezvous == NULL)) {\r
    //\r
    // Locate SMM cpu rendezvous protocol for the first time execute the function.\r
    //\r
    Status = gMmst->MmLocateProtocol (\r
                      &gEdkiiSmmCpuRendezvousProtocolGuid,\r
                      NULL,\r
                      (VOID **)&mSmmCpuRendezvous\r
                      );\r
    if (EFI_ERROR (Status)) {\r
      Status = gMmst->MmRegisterProtocolNotify (\r
                        &gEdkiiSmmCpuRendezvousProtocolGuid,\r
                        SmmCpuRendezvousProtocolNotify,\r
                        &mRegistration\r
                        );\r
      if (EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
    }\r
  }\r
\r
  //\r
  // The platform have not set up. It doesn't need smm cpu rendezvous.\r
  //\r
  if (mSmmCpuRendezvous == NULL) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  Status = mSmmCpuRendezvous->WaitForAllProcessor (\r
                                mSmmCpuRendezvous,\r
                                BlockingMode\r
                                );\r
  return Status;\r
}\r