| /** @file | |
| Main file for endfor and for shell level 1 functions. | |
| (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR> | |
| Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR> | |
| SPDX-License-Identifier: BSD-2-Clause-Patent | |
| **/ | |
| #include "UefiShellLevel1CommandsLib.h" | |
| #include <Library/PrintLib.h> | |
| /** | |
| Determine if a valid string is a valid number for the 'for' command. | |
| @param[in] Number The pointer to the string representation of the number to test. | |
| @retval TRUE The number is valid. | |
| @retval FALSE The number is not valid. | |
| **/ | |
| BOOLEAN | |
| ShellIsValidForNumber ( | |
| IN CONST CHAR16 *Number | |
| ) | |
| { | |
| if ((Number == NULL) || (*Number == CHAR_NULL)) { | |
| return (FALSE); | |
| } | |
| if (*Number == L'-') { | |
| Number++; | |
| } | |
| if (StrLen (Number) == 0) { | |
| return (FALSE); | |
| } | |
| if (StrLen (Number) >= 7) { | |
| if ((StrStr (Number, L" ") == NULL) || (((StrStr (Number, L" ") != NULL) && ((StrStr (Number, L" ") - Number) >= 7)))) { | |
| return (FALSE); | |
| } | |
| } | |
| if (!ShellIsDecimalDigitCharacter (*Number)) { | |
| return (FALSE); | |
| } | |
| return (TRUE); | |
| } | |
| /** | |
| Function for 'endfor' command. | |
| @param[in] ImageHandle Handle to the Image (NULL if Internal). | |
| @param[in] SystemTable Pointer to the System Table (NULL if Internal). | |
| **/ | |
| SHELL_STATUS | |
| EFIAPI | |
| ShellCommandRunEndFor ( | |
| IN EFI_HANDLE ImageHandle, | |
| IN EFI_SYSTEM_TABLE *SystemTable | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| BOOLEAN Found; | |
| SCRIPT_FILE *CurrentScriptFile; | |
| Status = CommandInit (); | |
| ASSERT_EFI_ERROR (Status); | |
| if (!gEfiShellProtocol->BatchIsActive ()) { | |
| ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_NO_SCRIPT), gShellLevel1HiiHandle, L"endfor"); | |
| return (SHELL_UNSUPPORTED); | |
| } | |
| if (gEfiShellParametersProtocol->Argc > 1) { | |
| ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel1HiiHandle, L"endfor"); | |
| return (SHELL_INVALID_PARAMETER); | |
| } | |
| Found = MoveToTag (GetPreviousNode, L"for", L"endfor", NULL, ShellCommandGetCurrentScriptFile (), FALSE, FALSE, FALSE); | |
| if (!Found) { | |
| CurrentScriptFile = ShellCommandGetCurrentScriptFile (); | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_SYNTAX_NO_MATCHING), | |
| gShellLevel1HiiHandle, | |
| L"For", | |
| L"EndFor", | |
| CurrentScriptFile != NULL | |
| && CurrentScriptFile->CurrentCommand != NULL | |
| ? CurrentScriptFile->CurrentCommand->Line : 0 | |
| ); | |
| return (SHELL_NOT_FOUND); | |
| } | |
| return (SHELL_SUCCESS); | |
| } | |
| typedef struct { | |
| UINT32 Signature; | |
| INTN Current; | |
| INTN End; | |
| INTN Step; | |
| CHAR16 *ReplacementName; | |
| CHAR16 *CurrentValue; | |
| BOOLEAN RemoveSubstAlias; | |
| CHAR16 Set[1]; | |
| } SHELL_FOR_INFO; | |
| #define SIZE_OF_SHELL_FOR_INFO OFFSET_OF (SHELL_FOR_INFO, Set) | |
| #define SHELL_FOR_INFO_SIGNATURE SIGNATURE_32 ('S', 'F', 'I', 's') | |
| /** | |
| Update the value of a given alias on the list. If the alias is not there then add it. | |
| @param[in] Alias The alias to test for. | |
| @param[in] CommandString The updated command string. | |
| @param[in, out] List The list to search. | |
| @retval EFI_SUCCESS The operation was completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES There was not enough free memory. | |
| **/ | |
| EFI_STATUS | |
| InternalUpdateAliasOnList ( | |
| IN CONST CHAR16 *Alias, | |
| IN CONST CHAR16 *CommandString, | |
| IN OUT LIST_ENTRY *List | |
| ) | |
| { | |
| ALIAS_LIST *Node; | |
| BOOLEAN Found; | |
| // | |
| // assert for NULL parameter | |
| // | |
| ASSERT (Alias != NULL); | |
| // | |
| // check for the Alias | |
| // | |
| for ( Node = (ALIAS_LIST *)GetFirstNode (List), Found = FALSE | |
| ; !IsNull (List, &Node->Link) | |
| ; Node = (ALIAS_LIST *)GetNextNode (List, &Node->Link) | |
| ) | |
| { | |
| ASSERT (Node->CommandString != NULL); | |
| ASSERT (Node->Alias != NULL); | |
| if (StrCmp (Node->Alias, Alias) == 0) { | |
| FreePool (Node->CommandString); | |
| Node->CommandString = NULL; | |
| Node->CommandString = StrnCatGrow (&Node->CommandString, NULL, CommandString, 0); | |
| Found = TRUE; | |
| break; | |
| } | |
| } | |
| if (!Found) { | |
| Node = AllocateZeroPool (sizeof (ALIAS_LIST)); | |
| if (Node == NULL) { | |
| return (EFI_OUT_OF_RESOURCES); | |
| } | |
| ASSERT (Node->Alias == NULL); | |
| Node->Alias = StrnCatGrow (&Node->Alias, NULL, Alias, 0); | |
| ASSERT (Node->CommandString == NULL); | |
| Node->CommandString = StrnCatGrow (&Node->CommandString, NULL, CommandString, 0); | |
| InsertTailList (List, &Node->Link); | |
| } | |
| return (EFI_SUCCESS); | |
| } | |
| /** | |
| Find out if an alias is on the given list. | |
| @param[in] Alias The alias to test for. | |
| @param[in] List The list to search. | |
| @retval TRUE The alias is on the list. | |
| @retval FALSE The alias is not on the list. | |
| **/ | |
| BOOLEAN | |
| InternalIsAliasOnList ( | |
| IN CONST CHAR16 *Alias, | |
| IN CONST LIST_ENTRY *List | |
| ) | |
| { | |
| ALIAS_LIST *Node; | |
| // | |
| // assert for NULL parameter | |
| // | |
| ASSERT (Alias != NULL); | |
| // | |
| // check for the Alias | |
| // | |
| for ( Node = (ALIAS_LIST *)GetFirstNode (List) | |
| ; !IsNull (List, &Node->Link) | |
| ; Node = (ALIAS_LIST *)GetNextNode (List, &Node->Link) | |
| ) | |
| { | |
| ASSERT (Node->CommandString != NULL); | |
| ASSERT (Node->Alias != NULL); | |
| if (StrCmp (Node->Alias, Alias) == 0) { | |
| return (TRUE); | |
| } | |
| } | |
| return (FALSE); | |
| } | |
| /** | |
| Remove an alias from the given list. | |
| @param[in] Alias The alias to remove. | |
| @param[in, out] List The list to search. | |
| **/ | |
| BOOLEAN | |
| InternalRemoveAliasFromList ( | |
| IN CONST CHAR16 *Alias, | |
| IN OUT LIST_ENTRY *List | |
| ) | |
| { | |
| ALIAS_LIST *Node; | |
| // | |
| // assert for NULL parameter | |
| // | |
| ASSERT (Alias != NULL); | |
| // | |
| // check for the Alias | |
| // | |
| for ( Node = (ALIAS_LIST *)GetFirstNode (List) | |
| ; !IsNull (List, &Node->Link) | |
| ; Node = (ALIAS_LIST *)GetNextNode (List, &Node->Link) | |
| ) | |
| { | |
| ASSERT (Node->CommandString != NULL); | |
| ASSERT (Node->Alias != NULL); | |
| if (StrCmp (Node->Alias, Alias) == 0) { | |
| RemoveEntryList (&Node->Link); | |
| FreePool (Node->Alias); | |
| FreePool (Node->CommandString); | |
| FreePool (Node); | |
| return (TRUE); | |
| } | |
| } | |
| return (FALSE); | |
| } | |
| /** | |
| Function to determine whether a string is decimal or hex representation of a number | |
| and return the number converted from the string. | |
| @param[in] String String representation of a number | |
| @return the number | |
| @retval (UINTN)(-1) An error ocurred. | |
| **/ | |
| UINTN | |
| ReturnUintn ( | |
| IN CONST CHAR16 *String | |
| ) | |
| { | |
| UINT64 RetVal; | |
| if (!EFI_ERROR (ShellConvertStringToUint64 (String, &RetVal, FALSE, TRUE))) { | |
| return ((UINTN)RetVal); | |
| } | |
| return ((UINTN)(-1)); | |
| } | |
| /** | |
| Function for 'for' command. | |
| @param[in] ImageHandle Handle to the Image (NULL if Internal). | |
| @param[in] SystemTable Pointer to the System Table (NULL if Internal). | |
| **/ | |
| SHELL_STATUS | |
| EFIAPI | |
| ShellCommandRunFor ( | |
| IN EFI_HANDLE ImageHandle, | |
| IN EFI_SYSTEM_TABLE *SystemTable | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| SHELL_STATUS ShellStatus; | |
| SCRIPT_FILE *CurrentScriptFile; | |
| CHAR16 *ArgSet; | |
| CHAR16 *ArgSetWalker; | |
| CHAR16 *Parameter; | |
| UINTN ArgSize; | |
| UINTN LoopVar; | |
| SHELL_FOR_INFO *Info; | |
| CHAR16 *TempString; | |
| CHAR16 *TempSpot; | |
| BOOLEAN FirstPass; | |
| EFI_SHELL_FILE_INFO *Node; | |
| EFI_SHELL_FILE_INFO *FileList; | |
| UINTN NewSize; | |
| ArgSet = NULL; | |
| ArgSize = 0; | |
| ShellStatus = SHELL_SUCCESS; | |
| ArgSetWalker = NULL; | |
| TempString = NULL; | |
| Parameter = NULL; | |
| FirstPass = FALSE; | |
| // | |
| // initialize the shell lib (we must be in non-auto-init...) | |
| // | |
| Status = ShellInitialize (); | |
| ASSERT_EFI_ERROR (Status); | |
| Status = CommandInit (); | |
| ASSERT_EFI_ERROR (Status); | |
| if (!gEfiShellProtocol->BatchIsActive ()) { | |
| ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_NO_SCRIPT), gShellLevel1HiiHandle, L"for"); | |
| return (SHELL_UNSUPPORTED); | |
| } | |
| if (gEfiShellParametersProtocol->Argc < 4) { | |
| ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel1HiiHandle, L"for"); | |
| return (SHELL_INVALID_PARAMETER); | |
| } | |
| CurrentScriptFile = ShellCommandGetCurrentScriptFile (); | |
| ASSERT (CurrentScriptFile != NULL); | |
| if ((CurrentScriptFile->CurrentCommand != NULL) && (CurrentScriptFile->CurrentCommand->Data == NULL)) { | |
| FirstPass = TRUE; | |
| // | |
| // Make sure that an End exists. | |
| // | |
| if (!MoveToTag (GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, TRUE, FALSE)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_SYNTAX_NO_MATCHING), | |
| gShellLevel1HiiHandle, | |
| L"EndFor", | |
| L"For", | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| return (SHELL_DEVICE_ERROR); | |
| } | |
| // | |
| // Process the line. | |
| // | |
| if ( (gEfiShellParametersProtocol->Argv[1][0] != L'%') || (gEfiShellParametersProtocol->Argv[1][2] != CHAR_NULL) | |
| || !( ((gEfiShellParametersProtocol->Argv[1][1] >= L'a') && (gEfiShellParametersProtocol->Argv[1][1] <= L'z')) | |
| || ((gEfiShellParametersProtocol->Argv[1][1] >= L'A') && (gEfiShellParametersProtocol->Argv[1][1] <= L'Z'))) | |
| ) | |
| { | |
| ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_VAR), gShellLevel1HiiHandle, gEfiShellParametersProtocol->Argv[1]); | |
| return (SHELL_INVALID_PARAMETER); | |
| } | |
| if (gUnicodeCollation->StriColl ( | |
| gUnicodeCollation, | |
| L"in", | |
| gEfiShellParametersProtocol->Argv[2] | |
| ) == 0) | |
| { | |
| for (LoopVar = 0x3; LoopVar < gEfiShellParametersProtocol->Argc; LoopVar++) { | |
| ASSERT ((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL)); | |
| if ( (StrStr (gEfiShellParametersProtocol->Argv[LoopVar], L"*") != NULL) | |
| || (StrStr (gEfiShellParametersProtocol->Argv[LoopVar], L"?") != NULL) | |
| || (StrStr (gEfiShellParametersProtocol->Argv[LoopVar], L"[") != NULL) | |
| || (StrStr (gEfiShellParametersProtocol->Argv[LoopVar], L"]") != NULL)) | |
| { | |
| FileList = NULL; | |
| Status = ShellOpenFileMetaArg ((CHAR16 *)gEfiShellParametersProtocol->Argv[LoopVar], EFI_FILE_MODE_READ, &FileList); | |
| if (EFI_ERROR (Status) || (FileList == NULL) || IsListEmpty (&FileList->Link)) { | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L" \"", 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L"\"", 0); | |
| } else { | |
| for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode (&FileList->Link) | |
| ; !IsNull (&FileList->Link, &Node->Link) | |
| ; Node = (EFI_SHELL_FILE_INFO *)GetNextNode (&FileList->Link, &Node->Link) | |
| ) | |
| { | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L" \"", 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, Node->FullName, 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L"\"", 0); | |
| } | |
| ShellCloseFileMetaArg (&FileList); | |
| } | |
| } else { | |
| Parameter = gEfiShellParametersProtocol->Argv[LoopVar]; | |
| if ((Parameter[0] == L'\"') && (Parameter[StrLen (Parameter)-1] == L'\"')) { | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L" ", 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, Parameter, 0); | |
| } else { | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L" \"", 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, Parameter, 0); | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L"\"", 0); | |
| } | |
| } | |
| } | |
| if (ArgSet == NULL) { | |
| ShellStatus = SHELL_OUT_OF_RESOURCES; | |
| } else { | |
| // | |
| // set up for an 'in' for loop | |
| // | |
| NewSize = StrSize (ArgSet); | |
| NewSize += sizeof (SHELL_FOR_INFO)+StrSize (gEfiShellParametersProtocol->Argv[1]); | |
| Info = AllocateZeroPool (NewSize); | |
| if (Info == NULL) { | |
| FreePool (ArgSet); | |
| return SHELL_OUT_OF_RESOURCES; | |
| } | |
| Info->Signature = SHELL_FOR_INFO_SIGNATURE; | |
| CopyMem (Info->Set, ArgSet, StrSize (ArgSet)); | |
| NewSize = StrSize (gEfiShellParametersProtocol->Argv[1]); | |
| CopyMem (Info->Set+(StrSize (ArgSet)/sizeof (Info->Set[0])), gEfiShellParametersProtocol->Argv[1], NewSize); | |
| Info->ReplacementName = Info->Set+StrSize (ArgSet)/sizeof (Info->Set[0]); | |
| Info->CurrentValue = (CHAR16 *)Info->Set; | |
| Info->Step = 0; | |
| Info->Current = 0; | |
| Info->End = 0; | |
| if (InternalIsAliasOnList (Info->ReplacementName, &CurrentScriptFile->SubstList)) { | |
| Info->RemoveSubstAlias = FALSE; | |
| } else { | |
| Info->RemoveSubstAlias = TRUE; | |
| } | |
| CurrentScriptFile->CurrentCommand->Data = Info; | |
| } | |
| } else if (gUnicodeCollation->StriColl ( | |
| gUnicodeCollation, | |
| L"run", | |
| gEfiShellParametersProtocol->Argv[2] | |
| ) == 0) | |
| { | |
| for (LoopVar = 0x3; LoopVar < gEfiShellParametersProtocol->Argc; LoopVar++) { | |
| ASSERT ((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL)); | |
| if ((StrStr (gEfiShellParametersProtocol->Argv[LoopVar], L")") != NULL) && | |
| ((LoopVar + 1) < gEfiShellParametersProtocol->Argc) | |
| ) | |
| { | |
| return (SHELL_INVALID_PARAMETER); | |
| } | |
| if (ArgSet == NULL) { | |
| // ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0); | |
| } else { | |
| ArgSet = StrnCatGrow (&ArgSet, &ArgSize, L" ", 0); | |
| } | |
| 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 | |
| // | |
| Info = AllocateZeroPool (sizeof (SHELL_FOR_INFO)+StrSize (gEfiShellParametersProtocol->Argv[1])); | |
| if (Info == NULL) { | |
| FreePool (ArgSet); | |
| return SHELL_OUT_OF_RESOURCES; | |
| } | |
| Info->Signature = SHELL_FOR_INFO_SIGNATURE; | |
| CopyMem (Info->Set, gEfiShellParametersProtocol->Argv[1], StrSize (gEfiShellParametersProtocol->Argv[1])); | |
| Info->ReplacementName = Info->Set; | |
| Info->CurrentValue = NULL; | |
| ArgSetWalker = ArgSet; | |
| if (ArgSetWalker[0] != L'(') { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| TempSpot = StrStr (ArgSetWalker, L")"); | |
| if (TempSpot != NULL) { | |
| TempString = TempSpot+1; | |
| if (*(TempString) != CHAR_NULL) { | |
| while (TempString != NULL && *TempString == L' ') { | |
| TempString++; | |
| } | |
| if (StrLen (TempString) > 0) { | |
| TempSpot = NULL; | |
| } | |
| } | |
| } | |
| if (TempSpot == NULL) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| *TempSpot = CHAR_NULL; | |
| ArgSetWalker++; | |
| while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') { | |
| ArgSetWalker++; | |
| } | |
| if (!ShellIsValidForNumber (ArgSetWalker)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| if (ArgSetWalker[0] == L'-') { | |
| Info->Current = 0 - (INTN)ReturnUintn (ArgSetWalker+1); | |
| } else { | |
| Info->Current = (INTN)ReturnUintn (ArgSetWalker); | |
| } | |
| ArgSetWalker = StrStr (ArgSetWalker, L" "); | |
| while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') { | |
| ArgSetWalker++; | |
| } | |
| if ((ArgSetWalker == NULL) || (*ArgSetWalker == CHAR_NULL) || !ShellIsValidForNumber (ArgSetWalker)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| if (ArgSetWalker[0] == L'-') { | |
| Info->End = 0 - (INTN)ReturnUintn (ArgSetWalker+1); | |
| } else { | |
| Info->End = (INTN)ReturnUintn (ArgSetWalker); | |
| } | |
| if (Info->Current < Info->End) { | |
| Info->Step = 1; | |
| } else { | |
| Info->Step = -1; | |
| } | |
| ArgSetWalker = StrStr (ArgSetWalker, L" "); | |
| while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') { | |
| ArgSetWalker++; | |
| } | |
| if ((ArgSetWalker != NULL) && (*ArgSetWalker != CHAR_NULL)) { | |
| if ((ArgSetWalker == NULL) || (*ArgSetWalker == CHAR_NULL) || !ShellIsValidForNumber (ArgSetWalker)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| if (*ArgSetWalker == L')') { | |
| ASSERT (Info->Step == 1 || Info->Step == -1); | |
| } else { | |
| if (ArgSetWalker[0] == L'-') { | |
| Info->Step = 0 - (INTN)ReturnUintn (ArgSetWalker+1); | |
| } else { | |
| Info->Step = (INTN)ReturnUintn (ArgSetWalker); | |
| } | |
| if (StrStr (ArgSetWalker, L" ") != NULL) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile->CurrentCommand->Line | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } | |
| } | |
| } | |
| } | |
| } | |
| } | |
| } | |
| } | |
| if (ShellStatus == SHELL_SUCCESS) { | |
| if (InternalIsAliasOnList (Info->ReplacementName, &CurrentScriptFile->SubstList)) { | |
| Info->RemoveSubstAlias = FALSE; | |
| } else { | |
| Info->RemoveSubstAlias = TRUE; | |
| } | |
| } | |
| if (CurrentScriptFile->CurrentCommand != NULL) { | |
| CurrentScriptFile->CurrentCommand->Data = Info; | |
| } | |
| } | |
| } else { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT), | |
| gShellLevel1HiiHandle, | |
| ArgSet, | |
| CurrentScriptFile != NULL | |
| && CurrentScriptFile->CurrentCommand != NULL | |
| ? CurrentScriptFile->CurrentCommand->Line : 0 | |
| ); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } | |
| } else { | |
| // | |
| // These need to be NULL since they are used to determine if this is the first pass later on... | |
| // | |
| ASSERT (ArgSetWalker == NULL); | |
| ASSERT (ArgSet == NULL); | |
| } | |
| if ((CurrentScriptFile != NULL) && (CurrentScriptFile->CurrentCommand != NULL)) { | |
| Info = (SHELL_FOR_INFO *)CurrentScriptFile->CurrentCommand->Data; | |
| if (CurrentScriptFile->CurrentCommand->Reset) { | |
| if (Info != NULL) { | |
| Info->CurrentValue = (CHAR16 *)Info->Set; | |
| } | |
| FirstPass = TRUE; | |
| CurrentScriptFile->CurrentCommand->Reset = FALSE; | |
| } | |
| } else { | |
| ShellStatus = SHELL_UNSUPPORTED; | |
| Info = NULL; | |
| } | |
| if (ShellStatus == SHELL_SUCCESS) { | |
| ASSERT (Info != NULL); | |
| if (Info->Step != 0) { | |
| // | |
| // only advance if not the first pass | |
| // | |
| if (!FirstPass) { | |
| // | |
| // sequence version of for loop... | |
| // | |
| Info->Current += Info->Step; | |
| } | |
| TempString = AllocateZeroPool (50*sizeof (CHAR16)); | |
| UnicodeSPrint (TempString, 50*sizeof (CHAR16), L"%d", Info->Current); | |
| InternalUpdateAliasOnList (Info->ReplacementName, TempString, &CurrentScriptFile->SubstList); | |
| FreePool (TempString); | |
| if (((Info->Step > 0) && (Info->Current > Info->End)) || ((Info->Step < 0) && (Info->Current < Info->End))) { | |
| CurrentScriptFile->CurrentCommand->Data = NULL; | |
| // | |
| // find the matching endfor (we're done with the loop) | |
| // | |
| if (!MoveToTag (GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_SYNTAX_NO_MATCHING), | |
| gShellLevel1HiiHandle, | |
| L"EndFor", | |
| L"For", | |
| CurrentScriptFile != NULL | |
| && CurrentScriptFile->CurrentCommand != NULL | |
| ? CurrentScriptFile->CurrentCommand->Line : 0 | |
| ); | |
| ShellStatus = SHELL_DEVICE_ERROR; | |
| } | |
| if (Info->RemoveSubstAlias) { | |
| // | |
| // remove item from list | |
| // | |
| InternalRemoveAliasFromList (Info->ReplacementName, &CurrentScriptFile->SubstList); | |
| } | |
| FreePool (Info); | |
| } | |
| } else { | |
| // | |
| // Must be in 'in' version of for loop... | |
| // | |
| ASSERT (Info->Set != NULL); | |
| if ((Info->CurrentValue != NULL) && (*Info->CurrentValue != CHAR_NULL)) { | |
| if (Info->CurrentValue[0] == L' ') { | |
| Info->CurrentValue++; | |
| } | |
| if (Info->CurrentValue[0] == L'\"') { | |
| Info->CurrentValue++; | |
| } | |
| // | |
| // do the next one of the set | |
| // | |
| 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; | |
| } | |
| while (TempString[StrLen (TempString)-1] == L'\"') { | |
| TempString[StrLen (TempString)-1] = CHAR_NULL; | |
| } | |
| InternalUpdateAliasOnList (Info->ReplacementName, TempString, &CurrentScriptFile->SubstList); | |
| Info->CurrentValue += StrLen (TempString); | |
| if (Info->CurrentValue[0] == L'\"') { | |
| Info->CurrentValue++; | |
| } | |
| FreePool (TempString); | |
| } | |
| } else { | |
| CurrentScriptFile->CurrentCommand->Data = NULL; | |
| // | |
| // find the matching endfor (we're done with the loop) | |
| // | |
| if (!MoveToTag (GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) { | |
| ShellPrintHiiEx ( | |
| -1, | |
| -1, | |
| NULL, | |
| STRING_TOKEN (STR_SYNTAX_NO_MATCHING), | |
| gShellLevel1HiiHandle, | |
| L"EndFor", | |
| L"For", | |
| CurrentScriptFile != NULL | |
| && CurrentScriptFile->CurrentCommand != NULL | |
| ? CurrentScriptFile->CurrentCommand->Line : 0 | |
| ); | |
| ShellStatus = SHELL_DEVICE_ERROR; | |
| } | |
| if (Info->RemoveSubstAlias) { | |
| // | |
| // remove item from list | |
| // | |
| InternalRemoveAliasFromList (Info->ReplacementName, &CurrentScriptFile->SubstList); | |
| } | |
| FreePool (Info); | |
| } | |
| } | |
| } | |
| if (ArgSet != NULL) { | |
| FreePool (ArgSet); | |
| } | |
| return (ShellStatus); | |
| } |