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

Ring3: Added IsUserImage field into LOADED_IMAGE_PRIVATE_DATA structure.

This commit is contained in:
Mikhail Krichanov 2024-01-15 10:53:36 +03:00
parent cf83eed077
commit f2a28c5a55
4 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
MdeModulePkg/Core/Dxe/DxeMain.h
View File

@ -223,7 +223,8 @@ typedef struct {
/// Status returned by LoadImage() service.
EFI_STATUS LoadImageStatus;
VOID *HiiData;
VOID *HiiData;
BOOLEAN IsUserImage;
} LOADED_IMAGE_PRIVATE_DATA;
#define LOADED_IMAGE_PRIVATE_DATA_FROM_THIS(a) \
@ -2727,12 +2728,14 @@ RemoveImageRecord (
@param[in] LoadedImage The loaded image protocol
@param[in] ImageOrigin Where File comes from.
@param[in] LoadedImageDevicePath The loaded image device path protocol
@param[out] IsUserImage Whether the loaded image is in user space.
**/
VOID
ProtectUefiImage (
IN EFI_LOADED_IMAGE_PROTOCOL *LoadedImage,
IN UINT8 ImageOrigin,
UEFI_IMAGE_LOADER_IMAGE_CONTEXT *ImageContext
IN EFI_LOADED_IMAGE_PROTOCOL *LoadedImage,
IN UINT8 ImageOrigin,
IN UEFI_IMAGE_LOADER_IMAGE_CONTEXT *ImageContext,
OUT BOOLEAN *IsUserImage
);
/**

2
MdeModulePkg/Core/Dxe/DxeMain/DxeMain.c
View File

@ -324,7 +324,7 @@ DxeMain (
CoreInitializeMemoryProtection ();
ProtectUefiImage (&mCurrentImage->Info, UefiImageOriginFv, &ImageContext);
ProtectUefiImage (&mCurrentImage->Info, UefiImageOriginFv, &ImageContext, &mCurrentImage->IsUserImage);
//
// Call constructor for all libraries

7
MdeModulePkg/Core/Dxe/Image/Image.c
View File

@ -64,7 +64,10 @@ LOADED_IMAGE_PRIVATE_DATA mCorePrivateImage = {
0, // Machine
NULL, // PeCoffEmu
NULL, // RuntimeData
NULL // LoadedImageDevicePath
NULL, // LoadedImageDevicePath
EFI_SUCCESS, // LoadImageStatus
NULL, // HiiData
FALSE // IsUserImage
};
//
// The field is define for Loading modules at fixed address feature to tracker the PEI code
@ -1435,7 +1438,7 @@ CoreLoadImageCommon (
}
Status = EFI_SUCCESS;
ProtectUefiImage (&Image->Info, ImageOrigin, &ImageContext);
ProtectUefiImage (&Image->Info, ImageOrigin, &ImageContext, &Image->IsUserImage);
RegisterMemoryProfileImage (
Image->LoadedImageDevicePath,

10
MdeModulePkg/Core/Dxe/Misc/MemoryProtection.c
View File

@ -168,12 +168,14 @@ IsMemoryProtectionSectionAligned (
@param[in] LoadedImage The loaded image protocol
@param[in] ImageOrigin Where File comes from.
@param[in] LoadedImageDevicePath The loaded image device path protocol
@param[out] IsUserImage Whether the loaded image is in user space.
**/
VOID
ProtectUefiImage (
IN EFI_LOADED_IMAGE_PROTOCOL *LoadedImage,
IN UINT8 ImageOrigin,
UEFI_IMAGE_LOADER_IMAGE_CONTEXT *ImageContext
IN EFI_LOADED_IMAGE_PROTOCOL *LoadedImage,
IN UINT8 ImageOrigin,
IN UEFI_IMAGE_LOADER_IMAGE_CONTEXT *ImageContext,
OUT BOOLEAN *IsUserImage
)
{
RETURN_STATUS PdbStatus;
@ -234,8 +236,10 @@ ProtectUefiImage (
//
if (AsciiStrStr (PdbPointer, "Ntfs") != NULL) {
SetUefiImageProtectionAttributes (ImageRecord, TRUE);
*IsUserImage = TRUE;
} else {
SetUefiImageProtectionAttributes (ImageRecord, FALSE);
*IsUserImage = FALSE;
}
}