summaryrefslogtreecommitdiffstats
path: root/src/plugins/lb
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/lb')
-rw-r--r--src/plugins/lb/node.c21
1 files changed, 12 insertions, 9 deletions
diff --git a/src/plugins/lb/node.c b/src/plugins/lb/node.c
index e19964d2f1c..44c7781e977 100644
--- a/src/plugins/lb/node.c
+++ b/src/plugins/lb/node.c
@@ -387,22 +387,25 @@ lb_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame,
ip4_header_t *ip40;
tcp_header_t *th0;
ip_csum_t csum;
- u32 old_dst;
- u32 old_dscp;
+ u32 old_dst, new_dst;
+ u8 old_tos, new_tos;
ip40 = vlib_buffer_get_current (p0);
old_dst = ip40->dst_address.as_u32;
- old_dscp = ip40->tos;
- ip40->dst_address = lbm->ass[asindex0].address.ip4;
+ new_dst = lbm->ass[asindex0].address.ip4.as_u32;
+ ip40->dst_address.as_u32 = lbm->ass[asindex0].address.ip4.as_u32;
/* Get and rewrite DSCP bit */
+ old_tos = ip40->tos;
+ new_tos = (u8) ((vip0->encap_args.dscp & 0x3F) << 2);
ip40->tos = (u8) ((vip0->encap_args.dscp & 0x3F) << 2);
csum = ip40->checksum;
- csum = ip_csum_sub_even (csum, old_dst);
- csum = ip_csum_sub_even (csum, old_dscp);
- csum = ip_csum_add_even (csum,
- lbm->ass[asindex0].address.ip4.as_u32);
- csum = ip_csum_add_even (csum, ip40->tos);
+ csum = ip_csum_update (csum, old_tos, new_tos,
+ ip4_header_t,
+ tos /* changed member */);
+ csum = ip_csum_update (csum, old_dst, new_dst,
+ ip4_header_t,
+ dst_address /* changed member */);
ip40->checksum = ip_csum_fold (csum);
/* Recomputing L4 checksum after dst-IP modifying */