[dpdk-dev] [PATCH v2 6/9] driver/virtio:enqueue vhost TX offload

[Jijiang Liu]

Enqueue vhost TX checksum and TSO4/6 offload in virtio-net lib.

Signed-off-by: Jijiang Liu <jijiang.liu at intel.com>
---
 drivers/net/virtio/virtio_rxtx.c |   61 ++++++++++++++++++++++++++++++++++++++
 1 files changed, 61 insertions(+), 0 deletions(-)

diff --git a/drivers/net/virtio/virtio_rxtx.c b/drivers/net/virtio/virtio_rxtx.c
index c5b53bb..b99f5b5 100644
--- a/drivers/net/virtio/virtio_rxtx.c
+++ b/drivers/net/virtio/virtio_rxtx.c
@@ -50,6 +50,10 @@
 #include <rte_string_fns.h>
 #include <rte_errno.h>
 #include <rte_byteorder.h>
+#include <rte_tcp.h>
+#include <rte_ip.h>
+#include <rte_udp.h>
+#include <rte_sctp.h>

 #include "virtio_logs.h"
 #include "virtio_ethdev.h"
@@ -199,6 +203,58 @@ virtqueue_enqueue_recv_refill(struct virtqueue *vq, struct 
rte_mbuf *cookie)
 }

 static int
+virtqueue_enqueue_offload(struct virtqueue *txvq, struct rte_mbuf *m,
+                       uint16_t idx, uint16_t hdr_sz)
+{
+       struct virtio_net_hdr *hdr = (struct virtio_net_hdr *)(uintptr_t)
+                               (txvq->virtio_net_hdr_addr + idx * hdr_sz);
+
+       hdr->flags = VIRTIO_NET_HDR_F_NEEDS_CSUM;
+
+       /* if vhost TX checksum offload is required */
+       if (m->ol_flags & PKT_TX_IP_CKSUM) {
+               hdr->csum_start = m->l2_len;
+               hdr->csum_offset = offsetof(struct ipv4_hdr, hdr_checksum);
+       } else if (m->ol_flags & PKT_TX_L4_MASK) {
+               hdr->csum_start = m->l2_len + m->l3_len;
+               switch (m->ol_flags & PKT_TX_L4_MASK) {
+               case PKT_TX_TCP_CKSUM:
+                       hdr->csum_offset = offsetof(struct tcp_hdr, cksum);
+                       break;
+               case PKT_TX_UDP_CKSUM:
+                       hdr->csum_offset = offsetof(struct udp_hdr,
+                                                       dgram_cksum);
+                       break;
+               case PKT_TX_SCTP_CKSUM:
+                       hdr->csum_offset = offsetof(struct sctp_hdr, cksum);
+                       break;
+               default:
+                       break;
+               }
+       } else
+               hdr->flags = 0;
+
+       /* if vhost TSO offload is required */
+       if (m->tso_segsz != 0 && m->ol_flags & PKT_TX_TCP_SEG) {
+               if (m->ol_flags & PKT_TX_IPV4) {
+                       if (!vtpci_with_feature(txvq->hw,
+                               VIRTIO_NET_F_HOST_TSO4))
+                               return -1;
+                       hdr->gso_type = VIRTIO_NET_HDR_GSO_TCPV4;
+               } else if (m->ol_flags & PKT_TX_IPV6) {
+                       if (!vtpci_with_feature(txvq->hw,
+                               VIRTIO_NET_F_HOST_TSO6))
+                               return -1;
+                       hdr->gso_type = VIRTIO_NET_HDR_GSO_TCPV6;
+               }
+               hdr->gso_size = m->tso_segsz;
+               hdr->hdr_len = m->l2_len + m->l3_len + m->l4_len;
+       } else
+               hdr->gso_type = VIRTIO_NET_HDR_GSO_NONE;
+       return 0;
+}
+
+static int
 virtqueue_enqueue_xmit(struct virtqueue *txvq, struct rte_mbuf *cookie)
 {
        struct vq_desc_extra *dxp;
@@ -221,6 +277,11 @@ virtqueue_enqueue_xmit(struct virtqueue *txvq, struct 
rte_mbuf *cookie)
        dxp->cookie = (void *)cookie;
        dxp->ndescs = needed;

+       if (vtpci_with_feature(txvq->hw, VIRTIO_NET_F_CSUM)) {
+               if (virtqueue_enqueue_offload(txvq, cookie, idx, head_size) < 0)
+                       return -EPERM;
+       }
+
        start_dp = txvq->vq_ring.desc;
        start_dp[idx].addr =
                txvq->virtio_net_hdr_mem + idx * head_size;
-- 
1.7.7.6