4a68176cb5
REF? https://bugzilla.tianocore.org/show_bug.cgi?id=3815 This patch define a new Protocol with the new services SmmWaitForAllProcessor(), which can be used by SMI handler to optionally wait for other APs to complete SMM rendezvous in relaxed AP mode. A new library SmmCpuRendezvousLib is provided to abstract the service into library API to simple SMI handler code. Cc: Eric Dong <eric.dong@intel.com> Reviewed-by: Ray Ni <ray.ni@intel.com> Cc: Rahul Kumar <rahul1.kumar@intel.com> Cc: Siyuan Fu <siyuan.fu@intel.com> Cc: Zhihao Li <zhihao.li@intel.com> Signed-off-by: Zhihao Li <zhihao.li@intel.com>
104 lines
2.9 KiB
C
104 lines
2.9 KiB
C
/** @file
|
|
SMM CPU Rendezvous sevice implement.
|
|
|
|
Copyright (c) 2022, Intel Corporation. All rights reserved.<BR>
|
|
SPDX-License-Identifier: BSD-2-Clause-Patent
|
|
|
|
**/
|
|
|
|
#include <Base.h>
|
|
#include <Uefi.h>
|
|
#include <Library/BaseLib.h>
|
|
#include <Library/DebugLib.h>
|
|
#include <Library/MmServicesTableLib.h>
|
|
#include <Protocol/SmmCpuService.h>
|
|
#include <Library/SmmCpuRendezvousLib.h>
|
|
|
|
STATIC EDKII_SMM_CPU_RENDEZVOUS_PROTOCOL *mSmmCpuRendezvous = NULL;
|
|
STATIC VOID *mRegistration = NULL;
|
|
|
|
/**
|
|
Callback function to wait Smm cpu rendezvous service located.
|
|
|
|
SmmCpuRendezvousLib need to support MM_STANDALONE and DXE_SMM_DRIVER driver.
|
|
So do not use library constructor to locate the protocol.
|
|
|
|
@param[in] Protocol Points to the protocol's unique identifier.
|
|
@param[in] Interface Points to the interface instance.
|
|
@param[in] Handle The handle on which the interface was installed.
|
|
|
|
@retval EFI_SUCCESS Notification runs successfully.
|
|
|
|
**/
|
|
EFI_STATUS
|
|
EFIAPI
|
|
SmmCpuRendezvousProtocolNotify (
|
|
IN CONST EFI_GUID *Protocol,
|
|
IN VOID *Interface,
|
|
IN EFI_HANDLE Handle
|
|
)
|
|
{
|
|
EFI_STATUS Status;
|
|
|
|
Status = gMmst->MmLocateProtocol (
|
|
&gEdkiiSmmCpuRendezvousProtocolGuid,
|
|
NULL,
|
|
(VOID **)&mSmmCpuRendezvous
|
|
);
|
|
ASSERT_EFI_ERROR (Status);
|
|
|
|
return EFI_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
This routine wait for all AP processors to arrive in SMM.
|
|
|
|
@param[in] BlockingMode Blocking mode or non-blocking mode.
|
|
|
|
@retval EFI_SUCCESS All avaiable APs arrived.
|
|
@retval EFI_TIMEOUT Wait for all APs until timeout.
|
|
@retval OTHER Fail to register SMM CPU Rendezvous service Protocol.
|
|
**/
|
|
EFI_STATUS
|
|
EFIAPI
|
|
SmmWaitForAllProcessor (
|
|
IN BOOLEAN BlockingMode
|
|
)
|
|
{
|
|
EFI_STATUS Status;
|
|
|
|
if ((mRegistration == NULL) && (mSmmCpuRendezvous == NULL)) {
|
|
//
|
|
// Locate SMM cpu rendezvous protocol for the first time execute the function.
|
|
//
|
|
Status = gMmst->MmLocateProtocol (
|
|
&gEdkiiSmmCpuRendezvousProtocolGuid,
|
|
NULL,
|
|
(VOID **)&mSmmCpuRendezvous
|
|
);
|
|
if (EFI_ERROR (Status)) {
|
|
Status = gMmst->MmRegisterProtocolNotify (
|
|
&gEdkiiSmmCpuRendezvousProtocolGuid,
|
|
SmmCpuRendezvousProtocolNotify,
|
|
&mRegistration
|
|
);
|
|
if (EFI_ERROR (Status)) {
|
|
return Status;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// The platform have not set up. It doesn't need smm cpu rendezvous.
|
|
//
|
|
if (mSmmCpuRendezvous == NULL) {
|
|
return EFI_SUCCESS;
|
|
}
|
|
|
|
Status = mSmmCpuRendezvous->WaitForAllProcessor (
|
|
mSmmCpuRendezvous,
|
|
BlockingMode
|
|
);
|
|
return Status;
|
|
}
|