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

/** @file\r
  Main file for LoadPciRom shell Debug1 function.\r
\r
  Copyright (c) 2005 - 2012, 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 "UefiShellDebug1CommandsLib.h"\r
#include <IndustryStandard/Pci22.h>\r
#include <IndustryStandard/Pci23.h>\r
#include <IndustryStandard/PeImage.h>\r
#include <Protocol/Decompress.h>\r
\r
/**\r
  Connects all available drives and controllers.\r
\r
  @retval EFI_SUCCESS     The operation was successful.\r
  @retval EFI_ABORTED     The abort mechanism was received.\r
**/\r
EFI_STATUS\r
EFIAPI\r
LoadPciRomConnectAllDriversToAllControllers (\r
  VOID\r
  );\r
\r
/**\r
  Command entry point.\r
\r
  @param[in] RomBar       The Rom Base address.\r
  @param[in] RomSize      The Rom size.\r
  @param[in] FileName     The file name.\r
\r
  @retval EFI_SUCCESS             The command completed successfully.\r
  @retval EFI_INVALID_PARAMETER   Command usage error.\r
  @retval EFI_UNSUPPORTED         Protocols unsupported.\r
  @retval EFI_OUT_OF_RESOURCES    Out of memory.\r
  @retval Other value             Unknown error.\r
**/\r
EFI_STATUS\r
EFIAPI\r
LoadEfiDriversFromRomImage (\r
  VOID                      *RomBar,\r
  UINTN                     RomSize,\r
  CONST CHAR16              *FileName\r
  );\r
\r
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-nc", TypeFlag},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'loadpcirom' 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
ShellCommandRunLoadPciRom (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_SHELL_FILE_INFO     *FileList;\r
  UINTN                   SourceSize;\r
  UINT8                   *File1Buffer;\r
  EFI_STATUS              Status;\r
  LIST_ENTRY              *Package;\r
  CHAR16                  *ProblemParam;\r
  SHELL_STATUS            ShellStatus;\r
  BOOLEAN                 Connect;\r
  CONST CHAR16            *Param;\r
  UINTN                   ParamCount;\r
  EFI_SHELL_FILE_INFO     *Node;\r
  //\r
  // Local variable initializations\r
  //\r
  File1Buffer = NULL;\r
  ShellStatus = SHELL_SUCCESS;\r
  FileList    = NULL;\r
\r
\r
  //\r
  // verify number of 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, ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\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);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      if (ShellCommandLineGetFlag(Package, L"-nc")) {\r
        Connect = FALSE;\r
      } else {\r
        Connect = TRUE;\r
      }\r
\r
      //\r
      // get a list with each file specified by parameters\r
      // if parameter is a directory then add all the files below it to the list\r
      //\r
      for ( ParamCount = 1, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
          ; Param != NULL\r
          ; ParamCount++, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
         ){\r
        Status = ShellOpenFileMetaArg((CHAR16*)Param, EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);\r
        if (EFI_ERROR(Status)) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, Param, Status);\r
          ShellStatus = SHELL_ACCESS_DENIED;\r
          break;\r
        }\r
      }\r
      if (ShellStatus == SHELL_SUCCESS  && FileList != NULL) {\r
        //\r
        // loop through the list and make sure we are not aborting...\r
        //\r
        for ( Node = (EFI_SHELL_FILE_INFO*)GetFirstNode(&FileList->Link)\r
            ; !IsNull(&FileList->Link, &Node->Link) && !ShellGetExecutionBreakFlag()\r
            ; Node = (EFI_SHELL_FILE_INFO*)GetNextNode(&FileList->Link, &Node->Link)\r
           ){\r
          if (EFI_ERROR(Node->Status)){\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, Node->FullName);\r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
            continue;\r
          }\r
          if (FileHandleIsDirectory(Node->Handle) == EFI_SUCCESS) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, Node->FullName);\r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
            continue;\r
          }\r
          SourceSize  = (UINTN) Node->Info->FileSize;\r
          File1Buffer = AllocateZeroPool (SourceSize);\r
          ASSERT(File1Buffer != NULL);\r
          Status = gEfiShellProtocol->ReadFile(Node->Handle, &SourceSize, File1Buffer);\r
          if (EFI_ERROR(Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_READ_FAIL), gShellDebug1HiiHandle, Node->FullName);\r
            ShellStatus = SHELL_INVALID_PARAMETER;\r
          } else {\r
            Status = LoadEfiDriversFromRomImage (\r
                      File1Buffer,\r
                      SourceSize,\r
                      Node->FullName\r
                     );\r
\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOAD_PCI_ROM_RES), gShellDebug1HiiHandle, Node->FullName, Status);\r
          }\r
          FreePool(File1Buffer);\r
        }\r
      } else if (ShellStatus == SHELL_SUCCESS) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_SPEC), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_NOT_FOUND;\r
      }\r
      if (FileList != NULL && !IsListEmpty(&FileList->Link)) {\r
        Status = ShellCloseFileMetaArg(&FileList);\r
      }\r
      FileList = NULL;\r
