vxlan:offload RX flow
[vpp.git] / src / vnet / vxlan / encap.c
index 4cfbbc2..c34b0e6 100644 (file)
@@ -1,3 +1,4 @@
+
 /*
  * Copyright (c) 2015 Cisco and/or its affiliates.
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -18,6 +19,8 @@
 #include <vnet/ip/ip.h>
 #include <vnet/ethernet/ethernet.h>
 #include <vnet/vxlan/vxlan.h>
+#include <vnet/qos/qos_types.h>
+#include <vnet/adj/rewrite.h>
 
 /* Statistics (not all errors) */
 #define foreach_vxlan_encap_error    \
@@ -58,38 +61,41 @@ u8 * format_vxlan_encap_trace (u8 * s, va_list * args)
   return s;
 }
 
-
-#define foreach_fixed_header4_offset            \
-    _(0) _(1) _(2) _(3)
-
-#define foreach_fixed_header6_offset            \
-    _(0) _(1) _(2) _(3) _(4) _(5) _(6)
-
 always_inline uword
 vxlan_encap_inline (vlib_main_t * vm,
                    vlib_node_runtime_t * node,
                    vlib_frame_t * from_frame,
-                   u32 is_ip4)
+                   u8 is_ip4, u8 csum_offload)
 {
   u32 n_left_from, next_index, * from, * to_next;
   vxlan_main_t * vxm = &vxlan_main;
   vnet_main_t * vnm = vxm->vnet_main;
   vnet_interface_main_t * im = &vnm->interface_main;
+  vlib_combined_counter_main_t * tx_counter = 
+      im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX;
   u32 pkts_encapsulated = 0;
-  u16 old_l0 = 0, old_l1 = 0;
   u32 thread_index = vlib_get_thread_index();
-  u32 stats_sw_if_index, stats_n_packets, stats_n_bytes;
   u32 sw_if_index0 = 0, sw_if_index1 = 0;
   u32 next0 = 0, next1 = 0;
-  vnet_hw_interface_t * hi0, * hi1;
   vxlan_tunnel_t * t0 = NULL, * t1 = NULL;
+  index_t dpoi_idx0 = INDEX_INVALID, dpoi_idx1 = INDEX_INVALID;
 
   from = vlib_frame_vector_args (from_frame);
   n_left_from = from_frame->n_vectors;
 
   next_index = node->cached_next_index;
-  stats_sw_if_index = node->runtime_data[0];
-  stats_n_packets = stats_n_bytes = 0;
+
+  STATIC_ASSERT_SIZEOF(ip6_vxlan_header_t, 56);
+  STATIC_ASSERT_SIZEOF(ip4_vxlan_header_t, 36);
+
+  u8 const underlay_hdr_len = is_ip4 ?
+    sizeof(ip4_vxlan_header_t) : sizeof(ip6_vxlan_header_t);
+  u8 const rw_hdr_offset = sizeof t0->rewrite_data - underlay_hdr_len;
+  u16 const l3_len = is_ip4 ? sizeof(ip4_header_t) : sizeof(ip6_header_t);
+  u32 const csum_flags = is_ip4 ?
+    VNET_BUFFER_F_OFFLOAD_IP_CKSUM | VNET_BUFFER_F_IS_IP4 |
+    VNET_BUFFER_F_OFFLOAD_UDP_CKSUM :
+    VNET_BUFFER_F_OFFLOAD_UDP_CKSUM;
 
   while (n_left_from > 0)
     {
@@ -100,20 +106,6 @@ vxlan_encap_inline (vlib_main_t * vm,
 
       while (n_left_from >= 4 && n_left_to_next >= 2)
        {
-          u32 bi0, bi1;
-         vlib_buffer_t * b0, * b1;
-          u32 flow_hash0, flow_hash1;
-         u32 len0, len1;
-          ip4_header_t * ip4_0, * ip4_1;
-          ip6_header_t * ip6_0, * ip6_1;
-          udp_header_t * udp0, * udp1;
-          u64 * copy_src0, * copy_dst0;
-          u64 * copy_src1, * copy_dst1;
-          u32 * copy_src_last0, * copy_dst_last0;
-          u32 * copy_src_last1, * copy_dst_last1;
-          u16 new_l0, new_l1;
-          ip_csum_t sum0, sum1;
-
          /* Prefetch next iteration. */
          {
            vlib_buffer_t * p2, * p3;
@@ -128,195 +120,183 @@ vxlan_encap_inline (vlib_main_t * vm,
            CLIB_PREFETCH (p3->data, 2*CLIB_CACHE_LINE_BYTES, LOAD);
          }
 
-         bi0 = from[0];
-         bi1 = from[1];
-         to_next[0] = bi0;
-         to_next[1] = bi1;
+         u32 bi0 = to_next[0] = from[0];
+         u32 bi1 = to_next[1] = from[1];
          from += 2;
          to_next += 2;
          n_left_to_next -= 2;
          n_left_from -= 2;
 
-         b0 = vlib_get_buffer (vm, bi0);
-         b1 = vlib_get_buffer (vm, bi1);
-
-          flow_hash0 = vnet_l2_compute_flow_hash (b0);
-          flow_hash1 = vnet_l2_compute_flow_hash (b1);
+         vlib_buffer_t * b0 = vlib_get_buffer (vm, bi0);
+         vlib_buffer_t * b1 = vlib_get_buffer (vm, bi1);
+          u32 flow_hash0 = vnet_l2_compute_flow_hash (b0);
+          u32 flow_hash1 = vnet_l2_compute_flow_hash (b1);
 
          /* Get next node index and adj index from tunnel next_dpo */
          if (sw_if_index0 != vnet_buffer(b0)->sw_if_index[VLIB_TX])
            {
              sw_if_index0 = vnet_buffer(b0)->sw_if_index[VLIB_TX];
-             hi0 = vnet_get_sup_hw_interface (vnm, sw_if_index0);
+             vnet_hw_interface_t *hi0 = 
+                 vnet_get_sup_hw_interface (vnm, sw_if_index0);
              t0 = &vxm->tunnels[hi0->dev_instance];
-             /* Note: change to always set next0 if it may be set to drop */
+             /* Note: change to always set next0 if it may set to drop */
              next0 = t0->next_dpo.dpoi_next_node;
+             dpoi_idx0 = t0->next_dpo.dpoi_index;
            }
-          vnet_buffer(b0)->ip.adj_index[VLIB_TX] = t0->next_dpo.dpoi_index;
 
          /* Get next node index and adj index from tunnel next_dpo */
          if (sw_if_index1 != vnet_buffer(b1)->sw_if_index[VLIB_TX])
            {
-             sw_if_index1 = vnet_buffer(b1)->sw_if_index[VLIB_TX];
-             hi1 = vnet_get_sup_hw_interface (vnm, sw_if_index1); 
-             t1 = &vxm->tunnels[hi1->dev_instance];
-             /* Note: change to always set next1 if it may be set to drop */
-             next1 = t1->next_dpo.dpoi_next_node;
+             if (sw_if_index0 == vnet_buffer(b1)->sw_if_index[VLIB_TX])
+               {
+                 sw_if_index1 = sw_if_index0;
+                 t1 = t0;
+                 next1 = next0;
+                 dpoi_idx1 = dpoi_idx0;
+               }
+             else
+               {
+                 sw_if_index1 = vnet_buffer(b1)->sw_if_index[VLIB_TX];
+                 vnet_hw_interface_t *hi1 = 
+                     vnet_get_sup_hw_interface (vnm, sw_if_index1);
+                 t1 = &vxm->tunnels[hi1->dev_instance];
+                 /* Note: change to always set next1 if it may set to drop */
+                 next1 = t1->next_dpo.dpoi_next_node;
+                 dpoi_idx1 = t1->next_dpo.dpoi_index;
+               }
            }
-          vnet_buffer(b1)->ip.adj_index[VLIB_TX] = t1->next_dpo.dpoi_index;
 
-          /* Apply the rewrite string. $$$$ vnet_rewrite? */
-          vlib_buffer_advance (b0, -(word)_vec_len(t0->rewrite));
-          vlib_buffer_advance (b1, -(word)_vec_len(t1->rewrite));
+          vnet_buffer(b0)->ip.adj_index[VLIB_TX] = dpoi_idx0;
+          vnet_buffer(b1)->ip.adj_index[VLIB_TX] = dpoi_idx1;
+
+          ASSERT(t0->rewrite_header.data_bytes == underlay_hdr_len);
+          ASSERT(t1->rewrite_header.data_bytes == underlay_hdr_len);
 
+          vlib_buffer_advance (b0, -underlay_hdr_len);
+          vlib_buffer_advance (b1, -underlay_hdr_len);
+
+         u32 len0 = vlib_buffer_length_in_chain (vm, b0);
+         u32 len1 = vlib_buffer_length_in_chain (vm, b1);
+          u16 payload_l0 = clib_host_to_net_u16 (len0 - l3_len);
+          u16 payload_l1 = clib_host_to_net_u16 (len1 - l3_len);
+
+          void * underlay0 = vlib_buffer_get_current(b0);
+          void * underlay1 = vlib_buffer_get_current(b1);
+
+         /* vnet_rewrite_two_header writes only in (uword) 8 bytes chunks
+           * and discards the first 4 bytes of the (36 bytes ip4 underlay)  rewrite
+           * use memcpy as a workaround */
+          clib_memcpy(underlay0, t0->rewrite_header.data + rw_hdr_offset, underlay_hdr_len);
+          clib_memcpy(underlay1, t1->rewrite_header.data + rw_hdr_offset, underlay_hdr_len);
+
+          ip4_header_t * ip4_0, * ip4_1;
+         qos_bits_t ip4_0_tos = 0, ip4_1_tos = 0;
+          ip6_header_t * ip6_0, * ip6_1;
+          udp_header_t * udp0, * udp1;
+          u8 * l3_0, * l3_1;
          if (is_ip4)
            {
-             /* IP4 VXLAN header should be 36 octects */
-              ASSERT(sizeof(ip4_vxlan_header_t) == 36);
-              ASSERT(vec_len(t0->rewrite) == sizeof(ip4_vxlan_header_t));
-             ASSERT(vec_len(t1->rewrite) == sizeof(ip4_vxlan_header_t));
-
-             ip4_0 = vlib_buffer_get_current(b0);
-             ip4_1 = vlib_buffer_get_current(b1);
-
-             /* Copy the fixed header */
-             copy_dst0 = (u64 *) ip4_0;
-             copy_src0 = (u64 *) t0->rewrite;
-             copy_dst1 = (u64 *) ip4_1;
-             copy_src1 = (u64 *) t1->rewrite;
-             /* Copy first 32 octets 8-bytes at a time */
-#define _(offs) copy_dst0[offs] = copy_src0[offs];
-             foreach_fixed_header4_offset;
-#undef _
-#define _(offs) copy_dst1[offs] = copy_src1[offs];
-             foreach_fixed_header4_offset;
-#undef _
-             /* Last 4 octets. Hopefully gcc will be our friend */
-              copy_dst_last0 = (u32 *)(&copy_dst0[4]);
-              copy_src_last0 = (u32 *)(&copy_src0[4]);
-              copy_dst_last0[0] = copy_src_last0[0];
-              copy_dst_last1 = (u32 *)(&copy_dst1[4]);
-              copy_src_last1 = (u32 *)(&copy_src1[4]);
-              copy_dst_last1[0] = copy_src_last1[0];
+              ip4_vxlan_header_t * hdr0 = underlay0;
+              ip4_vxlan_header_t * hdr1 = underlay1;
 
              /* Fix the IP4 checksum and length */
-             sum0 = ip4_0->checksum;
-             new_l0 = /* old_l0 always 0, see the rewrite setup */ 
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b0));
-              sum0 = ip_csum_update (sum0, old_l0, new_l0, ip4_header_t,
-                                    length /* changed member */);
-             ip4_0->checksum = ip_csum_fold (sum0);
-             ip4_0->length = new_l0;
-             sum1 = ip4_1->checksum;
-             new_l1 = /* old_l1 always 0, see the rewrite setup */
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b1));
-              sum1 = ip_csum_update (sum1, old_l1, new_l1, ip4_header_t,
-                                    length /* changed member */);
-             ip4_1->checksum = ip_csum_fold (sum1);
-             ip4_1->length = new_l1;
-
-             /* Fix UDP length and set source port */
-             udp0 = (udp_header_t *)(ip4_0+1);
-             new_l0 = clib_host_to_net_u16 (vlib_buffer_length_in_chain(vm, b0)
-                                            - sizeof (*ip4_0));
-             udp0->length = new_l0;
-             udp0->src_port = flow_hash0;
-             udp1 = (udp_header_t *)(ip4_1+1);
-             new_l1 = clib_host_to_net_u16 (vlib_buffer_length_in_chain(vm, b1)
-                                            - sizeof (*ip4_1));
-             udp1->length = new_l1;
-             udp1->src_port = flow_hash1;
+             ip4_0 = &hdr0->ip4;
+             ip4_1 = &hdr1->ip4;
+              ip4_0->length = clib_host_to_net_u16 (len0);
+              ip4_1->length = clib_host_to_net_u16 (len1);
+
+             if (PREDICT_FALSE (b0->flags & VNET_BUFFER_F_QOS_DATA_VALID))
+               {
+                 ip4_0_tos = vnet_buffer2 (b0)->qos.bits;
+                 ip4_0->tos = ip4_0_tos;
+               }
+             if (PREDICT_FALSE (b1->flags & VNET_BUFFER_F_QOS_DATA_VALID))
+               {
+                 ip4_1_tos = vnet_buffer2 (b1)->qos.bits;
+                 ip4_1->tos = ip4_1_tos;
+               }
+
+              l3_0 = (u8 *)ip4_0;
+              l3_1 = (u8 *)ip4_1;
+             udp0 = &hdr0->udp;
+             udp1 = &hdr1->udp;
            }
          else /* ipv6 */
            {
-              int bogus = 0;
+              ip6_vxlan_header_t * hdr0 = underlay0;
+              ip6_vxlan_header_t * hdr1 = underlay1;
 
-             /* IP6 VXLAN header should be 56 octects */
-              ASSERT(sizeof(ip6_vxlan_header_t) == 56);
-              ASSERT(vec_len(t0->rewrite) == sizeof(ip6_vxlan_header_t));
-             ASSERT(vec_len(t1->rewrite) == sizeof(ip6_vxlan_header_t));
-             ip6_0 = vlib_buffer_get_current(b0);
-             ip6_1 = vlib_buffer_get_current(b1);
-
-             /* Copy the fixed header */
-             copy_dst0 = (u64 *) ip6_0;
-             copy_src0 = (u64 *) t0->rewrite;
-             copy_dst1 = (u64 *) ip6_1;
-             copy_src1 = (u64 *) t1->rewrite;
-             /* Copy first 56 (ip6) octets 8-bytes at a time */
-#define _(offs) copy_dst0[offs] = copy_src0[offs];
-             foreach_fixed_header6_offset;
-#undef _
-#define _(offs) copy_dst1[offs] = copy_src1[offs];
-             foreach_fixed_header6_offset;
-#undef _
              /* Fix IP6 payload length */
-             new_l0 =
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b0)
-                                     - sizeof(*ip6_0));
-             ip6_0->payload_length = new_l0;
-             new_l1 =
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b1)
-                                     - sizeof(*ip6_1));
-             ip6_1->payload_length = new_l1;
-
-             /* Fix UDP length  and set source port */
-             udp0 = (udp_header_t *)(ip6_0+1);
-             udp0->length = new_l0;
-             udp0->src_port = flow_hash0;
-             udp1 = (udp_header_t *)(ip6_1+1);
-             udp1->length = new_l1;
-             udp1->src_port = flow_hash1;
-
-             /* IPv6 UDP checksum is mandatory */
-             udp0->checksum = ip6_tcp_udp_icmp_compute_checksum(vm, b0,
-                                                                ip6_0, &bogus);
-             ASSERT(bogus == 0);
-             if (udp0->checksum == 0)
-               udp0->checksum = 0xffff;
-             udp1->checksum = ip6_tcp_udp_icmp_compute_checksum(vm, b1,
-                                                        ip6_1, &bogus);
-             ASSERT(bogus == 0);
-             if (udp1->checksum == 0)
-               udp1->checksum = 0xffff;
+              ip6_0 = &hdr0->ip6;
+              ip6_1 = &hdr1->ip6;
+             ip6_0->payload_length = payload_l0;
+             ip6_1->payload_length = payload_l1;
+
+              l3_0 = (u8 *)ip6_0;
+              l3_1 = (u8 *)ip6_1;
+              udp0 = &hdr0->udp;
+              udp1 = &hdr1->udp;
            }
 
