MdeModulePkg DxeCore: Check pointer AllocInfoData before dereferencing.

[mirror_edk2.git] / MdeModulePkg / Core / Dxe / Mem / MemoryProfileRecord.c

diff --git a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
index dca0008f75aeec5f32b39fe7d2176b02f4d573ac..60971ba9f9be14801943575b4f507e8886dc9670 100644 (file)
--- a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
+++ b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
@@ -838,6 +838,7 @@ CoreUpdateProfileAllocate (
   if (EFI_ERROR (Status)) {\r
     return FALSE;\r
   }\r
+  ASSERT (AllocInfoData != NULL);\r
   AllocInfo = &AllocInfoData->AllocInfo;\r
   AllocInfoData->Signature      = MEMORY_PROFILE_ALLOC_INFO_SIGNATURE;\r
   AllocInfo->Header.Signature   = MEMORY_PROFILE_ALLOC_INFO_SIGNATURE;\r