| /** @file | |
| Main file for Help shell level 3 function. | |
| Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved. <BR> | |
| Copyright (c) 2014, ARM Limited. All rights reserved. <BR> | |
| (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR> | |
| SPDX-License-Identifier: BSD-2-Clause-Patent | |
| **/ | |
| #include "UefiShellLevel3CommandsLib.h" | |
| #include <Library/ShellLib.h> | |
| #include <Library/HandleParsingLib.h> | |
| #include <Protocol/ShellDynamicCommand.h> | |
| /** | |
| function to insert string items into a list in the correct alphabetical place | |
| the resultant list is a double NULL terminated list of NULL terminated strings. | |
| upon successful return the memory must be caller freed (unless passed back in | |
| via a loop where it will get reallocated). | |
| @param[in,out] DestList double pointer to the list. may be NULL. | |
| @param[in,out] DestSize pointer to the size of list. may be 0, if DestList is NULL. | |
| @param[in] Item the item to insert. | |
| @retval EFI_SUCCESS the operation was successful. | |
| **/ | |
| EFI_STATUS | |
| LexicalInsertIntoList ( | |
| IN OUT CHAR16 **DestList, | |
| IN OUT UINTN *DestSize, | |
| IN CONST CHAR16 *Item | |
| ) | |
| { | |
| CHAR16 *NewList; | |
| INTN LexicalMatchValue; | |
| CHAR16 *LexicalSpot; | |
| UINTN SizeOfAddedNameInBytes; | |
| // | |
| // If there are none, then just return with success | |
| // | |
| if ((Item == NULL) || (*Item == CHAR_NULL) || (StrLen (Item) == 0)) { | |
| return (EFI_SUCCESS); | |
| } | |
| NewList = *DestList; | |
| SizeOfAddedNameInBytes = StrSize (Item); | |
| NewList = ReallocatePool (*DestSize, (*DestSize) + SizeOfAddedNameInBytes, NewList); | |
| (*DestSize) = (*DestSize) + SizeOfAddedNameInBytes; | |
| // | |
| // Find the correct spot in the list | |
| // | |
| for (LexicalSpot = NewList | |
| ; LexicalSpot != NULL && LexicalSpot < NewList + (*DestSize) | |
| ; LexicalSpot += StrLen (LexicalSpot) + 1 | |
| ) | |
| { | |
| // | |
| // Get Lexical Comparison Value between PrevCommand and Command list entry | |
| // | |
| LexicalMatchValue = gUnicodeCollation->StriColl ( | |
| gUnicodeCollation, | |
| (CHAR16 *)LexicalSpot, | |
| (CHAR16 *)Item | |
| ); | |
| // | |
| // The new item goes before this one. | |
| // | |
| if ((LexicalMatchValue > 0) || (StrLen (LexicalSpot) == 0)) { | |
| if (StrLen (LexicalSpot) != 0) { | |
| // | |
| // Move this and all other items out of the way | |
| // | |
| CopyMem ( | |
| LexicalSpot + (SizeOfAddedNameInBytes/sizeof (CHAR16)), | |
| LexicalSpot, | |
| (*DestSize) - SizeOfAddedNameInBytes - ((LexicalSpot - NewList) * sizeof (CHAR16)) | |
| ); | |
| } | |
| // | |
| // Stick this one in place | |
| // | |
| StrCpyS (LexicalSpot, SizeOfAddedNameInBytes/sizeof (CHAR16), Item); | |
| break; | |
| } | |
| } | |
| *DestList = NewList; | |
| return (EFI_SUCCESS); | |
| } | |
| /** | |
| function to add each command name from the linked list to the string list. | |
| the resultant list is a double NULL terminated list of NULL terminated strings. | |
| @param[in,out] DestList double pointer to the list. may be NULL. | |
| @param[in,out] DestSize pointer to the size of list. may be 0, if DestList is NULL. | |
| @param[in] SourceList the double linked list of commands. | |
| @retval EFI_SUCCESS the operation was successful. | |
| **/ | |
| EFI_STATUS | |
| CopyListOfCommandNames ( | |
| IN OUT CHAR16 **DestList, | |
| IN OUT UINTN *DestSize, | |
| IN CONST COMMAND_LIST *SourceList | |
| ) | |
| { | |
| CONST COMMAND_LIST *Node; | |
| for ( Node = (COMMAND_LIST *)GetFirstNode (&SourceList->Link) | |
| ; SourceList != NULL && !IsListEmpty (&SourceList->Link) && !IsNull (&SourceList->Link, &Node->Link) | |
| ; Node = (COMMAND_LIST *)GetNextNode (&SourceList->Link, &Node->Link) | |
| ) | |
| { | |
| LexicalInsertIntoList (DestList, DestSize, Node->CommandString); | |
| } | |
| return (EFI_SUCCESS); | |
| } | |
| /** | |
| function to add each dynamic command name to the string list. | |
| the resultant list is a double NULL terminated list of NULL terminated strings. | |
| @param[in,out] DestList double pointer to the list. may be NULL. | |
| @param[in,out] DestSize pointer to the size of list. may be 0, if DestList is NULL. | |
| @retval EFI_SUCCESS the operation was successful. | |
| @return an error from HandleProtocol | |
| **/ | |
| STATIC | |
| EFI_STATUS | |
| CopyListOfCommandNamesWithDynamic ( | |
| IN OUT CHAR16 **DestList, | |
| IN OUT UINTN *DestSize | |
| ) | |
| { | |
| EFI_HANDLE *CommandHandleList; | |
| CONST EFI_HANDLE *NextCommand; | |
| EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL *DynamicCommand; | |
| EFI_STATUS Status; | |
| CommandHandleList = GetHandleListByProtocol (&gEfiShellDynamicCommandProtocolGuid); | |
| // | |
| // If there are none, then just return with success | |
| // | |
| if (CommandHandleList == NULL) { | |
| return (EFI_SUCCESS); | |
| } | |
| Status = EFI_SUCCESS; | |
| // | |
| // Append those to the list. | |
| // | |
| for (NextCommand = CommandHandleList; *NextCommand != NULL && !EFI_ERROR (Status); NextCommand++) { | |
| Status = gBS->HandleProtocol ( | |
| *NextCommand, | |
| &gEfiShellDynamicCommandProtocolGuid, | |
| (VOID **)&DynamicCommand | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| continue; | |
| } | |
| Status = LexicalInsertIntoList (DestList, DestSize, DynamicCommand->CommandName); | |
| } | |
| SHELL_FREE_NON_NULL (CommandHandleList); | |
| return (Status); | |
| } | |
| /** | |
| Attempt to print help from a dynamically added command. | |
| @param[in] CommandToGetHelpOn The unicode name of the command that help is | |
| requested on. | |
| @param[in] SectionToGetHelpOn Pointer to the section specifier(s). | |
| @param[in] PrintCommandText Print the command followed by the help content | |
| or just help. | |
| @retval EFI_SUCCESS The help was displayed | |
| @retval EFI_NOT_FOUND The command name could not be found | |
| @retval EFI_DEVICE_ERROR The help data format was incorrect. | |
| **/ | |
| EFI_STATUS | |
| PrintDynamicCommandHelp ( | |
| IN CONST CHAR16 *CommandToGetHelpOn, | |
| IN CONST CHAR16 *SectionToGetHelpOn, | |
| IN BOOLEAN PrintCommandText | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| BOOLEAN Found; | |
| EFI_HANDLE *CommandHandleList; | |
| EFI_HANDLE *NextCommand; | |
| EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL *DynamicCommand; | |
| Status = EFI_NOT_FOUND; | |
| Found = FALSE; | |
| CommandHandleList = NULL; | |
| CommandHandleList = GetHandleListByProtocol (&gEfiShellDynamicCommandProtocolGuid); | |
| if (CommandHandleList == NULL) { | |
| // | |
| // not found or out of resources | |
| // | |
| return Status; | |
| } | |
| for (NextCommand = CommandHandleList; *NextCommand != NULL; NextCommand++) { | |
| Status = gBS->HandleProtocol ( | |
| *NextCommand, | |
| &gEfiShellDynamicCommandProtocolGuid, | |
| (VOID **)&DynamicCommand | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| continue; | |
| } | |
| // | |
| // Check execution break flag when printing multiple command help information. | |
| // | |
| if (ShellGetExecutionBreakFlag ()) { | |
| break; | |
| } | |
| if ((gUnicodeCollation->MetaiMatch (gUnicodeCollation, (CHAR16 *)DynamicCommand->CommandName, (CHAR16 *)CommandToGetHelpOn)) || | |
| ((gEfiShellProtocol->GetAlias (CommandToGetHelpOn, NULL) != NULL) && (gUnicodeCollation->MetaiMatch (gUnicodeCollation, (CHAR16 *)DynamicCommand->CommandName, (CHAR16 *)(gEfiShellProtocol->GetAlias (CommandToGetHelpOn, NULL)))))) | |
| { | |
| // Print as Shell Help if in ManPage format. | |
| Status = ShellPrintHelp ( | |
| DynamicCommand->CommandName, | |
| SectionToGetHelpOn, | |
| PrintCommandText | |
| ); | |
| if (Status == EFI_DEVICE_ERROR) { | |
| ShellPrintHiiDefaultEx ( | |
| STRING_TOKEN (STR_HELP_INV), | |
| gShellLevel3HiiHandle, | |
| DynamicCommand->CommandName | |
| ); | |
| } else if (EFI_ERROR (Status)) { | |
| ShellPrintHiiDefaultEx ( | |
| STRING_TOKEN (STR_HELP_NF), | |
| gShellLevel3HiiHandle, | |
| DynamicCommand->CommandName | |
| ); | |
| } else { | |
| Found = TRUE; | |
| } | |
| } | |
| } | |
| SHELL_FREE_NON_NULL (CommandHandleList); | |
| return (Found ? EFI_SUCCESS : Status); | |
| } | |
| STATIC CONST SHELL_PARAM_ITEM ParamList[] = { | |
| { L"-usage", TypeFlag }, | |
| { L"-section", TypeMaxValue }, | |
| { L"-verbose", TypeFlag }, | |
| { L"-v", TypeFlag }, | |
| { NULL, TypeMax } | |
| }; | |
| /** | |
| Function for 'help' command. | |
| @param[in] ImageHandle Handle to the Image (NULL if Internal). | |
| @param[in] SystemTable Pointer to the System Table (NULL if Internal). | |
| **/ | |
| SHELL_STATUS | |
| EFIAPI | |
| ShellCommandRunHelp ( | |
| IN EFI_HANDLE ImageHandle, | |
| IN EFI_SYSTEM_TABLE *SystemTable | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| LIST_ENTRY *Package; | |
| CHAR16 *ProblemParam; | |
| SHELL_STATUS ShellStatus; | |
| CHAR16 *SortedCommandList; | |
| CONST CHAR16 *CurrentCommand; | |
| CHAR16 *CommandToGetHelpOn; | |
| CHAR16 *SectionToGetHelpOn; | |
| BOOLEAN Found; | |
| BOOLEAN PrintCommandText; | |
| UINTN SortedCommandListSize; | |
| PrintCommandText = TRUE; | |
| ProblemParam = NULL; | |
| ShellStatus = SHELL_SUCCESS; | |
| CommandToGetHelpOn = NULL; | |
| SectionToGetHelpOn = NULL; | |
| SortedCommandList = NULL; | |
| Found = FALSE; | |
| // | |
| // initialize the shell lib (we must be in non-auto-init...) | |
| // | |
| Status = ShellInitialize (); | |
| ASSERT_EFI_ERROR (Status); | |
| Status = CommandInit (); | |
| ASSERT_EFI_ERROR (Status); | |
| // | |
| // parse the command line | |
| // | |
| Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE); | |
| if (EFI_ERROR (Status)) { | |
| if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel3HiiHandle, L"help", ProblemParam); | |
| FreePool (ProblemParam); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| ASSERT (FALSE); | |
| } | |
| } else { | |
| // | |
| // Check for conflicting parameters. | |
| // | |
| if ( ShellCommandLineGetFlag (Package, L"-usage") | |
| && ShellCommandLineGetFlag (Package, L"-section") | |
| && (ShellCommandLineGetFlag (Package, L"-verbose") || ShellCommandLineGetFlag (Package, L"-v")) | |
| ) | |
| { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_GEN_PARAM_CON), gShellLevel3HiiHandle, L"help"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (ShellCommandLineGetRawValue (Package, 2) != NULL) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel3HiiHandle, L"help"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| // | |
| // Get the command name we are getting help on | |
| // | |
| ASSERT (CommandToGetHelpOn == NULL); | |
| StrnCatGrow (&CommandToGetHelpOn, NULL, ShellCommandLineGetRawValue (Package, 1), 0); | |
| if ((CommandToGetHelpOn == NULL) && ShellCommandLineGetFlag (Package, L"-?")) { | |
| // | |
| // If we dont have a command and we got a simple -? | |
| // we are looking for help on help command. | |
| // | |
| StrnCatGrow (&CommandToGetHelpOn, NULL, L"help", 0); | |
| } | |
| if (CommandToGetHelpOn == NULL) { | |
| StrnCatGrow (&CommandToGetHelpOn, NULL, L"*", 0); | |
| ASSERT (SectionToGetHelpOn == NULL); | |
| StrnCatGrow (&SectionToGetHelpOn, NULL, L"NAME", 0); | |
| } else { | |
| PrintCommandText = FALSE; | |
| ASSERT (SectionToGetHelpOn == NULL); | |
| // | |
| // Get the section name for the given command name | |
| // | |
| if (ShellCommandLineGetFlag (Package, L"-section")) { | |
| StrnCatGrow (&SectionToGetHelpOn, NULL, ShellCommandLineGetValue (Package, L"-section"), 0); | |
| } else if (ShellCommandLineGetFlag (Package, L"-usage")) { | |
| StrnCatGrow (&SectionToGetHelpOn, NULL, L"NAME,SYNOPSIS", 0); | |
| } else if (ShellCommandLineGetFlag (Package, L"-verbose") || ShellCommandLineGetFlag (Package, L"-v")) { | |
| } else { | |
| // | |
| // The output of help <command> will display NAME, SYNOPSIS, OPTIONS, DESCRIPTION, and EXAMPLES sections. | |
| // | |
| StrnCatGrow (&SectionToGetHelpOn, NULL, L"NAME,SYNOPSIS,OPTIONS,DESCRIPTION,EXAMPLES", 0); | |
| } | |
| } | |
| if (gUnicodeCollation->StriColl (gUnicodeCollation, CommandToGetHelpOn, L"special") == 0) { | |
| // | |
| // we need info on the special characters | |
| // | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_SC_HEADER), gShellLevel3HiiHandle); | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_SC_DATA), gShellLevel3HiiHandle); | |
| Found = TRUE; | |
| } else { | |
| SortedCommandList = NULL; | |
| SortedCommandListSize = 0; | |
| CopyListOfCommandNames (&SortedCommandList, &SortedCommandListSize, ShellCommandGetCommandList (TRUE)); | |
| CopyListOfCommandNamesWithDynamic (&SortedCommandList, &SortedCommandListSize); | |
| for (CurrentCommand = SortedCommandList | |
| ; CurrentCommand != NULL && CurrentCommand < SortedCommandList + SortedCommandListSize/sizeof (CHAR16) && *CurrentCommand != CHAR_NULL | |
| ; CurrentCommand += StrLen (CurrentCommand) + 1 | |
| ) | |
| { | |
| // | |
| // Checking execution break flag when print multiple command help information. | |
| // | |
| if (ShellGetExecutionBreakFlag ()) { | |
| break; | |
| } | |
| if ((gUnicodeCollation->MetaiMatch (gUnicodeCollation, (CHAR16 *)CurrentCommand, CommandToGetHelpOn)) || | |
| ((gEfiShellProtocol->GetAlias (CommandToGetHelpOn, NULL) != NULL) && (gUnicodeCollation->MetaiMatch (gUnicodeCollation, (CHAR16 *)CurrentCommand, (CHAR16 *)(gEfiShellProtocol->GetAlias (CommandToGetHelpOn, NULL)))))) | |
| { | |
| // | |
| // We have a command to look for help on. | |
| // | |
| Status = ShellPrintHelp (CurrentCommand, SectionToGetHelpOn, PrintCommandText); | |
| if (EFI_ERROR (Status)) { | |
| // | |
| // now try to match against the dynamic command list and print help | |
| // | |
| Status = PrintDynamicCommandHelp (CurrentCommand, SectionToGetHelpOn, PrintCommandText); | |
| } | |
| if (Status == EFI_DEVICE_ERROR) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_INV), gShellLevel3HiiHandle, CurrentCommand); | |
| } else if (EFI_ERROR (Status)) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_NF), gShellLevel3HiiHandle, CurrentCommand); | |
| } else { | |
| Found = TRUE; | |
| } | |
| } | |
| } | |
| // | |
| // Search the .man file for Shell applications (Shell external commands). | |
| // | |
| if (!Found) { | |
| Status = ShellPrintHelp (CommandToGetHelpOn, SectionToGetHelpOn, FALSE); | |
| if (Status == EFI_DEVICE_ERROR) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_INV), gShellLevel3HiiHandle, CommandToGetHelpOn); | |
| } else if (EFI_ERROR (Status)) { | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_NF), gShellLevel3HiiHandle, CommandToGetHelpOn); | |
| } else { | |
| Found = TRUE; | |
| } | |
| } | |
| } | |
| if (!Found) { | |
| ShellStatus = SHELL_NOT_FOUND; | |
| } | |
| // | |
| // free the command line package | |
| // | |
| ShellCommandLineFreeVarList (Package); | |
| } | |
| } | |
| if ((CommandToGetHelpOn != NULL) && (StrCmp (CommandToGetHelpOn, L"*") == 0)) { | |
| // | |
| // If '*' then the command entered was 'Help' without qualifiers, This footer | |
| // provides additional info on help switches | |
| // | |
| ShellPrintHiiDefaultEx (STRING_TOKEN (STR_HELP_FOOTER), gShellLevel3HiiHandle); | |
| } | |
| if (CommandToGetHelpOn != NULL) { | |
| FreePool (CommandToGetHelpOn); | |
| } | |
| if (SectionToGetHelpOn != NULL) { | |
| FreePool (SectionToGetHelpOn); | |
| } | |
| SHELL_FREE_NON_NULL (SortedCommandList); | |
| return (ShellStatus); | |
| } |