1 // SPDX-License-Identifier: ISC
2 /*
3  * Copyright (C) 2018 Lorenzo Bianconi <lorenzo.bianconi83@gmail.com>
4  */
5 
6 #include <linux/kernel.h>
7 #include <linux/module.h>
8 
9 #include "../mt76x02_usb.h"
10 #include "mt76x2u.h"
11 
12 static const struct usb_device_id mt76x2u_device_table[] = {
13 	{ USB_DEVICE(0x0b05, 0x1833) },	/* Asus USB-AC54 */
14 	{ USB_DEVICE(0x0b05, 0x17eb) },	/* Asus USB-AC55 */
15 	{ USB_DEVICE(0x0b05, 0x180b) },	/* Asus USB-N53 B1 */
16 	{ USB_DEVICE(0x0e8d, 0x7612) },	/* Aukey USBAC1200 - Alfa AWUS036ACM */
17 	{ USB_DEVICE(0x057c, 0x8503) },	/* Avm FRITZ!WLAN AC860 */
18 	{ USB_DEVICE(0x7392, 0xb711) },	/* Edimax EW 7722 UAC */
19 	{ USB_DEVICE(0x2c4e, 0x0103) },	/* Mercury UD13 */
20 	{ USB_DEVICE(0x0846, 0x9053) },	/* Netgear A6210 */
21 	{ USB_DEVICE(0x045e, 0x02e6) },	/* XBox One Wireless Adapter */
22 	{ USB_DEVICE(0x045e, 0x02fe) },	/* XBox One Wireless Adapter */
23 	{ },
24 };
25 
26 static int mt76x2u_probe(struct usb_interface *intf,
27 			 const struct usb_device_id *id)
28 {
29 	static const struct mt76_driver_ops drv_ops = {
30 		.drv_flags = MT_DRV_SW_RX_AIRTIME,
31 		.survey_flags = SURVEY_INFO_TIME_TX,
32 		.update_survey = mt76x02_update_channel,
33 		.tx_prepare_skb = mt76x02u_tx_prepare_skb,
34 		.tx_complete_skb = mt76x02u_tx_complete_skb,
35 		.tx_status_data = mt76x02_tx_status_data,
36 		.rx_skb = mt76x02_queue_rx_skb,
37 		.sta_ps = mt76x02_sta_ps,
38 		.sta_add = mt76x02_sta_add,
39 		.sta_remove = mt76x02_sta_remove,
40 	};
41 	struct usb_device *udev = interface_to_usbdev(intf);
42 	struct mt76x02_dev *dev;
43 	struct mt76_dev *mdev;
44 	int err;
45 
46 	mdev = mt76_alloc_device(&intf->dev, sizeof(*dev), &mt76x2u_ops,
47 				 &drv_ops);
48 	if (!mdev)
49 		return -ENOMEM;
50 
51 	dev = container_of(mdev, struct mt76x02_dev, mt76);
52 
53 	udev = usb_get_dev(udev);
54 	usb_reset_device(udev);
55 
56 	usb_set_intfdata(intf, dev);
57 
58 	mt76x02u_init_mcu(mdev);
59 	err = mt76u_init(mdev, intf, false);
60 	if (err < 0)
61 		goto err;
62 
63 	mdev->rev = mt76_rr(dev, MT_ASIC_VERSION);
64 	dev_info(mdev->dev, "ASIC revision: %08x\n", mdev->rev);
65 	if (!is_mt76x2(dev)) {
66 		err = -ENODEV;
67 		goto err;
68 	}
69 
70 	err = mt76x2u_register_device(dev);
71 	if (err < 0)
72 		goto err;
73 
74 	return 0;
75 
76 err:
77 	ieee80211_free_hw(mt76_hw(dev));
78 	mt76u_deinit(&dev->mt76);
79 	usb_set_intfdata(intf, NULL);
80 	usb_put_dev(udev);
81 
82 	return err;
83 }
84 
85 static void mt76x2u_disconnect(struct usb_interface *intf)
86 {
87 	struct usb_device *udev = interface_to_usbdev(intf);
88 	struct mt76x02_dev *dev = usb_get_intfdata(intf);
89 	struct ieee80211_hw *hw = mt76_hw(dev);
90 
91 	set_bit(MT76_REMOVED, &dev->mphy.state);
92 	ieee80211_unregister_hw(hw);
93 	mt76x2u_cleanup(dev);
94 	mt76u_deinit(&dev->mt76);
95 
96 	ieee80211_free_hw(hw);
97 	usb_set_intfdata(intf, NULL);
98 	usb_put_dev(udev);
99 }
100 
101 static int __maybe_unused mt76x2u_suspend(struct usb_interface *intf,
102 					  pm_message_t state)
103 {
104 	struct mt76x02_dev *dev = usb_get_intfdata(intf);
105 
106 	mt76u_stop_rx(&dev->mt76);
107 
108 	return 0;
109 }
110 
111 static int __maybe_unused mt76x2u_resume(struct usb_interface *intf)
112 {
113 	struct mt76x02_dev *dev = usb_get_intfdata(intf);
114 	int err;
115 
116 	err = mt76u_resume_rx(&dev->mt76);
117 	if (err < 0)
118 		goto err;
119 
120 	err = mt76x2u_init_hardware(dev);
121 	if (err < 0)
122 		goto err;
123 
124 	return 0;
125 
126 err:
127 	mt76x2u_cleanup(dev);
128 	return err;
129 }
130 
131 MODULE_DEVICE_TABLE(usb, mt76x2u_device_table);
132 MODULE_FIRMWARE(MT7662_FIRMWARE);
133 MODULE_FIRMWARE(MT7662_ROM_PATCH);
134 
135 static struct usb_driver mt76x2u_driver = {
136 	.name		= KBUILD_MODNAME,
137 	.id_table	= mt76x2u_device_table,
138 	.probe		= mt76x2u_probe,
139 	.disconnect	= mt76x2u_disconnect,
140 #ifdef CONFIG_PM
141 	.suspend	= mt76x2u_suspend,
142 	.resume		= mt76x2u_resume,
143 	.reset_resume	= mt76x2u_resume,
144 #endif /* CONFIG_PM */
145 	.soft_unbind	= 1,
146 	.disable_hub_initiated_lpm = 1,
147 };
148 module_usb_driver(mt76x2u_driver);
149 
150 MODULE_AUTHOR("Lorenzo Bianconi <lorenzo.bianconi83@gmail.com>");
151 MODULE_LICENSE("Dual BSD/GPL");
152