Linux Audio

Check our new training course

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