vlib: refactor checksum offload support

[vpp.git] / src / vnet / devices / af_packet / node.c

diff --git a/src/vnet/devices/af_packet/node.c b/src/vnet/devices/af_packet/node.c
index f4db639..9f1782e 100644 (file)
--- a/src/vnet/devices/af_packet/node.c
+++ b/src/vnet/devices/af_packet/node.c
@@ -120,6 +120,7 @@ static_always_inline void
 mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
 {
   ethernet_header_t *eth = vlib_buffer_get_current (b);
+  u32 oflags = 0;
   if (clib_net_to_host_u16 (eth->type) == ETHERNET_TYPE_IP4)
     {
       ip4_header_t *ip4 =
@@ -127,7 +128,7 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
       b->flags |= VNET_BUFFER_F_IS_IP4;
       if (ip4->protocol == IP_PROTOCOL_TCP)
        {
-         b->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
+         oflags |= VNET_BUFFER_OFFLOAD_F_TCP_CKSUM;
          tcp_header_t *tcp = (tcp_header_t *) (vlib_buffer_get_current (b) +
                                                sizeof (ethernet_header_t) +
                                                ip4_header_bytes (ip4));
@@ -136,7 +137,7 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
        }
       else if (ip4->protocol == IP_PROTOCOL_UDP)
        {
-         b->flags |= VNET_BUFFER_F_OFFLOAD_UDP_CKSUM;
+         oflags |= VNET_BUFFER_OFFLOAD_F_UDP_CKSUM;
          udp_header_t *udp = (udp_header_t *) (vlib_buffer_get_current (b) +
                                                sizeof (ethernet_header_t) +
                                                ip4_header_bytes (ip4));
@@ -146,6 +147,8 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
       vnet_buffer (b)->l3_hdr_offset = sizeof (ethernet_header_t);
       vnet_buffer (b)->l4_hdr_offset =
        sizeof (ethernet_header_t) + ip4_header_bytes (ip4);
+      if (oflags)
+       vnet_buffer_offload_flags_set (b, oflags);
     }
   else if (clib_net_to_host_u16 (eth->type) == ETHERNET_TYPE_IP6)
     {
@@ -165,7 +168,7 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
        }
       if (ip6->protocol == IP_PROTOCOL_TCP)
        {
-         b->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
+         oflags |= VNET_BUFFER_OFFLOAD_F_TCP_CKSUM;
          tcp_header_t *tcp =
            (tcp_header_t *) (vlib_buffer_get_current (b) +
                              sizeof (ethernet_header_t) + ip6_hdr_len);
@@ -174,7 +177,7 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
        }
       else if (ip6->protocol == IP_PROTOCOL_UDP)
        {
-         b->flags |= VNET_BUFFER_F_OFFLOAD_UDP_CKSUM;
+         oflags |= VNET_BUFFER_OFFLOAD_F_UDP_CKSUM;
          udp_header_t *udp =
            (udp_header_t *) (vlib_buffer_get_current (b) +
                              sizeof (ethernet_header_t) + ip6_hdr_len);
@@ -184,6 +187,8 @@ mark_tcp_udp_cksum_calc (vlib_buffer_t *b, u8 *l4_hdr_sz)
       vnet_buffer (b)->l3_hdr_offset = sizeof (ethernet_header_t);
       vnet_buffer (b)->l4_hdr_offset =
        sizeof (ethernet_header_t) + ip6_hdr_len;
+      if (oflags)
+       vnet_buffer_offload_flags_set (b, oflags);
     }
 }