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

StandaloneMmPkg/Core: Install Loaded Image Protocol for MM Core 

Retrieves the MM Core image info from Memory Allocation HOB reported
by MM IPL. Then install Loaded Image Protocol for MM Core with the
image info from HOB.

Cc: Ard Biesheuvel <ardb+tianocore@kernel.org>
Cc: Sami Mujawar <sami.mujawar@arm.com>
Cc: Ray Ni <ray.ni@intel.com>
Cc: Jiaxin Wu <jiaxin.wu@intel.com>
Signed-off-by: Wei6 Xu <wei6.xu@intel.com>
This commit is contained in:
Wei6 Xu 2024-05-07 01:59:46 +08:00 committed by mergify[bot]
parent a44830727a
commit dd775aa4d4
1 changed files with 85 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
StandaloneMmPkg/Core/StandaloneMmCore.c
View File

@ -317,6 +317,86 @@ MmEndOfDxeHandler (
return Status;
}
/**
Install LoadedImage protocol for MM Core.
**/
VOID
MmCoreInstallLoadedImage (
VOID
)
{
EFI_STATUS Status;
EFI_PHYSICAL_ADDRESS MmCoreImageBaseAddress;
UINT64 MmCoreImageLength;
EFI_PEI_HOB_POINTERS Hob;
EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;
EFI_HANDLE ImageHandle;
//
// Searching for Memory Allocation HOB
//
Hob.Raw = GetHobList ();
while ((Hob.Raw = GetNextHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, Hob.Raw)) != NULL) {
//
// Find MM Core HOB
//
if (CompareGuid (
&Hob.MemoryAllocationModule->MemoryAllocationHeader.Name,
&gEfiHobMemoryAllocModuleGuid
))
{
if (CompareGuid (&Hob.MemoryAllocationModule->ModuleName, &gEfiCallerIdGuid)) {
break;
}
}
Hob.Raw = GET_NEXT_HOB (Hob);
}
if (Hob.Raw == NULL) {
return;
}
MmCoreImageBaseAddress = Hob.MemoryAllocationModule->MemoryAllocationHeader.MemoryBaseAddress;
MmCoreImageLength = Hob.MemoryAllocationModule->MemoryAllocationHeader.MemoryLength;
//
// Allocate a Loaded Image Protocol in MM
//
LoadedImage = AllocatePool (sizeof (EFI_LOADED_IMAGE_PROTOCOL));
ASSERT (LoadedImage != NULL);
if (LoadedImage == NULL) {
return;
}
ZeroMem (LoadedImage, sizeof (EFI_LOADED_IMAGE_PROTOCOL));
//
// Fill in the remaining fields of the Loaded Image Protocol instance.
//
LoadedImage->Revision = EFI_LOADED_IMAGE_PROTOCOL_REVISION;
LoadedImage->ParentHandle = NULL;
LoadedImage->SystemTable = NULL;
LoadedImage->ImageBase = (VOID *)(UINTN)MmCoreImageBaseAddress;
LoadedImage->ImageSize = MmCoreImageLength;
LoadedImage->ImageCodeType = EfiRuntimeServicesCode;
LoadedImage->ImageDataType = EfiRuntimeServicesData;
//
// Create a new image handle in the MM handle database for the MM Core
//
ImageHandle = NULL;
Status = MmInstallProtocolInterface (
&ImageHandle,
&gEfiLoadedImageProtocolGuid,
EFI_NATIVE_INTERFACE,
LoadedImage
);
ASSERT_EFI_ERROR (Status);
}
/**
The main entry point to MM Foundation.
@ -657,6 +737,11 @@ StandaloneMmMain (
DEBUG ((DEBUG_INFO, "MmiHandlerRegister - GUID %g - Status %d\n", mMmCoreMmiHandlers[Index].HandlerType, Status));
}
//
// Install Loaded Image Protocol form MM Core
//
MmCoreInstallLoadedImage ();
DEBUG ((DEBUG_INFO, "MmMain Done!\n"));
return EFI_SUCCESS;