xref: /linux/drivers/memory/mtk-smi.c (revision 4f77c3462308c62ffe7129cc18b9ac937f44b5a5)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2015-2016 MediaTek Inc.
4  * Author: Yong Wu <yong.wu@mediatek.com>
5  */
6 #include <linux/arm-smccc.h>
7 #include <linux/clk.h>
8 #include <linux/component.h>
9 #include <linux/device.h>
10 #include <linux/err.h>
11 #include <linux/io.h>
12 #include <linux/iopoll.h>
13 #include <linux/module.h>
14 #include <linux/of.h>
15 #include <linux/of_platform.h>
16 #include <linux/platform_device.h>
17 #include <linux/pm_runtime.h>
18 #include <linux/soc/mediatek/mtk_sip_svc.h>
19 #include <soc/mediatek/smi.h>
20 #include <dt-bindings/memory/mt2701-larb-port.h>
21 #include <dt-bindings/memory/mtk-memory-port.h>
22 
23 /* SMI COMMON */
24 #define SMI_L1LEN			0x100
25 
26 #define SMI_L1_ARB			0x200
27 #define SMI_BUS_SEL			0x220
28 #define SMI_BUS_LARB_SHIFT(larbid)	((larbid) << 1)
29 /* All are MMU0 defaultly. Only specialize mmu1 here. */
30 #define F_MMU1_LARB(larbid)		(0x1 << SMI_BUS_LARB_SHIFT(larbid))
31 
32 #define SMI_READ_FIFO_TH		0x230
33 #define SMI_M4U_TH			0x234
34 #define SMI_FIFO_TH1			0x238
35 #define SMI_FIFO_TH2			0x23c
36 #define SMI_DCM				0x300
37 #define SMI_DUMMY			0x444
38 
39 /* SMI LARB */
40 #define SMI_LARB_SLP_CON                0xc
41 #define SLP_PROT_EN                     BIT(0)
42 #define SLP_PROT_RDY                    BIT(16)
43 
44 #define SMI_LARB_CMD_THRT_CON		0x24
45 #define SMI_LARB_THRT_RD_NU_LMT_MSK	GENMASK(7, 4)
46 #define SMI_LARB_THRT_RD_NU_LMT		(5 << 4)
47 
48 #define SMI_LARB_SW_FLAG		0x40
49 #define SMI_LARB_SW_FLAG_1		0x1
50 
51 #define SMI_LARB_OSTDL_PORT		0x200
52 #define SMI_LARB_OSTDL_PORTx(id)	(SMI_LARB_OSTDL_PORT + (((id) & 0x1f) << 2))
53 
54 /* Below are about mmu enable registers, they are different in SoCs */
55 /* gen1: mt2701 */
56 #define REG_SMI_SECUR_CON_BASE		0x5c0
57 
58 /* every register control 8 port, register offset 0x4 */
59 #define REG_SMI_SECUR_CON_OFFSET(id)	(((id) >> 3) << 2)
60 #define REG_SMI_SECUR_CON_ADDR(id)	\
61 	(REG_SMI_SECUR_CON_BASE + REG_SMI_SECUR_CON_OFFSET(id))
62 
63 /*
64  * every port have 4 bit to control, bit[port + 3] control virtual or physical,
65  * bit[port + 2 : port + 1] control the domain, bit[port] control the security
66  * or non-security.
67  */
68 #define SMI_SECUR_CON_VAL_MSK(id)	(~(0xf << (((id) & 0x7) << 2)))
69 #define SMI_SECUR_CON_VAL_VIRT(id)	BIT((((id) & 0x7) << 2) + 3)
70 /* mt2701 domain should be set to 3 */
71 #define SMI_SECUR_CON_VAL_DOMAIN(id)	(0x3 << ((((id) & 0x7) << 2) + 1))
72 
73 /* gen2: */
74 /* mt8167 */
75 #define MT8167_SMI_LARB_MMU_EN		0xfc0
76 
77 /* mt8173 */
78 #define MT8173_SMI_LARB_MMU_EN		0xf00
79 
80 /* general */
81 #define SMI_LARB_NONSEC_CON(id)		(0x380 + ((id) * 4))
82 #define F_MMU_EN			BIT(0)
83 #define BANK_SEL(id)			({		\
84 	u32 _id = (id) & 0x3;				\
85 	(_id << 8 | _id << 10 | _id << 12 | _id << 14);	\
86 })
87 
88 #define SMI_COMMON_INIT_REGS_NR		6
89 #define SMI_LARB_PORT_NR_MAX		32
90 
91 #define MTK_SMI_FLAG_THRT_UPDATE	BIT(0)
92 #define MTK_SMI_FLAG_SW_FLAG		BIT(1)
93 #define MTK_SMI_FLAG_SLEEP_CTL		BIT(2)
94 #define MTK_SMI_FLAG_CFG_PORT_SEC_CTL	BIT(3)
95 #define MTK_SMI_CAPS(flags, _x)		(!!((flags) & (_x)))
96 
97 struct mtk_smi_reg_pair {
98 	unsigned int		offset;
99 	u32			value;
100 };
101 
102 enum mtk_smi_type {
103 	MTK_SMI_GEN1,
104 	MTK_SMI_GEN2,		/* gen2 smi common */
105 	MTK_SMI_GEN2_SUB_COMM,	/* gen2 smi sub common */
106 };
107 
108 /* larbs: Require apb/smi clocks while gals is optional. */
109 static const char * const mtk_smi_larb_clks[] = {"apb", "smi", "gals"};
110 #define MTK_SMI_LARB_REQ_CLK_NR		2
111 #define MTK_SMI_LARB_OPT_CLK_NR		1
112 
113 /*
114  * common: Require these four clocks in has_gals case. Otherwise, only apb/smi are required.
115  * sub common: Require apb/smi/gals0 clocks in has_gals case. Otherwise, only apb/smi are required.
116  */
117 static const char * const mtk_smi_common_clks[] = {"apb", "smi", "gals0", "gals1"};
118 #define MTK_SMI_CLK_NR_MAX		ARRAY_SIZE(mtk_smi_common_clks)
119 #define MTK_SMI_COM_REQ_CLK_NR		2
120 #define MTK_SMI_COM_GALS_REQ_CLK_NR	MTK_SMI_CLK_NR_MAX
121 #define MTK_SMI_SUB_COM_GALS_REQ_CLK_NR 3
122 
123 struct mtk_smi_common_plat {
124 	enum mtk_smi_type	type;
125 	bool			has_gals;
126 	u32			bus_sel; /* Balance some larbs to enter mmu0 or mmu1 */
127 
128 	const struct mtk_smi_reg_pair	*init;
129 };
130 
131 struct mtk_smi_larb_gen {
132 	int port_in_larb[MTK_LARB_NR_MAX + 1];
133 	int				(*config_port)(struct device *dev);
134 	unsigned int			larb_direct_to_common_mask;
135 	unsigned int			flags_general;
136 	const u8			(*ostd)[SMI_LARB_PORT_NR_MAX];
137 };
138 
139 struct mtk_smi {
140 	struct device			*dev;
141 	unsigned int			clk_num;
142 	struct clk_bulk_data		clks[MTK_SMI_CLK_NR_MAX];
143 	struct clk			*clk_async; /*only needed by mt2701*/
144 	union {
145 		void __iomem		*smi_ao_base; /* only for gen1 */
146 		void __iomem		*base;	      /* only for gen2 */
147 	};
148 	struct device			*smi_common_dev; /* for sub common */
149 	const struct mtk_smi_common_plat *plat;
150 };
151 
152 struct mtk_smi_larb { /* larb: local arbiter */
153 	struct mtk_smi			smi;
154 	void __iomem			*base;
155 	struct device			*smi_common_dev; /* common or sub-common dev */
156 	const struct mtk_smi_larb_gen	*larb_gen;
157 	int				larbid;
158 	u32				*mmu;
159 	unsigned char			*bank;
160 };
161 
162 static int
163 mtk_smi_larb_bind(struct device *dev, struct device *master, void *data)
164 {
165 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
166 	struct mtk_smi_larb_iommu *larb_mmu = data;
167 	unsigned int         i;
168 
169 	for (i = 0; i < MTK_LARB_NR_MAX; i++) {
170 		if (dev == larb_mmu[i].dev) {
171 			larb->larbid = i;
172 			larb->mmu = &larb_mmu[i].mmu;
173 			larb->bank = larb_mmu[i].bank;
174 			return 0;
175 		}
176 	}
177 	return -ENODEV;
178 }
179 
180 static void
181 mtk_smi_larb_unbind(struct device *dev, struct device *master, void *data)
182 {
183 	/* Do nothing as the iommu is always enabled. */
184 }
185 
186 static const struct component_ops mtk_smi_larb_component_ops = {
187 	.bind = mtk_smi_larb_bind,
188 	.unbind = mtk_smi_larb_unbind,
189 };
190 
191 static int mtk_smi_larb_config_port_gen1(struct device *dev)
192 {
193 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
194 	const struct mtk_smi_larb_gen *larb_gen = larb->larb_gen;
195 	struct mtk_smi *common = dev_get_drvdata(larb->smi_common_dev);
196 	int i, m4u_port_id, larb_port_num;
197 	u32 sec_con_val, reg_val;
198 
199 	m4u_port_id = larb_gen->port_in_larb[larb->larbid];
200 	larb_port_num = larb_gen->port_in_larb[larb->larbid + 1]
201 			- larb_gen->port_in_larb[larb->larbid];
202 
203 	for (i = 0; i < larb_port_num; i++, m4u_port_id++) {
204 		if (*larb->mmu & BIT(i)) {
205 			/* bit[port + 3] controls the virtual or physical */
206 			sec_con_val = SMI_SECUR_CON_VAL_VIRT(m4u_port_id);
207 		} else {
208 			/* do not need to enable m4u for this port */
209 			continue;
210 		}
211 		reg_val = readl(common->smi_ao_base
212 			+ REG_SMI_SECUR_CON_ADDR(m4u_port_id));
213 		reg_val &= SMI_SECUR_CON_VAL_MSK(m4u_port_id);
214 		reg_val |= sec_con_val;
215 		reg_val |= SMI_SECUR_CON_VAL_DOMAIN(m4u_port_id);
216 		writel(reg_val,
217 			common->smi_ao_base
218 			+ REG_SMI_SECUR_CON_ADDR(m4u_port_id));
219 	}
220 	return 0;
221 }
222 
223 static int mtk_smi_larb_config_port_mt8167(struct device *dev)
224 {
225 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
226 
227 	writel(*larb->mmu, larb->base + MT8167_SMI_LARB_MMU_EN);
228 	return 0;
229 }
230 
231 static int mtk_smi_larb_config_port_mt8173(struct device *dev)
232 {
233 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
234 
235 	writel(*larb->mmu, larb->base + MT8173_SMI_LARB_MMU_EN);
236 	return 0;
237 }
238 
239 static int mtk_smi_larb_config_port_gen2_general(struct device *dev)
240 {
241 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
242 	u32 reg, flags_general = larb->larb_gen->flags_general;
243 	const u8 *larbostd = larb->larb_gen->ostd ? larb->larb_gen->ostd[larb->larbid] : NULL;
244 	struct arm_smccc_res res;
245 	int i;
246 
247 	if (BIT(larb->larbid) & larb->larb_gen->larb_direct_to_common_mask)
248 		return 0;
249 
250 	if (MTK_SMI_CAPS(flags_general, MTK_SMI_FLAG_THRT_UPDATE)) {
251 		reg = readl_relaxed(larb->base + SMI_LARB_CMD_THRT_CON);
252 		reg &= ~SMI_LARB_THRT_RD_NU_LMT_MSK;
253 		reg |= SMI_LARB_THRT_RD_NU_LMT;
254 		writel_relaxed(reg, larb->base + SMI_LARB_CMD_THRT_CON);
255 	}
256 
257 	if (MTK_SMI_CAPS(flags_general, MTK_SMI_FLAG_SW_FLAG))
258 		writel_relaxed(SMI_LARB_SW_FLAG_1, larb->base + SMI_LARB_SW_FLAG);
259 
260 	for (i = 0; i < SMI_LARB_PORT_NR_MAX && larbostd && !!larbostd[i]; i++)
261 		writel_relaxed(larbostd[i], larb->base + SMI_LARB_OSTDL_PORTx(i));
262 
263 	/*
264 	 * When mmu_en bits are in security world, the bank_sel still is in the
265 	 * LARB_NONSEC_CON below. And the mmu_en bits of LARB_NONSEC_CON have no
266 	 * effect in this case.
267 	 */
268 	if (MTK_SMI_CAPS(flags_general, MTK_SMI_FLAG_CFG_PORT_SEC_CTL)) {
269 		arm_smccc_smc(MTK_SIP_KERNEL_IOMMU_CONTROL, IOMMU_ATF_CMD_CONFIG_SMI_LARB,
270 			      larb->larbid, *larb->mmu, 0, 0, 0, 0, &res);
271 		if (res.a0 != 0) {
272 			dev_err(dev, "Enable iommu fail, ret %ld\n", res.a0);
273 			return -EINVAL;
274 		}
275 	}
276 
277 	for_each_set_bit(i, (unsigned long *)larb->mmu, 32) {
278 		reg = readl_relaxed(larb->base + SMI_LARB_NONSEC_CON(i));
279 		reg |= F_MMU_EN;
280 		reg |= BANK_SEL(larb->bank[i]);
281 		writel(reg, larb->base + SMI_LARB_NONSEC_CON(i));
282 	}
283 	return 0;
284 }
285 
286 static const u8 mtk_smi_larb_mt8188_ostd[][SMI_LARB_PORT_NR_MAX] = {
287 	[0] = {0x02, 0x18, 0x22, 0x22, 0x01, 0x02, 0x0a,},
288 	[1] = {0x12, 0x02, 0x14, 0x14, 0x01, 0x18, 0x0a,},
289 	[2] = {0x12, 0x12, 0x12, 0x12, 0x0a,},
290 	[3] = {0x12, 0x12, 0x12, 0x12, 0x28, 0x28, 0x0a,},
291 	[4] = {0x06, 0x01, 0x17, 0x06, 0x0a, 0x07, 0x07,},
292 	[5] = {0x02, 0x01, 0x04, 0x02, 0x06, 0x01, 0x06, 0x0a,},
293 	[6] = {0x06, 0x01, 0x06, 0x0a,},
294 	[7] = {0x0c, 0x0c, 0x12,},
295 	[8] = {0x0c, 0x01, 0x0a, 0x05, 0x02, 0x03, 0x01, 0x01, 0x14, 0x14,
296 	       0x0a, 0x14, 0x1e, 0x01, 0x0c, 0x0a, 0x05, 0x02, 0x02, 0x05,
297 	       0x03, 0x01, 0x1e, 0x01, 0x05,},
298 	[9] = {0x1e, 0x01, 0x0a, 0x0a, 0x01, 0x01, 0x03, 0x1e, 0x1e, 0x10,
299 	       0x07, 0x01, 0x0a, 0x06, 0x03, 0x03, 0x0e, 0x01, 0x04, 0x28,},
300 	[10] = {0x03, 0x20, 0x01, 0x20, 0x01, 0x01, 0x14, 0x0a, 0x0a, 0x0c,
301 		0x0a, 0x05, 0x02, 0x03, 0x02, 0x14, 0x0a, 0x0a, 0x14, 0x14,
302 		0x14, 0x01, 0x01, 0x14, 0x1e, 0x01, 0x05, 0x03, 0x02, 0x28,},
303 	[11] = {0x03, 0x20, 0x01, 0x20, 0x01, 0x01, 0x14, 0x0a, 0x0a, 0x0c,
304 		0x0a, 0x05, 0x02, 0x03, 0x02, 0x14, 0x0a, 0x0a, 0x14, 0x14,
305 		0x14, 0x01, 0x01, 0x14, 0x1e, 0x01, 0x05, 0x03, 0x02, 0x28,},
306 	[12] = {0x03, 0x20, 0x01, 0x20, 0x01, 0x01, 0x14, 0x0a, 0x0a, 0x0c,
307 		0x0a, 0x05, 0x02, 0x03, 0x02, 0x14, 0x0a, 0x0a, 0x14, 0x14,
308 		0x14, 0x01, 0x01, 0x14, 0x1e, 0x01, 0x05, 0x03, 0x02, 0x28,},
309 	[13] = {0x07, 0x02, 0x04, 0x02, 0x05, 0x05, 0x05, 0x05, 0x05, 0x05,
310 		0x07, 0x02, 0x04, 0x02, 0x05, 0x05,},
311 	[14] = {0x02, 0x02, 0x0c, 0x0c, 0x0c, 0x0c, 0x01, 0x01, 0x02, 0x02,
312 		0x02, 0x02, 0x0c, 0x0c, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
313 		0x02, 0x02, 0x01, 0x01,},
314 	[15] = {0x0c, 0x0c, 0x02, 0x02, 0x02, 0x02, 0x01, 0x01, 0x0c, 0x0c,
315 		0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x01, 0x02,
316 		0x0c, 0x01, 0x01,},
317 	[16] = {0x28, 0x28, 0x03, 0x01, 0x01, 0x03, 0x14, 0x14, 0x0a, 0x0d,
318 		0x03, 0x05, 0x0e, 0x01, 0x01, 0x05, 0x06, 0x0d, 0x01,},
319 	[17] = {0x28, 0x02, 0x02, 0x12, 0x02, 0x12, 0x10, 0x02, 0x02, 0x0a,
320 		0x12, 0x02, 0x02, 0x0a, 0x16, 0x02, 0x04,},
321 	[18] = {0x28, 0x02, 0x02, 0x12, 0x02, 0x12, 0x10, 0x02, 0x02, 0x0a,
322 		0x12, 0x02, 0x02, 0x0a, 0x16, 0x02, 0x04,},
323 	[19] = {0x1a, 0x0e, 0x0a, 0x0a, 0x0c, 0x0e, 0x10,},
324 	[20] = {0x1a, 0x0e, 0x0a, 0x0a, 0x0c, 0x0e, 0x10,},
325 	[21] = {0x01, 0x04, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04, 0x04, 0x01,
326 		0x01, 0x01, 0x04, 0x0a, 0x06, 0x01, 0x01, 0x01, 0x0a, 0x06,
327 		0x01, 0x01, 0x05, 0x03, 0x03, 0x04, 0x01,},
328 	[22] = {0x28, 0x19, 0x0c, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04,
329 		0x01,},
330 	[23] = {0x01, 0x01, 0x04, 0x01, 0x01, 0x01, 0x18, 0x01, 0x01,},
331 	[24] = {0x12, 0x06, 0x12, 0x06,},
332 	[25] = {0x01},
333 };
334 
335 static const u8 mtk_smi_larb_mt8195_ostd[][SMI_LARB_PORT_NR_MAX] = {
336 	[0] = {0x0a, 0xc, 0x22, 0x22, 0x01, 0x0a,}, /* larb0 */
337 	[1] = {0x0a, 0xc, 0x22, 0x22, 0x01, 0x0a,}, /* larb1 */
338 	[2] = {0x12, 0x12, 0x12, 0x12, 0x0a,},      /* ... */
339 	[3] = {0x12, 0x12, 0x12, 0x12, 0x28, 0x28, 0x0a,},
340 	[4] = {0x06, 0x01, 0x17, 0x06, 0x0a,},
341 	[5] = {0x06, 0x01, 0x17, 0x06, 0x06, 0x01, 0x06, 0x0a,},
342 	[6] = {0x06, 0x01, 0x06, 0x0a,},
343 	[7] = {0x0c, 0x0c, 0x12,},
344 	[8] = {0x0c, 0x0c, 0x12,},
345 	[9] = {0x0a, 0x08, 0x04, 0x06, 0x01, 0x01, 0x10, 0x18, 0x11, 0x0a,
346 		0x08, 0x04, 0x11, 0x06, 0x02, 0x06, 0x01, 0x11, 0x11, 0x06,},
347 	[10] = {0x18, 0x08, 0x01, 0x01, 0x20, 0x12, 0x18, 0x06, 0x05, 0x10,
348 		0x08, 0x08, 0x10, 0x08, 0x08, 0x18, 0x0c, 0x09, 0x0b, 0x0d,
349 		0x0d, 0x06, 0x10, 0x10,},
350 	[11] = {0x0e, 0x0e, 0x0e, 0x0e, 0x0e, 0x0e, 0x01, 0x01, 0x01, 0x01,},
351 	[12] = {0x09, 0x09, 0x05, 0x05, 0x0c, 0x18, 0x02, 0x02, 0x04, 0x02,},
352 	[13] = {0x02, 0x02, 0x12, 0x12, 0x02, 0x02, 0x02, 0x02, 0x08, 0x01,},
353 	[14] = {0x12, 0x12, 0x02, 0x02, 0x02, 0x02, 0x16, 0x01, 0x16, 0x01,
354 		0x01, 0x02, 0x02, 0x08, 0x02,},
355 	[15] = {},
356 	[16] = {0x28, 0x02, 0x02, 0x12, 0x02, 0x12, 0x10, 0x02, 0x02, 0x0a,
357 		0x12, 0x02, 0x0a, 0x16, 0x02, 0x04,},
358 	[17] = {0x1a, 0x0e, 0x0a, 0x0a, 0x0c, 0x0e, 0x10,},
359 	[18] = {0x12, 0x06, 0x12, 0x06,},
360 	[19] = {0x01, 0x04, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04, 0x04, 0x01,
361 		0x01, 0x01, 0x04, 0x0a, 0x06, 0x01, 0x01, 0x01, 0x0a, 0x06,
362 		0x01, 0x01, 0x05, 0x03, 0x03, 0x04, 0x01,},
363 	[20] = {0x01, 0x04, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04, 0x04, 0x01,
364 		0x01, 0x01, 0x04, 0x0a, 0x06, 0x01, 0x01, 0x01, 0x0a, 0x06,
365 		0x01, 0x01, 0x05, 0x03, 0x03, 0x04, 0x01,},
366 	[21] = {0x28, 0x19, 0x0c, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04,},
367 	[22] = {0x28, 0x19, 0x0c, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04,},
368 	[23] = {0x18, 0x01,},
369 	[24] = {0x01, 0x01, 0x04, 0x01, 0x01, 0x01, 0x01, 0x01, 0x04, 0x01,
370 		0x01, 0x01,},
371 	[25] = {0x02, 0x02, 0x02, 0x28, 0x16, 0x02, 0x02, 0x02, 0x12, 0x16,
372 		0x02, 0x01,},
373 	[26] = {0x02, 0x02, 0x02, 0x28, 0x16, 0x02, 0x02, 0x02, 0x12, 0x16,
374 		0x02, 0x01,},
375 	[27] = {0x02, 0x02, 0x02, 0x28, 0x16, 0x02, 0x02, 0x02, 0x12, 0x16,
376 		0x02, 0x01,},
377 	[28] = {0x1a, 0x0e, 0x0a, 0x0a, 0x0c, 0x0e, 0x10,},
378 };
379 
380 static const struct mtk_smi_larb_gen mtk_smi_larb_mt2701 = {
381 	.port_in_larb = {
382 		LARB0_PORT_OFFSET, LARB1_PORT_OFFSET,
383 		LARB2_PORT_OFFSET, LARB3_PORT_OFFSET
384 	},
385 	.config_port = mtk_smi_larb_config_port_gen1,
386 };
387 
388 static const struct mtk_smi_larb_gen mtk_smi_larb_mt2712 = {
389 	.config_port                = mtk_smi_larb_config_port_gen2_general,
390 	.larb_direct_to_common_mask = BIT(8) | BIT(9),      /* bdpsys */
391 };
392 
393 static const struct mtk_smi_larb_gen mtk_smi_larb_mt6779 = {
394 	.config_port  = mtk_smi_larb_config_port_gen2_general,
395 	.larb_direct_to_common_mask =
396 		BIT(4) | BIT(6) | BIT(11) | BIT(12) | BIT(13),
397 		/* DUMMY | IPU0 | IPU1 | CCU | MDLA */
398 };
399 
400 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8167 = {
401 	/* mt8167 do not need the port in larb */
402 	.config_port = mtk_smi_larb_config_port_mt8167,
403 };
404 
405 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8173 = {
406 	/* mt8173 do not need the port in larb */
407 	.config_port = mtk_smi_larb_config_port_mt8173,
408 };
409 
410 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8183 = {
411 	.config_port                = mtk_smi_larb_config_port_gen2_general,
412 	.larb_direct_to_common_mask = BIT(2) | BIT(3) | BIT(7),
413 				      /* IPU0 | IPU1 | CCU */
414 };
415 
416 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8186 = {
417 	.config_port                = mtk_smi_larb_config_port_gen2_general,
418 	.flags_general	            = MTK_SMI_FLAG_SLEEP_CTL,
419 };
420 
421 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8188 = {
422 	.config_port                = mtk_smi_larb_config_port_gen2_general,
423 	.flags_general	            = MTK_SMI_FLAG_THRT_UPDATE | MTK_SMI_FLAG_SW_FLAG |
424 				      MTK_SMI_FLAG_SLEEP_CTL | MTK_SMI_FLAG_CFG_PORT_SEC_CTL,
425 	.ostd		            = mtk_smi_larb_mt8188_ostd,
426 };
427 
428 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8192 = {
429 	.config_port                = mtk_smi_larb_config_port_gen2_general,
430 };
431 
432 static const struct mtk_smi_larb_gen mtk_smi_larb_mt8195 = {
433 	.config_port                = mtk_smi_larb_config_port_gen2_general,
434 	.flags_general	            = MTK_SMI_FLAG_THRT_UPDATE | MTK_SMI_FLAG_SW_FLAG |
435 				      MTK_SMI_FLAG_SLEEP_CTL,
436 	.ostd		            = mtk_smi_larb_mt8195_ostd,
437 };
438 
439 static const struct of_device_id mtk_smi_larb_of_ids[] = {
440 	{.compatible = "mediatek,mt2701-smi-larb", .data = &mtk_smi_larb_mt2701},
441 	{.compatible = "mediatek,mt2712-smi-larb", .data = &mtk_smi_larb_mt2712},
442 	{.compatible = "mediatek,mt6779-smi-larb", .data = &mtk_smi_larb_mt6779},
443 	{.compatible = "mediatek,mt6795-smi-larb", .data = &mtk_smi_larb_mt8173},
444 	{.compatible = "mediatek,mt8167-smi-larb", .data = &mtk_smi_larb_mt8167},
445 	{.compatible = "mediatek,mt8173-smi-larb", .data = &mtk_smi_larb_mt8173},
446 	{.compatible = "mediatek,mt8183-smi-larb", .data = &mtk_smi_larb_mt8183},
447 	{.compatible = "mediatek,mt8186-smi-larb", .data = &mtk_smi_larb_mt8186},
448 	{.compatible = "mediatek,mt8188-smi-larb", .data = &mtk_smi_larb_mt8188},
449 	{.compatible = "mediatek,mt8192-smi-larb", .data = &mtk_smi_larb_mt8192},
450 	{.compatible = "mediatek,mt8195-smi-larb", .data = &mtk_smi_larb_mt8195},
451 	{}
452 };
453 MODULE_DEVICE_TABLE(of, mtk_smi_larb_of_ids);
454 
455 static int mtk_smi_larb_sleep_ctrl_enable(struct mtk_smi_larb *larb)
456 {
457 	int ret;
458 	u32 tmp;
459 
460 	writel_relaxed(SLP_PROT_EN, larb->base + SMI_LARB_SLP_CON);
461 	ret = readl_poll_timeout_atomic(larb->base + SMI_LARB_SLP_CON,
462 					tmp, !!(tmp & SLP_PROT_RDY), 10, 1000);
463 	if (ret) {
464 		/* TODO: Reset this larb if it fails here. */
465 		dev_err(larb->smi.dev, "sleep ctrl is not ready(0x%x).\n", tmp);
466 	}
467 	return ret;
468 }
469 
470 static void mtk_smi_larb_sleep_ctrl_disable(struct mtk_smi_larb *larb)
471 {
472 	writel_relaxed(0, larb->base + SMI_LARB_SLP_CON);
473 }
474 
475 static int mtk_smi_device_link_common(struct device *dev, struct device **com_dev)
476 {
477 	struct platform_device *smi_com_pdev;
478 	struct device_node *smi_com_node;
479 	struct device *smi_com_dev;
480 	struct device_link *link;
481 
482 	smi_com_node = of_parse_phandle(dev->of_node, "mediatek,smi", 0);
483 	if (!smi_com_node)
484 		return -EINVAL;
485 
486 	smi_com_pdev = of_find_device_by_node(smi_com_node);
487 	of_node_put(smi_com_node);
488 	if (smi_com_pdev) {
489 		/* smi common is the supplier, Make sure it is ready before */
490 		if (!platform_get_drvdata(smi_com_pdev)) {
491 			put_device(&smi_com_pdev->dev);
492 			return -EPROBE_DEFER;
493 		}
494 		smi_com_dev = &smi_com_pdev->dev;
495 		link = device_link_add(dev, smi_com_dev,
496 				       DL_FLAG_PM_RUNTIME | DL_FLAG_STATELESS);
497 		if (!link) {
498 			dev_err(dev, "Unable to link smi-common dev\n");
499 			put_device(&smi_com_pdev->dev);
500 			return -ENODEV;
501 		}
502 		*com_dev = smi_com_dev;
503 	} else {
504 		dev_err(dev, "Failed to get the smi_common device\n");
505 		return -EINVAL;
506 	}
507 	return 0;
508 }
509 
510 static int mtk_smi_dts_clk_init(struct device *dev, struct mtk_smi *smi,
511 				const char * const clks[],
512 				unsigned int clk_nr_required,
513 				unsigned int clk_nr_optional)
514 {
515 	int i, ret;
516 
517 	for (i = 0; i < clk_nr_required; i++)
518 		smi->clks[i].id = clks[i];
519 	ret = devm_clk_bulk_get(dev, clk_nr_required, smi->clks);
520 	if (ret)
521 		return ret;
522 
523 	for (i = clk_nr_required; i < clk_nr_required + clk_nr_optional; i++)
524 		smi->clks[i].id = clks[i];
525 	ret = devm_clk_bulk_get_optional(dev, clk_nr_optional,
526 					 smi->clks + clk_nr_required);
527 	smi->clk_num = clk_nr_required + clk_nr_optional;
528 	return ret;
529 }
530 
531 static int mtk_smi_larb_probe(struct platform_device *pdev)
532 {
533 	struct mtk_smi_larb *larb;
534 	struct device *dev = &pdev->dev;
535 	int ret;
536 
537 	larb = devm_kzalloc(dev, sizeof(*larb), GFP_KERNEL);
538 	if (!larb)
539 		return -ENOMEM;
540 
541 	larb->larb_gen = of_device_get_match_data(dev);
542 	larb->base = devm_platform_ioremap_resource(pdev, 0);
543 	if (IS_ERR(larb->base))
544 		return PTR_ERR(larb->base);
545 
546 	ret = mtk_smi_dts_clk_init(dev, &larb->smi, mtk_smi_larb_clks,
547 				   MTK_SMI_LARB_REQ_CLK_NR, MTK_SMI_LARB_OPT_CLK_NR);
548 	if (ret)
549 		return ret;
550 
551 	larb->smi.dev = dev;
552 
553 	ret = mtk_smi_device_link_common(dev, &larb->smi_common_dev);
554 	if (ret < 0)
555 		return ret;
556 
557 	pm_runtime_enable(dev);
558 	platform_set_drvdata(pdev, larb);
559 	ret = component_add(dev, &mtk_smi_larb_component_ops);
560 	if (ret)
561 		goto err_pm_disable;
562 	return 0;
563 
564 err_pm_disable:
565 	pm_runtime_disable(dev);
566 	device_link_remove(dev, larb->smi_common_dev);
567 	return ret;
568 }
569 
570 static void mtk_smi_larb_remove(struct platform_device *pdev)
571 {
572 	struct mtk_smi_larb *larb = platform_get_drvdata(pdev);
573 
574 	device_link_remove(&pdev->dev, larb->smi_common_dev);
575 	pm_runtime_disable(&pdev->dev);
576 	component_del(&pdev->dev, &mtk_smi_larb_component_ops);
577 }
578 
579 static int __maybe_unused mtk_smi_larb_resume(struct device *dev)
580 {
581 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
582 	const struct mtk_smi_larb_gen *larb_gen = larb->larb_gen;
583 	int ret;
584 
585 	ret = clk_bulk_prepare_enable(larb->smi.clk_num, larb->smi.clks);
586 	if (ret)
587 		return ret;
588 
589 	if (MTK_SMI_CAPS(larb->larb_gen->flags_general, MTK_SMI_FLAG_SLEEP_CTL))
590 		mtk_smi_larb_sleep_ctrl_disable(larb);
591 
592 	/* Configure the basic setting for this larb */
593 	return larb_gen->config_port(dev);
594 }
595 
596 static int __maybe_unused mtk_smi_larb_suspend(struct device *dev)
597 {
598 	struct mtk_smi_larb *larb = dev_get_drvdata(dev);
599 	int ret;
600 
601 	if (MTK_SMI_CAPS(larb->larb_gen->flags_general, MTK_SMI_FLAG_SLEEP_CTL)) {
602 		ret = mtk_smi_larb_sleep_ctrl_enable(larb);
603 		if (ret)
604 			return ret;
605 	}
606 
607 	clk_bulk_disable_unprepare(larb->smi.clk_num, larb->smi.clks);
608 	return 0;
609 }
610 
611 static const struct dev_pm_ops smi_larb_pm_ops = {
612 	SET_RUNTIME_PM_OPS(mtk_smi_larb_suspend, mtk_smi_larb_resume, NULL)
613 	SET_LATE_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
614 				     pm_runtime_force_resume)
615 };
616 
617 static struct platform_driver mtk_smi_larb_driver = {
618 	.probe	= mtk_smi_larb_probe,
619 	.remove_new = mtk_smi_larb_remove,
620 	.driver	= {
621 		.name = "mtk-smi-larb",
622 		.of_match_table = mtk_smi_larb_of_ids,
623 		.pm             = &smi_larb_pm_ops,
624 	}
625 };
626 
627 static const struct mtk_smi_reg_pair mtk_smi_common_mt6795_init[SMI_COMMON_INIT_REGS_NR] = {
628 	{SMI_L1_ARB, 0x1b},
629 	{SMI_M4U_TH, 0xce810c85},
630 	{SMI_FIFO_TH1, 0x43214c8},
631 	{SMI_READ_FIFO_TH, 0x191f},
632 };
633 
634 static const struct mtk_smi_reg_pair mtk_smi_common_mt8195_init[SMI_COMMON_INIT_REGS_NR] = {
635 	{SMI_L1LEN, 0xb},
636 	{SMI_M4U_TH, 0xe100e10},
637 	{SMI_FIFO_TH1, 0x506090a},
638 	{SMI_FIFO_TH2, 0x506090a},
639 	{SMI_DCM, 0x4f1},
640 	{SMI_DUMMY, 0x1},
641 };
642 
643 static const struct mtk_smi_common_plat mtk_smi_common_gen1 = {
644 	.type     = MTK_SMI_GEN1,
645 };
646 
647 static const struct mtk_smi_common_plat mtk_smi_common_gen2 = {
648 	.type	  = MTK_SMI_GEN2,
649 };
650 
651 static const struct mtk_smi_common_plat mtk_smi_common_mt6779 = {
652 	.type	  = MTK_SMI_GEN2,
653 	.has_gals = true,
654 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(2) | F_MMU1_LARB(4) |
655 		    F_MMU1_LARB(5) | F_MMU1_LARB(6) | F_MMU1_LARB(7),
656 };
657 
658 static const struct mtk_smi_common_plat mtk_smi_common_mt6795 = {
659 	.type	  = MTK_SMI_GEN2,
660 	.bus_sel  = F_MMU1_LARB(0),
661 	.init     = mtk_smi_common_mt6795_init,
662 };
663 
664 static const struct mtk_smi_common_plat mtk_smi_common_mt8183 = {
665 	.type     = MTK_SMI_GEN2,
666 	.has_gals = true,
667 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(2) | F_MMU1_LARB(5) |
668 		    F_MMU1_LARB(7),
669 };
670 
671 static const struct mtk_smi_common_plat mtk_smi_common_mt8186 = {
672 	.type     = MTK_SMI_GEN2,
673 	.has_gals = true,
674 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(4) | F_MMU1_LARB(7),
675 };
676 
677 static const struct mtk_smi_common_plat mtk_smi_common_mt8188_vdo = {
678 	.type     = MTK_SMI_GEN2,
679 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(5) | F_MMU1_LARB(7),
680 	.init     = mtk_smi_common_mt8195_init,
681 };
682 
683 static const struct mtk_smi_common_plat mtk_smi_common_mt8188_vpp = {
684 	.type     = MTK_SMI_GEN2,
685 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(2) | F_MMU1_LARB(7),
686 	.init     = mtk_smi_common_mt8195_init,
687 };
688 
689 static const struct mtk_smi_common_plat mtk_smi_common_mt8192 = {
690 	.type     = MTK_SMI_GEN2,
691 	.has_gals = true,
692 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(2) | F_MMU1_LARB(5) |
693 		    F_MMU1_LARB(6),
694 };
695 
696 static const struct mtk_smi_common_plat mtk_smi_common_mt8195_vdo = {
697 	.type     = MTK_SMI_GEN2,
698 	.has_gals = true,
699 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(3) | F_MMU1_LARB(5) |
700 		    F_MMU1_LARB(7),
701 	.init     = mtk_smi_common_mt8195_init,
702 };
703 
704 static const struct mtk_smi_common_plat mtk_smi_common_mt8195_vpp = {
705 	.type     = MTK_SMI_GEN2,
706 	.has_gals = true,
707 	.bus_sel  = F_MMU1_LARB(1) | F_MMU1_LARB(2) | F_MMU1_LARB(7),
708 	.init     = mtk_smi_common_mt8195_init,
709 };
710 
711 static const struct mtk_smi_common_plat mtk_smi_sub_common_mt8195 = {
712 	.type     = MTK_SMI_GEN2_SUB_COMM,
713 	.has_gals = true,
714 };
715 
716 static const struct mtk_smi_common_plat mtk_smi_common_mt8365 = {
717 	.type     = MTK_SMI_GEN2,
718 	.bus_sel  = F_MMU1_LARB(2) | F_MMU1_LARB(4),
719 };
720 
721 static const struct of_device_id mtk_smi_common_of_ids[] = {
722 	{.compatible = "mediatek,mt2701-smi-common", .data = &mtk_smi_common_gen1},
723 	{.compatible = "mediatek,mt2712-smi-common", .data = &mtk_smi_common_gen2},
724 	{.compatible = "mediatek,mt6779-smi-common", .data = &mtk_smi_common_mt6779},
725 	{.compatible = "mediatek,mt6795-smi-common", .data = &mtk_smi_common_mt6795},
726 	{.compatible = "mediatek,mt8167-smi-common", .data = &mtk_smi_common_gen2},
727 	{.compatible = "mediatek,mt8173-smi-common", .data = &mtk_smi_common_gen2},
728 	{.compatible = "mediatek,mt8183-smi-common", .data = &mtk_smi_common_mt8183},
729 	{.compatible = "mediatek,mt8186-smi-common", .data = &mtk_smi_common_mt8186},
730 	{.compatible = "mediatek,mt8188-smi-common-vdo", .data = &mtk_smi_common_mt8188_vdo},
731 	{.compatible = "mediatek,mt8188-smi-common-vpp", .data = &mtk_smi_common_mt8188_vpp},
732 	{.compatible = "mediatek,mt8192-smi-common", .data = &mtk_smi_common_mt8192},
733 	{.compatible = "mediatek,mt8195-smi-common-vdo", .data = &mtk_smi_common_mt8195_vdo},
734 	{.compatible = "mediatek,mt8195-smi-common-vpp", .data = &mtk_smi_common_mt8195_vpp},
735 	{.compatible = "mediatek,mt8195-smi-sub-common", .data = &mtk_smi_sub_common_mt8195},
736 	{.compatible = "mediatek,mt8365-smi-common", .data = &mtk_smi_common_mt8365},
737 	{}
738 };
739 MODULE_DEVICE_TABLE(of, mtk_smi_common_of_ids);
740 
741 static int mtk_smi_common_probe(struct platform_device *pdev)
742 {
743 	struct device *dev = &pdev->dev;
744 	struct mtk_smi *common;
745 	int ret, clk_required = MTK_SMI_COM_REQ_CLK_NR;
746 
747 	common = devm_kzalloc(dev, sizeof(*common), GFP_KERNEL);
748 	if (!common)
749 		return -ENOMEM;
750 	common->dev = dev;
751 	common->plat = of_device_get_match_data(dev);
752 
753 	if (common->plat->has_gals) {
754 		if (common->plat->type == MTK_SMI_GEN2)
755 			clk_required = MTK_SMI_COM_GALS_REQ_CLK_NR;
756 		else if (common->plat->type == MTK_SMI_GEN2_SUB_COMM)
757 			clk_required = MTK_SMI_SUB_COM_GALS_REQ_CLK_NR;
758 	}
759 	ret = mtk_smi_dts_clk_init(dev, common, mtk_smi_common_clks, clk_required, 0);
760 	if (ret)
761 		return ret;
762 
763 	/*
764 	 * for mtk smi gen 1, we need to get the ao(always on) base to config
765 	 * m4u port, and we need to enable the aync clock for transform the smi
766 	 * clock into emi clock domain, but for mtk smi gen2, there's no smi ao
767 	 * base.
768 	 */
769 	if (common->plat->type == MTK_SMI_GEN1) {
770 		common->smi_ao_base = devm_platform_ioremap_resource(pdev, 0);
771 		if (IS_ERR(common->smi_ao_base))
772 			return PTR_ERR(common->smi_ao_base);
773 
774 		common->clk_async = devm_clk_get(dev, "async");
775 		if (IS_ERR(common->clk_async))
776 			return PTR_ERR(common->clk_async);
777 
778 		ret = clk_prepare_enable(common->clk_async);
779 		if (ret)
780 			return ret;
781 	} else {
782 		common->base = devm_platform_ioremap_resource(pdev, 0);
783 		if (IS_ERR(common->base))
784 			return PTR_ERR(common->base);
785 	}
786 
787 	/* link its smi-common if this is smi-sub-common */
788 	if (common->plat->type == MTK_SMI_GEN2_SUB_COMM) {
789 		ret = mtk_smi_device_link_common(dev, &common->smi_common_dev);
790 		if (ret < 0)
791 			return ret;
792 	}
793 
794 	pm_runtime_enable(dev);
795 	platform_set_drvdata(pdev, common);
796 	return 0;
797 }
798 
799 static void mtk_smi_common_remove(struct platform_device *pdev)
800 {
801 	struct mtk_smi *common = dev_get_drvdata(&pdev->dev);
802 
803 	if (common->plat->type == MTK_SMI_GEN2_SUB_COMM)
804 		device_link_remove(&pdev->dev, common->smi_common_dev);
805 	pm_runtime_disable(&pdev->dev);
806 }
807 
808 static int __maybe_unused mtk_smi_common_resume(struct device *dev)
809 {
810 	struct mtk_smi *common = dev_get_drvdata(dev);
811 	const struct mtk_smi_reg_pair *init = common->plat->init;
812 	u32 bus_sel = common->plat->bus_sel; /* default is 0 */
813 	int ret, i;
814 
815 	ret = clk_bulk_prepare_enable(common->clk_num, common->clks);
816 	if (ret)
817 		return ret;
818 
819 	if (common->plat->type != MTK_SMI_GEN2)
820 		return 0;
821 
822 	for (i = 0; i < SMI_COMMON_INIT_REGS_NR && init && init[i].offset; i++)
823 		writel_relaxed(init[i].value, common->base + init[i].offset);
824 
825 	writel(bus_sel, common->base + SMI_BUS_SEL);
826 	return 0;
827 }
828 
829 static int __maybe_unused mtk_smi_common_suspend(struct device *dev)
830 {
831 	struct mtk_smi *common = dev_get_drvdata(dev);
832 
833 	clk_bulk_disable_unprepare(common->clk_num, common->clks);
834 	return 0;
835 }
836 
837 static const struct dev_pm_ops smi_common_pm_ops = {
838 	SET_RUNTIME_PM_OPS(mtk_smi_common_suspend, mtk_smi_common_resume, NULL)
839 	SET_LATE_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
840 				     pm_runtime_force_resume)
841 };
842 
843 static struct platform_driver mtk_smi_common_driver = {
844 	.probe	= mtk_smi_common_probe,
845 	.remove_new = mtk_smi_common_remove,
846 	.driver	= {
847 		.name = "mtk-smi-common",
848 		.of_match_table = mtk_smi_common_of_ids,
849 		.pm             = &smi_common_pm_ops,
850 	}
851 };
852 
853 static struct platform_driver * const smidrivers[] = {
854 	&mtk_smi_common_driver,
855 	&mtk_smi_larb_driver,
856 };
857 
858 static int __init mtk_smi_init(void)
859 {
860 	return platform_register_drivers(smidrivers, ARRAY_SIZE(smidrivers));
861 }
862 module_init(mtk_smi_init);
863 
864 static void __exit mtk_smi_exit(void)
865 {
866 	platform_unregister_drivers(smidrivers, ARRAY_SIZE(smidrivers));
867 }
868 module_exit(mtk_smi_exit);
869 
870 MODULE_DESCRIPTION("MediaTek SMI driver");
871 MODULE_LICENSE("GPL v2");
872