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

/** @file\r
  Provides interface to advanced shell functionality for parsing both handle and protocol database.\r
\r
  Copyright (c) 2013 Hewlett-Packard Development Company, L.P.\r
  Copyright (c) 2010 - 2014, 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 "UefiHandleParsingLib.h"\r
#include "IndustryStandard/Acpi10.h"\r
\r
EFI_HANDLE mHandleParsingHiiHandle;\r
HANDLE_INDEX_LIST mHandleList = {{{NULL,NULL},0,0},0};\r
\r
/**\r
  Function to translate the EFI_MEMORY_TYPE into a string.\r
\r
  @param[in] Memory     The memory type.\r
\r
  @retval               A string representation of the type allocated from BS Pool.\r
**/\r
CHAR16*\r
EFIAPI\r
ConvertMemoryType (\r
  IN CONST EFI_MEMORY_TYPE Memory\r
  )\r
{\r
  CHAR16 *RetVal;\r
  RetVal = NULL;\r
\r
  switch (Memory) {\r
  case EfiReservedMemoryType:       StrnCatGrow(&RetVal, NULL, L"EfiReservedMemoryType", 0);        break;\r
  case EfiLoaderCode:               StrnCatGrow(&RetVal, NULL, L"EfiLoaderCode", 0);                break;\r
  case EfiLoaderData:               StrnCatGrow(&RetVal, NULL, L"EfiLoaderData", 0);                break;\r
  case EfiBootServicesCode:         StrnCatGrow(&RetVal, NULL, L"EfiBootServicesCode", 0);          break;\r
  case EfiBootServicesData:         StrnCatGrow(&RetVal, NULL, L"EfiBootServicesData", 0);          break;\r
  case EfiRuntimeServicesCode:      StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesCode", 0);       break;\r
  case EfiRuntimeServicesData:      StrnCatGrow(&RetVal, NULL, L"EfiRuntimeServicesData", 0);       break;\r
  case EfiConventionalMemory:       StrnCatGrow(&RetVal, NULL, L"EfiConventionalMemory", 0);        break;\r
  case EfiUnusableMemory:           StrnCatGrow(&RetVal, NULL, L"EfiUnusableMemory", 0);            break;\r
  case EfiACPIReclaimMemory:        StrnCatGrow(&RetVal, NULL, L"EfiACPIReclaimMemory", 0);         break;\r
  case EfiACPIMemoryNVS:            StrnCatGrow(&RetVal, NULL, L"EfiACPIMemoryNVS", 0);             break;\r
  case EfiMemoryMappedIO:           StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIO", 0);            break;\r
  case EfiMemoryMappedIOPortSpace:  StrnCatGrow(&RetVal, NULL, L"EfiMemoryMappedIOPortSpace", 0);   break;\r
  case EfiPalCode:                  StrnCatGrow(&RetVal, NULL, L"EfiPalCode", 0);                   break;\r
  case EfiMaxMemoryType:            StrnCatGrow(&RetVal, NULL, L"EfiMaxMemoryType", 0);             break;\r
  default: ASSERT(FALSE);\r
  }\r
  return (RetVal);\r
}\r
\r
/**\r
  Constructor for the library.\r
\r
  @param[in] ImageHandle    Ignored.\r
  @param[in] SystemTable    Ignored.\r
\r
  @retval EFI_SUCCESS   The operation was successful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
HandleParsingLibConstructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  mHandleParsingHiiHandle = HiiAddPackages (&gHandleParsingHiiGuid, gImageHandle, UefiHandleParsingLibStrings, NULL);\r
  if (mHandleParsingHiiHandle == NULL) {\r
    return (EFI_DEVICE_ERROR);\r
  }\r
\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Destructor for the library.  free any resources.\r
\r
  @param[in] ImageHandle    Ignored.\r
  @param[in] SystemTable    Ignored.\r
\r
  @retval EFI_SUCCESS   The operation was successful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
HandleParsingLibDestructor (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  if (mHandleParsingHiiHandle != NULL) {\r
    HiiRemovePackages(mHandleParsingHiiHandle);\r
  }\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Function to dump information about LoadedImage.\r
\r
  This will allocate the return buffer from boot services pool.\r
\r
  @param[in] TheHandle      The handle that has LoadedImage installed.\r
  @param[in] Verbose        TRUE for additional information, FALSE otherwise.\r
\r
  @retval A poitner to a string containing the information.\r
**/\r
CHAR16*\r
EFIAPI\r
LoadedImageProtocolDumpInformation(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  EFI_LOADED_IMAGE_PROTOCOL         *LoadedImage;\r
  EFI_STATUS                        Status;\r
  CHAR16                            *RetVal;\r
  CHAR16                            *Temp;\r
  CHAR16                            *CodeType;\r
  CHAR16                            *DataType;\r
\r
  if (!Verbose) {\r
    return (CatSPrint(NULL, L"LoadedImage"));\r
  }\r
\r
  Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_LI_DUMP_MAIN), NULL);\r
  RetVal = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));\r
  if (Temp == NULL || RetVal == NULL) {\r
    SHELL_FREE_NON_NULL(Temp);\r
    SHELL_FREE_NON_NULL(RetVal);\r
    return NULL;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                TheHandle,\r
                &gEfiLoadedImageProtocolGuid,\r
                (VOID**)&LoadedImage,\r
                gImageHandle,\r
                NULL,\r
                EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
               );\r
\r
  DataType = ConvertMemoryType(LoadedImage->ImageDataType);\r
  CodeType = ConvertMemoryType(LoadedImage->ImageCodeType);\r
\r
  RetVal = CatSPrint(RetVal,\r
                      Temp,\r
                      LoadedImage->Revision,\r
                      LoadedImage->ParentHandle,\r
                      LoadedImage->SystemTable,\r
                      LoadedImage->DeviceHandle,\r
                      LoadedImage->FilePath,\r
                      LoadedImage->LoadOptionsSize,\r
                      LoadedImage->LoadOptions,\r
                      LoadedImage->ImageBase,\r
                      LoadedImage->ImageSize,\r
                      CodeType,\r
                      DataType,\r
                      LoadedImage->Unload);\r
\r
  \r
  SHELL_FREE_NON_NULL(Temp);\r
  SHELL_FREE_NON_NULL(CodeType);\r
  SHELL_FREE_NON_NULL(DataType);\r
\r
  return RetVal;\r
}\r
\r
/**\r
  Function to dump information about PciRootBridgeIo.\r
\r
  This will allocate the return buffer from boot services pool.\r
\r
  @param[in] TheHandle      The handle that has PciRootBridgeIo installed.\r
  @param[in] Verbose        TRUE for additional information, FALSE otherwise.\r
\r
  @retval A poitner to a string containing the information.\r
**/\r
CHAR16*\r
EFIAPI\r
PciRootBridgeIoDumpInformation(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL   *PciRootBridgeIo;\r
  EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *Configuration;\r
  UINT64                            Supports;\r
  UINT64                            Attributes;\r
  CHAR16                            *Temp;\r
  CHAR16                            *Temp2;\r
  CHAR16                            *RetVal;\r
  EFI_STATUS                        Status;\r
\r
  RetVal  = NULL;\r
\r
  if (!Verbose) {\r
    return (CatSPrint(NULL, L"PciRootBridgeIo"));\r
  }\r
\r
  Status = gBS->HandleProtocol(\r
    TheHandle,\r
    &gEfiPciRootBridgeIoProtocolGuid,\r
    (VOID**)&PciRootBridgeIo);\r
\r
  if (EFI_ERROR(Status)) {\r
    return NULL;\r
  }\r
\r
  Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_PH), NULL);\r
  ASSERT (Temp != NULL);\r
  Temp2 = CatSPrint(L"\r\n", Temp, PciRootBridgeIo->ParentHandle);\r
  FreePool(Temp);\r
  RetVal = Temp2;\r
  Temp2 = NULL;\r
 \r
  Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_SEG), NULL);\r
  ASSERT (Temp != NULL);\r
  Temp2 = CatSPrint(RetVal, Temp, PciRootBridgeIo->SegmentNumber);\r
  FreePool(Temp);\r
  FreePool(RetVal);\r
  RetVal = Temp2;\r
  Temp2 = NULL;\r
\r
  Supports   = 0;\r
  Attributes = 0;\r
  Status = PciRootBridgeIo->GetAttributes (PciRootBridgeIo, &Supports, &Attributes);\r
  if (!EFI_ERROR(Status)) {\r
    Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_ATT), NULL);\r
    ASSERT (Temp != NULL);    \r
    Temp2 = CatSPrint(RetVal, Temp, Attributes);\r
    FreePool(Temp);\r
    FreePool(RetVal);\r
    RetVal = Temp2;\r
    Temp2 = NULL;\r
    \r
    Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_SUPPORTS), NULL);\r
    ASSERT (Temp != NULL);\r
    Temp2 = CatSPrint(RetVal, Temp, Supports);\r
    FreePool(Temp);\r
    FreePool(RetVal);\r
    RetVal = Temp2;\r
    Temp2 = NULL;\r
  }\r
