ShellPkg: stop using EFI_HANDLE in place of EFI_HII_HANDLE

[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / UefiShellDebug1CommandsLib.c

/** @file\r
  Main file for NULL named library for debug1 profile shell command functions.\r
\r
  Copyright (c) 2010 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UefiShellDebug1CommandsLib.h"\r
#include <Library/BcfgCommandLib.h>\r
\r
STATIC CONST CHAR16 mFileName[] = L"Debug1Commands";\r
EFI_HII_HANDLE gShellDebug1HiiHandle = NULL;\r
\r
/**\r
  Gets the debug file name.  This will be used if HII is not working.\r
\r
  @retval NULL    No file is available.\r
  @return         The NULL-terminated filename to get help from.\r
**/\r
CONST CHAR16*\r
EFIAPI\r
ShellCommandGetManFileNameDebug1 (\r
  VOID\r
  )\r
{\r
  return (mFileName);\r
}\r
\r
/**\r
  Constructor for the Shell Debug1 Commands library.\r
\r
  @param ImageHandle    the image handle of the process\r
  @param SystemTable    the EFI System Table pointer\r
\r
  @retval EFI_SUCCESS        the shell command handlers were installed sucessfully\r
  @retval EFI_UNSUPPORTED    the shell level required was not found.\r
**/\r
EFI_STATUS\r
EFIAPI\r
UefiShellDebug1CommandsLibConstructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  //\r
  // check our bit of the profiles mask\r
  //\r
  if ((PcdGet8(PcdShellProfileMask) & BIT1) == 0) {\r
    return (EFI_SUCCESS);\r
  }\r
\r
  //\r
  // install the HII stuff.\r
  //\r
  gShellDebug1HiiHandle = HiiAddPackages (&gShellDebug1HiiGuid, gImageHandle, UefiShellDebug1CommandsLibStrings, NULL);\r
  if (gShellDebug1HiiHandle == NULL) {\r
    return (EFI_DEVICE_ERROR);\r
  }\r
\r
  //\r
  // install our shell command handlers that are always installed\r
  //\r
  ShellCommandRegisterCommandName(L"setsize",       ShellCommandRunSetSize            , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SETSIZE)      );\r
  ShellCommandRegisterCommandName(L"comp",          ShellCommandRunComp               , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_COMP)         );\r
  ShellCommandRegisterCommandName(L"mode",          ShellCommandRunMode               , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MODE)         );\r
  ShellCommandRegisterCommandName(L"memmap",        ShellCommandRunMemMap             , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MEMMAP)       );\r
  ShellCommandRegisterCommandName(L"eficompress",   ShellCommandRunEfiCompress        , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EFICOMPRESS)  );\r
  ShellCommandRegisterCommandName(L"efidecompress", ShellCommandRunEfiDecompress      , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EFIDCOMPRESS) );\r
  ShellCommandRegisterCommandName(L"dmem",          ShellCommandRunDmem               , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DMEM)         );\r
  ShellCommandRegisterCommandName(L"loadpcirom",    ShellCommandRunLoadPciRom         , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_LOAD_PCI_ROM) );\r
  ShellCommandRegisterCommandName(L"mm",            ShellCommandRunMm                 , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MM)           );\r
  ShellCommandRegisterCommandName(L"setvar",        ShellCommandRunSetVar             , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SETVAR)       );\r
  ShellCommandRegisterCommandName(L"sermode",       ShellCommandRunSerMode            , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SERMODE)      );\r
  ShellCommandRegisterCommandName(L"pci",           ShellCommandRunPci                , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_PCI)          );\r
  ShellCommandRegisterCommandName(L"smbiosview",    ShellCommandRunSmbiosView         , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SMBIOSVIEW)   );\r
  ShellCommandRegisterCommandName(L"dmpstore",      ShellCommandRunDmpStore           , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DMPSTORE)     );\r
  ShellCommandRegisterCommandName(L"dblk",          ShellCommandRunDblk               , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DBLK)         );\r
  ShellCommandRegisterCommandName(L"edit",          ShellCommandRunEdit               , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EDIT)         );\r
  ShellCommandRegisterCommandName(L"hexedit",       ShellCommandRunHexEdit            , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_HEXEDIT)      );\r
\r
  ShellCommandRegisterAlias(L"dmem", L"mem");\r
\r
  BcfgLibraryRegisterBcfgCommand(ImageHandle, SystemTable, L"Debug1");\r
\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Destructor for the library.  free any resources.\r
\r
  @param ImageHandle            The image handle of the process.\r
  @param SystemTable            The EFI System Table pointer.\r
**/\r
EFI_STATUS\r
EFIAPI\r
UefiShellDebug1CommandsLibDestructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  if (gShellDebug1HiiHandle != NULL) {\r
    HiiRemovePackages(gShellDebug1HiiHandle);\r
  }\r
