drivers/pci/controller/pcie-rockchip-ep.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Rockchip AXI PCIe endpoint controller driver
   4  *
   5  * Copyright (c) 2018 Rockchip, Inc.
   6  *
   7  * Author: Shawn Lin <shawn.lin@rock-chips.com>
   8  *         Simon Xue <xxm@rock-chips.com>
   9  */
  10
  11 #include <linux/configfs.h>
  12 #include <linux/delay.h>
  13 #include <linux/kernel.h>
  14 #include <linux/of.h>
  15 #include <linux/pci-epc.h>
  16 #include <linux/platform_device.h>
  17 #include <linux/pci-epf.h>
  18 #include <linux/sizes.h>
  19
  20 #include "pcie-rockchip.h"
  21
  22 /**
  23  * struct rockchip_pcie_ep - private data for PCIe endpoint controller driver
  24  * @rockchip: Rockchip PCIe controller
  25  * @max_regions: maximum number of regions supported by hardware
  26  * @ob_region_map: bitmask of mapped outbound regions
  27  * @ob_addr: base addresses in the AXI bus where the outbound regions start
  28  * @irq_phys_addr: base address on the AXI bus where the MSI/legacy IRQ
  29  *                 dedicated outbound regions is mapped.
  30  * @irq_cpu_addr: base address in the CPU space where a write access triggers
  31  *                the sending of a memory write (MSI) / normal message (legacy
  32  *                IRQ) TLP through the PCIe bus.
  33  * @irq_pci_addr: used to save the current mapping of the MSI/legacy IRQ
  34  *                dedicated outbound region.
  35  * @irq_pci_fn: the latest PCI function that has updated the mapping of
  36  *              the MSI/legacy IRQ dedicated outbound region.
  37  * @irq_pending: bitmask of asserted legacy IRQs.
  38  */
  39 struct rockchip_pcie_ep {
  40         struct rockchip_pcie    rockchip;
  41         struct pci_epc          *epc;
  42         u32                     max_regions;
  43         unsigned long           ob_region_map;
  44         phys_addr_t             *ob_addr;
  45         phys_addr_t             irq_phys_addr;
  46         void __iomem            *irq_cpu_addr;
  47         u64                     irq_pci_addr;
  48         u8                      irq_pci_fn;
  49         u8                      irq_pending;
  50 };
  51
  52 static void rockchip_pcie_clear_ep_ob_atu(struct rockchip_pcie *rockchip,
  53                                           u32 region)
  54 {
  55         rockchip_pcie_write(rockchip, 0,
  56                             ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(region));
  57         rockchip_pcie_write(rockchip, 0,
  58                             ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(region));
  59         rockchip_pcie_write(rockchip, 0,
  60                             ROCKCHIP_PCIE_AT_OB_REGION_DESC0(region));
  61         rockchip_pcie_write(rockchip, 0,
  62                             ROCKCHIP_PCIE_AT_OB_REGION_DESC1(region));
  63         rockchip_pcie_write(rockchip, 0,
  64                             ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR0(region));
  65         rockchip_pcie_write(rockchip, 0,
  66                             ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(region));
  67 }
  68
  69 static void rockchip_pcie_prog_ep_ob_atu(struct rockchip_pcie *rockchip, u8 fn,
  70                                          u32 r, u32 type, u64 cpu_addr,
  71                                          u64 pci_addr, size_t size)
  72 {
  73         u64 sz = 1ULL << fls64(size - 1);
  74         int num_pass_bits = ilog2(sz);
  75         u32 addr0, addr1, desc0, desc1;
  76         bool is_nor_msg = (type == AXI_WRAPPER_NOR_MSG);
  77
  78         /* The minimal region size is 1MB */
  79         if (num_pass_bits < 8)
  80                 num_pass_bits = 8;
  81
  82         cpu_addr -= rockchip->mem_res->start;
  83         addr0 = ((is_nor_msg ? 0x10 : (num_pass_bits - 1)) &
  84                 PCIE_CORE_OB_REGION_ADDR0_NUM_BITS) |
  85                 (lower_32_bits(cpu_addr) & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR);
  86         addr1 = upper_32_bits(is_nor_msg ? cpu_addr : pci_addr);
  87         desc0 = ROCKCHIP_PCIE_AT_OB_REGION_DESC0_DEVFN(fn) | type;
  88         desc1 = 0;
  89
  90         if (is_nor_msg) {
  91                 rockchip_pcie_write(rockchip, 0,
  92                                     ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(r));
  93                 rockchip_pcie_write(rockchip, 0,
  94                                     ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(r));
  95                 rockchip_pcie_write(rockchip, desc0,
  96                                     ROCKCHIP_PCIE_AT_OB_REGION_DESC0(r));
  97                 rockchip_pcie_write(rockchip, desc1,
  98                                     ROCKCHIP_PCIE_AT_OB_REGION_DESC1(r));
  99         } else {
 100                 /* PCI bus address region */
 101                 rockchip_pcie_write(rockchip, addr0,
 102                                     ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR0(r));
 103                 rockchip_pcie_write(rockchip, addr1,
 104                                     ROCKCHIP_PCIE_AT_OB_REGION_PCI_ADDR1(r));
 105                 rockchip_pcie_write(rockchip, desc0,
 106                                     ROCKCHIP_PCIE_AT_OB_REGION_DESC0(r));
 107                 rockchip_pcie_write(rockchip, desc1,
 108                                     ROCKCHIP_PCIE_AT_OB_REGION_DESC1(r));
 109
 110                 addr0 =
 111                     ((num_pass_bits - 1) & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS) |
 112                     (lower_32_bits(cpu_addr) &
 113                      PCIE_CORE_OB_REGION_ADDR0_LO_ADDR);
 114                 addr1 = upper_32_bits(cpu_addr);
 115         }
 116
 117         /* CPU bus address region */
 118         rockchip_pcie_write(rockchip, addr0,
 119                             ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR0(r));
 120         rockchip_pcie_write(rockchip, addr1,
 121                             ROCKCHIP_PCIE_AT_OB_REGION_CPU_ADDR1(r));
 122 }
 123
 124 static int rockchip_pcie_ep_write_header(struct pci_epc *epc, u8 fn,
 125                                          struct pci_epf_header *hdr)
 126 {
 127         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 128         struct rockchip_pcie *rockchip = &ep->rockchip;
 129
 130         /* All functions share the same vendor ID with function 0 */
 131         if (fn == 0) {
 132                 u32 vid_regs = (hdr->vendorid & GENMASK(15, 0)) |
 133                                (hdr->subsys_vendor_id & GENMASK(31, 16)) << 16;
 134
 135                 rockchip_pcie_write(rockchip, vid_regs,
 136                                     PCIE_CORE_CONFIG_VENDOR);
 137         }
 138
 139         rockchip_pcie_write(rockchip, hdr->deviceid << 16,
 140                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) + PCI_VENDOR_ID);
 141
 142         rockchip_pcie_write(rockchip,
 143                             hdr->revid |
 144                             hdr->progif_code << 8 |
 145                             hdr->subclass_code << 16 |
 146                             hdr->baseclass_code << 24,
 147                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) + PCI_REVISION_ID);
 148         rockchip_pcie_write(rockchip, hdr->cache_line_size,
 149                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 150                             PCI_CACHE_LINE_SIZE);
 151         rockchip_pcie_write(rockchip, hdr->subsys_id << 16,
 152                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 153                             PCI_SUBSYSTEM_VENDOR_ID);
 154         rockchip_pcie_write(rockchip, hdr->interrupt_pin << 8,
 155                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 156                             PCI_INTERRUPT_LINE);
 157
 158         return 0;
 159 }
 160
 161 static int rockchip_pcie_ep_set_bar(struct pci_epc *epc, u8 fn,
 162                                     struct pci_epf_bar *epf_bar)
 163 {
 164         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 165         struct rockchip_pcie *rockchip = &ep->rockchip;
 166         dma_addr_t bar_phys = epf_bar->phys_addr;
 167         enum pci_barno bar = epf_bar->barno;
 168         int flags = epf_bar->flags;
 169         u32 addr0, addr1, reg, cfg, b, aperture, ctrl;
 170         u64 sz;
 171
 172         /* BAR size is 2^(aperture + 7) */
 173         sz = max_t(size_t, epf_bar->size, MIN_EP_APERTURE);
 174
 175         /*
 176          * roundup_pow_of_two() returns an unsigned long, which is not suited
 177          * for 64bit values.
 178          */
 179         sz = 1ULL << fls64(sz - 1);
 180         aperture = ilog2(sz) - 7; /* 128B -> 0, 256B -> 1, 512B -> 2, ... */
 181
 182         if ((flags & PCI_BASE_ADDRESS_SPACE) == PCI_BASE_ADDRESS_SPACE_IO) {
 183                 ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_IO_32BITS;
 184         } else {
 185                 bool is_prefetch = !!(flags & PCI_BASE_ADDRESS_MEM_PREFETCH);
 186                 bool is_64bits = sz > SZ_2G;
 187
 188                 if (is_64bits && (bar & 1))
 189                         return -EINVAL;
 190
 191                 if (is_64bits && is_prefetch)
 192                         ctrl =
 193                             ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_64BITS;
 194                 else if (is_prefetch)
 195                         ctrl =
 196                             ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_PREFETCH_MEM_32BITS;
 197                 else if (is_64bits)
 198                         ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_64BITS;
 199                 else
 200                         ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_MEM_32BITS;
 201         }
 202
 203         if (bar < BAR_4) {
 204                 reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG0(fn);
 205                 b = bar;
 206         } else {
 207                 reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG1(fn);
 208                 b = bar - BAR_4;
 209         }
 210
 211         addr0 = lower_32_bits(bar_phys);
 212         addr1 = upper_32_bits(bar_phys);
 213
 214         cfg = rockchip_pcie_read(rockchip, reg);
 215         cfg &= ~(ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b) |
 216                  ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b));
 217         cfg |= (ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE(b, aperture) |
 218                 ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL(b, ctrl));
 219
 220         rockchip_pcie_write(rockchip, cfg, reg);
 221         rockchip_pcie_write(rockchip, addr0,
 222                             ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR0(fn, bar));
 223         rockchip_pcie_write(rockchip, addr1,
 224                             ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar));
 225
 226         return 0;
 227 }
 228
 229 static void rockchip_pcie_ep_clear_bar(struct pci_epc *epc, u8 fn,
 230                                        struct pci_epf_bar *epf_bar)
 231 {
 232         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 233         struct rockchip_pcie *rockchip = &ep->rockchip;
 234         u32 reg, cfg, b, ctrl;
 235         enum pci_barno bar = epf_bar->barno;
 236
 237         if (bar < BAR_4) {
 238                 reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG0(fn);
 239                 b = bar;
 240         } else {
 241                 reg = ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG1(fn);
 242                 b = bar - BAR_4;
 243         }
 244
 245         ctrl = ROCKCHIP_PCIE_CORE_BAR_CFG_CTRL_DISABLED;
 246         cfg = rockchip_pcie_read(rockchip, reg);
 247         cfg &= ~(ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_APERTURE_MASK(b) |
 248                  ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL_MASK(b));
 249         cfg |= ROCKCHIP_PCIE_CORE_EP_FUNC_BAR_CFG_BAR_CTRL(b, ctrl);
 250
 251         rockchip_pcie_write(rockchip, cfg, reg);
 252         rockchip_pcie_write(rockchip, 0x0,
 253                             ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR0(fn, bar));
 254         rockchip_pcie_write(rockchip, 0x0,
 255                             ROCKCHIP_PCIE_AT_IB_EP_FUNC_BAR_ADDR1(fn, bar));
 256 }
 257
 258 static int rockchip_pcie_ep_map_addr(struct pci_epc *epc, u8 fn,
 259                                      phys_addr_t addr, u64 pci_addr,
 260                                      size_t size)
 261 {
 262         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 263         struct rockchip_pcie *pcie = &ep->rockchip;
 264         u32 r;
 265
 266         r = find_first_zero_bit(&ep->ob_region_map, BITS_PER_LONG);
 267         /*
 268          * Region 0 is reserved for configuration space and shouldn't
 269          * be used elsewhere per TRM, so leave it out.
 270          */
 271         if (r >= ep->max_regions - 1) {
 272                 dev_err(&epc->dev, "no free outbound region\n");
 273                 return -EINVAL;
 274         }
 275
 276         rockchip_pcie_prog_ep_ob_atu(pcie, fn, r, AXI_WRAPPER_MEM_WRITE, addr,
 277                                      pci_addr, size);
 278
 279         set_bit(r, &ep->ob_region_map);
 280         ep->ob_addr[r] = addr;
 281
 282         return 0;
 283 }
 284
 285 static void rockchip_pcie_ep_unmap_addr(struct pci_epc *epc, u8 fn,
 286                                         phys_addr_t addr)
 287 {
 288         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 289         struct rockchip_pcie *rockchip = &ep->rockchip;
 290         u32 r;
 291
 292         for (r = 0; r < ep->max_regions - 1; r++)
 293                 if (ep->ob_addr[r] == addr)
 294                         break;
 295
 296         /*
 297          * Region 0 is reserved for configuration space and shouldn't
 298          * be used elsewhere per TRM, so leave it out.
 299          */
 300         if (r == ep->max_regions - 1)
 301                 return;
 302
 303         rockchip_pcie_clear_ep_ob_atu(rockchip, r);
 304
 305         ep->ob_addr[r] = 0;
 306         clear_bit(r, &ep->ob_region_map);
 307 }
 308
 309 static int rockchip_pcie_ep_set_msi(struct pci_epc *epc, u8 fn,
 310                                     u8 multi_msg_cap)
 311 {
 312         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 313         struct rockchip_pcie *rockchip = &ep->rockchip;
 314         u16 flags;
 315
 316         flags = rockchip_pcie_read(rockchip,
 317                                    ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 318                                    ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
 319         flags &= ~ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_MASK;
 320         flags |=
 321            ((multi_msg_cap << 1) <<  ROCKCHIP_PCIE_EP_MSI_CTRL_MMC_OFFSET) |
 322            PCI_MSI_FLAGS_64BIT;
 323         flags &= ~ROCKCHIP_PCIE_EP_MSI_CTRL_MASK_MSI_CAP;
 324         rockchip_pcie_write(rockchip, flags,
 325                             ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 326                             ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
 327         return 0;
 328 }
 329
 330 static int rockchip_pcie_ep_get_msi(struct pci_epc *epc, u8 fn)
 331 {
 332         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 333         struct rockchip_pcie *rockchip = &ep->rockchip;
 334         u16 flags;
 335
 336         flags = rockchip_pcie_read(rockchip,
 337                                    ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 338                                    ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
 339         if (!(flags & ROCKCHIP_PCIE_EP_MSI_CTRL_ME))
 340                 return -EINVAL;
 341
 342         return ((flags & ROCKCHIP_PCIE_EP_MSI_CTRL_MME_MASK) >>
 343                         ROCKCHIP_PCIE_EP_MSI_CTRL_MME_OFFSET);
 344 }
 345
 346 static void rockchip_pcie_ep_assert_intx(struct rockchip_pcie_ep *ep, u8 fn,
 347                                          u8 intx, bool is_asserted)
 348 {
 349         struct rockchip_pcie *rockchip = &ep->rockchip;
 350         u32 r = ep->max_regions - 1;
 351         u32 offset;
 352         u32 status;
 353         u8 msg_code;
 354
 355         if (unlikely(ep->irq_pci_addr != ROCKCHIP_PCIE_EP_PCI_LEGACY_IRQ_ADDR ||
 356                      ep->irq_pci_fn != fn)) {
 357                 rockchip_pcie_prog_ep_ob_atu(rockchip, fn, r,
 358                                              AXI_WRAPPER_NOR_MSG,
 359                                              ep->irq_phys_addr, 0, 0);
 360                 ep->irq_pci_addr = ROCKCHIP_PCIE_EP_PCI_LEGACY_IRQ_ADDR;
 361                 ep->irq_pci_fn = fn;
 362         }
 363
 364         intx &= 3;
 365         if (is_asserted) {
 366                 ep->irq_pending |= BIT(intx);
 367                 msg_code = ROCKCHIP_PCIE_MSG_CODE_ASSERT_INTA + intx;
 368         } else {
 369                 ep->irq_pending &= ~BIT(intx);
 370                 msg_code = ROCKCHIP_PCIE_MSG_CODE_DEASSERT_INTA + intx;
 371         }
 372
 373         status = rockchip_pcie_read(rockchip,
 374                                     ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 375                                     ROCKCHIP_PCIE_EP_CMD_STATUS);
 376         status &= ROCKCHIP_PCIE_EP_CMD_STATUS_IS;
 377
 378         if ((status != 0) ^ (ep->irq_pending != 0)) {
 379                 status ^= ROCKCHIP_PCIE_EP_CMD_STATUS_IS;
 380                 rockchip_pcie_write(rockchip, status,
 381                                     ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 382                                     ROCKCHIP_PCIE_EP_CMD_STATUS);
 383         }
 384
 385         offset =
 386            ROCKCHIP_PCIE_MSG_ROUTING(ROCKCHIP_PCIE_MSG_ROUTING_LOCAL_INTX) |
 387            ROCKCHIP_PCIE_MSG_CODE(msg_code) | ROCKCHIP_PCIE_MSG_NO_DATA;
 388         writel(0, ep->irq_cpu_addr + offset);
 389 }
 390
 391 static int rockchip_pcie_ep_send_legacy_irq(struct rockchip_pcie_ep *ep, u8 fn,
 392                                             u8 intx)
 393 {
 394         u16 cmd;
 395
 396         cmd = rockchip_pcie_read(&ep->rockchip,
 397                                  ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 398                                  ROCKCHIP_PCIE_EP_CMD_STATUS);
 399
 400         if (cmd & PCI_COMMAND_INTX_DISABLE)
 401                 return -EINVAL;
 402
 403         /*
 404          * Should add some delay between toggling INTx per TRM vaguely saying
 405          * it depends on some cycles of the AHB bus clock to function it. So
 406          * add sufficient 1ms here.
 407          */
 408         rockchip_pcie_ep_assert_intx(ep, fn, intx, true);
 409         mdelay(1);
 410         rockchip_pcie_ep_assert_intx(ep, fn, intx, false);
 411         return 0;
 412 }
 413
 414 static int rockchip_pcie_ep_send_msi_irq(struct rockchip_pcie_ep *ep, u8 fn,
 415                                          u8 interrupt_num)
 416 {
 417         struct rockchip_pcie *rockchip = &ep->rockchip;
 418         u16 flags, mme, data, data_mask;
 419         u8 msi_count;
 420         u64 pci_addr, pci_addr_mask = 0xff;
 421
 422         /* Check MSI enable bit */
 423         flags = rockchip_pcie_read(&ep->rockchip,
 424                                    ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 425                                    ROCKCHIP_PCIE_EP_MSI_CTRL_REG);
 426         if (!(flags & ROCKCHIP_PCIE_EP_MSI_CTRL_ME))
 427                 return -EINVAL;
 428
 429         /* Get MSI numbers from MME */
 430         mme = ((flags & ROCKCHIP_PCIE_EP_MSI_CTRL_MME_MASK) >>
 431                         ROCKCHIP_PCIE_EP_MSI_CTRL_MME_OFFSET);
 432         msi_count = 1 << mme;
 433         if (!interrupt_num || interrupt_num > msi_count)
 434                 return -EINVAL;
 435
 436         /* Set MSI private data */
 437         data_mask = msi_count - 1;
 438         data = rockchip_pcie_read(rockchip,
 439                                   ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 440                                   ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
 441                                   PCI_MSI_DATA_64);
 442         data = (data & ~data_mask) | ((interrupt_num - 1) & data_mask);
 443
 444         /* Get MSI PCI address */
 445         pci_addr = rockchip_pcie_read(rockchip,
 446                                       ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 447                                       ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
 448                                       PCI_MSI_ADDRESS_HI);
 449         pci_addr <<= 32;
 450         pci_addr |= rockchip_pcie_read(rockchip,
 451                                        ROCKCHIP_PCIE_EP_FUNC_BASE(fn) +
 452                                        ROCKCHIP_PCIE_EP_MSI_CTRL_REG +
 453                                        PCI_MSI_ADDRESS_LO);
 454         pci_addr &= GENMASK_ULL(63, 2);
 455
 456         /* Set the outbound region if needed. */
 457         if (unlikely(ep->irq_pci_addr != (pci_addr & ~pci_addr_mask) ||
 458                      ep->irq_pci_fn != fn)) {
 459                 rockchip_pcie_prog_ep_ob_atu(rockchip, fn, ep->max_regions - 1,
 460                                              AXI_WRAPPER_MEM_WRITE,
 461                                              ep->irq_phys_addr,
 462                                              pci_addr & ~pci_addr_mask,
 463                                              pci_addr_mask + 1);
 464                 ep->irq_pci_addr = (pci_addr & ~pci_addr_mask);
 465                 ep->irq_pci_fn = fn;
 466         }
 467
 468         writew(data, ep->irq_cpu_addr + (pci_addr & pci_addr_mask));
 469         return 0;
 470 }
 471
 472 static int rockchip_pcie_ep_raise_irq(struct pci_epc *epc, u8 fn,
 473                                       enum pci_epc_irq_type type,
 474                                       u16 interrupt_num)
 475 {
 476         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 477
 478         switch (type) {
 479         case PCI_EPC_IRQ_LEGACY:
 480                 return rockchip_pcie_ep_send_legacy_irq(ep, fn, 0);
 481         case PCI_EPC_IRQ_MSI:
 482                 return rockchip_pcie_ep_send_msi_irq(ep, fn, interrupt_num);
 483         default:
 484                 return -EINVAL;
 485         }
 486 }
 487
 488 static int rockchip_pcie_ep_start(struct pci_epc *epc)
 489 {
 490         struct rockchip_pcie_ep *ep = epc_get_drvdata(epc);
 491         struct rockchip_pcie *rockchip = &ep->rockchip;
 492         struct pci_epf *epf;
 493         u32 cfg;
 494
 495         cfg = BIT(0);
 496         list_for_each_entry(epf, &epc->pci_epf, list)
 497                 cfg |= BIT(epf->func_no);
 498
 499         rockchip_pcie_write(rockchip, cfg, PCIE_CORE_PHY_FUNC_CFG);
 500
 501         list_for_each_entry(epf, &epc->pci_epf, list)
 502                 pci_epf_linkup(epf);
 503
 504         return 0;
 505 }
 506
 507 static const struct pci_epc_ops rockchip_pcie_epc_ops = {
 508         .write_header   = rockchip_pcie_ep_write_header,
 509         .set_bar        = rockchip_pcie_ep_set_bar,
 510         .clear_bar      = rockchip_pcie_ep_clear_bar,
 511         .map_addr       = rockchip_pcie_ep_map_addr,
 512         .unmap_addr     = rockchip_pcie_ep_unmap_addr,
 513         .set_msi        = rockchip_pcie_ep_set_msi,
 514         .get_msi        = rockchip_pcie_ep_get_msi,
 515         .raise_irq      = rockchip_pcie_ep_raise_irq,
 516         .start          = rockchip_pcie_ep_start,
 517 };
 518
 519 static int rockchip_pcie_parse_ep_dt(struct rockchip_pcie *rockchip,
 520                                      struct rockchip_pcie_ep *ep)
 521 {
 522         struct device *dev = rockchip->dev;
 523         int err;
 524
 525         err = rockchip_pcie_parse_dt(rockchip);
 526         if (err)
 527                 return err;
 528
 529         err = rockchip_pcie_get_phys(rockchip);
 530         if (err)
 531                 return err;
 532
 533         err = of_property_read_u32(dev->of_node,
 534                                    "rockchip,max-outbound-regions",
 535                                    &ep->max_regions);
 536         if (err < 0 || ep->max_regions > MAX_REGION_LIMIT)
 537                 ep->max_regions = MAX_REGION_LIMIT;
 538
 539         err = of_property_read_u8(dev->of_node, "max-functions",
 540                                   &ep->epc->max_functions);
 541         if (err < 0)
 542                 ep->epc->max_functions = 1;
 543
 544         return 0;
 545 }
 546
 547 static const struct of_device_id rockchip_pcie_ep_of_match[] = {
 548         { .compatible = "rockchip,rk3399-pcie-ep"},
 549         {},
 550 };
 551
 552 static int rockchip_pcie_ep_probe(struct platform_device *pdev)
 553 {
 554         struct device *dev = &pdev->dev;
 555         struct rockchip_pcie_ep *ep;
 556         struct rockchip_pcie *rockchip;
 557         struct pci_epc *epc;
 558         size_t max_regions;
 559         int err;
 560
 561         ep = devm_kzalloc(dev, sizeof(*ep), GFP_KERNEL);
 562         if (!ep)
 563                 return -ENOMEM;
 564
 565         rockchip = &ep->rockchip;
 566         rockchip->is_rc = false;
 567         rockchip->dev = dev;
 568
 569         epc = devm_pci_epc_create(dev, &rockchip_pcie_epc_ops);
 570         if (IS_ERR(epc)) {
 571                 dev_err(dev, "failed to create epc device\n");
 572                 return PTR_ERR(epc);
 573         }
 574
 575         ep->epc = epc;
 576         epc_set_drvdata(epc, ep);
 577
 578         err = rockchip_pcie_parse_ep_dt(rockchip, ep);
 579         if (err)
 580                 return err;
 581
 582         err = rockchip_pcie_enable_clocks(rockchip);
 583         if (err)
 584                 return err;
 585
 586         err = rockchip_pcie_init_port(rockchip);
 587         if (err)
 588                 goto err_disable_clocks;
 589
 590         /* Establish the link automatically */
 591         rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
 592                             PCIE_CLIENT_CONFIG);
 593
 594         max_regions = ep->max_regions;
 595         ep->ob_addr = devm_kcalloc(dev, max_regions, sizeof(*ep->ob_addr),
 596                                    GFP_KERNEL);
 597
 598         if (!ep->ob_addr) {
 599                 err = -ENOMEM;
 600                 goto err_uninit_port;
 601         }
 602
 603         /* Only enable function 0 by default */
 604         rockchip_pcie_write(rockchip, BIT(0), PCIE_CORE_PHY_FUNC_CFG);
 605
 606         err = pci_epc_mem_init(epc, rockchip->mem_res->start,
 607                                resource_size(rockchip->mem_res));
 608         if (err < 0) {
 609                 dev_err(dev, "failed to initialize the memory space\n");
 610                 goto err_uninit_port;
 611         }
 612
 613         ep->irq_cpu_addr = pci_epc_mem_alloc_addr(epc, &ep->irq_phys_addr,
 614                                                   SZ_128K);
 615         if (!ep->irq_cpu_addr) {
 616                 dev_err(dev, "failed to reserve memory space for MSI\n");
 617                 err = -ENOMEM;
 618                 goto err_epc_mem_exit;
 619         }
 620
 621         ep->irq_pci_addr = ROCKCHIP_PCIE_EP_DUMMY_IRQ_ADDR;
 622
 623         return 0;
 624 err_epc_mem_exit:
 625         pci_epc_mem_exit(epc);
 626 err_uninit_port:
 627         rockchip_pcie_deinit_phys(rockchip);
 628 err_disable_clocks:
 629         rockchip_pcie_disable_clocks(rockchip);
 630         return err;
 631 }
 632
 633 static struct platform_driver rockchip_pcie_ep_driver = {
 634         .driver = {
 635                 .name = "rockchip-pcie-ep",
 636                 .of_match_table = rockchip_pcie_ep_of_match,
 637         },
 638         .probe = rockchip_pcie_ep_probe,
 639 };
 640
 641 builtin_platform_driver(rockchip_pcie_ep_driver);