-          pkts_encapsulated += 2;
-         len0 = vlib_buffer_length_in_chain (vm, b0);
-         len1 = vlib_buffer_length_in_chain (vm, b1);
-         stats_n_packets += 2;
-         stats_n_bytes += len0 + len1;
-
-         /* Batch stats increment on the same vxlan tunnel so counter is not
-            incremented per packet. Note stats are still incremented for deleted
-            and admin-down tunnel where packets are dropped. It is not worthwhile
-            to check for this rare case and affect normal path performance. */
-         if (PREDICT_FALSE ((sw_if_index0 != stats_sw_if_index) ||
-                            (sw_if_index1 != stats_sw_if_index))) 
-           {
-             stats_n_packets -= 2;
-             stats_n_bytes -= len0 + len1;
-             if (sw_if_index0 == sw_if_index1) 
+          /* Fix UDP length  and set source port */
+          udp0->length = payload_l0;
+          udp0->src_port = flow_hash0;
+          udp1->length = payload_l1;
+          udp1->src_port = flow_hash1;
+
+          if (csum_offload)
+            {
+              b0->flags |= csum_flags;
+              vnet_buffer (b0)->l3_hdr_offset = l3_0 - b0->data;
+              vnet_buffer (b0)->l4_hdr_offset = (u8 *) udp0 - b0->data;
+              b1->flags |= csum_flags;
+              vnet_buffer (b1)->l3_hdr_offset = l3_1 - b1->data;
+              vnet_buffer (b1)->l4_hdr_offset = (u8 *) udp1 - b1->data;
+            }
+          /* IPv4 UDP checksum only if checksum offload is used */
+          else if (is_ip4)
+            {
+              ip_csum_t sum0 = ip4_0->checksum;
+              sum0 = ip_csum_update (sum0, 0, ip4_0->length, ip4_header_t,
+                  length /* changed member */);
+             if (PREDICT_FALSE (ip4_0_tos))
                {
-                 if (stats_n_packets) 
-                   vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      thread_index, stats_sw_if_index, 
-                      stats_n_packets, stats_n_bytes);
-                 stats_sw_if_index = sw_if_index0;
-                 stats_n_packets = 2;
-                 stats_n_bytes = len0 + len1;
-               }
-             else 
+                 sum0 = ip_csum_update (sum0, 0, ip4_0_tos, ip4_header_t,
+                     tos /* changed member */);
+               }
+              ip4_0->checksum = ip_csum_fold (sum0);
+              ip_csum_t sum1 = ip4_1->checksum;
+              sum1 = ip_csum_update (sum1, 0, ip4_1->length, ip4_header_t,
+                  length /* changed member */);
+             if (PREDICT_FALSE (ip4_1_tos))
                {
-                 vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      thread_index, sw_if_index0, 1, len0);
-                 vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      thread_index, sw_if_index1, 1, len1);
+                 sum1 = ip_csum_update (sum1, 0, ip4_1_tos, ip4_header_t,
+                     tos /* changed member */);
                }
