UefiCpuPkg/MpInitLib: Allocate and initialize memory of MP Data buffer

[mirror_edk2.git] / UefiCpuPkg / Library / MpInitLib / MpLib.c

/** @file\r
  CPU MP Initialize Library common functions.\r
\r
  Copyright (c) 2016, 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
\r
**/\r
\r
#include "MpLib.h"\r
\r
\r
/**\r
  Detect whether Mwait-monitor feature is supported.\r
\r
  @retval TRUE    Mwait-monitor feature is supported.\r
  @retval FALSE   Mwait-monitor feature is not supported.\r
**/\r
BOOLEAN\r
IsMwaitSupport (\r
  VOID\r
  )\r
{\r
  CPUID_VERSION_INFO_ECX        VersionInfoEcx;\r
\r
  AsmCpuid (CPUID_VERSION_INFO, NULL, NULL, &VersionInfoEcx.Uint32, NULL);\r
  return (VersionInfoEcx.Bits.MONITOR == 1) ? TRUE : FALSE;\r
}\r
\r
/**\r
  Get AP loop mode.\r
\r
  @param[out] MonitorFilterSize  Returns the largest monitor-line size in bytes.\r
\r
  @return The AP loop mode.\r
**/\r
UINT8\r
GetApLoopMode (\r
  OUT UINT32     *MonitorFilterSize\r
  )\r
{\r
  UINT8                         ApLoopMode;\r
  CPUID_MONITOR_MWAIT_EBX       MonitorMwaitEbx;\r
\r
  ASSERT (MonitorFilterSize != NULL);\r
\r
  ApLoopMode = PcdGet8 (PcdCpuApLoopMode);\r
  ASSERT (ApLoopMode >= ApInHltLoop && ApLoopMode <= ApInRunLoop);\r
  if (ApLoopMode == ApInMwaitLoop) {\r
    if (!IsMwaitSupport ()) {\r
      //\r
      // If processor does not support MONITOR/MWAIT feature,\r
      // force AP in Hlt-loop mode\r
      //\r
      ApLoopMode = ApInHltLoop;\r
    }\r
  }\r
\r
  if (ApLoopMode != ApInMwaitLoop) {\r
    *MonitorFilterSize = sizeof (UINT32);\r
  } else {\r
    //\r
    // CPUID.[EAX=05H]:EBX.BIT0-15: Largest monitor-line size in bytes\r
    // CPUID.[EAX=05H].EDX: C-states supported using MWAIT\r
    //\r
    AsmCpuid (CPUID_MONITOR_MWAIT, NULL, &MonitorMwaitEbx.Uint32, NULL, NULL);\r
    *MonitorFilterSize = MonitorMwaitEbx.Bits.LargestMonitorLineSize;\r
  }\r
\r
  return ApLoopMode;\r
}\r
/**\r
  MP Initialize Library initialization.\r
\r
  This service will allocate AP reset vector and wakeup all APs to do APs\r
  initialization.\r
\r
  This service must be invoked before all other MP Initialize Library\r
  service are invoked.\r
\r
  @retval  EFI_SUCCESS           MP initialization succeeds.\r
  @retval  Others                MP initialization fails.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
MpInitLibInitialize (\r
  VOID\r
  )\r
{\r
  UINT32                   MaxLogicalProcessorNumber;\r
  UINT32                   ApStackSize;\r
  MP_ASSEMBLY_ADDRESS_MAP  AddressMap;\r
  UINTN                    BufferSize;\r
  UINT32                   MonitorFilterSize;\r
  VOID                     *MpBuffer;\r
  UINTN                    Buffer;\r
  CPU_MP_DATA              *CpuMpData;\r
  UINT8                    ApLoopMode;\r
  UINT8                    *MonitorBuffer;\r
  UINTN                    ApResetVectorSize;\r
  UINTN                    BackupBufferAddr;\r
  MaxLogicalProcessorNumber = PcdGet32(PcdCpuMaxLogicalProcessorNumber);\r
\r
  AsmGetAddressMap (&AddressMap);\r
  ApResetVectorSize = AddressMap.RendezvousFunnelSize + sizeof (MP_CPU_EXCHANGE_INFO);\r
  ApStackSize = PcdGet32(PcdCpuApStackSize);\r
  ApLoopMode  = GetApLoopMode (&MonitorFilterSize);\r
\r
  BufferSize  = ApStackSize * MaxLogicalProcessorNumber;\r
  BufferSize += MonitorFilterSize * MaxLogicalProcessorNumber;\r
  BufferSize += sizeof (CPU_MP_DATA);\r
  BufferSize += ApResetVectorSize;\r
  BufferSize += (sizeof (CPU_AP_DATA) + sizeof (CPU_INFO_IN_HOB))* MaxLogicalProcessorNumber;\r
  MpBuffer    = AllocatePages (EFI_SIZE_TO_PAGES (BufferSize));\r
  ASSERT (MpBuffer != NULL);\r
  ZeroMem (MpBuffer, BufferSize);\r
  Buffer = (UINTN) MpBuffer;\r
\r
  MonitorBuffer    = (UINT8 *) (Buffer + ApStackSize * MaxLogicalProcessorNumber);\r
  BackupBufferAddr = (UINTN) MonitorBuffer + MonitorFilterSize * MaxLogicalProcessorNumber;\r
  CpuMpData = (CPU_MP_DATA *) (BackupBufferAddr + ApResetVectorSize);\r
  CpuMpData->Buffer           = Buffer;\r
  CpuMpData->CpuApStackSize   = ApStackSize;\r
  CpuMpData->BackupBuffer     = BackupBufferAddr;\r
  CpuMpData->BackupBufferSize = ApResetVectorSize;\r
  CpuMpData->EndOfPeiFlag     = FALSE;\r
  CpuMpData->WakeupBuffer     = (UINTN) -1;\r
  CpuMpData->CpuCount         = 1;\r
  CpuMpData->BspNumber        = 0;\r
  CpuMpData->WaitEvent        = NULL;\r
  CpuMpData->CpuData          = (CPU_AP_DATA *) (CpuMpData + 1);\r
  CpuMpData->CpuInfoInHob     = (UINT64) (UINTN) (CpuMpData->CpuData + MaxLogicalProcessorNumber);\r
  InitializeSpinLock(&CpuMpData->MpLock);\r
  //\r
  // Save assembly code information\r
  //\r
  CopyMem (&CpuMpData->AddressMap, &AddressMap, sizeof (MP_ASSEMBLY_ADDRESS_MAP));\r
  //\r
  // Finally set AP loop mode\r
  //\r
  CpuMpData->ApLoopMode = ApLoopMode;\r
  DEBUG ((DEBUG_INFO, "AP Loop Mode is %d\n", CpuMpData->ApLoopMode));\r
  //\r
  // Store BSP's MTRR setting\r
  //\r
  MtrrGetAllMtrrs (&CpuMpData->MtrrTable);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Gets detailed MP-related information on the requested processor at the\r
  instant this call is made. This service may only be called from the BSP.\r
\r
  @param[in]  ProcessorNumber       The handle number of processor.\r
  @param[out] ProcessorInfoBuffer   A pointer to the buffer where information for\r
                                    the requested processor is deposited.\r
  @param[out]  HealthData            Return processor health data.\r
\r
  @retval EFI_SUCCESS             Processor information was returned.\r
  @retval EFI_DEVICE_ERROR        The calling processor is an AP.\r
  @retval EFI_INVALID_PARAMETER   ProcessorInfoBuffer is NULL.\r
  @retval EFI_NOT_FOUND           The processor with the handle specified by\r
                                  ProcessorNumber does not exist in the platform.\r
  @retval EFI_NOT_READY           MP Initialize Library is not initialized.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
MpInitLibGetProcessorInfo (\r
  IN  UINTN                      ProcessorNumber,\r
  OUT EFI_PROCESSOR_INFORMATION  *ProcessorInfoBuffer,\r
  OUT EFI_HEALTH_FLAGS           *HealthData  OPTIONAL\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
/**\r
  This return the handle number for the calling processor.  This service may be\r
  called from the BSP and APs.\r
\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
                               MpInitLibGetNumberOfProcessors().\r
\r
  @retval EFI_SUCCESS             The current processor handle number was returned\r
                                  in ProcessorNumber.\r
  @retval EFI_INVALID_PARAMETER   ProcessorNumber is NULL.\r
  @retval EFI_NOT_READY           MP Initialize Library is not initialized.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
MpInitLibWhoAmI (\r
  OUT UINTN                    *ProcessorNumber\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r
/**\r
  Retrieves the number of logical processor in the platform and the number of\r
  those logical processors that are enabled on this boot. This service may only\r
  be called from the BSP.\r
\r
  @param[out] NumberOfProcessors          Pointer to the total number of logical\r
                                          processors in the system, including the BSP\r
                                          and disabled APs.\r
  @param[out] NumberOfEnabledProcessors   Pointer to the number of enabled logical\r
                                          processors that exist in system, including\r
                                          the BSP.\r
\r
  @retval EFI_SUCCESS             The number of logical processors and enabled\r
                                  logical processors was retrieved.\r
  @retval EFI_DEVICE_ERROR        The calling processor is an AP.\r
  @retval EFI_INVALID_PARAMETER   NumberOfProcessors is NULL and NumberOfEnabledProcessors\r
                                  is NULL.\r
  @retval EFI_NOT_READY           MP Initialize Library is not initialized.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
MpInitLibGetNumberOfProcessors (\r
  OUT UINTN                     *NumberOfProcessors,       OPTIONAL\r
  OUT UINTN                     *NumberOfEnabledProcessors OPTIONAL\r
  )\r
{\r
  return EFI_UNSUPPORTED;\r
}\r