ShellPkg/Library/UefiShellLevel1CommandsLib/For.c

   1 /** @file
   2   Main file for endfor and for shell level 1 functions.
   3
   4   Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
   5   This program and the accompanying materials
   6   are licensed and made available under the terms and conditions of the BSD License
   7   which accompanies this distribution.  The full text of the license may be found at
   8   http://opensource.org/licenses/bsd-license.php
   9
  10   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  11   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  12
  13 **/
  14
  15 #include "UefiShellLevel1CommandsLib.h"
  16 #include <Library/PrintLib.h>
  17
  18 /**
  19   Determine if a valid string is a valid number for the 'for' command.
  20
  21   @param[in] Number The pointer to the string representation of the number to test.
  22
  23   @retval TRUE    The number is valid.
  24   @retval FALSE   The number is not valid.
  25 **/
  26 BOOLEAN
  27 EFIAPI
  28 ShellIsValidForNumber (
  29   IN CONST CHAR16 *Number
  30   )
  31 {
  32   if (Number == NULL || *Number == CHAR_NULL) {
  33     return (FALSE);
  34   }
  35
  36   if (*Number == L'-') {
  37     Number++;
  38   }
  39
  40   if (StrLen(Number) == 0) {
  41     return (FALSE);
  42   }
  43
  44   if (StrLen(Number) >= 7) {
  45     if (StrStr(Number, L" ") != NULL && (StrStr(Number, L" ") - Number) >= 7) {
  46       return (FALSE);
  47     }
  48   }
  49
  50   if (!ShellIsDecimalDigitCharacter(*Number)) {
  51     return (FALSE);
  52   }
  53
  54   return (TRUE);
  55 }
  56
  57 /**
  58   Function for 'endfor' command.
  59
  60   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
  61   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
  62 **/
  63 SHELL_STATUS
  64 EFIAPI
  65 ShellCommandRunEndFor (
  66   IN EFI_HANDLE        ImageHandle,
  67   IN EFI_SYSTEM_TABLE  *SystemTable
  68   )
  69 {
  70   EFI_STATUS          Status;
  71   BOOLEAN             Found;
  72   SCRIPT_FILE         *CurrentScriptFile;
  73
  74   Status = CommandInit();
  75   ASSERT_EFI_ERROR(Status);
  76
  77   if (!gEfiShellProtocol->BatchIsActive()) {
  78     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_NO_SCRIPT), gShellLevel1HiiHandle, L"EndFor");
  79     return (SHELL_UNSUPPORTED);
  80   }
  81
  82   if (gEfiShellParametersProtocol->Argc > 1) {
  83     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel1HiiHandle);
  84     return (SHELL_INVALID_PARAMETER);
  85   }
  86
  87   Found = MoveToTag(GetPreviousNode, L"for", L"endfor", NULL, ShellCommandGetCurrentScriptFile(), FALSE, FALSE, FALSE);
  88
  89   if (!Found) {
  90     CurrentScriptFile = ShellCommandGetCurrentScriptFile();
  91     ShellPrintHiiEx(
  92       -1,
  93       -1,
  94       NULL,
  95       STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
  96       gShellLevel1HiiHandle,
  97       L"For",
  98       L"EndFor",
  99       CurrentScriptFile!=NULL
 100         && CurrentScriptFile->CurrentCommand!=NULL
 101           ? CurrentScriptFile->CurrentCommand->Line:0);
 102     return (SHELL_NOT_FOUND);
 103   }
 104   return (SHELL_SUCCESS);
 105 }
 106
 107 typedef struct {
 108   UINT32          Signature;
 109   INTN            Current;
 110   INTN            End;
 111   INTN            Step;
 112   CHAR16          *ReplacementName;
 113   CHAR16          *CurrentValue;
 114   BOOLEAN         RemoveSubstAlias;
 115   CHAR16          Set[1];
 116   } SHELL_FOR_INFO;
 117 #define SIZE_OF_SHELL_FOR_INFO OFFSET_OF (SHELL_FOR_INFO, Set)
 118 #define SHELL_FOR_INFO_SIGNATURE SIGNATURE_32 ('S', 'F', 'I', 's')
 119
 120 /**
 121   Update the value of a given alias on the list.  If the alias is not there then add it.
 122
 123   @param[in] Alias              The alias to test for.
 124   @param[in] CommandString      The updated command string.
 125   @param[in,out] List           The list to search.
 126
 127   @retval EFI_SUCCESS           The operation was completed successfully.
 128   @retval EFI_OUT_OF_RESOURCES  There was not enough free memory.
 129 **/
 130 EFI_STATUS
 131 EFIAPI
 132 InternalUpdateAliasOnList(
 133   IN CONST CHAR16       *Alias,
 134   IN CONST CHAR16       *CommandString,
 135   IN OUT LIST_ENTRY     *List
 136   )
 137 {
 138   ALIAS_LIST *Node;
 139   BOOLEAN    Found;
 140
 141   //
 142   // assert for NULL parameter
 143   //
 144   ASSERT(Alias != NULL);
 145
 146   //
 147   // check for the Alias
 148   //
 149   for ( Node = (ALIAS_LIST *)GetFirstNode(List), Found = FALSE
 150       ; !IsNull(List, &Node->Link)
 151       ; Node = (ALIAS_LIST *)GetNextNode(List, &Node->Link)
 152      ){
 153     ASSERT(Node->CommandString != NULL);
 154     ASSERT(Node->Alias != NULL);
 155     if (StrCmp(Node->Alias, Alias)==0) {
 156       FreePool(Node->CommandString);
 157       Node->CommandString = NULL;
 158       Node->CommandString = StrnCatGrow(&Node->CommandString, NULL, CommandString, 0);
 159       Found = TRUE;
 160       break;
 161     }
 162   }
 163   if (!Found) {
 164     Node = AllocateZeroPool(sizeof(ALIAS_LIST));
 165     if (Node == NULL) {
 166       return (EFI_OUT_OF_RESOURCES);
 167     }
 168     ASSERT(Node->Alias == NULL);
 169     Node->Alias         = StrnCatGrow(&Node->Alias, NULL, Alias, 0);
 170     ASSERT(Node->CommandString == NULL);
 171     Node->CommandString = StrnCatGrow(&Node->CommandString, NULL, CommandString, 0);
 172     InsertTailList(List, &Node->Link);
 173   }
 174   return (EFI_SUCCESS);
 175 }
 176
 177 /**
 178   Find out if an alias is on the given list.
 179
 180   @param[in] Alias              The alias to test for.
 181   @param[in] List               The list to search.
 182
 183   @retval TRUE                  The alias is on the list.
 184   @retval FALSE                 The alias is not on the list.
 185 **/
 186 BOOLEAN
 187 EFIAPI
 188 InternalIsAliasOnList(
 189   IN CONST CHAR16       *Alias,
 190   IN CONST LIST_ENTRY   *List
 191   )
 192 {
 193   ALIAS_LIST *Node;
 194
 195   //
 196   // assert for NULL parameter
 197   //
 198   ASSERT(Alias != NULL);
 199
 200   //
 201   // check for the Alias
 202   //
 203   for ( Node = (ALIAS_LIST *)GetFirstNode(List)
 204       ; !IsNull(List, &Node->Link)
 205       ; Node = (ALIAS_LIST *)GetNextNode(List, &Node->Link)
 206      ){
 207     ASSERT(Node->CommandString != NULL);
 208     ASSERT(Node->Alias != NULL);
 209     if (StrCmp(Node->Alias, Alias)==0) {
 210       return (TRUE);
 211     }
 212   }
 213   return (FALSE);
 214 }
 215
 216 /**
 217   Remove an alias from the given list.
 218
 219   @param[in] Alias              The alias to remove.
 220   @param[in,out] List           The list to search.
 221 **/
 222 BOOLEAN
 223 EFIAPI
 224 InternalRemoveAliasFromList(
 225   IN CONST CHAR16       *Alias,
 226   IN OUT LIST_ENTRY     *List
 227   )
 228 {
 229   ALIAS_LIST *Node;
 230
 231   //
 232   // assert for NULL parameter
 233   //
 234   ASSERT(Alias != NULL);
 235
 236   //
 237   // check for the Alias
 238   //
 239   for ( Node = (ALIAS_LIST *)GetFirstNode(List)
 240       ; !IsNull(List, &Node->Link)
 241       ; Node = (ALIAS_LIST *)GetNextNode(List, &Node->Link)
 242      ){
 243     ASSERT(Node->CommandString != NULL);
 244     ASSERT(Node->Alias != NULL);
 245     if (StrCmp(Node->Alias, Alias)==0) {
 246       RemoveEntryList(&Node->Link);
 247       FreePool(Node->Alias);
 248       FreePool(Node->CommandString);
 249       FreePool(Node);
 250       return (TRUE);
 251     }
 252   }
 253   return (FALSE);
 254 }
 255
 256 /**
 257   Function for 'for' command.
 258
 259   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
 260   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
 261 **/
 262 SHELL_STATUS
 263 EFIAPI
 264 ShellCommandRunFor (
 265   IN EFI_HANDLE        ImageHandle,
 266   IN EFI_SYSTEM_TABLE  *SystemTable
 267   )
 268 {
 269   EFI_STATUS          Status;
 270   SHELL_STATUS        ShellStatus;
 271   SCRIPT_FILE         *CurrentScriptFile;
 272   CHAR16              *ArgSet;
 273   CHAR16              *ArgSetWalker;
 274   UINTN               ArgSize;
 275   UINTN               LoopVar;
 276   SHELL_FOR_INFO      *Info;
 277   CHAR16              *TempString;
 278   CHAR16              *TempSpot;
 279   BOOLEAN             FirstPass;
 280   EFI_SHELL_FILE_INFO *Node;
 281   EFI_SHELL_FILE_INFO *FileList;
 282   UINTN               NewSize;
 283
 284   ArgSet              = NULL;
 285   ArgSize             = 0;
 286   ShellStatus         = SHELL_SUCCESS;
 287   ArgSetWalker        = NULL;
 288   TempString          = NULL;
 289   FirstPass           = FALSE;
 290
 291   //
 292   // initialize the shell lib (we must be in non-auto-init...)
 293   //
 294   Status = ShellInitialize();
 295   ASSERT_EFI_ERROR(Status);
 296
 297   Status = CommandInit();
 298   ASSERT_EFI_ERROR(Status);
 299
 300   if (!gEfiShellProtocol->BatchIsActive()) {
 301     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_NO_SCRIPT), gShellLevel1HiiHandle, L"For");
 302     return (SHELL_UNSUPPORTED);
 303   }
 304
 305   if (gEfiShellParametersProtocol->Argc < 4) {
 306     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel1HiiHandle);
 307     return (SHELL_INVALID_PARAMETER);
 308   }
 309
 310   CurrentScriptFile = ShellCommandGetCurrentScriptFile();
 311   ASSERT(CurrentScriptFile != NULL);
 312
 313   if (CurrentScriptFile->CurrentCommand->Data == NULL) {
 314     FirstPass = TRUE;
 315
 316     //
 317     // Make sure that an End exists.
 318     //
 319     if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, TRUE, FALSE)) {
 320       ShellPrintHiiEx(
 321         -1,
 322         -1,
 323         NULL,
 324         STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 325         gShellLevel1HiiHandle,
 326         L"EndFor",
 327         L"For",
 328         CurrentScriptFile->CurrentCommand!=NULL
 329           ?CurrentScriptFile->CurrentCommand->Line:0);
 330       return (SHELL_DEVICE_ERROR);
 331     }
 332
 333     //
 334     // Process the line.
 335     //
 336     if (gEfiShellParametersProtocol->Argv[1][0] != L'%' || gEfiShellParametersProtocol->Argv[1][2] != CHAR_NULL
 337       ||!((gEfiShellParametersProtocol->Argv[1][1] >= L'a' && gEfiShellParametersProtocol->Argv[1][1] <= L'z')
 338        ||(gEfiShellParametersProtocol->Argv[1][1] >= L'A' && gEfiShellParametersProtocol->Argv[1][1] <= L'Z'))
 339      ) {
 340       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_VAR), gShellLevel1HiiHandle, gEfiShellParametersProtocol->Argv[1]);
 341       return (SHELL_INVALID_PARAMETER);
 342     }
 343
 344     if (gUnicodeCollation->StriColl(
 345         gUnicodeCollation,
 346         L"in",
 347         gEfiShellParametersProtocol->Argv[2]) == 0) {
 348       for (LoopVar = 0x3 ; LoopVar < gEfiShellParametersProtocol->Argc ; LoopVar++) {
 349         ASSERT((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL));
 350         if (ArgSet == NULL) {
 351   //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 352         } else {
 353           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" \"", 0);
 354         }
 355         if (StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"*") != NULL
 356           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"?") != NULL
 357           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"[") != NULL
 358           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"]") != NULL) {
 359           FileList = NULL;
 360           Status = ShellOpenFileMetaArg ((CHAR16*)gEfiShellParametersProtocol->Argv[LoopVar], EFI_FILE_MODE_READ, &FileList);
 361           if (EFI_ERROR(Status) || FileList == NULL || IsListEmpty(&FileList->Link)) {
 362             ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 363           } else {
 364             for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)
 365               ;  !IsNull(&FileList->Link, &Node->Link)
 366               ;  Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)
 367              ){
 368               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" \"", 0);
 369               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, Node->FullName, 0);
 370               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 371             }
 372             ShellCloseFileMetaArg(&FileList);
 373           }
 374         } else {
 375           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 376         }
 377         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 378       }
 379       //
 380       // set up for an 'in' for loop
 381       //
 382       NewSize = StrSize(ArgSet);
 383       NewSize += sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]);
 384       Info = AllocateZeroPool(NewSize);
 385       ASSERT(Info != NULL);
 386       Info->Signature = SHELL_FOR_INFO_SIGNATURE;
 387       CopyMem(Info->Set, ArgSet, StrSize(ArgSet));
 388       NewSize = StrSize(gEfiShellParametersProtocol->Argv[1]);
 389       CopyMem(Info->Set+(StrSize(ArgSet)/sizeof(Info->Set[0])), gEfiShellParametersProtocol->Argv[1], NewSize);
 390       Info->ReplacementName = Info->Set+StrSize(ArgSet)/sizeof(Info->Set[0]);
 391       Info->CurrentValue  = (CHAR16*)Info->Set;
 392       Info->Step          = 0;
 393       Info->Current       = 0;
 394       Info->End           = 0;
 395
 396       if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {
 397         Info->RemoveSubstAlias  = FALSE;
 398       } else {
 399         Info->RemoveSubstAlias  = TRUE;
 400       }
 401       CurrentScriptFile->CurrentCommand->Data = Info;
 402     } else if (gUnicodeCollation->StriColl(
 403         gUnicodeCollation,
 404         L"run",
 405         gEfiShellParametersProtocol->Argv[2]) == 0) {
 406       for (LoopVar = 0x3 ; LoopVar < gEfiShellParametersProtocol->Argc ; LoopVar++) {
 407         ASSERT((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL));
 408         if (ArgSet == NULL) {
 409 //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 410         } else {
 411           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);
 412         }
 413         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 414 //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);
 415       }
 416       //
 417       // set up for a 'run' for loop
 418       //
 419       Info = AllocateZeroPool(sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]));
 420       ASSERT(Info != NULL);
 421       CopyMem(Info->Set, gEfiShellParametersProtocol->Argv[1], StrSize(gEfiShellParametersProtocol->Argv[1]));
 422       Info->ReplacementName = Info->Set;
 423       Info->CurrentValue    = NULL;
 424       ArgSetWalker            = ArgSet;
 425       if (ArgSetWalker[0] != L'(') {
 426         ShellPrintHiiEx(
 427           -1,
 428           -1,
 429           NULL,
 430           STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 431           gShellLevel1HiiHandle,
 432           ArgSet,
 433           CurrentScriptFile!=NULL
 434             && CurrentScriptFile->CurrentCommand!=NULL
 435             ? CurrentScriptFile->CurrentCommand->Line:0);
 436         ShellStatus = SHELL_INVALID_PARAMETER;
 437       } else {
 438         TempSpot = StrStr(ArgSetWalker, L")");
 439         if (TempSpot != NULL) {
 440           TempString = TempSpot+1;
 441           if (*(TempString) != CHAR_NULL) {
 442             while(TempString != NULL && *TempString == L' ') {
 443               TempString++;
 444             }
 445             if (StrLen(TempString) > 0) {
 446               TempSpot = NULL;
 447             }
 448           }
 449         }
 450         if (TempSpot == NULL) {
 451           ShellPrintHiiEx(
 452             -1,
 453             -1,
 454             NULL,
 455             STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 456             gShellLevel1HiiHandle,
 457             CurrentScriptFile!=NULL
 458               && CurrentScriptFile->CurrentCommand!=NULL
 459               ? CurrentScriptFile->CurrentCommand->Line:0);
 460           ShellStatus = SHELL_INVALID_PARAMETER;
 461         } else {
 462           *TempSpot = CHAR_NULL;
 463           ArgSetWalker++;
 464           while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 465             ArgSetWalker++;
 466           }
 467           if (!ShellIsValidForNumber(ArgSetWalker)) {
 468             ShellPrintHiiEx(
 469               -1,
 470               -1,
 471               NULL,
 472               STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 473               gShellLevel1HiiHandle,
 474               ArgSet,
 475               CurrentScriptFile!=NULL
 476                 && CurrentScriptFile->CurrentCommand!=NULL
 477                 ? CurrentScriptFile->CurrentCommand->Line:0);
 478             ShellStatus = SHELL_INVALID_PARAMETER;
 479           } else {
 480             if (ArgSetWalker[0] == L'-') {
 481               Info->Current = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);
 482             } else {
 483               Info->Current = (INTN)ShellStrToUintn(ArgSetWalker);
 484             }
 485             ArgSetWalker  = StrStr(ArgSetWalker, L" ");
 486             while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 487               ArgSetWalker++;
 488             }
 489             if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){
 490               ShellPrintHiiEx(
 491                 -1,
 492                 -1,
 493                 NULL,
 494                 STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 495                 gShellLevel1HiiHandle,
 496                 ArgSet,
 497                 CurrentScriptFile!=NULL
 498                   && CurrentScriptFile->CurrentCommand!=NULL
 499                   ? CurrentScriptFile->CurrentCommand->Line:0);
 500               ShellStatus = SHELL_INVALID_PARAMETER;
 501             } else {
 502               if (ArgSetWalker[0] == L'-') {
 503                 Info->End = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);
 504               } else {
 505                 Info->End = (INTN)ShellStrToUintn(ArgSetWalker);
 506               }
 507               if (Info->Current < Info->End) {
 508                 Info->Step            = 1;
 509               } else {
 510                 Info->Step            = -1;
 511               }
 512
 513               ArgSetWalker  = StrStr(ArgSetWalker, L" ");
 514               while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 515                 ArgSetWalker++;
 516               }
 517               if (ArgSetWalker != NULL && *ArgSetWalker != CHAR_NULL) {
 518                 if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){
 519                   ShellPrintHiiEx(
 520                     -1,
 521                     -1,
 522                     NULL,
 523                     STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 524                     gShellLevel1HiiHandle,
 525                     ArgSet,
 526                     CurrentScriptFile!=NULL
 527                       && CurrentScriptFile->CurrentCommand!=NULL
 528                       ? CurrentScriptFile->CurrentCommand->Line:0);
 529                   ShellStatus = SHELL_INVALID_PARAMETER;
 530                 } else {
 531                   if (*ArgSetWalker == L')') {
 532                     ASSERT(Info->Step == 1 || Info->Step == -1);
 533                   } else {
 534                     if (ArgSetWalker[0] == L'-') {
 535                       Info->Step = 0 - (INTN)ShellStrToUintn(ArgSetWalker+1);
 536                     } else {
 537                       Info->Step = (INTN)ShellStrToUintn(ArgSetWalker);
 538                     }
 539
 540                     if (StrStr(ArgSetWalker, L" ") != NULL) {
 541                       ShellPrintHiiEx(
 542                         -1,
 543                         -1,
 544                         NULL,
 545                         STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 546                         gShellLevel1HiiHandle,
 547                         ArgSet,
 548                         CurrentScriptFile!=NULL
 549                           && CurrentScriptFile->CurrentCommand!=NULL
 550                           ? CurrentScriptFile->CurrentCommand->Line:0);
 551                       ShellStatus = SHELL_INVALID_PARAMETER;
 552                     }
 553                   }
 554                 }
 555
 556               }
 557             }
 558           }
 559         }
 560       }
 561       if (ShellStatus == SHELL_SUCCESS) {
 562         if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {
 563           Info->RemoveSubstAlias  = FALSE;
 564         } else {
 565           Info->RemoveSubstAlias  = TRUE;
 566         }
 567       }
 568       if (CurrentScriptFile->CurrentCommand != NULL) {
 569         CurrentScriptFile->CurrentCommand->Data = Info;
 570       }
 571     } else {
 572       ShellPrintHiiEx(
 573         -1,
 574         -1,
 575         NULL,
 576         STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 577         gShellLevel1HiiHandle,
 578         ArgSet,
 579         CurrentScriptFile!=NULL
 580           && CurrentScriptFile->CurrentCommand!=NULL
 581           ? CurrentScriptFile->CurrentCommand->Line:0);
 582       ShellStatus = SHELL_INVALID_PARAMETER;
 583     }
 584   } else {
 585     //
 586     // These need to be NULL since they are used to determine if this is the first pass later on...
 587     //
 588     ASSERT(ArgSetWalker == NULL);
 589     ASSERT(ArgSet       == NULL);
 590   }
 591
 592   Info = (SHELL_FOR_INFO*)CurrentScriptFile->CurrentCommand->Data;
 593   if (CurrentScriptFile->CurrentCommand->Reset) {
 594     Info->CurrentValue  = (CHAR16*)Info->Set;
 595     FirstPass = TRUE;
 596     CurrentScriptFile->CurrentCommand->Reset = FALSE;
 597   }
 598   if (ShellStatus == SHELL_SUCCESS) {
 599     ASSERT(Info != NULL);
 600     if (Info->Step != 0) {
 601       //
 602       // only advance if not the first pass
 603       //
 604       if (!FirstPass) {
 605         //
 606         // sequence version of for loop...
 607         //
 608         Info->Current += Info->Step;
 609       }
 610
 611       TempString = AllocateZeroPool(50*sizeof(CHAR16));
 612       UnicodeSPrint(TempString, 50*sizeof(CHAR16), L"%d", Info->Current);
 613       InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);
 614       FreePool(TempString);
 615
 616       if ((Info->Step > 0 && Info->Current > Info->End) || (Info->Step < 0 && Info->Current < Info->End)) {
 617         CurrentScriptFile->CurrentCommand->Data = NULL;
 618         //
 619         // find the matching endfor (we're done with the loop)
 620         //
 621         if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) {
 622           ShellPrintHiiEx(
 623             -1,
 624             -1,
 625             NULL,
 626             STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 627             gShellLevel1HiiHandle,
 628             L"EndFor",
 629             L"For",
 630             CurrentScriptFile!=NULL
 631               && CurrentScriptFile->CurrentCommand!=NULL
 632               ? CurrentScriptFile->CurrentCommand->Line:0);
 633           ShellStatus = SHELL_DEVICE_ERROR;
 634         }
 635         if (Info->RemoveSubstAlias) {
 636           //
 637           // remove item from list
 638           //
 639           InternalRemoveAliasFromList(Info->ReplacementName, &CurrentScriptFile->SubstList);
 640         }
 641         FreePool(Info);
 642       }
 643     } else {
 644       //
 645       // Must be in 'in' version of for loop...
 646       //
 647       ASSERT(Info->Set != NULL);
 648       if (Info->CurrentValue != NULL && *Info->CurrentValue != CHAR_NULL) {
 649         if (Info->CurrentValue[0] == L'\"') {
 650           Info->CurrentValue++;
 651         }
 652 //        while (Info->CurrentValue[0] == L' ') {
 653 //          Info->CurrentValue++;
 654 //        }
 655         if (Info->CurrentValue[0] == L'\"') {
 656           Info->CurrentValue++;
 657         }
 658         //
 659         // do the next one of the set
 660         //
 661         ASSERT(TempString == NULL);
 662         TempString = StrnCatGrow(&TempString, NULL, Info->CurrentValue, 0);
 663         TempSpot   = StrStr(TempString, L"\" \"");
 664         if (TempSpot != NULL) {
 665           *TempSpot = CHAR_NULL;
 666         }
 667         while (TempString[StrLen(TempString)-1] == L'\"') {
 668           TempString[StrLen(TempString)-1] = CHAR_NULL;
 669         }
 670         InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);
 671         Info->CurrentValue += StrLen(TempString);
 672
 673         if (Info->CurrentValue[0] == L'\"') {
 674           Info->CurrentValue++;
 675         }
 676         while (Info->CurrentValue[0] == L' ') {
 677           Info->CurrentValue++;
 678         }
 679         if (Info->CurrentValue[0] == L'\"') {
 680           Info->CurrentValue++;
 681         }
 682         FreePool(TempString);
 683
 684       } else {
 685         CurrentScriptFile->CurrentCommand->Data = NULL;
 686         //
 687         // find the matching endfor (we're done with the loop)
 688         //
 689         if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) {
 690           ShellPrintHiiEx(
 691             -1,
 692             -1,
 693             NULL,
 694             STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 695             gShellLevel1HiiHandle,
 696             L"EndFor",
 697             L"For",
 698             CurrentScriptFile!=NULL
 699               && CurrentScriptFile->CurrentCommand!=NULL
 700               ? CurrentScriptFile->CurrentCommand->Line:0);
 701           ShellStatus = SHELL_DEVICE_ERROR;
 702         }
 703         if (Info->RemoveSubstAlias) {
 704           //
 705           // remove item from list
 706           //
 707           InternalRemoveAliasFromList(Info->ReplacementName, &CurrentScriptFile->SubstList);
 708         }
 709         FreePool(Info);
 710       }
 711     }
 712   }
 713   if (ArgSet != NULL) {
 714     FreePool(ArgSet);
 715   }
 716   return (ShellStatus);
 717 }
 718