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