ShellPkg/Library/UefiShellLevel1CommandsLib/For.c

   1 /** @file
   2   Main file for endfor and for shell level 1 functions.
   3
   4   Copyright (c) 2009 - 2012, 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" ") != 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 to determine whether a string is decimal or hex representation of a number
 258   and return the number converted from the string.
 259
 260   @param[in] String   String representation of a number
 261
 262   @return             the number
 263   @retval (UINTN)(-1) An error ocurred.
 264 **/
 265 UINTN
 266 EFIAPI
 267 ReturnUintn(
 268   IN CONST CHAR16 *String
 269   )
 270 {
 271   UINT64        RetVal;
 272
 273   if (!EFI_ERROR(ShellConvertStringToUint64(String, &RetVal, FALSE, FALSE))) {
 274     return ((UINTN)RetVal);
 275   }
 276   return ((UINTN)(-1));
 277 }
 278
 279 /**
 280   Function for 'for' command.
 281
 282   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
 283   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
 284 **/
 285 SHELL_STATUS
 286 EFIAPI
 287 ShellCommandRunFor (
 288   IN EFI_HANDLE        ImageHandle,
 289   IN EFI_SYSTEM_TABLE  *SystemTable
 290   )
 291 {
 292   EFI_STATUS          Status;
 293   SHELL_STATUS        ShellStatus;
 294   SCRIPT_FILE         *CurrentScriptFile;
 295   CHAR16              *ArgSet;
 296   CHAR16              *ArgSetWalker;
 297   UINTN               ArgSize;
 298   UINTN               LoopVar;
 299   SHELL_FOR_INFO      *Info;
 300   CHAR16              *TempString;
 301   CHAR16              *TempSpot;
 302   BOOLEAN             FirstPass;
 303   EFI_SHELL_FILE_INFO *Node;
 304   EFI_SHELL_FILE_INFO *FileList;
 305   UINTN               NewSize;
 306
 307   ArgSet              = NULL;
 308   ArgSize             = 0;
 309   ShellStatus         = SHELL_SUCCESS;
 310   ArgSetWalker        = NULL;
 311   TempString          = NULL;
 312   FirstPass           = FALSE;
 313
 314   //
 315   // initialize the shell lib (we must be in non-auto-init...)
 316   //
 317   Status = ShellInitialize();
 318   ASSERT_EFI_ERROR(Status);
 319
 320   Status = CommandInit();
 321   ASSERT_EFI_ERROR(Status);
 322
 323   if (!gEfiShellProtocol->BatchIsActive()) {
 324     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_NO_SCRIPT), gShellLevel1HiiHandle, L"For");
 325     return (SHELL_UNSUPPORTED);
 326   }
 327
 328   if (gEfiShellParametersProtocol->Argc < 4) {
 329     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel1HiiHandle);
 330     return (SHELL_INVALID_PARAMETER);
 331   }
 332
 333   CurrentScriptFile = ShellCommandGetCurrentScriptFile();
 334   ASSERT(CurrentScriptFile != NULL);
 335
 336   if (CurrentScriptFile->CurrentCommand->Data == NULL) {
 337     FirstPass = TRUE;
 338
 339     //
 340     // Make sure that an End exists.
 341     //
 342     if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, TRUE, FALSE)) {
 343       ShellPrintHiiEx(
 344         -1,
 345         -1,
 346         NULL,
 347         STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 348         gShellLevel1HiiHandle,
 349         L"EndFor",
 350         L"For",
 351         CurrentScriptFile->CurrentCommand!=NULL
 352           ?CurrentScriptFile->CurrentCommand->Line:0);
 353       return (SHELL_DEVICE_ERROR);
 354     }
 355
 356     //
 357     // Process the line.
 358     //
 359     if (gEfiShellParametersProtocol->Argv[1][0] != L'%' || gEfiShellParametersProtocol->Argv[1][2] != CHAR_NULL
 360       ||!((gEfiShellParametersProtocol->Argv[1][1] >= L'a' && gEfiShellParametersProtocol->Argv[1][1] <= L'z')
 361        ||(gEfiShellParametersProtocol->Argv[1][1] >= L'A' && gEfiShellParametersProtocol->Argv[1][1] <= L'Z'))
 362      ) {
 363       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_INV_VAR), gShellLevel1HiiHandle, gEfiShellParametersProtocol->Argv[1]);
 364       return (SHELL_INVALID_PARAMETER);
 365     }
 366
 367     if (gUnicodeCollation->StriColl(
 368         gUnicodeCollation,
 369         L"in",
 370         gEfiShellParametersProtocol->Argv[2]) == 0) {
 371       for (LoopVar = 0x3 ; LoopVar < gEfiShellParametersProtocol->Argc ; LoopVar++) {
 372         ASSERT((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL));
 373         if (ArgSet == NULL) {
 374   //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 375         } else {
 376           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" \"", 0);
 377         }
 378         if (StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"*") != NULL
 379           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"?") != NULL
 380           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"[") != NULL
 381           ||StrStr(gEfiShellParametersProtocol->Argv[LoopVar], L"]") != NULL) {
 382           FileList = NULL;
 383           Status = ShellOpenFileMetaArg ((CHAR16*)gEfiShellParametersProtocol->Argv[LoopVar], EFI_FILE_MODE_READ, &FileList);
 384           if (EFI_ERROR(Status) || FileList == NULL || IsListEmpty(&FileList->Link)) {
 385             ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 386           } else {
 387             for (Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&FileList->Link)
 388               ;  !IsNull(&FileList->Link, &Node->Link)
 389               ;  Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&FileList->Link, &Node->Link)
 390              ){
 391               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" \"", 0);
 392               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, Node->FullName, 0);
 393               ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 394             }
 395             ShellCloseFileMetaArg(&FileList);
 396           }
 397         } else {
 398           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 399         }
 400         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 401       }
 402       if (ArgSet == NULL) {
 403         ShellStatus = SHELL_OUT_OF_RESOURCES;
 404       } else {
 405         //
 406         // set up for an 'in' for loop
 407         //
 408         NewSize = StrSize(ArgSet);
 409         NewSize += sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]);
 410         Info = AllocateZeroPool(NewSize);
 411         ASSERT(Info != NULL);
 412         Info->Signature = SHELL_FOR_INFO_SIGNATURE;
 413         CopyMem(Info->Set, ArgSet, StrSize(ArgSet));
 414         NewSize = StrSize(gEfiShellParametersProtocol->Argv[1]);
 415         CopyMem(Info->Set+(StrSize(ArgSet)/sizeof(Info->Set[0])), gEfiShellParametersProtocol->Argv[1], NewSize);
 416         Info->ReplacementName = Info->Set+StrSize(ArgSet)/sizeof(Info->Set[0]);
 417         Info->CurrentValue  = (CHAR16*)Info->Set;
 418         Info->Step          = 0;
 419         Info->Current       = 0;
 420         Info->End           = 0;
 421
 422         if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {
 423           Info->RemoveSubstAlias  = FALSE;
 424         } else {
 425           Info->RemoveSubstAlias  = TRUE;
 426         }
 427         CurrentScriptFile->CurrentCommand->Data = Info;
 428       }
 429     } else if (gUnicodeCollation->StriColl(
 430         gUnicodeCollation,
 431         L"run",
 432         gEfiShellParametersProtocol->Argv[2]) == 0) {
 433       for (LoopVar = 0x3 ; LoopVar < gEfiShellParametersProtocol->Argc ; LoopVar++) {
 434         ASSERT((ArgSet == NULL && ArgSize == 0) || (ArgSet != NULL));
 435         if (ArgSet == NULL) {
 436 //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L"\"", 0);
 437         } else {
 438           ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);
 439         }
 440         ArgSet = StrnCatGrow(&ArgSet, &ArgSize, gEfiShellParametersProtocol->Argv[LoopVar], 0);
 441 //        ArgSet = StrnCatGrow(&ArgSet, &ArgSize, L" ", 0);
 442       }
 443       if (ArgSet == NULL) {
 444         ShellStatus = SHELL_OUT_OF_RESOURCES;
 445       } else {
 446         //
 447         // set up for a 'run' for loop
 448         //
 449         Info = AllocateZeroPool(sizeof(SHELL_FOR_INFO)+StrSize(gEfiShellParametersProtocol->Argv[1]));
 450         ASSERT(Info != NULL);
 451         Info->Signature = SHELL_FOR_INFO_SIGNATURE;
 452         CopyMem(Info->Set, gEfiShellParametersProtocol->Argv[1], StrSize(gEfiShellParametersProtocol->Argv[1]));
 453         Info->ReplacementName = Info->Set;
 454         Info->CurrentValue    = NULL;
 455         ArgSetWalker            = ArgSet;
 456         if (ArgSetWalker[0] != L'(') {
 457           ShellPrintHiiEx(
 458             -1,
 459             -1,
 460             NULL,
 461             STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 462             gShellLevel1HiiHandle,
 463             ArgSet,
 464             CurrentScriptFile!=NULL
 465               && CurrentScriptFile->CurrentCommand!=NULL
 466               ? CurrentScriptFile->CurrentCommand->Line:0);
 467           ShellStatus = SHELL_INVALID_PARAMETER;
 468         } else {
 469           TempSpot = StrStr(ArgSetWalker, L")");
 470           if (TempSpot != NULL) {
 471             TempString = TempSpot+1;
 472             if (*(TempString) != CHAR_NULL) {
 473               while(TempString != NULL && *TempString == L' ') {
 474                 TempString++;
 475               }
 476               if (StrLen(TempString) > 0) {
 477                 TempSpot = NULL;
 478               }
 479             }
 480           }
 481           if (TempSpot == NULL) {
 482             ShellPrintHiiEx(
 483               -1,
 484               -1,
 485               NULL,
 486               STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 487               gShellLevel1HiiHandle,
 488               CurrentScriptFile!=NULL
 489                 && CurrentScriptFile->CurrentCommand!=NULL
 490                 ? CurrentScriptFile->CurrentCommand->Line:0);
 491             ShellStatus = SHELL_INVALID_PARAMETER;
 492           } else {
 493             *TempSpot = CHAR_NULL;
 494             ArgSetWalker++;
 495             while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 496               ArgSetWalker++;
 497             }
 498             if (!ShellIsValidForNumber(ArgSetWalker)) {
 499               ShellPrintHiiEx(
 500                 -1,
 501                 -1,
 502                 NULL,
 503                 STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 504                 gShellLevel1HiiHandle,
 505                 ArgSet,
 506                 CurrentScriptFile!=NULL
 507                   && CurrentScriptFile->CurrentCommand!=NULL
 508                   ? CurrentScriptFile->CurrentCommand->Line:0);
 509               ShellStatus = SHELL_INVALID_PARAMETER;
 510             } else {
 511               if (ArgSetWalker[0] == L'-') {
 512                 Info->Current = 0 - (INTN)ReturnUintn(ArgSetWalker+1);
 513               } else {
 514                 Info->Current = (INTN)ReturnUintn(ArgSetWalker);
 515               }
 516               ArgSetWalker  = StrStr(ArgSetWalker, L" ");
 517               while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 518                 ArgSetWalker++;
 519               }
 520               if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){
 521                 ShellPrintHiiEx(
 522                   -1,
 523                   -1,
 524                   NULL,
 525                   STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 526                   gShellLevel1HiiHandle,
 527                   ArgSet,
 528                   CurrentScriptFile!=NULL
 529                     && CurrentScriptFile->CurrentCommand!=NULL
 530                     ? CurrentScriptFile->CurrentCommand->Line:0);
 531                 ShellStatus = SHELL_INVALID_PARAMETER;
 532               } else {
 533                 if (ArgSetWalker[0] == L'-') {
 534                   Info->End = 0 - (INTN)ReturnUintn(ArgSetWalker+1);
 535                 } else {
 536                   Info->End = (INTN)ReturnUintn(ArgSetWalker);
 537                 }
 538                 if (Info->Current < Info->End) {
 539                   Info->Step            = 1;
 540                 } else {
 541                   Info->Step            = -1;
 542                 }
 543
 544                 ArgSetWalker  = StrStr(ArgSetWalker, L" ");
 545                 while (ArgSetWalker != NULL && ArgSetWalker[0] == L' ') {
 546                   ArgSetWalker++;
 547                 }
 548                 if (ArgSetWalker != NULL && *ArgSetWalker != CHAR_NULL) {
 549                   if (ArgSetWalker == NULL || *ArgSetWalker == CHAR_NULL || !ShellIsValidForNumber(ArgSetWalker)){
 550                     ShellPrintHiiEx(
 551                       -1,
 552                       -1,
 553                       NULL,
 554                       STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 555                       gShellLevel1HiiHandle,
 556                       ArgSet,
 557                       CurrentScriptFile!=NULL
 558                         && CurrentScriptFile->CurrentCommand!=NULL
 559                         ? CurrentScriptFile->CurrentCommand->Line:0);
 560                     ShellStatus = SHELL_INVALID_PARAMETER;
 561                   } else {
 562                     if (*ArgSetWalker == L')') {
 563                       ASSERT(Info->Step == 1 || Info->Step == -1);
 564                     } else {
 565                       if (ArgSetWalker[0] == L'-') {
 566                         Info->Step = 0 - (INTN)ReturnUintn(ArgSetWalker+1);
 567                       } else {
 568                         Info->Step = (INTN)ReturnUintn(ArgSetWalker);
 569                       }
 570
 571                       if (StrStr(ArgSetWalker, L" ") != NULL) {
 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                     }
 585                   }
 586
 587                 }
 588               }
 589             }
 590           }
 591         }
 592         if (ShellStatus == SHELL_SUCCESS) {
 593           if (InternalIsAliasOnList(Info->ReplacementName, &CurrentScriptFile->SubstList)) {
 594             Info->RemoveSubstAlias  = FALSE;
 595           } else {
 596             Info->RemoveSubstAlias  = TRUE;
 597           }
 598         }
 599         if (CurrentScriptFile->CurrentCommand != NULL) {
 600           CurrentScriptFile->CurrentCommand->Data = Info;
 601         }
 602       }
 603     } else {
 604       ShellPrintHiiEx(
 605         -1,
 606         -1,
 607         NULL,
 608         STRING_TOKEN (STR_GEN_PROBLEM_SCRIPT),
 609         gShellLevel1HiiHandle,
 610         ArgSet,
 611         CurrentScriptFile!=NULL
 612           && CurrentScriptFile->CurrentCommand!=NULL
 613           ? CurrentScriptFile->CurrentCommand->Line:0);
 614       ShellStatus = SHELL_INVALID_PARAMETER;
 615     }
 616   } else {
 617     //
 618     // These need to be NULL since they are used to determine if this is the first pass later on...
 619     //
 620     ASSERT(ArgSetWalker == NULL);
 621     ASSERT(ArgSet       == NULL);
 622   }
 623
 624   if (CurrentScriptFile != NULL && CurrentScriptFile->CurrentCommand != NULL) {
 625     Info = (SHELL_FOR_INFO*)CurrentScriptFile->CurrentCommand->Data;
 626     if (CurrentScriptFile->CurrentCommand->Reset) {
 627       Info->CurrentValue  = (CHAR16*)Info->Set;
 628       FirstPass = TRUE;
 629       CurrentScriptFile->CurrentCommand->Reset = FALSE;
 630     }
 631   } else {
 632     ShellStatus = SHELL_UNSUPPORTED;
 633     Info = NULL;
 634   }
 635   if (ShellStatus == SHELL_SUCCESS) {
 636     ASSERT(Info != NULL);
 637     if (Info->Step != 0) {
 638       //
 639       // only advance if not the first pass
 640       //
 641       if (!FirstPass) {
 642         //
 643         // sequence version of for loop...
 644         //
 645         Info->Current += Info->Step;
 646       }
 647
 648       TempString = AllocateZeroPool(50*sizeof(CHAR16));
 649       UnicodeSPrint(TempString, 50*sizeof(CHAR16), L"%d", Info->Current);
 650       InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);
 651       FreePool(TempString);
 652
 653       if ((Info->Step > 0 && Info->Current > Info->End) || (Info->Step < 0 && Info->Current < Info->End)) {
 654         CurrentScriptFile->CurrentCommand->Data = NULL;
 655         //
 656         // find the matching endfor (we're done with the loop)
 657         //
 658         if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) {
 659           ShellPrintHiiEx(
 660             -1,
 661             -1,
 662             NULL,
 663             STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 664             gShellLevel1HiiHandle,
 665             L"EndFor",
 666             L"For",
 667             CurrentScriptFile!=NULL
 668               && CurrentScriptFile->CurrentCommand!=NULL
 669               ? CurrentScriptFile->CurrentCommand->Line:0);
 670           ShellStatus = SHELL_DEVICE_ERROR;
 671         }
 672         if (Info->RemoveSubstAlias) {
 673           //
 674           // remove item from list
 675           //
 676           InternalRemoveAliasFromList(Info->ReplacementName, &CurrentScriptFile->SubstList);
 677         }
 678         FreePool(Info);
 679       }
 680     } else {
 681       //
 682       // Must be in 'in' version of for loop...
 683       //
 684       ASSERT(Info->Set != NULL);
 685       if (Info->CurrentValue != NULL && *Info->CurrentValue != CHAR_NULL) {
 686         if (Info->CurrentValue[0] == L' ') {
 687           Info->CurrentValue++;
 688         }
 689         if (Info->CurrentValue[0] == L'\"') {
 690           Info->CurrentValue++;
 691         }
 692         //
 693         // do the next one of the set
 694         //
 695         ASSERT(TempString == NULL);
 696         TempString = StrnCatGrow(&TempString, NULL, Info->CurrentValue, 0);
 697         if (TempString == NULL) {
 698           ShellStatus = SHELL_OUT_OF_RESOURCES;
 699         } else {
 700           TempSpot   = StrStr(TempString, L"\" \"");
 701           if (TempSpot != NULL) {
 702             *TempSpot = CHAR_NULL;
 703           }
 704           while (TempString[StrLen(TempString)-1] == L'\"') {
 705             TempString[StrLen(TempString)-1] = CHAR_NULL;
 706           }
 707           InternalUpdateAliasOnList(Info->ReplacementName, TempString, &CurrentScriptFile->SubstList);
 708           Info->CurrentValue += StrLen(TempString);
 709
 710           if (Info->CurrentValue[0] == L'\"') {
 711             Info->CurrentValue++;
 712           }
 713           while (Info->CurrentValue[0] == L' ') {
 714             Info->CurrentValue++;
 715           }
 716           if (Info->CurrentValue[0] == L'\"') {
 717             Info->CurrentValue++;
 718           }
 719           FreePool(TempString);
 720         }
 721       } else {
 722         CurrentScriptFile->CurrentCommand->Data = NULL;
 723         //
 724         // find the matching endfor (we're done with the loop)
 725         //
 726         if (!MoveToTag(GetNextNode, L"endfor", L"for", NULL, CurrentScriptFile, TRUE, FALSE, FALSE)) {
 727           ShellPrintHiiEx(
 728             -1,
 729             -1,
 730             NULL,
 731             STRING_TOKEN (STR_SYNTAX_NO_MATCHING),
 732             gShellLevel1HiiHandle,
 733             L"EndFor",
 734             L"For",
 735             CurrentScriptFile!=NULL
 736               && CurrentScriptFile->CurrentCommand!=NULL
 737               ? CurrentScriptFile->CurrentCommand->Line:0);
 738           ShellStatus = SHELL_DEVICE_ERROR;
 739         }
 740         if (Info->RemoveSubstAlias) {
 741           //
 742           // remove item from list
 743           //
 744           InternalRemoveAliasFromList(Info->ReplacementName, &CurrentScriptFile->SubstList);
 745         }
 746         FreePool(Info);
 747       }
 748     }
 749   }
 750   if (ArgSet != NULL) {
 751     FreePool(ArgSet);
 752   }
 753   return (ShellStatus);
 754 }
 755