]> asedeno.scripts.mit.edu Git - linux.git/commitdiff
Merge branch 'lorenzo/pci/rockchip'
authorBjorn Helgaas <bhelgaas@google.com>
Wed, 6 Jun 2018 21:10:43 +0000 (16:10 -0500)
committerBjorn Helgaas <bhelgaas@google.com>
Wed, 6 Jun 2018 21:10:43 +0000 (16:10 -0500)
  - update arm64 defconfig for Rockchip (Shawn Lin)

  - refactor Rockchip code to facilitate both root port and endpoint mode
    (Shawn Lin)

  - add Rockchip endpoint mode driver (Shawn Lin)

* lorenzo/pci/rockchip:
  arm64: defconfig: update config for Rockchip PCIe
  dt-bindings: PCI: rockchip: Add DT bindings for Rockchip PCIe EP driver
  PCI: rockchip: Add EP driver for Rockchip PCIe controller
  dt-bindings: PCI: rockchip: Rename rockchip-pcie.txt to rockchip-pcie-host.txt
  PCI: rockchip: Split out common function to init controller
  PCI: rockchip: Split out rockchip_pcie_parse_dt() to parse DT
  PCI: rockchip: Separate common code from RC driver

# Conflicts:
# drivers/pci/host/pcie-rockchip.c

Documentation/devicetree/bindings/pci/rockchip-pcie-ep.txt [new file with mode: 0644]
Documentation/devicetree/bindings/pci/rockchip-pcie-host.txt [moved from Documentation/devicetree/bindings/pci/rockchip-pcie.txt with 100% similarity]
MAINTAINERS
arch/arm64/configs/defconfig
drivers/pci/host/Kconfig
drivers/pci/host/Makefile
drivers/pci/host/pcie-rockchip-ep.c [new file with mode: 0644]
drivers/pci/host/pcie-rockchip-host.c [new file with mode: 0644]
drivers/pci/host/pcie-rockchip.c
drivers/pci/host/pcie-rockchip.h [new file with mode: 0644]

