MdeModulePkg/Bus/Pci/PciBusDxe/PciRomTable.c

   1 /** @file
   2   Set up ROM Table for PCI Bus module.
   3
   4 Copyright (c) 2006 - 2017, Intel Corporation. All rights reserved.<BR>
   5 SPDX-License-Identifier: BSD-2-Clause-Patent
   6
   7 **/
   8
   9 #include "PciBus.h"
  10
  11 //
  12 // PCI ROM image information
  13 //
  14 typedef struct {
  15   EFI_HANDLE  ImageHandle;
  16   UINTN       Seg;
  17   UINT8       Bus;
  18   UINT8       Dev;
  19   UINT8       Func;
  20   VOID        *RomImage;
  21   UINT64      RomSize;
  22 } PCI_ROM_IMAGE;
  23
  24 UINTN          mNumberOfPciRomImages     = 0;
  25 UINTN          mMaxNumberOfPciRomImages  = 0;
  26 PCI_ROM_IMAGE  *mRomImageTable           = NULL;
  27
  28 /**
  29   Add the Rom Image to internal database for later PCI light enumeration.
  30
  31   @param ImageHandle    Option Rom image handle.
  32   @param Seg            Segment of PCI space.
  33   @param Bus            Bus NO of PCI space.
  34   @param Dev            Dev NO of PCI space.
  35   @param Func           Func NO of PCI space.
  36   @param RomImage       Option Rom buffer.
  37   @param RomSize        Size of Option Rom buffer.
  38 **/
  39 VOID
  40 PciRomAddImageMapping (
  41   IN  EFI_HANDLE  ImageHandle,
  42   IN  UINTN       Seg,
  43   IN  UINT8       Bus,
  44   IN  UINT8       Dev,
  45   IN  UINT8       Func,
  46   IN  VOID        *RomImage,
  47   IN  UINT64      RomSize
  48   )
  49 {
  50   UINTN           Index;
  51   PCI_ROM_IMAGE   *NewTable;
  52
  53   for (Index = 0; Index < mNumberOfPciRomImages; Index++) {
  54     if (mRomImageTable[Index].Seg  == Seg &&
  55         mRomImageTable[Index].Bus  == Bus &&
  56         mRomImageTable[Index].Dev  == Dev &&
  57         mRomImageTable[Index].Func == Func) {
  58       //
  59       // Expect once RomImage and RomSize are recorded, they will be passed in
  60       // later when updating ImageHandle
  61       //
  62       ASSERT ((mRomImageTable[Index].RomImage == NULL) || (RomImage == mRomImageTable[Index].RomImage));
  63       ASSERT ((mRomImageTable[Index].RomSize  == 0   ) || (RomSize  == mRomImageTable[Index].RomSize ));
  64       break;
  65     }
  66   }
  67
  68   if (Index == mNumberOfPciRomImages) {
  69     //
  70     // Rom Image Table buffer needs to grow.
  71     //
  72     if (mNumberOfPciRomImages == mMaxNumberOfPciRomImages) {
  73       NewTable = ReallocatePool (
  74                    mMaxNumberOfPciRomImages * sizeof (PCI_ROM_IMAGE),
  75                    (mMaxNumberOfPciRomImages + 0x20) * sizeof (PCI_ROM_IMAGE),
  76                    mRomImageTable
  77                    );
  78       if (NewTable == NULL) {
  79         return ;
  80       }
  81
  82       mRomImageTable            = NewTable;
  83       mMaxNumberOfPciRomImages += 0x20;
  84     }
  85     //
  86     // Record the new PCI device
  87     //
  88     mRomImageTable[Index].Seg  = Seg;
  89     mRomImageTable[Index].Bus  = Bus;
  90     mRomImageTable[Index].Dev  = Dev;
  91     mRomImageTable[Index].Func = Func;
  92     mNumberOfPciRomImages++;
  93   }
  94
  95   mRomImageTable[Index].ImageHandle = ImageHandle;
  96   mRomImageTable[Index].RomImage    = RomImage;
  97   mRomImageTable[Index].RomSize     = RomSize;
  98 }
  99
 100 /**
 101   Get Option rom driver's mapping for PCI device.
 102
 103   @param PciIoDevice Device instance.
 104
 105   @retval TRUE   Found Image mapping.
 106   @retval FALSE  Cannot found image mapping.
 107
 108 **/
 109 BOOLEAN
 110 PciRomGetImageMapping (
 111   IN  PCI_IO_DEVICE                       *PciIoDevice
 112   )
 113 {
 114   EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo;
 115   UINTN                           Index;
 116
 117   PciRootBridgeIo = PciIoDevice->PciRootBridgeIo;
 118
 119   for (Index = 0; Index < mNumberOfPciRomImages; Index++) {
 120     if (mRomImageTable[Index].Seg  == PciRootBridgeIo->SegmentNumber &&
 121         mRomImageTable[Index].Bus  == PciIoDevice->BusNumber         &&
 122         mRomImageTable[Index].Dev  == PciIoDevice->DeviceNumber      &&
 123         mRomImageTable[Index].Func == PciIoDevice->FunctionNumber    ) {
 124
 125       if (mRomImageTable[Index].ImageHandle != NULL) {
 126         AddDriver (PciIoDevice, mRomImageTable[Index].ImageHandle, NULL);
 127       }
 128       PciIoDevice->PciIo.RomImage = mRomImageTable[Index].RomImage;
 129       PciIoDevice->PciIo.RomSize  = mRomImageTable[Index].RomSize;
 130       return TRUE;
 131     }
 132   }
 133
 134   return FALSE;
 135 }