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

/** @file\r
  Main file for Mm shell Debug1 function.\r
\r
  (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2005 - 2017, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "UefiShellDebug1CommandsLib.h"\r
#include <Library/ShellLib.h>\r
#include <Library/IoLib.h>\r
#include <Protocol/PciRootBridgeIo.h>\r
#include <Protocol/DeviceIo.h>\r
\r
typedef enum {\r
  ShellMmMemory,\r
  ShellMmMemoryMappedIo,\r
  ShellMmIo,\r
  ShellMmPci,\r
  ShellMmPciExpress\r
} SHELL_MM_ACCESS_TYPE;\r
\r
CONST UINT16  mShellMmAccessTypeStr[] = {\r
  STRING_TOKEN (STR_MM_MEM),\r
  STRING_TOKEN (STR_MM_MMIO),\r
  STRING_TOKEN (STR_MM_IO),\r
  STRING_TOKEN (STR_MM_PCI),\r
  STRING_TOKEN (STR_MM_PCIE)\r
};\r
\r
STATIC CONST SHELL_PARAM_ITEM  ParamList[] = {\r
  { L"-mmio", TypeFlag  },\r
  { L"-mem",  TypeFlag  },\r
  { L"-io",   TypeFlag  },\r
  { L"-pci",  TypeFlag  },\r
  { L"-pcie", TypeFlag  },\r
  { L"-n",    TypeFlag  },\r
  { L"-w",    TypeValue },\r
  { NULL,     TypeMax   }\r
};\r
\r
CONST UINT64                                 mShellMmMaxNumber[] = {\r
  0, MAX_UINT8, MAX_UINT16, 0, MAX_UINT32, 0, 0, 0, MAX_UINT64\r
};\r
CONST EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL_WIDTH  mShellMmRootBridgeIoWidth[] = {\r
  0, EfiPciWidthUint8, EfiPciWidthUint16, 0, EfiPciWidthUint32, 0, 0, 0, EfiPciWidthUint64\r
};\r
CONST EFI_CPU_IO_PROTOCOL_WIDTH              mShellMmCpuIoWidth[] = {\r
  0, EfiCpuIoWidthUint8, EfiCpuIoWidthUint16, 0, EfiCpuIoWidthUint32, 0, 0, 0, EfiCpuIoWidthUint64\r
};\r
\r
/**\r
  Extract the PCI segment, bus, device, function, register from\r
  from a PCI or PCIE format of address..\r
\r
  @param[in]  PciFormat      Whether the address is of PCI format of PCIE format.\r
  @param[in]  Address        PCI or PCIE address.\r
  @param[out] Segment        PCI segment number.\r
  @param[out] Bus            PCI bus number.\r
  @param[out] Device         PCI device number.\r
  @param[out] Function       PCI function number.\r
  @param[out] Register       PCI register offset.\r
**/\r
VOID\r
ShellMmDecodePciAddress (\r
  IN BOOLEAN  PciFormat,\r
  IN UINT64   Address,\r
  OUT UINT32  *Segment,\r
  OUT UINT8   *Bus,\r
  OUT UINT8   *Device    OPTIONAL,\r
  OUT UINT8   *Function  OPTIONAL,\r
  OUT UINT32  *Register  OPTIONAL\r
  )\r
{\r
  if (PciFormat) {\r
    //\r
    // PCI Configuration Space.The address will have the format ssssbbddffrr,\r
    // where ssss = Segment, bb = Bus, dd = Device, ff = Function and rr = Register.\r
    //\r
    *Segment = (UINT32)(RShiftU64 (Address, 32) & 0xFFFF);\r
    *Bus     = (UINT8)(((UINT32)Address) >> 24);\r
\r
    if (Device != NULL) {\r
      *Device = (UINT8)(((UINT32)Address) >> 16);\r
    }\r
\r
    if (Function != NULL) {\r
      *Function = (UINT8)(((UINT32)Address) >> 8);\r
    }\r
\r
    if (Register != NULL) {\r
      *Register = (UINT8)Address;\r
    }\r
  } else {\r
    //\r
    // PCI Express Configuration Space.The address will have the format ssssssbbddffrrr,\r
    // where ssss = Segment, bb = Bus, dd = Device, ff = Function and rrr = Register.\r
    //\r
    *Segment = (UINT32)(RShiftU64 (Address, 36) & 0xFFFF);\r
    *Bus     = (UINT8)RShiftU64 (Address, 28);\r
    if (Device != NULL) {\r
      *Device = (UINT8)(((UINT32)Address) >> 20);\r
    }\r
\r
    if (Function != NULL) {\r
      *Function = (UINT8)(((UINT32)Address) >> 12);\r
    }\r
\r
    if (Register != NULL) {\r
      *Register = (UINT32)(Address & 0xFFF);\r
    }\r
  }\r
}\r
\r
/**\r
  Read or write some data from or into the Address.\r
\r
  @param[in]      AccessType      Access type.\r
  @param[in]      PciRootBridgeIo PciRootBridgeIo instance.\r
  @param[in]      CpuIo           CpuIo instance.\r
  @param[in]      Read            TRUE for read, FALSE for write.\r
  @param[in]      Addresss        The memory location to access.\r
  @param[in]      Size            The size of Buffer in Width sized units.\r
  @param[in, out] Buffer          The buffer to read into or write from.\r
**/\r
VOID\r
ShellMmAccess (\r
  IN     SHELL_MM_ACCESS_TYPE             AccessType,\r
  IN     EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL  *PciRootBridgeIo,\r
  IN     EFI_CPU_IO2_PROTOCOL             *CpuIo,\r
  IN     BOOLEAN                          Read,\r
  IN     UINT64                           Address,\r
  IN     UINTN                            Size,\r
  IN OUT VOID                             *Buffer\r
  )\r
{\r
  EFI_STATUS                              Status;\r
  EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL_IO_MEM  RootBridgeIoMem;\r
  EFI_CPU_IO_PROTOCOL_IO_MEM              CpuIoMem;\r
  UINT32                                  Segment;\r
  UINT8                                   Bus;\r
  UINT8                                   Device;\r
  UINT8                                   Function;\r
  UINT32                                  Register;\r
\r
  if (AccessType == ShellMmMemory) {\r
    if (Read) {\r
      CopyMem (Buffer, (VOID *)(UINTN)Address, Size);\r
    } else {\r
      CopyMem ((VOID *)(UINTN)Address, Buffer, Size);\r
    }\r
  } else {\r
    RootBridgeIoMem = NULL;\r
    CpuIoMem        = NULL;\r
    switch (AccessType) {\r
      case ShellMmPci:\r
      case ShellMmPciExpress:\r
        ASSERT (PciRootBridgeIo != NULL);\r
        ShellMmDecodePciAddress ((BOOLEAN)(AccessType == ShellMmPci), Address, &Segment, &Bus, &Device, &Function, &Register);\r
        if (Read) {\r
          Status = PciRootBridgeIo->Pci.Read (\r
                                          PciRootBridgeIo,\r
                                          mShellMmRootBridgeIoWidth[Size],\r
                                          EFI_PCI_ADDRESS (Bus, Device, Function, Register),\r
                                          1,\r
                                          Buffer\r
                                          );\r
        } else {\r
          Status = PciRootBridgeIo->Pci.Write (\r
                                          PciRootBridgeIo,\r
                                          mShellMmRootBridgeIoWidth[Size],\r
                                          EFI_PCI_ADDRESS (Bus, Device, Function, Register),\r
                                          1,\r
                                          Buffer\r
                                          );\r
        }\r
\r
        ASSERT_EFI_ERROR (Status);\r
        return;\r
\r
      case ShellMmMemoryMappedIo:\r
        if (PciRootBridgeIo != NULL) {\r
          RootBridgeIoMem = Read ? PciRootBridgeIo->Mem.Read : PciRootBridgeIo->Mem.Write;\r
        }\r
\r
        if (CpuIo != NULL) {\r
          CpuIoMem = Read ? CpuIo->Mem.Read : CpuIo->Mem.Write;\r
        }\r
\r
        break;\r
\r
      case ShellMmIo:\r
        if (PciRootBridgeIo != NULL) {\r
          RootBridgeIoMem = Read ? PciRootBridgeIo->Io.Read : PciRootBridgeIo->Io.Write;\r
        }\r
\r
        if (CpuIo != NULL) {\r
          CpuIoMem = Read ? CpuIo->Io.Read : CpuIo->Io.Write;\r
        }\r
\r
        break;\r
      default:\r
        ASSERT (FALSE);\r
        break;\r
    }\r
\r
    Status = EFI_UNSUPPORTED;\r
    if (RootBridgeIoMem != NULL) {\r
      Status = RootBridgeIoMem (PciRootBridgeIo, mShellMmRootBridgeIoWidth[Size], Address, 1, Buffer);\r
    }\r
\r
    if (EFI_ERROR (Status) && (CpuIoMem != NULL)) {\r
      Status = CpuIoMem (CpuIo, mShellMmCpuIoWidth[Size], Address, 1, Buffer);\r
    }\r
\r
    if (EFI_ERROR (Status)) {\r
      if (AccessType == ShellMmIo) {\r
        switch (Size) {\r
          case 1:\r
            if (Read) {\r
              *(UINT8 *)Buffer = IoRead8 ((UINTN)Address);\r
            } else {\r
              IoWrite8 ((UINTN)Address, *(UINT8 *)Buffer);\r
            }\r
\r
            break;\r
          case 2:\r
            if (Read) {\r
              *(UINT16 *)Buffer = IoRead16 ((UINTN)Address);\r
            } else {\r
              IoWrite16 ((UINTN)Address, *(UINT16 *)Buffer);\r
            }\r
\r
            break;\r
          case 4:\r
            if (Read) {\r
              *(UINT32 *)Buffer = IoRead32 ((UINTN)Address);\r
            } else {\r
              IoWrite32 ((UINTN)Address, *(UINT32 *)Buffer);\r
            }\r
\r
            break;\r
          case 8:\r
            if (Read) {\r
              *(UINT64 *)Buffer = IoRead64 ((UINTN)Address);\r
            } else {\r
              IoWrite64 ((UINTN)Address, *(UINT64 *)Buffer);\r
            }\r
\r
            break;\r
          default:\r
            ASSERT (FALSE);\r
            break;\r
        }\r
      } else {\r
        switch (Size) {\r
          case 1:\r
            if (Read) {\r
              *(UINT8 *)Buffer = MmioRead8 ((UINTN)Address);\r
            } else {\r
              MmioWrite8 ((UINTN)Address, *(UINT8 *)Buffer);\r
            }\r
\r
            break;\r
          case 2:\r
            if (Read) {\r
              *(UINT16 *)Buffer = MmioRead16 ((UINTN)Address);\r
            } else {\r
              MmioWrite16 ((UINTN)Address, *(UINT16 *)Buffer);\r
            }\r
\r
            break;\r
          case 4:\r
            if (Read) {\r
              *(UINT32 *)Buffer = MmioRead32 ((UINTN)Address);\r
            } else {\r
              MmioWrite32 ((UINTN)Address, *(UINT32 *)Buffer);\r
            }\r
\r
            break;\r
          case 8:\r
            if (Read) {\r
              *(UINT64 *)Buffer = MmioRead64 ((UINTN)Address);\r
            } else {\r
              MmioWrite64 ((UINTN)Address, *(UINT64 *)Buffer);\r
            }\r
\r
            break;\r
          default:\r
            ASSERT (FALSE);\r
            break;\r
        }\r
      }\r
    }\r
  }\r
}\r
\r
/**\r
  Find the CpuIo instance and PciRootBridgeIo instance in the platform.\r
  If there are multiple PciRootBridgeIo instances, the instance which manages\r
  the Address is returned.\r
\r
  @param[in]  AccessType      Access type.\r
  @param[in]  Address         Address to access.\r
  @param[out] CpuIo           Return the CpuIo instance.\r
  @param[out] PciRootBridgeIo Return the proper PciRootBridgeIo instance.\r
\r
  @retval TRUE  There are PciRootBridgeIo instances in the platform.\r
  @retval FALSE There isn't PciRootBridgeIo instance in the platform.\r
**/\r
BOOLEAN\r
ShellMmLocateIoProtocol (\r
  IN SHELL_MM_ACCESS_TYPE              AccessType,\r
  IN UINT64                            Address,\r
  OUT EFI_CPU_IO2_PROTOCOL             **CpuIo,\r
  OUT EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL  **PciRootBridgeIo\r
  )\r
{\r
  EFI_STATUS                         Status;\r
  UINTN                              Index;\r
  UINTN                              HandleCount;\r
  EFI_HANDLE                         *HandleBuffer;\r
  EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL    *Io;\r
  UINT32                             Segment;\r
  UINT8                              Bus;\r
  EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR  *Descriptors;\r
\r
  Status = gBS->LocateProtocol (&gEfiCpuIo2ProtocolGuid, NULL, (VOID **)CpuIo);\r
  if (EFI_ERROR (Status)) {\r
    *CpuIo = NULL;\r
  }\r
\r
  *PciRootBridgeIo = NULL;\r
  HandleBuffer     = NULL;\r
  Status           = gBS->LocateHandleBuffer (\r
                            ByProtocol,\r
                            &gEfiPciRootBridgeIoProtocolGuid,\r
                            NULL,\r
                            &HandleCount,\r
                            &HandleBuffer\r
                            );\r
  if (EFI_ERROR (Status) || (HandleCount == 0) || (HandleBuffer == NULL)) {\r
    return FALSE;\r
  }\r
\r
  Segment = 0;\r
  Bus     = 0;\r
  if ((AccessType == ShellMmPci) || (AccessType == ShellMmPciExpress)) {\r
    ShellMmDecodePciAddress ((BOOLEAN)(AccessType == ShellMmPci), Address, &Segment, &Bus, NULL, NULL, NULL);\r
  }\r
\r
  //\r
  // Find the EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL of the specified segment & bus number\r
  //\r
  for (Index = 0; (Index < HandleCount) && (*PciRootBridgeIo == NULL); Index++) {\r
    Status = gBS->HandleProtocol (\r
                    HandleBuffer[Index],\r
                    &gEfiPciRootBridgeIoProtocolGuid,\r
                    (VOID *)&Io\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    if ((((AccessType == ShellMmPci) || (AccessType == ShellMmPciExpress)) && (Io->SegmentNumber == Segment)) ||\r
        ((AccessType == ShellMmIo) || (AccessType == ShellMmMemoryMappedIo))\r
        )\r
    {\r
      Status = Io->Configuration (Io, (VOID **)&Descriptors);\r
      if (!EFI_ERROR (Status)) {\r
        while (Descriptors->Desc != ACPI_END_TAG_DESCRIPTOR) {\r
          //\r
          // Compare the segment and bus range for PCI/PCIE access\r
          //\r
          if ((Descriptors->ResType == ACPI_ADDRESS_SPACE_TYPE_BUS) &&\r
              ((AccessType == ShellMmPci) || (AccessType == ShellMmPciExpress)) &&\r
              ((Bus >= Descriptors->AddrRangeMin) && (Bus <= Descriptors->AddrRangeMax))\r
              )\r
          {\r
            *PciRootBridgeIo = Io;\r
            break;\r
\r
            //\r
            // Compare the address range for MMIO/IO access\r
            //\r
          } else if ((((Descriptors->ResType == ACPI_ADDRESS_SPACE_TYPE_IO) && (AccessType == ShellMmIo)) ||\r
                      ((Descriptors->ResType == ACPI_ADDRESS_SPACE_TYPE_MEM) && (AccessType == ShellMmMemoryMappedIo))\r
                      ) && ((Address >= Descriptors->AddrRangeMin) && (Address <= Descriptors->AddrRangeMax))\r
                     )\r
          {\r
            *PciRootBridgeIo = Io;\r
            break;\r
          }\r
\r
          Descriptors++;\r
        }\r
      }\r
    }\r
  }\r
\r
  if (HandleBuffer != NULL) {\r
    FreePool (HandleBuffer);\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Function for 'mm' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunMm (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS                       Status;\r
  EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL  *PciRootBridgeIo;\r
  EFI_CPU_IO2_PROTOCOL             *CpuIo;\r
  UINT64                           Address;\r
  UINT64                           Value;\r
  SHELL_MM_ACCESS_TYPE             AccessType;\r
  UINT64                           Buffer;\r
  UINTN                            Index;\r
  UINTN                            Size;\r
  BOOLEAN                          Complete;\r
  CHAR16                           *InputStr;\r
  BOOLEAN                          Interactive;\r
  LIST_ENTRY                       *Package;\r
  CHAR16                           *ProblemParam;\r
  SHELL_STATUS                     ShellStatus;\r
  CONST CHAR16                     *Temp;\r
  BOOLEAN                          HasPciRootBridgeIo;\r
\r
  Value       = 0;\r
  Address     = 0;\r
  ShellStatus = SHELL_SUCCESS;\r
  InputStr    = NULL;\r
  Size        = 1;\r
  AccessType  = ShellMmMemory;\r
\r
  //\r
  // Parse arguments\r
  //\r
  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
  if (EFI_ERROR (Status)) {\r
    if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"mm", ProblemParam);\r
      FreePool (ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    } else {\r
      ASSERT (FALSE);\r
    }\r
  } else {\r
    if (ShellCommandLineGetCount (Package) < 2) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"mm");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    } else if (ShellCommandLineGetCount (Package) > 3) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"mm");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    } else if (ShellCommandLineGetFlag (Package, L"-w") && (ShellCommandLineGetValue (Package, L"-w") == NULL)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDebug1HiiHandle, L"mm", L"-w");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    } else {\r
      if (ShellCommandLineGetFlag (Package, L"-mmio")) {\r
        AccessType = ShellMmMemoryMappedIo;\r
        if (  ShellCommandLineGetFlag (Package, L"-mem")\r
           || ShellCommandLineGetFlag (Package, L"-io")\r
           || ShellCommandLineGetFlag (Package, L"-pci")\r
           || ShellCommandLineGetFlag (Package, L"-pcie")\r
              )\r
        {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"mm");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto Done;\r
        }\r
      } else if (ShellCommandLineGetFlag (Package, L"-mem")) {\r
        AccessType = ShellMmMemory;\r
        if (  ShellCommandLineGetFlag (Package, L"-io")\r
           || ShellCommandLineGetFlag (Package, L"-pci")\r
           || ShellCommandLineGetFlag (Package, L"-pcie")\r
              )\r
        {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"mm");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto Done;\r
        }\r
      } else if (ShellCommandLineGetFlag (Package, L"-io")) {\r
        AccessType = ShellMmIo;\r
        if (  ShellCommandLineGetFlag (Package, L"-pci")\r
           || ShellCommandLineGetFlag (Package, L"-pcie")\r
              )\r
        {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"mm");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto Done;\r
        }\r
      } else if (ShellCommandLineGetFlag (Package, L"-pci")) {\r
        AccessType = ShellMmPci;\r
        if (ShellCommandLineGetFlag (Package, L"-pcie")\r
            )\r
        {\r
          ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"mm");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
          goto Done;\r
        }\r
      } else if (ShellCommandLineGetFlag (Package, L"-pcie")) {\r
        AccessType = ShellMmPciExpress;\r
      }\r
    }\r
