1 // SPDX-License-Identifier: ISC
2 /* Copyright (C) 2022 MediaTek Inc.
3  *
4  * Author: Lorenzo Bianconi <[email protected]>
5  */
6 
7 #include <linux/kernel.h>
8 #include <linux/module.h>
9 #include <linux/usb.h>
10 
11 #include "mt7921.h"
12 #include "mcu.h"
13 #include "../mt76_connac2_mac.h"
14 
15 static const struct usb_device_id mt7921u_device_table[] = {
16 	{ USB_DEVICE_AND_INTERFACE_INFO(0x0e8d, 0x7961, 0xff, 0xff, 0xff),
17 		.driver_info = (kernel_ulong_t)MT7921_FIRMWARE_WM },
18 	/* Comfast CF-952AX */
19 	{ USB_DEVICE_AND_INTERFACE_INFO(0x3574, 0x6211, 0xff, 0xff, 0xff),
20 		.driver_info = (kernel_ulong_t)MT7921_FIRMWARE_WM },
21 	/* Netgear, Inc. [A8000,AXE3000] */
22 	{ USB_DEVICE_AND_INTERFACE_INFO(0x0846, 0x9060, 0xff, 0xff, 0xff),
23 		.driver_info = (kernel_ulong_t)MT7921_FIRMWARE_WM },
24 	/* TP-Link TXE50UH */
25 	{ USB_DEVICE_AND_INTERFACE_INFO(0x35bc, 0x0107, 0xff, 0xff, 0xff),
26 		.driver_info = (kernel_ulong_t)MT7921_FIRMWARE_WM },
27 	{ },
28 };
29 
30 static int
mt7921u_mcu_send_message(struct mt76_dev * mdev,struct sk_buff * skb,int cmd,int * seq)31 mt7921u_mcu_send_message(struct mt76_dev *mdev, struct sk_buff *skb,
32 			 int cmd, int *seq)
33 {
34 	struct mt792x_dev *dev = container_of(mdev, struct mt792x_dev, mt76);
35 	u32 pad, ep;
36 	int ret;
37 
38 	ret = mt76_connac2_mcu_fill_message(mdev, skb, cmd, seq);
39 	if (ret)
40 		return ret;
41 
42 	mdev->mcu.timeout = 3 * HZ;
43 
44 	if (cmd != MCU_CMD(FW_SCATTER))
45 		ep = MT_EP_OUT_INBAND_CMD;
46 	else
47 		ep = MT_EP_OUT_AC_BE;
48 
49 	mt792x_skb_add_usb_sdio_hdr(dev, skb, 0);
50 	pad = round_up(skb->len, 4) + 4 - skb->len;
51 	__skb_put_zero(skb, pad);
52 
53 	ret = mt76u_bulk_msg(&dev->mt76, skb->data, skb->len, NULL,
54 			     1000, ep);
55 	dev_kfree_skb(skb);
56 
57 	return ret;
58 }
59 
mt7921u_mcu_init(struct mt792x_dev * dev)60 static int mt7921u_mcu_init(struct mt792x_dev *dev)
61 {
62 	static const struct mt76_mcu_ops mcu_ops = {
63 		.headroom = MT_SDIO_HDR_SIZE +
64 			    sizeof(struct mt76_connac2_mcu_txd),
65 		.tailroom = MT_USB_TAIL_SIZE,
66 		.mcu_skb_send_msg = mt7921u_mcu_send_message,
67 		.mcu_parse_response = mt7921_mcu_parse_response,
68 	};
69 	int ret;
70 
71 	dev->mt76.mcu_ops = &mcu_ops;
72 
73 	mt76_set(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
74 	ret = mt7921_run_firmware(dev);
75 	if (ret)
76 		return ret;
77 
78 	set_bit(MT76_STATE_MCU_RUNNING, &dev->mphy.state);
79 	mt76_clear(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
80 
81 	return 0;
82 }
83 
mt7921u_mac_reset(struct mt792x_dev * dev)84 static int mt7921u_mac_reset(struct mt792x_dev *dev)
85 {
86 	int err;
87 
88 	mt76_txq_schedule_all(&dev->mphy);
89 	mt76_worker_disable(&dev->mt76.tx_worker);
90 
91 	set_bit(MT76_RESET, &dev->mphy.state);
92 	set_bit(MT76_MCU_RESET, &dev->mphy.state);
93 
94 	wake_up(&dev->mt76.mcu.wait);
95 	skb_queue_purge(&dev->mt76.mcu.res_q);
96 
97 	mt76u_stop_rx(&dev->mt76);
98 	mt76u_stop_tx(&dev->mt76);
99 
100 	mt792xu_wfsys_reset(dev);
101 
102 	clear_bit(MT76_MCU_RESET, &dev->mphy.state);
103 	err = mt76u_resume_rx(&dev->mt76);
104 	if (err)
105 		goto out;
106 
107 	err = mt792xu_mcu_power_on(dev);
108 	if (err)
109 		goto out;
110 
111 	err = mt792xu_dma_init(dev, false);
112 	if (err)
113 		goto out;
114 
115 	mt76_wr(dev, MT_SWDEF_MODE, MT_SWDEF_NORMAL_MODE);
116 	mt76_set(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
117 
118 	err = mt7921_run_firmware(dev);
119 	if (err)
120 		goto out;
121 
122 	mt76_clear(dev, MT_UDMA_TX_QSEL, MT_FW_DL_EN);
123 
124 	err = mt7921_mcu_set_eeprom(dev);
125 	if (err)
126 		goto out;
127 
128 	err = mt7921_mac_init(dev);
129 	if (err)
130 		goto out;
131 
132 	err = __mt7921_start(&dev->phy);
133 out:
134 	clear_bit(MT76_RESET, &dev->mphy.state);
135 
136 	mt76_worker_enable(&dev->mt76.tx_worker);
137 
138 	return err;
139 }
140 
mt7921u_probe(struct usb_interface * usb_intf,const struct usb_device_id * id)141 static int mt7921u_probe(struct usb_interface *usb_intf,
142 			 const struct usb_device_id *id)
143 {
144 	static const struct mt76_driver_ops drv_ops = {
145 		.txwi_size = MT_SDIO_TXD_SIZE,
146 		.drv_flags = MT_DRV_RX_DMA_HDR | MT_DRV_HW_MGMT_TXQ |
147 			     MT_DRV_AMSDU_OFFLOAD,
148 		.survey_flags = SURVEY_INFO_TIME_TX |
149 				SURVEY_INFO_TIME_RX |
150 				SURVEY_INFO_TIME_BSS_RX,
151 		.tx_prepare_skb = mt7921_usb_sdio_tx_prepare_skb,
152 		.tx_complete_skb = mt7921_usb_sdio_tx_complete_skb,
153 		.tx_status_data = mt7921_usb_sdio_tx_status_data,
154 		.rx_skb = mt7921_queue_rx_skb,
155 		.rx_check = mt7921_rx_check,
156 		.sta_add = mt7921_mac_sta_add,
157 		.sta_event = mt7921_mac_sta_event,
158 		.sta_remove = mt7921_mac_sta_remove,
159 		.update_survey = mt792x_update_channel,
160 		.set_channel = mt7921_set_channel,
161 	};
162 	static const struct mt792x_hif_ops hif_ops = {
163 		.mcu_init = mt7921u_mcu_init,
164 		.init_reset = mt792xu_init_reset,
165 		.reset = mt7921u_mac_reset,
166 	};
167 	static struct mt76_bus_ops bus_ops = {
168 		.rr = mt792xu_rr,
169 		.wr = mt792xu_wr,
170 		.rmw = mt792xu_rmw,
171 		.read_copy = mt76u_read_copy,
172 		.write_copy = mt792xu_copy,
173 		.type = MT76_BUS_USB,
174 	};
175 	struct usb_device *udev = interface_to_usbdev(usb_intf);
176 	struct ieee80211_ops *ops;
177 	struct ieee80211_hw *hw;
178 	struct mt792x_dev *dev;
179 	struct mt76_dev *mdev;
180 	u8 features;
181 	int ret;
182 
183 	ops = mt792x_get_mac80211_ops(&usb_intf->dev, &mt7921_ops,
184 				      (void *)id->driver_info, &features);
185 	if (!ops)
186 		return -ENOMEM;
187 
188 	ops->stop = mt792xu_stop;
189 	mdev = mt76_alloc_device(&usb_intf->dev, sizeof(*dev), ops, &drv_ops);
190 	if (!mdev)
191 		return -ENOMEM;
192 
193 	dev = container_of(mdev, struct mt792x_dev, mt76);
194 	dev->fw_features = features;
195 	dev->hif_ops = &hif_ops;
196 
197 	udev = usb_get_dev(udev);
198 	usb_reset_device(udev);
199 
200 	usb_set_intfdata(usb_intf, dev);
201 
202 	ret = __mt76u_init(mdev, usb_intf, &bus_ops);
203 	if (ret < 0)
204 		goto error;
205 
206 	mdev->rev = (mt76_rr(dev, MT_HW_CHIPID) << 16) |
207 		    (mt76_rr(dev, MT_HW_REV) & 0xff);
208 	dev_dbg(mdev->dev, "ASIC revision: %04x\n", mdev->rev);
209 
210 	if (mt76_get_field(dev, MT_CONN_ON_MISC, MT_TOP_MISC2_FW_N9_RDY)) {
211 		ret = mt792xu_wfsys_reset(dev);
212 		if (ret)
213 			goto error;
214 	}
215 
216 	ret = mt792xu_mcu_power_on(dev);
217 	if (ret)
218 		goto error;
219 
220 	ret = mt76u_alloc_mcu_queue(&dev->mt76);
221 	if (ret)
222 		goto error;
223 
224 	ret = mt76u_alloc_queues(&dev->mt76);
225 	if (ret)
226 		goto error;
227 
228 	ret = mt792xu_dma_init(dev, false);
229 	if (ret)
230 		goto error;
231 
232 	hw = mt76_hw(dev);
233 	/* check hw sg support in order to enable AMSDU */
234 	hw->max_tx_fragments = mdev->usb.sg_en ? MT_HW_TXP_MAX_BUF_NUM : 1;
235 
236 	ret = mt7921_register_device(dev);
237 	if (ret)
238 		goto error;
239 
240 	return 0;
241 
242 error:
243 	mt76u_queues_deinit(&dev->mt76);
244 
245 	usb_set_intfdata(usb_intf, NULL);
246 	usb_put_dev(interface_to_usbdev(usb_intf));
247 
248 	mt76_free_device(&dev->mt76);
249 
250 	return ret;
251 }
252 
253 #ifdef CONFIG_PM
mt7921u_suspend(struct usb_interface * intf,pm_message_t state)254 static int mt7921u_suspend(struct usb_interface *intf, pm_message_t state)
255 {
256 	struct mt792x_dev *dev = usb_get_intfdata(intf);
257 	struct mt76_connac_pm *pm = &dev->pm;
258 	int err;
259 
260 	pm->suspended = true;
261 	flush_work(&dev->reset_work);
262 
263 	err = mt76_connac_mcu_set_hif_suspend(&dev->mt76, true, true);
264 	if (err)
265 		goto failed;
266 
267 	mt76u_stop_rx(&dev->mt76);
268 	mt76u_stop_tx(&dev->mt76);
269 
270 	return 0;
271 
272 failed:
273 	pm->suspended = false;
274 
275 	if (err < 0)
276 		mt792x_reset(&dev->mt76);
277 
278 	return err;
279 }
280 
mt7921u_resume(struct usb_interface * intf)281 static int mt7921u_resume(struct usb_interface *intf)
282 {
283 	struct mt792x_dev *dev = usb_get_intfdata(intf);
284 	struct mt76_connac_pm *pm = &dev->pm;
285 	bool reinit = true;
286 	int err, i;
287 
288 	for (i = 0; i < 10; i++) {
289 		u32 val = mt76_rr(dev, MT_WF_SW_DEF_CR_USB_MCU_EVENT);
290 
291 		if (!(val & MT_WF_SW_SER_TRIGGER_SUSPEND)) {
292 			reinit = false;
293 			break;
294 		}
295 		if (val & MT_WF_SW_SER_DONE_SUSPEND) {
296 			mt76_wr(dev, MT_WF_SW_DEF_CR_USB_MCU_EVENT, 0);
297 			break;
298 		}
299 
300 		msleep(20);
301 	}
302 
303 	if (reinit || mt792x_dma_need_reinit(dev)) {
304 		err = mt792xu_dma_init(dev, true);
305 		if (err)
306 			goto failed;
307 	}
308 
309 	err = mt76u_resume_rx(&dev->mt76);
310 	if (err < 0)
311 		goto failed;
312 
313 	err = mt76_connac_mcu_set_hif_suspend(&dev->mt76, false, true);
314 failed:
315 	pm->suspended = false;
316 
317 	if (err < 0)
318 		mt792x_reset(&dev->mt76);
319 
320 	return err;
321 }
322 #endif /* CONFIG_PM */
323 
324 MODULE_DEVICE_TABLE(usb, mt7921u_device_table);
325 MODULE_FIRMWARE(MT7921_FIRMWARE_WM);
326 MODULE_FIRMWARE(MT7921_ROM_PATCH);
327 
328 static struct usb_driver mt7921u_driver = {
329 	.name		= KBUILD_MODNAME,
330 	.id_table	= mt7921u_device_table,
331 	.probe		= mt7921u_probe,
332 	.disconnect	= mt792xu_disconnect,
333 #ifdef CONFIG_PM
334 	.suspend	= mt7921u_suspend,
335 	.resume		= mt7921u_resume,
336 	.reset_resume	= mt7921u_resume,
337 #endif /* CONFIG_PM */
338 	.soft_unbind	= 1,
339 	.disable_hub_initiated_lpm = 1,
340 };
341 module_usb_driver(mt7921u_driver);
342 
343 MODULE_DESCRIPTION("MediaTek MT7921U (USB) wireless driver");
344 MODULE_AUTHOR("Lorenzo Bianconi <[email protected]>");
345 MODULE_LICENSE("Dual BSD/GPL");
346