SmmBaseHelper: Update MP state in Framework SMST before transferring control to Frame...

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

diff --git a/EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c b/EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c
index c4f4aee61ff6362322fce461e384023917feecd1..83647b997358082264f290c1813e32c2d991bfa3 100644 (file)
--- a/EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c
+++ b/EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c
@@ -672,8 +672,9 @@ LoadImage (
   if (!EFI_ERROR (Status)) {\r
     ///\r
     /// Update MP state in Framework SMST before transferring control to Framework SMM driver entry point\r
-    /// in case it may invoke AP\r
     ///\r
+    mFrameworkSmst->SmmStartupThisAp      = gSmst->SmmStartupThisAp;\r
+    mFrameworkSmst->NumberOfCpus          = mNumberOfProcessors;\r
     mFrameworkSmst->CurrentlyExecutingCpu = gSmst->CurrentlyExecutingCpu;\r
 \r
     Status = gBS->StartImage (*ImageHandle, NULL, NULL);\r