Fix the bug that could make gMemoryTypeInformation unavailable when gMemoryTypeInform...

[mirror_edk2.git] / EdkModulePkg / Core / Dxe / Gcd / gcd.c

diff --git a/EdkModulePkg/Core/Dxe/Gcd/gcd.c b/EdkModulePkg/Core/Dxe/Gcd/gcd.c
index 208c05a0157b3305d9f59c32dea9f11472aaea1d..0735bf00b9bb5e041b028a9836c6d5c56b6d0b80 100644 (file)
--- a/EdkModulePkg/Core/Dxe/Gcd/gcd.c
+++ b/EdkModulePkg/Core/Dxe/Gcd/gcd.c
@@ -2089,7 +2089,7 @@ Returns:
   if (GuidHob != NULL) {\r
     EfiMemoryTypeInformation = GET_GUID_HOB_DATA (GuidHob);\r
     DataSize                 = GET_GUID_HOB_DATA_SIZE (GuidHob);\r
-    if (EfiMemoryTypeInformation != NULL && DataSize > 0 && DataSize < EfiMaxMemoryType * sizeof (EFI_MEMORY_TYPE_INFORMATION)) {\r
+    if (EfiMemoryTypeInformation != NULL && DataSize > 0 && DataSize <= (EfiMaxMemoryType + 1) * sizeof (EFI_MEMORY_TYPE_INFORMATION)) {\r
       CopyMem (&gMemoryTypeInformation, EfiMemoryTypeInformation, DataSize);\r
     }\r
   }\r