-           }
+              ip4_1->checksum = ip_csum_fold (sum1);
+            }
+          /* IPv6 UDP checksum is mandatory */
+          else
+            {
+              int bogus = 0;
+
+              udp0->checksum = ip6_tcp_udp_icmp_compute_checksum
+                (vm, b0, ip6_0, &bogus);
+              ASSERT(bogus == 0);
+              if (udp0->checksum == 0)
+                udp0->checksum = 0xffff;
+              udp1->checksum = ip6_tcp_udp_icmp_compute_checksum
+                (vm, b1, ip6_1, &bogus);
+              ASSERT(bogus == 0);
+              if (udp1->checksum == 0)
+                udp1->checksum = 0xffff;
+            }
+
+          vlib_increment_combined_counter (tx_counter, thread_index,
+              sw_if_index0, 1, len0);
+          vlib_increment_combined_counter (tx_counter, thread_index,
+              sw_if_index1, 1, len1);
+          pkts_encapsulated += 2;
 
          if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) 
             {
@@ -341,135 +321,113 @@ vxlan_encap_inline (vlib_main_t * vm,
 
       while (n_left_from > 0 && n_left_to_next > 0)
        {
-         u32 bi0;
-         vlib_buffer_t * b0;
-          u32 flow_hash0;
-         u32 len0;
-          ip4_header_t * ip4_0;
-          ip6_header_t * ip6_0;
-          udp_header_t * udp0;
-          u64 * copy_src0, * copy_dst0;
-          u32 * copy_src_last0, * copy_dst_last0;
-          u16 new_l0;
-          ip_csum_t sum0;
-
-         bi0 = from[0];
-         to_next[0] = bi0;
+         u32 bi0 = to_next[0] = from[0];
          from += 1;
          to_next += 1;
          n_left_from -= 1;
          n_left_to_next -= 1;
 
-         b0 = vlib_get_buffer (vm, bi0);
-
-          flow_hash0 = vnet_l2_compute_flow_hash(b0);
+         vlib_buffer_t * b0 = vlib_get_buffer (vm, bi0);
+          u32 flow_hash0 = vnet_l2_compute_flow_hash(b0);
 
          /* Get next node index and adj index from tunnel next_dpo */
          if (sw_if_index0 != vnet_buffer(b0)->sw_if_index[VLIB_TX])
            {
              sw_if_index0 = vnet_buffer(b0)->sw_if_index[VLIB_TX];
-             hi0 = vnet_get_sup_hw_interface (vnm, sw_if_index0);
+             vnet_hw_interface_t *hi0 = 
+                 vnet_get_sup_hw_interface (vnm, sw_if_index0);
              t0 = &vxm->tunnels[hi0->dev_instance];
              /* Note: change to always set next0 if it may be set to drop */
              next0 = t0->next_dpo.dpoi_next_node;
+             dpoi_idx0 = t0->next_dpo.dpoi_index;
            }
-         vnet_buffer(b0)->ip.adj_index[VLIB_TX] = t0->next_dpo.dpoi_index;
+         vnet_buffer(b0)->ip.adj_index[VLIB_TX] = dpoi_idx0;
+
+          ASSERT(t0->rewrite_header.data_bytes == underlay_hdr_len);
+
+          vlib_buffer_advance (b0, -underlay_hdr_len);
+          void * underlay0 = vlib_buffer_get_current(b0);
 
-          /* Apply the rewrite string. $$$$ vnet_rewrite? */
-          vlib_buffer_advance (b0, -(word)_vec_len(t0->rewrite));
+         /* vnet_rewrite_one_header writes only in (uword) 8 bytes chunks
+           * and discards the first 4 bytes of the (36 bytes ip4 underlay)  rewrite
+           * use memcpy as a workaround */
+          clib_memcpy(underlay0, t0->rewrite_header.data + rw_hdr_offset, underlay_hdr_len);
 
+         u32 len0 = vlib_buffer_length_in_chain (vm, b0);
+          u16 payload_l0 = clib_host_to_net_u16 (len0 - l3_len);
+
+          udp_header_t * udp0;
+          ip4_header_t * ip4_0;
+         qos_bits_t ip4_0_tos = 0;
+          ip6_header_t * ip6_0;
+          u8 * l3_0;
          if (is_ip4)
            {
-             /* IP4 VXLAN header should be 36 octects */
-              ASSERT(sizeof(ip4_vxlan_header_t) == 36);
-              ASSERT(vec_len(t0->rewrite) == sizeof(ip4_vxlan_header_t));
-             ip4_0 = vlib_buffer_get_current(b0);
-
-             /* Copy the fixed header */
-             copy_dst0 = (u64 *) ip4_0;
-             copy_src0 = (u64 *) t0->rewrite;
-             /* Copy first 32 octets 8-bytes at a time */
-#define _(offs) copy_dst0[offs] = copy_src0[offs];
-             foreach_fixed_header4_offset;
-#undef _
-             /* Last 4 octets. Hopefully gcc will be our friend */
-              copy_dst_last0 = (u32 *)(&copy_dst0[4]);
-              copy_src_last0 = (u32 *)(&copy_src0[4]);
-              copy_dst_last0[0] = copy_src_last0[0];
+              ip4_vxlan_header_t * hdr = underlay0;
 
              /* Fix the IP4 checksum and length */
-             sum0 = ip4_0->checksum;
-             new_l0 = /* old_l0 always 0, see the rewrite setup */ 
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b0));
-              sum0 = ip_csum_update (sum0, old_l0, new_l0, ip4_header_t,
-                                    length /* changed member */);
-             ip4_0->checksum = ip_csum_fold (sum0);
-             ip4_0->length = new_l0;
-
-             /* Fix UDP length and set source port */
-             udp0 = (udp_header_t *)(ip4_0+1);
-             new_l0 = clib_host_to_net_u16 (vlib_buffer_length_in_chain(vm, b0)
-                                            - sizeof (*ip4_0));
-             udp0->length = new_l0;
-             udp0->src_port = flow_hash0;
-           }
+              ip4_0 = &hdr->ip4;
+              ip4_0->length = clib_host_to_net_u16 (len0);
+
+             if (PREDICT_FALSE (b0->flags & VNET_BUFFER_F_QOS_DATA_VALID))
+               {
+                 ip4_0_tos = vnet_buffer2 (b0)->qos.bits;
+                 ip4_0->tos = ip4_0_tos;
+               }
 
+              l3_0 = (u8*)ip4_0;
+             udp0 = &hdr->udp;
+           }
          else /* ip6 path */
            {
-              int bogus = 0;
+              ip6_vxlan_header_t * hdr = underlay0;
 
-             /* IP6 VXLAN header should be 56 octects */
-              ASSERT(sizeof(ip6_vxlan_header_t) == 56);
-              ASSERT(vec_len(t0->rewrite) == sizeof(ip6_vxlan_header_t));
-             ip6_0 = vlib_buffer_get_current(b0);
-             /* Copy the fixed header */
-             copy_dst0 = (u64 *) ip6_0;
-             copy_src0 = (u64 *) t0->rewrite;
-             /* Copy first 56 (ip6) octets 8-bytes at a time */
-#define _(offs) copy_dst0[offs] = copy_src0[offs];
-             foreach_fixed_header6_offset;
-#undef _
              /* Fix IP6 payload length */
-             new_l0 =
-                clib_host_to_net_u16 (vlib_buffer_length_in_chain (vm, b0)
-                                     - sizeof(*ip6_0));
-             ip6_0->payload_length = new_l0;
-
-             /* Fix UDP length  and set source port */
-             udp0 = (udp_header_t *)(ip6_0+1);
-             udp0->length = new_l0;
-             udp0->src_port = flow_hash0;
-
-             /* IPv6 UDP checksum is mandatory */
-             udp0->checksum = ip6_tcp_udp_icmp_compute_checksum(vm, b0,
-                                                                ip6_0, &bogus);
-             ASSERT(bogus == 0);
-             if (udp0->checksum == 0)
-               udp0->checksum = 0xffff;
+              ip6_0 = &hdr->ip6;
+             ip6_0->payload_length = payload_l0;
+
+              l3_0 = (u8 *)ip6_0;
+              udp0 = &hdr->udp;
            }
 
