drivers/pci/controller/dwc/pci-keystone.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * PCIe host controller driver for Texas Instruments Keystone SoCs
   4  *
   5  * Copyright (C) 2013-2014 Texas Instruments., Ltd.
   6  *              https://www.ti.com
   7  *
   8  * Author: Murali Karicheri <[email protected]>
   9  * Implementation based on pci-exynos.c and pcie-designware.c
  10  */
  11
  12 #include <linux/clk.h>
  13 #include <linux/delay.h>
  14 #include <linux/gpio/consumer.h>
  15 #include <linux/init.h>
  16 #include <linux/interrupt.h>
  17 #include <linux/irqchip/chained_irq.h>
  18 #include <linux/irqdomain.h>
  19 #include <linux/mfd/syscon.h>
  20 #include <linux/msi.h>
  21 #include <linux/of.h>
  22 #include <linux/of_device.h>
  23 #include <linux/of_irq.h>
  24 #include <linux/of_pci.h>
  25 #include <linux/phy/phy.h>
  26 #include <linux/platform_device.h>
  27 #include <linux/regmap.h>
  28 #include <linux/resource.h>
  29 #include <linux/signal.h>
  30
  31 #include "../../pci.h"
  32 #include "pcie-designware.h"
  33
  34 #define PCIE_VENDORID_MASK      0xffff
  35 #define PCIE_DEVICEID_SHIFT     16
  36
  37 /* Application registers */
  38 #define CMD_STATUS                      0x004
  39 #define LTSSM_EN_VAL                    BIT(0)
  40 #define OB_XLAT_EN_VAL                  BIT(1)
  41 #define DBI_CS2                         BIT(5)
  42
  43 #define CFG_SETUP                       0x008
  44 #define CFG_BUS(x)                      (((x) & 0xff) << 16)
  45 #define CFG_DEVICE(x)                   (((x) & 0x1f) << 8)
  46 #define CFG_FUNC(x)                     ((x) & 0x7)
  47 #define CFG_TYPE1                       BIT(24)
  48
  49 #define OB_SIZE                         0x030
  50 #define OB_OFFSET_INDEX(n)              (0x200 + (8 * (n)))
  51 #define OB_OFFSET_HI(n)                 (0x204 + (8 * (n)))
  52 #define OB_ENABLEN                      BIT(0)
  53 #define OB_WIN_SIZE                     8       /* 8MB */
  54
  55 #define PCIE_LEGACY_IRQ_ENABLE_SET(n)   (0x188 + (0x10 * ((n) - 1)))
  56 #define PCIE_LEGACY_IRQ_ENABLE_CLR(n)   (0x18c + (0x10 * ((n) - 1)))
  57 #define PCIE_EP_IRQ_SET                 0x64
  58 #define PCIE_EP_IRQ_CLR                 0x68
  59 #define INT_ENABLE                      BIT(0)
  60
  61 /* IRQ register defines */
  62 #define IRQ_EOI                         0x050
  63
  64 #define MSI_IRQ                         0x054
  65 #define MSI_IRQ_STATUS(n)               (0x104 + ((n) << 4))
  66 #define MSI_IRQ_ENABLE_SET(n)           (0x108 + ((n) << 4))
  67 #define MSI_IRQ_ENABLE_CLR(n)           (0x10c + ((n) << 4))
  68 #define MSI_IRQ_OFFSET                  4
  69
  70 #define IRQ_STATUS(n)                   (0x184 + ((n) << 4))
  71 #define IRQ_ENABLE_SET(n)               (0x188 + ((n) << 4))
  72 #define INTx_EN                         BIT(0)
  73
  74 #define ERR_IRQ_STATUS                  0x1c4
  75 #define ERR_IRQ_ENABLE_SET              0x1c8
  76 #define ERR_AER                         BIT(5)  /* ECRC error */
  77 #define AM6_ERR_AER                     BIT(4)  /* AM6 ECRC error */
  78 #define ERR_AXI                         BIT(4)  /* AXI tag lookup fatal error */
  79 #define ERR_CORR                        BIT(3)  /* Correctable error */
  80 #define ERR_NONFATAL                    BIT(2)  /* Non-fatal error */
  81 #define ERR_FATAL                       BIT(1)  /* Fatal error */
  82 #define ERR_SYS                         BIT(0)  /* System error */
  83 #define ERR_IRQ_ALL                     (ERR_AER | ERR_AXI | ERR_CORR | \
  84                                          ERR_NONFATAL | ERR_FATAL | ERR_SYS)
  85
  86 /* PCIE controller device IDs */
  87 #define PCIE_RC_K2HK                    0xb008
  88 #define PCIE_RC_K2E                     0xb009
  89 #define PCIE_RC_K2L                     0xb00a
  90 #define PCIE_RC_K2G                     0xb00b
  91
  92 #define KS_PCIE_DEV_TYPE_MASK           (0x3 << 1)
  93 #define KS_PCIE_DEV_TYPE(mode)          ((mode) << 1)
  94
  95 #define EP                              0x0
  96 #define LEG_EP                          0x1
  97 #define RC                              0x2
  98
  99 #define KS_PCIE_SYSCLOCKOUTEN           BIT(0)
 100
 101 #define AM654_PCIE_DEV_TYPE_MASK        0x3
 102 #define AM654_WIN_SIZE                  SZ_64K
 103
 104 #define APP_ADDR_SPACE_0                (16 * SZ_1K)
 105
 106 #define to_keystone_pcie(x)             dev_get_drvdata((x)->dev)
 107
 108 struct ks_pcie_of_data {
 109         enum dw_pcie_device_mode mode;
 110         const struct dw_pcie_host_ops *host_ops;
 111         const struct dw_pcie_ep_ops *ep_ops;
 112         unsigned int version;
 113 };
 114
 115 struct keystone_pcie {
 116         struct dw_pcie          *pci;
 117         /* PCI Device ID */
 118         u32                     device_id;
 119         int                     legacy_host_irqs[PCI_NUM_INTX];
 120         struct                  device_node *legacy_intc_np;
 121
 122         int                     msi_host_irq;
 123         int                     num_lanes;
 124         struct phy              **phy;
 125         struct device_link      **link;
 126         struct                  device_node *msi_intc_np;
 127         struct irq_domain       *legacy_irq_domain;
 128         struct device_node      *np;
 129
 130         /* Application register space */
 131         void __iomem            *va_app_base;   /* DT 1st resource */
 132         struct resource         app;
 133         bool                    is_am6;
 134 };
 135
 136 static u32 ks_pcie_app_readl(struct keystone_pcie *ks_pcie, u32 offset)
 137 {
 138         return readl(ks_pcie->va_app_base + offset);
 139 }
 140
 141 static void ks_pcie_app_writel(struct keystone_pcie *ks_pcie, u32 offset,
 142                                u32 val)
 143 {
 144         writel(val, ks_pcie->va_app_base + offset);
 145 }
 146
 147 static void ks_pcie_msi_irq_ack(struct irq_data *data)
 148 {
 149         struct pcie_port *pp  = irq_data_get_irq_chip_data(data);
 150         struct keystone_pcie *ks_pcie;
 151         u32 irq = data->hwirq;
 152         struct dw_pcie *pci;
 153         u32 reg_offset;
 154         u32 bit_pos;
 155
 156         pci = to_dw_pcie_from_pp(pp);
 157         ks_pcie = to_keystone_pcie(pci);
 158
 159         reg_offset = irq % 8;
 160         bit_pos = irq >> 3;
 161
 162         ks_pcie_app_writel(ks_pcie, MSI_IRQ_STATUS(reg_offset),
 163                            BIT(bit_pos));
 164         ks_pcie_app_writel(ks_pcie, IRQ_EOI, reg_offset + MSI_IRQ_OFFSET);
 165 }
 166
 167 static void ks_pcie_compose_msi_msg(struct irq_data *data, struct msi_msg *msg)
 168 {
 169         struct pcie_port *pp = irq_data_get_irq_chip_data(data);
 170         struct keystone_pcie *ks_pcie;
 171         struct dw_pcie *pci;
 172         u64 msi_target;
 173
 174         pci = to_dw_pcie_from_pp(pp);
 175         ks_pcie = to_keystone_pcie(pci);
 176
 177         msi_target = ks_pcie->app.start + MSI_IRQ;
 178         msg->address_lo = lower_32_bits(msi_target);
 179         msg->address_hi = upper_32_bits(msi_target);
 180         msg->data = data->hwirq;
 181
 182         dev_dbg(pci->dev, "msi#%d address_hi %#x address_lo %#x\n",
 183                 (int)data->hwirq, msg->address_hi, msg->address_lo);
 184 }
 185
 186 static int ks_pcie_msi_set_affinity(struct irq_data *irq_data,
 187                                     const struct cpumask *mask, bool force)
 188 {
 189         return -EINVAL;
 190 }
 191
 192 static void ks_pcie_msi_mask(struct irq_data *data)
 193 {
 194         struct pcie_port *pp = irq_data_get_irq_chip_data(data);
 195         struct keystone_pcie *ks_pcie;
 196         u32 irq = data->hwirq;
 197         struct dw_pcie *pci;
 198         unsigned long flags;
 199         u32 reg_offset;
 200         u32 bit_pos;
 201
 202         raw_spin_lock_irqsave(&pp->lock, flags);
 203
 204         pci = to_dw_pcie_from_pp(pp);
 205         ks_pcie = to_keystone_pcie(pci);
 206
 207         reg_offset = irq % 8;
 208         bit_pos = irq >> 3;
 209
 210         ks_pcie_app_writel(ks_pcie, MSI_IRQ_ENABLE_CLR(reg_offset),
 211                            BIT(bit_pos));
 212
 213         raw_spin_unlock_irqrestore(&pp->lock, flags);
 214 }
 215
 216 static void ks_pcie_msi_unmask(struct irq_data *data)
 217 {
 218         struct pcie_port *pp = irq_data_get_irq_chip_data(data);
 219         struct keystone_pcie *ks_pcie;
 220         u32 irq = data->hwirq;
 221         struct dw_pcie *pci;
 222         unsigned long flags;
 223         u32 reg_offset;
 224         u32 bit_pos;
 225
 226         raw_spin_lock_irqsave(&pp->lock, flags);
 227
 228         pci = to_dw_pcie_from_pp(pp);
 229         ks_pcie = to_keystone_pcie(pci);
 230
 231         reg_offset = irq % 8;
 232         bit_pos = irq >> 3;
 233
 234         ks_pcie_app_writel(ks_pcie, MSI_IRQ_ENABLE_SET(reg_offset),
 235                            BIT(bit_pos));
 236
 237         raw_spin_unlock_irqrestore(&pp->lock, flags);
 238 }
 239
 240 static struct irq_chip ks_pcie_msi_irq_chip = {
 241         .name = "KEYSTONE-PCI-MSI",
 242         .irq_ack = ks_pcie_msi_irq_ack,
 243         .irq_compose_msi_msg = ks_pcie_compose_msi_msg,
 244         .irq_set_affinity = ks_pcie_msi_set_affinity,
 245         .irq_mask = ks_pcie_msi_mask,
 246         .irq_unmask = ks_pcie_msi_unmask,
 247 };
 248
 249 static int ks_pcie_msi_host_init(struct pcie_port *pp)
 250 {
 251         pp->msi_irq_chip = &ks_pcie_msi_irq_chip;
 252         return dw_pcie_allocate_domains(pp);
 253 }
 254
 255 static void ks_pcie_handle_legacy_irq(struct keystone_pcie *ks_pcie,
 256                                       int offset)
 257 {
 258         struct dw_pcie *pci = ks_pcie->pci;
 259         struct device *dev = pci->dev;
 260         u32 pending;
 261         int virq;
 262
 263         pending = ks_pcie_app_readl(ks_pcie, IRQ_STATUS(offset));
 264
 265         if (BIT(0) & pending) {
 266                 virq = irq_linear_revmap(ks_pcie->legacy_irq_domain, offset);
 267                 dev_dbg(dev, ": irq: irq_offset %d, virq %d\n", offset, virq);
 268                 generic_handle_irq(virq);
 269         }
 270
 271         /* EOI the INTx interrupt */
 272         ks_pcie_app_writel(ks_pcie, IRQ_EOI, offset);
 273 }
 274
 275 /*
 276  * Dummy function so that DW core doesn't configure MSI
 277  */
 278 static int ks_pcie_am654_msi_host_init(struct pcie_port *pp)
 279 {
 280         return 0;
 281 }
 282
 283 static void ks_pcie_enable_error_irq(struct keystone_pcie *ks_pcie)
 284 {
 285         ks_pcie_app_writel(ks_pcie, ERR_IRQ_ENABLE_SET, ERR_IRQ_ALL);
 286 }
 287
 288 static irqreturn_t ks_pcie_handle_error_irq(struct keystone_pcie *ks_pcie)
 289 {
 290         u32 reg;
 291         struct device *dev = ks_pcie->pci->dev;
 292
 293         reg = ks_pcie_app_readl(ks_pcie, ERR_IRQ_STATUS);
 294         if (!reg)
 295                 return IRQ_NONE;
 296
 297         if (reg & ERR_SYS)
 298                 dev_err(dev, "System Error\n");
 299
 300         if (reg & ERR_FATAL)
 301                 dev_err(dev, "Fatal Error\n");
 302
 303         if (reg & ERR_NONFATAL)
 304                 dev_dbg(dev, "Non Fatal Error\n");
 305
 306         if (reg & ERR_CORR)
 307                 dev_dbg(dev, "Correctable Error\n");
 308
 309         if (!ks_pcie->is_am6 && (reg & ERR_AXI))
 310                 dev_err(dev, "AXI tag lookup fatal Error\n");
 311
 312         if (reg & ERR_AER || (ks_pcie->is_am6 && (reg & AM6_ERR_AER)))
 313                 dev_err(dev, "ECRC Error\n");
 314
 315         ks_pcie_app_writel(ks_pcie, ERR_IRQ_STATUS, reg);
 316
 317         return IRQ_HANDLED;
 318 }
 319
 320 static void ks_pcie_ack_legacy_irq(struct irq_data *d)
 321 {
 322 }
 323
 324 static void ks_pcie_mask_legacy_irq(struct irq_data *d)
 325 {
 326 }
 327
 328 static void ks_pcie_unmask_legacy_irq(struct irq_data *d)
 329 {
 330 }
 331
 332 static struct irq_chip ks_pcie_legacy_irq_chip = {
 333         .name = "Keystone-PCI-Legacy-IRQ",
 334         .irq_ack = ks_pcie_ack_legacy_irq,
 335         .irq_mask = ks_pcie_mask_legacy_irq,
 336         .irq_unmask = ks_pcie_unmask_legacy_irq,
 337 };
 338
 339 static int ks_pcie_init_legacy_irq_map(struct irq_domain *d,
 340                                        unsigned int irq,
 341                                        irq_hw_number_t hw_irq)
 342 {
 343         irq_set_chip_and_handler(irq, &ks_pcie_legacy_irq_chip,
 344                                  handle_level_irq);
 345         irq_set_chip_data(irq, d->host_data);
 346
 347         return 0;
 348 }
 349
 350 static const struct irq_domain_ops ks_pcie_legacy_irq_domain_ops = {
 351         .map = ks_pcie_init_legacy_irq_map,
 352         .xlate = irq_domain_xlate_onetwocell,
 353 };
 354
 355 /**
 356  * ks_pcie_set_dbi_mode() - Set DBI mode to access overlaid BAR mask
 357  * registers
 358  *
 359  * Since modification of dbi_cs2 involves different clock domain, read the
 360  * status back to ensure the transition is complete.
 361  */
 362 static void ks_pcie_set_dbi_mode(struct keystone_pcie *ks_pcie)
 363 {
 364         u32 val;
 365
 366         val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 367         val |= DBI_CS2;
 368         ks_pcie_app_writel(ks_pcie, CMD_STATUS, val);
 369
 370         do {
 371                 val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 372         } while (!(val & DBI_CS2));
 373 }
 374
 375 /**
 376  * ks_pcie_clear_dbi_mode() - Disable DBI mode
 377  *
 378  * Since modification of dbi_cs2 involves different clock domain, read the
 379  * status back to ensure the transition is complete.
 380  */
 381 static void ks_pcie_clear_dbi_mode(struct keystone_pcie *ks_pcie)
 382 {
 383         u32 val;
 384
 385         val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 386         val &= ~DBI_CS2;
 387         ks_pcie_app_writel(ks_pcie, CMD_STATUS, val);
 388
 389         do {
 390                 val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 391         } while (val & DBI_CS2);
 392 }
 393
 394 static void ks_pcie_setup_rc_app_regs(struct keystone_pcie *ks_pcie)
 395 {
 396         u32 val;
 397         struct dw_pcie *pci = ks_pcie->pci;
 398         struct pcie_port *pp = &pci->pp;
 399         u32 num_viewport = pci->num_viewport;
 400         u64 start, end;
 401         struct resource *mem;
 402         int i;
 403
 404         mem = resource_list_first_type(&pp->bridge->windows, IORESOURCE_MEM)->res;
 405         start = mem->start;
 406         end = mem->end;
 407
 408         /* Disable BARs for inbound access */
 409         ks_pcie_set_dbi_mode(ks_pcie);
 410         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, 0);
 411         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_1, 0);
 412         ks_pcie_clear_dbi_mode(ks_pcie);
 413
 414         if (ks_pcie->is_am6)
 415                 return;
 416
 417         val = ilog2(OB_WIN_SIZE);
 418         ks_pcie_app_writel(ks_pcie, OB_SIZE, val);
 419
 420         /* Using Direct 1:1 mapping of RC <-> PCI memory space */
 421         for (i = 0; i < num_viewport && (start < end); i++) {
 422                 ks_pcie_app_writel(ks_pcie, OB_OFFSET_INDEX(i),
 423                                    lower_32_bits(start) | OB_ENABLEN);
 424                 ks_pcie_app_writel(ks_pcie, OB_OFFSET_HI(i),
 425                                    upper_32_bits(start));
 426                 start += OB_WIN_SIZE * SZ_1M;
 427         }
 428
 429         val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 430         val |= OB_XLAT_EN_VAL;
 431         ks_pcie_app_writel(ks_pcie, CMD_STATUS, val);
 432 }
 433
 434 static void __iomem *ks_pcie_other_map_bus(struct pci_bus *bus,
 435                                            unsigned int devfn, int where)
 436 {
 437         struct pcie_port *pp = bus->sysdata;
 438         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
 439         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 440         u32 reg;
 441
 442         reg = CFG_BUS(bus->number) | CFG_DEVICE(PCI_SLOT(devfn)) |
 443                 CFG_FUNC(PCI_FUNC(devfn));
 444         if (!pci_is_root_bus(bus->parent))
 445                 reg |= CFG_TYPE1;
 446         ks_pcie_app_writel(ks_pcie, CFG_SETUP, reg);
 447
 448         return pp->va_cfg0_base + where;
 449 }
 450
 451 static struct pci_ops ks_child_pcie_ops = {
 452         .map_bus = ks_pcie_other_map_bus,
 453         .read = pci_generic_config_read,
 454         .write = pci_generic_config_write,
 455 };
 456
 457 /**
 458  * ks_pcie_v3_65_add_bus() - keystone add_bus post initialization
 459  *
 460  * This sets BAR0 to enable inbound access for MSI_IRQ register
 461  */
 462 static int ks_pcie_v3_65_add_bus(struct pci_bus *bus)
 463 {
 464         struct pcie_port *pp = bus->sysdata;
 465         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
 466         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 467
 468         if (!pci_is_root_bus(bus))
 469                 return 0;
 470
 471         /* Configure and set up BAR0 */
 472         ks_pcie_set_dbi_mode(ks_pcie);
 473
 474         /* Enable BAR0 */
 475         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, 1);
 476         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, SZ_4K - 1);
 477
 478         ks_pcie_clear_dbi_mode(ks_pcie);
 479
 480          /*
 481           * For BAR0, just setting bus address for inbound writes (MSI) should
 482           * be sufficient.  Use physical address to avoid any conflicts.
 483           */
 484         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, ks_pcie->app.start);
 485
 486         return 0;
 487 }
 488
 489 static struct pci_ops ks_pcie_ops = {
 490         .map_bus = dw_pcie_own_conf_map_bus,
 491         .read = pci_generic_config_read,
 492         .write = pci_generic_config_write,
 493         .add_bus = ks_pcie_v3_65_add_bus,
 494 };
 495
 496 /**
 497  * ks_pcie_link_up() - Check if link up
 498  */
 499 static int ks_pcie_link_up(struct dw_pcie *pci)
 500 {
 501         u32 val;
 502
 503         val = dw_pcie_readl_dbi(pci, PCIE_PORT_DEBUG0);
 504         val &= PORT_LOGIC_LTSSM_STATE_MASK;
 505         return (val == PORT_LOGIC_LTSSM_STATE_L0);
 506 }
 507
 508 static void ks_pcie_stop_link(struct dw_pcie *pci)
 509 {
 510         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 511         u32 val;
 512
 513         /* Disable Link training */
 514         val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 515         val &= ~LTSSM_EN_VAL;
 516         ks_pcie_app_writel(ks_pcie, CMD_STATUS, val);
 517 }
 518
 519 static int ks_pcie_start_link(struct dw_pcie *pci)
 520 {
 521         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 522         struct device *dev = pci->dev;
 523         u32 val;
 524
 525         if (dw_pcie_link_up(pci)) {
 526                 dev_dbg(dev, "link is already up\n");
 527                 return 0;
 528         }
 529
 530         /* Initiate Link Training */
 531         val = ks_pcie_app_readl(ks_pcie, CMD_STATUS);
 532         ks_pcie_app_writel(ks_pcie, CMD_STATUS, LTSSM_EN_VAL | val);
 533
 534         return 0;
 535 }
 536
 537 static void ks_pcie_quirk(struct pci_dev *dev)
 538 {
 539         struct pci_bus *bus = dev->bus;
 540         struct pci_dev *bridge;
 541         static const struct pci_device_id rc_pci_devids[] = {
 542                 { PCI_DEVICE(PCI_VENDOR_ID_TI, PCIE_RC_K2HK),
 543                  .class = PCI_CLASS_BRIDGE_PCI << 8, .class_mask = ~0, },
 544                 { PCI_DEVICE(PCI_VENDOR_ID_TI, PCIE_RC_K2E),
 545                  .class = PCI_CLASS_BRIDGE_PCI << 8, .class_mask = ~0, },
 546                 { PCI_DEVICE(PCI_VENDOR_ID_TI, PCIE_RC_K2L),
 547                  .class = PCI_CLASS_BRIDGE_PCI << 8, .class_mask = ~0, },
 548                 { PCI_DEVICE(PCI_VENDOR_ID_TI, PCIE_RC_K2G),
 549                  .class = PCI_CLASS_BRIDGE_PCI << 8, .class_mask = ~0, },
 550                 { 0, },
 551         };
 552
 553         if (pci_is_root_bus(bus))
 554                 bridge = dev;
 555
 556         /* look for the host bridge */
 557         while (!pci_is_root_bus(bus)) {
 558                 bridge = bus->self;
 559                 bus = bus->parent;
 560         }
 561
 562         if (!bridge)
 563                 return;
 564
 565         /*
 566          * Keystone PCI controller has a h/w limitation of
 567          * 256 bytes maximum read request size.  It can't handle
 568          * anything higher than this.  So force this limit on
 569          * all downstream devices.
 570          */
 571         if (pci_match_id(rc_pci_devids, bridge)) {
 572                 if (pcie_get_readrq(dev) > 256) {
 573                         dev_info(&dev->dev, "limiting MRRS to 256\n");
 574                         pcie_set_readrq(dev, 256);
 575                 }
 576         }
 577 }
 578 DECLARE_PCI_FIXUP_ENABLE(PCI_ANY_ID, PCI_ANY_ID, ks_pcie_quirk);
 579
 580 static void ks_pcie_msi_irq_handler(struct irq_desc *desc)
 581 {
 582         unsigned int irq = desc->irq_data.hwirq;
 583         struct keystone_pcie *ks_pcie = irq_desc_get_handler_data(desc);
 584         u32 offset = irq - ks_pcie->msi_host_irq;
 585         struct dw_pcie *pci = ks_pcie->pci;
 586         struct pcie_port *pp = &pci->pp;
 587         struct device *dev = pci->dev;
 588         struct irq_chip *chip = irq_desc_get_chip(desc);
 589         u32 vector, virq, reg, pos;
 590
 591         dev_dbg(dev, "%s, irq %d\n", __func__, irq);
 592
 593         /*
 594          * The chained irq handler installation would have replaced normal
 595          * interrupt driver handler so we need to take care of mask/unmask and
 596          * ack operation.
 597          */
 598         chained_irq_enter(chip, desc);
 599
 600         reg = ks_pcie_app_readl(ks_pcie, MSI_IRQ_STATUS(offset));
 601         /*
 602          * MSI0 status bit 0-3 shows vectors 0, 8, 16, 24, MSI1 status bit
 603          * shows 1, 9, 17, 25 and so forth
 604          */
 605         for (pos = 0; pos < 4; pos++) {
 606                 if (!(reg & BIT(pos)))
 607                         continue;
 608
 609                 vector = offset + (pos << 3);
 610                 virq = irq_linear_revmap(pp->irq_domain, vector);
 611                 dev_dbg(dev, "irq: bit %d, vector %d, virq %d\n", pos, vector,
 612                         virq);
 613                 generic_handle_irq(virq);
 614         }
 615
 616         chained_irq_exit(chip, desc);
 617 }
 618
 619 /**
 620  * ks_pcie_legacy_irq_handler() - Handle legacy interrupt
 621  * @irq: IRQ line for legacy interrupts
 622  * @desc: Pointer to irq descriptor
 623  *
 624  * Traverse through pending legacy interrupts and invoke handler for each. Also
 625  * takes care of interrupt controller level mask/ack operation.
 626  */
 627 static void ks_pcie_legacy_irq_handler(struct irq_desc *desc)
 628 {
 629         unsigned int irq = irq_desc_get_irq(desc);
 630         struct keystone_pcie *ks_pcie = irq_desc_get_handler_data(desc);
 631         struct dw_pcie *pci = ks_pcie->pci;
 632         struct device *dev = pci->dev;
 633         u32 irq_offset = irq - ks_pcie->legacy_host_irqs[0];
 634         struct irq_chip *chip = irq_desc_get_chip(desc);
 635
 636         dev_dbg(dev, ": Handling legacy irq %d\n", irq);
 637
 638         /*
 639          * The chained irq handler installation would have replaced normal
 640          * interrupt driver handler so we need to take care of mask/unmask and
 641          * ack operation.
 642          */
 643         chained_irq_enter(chip, desc);
 644         ks_pcie_handle_legacy_irq(ks_pcie, irq_offset);
 645         chained_irq_exit(chip, desc);
 646 }
 647
 648 static int ks_pcie_config_msi_irq(struct keystone_pcie *ks_pcie)
 649 {
 650         struct device *dev = ks_pcie->pci->dev;
 651         struct device_node *np = ks_pcie->np;
 652         struct device_node *intc_np;
 653         struct irq_data *irq_data;
 654         int irq_count, irq, ret, i;
 655
 656         if (!IS_ENABLED(CONFIG_PCI_MSI))
 657                 return 0;
 658
 659         intc_np = of_get_child_by_name(np, "msi-interrupt-controller");
 660         if (!intc_np) {
 661                 if (ks_pcie->is_am6)
 662                         return 0;
 663                 dev_warn(dev, "msi-interrupt-controller node is absent\n");
 664                 return -EINVAL;
 665         }
 666
 667         irq_count = of_irq_count(intc_np);
 668         if (!irq_count) {
 669                 dev_err(dev, "No IRQ entries in msi-interrupt-controller\n");
 670                 ret = -EINVAL;
 671                 goto err;
 672         }
 673
 674         for (i = 0; i < irq_count; i++) {
 675                 irq = irq_of_parse_and_map(intc_np, i);
 676                 if (!irq) {
 677                         ret = -EINVAL;
 678                         goto err;
 679                 }
 680
 681                 if (!ks_pcie->msi_host_irq) {
 682                         irq_data = irq_get_irq_data(irq);
 683                         if (!irq_data) {
 684                                 ret = -EINVAL;
 685                                 goto err;
 686                         }
 687                         ks_pcie->msi_host_irq = irq_data->hwirq;
 688                 }
 689
 690                 irq_set_chained_handler_and_data(irq, ks_pcie_msi_irq_handler,
 691                                                  ks_pcie);
 692         }
 693
 694         of_node_put(intc_np);
 695         return 0;
 696
 697 err:
 698         of_node_put(intc_np);
 699         return ret;
 700 }
 701
 702 static int ks_pcie_config_legacy_irq(struct keystone_pcie *ks_pcie)
 703 {
 704         struct device *dev = ks_pcie->pci->dev;
 705         struct irq_domain *legacy_irq_domain;
 706         struct device_node *np = ks_pcie->np;
 707         struct device_node *intc_np;
 708         int irq_count, irq, ret = 0, i;
 709
 710         intc_np = of_get_child_by_name(np, "legacy-interrupt-controller");
 711         if (!intc_np) {
 712                 /*
 713                  * Since legacy interrupts are modeled as edge-interrupts in
 714                  * AM6, keep it disabled for now.
 715                  */
 716                 if (ks_pcie->is_am6)
 717                         return 0;
 718                 dev_warn(dev, "legacy-interrupt-controller node is absent\n");
 719                 return -EINVAL;
 720         }
 721
 722         irq_count = of_irq_count(intc_np);
 723         if (!irq_count) {
 724                 dev_err(dev, "No IRQ entries in legacy-interrupt-controller\n");
 725                 ret = -EINVAL;
 726                 goto err;
 727         }
 728
 729         for (i = 0; i < irq_count; i++) {
 730                 irq = irq_of_parse_and_map(intc_np, i);
 731                 if (!irq) {
 732                         ret = -EINVAL;
 733                         goto err;
 734                 }
 735                 ks_pcie->legacy_host_irqs[i] = irq;
 736
 737                 irq_set_chained_handler_and_data(irq,
 738                                                  ks_pcie_legacy_irq_handler,
 739                                                  ks_pcie);
 740         }
 741
 742         legacy_irq_domain =
 743                 irq_domain_add_linear(intc_np, PCI_NUM_INTX,
 744                                       &ks_pcie_legacy_irq_domain_ops, NULL);
 745         if (!legacy_irq_domain) {
 746                 dev_err(dev, "Failed to add irq domain for legacy irqs\n");
 747                 ret = -EINVAL;
 748                 goto err;
 749         }
 750         ks_pcie->legacy_irq_domain = legacy_irq_domain;
 751
 752         for (i = 0; i < PCI_NUM_INTX; i++)
 753                 ks_pcie_app_writel(ks_pcie, IRQ_ENABLE_SET(i), INTx_EN);
 754
 755 err:
 756         of_node_put(intc_np);
 757         return ret;
 758 }
 759
 760 #ifdef CONFIG_ARM
 761 /*
 762  * When a PCI device does not exist during config cycles, keystone host gets a
 763  * bus error instead of returning 0xffffffff. This handler always returns 0
 764  * for this kind of faults.
 765  */
 766 static int ks_pcie_fault(unsigned long addr, unsigned int fsr,
 767                          struct pt_regs *regs)
 768 {
 769         unsigned long instr = *(unsigned long *) instruction_pointer(regs);
 770
 771         if ((instr & 0x0e100090) == 0x00100090) {
 772                 int reg = (instr >> 12) & 15;
 773
 774                 regs->uregs[reg] = -1;
 775                 regs->ARM_pc += 4;
 776         }
 777
 778         return 0;
 779 }
 780 #endif
 781
 782 static int __init ks_pcie_init_id(struct keystone_pcie *ks_pcie)
 783 {
 784         int ret;
 785         unsigned int id;
 786         struct regmap *devctrl_regs;
 787         struct dw_pcie *pci = ks_pcie->pci;
 788         struct device *dev = pci->dev;
 789         struct device_node *np = dev->of_node;
 790
 791         devctrl_regs = syscon_regmap_lookup_by_phandle(np, "ti,syscon-pcie-id");
 792         if (IS_ERR(devctrl_regs))
 793                 return PTR_ERR(devctrl_regs);
 794
 795         ret = regmap_read(devctrl_regs, 0, &id);
 796         if (ret)
 797                 return ret;
 798
 799         dw_pcie_dbi_ro_wr_en(pci);
 800         dw_pcie_writew_dbi(pci, PCI_VENDOR_ID, id & PCIE_VENDORID_MASK);
 801         dw_pcie_writew_dbi(pci, PCI_DEVICE_ID, id >> PCIE_DEVICEID_SHIFT);
 802         dw_pcie_dbi_ro_wr_dis(pci);
 803
 804         return 0;
 805 }
 806
 807 static int __init ks_pcie_host_init(struct pcie_port *pp)
 808 {
 809         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
 810         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 811         int ret;
 812
 813         pp->bridge->ops = &ks_pcie_ops;
 814         pp->bridge->child_ops = &ks_child_pcie_ops;
 815
 816         ret = ks_pcie_config_legacy_irq(ks_pcie);
 817         if (ret)
 818                 return ret;
 819
 820         ret = ks_pcie_config_msi_irq(ks_pcie);
 821         if (ret)
 822                 return ret;
 823
 824         dw_pcie_setup_rc(pp);
 825
 826         ks_pcie_stop_link(pci);
 827         ks_pcie_setup_rc_app_regs(ks_pcie);
 828         writew(PCI_IO_RANGE_TYPE_32 | (PCI_IO_RANGE_TYPE_32 << 8),
 829                         pci->dbi_base + PCI_IO_BASE);
 830
 831         ret = ks_pcie_init_id(ks_pcie);
 832         if (ret < 0)
 833                 return ret;
 834
 835 #ifdef CONFIG_ARM
 836         /*
 837          * PCIe access errors that result into OCP errors are caught by ARM as
 838          * "External aborts"
 839          */
 840         hook_fault_code(17, ks_pcie_fault, SIGBUS, 0,
 841                         "Asynchronous external abort");
 842 #endif
 843
 844         ks_pcie_start_link(pci);
 845         dw_pcie_wait_for_link(pci);
 846
 847         return 0;
 848 }
 849
 850 static const struct dw_pcie_host_ops ks_pcie_host_ops = {
 851         .host_init = ks_pcie_host_init,
 852         .msi_host_init = ks_pcie_msi_host_init,
 853 };
 854
 855 static const struct dw_pcie_host_ops ks_pcie_am654_host_ops = {
 856         .host_init = ks_pcie_host_init,
 857         .msi_host_init = ks_pcie_am654_msi_host_init,
 858 };
 859
 860 static irqreturn_t ks_pcie_err_irq_handler(int irq, void *priv)
 861 {
 862         struct keystone_pcie *ks_pcie = priv;
 863
 864         return ks_pcie_handle_error_irq(ks_pcie);
 865 }
 866
 867 static int __init ks_pcie_add_pcie_port(struct keystone_pcie *ks_pcie,
 868                                         struct platform_device *pdev)
 869 {
 870         struct dw_pcie *pci = ks_pcie->pci;
 871         struct pcie_port *pp = &pci->pp;
 872         struct device *dev = &pdev->dev;
 873         int ret;
 874
 875         ret = dw_pcie_host_init(pp);
 876         if (ret) {
 877                 dev_err(dev, "failed to initialize host\n");
 878                 return ret;
 879         }
 880
 881         return 0;
 882 }
 883
 884 static void ks_pcie_am654_write_dbi2(struct dw_pcie *pci, void __iomem *base,
 885                                      u32 reg, size_t size, u32 val)
 886 {
 887         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 888
 889         ks_pcie_set_dbi_mode(ks_pcie);
 890         dw_pcie_write(base + reg, size, val);
 891         ks_pcie_clear_dbi_mode(ks_pcie);
 892 }
 893
 894 static const struct dw_pcie_ops ks_pcie_dw_pcie_ops = {
 895         .start_link = ks_pcie_start_link,
 896         .stop_link = ks_pcie_stop_link,
 897         .link_up = ks_pcie_link_up,
 898         .write_dbi2 = ks_pcie_am654_write_dbi2,
 899 };
 900
 901 static void ks_pcie_am654_ep_init(struct dw_pcie_ep *ep)
 902 {
 903         struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
 904         int flags;
 905
 906         ep->page_size = AM654_WIN_SIZE;
 907         flags = PCI_BASE_ADDRESS_SPACE_MEMORY | PCI_BASE_ADDRESS_MEM_TYPE_32;
 908         dw_pcie_writel_dbi2(pci, PCI_BASE_ADDRESS_0, APP_ADDR_SPACE_0 - 1);
 909         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, flags);
 910 }
 911
 912 static void ks_pcie_am654_raise_legacy_irq(struct keystone_pcie *ks_pcie)
 913 {
 914         struct dw_pcie *pci = ks_pcie->pci;
 915         u8 int_pin;
 916
 917         int_pin = dw_pcie_readb_dbi(pci, PCI_INTERRUPT_PIN);
 918         if (int_pin == 0 || int_pin > 4)
 919                 return;
 920
 921         ks_pcie_app_writel(ks_pcie, PCIE_LEGACY_IRQ_ENABLE_SET(int_pin),
 922                            INT_ENABLE);
 923         ks_pcie_app_writel(ks_pcie, PCIE_EP_IRQ_SET, INT_ENABLE);
 924         mdelay(1);
 925         ks_pcie_app_writel(ks_pcie, PCIE_EP_IRQ_CLR, INT_ENABLE);
 926         ks_pcie_app_writel(ks_pcie, PCIE_LEGACY_IRQ_ENABLE_CLR(int_pin),
 927                            INT_ENABLE);
 928 }
 929
 930 static int ks_pcie_am654_raise_irq(struct dw_pcie_ep *ep, u8 func_no,
 931                                    enum pci_epc_irq_type type,
 932                                    u16 interrupt_num)
 933 {
 934         struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
 935         struct keystone_pcie *ks_pcie = to_keystone_pcie(pci);
 936
 937         switch (type) {
 938         case PCI_EPC_IRQ_LEGACY:
 939                 ks_pcie_am654_raise_legacy_irq(ks_pcie);
 940                 break;
 941         case PCI_EPC_IRQ_MSI:
 942                 dw_pcie_ep_raise_msi_irq(ep, func_no, interrupt_num);
 943                 break;
 944         case PCI_EPC_IRQ_MSIX:
 945                 dw_pcie_ep_raise_msix_irq(ep, func_no, interrupt_num);
 946                 break;
 947         default:
 948                 dev_err(pci->dev, "UNKNOWN IRQ type\n");
 949                 return -EINVAL;
 950         }
 951
 952         return 0;
 953 }
 954
 955 static const struct pci_epc_features ks_pcie_am654_epc_features = {
 956         .linkup_notifier = false,
 957         .msi_capable = true,
 958         .msix_capable = true,
 959         .reserved_bar = 1 << BAR_0 | 1 << BAR_1,
 960         .bar_fixed_64bit = 1 << BAR_0,
 961         .bar_fixed_size[2] = SZ_1M,
 962         .bar_fixed_size[3] = SZ_64K,
 963         .bar_fixed_size[4] = 256,
 964         .bar_fixed_size[5] = SZ_1M,
 965         .align = SZ_1M,
 966 };
 967
 968 static const struct pci_epc_features*
 969 ks_pcie_am654_get_features(struct dw_pcie_ep *ep)
 970 {
 971         return &ks_pcie_am654_epc_features;
 972 }
 973
 974 static const struct dw_pcie_ep_ops ks_pcie_am654_ep_ops = {
 975         .ep_init = ks_pcie_am654_ep_init,
 976         .raise_irq = ks_pcie_am654_raise_irq,
 977         .get_features = &ks_pcie_am654_get_features,
 978 };
 979
 980 static int __init ks_pcie_add_pcie_ep(struct keystone_pcie *ks_pcie,
 981                                       struct platform_device *pdev)
 982 {
 983         int ret;
 984         struct dw_pcie_ep *ep;
 985         struct resource *res;
 986         struct device *dev = &pdev->dev;
 987         struct dw_pcie *pci = ks_pcie->pci;
 988
 989         ep = &pci->ep;
 990
 991         res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "addr_space");
 992         if (!res)
 993                 return -EINVAL;
 994
 995         ep->phys_base = res->start;
 996         ep->addr_size = resource_size(res);
 997
 998         ret = dw_pcie_ep_init(ep);
 999         if (ret) {
1000                 dev_err(dev, "failed to initialize endpoint\n");
1001                 return ret;
1002         }
1003
1004         return 0;
1005 }
1006
1007 static void ks_pcie_disable_phy(struct keystone_pcie *ks_pcie)
1008 {
1009         int num_lanes = ks_pcie->num_lanes;
1010
1011         while (num_lanes--) {
1012                 phy_power_off(ks_pcie->phy[num_lanes]);
1013                 phy_exit(ks_pcie->phy[num_lanes]);
1014         }
1015 }
1016
1017 static int ks_pcie_enable_phy(struct keystone_pcie *ks_pcie)
1018 {
1019         int i;
1020         int ret;
1021         int num_lanes = ks_pcie->num_lanes;
1022
1023         for (i = 0; i < num_lanes; i++) {
1024                 ret = phy_reset(ks_pcie->phy[i]);
1025                 if (ret < 0)
1026                         goto err_phy;
1027
1028                 ret = phy_init(ks_pcie->phy[i]);
1029                 if (ret < 0)
1030                         goto err_phy;
1031
1032                 ret = phy_power_on(ks_pcie->phy[i]);
1033                 if (ret < 0) {
1034                         phy_exit(ks_pcie->phy[i]);
1035                         goto err_phy;
1036                 }
1037         }
1038
1039         return 0;
1040
1041 err_phy:
1042         while (--i >= 0) {
1043                 phy_power_off(ks_pcie->phy[i]);
1044                 phy_exit(ks_pcie->phy[i]);
1045         }
1046
1047         return ret;
1048 }
1049
1050 static int ks_pcie_set_mode(struct device *dev)
1051 {
1052         struct device_node *np = dev->of_node;
1053         struct regmap *syscon;
1054         u32 val;
1055         u32 mask;
1056         int ret = 0;
1057
1058         syscon = syscon_regmap_lookup_by_phandle(np, "ti,syscon-pcie-mode");
1059         if (IS_ERR(syscon))
1060                 return 0;
1061
1062         mask = KS_PCIE_DEV_TYPE_MASK | KS_PCIE_SYSCLOCKOUTEN;
1063         val = KS_PCIE_DEV_TYPE(RC) | KS_PCIE_SYSCLOCKOUTEN;
1064
1065         ret = regmap_update_bits(syscon, 0, mask, val);
1066         if (ret) {
1067                 dev_err(dev, "failed to set pcie mode\n");
1068                 return ret;
1069         }
1070
1071         return 0;
1072 }
1073
1074 static int ks_pcie_am654_set_mode(struct device *dev,
1075                                   enum dw_pcie_device_mode mode)
1076 {
1077         struct device_node *np = dev->of_node;
1078         struct regmap *syscon;
1079         u32 val;
1080         u32 mask;
1081         int ret = 0;
1082
1083         syscon = syscon_regmap_lookup_by_phandle(np, "ti,syscon-pcie-mode");
1084         if (IS_ERR(syscon))
1085                 return 0;
1086
1087         mask = AM654_PCIE_DEV_TYPE_MASK;
1088
1089         switch (mode) {
1090         case DW_PCIE_RC_TYPE:
1091                 val = RC;
1092                 break;
1093         case DW_PCIE_EP_TYPE:
1094                 val = EP;
1095                 break;
1096         default:
1097                 dev_err(dev, "INVALID device type %d\n", mode);
1098                 return -EINVAL;
1099         }
1100
1101         ret = regmap_update_bits(syscon, 0, mask, val);
1102         if (ret) {
1103                 dev_err(dev, "failed to set pcie mode\n");
1104                 return ret;
1105         }
1106
1107         return 0;
1108 }
1109
1110 static const struct ks_pcie_of_data ks_pcie_rc_of_data = {
1111         .host_ops = &ks_pcie_host_ops,
1112         .version = 0x365A,
1113 };
1114
1115 static const struct ks_pcie_of_data ks_pcie_am654_rc_of_data = {
1116         .host_ops = &ks_pcie_am654_host_ops,
1117         .mode = DW_PCIE_RC_TYPE,
1118         .version = 0x490A,
1119 };
1120
1121 static const struct ks_pcie_of_data ks_pcie_am654_ep_of_data = {
1122         .ep_ops = &ks_pcie_am654_ep_ops,
1123         .mode = DW_PCIE_EP_TYPE,
1124         .version = 0x490A,
1125 };
1126
1127 static const struct of_device_id ks_pcie_of_match[] = {
1128         {
1129                 .type = "pci",
1130                 .data = &ks_pcie_rc_of_data,
1131                 .compatible = "ti,keystone-pcie",
1132         },
1133         {
1134                 .data = &ks_pcie_am654_rc_of_data,
1135                 .compatible = "ti,am654-pcie-rc",
1136         },
1137         {
1138                 .data = &ks_pcie_am654_ep_of_data,
1139                 .compatible = "ti,am654-pcie-ep",
1140         },
1141         { },
1142 };
1143
1144 static int __init ks_pcie_probe(struct platform_device *pdev)
1145 {
1146         const struct dw_pcie_host_ops *host_ops;
1147         const struct dw_pcie_ep_ops *ep_ops;
1148         struct device *dev = &pdev->dev;
1149         struct device_node *np = dev->of_node;
1150         const struct ks_pcie_of_data *data;
1151         const struct of_device_id *match;
1152         enum dw_pcie_device_mode mode;
1153         struct dw_pcie *pci;
1154         struct keystone_pcie *ks_pcie;
1155         struct device_link **link;
1156         struct gpio_desc *gpiod;
1157         struct resource *res;
1158         unsigned int version;
1159         void __iomem *base;
1160         struct phy **phy;
1161         u32 num_lanes;
1162         char name[10];
1163         int ret;
1164         int irq;
1165         int i;
1166
1167         match = of_match_device(of_match_ptr(ks_pcie_of_match), dev);
1168         data = (struct ks_pcie_of_data *)match->data;
1169         if (!data)
1170                 return -EINVAL;
1171
1172         version = data->version;
1173         host_ops = data->host_ops;
1174         ep_ops = data->ep_ops;
1175         mode = data->mode;
1176
1177         ks_pcie = devm_kzalloc(dev, sizeof(*ks_pcie), GFP_KERNEL);
1178         if (!ks_pcie)
1179                 return -ENOMEM;
1180
1181         pci = devm_kzalloc(dev, sizeof(*pci), GFP_KERNEL);
1182         if (!pci)
1183                 return -ENOMEM;
1184
1185         res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "app");
1186         ks_pcie->va_app_base = devm_ioremap_resource(dev, res);
1187         if (IS_ERR(ks_pcie->va_app_base))
1188                 return PTR_ERR(ks_pcie->va_app_base);
1189
1190         ks_pcie->app = *res;
1191
1192         res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "dbics");
1193         base = devm_pci_remap_cfg_resource(dev, res);
1194         if (IS_ERR(base))
1195                 return PTR_ERR(base);
1196
1197         if (of_device_is_compatible(np, "ti,am654-pcie-rc"))
1198                 ks_pcie->is_am6 = true;
1199
1200         pci->dbi_base = base;
1201         pci->dbi_base2 = base;
1202         pci->dev = dev;
1203         pci->ops = &ks_pcie_dw_pcie_ops;
1204         pci->version = version;
1205
1206         irq = platform_get_irq(pdev, 0);
1207         if (irq < 0)
1208                 return irq;
1209
1210         ret = request_irq(irq, ks_pcie_err_irq_handler, IRQF_SHARED,
1211                           "ks-pcie-error-irq", ks_pcie);
1212         if (ret < 0) {
1213                 dev_err(dev, "failed to request error IRQ %d\n",
1214                         irq);
1215                 return ret;
1216         }
1217
1218         ret = of_property_read_u32(np, "num-lanes", &num_lanes);
1219         if (ret)
1220                 num_lanes = 1;
1221
1222         phy = devm_kzalloc(dev, sizeof(*phy) * num_lanes, GFP_KERNEL);
1223         if (!phy)
1224                 return -ENOMEM;
1225
1226         link = devm_kzalloc(dev, sizeof(*link) * num_lanes, GFP_KERNEL);
1227         if (!link)
1228                 return -ENOMEM;
1229
1230         for (i = 0; i < num_lanes; i++) {
1231                 snprintf(name, sizeof(name), "pcie-phy%d", i);
1232                 phy[i] = devm_phy_optional_get(dev, name);
1233                 if (IS_ERR(phy[i])) {
1234                         ret = PTR_ERR(phy[i]);
1235                         goto err_link;
1236                 }
1237
1238                 if (!phy[i])
1239                         continue;
1240
1241                 link[i] = device_link_add(dev, &phy[i]->dev, DL_FLAG_STATELESS);
1242                 if (!link[i]) {
1243                         ret = -EINVAL;
1244                         goto err_link;
1245                 }
1246         }
1247
1248         ks_pcie->np = np;
1249         ks_pcie->pci = pci;
1250         ks_pcie->link = link;
1251         ks_pcie->num_lanes = num_lanes;
1252         ks_pcie->phy = phy;
1253
1254         gpiod = devm_gpiod_get_optional(dev, "reset",
1255                                         GPIOD_OUT_LOW);
1256         if (IS_ERR(gpiod)) {
1257                 ret = PTR_ERR(gpiod);
1258                 if (ret != -EPROBE_DEFER)
1259                         dev_err(dev, "Failed to get reset GPIO\n");
1260                 goto err_link;
1261         }
1262
1263         ret = ks_pcie_enable_phy(ks_pcie);
1264         if (ret) {
1265                 dev_err(dev, "failed to enable phy\n");
1266                 goto err_link;
1267         }
1268
1269         platform_set_drvdata(pdev, ks_pcie);
1270         pm_runtime_enable(dev);
1271         ret = pm_runtime_get_sync(dev);
1272         if (ret < 0) {
1273                 dev_err(dev, "pm_runtime_get_sync failed\n");
1274                 goto err_get_sync;
1275         }
1276
1277         if (pci->version >= 0x480A)
1278                 ret = ks_pcie_am654_set_mode(dev, mode);
1279         else
1280                 ret = ks_pcie_set_mode(dev);
1281         if (ret < 0)
1282                 goto err_get_sync;
1283
1284         switch (mode) {
1285         case DW_PCIE_RC_TYPE:
1286                 if (!IS_ENABLED(CONFIG_PCI_KEYSTONE_HOST)) {
1287                         ret = -ENODEV;
1288                         goto err_get_sync;
1289                 }
1290
1291                 /*
1292                  * "Power Sequencing and Reset Signal Timings" table in
1293                  * PCI EXPRESS CARD ELECTROMECHANICAL SPECIFICATION, REV. 2.0
1294                  * indicates PERST# should be deasserted after minimum of 100us
1295                  * once REFCLK is stable. The REFCLK to the connector in RC
1296                  * mode is selected while enabling the PHY. So deassert PERST#
1297                  * after 100 us.
1298                  */
1299                 if (gpiod) {
1300                         usleep_range(100, 200);
1301                         gpiod_set_value_cansleep(gpiod, 1);
1302                 }
1303
1304                 pci->pp.ops = host_ops;
1305                 ret = ks_pcie_add_pcie_port(ks_pcie, pdev);
1306                 if (ret < 0)
1307                         goto err_get_sync;
1308                 break;
1309         case DW_PCIE_EP_TYPE:
1310                 if (!IS_ENABLED(CONFIG_PCI_KEYSTONE_EP)) {
1311                         ret = -ENODEV;
1312                         goto err_get_sync;
1313                 }
1314
1315                 pci->ep.ops = ep_ops;
1316                 ret = ks_pcie_add_pcie_ep(ks_pcie, pdev);
1317                 if (ret < 0)
1318                         goto err_get_sync;
1319                 break;
1320         default:
1321                 dev_err(dev, "INVALID device type %d\n", mode);
1322         }
1323
1324         ks_pcie_enable_error_irq(ks_pcie);
1325
1326         return 0;
1327
1328 err_get_sync:
1329         pm_runtime_put(dev);
1330         pm_runtime_disable(dev);
1331         ks_pcie_disable_phy(ks_pcie);
1332
1333 err_link:
1334         while (--i >= 0 && link[i])
1335                 device_link_del(link[i]);
1336
1337         return ret;
1338 }
1339
1340 static int __exit ks_pcie_remove(struct platform_device *pdev)
1341 {
1342         struct keystone_pcie *ks_pcie = platform_get_drvdata(pdev);
1343         struct device_link **link = ks_pcie->link;
1344         int num_lanes = ks_pcie->num_lanes;
1345         struct device *dev = &pdev->dev;
1346
1347         pm_runtime_put(dev);
1348         pm_runtime_disable(dev);
1349         ks_pcie_disable_phy(ks_pcie);
1350         while (num_lanes--)
1351                 device_link_del(link[num_lanes]);
1352
1353         return 0;
1354 }
1355
1356 static struct platform_driver ks_pcie_driver __refdata = {
1357         .probe  = ks_pcie_probe,
1358         .remove = __exit_p(ks_pcie_remove),
1359         .driver = {
1360                 .name   = "keystone-pcie",
1361                 .of_match_table = of_match_ptr(ks_pcie_of_match),
1362         },
1363 };
1364 builtin_platform_driver(ks_pcie_driver);