1// SPDX-License-Identifier: ISC
2/* Copyright (C) 2019 MediaTek Inc.
3 *
4 * Author: Ryder Lee <ryder.lee@mediatek.com>
5 *         Felix Fietkau <nbd@nbd.name>
6 */
7
8#include <linux/kernel.h>
9#include <linux/module.h>
10#include <linux/pci.h>
11
12#include "mt7615.h"
13
14static const struct pci_device_id mt7615_pci_device_table[] = {
15	{ PCI_DEVICE(0x14c3, 0x7615) },
16	{ PCI_DEVICE(0x14c3, 0x7663) },
17	{ PCI_DEVICE(0x14c3, 0x7611) },
18	{ },
19};
20
21static int mt7615_pci_probe(struct pci_dev *pdev,
22			    const struct pci_device_id *id)
23{
24	const u32 *map;
25	int ret;
26
27	ret = pcim_enable_device(pdev);
28	if (ret)
29		return ret;
30
31	ret = pcim_iomap_regions(pdev, BIT(0), pci_name(pdev));
32	if (ret)
33		return ret;
34
35	pci_set_master(pdev);
36
37	ret = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_ALL_TYPES);
38	if (ret < 0)
39		return ret;
40
41	ret = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));
42	if (ret)
43		goto error;
44
45	mt76_pci_disable_aspm(pdev);
46
47	map = id->device == 0x7663 ? mt7663e_reg_map : mt7615e_reg_map;
48	ret = mt7615_mmio_probe(&pdev->dev, pcim_iomap_table(pdev)[0],
49				pdev->irq, map);
50	if (ret)
51		goto error;
52
53	return 0;
54error:
55	pci_free_irq_vectors(pdev);
56
57	return ret;
58}
59
60static void mt7615_pci_remove(struct pci_dev *pdev)
61{
62	struct mt76_dev *mdev = pci_get_drvdata(pdev);
63	struct mt7615_dev *dev = container_of(mdev, struct mt7615_dev, mt76);
64
65	mt7615_unregister_device(dev);
66	devm_free_irq(&pdev->dev, pdev->irq, dev);
67	pci_free_irq_vectors(pdev);
68}
69
70#ifdef CONFIG_PM
71static int mt7615_pci_suspend(struct pci_dev *pdev, pm_message_t state)
72{
73	struct mt76_dev *mdev = pci_get_drvdata(pdev);
74	struct mt7615_dev *dev = container_of(mdev, struct mt7615_dev, mt76);
75	bool hif_suspend;
76	int i, err;
77
78	err = mt7615_pm_wake(dev);
79	if (err < 0)
80		return err;
81
82	hif_suspend = !test_bit(MT76_STATE_SUSPEND, &dev->mphy.state) &&
83		      mt7615_firmware_offload(dev);
84	if (hif_suspend) {
85		err = mt7615_mcu_set_hif_suspend(dev, true);
86		if (err)
87			return err;
88	}
89
90	napi_disable(&mdev->tx_napi);
91	mt76_worker_disable(&mdev->tx_worker);
92
93	mt76_for_each_q_rx(mdev, i) {
94		napi_disable(&mdev->napi[i]);
95	}
96	tasklet_kill(&dev->irq_tasklet);
97
98	mt7615_dma_reset(dev);
99
100	err = mt7615_wait_pdma_busy(dev);
101	if (err)
102		goto restore;
103
104	if (is_mt7663(mdev)) {
105		mt76_set(dev, MT_PDMA_SLP_PROT, MT_PDMA_AXI_SLPPROT_ENABLE);
106		if (!mt76_poll_msec(dev, MT_PDMA_SLP_PROT,
107				    MT_PDMA_AXI_SLPPROT_RDY,
108				    MT_PDMA_AXI_SLPPROT_RDY, 1000)) {
109			dev_err(mdev->dev, "PDMA sleep protection failed\n");
110			err = -EIO;
111			goto restore;
112		}
113	}
114
115	pci_enable_wake(pdev, pci_choose_state(pdev, state), true);
116	pci_save_state(pdev);
117	err = pci_set_power_state(pdev, pci_choose_state(pdev, state));
118	if (err)
119		goto restore;
120
121	err = mt7615_mcu_set_fw_ctrl(dev);
122	if (err)
123		goto restore;
124
125	return 0;
126
127restore:
128	mt76_for_each_q_rx(mdev, i) {
129		napi_enable(&mdev->napi[i]);
130	}
131	napi_enable(&mdev->tx_napi);
132	if (hif_suspend)
133		mt7615_mcu_set_hif_suspend(dev, false);
134
135	return err;
136}
137
138static int mt7615_pci_resume(struct pci_dev *pdev)
139{
140	struct mt76_dev *mdev = pci_get_drvdata(pdev);
141	struct mt7615_dev *dev = container_of(mdev, struct mt7615_dev, mt76);
142	bool pdma_reset;
143	int i, err;
144
145	err = mt7615_mcu_set_drv_ctrl(dev);
146	if (err < 0)
147		return err;
148
149	err = pci_set_power_state(pdev, PCI_D0);
150	if (err)
151		return err;
152
153	pci_restore_state(pdev);
154
155	if (is_mt7663(&dev->mt76)) {
156		mt76_clear(dev, MT_PDMA_SLP_PROT, MT_PDMA_AXI_SLPPROT_ENABLE);
157		mt76_wr(dev, MT_PCIE_IRQ_ENABLE, 1);
158	}
159
160	pdma_reset = !mt76_rr(dev, MT_WPDMA_TX_RING0_CTRL0) &&
161		     !mt76_rr(dev, MT_WPDMA_TX_RING0_CTRL1);
162	if (pdma_reset)
163		dev_err(mdev->dev, "PDMA engine must be reinitialized\n");
164
165	mt76_worker_enable(&mdev->tx_worker);
166	mt76_for_each_q_rx(mdev, i) {
167		napi_enable(&mdev->napi[i]);
168		napi_schedule(&mdev->napi[i]);
169	}
170	napi_enable(&mdev->tx_napi);
171	napi_schedule(&mdev->tx_napi);
172
173	if (!test_bit(MT76_STATE_SUSPEND, &dev->mphy.state) &&
174	    mt7615_firmware_offload(dev))
175		err = mt7615_mcu_set_hif_suspend(dev, false);
176
177	return err;
178}
179#endif /* CONFIG_PM */
180
181struct pci_driver mt7615_pci_driver = {
182	.name		= KBUILD_MODNAME,
183	.id_table	= mt7615_pci_device_table,
184	.probe		= mt7615_pci_probe,
185	.remove		= mt7615_pci_remove,
186#ifdef CONFIG_PM
187	.suspend	= mt7615_pci_suspend,
188	.resume		= mt7615_pci_resume,
189#endif /* CONFIG_PM */
190};
191
192MODULE_DEVICE_TABLE(pci, mt7615_pci_device_table);
193MODULE_FIRMWARE(MT7615_FIRMWARE_CR4);
194MODULE_FIRMWARE(MT7615_FIRMWARE_N9);
195MODULE_FIRMWARE(MT7615_ROM_PATCH);
196MODULE_FIRMWARE(MT7663_OFFLOAD_FIRMWARE_N9);
197MODULE_FIRMWARE(MT7663_OFFLOAD_ROM_PATCH);
198MODULE_FIRMWARE(MT7663_FIRMWARE_N9);
199MODULE_FIRMWARE(MT7663_ROM_PATCH);
200