\r
      if (Connect) {\r
        Status = LoadPciRomConnectAllDriversToAllControllers ();\r
      }\r
    }\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
/**\r
  Command entry point.\r
\r
  @param[in] RomBar       The Rom Base address.\r
  @param[in] RomSize      The Rom size.\r
  @param[in] FileName     The file name.\r
\r
  @retval EFI_SUCCESS             The command completed successfully.\r
  @retval EFI_INVALID_PARAMETER   Command usage error.\r
  @retval EFI_UNSUPPORTED         Protocols unsupported.\r
  @retval EFI_OUT_OF_RESOURCES    Out of memory.\r
  @retval Other value             Unknown error.\r
**/\r
EFI_STATUS\r
EFIAPI\r
LoadEfiDriversFromRomImage (\r
  VOID                      *RomBar,\r
  UINTN                     RomSize,\r
  CONST CHAR16              *FileName\r
  )\r
\r
{\r
  EFI_PCI_EXPANSION_ROM_HEADER  *EfiRomHeader;\r
  PCI_DATA_STRUCTURE            *Pcir;\r
  UINTN                         ImageIndex;\r
  UINTN                         RomBarOffset;\r
  UINT32                        ImageSize;\r
  UINT16                        ImageOffset;\r
  EFI_HANDLE                    ImageHandle;\r
  EFI_STATUS                    Status;\r
  EFI_STATUS                    ReturnStatus;\r
  CHAR16                        RomFileName[280];\r
  EFI_DEVICE_PATH_PROTOCOL      *FilePath;\r
  BOOLEAN                       SkipImage;\r
  UINT32                        DestinationSize;\r
  UINT32                        ScratchSize;\r
  UINT8                         *Scratch;\r
  VOID                          *ImageBuffer;\r
  VOID                          *DecompressedImageBuffer;\r
  UINT32                        ImageLength;\r
  EFI_DECOMPRESS_PROTOCOL       *Decompress;\r
  UINT32                        InitializationSize;\r
\r
  ImageIndex    = 0;\r
  ReturnStatus     = EFI_NOT_FOUND;\r
  RomBarOffset  = (UINTN) RomBar;\r
\r
  do {\r
\r
    EfiRomHeader = (EFI_PCI_EXPANSION_ROM_HEADER *) (UINTN) RomBarOffset;\r
\r
    if (EfiRomHeader->Signature != PCI_EXPANSION_ROM_HEADER_SIGNATURE) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_CORRUPT), gShellDebug1HiiHandle, FileName, ImageIndex);\r
//      PrintToken (STRING_TOKEN (STR_LOADPCIROM_IMAGE_CORRUPT), HiiHandle, ImageIndex);\r
      return ReturnStatus;\r
    }\r
\r
    //\r
    // If the pointer to the PCI Data Structure is invalid, no further images can be located. \r
    // The PCI Data Structure must be DWORD aligned. \r
    //\r
    if (EfiRomHeader->PcirOffset == 0 ||\r
        (EfiRomHeader->PcirOffset & 3) != 0 ||\r
        RomBarOffset - (UINTN)RomBar + EfiRomHeader->PcirOffset + sizeof (PCI_DATA_STRUCTURE) > RomSize) {\r
      break;\r
    }\r
\r
    Pcir      = (PCI_DATA_STRUCTURE *) (UINTN) (RomBarOffset + EfiRomHeader->PcirOffset);\r
    //\r
    // If a valid signature is not present in the PCI Data Structure, no further images can be located.\r
    //\r
    if (Pcir->Signature != PCI_DATA_STRUCTURE_SIGNATURE) {\r
      break;\r
    }\r
    ImageSize = Pcir->ImageLength * 512;\r
    if (RomBarOffset - (UINTN)RomBar + ImageSize > RomSize) {\r
      break;\r
    }\r
