From bab57374fb8237add3c51b37b81daa7c631660ef Mon Sep 17 00:00:00 2001 From: Hao Wu Date: Thu, 7 Jul 2016 08:49:38 +0800 Subject: MdeModulePkg MemoryProfile: ASSERT to ensure 'DriverInfoData' is not NULL Code logic ensures that the pointer 'DriverInfoData' will not be NULL when it is used. Add ASSERT as warning for case that will not happen. Cc: Star Zeng Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Hao Wu Reviewed-by: Star Zeng (cherry picked from commit d8162f5b3283a06a6dc4e2e05cd0c45fc4358eb0) --- MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c | 1 + MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c | 1 + 2 files changed, 2 insertions(+) (limited to 'MdeModulePkg/Core') diff --git a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c index 1f17947808..7141373fd5 100644 --- a/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c +++ b/MdeModulePkg/Core/Dxe/Mem/MemoryProfileRecord.c @@ -319,6 +319,7 @@ BuildDriverInfo ( if (EFI_ERROR (Status)) { return NULL; } + ASSERT (DriverInfoData != NULL); ZeroMem (DriverInfoData, sizeof (*DriverInfoData)); diff --git a/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c b/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c index 57b34a018d..51fa7f947d 100644 --- a/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c +++ b/MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c @@ -272,6 +272,7 @@ BuildDriverInfo ( if (EFI_ERROR (Status)) { return NULL; } + ASSERT (DriverInfoData != NULL); ZeroMem (DriverInfoData, sizeof (*DriverInfoData)); -- cgit v1.2.3