ShellPkg: Handle pool allocation failure

FreePool() will receive NULL if AllocateZeroPool() fails.
So a check for NULL is needed.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Sergei Antonov <saproj@gmail.com>
reviewed-by: jaben carsey <jaben.carsey@gmail.com>

git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@14689 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
index fb9d4ec0be50978b806952b47dec0d4b8d06d710..2085f6c5218432d0c1eaef7d123c73196d1ae77a 100644 (file)
--- a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
+++ b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
@@ -84,9 +84,9 @@ FileHandleGetInfo (
     //\r
     // if we got an error free the memory and return NULL\r
     //\r
-    if (EFI_ERROR(Status)) {\r
+    if (EFI_ERROR(Status) && (FileInfo != NULL)) {\r
       FreePool(FileInfo);\r
-      return NULL;\r
+      FileInfo = NULL;\r
     }\r
   }\r
   return (FileInfo);\r