\r
    if ((Pcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) &&\r
        (EfiRomHeader->EfiSignature == EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE) &&\r
        ((EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER) ||\r
         (EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER))) {\r
\r
      ImageOffset             = EfiRomHeader->EfiImageHeaderOffset;\r
      InitializationSize      = EfiRomHeader->InitializationSize * 512;\r
\r
      if (InitializationSize <= ImageSize && ImageOffset < InitializationSize) {\r
\r
        ImageBuffer             = (VOID *) (UINTN) (RomBarOffset + ImageOffset);\r
        ImageLength             = InitializationSize - ImageOffset;\r
        DecompressedImageBuffer = NULL;\r
\r
        //\r
        // decompress here if needed\r
        //\r
        SkipImage = FALSE;\r
        if (EfiRomHeader->CompressionType > EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
          SkipImage = TRUE;\r
        }\r
\r
        if (EfiRomHeader->CompressionType == EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
          Status = gBS->LocateProtocol (&gEfiDecompressProtocolGuid, NULL, (VOID**)&Decompress);\r
          ASSERT_EFI_ERROR(Status);\r
          if (EFI_ERROR (Status)) {\r
            SkipImage = TRUE;\r
          } else {\r
            SkipImage = TRUE;\r
            Status = Decompress->GetInfo (\r
                                  Decompress,\r
                                  ImageBuffer,\r
                                  ImageLength,\r
                                  &DestinationSize,\r
                                  &ScratchSize\r
                                 );\r
            if (!EFI_ERROR (Status)) {\r
              DecompressedImageBuffer = AllocateZeroPool (DestinationSize);\r
              if (ImageBuffer != NULL) {\r
                Scratch = AllocateZeroPool (ScratchSize);\r
                if (Scratch != NULL) {\r
                  Status = Decompress->Decompress (\r
                                        Decompress,\r
                                        ImageBuffer,\r
                                        ImageLength,\r
                                        DecompressedImageBuffer,\r
                                        DestinationSize,\r
                                        Scratch,\r
                                        ScratchSize\r
                                       );\r
                  if (!EFI_ERROR (Status)) {\r
                    ImageBuffer = DecompressedImageBuffer;\r
                    ImageLength = DestinationSize;\r
                    SkipImage   = FALSE;\r
                  }\r
\r
                  FreePool (Scratch);\r
                }\r
              }\r
            }\r
          }\r
        }\r
\r
        if (!SkipImage) {\r
          //\r
          // load image and start image\r
          //\r
          UnicodeSPrint (RomFileName, sizeof (RomFileName), L"%s[%d]", FileName, ImageIndex);\r
          FilePath = FileDevicePath (NULL, RomFileName);\r
\r
          Status = gBS->LoadImage (\r
                        TRUE,\r
                        gImageHandle,\r
                        FilePath,\r
                        ImageBuffer,\r
                        ImageLength,\r
                        &ImageHandle\r
                       );\r
          if (EFI_ERROR (Status)) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_LOAD_FAIL), gShellDebug1HiiHandle, FileName, ImageIndex, Status);\r
//            PrintToken (STRING_TOKEN (STR_LOADPCIROM_LOAD_IMAGE_ERROR), HiiHandle, ImageIndex, Status);\r
          } else {\r
            Status = gBS->StartImage (ImageHandle, NULL, NULL);\r
            if (EFI_ERROR (Status)) {\r
              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_START_FAIL), gShellDebug1HiiHandle, FileName, ImageIndex, Status);\r
//              PrintToken (STRING_TOKEN (STR_LOADPCIROM_START_IMAGE), HiiHandle, ImageIndex, Status);\r
            } else {\r
              ReturnStatus = Status;\r
            }\r
          }\r
        }\r
\r
        if (DecompressedImageBuffer != NULL) {\r
          FreePool (DecompressedImageBuffer);\r
        }\r
\r
      }\r
    }\r
\r
    RomBarOffset = RomBarOffset + ImageSize;\r
    ImageIndex++;\r
  } while (((Pcir->Indicator & 0x80) == 0x00) && ((RomBarOffset - (UINTN) RomBar) < RomSize));\r
\r
  return ReturnStatus;\r
}\r
\r
/**\r
  Connects all available drives and controllers.\r
\r
  @retval EFI_SUCCESS     The operation was successful.\r
  @retval EFI_ABORTED     The abort mechanism was received.\r
**/\r
EFI_STATUS\r
EFIAPI\r
LoadPciRomConnectAllDriversToAllControllers (\r
  VOID\r
  )\r