diff --git a/Documentation/devicetree/bindings/pci/rockchip-pcie-ep.txt b/Documentation/devicetree/bindings/pci/rockchip-pcie-ep.txt
new file mode 100644 (file)
index 0000000..7784673
--- /dev/null
@@ -0,0 +1,62 @@
+* Rockchip AXI PCIe Endpoint Controller DT description
+
+Required properties:
+- compatible: Should contain "rockchip,rk3399-pcie-ep"
+- reg: Two register ranges as listed in the reg-names property
+- reg-names: Must include the following names
+       - "apb-base"
+       - "mem-base"
+- clocks: Must contain an entry for each entry in clock-names.
+               See ../clocks/clock-bindings.txt for details.
+- clock-names: Must include the following entries:
+       - "aclk"
+       - "aclk-perf"
+       - "hclk"
+       - "pm"
+- resets: Must contain seven entries for each entry in reset-names.
+          See ../reset/reset.txt for details.
+- reset-names: Must include the following names
+       - "core"
+       - "mgmt"
+       - "mgmt-sticky"
+       - "pipe"
+       - "pm"
+       - "aclk"
+       - "pclk"
+- pinctrl-names : The pin control state names
+- pinctrl-0: The "default" pinctrl state
+- phys: Must contain an phandle to a PHY for each entry in phy-names.
+- phy-names: Must include 4 entries for all 4 lanes even if some of
+  them won't be used for your cases. Entries are of the form "pcie-phy-N":
+  where N ranges from 0 to 3.
+  (see example below and you MUST also refer to ../phy/rockchip-pcie-phy.txt
+  for changing the #phy-cells of phy node to support it)
+- rockchip,max-outbound-regions: Maximum number of outbound regions
+
+Optional Property:
+- num-lanes: number of lanes to use
+- max-functions: Maximum number of functions that can be configured (default 1).
+
+pcie0-ep: pcie@f8000000 {
+       compatible = "rockchip,rk3399-pcie-ep";
+       #address-cells = <3>;
+       #size-cells = <2>;
+       rockchip,max-outbound-regions = <16>;
+       clocks = <&cru ACLK_PCIE>, <&cru ACLK_PERF_PCIE>,
+                <&cru PCLK_PCIE>, <&cru SCLK_PCIE_PM>;
+       clock-names = "aclk", "aclk-perf",
+                     "hclk", "pm";
+       max-functions = /bits/ 8 <8>;
+       num-lanes = <4>;
+       reg = <0x0 0xfd000000 0x0 0x1000000>, <0x0 0x80000000 0x0 0x20000>;
+       reg-names = "apb-base", "mem-base";
+       resets = <&cru SRST_PCIE_CORE>, <&cru SRST_PCIE_MGMT>,
+                <&cru SRST_PCIE_MGMT_STICKY>, <&cru SRST_PCIE_PIPE> ,
+                <&cru SRST_PCIE_PM>, <&cru SRST_P_PCIE>, <&cru SRST_A_PCIE>;
+       reset-names = "core", "mgmt", "mgmt-sticky", "pipe",
+                     "pm", "pclk", "aclk";
+       phys = <&pcie_phy 0>, <&pcie_phy 1>, <&pcie_phy 2>, <&pcie_phy 3>;
+       phy-names = "pcie-phy-0", "pcie-phy-1", "pcie-phy-2", "pcie-phy-3";
+       pinctrl-names = "default";
+       pinctrl-0 = <&pcie_clkreq>;
+};
index 0395ad608fc945eeeebb2e9d96e1bce38668c045..d70a026ccb84c45bae1abd2c0eca12224afa8d24 100644 (file)
@@ -10947,8 +10947,8 @@ M:      Shawn Lin <shawn.lin@rock-chips.com>
 L:     linux-pci@vger.kernel.org
 L:     linux-rockchip@lists.infradead.org
 S:     Maintained
-F:     Documentation/devicetree/bindings/pci/rockchip-pcie.txt
-F:     drivers/pci/host/pcie-rockchip.c
+F:     Documentation/devicetree/bindings/pci/rockchip-pcie*
+F:     drivers/pci/host/pcie-rockchip*
 
 PCI DRIVER FOR V3 SEMICONDUCTOR V360EPC
 M:     Linus Walleij <linus.walleij@linaro.org>
index ecf613761e78ee23168a73d9bc0e23a60bbb42cd..17ea72b1b3898bd0d9ccfb4e0f71128d0bcf1ed6 100644 (file)
@@ -78,7 +78,8 @@ CONFIG_PCIE_ARMADA_8K=y
 CONFIG_PCI_AARDVARK=y
 CONFIG_PCI_TEGRA=y
 CONFIG_PCIE_RCAR=y
-CONFIG_PCIE_ROCKCHIP=m
+CONFIG_PCIE_ROCKCHIP=y
+CONFIG_PCIE_ROCKCHIP_HOST=m
 CONFIG_PCI_HOST_GENERIC=y
 CONFIG_PCI_XGENE=y
 CONFIG_PCI_HOST_THUNDER_PEM=y
index 746eaf8a6e8fac2ae44bc4e9a35f956d603f2540..2593c8373f8f23c46a716b0d4b761fd44cfc45e3 100644 (file)
@@ -179,16 +179,33 @@ config PCI_HOST_THUNDER_ECAM
          Say Y here if you want ECAM support for CN88XX-Pass-1.x Cavium Thunder SoCs.
 
 config PCIE_ROCKCHIP
-       tristate "Rockchip PCIe controller"
+       bool
+       depends on PCI
+
+config PCIE_ROCKCHIP_HOST
+       tristate "Rockchip PCIe host controller"
        depends on ARCH_ROCKCHIP || COMPILE_TEST
        depends on OF
        depends on PCI_MSI_IRQ_DOMAIN
        select MFD_SYSCON
+       select PCIE_ROCKCHIP
        help
          Say Y here if you want internal PCI support on Rockchip SoC.
          There is 1 internal PCIe port available to support GEN2 with
          4 slots.
 
+config PCIE_ROCKCHIP_EP
+       bool "Rockchip PCIe endpoint controller"
+       depends on ARCH_ROCKCHIP || COMPILE_TEST
+       depends on OF
+       depends on PCI_ENDPOINT
+       select MFD_SYSCON
+       select PCIE_ROCKCHIP
+       help
+         Say Y here if you want to support Rockchip PCIe controller in
+         endpoint mode on Rockchip SoC. There is 1 internal PCIe port
+         available to support GEN2 with 4 slots.
+
 config PCIE_MEDIATEK
        bool "MediaTek PCIe controller"
        depends on ARCH_MEDIATEK || COMPILE_TEST
index 3b105919086750b0db88e1ff4ecee11f5e37990c..11d21b026d373335a1315a83dd9d4f0a6da6a149 100644 (file)
@@ -20,6 +20,8 @@ obj-$(CONFIG_PCIE_IPROC_BCMA) += pcie-iproc-bcma.o
 obj-$(CONFIG_PCIE_ALTERA) += pcie-altera.o
 obj-$(CONFIG_PCIE_ALTERA_MSI) += pcie-altera-msi.o
 obj-$(CONFIG_PCIE_ROCKCHIP) += pcie-rockchip.o
+obj-$(CONFIG_PCIE_ROCKCHIP_EP) += pcie-rockchip-ep.o
+obj-$(CONFIG_PCIE_ROCKCHIP_HOST) += pcie-rockchip-host.o
 obj-$(CONFIG_PCIE_MEDIATEK) += pcie-mediatek.o
 obj-$(CONFIG_PCIE_TANGO_SMP8759) += pcie-tango.o
 obj-$(CONFIG_VMD) += vmd.o
diff --git a/drivers/pci/host/pcie-rockchip-ep.c b/drivers/pci/host/pcie-rockchip-ep.c
new file mode 100644 (file)
index 0000000..fc267a4
--- /dev/null
@@ -0,0 +1,642 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Rockchip AXI PCIe endpoint controller driver
+ *
+ * Copyright (c) 2018 Rockchip, Inc.
+ *
+ * Author: Shawn Lin <shawn.lin@rock-chips.com>
+ *         Simon Xue <xxm@rock-chips.com>
+ */
+
+#include <linux/configfs.h>
+#include <linux/delay.h>
+#include <linux/kernel.h>
+#include <linux/of.h>
+#include <linux/pci-epc.h>
+#include <linux/platform_device.h>
+#include <linux/pci-epf.h>
+#include <linux/sizes.h>
+
+#include "pcie-rockchip.h"
+
+/**
+ * struct rockchip_pcie_ep - private data for PCIe endpoint controller driver
+ * @rockchip: Rockchip PCIe controller
+ * @max_regions: maximum number of regions supported by hardware
+ * @ob_region_map: bitmask of mapped outbound regions
+ * @ob_addr: base addresses in the AXI bus where the outbound regions start
+ * @irq_phys_addr: base address on the AXI bus where the MSI/legacy IRQ
+ *                dedicated outbound regions is mapped.
+ * @irq_cpu_addr: base address in the CPU space where a write access triggers
+ *               the sending of a memory write (MSI) / normal message (legacy
+ *               IRQ) TLP through the PCIe bus.
+ * @irq_pci_addr: used to save the current mapping of the MSI/legacy IRQ
+ *               dedicated outbound region.
+ * @irq_pci_fn: the latest PCI function that has updated the mapping of
+ *             the MSI/legacy IRQ dedicated outbound region.
+ * @irq_pending: bitmask of asserted legacy IRQs.
+ */
+struct rockchip_pcie_ep {
+       struct rockchip_pcie    rockchip;
+       struct pci_epc          *epc;
+       u32                     max_regions;
+       unsigned long           ob_region_map;
+       phys_addr_t             *ob_addr;
+       phys_addr_t             irq_phys_addr;
+       void __iomem            *irq_cpu_addr;
+       u64                     irq_pci_addr;
+       u8                      irq_pci_fn;
+       u8                      irq_pending;
+};
+
+static void rockchip_pcie_clear_ep_ob_atu(struct rockchip_pcie *rockchip,
+                                         u32 region)
+{
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(region));
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(region));
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_DESC0(region));
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_DESC1(region));
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR0(region));
+       rockchip_pcie_write(rockchip, 0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(region));
+}
+
+static void rockchip_pcie_prog_ep_ob_atu(struct rockchip_pcie *rockchip, u8 fn,
+                                        u32 r, u32 type, u64 cpu_addr,
+                                        u64 pci_addr, size_t size)
+{
+       u64 sz = 1ULL << fls64(size - 1);
+       int num_pass_bits = ilog2(sz);
+       u32 addr0, addr1, desc0, desc1;
+       bool is_nor_msg = (type == AXI_WRAPPER_NOR_MSG);
+
+       /* The minimal region size is 1MB */
+       if (num_pass_bits < 8)
+               num_pass_bits = 8;
+
+       cpu_addr -= rockchip->mem_res->start;
+       addr0 = ((is_nor_msg ? 0x10 : (num_pass_bits - 1)) &
+               PCIE_CORE_OB_REGION_ADDR0_NUM_BITS) |
+               (lower_32_bits(cpu_addr) & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR);
+       addr1 = upper_32_bits(is_nor_msg ? cpu_addr : pci_addr);
+       desc0 = ROCKCHIP_PCIE_AT_OB_REGION_DESC0_DEVFN(fn) | type;
+       desc1 = 0;
+
+       if (is_nor_msg) {
+               rockchip_pcie_write(rockchip, 0,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(r));
+               rockchip_pcie_write(rockchip, 0,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(r));
+               rockchip_pcie_write(rockchip, desc0,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_DESC0(r));
+               rockchip_pcie_write(rockchip, desc1,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_DESC1(r));
+       } else {
+               /* PCI bus address region */
+               rockchip_pcie_write(rockchip, addr0,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(r));
+               rockchip_pcie_write(rockchip, addr1,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(r));
+               rockchip_pcie_write(rockchip, desc0,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_DESC0(r));
+               rockchip_pcie_write(rockchip, desc1,
+                                   ROCKCHIP_PCIE_AT_OB_REGION_DESC1(r));
+
+               addr0 =
+                   ((num_pass_bits - 1) & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS) |
+                   (lower_32_bits(cpu_addr) &
+                    PCIE_CORE_OB_REGION_ADDR0_LO_ADDR);
+               addr1 = upper_32_bits(cpu_addr);
+       }
+
+       /* CPU bus address region */
+       rockchip_pcie_write(rockchip, addr0,
+                           ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR0(r));
+       rockchip_pcie_write(rockchip, addr1,
+                           ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(r));
+}
+
+static int rockchip_pcie_ep_write_header(struct pci_epc *epc, u8 fn,
+                                        struct pci_epf_header *hdr)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+
+       /* All functions share the same vendor ID with function 0 */
+       if (fn == 0) {
+               u32 vid_regs = (hdr->vendorid & GENMASK(15, 0)) |
+                              (hdr->subsys_vendor_id & GENMASK(31, 16)) << 16;
+
+               rockchip_pcie_write(rockchip, vid_regs,
+                                   PCIE_CORE_CONFIG_VENDOR);
+       }
+
+       rockchip_pcie_write(rockchip, hdr->deviceid << 16,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) + PCI_VENDOR_ID);
+
+       rockchip_pcie_write(rockchip,
+                           hdr->revid |
+                           hdr->progif_code << 8 |
+                           hdr->subclass_code << 16 |
+                           hdr->baseclass_code << 24,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) + PCI_REVISION_ID);
+       rockchip_pcie_write(rockchip, hdr->cache_line_size,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                           PCI_CACHE_LINE_SIZE);
+       rockchip_pcie_write(rockchip, hdr->subsys_id << 16,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                           PCI_SUBSYSTEM_VENDOR_ID);
+       rockchip_pcie_write(rockchip, hdr->interrupt_pin << 8,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                           PCI_INTERRUPT_LINE);
+
+       return 0;
+}
+
+static int rockchip_pcie_ep_set_bar(struct pci_epc *epc, u8 fn,
+                                   struct pci_epf_bar *epf_bar)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       dma_addr_t bar_phys = epf_bar->phys_addr;
+       enum pci_barno bar = epf_bar->barno;
+       int flags = epf_bar->flags;
+       u32 addr0, addr1, reg, cfg, b, aperture, ctrl;
+       u64 sz;
+
+       /* BAR size is 2^(aperture + 7) */
+       sz = max_t(size_t, epf_bar->size, MIN_EP_APERTURE);
+
+       /*
+        * roundup_pow_of_two() returns an unsigned long, which is not suited
+        * for 64bit values.
+        */
+       sz = 1ULL << fls64(sz - 1);
+       aperture = ilog2(sz) - 7; /* 128B -> 0, 256B -> 1, 512B -> 2, ... */
+
+       if ((flags & PCI_BASE_ADDRESS_SPACE) == PCI_BASE_ADDRESS_SPACE_IO) {
+               ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_IO_32BITS;
+       } else {
+               bool is_prefetch = !!(flags & PCI_BASE_ADDRESS_MEM_PREFETCH);
+               bool is_64bits = sz > SZ_2G;
+
+               if (is_64bits && (bar & 1))
+                       return -EINVAL;
+
+               if (is_64bits && is_prefetch)
+                       ctrl =
+                           ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_64BITS;
+               else if (is_prefetch)
+                       ctrl =
+                           ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_32BITS;
+               else if (is_64bits)
+                       ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_64BITS;
+               else
+                       ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_32BITS;
+       }
+
+       if (bar < BAR_4) {
+               reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG0(fn);
+               b = bar;
+       } else {
+               reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG1(fn);
+               b = bar - BAR_4;
+       }
+
+       addr0 = lower_32_bits(bar_phys);
+       addr1 = upper_32_bits(bar_phys);
+
+       cfg = rockchip_pcie_read(rockchip, reg);
+       cfg &= ~(ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b) |
+                ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b));
+       cfg |= (ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE(b, aperture) |
+               ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL(b, ctrl));
+
+       rockchip_pcie_write(rockchip, cfg, reg);
+       rockchip_pcie_write(rockchip, addr0,
+                           ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR0(fn, bar));
+       rockchip_pcie_write(rockchip, addr1,
+                           ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar));
+
+       return 0;
+}
+
+static void rockchip_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn,
+                                      struct pci_epf_bar *epf_bar)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u32 reg, cfg, b, ctrl;
+       enum pci_barno bar = epf_bar->barno;
+
+       if (bar < BAR_4) {
+               reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG0(fn);
+               b = bar;
+       } else {
+               reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG1(fn);
+               b = bar - BAR_4;
+       }
+
+       ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_DISABLED;
+       cfg = rockchip_pcie_read(rockchip, reg);
+       cfg &= ~(ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b) |
+                ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b));
+       cfg |= ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL(b, ctrl);
+
+       rockchip_pcie_write(rockchip, cfg, reg);
+       rockchip_pcie_write(rockchip, 0x0,
+                           ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR0(fn, bar));
+       rockchip_pcie_write(rockchip, 0x0,
+                           ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar));
+}
+
+static int rockchip_pcie_ep_map_addr(struct pci_epc *epc, u8 fn,
+                                    phys_addr_t addr, u64 pci_addr,
+                                    size_t size)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *pcie = &ep->rockchip;
+       u32 r;
+
+       r = find_first_zero_bit(&ep->ob_region_map,
+                               sizeof(ep->ob_region_map) * BITS_PER_LONG);
+       /*
+        * Region 0 is reserved for configuration space and shouldn't
+        * be used elsewhere per TRM, so leave it out.
+        */
+       if (r >= ep->max_regions - 1) {
+               dev_err(&epc->dev, "no free outbound region\n");
+               return -EINVAL;
+       }
+
+       rockchip_pcie_prog_ep_ob_atu(pcie, fn, r, AXI_WRAPPER_MEM_WRITE, addr,
+                                    pci_addr, size);
+
+       set_bit(r, &ep->ob_region_map);
+       ep->ob_addr[r] = addr;
+
+       return 0;
+}
+
+static void rockchip_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn,
+                                       phys_addr_t addr)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u32 r;
+
+       for (r = 0; r < ep->max_regions - 1; r++)
+               if (ep->ob_addr[r] == addr)
+                       break;
+
+       /*
+        * Region 0 is reserved for configuration space and shouldn't
+        * be used elsewhere per TRM, so leave it out.
+        */
+       if (r == ep->max_regions - 1)
+               return;
+
+       rockchip_pcie_clear_ep_ob_atu(rockchip, r);
+
+       ep->ob_addr[r] = 0;
+       clear_bit(r, &ep->ob_region_map);
+}
+
+static int rockchip_pcie_ep_set_msi(struct pci_epc *epc, u8 fn,
+                                   u8 multi_msg_cap)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u16 flags;
+
+       flags = rockchip_pcie_read(rockchip,
+                                  ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                  ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
+       flags &= ~ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_MASK;
+       flags |=
+          ((multi_msg_cap << 1) <<  ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_OFFSET) |
+          PCI_MSI_FLAGS_64BIT;
+       flags &= ~ROCKCHIP_PCIE_EP_MSI_CTRL_MASK_MSI_CAP;
+       rockchip_pcie_write(rockchip, flags,
+                           ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                           ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
+       return 0;
+}
+
+static int rockchip_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u16 flags;
+
+       flags = rockchip_pcie_read(rockchip,
+                                  ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                  ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
+       if (!(flags & ROCKCHIP_PCIE_EP_MSI_CTRL_ME))
+               return -EINVAL;
+
+       return ((flags & ROCKCHIP_PCIE_EP_MSI_CTRL_MME_MASK) >>
+                       ROCKCHIP_PCIE_EP_MSI_CTRL_MME_OFFSET);
+}
+
+static void rockchip_pcie_ep_assert_intx(struct rockchip_pcie_ep *ep, u8 fn,
+                                        u8 intx, bool is_asserted)
+{
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u32 r = ep->max_regions - 1;
+       u32 offset;
+       u16 status;
+       u8 msg_code;
+
+       if (unlikely(ep->irq_pci_addr != ROCKCHIP_PCIE_EP_PCI_LEGACY_IRQ_ADDR ||
+                    ep->irq_pci_fn != fn)) {
+               rockchip_pcie_prog_ep_ob_atu(rockchip, fn, r,
+                                            AXI_WRAPPER_NOR_MSG,
+                                            ep->irq_phys_addr, 0, 0);
+               ep->irq_pci_addr = ROCKCHIP_PCIE_EP_PCI_LEGACY_IRQ_ADDR;
+               ep->irq_pci_fn = fn;
+       }
+
+       intx &= 3;
+       if (is_asserted) {
+               ep->irq_pending |= BIT(intx);
+               msg_code = ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTA + intx;
+       } else {
+               ep->irq_pending &= ~BIT(intx);
+               msg_code = ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTA + intx;
+       }
+
+       status = rockchip_pcie_read(rockchip,
+                                   ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                   ROCKCHIP_PCIE_EP_CMD_STATUS);
+       status &= ROCKCHIP_PCIE_EP_CMD_STATUS_IS;
+
+       if ((status != 0) ^ (ep->irq_pending != 0)) {
+               status ^= ROCKCHIP_PCIE_EP_CMD_STATUS_IS;
+               rockchip_pcie_write(rockchip, status,
+                                   ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                   ROCKCHIP_PCIE_EP_CMD_STATUS);
+       }
+
+       offset =
+          ROCKCHIP_PCIE_MSG_ROUTING(ROCKCHIP_PCIE_MSG_ROUTING_LOCAL_INTX) |
+          ROCKCHIP_PCIE_MSG_CODE(msg_code) | ROCKCHIP_PCIE_MSG_NO_DATA;
+       writel(0, ep->irq_cpu_addr + offset);
+}
+
+static int rockchip_pcie_ep_send_legacy_irq(struct rockchip_pcie_ep *ep, u8 fn,
+                                           u8 intx)
+{
+       u16 cmd;
+
+       cmd = rockchip_pcie_read(&ep->rockchip,
+                                ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                ROCKCHIP_PCIE_EP_CMD_STATUS);
+
+       if (cmd & PCI_COMMAND_INTX_DISABLE)
+               return -EINVAL;
+
+       /*
+        * Should add some delay between toggling INTx per TRM vaguely saying
+        * it depends on some cycles of the AHB bus clock to function it. So
+        * add sufficient 1ms here.
+        */
+       rockchip_pcie_ep_assert_intx(ep, fn, intx, true);
+       mdelay(1);
+       rockchip_pcie_ep_assert_intx(ep, fn, intx, false);
+       return 0;
+}
+
+static int rockchip_pcie_ep_send_msi_irq(struct rockchip_pcie_ep *ep, u8 fn,
+                                        u8 interrupt_num)
+{
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       u16 flags, mme, data, data_mask;
+       u8 msi_count;
+       u64 pci_addr, pci_addr_mask = 0xff;
+
+       /* Check MSI enable bit */
+       flags = rockchip_pcie_read(&ep->rockchip,
+                                  ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                  ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
+       if (!(flags & ROCKCHIP_PCIE_EP_MSI_CTRL_ME))
+               return -EINVAL;
+
+       /* Get MSI numbers from MME */
+       mme = ((flags & ROCKCHIP_PCIE_EP_MSI_CTRL_MME_MASK) >>
+                       ROCKCHIP_PCIE_EP_MSI_CTRL_MME_OFFSET);
+       msi_count = 1 << mme;
+       if (!interrupt_num || interrupt_num > msi_count)
+               return -EINVAL;
+
+       /* Set MSI private data */
+       data_mask = msi_count - 1;
+       data = rockchip_pcie_read(rockchip,
+                                 ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                 ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
+                                 PCI_MSI_DATA_64);
+       data = (data & ~data_mask) | ((interrupt_num - 1) & data_mask);
+
+       /* Get MSI PCI address */
+       pci_addr = rockchip_pcie_read(rockchip,
+                                     ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                     ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
+                                     PCI_MSI_ADDRESS_HI);
+       pci_addr <<= 32;
+       pci_addr |= rockchip_pcie_read(rockchip,
+                                      ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
+                                      ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
+                                      PCI_MSI_ADDRESS_LO);
+       pci_addr &= GENMASK_ULL(63, 2);
+
+       /* Set the outbound region if needed. */
+       if (unlikely(ep->irq_pci_addr != (pci_addr & ~pci_addr_mask) ||
+                    ep->irq_pci_fn != fn)) {
+               rockchip_pcie_prog_ep_ob_atu(rockchip, fn, ep->max_regions - 1,
+                                            AXI_WRAPPER_MEM_WRITE,
+                                            ep->irq_phys_addr,
+                                            pci_addr & ~pci_addr_mask,
+                                            pci_addr_mask + 1);
+               ep->irq_pci_addr = (pci_addr & ~pci_addr_mask);
+               ep->irq_pci_fn = fn;
+       }
+
+       writew(data, ep->irq_cpu_addr + (pci_addr & pci_addr_mask));
+       return 0;
+}
+
+static int rockchip_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn,
+                                     enum pci_epc_irq_type type,
+                                     u8 interrupt_num)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+
+       switch (type) {
+       case PCI_EPC_IRQ_LEGACY:
+               return rockchip_pcie_ep_send_legacy_irq(ep, fn, 0);
+       case PCI_EPC_IRQ_MSI:
+               return rockchip_pcie_ep_send_msi_irq(ep, fn, interrupt_num);
+       default:
+               return -EINVAL;
+       }
+}
+
+static int rockchip_pcie_ep_start(struct pci_epc *epc)
+{
+       struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
+       struct rockchip_pcie *rockchip = &ep->rockchip;
+       struct pci_epf *epf;
+       u32 cfg;
+
+       cfg = BIT(0);
+       list_for_each_entry(epf, &epc->pci_epf, list)
+               cfg |= BIT(epf->func_no);
+
+       rockchip_pcie_write(rockchip, cfg, PCIE_CORE_PHY_FUNC_CFG);
+
+       list_for_each_entry(epf, &epc->pci_epf, list)
+               pci_epf_linkup(epf);
+
+       return 0;
+}
+
+static const struct pci_epc_ops rockchip_pcie_epc_ops = {
+       .write_header   = rockchip_pcie_ep_write_header,
+       .set_bar        = rockchip_pcie_ep_set_bar,
+       .clear_bar      = rockchip_pcie_ep_clear_bar,
+       .map_addr       = rockchip_pcie_ep_map_addr,
+       .unmap_addr     = rockchip_pcie_ep_unmap_addr,
+       .set_msi        = rockchip_pcie_ep_set_msi,
+       .get_msi        = rockchip_pcie_ep_get_msi,
+       .raise_irq      = rockchip_pcie_ep_raise_irq,
+       .start          = rockchip_pcie_ep_start,
+};
+
+static int rockchip_pcie_parse_ep_dt(struct rockchip_pcie *rockchip,
+                                    struct rockchip_pcie_ep *ep)
+{
+       struct device *dev = rockchip->dev;
+       int err;
+
+       err = rockchip_pcie_parse_dt(rockchip);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_get_phys(rockchip);
+       if (err)
+               return err;
+
+       err = of_property_read_u32(dev->of_node,
+                                  "rockchip,max-outbound-regions",
+                                  &ep->max_regions);
+       if (err < 0 || ep->max_regions > MAX_REGION_LIMIT)
+               ep->max_regions = MAX_REGION_LIMIT;
+
+       err = of_property_read_u8(dev->of_node, "max-functions",
+                                 &ep->epc->max_functions);
+       if (err < 0)
+               ep->epc->max_functions = 1;
+
+       return 0;
+}
+
+static const struct of_device_id rockchip_pcie_ep_of_match[] = {
+       { .compatible = "rockchip,rk3399-pcie-ep"},
+       {},
+};
+
+static int rockchip_pcie_ep_probe(struct platform_device *pdev)
+{
+       struct device *dev = &pdev->dev;
+       struct rockchip_pcie_ep *ep;
+       struct rockchip_pcie *rockchip;
+       struct pci_epc *epc;
+       size_t max_regions;
+       int err;
+
+       ep = devm_kzalloc(dev, sizeof(*ep), GFP_KERNEL);
+       if (!ep)
+               return -ENOMEM;
+
+       rockchip = &ep->rockchip;
+       rockchip->is_rc = false;
+       rockchip->dev = dev;
+
+       epc = devm_pci_epc_create(dev, &rockchip_pcie_epc_ops);
+       if (IS_ERR(epc)) {
+               dev_err(dev, "failed to create epc device\n");
+               return PTR_ERR(epc);
+       }
+
+       ep->epc = epc;
+       epc_set_drvdata(epc, ep);
+
+       err = rockchip_pcie_parse_ep_dt(rockchip, ep);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_enable_clocks(rockchip);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_init_port(rockchip);
+       if (err)
+               goto err_disable_clocks;
+
+       /* Establish the link automatically */
+       rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
+                           PCIE_CLIENT_CONFIG);
+
+       max_regions = ep->max_regions;
+       ep->ob_addr = devm_kzalloc(dev, max_regions * sizeof(*ep->ob_addr),
+                                  GFP_KERNEL);
+
+       if (!ep->ob_addr) {
+               err = -ENOMEM;
+               goto err_uninit_port;
+       }
+
+       /* Only enable function 0 by default */
+       rockchip_pcie_write(rockchip, BIT(0), PCIE_CORE_PHY_FUNC_CFG);
+
+       err = pci_epc_mem_init(epc, rockchip->mem_res->start,
+                              resource_size(rockchip->mem_res));
+       if (err < 0) {
+               dev_err(dev, "failed to initialize the memory space\n");
+               goto err_uninit_port;
+       }
+
+       ep->irq_cpu_addr = pci_epc_mem_alloc_addr(epc, &ep->irq_phys_addr,
+                                                 SZ_128K);
+       if (!ep->irq_cpu_addr) {
+               dev_err(dev, "failed to reserve memory space for MSI\n");
+               err = -ENOMEM;
+               goto err_epc_mem_exit;
+       }
+
+       ep->irq_pci_addr = ROCKCHIP_PCIE_EP_DUMMY_IRQ_ADDR;
+
+       return 0;
+err_epc_mem_exit:
+       pci_epc_mem_exit(epc);
+err_uninit_port:
+       rockchip_pcie_deinit_phys(rockchip);
+err_disable_clocks:
+       rockchip_pcie_disable_clocks(rockchip);
+       return err;
+}
+
+static struct platform_driver rockchip_pcie_ep_driver = {
+       .driver = {
+               .name = "rockchip-pcie-ep",
+               .of_match_table = rockchip_pcie_ep_of_match,
+       },
+       .probe = rockchip_pcie_ep_probe,
+};
+
+builtin_platform_driver(rockchip_pcie_ep_driver);
diff --git a/drivers/pci/host/pcie-rockchip-host.c b/drivers/pci/host/pcie-rockchip-host.c
new file mode 100644 (file)
index 0000000..1372d27
--- /dev/null
@@ -0,0 +1,1142 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Rockchip AXI PCIe host controller driver
+ *
+ * Copyright (c) 2016 Rockchip, Inc.
+ *
+ * Author: Shawn Lin <shawn.lin@rock-chips.com>
+ *         Wenrui Li <wenrui.li@rock-chips.com>
+ *
+ * Bits taken from Synopsys DesignWare Host controller driver and
+ * ARM PCI Host generic driver.
+ */
+
+#include <linux/bitrev.h>
+#include <linux/clk.h>
+#include <linux/delay.h>
+#include <linux/gpio/consumer.h>
+#include <linux/init.h>
+#include <linux/interrupt.h>
+#include <linux/iopoll.h>
+#include <linux/irq.h>
+#include <linux/irqchip/chained_irq.h>
+#include <linux/irqdomain.h>
+#include <linux/kernel.h>
+#include <linux/mfd/syscon.h>
+#include <linux/module.h>
+#include <linux/of_address.h>
+#include <linux/of_device.h>
+#include <linux/of_pci.h>
+#include <linux/of_platform.h>
+#include <linux/of_irq.h>
+#include <linux/pci.h>
+#include <linux/pci_ids.h>
+#include <linux/phy/phy.h>
+#include <linux/platform_device.h>
+#include <linux/reset.h>
+#include <linux/regmap.h>
+
+#include "../pci.h"
+#include "pcie-rockchip.h"
+
+static void rockchip_pcie_enable_bw_int(struct rockchip_pcie *rockchip)
+{
+       u32 status;
+
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
+       status |= (PCI_EXP_LNKCTL_LBMIE | PCI_EXP_LNKCTL_LABIE);
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
+}
+
+static void rockchip_pcie_clr_bw_int(struct rockchip_pcie *rockchip)
+{
+       u32 status;
+
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
+       status |= (PCI_EXP_LNKSTA_LBMS | PCI_EXP_LNKSTA_LABS) << 16;
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
+}
+
+static void rockchip_pcie_update_txcredit_mui(struct rockchip_pcie *rockchip)
+{
+       u32 val;
+
+       /* Update Tx credit maximum update interval */
+       val = rockchip_pcie_read(rockchip, PCIE_CORE_TXCREDIT_CFG1);
+       val &= ~PCIE_CORE_TXCREDIT_CFG1_MUI_MASK;
+       val |= PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(24000);       /* ns */
+       rockchip_pcie_write(rockchip, val, PCIE_CORE_TXCREDIT_CFG1);
+}
+
+static int rockchip_pcie_valid_device(struct rockchip_pcie *rockchip,
+                                     struct pci_bus *bus, int dev)
+{
+       /* access only one slot on each root port */
+       if (bus->number == rockchip->root_bus_nr && dev > 0)
+               return 0;
+
+       /*
+        * do not read more than one device on the bus directly attached
+        * to RC's downstream side.
+        */
+       if (bus->primary == rockchip->root_bus_nr && dev > 0)
+               return 0;
+
+       return 1;
+}
+
+static u8 rockchip_pcie_lane_map(struct rockchip_pcie *rockchip)
+{
+       u32 val;
+       u8 map;
+
+       if (rockchip->legacy_phy)
+               return GENMASK(MAX_LANE_NUM - 1, 0);
+
+       val = rockchip_pcie_read(rockchip, PCIE_CORE_LANE_MAP);
+       map = val & PCIE_CORE_LANE_MAP_MASK;
+
+       /* The link may be using a reverse-indexed mapping. */
+       if (val & PCIE_CORE_LANE_MAP_REVERSE)
+               map = bitrev8(map) >> 4;
+
+       return map;
+}
+
+static int rockchip_pcie_rd_own_conf(struct rockchip_pcie *rockchip,
+                                    int where, int size, u32 *val)
+{
+       void __iomem *addr;
+
+       addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + where;
+
+       if (!IS_ALIGNED((uintptr_t)addr, size)) {
+               *val = 0;
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+       }
+
+       if (size == 4) {
+               *val = readl(addr);
+       } else if (size == 2) {
+               *val = readw(addr);
+       } else if (size == 1) {
+               *val = readb(addr);
+       } else {
+               *val = 0;
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+       }
+       return PCIBIOS_SUCCESSFUL;
+}
+
+static int rockchip_pcie_wr_own_conf(struct rockchip_pcie *rockchip,
+                                    int where, int size, u32 val)
+{
+       u32 mask, tmp, offset;
+       void __iomem *addr;
+
+       offset = where & ~0x3;
+       addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + offset;
+
+       if (size == 4) {
+               writel(val, addr);
+               return PCIBIOS_SUCCESSFUL;
+       }
+
+       mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
+
+       /*
+        * N.B. This read/modify/write isn't safe in general because it can
+        * corrupt RW1C bits in adjacent registers.  But the hardware
+        * doesn't support smaller writes.
+        */
+       tmp = readl(addr) & mask;
+       tmp |= val << ((where & 0x3) * 8);
+       writel(tmp, addr);
+
+       return PCIBIOS_SUCCESSFUL;
+}
+
+static int rockchip_pcie_rd_other_conf(struct rockchip_pcie *rockchip,
+                                      struct pci_bus *bus, u32 devfn,
+                                      int where, int size, u32 *val)
+{
+       u32 busdev;
+
+       busdev = PCIE_ECAM_ADDR(bus->number, PCI_SLOT(devfn),
+                               PCI_FUNC(devfn), where);
+
+       if (!IS_ALIGNED(busdev, size)) {
+               *val = 0;
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+       }
+
+       if (bus->parent->number == rockchip->root_bus_nr)
+               rockchip_pcie_cfg_configuration_accesses(rockchip,
+                                               AXI_WRAPPER_TYPE0_CFG);
+       else
+               rockchip_pcie_cfg_configuration_accesses(rockchip,
+                                               AXI_WRAPPER_TYPE1_CFG);
+
+       if (size == 4) {
+               *val = readl(rockchip->reg_base + busdev);
+       } else if (size == 2) {
+               *val = readw(rockchip->reg_base + busdev);
+       } else if (size == 1) {
+               *val = readb(rockchip->reg_base + busdev);
+       } else {
+               *val = 0;
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+       }
+       return PCIBIOS_SUCCESSFUL;
+}
+
+static int rockchip_pcie_wr_other_conf(struct rockchip_pcie *rockchip,
+                                      struct pci_bus *bus, u32 devfn,
+                                      int where, int size, u32 val)
+{
+       u32 busdev;
+
+       busdev = PCIE_ECAM_ADDR(bus->number, PCI_SLOT(devfn),
+                               PCI_FUNC(devfn), where);
+       if (!IS_ALIGNED(busdev, size))
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+
+       if (bus->parent->number == rockchip->root_bus_nr)
+               rockchip_pcie_cfg_configuration_accesses(rockchip,
+                                               AXI_WRAPPER_TYPE0_CFG);
+       else
+               rockchip_pcie_cfg_configuration_accesses(rockchip,
+                                               AXI_WRAPPER_TYPE1_CFG);
+
+       if (size == 4)
+               writel(val, rockchip->reg_base + busdev);
+       else if (size == 2)
+               writew(val, rockchip->reg_base + busdev);
+       else if (size == 1)
+               writeb(val, rockchip->reg_base + busdev);
+       else
+               return PCIBIOS_BAD_REGISTER_NUMBER;
+
+       return PCIBIOS_SUCCESSFUL;
+}
+
+static int rockchip_pcie_rd_conf(struct pci_bus *bus, u32 devfn, int where,
+                                int size, u32 *val)
+{
+       struct rockchip_pcie *rockchip = bus->sysdata;
+
+       if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn))) {
+               *val = 0xffffffff;
+               return PCIBIOS_DEVICE_NOT_FOUND;
+       }
+
+       if (bus->number == rockchip->root_bus_nr)
+               return rockchip_pcie_rd_own_conf(rockchip, where, size, val);
+
+       return rockchip_pcie_rd_other_conf(rockchip, bus, devfn, where, size,
+                                          val);
+}
+
+static int rockchip_pcie_wr_conf(struct pci_bus *bus, u32 devfn,
+                                int where, int size, u32 val)
+{
+       struct rockchip_pcie *rockchip = bus->sysdata;
+
+       if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
+               return PCIBIOS_DEVICE_NOT_FOUND;
+
+       if (bus->number == rockchip->root_bus_nr)
+               return rockchip_pcie_wr_own_conf(rockchip, where, size, val);
+
+       return rockchip_pcie_wr_other_conf(rockchip, bus, devfn, where, size,
+                                          val);
+}
+
+static struct pci_ops rockchip_pcie_ops = {
+       .read = rockchip_pcie_rd_conf,
+       .write = rockchip_pcie_wr_conf,
+};
+
+static void rockchip_pcie_set_power_limit(struct rockchip_pcie *rockchip)
+{
+       int curr;
+       u32 status, scale, power;
+
+       if (IS_ERR(rockchip->vpcie3v3))
+               return;
+
+       /*
+        * Set RC's captured slot power limit and scale if
+        * vpcie3v3 available. The default values are both zero
+        * which means the software should set these two according
+        * to the actual power supply.
+        */
+       curr = regulator_get_current_limit(rockchip->vpcie3v3);
+       if (curr <= 0)
+               return;
+
+       scale = 3; /* 0.001x */
+       curr = curr / 1000; /* convert to mA */
+       power = (curr * 3300) / 1000; /* milliwatt */
+       while (power > PCIE_RC_CONFIG_DCR_CSPL_LIMIT) {
+               if (!scale) {
+                       dev_warn(rockchip->dev, "invalid power supply\n");
+                       return;
+               }
+               scale--;
+               power = power / 10;
+       }
+
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCR);
+       status |= (power << PCIE_RC_CONFIG_DCR_CSPL_SHIFT) |
+                 (scale << PCIE_RC_CONFIG_DCR_CPLS_SHIFT);
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCR);
+}
+
+/**
+ * rockchip_pcie_host_init_port - Initialize hardware
+ * @rockchip: PCIe port information
+ */
+static int rockchip_pcie_host_init_port(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       int err, i = MAX_LANE_NUM;
+       u32 status;
+
+       gpiod_set_value_cansleep(rockchip->ep_gpio, 0);
+
+       err = rockchip_pcie_init_port(rockchip);
+       if (err)
+               return err;
+
+       /* Fix the transmitted FTS count desired to exit from L0s. */
+       status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL_PLC1);
+       status = (status & ~PCIE_CORE_CTRL_PLC1_FTS_MASK) |
+                (PCIE_CORE_CTRL_PLC1_FTS_CNT << PCIE_CORE_CTRL_PLC1_FTS_SHIFT);
+       rockchip_pcie_write(rockchip, status, PCIE_CORE_CTRL_PLC1);
+
+       rockchip_pcie_set_power_limit(rockchip);
+
+       /* Set RC's clock architecture as common clock */
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
+       status |= PCI_EXP_LNKSTA_SLC << 16;
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
+
+       /* Set RC's RCB to 128 */
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
+       status |= PCI_EXP_LNKCTL_RCB;
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
+
+       /* Enable Gen1 training */
+       rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
+                           PCIE_CLIENT_CONFIG);
+
+       gpiod_set_value_cansleep(rockchip->ep_gpio, 1);
+
+       /* 500ms timeout value should be enough for Gen1/2 training */
+       err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_BASIC_STATUS1,
+                                status, PCIE_LINK_UP(status), 20,
+                                500 * USEC_PER_MSEC);
+       if (err) {
+               dev_err(dev, "PCIe link training gen1 timeout!\n");
+               goto err_power_off_phy;
+       }
+
+       if (rockchip->link_gen == 2) {
+               /*
+                * Enable retrain for gen2. This should be configured only after
+                * gen1 finished.
+                */
+               status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
+               status |= PCI_EXP_LNKCTL_RL;
+               rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
+
+               err = readl_poll_timeout(rockchip->apb_base + PCIE_CORE_CTRL,
+                                        status, PCIE_LINK_IS_GEN2(status), 20,
+                                        500 * USEC_PER_MSEC);
+               if (err)
+                       dev_dbg(dev, "PCIe link training gen2 timeout, fall back to gen1!\n");
+       }
+
+       /* Check the final link width from negotiated lane counter from MGMT */
+       status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL);
+       status = 0x1 << ((status & PCIE_CORE_PL_CONF_LANE_MASK) >>
+                         PCIE_CORE_PL_CONF_LANE_SHIFT);
+       dev_dbg(dev, "current link width is x%d\n", status);
+
+       /* Power off unused lane(s) */
+       rockchip->lanes_map = rockchip_pcie_lane_map(rockchip);
+       for (i = 0; i < MAX_LANE_NUM; i++) {
+               if (!(rockchip->lanes_map & BIT(i))) {
+                       dev_dbg(dev, "idling lane %d\n", i);
+                       phy_power_off(rockchip->phys[i]);
+               }
+       }
+
+       rockchip_pcie_write(rockchip, ROCKCHIP_VENDOR_ID,
+                           PCIE_CORE_CONFIG_VENDOR);
+       rockchip_pcie_write(rockchip,
+                           PCI_CLASS_BRIDGE_PCI << PCIE_RC_CONFIG_SCC_SHIFT,
+                           PCIE_RC_CONFIG_RID_CCR);
+
+       /* Clear THP cap's next cap pointer to remove L1 substate cap */
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_THP_CAP);
+       status &= ~PCIE_RC_CONFIG_THP_CAP_NEXT_MASK;
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_THP_CAP);
+
+       /* Clear L0s from RC's link cap */
+       if (of_property_read_bool(dev->of_node, "aspm-no-l0s")) {
+               status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LINK_CAP);
+               status &= ~PCIE_RC_CONFIG_LINK_CAP_L0S;
+               rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LINK_CAP);
+       }
+
+       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCSR);
+       status &= ~PCIE_RC_CONFIG_DCSR_MPS_MASK;
+       status |= PCIE_RC_CONFIG_DCSR_MPS_256;
+       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCSR);
+
+       return 0;
+err_power_off_phy:
+       while (i--)
+               phy_power_off(rockchip->phys[i]);
+       i = MAX_LANE_NUM;
+       while (i--)
+               phy_exit(rockchip->phys[i]);
+       return err;
+}
+
+static irqreturn_t rockchip_pcie_subsys_irq_handler(int irq, void *arg)
+{
+       struct rockchip_pcie *rockchip = arg;
+       struct device *dev = rockchip->dev;
+       u32 reg;
+       u32 sub_reg;
+
+       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
+       if (reg & PCIE_CLIENT_INT_LOCAL) {
+               dev_dbg(dev, "local interrupt received\n");
+               sub_reg = rockchip_pcie_read(rockchip, PCIE_CORE_INT_STATUS);
+               if (sub_reg & PCIE_CORE_INT_PRFPE)
+                       dev_dbg(dev, "parity error detected while reading from the PNP receive FIFO RAM\n");
+
+               if (sub_reg & PCIE_CORE_INT_CRFPE)
+                       dev_dbg(dev, "parity error detected while reading from the Completion Receive FIFO RAM\n");
+
+               if (sub_reg & PCIE_CORE_INT_RRPE)
+                       dev_dbg(dev, "parity error detected while reading from replay buffer RAM\n");
+
+               if (sub_reg & PCIE_CORE_INT_PRFO)
+                       dev_dbg(dev, "overflow occurred in the PNP receive FIFO\n");
+
+               if (sub_reg & PCIE_CORE_INT_CRFO)
+                       dev_dbg(dev, "overflow occurred in the completion receive FIFO\n");
+
+               if (sub_reg & PCIE_CORE_INT_RT)
+                       dev_dbg(dev, "replay timer timed out\n");
+
+               if (sub_reg & PCIE_CORE_INT_RTR)
+                       dev_dbg(dev, "replay timer rolled over after 4 transmissions of the same TLP\n");
+
+               if (sub_reg & PCIE_CORE_INT_PE)
+                       dev_dbg(dev, "phy error detected on receive side\n");
+
+               if (sub_reg & PCIE_CORE_INT_MTR)
+                       dev_dbg(dev, "malformed TLP received from the link\n");
+
+               if (sub_reg & PCIE_CORE_INT_UCR)
+                       dev_dbg(dev, "malformed TLP received from the link\n");
+
+               if (sub_reg & PCIE_CORE_INT_FCE)
+                       dev_dbg(dev, "an error was observed in the flow control advertisements from the other side\n");
+
+               if (sub_reg & PCIE_CORE_INT_CT)
+                       dev_dbg(dev, "a request timed out waiting for completion\n");
+
+               if (sub_reg & PCIE_CORE_INT_UTC)
+                       dev_dbg(dev, "unmapped TC error\n");
+
+               if (sub_reg & PCIE_CORE_INT_MMVC)
+                       dev_dbg(dev, "MSI mask register changes\n");
+
+               rockchip_pcie_write(rockchip, sub_reg, PCIE_CORE_INT_STATUS);
+       } else if (reg & PCIE_CLIENT_INT_PHY) {
+               dev_dbg(dev, "phy link changes\n");
+               rockchip_pcie_update_txcredit_mui(rockchip);
+               rockchip_pcie_clr_bw_int(rockchip);
+       }
+
+       rockchip_pcie_write(rockchip, reg & PCIE_CLIENT_INT_LOCAL,
+                           PCIE_CLIENT_INT_STATUS);
+
+       return IRQ_HANDLED;
+}
+
+static irqreturn_t rockchip_pcie_client_irq_handler(int irq, void *arg)
+{
+       struct rockchip_pcie *rockchip = arg;
+       struct device *dev = rockchip->dev;
+       u32 reg;
+
+       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
+       if (reg & PCIE_CLIENT_INT_LEGACY_DONE)
+               dev_dbg(dev, "legacy done interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_MSG)
+               dev_dbg(dev, "message done interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_HOT_RST)
+               dev_dbg(dev, "hot reset interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_DPA)
+               dev_dbg(dev, "dpa interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_FATAL_ERR)
+               dev_dbg(dev, "fatal error interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_NFATAL_ERR)
+               dev_dbg(dev, "no fatal error interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_CORR_ERR)
+               dev_dbg(dev, "correctable error interrupt received\n");
+
+       if (reg & PCIE_CLIENT_INT_PHY)
+               dev_dbg(dev, "phy interrupt received\n");
+
+       rockchip_pcie_write(rockchip, reg & (PCIE_CLIENT_INT_LEGACY_DONE |
+                             PCIE_CLIENT_INT_MSG | PCIE_CLIENT_INT_HOT_RST |
+                             PCIE_CLIENT_INT_DPA | PCIE_CLIENT_INT_FATAL_ERR |
+                             PCIE_CLIENT_INT_NFATAL_ERR |
+                             PCIE_CLIENT_INT_CORR_ERR |
+                             PCIE_CLIENT_INT_PHY),
+                  PCIE_CLIENT_INT_STATUS);
+
+       return IRQ_HANDLED;
+}
+
+static void rockchip_pcie_legacy_int_handler(struct irq_desc *desc)
+{
+       struct irq_chip *chip = irq_desc_get_chip(desc);
+       struct rockchip_pcie *rockchip = irq_desc_get_handler_data(desc);
+       struct device *dev = rockchip->dev;
+       u32 reg;
+       u32 hwirq;
+       u32 virq;
+
+       chained_irq_enter(chip, desc);
+
+       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
+       reg = (reg & PCIE_CLIENT_INTR_MASK) >> PCIE_CLIENT_INTR_SHIFT;
+
+       while (reg) {
+               hwirq = ffs(reg) - 1;
+               reg &= ~BIT(hwirq);
+
+               virq = irq_find_mapping(rockchip->irq_domain, hwirq);
+               if (virq)
+                       generic_handle_irq(virq);
+               else
+                       dev_err(dev, "unexpected IRQ, INT%d\n", hwirq);
+       }
+
+       chained_irq_exit(chip, desc);
+}
+
+static int rockchip_pcie_setup_irq(struct rockchip_pcie *rockchip)
+{
+       int irq, err;
+       struct device *dev = rockchip->dev;
+       struct platform_device *pdev = to_platform_device(dev);
+
+       irq = platform_get_irq_byname(pdev, "sys");
+       if (irq < 0) {
+               dev_err(dev, "missing sys IRQ resource\n");
+               return irq;
+       }
+
+       err = devm_request_irq(dev, irq, rockchip_pcie_subsys_irq_handler,
+                              IRQF_SHARED, "pcie-sys", rockchip);
+       if (err) {
+               dev_err(dev, "failed to request PCIe subsystem IRQ\n");
+               return err;
+       }
+
+       irq = platform_get_irq_byname(pdev, "legacy");
+       if (irq < 0) {
+               dev_err(dev, "missing legacy IRQ resource\n");
+               return irq;
+       }
+
+       irq_set_chained_handler_and_data(irq,
+                                        rockchip_pcie_legacy_int_handler,
+                                        rockchip);
+
+       irq = platform_get_irq_byname(pdev, "client");
+       if (irq < 0) {
+               dev_err(dev, "missing client IRQ resource\n");
+               return irq;
+       }
+
+       err = devm_request_irq(dev, irq, rockchip_pcie_client_irq_handler,
+                              IRQF_SHARED, "pcie-client", rockchip);
+       if (err) {
+               dev_err(dev, "failed to request PCIe client IRQ\n");
+               return err;
+       }
+
+       return 0;
+}
+
+/**
+ * rockchip_pcie_parse_host_dt - Parse Device Tree
+ * @rockchip: PCIe port information
+ *
+ * Return: '0' on success and error value on failure
+ */
+static int rockchip_pcie_parse_host_dt(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       int err;
+
+       err = rockchip_pcie_parse_dt(rockchip);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_setup_irq(rockchip);
+       if (err)
+               return err;
+
+       rockchip->vpcie12v = devm_regulator_get_optional(dev, "vpcie12v");
+       if (IS_ERR(rockchip->vpcie12v)) {
+               if (PTR_ERR(rockchip->vpcie12v) == -EPROBE_DEFER)
+                       return -EPROBE_DEFER;
+               dev_info(dev, "no vpcie12v regulator found\n");
+       }
+
+       rockchip->vpcie3v3 = devm_regulator_get_optional(dev, "vpcie3v3");
+       if (IS_ERR(rockchip->vpcie3v3)) {
+               if (PTR_ERR(rockchip->vpcie3v3) == -EPROBE_DEFER)
+                       return -EPROBE_DEFER;
+               dev_info(dev, "no vpcie3v3 regulator found\n");
+       }
+
+       rockchip->vpcie1v8 = devm_regulator_get_optional(dev, "vpcie1v8");
+       if (IS_ERR(rockchip->vpcie1v8)) {
+               if (PTR_ERR(rockchip->vpcie1v8) == -EPROBE_DEFER)
+                       return -EPROBE_DEFER;
+               dev_info(dev, "no vpcie1v8 regulator found\n");
+       }
+
+       rockchip->vpcie0v9 = devm_regulator_get_optional(dev, "vpcie0v9");
+       if (IS_ERR(rockchip->vpcie0v9)) {
+               if (PTR_ERR(rockchip->vpcie0v9) == -EPROBE_DEFER)
+                       return -EPROBE_DEFER;
+               dev_info(dev, "no vpcie0v9 regulator found\n");
+       }
+
+       return 0;
+}
+
+static int rockchip_pcie_set_vpcie(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       int err;
+
+       if (!IS_ERR(rockchip->vpcie12v)) {
+               err = regulator_enable(rockchip->vpcie12v);
+               if (err) {
+                       dev_err(dev, "fail to enable vpcie12v regulator\n");
+                       goto err_out;
+               }
+       }
+
+       if (!IS_ERR(rockchip->vpcie3v3)) {
+               err = regulator_enable(rockchip->vpcie3v3);
+               if (err) {
+                       dev_err(dev, "fail to enable vpcie3v3 regulator\n");
+                       goto err_disable_12v;
+               }
+       }
+
+       if (!IS_ERR(rockchip->vpcie1v8)) {
+               err = regulator_enable(rockchip->vpcie1v8);
+               if (err) {
+                       dev_err(dev, "fail to enable vpcie1v8 regulator\n");
+                       goto err_disable_3v3;
+               }
+       }
+
+       if (!IS_ERR(rockchip->vpcie0v9)) {
+               err = regulator_enable(rockchip->vpcie0v9);
+               if (err) {
+                       dev_err(dev, "fail to enable vpcie0v9 regulator\n");
+                       goto err_disable_1v8;
+               }
+       }
+
+       return 0;
+
+err_disable_1v8:
+       if (!IS_ERR(rockchip->vpcie1v8))
+               regulator_disable(rockchip->vpcie1v8);
+err_disable_3v3:
+       if (!IS_ERR(rockchip->vpcie3v3))
+               regulator_disable(rockchip->vpcie3v3);
+err_disable_12v:
+       if (!IS_ERR(rockchip->vpcie12v))
+               regulator_disable(rockchip->vpcie12v);
+err_out:
+       return err;
+}
+
+static void rockchip_pcie_enable_interrupts(struct rockchip_pcie *rockchip)
+{
+       rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) &
+                           (~PCIE_CLIENT_INT_CLI), PCIE_CLIENT_INT_MASK);
+       rockchip_pcie_write(rockchip, (u32)(~PCIE_CORE_INT),
+                           PCIE_CORE_INT_MASK);
+
+       rockchip_pcie_enable_bw_int(rockchip);
+}
+
+static int rockchip_pcie_intx_map(struct irq_domain *domain, unsigned int irq,
+                                 irq_hw_number_t hwirq)
+{
+       irq_set_chip_and_handler(irq, &dummy_irq_chip, handle_simple_irq);
+       irq_set_chip_data(irq, domain->host_data);
+
+       return 0;
+}
+
+static const struct irq_domain_ops intx_domain_ops = {
+       .map = rockchip_pcie_intx_map,
+};
+
+static int rockchip_pcie_init_irq_domain(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       struct device_node *intc = of_get_next_child(dev->of_node, NULL);
+
+       if (!intc) {
+               dev_err(dev, "missing child interrupt-controller node\n");
+               return -EINVAL;
+       }
+
+       rockchip->irq_domain = irq_domain_add_linear(intc, PCI_NUM_INTX,
+                                                   &intx_domain_ops, rockchip);
+       if (!rockchip->irq_domain) {
+               dev_err(dev, "failed to get a INTx IRQ domain\n");
+               return -EINVAL;
+       }
+
+       return 0;
+}
+
+static int rockchip_pcie_prog_ob_atu(struct rockchip_pcie *rockchip,
+                                    int region_no, int type, u8 num_pass_bits,
+                                    u32 lower_addr, u32 upper_addr)
+{
+       u32 ob_addr_0;
+       u32 ob_addr_1;
+       u32 ob_desc_0;
+       u32 aw_offset;
+
+       if (region_no >= MAX_AXI_WRAPPER_REGION_NUM)
+               return -EINVAL;
+       if (num_pass_bits + 1 < 8)
+               return -EINVAL;
+       if (num_pass_bits > 63)
+               return -EINVAL;
+       if (region_no == 0) {
+               if (AXI_REGION_0_SIZE < (2ULL << num_pass_bits))
+                       return -EINVAL;
+       }
+       if (region_no != 0) {
+               if (AXI_REGION_SIZE < (2ULL << num_pass_bits))
+                       return -EINVAL;
+       }
+
+       aw_offset = (region_no << OB_REG_SIZE_SHIFT);
+
+       ob_addr_0 = num_pass_bits & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS;
+       ob_addr_0 |= lower_addr & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR;
+       ob_addr_1 = upper_addr;
+       ob_desc_0 = (1 << 23 | type);
+
+       rockchip_pcie_write(rockchip, ob_addr_0,
+                           PCIE_CORE_OB_REGION_ADDR0 + aw_offset);
+       rockchip_pcie_write(rockchip, ob_addr_1,
+                           PCIE_CORE_OB_REGION_ADDR1 + aw_offset);
+       rockchip_pcie_write(rockchip, ob_desc_0,
+                           PCIE_CORE_OB_REGION_DESC0 + aw_offset);
+       rockchip_pcie_write(rockchip, 0,
+                           PCIE_CORE_OB_REGION_DESC1 + aw_offset);
+
+       return 0;
+}
+
+static int rockchip_pcie_prog_ib_atu(struct rockchip_pcie *rockchip,
+                                    int region_no, u8 num_pass_bits,
+                                    u32 lower_addr, u32 upper_addr)
+{
+       u32 ib_addr_0;
+       u32 ib_addr_1;
+       u32 aw_offset;
+
+       if (region_no > MAX_AXI_IB_ROOTPORT_REGION_NUM)
+               return -EINVAL;
+       if (num_pass_bits + 1 < MIN_AXI_ADDR_BITS_PASSED)
+               return -EINVAL;
+       if (num_pass_bits > 63)
+               return -EINVAL;
+
+       aw_offset = (region_no << IB_ROOT_PORT_REG_SIZE_SHIFT);
+
+       ib_addr_0 = num_pass_bits & PCIE_CORE_IB_REGION_ADDR0_NUM_BITS;
+       ib_addr_0 |= (lower_addr << 8) & PCIE_CORE_IB_REGION_ADDR0_LO_ADDR;
+       ib_addr_1 = upper_addr;
+
+       rockchip_pcie_write(rockchip, ib_addr_0, PCIE_RP_IB_ADDR0 + aw_offset);
+       rockchip_pcie_write(rockchip, ib_addr_1, PCIE_RP_IB_ADDR1 + aw_offset);
+
+       return 0;
+}
+
+static int rockchip_pcie_cfg_atu(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       int offset;
+       int err;
+       int reg_no;
+
+       rockchip_pcie_cfg_configuration_accesses(rockchip,
+                                                AXI_WRAPPER_TYPE0_CFG);
+
+       for (reg_no = 0; reg_no < (rockchip->mem_size >> 20); reg_no++) {
+               err = rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1,
+                                               AXI_WRAPPER_MEM_WRITE,
+                                               20 - 1,
+                                               rockchip->mem_bus_addr +
+                                               (reg_no << 20),
+                                               0);
+               if (err) {
+                       dev_err(dev, "program RC mem outbound ATU failed\n");
+                       return err;
+               }
+       }
+
+       err = rockchip_pcie_prog_ib_atu(rockchip, 2, 32 - 1, 0x0, 0);
+       if (err) {
+               dev_err(dev, "program RC mem inbound ATU failed\n");
+               return err;
+       }
+
+       offset = rockchip->mem_size >> 20;
+       for (reg_no = 0; reg_no < (rockchip->io_size >> 20); reg_no++) {
+               err = rockchip_pcie_prog_ob_atu(rockchip,
+                                               reg_no + 1 + offset,
+                                               AXI_WRAPPER_IO_WRITE,
+                                               20 - 1,
+                                               rockchip->io_bus_addr +
+                                               (reg_no << 20),
+                                               0);
+               if (err) {
+                       dev_err(dev, "program RC io outbound ATU failed\n");
+                       return err;
+               }
+       }
+
+       /* assign message regions */
+       rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1 + offset,
+                                 AXI_WRAPPER_NOR_MSG,
+                                 20 - 1, 0, 0);
+
+       rockchip->msg_bus_addr = rockchip->mem_bus_addr +
+                                       ((reg_no + offset) << 20);
+       return err;
+}
+
+static int rockchip_pcie_wait_l2(struct rockchip_pcie *rockchip)
+{
+       u32 value;
+       int err;
+
+       /* send PME_TURN_OFF message */
+       writel(0x0, rockchip->msg_region + PCIE_RC_SEND_PME_OFF);
+
+       /* read LTSSM and wait for falling into L2 link state */
+       err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_DEBUG_OUT_0,
+                                value, PCIE_LINK_IS_L2(value), 20,
+                                jiffies_to_usecs(5 * HZ));
+       if (err) {
+               dev_err(rockchip->dev, "PCIe link enter L2 timeout!\n");
+               return err;
+       }
+
+       return 0;
+}
+
+static int __maybe_unused rockchip_pcie_suspend_noirq(struct device *dev)
+{
+       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
+       int ret;
+
+       /* disable core and cli int since we don't need to ack PME_ACK */
+       rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) |
+                           PCIE_CLIENT_INT_CLI, PCIE_CLIENT_INT_MASK);
+       rockchip_pcie_write(rockchip, (u32)PCIE_CORE_INT, PCIE_CORE_INT_MASK);
+
+       ret = rockchip_pcie_wait_l2(rockchip);
+       if (ret) {
+               rockchip_pcie_enable_interrupts(rockchip);
+               return ret;
+       }
+
+       rockchip_pcie_deinit_phys(rockchip);
+
+       rockchip_pcie_disable_clocks(rockchip);
+
+       if (!IS_ERR(rockchip->vpcie0v9))
+               regulator_disable(rockchip->vpcie0v9);
+
+       return ret;
+}
+
+static int __maybe_unused rockchip_pcie_resume_noirq(struct device *dev)
+{
+       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
+       int err;
+
+       if (!IS_ERR(rockchip->vpcie0v9)) {
+               err = regulator_enable(rockchip->vpcie0v9);
+               if (err) {
+                       dev_err(dev, "fail to enable vpcie0v9 regulator\n");
+                       return err;
+               }
+       }
+
+       err = rockchip_pcie_enable_clocks(rockchip);
+       if (err)
+               goto err_disable_0v9;
+
+       err = rockchip_pcie_host_init_port(rockchip);
+       if (err)
+               goto err_pcie_resume;
+
+       err = rockchip_pcie_cfg_atu(rockchip);
+       if (err)
+               goto err_err_deinit_port;
+
+       /* Need this to enter L1 again */
+       rockchip_pcie_update_txcredit_mui(rockchip);
+       rockchip_pcie_enable_interrupts(rockchip);
+
+       return 0;
+
+err_err_deinit_port:
+       rockchip_pcie_deinit_phys(rockchip);
+err_pcie_resume:
+       rockchip_pcie_disable_clocks(rockchip);
+err_disable_0v9:
+       if (!IS_ERR(rockchip->vpcie0v9))
+               regulator_disable(rockchip->vpcie0v9);
+       return err;
+}
+
+static int rockchip_pcie_probe(struct platform_device *pdev)
+{
+       struct rockchip_pcie *rockchip;
+       struct device *dev = &pdev->dev;
+       struct pci_bus *bus, *child;
+       struct pci_host_bridge *bridge;
+       struct resource_entry *win;
+       resource_size_t io_base;
+       struct resource *mem;
+       struct resource *io;
+       int err;
+
+       LIST_HEAD(res);
+
+       if (!dev->of_node)
+               return -ENODEV;
+
+       bridge = devm_pci_alloc_host_bridge(dev, sizeof(*rockchip));
+       if (!bridge)
+               return -ENOMEM;
+
+       rockchip = pci_host_bridge_priv(bridge);
+
+       platform_set_drvdata(pdev, rockchip);
+       rockchip->dev = dev;
+       rockchip->is_rc = true;
+
+       err = rockchip_pcie_parse_host_dt(rockchip);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_enable_clocks(rockchip);
+       if (err)
+               return err;
+
+       err = rockchip_pcie_set_vpcie(rockchip);
+       if (err) {
+               dev_err(dev, "failed to set vpcie regulator\n");
+               goto err_set_vpcie;
+       }
+
+       err = rockchip_pcie_host_init_port(rockchip);
+       if (err)
+               goto err_vpcie;
+
+       rockchip_pcie_enable_interrupts(rockchip);
+
+       err = rockchip_pcie_init_irq_domain(rockchip);
+       if (err < 0)
+               goto err_deinit_port;
+
+       err = devm_of_pci_get_host_bridge_resources(dev, 0, 0xff,
+                                                   &res, &io_base);
+       if (err)
+               goto err_remove_irq_domain;
+
+       err = devm_request_pci_bus_resources(dev, &res);
+       if (err)
+               goto err_free_res;
+
+       /* Get the I/O and memory ranges from DT */
+       resource_list_for_each_entry(win, &res) {
+               switch (resource_type(win->res)) {
+               case IORESOURCE_IO:
+                       io = win->res;
+                       io->name = "I/O";
+                       rockchip->io_size = resource_size(io);
+                       rockchip->io_bus_addr = io->start - win->offset;
+                       err = pci_remap_iospace(io, io_base);
+                       if (err) {
+                               dev_warn(dev, "error %d: failed to map resource %pR\n",
+                                        err, io);
+                               continue;
+                       }
+                       rockchip->io = io;
+                       break;
+               case IORESOURCE_MEM:
+                       mem = win->res;
+                       mem->name = "MEM";
+                       rockchip->mem_size = resource_size(mem);
+                       rockchip->mem_bus_addr = mem->start - win->offset;
+                       break;
+               case IORESOURCE_BUS:
+                       rockchip->root_bus_nr = win->res->start;
+                       break;
+               default:
+                       continue;
+               }
+       }
+
+       err = rockchip_pcie_cfg_atu(rockchip);
+       if (err)
+               goto err_unmap_iospace;
+
+       rockchip->msg_region = devm_ioremap(dev, rockchip->msg_bus_addr, SZ_1M);
+       if (!rockchip->msg_region) {
+               err = -ENOMEM;
+               goto err_unmap_iospace;
+       }
+
+       list_splice_init(&res, &bridge->windows);
+       bridge->dev.parent = dev;
+       bridge->sysdata = rockchip;
+       bridge->busnr = 0;
+       bridge->ops = &rockchip_pcie_ops;
+       bridge->map_irq = of_irq_parse_and_map_pci;
+       bridge->swizzle_irq = pci_common_swizzle;
+
+       err = pci_scan_root_bus_bridge(bridge);
+       if (err < 0)
+               goto err_unmap_iospace;
+
+       bus = bridge->bus;
+
+       rockchip->root_bus = bus;
+
+       pci_bus_size_bridges(bus);
+       pci_bus_assign_resources(bus);
+       list_for_each_entry(child, &bus->children, node)
+               pcie_bus_configure_settings(child);
+
+       pci_bus_add_devices(bus);
+       return 0;
+
+err_unmap_iospace:
+       pci_unmap_iospace(rockchip->io);
+err_free_res:
+       pci_free_resource_list(&res);
+err_remove_irq_domain:
+       irq_domain_remove(rockchip->irq_domain);
+err_deinit_port:
+       rockchip_pcie_deinit_phys(rockchip);
+err_vpcie:
+       if (!IS_ERR(rockchip->vpcie12v))
+               regulator_disable(rockchip->vpcie12v);
+       if (!IS_ERR(rockchip->vpcie3v3))
+               regulator_disable(rockchip->vpcie3v3);
+       if (!IS_ERR(rockchip->vpcie1v8))
+               regulator_disable(rockchip->vpcie1v8);
+       if (!IS_ERR(rockchip->vpcie0v9))
+               regulator_disable(rockchip->vpcie0v9);
+err_set_vpcie:
+       rockchip_pcie_disable_clocks(rockchip);
+       return err;
+}
+
+static int rockchip_pcie_remove(struct platform_device *pdev)
+{
+       struct device *dev = &pdev->dev;
+       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
+
+       pci_stop_root_bus(rockchip->root_bus);
+       pci_remove_root_bus(rockchip->root_bus);
+       pci_unmap_iospace(rockchip->io);
+       irq_domain_remove(rockchip->irq_domain);
+
+       rockchip_pcie_deinit_phys(rockchip);
+
+       rockchip_pcie_disable_clocks(rockchip);
+
+       if (!IS_ERR(rockchip->vpcie12v))
+               regulator_disable(rockchip->vpcie12v);
+       if (!IS_ERR(rockchip->vpcie3v3))
+               regulator_disable(rockchip->vpcie3v3);
+       if (!IS_ERR(rockchip->vpcie1v8))
+               regulator_disable(rockchip->vpcie1v8);
+       if (!IS_ERR(rockchip->vpcie0v9))
+               regulator_disable(rockchip->vpcie0v9);
+
+       return 0;
+}
+
+static const struct dev_pm_ops rockchip_pcie_pm_ops = {
+       SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(rockchip_pcie_suspend_noirq,
+                                     rockchip_pcie_resume_noirq)
+};
+
+static const struct of_device_id rockchip_pcie_of_match[] = {
+       { .compatible = "rockchip,rk3399-pcie", },
+       {}
+};
+MODULE_DEVICE_TABLE(of, rockchip_pcie_of_match);
+
+static struct platform_driver rockchip_pcie_driver = {
+       .driver = {
+               .name = "rockchip-pcie",
+               .of_match_table = rockchip_pcie_of_match,
+               .pm = &rockchip_pcie_pm_ops,
+       },
+       .probe = rockchip_pcie_probe,
+       .remove = rockchip_pcie_remove,
+};
+module_platform_driver(rockchip_pcie_driver);
+
+MODULE_AUTHOR("Rockchip Inc");
+MODULE_DESCRIPTION("Rockchip AXI PCIe driver");
+MODULE_LICENSE("GPL v2");
index cd85de5fa928390cd8c4edcf69ab56ae7d0663f4..c53d1322a3d6cf873eafc4ba4a5c9d93c56c662c 100644 (file)
  * ARM PCI Host generic driver.
  */
 
