2 This is THE shell (application)
4 Copyright (c) 2009 - 2016, Intel Corporation. All rights reserved.<BR>
5 (C) Copyright 2013-2014 Hewlett-Packard Development Company, L.P.<BR>
6 This program and the accompanying materials
7 are licensed and made available under the terms and conditions of the BSD License
8 which accompanies this distribution. The full text of the license may be found at
9 http://opensource.org/licenses/bsd-license.php
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
19 // Initialize the global structure
21 SHELL_INFO ShellInfoObject
= {
57 {{NULL
, NULL
}, NULL
, NULL
},
58 {{NULL
, NULL
}, NULL
, NULL
},
70 STATIC CONST CHAR16 mScriptExtension
[] = L
".NSH";
71 STATIC CONST CHAR16 mExecutableExtensions
[] = L
".NSH;.EFI";
72 STATIC CONST CHAR16 mStartupScript
[] = L
"startup.nsh";
73 CONST CHAR16 mNoNestingEnvVarName
[] = L
"nonesting";
74 CONST CHAR16 mNoNestingTrue
[] = L
"True";
75 CONST CHAR16 mNoNestingFalse
[] = L
"False";
78 Cleans off leading and trailing spaces and tabs.
80 @param[in] String pointer to the string to trim them off.
87 ASSERT(String
!= NULL
);
88 ASSERT(*String
!= NULL
);
90 // Remove any spaces and tabs at the beginning of the (*String).
92 while (((*String
)[0] == L
' ') || ((*String
)[0] == L
'\t')) {
93 CopyMem((*String
), (*String
)+1, StrSize((*String
)) - sizeof((*String
)[0]));
97 // Remove any spaces and tabs at the end of the (*String).
99 while ((StrLen (*String
) > 0) && (((*String
)[StrLen((*String
))-1] == L
' ') || ((*String
)[StrLen((*String
))-1] == L
'\t'))) {
100 (*String
)[StrLen((*String
))-1] = CHAR_NULL
;
103 return (EFI_SUCCESS
);
107 Parse for the next instance of one string within another string. Can optionally make sure that
108 the string was not escaped (^ character) per the shell specification.
110 @param[in] SourceString The string to search within
111 @param[in] FindString The string to look for
112 @param[in] CheckForEscapeCharacter TRUE to skip escaped instances of FinfString, otherwise will return even escaped instances
116 IN CONST CHAR16
*SourceString
,
117 IN CONST CHAR16
*FindString
,
118 IN CONST BOOLEAN CheckForEscapeCharacter
122 if (SourceString
== NULL
) {
125 Temp
= StrStr(SourceString
, FindString
);
128 // If nothing found, or we don't care about escape characters
130 if (Temp
== NULL
|| !CheckForEscapeCharacter
) {
135 // If we found an escaped character, try again on the remainder of the string
137 if ((Temp
> (SourceString
)) && *(Temp
-1) == L
'^') {
138 return FindNextInstance(Temp
+1, FindString
, CheckForEscapeCharacter
);
142 // we found the right character
148 Check whether the string between a pair of % is a valid environment variable name.
150 @param[in] BeginPercent pointer to the first percent.
151 @param[in] EndPercent pointer to the last percent.
153 @retval TRUE is a valid environment variable name.
154 @retval FALSE is NOT a valid environment variable name.
157 IsValidEnvironmentVariableName(
158 IN CONST CHAR16
*BeginPercent
,
159 IN CONST CHAR16
*EndPercent
162 CONST CHAR16
*Walker
;
166 ASSERT (BeginPercent
!= NULL
);
167 ASSERT (EndPercent
!= NULL
);
168 ASSERT (BeginPercent
< EndPercent
);
170 if ((BeginPercent
+ 1) == EndPercent
) {
174 for (Walker
= BeginPercent
+ 1; Walker
< EndPercent
; Walker
++) {
176 (*Walker
>= L
'0' && *Walker
<= L
'9') ||
177 (*Walker
>= L
'A' && *Walker
<= L
'Z') ||
178 (*Walker
>= L
'a' && *Walker
<= L
'z') ||
181 if (Walker
== BeginPercent
+ 1 && (*Walker
>= L
'0' && *Walker
<= L
'9')) {
195 Determine if a command line contains a split operation
197 @param[in] CmdLine The command line to parse.
199 @retval TRUE CmdLine has a valid split.
200 @retval FALSE CmdLine does not have a valid split.
204 IN CONST CHAR16
*CmdLine
207 CONST CHAR16
*TempSpot
;
208 CONST CHAR16
*FirstQuote
;
209 CONST CHAR16
*SecondQuote
;
211 FirstQuote
= FindNextInstance (CmdLine
, L
"\"", TRUE
);
213 TempSpot
= FindFirstCharacter(CmdLine
, L
"|", L
'^');
215 if (FirstQuote
== NULL
||
217 TempSpot
== CHAR_NULL
||
218 FirstQuote
> TempSpot
220 return (BOOLEAN
) ((TempSpot
!= NULL
) && (*TempSpot
!= CHAR_NULL
));
223 while ((TempSpot
!= NULL
) && (*TempSpot
!= CHAR_NULL
)) {
224 if (FirstQuote
== NULL
|| FirstQuote
> TempSpot
) {
227 SecondQuote
= FindNextInstance (FirstQuote
+ 1, L
"\"", TRUE
);
228 if (SecondQuote
== NULL
) {
231 if (SecondQuote
< TempSpot
) {
232 FirstQuote
= FindNextInstance (SecondQuote
+ 1, L
"\"", TRUE
);
235 FirstQuote
= FindNextInstance (SecondQuote
+ 1, L
"\"", TRUE
);
236 TempSpot
= FindFirstCharacter(TempSpot
+ 1, L
"|", L
'^');
241 return (BOOLEAN
) ((TempSpot
!= NULL
) && (*TempSpot
!= CHAR_NULL
));
245 Function to start monitoring for CTRL-S using SimpleTextInputEx. This
246 feature's enabled state was not known when the shell initially launched.
248 @retval EFI_SUCCESS The feature is enabled.
249 @retval EFI_OUT_OF_RESOURCES There is not enough memory available.
252 InternalEfiShellStartCtrlSMonitor(
256 EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL
*SimpleEx
;
257 EFI_KEY_DATA KeyData
;
260 Status
= gBS
->OpenProtocol(
261 gST
->ConsoleInHandle
,
262 &gEfiSimpleTextInputExProtocolGuid
,
266 EFI_OPEN_PROTOCOL_GET_PROTOCOL
);
267 if (EFI_ERROR(Status
)) {
272 STRING_TOKEN (STR_SHELL_NO_IN_EX
),
273 ShellInfoObject
.HiiHandle
);
274 return (EFI_SUCCESS
);
277 KeyData
.KeyState
.KeyToggleState
= 0;
278 KeyData
.Key
.ScanCode
= 0;
279 KeyData
.KeyState
.KeyShiftState
= EFI_SHIFT_STATE_VALID
|EFI_LEFT_CONTROL_PRESSED
;
280 KeyData
.Key
.UnicodeChar
= L
's';
282 Status
= SimpleEx
->RegisterKeyNotify(
285 NotificationFunction
,
286 &ShellInfoObject
.CtrlSNotifyHandle1
);
288 KeyData
.KeyState
.KeyShiftState
= EFI_SHIFT_STATE_VALID
|EFI_RIGHT_CONTROL_PRESSED
;
289 if (!EFI_ERROR(Status
)) {
290 Status
= SimpleEx
->RegisterKeyNotify(
293 NotificationFunction
,
294 &ShellInfoObject
.CtrlSNotifyHandle2
);
296 KeyData
.KeyState
.KeyShiftState
= EFI_SHIFT_STATE_VALID
|EFI_LEFT_CONTROL_PRESSED
;
297 KeyData
.Key
.UnicodeChar
= 19;
299 if (!EFI_ERROR(Status
)) {
300 Status
= SimpleEx
->RegisterKeyNotify(
303 NotificationFunction
,
304 &ShellInfoObject
.CtrlSNotifyHandle3
);
306 KeyData
.KeyState
.KeyShiftState
= EFI_SHIFT_STATE_VALID
|EFI_RIGHT_CONTROL_PRESSED
;
307 if (!EFI_ERROR(Status
)) {
308 Status
= SimpleEx
->RegisterKeyNotify(
311 NotificationFunction
,
312 &ShellInfoObject
.CtrlSNotifyHandle4
);
320 The entry point for the application.
322 @param[in] ImageHandle The firmware allocated handle for the EFI image.
323 @param[in] SystemTable A pointer to the EFI System Table.
325 @retval EFI_SUCCESS The entry point is executed successfully.
326 @retval other Some error occurs when executing this entry point.
332 IN EFI_HANDLE ImageHandle
,
333 IN EFI_SYSTEM_TABLE
*SystemTable
339 EFI_HANDLE ConInHandle
;
340 EFI_SIMPLE_TEXT_INPUT_PROTOCOL
*OldConIn
;
343 if (PcdGet8(PcdShellSupportLevel
) > 3) {
344 return (EFI_UNSUPPORTED
);
350 Status
= gST
->ConOut
->ClearScreen(gST
->ConOut
);
351 if (EFI_ERROR(Status
)) {
356 // Populate the global structure from PCDs
358 ShellInfoObject
.ImageDevPath
= NULL
;
359 ShellInfoObject
.FileDevPath
= NULL
;
360 ShellInfoObject
.PageBreakEnabled
= PcdGetBool(PcdShellPageBreakDefault
);
361 ShellInfoObject
.ViewingSettings
.InsertMode
= PcdGetBool(PcdShellInsertModeDefault
);
362 ShellInfoObject
.LogScreenCount
= PcdGet8 (PcdShellScreenLogCount
);
365 // verify we dont allow for spec violation
367 ASSERT(ShellInfoObject
.LogScreenCount
>= 3);
370 // Initialize the LIST ENTRY objects...
372 InitializeListHead(&ShellInfoObject
.BufferToFreeList
.Link
);
373 InitializeListHead(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
);
374 InitializeListHead(&ShellInfoObject
.SplitList
.Link
);
377 // Check PCDs for optional features that are not implemented yet.
379 if ( PcdGetBool(PcdShellSupportOldProtocols
)
380 || !FeaturePcdGet(PcdShellRequireHiiPlatform
)
381 || FeaturePcdGet(PcdShellSupportFrameworkHii
)
383 return (EFI_UNSUPPORTED
);
387 // turn off the watchdog timer
389 gBS
->SetWatchdogTimer (0, 0, 0, NULL
);
392 // install our console logger. This will keep a log of the output for back-browsing
394 Status
= ConsoleLoggerInstall(ShellInfoObject
.LogScreenCount
, &ShellInfoObject
.ConsoleInfo
);
395 if (!EFI_ERROR(Status
)) {
397 // Enable the cursor to be visible
399 gST
->ConOut
->EnableCursor (gST
->ConOut
, TRUE
);
402 // If supporting EFI 1.1 we need to install HII protocol
403 // only do this if PcdShellRequireHiiPlatform == FALSE
405 // remove EFI_UNSUPPORTED check above when complete.
406 ///@todo add support for Framework HII
409 // install our (solitary) HII package
411 ShellInfoObject
.HiiHandle
= HiiAddPackages (&gEfiCallerIdGuid
, gImageHandle
, ShellStrings
, NULL
);
412 if (ShellInfoObject
.HiiHandle
== NULL
) {
413 if (PcdGetBool(PcdShellSupportFrameworkHii
)) {
414 ///@todo Add our package into Framework HII
416 if (ShellInfoObject
.HiiHandle
== NULL
) {
417 Status
= EFI_NOT_STARTED
;
423 // create and install the EfiShellParametersProtocol
425 Status
= CreatePopulateInstallShellParametersProtocol(&ShellInfoObject
.NewShellParametersProtocol
, &ShellInfoObject
.RootShellInstance
);
426 ASSERT_EFI_ERROR(Status
);
427 ASSERT(ShellInfoObject
.NewShellParametersProtocol
!= NULL
);
430 // create and install the EfiShellProtocol
432 Status
= CreatePopulateInstallShellProtocol(&ShellInfoObject
.NewEfiShellProtocol
);
433 ASSERT_EFI_ERROR(Status
);
434 ASSERT(ShellInfoObject
.NewEfiShellProtocol
!= NULL
);
437 // Now initialize the shell library (it requires Shell Parameters protocol)
439 Status
= ShellInitialize();
440 ASSERT_EFI_ERROR(Status
);
442 Status
= CommandInit();
443 ASSERT_EFI_ERROR(Status
);
445 Status
= ShellInitEnvVarList ();
448 // Check the command line
450 Status
= ProcessCommandLine ();
451 if (EFI_ERROR (Status
)) {
456 // If shell support level is >= 1 create the mappings and paths
458 if (PcdGet8(PcdShellSupportLevel
) >= 1) {
459 Status
= ShellCommandCreateInitialMappingsAndPaths();
463 // Set the environment variable for nesting support
467 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoNest
) {
469 // No change. require nesting in Shell Protocol Execute()
471 StrnCatGrow(&TempString
,
476 StrnCatGrow(&TempString
,
481 Status
= InternalEfiShellSetEnv(mNoNestingEnvVarName
, TempString
, TRUE
);
482 SHELL_FREE_NON_NULL(TempString
);
486 // save the device path for the loaded image and the device path for the filepath (under loaded image)
487 // These are where to look for the startup.nsh file
489 Status
= GetDevicePathsForImageAndFile(&ShellInfoObject
.ImageDevPath
, &ShellInfoObject
.FileDevPath
);
490 ASSERT_EFI_ERROR(Status
);
493 // Display the version
495 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoVersion
) {
498 gST
->ConOut
->Mode
->CursorRow
,
500 STRING_TOKEN (STR_VER_OUTPUT_MAIN_SHELL
),
501 ShellInfoObject
.HiiHandle
,
502 SupportLevel
[PcdGet8(PcdShellSupportLevel
)],
503 gEfiShellProtocol
->MajorVersion
,
504 gEfiShellProtocol
->MinorVersion
511 STRING_TOKEN (STR_VER_OUTPUT_MAIN_SUPPLIER
),
512 ShellInfoObject
.HiiHandle
,
513 (CHAR16
*) PcdGetPtr (PcdShellSupplier
)
520 STRING_TOKEN (STR_VER_OUTPUT_MAIN_UEFI
),
521 ShellInfoObject
.HiiHandle
,
522 (gST
->Hdr
.Revision
&0xffff0000)>>16,
523 (gST
->Hdr
.Revision
&0x0000ffff),
525 gST
->FirmwareRevision
530 // Display the mapping
532 if (PcdGet8(PcdShellSupportLevel
) >= 2 && !ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoMap
) {
533 Status
= RunCommand(L
"map");
534 ASSERT_EFI_ERROR(Status
);
538 // init all the built in alias'
540 Status
= SetBuiltInAlias();
541 ASSERT_EFI_ERROR(Status
);
544 // Initialize environment variables
546 if (ShellCommandGetProfileList() != NULL
) {
547 Status
= InternalEfiShellSetEnv(L
"profiles", ShellCommandGetProfileList(), TRUE
);
548 ASSERT_EFI_ERROR(Status
);
552 TempString
= AllocateZeroPool(Size
);
554 UnicodeSPrint(TempString
, Size
, L
"%d", PcdGet8(PcdShellSupportLevel
));
555 Status
= InternalEfiShellSetEnv(L
"uefishellsupport", TempString
, TRUE
);
556 ASSERT_EFI_ERROR(Status
);
558 UnicodeSPrint(TempString
, Size
, L
"%d.%d", ShellInfoObject
.NewEfiShellProtocol
->MajorVersion
, ShellInfoObject
.NewEfiShellProtocol
->MinorVersion
);
559 Status
= InternalEfiShellSetEnv(L
"uefishellversion", TempString
, TRUE
);
560 ASSERT_EFI_ERROR(Status
);
562 UnicodeSPrint(TempString
, Size
, L
"%d.%d", (gST
->Hdr
.Revision
& 0xFFFF0000) >> 16, gST
->Hdr
.Revision
& 0x0000FFFF);
563 Status
= InternalEfiShellSetEnv(L
"uefiversion", TempString
, TRUE
);
564 ASSERT_EFI_ERROR(Status
);
566 FreePool(TempString
);
568 if (!EFI_ERROR(Status
)) {
569 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoInterrupt
) {
571 // Set up the event for CTRL-C monitoring...
573 Status
= InernalEfiShellStartMonitor();
576 if (!EFI_ERROR(Status
) && !ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
) {
578 // Set up the event for CTRL-S monitoring...
580 Status
= InternalEfiShellStartCtrlSMonitor();
583 if (!EFI_ERROR(Status
) && ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
) {
585 // close off the gST->ConIn
587 OldConIn
= gST
->ConIn
;
588 ConInHandle
= gST
->ConsoleInHandle
;
589 gST
->ConIn
= CreateSimpleTextInOnFile((SHELL_FILE_HANDLE
)&FileInterfaceNulFile
, &gST
->ConsoleInHandle
);
595 if (!EFI_ERROR(Status
) && PcdGet8(PcdShellSupportLevel
) >= 1) {
597 // process the startup script or launch the called app.
599 Status
= DoStartupScript(ShellInfoObject
.ImageDevPath
, ShellInfoObject
.FileDevPath
);
602 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Exit
&& !ShellCommandGetExit() && (PcdGet8(PcdShellSupportLevel
) >= 3 || PcdGetBool(PcdShellForceConsole
)) && !EFI_ERROR(Status
) && !ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
) {
604 // begin the UI waiting loop
608 // clean out all the memory allocated for CONST <something> * return values
609 // between each shell prompt presentation
611 if (!IsListEmpty(&ShellInfoObject
.BufferToFreeList
.Link
)){
612 FreeBufferList(&ShellInfoObject
.BufferToFreeList
);
616 // Reset page break back to default.
618 ShellInfoObject
.PageBreakEnabled
= PcdGetBool(PcdShellPageBreakDefault
);
619 ASSERT (ShellInfoObject
.ConsoleInfo
!= NULL
);
620 ShellInfoObject
.ConsoleInfo
->Enabled
= TRUE
;
621 ShellInfoObject
.ConsoleInfo
->RowCounter
= 0;
624 // Reset the CTRL-C event (yes we ignore the return values)
626 Status
= gBS
->CheckEvent (ShellInfoObject
.NewEfiShellProtocol
->ExecutionBreak
);
631 Status
= DoShellPrompt();
632 } while (!ShellCommandGetExit());
634 if (OldConIn
!= NULL
&& ConInHandle
!= NULL
) {
635 CloseSimpleTextInOnFile (gST
->ConIn
);
636 gST
->ConIn
= OldConIn
;
637 gST
->ConsoleInHandle
= ConInHandle
;
644 // uninstall protocols / free memory / etc...
646 if (ShellInfoObject
.UserBreakTimer
!= NULL
) {
647 gBS
->CloseEvent(ShellInfoObject
.UserBreakTimer
);
648 DEBUG_CODE(ShellInfoObject
.UserBreakTimer
= NULL
;);
650 if (ShellInfoObject
.ImageDevPath
!= NULL
) {
651 FreePool(ShellInfoObject
.ImageDevPath
);
652 DEBUG_CODE(ShellInfoObject
.ImageDevPath
= NULL
;);
654 if (ShellInfoObject
.FileDevPath
!= NULL
) {
655 FreePool(ShellInfoObject
.FileDevPath
);
656 DEBUG_CODE(ShellInfoObject
.FileDevPath
= NULL
;);
658 if (ShellInfoObject
.NewShellParametersProtocol
!= NULL
) {
659 CleanUpShellParametersProtocol(ShellInfoObject
.NewShellParametersProtocol
);
660 DEBUG_CODE(ShellInfoObject
.NewShellParametersProtocol
= NULL
;);
662 if (ShellInfoObject
.NewEfiShellProtocol
!= NULL
){
663 if (ShellInfoObject
.NewEfiShellProtocol
->IsRootShell()){
664 InternalEfiShellSetEnv(L
"cwd", NULL
, TRUE
);
666 CleanUpShellEnvironment (ShellInfoObject
.NewEfiShellProtocol
);
667 DEBUG_CODE(ShellInfoObject
.NewEfiShellProtocol
= NULL
;);
670 if (!IsListEmpty(&ShellInfoObject
.BufferToFreeList
.Link
)){
671 FreeBufferList(&ShellInfoObject
.BufferToFreeList
);
674 if (!IsListEmpty(&ShellInfoObject
.SplitList
.Link
)){
675 ASSERT(FALSE
); ///@todo finish this de-allocation (free SplitStdIn/Out when needed).
677 for ( Split
= (SPLIT_LIST
*)GetFirstNode (&ShellInfoObject
.SplitList
.Link
)
678 ; !IsNull (&ShellInfoObject
.SplitList
.Link
, &Split
->Link
)
679 ; Split
= (SPLIT_LIST
*)GetNextNode (&ShellInfoObject
.SplitList
.Link
, &Split
->Link
)
681 RemoveEntryList (&Split
->Link
);
685 DEBUG_CODE (InitializeListHead (&ShellInfoObject
.SplitList
.Link
););
688 if (ShellInfoObject
.ShellInitSettings
.FileName
!= NULL
) {
689 FreePool(ShellInfoObject
.ShellInitSettings
.FileName
);
690 DEBUG_CODE(ShellInfoObject
.ShellInitSettings
.FileName
= NULL
;);
693 if (ShellInfoObject
.ShellInitSettings
.FileOptions
!= NULL
) {
694 FreePool(ShellInfoObject
.ShellInitSettings
.FileOptions
);
695 DEBUG_CODE(ShellInfoObject
.ShellInitSettings
.FileOptions
= NULL
;);
698 if (ShellInfoObject
.HiiHandle
!= NULL
) {
699 HiiRemovePackages(ShellInfoObject
.HiiHandle
);
700 DEBUG_CODE(ShellInfoObject
.HiiHandle
= NULL
;);
703 if (!IsListEmpty(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
)){
704 FreeBufferList(&ShellInfoObject
.ViewingSettings
.CommandHistory
);
707 ASSERT(ShellInfoObject
.ConsoleInfo
!= NULL
);
708 if (ShellInfoObject
.ConsoleInfo
!= NULL
) {
709 ConsoleLoggerUninstall(ShellInfoObject
.ConsoleInfo
);
710 FreePool(ShellInfoObject
.ConsoleInfo
);
711 DEBUG_CODE(ShellInfoObject
.ConsoleInfo
= NULL
;);
714 ShellFreeEnvVarList ();
716 if (ShellCommandGetExit()) {
717 return ((EFI_STATUS
)ShellCommandGetExitCode());
723 Sets all the alias' that were registered with the ShellCommandLib library.
725 @retval EFI_SUCCESS all init commands were run successfully.
732 CONST ALIAS_LIST
*List
;
736 // Get all the commands we want to run
738 List
= ShellCommandGetInitAliasList();
741 // for each command in the List
743 for ( Node
= (ALIAS_LIST
*)GetFirstNode(&List
->Link
)
744 ; !IsNull (&List
->Link
, &Node
->Link
)
745 ; Node
= (ALIAS_LIST
*)GetNextNode(&List
->Link
, &Node
->Link
)
748 // install the alias'
750 Status
= InternalSetAlias(Node
->CommandString
, Node
->Alias
, TRUE
);
751 ASSERT_EFI_ERROR(Status
);
753 return (EFI_SUCCESS
);
757 Internal function to determine if 2 command names are really the same.
759 @param[in] Command1 The pointer to the first command name.
760 @param[in] Command2 The pointer to the second command name.
762 @retval TRUE The 2 command names are the same.
763 @retval FALSE The 2 command names are not the same.
767 IN CONST CHAR16
*Command1
,
768 IN CONST CHAR16
*Command2
771 if (StringNoCaseCompare(&Command1
, &Command2
) == 0) {
778 Internal function to determine if a command is a script only command.
780 @param[in] CommandName The pointer to the command name.
782 @retval TRUE The command is a script only command.
783 @retval FALSE The command is not a script only command.
787 IN CONST CHAR16
*CommandName
790 if (IsCommand(CommandName
, L
"for")
791 ||IsCommand(CommandName
, L
"endfor")
792 ||IsCommand(CommandName
, L
"if")
793 ||IsCommand(CommandName
, L
"else")
794 ||IsCommand(CommandName
, L
"endif")
795 ||IsCommand(CommandName
, L
"goto")) {
802 This function will populate the 2 device path protocol parameters based on the
803 global gImageHandle. The DevPath will point to the device path for the handle that has
804 loaded image protocol installed on it. The FilePath will point to the device path
805 for the file that was loaded.
807 @param[in, out] DevPath On a successful return the device path to the loaded image.
808 @param[in, out] FilePath On a successful return the device path to the file.
810 @retval EFI_SUCCESS The 2 device paths were successfully returned.
811 @retval other A error from gBS->HandleProtocol.
816 GetDevicePathsForImageAndFile (
817 IN OUT EFI_DEVICE_PATH_PROTOCOL
**DevPath
,
818 IN OUT EFI_DEVICE_PATH_PROTOCOL
**FilePath
822 EFI_LOADED_IMAGE_PROTOCOL
*LoadedImage
;
823 EFI_DEVICE_PATH_PROTOCOL
*ImageDevicePath
;
825 ASSERT(DevPath
!= NULL
);
826 ASSERT(FilePath
!= NULL
);
828 Status
= gBS
->OpenProtocol (
830 &gEfiLoadedImageProtocolGuid
,
831 (VOID
**)&LoadedImage
,
834 EFI_OPEN_PROTOCOL_GET_PROTOCOL
836 if (!EFI_ERROR (Status
)) {
837 Status
= gBS
->OpenProtocol (
838 LoadedImage
->DeviceHandle
,
839 &gEfiDevicePathProtocolGuid
,
840 (VOID
**)&ImageDevicePath
,
843 EFI_OPEN_PROTOCOL_GET_PROTOCOL
845 if (!EFI_ERROR (Status
)) {
846 *DevPath
= DuplicateDevicePath (ImageDevicePath
);
847 *FilePath
= DuplicateDevicePath (LoadedImage
->FilePath
);
849 LoadedImage
->DeviceHandle
,
850 &gEfiDevicePathProtocolGuid
,
856 &gEfiLoadedImageProtocolGuid
,
864 Process all Uefi Shell 2.0 command line options.
866 see Uefi Shell 2.0 section 3.2 for full details.
868 the command line must resemble the following:
870 shell.efi [ShellOpt-options] [options] [file-name [file-name-options]]
872 ShellOpt-options Options which control the initialization behavior of the shell.
873 These options are read from the EFI global variable "ShellOpt"
874 and are processed before options or file-name.
876 options Options which control the initialization behavior of the shell.
878 file-name The name of a UEFI shell application or script to be executed
879 after initialization is complete. By default, if file-name is
880 specified, then -nostartup is implied. Scripts are not supported
883 file-name-options The command-line options that are passed to file-name when it
886 This will initialize the ShellInfoObject.ShellInitSettings global variable.
888 @retval EFI_SUCCESS The variable is initialized.
898 CHAR16
*DelayValueStr
;
901 EFI_UNICODE_COLLATION_PROTOCOL
*UnicodeCollation
;
903 // `file-name-options` will contain arguments to `file-name` that we don't
904 // know about. This would cause ShellCommandLineParse to error, so we parse
905 // arguments manually, ignoring those after the first thing that doesn't look
906 // like a shell option (which is assumed to be `file-name`).
908 Status
= gBS
->LocateProtocol (
909 &gEfiUnicodeCollation2ProtocolGuid
,
911 (VOID
**) &UnicodeCollation
913 if (EFI_ERROR (Status
)) {
914 Status
= gBS
->LocateProtocol (
915 &gEfiUnicodeCollationProtocolGuid
,
917 (VOID
**) &UnicodeCollation
919 if (EFI_ERROR (Status
)) {
924 // Set default options
925 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Startup
= FALSE
;
926 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoStartup
= FALSE
;
927 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleOut
= FALSE
;
928 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
= FALSE
;
929 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoInterrupt
= FALSE
;
930 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoMap
= FALSE
;
931 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoVersion
= FALSE
;
932 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Delay
= FALSE
;
933 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Exit
= FALSE
;
934 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoNest
= FALSE
;
935 ShellInfoObject
.ShellInitSettings
.Delay
= 5;
938 // Start LoopVar at 0 to parse only optional arguments at Argv[0]
939 // and parse other parameters from Argv[1]. This is for use case that
940 // UEFI Shell boot option is created, and OptionalData is provided
941 // that starts with shell command-line options.
943 for (LoopVar
= 0 ; LoopVar
< gEfiShellParametersProtocol
->Argc
; LoopVar
++) {
944 CurrentArg
= gEfiShellParametersProtocol
->Argv
[LoopVar
];
945 if (UnicodeCollation
->StriColl (
950 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Startup
= TRUE
;
952 else if (UnicodeCollation
->StriColl (
957 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoStartup
= TRUE
;
959 else if (UnicodeCollation
->StriColl (
964 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleOut
= TRUE
;
966 else if (UnicodeCollation
->StriColl (
971 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
= TRUE
;
973 else if (UnicodeCollation
->StriColl (
978 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoInterrupt
= TRUE
;
980 else if (UnicodeCollation
->StriColl (
985 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoMap
= TRUE
;
987 else if (UnicodeCollation
->StriColl (
992 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoVersion
= TRUE
;
994 else if (UnicodeCollation
->StriColl (
999 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoNest
= TRUE
;
1001 else if (UnicodeCollation
->StriColl (
1006 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Delay
= TRUE
;
1007 // Check for optional delay value following "-delay"
1008 DelayValueStr
= gEfiShellParametersProtocol
->Argv
[LoopVar
+ 1];
1009 if (DelayValueStr
!= NULL
){
1010 if (*DelayValueStr
== L
':') {
1013 if (!EFI_ERROR(ShellConvertStringToUint64 (
1019 ShellInfoObject
.ShellInitSettings
.Delay
= (UINTN
)DelayValue
;
1023 } else if (UnicodeCollation
->StriColl (
1028 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Exit
= TRUE
;
1029 } else if (StrnCmp (L
"-", CurrentArg
, 1) == 0) {
1030 // Unrecognized option
1031 ShellPrintHiiEx(-1, -1, NULL
,
1032 STRING_TOKEN (STR_GEN_PROBLEM
),
1033 ShellInfoObject
.HiiHandle
,
1036 return EFI_INVALID_PARAMETER
;
1039 // First argument should be Shell.efi image name
1045 ShellInfoObject
.ShellInitSettings
.FileName
= AllocateCopyPool(StrSize(CurrentArg
), CurrentArg
);
1046 if (ShellInfoObject
.ShellInitSettings
.FileName
== NULL
) {
1047 return (EFI_OUT_OF_RESOURCES
);
1050 // We found `file-name`.
1052 ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoStartup
= 1;
1055 // Add `file-name-options`
1056 for (Size
= 0 ; LoopVar
< gEfiShellParametersProtocol
->Argc
; LoopVar
++) {
1057 ASSERT((ShellInfoObject
.ShellInitSettings
.FileOptions
== NULL
&& Size
== 0) || (ShellInfoObject
.ShellInitSettings
.FileOptions
!= NULL
));
1058 StrnCatGrow(&ShellInfoObject
.ShellInitSettings
.FileOptions
,
1062 if (ShellInfoObject
.ShellInitSettings
.FileOptions
== NULL
) {
1063 SHELL_FREE_NON_NULL(ShellInfoObject
.ShellInitSettings
.FileName
);
1064 return (EFI_OUT_OF_RESOURCES
);
1066 StrnCatGrow(&ShellInfoObject
.ShellInitSettings
.FileOptions
,
1068 gEfiShellParametersProtocol
->Argv
[LoopVar
],
1070 if (ShellInfoObject
.ShellInitSettings
.FileOptions
== NULL
) {
1071 SHELL_FREE_NON_NULL(ShellInfoObject
.ShellInitSettings
.FileName
);
1072 return (EFI_OUT_OF_RESOURCES
);
1078 // "-nointerrupt" overrides "-delay"
1079 if (ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoInterrupt
) {
1080 ShellInfoObject
.ShellInitSettings
.Delay
= 0;
1087 Handles all interaction with the default startup script.
1089 this will check that the correct command line parameters were passed, handle the delay, and then start running the script.
1091 @param ImagePath the path to the image for shell. first place to look for the startup script
1092 @param FilePath the path to the file for shell. second place to look for the startup script.
1094 @retval EFI_SUCCESS the variable is initialized.
1098 IN EFI_DEVICE_PATH_PROTOCOL
*ImagePath
,
1099 IN EFI_DEVICE_PATH_PROTOCOL
*FilePath
1103 EFI_STATUS CalleeStatus
;
1106 SHELL_FILE_HANDLE FileHandle
;
1107 EFI_DEVICE_PATH_PROTOCOL
*NewPath
;
1108 EFI_DEVICE_PATH_PROTOCOL
*NamePath
;
1109 CHAR16
*FileStringPath
;
1112 CONST CHAR16
*MapName
;
1114 Key
.UnicodeChar
= CHAR_NULL
;
1118 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Startup
&& ShellInfoObject
.ShellInitSettings
.FileName
!= NULL
) {
1120 // launch something else instead
1122 NewSize
= StrSize(ShellInfoObject
.ShellInitSettings
.FileName
);
1123 if (ShellInfoObject
.ShellInitSettings
.FileOptions
!= NULL
) {
1124 NewSize
+= StrSize(ShellInfoObject
.ShellInitSettings
.FileOptions
) + sizeof(CHAR16
);
1126 FileStringPath
= AllocateZeroPool(NewSize
);
1127 if (FileStringPath
== NULL
) {
1128 return (EFI_OUT_OF_RESOURCES
);
1130 StrCpyS(FileStringPath
, NewSize
/sizeof(CHAR16
), ShellInfoObject
.ShellInitSettings
.FileName
);
1131 if (ShellInfoObject
.ShellInitSettings
.FileOptions
!= NULL
) {
1132 StrnCatS(FileStringPath
, NewSize
/sizeof(CHAR16
), L
" ", NewSize
/sizeof(CHAR16
) - StrLen(FileStringPath
) -1);
1133 StrnCatS(FileStringPath
, NewSize
/sizeof(CHAR16
), ShellInfoObject
.ShellInitSettings
.FileOptions
, NewSize
/sizeof(CHAR16
) - StrLen(FileStringPath
) -1);
1135 Status
= RunShellCommand(FileStringPath
, &CalleeStatus
);
1136 if (ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Exit
== TRUE
) {
1137 ShellCommandRegisterExit(gEfiShellProtocol
->BatchIsActive(), (UINT64
)CalleeStatus
);
1139 FreePool(FileStringPath
);
1145 // for shell level 0 we do no scripts
1146 // Without the Startup bit overriding we allow for nostartup to prevent scripts
1148 if ( (PcdGet8(PcdShellSupportLevel
) < 1)
1149 || (ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoStartup
&& !ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.Startup
)
1151 return (EFI_SUCCESS
);
1154 gST
->ConOut
->EnableCursor(gST
->ConOut
, FALSE
);
1156 // print out our warning and see if they press a key
1158 for ( Status
= EFI_UNSUPPORTED
, Delay
= ShellInfoObject
.ShellInitSettings
.Delay
1159 ; Delay
!= 0 && EFI_ERROR(Status
)
1162 ShellPrintHiiEx(0, gST
->ConOut
->Mode
->CursorRow
, NULL
, STRING_TOKEN (STR_SHELL_STARTUP_QUESTION
), ShellInfoObject
.HiiHandle
, Delay
);
1163 gBS
->Stall (1000000);
1164 if (!ShellInfoObject
.ShellInitSettings
.BitUnion
.Bits
.NoConsoleIn
) {
1165 Status
= gST
->ConIn
->ReadKeyStroke (gST
->ConIn
, &Key
);
1168 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_CRLF
), ShellInfoObject
.HiiHandle
);
1169 gST
->ConOut
->EnableCursor(gST
->ConOut
, TRUE
);
1174 if (Status
== EFI_SUCCESS
&& Key
.UnicodeChar
== 0 && Key
.ScanCode
== SCAN_ESC
) {
1175 return (EFI_SUCCESS
);
1179 // Try the first location (must be file system)
1181 MapName
= ShellInfoObject
.NewEfiShellProtocol
->GetMapFromDevicePath(&ImagePath
);
1182 if (MapName
!= NULL
) {
1183 FileStringPath
= NULL
;
1185 FileStringPath
= StrnCatGrow(&FileStringPath
, &NewSize
, MapName
, 0);
1186 if (FileStringPath
== NULL
) {
1187 Status
= EFI_OUT_OF_RESOURCES
;
1189 TempSpot
= StrStr(FileStringPath
, L
";");
1190 if (TempSpot
!= NULL
) {
1191 *TempSpot
= CHAR_NULL
;
1193 FileStringPath
= StrnCatGrow(&FileStringPath
, &NewSize
, ((FILEPATH_DEVICE_PATH
*)FilePath
)->PathName
, 0);
1194 PathRemoveLastItem(FileStringPath
);
1195 FileStringPath
= StrnCatGrow(&FileStringPath
, &NewSize
, mStartupScript
, 0);
1196 Status
= ShellInfoObject
.NewEfiShellProtocol
->OpenFileByName(FileStringPath
, &FileHandle
, EFI_FILE_MODE_READ
);
1197 FreePool(FileStringPath
);
1200 if (EFI_ERROR(Status
)) {
1201 NamePath
= FileDevicePath (NULL
, mStartupScript
);
1202 NewPath
= AppendDevicePathNode (ImagePath
, NamePath
);
1208 Status
= InternalOpenFileDevicePath(NewPath
, &FileHandle
, EFI_FILE_MODE_READ
, 0);
1212 // If we got a file, run it
1214 if (!EFI_ERROR(Status
) && FileHandle
!= NULL
) {
1215 Status
= RunScriptFile (mStartupScript
, FileHandle
, L
"", ShellInfoObject
.NewShellParametersProtocol
);
1216 ShellInfoObject
.NewEfiShellProtocol
->CloseFile(FileHandle
);
1218 FileStringPath
= ShellFindFilePath(mStartupScript
);
1219 if (FileStringPath
== NULL
) {
1221 // we return success since we don't need to have a startup script
1223 Status
= EFI_SUCCESS
;
1224 ASSERT(FileHandle
== NULL
);
1226 Status
= RunScriptFile(FileStringPath
, NULL
, L
"", ShellInfoObject
.NewShellParametersProtocol
);
1227 FreePool(FileStringPath
);
1236 Function to perform the shell prompt looping. It will do a single prompt,
1237 dispatch the result, and then return. It is expected that the caller will
1238 call this function in a loop many times.
1241 @retval RETURN_ABORTED
1251 CONST CHAR16
*CurDir
;
1254 LIST_ENTRY OldBufferList
;
1259 // Get screen setting to decide size of the command line buffer
1261 gST
->ConOut
->QueryMode (gST
->ConOut
, gST
->ConOut
->Mode
->Mode
, &Column
, &Row
);
1262 BufferSize
= Column
* Row
* sizeof (CHAR16
);
1263 CmdLine
= AllocateZeroPool (BufferSize
);
1264 if (CmdLine
== NULL
) {
1265 return EFI_OUT_OF_RESOURCES
;
1268 SaveBufferList(&OldBufferList
);
1269 CurDir
= ShellInfoObject
.NewEfiShellProtocol
->GetEnv(L
"cwd");
1274 gST
->ConOut
->SetCursorPosition (gST
->ConOut
, 0, gST
->ConOut
->Mode
->CursorRow
);
1276 if (CurDir
!= NULL
&& StrLen(CurDir
) > 1) {
1277 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_CURDIR
), ShellInfoObject
.HiiHandle
, CurDir
);
1279 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_SHELL
), ShellInfoObject
.HiiHandle
);
1283 // Read a line from the console
1285 Status
= ShellInfoObject
.NewEfiShellProtocol
->ReadFile(ShellInfoObject
.NewShellParametersProtocol
->StdIn
, &BufferSize
, CmdLine
);
1288 // Null terminate the string and parse it
1290 if (!EFI_ERROR (Status
)) {
1291 CmdLine
[BufferSize
/ sizeof (CHAR16
)] = CHAR_NULL
;
1292 Status
= RunCommand(CmdLine
);
1296 // Done with this command
1298 RestoreBufferList(&OldBufferList
);
1304 Add a buffer to the Buffer To Free List for safely returning buffers to other
1305 places without risking letting them modify internal shell information.
1307 @param Buffer Something to pass to FreePool when the shell is exiting.
1310 AddBufferToFreeList (
1314 BUFFER_LIST
*BufferListEntry
;
1316 if (Buffer
== NULL
) {
1320 BufferListEntry
= AllocateZeroPool (sizeof (BUFFER_LIST
));
1321 if (BufferListEntry
== NULL
) {
1325 BufferListEntry
->Buffer
= Buffer
;
1326 InsertTailList (&ShellInfoObject
.BufferToFreeList
.Link
, &BufferListEntry
->Link
);
1332 Create a new buffer list and stores the old one to OldBufferList
1334 @param OldBufferList The temporary list head used to store the nodes in BufferToFreeList.
1338 OUT LIST_ENTRY
*OldBufferList
1341 CopyMem (OldBufferList
, &ShellInfoObject
.BufferToFreeList
.Link
, sizeof (LIST_ENTRY
));
1342 InitializeListHead (&ShellInfoObject
.BufferToFreeList
.Link
);
1346 Restore previous nodes into BufferToFreeList .
1348 @param OldBufferList The temporary list head used to store the nodes in BufferToFreeList.
1352 IN OUT LIST_ENTRY
*OldBufferList
1355 FreeBufferList (&ShellInfoObject
.BufferToFreeList
);
1356 CopyMem (&ShellInfoObject
.BufferToFreeList
.Link
, OldBufferList
, sizeof (LIST_ENTRY
));
1361 Add a buffer to the Line History List
1363 @param Buffer The line buffer to add.
1366 AddLineToCommandHistory(
1367 IN CONST CHAR16
*Buffer
1371 BUFFER_LIST
*Walker
;
1372 UINT16 MaxHistoryCmdCount
;
1376 MaxHistoryCmdCount
= PcdGet16(PcdShellMaxHistoryCommandCount
);
1378 if (MaxHistoryCmdCount
== 0) {
1383 Node
= AllocateZeroPool(sizeof(BUFFER_LIST
));
1388 Node
->Buffer
= AllocateCopyPool (StrSize (Buffer
), Buffer
);
1389 if (Node
->Buffer
== NULL
) {
1394 for ( Walker
= (BUFFER_LIST
*)GetFirstNode(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
)
1395 ; !IsNull(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
, &Walker
->Link
)
1396 ; Walker
= (BUFFER_LIST
*)GetNextNode(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
, &Walker
->Link
)
1400 if (Count
< MaxHistoryCmdCount
){
1401 InsertTailList(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
, &Node
->Link
);
1403 Walker
= (BUFFER_LIST
*)GetFirstNode(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
);
1404 RemoveEntryList(&Walker
->Link
);
1405 if (Walker
->Buffer
!= NULL
) {
1406 FreePool(Walker
->Buffer
);
1409 InsertTailList(&ShellInfoObject
.ViewingSettings
.CommandHistory
.Link
, &Node
->Link
);
1414 Checks if a string is an alias for another command. If yes, then it replaces the alias name
1415 with the correct command name.
1417 @param[in, out] CommandString Upon entry the potential alias. Upon return the
1418 command name if it was an alias. If it was not
1419 an alias it will be unchanged. This function may
1420 change the buffer to fit the command name.
1422 @retval EFI_SUCCESS The name was changed.
1423 @retval EFI_SUCCESS The name was not an alias.
1424 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
1428 IN OUT CHAR16
**CommandString
1431 CONST CHAR16
*NewString
;
1433 NewString
= ShellInfoObject
.NewEfiShellProtocol
->GetAlias(*CommandString
, NULL
);
1434 if (NewString
== NULL
) {
1435 return (EFI_SUCCESS
);
1437 FreePool(*CommandString
);
1438 *CommandString
= AllocateCopyPool(StrSize(NewString
), NewString
);
1439 if (*CommandString
== NULL
) {
1440 return (EFI_OUT_OF_RESOURCES
);
1442 return (EFI_SUCCESS
);
1446 This function will eliminate unreplaced (and therefore non-found) environment variables.
1448 @param[in,out] CmdLine The command line to update.
1451 StripUnreplacedEnvironmentVariables(
1452 IN OUT CHAR16
*CmdLine
1455 CHAR16
*FirstPercent
;
1457 CHAR16
*SecondPercent
;
1458 CHAR16
*SecondQuote
;
1459 CHAR16
*CurrentLocator
;
1461 for (CurrentLocator
= CmdLine
; CurrentLocator
!= NULL
; ) {
1462 FirstQuote
= FindNextInstance(CurrentLocator
, L
"\"", TRUE
);
1463 FirstPercent
= FindNextInstance(CurrentLocator
, L
"%", TRUE
);
1464 SecondPercent
= FirstPercent
!=NULL
?FindNextInstance(FirstPercent
+1, L
"%", TRUE
):NULL
;
1465 if (FirstPercent
== NULL
|| SecondPercent
== NULL
) {
1467 // If we ever don't have 2 % we are done.
1472 if (FirstQuote
!= NULL
&& FirstQuote
< FirstPercent
) {
1473 SecondQuote
= FindNextInstance(FirstQuote
+1, L
"\"", TRUE
);
1475 // Quote is first found
1478 if (SecondQuote
< FirstPercent
) {
1480 // restart after the pair of "
1482 CurrentLocator
= SecondQuote
+ 1;
1483 } else /* FirstPercent < SecondQuote */{
1485 // Restart on the first percent
1487 CurrentLocator
= FirstPercent
;
1492 if (FirstQuote
== NULL
|| SecondPercent
< FirstQuote
) {
1493 if (IsValidEnvironmentVariableName(FirstPercent
, SecondPercent
)) {
1495 // We need to remove from FirstPercent to SecondPercent
1497 CopyMem(FirstPercent
, SecondPercent
+ 1, StrSize(SecondPercent
+ 1));
1499 // don't need to update the locator. both % characters are gone.
1502 CurrentLocator
= SecondPercent
+ 1;
1506 CurrentLocator
= FirstQuote
;
1508 return (EFI_SUCCESS
);
1512 Function allocates a new command line and replaces all instances of environment
1513 variable names that are correctly preset to their values.
1515 If the return value is not NULL the memory must be caller freed.
1517 @param[in] OriginalCommandLine The original command line
1519 @retval NULL An error occurred.
1520 @return The new command line with no environment variables present.
1523 ShellConvertVariables (
1524 IN CONST CHAR16
*OriginalCommandLine
1527 CONST CHAR16
*MasterEnvList
;
1529 CHAR16
*NewCommandLine1
;
1530 CHAR16
*NewCommandLine2
;
1534 SCRIPT_FILE
*CurrentScriptFile
;
1535 ALIAS_LIST
*AliasListNode
;
1537 ASSERT(OriginalCommandLine
!= NULL
);
1540 NewSize
= StrSize(OriginalCommandLine
);
1541 CurrentScriptFile
= ShellCommandGetCurrentScriptFile();
1544 ///@todo update this to handle the %0 - %9 for scripting only (borrow from line 1256 area) ? ? ?
1547 // calculate the size required for the post-conversion string...
1549 if (CurrentScriptFile
!= NULL
) {
1550 for (AliasListNode
= (ALIAS_LIST
*)GetFirstNode(&CurrentScriptFile
->SubstList
)
1551 ; !IsNull(&CurrentScriptFile
->SubstList
, &AliasListNode
->Link
)
1552 ; AliasListNode
= (ALIAS_LIST
*)GetNextNode(&CurrentScriptFile
->SubstList
, &AliasListNode
->Link
)
1554 for (Temp
= StrStr(OriginalCommandLine
, AliasListNode
->Alias
)
1556 ; Temp
= StrStr(Temp
+1, AliasListNode
->Alias
)
1559 // we need a preceding and if there is space no ^ preceding (if no space ignore)
1561 if ((((Temp
-OriginalCommandLine
)>2) && *(Temp
-2) != L
'^') || ((Temp
-OriginalCommandLine
)<=2)) {
1562 NewSize
+= StrSize(AliasListNode
->CommandString
);
1568 for (MasterEnvList
= EfiShellGetEnv(NULL
)
1569 ; MasterEnvList
!= NULL
&& *MasterEnvList
!= CHAR_NULL
//&& *(MasterEnvList+1) != CHAR_NULL
1570 ; MasterEnvList
+= StrLen(MasterEnvList
) + 1
1572 if (StrSize(MasterEnvList
) > ItemSize
) {
1573 ItemSize
= StrSize(MasterEnvList
);
1575 for (Temp
= StrStr(OriginalCommandLine
, MasterEnvList
)
1577 ; Temp
= StrStr(Temp
+1, MasterEnvList
)
1580 // we need a preceding and following % and if there is space no ^ preceding (if no space ignore)
1582 if (*(Temp
-1) == L
'%' && *(Temp
+StrLen(MasterEnvList
)) == L
'%' &&
1583 ((((Temp
-OriginalCommandLine
)>2) && *(Temp
-2) != L
'^') || ((Temp
-OriginalCommandLine
)<=2))) {
1584 NewSize
+=StrSize(EfiShellGetEnv(MasterEnvList
));
1590 // now do the replacements...
1592 NewCommandLine1
= AllocateCopyPool(NewSize
, OriginalCommandLine
);
1593 NewCommandLine2
= AllocateZeroPool(NewSize
);
1594 ItemTemp
= AllocateZeroPool(ItemSize
+(2*sizeof(CHAR16
)));
1595 if (NewCommandLine1
== NULL
|| NewCommandLine2
== NULL
|| ItemTemp
== NULL
) {
1596 SHELL_FREE_NON_NULL(NewCommandLine1
);
1597 SHELL_FREE_NON_NULL(NewCommandLine2
);
1598 SHELL_FREE_NON_NULL(ItemTemp
);
1601 for (MasterEnvList
= EfiShellGetEnv(NULL
)
1602 ; MasterEnvList
!= NULL
&& *MasterEnvList
!= CHAR_NULL
1603 ; MasterEnvList
+= StrLen(MasterEnvList
) + 1
1606 ((ItemSize
+(2*sizeof(CHAR16
)))/sizeof(CHAR16
)),
1610 ((ItemSize
+(2*sizeof(CHAR16
)))/sizeof(CHAR16
)),
1614 ((ItemSize
+(2*sizeof(CHAR16
)))/sizeof(CHAR16
)),
1617 ShellCopySearchAndReplace(NewCommandLine1
, NewCommandLine2
, NewSize
, ItemTemp
, EfiShellGetEnv(MasterEnvList
), TRUE
, FALSE
);
1618 StrCpyS(NewCommandLine1
, NewSize
/sizeof(CHAR16
), NewCommandLine2
);
1620 if (CurrentScriptFile
!= NULL
) {
1621 for (AliasListNode
= (ALIAS_LIST
*)GetFirstNode(&CurrentScriptFile
->SubstList
)
1622 ; !IsNull(&CurrentScriptFile
->SubstList
, &AliasListNode
->Link
)
1623 ; AliasListNode
= (ALIAS_LIST
*)GetNextNode(&CurrentScriptFile
->SubstList
, &AliasListNode
->Link
)
1625 ShellCopySearchAndReplace(NewCommandLine1
, NewCommandLine2
, NewSize
, AliasListNode
->Alias
, AliasListNode
->CommandString
, TRUE
, FALSE
);
1626 StrCpyS(NewCommandLine1
, NewSize
/sizeof(CHAR16
), NewCommandLine2
);
1631 // Remove non-existent environment variables
1633 StripUnreplacedEnvironmentVariables(NewCommandLine1
);
1636 // Now cleanup any straggler intentionally ignored "%" characters
1638 ShellCopySearchAndReplace(NewCommandLine1
, NewCommandLine2
, NewSize
, L
"^%", L
"%", TRUE
, FALSE
);
1639 StrCpyS(NewCommandLine1
, NewSize
/sizeof(CHAR16
), NewCommandLine2
);
1641 FreePool(NewCommandLine2
);
1644 return (NewCommandLine1
);
1648 Internal function to run a command line with pipe usage.
1650 @param[in] CmdLine The pointer to the command line.
1651 @param[in] StdIn The pointer to the Standard input.
1652 @param[in] StdOut The pointer to the Standard output.
1654 @retval EFI_SUCCESS The split command is executed successfully.
1655 @retval other Some error occurs when executing the split command.
1659 IN CONST CHAR16
*CmdLine
,
1660 IN SHELL_FILE_HANDLE
*StdIn
,
1661 IN SHELL_FILE_HANDLE
*StdOut
1665 CHAR16
*NextCommandLine
;
1666 CHAR16
*OurCommandLine
;
1670 SHELL_FILE_HANDLE
*TempFileHandle
;
1673 ASSERT(StdOut
== NULL
);
1675 ASSERT(StrStr(CmdLine
, L
"|") != NULL
);
1677 Status
= EFI_SUCCESS
;
1678 NextCommandLine
= NULL
;
1679 OurCommandLine
= NULL
;
1683 NextCommandLine
= StrnCatGrow(&NextCommandLine
, &Size1
, StrStr(CmdLine
, L
"|")+1, 0);
1684 OurCommandLine
= StrnCatGrow(&OurCommandLine
, &Size2
, CmdLine
, StrStr(CmdLine
, L
"|") - CmdLine
);
1686 if (NextCommandLine
== NULL
|| OurCommandLine
== NULL
) {
1687 SHELL_FREE_NON_NULL(OurCommandLine
);
1688 SHELL_FREE_NON_NULL(NextCommandLine
);
1689 return (EFI_OUT_OF_RESOURCES
);
1690 } else if (StrStr(OurCommandLine
, L
"|") != NULL
|| Size1
== 0 || Size2
== 0) {
1691 SHELL_FREE_NON_NULL(OurCommandLine
);
1692 SHELL_FREE_NON_NULL(NextCommandLine
);
1693 return (EFI_INVALID_PARAMETER
);
1694 } else if (NextCommandLine
[0] == L
'a' &&
1695 (NextCommandLine
[1] == L
' ' || NextCommandLine
[1] == CHAR_NULL
)
1697 CopyMem(NextCommandLine
, NextCommandLine
+1, StrSize(NextCommandLine
) - sizeof(NextCommandLine
[0]));
1698 while (NextCommandLine
[0] == L
' ') {
1699 CopyMem(NextCommandLine
, NextCommandLine
+1, StrSize(NextCommandLine
) - sizeof(NextCommandLine
[0]));
1701 if (NextCommandLine
[0] == CHAR_NULL
) {
1702 SHELL_FREE_NON_NULL(OurCommandLine
);
1703 SHELL_FREE_NON_NULL(NextCommandLine
);
1704 return (EFI_INVALID_PARAMETER
);
1713 // make a SPLIT_LIST item and add to list
1715 Split
= AllocateZeroPool(sizeof(SPLIT_LIST
));
1716 if (Split
== NULL
) {
1717 return EFI_OUT_OF_RESOURCES
;
1719 Split
->SplitStdIn
= StdIn
;
1720 Split
->SplitStdOut
= ConvertEfiFileProtocolToShellHandle(CreateFileInterfaceMem(Unicode
), NULL
);
1721 ASSERT(Split
->SplitStdOut
!= NULL
);
1722 InsertHeadList(&ShellInfoObject
.SplitList
.Link
, &Split
->Link
);
1724 Status
= RunCommand(OurCommandLine
);
1727 // move the output from the first to the in to the second.
1729 TempFileHandle
= Split
->SplitStdOut
;
1730 if (Split
->SplitStdIn
== StdIn
) {
1731 Split
->SplitStdOut
= NULL
;
1733 Split
->SplitStdOut
= Split
->SplitStdIn
;
1735 Split
->SplitStdIn
= TempFileHandle
;
1736 ShellInfoObject
.NewEfiShellProtocol
->SetFilePosition(ConvertShellHandleToEfiFileProtocol(Split
->SplitStdIn
), 0);
1738 if (!EFI_ERROR(Status
)) {
1739 Status
= RunCommand(NextCommandLine
);
1743 // remove the top level from the ScriptList
1745 ASSERT((SPLIT_LIST
*)GetFirstNode(&ShellInfoObject
.SplitList
.Link
) == Split
);
1746 RemoveEntryList(&Split
->Link
);
1749 // Note that the original StdIn is now the StdOut...
1751 if (Split
->SplitStdOut
!= NULL
) {
1752 ShellInfoObject
.NewEfiShellProtocol
->CloseFile(ConvertShellHandleToEfiFileProtocol(Split
->SplitStdOut
));
1754 if (Split
->SplitStdIn
!= NULL
) {
1755 ShellInfoObject
.NewEfiShellProtocol
->CloseFile(ConvertShellHandleToEfiFileProtocol(Split
->SplitStdIn
));
1756 FreePool (Split
->SplitStdIn
);
1760 FreePool(NextCommandLine
);
1761 FreePool(OurCommandLine
);
1767 Take the original command line, substitute any variables, free
1768 the original string, return the modified copy.
1770 @param[in] CmdLine pointer to the command line to update.
1772 @retval EFI_SUCCESS the function was successful.
1773 @retval EFI_OUT_OF_RESOURCES a memory allocation failed.
1776 ShellSubstituteVariables(
1781 NewCmdLine
= ShellConvertVariables(*CmdLine
);
1782 SHELL_FREE_NON_NULL(*CmdLine
);
1783 if (NewCmdLine
== NULL
) {
1784 return (EFI_OUT_OF_RESOURCES
);
1786 *CmdLine
= NewCmdLine
;
1787 return (EFI_SUCCESS
);
1791 Take the original command line, substitute any alias in the first group of space delimited characters, free
1792 the original string, return the modified copy.
1794 @param[in] CmdLine pointer to the command line to update.
1796 @retval EFI_SUCCESS the function was successful.
1797 @retval EFI_OUT_OF_RESOURCES a memory allocation failed.
1800 ShellSubstituteAliases(
1805 CHAR16
*CommandName
;
1807 UINTN PostAliasSize
;
1808 ASSERT(CmdLine
!= NULL
);
1809 ASSERT(*CmdLine
!= NULL
);
1813 if (StrStr((*CmdLine
), L
" ") == NULL
){
1814 StrnCatGrow(&CommandName
, NULL
, (*CmdLine
), 0);
1816 StrnCatGrow(&CommandName
, NULL
, (*CmdLine
), StrStr((*CmdLine
), L
" ") - (*CmdLine
));
1820 // This cannot happen 'inline' since the CmdLine can need extra space.
1823 if (!ShellCommandIsCommandOnList(CommandName
)) {
1825 // Convert via alias
1827 Status
= ShellConvertAlias(&CommandName
);
1828 if (EFI_ERROR(Status
)){
1832 NewCmdLine
= StrnCatGrow(&NewCmdLine
, &PostAliasSize
, CommandName
, 0);
1833 if (NewCmdLine
== NULL
) {
1834 SHELL_FREE_NON_NULL(CommandName
);
1835 SHELL_FREE_NON_NULL(*CmdLine
);
1836 return (EFI_OUT_OF_RESOURCES
);
1838 NewCmdLine
= StrnCatGrow(&NewCmdLine
, &PostAliasSize
, StrStr((*CmdLine
), L
" "), 0);
1839 if (NewCmdLine
== NULL
) {
1840 SHELL_FREE_NON_NULL(CommandName
);
1841 SHELL_FREE_NON_NULL(*CmdLine
);
1842 return (EFI_OUT_OF_RESOURCES
);
1845 NewCmdLine
= StrnCatGrow(&NewCmdLine
, NULL
, (*CmdLine
), 0);
1848 SHELL_FREE_NON_NULL(*CmdLine
);
1849 SHELL_FREE_NON_NULL(CommandName
);
1852 // re-assign the passed in double pointer to point to our newly allocated buffer
1854 *CmdLine
= NewCmdLine
;
1856 return (EFI_SUCCESS
);
1860 Takes the Argv[0] part of the command line and determine the meaning of it.
1862 @param[in] CmdName pointer to the command line to update.
1864 @retval Internal_Command The name is an internal command.
1865 @retval File_Sys_Change the name is a file system change.
1866 @retval Script_File_Name the name is a NSH script file.
1867 @retval Unknown_Invalid the name is unknown.
1868 @retval Efi_Application the name is an application (.EFI).
1870 SHELL_OPERATION_TYPES
1872 IN CONST CHAR16
*CmdName
1875 CHAR16
* FileWithPath
;
1876 CONST CHAR16
* TempLocation
;
1877 CONST CHAR16
* TempLocation2
;
1879 FileWithPath
= NULL
;
1881 // test for an internal command.
1883 if (ShellCommandIsCommandOnList(CmdName
)) {
1884 return (Internal_Command
);
1888 // Test for file system change request. anything ending with first : and cant have spaces.
1890 if (CmdName
[(StrLen(CmdName
)-1)] == L
':') {
1891 if ( StrStr(CmdName
, L
" ") != NULL
1892 || StrLen(StrStr(CmdName
, L
":")) > 1
1894 return (Unknown_Invalid
);
1896 return (File_Sys_Change
);
1902 if ((FileWithPath
= ShellFindFilePathEx(CmdName
, mExecutableExtensions
)) != NULL
) {
1904 // See if that file has a script file extension
1906 if (StrLen(FileWithPath
) > 4) {
1907 TempLocation
= FileWithPath
+StrLen(FileWithPath
)-4;
1908 TempLocation2
= mScriptExtension
;
1909 if (StringNoCaseCompare((VOID
*)(&TempLocation
), (VOID
*)(&TempLocation2
)) == 0) {
1910 SHELL_FREE_NON_NULL(FileWithPath
);
1911 return (Script_File_Name
);
1916 // Was a file, but not a script. we treat this as an application.
1918 SHELL_FREE_NON_NULL(FileWithPath
);
1919 return (Efi_Application
);
1922 SHELL_FREE_NON_NULL(FileWithPath
);
1924 // No clue what this is... return invalid flag...
1926 return (Unknown_Invalid
);
1930 Determine if the first item in a command line is valid.
1932 @param[in] CmdLine The command line to parse.
1934 @retval EFI_SUCCESS The item is valid.
1935 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
1936 @retval EFI_NOT_FOUND The operation type is unknown or invalid.
1940 IN CONST CHAR16
*CmdLine
1944 CHAR16
*FirstParameter
;
1950 Temp
= StrnCatGrow(&Temp
, NULL
, CmdLine
, 0);
1952 return (EFI_OUT_OF_RESOURCES
);
1955 FirstParameter
= StrStr(Temp
, L
"|");
1956 if (FirstParameter
!= NULL
) {
1957 *FirstParameter
= CHAR_NULL
;
1960 FirstParameter
= NULL
;
1963 // Process the command line
1965 Status
= ProcessCommandLineToFinal(&Temp
);
1967 if (!EFI_ERROR(Status
)) {
1968 FirstParameter
= AllocateZeroPool(StrSize(CmdLine
));
1969 if (FirstParameter
== NULL
) {
1970 SHELL_FREE_NON_NULL(Temp
);
1971 return (EFI_OUT_OF_RESOURCES
);
1973 TempWalker
= (CHAR16
*)Temp
;
1974 if (!EFI_ERROR(GetNextParameter(&TempWalker
, &FirstParameter
, StrSize(CmdLine
), TRUE
))) {
1975 if (GetOperationType(FirstParameter
) == Unknown_Invalid
) {
1976 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_NOT_FOUND
), ShellInfoObject
.HiiHandle
, FirstParameter
);
1977 SetLastError(SHELL_NOT_FOUND
);
1978 Status
= EFI_NOT_FOUND
;
1983 SHELL_FREE_NON_NULL(Temp
);
1984 SHELL_FREE_NON_NULL(FirstParameter
);
1989 Determine if a command line contains with a split contains only valid commands.
1991 @param[in] CmdLine The command line to parse.
1993 @retval EFI_SUCCESS CmdLine has only valid commands, application, or has no split.
1994 @retval EFI_ABORTED CmdLine has at least one invalid command or application.
1998 IN CONST CHAR16
*CmdLine
2001 CONST CHAR16
*TempSpot
;
2005 // If this was the only item, then get out
2007 if (!ContainsSplit(CmdLine
)) {
2008 return (EFI_SUCCESS
);
2012 // Verify up to the pipe or end character
2014 Status
= IsValidSplit(CmdLine
);
2015 if (EFI_ERROR(Status
)) {
2020 // recurse to verify the next item
2022 TempSpot
= FindFirstCharacter(CmdLine
, L
"|", L
'^') + 1;
2023 if (*TempSpot
== L
'a' &&
2024 (*(TempSpot
+ 1) == L
' ' || *(TempSpot
+ 1) == CHAR_NULL
)
2026 // If it's an ASCII pipe '|a'
2030 return (VerifySplit(TempSpot
));
2034 Process a split based operation.
2036 @param[in] CmdLine pointer to the command line to process
2038 @retval EFI_SUCCESS The operation was successful
2039 @return an error occurred.
2042 ProcessNewSplitCommandLine(
2043 IN CONST CHAR16
*CmdLine
2049 Status
= VerifySplit(CmdLine
);
2050 if (EFI_ERROR(Status
)) {
2057 // are we in an existing split???
2059 if (!IsListEmpty(&ShellInfoObject
.SplitList
.Link
)) {
2060 Split
= (SPLIT_LIST
*)GetFirstNode(&ShellInfoObject
.SplitList
.Link
);
2063 if (Split
== NULL
) {
2064 Status
= RunSplitCommand(CmdLine
, NULL
, NULL
);
2066 Status
= RunSplitCommand(CmdLine
, Split
->SplitStdIn
, Split
->SplitStdOut
);
2068 if (EFI_ERROR(Status
)) {
2069 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_INVALID_SPLIT
), ShellInfoObject
.HiiHandle
, CmdLine
);
2075 Handle a request to change the current file system.
2077 @param[in] CmdLine The passed in command line.
2079 @retval EFI_SUCCESS The operation was successful.
2083 IN CONST CHAR16
*CmdLine
2087 Status
= EFI_SUCCESS
;
2090 // make sure we are the right operation
2092 ASSERT(CmdLine
[(StrLen(CmdLine
)-1)] == L
':' && StrStr(CmdLine
, L
" ") == NULL
);
2095 // Call the protocol API to do the work
2097 Status
= ShellInfoObject
.NewEfiShellProtocol
->SetCurDir(NULL
, CmdLine
);
2100 // Report any errors
2102 if (EFI_ERROR(Status
)) {
2103 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_INVALID_MAPPING
), ShellInfoObject
.HiiHandle
, CmdLine
);
2110 Reprocess the command line to direct all -? to the help command.
2112 if found, will add "help" as argv[0], and move the rest later.
2114 @param[in,out] CmdLine pointer to the command line to update
2118 IN OUT CHAR16
**CmdLine
2121 CHAR16
*CurrentParameter
;
2123 CHAR16
*NewCommandLine
;
2125 UINTN NewCmdLineSize
;
2127 Status
= EFI_SUCCESS
;
2129 CurrentParameter
= AllocateZeroPool(StrSize(*CmdLine
));
2130 if (CurrentParameter
== NULL
) {
2131 return (EFI_OUT_OF_RESOURCES
);
2135 while(Walker
!= NULL
&& *Walker
!= CHAR_NULL
) {
2136 if (!EFI_ERROR(GetNextParameter(&Walker
, &CurrentParameter
, StrSize(*CmdLine
), TRUE
))) {
2137 if (StrStr(CurrentParameter
, L
"-?") == CurrentParameter
) {
2138 CurrentParameter
[0] = L
' ';
2139 CurrentParameter
[1] = L
' ';
2140 NewCmdLineSize
= StrSize(L
"help ") + StrSize(*CmdLine
);
2141 NewCommandLine
= AllocateZeroPool(NewCmdLineSize
);
2142 if (NewCommandLine
== NULL
) {
2143 Status
= EFI_OUT_OF_RESOURCES
;
2148 // We know the space is sufficient since we just calculated it.
2150 StrnCpyS(NewCommandLine
, NewCmdLineSize
/sizeof(CHAR16
), L
"help ", 5);
2151 StrnCatS(NewCommandLine
, NewCmdLineSize
/sizeof(CHAR16
), *CmdLine
, StrLen(*CmdLine
));
2152 SHELL_FREE_NON_NULL(*CmdLine
);
2153 *CmdLine
= NewCommandLine
;
2159 SHELL_FREE_NON_NULL(CurrentParameter
);
2165 Function to update the shell variable "lasterror".
2167 @param[in] ErrorCode the error code to put into lasterror.
2171 IN CONST SHELL_STATUS ErrorCode
2174 CHAR16 LeString
[19];
2175 if (sizeof(EFI_STATUS
) == sizeof(UINT64
)) {
2176 UnicodeSPrint(LeString
, sizeof(LeString
), L
"0x%Lx", ErrorCode
);
2178 UnicodeSPrint(LeString
, sizeof(LeString
), L
"0x%x", ErrorCode
);
2180 DEBUG_CODE(InternalEfiShellSetEnv(L
"debuglasterror", LeString
, TRUE
););
2181 InternalEfiShellSetEnv(L
"lasterror", LeString
, TRUE
);
2183 return (EFI_SUCCESS
);
2187 Converts the command line to it's post-processed form. this replaces variables and alias' per UEFI Shell spec.
2189 @param[in,out] CmdLine pointer to the command line to update
2191 @retval EFI_SUCCESS The operation was successful
2192 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.
2193 @return some other error occurred
2196 ProcessCommandLineToFinal(
2197 IN OUT CHAR16
**CmdLine
2201 TrimSpaces(CmdLine
);
2203 Status
= ShellSubstituteAliases(CmdLine
);
2204 if (EFI_ERROR(Status
)) {
2208 TrimSpaces(CmdLine
);
2210 Status
= ShellSubstituteVariables(CmdLine
);
2211 if (EFI_ERROR(Status
)) {
2214 ASSERT (*CmdLine
!= NULL
);
2216 TrimSpaces(CmdLine
);
2219 // update for help parsing
2221 if (StrStr(*CmdLine
, L
"?") != NULL
) {
2223 // This may do nothing if the ? does not indicate help.
2224 // Save all the details for in the API below.
2226 Status
= DoHelpUpdate(CmdLine
);
2229 TrimSpaces(CmdLine
);
2231 return (EFI_SUCCESS
);
2235 Run an internal shell command.
2237 This API will update the shell's environment since these commands are libraries.
2239 @param[in] CmdLine the command line to run.
2240 @param[in] FirstParameter the first parameter on the command line
2241 @param[in] ParamProtocol the shell parameters protocol pointer
2242 @param[out] CommandStatus the status from the command line.
2244 @retval EFI_SUCCESS The command was completed.
2245 @retval EFI_ABORTED The command's operation was aborted.
2249 IN CONST CHAR16
*CmdLine
,
2250 IN CHAR16
*FirstParameter
,
2251 IN EFI_SHELL_PARAMETERS_PROTOCOL
*ParamProtocol
,
2252 OUT EFI_STATUS
*CommandStatus
2258 SHELL_STATUS CommandReturnedStatus
;
2263 NewCmdLine
= AllocateCopyPool (StrSize (CmdLine
), CmdLine
);
2264 if (NewCmdLine
== NULL
) {
2265 return EFI_OUT_OF_RESOURCES
;
2268 for (Walker
= NewCmdLine
; Walker
!= NULL
&& *Walker
!= CHAR_NULL
; Walker
++) {
2269 if (*Walker
== L
'^' && *(Walker
+1) == L
'#') {
2270 CopyMem(Walker
, Walker
+1, StrSize(Walker
) - sizeof(Walker
[0]));
2275 // get the argc and argv updated for internal commands
2277 Status
= UpdateArgcArgv(ParamProtocol
, NewCmdLine
, Internal_Command
, &Argv
, &Argc
);
2278 if (!EFI_ERROR(Status
)) {
2280 // Run the internal command.
2282 Status
= ShellCommandRunCommandHandler(FirstParameter
, &CommandReturnedStatus
, &LastError
);
2284 if (!EFI_ERROR(Status
)) {
2285 if (CommandStatus
!= NULL
) {
2286 if (CommandReturnedStatus
!= SHELL_SUCCESS
) {
2287 *CommandStatus
= (EFI_STATUS
)(CommandReturnedStatus
| MAX_BIT
);
2289 *CommandStatus
= EFI_SUCCESS
;
2294 // Update last error status.
2295 // some commands do not update last error.
2298 SetLastError(CommandReturnedStatus
);
2302 // Pass thru the exitcode from the app.
2304 if (ShellCommandGetExit()) {
2306 // An Exit was requested ("exit" command), pass its value up.
2308 Status
= CommandReturnedStatus
;
2309 } else if (CommandReturnedStatus
!= SHELL_SUCCESS
&& IsScriptOnlyCommand(FirstParameter
)) {
2311 // Always abort when a script only command fails for any reason
2313 Status
= EFI_ABORTED
;
2314 } else if (ShellCommandGetCurrentScriptFile() != NULL
&& CommandReturnedStatus
== SHELL_ABORTED
) {
2316 // Abort when in a script and a command aborted
2318 Status
= EFI_ABORTED
;
2324 // This is guaranteed to be called after UpdateArgcArgv no matter what else happened.
2325 // This is safe even if the update API failed. In this case, it may be a no-op.
2327 RestoreArgcArgv(ParamProtocol
, &Argv
, &Argc
);
2330 // If a script is running and the command is not a script only command, then
2331 // change return value to success so the script won't halt (unless aborted).
2333 // Script only commands have to be able halt the script since the script will
2334 // not operate if they are failing.
2336 if ( ShellCommandGetCurrentScriptFile() != NULL
2337 && !IsScriptOnlyCommand(FirstParameter
)
2338 && Status
!= EFI_ABORTED
2340 Status
= EFI_SUCCESS
;
2343 FreePool (NewCmdLine
);
2348 Function to run the command or file.
2350 @param[in] Type the type of operation being run.
2351 @param[in] CmdLine the command line to run.
2352 @param[in] FirstParameter the first parameter on the command line
2353 @param[in] ParamProtocol the shell parameters protocol pointer
2354 @param[out] CommandStatus the status from the command line.
2356 @retval EFI_SUCCESS The command was completed.
2357 @retval EFI_ABORTED The command's operation was aborted.
2361 IN SHELL_OPERATION_TYPES Type
,
2362 IN CONST CHAR16
*CmdLine
,
2363 IN CHAR16
*FirstParameter
,
2364 IN EFI_SHELL_PARAMETERS_PROTOCOL
*ParamProtocol
,
2365 OUT EFI_STATUS
*CommandStatus
2369 EFI_STATUS StartStatus
;
2370 CHAR16
*CommandWithPath
;
2371 EFI_DEVICE_PATH_PROTOCOL
*DevPath
;
2372 SHELL_STATUS CalleeExitStatus
;
2374 Status
= EFI_SUCCESS
;
2375 CommandWithPath
= NULL
;
2377 CalleeExitStatus
= SHELL_INVALID_PARAMETER
;
2380 case Internal_Command
:
2381 Status
= RunInternalCommand(CmdLine
, FirstParameter
, ParamProtocol
, CommandStatus
);
2383 case Script_File_Name
:
2384 case Efi_Application
:
2386 // Process a fully qualified path
2388 if (StrStr(FirstParameter
, L
":") != NULL
) {
2389 ASSERT (CommandWithPath
== NULL
);
2390 if (ShellIsFile(FirstParameter
) == EFI_SUCCESS
) {
2391 CommandWithPath
= StrnCatGrow(&CommandWithPath
, NULL
, FirstParameter
, 0);
2396 // Process a relative path and also check in the path environment variable
2398 if (CommandWithPath
== NULL
) {
2399 CommandWithPath
= ShellFindFilePathEx(FirstParameter
, mExecutableExtensions
);
2403 // This should be impossible now.
2405 ASSERT(CommandWithPath
!= NULL
);
2408 // Make sure that path is not just a directory (or not found)
2410 if (!EFI_ERROR(ShellIsDirectory(CommandWithPath
))) {
2411 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_NOT_FOUND
), ShellInfoObject
.HiiHandle
, FirstParameter
);
2412 SetLastError(SHELL_NOT_FOUND
);
2415 case Script_File_Name
:
2416 Status
= RunScriptFile (CommandWithPath
, NULL
, CmdLine
, ParamProtocol
);
2418 case Efi_Application
:
2420 // Get the device path of the application image
2422 DevPath
= ShellInfoObject
.NewEfiShellProtocol
->GetDevicePathFromFilePath(CommandWithPath
);
2423 if (DevPath
== NULL
){
2424 Status
= EFI_OUT_OF_RESOURCES
;
2429 // Execute the device path
2431 Status
= InternalShellExecuteDevicePath(
2439 SHELL_FREE_NON_NULL(DevPath
);
2441 if(EFI_ERROR (Status
)) {
2442 CalleeExitStatus
= (SHELL_STATUS
) (Status
& (~MAX_BIT
));
2444 CalleeExitStatus
= (SHELL_STATUS
) StartStatus
;
2447 if (CommandStatus
!= NULL
) {
2448 *CommandStatus
= CalleeExitStatus
;
2452 // Update last error status.
2454 // Status is an EFI_STATUS. Clear top bit to convert to SHELL_STATUS
2455 SetLastError(CalleeExitStatus
);
2471 SHELL_FREE_NON_NULL(CommandWithPath
);
2477 Function to setup StdIn, StdErr, StdOut, and then run the command or file.
2479 @param[in] Type the type of operation being run.
2480 @param[in] CmdLine the command line to run.
2481 @param[in] FirstParameter the first parameter on the command line.
2482 @param[in] ParamProtocol the shell parameters protocol pointer
2483 @param[out] CommandStatus the status from the command line.
2485 @retval EFI_SUCCESS The command was completed.
2486 @retval EFI_ABORTED The command's operation was aborted.
2489 SetupAndRunCommandOrFile(
2490 IN SHELL_OPERATION_TYPES Type
,
2492 IN CHAR16
*FirstParameter
,
2493 IN EFI_SHELL_PARAMETERS_PROTOCOL
*ParamProtocol
,
2494 OUT EFI_STATUS
*CommandStatus
2498 SHELL_FILE_HANDLE OriginalStdIn
;
2499 SHELL_FILE_HANDLE OriginalStdOut
;
2500 SHELL_FILE_HANDLE OriginalStdErr
;
2501 SYSTEM_TABLE_INFO OriginalSystemTableInfo
;
2502 CONST SCRIPT_FILE
*ConstScriptFile
;
2505 // Update the StdIn, StdOut, and StdErr for redirection to environment variables, files, etc... unicode and ASCII
2507 Status
= UpdateStdInStdOutStdErr(ParamProtocol
, CmdLine
, &OriginalStdIn
, &OriginalStdOut
, &OriginalStdErr
, &OriginalSystemTableInfo
);
2510 // The StdIn, StdOut, and StdErr are set up.
2511 // Now run the command, script, or application
2513 if (!EFI_ERROR(Status
)) {
2514 TrimSpaces(&CmdLine
);
2515 Status
= RunCommandOrFile(Type
, CmdLine
, FirstParameter
, ParamProtocol
, CommandStatus
);
2521 if (EFI_ERROR(Status
)) {
2522 ConstScriptFile
= ShellCommandGetCurrentScriptFile();
2523 if (ConstScriptFile
== NULL
|| ConstScriptFile
->CurrentCommand
== NULL
) {
2524 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_ERROR
), ShellInfoObject
.HiiHandle
, (VOID
*)(Status
));
2526 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_ERROR_SCRIPT
), ShellInfoObject
.HiiHandle
, (VOID
*)(Status
), ConstScriptFile
->CurrentCommand
->Line
);
2531 // put back the original StdIn, StdOut, and StdErr
2533 RestoreStdInStdOutStdErr(ParamProtocol
, &OriginalStdIn
, &OriginalStdOut
, &OriginalStdErr
, &OriginalSystemTableInfo
);
2539 Function will process and run a command line.
2541 This will determine if the command line represents an internal shell
2542 command or dispatch an external application.
2544 @param[in] CmdLine The command line to parse.
2545 @param[out] CommandStatus The status from the command line.
2547 @retval EFI_SUCCESS The command was completed.
2548 @retval EFI_ABORTED The command's operation was aborted.
2552 IN CONST CHAR16
*CmdLine
,
2553 OUT EFI_STATUS
*CommandStatus
2557 CHAR16
*CleanOriginal
;
2558 CHAR16
*FirstParameter
;
2560 SHELL_OPERATION_TYPES Type
;
2562 ASSERT(CmdLine
!= NULL
);
2563 if (StrLen(CmdLine
) == 0) {
2564 return (EFI_SUCCESS
);
2567 Status
= EFI_SUCCESS
;
2568 CleanOriginal
= NULL
;
2570 CleanOriginal
= StrnCatGrow(&CleanOriginal
, NULL
, CmdLine
, 0);
2571 if (CleanOriginal
== NULL
) {
2572 return (EFI_OUT_OF_RESOURCES
);
2575 TrimSpaces(&CleanOriginal
);
2578 // NULL out comments (leveraged from RunScriptFileHandle() ).
2579 // The # character on a line is used to denote that all characters on the same line
2580 // and to the right of the # are to be ignored by the shell.
2581 // Afterwards, again remove spaces, in case any were between the last command-parameter and '#'.
2583 for (TempWalker
= CleanOriginal
; TempWalker
!= NULL
&& *TempWalker
!= CHAR_NULL
; TempWalker
++) {
2584 if (*TempWalker
== L
'^') {
2585 if (*(TempWalker
+ 1) == L
'#') {
2588 } else if (*TempWalker
== L
'#') {
2589 *TempWalker
= CHAR_NULL
;
2593 TrimSpaces(&CleanOriginal
);
2596 // Handle case that passed in command line is just 1 or more " " characters.
2598 if (StrLen (CleanOriginal
) == 0) {
2599 SHELL_FREE_NON_NULL(CleanOriginal
);
2600 return (EFI_SUCCESS
);
2603 Status
= ProcessCommandLineToFinal(&CleanOriginal
);
2604 if (EFI_ERROR(Status
)) {
2605 SHELL_FREE_NON_NULL(CleanOriginal
);
2610 // We don't do normal processing with a split command line (output from one command input to another)
2612 if (ContainsSplit(CleanOriginal
)) {
2613 Status
= ProcessNewSplitCommandLine(CleanOriginal
);
2614 SHELL_FREE_NON_NULL(CleanOriginal
);
2619 // We need the first parameter information so we can determine the operation type
2621 FirstParameter
= AllocateZeroPool(StrSize(CleanOriginal
));
2622 if (FirstParameter
== NULL
) {
2623 SHELL_FREE_NON_NULL(CleanOriginal
);
2624 return (EFI_OUT_OF_RESOURCES
);
2626 TempWalker
= CleanOriginal
;
2627 if (!EFI_ERROR(GetNextParameter(&TempWalker
, &FirstParameter
, StrSize(CleanOriginal
), TRUE
))) {
2629 // Depending on the first parameter we change the behavior
2631 switch (Type
= GetOperationType(FirstParameter
)) {
2632 case File_Sys_Change
:
2633 Status
= ChangeMappedDrive (FirstParameter
);
2635 case Internal_Command
:
2636 case Script_File_Name
:
2637 case Efi_Application
:
2638 Status
= SetupAndRunCommandOrFile(Type
, CleanOriginal
, FirstParameter
, ShellInfoObject
.NewShellParametersProtocol
, CommandStatus
);
2642 // Whatever was typed, it was invalid.
2644 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_NOT_FOUND
), ShellInfoObject
.HiiHandle
, FirstParameter
);
2645 SetLastError(SHELL_NOT_FOUND
);
2649 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_NOT_FOUND
), ShellInfoObject
.HiiHandle
, FirstParameter
);
2650 SetLastError(SHELL_NOT_FOUND
);
2653 SHELL_FREE_NON_NULL(CleanOriginal
);
2654 SHELL_FREE_NON_NULL(FirstParameter
);
2660 Function will process and run a command line.
2662 This will determine if the command line represents an internal shell
2663 command or dispatch an external application.
2665 @param[in] CmdLine The command line to parse.
2667 @retval EFI_SUCCESS The command was completed.
2668 @retval EFI_ABORTED The command's operation was aborted.
2672 IN CONST CHAR16
*CmdLine
2675 return (RunShellCommand(CmdLine
, NULL
));
2679 STATIC CONST UINT16 InvalidChars
[] = {L
'*', L
'?', L
'<', L
'>', L
'\\', L
'/', L
'\"', 0x0001, 0x0002};
2681 Function determines if the CommandName COULD be a valid command. It does not determine whether
2682 this is a valid command. It only checks for invalid characters.
2684 @param[in] CommandName The name to check
2686 @retval TRUE CommandName could be a command name
2687 @retval FALSE CommandName could not be a valid command name
2691 IN CONST CHAR16
*CommandName
2695 if (CommandName
== NULL
) {
2700 ; Count
< sizeof(InvalidChars
) / sizeof(InvalidChars
[0])
2703 if (ScanMem16(CommandName
, StrSize(CommandName
), InvalidChars
[Count
]) != NULL
) {
2711 Function to process a NSH script file via SHELL_FILE_HANDLE.
2713 @param[in] Handle The handle to the already opened file.
2714 @param[in] Name The name of the script file.
2716 @retval EFI_SUCCESS the script completed successfully
2719 RunScriptFileHandle (
2720 IN SHELL_FILE_HANDLE Handle
,
2721 IN CONST CHAR16
*Name
2725 SCRIPT_FILE
*NewScriptFile
;
2727 UINTN PrintBuffSize
;
2728 CHAR16
*CommandLine
;
2729 CHAR16
*CommandLine2
;
2730 CHAR16
*CommandLine3
;
2731 SCRIPT_COMMAND_LIST
*LastCommand
;
2733 BOOLEAN PreScriptEchoState
;
2734 BOOLEAN PreCommandEchoState
;
2735 CONST CHAR16
*CurDir
;
2737 CHAR16 LeString
[50];
2738 LIST_ENTRY OldBufferList
;
2740 ASSERT(!ShellCommandGetScriptExit());
2742 PreScriptEchoState
= ShellCommandGetEchoState();
2743 PrintBuffSize
= PcdGet16(PcdShellPrintBufferSize
);
2745 NewScriptFile
= (SCRIPT_FILE
*)AllocateZeroPool(sizeof(SCRIPT_FILE
));
2746 if (NewScriptFile
== NULL
) {
2747 return (EFI_OUT_OF_RESOURCES
);
2753 ASSERT(NewScriptFile
->ScriptName
== NULL
);
2754 NewScriptFile
->ScriptName
= StrnCatGrow(&NewScriptFile
->ScriptName
, NULL
, Name
, 0);
2755 if (NewScriptFile
->ScriptName
== NULL
) {
2756 DeleteScriptFileStruct(NewScriptFile
);
2757 return (EFI_OUT_OF_RESOURCES
);
2761 // Save the parameters (used to replace %0 to %9 later on)
2763 NewScriptFile
->Argc
= ShellInfoObject
.NewShellParametersProtocol
->Argc
;
2764 if (NewScriptFile
->Argc
!= 0) {
2765 NewScriptFile
->Argv
= (CHAR16
**)AllocateZeroPool(NewScriptFile
->Argc
* sizeof(CHAR16
*));
2766 if (NewScriptFile
->Argv
== NULL
) {
2767 DeleteScriptFileStruct(NewScriptFile
);
2768 return (EFI_OUT_OF_RESOURCES
);
2770 for (LoopVar
= 0 ; LoopVar
< 10 && LoopVar
< NewScriptFile
->Argc
; LoopVar
++) {
2771 ASSERT(NewScriptFile
->Argv
[LoopVar
] == NULL
);
2772 NewScriptFile
->Argv
[LoopVar
] = StrnCatGrow(&NewScriptFile
->Argv
[LoopVar
], NULL
, ShellInfoObject
.NewShellParametersProtocol
->Argv
[LoopVar
], 0);
2773 if (NewScriptFile
->Argv
[LoopVar
] == NULL
) {
2774 DeleteScriptFileStruct(NewScriptFile
);
2775 return (EFI_OUT_OF_RESOURCES
);
2779 NewScriptFile
->Argv
= NULL
;
2782 InitializeListHead(&NewScriptFile
->CommandList
);
2783 InitializeListHead(&NewScriptFile
->SubstList
);
2786 // Now build the list of all script commands.
2789 while(!ShellFileHandleEof(Handle
)) {
2790 CommandLine
= ShellFileHandleReturnLine(Handle
, &Ascii
);
2792 if (CommandLine
== NULL
|| StrLen(CommandLine
) == 0 || CommandLine
[0] == '#') {
2793 SHELL_FREE_NON_NULL(CommandLine
);
2796 NewScriptFile
->CurrentCommand
= AllocateZeroPool(sizeof(SCRIPT_COMMAND_LIST
));
2797 if (NewScriptFile
->CurrentCommand
== NULL
) {
2798 SHELL_FREE_NON_NULL(CommandLine
);
2799 DeleteScriptFileStruct(NewScriptFile
);
2800 return (EFI_OUT_OF_RESOURCES
);
2803 NewScriptFile
->CurrentCommand
->Cl
= CommandLine
;
2804 NewScriptFile
->CurrentCommand
->Data
= NULL
;
2805 NewScriptFile
->CurrentCommand
->Line
= LineCount
;
2807 InsertTailList(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
);
2811 // Add this as the topmost script file
2813 ShellCommandSetNewScript (NewScriptFile
);
2816 // Now enumerate through the commands and run each one.
2818 CommandLine
= AllocateZeroPool(PrintBuffSize
);
2819 if (CommandLine
== NULL
) {
2820 DeleteScriptFileStruct(NewScriptFile
);
2821 return (EFI_OUT_OF_RESOURCES
);
2823 CommandLine2
= AllocateZeroPool(PrintBuffSize
);
2824 if (CommandLine2
== NULL
) {
2825 FreePool(CommandLine
);
2826 DeleteScriptFileStruct(NewScriptFile
);
2827 return (EFI_OUT_OF_RESOURCES
);
2830 for ( NewScriptFile
->CurrentCommand
= (SCRIPT_COMMAND_LIST
*)GetFirstNode(&NewScriptFile
->CommandList
)
2831 ; !IsNull(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
)
2832 ; // conditional increment in the body of the loop
2834 ASSERT(CommandLine2
!= NULL
);
2835 StrnCpyS( CommandLine2
,
2836 PrintBuffSize
/sizeof(CHAR16
),
2837 NewScriptFile
->CurrentCommand
->Cl
,
2838 PrintBuffSize
/sizeof(CHAR16
) - 1
2841 SaveBufferList(&OldBufferList
);
2844 // NULL out comments
2846 for (CommandLine3
= CommandLine2
; CommandLine3
!= NULL
&& *CommandLine3
!= CHAR_NULL
; CommandLine3
++) {
2847 if (*CommandLine3
== L
'^') {
2848 if ( *(CommandLine3
+1) == L
':') {
2849 CopyMem(CommandLine3
, CommandLine3
+1, StrSize(CommandLine3
) - sizeof(CommandLine3
[0]));
2850 } else if (*(CommandLine3
+1) == L
'#') {
2853 } else if (*CommandLine3
== L
'#') {
2854 *CommandLine3
= CHAR_NULL
;
2858 if (CommandLine2
!= NULL
&& StrLen(CommandLine2
) >= 1) {
2860 // Due to variability in starting the find and replace action we need to have both buffers the same.
2862 StrnCpyS( CommandLine
,
2863 PrintBuffSize
/sizeof(CHAR16
),
2865 PrintBuffSize
/sizeof(CHAR16
) - 1
2869 // Remove the %0 to %9 from the command line (if we have some arguments)
2871 if (NewScriptFile
->Argv
!= NULL
) {
2872 switch (NewScriptFile
->Argc
) {
2874 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%9", NewScriptFile
->Argv
[9], FALSE
, FALSE
);
2875 ASSERT_EFI_ERROR(Status
);
2877 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%8", NewScriptFile
->Argv
[8], FALSE
, FALSE
);
2878 ASSERT_EFI_ERROR(Status
);
2880 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%7", NewScriptFile
->Argv
[7], FALSE
, FALSE
);
2881 ASSERT_EFI_ERROR(Status
);
2883 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%6", NewScriptFile
->Argv
[6], FALSE
, FALSE
);
2884 ASSERT_EFI_ERROR(Status
);
2886 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%5", NewScriptFile
->Argv
[5], FALSE
, FALSE
);
2887 ASSERT_EFI_ERROR(Status
);
2889 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%4", NewScriptFile
->Argv
[4], FALSE
, FALSE
);
2890 ASSERT_EFI_ERROR(Status
);
2892 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%3", NewScriptFile
->Argv
[3], FALSE
, FALSE
);
2893 ASSERT_EFI_ERROR(Status
);
2895 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%2", NewScriptFile
->Argv
[2], FALSE
, FALSE
);
2896 ASSERT_EFI_ERROR(Status
);
2898 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%1", NewScriptFile
->Argv
[1], FALSE
, FALSE
);
2899 ASSERT_EFI_ERROR(Status
);
2901 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%0", NewScriptFile
->Argv
[0], FALSE
, FALSE
);
2902 ASSERT_EFI_ERROR(Status
);
2908 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%1", L
"\"\"", FALSE
, FALSE
);
2909 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%2", L
"\"\"", FALSE
, FALSE
);
2910 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%3", L
"\"\"", FALSE
, FALSE
);
2911 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%4", L
"\"\"", FALSE
, FALSE
);
2912 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%5", L
"\"\"", FALSE
, FALSE
);
2913 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%6", L
"\"\"", FALSE
, FALSE
);
2914 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%7", L
"\"\"", FALSE
, FALSE
);
2915 Status
= ShellCopySearchAndReplace(CommandLine
, CommandLine2
, PrintBuffSize
, L
"%8", L
"\"\"", FALSE
, FALSE
);
2916 Status
= ShellCopySearchAndReplace(CommandLine2
, CommandLine
, PrintBuffSize
, L
"%9", L
"\"\"", FALSE
, FALSE
);
2918 StrnCpyS( CommandLine2
,
2919 PrintBuffSize
/sizeof(CHAR16
),
2921 PrintBuffSize
/sizeof(CHAR16
) - 1
2924 LastCommand
= NewScriptFile
->CurrentCommand
;
2926 for (CommandLine3
= CommandLine2
; CommandLine3
[0] == L
' ' ; CommandLine3
++);
2928 if (CommandLine3
!= NULL
&& CommandLine3
[0] == L
':' ) {
2930 // This line is a goto target / label
2933 if (CommandLine3
!= NULL
&& StrLen(CommandLine3
) > 0) {
2934 if (CommandLine3
[0] == L
'@') {
2936 // We need to save the current echo state
2937 // and disable echo for just this command.
2939 PreCommandEchoState
= ShellCommandGetEchoState();
2940 ShellCommandSetEchoState(FALSE
);
2941 Status
= RunCommand(CommandLine3
+1);
2944 // If command was "@echo -off" or "@echo -on" then don't restore echo state
2946 if (StrCmp (L
"@echo -off", CommandLine3
) != 0 &&
2947 StrCmp (L
"@echo -on", CommandLine3
) != 0) {
2949 // Now restore the pre-'@' echo state.
2951 ShellCommandSetEchoState(PreCommandEchoState
);
2954 if (ShellCommandGetEchoState()) {
2955 CurDir
= ShellInfoObject
.NewEfiShellProtocol
->GetEnv(L
"cwd");
2956 if (CurDir
!= NULL
&& StrLen(CurDir
) > 1) {
2957 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_CURDIR
), ShellInfoObject
.HiiHandle
, CurDir
);
2959 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SHELL_SHELL
), ShellInfoObject
.HiiHandle
);
2961 ShellPrintEx(-1, -1, L
"%s\r\n", CommandLine2
);
2963 Status
= RunCommand(CommandLine3
);
2967 if (ShellCommandGetScriptExit()) {
2969 // ShellCommandGetExitCode() always returns a UINT64
2971 UnicodeSPrint(LeString
, sizeof(LeString
), L
"0x%Lx", ShellCommandGetExitCode());
2972 DEBUG_CODE(InternalEfiShellSetEnv(L
"debuglasterror", LeString
, TRUE
););
2973 InternalEfiShellSetEnv(L
"lasterror", LeString
, TRUE
);
2975 ShellCommandRegisterExit(FALSE
, 0);
2976 Status
= EFI_SUCCESS
;
2977 RestoreBufferList(&OldBufferList
);
2980 if (ShellGetExecutionBreakFlag()) {
2981 RestoreBufferList(&OldBufferList
);
2984 if (EFI_ERROR(Status
)) {
2985 RestoreBufferList(&OldBufferList
);
2988 if (ShellCommandGetExit()) {
2989 RestoreBufferList(&OldBufferList
);
2994 // If that commend did not update the CurrentCommand then we need to advance it...
2996 if (LastCommand
== NewScriptFile
->CurrentCommand
) {
2997 NewScriptFile
->CurrentCommand
= (SCRIPT_COMMAND_LIST
*)GetNextNode(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
);
2998 if (!IsNull(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
)) {
2999 NewScriptFile
->CurrentCommand
->Reset
= TRUE
;
3003 NewScriptFile
->CurrentCommand
= (SCRIPT_COMMAND_LIST
*)GetNextNode(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
);
3004 if (!IsNull(&NewScriptFile
->CommandList
, &NewScriptFile
->CurrentCommand
->Link
)) {
3005 NewScriptFile
->CurrentCommand
->Reset
= TRUE
;
3008 RestoreBufferList(&OldBufferList
);
3012 FreePool(CommandLine
);
3013 FreePool(CommandLine2
);
3014 ShellCommandSetNewScript (NULL
);
3017 // Only if this was the last script reset the state.
3019 if (ShellCommandGetCurrentScriptFile()==NULL
) {
3020 ShellCommandSetEchoState(PreScriptEchoState
);
3022 return (EFI_SUCCESS
);
3026 Function to process a NSH script file.
3028 @param[in] ScriptPath Pointer to the script file name (including file system path).
3029 @param[in] Handle the handle of the script file already opened.
3030 @param[in] CmdLine the command line to run.
3031 @param[in] ParamProtocol the shell parameters protocol pointer
3033 @retval EFI_SUCCESS the script completed successfully
3037 IN CONST CHAR16
*ScriptPath
,
3038 IN SHELL_FILE_HANDLE Handle OPTIONAL
,
3039 IN CONST CHAR16
*CmdLine
,
3040 IN EFI_SHELL_PARAMETERS_PROTOCOL
*ParamProtocol
3044 SHELL_FILE_HANDLE FileHandle
;
3048 if (ShellIsFile(ScriptPath
) != EFI_SUCCESS
) {
3049 return (EFI_INVALID_PARAMETER
);
3053 // get the argc and argv updated for scripts
3055 Status
= UpdateArgcArgv(ParamProtocol
, CmdLine
, Script_File_Name
, &Argv
, &Argc
);
3056 if (!EFI_ERROR(Status
)) {
3058 if (Handle
== NULL
) {
3062 Status
= ShellOpenFileByName(ScriptPath
, &FileHandle
, EFI_FILE_MODE_READ
, 0);
3063 if (!EFI_ERROR(Status
)) {
3067 Status
= RunScriptFileHandle(FileHandle
, ScriptPath
);
3070 // now close the file
3072 ShellCloseFile(&FileHandle
);
3075 Status
= RunScriptFileHandle(Handle
, ScriptPath
);
3080 // This is guaranteed to be called after UpdateArgcArgv no matter what else happened.
3081 // This is safe even if the update API failed. In this case, it may be a no-op.
3083 RestoreArgcArgv(ParamProtocol
, &Argv
, &Argc
);
3089 Return the pointer to the first occurrence of any character from a list of characters.
3091 @param[in] String the string to parse
3092 @param[in] CharacterList the list of character to look for
3093 @param[in] EscapeCharacter An escape character to skip
3095 @return the location of the first character in the string
3096 @retval CHAR_NULL no instance of any character in CharacterList was found in String
3100 IN CONST CHAR16
*String
,
3101 IN CONST CHAR16
*CharacterList
,
3102 IN CONST CHAR16 EscapeCharacter
3108 for (WalkStr
= 0; WalkStr
< StrLen(String
); WalkStr
++) {
3109 if (String
[WalkStr
] == EscapeCharacter
) {
3113 for (WalkChar
= 0; WalkChar
< StrLen(CharacterList
); WalkChar
++) {
3114 if (String
[WalkStr
] == CharacterList
[WalkChar
]) {
3115 return (&String
[WalkStr
]);
3119 return (String
+ StrLen(String
));