\r
{\r
  EFI_STATUS  Status;\r
  UINTN       AllHandleCount;\r
  EFI_HANDLE  *AllHandleBuffer;\r
  UINTN       Index;\r
  UINTN       HandleCount;\r
  EFI_HANDLE  *HandleBuffer;\r
  UINTN       *HandleType;\r
  UINTN       HandleIndex;\r
  BOOLEAN     Parent;\r
  BOOLEAN     Device;\r
\r
  Status = gBS->LocateHandleBuffer(\r
            AllHandles,\r
            NULL,\r
            NULL,\r
            &AllHandleCount,\r
            &AllHandleBuffer\r
           );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  for (Index = 0; Index < AllHandleCount; Index++) {\r
    if (ShellGetExecutionBreakFlag ()) {\r
      Status = EFI_ABORTED;\r
      goto Done;\r
    }\r
    //\r
    // Scan the handle database\r
    //\r
    Status = ParseHandleDatabaseByRelationshipWithType(\r
      NULL,\r
      AllHandleBuffer[Index],\r
      &HandleCount,\r
      &HandleBuffer,\r
      &HandleType\r
     );\r
/*\r
    Status = LibScanHandleDatabase (\r
              NULL,\r
              NULL,\r
              AllHandleBuffer[Index],\r
              NULL,\r
              &HandleCount,\r
              &HandleBuffer,\r
              &HandleType\r
             );\r
*/\r
    if (EFI_ERROR (Status)) {\r
      goto Done;\r
    }\r
\r
    Device = TRUE;\r
    if ((HandleType[Index] & HR_DRIVER_BINDING_HANDLE) != 0) {\r
      Device = FALSE;\r
    }\r
\r
    if ((HandleType[Index] & HR_IMAGE_HANDLE) != 0) {\r
      Device = FALSE;\r
    }\r
\r
    if (Device) {\r
      Parent = FALSE;\r
      for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {\r
        if ((HandleType[HandleIndex] & HR_PARENT_HANDLE) != 0) {\r
          Parent = TRUE;\r
        }\r
      }\r
\r
      if (!Parent) {\r
        if ((HandleType[Index] & HR_DEVICE_HANDLE) != 0) {\r
          Status = gBS->ConnectController (\r
                        AllHandleBuffer[Index],\r
                        NULL,\r
                        NULL,\r
                        TRUE\r
                       );\r
        }\r
      }\r
    }\r
\r
    FreePool (HandleBuffer);\r
    FreePool (HandleType);\r
  }\r
\r
Done:\r
  FreePool (AllHandleBuffer);\r
  return Status;\r
}\r
Commit	Line	Data
5d73d92f	1	/** @file\r
	2	Main file for LoadPciRom shell Debug1 function.\r
	3	\r
94020bb4	4	Copyright (c) 2005 - 2012, Intel Corporation. All rights reserved.<BR>\r
5d73d92f	5	This program and the accompanying materials\r
	6	are licensed and made available under the terms and conditions of the BSD License\r
	7	which accompanies this distribution. The full text of the license may be found at\r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UefiShellDebug1CommandsLib.h"\r
	16	#include <IndustryStandard/Pci22.h>\r
	17	#include <IndustryStandard/Pci23.h>\r
	18	#include <IndustryStandard/PeImage.h>\r
	19	#include <Protocol/Decompress.h>\r
	20	\r
3737ac2b	21	/**\r
	22	Connects all available drives and controllers.\r
	23	\r
	24	@retval EFI_SUCCESS The operation was successful.\r
	25	@retval EFI_ABORTED The abort mechanism was received.\r
	26	**/\r
5d73d92f	27	EFI_STATUS\r
	28	EFIAPI\r
	29	LoadPciRomConnectAllDriversToAllControllers (\r
	30	VOID\r
	31	);\r
	32	\r
3737ac2b	33	/**\r
3737ac2b	34	Command entry point.\r
5d73d92f	35	\r
3737ac2b	36	@param[in] RomBar The Rom Base address.\r
	37	@param[in] RomSize The Rom size.\r
	38	@param[in] FileName The file name.\r
	39	\r
	40	@retval EFI_SUCCESS The command completed successfully.\r
	41	@retval EFI_INVALID_PARAMETER Command usage error.\r
	42	@retval EFI_UNSUPPORTED Protocols unsupported.\r
	43	@retval EFI_OUT_OF_RESOURCES Out of memory.\r
	44	@retval Other value Unknown error.\r
	45	**/\r
5d73d92f	46	EFI_STATUS\r
	47	EFIAPI\r
	48	LoadEfiDriversFromRomImage (\r
	49	VOID *RomBar,\r
	50	UINTN RomSize,\r
	51	CONST CHAR16 *FileName\r
	52	);\r
	53	\r
	54	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
	55	{L"-nc", TypeFlag},\r
	56	{NULL, TypeMax}\r
	57	};\r
	58	\r
3737ac2b	59	/**\r
	60	Function for 'loadpcirom' command.\r
	61	\r
	62	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	63	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	64	**/\r