\r
  Configuration   = NULL;\r
  Status = PciRootBridgeIo->Configuration (PciRootBridgeIo, (VOID **) &Configuration);\r
  if (!EFI_ERROR(Status) && Configuration != NULL) {\r
    Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_TITLE), NULL);\r
    ASSERT (Temp != NULL);\r
    Temp2 = CatSPrint(RetVal, Temp, Supports);\r
    FreePool(Temp);\r
    FreePool(RetVal);\r
    RetVal = Temp2;\r
    Temp2 = NULL;\r
    while (Configuration->Desc == ACPI_ADDRESS_SPACE_DESCRIPTOR) {\r
      Temp = NULL;\r
      switch (Configuration->ResType) {\r
      case ACPI_ADDRESS_SPACE_TYPE_MEM:\r
        Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_MEM), NULL);\r
        break;\r
      case ACPI_ADDRESS_SPACE_TYPE_IO:\r
        Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_IO), NULL);\r
        break;\r
      case ACPI_ADDRESS_SPACE_TYPE_BUS:\r
        Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_PCIRB_DUMP_BUS), NULL);\r
        break;\r
      }\r
      if (Temp != NULL) {\r
        Temp2 = CatSPrint(RetVal, L"%s", Temp);\r
        FreePool(Temp);\r
        FreePool(RetVal);\r
        RetVal = Temp2;\r
        Temp2 = NULL;\r
      }\r
\r
      Temp2 = CatSPrint(RetVal, \r
        L"%H%02x    %016lx  %016lx  %02x%N\r\n",\r
        Configuration->SpecificFlag,\r
        Configuration->AddrRangeMin,\r
        Configuration->AddrRangeMax,\r
        Configuration->AddrSpaceGranularity\r
        );\r
      FreePool(RetVal);\r
      RetVal = Temp2;\r
      Temp2 = NULL;\r
      Configuration++;\r
    }\r
  }\r
  return (RetVal);\r
}\r
\r
/**\r
  Function to dump information about SimpleTextOut.\r
\r
  This will allocate the return buffer from boot services pool.\r
\r
  @param[in] TheHandle      The handle that has SimpleTextOut installed.\r
  @param[in] Verbose        TRUE for additional information, FALSE otherwise.\r
\r
  @retval A poitner to a string containing the information.\r
**/\r
CHAR16*\r
EFIAPI\r
TxtOutProtocolDumpInformation(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *Dev;\r
  INTN                            Index;\r
  UINTN                           Col;\r
  UINTN                           Row;\r
  EFI_STATUS                      Status;\r
  CHAR16                          *RetVal;\r
  UINTN                           Size;\r
  CHAR16                          *Temp;\r
  UINTN                           NewSize;\r
\r
  if (!Verbose) {\r
    return (NULL);\r
  }\r
\r
  RetVal  = NULL;\r
  Size    = 0;\r
\r
  Status = gBS->HandleProtocol(\r
    TheHandle,\r
    &gEfiSimpleTextOutProtocolGuid,\r
    (VOID**)&Dev);\r
\r
  ASSERT_EFI_ERROR(Status);\r
  ASSERT (Dev != NULL && Dev->Mode != NULL);\r
\r
  Size = (Dev->Mode->MaxMode + 1) * 80;\r
  RetVal = AllocateZeroPool(Size);\r
\r
  Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_TXT_OUT_DUMP_HEADER), NULL);\r
  if (Temp != NULL) {\r
    UnicodeSPrint(RetVal, Size, Temp, Dev, Dev->Mode->Attribute);\r
    FreePool(Temp);\r
  }\r
