cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

usb.c (9372B)


      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
     14static 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
     46static 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
     73static 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
     80static 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
    102static 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
    120static 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
    147static 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
    175static 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
    204out_err:
    205	mt76x0u_cleanup(dev);
    206	return err;
    207}
    208
    209static 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);
    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
    279err:
    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
    288static 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
    305static 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
    317static 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;
    331err:
    332	mt76x0u_cleanup(dev);
    333	return ret;
    334}
    335
    336MODULE_DEVICE_TABLE(usb, mt76x0_device_table);
    337MODULE_FIRMWARE(MT7610E_FIRMWARE);
    338MODULE_FIRMWARE(MT7610U_FIRMWARE);
    339MODULE_LICENSE("GPL");
    340
    341static 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};
    352module_usb_driver(mt76x0_driver);