lb: lb_add_del_vip and lb_add_del_as doesn't work.

Ticket: FDIO-753
Type: fix

Change-Id: I4a8cf06970b658dfa15768459a3ff76571d6dfff
Signed-off-by: Hongjun Ni <hongjun.ni@intel.com>

2 files changed, 14 insertions, 4 deletions

diff --git a/src/plugins/lb/api.c b/src/plugins/lb/api.c
index 67a26d54902..bef6969ed84 100644
--- a/src/plugins/lb/api.c
+++ b/src/plugins/lb/api.c
@@ -93,7 +93,7 @@ vl_api_lb_add_del_vip_t_handler
       mp->protocol = ~0;
     }
 
-  ip_address_decode (&mp->pfx.address, &(args.prefix));
+  memcpy (&(args.prefix.ip6), &mp->pfx.address.un.ip6, sizeof(args.prefix.ip6));
 
   if (mp->is_del) {
     u32 vip_index;
@@ -183,10 +183,20 @@ vl_api_lb_add_del_as_t_handler
   int rv = 0;
   u32 vip_index;
   ip46_address_t vip_ip_prefix;
+
+  /* if port == 0, it means all-port VIP */
+  if (mp->port == 0)
+    {
+      mp->protocol = ~0;
+    }
+
+  memcpy(&vip_ip_prefix.ip6, &mp->pfx.address.un.ip6,
+              sizeof(vip_ip_prefix.ip6));
+
   ip46_address_t as_address;
 
-  ip_address_decode (&mp->pfx.address, &vip_ip_prefix);
-  ip_address_decode (&mp->as_address, &as_address);
+  memcpy(&as_address.ip6, &mp->as_address.un.ip6,
+         sizeof(as_address.ip6));
 
   if ((rv = lb_vip_find_index(&vip_ip_prefix, mp->pfx.len,
                               mp->protocol, ntohs(mp->port), &vip_index)))
diff --git a/src/plugins/lb/lb_test.c b/src/plugins/lb/lb_test.c
index 7176be8c8d8..007303f82c8 100644
--- a/src/plugins/lb/lb_test.c
+++ b/src/plugins/lb/lb_test.c
@@ -268,7 +268,7 @@ static int api_lb_add_del_as (vat_main_t * vam)
   mp->pfx.len = vip_plen;
   mp->protocol = (u8)protocol;
   mp->port = htons((u16)port);
-  clib_memcpy (&mp->as_address, &as_addr, sizeof (as_addr));
+  clib_memcpy (&mp->as_address.un.ip6, &as_addr.ip6, sizeof (as_addr.ip6));
   mp->is_del = is_del;
   mp->is_flush = is_flush;