tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-09-23 17:57:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

Ring3: Added EFI_MEMORY_USER attribute.

Browse Source
This commit is contained in:
Mikhail Krichanov 2023-12-25 16:08:53 +03:00
parent 138ecce134
commit 8c069a27e5
4 changed files with 38 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -94,7 +94,8 @@ SetUefiImageMemoryAttributes (
**/ **/
VOID VOID
SetUefiImageProtectionAttributes ( SetUefiImageProtectionAttributes (
IN UEFI_IMAGE_RECORD *ImageRecord IN UEFI_IMAGE_RECORD *ImageRecord,
IN BOOLEAN IsUser
) )
{ {
UEFI_IMAGE_RECORD_SEGMENT *ImageRecordSegment; UEFI_IMAGE_RECORD_SEGMENT *ImageRecordSegment;
@ -107,7 +108,7 @@ SetUefiImageProtectionAttributes (
SetUefiImageMemoryAttributes ( SetUefiImageMemoryAttributes (
SectionAddress, SectionAddress,
ImageRecordSegment->Size, ImageRecordSegment->Size,
ImageRecordSegment->Attributes IsUser ? ImageRecordSegment->Attributes | (UINT32)EFI_MEMORY_USER : ImageRecordSegment->Attributes
); );
SectionAddress += ImageRecordSegment->Size; SectionAddress += ImageRecordSegment->Size;
@ -231,7 +232,11 @@ ProtectUefiImage (
// //
// CPU ARCH present. Update memory attribute directly. // CPU ARCH present. Update memory attribute directly.
// //
SetUefiImageProtectionAttributes (ImageRecord); if (AsciiStrStr (PdbPointer, "Ntfs") != NULL) {
SetUefiImageProtectionAttributes (ImageRecord, TRUE);
} else {
SetUefiImageProtectionAttributes (ImageRecord, FALSE);
}
} }
Finish: Finish:
@ -663,7 +668,7 @@ MemoryProtectionCpuArchProtocolNotify (
// //
// CPU ARCH present. Update memory attribute directly. // CPU ARCH present. Update memory attribute directly.
// //
SetUefiImageProtectionAttributes (ImageRecord); SetUefiImageProtectionAttributes (ImageRecord, FALSE);
} }
Done: Done:

8
MdePkg/Include/Uefi/UefiSpec.h
View File

@ -105,6 +105,12 @@ typedef enum {
// //
#define EFI_MEMORY_CPU_CRYPTO 0x0000000000080000ULL #define EFI_MEMORY_CPU_CRYPTO 0x0000000000080000ULL
//
// If this flag is set, the memory region contains user code or data.
// If this flag is clear, the memory region contains supervisor code or data.
//
#define EFI_MEMORY_USER 0x0000000000100000ULL
// //
// Runtime memory attribute // Runtime memory attribute
// //
@ -130,7 +136,7 @@ typedef enum {
// //
#define EFI_CACHE_ATTRIBUTE_MASK (EFI_MEMORY_UC | EFI_MEMORY_WC | EFI_MEMORY_WT | EFI_MEMORY_WB | EFI_MEMORY_UCE | EFI_MEMORY_WP) #define EFI_CACHE_ATTRIBUTE_MASK (EFI_MEMORY_UC | EFI_MEMORY_WC | EFI_MEMORY_WT | EFI_MEMORY_WB | EFI_MEMORY_UCE | EFI_MEMORY_WP)
#define EFI_MEMORY_ACCESS_MASK (EFI_MEMORY_RP | EFI_MEMORY_XP | EFI_MEMORY_RO) #define EFI_MEMORY_ACCESS_MASK (EFI_MEMORY_RP | EFI_MEMORY_XP | EFI_MEMORY_RO)
#define EFI_MEMORY_ATTRIBUTE_MASK (EFI_MEMORY_ACCESS_MASK | EFI_MEMORY_SP | EFI_MEMORY_CPU_CRYPTO) #define EFI_MEMORY_ATTRIBUTE_MASK (EFI_MEMORY_ACCESS_MASK | EFI_MEMORY_SP | EFI_MEMORY_CPU_CRYPTO | EFI_MEMORY_USER)
/// ///
/// Memory descriptor version number. /// Memory descriptor version number.

21
UefiCpuPkg/Library/CpuArchLib/CpuPageTable.c
View File

@ -467,6 +467,27 @@ ConvertPageEntryAttribute (
} }
} }
if ((Attributes & EFI_MEMORY_USER) != 0) {
switch (PageAction) {
case PageActionAssign:
case PageActionSet:
NewPageEntry |= IA32_PG_U;
break;
case PageActionClear:
NewPageEntry &= ~(UINT64)IA32_PG_U;
break;
}
} else {
switch (PageAction) {
case PageActionAssign:
NewPageEntry &= ~(UINT64)IA32_PG_U;
break;
case PageActionSet:
case PageActionClear:
break;
}
}
GetPagingDetails (&PagingContext->ContextData, NULL, &PageAttributes); GetPagingDetails (&PagingContext->ContextData, NULL, &PageAttributes);
if ((*PageAttributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_XD_ACTIVATED) != 0) { if ((*PageAttributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_XD_ACTIVATED) != 0) {