5d73d92f	65	SHELL_STATUS\r
	66	EFIAPI\r
	67	ShellCommandRunLoadPciRom (\r
	68	IN EFI_HANDLE ImageHandle,\r
	69	IN EFI_SYSTEM_TABLE *SystemTable\r
	70	)\r
	71	{\r
	72	EFI_SHELL_FILE_INFO *FileList;\r
	73	UINTN SourceSize;\r
	74	UINT8 *File1Buffer;\r
	75	EFI_STATUS Status;\r
	76	LIST_ENTRY *Package;\r
	77	CHAR16 *ProblemParam;\r
	78	SHELL_STATUS ShellStatus;\r
	79	BOOLEAN Connect;\r
	80	CONST CHAR16 *Param;\r
	81	UINTN ParamCount;\r
	82	EFI_SHELL_FILE_INFO *Node;\r
	83	//\r
	84	// Local variable initializations\r
	85	//\r
	86	File1Buffer = NULL;\r
	87	ShellStatus = SHELL_SUCCESS;\r
	88	FileList = NULL;\r
	89	\r
	90	\r
	91	//\r
	92	// verify number of arguments\r
	93	//\r
	94	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	95	if (EFI_ERROR(Status)) {\r
	96	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
	97	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
	98	FreePool(ProblemParam);\r
	99	ShellStatus = SHELL_INVALID_PARAMETER;\r
	100	} else {\r
	101	ASSERT(FALSE);\r
	102	}\r
	103	} else {\r
3737ac2b	104	if (ShellCommandLineGetCount(Package) < 2) {\r
5d73d92f	105	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
	106	ShellStatus = SHELL_INVALID_PARAMETER;\r
	107	} else {\r
3737ac2b	108	if (ShellCommandLineGetFlag(Package, L"-nc")) {\r
5d73d92f	109	Connect = FALSE;\r
	110	} else {\r
	111	Connect = TRUE;\r
	112	}\r
	113	\r
	114	//\r
	115	// get a list with each file specified by parameters\r
	116	// if parameter is a directory then add all the files below it to the list\r
	117	//\r
	118	for ( ParamCount = 1, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
	119	; Param != NULL\r
	120	; ParamCount++, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
	121	){\r
	122	Status = ShellOpenFileMetaArg((CHAR16*)Param, EFI_FILE_MODE_WRITE\|EFI_FILE_MODE_READ, &FileList);\r
	123	if (EFI_ERROR(Status)) {\r
3737ac2b	124	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, Param, Status);\r
5d73d92f	125	ShellStatus = SHELL_ACCESS_DENIED;\r
	126	break;\r
	127	}\r
	128	}\r
3737ac2b	129	if (ShellStatus == SHELL_SUCCESS && FileList != NULL) {\r
5d73d92f	130	//\r
	131	// loop through the list and make sure we are not aborting...\r
	132	//\r
	133	for ( Node = (EFI_SHELL_FILE_INFO*)GetFirstNode(&FileList->Link)\r
	134	; !IsNull(&FileList->Link, &Node->Link) && !ShellGetExecutionBreakFlag()\r
	135	; Node = (EFI_SHELL_FILE_INFO*)GetNextNode(&FileList->Link, &Node->Link)\r
	136	){\r
	137	if (EFI_ERROR(Node->Status)){\r
	138	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, Node->FullName);\r
	139	ShellStatus = SHELL_INVALID_PARAMETER;\r
	140	continue;\r
	141	}\r
	142	if (FileHandleIsDirectory(Node->Handle) == EFI_SUCCESS) {\r
	143	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_DIR), gShellDebug1HiiHandle, Node->FullName);\r
	144	ShellStatus = SHELL_INVALID_PARAMETER;\r
	145	continue;\r
	146	}\r
	147	SourceSize = (UINTN) Node->Info->FileSize;\r
3737ac2b	148	File1Buffer = AllocateZeroPool (SourceSize);\r
5d73d92f	149	ASSERT(File1Buffer != NULL);\r
	150	Status = gEfiShellProtocol->ReadFile(Node->Handle, &SourceSize, File1Buffer);\r
	151	if (EFI_ERROR(Status)) {\r
	152	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_READ_FAIL), gShellDebug1HiiHandle, Node->FullName);\r
	153	ShellStatus = SHELL_INVALID_PARAMETER;\r
	154	} else {\r
	155	Status = LoadEfiDriversFromRomImage (\r
	156	File1Buffer,\r
	157	SourceSize,\r
	158	Node->FullName\r
	159	);\r
	160	\r
	161	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOAD_PCI_ROM_RES), gShellDebug1HiiHandle, Node->FullName, Status);\r
	162	}\r
	163	FreePool(File1Buffer);\r
	164	}\r
