1// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2/* Copyright(c) 2020  Realtek Corporation
3 */
4
5#include <linux/pci.h>
6
7#include "mac.h"
8#include "pci.h"
9#include "reg.h"
10#include "ser.h"
11
12static bool rtw89_pci_disable_clkreq;
13static bool rtw89_pci_disable_aspm_l1;
14static bool rtw89_pci_disable_l1ss;
15module_param_named(disable_clkreq, rtw89_pci_disable_clkreq, bool, 0644);
16module_param_named(disable_aspm_l1, rtw89_pci_disable_aspm_l1, bool, 0644);
17module_param_named(disable_aspm_l1ss, rtw89_pci_disable_l1ss, bool, 0644);
18MODULE_PARM_DESC(disable_clkreq, "Set Y to disable PCI clkreq support");
19MODULE_PARM_DESC(disable_aspm_l1, "Set Y to disable PCI ASPM L1 support");
20MODULE_PARM_DESC(disable_aspm_l1ss, "Set Y to disable PCI L1SS support");
21
22static int rtw89_pci_rst_bdram_ax(struct rtw89_dev *rtwdev)
23{
24	u32 val;
25	int ret;
26
27	rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_RST_BDRAM);
28
29	ret = read_poll_timeout_atomic(rtw89_read32, val, !(val & B_AX_RST_BDRAM),
30				       1, RTW89_PCI_POLL_BDRAM_RST_CNT, false,
31				       rtwdev, R_AX_PCIE_INIT_CFG1);
32
33	return ret;
34}
35
36static u32 rtw89_pci_dma_recalc(struct rtw89_dev *rtwdev,
37				struct rtw89_pci_dma_ring *bd_ring,
38				u32 cur_idx, bool tx)
39{
40	const struct rtw89_pci_info *info = rtwdev->pci_info;
41	u32 cnt, cur_rp, wp, rp, len;
42
43	rp = bd_ring->rp;
44	wp = bd_ring->wp;
45	len = bd_ring->len;
46
47	cur_rp = FIELD_GET(TXBD_HW_IDX_MASK, cur_idx);
48	if (tx) {
49		cnt = cur_rp >= rp ? cur_rp - rp : len - (rp - cur_rp);
50	} else {
51		if (info->rx_ring_eq_is_full)
52			wp += 1;
53
54		cnt = cur_rp >= wp ? cur_rp - wp : len - (wp - cur_rp);
55	}
56
57	bd_ring->rp = cur_rp;
58
59	return cnt;
60}
61
62static u32 rtw89_pci_txbd_recalc(struct rtw89_dev *rtwdev,
63				 struct rtw89_pci_tx_ring *tx_ring)
64{
65	struct rtw89_pci_dma_ring *bd_ring = &tx_ring->bd_ring;
66	u32 addr_idx = bd_ring->addr.idx;
67	u32 cnt, idx;
68
69	idx = rtw89_read32(rtwdev, addr_idx);
70	cnt = rtw89_pci_dma_recalc(rtwdev, bd_ring, idx, true);
71
72	return cnt;
73}
74
75static void rtw89_pci_release_fwcmd(struct rtw89_dev *rtwdev,
76				    struct rtw89_pci *rtwpci,
77				    u32 cnt, bool release_all)
78{
79	struct rtw89_pci_tx_data *tx_data;
80	struct sk_buff *skb;
81	u32 qlen;
82
83	while (cnt--) {
84		skb = skb_dequeue(&rtwpci->h2c_queue);
85		if (!skb) {
86			rtw89_err(rtwdev, "failed to pre-release fwcmd\n");
87			return;
88		}
89		skb_queue_tail(&rtwpci->h2c_release_queue, skb);
90	}
91
92	qlen = skb_queue_len(&rtwpci->h2c_release_queue);
93	if (!release_all)
94	       qlen = qlen > RTW89_PCI_MULTITAG ? qlen - RTW89_PCI_MULTITAG : 0;
95
96	while (qlen--) {
97		skb = skb_dequeue(&rtwpci->h2c_release_queue);
98		if (!skb) {
99			rtw89_err(rtwdev, "failed to release fwcmd\n");
100			return;
101		}
102		tx_data = RTW89_PCI_TX_SKB_CB(skb);
103		dma_unmap_single(&rtwpci->pdev->dev, tx_data->dma, skb->len,
104				 DMA_TO_DEVICE);
105		dev_kfree_skb_any(skb);
106	}
107}
108
109static void rtw89_pci_reclaim_tx_fwcmd(struct rtw89_dev *rtwdev,
110				       struct rtw89_pci *rtwpci)
111{
112	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[RTW89_TXCH_CH12];
113	u32 cnt;
114
115	cnt = rtw89_pci_txbd_recalc(rtwdev, tx_ring);
116	if (!cnt)
117		return;
118	rtw89_pci_release_fwcmd(rtwdev, rtwpci, cnt, false);
119}
120
121static u32 rtw89_pci_rxbd_recalc(struct rtw89_dev *rtwdev,
122				 struct rtw89_pci_rx_ring *rx_ring)
123{
124	struct rtw89_pci_dma_ring *bd_ring = &rx_ring->bd_ring;
125	u32 addr_idx = bd_ring->addr.idx;
126	u32 cnt, idx;
127
128	idx = rtw89_read32(rtwdev, addr_idx);
129	cnt = rtw89_pci_dma_recalc(rtwdev, bd_ring, idx, false);
130
131	return cnt;
132}
133
134static void rtw89_pci_sync_skb_for_cpu(struct rtw89_dev *rtwdev,
135				       struct sk_buff *skb)
136{
137	struct rtw89_pci_rx_info *rx_info;
138	dma_addr_t dma;
139
140	rx_info = RTW89_PCI_RX_SKB_CB(skb);
141	dma = rx_info->dma;
142	dma_sync_single_for_cpu(rtwdev->dev, dma, RTW89_PCI_RX_BUF_SIZE,
143				DMA_FROM_DEVICE);
144}
145
146static void rtw89_pci_sync_skb_for_device(struct rtw89_dev *rtwdev,
147					  struct sk_buff *skb)
148{
149	struct rtw89_pci_rx_info *rx_info;
150	dma_addr_t dma;
151
152	rx_info = RTW89_PCI_RX_SKB_CB(skb);
153	dma = rx_info->dma;
154	dma_sync_single_for_device(rtwdev->dev, dma, RTW89_PCI_RX_BUF_SIZE,
155				   DMA_FROM_DEVICE);
156}
157
158static void rtw89_pci_rxbd_info_update(struct rtw89_dev *rtwdev,
159				       struct sk_buff *skb)
160{
161	struct rtw89_pci_rxbd_info *rxbd_info;
162	struct rtw89_pci_rx_info *rx_info = RTW89_PCI_RX_SKB_CB(skb);
163
164	rxbd_info = (struct rtw89_pci_rxbd_info *)skb->data;
165	rx_info->fs = le32_get_bits(rxbd_info->dword, RTW89_PCI_RXBD_FS);
166	rx_info->ls = le32_get_bits(rxbd_info->dword, RTW89_PCI_RXBD_LS);
167	rx_info->len = le32_get_bits(rxbd_info->dword, RTW89_PCI_RXBD_WRITE_SIZE);
168	rx_info->tag = le32_get_bits(rxbd_info->dword, RTW89_PCI_RXBD_TAG);
169}
170
171static int rtw89_pci_validate_rx_tag(struct rtw89_dev *rtwdev,
172				     struct rtw89_pci_rx_ring *rx_ring,
173				     struct sk_buff *skb)
174{
175	struct rtw89_pci_rx_info *rx_info = RTW89_PCI_RX_SKB_CB(skb);
176	const struct rtw89_pci_info *info = rtwdev->pci_info;
177	u32 target_rx_tag;
178
179	if (!info->check_rx_tag)
180		return 0;
181
182	/* valid range is 1 ~ 0x1FFF */
183	if (rx_ring->target_rx_tag == 0)
184		target_rx_tag = 1;
185	else
186		target_rx_tag = rx_ring->target_rx_tag;
187
188	if (rx_info->tag != target_rx_tag) {
189		rtw89_debug(rtwdev, RTW89_DBG_UNEXP, "mismatch RX tag 0x%x 0x%x\n",
190			    rx_info->tag, target_rx_tag);
191		return -EAGAIN;
192	}
193
194	return 0;
195}
196
197static
198int rtw89_pci_sync_skb_for_device_and_validate_rx_info(struct rtw89_dev *rtwdev,
199						       struct rtw89_pci_rx_ring *rx_ring,
200						       struct sk_buff *skb)
201{
202	struct rtw89_pci_rx_info *rx_info = RTW89_PCI_RX_SKB_CB(skb);
203	int rx_tag_retry = 100;
204	int ret;
205
206	do {
207		rtw89_pci_sync_skb_for_cpu(rtwdev, skb);
208		rtw89_pci_rxbd_info_update(rtwdev, skb);
209
210		ret = rtw89_pci_validate_rx_tag(rtwdev, rx_ring, skb);
211		if (ret != -EAGAIN)
212			break;
213	} while (rx_tag_retry--);
214
215	/* update target rx_tag for next RX */
216	rx_ring->target_rx_tag = rx_info->tag + 1;
217
218	return ret;
219}
220
221static void rtw89_pci_ctrl_txdma_ch_ax(struct rtw89_dev *rtwdev, bool enable)
222{
223	const struct rtw89_pci_info *info = rtwdev->pci_info;
224	const struct rtw89_reg_def *dma_stop1 = &info->dma_stop1;
225	const struct rtw89_reg_def *dma_stop2 = &info->dma_stop2;
226
227	if (enable) {
228		rtw89_write32_clr(rtwdev, dma_stop1->addr, dma_stop1->mask);
229		if (dma_stop2->addr)
230			rtw89_write32_clr(rtwdev, dma_stop2->addr, dma_stop2->mask);
231	} else {
232		rtw89_write32_set(rtwdev, dma_stop1->addr, dma_stop1->mask);
233		if (dma_stop2->addr)
234			rtw89_write32_set(rtwdev, dma_stop2->addr, dma_stop2->mask);
235	}
236}
237
238static void rtw89_pci_ctrl_txdma_fw_ch_ax(struct rtw89_dev *rtwdev, bool enable)
239{
240	const struct rtw89_pci_info *info = rtwdev->pci_info;
241	const struct rtw89_reg_def *dma_stop1 = &info->dma_stop1;
242
243	if (enable)
244		rtw89_write32_clr(rtwdev, dma_stop1->addr, B_AX_STOP_CH12);
245	else
246		rtw89_write32_set(rtwdev, dma_stop1->addr, B_AX_STOP_CH12);
247}
248
249static bool
250rtw89_skb_put_rx_data(struct rtw89_dev *rtwdev, bool fs, bool ls,
251		      struct sk_buff *new,
252		      const struct sk_buff *skb, u32 offset,
253		      const struct rtw89_pci_rx_info *rx_info,
254		      const struct rtw89_rx_desc_info *desc_info)
255{
256	u32 copy_len = rx_info->len - offset;
257
258	if (unlikely(skb_tailroom(new) < copy_len)) {
259		rtw89_debug(rtwdev, RTW89_DBG_TXRX,
260			    "invalid rx data length bd_len=%d desc_len=%d offset=%d (fs=%d ls=%d)\n",
261			    rx_info->len, desc_info->pkt_size, offset, fs, ls);
262		rtw89_hex_dump(rtwdev, RTW89_DBG_TXRX, "rx_data: ",
263			       skb->data, rx_info->len);
264		/* length of a single segment skb is desc_info->pkt_size */
265		if (fs && ls) {
266			copy_len = desc_info->pkt_size;
267		} else {
268			rtw89_info(rtwdev, "drop rx data due to invalid length\n");
269			return false;
270		}
271	}
272
273	skb_put_data(new, skb->data + offset, copy_len);
274
275	return true;
276}
277
278static u32 rtw89_pci_get_rx_skb_idx(struct rtw89_dev *rtwdev,
279				    struct rtw89_pci_dma_ring *bd_ring)
280{
281	const struct rtw89_pci_info *info = rtwdev->pci_info;
282	u32 wp = bd_ring->wp;
283
284	if (!info->rx_ring_eq_is_full)
285		return wp;
286
287	if (++wp >= bd_ring->len)
288		wp = 0;
289
290	return wp;
291}
292
293static u32 rtw89_pci_rxbd_deliver_skbs(struct rtw89_dev *rtwdev,
294				       struct rtw89_pci_rx_ring *rx_ring)
295{
296	struct rtw89_pci_dma_ring *bd_ring = &rx_ring->bd_ring;
297	struct rtw89_pci_rx_info *rx_info;
298	struct rtw89_rx_desc_info *desc_info = &rx_ring->diliver_desc;
299	struct sk_buff *new = rx_ring->diliver_skb;
300	struct sk_buff *skb;
301	u32 rxinfo_size = sizeof(struct rtw89_pci_rxbd_info);
302	u32 skb_idx;
303	u32 offset;
304	u32 cnt = 1;
305	bool fs, ls;
306	int ret;
307
308	skb_idx = rtw89_pci_get_rx_skb_idx(rtwdev, bd_ring);
309	skb = rx_ring->buf[skb_idx];
310
311	ret = rtw89_pci_sync_skb_for_device_and_validate_rx_info(rtwdev, rx_ring, skb);
312	if (ret) {
313		rtw89_err(rtwdev, "failed to update %d RXBD info: %d\n",
314			  bd_ring->wp, ret);
315		goto err_sync_device;
316	}
317
318	rx_info = RTW89_PCI_RX_SKB_CB(skb);
319	fs = rx_info->fs;
320	ls = rx_info->ls;
321
322	if (fs) {
323		if (new) {
324			rtw89_debug(rtwdev, RTW89_DBG_UNEXP,
325				    "skb should not be ready before first segment start\n");
326			goto err_sync_device;
327		}
328		if (desc_info->ready) {
329			rtw89_warn(rtwdev, "desc info should not be ready before first segment start\n");
330			goto err_sync_device;
331		}
332
333		rtw89_chip_query_rxdesc(rtwdev, desc_info, skb->data, rxinfo_size);
334
335		new = rtw89_alloc_skb_for_rx(rtwdev, desc_info->pkt_size);
336		if (!new)
337			goto err_sync_device;
338
339		rx_ring->diliver_skb = new;
340
341		/* first segment has RX desc */
342		offset = desc_info->offset + desc_info->rxd_len;
343	} else {
344		offset = sizeof(struct rtw89_pci_rxbd_info);
345		if (!new) {
346			rtw89_debug(rtwdev, RTW89_DBG_UNEXP, "no last skb\n");
347			goto err_sync_device;
348		}
349	}
350	if (!rtw89_skb_put_rx_data(rtwdev, fs, ls, new, skb, offset, rx_info, desc_info))
351		goto err_sync_device;
352	rtw89_pci_sync_skb_for_device(rtwdev, skb);
353	rtw89_pci_rxbd_increase(rx_ring, 1);
354
355	if (!desc_info->ready) {
356		rtw89_warn(rtwdev, "no rx desc information\n");
357		goto err_free_resource;
358	}
359	if (ls) {
360		rtw89_core_rx(rtwdev, desc_info, new);
361		rx_ring->diliver_skb = NULL;
362		desc_info->ready = false;
363	}
364
365	return cnt;
366
367err_sync_device:
368	rtw89_pci_sync_skb_for_device(rtwdev, skb);
369	rtw89_pci_rxbd_increase(rx_ring, 1);
370err_free_resource:
371	if (new)
372		dev_kfree_skb_any(new);
373	rx_ring->diliver_skb = NULL;
374	desc_info->ready = false;
375
376	return cnt;
377}
378
379static void rtw89_pci_rxbd_deliver(struct rtw89_dev *rtwdev,
380				   struct rtw89_pci_rx_ring *rx_ring,
381				   u32 cnt)
382{
383	struct rtw89_pci_dma_ring *bd_ring = &rx_ring->bd_ring;
384	u32 rx_cnt;
385
386	while (cnt && rtwdev->napi_budget_countdown > 0) {
387		rx_cnt = rtw89_pci_rxbd_deliver_skbs(rtwdev, rx_ring);
388		if (!rx_cnt) {
389			rtw89_err(rtwdev, "failed to deliver RXBD skb\n");
390
391			/* skip the rest RXBD bufs */
392			rtw89_pci_rxbd_increase(rx_ring, cnt);
393			break;
394		}
395
396		cnt -= rx_cnt;
397	}
398
399	rtw89_write16(rtwdev, bd_ring->addr.idx, bd_ring->wp);
400}
401
402static int rtw89_pci_poll_rxq_dma(struct rtw89_dev *rtwdev,
403				  struct rtw89_pci *rtwpci, int budget)
404{
405	struct rtw89_pci_rx_ring *rx_ring;
406	int countdown = rtwdev->napi_budget_countdown;
407	u32 cnt;
408
409	rx_ring = &rtwpci->rx_rings[RTW89_RXCH_RXQ];
410
411	cnt = rtw89_pci_rxbd_recalc(rtwdev, rx_ring);
412	if (!cnt)
413		return 0;
414
415	cnt = min_t(u32, budget, cnt);
416
417	rtw89_pci_rxbd_deliver(rtwdev, rx_ring, cnt);
418
419	/* In case of flushing pending SKBs, the countdown may exceed. */
420	if (rtwdev->napi_budget_countdown <= 0)
421		return budget;
422
423	return budget - countdown;
424}
425
426static void rtw89_pci_tx_status(struct rtw89_dev *rtwdev,
427				struct rtw89_pci_tx_ring *tx_ring,
428				struct sk_buff *skb, u8 tx_status)
429{
430	struct rtw89_tx_skb_data *skb_data = RTW89_TX_SKB_CB(skb);
431	struct ieee80211_tx_info *info;
432
433	rtw89_core_tx_wait_complete(rtwdev, skb_data, tx_status == RTW89_TX_DONE);
434
435	info = IEEE80211_SKB_CB(skb);
436	ieee80211_tx_info_clear_status(info);
437
438	if (info->flags & IEEE80211_TX_CTL_NO_ACK)
439		info->flags |= IEEE80211_TX_STAT_NOACK_TRANSMITTED;
440	if (tx_status == RTW89_TX_DONE) {
441		info->flags |= IEEE80211_TX_STAT_ACK;
442		tx_ring->tx_acked++;
443	} else {
444		if (info->flags & IEEE80211_TX_CTL_REQ_TX_STATUS)
445			rtw89_debug(rtwdev, RTW89_DBG_FW,
446				    "failed to TX of status %x\n", tx_status);
447		switch (tx_status) {
448		case RTW89_TX_RETRY_LIMIT:
449			tx_ring->tx_retry_lmt++;
450			break;
451		case RTW89_TX_LIFE_TIME:
452			tx_ring->tx_life_time++;
453			break;
454		case RTW89_TX_MACID_DROP:
455			tx_ring->tx_mac_id_drop++;
456			break;
457		default:
458			rtw89_warn(rtwdev, "invalid TX status %x\n", tx_status);
459			break;
460		}
461	}
462
463	ieee80211_tx_status_ni(rtwdev->hw, skb);
464}
465
466static void rtw89_pci_reclaim_txbd(struct rtw89_dev *rtwdev, struct rtw89_pci_tx_ring *tx_ring)
467{
468	struct rtw89_pci_tx_wd *txwd;
469	u32 cnt;
470
471	cnt = rtw89_pci_txbd_recalc(rtwdev, tx_ring);
472	while (cnt--) {
473		txwd = list_first_entry_or_null(&tx_ring->busy_pages, struct rtw89_pci_tx_wd, list);
474		if (!txwd) {
475			rtw89_warn(rtwdev, "No busy txwd pages available\n");
476			break;
477		}
478
479		list_del_init(&txwd->list);
480
481		/* this skb has been freed by RPP */
482		if (skb_queue_len(&txwd->queue) == 0)
483			rtw89_pci_enqueue_txwd(tx_ring, txwd);
484	}
485}
486
487static void rtw89_pci_release_busy_txwd(struct rtw89_dev *rtwdev,
488					struct rtw89_pci_tx_ring *tx_ring)
489{
490	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
491	struct rtw89_pci_tx_wd *txwd;
492	int i;
493
494	for (i = 0; i < wd_ring->page_num; i++) {
495		txwd = list_first_entry_or_null(&tx_ring->busy_pages, struct rtw89_pci_tx_wd, list);
496		if (!txwd)
497			break;
498
499		list_del_init(&txwd->list);
500	}
501}
502
503static void rtw89_pci_release_txwd_skb(struct rtw89_dev *rtwdev,
504				       struct rtw89_pci_tx_ring *tx_ring,
505				       struct rtw89_pci_tx_wd *txwd, u16 seq,
506				       u8 tx_status)
507{
508	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
509	struct rtw89_pci_tx_data *tx_data;
510	struct sk_buff *skb, *tmp;
511	u8 txch = tx_ring->txch;
512
513	if (!list_empty(&txwd->list)) {
514		rtw89_pci_reclaim_txbd(rtwdev, tx_ring);
515		/* In low power mode, RPP can receive before updating of TX BD.
516		 * In normal mode, it should not happen so give it a warning.
517		 */
518		if (!rtwpci->low_power && !list_empty(&txwd->list))
519			rtw89_warn(rtwdev, "queue %d txwd %d is not idle\n",
520				   txch, seq);
521	}
522
523	skb_queue_walk_safe(&txwd->queue, skb, tmp) {
524		skb_unlink(skb, &txwd->queue);
525
526		tx_data = RTW89_PCI_TX_SKB_CB(skb);
527		dma_unmap_single(&rtwpci->pdev->dev, tx_data->dma, skb->len,
528				 DMA_TO_DEVICE);
529
530		rtw89_pci_tx_status(rtwdev, tx_ring, skb, tx_status);
531	}
532
533	if (list_empty(&txwd->list))
534		rtw89_pci_enqueue_txwd(tx_ring, txwd);
535}
536
537static void rtw89_pci_release_rpp(struct rtw89_dev *rtwdev,
538				  struct rtw89_pci_rpp_fmt *rpp)
539{
540	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
541	struct rtw89_pci_tx_ring *tx_ring;
542	struct rtw89_pci_tx_wd_ring *wd_ring;
543	struct rtw89_pci_tx_wd *txwd;
544	u16 seq;
545	u8 qsel, tx_status, txch;
546
547	seq = le32_get_bits(rpp->dword, RTW89_PCI_RPP_SEQ);
548	qsel = le32_get_bits(rpp->dword, RTW89_PCI_RPP_QSEL);
549	tx_status = le32_get_bits(rpp->dword, RTW89_PCI_RPP_TX_STATUS);
550	txch = rtw89_core_get_ch_dma(rtwdev, qsel);
551
552	if (txch == RTW89_TXCH_CH12) {
553		rtw89_warn(rtwdev, "should no fwcmd release report\n");
554		return;
555	}
556
557	tx_ring = &rtwpci->tx_rings[txch];
558	wd_ring = &tx_ring->wd_ring;
559	txwd = &wd_ring->pages[seq];
560
561	rtw89_pci_release_txwd_skb(rtwdev, tx_ring, txwd, seq, tx_status);
562}
563
564static void rtw89_pci_release_pending_txwd_skb(struct rtw89_dev *rtwdev,
565					       struct rtw89_pci_tx_ring *tx_ring)
566{
567	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
568	struct rtw89_pci_tx_wd *txwd;
569	int i;
570
571	for (i = 0; i < wd_ring->page_num; i++) {
572		txwd = &wd_ring->pages[i];
573
574		if (!list_empty(&txwd->list))
575			continue;
576
577		rtw89_pci_release_txwd_skb(rtwdev, tx_ring, txwd, i, RTW89_TX_MACID_DROP);
578	}
579}
580
581static u32 rtw89_pci_release_tx_skbs(struct rtw89_dev *rtwdev,
582				     struct rtw89_pci_rx_ring *rx_ring,
583				     u32 max_cnt)
584{
585	struct rtw89_pci_dma_ring *bd_ring = &rx_ring->bd_ring;
586	struct rtw89_pci_rx_info *rx_info;
587	struct rtw89_pci_rpp_fmt *rpp;
588	struct rtw89_rx_desc_info desc_info = {};
589	struct sk_buff *skb;
590	u32 cnt = 0;
591	u32 rpp_size = sizeof(struct rtw89_pci_rpp_fmt);
592	u32 rxinfo_size = sizeof(struct rtw89_pci_rxbd_info);
593	u32 skb_idx;
594	u32 offset;
595	int ret;
596
597	skb_idx = rtw89_pci_get_rx_skb_idx(rtwdev, bd_ring);
598	skb = rx_ring->buf[skb_idx];
599
600	ret = rtw89_pci_sync_skb_for_device_and_validate_rx_info(rtwdev, rx_ring, skb);
601	if (ret) {
602		rtw89_err(rtwdev, "failed to update %d RXBD info: %d\n",
603			  bd_ring->wp, ret);
604		goto err_sync_device;
605	}
606
607	rx_info = RTW89_PCI_RX_SKB_CB(skb);
608	if (!rx_info->fs || !rx_info->ls) {
609		rtw89_err(rtwdev, "cannot process RP frame not set FS/LS\n");
610		return cnt;
611	}
612
613	rtw89_chip_query_rxdesc(rtwdev, &desc_info, skb->data, rxinfo_size);
614
615	/* first segment has RX desc */
616	offset = desc_info.offset + desc_info.rxd_len;
617	for (; offset + rpp_size <= rx_info->len; offset += rpp_size) {
618		rpp = (struct rtw89_pci_rpp_fmt *)(skb->data + offset);
619		rtw89_pci_release_rpp(rtwdev, rpp);
620	}
621
622	rtw89_pci_sync_skb_for_device(rtwdev, skb);
623	rtw89_pci_rxbd_increase(rx_ring, 1);
624	cnt++;
625
626	return cnt;
627
628err_sync_device:
629	rtw89_pci_sync_skb_for_device(rtwdev, skb);
630	return 0;
631}
632
633static void rtw89_pci_release_tx(struct rtw89_dev *rtwdev,
634				 struct rtw89_pci_rx_ring *rx_ring,
635				 u32 cnt)
636{
637	struct rtw89_pci_dma_ring *bd_ring = &rx_ring->bd_ring;
638	u32 release_cnt;
639
640	while (cnt) {
641		release_cnt = rtw89_pci_release_tx_skbs(rtwdev, rx_ring, cnt);
642		if (!release_cnt) {
643			rtw89_err(rtwdev, "failed to release TX skbs\n");
644
645			/* skip the rest RXBD bufs */
646			rtw89_pci_rxbd_increase(rx_ring, cnt);
647			break;
648		}
649
650		cnt -= release_cnt;
651	}
652
653	rtw89_write16(rtwdev, bd_ring->addr.idx, bd_ring->wp);
654}
655
656static int rtw89_pci_poll_rpq_dma(struct rtw89_dev *rtwdev,
657				  struct rtw89_pci *rtwpci, int budget)
658{
659	struct rtw89_pci_rx_ring *rx_ring;
660	u32 cnt;
661	int work_done;
662
663	rx_ring = &rtwpci->rx_rings[RTW89_RXCH_RPQ];
664
665	spin_lock_bh(&rtwpci->trx_lock);
666
667	cnt = rtw89_pci_rxbd_recalc(rtwdev, rx_ring);
668	if (cnt == 0)
669		goto out_unlock;
670
671	rtw89_pci_release_tx(rtwdev, rx_ring, cnt);
672
673out_unlock:
674	spin_unlock_bh(&rtwpci->trx_lock);
675
676	/* always release all RPQ */
677	work_done = min_t(int, cnt, budget);
678	rtwdev->napi_budget_countdown -= work_done;
679
680	return work_done;
681}
682
683static void rtw89_pci_isr_rxd_unavail(struct rtw89_dev *rtwdev,
684				      struct rtw89_pci *rtwpci)
685{
686	struct rtw89_pci_rx_ring *rx_ring;
687	struct rtw89_pci_dma_ring *bd_ring;
688	u32 reg_idx;
689	u16 hw_idx, hw_idx_next, host_idx;
690	int i;
691
692	for (i = 0; i < RTW89_RXCH_NUM; i++) {
693		rx_ring = &rtwpci->rx_rings[i];
694		bd_ring = &rx_ring->bd_ring;
695
696		reg_idx = rtw89_read32(rtwdev, bd_ring->addr.idx);
697		hw_idx = FIELD_GET(TXBD_HW_IDX_MASK, reg_idx);
698		host_idx = FIELD_GET(TXBD_HOST_IDX_MASK, reg_idx);
699		hw_idx_next = (hw_idx + 1) % bd_ring->len;
700
701		if (hw_idx_next == host_idx)
702			rtw89_debug(rtwdev, RTW89_DBG_UNEXP, "%d RXD unavailable\n", i);
703
704		rtw89_debug(rtwdev, RTW89_DBG_TXRX,
705			    "%d RXD unavailable, idx=0x%08x, len=%d\n",
706			    i, reg_idx, bd_ring->len);
707	}
708}
709
710void rtw89_pci_recognize_intrs(struct rtw89_dev *rtwdev,
711			       struct rtw89_pci *rtwpci,
712			       struct rtw89_pci_isrs *isrs)
713{
714	isrs->halt_c2h_isrs = rtw89_read32(rtwdev, R_AX_HISR0) & rtwpci->halt_c2h_intrs;
715	isrs->isrs[0] = rtw89_read32(rtwdev, R_AX_PCIE_HISR00) & rtwpci->intrs[0];
716	isrs->isrs[1] = rtw89_read32(rtwdev, R_AX_PCIE_HISR10) & rtwpci->intrs[1];
717
718	rtw89_write32(rtwdev, R_AX_HISR0, isrs->halt_c2h_isrs);
719	rtw89_write32(rtwdev, R_AX_PCIE_HISR00, isrs->isrs[0]);
720	rtw89_write32(rtwdev, R_AX_PCIE_HISR10, isrs->isrs[1]);
721}
722EXPORT_SYMBOL(rtw89_pci_recognize_intrs);
723
724void rtw89_pci_recognize_intrs_v1(struct rtw89_dev *rtwdev,
725				  struct rtw89_pci *rtwpci,
726				  struct rtw89_pci_isrs *isrs)
727{
728	isrs->ind_isrs = rtw89_read32(rtwdev, R_AX_PCIE_HISR00_V1) & rtwpci->ind_intrs;
729	isrs->halt_c2h_isrs = isrs->ind_isrs & B_AX_HS0ISR_IND_INT_EN ?
730			      rtw89_read32(rtwdev, R_AX_HISR0) & rtwpci->halt_c2h_intrs : 0;
731	isrs->isrs[0] = isrs->ind_isrs & B_AX_HCI_AXIDMA_INT_EN ?
732			rtw89_read32(rtwdev, R_AX_HAXI_HISR00) & rtwpci->intrs[0] : 0;
733	isrs->isrs[1] = isrs->ind_isrs & B_AX_HS1ISR_IND_INT_EN ?
734			rtw89_read32(rtwdev, R_AX_HISR1) & rtwpci->intrs[1] : 0;
735
736	if (isrs->halt_c2h_isrs)
737		rtw89_write32(rtwdev, R_AX_HISR0, isrs->halt_c2h_isrs);
738	if (isrs->isrs[0])
739		rtw89_write32(rtwdev, R_AX_HAXI_HISR00, isrs->isrs[0]);
740	if (isrs->isrs[1])
741		rtw89_write32(rtwdev, R_AX_HISR1, isrs->isrs[1]);
742}
743EXPORT_SYMBOL(rtw89_pci_recognize_intrs_v1);
744
745void rtw89_pci_recognize_intrs_v2(struct rtw89_dev *rtwdev,
746				  struct rtw89_pci *rtwpci,
747				  struct rtw89_pci_isrs *isrs)
748{
749	isrs->ind_isrs = rtw89_read32(rtwdev, R_BE_PCIE_HISR) & rtwpci->ind_intrs;
750	isrs->halt_c2h_isrs = isrs->ind_isrs & B_BE_HS0ISR_IND_INT ?
751			      rtw89_read32(rtwdev, R_BE_HISR0) & rtwpci->halt_c2h_intrs : 0;
752	isrs->isrs[0] = isrs->ind_isrs & B_BE_HCI_AXIDMA_INT ?
753			rtw89_read32(rtwdev, R_BE_HAXI_HISR00) & rtwpci->intrs[0] : 0;
754	isrs->isrs[1] = rtw89_read32(rtwdev, R_BE_PCIE_DMA_ISR) & rtwpci->intrs[1];
755
756	if (isrs->halt_c2h_isrs)
757		rtw89_write32(rtwdev, R_BE_HISR0, isrs->halt_c2h_isrs);
758	if (isrs->isrs[0])
759		rtw89_write32(rtwdev, R_BE_HAXI_HISR00, isrs->isrs[0]);
760	if (isrs->isrs[1])
761		rtw89_write32(rtwdev, R_BE_PCIE_DMA_ISR, isrs->isrs[1]);
762	rtw89_write32(rtwdev, R_BE_PCIE_HISR, isrs->ind_isrs);
763}
764EXPORT_SYMBOL(rtw89_pci_recognize_intrs_v2);
765
766void rtw89_pci_enable_intr(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
767{
768	rtw89_write32(rtwdev, R_AX_HIMR0, rtwpci->halt_c2h_intrs);
769	rtw89_write32(rtwdev, R_AX_PCIE_HIMR00, rtwpci->intrs[0]);
770	rtw89_write32(rtwdev, R_AX_PCIE_HIMR10, rtwpci->intrs[1]);
771}
772EXPORT_SYMBOL(rtw89_pci_enable_intr);
773
774void rtw89_pci_disable_intr(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
775{
776	rtw89_write32(rtwdev, R_AX_HIMR0, 0);
777	rtw89_write32(rtwdev, R_AX_PCIE_HIMR00, 0);
778	rtw89_write32(rtwdev, R_AX_PCIE_HIMR10, 0);
779}
780EXPORT_SYMBOL(rtw89_pci_disable_intr);
781
782void rtw89_pci_enable_intr_v1(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
783{
784	rtw89_write32(rtwdev, R_AX_PCIE_HIMR00_V1, rtwpci->ind_intrs);
785	rtw89_write32(rtwdev, R_AX_HIMR0, rtwpci->halt_c2h_intrs);
786	rtw89_write32(rtwdev, R_AX_HAXI_HIMR00, rtwpci->intrs[0]);
787	rtw89_write32(rtwdev, R_AX_HIMR1, rtwpci->intrs[1]);
788}
789EXPORT_SYMBOL(rtw89_pci_enable_intr_v1);
790
791void rtw89_pci_disable_intr_v1(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
792{
793	rtw89_write32(rtwdev, R_AX_PCIE_HIMR00_V1, 0);
794}
795EXPORT_SYMBOL(rtw89_pci_disable_intr_v1);
796
797void rtw89_pci_enable_intr_v2(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
798{
799	rtw89_write32(rtwdev, R_BE_HIMR0, rtwpci->halt_c2h_intrs);
800	rtw89_write32(rtwdev, R_BE_HAXI_HIMR00, rtwpci->intrs[0]);
801	rtw89_write32(rtwdev, R_BE_PCIE_DMA_IMR_0_V1, rtwpci->intrs[1]);
802	rtw89_write32(rtwdev, R_BE_PCIE_HIMR0, rtwpci->ind_intrs);
803}
804EXPORT_SYMBOL(rtw89_pci_enable_intr_v2);
805
806void rtw89_pci_disable_intr_v2(struct rtw89_dev *rtwdev, struct rtw89_pci *rtwpci)
807{
808	rtw89_write32(rtwdev, R_BE_PCIE_HIMR0, 0);
809	rtw89_write32(rtwdev, R_BE_PCIE_DMA_IMR_0_V1, 0);
810}
811EXPORT_SYMBOL(rtw89_pci_disable_intr_v2);
812
813static void rtw89_pci_ops_recovery_start(struct rtw89_dev *rtwdev)
814{
815	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
816	unsigned long flags;
817
818	spin_lock_irqsave(&rtwpci->irq_lock, flags);
819	rtw89_chip_disable_intr(rtwdev, rtwpci);
820	rtw89_chip_config_intr_mask(rtwdev, RTW89_PCI_INTR_MASK_RECOVERY_START);
821	rtw89_chip_enable_intr(rtwdev, rtwpci);
822	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
823}
824
825static void rtw89_pci_ops_recovery_complete(struct rtw89_dev *rtwdev)
826{
827	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
828	unsigned long flags;
829
830	spin_lock_irqsave(&rtwpci->irq_lock, flags);
831	rtw89_chip_disable_intr(rtwdev, rtwpci);
832	rtw89_chip_config_intr_mask(rtwdev, RTW89_PCI_INTR_MASK_RECOVERY_COMPLETE);
833	rtw89_chip_enable_intr(rtwdev, rtwpci);
834	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
835}
836
837static void rtw89_pci_low_power_interrupt_handler(struct rtw89_dev *rtwdev)
838{
839	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
840	int budget = NAPI_POLL_WEIGHT;
841
842	/* To prevent RXQ get stuck due to run out of budget. */
843	rtwdev->napi_budget_countdown = budget;
844
845	rtw89_pci_poll_rpq_dma(rtwdev, rtwpci, budget);
846	rtw89_pci_poll_rxq_dma(rtwdev, rtwpci, budget);
847}
848
849static irqreturn_t rtw89_pci_interrupt_threadfn(int irq, void *dev)
850{
851	struct rtw89_dev *rtwdev = dev;
852	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
853	const struct rtw89_pci_info *info = rtwdev->pci_info;
854	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
855	struct rtw89_pci_isrs isrs;
856	unsigned long flags;
857
858	spin_lock_irqsave(&rtwpci->irq_lock, flags);
859	rtw89_chip_recognize_intrs(rtwdev, rtwpci, &isrs);
860	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
861
862	if (unlikely(isrs.isrs[0] & gen_def->isr_rdu))
863		rtw89_pci_isr_rxd_unavail(rtwdev, rtwpci);
864
865	if (unlikely(isrs.halt_c2h_isrs & gen_def->isr_halt_c2h))
866		rtw89_ser_notify(rtwdev, rtw89_mac_get_err_status(rtwdev));
867
868	if (unlikely(isrs.halt_c2h_isrs & gen_def->isr_wdt_timeout))
869		rtw89_ser_notify(rtwdev, MAC_AX_ERR_L2_ERR_WDT_TIMEOUT_INT);
870
871	if (unlikely(rtwpci->under_recovery))
872		goto enable_intr;
873
874	if (unlikely(rtwpci->low_power)) {
875		rtw89_pci_low_power_interrupt_handler(rtwdev);
876		goto enable_intr;
877	}
878
879	if (likely(rtwpci->running)) {
880		local_bh_disable();
881		napi_schedule(&rtwdev->napi);
882		local_bh_enable();
883	}
884
885	return IRQ_HANDLED;
886
887enable_intr:
888	spin_lock_irqsave(&rtwpci->irq_lock, flags);
889	if (likely(rtwpci->running))
890		rtw89_chip_enable_intr(rtwdev, rtwpci);
891	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
892	return IRQ_HANDLED;
893}
894
895static irqreturn_t rtw89_pci_interrupt_handler(int irq, void *dev)
896{
897	struct rtw89_dev *rtwdev = dev;
898	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
899	unsigned long flags;
900	irqreturn_t irqret = IRQ_WAKE_THREAD;
901
902	spin_lock_irqsave(&rtwpci->irq_lock, flags);
903
904	/* If interrupt event is on the road, it is still trigger interrupt
905	 * even we have done pci_stop() to turn off IMR.
906	 */
907	if (unlikely(!rtwpci->running)) {
908		irqret = IRQ_HANDLED;
909		goto exit;
910	}
911
912	rtw89_chip_disable_intr(rtwdev, rtwpci);
913exit:
914	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
915
916	return irqret;
917}
918
919#define DEF_TXCHADDRS_TYPE2(gen, ch_idx, txch, v...) \
920	[RTW89_TXCH_##ch_idx] = { \
921		.num = R_##gen##_##txch##_TXBD_NUM ##v, \
922		.idx = R_##gen##_##txch##_TXBD_IDX ##v, \
923		.bdram = 0, \
924		.desa_l = R_##gen##_##txch##_TXBD_DESA_L ##v, \
925		.desa_h = R_##gen##_##txch##_TXBD_DESA_H ##v, \
926	}
927
928#define DEF_TXCHADDRS_TYPE1(info, txch, v...) \
929	[RTW89_TXCH_##txch] = { \
930		.num = R_AX_##txch##_TXBD_NUM ##v, \
931		.idx = R_AX_##txch##_TXBD_IDX ##v, \
932		.bdram = R_AX_##txch##_BDRAM_CTRL ##v, \
933		.desa_l = R_AX_##txch##_TXBD_DESA_L ##v, \
934		.desa_h = R_AX_##txch##_TXBD_DESA_H ##v, \
935	}
936
937#define DEF_TXCHADDRS(info, txch, v...) \
938	[RTW89_TXCH_##txch] = { \
939		.num = R_AX_##txch##_TXBD_NUM, \
940		.idx = R_AX_##txch##_TXBD_IDX, \
941		.bdram = R_AX_##txch##_BDRAM_CTRL ##v, \
942		.desa_l = R_AX_##txch##_TXBD_DESA_L ##v, \
943		.desa_h = R_AX_##txch##_TXBD_DESA_H ##v, \
944	}
945
946#define DEF_RXCHADDRS(gen, ch_idx, rxch, v...) \
947	[RTW89_RXCH_##ch_idx] = { \
948		.num = R_##gen##_##rxch##_RXBD_NUM ##v, \
949		.idx = R_##gen##_##rxch##_RXBD_IDX ##v, \
950		.desa_l = R_##gen##_##rxch##_RXBD_DESA_L ##v, \
951		.desa_h = R_##gen##_##rxch##_RXBD_DESA_H ##v, \
952	}
953
954const struct rtw89_pci_ch_dma_addr_set rtw89_pci_ch_dma_addr_set = {
955	.tx = {
956		DEF_TXCHADDRS(info, ACH0),
957		DEF_TXCHADDRS(info, ACH1),
958		DEF_TXCHADDRS(info, ACH2),
959		DEF_TXCHADDRS(info, ACH3),
960		DEF_TXCHADDRS(info, ACH4),
961		DEF_TXCHADDRS(info, ACH5),
962		DEF_TXCHADDRS(info, ACH6),
963		DEF_TXCHADDRS(info, ACH7),
964		DEF_TXCHADDRS(info, CH8),
965		DEF_TXCHADDRS(info, CH9),
966		DEF_TXCHADDRS_TYPE1(info, CH10),
967		DEF_TXCHADDRS_TYPE1(info, CH11),
968		DEF_TXCHADDRS(info, CH12),
969	},
970	.rx = {
971		DEF_RXCHADDRS(AX, RXQ, RXQ),
972		DEF_RXCHADDRS(AX, RPQ, RPQ),
973	},
974};
975EXPORT_SYMBOL(rtw89_pci_ch_dma_addr_set);
976
977const struct rtw89_pci_ch_dma_addr_set rtw89_pci_ch_dma_addr_set_v1 = {
978	.tx = {
979		DEF_TXCHADDRS(info, ACH0, _V1),
980		DEF_TXCHADDRS(info, ACH1, _V1),
981		DEF_TXCHADDRS(info, ACH2, _V1),
982		DEF_TXCHADDRS(info, ACH3, _V1),
983		DEF_TXCHADDRS(info, ACH4, _V1),
984		DEF_TXCHADDRS(info, ACH5, _V1),
985		DEF_TXCHADDRS(info, ACH6, _V1),
986		DEF_TXCHADDRS(info, ACH7, _V1),
987		DEF_TXCHADDRS(info, CH8, _V1),
988		DEF_TXCHADDRS(info, CH9, _V1),
989		DEF_TXCHADDRS_TYPE1(info, CH10, _V1),
990		DEF_TXCHADDRS_TYPE1(info, CH11, _V1),
991		DEF_TXCHADDRS(info, CH12, _V1),
992	},
993	.rx = {
994		DEF_RXCHADDRS(AX, RXQ, RXQ, _V1),
995		DEF_RXCHADDRS(AX, RPQ, RPQ, _V1),
996	},
997};
998EXPORT_SYMBOL(rtw89_pci_ch_dma_addr_set_v1);
999
1000const struct rtw89_pci_ch_dma_addr_set rtw89_pci_ch_dma_addr_set_be = {
1001	.tx = {
1002		DEF_TXCHADDRS_TYPE2(BE, ACH0, CH0, _V1),
1003		DEF_TXCHADDRS_TYPE2(BE, ACH1, CH1, _V1),
1004		DEF_TXCHADDRS_TYPE2(BE, ACH2, CH2, _V1),
1005		DEF_TXCHADDRS_TYPE2(BE, ACH3, CH3, _V1),
1006		DEF_TXCHADDRS_TYPE2(BE, ACH4, CH4, _V1),
1007		DEF_TXCHADDRS_TYPE2(BE, ACH5, CH5, _V1),
1008		DEF_TXCHADDRS_TYPE2(BE, ACH6, CH6, _V1),
1009		DEF_TXCHADDRS_TYPE2(BE, ACH7, CH7, _V1),
1010		DEF_TXCHADDRS_TYPE2(BE, CH8, CH8, _V1),
1011		DEF_TXCHADDRS_TYPE2(BE, CH9, CH9, _V1),
1012		DEF_TXCHADDRS_TYPE2(BE, CH10, CH10, _V1),
1013		DEF_TXCHADDRS_TYPE2(BE, CH11, CH11, _V1),
1014		DEF_TXCHADDRS_TYPE2(BE, CH12, CH12, _V1),
1015	},
1016	.rx = {
1017		DEF_RXCHADDRS(BE, RXQ, RXQ0, _V1),
1018		DEF_RXCHADDRS(BE, RPQ, RPQ0, _V1),
1019	},
1020};
1021EXPORT_SYMBOL(rtw89_pci_ch_dma_addr_set_be);
1022
1023#undef DEF_TXCHADDRS_TYPE1
1024#undef DEF_TXCHADDRS
1025#undef DEF_RXCHADDRS
1026
1027static int rtw89_pci_get_txch_addrs(struct rtw89_dev *rtwdev,
1028				    enum rtw89_tx_channel txch,
1029				    const struct rtw89_pci_ch_dma_addr **addr)
1030{
1031	const struct rtw89_pci_info *info = rtwdev->pci_info;
1032
1033	if (txch >= RTW89_TXCH_NUM)
1034		return -EINVAL;
1035
1036	*addr = &info->dma_addr_set->tx[txch];
1037
1038	return 0;
1039}
1040
1041static int rtw89_pci_get_rxch_addrs(struct rtw89_dev *rtwdev,
1042				    enum rtw89_rx_channel rxch,
1043				    const struct rtw89_pci_ch_dma_addr **addr)
1044{
1045	const struct rtw89_pci_info *info = rtwdev->pci_info;
1046
1047	if (rxch >= RTW89_RXCH_NUM)
1048		return -EINVAL;
1049
1050	*addr = &info->dma_addr_set->rx[rxch];
1051
1052	return 0;
1053}
1054
1055static u32 rtw89_pci_get_avail_txbd_num(struct rtw89_pci_tx_ring *ring)
1056{
1057	struct rtw89_pci_dma_ring *bd_ring = &ring->bd_ring;
1058
1059	/* reserved 1 desc check ring is full or not */
1060	if (bd_ring->rp > bd_ring->wp)
1061		return bd_ring->rp - bd_ring->wp - 1;
1062
1063	return bd_ring->len - (bd_ring->wp - bd_ring->rp) - 1;
1064}
1065
1066static
1067u32 __rtw89_pci_check_and_reclaim_tx_fwcmd_resource(struct rtw89_dev *rtwdev)
1068{
1069	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1070	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[RTW89_TXCH_CH12];
1071	u32 cnt;
1072
1073	spin_lock_bh(&rtwpci->trx_lock);
1074	rtw89_pci_reclaim_tx_fwcmd(rtwdev, rtwpci);
1075	cnt = rtw89_pci_get_avail_txbd_num(tx_ring);
1076	spin_unlock_bh(&rtwpci->trx_lock);
1077
1078	return cnt;
1079}
1080
1081static
1082u32 __rtw89_pci_check_and_reclaim_tx_resource_noio(struct rtw89_dev *rtwdev,
1083						   u8 txch)
1084{
1085	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1086	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[txch];
1087	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
1088	u32 cnt;
1089
1090	spin_lock_bh(&rtwpci->trx_lock);
1091	cnt = rtw89_pci_get_avail_txbd_num(tx_ring);
1092	cnt = min(cnt, wd_ring->curr_num);
1093	spin_unlock_bh(&rtwpci->trx_lock);
1094
1095	return cnt;
1096}
1097
1098static u32 __rtw89_pci_check_and_reclaim_tx_resource(struct rtw89_dev *rtwdev,
1099						     u8 txch)
1100{
1101	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1102	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[txch];
1103	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
1104	const struct rtw89_chip_info *chip = rtwdev->chip;
1105	u32 bd_cnt, wd_cnt, min_cnt = 0;
1106	struct rtw89_pci_rx_ring *rx_ring;
1107	enum rtw89_debug_mask debug_mask;
1108	u32 cnt;
1109
1110	rx_ring = &rtwpci->rx_rings[RTW89_RXCH_RPQ];
1111
1112	spin_lock_bh(&rtwpci->trx_lock);
1113	bd_cnt = rtw89_pci_get_avail_txbd_num(tx_ring);
1114	wd_cnt = wd_ring->curr_num;
1115
1116	if (wd_cnt == 0 || bd_cnt == 0) {
1117		cnt = rtw89_pci_rxbd_recalc(rtwdev, rx_ring);
1118		if (cnt)
1119			rtw89_pci_release_tx(rtwdev, rx_ring, cnt);
1120		else if (wd_cnt == 0)
1121			goto out_unlock;
1122
1123		bd_cnt = rtw89_pci_get_avail_txbd_num(tx_ring);
1124		if (bd_cnt == 0)
1125			rtw89_pci_reclaim_txbd(rtwdev, tx_ring);
1126	}
1127
1128	bd_cnt = rtw89_pci_get_avail_txbd_num(tx_ring);
1129	wd_cnt = wd_ring->curr_num;
1130	min_cnt = min(bd_cnt, wd_cnt);
1131	if (min_cnt == 0) {
1132		/* This message can be frequently shown in low power mode or
1133		 * high traffic with small FIFO chips, and we have recognized it as normal
1134		 * behavior, so print with mask RTW89_DBG_TXRX in these situations.
1135		 */
1136		if (rtwpci->low_power || chip->small_fifo_size)
1137			debug_mask = RTW89_DBG_TXRX;
1138		else
1139			debug_mask = RTW89_DBG_UNEXP;
1140
1141		rtw89_debug(rtwdev, debug_mask,
1142			    "still no tx resource after reclaim: wd_cnt=%d bd_cnt=%d\n",
1143			    wd_cnt, bd_cnt);
1144	}
1145
1146out_unlock:
1147	spin_unlock_bh(&rtwpci->trx_lock);
1148
1149	return min_cnt;
1150}
1151
1152static u32 rtw89_pci_check_and_reclaim_tx_resource(struct rtw89_dev *rtwdev,
1153						   u8 txch)
1154{
1155	if (rtwdev->hci.paused)
1156		return __rtw89_pci_check_and_reclaim_tx_resource_noio(rtwdev, txch);
1157
1158	if (txch == RTW89_TXCH_CH12)
1159		return __rtw89_pci_check_and_reclaim_tx_fwcmd_resource(rtwdev);
1160
1161	return __rtw89_pci_check_and_reclaim_tx_resource(rtwdev, txch);
1162}
1163
1164static void __rtw89_pci_tx_kick_off(struct rtw89_dev *rtwdev, struct rtw89_pci_tx_ring *tx_ring)
1165{
1166	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1167	struct rtw89_pci_dma_ring *bd_ring = &tx_ring->bd_ring;
1168	u32 host_idx, addr;
1169
1170	spin_lock_bh(&rtwpci->trx_lock);
1171
1172	addr = bd_ring->addr.idx;
1173	host_idx = bd_ring->wp;
1174	rtw89_write16(rtwdev, addr, host_idx);
1175
1176	spin_unlock_bh(&rtwpci->trx_lock);
1177}
1178
1179static void rtw89_pci_tx_bd_ring_update(struct rtw89_dev *rtwdev, struct rtw89_pci_tx_ring *tx_ring,
1180					int n_txbd)
1181{
1182	struct rtw89_pci_dma_ring *bd_ring = &tx_ring->bd_ring;
1183	u32 host_idx, len;
1184
1185	len = bd_ring->len;
1186	host_idx = bd_ring->wp + n_txbd;
1187	host_idx = host_idx < len ? host_idx : host_idx - len;
1188
1189	bd_ring->wp = host_idx;
1190}
1191
1192static void rtw89_pci_ops_tx_kick_off(struct rtw89_dev *rtwdev, u8 txch)
1193{
1194	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1195	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[txch];
1196
1197	if (rtwdev->hci.paused) {
1198		set_bit(txch, rtwpci->kick_map);
1199		return;
1200	}
1201
1202	__rtw89_pci_tx_kick_off(rtwdev, tx_ring);
1203}
1204
1205static void rtw89_pci_tx_kick_off_pending(struct rtw89_dev *rtwdev)
1206{
1207	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1208	struct rtw89_pci_tx_ring *tx_ring;
1209	int txch;
1210
1211	for (txch = 0; txch < RTW89_TXCH_NUM; txch++) {
1212		if (!test_and_clear_bit(txch, rtwpci->kick_map))
1213			continue;
1214
1215		tx_ring = &rtwpci->tx_rings[txch];
1216		__rtw89_pci_tx_kick_off(rtwdev, tx_ring);
1217	}
1218}
1219
1220static void __pci_flush_txch(struct rtw89_dev *rtwdev, u8 txch, bool drop)
1221{
1222	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1223	struct rtw89_pci_tx_ring *tx_ring = &rtwpci->tx_rings[txch];
1224	struct rtw89_pci_dma_ring *bd_ring = &tx_ring->bd_ring;
1225	u32 cur_idx, cur_rp;
1226	u8 i;
1227
1228	/* Because the time taked by the I/O is a bit dynamic, it's hard to
1229	 * define a reasonable fixed total timeout to use read_poll_timeout*
1230	 * helper. Instead, we can ensure a reasonable polling times, so we
1231	 * just use for loop with udelay here.
1232	 */
1233	for (i = 0; i < 60; i++) {
1234		cur_idx = rtw89_read32(rtwdev, bd_ring->addr.idx);
1235		cur_rp = FIELD_GET(TXBD_HW_IDX_MASK, cur_idx);
1236		if (cur_rp == bd_ring->wp)
1237			return;
1238
1239		udelay(1);
1240	}
1241
1242	if (!drop)
1243		rtw89_info(rtwdev, "timed out to flush pci txch: %d\n", txch);
1244}
1245
1246static void __rtw89_pci_ops_flush_txchs(struct rtw89_dev *rtwdev, u32 txchs,
1247					bool drop)
1248{
1249	const struct rtw89_pci_info *info = rtwdev->pci_info;
1250	u8 i;
1251
1252	for (i = 0; i < RTW89_TXCH_NUM; i++) {
1253		/* It may be unnecessary to flush FWCMD queue. */
1254		if (i == RTW89_TXCH_CH12)
1255			continue;
1256		if (info->tx_dma_ch_mask & BIT(i))
1257			continue;
1258
1259		if (txchs & BIT(i))
1260			__pci_flush_txch(rtwdev, i, drop);
1261	}
1262}
1263
1264static void rtw89_pci_ops_flush_queues(struct rtw89_dev *rtwdev, u32 queues,
1265				       bool drop)
1266{
1267	__rtw89_pci_ops_flush_txchs(rtwdev, BIT(RTW89_TXCH_NUM) - 1, drop);
1268}
1269
1270u32 rtw89_pci_fill_txaddr_info(struct rtw89_dev *rtwdev,
1271			       void *txaddr_info_addr, u32 total_len,
1272			       dma_addr_t dma, u8 *add_info_nr)
1273{
1274	struct rtw89_pci_tx_addr_info_32 *txaddr_info = txaddr_info_addr;
1275
1276	txaddr_info->length = cpu_to_le16(total_len);
1277	txaddr_info->option = cpu_to_le16(RTW89_PCI_ADDR_MSDU_LS |
1278					  RTW89_PCI_ADDR_NUM(1));
1279	txaddr_info->dma = cpu_to_le32(dma);
1280
1281	*add_info_nr = 1;
1282
1283	return sizeof(*txaddr_info);
1284}
1285EXPORT_SYMBOL(rtw89_pci_fill_txaddr_info);
1286
1287u32 rtw89_pci_fill_txaddr_info_v1(struct rtw89_dev *rtwdev,
1288				  void *txaddr_info_addr, u32 total_len,
1289				  dma_addr_t dma, u8 *add_info_nr)
1290{
1291	struct rtw89_pci_tx_addr_info_32_v1 *txaddr_info = txaddr_info_addr;
1292	u32 remain = total_len;
1293	u32 len;
1294	u16 length_option;
1295	int n;
1296
1297	for (n = 0; n < RTW89_TXADDR_INFO_NR_V1 && remain; n++) {
1298		len = remain >= TXADDR_INFO_LENTHG_V1_MAX ?
1299		      TXADDR_INFO_LENTHG_V1_MAX : remain;
1300		remain -= len;
1301
1302		length_option = FIELD_PREP(B_PCIADDR_LEN_V1_MASK, len) |
1303				FIELD_PREP(B_PCIADDR_HIGH_SEL_V1_MASK, 0) |
1304				FIELD_PREP(B_PCIADDR_LS_V1_MASK, remain == 0);
1305		txaddr_info->length_opt = cpu_to_le16(length_option);
1306		txaddr_info->dma_low_lsb = cpu_to_le16(FIELD_GET(GENMASK(15, 0), dma));
1307		txaddr_info->dma_low_msb = cpu_to_le16(FIELD_GET(GENMASK(31, 16), dma));
1308
1309		dma += len;
1310		txaddr_info++;
1311	}
1312
1313	WARN_ONCE(remain, "length overflow remain=%u total_len=%u",
1314		  remain, total_len);
1315
1316	*add_info_nr = n;
1317
1318	return n * sizeof(*txaddr_info);
1319}
1320EXPORT_SYMBOL(rtw89_pci_fill_txaddr_info_v1);
1321
1322static int rtw89_pci_txwd_submit(struct rtw89_dev *rtwdev,
1323				 struct rtw89_pci_tx_ring *tx_ring,
1324				 struct rtw89_pci_tx_wd *txwd,
1325				 struct rtw89_core_tx_request *tx_req)
1326{
1327	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1328	const struct rtw89_chip_info *chip = rtwdev->chip;
1329	struct rtw89_tx_desc_info *desc_info = &tx_req->desc_info;
1330	struct rtw89_pci_tx_wp_info *txwp_info;
1331	void *txaddr_info_addr;
1332	struct pci_dev *pdev = rtwpci->pdev;
1333	struct sk_buff *skb = tx_req->skb;
1334	struct rtw89_pci_tx_data *tx_data = RTW89_PCI_TX_SKB_CB(skb);
1335	struct rtw89_tx_skb_data *skb_data = RTW89_TX_SKB_CB(skb);
1336	bool en_wd_info = desc_info->en_wd_info;
1337	u32 txwd_len;
1338	u32 txwp_len;
1339	u32 txaddr_info_len;
1340	dma_addr_t dma;
1341	int ret;
1342
1343	dma = dma_map_single(&pdev->dev, skb->data, skb->len, DMA_TO_DEVICE);
1344	if (dma_mapping_error(&pdev->dev, dma)) {
1345		rtw89_err(rtwdev, "failed to map skb dma data\n");
1346		ret = -EBUSY;
1347		goto err;
1348	}
1349
1350	tx_data->dma = dma;
1351	rcu_assign_pointer(skb_data->wait, NULL);
1352
1353	txwp_len = sizeof(*txwp_info);
1354	txwd_len = chip->txwd_body_size;
1355	txwd_len += en_wd_info ? chip->txwd_info_size : 0;
1356
1357	txwp_info = txwd->vaddr + txwd_len;
1358	txwp_info->seq0 = cpu_to_le16(txwd->seq | RTW89_PCI_TXWP_VALID);
1359	txwp_info->seq1 = 0;
1360	txwp_info->seq2 = 0;
1361	txwp_info->seq3 = 0;
1362
1363	tx_ring->tx_cnt++;
1364	txaddr_info_addr = txwd->vaddr + txwd_len + txwp_len;
1365	txaddr_info_len =
1366		rtw89_chip_fill_txaddr_info(rtwdev, txaddr_info_addr, skb->len,
1367					    dma, &desc_info->addr_info_nr);
1368
1369	txwd->len = txwd_len + txwp_len + txaddr_info_len;
1370
1371	rtw89_chip_fill_txdesc(rtwdev, desc_info, txwd->vaddr);
1372
1373	skb_queue_tail(&txwd->queue, skb);
1374
1375	return 0;
1376
1377err:
1378	return ret;
1379}
1380
1381static int rtw89_pci_fwcmd_submit(struct rtw89_dev *rtwdev,
1382				  struct rtw89_pci_tx_ring *tx_ring,
1383				  struct rtw89_pci_tx_bd_32 *txbd,
1384				  struct rtw89_core_tx_request *tx_req)
1385{
1386	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1387	const struct rtw89_chip_info *chip = rtwdev->chip;
1388	struct rtw89_tx_desc_info *desc_info = &tx_req->desc_info;
1389	void *txdesc;
1390	int txdesc_size = chip->h2c_desc_size;
1391	struct pci_dev *pdev = rtwpci->pdev;
1392	struct sk_buff *skb = tx_req->skb;
1393	struct rtw89_pci_tx_data *tx_data = RTW89_PCI_TX_SKB_CB(skb);
1394	dma_addr_t dma;
1395
1396	txdesc = skb_push(skb, txdesc_size);
1397	memset(txdesc, 0, txdesc_size);
1398	rtw89_chip_fill_txdesc_fwcmd(rtwdev, desc_info, txdesc);
1399
1400	dma = dma_map_single(&pdev->dev, skb->data, skb->len, DMA_TO_DEVICE);
1401	if (dma_mapping_error(&pdev->dev, dma)) {
1402		rtw89_err(rtwdev, "failed to map fwcmd dma data\n");
1403		return -EBUSY;
1404	}
1405
1406	tx_data->dma = dma;
1407	txbd->option = cpu_to_le16(RTW89_PCI_TXBD_OPTION_LS);
1408	txbd->length = cpu_to_le16(skb->len);
1409	txbd->dma = cpu_to_le32(tx_data->dma);
1410	skb_queue_tail(&rtwpci->h2c_queue, skb);
1411
1412	rtw89_pci_tx_bd_ring_update(rtwdev, tx_ring, 1);
1413
1414	return 0;
1415}
1416
1417static int rtw89_pci_txbd_submit(struct rtw89_dev *rtwdev,
1418				 struct rtw89_pci_tx_ring *tx_ring,
1419				 struct rtw89_pci_tx_bd_32 *txbd,
1420				 struct rtw89_core_tx_request *tx_req)
1421{
1422	struct rtw89_pci_tx_wd *txwd;
1423	int ret;
1424
1425	/* FWCMD queue doesn't have wd pages. Instead, it submits the CMD
1426	 * buffer with WD BODY only. So here we don't need to check the free
1427	 * pages of the wd ring.
1428	 */
1429	if (tx_ring->txch == RTW89_TXCH_CH12)
1430		return rtw89_pci_fwcmd_submit(rtwdev, tx_ring, txbd, tx_req);
1431
1432	txwd = rtw89_pci_dequeue_txwd(tx_ring);
1433	if (!txwd) {
1434		rtw89_err(rtwdev, "no available TXWD\n");
1435		ret = -ENOSPC;
1436		goto err;
1437	}
1438
1439	ret = rtw89_pci_txwd_submit(rtwdev, tx_ring, txwd, tx_req);
1440	if (ret) {
1441		rtw89_err(rtwdev, "failed to submit TXWD %d\n", txwd->seq);
1442		goto err_enqueue_wd;
1443	}
1444
1445	list_add_tail(&txwd->list, &tx_ring->busy_pages);
1446
1447	txbd->option = cpu_to_le16(RTW89_PCI_TXBD_OPTION_LS);
1448	txbd->length = cpu_to_le16(txwd->len);
1449	txbd->dma = cpu_to_le32(txwd->paddr);
1450
1451	rtw89_pci_tx_bd_ring_update(rtwdev, tx_ring, 1);
1452
1453	return 0;
1454
1455err_enqueue_wd:
1456	rtw89_pci_enqueue_txwd(tx_ring, txwd);
1457err:
1458	return ret;
1459}
1460
1461static int rtw89_pci_tx_write(struct rtw89_dev *rtwdev, struct rtw89_core_tx_request *tx_req,
1462			      u8 txch)
1463{
1464	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1465	struct rtw89_pci_tx_ring *tx_ring;
1466	struct rtw89_pci_tx_bd_32 *txbd;
1467	u32 n_avail_txbd;
1468	int ret = 0;
1469
1470	/* check the tx type and dma channel for fw cmd queue */
1471	if ((txch == RTW89_TXCH_CH12 ||
1472	     tx_req->tx_type == RTW89_CORE_TX_TYPE_FWCMD) &&
1473	    (txch != RTW89_TXCH_CH12 ||
1474	     tx_req->tx_type != RTW89_CORE_TX_TYPE_FWCMD)) {
1475		rtw89_err(rtwdev, "only fw cmd uses dma channel 12\n");
1476		return -EINVAL;
1477	}
1478
1479	tx_ring = &rtwpci->tx_rings[txch];
1480	spin_lock_bh(&rtwpci->trx_lock);
1481
1482	n_avail_txbd = rtw89_pci_get_avail_txbd_num(tx_ring);
1483	if (n_avail_txbd == 0) {
1484		rtw89_err(rtwdev, "no available TXBD\n");
1485		ret = -ENOSPC;
1486		goto err_unlock;
1487	}
1488
1489	txbd = rtw89_pci_get_next_txbd(tx_ring);
1490	ret = rtw89_pci_txbd_submit(rtwdev, tx_ring, txbd, tx_req);
1491	if (ret) {
1492		rtw89_err(rtwdev, "failed to submit TXBD\n");
1493		goto err_unlock;
1494	}
1495
1496	spin_unlock_bh(&rtwpci->trx_lock);
1497	return 0;
1498
1499err_unlock:
1500	spin_unlock_bh(&rtwpci->trx_lock);
1501	return ret;
1502}
1503
1504static int rtw89_pci_ops_tx_write(struct rtw89_dev *rtwdev, struct rtw89_core_tx_request *tx_req)
1505{
1506	struct rtw89_tx_desc_info *desc_info = &tx_req->desc_info;
1507	int ret;
1508
1509	ret = rtw89_pci_tx_write(rtwdev, tx_req, desc_info->ch_dma);
1510	if (ret) {
1511		rtw89_err(rtwdev, "failed to TX Queue %d\n", desc_info->ch_dma);
1512		return ret;
1513	}
1514
1515	return 0;
1516}
1517
1518const struct rtw89_pci_bd_ram rtw89_bd_ram_table_dual[RTW89_TXCH_NUM] = {
1519	[RTW89_TXCH_ACH0] = {.start_idx = 0,  .max_num = 5, .min_num = 2},
1520	[RTW89_TXCH_ACH1] = {.start_idx = 5,  .max_num = 5, .min_num = 2},
1521	[RTW89_TXCH_ACH2] = {.start_idx = 10, .max_num = 5, .min_num = 2},
1522	[RTW89_TXCH_ACH3] = {.start_idx = 15, .max_num = 5, .min_num = 2},
1523	[RTW89_TXCH_ACH4] = {.start_idx = 20, .max_num = 5, .min_num = 2},
1524	[RTW89_TXCH_ACH5] = {.start_idx = 25, .max_num = 5, .min_num = 2},
1525	[RTW89_TXCH_ACH6] = {.start_idx = 30, .max_num = 5, .min_num = 2},
1526	[RTW89_TXCH_ACH7] = {.start_idx = 35, .max_num = 5, .min_num = 2},
1527	[RTW89_TXCH_CH8]  = {.start_idx = 40, .max_num = 5, .min_num = 1},
1528	[RTW89_TXCH_CH9]  = {.start_idx = 45, .max_num = 5, .min_num = 1},
1529	[RTW89_TXCH_CH10] = {.start_idx = 50, .max_num = 5, .min_num = 1},
1530	[RTW89_TXCH_CH11] = {.start_idx = 55, .max_num = 5, .min_num = 1},
1531	[RTW89_TXCH_CH12] = {.start_idx = 60, .max_num = 4, .min_num = 1},
1532};
1533EXPORT_SYMBOL(rtw89_bd_ram_table_dual);
1534
1535const struct rtw89_pci_bd_ram rtw89_bd_ram_table_single[RTW89_TXCH_NUM] = {
1536	[RTW89_TXCH_ACH0] = {.start_idx = 0,  .max_num = 5, .min_num = 2},
1537	[RTW89_TXCH_ACH1] = {.start_idx = 5,  .max_num = 5, .min_num = 2},
1538	[RTW89_TXCH_ACH2] = {.start_idx = 10, .max_num = 5, .min_num = 2},
1539	[RTW89_TXCH_ACH3] = {.start_idx = 15, .max_num = 5, .min_num = 2},
1540	[RTW89_TXCH_CH8]  = {.start_idx = 20, .max_num = 4, .min_num = 1},
1541	[RTW89_TXCH_CH9]  = {.start_idx = 24, .max_num = 4, .min_num = 1},
1542	[RTW89_TXCH_CH12] = {.start_idx = 28, .max_num = 4, .min_num = 1},
1543};
1544EXPORT_SYMBOL(rtw89_bd_ram_table_single);
1545
1546static void rtw89_pci_reset_trx_rings(struct rtw89_dev *rtwdev)
1547{
1548	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1549	const struct rtw89_pci_info *info = rtwdev->pci_info;
1550	const struct rtw89_pci_bd_ram *bd_ram_table = *info->bd_ram_table;
1551	struct rtw89_pci_tx_ring *tx_ring;
1552	struct rtw89_pci_rx_ring *rx_ring;
1553	struct rtw89_pci_dma_ring *bd_ring;
1554	const struct rtw89_pci_bd_ram *bd_ram;
1555	u32 addr_num;
1556	u32 addr_idx;
1557	u32 addr_bdram;
1558	u32 addr_desa_l;
1559	u32 val32;
1560	int i;
1561
1562	for (i = 0; i < RTW89_TXCH_NUM; i++) {
1563		if (info->tx_dma_ch_mask & BIT(i))
1564			continue;
1565
1566		tx_ring = &rtwpci->tx_rings[i];
1567		bd_ring = &tx_ring->bd_ring;
1568		bd_ram = bd_ram_table ? &bd_ram_table[i] : NULL;
1569		addr_num = bd_ring->addr.num;
1570		addr_bdram = bd_ring->addr.bdram;
1571		addr_desa_l = bd_ring->addr.desa_l;
1572		bd_ring->wp = 0;
1573		bd_ring->rp = 0;
1574
1575		rtw89_write16(rtwdev, addr_num, bd_ring->len);
1576		if (addr_bdram && bd_ram) {
1577			val32 = FIELD_PREP(BDRAM_SIDX_MASK, bd_ram->start_idx) |
1578				FIELD_PREP(BDRAM_MAX_MASK, bd_ram->max_num) |
1579				FIELD_PREP(BDRAM_MIN_MASK, bd_ram->min_num);
1580
1581			rtw89_write32(rtwdev, addr_bdram, val32);
1582		}
1583		rtw89_write32(rtwdev, addr_desa_l, bd_ring->dma);
1584	}
1585
1586	for (i = 0; i < RTW89_RXCH_NUM; i++) {
1587		rx_ring = &rtwpci->rx_rings[i];
1588		bd_ring = &rx_ring->bd_ring;
1589		addr_num = bd_ring->addr.num;
1590		addr_idx = bd_ring->addr.idx;
1591		addr_desa_l = bd_ring->addr.desa_l;
1592		if (info->rx_ring_eq_is_full)
1593			bd_ring->wp = bd_ring->len - 1;
1594		else
1595			bd_ring->wp = 0;
1596		bd_ring->rp = 0;
1597		rx_ring->diliver_skb = NULL;
1598		rx_ring->diliver_desc.ready = false;
1599		rx_ring->target_rx_tag = 0;
1600
1601		rtw89_write16(rtwdev, addr_num, bd_ring->len);
1602		rtw89_write32(rtwdev, addr_desa_l, bd_ring->dma);
1603
1604		if (info->rx_ring_eq_is_full)
1605			rtw89_write16(rtwdev, addr_idx, bd_ring->wp);
1606	}
1607}
1608
1609static void rtw89_pci_release_tx_ring(struct rtw89_dev *rtwdev,
1610				      struct rtw89_pci_tx_ring *tx_ring)
1611{
1612	rtw89_pci_release_busy_txwd(rtwdev, tx_ring);
1613	rtw89_pci_release_pending_txwd_skb(rtwdev, tx_ring);
1614}
1615
1616void rtw89_pci_ops_reset(struct rtw89_dev *rtwdev)
1617{
1618	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1619	const struct rtw89_pci_info *info = rtwdev->pci_info;
1620	int txch;
1621
1622	rtw89_pci_reset_trx_rings(rtwdev);
1623
1624	spin_lock_bh(&rtwpci->trx_lock);
1625	for (txch = 0; txch < RTW89_TXCH_NUM; txch++) {
1626		if (info->tx_dma_ch_mask & BIT(txch))
1627			continue;
1628		if (txch == RTW89_TXCH_CH12) {
1629			rtw89_pci_release_fwcmd(rtwdev, rtwpci,
1630						skb_queue_len(&rtwpci->h2c_queue), true);
1631			continue;
1632		}
1633		rtw89_pci_release_tx_ring(rtwdev, &rtwpci->tx_rings[txch]);
1634	}
1635	spin_unlock_bh(&rtwpci->trx_lock);
1636}
1637
1638static void rtw89_pci_enable_intr_lock(struct rtw89_dev *rtwdev)
1639{
1640	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1641	unsigned long flags;
1642
1643	spin_lock_irqsave(&rtwpci->irq_lock, flags);
1644	rtwpci->running = true;
1645	rtw89_chip_enable_intr(rtwdev, rtwpci);
1646	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
1647}
1648
1649static void rtw89_pci_disable_intr_lock(struct rtw89_dev *rtwdev)
1650{
1651	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1652	unsigned long flags;
1653
1654	spin_lock_irqsave(&rtwpci->irq_lock, flags);
1655	rtwpci->running = false;
1656	rtw89_chip_disable_intr(rtwdev, rtwpci);
1657	spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
1658}
1659
1660static int rtw89_pci_ops_start(struct rtw89_dev *rtwdev)
1661{
1662	rtw89_core_napi_start(rtwdev);
1663	rtw89_pci_enable_intr_lock(rtwdev);
1664
1665	return 0;
1666}
1667
1668static void rtw89_pci_ops_stop(struct rtw89_dev *rtwdev)
1669{
1670	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1671	struct pci_dev *pdev = rtwpci->pdev;
1672
1673	rtw89_pci_disable_intr_lock(rtwdev);
1674	synchronize_irq(pdev->irq);
1675	rtw89_core_napi_stop(rtwdev);
1676}
1677
1678static void rtw89_pci_ops_pause(struct rtw89_dev *rtwdev, bool pause)
1679{
1680	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1681	struct pci_dev *pdev = rtwpci->pdev;
1682
1683	if (pause) {
1684		rtw89_pci_disable_intr_lock(rtwdev);
1685		synchronize_irq(pdev->irq);
1686		if (test_bit(RTW89_FLAG_NAPI_RUNNING, rtwdev->flags))
1687			napi_synchronize(&rtwdev->napi);
1688	} else {
1689		rtw89_pci_enable_intr_lock(rtwdev);
1690		rtw89_pci_tx_kick_off_pending(rtwdev);
1691	}
1692}
1693
1694static
1695void rtw89_pci_switch_bd_idx_addr(struct rtw89_dev *rtwdev, bool low_power)
1696{
1697	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1698	const struct rtw89_pci_info *info = rtwdev->pci_info;
1699	const struct rtw89_pci_bd_idx_addr *bd_idx_addr = info->bd_idx_addr_low_power;
1700	const struct rtw89_pci_ch_dma_addr_set *dma_addr_set = info->dma_addr_set;
1701	struct rtw89_pci_tx_ring *tx_ring;
1702	struct rtw89_pci_rx_ring *rx_ring;
1703	int i;
1704
1705	if (WARN(!bd_idx_addr, "only HCI with low power mode needs this\n"))
1706		return;
1707
1708	for (i = 0; i < RTW89_TXCH_NUM; i++) {
1709		tx_ring = &rtwpci->tx_rings[i];
1710		tx_ring->bd_ring.addr.idx = low_power ?
1711					    bd_idx_addr->tx_bd_addrs[i] :
1712					    dma_addr_set->tx[i].idx;
1713	}
1714
1715	for (i = 0; i < RTW89_RXCH_NUM; i++) {
1716		rx_ring = &rtwpci->rx_rings[i];
1717		rx_ring->bd_ring.addr.idx = low_power ?
1718					    bd_idx_addr->rx_bd_addrs[i] :
1719					    dma_addr_set->rx[i].idx;
1720	}
1721}
1722
1723static void rtw89_pci_ops_switch_mode(struct rtw89_dev *rtwdev, bool low_power)
1724{
1725	enum rtw89_pci_intr_mask_cfg cfg;
1726
1727	WARN(!rtwdev->hci.paused, "HCI isn't paused\n");
1728
1729	cfg = low_power ? RTW89_PCI_INTR_MASK_LOW_POWER : RTW89_PCI_INTR_MASK_NORMAL;
1730	rtw89_chip_config_intr_mask(rtwdev, cfg);
1731	rtw89_pci_switch_bd_idx_addr(rtwdev, low_power);
1732}
1733
1734static void rtw89_pci_ops_write32(struct rtw89_dev *rtwdev, u32 addr, u32 data);
1735
1736static u32 rtw89_pci_ops_read32_cmac(struct rtw89_dev *rtwdev, u32 addr)
1737{
1738	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1739	u32 val = readl(rtwpci->mmap + addr);
1740	int count;
1741
1742	for (count = 0; ; count++) {
1743		if (val != RTW89_R32_DEAD)
1744			return val;
1745		if (count >= MAC_REG_POOL_COUNT) {
1746			rtw89_warn(rtwdev, "addr %#x = %#x\n", addr, val);
1747			return RTW89_R32_DEAD;
1748		}
1749		rtw89_pci_ops_write32(rtwdev, R_AX_CK_EN, B_AX_CMAC_ALLCKEN);
1750		val = readl(rtwpci->mmap + addr);
1751	}
1752
1753	return val;
1754}
1755
1756static u8 rtw89_pci_ops_read8(struct rtw89_dev *rtwdev, u32 addr)
1757{
1758	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1759	u32 addr32, val32, shift;
1760
1761	if (!ACCESS_CMAC(addr))
1762		return readb(rtwpci->mmap + addr);
1763
1764	addr32 = addr & ~0x3;
1765	shift = (addr & 0x3) * 8;
1766	val32 = rtw89_pci_ops_read32_cmac(rtwdev, addr32);
1767	return val32 >> shift;
1768}
1769
1770static u16 rtw89_pci_ops_read16(struct rtw89_dev *rtwdev, u32 addr)
1771{
1772	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1773	u32 addr32, val32, shift;
1774
1775	if (!ACCESS_CMAC(addr))
1776		return readw(rtwpci->mmap + addr);
1777
1778	addr32 = addr & ~0x3;
1779	shift = (addr & 0x3) * 8;
1780	val32 = rtw89_pci_ops_read32_cmac(rtwdev, addr32);
1781	return val32 >> shift;
1782}
1783
1784static u32 rtw89_pci_ops_read32(struct rtw89_dev *rtwdev, u32 addr)
1785{
1786	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1787
1788	if (!ACCESS_CMAC(addr))
1789		return readl(rtwpci->mmap + addr);
1790
1791	return rtw89_pci_ops_read32_cmac(rtwdev, addr);
1792}
1793
1794static void rtw89_pci_ops_write8(struct rtw89_dev *rtwdev, u32 addr, u8 data)
1795{
1796	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1797
1798	writeb(data, rtwpci->mmap + addr);
1799}
1800
1801static void rtw89_pci_ops_write16(struct rtw89_dev *rtwdev, u32 addr, u16 data)
1802{
1803	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1804
1805	writew(data, rtwpci->mmap + addr);
1806}
1807
1808static void rtw89_pci_ops_write32(struct rtw89_dev *rtwdev, u32 addr, u32 data)
1809{
1810	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
1811
1812	writel(data, rtwpci->mmap + addr);
1813}
1814
1815static void rtw89_pci_ctrl_dma_trx(struct rtw89_dev *rtwdev, bool enable)
1816{
1817	const struct rtw89_pci_info *info = rtwdev->pci_info;
1818
1819	if (enable)
1820		rtw89_write32_set(rtwdev, info->init_cfg_reg,
1821				  info->rxhci_en_bit | info->txhci_en_bit);
1822	else
1823		rtw89_write32_clr(rtwdev, info->init_cfg_reg,
1824				  info->rxhci_en_bit | info->txhci_en_bit);
1825}
1826
1827static void rtw89_pci_ctrl_dma_io(struct rtw89_dev *rtwdev, bool enable)
1828{
1829	const struct rtw89_pci_info *info = rtwdev->pci_info;
1830	const struct rtw89_reg_def *reg = &info->dma_io_stop;
1831
1832	if (enable)
1833		rtw89_write32_clr(rtwdev, reg->addr, reg->mask);
1834	else
1835		rtw89_write32_set(rtwdev, reg->addr, reg->mask);
1836}
1837
1838void rtw89_pci_ctrl_dma_all(struct rtw89_dev *rtwdev, bool enable)
1839{
1840	rtw89_pci_ctrl_dma_io(rtwdev, enable);
1841	rtw89_pci_ctrl_dma_trx(rtwdev, enable);
1842}
1843
1844static int rtw89_pci_check_mdio(struct rtw89_dev *rtwdev, u8 addr, u8 speed, u16 rw_bit)
1845{
1846	u16 val;
1847
1848	rtw89_write8(rtwdev, R_AX_MDIO_CFG, addr & 0x1F);
1849
1850	val = rtw89_read16(rtwdev, R_AX_MDIO_CFG);
1851	switch (speed) {
1852	case PCIE_PHY_GEN1:
1853		if (addr < 0x20)
1854			val = u16_replace_bits(val, MDIO_PG0_G1, B_AX_MDIO_PHY_ADDR_MASK);
1855		else
1856			val = u16_replace_bits(val, MDIO_PG1_G1, B_AX_MDIO_PHY_ADDR_MASK);
1857		break;
1858	case PCIE_PHY_GEN2:
1859		if (addr < 0x20)
1860			val = u16_replace_bits(val, MDIO_PG0_G2, B_AX_MDIO_PHY_ADDR_MASK);
1861		else
1862			val = u16_replace_bits(val, MDIO_PG1_G2, B_AX_MDIO_PHY_ADDR_MASK);
1863		break;
1864	default:
1865		rtw89_err(rtwdev, "[ERR]Error Speed %d!\n", speed);
1866		return -EINVAL;
1867	}
1868	rtw89_write16(rtwdev, R_AX_MDIO_CFG, val);
1869	rtw89_write16_set(rtwdev, R_AX_MDIO_CFG, rw_bit);
1870
1871	return read_poll_timeout(rtw89_read16, val, !(val & rw_bit), 10, 2000,
1872				 false, rtwdev, R_AX_MDIO_CFG);
1873}
1874
1875static int
1876rtw89_read16_mdio(struct rtw89_dev *rtwdev, u8 addr, u8 speed, u16 *val)
1877{
1878	int ret;
1879
1880	ret = rtw89_pci_check_mdio(rtwdev, addr, speed, B_AX_MDIO_RFLAG);
1881	if (ret) {
1882		rtw89_err(rtwdev, "[ERR]MDIO R16 0x%X fail ret=%d!\n", addr, ret);
1883		return ret;
1884	}
1885	*val = rtw89_read16(rtwdev, R_AX_MDIO_RDATA);
1886
1887	return 0;
1888}
1889
1890static int
1891rtw89_write16_mdio(struct rtw89_dev *rtwdev, u8 addr, u16 data, u8 speed)
1892{
1893	int ret;
1894
1895	rtw89_write16(rtwdev, R_AX_MDIO_WDATA, data);
1896	ret = rtw89_pci_check_mdio(rtwdev, addr, speed, B_AX_MDIO_WFLAG);
1897	if (ret) {
1898		rtw89_err(rtwdev, "[ERR]MDIO W16 0x%X = %x fail ret=%d!\n", addr, data, ret);
1899		return ret;
1900	}
1901
1902	return 0;
1903}
1904
1905static int
1906rtw89_write16_mdio_mask(struct rtw89_dev *rtwdev, u8 addr, u16 mask, u16 data, u8 speed)
1907{
1908	u32 shift;
1909	int ret;
1910	u16 val;
1911
1912	ret = rtw89_read16_mdio(rtwdev, addr, speed, &val);
1913	if (ret)
1914		return ret;
1915
1916	shift = __ffs(mask);
1917	val &= ~mask;
1918	val |= ((data << shift) & mask);
1919
1920	ret = rtw89_write16_mdio(rtwdev, addr, val, speed);
1921	if (ret)
1922		return ret;
1923
1924	return 0;
1925}
1926
1927static int rtw89_write16_mdio_set(struct rtw89_dev *rtwdev, u8 addr, u16 mask, u8 speed)
1928{
1929	int ret;
1930	u16 val;
1931
1932	ret = rtw89_read16_mdio(rtwdev, addr, speed, &val);
1933	if (ret)
1934		return ret;
1935	ret = rtw89_write16_mdio(rtwdev, addr, val | mask, speed);
1936	if (ret)
1937		return ret;
1938
1939	return 0;
1940}
1941
1942static int rtw89_write16_mdio_clr(struct rtw89_dev *rtwdev, u8 addr, u16 mask, u8 speed)
1943{
1944	int ret;
1945	u16 val;
1946
1947	ret = rtw89_read16_mdio(rtwdev, addr, speed, &val);
1948	if (ret)
1949		return ret;
1950	ret = rtw89_write16_mdio(rtwdev, addr, val & ~mask, speed);
1951	if (ret)
1952		return ret;
1953
1954	return 0;
1955}
1956
1957static int rtw89_dbi_write8(struct rtw89_dev *rtwdev, u16 addr, u8 data)
1958{
1959	u16 addr_2lsb = addr & B_AX_DBI_2LSB;
1960	u16 write_addr;
1961	u8 flag;
1962	int ret;
1963
1964	write_addr = addr & B_AX_DBI_ADDR_MSK;
1965	write_addr |= u16_encode_bits(BIT(addr_2lsb), B_AX_DBI_WREN_MSK);
1966	rtw89_write8(rtwdev, R_AX_DBI_WDATA + addr_2lsb, data);
1967	rtw89_write16(rtwdev, R_AX_DBI_FLAG, write_addr);
1968	rtw89_write8(rtwdev, R_AX_DBI_FLAG + 2, B_AX_DBI_WFLAG >> 16);
1969
1970	ret = read_poll_timeout_atomic(rtw89_read8, flag, !flag, 10,
1971				       10 * RTW89_PCI_WR_RETRY_CNT, false,
1972				       rtwdev, R_AX_DBI_FLAG + 2);
1973	if (ret)
1974		rtw89_err(rtwdev, "failed to write DBI register, addr=0x%X\n",
1975			  addr);
1976
1977	return ret;
1978}
1979
1980static int rtw89_dbi_read8(struct rtw89_dev *rtwdev, u16 addr, u8 *value)
1981{
1982	u16 read_addr = addr & B_AX_DBI_ADDR_MSK;
1983	u8 flag;
1984	int ret;
1985
1986	rtw89_write16(rtwdev, R_AX_DBI_FLAG, read_addr);
1987	rtw89_write8(rtwdev, R_AX_DBI_FLAG + 2, B_AX_DBI_RFLAG >> 16);
1988
1989	ret = read_poll_timeout_atomic(rtw89_read8, flag, !flag, 10,
1990				       10 * RTW89_PCI_WR_RETRY_CNT, false,
1991				       rtwdev, R_AX_DBI_FLAG + 2);
1992	if (ret) {
1993		rtw89_err(rtwdev, "failed to read DBI register, addr=0x%X\n",
1994			  addr);
1995		return ret;
1996	}
1997
1998	read_addr = R_AX_DBI_RDATA + (addr & 3);
1999	*value = rtw89_read8(rtwdev, read_addr);
2000
2001	return 0;
2002}
2003
2004static int rtw89_pci_write_config_byte(struct rtw89_dev *rtwdev, u16 addr,
2005				       u8 data)
2006{
2007	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2008	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2009	struct pci_dev *pdev = rtwpci->pdev;
2010	int ret;
2011
2012	ret = pci_write_config_byte(pdev, addr, data);
2013	if (!ret)
2014		return 0;
2015
2016	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B)
2017		ret = rtw89_dbi_write8(rtwdev, addr, data);
2018
2019	return ret;
2020}
2021
2022static int rtw89_pci_read_config_byte(struct rtw89_dev *rtwdev, u16 addr,
2023				      u8 *value)
2024{
2025	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2026	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2027	struct pci_dev *pdev = rtwpci->pdev;
2028	int ret;
2029
2030	ret = pci_read_config_byte(pdev, addr, value);
2031	if (!ret)
2032		return 0;
2033
2034	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B)
2035		ret = rtw89_dbi_read8(rtwdev, addr, value);
2036
2037	return ret;
2038}
2039
2040static int rtw89_pci_config_byte_set(struct rtw89_dev *rtwdev, u16 addr,
2041				     u8 bit)
2042{
2043	u8 value;
2044	int ret;
2045
2046	ret = rtw89_pci_read_config_byte(rtwdev, addr, &value);
2047	if (ret)
2048		return ret;
2049
2050	value |= bit;
2051	ret = rtw89_pci_write_config_byte(rtwdev, addr, value);
2052
2053	return ret;
2054}
2055
2056static int rtw89_pci_config_byte_clr(struct rtw89_dev *rtwdev, u16 addr,
2057				     u8 bit)
2058{
2059	u8 value;
2060	int ret;
2061
2062	ret = rtw89_pci_read_config_byte(rtwdev, addr, &value);
2063	if (ret)
2064		return ret;
2065
2066	value &= ~bit;
2067	ret = rtw89_pci_write_config_byte(rtwdev, addr, value);
2068
2069	return ret;
2070}
2071
2072static int
2073__get_target(struct rtw89_dev *rtwdev, u16 *target, enum rtw89_pcie_phy phy_rate)
2074{
2075	u16 val, tar;
2076	int ret;
2077
2078	/* Enable counter */
2079	ret = rtw89_read16_mdio(rtwdev, RAC_CTRL_PPR_V1, phy_rate, &val);
2080	if (ret)
2081		return ret;
2082	ret = rtw89_write16_mdio(rtwdev, RAC_CTRL_PPR_V1, val & ~B_AX_CLK_CALIB_EN,
2083				 phy_rate);
2084	if (ret)
2085		return ret;
2086	ret = rtw89_write16_mdio(rtwdev, RAC_CTRL_PPR_V1, val | B_AX_CLK_CALIB_EN,
2087				 phy_rate);
2088	if (ret)
2089		return ret;
2090
2091	fsleep(300);
2092
2093	ret = rtw89_read16_mdio(rtwdev, RAC_CTRL_PPR_V1, phy_rate, &tar);
2094	if (ret)
2095		return ret;
2096	ret = rtw89_write16_mdio(rtwdev, RAC_CTRL_PPR_V1, val & ~B_AX_CLK_CALIB_EN,
2097				 phy_rate);
2098	if (ret)
2099		return ret;
2100
2101	tar = tar & 0x0FFF;
2102	if (tar == 0 || tar == 0x0FFF) {
2103		rtw89_err(rtwdev, "[ERR]Get target failed.\n");
2104		return -EINVAL;
2105	}
2106
2107	*target = tar;
2108
2109	return 0;
2110}
2111
2112static int rtw89_pci_autok_x(struct rtw89_dev *rtwdev)
2113{
2114	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2115	int ret;
2116
2117	if (chip_id != RTL8852B && chip_id != RTL8851B)
2118		return 0;
2119
2120	ret = rtw89_write16_mdio_mask(rtwdev, RAC_REG_FLD_0, BAC_AUTOK_N_MASK,
2121				      PCIE_AUTOK_4, PCIE_PHY_GEN1);
2122	return ret;
2123}
2124
2125static int rtw89_pci_auto_refclk_cal(struct rtw89_dev *rtwdev, bool autook_en)
2126{
2127	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2128	enum rtw89_pcie_phy phy_rate;
2129	u16 val16, mgn_set, div_set, tar;
2130	u8 val8, bdr_ori;
2131	bool l1_flag = false;
2132	int ret = 0;
2133
2134	if (chip_id != RTL8852B && chip_id != RTL8851B)
2135		return 0;
2136
2137	ret = rtw89_pci_read_config_byte(rtwdev, RTW89_PCIE_PHY_RATE, &val8);
2138	if (ret) {
2139		rtw89_err(rtwdev, "[ERR]pci config read %X\n",
2140			  RTW89_PCIE_PHY_RATE);
2141		return ret;
2142	}
2143
2144	if (FIELD_GET(RTW89_PCIE_PHY_RATE_MASK, val8) == 0x1) {
2145		phy_rate = PCIE_PHY_GEN1;
2146	} else if (FIELD_GET(RTW89_PCIE_PHY_RATE_MASK, val8) == 0x2) {
2147		phy_rate = PCIE_PHY_GEN2;
2148	} else {
2149		rtw89_err(rtwdev, "[ERR]PCIe PHY rate %#x not support\n", val8);
2150		return -EOPNOTSUPP;
2151	}
2152	/* Disable L1BD */
2153	ret = rtw89_pci_read_config_byte(rtwdev, RTW89_PCIE_L1_CTRL, &bdr_ori);
2154	if (ret) {
2155		rtw89_err(rtwdev, "[ERR]pci config read %X\n", RTW89_PCIE_L1_CTRL);
2156		return ret;
2157	}
2158
2159	if (bdr_ori & RTW89_PCIE_BIT_L1) {
2160		ret = rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_L1_CTRL,
2161						  bdr_ori & ~RTW89_PCIE_BIT_L1);
2162		if (ret) {
2163			rtw89_err(rtwdev, "[ERR]pci config write %X\n",
2164				  RTW89_PCIE_L1_CTRL);
2165			return ret;
2166		}
2167		l1_flag = true;
2168	}
2169
2170	ret = rtw89_read16_mdio(rtwdev, RAC_CTRL_PPR_V1, phy_rate, &val16);
2171	if (ret) {
2172		rtw89_err(rtwdev, "[ERR]mdio_r16_pcie %X\n", RAC_CTRL_PPR_V1);
2173		goto end;
2174	}
2175
2176	if (val16 & B_AX_CALIB_EN) {
2177		ret = rtw89_write16_mdio(rtwdev, RAC_CTRL_PPR_V1,
2178					 val16 & ~B_AX_CALIB_EN, phy_rate);
2179		if (ret) {
2180			rtw89_err(rtwdev, "[ERR]mdio_w16_pcie %X\n", RAC_CTRL_PPR_V1);
2181			goto end;
2182		}
2183	}
2184
2185	if (!autook_en)
2186		goto end;
2187	/* Set div */
2188	ret = rtw89_write16_mdio_clr(rtwdev, RAC_CTRL_PPR_V1, B_AX_DIV, phy_rate);
2189	if (ret) {
2190		rtw89_err(rtwdev, "[ERR]mdio_w16_pcie %X\n", RAC_CTRL_PPR_V1);
2191		goto end;
2192	}
2193
2194	/* Obtain div and margin */
2195	ret = __get_target(rtwdev, &tar, phy_rate);
2196	if (ret) {
2197		rtw89_err(rtwdev, "[ERR]1st get target fail %d\n", ret);
2198		goto end;
2199	}
2200
2201	mgn_set = tar * INTF_INTGRA_HOSTREF_V1 / INTF_INTGRA_MINREF_V1 - tar;
2202
2203	if (mgn_set >= 128) {
2204		div_set = 0x0003;
2205		mgn_set = 0x000F;
2206	} else if (mgn_set >= 64) {
2207		div_set = 0x0003;
2208		mgn_set >>= 3;
2209	} else if (mgn_set >= 32) {
2210		div_set = 0x0002;
2211		mgn_set >>= 2;
2212	} else if (mgn_set >= 16) {
2213		div_set = 0x0001;
2214		mgn_set >>= 1;
2215	} else if (mgn_set == 0) {
2216		rtw89_err(rtwdev, "[ERR]cal mgn is 0,tar = %d\n", tar);
2217		goto end;
2218	} else {
2219		div_set = 0x0000;
2220	}
2221
2222	ret = rtw89_read16_mdio(rtwdev, RAC_CTRL_PPR_V1, phy_rate, &val16);
2223	if (ret) {
2224		rtw89_err(rtwdev, "[ERR]mdio_r16_pcie %X\n", RAC_CTRL_PPR_V1);
2225		goto end;
2226	}
2227
2228	val16 |= u16_encode_bits(div_set, B_AX_DIV);
2229
2230	ret = rtw89_write16_mdio(rtwdev, RAC_CTRL_PPR_V1, val16, phy_rate);
2231	if (ret) {
2232		rtw89_err(rtwdev, "[ERR]mdio_w16_pcie %X\n", RAC_CTRL_PPR_V1);
2233		goto end;
2234	}
2235
2236	ret = __get_target(rtwdev, &tar, phy_rate);
2237	if (ret) {
2238		rtw89_err(rtwdev, "[ERR]2nd get target fail %d\n", ret);
2239		goto end;
2240	}
2241
2242	rtw89_debug(rtwdev, RTW89_DBG_HCI, "[TRACE]target = 0x%X, div = 0x%X, margin = 0x%X\n",
2243		    tar, div_set, mgn_set);
2244	ret = rtw89_write16_mdio(rtwdev, RAC_SET_PPR_V1,
2245				 (tar & 0x0FFF) | (mgn_set << 12), phy_rate);
2246	if (ret) {
2247		rtw89_err(rtwdev, "[ERR]mdio_w16_pcie %X\n", RAC_SET_PPR_V1);
2248		goto end;
2249	}
2250
2251	/* Enable function */
2252	ret = rtw89_write16_mdio_set(rtwdev, RAC_CTRL_PPR_V1, B_AX_CALIB_EN, phy_rate);
2253	if (ret) {
2254		rtw89_err(rtwdev, "[ERR]mdio_w16_pcie %X\n", RAC_CTRL_PPR_V1);
2255		goto end;
2256	}
2257
2258	/* CLK delay = 0 */
2259	ret = rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_CLK_CTRL,
2260					  PCIE_CLKDLY_HW_0);
2261
2262end:
2263	/* Set L1BD to ori */
2264	if (l1_flag) {
2265		ret = rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_L1_CTRL,
2266						  bdr_ori);
2267		if (ret) {
2268			rtw89_err(rtwdev, "[ERR]pci config write %X\n",
2269				  RTW89_PCIE_L1_CTRL);
2270			return ret;
2271		}
2272	}
2273
2274	return ret;
2275}
2276
2277static int rtw89_pci_deglitch_setting(struct rtw89_dev *rtwdev)
2278{
2279	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2280	int ret;
2281
2282	if (chip_id == RTL8852A) {
2283		ret = rtw89_write16_mdio_clr(rtwdev, RAC_ANA24, B_AX_DEGLITCH,
2284					     PCIE_PHY_GEN1);
2285		if (ret)
2286			return ret;
2287		ret = rtw89_write16_mdio_clr(rtwdev, RAC_ANA24, B_AX_DEGLITCH,
2288					     PCIE_PHY_GEN2);
2289		if (ret)
2290			return ret;
2291	} else if (chip_id == RTL8852C) {
2292		rtw89_write16_clr(rtwdev, R_RAC_DIRECT_OFFSET_G1 + RAC_ANA24 * 2,
2293				  B_AX_DEGLITCH);
2294		rtw89_write16_clr(rtwdev, R_RAC_DIRECT_OFFSET_G2 + RAC_ANA24 * 2,
2295				  B_AX_DEGLITCH);
2296	}
2297
2298	return 0;
2299}
2300
2301static void rtw89_pci_rxdma_prefth(struct rtw89_dev *rtwdev)
2302{
2303	if (rtwdev->chip->chip_id != RTL8852A)
2304		return;
2305
2306	rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_DIS_RXDMA_PRE);
2307}
2308
2309static void rtw89_pci_l1off_pwroff(struct rtw89_dev *rtwdev)
2310{
2311	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2312
2313	if (chip_id != RTL8852A && chip_id != RTL8852B && chip_id != RTL8851B)
2314		return;
2315
2316	rtw89_write32_clr(rtwdev, R_AX_PCIE_PS_CTRL, B_AX_L1OFF_PWR_OFF_EN);
2317}
2318
2319static u32 rtw89_pci_l2_rxen_lat(struct rtw89_dev *rtwdev)
2320{
2321	int ret;
2322
2323	if (rtwdev->chip->chip_id != RTL8852A)
2324		return 0;
2325
2326	ret = rtw89_write16_mdio_clr(rtwdev, RAC_ANA26, B_AX_RXEN,
2327				     PCIE_PHY_GEN1);
2328	if (ret)
2329		return ret;
2330
2331	ret = rtw89_write16_mdio_clr(rtwdev, RAC_ANA26, B_AX_RXEN,
2332				     PCIE_PHY_GEN2);
2333	if (ret)
2334		return ret;
2335
2336	return 0;
2337}
2338
2339static void rtw89_pci_aphy_pwrcut(struct rtw89_dev *rtwdev)
2340{
2341	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2342
2343	if (chip_id != RTL8852A && chip_id != RTL8852B && chip_id != RTL8851B)
2344		return;
2345
2346	rtw89_write32_clr(rtwdev, R_AX_SYS_PW_CTRL, B_AX_PSUS_OFF_CAPC_EN);
2347}
2348
2349static void rtw89_pci_hci_ldo(struct rtw89_dev *rtwdev)
2350{
2351	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2352
2353	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
2354		rtw89_write32_set(rtwdev, R_AX_SYS_SDIO_CTRL,
2355				  B_AX_PCIE_DIS_L2_CTRL_LDO_HCI);
2356		rtw89_write32_clr(rtwdev, R_AX_SYS_SDIO_CTRL,
2357				  B_AX_PCIE_DIS_WLSUS_AFT_PDN);
2358	} else if (rtwdev->chip->chip_id == RTL8852C) {
2359		rtw89_write32_clr(rtwdev, R_AX_SYS_SDIO_CTRL,
2360				  B_AX_PCIE_DIS_L2_CTRL_LDO_HCI);
2361	}
2362}
2363
2364static int rtw89_pci_dphy_delay(struct rtw89_dev *rtwdev)
2365{
2366	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2367
2368	if (chip_id != RTL8852B && chip_id != RTL8851B)
2369		return 0;
2370
2371	return rtw89_write16_mdio_mask(rtwdev, RAC_REG_REV2, BAC_CMU_EN_DLY_MASK,
2372				       PCIE_DPHY_DLY_25US, PCIE_PHY_GEN1);
2373}
2374
2375static void rtw89_pci_power_wake(struct rtw89_dev *rtwdev, bool pwr_up)
2376{
2377	if (pwr_up)
2378		rtw89_write32_set(rtwdev, R_AX_HCI_OPT_CTRL, BIT_WAKE_CTRL);
2379	else
2380		rtw89_write32_clr(rtwdev, R_AX_HCI_OPT_CTRL, BIT_WAKE_CTRL);
2381}
2382
2383static void rtw89_pci_autoload_hang(struct rtw89_dev *rtwdev)
2384{
2385	if (rtwdev->chip->chip_id != RTL8852C)
2386		return;
2387
2388	rtw89_write32_set(rtwdev, R_AX_PCIE_BG_CLR, B_AX_BG_CLR_ASYNC_M3);
2389	rtw89_write32_clr(rtwdev, R_AX_PCIE_BG_CLR, B_AX_BG_CLR_ASYNC_M3);
2390}
2391
2392static void rtw89_pci_l12_vmain(struct rtw89_dev *rtwdev)
2393{
2394	if (!(rtwdev->chip->chip_id == RTL8852C && rtwdev->hal.cv == CHIP_CAV))
2395		return;
2396
2397	rtw89_write32_set(rtwdev, R_AX_SYS_SDIO_CTRL, B_AX_PCIE_FORCE_PWR_NGAT);
2398}
2399
2400static void rtw89_pci_gen2_force_ib(struct rtw89_dev *rtwdev)
2401{
2402	if (!(rtwdev->chip->chip_id == RTL8852C && rtwdev->hal.cv == CHIP_CAV))
2403		return;
2404
2405	rtw89_write32_set(rtwdev, R_AX_PMC_DBG_CTRL2,
2406			  B_AX_SYSON_DIS_PMCR_AX_WRMSK);
2407	rtw89_write32_set(rtwdev, R_AX_HCI_BG_CTRL, B_AX_BG_CLR_ASYNC_M3);
2408	rtw89_write32_clr(rtwdev, R_AX_PMC_DBG_CTRL2,
2409			  B_AX_SYSON_DIS_PMCR_AX_WRMSK);
2410}
2411
2412static void rtw89_pci_l1_ent_lat(struct rtw89_dev *rtwdev)
2413{
2414	if (rtwdev->chip->chip_id != RTL8852C)
2415		return;
2416
2417	rtw89_write32_clr(rtwdev, R_AX_PCIE_PS_CTRL_V1, B_AX_SEL_REQ_ENTR_L1);
2418}
2419
2420static void rtw89_pci_wd_exit_l1(struct rtw89_dev *rtwdev)
2421{
2422	if (rtwdev->chip->chip_id != RTL8852C)
2423		return;
2424
2425	rtw89_write32_set(rtwdev, R_AX_PCIE_PS_CTRL_V1, B_AX_DMAC0_EXIT_L1_EN);
2426}
2427
2428static void rtw89_pci_set_sic(struct rtw89_dev *rtwdev)
2429{
2430	if (rtwdev->chip->chip_id == RTL8852C)
2431		return;
2432
2433	rtw89_write32_clr(rtwdev, R_AX_PCIE_EXP_CTRL,
2434			  B_AX_SIC_EN_FORCE_CLKREQ);
2435}
2436
2437static void rtw89_pci_set_lbc(struct rtw89_dev *rtwdev)
2438{
2439	const struct rtw89_pci_info *info = rtwdev->pci_info;
2440	u32 lbc;
2441
2442	if (rtwdev->chip->chip_id == RTL8852C)
2443		return;
2444
2445	lbc = rtw89_read32(rtwdev, R_AX_LBC_WATCHDOG);
2446	if (info->lbc_en == MAC_AX_PCIE_ENABLE) {
2447		lbc = u32_replace_bits(lbc, info->lbc_tmr, B_AX_LBC_TIMER);
2448		lbc |= B_AX_LBC_FLAG | B_AX_LBC_EN;
2449		rtw89_write32(rtwdev, R_AX_LBC_WATCHDOG, lbc);
2450	} else {
2451		lbc &= ~B_AX_LBC_EN;
2452	}
2453	rtw89_write32_set(rtwdev, R_AX_LBC_WATCHDOG, lbc);
2454}
2455
2456static void rtw89_pci_set_io_rcy(struct rtw89_dev *rtwdev)
2457{
2458	const struct rtw89_pci_info *info = rtwdev->pci_info;
2459	u32 val32;
2460
2461	if (rtwdev->chip->chip_id != RTL8852C)
2462		return;
2463
2464	if (info->io_rcy_en == MAC_AX_PCIE_ENABLE) {
2465		val32 = FIELD_PREP(B_AX_PCIE_WDT_TIMER_M1_MASK,
2466				   info->io_rcy_tmr);
2467		rtw89_write32(rtwdev, R_AX_PCIE_WDT_TIMER_M1, val32);
2468		rtw89_write32(rtwdev, R_AX_PCIE_WDT_TIMER_M2, val32);
2469		rtw89_write32(rtwdev, R_AX_PCIE_WDT_TIMER_E0, val32);
2470
2471		rtw89_write32_set(rtwdev, R_AX_PCIE_IO_RCY_M1, B_AX_PCIE_IO_RCY_WDT_MODE_M1);
2472		rtw89_write32_set(rtwdev, R_AX_PCIE_IO_RCY_M2, B_AX_PCIE_IO_RCY_WDT_MODE_M2);
2473		rtw89_write32_set(rtwdev, R_AX_PCIE_IO_RCY_E0, B_AX_PCIE_IO_RCY_WDT_MODE_E0);
2474	} else {
2475		rtw89_write32_clr(rtwdev, R_AX_PCIE_IO_RCY_M1, B_AX_PCIE_IO_RCY_WDT_MODE_M1);
2476		rtw89_write32_clr(rtwdev, R_AX_PCIE_IO_RCY_M2, B_AX_PCIE_IO_RCY_WDT_MODE_M2);
2477		rtw89_write32_clr(rtwdev, R_AX_PCIE_IO_RCY_E0, B_AX_PCIE_IO_RCY_WDT_MODE_E0);
2478	}
2479
2480	rtw89_write32_clr(rtwdev, R_AX_PCIE_IO_RCY_S1, B_AX_PCIE_IO_RCY_WDT_MODE_S1);
2481}
2482
2483static void rtw89_pci_set_dbg(struct rtw89_dev *rtwdev)
2484{
2485	if (rtwdev->chip->chip_id == RTL8852C)
2486		return;
2487
2488	rtw89_write32_set(rtwdev, R_AX_PCIE_DBG_CTRL,
2489			  B_AX_ASFF_FULL_NO_STK | B_AX_EN_STUCK_DBG);
2490
2491	if (rtwdev->chip->chip_id == RTL8852A)
2492		rtw89_write32_set(rtwdev, R_AX_PCIE_EXP_CTRL,
2493				  B_AX_EN_CHKDSC_NO_RX_STUCK);
2494}
2495
2496static void rtw89_pci_set_keep_reg(struct rtw89_dev *rtwdev)
2497{
2498	if (rtwdev->chip->chip_id == RTL8852C)
2499		return;
2500
2501	rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1,
2502			  B_AX_PCIE_TXRST_KEEP_REG | B_AX_PCIE_RXRST_KEEP_REG);
2503}
2504
2505static void rtw89_pci_clr_idx_all_ax(struct rtw89_dev *rtwdev)
2506{
2507	const struct rtw89_pci_info *info = rtwdev->pci_info;
2508	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2509	u32 val = B_AX_CLR_ACH0_IDX | B_AX_CLR_ACH1_IDX | B_AX_CLR_ACH2_IDX |
2510		  B_AX_CLR_ACH3_IDX | B_AX_CLR_CH8_IDX | B_AX_CLR_CH9_IDX |
2511		  B_AX_CLR_CH12_IDX;
2512	u32 rxbd_rwptr_clr = info->rxbd_rwptr_clr_reg;
2513	u32 txbd_rwptr_clr2 = info->txbd_rwptr_clr2_reg;
2514
2515	if (chip_id == RTL8852A || chip_id == RTL8852C)
2516		val |= B_AX_CLR_ACH4_IDX | B_AX_CLR_ACH5_IDX |
2517		       B_AX_CLR_ACH6_IDX | B_AX_CLR_ACH7_IDX;
2518	/* clear DMA indexes */
2519	rtw89_write32_set(rtwdev, R_AX_TXBD_RWPTR_CLR1, val);
2520	if (chip_id == RTL8852A || chip_id == RTL8852C)
2521		rtw89_write32_set(rtwdev, txbd_rwptr_clr2,
2522				  B_AX_CLR_CH10_IDX | B_AX_CLR_CH11_IDX);
2523	rtw89_write32_set(rtwdev, rxbd_rwptr_clr,
2524			  B_AX_CLR_RXQ_IDX | B_AX_CLR_RPQ_IDX);
2525}
2526
2527static int rtw89_pci_poll_txdma_ch_idle_ax(struct rtw89_dev *rtwdev)
2528{
2529	const struct rtw89_pci_info *info = rtwdev->pci_info;
2530	u32 ret, check, dma_busy;
2531	u32 dma_busy1 = info->dma_busy1.addr;
2532	u32 dma_busy2 = info->dma_busy2_reg;
2533
2534	check = info->dma_busy1.mask;
2535
2536	ret = read_poll_timeout(rtw89_read32, dma_busy, (dma_busy & check) == 0,
2537				10, 100, false, rtwdev, dma_busy1);
2538	if (ret)
2539		return ret;
2540
2541	if (!dma_busy2)
2542		return 0;
2543
2544	check = B_AX_CH10_BUSY | B_AX_CH11_BUSY;
2545
2546	ret = read_poll_timeout(rtw89_read32, dma_busy, (dma_busy & check) == 0,
2547				10, 100, false, rtwdev, dma_busy2);
2548	if (ret)
2549		return ret;
2550
2551	return 0;
2552}
2553
2554static int rtw89_pci_poll_rxdma_ch_idle_ax(struct rtw89_dev *rtwdev)
2555{
2556	const struct rtw89_pci_info *info = rtwdev->pci_info;
2557	u32 ret, check, dma_busy;
2558	u32 dma_busy3 = info->dma_busy3_reg;
2559
2560	check = B_AX_RXQ_BUSY | B_AX_RPQ_BUSY;
2561
2562	ret = read_poll_timeout(rtw89_read32, dma_busy, (dma_busy & check) == 0,
2563				10, 100, false, rtwdev, dma_busy3);
2564	if (ret)
2565		return ret;
2566
2567	return 0;
2568}
2569
2570static int rtw89_pci_poll_dma_all_idle(struct rtw89_dev *rtwdev)
2571{
2572	u32 ret;
2573
2574	ret = rtw89_pci_poll_txdma_ch_idle_ax(rtwdev);
2575	if (ret) {
2576		rtw89_err(rtwdev, "txdma ch busy\n");
2577		return ret;
2578	}
2579
2580	ret = rtw89_pci_poll_rxdma_ch_idle_ax(rtwdev);
2581	if (ret) {
2582		rtw89_err(rtwdev, "rxdma ch busy\n");
2583		return ret;
2584	}
2585
2586	return 0;
2587}
2588
2589static int rtw89_pci_mode_op(struct rtw89_dev *rtwdev)
2590{
2591	const struct rtw89_pci_info *info = rtwdev->pci_info;
2592	enum mac_ax_bd_trunc_mode txbd_trunc_mode = info->txbd_trunc_mode;
2593	enum mac_ax_bd_trunc_mode rxbd_trunc_mode = info->rxbd_trunc_mode;
2594	enum mac_ax_rxbd_mode rxbd_mode = info->rxbd_mode;
2595	enum mac_ax_tag_mode tag_mode = info->tag_mode;
2596	enum mac_ax_wd_dma_intvl wd_dma_idle_intvl = info->wd_dma_idle_intvl;
2597	enum mac_ax_wd_dma_intvl wd_dma_act_intvl = info->wd_dma_act_intvl;
2598	enum mac_ax_tx_burst tx_burst = info->tx_burst;
2599	enum mac_ax_rx_burst rx_burst = info->rx_burst;
2600	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2601	u8 cv = rtwdev->hal.cv;
2602	u32 val32;
2603
2604	if (txbd_trunc_mode == MAC_AX_BD_TRUNC) {
2605		if (chip_id == RTL8852A && cv == CHIP_CBV)
2606			rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_TX_TRUNC_MODE);
2607	} else if (txbd_trunc_mode == MAC_AX_BD_NORM) {
2608		if (chip_id == RTL8852A || chip_id == RTL8852B)
2609			rtw89_write32_clr(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_TX_TRUNC_MODE);
2610	}
2611
2612	if (rxbd_trunc_mode == MAC_AX_BD_TRUNC) {
2613		if (chip_id == RTL8852A && cv == CHIP_CBV)
2614			rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_RX_TRUNC_MODE);
2615	} else if (rxbd_trunc_mode == MAC_AX_BD_NORM) {
2616		if (chip_id == RTL8852A || chip_id == RTL8852B)
2617			rtw89_write32_clr(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_RX_TRUNC_MODE);
2618	}
2619
2620	if (rxbd_mode == MAC_AX_RXBD_PKT) {
2621		rtw89_write32_clr(rtwdev, info->init_cfg_reg, info->rxbd_mode_bit);
2622	} else if (rxbd_mode == MAC_AX_RXBD_SEP) {
2623		rtw89_write32_set(rtwdev, info->init_cfg_reg, info->rxbd_mode_bit);
2624
2625		if (chip_id == RTL8852A || chip_id == RTL8852B)
2626			rtw89_write32_mask(rtwdev, R_AX_PCIE_INIT_CFG2,
2627					   B_AX_PCIE_RX_APPLEN_MASK, 0);
2628	}
2629
2630	if (chip_id == RTL8852A || chip_id == RTL8852B) {
2631		rtw89_write32_mask(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_PCIE_MAX_TXDMA_MASK, tx_burst);
2632		rtw89_write32_mask(rtwdev, R_AX_PCIE_INIT_CFG1, B_AX_PCIE_MAX_RXDMA_MASK, rx_burst);
2633	} else if (chip_id == RTL8852C) {
2634		rtw89_write32_mask(rtwdev, R_AX_HAXI_INIT_CFG1, B_AX_HAXI_MAX_TXDMA_MASK, tx_burst);
2635		rtw89_write32_mask(rtwdev, R_AX_HAXI_INIT_CFG1, B_AX_HAXI_MAX_RXDMA_MASK, rx_burst);
2636	}
2637
2638	if (chip_id == RTL8852A || chip_id == RTL8852B) {
2639		if (tag_mode == MAC_AX_TAG_SGL) {
2640			val32 = rtw89_read32(rtwdev, R_AX_PCIE_INIT_CFG1) &
2641					    ~B_AX_LATENCY_CONTROL;
2642			rtw89_write32(rtwdev, R_AX_PCIE_INIT_CFG1, val32);
2643		} else if (tag_mode == MAC_AX_TAG_MULTI) {
2644			val32 = rtw89_read32(rtwdev, R_AX_PCIE_INIT_CFG1) |
2645					    B_AX_LATENCY_CONTROL;
2646			rtw89_write32(rtwdev, R_AX_PCIE_INIT_CFG1, val32);
2647		}
2648	}
2649
2650	rtw89_write32_mask(rtwdev, info->exp_ctrl_reg, info->max_tag_num_mask,
2651			   info->multi_tag_num);
2652
2653	if (chip_id == RTL8852A || chip_id == RTL8852B) {
2654		rtw89_write32_mask(rtwdev, R_AX_PCIE_INIT_CFG2, B_AX_WD_ITVL_IDLE,
2655				   wd_dma_idle_intvl);
2656		rtw89_write32_mask(rtwdev, R_AX_PCIE_INIT_CFG2, B_AX_WD_ITVL_ACT,
2657				   wd_dma_act_intvl);
2658	} else if (chip_id == RTL8852C) {
2659		rtw89_write32_mask(rtwdev, R_AX_HAXI_INIT_CFG1, B_AX_WD_ITVL_IDLE_V1_MASK,
2660				   wd_dma_idle_intvl);
2661		rtw89_write32_mask(rtwdev, R_AX_HAXI_INIT_CFG1, B_AX_WD_ITVL_ACT_V1_MASK,
2662				   wd_dma_act_intvl);
2663	}
2664
2665	if (txbd_trunc_mode == MAC_AX_BD_TRUNC) {
2666		rtw89_write32_set(rtwdev, R_AX_TX_ADDRESS_INFO_MODE_SETTING,
2667				  B_AX_HOST_ADDR_INFO_8B_SEL);
2668		rtw89_write32_clr(rtwdev, R_AX_PKTIN_SETTING, B_AX_WD_ADDR_INFO_LENGTH);
2669	} else if (txbd_trunc_mode == MAC_AX_BD_NORM) {
2670		rtw89_write32_clr(rtwdev, R_AX_TX_ADDRESS_INFO_MODE_SETTING,
2671				  B_AX_HOST_ADDR_INFO_8B_SEL);
2672		rtw89_write32_set(rtwdev, R_AX_PKTIN_SETTING, B_AX_WD_ADDR_INFO_LENGTH);
2673	}
2674
2675	return 0;
2676}
2677
2678static int rtw89_pci_ops_deinit(struct rtw89_dev *rtwdev)
2679{
2680	const struct rtw89_pci_info *info = rtwdev->pci_info;
2681
2682	if (rtwdev->chip->chip_id == RTL8852A) {
2683		/* ltr sw trigger */
2684		rtw89_write32_set(rtwdev, R_AX_LTR_CTRL_0, B_AX_APP_LTR_IDLE);
2685	}
2686	info->ltr_set(rtwdev, false);
2687	rtw89_pci_ctrl_dma_all(rtwdev, false);
2688	rtw89_pci_clr_idx_all(rtwdev);
2689
2690	return 0;
2691}
2692
2693static int rtw89_pci_ops_mac_pre_init_ax(struct rtw89_dev *rtwdev)
2694{
2695	const struct rtw89_pci_info *info = rtwdev->pci_info;
2696	int ret;
2697
2698	rtw89_pci_rxdma_prefth(rtwdev);
2699	rtw89_pci_l1off_pwroff(rtwdev);
2700	rtw89_pci_deglitch_setting(rtwdev);
2701	ret = rtw89_pci_l2_rxen_lat(rtwdev);
2702	if (ret) {
2703		rtw89_err(rtwdev, "[ERR] pcie l2 rxen lat %d\n", ret);
2704		return ret;
2705	}
2706
2707	rtw89_pci_aphy_pwrcut(rtwdev);
2708	rtw89_pci_hci_ldo(rtwdev);
2709	rtw89_pci_dphy_delay(rtwdev);
2710
2711	ret = rtw89_pci_autok_x(rtwdev);
2712	if (ret) {
2713		rtw89_err(rtwdev, "[ERR] pcie autok_x fail %d\n", ret);
2714		return ret;
2715	}
2716
2717	ret = rtw89_pci_auto_refclk_cal(rtwdev, false);
2718	if (ret) {
2719		rtw89_err(rtwdev, "[ERR] pcie autok fail %d\n", ret);
2720		return ret;
2721	}
2722
2723	rtw89_pci_power_wake(rtwdev, true);
2724	rtw89_pci_autoload_hang(rtwdev);
2725	rtw89_pci_l12_vmain(rtwdev);
2726	rtw89_pci_gen2_force_ib(rtwdev);
2727	rtw89_pci_l1_ent_lat(rtwdev);
2728	rtw89_pci_wd_exit_l1(rtwdev);
2729	rtw89_pci_set_sic(rtwdev);
2730	rtw89_pci_set_lbc(rtwdev);
2731	rtw89_pci_set_io_rcy(rtwdev);
2732	rtw89_pci_set_dbg(rtwdev);
2733	rtw89_pci_set_keep_reg(rtwdev);
2734
2735	rtw89_write32_set(rtwdev, info->dma_stop1.addr, B_AX_STOP_WPDMA);
2736
2737	/* stop DMA activities */
2738	rtw89_pci_ctrl_dma_all(rtwdev, false);
2739
2740	ret = rtw89_pci_poll_dma_all_idle(rtwdev);
2741	if (ret) {
2742		rtw89_err(rtwdev, "[ERR] poll pcie dma all idle\n");
2743		return ret;
2744	}
2745
2746	rtw89_pci_clr_idx_all(rtwdev);
2747	rtw89_pci_mode_op(rtwdev);
2748
2749	/* fill TRX BD indexes */
2750	rtw89_pci_ops_reset(rtwdev);
2751
2752	ret = rtw89_pci_rst_bdram_ax(rtwdev);
2753	if (ret) {
2754		rtw89_warn(rtwdev, "reset bdram busy\n");
2755		return ret;
2756	}
2757
2758	/* disable all channels except to FW CMD channel to download firmware */
2759	rtw89_pci_ctrl_txdma_ch_ax(rtwdev, false);
2760	rtw89_pci_ctrl_txdma_fw_ch_ax(rtwdev, true);
2761
2762	/* start DMA activities */
2763	rtw89_pci_ctrl_dma_all(rtwdev, true);
2764
2765	return 0;
2766}
2767
2768int rtw89_pci_ltr_set(struct rtw89_dev *rtwdev, bool en)
2769{
2770	u32 val;
2771
2772	if (!en)
2773		return 0;
2774
2775	val = rtw89_read32(rtwdev, R_AX_LTR_CTRL_0);
2776	if (rtw89_pci_ltr_is_err_reg_val(val))
2777		return -EINVAL;
2778	val = rtw89_read32(rtwdev, R_AX_LTR_CTRL_1);
2779	if (rtw89_pci_ltr_is_err_reg_val(val))
2780		return -EINVAL;
2781	val = rtw89_read32(rtwdev, R_AX_LTR_IDLE_LATENCY);
2782	if (rtw89_pci_ltr_is_err_reg_val(val))
2783		return -EINVAL;
2784	val = rtw89_read32(rtwdev, R_AX_LTR_ACTIVE_LATENCY);
2785	if (rtw89_pci_ltr_is_err_reg_val(val))
2786		return -EINVAL;
2787
2788	rtw89_write32_set(rtwdev, R_AX_LTR_CTRL_0, B_AX_LTR_HW_EN | B_AX_LTR_EN |
2789						   B_AX_LTR_WD_NOEMP_CHK);
2790	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_0, B_AX_LTR_SPACE_IDX_MASK,
2791			   PCI_LTR_SPC_500US);
2792	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_0, B_AX_LTR_IDLE_TIMER_IDX_MASK,
2793			   PCI_LTR_IDLE_TIMER_3_2MS);
2794	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_1, B_AX_LTR_RX0_TH_MASK, 0x28);
2795	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_1, B_AX_LTR_RX1_TH_MASK, 0x28);
2796	rtw89_write32(rtwdev, R_AX_LTR_IDLE_LATENCY, 0x90039003);
2797	rtw89_write32(rtwdev, R_AX_LTR_ACTIVE_LATENCY, 0x880b880b);
2798
2799	return 0;
2800}
2801EXPORT_SYMBOL(rtw89_pci_ltr_set);
2802
2803int rtw89_pci_ltr_set_v1(struct rtw89_dev *rtwdev, bool en)
2804{
2805	u32 dec_ctrl;
2806	u32 val32;
2807
2808	val32 = rtw89_read32(rtwdev, R_AX_LTR_CTRL_0);
2809	if (rtw89_pci_ltr_is_err_reg_val(val32))
2810		return -EINVAL;
2811	val32 = rtw89_read32(rtwdev, R_AX_LTR_CTRL_1);
2812	if (rtw89_pci_ltr_is_err_reg_val(val32))
2813		return -EINVAL;
2814	dec_ctrl = rtw89_read32(rtwdev, R_AX_LTR_DEC_CTRL);
2815	if (rtw89_pci_ltr_is_err_reg_val(dec_ctrl))
2816		return -EINVAL;
2817	val32 = rtw89_read32(rtwdev, R_AX_LTR_LATENCY_IDX3);
2818	if (rtw89_pci_ltr_is_err_reg_val(val32))
2819		return -EINVAL;
2820	val32 = rtw89_read32(rtwdev, R_AX_LTR_LATENCY_IDX0);
2821	if (rtw89_pci_ltr_is_err_reg_val(val32))
2822		return -EINVAL;
2823
2824	if (!en) {
2825		dec_ctrl &= ~(LTR_EN_BITS | B_AX_LTR_IDX_DRV_MASK | B_AX_LTR_HW_DEC_EN);
2826		dec_ctrl |= FIELD_PREP(B_AX_LTR_IDX_DRV_MASK, PCIE_LTR_IDX_IDLE) |
2827			    B_AX_LTR_REQ_DRV;
2828	} else {
2829		dec_ctrl |= B_AX_LTR_HW_DEC_EN;
2830	}
2831
2832	dec_ctrl &= ~B_AX_LTR_SPACE_IDX_V1_MASK;
2833	dec_ctrl |= FIELD_PREP(B_AX_LTR_SPACE_IDX_V1_MASK, PCI_LTR_SPC_500US);
2834
2835	if (en)
2836		rtw89_write32_set(rtwdev, R_AX_LTR_CTRL_0,
2837				  B_AX_LTR_WD_NOEMP_CHK_V1 | B_AX_LTR_HW_EN);
2838	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_0, B_AX_LTR_IDLE_TIMER_IDX_MASK,
2839			   PCI_LTR_IDLE_TIMER_3_2MS);
2840	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_1, B_AX_LTR_RX0_TH_MASK, 0x28);
2841	rtw89_write32_mask(rtwdev, R_AX_LTR_CTRL_1, B_AX_LTR_RX1_TH_MASK, 0x28);
2842	rtw89_write32(rtwdev, R_AX_LTR_DEC_CTRL, dec_ctrl);
2843	rtw89_write32(rtwdev, R_AX_LTR_LATENCY_IDX3, 0x90039003);
2844	rtw89_write32(rtwdev, R_AX_LTR_LATENCY_IDX0, 0x880b880b);
2845
2846	return 0;
2847}
2848EXPORT_SYMBOL(rtw89_pci_ltr_set_v1);
2849
2850static int rtw89_pci_ops_mac_post_init_ax(struct rtw89_dev *rtwdev)
2851{
2852	const struct rtw89_pci_info *info = rtwdev->pci_info;
2853	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
2854	int ret;
2855
2856	ret = info->ltr_set(rtwdev, true);
2857	if (ret) {
2858		rtw89_err(rtwdev, "pci ltr set fail\n");
2859		return ret;
2860	}
2861	if (chip_id == RTL8852A) {
2862		/* ltr sw trigger */
2863		rtw89_write32_set(rtwdev, R_AX_LTR_CTRL_0, B_AX_APP_LTR_ACT);
2864	}
2865	if (chip_id == RTL8852A || chip_id == RTL8852B) {
2866		/* ADDR info 8-byte mode */
2867		rtw89_write32_set(rtwdev, R_AX_TX_ADDRESS_INFO_MODE_SETTING,
2868				  B_AX_HOST_ADDR_INFO_8B_SEL);
2869		rtw89_write32_clr(rtwdev, R_AX_PKTIN_SETTING, B_AX_WD_ADDR_INFO_LENGTH);
2870	}
2871
2872	/* enable DMA for all queues */
2873	rtw89_pci_ctrl_txdma_ch_ax(rtwdev, true);
2874
2875	/* Release PCI IO */
2876	rtw89_write32_clr(rtwdev, info->dma_stop1.addr,
2877			  B_AX_STOP_WPDMA | B_AX_STOP_PCIEIO);
2878
2879	return 0;
2880}
2881
2882static int rtw89_pci_claim_device(struct rtw89_dev *rtwdev,
2883				  struct pci_dev *pdev)
2884{
2885	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2886	int ret;
2887
2888	ret = pci_enable_device(pdev);
2889	if (ret) {
2890		rtw89_err(rtwdev, "failed to enable pci device\n");
2891		return ret;
2892	}
2893
2894	pci_set_master(pdev);
2895	pci_set_drvdata(pdev, rtwdev->hw);
2896
2897	rtwpci->pdev = pdev;
2898
2899	return 0;
2900}
2901
2902static void rtw89_pci_declaim_device(struct rtw89_dev *rtwdev,
2903				     struct pci_dev *pdev)
2904{
2905	pci_disable_device(pdev);
2906}
2907
2908static int rtw89_pci_setup_mapping(struct rtw89_dev *rtwdev,
2909				   struct pci_dev *pdev)
2910{
2911	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2912	unsigned long resource_len;
2913	u8 bar_id = 2;
2914	int ret;
2915
2916	ret = pci_request_regions(pdev, KBUILD_MODNAME);
2917	if (ret) {
2918		rtw89_err(rtwdev, "failed to request pci regions\n");
2919		goto err;
2920	}
2921
2922	ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(32));
2923	if (ret) {
2924		rtw89_err(rtwdev, "failed to set dma mask to 32-bit\n");
2925		goto err_release_regions;
2926	}
2927
2928	ret = dma_set_coherent_mask(&pdev->dev, DMA_BIT_MASK(32));
2929	if (ret) {
2930		rtw89_err(rtwdev, "failed to set consistent dma mask to 32-bit\n");
2931		goto err_release_regions;
2932	}
2933
2934	resource_len = pci_resource_len(pdev, bar_id);
2935	rtwpci->mmap = pci_iomap(pdev, bar_id, resource_len);
2936	if (!rtwpci->mmap) {
2937		rtw89_err(rtwdev, "failed to map pci io\n");
2938		ret = -EIO;
2939		goto err_release_regions;
2940	}
2941
2942	return 0;
2943
2944err_release_regions:
2945	pci_release_regions(pdev);
2946err:
2947	return ret;
2948}
2949
2950static void rtw89_pci_clear_mapping(struct rtw89_dev *rtwdev,
2951				    struct pci_dev *pdev)
2952{
2953	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2954
2955	if (rtwpci->mmap) {
2956		pci_iounmap(pdev, rtwpci->mmap);
2957		pci_release_regions(pdev);
2958	}
2959}
2960
2961static void rtw89_pci_free_tx_wd_ring(struct rtw89_dev *rtwdev,
2962				      struct pci_dev *pdev,
2963				      struct rtw89_pci_tx_ring *tx_ring)
2964{
2965	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
2966	u8 *head = wd_ring->head;
2967	dma_addr_t dma = wd_ring->dma;
2968	u32 page_size = wd_ring->page_size;
2969	u32 page_num = wd_ring->page_num;
2970	u32 ring_sz = page_size * page_num;
2971
2972	dma_free_coherent(&pdev->dev, ring_sz, head, dma);
2973	wd_ring->head = NULL;
2974}
2975
2976static void rtw89_pci_free_tx_ring(struct rtw89_dev *rtwdev,
2977				   struct pci_dev *pdev,
2978				   struct rtw89_pci_tx_ring *tx_ring)
2979{
2980	int ring_sz;
2981	u8 *head;
2982	dma_addr_t dma;
2983
2984	head = tx_ring->bd_ring.head;
2985	dma = tx_ring->bd_ring.dma;
2986	ring_sz = tx_ring->bd_ring.desc_size * tx_ring->bd_ring.len;
2987	dma_free_coherent(&pdev->dev, ring_sz, head, dma);
2988
2989	tx_ring->bd_ring.head = NULL;
2990}
2991
2992static void rtw89_pci_free_tx_rings(struct rtw89_dev *rtwdev,
2993				    struct pci_dev *pdev)
2994{
2995	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
2996	const struct rtw89_pci_info *info = rtwdev->pci_info;
2997	struct rtw89_pci_tx_ring *tx_ring;
2998	int i;
2999
3000	for (i = 0; i < RTW89_TXCH_NUM; i++) {
3001		if (info->tx_dma_ch_mask & BIT(i))
3002			continue;
3003		tx_ring = &rtwpci->tx_rings[i];
3004		rtw89_pci_free_tx_wd_ring(rtwdev, pdev, tx_ring);
3005		rtw89_pci_free_tx_ring(rtwdev, pdev, tx_ring);
3006	}
3007}
3008
3009static void rtw89_pci_free_rx_ring(struct rtw89_dev *rtwdev,
3010				   struct pci_dev *pdev,
3011				   struct rtw89_pci_rx_ring *rx_ring)
3012{
3013	struct rtw89_pci_rx_info *rx_info;
3014	struct sk_buff *skb;
3015	dma_addr_t dma;
3016	u32 buf_sz;
3017	u8 *head;
3018	int ring_sz = rx_ring->bd_ring.desc_size * rx_ring->bd_ring.len;
3019	int i;
3020
3021	buf_sz = rx_ring->buf_sz;
3022	for (i = 0; i < rx_ring->bd_ring.len; i++) {
3023		skb = rx_ring->buf[i];
3024		if (!skb)
3025			continue;
3026
3027		rx_info = RTW89_PCI_RX_SKB_CB(skb);
3028		dma = rx_info->dma;
3029		dma_unmap_single(&pdev->dev, dma, buf_sz, DMA_FROM_DEVICE);
3030		dev_kfree_skb(skb);
3031		rx_ring->buf[i] = NULL;
3032	}
3033
3034	head = rx_ring->bd_ring.head;
3035	dma = rx_ring->bd_ring.dma;
3036	dma_free_coherent(&pdev->dev, ring_sz, head, dma);
3037
3038	rx_ring->bd_ring.head = NULL;
3039}
3040
3041static void rtw89_pci_free_rx_rings(struct rtw89_dev *rtwdev,
3042				    struct pci_dev *pdev)
3043{
3044	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3045	struct rtw89_pci_rx_ring *rx_ring;
3046	int i;
3047
3048	for (i = 0; i < RTW89_RXCH_NUM; i++) {
3049		rx_ring = &rtwpci->rx_rings[i];
3050		rtw89_pci_free_rx_ring(rtwdev, pdev, rx_ring);
3051	}
3052}
3053
3054static void rtw89_pci_free_trx_rings(struct rtw89_dev *rtwdev,
3055				     struct pci_dev *pdev)
3056{
3057	rtw89_pci_free_rx_rings(rtwdev, pdev);
3058	rtw89_pci_free_tx_rings(rtwdev, pdev);
3059}
3060
3061static int rtw89_pci_init_rx_bd(struct rtw89_dev *rtwdev, struct pci_dev *pdev,
3062				struct rtw89_pci_rx_ring *rx_ring,
3063				struct sk_buff *skb, int buf_sz, u32 idx)
3064{
3065	struct rtw89_pci_rx_info *rx_info;
3066	struct rtw89_pci_rx_bd_32 *rx_bd;
3067	dma_addr_t dma;
3068
3069	if (!skb)
3070		return -EINVAL;
3071
3072	dma = dma_map_single(&pdev->dev, skb->data, buf_sz, DMA_FROM_DEVICE);
3073	if (dma_mapping_error(&pdev->dev, dma))
3074		return -EBUSY;
3075
3076	rx_info = RTW89_PCI_RX_SKB_CB(skb);
3077	rx_bd = RTW89_PCI_RX_BD(rx_ring, idx);
3078
3079	memset(rx_bd, 0, sizeof(*rx_bd));
3080	rx_bd->buf_size = cpu_to_le16(buf_sz);
3081	rx_bd->dma = cpu_to_le32(dma);
3082	rx_info->dma = dma;
3083
3084	return 0;
3085}
3086
3087static int rtw89_pci_alloc_tx_wd_ring(struct rtw89_dev *rtwdev,
3088				      struct pci_dev *pdev,
3089				      struct rtw89_pci_tx_ring *tx_ring,
3090				      enum rtw89_tx_channel txch)
3091{
3092	struct rtw89_pci_tx_wd_ring *wd_ring = &tx_ring->wd_ring;
3093	struct rtw89_pci_tx_wd *txwd;
3094	dma_addr_t dma;
3095	dma_addr_t cur_paddr;
3096	u8 *head;
3097	u8 *cur_vaddr;
3098	u32 page_size = RTW89_PCI_TXWD_PAGE_SIZE;
3099	u32 page_num = RTW89_PCI_TXWD_NUM_MAX;
3100	u32 ring_sz = page_size * page_num;
3101	u32 page_offset;
3102	int i;
3103
3104	/* FWCMD queue doesn't use txwd as pages */
3105	if (txch == RTW89_TXCH_CH12)
3106		return 0;
3107
3108	head = dma_alloc_coherent(&pdev->dev, ring_sz, &dma, GFP_KERNEL);
3109	if (!head)
3110		return -ENOMEM;
3111
3112	INIT_LIST_HEAD(&wd_ring->free_pages);
3113	wd_ring->head = head;
3114	wd_ring->dma = dma;
3115	wd_ring->page_size = page_size;
3116	wd_ring->page_num = page_num;
3117
3118	page_offset = 0;
3119	for (i = 0; i < page_num; i++) {
3120		txwd = &wd_ring->pages[i];
3121		cur_paddr = dma + page_offset;
3122		cur_vaddr = head + page_offset;
3123
3124		skb_queue_head_init(&txwd->queue);
3125		INIT_LIST_HEAD(&txwd->list);
3126		txwd->paddr = cur_paddr;
3127		txwd->vaddr = cur_vaddr;
3128		txwd->len = page_size;
3129		txwd->seq = i;
3130		rtw89_pci_enqueue_txwd(tx_ring, txwd);
3131
3132		page_offset += page_size;
3133	}
3134
3135	return 0;
3136}
3137
3138static int rtw89_pci_alloc_tx_ring(struct rtw89_dev *rtwdev,
3139				   struct pci_dev *pdev,
3140				   struct rtw89_pci_tx_ring *tx_ring,
3141				   u32 desc_size, u32 len,
3142				   enum rtw89_tx_channel txch)
3143{
3144	const struct rtw89_pci_ch_dma_addr *txch_addr;
3145	int ring_sz = desc_size * len;
3146	u8 *head;
3147	dma_addr_t dma;
3148	int ret;
3149
3150	ret = rtw89_pci_alloc_tx_wd_ring(rtwdev, pdev, tx_ring, txch);
3151	if (ret) {
3152		rtw89_err(rtwdev, "failed to alloc txwd ring of txch %d\n", txch);
3153		goto err;
3154	}
3155
3156	ret = rtw89_pci_get_txch_addrs(rtwdev, txch, &txch_addr);
3157	if (ret) {
3158		rtw89_err(rtwdev, "failed to get address of txch %d", txch);
3159		goto err_free_wd_ring;
3160	}
3161
3162	head = dma_alloc_coherent(&pdev->dev, ring_sz, &dma, GFP_KERNEL);
3163	if (!head) {
3164		ret = -ENOMEM;
3165		goto err_free_wd_ring;
3166	}
3167
3168	INIT_LIST_HEAD(&tx_ring->busy_pages);
3169	tx_ring->bd_ring.head = head;
3170	tx_ring->bd_ring.dma = dma;
3171	tx_ring->bd_ring.len = len;
3172	tx_ring->bd_ring.desc_size = desc_size;
3173	tx_ring->bd_ring.addr = *txch_addr;
3174	tx_ring->bd_ring.wp = 0;
3175	tx_ring->bd_ring.rp = 0;
3176	tx_ring->txch = txch;
3177
3178	return 0;
3179
3180err_free_wd_ring:
3181	rtw89_pci_free_tx_wd_ring(rtwdev, pdev, tx_ring);
3182err:
3183	return ret;
3184}
3185
3186static int rtw89_pci_alloc_tx_rings(struct rtw89_dev *rtwdev,
3187				    struct pci_dev *pdev)
3188{
3189	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3190	const struct rtw89_pci_info *info = rtwdev->pci_info;
3191	struct rtw89_pci_tx_ring *tx_ring;
3192	u32 desc_size;
3193	u32 len;
3194	u32 i, tx_allocated;
3195	int ret;
3196
3197	for (i = 0; i < RTW89_TXCH_NUM; i++) {
3198		if (info->tx_dma_ch_mask & BIT(i))
3199			continue;
3200		tx_ring = &rtwpci->tx_rings[i];
3201		desc_size = sizeof(struct rtw89_pci_tx_bd_32);
3202		len = RTW89_PCI_TXBD_NUM_MAX;
3203		ret = rtw89_pci_alloc_tx_ring(rtwdev, pdev, tx_ring,
3204					      desc_size, len, i);
3205		if (ret) {
3206			rtw89_err(rtwdev, "failed to alloc tx ring %d\n", i);
3207			goto err_free;
3208		}
3209	}
3210
3211	return 0;
3212
3213err_free:
3214	tx_allocated = i;
3215	for (i = 0; i < tx_allocated; i++) {
3216		tx_ring = &rtwpci->tx_rings[i];
3217		rtw89_pci_free_tx_ring(rtwdev, pdev, tx_ring);
3218	}
3219
3220	return ret;
3221}
3222
3223static int rtw89_pci_alloc_rx_ring(struct rtw89_dev *rtwdev,
3224				   struct pci_dev *pdev,
3225				   struct rtw89_pci_rx_ring *rx_ring,
3226				   u32 desc_size, u32 len, u32 rxch)
3227{
3228	const struct rtw89_pci_info *info = rtwdev->pci_info;
3229	const struct rtw89_pci_ch_dma_addr *rxch_addr;
3230	struct sk_buff *skb;
3231	u8 *head;
3232	dma_addr_t dma;
3233	int ring_sz = desc_size * len;
3234	int buf_sz = RTW89_PCI_RX_BUF_SIZE;
3235	int i, allocated;
3236	int ret;
3237
3238	ret = rtw89_pci_get_rxch_addrs(rtwdev, rxch, &rxch_addr);
3239	if (ret) {
3240		rtw89_err(rtwdev, "failed to get address of rxch %d", rxch);
3241		return ret;
3242	}
3243
3244	head = dma_alloc_coherent(&pdev->dev, ring_sz, &dma, GFP_KERNEL);
3245	if (!head) {
3246		ret = -ENOMEM;
3247		goto err;
3248	}
3249
3250	rx_ring->bd_ring.head = head;
3251	rx_ring->bd_ring.dma = dma;
3252	rx_ring->bd_ring.len = len;
3253	rx_ring->bd_ring.desc_size = desc_size;
3254	rx_ring->bd_ring.addr = *rxch_addr;
3255	if (info->rx_ring_eq_is_full)
3256		rx_ring->bd_ring.wp = len - 1;
3257	else
3258		rx_ring->bd_ring.wp = 0;
3259	rx_ring->bd_ring.rp = 0;
3260	rx_ring->buf_sz = buf_sz;
3261	rx_ring->diliver_skb = NULL;
3262	rx_ring->diliver_desc.ready = false;
3263	rx_ring->target_rx_tag = 0;
3264
3265	for (i = 0; i < len; i++) {
3266		skb = dev_alloc_skb(buf_sz);
3267		if (!skb) {
3268			ret = -ENOMEM;
3269			goto err_free;
3270		}
3271
3272		memset(skb->data, 0, buf_sz);
3273		rx_ring->buf[i] = skb;
3274		ret = rtw89_pci_init_rx_bd(rtwdev, pdev, rx_ring, skb,
3275					   buf_sz, i);
3276		if (ret) {
3277			rtw89_err(rtwdev, "failed to init rx buf %d\n", i);
3278			dev_kfree_skb_any(skb);
3279			rx_ring->buf[i] = NULL;
3280			goto err_free;
3281		}
3282	}
3283
3284	return 0;
3285
3286err_free:
3287	allocated = i;
3288	for (i = 0; i < allocated; i++) {
3289		skb = rx_ring->buf[i];
3290		if (!skb)
3291			continue;
3292		dma = *((dma_addr_t *)skb->cb);
3293		dma_unmap_single(&pdev->dev, dma, buf_sz, DMA_FROM_DEVICE);
3294		dev_kfree_skb(skb);
3295		rx_ring->buf[i] = NULL;
3296	}
3297
3298	head = rx_ring->bd_ring.head;
3299	dma = rx_ring->bd_ring.dma;
3300	dma_free_coherent(&pdev->dev, ring_sz, head, dma);
3301
3302	rx_ring->bd_ring.head = NULL;
3303err:
3304	return ret;
3305}
3306
3307static int rtw89_pci_alloc_rx_rings(struct rtw89_dev *rtwdev,
3308				    struct pci_dev *pdev)
3309{
3310	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3311	struct rtw89_pci_rx_ring *rx_ring;
3312	u32 desc_size;
3313	u32 len;
3314	int i, rx_allocated;
3315	int ret;
3316
3317	for (i = 0; i < RTW89_RXCH_NUM; i++) {
3318		rx_ring = &rtwpci->rx_rings[i];
3319		desc_size = sizeof(struct rtw89_pci_rx_bd_32);
3320		len = RTW89_PCI_RXBD_NUM_MAX;
3321		ret = rtw89_pci_alloc_rx_ring(rtwdev, pdev, rx_ring,
3322					      desc_size, len, i);
3323		if (ret) {
3324			rtw89_err(rtwdev, "failed to alloc rx ring %d\n", i);
3325			goto err_free;
3326		}
3327	}
3328
3329	return 0;
3330
3331err_free:
3332	rx_allocated = i;
3333	for (i = 0; i < rx_allocated; i++) {
3334		rx_ring = &rtwpci->rx_rings[i];
3335		rtw89_pci_free_rx_ring(rtwdev, pdev, rx_ring);
3336	}
3337
3338	return ret;
3339}
3340
3341static int rtw89_pci_alloc_trx_rings(struct rtw89_dev *rtwdev,
3342				     struct pci_dev *pdev)
3343{
3344	int ret;
3345
3346	ret = rtw89_pci_alloc_tx_rings(rtwdev, pdev);
3347	if (ret) {
3348		rtw89_err(rtwdev, "failed to alloc dma tx rings\n");
3349		goto err;
3350	}
3351
3352	ret = rtw89_pci_alloc_rx_rings(rtwdev, pdev);
3353	if (ret) {
3354		rtw89_err(rtwdev, "failed to alloc dma rx rings\n");
3355		goto err_free_tx_rings;
3356	}
3357
3358	return 0;
3359
3360err_free_tx_rings:
3361	rtw89_pci_free_tx_rings(rtwdev, pdev);
3362err:
3363	return ret;
3364}
3365
3366static void rtw89_pci_h2c_init(struct rtw89_dev *rtwdev,
3367			       struct rtw89_pci *rtwpci)
3368{
3369	skb_queue_head_init(&rtwpci->h2c_queue);
3370	skb_queue_head_init(&rtwpci->h2c_release_queue);
3371}
3372
3373static int rtw89_pci_setup_resource(struct rtw89_dev *rtwdev,
3374				    struct pci_dev *pdev)
3375{
3376	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3377	int ret;
3378
3379	ret = rtw89_pci_setup_mapping(rtwdev, pdev);
3380	if (ret) {
3381		rtw89_err(rtwdev, "failed to setup pci mapping\n");
3382		goto err;
3383	}
3384
3385	ret = rtw89_pci_alloc_trx_rings(rtwdev, pdev);
3386	if (ret) {
3387		rtw89_err(rtwdev, "failed to alloc pci trx rings\n");
3388		goto err_pci_unmap;
3389	}
3390
3391	rtw89_pci_h2c_init(rtwdev, rtwpci);
3392
3393	spin_lock_init(&rtwpci->irq_lock);
3394	spin_lock_init(&rtwpci->trx_lock);
3395
3396	return 0;
3397
3398err_pci_unmap:
3399	rtw89_pci_clear_mapping(rtwdev, pdev);
3400err:
3401	return ret;
3402}
3403
3404static void rtw89_pci_clear_resource(struct rtw89_dev *rtwdev,
3405				     struct pci_dev *pdev)
3406{
3407	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3408
3409	rtw89_pci_free_trx_rings(rtwdev, pdev);
3410	rtw89_pci_clear_mapping(rtwdev, pdev);
3411	rtw89_pci_release_fwcmd(rtwdev, rtwpci,
3412				skb_queue_len(&rtwpci->h2c_queue), true);
3413}
3414
3415void rtw89_pci_config_intr_mask(struct rtw89_dev *rtwdev)
3416{
3417	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3418	const struct rtw89_chip_info *chip = rtwdev->chip;
3419	u32 hs0isr_ind_int_en = B_AX_HS0ISR_IND_INT_EN;
3420
3421	if (chip->chip_id == RTL8851B)
3422		hs0isr_ind_int_en = B_AX_HS0ISR_IND_INT_EN_WKARND;
3423
3424	rtwpci->halt_c2h_intrs = B_AX_HALT_C2H_INT_EN | 0;
3425
3426	if (rtwpci->under_recovery) {
3427		rtwpci->intrs[0] = hs0isr_ind_int_en;
3428		rtwpci->intrs[1] = 0;
3429	} else {
3430		rtwpci->intrs[0] = B_AX_TXDMA_STUCK_INT_EN |
3431				   B_AX_RXDMA_INT_EN |
3432				   B_AX_RXP1DMA_INT_EN |
3433				   B_AX_RPQDMA_INT_EN |
3434				   B_AX_RXDMA_STUCK_INT_EN |
3435				   B_AX_RDU_INT_EN |
3436				   B_AX_RPQBD_FULL_INT_EN |
3437				   hs0isr_ind_int_en;
3438
3439		rtwpci->intrs[1] = B_AX_HC10ISR_IND_INT_EN;
3440	}
3441}
3442EXPORT_SYMBOL(rtw89_pci_config_intr_mask);
3443
3444static void rtw89_pci_recovery_intr_mask_v1(struct rtw89_dev *rtwdev)
3445{
3446	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3447
3448	rtwpci->ind_intrs = B_AX_HS0ISR_IND_INT_EN;
3449	rtwpci->halt_c2h_intrs = B_AX_HALT_C2H_INT_EN | B_AX_WDT_TIMEOUT_INT_EN;
3450	rtwpci->intrs[0] = 0;
3451	rtwpci->intrs[1] = 0;
3452}
3453
3454static void rtw89_pci_default_intr_mask_v1(struct rtw89_dev *rtwdev)
3455{
3456	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3457
3458	rtwpci->ind_intrs = B_AX_HCI_AXIDMA_INT_EN |
3459			    B_AX_HS1ISR_IND_INT_EN |
3460			    B_AX_HS0ISR_IND_INT_EN;
3461	rtwpci->halt_c2h_intrs = B_AX_HALT_C2H_INT_EN | B_AX_WDT_TIMEOUT_INT_EN;
3462	rtwpci->intrs[0] = B_AX_TXDMA_STUCK_INT_EN |
3463			   B_AX_RXDMA_INT_EN |
3464			   B_AX_RXP1DMA_INT_EN |
3465			   B_AX_RPQDMA_INT_EN |
3466			   B_AX_RXDMA_STUCK_INT_EN |
3467			   B_AX_RDU_INT_EN |
3468			   B_AX_RPQBD_FULL_INT_EN;
3469	rtwpci->intrs[1] = B_AX_GPIO18_INT_EN;
3470}
3471
3472static void rtw89_pci_low_power_intr_mask_v1(struct rtw89_dev *rtwdev)
3473{
3474	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3475
3476	rtwpci->ind_intrs = B_AX_HS1ISR_IND_INT_EN |
3477			    B_AX_HS0ISR_IND_INT_EN;
3478	rtwpci->halt_c2h_intrs = B_AX_HALT_C2H_INT_EN | B_AX_WDT_TIMEOUT_INT_EN;
3479	rtwpci->intrs[0] = 0;
3480	rtwpci->intrs[1] = B_AX_GPIO18_INT_EN;
3481}
3482
3483void rtw89_pci_config_intr_mask_v1(struct rtw89_dev *rtwdev)
3484{
3485	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3486
3487	if (rtwpci->under_recovery)
3488		rtw89_pci_recovery_intr_mask_v1(rtwdev);
3489	else if (rtwpci->low_power)
3490		rtw89_pci_low_power_intr_mask_v1(rtwdev);
3491	else
3492		rtw89_pci_default_intr_mask_v1(rtwdev);
3493}
3494EXPORT_SYMBOL(rtw89_pci_config_intr_mask_v1);
3495
3496static void rtw89_pci_recovery_intr_mask_v2(struct rtw89_dev *rtwdev)
3497{
3498	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3499
3500	rtwpci->ind_intrs = B_BE_HS0_IND_INT_EN0;
3501	rtwpci->halt_c2h_intrs = B_BE_HALT_C2H_INT_EN | B_BE_WDT_TIMEOUT_INT_EN;
3502	rtwpci->intrs[0] = 0;
3503	rtwpci->intrs[1] = 0;
3504}
3505
3506static void rtw89_pci_default_intr_mask_v2(struct rtw89_dev *rtwdev)
3507{
3508	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3509
3510	rtwpci->ind_intrs = B_BE_HCI_AXIDMA_INT_EN0 |
3511			    B_BE_HS0_IND_INT_EN0;
3512	rtwpci->halt_c2h_intrs = B_BE_HALT_C2H_INT_EN | B_BE_WDT_TIMEOUT_INT_EN;
3513	rtwpci->intrs[0] = B_BE_RDU_CH1_INT_IMR_V1 |
3514			   B_BE_RDU_CH0_INT_IMR_V1;
3515	rtwpci->intrs[1] = B_BE_PCIE_RX_RX0P2_IMR0_V1 |
3516			   B_BE_PCIE_RX_RPQ0_IMR0_V1;
3517}
3518
3519static void rtw89_pci_low_power_intr_mask_v2(struct rtw89_dev *rtwdev)
3520{
3521	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3522
3523	rtwpci->ind_intrs = B_BE_HS0_IND_INT_EN0 |
3524			    B_BE_HS1_IND_INT_EN0;
3525	rtwpci->halt_c2h_intrs = B_BE_HALT_C2H_INT_EN | B_BE_WDT_TIMEOUT_INT_EN;
3526	rtwpci->intrs[0] = 0;
3527	rtwpci->intrs[1] = B_BE_PCIE_RX_RX0P2_IMR0_V1 |
3528			   B_BE_PCIE_RX_RPQ0_IMR0_V1;
3529}
3530
3531void rtw89_pci_config_intr_mask_v2(struct rtw89_dev *rtwdev)
3532{
3533	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3534
3535	if (rtwpci->under_recovery)
3536		rtw89_pci_recovery_intr_mask_v2(rtwdev);
3537	else if (rtwpci->low_power)
3538		rtw89_pci_low_power_intr_mask_v2(rtwdev);
3539	else
3540		rtw89_pci_default_intr_mask_v2(rtwdev);
3541}
3542EXPORT_SYMBOL(rtw89_pci_config_intr_mask_v2);
3543
3544static int rtw89_pci_request_irq(struct rtw89_dev *rtwdev,
3545				 struct pci_dev *pdev)
3546{
3547	unsigned long flags = 0;
3548	int ret;
3549
3550	flags |= PCI_IRQ_LEGACY | PCI_IRQ_MSI;
3551	ret = pci_alloc_irq_vectors(pdev, 1, 1, flags);
3552	if (ret < 0) {
3553		rtw89_err(rtwdev, "failed to alloc irq vectors, ret %d\n", ret);
3554		goto err;
3555	}
3556
3557	ret = devm_request_threaded_irq(rtwdev->dev, pdev->irq,
3558					rtw89_pci_interrupt_handler,
3559					rtw89_pci_interrupt_threadfn,
3560					IRQF_SHARED, KBUILD_MODNAME, rtwdev);
3561	if (ret) {
3562		rtw89_err(rtwdev, "failed to request threaded irq\n");
3563		goto err_free_vector;
3564	}
3565
3566	rtw89_chip_config_intr_mask(rtwdev, RTW89_PCI_INTR_MASK_RESET);
3567
3568	return 0;
3569
3570err_free_vector:
3571	pci_free_irq_vectors(pdev);
3572err:
3573	return ret;
3574}
3575
3576static void rtw89_pci_free_irq(struct rtw89_dev *rtwdev,
3577			       struct pci_dev *pdev)
3578{
3579	devm_free_irq(rtwdev->dev, pdev->irq, rtwdev);
3580	pci_free_irq_vectors(pdev);
3581}
3582
3583static u16 gray_code_to_bin(u16 gray_code, u32 bit_num)
3584{
3585	u16 bin = 0, gray_bit;
3586	u32 bit_idx;
3587
3588	for (bit_idx = 0; bit_idx < bit_num; bit_idx++) {
3589		gray_bit = (gray_code >> bit_idx) & 0x1;
3590		if (bit_num - bit_idx > 1)
3591			gray_bit ^= (gray_code >> (bit_idx + 1)) & 0x1;
3592		bin |= (gray_bit << bit_idx);
3593	}
3594
3595	return bin;
3596}
3597
3598static int rtw89_pci_filter_out(struct rtw89_dev *rtwdev)
3599{
3600	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3601	struct pci_dev *pdev = rtwpci->pdev;
3602	u16 val16, filter_out_val;
3603	u32 val, phy_offset;
3604	int ret;
3605
3606	if (rtwdev->chip->chip_id != RTL8852C)
3607		return 0;
3608
3609	val = rtw89_read32_mask(rtwdev, R_AX_PCIE_MIX_CFG_V1, B_AX_ASPM_CTRL_MASK);
3610	if (val == B_AX_ASPM_CTRL_L1)
3611		return 0;
3612
3613	ret = pci_read_config_dword(pdev, RTW89_PCIE_L1_STS_V1, &val);
3614	if (ret)
3615		return ret;
3616
3617	val = FIELD_GET(RTW89_BCFG_LINK_SPEED_MASK, val);
3618	if (val == RTW89_PCIE_GEN1_SPEED) {
3619		phy_offset = R_RAC_DIRECT_OFFSET_G1;
3620	} else if (val == RTW89_PCIE_GEN2_SPEED) {
3621		phy_offset = R_RAC_DIRECT_OFFSET_G2;
3622		val16 = rtw89_read16(rtwdev, phy_offset + RAC_ANA10 * RAC_MULT);
3623		rtw89_write16_set(rtwdev, phy_offset + RAC_ANA10 * RAC_MULT,
3624				  val16 | B_PCIE_BIT_PINOUT_DIS);
3625		rtw89_write16_set(rtwdev, phy_offset + RAC_ANA19 * RAC_MULT,
3626				  val16 & ~B_PCIE_BIT_RD_SEL);
3627
3628		val16 = rtw89_read16_mask(rtwdev,
3629					  phy_offset + RAC_ANA1F * RAC_MULT,
3630					  FILTER_OUT_EQ_MASK);
3631		val16 = gray_code_to_bin(val16, hweight16(val16));
3632		filter_out_val = rtw89_read16(rtwdev, phy_offset + RAC_ANA24 *
3633					      RAC_MULT);
3634		filter_out_val &= ~REG_FILTER_OUT_MASK;
3635		filter_out_val |= FIELD_PREP(REG_FILTER_OUT_MASK, val16);
3636
3637		rtw89_write16(rtwdev, phy_offset + RAC_ANA24 * RAC_MULT,
3638			      filter_out_val);
3639		rtw89_write16_set(rtwdev, phy_offset + RAC_ANA0A * RAC_MULT,
3640				  B_BAC_EQ_SEL);
3641		rtw89_write16_set(rtwdev,
3642				  R_RAC_DIRECT_OFFSET_G1 + RAC_ANA0C * RAC_MULT,
3643				  B_PCIE_BIT_PSAVE);
3644	} else {
3645		return -EOPNOTSUPP;
3646	}
3647	rtw89_write16_set(rtwdev, phy_offset + RAC_ANA0C * RAC_MULT,
3648			  B_PCIE_BIT_PSAVE);
3649
3650	return 0;
3651}
3652
3653static void rtw89_pci_clkreq_set(struct rtw89_dev *rtwdev, bool enable)
3654{
3655	const struct rtw89_pci_info *info = rtwdev->pci_info;
3656	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
3657
3658	if (rtw89_pci_disable_clkreq)
3659		return;
3660
3661	gen_def->clkreq_set(rtwdev, enable);
3662}
3663
3664static void rtw89_pci_clkreq_set_ax(struct rtw89_dev *rtwdev, bool enable)
3665{
3666	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
3667	int ret;
3668
3669	ret = rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_CLK_CTRL,
3670					  PCIE_CLKDLY_HW_30US);
3671	if (ret)
3672		rtw89_err(rtwdev, "failed to set CLKREQ Delay\n");
3673
3674	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
3675		if (enable)
3676			ret = rtw89_pci_config_byte_set(rtwdev,
3677							RTW89_PCIE_L1_CTRL,
3678							RTW89_PCIE_BIT_CLK);
3679		else
3680			ret = rtw89_pci_config_byte_clr(rtwdev,
3681							RTW89_PCIE_L1_CTRL,
3682							RTW89_PCIE_BIT_CLK);
3683		if (ret)
3684			rtw89_err(rtwdev, "failed to %s CLKREQ_L1, ret=%d",
3685				  enable ? "set" : "unset", ret);
3686	} else if (chip_id == RTL8852C) {
3687		rtw89_write32_set(rtwdev, R_AX_PCIE_LAT_CTRL,
3688				  B_AX_CLK_REQ_SEL_OPT | B_AX_CLK_REQ_SEL);
3689		if (enable)
3690			rtw89_write32_set(rtwdev, R_AX_L1_CLK_CTRL,
3691					  B_AX_CLK_REQ_N);
3692		else
3693			rtw89_write32_clr(rtwdev, R_AX_L1_CLK_CTRL,
3694					  B_AX_CLK_REQ_N);
3695	}
3696}
3697
3698static void rtw89_pci_aspm_set(struct rtw89_dev *rtwdev, bool enable)
3699{
3700	const struct rtw89_pci_info *info = rtwdev->pci_info;
3701	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
3702
3703	if (rtw89_pci_disable_aspm_l1)
3704		return;
3705
3706	gen_def->aspm_set(rtwdev, enable);
3707}
3708
3709static void rtw89_pci_aspm_set_ax(struct rtw89_dev *rtwdev, bool enable)
3710{
3711	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
3712	u8 value = 0;
3713	int ret;
3714
3715	ret = rtw89_pci_read_config_byte(rtwdev, RTW89_PCIE_ASPM_CTRL, &value);
3716	if (ret)
3717		rtw89_warn(rtwdev, "failed to read ASPM Delay\n");
3718
3719	u8p_replace_bits(&value, PCIE_L1DLY_16US, RTW89_L1DLY_MASK);
3720	u8p_replace_bits(&value, PCIE_L0SDLY_4US, RTW89_L0DLY_MASK);
3721
3722	ret = rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_ASPM_CTRL, value);
3723	if (ret)
3724		rtw89_warn(rtwdev, "failed to read ASPM Delay\n");
3725
3726	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
3727		if (enable)
3728			ret = rtw89_pci_config_byte_set(rtwdev,
3729							RTW89_PCIE_L1_CTRL,
3730							RTW89_PCIE_BIT_L1);
3731		else
3732			ret = rtw89_pci_config_byte_clr(rtwdev,
3733							RTW89_PCIE_L1_CTRL,
3734							RTW89_PCIE_BIT_L1);
3735	} else if (chip_id == RTL8852C) {
3736		if (enable)
3737			rtw89_write32_set(rtwdev, R_AX_PCIE_MIX_CFG_V1,
3738					  B_AX_ASPM_CTRL_L1);
3739		else
3740			rtw89_write32_clr(rtwdev, R_AX_PCIE_MIX_CFG_V1,
3741					  B_AX_ASPM_CTRL_L1);
3742	}
3743	if (ret)
3744		rtw89_err(rtwdev, "failed to %s ASPM L1, ret=%d",
3745			  enable ? "set" : "unset", ret);
3746}
3747
3748static void rtw89_pci_recalc_int_mit(struct rtw89_dev *rtwdev)
3749{
3750	enum rtw89_chip_gen chip_gen = rtwdev->chip->chip_gen;
3751	const struct rtw89_pci_info *info = rtwdev->pci_info;
3752	struct rtw89_traffic_stats *stats = &rtwdev->stats;
3753	enum rtw89_tfc_lv tx_tfc_lv = stats->tx_tfc_lv;
3754	enum rtw89_tfc_lv rx_tfc_lv = stats->rx_tfc_lv;
3755	u32 val = 0;
3756
3757	if (rtwdev->scanning ||
3758	    (tx_tfc_lv < RTW89_TFC_HIGH && rx_tfc_lv < RTW89_TFC_HIGH))
3759		goto out;
3760
3761	if (chip_gen == RTW89_CHIP_BE)
3762		val = B_BE_PCIE_MIT_RX0P2_EN | B_BE_PCIE_MIT_RX0P1_EN;
3763	else
3764		val = B_AX_RXMIT_RXP2_SEL | B_AX_RXMIT_RXP1_SEL |
3765		      FIELD_PREP(B_AX_RXCOUNTER_MATCH_MASK, RTW89_PCI_RXBD_NUM_MAX / 2) |
3766		      FIELD_PREP(B_AX_RXTIMER_UNIT_MASK, AX_RXTIMER_UNIT_64US) |
3767		      FIELD_PREP(B_AX_RXTIMER_MATCH_MASK, 2048 / 64);
3768
3769out:
3770	rtw89_write32(rtwdev, info->mit_addr, val);
3771}
3772
3773static void rtw89_pci_link_cfg(struct rtw89_dev *rtwdev)
3774{
3775	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3776	struct pci_dev *pdev = rtwpci->pdev;
3777	u16 link_ctrl;
3778	int ret;
3779
3780	/* Though there is standard PCIE configuration space to set the
3781	 * link control register, but by Realtek's design, driver should
3782	 * check if host supports CLKREQ/ASPM to enable the HW module.
3783	 *
3784	 * These functions are implemented by two HW modules associated,
3785	 * one is responsible to access PCIE configuration space to
3786	 * follow the host settings, and another is in charge of doing
3787	 * CLKREQ/ASPM mechanisms, it is default disabled. Because sometimes
3788	 * the host does not support it, and due to some reasons or wrong
3789	 * settings (ex. CLKREQ# not Bi-Direction), it could lead to device
3790	 * loss if HW misbehaves on the link.
3791	 *
3792	 * Hence it's designed that driver should first check the PCIE
3793	 * configuration space is sync'ed and enabled, then driver can turn
3794	 * on the other module that is actually working on the mechanism.
3795	 */
3796	ret = pcie_capability_read_word(pdev, PCI_EXP_LNKCTL, &link_ctrl);
3797	if (ret) {
3798		rtw89_err(rtwdev, "failed to read PCI cap, ret=%d\n", ret);
3799		return;
3800	}
3801
3802	if (link_ctrl & PCI_EXP_LNKCTL_CLKREQ_EN)
3803		rtw89_pci_clkreq_set(rtwdev, true);
3804
3805	if (link_ctrl & PCI_EXP_LNKCTL_ASPM_L1)
3806		rtw89_pci_aspm_set(rtwdev, true);
3807}
3808
3809static void rtw89_pci_l1ss_set(struct rtw89_dev *rtwdev, bool enable)
3810{
3811	const struct rtw89_pci_info *info = rtwdev->pci_info;
3812	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
3813
3814	if (rtw89_pci_disable_l1ss)
3815		return;
3816
3817	gen_def->l1ss_set(rtwdev, enable);
3818}
3819
3820static void rtw89_pci_l1ss_set_ax(struct rtw89_dev *rtwdev, bool enable)
3821{
3822	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
3823	int ret;
3824
3825	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
3826		if (enable)
3827			ret = rtw89_pci_config_byte_set(rtwdev,
3828							RTW89_PCIE_TIMER_CTRL,
3829							RTW89_PCIE_BIT_L1SUB);
3830		else
3831			ret = rtw89_pci_config_byte_clr(rtwdev,
3832							RTW89_PCIE_TIMER_CTRL,
3833							RTW89_PCIE_BIT_L1SUB);
3834		if (ret)
3835			rtw89_err(rtwdev, "failed to %s L1SS, ret=%d",
3836				  enable ? "set" : "unset", ret);
3837	} else if (chip_id == RTL8852C) {
3838		ret = rtw89_pci_config_byte_clr(rtwdev, RTW89_PCIE_L1SS_STS_V1,
3839						RTW89_PCIE_BIT_ASPM_L11 |
3840						RTW89_PCIE_BIT_PCI_L11);
3841		if (ret)
3842			rtw89_warn(rtwdev, "failed to unset ASPM L1.1, ret=%d", ret);
3843		if (enable)
3844			rtw89_write32_clr(rtwdev, R_AX_PCIE_MIX_CFG_V1,
3845					  B_AX_L1SUB_DISABLE);
3846		else
3847			rtw89_write32_set(rtwdev, R_AX_PCIE_MIX_CFG_V1,
3848					  B_AX_L1SUB_DISABLE);
3849	}
3850}
3851
3852static void rtw89_pci_l1ss_cfg(struct rtw89_dev *rtwdev)
3853{
3854	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3855	struct pci_dev *pdev = rtwpci->pdev;
3856	u32 l1ss_cap_ptr, l1ss_ctrl;
3857
3858	if (rtw89_pci_disable_l1ss)
3859		return;
3860
3861	l1ss_cap_ptr = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_L1SS);
3862	if (!l1ss_cap_ptr)
3863		return;
3864
3865	pci_read_config_dword(pdev, l1ss_cap_ptr + PCI_L1SS_CTL1, &l1ss_ctrl);
3866
3867	if (l1ss_ctrl & PCI_L1SS_CTL1_L1SS_MASK)
3868		rtw89_pci_l1ss_set(rtwdev, true);
3869}
3870
3871static int rtw89_pci_poll_io_idle_ax(struct rtw89_dev *rtwdev)
3872{
3873	int ret = 0;
3874	u32 sts;
3875	u32 busy = B_AX_PCIEIO_BUSY | B_AX_PCIEIO_TX_BUSY | B_AX_PCIEIO_RX_BUSY;
3876
3877	ret = read_poll_timeout_atomic(rtw89_read32, sts, (sts & busy) == 0x0,
3878				       10, 1000, false, rtwdev,
3879				       R_AX_PCIE_DMA_BUSY1);
3880	if (ret) {
3881		rtw89_err(rtwdev, "pci dmach busy1 0x%X\n",
3882			  rtw89_read32(rtwdev, R_AX_PCIE_DMA_BUSY1));
3883		return -EINVAL;
3884	}
3885	return ret;
3886}
3887
3888static int rtw89_pci_lv1rst_stop_dma_ax(struct rtw89_dev *rtwdev)
3889{
3890	u32 val;
3891	int ret;
3892
3893	if (rtwdev->chip->chip_id == RTL8852C)
3894		return 0;
3895
3896	rtw89_pci_ctrl_dma_all(rtwdev, false);
3897	ret = rtw89_pci_poll_io_idle_ax(rtwdev);
3898	if (ret) {
3899		val = rtw89_read32(rtwdev, R_AX_DBG_ERR_FLAG);
3900		rtw89_debug(rtwdev, RTW89_DBG_HCI,
3901			    "[PCIe] poll_io_idle fail, before 0x%08x: 0x%08x\n",
3902			    R_AX_DBG_ERR_FLAG, val);
3903		if (val & B_AX_TX_STUCK || val & B_AX_PCIE_TXBD_LEN0)
3904			rtw89_mac_ctrl_hci_dma_tx(rtwdev, false);
3905		if (val & B_AX_RX_STUCK)
3906			rtw89_mac_ctrl_hci_dma_rx(rtwdev, false);
3907		rtw89_mac_ctrl_hci_dma_trx(rtwdev, true);
3908		ret = rtw89_pci_poll_io_idle_ax(rtwdev);
3909		val = rtw89_read32(rtwdev, R_AX_DBG_ERR_FLAG);
3910		rtw89_debug(rtwdev, RTW89_DBG_HCI,
3911			    "[PCIe] poll_io_idle fail, after 0x%08x: 0x%08x\n",
3912			    R_AX_DBG_ERR_FLAG, val);
3913	}
3914
3915	return ret;
3916}
3917
3918static int rtw89_pci_lv1rst_start_dma_ax(struct rtw89_dev *rtwdev)
3919{
3920	u32 ret;
3921
3922	if (rtwdev->chip->chip_id == RTL8852C)
3923		return 0;
3924
3925	rtw89_mac_ctrl_hci_dma_trx(rtwdev, false);
3926	rtw89_mac_ctrl_hci_dma_trx(rtwdev, true);
3927	rtw89_pci_clr_idx_all(rtwdev);
3928
3929	ret = rtw89_pci_rst_bdram_ax(rtwdev);
3930	if (ret)
3931		return ret;
3932
3933	rtw89_pci_ctrl_dma_all(rtwdev, true);
3934	return ret;
3935}
3936
3937static int rtw89_pci_ops_mac_lv1_recovery(struct rtw89_dev *rtwdev,
3938					  enum rtw89_lv1_rcvy_step step)
3939{
3940	const struct rtw89_pci_info *info = rtwdev->pci_info;
3941	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
3942	int ret;
3943
3944	switch (step) {
3945	case RTW89_LV1_RCVY_STEP_1:
3946		ret = gen_def->lv1rst_stop_dma(rtwdev);
3947		if (ret)
3948			rtw89_err(rtwdev, "lv1 rcvy pci stop dma fail\n");
3949
3950		break;
3951
3952	case RTW89_LV1_RCVY_STEP_2:
3953		ret = gen_def->lv1rst_start_dma(rtwdev);
3954		if (ret)
3955			rtw89_err(rtwdev, "lv1 rcvy pci start dma fail\n");
3956		break;
3957
3958	default:
3959		return -EINVAL;
3960	}
3961
3962	return ret;
3963}
3964
3965static void rtw89_pci_ops_dump_err_status(struct rtw89_dev *rtwdev)
3966{
3967	if (rtwdev->chip->chip_gen == RTW89_CHIP_BE)
3968		return;
3969
3970	if (rtwdev->chip->chip_id == RTL8852C) {
3971		rtw89_info(rtwdev, "R_AX_DBG_ERR_FLAG=0x%08x\n",
3972			   rtw89_read32(rtwdev, R_AX_DBG_ERR_FLAG_V1));
3973		rtw89_info(rtwdev, "R_AX_LBC_WATCHDOG=0x%08x\n",
3974			   rtw89_read32(rtwdev, R_AX_LBC_WATCHDOG_V1));
3975	} else {
3976		rtw89_info(rtwdev, "R_AX_RPQ_RXBD_IDX =0x%08x\n",
3977			   rtw89_read32(rtwdev, R_AX_RPQ_RXBD_IDX));
3978		rtw89_info(rtwdev, "R_AX_DBG_ERR_FLAG=0x%08x\n",
3979			   rtw89_read32(rtwdev, R_AX_DBG_ERR_FLAG));
3980		rtw89_info(rtwdev, "R_AX_LBC_WATCHDOG=0x%08x\n",
3981			   rtw89_read32(rtwdev, R_AX_LBC_WATCHDOG));
3982	}
3983}
3984
3985static int rtw89_pci_napi_poll(struct napi_struct *napi, int budget)
3986{
3987	struct rtw89_dev *rtwdev = container_of(napi, struct rtw89_dev, napi);
3988	struct rtw89_pci *rtwpci = (struct rtw89_pci *)rtwdev->priv;
3989	const struct rtw89_pci_info *info = rtwdev->pci_info;
3990	const struct rtw89_pci_gen_def *gen_def = info->gen_def;
3991	unsigned long flags;
3992	int work_done;
3993
3994	rtwdev->napi_budget_countdown = budget;
3995
3996	rtw89_write32(rtwdev, gen_def->isr_clear_rpq.addr, gen_def->isr_clear_rpq.data);
3997	work_done = rtw89_pci_poll_rpq_dma(rtwdev, rtwpci, rtwdev->napi_budget_countdown);
3998	if (work_done == budget)
3999		return budget;
4000
4001	rtw89_write32(rtwdev, gen_def->isr_clear_rxq.addr, gen_def->isr_clear_rxq.data);
4002	work_done += rtw89_pci_poll_rxq_dma(rtwdev, rtwpci, rtwdev->napi_budget_countdown);
4003	if (work_done < budget && napi_complete_done(napi, work_done)) {
4004		spin_lock_irqsave(&rtwpci->irq_lock, flags);
4005		if (likely(rtwpci->running))
4006			rtw89_chip_enable_intr(rtwdev, rtwpci);
4007		spin_unlock_irqrestore(&rtwpci->irq_lock, flags);
4008	}
4009
4010	return work_done;
4011}
4012
4013static int __maybe_unused rtw89_pci_suspend(struct device *dev)
4014{
4015	struct ieee80211_hw *hw = dev_get_drvdata(dev);
4016	struct rtw89_dev *rtwdev = hw->priv;
4017	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
4018
4019	rtw89_write32_set(rtwdev, R_AX_RSV_CTRL, B_AX_WLOCK_1C_BIT6);
4020	rtw89_write32_set(rtwdev, R_AX_RSV_CTRL, B_AX_R_DIS_PRST);
4021	rtw89_write32_clr(rtwdev, R_AX_RSV_CTRL, B_AX_WLOCK_1C_BIT6);
4022	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
4023		rtw89_write32_clr(rtwdev, R_AX_SYS_SDIO_CTRL,
4024				  B_AX_PCIE_DIS_L2_CTRL_LDO_HCI);
4025		rtw89_write32_set(rtwdev, R_AX_PCIE_INIT_CFG1,
4026				  B_AX_PCIE_PERST_KEEP_REG | B_AX_PCIE_TRAIN_KEEP_REG);
4027	} else {
4028		rtw89_write32_clr(rtwdev, R_AX_PCIE_PS_CTRL_V1,
4029				  B_AX_CMAC_EXIT_L1_EN | B_AX_DMAC0_EXIT_L1_EN);
4030	}
4031
4032	return 0;
4033}
4034
4035static void rtw89_pci_l2_hci_ldo(struct rtw89_dev *rtwdev)
4036{
4037	if (rtwdev->chip->chip_id == RTL8852C)
4038		return;
4039
4040	/* Hardware need write the reg twice to ensure the setting work */
4041	rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_RST_MSTATE,
4042				    RTW89_PCIE_BIT_CFG_RST_MSTATE);
4043	rtw89_pci_write_config_byte(rtwdev, RTW89_PCIE_RST_MSTATE,
4044				    RTW89_PCIE_BIT_CFG_RST_MSTATE);
4045}
4046
4047static int __maybe_unused rtw89_pci_resume(struct device *dev)
4048{
4049	struct ieee80211_hw *hw = dev_get_drvdata(dev);
4050	struct rtw89_dev *rtwdev = hw->priv;
4051	enum rtw89_core_chip_id chip_id = rtwdev->chip->chip_id;
4052
4053	rtw89_write32_set(rtwdev, R_AX_RSV_CTRL, B_AX_WLOCK_1C_BIT6);
4054	rtw89_write32_clr(rtwdev, R_AX_RSV_CTRL, B_AX_R_DIS_PRST);
4055	rtw89_write32_clr(rtwdev, R_AX_RSV_CTRL, B_AX_WLOCK_1C_BIT6);
4056	if (chip_id == RTL8852A || chip_id == RTL8852B || chip_id == RTL8851B) {
4057		rtw89_write32_set(rtwdev, R_AX_SYS_SDIO_CTRL,
4058				  B_AX_PCIE_DIS_L2_CTRL_LDO_HCI);
4059		rtw89_write32_clr(rtwdev, R_AX_PCIE_INIT_CFG1,
4060				  B_AX_PCIE_PERST_KEEP_REG | B_AX_PCIE_TRAIN_KEEP_REG);
4061	} else {
4062		rtw89_write32_set(rtwdev, R_AX_PCIE_PS_CTRL_V1,
4063				  B_AX_CMAC_EXIT_L1_EN | B_AX_DMAC0_EXIT_L1_EN);
4064		rtw89_write32_clr(rtwdev, R_AX_PCIE_PS_CTRL_V1,
4065				  B_AX_SEL_REQ_ENTR_L1);
4066	}
4067	rtw89_pci_l2_hci_ldo(rtwdev);
4068	rtw89_pci_filter_out(rtwdev);
4069	rtw89_pci_link_cfg(rtwdev);
4070	rtw89_pci_l1ss_cfg(rtwdev);
4071
4072	return 0;
4073}
4074
4075SIMPLE_DEV_PM_OPS(rtw89_pm_ops, rtw89_pci_suspend, rtw89_pci_resume);
4076EXPORT_SYMBOL(rtw89_pm_ops);
4077
4078const struct rtw89_pci_gen_def rtw89_pci_gen_ax = {
4079	.isr_rdu = B_AX_RDU_INT,
4080	.isr_halt_c2h = B_AX_HALT_C2H_INT_EN,
4081	.isr_wdt_timeout = B_AX_WDT_TIMEOUT_INT_EN,
4082	.isr_clear_rpq = {R_AX_PCIE_HISR00, B_AX_RPQDMA_INT | B_AX_RPQBD_FULL_INT},
4083	.isr_clear_rxq = {R_AX_PCIE_HISR00, B_AX_RXP1DMA_INT | B_AX_RXDMA_INT |
4084					    B_AX_RDU_INT},
4085
4086	.mac_pre_init = rtw89_pci_ops_mac_pre_init_ax,
4087	.mac_pre_deinit = NULL,
4088	.mac_post_init = rtw89_pci_ops_mac_post_init_ax,
4089
4090	.clr_idx_all = rtw89_pci_clr_idx_all_ax,
4091	.rst_bdram = rtw89_pci_rst_bdram_ax,
4092
4093	.lv1rst_stop_dma = rtw89_pci_lv1rst_stop_dma_ax,
4094	.lv1rst_start_dma = rtw89_pci_lv1rst_start_dma_ax,
4095
4096	.ctrl_txdma_ch = rtw89_pci_ctrl_txdma_ch_ax,
4097	.ctrl_txdma_fw_ch = rtw89_pci_ctrl_txdma_fw_ch_ax,
4098	.poll_txdma_ch_idle = rtw89_pci_poll_txdma_ch_idle_ax,
4099
4100	.aspm_set = rtw89_pci_aspm_set_ax,
4101	.clkreq_set = rtw89_pci_clkreq_set_ax,
4102	.l1ss_set = rtw89_pci_l1ss_set_ax,
4103};
4104EXPORT_SYMBOL(rtw89_pci_gen_ax);
4105
4106static const struct rtw89_hci_ops rtw89_pci_ops = {
4107	.tx_write	= rtw89_pci_ops_tx_write,
4108	.tx_kick_off	= rtw89_pci_ops_tx_kick_off,
4109	.flush_queues	= rtw89_pci_ops_flush_queues,
4110	.reset		= rtw89_pci_ops_reset,
4111	.start		= rtw89_pci_ops_start,
4112	.stop		= rtw89_pci_ops_stop,
4113	.pause		= rtw89_pci_ops_pause,
4114	.switch_mode	= rtw89_pci_ops_switch_mode,
4115	.recalc_int_mit = rtw89_pci_recalc_int_mit,
4116
4117	.read8		= rtw89_pci_ops_read8,
4118	.read16		= rtw89_pci_ops_read16,
4119	.read32		= rtw89_pci_ops_read32,
4120	.write8		= rtw89_pci_ops_write8,
4121	.write16	= rtw89_pci_ops_write16,
4122	.write32	= rtw89_pci_ops_write32,
4123
4124	.mac_pre_init	= rtw89_pci_ops_mac_pre_init,
4125	.mac_pre_deinit	= rtw89_pci_ops_mac_pre_deinit,
4126	.mac_post_init	= rtw89_pci_ops_mac_post_init,
4127	.deinit		= rtw89_pci_ops_deinit,
4128
4129	.check_and_reclaim_tx_resource = rtw89_pci_check_and_reclaim_tx_resource,
4130	.mac_lv1_rcvy	= rtw89_pci_ops_mac_lv1_recovery,
4131	.dump_err_status = rtw89_pci_ops_dump_err_status,
4132	.napi_poll	= rtw89_pci_napi_poll,
4133
4134	.recovery_start = rtw89_pci_ops_recovery_start,
4135	.recovery_complete = rtw89_pci_ops_recovery_complete,
4136
4137	.ctrl_txdma_ch	= rtw89_pci_ctrl_txdma_ch,
4138	.ctrl_txdma_fw_ch = rtw89_pci_ctrl_txdma_fw_ch,
4139	.ctrl_trxhci	= rtw89_pci_ctrl_dma_trx,
4140	.poll_txdma_ch_idle = rtw89_pci_poll_txdma_ch_idle,
4141
4142	.clr_idx_all	= rtw89_pci_clr_idx_all,
4143	.clear		= rtw89_pci_clear_resource,
4144	.disable_intr	= rtw89_pci_disable_intr_lock,
4145	.enable_intr	= rtw89_pci_enable_intr_lock,
4146	.rst_bdram	= rtw89_pci_reset_bdram,
4147};
4148
4149int rtw89_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
4150{
4151	struct rtw89_dev *rtwdev;
4152	const struct rtw89_driver_info *info;
4153	const struct rtw89_pci_info *pci_info;
4154	int ret;
4155
4156	info = (const struct rtw89_driver_info *)id->driver_data;
4157
4158	rtwdev = rtw89_alloc_ieee80211_hw(&pdev->dev,
4159					  sizeof(struct rtw89_pci),
4160					  info->chip);
4161	if (!rtwdev) {
4162		dev_err(&pdev->dev, "failed to allocate hw\n");
4163		return -ENOMEM;
4164	}
4165
4166	pci_info = info->bus.pci;
4167
4168	rtwdev->pci_info = info->bus.pci;
4169	rtwdev->hci.ops = &rtw89_pci_ops;
4170	rtwdev->hci.type = RTW89_HCI_TYPE_PCIE;
4171	rtwdev->hci.rpwm_addr = pci_info->rpwm_addr;
4172	rtwdev->hci.cpwm_addr = pci_info->cpwm_addr;
4173
4174	SET_IEEE80211_DEV(rtwdev->hw, &pdev->dev);
4175
4176	ret = rtw89_core_init(rtwdev);
4177	if (ret) {
4178		rtw89_err(rtwdev, "failed to initialise core\n");
4179		goto err_release_hw;
4180	}
4181
4182	ret = rtw89_pci_claim_device(rtwdev, pdev);
4183	if (ret) {
4184		rtw89_err(rtwdev, "failed to claim pci device\n");
4185		goto err_core_deinit;
4186	}
4187
4188	ret = rtw89_pci_setup_resource(rtwdev, pdev);
4189	if (ret) {
4190		rtw89_err(rtwdev, "failed to setup pci resource\n");
4191		goto err_declaim_pci;
4192	}
4193
4194	ret = rtw89_chip_info_setup(rtwdev);
4195	if (ret) {
4196		rtw89_err(rtwdev, "failed to setup chip information\n");
4197		goto err_clear_resource;
4198	}
4199
4200	rtw89_pci_filter_out(rtwdev);
4201	rtw89_pci_link_cfg(rtwdev);
4202	rtw89_pci_l1ss_cfg(rtwdev);
4203
4204	rtw89_core_napi_init(rtwdev);
4205
4206	ret = rtw89_pci_request_irq(rtwdev, pdev);
4207	if (ret) {
4208		rtw89_err(rtwdev, "failed to request pci irq\n");
4209		goto err_deinit_napi;
4210	}
4211
4212	ret = rtw89_core_register(rtwdev);
4213	if (ret) {
4214		rtw89_err(rtwdev, "failed to register core\n");
4215		goto err_free_irq;
4216	}
4217
4218	set_bit(RTW89_FLAG_PROBE_DONE, rtwdev->flags);
4219
4220	return 0;
4221
4222err_free_irq:
4223	rtw89_pci_free_irq(rtwdev, pdev);
4224err_deinit_napi:
4225	rtw89_core_napi_deinit(rtwdev);
4226err_clear_resource:
4227	rtw89_pci_clear_resource(rtwdev, pdev);
4228err_declaim_pci:
4229	rtw89_pci_declaim_device(rtwdev, pdev);
4230err_core_deinit:
4231	rtw89_core_deinit(rtwdev);
4232err_release_hw:
4233	rtw89_free_ieee80211_hw(rtwdev);
4234
4235	return ret;
4236}
4237EXPORT_SYMBOL(rtw89_pci_probe);
4238
4239void rtw89_pci_remove(struct pci_dev *pdev)
4240{
4241	struct ieee80211_hw *hw = pci_get_drvdata(pdev);
4242	struct rtw89_dev *rtwdev;
4243
4244	rtwdev = hw->priv;
4245
4246	rtw89_pci_free_irq(rtwdev, pdev);
4247	rtw89_core_napi_deinit(rtwdev);
4248	rtw89_core_unregister(rtwdev);
4249	rtw89_pci_clear_resource(rtwdev, pdev);
4250	rtw89_pci_declaim_device(rtwdev, pdev);
4251	rtw89_core_deinit(rtwdev);
4252	rtw89_free_ieee80211_hw(rtwdev);
4253}
4254EXPORT_SYMBOL(rtw89_pci_remove);
4255
4256MODULE_AUTHOR("Realtek Corporation");
4257MODULE_DESCRIPTION("Realtek PCI 802.11ax wireless driver");
4258MODULE_LICENSE("Dual BSD/GPL");
4259