PCI: dwc: Add dw_pcie_ep_{read,write}_dbi[2] helpers
authorYoshihiro Shimoda <yoshihiro.shimoda.uh@renesas.com>
Wed, 20 Dec 2023 05:38:27 +0000 (14:38 +0900)
committerKrzysztof Wilczyński <kwilczynski@kernel.org>
Sat, 6 Jan 2024 07:51:09 +0000 (07:51 +0000)
The current code calculated some dbi[2] registers' offset by calling
dw_pcie_ep_get_dbi[2]_offset() in each function. To improve the code
readability, add dw_pcie_ep_{read,write}_dbi[2} and some data-width
related helpers.

Link: https://lore.kernel.org/linux-pci/20231220053829.1921187-5-yoshihiro.shimoda.uh@renesas.com
Signed-off-by: Yoshihiro Shimoda <yoshihiro.shimoda.uh@renesas.com>
Signed-off-by: Krzysztof Wilczyński <kwilczynski@kernel.org>
Reviewed-by: Serge Semin <fancer.lancer@gmail.com>
Reviewed-by: Manivannan Sadhasivam <manivannan.sadhasivam@linaro.org>
drivers/pci/controller/dwc/pcie-designware-ep.c
drivers/pci/controller/dwc/pcie-designware.h

index adef0bc..e091783 100644 (file)
@@ -43,46 +43,19 @@ dw_pcie_ep_get_func_from_ep(struct dw_pcie_ep *ep, u8 func_no)
        return NULL;
 }
 
-static unsigned int dw_pcie_ep_get_dbi_offset(struct dw_pcie_ep *ep, u8 func_no)
-{
-       unsigned int dbi_offset = 0;
-
-       if (ep->ops->get_dbi_offset)
-               dbi_offset = ep->ops->get_dbi_offset(ep, func_no);
-
-       return dbi_offset;
-}
-
-static unsigned int dw_pcie_ep_get_dbi2_offset(struct dw_pcie_ep *ep, u8 func_no)
-{
-       unsigned int dbi2_offset = 0;
-
-       if (ep->ops->get_dbi2_offset)
-               dbi2_offset = ep->ops->get_dbi2_offset(ep, func_no);
-       else if (ep->ops->get_dbi_offset)     /* for backward compatibility */
-               dbi2_offset = ep->ops->get_dbi_offset(ep, func_no);
-
-       return dbi2_offset;
-}
-
 static void __dw_pcie_ep_reset_bar(struct dw_pcie *pci, u8 func_no,
                                   enum pci_barno bar, int flags)
 {
-       unsigned int dbi_offset, dbi2_offset;
        struct dw_pcie_ep *ep = &pci->ep;
-       u32 reg, reg_dbi2;
-
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-       dbi2_offset = dw_pcie_ep_get_dbi2_offset(ep, func_no);
+       u32 reg;
 
-       reg = dbi_offset + PCI_BASE_ADDRESS_0 + (4 * bar);
-       reg_dbi2 = dbi2_offset + PCI_BASE_ADDRESS_0 + (4 * bar);
+       reg = PCI_BASE_ADDRESS_0 + (4 * bar);
        dw_pcie_dbi_ro_wr_en(pci);
-       dw_pcie_writel_dbi2(pci, reg_dbi2, 0x0);
-       dw_pcie_writel_dbi(pci, reg, 0x0);
+       dw_pcie_ep_writel_dbi2(ep, func_no, reg, 0x0);
+       dw_pcie_ep_writel_dbi(ep, func_no, reg, 0x0);
        if (flags & PCI_BASE_ADDRESS_MEM_TYPE_64) {
-               dw_pcie_writel_dbi2(pci, reg_dbi2 + 4, 0x0);
-               dw_pcie_writel_dbi(pci, reg + 4, 0x0);
+               dw_pcie_ep_writel_dbi2(ep, func_no, reg + 4, 0x0);
+               dw_pcie_ep_writel_dbi(ep, func_no, reg + 4, 0x0);
        }
        dw_pcie_dbi_ro_wr_dis(pci);
 }
