[mirror_edk2.git] / EdkModulePkg / Bus / Pci / PciBus / Dxe / PciRomTable.c

/*++\r
\r
Copyright (c) 2006, Intel Corporation                                                         \r
All rights reserved. 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
Module Name:\r
\r
  PciRomTable.c\r
  \r
Abstract:\r
\r
  Option Rom Support for PCI Bus Driver\r
\r
Revision History\r
\r
--*/\r
\r
#include "pcibus.h"\r
#include "PciRomTable.h"\r
\r
typedef struct {\r
  EFI_HANDLE  ImageHandle;\r
  UINTN       Seg;\r
  UINT8       Bus;\r
  UINT8       Dev;\r
  UINT8       Func;\r
  UINT64      RomAddress;\r
  UINT64      RomLength;\r
} EFI_PCI_ROM_IMAGE_MAPPING;\r
\r
static UINTN                      mNumberOfPciRomImages     = 0;\r
static UINTN                      mMaxNumberOfPciRomImages  = 0;\r
static EFI_PCI_ROM_IMAGE_MAPPING  *mRomImageTable           = NULL;\r
\r
VOID\r
PciRomAddImageMapping (\r
  IN EFI_HANDLE  ImageHandle,\r
  IN UINTN       Seg,\r
  IN UINT8       Bus,\r
  IN UINT8       Dev,\r
  IN UINT8       Func,\r
  IN UINT64      RomAddress,\r
  IN UINT64      RomLength\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  ImageHandle - TODO: add argument description\r
  Seg         - TODO: add argument description\r
  Bus         - TODO: add argument description\r
  Dev         - TODO: add argument description\r
  Func        - TODO: add argument description\r
  RomAddress  - TODO: add argument description\r
  RomLength   - TODO: add argument description\r
\r
Returns:\r
\r
  TODO: add return values\r
\r
--*/\r
{\r
  EFI_PCI_ROM_IMAGE_MAPPING *TempMapping;\r
\r
  if (mNumberOfPciRomImages >= mMaxNumberOfPciRomImages) {\r
\r
    mMaxNumberOfPciRomImages += 0x20;\r
\r
    TempMapping = NULL;\r
    TempMapping = AllocatePool (mMaxNumberOfPciRomImages * sizeof (EFI_PCI_ROM_IMAGE_MAPPING));\r
    if (TempMapping == NULL) {\r
      return ;\r
    }\r
\r
    CopyMem (TempMapping, mRomImageTable, mNumberOfPciRomImages * sizeof (EFI_PCI_ROM_IMAGE_MAPPING));\r
\r
    if (mRomImageTable != NULL) {\r
      gBS->FreePool (mRomImageTable);\r
    }\r
\r
    mRomImageTable = TempMapping;\r
  }\r
\r
  mRomImageTable[mNumberOfPciRomImages].ImageHandle = ImageHandle;\r
  mRomImageTable[mNumberOfPciRomImages].Seg         = Seg;\r
  mRomImageTable[mNumberOfPciRomImages].Bus         = Bus;\r
  mRomImageTable[mNumberOfPciRomImages].Dev         = Dev;\r
  mRomImageTable[mNumberOfPciRomImages].Func        = Func;\r
  mRomImageTable[mNumberOfPciRomImages].RomAddress  = RomAddress;\r
  mRomImageTable[mNumberOfPciRomImages].RomLength   = RomLength;\r
  mNumberOfPciRomImages++;\r
}\r
\r
STATIC\r
VOID\r
HexToString (\r
  CHAR16  *String,\r
  UINTN   Value,\r
  UINTN   Digits\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  TODO: Add function description\r
\r
Arguments:\r
\r
  String  - TODO: add argument description\r
  Value   - TODO: add argument description\r
  Digits  - TODO: add argument description\r
\r
Returns:\r
\r
  TODO: add return values\r
\r
--*/\r
{\r
  for (; Digits > 0; Digits--, String++) {\r
    *String = (CHAR16) ((Value >> (4 * (Digits - 1))) & 0x0f);\r
    if (*String > 9) {\r
      (*String) += ('A' - 10);\r
    } else {\r
      (*String) += '0';\r
    }\r
  }\r
}\r
\r
STATIC\r
EFI_STATUS\r
PciRomLoadEfiDriversFromRomImage (\r
  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
  IN EFI_PCI_OPTION_ROM_DESCRIPTOR  *PciOptionRomDescriptor\r
  )\r
/*++\r
\r
Routine Description:\r
\r
Arguments:\r
\r
Returns:\r
\r
--*/\r
// TODO:    This - add argument and description to function comment\r
// TODO:    PciOptionRomDescriptor - add argument and description to function comment\r
{\r
  VOID                          *RomBar;\r
  UINTN                         RomSize;\r
  CHAR16                        *FileName;\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                    retStatus;\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
\r
  RomBar = (VOID *) (UINTN) PciOptionRomDescriptor->RomAddress;\r
  RomSize = (UINTN) PciOptionRomDescriptor->RomLength;\r
  FileName = L"PciRom Seg=00000000 Bus=00 Dev=00 Func=00 Image=0000";\r
\r
  HexToString (&FileName[11], PciOptionRomDescriptor->Seg, 8);\r
  HexToString (&FileName[24], PciOptionRomDescriptor->Bus, 2);\r
  HexToString (&FileName[31], PciOptionRomDescriptor->Dev, 2);\r
  HexToString (&FileName[39], PciOptionRomDescriptor->Func, 2);\r
\r
  ImageIndex    = 0;\r
  retStatus     = 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 != 0xaa55) {\r
      return retStatus;\r
    }\r
\r
    Pcir      = (PCI_DATA_STRUCTURE *) (UINTN) (RomBarOffset + EfiRomHeader->PcirOffset);\r
    ImageSize = Pcir->ImageLength * 512;\r
\r
    if ((Pcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) &&\r
        (EfiRomHeader->EfiSignature == EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE) ) {\r
\r
      if ((EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER) ||\r
          (EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER) ) {\r
\r
        ImageOffset             = EfiRomHeader->EfiImageHeaderOffset;\r
        ImageSize               = EfiRomHeader->InitializationSize * 512;\r
\r
        ImageBuffer             = (VOID *) (UINTN) (RomBarOffset + ImageOffset);\r
        ImageLength             = ImageSize - 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
          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 = NULL;\r
              DecompressedImageBuffer = AllocatePool (DestinationSize);\r
              if (DecompressedImageBuffer != NULL) {\r
                Scratch = AllocatePool (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
                  gBS->FreePool (Scratch);\r
                }\r
              }\r
            }\r
          }\r
        }\r
\r
        if (!SkipImage) {\r
\r
          //\r
          // load image and start image\r
          //\r
\r
          HexToString (&FileName[48], ImageIndex, 4);\r
          FilePath = FileDevicePath (NULL, FileName);\r
\r
          Status = gBS->LoadImage (\r
                          FALSE,\r
                          This->ImageHandle,\r
                          FilePath,\r
                          ImageBuffer,\r
                          ImageLength,\r
                          &ImageHandle\r
                          );\r
          if (!EFI_ERROR (Status)) {\r
            Status = gBS->StartImage (ImageHandle, NULL, NULL);\r
            if (!EFI_ERROR (Status)) {\r
              PciRomAddImageMapping (\r
                ImageHandle,\r
                PciOptionRomDescriptor->Seg,\r
                PciOptionRomDescriptor->Bus,\r
                PciOptionRomDescriptor->Dev,\r
                PciOptionRomDescriptor->Func,\r
                PciOptionRomDescriptor->RomAddress,\r
                PciOptionRomDescriptor->RomLength\r
                );\r
              retStatus = Status;\r
            }\r
          }\r
        }\r
\r
        if (DecompressedImageBuffer != NULL) {\r
          gBS->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 retStatus;\r
}\r
\r
STATIC\r
EFI_STATUS\r
PciRomLoadEfiDriversFromOptionRomTable (\r
  IN EFI_DRIVER_BINDING_PROTOCOL      *This,\r
  IN EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL  *PciRootBridgeIo\r
  )\r
/*++\r
\r
Routine Description:\r
\r
Arguments:\r
\r
Returns:\r
\r
--*/\r
// TODO:    This - add argument and description to function comment\r
// TODO:    PciRootBridgeIo - add argument and description to function comment\r
// TODO:    EFI_NOT_FOUND - add return value to function comment\r
{\r
  EFI_STATUS                        Status;\r
  EFI_PCI_OPTION_ROM_TABLE          *PciOptionRomTable;\r
  EFI_PCI_OPTION_ROM_DESCRIPTOR     *PciOptionRomDescriptor;\r
  UINTN                             Index;\r
  EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *Descriptors;\r
  UINT16                            MinBus;\r
  UINT16                            MaxBus;\r
\r
  Status = EfiGetSystemConfigurationTable (&gEfiPciOptionRomTableGuid, (VOID **) &PciOptionRomTable);\r
  if (EFI_ERROR (Status)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  Status = EFI_NOT_FOUND;\r
\r
  for (Index = 0; Index < PciOptionRomTable->PciOptionRomCount; Index++) {\r
    PciOptionRomDescriptor = &PciOptionRomTable->PciOptionRomDescriptors[Index];\r
    if (!PciOptionRomDescriptor->DontLoadEfiRom) {\r
      if (PciOptionRomDescriptor->Seg == PciRootBridgeIo->SegmentNumber) {\r
        Status = PciRootBridgeIo->Configuration (PciRootBridgeIo, (VOID **) &Descriptors);\r
        if (EFI_ERROR (Status)) {\r
          return Status;\r
        }\r
\r
        PciGetBusRange (&Descriptors, &MinBus, &MaxBus, NULL);\r
        if ((MinBus <= PciOptionRomDescriptor->Bus) && (PciOptionRomDescriptor->Bus <= MaxBus)) {\r
          Status = PciRomLoadEfiDriversFromRomImage (This, PciOptionRomDescriptor);\r
          PciOptionRomDescriptor->DontLoadEfiRom |= 2;\r
        }\r
      }\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
PciRomGetRomResourceFromPciOptionRomTable (\r
  IN EFI_DRIVER_BINDING_PROTOCOL      *This,\r
  IN EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL  *PciRootBridgeIo,\r
  PCI_IO_DEVICE                       *PciIoDevice\r
  )\r
/*++\r
\r
Routine Description:\r
\r
Arguments:\r
\r
Returns:\r
\r
--*/\r
// TODO:    This - add argument and description to function comment\r
// TODO:    PciRootBridgeIo - add argument and description to function comment\r
// TODO:    PciIoDevice - add argument and description to function comment\r
// TODO:    EFI_NOT_FOUND - add return value to function comment\r
// TODO:    EFI_SUCCESS - add return value to function comment\r
{\r
  EFI_STATUS                    Status;\r
  EFI_PCI_OPTION_ROM_TABLE      *PciOptionRomTable;\r
  EFI_PCI_OPTION_ROM_DESCRIPTOR *PciOptionRomDescriptor;\r
  UINTN                         Index;\r
\r
  Status = EfiGetSystemConfigurationTable (&gEfiPciOptionRomTableGuid, (VOID **) &PciOptionRomTable);\r
  if (EFI_ERROR (Status)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  for (Index = 0; Index < PciOptionRomTable->PciOptionRomCount; Index++) {\r
    PciOptionRomDescriptor = &PciOptionRomTable->PciOptionRomDescriptors[Index];\r
    if (PciOptionRomDescriptor->Seg == PciRootBridgeIo->SegmentNumber &&\r
        PciOptionRomDescriptor->Bus == PciIoDevice->BusNumber         &&\r
        PciOptionRomDescriptor->Dev == PciIoDevice->DeviceNumber      &&\r
        PciOptionRomDescriptor->Func == PciIoDevice->FunctionNumber ) {\r
\r
      PciIoDevice->PciIo.RomImage = (VOID *) (UINTN) PciOptionRomDescriptor->RomAddress;\r
      PciIoDevice->PciIo.RomSize  = (UINTN) PciOptionRomDescriptor->RomLength;\r
    }\r
  }\r
\r
  for (Index = 0; Index < mNumberOfPciRomImages; Index++) {\r
    if (mRomImageTable[Index].Seg  == PciRootBridgeIo->SegmentNumber &&\r
        mRomImageTable[Index].Bus  == PciIoDevice->BusNumber         &&\r
        mRomImageTable[Index].Dev  == PciIoDevice->DeviceNumber      &&\r
        mRomImageTable[Index].Func == PciIoDevice->FunctionNumber    ) {\r
\r
      AddDriver (PciIoDevice, mRomImageTable[Index].ImageHandle);\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
PciRomGetImageMapping (\r
  PCI_IO_DEVICE                       *PciIoDevice\r
  )\r
/*++\r
\r
Routine Description:\r
\r
Arguments:\r
\r
Returns:\r
\r
--*/\r
// TODO:    PciIoDevice - add argument and description to function comment\r
// TODO:    EFI_SUCCESS - add return value to function comment\r
{\r
  EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo;\r
  UINTN                           Index;\r
\r
  PciRootBridgeIo = PciIoDevice->PciRootBridgeIo;\r
\r
  for (Index = 0; Index < mNumberOfPciRomImages; Index++) {\r
    if (mRomImageTable[Index].Seg  == PciRootBridgeIo->SegmentNumber &&\r
        mRomImageTable[Index].Bus  == PciIoDevice->BusNumber         &&\r
        mRomImageTable[Index].Dev  == PciIoDevice->DeviceNumber      &&\r
        mRomImageTable[Index].Func == PciIoDevice->FunctionNumber    ) {\r
\r
      if (mRomImageTable[Index].ImageHandle != NULL) {\r
        AddDriver (PciIoDevice, mRomImageTable[Index].ImageHandle);\r
      } else {\r
        PciIoDevice->PciIo.RomImage = (VOID *) (UINTN) mRomImageTable[Index].RomAddress;\r
        PciIoDevice->PciIo.RomSize  = (UINTN) mRomImageTable[Index].RomLength;\r
      }\r
    }\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
Commit	Line	Data
878ddf1f	1	/*++\r
	2	\r
	3	Copyright (c) 2006, Intel Corporation \r
	4	All rights reserved. This program and the accompanying materials \r
	5	are licensed and made available under the terms and conditions of the BSD License \r
	6	which accompanies this distribution. The full text of the license may be found at \r
	7	http://opensource.org/licenses/bsd-license.php \r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r
	11	\r
	12	Module Name:\r
	13	\r
	14	PciRomTable.c\r
	15	\r
	16	Abstract:\r
	17	\r
	18	Option Rom Support for PCI Bus Driver\r
	19	\r
	20	Revision History\r
	21	\r
	22	--*/\r
	23	\r
	24	#include "pcibus.h"\r
	25	#include "PciRomTable.h"\r
	26	\r
	27	typedef struct {\r
	28	EFI_HANDLE ImageHandle;\r
	29	UINTN Seg;\r
	30	UINT8 Bus;\r
	31	UINT8 Dev;\r
	32	UINT8 Func;\r
	33	UINT64 RomAddress;\r
	34	UINT64 RomLength;\r
	35	} EFI_PCI_ROM_IMAGE_MAPPING;\r
	36	\r
	37	static UINTN mNumberOfPciRomImages = 0;\r
	38	static UINTN mMaxNumberOfPciRomImages = 0;\r
	39	static EFI_PCI_ROM_IMAGE_MAPPING *mRomImageTable = NULL;\r
	40	\r
	41	VOID\r
	42	PciRomAddImageMapping (\r
	43	IN EFI_HANDLE ImageHandle,\r
	44	IN UINTN Seg,\r
	45	IN UINT8 Bus,\r
	46	IN UINT8 Dev,\r
	47	IN UINT8 Func,\r
	48	IN UINT64 RomAddress,\r
	49	IN UINT64 RomLength\r
	50	)\r
	51	/*++\r
	52	\r
	53	Routine Description:\r
	54	\r
	55	TODO: Add function description\r
	56	\r
	57	Arguments:\r
	58	\r
	59	ImageHandle - TODO: add argument description\r
	60	Seg - TODO: add argument description\r
	61	Bus - TODO: add argument description\r
	62	Dev - TODO: add argument description\r
	63	Func - TODO: add argument description\r
	64	RomAddress - TODO: add argument description\r
65	RomLength - TODO: add argument description\r
66	\r
67	Returns:\r
68	\r
69	TODO: add return values\r
70	\r
71	--*/\r
72	{\r
73	EFI_PCI_ROM_IMAGE_MAPPING *TempMapping;\r
74	\r
75	if (mNumberOfPciRomImages >= mMaxNumberOfPciRomImages) {\r
76	\r
77	mMaxNumberOfPciRomImages += 0x20;\r
78	\r
79	TempMapping = NULL;\r
80	TempMapping = AllocatePool (mMaxNumberOfPciRomImages * sizeof (EFI_PCI_ROM_IMAGE_MAPPING));\r
81	if (TempMapping == NULL) {\r
82	return ;\r
83	}\r
84	\r
85	CopyMem (TempMapping, mRomImageTable, mNumberOfPciRomImages * sizeof (EFI_PCI_ROM_IMAGE_MAPPING));\r
86	\r
87	if (mRomImageTable != NULL) {\r
88	gBS->FreePool (mRomImageTable);\r
89	}\r
90	\r
91	mRomImageTable = TempMapping;\r
92	}\r
93	\r
94	mRomImageTable[mNumberOfPciRomImages].ImageHandle = ImageHandle;\r
95	mRomImageTable[mNumberOfPciRomImages].Seg = Seg;\r
96	mRomImageTable[mNumberOfPciRomImages].Bus = Bus;\r
97	mRomImageTable[mNumberOfPciRomImages].Dev = Dev;\r
98	mRomImageTable[mNumberOfPciRomImages].Func = Func;\r
99	mRomImageTable[mNumberOfPciRomImages].RomAddress = RomAddress;\r
100	mRomImageTable[mNumberOfPciRomImages].RomLength = RomLength;\r
101	mNumberOfPciRomImages++;\r
102	}\r
103	\r
1cc8ee78	104	STATIC\r
878ddf1f	105	VOID\r
	106	HexToString (\r
	107	CHAR16 *String,\r
	108	UINTN Value,\r
	109	UINTN Digits\r
	110	)\r
	111	/*++\r
	112	\r
	113	Routine Description:\r
	114	\r
	115	TODO: Add function description\r
	116	\r
	117	Arguments:\r
	118	\r
	119	String - TODO: add argument description\r
	120	Value - TODO: add argument description\r
	121	Digits - TODO: add argument description\r
	122	\r
	123	Returns:\r
	124	\r
	125	TODO: add return values\r
	126	\r
	127	--*/\r
	128	{\r
	129	for (; Digits > 0; Digits--, String++) {\r
	130	String = (CHAR16) ((Value >> (4 (Digits - 1))) & 0x0f);\r
	131	if (*String > 9) {\r
	132	(*String) += ('A' - 10);\r
	133	} else {\r
	134	(*String) += '0';\r
	135	}\r
	136	}\r
	137	}\r
1cc8ee78	138	\r
1cc8ee78	139	STATIC\r
878ddf1f	140	EFI_STATUS\r
	141	PciRomLoadEfiDriversFromRomImage (\r
	142	IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
	143	IN EFI_PCI_OPTION_ROM_DESCRIPTOR *PciOptionRomDescriptor\r
	144	)\r
	145	/*++\r
	146	\r
	147	Routine Description:\r
	148	\r
	149	Arguments:\r
	150	\r
	151	Returns:\r
	152	\r
	153	--*/\r
	154	// TODO: This - add argument and description to function comment\r
	155	// TODO: PciOptionRomDescriptor - add argument and description to function comment\r
	156	{\r
	157	VOID *RomBar;\r
	158	UINTN RomSize;\r
	159	CHAR16 *FileName;\r
	160	EFI_PCI_EXPANSION_ROM_HEADER *EfiRomHeader;\r
	161	PCI_DATA_STRUCTURE *Pcir;\r
	162	UINTN ImageIndex;\r
	163	UINTN RomBarOffset;\r
	164	UINT32 ImageSize;\r
	165	UINT16 ImageOffset;\r
	166	EFI_HANDLE ImageHandle;\r
	167	EFI_STATUS Status;\r
	168	EFI_STATUS retStatus;\r
	169	EFI_DEVICE_PATH_PROTOCOL *FilePath;\r
	170	BOOLEAN SkipImage;\r
	171	UINT32 DestinationSize;\r
	172	UINT32 ScratchSize;\r
	173	UINT8 *Scratch;\r
	174	VOID *ImageBuffer;\r
	175	VOID *DecompressedImageBuffer;\r
	176	UINT32 ImageLength;\r
	177	EFI_DECOMPRESS_PROTOCOL *Decompress;\r
	178	\r
	179	RomBar = (VOID *) (UINTN) PciOptionRomDescriptor->RomAddress;\r
	180	RomSize = (UINTN) PciOptionRomDescriptor->RomLength;\r
	181	FileName = L"PciRom Seg=00000000 Bus=00 Dev=00 Func=00 Image=0000";\r
	182	\r
	183	HexToString (&FileName[11], PciOptionRomDescriptor->Seg, 8);\r
	184	HexToString (&FileName[24], PciOptionRomDescriptor->Bus, 2);\r
	185	HexToString (&FileName[31], PciOptionRomDescriptor->Dev, 2);\r
	186	HexToString (&FileName[39], PciOptionRomDescriptor->Func, 2);\r
	187	\r
	188	ImageIndex = 0;\r
	189	retStatus = EFI_NOT_FOUND;\r
	190	RomBarOffset = (UINTN) RomBar;\r
	191	\r
	192	do {\r
	193	\r
	194	EfiRomHeader = (EFI_PCI_EXPANSION_ROM_HEADER *) (UINTN) RomBarOffset;\r
	195	\r
	196	if (EfiRomHeader->Signature != 0xaa55) {\r
	197	return retStatus;\r
	198	}\r
	199	\r
	200	Pcir = (PCI_DATA_STRUCTURE *) (UINTN) (RomBarOffset + EfiRomHeader->PcirOffset);\r
	201	ImageSize = Pcir->ImageLength * 512;\r
	202	\r
	203	if ((Pcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) &&\r
204	(EfiRomHeader->EfiSignature == EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE) ) {\r
205	\r
206	if ((EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER) \|\|\r
207	(EfiRomHeader->EfiSubsystem == EFI_IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER) ) {\r
208	\r
209	ImageOffset = EfiRomHeader->EfiImageHeaderOffset;\r
210	ImageSize = EfiRomHeader->InitializationSize * 512;\r
211	\r
212	ImageBuffer = (VOID *) (UINTN) (RomBarOffset + ImageOffset);\r
213	ImageLength = ImageSize - ImageOffset;\r
214	DecompressedImageBuffer = NULL;\r
215	\r
216	//\r
217	// decompress here if needed\r
218	//\r
219	SkipImage = FALSE;\r
220	if (EfiRomHeader->CompressionType > EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
221	SkipImage = TRUE;\r
222	}\r
223	\r
224	if (EfiRomHeader->CompressionType == EFI_PCI_EXPANSION_ROM_HEADER_COMPRESSED) {\r
225	Status = gBS->LocateProtocol (&gEfiDecompressProtocolGuid, NULL, (VOID **) &Decompress);\r
226	if (EFI_ERROR (Status)) {\r
227	SkipImage = TRUE;\r
228	} else {\r
229	SkipImage = TRUE;\r
230	Status = Decompress->GetInfo (\r
231	Decompress,\r
232	ImageBuffer,\r
233	ImageLength,\r
234	&DestinationSize,\r
235	&ScratchSize\r
236	);\r
237	if (!EFI_ERROR (Status)) {\r
238	DecompressedImageBuffer = NULL;\r
239	DecompressedImageBuffer = AllocatePool (DestinationSize);\r
240	if (DecompressedImageBuffer != NULL) {\r
241	Scratch = AllocatePool (ScratchSize);\r
242	if (Scratch != NULL) {\r
243	Status = Decompress->Decompress (\r
244	Decompress,\r
245	ImageBuffer,\r
246	ImageLength,\r
247	DecompressedImageBuffer,\r
248	DestinationSize,\r
249	Scratch,\r
250	ScratchSize\r
251	);\r
252	if (!EFI_ERROR (Status)) {\r
253	ImageBuffer = DecompressedImageBuffer;\r
254	ImageLength = DestinationSize;\r
255	SkipImage = FALSE;\r
256	}\r
257	\r
258	gBS->FreePool (Scratch);\r
259	}\r
260	}\r
261	}\r
262	}\r
263	}\r
264	\r
265	if (!SkipImage) {\r
266	\r
267	//\r
268	// load image and start image\r
269	//\r
270	\r
271	HexToString (&FileName[48], ImageIndex, 4);\r
272	FilePath = FileDevicePath (NULL, FileName);\r
273	\r
274	Status = gBS->LoadImage (\r
275	FALSE,\r
276	This->ImageHandle,\r
277	FilePath,\r
278	ImageBuffer,\r
279	ImageLength,\r
280	&ImageHandle\r
281	);\r
282	if (!EFI_ERROR (Status)) {\r
283	Status = gBS->StartImage (ImageHandle, NULL, NULL);\r
284	if (!EFI_ERROR (Status)) {\r
285	PciRomAddImageMapping (\r
286	ImageHandle,\r
287	PciOptionRomDescriptor->Seg,\r
288	PciOptionRomDescriptor->Bus,\r
289	PciOptionRomDescriptor->Dev,\r
290	PciOptionRomDescriptor->Func,\r
291	PciOptionRomDescriptor->RomAddress,\r
292	PciOptionRomDescriptor->RomLength\r
293	);\r
294	retStatus = Status;\r
295	}\r
296	}\r
297	}\r
298	\r
299	if (DecompressedImageBuffer != NULL) {\r
300	gBS->FreePool (DecompressedImageBuffer);\r
301	}\r
302	\r
303	}\r
304	}\r
305	\r
306	RomBarOffset = RomBarOffset + ImageSize;\r
307	ImageIndex++;\r
308	} while (((Pcir->Indicator & 0x80) == 0x00) && ((RomBarOffset - (UINTN) RomBar) < RomSize));\r
309	\r
310	return retStatus;\r
311	}\r
312	\r
1cc8ee78	313	STATIC\r
878ddf1f	314	EFI_STATUS\r
	315	PciRomLoadEfiDriversFromOptionRomTable (\r
	316	IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
	317	IN EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo\r
	318	)\r
	319	/*++\r
	320	\r
	321	Routine Description:\r
	322	\r
	323	Arguments:\r
	324	\r
	325	Returns:\r
	326	\r
	327	--*/\r
	328	// TODO: This - add argument and description to function comment\r
	329	// TODO: PciRootBridgeIo - add argument and description to function comment\r
	330	// TODO: EFI_NOT_FOUND - add return value to function comment\r
	331	{\r
	332	EFI_STATUS Status;\r
	333	EFI_PCI_OPTION_ROM_TABLE *PciOptionRomTable;\r
	334	EFI_PCI_OPTION_ROM_DESCRIPTOR *PciOptionRomDescriptor;\r
	335	UINTN Index;\r
	336	EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *Descriptors;\r
	337	UINT16 MinBus;\r
	338	UINT16 MaxBus;\r
	339	\r
	340	Status = EfiGetSystemConfigurationTable (&gEfiPciOptionRomTableGuid, (VOID **) &PciOptionRomTable);\r
	341	if (EFI_ERROR (Status)) {\r
	342	return EFI_NOT_FOUND;\r
	343	}\r
	344	\r
	345	Status = EFI_NOT_FOUND;\r
	346	\r
	347	for (Index = 0; Index < PciOptionRomTable->PciOptionRomCount; Index++) {\r
	348	PciOptionRomDescriptor = &PciOptionRomTable->PciOptionRomDescriptors[Index];\r
	349	if (!PciOptionRomDescriptor->DontLoadEfiRom) {\r
	350	if (PciOptionRomDescriptor->Seg == PciRootBridgeIo->SegmentNumber) {\r
	351	Status = PciRootBridgeIo->Configuration (PciRootBridgeIo, (VOID **) &Descriptors);\r
	352	if (EFI_ERROR (Status)) {\r
	353	return Status;\r
	354	}\r
	355	\r
	356	PciGetBusRange (&Descriptors, &MinBus, &MaxBus, NULL);\r
	357	if ((MinBus <= PciOptionRomDescriptor->Bus) && (PciOptionRomDescriptor->Bus <= MaxBus)) {\r
	358	Status = PciRomLoadEfiDriversFromRomImage (This, PciOptionRomDescriptor);\r
	359	PciOptionRomDescriptor->DontLoadEfiRom \|= 2;\r
	360	}\r
	361	}\r
	362	}\r
	363	}\r
	364	\r
	365	return Status;\r
	366	}\r
	367	\r
	368	EFI_STATUS\r
	369	PciRomGetRomResourceFromPciOptionRomTable (\r
	370	IN EFI_DRIVER_BINDING_PROTOCOL *This,\r
	371	IN EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo,\r
	372	PCI_IO_DEVICE *PciIoDevice\r
	373	)\r
	374	/*++\r
	375	\r
	376	Routine Description:\r
	377	\r
378	Arguments:\r
379	\r
380	Returns:\r
381	\r
382	--*/\r
383	// TODO: This - add argument and description to function comment\r
384	// TODO: PciRootBridgeIo - add argument and description to function comment\r
385	// TODO: PciIoDevice - add argument and description to function comment\r
386	// TODO: EFI_NOT_FOUND - add return value to function comment\r
387	// TODO: EFI_SUCCESS - add return value to function comment\r
388	{\r
389	EFI_STATUS Status;\r
390	EFI_PCI_OPTION_ROM_TABLE *PciOptionRomTable;\r
391	EFI_PCI_OPTION_ROM_DESCRIPTOR *PciOptionRomDescriptor;\r
392	UINTN Index;\r
393	\r
394	Status = EfiGetSystemConfigurationTable (&gEfiPciOptionRomTableGuid, (VOID **) &PciOptionRomTable);\r
395	if (EFI_ERROR (Status)) {\r
396	return EFI_NOT_FOUND;\r
397	}\r
398	\r
399	for (Index = 0; Index < PciOptionRomTable->PciOptionRomCount; Index++) {\r
400	PciOptionRomDescriptor = &PciOptionRomTable->PciOptionRomDescriptors[Index];\r
401	if (PciOptionRomDescriptor->Seg == PciRootBridgeIo->SegmentNumber &&\r
402	PciOptionRomDescriptor->Bus == PciIoDevice->BusNumber &&\r
403	PciOptionRomDescriptor->Dev == PciIoDevice->DeviceNumber &&\r
404	PciOptionRomDescriptor->Func == PciIoDevice->FunctionNumber ) {\r
405	\r
406	PciIoDevice->PciIo.RomImage = (VOID *) (UINTN) PciOptionRomDescriptor->RomAddress;\r
407	PciIoDevice->PciIo.RomSize = (UINTN) PciOptionRomDescriptor->RomLength;\r
408	}\r
409	}\r
410	\r
411	for (Index = 0; Index < mNumberOfPciRomImages; Index++) {\r
412	if (mRomImageTable[Index].Seg == PciRootBridgeIo->SegmentNumber &&\r
413	mRomImageTable[Index].Bus == PciIoDevice->BusNumber &&\r
414	mRomImageTable[Index].Dev == PciIoDevice->DeviceNumber &&\r
415	mRomImageTable[Index].Func == PciIoDevice->FunctionNumber ) {\r
416	\r
417	AddDriver (PciIoDevice, mRomImageTable[Index].ImageHandle);\r
418	}\r
419	}\r
420	\r
421	return EFI_SUCCESS;\r
422	}\r
423	\r
424	EFI_STATUS\r
425	PciRomGetImageMapping (\r
426	PCI_IO_DEVICE *PciIoDevice\r
427	)\r
428	/*++\r
429	\r
430	Routine Description:\r
431	\r
432	Arguments:\r
433	\r
434	Returns:\r
435	\r
436	--*/\r
437	// TODO: PciIoDevice - add argument and description to function comment\r
438	// TODO: EFI_SUCCESS - add return value to function comment\r
439	{\r
440	EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo;\r
441	UINTN Index;\r
442	\r
443	PciRootBridgeIo = PciIoDevice->PciRootBridgeIo;\r
444	\r
445	for (Index = 0; Index < mNumberOfPciRomImages; Index++) {\r
446	if (mRomImageTable[Index].Seg == PciRootBridgeIo->SegmentNumber &&\r
447	mRomImageTable[Index].Bus == PciIoDevice->BusNumber &&\r
448	mRomImageTable[Index].Dev == PciIoDevice->DeviceNumber &&\r
449	mRomImageTable[Index].Func == PciIoDevice->FunctionNumber ) {\r
450	\r
451	if (mRomImageTable[Index].ImageHandle != NULL) {\r
452	AddDriver (PciIoDevice, mRomImageTable[Index].ImageHandle);\r
453	} else {\r
454	PciIoDevice->PciIo.RomImage = (VOID *) (UINTN) mRomImageTable[Index].RomAddress;\r
455	PciIoDevice->PciIo.RomSize = (UINTN) mRomImageTable[Index].RomLength;\r
456	}\r
457	}\r
458	}\r
459	\r
460	return EFI_SUCCESS;\r
461	}\r