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

2 files changed, 6 insertions, 0 deletions

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);

diff --git a/ShellPkg/Library/UefiShellLib/UefiShellLib.c b/ShellPkg/Library/UefiShellLib/UefiShellLib.c
index 598a2b479a..06e2386378 100644
--- a/ShellPkg/Library/UefiShellLib/UefiShellLib.c
+++ b/ShellPkg/Library/UefiShellLib/UefiShellLib.c
@@ -1933,6 +1933,9 @@ InternalCommandLineParse (
   // initialize the linked list

   //

   *CheckPackage = (LIST_ENTRY*)AllocateZeroPool(sizeof(LIST_ENTRY));

+  if (*CheckPackage == NULL) {

+    return EFI_OUT_OF_RESOURCES;

+  }

   InitializeListHead(*CheckPackage);

 

   //