PCI: Handle Enhanced Allocation capability for SR-IOV devices

[firefly-linux-kernel-4.4.55.git] / drivers / pci / pci.c

diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
index 8d796c02ee9022f079e71f3270cce1b956480678..364f98b301302e80f80727f57f39fd5cd0e87318 100644 (file)
--- a/drivers/pci/pci.c
+++ b/drivers/pci/pci.c
@@ -2176,6 +2176,12 @@ static struct resource *pci_ea_get_resource(struct pci_dev *dev, u8 bei,
 {
        if (bei <= PCI_EA_BEI_BAR5 && prop <= PCI_EA_P_IO)
                return &dev->resource[bei];
+#ifdef CONFIG_PCI_IOV
+       else if (bei >= PCI_EA_BEI_VF_BAR0 && bei <= PCI_EA_BEI_VF_BAR5 &&
+                (prop == PCI_EA_P_VF_MEM || prop == PCI_EA_P_VF_MEM_PREFETCH))
+               return &dev->resource[PCI_IOV_RESOURCES +
+                                     bei - PCI_EA_BEI_VF_BAR0];
+#endif
        else if (bei == PCI_EA_BEI_ROM)
                return &dev->resource[PCI_ROM_RESOURCE];
        else