\r
    //\r
    // Non interactive for a script file or for the specific parameter\r
    //\r
    Interactive = TRUE;\r
    if (gEfiShellProtocol->BatchIsActive () || ShellCommandLineGetFlag (Package, L"-n")) {\r
      Interactive = FALSE;\r
    }\r
\r
    Temp = ShellCommandLineGetValue (Package, L"-w");\r
    if (Temp != NULL) {\r
      Size = ShellStrToUintn (Temp);\r
    }\r
\r
    if ((Size != 1) && (Size != 2) && (Size != 4) && (Size != 8)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, L"mm", Temp, L"-w");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    }\r
\r
    Temp   = ShellCommandLineGetRawValue (Package, 1);\r
    Status = ShellConvertStringToUint64 (Temp, &Address, TRUE, FALSE);\r
    if (EFI_ERROR (Status)) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"mm", Temp);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    }\r
\r
    if ((Address & (Size - 1)) != 0) {\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_NOT_ALIGNED), gShellDebug1HiiHandle, L"mm", Address);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
      goto Done;\r
    }\r
\r
    //\r
    // locate IO protocol interface\r
    //\r
    HasPciRootBridgeIo = ShellMmLocateIoProtocol (AccessType, Address, &CpuIo, &PciRootBridgeIo);\r
    if ((AccessType == ShellMmPci) || (AccessType == ShellMmPciExpress)) {\r
      if (!HasPciRootBridgeIo) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PCIRBIO_NF), gShellDebug1HiiHandle, L"mm");\r
        ShellStatus = SHELL_NOT_FOUND;\r
        goto Done;\r
      }\r
