1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * PCIe host controller driver for Amlogic MESON SoCs
4  *
5  * Copyright (c) 2018 Amlogic, inc.
6  * Author: Yue Wang <yue.wang@amlogic.com>
7  */
8 
9 #include <linux/clk.h>
10 #include <linux/delay.h>
11 #include <linux/gpio/consumer.h>
12 #include <linux/of_device.h>
13 #include <linux/of_gpio.h>
14 #include <linux/pci.h>
15 #include <linux/platform_device.h>
16 #include <linux/reset.h>
17 #include <linux/resource.h>
18 #include <linux/types.h>
19 #include <linux/phy/phy.h>
20 #include <linux/module.h>
21 
22 #include "pcie-designware.h"
23 
24 #define to_meson_pcie(x) dev_get_drvdata((x)->dev)
25 
26 #define PCIE_CAP_MAX_PAYLOAD_SIZE(x)	((x) << 5)
27 #define PCIE_CAP_MAX_READ_REQ_SIZE(x)	((x) << 12)
28 
29 /* PCIe specific config registers */
30 #define PCIE_CFG0			0x0
31 #define APP_LTSSM_ENABLE		BIT(7)
32 
33 #define PCIE_CFG_STATUS12		0x30
34 #define IS_SMLH_LINK_UP(x)		((x) & (1 << 6))
35 #define IS_RDLH_LINK_UP(x)		((x) & (1 << 16))
36 #define IS_LTSSM_UP(x)			((((x) >> 10) & 0x1f) == 0x11)
37 
38 #define PCIE_CFG_STATUS17		0x44
39 #define PM_CURRENT_STATE(x)		(((x) >> 7) & 0x1)
40 
41 #define WAIT_LINKUP_TIMEOUT		4000
42 #define PORT_CLK_RATE			100000000UL
43 #define MAX_PAYLOAD_SIZE		256
44 #define MAX_READ_REQ_SIZE		256
45 #define PCIE_RESET_DELAY		500
46 #define PCIE_SHARED_RESET		1
47 #define PCIE_NORMAL_RESET		0
48 
49 enum pcie_data_rate {
50 	PCIE_GEN1,
51 	PCIE_GEN2,
52 	PCIE_GEN3,
53 	PCIE_GEN4
54 };
55 
56 struct meson_pcie_clk_res {
57 	struct clk *clk;
58 	struct clk *port_clk;
59 	struct clk *general_clk;
60 };
61 
62 struct meson_pcie_rc_reset {
63 	struct reset_control *port;
64 	struct reset_control *apb;
65 };
66 
67 struct meson_pcie {
68 	struct dw_pcie pci;
69 	void __iomem *cfg_base;
70 	struct meson_pcie_clk_res clk_res;
71 	struct meson_pcie_rc_reset mrst;
72 	struct gpio_desc *reset_gpio;
73 	struct phy *phy;
74 };
75 
76 static struct reset_control *meson_pcie_get_reset(struct meson_pcie *mp,
77 						  const char *id,
78 						  u32 reset_type)
79 {
80 	struct device *dev = mp->pci.dev;
81 	struct reset_control *reset;
82 
83 	if (reset_type == PCIE_SHARED_RESET)
84 		reset = devm_reset_control_get_shared(dev, id);
85 	else
86 		reset = devm_reset_control_get(dev, id);
87 
88 	return reset;
89 }
90 
91 static int meson_pcie_get_resets(struct meson_pcie *mp)
92 {
93 	struct meson_pcie_rc_reset *mrst = &mp->mrst;
94 
95 	mrst->port = meson_pcie_get_reset(mp, "port", PCIE_NORMAL_RESET);
96 	if (IS_ERR(mrst->port))
97 		return PTR_ERR(mrst->port);
98 	reset_control_deassert(mrst->port);
99 
100 	mrst->apb = meson_pcie_get_reset(mp, "apb", PCIE_SHARED_RESET);
101 	if (IS_ERR(mrst->apb))
102 		return PTR_ERR(mrst->apb);
103 	reset_control_deassert(mrst->apb);
104 
105 	return 0;
106 }
107 
108 static int meson_pcie_get_mems(struct platform_device *pdev,
109 			       struct meson_pcie *mp)
110 {
111 	struct dw_pcie *pci = &mp->pci;
112 
113 	pci->dbi_base = devm_platform_ioremap_resource_byname(pdev, "elbi");
114 	if (IS_ERR(pci->dbi_base))
115 		return PTR_ERR(pci->dbi_base);
116 
117 	mp->cfg_base = devm_platform_ioremap_resource_byname(pdev, "cfg");
118 	if (IS_ERR(mp->cfg_base))
119 		return PTR_ERR(mp->cfg_base);
120 
121 	return 0;
122 }
123 
124 static int meson_pcie_power_on(struct meson_pcie *mp)
125 {
126 	int ret = 0;
127 
128 	ret = phy_init(mp->phy);
129 	if (ret)
130 		return ret;
131 
132 	ret = phy_power_on(mp->phy);
133 	if (ret) {
134 		phy_exit(mp->phy);
135 		return ret;
136 	}
137 
138 	return 0;
139 }
140 
141 static void meson_pcie_power_off(struct meson_pcie *mp)
142 {
143 	phy_power_off(mp->phy);
144 	phy_exit(mp->phy);
145 }
146 
147 static int meson_pcie_reset(struct meson_pcie *mp)
148 {
149 	struct meson_pcie_rc_reset *mrst = &mp->mrst;
150 	int ret = 0;
151 
152 	ret = phy_reset(mp->phy);
153 	if (ret)
154 		return ret;
155 
156 	reset_control_assert(mrst->port);
157 	reset_control_assert(mrst->apb);
158 	udelay(PCIE_RESET_DELAY);
159 	reset_control_deassert(mrst->port);
160 	reset_control_deassert(mrst->apb);
161 	udelay(PCIE_RESET_DELAY);
162 
163 	return 0;
164 }
165 
166 static inline struct clk *meson_pcie_probe_clock(struct device *dev,
167 						 const char *id, u64 rate)
168 {
169 	struct clk *clk;
170 	int ret;
171 
172 	clk = devm_clk_get(dev, id);
173 	if (IS_ERR(clk))
174 		return clk;
175 
176 	if (rate) {
177 		ret = clk_set_rate(clk, rate);
178 		if (ret) {
179 			dev_err(dev, "set clk rate failed, ret = %d\n", ret);
180 			return ERR_PTR(ret);
181 		}
182 	}
183 
184 	ret = clk_prepare_enable(clk);
185 	if (ret) {
186 		dev_err(dev, "couldn't enable clk\n");
187 		return ERR_PTR(ret);
188 	}
189 
190 	devm_add_action_or_reset(dev,
191 				 (void (*) (void *))clk_disable_unprepare,
192 				 clk);
193 
194 	return clk;
195 }
196 
197 static int meson_pcie_probe_clocks(struct meson_pcie *mp)
198 {
199 	struct device *dev = mp->pci.dev;
200 	struct meson_pcie_clk_res *res = &mp->clk_res;
201 
202 	res->port_clk = meson_pcie_probe_clock(dev, "port", PORT_CLK_RATE);
203 	if (IS_ERR(res->port_clk))
204 		return PTR_ERR(res->port_clk);
205 
206 	res->general_clk = meson_pcie_probe_clock(dev, "general", 0);
207 	if (IS_ERR(res->general_clk))
208 		return PTR_ERR(res->general_clk);
209 
210 	res->clk = meson_pcie_probe_clock(dev, "pclk", 0);
211 	if (IS_ERR(res->clk))
212 		return PTR_ERR(res->clk);
213 
214 	return 0;
215 }
216 
217 static inline u32 meson_cfg_readl(struct meson_pcie *mp, u32 reg)
218 {
219 	return readl(mp->cfg_base + reg);
220 }
221 
222 static inline void meson_cfg_writel(struct meson_pcie *mp, u32 val, u32 reg)
223 {
224 	writel(val, mp->cfg_base + reg);
225 }
226 
227 static void meson_pcie_assert_reset(struct meson_pcie *mp)
228 {
229 	gpiod_set_value_cansleep(mp->reset_gpio, 1);
230 	udelay(500);
231 	gpiod_set_value_cansleep(mp->reset_gpio, 0);
232 }
233 
234 static void meson_pcie_ltssm_enable(struct meson_pcie *mp)
235 {
236 	u32 val;
237 
238 	val = meson_cfg_readl(mp, PCIE_CFG0);
239 	val |= APP_LTSSM_ENABLE;
240 	meson_cfg_writel(mp, val, PCIE_CFG0);
241 }
242 
243 static int meson_size_to_payload(struct meson_pcie *mp, int size)
244 {
245 	struct device *dev = mp->pci.dev;
246 
247 	/*
248 	 * dwc supports 2^(val+7) payload size, which val is 0~5 default to 1.
249 	 * So if input size is not 2^order alignment or less than 2^7 or bigger
250 	 * than 2^12, just set to default size 2^(1+7).
251 	 */
252 	if (!is_power_of_2(size) || size < 128 || size > 4096) {
253 		dev_warn(dev, "payload size %d, set to default 256\n", size);
254 		return 1;
255 	}
256 
257 	return fls(size) - 8;
258 }
259 
260 static void meson_set_max_payload(struct meson_pcie *mp, int size)
261 {
262 	struct dw_pcie *pci = &mp->pci;
263 	u32 val;
264 	u16 offset = dw_pcie_find_capability(pci, PCI_CAP_ID_EXP);
265 	int max_payload_size = meson_size_to_payload(mp, size);
266 
267 	val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_DEVCTL);
268 	val &= ~PCI_EXP_DEVCTL_PAYLOAD;
269 	dw_pcie_writel_dbi(pci, offset + PCI_EXP_DEVCTL, val);
270 
271 	val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_DEVCTL);
272 	val |= PCIE_CAP_MAX_PAYLOAD_SIZE(max_payload_size);
273 	dw_pcie_writel_dbi(pci, offset + PCI_EXP_DEVCTL, val);
274 }
275 
276 static void meson_set_max_rd_req_size(struct meson_pcie *mp, int size)
277 {
278 	struct dw_pcie *pci = &mp->pci;
279 	u32 val;
280 	u16 offset = dw_pcie_find_capability(pci, PCI_CAP_ID_EXP);
281 	int max_rd_req_size = meson_size_to_payload(mp, size);
282 
283 	val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_DEVCTL);
284 	val &= ~PCI_EXP_DEVCTL_READRQ;
285 	dw_pcie_writel_dbi(pci, offset + PCI_EXP_DEVCTL, val);
286 
287 	val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_DEVCTL);
288 	val |= PCIE_CAP_MAX_READ_REQ_SIZE(max_rd_req_size);
289 	dw_pcie_writel_dbi(pci, offset + PCI_EXP_DEVCTL, val);
290 }
291 
292 static int meson_pcie_start_link(struct dw_pcie *pci)
293 {
294 	struct meson_pcie *mp = to_meson_pcie(pci);
295 
296 	meson_pcie_ltssm_enable(mp);
297 	meson_pcie_assert_reset(mp);
298 
299 	return 0;
300 }
301 
302 static int meson_pcie_rd_own_conf(struct pci_bus *bus, u32 devfn,
303 				  int where, int size, u32 *val)
304 {
305 	int ret;
306 
307 	ret = pci_generic_config_read(bus, devfn, where, size, val);
308 	if (ret != PCIBIOS_SUCCESSFUL)
309 		return ret;
310 
311 	/*
312 	 * There is a bug in the MESON AXG PCIe controller whereby software
313 	 * cannot program the PCI_CLASS_DEVICE register, so we must fabricate
314 	 * the return value in the config accessors.
315 	 */
316 	if ((where & ~3) == PCI_CLASS_REVISION) {
317 		if (size <= 2)
318 			*val = (*val & ((1 << (size * 8)) - 1)) << (8 * (where & 3));
319 		*val &= ~0xffffff00;
320 		*val |= PCI_CLASS_BRIDGE_PCI_NORMAL << 8;
321 		if (size <= 2)
322 			*val = (*val >> (8 * (where & 3))) & ((1 << (size * 8)) - 1);
323 	}
324 
325 	return PCIBIOS_SUCCESSFUL;
326 }
327 
328 static struct pci_ops meson_pci_ops = {
329 	.map_bus = dw_pcie_own_conf_map_bus,
330 	.read = meson_pcie_rd_own_conf,
331 	.write = pci_generic_config_write,
332 };
333 
334 static int meson_pcie_link_up(struct dw_pcie *pci)
335 {
336 	struct meson_pcie *mp = to_meson_pcie(pci);
337 	struct device *dev = pci->dev;
338 	u32 speed_okay = 0;
339 	u32 cnt = 0;
340 	u32 state12, state17, smlh_up, ltssm_up, rdlh_up;
341 
342 	do {
343 		state12 = meson_cfg_readl(mp, PCIE_CFG_STATUS12);
344 		state17 = meson_cfg_readl(mp, PCIE_CFG_STATUS17);
345 		smlh_up = IS_SMLH_LINK_UP(state12);
346 		rdlh_up = IS_RDLH_LINK_UP(state12);
347 		ltssm_up = IS_LTSSM_UP(state12);
348 
349 		if (PM_CURRENT_STATE(state17) < PCIE_GEN3)
350 			speed_okay = 1;
351 
352 		if (smlh_up)
353 			dev_dbg(dev, "smlh_link_up is on\n");
354 		if (rdlh_up)
355 			dev_dbg(dev, "rdlh_link_up is on\n");
356 		if (ltssm_up)
357 			dev_dbg(dev, "ltssm_up is on\n");
358 		if (speed_okay)
359 			dev_dbg(dev, "speed_okay\n");
360 
361 		if (smlh_up && rdlh_up && ltssm_up && speed_okay)
362 			return 1;
363 
364 		cnt++;
365 
366 		udelay(10);
367 	} while (cnt < WAIT_LINKUP_TIMEOUT);
368 
369 	dev_err(dev, "error: wait linkup timeout\n");
370 	return 0;
371 }
372 
373 static int meson_pcie_host_init(struct pcie_port *pp)
374 {
375 	struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
376 	struct meson_pcie *mp = to_meson_pcie(pci);
377 
378 	pp->bridge->ops = &meson_pci_ops;
379 
380 	meson_set_max_payload(mp, MAX_PAYLOAD_SIZE);
381 	meson_set_max_rd_req_size(mp, MAX_READ_REQ_SIZE);
382 
383 	return 0;
384 }
385 
386 static const struct dw_pcie_host_ops meson_pcie_host_ops = {
387 	.host_init = meson_pcie_host_init,
388 };
389 
390 static const struct dw_pcie_ops dw_pcie_ops = {
391 	.link_up = meson_pcie_link_up,
392 	.start_link = meson_pcie_start_link,
393 };
394 
395 static int meson_pcie_probe(struct platform_device *pdev)
396 {
397 	struct device *dev = &pdev->dev;
398 	struct dw_pcie *pci;
399 	struct meson_pcie *mp;
400 	int ret;
401 
402 	mp = devm_kzalloc(dev, sizeof(*mp), GFP_KERNEL);
403 	if (!mp)
404 		return -ENOMEM;
405 
406 	pci = &mp->pci;
407 	pci->dev = dev;
408 	pci->ops = &dw_pcie_ops;
409 	pci->pp.ops = &meson_pcie_host_ops;
410 	pci->num_lanes = 1;
411 
412 	mp->phy = devm_phy_get(dev, "pcie");
413 	if (IS_ERR(mp->phy)) {
414 		dev_err(dev, "get phy failed, %ld\n", PTR_ERR(mp->phy));
415 		return PTR_ERR(mp->phy);
416 	}
417 
418 	mp->reset_gpio = devm_gpiod_get(dev, "reset", GPIOD_OUT_LOW);
419 	if (IS_ERR(mp->reset_gpio)) {
420 		dev_err(dev, "get reset gpio failed\n");
421 		return PTR_ERR(mp->reset_gpio);
422 	}
423 
424 	ret = meson_pcie_get_resets(mp);
425 	if (ret) {
426 		dev_err(dev, "get reset resource failed, %d\n", ret);
427 		return ret;
428 	}
429 
430 	ret = meson_pcie_get_mems(pdev, mp);
431 	if (ret) {
432 		dev_err(dev, "get memory resource failed, %d\n", ret);
433 		return ret;
434 	}
435 
436 	ret = meson_pcie_power_on(mp);
437 	if (ret) {
438 		dev_err(dev, "phy power on failed, %d\n", ret);
439 		return ret;
440 	}
441 
442 	ret = meson_pcie_reset(mp);
443 	if (ret) {
444 		dev_err(dev, "reset failed, %d\n", ret);
445 		goto err_phy;
446 	}
447 
448 	ret = meson_pcie_probe_clocks(mp);
449 	if (ret) {
450 		dev_err(dev, "init clock resources failed, %d\n", ret);
451 		goto err_phy;
452 	}
453 
454 	platform_set_drvdata(pdev, mp);
455 
456 	ret = dw_pcie_host_init(&pci->pp);
457 	if (ret < 0) {
458 		dev_err(dev, "Add PCIe port failed, %d\n", ret);
459 		goto err_phy;
460 	}
461 
462 	return 0;
463 
464 err_phy:
465 	meson_pcie_power_off(mp);
466 	return ret;
467 }
468 
469 static const struct of_device_id meson_pcie_of_match[] = {
470 	{
471 		.compatible = "amlogic,axg-pcie",
472 	},
473 	{
474 		.compatible = "amlogic,g12a-pcie",
475 	},
476 	{},
477 };
478 MODULE_DEVICE_TABLE(of, meson_pcie_of_match);
479 
480 static struct platform_driver meson_pcie_driver = {
481 	.probe = meson_pcie_probe,
482 	.driver = {
483 		.name = "meson-pcie",
484 		.of_match_table = meson_pcie_of_match,
485 	},
486 };
487 
488 module_platform_driver(meson_pcie_driver);
489 
490 MODULE_AUTHOR("Yue Wang <yue.wang@amlogic.com>");
491 MODULE_DESCRIPTION("Amlogic PCIe Controller driver");
492 MODULE_LICENSE("GPL v2");
493