Verify memory allocations were successful.

[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Cp.c

diff --git a/ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c b/ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c
index 94161cfecb3bf147b61e963018456ae5d38c55de..9344c331e58eb0d70c5ac6e1136b42848a7b33ac 100644 (file)
--- a/ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c
+++ b/ShellPkg/Library/UefiShellLevel2CommandsLib/Cp.c
@@ -287,6 +287,13 @@ ValidateAndCopyFiles(
   HiiResultOk = HiiGetString (gShellLevel2HiiHandle, STRING_TOKEN (STR_GEN_RES_OK), NULL);\r
   DestPath    = AllocatePool(PathLen);\r
 \r
+  if (HiiOutput == NULL || HiiOutput == NULL || HiiResultOk == NULL) {\r
+    SHELL_FREE_NON_NULL(DestPath);\r
+    SHELL_FREE_NON_NULL(HiiOutput);\r
+    SHELL_FREE_NON_NULL(HiiResultOk);\r
+    return (SHELL_OUT_OF_RESOURCES);\r
+  }\r
+\r
   //\r
   // Go through the list of files to copy...\r
   //\r