From 523f48e771b1088c6e2cfddc5c56cde1ef9a2ab0 Mon Sep 17 00:00:00 2001 From: Eric Dong Date: Thu, 26 Jun 2014 03:24:25 +0000 Subject: Refine code to make it more safely. Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Eric Dong Reviewed-by: Feng Tian git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15598 6f19259b-4bc3-4df7-8a09-765794883524 --- MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c') diff --git a/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c b/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c index 3c55a785fc..0f0da166ad 100644 --- a/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c +++ b/MdeModulePkg/Universal/PlatformDriOverrideDxe/PlatDriOverrideDxe.c @@ -123,10 +123,11 @@ DevicePathToStr ( TRUE ); if (Text == NULL) { - return AllocateCopyPool (sizeof (L"?"), L"?"); - } else { - return Text; + Text = AllocateCopyPool (sizeof (L"?"), L"?"); + ASSERT (Text != NULL); } + + return Text; } /** -- cgit v1.2.3