From: Hao Wu <hao.a.wu@intel.com>
Date: Mon, 4 Jul 2016 04:38:28 +0000 (+0800)
Subject: MdeModulePkg MemoryProfile: Add ASSERTs to make sure pointers are not NULL
X-Git-Tag: edk2-stable201903~6406
X-Git-Url: https://git.proxmox.com/?a=commitdiff_plain;h=9cda0af7ecfcfc5446da13e63b1930a0d83c0f3f;p=mirror_edk2.git

MdeModulePkg MemoryProfile: Add ASSERTs to make sure pointers are not NULL

Code logic ensures that both pointers 'DriverInfoData' and 'AllocInfoData'
will not be NULL when they are used.

Add ASSERTs as warning for cases that will not happen.

Cc: Star Zeng <star.zeng@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Hao Wu <hao.a.wu@intel.com>
Reviewed-by: Star Zeng <star.zeng@intel.com>
---

diff --git a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
index 14c4959f9f..30c0df46d6 100644
--- a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
+++ b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c
@@ -1283,6 +1283,9 @@ CoreUpdateProfileFree (
       }
     }
 
+    ASSERT (DriverInfoData != NULL);
+    ASSERT (AllocInfoData != NULL);
+
     Found = TRUE;
 
     Context = &ContextData->Context;
diff --git a/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c b/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c
index 281e382d5c..ac832ce38c 100644
--- a/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c
+++ b/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c
@@ -1403,6 +1403,9 @@ SmmCoreUpdateProfileFree (
       }
     }
 
+    ASSERT (DriverInfoData != NULL);
+    ASSERT (AllocInfoData != NULL);
+
     Found = TRUE;
 
     Context = &ContextData->Context;