lb: Loadbalancer plugin, correctly rewrite checksum in L3DSR, non-TCP cases

[vpp.git] / src / plugins / lb / lb.c

diff --git a/src/plugins/lb/lb.c b/src/plugins/lb/lb.c
index 2fc38e2..21a1484 100644 (file)
--- a/src/plugins/lb/lb.c
+++ b/src/plugins/lb/lb.c
@@ -17,7 +17,7 @@
 #include <vnet/plugin/plugin.h>
 #include <vpp/app/version.h>
 #include <vnet/api_errno.h>
-#include <vnet/udp/udp.h>
+#include <vnet/udp/udp_local.h>
 #include <vppinfra/lock.h>
 
 //GC runs at most once every so many seconds
@@ -907,7 +907,7 @@ static void lb_vip_add_adjacency(lb_main_t *lbm, lb_vip_t *vip,
                                       vip->protocol, vip->port, &vip_idx))
         {
           lb_vip_t *exists_vip = lb_vip_get_by_index(vip_idx);
-          *vip_prefix_index = exists_vip->vip_prefix_index;
+          *vip_prefix_index = exists_vip ? exists_vip->vip_prefix_index : ~0;
           return;
         }