REBASE ArmPkg: ArmPsciMpServicesDxe: Fix CPU resource leakage

This change will fix an issue where the timeout of an AP function will cause this AP never being to accept task again. Signed-off-by: Kun Qin <kuqin@microsoft.com>
microsoft · Dec 19, 2023 · 324728d · 324728d
1 parent 31b11c9
commit 324728d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ArmPkg/Drivers/ArmPsciMpServicesDxe/ArmPsciMpServicesDxe.c b/ArmPkg/Drivers/ArmPsciMpServicesDxe/ArmPsciMpServicesDxe.c
@@ -754,6 +754,13 @@ StartupThisAP (
     return EFI_SUCCESS;
   }
 
+  // MU_CHANGE: Set the timer anyway, otherwise this AP is spent on this boot if the AP routine timeout.
+  gBS->SetTimer (
+         CpuData->CheckThisAPEvent,
+         TimerPeriodic,
+         POLL_INTERVAL_US
+         );
+
   // Blocking
   while (TRUE) {
     if (GetApState (CpuData) == CpuStateFinished) {