3737ac2b	165	} else if (ShellStatus == SHELL_SUCCESS) {\r
	166	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_NOT_SPEC), gShellDebug1HiiHandle);\r
	167	ShellStatus = SHELL_NOT_FOUND;\r
5d73d92f	168	}\r
	169	if (FileList != NULL && !IsListEmpty(&FileList->Link)) {\r
	170	Status = ShellCloseFileMetaArg(&FileList);\r
	171	}\r
	172	FileList = NULL;\r
	173	\r
	174	if (Connect) {\r
	175	Status = LoadPciRomConnectAllDriversToAllControllers ();\r
	176	}\r
	177	}\r
	178	}\r
	179	\r
	180	return (ShellStatus);\r
	181	}\r
	182	\r
3737ac2b	183	/**\r
	184	Command entry point.\r
	185	\r
	186	@param[in] RomBar The Rom Base address.\r
	187	@param[in] RomSize The Rom size.\r
	188	@param[in] FileName The file name.\r
	189	\r
	190	@retval EFI_SUCCESS The command completed successfully.\r
	191	@retval EFI_INVALID_PARAMETER Command usage error.\r
	192	@retval EFI_UNSUPPORTED Protocols unsupported.\r
	193	@retval EFI_OUT_OF_RESOURCES Out of memory.\r
	194	@retval Other value Unknown error.\r
	195	**/\r
5d73d92f	196	EFI_STATUS\r
3737ac2b	197	EFIAPI\r
5d73d92f	198	LoadEfiDriversFromRomImage (\r
	199	VOID *RomBar,\r
	200	UINTN RomSize,\r
	201	CONST CHAR16 *FileName\r
	202	)\r
5d73d92f	203	\r
5d73d92f	204	{\r
	205	EFI_PCI_EXPANSION_ROM_HEADER *EfiRomHeader;\r
	206	PCI_DATA_STRUCTURE *Pcir;\r
	207	UINTN ImageIndex;\r
	208	UINTN RomBarOffset;\r
	209	UINT32 ImageSize;\r
	210	UINT16 ImageOffset;\r
	211	EFI_HANDLE ImageHandle;\r
	212	EFI_STATUS Status;\r
3737ac2b	213	EFI_STATUS ReturnStatus;\r
5d73d92f	214	CHAR16 RomFileName[280];\r
	215	EFI_DEVICE_PATH_PROTOCOL *FilePath;\r
	216	BOOLEAN SkipImage;\r
	217	UINT32 DestinationSize;\r
	218	UINT32 ScratchSize;\r
	219	UINT8 *Scratch;\r
	220	VOID *ImageBuffer;\r
	221	VOID *DecompressedImageBuffer;\r
	222	UINT32 ImageLength;\r
	223	EFI_DECOMPRESS_PROTOCOL *Decompress;\r
94020bb4	224	UINT32 InitializationSize;\r
5d73d92f	225	\r
5d73d92f	226	ImageIndex = 0;\r
3737ac2b	227	ReturnStatus = EFI_NOT_FOUND;\r
5d73d92f	228	RomBarOffset = (UINTN) RomBar;\r
	229	\r
	230	do {\r
	231	\r
	232	EfiRomHeader = (EFI_PCI_EXPANSION_ROM_HEADER *) (UINTN) RomBarOffset;\r
	233	\r
94020bb4	234	if (EfiRomHeader->Signature != PCI_EXPANSION_ROM_HEADER_SIGNATURE) {\r
5d73d92f	235	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_CORRUPT), gShellDebug1HiiHandle, FileName, ImageIndex);\r
5d73d92f	236	// PrintToken (STRING_TOKEN (STR_LOADPCIROM_IMAGE_CORRUPT), HiiHandle, ImageIndex);\r
3737ac2b	237	return ReturnStatus;\r
5d73d92f	238	}\r
5d73d92f	239	\r
94020bb4	240	//\r
	241	// If the pointer to the PCI Data Structure is invalid, no further images can be located. \r
	242	// The PCI Data Structure must be DWORD aligned. \r
	243	//\r
	244	if (EfiRomHeader->PcirOffset == 0 \|\|\r
	245	(EfiRomHeader->PcirOffset & 3) != 0 \|\|\r
	246	RomBarOffset - (UINTN)RomBar + EfiRomHeader->PcirOffset + sizeof (PCI_DATA_STRUCTURE) > RomSize) {\r
	247	break;\r
	248	}\r
	249	\r
