1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (C) 2015 Jakub Kicinski <kubakici@wp.pl>
4  */
5 
6 #include <linux/kernel.h>
7 #include <linux/module.h>
8 #include <linux/usb.h>
9 
10 #include "mt76x0.h"
11 #include "mcu.h"
12 #include "../mt76x02_usb.h"
13 
14 static struct usb_device_id mt76x0_device_table[] = {
15 	{ USB_DEVICE(0x148F, 0x7610) },	/* MT7610U */
16 	{ USB_DEVICE(0x13B1, 0x003E) },	/* Linksys AE6000 */
17 	{ USB_DEVICE(0x0E8D, 0x7610) },	/* Sabrent NTWLAC */
18 	{ USB_DEVICE(0x7392, 0xa711) },	/* Edimax 7711mac */
19 	{ USB_DEVICE(0x7392, 0xb711) },	/* Edimax / Elecom  */
20 	{ USB_DEVICE(0x148f, 0x761a) },	/* TP-Link TL-WDN5200 */
21 	{ USB_DEVICE(0x148f, 0x760a) },	/* TP-Link unknown */
22 	{ USB_DEVICE(0x0b05, 0x17d1) },	/* Asus USB-AC51 */
23 	{ USB_DEVICE(0x0b05, 0x17db) },	/* Asus USB-AC50 */
24 	{ USB_DEVICE(0x0df6, 0x0075) },	/* Sitecom WLA-3100 */
25 	{ USB_DEVICE(0x2019, 0xab31) },	/* Planex GW-450D */
26 	{ USB_DEVICE(0x2001, 0x3d02) },	/* D-LINK DWA-171 rev B1 */
27 	{ USB_DEVICE(0x0586, 0x3425) },	/* Zyxel NWD6505 */
28 	{ USB_DEVICE(0x07b8, 0x7610) },	/* AboCom AU7212 */
29 	{ USB_DEVICE(0x04bb, 0x0951) },	/* I-O DATA WN-AC433UK */
30 	{ USB_DEVICE(0x057c, 0x8502) },	/* AVM FRITZ!WLAN USB Stick AC 430 */
31 	{ USB_DEVICE(0x293c, 0x5702) },	/* Comcast Xfinity KXW02AAA  */
32 	{ USB_DEVICE(0x20f4, 0x806b) },	/* TRENDnet TEW-806UBH  */
33 	{ USB_DEVICE(0x7392, 0xc711) }, /* Devolo Wifi ac Stick */
34 	{ USB_DEVICE(0x0df6, 0x0079) }, /* Sitecom Europe B.V. ac  Stick */
35 	{ USB_DEVICE(0x2357, 0x0123) }, /* TP-LINK T2UHP_US_v1 */
36 	{ USB_DEVICE(0x2357, 0x010b) }, /* TP-LINK T2UHP_UN_v1 */
37 	/* TP-LINK Archer T1U */
38 	{ USB_DEVICE(0x2357, 0x0105), .driver_info = 1, },
39 	/* MT7630U */
40 	{ USB_DEVICE_AND_INTERFACE_INFO(0x0E8D, 0x7630, 0xff, 0x2, 0xff)},
41 	/* MT7650U */
42 	{ USB_DEVICE_AND_INTERFACE_INFO(0x0E8D, 0x7650, 0xff, 0x2, 0xff)},
43 	{ 0, }
44 };
45 
46 static void mt76x0_init_usb_dma(struct mt76x02_dev *dev)
47 {
48 	u32 val;
49 
50 	val = mt76_rr(dev, MT_USB_DMA_CFG);
51 
52 	val |= MT_USB_DMA_CFG_RX_BULK_EN |
53 	       MT_USB_DMA_CFG_TX_BULK_EN;
54 
55 	/* disable AGGR_BULK_RX in order to receive one
56 	 * frame in each rx urb and avoid copies
57 	 */
58 	val &= ~MT_USB_DMA_CFG_RX_BULK_AGG_EN;
59 	mt76_wr(dev, MT_USB_DMA_CFG, val);
60 
61 	val = mt76_rr(dev, MT_COM_REG0);
62 	if (val & 1)
63 		dev_dbg(dev->mt76.dev, "MCU not ready\n");
64 
65 	val = mt76_rr(dev, MT_USB_DMA_CFG);
66 
67 	val |= MT_USB_DMA_CFG_RX_DROP_OR_PAD;
68 	mt76_wr(dev, MT_USB_DMA_CFG, val);
69 	val &= ~MT_USB_DMA_CFG_RX_DROP_OR_PAD;
70 	mt76_wr(dev, MT_USB_DMA_CFG, val);
71 }
72 
73 static void mt76x0u_cleanup(struct mt76x02_dev *dev)
74 {
75 	clear_bit(MT76_STATE_INITIALIZED, &dev->mphy.state);
76 	mt76x0_chip_onoff(dev, false, false);
77 	mt76u_queues_deinit(&dev->mt76);
78 }
79 
80 static void mt76x0u_stop(struct ieee80211_hw *hw)
81 {
82 	struct mt76x02_dev *dev = hw->priv;
83 
84 	clear_bit(MT76_STATE_RUNNING, &dev->mphy.state);
85 	cancel_delayed_work_sync(&dev->cal_work);
86 	cancel_delayed_work_sync(&dev->mphy.mac_work);
87 	mt76u_stop_tx(&dev->mt76);
88 	mt76x02u_exit_beacon_config(dev);
89 
90 	if (test_bit(MT76_REMOVED, &dev->mphy.state))
91 		return;
92 
93 	if (!mt76_poll(dev, MT_USB_DMA_CFG, MT_USB_DMA_CFG_TX_BUSY, 0, 1000))
94 		dev_warn(dev->mt76.dev, "TX DMA did not stop\n");
95 
96 	mt76x0_mac_stop(dev);
97 
98 	if (!mt76_poll(dev, MT_USB_DMA_CFG, MT_USB_DMA_CFG_RX_BUSY, 0, 1000))
99 		dev_warn(dev->mt76.dev, "RX DMA did not stop\n");
100 }
101 
102 static int mt76x0u_start(struct ieee80211_hw *hw)
103 {
104 	struct mt76x02_dev *dev = hw->priv;
105 	int ret;
106 
107 	ret = mt76x02u_mac_start(dev);
108 	if (ret)
109 		return ret;
110 
111 	mt76x0_phy_calibrate(dev, true);
112 	ieee80211_queue_delayed_work(dev->mt76.hw, &dev->mphy.mac_work,
113 				     MT_MAC_WORK_INTERVAL);
114 	ieee80211_queue_delayed_work(dev->mt76.hw, &dev->cal_work,
115 				     MT_CALIBRATE_INTERVAL);
116 	set_bit(MT76_STATE_RUNNING, &dev->mphy.state);
117 	return 0;
118 }
119 
120 static const struct ieee80211_ops mt76x0u_ops = {
121 	.tx = mt76x02_tx,
122 	.start = mt76x0u_start,
123 	.stop = mt76x0u_stop,
124 	.add_interface = mt76x02_add_interface,
125 	.remove_interface = mt76x02_remove_interface,
126 	.config = mt76x0_config,
127 	.configure_filter = mt76x02_configure_filter,
128 	.bss_info_changed = mt76x02_bss_info_changed,
129 	.sta_state = mt76_sta_state,
130 	.sta_pre_rcu_remove = mt76_sta_pre_rcu_remove,
131 	.set_key = mt76x02_set_key,
132 	.conf_tx = mt76x02_conf_tx,
133 	.sw_scan_start = mt76_sw_scan,
134 	.sw_scan_complete = mt76x02_sw_scan_complete,
135 	.ampdu_action = mt76x02_ampdu_action,
136 	.sta_rate_tbl_update = mt76x02_sta_rate_tbl_update,
137 	.set_rts_threshold = mt76x02_set_rts_threshold,
138 	.wake_tx_queue = mt76_wake_tx_queue,
139 	.get_txpower = mt76_get_txpower,
140 	.get_survey = mt76_get_survey,
141 	.set_tim = mt76_set_tim,
142 	.release_buffered_frames = mt76_release_buffered_frames,
143 	.get_antenna = mt76_get_antenna,
144 	.set_sar_specs = mt76x0_set_sar_specs,
145 };
146 
147 static int mt76x0u_init_hardware(struct mt76x02_dev *dev, bool reset)
148 {
149 	int err;
150 
151 	mt76x0_chip_onoff(dev, true, reset);
152 
153 	if (!mt76x02_wait_for_mac(&dev->mt76))
154 		return -ETIMEDOUT;
155 
156 	err = mt76x0u_mcu_init(dev);
157 	if (err < 0)
158 		return err;
159 
160 	mt76x0_init_usb_dma(dev);
161 	err = mt76x0_init_hardware(dev);
162 	if (err < 0)
163 		return err;
164 
165 	mt76x02u_init_beacon_config(dev);
166 
167 	mt76_rmw(dev, MT_US_CYC_CFG, MT_US_CYC_CNT, 0x1e);
168 	mt76_wr(dev, MT_TXOP_CTRL_CFG,
169 		FIELD_PREP(MT_TXOP_TRUN_EN, 0x3f) |
170 		FIELD_PREP(MT_TXOP_EXT_CCA_DLY, 0x58));
171 
172 	return 0;
173 }
174 
175 static int mt76x0u_register_device(struct mt76x02_dev *dev)
176 {
177 	struct ieee80211_hw *hw = dev->mt76.hw;
178 	struct mt76_usb *usb = &dev->mt76.usb;
179 	int err;
180 
181 	usb->mcu.data = devm_kmalloc(dev->mt76.dev, MCU_RESP_URB_SIZE,
182 				     GFP_KERNEL);
183 	if (!usb->mcu.data)
184 		return -ENOMEM;
185 
186 	err = mt76u_alloc_queues(&dev->mt76);
187 	if (err < 0)
188 		goto out_err;
189 
190 	err = mt76x0u_init_hardware(dev, true);
191 	if (err < 0)
192 		goto out_err;
193 
194 	/* check hw sg support in order to enable AMSDU */
195 	hw->max_tx_fragments = dev->mt76.usb.sg_en ? MT_TX_SG_MAX_SIZE : 1;
196 	err = mt76x0_register_device(dev);
197 	if (err < 0)
198 		goto out_err;
199 
200 	set_bit(MT76_STATE_INITIALIZED, &dev->mphy.state);
201 
202 	return 0;
203 
204 out_err:
205 	mt76x0u_cleanup(dev);
206 	return err;
207 }
208 
209 static int mt76x0u_probe(struct usb_interface *usb_intf,
210 			 const struct usb_device_id *id)
211 {
212 	static const struct mt76_driver_ops drv_ops = {
213 		.drv_flags = MT_DRV_SW_RX_AIRTIME,
214 		.survey_flags = SURVEY_INFO_TIME_TX,
215 		.update_survey = mt76x02_update_channel,
216 		.tx_prepare_skb = mt76x02u_tx_prepare_skb,
217 		.tx_complete_skb = mt76x02u_tx_complete_skb,
218 		.tx_status_data = mt76x02_tx_status_data,
219 		.rx_skb = mt76x02_queue_rx_skb,
220 		.sta_ps = mt76x02_sta_ps,
221 		.sta_add = mt76x02_sta_add,
222 		.sta_remove = mt76x02_sta_remove,
223 	};
224 	struct usb_device *usb_dev = interface_to_usbdev(usb_intf);
225 	struct mt76x02_dev *dev;
226 	struct mt76_dev *mdev;
227 	u32 mac_rev;
228 	int ret;
229 
230 	mdev = mt76_alloc_device(&usb_intf->dev, sizeof(*dev), &mt76x0u_ops,
231 				 &drv_ops);
232 	if (!mdev)
233 		return -ENOMEM;
234 
235 	dev = container_of(mdev, struct mt76x02_dev, mt76);
236 	mutex_init(&dev->phy_mutex);
237 
238 	/* Quirk for Archer T1U */
239 	if (id->driver_info)
240 		dev->no_2ghz = true;
241 
242 	usb_dev = usb_get_dev(usb_dev);
243 	usb_reset_device(usb_dev);
244 
245 	usb_set_intfdata(usb_intf, dev);
246 
247 	mt76x02u_init_mcu(mdev);
248 	ret = mt76u_init(mdev, usb_intf, false);
249 	if (ret)
250 		goto err;
251 
252 	/* Disable the HW, otherwise MCU fail to initialize on hot reboot */
253 	mt76x0_chip_onoff(dev, false, false);
254 
255 	if (!mt76x02_wait_for_mac(mdev)) {
256 		ret = -ETIMEDOUT;
257 		goto err;
258 	}
259 
260 	mdev->rev = mt76_rr(dev, MT_ASIC_VERSION);
261 	mac_rev = mt76_rr(dev, MT_MAC_CSR0);
262 	dev_info(mdev->dev, "ASIC revision: %08x MAC revision: %08x\n",
263 		 mdev->rev, mac_rev);
264 	if (!is_mt76x0(dev)) {
265 		ret = -ENODEV;
266 		goto err;
267 	}
268 
269 	/* Note: vendor driver skips this check for MT76X0U */
270 	if (!(mt76_rr(dev, MT_EFUSE_CTRL) & MT_EFUSE_CTRL_SEL))
271 		dev_warn(mdev->dev, "Warning: eFUSE not present\n");
272 
273 	ret = mt76x0u_register_device(dev);
274 	if (ret < 0)
275 		goto err;
276 
277 	return 0;
278 
279 err:
280 	usb_set_intfdata(usb_intf, NULL);
281 	usb_put_dev(interface_to_usbdev(usb_intf));
282 	mt76u_queues_deinit(&dev->mt76);
283 	mt76_free_device(&dev->mt76);
284 
285 	return ret;
286 }
287 
288 static void mt76x0_disconnect(struct usb_interface *usb_intf)
289 {
290 	struct mt76x02_dev *dev = usb_get_intfdata(usb_intf);
291 	bool initialized = test_bit(MT76_STATE_INITIALIZED, &dev->mphy.state);
292 
293 	if (!initialized)
294 		return;
295 
296 	ieee80211_unregister_hw(dev->mt76.hw);
297 	mt76x0u_cleanup(dev);
298 
299 	usb_set_intfdata(usb_intf, NULL);
300 	usb_put_dev(interface_to_usbdev(usb_intf));
301 
302 	mt76_free_device(&dev->mt76);
303 }
304 
305 static int __maybe_unused mt76x0_suspend(struct usb_interface *usb_intf,
306 					 pm_message_t state)
307 {
308 	struct mt76x02_dev *dev = usb_get_intfdata(usb_intf);
309 
310 	mt76u_stop_rx(&dev->mt76);
311 	clear_bit(MT76_STATE_MCU_RUNNING, &dev->mphy.state);
312 	mt76x0_chip_onoff(dev, false, false);
313 
314 	return 0;
315 }
316 
317 static int __maybe_unused mt76x0_resume(struct usb_interface *usb_intf)
318 {
319 	struct mt76x02_dev *dev = usb_get_intfdata(usb_intf);
320 	int ret;
321 
322 	ret = mt76u_resume_rx(&dev->mt76);
323 	if (ret < 0)
324 		goto err;
325 
326 	ret = mt76x0u_init_hardware(dev, false);
327 	if (ret)
328 		goto err;
329 
330 	return 0;
331 err:
332 	mt76x0u_cleanup(dev);
333 	return ret;
334 }
335 
336 MODULE_DEVICE_TABLE(usb, mt76x0_device_table);
337 MODULE_FIRMWARE(MT7610E_FIRMWARE);
338 MODULE_FIRMWARE(MT7610U_FIRMWARE);
339 MODULE_LICENSE("GPL");
340 
341 static struct usb_driver mt76x0_driver = {
342 	.name		= KBUILD_MODNAME,
343 	.id_table	= mt76x0_device_table,
344 	.probe		= mt76x0u_probe,
345 	.disconnect	= mt76x0_disconnect,
346 	.suspend	= mt76x0_suspend,
347 	.resume		= mt76x0_resume,
348 	.reset_resume	= mt76x0_resume,
349 	.soft_unbind	= 1,
350 	.disable_hub_initiated_lpm = 1,
351 };
352 module_usb_driver(mt76x0_driver);
353