UefiCpuPkg/PiSmmCpuDxeSmm: Consume SmmCpuPlatformHookAfterCpuSync func

This patch is for PiSmmCpuDxeSmm driver to consume the function of SmmCpuPlatformHookAfterCpuSync(). With the function, SMM CPU driver can perform the platform specific items after BSP and AP finish the sync and before the SMI handlers. Signed-off-by: Jiaxin Wu <jiaxin.wu@intel.com>
tianocore · Sep 10, 2024 · 06504e7 · 06504e7
1 parent c99de4a
commit 06504e7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/UefiCpuPkg/PiSmmCpuDxeSmm/MpService.c b/UefiCpuPkg/PiSmmCpuDxeSmm/MpService.c
@@ -731,6 +731,9 @@ APHandler (
   UINTN          BspIndex;
   MTRR_SETTINGS  Mtrrs;
   EFI_STATUS     ProcedureStatus;
+  BOOLEAN        PlatformHookDone;
+
+  PlatformHookDone = FALSE;
 
   //
   // Timeout BSP
@@ -846,6 +849,14 @@ APHandler (
     //
     SmmCpuSyncWaitForBsp (mSmmMpSyncData->SyncContext, CpuIndex, BspIndex);
 
+    //
+    // Perform SMM CPU Platform Hook After BSP and AP Sync
+    //
+    if (!PlatformHookDone) {
+      SmmCpuPlatformHookAfterCpuSync ();
+      PlatformHookDone = TRUE;
+    }
+
     //
     // Check if BSP wants to exit SMM
     //

diff --git a/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuCommon.c b/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuCommon.c
@@ -1534,5 +1534,10 @@ PerformPreTasks (
   VOID
   )
 {
+  //
+  // Perform SMM CPU Platform Hook After BSP and AP Sync
+  //
+  SmmCpuPlatformHookAfterCpuSync ();
+
   RestoreSmmConfigurationInS3 ();
 }