5d73d92f	250	Pcir = (PCI_DATA_STRUCTURE *) (UINTN) (RomBarOffset + EfiRomHeader->PcirOffset);\r
94020bb4	251	//\r
	252	// If a valid signature is not present in the PCI Data Structure, no further images can be located.\r
	253	//\r
	254	if (Pcir->Signature != PCI_DATA_STRUCTURE_SIGNATURE) {\r
	255	break;\r
	256	}\r
5d73d92f	257	ImageSize = Pcir->ImageLength * 512;\r
94020bb4	258	if (RomBarOffset - (UINTN)RomBar + ImageSize > RomSize) {\r
	259	break;\r
	260	}\r
5d73d92f	261	\r
5d73d92f	262	if ((Pcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) &&\r
94020bb4	263	(EfiRomHeader->EfiSignature == EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE) &&\r
	264	((EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER) \|\|\r
	265	(EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER))) {\r
	266	\r
	267	ImageOffset = EfiRomHeader->EfiImageHeaderOffset;\r
	268	InitializationSize = EfiRomHeader->InitializationSize * 512;\r
5d73d92f	269	\r
94020bb4	270	if (InitializationSize <= ImageSize && ImageOffset < InitializationSize) {\r
5d73d92f	271	\r
5d73d92f	272	ImageBuffer = (VOID *) (UINTN) (RomBarOffset + ImageOffset);\r
94020bb4	273	ImageLength = InitializationSize - ImageOffset;\r
5d73d92f	274	DecompressedImageBuffer = NULL;\r
	275	\r
	276	//\r
	277	// decompress here if needed\r
	278	//\r
	279	SkipImage = FALSE;\r
	280	if (EfiRomHeader->CompressionType > EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
	281	SkipImage = TRUE;\r
	282	}\r
	283	\r
	284	if (EfiRomHeader->CompressionType == EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
	285	Status = gBS->LocateProtocol (&gEfiDecompressProtocolGuid, NULL, (VOID**)&Decompress);\r
	286	ASSERT_EFI_ERROR(Status);\r
	287	if (EFI_ERROR (Status)) {\r
	288	SkipImage = TRUE;\r
	289	} else {\r
	290	SkipImage = TRUE;\r
	291	Status = Decompress->GetInfo (\r
	292	Decompress,\r
	293	ImageBuffer,\r
	294	ImageLength,\r
	295	&DestinationSize,\r
	296	&ScratchSize\r
	297	);\r
	298	if (!EFI_ERROR (Status)) {\r
3737ac2b	299	DecompressedImageBuffer = AllocateZeroPool (DestinationSize);\r
5d73d92f	300	if (ImageBuffer != NULL) {\r
3737ac2b	301	Scratch = AllocateZeroPool (ScratchSize);\r
5d73d92f	302	if (Scratch != NULL) {\r
	303	Status = Decompress->Decompress (\r
	304	Decompress,\r
	305	ImageBuffer,\r
	306	ImageLength,\r
	307	DecompressedImageBuffer,\r
	308	DestinationSize,\r
	309	Scratch,\r
	310	ScratchSize\r
	311	);\r
	312	if (!EFI_ERROR (Status)) {\r
	313	ImageBuffer = DecompressedImageBuffer;\r
	314	ImageLength = DestinationSize;\r
	315	SkipImage = FALSE;\r
	316	}\r
	317	\r
	318	FreePool (Scratch);\r
	319	}\r
	320	}\r
	321	}\r
	322	}\r
	323	}\r
	324	\r
	325	if (!SkipImage) {\r
	326	//\r
	327	// load image and start image\r
	328	//\r
	329	UnicodeSPrint (RomFileName, sizeof (RomFileName), L"%s[%d]", FileName, ImageIndex);\r
	330	FilePath = FileDevicePath (NULL, RomFileName);\r
	331	\r
	332	Status = gBS->LoadImage (\r
	333	TRUE,\r
	334	gImageHandle,\r
	335	FilePath,\r
	336	ImageBuffer,\r
	337	ImageLength,\r
	338	&ImageHandle\r
	339	);\r
	340	if (EFI_ERROR (Status)) {\r
	341	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_LOAD_FAIL), gShellDebug1HiiHandle, FileName, ImageIndex, Status);\r
	342	// PrintToken (STRING_TOKEN (STR_LOADPCIROM_LOAD_IMAGE_ERROR), HiiHandle, ImageIndex, Status);\r
	343	} else {\r
	344	Status = gBS->StartImage (ImageHandle, NULL, NULL);\r
	345	if (EFI_ERROR (Status)) {\r
	346	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_LOADPCIROM_START_FAIL), gShellDebug1HiiHandle, FileName, ImageIndex, Status);\r
	347	// PrintToken (STRING_TOKEN (STR_LOADPCIROM_START_IMAGE), HiiHandle, ImageIndex, Status);\r
	348	} else {\r
3737ac2b	349	ReturnStatus = Status;\r
5d73d92f	350	}\r
	351	}\r
	352	}\r
	353	\r
	354	if (DecompressedImageBuffer != NULL) {\r
	355	FreePool (DecompressedImageBuffer);\r
	356	}\r
	357	\r
	358	}\r
	359	}\r
	360	\r
	361	RomBarOffset = RomBarOffset + ImageSize;\r
	362	ImageIndex++;\r
	363	} while (((Pcir->Indicator & 0x80) == 0x00) && ((RomBarOffset - (UINTN) RomBar) < RomSize));\r
	364	\r
