Add virtual function number in pci_epc ops. EPC controller driver
can perform virtual function specific initialization based on the
virtual function number.

Signed-off-by: Kishon Vijay Abraham I <kis...@ti.com>
---
 .../pci/controller/cadence/pcie-cadence-ep.c  |  44 +++---
 .../pci/controller/dwc/pcie-designware-ep.c   |  36 ++---
 drivers/pci/controller/pcie-rcar-ep.c         |  19 +--
 drivers/pci/controller/pcie-rockchip-ep.c     |  18 +--
 drivers/pci/endpoint/functions/pci-epf-ntb.c  |  79 ++++++-----
 drivers/pci/endpoint/functions/pci-epf-test.c |  66 +++++----
 drivers/pci/endpoint/pci-epc-core.c           | 128 ++++++++++++------
 drivers/pci/endpoint/pci-epf-core.c           |  48 ++++++-
 include/linux/pci-epc.h                       |  57 ++++----
 9 files changed, 313 insertions(+), 182 deletions(-)

diff --git a/drivers/pci/controller/cadence/pcie-cadence-ep.c 
b/drivers/pci/controller/cadence/pcie-cadence-ep.c
index 897cdde02bd8..912a15be8bfd 100644
--- a/drivers/pci/controller/cadence/pcie-cadence-ep.c
+++ b/drivers/pci/controller/cadence/pcie-cadence-ep.c
@@ -16,7 +16,7 @@
 #define CDNS_PCIE_EP_IRQ_PCI_ADDR_NONE         0x1
 #define CDNS_PCIE_EP_IRQ_PCI_ADDR_LEGACY       0x3
 
-static int cdns_pcie_ep_write_header(struct pci_epc *epc, u8 fn,
+static int cdns_pcie_ep_write_header(struct pci_epc *epc, u8 fn, u8 vfn,
                                     struct pci_epf_header *hdr)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
@@ -47,7 +47,7 @@ static int cdns_pcie_ep_write_header(struct pci_epc *epc, u8 
fn,
        return 0;
 }
 
-static int cdns_pcie_ep_set_bar(struct pci_epc *epc, u8 fn,
+static int cdns_pcie_ep_set_bar(struct pci_epc *epc, u8 fn, u8 vfn,
                                struct pci_epf_bar *epf_bar)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
@@ -117,7 +117,7 @@ static int cdns_pcie_ep_set_bar(struct pci_epc *epc, u8 fn,
        return 0;
 }
 
-static void cdns_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn,
+static void cdns_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn, u8 vfn,
                                   struct pci_epf_bar *epf_bar)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
@@ -147,8 +147,8 @@ static void cdns_pcie_ep_clear_bar(struct pci_epc *epc, u8 
fn,
        epf->epf_bar[bar] = NULL;
 }
 
-static int cdns_pcie_ep_map_addr(struct pci_epc *epc, u8 fn, phys_addr_t addr,
-                                u64 pci_addr, size_t size)
+static int cdns_pcie_ep_map_addr(struct pci_epc *epc, u8 fn, u8 vfn,
+                                phys_addr_t addr, u64 pci_addr, size_t size)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
        struct cdns_pcie *pcie = &ep->pcie;
@@ -169,7 +169,7 @@ static int cdns_pcie_ep_map_addr(struct pci_epc *epc, u8 
fn, phys_addr_t addr,
        return 0;
 }
 
-static void cdns_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn,
+static void cdns_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn, u8 vfn,
                                    phys_addr_t addr)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
@@ -189,7 +189,7 @@ static void cdns_pcie_ep_unmap_addr(struct pci_epc *epc, u8 
fn,
        clear_bit(r, &ep->ob_region_map);
 }
 
-static int cdns_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, u8 mmc)
+static int cdns_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, u8 vfn, u8 mmc)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
        struct cdns_pcie *pcie = &ep->pcie;
@@ -209,7 +209,7 @@ static int cdns_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, 
u8 mmc)
        return 0;
 }
 
-static int cdns_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
+static int cdns_pcie_ep_get_msi(struct pci_epc *epc, u8 fn, u8 vfn)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
        struct cdns_pcie *pcie = &ep->pcie;
@@ -230,7 +230,7 @@ static int cdns_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
        return mme;
 }
 
-static int cdns_pcie_ep_get_msix(struct pci_epc *epc, u8 func_no)
+static int cdns_pcie_ep_get_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
        struct cdns_pcie *pcie = &ep->pcie;
@@ -247,8 +247,9 @@ static int cdns_pcie_ep_get_msix(struct pci_epc *epc, u8 
func_no)
        return val;
 }
 
