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