+          /* Fix UDP length  and set source port */
+          udp0->length = payload_l0;
+          udp0->src_port = flow_hash0;
+
+          if (csum_offload)
+            {
+              b0->flags |= csum_flags;
+              vnet_buffer (b0)->l3_hdr_offset = l3_0 - b0->data;
+              vnet_buffer (b0)->l4_hdr_offset = (u8 *) udp0 - b0->data;
+            }
+          /* IPv4 UDP checksum only if checksum offload is used */
+          else if (is_ip4)
+            {
+              ip_csum_t sum0 = ip4_0->checksum;
+              sum0 = ip_csum_update (sum0, 0, ip4_0->length, ip4_header_t,
+                  length /* changed member */);
+             if (PREDICT_FALSE (ip4_0_tos))
+               {
+                 sum0 = ip_csum_update (sum0, 0, ip4_0_tos, ip4_header_t,
+                     tos /* changed member */);
+               }
+              ip4_0->checksum = ip_csum_fold (sum0);
+            }
+          /* IPv6 UDP checksum is mandatory */
+          else
+            {
+              int bogus = 0;
+
+              udp0->checksum = ip6_tcp_udp_icmp_compute_checksum
+                (vm, b0, ip6_0, &bogus);
+              ASSERT(bogus == 0);
+              if (udp0->checksum == 0)
+                udp0->checksum = 0xffff;
+            }
+
+          vlib_increment_combined_counter (tx_counter, thread_index,
+              sw_if_index0, 1, len0);
           pkts_encapsulated ++;