\r
  //\r
  // Dump TextOut Info\r
  //\r
  Temp = HiiGetString(mHandleParsingHiiHandle, STRING_TOKEN(STR_TXT_OUT_DUMP_LINE), NULL);\r
  for (Index = 0; Index < Dev->Mode->MaxMode; Index++) {\r
    Status = Dev->QueryMode (Dev, Index, &Col, &Row);\r
    NewSize = Size - StrSize(RetVal);\r
    UnicodeSPrint(\r
      RetVal + StrLen(RetVal),\r
      NewSize,\r
      Temp == NULL?L"":Temp,\r
      Index == Dev->Mode->Mode ? L'*' : L' ',\r
      Index,\r
      !EFI_ERROR(Status)?(INTN)Col:-1,\r
      !EFI_ERROR(Status)?(INTN)Row:-1\r
     );\r
  }\r
  FreePool(Temp);\r
  return (RetVal);\r
}\r
\r
STATIC CONST UINTN VersionStringSize = 60;\r
\r
/**\r
  Function to dump information about EfiDriverSupportedEfiVersion protocol.\r
\r
  This will allocate the return buffer from boot services pool.\r
\r
  @param[in] TheHandle      The handle that has the protocol installed.\r
  @param[in] Verbose        TRUE for additional information, FALSE otherwise.\r
\r
  @retval A poitner to a string containing the information.\r
**/\r
CHAR16*\r
EFIAPI\r
DriverEfiVersionProtocolDumpInformation(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  EFI_DRIVER_SUPPORTED_EFI_VERSION_PROTOCOL *DriverEfiVersion;\r
  EFI_STATUS                                Status;\r
  CHAR16                                    *RetVal;\r
\r
  Status = gBS->HandleProtocol(\r
    TheHandle,\r
    &gEfiDriverSupportedEfiVersionProtocolGuid,\r
    (VOID**)&DriverEfiVersion);\r
\r
  ASSERT_EFI_ERROR(Status);\r
\r
  RetVal = AllocateZeroPool(VersionStringSize);\r
  ASSERT(RetVal != NULL);\r
  UnicodeSPrint(RetVal, VersionStringSize, L"0x%08x", DriverEfiVersion->FirmwareVersion);\r
  return (RetVal);\r
}\r
\r
/**\r
  Function to dump information about DevicePath protocol.\r
\r
  This will allocate the return buffer from boot services pool.\r
\r
  @param[in] TheHandle      The handle that has the protocol installed.\r
  @param[in] Verbose        TRUE for additional information, FALSE otherwise.\r
\r
  @retval A poitner to a string containing the information.\r
**/\r
CHAR16*\r
EFIAPI\r
DevicePathProtocolDumpInformation(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL          *DevPath;\r
  CHAR16                            *Temp;\r
  CHAR16                            *Temp2;\r
  EFI_STATUS                        Status;\r
  Temp = NULL;\r
\r
  Status = gBS->OpenProtocol(TheHandle, &gEfiDevicePathProtocolGuid, (VOID**)&DevPath, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
  if (!EFI_ERROR(Status)) {\r
    //\r
    // I cannot decide whether to allow shortcuts here (the second BOOLEAN on the next line)\r
    //\r
    Temp = ConvertDevicePathToText(DevPath, TRUE, TRUE);\r
    gBS->CloseProtocol(TheHandle, &gEfiDevicePathProtocolGuid, gImageHandle, NULL);\r
  }\r
  if (!Verbose && Temp != NULL && StrLen(Temp) > 30) {\r
    Temp2 = NULL;\r
    Temp2 = StrnCatGrow(&Temp2, NULL, Temp+(StrLen(Temp) - 30), 30);\r
    FreePool(Temp);\r
    Temp = Temp2;\r
  }\r
  return (Temp);\r
}\r
\r
//\r
// Put the information on the NT32 protocol GUIDs here so we are not dependant on the Nt32Pkg\r
//\r
#define LOCAL_EFI_WIN_NT_THUNK_PROTOCOL_GUID \\r
  { \\r
    0x58c518b1, 0x76f3, 0x11d4, { 0xbc, 0xea, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
  }\r
\r
#define LOCAL_EFI_WIN_NT_BUS_DRIVER_IO_PROTOCOL_GUID \\r
  { \\r
    0x96eb4ad6, 0xa32a, 0x11d4, { 0xbc, 0xfd, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
  }\r
\r
#define LOCAL_EFI_WIN_NT_SERIAL_PORT_GUID \\r
  { \\r
    0xc95a93d, 0xa006, 0x11d4, { 0xbc, 0xfa, 0x0, 0x80, 0xc7, 0x3c, 0x88, 0x81 } \\r
  }\r
STATIC CONST EFI_GUID WinNtThunkProtocolGuid = LOCAL_EFI_WIN_NT_THUNK_PROTOCOL_GUID;\r
STATIC CONST EFI_GUID WinNtIoProtocolGuid    = LOCAL_EFI_WIN_NT_BUS_DRIVER_IO_PROTOCOL_GUID;\r
STATIC CONST EFI_GUID WinNtSerialPortGuid    = LOCAL_EFI_WIN_NT_SERIAL_PORT_GUID;\r
\r
STATIC CONST GUID_INFO_BLOCK mGuidStringListNT[] = {\r
  {STRING_TOKEN(STR_WINNT_THUNK),           (EFI_GUID*)&WinNtThunkProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_WINNT_DRIVER_IO),       (EFI_GUID*)&WinNtIoProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_WINNT_SERIAL_PORT),     (EFI_GUID*)&WinNtSerialPortGuid,                  NULL},\r
  {STRING_TOKEN(STR_UNKNOWN_DEVICE),        NULL,                                             NULL},\r
};\r
\r
STATIC CONST GUID_INFO_BLOCK mGuidStringList[] = {\r
  {STRING_TOKEN(STR_LOADED_IMAGE),          &gEfiLoadedImageProtocolGuid,                     LoadedImageProtocolDumpInformation},\r
  {STRING_TOKEN(STR_DEVICE_PATH),           &gEfiDevicePathProtocolGuid,                      DevicePathProtocolDumpInformation},\r
  {STRING_TOKEN(STR_IMAGE_PATH),            &gEfiLoadedImageDevicePathProtocolGuid,           DevicePathProtocolDumpInformation},\r
  {STRING_TOKEN(STR_DEVICE_PATH_UTIL),      &gEfiDevicePathUtilitiesProtocolGuid,             NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_TXT),       &gEfiDevicePathToTextProtocolGuid,                NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_FTXT),      &gEfiDevicePathFromTextProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_PC),        &gEfiPcAnsiGuid,                                  NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_VT100),     &gEfiVT100Guid,                                   NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_VT100P),    &gEfiVT100PlusGuid,                               NULL},\r
  {STRING_TOKEN(STR_DEVICE_PATH_VTUTF8),    &gEfiVTUTF8Guid,                                  NULL},\r
  {STRING_TOKEN(STR_DRIVER_BINDING),        &gEfiDriverBindingProtocolGuid,                   NULL},\r
  {STRING_TOKEN(STR_PLATFORM_OVERRIDE),     &gEfiPlatformDriverOverrideProtocolGuid,          NULL},\r
  {STRING_TOKEN(STR_BUS_OVERRIDE),          &gEfiBusSpecificDriverOverrideProtocolGuid,       NULL},\r
  {STRING_TOKEN(STR_DRIVER_DIAG),           &gEfiDriverDiagnosticsProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_DRIVER_DIAG2),          &gEfiDriverDiagnostics2ProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_DRIVER_CN),             &gEfiComponentNameProtocolGuid,                   NULL},\r
  {STRING_TOKEN(STR_DRIVER_CN2),            &gEfiComponentName2ProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_PLAT_DRV_CFG),          &gEfiPlatformToDriverConfigurationProtocolGuid,   NULL},\r
  {STRING_TOKEN(STR_DRIVER_VERSION),        &gEfiDriverSupportedEfiVersionProtocolGuid,       DriverEfiVersionProtocolDumpInformation},\r
  {STRING_TOKEN(STR_TXT_IN),                &gEfiSimpleTextInProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_TXT_IN_EX),             &gEfiSimpleTextInputExProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_TXT_OUT),               &gEfiSimpleTextOutProtocolGuid,                   TxtOutProtocolDumpInformation},\r
  {STRING_TOKEN(STR_SIM_POINTER),           &gEfiSimplePointerProtocolGuid,                   NULL},\r
  {STRING_TOKEN(STR_ABS_POINTER),           &gEfiAbsolutePointerProtocolGuid,                 NULL},\r
  {STRING_TOKEN(STR_SERIAL_IO),             &gEfiSerialIoProtocolGuid,                        NULL},\r
  {STRING_TOKEN(STR_GRAPHICS_OUTPUT),       &gEfiGraphicsOutputProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_EDID_DISCOVERED),       &gEfiEdidDiscoveredProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_EDID_ACTIVE),           &gEfiEdidActiveProtocolGuid,                      NULL},\r
  {STRING_TOKEN(STR_EDID_OVERRIDE),         &gEfiEdidOverrideProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_CON_IN),                &gEfiConsoleInDeviceGuid,                         NULL},\r
  {STRING_TOKEN(STR_CON_OUT),               &gEfiConsoleOutDeviceGuid,                        NULL},\r
  {STRING_TOKEN(STR_STD_ERR),               &gEfiStandardErrorDeviceGuid,                     NULL},\r
  {STRING_TOKEN(STR_LOAD_FILE),             &gEfiLoadFileProtocolGuid,                        NULL},\r
  {STRING_TOKEN(STR_LOAD_FILE2),            &gEfiLoadFile2ProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_SIMPLE_FILE_SYS),       &gEfiSimpleFileSystemProtocolGuid,                NULL},\r
  {STRING_TOKEN(STR_TAPE_IO),               &gEfiTapeIoProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_DISK_IO),               &gEfiDiskIoProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_BLK_IO),                &gEfiBlockIoProtocolGuid,                         NULL},\r
  {STRING_TOKEN(STR_UC),                    &gEfiUnicodeCollationProtocolGuid,                NULL},\r
  {STRING_TOKEN(STR_UC2),                   &gEfiUnicodeCollation2ProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_PCIRB_IO),              &gEfiPciRootBridgeIoProtocolGuid,                 PciRootBridgeIoDumpInformation},\r
  {STRING_TOKEN(STR_PCI_IO),                &gEfiPciIoProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_SCSI_PT),               &gEfiScsiPassThruProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_SCSI_IO),               &gEfiScsiIoProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_SCSI_PT_EXT),           &gEfiExtScsiPassThruProtocolGuid,                 NULL},\r
  {STRING_TOKEN(STR_ISCSI),                 &gEfiIScsiInitiatorNameProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_USB_IO),                &gEfiUsbIoProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_USB_HC),                &gEfiUsbHcProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_USB_HC2),               &gEfiUsb2HcProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_DEBUG_SUPPORT),         &gEfiDebugSupportProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_DEBUG_PORT),            &gEfiDebugPortProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_DECOMPRESS),            &gEfiDecompressProtocolGuid,                      NULL},\r
  {STRING_TOKEN(STR_ACPI_TABLE),            &gEfiAcpiTableProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_EBC_INTERPRETER),       &gEfiEbcProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_SNP),                   &gEfiSimpleNetworkProtocolGuid,                   NULL},\r
  {STRING_TOKEN(STR_NII),                   &gEfiNetworkInterfaceIdentifierProtocolGuid,      NULL},\r
  {STRING_TOKEN(STR_NII_31),                &gEfiNetworkInterfaceIdentifierProtocolGuid_31,   NULL},\r
  {STRING_TOKEN(STR_PXE_BC),                &gEfiPxeBaseCodeProtocolGuid,                     NULL},\r
  {STRING_TOKEN(STR_PXE_CB),                &gEfiPxeBaseCodeCallbackProtocolGuid,             NULL},\r
  {STRING_TOKEN(STR_BIS),                   &gEfiBisProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_MNP_SB),                &gEfiManagedNetworkServiceBindingProtocolGuid,    NULL},\r
  {STRING_TOKEN(STR_MNP),                   &gEfiManagedNetworkProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_ARP_SB),                &gEfiArpServiceBindingProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_ARP),                   &gEfiArpProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_DHCPV4_SB),             &gEfiDhcp4ServiceBindingProtocolGuid,             NULL},\r
  {STRING_TOKEN(STR_DHCPV4),                &gEfiDhcp4ProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_TCPV4_SB),              &gEfiTcp4ServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_TCPV4),                 &gEfiTcp4ProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_IPV4_SB),               &gEfiIp4ServiceBindingProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_IPV4),                  &gEfiIp4ProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_IPV4_CFG),              &gEfiIp4ConfigProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_SHELL_PARAMETERS),      &gEfiShellParametersProtocolGuid,                 NULL},\r
  {STRING_TOKEN(STR_SHELL),                 &gEfiShellProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_UDPV4_SB),              &gEfiUdp4ServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_UDPV4),                 &gEfiUdp4ProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_MTFTPV4_SB),            &gEfiMtftp4ServiceBindingProtocolGuid,            NULL},\r
  {STRING_TOKEN(STR_MTFTPV4),               &gEfiMtftp4ProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_AUTH_INFO),             &gEfiAuthenticationInfoProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_HASH_SB),               &gEfiHashServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_HASH),                  &gEfiHashProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_HII_FONT),              &gEfiHiiFontProtocolGuid,                         NULL},\r
  {STRING_TOKEN(STR_HII_STRING),            &gEfiHiiStringProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_HII_IMAGE),             &gEfiHiiImageProtocolGuid,                        NULL},\r
  {STRING_TOKEN(STR_HII_DATABASE),          &gEfiHiiDatabaseProtocolGuid,                     NULL},\r
  {STRING_TOKEN(STR_HII_CONFIG_ROUT),       &gEfiHiiConfigRoutingProtocolGuid,                NULL},\r
  {STRING_TOKEN(STR_HII_CONFIG_ACC),        &gEfiHiiConfigAccessProtocolGuid,                 NULL},\r
  {STRING_TOKEN(STR_HII_FORM_BROWSER2),     &gEfiFormBrowser2ProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_DRIVER_FAM_OVERRIDE),   &gEfiDriverFamilyOverrideProtocolGuid,            NULL},\r
  {STRING_TOKEN(STR_PCD),                   &gPcdProtocolGuid,                                NULL},\r
  {STRING_TOKEN(STR_TCG),                   &gEfiTcgProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_HII_PACKAGE_LIST),      &gEfiHiiPackageListProtocolGuid,                  NULL},\r
\r
//\r
// the ones under this are deprecated by the current UEFI Spec, but may be found anyways...\r
//\r
  {STRING_TOKEN(STR_SHELL_INTERFACE),       &gEfiShellInterfaceGuid,                          NULL},\r
  {STRING_TOKEN(STR_SHELL_ENV2),            &gEfiShellEnvironment2Guid,                       NULL},\r
  {STRING_TOKEN(STR_SHELL_ENV),             &gEfiShellEnvironment2Guid,                       NULL},\r
  {STRING_TOKEN(STR_DEVICE_IO),             &gEfiDeviceIoProtocolGuid,                        NULL},\r
  {STRING_TOKEN(STR_UGA_DRAW),              &gEfiUgaDrawProtocolGuid,                         NULL},\r
  {STRING_TOKEN(STR_UGA_IO),                &gEfiUgaIoProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_ESP),                   &gEfiPartTypeSystemPartGuid,                      NULL},\r
  {STRING_TOKEN(STR_GPT_NBR),               &gEfiPartTypeLegacyMbrGuid,                       NULL},\r
  {STRING_TOKEN(STR_DRIVER_CONFIG),         &gEfiDriverConfigurationProtocolGuid,             NULL},\r
  {STRING_TOKEN(STR_DRIVER_CONFIG2),        &gEfiDriverConfiguration2ProtocolGuid,            NULL},\r
