1// SPDX-License-Identifier: GPL-2.0-only
2/* Copyright (C) 2022 MediaTek Inc.
3 *
4 * Author: Lorenzo Bianconi <lorenzo@kernel.org>
5 *	   Sujuan Chen <sujuan.chen@mediatek.com>
6 */
7
8#include <linux/firmware.h>
9#include <linux/of_address.h>
10#include <linux/of_reserved_mem.h>
11#include <linux/mfd/syscon.h>
12#include <linux/soc/mediatek/mtk_wed.h>
13#include <asm/unaligned.h>
14
15#include "mtk_wed_regs.h"
16#include "mtk_wed_wo.h"
17#include "mtk_wed.h"
18
19static u32 wo_r32(struct mtk_wed_wo *wo, u32 reg)
20{
21	return readl(wo->boot.addr + reg);
22}
23
24static void wo_w32(struct mtk_wed_wo *wo, u32 reg, u32 val)
25{
26	writel(val, wo->boot.addr + reg);
27}
28
29static struct sk_buff *
30mtk_wed_mcu_msg_alloc(const void *data, int data_len)
31{
32	int length = sizeof(struct mtk_wed_mcu_hdr) + data_len;
33	struct sk_buff *skb;
34
35	skb = alloc_skb(length, GFP_KERNEL);
36	if (!skb)
37		return NULL;
38
39	memset(skb->head, 0, length);
40	skb_reserve(skb, sizeof(struct mtk_wed_mcu_hdr));
41	if (data && data_len)
42		skb_put_data(skb, data, data_len);
43
44	return skb;
45}
46
47static struct sk_buff *
48mtk_wed_mcu_get_response(struct mtk_wed_wo *wo, unsigned long expires)
49{
50	if (!time_is_after_jiffies(expires))
51		return NULL;
52
53	wait_event_timeout(wo->mcu.wait, !skb_queue_empty(&wo->mcu.res_q),
54			   expires - jiffies);
55	return skb_dequeue(&wo->mcu.res_q);
56}
57
58void mtk_wed_mcu_rx_event(struct mtk_wed_wo *wo, struct sk_buff *skb)
59{
60	skb_queue_tail(&wo->mcu.res_q, skb);
61	wake_up(&wo->mcu.wait);
62}
63
64static void
65mtk_wed_update_rx_stats(struct mtk_wed_device *wed, struct sk_buff *skb)
66{
67	u32 count = get_unaligned_le32(skb->data);
68	struct mtk_wed_wo_rx_stats *stats;
69	int i;
70
71	if (count * sizeof(*stats) > skb->len - sizeof(u32))
72		return;
73
74	stats = (struct mtk_wed_wo_rx_stats *)(skb->data + sizeof(u32));
75	for (i = 0 ; i < count ; i++)
76		wed->wlan.update_wo_rx_stats(wed, &stats[i]);
77}
78
79void mtk_wed_mcu_rx_unsolicited_event(struct mtk_wed_wo *wo,
80				      struct sk_buff *skb)
81{
82	struct mtk_wed_mcu_hdr *hdr = (struct mtk_wed_mcu_hdr *)skb->data;
83
84	skb_pull(skb, sizeof(*hdr));
85
86	switch (hdr->cmd) {
87	case MTK_WED_WO_EVT_LOG_DUMP:
88		dev_notice(wo->hw->dev, "%s\n", skb->data);
89		break;
90	case MTK_WED_WO_EVT_PROFILING: {
91		struct mtk_wed_wo_log_info *info = (void *)skb->data;
92		u32 count = skb->len / sizeof(*info);
93		int i;
94
95		for (i = 0 ; i < count ; i++)
96			dev_notice(wo->hw->dev,
97				   "SN:%u latency: total=%u, rro:%u, mod:%u\n",
98				   le32_to_cpu(info[i].sn),
99				   le32_to_cpu(info[i].total),
100				   le32_to_cpu(info[i].rro),
101				   le32_to_cpu(info[i].mod));
102		break;
103	}
104	case MTK_WED_WO_EVT_RXCNT_INFO:
105		mtk_wed_update_rx_stats(wo->hw->wed_dev, skb);
106		break;
107	default:
108		break;
109	}
110
111	dev_kfree_skb(skb);
112}
113
114static int
115mtk_wed_mcu_skb_send_msg(struct mtk_wed_wo *wo, struct sk_buff *skb,
116			 int id, int cmd, u16 *wait_seq, bool wait_resp)
117{
118	struct mtk_wed_mcu_hdr *hdr;
119
120	/* TODO: make it dynamic based on cmd */
121	wo->mcu.timeout = 20 * HZ;
122
123	hdr = (struct mtk_wed_mcu_hdr *)skb_push(skb, sizeof(*hdr));
124	hdr->cmd = cmd;
125	hdr->length = cpu_to_le16(skb->len);
126
127	if (wait_resp && wait_seq) {
128		u16 seq = ++wo->mcu.seq;
129
130		if (!seq)
131			seq = ++wo->mcu.seq;
132		*wait_seq = seq;
133
134		hdr->flag |= cpu_to_le16(MTK_WED_WARP_CMD_FLAG_NEED_RSP);
135		hdr->seq = cpu_to_le16(seq);
136	}
137	if (id == MTK_WED_MODULE_ID_WO)
138		hdr->flag |= cpu_to_le16(MTK_WED_WARP_CMD_FLAG_FROM_TO_WO);
139
140	return mtk_wed_wo_queue_tx_skb(wo, &wo->q_tx, skb);
141}
142
143static int
144mtk_wed_mcu_parse_response(struct mtk_wed_wo *wo, struct sk_buff *skb,
145			   int cmd, int seq)
146{
147	struct mtk_wed_mcu_hdr *hdr;
148
149	if (!skb) {
150		dev_err(wo->hw->dev, "Message %08x (seq %d) timeout\n",
151			cmd, seq);
152		return -ETIMEDOUT;
153	}
154
155	hdr = (struct mtk_wed_mcu_hdr *)skb->data;
156	if (le16_to_cpu(hdr->seq) != seq)
157		return -EAGAIN;
158
159	skb_pull(skb, sizeof(*hdr));
160	switch (cmd) {
161	case MTK_WED_WO_CMD_RXCNT_INFO:
162		mtk_wed_update_rx_stats(wo->hw->wed_dev, skb);
163		break;
164	default:
165		break;
166	}
167
168	return 0;
169}
170
171int mtk_wed_mcu_send_msg(struct mtk_wed_wo *wo, int id, int cmd,
172			 const void *data, int len, bool wait_resp)
173{
174	unsigned long expires;
175	struct sk_buff *skb;
176	u16 seq;
177	int ret;
178
179	skb = mtk_wed_mcu_msg_alloc(data, len);
180	if (!skb)
181		return -ENOMEM;
182
183	mutex_lock(&wo->mcu.mutex);
184
185	ret = mtk_wed_mcu_skb_send_msg(wo, skb, id, cmd, &seq, wait_resp);
186	if (ret || !wait_resp)
187		goto unlock;
188
189	expires = jiffies + wo->mcu.timeout;
190	do {
191		skb = mtk_wed_mcu_get_response(wo, expires);
192		ret = mtk_wed_mcu_parse_response(wo, skb, cmd, seq);
193		dev_kfree_skb(skb);
194	} while (ret == -EAGAIN);
195
196unlock:
197	mutex_unlock(&wo->mcu.mutex);
198
199	return ret;
200}
201
202int mtk_wed_mcu_msg_update(struct mtk_wed_device *dev, int id, void *data,
203			   int len)
204{
205	struct mtk_wed_wo *wo = dev->hw->wed_wo;
206
207	if (dev->hw->version == 1)
208		return 0;
209
210	if (WARN_ON(!wo))
211		return -ENODEV;
212
213	return mtk_wed_mcu_send_msg(wo, MTK_WED_MODULE_ID_WO, id, data, len,
214				    true);
215}
216
217static int
218mtk_wed_get_memory_region(struct mtk_wed_wo *wo,
219			  struct mtk_wed_wo_memory_region *region)
220{
221	struct reserved_mem *rmem;
222	struct device_node *np;
223	int index;
224
225	index = of_property_match_string(wo->hw->node, "memory-region-names",
226					 region->name);
227	if (index < 0)
228		return index;
229
230	np = of_parse_phandle(wo->hw->node, "memory-region", index);
231	if (!np)
232		return -ENODEV;
233
234	rmem = of_reserved_mem_lookup(np);
235	of_node_put(np);
236
237	if (!rmem)
238		return -ENODEV;
239
240	region->phy_addr = rmem->base;
241	region->size = rmem->size;
242	region->addr = devm_ioremap(wo->hw->dev, region->phy_addr, region->size);
243
244	return !region->addr ? -EINVAL : 0;
245}
246
247static int
248mtk_wed_mcu_run_firmware(struct mtk_wed_wo *wo, const struct firmware *fw,
249			 struct mtk_wed_wo_memory_region *region)
250{
251	const u8 *first_region_ptr, *region_ptr, *trailer_ptr, *ptr = fw->data;
252	const struct mtk_wed_fw_trailer *trailer;
253	const struct mtk_wed_fw_region *fw_region;
254
255	trailer_ptr = fw->data + fw->size - sizeof(*trailer);
256	trailer = (const struct mtk_wed_fw_trailer *)trailer_ptr;
257	region_ptr = trailer_ptr - trailer->num_region * sizeof(*fw_region);
258	first_region_ptr = region_ptr;
259
260	while (region_ptr < trailer_ptr) {
261		u32 length;
262
263		fw_region = (const struct mtk_wed_fw_region *)region_ptr;
264		length = le32_to_cpu(fw_region->len);
265
266		if (region->phy_addr != le32_to_cpu(fw_region->addr))
267			goto next;
268
269		if (region->size < length)
270			goto next;
271
272		if (first_region_ptr < ptr + length)
273			goto next;
274
275		if (region->shared && region->consumed)
276			return 0;
277
278		if (!region->shared || !region->consumed) {
279			memcpy_toio(region->addr, ptr, length);
280			region->consumed = true;
281			return 0;
282		}
283next:
284		region_ptr += sizeof(*fw_region);
285		ptr += length;
286	}
287
288	return -EINVAL;
289}
290
291static int
292mtk_wed_mcu_load_firmware(struct mtk_wed_wo *wo)
293{
294	static struct mtk_wed_wo_memory_region mem_region[] = {
295		[MTK_WED_WO_REGION_EMI] = {
296			.name = "wo-emi",
297		},
298		[MTK_WED_WO_REGION_ILM] = {
299			.name = "wo-ilm",
300		},
301		[MTK_WED_WO_REGION_DATA] = {
302			.name = "wo-data",
303			.shared = true,
304		},
305	};
306	const struct mtk_wed_fw_trailer *trailer;
307	const struct firmware *fw;
308	const char *fw_name;
309	u32 val, boot_cr;
310	int ret, i;
311
312	/* load firmware region metadata */
313	for (i = 0; i < ARRAY_SIZE(mem_region); i++) {
314		ret = mtk_wed_get_memory_region(wo, &mem_region[i]);
315		if (ret)
316			return ret;
317	}
318
319	wo->boot.name = "wo-boot";
320	ret = mtk_wed_get_memory_region(wo, &wo->boot);
321	if (ret)
322		return ret;
323
324	/* set dummy cr */
325	wed_w32(wo->hw->wed_dev, MTK_WED_SCR0 + 4 * MTK_WED_DUMMY_CR_FWDL,
326		wo->hw->index + 1);
327
328	/* load firmware */
329	if (of_device_is_compatible(wo->hw->node, "mediatek,mt7981-wed"))
330		fw_name = MT7981_FIRMWARE_WO;
331	else
332		fw_name = wo->hw->index ? MT7986_FIRMWARE_WO1 : MT7986_FIRMWARE_WO0;
333
334	ret = request_firmware(&fw, fw_name, wo->hw->dev);
335	if (ret)
336		return ret;
337
338	trailer = (void *)(fw->data + fw->size -
339			   sizeof(struct mtk_wed_fw_trailer));
340	dev_info(wo->hw->dev,
341		 "MTK WED WO Firmware Version: %.10s, Build Time: %.15s\n",
342		 trailer->fw_ver, trailer->build_date);
343	dev_info(wo->hw->dev, "MTK WED WO Chip ID %02x Region %d\n",
344		 trailer->chip_id, trailer->num_region);
345
346	for (i = 0; i < ARRAY_SIZE(mem_region); i++) {
347		ret = mtk_wed_mcu_run_firmware(wo, fw, &mem_region[i]);
348		if (ret)
349			goto out;
350	}
351
352	/* set the start address */
353	boot_cr = wo->hw->index ? MTK_WO_MCU_CFG_LS_WA_BOOT_ADDR_ADDR
354				: MTK_WO_MCU_CFG_LS_WM_BOOT_ADDR_ADDR;
355	wo_w32(wo, boot_cr, mem_region[MTK_WED_WO_REGION_EMI].phy_addr >> 16);
356	/* wo firmware reset */
357	wo_w32(wo, MTK_WO_MCU_CFG_LS_WF_MCCR_CLR_ADDR, 0xc00);
358
359	val = wo_r32(wo, MTK_WO_MCU_CFG_LS_WF_MCU_CFG_WM_WA_ADDR);
360	val |= wo->hw->index ? MTK_WO_MCU_CFG_LS_WF_WM_WA_WA_CPU_RSTB_MASK
361			     : MTK_WO_MCU_CFG_LS_WF_WM_WA_WM_CPU_RSTB_MASK;
362	wo_w32(wo, MTK_WO_MCU_CFG_LS_WF_MCU_CFG_WM_WA_ADDR, val);
363out:
364	release_firmware(fw);
365
366	return ret;
367}
368
369static u32
370mtk_wed_mcu_read_fw_dl(struct mtk_wed_wo *wo)
371{
372	return wed_r32(wo->hw->wed_dev,
373		       MTK_WED_SCR0 + 4 * MTK_WED_DUMMY_CR_FWDL);
374}
375
376int mtk_wed_mcu_init(struct mtk_wed_wo *wo)
377{
378	u32 val;
379	int ret;
380
381	skb_queue_head_init(&wo->mcu.res_q);
382	init_waitqueue_head(&wo->mcu.wait);
383	mutex_init(&wo->mcu.mutex);
384
385	ret = mtk_wed_mcu_load_firmware(wo);
386	if (ret)
387		return ret;
388
389	return readx_poll_timeout(mtk_wed_mcu_read_fw_dl, wo, val, !val,
390				  100, MTK_FW_DL_TIMEOUT);
391}
392
393MODULE_FIRMWARE(MT7981_FIRMWARE_WO);
394MODULE_FIRMWARE(MT7986_FIRMWARE_WO0);
395MODULE_FIRMWARE(MT7986_FIRMWARE_WO1);
396