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

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