From cab102c36bac10af35cbcdf8cf2fb5aa931f2953 Mon Sep 17 00:00:00 2001 From: Jaben Carsey Date: Tue, 5 Aug 2014 20:56:07 +0000 Subject: [PATCH] Updates the logic to allow help to ouput help information from a dynamic shell command Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jaben Carsey Reviewed-by: Lee Rosenbaum git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15756 6f19259b-4bc3-4df7-8a09-765794883524 --- .../Library/UefiShellLevel3CommandsLib/Help.c | 118 ++++++++++++++++-- .../UefiShellLevel3CommandsLib.inf | 5 +- 2 files changed, 112 insertions(+), 11 deletions(-) diff --git a/ShellPkg/Library/UefiShellLevel3CommandsLib/Help.c b/ShellPkg/Library/UefiShellLevel3CommandsLib/Help.c index cd03ff7f07..a5dfade8f1 100644 --- a/ShellPkg/Library/UefiShellLevel3CommandsLib/Help.c +++ b/ShellPkg/Library/UefiShellLevel3CommandsLib/Help.c @@ -1,7 +1,7 @@ /** @file Main file for Help shell level 3 function. - Copyright (c) 2009 - 2013, Intel Corporation. All rights reserved.
+ Copyright (c) 2009 - 2014, Intel Corporation. All rights reserved.
This program and the accompanying materials are licensed and made available under the terms and conditions of the BSD License which accompanies this distribution. The full text of the license may be found at @@ -15,6 +15,96 @@ #include "UefiShellLevel3CommandsLib.h" #include +#include + +#include + +/** + Attempt to print help from a dynamically added command. + + @param[in] CommandToGetHelpOn The unicode name of the command that help is requested on. + + @retval EFI_SUCCESS The help was displayed + @retval FI_NOT_FOUND The command name could not be found +**/ +STATIC +EFI_STATUS +EFIAPI +PrintDynamicCommandHelp( + IN CHAR16 *CommandToGetHelpOn + ) +{ + EFI_STATUS Status; + BOOLEAN Found = FALSE; + EFI_HANDLE *CommandHandleList = NULL; + EFI_HANDLE *NextCommand; + EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL *DynamicCommand; + CHAR16 *OutText = NULL; + + CommandHandleList = GetHandleListByProtocol(&gEfiShellDynamicCommandProtocolGuid); + + if (CommandHandleList == NULL) { + // + // not found or out of resources + // + return FALSE; + } + + for (NextCommand = CommandHandleList; *NextCommand != NULL; NextCommand++) { + Status = gBS->HandleProtocol( + *NextCommand, + &gEfiShellDynamicCommandProtocolGuid, + (VOID **)&DynamicCommand + ); + + if (EFI_ERROR(Status)) { + continue; + } + + if ((gUnicodeCollation->MetaiMatch(gUnicodeCollation, (CHAR16 *)DynamicCommand->CommandName, CommandToGetHelpOn)) || + (gEfiShellProtocol->GetAlias(CommandToGetHelpOn, NULL) != NULL && (gUnicodeCollation->MetaiMatch(gUnicodeCollation, (CHAR16 *)DynamicCommand->CommandName, (CHAR16*)(gEfiShellProtocol->GetAlias(CommandToGetHelpOn, NULL)))))) { + // + // TODO: how to get proper language? + // + OutText = DynamicCommand->GetHelp(DynamicCommand, "en"); + + if (OutText == NULL) { + continue; + } + + // + // Trim extra characters from the end the the string before printing + // + while (StrLen(OutText) > 0 + && (OutText[StrLen(OutText) - 1] == L'\r' || OutText[StrLen(OutText) - 1] == L'\n' || OutText[StrLen(OutText) - 1] == L' ')) { + OutText[StrLen(OutText) - 1] = CHAR_NULL; + } + + // + // Make sure we have something to print still. + // + if (StrLen(OutText) == 0) { + FreePool(OutText); + OutText = NULL; + continue; + } + + // + // Print and move on. + // + ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_HELP_COMMAND), gShellLevel3HiiHandle, DynamicCommand->CommandName, OutText); + FreePool(OutText); + OutText = NULL; + Found = TRUE; + break; + } + + } + + FreePool(CommandHandleList); + + return Found ? EFI_SUCCESS : EFI_NOT_FOUND; +} STATIC CONST SHELL_PARAM_ITEM ParamList[] = { {L"-usage", TypeFlag}, @@ -168,15 +258,24 @@ ShellCommandRunHelp ( // Search the .man file for Shell applications (Shell external commands). // if (!Found) { - Status = ShellPrintHelp(CommandToGetHelpOn, SectionToGetHelpOn, FALSE); - if (Status == EFI_DEVICE_ERROR) { - ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HELP_INV), gShellLevel3HiiHandle, CommandToGetHelpOn); - } else if (EFI_ERROR(Status)) { - ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HELP_NF), gShellLevel3HiiHandle, CommandToGetHelpOn); - } else { - Found = TRUE; - } + Status = ShellPrintHelp(CommandToGetHelpOn, SectionToGetHelpOn, FALSE); + if (Status == EFI_DEVICE_ERROR) { + ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HELP_INV), gShellLevel3HiiHandle, CommandToGetHelpOn); + } else if (EFI_ERROR(Status)) { + ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HELP_NF), gShellLevel3HiiHandle, CommandToGetHelpOn); + } else { + Found = TRUE; + } } + + // + // now try to match against the dynamic command list and print help + // + Status = PrintDynamicCommandHelp(CommandToGetHelpOn); + if (Status == EFI_SUCCESS) { + Found = TRUE; + } + } if (!Found) { @@ -206,4 +305,3 @@ ShellCommandRunHelp ( return (ShellStatus); } - diff --git a/ShellPkg/Library/UefiShellLevel3CommandsLib/UefiShellLevel3CommandsLib.inf b/ShellPkg/Library/UefiShellLevel3CommandsLib/UefiShellLevel3CommandsLib.inf index 7316750fd8..79a4060833 100644 --- a/ShellPkg/Library/UefiShellLevel3CommandsLib/UefiShellLevel3CommandsLib.inf +++ b/ShellPkg/Library/UefiShellLevel3CommandsLib/UefiShellLevel3CommandsLib.inf @@ -3,7 +3,7 @@ # Note that the interactive versions of the time, date, and timezone functions are handled in the level 2 library. # # Copyright (c) 2013, Hewlett-Packard Development Company, L.P. -# Copyright (c) 2009-2011, Intel Corporation. All rights reserved.
+# Copyright (c) 2009-2014, Intel Corporation. All rights reserved.
# # This program and the accompanying materials # are licensed and made available under the terms and conditions of the BSD License @@ -59,6 +59,7 @@ PcdLib HiiLib FileHandleLib + HandleParsingLib [Guids] gEfiFileInfoGuid @@ -69,3 +70,5 @@ gEfiShellPkgTokenSpaceGuid.PcdShellFileOperationSize gEfiShellPkgTokenSpaceGuid.PcdShellSupplier +[Protocols] + gEfiShellDynamicCommandProtocolGuid # SOMETIMES_CONSUMED