]> git.proxmox.com Git - mirror_edk2.git/blobdiff - UefiCpuPkg/CpuDxe/CpuMp.c
UefiCpuPkg: Replace BSD License with BSD+Patent License
[mirror_edk2.git] / UefiCpuPkg / CpuDxe / CpuMp.c
index f864de1f6f4db3eb31d5a5fb472a24826a014ecf..de6c6a5c330c97fadc23df164ad0f3af98bfa0e4 100644 (file)
@@ -1,33 +1,16 @@
 /** @file\r
-  CPU DXE Module.\r
+  CPU DXE Module to produce CPU MP Protocol.\r
 \r
-  Copyright (c) 2008 - 2014, Intel Corporation. All rights reserved.<BR>\r
-  This program and the accompanying materials\r
-  are licensed and made available under the terms and conditions of the BSD License\r
-  which accompanies this distribution.  The full text of the license may be found at\r
-  http://opensource.org/licenses/bsd-license.php\r
-\r
-  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+  Copyright (c) 2008 - 2017, Intel Corporation. All rights reserved.<BR>\r
+  SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
 #include "CpuDxe.h"\r
 #include "CpuMp.h"\r
 \r
-UINTN gMaxLogicalProcessorNumber;\r
-UINTN gApStackSize;\r
-UINTN gPollInterval = 100; // 100 microseconds\r
-\r
-MP_SYSTEM_DATA mMpSystemData;\r
-EFI_HANDLE     mMpServiceHandle = NULL;\r
-\r
-VOID *mCommonStack = 0;\r
-VOID *mTopOfApCommonStack = 0;\r
-VOID *mApStackStart = 0;\r
-\r
-volatile BOOLEAN mAPsAlreadyInitFinished = FALSE;\r
-volatile BOOLEAN mStopCheckAllAPsStatus = TRUE;\r
+EFI_HANDLE     mMpServiceHandle       = NULL;\r
+UINTN          mNumberOfProcessors    = 1;\r
 \r
 EFI_MP_SERVICES_PROTOCOL  mMpServicesTemplate = {\r
   GetNumberOfProcessors,\r
@@ -39,322 +22,6 @@ EFI_MP_SERVICES_PROTOCOL  mMpServicesTemplate = {
   WhoAmI\r
 };\r
 \r
-/**\r
-   Get Mp Service Lock.\r
-\r
-  @param   CpuData    the pointer to CPU_DATA_BLOCK of specified processor\r
-\r
-**/\r
-VOID\r
-GetMpSpinLock (\r
-  IN  CPU_DATA_BLOCK  *CpuData\r
-  )\r
-{\r
-  while (!AcquireSpinLockOrFail (&CpuData->CpuDataLock)) {\r
-    CpuPause ();\r
-  }\r
-  CpuData->LockSelf = GetApicId ();\r
-}\r
-\r
-/**\r
-   Release Mp Service Lock.\r
-\r
-  @param   CpuData    the pointer to CPU_DATA_BLOCK of specified processor\r
-\r
-**/\r
-VOID\r
-ReleaseMpSpinLock (\r
-  IN  CPU_DATA_BLOCK  *CpuData\r
-  )\r
-{\r
-  ReleaseSpinLock (&CpuData->CpuDataLock);\r
-}\r
-\r
-/**\r
-  Check whether caller processor is BSP.\r
-\r
-  @retval  TRUE       the caller is BSP\r
-  @retval  FALSE      the caller is AP\r
-\r
-**/\r
-BOOLEAN\r
-IsBSP (\r
-  VOID\r
-  )\r
-{\r
-  UINTN           CpuIndex;\r
-  CPU_DATA_BLOCK  *CpuData;\r
-\r
-  CpuData = NULL;\r
-\r
-  WhoAmI (&mMpServicesTemplate, &CpuIndex);\r
-  CpuData = &mMpSystemData.CpuDatas[CpuIndex];\r
-\r
-  return CpuData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT ? TRUE : FALSE;\r
-}\r
-\r
-/**\r
-  Get the Application Processors state.\r
-\r
-  @param   CpuData    the pointer to CPU_DATA_BLOCK of specified AP\r
-\r
-  @retval  CPU_STATE  the AP status\r
-\r
-**/\r
-CPU_STATE\r
-GetApState (\r
-  IN  CPU_DATA_BLOCK  *CpuData\r
-  )\r
-{\r
-  CPU_STATE State;\r
-\r
-  GetMpSpinLock (CpuData);\r
-  State = CpuData->State;\r
-  ReleaseMpSpinLock (CpuData);\r
-\r
-  return State;\r
-}\r
-\r
-/**\r
-  Set the Application Processors state.\r
-\r
-  @param   CpuData    The pointer to CPU_DATA_BLOCK of specified AP\r
-  @param   State      The AP status\r
-\r
-**/\r
-VOID\r
-SetApState (\r
-  IN  CPU_DATA_BLOCK   *CpuData,\r
-  IN  CPU_STATE        State\r
-  )\r
-{\r
-  GetMpSpinLock (CpuData);\r
-  CpuData->State = State;\r
-  ReleaseMpSpinLock (CpuData);\r
-}\r
-\r
-/**\r
-  Set the Application Processor prepare to run a function specified\r
-  by Params.\r
-\r
-  @param CpuData           the pointer to CPU_DATA_BLOCK of specified AP\r
-  @param Procedure         A pointer to the function to be run on enabled APs of the system\r
-  @param ProcedureArgument Pointer to the optional parameter of the assigned function\r
-\r
-**/\r
-VOID\r
-SetApProcedure (\r
-  IN   CPU_DATA_BLOCK        *CpuData,\r
-  IN   EFI_AP_PROCEDURE      Procedure,\r
-  IN   VOID                  *ProcedureArgument\r
-  )\r
-{\r
-  GetMpSpinLock (CpuData);\r
-  CpuData->Parameter  = ProcedureArgument;\r
-  CpuData->Procedure  = Procedure;\r
-  ReleaseMpSpinLock (CpuData);\r
-}\r
-\r
-/**\r
-  Check the Application Processors Status whether contains the Flags.\r
-\r
-  @param     CpuData  the pointer to CPU_DATA_BLOCK of specified AP\r
-  @param     Flags    the StatusFlag describing in EFI_PROCESSOR_INFORMATION\r
-\r
-  @retval    TRUE     the AP status includes the StatusFlag\r
-  @retval    FALSE    the AP status excludes the StatusFlag\r
-\r
-**/\r
-BOOLEAN\r
-TestCpuStatusFlag (\r
-  IN  CPU_DATA_BLOCK  *CpuData,\r
-  IN  UINT32          Flags\r
-  )\r
-{\r
-  UINT32 Ret;\r
-\r
-  GetMpSpinLock (CpuData);\r
-  Ret = CpuData->Info.StatusFlag & Flags;\r
-  ReleaseMpSpinLock (CpuData);\r
-\r
-  return !!(Ret);\r
-}\r
-\r
-/**\r
-  Bitwise-Or of the Application Processors Status with the Flags.\r
-\r
-  @param     CpuData  the pointer to CPU_DATA_BLOCK of specified AP\r
-  @param     Flags    the StatusFlag describing in EFI_PROCESSOR_INFORMATION\r
-\r
-**/\r
-VOID\r
-CpuStatusFlagOr (\r
-  IN  CPU_DATA_BLOCK  *CpuData,\r
-  IN  UINT32          Flags\r
-  )\r
-{\r
-  GetMpSpinLock (CpuData);\r
-  CpuData->Info.StatusFlag |= Flags;\r
-  ReleaseMpSpinLock (CpuData);\r
-}\r
-\r
-/**\r
-  Bitwise-AndNot of the Application Processors Status with the Flags.\r
-\r
-  @param     CpuData  the pointer to CPU_DATA_BLOCK of specified AP\r
-  @param     Flags    the StatusFlag describing in EFI_PROCESSOR_INFORMATION\r
-\r
-**/\r
-VOID\r
-CpuStatusFlagAndNot (\r
-  IN  CPU_DATA_BLOCK  *CpuData,\r
-  IN  UINT32          Flags\r
-  )\r
-{\r
-  GetMpSpinLock (CpuData);\r
-  CpuData->Info.StatusFlag &= ~Flags;\r
-  ReleaseMpSpinLock (CpuData);\r
-}\r
-\r
-/**\r
-  Searches for the next blocking AP.\r
-\r
-  Search for the next AP that is put in blocking state by single-threaded StartupAllAPs().\r
-\r
-  @param  NextNumber           Pointer to the processor number of the next blocking AP.\r
-\r
-  @retval EFI_SUCCESS          The next blocking AP has been found.\r
-  @retval EFI_NOT_FOUND        No blocking AP exists.\r
-\r
-**/\r
-EFI_STATUS\r
-GetNextBlockedNumber (\r
-  OUT UINTN  *NextNumber\r
-  )\r
-{\r
-  UINTN                 Number;\r
-  CPU_STATE             CpuState;\r
-  CPU_DATA_BLOCK        *CpuData;\r
-\r
-  for (Number = 0; Number < mMpSystemData.NumberOfProcessors; Number++) {\r
-    CpuData = &mMpSystemData.CpuDatas[Number];\r
-    if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT)) {\r
-      //\r
-      // Skip BSP\r
-      //\r
-      continue;\r
-    }\r
-\r
-    CpuState = GetApState (CpuData);\r
-    if (CpuState == CpuStateBlocked) {\r
-      *NextNumber = Number;\r
-      return EFI_SUCCESS;\r
-    }\r
-  }\r
-\r
-  return EFI_NOT_FOUND;\r
-}\r
-\r
-/**\r
-  Check if the APs state are finished, and update them to idle state\r
-  by StartupAllAPs().\r
-\r
-**/\r
-VOID\r
-CheckAndUpdateAllAPsToIdleState (\r
-  VOID\r
-  )\r
-{\r
-  UINTN                 ProcessorNumber;\r
-  UINTN                 NextNumber;\r
-  CPU_DATA_BLOCK        *CpuData;\r
-  EFI_STATUS            Status;\r
-  CPU_STATE             CpuState;\r
-\r
-  for (ProcessorNumber = 0; ProcessorNumber < mMpSystemData.NumberOfProcessors; ProcessorNumber++) {\r
-    CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
-    if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT)) {\r
-      //\r
-      // Skip BSP\r
-      //\r
-      continue;\r
-    }\r
-\r
-    if (!TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-      //\r
-      // Skip Disabled processors\r
-      //\r
-      continue;\r
-    }\r
-\r
-    CpuState = GetApState (CpuData);\r
-    if (CpuState == CpuStateFinished) {\r
-      mMpSystemData.FinishCount++;\r
-      if (mMpSystemData.SingleThread) {\r
-        Status = GetNextBlockedNumber (&NextNumber);\r
-        if (!EFI_ERROR (Status)) {\r
-          SetApState (&mMpSystemData.CpuDatas[NextNumber], CpuStateReady);\r
-          SetApProcedure (&mMpSystemData.CpuDatas[NextNumber],\r
-                          mMpSystemData.Procedure,\r
-                          mMpSystemData.ProcedureArgument);\r
-        }\r
-      }\r
-\r
-      SetApState (CpuData, CpuStateIdle);\r
-    }\r
-  }\r
-}\r
-\r
-/**\r
-  If the timeout expires before all APs returns from Procedure,\r
-  we should forcibly terminate the executing AP and fill FailedList back\r
-  by StartupAllAPs().\r
-\r
-**/\r
-VOID\r
-ResetAllFailedAPs (\r
-  VOID\r
-  )\r
-{\r
-  CPU_DATA_BLOCK        *CpuData;\r
-  UINTN                 Number;\r
-  CPU_STATE             CpuState;\r
-\r
-  if (mMpSystemData.FailedList != NULL) {\r
-     *mMpSystemData.FailedList = AllocatePool ((mMpSystemData.StartCount - mMpSystemData.FinishCount + 1) * sizeof(UINTN));\r
-     ASSERT (*mMpSystemData.FailedList != NULL);\r
-  }\r
-\r
-  for (Number = 0; Number < mMpSystemData.NumberOfProcessors; Number++) {\r
-    CpuData = &mMpSystemData.CpuDatas[Number];\r
-    if (TestCpuStatusFlag (CpuData,  PROCESSOR_AS_BSP_BIT)) {\r
-      //\r
-      // Skip BSP\r
-      //\r
-      continue;\r
-    }\r
-\r
-    if (!TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-      //\r
-      // Skip Disabled processors\r
-      //\r
-      continue;\r
-    }\r
-\r
-    CpuState = GetApState (CpuData);\r
-    if (CpuState != CpuStateIdle) {\r
-      if (mMpSystemData.FailedList != NULL) {\r
-        (*mMpSystemData.FailedList)[mMpSystemData.FailedListIndex++] = Number;\r
-      }\r
-      ResetProcessorToIdleState (CpuData);\r
-    }\r
-  }\r
-\r
-  if (mMpSystemData.FailedList != NULL) {\r
-    (*mMpSystemData.FailedList)[mMpSystemData.FailedListIndex] = END_OF_CPU_LIST;\r
-  }\r
-}\r
-\r
 /**\r
   This service retrieves the number of logical processor in the platform\r
   and the number of those logical processors that are enabled on this boot.\r
@@ -403,13 +70,10 @@ GetNumberOfProcessors (
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  if (!IsBSP ()) {\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  *NumberOfProcessors        = mMpSystemData.NumberOfProcessors;\r
-  *NumberOfEnabledProcessors = mMpSystemData.NumberOfEnabledProcessors;\r
-  return EFI_SUCCESS;\r
+  return MpInitLibGetNumberOfProcessors (\r
+           NumberOfProcessors,\r
+           NumberOfEnabledProcessors\r
+           );\r
 }\r
 \r
 /**\r
@@ -446,20 +110,7 @@ GetProcessorInfo (
   OUT EFI_PROCESSOR_INFORMATION  *ProcessorInfoBuffer\r
   )\r
 {\r
-  if (ProcessorInfoBuffer == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  if (!IsBSP ()) {\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  if (ProcessorNumber >= mMpSystemData.NumberOfProcessors) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
-  CopyMem (ProcessorInfoBuffer, &mMpSystemData.CpuDatas[ProcessorNumber], sizeof (EFI_PROCESSOR_INFORMATION));\r
-  return EFI_SUCCESS;\r
+  return MpInitLibGetProcessorInfo (ProcessorNumber, ProcessorInfoBuffer, NULL);\r
 }\r
 \r
 /**\r
@@ -610,141 +261,14 @@ StartupAllAPs (
   OUT UINTN                     **FailedCpuList         OPTIONAL\r
   )\r
 {\r
-  EFI_STATUS            Status;\r
-  CPU_DATA_BLOCK        *CpuData;\r
-  UINTN                 Number;\r
-  CPU_STATE             APInitialState;\r
-\r
-  CpuData = NULL;\r
-\r
-  if (FailedCpuList != NULL) {\r
-    *FailedCpuList = NULL;\r
-  }\r
-\r
-  if (!IsBSP ()) {\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  if (mMpSystemData.NumberOfProcessors == 1) {\r
-    return EFI_NOT_STARTED;\r
-  }\r
-\r
-  if (Procedure == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  //\r
-  // temporarily stop checkAllAPsStatus for avoid resource dead-lock.\r
-  //\r
-  mStopCheckAllAPsStatus = TRUE;\r
-\r
-  for (Number = 0; Number < mMpSystemData.NumberOfProcessors; Number++) {\r
-    CpuData = &mMpSystemData.CpuDatas[Number];\r
-    if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT)) {\r
-      //\r
-      // Skip BSP\r
-      //\r
-      continue;\r
-    }\r
-\r
-    if (!TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-      //\r
-      // Skip Disabled processors\r
-      //\r
-      continue;\r
-    }\r
-\r
-    if (GetApState (CpuData) != CpuStateIdle) {\r
-      return EFI_NOT_READY;\r
-    }\r
-  }\r
-\r
-  mMpSystemData.Procedure         = Procedure;\r
-  mMpSystemData.ProcedureArgument = ProcedureArgument;\r
-  mMpSystemData.WaitEvent         = WaitEvent;\r
-  mMpSystemData.Timeout           = TimeoutInMicroseconds;\r
-  mMpSystemData.TimeoutActive     = !!(TimeoutInMicroseconds);\r
-  mMpSystemData.FinishCount       = 0;\r
-  mMpSystemData.StartCount        = 0;\r
-  mMpSystemData.SingleThread      = SingleThread;\r
-  mMpSystemData.FailedList        = FailedCpuList;\r
-  mMpSystemData.FailedListIndex   = 0;\r
-  APInitialState                  = CpuStateReady;\r
-\r
-  for (Number = 0; Number < mMpSystemData.NumberOfProcessors; Number++) {\r
-    CpuData = &mMpSystemData.CpuDatas[Number];\r
-    if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT)) {\r
-      //\r
-      // Skip BSP\r
-      //\r
-      continue;\r
-    }\r
-\r
-    if (!TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-      //\r
-      // Skip Disabled processors\r
-      //\r
-      continue;\r
-    }\r
-\r
-    //\r
-    // Get APs prepared, and put failing APs into FailedCpuList\r
-    // if "SingleThread", only 1 AP will put to ready state, other AP will be put to ready\r
-    // state 1 by 1, until the previous 1 finished its task\r
-    // if not "SingleThread", all APs are put to ready state from the beginning\r
-    //\r
-    if (GetApState (CpuData) == CpuStateIdle) {\r
-      mMpSystemData.StartCount++;\r
-\r
-      SetApState (CpuData, APInitialState);\r
-\r
-      if (APInitialState == CpuStateReady) {\r
-        SetApProcedure (CpuData, Procedure, ProcedureArgument);\r
-      }\r
-\r
-      if (SingleThread) {\r
-        APInitialState = CpuStateBlocked;\r
-      }\r
-    }\r
-  }\r
-\r
-  mStopCheckAllAPsStatus = FALSE;\r
-\r
-  if (WaitEvent != NULL) {\r
-    //\r
-    // non blocking\r
-    //\r
-    return EFI_SUCCESS;\r
-  }\r
-\r
-  //\r
-  // Blocking temporarily stop CheckAllAPsStatus()\r
-  //\r
-  mStopCheckAllAPsStatus = TRUE;\r
-\r
-  while (TRUE) {\r
-    CheckAndUpdateAllAPsToIdleState ();\r
-    if (mMpSystemData.FinishCount == mMpSystemData.StartCount) {\r
-      Status = EFI_SUCCESS;\r
-      goto Done;\r
-    }\r
-\r
-    //\r
-    // task timeout\r
-    //\r
-    if (mMpSystemData.TimeoutActive && mMpSystemData.Timeout < 0) {\r
-      ResetAllFailedAPs();\r
-      Status = EFI_TIMEOUT;\r
-      goto Done;\r
-    }\r
-\r
-    gBS->Stall (gPollInterval);\r
-    mMpSystemData.Timeout -= gPollInterval;\r
-  }\r
-\r
-Done:\r
-\r
-  return Status;\r
+  return MpInitLibStartupAllAPs (\r
+           Procedure,\r
+           SingleThread,\r
+           WaitEvent,\r
+           TimeoutInMicroseconds,\r
+           ProcedureArgument,\r
+           FailedCpuList\r
+           );\r
 }\r
 \r
 /**\r
@@ -769,8 +293,8 @@ Done:
 \r
   @param[in]  This                    A pointer to the EFI_MP_SERVICES_PROTOCOL\r
                                       instance.\r
-  @param[in]  Procedure               A pointer to the function to be run on\r
-                                      enabled APs of the system. See type\r
+  @param[in]  Procedure               A pointer to the function to be run on the\r
+                                      designated AP of the system. See type\r
                                       EFI_AP_PROCEDURE.\r
   @param[in]  ProcessorNumber         The handle number of the AP. The range is\r
                                       from 0 to the total number of logical\r
@@ -779,13 +303,13 @@ Done:
                                       EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().\r
   @param[in]  WaitEvent               The event created by the caller with CreateEvent()\r
                                       service.  If it is NULL, then execute in\r
-                                      blocking mode. BSP waits until all APs finish\r
-                                      or TimeoutInMicroseconds expires.  If it's\r
+                                      blocking mode. BSP waits until this AP finish\r
+                                      or TimeoutInMicroSeconds expires.  If it's\r
                                       not NULL, then execute in non-blocking mode.\r
                                       BSP requests the function specified by\r
-                                      Procedure to be started on all the enabled\r
-                                      APs, and go on executing immediately. If\r
-                                      all return from Procedure or TimeoutInMicroseconds\r
+                                      Procedure to be started on this AP,\r
+                                      and go on executing immediately. If this AP\r
+                                      return from Procedure or TimeoutInMicroSeconds\r
                                       expires, this event is signaled. The BSP\r
                                       can use the CheckEvent() or WaitForEvent()\r
                                       services to check the state of event.  Type\r
@@ -793,20 +317,20 @@ Done:
                                       the Unified Extensible Firmware Interface\r
                                       Specification.\r
   @param[in]  TimeoutInMicroseconds   Indicates the time limit in microseconds for\r
-                                      APs to return from Procedure, either for\r
+                                      this AP to finish this Procedure, either for\r
                                       blocking or non-blocking mode. Zero means\r
                                       infinity.  If the timeout expires before\r
-                                      all APs return from Procedure, then Procedure\r
-                                      on the failed APs is terminated. All enabled\r
-                                      APs are available for next function assigned\r
+                                      this AP returns from Procedure, then Procedure\r
+                                      on the AP is terminated. The\r
+                                      AP is available for next function assigned\r
                                       by EFI_MP_SERVICES_PROTOCOL.StartupAllAPs()\r
                                       or EFI_MP_SERVICES_PROTOCOL.StartupThisAP().\r
                                       If the timeout expires in blocking mode,\r
                                       BSP returns EFI_TIMEOUT.  If the timeout\r
                                       expires in non-blocking mode, WaitEvent\r
                                       is signaled with SignalEvent().\r
-  @param[in]  ProcedureArgument       The parameter passed into Procedure for\r
-                                      all APs.\r
+  @param[in]  ProcedureArgument       The parameter passed into Procedure on the\r
+                                      specified AP.\r
   @param[out] Finished                If NULL, this parameter is ignored.  In\r
                                       blocking mode, this parameter is ignored.\r
                                       In non-blocking mode, if AP returns from\r
@@ -845,78 +369,14 @@ StartupThisAP (
   OUT BOOLEAN                   *Finished               OPTIONAL\r
   )\r
 {\r
-  CPU_DATA_BLOCK        *CpuData;\r
-\r
-  CpuData = NULL;\r
-\r
-  if (Finished != NULL) {\r
-    *Finished = FALSE;\r
-  }\r
-\r
-  if (!IsBSP ()) {\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  if (Procedure == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  if (ProcessorNumber >= mMpSystemData.NumberOfProcessors) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
-  //\r
-  // temporarily stop checkAllAPsStatus for avoid resource dead-lock.\r
-  //\r
-  mStopCheckAllAPsStatus = TRUE;\r
-\r
-  CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
-  if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT) ||\r
-      !TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  if (GetApState (CpuData) != CpuStateIdle) {\r
-    return EFI_NOT_READY;\r
-  }\r
-\r
-  SetApState (CpuData, CpuStateReady);\r
-\r
-  SetApProcedure (CpuData, Procedure, ProcedureArgument);\r
-\r
-  CpuData->Timeout = TimeoutInMicroseconds;\r
-  CpuData->WaitEvent = WaitEvent;\r
-  CpuData->TimeoutActive = !!(TimeoutInMicroseconds);\r
-  CpuData->Finished = Finished;\r
-\r
-  mStopCheckAllAPsStatus = FALSE;\r
-\r
-  if (WaitEvent != NULL) {\r
-    //\r
-    // Non Blocking\r
-    //\r
-    return EFI_SUCCESS;\r
-  }\r
-\r
-  //\r
-  // Blocking\r
-  //\r
-  while (TRUE) {\r
-    if (GetApState (CpuData) == CpuStateFinished) {\r
-      SetApState (CpuData, CpuStateIdle);\r
-      break;\r
-    }\r
-\r
-    if (CpuData->TimeoutActive && CpuData->Timeout < 0) {\r
-      ResetProcessorToIdleState (CpuData);\r
-      return EFI_TIMEOUT;\r
-    }\r
-\r
-    gBS->Stall (gPollInterval);\r
-    CpuData->Timeout -= gPollInterval;\r
-  }\r
-\r
-  return EFI_SUCCESS;\r
+  return MpInitLibStartupThisAP (\r
+           Procedure,\r
+           ProcessorNumber,\r
+           WaitEvent,\r
+           TimeoutInMicroseconds,\r
+           ProcedureArgument,\r
+           Finished\r
+           );\r
 }\r
 \r
 /**\r
@@ -946,7 +406,7 @@ StartupThisAP (
   @retval EFI_UNSUPPORTED         Switching the BSP cannot be completed prior to\r
                                   this service returning.\r
   @retval EFI_UNSUPPORTED         Switching the BSP is not supported.\r
-  @retval EFI_SUCCESS             The calling processor is an AP.\r
+  @retval EFI_DEVICE_ERROR        The calling processor is an AP.\r
   @retval EFI_NOT_FOUND           The processor with the handle specified by\r
                                   ProcessorNumber does not exist.\r
   @retval EFI_INVALID_PARAMETER   ProcessorNumber specifies the current BSP or\r
@@ -962,10 +422,7 @@ SwitchBSP (
   IN  BOOLEAN                  EnableOldBSP\r
   )\r
 {\r
-   //\r
-   // Current always return unsupported.\r
-   //\r
-   return EFI_UNSUPPORTED;\r
+  return MpInitLibSwitchBSP (ProcessorNumber, EnableOldBSP);\r
 }\r
 \r
 /**\r
@@ -984,8 +441,8 @@ SwitchBSP (
   from this service, then EFI_UNSUPPORTED must be returned.\r
 \r
   @param[in] This              A pointer to the EFI_MP_SERVICES_PROTOCOL instance.\r
-  @param[in] ProcessorNumber   The handle number of AP that is to become the new\r
-                               BSP. The range is from 0 to the total number of\r
+  @param[in] ProcessorNumber   The handle number of AP.\r
+                               The range is from 0 to the total number of\r
                                logical processors minus 1. The total number of\r
                                logical processors can be retrieved by\r
                                EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().\r
@@ -1018,59 +475,7 @@ EnableDisableAP (
   IN  UINT32                    *HealthFlag OPTIONAL\r
   )\r
 {\r
-  CPU_DATA_BLOCK *CpuData;\r
-  BOOLEAN        TempStopCheckState;\r
-\r
-  CpuData = NULL;\r
-  TempStopCheckState = FALSE;\r
-\r
-  if (!IsBSP ()) {\r
-    return EFI_DEVICE_ERROR;\r
-  }\r
-\r
-  if (ProcessorNumber >= mMpSystemData.NumberOfProcessors) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-\r
-  //\r
-  // temporarily stop checkAllAPsStatus for initialize parameters.\r
-  //\r
-  if (!mStopCheckAllAPsStatus) {\r
-    mStopCheckAllAPsStatus = TRUE;\r
-    TempStopCheckState = TRUE;\r
-  }\r
-\r
-  CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
-  if (TestCpuStatusFlag (CpuData, PROCESSOR_AS_BSP_BIT)) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  if (GetApState (CpuData) != CpuStateIdle) {\r
-    return EFI_UNSUPPORTED;\r
-  }\r
-\r
-  if (EnableAP) {\r
-    if (!(TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT))) {\r
-      mMpSystemData.NumberOfEnabledProcessors++;\r
-    }\r
-    CpuStatusFlagOr (CpuData, PROCESSOR_ENABLED_BIT);\r
-  } else {\r
-    if (TestCpuStatusFlag (CpuData, PROCESSOR_ENABLED_BIT)) {\r
-      mMpSystemData.NumberOfEnabledProcessors--;\r
-    }\r
-    CpuStatusFlagAndNot (CpuData, PROCESSOR_ENABLED_BIT);\r
-  }\r
-\r
-  if (HealthFlag != NULL) {\r
-    CpuStatusFlagAndNot (CpuData, (UINT32)~PROCESSOR_HEALTH_STATUS_BIT);\r
-    CpuStatusFlagOr (CpuData, (*HealthFlag & PROCESSOR_HEALTH_STATUS_BIT));\r
-  }\r
-\r
-  if (TempStopCheckState) {\r
-    mStopCheckAllAPsStatus = FALSE;\r
-  }\r
-\r
-  return EFI_SUCCESS;\r
+  return MpInitLibEnableDisableAP (ProcessorNumber, EnableAP, HealthFlag);\r
 }\r
 \r
 /**\r
@@ -1086,8 +491,8 @@ EnableDisableAP (
   ProcessorNumber, and EFI_SUCCESS is returned.\r
 \r
   @param[in]  This             A pointer to the EFI_MP_SERVICES_PROTOCOL instance.\r
-  @param[out] ProcessorNumber  The handle number of AP that is to become the new\r
-                               BSP. The range is from 0 to the total number of\r
+  @param[out] ProcessorNumber  Pointer to the handle number of AP.\r
+                               The range is from 0 to the total number of\r
                                logical processors minus 1. The total number of\r
                                logical processors can be retrieved by\r
                                EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().\r
@@ -1104,364 +509,301 @@ WhoAmI (
   OUT UINTN                    *ProcessorNumber\r
   )\r
 {\r
-  UINTN   Index;\r
-  UINT32  ProcessorId;\r
-\r
-  if (ProcessorNumber == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  ProcessorId = GetApicId ();\r
-  for (Index = 0; Index < mMpSystemData.NumberOfProcessors; Index++) {\r
-    if (mMpSystemData.CpuDatas[Index].Info.ProcessorId == ProcessorId) {\r
-      break;\r
-    }\r
-  }\r
-\r
-  *ProcessorNumber = Index;\r
-  return EFI_SUCCESS;\r
-}\r
-\r
-/**\r
-  Terminate AP's task and set it to idle state.\r
-\r
-  This function terminates AP's task due to timeout by sending INIT-SIPI,\r
-  and sends it to idle state.\r
-\r
-  @param CpuData           the pointer to CPU_DATA_BLOCK of specified AP\r
-\r
-**/\r
-VOID\r
-ResetProcessorToIdleState (\r
-  IN CPU_DATA_BLOCK  *CpuData\r
-  )\r
-{\r
-  ResetApStackless ((UINT32)CpuData->Info.ProcessorId);\r
+  return MpInitLibWhoAmI (ProcessorNumber);;\r
 }\r
 \r
 /**\r
-  Application Processors do loop routine\r
-  after switch to its own stack.\r
+  Collects BIST data from HOB.\r
 \r
-  @param  Context1    A pointer to the context to pass into the function.\r
-  @param  Context2    A pointer to the context to pass into the function.\r
+  This function collects BIST data from HOB built from Sec Platform Information\r
+  PPI or SEC Platform Information2 PPI.\r
 \r
 **/\r
 VOID\r
