OvmfPkg/IoMmuDxe: abort harder on memory encryption mask failures

[mirror_edk2.git] / OvmfPkg / IoMmuDxe / AmdSevIoMmu.c

diff --git a/OvmfPkg/IoMmuDxe/AmdSevIoMmu.c b/OvmfPkg/IoMmuDxe/AmdSevIoMmu.c
index 1dafe0df1127259281427a031281d82789f59e6c..452d5c47753bb9b013d59ab6dcb752d43b6491c0 100644 (file)
--- a/OvmfPkg/IoMmuDxe/AmdSevIoMmu.c
+++ b/OvmfPkg/IoMmuDxe/AmdSevIoMmu.c
@@ -223,7 +223,10 @@ IoMmuMap (
              MapInfo->NumberOfPages,\r
              TRUE\r
              );\r
-  ASSERT_EFI_ERROR(Status);\r
+  ASSERT_EFI_ERROR (Status);\r
+  if (EFI_ERROR (Status)) {\r
+    CpuDeadLoop ();\r
+  }\r
 \r
   //\r
   // If this is a read operation from the Bus Master's point of view,\r
@@ -365,7 +368,10 @@ IoMmuUnmap (
              MapInfo->NumberOfPages,\r
              TRUE\r
              );\r
-  ASSERT_EFI_ERROR(Status);\r
+  ASSERT_EFI_ERROR (Status);\r
+  if (EFI_ERROR (Status)) {\r
+    CpuDeadLoop ();\r
+  }\r
 \r
   //\r
   // For BusMasterCommonBuffer[64] operations, copy the stashed data to the\r