123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392 |
- /** @file
- Construct MP Services Protocol on top of the EMU Thread protocol.
- This code makes APs show up in the emulator. PcdEmuApCount is the
- number of APs the emulator should produce.
- The MP Services Protocol provides a generalized way of performing following tasks:
- - Retrieving information of multi-processor environment and MP-related status of
- specific processors.
- - Dispatching user-provided function to APs.
- - Maintain MP-related processor status.
- The MP Services Protocol must be produced on any system with more than one logical
- processor.
- The Protocol is available only during boot time.
- MP Services Protocol is hardware-independent. Most of the logic of this protocol
- is architecturally neutral. It abstracts the multi-processor environment and
- status of processors, and provides interfaces to retrieve information, maintain,
- and dispatch.
- MP Services Protocol may be consumed by ACPI module. The ACPI module may use this
- protocol to retrieve data that are needed for an MP platform and report them to OS.
- MP Services Protocol may also be used to program and configure processors, such
- as MTRR synchronization for memory space attributes setting in DXE Services.
- MP Services Protocol may be used by non-CPU DXE drivers to speed up platform boot
- by taking advantage of the processing capabilities of the APs, for example, using
- APs to help test system memory in parallel with other device initialization.
- Diagnostics applications may also use this protocol for multi-processor.
- Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>
- Portitions Copyright (c) 2011, Apple Inc. All rights reserved.
- SPDX-License-Identifier: BSD-2-Clause-Patent
- **/
- #include "CpuDriver.h"
- MP_SYSTEM_DATA gMPSystem;
- EMU_THREAD_THUNK_PROTOCOL *gThread = NULL;
- EFI_EVENT gReadToBootEvent;
- BOOLEAN gReadToBoot = FALSE;
- UINTN gPollInterval;
- BOOLEAN
- IsBSP (
- VOID
- )
- {
- EFI_STATUS Status;
- UINTN ProcessorNumber;
- Status = CpuMpServicesWhoAmI (&mMpServicesTemplate, &ProcessorNumber);
- if (EFI_ERROR (Status)) {
- return FALSE;
- }
- return (gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0;
- }
- VOID
- SetApProcedure (
- IN PROCESSOR_DATA_BLOCK *Processor,
- IN EFI_AP_PROCEDURE Procedure,
- IN VOID *ProcedureArgument
- )
- {
- gThread->MutexLock (Processor->ProcedureLock);
- Processor->Parameter = ProcedureArgument;
- Processor->Procedure = Procedure;
- gThread->MutexUnlock (Processor->ProcedureLock);
- }
- EFI_STATUS
- GetNextBlockedNumber (
- OUT UINTN *NextNumber
- )
- {
- UINTN Number;
- PROCESSOR_STATE ProcessorState;
- PROCESSOR_DATA_BLOCK *Data;
- for (Number = 0; Number < gMPSystem.NumberOfProcessors; Number++) {
- Data = &gMPSystem.ProcessorData[Number];
- if ((Data->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0) {
- // Skip BSP
- continue;
- }
- gThread->MutexLock (Data->StateLock);
- ProcessorState = Data->State;
- gThread->MutexUnlock (Data->StateLock);
- if (ProcessorState == CPU_STATE_BLOCKED) {
- *NextNumber = Number;
- return EFI_SUCCESS;
- }
- }
- return EFI_NOT_FOUND;
- }
- /**
- * Calculated and stalled the interval time by BSP to check whether
- * the APs have finished.
- *
- * @param[in] Timeout The time limit in microseconds for
- * APs to return from Procedure.
- *
- * @retval StallTime Time of execution stall.
- **/
- UINTN
- CalculateAndStallInterval (
- IN UINTN Timeout
- )
- {
- UINTN StallTime;
- if ((Timeout < gPollInterval) && (Timeout != 0)) {
- StallTime = Timeout;
- } else {
- StallTime = gPollInterval;
- }
- gBS->Stall (StallTime);
- return StallTime;
- }
- /**
- This service retrieves the number of logical processor in the platform
- and the number of those logical processors that are enabled on this boot.
- This service may only be called from the BSP.
- This function is used to retrieve the following information:
- - The number of logical processors that are present in the system.
- - The number of enabled logical processors in the system at the instant
- this call is made.
- Because MP Service Protocol provides services to enable and disable processors
- dynamically, the number of enabled logical processors may vary during the
- course of a boot session.
- If this service is called from an AP, then EFI_DEVICE_ERROR is returned.
- If NumberOfProcessors or NumberOfEnabledProcessors is NULL, then
- EFI_INVALID_PARAMETER is returned. Otherwise, the total number of processors
- is returned in NumberOfProcessors, the number of currently enabled processor
- is returned in NumberOfEnabledProcessors, and EFI_SUCCESS is returned.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL
- instance.
- @param[out] NumberOfProcessors Pointer to the total number of logical
- processors in the system, including the BSP
- and disabled APs.
- @param[out] NumberOfEnabledProcessors Pointer to the number of enabled logical
- processors that exist in system, including
- the BSP.
- @retval EFI_SUCCESS The number of logical processors and enabled
- logical processors was retrieved.
- @retval EFI_DEVICE_ERROR The calling processor is an AP.
- @retval EFI_INVALID_PARAMETER NumberOfProcessors is NULL.
- @retval EFI_INVALID_PARAMETER NumberOfEnabledProcessors is NULL.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesGetNumberOfProcessors (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- OUT UINTN *NumberOfProcessors,
- OUT UINTN *NumberOfEnabledProcessors
- )
- {
- if ((NumberOfProcessors == NULL) || (NumberOfEnabledProcessors == NULL)) {
- return EFI_INVALID_PARAMETER;
- }
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- *NumberOfProcessors = gMPSystem.NumberOfProcessors;
- *NumberOfEnabledProcessors = gMPSystem.NumberOfEnabledProcessors;
- return EFI_SUCCESS;
- }
- /**
- Gets detailed MP-related information on the requested processor at the
- instant this call is made. This service may only be called from the BSP.
- This service retrieves detailed MP-related information about any processor
- on the platform. Note the following:
- - The processor information may change during the course of a boot session.
- - The information presented here is entirely MP related.
- Information regarding the number of caches and their sizes, frequency of operation,
- slot numbers is all considered platform-related information and is not provided
- by this service.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL
- instance.
- @param[in] ProcessorNumber The handle number of processor.
- @param[out] ProcessorInfoBuffer A pointer to the buffer where information for
- the requested processor is deposited.
- @retval EFI_SUCCESS Processor information was returned.
- @retval EFI_DEVICE_ERROR The calling processor is an AP.
- @retval EFI_INVALID_PARAMETER ProcessorInfoBuffer is NULL.
- @retval EFI_NOT_FOUND The processor with the handle specified by
- ProcessorNumber does not exist in the platform.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesGetProcessorInfo (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- IN UINTN ProcessorNumber,
- OUT EFI_PROCESSOR_INFORMATION *ProcessorInfoBuffer
- )
- {
- if (ProcessorInfoBuffer == NULL) {
- return EFI_INVALID_PARAMETER;
- }
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- if (ProcessorNumber >= gMPSystem.NumberOfProcessors) {
- return EFI_NOT_FOUND;
- }
- CopyMem (ProcessorInfoBuffer, &gMPSystem.ProcessorData[ProcessorNumber], sizeof (EFI_PROCESSOR_INFORMATION));
- return EFI_SUCCESS;
- }
- /**
- This service executes a caller provided function on all enabled APs. APs can
- run either simultaneously or one at a time in sequence. This service supports
- both blocking and non-blocking requests. The non-blocking requests use EFI
- events so the BSP can detect when the APs have finished. This service may only
- be called from the BSP.
- This function is used to dispatch all the enabled APs to the function specified
- by Procedure. If any enabled AP is busy, then EFI_NOT_READY is returned
- immediately and Procedure is not started on any AP.
- If SingleThread is TRUE, all the enabled APs execute the function specified by
- Procedure one by one, in ascending order of processor handle number. Otherwise,
- all the enabled APs execute the function specified by Procedure simultaneously.
- If WaitEvent is NULL, execution is in blocking mode. The BSP waits until all
- APs finish or TimeoutInMicroseconds expires. Otherwise, execution is in non-blocking
- mode, and the BSP returns from this service without waiting for APs. If a
- non-blocking mode is requested after the UEFI Event EFI_EVENT_GROUP_READY_TO_BOOT
- is signaled, then EFI_UNSUPPORTED must be returned.
- If the timeout specified by TimeoutInMicroseconds expires before all APs return
- from Procedure, then Procedure on the failed APs is terminated. All enabled APs
- are always available for further calls to EFI_MP_SERVICES_PROTOCOL.StartupAllAPs()
- and EFI_MP_SERVICES_PROTOCOL.StartupThisAP(). If FailedCpuList is not NULL, its
- content points to the list of processor handle numbers in which Procedure was
- terminated.
- Note: It is the responsibility of the consumer of the EFI_MP_SERVICES_PROTOCOL.StartupAllAPs()
- to make sure that the nature of the code that is executed on the BSP and the
- dispatched APs is well controlled. The MP Services Protocol does not guarantee
- that the Procedure function is MP-safe. Hence, the tasks that can be run in
- parallel are limited to certain independent tasks and well-controlled exclusive
- code. EFI services and protocols may not be called by APs unless otherwise
- specified.
- In blocking execution mode, BSP waits until all APs finish or
- TimeoutInMicroseconds expires.
- In non-blocking execution mode, BSP is freed to return to the caller and then
- proceed to the next task without having to wait for APs. The following
- sequence needs to occur in a non-blocking execution mode:
- -# The caller that intends to use this MP Services Protocol in non-blocking
- mode creates WaitEvent by calling the EFI CreateEvent() service. The caller
- invokes EFI_MP_SERVICES_PROTOCOL.StartupAllAPs(). If the parameter WaitEvent
- is not NULL, then StartupAllAPs() executes in non-blocking mode. It requests
- the function specified by Procedure to be started on all the enabled APs,
- and releases the BSP to continue with other tasks.
- -# The caller can use the CheckEvent() and WaitForEvent() services to check
- the state of the WaitEvent created in step 1.
- -# When the APs complete their task or TimeoutInMicroSecondss expires, the MP
- Service signals WaitEvent by calling the EFI SignalEvent() function. If
- FailedCpuList is not NULL, its content is available when WaitEvent is
- signaled. If all APs returned from Procedure prior to the timeout, then
- FailedCpuList is set to NULL. If not all APs return from Procedure before
- the timeout, then FailedCpuList is filled in with the list of the failed
- APs. The buffer is allocated by MP Service Protocol using AllocatePool().
- It is the caller's responsibility to free the buffer with FreePool() service.
- -# This invocation of SignalEvent() function informs the caller that invoked
- EFI_MP_SERVICES_PROTOCOL.StartupAllAPs() that either all the APs completed
- the specified task or a timeout occurred. The contents of FailedCpuList
- can be examined to determine which APs did not complete the specified task
- prior to the timeout.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL
- instance.
- @param[in] Procedure A pointer to the function to be run on
- enabled APs of the system. See type
- EFI_AP_PROCEDURE.
- @param[in] SingleThread If TRUE, then all the enabled APs execute
- the function specified by Procedure one by
- one, in ascending order of processor handle
- number. If FALSE, then all the enabled APs
- execute the function specified by Procedure
- simultaneously.
- @param[in] WaitEvent The event created by the caller with CreateEvent()
- service. If it is NULL, then execute in
- blocking mode. BSP waits until all APs finish
- or TimeoutInMicroseconds expires. If it's
- not NULL, then execute in non-blocking mode.
- BSP requests the function specified by
- Procedure to be started on all the enabled
- APs, and go on executing immediately. If
- all return from Procedure, or TimeoutInMicroseconds
- expires, this event is signaled. The BSP
- can use the CheckEvent() or WaitForEvent()
- services to check the state of event. Type
- EFI_EVENT is defined in CreateEvent() in
- the Unified Extensible Firmware Interface
- Specification.
- @param[in] TimeoutInMicrosecsond Indicates the time limit in microseconds for
- APs to return from Procedure, either for
- blocking or non-blocking mode. Zero means
- infinity. If the timeout expires before
- all APs return from Procedure, then Procedure
- on the failed APs is terminated. All enabled
- APs are available for next function assigned
- by EFI_MP_SERVICES_PROTOCOL.StartupAllAPs()
- or EFI_MP_SERVICES_PROTOCOL.StartupThisAP().
- If the timeout expires in blocking mode,
- BSP returns EFI_TIMEOUT. If the timeout
- expires in non-blocking mode, WaitEvent
- is signaled with SignalEvent().
- @param[in] ProcedureArgument The parameter passed into Procedure for
- all APs.
- @param[out] FailedCpuList If NULL, this parameter is ignored. Otherwise,
- if all APs finish successfully, then its
- content is set to NULL. If not all APs
- finish before timeout expires, then its
- content is set to address of the buffer
- holding handle numbers of the failed APs.
- The buffer is allocated by MP Service Protocol,
- and it's the caller's responsibility to
- free the buffer with FreePool() service.
- In blocking mode, it is ready for consumption
- when the call returns. In non-blocking mode,
- it is ready when WaitEvent is signaled. The
- list of failed CPU is terminated by
- END_OF_CPU_LIST.
- @retval EFI_SUCCESS In blocking mode, all APs have finished before
- the timeout expired.
- @retval EFI_SUCCESS In non-blocking mode, function has been dispatched
- to all enabled APs.
- @retval EFI_UNSUPPORTED A non-blocking mode request was made after the
- UEFI event EFI_EVENT_GROUP_READY_TO_BOOT was
- signaled.
- @retval EFI_DEVICE_ERROR Caller processor is AP.
- @retval EFI_NOT_STARTED No enabled APs exist in the system.
- @retval EFI_NOT_READY Any enabled APs are busy.
- @retval EFI_TIMEOUT In blocking mode, the timeout expired before
- all enabled APs have finished.
- @retval EFI_INVALID_PARAMETER Procedure is NULL.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesStartupAllAps (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- IN EFI_AP_PROCEDURE Procedure,
- IN BOOLEAN SingleThread,
- IN EFI_EVENT WaitEvent OPTIONAL,
- IN UINTN TimeoutInMicroseconds,
- IN VOID *ProcedureArgument OPTIONAL,
- OUT UINTN **FailedCpuList OPTIONAL
- )
- {
- EFI_STATUS Status;
- PROCESSOR_DATA_BLOCK *ProcessorData;
- UINTN Number;
- UINTN NextNumber;
- PROCESSOR_STATE APInitialState;
- PROCESSOR_STATE ProcessorState;
- UINTN Timeout;
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- if (gMPSystem.NumberOfProcessors == 1) {
- return EFI_NOT_STARTED;
- }
- if (Procedure == NULL) {
- return EFI_INVALID_PARAMETER;
- }
- if ((WaitEvent != NULL) && gReadToBoot) {
- return EFI_UNSUPPORTED;
- }
- for (Number = 0; Number < gMPSystem.NumberOfProcessors; Number++) {
- ProcessorData = &gMPSystem.ProcessorData[Number];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- continue;
- }
- gThread->MutexLock (ProcessorData->StateLock);
- if (ProcessorData->State != CPU_STATE_IDLE) {
- gThread->MutexUnlock (ProcessorData->StateLock);
- return EFI_NOT_READY;
- }
- gThread->MutexUnlock (ProcessorData->StateLock);
- }
- if (FailedCpuList != NULL) {
- gMPSystem.FailedList = AllocatePool ((gMPSystem.NumberOfProcessors + 1) * sizeof (UINTN));
- if (gMPSystem.FailedList == NULL) {
- return EFI_OUT_OF_RESOURCES;
- }
- SetMemN (gMPSystem.FailedList, (gMPSystem.NumberOfProcessors + 1) * sizeof (UINTN), END_OF_CPU_LIST);
- gMPSystem.FailedListIndex = 0;
- *FailedCpuList = gMPSystem.FailedList;
- }
- Timeout = TimeoutInMicroseconds;
- ProcessorData = NULL;
- gMPSystem.FinishCount = 0;
- gMPSystem.StartCount = 0;
- gMPSystem.SingleThread = SingleThread;
- APInitialState = CPU_STATE_READY;
- for (Number = 0; Number < gMPSystem.NumberOfProcessors; Number++) {
- ProcessorData = &gMPSystem.ProcessorData[Number];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- gMPSystem.FailedList[gMPSystem.FailedListIndex++] = Number;
- continue;
- }
- //
- // Get APs prepared, and put failing APs into FailedCpuList
- // if "SingleThread", only 1 AP will put to ready state, other AP will be put to ready
- // state 1 by 1, until the previous 1 finished its task
- // if not "SingleThread", all APs are put to ready state from the beginning
- //
- gThread->MutexLock (ProcessorData->StateLock);
- ASSERT (ProcessorData->State == CPU_STATE_IDLE);
- ProcessorData->State = APInitialState;
- gThread->MutexUnlock (ProcessorData->StateLock);
- gMPSystem.StartCount++;
- if (SingleThread) {
- APInitialState = CPU_STATE_BLOCKED;
- }
- }
- if (WaitEvent != NULL) {
- for (Number = 0; Number < gMPSystem.NumberOfProcessors; Number++) {
- ProcessorData = &gMPSystem.ProcessorData[Number];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- continue;
- }
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorState = ProcessorData->State;
- gThread->MutexUnlock (ProcessorData->StateLock);
- if (ProcessorState == CPU_STATE_READY) {
- SetApProcedure (ProcessorData, Procedure, ProcedureArgument);
- }
- }
- //
- // Save data into private data structure, and create timer to poll AP state before exiting
- //
- gMPSystem.Procedure = Procedure;
- gMPSystem.ProcedureArgument = ProcedureArgument;
- gMPSystem.WaitEvent = WaitEvent;
- gMPSystem.Timeout = TimeoutInMicroseconds;
- gMPSystem.TimeoutActive = (BOOLEAN)(TimeoutInMicroseconds != 0);
- Status = gBS->SetTimer (
- gMPSystem.CheckAllAPsEvent,
- TimerPeriodic,
- gPollInterval
- );
- return Status;
- }
- while (TRUE) {
- for (Number = 0; Number < gMPSystem.NumberOfProcessors; Number++) {
- ProcessorData = &gMPSystem.ProcessorData[Number];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- continue;
- }
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorState = ProcessorData->State;
- gThread->MutexUnlock (ProcessorData->StateLock);
- switch (ProcessorState) {
- case CPU_STATE_READY:
- SetApProcedure (ProcessorData, Procedure, ProcedureArgument);
- break;
- case CPU_STATE_FINISHED:
- gMPSystem.FinishCount++;
- if (SingleThread) {
- Status = GetNextBlockedNumber (&NextNumber);
- if (!EFI_ERROR (Status)) {
- gThread->MutexLock (gMPSystem.ProcessorData[NextNumber].StateLock);
- gMPSystem.ProcessorData[NextNumber].State = CPU_STATE_READY;
- gThread->MutexUnlock (gMPSystem.ProcessorData[NextNumber].StateLock);
- }
- }
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorData->State = CPU_STATE_IDLE;
- gThread->MutexUnlock (ProcessorData->StateLock);
- break;
- default:
- break;
- }
- }
- if (gMPSystem.FinishCount == gMPSystem.StartCount) {
- Status = EFI_SUCCESS;
- goto Done;
- }
- if ((TimeoutInMicroseconds != 0) && (Timeout == 0)) {
- Status = EFI_TIMEOUT;
- goto Done;
- }
- Timeout -= CalculateAndStallInterval (Timeout);
- }
- Done:
- if (FailedCpuList != NULL) {
- if (gMPSystem.FailedListIndex == 0) {
- FreePool (*FailedCpuList);
- *FailedCpuList = NULL;
- }
- }
- return EFI_SUCCESS;
- }
- /**
- This service lets the caller get one enabled AP to execute a caller-provided
- function. The caller can request the BSP to either wait for the completion
- of the AP or just proceed with the next task by using the EFI event mechanism.
- See EFI_MP_SERVICES_PROTOCOL.StartupAllAPs() for more details on non-blocking
- execution support. This service may only be called from the BSP.
- This function is used to dispatch one enabled AP to the function specified by
- Procedure passing in the argument specified by ProcedureArgument. If WaitEvent
- is NULL, execution is in blocking mode. The BSP waits until the AP finishes or
- TimeoutInMicroSecondss expires. Otherwise, execution is in non-blocking mode.
- BSP proceeds to the next task without waiting for the AP. If a non-blocking mode
- is requested after the UEFI Event EFI_EVENT_GROUP_READY_TO_BOOT is signaled,
- then EFI_UNSUPPORTED must be returned.
- If the timeout specified by TimeoutInMicroseconds expires before the AP returns
- from Procedure, then execution of Procedure by the AP is terminated. The AP is
- available for subsequent calls to EFI_MP_SERVICES_PROTOCOL.StartupAllAPs() and
- EFI_MP_SERVICES_PROTOCOL.StartupThisAP().
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL
- instance.
- @param[in] Procedure A pointer to the function to be run on
- enabled APs of the system. See type
- EFI_AP_PROCEDURE.
- @param[in] ProcessorNumber The handle number of the AP. The range is
- from 0 to the total number of logical
- processors minus 1. The total number of
- logical processors can be retrieved by
- EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().
- @param[in] WaitEvent The event created by the caller with CreateEvent()
- service. If it is NULL, then execute in
- blocking mode. BSP waits until all APs finish
- or TimeoutInMicroseconds expires. If it's
- not NULL, then execute in non-blocking mode.
- BSP requests the function specified by
- Procedure to be started on all the enabled
- APs, and go on executing immediately. If
- all return from Procedure or TimeoutInMicroseconds
- expires, this event is signaled. The BSP
- can use the CheckEvent() or WaitForEvent()
- services to check the state of event. Type
- EFI_EVENT is defined in CreateEvent() in
- the Unified Extensible Firmware Interface
- Specification.
- @param[in] TimeoutInMicrosecsond Indicates the time limit in microseconds for
- APs to return from Procedure, either for
- blocking or non-blocking mode. Zero means
- infinity. If the timeout expires before
- all APs return from Procedure, then Procedure
- on the failed APs is terminated. All enabled
- APs are available for next function assigned
- by EFI_MP_SERVICES_PROTOCOL.StartupAllAPs()
- or EFI_MP_SERVICES_PROTOCOL.StartupThisAP().
- If the timeout expires in blocking mode,
- BSP returns EFI_TIMEOUT. If the timeout
- expires in non-blocking mode, WaitEvent
- is signaled with SignalEvent().
- @param[in] ProcedureArgument The parameter passed into Procedure for
- all APs.
- @param[out] Finished If NULL, this parameter is ignored. In
- blocking mode, this parameter is ignored.
- In non-blocking mode, if AP returns from
- Procedure before the timeout expires, its
- content is set to TRUE. Otherwise, the
- value is set to FALSE. The caller can
- determine if the AP returned from Procedure
- by evaluating this value.
- @retval EFI_SUCCESS In blocking mode, specified AP finished before
- the timeout expires.
- @retval EFI_SUCCESS In non-blocking mode, the function has been
- dispatched to specified AP.
- @retval EFI_UNSUPPORTED A non-blocking mode request was made after the
- UEFI event EFI_EVENT_GROUP_READY_TO_BOOT was
- signaled.
- @retval EFI_DEVICE_ERROR The calling processor is an AP.
- @retval EFI_TIMEOUT In blocking mode, the timeout expired before
- the specified AP has finished.
- @retval EFI_NOT_READY The specified AP is busy.
- @retval EFI_NOT_FOUND The processor with the handle specified by
- ProcessorNumber does not exist.
- @retval EFI_INVALID_PARAMETER ProcessorNumber specifies the BSP or disabled AP.
- @retval EFI_INVALID_PARAMETER Procedure is NULL.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesStartupThisAP (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- IN EFI_AP_PROCEDURE Procedure,
- IN UINTN ProcessorNumber,
- IN EFI_EVENT WaitEvent OPTIONAL,
- IN UINTN TimeoutInMicroseconds,
- IN VOID *ProcedureArgument OPTIONAL,
- OUT BOOLEAN *Finished OPTIONAL
- )
- {
- UINTN Timeout;
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- if (Procedure == NULL) {
- return EFI_INVALID_PARAMETER;
- }
- if (ProcessorNumber >= gMPSystem.NumberOfProcessors) {
- return EFI_NOT_FOUND;
- }
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0) {
- return EFI_INVALID_PARAMETER;
- }
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- return EFI_INVALID_PARAMETER;
- }
- gThread->MutexLock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if (gMPSystem.ProcessorData[ProcessorNumber].State != CPU_STATE_IDLE) {
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- return EFI_NOT_READY;
- }
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if ((WaitEvent != NULL) && gReadToBoot) {
- return EFI_UNSUPPORTED;
- }
- Timeout = TimeoutInMicroseconds;
- gMPSystem.StartCount = 1;
- gMPSystem.FinishCount = 0;
- SetApProcedure (&gMPSystem.ProcessorData[ProcessorNumber], Procedure, ProcedureArgument);
- if (WaitEvent != NULL) {
- // Non Blocking
- gMPSystem.WaitEvent = WaitEvent;
- gBS->SetTimer (
- gMPSystem.ProcessorData[ProcessorNumber].CheckThisAPEvent,
- TimerPeriodic,
- gPollInterval
- );
- return EFI_SUCCESS;
- }
- // Blocking
- while (TRUE) {
- gThread->MutexLock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if (gMPSystem.ProcessorData[ProcessorNumber].State == CPU_STATE_FINISHED) {
- gMPSystem.ProcessorData[ProcessorNumber].State = CPU_STATE_IDLE;
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- break;
- }
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if ((TimeoutInMicroseconds != 0) && (Timeout == 0)) {
- return EFI_TIMEOUT;
- }
- Timeout -= CalculateAndStallInterval (Timeout);
- }
- return EFI_SUCCESS;
- }
- /**
- This service switches the requested AP to be the BSP from that point onward.
- This service changes the BSP for all purposes. This call can only be performed
- by the current BSP.
- This service switches the requested AP to be the BSP from that point onward.
- This service changes the BSP for all purposes. The new BSP can take over the
- execution of the old BSP and continue seamlessly from where the old one left
- off. This service may not be supported after the UEFI Event EFI_EVENT_GROUP_READY_TO_BOOT
- is signaled.
- If the BSP cannot be switched prior to the return from this service, then
- EFI_UNSUPPORTED must be returned.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL instance.
- @param[in] ProcessorNumber The handle number of AP that is to become the new
- BSP. The range is from 0 to the total number of
- logical processors minus 1. The total number of
- logical processors can be retrieved by
- EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().
- @param[in] EnableOldBSP If TRUE, then the old BSP will be listed as an
- enabled AP. Otherwise, it will be disabled.
- @retval EFI_SUCCESS BSP successfully switched.
- @retval EFI_UNSUPPORTED Switching the BSP cannot be completed prior to
- this service returning.
- @retval EFI_UNSUPPORTED Switching the BSP is not supported.
- @retval EFI_SUCCESS The calling processor is an AP.
- @retval EFI_NOT_FOUND The processor with the handle specified by
- ProcessorNumber does not exist.
- @retval EFI_INVALID_PARAMETER ProcessorNumber specifies the current BSP or
- a disabled AP.
- @retval EFI_NOT_READY The specified AP is busy.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesSwitchBSP (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- IN UINTN ProcessorNumber,
- IN BOOLEAN EnableOldBSP
- )
- {
- UINTN Index;
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- if (ProcessorNumber >= gMPSystem.NumberOfProcessors) {
- return EFI_NOT_FOUND;
- }
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- return EFI_INVALID_PARAMETER;
- }
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0) {
- return EFI_INVALID_PARAMETER;
- }
- for (Index = 0; Index < gMPSystem.NumberOfProcessors; Index++) {
- if ((gMPSystem.ProcessorData[Index].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0) {
- break;
- }
- }
- ASSERT (Index != gMPSystem.NumberOfProcessors);
- gThread->MutexLock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if (gMPSystem.ProcessorData[ProcessorNumber].State != CPU_STATE_IDLE) {
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- return EFI_NOT_READY;
- }
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- // Skip for now as we need switch a bunch of stack stuff around and it's complex
- // May not be worth it?
- return EFI_NOT_READY;
- }
- /**
- This service lets the caller enable or disable an AP from this point onward.
- This service may only be called from the BSP.
- This service allows the caller enable or disable an AP from this point onward.
- The caller can optionally specify the health status of the AP by Health. If
- an AP is being disabled, then the state of the disabled AP is implementation
- dependent. If an AP is enabled, then the implementation must guarantee that a
- complete initialization sequence is performed on the AP, so the AP is in a state
- that is compatible with an MP operating system. This service may not be supported
- after the UEFI Event EFI_EVENT_GROUP_READY_TO_BOOT is signaled.
- If the enable or disable AP operation cannot be completed prior to the return
- from this service, then EFI_UNSUPPORTED must be returned.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL instance.
- @param[in] ProcessorNumber The handle number of AP that is to become the new
- BSP. The range is from 0 to the total number of
- logical processors minus 1. The total number of
- logical processors can be retrieved by
- EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().
- @param[in] EnableAP Specifies the new state for the processor for
- enabled, FALSE for disabled.
- @param[in] HealthFlag If not NULL, a pointer to a value that specifies
- the new health status of the AP. This flag
- corresponds to StatusFlag defined in
- EFI_MP_SERVICES_PROTOCOL.GetProcessorInfo(). Only
- the PROCESSOR_HEALTH_STATUS_BIT is used. All other
- bits are ignored. If it is NULL, this parameter
- is ignored.
- @retval EFI_SUCCESS The specified AP was enabled or disabled successfully.
- @retval EFI_UNSUPPORTED Enabling or disabling an AP cannot be completed
- prior to this service returning.
- @retval EFI_UNSUPPORTED Enabling or disabling an AP is not supported.
- @retval EFI_DEVICE_ERROR The calling processor is an AP.
- @retval EFI_NOT_FOUND Processor with the handle specified by ProcessorNumber
- does not exist.
- @retval EFI_INVALID_PARAMETER ProcessorNumber specifies the BSP.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesEnableDisableAP (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- IN UINTN ProcessorNumber,
- IN BOOLEAN EnableAP,
- IN UINT32 *HealthFlag OPTIONAL
- )
- {
- if (!IsBSP ()) {
- return EFI_DEVICE_ERROR;
- }
- if (ProcessorNumber >= gMPSystem.NumberOfProcessors) {
- return EFI_NOT_FOUND;
- }
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) != 0) {
- return EFI_INVALID_PARAMETER;
- }
- gThread->MutexLock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if (gMPSystem.ProcessorData[ProcessorNumber].State != CPU_STATE_IDLE) {
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- return EFI_UNSUPPORTED;
- }
- gThread->MutexUnlock (gMPSystem.ProcessorData[ProcessorNumber].StateLock);
- if (EnableAP) {
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0 ) {
- gMPSystem.NumberOfEnabledProcessors++;
- }
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag |= PROCESSOR_ENABLED_BIT;
- } else {
- if ((gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag & PROCESSOR_ENABLED_BIT) == PROCESSOR_ENABLED_BIT ) {
- gMPSystem.NumberOfEnabledProcessors--;
- }
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag &= ~PROCESSOR_ENABLED_BIT;
- }
- if (HealthFlag != NULL) {
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag &= ~PROCESSOR_HEALTH_STATUS_BIT;
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag |= (*HealthFlag & PROCESSOR_HEALTH_STATUS_BIT);
- }
- return EFI_SUCCESS;
- }
- /**
- This return the handle number for the calling processor. This service may be
- called from the BSP and APs.
- This service returns the processor handle number for the calling processor.
- The returned value is in the range from 0 to the total number of logical
- processors minus 1. The total number of logical processors can be retrieved
- with EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors(). This service may be
- called from the BSP and APs. If ProcessorNumber is NULL, then EFI_INVALID_PARAMETER
- is returned. Otherwise, the current processors handle number is returned in
- ProcessorNumber, and EFI_SUCCESS is returned.
- @param[in] This A pointer to the EFI_MP_SERVICES_PROTOCOL instance.
- @param[in] ProcessorNumber The handle number of AP that is to become the new
- BSP. The range is from 0 to the total number of
- logical processors minus 1. The total number of
- logical processors can be retrieved by
- EFI_MP_SERVICES_PROTOCOL.GetNumberOfProcessors().
- @retval EFI_SUCCESS The current processor handle number was returned
- in ProcessorNumber.
- @retval EFI_INVALID_PARAMETER ProcessorNumber is NULL.
- **/
- EFI_STATUS
- EFIAPI
- CpuMpServicesWhoAmI (
- IN EFI_MP_SERVICES_PROTOCOL *This,
- OUT UINTN *ProcessorNumber
- )
- {
- UINTN Index;
- UINT64 ProcessorId;
- if (ProcessorNumber == NULL) {
- return EFI_INVALID_PARAMETER;
- }
- ProcessorId = gThread->Self ();
- for (Index = 0; Index < gMPSystem.NumberOfProcessors; Index++) {
- if (gMPSystem.ProcessorData[Index].Info.ProcessorId == ProcessorId) {
- break;
- }
- }
- *ProcessorNumber = Index;
- return EFI_SUCCESS;
- }
- EFI_MP_SERVICES_PROTOCOL mMpServicesTemplate = {
- CpuMpServicesGetNumberOfProcessors,
- CpuMpServicesGetProcessorInfo,
- CpuMpServicesStartupAllAps,
- CpuMpServicesStartupThisAP,
- CpuMpServicesSwitchBSP,
- CpuMpServicesEnableDisableAP,
- CpuMpServicesWhoAmI
- };
- /*++
- If timeout occurs in StartupAllAps(), a timer is set, which invokes this
- procedure periodically to check whether all APs have finished.
- --*/
- VOID
- EFIAPI
- CpuCheckAllAPsStatus (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- UINTN ProcessorNumber;
- UINTN NextNumber;
- PROCESSOR_DATA_BLOCK *ProcessorData;
- PROCESSOR_DATA_BLOCK *NextData;
- EFI_STATUS Status;
- PROCESSOR_STATE ProcessorState;
- UINTN Cpu;
- BOOLEAN Found;
- if (gMPSystem.TimeoutActive) {
- gMPSystem.Timeout -= CalculateAndStallInterval (gMPSystem.Timeout);
- }
- for (ProcessorNumber = 0; ProcessorNumber < gMPSystem.NumberOfProcessors; ProcessorNumber++) {
- ProcessorData = &gMPSystem.ProcessorData[ProcessorNumber];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- continue;
- }
- // This is an Interrupt Service routine.
- // This can grab a lock that is held in a non-interrupt
- // context. Meaning deadlock. Which is a bad thing.
- // So, try lock it. If we can get it, cool, do our thing.
- // otherwise, just dump out & try again on the next iteration.
- Status = gThread->MutexTryLock (ProcessorData->StateLock);
- if (EFI_ERROR (Status)) {
- return;
- }
- ProcessorState = ProcessorData->State;
- gThread->MutexUnlock (ProcessorData->StateLock);
- switch (ProcessorState) {
- case CPU_STATE_FINISHED:
- if (gMPSystem.SingleThread) {
- Status = GetNextBlockedNumber (&NextNumber);
- if (!EFI_ERROR (Status)) {
- NextData = &gMPSystem.ProcessorData[NextNumber];
- gThread->MutexLock (NextData->StateLock);
- NextData->State = CPU_STATE_READY;
- gThread->MutexUnlock (NextData->StateLock);
- SetApProcedure (NextData, gMPSystem.Procedure, gMPSystem.ProcedureArgument);
- }
- }
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorData->State = CPU_STATE_IDLE;
- gThread->MutexUnlock (ProcessorData->StateLock);
- gMPSystem.FinishCount++;
- break;
- default:
- break;
- }
- }
- if (gMPSystem.TimeoutActive && (gMPSystem.Timeout == 0)) {
- //
- // Timeout
- //
- if (gMPSystem.FailedList != NULL) {
- for (ProcessorNumber = 0; ProcessorNumber < gMPSystem.NumberOfProcessors; ProcessorNumber++) {
- ProcessorData = &gMPSystem.ProcessorData[ProcessorNumber];
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- if ((ProcessorData->Info.StatusFlag & PROCESSOR_ENABLED_BIT) == 0) {
- // Skip Disabled processors
- continue;
- }
- // Mark the
- Status = gThread->MutexTryLock (ProcessorData->StateLock);
- if (EFI_ERROR (Status)) {
- return;
- }
- ProcessorState = ProcessorData->State;
- gThread->MutexUnlock (ProcessorData->StateLock);
- if (ProcessorState != CPU_STATE_IDLE) {
- // If we are retrying make sure we don't double count
- for (Cpu = 0, Found = FALSE; Cpu < gMPSystem.NumberOfProcessors; Cpu++) {
- if (gMPSystem.FailedList[Cpu] == END_OF_CPU_LIST) {
- break;
- }
- if (gMPSystem.FailedList[ProcessorNumber] == Cpu) {
- Found = TRUE;
- break;
- }
- }
- if (!Found) {
- gMPSystem.FailedList[gMPSystem.FailedListIndex++] = Cpu;
- }
- }
- }
- }
- // Force terminal exit
- gMPSystem.FinishCount = gMPSystem.StartCount;
- }
- if (gMPSystem.FinishCount != gMPSystem.StartCount) {
- return;
- }
- gBS->SetTimer (
- gMPSystem.CheckAllAPsEvent,
- TimerCancel,
- 0
- );
- if (gMPSystem.FailedListIndex == 0) {
- if (gMPSystem.FailedList != NULL) {
- FreePool (gMPSystem.FailedList);
- gMPSystem.FailedList = NULL;
- }
- }
- Status = gBS->SignalEvent (gMPSystem.WaitEvent);
- return;
- }
- VOID
- EFIAPI
- CpuCheckThisAPStatus (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- EFI_STATUS Status;
- PROCESSOR_DATA_BLOCK *ProcessorData;
- PROCESSOR_STATE ProcessorState;
- ProcessorData = (PROCESSOR_DATA_BLOCK *)Context;
- //
- // This is an Interrupt Service routine.
- // that can grab a lock that is held in a non-interrupt
- // context. Meaning deadlock. Which is a badddd thing.
- // So, try lock it. If we can get it, cool, do our thing.
- // otherwise, just dump out & try again on the next iteration.
- //
- Status = gThread->MutexTryLock (ProcessorData->StateLock);
- if (EFI_ERROR (Status)) {
- return;
- }
- ProcessorState = ProcessorData->State;
- gThread->MutexUnlock (ProcessorData->StateLock);
- if (ProcessorState == CPU_STATE_FINISHED) {
- Status = gBS->SetTimer (ProcessorData->CheckThisAPEvent, TimerCancel, 0);
- ASSERT_EFI_ERROR (Status);
- Status = gBS->SignalEvent (gMPSystem.WaitEvent);
- ASSERT_EFI_ERROR (Status);
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorData->State = CPU_STATE_IDLE;
- gThread->MutexUnlock (ProcessorData->StateLock);
- }
- return;
- }
- /*++
- This function is called by all processors (both BSP and AP) once and collects MP related data
- MPSystemData - Pointer to the data structure containing MP related data
- BSP - TRUE if the CPU is BSP
- EFI_SUCCESS - Data for the processor collected and filled in
- --*/
- EFI_STATUS
- FillInProcessorInformation (
- IN BOOLEAN BSP,
- IN UINTN ProcessorNumber
- )
- {
- gMPSystem.ProcessorData[ProcessorNumber].Info.ProcessorId = gThread->Self ();
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag = PROCESSOR_ENABLED_BIT | PROCESSOR_HEALTH_STATUS_BIT;
- if (BSP) {
- gMPSystem.ProcessorData[ProcessorNumber].Info.StatusFlag |= PROCESSOR_AS_BSP_BIT;
- }
- gMPSystem.ProcessorData[ProcessorNumber].Info.Location.Package = (UINT32)ProcessorNumber;
- gMPSystem.ProcessorData[ProcessorNumber].Info.Location.Core = 0;
- gMPSystem.ProcessorData[ProcessorNumber].Info.Location.Thread = 0;
- gMPSystem.ProcessorData[ProcessorNumber].State = BSP ? CPU_STATE_BUSY : CPU_STATE_IDLE;
- gMPSystem.ProcessorData[ProcessorNumber].Procedure = NULL;
- gMPSystem.ProcessorData[ProcessorNumber].Parameter = NULL;
- gMPSystem.ProcessorData[ProcessorNumber].StateLock = gThread->MutexInit ();
- gMPSystem.ProcessorData[ProcessorNumber].ProcedureLock = gThread->MutexInit ();
- return EFI_SUCCESS;
- }
- VOID *
- EFIAPI
- CpuDriverApIdolLoop (
- VOID *Context
- )
- {
- EFI_AP_PROCEDURE Procedure;
- VOID *Parameter;
- UINTN ProcessorNumber;
- PROCESSOR_DATA_BLOCK *ProcessorData;
- ProcessorNumber = (UINTN)Context;
- ProcessorData = &gMPSystem.ProcessorData[ProcessorNumber];
- ProcessorData->Info.ProcessorId = gThread->Self ();
- while (TRUE) {
- //
- // Make a local copy on the stack to be extra safe
- //
- gThread->MutexLock (ProcessorData->ProcedureLock);
- Procedure = ProcessorData->Procedure;
- Parameter = ProcessorData->Parameter;
- gThread->MutexUnlock (ProcessorData->ProcedureLock);
- if (Procedure != NULL) {
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorData->State = CPU_STATE_BUSY;
- gThread->MutexUnlock (ProcessorData->StateLock);
- Procedure (Parameter);
- gThread->MutexLock (ProcessorData->ProcedureLock);
- ProcessorData->Procedure = NULL;
- gThread->MutexUnlock (ProcessorData->ProcedureLock);
- gThread->MutexLock (ProcessorData->StateLock);
- ProcessorData->State = CPU_STATE_FINISHED;
- gThread->MutexUnlock (ProcessorData->StateLock);
- }
- // Poll 5 times a seconds, 200ms
- // Don't want to burn too many system resources doing nothing.
- gEmuThunk->Sleep (200 * 1000);
- }
- return 0;
- }
- EFI_STATUS
- InitializeMpSystemData (
- IN UINTN NumberOfProcessors
- )
- {
- EFI_STATUS Status;
- UINTN Index;
- //
- // Clear the data structure area first.
- //
- ZeroMem (&gMPSystem, sizeof (MP_SYSTEM_DATA));
- //
- // First BSP fills and inits all known values, including it's own records.
- //
- gMPSystem.NumberOfProcessors = NumberOfProcessors;
- gMPSystem.NumberOfEnabledProcessors = NumberOfProcessors;
- gMPSystem.ProcessorData = AllocateZeroPool (gMPSystem.NumberOfProcessors * sizeof (PROCESSOR_DATA_BLOCK));
- ASSERT (gMPSystem.ProcessorData != NULL);
- FillInProcessorInformation (TRUE, 0);
- Status = gBS->CreateEvent (
- EVT_TIMER | EVT_NOTIFY_SIGNAL,
- TPL_CALLBACK,
- CpuCheckAllAPsStatus,
- NULL,
- &gMPSystem.CheckAllAPsEvent
- );
- ASSERT_EFI_ERROR (Status);
- for (Index = 0; Index < gMPSystem.NumberOfProcessors; Index++) {
- if ((gMPSystem.ProcessorData[Index].Info.StatusFlag & PROCESSOR_AS_BSP_BIT) == PROCESSOR_AS_BSP_BIT) {
- // Skip BSP
- continue;
- }
- FillInProcessorInformation (FALSE, Index);
- Status = gThread->CreateThread (
- (VOID *)&gMPSystem.ProcessorData[Index].Info.ProcessorId,
- NULL,
- CpuDriverApIdolLoop,
- (VOID *)Index
- );
- Status = gBS->CreateEvent (
- EVT_TIMER | EVT_NOTIFY_SIGNAL,
- TPL_CALLBACK,
- CpuCheckThisAPStatus,
- (VOID *)&gMPSystem.ProcessorData[Index],
- &gMPSystem.ProcessorData[Index].CheckThisAPEvent
- );
- }
- return EFI_SUCCESS;
- }
- /**
- Invoke a notification event
- @param Event Event whose notification function is being invoked.
- @param Context The pointer to the notification function's context,
- which is implementation-dependent.
- **/
- VOID
- EFIAPI
- CpuReadToBootFunction (
- IN EFI_EVENT Event,
- IN VOID *Context
- )
- {
- gReadToBoot = TRUE;
- }
- EFI_STATUS
- CpuMpServicesInit (
- OUT UINTN *MaxCpus
- )
- {
- EFI_STATUS Status;
- EFI_HANDLE Handle;
- EMU_IO_THUNK_PROTOCOL *IoThunk;
- *MaxCpus = 1; // BSP
- IoThunk = GetIoThunkInstance (&gEmuThreadThunkProtocolGuid, 0);
- if (IoThunk != NULL) {
- Status = IoThunk->Open (IoThunk);
- if (!EFI_ERROR (Status)) {
- if (IoThunk->ConfigString != NULL) {
- *MaxCpus += StrDecimalToUintn (IoThunk->ConfigString);
- gThread = IoThunk->Interface;
- }
- }
- }
- if (*MaxCpus == 1) {
- // We are not MP so nothing to do
- return EFI_SUCCESS;
- }
- gPollInterval = (UINTN)PcdGet64 (PcdEmuMpServicesPollingInterval);
- Status = InitializeMpSystemData (*MaxCpus);
- if (EFI_ERROR (Status)) {
- return Status;
- }
- Status = EfiCreateEventReadyToBootEx (TPL_CALLBACK, CpuReadToBootFunction, NULL, &gReadToBootEvent);
- ASSERT_EFI_ERROR (Status);
- //
- // Now install the MP services protocol.
- //
- Handle = NULL;
- Status = gBS->InstallMultipleProtocolInterfaces (
- &Handle,
- &gEfiMpServiceProtocolGuid,
- &mMpServicesTemplate,
- NULL
- );
- return Status;
- }
|