vmxnet3: gso fixes

[vpp.git] / src / plugins / vmxnet3 / output.c

diff --git a/src/plugins/vmxnet3/output.c b/src/plugins/vmxnet3/output.c
index 5c48549..4c9b709 100644 (file)
--- a/src/plugins/vmxnet3/output.c
+++ b/src/plugins/vmxnet3/output.c
@@ -20,6 +20,8 @@
 #include <vlib/pci/pci.h>
 #include <vnet/ethernet/ethernet.h>
 #include <vnet/devices/devices.h>
+#include <vnet/ip/ip6_packet.h>
+#include <vnet/ip/ip4_packet.h>
 
 #include <vmxnet3/vmxnet3.h>
 
@@ -127,7 +129,10 @@ VNET_DEVICE_CLASS_TX_FN (vmxnet3_device_class) (vlib_main_t * vm,
   while (PREDICT_TRUE (n_left))
     {
       u16 space_needed = 1, i;
+      u32 gso_size = 0;
+      u32 l4_hdr_sz;
       vlib_buffer_t *b;
+      u32 hdr_len = 0;
 
       bi0 = buffers[0];
       b0 = vlib_get_buffer (vm, bi0);
@@ -178,17 +183,38 @@ VNET_DEVICE_CLASS_TX_FN (vmxnet3_device_class) (vlib_main_t * vm,
          txq->tx_ring.bufs[desc_idx] = bi0;
 
          txd = &txq->tx_desc[desc_idx];
+
          txd->address = vlib_buffer_get_current_pa (vm, b0);
 
          txd->flags[0] = generation | b0->current_length;
+         txd->flags[1] = 0;
+         if (PREDICT_FALSE (b0->flags & VNET_BUFFER_F_GSO))
+           {
+             /*
+              * We should not be getting GSO outbound traffic unless it is
+              * lro is enable
+              */
+             ASSERT (vd->gso_enable == 1);
+             gso_size = vnet_buffer2 (b0)->gso_size;
+             l4_hdr_sz = vnet_buffer2 (b0)->gso_l4_hdr_sz;
+             if (b0->flags & VNET_BUFFER_F_IS_IP6)
+               hdr_len = sizeof (ethernet_header_t) + sizeof (ip6_header_t) +
+                 l4_hdr_sz;
+             else
+               hdr_len = sizeof (ethernet_header_t) + sizeof (ip4_header_t) +
+                 l4_hdr_sz;
+           }
 
          generation = txq->tx_ring.gen;
-
-         txd->flags[1] = 0;
          bi0 = b0->next_buffer;
        }
-
-      txd->flags[1] = VMXNET3_TXF_CQ | VMXNET3_TXF_EOP;
+      if (PREDICT_FALSE (gso_size != 0))
+       {
+         txq->tx_desc[first_idx].flags[1] = hdr_len;
+         txq->tx_desc[first_idx].flags[1] |= VMXNET3_TXF_OM (VMXNET3_OM_TSO);
+         txq->tx_desc[first_idx].flags[0] |= VMXNET3_TXF_MSSCOF (gso_size);
+       }
+      txd->flags[1] |= VMXNET3_TXF_CQ | VMXNET3_TXF_EOP;
       asm volatile ("":::"memory");
       /*
        * Now toggle back the generation bit for the first segment.