Ring3: Small fix.

acidanthera · Feb 22, 2024 · 031adc2 · 031adc2
1 parent b9aa237
commit 031adc2
Showing 1 changed file with 17 additions and 13 deletions.
diff --git a/MdeModulePkg/Core/Dxe/SysCall/SupportedProtocols.c b/MdeModulePkg/Core/Dxe/SysCall/SupportedProtocols.c
@@ -44,22 +44,26 @@ GoToRing3 (
   VA_END (Marker);
   EnableSMAP ();
 
-  //
-  // Necessary fix for ProcessLibraryConstructorList() -> DxeCcProbeLibConstructor()
-  //
-  SetUefiImageMemoryAttributes (
-    FixedPcdGet32 (PcdOvmfWorkAreaBase),
-    FixedPcdGet32 (PcdOvmfWorkAreaSize),
-    EFI_MEMORY_XP | EFI_MEMORY_USER
-    );
+  if (Number == 2) {
+    //
+    // Necessary fix for ProcessLibraryConstructorList() -> DxeCcProbeLibConstructor()
+    //
+    SetUefiImageMemoryAttributes (
+      FixedPcdGet32 (PcdOvmfWorkAreaBase),
+      FixedPcdGet32 (PcdOvmfWorkAreaSize),
+      EFI_MEMORY_XP | EFI_MEMORY_USER
+      );
+  }
 
   Status = CallRing3 (Input);
 
-  SetUefiImageMemoryAttributes (
-    FixedPcdGet32 (PcdOvmfWorkAreaBase),
-    FixedPcdGet32 (PcdOvmfWorkAreaSize),
-    EFI_MEMORY_XP
-    );
+  if (Number == 2) {
+    SetUefiImageMemoryAttributes (
+      FixedPcdGet32 (PcdOvmfWorkAreaBase),
+      FixedPcdGet32 (PcdOvmfWorkAreaSize),
+      EFI_MEMORY_XP
+      );
+  }
 
   return Status;
 }