MdePkg: Probe Cc guest in BaseIoLibIntrinsicSev

[mirror_edk2.git] / MdePkg / Library / BaseIoLibIntrinsic / IoLibInternalTdx.c

diff --git a/MdePkg/Library/BaseIoLibIntrinsic/IoLibInternalTdx.c b/MdePkg/Library/BaseIoLibIntrinsic/IoLibInternalTdx.c
index 1e539dbfbbade1de7c9ef583483ff11d47a12acc..ec837f5eb03e78ae64ce180280196bff656d6ce2 100644 (file)
--- a/MdePkg/Library/BaseIoLibIntrinsic/IoLibInternalTdx.c
+++ b/MdePkg/Library/BaseIoLibIntrinsic/IoLibInternalTdx.c
@@ -10,6 +10,7 @@
 #include <Include/IndustryStandard/Tdx.h>\r
 #include <Library/TdxLib.h>\r
 #include <Register/Intel/Cpuid.h>\r
+#include <Library/CcProbeLib.h>\r
 #include "IoLibTdx.h"\r
 \r
 // Size of TDVMCALL Access, including IO and MMIO\r
@@ -22,9 +23,6 @@
 #define TDVMCALL_ACCESS_READ   0\r
 #define TDVMCALL_ACCESS_WRITE  1\r
 \r
-BOOLEAN  mTdxEnabled = FALSE;\r
-BOOLEAN  mTdxProbed  = FALSE;\r
-\r
 /**\r
   Check if it is Tdx guest.\r
 \r
@@ -38,14 +36,7 @@ IsTdxGuest (
   VOID\r
   )\r
 {\r
-  if (mTdxProbed) {\r
-    return mTdxEnabled;\r
-  }\r
-\r
-  mTdxEnabled = TdIsEnabled ();\r
-  mTdxProbed  = TRUE;\r
-\r
-  return mTdxEnabled;\r
+  return CcProbe () == CcGuestTypeIntelTdx;\r
 }\r
 \r
 /**\r