\r
      if (PciRootBridgeIo == NULL) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_PCIE_ADDRESS_RANGE), gShellDebug1HiiHandle, L"mm", Address);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
        goto Done;\r
      }\r
    }\r
\r
    //\r
    // Mode 1: Directly set a value\r
    //\r
    Temp = ShellCommandLineGetRawValue (Package, 2);\r
    if (Temp != NULL) {\r
      Status = ShellConvertStringToUint64 (Temp, &Value, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"mm", Temp);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
        goto Done;\r
      }\r
\r
      if (Value > mShellMmMaxNumber[Size]) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"mm", Temp);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
        goto Done;\r
      }\r
\r
      ShellMmAccess (AccessType, PciRootBridgeIo, CpuIo, FALSE, Address, Size, &Value);\r
      goto Done;\r
    }\r
\r
    //\r
    // Mode 2: Directly show a value\r
    //\r
    if (!Interactive) {\r
      if (!gEfiShellProtocol->BatchIsActive ()) {\r
        ShellPrintHiiEx (-1, -1, NULL, mShellMmAccessTypeStr[AccessType], gShellDebug1HiiHandle);\r
      }\r
\r
      ShellMmAccess (AccessType, PciRootBridgeIo, CpuIo, TRUE, Address, Size, &Buffer);\r
\r
      if (!gEfiShellProtocol->BatchIsActive ()) {\r
        ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_ADDRESS), gShellDebug1HiiHandle, Address);\r
      }\r