\r
  BcfgLibraryUnregisterBcfgCommand(ImageHandle, SystemTable);\r
  return (EFI_SUCCESS);\r
}\r
\r
\r
/**\r
  Function returns a system configuration table that is stored in the\r
  EFI System Table based on the provided GUID.\r
\r
  @param[in]  TableGuid     A pointer to the table's GUID type.\r
  @param[in, out] Table     On exit, a pointer to a system configuration table.\r
\r
  @retval EFI_SUCCESS      A configuration table matching TableGuid was found.\r
  @retval EFI_NOT_FOUND    A configuration table matching TableGuid was not found.\r
**/\r
EFI_STATUS\r
GetSystemConfigurationTable (\r
  IN EFI_GUID *TableGuid,\r
  IN OUT VOID **Table\r
  )\r
{\r
  UINTN Index;\r
  ASSERT (Table != NULL);\r
\r
  for (Index = 0; Index < gST->NumberOfTableEntries; Index++) {\r
    if (CompareGuid (TableGuid, &(gST->ConfigurationTable[Index].VendorGuid))) {\r
      *Table = gST->ConfigurationTable[Index].VendorTable;\r
      return EFI_SUCCESS;\r
    }\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
/**\r
  Clear the line at the specified Row.\r
\r
  @param[in] Row                The row number to be cleared ( start from 1 )\r
  @param[in] LastCol            The last printable column.\r
  @param[in] LastRow            The last printable row.\r
**/\r
VOID\r
EditorClearLine (\r
  IN UINTN Row,\r
  IN UINTN LastCol,\r
  IN UINTN LastRow\r
  )\r
{\r
  UINTN  Col;\r
  CHAR16 Line[200];\r
\r
  if (Row == 0) {\r
    Row = 1;\r
  }\r
\r
  //\r
  // prepare a blank line\r
  // If max column is larger, split to multiple prints.\r
  //\r
  SetMem16 (Line, sizeof (Line), L' ');\r
  Line[ARRAY_SIZE (Line) - 1] = CHAR_NULL;\r
\r
  for (Col = 1; Col <= LastCol; Col += ARRAY_SIZE (Line) - 1) {\r
    if (Col + ARRAY_SIZE (Line) - 1 > LastCol) {\r
      if (Row == LastRow) {\r
        //\r
        // if CHAR_NULL is still at position LastCol, it will cause first line error\r
        //\r
        Line[(LastCol - 1) % (ARRAY_SIZE (Line) - 1)] = CHAR_NULL;\r
      } else {\r
        Line[LastCol % (ARRAY_SIZE (Line) - 1)] = CHAR_NULL;\r
      }\r
    }\r
\r
    //\r
    // print out the blank line\r
    //\r
    ShellPrintEx ((INT32) Col - 1, (INT32) Row - 1, Line);\r
  }\r
}\r
\r
/**\r
  Determine if the character is valid for a filename.\r
\r
  @param[in] Ch     The character to test.\r
\r
  @retval TRUE      The character is valid.\r
  @retval FALSE     The character is not valid.\r
**/\r
BOOLEAN\r
IsValidFileNameChar (\r
  IN CONST CHAR16 Ch\r
  )\r
{\r
  //\r
  // See if there are any illegal characters within the name\r
  //\r
  if (Ch < 0x20 || Ch == L'\"' || Ch == L'*' || Ch == L'/' || Ch == L'<' || Ch == L'>' || Ch == L'?' || Ch == L'|') {\r
    return FALSE;\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Check if file name has illegal characters.\r
\r
  @param Name       The filename to check.\r
\r
  @retval TRUE      The filename is ok.\r
  @retval FALSE     The filename is not ok.\r
**/\r
BOOLEAN\r
IsValidFileName (\r
  IN CONST CHAR16 *Name\r
  )\r
{\r
\r
  UINTN Index;\r
  UINTN Len;\r
\r
  //\r
  // check the length of Name\r
  //\r
  for (Len = 0, Index = StrLen (Name) - 1; Index + 1 != 0; Index--, Len++) {\r
    if (Name[Index] == '\\' || Name[Index] == ':') {\r
      break;\r
    }\r
  }\r
\r
  if (Len == 0 || Len > 255) {\r
    return FALSE;\r
  }\r
  //\r
  // check whether any char in Name not appears in valid file name char\r
  //\r
  for (Index = 0; Index < StrLen (Name); Index++) {\r
    if (!IsValidFileNameChar (Name[Index])) {\r
      return FALSE;\r
    }\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Find a filename that is valid (not taken) with the given extension.\r
\r
  @param[in] Extension      The file extension.\r
\r
  @retval NULL  Something went wrong.\r
  @return the valid filename.\r
**/\r
CHAR16 *\r
EditGetDefaultFileName (\r
  IN CONST CHAR16 *Extension\r
  )\r
{\r
  EFI_STATUS         Status;\r
  UINTN              Suffix;\r
  CHAR16             *FileNameTmp;\r
\r
  Suffix       = 0;\r
\r
  do {\r
    FileNameTmp = CatSPrint (NULL, L"NewFile%d.%s", Suffix, Extension);\r
\r
    //\r
    // after that filename changed to path\r
    //\r
    Status = ShellFileExists (FileNameTmp);\r
\r
    if (Status == EFI_NOT_FOUND) {\r
      return FileNameTmp;\r
    }\r
\r
    FreePool (FileNameTmp);\r
    FileNameTmp = NULL;\r
    Suffix++;\r
  } while (Suffix != 0);\r
\r
  FreePool (FileNameTmp);\r
  return NULL;\r
}\r
\r
/**\r
  Read a file into an allocated buffer.  The buffer is the responsibility\r
  of the caller to free.\r
\r
  @param[in]  FileName          The filename of the file to open.\r
  @param[out] Buffer            Upon successful return, the pointer to the\r
                                address of the allocated buffer.\r
  @param[out] BufferSize        If not NULL, then the pointer to the size\r
                                of the allocated buffer.\r
  @param[out] ReadOnly          Upon successful return TRUE if the file is\r
                                read only.  FALSE otherwise.\r
\r
  @retval EFI_NOT_FOUND         The filename did not represent a file in the\r
                                file system.\r
  @retval EFI_SUCCESS           The file was read into the buffer.\r
  @retval EFI_OUT_OF_RESOURCES  A memory allocation failed.\r
  @retval EFI_LOAD_ERROR        The file read operation failed.\r
  @retval EFI_INVALID_PARAMETER A parameter was invalid.\r
  @retval EFI_INVALID_PARAMETER FileName was NULL.\r
  @retval EFI_INVALID_PARAMETER FileName was a directory.\r
**/\r
EFI_STATUS\r
ReadFileIntoBuffer (\r
  IN CONST CHAR16 *FileName,\r
  OUT VOID        **Buffer,\r
  OUT UINTN       *BufferSize OPTIONAL,\r
  OUT BOOLEAN     *ReadOnly\r
  )\r
{\r
  VOID              *InternalBuffer;\r
  UINTN             FileSize;\r
  SHELL_FILE_HANDLE FileHandle;\r
  BOOLEAN           CreateFile;\r
  EFI_STATUS        Status;\r
  EFI_FILE_INFO     *Info;\r
\r
  InternalBuffer  = NULL;\r
  FileSize        = 0;\r
  FileHandle      = NULL;\r
  CreateFile      = FALSE;\r
  Status          = EFI_SUCCESS;\r
  Info            = NULL;\r
\r
  if (FileName == NULL || Buffer == NULL || ReadOnly == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  //\r
  // try to open the file\r
  //\r
  Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ, 0);\r
\r
  if (!EFI_ERROR(Status)) {\r
    ASSERT(CreateFile == FALSE);\r
    if (FileHandle == NULL) {\r
      return EFI_LOAD_ERROR;\r
    }\r
\r
    Info = ShellGetFileInfo(FileHandle);\r
\r
    if (Info->Attribute & EFI_FILE_DIRECTORY) {\r
      FreePool (Info);\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    if (Info->Attribute & EFI_FILE_READ_ONLY) {\r
      *ReadOnly = TRUE;\r
    } else {\r
      *ReadOnly = FALSE;\r
    }\r
    //\r
    // get file size\r
    //\r
    FileSize = (UINTN) Info->FileSize;\r
\r
    FreePool (Info);\r
  } else if (Status == EFI_NOT_FOUND) {\r
    //\r
    // file not exists.  add create and try again\r
    //\r
    Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    } else {\r
      //\r
      // it worked.  now delete it and move on with the name (now validated)\r
      //\r
      Status = ShellDeleteFile (&FileHandle);\r
      if (Status == EFI_WARN_DELETE_FAILURE) {\r
        Status = EFI_ACCESS_DENIED;\r
      }\r
      if (EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
    }\r
    //\r
    // file doesn't exist, so set CreateFile to TRUE and can't be read-only\r
    //\r
    CreateFile = TRUE;\r
    *ReadOnly  = FALSE;\r
  }\r
\r
  //\r
  // the file exists\r
  //\r
  if (!CreateFile) {\r
    //\r
    // allocate buffer to read file\r
    //\r
    InternalBuffer = AllocateZeroPool (FileSize);\r
    if (InternalBuffer == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
    //\r
    // read file into InternalBuffer\r
    //\r
    Status = ShellReadFile (FileHandle, &FileSize, InternalBuffer);\r
    ShellCloseFile(&FileHandle);\r
    FileHandle = NULL;\r
    if (EFI_ERROR (Status)) {\r
      SHELL_FREE_NON_NULL (InternalBuffer);\r
      return EFI_LOAD_ERROR;\r
    }\r
  }\r
  *Buffer = InternalBuffer;\r
  if (BufferSize != NULL) {\r
    *BufferSize = FileSize;\r
  }\r
  return (EFI_SUCCESS);\r
\r
}\r