UefiCpuPkg/CpuDxe: Ap do loop routine to execute procedure

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Chen Fan <chen.fan.fnst@cn.fujitsu.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@16361 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/UefiCpuPkg/CpuDxe/CpuMp.c b/UefiCpuPkg/CpuDxe/CpuMp.c
index 878cdf25dbe13708c41f397398193b2f0b68aa9c..1b449a7a8de5bd78c2338f68fa9f6d025d8bdd5b 100644 (file)
--- a/UefiCpuPkg/CpuDxe/CpuMp.c
+++ b/UefiCpuPkg/CpuDxe/CpuMp.c
@@ -1108,10 +1108,41 @@ ProcessorToIdleState (
   IN      VOID                      *Context2   OPTIONAL\r
   )\r
 {\r
-  DEBUG ((DEBUG_INFO, "Ap apicid is %d\n", GetApicId ()));\r
+  UINTN                 ProcessorNumber;\r
+  CPU_DATA_BLOCK        *CpuData;\r
+  EFI_AP_PROCEDURE      Procedure;\r
+  VOID                  *ProcedureArgument;\r
+\r
+  WhoAmI (&mMpServicesTemplate, &ProcessorNumber);\r
+  CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
 \r
   AsmApDoneWithCommonStack ();\r
 \r
+  while (TRUE) {\r
+    while (!AcquireSpinLockOrFail (&CpuData->CpuDataLock)) {\r
+      CpuPause ();\r
+    }\r
+\r
+    ProcedureArgument = CpuData->Parameter;\r
+    Procedure = CpuData->Procedure;\r
+    ReleaseSpinLock (&CpuData->CpuDataLock);\r
+\r
+    if (Procedure != NULL) {\r
+      Procedure (ProcedureArgument);\r
+\r
+      while (!AcquireSpinLockOrFail (&CpuData->CpuDataLock)) {\r
+        CpuPause ();\r
+      }\r
+\r
+      CpuData->Procedure = NULL;\r
+      ReleaseSpinLock (&CpuData->CpuDataLock);\r
+\r
+      SetApState (CpuData, CpuStateFinished);\r
+    }\r
+\r
+    CpuPause ();\r
+  }\r
+\r
   CpuSleep ();\r
   CpuDeadLoop ();\r
 }\r