FileStringPath = NULL;\r
NewSize = 0;\r
FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, MapName, 0);\r
- TempSpot = StrStr(FileStringPath, L";");\r
- if (TempSpot != NULL) {\r
- *TempSpot = CHAR_NULL;\r
+ if (FileStringPath == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ } else {\r
+ TempSpot = StrStr(FileStringPath, L";");\r
+ if (TempSpot != NULL) {\r
+ *TempSpot = CHAR_NULL;\r
+ }\r
+ FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, ((FILEPATH_DEVICE_PATH*)FilePath)->PathName, 0);\r
+ PathRemoveLastItem(FileStringPath);\r
+ FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, mStartupScript, 0);\r
+ Status = ShellInfoObject.NewEfiShellProtocol->OpenFileByName(FileStringPath, &FileHandle, EFI_FILE_MODE_READ);\r
+ FreePool(FileStringPath);\r
}\r
- FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, ((FILEPATH_DEVICE_PATH*)FilePath)->PathName, 0);\r
- PathRemoveLastItem(FileStringPath);\r
- FileStringPath = StrnCatGrow(&FileStringPath, &NewSize, mStartupScript, 0);\r
- Status = ShellInfoObject.NewEfiShellProtocol->OpenFileByName(FileStringPath, &FileHandle, EFI_FILE_MODE_READ);\r
- FreePool(FileStringPath);\r
}\r
if (EFI_ERROR(Status)) {\r
NamePath = FileDevicePath (NULL, mStartupScript);\r
\r
NextCommandLine = StrnCatGrow(&NextCommandLine, &Size1, StrStr(CmdLine, L"|")+1, 0);\r
OurCommandLine = StrnCatGrow(&OurCommandLine , &Size2, CmdLine , StrStr(CmdLine, L"|") - CmdLine);\r
+\r
+ if (NextCommandLine == NULL || OurCommandLine == NULL) {\r
+ SHELL_FREE_NON_NULL(OurCommandLine);\r
+ SHELL_FREE_NON_NULL(NextCommandLine);\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
if (NextCommandLine[0] != CHAR_NULL &&\r
NextCommandLine[0] == L'a' &&\r
NextCommandLine[1] == L' '\r
Split = NULL;\r
\r
CleanOriginal = StrnCatGrow(&CleanOriginal, NULL, CmdLine, 0);\r
+ if (CleanOriginal == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
while (CleanOriginal[StrLen(CleanOriginal)-1] == L' ') {\r
CleanOriginal[StrLen(CleanOriginal)-1] = CHAR_NULL;\r
}\r
This function converts a file system style name to a device path, by replacing any\r
mapping references to the associated device path.\r
\r
- @param Path the pointer to the path\r
+ @param[in] Path The pointer to the path.\r
\r
- @return all The pointer of the file path. The file path is callee\r
+ @return The pointer of the file path. The file path is callee\r
allocated and should be freed by the caller.\r
+ @retval NULL The path could not be found.\r
+ @retval NULL There was not enough available memory.\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
Size = StrSize(Cwd);\r
Size += StrSize(Path);\r
NewPath = AllocateZeroPool(Size);\r
- ASSERT(NewPath != NULL);\r
+ if (NewPath == NULL) {\r
+ return (NULL);\r
+ }\r
StrCpy(NewPath, Cwd);\r
if (*Path == L'\\') {\r
Path++;\r
//\r
ASSERT((MapName == NULL && Size == 0) || (MapName != NULL));\r
MapName = StrnCatGrow(&MapName, &Size, Path, (StrStr(Path, L":")-Path+1));\r
- if (MapName[StrLen(MapName)-1] != L':') {\r
- ASSERT(FALSE);\r
+ if (MapName == NULL || MapName[StrLen(MapName)-1] != L':') {\r
return (NULL);\r
}\r
\r
//\r
DevicePathCopyForFree = DevicePathCopy = DuplicateDevicePath(DevicePath);\r
if (DevicePathCopy == NULL) {\r
- ASSERT(FALSE);\r
FreePool(MapName);\r
return (NULL);\r
}\r
TempString = NULL;\r
Size = 0;\r
TempString = StrnCatGrow(&TempString, &Size, ShellFileHandleGetPath(FileDirHandle), 0);\r
+ if (TempString == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
TempSpot = StrStr(TempString, L";");\r
\r
if (TempSpot != NULL) {\r
}\r
\r
TempString = StrnCatGrow(&TempString, &Size, BasePath, 0);\r
+ if (TempString == NULL) {\r
+ return (EFI_OUT_OF_RESOURCES);\r
+ }\r
BasePath = TempString;\r
}\r
\r
\r
ASSERT(MapName == NULL);\r
MapName = StrnCatGrow(&MapName, NULL, PatternCopy, Count);\r
+ if (MapName == NULL) {\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ }\r
\r
if (!EFI_ERROR(Status)) {\r
RootDevicePath = EfiShellGetDevicePathFromFilePath(PatternCopy);\r
MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, L"\\", 0);\r
ASSERT((MapListItem->CurrentDirectoryPath == NULL && Size == 0) || (MapListItem->CurrentDirectoryPath != NULL));\r
MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, DirectoryName, 0);\r
- if (MapListItem->CurrentDirectoryPath[StrLen(MapListItem->CurrentDirectoryPath)-1] != L'\\') {\r
+ if (MapListItem->CurrentDirectoryPath != NULL && MapListItem->CurrentDirectoryPath[StrLen(MapListItem->CurrentDirectoryPath)-1] != L'\\') {\r
ASSERT((MapListItem->CurrentDirectoryPath == NULL && Size == 0) || (MapListItem->CurrentDirectoryPath != NULL));\r
MapListItem->CurrentDirectoryPath = StrnCatGrow(&MapListItem->CurrentDirectoryPath, &Size, L"\\", 0);\r
}\r
}\r
ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);\r
}\r
- //\r
- // set up for an 'in' for loop\r
- //\r
- NewSize = StrSize(ArgSet);\r
- NewSize += sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]);\r
- Info = AllocateZeroPool(NewSize);\r
- ASSERT(Info != NULL);\r
- Info->Signature = SHELL_FOR_INFO_SIGNATURE;\r
- CopyMem(Info->Set, ArgSet, StrSize(ArgSet));\r
- NewSize = StrSize(gEfiShellParametersProtocol->Argv[1]);\r
- CopyMem(Info->Set+(StrSize(ArgSet)/sizeof(Info->Set[0])), gEfiShellParametersProtocol->Argv[1], NewSize);\r
- Info->ReplacementName = Info->Set+StrSize(ArgSet)/sizeof(Info->Set[0]);\r
- Info->CurrentValue = (CHAR16*)Info->Set;\r
- Info->Step = 0;\r
- Info->Current = 0;\r
- Info->End = 0;\r
-\r
- if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {\r
- Info->RemoveSubstAlias = FALSE;\r
+ if (ArgSet == NULL) {\r
+ ShellStatus = SHELL_OUT_OF_RESOURCES;\r
} else {\r
- Info->RemoveSubstAlias = TRUE;\r
+ //\r
+ // set up for an 'in' for loop\r
+ //\r
+ NewSize = StrSize(ArgSet);\r
+ NewSize += sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]);\r
+ Info = AllocateZeroPool(NewSize);\r
+ ASSERT(Info != NULL);\r
+ Info->Signature = SHELL_FOR_INFO_SIGNATURE;\r
+ CopyMem(Info->Set, ArgSet, StrSize(ArgSet));\r
+ NewSize = StrSize(gEfiShellParametersProtocol->Argv[1]);\r
+ CopyMem(Info->Set+(StrSize(ArgSet)/sizeof(Info->Set[0])), gEfiShellParametersProtocol->Argv[1], NewSize);\r
+ Info->ReplacementName = Info->Set+StrSize(ArgSet)/sizeof(Info->Set[0]);\r
+ Info->CurrentValue = (CHAR16*)Info->Set;\r
+ Info->Step = 0;\r
+ Info->Current = 0;\r
+ Info->End = 0;\r
+\r
+ if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {\r
+ Info->RemoveSubstAlias = FALSE;\r
+ } else {\r
+ Info->RemoveSubstAlias = TRUE;\r
+ }\r
+ CurrentScriptFile->CurrentCommand->Data = Info;\r
}\r
- CurrentScriptFile->CurrentCommand->Data = Info;\r
} else if (gUnicodeCollation->StriColl(\r
gUnicodeCollation,\r
L"run",\r
ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);\r
// ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);\r
}\r
- //\r
- // set up for a 'run' for loop\r
- //\r
- Info = AllocateZeroPool(sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]));\r
- ASSERT(Info != NULL);\r
- CopyMem(Info->Set, gEfiShellParametersProtocol->Argv[1], StrSize(gEfiShellParametersProtocol->Argv[1]));\r
- Info->ReplacementName = Info->Set;\r
- Info->CurrentValue = NULL;\r
- ArgSetWalker = ArgSet;\r
- if (ArgSetWalker[0] != L'(') {\r
- ShellPrintHiiEx(\r
- -1, \r
- -1, \r
- NULL, \r
- STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
- gShellLevel1HiiHandle, \r
- ArgSet, \r
- CurrentScriptFile!=NULL \r
- && CurrentScriptFile->CurrentCommand!=NULL\r
- ? CurrentScriptFile->CurrentCommand->Line:0);\r
- ShellStatus = SHELL_INVALID_PARAMETER;\r
+ if (ArgSet == NULL) {\r
+ ShellStatus = SHELL_OUT_OF_RESOURCES;\r
} else {\r
- TempSpot = StrStr(ArgSetWalker, L")");\r
- if (TempSpot != NULL) {\r
- TempString = TempSpot+1;\r
- if (*(TempString) != CHAR_NULL) {\r
- while(TempString != NULL && *TempString == L' ') {\r
- TempString++;\r
- }\r
- if (StrLen(TempString) > 0) {\r
- TempSpot = NULL;\r
- }\r
- }\r
- }\r
- if (TempSpot == NULL) {\r
+ //\r
+ // set up for a 'run' for loop\r
+ //\r
+ Info = AllocateZeroPool(sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]));\r
+ ASSERT(Info != NULL);\r
+ CopyMem(Info->Set, gEfiShellParametersProtocol->Argv[1], StrSize(gEfiShellParametersProtocol->Argv[1]));\r
+ Info->ReplacementName = Info->Set;\r
+ Info->CurrentValue = NULL;\r
+ ArgSetWalker = ArgSet;\r
+ if (ArgSetWalker[0] != L'(') {\r
ShellPrintHiiEx(\r
-1, \r
-1, \r
NULL, \r
STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
gShellLevel1HiiHandle, \r
+ ArgSet, \r
CurrentScriptFile!=NULL \r
&& CurrentScriptFile->CurrentCommand!=NULL\r
? CurrentScriptFile->CurrentCommand->Line:0);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- *TempSpot = CHAR_NULL;\r
- ArgSetWalker++;\r
- while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {\r
- ArgSetWalker++;\r
+ TempSpot = StrStr(ArgSetWalker, L")");\r
+ if (TempSpot != NULL) {\r
+ TempString = TempSpot+1;\r
+ if (*(TempString) != CHAR_NULL) {\r
+ while(TempString != NULL && *TempString == L' ') {\r
+ TempString++;\r
+ }\r
+ if (StrLen(TempString) > 0) {\r
+ TempSpot = NULL;\r
+ }\r
+ }\r
}\r
- if (!ShellIsValidForNumber(ArgSetWalker)) {\r
+ if (TempSpot == NULL) {\r
ShellPrintHiiEx(\r
-1, \r
-1, \r
NULL, \r
STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
gShellLevel1HiiHandle, \r
- ArgSet, \r
CurrentScriptFile!=NULL \r
&& CurrentScriptFile->CurrentCommand!=NULL\r
? CurrentScriptFile->CurrentCommand->Line:0);\r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- if (ArgSetWalker[0] == L'-') {\r
- Info->Current = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
- } else {\r
- Info->Current = (INTN)ShellStrToUintn(ArgSetWalker);\r
- }\r
- ArgSetWalker = StrStr(ArgSetWalker, L" ");\r
+ *TempSpot = CHAR_NULL;\r
+ ArgSetWalker++;\r
while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {\r
ArgSetWalker++;\r
}\r
- if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){\r
+ if (!ShellIsValidForNumber(ArgSetWalker)) {\r
ShellPrintHiiEx(\r
-1, \r
-1, \r
ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
if (ArgSetWalker[0] == L'-') {\r
- Info->End = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
- } else {\r
- Info->End = (INTN)ShellStrToUintn(ArgSetWalker);\r
- }\r
- if (Info->Current < Info->End) {\r
- Info->Step = 1;\r
+ Info->Current = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
} else {\r
- Info->Step = -1;\r
+ Info->Current = (INTN)ShellStrToUintn(ArgSetWalker);\r
}\r
-\r
ArgSetWalker = StrStr(ArgSetWalker, L" ");\r
while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {\r
ArgSetWalker++;\r
}\r
- if (ArgSetWalker != NULL && *ArgSetWalker != CHAR_NULL) {\r
- if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){\r
- ShellPrintHiiEx(\r
- -1, \r
- -1, \r
- NULL, \r
- STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
- gShellLevel1HiiHandle, \r
- ArgSet, \r
- CurrentScriptFile!=NULL \r
- && CurrentScriptFile->CurrentCommand!=NULL\r
- ? CurrentScriptFile->CurrentCommand->Line:0);\r
- ShellStatus = SHELL_INVALID_PARAMETER;\r
+ if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){\r
+ ShellPrintHiiEx(\r
+ -1, \r
+ -1, \r
+ NULL, \r
+ STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
+ gShellLevel1HiiHandle, \r
+ ArgSet, \r
+ CurrentScriptFile!=NULL \r
+ && CurrentScriptFile->CurrentCommand!=NULL\r
+ ? CurrentScriptFile->CurrentCommand->Line:0);\r
+ ShellStatus = SHELL_INVALID_PARAMETER;\r
+ } else {\r
+ if (ArgSetWalker[0] == L'-') {\r
+ Info->End = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
} else {\r
- if (*ArgSetWalker == L')') {\r
- ASSERT(Info->Step == 1 || Info->Step == -1);\r
+ Info->End = (INTN)ShellStrToUintn(ArgSetWalker);\r
+ }\r
+ if (Info->Current < Info->End) {\r
+ Info->Step = 1;\r
+ } else {\r
+ Info->Step = -1;\r
+ }\r
+\r
+ ArgSetWalker = StrStr(ArgSetWalker, L" ");\r
+ while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {\r
+ ArgSetWalker++;\r
+ }\r
+ if (ArgSetWalker != NULL && *ArgSetWalker != CHAR_NULL) {\r
+ if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){\r
+ ShellPrintHiiEx(\r
+ -1, \r
+ -1, \r
+ NULL, \r
+ STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
+ gShellLevel1HiiHandle, \r
+ ArgSet, \r
+ CurrentScriptFile!=NULL \r
+ && CurrentScriptFile->CurrentCommand!=NULL\r
+ ? CurrentScriptFile->CurrentCommand->Line:0);\r
+ ShellStatus = SHELL_INVALID_PARAMETER;\r
} else {\r
- if (ArgSetWalker[0] == L'-') {\r
- Info->Step = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
+ if (*ArgSetWalker == L')') {\r
+ ASSERT(Info->Step == 1 || Info->Step == -1);\r
} else {\r
- Info->Step = (INTN)ShellStrToUintn(ArgSetWalker);\r
- }\r
-\r
- if (StrStr(ArgSetWalker, L" ") != NULL) {\r
- ShellPrintHiiEx(\r
- -1, \r
- -1, \r
- NULL, \r
- STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
- gShellLevel1HiiHandle, \r
- ArgSet, \r
- CurrentScriptFile!=NULL \r
- && CurrentScriptFile->CurrentCommand!=NULL\r
- ? CurrentScriptFile->CurrentCommand->Line:0);\r
- ShellStatus = SHELL_INVALID_PARAMETER;\r
+ if (ArgSetWalker[0] == L'-') {\r
+ Info->Step = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);\r
+ } else {\r
+ Info->Step = (INTN)ShellStrToUintn(ArgSetWalker);\r
+ }\r
+\r
+ if (StrStr(ArgSetWalker, L" ") != NULL) {\r
+ ShellPrintHiiEx(\r
+ -1, \r
+ -1, \r
+ NULL, \r
+ STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), \r
+ gShellLevel1HiiHandle, \r
+ ArgSet, \r
+ CurrentScriptFile!=NULL \r
+ && CurrentScriptFile->CurrentCommand!=NULL\r
+ ? CurrentScriptFile->CurrentCommand->Line:0);\r
+ ShellStatus = SHELL_INVALID_PARAMETER;\r
+ }\r
}\r
}\r
+ \r
}\r
- \r
}\r
}\r
}\r
}\r
- }\r
- if (ShellStatus == SHELL_SUCCESS) {\r
- if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {\r
- Info->RemoveSubstAlias = FALSE;\r
- } else {\r
- Info->RemoveSubstAlias = TRUE;\r
+ if (ShellStatus == SHELL_SUCCESS) {\r
+ if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {\r
+ Info->RemoveSubstAlias = FALSE;\r
+ } else {\r
+ Info->RemoveSubstAlias = TRUE;\r
+ }\r
+ }\r
+ if (CurrentScriptFile->CurrentCommand != NULL) {\r
+ CurrentScriptFile->CurrentCommand->Data = Info;\r
}\r
- }\r
- if (CurrentScriptFile->CurrentCommand != NULL) {\r
- CurrentScriptFile->CurrentCommand->Data = Info;\r
}\r
} else {\r
ShellPrintHiiEx(\r
//\r
ASSERT(TempString == NULL);\r
TempString = StrnCatGrow(&TempString, NULL, Info->CurrentValue, 0);\r
- TempSpot = StrStr(TempString, L"\" \"");\r
- if (TempSpot != NULL) {\r
- *TempSpot = CHAR_NULL;\r
- }\r
- while (TempString[StrLen(TempString)-1] == L'\"') {\r
- TempString[StrLen(TempString)-1] = CHAR_NULL;\r
- }\r
- InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);\r
- Info->CurrentValue += StrLen(TempString);\r
+ if (TempString == NULL) {\r
+ ShellStatus = SHELL_OUT_OF_RESOURCES;\r
+ } else {\r
+ TempSpot = StrStr(TempString, L"\" \"");\r
+ if (TempSpot != NULL) {\r
+ *TempSpot = CHAR_NULL;\r
+ }\r
+ while (TempString[StrLen(TempString)-1] == L'\"') {\r
+ TempString[StrLen(TempString)-1] = CHAR_NULL;\r
+ }\r
+ InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);\r
+ Info->CurrentValue += StrLen(TempString);\r
\r
- if (Info->CurrentValue[0] == L'\"') {\r
- Info->CurrentValue++;\r
- }\r
- while (Info->CurrentValue[0] == L' ') {\r
- Info->CurrentValue++;\r
- }\r
- if (Info->CurrentValue[0] == L'\"') {\r
- Info->CurrentValue++;\r
+ if (Info->CurrentValue[0] == L'\"') {\r
+ Info->CurrentValue++;\r
+ }\r
+ while (Info->CurrentValue[0] == L' ') {\r
+ Info->CurrentValue++;\r
+ }\r
+ if (Info->CurrentValue[0] == L'\"') {\r
+ Info->CurrentValue++;\r
+ }\r
+ FreePool(TempString);\r
}\r
- FreePool(TempString);\r
-\r
} else {\r
CurrentScriptFile->CurrentCommand->Data = NULL;\r
//\r