\r
//\r
// the ones under this are GUID identified structs, not protocols\r
//\r
  {STRING_TOKEN(STR_FILE_INFO),             &gEfiFileInfoGuid,                                NULL},\r
  {STRING_TOKEN(STR_FILE_SYS_INFO),         &gEfiFileSystemInfoGuid,                          NULL},\r
\r
//\r
// the ones under this are misc GUIDS.\r
//\r
  {STRING_TOKEN(STR_EFI_GLOBAL_VARIABLE),   &gEfiGlobalVariableGuid,                          NULL},\r
\r
//\r
// UEFI 2.2\r
//\r
  {STRING_TOKEN(STR_IP6_SB),                &gEfiIp6ServiceBindingProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_IP6),                   &gEfiIp6ProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_IP6_CONFIG),            &gEfiIp6ConfigProtocolGuid,                       NULL},\r
  {STRING_TOKEN(STR_MTFTP6_SB),             &gEfiMtftp6ServiceBindingProtocolGuid,            NULL},\r
  {STRING_TOKEN(STR_MTFTP6),                &gEfiMtftp6ProtocolGuid,                          NULL},\r
  {STRING_TOKEN(STR_DHCP6_SB),              &gEfiDhcp6ServiceBindingProtocolGuid,             NULL},\r
  {STRING_TOKEN(STR_DHCP6),                 &gEfiDhcp6ProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_UDP6_SB),               &gEfiUdp6ServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_UDP6),                  &gEfiUdp6ProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_TCP6_SB),               &gEfiTcp6ServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_TCP6),                  &gEfiTcp6ProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_VLAN_CONFIG),           &gEfiVlanConfigProtocolGuid,                      NULL},\r
  {STRING_TOKEN(STR_EAP),                   &gEfiEapProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_EAP_MGMT),              &gEfiEapManagementProtocolGuid,                   NULL},\r
  {STRING_TOKEN(STR_FTP4_SB),               &gEfiFtp4ServiceBindingProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_FTP4),                  &gEfiFtp4ProtocolGuid,                            NULL},\r
  {STRING_TOKEN(STR_IP_SEC_CONFIG),         &gEfiIpSecConfigProtocolGuid,                     NULL},\r
  {STRING_TOKEN(STR_DH),                    &gEfiDriverHealthProtocolGuid,                    NULL},\r
  {STRING_TOKEN(STR_DEF_IMG_LOAD),          &gEfiDeferredImageLoadProtocolGuid,               NULL},\r
  {STRING_TOKEN(STR_USER_CRED),             &gEfiUserCredentialProtocolGuid,                  NULL},\r
  {STRING_TOKEN(STR_USER_MNGR),             &gEfiUserManagerProtocolGuid,                     NULL},\r
  {STRING_TOKEN(STR_ATA_PASS_THRU),         &gEfiAtaPassThruProtocolGuid,                     NULL},\r
\r
//\r
// UEFI 2.3\r
//\r
  {STRING_TOKEN(STR_FW_MGMT),               &gEfiFirmwareManagementProtocolGuid,              NULL},\r
  {STRING_TOKEN(STR_IP_SEC),                &gEfiIpSecProtocolGuid,                           NULL},\r
  {STRING_TOKEN(STR_IP_SEC2),               &gEfiIpSec2ProtocolGuid,                          NULL},\r
\r
//\r
// UEFI 2.3.1\r
//\r
  {STRING_TOKEN(STR_KMS),                   &gEfiKmsProtocolGuid,                             NULL},\r
  {STRING_TOKEN(STR_BLK_IO2),               &gEfiBlockIo2ProtocolGuid,                        NULL},\r
  {STRING_TOKEN(STR_SSC),                   &gEfiStorageSecurityCommandProtocolGuid,          NULL},\r
  {STRING_TOKEN(STR_UCRED2),                &gEfiUserCredential2ProtocolGuid,                 NULL},\r
\r
//\r
// UEFI 2.4\r
//\r
  {STRING_TOKEN(STR_DISK_IO2),              &gEfiDiskIo2ProtocolGuid,                         NULL},\r
\r
//\r
// PI Spec ones\r
//\r
  {STRING_TOKEN(STR_IDE_CONT_INIT),         &gEfiIdeControllerInitProtocolGuid,               NULL},\r
