Home | History | Annotate | Download | only in PciBusNoEnumerationDxe 
      1 /*++
      2 
      3 Copyright (c) 2005 - 2007, Intel Corporation. All rights reserved.<BR>
      4 This program and the accompanying materials
      5 are licensed and made available under the terms and conditions of the BSD License
      6 which accompanies this distribution.  The full text of the license may be found at
      7 http://opensource.org/licenses/bsd-license.php
      8 
      9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
     10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
     11 
     12 Module Name:
     13 
     14   PciDriverOverride.c
     15 
     16 Abstract:
     17 
     18   PCI Bus Driver
     19 
     20 Revision History
     21 
     22 --*/
     23 
     24 #include "PciBus.h"
     25 
     26 EFI_STATUS
     27 EFIAPI
     28 GetDriver(
     29   IN     EFI_BUS_SPECIFIC_DRIVER_OVERRIDE_PROTOCOL          *This,
     30   IN OUT EFI_HANDLE                                         *DriverImageHandle
     31   );
     32 
     33 
     34 
     35 EFI_STATUS
     36 InitializePciDriverOverrideInstance (
     37   PCI_IO_DEVICE  *PciIoDevice
     38   )
     39 /*++
     40 
     41 Routine Description:
     42 
     43   Initializes a PCI Driver Override Instance
     44 
     45 Arguments:
     46 
     47 Returns:
     48 
     49   None
     50 
     51 --*/
     52 
     53 {
     54   PciIoDevice->PciDriverOverride.GetDriver = GetDriver;
     55   return EFI_SUCCESS;
     56 }
     57 
     58 EFI_STATUS
     59 EFIAPI
     60 GetDriver (
     61   IN EFI_BUS_SPECIFIC_DRIVER_OVERRIDE_PROTOCOL              *This,
     62   IN OUT EFI_HANDLE                                         *DriverImageHandle
     63   )
     64 /*++
     65 
     66 Routine Description:
     67 
     68   Get a overriding driver image
     69 
     70 Arguments:
     71 
     72 Returns:
     73 
     74   None
     75 
     76 --*/
     77 {
     78   PCI_IO_DEVICE             *PciIoDevice;
     79   LIST_ENTRY            *CurrentLink;
     80   PCI_DRIVER_OVERRIDE_LIST  *Node;
     81 
     82   PciIoDevice = PCI_IO_DEVICE_FROM_PCI_DRIVER_OVERRIDE_THIS (This);
     83 
     84   CurrentLink = PciIoDevice->OptionRomDriverList.ForwardLink;
     85 
     86   while (CurrentLink && CurrentLink != &PciIoDevice->OptionRomDriverList) {
     87 
     88     Node = DRIVER_OVERRIDE_FROM_LINK (CurrentLink);
     89 
     90     if (*DriverImageHandle == NULL) {
     91 
     92       *DriverImageHandle = Node->DriverImageHandle;
     93       return EFI_SUCCESS;
     94     }
     95 
     96     if (*DriverImageHandle == Node->DriverImageHandle) {
     97 
     98       if (CurrentLink->ForwardLink == &PciIoDevice->OptionRomDriverList ||
     99           CurrentLink->ForwardLink == NULL) {
    100         return EFI_NOT_FOUND;
    101       }
    102 
    103       //
    104       // Get next node
    105       //
    106       Node                = DRIVER_OVERRIDE_FROM_LINK (CurrentLink->ForwardLink);
    107       *DriverImageHandle  = Node->DriverImageHandle;
    108       return EFI_SUCCESS;
    109     }
    110 
    111     CurrentLink = CurrentLink->ForwardLink;
    112   }
    113 
    114   return EFI_INVALID_PARAMETER;
    115 }
    116 
    117 EFI_STATUS
    118 AddDriver (
    119   IN PCI_IO_DEVICE     *PciIoDevice,
    120   IN EFI_HANDLE        DriverImageHandle
    121   )
    122 /*++
    123 
    124 Routine Description:
    125 
    126   Add a overriding driver image
    127 
    128 Arguments:
    129 
    130 Returns:
    131 
    132   None
    133 
    134 --*/
    135 
    136 {
    137   EFI_STATUS                    Status;
    138   EFI_LOADED_IMAGE_PROTOCOL     *LoadedImage;
    139   PE_COFF_LOADER_IMAGE_CONTEXT  ImageContext;
    140   PCI_DRIVER_OVERRIDE_LIST      *Node;
    141 
    142   Status = gBS->HandleProtocol (DriverImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **) &LoadedImage);
    143   if (EFI_ERROR (Status)) {
    144     return Status;
    145   }
    146 
    147   Node = AllocatePool (sizeof (PCI_DRIVER_OVERRIDE_LIST));
    148   if (Node == NULL) {
    149     return EFI_OUT_OF_RESOURCES;
    150   }
    151 
    152   Node->Signature         = DRIVER_OVERRIDE_SIGNATURE;
    153   Node->DriverImageHandle = DriverImageHandle;
    154 
    155   InsertTailList (&PciIoDevice->OptionRomDriverList, &(Node->Link));
    156 
    157   PciIoDevice->BusOverride  = TRUE;
    158 
    159 
    160   ImageContext.Handle    = LoadedImage->ImageBase;
    161   ImageContext.ImageRead = PeCoffLoaderImageReadFromMemory;
    162 
    163   //
    164   // Get information about the image
    165   //
    166   Status = PeCoffLoaderGetImageInfo (&ImageContext);
    167   if (EFI_ERROR (Status)) {
    168     return EFI_SUCCESS;
    169   }
    170 
    171   if (ImageContext.Machine != EFI_IMAGE_MACHINE_EBC) {
    172     return EFI_SUCCESS;
    173   }
    174 
    175   return EFI_SUCCESS;
    176 }
    177