2 CPU PEI Module installs CPU Multiple Processor PPI.
4 Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
18 // Global Descriptor Table (GDT)
20 GLOBAL_REMOVE_IF_UNREFERENCED IA32_GDT mGdtEntries
[] = {
21 /* selector { Global Segment Descriptor } */
22 /* 0x00 */ {{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}}, //null descriptor
23 /* 0x08 */ {{0xffff, 0, 0, 0x2, 1, 0, 1, 0xf, 0, 0, 1, 1, 0}}, //linear data segment descriptor
24 /* 0x10 */ {{0xffff, 0, 0, 0xf, 1, 0, 1, 0xf, 0, 0, 1, 1, 0}}, //linear code segment descriptor
25 /* 0x18 */ {{0xffff, 0, 0, 0x3, 1, 0, 1, 0xf, 0, 0, 1, 1, 0}}, //system data segment descriptor
26 /* 0x20 */ {{0xffff, 0, 0, 0xa, 1, 0, 1, 0xf, 0, 0, 1, 1, 0}}, //system code segment descriptor
27 /* 0x28 */ {{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}}, //spare segment descriptor
28 /* 0x30 */ {{0xffff, 0, 0, 0x2, 1, 0, 1, 0xf, 0, 0, 1, 1, 0}}, //system data segment descriptor
29 /* 0x38 */ {{0xffff, 0, 0, 0xa, 1, 0, 1, 0xf, 0, 1, 0, 1, 0}}, //system code segment descriptor
30 /* 0x40 */ {{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}}, //spare segment descriptor
36 GLOBAL_REMOVE_IF_UNREFERENCED IA32_DESCRIPTOR mGdt
= {
37 sizeof (mGdtEntries
) - 1,
42 Sort the APIC ID of all processors.
44 This function sorts the APIC ID of all processors so that processor number is
45 assigned in the ascending order of APIC ID which eases MP debugging.
47 @param PeiCpuMpData Pointer to PEI CPU MP Data
51 IN PEI_CPU_MP_DATA
*PeiCpuMpData
58 EFI_HEALTH_FLAGS Health
;
61 ApCount
= PeiCpuMpData
->CpuCount
- 1;
64 for (Index1
= 0; Index1
< ApCount
; Index1
++) {
67 // Sort key is the hardware default APIC ID
69 ApicId
= PeiCpuMpData
->CpuData
[Index1
].ApicId
;
70 for (Index2
= Index1
+ 1; Index2
<= ApCount
; Index2
++) {
71 if (ApicId
> PeiCpuMpData
->CpuData
[Index2
].ApicId
) {
73 ApicId
= PeiCpuMpData
->CpuData
[Index2
].ApicId
;
76 if (Index3
!= Index1
) {
77 PeiCpuMpData
->CpuData
[Index3
].ApicId
= PeiCpuMpData
->CpuData
[Index1
].ApicId
;
78 PeiCpuMpData
->CpuData
[Index1
].ApicId
= ApicId
;
79 Health
= PeiCpuMpData
->CpuData
[Index3
].Health
;
80 PeiCpuMpData
->CpuData
[Index3
].Health
= PeiCpuMpData
->CpuData
[Index1
].Health
;
81 PeiCpuMpData
->CpuData
[Index1
].Health
= Health
;
86 // Get the processor number for the BSP
88 ApicId
= GetInitialApicId ();
89 for (Index1
= 0; Index1
< PeiCpuMpData
->CpuCount
; Index1
++) {
90 if (PeiCpuMpData
->CpuData
[Index1
].ApicId
== ApicId
) {
91 PeiCpuMpData
->BspNumber
= (UINT32
) Index1
;
98 This function will be called from AP reset code if BSP uses WakeUpAP.
100 @param ExchangeInfo Pointer to the MP exchange info buffer
101 @param NumApsExecuting Number of curret executing AP
106 IN MP_CPU_EXCHANGE_INFO
*ExchangeInfo
,
107 IN UINTN NumApsExecuting
110 PEI_CPU_MP_DATA
*PeiCpuMpData
;
113 PeiCpuMpData
= ExchangeInfo
->PeiCpuMpData
;
114 if (PeiCpuMpData
->InitFlag
) {
116 // This is first time AP wakeup, get BIST inforamtion from AP stack
118 BistData
= *(UINTN
*) (PeiCpuMpData
->Buffer
+ NumApsExecuting
* PeiCpuMpData
->CpuApStackSize
- sizeof (UINTN
));
119 PeiCpuMpData
->CpuData
[NumApsExecuting
].ApicId
= GetInitialApicId ();
120 PeiCpuMpData
->CpuData
[NumApsExecuting
].Health
.Uint32
= (UINT32
) BistData
;
124 // AP finished executing C code
126 InterlockedIncrement ((UINT32
*)&PeiCpuMpData
->FinishedCount
);
131 This function will be called by BSP to wakeup AP.
133 @param PeiCpuMpData Pointer to PEI CPU MP Data
134 @param Broadcast TRUE: Send broadcast IPI to all APs
135 FALSE: Send IPI to AP by ApicId
136 @param ApicId Apic ID for the processor to be waked
137 @param Procedure The function to be invoked by AP
138 @param ProcedureArgument The argument to be passed into AP function
142 IN PEI_CPU_MP_DATA
*PeiCpuMpData
,
143 IN BOOLEAN Broadcast
,
145 IN EFI_AP_PROCEDURE Procedure
, OPTIONAL
146 IN VOID
*ProcedureArgument OPTIONAL
149 volatile MP_CPU_EXCHANGE_INFO
*ExchangeInfo
;
151 PeiCpuMpData
->ApFunction
= (UINTN
) Procedure
;
152 PeiCpuMpData
->ApFunctionArgument
= (UINTN
) ProcedureArgument
;
153 PeiCpuMpData
->FinishedCount
= 0;
155 ExchangeInfo
= PeiCpuMpData
->MpCpuExchangeInfo
;
156 ExchangeInfo
->Lock
= 0;
157 ExchangeInfo
->StackStart
= PeiCpuMpData
->Buffer
;
158 ExchangeInfo
->StackSize
= PeiCpuMpData
->CpuApStackSize
;
159 ExchangeInfo
->BufferStart
= PeiCpuMpData
->WakeupBuffer
;
160 ExchangeInfo
->PmodeOffset
= PeiCpuMpData
->AddressMap
.PModeEntryOffset
;
161 ExchangeInfo
->LmodeOffset
= PeiCpuMpData
->AddressMap
.LModeEntryOffset
;
162 ExchangeInfo
->Cr3
= AsmReadCr3 ();
163 ExchangeInfo
->CFunction
= (UINTN
) ApCFunction
;
164 ExchangeInfo
->NumApsExecuting
= 0;
165 ExchangeInfo
->PeiCpuMpData
= PeiCpuMpData
;
168 // Get the BSP's data of GDT and IDT
170 CopyMem ((VOID
*)&ExchangeInfo
->GdtrProfile
, &mGdt
, sizeof(mGdt
));
171 AsmReadIdtr ((IA32_DESCRIPTOR
*) &ExchangeInfo
->IdtrProfile
);
174 SendInitSipiSipiAllExcludingSelf ((UINT32
) ExchangeInfo
->BufferStart
);
176 SendInitSipiSipi (ApicId
, (UINT32
) ExchangeInfo
->BufferStart
);
183 Get available system memory below 1MB by specified size.
185 @param WakeupBufferSize Wakeup buffer size required
187 @retval other Return wakeup buffer address below 1MB.
188 @retval -1 Cannot find free memory below 1MB.
192 IN UINTN WakeupBufferSize
195 EFI_PEI_HOB_POINTERS Hob
;
196 UINTN WakeupBufferStart
;
197 UINTN WakeupBufferEnd
;
200 // Get the HOB list for processing
202 Hob
.Raw
= GetHobList ();
205 // Collect memory ranges
207 while (!END_OF_HOB_LIST (Hob
)) {
208 if (Hob
.Header
->HobType
== EFI_HOB_TYPE_RESOURCE_DESCRIPTOR
) {
209 if ((Hob
.ResourceDescriptor
->PhysicalStart
< BASE_1MB
) &&
210 (Hob
.ResourceDescriptor
->ResourceType
== EFI_RESOURCE_SYSTEM_MEMORY
) &&
211 ((Hob
.ResourceDescriptor
->ResourceAttribute
&
212 (EFI_RESOURCE_ATTRIBUTE_READ_PROTECTED
|
213 EFI_RESOURCE_ATTRIBUTE_WRITE_PROTECTED
|
214 EFI_RESOURCE_ATTRIBUTE_EXECUTION_PROTECTED
218 // Need memory under 1MB to be collected here
220 WakeupBufferEnd
= (UINTN
) (Hob
.ResourceDescriptor
->PhysicalStart
+ Hob
.ResourceDescriptor
->ResourceLength
);
221 if (WakeupBufferEnd
> BASE_1MB
) {
223 // Wakeup buffer should be under 1MB
225 WakeupBufferEnd
= BASE_1MB
;
228 // Wakeup buffer should be aligned on 4KB
230 WakeupBufferStart
= (WakeupBufferEnd
- WakeupBufferSize
) & ~(SIZE_4KB
- 1);
231 if (WakeupBufferStart
< Hob
.ResourceDescriptor
->PhysicalStart
) {
235 // Create a memory allocation HOB.
237 BuildMemoryAllocationHob (
242 return WakeupBufferStart
;
248 Hob
.Raw
= GET_NEXT_HOB (Hob
);
255 Get available system memory below 1MB by specified size.
257 @param PeiCpuMpData Pointer to PEI CPU MP Data
260 BackupAndPrepareWakeupBuffer(
261 IN PEI_CPU_MP_DATA
*PeiCpuMpData
265 (VOID
*) PeiCpuMpData
->BackupBuffer
,
266 (VOID
*) PeiCpuMpData
->WakeupBuffer
,
267 PeiCpuMpData
->BackupBufferSize
270 (VOID
*) PeiCpuMpData
->WakeupBuffer
,
271 (VOID
*) PeiCpuMpData
->AddressMap
.RendezvousFunnelAddress
,
272 PeiCpuMpData
->AddressMap
.RendezvousFunnelSize
276 This function will get CPU count in the system.
278 @param PeiCpuMpData Pointer to PEI CPU MP Data
280 @return AP processor count
283 CountProcessorNumber (
284 IN PEI_CPU_MP_DATA
*PeiCpuMpData
289 // Send broadcast IPI to APs to wakeup APs
291 PeiCpuMpData
->InitFlag
= 1;
292 WakeUpAP (PeiCpuMpData
, TRUE
, 0, NULL
, NULL
);
294 // Wait for AP task to complete and then exit.
296 MicroSecondDelay (PcdGet32 (PcdCpuApInitTimeOutInMicroSeconds
));
297 PeiCpuMpData
->InitFlag
= 0;
298 PeiCpuMpData
->CpuCount
+= (UINT32
) PeiCpuMpData
->MpCpuExchangeInfo
->NumApsExecuting
;
300 // Sort BSP/Aps by CPU APIC ID in ascending order
302 SortApicId (PeiCpuMpData
);
304 DEBUG ((EFI_D_INFO
, "CpuMpPei: Find %d processors in system.\n", PeiCpuMpData
->CpuCount
));
305 return PeiCpuMpData
->CpuCount
;
309 Prepare for AP wakeup buffer and copy AP reset code into it.
311 Get wakeup buffer below 1MB. Allocate memory for CPU MP Data and APs Stack.
313 @return Pointer to PEI CPU MP Data
316 PrepareAPStartupVector (
322 PEI_CPU_MP_DATA
*PeiCpuMpData
;
323 EFI_PHYSICAL_ADDRESS Buffer
;
326 UINTN WakeupBufferSize
;
327 MP_ASSEMBLY_ADDRESS_MAP AddressMap
;
329 AsmGetAddressMap (&AddressMap
);
330 WakeupBufferSize
= AddressMap
.RendezvousFunnelSize
+ sizeof (MP_CPU_EXCHANGE_INFO
);
331 WakeupBuffer
= GetWakeupBuffer ((WakeupBufferSize
+ SIZE_4KB
- 1) & ~(SIZE_4KB
- 1));
332 DEBUG ((EFI_D_INFO
, "CpuMpPei: WakeupBuffer = 0x%x\n", WakeupBuffer
));
335 // Allocate Pages for APs stack, CPU MP Data and backup buffer for wakeup buffer
337 MaxCpuCount
= PcdGet32(PcdCpuMaxLogicalProcessorNumber
);
338 BufferSize
= PcdGet32 (PcdCpuApStackSize
) * MaxCpuCount
+ sizeof (PEI_CPU_MP_DATA
)
339 + WakeupBufferSize
+ sizeof (PEI_CPU_DATA
) * MaxCpuCount
;
340 Status
= PeiServicesAllocatePages (
342 EFI_SIZE_TO_PAGES (BufferSize
),
345 ASSERT_EFI_ERROR (Status
);
347 PeiCpuMpData
= (PEI_CPU_MP_DATA
*) (UINTN
) (Buffer
+ PcdGet32 (PcdCpuApStackSize
) * MaxCpuCount
);
348 PeiCpuMpData
->Buffer
= (UINTN
) Buffer
;
349 PeiCpuMpData
->CpuApStackSize
= PcdGet32 (PcdCpuApStackSize
);
350 PeiCpuMpData
->WakeupBuffer
= WakeupBuffer
;
351 PeiCpuMpData
->BackupBuffer
= (UINTN
)PeiCpuMpData
+ sizeof (PEI_CPU_MP_DATA
);
352 PeiCpuMpData
->BackupBufferSize
= WakeupBufferSize
;
353 PeiCpuMpData
->MpCpuExchangeInfo
= (MP_CPU_EXCHANGE_INFO
*) (UINTN
) (WakeupBuffer
+ AddressMap
.RendezvousFunnelSize
);
355 PeiCpuMpData
->CpuCount
= 1;
356 PeiCpuMpData
->BspNumber
= 0;
357 PeiCpuMpData
->CpuData
= (PEI_CPU_DATA
*) (PeiCpuMpData
->MpCpuExchangeInfo
+ 1);
358 PeiCpuMpData
->CpuData
[0].ApicId
= GetInitialApicId ();
359 PeiCpuMpData
->CpuData
[0].Health
.Uint32
= 0;
360 CopyMem (&PeiCpuMpData
->AddressMap
, &AddressMap
, sizeof (MP_ASSEMBLY_ADDRESS_MAP
));
363 // Backup original data and copy AP reset code in it
365 BackupAndPrepareWakeupBuffer(PeiCpuMpData
);
370 The Entry point of the MP CPU PEIM.
372 This function will wakeup APs and collect CPU AP count and install the
375 @param FileHandle Handle of the file being invoked.
376 @param PeiServices Describes the list of possible PEI Services.
378 @retval EFI_SUCCESS MpServicePpi is installed successfully.
384 IN EFI_PEI_FILE_HANDLE FileHandle
,
385 IN CONST EFI_PEI_SERVICES
**PeiServices
389 PEI_CPU_MP_DATA
*PeiCpuMpData
;
390 UINT32 ProcessorCount
;
393 // Load new GDT table on BSP
395 AsmInitializeGdt (&mGdt
);
397 // Get wakeup buffer and copy AP reset code in it
399 PeiCpuMpData
= PrepareAPStartupVector ();
401 // Count processor number and collect processor information
403 ProcessorCount
= CountProcessorNumber (PeiCpuMpData
);