-#include <linux/bitrev.h>
 #include <linux/clk.h>
 #include <linux/delay.h>
 #include <linux/gpio/consumer.h>
-#include <linux/init.h>
-#include <linux/interrupt.h>
-#include <linux/iopoll.h>
-#include <linux/irq.h>
-#include <linux/irqchip/chained_irq.h>
-#include <linux/irqdomain.h>
-#include <linux/kernel.h>
-#include <linux/mfd/syscon.h>
-#include <linux/module.h>
-#include <linux/of_address.h>
-#include <linux/of_device.h>
 #include <linux/of_pci.h>
-#include <linux/of_platform.h>
-#include <linux/of_irq.h>
-#include <linux/pci.h>
-#include <linux/pci_ids.h>
 #include <linux/phy/phy.h>
 #include <linux/platform_device.h>
 #include <linux/reset.h>
-#include <linux/regmap.h>
 
 #include "../pci.h"
+#include "pcie-rockchip.h"
 
-/*
- * The upper 16 bits of PCIE_CLIENT_CONFIG are a write mask for the lower 16
- * bits.  This allows atomic updates of the register without locking.
- */
-#define HIWORD_UPDATE(mask, val)       (((mask) << 16) | (val))
-#define HIWORD_UPDATE_BIT(val)         HIWORD_UPDATE(val, val)
-
-#define ENCODE_LANES(x)                        ((((x) >> 1) & 3) << 4)
-#define MAX_LANE_NUM                   4
-
-#define PCIE_CLIENT_BASE               0x0
-#define PCIE_CLIENT_CONFIG             (PCIE_CLIENT_BASE + 0x00)
-#define   PCIE_CLIENT_CONF_ENABLE        HIWORD_UPDATE_BIT(0x0001)
-#define   PCIE_CLIENT_LINK_TRAIN_ENABLE          HIWORD_UPDATE_BIT(0x0002)
-#define   PCIE_CLIENT_ARI_ENABLE         HIWORD_UPDATE_BIT(0x0008)
-#define   PCIE_CLIENT_CONF_LANE_NUM(x)   HIWORD_UPDATE(0x0030, ENCODE_LANES(x))
-#define   PCIE_CLIENT_MODE_RC            HIWORD_UPDATE_BIT(0x0040)
-#define   PCIE_CLIENT_GEN_SEL_1                  HIWORD_UPDATE(0x0080, 0)
-#define   PCIE_CLIENT_GEN_SEL_2                  HIWORD_UPDATE_BIT(0x0080)
-#define PCIE_CLIENT_DEBUG_OUT_0                (PCIE_CLIENT_BASE + 0x3c)
-#define   PCIE_CLIENT_DEBUG_LTSSM_MASK         GENMASK(5, 0)
-#define   PCIE_CLIENT_DEBUG_LTSSM_L1           0x18
-#define   PCIE_CLIENT_DEBUG_LTSSM_L2           0x19
-#define PCIE_CLIENT_BASIC_STATUS1      (PCIE_CLIENT_BASE + 0x48)
-#define   PCIE_CLIENT_LINK_STATUS_UP           0x00300000
-#define   PCIE_CLIENT_LINK_STATUS_MASK         0x00300000
-#define PCIE_CLIENT_INT_MASK           (PCIE_CLIENT_BASE + 0x4c)
-#define PCIE_CLIENT_INT_STATUS         (PCIE_CLIENT_BASE + 0x50)
-#define   PCIE_CLIENT_INTR_MASK                        GENMASK(8, 5)
-#define   PCIE_CLIENT_INTR_SHIFT               5
-#define   PCIE_CLIENT_INT_LEGACY_DONE          BIT(15)
-#define   PCIE_CLIENT_INT_MSG                  BIT(14)
-#define   PCIE_CLIENT_INT_HOT_RST              BIT(13)
-#define   PCIE_CLIENT_INT_DPA                  BIT(12)
-#define   PCIE_CLIENT_INT_FATAL_ERR            BIT(11)
-#define   PCIE_CLIENT_INT_NFATAL_ERR           BIT(10)
-#define   PCIE_CLIENT_INT_CORR_ERR             BIT(9)
-#define   PCIE_CLIENT_INT_INTD                 BIT(8)
-#define   PCIE_CLIENT_INT_INTC                 BIT(7)
-#define   PCIE_CLIENT_INT_INTB                 BIT(6)
-#define   PCIE_CLIENT_INT_INTA                 BIT(5)
-#define   PCIE_CLIENT_INT_LOCAL                        BIT(4)
-#define   PCIE_CLIENT_INT_UDMA                 BIT(3)
-#define   PCIE_CLIENT_INT_PHY                  BIT(2)
-#define   PCIE_CLIENT_INT_HOT_PLUG             BIT(1)
-#define   PCIE_CLIENT_INT_PWR_STCG             BIT(0)
-
-#define PCIE_CLIENT_INT_LEGACY \
-       (PCIE_CLIENT_INT_INTA | PCIE_CLIENT_INT_INTB | \
-       PCIE_CLIENT_INT_INTC | PCIE_CLIENT_INT_INTD)
-
-#define PCIE_CLIENT_INT_CLI \
-       (PCIE_CLIENT_INT_CORR_ERR | PCIE_CLIENT_INT_NFATAL_ERR | \
-       PCIE_CLIENT_INT_FATAL_ERR | PCIE_CLIENT_INT_DPA | \
-       PCIE_CLIENT_INT_HOT_RST | PCIE_CLIENT_INT_MSG | \
-       PCIE_CLIENT_INT_LEGACY_DONE | PCIE_CLIENT_INT_LEGACY | \
-       PCIE_CLIENT_INT_PHY)
-
-#define PCIE_CORE_CTRL_MGMT_BASE       0x900000
-#define PCIE_CORE_CTRL                 (PCIE_CORE_CTRL_MGMT_BASE + 0x000)
-#define   PCIE_CORE_PL_CONF_SPEED_5G           0x00000008
-#define   PCIE_CORE_PL_CONF_SPEED_MASK         0x00000018
-#define   PCIE_CORE_PL_CONF_LANE_MASK          0x00000006
-#define   PCIE_CORE_PL_CONF_LANE_SHIFT         1
-#define PCIE_CORE_CTRL_PLC1            (PCIE_CORE_CTRL_MGMT_BASE + 0x004)
-#define   PCIE_CORE_CTRL_PLC1_FTS_MASK         GENMASK(23, 8)
-#define   PCIE_CORE_CTRL_PLC1_FTS_SHIFT                8
-#define   PCIE_CORE_CTRL_PLC1_FTS_CNT          0xffff
-#define PCIE_CORE_TXCREDIT_CFG1                (PCIE_CORE_CTRL_MGMT_BASE + 0x020)
-#define   PCIE_CORE_TXCREDIT_CFG1_MUI_MASK     0xFFFF0000
-#define   PCIE_CORE_TXCREDIT_CFG1_MUI_SHIFT    16
-#define   PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(x) \
-               (((x) >> 3) << PCIE_CORE_TXCREDIT_CFG1_MUI_SHIFT)
-#define PCIE_CORE_LANE_MAP             (PCIE_CORE_CTRL_MGMT_BASE + 0x200)
-#define   PCIE_CORE_LANE_MAP_MASK              0x0000000f
-#define   PCIE_CORE_LANE_MAP_REVERSE           BIT(16)
-#define PCIE_CORE_INT_STATUS           (PCIE_CORE_CTRL_MGMT_BASE + 0x20c)
-#define   PCIE_CORE_INT_PRFPE                  BIT(0)
-#define   PCIE_CORE_INT_CRFPE                  BIT(1)
-#define   PCIE_CORE_INT_RRPE                   BIT(2)
-#define   PCIE_CORE_INT_PRFO                   BIT(3)
-#define   PCIE_CORE_INT_CRFO                   BIT(4)
-#define   PCIE_CORE_INT_RT                     BIT(5)
-#define   PCIE_CORE_INT_RTR                    BIT(6)
-#define   PCIE_CORE_INT_PE                     BIT(7)
-#define   PCIE_CORE_INT_MTR                    BIT(8)
-#define   PCIE_CORE_INT_UCR                    BIT(9)
-#define   PCIE_CORE_INT_FCE                    BIT(10)
-#define   PCIE_CORE_INT_CT                     BIT(11)
-#define   PCIE_CORE_INT_UTC                    BIT(18)
-#define   PCIE_CORE_INT_MMVC                   BIT(19)
-#define PCIE_CORE_CONFIG_VENDOR                (PCIE_CORE_CTRL_MGMT_BASE + 0x44)
-#define PCIE_CORE_INT_MASK             (PCIE_CORE_CTRL_MGMT_BASE + 0x210)
-#define PCIE_RC_BAR_CONF               (PCIE_CORE_CTRL_MGMT_BASE + 0x300)
-
-#define PCIE_CORE_INT \
-               (PCIE_CORE_INT_PRFPE | PCIE_CORE_INT_CRFPE | \
-                PCIE_CORE_INT_RRPE | PCIE_CORE_INT_CRFO | \
-                PCIE_CORE_INT_RT | PCIE_CORE_INT_RTR | \
-                PCIE_CORE_INT_PE | PCIE_CORE_INT_MTR | \
-                PCIE_CORE_INT_UCR | PCIE_CORE_INT_FCE | \
-                PCIE_CORE_INT_CT | PCIE_CORE_INT_UTC | \
-                PCIE_CORE_INT_MMVC)
-
-#define PCIE_RC_CONFIG_NORMAL_BASE     0x800000
-#define PCIE_RC_CONFIG_BASE            0xa00000
-#define PCIE_RC_CONFIG_RID_CCR         (PCIE_RC_CONFIG_BASE + 0x08)
-#define   PCIE_RC_CONFIG_SCC_SHIFT             16
-#define PCIE_RC_CONFIG_DCR             (PCIE_RC_CONFIG_BASE + 0xc4)
-#define   PCIE_RC_CONFIG_DCR_CSPL_SHIFT                18
-#define   PCIE_RC_CONFIG_DCR_CSPL_LIMIT                0xff
-#define   PCIE_RC_CONFIG_DCR_CPLS_SHIFT                26
-#define PCIE_RC_CONFIG_DCSR            (PCIE_RC_CONFIG_BASE + 0xc8)
-#define   PCIE_RC_CONFIG_DCSR_MPS_MASK         GENMASK(7, 5)
-#define   PCIE_RC_CONFIG_DCSR_MPS_256          (0x1 << 5)
-#define PCIE_RC_CONFIG_LINK_CAP                (PCIE_RC_CONFIG_BASE + 0xcc)
-#define   PCIE_RC_CONFIG_LINK_CAP_L0S          BIT(10)
-#define PCIE_RC_CONFIG_LCS             (PCIE_RC_CONFIG_BASE + 0xd0)
-#define PCIE_RC_CONFIG_L1_SUBSTATE_CTRL2 (PCIE_RC_CONFIG_BASE + 0x90c)
-#define PCIE_RC_CONFIG_THP_CAP         (PCIE_RC_CONFIG_BASE + 0x274)
-#define   PCIE_RC_CONFIG_THP_CAP_NEXT_MASK     GENMASK(31, 20)
-
-#define PCIE_CORE_AXI_CONF_BASE                0xc00000
-#define PCIE_CORE_OB_REGION_ADDR0      (PCIE_CORE_AXI_CONF_BASE + 0x0)
-#define   PCIE_CORE_OB_REGION_ADDR0_NUM_BITS   0x3f
-#define   PCIE_CORE_OB_REGION_ADDR0_LO_ADDR    0xffffff00
-#define PCIE_CORE_OB_REGION_ADDR1      (PCIE_CORE_AXI_CONF_BASE + 0x4)
-#define PCIE_CORE_OB_REGION_DESC0      (PCIE_CORE_AXI_CONF_BASE + 0x8)
-#define PCIE_CORE_OB_REGION_DESC1      (PCIE_CORE_AXI_CONF_BASE + 0xc)
-
-#define PCIE_CORE_AXI_INBOUND_BASE     0xc00800
-#define PCIE_RP_IB_ADDR0               (PCIE_CORE_AXI_INBOUND_BASE + 0x0)
-#define   PCIE_CORE_IB_REGION_ADDR0_NUM_BITS   0x3f
-#define   PCIE_CORE_IB_REGION_ADDR0_LO_ADDR    0xffffff00
-#define PCIE_RP_IB_ADDR1               (PCIE_CORE_AXI_INBOUND_BASE + 0x4)
-
-/* Size of one AXI Region (not Region 0) */
-#define AXI_REGION_SIZE                                BIT(20)
-/* Size of Region 0, equal to sum of sizes of other regions */
-#define AXI_REGION_0_SIZE                      (32 * (0x1 << 20))
-#define OB_REG_SIZE_SHIFT                      5
-#define IB_ROOT_PORT_REG_SIZE_SHIFT            3
-#define AXI_WRAPPER_IO_WRITE                   0x6
-#define AXI_WRAPPER_MEM_WRITE                  0x2
-#define AXI_WRAPPER_TYPE0_CFG                  0xa
-#define AXI_WRAPPER_TYPE1_CFG                  0xb
-#define AXI_WRAPPER_NOR_MSG                    0xc
-
-#define MAX_AXI_IB_ROOTPORT_REGION_NUM         3
-#define MIN_AXI_ADDR_BITS_PASSED               8
-#define PCIE_RC_SEND_PME_OFF                   0x11960
-#define ROCKCHIP_VENDOR_ID                     0x1d87
-#define PCIE_ECAM_BUS(x)                       (((x) & 0xff) << 20)
-#define PCIE_ECAM_DEV(x)                       (((x) & 0x1f) << 15)
-#define PCIE_ECAM_FUNC(x)                      (((x) & 0x7) << 12)
-#define PCIE_ECAM_REG(x)                       (((x) & 0xfff) << 0)
-#define PCIE_ECAM_ADDR(bus, dev, func, reg) \
-         (PCIE_ECAM_BUS(bus) | PCIE_ECAM_DEV(dev) | \
-          PCIE_ECAM_FUNC(func) | PCIE_ECAM_REG(reg))
-#define PCIE_LINK_IS_L2(x) \
-       (((x) & PCIE_CLIENT_DEBUG_LTSSM_MASK) == PCIE_CLIENT_DEBUG_LTSSM_L2)
-#define PCIE_LINK_UP(x) \
-       (((x) & PCIE_CLIENT_LINK_STATUS_MASK) == PCIE_CLIENT_LINK_STATUS_UP)
-#define PCIE_LINK_IS_GEN2(x) \
-       (((x) & PCIE_CORE_PL_CONF_SPEED_MASK) == PCIE_CORE_PL_CONF_SPEED_5G)
-
-#define RC_REGION_0_ADDR_TRANS_H               0x00000000
-#define RC_REGION_0_ADDR_TRANS_L               0x00000000
-#define RC_REGION_0_PASS_BITS                  (25 - 1)
-#define RC_REGION_0_TYPE_MASK                  GENMASK(3, 0)
-#define MAX_AXI_WRAPPER_REGION_NUM             33
-
-struct rockchip_pcie {
-       void    __iomem *reg_base;              /* DT axi-base */
-       void    __iomem *apb_base;              /* DT apb-base */
-       bool    legacy_phy;
-       struct  phy *phys[MAX_LANE_NUM];
-       struct  reset_control *core_rst;
-       struct  reset_control *mgmt_rst;
-       struct  reset_control *mgmt_sticky_rst;
-       struct  reset_control *pipe_rst;
-       struct  reset_control *pm_rst;
-       struct  reset_control *aclk_rst;
-       struct  reset_control *pclk_rst;
-       struct  clk *aclk_pcie;
-       struct  clk *aclk_perf_pcie;
-       struct  clk *hclk_pcie;
-       struct  clk *clk_pcie_pm;
-       struct  regulator *vpcie12v; /* 12V power supply */
-       struct  regulator *vpcie3v3; /* 3.3V power supply */
-       struct  regulator *vpcie1v8; /* 1.8V power supply */
-       struct  regulator *vpcie0v9; /* 0.9V power supply */
-       struct  gpio_desc *ep_gpio;
-       u32     lanes;
-       u8      lanes_map;
-       u8      root_bus_nr;
-       int     link_gen;
-       struct  device *dev;
-       struct  irq_domain *irq_domain;
-       int     offset;
-       struct pci_bus *root_bus;
-       struct resource *io;
-       phys_addr_t io_bus_addr;
-       u32     io_size;
-       void    __iomem *msg_region;
-       u32     mem_size;
-       phys_addr_t msg_bus_addr;
-       phys_addr_t mem_bus_addr;
-};
-
-static u32 rockchip_pcie_read(struct rockchip_pcie *rockchip, u32 reg)
-{
-       return readl(rockchip->apb_base + reg);
-}
-
-static void rockchip_pcie_write(struct rockchip_pcie *rockchip, u32 val,
-                               u32 reg)
-{
-       writel(val, rockchip->apb_base + reg);
-}
-
-static void rockchip_pcie_enable_bw_int(struct rockchip_pcie *rockchip)
-{
-       u32 status;
-
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
-       status |= (PCI_EXP_LNKCTL_LBMIE | PCI_EXP_LNKCTL_LABIE);
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
-}
-
-static void rockchip_pcie_clr_bw_int(struct rockchip_pcie *rockchip)
-{
-       u32 status;
-
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
-       status |= (PCI_EXP_LNKSTA_LBMS | PCI_EXP_LNKSTA_LABS) << 16;
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
-}
-
-static void rockchip_pcie_update_txcredit_mui(struct rockchip_pcie *rockchip)
-{
-       u32 val;
-
-       /* Update Tx credit maximum update interval */
-       val = rockchip_pcie_read(rockchip, PCIE_CORE_TXCREDIT_CFG1);
-       val &= ~PCIE_CORE_TXCREDIT_CFG1_MUI_MASK;
-       val |= PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(24000);       /* ns */
-       rockchip_pcie_write(rockchip, val, PCIE_CORE_TXCREDIT_CFG1);
-}
-
-static int rockchip_pcie_valid_device(struct rockchip_pcie *rockchip,
-                                     struct pci_bus *bus, int dev)
-{
-       /* access only one slot on each root port */
-       if (bus->number == rockchip->root_bus_nr && dev > 0)
-               return 0;
-
-       /*
-        * do not read more than one device on the bus directly attached
-        * to RC's downstream side.
-        */
-       if (bus->primary == rockchip->root_bus_nr && dev > 0)
-               return 0;
-
-       return 1;
-}
-
-static u8 rockchip_pcie_lane_map(struct rockchip_pcie *rockchip)
-{
-       u32 val;
-       u8 map;
-
-       if (rockchip->legacy_phy)
-               return GENMASK(MAX_LANE_NUM - 1, 0);
-
-       val = rockchip_pcie_read(rockchip, PCIE_CORE_LANE_MAP);
-       map = val & PCIE_CORE_LANE_MAP_MASK;
-
-       /* The link may be using a reverse-indexed mapping. */
-       if (val & PCIE_CORE_LANE_MAP_REVERSE)
-               map = bitrev8(map) >> 4;
-
-       return map;
-}
-
-static int rockchip_pcie_rd_own_conf(struct rockchip_pcie *rockchip,
-                                    int where, int size, u32 *val)
+int rockchip_pcie_parse_dt(struct rockchip_pcie *rockchip)
 {
-       void __iomem *addr;
-
-       addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + where;
-
-       if (!IS_ALIGNED((uintptr_t)addr, size)) {
-               *val = 0;
-               return PCIBIOS_BAD_REGISTER_NUMBER;
-       }
+       struct device *dev = rockchip->dev;
+       struct platform_device *pdev = to_platform_device(dev);
+       struct device_node *node = dev->of_node;
+       struct resource *regs;
+       int err;
 
-       if (size == 4) {
-               *val = readl(addr);
-       } else if (size == 2) {
-               *val = readw(addr);
-       } else if (size == 1) {
-               *val = readb(addr);
+       if (rockchip->is_rc) {
+               regs = platform_get_resource_byname(pdev,
+                                                   IORESOURCE_MEM,
+                                                   "axi-base");
+               rockchip->reg_base = devm_pci_remap_cfg_resource(dev, regs);
+               if (IS_ERR(rockchip->reg_base))
+                       return PTR_ERR(rockchip->reg_base);
        } else {
-               *val = 0;
-               return PCIBIOS_BAD_REGISTER_NUMBER;
+               rockchip->mem_res =
+                       platform_get_resource_byname(pdev, IORESOURCE_MEM,
+                                                    "mem-base");
+               if (!rockchip->mem_res)
+                       return -EINVAL;
        }
-       return PCIBIOS_SUCCESSFUL;
-}
 
-static int rockchip_pcie_wr_own_conf(struct rockchip_pcie *rockchip,
-                                    int where, int size, u32 val)
-{
-       u32 mask, tmp, offset;
-       void __iomem *addr;
+       regs = platform_get_resource_byname(pdev, IORESOURCE_MEM,
+                                           "apb-base");
+       rockchip->apb_base = devm_ioremap_resource(dev, regs);
+       if (IS_ERR(rockchip->apb_base))
+               return PTR_ERR(rockchip->apb_base);
 
-       offset = where & ~0x3;
-       addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + offset;
+       err = rockchip_pcie_get_phys(rockchip);
+       if (err)
+               return err;
 
-       if (size == 4) {
-               writel(val, addr);
-               return PCIBIOS_SUCCESSFUL;
+       rockchip->lanes = 1;
+       err = of_property_read_u32(node, "num-lanes", &rockchip->lanes);
+       if (!err && (rockchip->lanes == 0 ||
+                    rockchip->lanes == 3 ||
+                    rockchip->lanes > 4)) {
+               dev_warn(dev, "invalid num-lanes, default to use one lane\n");
+               rockchip->lanes = 1;
        }
 
-       mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
-
-       /*
-        * N.B. This read/modify/write isn't safe in general because it can
-        * corrupt RW1C bits in adjacent registers.  But the hardware
-        * doesn't support smaller writes.
-        */
-       tmp = readl(addr) & mask;
-       tmp |= val << ((where & 0x3) * 8);
-       writel(tmp, addr);
-
-       return PCIBIOS_SUCCESSFUL;
-}
-
-static void rockchip_pcie_cfg_configuration_accesses(
-               struct rockchip_pcie *rockchip, u32 type)
-{
-       u32 ob_desc_0;
-
-       /* Configuration Accesses for region 0 */
-       rockchip_pcie_write(rockchip, 0x0, PCIE_RC_BAR_CONF);
-
-       rockchip_pcie_write(rockchip,
-                           (RC_REGION_0_ADDR_TRANS_L + RC_REGION_0_PASS_BITS),
-                           PCIE_CORE_OB_REGION_ADDR0);
-       rockchip_pcie_write(rockchip, RC_REGION_0_ADDR_TRANS_H,
-                           PCIE_CORE_OB_REGION_ADDR1);
-       ob_desc_0 = rockchip_pcie_read(rockchip, PCIE_CORE_OB_REGION_DESC0);
-       ob_desc_0 &= ~(RC_REGION_0_TYPE_MASK);
-       ob_desc_0 |= (type | (0x1 << 23));
-       rockchip_pcie_write(rockchip, ob_desc_0, PCIE_CORE_OB_REGION_DESC0);
-       rockchip_pcie_write(rockchip, 0x0, PCIE_CORE_OB_REGION_DESC1);
-}
-
-static int rockchip_pcie_rd_other_conf(struct rockchip_pcie *rockchip,
-                                      struct pci_bus *bus, u32 devfn,
-                                      int where, int size, u32 *val)
-{
-       u32 busdev;
-
-       busdev = PCIE_ECAM_ADDR(bus->number, PCI_SLOT(devfn),
-                               PCI_FUNC(devfn), where);
+       rockchip->link_gen = of_pci_get_max_link_speed(node);
+       if (rockchip->link_gen < 0 || rockchip->link_gen > 2)
+               rockchip->link_gen = 2;
 
-       if (!IS_ALIGNED(busdev, size)) {
-               *val = 0;
-               return PCIBIOS_BAD_REGISTER_NUMBER;
+       rockchip->core_rst = devm_reset_control_get_exclusive(dev, "core");
+       if (IS_ERR(rockchip->core_rst)) {
+               if (PTR_ERR(rockchip->core_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing core reset property in node\n");
+               return PTR_ERR(rockchip->core_rst);
        }
 
-       if (bus->parent->number == rockchip->root_bus_nr)
-               rockchip_pcie_cfg_configuration_accesses(rockchip,
-                                               AXI_WRAPPER_TYPE0_CFG);
-       else
-               rockchip_pcie_cfg_configuration_accesses(rockchip,
-                                               AXI_WRAPPER_TYPE1_CFG);
-
-       if (size == 4) {
-               *val = readl(rockchip->reg_base + busdev);
-       } else if (size == 2) {
-               *val = readw(rockchip->reg_base + busdev);
-       } else if (size == 1) {
-               *val = readb(rockchip->reg_base + busdev);
-       } else {
-               *val = 0;
-               return PCIBIOS_BAD_REGISTER_NUMBER;
+       rockchip->mgmt_rst = devm_reset_control_get_exclusive(dev, "mgmt");
+       if (IS_ERR(rockchip->mgmt_rst)) {
+               if (PTR_ERR(rockchip->mgmt_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing mgmt reset property in node\n");
+               return PTR_ERR(rockchip->mgmt_rst);
        }
-       return PCIBIOS_SUCCESSFUL;
-}
-
-static int rockchip_pcie_wr_other_conf(struct rockchip_pcie *rockchip,
-                                      struct pci_bus *bus, u32 devfn,
-                                      int where, int size, u32 val)
-{
-       u32 busdev;
-
-       busdev = PCIE_ECAM_ADDR(bus->number, PCI_SLOT(devfn),
-                               PCI_FUNC(devfn), where);
-       if (!IS_ALIGNED(busdev, size))
-               return PCIBIOS_BAD_REGISTER_NUMBER;
-
-       if (bus->parent->number == rockchip->root_bus_nr)
-               rockchip_pcie_cfg_configuration_accesses(rockchip,
-                                               AXI_WRAPPER_TYPE0_CFG);
-       else
-               rockchip_pcie_cfg_configuration_accesses(rockchip,
-                                               AXI_WRAPPER_TYPE1_CFG);
-
-       if (size == 4)
-               writel(val, rockchip->reg_base + busdev);
-       else if (size == 2)
-               writew(val, rockchip->reg_base + busdev);
-       else if (size == 1)
-               writeb(val, rockchip->reg_base + busdev);
-       else
-               return PCIBIOS_BAD_REGISTER_NUMBER;
-
-       return PCIBIOS_SUCCESSFUL;
-}
 
-static int rockchip_pcie_rd_conf(struct pci_bus *bus, u32 devfn, int where,
-                                int size, u32 *val)
-{
-       struct rockchip_pcie *rockchip = bus->sysdata;
-
-       if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn))) {
-               *val = 0xffffffff;
-               return PCIBIOS_DEVICE_NOT_FOUND;
+       rockchip->mgmt_sticky_rst = devm_reset_control_get_exclusive(dev,
+                                                                    "mgmt-sticky");
+       if (IS_ERR(rockchip->mgmt_sticky_rst)) {
+               if (PTR_ERR(rockchip->mgmt_sticky_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing mgmt-sticky reset property in node\n");
+               return PTR_ERR(rockchip->mgmt_sticky_rst);
        }
 
-       if (bus->number == rockchip->root_bus_nr)
-               return rockchip_pcie_rd_own_conf(rockchip, where, size, val);
-
-       return rockchip_pcie_rd_other_conf(rockchip, bus, devfn, where, size, val);
-}
+       rockchip->pipe_rst = devm_reset_control_get_exclusive(dev, "pipe");
+       if (IS_ERR(rockchip->pipe_rst)) {
+               if (PTR_ERR(rockchip->pipe_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing pipe reset property in node\n");
+               return PTR_ERR(rockchip->pipe_rst);
+       }
 
-static int rockchip_pcie_wr_conf(struct pci_bus *bus, u32 devfn,
-                                int where, int size, u32 val)
-{
-       struct rockchip_pcie *rockchip = bus->sysdata;
+       rockchip->pm_rst = devm_reset_control_get_exclusive(dev, "pm");
+       if (IS_ERR(rockchip->pm_rst)) {
+               if (PTR_ERR(rockchip->pm_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing pm reset property in node\n");
+               return PTR_ERR(rockchip->pm_rst);
+       }
 
-       if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
-               return PCIBIOS_DEVICE_NOT_FOUND;
+       rockchip->pclk_rst = devm_reset_control_get_exclusive(dev, "pclk");
+       if (IS_ERR(rockchip->pclk_rst)) {
+               if (PTR_ERR(rockchip->pclk_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing pclk reset property in node\n");
+               return PTR_ERR(rockchip->pclk_rst);
+       }
 
-       if (bus->number == rockchip->root_bus_nr)
-               return rockchip_pcie_wr_own_conf(rockchip, where, size, val);
+       rockchip->aclk_rst = devm_reset_control_get_exclusive(dev, "aclk");
+       if (IS_ERR(rockchip->aclk_rst)) {
+               if (PTR_ERR(rockchip->aclk_rst) != -EPROBE_DEFER)
+                       dev_err(dev, "missing aclk reset property in node\n");
+               return PTR_ERR(rockchip->aclk_rst);
+       }
 
-       return rockchip_pcie_wr_other_conf(rockchip, bus, devfn, where, size, val);
-}
+       if (rockchip->is_rc) {
+               rockchip->ep_gpio = devm_gpiod_get(dev, "ep", GPIOD_OUT_HIGH);
+               if (IS_ERR(rockchip->ep_gpio)) {
+                       dev_err(dev, "missing ep-gpios property in node\n");
+                       return PTR_ERR(rockchip->ep_gpio);
+               }
+       }
 
-static struct pci_ops rockchip_pcie_ops = {
-       .read = rockchip_pcie_rd_conf,
-       .write = rockchip_pcie_wr_conf,
-};
+       rockchip->aclk_pcie = devm_clk_get(dev, "aclk");
+       if (IS_ERR(rockchip->aclk_pcie)) {
+               dev_err(dev, "aclk clock not found\n");
+               return PTR_ERR(rockchip->aclk_pcie);
+       }
 
-static void rockchip_pcie_set_power_limit(struct rockchip_pcie *rockchip)
-{
-       int curr;
-       u32 status, scale, power;
+       rockchip->aclk_perf_pcie = devm_clk_get(dev, "aclk-perf");
+       if (IS_ERR(rockchip->aclk_perf_pcie)) {
+               dev_err(dev, "aclk_perf clock not found\n");
+               return PTR_ERR(rockchip->aclk_perf_pcie);
+       }
 
-       if (IS_ERR(rockchip->vpcie3v3))
-               return;
+       rockchip->hclk_pcie = devm_clk_get(dev, "hclk");
+       if (IS_ERR(rockchip->hclk_pcie)) {
+               dev_err(dev, "hclk clock not found\n");
+               return PTR_ERR(rockchip->hclk_pcie);
+       }
 
-       /*
-        * Set RC's captured slot power limit and scale if
-        * vpcie3v3 available. The default values are both zero
-        * which means the software should set these two according
-        * to the actual power supply.
-        */
-       curr = regulator_get_current_limit(rockchip->vpcie3v3);
-       if (curr <= 0)
-               return;
-
-       scale = 3; /* 0.001x */
-       curr = curr / 1000; /* convert to mA */
-       power = (curr * 3300) / 1000; /* milliwatt */
-       while (power > PCIE_RC_CONFIG_DCR_CSPL_LIMIT) {
-               if (!scale) {
-                       dev_warn(rockchip->dev, "invalid power supply\n");
-                       return;
-               }
-               scale--;
-               power = power / 10;
+       rockchip->clk_pcie_pm = devm_clk_get(dev, "pm");
+       if (IS_ERR(rockchip->clk_pcie_pm)) {
+               dev_err(dev, "pm clock not found\n");
+               return PTR_ERR(rockchip->clk_pcie_pm);
        }
 
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCR);
-       status |= (power << PCIE_RC_CONFIG_DCR_CSPL_SHIFT) |
-                 (scale << PCIE_RC_CONFIG_DCR_CPLS_SHIFT);
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCR);
+       return 0;
 }
+EXPORT_SYMBOL_GPL(rockchip_pcie_parse_dt);
 
-/**
- * rockchip_pcie_init_port - Initialize hardware
- * @rockchip: PCIe port information
- */
-static int rockchip_pcie_init_port(struct rockchip_pcie *rockchip)
+int rockchip_pcie_init_port(struct rockchip_pcie *rockchip)
 {
        struct device *dev = rockchip->dev;
        int err, i;
-       u32 status;
-
-       gpiod_set_value_cansleep(rockchip->ep_gpio, 0);
+       u32 regs;
 
        err = reset_control_assert(rockchip->aclk_rst);
        if (err) {
@@ -620,13 +237,15 @@ static int rockchip_pcie_init_port(struct rockchip_pcie *rockchip)
                rockchip_pcie_write(rockchip, PCIE_CLIENT_GEN_SEL_1,
                                    PCIE_CLIENT_CONFIG);
 
-       rockchip_pcie_write(rockchip,
-                           PCIE_CLIENT_CONF_ENABLE |
-                           PCIE_CLIENT_LINK_TRAIN_ENABLE |
-                           PCIE_CLIENT_ARI_ENABLE |
-                           PCIE_CLIENT_CONF_LANE_NUM(rockchip->lanes) |
-                           PCIE_CLIENT_MODE_RC,
-                           PCIE_CLIENT_CONFIG);
+       regs = PCIE_CLIENT_LINK_TRAIN_ENABLE | PCIE_CLIENT_ARI_ENABLE |
+              PCIE_CLIENT_CONF_LANE_NUM(rockchip->lanes);
+
+       if (rockchip->is_rc)
+               regs |= PCIE_CLIENT_CONF_ENABLE | PCIE_CLIENT_MODE_RC;
+       else
+               regs |= PCIE_CLIENT_CONF_DISABLE | PCIE_CLIENT_MODE_EP;
+
+       rockchip_pcie_write(rockchip, regs, PCIE_CLIENT_CONFIG);
 
        for (i = 0; i < MAX_LANE_NUM; i++) {
                err = phy_power_on(rockchip->phys[i]);
@@ -664,105 +283,61 @@ static int rockchip_pcie_init_port(struct rockchip_pcie *rockchip)
                goto err_power_off_phy;
        }
 
-       /* Fix the transmitted FTS count desired to exit from L0s. */
-       status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL_PLC1);
-       status = (status & ~PCIE_CORE_CTRL_PLC1_FTS_MASK) |
-                (PCIE_CORE_CTRL_PLC1_FTS_CNT << PCIE_CORE_CTRL_PLC1_FTS_SHIFT);
-       rockchip_pcie_write(rockchip, status, PCIE_CORE_CTRL_PLC1);
-
-       rockchip_pcie_set_power_limit(rockchip);
-
-       /* Set RC's clock architecture as common clock */
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
-       status |= PCI_EXP_LNKSTA_SLC << 16;
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
-
-       /* Set RC's RCB to 128 */
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
-       status |= PCI_EXP_LNKCTL_RCB;
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
-
-       /* Enable Gen1 training */
-       rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
-                           PCIE_CLIENT_CONFIG);
+       return 0;
+err_power_off_phy:
+       while (i--)
+               phy_power_off(rockchip->phys[i]);
+       i = MAX_LANE_NUM;
+err_exit_phy:
+       while (i--)
+               phy_exit(rockchip->phys[i]);
+       return err;
+}
+EXPORT_SYMBOL_GPL(rockchip_pcie_init_port);
 
-       gpiod_set_value_cansleep(rockchip->ep_gpio, 1);
+int rockchip_pcie_get_phys(struct rockchip_pcie *rockchip)
+{
+       struct device *dev = rockchip->dev;
+       struct phy *phy;
+       char *name;
+       u32 i;
 
-       /* 500ms timeout value should be enough for Gen1/2 training */
-       err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_BASIC_STATUS1,
-                                status, PCIE_LINK_UP(status), 20,
-                                500 * USEC_PER_MSEC);
-       if (err) {
-               dev_err(dev, "PCIe link training gen1 timeout!\n");
-               goto err_power_off_phy;
+       phy = devm_phy_get(dev, "pcie-phy");
+       if (!IS_ERR(phy)) {
+               rockchip->legacy_phy = true;
+               rockchip->phys[0] = phy;
+               dev_warn(dev, "legacy phy model is deprecated!\n");
+               return 0;
        }
 
-       if (rockchip->link_gen == 2) {
-               /*
-                * Enable retrain for gen2. This should be configured only after
-                * gen1 finished.
-                */
-               status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
-               status |= PCI_EXP_LNKCTL_RL;
-               rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
-
-               err = readl_poll_timeout(rockchip->apb_base + PCIE_CORE_CTRL,
-                                        status, PCIE_LINK_IS_GEN2(status), 20,
-                                        500 * USEC_PER_MSEC);
-               if (err)
-                       dev_dbg(dev, "PCIe link training gen2 timeout, fall back to gen1!\n");
-       }
+       if (PTR_ERR(phy) == -EPROBE_DEFER)
+               return PTR_ERR(phy);
 
-       /* Check the final link width from negotiated lane counter from MGMT */
-       status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL);
-       status = 0x1 << ((status & PCIE_CORE_PL_CONF_LANE_MASK) >>
-                         PCIE_CORE_PL_CONF_LANE_SHIFT);
-       dev_dbg(dev, "current link width is x%d\n", status);
+       dev_dbg(dev, "missing legacy phy; search for per-lane PHY\n");
 
-       /* Power off unused lane(s) */
-       rockchip->lanes_map = rockchip_pcie_lane_map(rockchip);
        for (i = 0; i < MAX_LANE_NUM; i++) {
-               if (!(rockchip->lanes_map & BIT(i))) {
-                       dev_dbg(dev, "idling lane %d\n", i);
-                       phy_power_off(rockchip->phys[i]);
+               name = kasprintf(GFP_KERNEL, "pcie-phy-%u", i);
+               if (!name)
+                       return -ENOMEM;
+
+               phy = devm_of_phy_get(dev, dev->of_node, name);
+               kfree(name);
+
+               if (IS_ERR(phy)) {
+                       if (PTR_ERR(phy) != -EPROBE_DEFER)
+                               dev_err(dev, "missing phy for lane %d: %ld\n",
+                                       i, PTR_ERR(phy));
+                       return PTR_ERR(phy);
                }
-       }
 
-       rockchip_pcie_write(rockchip, ROCKCHIP_VENDOR_ID,
-                           PCIE_CORE_CONFIG_VENDOR);
-       rockchip_pcie_write(rockchip,
-                           PCI_CLASS_BRIDGE_PCI << PCIE_RC_CONFIG_SCC_SHIFT,
-                           PCIE_RC_CONFIG_RID_CCR);
-
-       /* Clear THP cap's next cap pointer to remove L1 substate cap */
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_THP_CAP);
-       status &= ~PCIE_RC_CONFIG_THP_CAP_NEXT_MASK;
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_THP_CAP);
-
-       /* Clear L0s from RC's link cap */
-       if (of_property_read_bool(dev->of_node, "aspm-no-l0s")) {
-               status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LINK_CAP);
-               status &= ~PCIE_RC_CONFIG_LINK_CAP_L0S;
-               rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LINK_CAP);
+               rockchip->phys[i] = phy;
        }
 
-       status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCSR);
-       status &= ~PCIE_RC_CONFIG_DCSR_MPS_MASK;
-       status |= PCIE_RC_CONFIG_DCSR_MPS_256;
-       rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCSR);
-
        return 0;
-err_power_off_phy:
-       while (i--)
-               phy_power_off(rockchip->phys[i]);
-       i = MAX_LANE_NUM;
-err_exit_phy:
-       while (i--)
-               phy_exit(rockchip->phys[i]);
-       return err;
 }
+EXPORT_SYMBOL_GPL(rockchip_pcie_get_phys);
 
-static void rockchip_pcie_deinit_phys(struct rockchip_pcie *rockchip)
+void rockchip_pcie_deinit_phys(struct rockchip_pcie *rockchip)
 {
        int i;
 
@@ -773,650 +348,30 @@ static void rockchip_pcie_deinit_phys(struct rockchip_pcie *rockchip)
                phy_exit(rockchip->phys[i]);
        }
 }
+EXPORT_SYMBOL_GPL(rockchip_pcie_deinit_phys);
 
-static irqreturn_t rockchip_pcie_subsys_irq_handler(int irq, void *arg)
+int rockchip_pcie_enable_clocks(struct rockchip_pcie *rockchip)
 {
-       struct rockchip_pcie *rockchip = arg;
        struct device *dev = rockchip->dev;
-       u32 reg;
-       u32 sub_reg;
-
-       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
-       if (reg & PCIE_CLIENT_INT_LOCAL) {
-               dev_dbg(dev, "local interrupt received\n");
-               sub_reg = rockchip_pcie_read(rockchip, PCIE_CORE_INT_STATUS);
-               if (sub_reg & PCIE_CORE_INT_PRFPE)
-                       dev_dbg(dev, "parity error detected while reading from the PNP receive FIFO RAM\n");
-
-               if (sub_reg & PCIE_CORE_INT_CRFPE)
-                       dev_dbg(dev, "parity error detected while reading from the Completion Receive FIFO RAM\n");
-
-               if (sub_reg & PCIE_CORE_INT_RRPE)
-                       dev_dbg(dev, "parity error detected while reading from replay buffer RAM\n");
-
-               if (sub_reg & PCIE_CORE_INT_PRFO)
-                       dev_dbg(dev, "overflow occurred in the PNP receive FIFO\n");
+       int err;
 
-               if (sub_reg & PCIE_CORE_INT_CRFO)
-                       dev_dbg(dev, "overflow occurred in the completion receive FIFO\n");
+       err = clk_prepare_enable(rockchip->aclk_pcie);
+       if (err) {
+               dev_err(dev, "unable to enable aclk_pcie clock\n");
+               return err;
+       }
 
-               if (sub_reg & PCIE_CORE_INT_RT)
-                       dev_dbg(dev, "replay timer timed out\n");
+       err = clk_prepare_enable(rockchip->aclk_perf_pcie);
+       if (err) {
+               dev_err(dev, "unable to enable aclk_perf_pcie clock\n");
+               goto err_aclk_perf_pcie;
+       }
 
-               if (sub_reg & PCIE_CORE_INT_RTR)
-                       dev_dbg(dev, "replay timer rolled over after 4 transmissions of the same TLP\n");
-
-               if (sub_reg & PCIE_CORE_INT_PE)
-                       dev_dbg(dev, "phy error detected on receive side\n");
-
-               if (sub_reg & PCIE_CORE_INT_MTR)
-                       dev_dbg(dev, "malformed TLP received from the link\n");
-
-               if (sub_reg & PCIE_CORE_INT_UCR)
-                       dev_dbg(dev, "malformed TLP received from the link\n");
-
-               if (sub_reg & PCIE_CORE_INT_FCE)
-                       dev_dbg(dev, "an error was observed in the flow control advertisements from the other side\n");
-
-               if (sub_reg & PCIE_CORE_INT_CT)
-                       dev_dbg(dev, "a request timed out waiting for completion\n");
-
-               if (sub_reg & PCIE_CORE_INT_UTC)
-                       dev_dbg(dev, "unmapped TC error\n");
-
-               if (sub_reg & PCIE_CORE_INT_MMVC)
-                       dev_dbg(dev, "MSI mask register changes\n");
-
-               rockchip_pcie_write(rockchip, sub_reg, PCIE_CORE_INT_STATUS);
-       } else if (reg & PCIE_CLIENT_INT_PHY) {
-               dev_dbg(dev, "phy link changes\n");
-               rockchip_pcie_update_txcredit_mui(rockchip);
-               rockchip_pcie_clr_bw_int(rockchip);
-       }
-
-       rockchip_pcie_write(rockchip, reg & PCIE_CLIENT_INT_LOCAL,
-                           PCIE_CLIENT_INT_STATUS);
-
-       return IRQ_HANDLED;
-}
-
-static irqreturn_t rockchip_pcie_client_irq_handler(int irq, void *arg)
-{
-       struct rockchip_pcie *rockchip = arg;
-       struct device *dev = rockchip->dev;
-       u32 reg;
-
-       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
-       if (reg & PCIE_CLIENT_INT_LEGACY_DONE)
-               dev_dbg(dev, "legacy done interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_MSG)
-               dev_dbg(dev, "message done interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_HOT_RST)
-               dev_dbg(dev, "hot reset interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_DPA)
-               dev_dbg(dev, "dpa interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_FATAL_ERR)
-               dev_dbg(dev, "fatal error interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_NFATAL_ERR)
-               dev_dbg(dev, "no fatal error interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_CORR_ERR)
-               dev_dbg(dev, "correctable error interrupt received\n");
-
-       if (reg & PCIE_CLIENT_INT_PHY)
-               dev_dbg(dev, "phy interrupt received\n");
-
-       rockchip_pcie_write(rockchip, reg & (PCIE_CLIENT_INT_LEGACY_DONE |
-                             PCIE_CLIENT_INT_MSG | PCIE_CLIENT_INT_HOT_RST |
-                             PCIE_CLIENT_INT_DPA | PCIE_CLIENT_INT_FATAL_ERR |
-                             PCIE_CLIENT_INT_NFATAL_ERR |
-                             PCIE_CLIENT_INT_CORR_ERR |
-                             PCIE_CLIENT_INT_PHY),
-                  PCIE_CLIENT_INT_STATUS);
-
-       return IRQ_HANDLED;
-}
-
-static void rockchip_pcie_legacy_int_handler(struct irq_desc *desc)
-{
-       struct irq_chip *chip = irq_desc_get_chip(desc);
-       struct rockchip_pcie *rockchip = irq_desc_get_handler_data(desc);
-       struct device *dev = rockchip->dev;
-       u32 reg;
-       u32 hwirq;
-       u32 virq;
-
-       chained_irq_enter(chip, desc);
-
-       reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
-       reg = (reg & PCIE_CLIENT_INTR_MASK) >> PCIE_CLIENT_INTR_SHIFT;
-
-       while (reg) {
-               hwirq = ffs(reg) - 1;
-               reg &= ~BIT(hwirq);
-
-               virq = irq_find_mapping(rockchip->irq_domain, hwirq);
-               if (virq)
-                       generic_handle_irq(virq);
-               else
-                       dev_err(dev, "unexpected IRQ, INT%d\n", hwirq);
-       }
-
-       chained_irq_exit(chip, desc);
-}
-
-static int rockchip_pcie_get_phys(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       struct phy *phy;
-       char *name;
-       u32 i;
-
-       phy = devm_phy_get(dev, "pcie-phy");
-       if (!IS_ERR(phy)) {
-               rockchip->legacy_phy = true;
-               rockchip->phys[0] = phy;
-               dev_warn(dev, "legacy phy model is deprecated!\n");
-               return 0;
-       }
-
-       if (PTR_ERR(phy) == -EPROBE_DEFER)
-               return PTR_ERR(phy);
-
-       dev_dbg(dev, "missing legacy phy; search for per-lane PHY\n");
-
-       for (i = 0; i < MAX_LANE_NUM; i++) {
-               name = kasprintf(GFP_KERNEL, "pcie-phy-%u", i);
-               if (!name)
-                       return -ENOMEM;
-
-               phy = devm_of_phy_get(dev, dev->of_node, name);
-               kfree(name);
-
-               if (IS_ERR(phy)) {
-                       if (PTR_ERR(phy) != -EPROBE_DEFER)
-                               dev_err(dev, "missing phy for lane %d: %ld\n",
-                                       i, PTR_ERR(phy));
-                       return PTR_ERR(phy);
-               }
-
-               rockchip->phys[i] = phy;
-       }
-
-       return 0;
-}
-
-static int rockchip_pcie_setup_irq(struct rockchip_pcie *rockchip)
-{
-       int irq, err;
-       struct device *dev = rockchip->dev;
-       struct platform_device *pdev = to_platform_device(dev);
-
-       irq = platform_get_irq_byname(pdev, "sys");
-       if (irq < 0) {
-               dev_err(dev, "missing sys IRQ resource\n");
-               return irq;
-       }
-
-       err = devm_request_irq(dev, irq, rockchip_pcie_subsys_irq_handler,
-                              IRQF_SHARED, "pcie-sys", rockchip);
-       if (err) {
-               dev_err(dev, "failed to request PCIe subsystem IRQ\n");
-               return err;
-       }
-
-       irq = platform_get_irq_byname(pdev, "legacy");
-       if (irq < 0) {
-               dev_err(dev, "missing legacy IRQ resource\n");
-               return irq;
-       }
-
-       irq_set_chained_handler_and_data(irq,
-                                        rockchip_pcie_legacy_int_handler,
-                                        rockchip);
-
-       irq = platform_get_irq_byname(pdev, "client");
-       if (irq < 0) {
-               dev_err(dev, "missing client IRQ resource\n");
-               return irq;
-       }
-
-       err = devm_request_irq(dev, irq, rockchip_pcie_client_irq_handler,
-                              IRQF_SHARED, "pcie-client", rockchip);
-       if (err) {
-               dev_err(dev, "failed to request PCIe client IRQ\n");
-               return err;
-       }
-
-       return 0;
-}
-
-/**
- * rockchip_pcie_parse_dt - Parse Device Tree
- * @rockchip: PCIe port information
- *
- * Return: '0' on success and error value on failure
- */
-static int rockchip_pcie_parse_dt(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       struct platform_device *pdev = to_platform_device(dev);
-       struct device_node *node = dev->of_node;
-       struct resource *regs;
-       int err;
-
-       regs = platform_get_resource_byname(pdev,
-                                           IORESOURCE_MEM,
-                                           "axi-base");
-       rockchip->reg_base = devm_pci_remap_cfg_resource(dev, regs);
-       if (IS_ERR(rockchip->reg_base))
-               return PTR_ERR(rockchip->reg_base);
-
-       regs = platform_get_resource_byname(pdev,
-                                           IORESOURCE_MEM,
-                                           "apb-base");
-       rockchip->apb_base = devm_ioremap_resource(dev, regs);
-       if (IS_ERR(rockchip->apb_base))
-               return PTR_ERR(rockchip->apb_base);
-
-       err = rockchip_pcie_get_phys(rockchip);
-       if (err)
-               return err;
-
-       rockchip->lanes = 1;
-       err = of_property_read_u32(node, "num-lanes", &rockchip->lanes);
-       if (!err && (rockchip->lanes == 0 ||
-                    rockchip->lanes == 3 ||
-                    rockchip->lanes > 4)) {
-               dev_warn(dev, "invalid num-lanes, default to use one lane\n");
-               rockchip->lanes = 1;
-       }
-
-       rockchip->link_gen = of_pci_get_max_link_speed(node);
-       if (rockchip->link_gen < 0 || rockchip->link_gen > 2)
-               rockchip->link_gen = 2;
-
-       rockchip->core_rst = devm_reset_control_get_exclusive(dev, "core");
-       if (IS_ERR(rockchip->core_rst)) {
-               if (PTR_ERR(rockchip->core_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing core reset property in node\n");
-               return PTR_ERR(rockchip->core_rst);
-       }
-
-       rockchip->mgmt_rst = devm_reset_control_get_exclusive(dev, "mgmt");
-       if (IS_ERR(rockchip->mgmt_rst)) {
-               if (PTR_ERR(rockchip->mgmt_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing mgmt reset property in node\n");
-               return PTR_ERR(rockchip->mgmt_rst);
-       }
-
-       rockchip->mgmt_sticky_rst = devm_reset_control_get_exclusive(dev,
-                                                                    "mgmt-sticky");
-       if (IS_ERR(rockchip->mgmt_sticky_rst)) {
-               if (PTR_ERR(rockchip->mgmt_sticky_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing mgmt-sticky reset property in node\n");
-               return PTR_ERR(rockchip->mgmt_sticky_rst);
-       }
-
-       rockchip->pipe_rst = devm_reset_control_get_exclusive(dev, "pipe");
-       if (IS_ERR(rockchip->pipe_rst)) {
-               if (PTR_ERR(rockchip->pipe_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing pipe reset property in node\n");
-               return PTR_ERR(rockchip->pipe_rst);
-       }
-
-       rockchip->pm_rst = devm_reset_control_get_exclusive(dev, "pm");
-       if (IS_ERR(rockchip->pm_rst)) {
-               if (PTR_ERR(rockchip->pm_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing pm reset property in node\n");
-               return PTR_ERR(rockchip->pm_rst);
-       }
-
-       rockchip->pclk_rst = devm_reset_control_get_exclusive(dev, "pclk");
-       if (IS_ERR(rockchip->pclk_rst)) {
-               if (PTR_ERR(rockchip->pclk_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing pclk reset property in node\n");
-               return PTR_ERR(rockchip->pclk_rst);
-       }
-
-       rockchip->aclk_rst = devm_reset_control_get_exclusive(dev, "aclk");
-       if (IS_ERR(rockchip->aclk_rst)) {
-               if (PTR_ERR(rockchip->aclk_rst) != -EPROBE_DEFER)
-                       dev_err(dev, "missing aclk reset property in node\n");
-               return PTR_ERR(rockchip->aclk_rst);
-       }
-
-       rockchip->ep_gpio = devm_gpiod_get(dev, "ep", GPIOD_OUT_HIGH);
-       if (IS_ERR(rockchip->ep_gpio)) {
-               dev_err(dev, "missing ep-gpios property in node\n");
-               return PTR_ERR(rockchip->ep_gpio);
-       }
-
-       rockchip->aclk_pcie = devm_clk_get(dev, "aclk");
-       if (IS_ERR(rockchip->aclk_pcie)) {
-               dev_err(dev, "aclk clock not found\n");
-               return PTR_ERR(rockchip->aclk_pcie);
-       }
-
-       rockchip->aclk_perf_pcie = devm_clk_get(dev, "aclk-perf");
-       if (IS_ERR(rockchip->aclk_perf_pcie)) {
-               dev_err(dev, "aclk_perf clock not found\n");
-               return PTR_ERR(rockchip->aclk_perf_pcie);
-       }
-
-       rockchip->hclk_pcie = devm_clk_get(dev, "hclk");
-       if (IS_ERR(rockchip->hclk_pcie)) {
-               dev_err(dev, "hclk clock not found\n");
-               return PTR_ERR(rockchip->hclk_pcie);
-       }
-
-       rockchip->clk_pcie_pm = devm_clk_get(dev, "pm");
-       if (IS_ERR(rockchip->clk_pcie_pm)) {
-               dev_err(dev, "pm clock not found\n");
-               return PTR_ERR(rockchip->clk_pcie_pm);
-       }
-
-       err = rockchip_pcie_setup_irq(rockchip);
-       if (err)
-               return err;
-
-       rockchip->vpcie12v = devm_regulator_get_optional(dev, "vpcie12v");
-       if (IS_ERR(rockchip->vpcie12v)) {
-               if (PTR_ERR(rockchip->vpcie12v) == -EPROBE_DEFER)
-                       return -EPROBE_DEFER;
-               dev_info(dev, "no vpcie12v regulator found\n");
-       }
-
-       rockchip->vpcie3v3 = devm_regulator_get_optional(dev, "vpcie3v3");
-       if (IS_ERR(rockchip->vpcie3v3)) {
-               if (PTR_ERR(rockchip->vpcie3v3) == -EPROBE_DEFER)
-                       return -EPROBE_DEFER;
-               dev_info(dev, "no vpcie3v3 regulator found\n");
-       }
-
-       rockchip->vpcie1v8 = devm_regulator_get_optional(dev, "vpcie1v8");
-       if (IS_ERR(rockchip->vpcie1v8)) {
-               if (PTR_ERR(rockchip->vpcie1v8) == -EPROBE_DEFER)
-                       return -EPROBE_DEFER;
-               dev_info(dev, "no vpcie1v8 regulator found\n");
-       }
-
-       rockchip->vpcie0v9 = devm_regulator_get_optional(dev, "vpcie0v9");
-       if (IS_ERR(rockchip->vpcie0v9)) {
-               if (PTR_ERR(rockchip->vpcie0v9) == -EPROBE_DEFER)
-                       return -EPROBE_DEFER;
-               dev_info(dev, "no vpcie0v9 regulator found\n");
-       }
-
-       return 0;
-}
-
-static int rockchip_pcie_set_vpcie(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       int err;
-
-       if (!IS_ERR(rockchip->vpcie12v)) {
-               err = regulator_enable(rockchip->vpcie12v);
-               if (err) {
-                       dev_err(dev, "fail to enable vpcie12v regulator\n");
-                       goto err_out;
-               }
-       }
-
-       if (!IS_ERR(rockchip->vpcie3v3)) {
-               err = regulator_enable(rockchip->vpcie3v3);
-               if (err) {
-                       dev_err(dev, "fail to enable vpcie3v3 regulator\n");
-                       goto err_disable_12v;
-               }
-       }
-
-       if (!IS_ERR(rockchip->vpcie1v8)) {
-               err = regulator_enable(rockchip->vpcie1v8);
-               if (err) {
-                       dev_err(dev, "fail to enable vpcie1v8 regulator\n");
-                       goto err_disable_3v3;
-               }
-       }
-
-       if (!IS_ERR(rockchip->vpcie0v9)) {
-               err = regulator_enable(rockchip->vpcie0v9);
-               if (err) {
-                       dev_err(dev, "fail to enable vpcie0v9 regulator\n");
-                       goto err_disable_1v8;
-               }
-       }
-
-       return 0;
-
-err_disable_1v8:
-       if (!IS_ERR(rockchip->vpcie1v8))
-               regulator_disable(rockchip->vpcie1v8);
-err_disable_3v3:
-       if (!IS_ERR(rockchip->vpcie3v3))
-               regulator_disable(rockchip->vpcie3v3);
-err_disable_12v:
-       if (!IS_ERR(rockchip->vpcie12v))
-               regulator_disable(rockchip->vpcie12v);
-err_out:
-       return err;
-}
-
-static void rockchip_pcie_enable_interrupts(struct rockchip_pcie *rockchip)
-{
-       rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) &
-                           (~PCIE_CLIENT_INT_CLI), PCIE_CLIENT_INT_MASK);
-       rockchip_pcie_write(rockchip, (u32)(~PCIE_CORE_INT),
-                           PCIE_CORE_INT_MASK);
-
-       rockchip_pcie_enable_bw_int(rockchip);
-}
-
-static int rockchip_pcie_intx_map(struct irq_domain *domain, unsigned int irq,
-                                 irq_hw_number_t hwirq)
-{
-       irq_set_chip_and_handler(irq, &dummy_irq_chip, handle_simple_irq);
-       irq_set_chip_data(irq, domain->host_data);
-
-       return 0;
-}
-
-static const struct irq_domain_ops intx_domain_ops = {
-       .map = rockchip_pcie_intx_map,
-};
-
-static int rockchip_pcie_init_irq_domain(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       struct device_node *intc = of_get_next_child(dev->of_node, NULL);
-
-       if (!intc) {
-               dev_err(dev, "missing child interrupt-controller node\n");
-               return -EINVAL;
-       }
-
-       rockchip->irq_domain = irq_domain_add_linear(intc, PCI_NUM_INTX,
-                                                   &intx_domain_ops, rockchip);
-       if (!rockchip->irq_domain) {
-               dev_err(dev, "failed to get a INTx IRQ domain\n");
-               return -EINVAL;
-       }
-
-       return 0;
-}
-
-static int rockchip_pcie_prog_ob_atu(struct rockchip_pcie *rockchip,
-                                    int region_no, int type, u8 num_pass_bits,
-                                    u32 lower_addr, u32 upper_addr)
-{
-       u32 ob_addr_0;
-       u32 ob_addr_1;
-       u32 ob_desc_0;
-       u32 aw_offset;
-
-       if (region_no >= MAX_AXI_WRAPPER_REGION_NUM)
-               return -EINVAL;
-       if (num_pass_bits + 1 < 8)
-               return -EINVAL;
-       if (num_pass_bits > 63)
-               return -EINVAL;
-       if (region_no == 0) {
-               if (AXI_REGION_0_SIZE < (2ULL << num_pass_bits))
-                       return -EINVAL;
-       }
-       if (region_no != 0) {
-               if (AXI_REGION_SIZE < (2ULL << num_pass_bits))
-                       return -EINVAL;
-       }
-
-       aw_offset = (region_no << OB_REG_SIZE_SHIFT);
-
-       ob_addr_0 = num_pass_bits & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS;
-       ob_addr_0 |= lower_addr & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR;
-       ob_addr_1 = upper_addr;
-       ob_desc_0 = (1 << 23 | type);
-
-       rockchip_pcie_write(rockchip, ob_addr_0,
-                           PCIE_CORE_OB_REGION_ADDR0 + aw_offset);
-       rockchip_pcie_write(rockchip, ob_addr_1,
-                           PCIE_CORE_OB_REGION_ADDR1 + aw_offset);
-       rockchip_pcie_write(rockchip, ob_desc_0,
-                           PCIE_CORE_OB_REGION_DESC0 + aw_offset);
-       rockchip_pcie_write(rockchip, 0,
-                           PCIE_CORE_OB_REGION_DESC1 + aw_offset);
-
-       return 0;
-}
-
-static int rockchip_pcie_prog_ib_atu(struct rockchip_pcie *rockchip,
-                                    int region_no, u8 num_pass_bits,
-                                    u32 lower_addr, u32 upper_addr)
-{
-       u32 ib_addr_0;
-       u32 ib_addr_1;
-       u32 aw_offset;
-
-       if (region_no > MAX_AXI_IB_ROOTPORT_REGION_NUM)
-               return -EINVAL;
-       if (num_pass_bits + 1 < MIN_AXI_ADDR_BITS_PASSED)
-               return -EINVAL;
-       if (num_pass_bits > 63)
-               return -EINVAL;
-
-       aw_offset = (region_no << IB_ROOT_PORT_REG_SIZE_SHIFT);
-
-       ib_addr_0 = num_pass_bits & PCIE_CORE_IB_REGION_ADDR0_NUM_BITS;
-       ib_addr_0 |= (lower_addr << 8) & PCIE_CORE_IB_REGION_ADDR0_LO_ADDR;
-       ib_addr_1 = upper_addr;
-
-       rockchip_pcie_write(rockchip, ib_addr_0, PCIE_RP_IB_ADDR0 + aw_offset);
-       rockchip_pcie_write(rockchip, ib_addr_1, PCIE_RP_IB_ADDR1 + aw_offset);
-
-       return 0;
-}
-
-static int rockchip_pcie_cfg_atu(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       int offset;
-       int err;
-       int reg_no;
-
-       rockchip_pcie_cfg_configuration_accesses(rockchip,
-                                                AXI_WRAPPER_TYPE0_CFG);
-
-       for (reg_no = 0; reg_no < (rockchip->mem_size >> 20); reg_no++) {
-               err = rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1,
-                                               AXI_WRAPPER_MEM_WRITE,
-                                               20 - 1,
-                                               rockchip->mem_bus_addr +
-                                               (reg_no << 20),
-                                               0);
-               if (err) {
-                       dev_err(dev, "program RC mem outbound ATU failed\n");
-                       return err;
-               }
-       }
-
-       err = rockchip_pcie_prog_ib_atu(rockchip, 2, 32 - 1, 0x0, 0);
-       if (err) {
-               dev_err(dev, "program RC mem inbound ATU failed\n");
-               return err;
-       }
-
-       offset = rockchip->mem_size >> 20;
-       for (reg_no = 0; reg_no < (rockchip->io_size >> 20); reg_no++) {
-               err = rockchip_pcie_prog_ob_atu(rockchip,
-                                               reg_no + 1 + offset,
-                                               AXI_WRAPPER_IO_WRITE,
-                                               20 - 1,
-                                               rockchip->io_bus_addr +
-                                               (reg_no << 20),
-                                               0);
-               if (err) {
-                       dev_err(dev, "program RC io outbound ATU failed\n");
-                       return err;
-               }
-       }
-
-       /* assign message regions */
-       rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1 + offset,
-                                 AXI_WRAPPER_NOR_MSG,
-                                 20 - 1, 0, 0);
-
-       rockchip->msg_bus_addr = rockchip->mem_bus_addr +
-                                       ((reg_no + offset) << 20);
-       return err;
-}
-
-static int rockchip_pcie_wait_l2(struct rockchip_pcie *rockchip)
-{
-       u32 value;
-       int err;
-
-       /* send PME_TURN_OFF message */
-       writel(0x0, rockchip->msg_region + PCIE_RC_SEND_PME_OFF);
-
-       /* read LTSSM and wait for falling into L2 link state */
-       err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_DEBUG_OUT_0,
-                                value, PCIE_LINK_IS_L2(value), 20,
-                                jiffies_to_usecs(5 * HZ));
-       if (err) {
-               dev_err(rockchip->dev, "PCIe link enter L2 timeout!\n");
-               return err;
-       }
-
-       return 0;
-}
-
-static int rockchip_pcie_enable_clocks(struct rockchip_pcie *rockchip)
-{
-       struct device *dev = rockchip->dev;
-       int err;
-
-       err = clk_prepare_enable(rockchip->aclk_pcie);
-       if (err) {
-               dev_err(dev, "unable to enable aclk_pcie clock\n");
-               return err;
-       }
-
-       err = clk_prepare_enable(rockchip->aclk_perf_pcie);
-       if (err) {
-               dev_err(dev, "unable to enable aclk_perf_pcie clock\n");
-               goto err_aclk_perf_pcie;
-       }
-
-       err = clk_prepare_enable(rockchip->hclk_pcie);
-       if (err) {
-               dev_err(dev, "unable to enable hclk_pcie clock\n");
-               goto err_hclk_pcie;
-       }
+       err = clk_prepare_enable(rockchip->hclk_pcie);
+       if (err) {
+               dev_err(dev, "unable to enable hclk_pcie clock\n");
+               goto err_hclk_pcie;
+       }
 
        err = clk_prepare_enable(rockchip->clk_pcie_pm);
        if (err) {
@@ -1434,8 +389,9 @@ static int rockchip_pcie_enable_clocks(struct rockchip_pcie *rockchip)
        clk_disable_unprepare(rockchip->aclk_pcie);
        return err;
 }
+EXPORT_SYMBOL_GPL(rockchip_pcie_enable_clocks);
 
-static void rockchip_pcie_disable_clocks(void *data)
+void rockchip_pcie_disable_clocks(void *data)
 {
        struct rockchip_pcie *rockchip = data;
 
@@ -1444,267 +400,25 @@ static void rockchip_pcie_disable_clocks(void *data)
        clk_disable_unprepare(rockchip->aclk_perf_pcie);
        clk_disable_unprepare(rockchip->aclk_pcie);
 }
+EXPORT_SYMBOL_GPL(rockchip_pcie_disable_clocks);
 
-static int __maybe_unused rockchip_pcie_suspend_noirq(struct device *dev)
-{
-       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
-       int ret;
-
-       /* disable core and cli int since we don't need to ack PME_ACK */
-       rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) |
-                           PCIE_CLIENT_INT_CLI, PCIE_CLIENT_INT_MASK);
-       rockchip_pcie_write(rockchip, (u32)PCIE_CORE_INT, PCIE_CORE_INT_MASK);
-
-       ret = rockchip_pcie_wait_l2(rockchip);
-       if (ret) {
-               rockchip_pcie_enable_interrupts(rockchip);
-               return ret;
-       }
-
-       rockchip_pcie_deinit_phys(rockchip);
-
-       rockchip_pcie_disable_clocks(rockchip);
-
-       if (!IS_ERR(rockchip->vpcie0v9))
-               regulator_disable(rockchip->vpcie0v9);
-
-       return ret;
-}
-
-static int __maybe_unused rockchip_pcie_resume_noirq(struct device *dev)
-{
-       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
-       int err;
-
-       if (!IS_ERR(rockchip->vpcie0v9)) {
-               err = regulator_enable(rockchip->vpcie0v9);
-               if (err) {
-                       dev_err(dev, "fail to enable vpcie0v9 regulator\n");
-                       return err;
-               }
-       }
-
-       err = rockchip_pcie_enable_clocks(rockchip);
-       if (err)
-               goto err_disable_0v9;
-
-       err = rockchip_pcie_init_port(rockchip);
-       if (err)
-               goto err_pcie_resume;
-
-       err = rockchip_pcie_cfg_atu(rockchip);
-       if (err)
-               goto err_err_deinit_port;
-
-       /* Need this to enter L1 again */
-       rockchip_pcie_update_txcredit_mui(rockchip);
-       rockchip_pcie_enable_interrupts(rockchip);
-
-       return 0;
-
-err_err_deinit_port:
-       rockchip_pcie_deinit_phys(rockchip);
-err_pcie_resume:
-       rockchip_pcie_disable_clocks(rockchip);
-err_disable_0v9:
-       if (!IS_ERR(rockchip->vpcie0v9))
-               regulator_disable(rockchip->vpcie0v9);
-       return err;
-}
-
-static int rockchip_pcie_probe(struct platform_device *pdev)
-{
-       struct rockchip_pcie *rockchip;
-       struct device *dev = &pdev->dev;
-       struct pci_bus *bus, *child;
-       struct pci_host_bridge *bridge;
-       struct resource_entry *win;
-       resource_size_t io_base;
-       struct resource *mem;
-       struct resource *io;
-       int err;
-
-       LIST_HEAD(res);
-
-       if (!dev->of_node)
-               return -ENODEV;
-
-       bridge = devm_pci_alloc_host_bridge(dev, sizeof(*rockchip));
-       if (!bridge)
-               return -ENOMEM;
-
-       rockchip = pci_host_bridge_priv(bridge);
-
-       platform_set_drvdata(pdev, rockchip);
-       rockchip->dev = dev;
-
-       err = rockchip_pcie_parse_dt(rockchip);
-       if (err)
-               return err;
-
-       err = rockchip_pcie_enable_clocks(rockchip);
-       if (err)
-               return err;
-
-       err = rockchip_pcie_set_vpcie(rockchip);
-       if (err) {
-               dev_err(dev, "failed to set vpcie regulator\n");
-               goto err_set_vpcie;
-       }
-
-       err = rockchip_pcie_init_port(rockchip);
-       if (err)
-               goto err_vpcie;
-
-       rockchip_pcie_enable_interrupts(rockchip);
-
-       err = rockchip_pcie_init_irq_domain(rockchip);
-       if (err < 0)
-               goto err_deinit_port;
-
-       err = devm_of_pci_get_host_bridge_resources(dev, 0, 0xff,
-                                                   &res, &io_base);
-       if (err)
-               goto err_remove_irq_domain;
-
-       err = devm_request_pci_bus_resources(dev, &res);
-       if (err)
-               goto err_free_res;
-
-       /* Get the I/O and memory ranges from DT */
-       resource_list_for_each_entry(win, &res) {
-               switch (resource_type(win->res)) {
-               case IORESOURCE_IO:
-                       io = win->res;
-                       io->name = "I/O";
-                       rockchip->io_size = resource_size(io);
-                       rockchip->io_bus_addr = io->start - win->offset;
-                       err = pci_remap_iospace(io, io_base);
-                       if (err) {
-                               dev_warn(dev, "error %d: failed to map resource %pR\n",
-                                        err, io);
-                               continue;
-                       }
-                       rockchip->io = io;
-                       break;
-               case IORESOURCE_MEM:
-                       mem = win->res;
-                       mem->name = "MEM";
-                       rockchip->mem_size = resource_size(mem);
-                       rockchip->mem_bus_addr = mem->start - win->offset;
-                       break;
-               case IORESOURCE_BUS:
-                       rockchip->root_bus_nr = win->res->start;
-                       break;
-               default:
-                       continue;
-               }
-       }
-
-       err = rockchip_pcie_cfg_atu(rockchip);
-       if (err)
-               goto err_unmap_iospace;
-
-       rockchip->msg_region = devm_ioremap(dev, rockchip->msg_bus_addr, SZ_1M);
-       if (!rockchip->msg_region) {
-               err = -ENOMEM;
-               goto err_unmap_iospace;
-       }
-
-       list_splice_init(&res, &bridge->windows);
-       bridge->dev.parent = dev;
-       bridge->sysdata = rockchip;
-       bridge->busnr = 0;
-       bridge->ops = &rockchip_pcie_ops;
-       bridge->map_irq = of_irq_parse_and_map_pci;
-       bridge->swizzle_irq = pci_common_swizzle;
-
-       err = pci_scan_root_bus_bridge(bridge);
-       if (err < 0)
-               goto err_unmap_iospace;
-
-       bus = bridge->bus;
-
-       rockchip->root_bus = bus;
-
-       pci_bus_size_bridges(bus);
-       pci_bus_assign_resources(bus);
-       list_for_each_entry(child, &bus->children, node)
-               pcie_bus_configure_settings(child);
-
-       pci_bus_add_devices(bus);
-       return 0;
-
-err_unmap_iospace:
-       pci_unmap_iospace(rockchip->io);
-err_free_res:
-       pci_free_resource_list(&res);
-err_remove_irq_domain:
-       irq_domain_remove(rockchip->irq_domain);
-err_deinit_port:
-       rockchip_pcie_deinit_phys(rockchip);
-err_vpcie:
-       if (!IS_ERR(rockchip->vpcie12v))
-               regulator_disable(rockchip->vpcie12v);
-       if (!IS_ERR(rockchip->vpcie3v3))
-               regulator_disable(rockchip->vpcie3v3);
-       if (!IS_ERR(rockchip->vpcie1v8))
-               regulator_disable(rockchip->vpcie1v8);
-       if (!IS_ERR(rockchip->vpcie0v9))
-               regulator_disable(rockchip->vpcie0v9);
-err_set_vpcie:
-       rockchip_pcie_disable_clocks(rockchip);
-       return err;
-}
-
-static int rockchip_pcie_remove(struct platform_device *pdev)
+void rockchip_pcie_cfg_configuration_accesses(
+               struct rockchip_pcie *rockchip, u32 type)
 {
-       struct device *dev = &pdev->dev;
-       struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
-
-       pci_stop_root_bus(rockchip->root_bus);
-       pci_remove_root_bus(rockchip->root_bus);
-       pci_unmap_iospace(rockchip->io);
-       irq_domain_remove(rockchip->irq_domain);
-
-       rockchip_pcie_deinit_phys(rockchip);
-
-       rockchip_pcie_disable_clocks(rockchip);
+       u32 ob_desc_0;
 
-       if (!IS_ERR(rockchip->vpcie12v))
-               regulator_disable(rockchip->vpcie12v);
-       if (!IS_ERR(rockchip->vpcie3v3))
-               regulator_disable(rockchip->vpcie3v3);
-       if (!IS_ERR(rockchip->vpcie1v8))
-               regulator_disable(rockchip->vpcie1v8);
-       if (!IS_ERR(rockchip->vpcie0v9))
-               regulator_disable(rockchip->vpcie0v9);
+       /* Configuration Accesses for region 0 */
+       rockchip_pcie_write(rockchip, 0x0, PCIE_RC_BAR_CONF);
 
-       return 0;
+       rockchip_pcie_write(rockchip,
+                           (RC_REGION_0_ADDR_TRANS_L + RC_REGION_0_PASS_BITS),
+                           PCIE_CORE_OB_REGION_ADDR0);
+       rockchip_pcie_write(rockchip, RC_REGION_0_ADDR_TRANS_H,
+                           PCIE_CORE_OB_REGION_ADDR1);
+       ob_desc_0 = rockchip_pcie_read(rockchip, PCIE_CORE_OB_REGION_DESC0);
+       ob_desc_0 &= ~(RC_REGION_0_TYPE_MASK);
+       ob_desc_0 |= (type | (0x1 << 23));
+       rockchip_pcie_write(rockchip, ob_desc_0, PCIE_CORE_OB_REGION_DESC0);
+       rockchip_pcie_write(rockchip, 0x0, PCIE_CORE_OB_REGION_DESC1);
 }
-
-static const struct dev_pm_ops rockchip_pcie_pm_ops = {
-       SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(rockchip_pcie_suspend_noirq,
-                                     rockchip_pcie_resume_noirq)
-};
-
-static const struct of_device_id rockchip_pcie_of_match[] = {
-       { .compatible = "rockchip,rk3399-pcie", },
-       {}
-};
-MODULE_DEVICE_TABLE(of, rockchip_pcie_of_match);
-
-static struct platform_driver rockchip_pcie_driver = {
-       .driver = {
-               .name = "rockchip-pcie",
-               .of_match_table = rockchip_pcie_of_match,
-               .pm = &rockchip_pcie_pm_ops,
-       },
-       .probe = rockchip_pcie_probe,
-       .remove = rockchip_pcie_remove,
-};
-module_platform_driver(rockchip_pcie_driver);
-
-MODULE_AUTHOR("Rockchip Inc");
-MODULE_DESCRIPTION("Rockchip AXI PCIe driver");
-MODULE_LICENSE("GPL v2");
+EXPORT_SYMBOL_GPL(rockchip_pcie_cfg_configuration_accesses);
diff --git a/drivers/pci/host/pcie-rockchip.h b/drivers/pci/host/pcie-rockchip.h
new file mode 100644 (file)
index 0000000..8e87a05
--- /dev/null
@@ -0,0 +1,338 @@
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Rockchip AXI PCIe controller driver
+ *
+ * Copyright (c) 2018 Rockchip, Inc.
+ *
+ * Author: Shawn Lin <shawn.lin@rock-chips.com>
+ *
+ */
+
+#ifndef _PCIE_ROCKCHIP_H
+#define _PCIE_ROCKCHIP_H
+
+#include <linux/kernel.h>
+#include <linux/pci.h>
+
+/*
+ * The upper 16 bits of PCIE_CLIENT_CONFIG are a write mask for the lower 16
+ * bits.  This allows atomic updates of the register without locking.
+ */
+#define HIWORD_UPDATE(mask, val)       (((mask) << 16) | (val))
+#define HIWORD_UPDATE_BIT(val)         HIWORD_UPDATE(val, val)
+
+#define ENCODE_LANES(x)                        ((((x) >> 1) & 3) << 4)
+#define MAX_LANE_NUM                   4
+#define MAX_REGION_LIMIT               32
+#define MIN_EP_APERTURE                        28
+
+#define PCIE_CLIENT_BASE               0x0
+#define PCIE_CLIENT_CONFIG             (PCIE_CLIENT_BASE + 0x00)
+#define   PCIE_CLIENT_CONF_ENABLE        HIWORD_UPDATE_BIT(0x0001)
+#define   PCIE_CLIENT_CONF_DISABLE       HIWORD_UPDATE(0x0001, 0)
+#define   PCIE_CLIENT_LINK_TRAIN_ENABLE          HIWORD_UPDATE_BIT(0x0002)
+#define   PCIE_CLIENT_ARI_ENABLE         HIWORD_UPDATE_BIT(0x0008)
+#define   PCIE_CLIENT_CONF_LANE_NUM(x)   HIWORD_UPDATE(0x0030, ENCODE_LANES(x))
+#define   PCIE_CLIENT_MODE_RC            HIWORD_UPDATE_BIT(0x0040)
+#define   PCIE_CLIENT_MODE_EP            HIWORD_UPDATE(0x0040, 0)
+#define   PCIE_CLIENT_GEN_SEL_1                  HIWORD_UPDATE(0x0080, 0)
+#define   PCIE_CLIENT_GEN_SEL_2                  HIWORD_UPDATE_BIT(0x0080)
+#define PCIE_CLIENT_DEBUG_OUT_0                (PCIE_CLIENT_BASE + 0x3c)
+#define   PCIE_CLIENT_DEBUG_LTSSM_MASK         GENMASK(5, 0)
+#define   PCIE_CLIENT_DEBUG_LTSSM_L1           0x18
+#define   PCIE_CLIENT_DEBUG_LTSSM_L2           0x19
+#define PCIE_CLIENT_BASIC_STATUS1      (PCIE_CLIENT_BASE + 0x48)
+#define   PCIE_CLIENT_LINK_STATUS_UP           0x00300000
+#define   PCIE_CLIENT_LINK_STATUS_MASK         0x00300000
+#define PCIE_CLIENT_INT_MASK           (PCIE_CLIENT_BASE + 0x4c)
+#define PCIE_CLIENT_INT_STATUS         (PCIE_CLIENT_BASE + 0x50)
+#define   PCIE_CLIENT_INTR_MASK                        GENMASK(8, 5)
+#define   PCIE_CLIENT_INTR_SHIFT               5
+#define   PCIE_CLIENT_INT_LEGACY_DONE          BIT(15)
+#define   PCIE_CLIENT_INT_MSG                  BIT(14)
+#define   PCIE_CLIENT_INT_HOT_RST              BIT(13)
+#define   PCIE_CLIENT_INT_DPA                  BIT(12)
+#define   PCIE_CLIENT_INT_FATAL_ERR            BIT(11)
+#define   PCIE_CLIENT_INT_NFATAL_ERR           BIT(10)
+#define   PCIE_CLIENT_INT_CORR_ERR             BIT(9)
+#define   PCIE_CLIENT_INT_INTD                 BIT(8)
+#define   PCIE_CLIENT_INT_INTC                 BIT(7)
+#define   PCIE_CLIENT_INT_INTB                 BIT(6)
+#define   PCIE_CLIENT_INT_INTA                 BIT(5)
+#define   PCIE_CLIENT_INT_LOCAL                        BIT(4)
+#define   PCIE_CLIENT_INT_UDMA                 BIT(3)
+#define   PCIE_CLIENT_INT_PHY                  BIT(2)
+#define   PCIE_CLIENT_INT_HOT_PLUG             BIT(1)
+#define   PCIE_CLIENT_INT_PWR_STCG             BIT(0)
+
+#define PCIE_CLIENT_INT_LEGACY \
+       (PCIE_CLIENT_INT_INTA | PCIE_CLIENT_INT_INTB | \
+       PCIE_CLIENT_INT_INTC | PCIE_CLIENT_INT_INTD)
+
+#define PCIE_CLIENT_INT_CLI \
+       (PCIE_CLIENT_INT_CORR_ERR | PCIE_CLIENT_INT_NFATAL_ERR | \
+       PCIE_CLIENT_INT_FATAL_ERR | PCIE_CLIENT_INT_DPA | \
+       PCIE_CLIENT_INT_HOT_RST | PCIE_CLIENT_INT_MSG | \
+       PCIE_CLIENT_INT_LEGACY_DONE | PCIE_CLIENT_INT_LEGACY | \
+       PCIE_CLIENT_INT_PHY)
+
+#define PCIE_CORE_CTRL_MGMT_BASE       0x900000
+#define PCIE_CORE_CTRL                 (PCIE_CORE_CTRL_MGMT_BASE + 0x000)
+#define   PCIE_CORE_PL_CONF_SPEED_5G           0x00000008
+#define   PCIE_CORE_PL_CONF_SPEED_MASK         0x00000018
+#define   PCIE_CORE_PL_CONF_LANE_MASK          0x00000006
+#define   PCIE_CORE_PL_CONF_LANE_SHIFT         1
+#define PCIE_CORE_CTRL_PLC1            (PCIE_CORE_CTRL_MGMT_BASE + 0x004)
+#define   PCIE_CORE_CTRL_PLC1_FTS_MASK         GENMASK(23, 8)
+#define   PCIE_CORE_CTRL_PLC1_FTS_SHIFT                8
+#define   PCIE_CORE_CTRL_PLC1_FTS_CNT          0xffff
+#define PCIE_CORE_TXCREDIT_CFG1                (PCIE_CORE_CTRL_MGMT_BASE + 0x020)
+#define   PCIE_CORE_TXCREDIT_CFG1_MUI_MASK     0xFFFF0000
+#define   PCIE_CORE_TXCREDIT_CFG1_MUI_SHIFT    16
+#define   PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(x) \
+               (((x) >> 3) << PCIE_CORE_TXCREDIT_CFG1_MUI_SHIFT)
+#define PCIE_CORE_LANE_MAP             (PCIE_CORE_CTRL_MGMT_BASE + 0x200)
+#define   PCIE_CORE_LANE_MAP_MASK              0x0000000f
+#define   PCIE_CORE_LANE_MAP_REVERSE           BIT(16)
+#define PCIE_CORE_INT_STATUS           (PCIE_CORE_CTRL_MGMT_BASE + 0x20c)
+#define   PCIE_CORE_INT_PRFPE                  BIT(0)
+#define   PCIE_CORE_INT_CRFPE                  BIT(1)
+#define   PCIE_CORE_INT_RRPE                   BIT(2)
+#define   PCIE_CORE_INT_PRFO                   BIT(3)
+#define   PCIE_CORE_INT_CRFO                   BIT(4)
+#define   PCIE_CORE_INT_RT                     BIT(5)
+#define   PCIE_CORE_INT_RTR                    BIT(6)
+#define   PCIE_CORE_INT_PE                     BIT(7)
+#define   PCIE_CORE_INT_MTR                    BIT(8)
+#define   PCIE_CORE_INT_UCR                    BIT(9)
+#define   PCIE_CORE_INT_FCE                    BIT(10)
+#define   PCIE_CORE_INT_CT                     BIT(11)
+#define   PCIE_CORE_INT_UTC                    BIT(18)
+#define   PCIE_CORE_INT_MMVC                   BIT(19)
+#define PCIE_CORE_CONFIG_VENDOR                (PCIE_CORE_CTRL_MGMT_BASE + 0x44)
+#define PCIE_CORE_INT_MASK             (PCIE_CORE_CTRL_MGMT_BASE + 0x210)
+#define PCIE_CORE_PHY_FUNC_CFG         (PCIE_CORE_CTRL_MGMT_BASE + 0x2c0)
+#define PCIE_RC_BAR_CONF               (PCIE_CORE_CTRL_MGMT_BASE + 0x300)
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_DISABLED               0x0
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_IO_32BITS              0x1
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_32BITS             0x4
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_32BITS    0x5
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_64BITS             0x6
+#define ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_64BITS    0x7
+
+#define PCIE_CORE_INT \
+               (PCIE_CORE_INT_PRFPE | PCIE_CORE_INT_CRFPE | \
+                PCIE_CORE_INT_RRPE | PCIE_CORE_INT_CRFO | \
+                PCIE_CORE_INT_RT | PCIE_CORE_INT_RTR | \
+                PCIE_CORE_INT_PE | PCIE_CORE_INT_MTR | \
+                PCIE_CORE_INT_UCR | PCIE_CORE_INT_FCE | \
+                PCIE_CORE_INT_CT | PCIE_CORE_INT_UTC | \
+                PCIE_CORE_INT_MMVC)
+
+#define PCIE_RC_RP_ATS_BASE            0x400000
+#define PCIE_RC_CONFIG_NORMAL_BASE     0x800000
+#define PCIE_RC_CONFIG_BASE            0xa00000
+#define PCIE_RC_CONFIG_RID_CCR         (PCIE_RC_CONFIG_BASE + 0x08)
+#define   PCIE_RC_CONFIG_SCC_SHIFT             16
+#define PCIE_RC_CONFIG_DCR             (PCIE_RC_CONFIG_BASE + 0xc4)
+#define   PCIE_RC_CONFIG_DCR_CSPL_SHIFT                18
+#define   PCIE_RC_CONFIG_DCR_CSPL_LIMIT                0xff
+#define   PCIE_RC_CONFIG_DCR_CPLS_SHIFT                26
+#define PCIE_RC_CONFIG_DCSR            (PCIE_RC_CONFIG_BASE + 0xc8)
+#define   PCIE_RC_CONFIG_DCSR_MPS_MASK         GENMASK(7, 5)
+#define   PCIE_RC_CONFIG_DCSR_MPS_256          (0x1 << 5)
+#define PCIE_RC_CONFIG_LINK_CAP                (PCIE_RC_CONFIG_BASE + 0xcc)
+#define   PCIE_RC_CONFIG_LINK_CAP_L0S          BIT(10)
+#define PCIE_RC_CONFIG_LCS             (PCIE_RC_CONFIG_BASE + 0xd0)
+#define PCIE_RC_CONFIG_L1_SUBSTATE_CTRL2 (PCIE_RC_CONFIG_BASE + 0x90c)
+#define PCIE_RC_CONFIG_THP_CAP         (PCIE_RC_CONFIG_BASE + 0x274)
+#define   PCIE_RC_CONFIG_THP_CAP_NEXT_MASK     GENMASK(31, 20)
+
+#define PCIE_CORE_AXI_CONF_BASE                0xc00000
+#define PCIE_CORE_OB_REGION_ADDR0      (PCIE_CORE_AXI_CONF_BASE + 0x0)
+#define   PCIE_CORE_OB_REGION_ADDR0_NUM_BITS   0x3f
+#define   PCIE_CORE_OB_REGION_ADDR0_LO_ADDR    0xffffff00
+#define PCIE_CORE_OB_REGION_ADDR1      (PCIE_CORE_AXI_CONF_BASE + 0x4)
+#define PCIE_CORE_OB_REGION_DESC0      (PCIE_CORE_AXI_CONF_BASE + 0x8)
+#define PCIE_CORE_OB_REGION_DESC1      (PCIE_CORE_AXI_CONF_BASE + 0xc)
+
+#define PCIE_CORE_AXI_INBOUND_BASE     0xc00800
+#define PCIE_RP_IB_ADDR0               (PCIE_CORE_AXI_INBOUND_BASE + 0x0)
+#define   PCIE_CORE_IB_REGION_ADDR0_NUM_BITS   0x3f
+#define   PCIE_CORE_IB_REGION_ADDR0_LO_ADDR    0xffffff00
+#define PCIE_RP_IB_ADDR1               (PCIE_CORE_AXI_INBOUND_BASE + 0x4)
+
+/* Size of one AXI Region (not Region 0) */
+#define AXI_REGION_SIZE                                BIT(20)
+/* Size of Region 0, equal to sum of sizes of other regions */
+#define AXI_REGION_0_SIZE                      (32 * (0x1 << 20))
+#define OB_REG_SIZE_SHIFT                      5
+#define IB_ROOT_PORT_REG_SIZE_SHIFT            3
+#define AXI_WRAPPER_IO_WRITE                   0x6
+#define AXI_WRAPPER_MEM_WRITE                  0x2
+#define AXI_WRAPPER_TYPE0_CFG                  0xa
+#define AXI_WRAPPER_TYPE1_CFG                  0xb
+#define AXI_WRAPPER_NOR_MSG                    0xc
+
+#define MAX_AXI_IB_ROOTPORT_REGION_NUM         3
+#define MIN_AXI_ADDR_BITS_PASSED               8
+#define PCIE_RC_SEND_PME_OFF                   0x11960
+#define ROCKCHIP_VENDOR_ID                     0x1d87
+#define PCIE_ECAM_BUS(x)                       (((x) & 0xff) << 20)
+#define PCIE_ECAM_DEV(x)                       (((x) & 0x1f) << 15)
+#define PCIE_ECAM_FUNC(x)                      (((x) & 0x7) << 12)
+#define PCIE_ECAM_REG(x)                       (((x) & 0xfff) << 0)
+#define PCIE_ECAM_ADDR(bus, dev, func, reg) \
+         (PCIE_ECAM_BUS(bus) | PCIE_ECAM_DEV(dev) | \
+          PCIE_ECAM_FUNC(func) | PCIE_ECAM_REG(reg))
+#define PCIE_LINK_IS_L2(x) \
+       (((x) & PCIE_CLIENT_DEBUG_LTSSM_MASK) == PCIE_CLIENT_DEBUG_LTSSM_L2)
+#define PCIE_LINK_UP(x) \
+       (((x) & PCIE_CLIENT_LINK_STATUS_MASK) == PCIE_CLIENT_LINK_STATUS_UP)
+#define PCIE_LINK_IS_GEN2(x) \
+       (((x) & PCIE_CORE_PL_CONF_SPEED_MASK) == PCIE_CORE_PL_CONF_SPEED_5G)
+
+#define RC_REGION_0_ADDR_TRANS_H               0x00000000
+#define RC_REGION_0_ADDR_TRANS_L               0x00000000
+#define RC_REGION_0_PASS_BITS                  (25 - 1)
+#define RC_REGION_0_TYPE_MASK                  GENMASK(3, 0)
+#define MAX_AXI_WRAPPER_REGION_NUM             33
+
+#define ROCKCHIP_PCIE_MSG_ROUTING_TO_RC                0x0
+#define ROCKCHIP_PCIE_MSG_ROUTING_VIA_ADDR             0x1
+#define ROCKCHIP_PCIE_MSG_ROUTING_VIA_ID               0x2
+#define ROCKCHIP_PCIE_MSG_ROUTING_BROADCAST            0x3
+#define ROCKCHIP_PCIE_MSG_ROUTING_LOCAL_INTX           0x4
+#define ROCKCHIP_PCIE_MSG_ROUTING_PME_ACK              0x5
+#define ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTA             0x20
+#define ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTB             0x21
+#define ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTC             0x22
+#define ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTD             0x23
+#define ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTA           0x24
+#define ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTB           0x25
+#define ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTC           0x26
+#define ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTD           0x27
+#define ROCKCHIP_PCIE_MSG_ROUTING_MASK                 GENMASK(7, 5)
+#define ROCKCHIP_PCIE_MSG_ROUTING(route) \
+       (((route) << 5) & ROCKCHIP_PCIE_MSG_ROUTING_MASK)
+#define ROCKCHIP_PCIE_MSG_CODE_MASK                    GENMASK(15, 8)
+#define ROCKCHIP_PCIE_MSG_CODE(code) \
+       (((code) << 8) & ROCKCHIP_PCIE_MSG_CODE_MASK)
+#define ROCKCHIP_PCIE_MSG_NO_DATA                      BIT(16)
+
+#define ROCKCHIP_PCIE_EP_CMD_STATUS                    0x4
+#define   ROCKCHIP_PCIE_EP_CMD_STATUS_IS               BIT(19)
+#define ROCKCHIP_PCIE_EP_MSI_CTRL_REG                  0x90
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_OFFSET         17
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_MASK           GENMASK(19, 17)
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_MME_OFFSET         20
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_MME_MASK           GENMASK(22, 20)
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_ME                         BIT(16)
+#define   ROCKCHIP_PCIE_EP_MSI_CTRL_MASK_MSI_CAP       BIT(24)
+#define ROCKCHIP_PCIE_EP_DUMMY_IRQ_ADDR                                0x1
+#define ROCKCHIP_PCIE_EP_PCI_LEGACY_IRQ_ADDR           0x3
+#define ROCKCHIP_PCIE_EP_FUNC_BASE(fn) (((fn) << 12) & GENMASK(19, 12))
+#define ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR0(fn, bar) \
+       (PCIE_RC_RP_ATS_BASE + 0x0840 + (fn) * 0x0040 + (bar) * 0x0008)
+#define ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar) \
+       (PCIE_RC_RP_ATS_BASE + 0x0844 + (fn) * 0x0040 + (bar) * 0x0008)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(r) \
+       (PCIE_RC_RP_ATS_BASE + 0x0000 + ((r) & 0x1f) * 0x0020)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_DEVFN_MASK        GENMASK(19, 12)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_DEVFN(devfn) \
+       (((devfn) << 12) & \
+                ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_DEVFN_MASK)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_BUS_MASK  GENMASK(27, 20)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_BUS(bus) \
+               (((bus) << 20) & ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0_BUS_MASK)
+#define ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(r) \
+               (PCIE_RC_RP_ATS_BASE + 0x0004 + ((r) & 0x1f) * 0x0020)
+#define ROCKCHIP_PCIE_AT_OB_REGION_DESC0_HARDCODED_RID BIT(23)
+#define ROCKCHIP_PCIE_AT_OB_REGION_DESC0_DEVFN_MASK    GENMASK(31, 24)
+#define ROCKCHIP_PCIE_AT_OB_REGION_DESC0_DEVFN(devfn) \
+               (((devfn) << 24) & ROCKCHIP_PCIE_AT_OB_REGION_DESC0_DEVFN_MASK)
+#define ROCKCHIP_PCIE_AT_OB_REGION_DESC0(r) \
+               (PCIE_RC_RP_ATS_BASE + 0x0008 + ((r) & 0x1f) * 0x0020)
+#define ROCKCHIP_PCIE_AT_OB_REGION_DESC1(r)    \
+               (PCIE_RC_RP_ATS_BASE + 0x000c + ((r) & 0x1f) * 0x0020)
+#define ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR0(r) \
+               (PCIE_RC_RP_ATS_BASE + 0x0018 + ((r) & 0x1f) * 0x0020)
+#define ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(r) \
+               (PCIE_RC_RP_ATS_BASE + 0x001c + ((r) & 0x1f) * 0x0020)
+
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG0(fn) \
+               (PCIE_CORE_CTRL_MGMT_BASE + 0x0240 + (fn) * 0x0008)
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG1(fn) \
+               (PCIE_CORE_CTRL_MGMT_BASE + 0x0244 + (fn) * 0x0008)
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b) \
+               (GENMASK(4, 0) << ((b) * 8))
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE(b, a) \
+               (((a) << ((b) * 8)) & \
+                ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b))
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b) \
+               (GENMASK(7, 5) << ((b) * 8))
+#define ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL(b, c) \
+               (((c) << ((b) * 8 + 5)) & \
+                ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b))
+
+struct rockchip_pcie {
+       void    __iomem *reg_base;              /* DT axi-base */
+       void    __iomem *apb_base;              /* DT apb-base */
+       bool    legacy_phy;
+       struct  phy *phys[MAX_LANE_NUM];
+       struct  reset_control *core_rst;
+       struct  reset_control *mgmt_rst;
+       struct  reset_control *mgmt_sticky_rst;
+       struct  reset_control *pipe_rst;
+       struct  reset_control *pm_rst;
+       struct  reset_control *aclk_rst;
+       struct  reset_control *pclk_rst;
+       struct  clk *aclk_pcie;
+       struct  clk *aclk_perf_pcie;
+       struct  clk *hclk_pcie;
+       struct  clk *clk_pcie_pm;
+       struct  regulator *vpcie12v; /* 12V power supply */
+       struct  regulator *vpcie3v3; /* 3.3V power supply */
+       struct  regulator *vpcie1v8; /* 1.8V power supply */
+       struct  regulator *vpcie0v9; /* 0.9V power supply */
+       struct  gpio_desc *ep_gpio;
+       u32     lanes;
+       u8      lanes_map;
+       u8      root_bus_nr;
+       int     link_gen;
+       struct  device *dev;
+       struct  irq_domain *irq_domain;
+       int     offset;
+       struct pci_bus *root_bus;
+       struct resource *io;
+       phys_addr_t io_bus_addr;
+       u32     io_size;
+       void    __iomem *msg_region;
+       u32     mem_size;
+       phys_addr_t msg_bus_addr;
+       phys_addr_t mem_bus_addr;
+       bool is_rc;
+       struct resource *mem_res;
+};
+
+static u32 rockchip_pcie_read(struct rockchip_pcie *rockchip, u32 reg)
+{
+       return readl(rockchip->apb_base + reg);
+}
+
+static void rockchip_pcie_write(struct rockchip_pcie *rockchip, u32 val,
+                               u32 reg)
+{
+       writel(val, rockchip->apb_base + reg);
+}
+
+int rockchip_pcie_parse_dt(struct rockchip_pcie *rockchip);
+int rockchip_pcie_init_port(struct rockchip_pcie *rockchip);
+int rockchip_pcie_get_phys(struct rockchip_pcie *rockchip);
+void rockchip_pcie_deinit_phys(struct rockchip_pcie *rockchip);
+int rockchip_pcie_enable_clocks(struct rockchip_pcie *rockchip);
+void rockchip_pcie_disable_clocks(void *data);
+void rockchip_pcie_cfg_configuration_accesses(
+               struct rockchip_pcie *rockchip, u32 type);
+
+#endif /* _PCIE_ROCKCHIP_H */