1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Rockchip AXI PCIe host controller driver
4  *
5  * Copyright (c) 2016 Rockchip, Inc.
6  *
7  * Author: Shawn Lin <shawn.lin@rock-chips.com>
8  *         Wenrui Li <wenrui.li@rock-chips.com>
9  *
10  * Bits taken from Synopsys DesignWare Host controller driver and
11  * ARM PCI Host generic driver.
12  */
13 
14 #include <linux/bitrev.h>
15 #include <linux/clk.h>
16 #include <linux/delay.h>
17 #include <linux/gpio/consumer.h>
18 #include <linux/init.h>
19 #include <linux/interrupt.h>
20 #include <linux/iopoll.h>
21 #include <linux/irq.h>
22 #include <linux/irqchip/chained_irq.h>
23 #include <linux/irqdomain.h>
24 #include <linux/kernel.h>
25 #include <linux/mfd/syscon.h>
26 #include <linux/module.h>
27 #include <linux/of_address.h>
28 #include <linux/of_device.h>
29 #include <linux/of_pci.h>
30 #include <linux/of_platform.h>
31 #include <linux/of_irq.h>
32 #include <linux/pci.h>
33 #include <linux/pci_ids.h>
34 #include <linux/phy/phy.h>
35 #include <linux/platform_device.h>
36 #include <linux/reset.h>
37 #include <linux/regmap.h>
38 
39 #include "../pci.h"
40 #include "pcie-rockchip.h"
41 
42 static void rockchip_pcie_enable_bw_int(struct rockchip_pcie *rockchip)
43 {
44 	u32 status;
45 
46 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
47 	status |= (PCI_EXP_LNKCTL_LBMIE | PCI_EXP_LNKCTL_LABIE);
48 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
49 }
50 
51 static void rockchip_pcie_clr_bw_int(struct rockchip_pcie *rockchip)
52 {
53 	u32 status;
54 
55 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
56 	status |= (PCI_EXP_LNKSTA_LBMS | PCI_EXP_LNKSTA_LABS) << 16;
57 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
58 }
59 
60 static void rockchip_pcie_update_txcredit_mui(struct rockchip_pcie *rockchip)
61 {
62 	u32 val;
63 
64 	/* Update Tx credit maximum update interval */
65 	val = rockchip_pcie_read(rockchip, PCIE_CORE_TXCREDIT_CFG1);
66 	val &= ~PCIE_CORE_TXCREDIT_CFG1_MUI_MASK;
67 	val |= PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(24000);	/* ns */
68 	rockchip_pcie_write(rockchip, val, PCIE_CORE_TXCREDIT_CFG1);
69 }
70 
71 static int rockchip_pcie_valid_device(struct rockchip_pcie *rockchip,
72 				      struct pci_bus *bus, int dev)
73 {
74 	/*
75 	 * Access only one slot on each root port.
76 	 * Do not read more than one device on the bus directly attached
77 	 * to RC's downstream side.
78 	 */
79 	if (pci_is_root_bus(bus) || pci_is_root_bus(bus->parent))
80 		return dev == 0;
81 
82 	return 1;
83 }
84 
85 static u8 rockchip_pcie_lane_map(struct rockchip_pcie *rockchip)
86 {
87 	u32 val;
88 	u8 map;
89 
90 	if (rockchip->legacy_phy)
91 		return GENMASK(MAX_LANE_NUM - 1, 0);
92 
93 	val = rockchip_pcie_read(rockchip, PCIE_CORE_LANE_MAP);
94 	map = val & PCIE_CORE_LANE_MAP_MASK;
95 
96 	/* The link may be using a reverse-indexed mapping. */
97 	if (val & PCIE_CORE_LANE_MAP_REVERSE)
98 		map = bitrev8(map) >> 4;
99 
100 	return map;
101 }
102 
103 static int rockchip_pcie_rd_own_conf(struct rockchip_pcie *rockchip,
104 				     int where, int size, u32 *val)
105 {
106 	void __iomem *addr;
107 
108 	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + where;
109 
110 	if (!IS_ALIGNED((uintptr_t)addr, size)) {
111 		*val = 0;
112 		return PCIBIOS_BAD_REGISTER_NUMBER;
113 	}
114 
115 	if (size == 4) {
116 		*val = readl(addr);
117 	} else if (size == 2) {
118 		*val = readw(addr);
119 	} else if (size == 1) {
120 		*val = readb(addr);
121 	} else {
122 		*val = 0;
123 		return PCIBIOS_BAD_REGISTER_NUMBER;
124 	}
125 	return PCIBIOS_SUCCESSFUL;
126 }
127 
128 static int rockchip_pcie_wr_own_conf(struct rockchip_pcie *rockchip,
129 				     int where, int size, u32 val)
130 {
131 	u32 mask, tmp, offset;
132 	void __iomem *addr;
133 
134 	offset = where & ~0x3;
135 	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + offset;
136 
137 	if (size == 4) {
138 		writel(val, addr);
139 		return PCIBIOS_SUCCESSFUL;
140 	}
141 
142 	mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
143 
144 	/*
145 	 * N.B. This read/modify/write isn't safe in general because it can
146 	 * corrupt RW1C bits in adjacent registers.  But the hardware
147 	 * doesn't support smaller writes.
148 	 */
149 	tmp = readl(addr) & mask;
150 	tmp |= val << ((where & 0x3) * 8);
151 	writel(tmp, addr);
152 
153 	return PCIBIOS_SUCCESSFUL;
154 }
155 
156 static int rockchip_pcie_rd_other_conf(struct rockchip_pcie *rockchip,
157 				       struct pci_bus *bus, u32 devfn,
158 				       int where, int size, u32 *val)
159 {
160 	void __iomem *addr;
161 
162 	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
163 
164 	if (!IS_ALIGNED((uintptr_t)addr, size)) {
165 		*val = 0;
166 		return PCIBIOS_BAD_REGISTER_NUMBER;
167 	}
168 
169 	if (pci_is_root_bus(bus->parent))
170 		rockchip_pcie_cfg_configuration_accesses(rockchip,
171 						AXI_WRAPPER_TYPE0_CFG);
172 	else
173 		rockchip_pcie_cfg_configuration_accesses(rockchip,
174 						AXI_WRAPPER_TYPE1_CFG);
175 
176 	if (size == 4) {
177 		*val = readl(addr);
178 	} else if (size == 2) {
179 		*val = readw(addr);
180 	} else if (size == 1) {
181 		*val = readb(addr);
182 	} else {
183 		*val = 0;
184 		return PCIBIOS_BAD_REGISTER_NUMBER;
185 	}
186 	return PCIBIOS_SUCCESSFUL;
187 }
188 
189 static int rockchip_pcie_wr_other_conf(struct rockchip_pcie *rockchip,
190 				       struct pci_bus *bus, u32 devfn,
191 				       int where, int size, u32 val)
192 {
193 	void __iomem *addr;
194 
195 	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
196 
197 	if (!IS_ALIGNED((uintptr_t)addr, size))
198 		return PCIBIOS_BAD_REGISTER_NUMBER;
199 
200 	if (pci_is_root_bus(bus->parent))
201 		rockchip_pcie_cfg_configuration_accesses(rockchip,
202 						AXI_WRAPPER_TYPE0_CFG);
203 	else
204 		rockchip_pcie_cfg_configuration_accesses(rockchip,
205 						AXI_WRAPPER_TYPE1_CFG);
206 
207 	if (size == 4)
208 		writel(val, addr);
209 	else if (size == 2)
210 		writew(val, addr);
211 	else if (size == 1)
212 		writeb(val, addr);
213 	else
214 		return PCIBIOS_BAD_REGISTER_NUMBER;
215 
216 	return PCIBIOS_SUCCESSFUL;
217 }
218 
219 static int rockchip_pcie_rd_conf(struct pci_bus *bus, u32 devfn, int where,
220 				 int size, u32 *val)
221 {
222 	struct rockchip_pcie *rockchip = bus->sysdata;
223 
224 	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
225 		return PCIBIOS_DEVICE_NOT_FOUND;
226 
227 	if (pci_is_root_bus(bus))
228 		return rockchip_pcie_rd_own_conf(rockchip, where, size, val);
229 
230 	return rockchip_pcie_rd_other_conf(rockchip, bus, devfn, where, size,
231 					   val);
232 }
233 
234 static int rockchip_pcie_wr_conf(struct pci_bus *bus, u32 devfn,
235 				 int where, int size, u32 val)
236 {
237 	struct rockchip_pcie *rockchip = bus->sysdata;
238 
239 	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
240 		return PCIBIOS_DEVICE_NOT_FOUND;
241 
242 	if (pci_is_root_bus(bus))
243 		return rockchip_pcie_wr_own_conf(rockchip, where, size, val);
244 
245 	return rockchip_pcie_wr_other_conf(rockchip, bus, devfn, where, size,
246 					   val);
247 }
248 
249 static struct pci_ops rockchip_pcie_ops = {
250 	.read = rockchip_pcie_rd_conf,
251 	.write = rockchip_pcie_wr_conf,
252 };
253 
254 static void rockchip_pcie_set_power_limit(struct rockchip_pcie *rockchip)
255 {
256 	int curr;
257 	u32 status, scale, power;
258 
259 	if (IS_ERR(rockchip->vpcie3v3))
260 		return;
261 
262 	/*
263 	 * Set RC's captured slot power limit and scale if
264 	 * vpcie3v3 available. The default values are both zero
265 	 * which means the software should set these two according
266 	 * to the actual power supply.
267 	 */
268 	curr = regulator_get_current_limit(rockchip->vpcie3v3);
269 	if (curr <= 0)
270 		return;
271 
272 	scale = 3; /* 0.001x */
273 	curr = curr / 1000; /* convert to mA */
274 	power = (curr * 3300) / 1000; /* milliwatt */
275 	while (power > PCIE_RC_CONFIG_DCR_CSPL_LIMIT) {
276 		if (!scale) {
277 			dev_warn(rockchip->dev, "invalid power supply\n");
278 			return;
279 		}
280 		scale--;
281 		power = power / 10;
282 	}
283 
284 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCR);
285 	status |= (power << PCIE_RC_CONFIG_DCR_CSPL_SHIFT) |
286 		  (scale << PCIE_RC_CONFIG_DCR_CPLS_SHIFT);
287 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCR);
288 }
289 
290 /**
291  * rockchip_pcie_host_init_port - Initialize hardware
292  * @rockchip: PCIe port information
293  */
294 static int rockchip_pcie_host_init_port(struct rockchip_pcie *rockchip)
295 {
296 	struct device *dev = rockchip->dev;
297 	int err, i = MAX_LANE_NUM;
298 	u32 status;
299 
300 	gpiod_set_value_cansleep(rockchip->ep_gpio, 0);
301 
302 	err = rockchip_pcie_init_port(rockchip);
303 	if (err)
304 		return err;
305 
306 	/* Fix the transmitted FTS count desired to exit from L0s. */
307 	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL_PLC1);
308 	status = (status & ~PCIE_CORE_CTRL_PLC1_FTS_MASK) |
309 		 (PCIE_CORE_CTRL_PLC1_FTS_CNT << PCIE_CORE_CTRL_PLC1_FTS_SHIFT);
310 	rockchip_pcie_write(rockchip, status, PCIE_CORE_CTRL_PLC1);
311 
312 	rockchip_pcie_set_power_limit(rockchip);
313 
314 	/* Set RC's clock architecture as common clock */
315 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
316 	status |= PCI_EXP_LNKSTA_SLC << 16;
317 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
318 
319 	/* Set RC's RCB to 128 */
320 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
321 	status |= PCI_EXP_LNKCTL_RCB;
322 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
323 
324 	/* Enable Gen1 training */
325 	rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
326 			    PCIE_CLIENT_CONFIG);
327 
328 	gpiod_set_value_cansleep(rockchip->ep_gpio, 1);
329 
330 	/* 500ms timeout value should be enough for Gen1/2 training */
331 	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_BASIC_STATUS1,
332 				 status, PCIE_LINK_UP(status), 20,
333 				 500 * USEC_PER_MSEC);
334 	if (err) {
335 		dev_err(dev, "PCIe link training gen1 timeout!\n");
336 		goto err_power_off_phy;
337 	}
338 
339 	if (rockchip->link_gen == 2) {
340 		/*
341 		 * Enable retrain for gen2. This should be configured only after
342 		 * gen1 finished.
343 		 */
344 		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
345 		status |= PCI_EXP_LNKCTL_RL;
346 		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
347 
348 		err = readl_poll_timeout(rockchip->apb_base + PCIE_CORE_CTRL,
349 					 status, PCIE_LINK_IS_GEN2(status), 20,
350 					 500 * USEC_PER_MSEC);
351 		if (err)
352 			dev_dbg(dev, "PCIe link training gen2 timeout, fall back to gen1!\n");
353 	}
354 
355 	/* Check the final link width from negotiated lane counter from MGMT */
356 	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL);
357 	status = 0x1 << ((status & PCIE_CORE_PL_CONF_LANE_MASK) >>
358 			  PCIE_CORE_PL_CONF_LANE_SHIFT);
359 	dev_dbg(dev, "current link width is x%d\n", status);
360 
361 	/* Power off unused lane(s) */
362 	rockchip->lanes_map = rockchip_pcie_lane_map(rockchip);
363 	for (i = 0; i < MAX_LANE_NUM; i++) {
364 		if (!(rockchip->lanes_map & BIT(i))) {
365 			dev_dbg(dev, "idling lane %d\n", i);
366 			phy_power_off(rockchip->phys[i]);
367 		}
368 	}
369 
370 	rockchip_pcie_write(rockchip, ROCKCHIP_VENDOR_ID,
371 			    PCIE_CORE_CONFIG_VENDOR);
372 	rockchip_pcie_write(rockchip,
373 			    PCI_CLASS_BRIDGE_PCI_NORMAL << 8,
374 			    PCIE_RC_CONFIG_RID_CCR);
375 
376 	/* Clear THP cap's next cap pointer to remove L1 substate cap */
377 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_THP_CAP);
378 	status &= ~PCIE_RC_CONFIG_THP_CAP_NEXT_MASK;
379 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_THP_CAP);
380 
381 	/* Clear L0s from RC's link cap */
382 	if (of_property_read_bool(dev->of_node, "aspm-no-l0s")) {
383 		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LINK_CAP);
384 		status &= ~PCIE_RC_CONFIG_LINK_CAP_L0S;
385 		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LINK_CAP);
386 	}
387 
388 	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCSR);
389 	status &= ~PCIE_RC_CONFIG_DCSR_MPS_MASK;
390 	status |= PCIE_RC_CONFIG_DCSR_MPS_256;
391 	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCSR);
392 
393 	return 0;
394 err_power_off_phy:
395 	while (i--)
396 		phy_power_off(rockchip->phys[i]);
397 	i = MAX_LANE_NUM;
398 	while (i--)
399 		phy_exit(rockchip->phys[i]);
400 	return err;
401 }
402 
403 static irqreturn_t rockchip_pcie_subsys_irq_handler(int irq, void *arg)
404 {
405 	struct rockchip_pcie *rockchip = arg;
406 	struct device *dev = rockchip->dev;
407 	u32 reg;
408 	u32 sub_reg;
409 
410 	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
411 	if (reg & PCIE_CLIENT_INT_LOCAL) {
412 		dev_dbg(dev, "local interrupt received\n");
413 		sub_reg = rockchip_pcie_read(rockchip, PCIE_CORE_INT_STATUS);
414 		if (sub_reg & PCIE_CORE_INT_PRFPE)
415 			dev_dbg(dev, "parity error detected while reading from the PNP receive FIFO RAM\n");
416 
417 		if (sub_reg & PCIE_CORE_INT_CRFPE)
418 			dev_dbg(dev, "parity error detected while reading from the Completion Receive FIFO RAM\n");
419 
420 		if (sub_reg & PCIE_CORE_INT_RRPE)
421 			dev_dbg(dev, "parity error detected while reading from replay buffer RAM\n");
422 
423 		if (sub_reg & PCIE_CORE_INT_PRFO)
424 			dev_dbg(dev, "overflow occurred in the PNP receive FIFO\n");
425 
426 		if (sub_reg & PCIE_CORE_INT_CRFO)
427 			dev_dbg(dev, "overflow occurred in the completion receive FIFO\n");
428 
429 		if (sub_reg & PCIE_CORE_INT_RT)
430 			dev_dbg(dev, "replay timer timed out\n");
431 
432 		if (sub_reg & PCIE_CORE_INT_RTR)
433 			dev_dbg(dev, "replay timer rolled over after 4 transmissions of the same TLP\n");
434 
435 		if (sub_reg & PCIE_CORE_INT_PE)
436 			dev_dbg(dev, "phy error detected on receive side\n");
437 
438 		if (sub_reg & PCIE_CORE_INT_MTR)
439 			dev_dbg(dev, "malformed TLP received from the link\n");
440 
441 		if (sub_reg & PCIE_CORE_INT_UCR)
442 			dev_dbg(dev, "malformed TLP received from the link\n");
443 
444 		if (sub_reg & PCIE_CORE_INT_FCE)
445 			dev_dbg(dev, "an error was observed in the flow control advertisements from the other side\n");
446 
447 		if (sub_reg & PCIE_CORE_INT_CT)
448 			dev_dbg(dev, "a request timed out waiting for completion\n");
449 
450 		if (sub_reg & PCIE_CORE_INT_UTC)
451 			dev_dbg(dev, "unmapped TC error\n");
452 
453 		if (sub_reg & PCIE_CORE_INT_MMVC)
454 			dev_dbg(dev, "MSI mask register changes\n");
455 
456 		rockchip_pcie_write(rockchip, sub_reg, PCIE_CORE_INT_STATUS);
457 	} else if (reg & PCIE_CLIENT_INT_PHY) {
458 		dev_dbg(dev, "phy link changes\n");
459 		rockchip_pcie_update_txcredit_mui(rockchip);
460 		rockchip_pcie_clr_bw_int(rockchip);
461 	}
462 
463 	rockchip_pcie_write(rockchip, reg & PCIE_CLIENT_INT_LOCAL,
464 			    PCIE_CLIENT_INT_STATUS);
465 
466 	return IRQ_HANDLED;
467 }
468 
469 static irqreturn_t rockchip_pcie_client_irq_handler(int irq, void *arg)
470 {
471 	struct rockchip_pcie *rockchip = arg;
472 	struct device *dev = rockchip->dev;
473 	u32 reg;
474 
475 	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
476 	if (reg & PCIE_CLIENT_INT_LEGACY_DONE)
477 		dev_dbg(dev, "legacy done interrupt received\n");
478 
479 	if (reg & PCIE_CLIENT_INT_MSG)
480 		dev_dbg(dev, "message done interrupt received\n");
481 
482 	if (reg & PCIE_CLIENT_INT_HOT_RST)
483 		dev_dbg(dev, "hot reset interrupt received\n");
484 
485 	if (reg & PCIE_CLIENT_INT_DPA)
486 		dev_dbg(dev, "dpa interrupt received\n");
487 
488 	if (reg & PCIE_CLIENT_INT_FATAL_ERR)
489 		dev_dbg(dev, "fatal error interrupt received\n");
490 
491 	if (reg & PCIE_CLIENT_INT_NFATAL_ERR)
492 		dev_dbg(dev, "no fatal error interrupt received\n");
493 
494 	if (reg & PCIE_CLIENT_INT_CORR_ERR)
495 		dev_dbg(dev, "correctable error interrupt received\n");
496 
497 	if (reg & PCIE_CLIENT_INT_PHY)
498 		dev_dbg(dev, "phy interrupt received\n");
499 
500 	rockchip_pcie_write(rockchip, reg & (PCIE_CLIENT_INT_LEGACY_DONE |
501 			      PCIE_CLIENT_INT_MSG | PCIE_CLIENT_INT_HOT_RST |
502 			      PCIE_CLIENT_INT_DPA | PCIE_CLIENT_INT_FATAL_ERR |
503 			      PCIE_CLIENT_INT_NFATAL_ERR |
504 			      PCIE_CLIENT_INT_CORR_ERR |
505 			      PCIE_CLIENT_INT_PHY),
506 		   PCIE_CLIENT_INT_STATUS);
507 
508 	return IRQ_HANDLED;
509 }
510 
511 static void rockchip_pcie_legacy_int_handler(struct irq_desc *desc)
512 {
513 	struct irq_chip *chip = irq_desc_get_chip(desc);
514 	struct rockchip_pcie *rockchip = irq_desc_get_handler_data(desc);
515 	struct device *dev = rockchip->dev;
516 	u32 reg;
517 	u32 hwirq;
518 	int ret;
519 
520 	chained_irq_enter(chip, desc);
521 
522 	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
523 	reg = (reg & PCIE_CLIENT_INTR_MASK) >> PCIE_CLIENT_INTR_SHIFT;
524 
525 	while (reg) {
526 		hwirq = ffs(reg) - 1;
527 		reg &= ~BIT(hwirq);
528 
529 		ret = generic_handle_domain_irq(rockchip->irq_domain, hwirq);
530 		if (ret)
531 			dev_err(dev, "unexpected IRQ, INT%d\n", hwirq);
532 	}
533 
534 	chained_irq_exit(chip, desc);
535 }
536 
537 static int rockchip_pcie_setup_irq(struct rockchip_pcie *rockchip)
538 {
539 	int irq, err;
540 	struct device *dev = rockchip->dev;
541 	struct platform_device *pdev = to_platform_device(dev);
542 
543 	irq = platform_get_irq_byname(pdev, "sys");
544 	if (irq < 0)
545 		return irq;
546 
547 	err = devm_request_irq(dev, irq, rockchip_pcie_subsys_irq_handler,
548 			       IRQF_SHARED, "pcie-sys", rockchip);
549 	if (err) {
550 		dev_err(dev, "failed to request PCIe subsystem IRQ\n");
551 		return err;
552 	}
553 
554 	irq = platform_get_irq_byname(pdev, "legacy");
555 	if (irq < 0)
556 		return irq;
557 
558 	irq_set_chained_handler_and_data(irq,
559 					 rockchip_pcie_legacy_int_handler,
560 					 rockchip);
561 
562 	irq = platform_get_irq_byname(pdev, "client");
563 	if (irq < 0)
564 		return irq;
565 
566 	err = devm_request_irq(dev, irq, rockchip_pcie_client_irq_handler,
567 			       IRQF_SHARED, "pcie-client", rockchip);
568 	if (err) {
569 		dev_err(dev, "failed to request PCIe client IRQ\n");
570 		return err;
571 	}
572 
573 	return 0;
574 }
575 
576 /**
577  * rockchip_pcie_parse_host_dt - Parse Device Tree
578  * @rockchip: PCIe port information
579  *
580  * Return: '0' on success and error value on failure
581  */
582 static int rockchip_pcie_parse_host_dt(struct rockchip_pcie *rockchip)
583 {
584 	struct device *dev = rockchip->dev;
585 	int err;
586 
587 	err = rockchip_pcie_parse_dt(rockchip);
588 	if (err)
589 		return err;
590 
591 	rockchip->vpcie12v = devm_regulator_get_optional(dev, "vpcie12v");
592 	if (IS_ERR(rockchip->vpcie12v)) {
593 		if (PTR_ERR(rockchip->vpcie12v) != -ENODEV)
594 			return PTR_ERR(rockchip->vpcie12v);
595 		dev_info(dev, "no vpcie12v regulator found\n");
596 	}
597 
598 	rockchip->vpcie3v3 = devm_regulator_get_optional(dev, "vpcie3v3");
599 	if (IS_ERR(rockchip->vpcie3v3)) {
600 		if (PTR_ERR(rockchip->vpcie3v3) != -ENODEV)
601 			return PTR_ERR(rockchip->vpcie3v3);
602 		dev_info(dev, "no vpcie3v3 regulator found\n");
603 	}
604 
605 	rockchip->vpcie1v8 = devm_regulator_get(dev, "vpcie1v8");
606 	if (IS_ERR(rockchip->vpcie1v8))
607 		return PTR_ERR(rockchip->vpcie1v8);
608 
609 	rockchip->vpcie0v9 = devm_regulator_get(dev, "vpcie0v9");
610 	if (IS_ERR(rockchip->vpcie0v9))
611 		return PTR_ERR(rockchip->vpcie0v9);
612 
613 	return 0;
614 }
615 
616 static int rockchip_pcie_set_vpcie(struct rockchip_pcie *rockchip)
617 {
618 	struct device *dev = rockchip->dev;
619 	int err;
620 
621 	if (!IS_ERR(rockchip->vpcie12v)) {
622 		err = regulator_enable(rockchip->vpcie12v);
623 		if (err) {
624 			dev_err(dev, "fail to enable vpcie12v regulator\n");
625 			goto err_out;
626 		}
627 	}
628 
629 	if (!IS_ERR(rockchip->vpcie3v3)) {
630 		err = regulator_enable(rockchip->vpcie3v3);
631 		if (err) {
632 			dev_err(dev, "fail to enable vpcie3v3 regulator\n");
633 			goto err_disable_12v;
634 		}
635 	}
636 
637 	err = regulator_enable(rockchip->vpcie1v8);
638 	if (err) {
639 		dev_err(dev, "fail to enable vpcie1v8 regulator\n");
640 		goto err_disable_3v3;
641 	}
642 
643 	err = regulator_enable(rockchip->vpcie0v9);
644 	if (err) {
645 		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
646 		goto err_disable_1v8;
647 	}
648 
649 	return 0;
650 
651 err_disable_1v8:
652 	regulator_disable(rockchip->vpcie1v8);
653 err_disable_3v3:
654 	if (!IS_ERR(rockchip->vpcie3v3))
655 		regulator_disable(rockchip->vpcie3v3);
656 err_disable_12v:
657 	if (!IS_ERR(rockchip->vpcie12v))
658 		regulator_disable(rockchip->vpcie12v);
659 err_out:
660 	return err;
661 }
662 
663 static void rockchip_pcie_enable_interrupts(struct rockchip_pcie *rockchip)
664 {
665 	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) &
666 			    (~PCIE_CLIENT_INT_CLI), PCIE_CLIENT_INT_MASK);
667 	rockchip_pcie_write(rockchip, (u32)(~PCIE_CORE_INT),
668 			    PCIE_CORE_INT_MASK);
669 
670 	rockchip_pcie_enable_bw_int(rockchip);
671 }
672 
673 static int rockchip_pcie_intx_map(struct irq_domain *domain, unsigned int irq,
674 				  irq_hw_number_t hwirq)
675 {
676 	irq_set_chip_and_handler(irq, &dummy_irq_chip, handle_simple_irq);
677 	irq_set_chip_data(irq, domain->host_data);
678 
679 	return 0;
680 }
681 
682 static const struct irq_domain_ops intx_domain_ops = {
683 	.map = rockchip_pcie_intx_map,
684 };
685 
686 static int rockchip_pcie_init_irq_domain(struct rockchip_pcie *rockchip)
687 {
688 	struct device *dev = rockchip->dev;
689 	struct device_node *intc = of_get_next_child(dev->of_node, NULL);
690 
691 	if (!intc) {
692 		dev_err(dev, "missing child interrupt-controller node\n");
693 		return -EINVAL;
694 	}
695 
696 	rockchip->irq_domain = irq_domain_add_linear(intc, PCI_NUM_INTX,
697 						    &intx_domain_ops, rockchip);
698 	of_node_put(intc);
699 	if (!rockchip->irq_domain) {
700 		dev_err(dev, "failed to get a INTx IRQ domain\n");
701 		return -EINVAL;
702 	}
703 
704 	return 0;
705 }
706 
707 static int rockchip_pcie_prog_ob_atu(struct rockchip_pcie *rockchip,
708 				     int region_no, int type, u8 num_pass_bits,
709 				     u32 lower_addr, u32 upper_addr)
710 {
711 	u32 ob_addr_0;
712 	u32 ob_addr_1;
713 	u32 ob_desc_0;
714 	u32 aw_offset;
715 
716 	if (region_no >= MAX_AXI_WRAPPER_REGION_NUM)
717 		return -EINVAL;
718 	if (num_pass_bits + 1 < 8)
719 		return -EINVAL;
720 	if (num_pass_bits > 63)
721 		return -EINVAL;
722 	if (region_no == 0) {
723 		if (AXI_REGION_0_SIZE < (2ULL << num_pass_bits))
724 			return -EINVAL;
725 	}
726 	if (region_no != 0) {
727 		if (AXI_REGION_SIZE < (2ULL << num_pass_bits))
728 			return -EINVAL;
729 	}
730 
731 	aw_offset = (region_no << OB_REG_SIZE_SHIFT);
732 
733 	ob_addr_0 = num_pass_bits & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS;
734 	ob_addr_0 |= lower_addr & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR;
735 	ob_addr_1 = upper_addr;
736 	ob_desc_0 = (1 << 23 | type);
737 
738 	rockchip_pcie_write(rockchip, ob_addr_0,
739 			    PCIE_CORE_OB_REGION_ADDR0 + aw_offset);
740 	rockchip_pcie_write(rockchip, ob_addr_1,
741 			    PCIE_CORE_OB_REGION_ADDR1 + aw_offset);
742 	rockchip_pcie_write(rockchip, ob_desc_0,
743 			    PCIE_CORE_OB_REGION_DESC0 + aw_offset);
744 	rockchip_pcie_write(rockchip, 0,
745 			    PCIE_CORE_OB_REGION_DESC1 + aw_offset);
746 
747 	return 0;
748 }
749 
750 static int rockchip_pcie_prog_ib_atu(struct rockchip_pcie *rockchip,
751 				     int region_no, u8 num_pass_bits,
752 				     u32 lower_addr, u32 upper_addr)
753 {
754 	u32 ib_addr_0;
755 	u32 ib_addr_1;
756 	u32 aw_offset;
757 
758 	if (region_no > MAX_AXI_IB_ROOTPORT_REGION_NUM)
759 		return -EINVAL;
760 	if (num_pass_bits + 1 < MIN_AXI_ADDR_BITS_PASSED)
761 		return -EINVAL;
762 	if (num_pass_bits > 63)
763 		return -EINVAL;
764 
765 	aw_offset = (region_no << IB_ROOT_PORT_REG_SIZE_SHIFT);
766 
767 	ib_addr_0 = num_pass_bits & PCIE_CORE_IB_REGION_ADDR0_NUM_BITS;
768 	ib_addr_0 |= (lower_addr << 8) & PCIE_CORE_IB_REGION_ADDR0_LO_ADDR;
769 	ib_addr_1 = upper_addr;
770 
771 	rockchip_pcie_write(rockchip, ib_addr_0, PCIE_RP_IB_ADDR0 + aw_offset);
772 	rockchip_pcie_write(rockchip, ib_addr_1, PCIE_RP_IB_ADDR1 + aw_offset);
773 
774 	return 0;
775 }
776 
777 static int rockchip_pcie_cfg_atu(struct rockchip_pcie *rockchip)
778 {
779 	struct device *dev = rockchip->dev;
780 	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
781 	struct resource_entry *entry;
782 	u64 pci_addr, size;
783 	int offset;
784 	int err;
785 	int reg_no;
786 
787 	rockchip_pcie_cfg_configuration_accesses(rockchip,
788 						 AXI_WRAPPER_TYPE0_CFG);
789 	entry = resource_list_first_type(&bridge->windows, IORESOURCE_MEM);
790 	if (!entry)
791 		return -ENODEV;
792 
793 	size = resource_size(entry->res);
794 	pci_addr = entry->res->start - entry->offset;
795 	rockchip->msg_bus_addr = pci_addr;
796 
797 	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
798 		err = rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1,
799 						AXI_WRAPPER_MEM_WRITE,
800 						20 - 1,
801 						pci_addr + (reg_no << 20),
802 						0);
803 		if (err) {
804 			dev_err(dev, "program RC mem outbound ATU failed\n");
805 			return err;
806 		}
807 	}
808 
809 	err = rockchip_pcie_prog_ib_atu(rockchip, 2, 32 - 1, 0x0, 0);
810 	if (err) {
811 		dev_err(dev, "program RC mem inbound ATU failed\n");
812 		return err;
813 	}
814 
815 	entry = resource_list_first_type(&bridge->windows, IORESOURCE_IO);
816 	if (!entry)
817 		return -ENODEV;
818 
819 	/* store the register number offset to program RC io outbound ATU */
820 	offset = size >> 20;
821 
822 	size = resource_size(entry->res);
823 	pci_addr = entry->res->start - entry->offset;
824 
825 	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
826 		err = rockchip_pcie_prog_ob_atu(rockchip,
827 						reg_no + 1 + offset,
828 						AXI_WRAPPER_IO_WRITE,
829 						20 - 1,
830 						pci_addr + (reg_no << 20),
831 						0);
832 		if (err) {
833 			dev_err(dev, "program RC io outbound ATU failed\n");
834 			return err;
835 		}
836 	}
837 
838 	/* assign message regions */
839 	rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1 + offset,
840 				  AXI_WRAPPER_NOR_MSG,
841 				  20 - 1, 0, 0);
842 
843 	rockchip->msg_bus_addr += ((reg_no + offset) << 20);
844 	return err;
845 }
846 
847 static int rockchip_pcie_wait_l2(struct rockchip_pcie *rockchip)
848 {
849 	u32 value;
850 	int err;
851 
852 	/* send PME_TURN_OFF message */
853 	writel(0x0, rockchip->msg_region + PCIE_RC_SEND_PME_OFF);
854 
855 	/* read LTSSM and wait for falling into L2 link state */
856 	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_DEBUG_OUT_0,
857 				 value, PCIE_LINK_IS_L2(value), 20,
858 				 jiffies_to_usecs(5 * HZ));
859 	if (err) {
860 		dev_err(rockchip->dev, "PCIe link enter L2 timeout!\n");
861 		return err;
862 	}
863 
864 	return 0;
865 }
866 
867 static int rockchip_pcie_suspend_noirq(struct device *dev)
868 {
869 	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
870 	int ret;
871 
872 	/* disable core and cli int since we don't need to ack PME_ACK */
873 	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) |
874 			    PCIE_CLIENT_INT_CLI, PCIE_CLIENT_INT_MASK);
875 	rockchip_pcie_write(rockchip, (u32)PCIE_CORE_INT, PCIE_CORE_INT_MASK);
876 
877 	ret = rockchip_pcie_wait_l2(rockchip);
878 	if (ret) {
879 		rockchip_pcie_enable_interrupts(rockchip);
880 		return ret;
881 	}
882 
883 	rockchip_pcie_deinit_phys(rockchip);
884 
885 	rockchip_pcie_disable_clocks(rockchip);
886 
887 	regulator_disable(rockchip->vpcie0v9);
888 
889 	return ret;
890 }
891 
892 static int rockchip_pcie_resume_noirq(struct device *dev)
893 {
894 	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
895 	int err;
896 
897 	err = regulator_enable(rockchip->vpcie0v9);
898 	if (err) {
899 		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
900 		return err;
901 	}
902 
903 	err = rockchip_pcie_enable_clocks(rockchip);
904 	if (err)
905 		goto err_disable_0v9;
906 
907 	err = rockchip_pcie_host_init_port(rockchip);
908 	if (err)
909 		goto err_pcie_resume;
910 
911 	err = rockchip_pcie_cfg_atu(rockchip);
912 	if (err)
913 		goto err_err_deinit_port;
914 
915 	/* Need this to enter L1 again */
916 	rockchip_pcie_update_txcredit_mui(rockchip);
917 	rockchip_pcie_enable_interrupts(rockchip);
918 
919 	return 0;
920 
921 err_err_deinit_port:
922 	rockchip_pcie_deinit_phys(rockchip);
923 err_pcie_resume:
924 	rockchip_pcie_disable_clocks(rockchip);
925 err_disable_0v9:
926 	regulator_disable(rockchip->vpcie0v9);
927 	return err;
928 }
929 
930 static int rockchip_pcie_probe(struct platform_device *pdev)
931 {
932 	struct rockchip_pcie *rockchip;
933 	struct device *dev = &pdev->dev;
934 	struct pci_host_bridge *bridge;
935 	int err;
936 
937 	if (!dev->of_node)
938 		return -ENODEV;
939 
940 	bridge = devm_pci_alloc_host_bridge(dev, sizeof(*rockchip));
941 	if (!bridge)
942 		return -ENOMEM;
943 
944 	rockchip = pci_host_bridge_priv(bridge);
945 
946 	platform_set_drvdata(pdev, rockchip);
947 	rockchip->dev = dev;
948 	rockchip->is_rc = true;
949 
950 	err = rockchip_pcie_parse_host_dt(rockchip);
951 	if (err)
952 		return err;
953 
954 	err = rockchip_pcie_enable_clocks(rockchip);
955 	if (err)
956 		return err;
957 
958 	err = rockchip_pcie_set_vpcie(rockchip);
959 	if (err) {
960 		dev_err(dev, "failed to set vpcie regulator\n");
961 		goto err_set_vpcie;
962 	}
963 
964 	err = rockchip_pcie_host_init_port(rockchip);
965 	if (err)
966 		goto err_vpcie;
967 
968 	err = rockchip_pcie_init_irq_domain(rockchip);
969 	if (err < 0)
970 		goto err_deinit_port;
971 
972 	err = rockchip_pcie_cfg_atu(rockchip);
973 	if (err)
974 		goto err_remove_irq_domain;
975 
976 	rockchip->msg_region = devm_ioremap(dev, rockchip->msg_bus_addr, SZ_1M);
977 	if (!rockchip->msg_region) {
978 		err = -ENOMEM;
979 		goto err_remove_irq_domain;
980 	}
981 
982 	bridge->sysdata = rockchip;
983 	bridge->ops = &rockchip_pcie_ops;
984 
985 	err = rockchip_pcie_setup_irq(rockchip);
986 	if (err)
987 		goto err_remove_irq_domain;
988 
989 	rockchip_pcie_enable_interrupts(rockchip);
990 
991 	err = pci_host_probe(bridge);
992 	if (err < 0)
993 		goto err_remove_irq_domain;
994 
995 	return 0;
996 
997 err_remove_irq_domain:
998 	irq_domain_remove(rockchip->irq_domain);
999 err_deinit_port:
1000 	rockchip_pcie_deinit_phys(rockchip);
1001 err_vpcie:
1002 	if (!IS_ERR(rockchip->vpcie12v))
1003 		regulator_disable(rockchip->vpcie12v);
1004 	if (!IS_ERR(rockchip->vpcie3v3))
1005 		regulator_disable(rockchip->vpcie3v3);
1006 	regulator_disable(rockchip->vpcie1v8);
1007 	regulator_disable(rockchip->vpcie0v9);
1008 err_set_vpcie:
1009 	rockchip_pcie_disable_clocks(rockchip);
1010 	return err;
1011 }
1012 
1013 static int rockchip_pcie_remove(struct platform_device *pdev)
1014 {
1015 	struct device *dev = &pdev->dev;
1016 	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
1017 	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
1018 
1019 	pci_stop_root_bus(bridge->bus);
1020 	pci_remove_root_bus(bridge->bus);
1021 	irq_domain_remove(rockchip->irq_domain);
1022 
1023 	rockchip_pcie_deinit_phys(rockchip);
1024 
1025 	rockchip_pcie_disable_clocks(rockchip);
1026 
1027 	if (!IS_ERR(rockchip->vpcie12v))
1028 		regulator_disable(rockchip->vpcie12v);
1029 	if (!IS_ERR(rockchip->vpcie3v3))
1030 		regulator_disable(rockchip->vpcie3v3);
1031 	regulator_disable(rockchip->vpcie1v8);
1032 	regulator_disable(rockchip->vpcie0v9);
1033 
1034 	return 0;
1035 }
1036 
1037 static const struct dev_pm_ops rockchip_pcie_pm_ops = {
1038 	NOIRQ_SYSTEM_SLEEP_PM_OPS(rockchip_pcie_suspend_noirq,
1039 				  rockchip_pcie_resume_noirq)
1040 };
1041 
1042 static const struct of_device_id rockchip_pcie_of_match[] = {
1043 	{ .compatible = "rockchip,rk3399-pcie", },
1044 	{}
1045 };
1046 MODULE_DEVICE_TABLE(of, rockchip_pcie_of_match);
1047 
1048 static struct platform_driver rockchip_pcie_driver = {
1049 	.driver = {
1050 		.name = "rockchip-pcie",
1051 		.of_match_table = rockchip_pcie_of_match,
1052 		.pm = &rockchip_pcie_pm_ops,
1053 	},
1054 	.probe = rockchip_pcie_probe,
1055 	.remove = rockchip_pcie_remove,
1056 };
1057 module_platform_driver(rockchip_pcie_driver);
1058 
1059 MODULE_AUTHOR("Rockchip Inc");
1060 MODULE_DESCRIPTION("Rockchip AXI PCIe driver");
1061 MODULE_LICENSE("GPL v2");
1062