udk2010.up2.shell initial release.

[mirror_edk2.git] / ShellPkg / Application / Shell / ShellManParser.c

/** @file\r
  Provides interface to shell MAN file parser.\r
\r
  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
  This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "Shell.h"\r
\r
/**\r
  Verifies that the filename has .MAN on the end.\r
\r
  allocates a new buffer and copies the name (appending .MAN if necessary)\r
\r
  ASSERT if ManFileName is NULL\r
\r
  @param[in] ManFileName            original filename\r
\r
  @return the new filename with .man as the extension.\r
**/\r
CHAR16 *\r
EFIAPI\r
GetManFileName(\r
  IN CONST CHAR16 *ManFileName\r
  )\r
{\r
  CHAR16 *Buffer;\r
  ASSERT(ManFileName != NULL);\r
  //\r
  // Fix the file name\r
  //\r
  if (StrnCmp(ManFileName+StrLen(ManFileName)-4, L".man", 4)==0) {\r
    Buffer = AllocateZeroPool(StrSize(ManFileName));\r
    StrCpy(Buffer, ManFileName);\r
  } else {\r
    Buffer = AllocateZeroPool(StrSize(ManFileName) + 4*sizeof(CHAR16));\r
    StrCpy(Buffer, ManFileName);\r
    StrCat(Buffer, L".man");\r
  }\r
  return (Buffer);\r
}\r
\r
/**\r
  Search the path environment variable for possible locations and test for\r
  which one contains a man file with the name specified.  If a valid file is found\r
  stop searching and return the (opened) SHELL_FILE_HANDLE for that file.\r
\r
  @param[in] FileName           Name of the file to find and open.\r
  @param[out] Handle            Pointer to the handle of the found file.  The\r
                                value of this is undefined for return values\r
                                except EFI_SUCCESS.\r
\r
  @retval EFI_SUCCESS           The file was found.  Handle is a valid SHELL_FILE_HANDLE\r
  @retval EFI_INVALID_PARAMETER A parameter had an invalid value.\r
  @retval EFI_NOT_FOUND         The file was not found.\r
**/\r
EFI_STATUS\r
EFIAPI\r
SearchPathForFile(\r
  IN CONST CHAR16             *FileName,\r
  OUT SHELL_FILE_HANDLE       *Handle\r
  )\r
{\r
  CHAR16          *FullFileName;\r
  EFI_STATUS      Status;\r
\r
  if ( FileName     == NULL\r
    || Handle       == NULL\r
    || StrLen(FileName) == 0\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  FullFileName = ShellFindFilePath(FileName);\r
  if (FullFileName == NULL) {\r
    return (EFI_NOT_FOUND);\r
  }\r
\r
  //\r
  // now open that file\r
  //\r
  Status = EfiShellOpenFileByName(FullFileName, Handle, EFI_FILE_MODE_READ);\r
  FreePool(FullFileName);\r
\r
  return (Status);\r
}\r
\r
/**\r
  parses through Buffer (which is MAN file formatted) and returns the\r
  detailed help for any sub section specified in the comma seperated list of\r
  sections provided.  If the end of the file or a .TH section is found then\r
  return.\r
\r
  Upon a sucessful return the caller is responsible to free the memory in *HelpText\r
\r
  @param[in] Buffer             Buffer to read from\r
  @param[in] Sections           name of command's sub sections to find\r
  @param[in] HelpText           pointer to pointer to string where text goes.\r
  @param[in] HelpSize           pointer to size of allocated HelpText (may be updated)\r
\r
  @retval EFI_OUT_OF_RESOURCES  a memory allocation failed.\r
  @retval EFI_SUCCESS           the section was found and its description sotred in\r
                                an alloceted buffer.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ManBufferFindSections(\r
  IN CONST CHAR16 *Buffer,\r
  IN CONST CHAR16 *Sections,\r
  IN CHAR16       **HelpText,\r
  IN UINTN        *HelpSize\r
  )\r
{\r
  EFI_STATUS          Status;\r
  CONST CHAR16        *CurrentLocation;\r
  BOOLEAN             CurrentlyReading;\r
  CHAR16              *SectionName;\r
  UINTN               SectionLen;\r
  BOOLEAN             Found;\r
  CHAR16              *TempString;\r
  CHAR16              *TempString2;\r
\r
  if ( Buffer     == NULL\r
    || HelpText   == NULL\r
    || HelpSize   == NULL\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  Status            = EFI_SUCCESS;\r
  CurrentlyReading  = FALSE;\r
  Found             = FALSE;\r
\r
  for (CurrentLocation = Buffer,TempString = NULL\r
    ;  CurrentLocation != NULL && *CurrentLocation != CHAR_NULL\r
    ;  CurrentLocation=StrStr(CurrentLocation, L"\r\n"),TempString = NULL\r
   ){\r
    while(CurrentLocation[0] == L'\r' || CurrentLocation[0] == L'\n') {\r
      CurrentLocation++;\r
    }\r
    if (CurrentLocation[0] == L'#') {\r
      //\r
      // Skip comment lines\r
      //\r
      continue;\r
    }\r
    if (StrnCmp(CurrentLocation, L".TH", 3) == 0) {\r
      //\r
      // we hit the end of this commands section so stop.\r
      //\r
      break;\r
    }\r
    if (StrnCmp(CurrentLocation, L".SH ", 4) == 0) {\r
      if (Sections == NULL) {\r
        CurrentlyReading = TRUE;\r
        continue;\r
      } else if (CurrentlyReading) {\r
        CurrentlyReading = FALSE;\r
      }\r
      CurrentLocation += 4;\r
      //\r
      // is this a section we want to read in?\r
      //\r
      if (StrLen(CurrentLocation)!=0) {\r
        TempString2 = StrStr(CurrentLocation, L" ");\r
        TempString2 = MIN(TempString2, StrStr(CurrentLocation, L"\r"));\r
        TempString2 = MIN(TempString2, StrStr(CurrentLocation, L"\n"));\r
        ASSERT(TempString == NULL);\r
        TempString = StrnCatGrow(&TempString, NULL, CurrentLocation, TempString2==NULL?0:TempString2 - CurrentLocation);\r
        SectionName = TempString;\r
        SectionLen = StrLen(SectionName);\r
        SectionName = StrStr(Sections, SectionName);\r
        if (SectionName == NULL) {\r
          continue;\r
        }\r
        if (*(SectionName + SectionLen) == CHAR_NULL || *(SectionName + SectionLen) == L',') {\r
          CurrentlyReading = TRUE;\r
        }\r
      }\r
    } else if (CurrentlyReading) {\r
      Found = TRUE;\r
      if (StrLen(CurrentLocation)!=0) {\r
        TempString2 = StrStr(CurrentLocation, L"\r");\r
        TempString2 = MIN(TempString2, StrStr(CurrentLocation, L"\n"));\r
        ASSERT(TempString == NULL);\r
        TempString = StrnCatGrow(&TempString, NULL, CurrentLocation, TempString2==NULL?0:TempString2 - CurrentLocation);\r
        //\r
        // copy and save the current line.\r
        //\r
        ASSERT((*HelpText == NULL && *HelpSize == 0) || (*HelpText != NULL));\r
        StrnCatGrow (HelpText, HelpSize, TempString, 0);\r
        StrnCatGrow (HelpText, HelpSize, L"\r\n", 0);\r
      }\r
    }\r
    SHELL_FREE_NON_NULL(TempString);\r
  }\r
  if (!Found && !EFI_ERROR(Status)) {\r
    return (EFI_NOT_FOUND);\r
  }\r
  return (Status);\r
}\r
\r
/**\r
  parses through the MAN file specified by SHELL_FILE_HANDLE and returns the\r
  detailed help for any sub section specified in the comma seperated list of\r
  sections provided.  If the end of the file or a .TH section is found then\r
  return.\r
\r
  Upon a sucessful return the caller is responsible to free the memory in *HelpText\r
\r
  @param[in] Handle             FileHandle to read from\r
  @param[in] Sections           name of command's sub sections to find\r
  @param[out] HelpText          pointer to pointer to string where text goes.\r
  @param[out] HelpSize          pointer to size of allocated HelpText (may be updated)\r
  @param[in] Ascii              TRUE if the file is ASCII, FALSE otherwise.\r
\r
  @retval EFI_OUT_OF_RESOURCES  a memory allocation failed.\r
  @retval EFI_SUCCESS           the section was found and its description sotred in\r
                                an alloceted buffer.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ManFileFindSections(\r
  IN SHELL_FILE_HANDLE  Handle,\r
  IN CONST CHAR16       *Sections,\r
  OUT CHAR16            **HelpText,\r
  OUT UINTN             *HelpSize,\r
  IN BOOLEAN            Ascii\r
  )\r
{\r
  EFI_STATUS          Status;\r
  CHAR16              *ReadLine;\r
  UINTN               Size;\r
  BOOLEAN             CurrentlyReading;\r
  CHAR16              *SectionName;\r
  UINTN               SectionLen;\r
  BOOLEAN             Found;\r
\r
  if ( Handle     == NULL\r
    || HelpText   == NULL\r
    || HelpSize   == NULL\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  Status            = EFI_SUCCESS;\r
  CurrentlyReading  = FALSE;\r
  Size              = 1024;\r
  Found             = FALSE;\r
\r
  ReadLine          = AllocateZeroPool(Size);\r
  if (ReadLine == NULL) {\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
\r
  for (;!ShellFileHandleEof(Handle);Size = 1024) {\r
    Status = ShellFileHandleReadLine(Handle, ReadLine, &Size, TRUE, &Ascii);\r
    if (ReadLine[0] == L'#') {\r
      //\r
      // Skip comment lines\r
      //\r
      continue;\r
    }\r
    //\r
    // ignore too small of buffer...\r
    //\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      Status = EFI_SUCCESS;\r
    }\r
    if (EFI_ERROR(Status)) {\r
      break;\r
    } else if (StrnCmp(ReadLine, L".TH", 3) == 0) {\r
      //\r
      // we hit the end of this commands section so stop.\r
      //\r
      break;\r
    } else if (StrnCmp(ReadLine, L".SH", 3) == 0) {\r
      if (Sections == NULL) {\r
        CurrentlyReading = TRUE;\r
        continue;\r
      }\r
      //\r
      // we found a section\r
      //\r
      if (CurrentlyReading) {\r
        CurrentlyReading = FALSE;\r
      }\r
      //\r
      // is this a section we want to read in?\r
      //\r
      for ( SectionName = ReadLine + 3\r
          ; *SectionName == L' '\r
          ; SectionName++);\r
      SectionLen = StrLen(SectionName);\r
      SectionName = StrStr(Sections, SectionName);\r
      if (SectionName == NULL) {\r
        continue;\r
      }\r
      if (*(SectionName + SectionLen) == CHAR_NULL || *(SectionName + SectionLen) == L',') {\r
        CurrentlyReading = TRUE;\r
      }\r
    } else if (CurrentlyReading) {\r
      Found = TRUE;\r
      //\r
      // copy and save the current line.\r
      //\r
      ASSERT((*HelpText == NULL && *HelpSize == 0) || (*HelpText != NULL));\r
      StrnCatGrow (HelpText, HelpSize, ReadLine, 0);\r
      StrnCatGrow (HelpText, HelpSize, L"\r\n", 0);\r
    }\r
  }\r
  FreePool(ReadLine);\r
  if (!Found && !EFI_ERROR(Status)) {\r
    return (EFI_NOT_FOUND);\r
  }\r
  return (Status);\r
}\r
\r
/**\r
  parses through the MAN file formatted Buffer and returns the\r
  "Brief Description" for the .TH section as specified by Command.  If the\r
  command section is not found return EFI_NOT_FOUND.\r
\r
  Upon a sucessful return the caller is responsible to free the memory in *BriefDesc\r
\r
  @param[in] Handle             Buffer to read from\r
  @param[in] Command            name of command's section to find\r
  @param[in] BriefDesc          pointer to pointer to string where description goes.\r
  @param[in] BriefSize          pointer to size of allocated BriefDesc\r
\r
  @retval EFI_OUT_OF_RESOURCES  a memory allocation failed.\r
  @retval EFI_SUCCESS           the section was found and its description sotred in\r
                                an alloceted buffer.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ManBufferFindTitleSection(\r
  IN CHAR16         **Buffer,\r
  IN CONST CHAR16   *Command,\r
  IN CHAR16         **BriefDesc,\r
  IN UINTN          *BriefSize\r
  )\r
{\r
  EFI_STATUS    Status;\r
  CHAR16        *TitleString;\r
  CHAR16        *TitleEnd;\r
  CHAR16        *CurrentLocation;\r
\r
  if ( Buffer     == NULL\r
    || Command    == NULL\r
    || (BriefDesc != NULL && BriefSize == NULL)\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  Status    = EFI_SUCCESS;\r
\r
  TitleString = AllocatePool((7*sizeof(CHAR16)) + StrSize(Command));\r
  if (TitleString == NULL) {\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
  StrCpy(TitleString, L".TH ");\r
  StrCat(TitleString, Command);\r
  StrCat(TitleString, L" 0 ");\r
\r
  CurrentLocation = StrStr(*Buffer, TitleString);\r
  if (CurrentLocation == NULL){\r
    Status = EFI_NOT_FOUND;\r
  } else {\r
    //\r
    // we found it so copy out the rest of the line into BriefDesc\r
    // After skipping any spaces or zeroes\r
    //\r
    for (CurrentLocation += StrLen(TitleString)\r
      ;  *CurrentLocation == L' ' || *CurrentLocation == L'0' || *CurrentLocation == L'1' || *CurrentLocation == L'\"'\r
      ;  CurrentLocation++);\r
\r
    TitleEnd = StrStr(CurrentLocation, L"\"");\r
    ASSERT(TitleEnd != NULL);\r
    if (BriefDesc != NULL) {\r
      *BriefSize = StrSize(TitleEnd);\r
      *BriefDesc = AllocateZeroPool(*BriefSize);\r
      if (*BriefDesc == NULL) {\r
        Status = EFI_OUT_OF_RESOURCES;\r
      } else {\r
        StrnCpy(*BriefDesc, CurrentLocation, TitleEnd-CurrentLocation);\r
      }\r
    }\r
\r
    for (CurrentLocation = TitleEnd\r
      ;  *CurrentLocation != L'\n'\r
      ;  CurrentLocation++);\r
    for (\r
      ;  *CurrentLocation == L' ' || *CurrentLocation == L'\n' || *CurrentLocation == L'\r'\r
      ;  CurrentLocation++);\r
    *Buffer = CurrentLocation;\r
  }\r
\r
  FreePool(TitleString);\r
  return (Status);\r
}\r
\r
/**\r
  parses through the MAN file specified by SHELL_FILE_HANDLE and returns the\r
  "Brief Description" for the .TH section as specified by Command.  if the\r
  command section is not found return EFI_NOT_FOUND.\r
\r
  Upon a sucessful return the caller is responsible to free the memory in *BriefDesc\r
\r
  @param[in] Handle             FileHandle to read from\r
  @param[in] Command            name of command's section to find\r
  @param[out] BriefDesc         pointer to pointer to string where description goes.\r
  @param[out] BriefSize         pointer to size of allocated BriefDesc\r
  @param[in,out] Ascii          TRUE if the file is ASCII, FALSE otherwise, will be\r
                                set if the file handle is at the 0 position.\r
\r
  @retval EFI_OUT_OF_RESOURCES  a memory allocation failed.\r
  @retval EFI_SUCCESS           the section was found and its description sotred in\r
                                an alloceted buffer.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ManFileFindTitleSection(\r
  IN SHELL_FILE_HANDLE  Handle,\r
  IN CONST CHAR16       *Command,\r
  OUT CHAR16            **BriefDesc OPTIONAL,\r
  OUT UINTN             *BriefSize OPTIONAL,\r
  IN OUT BOOLEAN        *Ascii\r
  )\r
{\r
  EFI_STATUS  Status;\r
  CHAR16      *TitleString;\r
  CHAR16      *ReadLine;\r
  UINTN       Size;\r
  CHAR16      *TitleEnd;\r
  UINTN       TitleLen;\r
  BOOLEAN     Found;\r
\r
  if ( Handle     == NULL\r
    || Command    == NULL\r
    || (BriefDesc != NULL && BriefSize == NULL)\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  Status    = EFI_SUCCESS;\r
  Size      = 1024;\r
  Found     = FALSE;\r
\r
  ReadLine  = AllocateZeroPool(Size);\r
  if (ReadLine == NULL) {\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
\r
  TitleString = AllocatePool((4*sizeof(CHAR16)) + StrSize(Command));\r
  if (TitleString == NULL) {\r
    FreePool(ReadLine);\r
    return (EFI_OUT_OF_RESOURCES);\r
  }\r
  StrCpy(TitleString, L".TH ");\r
  StrCat(TitleString, Command);\r
  TitleLen = StrLen(TitleString);\r
  for (;!ShellFileHandleEof(Handle);Size = 1024) {\r
   Status = ShellFileHandleReadLine(Handle, ReadLine, &Size, TRUE, Ascii);\r
    if (ReadLine[0] == L'#') {\r
      //\r
      // Skip comment lines\r
      //\r
      continue;\r
    }\r
    //\r
    // ignore too small of buffer...\r
    //\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      Status = EFI_SUCCESS;\r
    }\r
    if (EFI_ERROR(Status)) {\r
      break;\r
    }\r
    if (StrnCmp(ReadLine, TitleString, TitleLen) == 0) {\r
      Found = TRUE;\r
      //\r
      // we found it so copy out the rest of the line into BriefDesc\r
      // After skipping any spaces or zeroes\r
      //\r
      for ( TitleEnd = ReadLine+TitleLen\r
          ; *TitleEnd == L' ' || *TitleEnd == L'0' || *TitleEnd == L'1'\r
          ; TitleEnd++);\r
      if (BriefDesc != NULL) {\r
        *BriefSize = StrSize(TitleEnd);\r
        *BriefDesc = AllocateZeroPool(*BriefSize);\r
        if (*BriefDesc == NULL) {\r
          Status = EFI_OUT_OF_RESOURCES;\r
          break;\r
        }\r
        StrCpy(*BriefDesc, TitleEnd);\r
      }\r
      break;\r
    }\r
  }\r
  FreePool(ReadLine);\r
  FreePool(TitleString);\r
  if (!Found && !EFI_ERROR(Status)) {\r
    return (EFI_NOT_FOUND);\r
  }\r
  return (Status);\r
}\r
\r
/**\r
  This function returns the help information for the specified command. The help text\r
  will be parsed from a UEFI Shell manual page. (see UEFI Shell 2.0 Appendix B)\r
\r
  If Sections is specified, then each section name listed will be compared in a casesensitive\r
  manner, to the section names described in Appendix B. If the section exists,\r
  it will be appended to the returned help text. If the section does not exist, no\r
  information will be returned. If Sections is NULL, then all help text information\r
  available will be returned.\r
\r
  if BriefDesc is NULL, then the breif description will not be savedd seperatly,\r
  but placed first in the main HelpText.\r
\r
  @param[in] ManFileName        Points to the NULL-terminated UEFI Shell MAN file name.\r
  @param[in] Command            Points to the NULL-terminated UEFI Shell command name.\r
  @param[in] Sections           Points to the NULL-terminated comma-delimited\r
                                section names to return. If NULL, then all\r
                                sections will be returned.\r
  @param[out] BriefDesc         On return, points to a callee-allocated buffer\r
                                containing brief description text.\r
  @param[out] HelpText          On return, points to a callee-allocated buffer\r
                                containing all specified help text.\r
\r
  @retval EFI_SUCCESS           The help text was returned.\r
  @retval EFI_OUT_OF_RESOURCES  The necessary buffer could not be allocated to hold the\r
                                returned help text.\r
  @retval EFI_INVALID_PARAMETER HelpText is NULL\r
  @retval EFI_NOT_FOUND         There is no help text available for Command.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ProcessManFile(\r
  IN CONST CHAR16 *ManFileName,\r
  IN CONST CHAR16 *Command,\r
  IN CONST CHAR16 *Sections OPTIONAL,\r
  OUT CHAR16      **BriefDesc OPTIONAL,\r
  OUT CHAR16      **HelpText\r
  )\r
{\r
  CHAR16            *TempString;\r
  SHELL_FILE_HANDLE FileHandle;\r
  EFI_STATUS        Status;\r
  UINTN             HelpSize;\r
  UINTN             BriefSize;\r
  BOOLEAN           Ascii;\r
  CHAR16            *TempString2;\r
  EFI_DEVICE_PATH_PROTOCOL  *FileDevPath;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevPath;\r
\r
  if ( ManFileName == NULL\r
    || Command     == NULL\r
    || HelpText    == NULL\r
   ){\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  HelpSize    = 0;\r
  BriefSize   = 0;\r
  TempString  = NULL;\r
  //\r
  // See if it's in HII first\r
  //\r
  TempString = ShellCommandGetCommandHelp(Command);\r
  if (TempString != NULL) {\r
    TempString2 = TempString;\r
    Status = ManBufferFindTitleSection(&TempString2, Command, BriefDesc, &BriefSize);\r
    if (!EFI_ERROR(Status) && HelpText != NULL){\r
      Status = ManBufferFindSections(TempString2, Sections, HelpText, &HelpSize);\r
    }\r
  } else {\r
    FileHandle    = NULL;\r
    TempString  = GetManFileName(ManFileName);\r
\r
    Status = SearchPathForFile(TempString, &FileHandle);\r
    if (EFI_ERROR(Status)) {\r
      FileDevPath = FileDevicePath(NULL, TempString);\r
      DevPath = AppendDevicePath (ShellInfoObject.ImageDevPath, FileDevPath);\r
      Status = InternalOpenFileDevicePath(DevPath, &FileHandle, EFI_FILE_MODE_READ, 0);\r
      FreePool(FileDevPath);\r
      FreePool(DevPath);\r
    }\r
\r
    if (!EFI_ERROR(Status)) {\r
      HelpSize  = 0;\r
      BriefSize = 0;\r
      Status = ManFileFindTitleSection(FileHandle, Command, BriefDesc, &BriefSize, &Ascii);\r
      if (!EFI_ERROR(Status) && HelpText != NULL){\r
        Status = ManFileFindSections(FileHandle, Sections, HelpText, &HelpSize, Ascii);\r
      }\r
      ShellInfoObject.NewEfiShellProtocol->CloseFile(FileHandle);\r
    } else {\r
      *HelpText = NULL;\r
    }\r
  }\r
  if (TempString != NULL) {\r
    FreePool(TempString);\r
  }\r
\r
  return (Status);\r
}\r