tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

UefiCpuPkg: x86 CpuDxe: Allocate AP Exception Stack Below 4GB 

When setting up the APs' exception stacks, the x86 CpuDxe allocates
any range and then copies over the existing GDT and IDT and adds the
appropriate new entries for this AP, then installs them.

This can cause an issue if the allocated buffer is over 4GB because
the next time the AP is started, it goes through an INIT-SIPI-SIPI,
stepping through real mode -> protected mode -> long mode and when it
is in protected mode it needs a 32 code segment descriptor or else it
will fault when trying to execute. If the GDT lives above 4GB, it
cannot be accessed by the protected mode code and the triple fault
is seen.

This patch updates CpuDxe's MP management code to allocate the
exception stacks for all APs below 4GB explicitly to avoid this
problem, such as it does with the BSP's GDT that first gets
populated to the APs.

Signed-off-by: Oliver Smith-Denny <osde@microsoft.com>
This commit is contained in:
Oliver Smith-Denny 2024-12-13 13:04:04 -08:00 committed by mergify[bot]
parent cef65b2e93
commit 924780f2cc
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
UefiCpuPkg/CpuDxe/CpuMp.c
View File

@ -661,7 +661,12 @@ InitializeMpExceptionStackSwitchHandlers (
UINT8 *Buffer;
SwitchStackData = AllocateZeroPool (mNumberOfProcessors * sizeof (EXCEPTION_STACK_SWITCH_CONTEXT));
ASSERT (SwitchStackData != NULL);
if (SwitchStackData == NULL) {
DEBUG ((DEBUG_ERROR, "%a Failed to allocate buffer for SwitchStackData\n", __func__));
ASSERT (SwitchStackData != NULL);
return;
}
for (Index = 0; Index < mNumberOfProcessors; ++Index) {
//
// Because the procedure may runs multiple times, use the status EFI_NOT_STARTED
@ -689,8 +694,24 @@ InitializeMpExceptionStackSwitchHandlers (
}
if (BufferSize != 0) {
Buffer = AllocateRuntimeZeroPool (BufferSize);
ASSERT (Buffer != NULL);
// we are allocating the buffer that will hold the new GDT and IDT for the APs. These must be allocated below
// 4GB as they are used by protected mode code on the APs when they are started up after this point. If they are
// above 4GB, the APs will triple fault because the 32 bit code segment is invalid
Buffer = (UINT8 *)(UINTN)(BASE_4GB - 1);
Status = gBS->AllocatePages (
AllocateMaxAddress,
EfiRuntimeServicesData,
EFI_SIZE_TO_PAGES (BufferSize),
(EFI_PHYSICAL_ADDRESS *)&Buffer
);
if (EFI_ERROR (Status)) {
DEBUG ((DEBUG_ERROR, "Failed to allocate buffer for InitializeExceptionStackSwitchHandlers Status %r\n", Status));
ASSERT_EFI_ERROR (Status);
goto Exit;
}
ZeroMem (Buffer, BufferSize);
BufferSize = 0;
for (Index = 0; Index < mNumberOfProcessors; ++Index) {
if (SwitchStackData[Index].Status == EFI_BUFFER_TOO_SMALL) {
@ -717,6 +738,7 @@ InitializeMpExceptionStackSwitchHandlers (
}
}
Exit:
FreePool (SwitchStackData);
}