brintos

brintos / linux-shallow public Read only

0
0
Text · 24.2 KiB · f21de0c Raw
974 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/* Atlantic Network Driver3 *4 * Copyright (C) 2014-2019 aQuantia Corporation5 * Copyright (C) 2019-2020 Marvell International Ltd.6 */7 8/* File aq_ring.c: Definition of functions for Rx/Tx rings. */9 10#include "aq_nic.h"11#include "aq_hw.h"12#include "aq_hw_utils.h"13#include "aq_ptp.h"14#include "aq_vec.h"15#include "aq_main.h"16 17#include <net/xdp.h>18#include <linux/filter.h>19#include <linux/bpf_trace.h>20#include <linux/netdevice.h>21#include <linux/etherdevice.h>22 23static void aq_get_rxpages_xdp(struct aq_ring_buff_s *buff,24			       struct xdp_buff *xdp)25{26	struct skb_shared_info *sinfo;27	int i;28 29	if (xdp_buff_has_frags(xdp)) {30		sinfo = xdp_get_shared_info_from_buff(xdp);31 32		for (i = 0; i < sinfo->nr_frags; i++) {33			skb_frag_t *frag = &sinfo->frags[i];34 35			page_ref_inc(skb_frag_page(frag));36		}37	}38	page_ref_inc(buff->rxdata.page);39}40 41static inline void aq_free_rxpage(struct aq_rxpage *rxpage, struct device *dev)42{43	unsigned int len = PAGE_SIZE << rxpage->order;44 45	dma_unmap_page(dev, rxpage->daddr, len, DMA_FROM_DEVICE);46 47	/* Drop the ref for being in the ring. */48	__free_pages(rxpage->page, rxpage->order);49	rxpage->page = NULL;50}51 52static int aq_alloc_rxpages(struct aq_rxpage *rxpage, struct aq_ring_s *rx_ring)53{54	struct device *dev = aq_nic_get_dev(rx_ring->aq_nic);55	unsigned int order = rx_ring->page_order;56	struct page *page;57	int ret = -ENOMEM;58	dma_addr_t daddr;59 60	page = dev_alloc_pages(order);61	if (unlikely(!page))62		goto err_exit;63 64	daddr = dma_map_page(dev, page, 0, PAGE_SIZE << order,65			     DMA_FROM_DEVICE);66 67	if (unlikely(dma_mapping_error(dev, daddr)))68		goto free_page;69 70	rxpage->page = page;71	rxpage->daddr = daddr;72	rxpage->order = order;73	rxpage->pg_off = rx_ring->page_offset;74 75	return 0;76 77free_page:78	__free_pages(page, order);79 80err_exit:81	return ret;82}83 84static int aq_get_rxpages(struct aq_ring_s *self, struct aq_ring_buff_s *rxbuf)85{86	unsigned int order = self->page_order;87	u16 page_offset = self->page_offset;88	u16 frame_max = self->frame_max;89	u16 tail_size = self->tail_size;90	int ret;91 92	if (rxbuf->rxdata.page) {93		/* One means ring is the only user and can reuse */94		if (page_ref_count(rxbuf->rxdata.page) > 1) {95			/* Try reuse buffer */96			rxbuf->rxdata.pg_off += frame_max + page_offset +97						tail_size;98			if (rxbuf->rxdata.pg_off + frame_max + tail_size <=99			    (PAGE_SIZE << order)) {100				u64_stats_update_begin(&self->stats.rx.syncp);101				self->stats.rx.pg_flips++;102				u64_stats_update_end(&self->stats.rx.syncp);103 104			} else {105				/* Buffer exhausted. We have other users and106				 * should release this page and realloc107				 */108				aq_free_rxpage(&rxbuf->rxdata,109					       aq_nic_get_dev(self->aq_nic));110				u64_stats_update_begin(&self->stats.rx.syncp);111				self->stats.rx.pg_losts++;112				u64_stats_update_end(&self->stats.rx.syncp);113			}114		} else {115			rxbuf->rxdata.pg_off = page_offset;116			u64_stats_update_begin(&self->stats.rx.syncp);117			self->stats.rx.pg_reuses++;118			u64_stats_update_end(&self->stats.rx.syncp);119		}120	}121 122	if (!rxbuf->rxdata.page) {123		ret = aq_alloc_rxpages(&rxbuf->rxdata, self);124		if (ret) {125			u64_stats_update_begin(&self->stats.rx.syncp);126			self->stats.rx.alloc_fails++;127			u64_stats_update_end(&self->stats.rx.syncp);128		}129		return ret;130	}131 132	return 0;133}134 135static int aq_ring_alloc(struct aq_ring_s *self,136			 struct aq_nic_s *aq_nic)137{138	int err = 0;139 140	self->buff_ring =141		kcalloc(self->size, sizeof(struct aq_ring_buff_s), GFP_KERNEL);142 143	if (!self->buff_ring) {144		err = -ENOMEM;145		goto err_exit;146	}147 148	self->dx_ring = dma_alloc_coherent(aq_nic_get_dev(aq_nic),149					   self->size * self->dx_size,150					   &self->dx_ring_pa, GFP_KERNEL);151	if (!self->dx_ring) {152		err = -ENOMEM;153		goto err_exit;154	}155 156err_exit:157	if (err < 0) {158		aq_ring_free(self);159	}160 161	return err;162}163 164int aq_ring_tx_alloc(struct aq_ring_s *self,165		     struct aq_nic_s *aq_nic,166		     unsigned int idx,167		     struct aq_nic_cfg_s *aq_nic_cfg)168{169	self->aq_nic = aq_nic;170	self->idx = idx;171	self->size = aq_nic_cfg->txds;172	self->dx_size = aq_nic_cfg->aq_hw_caps->txd_size;173 174	return aq_ring_alloc(self, aq_nic);175}176 177int aq_ring_rx_alloc(struct aq_ring_s *self,178		     struct aq_nic_s *aq_nic,179		     unsigned int idx,180		     struct aq_nic_cfg_s *aq_nic_cfg)181{182	self->aq_nic = aq_nic;183	self->idx = idx;184	self->size = aq_nic_cfg->rxds;185	self->dx_size = aq_nic_cfg->aq_hw_caps->rxd_size;186	self->xdp_prog = aq_nic->xdp_prog;187	self->frame_max = AQ_CFG_RX_FRAME_MAX;188 189	/* Only order-2 is allowed if XDP is enabled */190	if (READ_ONCE(self->xdp_prog)) {191		self->page_offset = AQ_XDP_HEADROOM;192		self->page_order = AQ_CFG_XDP_PAGEORDER;193		self->tail_size = AQ_XDP_TAILROOM;194	} else {195		self->page_offset = 0;196		self->page_order = fls(self->frame_max / PAGE_SIZE +197				       (self->frame_max % PAGE_SIZE ? 1 : 0)) - 1;198		if (aq_nic_cfg->rxpageorder > self->page_order)199			self->page_order = aq_nic_cfg->rxpageorder;200		self->tail_size = 0;201	}202 203	return aq_ring_alloc(self, aq_nic);204}205 206int207aq_ring_hwts_rx_alloc(struct aq_ring_s *self, struct aq_nic_s *aq_nic,208		      unsigned int idx, unsigned int size, unsigned int dx_size)209{210	struct device *dev = aq_nic_get_dev(aq_nic);211	size_t sz = size * dx_size + AQ_CFG_RXDS_DEF;212 213	memset(self, 0, sizeof(*self));214 215	self->aq_nic = aq_nic;216	self->idx = idx;217	self->size = size;218	self->dx_size = dx_size;219 220	self->dx_ring = dma_alloc_coherent(dev, sz, &self->dx_ring_pa,221					   GFP_KERNEL);222	if (!self->dx_ring) {223		aq_ring_free(self);224		return -ENOMEM;225	}226 227	return 0;228}229 230int aq_ring_init(struct aq_ring_s *self, const enum atl_ring_type ring_type)231{232	self->hw_head = 0;233	self->sw_head = 0;234	self->sw_tail = 0;235	self->ring_type = ring_type;236 237	if (self->ring_type == ATL_RING_RX)238		u64_stats_init(&self->stats.rx.syncp);239	else240		u64_stats_init(&self->stats.tx.syncp);241 242	return 0;243}244 245static inline bool aq_ring_dx_in_range(unsigned int h, unsigned int i,246				       unsigned int t)247{248	return (h < t) ? ((h < i) && (i < t)) : ((h < i) || (i < t));249}250 251void aq_ring_update_queue_state(struct aq_ring_s *ring)252{253	if (aq_ring_avail_dx(ring) <= AQ_CFG_SKB_FRAGS_MAX)254		aq_ring_queue_stop(ring);255	else if (aq_ring_avail_dx(ring) > AQ_CFG_RESTART_DESC_THRES)256		aq_ring_queue_wake(ring);257}258 259void aq_ring_queue_wake(struct aq_ring_s *ring)260{261	struct net_device *ndev = aq_nic_get_ndev(ring->aq_nic);262 263	if (__netif_subqueue_stopped(ndev,264				     AQ_NIC_RING2QMAP(ring->aq_nic,265						      ring->idx))) {266		netif_wake_subqueue(ndev,267				    AQ_NIC_RING2QMAP(ring->aq_nic, ring->idx));268		u64_stats_update_begin(&ring->stats.tx.syncp);269		ring->stats.tx.queue_restarts++;270		u64_stats_update_end(&ring->stats.tx.syncp);271	}272}273 274void aq_ring_queue_stop(struct aq_ring_s *ring)275{276	struct net_device *ndev = aq_nic_get_ndev(ring->aq_nic);277 278	if (!__netif_subqueue_stopped(ndev,279				      AQ_NIC_RING2QMAP(ring->aq_nic,280						       ring->idx)))281		netif_stop_subqueue(ndev,282				    AQ_NIC_RING2QMAP(ring->aq_nic, ring->idx));283}284 285bool aq_ring_tx_clean(struct aq_ring_s *self)286{287	struct device *dev = aq_nic_get_dev(self->aq_nic);288	unsigned int budget;289 290	for (budget = AQ_CFG_TX_CLEAN_BUDGET;291	     budget && self->sw_head != self->hw_head; budget--) {292		struct aq_ring_buff_s *buff = &self->buff_ring[self->sw_head];293 294		if (likely(buff->is_mapped)) {295			if (unlikely(buff->is_sop)) {296				if (!buff->is_eop &&297				    buff->eop_index != 0xffffU &&298				    (!aq_ring_dx_in_range(self->sw_head,299						buff->eop_index,300						self->hw_head)))301					break;302 303				dma_unmap_single(dev, buff->pa, buff->len,304						 DMA_TO_DEVICE);305			} else {306				dma_unmap_page(dev, buff->pa, buff->len,307					       DMA_TO_DEVICE);308			}309		}310 311		if (likely(!buff->is_eop))312			goto out;313 314		if (buff->skb) {315			u64_stats_update_begin(&self->stats.tx.syncp);316			++self->stats.tx.packets;317			self->stats.tx.bytes += buff->skb->len;318			u64_stats_update_end(&self->stats.tx.syncp);319			dev_kfree_skb_any(buff->skb);320		} else if (buff->xdpf) {321			u64_stats_update_begin(&self->stats.tx.syncp);322			++self->stats.tx.packets;323			self->stats.tx.bytes += xdp_get_frame_len(buff->xdpf);324			u64_stats_update_end(&self->stats.tx.syncp);325			xdp_return_frame_rx_napi(buff->xdpf);326		}327 328out:329		buff->skb = NULL;330		buff->xdpf = NULL;331		buff->pa = 0U;332		buff->eop_index = 0xffffU;333		self->sw_head = aq_ring_next_dx(self, self->sw_head);334	}335 336	return !!budget;337}338 339static void aq_rx_checksum(struct aq_ring_s *self,340			   struct aq_ring_buff_s *buff,341			   struct sk_buff *skb)342{343	if (!(self->aq_nic->ndev->features & NETIF_F_RXCSUM))344		return;345 346	if (unlikely(buff->is_cso_err)) {347		u64_stats_update_begin(&self->stats.rx.syncp);348		++self->stats.rx.errors;349		u64_stats_update_end(&self->stats.rx.syncp);350		skb->ip_summed = CHECKSUM_NONE;351		return;352	}353	if (buff->is_ip_cso) {354		__skb_incr_checksum_unnecessary(skb);355	} else {356		skb->ip_summed = CHECKSUM_NONE;357	}358 359	if (buff->is_udp_cso || buff->is_tcp_cso)360		__skb_incr_checksum_unnecessary(skb);361}362 363int aq_xdp_xmit(struct net_device *dev, int num_frames,364		struct xdp_frame **frames, u32 flags)365{366	struct aq_nic_s *aq_nic = netdev_priv(dev);367	unsigned int vec, i, drop = 0;368	int cpu = smp_processor_id();369	struct aq_nic_cfg_s *aq_cfg;370	struct aq_ring_s *ring;371 372	aq_cfg = aq_nic_get_cfg(aq_nic);373	vec = cpu % aq_cfg->vecs;374	ring = aq_nic->aq_ring_tx[AQ_NIC_CFG_TCVEC2RING(aq_cfg, 0, vec)];375 376	for (i = 0; i < num_frames; i++) {377		struct xdp_frame *xdpf = frames[i];378 379		if (aq_nic_xmit_xdpf(aq_nic, ring, xdpf) == NETDEV_TX_BUSY)380			drop++;381	}382 383	return num_frames - drop;384}385 386static struct sk_buff *aq_xdp_build_skb(struct xdp_buff *xdp,387					struct net_device *dev,388					struct aq_ring_buff_s *buff)389{390	struct xdp_frame *xdpf;391	struct sk_buff *skb;392 393	xdpf = xdp_convert_buff_to_frame(xdp);394	if (unlikely(!xdpf))395		return NULL;396 397	skb = xdp_build_skb_from_frame(xdpf, dev);398	if (!skb)399		return NULL;400 401	aq_get_rxpages_xdp(buff, xdp);402	return skb;403}404 405static struct sk_buff *aq_xdp_run_prog(struct aq_nic_s *aq_nic,406				       struct xdp_buff *xdp,407				       struct aq_ring_s *rx_ring,408				       struct aq_ring_buff_s *buff)409{410	int result = NETDEV_TX_BUSY;411	struct aq_ring_s *tx_ring;412	struct xdp_frame *xdpf;413	struct bpf_prog *prog;414	u32 act = XDP_ABORTED;415	struct sk_buff *skb;416 417	u64_stats_update_begin(&rx_ring->stats.rx.syncp);418	++rx_ring->stats.rx.packets;419	rx_ring->stats.rx.bytes += xdp_get_buff_len(xdp);420	u64_stats_update_end(&rx_ring->stats.rx.syncp);421 422	prog = READ_ONCE(rx_ring->xdp_prog);423	if (!prog)424		return aq_xdp_build_skb(xdp, aq_nic->ndev, buff);425 426	prefetchw(xdp->data_hard_start); /* xdp_frame write */427 428	/* single buffer XDP program, but packet is multi buffer, aborted */429	if (xdp_buff_has_frags(xdp) && !prog->aux->xdp_has_frags)430		goto out_aborted;431 432	act = bpf_prog_run_xdp(prog, xdp);433	switch (act) {434	case XDP_PASS:435		skb = aq_xdp_build_skb(xdp, aq_nic->ndev, buff);436		if (!skb)437			goto out_aborted;438		u64_stats_update_begin(&rx_ring->stats.rx.syncp);439		++rx_ring->stats.rx.xdp_pass;440		u64_stats_update_end(&rx_ring->stats.rx.syncp);441		return skb;442	case XDP_TX:443		xdpf = xdp_convert_buff_to_frame(xdp);444		if (unlikely(!xdpf))445			goto out_aborted;446		tx_ring = aq_nic->aq_ring_tx[rx_ring->idx];447		result = aq_nic_xmit_xdpf(aq_nic, tx_ring, xdpf);448		if (result == NETDEV_TX_BUSY)449			goto out_aborted;450		u64_stats_update_begin(&rx_ring->stats.rx.syncp);451		++rx_ring->stats.rx.xdp_tx;452		u64_stats_update_end(&rx_ring->stats.rx.syncp);453		aq_get_rxpages_xdp(buff, xdp);454		break;455	case XDP_REDIRECT:456		if (xdp_do_redirect(aq_nic->ndev, xdp, prog) < 0)457			goto out_aborted;458		xdp_do_flush();459		u64_stats_update_begin(&rx_ring->stats.rx.syncp);460		++rx_ring->stats.rx.xdp_redirect;461		u64_stats_update_end(&rx_ring->stats.rx.syncp);462		aq_get_rxpages_xdp(buff, xdp);463		break;464	default:465		fallthrough;466	case XDP_ABORTED:467out_aborted:468		u64_stats_update_begin(&rx_ring->stats.rx.syncp);469		++rx_ring->stats.rx.xdp_aborted;470		u64_stats_update_end(&rx_ring->stats.rx.syncp);471		trace_xdp_exception(aq_nic->ndev, prog, act);472		bpf_warn_invalid_xdp_action(aq_nic->ndev, prog, act);473		break;474	case XDP_DROP:475		u64_stats_update_begin(&rx_ring->stats.rx.syncp);476		++rx_ring->stats.rx.xdp_drop;477		u64_stats_update_end(&rx_ring->stats.rx.syncp);478		break;479	}480 481	return ERR_PTR(-result);482}483 484static bool aq_add_rx_fragment(struct device *dev,485			       struct aq_ring_s *ring,486			       struct aq_ring_buff_s *buff,487			       struct xdp_buff *xdp)488{489	struct skb_shared_info *sinfo = xdp_get_shared_info_from_buff(xdp);490	struct aq_ring_buff_s *buff_ = buff;491 492	memset(sinfo, 0, sizeof(*sinfo));493	do {494		skb_frag_t *frag;495 496		if (unlikely(sinfo->nr_frags >= MAX_SKB_FRAGS))497			return true;498 499		frag = &sinfo->frags[sinfo->nr_frags++];500		buff_ = &ring->buff_ring[buff_->next];501		dma_sync_single_range_for_cpu(dev,502					      buff_->rxdata.daddr,503					      buff_->rxdata.pg_off,504					      buff_->len,505					      DMA_FROM_DEVICE);506		sinfo->xdp_frags_size += buff_->len;507		skb_frag_fill_page_desc(frag, buff_->rxdata.page,508					buff_->rxdata.pg_off,509					buff_->len);510 511		buff_->is_cleaned = 1;512 513		buff->is_ip_cso &= buff_->is_ip_cso;514		buff->is_udp_cso &= buff_->is_udp_cso;515		buff->is_tcp_cso &= buff_->is_tcp_cso;516		buff->is_cso_err |= buff_->is_cso_err;517 518		if (page_is_pfmemalloc(buff_->rxdata.page))519			xdp_buff_set_frag_pfmemalloc(xdp);520 521	} while (!buff_->is_eop);522 523	xdp_buff_set_frags_flag(xdp);524 525	return false;526}527 528static int __aq_ring_rx_clean(struct aq_ring_s *self, struct napi_struct *napi,529			      int *work_done, int budget)530{531	struct net_device *ndev = aq_nic_get_ndev(self->aq_nic);532	int err = 0;533 534	for (; (self->sw_head != self->hw_head) && budget;535		self->sw_head = aq_ring_next_dx(self, self->sw_head),536		--budget, ++(*work_done)) {537		struct aq_ring_buff_s *buff = &self->buff_ring[self->sw_head];538		bool is_ptp_ring = aq_ptp_ring(self->aq_nic, self);539		struct aq_ring_buff_s *buff_ = NULL;540		struct sk_buff *skb = NULL;541		unsigned int next_ = 0U;542		unsigned int i = 0U;543		u16 hdr_len;544 545		if (buff->is_cleaned)546			continue;547 548		if (!buff->is_eop) {549			unsigned int frag_cnt = 0U;550			buff_ = buff;551			do {552				bool is_rsc_completed = true;553 554				if (buff_->next >= self->size) {555					err = -EIO;556					goto err_exit;557				}558 559				frag_cnt++;560				next_ = buff_->next;561				buff_ = &self->buff_ring[next_];562				is_rsc_completed =563					aq_ring_dx_in_range(self->sw_head,564							    next_,565							    self->hw_head);566 567				if (unlikely(!is_rsc_completed) ||568						frag_cnt > MAX_SKB_FRAGS) {569					err = 0;570					goto err_exit;571				}572 573				buff->is_error |= buff_->is_error;574				buff->is_cso_err |= buff_->is_cso_err;575 576			} while (!buff_->is_eop);577 578			if (buff->is_error ||579			    (buff->is_lro && buff->is_cso_err)) {580				buff_ = buff;581				do {582					if (buff_->next >= self->size) {583						err = -EIO;584						goto err_exit;585					}586					next_ = buff_->next;587					buff_ = &self->buff_ring[next_];588 589					buff_->is_cleaned = true;590				} while (!buff_->is_eop);591 592				u64_stats_update_begin(&self->stats.rx.syncp);593				++self->stats.rx.errors;594				u64_stats_update_end(&self->stats.rx.syncp);595				continue;596			}597		}598 599		if (buff->is_error) {600			u64_stats_update_begin(&self->stats.rx.syncp);601			++self->stats.rx.errors;602			u64_stats_update_end(&self->stats.rx.syncp);603			continue;604		}605 606		dma_sync_single_range_for_cpu(aq_nic_get_dev(self->aq_nic),607					      buff->rxdata.daddr,608					      buff->rxdata.pg_off,609					      buff->len, DMA_FROM_DEVICE);610 611		skb = napi_alloc_skb(napi, AQ_CFG_RX_HDR_SIZE);612		if (unlikely(!skb)) {613			u64_stats_update_begin(&self->stats.rx.syncp);614			self->stats.rx.skb_alloc_fails++;615			u64_stats_update_end(&self->stats.rx.syncp);616			err = -ENOMEM;617			goto err_exit;618		}619		if (is_ptp_ring)620			buff->len -=621				aq_ptp_extract_ts(self->aq_nic, skb_hwtstamps(skb),622						  aq_buf_vaddr(&buff->rxdata),623						  buff->len);624 625		hdr_len = buff->len;626		if (hdr_len > AQ_CFG_RX_HDR_SIZE)627			hdr_len = eth_get_headlen(skb->dev,628						  aq_buf_vaddr(&buff->rxdata),629						  AQ_CFG_RX_HDR_SIZE);630 631		memcpy(__skb_put(skb, hdr_len), aq_buf_vaddr(&buff->rxdata),632		       ALIGN(hdr_len, sizeof(long)));633 634		if (buff->len - hdr_len > 0) {635			skb_add_rx_frag(skb, i++, buff->rxdata.page,636					buff->rxdata.pg_off + hdr_len,637					buff->len - hdr_len,638					self->frame_max);639			page_ref_inc(buff->rxdata.page);640		}641 642		if (!buff->is_eop) {643			buff_ = buff;644			do {645				next_ = buff_->next;646				buff_ = &self->buff_ring[next_];647 648				dma_sync_single_range_for_cpu(aq_nic_get_dev(self->aq_nic),649							      buff_->rxdata.daddr,650							      buff_->rxdata.pg_off,651							      buff_->len,652							      DMA_FROM_DEVICE);653				skb_add_rx_frag(skb, i++,654						buff_->rxdata.page,655						buff_->rxdata.pg_off,656						buff_->len,657						self->frame_max);658				page_ref_inc(buff_->rxdata.page);659				buff_->is_cleaned = 1;660 661				buff->is_ip_cso &= buff_->is_ip_cso;662				buff->is_udp_cso &= buff_->is_udp_cso;663				buff->is_tcp_cso &= buff_->is_tcp_cso;664				buff->is_cso_err |= buff_->is_cso_err;665 666			} while (!buff_->is_eop);667		}668 669		if (buff->is_vlan)670			__vlan_hwaccel_put_tag(skb, htons(ETH_P_8021Q),671					       buff->vlan_rx_tag);672 673		skb->protocol = eth_type_trans(skb, ndev);674 675		aq_rx_checksum(self, buff, skb);676 677		skb_set_hash(skb, buff->rss_hash,678			     buff->is_hash_l4 ? PKT_HASH_TYPE_L4 :679			     PKT_HASH_TYPE_NONE);680		/* Send all PTP traffic to 0 queue */681		skb_record_rx_queue(skb,682				    is_ptp_ring ? 0683						: AQ_NIC_RING2QMAP(self->aq_nic,684								   self->idx));685 686		u64_stats_update_begin(&self->stats.rx.syncp);687		++self->stats.rx.packets;688		self->stats.rx.bytes += skb->len;689		u64_stats_update_end(&self->stats.rx.syncp);690 691		napi_gro_receive(napi, skb);692	}693 694err_exit:695	return err;696}697 698static int __aq_ring_xdp_clean(struct aq_ring_s *rx_ring,699			       struct napi_struct *napi, int *work_done,700			       int budget)701{702	int frame_sz = rx_ring->page_offset + rx_ring->frame_max +703		       rx_ring->tail_size;704	struct aq_nic_s *aq_nic = rx_ring->aq_nic;705	bool is_rsc_completed = true;706	struct device *dev;707	int err = 0;708 709	dev = aq_nic_get_dev(aq_nic);710	for (; (rx_ring->sw_head != rx_ring->hw_head) && budget;711		rx_ring->sw_head = aq_ring_next_dx(rx_ring, rx_ring->sw_head),712		--budget, ++(*work_done)) {713		struct aq_ring_buff_s *buff = &rx_ring->buff_ring[rx_ring->sw_head];714		bool is_ptp_ring = aq_ptp_ring(rx_ring->aq_nic, rx_ring);715		struct aq_ring_buff_s *buff_ = NULL;716		u16 ptp_hwtstamp_len = 0;717		struct skb_shared_hwtstamps shhwtstamps;718		struct sk_buff *skb = NULL;719		unsigned int next_ = 0U;720		struct xdp_buff xdp;721		void *hard_start;722 723		if (buff->is_cleaned)724			continue;725 726		if (!buff->is_eop) {727			buff_ = buff;728			do {729				if (buff_->next >= rx_ring->size) {730					err = -EIO;731					goto err_exit;732				}733				next_ = buff_->next;734				buff_ = &rx_ring->buff_ring[next_];735				is_rsc_completed =736					aq_ring_dx_in_range(rx_ring->sw_head,737							    next_,738							    rx_ring->hw_head);739 740				if (unlikely(!is_rsc_completed))741					break;742 743				buff->is_error |= buff_->is_error;744				buff->is_cso_err |= buff_->is_cso_err;745			} while (!buff_->is_eop);746 747			if (!is_rsc_completed) {748				err = 0;749				goto err_exit;750			}751			if (buff->is_error ||752			    (buff->is_lro && buff->is_cso_err)) {753				buff_ = buff;754				do {755					if (buff_->next >= rx_ring->size) {756						err = -EIO;757						goto err_exit;758					}759					next_ = buff_->next;760					buff_ = &rx_ring->buff_ring[next_];761 762					buff_->is_cleaned = true;763				} while (!buff_->is_eop);764 765				u64_stats_update_begin(&rx_ring->stats.rx.syncp);766				++rx_ring->stats.rx.errors;767				u64_stats_update_end(&rx_ring->stats.rx.syncp);768				continue;769			}770		}771 772		if (buff->is_error) {773			u64_stats_update_begin(&rx_ring->stats.rx.syncp);774			++rx_ring->stats.rx.errors;775			u64_stats_update_end(&rx_ring->stats.rx.syncp);776			continue;777		}778 779		dma_sync_single_range_for_cpu(dev,780					      buff->rxdata.daddr,781					      buff->rxdata.pg_off,782					      buff->len, DMA_FROM_DEVICE);783		hard_start = page_address(buff->rxdata.page) +784			     buff->rxdata.pg_off - rx_ring->page_offset;785 786		if (is_ptp_ring) {787			ptp_hwtstamp_len = aq_ptp_extract_ts(rx_ring->aq_nic, &shhwtstamps,788							     aq_buf_vaddr(&buff->rxdata),789							     buff->len);790			buff->len -= ptp_hwtstamp_len;791		}792 793		xdp_init_buff(&xdp, frame_sz, &rx_ring->xdp_rxq);794		xdp_prepare_buff(&xdp, hard_start, rx_ring->page_offset,795				 buff->len, false);796		if (!buff->is_eop) {797			if (aq_add_rx_fragment(dev, rx_ring, buff, &xdp)) {798				u64_stats_update_begin(&rx_ring->stats.rx.syncp);799				++rx_ring->stats.rx.packets;800				rx_ring->stats.rx.bytes += xdp_get_buff_len(&xdp);801				++rx_ring->stats.rx.xdp_aborted;802				u64_stats_update_end(&rx_ring->stats.rx.syncp);803				continue;804			}805		}806 807		skb = aq_xdp_run_prog(aq_nic, &xdp, rx_ring, buff);808		if (IS_ERR(skb) || !skb)809			continue;810 811		if (ptp_hwtstamp_len > 0)812			*skb_hwtstamps(skb) = shhwtstamps;813 814		if (buff->is_vlan)815			__vlan_hwaccel_put_tag(skb, htons(ETH_P_8021Q),816					       buff->vlan_rx_tag);817 818		aq_rx_checksum(rx_ring, buff, skb);819 820		skb_set_hash(skb, buff->rss_hash,821			     buff->is_hash_l4 ? PKT_HASH_TYPE_L4 :822			     PKT_HASH_TYPE_NONE);823		/* Send all PTP traffic to 0 queue */824		skb_record_rx_queue(skb,825				    is_ptp_ring ? 0826						: AQ_NIC_RING2QMAP(rx_ring->aq_nic,827								   rx_ring->idx));828 829		napi_gro_receive(napi, skb);830	}831 832err_exit:833	return err;834}835 836int aq_ring_rx_clean(struct aq_ring_s *self,837		     struct napi_struct *napi,838		     int *work_done,839		     int budget)840{841	if (static_branch_unlikely(&aq_xdp_locking_key))842		return __aq_ring_xdp_clean(self, napi, work_done, budget);843	else844		return __aq_ring_rx_clean(self, napi, work_done, budget);845}846 847void aq_ring_hwts_rx_clean(struct aq_ring_s *self, struct aq_nic_s *aq_nic)848{849#if IS_REACHABLE(CONFIG_PTP_1588_CLOCK)850	while (self->sw_head != self->hw_head) {851		u64 ns;852 853		aq_nic->aq_hw_ops->extract_hwts(aq_nic->aq_hw,854						self->dx_ring +855						(self->sw_head * self->dx_size),856						self->dx_size, &ns);857		aq_ptp_tx_hwtstamp(aq_nic, ns);858 859		self->sw_head = aq_ring_next_dx(self, self->sw_head);860	}861#endif862}863 864int aq_ring_rx_fill(struct aq_ring_s *self)865{866	struct aq_ring_buff_s *buff = NULL;867	int err = 0;868	int i = 0;869 870	if (aq_ring_avail_dx(self) < min_t(unsigned int, AQ_CFG_RX_REFILL_THRES,871					   self->size / 2))872		return err;873 874	for (i = aq_ring_avail_dx(self); i--;875		self->sw_tail = aq_ring_next_dx(self, self->sw_tail)) {876		buff = &self->buff_ring[self->sw_tail];877 878		buff->flags = 0U;879		buff->len = self->frame_max;880 881		err = aq_get_rxpages(self, buff);882		if (err)883			goto err_exit;884 885		buff->pa = aq_buf_daddr(&buff->rxdata);886		buff = NULL;887	}888 889err_exit:890	return err;891}892 893void aq_ring_rx_deinit(struct aq_ring_s *self)894{895	if (!self)896		return;897 898	for (; self->sw_head != self->sw_tail;899		self->sw_head = aq_ring_next_dx(self, self->sw_head)) {900		struct aq_ring_buff_s *buff = &self->buff_ring[self->sw_head];901 902		aq_free_rxpage(&buff->rxdata, aq_nic_get_dev(self->aq_nic));903	}904}905 906void aq_ring_free(struct aq_ring_s *self)907{908	if (!self)909		return;910 911	kfree(self->buff_ring);912	self->buff_ring = NULL;913 914	if (self->dx_ring) {915		dma_free_coherent(aq_nic_get_dev(self->aq_nic),916				  self->size * self->dx_size, self->dx_ring,917				  self->dx_ring_pa);918		self->dx_ring = NULL;919	}920}921 922void aq_ring_hwts_rx_free(struct aq_ring_s *self)923{924	if (!self)925		return;926 927	if (self->dx_ring) {928		dma_free_coherent(aq_nic_get_dev(self->aq_nic),929				  self->size * self->dx_size + AQ_CFG_RXDS_DEF,930				  self->dx_ring, self->dx_ring_pa);931		self->dx_ring = NULL;932	}933}934 935unsigned int aq_ring_fill_stats_data(struct aq_ring_s *self, u64 *data)936{937	unsigned int count;938	unsigned int start;939 940	if (self->ring_type == ATL_RING_RX) {941		/* This data should mimic aq_ethtool_queue_rx_stat_names structure */942		do {943			count = 0;944			start = u64_stats_fetch_begin(&self->stats.rx.syncp);945			data[count] = self->stats.rx.packets;946			data[++count] = self->stats.rx.jumbo_packets;947			data[++count] = self->stats.rx.lro_packets;948			data[++count] = self->stats.rx.errors;949			data[++count] = self->stats.rx.alloc_fails;950			data[++count] = self->stats.rx.skb_alloc_fails;951			data[++count] = self->stats.rx.polls;952			data[++count] = self->stats.rx.pg_flips;953			data[++count] = self->stats.rx.pg_reuses;954			data[++count] = self->stats.rx.pg_losts;955			data[++count] = self->stats.rx.xdp_aborted;956			data[++count] = self->stats.rx.xdp_drop;957			data[++count] = self->stats.rx.xdp_pass;958			data[++count] = self->stats.rx.xdp_tx;959			data[++count] = self->stats.rx.xdp_invalid;960			data[++count] = self->stats.rx.xdp_redirect;961		} while (u64_stats_fetch_retry(&self->stats.rx.syncp, start));962	} else {963		/* This data should mimic aq_ethtool_queue_tx_stat_names structure */964		do {965			count = 0;966			start = u64_stats_fetch_begin(&self->stats.tx.syncp);967			data[count] = self->stats.tx.packets;968			data[++count] = self->stats.tx.queue_restarts;969		} while (u64_stats_fetch_retry(&self->stats.tx.syncp, start));970	}971 972	return ++count;973}974