\r
//\r
// terminator\r
//\r
  {STRING_TOKEN(STR_UNKNOWN_DEVICE),        NULL,                                             NULL},\r
};\r
\r
/**\r
  Function to get the node for a protocol or struct from it's GUID.\r
\r
  if Guid is NULL, then ASSERT.\r
\r
  @param[in] Guid               The GUID to look for the name of.\r
\r
  @return                       The node.\r
**/\r
CONST GUID_INFO_BLOCK *\r
EFIAPI\r
InternalShellGetNodeFromGuid(\r
  IN CONST EFI_GUID* Guid\r
  )\r
{\r
  CONST GUID_INFO_BLOCK *ListWalker;\r
\r
  ASSERT(Guid != NULL);\r
\r
  if (PcdGetBool(PcdShellIncludeNtGuids)) {\r
    for (ListWalker = mGuidStringListNT ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
      if (CompareGuid(ListWalker->GuidId, Guid)) {\r
        return (ListWalker);\r
      }\r
    }\r
  }\r
  for (ListWalker = mGuidStringList ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
    if (CompareGuid(ListWalker->GuidId, Guid)) {\r
      return (ListWalker);\r
    }\r
  }\r
  return (ListWalker);\r
}\r
\r
/**\r
  Function to get the name of a protocol or struct from it's GUID.\r
\r
  if Guid is NULL, then ASSERT.\r
\r
  @param[in] Guid               The GUID to look for the name of.\r
  @param[in] Lang               The language to use.\r
\r
  @return                       pointer to string of the name.  The caller\r
                                is responsible to free this memory.\r
**/\r
CHAR16*\r
EFIAPI\r
GetStringNameFromGuid(\r
  IN CONST EFI_GUID *Guid,\r
  IN CONST CHAR8    *Lang OPTIONAL\r
  )\r
{\r
  CONST GUID_INFO_BLOCK *Id;\r
\r
  Id = InternalShellGetNodeFromGuid(Guid);\r
  return (HiiGetString(mHandleParsingHiiHandle, Id->StringId, Lang));\r
}\r
\r
/**\r
  Function to dump protocol information from a handle.\r
\r
  This function will return a allocated string buffer containing the\r
  information.  The caller is responsible for freeing the memory.\r
\r
  If Guid is NULL, ASSERT().\r
  If TheHandle is NULL, ASSERT().\r
\r
  @param[in] TheHandle      The handle to dump information from.\r
  @param[in] Guid           The GUID of the protocol to dump.\r
  @param[in] Verbose        TRUE for extra info.  FALSE otherwise.\r
\r
  @return                   The pointer to string.\r
  @retval NULL              An error was encountered.\r
**/\r
CHAR16*\r
EFIAPI\r
GetProtocolInformationDump(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST EFI_GUID   *Guid,\r
  IN CONST BOOLEAN    Verbose\r
  )\r
{\r
  CONST GUID_INFO_BLOCK *Id;\r
\r
  ASSERT(TheHandle  != NULL);\r
  ASSERT(Guid       != NULL);\r
\r
  if (TheHandle == NULL || Guid == NULL) {\r
    return (NULL);\r
  }\r
\r
  Id = InternalShellGetNodeFromGuid(Guid);\r
  if (Id != NULL && Id->DumpInfo != NULL) {\r
    return (Id->DumpInfo(TheHandle, Verbose));\r
  }\r
  return (NULL);\r
}\r
\r
/**\r
  Function to get the Guid for a protocol or struct based on it's string name.\r
\r
  @param[in] Name           The pointer to the string name.\r
  @param[in] Lang           The pointer to the language code.\r
  @param[in] Guid           The pointer to the Guid.\r
\r
  @retval EFI_SUCCESS       The operation was sucessful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
GetGuidFromStringName(\r
  IN CONST CHAR16 *Name,\r
  IN CONST CHAR8  *Lang OPTIONAL,\r
  IN EFI_GUID     **Guid\r
  )\r
{\r
  CONST GUID_INFO_BLOCK  *ListWalker;\r
  CHAR16                     *String;\r
\r
  ASSERT(Guid != NULL);\r
  if (Guid == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
  *Guid = NULL;\r
\r
  if (PcdGetBool(PcdShellIncludeNtGuids)) {\r
    for (ListWalker = mGuidStringListNT ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
      String = HiiGetString(mHandleParsingHiiHandle, ListWalker->StringId, Lang);\r
      if (Name != NULL && String != NULL && StringNoCaseCompare (&Name, &String) == 0) {\r
        *Guid = ListWalker->GuidId;\r
      }\r
      SHELL_FREE_NON_NULL(String);\r
      if (*Guid != NULL) {\r
        return (EFI_SUCCESS);\r
      }\r
    }\r
  }\r
  for (ListWalker = mGuidStringList ; ListWalker != NULL && ListWalker->GuidId != NULL ; ListWalker++) {\r
    String = HiiGetString(mHandleParsingHiiHandle, ListWalker->StringId, Lang);\r
    if (Name != NULL && String != NULL && StringNoCaseCompare (&Name, &String) == 0) {\r
      *Guid = ListWalker->GuidId;\r
    }\r
    SHELL_FREE_NON_NULL(String);\r
    if (*Guid != NULL) {\r
      return (EFI_SUCCESS);\r
    }\r
  }\r
  return (EFI_NOT_FOUND);\r
}\r
\r
/**\r
  Get best support language for this driver.\r
  \r
  First base on the user input language  to search, second base on the current \r
  platform used language to search, third get the first language from the \r
  support language list. The caller need to free the buffer of the best language.\r
\r
  @param[in] SupportedLanguages      The support languages for this driver.\r
  @param[in] InputLanguage           The user input language.\r
  @param[in] Iso639Language          Whether get language for ISO639.\r
\r
  @return                            The best support language for this driver.\r
**/\r
CHAR8 *\r
EFIAPI\r
GetBestLanguageForDriver (\r
  IN CONST CHAR8  *SupportedLanguages,\r
  IN CONST CHAR8  *InputLanguage,\r
  IN BOOLEAN      Iso639Language\r
  )\r
{\r
  CHAR8                         *LanguageVariable;\r
  CHAR8                         *BestLanguage;\r
\r
  LanguageVariable = GetVariable (Iso639Language ? L"Lang" : L"PlatformLang", &gEfiGlobalVariableGuid);\r
\r
  BestLanguage = GetBestLanguage(\r
                   SupportedLanguages,\r
                   Iso639Language,\r
                   (InputLanguage != NULL) ? InputLanguage : "",\r
                   (LanguageVariable != NULL) ? LanguageVariable : "",\r
                   SupportedLanguages,\r
                   NULL\r
                   );\r
\r
  if (LanguageVariable != NULL) {\r
    FreePool (LanguageVariable);\r
  }\r
\r
  return BestLanguage;\r
}\r
\r
/**\r
  Function to retrieve the driver name (if possible) from the ComponentName or\r
  ComponentName2 protocol\r
\r
  @param[in] TheHandle      The driver handle to get the name of.\r
  @param[in] Language       The language to use.\r
\r
  @retval NULL              The name could not be found.\r
  @return                   A pointer to the string name.  Do not de-allocate the memory.\r
**/\r
CONST CHAR16*\r
EFIAPI\r
GetStringNameFromHandle(\r
  IN CONST EFI_HANDLE TheHandle,\r
  IN CONST CHAR8      *Language\r
  )\r
{\r
  EFI_COMPONENT_NAME2_PROTOCOL  *CompNameStruct;\r
  EFI_STATUS                    Status;\r
  CHAR16                        *RetVal;\r
  CHAR8                         *BestLang;\r
\r
  BestLang = NULL;\r
\r
  Status = gBS->OpenProtocol(\r
    TheHandle,\r
    &gEfiComponentName2ProtocolGuid,\r
    (VOID**)&CompNameStruct,\r
    gImageHandle,\r
    NULL,\r
    EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
  if (!EFI_ERROR(Status)) {\r
    BestLang = GetBestLanguageForDriver (CompNameStruct->SupportedLanguages, Language, FALSE);\r
    Status = CompNameStruct->GetDriverName(CompNameStruct, BestLang, &RetVal);\r
    if (BestLang != NULL) {\r
      FreePool (BestLang);\r
      BestLang = NULL;\r
    }\r
    if (!EFI_ERROR(Status)) {\r
      return (RetVal);\r
    }\r
  }\r
  Status = gBS->OpenProtocol(\r
    TheHandle,\r
    &gEfiComponentNameProtocolGuid,\r
    (VOID**)&CompNameStruct,\r
    gImageHandle,\r
    NULL,\r
    EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
  if (!EFI_ERROR(Status)) {\r
    BestLang = GetBestLanguageForDriver (CompNameStruct->SupportedLanguages, Language, FALSE);\r
    Status = CompNameStruct->GetDriverName(CompNameStruct, BestLang, &RetVal);\r
    if (BestLang != NULL) {\r
      FreePool (BestLang);\r
    }\r
    if (!EFI_ERROR(Status)) {\r
      return (RetVal);\r
    }\r
  }\r
  return (NULL);\r
}\r
\r
/**\r
  Function to initialize the file global mHandleList object for use in\r
  vonverting handles to index and index to handle.\r
\r
  @retval EFI_SUCCESS     The operation was successful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
InternalShellInitHandleList(\r
  VOID\r
  )\r
{\r
  EFI_STATUS   Status;\r
  EFI_HANDLE   *HandleBuffer;\r
  UINTN        HandleCount;\r
  HANDLE_LIST  *ListWalker;\r
\r
  if (mHandleList.NextIndex != 0) {\r
    return EFI_SUCCESS;\r
  }\r
  InitializeListHead(&mHandleList.List.Link);\r
  mHandleList.NextIndex = 1;\r
  Status = gBS->LocateHandleBuffer (\r
                AllHandles,\r
                NULL,\r
                NULL,\r
                &HandleCount,\r
                &HandleBuffer\r
               );\r
  ASSERT_EFI_ERROR(Status);\r
  if (EFI_ERROR(Status)) {\r
    return (Status);\r
  }\r
  for (mHandleList.NextIndex = 1 ; mHandleList.NextIndex <= HandleCount ; mHandleList.NextIndex++){\r
    ListWalker = AllocateZeroPool(sizeof(HANDLE_LIST));\r
    ASSERT(ListWalker != NULL);\r
    ListWalker->TheHandle = HandleBuffer[mHandleList.NextIndex-1];\r
    ListWalker->TheIndex  = mHandleList.NextIndex;\r
    InsertTailList(&mHandleList.List.Link,&ListWalker->Link);\r
  }\r
  FreePool(HandleBuffer);\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Function to retrieve the human-friendly index of a given handle.  If the handle\r
  does not have a index one will be automatically assigned.  The index value is valid\r
  until the termination of the shell application.\r
\r
  @param[in] TheHandle    The handle to retrieve an index for.\r
\r
  @retval 0               A memory allocation failed.\r
  @return                 The index of the handle.\r
\r
**/\r
UINTN\r
EFIAPI\r
ConvertHandleToHandleIndex(\r
  IN CONST EFI_HANDLE TheHandle\r
  )\r
{\r
  EFI_STATUS   Status;\r
  EFI_GUID     **ProtocolBuffer;\r
  UINTN        ProtocolCount;\r
  HANDLE_LIST  *ListWalker;\r
\r
  if (TheHandle == NULL) {\r
    return 0;\r
  }\r
\r
  InternalShellInitHandleList();\r
\r
  for (ListWalker = (HANDLE_LIST*)GetFirstNode(&mHandleList.List.Link)\r
    ;  !IsNull(&mHandleList.List.Link,&ListWalker->Link)\r
    ;  ListWalker = (HANDLE_LIST*)GetNextNode(&mHandleList.List.Link,&ListWalker->Link)\r
   ){\r
    if (ListWalker->TheHandle == TheHandle) {\r
      //\r
      // Verify that TheHandle is still present in the Handle Database\r
      //\r
      Status = gBS->ProtocolsPerHandle(TheHandle, &ProtocolBuffer, &ProtocolCount);\r
      if (EFI_ERROR (Status)) {\r
        //\r
        // TheHandle is not present in the Handle Database, so delete from the handle list\r
        //\r
        RemoveEntryList (&ListWalker->Link);\r
        return 0;\r
      }\r
      FreePool (ProtocolBuffer);\r
      return (ListWalker->TheIndex);\r
    }\r
  }\r
\r
  //\r
  // Verify that TheHandle is valid handle\r
  //\r
  Status = gBS->ProtocolsPerHandle(TheHandle, &ProtocolBuffer, &ProtocolCount);\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // TheHandle is not valid, so do not add to handle list\r
    //\r
    return 0;\r
  }\r
  FreePool (ProtocolBuffer);\r
\r
  ListWalker = AllocateZeroPool(sizeof(HANDLE_LIST));\r
  ASSERT(ListWalker != NULL);\r
  ListWalker->TheHandle = TheHandle;\r
  ListWalker->TheIndex  = mHandleList.NextIndex++;\r
  InsertTailList(&mHandleList.List.Link,&ListWalker->Link);\r
  return (ListWalker->TheIndex);\r
}\r
\r
\r
\r
/**\r
  Function to retrieve the EFI_HANDLE from the human-friendly index.\r
\r
  @param[in] TheIndex     The index to retrieve the EFI_HANDLE for.\r
\r
  @retval NULL            The index was invalid.\r
  @return                 The EFI_HANDLE that index represents.\r
\r
**/\r
EFI_HANDLE\r
EFIAPI\r
ConvertHandleIndexToHandle(\r
  IN CONST UINTN TheIndex\r
  )\r
{\r
  EFI_STATUS   Status;\r
  EFI_GUID     **ProtocolBuffer;\r
  UINTN        ProtocolCount;\r
  HANDLE_LIST *ListWalker;\r
\r
  InternalShellInitHandleList();\r
\r
  if (TheIndex >= mHandleList.NextIndex) {\r
    return NULL;\r
  }\r
\r
  for (ListWalker = (HANDLE_LIST*)GetFirstNode(&mHandleList.List.Link)\r
    ;  !IsNull(&mHandleList.List.Link,&ListWalker->Link)\r
    ;  ListWalker = (HANDLE_LIST*)GetNextNode(&mHandleList.List.Link,&ListWalker->Link)\r
   ){\r
    if (ListWalker->TheIndex == TheIndex && ListWalker->TheHandle != NULL) {\r
      //\r
      // Verify that LinkWalker->TheHandle is valid handle\r
      //\r
      Status = gBS->ProtocolsPerHandle(ListWalker->TheHandle, &ProtocolBuffer, &ProtocolCount);\r
      if (EFI_ERROR (Status)) {\r
        //\r
        // TheHandle is not valid, so do not add to handle list\r
        //\r
        ListWalker->TheHandle = NULL;\r
      }\r
      return (ListWalker->TheHandle);\r
    }\r
  }\r
  return NULL;\r
}\r
\r
/**\r
  Gets all the related EFI_HANDLEs based on the mask supplied.\r
\r
  This function scans all EFI_HANDLES in the UEFI environment's handle database\r
  and returns the ones with the specified relationship (Mask) to the specified\r
  controller handle.\r
\r
  If both DriverBindingHandle and ControllerHandle are NULL, then ASSERT.\r
  If MatchingHandleCount is NULL, then ASSERT.\r
\r
  If MatchingHandleBuffer is not NULL upon a successful return the memory must be\r
  caller freed.\r
\r
  @param[in] DriverBindingHandle    The handle with Driver Binding protocol on it.\r
  @param[in] ControllerHandle       The handle with Device Path protocol on it.\r
  @param[in] MatchingHandleCount    The pointer to UINTN that specifies the number of HANDLES in\r
                                    MatchingHandleBuffer.\r
  @param[out] MatchingHandleBuffer  On a successful return, a buffer of MatchingHandleCount\r
                                    EFI_HANDLEs with a terminating NULL EFI_HANDLE.\r
  @param[out] HandleType            An array of type information.\r
\r
  @retval EFI_SUCCESS               The operation was successful, and any related handles\r
                                    are in MatchingHandleBuffer.\r
  @retval EFI_NOT_FOUND             No matching handles were found.\r
  @retval EFI_INVALID_PARAMETER     A parameter was invalid or out of range.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ParseHandleDatabaseByRelationshipWithType (\r
  IN CONST EFI_HANDLE DriverBindingHandle OPTIONAL,\r
  IN CONST EFI_HANDLE ControllerHandle OPTIONAL,\r
  IN UINTN            *HandleCount,\r
  OUT EFI_HANDLE      **HandleBuffer,\r
  OUT UINTN           **HandleType\r
  )\r
{\r
  EFI_STATUS                          Status;\r
  UINTN                               HandleIndex;\r
  EFI_GUID                            **ProtocolGuidArray;\r
  UINTN                               ArrayCount;\r
  UINTN                               ProtocolIndex;\r
  EFI_OPEN_PROTOCOL_INFORMATION_ENTRY *OpenInfo;\r
  UINTN                               OpenInfoCount;\r
  UINTN                               OpenInfoIndex;\r
  UINTN                               ChildIndex;\r
  INTN                                DriverBindingHandleIndex;\r
\r
  ASSERT(HandleCount  != NULL);\r
  ASSERT(HandleBuffer != NULL);\r
  ASSERT(HandleType   != NULL);\r
  ASSERT(DriverBindingHandle != NULL || ControllerHandle != NULL);\r
\r
  *HandleCount                  = 0;\r
  *HandleBuffer                 = NULL;\r
  *HandleType                   = NULL;\r
\r
  //\r
  // Retrieve the list of all handles from the handle database\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                AllHandles,\r
                NULL,\r
                NULL,\r
                HandleCount,\r
                HandleBuffer\r
               );\r
  if (EFI_ERROR (Status)) {\r
    return (Status);\r
  }\r
\r
  *HandleType = AllocateZeroPool (*HandleCount * sizeof (UINTN));\r
  ASSERT(*HandleType != NULL);\r
\r
  DriverBindingHandleIndex = -1;\r
  for (HandleIndex = 0; HandleIndex < *HandleCount; HandleIndex++) {\r
    if (DriverBindingHandle != NULL && (*HandleBuffer)[HandleIndex] == DriverBindingHandle) {\r
      DriverBindingHandleIndex = (INTN)HandleIndex;\r
    }\r
  }\r
\r
  for (HandleIndex = 0; HandleIndex < *HandleCount; HandleIndex++) {\r
    //\r
    // Retrieve the list of all the protocols on each handle\r
    //\r
    Status = gBS->ProtocolsPerHandle (\r
                  (*HandleBuffer)[HandleIndex],\r
                  &ProtocolGuidArray,\r
                  &ArrayCount\r
                 );\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    for (ProtocolIndex = 0; ProtocolIndex < ArrayCount; ProtocolIndex++) {\r
\r
      //\r
      // Set the bit describing what this handle has\r
      //\r
      if        (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiLoadedImageProtocolGuid)         ) {\r
        (*HandleType)[HandleIndex] |= HR_IMAGE_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverBindingProtocolGuid)       ) {\r
        (*HandleType)[HandleIndex] |= HR_DRIVER_BINDING_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverConfiguration2ProtocolGuid)) {\r
        (*HandleType)[HandleIndex] |= HR_DRIVER_CONFIGURATION_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverConfigurationProtocolGuid) ) {\r
        (*HandleType)[HandleIndex] |= HR_DRIVER_CONFIGURATION_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverDiagnostics2ProtocolGuid)  ) {\r
        (*HandleType)[HandleIndex] |= HR_DRIVER_DIAGNOSTICS_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDriverDiagnosticsProtocolGuid)   ) {\r
        (*HandleType)[HandleIndex] |= HR_DRIVER_DIAGNOSTICS_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiComponentName2ProtocolGuid)      ) {\r
        (*HandleType)[HandleIndex] |= HR_COMPONENT_NAME_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiComponentNameProtocolGuid)       ) {\r
        (*HandleType)[HandleIndex] |= HR_COMPONENT_NAME_HANDLE;\r
      } else if (CompareGuid (ProtocolGuidArray[ProtocolIndex], &gEfiDevicePathProtocolGuid)          ) {\r
        (*HandleType)[HandleIndex] |= HR_DEVICE_HANDLE;\r
      } else {\r
        DEBUG_CODE_BEGIN();\r
        ASSERT((*HandleType)[HandleIndex] == (*HandleType)[HandleIndex]);\r
        DEBUG_CODE_END();\r
      }\r
      //\r
      // Retrieve the list of agents that have opened each protocol\r
      //\r
      Status = gBS->OpenProtocolInformation (\r
                      (*HandleBuffer)[HandleIndex],\r
                      ProtocolGuidArray[ProtocolIndex],\r
                      &OpenInfo,\r
                      &OpenInfoCount\r
                     );\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      if (ControllerHandle == NULL) {\r
        //\r
        // ControllerHandle == NULL and DriverBindingHandle != NULL.  \r
        // Return information on all the controller handles that the driver specified by DriverBindingHandle is managing\r
        //\r
        for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
          if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle && (OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
            (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
            if (DriverBindingHandleIndex != -1) {\r
              (*HandleType)[DriverBindingHandleIndex] |= HR_DEVICE_DRIVER;\r
            }\r
          }\r
          if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle && (OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
            (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
            if (DriverBindingHandleIndex != -1) {\r
              (*HandleType)[DriverBindingHandleIndex] |= (HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
            }\r
            for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
              if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
                (*HandleType)[ChildIndex] |= (HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
              }\r
            }\r
          }\r
        }\r
      }\r
      if (DriverBindingHandle == NULL && ControllerHandle != NULL) {\r
        if (ControllerHandle == (*HandleBuffer)[HandleIndex]) {\r
          (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
          for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
              for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
                if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
                  (*HandleType)[ChildIndex] |= HR_DEVICE_DRIVER;\r
                }\r
              }\r
            }\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
              for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
                if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
                  (*HandleType)[ChildIndex] |= (HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
                }\r
                if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
                  (*HandleType)[ChildIndex] |= (HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
                }\r
              }\r
            }\r
          }\r
        } else {\r
          for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
              if (OpenInfo[OpenInfoIndex].ControllerHandle == ControllerHandle) {\r
                (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_PARENT_HANDLE);\r
              }\r
            }\r
          }\r
        }\r
      }\r
      if (DriverBindingHandle != NULL && ControllerHandle != NULL) {\r
        if (ControllerHandle == (*HandleBuffer)[HandleIndex]) {\r
          (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_CONTROLLER_HANDLE);\r
          for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) != 0) {\r
              if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle) {\r
                if (DriverBindingHandleIndex != -1) {\r
                  (*HandleType)[DriverBindingHandleIndex] |= HR_DEVICE_DRIVER;\r
                }\r
              }\r
            }\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
              if (OpenInfo[OpenInfoIndex].AgentHandle == DriverBindingHandle) {\r
                for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
                  if (OpenInfo[OpenInfoIndex].ControllerHandle == (*HandleBuffer)[ChildIndex]) {\r
                    (*HandleType)[ChildIndex] |= (HR_DEVICE_HANDLE | HR_CHILD_HANDLE);\r
                  }\r
                }\r
              }\r
\r
              for (ChildIndex = 0; ChildIndex < *HandleCount; ChildIndex++) {\r
                if (OpenInfo[OpenInfoIndex].AgentHandle == (*HandleBuffer)[ChildIndex]) {\r
                  (*HandleType)[ChildIndex] |= (HR_BUS_DRIVER | HR_DEVICE_DRIVER);\r
                }\r
              }\r
            }\r
          }\r
        } else {\r
          for (OpenInfoIndex = 0; OpenInfoIndex < OpenInfoCount; OpenInfoIndex++) {\r
            if ((OpenInfo[OpenInfoIndex].Attributes & EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) != 0) {\r
              if (OpenInfo[OpenInfoIndex].ControllerHandle == ControllerHandle) {\r
                (*HandleType)[HandleIndex] |= (HR_DEVICE_HANDLE | HR_PARENT_HANDLE);\r
              }\r
            }\r
          }\r
        }\r
      }\r
      FreePool (OpenInfo);\r
    }\r
    FreePool (ProtocolGuidArray);\r
  }\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Gets all the related EFI_HANDLEs based on the single EFI_HANDLE and the mask\r
  supplied.\r
\r
  This function will scan all EFI_HANDLES in the UEFI environment's handle database\r
  and return all the ones with the specified relationship (Mask) to the specified\r
  controller handle.\r
\r
  If both DriverBindingHandle and ControllerHandle are NULL, then ASSERT.\r
  If MatchingHandleCount is NULL, then ASSERT.\r
\r
  If MatchingHandleBuffer is not NULL upon a sucessful return the memory must be\r
  caller freed.\r
\r
  @param[in] DriverBindingHandle    Handle to a object with Driver Binding protocol\r
                                    on it.\r
  @param[in] ControllerHandle       Handle to a device with Device Path protocol on it.\r
  @param[in] Mask                   Mask of what relationship(s) is desired.\r
  @param[in] MatchingHandleCount    Poitner to UINTN specifying number of HANDLES in\r
                                    MatchingHandleBuffer.\r
  @param[out] MatchingHandleBuffer  On a sucessful return a buffer of MatchingHandleCount\r
                                    EFI_HANDLEs and a terminating NULL EFI_HANDLE.\r
\r
  @retval EFI_SUCCESS               The operation was sucessful and any related handles\r
                                    are in MatchingHandleBuffer;\r
  @retval EFI_NOT_FOUND             No matching handles were found.\r
  @retval EFI_INVALID_PARAMETER     A parameter was invalid or out of range.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ParseHandleDatabaseByRelationship (\r
  IN CONST EFI_HANDLE       DriverBindingHandle OPTIONAL,\r
  IN CONST EFI_HANDLE       ControllerHandle OPTIONAL,\r
  IN CONST UINTN            Mask,\r
  IN UINTN                  *MatchingHandleCount,\r
  OUT EFI_HANDLE            **MatchingHandleBuffer OPTIONAL\r
  )\r
{\r
  EFI_STATUS            Status;\r
  UINTN                 HandleCount;\r
  EFI_HANDLE            *HandleBuffer;\r
  UINTN                 *HandleType;\r
  UINTN                 HandleIndex;\r
\r
  ASSERT(MatchingHandleCount != NULL);\r
  ASSERT(DriverBindingHandle != NULL || ControllerHandle != NULL);\r
\r
  if ((Mask & HR_VALID_MASK) != Mask) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  if ((Mask & HR_CHILD_HANDLE) != 0 && DriverBindingHandle == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
\r
  *MatchingHandleCount = 0;\r
  if (MatchingHandleBuffer != NULL) {\r
    *MatchingHandleBuffer = NULL;\r
  }\r
\r
  HandleBuffer  = NULL;\r
  HandleType    = NULL;\r
\r
  Status = ParseHandleDatabaseByRelationshipWithType (\r
            DriverBindingHandle,\r
            ControllerHandle,\r
            &HandleCount,\r
            &HandleBuffer,\r
            &HandleType\r
           );\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Count the number of handles that match the attributes in Mask\r
    //\r
    for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {\r
      if ((HandleType[HandleIndex] & Mask) == Mask) {\r
        (*MatchingHandleCount)++;\r
      }\r
    }\r
    //\r
    // If no handles match the attributes in Mask then return EFI_NOT_FOUND\r
    //\r
    if (*MatchingHandleCount == 0) {\r
      Status = EFI_NOT_FOUND;\r
    } else {\r
\r
      if (MatchingHandleBuffer == NULL) {\r
        //\r
        // Someone just wanted the count...\r
        //\r
        Status = EFI_SUCCESS;\r
      } else {\r
        //\r
        // Allocate a handle buffer for the number of handles that matched the attributes in Mask\r
        //\r
        *MatchingHandleBuffer = AllocateZeroPool ((*MatchingHandleCount +1)* sizeof (EFI_HANDLE));\r
        ASSERT(*MatchingHandleBuffer != NULL);\r
\r
        for (HandleIndex = 0,*MatchingHandleCount = 0\r
          ;  HandleIndex < HandleCount\r
          ;  HandleIndex++\r
         ){\r
          //\r
          // Fill the allocated buffer with the handles that matched the attributes in Mask\r
          //\r
          if ((HandleType[HandleIndex] & Mask) == Mask) {\r
            (*MatchingHandleBuffer)[(*MatchingHandleCount)++] = HandleBuffer[HandleIndex];\r
          }\r
        }\r
\r
        //\r
        // Make the last one NULL\r
        //\r
        (*MatchingHandleBuffer)[*MatchingHandleCount] = NULL;\r
\r
        Status = EFI_SUCCESS;\r
      } // MacthingHandleBuffer == NULL (ELSE)\r
    } // *MatchingHandleCount  == 0 (ELSE)\r
  } // no error on ParseHandleDatabaseByRelationshipWithType\r
\r
  if (HandleBuffer != NULL) {\r
    FreePool (HandleBuffer);\r
  }\r
\r
  if (HandleType != NULL) {\r
    FreePool (HandleType);\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Gets handles for any child controllers of the passed in controller.\r
\r
  @param[in] ControllerHandle       The handle of the "parent controller"\r
  @param[in] MatchingHandleCount    Pointer to the number of handles in\r
                                    MatchingHandleBuffer on return.\r
  @param[out] MatchingHandleBuffer  Buffer containing handles on a successful\r
                                    return.\r
\r
\r
  @retval EFI_SUCCESS               The operation was sucessful.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ParseHandleDatabaseForChildControllers(\r
  IN CONST EFI_HANDLE       ControllerHandle,\r
  IN UINTN                  *MatchingHandleCount,\r
  OUT EFI_HANDLE            **MatchingHandleBuffer OPTIONAL\r
  )\r
{\r
  EFI_STATUS  Status;\r
  UINTN       HandleIndex;\r
  UINTN       DriverBindingHandleCount;\r
  EFI_HANDLE  *DriverBindingHandleBuffer;\r
  UINTN       DriverBindingHandleIndex;\r
  UINTN       ChildControllerHandleCount;\r
  EFI_HANDLE  *ChildControllerHandleBuffer;\r
  UINTN       ChildControllerHandleIndex;\r
  EFI_HANDLE  *HandleBufferForReturn;\r
\r
  if (MatchingHandleCount == NULL) {\r
    return (EFI_INVALID_PARAMETER);\r
  }\r
  *MatchingHandleCount = 0;\r
\r
  Status = PARSE_HANDLE_DATABASE_UEFI_DRIVERS (\r
            ControllerHandle,\r
            &DriverBindingHandleCount,\r
            &DriverBindingHandleBuffer\r
           );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Get a buffer big enough for all the controllers.\r
  //\r
  HandleBufferForReturn = GetHandleListByProtocol(NULL);\r
  if (HandleBufferForReturn == NULL) {\r
    FreePool (DriverBindingHandleBuffer);\r
    return (EFI_NOT_FOUND);\r
  }\r
\r
  for (DriverBindingHandleIndex = 0; DriverBindingHandleIndex < DriverBindingHandleCount; DriverBindingHandleIndex++) {\r
    Status = PARSE_HANDLE_DATABASE_MANAGED_CHILDREN (\r
              DriverBindingHandleBuffer[DriverBindingHandleIndex],\r
              ControllerHandle,\r
              &ChildControllerHandleCount,\r
              &ChildControllerHandleBuffer\r
             );\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    for (ChildControllerHandleIndex = 0;\r
         ChildControllerHandleIndex < ChildControllerHandleCount;\r
         ChildControllerHandleIndex++\r
       ) {\r
      for (HandleIndex = 0; HandleIndex < *MatchingHandleCount; HandleIndex++) {\r
        if (HandleBufferForReturn[HandleIndex] == ChildControllerHandleBuffer[ChildControllerHandleIndex]) {\r
          break;\r
        }\r
      }\r
      if (HandleIndex >= *MatchingHandleCount) {\r
        HandleBufferForReturn[(*MatchingHandleCount)++] = ChildControllerHandleBuffer[ChildControllerHandleIndex];\r
      }\r
    }\r
\r
    FreePool (ChildControllerHandleBuffer);\r
  }\r
\r
  FreePool (DriverBindingHandleBuffer);\r
\r
  if (MatchingHandleBuffer != NULL) {\r
    *MatchingHandleBuffer = HandleBufferForReturn;\r
  } else {\r
    FreePool(HandleBufferForReturn);\r
  }\r
\r
  return (EFI_SUCCESS);\r
}\r
\r
/**\r
  Appends 1 buffer to another buffer.  This will re-allocate the destination buffer\r
  if necessary to fit all of the data.\r
\r
  If DestinationBuffer is NULL, then ASSERT().\r
\r
  @param[in, out]  DestinationBuffer The pointer to the pointer to the buffer to append onto.\r
  @param[in, out]  DestinationSize   The pointer to the size of DestinationBuffer.\r
  @param[in]       SourceBuffer      The pointer to the buffer to append onto DestinationBuffer.\r
  @param[in]       SourceSize        The number of bytes of SourceBuffer to append.\r
\r
  @retval NULL                      A memory allocation failed.\r
  @retval NULL                      A parameter was invalid.\r
  @return                           A pointer to (*DestinationBuffer).\r
**/\r
VOID*\r
EFIAPI\r
BuffernCatGrow (\r
  IN OUT VOID   **DestinationBuffer,\r
  IN OUT UINTN  *DestinationSize,\r
  IN     VOID   *SourceBuffer,\r
  IN     UINTN  SourceSize\r
  )\r
{\r
  UINTN LocalDestinationSize;\r
  UINTN LocalDestinationFinalSize;\r
\r
  ASSERT(DestinationBuffer != NULL);\r
\r
  if (SourceSize == 0 || SourceBuffer == NULL) {\r
    return (*DestinationBuffer);\r
  }\r
\r
  if (DestinationSize == NULL) {\r
    LocalDestinationSize = 0;\r
  } else {\r
    LocalDestinationSize = *DestinationSize;\r
  }\r
\r
  LocalDestinationFinalSize = LocalDestinationSize + SourceSize;\r
\r
  if (DestinationSize != NULL) {\r
    *DestinationSize = LocalDestinationSize;\r
  }\r
\r
  if (LocalDestinationSize == 0) {\r
    // allcoate\r
    *DestinationBuffer = AllocateZeroPool(LocalDestinationFinalSize);\r
  } else {\r
    // reallocate\r
    *DestinationBuffer = ReallocatePool(LocalDestinationSize, LocalDestinationFinalSize, *DestinationBuffer);\r
  }\r
\r
  ASSERT(*DestinationBuffer != NULL);\r
\r
  // copy\r
  return (CopyMem(((UINT8*)(*DestinationBuffer)) + LocalDestinationSize, SourceBuffer, SourceSize));\r
}\r
\r
/**\r
  Gets handles for any child devices produced by the passed in driver.\r
\r
  @param[in] DriverHandle           The handle of the driver.\r
  @param[in] MatchingHandleCount    Pointer to the number of handles in\r
                                    MatchingHandleBuffer on return.\r
  @param[out] MatchingHandleBuffer  Buffer containing handles on a successful\r
                                    return.\r
  @retval EFI_SUCCESS               The operation was sucessful.\r
  @sa ParseHandleDatabaseByRelationship\r
**/\r
EFI_STATUS\r
EFIAPI\r
ParseHandleDatabaseForChildDevices(\r
  IN CONST EFI_HANDLE       DriverHandle,\r
  IN UINTN                  *MatchingHandleCount,\r
  OUT EFI_HANDLE            **MatchingHandleBuffer OPTIONAL\r
  )\r
{\r
  EFI_HANDLE      *Buffer;\r
  EFI_HANDLE      *Buffer2;\r
  UINTN           Count1;\r
  UINTN           Count2;\r
  UINTN           HandleIndex;\r
  EFI_STATUS      Status;\r
  UINTN           HandleBufferSize;\r
\r
  ASSERT(MatchingHandleCount != NULL);\r
\r
  HandleBufferSize      = 0;\r
  Buffer                = NULL;\r
  Buffer2               = NULL;\r
  *MatchingHandleCount  = 0;\r
\r
  Status = PARSE_HANDLE_DATABASE_DEVICES (\r
            DriverHandle,\r
            &Count1,\r
            &Buffer\r
           );\r
  if (!EFI_ERROR (Status)) {\r
    for (HandleIndex = 0; HandleIndex < Count1; HandleIndex++) {\r
      //\r
      // now find the children\r
      //\r
      Status = PARSE_HANDLE_DATABASE_MANAGED_CHILDREN (\r
                DriverHandle,\r
                Buffer[HandleIndex],\r
                &Count2,\r
                &Buffer2\r
               );\r
      if (EFI_ERROR(Status)) {\r
        break;\r
      }\r
      //\r
      // save out required and optional data elements\r
      //\r
      *MatchingHandleCount += Count2;\r
      if (MatchingHandleBuffer != NULL) {\r
        *MatchingHandleBuffer = BuffernCatGrow((VOID**)MatchingHandleBuffer, &HandleBufferSize, Buffer2, Count2 * sizeof(Buffer2[0]));\r
      }\r
\r
      //\r
      // free the memory\r
      //\r
      if (Buffer2 != NULL) {\r
        FreePool(Buffer2);\r
      }\r
    }\r
  }\r
\r
  if (Buffer != NULL) {\r
    FreePool(Buffer);\r
  }\r
  return (Status);\r
}\r
\r
/**\r
  Function to get all handles that support a given protocol or all handles.\r
\r
  @param[in] ProtocolGuid The guid of the protocol to get handles for.  If NULL\r
                          then the function will return all handles.\r
\r
  @retval NULL            A memory allocation failed.\r
  @return                 A NULL terminated list of handles.\r
**/\r
EFI_HANDLE*\r
EFIAPI\r
GetHandleListByProtocol (\r
  IN CONST EFI_GUID *ProtocolGuid OPTIONAL\r
  )\r
{\r
  EFI_HANDLE          *HandleList;\r
  UINTN               Size;\r
  EFI_STATUS          Status;\r
\r
  Size = 0;\r
  HandleList = NULL;\r
\r
  //\r
  // We cannot use LocateHandleBuffer since we need that NULL item on the ends of the list!\r
  //\r
  if (ProtocolGuid == NULL) {\r
    Status = gBS->LocateHandle(AllHandles, NULL, NULL, &Size, HandleList);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      HandleList = AllocateZeroPool(Size + sizeof(EFI_HANDLE));\r
      if (HandleList == NULL) {\r
        return (NULL);\r
      }\r
      Status = gBS->LocateHandle(AllHandles, NULL, NULL, &Size, HandleList);\r
      HandleList[Size/sizeof(EFI_HANDLE)] = NULL;\r
    }\r
  } else {\r
    Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)ProtocolGuid, NULL, &Size, HandleList);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      HandleList = AllocateZeroPool(Size + sizeof(EFI_HANDLE));\r
      if (HandleList == NULL) {\r
        return (NULL);\r
      }\r
      Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)ProtocolGuid, NULL, &Size, HandleList);\r
      HandleList[Size/sizeof(EFI_HANDLE)] = NULL;\r
    }\r
  }\r
  if (EFI_ERROR(Status)) {\r
    if (HandleList != NULL) {\r
      FreePool(HandleList);\r
    }\r
    return (NULL);\r
  }\r
  return (HandleList);\r
}\r
\r
/**\r
  Function to get all handles that support some protocols.\r
\r
  @param[in] ProtocolGuids  A NULL terminated list of protocol GUIDs.\r
\r
  @retval NULL              A memory allocation failed.\r
  @retval NULL              ProtocolGuids was NULL.\r
  @return                   A NULL terminated list of EFI_HANDLEs.\r
**/\r
EFI_HANDLE*\r
EFIAPI\r
GetHandleListByProtocolList (\r
  IN CONST EFI_GUID **ProtocolGuids\r
  )\r
{\r
  EFI_HANDLE          *HandleList;\r
  UINTN               Size;\r
  UINTN               TotalSize;\r
  UINTN               TempSize;\r
  EFI_STATUS          Status;\r
  CONST EFI_GUID      **GuidWalker;\r
  EFI_HANDLE          *HandleWalker1;\r
  EFI_HANDLE          *HandleWalker2;\r
\r
  Size        = 0;\r
  HandleList  = NULL;\r
  TotalSize   = sizeof(EFI_HANDLE);\r
\r
  for (GuidWalker = ProtocolGuids ; GuidWalker != NULL && *GuidWalker != NULL ; GuidWalker++,Size = 0){\r
    Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)(*GuidWalker), NULL, &Size, NULL);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      TotalSize += Size;\r
    }\r
  }\r
