OvmfPkg/IncompatiblePciDeviceSupportDxe: Ignore OptionRom in Sev guest

[mirror_edk2.git] / OvmfPkg / IncompatiblePciDeviceSupportDxe / IncompatiblePciDeviceSupport.c

diff --git a/OvmfPkg/IncompatiblePciDeviceSupportDxe/IncompatiblePciDeviceSupport.c b/OvmfPkg/IncompatiblePciDeviceSupportDxe/IncompatiblePciDeviceSupport.c
index 2d385d26ef2837a584a6059262d6d6b98ce76113..686d85633e7c253f39a4e19906d0786bd06a0add 100644 (file)
--- a/OvmfPkg/IncompatiblePciDeviceSupportDxe/IncompatiblePciDeviceSupport.c
+++ b/OvmfPkg/IncompatiblePciDeviceSupportDxe/IncompatiblePciDeviceSupport.c
@@ -18,6 +18,7 @@
 #include <Library/MemoryAllocationLib.h>\r
 #include <Library/PcdLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
+#include <Library/CcProbeLib.h>\r
 \r
 #include <Protocol/IncompatiblePciDeviceSupport.h>\r
 #include <Protocol/LegacyBios.h>\r
@@ -264,7 +265,7 @@ CheckDevice (
   //\r
   // In Td guest OptionRom is not allowed.\r
   //\r
-  if (TdIsEnabled ()) {\r
+  if (CcProbe ()) {\r
     Length += sizeof mOptionRomConfiguration;\r
   }\r
 \r
@@ -286,7 +287,7 @@ CheckDevice (
   CopyMem (Ptr, &mMmio64Configuration, sizeof mMmio64Configuration);\r
   Length = sizeof mMmio64Configuration;\r
 \r
-  if (TdIsEnabled ()) {\r
+  if (CcProbe ()) {\r
     CopyMem (Ptr + Length, &mOptionRomConfiguration, sizeof mOptionRomConfiguration);\r
     Length += sizeof mOptionRomConfiguration;\r
   }\r