3737ac2b	365	return ReturnStatus;\r
5d73d92f	366	}\r
5d73d92f	367	\r
3737ac2b	368	/**\r
	369	Connects all available drives and controllers.\r
	370	\r
	371	@retval EFI_SUCCESS The operation was successful.\r
	372	@retval EFI_ABORTED The abort mechanism was received.\r
	373	**/\r
5d73d92f	374	EFI_STATUS\r
3737ac2b	375	EFIAPI\r
5d73d92f	376	LoadPciRomConnectAllDriversToAllControllers (\r
	377	VOID\r
	378	)\r
	379	\r
	380	{\r
	381	EFI_STATUS Status;\r
	382	UINTN AllHandleCount;\r
	383	EFI_HANDLE *AllHandleBuffer;\r
	384	UINTN Index;\r
	385	UINTN HandleCount;\r
	386	EFI_HANDLE *HandleBuffer;\r
	387	UINTN *HandleType;\r
	388	UINTN HandleIndex;\r
	389	BOOLEAN Parent;\r
	390	BOOLEAN Device;\r
	391	\r
	392	Status = gBS->LocateHandleBuffer(\r
	393	AllHandles,\r
	394	NULL,\r
	395	NULL,\r
	396	&AllHandleCount,\r
	397	&AllHandleBuffer\r
	398	);\r
	399	if (EFI_ERROR (Status)) {\r
	400	return Status;\r
	401	}\r
	402	\r
	403	for (Index = 0; Index < AllHandleCount; Index++) {\r
	404	if (ShellGetExecutionBreakFlag ()) {\r
	405	Status = EFI_ABORTED;\r
	406	goto Done;\r
	407	}\r
	408	//\r
	409	// Scan the handle database\r
	410	//\r
	411	Status = ParseHandleDatabaseByRelationshipWithType(\r
	412	NULL,\r
	413	AllHandleBuffer[Index],\r
	414	&HandleCount,\r
	415	&HandleBuffer,\r
	416	&HandleType\r
	417	);\r
	418	/*\r
	419	Status = LibScanHandleDatabase (\r
	420	NULL,\r
	421	NULL,\r
	422	AllHandleBuffer[Index],\r
	423	NULL,\r
	424	&HandleCount,\r
	425	&HandleBuffer,\r
	426	&HandleType\r
	427	);\r
	428	*/\r
	429	if (EFI_ERROR (Status)) {\r
	430	goto Done;\r
	431	}\r
	432	\r
	433	Device = TRUE;\r
	434	if ((HandleType[Index] & HR_DRIVER_BINDING_HANDLE) != 0) {\r
	435	Device = FALSE;\r
	436	}\r
	437	\r
	438	if ((HandleType[Index] & HR_IMAGE_HANDLE) != 0) {\r
	439	Device = FALSE;\r
440	}\r
441	\r
442	if (Device) {\r
443	Parent = FALSE;\r
444	for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {\r
445	if ((HandleType[HandleIndex] & HR_PARENT_HANDLE) != 0) {\r
446	Parent = TRUE;\r
447	}\r
448	}\r
449	\r
450	if (!Parent) {\r
451	if ((HandleType[Index] & HR_DEVICE_HANDLE) != 0) {\r
452	Status = gBS->ConnectController (\r
453	AllHandleBuffer[Index],\r
454	NULL,\r
455	NULL,\r
456	TRUE\r
457	);\r
458	}\r
459	}\r
460	}\r
461	\r
462	FreePool (HandleBuffer);\r
463	FreePool (HandleType);\r
464	}\r
465	\r
466	Done:\r
467	FreePool (AllHandleBuffer);\r
468	return Status;\r
469	}\r