Ring3: Defined CpuGetMemoryAttributes(), DisableSMAP(), EnableSMAP() for AARCH64.

ArmPkg/Library/CpuArchLib/AArch64/Mmu.c

@@ -95,6 +95,12 @@ PageAttributeToGcdAttribute (
     GcdAttributes |= EFI_MEMORY_XP;
   }
 
+  if (((PageAttributes & TT_AP_MASK) == TT_AP_RW_RW) ||
+      ((PageAttributes & TT_AP_MASK) == TT_AP_RO_RO))
+  {
+    GcdAttributes |= EFI_MEMORY_USER;
+  }
+
   return GcdAttributes;
 }
 

ArmPkg/Library/CpuArchLib/CpuDxe.c

@@ -218,6 +218,7 @@ EFI_CPU_ARCH_PROTOCOL  mCpu       = {
   CpuSetMemoryAttributes,
   0,          // NumberOfTimers
   2048,       // DmaBufferAlignment
+  CpuGetMemoryAttributes
 };
 
 STATIC

ArmPkg/Library/CpuArchLib/CpuDxe.h

@@ -96,6 +96,14 @@ CpuSetMemoryAttributes (
   IN UINT64                 Attributes
   );
 
+EFI_STATUS
+EFIAPI
+CpuGetMemoryAttributes (
+  IN  EFI_CPU_ARCH_PROTOCOL  *This,
+  IN  EFI_PHYSICAL_ADDRESS   Address,
+  OUT UINT64                 *Attributes
+  );
+
 EFI_STATUS
 InitializeExceptions (
   IN EFI_CPU_ARCH_PROTOCOL  *Cpu

ArmPkg/Library/CpuArchLib/CpuMmuCommon.c

@@ -222,3 +222,28 @@ CpuSetMemoryAttributes (
     return EFI_SUCCESS;
   }
 }
+
+EFI_STATUS
+EFIAPI
+CpuGetMemoryAttributes (
+  IN  EFI_CPU_ARCH_PROTOCOL  *This,
+  IN  EFI_PHYSICAL_ADDRESS   Address,
+  OUT UINT64                 *Attributes
+  )
+{
+  EFI_STATUS  Status;
+  UINTN       RegionBaseAddress;
+  UINTN       RegionLength;
+  UINTN       RegionArmAttributes;
+
+  RegionBaseAddress = Address;
+  Status            = GetMemoryRegion (&RegionBaseAddress, &RegionLength, &RegionArmAttributes);
+
+  if (EFI_ERROR (Status)) {
+    return EFI_NOT_FOUND;
+  }
+
+  *Attributes = RegionAttributeToGcdAttribute (RegionArmAttributes);
+
+  return Status;
+}

MdeModulePkg/Core/Dxe/SysCall/AARCH64/CoreBootServices.S

@@ -7,26 +7,6 @@
 
 #include <AsmMacroIoLibV8.h>
 
-//------------------------------------------------------------------------------
-// VOID
-// EFIAPI
-// DisableSMAP (
-//   VOID
-//   );
-//------------------------------------------------------------------------------
-ASM_FUNC(DisableSMAP)
-    ret
-
-//------------------------------------------------------------------------------
-// VOID
-// EFIAPI
-// EnableSMAP (
-//   VOID
-//   );
-//------------------------------------------------------------------------------
-ASM_FUNC(EnableSMAP)
-    ret
-
 //------------------------------------------------------------------------------
 // EFI_STATUS
 // EFIAPI

MdeModulePkg/Core/Dxe/SysCall/AARCH64/InitializeMsr.c

@@ -41,3 +41,21 @@ InitializeMsr (
     ASSERT (FALSE);
   }
 }
+
+VOID
+EFIAPI
+DisableSMAP (
+  VOID
+  )
+{
+  ArmClearPan ();
+}
+
+VOID
+EFIAPI
+EnableSMAP (
+  VOID
+  )
+{
+  ArmSetPan ();
+}