Merge branch 'remotes/lorenzo/pci/qcom'
[linux-2.6-microblaze.git] / 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 <m-karicheri2@ti.com>
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);