-ProcessorToIdleState (\r
-  IN      VOID                      *Context1,  OPTIONAL\r
-  IN      VOID                      *Context2   OPTIONAL\r
+CollectBistDataFromHob (\r
+  VOID\r
   )\r
 {\r
-  UINTN                 ProcessorNumber;\r
-  CPU_DATA_BLOCK        *CpuData;\r
-  EFI_AP_PROCEDURE      Procedure;\r
-  VOID                  *ProcedureArgument;\r
-\r
-  AsmApDoneWithCommonStack ();\r
-\r
-  while (!mAPsAlreadyInitFinished) {\r
-    CpuPause ();\r
-  }\r
-\r
-  WhoAmI (&mMpServicesTemplate, &ProcessorNumber);\r
-  CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
+  EFI_HOB_GUID_TYPE                     *GuidHob;\r
+  EFI_SEC_PLATFORM_INFORMATION_RECORD2  *SecPlatformInformation2;\r
+  EFI_SEC_PLATFORM_INFORMATION_RECORD   *SecPlatformInformation;\r
+  UINTN                                 NumberOfData;\r
+  EFI_SEC_PLATFORM_INFORMATION_CPU      *CpuInstance;\r
+  EFI_SEC_PLATFORM_INFORMATION_CPU      BspCpuInstance;\r
+  UINTN                                 ProcessorNumber;\r
+  EFI_PROCESSOR_INFORMATION             ProcessorInfo;\r
+  EFI_HEALTH_FLAGS                      BistData;\r
+  UINTN                                 CpuInstanceNumber;\r
+\r
+  SecPlatformInformation2 = NULL;\r
+  SecPlatformInformation  = NULL;\r
 \r
   //\r
-  // Avoid forcibly reset AP caused the AP got lock not release.\r
+  // Get gEfiSecPlatformInformation2PpiGuid Guided HOB firstly\r
   //\r
-  if (CpuData->LockSelf == (INTN) GetApicId ()) {\r
-    ReleaseSpinLock (&CpuData->CpuDataLock);\r
+  GuidHob = GetFirstGuidHob (&gEfiSecPlatformInformation2PpiGuid);\r
+  if (GuidHob != NULL) {\r
+    //\r
+    // Sec Platform Information2 PPI includes BSP/APs' BIST information\r
+    //\r
+    SecPlatformInformation2 = GET_GUID_HOB_DATA (GuidHob);\r
+    NumberOfData = SecPlatformInformation2->NumberOfCpus;\r
+    CpuInstance  = SecPlatformInformation2->CpuInstance;\r
+  } else {\r
+    //\r
+    // Otherwise, get gEfiSecPlatformInformationPpiGuid Guided HOB\r
+    //\r
+    GuidHob = GetFirstGuidHob (&gEfiSecPlatformInformationPpiGuid);\r
+    if (GuidHob != NULL) {\r
+      SecPlatformInformation = GET_GUID_HOB_DATA (GuidHob);\r
+      NumberOfData = 1;\r
+      //\r
+      // SEC Platform Information only includes BSP's BIST information\r
+      // does not have BSP's APIC ID\r
+      //\r
+      BspCpuInstance.CpuLocation = GetApicId ();\r
+      BspCpuInstance.InfoRecord.IA32HealthFlags.Uint32  = SecPlatformInformation->IA32HealthFlags.Uint32;\r
+      CpuInstance = &BspCpuInstance;\r
+    } else {\r
+      DEBUG ((DEBUG_INFO, "Does not find any HOB stored CPU BIST information!\n"));\r
+      //\r
+      // Does not find any HOB stored BIST information\r
+      //\r
+      return;\r
+    }\r
   }\r
 \r
-  //\r
-  // Avoid forcibly reset AP caused the AP State is not updated.\r
-  //\r
-  GetMpSpinLock (CpuData);\r
-  CpuData->State = CpuStateIdle;\r
-  CpuData->Procedure = NULL;\r
-  ReleaseMpSpinLock (CpuData);\r
-\r
-  while (TRUE) {\r
-    GetMpSpinLock (CpuData);\r
-    ProcedureArgument = CpuData->Parameter;\r
-    Procedure = CpuData->Procedure;\r
-    ReleaseMpSpinLock (CpuData);\r
-\r
-    if (Procedure != NULL) {\r
-      Procedure (ProcedureArgument);\r
-\r
-      GetMpSpinLock (CpuData);\r
-      CpuData->Procedure = NULL;\r
-      CpuData->State = CpuStateFinished;\r
-      ReleaseMpSpinLock (CpuData);\r
+  for (ProcessorNumber = 0; ProcessorNumber < mNumberOfProcessors; ProcessorNumber++) {\r
+    MpInitLibGetProcessorInfo (ProcessorNumber, &ProcessorInfo, &BistData);\r
+    for (CpuInstanceNumber = 0; CpuInstanceNumber < NumberOfData; CpuInstanceNumber++) {\r
+      if (ProcessorInfo.ProcessorId == CpuInstance[CpuInstanceNumber].CpuLocation) {\r
+        //\r
+        // Update CPU health status for MP Services Protocol according to BIST data.\r
+        //\r
+        BistData = CpuInstance[CpuInstanceNumber].InfoRecord.IA32HealthFlags;\r
+      }\r
+    }\r
+    if (BistData.Uint32 != 0) {\r
+      //\r
+      // Report Status Code that self test is failed\r
+      //\r
+      REPORT_STATUS_CODE (\r
+        EFI_ERROR_CODE | EFI_ERROR_MAJOR,\r
+        (EFI_COMPUTING_UNIT_HOST_PROCESSOR | EFI_CU_HP_EC_SELF_TEST)\r
+        );\r
     }\r
-\r
-    CpuPause ();\r
   }\r
-\r
-  CpuSleep ();\r
-  CpuDeadLoop ();\r
 }\r
 \r
 /**\r
-  Checks AP' status periodically.\r
+  Get GDT register value.\r
 \r
-  This function is triggerred by timer perodically to check the\r
-  state of AP forStartupThisAP() executed in non-blocking mode.\r
+  This function is mainly for AP purpose because AP may have different GDT\r
+  table than BSP.\r
 \r
-  @param  Event    Event triggered.\r
-  @param  Context  Parameter passed with the event.\r
+  @param[in,out] Buffer  The pointer to private data buffer.\r
 \r
 **/\r
 VOID\r
 EFIAPI\r
-CheckThisAPStatus (\r
-  IN  EFI_EVENT        Event,\r
-  IN  VOID             *Context\r
+GetGdtr (\r
+  IN OUT VOID *Buffer\r
   )\r
 {\r
-  CPU_DATA_BLOCK  *CpuData;\r
-  CPU_STATE       CpuState;\r
-\r
-  CpuData = (CPU_DATA_BLOCK *) Context;\r
-  if (CpuData->TimeoutActive) {\r
-    CpuData->Timeout -= gPollInterval;\r
-  }\r
-\r
-  CpuState = GetApState (CpuData);\r
-\r
-  if (CpuState == CpuStateFinished) {\r
-    if (CpuData->Finished) {\r
-      *CpuData->Finished = TRUE;\r
-    }\r
-    SetApState (CpuData, CpuStateIdle);\r
-    goto out;\r
-  }\r
-\r
-  if (CpuData->TimeoutActive && CpuData->Timeout < 0) {\r
-    if (CpuState != CpuStateIdle &&\r
-        CpuData->Finished) {\r
-      *CpuData->Finished = FALSE;\r
-    }\r
-    ResetProcessorToIdleState (CpuData);\r
-    goto out;\r
-  }\r
-\r
-  return;\r
-\r
-out:\r
-  CpuData->TimeoutActive = FALSE;\r
-  gBS->SignalEvent (CpuData->WaitEvent);\r
-  CpuData->WaitEvent = NULL;\r
+  AsmReadGdtr ((IA32_DESCRIPTOR *)Buffer);\r
 }\r
 \r
 /**\r
-  Checks APs' status periodically.\r
+  Initializes CPU exceptions handlers for the sake of stack switch requirement.\r
 \r
-  This function is triggerred by timer perodically to check the\r
-  state of APs for StartupAllAPs() executed in non-blocking mode.\r
+  This function is a wrapper of InitializeCpuExceptionHandlersEx. It's mainly\r
+  for the sake of AP's init because of EFI_AP_PROCEDURE API requirement.\r
 \r
-  @param  Event    Event triggered.\r
-  @param  Context  Parameter passed with the event.\r
+  @param[in,out] Buffer  The pointer to private data buffer.\r
 \r
 **/\r
 VOID\r
 EFIAPI\r
-CheckAllAPsStatus (\r
-  IN  EFI_EVENT        Event,\r
-  IN  VOID             *Context\r
+InitializeExceptionStackSwitchHandlers (\r
+  IN OUT VOID *Buffer\r
   )\r
 {\r
-  CPU_DATA_BLOCK *CpuData;\r
-  UINTN          Number;\r
-  EFI_STATUS     Status;\r
-\r
-  if (mMpSystemData.TimeoutActive) {\r
-    mMpSystemData.Timeout -= gPollInterval;\r
-  }\r
-\r
-  if (mStopCheckAllAPsStatus) {\r
-    return;\r
-  }\r
+  CPU_EXCEPTION_INIT_DATA           *EssData;\r
+  IA32_DESCRIPTOR                   Idtr;\r
+  EFI_STATUS                        Status;\r
 \r
+  EssData = Buffer;\r
   //\r
-  // avoid next timer enter.\r
+  // We don't plan to replace IDT table with a new one, but we should not assume\r
+  // the AP's IDT is the same as BSP's IDT either.\r
   //\r
-  Status = gBS->SetTimer (\r
-                  mMpSystemData.CheckAllAPsEvent,\r
-                  TimerCancel,\r
-                  0\r
-                  );\r
-  ASSERT_EFI_ERROR (Status);\r
-\r
-  if (mMpSystemData.WaitEvent != NULL) {\r
-    CheckAndUpdateAllAPsToIdleState ();\r
-    //\r
-    // task timeout\r
-    //\r
-    if (mMpSystemData.TimeoutActive && mMpSystemData.Timeout < 0) {\r
-      ResetAllFailedAPs();\r
-      //\r
-      // force exit\r
-      //\r
-      mMpSystemData.FinishCount = mMpSystemData.StartCount;\r
-    }\r
-\r
-    if (mMpSystemData.FinishCount != mMpSystemData.StartCount) {\r
-      goto EXIT;\r
-    }\r
-\r
-    mMpSystemData.TimeoutActive = FALSE;\r
-    gBS->SignalEvent (mMpSystemData.WaitEvent);\r
-    mMpSystemData.WaitEvent = NULL;\r
-    mStopCheckAllAPsStatus = TRUE;\r
-\r
-    goto EXIT;\r
-  }\r
-\r
-  //\r
-  // check each AP status for StartupThisAP\r
-  //\r
-  for (Number = 0; Number < mMpSystemData.NumberOfProcessors; Number++) {\r
-    CpuData = &mMpSystemData.CpuDatas[Number];\r
-    if (CpuData->WaitEvent) {\r
-      CheckThisAPStatus (NULL, (VOID *)CpuData);\r
-    }\r
-  }\r
-\r
-EXIT:\r
-  Status = gBS->SetTimer (\r
-                  mMpSystemData.CheckAllAPsEvent,\r
-                  TimerPeriodic,\r
-                  EFI_TIMER_PERIOD_MICROSECONDS (100)\r
-                  );\r
+  AsmReadIdtr (&Idtr);\r
+  EssData->Ia32.IdtTable = (VOID *)Idtr.Base;\r
+  EssData->Ia32.IdtTableSize = Idtr.Limit + 1;\r
+  Status = InitializeCpuExceptionHandlersEx (NULL, EssData);\r
   ASSERT_EFI_ERROR (Status);\r
 }\r
 \r
 /**\r
-  Application Processor C code entry point.\r
+  Initializes MP exceptions handlers for the sake of stack switch requirement.\r
+\r
+  This function will allocate required resources required to setup stack switch\r
+  and pass them through CPU_EXCEPTION_INIT_DATA to each logic processor.\r
 \r
 **/\r
 VOID\r
-EFIAPI\r
-ApEntryPointInC (\r
+InitializeMpExceptionStackSwitchHandlers (\r
   VOID\r
   )\r
 {\r
-  VOID*           TopOfApStack;\r
-  UINTN           ProcessorNumber;\r
+  UINTN                           Index;\r
+  UINTN                           Bsp;\r
+  UINTN                           ExceptionNumber;\r
+  UINTN                           OldGdtSize;\r
+  UINTN                           NewGdtSize;\r
+  UINTN                           NewStackSize;\r
+  IA32_DESCRIPTOR                 Gdtr;\r
+  CPU_EXCEPTION_INIT_DATA         EssData;\r
+  UINT8                           *GdtBuffer;\r
+  UINT8                           *StackTop;\r
+\r
+  ExceptionNumber = FixedPcdGetSize (PcdCpuStackSwitchExceptionList);\r
+  NewStackSize = FixedPcdGet32 (PcdCpuKnownGoodStackSize) * ExceptionNumber;\r
+\r
+  StackTop = AllocateRuntimeZeroPool (NewStackSize * mNumberOfProcessors);\r
+  ASSERT (StackTop != NULL);\r
+  StackTop += NewStackSize  * mNumberOfProcessors;\r
 \r
-  if (!mAPsAlreadyInitFinished) {\r
-    FillInProcessorInformation (FALSE, mMpSystemData.NumberOfProcessors);\r
-    TopOfApStack  = (UINT8*)mApStackStart + gApStackSize;\r
-    mApStackStart = TopOfApStack;\r
+  //\r
+  // The default exception handlers must have been initialized. Let's just skip\r
+  // it in this method.\r
+  //\r
+  EssData.Ia32.Revision = CPU_EXCEPTION_INIT_DATA_REV;\r
+  EssData.Ia32.InitDefaultHandlers = FALSE;\r
+\r
+  EssData.Ia32.StackSwitchExceptions = FixedPcdGetPtr(PcdCpuStackSwitchExceptionList);\r
+  EssData.Ia32.StackSwitchExceptionNumber = ExceptionNumber;\r
+  EssData.Ia32.KnownGoodStackSize = FixedPcdGet32(PcdCpuKnownGoodStackSize);\r
 \r
+  //\r
+  // Initialize Gdtr to suppress incorrect compiler/analyzer warnings.\r
+  //\r
+  Gdtr.Base = 0;\r
+  Gdtr.Limit = 0;\r
+  MpInitLibWhoAmI (&Bsp);\r
+  for (Index = 0; Index < mNumberOfProcessors; ++Index) {\r
     //\r
-    // Store the Stack address, when reset the AP, We can found the original address.\r
+    // To support stack switch, we need to re-construct GDT but not IDT.\r
     //\r
-    mMpSystemData.CpuDatas[mMpSystemData.NumberOfProcessors].TopOfStack = TopOfApStack;\r
-    mMpSystemData.NumberOfProcessors++;\r
-    mMpSystemData.NumberOfEnabledProcessors++;\r
-  } else {\r
-    WhoAmI (&mMpServicesTemplate, &ProcessorNumber);\r
+    if (Index == Bsp) {\r
+      GetGdtr (&Gdtr);\r
+    } else {\r
+      //\r
+      // AP might have different size of GDT from BSP.\r
+      //\r
+      MpInitLibStartupThisAP (GetGdtr, Index, NULL, 0, (VOID *)&Gdtr, NULL);\r
+    }\r
+\r
     //\r
-    // Get the original stack address.\r
+    // X64 needs only one TSS of current task working for all exceptions\r
+    // because of its IST feature. IA32 needs one TSS for each exception\r
+    // in addition to current task. Since AP is not supposed to allocate\r
+    // memory, we have to do it in BSP. To simplify the code, we allocate\r
+    // memory for IA32 case to cover both IA32 and X64 exception stack\r
+    // switch.\r
     //\r
-    TopOfApStack = mMpSystemData.CpuDatas[ProcessorNumber].TopOfStack;\r
-  }\r
-\r
-  SwitchStack (\r
-    (SWITCH_STACK_ENTRY_POINT)(UINTN)ProcessorToIdleState,\r
-    NULL,\r
-    NULL,\r
-    TopOfApStack);\r
-}\r
-\r
-/**\r
-  This function is called by all processors (both BSP and AP) once and collects MP related data.\r
-\r
-  @param Bsp             TRUE if the CPU is BSP\r
-  @param ProcessorNumber The specific processor number\r
+    // Layout of memory to allocate for each processor:\r
+    //    --------------------------------\r
+    //    |            Alignment         |  (just in case)\r
+    //    --------------------------------\r
+    //    |                              |\r
+    //    |        Original GDT          |\r
+    //    |                              |\r
+    //    --------------------------------\r
+    //    |    Current task descriptor   |\r
+    //    --------------------------------\r
+    //    |                              |\r
+    //    |  Exception task descriptors  |  X ExceptionNumber\r
+    //    |                              |\r
+    //    --------------------------------\r
+    //    |  Current task-state segment  |\r
+    //    --------------------------------\r
+    //    |                              |\r
+    //    | Exception task-state segment |  X ExceptionNumber\r
+    //    |                              |\r
+    //    --------------------------------\r
+    //\r
+    OldGdtSize = Gdtr.Limit + 1;\r
+    EssData.Ia32.ExceptionTssDescSize = sizeof (IA32_TSS_DESCRIPTOR) *\r
+                                        (ExceptionNumber + 1);\r
+    EssData.Ia32.ExceptionTssSize = sizeof (IA32_TASK_STATE_SEGMENT) *\r
+                                    (ExceptionNumber + 1);\r
+    NewGdtSize = sizeof (IA32_TSS_DESCRIPTOR) +\r
+                 OldGdtSize +\r
+                 EssData.Ia32.ExceptionTssDescSize +\r
+                 EssData.Ia32.ExceptionTssSize;\r
+\r
+    GdtBuffer = AllocateRuntimeZeroPool (NewGdtSize);\r
+    ASSERT (GdtBuffer != NULL);\r
 \r
-  @retval EFI_SUCCESS    Data for the processor collected and filled in\r
+    //\r
+    // Make sure GDT table alignment\r
+    //\r
+    EssData.Ia32.GdtTable = ALIGN_POINTER(GdtBuffer, sizeof (IA32_TSS_DESCRIPTOR));\r
+    NewGdtSize -= ((UINT8 *)EssData.Ia32.GdtTable - GdtBuffer);\r
+    EssData.Ia32.GdtTableSize = NewGdtSize;\r
+\r
+    EssData.Ia32.ExceptionTssDesc = ((UINT8 *)EssData.Ia32.GdtTable + OldGdtSize);\r
+    EssData.Ia32.ExceptionTss = ((UINT8 *)EssData.Ia32.GdtTable + OldGdtSize +\r
+                                 EssData.Ia32.ExceptionTssDescSize);\r
+\r
+    EssData.Ia32.KnownGoodStackTop = (UINTN)StackTop;\r
+    DEBUG ((DEBUG_INFO,\r
+            "Exception stack top[cpu%lu]: 0x%lX\n",\r
+            (UINT64)(UINTN)Index,\r
+            (UINT64)(UINTN)StackTop));\r
+\r
+    if (Index == Bsp) {\r
+      InitializeExceptionStackSwitchHandlers (&EssData);\r
+    } else {\r
+      MpInitLibStartupThisAP (\r
+        InitializeExceptionStackSwitchHandlers,\r
+        Index,\r
+        NULL,\r
+        0,\r
+        (VOID *)&EssData,\r
+        NULL\r
+        );\r
+    }\r
 \r
-**/\r
-EFI_STATUS\r
-FillInProcessorInformation (\r
-  IN     BOOLEAN              Bsp,\r
-  IN     UINTN                ProcessorNumber\r
-  )\r
-{\r
-  CPU_DATA_BLOCK  *CpuData;\r
-  UINT32          ProcessorId;\r
-\r
-  CpuData = &mMpSystemData.CpuDatas[ProcessorNumber];\r
-  ProcessorId  = GetApicId ();\r
-  CpuData->Info.ProcessorId  = ProcessorId;\r
-  CpuData->Info.StatusFlag   = PROCESSOR_ENABLED_BIT | PROCESSOR_HEALTH_STATUS_BIT;\r
-  if (Bsp) {\r
-    CpuData->Info.StatusFlag |= PROCESSOR_AS_BSP_BIT;\r
+    StackTop  -= NewStackSize;\r
   }\r
-  CpuData->Info.Location.Package = ProcessorId;\r
-  CpuData->Info.Location.Core    = 0;\r
-  CpuData->Info.Location.Thread  = 0;\r
-  CpuData->State = Bsp ? CpuStateBuzy : CpuStateIdle;\r
-\r
-  CpuData->Procedure        = NULL;\r
-  CpuData->Parameter        = NULL;\r
-  InitializeSpinLock (&CpuData->CpuDataLock);\r
-  CpuData->LockSelf         = -1;\r
-\r
-  return EFI_SUCCESS;\r
 }\r
 \r
 /**\r
-  Prepare the System Data.\r
-\r
-  @retval EFI_SUCCESS     the System Data finished initilization.\r
-\r
+  Initializes MP exceptions handlers for special features, such as Heap Guard\r
+  and Stack Guard.\r
 **/\r
-EFI_STATUS\r
-InitMpSystemData (\r
+VOID\r
+InitializeMpExceptionHandlers (\r
   VOID\r
   )\r
 {\r
-  EFI_STATUS     Status;\r
-\r
-  ZeroMem (&mMpSystemData, sizeof (MP_SYSTEM_DATA));\r
-\r
-  mMpSystemData.NumberOfProcessors = 1;\r
-  mMpSystemData.NumberOfEnabledProcessors = 1;\r
-\r
-  mMpSystemData.CpuDatas = AllocateZeroPool (sizeof (CPU_DATA_BLOCK) * gMaxLogicalProcessorNumber);\r
-  ASSERT(mMpSystemData.CpuDatas != NULL);\r
-\r
-  Status = gBS->CreateEvent (\r
-                  EVT_TIMER | EVT_NOTIFY_SIGNAL,\r
-                  TPL_CALLBACK,\r
-                  CheckAllAPsStatus,\r
-                  NULL,\r
-                  &mMpSystemData.CheckAllAPsEvent\r
-                  );\r
-  ASSERT_EFI_ERROR (Status);\r
-\r
   //\r
-  // Set timer to check all APs status.\r
+  // Enable non-stop mode for #PF triggered by Heap Guard or NULL Pointer\r
+  // Detection.\r
   //\r
-  Status = gBS->SetTimer (\r
-                  mMpSystemData.CheckAllAPsEvent,\r
-                  TimerPeriodic,\r
-                  EFI_TIMER_PERIOD_MICROSECONDS (100)\r
-                  );\r
-  ASSERT_EFI_ERROR (Status);\r
+  if (HEAP_GUARD_NONSTOP_MODE || NULL_DETECTION_NONSTOP_MODE) {\r
+    RegisterCpuInterruptHandler (EXCEPT_IA32_DEBUG, DebugExceptionHandler);\r
+    RegisterCpuInterruptHandler (EXCEPT_IA32_PAGE_FAULT, PageFaultExceptionHandler);\r
+  }\r
 \r
   //\r
-  // BSP\r
+  // Setup stack switch for Stack Guard feature.\r
   //\r
-  FillInProcessorInformation (TRUE, 0);\r
-\r
-  return EFI_SUCCESS;\r
+  if (PcdGetBool (PcdCpuStackGuard)) {\r
+    InitializeMpExceptionStackSwitchHandlers ();\r
+  }\r
 }\r
 \r
 /**\r
@@ -1473,51 +815,29 @@ InitializeMpSupport (
   VOID\r
   )\r
 {\r
-  EFI_STATUS Status;\r
-\r
-  gMaxLogicalProcessorNumber = (UINTN) PcdGet32 (PcdCpuMaxLogicalProcessorNumber);\r
-  if (gMaxLogicalProcessorNumber < 1) {\r
-    DEBUG ((DEBUG_ERROR, "Setting PcdCpuMaxLogicalProcessorNumber should be more than zero.\n"));\r
-    return;\r
-  }\r
-\r
-  if (gMaxLogicalProcessorNumber == 1) {\r
-    return;\r
-  }\r
-\r
-  gApStackSize = (UINTN) PcdGet32 (PcdCpuApStackSize);\r
-  ASSERT ((gApStackSize & (SIZE_4KB - 1)) == 0);\r
-\r
-  mApStackStart = AllocatePages (EFI_SIZE_TO_PAGES (gMaxLogicalProcessorNumber * gApStackSize));\r
-  ASSERT (mApStackStart != NULL);\r
+  EFI_STATUS     Status;\r
+  UINTN          NumberOfProcessors;\r
+  UINTN          NumberOfEnabledProcessors;\r
 \r
   //\r
-  // the first buffer of stack size used for common stack, when the amount of AP\r
-  // more than 1, we should never free the common stack which maybe used for AP reset.\r
+  // Wakeup APs to do initialization\r
   //\r
-  mCommonStack = mApStackStart;\r
-  mTopOfApCommonStack = (UINT8*) mApStackStart + gApStackSize;\r
-  mApStackStart = mTopOfApCommonStack;\r
-\r
-  InitMpSystemData ();\r
-\r
-  PrepareAPStartupCode ();\r
-\r
-  StartApsStackless ();\r
+  Status = MpInitLibInitialize ();\r
+  ASSERT_EFI_ERROR (Status);\r
 \r
-  DEBUG ((DEBUG_INFO, "Detect CPU count: %d\n", mMpSystemData.NumberOfProcessors));\r
-  if (mMpSystemData.NumberOfProcessors == 1) {\r
-    FreeApStartupCode ();\r
-    FreePages (mCommonStack, EFI_SIZE_TO_PAGES (gMaxLogicalProcessorNumber * gApStackSize));\r
-    return;\r
-  }\r
+  MpInitLibGetNumberOfProcessors (&NumberOfProcessors, &NumberOfEnabledProcessors);\r
+  mNumberOfProcessors = NumberOfProcessors;\r
+  DEBUG ((DEBUG_INFO, "Detect CPU count: %d\n", mNumberOfProcessors));\r
 \r
-  mMpSystemData.CpuDatas = ReallocatePool (\r
-                             sizeof (CPU_DATA_BLOCK) * gMaxLogicalProcessorNumber,\r
-                             sizeof (CPU_DATA_BLOCK) * mMpSystemData.NumberOfProcessors,\r
-                             mMpSystemData.CpuDatas);\r
+  //\r
+  // Initialize special exception handlers for each logic processor.\r
+  //\r
+  InitializeMpExceptionHandlers ();\r
 \r
-  mAPsAlreadyInitFinished = TRUE;\r
+  //\r
+  // Update CPU healthy information from Guided HOB\r
+  //\r
+  CollectBistDataFromHob ();\r
 \r
   Status = gBS->InstallMultipleProtocolInterfaces (\r
                   &mMpServiceHandle,\r
@@ -1525,10 +845,5 @@ InitializeMpSupport (
                   NULL\r
                   );\r
   ASSERT_EFI_ERROR (Status);\r
-\r
-  if (mMpSystemData.NumberOfProcessors < gMaxLogicalProcessorNumber) {\r
-    FreePages (mApStackStart, EFI_SIZE_TO_PAGES (\r
-                                (gMaxLogicalProcessorNumber - mMpSystemData.NumberOfProcessors) *\r
-                                gApStackSize));\r
-  }\r
 }\r
+\r