@@ -99,19 +72,15 @@ void dw_pcie_ep_reset_bar(struct dw_pcie *pci, enum pci_barno bar)
 EXPORT_SYMBOL_GPL(dw_pcie_ep_reset_bar);
 
 static u8 __dw_pcie_ep_find_next_cap(struct dw_pcie_ep *ep, u8 func_no,
-               u8 cap_ptr, u8 cap)
+                                    u8 cap_ptr, u8 cap)
 {
-       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
-       unsigned int dbi_offset = 0;
        u8 cap_id, next_cap_ptr;
        u16 reg;
 
        if (!cap_ptr)
                return 0;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = dw_pcie_readw_dbi(pci, dbi_offset + cap_ptr);
+       reg = dw_pcie_ep_readw_dbi(ep, func_no, cap_ptr);
        cap_id = (reg & 0x00ff);
 
        if (cap_id > PCI_CAP_ID_MAX)
@@ -126,14 +95,10 @@ static u8 __dw_pcie_ep_find_next_cap(struct dw_pcie_ep *ep, u8 func_no,
 
 static u8 dw_pcie_ep_find_capability(struct dw_pcie_ep *ep, u8 func_no, u8 cap)
 {
-       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
-       unsigned int dbi_offset = 0;
        u8 next_cap_ptr;
        u16 reg;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = dw_pcie_readw_dbi(pci, dbi_offset + PCI_CAPABILITY_LIST);
+       reg = dw_pcie_ep_readw_dbi(ep, func_no, PCI_CAPABILITY_LIST);
        next_cap_ptr = (reg & 0x00ff);
 
        return __dw_pcie_ep_find_next_cap(ep, func_no, next_cap_ptr, cap);
@@ -144,24 +109,21 @@ static int dw_pcie_ep_write_header(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);
-       unsigned int dbi_offset = 0;
-
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
 
        dw_pcie_dbi_ro_wr_en(pci);
-       dw_pcie_writew_dbi(pci, dbi_offset + PCI_VENDOR_ID, hdr->vendorid);
-       dw_pcie_writew_dbi(pci, dbi_offset + PCI_DEVICE_ID, hdr->deviceid);
-       dw_pcie_writeb_dbi(pci, dbi_offset + PCI_REVISION_ID, hdr->revid);
-       dw_pcie_writeb_dbi(pci, dbi_offset + PCI_CLASS_PROG, hdr->progif_code);
-       dw_pcie_writew_dbi(pci, dbi_offset + PCI_CLASS_DEVICE,
-                          hdr->subclass_code | hdr->baseclass_code << 8);
-       dw_pcie_writeb_dbi(pci, dbi_offset + PCI_CACHE_LINE_SIZE,
-                          hdr->cache_line_size);
-       dw_pcie_writew_dbi(pci, dbi_offset + PCI_SUBSYSTEM_VENDOR_ID,
-                          hdr->subsys_vendor_id);
-       dw_pcie_writew_dbi(pci, dbi_offset + PCI_SUBSYSTEM_ID, hdr->subsys_id);
-       dw_pcie_writeb_dbi(pci, dbi_offset + PCI_INTERRUPT_PIN,
-                          hdr->interrupt_pin);
+       dw_pcie_ep_writew_dbi(ep, func_no, PCI_VENDOR_ID, hdr->vendorid);
+       dw_pcie_ep_writew_dbi(ep, func_no, PCI_DEVICE_ID, hdr->deviceid);
+       dw_pcie_ep_writeb_dbi(ep, func_no, PCI_REVISION_ID, hdr->revid);
+       dw_pcie_ep_writeb_dbi(ep, func_no, PCI_CLASS_PROG, hdr->progif_code);
+       dw_pcie_ep_writew_dbi(ep, func_no, PCI_CLASS_DEVICE,
+                             hdr->subclass_code | hdr->baseclass_code << 8);
+       dw_pcie_ep_writeb_dbi(ep, func_no, PCI_CACHE_LINE_SIZE,
+                             hdr->cache_line_size);
+       dw_pcie_ep_writew_dbi(ep, func_no, PCI_SUBSYSTEM_VENDOR_ID,
+                             hdr->subsys_vendor_id);
+       dw_pcie_ep_writew_dbi(ep, func_no, PCI_SUBSYSTEM_ID, hdr->subsys_id);
+       dw_pcie_ep_writeb_dbi(ep, func_no, PCI_INTERRUPT_PIN,
+                             hdr->interrupt_pin);
        dw_pcie_dbi_ro_wr_dis(pci);
 
        return 0;
@@ -243,18 +205,13 @@ static int dw_pcie_ep_set_bar(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);
-       unsigned int dbi_offset, dbi2_offset;
        enum pci_barno bar = epf_bar->barno;
        size_t size = epf_bar->size;
        int flags = epf_bar->flags;
-       u32 reg, reg_dbi2;
        int ret, type;
+       u32 reg;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-       dbi2_offset = dw_pcie_ep_get_dbi2_offset(ep, func_no);
-
-       reg = PCI_BASE_ADDRESS_0 + (4 * bar) + dbi_offset;
-       reg_dbi2 = PCI_BASE_ADDRESS_0 + (4 * bar) + dbi2_offset;
+       reg = PCI_BASE_ADDRESS_0 + (4 * bar);
 
        if (!(flags & PCI_BASE_ADDRESS_SPACE))
                type = PCIE_ATU_TYPE_MEM;
@@ -270,12 +227,12 @@ static int dw_pcie_ep_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 
        dw_pcie_dbi_ro_wr_en(pci);
 
-       dw_pcie_writel_dbi2(pci, reg_dbi2, lower_32_bits(size - 1));
-       dw_pcie_writel_dbi(pci, reg, flags);
+       dw_pcie_ep_writel_dbi2(ep, func_no, reg, lower_32_bits(size - 1));
+       dw_pcie_ep_writel_dbi(ep, func_no, reg, flags);
 
        if (flags & PCI_BASE_ADDRESS_MEM_TYPE_64) {
-               dw_pcie_writel_dbi2(pci, reg_dbi2 + 4, upper_32_bits(size - 1));
-               dw_pcie_writel_dbi(pci, reg + 4, 0);
+               dw_pcie_ep_writel_dbi2(ep, func_no, reg + 4, upper_32_bits(size - 1));
+               dw_pcie_ep_writel_dbi(ep, func_no, reg + 4, 0);
        }
 
        ep->epf_bar[bar] = epf_bar;
@@ -335,19 +292,15 @@ static int dw_pcie_ep_map_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_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);
-       u32 val, reg;
-       unsigned int dbi_offset = 0;
        struct dw_pcie_ep_func *ep_func;
+       u32 val, reg;
 
        ep_func = dw_pcie_ep_get_func_from_ep(ep, func_no);
        if (!ep_func || !ep_func->msi_cap)
                return -EINVAL;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = ep_func->msi_cap + dbi_offset + PCI_MSI_FLAGS;
-       val = dw_pcie_readw_dbi(pci, reg);
+       reg = ep_func->msi_cap + PCI_MSI_FLAGS;
+       val = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        if (!(val & PCI_MSI_FLAGS_ENABLE))
                return -EINVAL;
 
@@ -361,22 +314,19 @@ static int dw_pcie_ep_set_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);
-       u32 val, reg;
-       unsigned int dbi_offset = 0;
        struct dw_pcie_ep_func *ep_func;
+       u32 val, reg;
 
        ep_func = dw_pcie_ep_get_func_from_ep(ep, func_no);
        if (!ep_func || !ep_func->msi_cap)
                return -EINVAL;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = ep_func->msi_cap + dbi_offset + PCI_MSI_FLAGS;
-       val = dw_pcie_readw_dbi(pci, reg);
+       reg = ep_func->msi_cap + PCI_MSI_FLAGS;
+       val = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        val &= ~PCI_MSI_FLAGS_QMASK;
        val |= FIELD_PREP(PCI_MSI_FLAGS_QMASK, interrupts);
        dw_pcie_dbi_ro_wr_en(pci);
-       dw_pcie_writew_dbi(pci, reg, val);
+       dw_pcie_ep_writew_dbi(ep, func_no, reg, val);
        dw_pcie_dbi_ro_wr_dis(pci);
 
        return 0;
@@ -385,19 +335,15 @@ static int dw_pcie_ep_set_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_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);
-       u32 val, reg;
-       unsigned int dbi_offset = 0;
        struct dw_pcie_ep_func *ep_func;
