From 5b9fc2f0890019cbb204f187abd1c41db51f4b93 Mon Sep 17 00:00:00 2001
From: rsun3 <rsun3@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 17 Aug 2010 09:43:50 +0000
Subject: SmmBaseHelper: Update MP state in Framework SMST before transferring
 control to Framework SMM driver entry point.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10802 6f19259b-4bc3-4df7-8a09-765794883524
---
 EdkCompatibilityPkg/Compatibility/SmmBaseHelper/SmmBaseHelper.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'EdkCompatibilityPkg')

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