GNU Linux-libre 5.10.153-gnu1
[releases.git] / drivers / pci / controller / dwc / pcie-tegra194.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * PCIe host controller driver for Tegra194 SoC
4  *
5  * Copyright (C) 2019 NVIDIA Corporation.
6  *
7  * Author: Vidya Sagar <vidyas@nvidia.com>
8  */
9
10 #include <linux/clk.h>
11 #include <linux/debugfs.h>
12 #include <linux/delay.h>
13 #include <linux/gpio.h>
14 #include <linux/gpio/consumer.h>
15 #include <linux/interrupt.h>
16 #include <linux/iopoll.h>
17 #include <linux/kernel.h>
18 #include <linux/module.h>
19 #include <linux/of.h>
20 #include <linux/of_device.h>
21 #include <linux/of_gpio.h>
22 #include <linux/of_irq.h>
23 #include <linux/of_pci.h>
24 #include <linux/pci.h>
25 #include <linux/phy/phy.h>
26 #include <linux/pinctrl/consumer.h>
27 #include <linux/platform_device.h>
28 #include <linux/pm_runtime.h>
29 #include <linux/random.h>
30 #include <linux/reset.h>
31 #include <linux/resource.h>
32 #include <linux/types.h>
33 #include "pcie-designware.h"
34 #include <soc/tegra/bpmp.h>
35 #include <soc/tegra/bpmp-abi.h>
36 #include "../../pci.h"
37
38 #define APPL_PINMUX                             0x0
39 #define APPL_PINMUX_PEX_RST                     BIT(0)
40 #define APPL_PINMUX_CLKREQ_OVERRIDE_EN          BIT(2)
41 #define APPL_PINMUX_CLKREQ_OVERRIDE             BIT(3)
42 #define APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE_EN   BIT(4)
43 #define APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE      BIT(5)
44
45 #define APPL_CTRL                               0x4
46 #define APPL_CTRL_SYS_PRE_DET_STATE             BIT(6)
47 #define APPL_CTRL_LTSSM_EN                      BIT(7)
48 #define APPL_CTRL_HW_HOT_RST_EN                 BIT(20)
49 #define APPL_CTRL_HW_HOT_RST_MODE_MASK          GENMASK(1, 0)
50 #define APPL_CTRL_HW_HOT_RST_MODE_SHIFT         22
51 #define APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST      0x1
52
53 #define APPL_INTR_EN_L0_0                       0x8
54 #define APPL_INTR_EN_L0_0_LINK_STATE_INT_EN     BIT(0)
55 #define APPL_INTR_EN_L0_0_MSI_RCV_INT_EN        BIT(4)
56 #define APPL_INTR_EN_L0_0_INT_INT_EN            BIT(8)
57 #define APPL_INTR_EN_L0_0_PCI_CMD_EN_INT_EN     BIT(15)
58 #define APPL_INTR_EN_L0_0_CDM_REG_CHK_INT_EN    BIT(19)
59 #define APPL_INTR_EN_L0_0_SYS_INTR_EN           BIT(30)
60 #define APPL_INTR_EN_L0_0_SYS_MSI_INTR_EN       BIT(31)
61
62 #define APPL_INTR_STATUS_L0                     0xC
63 #define APPL_INTR_STATUS_L0_LINK_STATE_INT      BIT(0)
64 #define APPL_INTR_STATUS_L0_INT_INT             BIT(8)
65 #define APPL_INTR_STATUS_L0_PCI_CMD_EN_INT      BIT(15)
66 #define APPL_INTR_STATUS_L0_PEX_RST_INT         BIT(16)
67 #define APPL_INTR_STATUS_L0_CDM_REG_CHK_INT     BIT(18)
68
69 #define APPL_INTR_EN_L1_0_0                             0x1C
70 #define APPL_INTR_EN_L1_0_0_LINK_REQ_RST_NOT_INT_EN     BIT(1)
71 #define APPL_INTR_EN_L1_0_0_RDLH_LINK_UP_INT_EN         BIT(3)
72 #define APPL_INTR_EN_L1_0_0_HOT_RESET_DONE_INT_EN       BIT(30)
73
74 #define APPL_INTR_STATUS_L1_0_0                         0x20
75 #define APPL_INTR_STATUS_L1_0_0_LINK_REQ_RST_NOT_CHGED  BIT(1)
76 #define APPL_INTR_STATUS_L1_0_0_RDLH_LINK_UP_CHGED      BIT(3)
77 #define APPL_INTR_STATUS_L1_0_0_HOT_RESET_DONE          BIT(30)
78
79 #define APPL_INTR_STATUS_L1_1                   0x2C
80 #define APPL_INTR_STATUS_L1_2                   0x30
81 #define APPL_INTR_STATUS_L1_3                   0x34
82 #define APPL_INTR_STATUS_L1_6                   0x3C
83 #define APPL_INTR_STATUS_L1_7                   0x40
84 #define APPL_INTR_STATUS_L1_15_CFG_BME_CHGED    BIT(1)
85
86 #define APPL_INTR_EN_L1_8_0                     0x44
87 #define APPL_INTR_EN_L1_8_BW_MGT_INT_EN         BIT(2)
88 #define APPL_INTR_EN_L1_8_AUTO_BW_INT_EN        BIT(3)
89 #define APPL_INTR_EN_L1_8_INTX_EN               BIT(11)
90 #define APPL_INTR_EN_L1_8_AER_INT_EN            BIT(15)
91
92 #define APPL_INTR_STATUS_L1_8_0                 0x4C
93 #define APPL_INTR_STATUS_L1_8_0_EDMA_INT_MASK   GENMASK(11, 6)
94 #define APPL_INTR_STATUS_L1_8_0_BW_MGT_INT_STS  BIT(2)
95 #define APPL_INTR_STATUS_L1_8_0_AUTO_BW_INT_STS BIT(3)
96
97 #define APPL_INTR_STATUS_L1_9                   0x54
98 #define APPL_INTR_STATUS_L1_10                  0x58
99 #define APPL_INTR_STATUS_L1_11                  0x64
100 #define APPL_INTR_STATUS_L1_13                  0x74
101 #define APPL_INTR_STATUS_L1_14                  0x78
102 #define APPL_INTR_STATUS_L1_15                  0x7C
103 #define APPL_INTR_STATUS_L1_17                  0x88
104
105 #define APPL_INTR_EN_L1_18                              0x90
106 #define APPL_INTR_EN_L1_18_CDM_REG_CHK_CMPLT            BIT(2)
107 #define APPL_INTR_EN_L1_18_CDM_REG_CHK_CMP_ERR          BIT(1)
108 #define APPL_INTR_EN_L1_18_CDM_REG_CHK_LOGIC_ERR        BIT(0)
109
110 #define APPL_INTR_STATUS_L1_18                          0x94
111 #define APPL_INTR_STATUS_L1_18_CDM_REG_CHK_CMPLT        BIT(2)
112 #define APPL_INTR_STATUS_L1_18_CDM_REG_CHK_CMP_ERR      BIT(1)
113 #define APPL_INTR_STATUS_L1_18_CDM_REG_CHK_LOGIC_ERR    BIT(0)
114
115 #define APPL_MSI_CTRL_1                         0xAC
116
117 #define APPL_MSI_CTRL_2                         0xB0
118
119 #define APPL_LEGACY_INTX                        0xB8
120
121 #define APPL_LTR_MSG_1                          0xC4
122 #define LTR_MSG_REQ                             BIT(15)
123 #define LTR_MST_NO_SNOOP_SHIFT                  16
124
125 #define APPL_LTR_MSG_2                          0xC8
126 #define APPL_LTR_MSG_2_LTR_MSG_REQ_STATE        BIT(3)
127
128 #define APPL_LINK_STATUS                        0xCC
129 #define APPL_LINK_STATUS_RDLH_LINK_UP           BIT(0)
130
131 #define APPL_DEBUG                              0xD0
132 #define APPL_DEBUG_PM_LINKST_IN_L2_LAT          BIT(21)
133 #define APPL_DEBUG_PM_LINKST_IN_L0              0x11
134 #define APPL_DEBUG_LTSSM_STATE_MASK             GENMASK(8, 3)
135 #define APPL_DEBUG_LTSSM_STATE_SHIFT            3
136 #define LTSSM_STATE_PRE_DETECT                  5
137
138 #define APPL_RADM_STATUS                        0xE4
139 #define APPL_PM_XMT_TURNOFF_STATE               BIT(0)
140
141 #define APPL_DM_TYPE                            0x100
142 #define APPL_DM_TYPE_MASK                       GENMASK(3, 0)
143 #define APPL_DM_TYPE_RP                         0x4
144 #define APPL_DM_TYPE_EP                         0x0
145
146 #define APPL_CFG_BASE_ADDR                      0x104
147 #define APPL_CFG_BASE_ADDR_MASK                 GENMASK(31, 12)
148
149 #define APPL_CFG_IATU_DMA_BASE_ADDR             0x108
150 #define APPL_CFG_IATU_DMA_BASE_ADDR_MASK        GENMASK(31, 18)
151
152 #define APPL_CFG_MISC                           0x110
153 #define APPL_CFG_MISC_SLV_EP_MODE               BIT(14)
154 #define APPL_CFG_MISC_ARCACHE_MASK              GENMASK(13, 10)
155 #define APPL_CFG_MISC_ARCACHE_SHIFT             10
156 #define APPL_CFG_MISC_ARCACHE_VAL               3
157
158 #define APPL_CFG_SLCG_OVERRIDE                  0x114
159 #define APPL_CFG_SLCG_OVERRIDE_SLCG_EN_MASTER   BIT(0)
160
161 #define APPL_CAR_RESET_OVRD                             0x12C
162 #define APPL_CAR_RESET_OVRD_CYA_OVERRIDE_CORE_RST_N     BIT(0)
163
164 #define IO_BASE_IO_DECODE                               BIT(0)
165 #define IO_BASE_IO_DECODE_BIT8                          BIT(8)
166
167 #define CFG_PREF_MEM_LIMIT_BASE_MEM_DECODE              BIT(0)
168 #define CFG_PREF_MEM_LIMIT_BASE_MEM_LIMIT_DECODE        BIT(16)
169
170 #define CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF 0x718
171 #define CFG_TIMER_CTRL_ACK_NAK_SHIFT    (19)
172
173 #define EVENT_COUNTER_ALL_CLEAR         0x3
174 #define EVENT_COUNTER_ENABLE_ALL        0x7
175 #define EVENT_COUNTER_ENABLE_SHIFT      2
176 #define EVENT_COUNTER_EVENT_SEL_MASK    GENMASK(7, 0)
177 #define EVENT_COUNTER_EVENT_SEL_SHIFT   16
178 #define EVENT_COUNTER_EVENT_Tx_L0S      0x2
179 #define EVENT_COUNTER_EVENT_Rx_L0S      0x3
180 #define EVENT_COUNTER_EVENT_L1          0x5
181 #define EVENT_COUNTER_EVENT_L1_1        0x7
182 #define EVENT_COUNTER_EVENT_L1_2        0x8
183 #define EVENT_COUNTER_GROUP_SEL_SHIFT   24
184 #define EVENT_COUNTER_GROUP_5           0x5
185
186 #define N_FTS_VAL                                       52
187 #define FTS_VAL                                         52
188
189 #define PORT_LOGIC_MSI_CTRL_INT_0_EN            0x828
190
191 #define GEN3_EQ_CONTROL_OFF                     0x8a8
192 #define GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_SHIFT  8
193 #define GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_MASK   GENMASK(23, 8)
194 #define GEN3_EQ_CONTROL_OFF_FB_MODE_MASK        GENMASK(3, 0)
195
196 #define GEN3_RELATED_OFF                        0x890
197 #define GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL    BIT(0)
198 #define GEN3_RELATED_OFF_GEN3_EQ_DISABLE        BIT(16)
199 #define GEN3_RELATED_OFF_RATE_SHADOW_SEL_SHIFT  24
200 #define GEN3_RELATED_OFF_RATE_SHADOW_SEL_MASK   GENMASK(25, 24)
201
202 #define PORT_LOGIC_AMBA_ERROR_RESPONSE_DEFAULT  0x8D0
203 #define AMBA_ERROR_RESPONSE_CRS_SHIFT           3
204 #define AMBA_ERROR_RESPONSE_CRS_MASK            GENMASK(1, 0)
205 #define AMBA_ERROR_RESPONSE_CRS_OKAY            0
206 #define AMBA_ERROR_RESPONSE_CRS_OKAY_FFFFFFFF   1
207 #define AMBA_ERROR_RESPONSE_CRS_OKAY_FFFF0001   2
208
209 #define MSIX_ADDR_MATCH_LOW_OFF                 0x940
210 #define MSIX_ADDR_MATCH_LOW_OFF_EN              BIT(0)
211 #define MSIX_ADDR_MATCH_LOW_OFF_MASK            GENMASK(31, 2)
212
213 #define MSIX_ADDR_MATCH_HIGH_OFF                0x944
214 #define MSIX_ADDR_MATCH_HIGH_OFF_MASK           GENMASK(31, 0)
215
216 #define PORT_LOGIC_MSIX_DOORBELL                        0x948
217
218 #define CAP_SPCIE_CAP_OFF                       0x154
219 #define CAP_SPCIE_CAP_OFF_DSP_TX_PRESET0_MASK   GENMASK(3, 0)
220 #define CAP_SPCIE_CAP_OFF_USP_TX_PRESET0_MASK   GENMASK(11, 8)
221 #define CAP_SPCIE_CAP_OFF_USP_TX_PRESET0_SHIFT  8
222
223 #define PME_ACK_TIMEOUT 10000
224
225 #define LTSSM_TIMEOUT 50000     /* 50ms */
226
227 #define GEN3_GEN4_EQ_PRESET_INIT        5
228
229 #define GEN1_CORE_CLK_FREQ      62500000
230 #define GEN2_CORE_CLK_FREQ      125000000
231 #define GEN3_CORE_CLK_FREQ      250000000
232 #define GEN4_CORE_CLK_FREQ      500000000
233
234 #define LTR_MSG_TIMEOUT         (100 * 1000)
235
236 #define PERST_DEBOUNCE_TIME     (5 * 1000)
237
238 #define EP_STATE_DISABLED       0
239 #define EP_STATE_ENABLED        1
240
241 static const unsigned int pcie_gen_freq[] = {
242         GEN1_CORE_CLK_FREQ,
243         GEN2_CORE_CLK_FREQ,
244         GEN3_CORE_CLK_FREQ,
245         GEN4_CORE_CLK_FREQ
246 };
247
248 static const u32 event_cntr_ctrl_offset[] = {
249         0x1d8,
250         0x1a8,
251         0x1a8,
252         0x1a8,
253         0x1c4,
254         0x1d8
255 };
256
257 static const u32 event_cntr_data_offset[] = {
258         0x1dc,
259         0x1ac,
260         0x1ac,
261         0x1ac,
262         0x1c8,
263         0x1dc
264 };
265
266 struct tegra_pcie_dw {
267         struct device *dev;
268         struct resource *appl_res;
269         struct resource *dbi_res;
270         struct resource *atu_dma_res;
271         void __iomem *appl_base;
272         struct clk *core_clk;
273         struct reset_control *core_apb_rst;
274         struct reset_control *core_rst;
275         struct dw_pcie pci;
276         struct tegra_bpmp *bpmp;
277
278         enum dw_pcie_device_mode mode;
279
280         bool supports_clkreq;
281         bool enable_cdm_check;
282         bool link_state;
283         bool update_fc_fixup;
284         u8 init_link_width;
285         u32 msi_ctrl_int;
286         u32 num_lanes;
287         u32 cid;
288         u32 cfg_link_cap_l1sub;
289         u32 pcie_cap_base;
290         u32 aspm_cmrt;
291         u32 aspm_pwr_on_t;
292         u32 aspm_l0s_enter_lat;
293
294         struct regulator *pex_ctl_supply;
295         struct regulator *slot_ctl_3v3;
296         struct regulator *slot_ctl_12v;
297
298         unsigned int phy_count;
299         struct phy **phys;
300
301         struct dentry *debugfs;
302
303         /* Endpoint mode specific */
304         struct gpio_desc *pex_rst_gpiod;
305         struct gpio_desc *pex_refclk_sel_gpiod;
306         unsigned int pex_rst_irq;
307         int ep_state;
308 };
309
310 struct tegra_pcie_dw_of_data {
311         enum dw_pcie_device_mode mode;
312 };
313
314 static inline struct tegra_pcie_dw *to_tegra_pcie(struct dw_pcie *pci)
315 {
316         return container_of(pci, struct tegra_pcie_dw, pci);
317 }
318
319 static inline void appl_writel(struct tegra_pcie_dw *pcie, const u32 value,
320                                const u32 reg)
321 {
322         writel_relaxed(value, pcie->appl_base + reg);
323 }
324
325 static inline u32 appl_readl(struct tegra_pcie_dw *pcie, const u32 reg)
326 {
327         return readl_relaxed(pcie->appl_base + reg);
328 }
329
330 struct tegra_pcie_soc {
331         enum dw_pcie_device_mode mode;
332 };
333
334 static void apply_bad_link_workaround(struct pcie_port *pp)
335 {
336         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
337         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
338         u32 current_link_width;
339         u16 val;
340
341         /*
342          * NOTE:- Since this scenario is uncommon and link as such is not
343          * stable anyway, not waiting to confirm if link is really
344          * transitioning to Gen-2 speed
345          */
346         val = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA);
347         if (val & PCI_EXP_LNKSTA_LBMS) {
348                 current_link_width = (val & PCI_EXP_LNKSTA_NLW) >>
349                                      PCI_EXP_LNKSTA_NLW_SHIFT;
350                 if (pcie->init_link_width > current_link_width) {
351                         dev_warn(pci->dev, "PCIe link is bad, width reduced\n");
352                         val = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base +
353                                                 PCI_EXP_LNKCTL2);
354                         val &= ~PCI_EXP_LNKCTL2_TLS;
355                         val |= PCI_EXP_LNKCTL2_TLS_2_5GT;
356                         dw_pcie_writew_dbi(pci, pcie->pcie_cap_base +
357                                            PCI_EXP_LNKCTL2, val);
358
359                         val = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base +
360                                                 PCI_EXP_LNKCTL);
361                         val |= PCI_EXP_LNKCTL_RL;
362                         dw_pcie_writew_dbi(pci, pcie->pcie_cap_base +
363                                            PCI_EXP_LNKCTL, val);
364                 }
365         }
366 }
367
368 static irqreturn_t tegra_pcie_rp_irq_handler(int irq, void *arg)
369 {
370         struct tegra_pcie_dw *pcie = arg;
371         struct dw_pcie *pci = &pcie->pci;
372         struct pcie_port *pp = &pci->pp;
373         u32 val, status_l0, status_l1;
374         u16 val_w;
375
376         status_l0 = appl_readl(pcie, APPL_INTR_STATUS_L0);
377         if (status_l0 & APPL_INTR_STATUS_L0_LINK_STATE_INT) {
378                 status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_0_0);
379                 appl_writel(pcie, status_l1, APPL_INTR_STATUS_L1_0_0);
380                 if (status_l1 & APPL_INTR_STATUS_L1_0_0_LINK_REQ_RST_NOT_CHGED) {
381                         /* SBR & Surprise Link Down WAR */
382                         val = appl_readl(pcie, APPL_CAR_RESET_OVRD);
383                         val &= ~APPL_CAR_RESET_OVRD_CYA_OVERRIDE_CORE_RST_N;
384                         appl_writel(pcie, val, APPL_CAR_RESET_OVRD);
385                         udelay(1);
386                         val = appl_readl(pcie, APPL_CAR_RESET_OVRD);
387                         val |= APPL_CAR_RESET_OVRD_CYA_OVERRIDE_CORE_RST_N;
388                         appl_writel(pcie, val, APPL_CAR_RESET_OVRD);
389
390                         val = dw_pcie_readl_dbi(pci, PCIE_LINK_WIDTH_SPEED_CONTROL);
391                         val |= PORT_LOGIC_SPEED_CHANGE;
392                         dw_pcie_writel_dbi(pci, PCIE_LINK_WIDTH_SPEED_CONTROL, val);
393                 }
394         }
395
396         if (status_l0 & APPL_INTR_STATUS_L0_INT_INT) {
397                 status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_8_0);
398                 if (status_l1 & APPL_INTR_STATUS_L1_8_0_AUTO_BW_INT_STS) {
399                         appl_writel(pcie,
400                                     APPL_INTR_STATUS_L1_8_0_AUTO_BW_INT_STS,
401                                     APPL_INTR_STATUS_L1_8_0);
402                         apply_bad_link_workaround(pp);
403                 }
404                 if (status_l1 & APPL_INTR_STATUS_L1_8_0_BW_MGT_INT_STS) {
405                         appl_writel(pcie,
406                                     APPL_INTR_STATUS_L1_8_0_BW_MGT_INT_STS,
407                                     APPL_INTR_STATUS_L1_8_0);
408
409                         val_w = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base +
410                                                   PCI_EXP_LNKSTA);
411                         dev_dbg(pci->dev, "Link Speed : Gen-%u\n", val_w &
412                                 PCI_EXP_LNKSTA_CLS);
413                 }
414         }
415
416         if (status_l0 & APPL_INTR_STATUS_L0_CDM_REG_CHK_INT) {
417                 status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_18);
418                 val = dw_pcie_readl_dbi(pci, PCIE_PL_CHK_REG_CONTROL_STATUS);
419                 if (status_l1 & APPL_INTR_STATUS_L1_18_CDM_REG_CHK_CMPLT) {
420                         dev_info(pci->dev, "CDM check complete\n");
421                         val |= PCIE_PL_CHK_REG_CHK_REG_COMPLETE;
422                 }
423                 if (status_l1 & APPL_INTR_STATUS_L1_18_CDM_REG_CHK_CMP_ERR) {
424                         dev_err(pci->dev, "CDM comparison mismatch\n");
425                         val |= PCIE_PL_CHK_REG_CHK_REG_COMPARISON_ERROR;
426                 }
427                 if (status_l1 & APPL_INTR_STATUS_L1_18_CDM_REG_CHK_LOGIC_ERR) {
428                         dev_err(pci->dev, "CDM Logic error\n");
429                         val |= PCIE_PL_CHK_REG_CHK_REG_LOGIC_ERROR;
430                 }
431                 dw_pcie_writel_dbi(pci, PCIE_PL_CHK_REG_CONTROL_STATUS, val);
432                 val = dw_pcie_readl_dbi(pci, PCIE_PL_CHK_REG_ERR_ADDR);
433                 dev_err(pci->dev, "CDM Error Address Offset = 0x%08X\n", val);
434         }
435
436         return IRQ_HANDLED;
437 }
438
439 static void pex_ep_event_hot_rst_done(struct tegra_pcie_dw *pcie)
440 {
441         u32 val;
442
443         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L0);
444         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_0_0);
445         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_1);
446         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_2);
447         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_3);
448         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_6);
449         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_7);
450         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_8_0);
451         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_9);
452         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_10);
453         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_11);
454         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_13);
455         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_14);
456         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_15);
457         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_17);
458         appl_writel(pcie, 0xFFFFFFFF, APPL_MSI_CTRL_2);
459
460         val = appl_readl(pcie, APPL_CTRL);
461         val |= APPL_CTRL_LTSSM_EN;
462         appl_writel(pcie, val, APPL_CTRL);
463 }
464
465 static irqreturn_t tegra_pcie_ep_irq_thread(int irq, void *arg)
466 {
467         struct tegra_pcie_dw *pcie = arg;
468         struct dw_pcie *pci = &pcie->pci;
469         u32 val, speed;
470
471         speed = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA) &
472                 PCI_EXP_LNKSTA_CLS;
473         clk_set_rate(pcie->core_clk, pcie_gen_freq[speed - 1]);
474
475         /* If EP doesn't advertise L1SS, just return */
476         val = dw_pcie_readl_dbi(pci, pcie->cfg_link_cap_l1sub);
477         if (!(val & (PCI_L1SS_CAP_ASPM_L1_1 | PCI_L1SS_CAP_ASPM_L1_2)))
478                 return IRQ_HANDLED;
479
480         /* Check if BME is set to '1' */
481         val = dw_pcie_readl_dbi(pci, PCI_COMMAND);
482         if (val & PCI_COMMAND_MASTER) {
483                 ktime_t timeout;
484
485                 /* 110us for both snoop and no-snoop */
486                 val = 110 | (2 << PCI_LTR_SCALE_SHIFT) | LTR_MSG_REQ;
487                 val |= (val << LTR_MST_NO_SNOOP_SHIFT);
488                 appl_writel(pcie, val, APPL_LTR_MSG_1);
489
490                 /* Send LTR upstream */
491                 val = appl_readl(pcie, APPL_LTR_MSG_2);
492                 val |= APPL_LTR_MSG_2_LTR_MSG_REQ_STATE;
493                 appl_writel(pcie, val, APPL_LTR_MSG_2);
494
495                 timeout = ktime_add_us(ktime_get(), LTR_MSG_TIMEOUT);
496                 for (;;) {
497                         val = appl_readl(pcie, APPL_LTR_MSG_2);
498                         if (!(val & APPL_LTR_MSG_2_LTR_MSG_REQ_STATE))
499                                 break;
500                         if (ktime_after(ktime_get(), timeout))
501                                 break;
502                         usleep_range(1000, 1100);
503                 }
504                 if (val & APPL_LTR_MSG_2_LTR_MSG_REQ_STATE)
505                         dev_err(pcie->dev, "Failed to send LTR message\n");
506         }
507
508         return IRQ_HANDLED;
509 }
510
511 static irqreturn_t tegra_pcie_ep_hard_irq(int irq, void *arg)
512 {
513         struct tegra_pcie_dw *pcie = arg;
514         struct dw_pcie_ep *ep = &pcie->pci.ep;
515         int spurious = 1;
516         u32 status_l0, status_l1, link_status;
517
518         status_l0 = appl_readl(pcie, APPL_INTR_STATUS_L0);
519         if (status_l0 & APPL_INTR_STATUS_L0_LINK_STATE_INT) {
520                 status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_0_0);
521                 appl_writel(pcie, status_l1, APPL_INTR_STATUS_L1_0_0);
522
523                 if (status_l1 & APPL_INTR_STATUS_L1_0_0_HOT_RESET_DONE)
524                         pex_ep_event_hot_rst_done(pcie);
525
526                 if (status_l1 & APPL_INTR_STATUS_L1_0_0_RDLH_LINK_UP_CHGED) {
527                         link_status = appl_readl(pcie, APPL_LINK_STATUS);
528                         if (link_status & APPL_LINK_STATUS_RDLH_LINK_UP) {
529                                 dev_dbg(pcie->dev, "Link is up with Host\n");
530                                 dw_pcie_ep_linkup(ep);
531                         }
532                 }
533
534                 spurious = 0;
535         }
536
537         if (status_l0 & APPL_INTR_STATUS_L0_PCI_CMD_EN_INT) {
538                 status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_15);
539                 appl_writel(pcie, status_l1, APPL_INTR_STATUS_L1_15);
540
541                 if (status_l1 & APPL_INTR_STATUS_L1_15_CFG_BME_CHGED)
542                         return IRQ_WAKE_THREAD;
543
544                 spurious = 0;
545         }
546
547         if (spurious) {
548                 dev_warn(pcie->dev, "Random interrupt (STATUS = 0x%08X)\n",
549                          status_l0);
550                 appl_writel(pcie, status_l0, APPL_INTR_STATUS_L0);
551         }
552
553         return IRQ_HANDLED;
554 }
555
556 static int tegra_pcie_dw_rd_own_conf(struct pci_bus *bus, u32 devfn, int where,
557                                      int size, u32 *val)
558 {
559         /*
560          * This is an endpoint mode specific register happen to appear even
561          * when controller is operating in root port mode and system hangs
562          * when it is accessed with link being in ASPM-L1 state.
563          * So skip accessing it altogether
564          */
565         if (!PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL) {
566                 *val = 0x00000000;
567                 return PCIBIOS_SUCCESSFUL;
568         }
569
570         return pci_generic_config_read(bus, devfn, where, size, val);
571 }
572
573 static int tegra_pcie_dw_wr_own_conf(struct pci_bus *bus, u32 devfn, int where,
574                                      int size, u32 val)
575 {
576         /*
577          * This is an endpoint mode specific register happen to appear even
578          * when controller is operating in root port mode and system hangs
579          * when it is accessed with link being in ASPM-L1 state.
580          * So skip accessing it altogether
581          */
582         if (!PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL)
583                 return PCIBIOS_SUCCESSFUL;
584
585         return pci_generic_config_write(bus, devfn, where, size, val);
586 }
587
588 static struct pci_ops tegra_pci_ops = {
589         .map_bus = dw_pcie_own_conf_map_bus,
590         .read = tegra_pcie_dw_rd_own_conf,
591         .write = tegra_pcie_dw_wr_own_conf,
592 };
593
594 #if defined(CONFIG_PCIEASPM)
595 static void disable_aspm_l11(struct tegra_pcie_dw *pcie)
596 {
597         u32 val;
598
599         val = dw_pcie_readl_dbi(&pcie->pci, pcie->cfg_link_cap_l1sub);
600         val &= ~PCI_L1SS_CAP_ASPM_L1_1;
601         dw_pcie_writel_dbi(&pcie->pci, pcie->cfg_link_cap_l1sub, val);
602 }
603
604 static void disable_aspm_l12(struct tegra_pcie_dw *pcie)
605 {
606         u32 val;
607
608         val = dw_pcie_readl_dbi(&pcie->pci, pcie->cfg_link_cap_l1sub);
609         val &= ~PCI_L1SS_CAP_ASPM_L1_2;
610         dw_pcie_writel_dbi(&pcie->pci, pcie->cfg_link_cap_l1sub, val);
611 }
612
613 static inline u32 event_counter_prog(struct tegra_pcie_dw *pcie, u32 event)
614 {
615         u32 val;
616
617         val = dw_pcie_readl_dbi(&pcie->pci, event_cntr_ctrl_offset[pcie->cid]);
618         val &= ~(EVENT_COUNTER_EVENT_SEL_MASK << EVENT_COUNTER_EVENT_SEL_SHIFT);
619         val |= EVENT_COUNTER_GROUP_5 << EVENT_COUNTER_GROUP_SEL_SHIFT;
620         val |= event << EVENT_COUNTER_EVENT_SEL_SHIFT;
621         val |= EVENT_COUNTER_ENABLE_ALL << EVENT_COUNTER_ENABLE_SHIFT;
622         dw_pcie_writel_dbi(&pcie->pci, event_cntr_ctrl_offset[pcie->cid], val);
623         val = dw_pcie_readl_dbi(&pcie->pci, event_cntr_data_offset[pcie->cid]);
624
625         return val;
626 }
627
628 static int aspm_state_cnt(struct seq_file *s, void *data)
629 {
630         struct tegra_pcie_dw *pcie = (struct tegra_pcie_dw *)
631                                      dev_get_drvdata(s->private);
632         u32 val;
633
634         seq_printf(s, "Tx L0s entry count : %u\n",
635                    event_counter_prog(pcie, EVENT_COUNTER_EVENT_Tx_L0S));
636
637         seq_printf(s, "Rx L0s entry count : %u\n",
638                    event_counter_prog(pcie, EVENT_COUNTER_EVENT_Rx_L0S));
639
640         seq_printf(s, "Link L1 entry count : %u\n",
641                    event_counter_prog(pcie, EVENT_COUNTER_EVENT_L1));
642
643         seq_printf(s, "Link L1.1 entry count : %u\n",
644                    event_counter_prog(pcie, EVENT_COUNTER_EVENT_L1_1));
645
646         seq_printf(s, "Link L1.2 entry count : %u\n",
647                    event_counter_prog(pcie, EVENT_COUNTER_EVENT_L1_2));
648
649         /* Clear all counters */
650         dw_pcie_writel_dbi(&pcie->pci, event_cntr_ctrl_offset[pcie->cid],
651                            EVENT_COUNTER_ALL_CLEAR);
652
653         /* Re-enable counting */
654         val = EVENT_COUNTER_ENABLE_ALL << EVENT_COUNTER_ENABLE_SHIFT;
655         val |= EVENT_COUNTER_GROUP_5 << EVENT_COUNTER_GROUP_SEL_SHIFT;
656         dw_pcie_writel_dbi(&pcie->pci, event_cntr_ctrl_offset[pcie->cid], val);
657
658         return 0;
659 }
660
661 static void init_host_aspm(struct tegra_pcie_dw *pcie)
662 {
663         struct dw_pcie *pci = &pcie->pci;
664         u32 val;
665
666         val = dw_pcie_find_ext_capability(pci, PCI_EXT_CAP_ID_L1SS);
667         pcie->cfg_link_cap_l1sub = val + PCI_L1SS_CAP;
668
669         /* Enable ASPM counters */
670         val = EVENT_COUNTER_ENABLE_ALL << EVENT_COUNTER_ENABLE_SHIFT;
671         val |= EVENT_COUNTER_GROUP_5 << EVENT_COUNTER_GROUP_SEL_SHIFT;
672         dw_pcie_writel_dbi(pci, event_cntr_ctrl_offset[pcie->cid], val);
673
674         /* Program T_cmrt and T_pwr_on values */
675         val = dw_pcie_readl_dbi(pci, pcie->cfg_link_cap_l1sub);
676         val &= ~(PCI_L1SS_CAP_CM_RESTORE_TIME | PCI_L1SS_CAP_P_PWR_ON_VALUE);
677         val |= (pcie->aspm_cmrt << 8);
678         val |= (pcie->aspm_pwr_on_t << 19);
679         dw_pcie_writel_dbi(pci, pcie->cfg_link_cap_l1sub, val);
680
681         /* Program L0s and L1 entrance latencies */
682         val = dw_pcie_readl_dbi(pci, PCIE_PORT_AFR);
683         val &= ~PORT_AFR_L0S_ENTRANCE_LAT_MASK;
684         val |= (pcie->aspm_l0s_enter_lat << PORT_AFR_L0S_ENTRANCE_LAT_SHIFT);
685         val |= PORT_AFR_ENTER_ASPM;
686         dw_pcie_writel_dbi(pci, PCIE_PORT_AFR, val);
687 }
688
689 static void init_debugfs(struct tegra_pcie_dw *pcie)
690 {
691         debugfs_create_devm_seqfile(pcie->dev, "aspm_state_cnt", pcie->debugfs,
692                                     aspm_state_cnt);
693 }
694 #else
695 static inline void disable_aspm_l12(struct tegra_pcie_dw *pcie) { return; }
696 static inline void disable_aspm_l11(struct tegra_pcie_dw *pcie) { return; }
697 static inline void init_host_aspm(struct tegra_pcie_dw *pcie) { return; }
698 static inline void init_debugfs(struct tegra_pcie_dw *pcie) { return; }
699 #endif
700
701 static void tegra_pcie_enable_system_interrupts(struct pcie_port *pp)
702 {
703         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
704         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
705         u32 val;
706         u16 val_w;
707
708         val = appl_readl(pcie, APPL_INTR_EN_L0_0);
709         val |= APPL_INTR_EN_L0_0_LINK_STATE_INT_EN;
710         appl_writel(pcie, val, APPL_INTR_EN_L0_0);
711
712         val = appl_readl(pcie, APPL_INTR_EN_L1_0_0);
713         val |= APPL_INTR_EN_L1_0_0_LINK_REQ_RST_NOT_INT_EN;
714         appl_writel(pcie, val, APPL_INTR_EN_L1_0_0);
715
716         if (pcie->enable_cdm_check) {
717                 val = appl_readl(pcie, APPL_INTR_EN_L0_0);
718                 val |= APPL_INTR_EN_L0_0_CDM_REG_CHK_INT_EN;
719                 appl_writel(pcie, val, APPL_INTR_EN_L0_0);
720
721                 val = appl_readl(pcie, APPL_INTR_EN_L1_18);
722                 val |= APPL_INTR_EN_L1_18_CDM_REG_CHK_CMP_ERR;
723                 val |= APPL_INTR_EN_L1_18_CDM_REG_CHK_LOGIC_ERR;
724                 appl_writel(pcie, val, APPL_INTR_EN_L1_18);
725         }
726
727         val_w = dw_pcie_readw_dbi(&pcie->pci, pcie->pcie_cap_base +
728                                   PCI_EXP_LNKSTA);
729         pcie->init_link_width = (val_w & PCI_EXP_LNKSTA_NLW) >>
730                                 PCI_EXP_LNKSTA_NLW_SHIFT;
731
732         val_w = dw_pcie_readw_dbi(&pcie->pci, pcie->pcie_cap_base +
733                                   PCI_EXP_LNKCTL);
734         val_w |= PCI_EXP_LNKCTL_LBMIE;
735         dw_pcie_writew_dbi(&pcie->pci, pcie->pcie_cap_base + PCI_EXP_LNKCTL,
736                            val_w);
737 }
738
739 static void tegra_pcie_enable_legacy_interrupts(struct pcie_port *pp)
740 {
741         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
742         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
743         u32 val;
744
745         /* Enable legacy interrupt generation */
746         val = appl_readl(pcie, APPL_INTR_EN_L0_0);
747         val |= APPL_INTR_EN_L0_0_SYS_INTR_EN;
748         val |= APPL_INTR_EN_L0_0_INT_INT_EN;
749         appl_writel(pcie, val, APPL_INTR_EN_L0_0);
750
751         val = appl_readl(pcie, APPL_INTR_EN_L1_8_0);
752         val |= APPL_INTR_EN_L1_8_INTX_EN;
753         val |= APPL_INTR_EN_L1_8_AUTO_BW_INT_EN;
754         val |= APPL_INTR_EN_L1_8_BW_MGT_INT_EN;
755         if (IS_ENABLED(CONFIG_PCIEAER))
756                 val |= APPL_INTR_EN_L1_8_AER_INT_EN;
757         appl_writel(pcie, val, APPL_INTR_EN_L1_8_0);
758 }
759
760 static void tegra_pcie_enable_msi_interrupts(struct pcie_port *pp)
761 {
762         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
763         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
764         u32 val;
765
766         dw_pcie_msi_init(pp);
767
768         /* Enable MSI interrupt generation */
769         val = appl_readl(pcie, APPL_INTR_EN_L0_0);
770         val |= APPL_INTR_EN_L0_0_SYS_MSI_INTR_EN;
771         val |= APPL_INTR_EN_L0_0_MSI_RCV_INT_EN;
772         appl_writel(pcie, val, APPL_INTR_EN_L0_0);
773 }
774
775 static void tegra_pcie_enable_interrupts(struct pcie_port *pp)
776 {
777         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
778         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
779
780         /* Clear interrupt statuses before enabling interrupts */
781         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L0);
782         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_0_0);
783         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_1);
784         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_2);
785         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_3);
786         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_6);
787         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_7);
788         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_8_0);
789         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_9);
790         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_10);
791         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_11);
792         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_13);
793         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_14);
794         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_15);
795         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_17);
796
797         tegra_pcie_enable_system_interrupts(pp);
798         tegra_pcie_enable_legacy_interrupts(pp);
799         if (IS_ENABLED(CONFIG_PCI_MSI))
800                 tegra_pcie_enable_msi_interrupts(pp);
801 }
802
803 static void config_gen3_gen4_eq_presets(struct tegra_pcie_dw *pcie)
804 {
805         struct dw_pcie *pci = &pcie->pci;
806         u32 val, offset, i;
807
808         /* Program init preset */
809         for (i = 0; i < pcie->num_lanes; i++) {
810                 val = dw_pcie_readw_dbi(pci, CAP_SPCIE_CAP_OFF + (i * 2));
811                 val &= ~CAP_SPCIE_CAP_OFF_DSP_TX_PRESET0_MASK;
812                 val |= GEN3_GEN4_EQ_PRESET_INIT;
813                 val &= ~CAP_SPCIE_CAP_OFF_USP_TX_PRESET0_MASK;
814                 val |= (GEN3_GEN4_EQ_PRESET_INIT <<
815                            CAP_SPCIE_CAP_OFF_USP_TX_PRESET0_SHIFT);
816                 dw_pcie_writew_dbi(pci, CAP_SPCIE_CAP_OFF + (i * 2), val);
817
818                 offset = dw_pcie_find_ext_capability(pci,
819                                                      PCI_EXT_CAP_ID_PL_16GT) +
820                                 PCI_PL_16GT_LE_CTRL;
821                 val = dw_pcie_readb_dbi(pci, offset + i);
822                 val &= ~PCI_PL_16GT_LE_CTRL_DSP_TX_PRESET_MASK;
823                 val |= GEN3_GEN4_EQ_PRESET_INIT;
824                 val &= ~PCI_PL_16GT_LE_CTRL_USP_TX_PRESET_MASK;
825                 val |= (GEN3_GEN4_EQ_PRESET_INIT <<
826                         PCI_PL_16GT_LE_CTRL_USP_TX_PRESET_SHIFT);
827                 dw_pcie_writeb_dbi(pci, offset + i, val);
828         }
829
830         val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
831         val &= ~GEN3_RELATED_OFF_RATE_SHADOW_SEL_MASK;
832         dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
833
834         val = dw_pcie_readl_dbi(pci, GEN3_EQ_CONTROL_OFF);
835         val &= ~GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_MASK;
836         val |= (0x3ff << GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_SHIFT);
837         val &= ~GEN3_EQ_CONTROL_OFF_FB_MODE_MASK;
838         dw_pcie_writel_dbi(pci, GEN3_EQ_CONTROL_OFF, val);
839
840         val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
841         val &= ~GEN3_RELATED_OFF_RATE_SHADOW_SEL_MASK;
842         val |= (0x1 << GEN3_RELATED_OFF_RATE_SHADOW_SEL_SHIFT);
843         dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
844
845         val = dw_pcie_readl_dbi(pci, GEN3_EQ_CONTROL_OFF);
846         val &= ~GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_MASK;
847         val |= (0x360 << GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_SHIFT);
848         val &= ~GEN3_EQ_CONTROL_OFF_FB_MODE_MASK;
849         dw_pcie_writel_dbi(pci, GEN3_EQ_CONTROL_OFF, val);
850
851         val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
852         val &= ~GEN3_RELATED_OFF_RATE_SHADOW_SEL_MASK;
853         dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
854 }
855
856 static void tegra_pcie_prepare_host(struct pcie_port *pp)
857 {
858         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
859         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
860         u32 val;
861
862         val = dw_pcie_readl_dbi(pci, PCI_IO_BASE);
863         val &= ~(IO_BASE_IO_DECODE | IO_BASE_IO_DECODE_BIT8);
864         dw_pcie_writel_dbi(pci, PCI_IO_BASE, val);
865
866         val = dw_pcie_readl_dbi(pci, PCI_PREF_MEMORY_BASE);
867         val |= CFG_PREF_MEM_LIMIT_BASE_MEM_DECODE;
868         val |= CFG_PREF_MEM_LIMIT_BASE_MEM_LIMIT_DECODE;
869         dw_pcie_writel_dbi(pci, PCI_PREF_MEMORY_BASE, val);
870
871         dw_pcie_writel_dbi(pci, PCI_BASE_ADDRESS_0, 0);
872
873         /* Enable as 0xFFFF0001 response for CRS */
874         val = dw_pcie_readl_dbi(pci, PORT_LOGIC_AMBA_ERROR_RESPONSE_DEFAULT);
875         val &= ~(AMBA_ERROR_RESPONSE_CRS_MASK << AMBA_ERROR_RESPONSE_CRS_SHIFT);
876         val |= (AMBA_ERROR_RESPONSE_CRS_OKAY_FFFF0001 <<
877                 AMBA_ERROR_RESPONSE_CRS_SHIFT);
878         dw_pcie_writel_dbi(pci, PORT_LOGIC_AMBA_ERROR_RESPONSE_DEFAULT, val);
879
880         /* Configure Max lane width from DT */
881         val = dw_pcie_readl_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKCAP);
882         val &= ~PCI_EXP_LNKCAP_MLW;
883         val |= (pcie->num_lanes << PCI_EXP_LNKSTA_NLW_SHIFT);
884         dw_pcie_writel_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKCAP, val);
885
886         config_gen3_gen4_eq_presets(pcie);
887
888         init_host_aspm(pcie);
889
890         val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
891         val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
892         dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
893
894         if (pcie->update_fc_fixup) {
895                 val = dw_pcie_readl_dbi(pci, CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF);
896                 val |= 0x1 << CFG_TIMER_CTRL_ACK_NAK_SHIFT;
897                 dw_pcie_writel_dbi(pci, CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF, val);
898         }
899
900         dw_pcie_setup_rc(pp);
901
902         clk_set_rate(pcie->core_clk, GEN4_CORE_CLK_FREQ);
903
904         /* Assert RST */
905         val = appl_readl(pcie, APPL_PINMUX);
906         val &= ~APPL_PINMUX_PEX_RST;
907         appl_writel(pcie, val, APPL_PINMUX);
908
909         usleep_range(100, 200);
910
911         /* Enable LTSSM */
912         val = appl_readl(pcie, APPL_CTRL);
913         val |= APPL_CTRL_LTSSM_EN;
914         appl_writel(pcie, val, APPL_CTRL);
915
916         /* De-assert RST */
917         val = appl_readl(pcie, APPL_PINMUX);
918         val |= APPL_PINMUX_PEX_RST;
919         appl_writel(pcie, val, APPL_PINMUX);
920
921         msleep(100);
922 }
923
924 static int tegra_pcie_dw_host_init(struct pcie_port *pp)
925 {
926         struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
927         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
928         u32 val, tmp, offset, speed;
929
930         pp->bridge->ops = &tegra_pci_ops;
931
932         tegra_pcie_prepare_host(pp);
933
934         if (dw_pcie_wait_for_link(pci)) {
935                 /*
936                  * There are some endpoints which can't get the link up if
937                  * root port has Data Link Feature (DLF) enabled.
938                  * Refer Spec rev 4.0 ver 1.0 sec 3.4.2 & 7.7.4 for more info
939                  * on Scaled Flow Control and DLF.
940                  * So, need to confirm that is indeed the case here and attempt
941                  * link up once again with DLF disabled.
942                  */
943                 val = appl_readl(pcie, APPL_DEBUG);
944                 val &= APPL_DEBUG_LTSSM_STATE_MASK;
945                 val >>= APPL_DEBUG_LTSSM_STATE_SHIFT;
946                 tmp = appl_readl(pcie, APPL_LINK_STATUS);
947                 tmp &= APPL_LINK_STATUS_RDLH_LINK_UP;
948                 if (!(val == 0x11 && !tmp)) {
949                         /* Link is down for all good reasons */
950                         return 0;
951                 }
952
953                 dev_info(pci->dev, "Link is down in DLL");
954                 dev_info(pci->dev, "Trying again with DLFE disabled\n");
955                 /* Disable LTSSM */
956                 val = appl_readl(pcie, APPL_CTRL);
957                 val &= ~APPL_CTRL_LTSSM_EN;
958                 appl_writel(pcie, val, APPL_CTRL);
959
960                 reset_control_assert(pcie->core_rst);
961                 reset_control_deassert(pcie->core_rst);
962
963                 offset = dw_pcie_find_ext_capability(pci, PCI_EXT_CAP_ID_DLF);
964                 val = dw_pcie_readl_dbi(pci, offset + PCI_DLF_CAP);
965                 val &= ~PCI_DLF_EXCHANGE_ENABLE;
966                 dw_pcie_writel_dbi(pci, offset + PCI_DLF_CAP, val);
967
968                 tegra_pcie_prepare_host(pp);
969
970                 if (dw_pcie_wait_for_link(pci))
971                         return 0;
972         }
973
974         speed = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA) &
975                 PCI_EXP_LNKSTA_CLS;
976         clk_set_rate(pcie->core_clk, pcie_gen_freq[speed - 1]);
977
978         tegra_pcie_enable_interrupts(pp);
979
980         return 0;
981 }
982
983 static int tegra_pcie_dw_link_up(struct dw_pcie *pci)
984 {
985         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
986         u32 val = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA);
987
988         return !!(val & PCI_EXP_LNKSTA_DLLLA);
989 }
990
991 static void tegra_pcie_set_msi_vec_num(struct pcie_port *pp)
992 {
993         pp->num_vectors = MAX_MSI_IRQS;
994 }
995
996 static int tegra_pcie_dw_start_link(struct dw_pcie *pci)
997 {
998         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
999
1000         enable_irq(pcie->pex_rst_irq);
1001
1002         return 0;
1003 }
1004
1005 static void tegra_pcie_dw_stop_link(struct dw_pcie *pci)
1006 {
1007         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
1008
1009         disable_irq(pcie->pex_rst_irq);
1010 }
1011
1012 static const struct dw_pcie_ops tegra_dw_pcie_ops = {
1013         .link_up = tegra_pcie_dw_link_up,
1014         .start_link = tegra_pcie_dw_start_link,
1015         .stop_link = tegra_pcie_dw_stop_link,
1016 };
1017
1018 static struct dw_pcie_host_ops tegra_pcie_dw_host_ops = {
1019         .host_init = tegra_pcie_dw_host_init,
1020         .set_num_vectors = tegra_pcie_set_msi_vec_num,
1021 };
1022
1023 static void tegra_pcie_disable_phy(struct tegra_pcie_dw *pcie)
1024 {
1025         unsigned int phy_count = pcie->phy_count;
1026
1027         while (phy_count--) {
1028                 phy_power_off(pcie->phys[phy_count]);
1029                 phy_exit(pcie->phys[phy_count]);
1030         }
1031 }
1032
1033 static int tegra_pcie_enable_phy(struct tegra_pcie_dw *pcie)
1034 {
1035         unsigned int i;
1036         int ret;
1037
1038         for (i = 0; i < pcie->phy_count; i++) {
1039                 ret = phy_init(pcie->phys[i]);
1040                 if (ret < 0)
1041                         goto phy_power_off;
1042
1043                 ret = phy_power_on(pcie->phys[i]);
1044                 if (ret < 0)
1045                         goto phy_exit;
1046         }
1047
1048         return 0;
1049
1050 phy_power_off:
1051         while (i--) {
1052                 phy_power_off(pcie->phys[i]);
1053 phy_exit:
1054                 phy_exit(pcie->phys[i]);
1055         }
1056
1057         return ret;
1058 }
1059
1060 static int tegra_pcie_dw_parse_dt(struct tegra_pcie_dw *pcie)
1061 {
1062         struct device_node *np = pcie->dev->of_node;
1063         int ret;
1064
1065         ret = of_property_read_u32(np, "nvidia,aspm-cmrt-us", &pcie->aspm_cmrt);
1066         if (ret < 0) {
1067                 dev_info(pcie->dev, "Failed to read ASPM T_cmrt: %d\n", ret);
1068                 return ret;
1069         }
1070
1071         ret = of_property_read_u32(np, "nvidia,aspm-pwr-on-t-us",
1072                                    &pcie->aspm_pwr_on_t);
1073         if (ret < 0)
1074                 dev_info(pcie->dev, "Failed to read ASPM Power On time: %d\n",
1075                          ret);
1076
1077         ret = of_property_read_u32(np, "nvidia,aspm-l0s-entrance-latency-us",
1078                                    &pcie->aspm_l0s_enter_lat);
1079         if (ret < 0)
1080                 dev_info(pcie->dev,
1081                          "Failed to read ASPM L0s Entrance latency: %d\n", ret);
1082
1083         ret = of_property_read_u32(np, "num-lanes", &pcie->num_lanes);
1084         if (ret < 0) {
1085                 dev_err(pcie->dev, "Failed to read num-lanes: %d\n", ret);
1086                 return ret;
1087         }
1088
1089         ret = of_property_read_u32_index(np, "nvidia,bpmp", 1, &pcie->cid);
1090         if (ret) {
1091                 dev_err(pcie->dev, "Failed to read Controller-ID: %d\n", ret);
1092                 return ret;
1093         }
1094
1095         ret = of_property_count_strings(np, "phy-names");
1096         if (ret < 0) {
1097                 dev_err(pcie->dev, "Failed to find PHY entries: %d\n",
1098                         ret);
1099                 return ret;
1100         }
1101         pcie->phy_count = ret;
1102
1103         if (of_property_read_bool(np, "nvidia,update-fc-fixup"))
1104                 pcie->update_fc_fixup = true;
1105
1106         pcie->supports_clkreq =
1107                 of_property_read_bool(pcie->dev->of_node, "supports-clkreq");
1108
1109         pcie->enable_cdm_check =
1110                 of_property_read_bool(np, "snps,enable-cdm-check");
1111
1112         if (pcie->mode == DW_PCIE_RC_TYPE)
1113                 return 0;
1114
1115         /* Endpoint mode specific DT entries */
1116         pcie->pex_rst_gpiod = devm_gpiod_get(pcie->dev, "reset", GPIOD_IN);
1117         if (IS_ERR(pcie->pex_rst_gpiod)) {
1118                 int err = PTR_ERR(pcie->pex_rst_gpiod);
1119                 const char *level = KERN_ERR;
1120
1121                 if (err == -EPROBE_DEFER)
1122                         level = KERN_DEBUG;
1123
1124                 dev_printk(level, pcie->dev,
1125                            dev_fmt("Failed to get PERST GPIO: %d\n"),
1126                            err);
1127                 return err;
1128         }
1129
1130         pcie->pex_refclk_sel_gpiod = devm_gpiod_get(pcie->dev,
1131                                                     "nvidia,refclk-select",
1132                                                     GPIOD_OUT_HIGH);
1133         if (IS_ERR(pcie->pex_refclk_sel_gpiod)) {
1134                 int err = PTR_ERR(pcie->pex_refclk_sel_gpiod);
1135                 const char *level = KERN_ERR;
1136
1137                 if (err == -EPROBE_DEFER)
1138                         level = KERN_DEBUG;
1139
1140                 dev_printk(level, pcie->dev,
1141                            dev_fmt("Failed to get REFCLK select GPIOs: %d\n"),
1142                            err);
1143                 pcie->pex_refclk_sel_gpiod = NULL;
1144         }
1145
1146         return 0;
1147 }
1148
1149 static int tegra_pcie_bpmp_set_ctrl_state(struct tegra_pcie_dw *pcie,
1150                                           bool enable)
1151 {
1152         struct mrq_uphy_response resp;
1153         struct tegra_bpmp_message msg;
1154         struct mrq_uphy_request req;
1155
1156         /* Controller-5 doesn't need to have its state set by BPMP-FW */
1157         if (pcie->cid == 5)
1158                 return 0;
1159
1160         memset(&req, 0, sizeof(req));
1161         memset(&resp, 0, sizeof(resp));
1162
1163         req.cmd = CMD_UPHY_PCIE_CONTROLLER_STATE;
1164         req.controller_state.pcie_controller = pcie->cid;
1165         req.controller_state.enable = enable;
1166
1167         memset(&msg, 0, sizeof(msg));
1168         msg.mrq = MRQ_UPHY;
1169         msg.tx.data = &req;
1170         msg.tx.size = sizeof(req);
1171         msg.rx.data = &resp;
1172         msg.rx.size = sizeof(resp);
1173
1174         return tegra_bpmp_transfer(pcie->bpmp, &msg);
1175 }
1176
1177 static int tegra_pcie_bpmp_set_pll_state(struct tegra_pcie_dw *pcie,
1178                                          bool enable)
1179 {
1180         struct mrq_uphy_response resp;
1181         struct tegra_bpmp_message msg;
1182         struct mrq_uphy_request req;
1183
1184         memset(&req, 0, sizeof(req));
1185         memset(&resp, 0, sizeof(resp));
1186
1187         if (enable) {
1188                 req.cmd = CMD_UPHY_PCIE_EP_CONTROLLER_PLL_INIT;
1189                 req.ep_ctrlr_pll_init.ep_controller = pcie->cid;
1190         } else {
1191                 req.cmd = CMD_UPHY_PCIE_EP_CONTROLLER_PLL_OFF;
1192                 req.ep_ctrlr_pll_off.ep_controller = pcie->cid;
1193         }
1194
1195         memset(&msg, 0, sizeof(msg));
1196         msg.mrq = MRQ_UPHY;
1197         msg.tx.data = &req;
1198         msg.tx.size = sizeof(req);
1199         msg.rx.data = &resp;
1200         msg.rx.size = sizeof(resp);
1201
1202         return tegra_bpmp_transfer(pcie->bpmp, &msg);
1203 }
1204
1205 static void tegra_pcie_downstream_dev_to_D0(struct tegra_pcie_dw *pcie)
1206 {
1207         struct pcie_port *pp = &pcie->pci.pp;
1208         struct pci_bus *child, *root_bus = NULL;
1209         struct pci_dev *pdev;
1210
1211         /*
1212          * link doesn't go into L2 state with some of the endpoints with Tegra
1213          * if they are not in D0 state. So, need to make sure that immediate
1214          * downstream devices are in D0 state before sending PME_TurnOff to put
1215          * link into L2 state.
1216          * This is as per PCI Express Base r4.0 v1.0 September 27-2017,
1217          * 5.2 Link State Power Management (Page #428).
1218          */
1219
1220         list_for_each_entry(child, &pp->bridge->bus->children, node) {
1221                 /* Bring downstream devices to D0 if they are not already in */
1222                 if (child->parent == pp->bridge->bus) {
1223                         root_bus = child;
1224                         break;
1225                 }
1226         }
1227
1228         if (!root_bus) {
1229                 dev_err(pcie->dev, "Failed to find downstream devices\n");
1230                 return;
1231         }
1232
1233         list_for_each_entry(pdev, &root_bus->devices, bus_list) {
1234                 if (PCI_SLOT(pdev->devfn) == 0) {
1235                         if (pci_set_power_state(pdev, PCI_D0))
1236                                 dev_err(pcie->dev,
1237                                         "Failed to transition %s to D0 state\n",
1238                                         dev_name(&pdev->dev));
1239                 }
1240         }
1241 }
1242
1243 static int tegra_pcie_get_slot_regulators(struct tegra_pcie_dw *pcie)
1244 {
1245         pcie->slot_ctl_3v3 = devm_regulator_get_optional(pcie->dev, "vpcie3v3");
1246         if (IS_ERR(pcie->slot_ctl_3v3)) {
1247                 if (PTR_ERR(pcie->slot_ctl_3v3) != -ENODEV)
1248                         return PTR_ERR(pcie->slot_ctl_3v3);
1249
1250                 pcie->slot_ctl_3v3 = NULL;
1251         }
1252
1253         pcie->slot_ctl_12v = devm_regulator_get_optional(pcie->dev, "vpcie12v");
1254         if (IS_ERR(pcie->slot_ctl_12v)) {
1255                 if (PTR_ERR(pcie->slot_ctl_12v) != -ENODEV)
1256                         return PTR_ERR(pcie->slot_ctl_12v);
1257
1258                 pcie->slot_ctl_12v = NULL;
1259         }
1260
1261         return 0;
1262 }
1263
1264 static int tegra_pcie_enable_slot_regulators(struct tegra_pcie_dw *pcie)
1265 {
1266         int ret;
1267
1268         if (pcie->slot_ctl_3v3) {
1269                 ret = regulator_enable(pcie->slot_ctl_3v3);
1270                 if (ret < 0) {
1271                         dev_err(pcie->dev,
1272                                 "Failed to enable 3.3V slot supply: %d\n", ret);
1273                         return ret;
1274                 }
1275         }
1276
1277         if (pcie->slot_ctl_12v) {
1278                 ret = regulator_enable(pcie->slot_ctl_12v);
1279                 if (ret < 0) {
1280                         dev_err(pcie->dev,
1281                                 "Failed to enable 12V slot supply: %d\n", ret);
1282                         goto fail_12v_enable;
1283                 }
1284         }
1285
1286         /*
1287          * According to PCI Express Card Electromechanical Specification
1288          * Revision 1.1, Table-2.4, T_PVPERL (Power stable to PERST# inactive)
1289          * should be a minimum of 100ms.
1290          */
1291         if (pcie->slot_ctl_3v3 || pcie->slot_ctl_12v)
1292                 msleep(100);
1293
1294         return 0;
1295
1296 fail_12v_enable:
1297         if (pcie->slot_ctl_3v3)
1298                 regulator_disable(pcie->slot_ctl_3v3);
1299         return ret;
1300 }
1301
1302 static void tegra_pcie_disable_slot_regulators(struct tegra_pcie_dw *pcie)
1303 {
1304         if (pcie->slot_ctl_12v)
1305                 regulator_disable(pcie->slot_ctl_12v);
1306         if (pcie->slot_ctl_3v3)
1307                 regulator_disable(pcie->slot_ctl_3v3);
1308 }
1309
1310 static int tegra_pcie_config_controller(struct tegra_pcie_dw *pcie,
1311                                         bool en_hw_hot_rst)
1312 {
1313         int ret;
1314         u32 val;
1315
1316         ret = tegra_pcie_bpmp_set_ctrl_state(pcie, true);
1317         if (ret) {
1318                 dev_err(pcie->dev,
1319                         "Failed to enable controller %u: %d\n", pcie->cid, ret);
1320                 return ret;
1321         }
1322
1323         ret = tegra_pcie_enable_slot_regulators(pcie);
1324         if (ret < 0)
1325                 goto fail_slot_reg_en;
1326
1327         ret = regulator_enable(pcie->pex_ctl_supply);
1328         if (ret < 0) {
1329                 dev_err(pcie->dev, "Failed to enable regulator: %d\n", ret);
1330                 goto fail_reg_en;
1331         }
1332
1333         ret = clk_prepare_enable(pcie->core_clk);
1334         if (ret) {
1335                 dev_err(pcie->dev, "Failed to enable core clock: %d\n", ret);
1336                 goto fail_core_clk;
1337         }
1338
1339         ret = reset_control_deassert(pcie->core_apb_rst);
1340         if (ret) {
1341                 dev_err(pcie->dev, "Failed to deassert core APB reset: %d\n",
1342                         ret);
1343                 goto fail_core_apb_rst;
1344         }
1345
1346         if (en_hw_hot_rst) {
1347                 /* Enable HW_HOT_RST mode */
1348                 val = appl_readl(pcie, APPL_CTRL);
1349                 val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
1350                          APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
1351                 val |= APPL_CTRL_HW_HOT_RST_EN;
1352                 appl_writel(pcie, val, APPL_CTRL);
1353         }
1354
1355         ret = tegra_pcie_enable_phy(pcie);
1356         if (ret) {
1357                 dev_err(pcie->dev, "Failed to enable PHY: %d\n", ret);
1358                 goto fail_phy;
1359         }
1360
1361         /* Update CFG base address */
1362         appl_writel(pcie, pcie->dbi_res->start & APPL_CFG_BASE_ADDR_MASK,
1363                     APPL_CFG_BASE_ADDR);
1364
1365         /* Configure this core for RP mode operation */
1366         appl_writel(pcie, APPL_DM_TYPE_RP, APPL_DM_TYPE);
1367
1368         appl_writel(pcie, 0x0, APPL_CFG_SLCG_OVERRIDE);
1369
1370         val = appl_readl(pcie, APPL_CTRL);
1371         appl_writel(pcie, val | APPL_CTRL_SYS_PRE_DET_STATE, APPL_CTRL);
1372
1373         val = appl_readl(pcie, APPL_CFG_MISC);
1374         val |= (APPL_CFG_MISC_ARCACHE_VAL << APPL_CFG_MISC_ARCACHE_SHIFT);
1375         appl_writel(pcie, val, APPL_CFG_MISC);
1376
1377         if (!pcie->supports_clkreq) {
1378                 val = appl_readl(pcie, APPL_PINMUX);
1379                 val |= APPL_PINMUX_CLKREQ_OVERRIDE_EN;
1380                 val &= ~APPL_PINMUX_CLKREQ_OVERRIDE;
1381                 appl_writel(pcie, val, APPL_PINMUX);
1382         }
1383
1384         /* Update iATU_DMA base address */
1385         appl_writel(pcie,
1386                     pcie->atu_dma_res->start & APPL_CFG_IATU_DMA_BASE_ADDR_MASK,
1387                     APPL_CFG_IATU_DMA_BASE_ADDR);
1388
1389         reset_control_deassert(pcie->core_rst);
1390
1391         pcie->pcie_cap_base = dw_pcie_find_capability(&pcie->pci,
1392                                                       PCI_CAP_ID_EXP);
1393
1394         /* Disable ASPM-L1SS advertisement as there is no CLKREQ routing */
1395         if (!pcie->supports_clkreq) {
1396                 disable_aspm_l11(pcie);
1397                 disable_aspm_l12(pcie);
1398         }
1399
1400         return ret;
1401
1402 fail_phy:
1403         reset_control_assert(pcie->core_apb_rst);
1404 fail_core_apb_rst:
1405         clk_disable_unprepare(pcie->core_clk);
1406 fail_core_clk:
1407         regulator_disable(pcie->pex_ctl_supply);
1408 fail_reg_en:
1409         tegra_pcie_disable_slot_regulators(pcie);
1410 fail_slot_reg_en:
1411         tegra_pcie_bpmp_set_ctrl_state(pcie, false);
1412
1413         return ret;
1414 }
1415
1416 static int __deinit_controller(struct tegra_pcie_dw *pcie)
1417 {
1418         int ret;
1419
1420         ret = reset_control_assert(pcie->core_rst);
1421         if (ret) {
1422                 dev_err(pcie->dev, "Failed to assert \"core\" reset: %d\n",
1423                         ret);
1424                 return ret;
1425         }
1426
1427         tegra_pcie_disable_phy(pcie);
1428
1429         ret = reset_control_assert(pcie->core_apb_rst);
1430         if (ret) {
1431                 dev_err(pcie->dev, "Failed to assert APB reset: %d\n", ret);
1432                 return ret;
1433         }
1434
1435         clk_disable_unprepare(pcie->core_clk);
1436
1437         ret = regulator_disable(pcie->pex_ctl_supply);
1438         if (ret) {
1439                 dev_err(pcie->dev, "Failed to disable regulator: %d\n", ret);
1440                 return ret;
1441         }
1442
1443         tegra_pcie_disable_slot_regulators(pcie);
1444
1445         ret = tegra_pcie_bpmp_set_ctrl_state(pcie, false);
1446         if (ret) {
1447                 dev_err(pcie->dev, "Failed to disable controller %d: %d\n",
1448                         pcie->cid, ret);
1449                 return ret;
1450         }
1451
1452         return ret;
1453 }
1454
1455 static int tegra_pcie_init_controller(struct tegra_pcie_dw *pcie)
1456 {
1457         struct dw_pcie *pci = &pcie->pci;
1458         struct pcie_port *pp = &pci->pp;
1459         int ret;
1460
1461         ret = tegra_pcie_config_controller(pcie, false);
1462         if (ret < 0)
1463                 return ret;
1464
1465         pp->ops = &tegra_pcie_dw_host_ops;
1466
1467         ret = dw_pcie_host_init(pp);
1468         if (ret < 0) {
1469                 dev_err(pcie->dev, "Failed to add PCIe port: %d\n", ret);
1470                 goto fail_host_init;
1471         }
1472
1473         return 0;
1474
1475 fail_host_init:
1476         return __deinit_controller(pcie);
1477 }
1478
1479 static int tegra_pcie_try_link_l2(struct tegra_pcie_dw *pcie)
1480 {
1481         u32 val;
1482
1483         if (!tegra_pcie_dw_link_up(&pcie->pci))
1484                 return 0;
1485
1486         val = appl_readl(pcie, APPL_RADM_STATUS);
1487         val |= APPL_PM_XMT_TURNOFF_STATE;
1488         appl_writel(pcie, val, APPL_RADM_STATUS);
1489
1490         return readl_poll_timeout_atomic(pcie->appl_base + APPL_DEBUG, val,
1491                                  val & APPL_DEBUG_PM_LINKST_IN_L2_LAT,
1492                                  1, PME_ACK_TIMEOUT);
1493 }
1494
1495 static void tegra_pcie_dw_pme_turnoff(struct tegra_pcie_dw *pcie)
1496 {
1497         u32 data;
1498         int err;
1499
1500         if (!tegra_pcie_dw_link_up(&pcie->pci)) {
1501                 dev_dbg(pcie->dev, "PCIe link is not up...!\n");
1502                 return;
1503         }
1504
1505         if (tegra_pcie_try_link_l2(pcie)) {
1506                 dev_info(pcie->dev, "Link didn't transition to L2 state\n");
1507                 /*
1508                  * TX lane clock freq will reset to Gen1 only if link is in L2
1509                  * or detect state.
1510                  * So apply pex_rst to end point to force RP to go into detect
1511                  * state
1512                  */
1513                 data = appl_readl(pcie, APPL_PINMUX);
1514                 data &= ~APPL_PINMUX_PEX_RST;
1515                 appl_writel(pcie, data, APPL_PINMUX);
1516
1517                 err = readl_poll_timeout_atomic(pcie->appl_base + APPL_DEBUG,
1518                                                 data,
1519                                                 ((data &
1520                                                 APPL_DEBUG_LTSSM_STATE_MASK) >>
1521                                                 APPL_DEBUG_LTSSM_STATE_SHIFT) ==
1522                                                 LTSSM_STATE_PRE_DETECT,
1523                                                 1, LTSSM_TIMEOUT);
1524                 if (err) {
1525                         dev_info(pcie->dev, "Link didn't go to detect state\n");
1526                 } else {
1527                         /* Disable LTSSM after link is in detect state */
1528                         data = appl_readl(pcie, APPL_CTRL);
1529                         data &= ~APPL_CTRL_LTSSM_EN;
1530                         appl_writel(pcie, data, APPL_CTRL);
1531                 }
1532         }
1533         /*
1534          * DBI registers may not be accessible after this as PLL-E would be
1535          * down depending on how CLKREQ is pulled by end point
1536          */
1537         data = appl_readl(pcie, APPL_PINMUX);
1538         data |= (APPL_PINMUX_CLKREQ_OVERRIDE_EN | APPL_PINMUX_CLKREQ_OVERRIDE);
1539         /* Cut REFCLK to slot */
1540         data |= APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE_EN;
1541         data &= ~APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE;
1542         appl_writel(pcie, data, APPL_PINMUX);
1543 }
1544
1545 static int tegra_pcie_deinit_controller(struct tegra_pcie_dw *pcie)
1546 {
1547         tegra_pcie_downstream_dev_to_D0(pcie);
1548         dw_pcie_host_deinit(&pcie->pci.pp);
1549         tegra_pcie_dw_pme_turnoff(pcie);
1550
1551         return __deinit_controller(pcie);
1552 }
1553
1554 static int tegra_pcie_config_rp(struct tegra_pcie_dw *pcie)
1555 {
1556         struct pcie_port *pp = &pcie->pci.pp;
1557         struct device *dev = pcie->dev;
1558         char *name;
1559         int ret;
1560
1561         if (IS_ENABLED(CONFIG_PCI_MSI)) {
1562                 pp->msi_irq = of_irq_get_byname(dev->of_node, "msi");
1563                 if (!pp->msi_irq) {
1564                         dev_err(dev, "Failed to get MSI interrupt\n");
1565                         return -ENODEV;
1566                 }
1567         }
1568
1569         pm_runtime_enable(dev);
1570
1571         ret = pm_runtime_get_sync(dev);
1572         if (ret < 0) {
1573                 dev_err(dev, "Failed to get runtime sync for PCIe dev: %d\n",
1574                         ret);
1575                 goto fail_pm_get_sync;
1576         }
1577
1578         ret = pinctrl_pm_select_default_state(dev);
1579         if (ret < 0) {
1580                 dev_err(dev, "Failed to configure sideband pins: %d\n", ret);
1581                 goto fail_pm_get_sync;
1582         }
1583
1584         tegra_pcie_init_controller(pcie);
1585
1586         pcie->link_state = tegra_pcie_dw_link_up(&pcie->pci);
1587         if (!pcie->link_state) {
1588                 ret = -ENOMEDIUM;
1589                 goto fail_host_init;
1590         }
1591
1592         name = devm_kasprintf(dev, GFP_KERNEL, "%pOFP", dev->of_node);
1593         if (!name) {
1594                 ret = -ENOMEM;
1595                 goto fail_host_init;
1596         }
1597
1598         pcie->debugfs = debugfs_create_dir(name, NULL);
1599         init_debugfs(pcie);
1600
1601         return ret;
1602
1603 fail_host_init:
1604         tegra_pcie_deinit_controller(pcie);
1605 fail_pm_get_sync:
1606         pm_runtime_put_sync(dev);
1607         pm_runtime_disable(dev);
1608         return ret;
1609 }
1610
1611 static void pex_ep_event_pex_rst_assert(struct tegra_pcie_dw *pcie)
1612 {
1613         u32 val;
1614         int ret;
1615
1616         if (pcie->ep_state == EP_STATE_DISABLED)
1617                 return;
1618
1619         /* Disable LTSSM */
1620         val = appl_readl(pcie, APPL_CTRL);
1621         val &= ~APPL_CTRL_LTSSM_EN;
1622         appl_writel(pcie, val, APPL_CTRL);
1623
1624         ret = readl_poll_timeout(pcie->appl_base + APPL_DEBUG, val,
1625                                  ((val & APPL_DEBUG_LTSSM_STATE_MASK) >>
1626                                  APPL_DEBUG_LTSSM_STATE_SHIFT) ==
1627                                  LTSSM_STATE_PRE_DETECT,
1628                                  1, LTSSM_TIMEOUT);
1629         if (ret)
1630                 dev_err(pcie->dev, "Failed to go Detect state: %d\n", ret);
1631
1632         reset_control_assert(pcie->core_rst);
1633
1634         tegra_pcie_disable_phy(pcie);
1635
1636         reset_control_assert(pcie->core_apb_rst);
1637
1638         clk_disable_unprepare(pcie->core_clk);
1639
1640         pm_runtime_put_sync(pcie->dev);
1641
1642         ret = tegra_pcie_bpmp_set_pll_state(pcie, false);
1643         if (ret)
1644                 dev_err(pcie->dev, "Failed to turn off UPHY: %d\n", ret);
1645
1646         pcie->ep_state = EP_STATE_DISABLED;
1647         dev_dbg(pcie->dev, "Uninitialization of endpoint is completed\n");
1648 }
1649
1650 static void pex_ep_event_pex_rst_deassert(struct tegra_pcie_dw *pcie)
1651 {
1652         struct dw_pcie *pci = &pcie->pci;
1653         struct dw_pcie_ep *ep = &pci->ep;
1654         struct device *dev = pcie->dev;
1655         u32 val;
1656         int ret;
1657
1658         if (pcie->ep_state == EP_STATE_ENABLED)
1659                 return;
1660
1661         ret = pm_runtime_resume_and_get(dev);
1662         if (ret < 0) {
1663                 dev_err(dev, "Failed to get runtime sync for PCIe dev: %d\n",
1664                         ret);
1665                 return;
1666         }
1667
1668         ret = tegra_pcie_bpmp_set_pll_state(pcie, true);
1669         if (ret) {
1670                 dev_err(dev, "Failed to init UPHY for PCIe EP: %d\n", ret);
1671                 goto fail_pll_init;
1672         }
1673
1674         ret = clk_prepare_enable(pcie->core_clk);
1675         if (ret) {
1676                 dev_err(dev, "Failed to enable core clock: %d\n", ret);
1677                 goto fail_core_clk_enable;
1678         }
1679
1680         ret = reset_control_deassert(pcie->core_apb_rst);
1681         if (ret) {
1682                 dev_err(dev, "Failed to deassert core APB reset: %d\n", ret);
1683                 goto fail_core_apb_rst;
1684         }
1685
1686         ret = tegra_pcie_enable_phy(pcie);
1687         if (ret) {
1688                 dev_err(dev, "Failed to enable PHY: %d\n", ret);
1689                 goto fail_phy;
1690         }
1691
1692         /* Clear any stale interrupt statuses */
1693         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L0);
1694         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_0_0);
1695         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_1);
1696         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_2);
1697         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_3);
1698         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_6);
1699         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_7);
1700         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_8_0);
1701         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_9);
1702         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_10);
1703         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_11);
1704         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_13);
1705         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_14);
1706         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_15);
1707         appl_writel(pcie, 0xFFFFFFFF, APPL_INTR_STATUS_L1_17);
1708
1709         /* configure this core for EP mode operation */
1710         val = appl_readl(pcie, APPL_DM_TYPE);
1711         val &= ~APPL_DM_TYPE_MASK;
1712         val |= APPL_DM_TYPE_EP;
1713         appl_writel(pcie, val, APPL_DM_TYPE);
1714
1715         appl_writel(pcie, 0x0, APPL_CFG_SLCG_OVERRIDE);
1716
1717         val = appl_readl(pcie, APPL_CTRL);
1718         val |= APPL_CTRL_SYS_PRE_DET_STATE;
1719         val |= APPL_CTRL_HW_HOT_RST_EN;
1720         appl_writel(pcie, val, APPL_CTRL);
1721
1722         val = appl_readl(pcie, APPL_CFG_MISC);
1723         val |= APPL_CFG_MISC_SLV_EP_MODE;
1724         val |= (APPL_CFG_MISC_ARCACHE_VAL << APPL_CFG_MISC_ARCACHE_SHIFT);
1725         appl_writel(pcie, val, APPL_CFG_MISC);
1726
1727         val = appl_readl(pcie, APPL_PINMUX);
1728         val |= APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE_EN;
1729         val |= APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE;
1730         appl_writel(pcie, val, APPL_PINMUX);
1731
1732         appl_writel(pcie, pcie->dbi_res->start & APPL_CFG_BASE_ADDR_MASK,
1733                     APPL_CFG_BASE_ADDR);
1734
1735         appl_writel(pcie, pcie->atu_dma_res->start &
1736                     APPL_CFG_IATU_DMA_BASE_ADDR_MASK,
1737                     APPL_CFG_IATU_DMA_BASE_ADDR);
1738
1739         val = appl_readl(pcie, APPL_INTR_EN_L0_0);
1740         val |= APPL_INTR_EN_L0_0_SYS_INTR_EN;
1741         val |= APPL_INTR_EN_L0_0_LINK_STATE_INT_EN;
1742         val |= APPL_INTR_EN_L0_0_PCI_CMD_EN_INT_EN;
1743         appl_writel(pcie, val, APPL_INTR_EN_L0_0);
1744
1745         val = appl_readl(pcie, APPL_INTR_EN_L1_0_0);
1746         val |= APPL_INTR_EN_L1_0_0_HOT_RESET_DONE_INT_EN;
1747         val |= APPL_INTR_EN_L1_0_0_RDLH_LINK_UP_INT_EN;
1748         appl_writel(pcie, val, APPL_INTR_EN_L1_0_0);
1749
1750         reset_control_deassert(pcie->core_rst);
1751
1752         if (pcie->update_fc_fixup) {
1753                 val = dw_pcie_readl_dbi(pci, CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF);
1754                 val |= 0x1 << CFG_TIMER_CTRL_ACK_NAK_SHIFT;
1755                 dw_pcie_writel_dbi(pci, CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF, val);
1756         }
1757
1758         config_gen3_gen4_eq_presets(pcie);
1759
1760         init_host_aspm(pcie);
1761
1762         /* Disable ASPM-L1SS advertisement if there is no CLKREQ routing */
1763         if (!pcie->supports_clkreq) {
1764                 disable_aspm_l11(pcie);
1765                 disable_aspm_l12(pcie);
1766         }
1767
1768         val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
1769         val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
1770         dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
1771
1772         pcie->pcie_cap_base = dw_pcie_find_capability(&pcie->pci,
1773                                                       PCI_CAP_ID_EXP);
1774         clk_set_rate(pcie->core_clk, GEN4_CORE_CLK_FREQ);
1775
1776         val = (ep->msi_mem_phys & MSIX_ADDR_MATCH_LOW_OFF_MASK);
1777         val |= MSIX_ADDR_MATCH_LOW_OFF_EN;
1778         dw_pcie_writel_dbi(pci, MSIX_ADDR_MATCH_LOW_OFF, val);
1779         val = (upper_32_bits(ep->msi_mem_phys) & MSIX_ADDR_MATCH_HIGH_OFF_MASK);
1780         dw_pcie_writel_dbi(pci, MSIX_ADDR_MATCH_HIGH_OFF, val);
1781
1782         ret = dw_pcie_ep_init_complete(ep);
1783         if (ret) {
1784                 dev_err(dev, "Failed to complete initialization: %d\n", ret);
1785                 goto fail_init_complete;
1786         }
1787
1788         dw_pcie_ep_init_notify(ep);
1789
1790         /* Enable LTSSM */
1791         val = appl_readl(pcie, APPL_CTRL);
1792         val |= APPL_CTRL_LTSSM_EN;
1793         appl_writel(pcie, val, APPL_CTRL);
1794
1795         pcie->ep_state = EP_STATE_ENABLED;
1796         dev_dbg(dev, "Initialization of endpoint is completed\n");
1797
1798         return;
1799
1800 fail_init_complete:
1801         reset_control_assert(pcie->core_rst);
1802         tegra_pcie_disable_phy(pcie);
1803 fail_phy:
1804         reset_control_assert(pcie->core_apb_rst);
1805 fail_core_apb_rst:
1806         clk_disable_unprepare(pcie->core_clk);
1807 fail_core_clk_enable:
1808         tegra_pcie_bpmp_set_pll_state(pcie, false);
1809 fail_pll_init:
1810         pm_runtime_put_sync(dev);
1811 }
1812
1813 static irqreturn_t tegra_pcie_ep_pex_rst_irq(int irq, void *arg)
1814 {
1815         struct tegra_pcie_dw *pcie = arg;
1816
1817         if (gpiod_get_value(pcie->pex_rst_gpiod))
1818                 pex_ep_event_pex_rst_assert(pcie);
1819         else
1820                 pex_ep_event_pex_rst_deassert(pcie);
1821
1822         return IRQ_HANDLED;
1823 }
1824
1825 static int tegra_pcie_ep_raise_legacy_irq(struct tegra_pcie_dw *pcie, u16 irq)
1826 {
1827         /* Tegra194 supports only INTA */
1828         if (irq > 1)
1829                 return -EINVAL;
1830
1831         appl_writel(pcie, 1, APPL_LEGACY_INTX);
1832         usleep_range(1000, 2000);
1833         appl_writel(pcie, 0, APPL_LEGACY_INTX);
1834         return 0;
1835 }
1836
1837 static int tegra_pcie_ep_raise_msi_irq(struct tegra_pcie_dw *pcie, u16 irq)
1838 {
1839         if (unlikely(irq > 31))
1840                 return -EINVAL;
1841
1842         appl_writel(pcie, BIT(irq), APPL_MSI_CTRL_1);
1843
1844         return 0;
1845 }
1846
1847 static int tegra_pcie_ep_raise_msix_irq(struct tegra_pcie_dw *pcie, u16 irq)
1848 {
1849         struct dw_pcie_ep *ep = &pcie->pci.ep;
1850
1851         writel(irq, ep->msi_mem);
1852
1853         return 0;
1854 }
1855
1856 static int tegra_pcie_ep_raise_irq(struct dw_pcie_ep *ep, u8 func_no,
1857                                    enum pci_epc_irq_type type,
1858                                    u16 interrupt_num)
1859 {
1860         struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
1861         struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
1862
1863         switch (type) {
1864         case PCI_EPC_IRQ_LEGACY:
1865                 return tegra_pcie_ep_raise_legacy_irq(pcie, interrupt_num);
1866
1867         case PCI_EPC_IRQ_MSI:
1868                 return tegra_pcie_ep_raise_msi_irq(pcie, interrupt_num);
1869
1870         case PCI_EPC_IRQ_MSIX:
1871                 return tegra_pcie_ep_raise_msix_irq(pcie, interrupt_num);
1872
1873         default:
1874                 dev_err(pci->dev, "Unknown IRQ type\n");
1875                 return -EPERM;
1876         }
1877
1878         return 0;
1879 }
1880
1881 static const struct pci_epc_features tegra_pcie_epc_features = {
1882         .linkup_notifier = true,
1883         .core_init_notifier = true,
1884         .msi_capable = false,
1885         .msix_capable = false,
1886         .reserved_bar = 1 << BAR_2 | 1 << BAR_3 | 1 << BAR_4 | 1 << BAR_5,
1887         .bar_fixed_64bit = 1 << BAR_0,
1888         .bar_fixed_size[0] = SZ_1M,
1889 };
1890
1891 static const struct pci_epc_features*
1892 tegra_pcie_ep_get_features(struct dw_pcie_ep *ep)
1893 {
1894         return &tegra_pcie_epc_features;
1895 }
1896
1897 static struct dw_pcie_ep_ops pcie_ep_ops = {
1898         .raise_irq = tegra_pcie_ep_raise_irq,
1899         .get_features = tegra_pcie_ep_get_features,
1900 };
1901
1902 static int tegra_pcie_config_ep(struct tegra_pcie_dw *pcie,
1903                                 struct platform_device *pdev)
1904 {
1905         struct dw_pcie *pci = &pcie->pci;
1906         struct device *dev = pcie->dev;
1907         struct dw_pcie_ep *ep;
1908         struct resource *res;
1909         char *name;
1910         int ret;
1911
1912         ep = &pci->ep;
1913         ep->ops = &pcie_ep_ops;
1914
1915         res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "addr_space");
1916         if (!res)
1917                 return -EINVAL;
1918
1919         ep->phys_base = res->start;
1920         ep->addr_size = resource_size(res);
1921         ep->page_size = SZ_64K;
1922
1923         ret = gpiod_set_debounce(pcie->pex_rst_gpiod, PERST_DEBOUNCE_TIME);
1924         if (ret < 0) {
1925                 dev_err(dev, "Failed to set PERST GPIO debounce time: %d\n",
1926                         ret);
1927                 return ret;
1928         }
1929
1930         ret = gpiod_to_irq(pcie->pex_rst_gpiod);
1931         if (ret < 0) {
1932                 dev_err(dev, "Failed to get IRQ for PERST GPIO: %d\n", ret);
1933                 return ret;
1934         }
1935         pcie->pex_rst_irq = (unsigned int)ret;
1936
1937         name = devm_kasprintf(dev, GFP_KERNEL, "tegra_pcie_%u_pex_rst_irq",
1938                               pcie->cid);
1939         if (!name) {
1940                 dev_err(dev, "Failed to create PERST IRQ string\n");
1941                 return -ENOMEM;
1942         }
1943
1944         irq_set_status_flags(pcie->pex_rst_irq, IRQ_NOAUTOEN);
1945
1946         pcie->ep_state = EP_STATE_DISABLED;
1947
1948         ret = devm_request_threaded_irq(dev, pcie->pex_rst_irq, NULL,
1949                                         tegra_pcie_ep_pex_rst_irq,
1950                                         IRQF_TRIGGER_RISING |
1951                                         IRQF_TRIGGER_FALLING | IRQF_ONESHOT,
1952                                         name, (void *)pcie);
1953         if (ret < 0) {
1954                 dev_err(dev, "Failed to request IRQ for PERST: %d\n", ret);
1955                 return ret;
1956         }
1957
1958         name = devm_kasprintf(dev, GFP_KERNEL, "tegra_pcie_%u_ep_work",
1959                               pcie->cid);
1960         if (!name) {
1961                 dev_err(dev, "Failed to create PCIe EP work thread string\n");
1962                 return -ENOMEM;
1963         }
1964
1965         pm_runtime_enable(dev);
1966
1967         ret = dw_pcie_ep_init(ep);
1968         if (ret) {
1969                 dev_err(dev, "Failed to initialize DWC Endpoint subsystem: %d\n",
1970                         ret);
1971                 pm_runtime_disable(dev);
1972                 return ret;
1973         }
1974
1975         return 0;
1976 }
1977
1978 static int tegra_pcie_dw_probe(struct platform_device *pdev)
1979 {
1980         const struct tegra_pcie_dw_of_data *data;
1981         struct device *dev = &pdev->dev;
1982         struct resource *atu_dma_res;
1983         struct tegra_pcie_dw *pcie;
1984         struct resource *dbi_res;
1985         struct pcie_port *pp;
1986         struct dw_pcie *pci;
1987         struct phy **phys;
1988         char *name;
1989         int ret;
1990         u32 i;
1991
1992         data = of_device_get_match_data(dev);
1993
1994         pcie = devm_kzalloc(dev, sizeof(*pcie), GFP_KERNEL);
1995         if (!pcie)
1996                 return -ENOMEM;
1997
1998         pci = &pcie->pci;
1999         pci->dev = &pdev->dev;
2000         pci->ops = &tegra_dw_pcie_ops;
2001         pci->n_fts[0] = N_FTS_VAL;
2002         pci->n_fts[1] = FTS_VAL;
2003
2004         pp = &pci->pp;
2005         pcie->dev = &pdev->dev;
2006         pcie->mode = (enum dw_pcie_device_mode)data->mode;
2007
2008         ret = tegra_pcie_dw_parse_dt(pcie);
2009         if (ret < 0) {
2010                 const char *level = KERN_ERR;
2011
2012                 if (ret == -EPROBE_DEFER)
2013                         level = KERN_DEBUG;
2014
2015                 dev_printk(level, dev,
2016                            dev_fmt("Failed to parse device tree: %d\n"),
2017                            ret);
2018                 return ret;
2019         }
2020
2021         ret = tegra_pcie_get_slot_regulators(pcie);
2022         if (ret < 0) {
2023                 const char *level = KERN_ERR;
2024
2025                 if (ret == -EPROBE_DEFER)
2026                         level = KERN_DEBUG;
2027
2028                 dev_printk(level, dev,
2029                            dev_fmt("Failed to get slot regulators: %d\n"),
2030                            ret);
2031                 return ret;
2032         }
2033
2034         if (pcie->pex_refclk_sel_gpiod)
2035                 gpiod_set_value(pcie->pex_refclk_sel_gpiod, 1);
2036
2037         pcie->pex_ctl_supply = devm_regulator_get(dev, "vddio-pex-ctl");
2038         if (IS_ERR(pcie->pex_ctl_supply)) {
2039                 ret = PTR_ERR(pcie->pex_ctl_supply);
2040                 if (ret != -EPROBE_DEFER)
2041                         dev_err(dev, "Failed to get regulator: %ld\n",
2042                                 PTR_ERR(pcie->pex_ctl_supply));
2043                 return ret;
2044         }
2045
2046         pcie->core_clk = devm_clk_get(dev, "core");
2047         if (IS_ERR(pcie->core_clk)) {
2048                 dev_err(dev, "Failed to get core clock: %ld\n",
2049                         PTR_ERR(pcie->core_clk));
2050                 return PTR_ERR(pcie->core_clk);
2051         }
2052
2053         pcie->appl_res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
2054                                                       "appl");
2055         if (!pcie->appl_res) {
2056                 dev_err(dev, "Failed to find \"appl\" region\n");
2057                 return -ENODEV;
2058         }
2059
2060         pcie->appl_base = devm_ioremap_resource(dev, pcie->appl_res);
2061         if (IS_ERR(pcie->appl_base))
2062                 return PTR_ERR(pcie->appl_base);
2063
2064         pcie->core_apb_rst = devm_reset_control_get(dev, "apb");
2065         if (IS_ERR(pcie->core_apb_rst)) {
2066                 dev_err(dev, "Failed to get APB reset: %ld\n",
2067                         PTR_ERR(pcie->core_apb_rst));
2068                 return PTR_ERR(pcie->core_apb_rst);
2069         }
2070
2071         phys = devm_kcalloc(dev, pcie->phy_count, sizeof(*phys), GFP_KERNEL);
2072         if (!phys)
2073                 return -ENOMEM;
2074
2075         for (i = 0; i < pcie->phy_count; i++) {
2076                 name = kasprintf(GFP_KERNEL, "p2u-%u", i);
2077                 if (!name) {
2078                         dev_err(dev, "Failed to create P2U string\n");
2079                         return -ENOMEM;
2080                 }
2081                 phys[i] = devm_phy_get(dev, name);
2082                 kfree(name);
2083                 if (IS_ERR(phys[i])) {
2084                         ret = PTR_ERR(phys[i]);
2085                         if (ret != -EPROBE_DEFER)
2086                                 dev_err(dev, "Failed to get PHY: %d\n", ret);
2087                         return ret;
2088                 }
2089         }
2090
2091         pcie->phys = phys;
2092
2093         dbi_res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "dbi");
2094         if (!dbi_res) {
2095                 dev_err(dev, "Failed to find \"dbi\" region\n");
2096                 return -ENODEV;
2097         }
2098         pcie->dbi_res = dbi_res;
2099
2100         pci->dbi_base = devm_ioremap_resource(dev, dbi_res);
2101         if (IS_ERR(pci->dbi_base))
2102                 return PTR_ERR(pci->dbi_base);
2103
2104         /* Tegra HW locates DBI2 at a fixed offset from DBI */
2105         pci->dbi_base2 = pci->dbi_base + 0x1000;
2106
2107         atu_dma_res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
2108                                                    "atu_dma");
2109         if (!atu_dma_res) {
2110                 dev_err(dev, "Failed to find \"atu_dma\" region\n");
2111                 return -ENODEV;
2112         }
2113         pcie->atu_dma_res = atu_dma_res;
2114
2115         pci->atu_base = devm_ioremap_resource(dev, atu_dma_res);
2116         if (IS_ERR(pci->atu_base))
2117                 return PTR_ERR(pci->atu_base);
2118
2119         pcie->core_rst = devm_reset_control_get(dev, "core");
2120         if (IS_ERR(pcie->core_rst)) {
2121                 dev_err(dev, "Failed to get core reset: %ld\n",
2122                         PTR_ERR(pcie->core_rst));
2123                 return PTR_ERR(pcie->core_rst);
2124         }
2125
2126         pp->irq = platform_get_irq_byname(pdev, "intr");
2127         if (pp->irq < 0)
2128                 return pp->irq;
2129
2130         pcie->bpmp = tegra_bpmp_get(dev);
2131         if (IS_ERR(pcie->bpmp))
2132                 return PTR_ERR(pcie->bpmp);
2133
2134         platform_set_drvdata(pdev, pcie);
2135
2136         switch (pcie->mode) {
2137         case DW_PCIE_RC_TYPE:
2138                 ret = devm_request_irq(dev, pp->irq, tegra_pcie_rp_irq_handler,
2139                                        IRQF_SHARED, "tegra-pcie-intr", pcie);
2140                 if (ret) {
2141                         dev_err(dev, "Failed to request IRQ %d: %d\n", pp->irq,
2142                                 ret);
2143                         goto fail;
2144                 }
2145
2146                 ret = tegra_pcie_config_rp(pcie);
2147                 if (ret && ret != -ENOMEDIUM)
2148                         goto fail;
2149                 else
2150                         return 0;
2151                 break;
2152
2153         case DW_PCIE_EP_TYPE:
2154                 ret = devm_request_threaded_irq(dev, pp->irq,
2155                                                 tegra_pcie_ep_hard_irq,
2156                                                 tegra_pcie_ep_irq_thread,
2157                                                 IRQF_SHARED | IRQF_ONESHOT,
2158                                                 "tegra-pcie-ep-intr", pcie);
2159                 if (ret) {
2160                         dev_err(dev, "Failed to request IRQ %d: %d\n", pp->irq,
2161                                 ret);
2162                         goto fail;
2163                 }
2164
2165                 ret = tegra_pcie_config_ep(pcie, pdev);
2166                 if (ret < 0)
2167                         goto fail;
2168                 break;
2169
2170         default:
2171                 dev_err(dev, "Invalid PCIe device type %d\n", pcie->mode);
2172         }
2173
2174 fail:
2175         tegra_bpmp_put(pcie->bpmp);
2176         return ret;
2177 }
2178
2179 static int tegra_pcie_dw_remove(struct platform_device *pdev)
2180 {
2181         struct tegra_pcie_dw *pcie = platform_get_drvdata(pdev);
2182
2183         if (!pcie->link_state)
2184                 return 0;
2185
2186         debugfs_remove_recursive(pcie->debugfs);
2187         tegra_pcie_deinit_controller(pcie);
2188         pm_runtime_put_sync(pcie->dev);
2189         pm_runtime_disable(pcie->dev);
2190         tegra_bpmp_put(pcie->bpmp);
2191         if (pcie->pex_refclk_sel_gpiod)
2192                 gpiod_set_value(pcie->pex_refclk_sel_gpiod, 0);
2193
2194         return 0;
2195 }
2196
2197 static int tegra_pcie_dw_suspend_late(struct device *dev)
2198 {
2199         struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
2200         u32 val;
2201
2202         if (!pcie->link_state)
2203                 return 0;
2204
2205         /* Enable HW_HOT_RST mode */
2206         val = appl_readl(pcie, APPL_CTRL);
2207         val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
2208                  APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
2209         val |= APPL_CTRL_HW_HOT_RST_EN;
2210         appl_writel(pcie, val, APPL_CTRL);
2211
2212         return 0;
2213 }
2214
2215 static int tegra_pcie_dw_suspend_noirq(struct device *dev)
2216 {
2217         struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
2218
2219         if (!pcie->link_state)
2220                 return 0;
2221
2222         /* Save MSI interrupt vector */
2223         pcie->msi_ctrl_int = dw_pcie_readl_dbi(&pcie->pci,
2224                                                PORT_LOGIC_MSI_CTRL_INT_0_EN);
2225         tegra_pcie_downstream_dev_to_D0(pcie);
2226         tegra_pcie_dw_pme_turnoff(pcie);
2227
2228         return __deinit_controller(pcie);
2229 }
2230
2231 static int tegra_pcie_dw_resume_noirq(struct device *dev)
2232 {
2233         struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
2234         int ret;
2235
2236         if (!pcie->link_state)
2237                 return 0;
2238
2239         ret = tegra_pcie_config_controller(pcie, true);
2240         if (ret < 0)
2241                 return ret;
2242
2243         ret = tegra_pcie_dw_host_init(&pcie->pci.pp);
2244         if (ret < 0) {
2245                 dev_err(dev, "Failed to init host: %d\n", ret);
2246                 goto fail_host_init;
2247         }
2248
2249         /* Restore MSI interrupt vector */
2250         dw_pcie_writel_dbi(&pcie->pci, PORT_LOGIC_MSI_CTRL_INT_0_EN,
2251                            pcie->msi_ctrl_int);
2252
2253         return 0;
2254
2255 fail_host_init:
2256         return __deinit_controller(pcie);
2257 }
2258
2259 static int tegra_pcie_dw_resume_early(struct device *dev)
2260 {
2261         struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
2262         u32 val;
2263
2264         if (!pcie->link_state)
2265                 return 0;
2266
2267         /* Disable HW_HOT_RST mode */
2268         val = appl_readl(pcie, APPL_CTRL);
2269         val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
2270                  APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
2271         val |= APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST <<
2272                APPL_CTRL_HW_HOT_RST_MODE_SHIFT;
2273         val &= ~APPL_CTRL_HW_HOT_RST_EN;
2274         appl_writel(pcie, val, APPL_CTRL);
2275
2276         return 0;
2277 }
2278
2279 static void tegra_pcie_dw_shutdown(struct platform_device *pdev)
2280 {
2281         struct tegra_pcie_dw *pcie = platform_get_drvdata(pdev);
2282
2283         if (!pcie->link_state)
2284                 return;
2285
2286         debugfs_remove_recursive(pcie->debugfs);
2287         tegra_pcie_downstream_dev_to_D0(pcie);
2288
2289         disable_irq(pcie->pci.pp.irq);
2290         if (IS_ENABLED(CONFIG_PCI_MSI))
2291                 disable_irq(pcie->pci.pp.msi_irq);
2292
2293         tegra_pcie_dw_pme_turnoff(pcie);
2294         __deinit_controller(pcie);
2295 }
2296
2297 static const struct tegra_pcie_dw_of_data tegra_pcie_dw_rc_of_data = {
2298         .mode = DW_PCIE_RC_TYPE,
2299 };
2300
2301 static const struct tegra_pcie_dw_of_data tegra_pcie_dw_ep_of_data = {
2302         .mode = DW_PCIE_EP_TYPE,
2303 };
2304
2305 static const struct of_device_id tegra_pcie_dw_of_match[] = {
2306         {
2307                 .compatible = "nvidia,tegra194-pcie",
2308                 .data = &tegra_pcie_dw_rc_of_data,
2309         },
2310         {
2311                 .compatible = "nvidia,tegra194-pcie-ep",
2312                 .data = &tegra_pcie_dw_ep_of_data,
2313         },
2314         {},
2315 };
2316
2317 static const struct dev_pm_ops tegra_pcie_dw_pm_ops = {
2318         .suspend_late = tegra_pcie_dw_suspend_late,
2319         .suspend_noirq = tegra_pcie_dw_suspend_noirq,
2320         .resume_noirq = tegra_pcie_dw_resume_noirq,
2321         .resume_early = tegra_pcie_dw_resume_early,
2322 };
2323
2324 static struct platform_driver tegra_pcie_dw_driver = {
2325         .probe = tegra_pcie_dw_probe,
2326         .remove = tegra_pcie_dw_remove,
2327         .shutdown = tegra_pcie_dw_shutdown,
2328         .driver = {
2329                 .name   = "tegra194-pcie",
2330                 .pm = &tegra_pcie_dw_pm_ops,
2331                 .of_match_table = tegra_pcie_dw_of_match,
2332         },
2333 };
2334 module_platform_driver(tegra_pcie_dw_driver);
2335
2336 MODULE_DEVICE_TABLE(of, tegra_pcie_dw_of_match);
2337
2338 MODULE_AUTHOR("Vidya Sagar <vidyas@nvidia.com>");
2339 MODULE_DESCRIPTION("NVIDIA PCIe host controller driver");
2340 MODULE_LICENSE("GPL v2");