+       u32 val, reg;
 
        ep_func = dw_pcie_ep_get_func_from_ep(ep, func_no);
        if (!ep_func || !ep_func->msix_cap)
                return -EINVAL;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = ep_func->msix_cap + dbi_offset + PCI_MSIX_FLAGS;
-       val = dw_pcie_readw_dbi(pci, reg);
+       reg = ep_func->msix_cap + PCI_MSIX_FLAGS;
+       val = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        if (!(val & PCI_MSIX_FLAGS_ENABLE))
                return -EINVAL;
 
@@ -411,9 +357,8 @@ static int dw_pcie_ep_set_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);
-       u32 val, reg;
-       unsigned int dbi_offset = 0;
        struct dw_pcie_ep_func *ep_func;
+       u32 val, reg;
 
        ep_func = dw_pcie_ep_get_func_from_ep(ep, func_no);
        if (!ep_func || !ep_func->msix_cap)
@@ -421,21 +366,19 @@ static int dw_pcie_ep_set_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 
        dw_pcie_dbi_ro_wr_en(pci);
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = ep_func->msix_cap + dbi_offset + PCI_MSIX_FLAGS;
-       val = dw_pcie_readw_dbi(pci, reg);
+       reg = ep_func->msix_cap + PCI_MSIX_FLAGS;
+       val = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        val &= ~PCI_MSIX_FLAGS_QSIZE;
        val |= interrupts;
        dw_pcie_writew_dbi(pci, reg, val);
 