-static int cdns_pcie_ep_set_msix(struct pci_epc *epc, u8 fn, u16 interrupts,
-                                enum pci_barno bir, u32 offset)
+static int cdns_pcie_ep_set_msix(struct pci_epc *epc, u8 fn, u8 vfn,
+                                u16 interrupts, enum pci_barno bir,
+                                u32 offset)
 {
        struct cdns_pcie_ep *ep = epc_get_drvdata(epc);
        struct cdns_pcie *pcie = &ep->pcie;
@@ -317,7 +318,8 @@ static void cdns_pcie_ep_assert_intx(struct cdns_pcie_ep 
*ep, u8 fn,
        writel(0, ep->irq_cpu_addr + offset);
 }
 
-static int cdns_pcie_ep_send_legacy_irq(struct cdns_pcie_ep *ep, u8 fn, u8 
intx)
+static int cdns_pcie_ep_send_legacy_irq(struct cdns_pcie_ep *ep, u8 fn, u8 vfn,
+                                       u8 intx)
 {
        u16 cmd;
 
@@ -334,7 +336,7 @@ static int cdns_pcie_ep_send_legacy_irq(struct cdns_pcie_ep 
*ep, u8 fn, u8 intx)
        return 0;
 }
 
-static int cdns_pcie_ep_send_msi_irq(struct cdns_pcie_ep *ep, u8 fn,
+static int cdns_pcie_ep_send_msi_irq(struct cdns_pcie_ep *ep, u8 fn, u8 vfn,
                                     u8 interrupt_num)
 {
        struct cdns_pcie *pcie = &ep->pcie;
@@ -382,7 +384,7 @@ static int cdns_pcie_ep_send_msi_irq(struct cdns_pcie_ep 
*ep, u8 fn,
        return 0;
 }
 
-static int cdns_pcie_ep_map_msi_irq(struct pci_epc *epc, u8 fn,
+static int cdns_pcie_ep_map_msi_irq(struct pci_epc *epc, u8 fn, u8 vfn,
                                    phys_addr_t addr, u8 interrupt_num,
                                    u32 entry_size, u32 *msi_data,
                                    u32 *msi_addr_offset)
@@ -419,7 +421,7 @@ static int cdns_pcie_ep_map_msi_irq(struct pci_epc *epc, u8 
fn,
        pci_addr &= GENMASK_ULL(63, 2);
 
        for (i = 0; i < interrupt_num; i++) {
-               ret = cdns_pcie_ep_map_addr(epc, fn, addr,
+               ret = cdns_pcie_ep_map_addr(epc, fn, vfn, addr,
                                            pci_addr & ~pci_addr_mask,
                                            entry_size);
                if (ret)
@@ -433,7 +435,7 @@ static int cdns_pcie_ep_map_msi_irq(struct pci_epc *epc, u8 
fn,
        return 0;
 }
 
-static int cdns_pcie_ep_send_msix_irq(struct cdns_pcie_ep *ep, u8 fn,
+static int cdns_pcie_ep_send_msix_irq(struct cdns_pcie_ep *ep, u8 fn, u8 vfn,
                                      u16 interrupt_num)
 {
        u32 cap = CDNS_PCIE_EP_FUNC_MSIX_CAP_OFFSET;
@@ -478,7 +480,7 @@ static int cdns_pcie_ep_send_msix_irq(struct cdns_pcie_ep 
*ep, u8 fn,
        return 0;
 }
 
-static int cdns_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn,
+static int cdns_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn, u8 vfn,
                                  enum pci_epc_irq_type type,
                                  u16 interrupt_num)
 {
@@ -486,13 +488,13 @@ static int cdns_pcie_ep_raise_irq(struct pci_epc *epc, u8 
fn,
 
        switch (type) {
        case PCI_EPC_IRQ_LEGACY:
-               return cdns_pcie_ep_send_legacy_irq(ep, fn, 0);
+               return cdns_pcie_ep_send_legacy_irq(ep, fn, vfn, 0);
 
        case PCI_EPC_IRQ_MSI:
-               return cdns_pcie_ep_send_msi_irq(ep, fn, interrupt_num);
+               return cdns_pcie_ep_send_msi_irq(ep, fn, vfn, interrupt_num);
 
        case PCI_EPC_IRQ_MSIX:
-               return cdns_pcie_ep_send_msix_irq(ep, fn, interrupt_num);
+               return cdns_pcie_ep_send_msix_irq(ep, fn, vfn, interrupt_num);
 
        default:
                break;
@@ -531,7 +533,7 @@ static const struct pci_epc_features cdns_pcie_epc_features 
= {
 };
 
 static const struct pci_epc_features*
-cdns_pcie_ep_get_features(struct pci_epc *epc, u8 func_no)
+cdns_pcie_ep_get_features(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        return &cdns_pcie_epc_features;
 }
diff --git a/drivers/pci/controller/dwc/pcie-designware-ep.c 
b/drivers/pci/controller/dwc/pcie-designware-ep.c
index 1c25d8337151..0f987b8dfb8a 100644
--- a/drivers/pci/controller/dwc/pcie-designware-ep.c
+++ b/drivers/pci/controller/dwc/pcie-designware-ep.c
@@ -125,7 +125,7 @@ static u8 dw_pcie_ep_find_capability(struct dw_pcie_ep *ep, 
u8 func_no, u8 cap)
        return __dw_pcie_ep_find_next_cap(ep, func_no, next_cap_ptr, cap);
 }
 
-static int dw_pcie_ep_write_header(struct pci_epc *epc, u8 func_no,
+static int dw_pcie_ep_write_header(struct pci_epc *epc, u8 func_no, u8 
vfunc_no,
                                   struct pci_epf_header *hdr)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
@@ -202,7 +202,7 @@ static int dw_pcie_ep_outbound_atu(struct dw_pcie_ep *ep, 
u8 func_no,
        return 0;
 }
 
-static void dw_pcie_ep_clear_bar(struct pci_epc *epc, u8 func_no,
+static void dw_pcie_ep_clear_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                                 struct pci_epf_bar *epf_bar)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
@@ -217,7 +217,7 @@ static void dw_pcie_ep_clear_bar(struct pci_epc *epc, u8 
func_no,
        ep->epf_bar[bar] = NULL;
 }
 
-static int dw_pcie_ep_set_bar(struct pci_epc *epc, u8 func_no,
+static int dw_pcie_ep_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                              struct pci_epf_bar *epf_bar)
 {
        int ret;
@@ -276,7 +276,7 @@ static int dw_pcie_find_index(struct dw_pcie_ep *ep, 
phys_addr_t addr,
        return -EINVAL;
 }
 
-static void dw_pcie_ep_unmap_addr(struct pci_epc *epc, u8 func_no,
+static void dw_pcie_ep_unmap_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                                  phys_addr_t addr)
 {
        int ret;
@@ -292,9 +292,8 @@ static void dw_pcie_ep_unmap_addr(struct pci_epc *epc, u8 
func_no,
        clear_bit(atu_index, ep->ob_window_map);
 }
 
-static int dw_pcie_ep_map_addr(struct pci_epc *epc, u8 func_no,
-                              phys_addr_t addr,
-                              u64 pci_addr, size_t size)
+static int dw_pcie_ep_map_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                              phys_addr_t addr, u64 pci_addr, size_t size)
 {
        int ret;
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
@@ -309,7 +308,7 @@ static int dw_pcie_ep_map_addr(struct pci_epc *epc, u8 
func_no,
        return 0;
 }
 
-static int dw_pcie_ep_get_msi(struct pci_epc *epc, u8 func_no)
+static int dw_pcie_ep_get_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
@@ -333,7 +332,8 @@ static int dw_pcie_ep_get_msi(struct pci_epc *epc, u8 
func_no)
        return val;
 }
 
-static int dw_pcie_ep_set_msi(struct pci_epc *epc, u8 func_no, u8 interrupts)
+static int dw_pcie_ep_set_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                             u8 interrupts)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
@@ -358,7 +358,7 @@ static int dw_pcie_ep_set_msi(struct pci_epc *epc, u8 
func_no, u8 interrupts)
        return 0;
 }
 
-static int dw_pcie_ep_get_msix(struct pci_epc *epc, u8 func_no)
+static int dw_pcie_ep_get_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
@@ -382,8 +382,8 @@ static int dw_pcie_ep_get_msix(struct pci_epc *epc, u8 
func_no)
        return val;
 }
 
-static int dw_pcie_ep_set_msix(struct pci_epc *epc, u8 func_no, u16 interrupts,
-                              enum pci_barno bir, u32 offset)
+static int dw_pcie_ep_set_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                              u16 interrupts, enum pci_barno bir, u32 offset)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
@@ -418,7 +418,7 @@ static int dw_pcie_ep_set_msix(struct pci_epc *epc, u8 
func_no, u16 interrupts,
        return 0;
 }
 
-static int dw_pcie_ep_raise_irq(struct pci_epc *epc, u8 func_no,
+static int dw_pcie_ep_raise_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                                enum pci_epc_irq_type type, u16 interrupt_num)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
@@ -450,7 +450,7 @@ static int dw_pcie_ep_start(struct pci_epc *epc)
 }
 
 static const struct pci_epc_features*
-dw_pcie_ep_get_features(struct pci_epc *epc, u8 func_no)
+dw_pcie_ep_get_features(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
 
@@ -525,14 +525,14 @@ int dw_pcie_ep_raise_msi_irq(struct dw_pcie_ep *ep, u8 
func_no,
        aligned_offset = msg_addr_lower & (epc->mem->window.page_size - 1);
        msg_addr = ((u64)msg_addr_upper) << 32 |
                        (msg_addr_lower & ~aligned_offset);
-       ret = dw_pcie_ep_map_addr(epc, func_no, ep->msi_mem_phys, msg_addr,
+       ret = dw_pcie_ep_map_addr(epc, func_no, 0, ep->msi_mem_phys, msg_addr,
                                  epc->mem->window.page_size);
        if (ret)
                return ret;
 
        writel(msg_data | (interrupt_num - 1), ep->msi_mem + aligned_offset);
 
-       dw_pcie_ep_unmap_addr(epc, func_no, ep->msi_mem_phys);
+       dw_pcie_ep_unmap_addr(epc, func_no, 0, ep->msi_mem_phys);
 
        return 0;
 }
@@ -593,14 +593,14 @@ int dw_pcie_ep_raise_msix_irq(struct dw_pcie_ep *ep, u8 
func_no,
        }
 
        aligned_offset = msg_addr & (epc->mem->window.page_size - 1);
-       ret = dw_pcie_ep_map_addr(epc, func_no, ep->msi_mem_phys,  msg_addr,
+       ret = dw_pcie_ep_map_addr(epc, func_no, 0, ep->msi_mem_phys, msg_addr,
                                  epc->mem->window.page_size);
        if (ret)
                return ret;
 
        writel(msg_data, ep->msi_mem + aligned_offset);
 
-       dw_pcie_ep_unmap_addr(epc, func_no, ep->msi_mem_phys);
+       dw_pcie_ep_unmap_addr(epc, func_no, 0, ep->msi_mem_phys);
 
        return 0;
 }
diff --git a/drivers/pci/controller/pcie-rcar-ep.c 
b/drivers/pci/controller/pcie-rcar-ep.c
index b4a288e24aaf..6cee4e09acca 100644
--- a/drivers/pci/controller/pcie-rcar-ep.c
+++ b/drivers/pci/controller/pcie-rcar-ep.c
@@ -159,7 +159,7 @@ static int rcar_pcie_ep_get_pdata(struct rcar_pcie_endpoint 
*ep,
        return 0;
 }
 
-static int rcar_pcie_ep_write_header(struct pci_epc *epc, u8 fn,
+static int rcar_pcie_ep_write_header(struct pci_epc *epc, u8 fn, u8 vfn,
                                     struct pci_epf_header *hdr)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
@@ -195,7 +195,7 @@ static int rcar_pcie_ep_write_header(struct pci_epc *epc, 
u8 fn,
        return 0;
 }
 
-static int rcar_pcie_ep_set_bar(struct pci_epc *epc, u8 func_no,
+static int rcar_pcie_ep_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                                struct pci_epf_bar *epf_bar)
 {
        int flags = epf_bar->flags | LAR_ENABLE | LAM_64BIT;
@@ -246,7 +246,7 @@ static int rcar_pcie_ep_set_bar(struct pci_epc *epc, u8 
func_no,
        return 0;
 }
 
-static void rcar_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn,
+static void rcar_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn, u8 vfn,
                                   struct pci_epf_bar *epf_bar)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
@@ -259,7 +259,8 @@ static void rcar_pcie_ep_clear_bar(struct pci_epc *epc, u8 
fn,
        clear_bit(atu_index + 1, ep->ib_window_map);
 }
 
-static int rcar_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, u8 interrupts)
+static int rcar_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, u8 vfn,
+                               u8 interrupts)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
        struct rcar_pcie *pcie = &ep->pcie;
@@ -272,7 +273,7 @@ static int rcar_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, 
u8 interrupts)
        return 0;
 }
 
-static int rcar_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
+static int rcar_pcie_ep_get_msi(struct pci_epc *epc, u8 fn, u8 vfn)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
        struct rcar_pcie *pcie = &ep->pcie;
@@ -285,7 +286,7 @@ static int rcar_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
        return ((flags & MSICAP0_MMESE_MASK) >> MSICAP0_MMESE_OFFSET);
 }
 
-static int rcar_pcie_ep_map_addr(struct pci_epc *epc, u8 fn,
+static int rcar_pcie_ep_map_addr(struct pci_epc *epc, u8 fn, u8 vfn,
                                 phys_addr_t addr, u64 pci_addr, size_t size)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
@@ -322,7 +323,7 @@ static int rcar_pcie_ep_map_addr(struct pci_epc *epc, u8 fn,
        return 0;
 }
 
-static void rcar_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn,
+static void rcar_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn, u8 vfn,
                                    phys_addr_t addr)
 {
        struct rcar_pcie_endpoint *ep = epc_get_drvdata(epc);
@@ -403,7 +404,7 @@ static int rcar_pcie_ep_assert_msi(struct rcar_pcie *pcie,
        return 0;
 }
 
-static int rcar_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn,
+static int rcar_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn, u8 vfn,
                                  enum pci_epc_irq_type type,
                                  u16 interrupt_num)
 {
@@ -451,7 +452,7 @@ static const struct pci_epc_features rcar_pcie_epc_features 
= {
 };
 
 static const struct pci_epc_features*
-rcar_pcie_ep_get_features(struct pci_epc *epc, u8 func_no)
+rcar_pcie_ep_get_features(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        return &rcar_pcie_epc_features;
 }
diff --git a/drivers/pci/controller/pcie-rockchip-ep.c 
b/drivers/pci/controller/pcie-rockchip-ep.c
index 7631dc3961c1..5fb9ce6e536e 100644
--- a/drivers/pci/controller/pcie-rockchip-ep.c
+++ b/drivers/pci/controller/pcie-rockchip-ep.c
@@ -122,7 +122,7 @@ static void rockchip_pcie_prog_ep_ob_atu(struct 
rockchip_pcie *rockchip, u8 fn,
                            ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(r));
 }
 
-static int rockchip_pcie_ep_write_header(struct pci_epc *epc, u8 fn,
+static int rockchip_pcie_ep_write_header(struct pci_epc *epc, u8 fn, u8 vfn,
                                         struct pci_epf_header *hdr)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
@@ -159,7 +159,7 @@ static int rockchip_pcie_ep_write_header(struct pci_epc 
*epc, u8 fn,
        return 0;
 }
 
-static int rockchip_pcie_ep_set_bar(struct pci_epc *epc, u8 fn,
+static int rockchip_pcie_ep_set_bar(struct pci_epc *epc, u8 fn, u8 vfn,
                                    struct pci_epf_bar *epf_bar)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
@@ -227,7 +227,7 @@ static int rockchip_pcie_ep_set_bar(struct pci_epc *epc, u8 
fn,
        return 0;
 }
 
-static void rockchip_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn,
+static void rockchip_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn, u8 vfn,
                                       struct pci_epf_bar *epf_bar)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
@@ -256,7 +256,7 @@ static void rockchip_pcie_ep_clear_bar(struct pci_epc *epc, 
u8 fn,
                            ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar));
 }
 
-static int rockchip_pcie_ep_map_addr(struct pci_epc *epc, u8 fn,
+static int rockchip_pcie_ep_map_addr(struct pci_epc *epc, u8 fn, u8 vfn,
                                     phys_addr_t addr, u64 pci_addr,
                                     size_t size)
 {
@@ -284,7 +284,7 @@ static int rockchip_pcie_ep_map_addr(struct pci_epc *epc, 
u8 fn,
        return 0;
 }
 
-static void rockchip_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn,
+static void rockchip_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn, u8 vfn,
                                        phys_addr_t addr)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
@@ -308,7 +308,7 @@ static void rockchip_pcie_ep_unmap_addr(struct pci_epc 
*epc, u8 fn,
        clear_bit(r, &ep->ob_region_map);
 }
 
-static int rockchip_pcie_ep_set_msi(struct pci_epc *epc, u8 fn,
+static int rockchip_pcie_ep_set_msi(struct pci_epc *epc, u8 fn, u8 vfn,
                                    u8 multi_msg_cap)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
@@ -329,7 +329,7 @@ static int rockchip_pcie_ep_set_msi(struct pci_epc *epc, u8 
fn,
        return 0;
 }
 
-static int rockchip_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
+static int rockchip_pcie_ep_get_msi(struct pci_epc *epc, u8 fn, u8 vfn)
 {
        struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
        struct rockchip_pcie *rockchip = &ep->rockchip;
@@ -471,7 +471,7 @@ static int rockchip_pcie_ep_send_msi_irq(struct 
rockchip_pcie_ep *ep, u8 fn,
        return 0;
 }
 
-static int rockchip_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn,
+static int rockchip_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn, u8 vfn,
                                      enum pci_epc_irq_type type,
                                      u16 interrupt_num)
 {
@@ -510,7 +510,7 @@ static const struct pci_epc_features 
rockchip_pcie_epc_features = {
 };
 
 static const struct pci_epc_features*
-rockchip_pcie_ep_get_features(struct pci_epc *epc, u8 func_no)
+rockchip_pcie_ep_get_features(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        return &rockchip_pcie_epc_features;
 }
diff --git a/drivers/pci/endpoint/functions/pci-epf-ntb.c 
b/drivers/pci/endpoint/functions/pci-epf-ntb.c
index 338148cf56f5..73714080a166 100644
--- a/drivers/pci/endpoint/functions/pci-epf-ntb.c
+++ b/drivers/pci/endpoint/functions/pci-epf-ntb.c
@@ -87,6 +87,7 @@ struct epf_ntb {
 
 struct epf_ntb_epc {
        u8 func_no;
+       u8 vfunc_no;
        bool linkup;
        bool is_msix;
        int msix_bar;
@@ -143,14 +144,15 @@ static int epf_ntb_link_up(struct epf_ntb *ntb, bool 
link_up)
        struct epf_ntb_epc *ntb_epc;
        struct epf_ntb_ctrl *ctrl;
        struct pci_epc *epc;
+       u8 func_no, vfunc_no;
        bool is_msix;
-       u8 func_no;
        int ret;
 
        for (type = PRIMARY_INTERFACE; type <= SECONDARY_INTERFACE; type++) {
                ntb_epc = ntb->epc[type];
                epc = ntb_epc->epc;
                func_no = ntb_epc->func_no;
+               vfunc_no = ntb_epc->vfunc_no;
                is_msix = ntb_epc->is_msix;
                ctrl = ntb_epc->reg;
                if (link_up)
@@ -158,7 +160,7 @@ static int epf_ntb_link_up(struct epf_ntb *ntb, bool 
link_up)
                else
                        ctrl->link_status &= ~LINK_STATUS_UP;
                irq_type = is_msix ? PCI_EPC_IRQ_MSIX : PCI_EPC_IRQ_MSI;
-               ret = pci_epc_raise_irq(epc, func_no, irq_type, 1);
+               ret = pci_epc_raise_irq(epc, func_no, vfunc_no, irq_type, 1);
                if (ret) {
                        dev_err(&epc->dev,
                                "%s intf: Failed to raise Link Up IRQ\n",
@@ -238,10 +240,10 @@ static int epf_ntb_configure_mw(struct epf_ntb *ntb,
        enum pci_barno peer_barno;
        struct epf_ntb_ctrl *ctrl;
        phys_addr_t phys_addr;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        u64 addr, size;
        int ret = 0;
-       u8 func_no;
 
        ntb_epc = ntb->epc[type];
        epc = ntb_epc->epc;
@@ -267,8 +269,9 @@ static int epf_ntb_configure_mw(struct epf_ntb *ntb,
        }
 
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
-       ret = pci_epc_map_addr(epc, func_no, phys_addr, addr, size);
+       ret = pci_epc_map_addr(epc, func_no, vfunc_no, phys_addr, addr, size);
        if (ret)
                dev_err(&epc->dev,
                        "%s intf: Failed to map memory window %d address\n",
@@ -296,8 +299,8 @@ static void epf_ntb_teardown_mw(struct epf_ntb *ntb,
        enum pci_barno peer_barno;
        struct epf_ntb_ctrl *ctrl;
        phys_addr_t phys_addr;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
 
        ntb_epc = ntb->epc[type];
        epc = ntb_epc->epc;
@@ -311,8 +314,9 @@ static void epf_ntb_teardown_mw(struct epf_ntb *ntb,
        if (mw + NTB_MW_OFFSET == BAR_DB_MW1)
                phys_addr += ctrl->mw1_offset;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
-       pci_epc_unmap_addr(epc, func_no, phys_addr);
+       pci_epc_unmap_addr(epc, func_no, vfunc_no, phys_addr);
 }
 
 /**
@@ -385,8 +389,8 @@ static int epf_ntb_configure_msi(struct epf_ntb *ntb,
        struct epf_ntb_ctrl *peer_ctrl;
        enum pci_barno peer_barno;
        phys_addr_t phys_addr;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
        int ret, i;
 
        ntb_epc = ntb->epc[type];
@@ -400,8 +404,9 @@ static int epf_ntb_configure_msi(struct epf_ntb *ntb,
 
        phys_addr = peer_epf_bar->phys_addr;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
-       ret = pci_epc_map_msi_irq(epc, func_no, phys_addr, db_count,
+       ret = pci_epc_map_msi_irq(epc, func_no, vfunc_no, phys_addr, db_count,
                                  db_entry_size, &db_data, &db_offset);
        if (ret) {
                dev_err(&epc->dev, "%s intf: Failed to map MSI IRQ\n",
@@ -491,10 +496,10 @@ static int epf_ntb_configure_msix(struct epf_ntb *ntb,
        u32 db_entry_size, msg_data;
        enum pci_barno peer_barno;
        phys_addr_t phys_addr;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        size_t align;
        u64 msg_addr;
-       u8 func_no;
        int ret, i;
 
        ntb_epc = ntb->epc[type];
@@ -512,12 +517,13 @@ static int epf_ntb_configure_msix(struct epf_ntb *ntb,
        align = epc_features->align;
 
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        db_entry_size = peer_ctrl->db_entry_size;
 
        for (i = 0; i < db_count; i++) {
                msg_addr = ALIGN_DOWN(msix_tbl[i].msg_addr, align);
                msg_data = msix_tbl[i].msg_data;
-               ret = pci_epc_map_addr(epc, func_no, phys_addr, msg_addr,
+               ret = pci_epc_map_addr(epc, func_no, vfunc_no, phys_addr, 
msg_addr,
                                       db_entry_size);
                if (ret) {
                        dev_err(&epc->dev,
@@ -586,8 +592,8 @@ epf_ntb_teardown_db(struct epf_ntb *ntb, enum 
pci_epc_interface_type type)
        struct pci_epf_bar *peer_epf_bar;
        enum pci_barno peer_barno;
        phys_addr_t phys_addr;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
 
        ntb_epc = ntb->epc[type];
        epc = ntb_epc->epc;
@@ -597,8 +603,9 @@ epf_ntb_teardown_db(struct epf_ntb *ntb, enum 
pci_epc_interface_type type)
        peer_epf_bar = &peer_ntb_epc->epf_bar[peer_barno];
        phys_addr = peer_epf_bar->phys_addr;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
-       pci_epc_unmap_addr(epc, func_no, phys_addr);
+       pci_epc_unmap_addr(epc, func_no, vfunc_no, phys_addr);
 }
 
 /**
@@ -727,14 +734,15 @@ static void epf_ntb_peer_spad_bar_clear(struct 
epf_ntb_epc *ntb_epc)
 {
        struct pci_epf_bar *epf_bar;
        enum pci_barno barno;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
 
        epc = ntb_epc->epc;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        barno = ntb_epc->epf_ntb_bar[BAR_PEER_SPAD];
        epf_bar = &ntb_epc->epf_bar[barno];
-       pci_epc_clear_bar(epc, func_no, epf_bar);
+       pci_epc_clear_bar(epc, func_no, vfunc_no, epf_bar);
 }
 
 /**
@@ -773,9 +781,9 @@ static int epf_ntb_peer_spad_bar_set(struct epf_ntb *ntb,
        struct pci_epf_bar *peer_epf_bar, *epf_bar;
        enum pci_barno peer_barno, barno;
        u32 peer_spad_offset;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        struct device *dev;
-       u8 func_no;
        int ret;
 
        dev = &ntb->epf->dev;
@@ -788,6 +796,7 @@ static int epf_ntb_peer_spad_bar_set(struct epf_ntb *ntb,
        barno = ntb_epc->epf_ntb_bar[BAR_PEER_SPAD];
        epf_bar = &ntb_epc->epf_bar[barno];
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        epc = ntb_epc->epc;
 
        peer_spad_offset = peer_ntb_epc->reg->spad_offset;
@@ -796,7 +805,7 @@ static int epf_ntb_peer_spad_bar_set(struct epf_ntb *ntb,
        epf_bar->barno = barno;
        epf_bar->flags = PCI_BASE_ADDRESS_MEM_TYPE_32;
 
-       ret = pci_epc_set_bar(epc, func_no, epf_bar);
+       ret = pci_epc_set_bar(epc, func_no, vfunc_no, epf_bar);
        if (ret) {
                dev_err(dev, "%s intf: peer SPAD BAR set failed\n",
                        pci_epc_interface_string(type));
@@ -839,14 +848,15 @@ static void epf_ntb_config_sspad_bar_clear(struct 
epf_ntb_epc *ntb_epc)
 {
        struct pci_epf_bar *epf_bar;
        enum pci_barno barno;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
 
        epc = ntb_epc->epc;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        barno = ntb_epc->epf_ntb_bar[BAR_CONFIG];
        epf_bar = &ntb_epc->epf_bar[barno];
-       pci_epc_clear_bar(epc, func_no, epf_bar);
+       pci_epc_clear_bar(epc, func_no, vfunc_no, epf_bar);
 }
 
 /**
@@ -882,10 +892,10 @@ static int epf_ntb_config_sspad_bar_set(struct 
epf_ntb_epc *ntb_epc)
 {
        struct pci_epf_bar *epf_bar;
        enum pci_barno barno;
+       u8 func_no, vfunc_no;
        struct epf_ntb *ntb;
        struct pci_epc *epc;
        struct device *dev;
-       u8 func_no;
        int ret;
 
        ntb = ntb_epc->epf_ntb;
@@ -893,10 +903,11 @@ static int epf_ntb_config_sspad_bar_set(struct 
epf_ntb_epc *ntb_epc)
 
        epc = ntb_epc->epc;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        barno = ntb_epc->epf_ntb_bar[BAR_CONFIG];
        epf_bar = &ntb_epc->epf_bar[barno];
 
-       ret = pci_epc_set_bar(epc, func_no, epf_bar);
+       ret = pci_epc_set_bar(epc, func_no, vfunc_no, epf_bar);
        if (ret) {
                dev_err(dev, "%s inft: Config/Status/SPAD BAR set failed\n",
                        pci_epc_interface_string(ntb_epc->type));
@@ -1210,17 +1221,18 @@ static void epf_ntb_db_mw_bar_clear(struct epf_ntb_epc 
*ntb_epc)
        struct pci_epf_bar *epf_bar;
        enum epf_ntb_bar bar;
        enum pci_barno barno;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
-       u8 func_no;
 
        epc = ntb_epc->epc;
 
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
        for (bar = BAR_DB_MW1; bar < BAR_MW4; bar++) {
                barno = ntb_epc->epf_ntb_bar[bar];
                epf_bar = &ntb_epc->epf_bar[barno];
-               pci_epc_clear_bar(epc, func_no, epf_bar);
+               pci_epc_clear_bar(epc, func_no, vfunc_no, epf_bar);
        }
 }
 
@@ -1259,10 +1271,10 @@ static int epf_ntb_configure_interrupt(struct epf_ntb 
*ntb,
        const struct pci_epc_features *epc_features;
        bool msix_capable, msi_capable;
        struct epf_ntb_epc *ntb_epc;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        struct device *dev;
        u32 db_count;
-       u8 func_no;
        int ret;
 
        ntb_epc = ntb->epc[type];
@@ -1278,6 +1290,7 @@ static int epf_ntb_configure_interrupt(struct epf_ntb 
*ntb,
        }
 
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
        db_count = ntb->db_count;
        if (db_count > MAX_DB_COUNT) {
@@ -1289,7 +1302,7 @@ static int epf_ntb_configure_interrupt(struct epf_ntb 
*ntb,
        epc = ntb_epc->epc;
 
        if (msi_capable) {
-               ret = pci_epc_set_msi(epc, func_no, db_count);
+               ret = pci_epc_set_msi(epc, func_no, vfunc_no, db_count);
                if (ret) {
                        dev_err(dev, "%s intf: MSI configuration failed\n",
                                pci_epc_interface_string(type));
@@ -1298,7 +1311,7 @@ static int epf_ntb_configure_interrupt(struct epf_ntb 
*ntb,
        }
 
        if (msix_capable) {
-               ret = pci_epc_set_msix(epc, func_no, db_count,
+               ret = pci_epc_set_msix(epc, func_no, vfunc_no, db_count,
                                       ntb_epc->msix_bar,
                                       ntb_epc->msix_table_offset);
                if (ret) {
@@ -1418,11 +1431,11 @@ static int epf_ntb_db_mw_bar_init(struct epf_ntb *ntb,
        u32 num_mws, db_count;
        enum epf_ntb_bar bar;
        enum pci_barno barno;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        struct device *dev;
        size_t align;
        int ret, i;
-       u8 func_no;
        u64 size;
 
        ntb_epc = ntb->epc[type];
@@ -1432,6 +1445,7 @@ static int epf_ntb_db_mw_bar_init(struct epf_ntb *ntb,
        epc_features = ntb_epc->epc_features;
        align = epc_features->align;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
        epc = ntb_epc->epc;
        num_mws = ntb->num_mws;
        db_count = ntb->db_count;
@@ -1459,7 +1473,7 @@ static int epf_ntb_db_mw_bar_init(struct epf_ntb *ntb,
                barno = ntb_epc->epf_ntb_bar[bar];
                epf_bar = &ntb_epc->epf_bar[barno];
 
-               ret = pci_epc_set_bar(epc, func_no, epf_bar);
+               ret = pci_epc_set_bar(epc, func_no, vfunc_no, epf_bar);
                if (ret) {
                        dev_err(dev, "%s intf: DoorBell BAR set failed\n",
                                pci_epc_interface_string(type));
@@ -1531,9 +1545,9 @@ static int epf_ntb_epc_create_interface(struct epf_ntb 
*ntb,
        const struct pci_epc_features *epc_features;
        struct pci_epf_bar *epf_bar;
        struct epf_ntb_epc *ntb_epc;
+       u8 func_no, vfunc_no;
        struct pci_epf *epf;
        struct device *dev;
-       u8 func_no;
 
        dev = &ntb->epf->dev;
 
@@ -1542,6 +1556,7 @@ static int epf_ntb_epc_create_interface(struct epf_ntb 
*ntb,
                return -ENOMEM;
 
        epf = ntb->epf;
+       vfunc_no = epf->vfunc_no;
        if (type == PRIMARY_INTERFACE) {
                func_no = epf->func_no;
                epf_bar = epf->bar;
@@ -1553,11 +1568,12 @@ static int epf_ntb_epc_create_interface(struct epf_ntb 
*ntb,
        ntb_epc->linkup = false;
        ntb_epc->epc = epc;
        ntb_epc->func_no = func_no;
+       ntb_epc->vfunc_no = vfunc_no;
        ntb_epc->type = type;
        ntb_epc->epf_bar = epf_bar;
        ntb_epc->epf_ntb = ntb;
 
-       epc_features = pci_epc_get_features(epc, func_no);
+       epc_features = pci_epc_get_features(epc, func_no, vfunc_no);
        if (!epc_features)
                return -EINVAL;
        ntb_epc->epc_features = epc_features;
@@ -1698,10 +1714,10 @@ static int epf_ntb_epc_init_interface(struct epf_ntb 
*ntb,
                                      enum pci_epc_interface_type type)
 {
        struct epf_ntb_epc *ntb_epc;
+       u8 func_no, vfunc_no;
        struct pci_epc *epc;
        struct pci_epf *epf;
        struct device *dev;
-       u8 func_no;
        int ret;
 
        ntb_epc = ntb->epc[type];
@@ -1709,6 +1725,7 @@ static int epf_ntb_epc_init_interface(struct epf_ntb *ntb,
        dev = &epf->dev;
        epc = ntb_epc->epc;
        func_no = ntb_epc->func_no;
+       vfunc_no = ntb_epc->vfunc_no;
 
        ret = epf_ntb_config_sspad_bar_set(ntb->epc[type]);
        if (ret) {
@@ -1738,7 +1755,7 @@ static int epf_ntb_epc_init_interface(struct epf_ntb *ntb,
                goto err_db_mw_bar_init;
        }
 
-       ret = pci_epc_write_header(epc, func_no, epf->header);
+       ret = pci_epc_write_header(epc, func_no, vfunc_no, epf->header);
        if (ret) {
                dev_err(dev, "%s intf: Configuration header write failed\n",
                        pci_epc_interface_string(type));
diff --git a/drivers/pci/endpoint/functions/pci-epf-test.c 
b/drivers/pci/endpoint/functions/pci-epf-test.c
index c0ac4e9cbe72..b588c94d97e9 100644
--- a/drivers/pci/endpoint/functions/pci-epf-test.c
+++ b/drivers/pci/endpoint/functions/pci-epf-test.c
@@ -247,8 +247,8 @@ static int pci_epf_test_copy(struct pci_epf_test *epf_test)
                goto err;
        }
 
-       ret = pci_epc_map_addr(epc, epf->func_no, src_phys_addr, reg->src_addr,
-                              reg->size);
+       ret = pci_epc_map_addr(epc, epf->func_no, epf->vfunc_no, src_phys_addr,
+                              reg->src_addr, reg->size);
        if (ret) {
                dev_err(dev, "Failed to map source address\n");
                reg->status = STATUS_SRC_ADDR_INVALID;
@@ -263,8 +263,8 @@ static int pci_epf_test_copy(struct pci_epf_test *epf_test)
                goto err_src_map_addr;
        }
 
-       ret = pci_epc_map_addr(epc, epf->func_no, dst_phys_addr, reg->dst_addr,
-                              reg->size);
+       ret = pci_epc_map_addr(epc, epf->func_no, epf->vfunc_no, dst_phys_addr,
+                              reg->dst_addr, reg->size);
        if (ret) {
                dev_err(dev, "Failed to map destination address\n");
                reg->status = STATUS_DST_ADDR_INVALID;
@@ -291,13 +291,13 @@ static int pci_epf_test_copy(struct pci_epf_test 
*epf_test)
        pci_epf_test_print_rate("COPY", reg->size, &start, &end, use_dma);
 
 err_map_addr:
-       pci_epc_unmap_addr(epc, epf->func_no, dst_phys_addr);
+       pci_epc_unmap_addr(epc, epf->func_no, epf->vfunc_no, dst_phys_addr);
 
 err_dst_addr:
        pci_epc_mem_free_addr(epc, dst_phys_addr, dst_addr, reg->size);
 
 err_src_map_addr:
-       pci_epc_unmap_addr(epc, epf->func_no, src_phys_addr);
+       pci_epc_unmap_addr(epc, epf->func_no, epf->vfunc_no, src_phys_addr);
 
 err_src_addr:
        pci_epc_mem_free_addr(epc, src_phys_addr, src_addr, reg->size);
@@ -331,8 +331,8 @@ static int pci_epf_test_read(struct pci_epf_test *epf_test)
                goto err;
        }
 
-       ret = pci_epc_map_addr(epc, epf->func_no, phys_addr, reg->src_addr,
-                              reg->size);
+       ret = pci_epc_map_addr(epc, epf->func_no, epf->vfunc_no, phys_addr,
+                              reg->src_addr, reg->size);
        if (ret) {
                dev_err(dev, "Failed to map address\n");
                reg->status = STATUS_SRC_ADDR_INVALID;
@@ -386,7 +386,7 @@ static int pci_epf_test_read(struct pci_epf_test *epf_test)
        kfree(buf);
 
 err_map_addr:
-       pci_epc_unmap_addr(epc, epf->func_no, phys_addr);
+       pci_epc_unmap_addr(epc, epf->func_no, epf->vfunc_no, phys_addr);
 
 err_addr:
        pci_epc_mem_free_addr(epc, phys_addr, src_addr, reg->size);
@@ -419,8 +419,8 @@ static int pci_epf_test_write(struct pci_epf_test *epf_test)
                goto err;
        }
 
-       ret = pci_epc_map_addr(epc, epf->func_no, phys_addr, reg->dst_addr,
-                              reg->size);
+       ret = pci_epc_map_addr(epc, epf->func_no, epf->vfunc_no, phys_addr,
+                              reg->dst_addr, reg->size);
        if (ret) {
                dev_err(dev, "Failed to map address\n");
                reg->status = STATUS_DST_ADDR_INVALID;
@@ -479,7 +479,7 @@ static int pci_epf_test_write(struct pci_epf_test *epf_test)
        kfree(buf);
 
 err_map_addr:
-       pci_epc_unmap_addr(epc, epf->func_no, phys_addr);
+       pci_epc_unmap_addr(epc, epf->func_no, epf->vfunc_no, phys_addr);
 
 err_addr:
        pci_epc_mem_free_addr(epc, phys_addr, dst_addr, reg->size);
@@ -501,13 +501,16 @@ static void pci_epf_test_raise_irq(struct pci_epf_test 
*epf_test, u8 irq_type,
 
        switch (irq_type) {
        case IRQ_TYPE_LEGACY:
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_LEGACY, 0);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_LEGACY, 0);
                break;
        case IRQ_TYPE_MSI:
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_MSI, irq);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_MSI, irq);
                break;
        case IRQ_TYPE_MSIX:
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_MSIX, irq);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_MSIX, irq);
                break;
        default:
                dev_err(dev, "Failed to raise IRQ, unknown type\n");
@@ -542,7 +545,8 @@ static void pci_epf_test_cmd_handler(struct work_struct 
*work)
 
        if (command & COMMAND_RAISE_LEGACY_IRQ) {
                reg->status = STATUS_IRQ_RAISED;
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_LEGACY, 0);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_LEGACY, 0);
                goto reset_handler;
        }
 
@@ -580,22 +584,22 @@ static void pci_epf_test_cmd_handler(struct work_struct 
*work)
        }
 
        if (command & COMMAND_RAISE_MSI_IRQ) {
-               count = pci_epc_get_msi(epc, epf->func_no);
+               count = pci_epc_get_msi(epc, epf->func_no, epf->vfunc_no);
                if (reg->irq_number > count || count <= 0)
                        goto reset_handler;
                reg->status = STATUS_IRQ_RAISED;
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_MSI,
-                                 reg->irq_number);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_MSI, reg->irq_number);
                goto reset_handler;
        }
 
        if (command & COMMAND_RAISE_MSIX_IRQ) {
-               count = pci_epc_get_msix(epc, epf->func_no);
+               count = pci_epc_get_msix(epc, epf->func_no, epf->vfunc_no);
                if (reg->irq_number > count || count <= 0)
                        goto reset_handler;
                reg->status = STATUS_IRQ_RAISED;
-               pci_epc_raise_irq(epc, epf->func_no, PCI_EPC_IRQ_MSIX,
-                                 reg->irq_number);
+               pci_epc_raise_irq(epc, epf->func_no, epf->vfunc_no,
+                                 PCI_EPC_IRQ_MSIX, reg->irq_number);
                goto reset_handler;
        }
 
@@ -618,7 +622,8 @@ static void pci_epf_test_unbind(struct pci_epf *epf)
                epf_bar = &epf->bar[bar];
 
                if (epf_test->reg[bar]) {
-                       pci_epc_clear_bar(epc, epf->func_no, epf_bar);
+                       pci_epc_clear_bar(epc, epf->func_no, epf->vfunc_no,
+                                         epf_bar);
                        pci_epf_free_space(epf, epf_test->reg[bar], bar,
                                           PRIMARY_INTERFACE);
                }
@@ -650,7 +655,8 @@ static int pci_epf_test_set_bar(struct pci_epf *epf)
                if (!!(epc_features->reserved_bar & (1 << bar)))
                        continue;
 
-               ret = pci_epc_set_bar(epc, epf->func_no, epf_bar);
+               ret = pci_epc_set_bar(epc, epf->func_no, epf->vfunc_no,
+                                     epf_bar);
                if (ret) {
                        pci_epf_free_space(epf, epf_test->reg[bar], bar,
                                           PRIMARY_INTERFACE);
@@ -674,13 +680,13 @@ static int pci_epf_test_core_init(struct pci_epf *epf)
        bool msi_capable = true;
        int ret;
 
-       epc_features = pci_epc_get_features(epc, epf->func_no);
+       epc_features = pci_epc_get_features(epc, epf->func_no, epf->vfunc_no);
        if (epc_features) {
                msix_capable = epc_features->msix_capable;
                msi_capable = epc_features->msi_capable;
        }
 
-       ret = pci_epc_write_header(epc, epf->func_no, header);
+       ret = pci_epc_write_header(epc, epf->func_no, epf->vfunc_no, header);
        if (ret) {
                dev_err(dev, "Configuration header write failed\n");
                return ret;
@@ -691,7 +697,8 @@ static int pci_epf_test_core_init(struct pci_epf *epf)
                return ret;
 
        if (msi_capable) {
-               ret = pci_epc_set_msi(epc, epf->func_no, epf->msi_interrupts);
+               ret = pci_epc_set_msi(epc, epf->func_no, epf->vfunc_no,
+                                     epf->msi_interrupts);
                if (ret) {
                        dev_err(dev, "MSI configuration failed\n");
                        return ret;
@@ -699,7 +706,8 @@ static int pci_epf_test_core_init(struct pci_epf *epf)
        }
 
        if (msix_capable) {
-               ret = pci_epc_set_msix(epc, epf->func_no, epf->msix_interrupts,
+               ret = pci_epc_set_msix(epc, epf->func_no, epf->vfunc_no,
+                                      epf->msix_interrupts,
                                       epf_test->test_reg_bar,
                                       epf_test->msix_table_offset);
                if (ret) {
@@ -832,7 +840,7 @@ static int pci_epf_test_bind(struct pci_epf *epf)
        if (WARN_ON_ONCE(!epc))
                return -EINVAL;
 
-       epc_features = pci_epc_get_features(epc, epf->func_no);
+       epc_features = pci_epc_get_features(epc, epf->func_no, epf->vfunc_no);
        if (epc_features) {
                linkup_notifier = epc_features->linkup_notifier;
                core_init_notifier = epc_features->core_init_notifier;
diff --git a/drivers/pci/endpoint/pci-epc-core.c 
b/drivers/pci/endpoint/pci-epc-core.c
index fe6d23849502..23594165fa68 100644
--- a/drivers/pci/endpoint/pci-epc-core.c
+++ b/drivers/pci/endpoint/pci-epc-core.c
@@ -137,24 +137,29 @@ EXPORT_SYMBOL_GPL(pci_epc_get_next_free_bar);
  * @epc: the features supported by *this* EPC device will be returned
  * @func_no: the features supported by the EPC device specific to the
  *          endpoint function with func_no will be returned
+ * @vfunc_no: the features supported by the EPC device specific to the
+ *          virtual endpoint function with vfunc_no will be returned
  *
  * Invoke to get the features provided by the EPC which may be
  * specific to an endpoint function. Returns pci_epc_features on success
  * and NULL for any failures.
  */
 const struct pci_epc_features *pci_epc_get_features(struct pci_epc *epc,
-                                                   u8 func_no)
+                                                   u8 func_no, u8 vfunc_no)
 {
        const struct pci_epc_features *epc_features;
 
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return NULL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return NULL;
+
        if (!epc->ops->get_features)
                return NULL;
 
        mutex_lock(&epc->lock);
-       epc_features = epc->ops->get_features(epc, func_no);
+       epc_features = epc->ops->get_features(epc, func_no, vfunc_no);
        mutex_unlock(&epc->lock);
 
        return epc_features;
@@ -205,13 +210,14 @@ EXPORT_SYMBOL_GPL(pci_epc_start);
 /**
  * pci_epc_raise_irq() - interrupt the host system
  * @epc: the EPC device which has to interrupt the host
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @type: specify the type of interrupt; legacy, MSI or MSI-X
  * @interrupt_num: the MSI or MSI-X interrupt number
  *
  * Invoke to raise an legacy, MSI or MSI-X interrupt
  */
-int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no,
+int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                      enum pci_epc_irq_type type, u16 interrupt_num)
 {
        int ret;
@@ -219,11 +225,14 @@ int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no,
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->raise_irq)
                return 0;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->raise_irq(epc, func_no, type, interrupt_num);
+       ret = epc->ops->raise_irq(epc, func_no, vfunc_no, type, interrupt_num);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -235,6 +244,7 @@ EXPORT_SYMBOL_GPL(pci_epc_raise_irq);
  *                         MSI data
  * @epc: the EPC device which has the MSI capability
  * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @phys_addr: the physical address of the outbound region
  * @interrupt_num: the MSI interrupt number
  * @entry_size: Size of Outbound address region for each interrupt
@@ -250,21 +260,25 @@ EXPORT_SYMBOL_GPL(pci_epc_raise_irq);
  * physical address (in outbound region) of the other interface to ring
  * doorbell.
  */
-int pci_epc_map_msi_irq(struct pci_epc *epc, u8 func_no, phys_addr_t phys_addr,
-                       u8 interrupt_num, u32 entry_size, u32 *msi_data,
-                       u32 *msi_addr_offset)
+int pci_epc_map_msi_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                       phys_addr_t phys_addr, u8 interrupt_num, u32 entry_size,
+                       u32 *msi_data, u32 *msi_addr_offset)
 {
        int ret;
 
        if (IS_ERR_OR_NULL(epc))
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->map_msi_irq)
                return -EINVAL;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->map_msi_irq(epc, func_no, phys_addr, interrupt_num,
-                                   entry_size, msi_data, msi_addr_offset);
+       ret = epc->ops->map_msi_irq(epc, func_no, vfunc_no, phys_addr,
+                                   interrupt_num, entry_size, msi_data,
+                                   msi_addr_offset);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -274,22 +288,26 @@ EXPORT_SYMBOL_GPL(pci_epc_map_msi_irq);
 /**
  * pci_epc_get_msi() - get the number of MSI interrupt numbers allocated
  * @epc: the EPC device to which MSI interrupts was requested
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  *
  * Invoke to get the number of MSI interrupts allocated by the RC
  */
-int pci_epc_get_msi(struct pci_epc *epc, u8 func_no)
+int pci_epc_get_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        int interrupt;
 
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return 0;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return 0;
+
        if (!epc->ops->get_msi)
                return 0;
 
        mutex_lock(&epc->lock);
-       interrupt = epc->ops->get_msi(epc, func_no);
+       interrupt = epc->ops->get_msi(epc, func_no, vfunc_no);
        mutex_unlock(&epc->lock);
 
        if (interrupt < 0)
@@ -304,12 +322,13 @@ EXPORT_SYMBOL_GPL(pci_epc_get_msi);
 /**
  * pci_epc_set_msi() - set the number of MSI interrupt numbers required
  * @epc: the EPC device on which MSI has to be configured
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @interrupts: number of MSI interrupts required by the EPF
  *
  * Invoke to set the required number of MSI interrupts.
  */
-int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 interrupts)
+int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no, u8 
interrupts)
 {
        int ret;
        u8 encode_int;
@@ -318,13 +337,16 @@ int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 
interrupts)
            interrupts > 32)
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->set_msi)
                return 0;
 
        encode_int = order_base_2(interrupts);
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->set_msi(epc, func_no, encode_int);
+       ret = epc->ops->set_msi(epc, func_no, vfunc_no, encode_int);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -334,22 +356,26 @@ EXPORT_SYMBOL_GPL(pci_epc_set_msi);
 /**
  * pci_epc_get_msix() - get the number of MSI-X interrupt numbers allocated
  * @epc: the EPC device to which MSI-X interrupts was requested
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  *
  * Invoke to get the number of MSI-X interrupts allocated by the RC
  */
-int pci_epc_get_msix(struct pci_epc *epc, u8 func_no)
+int pci_epc_get_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        int interrupt;
 
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return 0;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return 0;
+
        if (!epc->ops->get_msix)
                return 0;
 
        mutex_lock(&epc->lock);
-       interrupt = epc->ops->get_msix(epc, func_no);
+       interrupt = epc->ops->get_msix(epc, func_no, vfunc_no);
        mutex_unlock(&epc->lock);
 
        if (interrupt < 0)
@@ -362,15 +388,16 @@ EXPORT_SYMBOL_GPL(pci_epc_get_msix);
 /**
  * pci_epc_set_msix() - set the number of MSI-X interrupt numbers required
  * @epc: the EPC device on which MSI-X has to be configured
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @interrupts: number of MSI-X interrupts required by the EPF
  * @bir: BAR where the MSI-X table resides
  * @offset: Offset pointing to the start of MSI-X table
  *
  * Invoke to set the required number of MSI-X interrupts.
  */
-int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u16 interrupts,
-                    enum pci_barno bir, u32 offset)
+int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                    u16 interrupts, enum pci_barno bir, u32 offset)
 {
        int ret;
 
@@ -378,11 +405,15 @@ int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u16 
interrupts,
            interrupts < 1 || interrupts > 2048)
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->set_msix)
                return 0;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->set_msix(epc, func_no, interrupts - 1, bir, offset);
+       ret = epc->ops->set_msix(epc, func_no, vfunc_no, interrupts - 1, bir,
+                                offset);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -392,22 +423,26 @@ EXPORT_SYMBOL_GPL(pci_epc_set_msix);
 /**
  * pci_epc_unmap_addr() - unmap CPU address from PCI address
  * @epc: the EPC device on which address is allocated
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @phys_addr: physical address of the local system
  *
  * Invoke to unmap the CPU address from PCI address.
  */
-void pci_epc_unmap_addr(struct pci_epc *epc, u8 func_no,
+void pci_epc_unmap_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                        phys_addr_t phys_addr)
 {
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return;
+
        if (!epc->ops->unmap_addr)
                return;
 
        mutex_lock(&epc->lock);
-       epc->ops->unmap_addr(epc, func_no, phys_addr);
+       epc->ops->unmap_addr(epc, func_no, vfunc_no, phys_addr);
        mutex_unlock(&epc->lock);
 }
 EXPORT_SYMBOL_GPL(pci_epc_unmap_addr);
@@ -415,14 +450,15 @@ EXPORT_SYMBOL_GPL(pci_epc_unmap_addr);
 /**
  * pci_epc_map_addr() - map CPU address to PCI address
  * @epc: the EPC device on which address is allocated
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @phys_addr: physical address of the local system
  * @pci_addr: PCI address to which the physical address should be mapped
  * @size: the size of the allocation
  *
  * Invoke to map CPU address with PCI address.
  */
-int pci_epc_map_addr(struct pci_epc *epc, u8 func_no,
+int pci_epc_map_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                     phys_addr_t phys_addr, u64 pci_addr, size_t size)
 {
        int ret;
@@ -430,11 +466,15 @@ int pci_epc_map_addr(struct pci_epc *epc, u8 func_no,
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->map_addr)
                return 0;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->map_addr(epc, func_no, phys_addr, pci_addr, size);
+       ret = epc->ops->map_addr(epc, func_no, vfunc_no, phys_addr, pci_addr,
+                                size);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -444,12 +484,13 @@ EXPORT_SYMBOL_GPL(pci_epc_map_addr);
 /**
  * pci_epc_clear_bar() - reset the BAR
  * @epc: the EPC device for which the BAR has to be cleared
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @epf_bar: the struct epf_bar that contains the BAR information
  *
  * Invoke to reset the BAR of the endpoint device.
  */
-void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no,
+void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                       struct pci_epf_bar *epf_bar)
 {
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions ||
@@ -457,11 +498,14 @@ void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no,
             epf_bar->flags & PCI_BASE_ADDRESS_MEM_TYPE_64))
                return;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return;
+
        if (!epc->ops->clear_bar)
                return;
 
        mutex_lock(&epc->lock);
-       epc->ops->clear_bar(epc, func_no, epf_bar);
+       epc->ops->clear_bar(epc, func_no, vfunc_no, epf_bar);
        mutex_unlock(&epc->lock);
 }
 EXPORT_SYMBOL_GPL(pci_epc_clear_bar);
@@ -469,12 +513,13 @@ EXPORT_SYMBOL_GPL(pci_epc_clear_bar);
 /**
  * pci_epc_set_bar() - configure BAR in order for host to assign PCI addr space
  * @epc: the EPC device on which BAR has to be configured
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @epf_bar: the struct epf_bar that contains the BAR information
  *
  * Invoke to configure the BAR of the endpoint device.
  */
-int pci_epc_set_bar(struct pci_epc *epc, u8 func_no,
+int pci_epc_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                    struct pci_epf_bar *epf_bar)
 {
        int ret;
@@ -489,11 +534,14 @@ int pci_epc_set_bar(struct pci_epc *epc, u8 func_no,
             !(flags & PCI_BASE_ADDRESS_MEM_TYPE_64)))
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->set_bar)
                return 0;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->set_bar(epc, func_no, epf_bar);
+       ret = epc->ops->set_bar(epc, func_no, vfunc_no, epf_bar);
        mutex_unlock(&epc->lock);
 
        return ret;
@@ -503,7 +551,8 @@ EXPORT_SYMBOL_GPL(pci_epc_set_bar);
 /**
  * pci_epc_write_header() - write standard configuration header
  * @epc: the EPC device to which the configuration header should be written
- * @func_no: the endpoint function number in the EPC device
+ * @func_no: the physical endpoint function number in the EPC device
+ * @vfunc_no: the virtual endpoint function number in the physical function
  * @header: standard configuration header fields
  *
  * Invoke to write the configuration header to the endpoint controller. Every
@@ -511,7 +560,7 @@ EXPORT_SYMBOL_GPL(pci_epc_set_bar);
  * configuration header would be written. The callback function should write
  * the header fields to this dedicated location.
  */
-int pci_epc_write_header(struct pci_epc *epc, u8 func_no,
+int pci_epc_write_header(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                         struct pci_epf_header *header)
 {
        int ret;
@@ -519,11 +568,14 @@ int pci_epc_write_header(struct pci_epc *epc, u8 func_no,
        if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
                return -EINVAL;
 
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return -EINVAL;
+
        if (!epc->ops->write_header)
                return 0;
 
        mutex_lock(&epc->lock);
-       ret = epc->ops->write_header(epc, func_no, header);
+       ret = epc->ops->write_header(epc, func_no, vfunc_no, header);
        mutex_unlock(&epc->lock);
 
        return ret;
diff --git a/drivers/pci/endpoint/pci-epf-core.c 
b/drivers/pci/endpoint/pci-epf-core.c
index 3f1271a63be5..ff39b54e5d05 100644
--- a/drivers/pci/endpoint/pci-epf-core.c
+++ b/drivers/pci/endpoint/pci-epf-core.c
@@ -90,11 +90,14 @@ EXPORT_SYMBOL_GPL(pci_epf_unbind);
  */
 int pci_epf_bind(struct pci_epf *epf)
 {
+       struct device *dev = &epf->dev;
        struct pci_epf *epf_vf;
+       u8 func_no, vfunc_no;
+       struct pci_epc *epc;
        int ret;
 
        if (!epf->driver) {
-               dev_WARN(&epf->dev, "epf device not bound to driver\n");
+               dev_WARN(dev, "epf device not bound to driver\n");
                return -EINVAL;
        }
 
@@ -103,7 +106,50 @@ int pci_epf_bind(struct pci_epf *epf)
 
        mutex_lock(&epf->lock);
        list_for_each_entry(epf_vf, &epf->pci_vepf, list) {
+               vfunc_no = epf_vf->vfunc_no;
+
+               if (vfunc_no < 1) {
+                       dev_err(dev, "Invalid virtual function number\n");
+                       ret = -EINVAL;
+                       goto ret;
+               }
+
+               epc = epf->epc;
+               func_no = epf->func_no;
+               if (!IS_ERR_OR_NULL(epc)) {
+                       if (!epc->max_vfs) {
+                               dev_err(dev, "No support for virt function\n");
+                               ret = -EINVAL;
+                               goto ret;
+                       }
+
+                       if (vfunc_no > epc->max_vfs[func_no]) {
+                               dev_err(dev, "PF%d: Exceeds max vfunc number\n",
+                                       func_no);
+                               ret = -EINVAL;
+                               goto ret;
+                       }
+               }
+
+               epc = epf->sec_epc;
+               func_no = epf->sec_epc_func_no;
+               if (!IS_ERR_OR_NULL(epc)) {
+                       if (!epc->max_vfs) {
+                               dev_err(dev, "No support for virt function\n");
+                               ret = -EINVAL;
+                               goto ret;
+                       }
+
+                       if (vfunc_no > epc->max_vfs[func_no]) {
+                               dev_err(dev, "PF%d: Exceeds max vfunc number\n",
+                                       func_no);
+                               ret = -EINVAL;
+                               goto ret;
+                       }
+               }
+
                epf_vf->func_no = epf->func_no;
+               epf_vf->sec_epc_func_no = epf->sec_epc_func_no;
                epf_vf->epc = epf->epc;
                epf_vf->sec_epc = epf->sec_epc;
                ret = epf_vf->driver->ops->bind(epf_vf);
diff --git a/include/linux/pci-epc.h b/include/linux/pci-epc.h
index b82c9b100e97..3485d1a98682 100644
--- a/include/linux/pci-epc.h
+++ b/include/linux/pci-epc.h
@@ -61,31 +61,32 @@ pci_epc_interface_string(enum pci_epc_interface_type type)
  * @owner: the module owner containing the ops
  */
 struct pci_epc_ops {
-       int     (*write_header)(struct pci_epc *epc, u8 func_no,
+       int     (*write_header)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                                struct pci_epf_header *hdr);
-       int     (*set_bar)(struct pci_epc *epc, u8 func_no,
+       int     (*set_bar)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                           struct pci_epf_bar *epf_bar);
-       void    (*clear_bar)(struct pci_epc *epc, u8 func_no,
+       void    (*clear_bar)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                             struct pci_epf_bar *epf_bar);
-       int     (*map_addr)(struct pci_epc *epc, u8 func_no,
+       int     (*map_addr)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                            phys_addr_t addr, u64 pci_addr, size_t size);
-       void    (*unmap_addr)(struct pci_epc *epc, u8 func_no,
+       void    (*unmap_addr)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                              phys_addr_t addr);
-       int     (*set_msi)(struct pci_epc *epc, u8 func_no, u8 interrupts);
-       int     (*get_msi)(struct pci_epc *epc, u8 func_no);
-       int     (*set_msix)(struct pci_epc *epc, u8 func_no, u16 interrupts,
-                           enum pci_barno, u32 offset);
-       int     (*get_msix)(struct pci_epc *epc, u8 func_no);
-       int     (*raise_irq)(struct pci_epc *epc, u8 func_no,
+       int     (*set_msi)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                          u8 interrupts);
+       int     (*get_msi)(struct pci_epc *epc, u8 func_no, u8 vfunc_no);
+       int     (*set_msix)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                           u16 interrupts, enum pci_barno, u32 offset);
+       int     (*get_msix)(struct pci_epc *epc, u8 func_no, u8 vfunc_no);
+       int     (*raise_irq)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                             enum pci_epc_irq_type type, u16 interrupt_num);
-       int     (*map_msi_irq)(struct pci_epc *epc, u8 func_no,
+       int     (*map_msi_irq)(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                               phys_addr_t phys_addr, u8 interrupt_num,
                               u32 entry_size, u32 *msi_data,
                               u32 *msi_addr_offset);
        int     (*start)(struct pci_epc *epc);
        void    (*stop)(struct pci_epc *epc);
        const struct pci_epc_features* (*get_features)(struct pci_epc *epc,
-                                                      u8 func_no);
+                                                      u8 func_no, u8 vfunc_no);
        struct module *owner;
 };
 
@@ -127,6 +128,8 @@ struct pci_epc_mem {
  *       single window.
  * @num_windows: number of windows supported by device
  * @max_functions: max number of functions that can be configured in this EPC
+ * @max_vfs: Array indicating the maximum number of virtual functions that can
+ *   be associated with each physical function
  * @group: configfs group representing the PCI EPC device
  * @lock: mutex to protect pci_epc ops
  * @function_num_map: bitmap to manage physical function number
@@ -140,6 +143,7 @@ struct pci_epc {
        struct pci_epc_mem              *mem;
        unsigned int                    num_windows;
        u8                              max_functions;
+       u8                              *max_vfs;
        struct config_group             *group;
        /* mutex to protect against concurrent access of EP controller */
        struct mutex                    lock;
@@ -205,31 +209,32 @@ void pci_epc_linkup(struct pci_epc *epc);
 void pci_epc_init_notify(struct pci_epc *epc);
 void pci_epc_remove_epf(struct pci_epc *epc, struct pci_epf *epf,
                        enum pci_epc_interface_type type);
-int pci_epc_write_header(struct pci_epc *epc, u8 func_no,
+int pci_epc_write_header(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                         struct pci_epf_header *hdr);
-int pci_epc_set_bar(struct pci_epc *epc, u8 func_no,
+int pci_epc_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                    struct pci_epf_bar *epf_bar);
-void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no,
+void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                       struct pci_epf_bar *epf_bar);
-int pci_epc_map_addr(struct pci_epc *epc, u8 func_no,
+int pci_epc_map_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                     phys_addr_t phys_addr,
                     u64 pci_addr, size_t size);
-void pci_epc_unmap_addr(struct pci_epc *epc, u8 func_no,
+void pci_epc_unmap_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                        phys_addr_t phys_addr);
-int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 interrupts);
-int pci_epc_get_msi(struct pci_epc *epc, u8 func_no);
-int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u16 interrupts,
-                    enum pci_barno, u32 offset);
-int pci_epc_get_msix(struct pci_epc *epc, u8 func_no);
-int pci_epc_map_msi_irq(struct pci_epc *epc, u8 func_no,
+int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                   u8 interrupts);
+int pci_epc_get_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no);
+int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
+                    u16 interrupts, enum pci_barno, u32 offset);
+int pci_epc_get_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no);
+int pci_epc_map_msi_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                        phys_addr_t phys_addr, u8 interrupt_num,
                        u32 entry_size, u32 *msi_data, u32 *msi_addr_offset);
-int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no,
+int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                      enum pci_epc_irq_type type, u16 interrupt_num);
 int pci_epc_start(struct pci_epc *epc);
 void pci_epc_stop(struct pci_epc *epc);
 const struct pci_epc_features *pci_epc_get_features(struct pci_epc *epc,
-                                                   u8 func_no);
+                                                   u8 func_no, u8 vfunc_no);
 enum pci_barno
 pci_epc_get_first_free_bar(const struct pci_epc_features *epc_features);
 enum pci_barno pci_epc_get_next_free_bar(const struct pci_epc_features
-- 
2.17.1

Reply via email to