ShellPkg: Add checking for memory allocation and pointer returns from functions.

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

diff --git a/ShellPkg/Library/UefiShellLevel2CommandsLib/Map.c b/ShellPkg/Library/UefiShellLevel2CommandsLib/Map.c
index 358f63d8db719208389cb51e8cbdc5d5add01156..a1f9448df92854fb63aabea2549fc96c798f5e3b 100644 (file)
--- a/ShellPkg/Library/UefiShellLevel2CommandsLib/Map.c
+++ b/ShellPkg/Library/UefiShellLevel2CommandsLib/Map.c
@@ -414,6 +414,9 @@ PerformSingleMappingDisplay(
 \r
   CurrentName = NULL;\r
   CurrentName = StrnCatGrow(&CurrentName, 0, MapList, 0);\r
+  if (CurrentName == NULL) {\r
+    return (EFI_OUT_OF_RESOURCES);\r
+  }\r
   TempSpot = StrStr(CurrentName, L";");\r
   if (TempSpot != NULL) {\r
     *TempSpot = CHAR_NULL;\r