From 02a758cb0b5bc8775d95e0a52acc483f850124a1 Mon Sep 17 00:00:00 2001
From: sfu5 <sfu5@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Sat, 8 Oct 2011 02:55:30 +0000
Subject: Add pointer check for NULL before dereference it.

Signed-off-by: sfu5
Reviewed-by: xdu2
Reviewed-by: ydong10


git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12514 6f19259b-4bc3-4df7-8a09-765794883524
---
 ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c')

diff --git a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
index a4820c9970..116c78f38b 100644
--- a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
+++ b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
@@ -765,6 +765,9 @@ StrnCatGrowLeft (
   } else {
     *Destination = AllocateZeroPool(Count+sizeof(CHAR16));
   }
+  if (*Destination == NULL) {
+    return NULL;
+  }
 
   CopySize = StrSize(*Destination);
   CopyMem((*Destination)+((Count-2)/sizeof(CHAR16)), *Destination, CopySize);
-- 
cgit v1.2.3