162306a36Sopenharmony_ci// SPDX-License-Identifier: ISC
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2018 Lorenzo Bianconi <lorenzo.bianconi83@gmail.com>
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#include <linux/kernel.h>
762306a36Sopenharmony_ci#include <linux/module.h>
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include "../mt76x02_usb.h"
1062306a36Sopenharmony_ci#include "mt76x2u.h"
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistatic const struct usb_device_id mt76x2u_device_table[] = {
1362306a36Sopenharmony_ci	{ USB_DEVICE(0x0b05, 0x1833) },	/* Asus USB-AC54 */
1462306a36Sopenharmony_ci	{ USB_DEVICE(0x0b05, 0x17eb) },	/* Asus USB-AC55 */
1562306a36Sopenharmony_ci	{ USB_DEVICE(0x0b05, 0x180b) },	/* Asus USB-N53 B1 */
1662306a36Sopenharmony_ci	{ USB_DEVICE(0x0e8d, 0x7612) },	/* Aukey USBAC1200 - Alfa AWUS036ACM */
1762306a36Sopenharmony_ci	{ USB_DEVICE(0x057c, 0x8503) },	/* Avm FRITZ!WLAN AC860 */
1862306a36Sopenharmony_ci	{ USB_DEVICE(0x7392, 0xb711) },	/* Edimax EW 7722 UAC */
1962306a36Sopenharmony_ci	{ USB_DEVICE(0x0e8d, 0x7632) },	/* HC-M7662BU1 */
2062306a36Sopenharmony_ci	{ USB_DEVICE(0x2c4e, 0x0103) },	/* Mercury UD13 */
2162306a36Sopenharmony_ci	{ USB_DEVICE(0x0846, 0x9053) },	/* Netgear A6210 */
2262306a36Sopenharmony_ci	{ USB_DEVICE(0x045e, 0x02e6) },	/* XBox One Wireless Adapter */
2362306a36Sopenharmony_ci	{ USB_DEVICE(0x045e, 0x02fe) },	/* XBox One Wireless Adapter */
2462306a36Sopenharmony_ci	{ },
2562306a36Sopenharmony_ci};
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_cistatic int mt76x2u_probe(struct usb_interface *intf,
2862306a36Sopenharmony_ci			 const struct usb_device_id *id)
2962306a36Sopenharmony_ci{
3062306a36Sopenharmony_ci	static const struct mt76_driver_ops drv_ops = {
3162306a36Sopenharmony_ci		.drv_flags = MT_DRV_SW_RX_AIRTIME,
3262306a36Sopenharmony_ci		.survey_flags = SURVEY_INFO_TIME_TX,
3362306a36Sopenharmony_ci		.update_survey = mt76x02_update_channel,
3462306a36Sopenharmony_ci		.tx_prepare_skb = mt76x02u_tx_prepare_skb,
3562306a36Sopenharmony_ci		.tx_complete_skb = mt76x02u_tx_complete_skb,
3662306a36Sopenharmony_ci		.tx_status_data = mt76x02_tx_status_data,
3762306a36Sopenharmony_ci		.rx_skb = mt76x02_queue_rx_skb,
3862306a36Sopenharmony_ci		.sta_ps = mt76x02_sta_ps,
3962306a36Sopenharmony_ci		.sta_add = mt76x02_sta_add,
4062306a36Sopenharmony_ci		.sta_remove = mt76x02_sta_remove,
4162306a36Sopenharmony_ci	};
4262306a36Sopenharmony_ci	struct usb_device *udev = interface_to_usbdev(intf);
4362306a36Sopenharmony_ci	struct mt76x02_dev *dev;
4462306a36Sopenharmony_ci	struct mt76_dev *mdev;
4562306a36Sopenharmony_ci	int err;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	mdev = mt76_alloc_device(&intf->dev, sizeof(*dev), &mt76x2u_ops,
4862306a36Sopenharmony_ci				 &drv_ops);
4962306a36Sopenharmony_ci	if (!mdev)
5062306a36Sopenharmony_ci		return -ENOMEM;
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	dev = container_of(mdev, struct mt76x02_dev, mt76);
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	udev = usb_get_dev(udev);
5562306a36Sopenharmony_ci	usb_reset_device(udev);
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	usb_set_intfdata(intf, dev);
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	mt76x02u_init_mcu(mdev);
6062306a36Sopenharmony_ci	err = mt76u_init(mdev, intf);
6162306a36Sopenharmony_ci	if (err < 0)
6262306a36Sopenharmony_ci		goto err;
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	mdev->rev = mt76_rr(dev, MT_ASIC_VERSION);
6562306a36Sopenharmony_ci	dev_info(mdev->dev, "ASIC revision: %08x\n", mdev->rev);
6662306a36Sopenharmony_ci	if (!is_mt76x2(dev)) {
6762306a36Sopenharmony_ci		err = -ENODEV;
6862306a36Sopenharmony_ci		goto err;
6962306a36Sopenharmony_ci	}
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_ci	err = mt76x2u_register_device(dev);
7262306a36Sopenharmony_ci	if (err < 0)
7362306a36Sopenharmony_ci		goto err;
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	return 0;
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_cierr:
7862306a36Sopenharmony_ci	mt76u_queues_deinit(&dev->mt76);
7962306a36Sopenharmony_ci	mt76_free_device(&dev->mt76);
8062306a36Sopenharmony_ci	usb_set_intfdata(intf, NULL);
8162306a36Sopenharmony_ci	usb_put_dev(udev);
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	return err;
8462306a36Sopenharmony_ci}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_cistatic void mt76x2u_disconnect(struct usb_interface *intf)
8762306a36Sopenharmony_ci{
8862306a36Sopenharmony_ci	struct usb_device *udev = interface_to_usbdev(intf);
8962306a36Sopenharmony_ci	struct mt76x02_dev *dev = usb_get_intfdata(intf);
9062306a36Sopenharmony_ci	struct ieee80211_hw *hw = mt76_hw(dev);
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	set_bit(MT76_REMOVED, &dev->mphy.state);
9362306a36Sopenharmony_ci	ieee80211_unregister_hw(hw);
9462306a36Sopenharmony_ci	mt76x2u_cleanup(dev);
9562306a36Sopenharmony_ci	mt76_free_device(&dev->mt76);
9662306a36Sopenharmony_ci	usb_set_intfdata(intf, NULL);
9762306a36Sopenharmony_ci	usb_put_dev(udev);
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic int __maybe_unused mt76x2u_suspend(struct usb_interface *intf,
10162306a36Sopenharmony_ci					  pm_message_t state)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	struct mt76x02_dev *dev = usb_get_intfdata(intf);
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	mt76u_stop_rx(&dev->mt76);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return 0;
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic int __maybe_unused mt76x2u_resume(struct usb_interface *intf)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	struct mt76x02_dev *dev = usb_get_intfdata(intf);
11362306a36Sopenharmony_ci	int err;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	err = mt76u_resume_rx(&dev->mt76);
11662306a36Sopenharmony_ci	if (err < 0)
11762306a36Sopenharmony_ci		goto err;
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	err = mt76x2u_init_hardware(dev);
12062306a36Sopenharmony_ci	if (err < 0)
12162306a36Sopenharmony_ci		goto err;
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	return 0;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_cierr:
12662306a36Sopenharmony_ci	mt76x2u_cleanup(dev);
12762306a36Sopenharmony_ci	return err;
12862306a36Sopenharmony_ci}
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ciMODULE_DEVICE_TABLE(usb, mt76x2u_device_table);
13162306a36Sopenharmony_ciMODULE_FIRMWARE(MT7662_FIRMWARE);
13262306a36Sopenharmony_ciMODULE_FIRMWARE(MT7662_ROM_PATCH);
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_cistatic struct usb_driver mt76x2u_driver = {
13562306a36Sopenharmony_ci	.name		= KBUILD_MODNAME,
13662306a36Sopenharmony_ci	.id_table	= mt76x2u_device_table,
13762306a36Sopenharmony_ci	.probe		= mt76x2u_probe,
13862306a36Sopenharmony_ci	.disconnect	= mt76x2u_disconnect,
13962306a36Sopenharmony_ci#ifdef CONFIG_PM
14062306a36Sopenharmony_ci	.suspend	= mt76x2u_suspend,
14162306a36Sopenharmony_ci	.resume		= mt76x2u_resume,
14262306a36Sopenharmony_ci	.reset_resume	= mt76x2u_resume,
14362306a36Sopenharmony_ci#endif /* CONFIG_PM */
14462306a36Sopenharmony_ci	.soft_unbind	= 1,
14562306a36Sopenharmony_ci	.disable_hub_initiated_lpm = 1,
14662306a36Sopenharmony_ci};
14762306a36Sopenharmony_cimodule_usb_driver(mt76x2u_driver);
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ciMODULE_AUTHOR("Lorenzo Bianconi <lorenzo.bianconi83@gmail.com>");
15062306a36Sopenharmony_ciMODULE_LICENSE("Dual BSD/GPL");
151