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