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

Make sure the base address of stack base and base address of hob start is in page alignment in InstallPeiMemory() interface. 

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@4646 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
klu2 2008-01-31 08:41:14 +00:00
parent 6be065d4be
commit 1eabd8e747
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
MdeModulePkg/Core/Pei/Memory/MemoryServices.c
View File

@ -115,6 +115,10 @@ Returns:
UINT64 EfiFreeMemorySize; UINT64 EfiFreeMemorySize;
EFI_PHYSICAL_ADDRESS PhysicalAddressOfOldHob; EFI_PHYSICAL_ADDRESS PhysicalAddressOfOldHob;
if (MemoryLength > (MAX_ADDRESS - MemoryBegin + 1))
return EFI_INVALID_PARAMETER;
DEBUG ((EFI_D_INFO, "PeiInstallPeiMemory MemoryBegin 0x%LX, MemoryLength 0x%LX\n", MemoryBegin, MemoryLength)); DEBUG ((EFI_D_INFO, "PeiInstallPeiMemory MemoryBegin 0x%LX, MemoryLength 0x%LX\n", MemoryBegin, MemoryLength));
PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices); PrivateData = PEI_CORE_INSTANCE_FROM_PS_THIS (PeiServices);
@ -122,9 +126,12 @@ Returns:
PrivateData->SwitchStackSignal = TRUE; PrivateData->SwitchStackSignal = TRUE;
PrivateData->PeiMemoryInstalled = TRUE; PrivateData->PeiMemoryInstalled = TRUE;
PrivateData->StackBase = MemoryBegin; //
// Ensure the stack base is in page alignment
//
PrivateData->StackBase = ((UINTN)MemoryBegin + (EFI_PAGE_SIZE - 1)) & EFI_PAGE_SIZE;
PeiStackSize = RShiftU64 (MemoryLength, 1); PeiStackSize = (RShiftU64 (MemoryLength, 1) + (EFI_PAGE_SIZE - 1)) & EFI_PAGE_SIZE;
if (PEI_STACK_SIZE > PeiStackSize) { if (PEI_STACK_SIZE > PeiStackSize) {
PrivateData->StackSize = PeiStackSize; PrivateData->StackSize = PeiStackSize;
} else { } else {
@ -133,7 +140,7 @@ Returns:
OldHandOffHob = PrivateData->HobList.HandoffInformationTable; OldHandOffHob = PrivateData->HobList.HandoffInformationTable;
PrivateData->HobList.Raw = (VOID *)((UINTN)(MemoryBegin + PrivateData->StackSize)); PrivateData->HobList.Raw = (VOID *)((UINTN)(PrivateData->StackBase + PrivateData->StackSize));
NewHandOffHob = PrivateData->HobList.HandoffInformationTable; NewHandOffHob = PrivateData->HobList.HandoffInformationTable;
PhysicalAddressOfOldHob = (EFI_PHYSICAL_ADDRESS) (UINTN) OldHandOffHob; PhysicalAddressOfOldHob = (EFI_PHYSICAL_ADDRESS) (UINTN) OldHandOffHob;