SecurityPkg-Opal(2): Enhance AHCI Bar MMIO region check.

[mirror_edk2.git] / SecurityPkg / Tcg / Opal / OpalPasswordSmm / OpalPasswordSmm.h

diff --git a/SecurityPkg/Tcg/Opal/OpalPasswordSmm/OpalPasswordSmm.h b/SecurityPkg/Tcg/Opal/OpalPasswordSmm/OpalPasswordSmm.h
index f365136918d979501851fd84cb74b1f7994d345b..ab31a6bfe7bc59b8b2dc0a34468047f8861961da 100644 (file)
--- a/SecurityPkg/Tcg/Opal/OpalPasswordSmm/OpalPasswordSmm.h
+++ b/SecurityPkg/Tcg/Opal/OpalPasswordSmm/OpalPasswordSmm.h
@@ -25,6 +25,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Protocol/SmmReadyToLock.h>\r
 #include <Protocol/SmmVariable.h>\r
 #include <Protocol/VariableLock.h>\r
+#include <Protocol/SmmEndOfDxe.h>\r
 #include <Protocol/StorageSecurityCommand.h>\r
 \r
 #include <Library/OpalPasswordSupportLib.h>\r
@@ -43,8 +44,6 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Library/UefiLib.h>\r
 #include <Library/S3BootScriptLib.h>\r
 #include <Library/DevicePathLib.h>\r
-#include <Library/SmmMemLib.h>\r
-\r
 #include <Library/DxeServicesTableLib.h>\r
 \r
 #include <IndustryStandard/Pci22.h>\r
@@ -71,6 +70,8 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 extern VOID                              *mBuffer;\r
 \r
+extern EFI_GCD_MEMORY_SPACE_DESCRIPTOR   *mGcdMemSpace;\r
+extern UINTN                             mNumberOfDescriptors;\r
 #pragma pack(1)\r
 \r
 typedef struct {\r