/** @file\r
PCI Rom supporting funtions implementation for PCI Bus module.\r
\r
-Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2018, 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
return EFI_NOT_FOUND;\r
}\r
\r
- PciIoDevice->RomSize = (UINT64) ((~AllOnes) + 1);\r
+ PciIoDevice->RomSize = (~AllOnes) + 1;\r
return EFI_SUCCESS;\r
}\r
\r
+/**\r
+ Check if the RomImage contains EFI Images.\r
+\r
+ @param RomImage The ROM address of Image for check.\r
+ @param RomSize Size of ROM for check.\r
+\r
+ @retval TRUE ROM contain EFI Image.\r
+ @retval FALSE ROM not contain EFI Image.\r
+\r
+**/\r
+BOOLEAN\r
+ContainEfiImage (\r
+ IN VOID *RomImage,\r
+ IN UINT64 RomSize\r
+ )\r
+{\r
+ PCI_EXPANSION_ROM_HEADER *RomHeader;\r
+ PCI_DATA_STRUCTURE *RomPcir;\r
+ UINT8 Indicator;\r
+\r
+ Indicator = 0;\r
+ RomHeader = RomImage;\r
+ if (RomHeader == NULL) {\r
+ return FALSE;\r
+ }\r
+\r
+ do {\r
+ if (RomHeader->Signature != PCI_EXPANSION_ROM_HEADER_SIGNATURE) {\r
+ RomHeader = (PCI_EXPANSION_ROM_HEADER *) ((UINT8 *) RomHeader + 512);\r
+ continue;\r
+ }\r
+\r
+ //\r
+ // The PCI Data Structure must be DWORD aligned.\r
+ //\r
+ if (RomHeader->PcirOffset == 0 ||\r
+ (RomHeader->PcirOffset & 3) != 0 ||\r
+ (UINT8 *) RomHeader + RomHeader->PcirOffset + sizeof (PCI_DATA_STRUCTURE) > (UINT8 *) RomImage + RomSize) {\r
+ break;\r
+ }\r
+\r
+ RomPcir = (PCI_DATA_STRUCTURE *) ((UINT8 *) RomHeader + RomHeader->PcirOffset);\r
+ if (RomPcir->Signature != PCI_DATA_STRUCTURE_SIGNATURE) {\r
+ break;\r
+ }\r
+\r
+ if (RomPcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) {\r
+ return TRUE;\r
+ }\r
+\r
+ Indicator = RomPcir->Indicator;\r
+ RomHeader = (PCI_EXPANSION_ROM_HEADER *) ((UINT8 *) RomHeader + RomPcir->ImageLength * 512);\r
+ } while (((UINT8 *) RomHeader < (UINT8 *) RomImage + RomSize) && ((Indicator & 0x80) == 0x00));\r
+\r
+ return FALSE;\r
+}\r
+\r
/**\r
Load Option Rom image for specified PCI device.\r
\r
UINT32 LegacyImageLength;\r
UINT8 *RomInMemory;\r
UINT8 CodeType;\r
- BOOLEAN HasEfiOpRom;\r
\r
RomSize = PciDevice->RomSize;\r
\r
RetStatus = EFI_NOT_FOUND;\r
FirstCheck = TRUE;\r
LegacyImageLength = 0;\r
- HasEfiOpRom = FALSE;\r
\r
do {\r
PciDevice->PciRootBridgeIo->Mem.Read (\r
FirstCheck = FALSE;\r
OffsetPcir = RomHeader->PcirOffset;\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
+ // 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 (OffsetPcir == 0 ||\r
(OffsetPcir & 3) != 0 ||\r
if (RomPcir->CodeType == PCI_CODE_TYPE_PCAT_IMAGE) {\r
CodeType = PCI_CODE_TYPE_PCAT_IMAGE;\r
LegacyImageLength = ((UINT32)((EFI_LEGACY_EXPANSION_ROM_HEADER *)RomHeader)->Size512) * 512;\r
- } else if (RomPcir->CodeType == PCI_CODE_TYPE_EFI_IMAGE) {\r
- HasEfiOpRom = TRUE;\r
}\r
Indicator = RomPcir->Indicator;\r
RomImageSize = RomImageSize + RomPcir->ImageLength * 512;\r
\r
RomDecode (PciDevice, RomBarIndex, RomBar, FALSE);\r
\r
- PciDevice->HasEfiOpRom = HasEfiOpRom;\r
PciDevice->EmbeddedRom = TRUE;\r
PciDevice->PciIo.RomSize = RomImageSize;\r
PciDevice->PciIo.RomImage = RomInMemory;\r
PciDevice->BusNumber,\r
PciDevice->DeviceNumber,\r
PciDevice->FunctionNumber,\r
- (UINT64) (UINTN) PciDevice->PciIo.RomImage,\r
+ PciDevice->PciIo.RomImage,\r
PciDevice->PciIo.RomSize\r
);\r
\r
BufferSize,\r
&ImageHandle\r
);\r
-\r
- FreePool (PciOptionRomImageDevicePath);\r
-\r
- if (!EFI_ERROR (Status)) {\r
+ if (EFI_ERROR (Status)) {\r
+ //\r
+ // Record the Option ROM Image device path when LoadImage fails.\r
+ // PciOverride.GetDriver() will try to look for the Image Handle using the device path later.\r
+ //\r
+ AddDriver (PciDevice, NULL, PciOptionRomImageDevicePath);\r
+ } else {\r
Status = gBS->StartImage (ImageHandle, NULL, NULL);\r
if (!EFI_ERROR (Status)) {\r
- AddDriver (PciDevice, ImageHandle);\r
+ //\r
+ // Record the Option ROM Image Handle\r
+ //\r
+ AddDriver (PciDevice, ImageHandle, NULL);\r
PciRomAddImageMapping (\r
ImageHandle,\r
PciDevice->PciRootBridgeIo->SegmentNumber,\r
PciDevice->BusNumber,\r
PciDevice->DeviceNumber,\r
PciDevice->FunctionNumber,\r
- (UINT64) (UINTN) PciDevice->PciIo.RomImage,\r
+ PciDevice->PciIo.RomImage,\r
PciDevice->PciIo.RomSize\r
);\r
RetStatus = EFI_SUCCESS;\r
}\r
}\r
+ FreePool (PciOptionRomImageDevicePath);\r
\r
NextImage:\r
RomBarOffset += ImageSize;\r
\r
- } while (((Indicator & 0x80) == 0x00) && ((UINTN) (RomBarOffset - (UINT8 *) RomBar) < PciDevice->RomSize));\r
+ } while (((Indicator & 0x80) == 0x00) && (((UINTN) RomBarOffset - (UINTN) RomBar) < PciDevice->RomSize));\r
\r
return RetStatus;\r
}\r