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

signed-off-by: jcarsey
reviewed-by: geekboy15a

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12540 6f19259b-4bc3-4df7-8a09-765794883524

ShellPkg/Application/Shell/Shell.c

@@ -857,6 +857,9 @@ DoStartupScript(
     FileStringPath = NULL;
     NewSize = 0;
     FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, MapName, 0);
+    if (FileStringPath == NULL) {
+      Status = EFI_OUT_OF_RESOURCES;
+    } else {
       TempSpot = StrStr(FileStringPath, L";");
       if (TempSpot != NULL) {
         *TempSpot = CHAR_NULL;
@@ -867,6 +870,7 @@ DoStartupScript(
       Status = ShellInfoObject.NewEfiShellProtocol->OpenFileByName(FileStringPath, &FileHandle, EFI_FILE_MODE_READ);
       FreePool(FileStringPath);
     }
+  }
   if (EFI_ERROR(Status)) {
     NamePath = FileDevicePath (NULL, mStartupScript);
     NewPath = AppendDevicePathNode (ImagePath, NamePath);
@@ -1215,6 +1219,12 @@ RunSplitCommand(
 
   NextCommandLine = StrnCatGrow(&NextCommandLine, &Size1, StrStr(CmdLine, L"|")+1, 0);
   OurCommandLine  = StrnCatGrow(&OurCommandLine , &Size2, CmdLine                , StrStr(CmdLine, L"|") - CmdLine);
+
+  if (NextCommandLine == NULL || OurCommandLine == NULL) {
+    SHELL_FREE_NON_NULL(OurCommandLine);
+    SHELL_FREE_NON_NULL(NextCommandLine);
+    return (EFI_OUT_OF_RESOURCES);
+  }
   if (NextCommandLine[0] != CHAR_NULL &&
       NextCommandLine[0] == L'a' &&
       NextCommandLine[1] == L' '
@@ -1334,6 +1344,9 @@ RunCommand(
   Split               = NULL;
 
   CleanOriginal = StrnCatGrow(&CleanOriginal, NULL, CmdLine, 0);
+  if (CleanOriginal == NULL) {
+    return (EFI_OUT_OF_RESOURCES);
+  }
   while (CleanOriginal[StrLen(CleanOriginal)-1] == L' ') {
     CleanOriginal[StrLen(CleanOriginal)-1] = CHAR_NULL;
   }

ShellPkg/Application/Shell/ShellManParser.c

@@ -180,6 +180,10 @@ ManBufferFindSections(
         TempString2 = MIN(TempString2, StrStr(CurrentLocation, L"\n"));
         ASSERT(TempString == NULL);
         TempString = StrnCatGrow(&TempString, NULL, CurrentLocation, TempString2==NULL?0:TempString2 - CurrentLocation);
+        if (TempString == NULL) {
+          Status = EFI_OUT_OF_RESOURCES;
+          break;
+        }
         SectionName = TempString;
         SectionLen = StrLen(SectionName);
         SectionName = StrStr(Sections, SectionName);
@@ -197,12 +201,24 @@ ManBufferFindSections(
         TempString2 = MIN(TempString2, StrStr(CurrentLocation, L"\n"));
         ASSERT(TempString == NULL);
         TempString = StrnCatGrow(&TempString, NULL, CurrentLocation, TempString2==NULL?0:TempString2 - CurrentLocation);
+        if (TempString == NULL) {
+          Status = EFI_OUT_OF_RESOURCES;
+          break;
+        }
         //
         // copy and save the current line.
         //
         ASSERT((*HelpText == NULL && *HelpSize == 0) || (*HelpText != NULL));
         StrnCatGrow (HelpText, HelpSize, TempString, 0);
+        if (HelpText == NULL) {
+          Status = EFI_OUT_OF_RESOURCES;
+          break;
+        }
         StrnCatGrow (HelpText, HelpSize, L"\r\n", 0);
+        if (HelpText == NULL) {
+          Status = EFI_OUT_OF_RESOURCES;
+          break;
+        }
       }
     }
     SHELL_FREE_NON_NULL(TempString);

ShellPkg/Application/Shell/ShellParametersProtocol.c

@@ -577,6 +577,9 @@ UpdateStdInStdOutStdErr(
   }
 
   CommandLineCopy = StrnCatGrow(&CommandLineCopy, NULL, NewCommandLine, 0);
+  if (CommandLineCopy == NULL) {
+    return (EFI_OUT_OF_RESOURCES);
+  }
   Status          = EFI_SUCCESS;
   Split           = NULL;
   FirstLocation   = CommandLineCopy + StrLen(CommandLineCopy);
@@ -1030,11 +1033,13 @@ UpdateStdInStdOutStdErr(
       //
       if (!EFI_ERROR(Status) && StdInVarName != NULL) {
         TempHandle = CreateFileInterfaceEnv(StdInVarName);
+        if (TempHandle == NULL) {
+          Status = EFI_OUT_OF_RESOURCES;
+        } else {
           if (!InUnicode) {
             TempHandle = CreateFileInterfaceFile(TempHandle, FALSE);
           }
           Size = 0;
-        ASSERT(TempHandle != NULL);
           if (((EFI_FILE_PROTOCOL*)TempHandle)->Read(TempHandle, &Size, NULL) != EFI_BUFFER_TOO_SMALL) {
             Status = EFI_INVALID_PARAMETER;
           } else {
@@ -1042,6 +1047,7 @@ UpdateStdInStdOutStdErr(
             gST->ConIn = CreateSimpleTextInOnFile(TempHandle, &gST->ConsoleInHandle);
           }
         }
+      }
 
       //
       // StdIn from a file

ShellPkg/Application/Shell/ShellProtocol.c

@@ -488,10 +488,12 @@ EfiShellGetFilePathFromDevicePath(
   This function converts a file system style name to a device path, by replacing any
   mapping references to the associated device path.
 
-  @param Path                   the pointer to the path
+  @param[in] Path               The pointer to the path.
 
-  @return all                   The pointer of the file path. The file path is callee
+  @return                       The pointer of the file path. The file path is callee
                                 allocated and should be freed by the caller.
+  @retval NULL                  The path could not be found.
+  @retval NULL                  There was not enough available memory.
 **/
 EFI_DEVICE_PATH_PROTOCOL *
 EFIAPI
@@ -525,7 +527,9 @@ EfiShellGetDevicePathFromFilePath(
     Size = StrSize(Cwd);
     Size += StrSize(Path);
     NewPath = AllocateZeroPool(Size);
-    ASSERT(NewPath != NULL);
+    if (NewPath == NULL) {
+      return (NULL);
+    }
     StrCpy(NewPath, Cwd);
     if (*Path == L'\\') {
       Path++;
@@ -543,8 +547,7 @@ EfiShellGetDevicePathFromFilePath(
   //
   ASSERT((MapName == NULL && Size == 0) || (MapName != NULL));
   MapName = StrnCatGrow(&MapName, &Size, Path, (StrStr(Path, L":")-Path+1));
-  if (MapName[StrLen(MapName)-1] != L':') {
+  if (MapName == NULL || MapName[StrLen(MapName)-1] != L':') {
-    ASSERT(FALSE);
     return (NULL);
   }
 
@@ -564,7 +567,6 @@ EfiShellGetDevicePathFromFilePath(
   //
   DevicePathCopyForFree = DevicePathCopy = DuplicateDevicePath(DevicePath);
   if (DevicePathCopy == NULL) {
-    ASSERT(FALSE);
     FreePool(MapName);
     return (NULL);
   }
@@ -1888,6 +1890,9 @@ EfiShellFindFilesInDir(
     TempString        = NULL;
     Size              = 0;
     TempString        = StrnCatGrow(&TempString, &Size, ShellFileHandleGetPath(FileDirHandle), 0);
+    if (TempString == NULL) {
+      return (EFI_OUT_OF_RESOURCES);
+    }
     TempSpot          = StrStr(TempString, L";");
 
     if (TempSpot != NULL) {
@@ -1895,6 +1900,9 @@ EfiShellFindFilesInDir(
     }
 
     TempString        = StrnCatGrow(&TempString, &Size, BasePath, 0);
+    if (TempString == NULL) {
+      return (EFI_OUT_OF_RESOURCES);
+    }
     BasePath          = TempString;
   }
 
@@ -2239,6 +2247,9 @@ EfiShellFindFiles(
 
   ASSERT(MapName == NULL);
   MapName = StrnCatGrow(&MapName, NULL, PatternCopy, Count);
+  if (MapName == NULL) {
+    Status = EFI_OUT_OF_RESOURCES;
+  }
 
   if (!EFI_ERROR(Status)) {
     RootDevicePath = EfiShellGetDevicePathFromFilePath(PatternCopy);
@@ -2759,7 +2770,7 @@ EfiShellSetCurDir(
       MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, L"\\", 0);
       ASSERT((MapListItem->CurrentDirectoryPath == NULL && Size == 0) || (MapListItem->CurrentDirectoryPath != NULL));
       MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, DirectoryName, 0);
-      if (MapListItem->CurrentDirectoryPath[StrLen(MapListItem->CurrentDirectoryPath)-1] != L'\\') {
+      if (MapListItem->CurrentDirectoryPath != NULL && MapListItem->CurrentDirectoryPath[StrLen(MapListItem->CurrentDirectoryPath)-1] != L'\\') {
         ASSERT((MapListItem->CurrentDirectoryPath == NULL && Size == 0) || (MapListItem->CurrentDirectoryPath != NULL));
         MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, L"\\", 0);
       }

ShellPkg/Library/UefiShellDebug1CommandsLib/Bcfg.c

@@ -978,7 +978,7 @@ ShellCommandRunBcfg (
           ShellStatus = BcfgAddDebug1(
             CurrentOperation.Number1,
             CurrentOperation.FileName,
-            CurrentOperation.Description,
+            CurrentOperation.Description==NULL?L"":CurrentOperation.Description,
             CurrentOperation.Order,
             Length / sizeof(CurrentOperation.Order[0]),
             CurrentOperation.Target,

ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/FileImage.c

@@ -178,6 +178,7 @@ HFileImageRead (
 
     StatusBarSetStatusString (UnicodeBuffer);
     FreePool (UnicodeBuffer);
+    return EFI_OUT_OF_RESOURCES;
   }
 
   HFileImageSetFileName (FileName);

ShellPkg/Library/UefiShellDebug1CommandsLib/UefiShellDebug1CommandsLib.c

@@ -303,6 +303,9 @@ ConvertStringToGuid (
   } 
   TempCopy = NULL;
   TempCopy = StrnCatGrow(&TempCopy, NULL, StringGuid, 0);
+  if (TempCopy == NULL) {
+    return (EFI_OUT_OF_RESOURCES);
+  }
   Walker   = TempCopy;
   TempSpot = StrStr(Walker, L"-");
   if (TempSpot != NULL) {

ShellPkg/Library/UefiShellInstall1CommandsLib/Bcfg.c

@@ -978,7 +978,7 @@ ShellCommandRunBcfgInstall (
           ShellStatus = BcfgAddInstall1(
             CurrentOperation.Number1,
             CurrentOperation.FileName,
-            CurrentOperation.Description,
+            CurrentOperation.Description==NULL?L"":CurrentOperation.Description,
             CurrentOperation.Order,
             Length / sizeof(CurrentOperation.Order[0]),
             CurrentOperation.Target,

ShellPkg/Library/UefiShellLevel1CommandsLib/For.c

@@ -376,6 +376,9 @@ ShellCommandRunFor (
         }
         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
       }
+      if (ArgSet == NULL) {
+        ShellStatus = SHELL_OUT_OF_RESOURCES;
+      } else {
         //
         // set up for an 'in' for loop
         //
@@ -399,6 +402,7 @@ ShellCommandRunFor (
           Info->RemoveSubstAlias  = TRUE;
         }
         CurrentScriptFile->CurrentCommand->Data = Info;
+      }
     } else if (gUnicodeCollation->StriColl(
         gUnicodeCollation,
         L"run",
@@ -413,6 +417,9 @@ ShellCommandRunFor (
         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);
       }
+      if (ArgSet == NULL) {
+        ShellStatus = SHELL_OUT_OF_RESOURCES;
+      } else {
         //
         // set up for a 'run' for loop
         //
@@ -568,6 +575,7 @@ ShellCommandRunFor (
         if (CurrentScriptFile->CurrentCommand != NULL) {
           CurrentScriptFile->CurrentCommand->Data = Info;
         }
+      }
     } else {
       ShellPrintHiiEx(
         -1, 
@@ -665,6 +673,9 @@ ShellCommandRunFor (
         //
         ASSERT(TempString == NULL);
         TempString = StrnCatGrow(&TempString, NULL, Info->CurrentValue, 0);
+        if (TempString == NULL) {
+          ShellStatus = SHELL_OUT_OF_RESOURCES;
+        } else {
           TempSpot   = StrStr(TempString, L"\" \"");
           if (TempSpot != NULL) {
             *TempSpot = CHAR_NULL;
@@ -685,7 +696,7 @@ ShellCommandRunFor (
             Info->CurrentValue++;
           }
           FreePool(TempString);
-
+        }
       } else {
         CurrentScriptFile->CurrentCommand->Data = NULL;
         //

ShellPkg/Library/UefiShellLevel1CommandsLib/If.c

@@ -739,6 +739,9 @@ MoveToTagSpecial (
     //
     CommandName   = NULL;
     CommandName   = StrnCatGrow(&CommandName, NULL, CommandNode->Cl, 0);
+    if (CommandName == NULL) {
+      continue;
+    }
     CommandWalker = CommandName;
     while (CommandWalker[0] == L' ') {
       CommandWalker++;

ShellPkg/Library/UefiShellLevel1CommandsLib/UefiShellLevel1CommandsLib.c

@@ -145,6 +145,10 @@ TestNodeForMove (
   //
   CommandName   = NULL;
   CommandName   = StrnCatGrow(&CommandName, NULL, CommandNode->Cl, 0);
+  if (CommandName == NULL) {
+    return (FALSE);
+  }
+
   CommandNameWalker = CommandName;
   while(CommandNameWalker[0] == L' ') {
     CommandNameWalker++;

ShellPkg/Library/UefiShellLevel2CommandsLib/Map.c

@@ -414,6 +414,9 @@ PerformSingleMappingDisplay(
 
   CurrentName = NULL;
   CurrentName = StrnCatGrow(&CurrentName, 0, MapList, 0);
+  if (CurrentName == NULL) {
+    return (EFI_OUT_OF_RESOURCES);
+  }
   TempSpot = StrStr(CurrentName, L";");
   if (TempSpot != NULL) {
     *TempSpot = CHAR_NULL;

ShellPkg/Library/UefiShellLevel2CommandsLib/TimeDate.c

@@ -642,10 +642,15 @@ CheckAndSetTimeZone (
   }
 
   Status = gRT->GetTime(&TheTime, NULL);
-  ASSERT_EFI_ERROR(Status);
+  if (EFI_ERROR(Status)) {
+    return (SHELL_DEVICE_ERROR);
+  }
 
   TimeZoneCopy = NULL;
   TimeZoneCopy = StrnCatGrow(&TimeZoneCopy, NULL, TimeZoneString, 0);
+  if (TimeZoneCopy == NULL) {
+    return (SHELL_OUT_OF_RESOURCES);
+  }
   Walker = TimeZoneCopy;
   Walker2 = StrStr(Walker, L":");
   if (Walker2 != NULL && *Walker2 == L':') {

ShellPkg/Library/UefiShellLevel2CommandsLib/UefiShellLevel2CommandsLib.c

@@ -226,6 +226,10 @@ VerifyIntermediateDirectories (
   PathCopy    = StrnCatGrow(&PathCopy, NULL, Path, 0);
   FileHandle  = NULL;
 
+  if (PathCopy == NULL) {
+    return (EFI_OUT_OF_RESOURCES);
+  }
+
   for (TempSpot = &PathCopy[StrLen(PathCopy)-1] ; *TempSpot != CHAR_NULL && *TempSpot != L'\\' ; TempSpot = &PathCopy[StrLen(PathCopy)-1]){
     *TempSpot = CHAR_NULL;
   }

ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c

@@ -79,11 +79,16 @@ DoTouchByHandle (
 
   if (FS == NULL) {
     FS = StrnCatGrow(&FS, NULL, Name, 0);
+    if (FS != NULL) {
       TempSpot = StrStr(FS, L"\\");
       if (TempSpot != NULL) {
         *TempSpot = CHAR_NULL;
       }
     }
+  }
+  if (FS == NULL) {
+    return (EFI_INVALID_PARAMETER);
+  }
 
   //
   // do it