10e3d6777SRyder Lee // SPDX-License-Identifier: ISC
21613c621SLorenzo Bianconi /*
31613c621SLorenzo Bianconi * Copyright (C) 2016 Felix Fietkau <nbd@nbd.name>
41613c621SLorenzo Bianconi */
51613c621SLorenzo Bianconi
61613c621SLorenzo Bianconi #include <linux/kernel.h>
71613c621SLorenzo Bianconi #include <linux/module.h>
81613c621SLorenzo Bianconi #include <linux/pci.h>
91613c621SLorenzo Bianconi
101613c621SLorenzo Bianconi #include "mt76x2.h"
111613c621SLorenzo Bianconi
126ed942afSLorenzo Bianconi static const struct pci_device_id mt76x2e_device_table[] = {
137715a1d5SRyder Lee { PCI_DEVICE(PCI_VENDOR_ID_MEDIATEK, 0x7662) },
147715a1d5SRyder Lee { PCI_DEVICE(PCI_VENDOR_ID_MEDIATEK, 0x7612) },
157715a1d5SRyder Lee { PCI_DEVICE(PCI_VENDOR_ID_MEDIATEK, 0x7602) },
161613c621SLorenzo Bianconi { },
171613c621SLorenzo Bianconi };
181613c621SLorenzo Bianconi
191613c621SLorenzo Bianconi static int
mt76x2e_probe(struct pci_dev * pdev,const struct pci_device_id * id)206ed942afSLorenzo Bianconi mt76x2e_probe(struct pci_dev *pdev, const struct pci_device_id *id)
211613c621SLorenzo Bianconi {
22ecd25b54SLorenzo Bianconi static const struct mt76_driver_ops drv_ops = {
23ecd25b54SLorenzo Bianconi .txwi_size = sizeof(struct mt76x02_txwi),
245ce09c1aSFelix Fietkau .drv_flags = MT_DRV_TX_ALIGNED4_SKBS |
255ce09c1aSFelix Fietkau MT_DRV_SW_RX_AIRTIME,
26355f8d00SFelix Fietkau .survey_flags = SURVEY_INFO_TIME_TX,
27ecd25b54SLorenzo Bianconi .update_survey = mt76x02_update_channel,
28*f4fdd771SFelix Fietkau .set_channel = mt76x2e_set_channel,
29ecd25b54SLorenzo Bianconi .tx_prepare_skb = mt76x02_tx_prepare_skb,
30ecd25b54SLorenzo Bianconi .tx_complete_skb = mt76x02_tx_complete_skb,
31ecd25b54SLorenzo Bianconi .rx_skb = mt76x02_queue_rx_skb,
32ecd25b54SLorenzo Bianconi .rx_poll_complete = mt76x02_rx_poll_complete,
33ecd25b54SLorenzo Bianconi .sta_ps = mt76x02_sta_ps,
34ecd25b54SLorenzo Bianconi .sta_add = mt76x02_sta_add,
35ecd25b54SLorenzo Bianconi .sta_remove = mt76x02_sta_remove,
36ecd25b54SLorenzo Bianconi };
37e40803f2SLorenzo Bianconi struct mt76x02_dev *dev;
38ecd25b54SLorenzo Bianconi struct mt76_dev *mdev;
391613c621SLorenzo Bianconi int ret;
401613c621SLorenzo Bianconi
411613c621SLorenzo Bianconi ret = pcim_enable_device(pdev);
421613c621SLorenzo Bianconi if (ret)
431613c621SLorenzo Bianconi return ret;
441613c621SLorenzo Bianconi
451613c621SLorenzo Bianconi ret = pcim_iomap_regions(pdev, BIT(0), pci_name(pdev));
461613c621SLorenzo Bianconi if (ret)
471613c621SLorenzo Bianconi return ret;
481613c621SLorenzo Bianconi
491613c621SLorenzo Bianconi pci_set_master(pdev);
501613c621SLorenzo Bianconi
51a2e75961SChristophe JAILLET ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(32));
521613c621SLorenzo Bianconi if (ret)
531613c621SLorenzo Bianconi return ret;
541613c621SLorenzo Bianconi
55ecd25b54SLorenzo Bianconi mdev = mt76_alloc_device(&pdev->dev, sizeof(*dev), &mt76x2_ops,
56ecd25b54SLorenzo Bianconi &drv_ops);
57ecd25b54SLorenzo Bianconi if (!mdev)
581613c621SLorenzo Bianconi return -ENOMEM;
591613c621SLorenzo Bianconi
60ecd25b54SLorenzo Bianconi dev = container_of(mdev, struct mt76x02_dev, mt76);
61ecd25b54SLorenzo Bianconi mt76_mmio_init(mdev, pcim_iomap_table(pdev)[0]);
621613c621SLorenzo Bianconi mt76x2_reset_wlan(dev, false);
631613c621SLorenzo Bianconi
64ecd25b54SLorenzo Bianconi mdev->rev = mt76_rr(dev, MT_ASIC_VERSION);
65ecd25b54SLorenzo Bianconi dev_info(mdev->dev, "ASIC revision: %08x\n", mdev->rev);
661613c621SLorenzo Bianconi
679df8c539SFelix Fietkau mt76_wr(dev, MT_INT_MASK_CSR, 0);
689df8c539SFelix Fietkau
69ecd25b54SLorenzo Bianconi ret = devm_request_irq(mdev->dev, pdev->irq, mt76x02_irq_handler,
701613c621SLorenzo Bianconi IRQF_SHARED, KBUILD_MODNAME, dev);
711613c621SLorenzo Bianconi if (ret)
721613c621SLorenzo Bianconi goto error;
731613c621SLorenzo Bianconi
741613c621SLorenzo Bianconi ret = mt76x2_register_device(dev);
751613c621SLorenzo Bianconi if (ret)
761613c621SLorenzo Bianconi goto error;
771613c621SLorenzo Bianconi
781613c621SLorenzo Bianconi /* Fix up ASPM configuration */
791613c621SLorenzo Bianconi
801613c621SLorenzo Bianconi /* RG_SSUSB_G1_CDR_BIR_LTR = 0x9 */
811613c621SLorenzo Bianconi mt76_rmw_field(dev, 0x15a10, 0x1f << 16, 0x9);
821613c621SLorenzo Bianconi
831613c621SLorenzo Bianconi /* RG_SSUSB_G1_CDR_BIC_LTR = 0xf */
84dbc2b176SBorislav Petkov mt76_rmw_field(dev, 0x15a0c, 0xfU << 28, 0xf);
851613c621SLorenzo Bianconi
861613c621SLorenzo Bianconi /* RG_SSUSB_CDR_BR_PE1D = 0x3 */
871613c621SLorenzo Bianconi mt76_rmw_field(dev, 0x15c58, 0x3 << 6, 0x3);
881613c621SLorenzo Bianconi
89f37f0550SLorenzo Bianconi mt76_pci_disable_aspm(pdev);
90f37f0550SLorenzo Bianconi
911613c621SLorenzo Bianconi return 0;
921613c621SLorenzo Bianconi
931613c621SLorenzo Bianconi error:
94bc348defSLorenzo Bianconi mt76_free_device(&dev->mt76);
95bc348defSLorenzo Bianconi
961613c621SLorenzo Bianconi return ret;
971613c621SLorenzo Bianconi }
981613c621SLorenzo Bianconi
991613c621SLorenzo Bianconi static void
mt76x2e_remove(struct pci_dev * pdev)1006ed942afSLorenzo Bianconi mt76x2e_remove(struct pci_dev *pdev)
1011613c621SLorenzo Bianconi {
1021613c621SLorenzo Bianconi struct mt76_dev *mdev = pci_get_drvdata(pdev);
103e40803f2SLorenzo Bianconi struct mt76x02_dev *dev = container_of(mdev, struct mt76x02_dev, mt76);
1041613c621SLorenzo Bianconi
1051613c621SLorenzo Bianconi mt76_unregister_device(mdev);
1061613c621SLorenzo Bianconi mt76x2_cleanup(dev);
107def34a2fSLorenzo Bianconi mt76_free_device(mdev);
1081613c621SLorenzo Bianconi }
1091613c621SLorenzo Bianconi
11040ba9a93SLorenzo Bianconi static int __maybe_unused
mt76x2e_suspend(struct pci_dev * pdev,pm_message_t state)11140ba9a93SLorenzo Bianconi mt76x2e_suspend(struct pci_dev *pdev, pm_message_t state)
11240ba9a93SLorenzo Bianconi {
11340ba9a93SLorenzo Bianconi struct mt76_dev *mdev = pci_get_drvdata(pdev);
11440ba9a93SLorenzo Bianconi int i, err;
11540ba9a93SLorenzo Bianconi
11640ba9a93SLorenzo Bianconi napi_disable(&mdev->tx_napi);
11740ba9a93SLorenzo Bianconi tasklet_kill(&mdev->pre_tbtt_tasklet);
118781eef5bSFelix Fietkau mt76_worker_disable(&mdev->tx_worker);
11940ba9a93SLorenzo Bianconi
12040ba9a93SLorenzo Bianconi mt76_for_each_q_rx(mdev, i)
12140ba9a93SLorenzo Bianconi napi_disable(&mdev->napi[i]);
12240ba9a93SLorenzo Bianconi
12340ba9a93SLorenzo Bianconi pci_enable_wake(pdev, pci_choose_state(pdev, state), true);
12440ba9a93SLorenzo Bianconi pci_save_state(pdev);
12540ba9a93SLorenzo Bianconi err = pci_set_power_state(pdev, pci_choose_state(pdev, state));
12640ba9a93SLorenzo Bianconi if (err)
12740ba9a93SLorenzo Bianconi goto restore;
12840ba9a93SLorenzo Bianconi
12940ba9a93SLorenzo Bianconi return 0;
13040ba9a93SLorenzo Bianconi
13140ba9a93SLorenzo Bianconi restore:
13240ba9a93SLorenzo Bianconi mt76_for_each_q_rx(mdev, i)
13340ba9a93SLorenzo Bianconi napi_enable(&mdev->napi[i]);
13440ba9a93SLorenzo Bianconi napi_enable(&mdev->tx_napi);
13540ba9a93SLorenzo Bianconi
13640ba9a93SLorenzo Bianconi return err;
13740ba9a93SLorenzo Bianconi }
13840ba9a93SLorenzo Bianconi
13940ba9a93SLorenzo Bianconi static int __maybe_unused
mt76x2e_resume(struct pci_dev * pdev)14040ba9a93SLorenzo Bianconi mt76x2e_resume(struct pci_dev *pdev)
14140ba9a93SLorenzo Bianconi {
14240ba9a93SLorenzo Bianconi struct mt76_dev *mdev = pci_get_drvdata(pdev);
14340ba9a93SLorenzo Bianconi struct mt76x02_dev *dev = container_of(mdev, struct mt76x02_dev, mt76);
14440ba9a93SLorenzo Bianconi int i, err;
14540ba9a93SLorenzo Bianconi
14640ba9a93SLorenzo Bianconi err = pci_set_power_state(pdev, PCI_D0);
14740ba9a93SLorenzo Bianconi if (err)
14840ba9a93SLorenzo Bianconi return err;
14940ba9a93SLorenzo Bianconi
15040ba9a93SLorenzo Bianconi pci_restore_state(pdev);
15140ba9a93SLorenzo Bianconi
152781eef5bSFelix Fietkau mt76_worker_enable(&mdev->tx_worker);
153970be1dfSFelix Fietkau
154970be1dfSFelix Fietkau local_bh_disable();
15540ba9a93SLorenzo Bianconi mt76_for_each_q_rx(mdev, i) {
15640ba9a93SLorenzo Bianconi napi_enable(&mdev->napi[i]);
15740ba9a93SLorenzo Bianconi napi_schedule(&mdev->napi[i]);
15840ba9a93SLorenzo Bianconi }
15940ba9a93SLorenzo Bianconi napi_enable(&mdev->tx_napi);
16040ba9a93SLorenzo Bianconi napi_schedule(&mdev->tx_napi);
161970be1dfSFelix Fietkau local_bh_enable();
16240ba9a93SLorenzo Bianconi
16340ba9a93SLorenzo Bianconi return mt76x2_resume_device(dev);
16440ba9a93SLorenzo Bianconi }
16540ba9a93SLorenzo Bianconi
1666ed942afSLorenzo Bianconi MODULE_DEVICE_TABLE(pci, mt76x2e_device_table);
1671613c621SLorenzo Bianconi MODULE_FIRMWARE(MT7662_FIRMWARE);
1681613c621SLorenzo Bianconi MODULE_FIRMWARE(MT7662_ROM_PATCH);
169f3f8f050SBreno Leitao MODULE_DESCRIPTION("MediaTek MT76x2E (PCIe) wireless driver");
1701613c621SLorenzo Bianconi MODULE_LICENSE("Dual BSD/GPL");
1711613c621SLorenzo Bianconi
1721613c621SLorenzo Bianconi static struct pci_driver mt76pci_driver = {
1731613c621SLorenzo Bianconi .name = KBUILD_MODNAME,
1746ed942afSLorenzo Bianconi .id_table = mt76x2e_device_table,
1756ed942afSLorenzo Bianconi .probe = mt76x2e_probe,
1766ed942afSLorenzo Bianconi .remove = mt76x2e_remove,
17740ba9a93SLorenzo Bianconi #ifdef CONFIG_PM
17840ba9a93SLorenzo Bianconi .suspend = mt76x2e_suspend,
17940ba9a93SLorenzo Bianconi .resume = mt76x2e_resume,
18040ba9a93SLorenzo Bianconi #endif /* CONFIG_PM */
1811613c621SLorenzo Bianconi };
1821613c621SLorenzo Bianconi
1831613c621SLorenzo Bianconi module_pci_driver(mt76pci_driver);
184