DxeRegisterCpuFeaturesLib.c

Go to the documentation of this file.

 
#include <PiDxe.h>
 
#include <Library/UefiBootServicesTableLib.h>
#include <Library/UefiLib.h>
 
#include "RegisterCpuFeatures.h"
 
CPU_FEATURES_DATA  mCpuFeaturesData = { 0 };
 
CPU_FEATURES_DATA *
GetCpuFeaturesData (
  VOID
  )
{
  return &mCpuFeaturesData;
}
 
MP_SERVICES
GetMpService (
  VOID
  )
{
  EFI_STATUS   Status;
  MP_SERVICES  MpService;
 
  //
  // Get MP Services Protocol
  //
  Status = gBS->LocateProtocol (
                  &gEfiMpServiceProtocolGuid,
                  NULL,
                  (VOID **)&MpService.Protocol
                  );
  ASSERT_EFI_ERROR (Status);
 
  return MpService;
}
 
UINTN
GetProcessorIndex (
  IN CPU_FEATURES_DATA  *CpuFeaturesData
  )
{
  EFI_STATUS                Status;
  UINTN                     ProcessorIndex;
  EFI_MP_SERVICES_PROTOCOL  *MpServices;
 
  MpServices = CpuFeaturesData->MpService.Protocol;
  Status     = MpServices->WhoAmI (MpServices, &ProcessorIndex);
  ASSERT_EFI_ERROR (Status);
  return ProcessorIndex;
}
 
EFI_STATUS
GetProcessorInformation (
  IN  UINTN                      ProcessorNumber,
  OUT EFI_PROCESSOR_INFORMATION  *ProcessorInfoBuffer
  )
{
  EFI_STATUS                Status;
  EFI_MP_SERVICES_PROTOCOL  *MpServices;
  CPU_FEATURES_DATA         *CpuFeaturesData;
 
  CpuFeaturesData = GetCpuFeaturesData ();
  MpServices      = CpuFeaturesData->MpService.Protocol;
 
  Status = MpServices->GetProcessorInfo (
                         MpServices,
                         ProcessorNumber | CPU_V2_EXTENDED_TOPOLOGY,
                         ProcessorInfoBuffer
                         );
  return Status;
}
 
VOID
StartupAllAPsWorker (
  IN  EFI_AP_PROCEDURE  Procedure,
  IN  EFI_EVENT         MpEvent
  )
{
  EFI_STATUS                Status;
  EFI_MP_SERVICES_PROTOCOL  *MpServices;
  CPU_FEATURES_DATA         *CpuFeaturesData;
 
  CpuFeaturesData = GetCpuFeaturesData ();
  MpServices      = CpuFeaturesData->MpService.Protocol;
 
  //
  // Wakeup all APs
  //
  Status = MpServices->StartupAllAPs (
                         MpServices,
                         Procedure,
                         FALSE,
                         MpEvent,
                         0,
                         CpuFeaturesData,
                         NULL
                         );
  ASSERT_EFI_ERROR (Status);
}
 
VOID
SwitchNewBsp (
  IN  UINTN  ProcessorNumber
  )
{
  EFI_STATUS                Status;
  EFI_MP_SERVICES_PROTOCOL  *MpServices;
  CPU_FEATURES_DATA         *CpuFeaturesData;
 
  CpuFeaturesData = GetCpuFeaturesData ();
  MpServices      = CpuFeaturesData->MpService.Protocol;
 
  //
  // Wakeup all APs
  //
  Status = MpServices->SwitchBSP (
                         MpServices,
                         ProcessorNumber,
                         TRUE
                         );
  ASSERT_EFI_ERROR (Status);
}
 
VOID
GetNumberOfProcessor (
  OUT UINTN  *NumberOfCpus,
  OUT UINTN  *NumberOfEnabledProcessors
  )
{
  EFI_STATUS                Status;
  EFI_MP_SERVICES_PROTOCOL  *MpServices;
  CPU_FEATURES_DATA         *CpuFeaturesData;
 
  CpuFeaturesData = GetCpuFeaturesData ();
  MpServices      = CpuFeaturesData->MpService.Protocol;
 
  //
  // Get the number of CPUs
  //
  Status = MpServices->GetNumberOfProcessors (
                         MpServices,
                         NumberOfCpus,
                         NumberOfEnabledProcessors
                         );
  ASSERT_EFI_ERROR (Status);
}
 
VOID
EFIAPI
CpuFeaturesInitialize (
  VOID
  )
{
  CPU_FEATURES_DATA  *CpuFeaturesData;
  UINTN              OldBspNumber;
  EFI_EVENT          MpEvent;
  EFI_STATUS         Status;
 
  CpuFeaturesData = GetCpuFeaturesData ();
 
  OldBspNumber               = GetProcessorIndex (CpuFeaturesData);
  CpuFeaturesData->BspNumber = OldBspNumber;
 
  //
  //
  // Initialize MpEvent to suppress incorrect compiler/analyzer warnings.
  //
  MpEvent = NULL;
 
  if (CpuFeaturesData->NumberOfCpus > 1) {
    Status = gBS->CreateEvent (
                    EVT_NOTIFY_WAIT,
                    TPL_CALLBACK,
                    EfiEventEmptyFunction,
                    NULL,
                    &MpEvent
                    );
    ASSERT_EFI_ERROR (Status);
 
    //
    // Wakeup all APs for programming.
    //
    StartupAllAPsWorker (SetProcessorRegister, MpEvent);
  }
 
  //
  // Programming BSP
  //
  SetProcessorRegister (CpuFeaturesData);
 
  if (CpuFeaturesData->NumberOfCpus > 1) {
    //
    // Wait all processors to finish the task.
    //
    do {
      Status = gBS->CheckEvent (MpEvent);
    } while (Status == EFI_NOT_READY);
 
    ASSERT_EFI_ERROR (Status);
  }
 
  //
  // Switch to new BSP if required
  //
  if (CpuFeaturesData->BspNumber != OldBspNumber) {
    SwitchNewBsp (CpuFeaturesData->BspNumber);
  }
}