tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-07-21 12:44:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update DUET PCI Bus Driver to use PeCoffLib instead of parsing the PE/COFF image itself

Browse Source 
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7439 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
mdkinney 2009-02-05 19:31:45 +00:00
parent 4ab0dff338
commit e8ba34ff3d
3 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
DuetPkg/PciBusNoEnumerationDxe/PciBus.h
View File

@ -50,7 +50,7 @@ Revision History
#include <Library/UefiBootServicesTableLib.h> #include <Library/UefiBootServicesTableLib.h>
#include <Library/DevicePathLib.h> #include <Library/DevicePathLib.h>
#include <Library/PcdLib.h> #include <Library/PcdLib.h>
#include <Library/PeCoffLib.h>
// //
// Driver Produced Protocol Prototypes // Driver Produced Protocol Prototypes

1
DuetPkg/PciBusNoEnumerationDxe/PciBusNoEnumeration.inf
View File

@ -39,6 +39,7 @@
BaseMemoryLib BaseMemoryLib
ReportStatusCodeLib ReportStatusCodeLib
DevicePathLib DevicePathLib
PeCoffLib
[Sources] [Sources]
PciBus.h PciBus.h

18
DuetPkg/PciBusNoEnumerationDxe/PciDriverOverride.c
View File

@ -135,9 +135,8 @@ Returns:
{ {
EFI_STATUS Status; EFI_STATUS Status;
EFI_IMAGE_DOS_HEADER *DosHdr;
EFI_IMAGE_NT_HEADERS *PeHdr;
EFI_LOADED_IMAGE_PROTOCOL *LoadedImage; EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;
PE_COFF_LOADER_IMAGE_CONTEXT ImageContext;
PCI_DRIVER_OVERRIDE_LIST *Node; PCI_DRIVER_OVERRIDE_LIST *Node;
#if (EFI_SPECIFICATION_VERSION < 0x00020000) #if (EFI_SPECIFICATION_VERSION < 0x00020000)
EFI_DRIVER_OS_HANDOFF_HEADER *DriverOsHandoffHeader; EFI_DRIVER_OS_HANDOFF_HEADER *DriverOsHandoffHeader;
@ -167,14 +166,19 @@ Returns:
PciIoDevice->BusOverride = TRUE; PciIoDevice->BusOverride = TRUE;
DosHdr = (EFI_IMAGE_DOS_HEADER *) LoadedImage->ImageBase;
if (DosHdr->e_magic != EFI_IMAGE_DOS_SIGNATURE) { ImageContext.Handle = LoadedImage->ImageBase;
ImageContext.ImageRead = PeCoffLoaderImageReadFromMemory;
//
// Get information about the image
//
Status = PeCoffLoaderGetImageInfo (&ImageContext);
if (EFI_ERROR (Status)) {
return EFI_SUCCESS; return EFI_SUCCESS;
} }
PeHdr = (EFI_IMAGE_NT_HEADERS *) ((UINTN) LoadedImage->ImageBase + DosHdr->e_lfanew); if (ImageContext.Machine != EFI_IMAGE_MACHINE_EBC) {
if (PeHdr->FileHeader.Machine != EFI_IMAGE_MACHINE_EBC) {
return EFI_SUCCESS; return EFI_SUCCESS;
} }