\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_BUF), gShellDebug1HiiHandle, Size * 2, Buffer & mShellMmMaxNumber[Size]);\r
      ShellPrintEx (-1, -1, L"\r\n");\r
      goto Done;\r
    }\r
\r
    //\r
    // Mode 3: Show or set values in interactive mode\r
    //\r
    Complete = FALSE;\r
    do {\r
      ShellMmAccess (AccessType, PciRootBridgeIo, CpuIo, TRUE, Address, Size, &Buffer);\r
      ShellPrintHiiEx (-1, -1, NULL, mShellMmAccessTypeStr[AccessType], gShellDebug1HiiHandle);\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_ADDRESS), gShellDebug1HiiHandle, Address);\r
      ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_BUF), gShellDebug1HiiHandle, Size * 2, Buffer & mShellMmMaxNumber[Size]);\r
      ShellPrintEx (-1, -1, L" > ");\r
      //\r
      // wait user input to modify\r
      //\r
      if (InputStr != NULL) {\r
        FreePool (InputStr);\r
        InputStr = NULL;\r
      }\r
\r
      ShellPromptForResponse (ShellPromptResponseTypeFreeform, NULL, (VOID **)&InputStr);\r
\r
      if (InputStr != NULL) {\r
        //\r
        // skip space characters\r
        //\r
        for (Index = 0; InputStr[Index] == ' '; Index++) {\r
        }\r
\r
        if (InputStr[Index] != CHAR_NULL) {\r
          if ((InputStr[Index] == '.') || (InputStr[Index] == 'q') || (InputStr[Index] == 'Q')) {\r
            Complete = TRUE;\r
          } else if (!EFI_ERROR (ShellConvertStringToUint64 (InputStr + Index, &Buffer, TRUE, TRUE)) &&\r
                     (Buffer <= mShellMmMaxNumber[Size])\r
                     )\r
          {\r
            ShellMmAccess (AccessType, PciRootBridgeIo, CpuIo, FALSE, Address, Size, &Buffer);\r
          } else {\r
            ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_MM_ERROR), gShellDebug1HiiHandle, L"mm");\r
            continue;\r
          }\r
        }\r
      }\r
\r
      Address += Size;\r
      ShellPrintEx (-1, -1, L"\r\n");\r
    } while (!Complete);\r
  }\r
\r
  ASSERT (ShellStatus == SHELL_SUCCESS);\r
\r
Done:\r
  if (InputStr != NULL) {\r
    FreePool (InputStr);\r
  }\r
\r
  if (Package != NULL) {\r
    ShellCommandLineFreeVarList (Package);\r
  }\r
\r
  return ShellStatus;\r
}\r