-       reg = ep_func->msix_cap + dbi_offset + PCI_MSIX_TABLE;
+       reg = ep_func->msix_cap + PCI_MSIX_TABLE;
        val = offset | bir;
-       dw_pcie_writel_dbi(pci, reg, val);
+       dw_pcie_ep_writel_dbi(ep, func_no, reg, val);
 
-       reg = ep_func->msix_cap + dbi_offset + PCI_MSIX_PBA;
+       reg = ep_func->msix_cap + PCI_MSIX_PBA;
        val = (offset + (interrupts * PCI_MSIX_ENTRY_SIZE)) | bir;
-       dw_pcie_writel_dbi(pci, reg, val);
+       dw_pcie_ep_writel_dbi(ep, func_no, reg, val);
 
        dw_pcie_dbi_ro_wr_dis(pci);
 
@@ -510,38 +453,34 @@ EXPORT_SYMBOL_GPL(dw_pcie_ep_raise_legacy_irq);
 int dw_pcie_ep_raise_msi_irq(struct dw_pcie_ep *ep, u8 func_no,
                             u8 interrupt_num)
 {
-       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+       u32 msg_addr_lower, msg_addr_upper, reg;
        struct dw_pcie_ep_func *ep_func;
        struct pci_epc *epc = ep->epc;
        unsigned int aligned_offset;
-       unsigned int dbi_offset = 0;
        u16 msg_ctrl, msg_data;
-       u32 msg_addr_lower, msg_addr_upper, reg;
-       u64 msg_addr;
        bool has_upper;
+       u64 msg_addr;
        int ret;
 
        ep_func = dw_pcie_ep_get_func_from_ep(ep, func_no);
        if (!ep_func || !ep_func->msi_cap)
                return -EINVAL;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
        /* Raise MSI per the PCI Local Bus Specification Revision 3.0, 6.8.1. */
-       reg = ep_func->msi_cap + dbi_offset + PCI_MSI_FLAGS;
-       msg_ctrl = dw_pcie_readw_dbi(pci, reg);
+       reg = ep_func->msi_cap + PCI_MSI_FLAGS;
+       msg_ctrl = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        has_upper = !!(msg_ctrl & PCI_MSI_FLAGS_64BIT);
-       reg = ep_func->msi_cap + dbi_offset + PCI_MSI_ADDRESS_LO;
-       msg_addr_lower = dw_pcie_readl_dbi(pci, reg);
+       reg = ep_func->msi_cap + PCI_MSI_ADDRESS_LO;
+       msg_addr_lower = dw_pcie_ep_readl_dbi(ep, func_no, reg);
        if (has_upper) {
-               reg = ep_func->msi_cap + dbi_offset + PCI_MSI_ADDRESS_HI;
-               msg_addr_upper = dw_pcie_readl_dbi(pci, reg);
-               reg = ep_func->msi_cap + dbi_offset + PCI_MSI_DATA_64;
-               msg_data = dw_pcie_readw_dbi(pci, reg);
+               reg = ep_func->msi_cap + PCI_MSI_ADDRESS_HI;
+               msg_addr_upper = dw_pcie_ep_readl_dbi(ep, func_no, reg);
+               reg = ep_func->msi_cap + PCI_MSI_DATA_64;
+               msg_data = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        } else {
                msg_addr_upper = 0;
-               reg = ep_func->msi_cap + dbi_offset + PCI_MSI_DATA_32;
-               msg_data = dw_pcie_readw_dbi(pci, reg);
+               reg = ep_func->msi_cap + PCI_MSI_DATA_32;
+               msg_data = dw_pcie_ep_readw_dbi(ep, func_no, reg);
        }
        aligned_offset = msg_addr_lower & (epc->mem->window.page_size - 1);
        msg_addr = ((u64)msg_addr_upper) << 32 |
@@ -582,10 +521,9 @@ int dw_pcie_ep_raise_msix_irq(struct dw_pcie_ep *ep, u8 func_no,
                              u16 interrupt_num)
 {
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
-       struct dw_pcie_ep_func *ep_func;
        struct pci_epf_msix_tbl *msix_tbl;
+       struct dw_pcie_ep_func *ep_func;
        struct pci_epc *epc = ep->epc;
-       unsigned int dbi_offset = 0;
        u32 reg, msg_data, vec_ctrl;
        unsigned int aligned_offset;
        u32 tbl_offset;
@@ -597,10 +535,8 @@ int dw_pcie_ep_raise_msix_irq(struct dw_pcie_ep *ep, u8 func_no,
        if (!ep_func || !ep_func->msix_cap)
                return -EINVAL;
 
-       dbi_offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
-
-       reg = ep_func->msix_cap + dbi_offset + PCI_MSIX_TABLE;
-       tbl_offset = dw_pcie_readl_dbi(pci, reg);
+       reg = ep_func->msix_cap + PCI_MSIX_TABLE;
+       tbl_offset = dw_pcie_ep_readl_dbi(ep, func_no, reg);
        bir = FIELD_GET(PCI_MSIX_TABLE_BIR, tbl_offset);
        tbl_offset &= PCI_MSIX_TABLE_OFFSET;
 
index 5e36da1..e8ec4f1 100644 (file)
@@ -486,6 +486,99 @@ static inline void dw_pcie_writel_dbi2(struct dw_pcie *pci, u32 reg, u32 val)
        dw_pcie_write_dbi2(pci, reg, 0x4, val);
 }
 
+static inline unsigned int dw_pcie_ep_get_dbi_offset(struct dw_pcie_ep *ep,
+                                                    u8 func_no)
+{
+       unsigned int dbi_offset = 0;
+
+       if (ep->ops->get_dbi_offset)
+               dbi_offset = ep->ops->get_dbi_offset(ep, func_no);
+
+       return dbi_offset;
+}
+
+static inline u32 dw_pcie_ep_read_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                     u32 reg, size_t size)
+{
+       unsigned int offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
+       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+
+       return dw_pcie_read_dbi(pci, offset + reg, size);
+}
+
+static inline void dw_pcie_ep_write_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                       u32 reg, size_t size, u32 val)
+{
+       unsigned int offset = dw_pcie_ep_get_dbi_offset(ep, func_no);
+       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+
+       dw_pcie_write_dbi(pci, offset + reg, size, val);
+}
+
+static inline void dw_pcie_ep_writel_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                        u32 reg, u32 val)
+{
+       dw_pcie_ep_write_dbi(ep, func_no, reg, 0x4, val);
+}
+
+static inline u32 dw_pcie_ep_readl_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                      u32 reg)
+{
+       return dw_pcie_ep_read_dbi(ep, func_no, reg, 0x4);
+}
+
+static inline void dw_pcie_ep_writew_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                        u32 reg, u16 val)
+{
+       dw_pcie_ep_write_dbi(ep, func_no, reg, 0x2, val);
+}
+
+static inline u16 dw_pcie_ep_readw_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                      u32 reg)
+{
+       return dw_pcie_ep_read_dbi(ep, func_no, reg, 0x2);
+}
+
+static inline void dw_pcie_ep_writeb_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                        u32 reg, u8 val)
+{
+       dw_pcie_ep_write_dbi(ep, func_no, reg, 0x1, val);
+}
+
+static inline u8 dw_pcie_ep_readb_dbi(struct dw_pcie_ep *ep, u8 func_no,
+                                     u32 reg)
+{
+       return dw_pcie_ep_read_dbi(ep, func_no, reg, 0x1);
+}
+
+static inline unsigned int dw_pcie_ep_get_dbi2_offset(struct dw_pcie_ep *ep,
+                                                     u8 func_no)
+{
+       unsigned int dbi2_offset = 0;
+
+       if (ep->ops->get_dbi2_offset)
+               dbi2_offset = ep->ops->get_dbi2_offset(ep, func_no);
+       else if (ep->ops->get_dbi_offset)     /* for backward compatibility */
+               dbi2_offset = ep->ops->get_dbi_offset(ep, func_no);
+
+       return dbi2_offset;
+}
+
+static inline void dw_pcie_ep_write_dbi2(struct dw_pcie_ep *ep, u8 func_no,
+                                        u32 reg, size_t size, u32 val)
+{
+       unsigned int offset = dw_pcie_ep_get_dbi2_offset(ep, func_no);
+       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+
+       dw_pcie_write_dbi2(pci, offset + reg, size, val);
+}
+
+static inline void dw_pcie_ep_writel_dbi2(struct dw_pcie_ep *ep, u8 func_no,
+                                         u32 reg, u32 val)
+{
+       dw_pcie_ep_write_dbi2(ep, func_no, reg, 0x4, val);
+}
+
 static inline void dw_pcie_dbi_ro_wr_en(struct dw_pcie *pci)
 {
        u32 reg;