#include <linux/pci-epc.h>
 #include <linux/pci-epf.h>
 
-/**
- * dw_pcie_ep_linkup - Notify EPF drivers about Link Up event
- * @ep: DWC EP device
- */
-void dw_pcie_ep_linkup(struct dw_pcie_ep *ep)
-{
-       struct pci_epc *epc = ep->epc;
-
-       pci_epc_linkup(epc);
-}
-EXPORT_SYMBOL_GPL(dw_pcie_ep_linkup);
-
 /**
  * dw_pcie_ep_init_notify - Notify EPF drivers about EPC initialization complete
  * @ep: DWC EP device
        return 0;
 }
 
+static void dw_pcie_ep_init_non_sticky_registers(struct dw_pcie *pci)
+{
+       unsigned int offset;
+       unsigned int nbars;
+       u32 reg, i;
+
+       offset = dw_pcie_ep_find_ext_capability(pci, PCI_EXT_CAP_ID_REBAR);
+
+       dw_pcie_dbi_ro_wr_en(pci);
+
+       if (offset) {
+               reg = dw_pcie_readl_dbi(pci, offset + PCI_REBAR_CTRL);
+               nbars = (reg & PCI_REBAR_CTRL_NBAR_MASK) >>
+                       PCI_REBAR_CTRL_NBAR_SHIFT;
+
+               /*
+                * PCIe r6.0, sec 7.8.6.2 require us to support at least one
+                * size in the range from 1 MB to 512 GB. Advertise support
+                * for 1 MB BAR size only.
+                */
+               for (i = 0; i < nbars; i++, offset += PCI_REBAR_CTRL)
+                       dw_pcie_writel_dbi(pci, offset + PCI_REBAR_CAP, 0x0);
+       }
+
+       dw_pcie_setup(pci);
+       dw_pcie_dbi_ro_wr_dis(pci);
+}
+
 /**
  * dw_pcie_ep_init_registers - Initialize DWC EP specific registers
  * @ep: DWC EP device
        struct dw_pcie_ep_func *ep_func;
        struct device *dev = pci->dev;
        struct pci_epc *epc = ep->epc;
-       unsigned int offset, ptm_cap_base;
-       unsigned int nbars;
+       u32 ptm_cap_base, reg;
        u8 hdr_type;
        u8 func_no;
-       int i, ret;
        void *addr;
-       u32 reg;
+       int ret;
 
        hdr_type = dw_pcie_readb_dbi(pci, PCI_HEADER_TYPE) &
                   PCI_HEADER_TYPE_MASK;
        if (ep->ops->init)
                ep->ops->init(ep);
 
-       offset = dw_pcie_ep_find_ext_capability(pci, PCI_EXT_CAP_ID_REBAR);
        ptm_cap_base = dw_pcie_ep_find_ext_capability(pci, PCI_EXT_CAP_ID_PTM);
 
-       dw_pcie_dbi_ro_wr_en(pci);
-
-       if (offset) {
-               reg = dw_pcie_readl_dbi(pci, offset + PCI_REBAR_CTRL);
-               nbars = (reg & PCI_REBAR_CTRL_NBAR_MASK) >>
-                       PCI_REBAR_CTRL_NBAR_SHIFT;
-
-               /*
-                * PCIe r6.0, sec 7.8.6.2 require us to support at least one
-                * size in the range from 1 MB to 512 GB. Advertise support
-                * for 1 MB BAR size only.
-                */
-               for (i = 0; i < nbars; i++, offset += PCI_REBAR_CTRL)
-                       dw_pcie_writel_dbi(pci, offset + PCI_REBAR_CAP, BIT(4));
-       }
-
        /*
         * PTM responder capability can be disabled only after disabling
         * PTM root capability.
                dw_pcie_dbi_ro_wr_dis(pci);
        }
 
-       dw_pcie_setup(pci);
-       dw_pcie_dbi_ro_wr_dis(pci);
+       dw_pcie_ep_init_non_sticky_registers(pci);
 
        return 0;
 
 }
 EXPORT_SYMBOL_GPL(dw_pcie_ep_init_registers);
 
+/**
+ * dw_pcie_ep_linkup - Notify EPF drivers about Link Up event
+ * @ep: DWC EP device
+ */
+void dw_pcie_ep_linkup(struct dw_pcie_ep *ep)
+{
+       struct pci_epc *epc = ep->epc;
+
+       pci_epc_linkup(epc);
+}
+EXPORT_SYMBOL_GPL(dw_pcie_ep_linkup);
+
+/**
+ * dw_pcie_ep_linkdown - Notify EPF drivers about Link Down event
+ * @ep: DWC EP device
+ *
+ * Non-sticky registers are also initialized before sending the notification to
+ * the EPF drivers. This is needed since the registers need to be initialized
+ * before the link comes back again.
+ */
+void dw_pcie_ep_linkdown(struct dw_pcie_ep *ep)
+{
+       struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+       struct pci_epc *epc = ep->epc;
+
+       /*
+        * Initialize the non-sticky DWC registers as they would've reset post
+        * Link Down. This is specifically needed for drivers not supporting
+        * PERST# as they have no way to reinitialize the registers before the
+        * link comes back again.
+        */
+       dw_pcie_ep_init_non_sticky_registers(pci);
+
+       pci_epc_linkdown(epc);
+}
+EXPORT_SYMBOL_GPL(dw_pcie_ep_linkdown);
+
 /**
  * dw_pcie_ep_init - Initialize the endpoint device
  * @ep: DWC EP device