vxlan: Lookup FIB in either IPv4 or IPv6 families.
[vpp.git] / src / vnet / vxlan / encap.c
index 5b63064..062e338 100644 (file)
@@ -1,3 +1,4 @@
+
 /*
  * Copyright (c) 2015 Cisco and/or its affiliates.
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -58,26 +59,19 @@ 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 cpu_index = os_get_cpu_number();
+  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;
@@ -91,6 +85,17 @@ vxlan_encap_inline (vlib_main_t * vm,
   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);
+
+  word const underlay_hdr_len = is_ip4 ?
+    sizeof(ip4_vxlan_header_t) : sizeof(ip6_vxlan_header_t);
+  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)
     {
       u32 n_left_to_next;
@@ -100,20 +105,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,8 +119,14 @@ vxlan_encap_inline (vlib_main_t * vm,
            CLIB_PREFETCH (p3->data, 2*CLIB_CACHE_LINE_BYTES, LOAD);
          }
 
-         bi0 = from[0];
-         bi1 = from[1];
+         u32 bi0 = from[0];
+         u32 bi1 = from[1];
+
+         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);
+
          to_next[0] = bi0;
          to_next[1] = bi1;
          from += 2;
@@ -137,12 +134,6 @@ vxlan_encap_inline (vlib_main_t * vm,
          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);
-
          /* Get next node index and adj index from tunnel next_dpo */
          if (sw_if_index0 != vnet_buffer(b0)->sw_if_index[VLIB_TX])
            {
@@ -165,158 +156,133 @@ vxlan_encap_inline (vlib_main_t * vm,
            }
           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));
+          ASSERT(vec_len(t0->rewrite) == underlay_hdr_len);
+          ASSERT(vec_len(t1->rewrite) == 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);
+
+          ip4_header_t * ip4_0, * ip4_1;
+          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 = vlib_buffer_get_current(b0);
+              ip4_vxlan_header_t * rewrite0 = (void *)t0->rewrite;
+              ip4_vxlan_header_t * hdr1 = vlib_buffer_get_current(b1);
+              ip4_vxlan_header_t * rewrite1 = (void *)t1->rewrite;
+              *hdr0 = *rewrite0;
+              *hdr1 = *rewrite1;
 
              /* 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);
+
+              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 = vlib_buffer_get_current(b0);
+              ip6_vxlan_header_t * rewrite0 = (void *) t0->rewrite;
+              ip6_vxlan_header_t * hdr1 = vlib_buffer_get_current(b0);
+              ip6_vxlan_header_t * rewrite1 = (void *) t1->rewrite;
+              *hdr0 = *rewrite0;
+              *hdr1 = *rewrite1;
 
-             /* 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;
+          /* 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 */);
+              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 */);
+              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;
+            }
 
          /* 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) 
-               {
-                 if (stats_n_packets) 
-                   vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      cpu_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 
-               {
-                 vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      cpu_index, sw_if_index0, 1, len0);
-                 vlib_increment_combined_counter 
-                     (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-                      cpu_index, sw_if_index1, 1, len1);
-               }
-           }
+          if (sw_if_index0 == sw_if_index1)
+            {
+              if (PREDICT_FALSE(sw_if_index0 != stats_sw_if_index))
+                {
+                  if (stats_n_packets) 
+                    {
+                      vlib_increment_combined_counter (tx_counter, thread_index,
+                          stats_sw_if_index, stats_n_packets, stats_n_bytes);
+                      stats_n_packets = stats_n_bytes = 0;
+                    }
+                  stats_sw_if_index = sw_if_index0;
+                }
+              stats_n_packets += 2;
+              stats_n_bytes += len0 + len1;
+            }
+          else
+            {
+              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,29 +307,16 @@ 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;
+         u32 bi0 = from[0];
+         vlib_buffer_t * b0 = vlib_get_buffer (vm, bi0);
+          u32 flow_hash0 = vnet_l2_compute_flow_hash(b0);
 
-         bi0 = from[0];
          to_next[0] = bi0;
          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);
-
          /* Get next node index and adj index from tunnel next_dpo */
          if (sw_if_index0 != vnet_buffer(b0)->sw_if_index[VLIB_TX])
            {
@@ -375,83 +328,72 @@ vxlan_encap_inline (vlib_main_t * vm,
            }
          vnet_buffer(b0)->ip.adj_index[VLIB_TX] = t0->next_dpo.dpoi_index;
 
-          /* Apply the rewrite string. $$$$ vnet_rewrite? */
-          vlib_buffer_advance (b0, -(word)_vec_len(t0->rewrite));
+          ASSERT(vec_len(t0->rewrite) == underlay_hdr_len);
+          vlib_buffer_advance (b0, -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;
+          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 * rewrite = (void *)t0->rewrite;
+              ip4_vxlan_header_t * hdr = vlib_buffer_get_current(b0);
+              *hdr = *rewrite;
 
              /* 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);
 
+              l3_0 = (u8*)ip4_0;
+             udp0 = &hdr->udp;
+           }
          else /* ip6 path */
            {
-              int bogus = 0;
+              ip6_vxlan_header_t * hdr = vlib_buffer_get_current(b0);
+              ip6_vxlan_header_t * rewrite = (void *) t0->rewrite;
+              *hdr = *rewrite;
 
-             /* 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;
            }
 
-          pkts_encapsulated ++;
-         len0 = vlib_buffer_length_in_chain (vm, b0);
-         stats_n_packets += 1;
-         stats_n_bytes += len0;
+          /* 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 */);
+              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;
+            }
 
          /* Batch stats increment on the same vxlan tunnel so counter is not
             incremented per packet. Note stats are still incremented for deleted
@@ -459,17 +401,17 @@ vxlan_encap_inline (vlib_main_t * vm,
             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,
-                  cpu_index, stats_sw_if_index, 
-                  stats_n_packets, stats_n_bytes);
-             stats_n_packets = 1;
-             stats_n_bytes = len0;
+                {
+                  vlib_increment_combined_counter (tx_counter, thread_index,
+                      stats_sw_if_index, stats_n_packets, stats_n_bytes);
+                  stats_n_bytes = stats_n_packets = 0;
+                }
              stats_sw_if_index = sw_if_index0;
            }
+         stats_n_packets += 1;
+         stats_n_bytes += len0;
+          pkts_encapsulated ++;
 
           if (PREDICT_FALSE(b0->flags & VLIB_BUFFER_IS_TRACED)) 
             {
@@ -494,9 +436,8 @@ vxlan_encap_inline (vlib_main_t * vm,
   /* Increment any remaining batch stats */
   if (stats_n_packets)
     {
-      vlib_increment_combined_counter 
-       (im->combined_sw_if_counters + VNET_INTERFACE_COUNTER_TX,
-        cpu_index, stats_sw_if_index, stats_n_packets, stats_n_bytes);
+      vlib_increment_combined_counter (tx_counter, thread_index,
+          stats_sw_if_index, stats_n_packets, stats_n_bytes);
       node->runtime_data[0] = stats_sw_if_index;
     }
 
@@ -508,7 +449,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 +460,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) = {