xref: /dpdk/drivers/net/virtio/virtio_rxtx.c (revision 4b058fdd27ceac277aa715df6f6bb0f7b37b150b)
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2010-2014 Intel Corporation
3  */
4 
5 #include <stdint.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <errno.h>
10 
11 #include <rte_cycles.h>
12 #include <rte_memory.h>
13 #include <rte_branch_prediction.h>
14 #include <rte_mempool.h>
15 #include <rte_malloc.h>
16 #include <rte_mbuf.h>
17 #include <rte_ether.h>
18 #include <rte_ethdev_driver.h>
19 #include <rte_prefetch.h>
20 #include <rte_string_fns.h>
21 #include <rte_errno.h>
22 #include <rte_byteorder.h>
23 #include <rte_net.h>
24 #include <rte_ip.h>
25 #include <rte_udp.h>
26 #include <rte_tcp.h>
27 
28 #include "virtio_logs.h"
29 #include "virtio_ethdev.h"
30 #include "virtio_pci.h"
31 #include "virtqueue.h"
32 #include "virtio_rxtx.h"
33 #include "virtio_rxtx_simple.h"
34 #include "virtio_ring.h"
35 
36 #ifdef RTE_LIBRTE_VIRTIO_DEBUG_DUMP
37 #define VIRTIO_DUMP_PACKET(m, len) rte_pktmbuf_dump(stdout, m, len)
38 #else
39 #define  VIRTIO_DUMP_PACKET(m, len) do { } while (0)
40 #endif
41 
42 int
43 virtio_dev_rx_queue_done(void *rxq, uint16_t offset)
44 {
45 	struct virtnet_rx *rxvq = rxq;
46 	struct virtqueue *vq = rxvq->vq;
47 
48 	return VIRTQUEUE_NUSED(vq) >= offset;
49 }
50 
51 void
52 vq_ring_free_inorder(struct virtqueue *vq, uint16_t desc_idx, uint16_t num)
53 {
54 	vq->vq_free_cnt += num;
55 	vq->vq_desc_tail_idx = desc_idx & (vq->vq_nentries - 1);
56 }
57 
58 void
59 vq_ring_free_chain(struct virtqueue *vq, uint16_t desc_idx)
60 {
61 	struct vring_desc *dp, *dp_tail;
62 	struct vq_desc_extra *dxp;
63 	uint16_t desc_idx_last = desc_idx;
64 
65 	dp  = &vq->vq_ring.desc[desc_idx];
66 	dxp = &vq->vq_descx[desc_idx];
67 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt + dxp->ndescs);
68 	if ((dp->flags & VRING_DESC_F_INDIRECT) == 0) {
69 		while (dp->flags & VRING_DESC_F_NEXT) {
70 			desc_idx_last = dp->next;
71 			dp = &vq->vq_ring.desc[dp->next];
72 		}
73 	}
74 	dxp->ndescs = 0;
75 
76 	/*
77 	 * We must append the existing free chain, if any, to the end of
78 	 * newly freed chain. If the virtqueue was completely used, then
79 	 * head would be VQ_RING_DESC_CHAIN_END (ASSERTed above).
80 	 */
81 	if (vq->vq_desc_tail_idx == VQ_RING_DESC_CHAIN_END) {
82 		vq->vq_desc_head_idx = desc_idx;
83 	} else {
84 		dp_tail = &vq->vq_ring.desc[vq->vq_desc_tail_idx];
85 		dp_tail->next = desc_idx;
86 	}
87 
88 	vq->vq_desc_tail_idx = desc_idx_last;
89 	dp->next = VQ_RING_DESC_CHAIN_END;
90 }
91 
92 static void
93 vq_ring_free_id_packed(struct virtqueue *vq, uint16_t id)
94 {
95 	struct vq_desc_extra *dxp;
96 
97 	dxp = &vq->vq_descx[id];
98 	vq->vq_free_cnt += dxp->ndescs;
99 
100 	if (vq->vq_desc_tail_idx == VQ_RING_DESC_CHAIN_END)
101 		vq->vq_desc_head_idx = id;
102 	else
103 		vq->vq_descx[vq->vq_desc_tail_idx].next = id;
104 
105 	vq->vq_desc_tail_idx = id;
106 	dxp->next = VQ_RING_DESC_CHAIN_END;
107 }
108 
109 static uint16_t
110 virtqueue_dequeue_burst_rx_packed(struct virtqueue *vq,
111 				  struct rte_mbuf **rx_pkts,
112 				  uint32_t *len,
113 				  uint16_t num)
114 {
115 	struct rte_mbuf *cookie;
116 	uint16_t used_idx;
117 	uint16_t id;
118 	struct vring_packed_desc *desc;
119 	uint16_t i;
120 
121 	desc = vq->ring_packed.desc_packed;
122 
123 	for (i = 0; i < num; i++) {
124 		used_idx = vq->vq_used_cons_idx;
125 		if (!desc_is_used(&desc[used_idx], vq))
126 			return i;
127 		virtio_rmb(vq->hw->weak_barriers);
128 		len[i] = desc[used_idx].len;
129 		id = desc[used_idx].id;
130 		cookie = (struct rte_mbuf *)vq->vq_descx[id].cookie;
131 		if (unlikely(cookie == NULL)) {
132 			PMD_DRV_LOG(ERR, "vring descriptor with no mbuf cookie at %u",
133 				vq->vq_used_cons_idx);
134 			break;
135 		}
136 		rte_prefetch0(cookie);
137 		rte_packet_prefetch(rte_pktmbuf_mtod(cookie, void *));
138 		rx_pkts[i] = cookie;
139 
140 		vq->vq_free_cnt++;
141 		vq->vq_used_cons_idx++;
142 		if (vq->vq_used_cons_idx >= vq->vq_nentries) {
143 			vq->vq_used_cons_idx -= vq->vq_nentries;
144 			vq->used_wrap_counter ^= 1;
145 		}
146 	}
147 
148 	return i;
149 }
150 
151 static uint16_t
152 virtqueue_dequeue_burst_rx(struct virtqueue *vq, struct rte_mbuf **rx_pkts,
153 			   uint32_t *len, uint16_t num)
154 {
155 	struct vring_used_elem *uep;
156 	struct rte_mbuf *cookie;
157 	uint16_t used_idx, desc_idx;
158 	uint16_t i;
159 
160 	/*  Caller does the check */
161 	for (i = 0; i < num ; i++) {
162 		used_idx = (uint16_t)(vq->vq_used_cons_idx & (vq->vq_nentries - 1));
163 		uep = &vq->vq_ring.used->ring[used_idx];
164 		desc_idx = (uint16_t) uep->id;
165 		len[i] = uep->len;
166 		cookie = (struct rte_mbuf *)vq->vq_descx[desc_idx].cookie;
167 
168 		if (unlikely(cookie == NULL)) {
169 			PMD_DRV_LOG(ERR, "vring descriptor with no mbuf cookie at %u",
170 				vq->vq_used_cons_idx);
171 			break;
172 		}
173 
174 		rte_prefetch0(cookie);
175 		rte_packet_prefetch(rte_pktmbuf_mtod(cookie, void *));
176 		rx_pkts[i]  = cookie;
177 		vq->vq_used_cons_idx++;
178 		vq_ring_free_chain(vq, desc_idx);
179 		vq->vq_descx[desc_idx].cookie = NULL;
180 	}
181 
182 	return i;
183 }
184 
185 static uint16_t
186 virtqueue_dequeue_rx_inorder(struct virtqueue *vq,
187 			struct rte_mbuf **rx_pkts,
188 			uint32_t *len,
189 			uint16_t num)
190 {
191 	struct vring_used_elem *uep;
192 	struct rte_mbuf *cookie;
193 	uint16_t used_idx = 0;
194 	uint16_t i;
195 
196 	if (unlikely(num == 0))
197 		return 0;
198 
199 	for (i = 0; i < num; i++) {
200 		used_idx = vq->vq_used_cons_idx & (vq->vq_nentries - 1);
201 		/* Desc idx same as used idx */
202 		uep = &vq->vq_ring.used->ring[used_idx];
203 		len[i] = uep->len;
204 		cookie = (struct rte_mbuf *)vq->vq_descx[used_idx].cookie;
205 
206 		if (unlikely(cookie == NULL)) {
207 			PMD_DRV_LOG(ERR, "vring descriptor with no mbuf cookie at %u",
208 				vq->vq_used_cons_idx);
209 			break;
210 		}
211 
212 		rte_prefetch0(cookie);
213 		rte_packet_prefetch(rte_pktmbuf_mtod(cookie, void *));
214 		rx_pkts[i]  = cookie;
215 		vq->vq_used_cons_idx++;
216 		vq->vq_descx[used_idx].cookie = NULL;
217 	}
218 
219 	vq_ring_free_inorder(vq, used_idx, i);
220 	return i;
221 }
222 
223 #ifndef DEFAULT_TX_FREE_THRESH
224 #define DEFAULT_TX_FREE_THRESH 32
225 #endif
226 
227 /* Cleanup from completed transmits. */
228 static void
229 virtio_xmit_cleanup_packed(struct virtqueue *vq, int num)
230 {
231 	uint16_t used_idx, id;
232 	uint16_t size = vq->vq_nentries;
233 	struct vring_packed_desc *desc = vq->ring_packed.desc_packed;
234 	struct vq_desc_extra *dxp;
235 
236 	used_idx = vq->vq_used_cons_idx;
237 	while (num-- && desc_is_used(&desc[used_idx], vq)) {
238 		virtio_rmb(vq->hw->weak_barriers);
239 		id = desc[used_idx].id;
240 		dxp = &vq->vq_descx[id];
241 		vq->vq_used_cons_idx += dxp->ndescs;
242 		if (vq->vq_used_cons_idx >= size) {
243 			vq->vq_used_cons_idx -= size;
244 			vq->used_wrap_counter ^= 1;
245 		}
246 		vq_ring_free_id_packed(vq, id);
247 		if (dxp->cookie != NULL) {
248 			rte_pktmbuf_free(dxp->cookie);
249 			dxp->cookie = NULL;
250 		}
251 		used_idx = vq->vq_used_cons_idx;
252 	}
253 }
254 
255 static void
256 virtio_xmit_cleanup(struct virtqueue *vq, uint16_t num)
257 {
258 	uint16_t i, used_idx, desc_idx;
259 	for (i = 0; i < num; i++) {
260 		struct vring_used_elem *uep;
261 		struct vq_desc_extra *dxp;
262 
263 		used_idx = (uint16_t)(vq->vq_used_cons_idx & (vq->vq_nentries - 1));
264 		uep = &vq->vq_ring.used->ring[used_idx];
265 
266 		desc_idx = (uint16_t) uep->id;
267 		dxp = &vq->vq_descx[desc_idx];
268 		vq->vq_used_cons_idx++;
269 		vq_ring_free_chain(vq, desc_idx);
270 
271 		if (dxp->cookie != NULL) {
272 			rte_pktmbuf_free(dxp->cookie);
273 			dxp->cookie = NULL;
274 		}
275 	}
276 }
277 
278 /* Cleanup from completed inorder transmits. */
279 static void
280 virtio_xmit_cleanup_inorder(struct virtqueue *vq, uint16_t num)
281 {
282 	uint16_t i, used_idx, desc_idx = 0, last_idx;
283 	int16_t free_cnt = 0;
284 	struct vq_desc_extra *dxp = NULL;
285 
286 	if (unlikely(num == 0))
287 		return;
288 
289 	for (i = 0; i < num; i++) {
290 		struct vring_used_elem *uep;
291 
292 		used_idx = vq->vq_used_cons_idx & (vq->vq_nentries - 1);
293 		uep = &vq->vq_ring.used->ring[used_idx];
294 		desc_idx = (uint16_t)uep->id;
295 
296 		dxp = &vq->vq_descx[desc_idx];
297 		vq->vq_used_cons_idx++;
298 
299 		if (dxp->cookie != NULL) {
300 			rte_pktmbuf_free(dxp->cookie);
301 			dxp->cookie = NULL;
302 		}
303 	}
304 
305 	last_idx = desc_idx + dxp->ndescs - 1;
306 	free_cnt = last_idx - vq->vq_desc_tail_idx;
307 	if (free_cnt <= 0)
308 		free_cnt += vq->vq_nentries;
309 
310 	vq_ring_free_inorder(vq, last_idx, free_cnt);
311 }
312 
313 static inline int
314 virtqueue_enqueue_refill_inorder(struct virtqueue *vq,
315 			struct rte_mbuf **cookies,
316 			uint16_t num)
317 {
318 	struct vq_desc_extra *dxp;
319 	struct virtio_hw *hw = vq->hw;
320 	struct vring_desc *start_dp;
321 	uint16_t head_idx, idx, i = 0;
322 
323 	if (unlikely(vq->vq_free_cnt == 0))
324 		return -ENOSPC;
325 	if (unlikely(vq->vq_free_cnt < num))
326 		return -EMSGSIZE;
327 
328 	head_idx = vq->vq_desc_head_idx & (vq->vq_nentries - 1);
329 	start_dp = vq->vq_ring.desc;
330 
331 	while (i < num) {
332 		idx = head_idx & (vq->vq_nentries - 1);
333 		dxp = &vq->vq_descx[idx];
334 		dxp->cookie = (void *)cookies[i];
335 		dxp->ndescs = 1;
336 
337 		start_dp[idx].addr =
338 				VIRTIO_MBUF_ADDR(cookies[i], vq) +
339 				RTE_PKTMBUF_HEADROOM - hw->vtnet_hdr_size;
340 		start_dp[idx].len =
341 				cookies[i]->buf_len -
342 				RTE_PKTMBUF_HEADROOM +
343 				hw->vtnet_hdr_size;
344 		start_dp[idx].flags =  VRING_DESC_F_WRITE;
345 
346 		vq_update_avail_ring(vq, idx);
347 		head_idx++;
348 		i++;
349 	}
350 
351 	vq->vq_desc_head_idx += num;
352 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - num);
353 	return 0;
354 }
355 
356 static inline int
357 virtqueue_enqueue_recv_refill(struct virtqueue *vq, struct rte_mbuf **cookie,
358 				uint16_t num)
359 {
360 	struct vq_desc_extra *dxp;
361 	struct virtio_hw *hw = vq->hw;
362 	struct vring_desc *start_dp = vq->vq_ring.desc;
363 	uint16_t idx, i;
364 
365 	if (unlikely(vq->vq_free_cnt == 0))
366 		return -ENOSPC;
367 	if (unlikely(vq->vq_free_cnt < num))
368 		return -EMSGSIZE;
369 
370 	if (unlikely(vq->vq_desc_head_idx >= vq->vq_nentries))
371 		return -EFAULT;
372 
373 	for (i = 0; i < num; i++) {
374 		idx = vq->vq_desc_head_idx;
375 		dxp = &vq->vq_descx[idx];
376 		dxp->cookie = (void *)cookie[i];
377 		dxp->ndescs = 1;
378 
379 		start_dp[idx].addr =
380 			VIRTIO_MBUF_ADDR(cookie[i], vq) +
381 			RTE_PKTMBUF_HEADROOM - hw->vtnet_hdr_size;
382 		start_dp[idx].len =
383 			cookie[i]->buf_len - RTE_PKTMBUF_HEADROOM +
384 			hw->vtnet_hdr_size;
385 		start_dp[idx].flags = VRING_DESC_F_WRITE;
386 		vq->vq_desc_head_idx = start_dp[idx].next;
387 		vq_update_avail_ring(vq, idx);
388 		if (vq->vq_desc_head_idx == VQ_RING_DESC_CHAIN_END) {
389 			vq->vq_desc_tail_idx = vq->vq_desc_head_idx;
390 			break;
391 		}
392 	}
393 
394 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - num);
395 
396 	return 0;
397 }
398 
399 static inline int
400 virtqueue_enqueue_recv_refill_packed(struct virtqueue *vq,
401 				     struct rte_mbuf **cookie, uint16_t num)
402 {
403 	struct vring_packed_desc *start_dp = vq->ring_packed.desc_packed;
404 	uint16_t flags = VRING_DESC_F_WRITE | vq->avail_used_flags;
405 	struct virtio_hw *hw = vq->hw;
406 	struct vq_desc_extra *dxp;
407 	uint16_t idx;
408 	int i;
409 
410 	if (unlikely(vq->vq_free_cnt == 0))
411 		return -ENOSPC;
412 	if (unlikely(vq->vq_free_cnt < num))
413 		return -EMSGSIZE;
414 
415 	for (i = 0; i < num; i++) {
416 		idx = vq->vq_avail_idx;
417 		dxp = &vq->vq_descx[idx];
418 		dxp->cookie = (void *)cookie[i];
419 		dxp->ndescs = 1;
420 
421 		start_dp[idx].addr = VIRTIO_MBUF_ADDR(cookie[i], vq) +
422 				RTE_PKTMBUF_HEADROOM - hw->vtnet_hdr_size;
423 		start_dp[idx].len = cookie[i]->buf_len - RTE_PKTMBUF_HEADROOM
424 					+ hw->vtnet_hdr_size;
425 
426 		vq->vq_desc_head_idx = dxp->next;
427 		if (vq->vq_desc_head_idx == VQ_RING_DESC_CHAIN_END)
428 			vq->vq_desc_tail_idx = vq->vq_desc_head_idx;
429 		virtio_wmb(hw->weak_barriers);
430 		start_dp[idx].flags = flags;
431 		if (++vq->vq_avail_idx >= vq->vq_nentries) {
432 			vq->vq_avail_idx -= vq->vq_nentries;
433 			vq->avail_wrap_counter ^= 1;
434 			vq->avail_used_flags =
435 				VRING_DESC_F_AVAIL(vq->avail_wrap_counter) |
436 				VRING_DESC_F_USED(!vq->avail_wrap_counter);
437 			flags = VRING_DESC_F_WRITE | vq->avail_used_flags;
438 		}
439 	}
440 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - num);
441 	return 0;
442 }
443 
444 /* When doing TSO, the IP length is not included in the pseudo header
445  * checksum of the packet given to the PMD, but for virtio it is
446  * expected.
447  */
448 static void
449 virtio_tso_fix_cksum(struct rte_mbuf *m)
450 {
451 	/* common case: header is not fragmented */
452 	if (likely(rte_pktmbuf_data_len(m) >= m->l2_len + m->l3_len +
453 			m->l4_len)) {
454 		struct ipv4_hdr *iph;
455 		struct ipv6_hdr *ip6h;
456 		struct tcp_hdr *th;
457 		uint16_t prev_cksum, new_cksum, ip_len, ip_paylen;
458 		uint32_t tmp;
459 
460 		iph = rte_pktmbuf_mtod_offset(m, struct ipv4_hdr *, m->l2_len);
461 		th = RTE_PTR_ADD(iph, m->l3_len);
462 		if ((iph->version_ihl >> 4) == 4) {
463 			iph->hdr_checksum = 0;
464 			iph->hdr_checksum = rte_ipv4_cksum(iph);
465 			ip_len = iph->total_length;
466 			ip_paylen = rte_cpu_to_be_16(rte_be_to_cpu_16(ip_len) -
467 				m->l3_len);
468 		} else {
469 			ip6h = (struct ipv6_hdr *)iph;
470 			ip_paylen = ip6h->payload_len;
471 		}
472 
473 		/* calculate the new phdr checksum not including ip_paylen */
474 		prev_cksum = th->cksum;
475 		tmp = prev_cksum;
476 		tmp += ip_paylen;
477 		tmp = (tmp & 0xffff) + (tmp >> 16);
478 		new_cksum = tmp;
479 
480 		/* replace it in the packet */
481 		th->cksum = new_cksum;
482 	}
483 }
484 
485 
486 /* avoid write operation when necessary, to lessen cache issues */
487 #define ASSIGN_UNLESS_EQUAL(var, val) do {	\
488 	if ((var) != (val))			\
489 		(var) = (val);			\
490 } while (0)
491 
492 static inline void
493 virtqueue_xmit_offload(struct virtio_net_hdr *hdr,
494 			struct rte_mbuf *cookie,
495 			bool offload)
496 {
497 	if (offload) {
498 		if (cookie->ol_flags & PKT_TX_TCP_SEG)
499 			cookie->ol_flags |= PKT_TX_TCP_CKSUM;
500 
501 		switch (cookie->ol_flags & PKT_TX_L4_MASK) {
502 		case PKT_TX_UDP_CKSUM:
503 			hdr->csum_start = cookie->l2_len + cookie->l3_len;
504 			hdr->csum_offset = offsetof(struct udp_hdr,
505 				dgram_cksum);
506 			hdr->flags = VIRTIO_NET_HDR_F_NEEDS_CSUM;
507 			break;
508 
509 		case PKT_TX_TCP_CKSUM:
510 			hdr->csum_start = cookie->l2_len + cookie->l3_len;
511 			hdr->csum_offset = offsetof(struct tcp_hdr, cksum);
512 			hdr->flags = VIRTIO_NET_HDR_F_NEEDS_CSUM;
513 			break;
514 
515 		default:
516 			ASSIGN_UNLESS_EQUAL(hdr->csum_start, 0);
517 			ASSIGN_UNLESS_EQUAL(hdr->csum_offset, 0);
518 			ASSIGN_UNLESS_EQUAL(hdr->flags, 0);
519 			break;
520 		}
521 
522 		/* TCP Segmentation Offload */
523 		if (cookie->ol_flags & PKT_TX_TCP_SEG) {
524 			virtio_tso_fix_cksum(cookie);
525 			hdr->gso_type = (cookie->ol_flags & PKT_TX_IPV6) ?
526 				VIRTIO_NET_HDR_GSO_TCPV6 :
527 				VIRTIO_NET_HDR_GSO_TCPV4;
528 			hdr->gso_size = cookie->tso_segsz;
529 			hdr->hdr_len =
530 				cookie->l2_len +
531 				cookie->l3_len +
532 				cookie->l4_len;
533 		} else {
534 			ASSIGN_UNLESS_EQUAL(hdr->gso_type, 0);
535 			ASSIGN_UNLESS_EQUAL(hdr->gso_size, 0);
536 			ASSIGN_UNLESS_EQUAL(hdr->hdr_len, 0);
537 		}
538 	}
539 }
540 
541 static inline void
542 virtqueue_enqueue_xmit_inorder(struct virtnet_tx *txvq,
543 			struct rte_mbuf **cookies,
544 			uint16_t num)
545 {
546 	struct vq_desc_extra *dxp;
547 	struct virtqueue *vq = txvq->vq;
548 	struct vring_desc *start_dp;
549 	struct virtio_net_hdr *hdr;
550 	uint16_t idx;
551 	uint16_t head_size = vq->hw->vtnet_hdr_size;
552 	uint16_t i = 0;
553 
554 	idx = vq->vq_desc_head_idx;
555 	start_dp = vq->vq_ring.desc;
556 
557 	while (i < num) {
558 		idx = idx & (vq->vq_nentries - 1);
559 		dxp = &vq->vq_descx[idx];
560 		dxp->cookie = (void *)cookies[i];
561 		dxp->ndescs = 1;
562 
563 		hdr = (struct virtio_net_hdr *)
564 			rte_pktmbuf_prepend(cookies[i], head_size);
565 		cookies[i]->pkt_len -= head_size;
566 
567 		/* if offload disabled, it is not zeroed below, do it now */
568 		if (!vq->hw->has_tx_offload) {
569 			ASSIGN_UNLESS_EQUAL(hdr->csum_start, 0);
570 			ASSIGN_UNLESS_EQUAL(hdr->csum_offset, 0);
571 			ASSIGN_UNLESS_EQUAL(hdr->flags, 0);
572 			ASSIGN_UNLESS_EQUAL(hdr->gso_type, 0);
573 			ASSIGN_UNLESS_EQUAL(hdr->gso_size, 0);
574 			ASSIGN_UNLESS_EQUAL(hdr->hdr_len, 0);
575 		}
576 
577 		virtqueue_xmit_offload(hdr, cookies[i],
578 				vq->hw->has_tx_offload);
579 
580 		start_dp[idx].addr  = VIRTIO_MBUF_DATA_DMA_ADDR(cookies[i], vq);
581 		start_dp[idx].len   = cookies[i]->data_len;
582 		start_dp[idx].flags = 0;
583 
584 		vq_update_avail_ring(vq, idx);
585 
586 		idx++;
587 		i++;
588 	};
589 
590 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - num);
591 	vq->vq_desc_head_idx = idx & (vq->vq_nentries - 1);
592 }
593 
594 static inline void
595 virtqueue_enqueue_xmit_packed(struct virtnet_tx *txvq, struct rte_mbuf *cookie,
596 			      uint16_t needed, int can_push)
597 {
598 	struct virtio_tx_region *txr = txvq->virtio_net_hdr_mz->addr;
599 	struct vq_desc_extra *dxp;
600 	struct virtqueue *vq = txvq->vq;
601 	struct vring_packed_desc *start_dp, *head_dp;
602 	uint16_t idx, id, head_idx, head_flags;
603 	uint16_t head_size = vq->hw->vtnet_hdr_size;
604 	struct virtio_net_hdr *hdr;
605 	uint16_t prev;
606 
607 	id = vq->vq_desc_head_idx;
608 
609 	dxp = &vq->vq_descx[id];
610 	dxp->ndescs = needed;
611 	dxp->cookie = cookie;
612 
613 	head_idx = vq->vq_avail_idx;
614 	idx = head_idx;
615 	prev = head_idx;
616 	start_dp = vq->ring_packed.desc_packed;
617 
618 	head_dp = &vq->ring_packed.desc_packed[idx];
619 	head_flags = cookie->next ? VRING_DESC_F_NEXT : 0;
620 	head_flags |= vq->avail_used_flags;
621 
622 	if (can_push) {
623 		/* prepend cannot fail, checked by caller */
624 		hdr = (struct virtio_net_hdr *)
625 			rte_pktmbuf_prepend(cookie, head_size);
626 		/* rte_pktmbuf_prepend() counts the hdr size to the pkt length,
627 		 * which is wrong. Below subtract restores correct pkt size.
628 		 */
629 		cookie->pkt_len -= head_size;
630 
631 		/* if offload disabled, it is not zeroed below, do it now */
632 		if (!vq->hw->has_tx_offload) {
633 			ASSIGN_UNLESS_EQUAL(hdr->csum_start, 0);
634 			ASSIGN_UNLESS_EQUAL(hdr->csum_offset, 0);
635 			ASSIGN_UNLESS_EQUAL(hdr->flags, 0);
636 			ASSIGN_UNLESS_EQUAL(hdr->gso_type, 0);
637 			ASSIGN_UNLESS_EQUAL(hdr->gso_size, 0);
638 			ASSIGN_UNLESS_EQUAL(hdr->hdr_len, 0);
639 		}
640 	} else {
641 		/* setup first tx ring slot to point to header
642 		 * stored in reserved region.
643 		 */
644 		start_dp[idx].addr  = txvq->virtio_net_hdr_mem +
645 			RTE_PTR_DIFF(&txr[idx].tx_hdr, txr);
646 		start_dp[idx].len   = vq->hw->vtnet_hdr_size;
647 		hdr = (struct virtio_net_hdr *)&txr[idx].tx_hdr;
648 		idx++;
649 		if (idx >= vq->vq_nentries) {
650 			idx -= vq->vq_nentries;
651 			vq->avail_wrap_counter ^= 1;
652 			vq->avail_used_flags =
653 				VRING_DESC_F_AVAIL(vq->avail_wrap_counter) |
654 				VRING_DESC_F_USED(!vq->avail_wrap_counter);
655 		}
656 	}
657 
658 	virtqueue_xmit_offload(hdr, cookie, vq->hw->has_tx_offload);
659 
660 	do {
661 		uint16_t flags;
662 
663 		start_dp[idx].addr = VIRTIO_MBUF_DATA_DMA_ADDR(cookie, vq);
664 		start_dp[idx].len  = cookie->data_len;
665 		if (likely(idx != head_idx)) {
666 			flags = cookie->next ? VRING_DESC_F_NEXT : 0;
667 			flags |= vq->avail_used_flags;
668 			start_dp[idx].flags = flags;
669 		}
670 		prev = idx;
671 		idx++;
672 		if (idx >= vq->vq_nentries) {
673 			idx -= vq->vq_nentries;
674 			vq->avail_wrap_counter ^= 1;
675 			vq->avail_used_flags =
676 				VRING_DESC_F_AVAIL(vq->avail_wrap_counter) |
677 				VRING_DESC_F_USED(!vq->avail_wrap_counter);
678 		}
679 	} while ((cookie = cookie->next) != NULL);
680 
681 	start_dp[prev].id = id;
682 
683 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - needed);
684 
685 	vq->vq_desc_head_idx = dxp->next;
686 	if (vq->vq_desc_head_idx == VQ_RING_DESC_CHAIN_END)
687 		vq->vq_desc_tail_idx = VQ_RING_DESC_CHAIN_END;
688 
689 	vq->vq_avail_idx = idx;
690 
691 	virtio_wmb(vq->hw->weak_barriers);
692 	head_dp->flags = head_flags;
693 }
694 
695 static inline void
696 virtqueue_enqueue_xmit(struct virtnet_tx *txvq, struct rte_mbuf *cookie,
697 			uint16_t needed, int use_indirect, int can_push,
698 			int in_order)
699 {
700 	struct virtio_tx_region *txr = txvq->virtio_net_hdr_mz->addr;
701 	struct vq_desc_extra *dxp;
702 	struct virtqueue *vq = txvq->vq;
703 	struct vring_desc *start_dp;
704 	uint16_t seg_num = cookie->nb_segs;
705 	uint16_t head_idx, idx;
706 	uint16_t head_size = vq->hw->vtnet_hdr_size;
707 	struct virtio_net_hdr *hdr;
708 
709 	head_idx = vq->vq_desc_head_idx;
710 	idx = head_idx;
711 	dxp = &vq->vq_descx[idx];
712 	dxp->cookie = (void *)cookie;
713 	dxp->ndescs = needed;
714 
715 	start_dp = vq->vq_ring.desc;
716 
717 	if (can_push) {
718 		/* prepend cannot fail, checked by caller */
719 		hdr = (struct virtio_net_hdr *)
720 			rte_pktmbuf_prepend(cookie, head_size);
721 		/* rte_pktmbuf_prepend() counts the hdr size to the pkt length,
722 		 * which is wrong. Below subtract restores correct pkt size.
723 		 */
724 		cookie->pkt_len -= head_size;
725 
726 		/* if offload disabled, it is not zeroed below, do it now */
727 		if (!vq->hw->has_tx_offload) {
728 			ASSIGN_UNLESS_EQUAL(hdr->csum_start, 0);
729 			ASSIGN_UNLESS_EQUAL(hdr->csum_offset, 0);
730 			ASSIGN_UNLESS_EQUAL(hdr->flags, 0);
731 			ASSIGN_UNLESS_EQUAL(hdr->gso_type, 0);
732 			ASSIGN_UNLESS_EQUAL(hdr->gso_size, 0);
733 			ASSIGN_UNLESS_EQUAL(hdr->hdr_len, 0);
734 		}
735 	} else if (use_indirect) {
736 		/* setup tx ring slot to point to indirect
737 		 * descriptor list stored in reserved region.
738 		 *
739 		 * the first slot in indirect ring is already preset
740 		 * to point to the header in reserved region
741 		 */
742 		start_dp[idx].addr  = txvq->virtio_net_hdr_mem +
743 			RTE_PTR_DIFF(&txr[idx].tx_indir, txr);
744 		start_dp[idx].len   = (seg_num + 1) * sizeof(struct vring_desc);
745 		start_dp[idx].flags = VRING_DESC_F_INDIRECT;
746 		hdr = (struct virtio_net_hdr *)&txr[idx].tx_hdr;
747 
748 		/* loop below will fill in rest of the indirect elements */
749 		start_dp = txr[idx].tx_indir;
750 		idx = 1;
751 	} else {
752 		/* setup first tx ring slot to point to header
753 		 * stored in reserved region.
754 		 */
755 		start_dp[idx].addr  = txvq->virtio_net_hdr_mem +
756 			RTE_PTR_DIFF(&txr[idx].tx_hdr, txr);
757 		start_dp[idx].len   = vq->hw->vtnet_hdr_size;
758 		start_dp[idx].flags = VRING_DESC_F_NEXT;
759 		hdr = (struct virtio_net_hdr *)&txr[idx].tx_hdr;
760 
761 		idx = start_dp[idx].next;
762 	}
763 
764 	virtqueue_xmit_offload(hdr, cookie, vq->hw->has_tx_offload);
765 
766 	do {
767 		start_dp[idx].addr  = VIRTIO_MBUF_DATA_DMA_ADDR(cookie, vq);
768 		start_dp[idx].len   = cookie->data_len;
769 		start_dp[idx].flags = cookie->next ? VRING_DESC_F_NEXT : 0;
770 		idx = start_dp[idx].next;
771 	} while ((cookie = cookie->next) != NULL);
772 
773 	if (use_indirect)
774 		idx = vq->vq_ring.desc[head_idx].next;
775 
776 	vq->vq_free_cnt = (uint16_t)(vq->vq_free_cnt - needed);
777 
778 	vq->vq_desc_head_idx = idx;
779 	vq_update_avail_ring(vq, head_idx);
780 
781 	if (!in_order) {
782 		if (vq->vq_desc_head_idx == VQ_RING_DESC_CHAIN_END)
783 			vq->vq_desc_tail_idx = idx;
784 	}
785 }
786 
787 void
788 virtio_dev_cq_start(struct rte_eth_dev *dev)
789 {
790 	struct virtio_hw *hw = dev->data->dev_private;
791 
792 	if (hw->cvq && hw->cvq->vq) {
793 		rte_spinlock_init(&hw->cvq->lock);
794 		VIRTQUEUE_DUMP((struct virtqueue *)hw->cvq->vq);
795 	}
796 }
797 
798 int
799 virtio_dev_rx_queue_setup(struct rte_eth_dev *dev,
800 			uint16_t queue_idx,
801 			uint16_t nb_desc,
802 			unsigned int socket_id __rte_unused,
803 			const struct rte_eth_rxconf *rx_conf __rte_unused,
804 			struct rte_mempool *mp)
805 {
806 	uint16_t vtpci_queue_idx = 2 * queue_idx + VTNET_SQ_RQ_QUEUE_IDX;
807 	struct virtio_hw *hw = dev->data->dev_private;
808 	struct virtqueue *vq = hw->vqs[vtpci_queue_idx];
809 	struct virtnet_rx *rxvq;
810 
811 	PMD_INIT_FUNC_TRACE();
812 
813 	if (nb_desc == 0 || nb_desc > vq->vq_nentries)
814 		nb_desc = vq->vq_nentries;
815 	vq->vq_free_cnt = RTE_MIN(vq->vq_free_cnt, nb_desc);
816 
817 	rxvq = &vq->rxq;
818 	rxvq->queue_id = queue_idx;
819 	rxvq->mpool = mp;
820 	if (rxvq->mpool == NULL) {
821 		rte_exit(EXIT_FAILURE,
822 			"Cannot allocate mbufs for rx virtqueue");
823 	}
824 
825 	dev->data->rx_queues[queue_idx] = rxvq;
826 
827 	return 0;
828 }
829 
830 int
831 virtio_dev_rx_queue_setup_finish(struct rte_eth_dev *dev, uint16_t queue_idx)
832 {
833 	uint16_t vtpci_queue_idx = 2 * queue_idx + VTNET_SQ_RQ_QUEUE_IDX;
834 	struct virtio_hw *hw = dev->data->dev_private;
835 	struct virtqueue *vq = hw->vqs[vtpci_queue_idx];
836 	struct virtnet_rx *rxvq = &vq->rxq;
837 	struct rte_mbuf *m;
838 	uint16_t desc_idx;
839 	int error, nbufs, i;
840 
841 	PMD_INIT_FUNC_TRACE();
842 
843 	/* Allocate blank mbufs for the each rx descriptor */
844 	nbufs = 0;
845 
846 	if (hw->use_simple_rx) {
847 		for (desc_idx = 0; desc_idx < vq->vq_nentries;
848 		     desc_idx++) {
849 			vq->vq_ring.avail->ring[desc_idx] = desc_idx;
850 			vq->vq_ring.desc[desc_idx].flags =
851 				VRING_DESC_F_WRITE;
852 		}
853 
854 		virtio_rxq_vec_setup(rxvq);
855 	}
856 
857 	memset(&rxvq->fake_mbuf, 0, sizeof(rxvq->fake_mbuf));
858 	for (desc_idx = 0; desc_idx < RTE_PMD_VIRTIO_RX_MAX_BURST;
859 	     desc_idx++) {
860 		vq->sw_ring[vq->vq_nentries + desc_idx] =
861 			&rxvq->fake_mbuf;
862 	}
863 
864 	if (hw->use_simple_rx) {
865 		while (vq->vq_free_cnt >= RTE_VIRTIO_VPMD_RX_REARM_THRESH) {
866 			virtio_rxq_rearm_vec(rxvq);
867 			nbufs += RTE_VIRTIO_VPMD_RX_REARM_THRESH;
868 		}
869 	} else if (hw->use_inorder_rx) {
870 		if ((!virtqueue_full(vq))) {
871 			uint16_t free_cnt = vq->vq_free_cnt;
872 			struct rte_mbuf *pkts[free_cnt];
873 
874 			if (!rte_pktmbuf_alloc_bulk(rxvq->mpool, pkts,
875 				free_cnt)) {
876 				error = virtqueue_enqueue_refill_inorder(vq,
877 						pkts,
878 						free_cnt);
879 				if (unlikely(error)) {
880 					for (i = 0; i < free_cnt; i++)
881 						rte_pktmbuf_free(pkts[i]);
882 				}
883 			}
884 
885 			nbufs += free_cnt;
886 			vq_update_avail_idx(vq);
887 		}
888 	} else {
889 		while (!virtqueue_full(vq)) {
890 			m = rte_mbuf_raw_alloc(rxvq->mpool);
891 			if (m == NULL)
892 				break;
893 
894 			/* Enqueue allocated buffers */
895 			if (vtpci_packed_queue(vq->hw))
896 				error = virtqueue_enqueue_recv_refill_packed(vq,
897 						&m, 1);
898 			else
899 				error = virtqueue_enqueue_recv_refill(vq,
900 						&m, 1);
901 			if (error) {
902 				rte_pktmbuf_free(m);
903 				break;
904 			}
905 			nbufs++;
906 		}
907 
908 		if (!vtpci_packed_queue(vq->hw))
909 			vq_update_avail_idx(vq);
910 	}
911 
912 	PMD_INIT_LOG(DEBUG, "Allocated %d bufs", nbufs);
913 
914 	VIRTQUEUE_DUMP(vq);
915 
916 	return 0;
917 }
918 
919 /*
920  * struct rte_eth_dev *dev: Used to update dev
921  * uint16_t nb_desc: Defaults to values read from config space
922  * unsigned int socket_id: Used to allocate memzone
923  * const struct rte_eth_txconf *tx_conf: Used to setup tx engine
924  * uint16_t queue_idx: Just used as an index in dev txq list
925  */
926 int
927 virtio_dev_tx_queue_setup(struct rte_eth_dev *dev,
928 			uint16_t queue_idx,
929 			uint16_t nb_desc,
930 			unsigned int socket_id __rte_unused,
931 			const struct rte_eth_txconf *tx_conf)
932 {
933 	uint8_t vtpci_queue_idx = 2 * queue_idx + VTNET_SQ_TQ_QUEUE_IDX;
934 	struct virtio_hw *hw = dev->data->dev_private;
935 	struct virtqueue *vq = hw->vqs[vtpci_queue_idx];
936 	struct virtnet_tx *txvq;
937 	uint16_t tx_free_thresh;
938 
939 	PMD_INIT_FUNC_TRACE();
940 
941 	if (nb_desc == 0 || nb_desc > vq->vq_nentries)
942 		nb_desc = vq->vq_nentries;
943 	vq->vq_free_cnt = RTE_MIN(vq->vq_free_cnt, nb_desc);
944 
945 	txvq = &vq->txq;
946 	txvq->queue_id = queue_idx;
947 
948 	tx_free_thresh = tx_conf->tx_free_thresh;
949 	if (tx_free_thresh == 0)
950 		tx_free_thresh =
951 			RTE_MIN(vq->vq_nentries / 4, DEFAULT_TX_FREE_THRESH);
952 
953 	if (tx_free_thresh >= (vq->vq_nentries - 3)) {
954 		RTE_LOG(ERR, PMD, "tx_free_thresh must be less than the "
955 			"number of TX entries minus 3 (%u)."
956 			" (tx_free_thresh=%u port=%u queue=%u)\n",
957 			vq->vq_nentries - 3,
958 			tx_free_thresh, dev->data->port_id, queue_idx);
959 		return -EINVAL;
960 	}
961 
962 	vq->vq_free_thresh = tx_free_thresh;
963 
964 	dev->data->tx_queues[queue_idx] = txvq;
965 	return 0;
966 }
967 
968 int
969 virtio_dev_tx_queue_setup_finish(struct rte_eth_dev *dev,
970 				uint16_t queue_idx)
971 {
972 	uint8_t vtpci_queue_idx = 2 * queue_idx + VTNET_SQ_TQ_QUEUE_IDX;
973 	struct virtio_hw *hw = dev->data->dev_private;
974 	struct virtqueue *vq = hw->vqs[vtpci_queue_idx];
975 
976 	PMD_INIT_FUNC_TRACE();
977 
978 	if (!vtpci_packed_queue(hw)) {
979 		if (hw->use_inorder_tx)
980 			vq->vq_ring.desc[vq->vq_nentries - 1].next = 0;
981 	}
982 
983 	VIRTQUEUE_DUMP(vq);
984 
985 	return 0;
986 }
987 
988 static inline void
989 virtio_discard_rxbuf(struct virtqueue *vq, struct rte_mbuf *m)
990 {
991 	int error;
992 	/*
993 	 * Requeue the discarded mbuf. This should always be
994 	 * successful since it was just dequeued.
995 	 */
996 	if (vtpci_packed_queue(vq->hw))
997 		error = virtqueue_enqueue_recv_refill_packed(vq, &m, 1);
998 	else
999 		error = virtqueue_enqueue_recv_refill(vq, &m, 1);
1000 
1001 	if (unlikely(error)) {
1002 		RTE_LOG(ERR, PMD, "cannot requeue discarded mbuf");
1003 		rte_pktmbuf_free(m);
1004 	}
1005 }
1006 
1007 static inline void
1008 virtio_discard_rxbuf_inorder(struct virtqueue *vq, struct rte_mbuf *m)
1009 {
1010 	int error;
1011 
1012 	error = virtqueue_enqueue_refill_inorder(vq, &m, 1);
1013 	if (unlikely(error)) {
1014 		RTE_LOG(ERR, PMD, "cannot requeue discarded mbuf");
1015 		rte_pktmbuf_free(m);
1016 	}
1017 }
1018 
1019 static inline void
1020 virtio_update_packet_stats(struct virtnet_stats *stats, struct rte_mbuf *mbuf)
1021 {
1022 	uint32_t s = mbuf->pkt_len;
1023 	struct ether_addr *ea;
1024 
1025 	stats->bytes += s;
1026 
1027 	if (s == 64) {
1028 		stats->size_bins[1]++;
1029 	} else if (s > 64 && s < 1024) {
1030 		uint32_t bin;
1031 
1032 		/* count zeros, and offset into correct bin */
1033 		bin = (sizeof(s) * 8) - __builtin_clz(s) - 5;
1034 		stats->size_bins[bin]++;
1035 	} else {
1036 		if (s < 64)
1037 			stats->size_bins[0]++;
1038 		else if (s < 1519)
1039 			stats->size_bins[6]++;
1040 		else if (s >= 1519)
1041 			stats->size_bins[7]++;
1042 	}
1043 
1044 	ea = rte_pktmbuf_mtod(mbuf, struct ether_addr *);
1045 	if (is_multicast_ether_addr(ea)) {
1046 		if (is_broadcast_ether_addr(ea))
1047 			stats->broadcast++;
1048 		else
1049 			stats->multicast++;
1050 	}
1051 }
1052 
1053 static inline void
1054 virtio_rx_stats_updated(struct virtnet_rx *rxvq, struct rte_mbuf *m)
1055 {
1056 	VIRTIO_DUMP_PACKET(m, m->data_len);
1057 
1058 	virtio_update_packet_stats(&rxvq->stats, m);
1059 }
1060 
1061 /* Optionally fill offload information in structure */
1062 static inline int
1063 virtio_rx_offload(struct rte_mbuf *m, struct virtio_net_hdr *hdr)
1064 {
1065 	struct rte_net_hdr_lens hdr_lens;
1066 	uint32_t hdrlen, ptype;
1067 	int l4_supported = 0;
1068 
1069 	/* nothing to do */
1070 	if (hdr->flags == 0 && hdr->gso_type == VIRTIO_NET_HDR_GSO_NONE)
1071 		return 0;
1072 
1073 	m->ol_flags |= PKT_RX_IP_CKSUM_UNKNOWN;
1074 
1075 	ptype = rte_net_get_ptype(m, &hdr_lens, RTE_PTYPE_ALL_MASK);
1076 	m->packet_type = ptype;
1077 	if ((ptype & RTE_PTYPE_L4_MASK) == RTE_PTYPE_L4_TCP ||
1078 	    (ptype & RTE_PTYPE_L4_MASK) == RTE_PTYPE_L4_UDP ||
1079 	    (ptype & RTE_PTYPE_L4_MASK) == RTE_PTYPE_L4_SCTP)
1080 		l4_supported = 1;
1081 
1082 	if (hdr->flags & VIRTIO_NET_HDR_F_NEEDS_CSUM) {
1083 		hdrlen = hdr_lens.l2_len + hdr_lens.l3_len + hdr_lens.l4_len;
1084 		if (hdr->csum_start <= hdrlen && l4_supported) {
1085 			m->ol_flags |= PKT_RX_L4_CKSUM_NONE;
1086 		} else {
1087 			/* Unknown proto or tunnel, do sw cksum. We can assume
1088 			 * the cksum field is in the first segment since the
1089 			 * buffers we provided to the host are large enough.
1090 			 * In case of SCTP, this will be wrong since it's a CRC
1091 			 * but there's nothing we can do.
1092 			 */
1093 			uint16_t csum = 0, off;
1094 
1095 			rte_raw_cksum_mbuf(m, hdr->csum_start,
1096 				rte_pktmbuf_pkt_len(m) - hdr->csum_start,
1097 				&csum);
1098 			if (likely(csum != 0xffff))
1099 				csum = ~csum;
1100 			off = hdr->csum_offset + hdr->csum_start;
1101 			if (rte_pktmbuf_data_len(m) >= off + 1)
1102 				*rte_pktmbuf_mtod_offset(m, uint16_t *,
1103 					off) = csum;
1104 		}
1105 	} else if (hdr->flags & VIRTIO_NET_HDR_F_DATA_VALID && l4_supported) {
1106 		m->ol_flags |= PKT_RX_L4_CKSUM_GOOD;
1107 	}
1108 
1109 	/* GSO request, save required information in mbuf */
1110 	if (hdr->gso_type != VIRTIO_NET_HDR_GSO_NONE) {
1111 		/* Check unsupported modes */
1112 		if ((hdr->gso_type & VIRTIO_NET_HDR_GSO_ECN) ||
1113 		    (hdr->gso_size == 0)) {
1114 			return -EINVAL;
1115 		}
1116 
1117 		/* Update mss lengthes in mbuf */
1118 		m->tso_segsz = hdr->gso_size;
1119 		switch (hdr->gso_type & ~VIRTIO_NET_HDR_GSO_ECN) {
1120 			case VIRTIO_NET_HDR_GSO_TCPV4:
1121 			case VIRTIO_NET_HDR_GSO_TCPV6:
1122 				m->ol_flags |= PKT_RX_LRO | \
1123 					PKT_RX_L4_CKSUM_NONE;
1124 				break;
1125 			default:
1126 				return -EINVAL;
1127 		}
1128 	}
1129 
1130 	return 0;
1131 }
1132 
1133 #define VIRTIO_MBUF_BURST_SZ 64
1134 #define DESC_PER_CACHELINE (RTE_CACHE_LINE_SIZE / sizeof(struct vring_desc))
1135 uint16_t
1136 virtio_recv_pkts(void *rx_queue, struct rte_mbuf **rx_pkts, uint16_t nb_pkts)
1137 {
1138 	struct virtnet_rx *rxvq = rx_queue;
1139 	struct virtqueue *vq = rxvq->vq;
1140 	struct virtio_hw *hw = vq->hw;
1141 	struct rte_mbuf *rxm, *new_mbuf;
1142 	uint16_t nb_used, num, nb_rx;
1143 	uint32_t len[VIRTIO_MBUF_BURST_SZ];
1144 	struct rte_mbuf *rcv_pkts[VIRTIO_MBUF_BURST_SZ];
1145 	int error;
1146 	uint32_t i, nb_enqueued;
1147 	uint32_t hdr_size;
1148 	struct virtio_net_hdr *hdr;
1149 
1150 	nb_rx = 0;
1151 	if (unlikely(hw->started == 0))
1152 		return nb_rx;
1153 
1154 	nb_used = VIRTQUEUE_NUSED(vq);
1155 
1156 	virtio_rmb(hw->weak_barriers);
1157 
1158 	num = likely(nb_used <= nb_pkts) ? nb_used : nb_pkts;
1159 	if (unlikely(num > VIRTIO_MBUF_BURST_SZ))
1160 		num = VIRTIO_MBUF_BURST_SZ;
1161 	if (likely(num > DESC_PER_CACHELINE))
1162 		num = num - ((vq->vq_used_cons_idx + num) % DESC_PER_CACHELINE);
1163 
1164 	num = virtqueue_dequeue_burst_rx(vq, rcv_pkts, len, num);
1165 	PMD_RX_LOG(DEBUG, "used:%d dequeue:%d", nb_used, num);
1166 
1167 	nb_enqueued = 0;
1168 	hdr_size = hw->vtnet_hdr_size;
1169 
1170 	for (i = 0; i < num ; i++) {
1171 		rxm = rcv_pkts[i];
1172 
1173 		PMD_RX_LOG(DEBUG, "packet len:%d", len[i]);
1174 
1175 		if (unlikely(len[i] < hdr_size + ETHER_HDR_LEN)) {
1176 			PMD_RX_LOG(ERR, "Packet drop");
1177 			nb_enqueued++;
1178 			virtio_discard_rxbuf(vq, rxm);
1179 			rxvq->stats.errors++;
1180 			continue;
1181 		}
1182 
1183 		rxm->port = rxvq->port_id;
1184 		rxm->data_off = RTE_PKTMBUF_HEADROOM;
1185 		rxm->ol_flags = 0;
1186 		rxm->vlan_tci = 0;
1187 
1188 		rxm->pkt_len = (uint32_t)(len[i] - hdr_size);
1189 		rxm->data_len = (uint16_t)(len[i] - hdr_size);
1190 
1191 		hdr = (struct virtio_net_hdr *)((char *)rxm->buf_addr +
1192 			RTE_PKTMBUF_HEADROOM - hdr_size);
1193 
1194 		if (hw->vlan_strip)
1195 			rte_vlan_strip(rxm);
1196 
1197 		if (hw->has_rx_offload && virtio_rx_offload(rxm, hdr) < 0) {
1198 			virtio_discard_rxbuf(vq, rxm);
1199 			rxvq->stats.errors++;
1200 			continue;
1201 		}
1202 
1203 		virtio_rx_stats_updated(rxvq, rxm);
1204 
1205 		rx_pkts[nb_rx++] = rxm;
1206 	}
1207 
1208 	rxvq->stats.packets += nb_rx;
1209 
1210 	/* Allocate new mbuf for the used descriptor */
1211 	while (likely(!virtqueue_full(vq))) {
1212 		new_mbuf = rte_mbuf_raw_alloc(rxvq->mpool);
1213 		if (unlikely(new_mbuf == NULL)) {
1214 			struct rte_eth_dev *dev
1215 				= &rte_eth_devices[rxvq->port_id];
1216 			dev->data->rx_mbuf_alloc_failed++;
1217 			break;
1218 		}
1219 		error = virtqueue_enqueue_recv_refill(vq, &new_mbuf, 1);
1220 		if (unlikely(error)) {
1221 			rte_pktmbuf_free(new_mbuf);
1222 			break;
1223 		}
1224 		nb_enqueued++;
1225 	}
1226 
1227 	if (likely(nb_enqueued)) {
1228 		vq_update_avail_idx(vq);
1229 
1230 		if (unlikely(virtqueue_kick_prepare(vq))) {
1231 			virtqueue_notify(vq);
1232 			PMD_RX_LOG(DEBUG, "Notified");
1233 		}
1234 	}
1235 
1236 	return nb_rx;
1237 }
1238 
1239 uint16_t
1240 virtio_recv_pkts_packed(void *rx_queue, struct rte_mbuf **rx_pkts,
1241 			uint16_t nb_pkts)
1242 {
1243 	struct virtnet_rx *rxvq = rx_queue;
1244 	struct virtqueue *vq = rxvq->vq;
1245 	struct virtio_hw *hw = vq->hw;
1246 	struct rte_mbuf *rxm, *new_mbuf;
1247 	uint16_t num, nb_rx;
1248 	uint32_t len[VIRTIO_MBUF_BURST_SZ];
1249 	struct rte_mbuf *rcv_pkts[VIRTIO_MBUF_BURST_SZ];
1250 	int error;
1251 	uint32_t i, nb_enqueued;
1252 	uint32_t hdr_size;
1253 	struct virtio_net_hdr *hdr;
1254 
1255 	nb_rx = 0;
1256 	if (unlikely(hw->started == 0))
1257 		return nb_rx;
1258 
1259 	num = RTE_MIN(VIRTIO_MBUF_BURST_SZ, nb_pkts);
1260 	if (likely(num > DESC_PER_CACHELINE))
1261 		num = num - ((vq->vq_used_cons_idx + num) % DESC_PER_CACHELINE);
1262 
1263 	num = virtqueue_dequeue_burst_rx_packed(vq, rcv_pkts, len, num);
1264 	PMD_RX_LOG(DEBUG, "dequeue:%d", num);
1265 
1266 	nb_enqueued = 0;
1267 	hdr_size = hw->vtnet_hdr_size;
1268 
1269 	for (i = 0; i < num; i++) {
1270 		rxm = rcv_pkts[i];
1271 
1272 		PMD_RX_LOG(DEBUG, "packet len:%d", len[i]);
1273 
1274 		if (unlikely(len[i] < hdr_size + ETHER_HDR_LEN)) {
1275 			PMD_RX_LOG(ERR, "Packet drop");
1276 			nb_enqueued++;
1277 			virtio_discard_rxbuf(vq, rxm);
1278 			rxvq->stats.errors++;
1279 			continue;
1280 		}
1281 
1282 		rxm->port = rxvq->port_id;
1283 		rxm->data_off = RTE_PKTMBUF_HEADROOM;
1284 		rxm->ol_flags = 0;
1285 		rxm->vlan_tci = 0;
1286 
1287 		rxm->pkt_len = (uint32_t)(len[i] - hdr_size);
1288 		rxm->data_len = (uint16_t)(len[i] - hdr_size);
1289 
1290 		hdr = (struct virtio_net_hdr *)((char *)rxm->buf_addr +
1291 			RTE_PKTMBUF_HEADROOM - hdr_size);
1292 
1293 		if (hw->vlan_strip)
1294 			rte_vlan_strip(rxm);
1295 
1296 		if (hw->has_rx_offload && virtio_rx_offload(rxm, hdr) < 0) {
1297 			virtio_discard_rxbuf(vq, rxm);
1298 			rxvq->stats.errors++;
1299 			continue;
1300 		}
1301 
1302 		virtio_rx_stats_updated(rxvq, rxm);
1303 
1304 		rx_pkts[nb_rx++] = rxm;
1305 	}
1306 
1307 	rxvq->stats.packets += nb_rx;
1308 
1309 	/* Allocate new mbuf for the used descriptor */
1310 	while (likely(!virtqueue_full(vq))) {
1311 		new_mbuf = rte_mbuf_raw_alloc(rxvq->mpool);
1312 		if (unlikely(new_mbuf == NULL)) {
1313 			struct rte_eth_dev *dev =
1314 				&rte_eth_devices[rxvq->port_id];
1315 			dev->data->rx_mbuf_alloc_failed++;
1316 			break;
1317 		}
1318 		error = virtqueue_enqueue_recv_refill_packed(vq, &new_mbuf, 1);
1319 		if (unlikely(error)) {
1320 			rte_pktmbuf_free(new_mbuf);
1321 			break;
1322 		}
1323 		nb_enqueued++;
1324 	}
1325 
1326 	if (likely(nb_enqueued)) {
1327 		if (unlikely(virtqueue_kick_prepare_packed(vq))) {
1328 			virtqueue_notify(vq);
1329 			PMD_RX_LOG(DEBUG, "Notified");
1330 		}
1331 	}
1332 
1333 	return nb_rx;
1334 }
1335 
1336 
1337 uint16_t
1338 virtio_recv_pkts_inorder(void *rx_queue,
1339 			struct rte_mbuf **rx_pkts,
1340 			uint16_t nb_pkts)
1341 {
1342 	struct virtnet_rx *rxvq = rx_queue;
1343 	struct virtqueue *vq = rxvq->vq;
1344 	struct virtio_hw *hw = vq->hw;
1345 	struct rte_mbuf *rxm;
1346 	struct rte_mbuf *prev;
1347 	uint16_t nb_used, num, nb_rx;
1348 	uint32_t len[VIRTIO_MBUF_BURST_SZ];
1349 	struct rte_mbuf *rcv_pkts[VIRTIO_MBUF_BURST_SZ];
1350 	int error;
1351 	uint32_t nb_enqueued;
1352 	uint32_t seg_num;
1353 	uint32_t seg_res;
1354 	uint32_t hdr_size;
1355 	int32_t i;
1356 
1357 	nb_rx = 0;
1358 	if (unlikely(hw->started == 0))
1359 		return nb_rx;
1360 
1361 	nb_used = VIRTQUEUE_NUSED(vq);
1362 	nb_used = RTE_MIN(nb_used, nb_pkts);
1363 	nb_used = RTE_MIN(nb_used, VIRTIO_MBUF_BURST_SZ);
1364 
1365 	virtio_rmb(hw->weak_barriers);
1366 
1367 	PMD_RX_LOG(DEBUG, "used:%d", nb_used);
1368 
1369 	nb_enqueued = 0;
1370 	seg_num = 1;
1371 	seg_res = 0;
1372 	hdr_size = hw->vtnet_hdr_size;
1373 
1374 	num = virtqueue_dequeue_rx_inorder(vq, rcv_pkts, len, nb_used);
1375 
1376 	for (i = 0; i < num; i++) {
1377 		struct virtio_net_hdr_mrg_rxbuf *header;
1378 
1379 		PMD_RX_LOG(DEBUG, "dequeue:%d", num);
1380 		PMD_RX_LOG(DEBUG, "packet len:%d", len[i]);
1381 
1382 		rxm = rcv_pkts[i];
1383 
1384 		if (unlikely(len[i] < hdr_size + ETHER_HDR_LEN)) {
1385 			PMD_RX_LOG(ERR, "Packet drop");
1386 			nb_enqueued++;
1387 			virtio_discard_rxbuf_inorder(vq, rxm);
1388 			rxvq->stats.errors++;
1389 			continue;
1390 		}
1391 
1392 		header = (struct virtio_net_hdr_mrg_rxbuf *)
1393 			 ((char *)rxm->buf_addr + RTE_PKTMBUF_HEADROOM
1394 			 - hdr_size);
1395 
1396 		if (vtpci_with_feature(hw, VIRTIO_NET_F_MRG_RXBUF)) {
1397 			seg_num = header->num_buffers;
1398 			if (seg_num == 0)
1399 				seg_num = 1;
1400 		} else {
1401 			seg_num = 1;
1402 		}
1403 
1404 		rxm->data_off = RTE_PKTMBUF_HEADROOM;
1405 		rxm->nb_segs = seg_num;
1406 		rxm->ol_flags = 0;
1407 		rxm->vlan_tci = 0;
1408 		rxm->pkt_len = (uint32_t)(len[i] - hdr_size);
1409 		rxm->data_len = (uint16_t)(len[i] - hdr_size);
1410 
1411 		rxm->port = rxvq->port_id;
1412 
1413 		rx_pkts[nb_rx] = rxm;
1414 		prev = rxm;
1415 
1416 		if (vq->hw->has_rx_offload &&
1417 				virtio_rx_offload(rxm, &header->hdr) < 0) {
1418 			virtio_discard_rxbuf_inorder(vq, rxm);
1419 			rxvq->stats.errors++;
1420 			continue;
1421 		}
1422 
1423 		if (hw->vlan_strip)
1424 			rte_vlan_strip(rx_pkts[nb_rx]);
1425 
1426 		seg_res = seg_num - 1;
1427 
1428 		/* Merge remaining segments */
1429 		while (seg_res != 0 && i < (num - 1)) {
1430 			i++;
1431 
1432 			rxm = rcv_pkts[i];
1433 			rxm->data_off = RTE_PKTMBUF_HEADROOM - hdr_size;
1434 			rxm->pkt_len = (uint32_t)(len[i]);
1435 			rxm->data_len = (uint16_t)(len[i]);
1436 
1437 			rx_pkts[nb_rx]->pkt_len += (uint32_t)(len[i]);
1438 			rx_pkts[nb_rx]->data_len += (uint16_t)(len[i]);
1439 
1440 			if (prev)
1441 				prev->next = rxm;
1442 
1443 			prev = rxm;
1444 			seg_res -= 1;
1445 		}
1446 
1447 		if (!seg_res) {
1448 			virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1449 			nb_rx++;
1450 		}
1451 	}
1452 
1453 	/* Last packet still need merge segments */
1454 	while (seg_res != 0) {
1455 		uint16_t rcv_cnt = RTE_MIN((uint16_t)seg_res,
1456 					VIRTIO_MBUF_BURST_SZ);
1457 
1458 		prev = rcv_pkts[nb_rx];
1459 		if (likely(VIRTQUEUE_NUSED(vq) >= rcv_cnt)) {
1460 			virtio_rmb(hw->weak_barriers);
1461 			num = virtqueue_dequeue_rx_inorder(vq, rcv_pkts, len,
1462 							   rcv_cnt);
1463 			uint16_t extra_idx = 0;
1464 
1465 			rcv_cnt = num;
1466 			while (extra_idx < rcv_cnt) {
1467 				rxm = rcv_pkts[extra_idx];
1468 				rxm->data_off =
1469 					RTE_PKTMBUF_HEADROOM - hdr_size;
1470 				rxm->pkt_len = (uint32_t)(len[extra_idx]);
1471 				rxm->data_len = (uint16_t)(len[extra_idx]);
1472 				prev->next = rxm;
1473 				prev = rxm;
1474 				rx_pkts[nb_rx]->pkt_len += len[extra_idx];
1475 				rx_pkts[nb_rx]->data_len += len[extra_idx];
1476 				extra_idx += 1;
1477 			};
1478 			seg_res -= rcv_cnt;
1479 
1480 			if (!seg_res) {
1481 				virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1482 				nb_rx++;
1483 			}
1484 		} else {
1485 			PMD_RX_LOG(ERR,
1486 					"No enough segments for packet.");
1487 			virtio_discard_rxbuf_inorder(vq, prev);
1488 			rxvq->stats.errors++;
1489 			break;
1490 		}
1491 	}
1492 
1493 	rxvq->stats.packets += nb_rx;
1494 
1495 	/* Allocate new mbuf for the used descriptor */
1496 
1497 	if (likely(!virtqueue_full(vq))) {
1498 		/* free_cnt may include mrg descs */
1499 		uint16_t free_cnt = vq->vq_free_cnt;
1500 		struct rte_mbuf *new_pkts[free_cnt];
1501 
1502 		if (!rte_pktmbuf_alloc_bulk(rxvq->mpool, new_pkts, free_cnt)) {
1503 			error = virtqueue_enqueue_refill_inorder(vq, new_pkts,
1504 					free_cnt);
1505 			if (unlikely(error)) {
1506 				for (i = 0; i < free_cnt; i++)
1507 					rte_pktmbuf_free(new_pkts[i]);
1508 			}
1509 			nb_enqueued += free_cnt;
1510 		} else {
1511 			struct rte_eth_dev *dev =
1512 				&rte_eth_devices[rxvq->port_id];
1513 			dev->data->rx_mbuf_alloc_failed += free_cnt;
1514 		}
1515 	}
1516 
1517 	if (likely(nb_enqueued)) {
1518 		vq_update_avail_idx(vq);
1519 
1520 		if (unlikely(virtqueue_kick_prepare(vq))) {
1521 			virtqueue_notify(vq);
1522 			PMD_RX_LOG(DEBUG, "Notified");
1523 		}
1524 	}
1525 
1526 	return nb_rx;
1527 }
1528 
1529 uint16_t
1530 virtio_recv_mergeable_pkts(void *rx_queue,
1531 			struct rte_mbuf **rx_pkts,
1532 			uint16_t nb_pkts)
1533 {
1534 	struct virtnet_rx *rxvq = rx_queue;
1535 	struct virtqueue *vq = rxvq->vq;
1536 	struct virtio_hw *hw = vq->hw;
1537 	struct rte_mbuf *rxm;
1538 	struct rte_mbuf *prev;
1539 	uint16_t nb_used, num, nb_rx = 0;
1540 	uint32_t len[VIRTIO_MBUF_BURST_SZ];
1541 	struct rte_mbuf *rcv_pkts[VIRTIO_MBUF_BURST_SZ];
1542 	int error;
1543 	uint32_t nb_enqueued = 0;
1544 	uint32_t seg_num = 0;
1545 	uint32_t seg_res = 0;
1546 	uint32_t hdr_size = hw->vtnet_hdr_size;
1547 	int32_t i;
1548 
1549 	if (unlikely(hw->started == 0))
1550 		return nb_rx;
1551 
1552 	nb_used = VIRTQUEUE_NUSED(vq);
1553 
1554 	virtio_rmb(hw->weak_barriers);
1555 
1556 	PMD_RX_LOG(DEBUG, "used:%d", nb_used);
1557 
1558 	num = likely(nb_used <= nb_pkts) ? nb_used : nb_pkts;
1559 	if (unlikely(num > VIRTIO_MBUF_BURST_SZ))
1560 		num = VIRTIO_MBUF_BURST_SZ;
1561 	if (likely(num > DESC_PER_CACHELINE))
1562 		num = num - ((vq->vq_used_cons_idx + num) %
1563 				DESC_PER_CACHELINE);
1564 
1565 
1566 	num = virtqueue_dequeue_burst_rx(vq, rcv_pkts, len, num);
1567 
1568 	for (i = 0; i < num; i++) {
1569 		struct virtio_net_hdr_mrg_rxbuf *header;
1570 
1571 		PMD_RX_LOG(DEBUG, "dequeue:%d", num);
1572 		PMD_RX_LOG(DEBUG, "packet len:%d", len[i]);
1573 
1574 		rxm = rcv_pkts[i];
1575 
1576 		if (unlikely(len[i] < hdr_size + ETHER_HDR_LEN)) {
1577 			PMD_RX_LOG(ERR, "Packet drop");
1578 			nb_enqueued++;
1579 			virtio_discard_rxbuf(vq, rxm);
1580 			rxvq->stats.errors++;
1581 			continue;
1582 		}
1583 
1584 		header = (struct virtio_net_hdr_mrg_rxbuf *)
1585 			 ((char *)rxm->buf_addr + RTE_PKTMBUF_HEADROOM
1586 			 - hdr_size);
1587 		seg_num = header->num_buffers;
1588 		if (seg_num == 0)
1589 			seg_num = 1;
1590 
1591 		rxm->data_off = RTE_PKTMBUF_HEADROOM;
1592 		rxm->nb_segs = seg_num;
1593 		rxm->ol_flags = 0;
1594 		rxm->vlan_tci = 0;
1595 		rxm->pkt_len = (uint32_t)(len[i] - hdr_size);
1596 		rxm->data_len = (uint16_t)(len[i] - hdr_size);
1597 
1598 		rxm->port = rxvq->port_id;
1599 
1600 		rx_pkts[nb_rx] = rxm;
1601 		prev = rxm;
1602 
1603 		if (hw->has_rx_offload &&
1604 				virtio_rx_offload(rxm, &header->hdr) < 0) {
1605 			virtio_discard_rxbuf(vq, rxm);
1606 			rxvq->stats.errors++;
1607 			continue;
1608 		}
1609 
1610 		if (hw->vlan_strip)
1611 			rte_vlan_strip(rx_pkts[nb_rx]);
1612 
1613 		seg_res = seg_num - 1;
1614 
1615 		/* Merge remaining segments */
1616 		while (seg_res != 0 && i < (num - 1)) {
1617 			i++;
1618 
1619 			rxm = rcv_pkts[i];
1620 			rxm->data_off = RTE_PKTMBUF_HEADROOM - hdr_size;
1621 			rxm->pkt_len = (uint32_t)(len[i]);
1622 			rxm->data_len = (uint16_t)(len[i]);
1623 
1624 			rx_pkts[nb_rx]->pkt_len += (uint32_t)(len[i]);
1625 			rx_pkts[nb_rx]->data_len += (uint16_t)(len[i]);
1626 
1627 			if (prev)
1628 				prev->next = rxm;
1629 
1630 			prev = rxm;
1631 			seg_res -= 1;
1632 		}
1633 
1634 		if (!seg_res) {
1635 			virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1636 			nb_rx++;
1637 		}
1638 	}
1639 
1640 	/* Last packet still need merge segments */
1641 	while (seg_res != 0) {
1642 		uint16_t rcv_cnt = RTE_MIN((uint16_t)seg_res,
1643 					VIRTIO_MBUF_BURST_SZ);
1644 
1645 		prev = rcv_pkts[nb_rx];
1646 		if (likely(VIRTQUEUE_NUSED(vq) >= rcv_cnt)) {
1647 			virtio_rmb(hw->weak_barriers);
1648 			num = virtqueue_dequeue_burst_rx(vq, rcv_pkts, len,
1649 							   rcv_cnt);
1650 			uint16_t extra_idx = 0;
1651 
1652 			rcv_cnt = num;
1653 			while (extra_idx < rcv_cnt) {
1654 				rxm = rcv_pkts[extra_idx];
1655 				rxm->data_off =
1656 					RTE_PKTMBUF_HEADROOM - hdr_size;
1657 				rxm->pkt_len = (uint32_t)(len[extra_idx]);
1658 				rxm->data_len = (uint16_t)(len[extra_idx]);
1659 				prev->next = rxm;
1660 				prev = rxm;
1661 				rx_pkts[nb_rx]->pkt_len += len[extra_idx];
1662 				rx_pkts[nb_rx]->data_len += len[extra_idx];
1663 				extra_idx += 1;
1664 			};
1665 			seg_res -= rcv_cnt;
1666 
1667 			if (!seg_res) {
1668 				virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1669 				nb_rx++;
1670 			}
1671 		} else {
1672 			PMD_RX_LOG(ERR,
1673 					"No enough segments for packet.");
1674 			virtio_discard_rxbuf(vq, prev);
1675 			rxvq->stats.errors++;
1676 			break;
1677 		}
1678 	}
1679 
1680 	rxvq->stats.packets += nb_rx;
1681 
1682 	/* Allocate new mbuf for the used descriptor */
1683 	if (likely(!virtqueue_full(vq))) {
1684 		/* free_cnt may include mrg descs */
1685 		uint16_t free_cnt = vq->vq_free_cnt;
1686 		struct rte_mbuf *new_pkts[free_cnt];
1687 
1688 		if (!rte_pktmbuf_alloc_bulk(rxvq->mpool, new_pkts, free_cnt)) {
1689 			error = virtqueue_enqueue_recv_refill(vq, new_pkts,
1690 					free_cnt);
1691 			if (unlikely(error)) {
1692 				for (i = 0; i < free_cnt; i++)
1693 					rte_pktmbuf_free(new_pkts[i]);
1694 			}
1695 			nb_enqueued += free_cnt;
1696 		} else {
1697 			struct rte_eth_dev *dev =
1698 				&rte_eth_devices[rxvq->port_id];
1699 			dev->data->rx_mbuf_alloc_failed += free_cnt;
1700 		}
1701 	}
1702 
1703 	if (likely(nb_enqueued)) {
1704 		vq_update_avail_idx(vq);
1705 
1706 		if (unlikely(virtqueue_kick_prepare(vq))) {
1707 			virtqueue_notify(vq);
1708 			PMD_RX_LOG(DEBUG, "Notified");
1709 		}
1710 	}
1711 
1712 	return nb_rx;
1713 }
1714 
1715 uint16_t
1716 virtio_recv_mergeable_pkts_packed(void *rx_queue,
1717 			struct rte_mbuf **rx_pkts,
1718 			uint16_t nb_pkts)
1719 {
1720 	struct virtnet_rx *rxvq = rx_queue;
1721 	struct virtqueue *vq = rxvq->vq;
1722 	struct virtio_hw *hw = vq->hw;
1723 	struct rte_mbuf *rxm;
1724 	struct rte_mbuf *prev = NULL;
1725 	uint16_t num, nb_rx = 0;
1726 	uint32_t len[VIRTIO_MBUF_BURST_SZ];
1727 	struct rte_mbuf *rcv_pkts[VIRTIO_MBUF_BURST_SZ];
1728 	uint32_t nb_enqueued = 0;
1729 	uint32_t seg_num = 0;
1730 	uint32_t seg_res = 0;
1731 	uint32_t hdr_size = hw->vtnet_hdr_size;
1732 	int32_t i;
1733 	int error;
1734 
1735 	if (unlikely(hw->started == 0))
1736 		return nb_rx;
1737 
1738 
1739 	num = nb_pkts;
1740 	if (unlikely(num > VIRTIO_MBUF_BURST_SZ))
1741 		num = VIRTIO_MBUF_BURST_SZ;
1742 	if (likely(num > DESC_PER_CACHELINE))
1743 		num = num - ((vq->vq_used_cons_idx + num) % DESC_PER_CACHELINE);
1744 
1745 	num = virtqueue_dequeue_burst_rx_packed(vq, rcv_pkts, len, num);
1746 
1747 	for (i = 0; i < num; i++) {
1748 		struct virtio_net_hdr_mrg_rxbuf *header;
1749 
1750 		PMD_RX_LOG(DEBUG, "dequeue:%d", num);
1751 		PMD_RX_LOG(DEBUG, "packet len:%d", len[i]);
1752 
1753 		rxm = rcv_pkts[i];
1754 
1755 		if (unlikely(len[i] < hdr_size + ETHER_HDR_LEN)) {
1756 			PMD_RX_LOG(ERR, "Packet drop");
1757 			nb_enqueued++;
1758 			virtio_discard_rxbuf(vq, rxm);
1759 			rxvq->stats.errors++;
1760 			continue;
1761 		}
1762 
1763 		header = (struct virtio_net_hdr_mrg_rxbuf *)((char *)
1764 			  rxm->buf_addr + RTE_PKTMBUF_HEADROOM - hdr_size);
1765 		seg_num = header->num_buffers;
1766 
1767 		if (seg_num == 0)
1768 			seg_num = 1;
1769 
1770 		rxm->data_off = RTE_PKTMBUF_HEADROOM;
1771 		rxm->nb_segs = seg_num;
1772 		rxm->ol_flags = 0;
1773 		rxm->vlan_tci = 0;
1774 		rxm->pkt_len = (uint32_t)(len[i] - hdr_size);
1775 		rxm->data_len = (uint16_t)(len[i] - hdr_size);
1776 
1777 		rxm->port = rxvq->port_id;
1778 		rx_pkts[nb_rx] = rxm;
1779 		prev = rxm;
1780 
1781 		if (hw->has_rx_offload &&
1782 				virtio_rx_offload(rxm, &header->hdr) < 0) {
1783 			virtio_discard_rxbuf(vq, rxm);
1784 			rxvq->stats.errors++;
1785 			continue;
1786 		}
1787 
1788 		if (hw->vlan_strip)
1789 			rte_vlan_strip(rx_pkts[nb_rx]);
1790 
1791 		seg_res = seg_num - 1;
1792 
1793 		/* Merge remaining segments */
1794 		while (seg_res != 0 && i < (num - 1)) {
1795 			i++;
1796 
1797 			rxm = rcv_pkts[i];
1798 			rxm->data_off = RTE_PKTMBUF_HEADROOM - hdr_size;
1799 			rxm->pkt_len = (uint32_t)(len[i]);
1800 			rxm->data_len = (uint16_t)(len[i]);
1801 
1802 			rx_pkts[nb_rx]->pkt_len += (uint32_t)(len[i]);
1803 			rx_pkts[nb_rx]->data_len += (uint16_t)(len[i]);
1804 
1805 			if (prev)
1806 				prev->next = rxm;
1807 
1808 			prev = rxm;
1809 			seg_res -= 1;
1810 		}
1811 
1812 		if (!seg_res) {
1813 			virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1814 			nb_rx++;
1815 		}
1816 	}
1817 
1818 	/* Last packet still need merge segments */
1819 	while (seg_res != 0) {
1820 		uint16_t rcv_cnt = RTE_MIN((uint16_t)seg_res,
1821 					VIRTIO_MBUF_BURST_SZ);
1822 		if (likely(vq->vq_free_cnt >= rcv_cnt)) {
1823 			num = virtqueue_dequeue_burst_rx_packed(vq, rcv_pkts,
1824 					len, rcv_cnt);
1825 			uint16_t extra_idx = 0;
1826 
1827 			rcv_cnt = num;
1828 
1829 			while (extra_idx < rcv_cnt) {
1830 				rxm = rcv_pkts[extra_idx];
1831 
1832 				rxm->data_off =
1833 					RTE_PKTMBUF_HEADROOM - hdr_size;
1834 				rxm->pkt_len = (uint32_t)(len[extra_idx]);
1835 				rxm->data_len = (uint16_t)(len[extra_idx]);
1836 
1837 				prev->next = rxm;
1838 				prev = rxm;
1839 				rx_pkts[nb_rx]->pkt_len += len[extra_idx];
1840 				rx_pkts[nb_rx]->data_len += len[extra_idx];
1841 				extra_idx += 1;
1842 			}
1843 			seg_res -= rcv_cnt;
1844 			if (!seg_res) {
1845 				virtio_rx_stats_updated(rxvq, rx_pkts[nb_rx]);
1846 				nb_rx++;
1847 			}
1848 		} else {
1849 			PMD_RX_LOG(ERR,
1850 					"No enough segments for packet.");
1851 			if (prev)
1852 				virtio_discard_rxbuf(vq, prev);
1853 			rxvq->stats.errors++;
1854 			break;
1855 		}
1856 	}
1857 
1858 	rxvq->stats.packets += nb_rx;
1859 
1860 	/* Allocate new mbuf for the used descriptor */
1861 	if (likely(!virtqueue_full(vq))) {
1862 		/* free_cnt may include mrg descs */
1863 		uint16_t free_cnt = vq->vq_free_cnt;
1864 		struct rte_mbuf *new_pkts[free_cnt];
1865 
1866 		if (!rte_pktmbuf_alloc_bulk(rxvq->mpool, new_pkts, free_cnt)) {
1867 			error = virtqueue_enqueue_recv_refill_packed(vq,
1868 					new_pkts, free_cnt);
1869 			if (unlikely(error)) {
1870 				for (i = 0; i < free_cnt; i++)
1871 					rte_pktmbuf_free(new_pkts[i]);
1872 			}
1873 			nb_enqueued += free_cnt;
1874 		} else {
1875 			struct rte_eth_dev *dev =
1876 				&rte_eth_devices[rxvq->port_id];
1877 			dev->data->rx_mbuf_alloc_failed += free_cnt;
1878 		}
1879 	}
1880 
1881 	if (likely(nb_enqueued)) {
1882 		if (unlikely(virtqueue_kick_prepare_packed(vq))) {
1883 			virtqueue_notify(vq);
1884 			PMD_RX_LOG(DEBUG, "Notified");
1885 		}
1886 	}
1887 
1888 	return nb_rx;
1889 }
1890 
1891 uint16_t
1892 virtio_xmit_pkts_packed(void *tx_queue, struct rte_mbuf **tx_pkts,
1893 			uint16_t nb_pkts)
1894 {
1895 	struct virtnet_tx *txvq = tx_queue;
1896 	struct virtqueue *vq = txvq->vq;
1897 	struct virtio_hw *hw = vq->hw;
1898 	uint16_t hdr_size = hw->vtnet_hdr_size;
1899 	uint16_t nb_tx = 0;
1900 	int error;
1901 
1902 	if (unlikely(hw->started == 0 && tx_pkts != hw->inject_pkts))
1903 		return nb_tx;
1904 
1905 	if (unlikely(nb_pkts < 1))
1906 		return nb_pkts;
1907 
1908 	PMD_TX_LOG(DEBUG, "%d packets to xmit", nb_pkts);
1909 
1910 	if (nb_pkts > vq->vq_free_cnt)
1911 		virtio_xmit_cleanup_packed(vq, nb_pkts - vq->vq_free_cnt);
1912 
1913 	for (nb_tx = 0; nb_tx < nb_pkts; nb_tx++) {
1914 		struct rte_mbuf *txm = tx_pkts[nb_tx];
1915 		int can_push = 0, slots, need;
1916 
1917 		/* Do VLAN tag insertion */
1918 		if (unlikely(txm->ol_flags & PKT_TX_VLAN_PKT)) {
1919 			error = rte_vlan_insert(&txm);
1920 			if (unlikely(error)) {
1921 				rte_pktmbuf_free(txm);
1922 				continue;
1923 			}
1924 		}
1925 
1926 		/* optimize ring usage */
1927 		if ((vtpci_with_feature(hw, VIRTIO_F_ANY_LAYOUT) ||
1928 		      vtpci_with_feature(hw, VIRTIO_F_VERSION_1)) &&
1929 		    rte_mbuf_refcnt_read(txm) == 1 &&
1930 		    RTE_MBUF_DIRECT(txm) &&
1931 		    txm->nb_segs == 1 &&
1932 		    rte_pktmbuf_headroom(txm) >= hdr_size &&
1933 		    rte_is_aligned(rte_pktmbuf_mtod(txm, char *),
1934 			   __alignof__(struct virtio_net_hdr_mrg_rxbuf)))
1935 			can_push = 1;
1936 
1937 		/* How many main ring entries are needed to this Tx?
1938 		 * any_layout => number of segments
1939 		 * default    => number of segments + 1
1940 		 */
1941 		slots = txm->nb_segs + !can_push;
1942 		need = slots - vq->vq_free_cnt;
1943 
1944 		/* Positive value indicates it need free vring descriptors */
1945 		if (unlikely(need > 0)) {
1946 			need = RTE_MIN(need, (int)nb_pkts);
1947 			virtio_xmit_cleanup_packed(vq, need);
1948 			need = slots - vq->vq_free_cnt;
1949 			if (unlikely(need > 0)) {
1950 				PMD_TX_LOG(ERR,
1951 					   "No free tx descriptors to transmit");
1952 				break;
1953 			}
1954 		}
1955 
1956 		/* Enqueue Packet buffers */
1957 		virtqueue_enqueue_xmit_packed(txvq, txm, slots, can_push);
1958 
1959 		virtio_update_packet_stats(&txvq->stats, txm);
1960 	}
1961 
1962 	txvq->stats.packets += nb_tx;
1963 
1964 	if (likely(nb_tx)) {
1965 		if (unlikely(virtqueue_kick_prepare_packed(vq))) {
1966 			virtqueue_notify(vq);
1967 			PMD_TX_LOG(DEBUG, "Notified backend after xmit");
1968 		}
1969 	}
1970 
1971 	return nb_tx;
1972 }
1973 
1974 uint16_t
1975 virtio_xmit_pkts(void *tx_queue, struct rte_mbuf **tx_pkts, uint16_t nb_pkts)
1976 {
1977 	struct virtnet_tx *txvq = tx_queue;
1978 	struct virtqueue *vq = txvq->vq;
1979 	struct virtio_hw *hw = vq->hw;
1980 	uint16_t hdr_size = hw->vtnet_hdr_size;
1981 	uint16_t nb_used, nb_tx = 0;
1982 	int error;
1983 
1984 	if (unlikely(hw->started == 0 && tx_pkts != hw->inject_pkts))
1985 		return nb_tx;
1986 
1987 	if (unlikely(nb_pkts < 1))
1988 		return nb_pkts;
1989 
1990 	PMD_TX_LOG(DEBUG, "%d packets to xmit", nb_pkts);
1991 	nb_used = VIRTQUEUE_NUSED(vq);
1992 
1993 	virtio_rmb(hw->weak_barriers);
1994 	if (likely(nb_used > vq->vq_nentries - vq->vq_free_thresh))
1995 		virtio_xmit_cleanup(vq, nb_used);
1996 
1997 	for (nb_tx = 0; nb_tx < nb_pkts; nb_tx++) {
1998 		struct rte_mbuf *txm = tx_pkts[nb_tx];
1999 		int can_push = 0, use_indirect = 0, slots, need;
2000 
2001 		/* Do VLAN tag insertion */
2002 		if (unlikely(txm->ol_flags & PKT_TX_VLAN_PKT)) {
2003 			error = rte_vlan_insert(&txm);
2004 			if (unlikely(error)) {
2005 				rte_pktmbuf_free(txm);
2006 				continue;
2007 			}
2008 		}
2009 
2010 		/* optimize ring usage */
2011 		if ((vtpci_with_feature(hw, VIRTIO_F_ANY_LAYOUT) ||
2012 		      vtpci_with_feature(hw, VIRTIO_F_VERSION_1)) &&
2013 		    rte_mbuf_refcnt_read(txm) == 1 &&
2014 		    RTE_MBUF_DIRECT(txm) &&
2015 		    txm->nb_segs == 1 &&
2016 		    rte_pktmbuf_headroom(txm) >= hdr_size &&
2017 		    rte_is_aligned(rte_pktmbuf_mtod(txm, char *),
2018 				   __alignof__(struct virtio_net_hdr_mrg_rxbuf)))
2019 			can_push = 1;
2020 		else if (vtpci_with_feature(hw, VIRTIO_RING_F_INDIRECT_DESC) &&
2021 			 txm->nb_segs < VIRTIO_MAX_TX_INDIRECT)
2022 			use_indirect = 1;
2023 
2024 		/* How many main ring entries are needed to this Tx?
2025 		 * any_layout => number of segments
2026 		 * indirect   => 1
2027 		 * default    => number of segments + 1
2028 		 */
2029 		slots = use_indirect ? 1 : (txm->nb_segs + !can_push);
2030 		need = slots - vq->vq_free_cnt;
2031 
2032 		/* Positive value indicates it need free vring descriptors */
2033 		if (unlikely(need > 0)) {
2034 			nb_used = VIRTQUEUE_NUSED(vq);
2035 			virtio_rmb(hw->weak_barriers);
2036 			need = RTE_MIN(need, (int)nb_used);
2037 
2038 			virtio_xmit_cleanup(vq, need);
2039 			need = slots - vq->vq_free_cnt;
2040 			if (unlikely(need > 0)) {
2041 				PMD_TX_LOG(ERR,
2042 					   "No free tx descriptors to transmit");
2043 				break;
2044 			}
2045 		}
2046 
2047 		/* Enqueue Packet buffers */
2048 		virtqueue_enqueue_xmit(txvq, txm, slots, use_indirect,
2049 			can_push, 0);
2050 
2051 		virtio_update_packet_stats(&txvq->stats, txm);
2052 	}
2053 
2054 	txvq->stats.packets += nb_tx;
2055 
2056 	if (likely(nb_tx)) {
2057 		vq_update_avail_idx(vq);
2058 
2059 		if (unlikely(virtqueue_kick_prepare(vq))) {
2060 			virtqueue_notify(vq);
2061 			PMD_TX_LOG(DEBUG, "Notified backend after xmit");
2062 		}
2063 	}
2064 
2065 	return nb_tx;
2066 }
2067 
2068 uint16_t
2069 virtio_xmit_pkts_inorder(void *tx_queue,
2070 			struct rte_mbuf **tx_pkts,
2071 			uint16_t nb_pkts)
2072 {
2073 	struct virtnet_tx *txvq = tx_queue;
2074 	struct virtqueue *vq = txvq->vq;
2075 	struct virtio_hw *hw = vq->hw;
2076 	uint16_t hdr_size = hw->vtnet_hdr_size;
2077 	uint16_t nb_used, nb_avail, nb_tx = 0, nb_inorder_pkts = 0;
2078 	struct rte_mbuf *inorder_pkts[nb_pkts];
2079 	int error;
2080 
2081 	if (unlikely(hw->started == 0 && tx_pkts != hw->inject_pkts))
2082 		return nb_tx;
2083 
2084 	if (unlikely(nb_pkts < 1))
2085 		return nb_pkts;
2086 
2087 	VIRTQUEUE_DUMP(vq);
2088 	PMD_TX_LOG(DEBUG, "%d packets to xmit", nb_pkts);
2089 	nb_used = VIRTQUEUE_NUSED(vq);
2090 
2091 	virtio_rmb(hw->weak_barriers);
2092 	if (likely(nb_used > vq->vq_nentries - vq->vq_free_thresh))
2093 		virtio_xmit_cleanup_inorder(vq, nb_used);
2094 
2095 	if (unlikely(!vq->vq_free_cnt))
2096 		virtio_xmit_cleanup_inorder(vq, nb_used);
2097 
2098 	nb_avail = RTE_MIN(vq->vq_free_cnt, nb_pkts);
2099 
2100 	for (nb_tx = 0; nb_tx < nb_avail; nb_tx++) {
2101 		struct rte_mbuf *txm = tx_pkts[nb_tx];
2102 		int slots, need;
2103 
2104 		/* Do VLAN tag insertion */
2105 		if (unlikely(txm->ol_flags & PKT_TX_VLAN_PKT)) {
2106 			error = rte_vlan_insert(&txm);
2107 			if (unlikely(error)) {
2108 				rte_pktmbuf_free(txm);
2109 				continue;
2110 			}
2111 		}
2112 
2113 		/* optimize ring usage */
2114 		if ((vtpci_with_feature(hw, VIRTIO_F_ANY_LAYOUT) ||
2115 		     vtpci_with_feature(hw, VIRTIO_F_VERSION_1)) &&
2116 		     rte_mbuf_refcnt_read(txm) == 1 &&
2117 		     RTE_MBUF_DIRECT(txm) &&
2118 		     txm->nb_segs == 1 &&
2119 		     rte_pktmbuf_headroom(txm) >= hdr_size &&
2120 		     rte_is_aligned(rte_pktmbuf_mtod(txm, char *),
2121 				__alignof__(struct virtio_net_hdr_mrg_rxbuf))) {
2122 			inorder_pkts[nb_inorder_pkts] = txm;
2123 			nb_inorder_pkts++;
2124 
2125 			virtio_update_packet_stats(&txvq->stats, txm);
2126 			continue;
2127 		}
2128 
2129 		if (nb_inorder_pkts) {
2130 			virtqueue_enqueue_xmit_inorder(txvq, inorder_pkts,
2131 							nb_inorder_pkts);
2132 			nb_inorder_pkts = 0;
2133 		}
2134 
2135 		slots = txm->nb_segs + 1;
2136 		need = slots - vq->vq_free_cnt;
2137 		if (unlikely(need > 0)) {
2138 			nb_used = VIRTQUEUE_NUSED(vq);
2139 			virtio_rmb(hw->weak_barriers);
2140 			need = RTE_MIN(need, (int)nb_used);
2141 
2142 			virtio_xmit_cleanup_inorder(vq, need);
2143 
2144 			need = slots - vq->vq_free_cnt;
2145 
2146 			if (unlikely(need > 0)) {
2147 				PMD_TX_LOG(ERR,
2148 					"No free tx descriptors to transmit");
2149 				break;
2150 			}
2151 		}
2152 		/* Enqueue Packet buffers */
2153 		virtqueue_enqueue_xmit(txvq, txm, slots, 0, 0, 1);
2154 
2155 		virtio_update_packet_stats(&txvq->stats, txm);
2156 	}
2157 
2158 	/* Transmit all inorder packets */
2159 	if (nb_inorder_pkts)
2160 		virtqueue_enqueue_xmit_inorder(txvq, inorder_pkts,
2161 						nb_inorder_pkts);
2162 
2163 	txvq->stats.packets += nb_tx;
2164 
2165 	if (likely(nb_tx)) {
2166 		vq_update_avail_idx(vq);
2167 
2168 		if (unlikely(virtqueue_kick_prepare(vq))) {
2169 			virtqueue_notify(vq);
2170 			PMD_TX_LOG(DEBUG, "Notified backend after xmit");
2171 		}
2172 	}
2173 
2174 	VIRTQUEUE_DUMP(vq);
2175 
2176 	return nb_tx;
2177 }
2178