\r
  //\r
  // No handles were found... \r
  //\r
  if (TotalSize == sizeof(EFI_HANDLE)) {\r
    return (NULL);\r
  }\r
\r
  HandleList = AllocateZeroPool(TotalSize);\r
  if (HandleList == NULL) {\r
    return (NULL);\r
  }\r
\r
  Size = 0;\r
  for (GuidWalker = ProtocolGuids ; GuidWalker != NULL && *GuidWalker != NULL ; GuidWalker++){\r
    TempSize = TotalSize - Size;\r
    Status = gBS->LocateHandle(ByProtocol, (EFI_GUID*)(*GuidWalker), NULL, &TempSize, HandleList+(Size/sizeof(EFI_HANDLE)));\r
\r
    //\r
    // Allow for missing protocols... Only update the 'used' size upon success.\r
    //\r
    if (!EFI_ERROR(Status)) {\r
      Size += TempSize;\r
    }\r
  }\r
  ASSERT(HandleList[(TotalSize/sizeof(EFI_HANDLE))-1] == NULL);\r
\r
  for (HandleWalker1 = HandleList ; HandleWalker1 != NULL && *HandleWalker1 != NULL ; HandleWalker1++) {\r
    for (HandleWalker2 = HandleWalker1 + 1; HandleWalker2 != NULL && *HandleWalker2 != NULL ; HandleWalker2++) {\r
      if (*HandleWalker1 == *HandleWalker2) {\r
        //\r
        // copy memory back 1 handle width.\r
        //\r
        CopyMem(HandleWalker2, HandleWalker2 + 1, TotalSize - ((HandleWalker2-HandleList+1)*sizeof(EFI_HANDLE)));\r
      }\r
    }\r
  }\r
\r
  return (HandleList);\r
}\r
\r
\r
\r
\r
\r
\r
\r
\r
\r
\r