-         len0 = vlib_buffer_length_in_chain (vm, b0);
-         stats_n_packets += 1;
-         stats_n_bytes += len0;
-
-         /* Batch stats increment on the same vxlan tunnel so counter is not
-            incremented per packet. Note stats are still incremented for deleted
-            and admin-down tunnel where packets are dropped. It is not worthwhile
-            to check for this rare case and affect normal path performance. */
-         if (PREDICT_FALSE (sw_if_index0 != stats_sw_if_index)) 
-           {
-             stats_n_packets -= 1;
-             stats_n_bytes -= len0;
-             if (stats_n_packets)
-               vlib_increment_combined_counter 
-                 (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                  thread_index, stats_sw_if_index, 
-                  stats_n_packets, stats_n_bytes);
-             stats_n_packets = 1;
-             stats_n_bytes = len0;
-             stats_sw_if_index = sw_if_index0;
-           }
 
           if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) 
             {
@@ -491,15 +449,6 @@ vxlan_encap_inline (vlib_main_t * vm,
                                VXLAN_ENCAP_ERROR_ENCAPSULATED, 
                                pkts_encapsulated);
 
-  /* Increment any remaining batch stats */
-  if (stats_n_packets)
-    {
-      vlib_increment_combined_counter 
-       (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-        thread_index, stats_sw_if_index, stats_n_packets, stats_n_bytes);
-      node->runtime_data[0] = stats_sw_if_index;
-    }
-
   return from_frame->n_vectors;
 }
 
@@ -508,7 +457,10 @@ vxlan4_encap (vlib_main_t * vm,
              vlib_node_runtime_t * node,
              vlib_frame_t * from_frame)
 {
-  return vxlan_encap_inline (vm, node, from_frame, /* is_ip4 */ 1);
+  /* Disable chksum offload as setup overhead in tx node is not worthwhile
+     for ip4 header checksum only, unless udp checksum is also required */
+  return vxlan_encap_inline (vm, node, from_frame, /* is_ip4 */ 1, 
+                            /* csum_offload */ 0);
 }
 
 static uword
@@ -516,7 +468,9 @@ vxlan6_encap (vlib_main_t * vm,
              vlib_node_runtime_t * node,
              vlib_frame_t * from_frame)
 {
-  return vxlan_encap_inline (vm, node, from_frame, /* is_ip4 */ 0);
+  /* Enable checksum offload for ip6 as udp checksum is mandatory, */
+  return vxlan_encap_inline (vm, node, from_frame, /* is_ip4 */ 0, 
+                            /* csum_offload */ 1);
 }
 
 VLIB_REGISTER_NODE (vxlan4_encap_node) = {