Linux Audio

Check our new training course

Loading...
v6.13.7
  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * (c) Copyright 2002-2010, Ralink Technology, Inc.
  4 * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
  5 * Copyright (C) 2015 Jakub Kicinski <kubakici@wp.pl>
 
 
 
 
 
 
 
 
 
  6 */
  7
  8#include <linux/kernel.h>
  9#include <linux/firmware.h>
 10#include <linux/delay.h>
 11#include <linux/usb.h>
 12#include <linux/skbuff.h>
 13
 14#include "mt7601u.h"
 15#include "dma.h"
 16#include "mcu.h"
 17#include "usb.h"
 18#include "trace.h"
 19
 20#define MCU_FW_URB_MAX_PAYLOAD		0x3800
 21#define MCU_FW_URB_SIZE			(MCU_FW_URB_MAX_PAYLOAD + 12)
 22#define MCU_RESP_URB_SIZE		1024
 23
 24static inline int firmware_running(struct mt7601u_dev *dev)
 25{
 26	return mt7601u_rr(dev, MT_MCU_COM_REG0) == 1;
 27}
 28
 29static inline void skb_put_le32(struct sk_buff *skb, u32 val)
 30{
 31	put_unaligned_le32(val, skb_put(skb, 4));
 32}
 33
 34static inline void mt7601u_dma_skb_wrap_cmd(struct sk_buff *skb,
 35					    u8 seq, enum mcu_cmd cmd)
 36{
 37	WARN_ON(mt7601u_dma_skb_wrap(skb, CPU_TX_PORT, DMA_COMMAND,
 38				     FIELD_PREP(MT_TXD_CMD_INFO_SEQ, seq) |
 39				     FIELD_PREP(MT_TXD_CMD_INFO_TYPE, cmd)));
 40}
 41
 42static inline void trace_mt_mcu_msg_send_cs(struct mt7601u_dev *dev,
 43					    struct sk_buff *skb, bool need_resp)
 44{
 45	u32 i, csum = 0;
 46
 47	for (i = 0; i < skb->len / 4; i++)
 48		csum ^= get_unaligned_le32(skb->data + i * 4);
 49
 50	trace_mt_mcu_msg_send(dev, skb, csum, need_resp);
 51}
 52
 53static struct sk_buff *mt7601u_mcu_msg_alloc(const void *data, int len)
 
 54{
 55	struct sk_buff *skb;
 56
 57	WARN_ON(len % 4); /* if length is not divisible by 4 we need to pad */
 58
 59	skb = alloc_skb(len + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
 60	if (skb) {
 61		skb_reserve(skb, MT_DMA_HDR_LEN);
 62		skb_put_data(skb, data, len);
 63	}
 64
 65	return skb;
 66}
 67
 68static int mt7601u_mcu_wait_resp(struct mt7601u_dev *dev, u8 seq)
 69{
 70	struct urb *urb = dev->mcu.resp.urb;
 71	u32 rxfce;
 72	int urb_status, ret, i = 5;
 73
 74	while (i--) {
 75		if (!wait_for_completion_timeout(&dev->mcu.resp_cmpl,
 76						 msecs_to_jiffies(300))) {
 77			dev_warn(dev->dev, "Warning: %s retrying\n", __func__);
 78			continue;
 79		}
 80
 81		/* Make copies of important data before reusing the urb */
 82		rxfce = get_unaligned_le32(dev->mcu.resp.buf);
 83		urb_status = urb->status * mt7601u_urb_has_error(urb);
 84
 85		ret = mt7601u_usb_submit_buf(dev, USB_DIR_IN, MT_EP_IN_CMD_RESP,
 86					     &dev->mcu.resp, GFP_KERNEL,
 87					     mt7601u_complete_urb,
 88					     &dev->mcu.resp_cmpl);
 89		if (ret)
 90			return ret;
 91
 92		if (urb_status)
 93			dev_err(dev->dev, "Error: MCU resp urb failed:%d\n",
 94				urb_status);
 95
 96		if (FIELD_GET(MT_RXD_CMD_INFO_CMD_SEQ, rxfce) == seq &&
 97		    FIELD_GET(MT_RXD_CMD_INFO_EVT_TYPE, rxfce) == CMD_DONE)
 98			return 0;
 99
100		dev_err(dev->dev, "Error: MCU resp evt:%lx seq:%hhx-%lx!\n",
101			FIELD_GET(MT_RXD_CMD_INFO_EVT_TYPE, rxfce),
102			seq, FIELD_GET(MT_RXD_CMD_INFO_CMD_SEQ, rxfce));
103	}
104
105	dev_err(dev->dev, "Error: %s timed out\n", __func__);
106	return -ETIMEDOUT;
107}
108
109static int
110mt7601u_mcu_msg_send(struct mt7601u_dev *dev, struct sk_buff *skb,
111		     enum mcu_cmd cmd, bool wait_resp)
112{
113	struct usb_device *usb_dev = mt7601u_to_usb_dev(dev);
114	unsigned cmd_pipe = usb_sndbulkpipe(usb_dev,
115					    dev->out_eps[MT_EP_OUT_INBAND_CMD]);
116	int sent, ret;
117	u8 seq = 0;
118
119	if (test_bit(MT7601U_STATE_REMOVED, &dev->state)) {
120		consume_skb(skb);
121		return 0;
122	}
123
124	mutex_lock(&dev->mcu.mutex);
125
126	if (wait_resp)
127		while (!seq)
128			seq = ++dev->mcu.msg_seq & 0xf;
129
130	mt7601u_dma_skb_wrap_cmd(skb, seq, cmd);
131
132	if (dev->mcu.resp_cmpl.done)
133		dev_err(dev->dev, "Error: MCU response pre-completed!\n");
134
135	trace_mt_mcu_msg_send_cs(dev, skb, wait_resp);
136	trace_mt_submit_urb_sync(dev, cmd_pipe, skb->len);
137	ret = usb_bulk_msg(usb_dev, cmd_pipe, skb->data, skb->len, &sent, 500);
138	if (ret) {
139		dev_err(dev->dev, "Error: send MCU cmd failed:%d\n", ret);
140		goto out;
141	}
142	if (sent != skb->len)
143		dev_err(dev->dev, "Error: %s sent != skb->len\n", __func__);
144
145	if (wait_resp)
146		ret = mt7601u_mcu_wait_resp(dev, seq);
147out:
148	mutex_unlock(&dev->mcu.mutex);
149
150	consume_skb(skb);
151
152	return ret;
153}
154
155static int mt7601u_mcu_function_select(struct mt7601u_dev *dev,
156				       enum mcu_function func, u32 val)
157{
158	struct sk_buff *skb;
159	struct {
160		__le32 id;
161		__le32 value;
162	} __packed __aligned(4) msg = {
163		.id = cpu_to_le32(func),
164		.value = cpu_to_le32(val),
165	};
166
167	skb = mt7601u_mcu_msg_alloc(&msg, sizeof(msg));
168	if (!skb)
169		return -ENOMEM;
170	return mt7601u_mcu_msg_send(dev, skb, CMD_FUN_SET_OP, func == 5);
171}
172
173int mt7601u_mcu_tssi_read_kick(struct mt7601u_dev *dev, int use_hvga)
174{
175	int ret;
176
177	if (!test_bit(MT7601U_STATE_MCU_RUNNING, &dev->state))
178		return 0;
179
180	ret = mt7601u_mcu_function_select(dev, ATOMIC_TSSI_SETTING,
181					  use_hvga);
182	if (ret) {
183		dev_warn(dev->dev, "Warning: MCU TSSI read kick failed\n");
184		return ret;
185	}
186
187	dev->tssi_read_trig = true;
188
189	return 0;
190}
191
192int
193mt7601u_mcu_calibrate(struct mt7601u_dev *dev, enum mcu_calibrate cal, u32 val)
194{
195	struct sk_buff *skb;
196	struct {
197		__le32 id;
198		__le32 value;
199	} __packed __aligned(4) msg = {
200		.id = cpu_to_le32(cal),
201		.value = cpu_to_le32(val),
202	};
203
204	skb = mt7601u_mcu_msg_alloc(&msg, sizeof(msg));
205	if (!skb)
206		return -ENOMEM;
207	return mt7601u_mcu_msg_send(dev, skb, CMD_CALIBRATION_OP, true);
208}
209
210int mt7601u_write_reg_pairs(struct mt7601u_dev *dev, u32 base,
211			    const struct mt76_reg_pair *data, int n)
212{
213	const int max_vals_per_cmd = INBAND_PACKET_MAX_LEN / 8;
214	struct sk_buff *skb;
215	int cnt, i, ret;
216
217	if (!n)
218		return 0;
219
220	cnt = min(max_vals_per_cmd, n);
221
222	skb = alloc_skb(cnt * 8 + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
223	if (!skb)
224		return -ENOMEM;
225	skb_reserve(skb, MT_DMA_HDR_LEN);
226
227	for (i = 0; i < cnt; i++) {
228		skb_put_le32(skb, base + data[i].reg);
229		skb_put_le32(skb, data[i].value);
230	}
231
232	ret = mt7601u_mcu_msg_send(dev, skb, CMD_RANDOM_WRITE, cnt == n);
233	if (ret)
234		return ret;
235
236	return mt7601u_write_reg_pairs(dev, base, data + cnt, n - cnt);
237}
238
239int mt7601u_burst_write_regs(struct mt7601u_dev *dev, u32 offset,
240			     const u32 *data, int n)
241{
242	const int max_regs_per_cmd = INBAND_PACKET_MAX_LEN / 4 - 1;
243	struct sk_buff *skb;
244	int cnt, i, ret;
245
246	if (!n)
247		return 0;
248
249	cnt = min(max_regs_per_cmd, n);
250
251	skb = alloc_skb(cnt * 4 + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
252	if (!skb)
253		return -ENOMEM;
254	skb_reserve(skb, MT_DMA_HDR_LEN);
255
256	skb_put_le32(skb, MT_MCU_MEMMAP_WLAN + offset);
257	for (i = 0; i < cnt; i++)
258		skb_put_le32(skb, data[i]);
259
260	ret = mt7601u_mcu_msg_send(dev, skb, CMD_BURST_WRITE, cnt == n);
261	if (ret)
262		return ret;
263
264	return mt7601u_burst_write_regs(dev, offset + cnt * 4,
265					data + cnt, n - cnt);
266}
267
268struct mt76_fw_header {
269	__le32 ilm_len;
270	__le32 dlm_len;
271	__le16 build_ver;
272	__le16 fw_ver;
273	u8 pad[4];
274	char build_time[16];
275};
276
277struct mt76_fw {
278	struct mt76_fw_header hdr;
279	u8 ivb[MT_MCU_IVB_SIZE];
280	u8 ilm[];
281};
282
283static int __mt7601u_dma_fw(struct mt7601u_dev *dev,
284			    const struct mt7601u_dma_buf *dma_buf,
285			    const void *data, u32 len, u32 dst_addr)
286{
287	DECLARE_COMPLETION_ONSTACK(cmpl);
288	struct mt7601u_dma_buf buf = *dma_buf; /* we need to fake length */
289	__le32 reg;
290	u32 val;
291	int ret;
292
293	reg = cpu_to_le32(FIELD_PREP(MT_TXD_INFO_TYPE, DMA_PACKET) |
294			  FIELD_PREP(MT_TXD_INFO_D_PORT, CPU_TX_PORT) |
295			  FIELD_PREP(MT_TXD_INFO_LEN, len));
296	memcpy(buf.buf, &reg, sizeof(reg));
297	memcpy(buf.buf + sizeof(reg), data, len);
298	memset(buf.buf + sizeof(reg) + len, 0, 8);
299
300	ret = mt7601u_vendor_single_wr(dev, MT_VEND_WRITE_FCE,
301				       MT_FCE_DMA_ADDR, dst_addr);
302	if (ret)
303		return ret;
304	len = roundup(len, 4);
305	ret = mt7601u_vendor_single_wr(dev, MT_VEND_WRITE_FCE,
306				       MT_FCE_DMA_LEN, len << 16);
307	if (ret)
308		return ret;
309
310	buf.len = MT_DMA_HDR_LEN + len + 4;
311	ret = mt7601u_usb_submit_buf(dev, USB_DIR_OUT, MT_EP_OUT_INBAND_CMD,
312				     &buf, GFP_KERNEL,
313				     mt7601u_complete_urb, &cmpl);
314	if (ret)
315		return ret;
316
317	if (!wait_for_completion_timeout(&cmpl, msecs_to_jiffies(1000))) {
318		dev_err(dev->dev, "Error: firmware upload timed out\n");
319		usb_kill_urb(buf.urb);
320		return -ETIMEDOUT;
321	}
322	if (mt7601u_urb_has_error(buf.urb)) {
323		dev_err(dev->dev, "Error: firmware upload urb failed:%d\n",
324			buf.urb->status);
325		return buf.urb->status;
326	}
327
328	val = mt7601u_rr(dev, MT_TX_CPU_FROM_FCE_CPU_DESC_IDX);
329	val++;
330	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_CPU_DESC_IDX, val);
331
332	return 0;
333}
334
335static int
336mt7601u_dma_fw(struct mt7601u_dev *dev, struct mt7601u_dma_buf *dma_buf,
337	       const void *data, int len, u32 dst_addr)
338{
339	int n, ret;
340
341	if (len == 0)
342		return 0;
343
344	n = min(MCU_FW_URB_MAX_PAYLOAD, len);
345	ret = __mt7601u_dma_fw(dev, dma_buf, data, n, dst_addr);
346	if (ret)
347		return ret;
348
349	if (!mt76_poll_msec(dev, MT_MCU_COM_REG1, BIT(31), BIT(31), 500))
350		return -ETIMEDOUT;
351
352	return mt7601u_dma_fw(dev, dma_buf, data + n, len - n, dst_addr + n);
353}
354
355static int
356mt7601u_upload_firmware(struct mt7601u_dev *dev, const struct mt76_fw *fw)
357{
358	struct mt7601u_dma_buf dma_buf;
359	void *ivb;
360	u32 ilm_len, dlm_len;
361	int i, ret;
362
363	ivb = kmemdup(fw->ivb, sizeof(fw->ivb), GFP_KERNEL);
364	if (!ivb)
365		return -ENOMEM;
366	if (mt7601u_usb_alloc_buf(dev, MCU_FW_URB_SIZE, &dma_buf)) {
367		ret = -ENOMEM;
368		goto error;
369	}
370
371	ilm_len = le32_to_cpu(fw->hdr.ilm_len) - sizeof(fw->ivb);
372	dev_dbg(dev->dev, "loading FW - ILM %u + IVB %zu\n",
373		ilm_len, sizeof(fw->ivb));
374	ret = mt7601u_dma_fw(dev, &dma_buf, fw->ilm, ilm_len, sizeof(fw->ivb));
375	if (ret)
376		goto error;
377
378	dlm_len = le32_to_cpu(fw->hdr.dlm_len);
379	dev_dbg(dev->dev, "loading FW - DLM %u\n", dlm_len);
380	ret = mt7601u_dma_fw(dev, &dma_buf, fw->ilm + ilm_len,
381			     dlm_len, MT_MCU_DLM_OFFSET);
382	if (ret)
383		goto error;
384
385	ret = mt7601u_vendor_request(dev, MT_VEND_DEV_MODE, USB_DIR_OUT,
386				     0x12, 0, ivb, sizeof(fw->ivb));
387	if (ret < 0)
388		goto error;
389	ret = 0;
390
391	for (i = 100; i && !firmware_running(dev); i--)
392		msleep(10);
393	if (!i) {
394		ret = -ETIMEDOUT;
395		goto error;
396	}
397
398	dev_dbg(dev->dev, "Firmware running!\n");
399error:
400	kfree(ivb);
401	mt7601u_usb_free_buf(dev, &dma_buf);
402
403	return ret;
404}
405
406static int mt7601u_load_firmware(struct mt7601u_dev *dev)
407{
408	const struct firmware *fw;
409	const struct mt76_fw_header *hdr;
410	int len, ret;
411	u32 val;
412
413	mt7601u_wr(dev, MT_USB_DMA_CFG, (MT_USB_DMA_CFG_RX_BULK_EN |
414					 MT_USB_DMA_CFG_TX_BULK_EN));
415
416	if (firmware_running(dev))
417		return firmware_request_cache(dev->dev, MT7601U_FIRMWARE);
418
419	ret = request_firmware(&fw, MT7601U_FIRMWARE, dev->dev);
420	if (ret)
421		return ret;
422
423	if (!fw || !fw->data || fw->size < sizeof(*hdr))
424		goto err_inv_fw;
425
426	hdr = (const struct mt76_fw_header *) fw->data;
427
428	if (le32_to_cpu(hdr->ilm_len) <= MT_MCU_IVB_SIZE)
429		goto err_inv_fw;
430
431	len = sizeof(*hdr);
432	len += le32_to_cpu(hdr->ilm_len);
433	len += le32_to_cpu(hdr->dlm_len);
434
435	if (fw->size != len)
436		goto err_inv_fw;
437
438	val = le16_to_cpu(hdr->fw_ver);
439	dev_info(dev->dev,
440		 "Firmware Version: %d.%d.%02d Build: %x Build time: %.16s\n",
441		 (val >> 12) & 0xf, (val >> 8) & 0xf, val & 0xf,
442		 le16_to_cpu(hdr->build_ver), hdr->build_time);
443
444	len = le32_to_cpu(hdr->ilm_len);
445
446	mt7601u_wr(dev, 0x94c, 0);
447	mt7601u_wr(dev, MT_FCE_PSE_CTRL, 0);
448
449	mt7601u_vendor_reset(dev);
450	msleep(5);
451
452	mt7601u_wr(dev, 0xa44, 0);
453	mt7601u_wr(dev, 0x230, 0x84210);
454	mt7601u_wr(dev, 0x400, 0x80c00);
455	mt7601u_wr(dev, 0x800, 1);
456
457	mt7601u_rmw(dev, MT_PBF_CFG, 0, (MT_PBF_CFG_TX0Q_EN |
458					 MT_PBF_CFG_TX1Q_EN |
459					 MT_PBF_CFG_TX2Q_EN |
460					 MT_PBF_CFG_TX3Q_EN));
461
462	mt7601u_wr(dev, MT_FCE_PSE_CTRL, 1);
463
464	mt7601u_wr(dev, MT_USB_DMA_CFG, (MT_USB_DMA_CFG_RX_BULK_EN |
465					 MT_USB_DMA_CFG_TX_BULK_EN));
466	val = mt76_set(dev, MT_USB_DMA_CFG, MT_USB_DMA_CFG_TX_CLR);
467	val &= ~MT_USB_DMA_CFG_TX_CLR;
468	mt7601u_wr(dev, MT_USB_DMA_CFG, val);
469
470	/* FCE tx_fs_base_ptr */
471	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_BASE_PTR, 0x400230);
472	/* FCE tx_fs_max_cnt */
473	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_MAX_COUNT, 1);
474	/* FCE pdma enable */
475	mt7601u_wr(dev, MT_FCE_PDMA_GLOBAL_CONF, 0x44);
476	/* FCE skip_fs_en */
477	mt7601u_wr(dev, MT_FCE_SKIP_FS, 3);
478
479	ret = mt7601u_upload_firmware(dev, (const struct mt76_fw *)fw->data);
480
481	release_firmware(fw);
482
483	return ret;
484
485err_inv_fw:
486	dev_err(dev->dev, "Invalid firmware image\n");
487	release_firmware(fw);
488	return -ENOENT;
489}
490
491int mt7601u_mcu_init(struct mt7601u_dev *dev)
492{
493	int ret;
494
495	mutex_init(&dev->mcu.mutex);
496
497	ret = mt7601u_load_firmware(dev);
498	if (ret)
499		return ret;
500
501	set_bit(MT7601U_STATE_MCU_RUNNING, &dev->state);
502
503	return 0;
504}
505
506int mt7601u_mcu_cmd_init(struct mt7601u_dev *dev)
507{
508	int ret;
509
510	ret = mt7601u_mcu_function_select(dev, Q_SELECT, 1);
511	if (ret)
512		return ret;
513
514	init_completion(&dev->mcu.resp_cmpl);
515	if (mt7601u_usb_alloc_buf(dev, MCU_RESP_URB_SIZE, &dev->mcu.resp)) {
516		mt7601u_usb_free_buf(dev, &dev->mcu.resp);
517		return -ENOMEM;
518	}
519
520	ret = mt7601u_usb_submit_buf(dev, USB_DIR_IN, MT_EP_IN_CMD_RESP,
521				     &dev->mcu.resp, GFP_KERNEL,
522				     mt7601u_complete_urb, &dev->mcu.resp_cmpl);
523	if (ret) {
524		mt7601u_usb_free_buf(dev, &dev->mcu.resp);
525		return ret;
526	}
527
528	return 0;
529}
530
531void mt7601u_mcu_cmd_deinit(struct mt7601u_dev *dev)
532{
533	usb_kill_urb(dev->mcu.resp.urb);
534	mt7601u_usb_free_buf(dev, &dev->mcu.resp);
535}
v4.10.11
 
  1/*
  2 * (c) Copyright 2002-2010, Ralink Technology, Inc.
  3 * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
  4 * Copyright (C) 2015 Jakub Kicinski <kubakici@wp.pl>
  5 *
  6 * This program is free software; you can redistribute it and/or modify
  7 * it under the terms of the GNU General Public License version 2
  8 * as published by the Free Software Foundation
  9 *
 10 * This program is distributed in the hope that it will be useful,
 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 13 * GNU General Public License for more details.
 14 */
 15
 16#include <linux/kernel.h>
 17#include <linux/firmware.h>
 18#include <linux/delay.h>
 19#include <linux/usb.h>
 20#include <linux/skbuff.h>
 21
 22#include "mt7601u.h"
 23#include "dma.h"
 24#include "mcu.h"
 25#include "usb.h"
 26#include "trace.h"
 27
 28#define MCU_FW_URB_MAX_PAYLOAD		0x3800
 29#define MCU_FW_URB_SIZE			(MCU_FW_URB_MAX_PAYLOAD + 12)
 30#define MCU_RESP_URB_SIZE		1024
 31
 32static inline int firmware_running(struct mt7601u_dev *dev)
 33{
 34	return mt7601u_rr(dev, MT_MCU_COM_REG0) == 1;
 35}
 36
 37static inline void skb_put_le32(struct sk_buff *skb, u32 val)
 38{
 39	put_unaligned_le32(val, skb_put(skb, 4));
 40}
 41
 42static inline void mt7601u_dma_skb_wrap_cmd(struct sk_buff *skb,
 43					    u8 seq, enum mcu_cmd cmd)
 44{
 45	WARN_ON(mt7601u_dma_skb_wrap(skb, CPU_TX_PORT, DMA_COMMAND,
 46				     FIELD_PREP(MT_TXD_CMD_INFO_SEQ, seq) |
 47				     FIELD_PREP(MT_TXD_CMD_INFO_TYPE, cmd)));
 48}
 49
 50static inline void trace_mt_mcu_msg_send_cs(struct mt7601u_dev *dev,
 51					    struct sk_buff *skb, bool need_resp)
 52{
 53	u32 i, csum = 0;
 54
 55	for (i = 0; i < skb->len / 4; i++)
 56		csum ^= get_unaligned_le32(skb->data + i * 4);
 57
 58	trace_mt_mcu_msg_send(dev, skb, csum, need_resp);
 59}
 60
 61static struct sk_buff *
 62mt7601u_mcu_msg_alloc(struct mt7601u_dev *dev, const void *data, int len)
 63{
 64	struct sk_buff *skb;
 65
 66	WARN_ON(len % 4); /* if length is not divisible by 4 we need to pad */
 67
 68	skb = alloc_skb(len + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
 69	skb_reserve(skb, MT_DMA_HDR_LEN);
 70	memcpy(skb_put(skb, len), data, len);
 
 
 71
 72	return skb;
 73}
 74
 75static int mt7601u_mcu_wait_resp(struct mt7601u_dev *dev, u8 seq)
 76{
 77	struct urb *urb = dev->mcu.resp.urb;
 78	u32 rxfce;
 79	int urb_status, ret, i = 5;
 80
 81	while (i--) {
 82		if (!wait_for_completion_timeout(&dev->mcu.resp_cmpl,
 83						 msecs_to_jiffies(300))) {
 84			dev_warn(dev->dev, "Warning: %s retrying\n", __func__);
 85			continue;
 86		}
 87
 88		/* Make copies of important data before reusing the urb */
 89		rxfce = get_unaligned_le32(dev->mcu.resp.buf);
 90		urb_status = urb->status * mt7601u_urb_has_error(urb);
 91
 92		ret = mt7601u_usb_submit_buf(dev, USB_DIR_IN, MT_EP_IN_CMD_RESP,
 93					     &dev->mcu.resp, GFP_KERNEL,
 94					     mt7601u_complete_urb,
 95					     &dev->mcu.resp_cmpl);
 96		if (ret)
 97			return ret;
 98
 99		if (urb_status)
100			dev_err(dev->dev, "Error: MCU resp urb failed:%d\n",
101				urb_status);
102
103		if (FIELD_GET(MT_RXD_CMD_INFO_CMD_SEQ, rxfce) == seq &&
104		    FIELD_GET(MT_RXD_CMD_INFO_EVT_TYPE, rxfce) == CMD_DONE)
105			return 0;
106
107		dev_err(dev->dev, "Error: MCU resp evt:%lx seq:%hhx-%lx!\n",
108			FIELD_GET(MT_RXD_CMD_INFO_EVT_TYPE, rxfce),
109			seq, FIELD_GET(MT_RXD_CMD_INFO_CMD_SEQ, rxfce));
110	}
111
112	dev_err(dev->dev, "Error: %s timed out\n", __func__);
113	return -ETIMEDOUT;
114}
115
116static int
117mt7601u_mcu_msg_send(struct mt7601u_dev *dev, struct sk_buff *skb,
118		     enum mcu_cmd cmd, bool wait_resp)
119{
120	struct usb_device *usb_dev = mt7601u_to_usb_dev(dev);
121	unsigned cmd_pipe = usb_sndbulkpipe(usb_dev,
122					    dev->out_eps[MT_EP_OUT_INBAND_CMD]);
123	int sent, ret;
124	u8 seq = 0;
125
126	if (test_bit(MT7601U_STATE_REMOVED, &dev->state))
 
127		return 0;
 
128
129	mutex_lock(&dev->mcu.mutex);
130
131	if (wait_resp)
132		while (!seq)
133			seq = ++dev->mcu.msg_seq & 0xf;
134
135	mt7601u_dma_skb_wrap_cmd(skb, seq, cmd);
136
137	if (dev->mcu.resp_cmpl.done)
138		dev_err(dev->dev, "Error: MCU response pre-completed!\n");
139
140	trace_mt_mcu_msg_send_cs(dev, skb, wait_resp);
141	trace_mt_submit_urb_sync(dev, cmd_pipe, skb->len);
142	ret = usb_bulk_msg(usb_dev, cmd_pipe, skb->data, skb->len, &sent, 500);
143	if (ret) {
144		dev_err(dev->dev, "Error: send MCU cmd failed:%d\n", ret);
145		goto out;
146	}
147	if (sent != skb->len)
148		dev_err(dev->dev, "Error: %s sent != skb->len\n", __func__);
149
150	if (wait_resp)
151		ret = mt7601u_mcu_wait_resp(dev, seq);
152out:
153	mutex_unlock(&dev->mcu.mutex);
154
155	consume_skb(skb);
156
157	return ret;
158}
159
160static int mt7601u_mcu_function_select(struct mt7601u_dev *dev,
161				       enum mcu_function func, u32 val)
162{
163	struct sk_buff *skb;
164	struct {
165		__le32 id;
166		__le32 value;
167	} __packed __aligned(4) msg = {
168		.id = cpu_to_le32(func),
169		.value = cpu_to_le32(val),
170	};
171
172	skb = mt7601u_mcu_msg_alloc(dev, &msg, sizeof(msg));
 
 
173	return mt7601u_mcu_msg_send(dev, skb, CMD_FUN_SET_OP, func == 5);
174}
175
176int mt7601u_mcu_tssi_read_kick(struct mt7601u_dev *dev, int use_hvga)
177{
178	int ret;
179
180	if (!test_bit(MT7601U_STATE_MCU_RUNNING, &dev->state))
181		return 0;
182
183	ret = mt7601u_mcu_function_select(dev, ATOMIC_TSSI_SETTING,
184					  use_hvga);
185	if (ret) {
186		dev_warn(dev->dev, "Warning: MCU TSSI read kick failed\n");
187		return ret;
188	}
189
190	dev->tssi_read_trig = true;
191
192	return 0;
193}
194
195int
196mt7601u_mcu_calibrate(struct mt7601u_dev *dev, enum mcu_calibrate cal, u32 val)
197{
198	struct sk_buff *skb;
199	struct {
200		__le32 id;
201		__le32 value;
202	} __packed __aligned(4) msg = {
203		.id = cpu_to_le32(cal),
204		.value = cpu_to_le32(val),
205	};
206
207	skb = mt7601u_mcu_msg_alloc(dev, &msg, sizeof(msg));
 
 
208	return mt7601u_mcu_msg_send(dev, skb, CMD_CALIBRATION_OP, true);
209}
210
211int mt7601u_write_reg_pairs(struct mt7601u_dev *dev, u32 base,
212			    const struct mt76_reg_pair *data, int n)
213{
214	const int max_vals_per_cmd = INBAND_PACKET_MAX_LEN / 8;
215	struct sk_buff *skb;
216	int cnt, i, ret;
217
218	if (!n)
219		return 0;
220
221	cnt = min(max_vals_per_cmd, n);
222
223	skb = alloc_skb(cnt * 8 + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
224	if (!skb)
225		return -ENOMEM;
226	skb_reserve(skb, MT_DMA_HDR_LEN);
227
228	for (i = 0; i < cnt; i++) {
229		skb_put_le32(skb, base + data[i].reg);
230		skb_put_le32(skb, data[i].value);
231	}
232
233	ret = mt7601u_mcu_msg_send(dev, skb, CMD_RANDOM_WRITE, cnt == n);
234	if (ret)
235		return ret;
236
237	return mt7601u_write_reg_pairs(dev, base, data + cnt, n - cnt);
238}
239
240int mt7601u_burst_write_regs(struct mt7601u_dev *dev, u32 offset,
241			     const u32 *data, int n)
242{
243	const int max_regs_per_cmd = INBAND_PACKET_MAX_LEN / 4 - 1;
244	struct sk_buff *skb;
245	int cnt, i, ret;
246
247	if (!n)
248		return 0;
249
250	cnt = min(max_regs_per_cmd, n);
251
252	skb = alloc_skb(cnt * 4 + MT_DMA_HDR_LEN + 4, GFP_KERNEL);
253	if (!skb)
254		return -ENOMEM;
255	skb_reserve(skb, MT_DMA_HDR_LEN);
256
257	skb_put_le32(skb, MT_MCU_MEMMAP_WLAN + offset);
258	for (i = 0; i < cnt; i++)
259		skb_put_le32(skb, data[i]);
260
261	ret = mt7601u_mcu_msg_send(dev, skb, CMD_BURST_WRITE, cnt == n);
262	if (ret)
263		return ret;
264
265	return mt7601u_burst_write_regs(dev, offset + cnt * 4,
266					data + cnt, n - cnt);
267}
268
269struct mt76_fw_header {
270	__le32 ilm_len;
271	__le32 dlm_len;
272	__le16 build_ver;
273	__le16 fw_ver;
274	u8 pad[4];
275	char build_time[16];
276};
277
278struct mt76_fw {
279	struct mt76_fw_header hdr;
280	u8 ivb[MT_MCU_IVB_SIZE];
281	u8 ilm[];
282};
283
284static int __mt7601u_dma_fw(struct mt7601u_dev *dev,
285			    const struct mt7601u_dma_buf *dma_buf,
286			    const void *data, u32 len, u32 dst_addr)
287{
288	DECLARE_COMPLETION_ONSTACK(cmpl);
289	struct mt7601u_dma_buf buf = *dma_buf; /* we need to fake length */
290	__le32 reg;
291	u32 val;
292	int ret;
293
294	reg = cpu_to_le32(FIELD_PREP(MT_TXD_INFO_TYPE, DMA_PACKET) |
295			  FIELD_PREP(MT_TXD_INFO_D_PORT, CPU_TX_PORT) |
296			  FIELD_PREP(MT_TXD_INFO_LEN, len));
297	memcpy(buf.buf, &reg, sizeof(reg));
298	memcpy(buf.buf + sizeof(reg), data, len);
299	memset(buf.buf + sizeof(reg) + len, 0, 8);
300
301	ret = mt7601u_vendor_single_wr(dev, MT_VEND_WRITE_FCE,
302				       MT_FCE_DMA_ADDR, dst_addr);
303	if (ret)
304		return ret;
305	len = roundup(len, 4);
306	ret = mt7601u_vendor_single_wr(dev, MT_VEND_WRITE_FCE,
307				       MT_FCE_DMA_LEN, len << 16);
308	if (ret)
309		return ret;
310
311	buf.len = MT_DMA_HDR_LEN + len + 4;
312	ret = mt7601u_usb_submit_buf(dev, USB_DIR_OUT, MT_EP_OUT_INBAND_CMD,
313				     &buf, GFP_KERNEL,
314				     mt7601u_complete_urb, &cmpl);
315	if (ret)
316		return ret;
317
318	if (!wait_for_completion_timeout(&cmpl, msecs_to_jiffies(1000))) {
319		dev_err(dev->dev, "Error: firmware upload timed out\n");
320		usb_kill_urb(buf.urb);
321		return -ETIMEDOUT;
322	}
323	if (mt7601u_urb_has_error(buf.urb)) {
324		dev_err(dev->dev, "Error: firmware upload urb failed:%d\n",
325			buf.urb->status);
326		return buf.urb->status;
327	}
328
329	val = mt7601u_rr(dev, MT_TX_CPU_FROM_FCE_CPU_DESC_IDX);
330	val++;
331	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_CPU_DESC_IDX, val);
332
333	return 0;
334}
335
336static int
337mt7601u_dma_fw(struct mt7601u_dev *dev, struct mt7601u_dma_buf *dma_buf,
338	       const void *data, int len, u32 dst_addr)
339{
340	int n, ret;
341
342	if (len == 0)
343		return 0;
344
345	n = min(MCU_FW_URB_MAX_PAYLOAD, len);
346	ret = __mt7601u_dma_fw(dev, dma_buf, data, n, dst_addr);
347	if (ret)
348		return ret;
349
350	if (!mt76_poll_msec(dev, MT_MCU_COM_REG1, BIT(31), BIT(31), 500))
351		return -ETIMEDOUT;
352
353	return mt7601u_dma_fw(dev, dma_buf, data + n, len - n, dst_addr + n);
354}
355
356static int
357mt7601u_upload_firmware(struct mt7601u_dev *dev, const struct mt76_fw *fw)
358{
359	struct mt7601u_dma_buf dma_buf;
360	void *ivb;
361	u32 ilm_len, dlm_len;
362	int i, ret;
363
364	ivb = kmemdup(fw->ivb, sizeof(fw->ivb), GFP_KERNEL);
365	if (!ivb)
366		return -ENOMEM;
367	if (mt7601u_usb_alloc_buf(dev, MCU_FW_URB_SIZE, &dma_buf)) {
368		ret = -ENOMEM;
369		goto error;
370	}
371
372	ilm_len = le32_to_cpu(fw->hdr.ilm_len) - sizeof(fw->ivb);
373	dev_dbg(dev->dev, "loading FW - ILM %u + IVB %zu\n",
374		ilm_len, sizeof(fw->ivb));
375	ret = mt7601u_dma_fw(dev, &dma_buf, fw->ilm, ilm_len, sizeof(fw->ivb));
376	if (ret)
377		goto error;
378
379	dlm_len = le32_to_cpu(fw->hdr.dlm_len);
380	dev_dbg(dev->dev, "loading FW - DLM %u\n", dlm_len);
381	ret = mt7601u_dma_fw(dev, &dma_buf, fw->ilm + ilm_len,
382			     dlm_len, MT_MCU_DLM_OFFSET);
383	if (ret)
384		goto error;
385
386	ret = mt7601u_vendor_request(dev, MT_VEND_DEV_MODE, USB_DIR_OUT,
387				     0x12, 0, ivb, sizeof(fw->ivb));
388	if (ret < 0)
389		goto error;
390	ret = 0;
391
392	for (i = 100; i && !firmware_running(dev); i--)
393		msleep(10);
394	if (!i) {
395		ret = -ETIMEDOUT;
396		goto error;
397	}
398
399	dev_dbg(dev->dev, "Firmware running!\n");
400error:
401	kfree(ivb);
402	mt7601u_usb_free_buf(dev, &dma_buf);
403
404	return ret;
405}
406
407static int mt7601u_load_firmware(struct mt7601u_dev *dev)
408{
409	const struct firmware *fw;
410	const struct mt76_fw_header *hdr;
411	int len, ret;
412	u32 val;
413
414	mt7601u_wr(dev, MT_USB_DMA_CFG, (MT_USB_DMA_CFG_RX_BULK_EN |
415					 MT_USB_DMA_CFG_TX_BULK_EN));
416
417	if (firmware_running(dev))
418		return 0;
419
420	ret = request_firmware(&fw, MT7601U_FIRMWARE, dev->dev);
421	if (ret)
422		return ret;
423
424	if (!fw || !fw->data || fw->size < sizeof(*hdr))
425		goto err_inv_fw;
426
427	hdr = (const struct mt76_fw_header *) fw->data;
428
429	if (le32_to_cpu(hdr->ilm_len) <= MT_MCU_IVB_SIZE)
430		goto err_inv_fw;
431
432	len = sizeof(*hdr);
433	len += le32_to_cpu(hdr->ilm_len);
434	len += le32_to_cpu(hdr->dlm_len);
435
436	if (fw->size != len)
437		goto err_inv_fw;
438
439	val = le16_to_cpu(hdr->fw_ver);
440	dev_info(dev->dev,
441		 "Firmware Version: %d.%d.%02d Build: %x Build time: %.16s\n",
442		 (val >> 12) & 0xf, (val >> 8) & 0xf, val & 0xf,
443		 le16_to_cpu(hdr->build_ver), hdr->build_time);
444
445	len = le32_to_cpu(hdr->ilm_len);
446
447	mt7601u_wr(dev, 0x94c, 0);
448	mt7601u_wr(dev, MT_FCE_PSE_CTRL, 0);
449
450	mt7601u_vendor_reset(dev);
451	msleep(5);
452
453	mt7601u_wr(dev, 0xa44, 0);
454	mt7601u_wr(dev, 0x230, 0x84210);
455	mt7601u_wr(dev, 0x400, 0x80c00);
456	mt7601u_wr(dev, 0x800, 1);
457
458	mt7601u_rmw(dev, MT_PBF_CFG, 0, (MT_PBF_CFG_TX0Q_EN |
459					 MT_PBF_CFG_TX1Q_EN |
460					 MT_PBF_CFG_TX2Q_EN |
461					 MT_PBF_CFG_TX3Q_EN));
462
463	mt7601u_wr(dev, MT_FCE_PSE_CTRL, 1);
464
465	mt7601u_wr(dev, MT_USB_DMA_CFG, (MT_USB_DMA_CFG_RX_BULK_EN |
466					 MT_USB_DMA_CFG_TX_BULK_EN));
467	val = mt76_set(dev, MT_USB_DMA_CFG, MT_USB_DMA_CFG_TX_CLR);
468	val &= ~MT_USB_DMA_CFG_TX_CLR;
469	mt7601u_wr(dev, MT_USB_DMA_CFG, val);
470
471	/* FCE tx_fs_base_ptr */
472	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_BASE_PTR, 0x400230);
473	/* FCE tx_fs_max_cnt */
474	mt7601u_wr(dev, MT_TX_CPU_FROM_FCE_MAX_COUNT, 1);
475	/* FCE pdma enable */
476	mt7601u_wr(dev, MT_FCE_PDMA_GLOBAL_CONF, 0x44);
477	/* FCE skip_fs_en */
478	mt7601u_wr(dev, MT_FCE_SKIP_FS, 3);
479
480	ret = mt7601u_upload_firmware(dev, (const struct mt76_fw *)fw->data);
481
482	release_firmware(fw);
483
484	return ret;
485
486err_inv_fw:
487	dev_err(dev->dev, "Invalid firmware image\n");
488	release_firmware(fw);
489	return -ENOENT;
490}
491
492int mt7601u_mcu_init(struct mt7601u_dev *dev)
493{
494	int ret;
495
496	mutex_init(&dev->mcu.mutex);
497
498	ret = mt7601u_load_firmware(dev);
499	if (ret)
500		return ret;
501
502	set_bit(MT7601U_STATE_MCU_RUNNING, &dev->state);
503
504	return 0;
505}
506
507int mt7601u_mcu_cmd_init(struct mt7601u_dev *dev)
508{
509	int ret;
510
511	ret = mt7601u_mcu_function_select(dev, Q_SELECT, 1);
512	if (ret)
513		return ret;
514
515	init_completion(&dev->mcu.resp_cmpl);
516	if (mt7601u_usb_alloc_buf(dev, MCU_RESP_URB_SIZE, &dev->mcu.resp)) {
517		mt7601u_usb_free_buf(dev, &dev->mcu.resp);
518		return -ENOMEM;
519	}
520
521	ret = mt7601u_usb_submit_buf(dev, USB_DIR_IN, MT_EP_IN_CMD_RESP,
522				     &dev->mcu.resp, GFP_KERNEL,
523				     mt7601u_complete_urb, &dev->mcu.resp_cmpl);
524	if (ret) {
525		mt7601u_usb_free_buf(dev, &dev->mcu.resp);
526		return ret;
527	}
528
529	return 0;
530}
531
532void mt7601u_mcu_cmd_deinit(struct mt7601u_dev *dev)
533{
534	usb_kill_urb(dev->mcu.resp.urb);
535	mt7601u_usb_free_buf(dev, &dev->mcu.resp);
536}