src/vnet/lisp-cp/control.c

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <vlibmemory/api.h>
  17 #include <vnet/lisp-cp/control.h>
  18 #include <vnet/lisp-cp/packets.h>
  19 #include <vnet/lisp-cp/lisp_msg_serdes.h>
  20 #include <vnet/lisp-gpe/lisp_gpe_fwd_entry.h>
  21 #include <vnet/lisp-gpe/lisp_gpe_tenant.h>
  22 #include <vnet/lisp-gpe/lisp_gpe_tunnel.h>
  23 #include <vnet/fib/fib_entry.h>
  24 #include <vnet/fib/fib_table.h>
  25 #include <vnet/ethernet/arp_packet.h>
  26 #include <vnet/ethernet/packet.h>
  27
  28 #include <openssl/evp.h>
  29 #include <openssl/hmac.h>
  30
  31 #define MAX_VALUE_U24 0xffffff
  32
  33 /* mapping timer control constants (in seconds) */
  34 #define TIME_UNTIL_REFETCH_OR_DELETE  20
  35 #define MAPPING_TIMEOUT (((m->ttl) * 60) - TIME_UNTIL_REFETCH_OR_DELETE)
  36
  37 lisp_cp_main_t lisp_control_main;
  38
  39 u8 *format_lisp_cp_input_trace (u8 * s, va_list * args);
  40 static void *send_map_request_thread_fn (void *arg);
  41
  42 typedef enum
  43 {
  44   LISP_CP_INPUT_NEXT_DROP,
  45   LISP_CP_INPUT_N_NEXT,
  46 } lisp_cp_input_next_t;
  47
  48 typedef struct
  49 {
  50   u8 is_resend;
  51   gid_address_t seid;
  52   gid_address_t deid;
  53   u8 smr_invoked;
  54 } map_request_args_t;
  55
  56 u8
  57 vnet_lisp_get_map_request_mode (void)
  58 {
  59   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
  60   return lcm->map_request_mode;
  61 }
  62
  63 static u16
  64 auth_data_len_by_key_id (lisp_key_type_t key_id)
  65 {
  66   switch (key_id)
  67     {
  68     case HMAC_SHA_1_96:
  69       return SHA1_AUTH_DATA_LEN;
  70     case HMAC_SHA_256_128:
  71       return SHA256_AUTH_DATA_LEN;
  72     default:
  73       clib_warning ("unsupported key type: %d!", key_id);
  74       return (u16) ~ 0;
  75     }
  76   return (u16) ~ 0;
  77 }
  78
  79 static const EVP_MD *
  80 get_encrypt_fcn (lisp_key_type_t key_id)
  81 {
  82   switch (key_id)
  83     {
  84     case HMAC_SHA_1_96:
  85       return EVP_sha1 ();
  86     case HMAC_SHA_256_128:
  87       return EVP_sha256 ();
  88     default:
  89       clib_warning ("unsupported encryption key type: %d!", key_id);
  90       break;
  91     }
  92   return 0;
  93 }
  94
  95 static int
  96 queue_map_request (gid_address_t * seid, gid_address_t * deid,
  97                    u8 smr_invoked, u8 is_resend);
  98
  99 ip_interface_address_t *
 100 ip_interface_get_first_interface_address (ip_lookup_main_t * lm,
 101                                           u32 sw_if_index, u8 loop)
 102 {
 103   vnet_main_t *vnm = vnet_get_main ();
 104   vnet_sw_interface_t *swif = vnet_get_sw_interface (vnm, sw_if_index);
 105   if (loop && swif->flags & VNET_SW_INTERFACE_FLAG_UNNUMBERED)
 106     sw_if_index = swif->unnumbered_sw_if_index;
 107   u32 ia =
 108     (vec_len ((lm)->if_address_pool_index_by_sw_if_index) > (sw_if_index)) ?
 109     vec_elt ((lm)->if_address_pool_index_by_sw_if_index, (sw_if_index)) :
 110     (u32) ~ 0;
 111   return pool_elt_at_index ((lm)->if_address_pool, ia);
 112 }
 113
 114 void *
 115 ip_interface_get_first_address (ip_lookup_main_t * lm, u32 sw_if_index,
 116                                 u8 version)
 117 {
 118   ip_interface_address_t *ia;
 119
 120   ia = ip_interface_get_first_interface_address (lm, sw_if_index, 1);
 121   if (!ia)
 122     return 0;
 123   return ip_interface_address_get_address (lm, ia);
 124 }
 125
 126 int
 127 ip_interface_get_first_ip_address (lisp_cp_main_t * lcm, u32 sw_if_index,
 128                                    u8 version, ip_address_t * result)
 129 {
 130   ip_lookup_main_t *lm;
 131   void *addr;
 132
 133   lm = (version == IP4) ? &lcm->im4->lookup_main : &lcm->im6->lookup_main;
 134   addr = ip_interface_get_first_address (lm, sw_if_index, version);
 135   if (!addr)
 136     return 0;
 137
 138   ip_address_set (result, addr, version);
 139   return 1;
 140 }
 141
 142 /**
 143  * convert from a LISP address to a FIB prefix
 144  */
 145 void
 146 ip_address_to_fib_prefix (const ip_address_t * addr, fib_prefix_t * prefix)
 147 {
 148   if (addr->version == IP4)
 149     {
 150       prefix->fp_len = 32;
 151       prefix->fp_proto = FIB_PROTOCOL_IP4;
 152       clib_memset (&prefix->fp_addr.pad, 0, sizeof (prefix->fp_addr.pad));
 153       memcpy (&prefix->fp_addr.ip4, &addr->ip, sizeof (prefix->fp_addr.ip4));
 154     }
 155   else
 156     {
 157       prefix->fp_len = 128;
 158       prefix->fp_proto = FIB_PROTOCOL_IP6;
 159       memcpy (&prefix->fp_addr.ip6, &addr->ip, sizeof (prefix->fp_addr.ip6));
 160     }
 161 }
 162
 163 /**
 164  * convert from a LISP to a FIB prefix
 165  */
 166 void
 167 ip_prefix_to_fib_prefix (const ip_prefix_t * ip_prefix,
 168                          fib_prefix_t * fib_prefix)
 169 {
 170   ip_address_to_fib_prefix (&ip_prefix->addr, fib_prefix);
 171   fib_prefix->fp_len = ip_prefix->len;
 172 }
 173
 174 /**
 175  * Find the sw_if_index of the interface that would be used to egress towards
 176  * dst.
 177  */
 178 u32
 179 ip_fib_get_egress_iface_for_dst (lisp_cp_main_t * lcm, ip_address_t * dst)
 180 {
 181   fib_node_index_t fei;
 182   fib_prefix_t prefix;
 183
 184   ip_address_to_fib_prefix (dst, &prefix);
 185
 186   fei = fib_table_lookup (0, &prefix);
 187
 188   return (fib_entry_get_resolving_interface (fei));
 189 }
 190
 191 /**
 192  * Find first IP of the interface that would be used to egress towards dst.
 193  * Returns 1 if the address is found 0 otherwise.
 194  */
 195 int
 196 ip_fib_get_first_egress_ip_for_dst (lisp_cp_main_t * lcm, ip_address_t * dst,
 197                                     ip_address_t * result)
 198 {
 199   u32 si;
 200   ip_lookup_main_t *lm;
 201   void *addr = 0;
 202   u8 ipver;
 203
 204   ASSERT (result != 0);
 205
 206   ipver = ip_addr_version (dst);
 207
 208   lm = (ipver == IP4) ? &lcm->im4->lookup_main : &lcm->im6->lookup_main;
 209   si = ip_fib_get_egress_iface_for_dst (lcm, dst);
 210
 211   if ((u32) ~ 0 == si)
 212     return 0;
 213
 214   /* find the first ip address */
 215   addr = ip_interface_get_first_address (lm, si, ipver);
 216   if (0 == addr)
 217     return 0;
 218
 219   ip_address_set (result, addr, ipver);
 220   return 1;
 221 }
 222
 223 static int
 224 dp_add_del_iface (lisp_cp_main_t * lcm, u32 vni, u8 is_l2, u8 is_add,
 225                   u8 with_default_route)
 226 {
 227   uword *dp_table;
 228
 229   if (!is_l2)
 230     {
 231       dp_table = hash_get (lcm->table_id_by_vni, vni);
 232
 233       if (!dp_table)
 234         {
 235           clib_warning ("vni %d not associated to a vrf!", vni);
 236           return VNET_API_ERROR_INVALID_VALUE;
 237         }
 238     }
 239   else
 240     {
 241       dp_table = hash_get (lcm->bd_id_by_vni, vni);
 242       if (!dp_table)
 243         {
 244           clib_warning ("vni %d not associated to a bridge domain!", vni);
 245           return VNET_API_ERROR_INVALID_VALUE;
 246         }
 247     }
 248
 249   /* enable/disable data-plane interface */
 250   if (is_add)
 251     {
 252       if (is_l2)
 253         lisp_gpe_tenant_l2_iface_add_or_lock (vni, dp_table[0]);
 254       else
 255         lisp_gpe_tenant_l3_iface_add_or_lock (vni, dp_table[0],
 256                                               with_default_route);
 257     }
 258   else
 259     {
 260       if (is_l2)
 261         lisp_gpe_tenant_l2_iface_unlock (vni);
 262       else
 263         lisp_gpe_tenant_l3_iface_unlock (vni);
 264     }
 265
 266   return 0;
 267 }
 268
 269 static void
 270 dp_del_fwd_entry (lisp_cp_main_t * lcm, u32 dst_map_index)
 271 {
 272   vnet_lisp_gpe_add_del_fwd_entry_args_t _a, *a = &_a;
 273   fwd_entry_t *fe = 0;
 274   uword *feip = 0;
 275   clib_memset (a, 0, sizeof (*a));
 276
 277   feip = hash_get (lcm->fwd_entry_by_mapping_index, dst_map_index);
 278   if (!feip)
 279     return;
 280
 281   fe = pool_elt_at_index (lcm->fwd_entry_pool, feip[0]);
 282
 283   /* delete dp fwd entry */
 284   u32 sw_if_index;
 285   a->is_add = 0;
 286   a->locator_pairs = fe->locator_pairs;
 287   a->vni = gid_address_vni (&fe->reid);
 288   gid_address_copy (&a->rmt_eid, &fe->reid);
 289   if (fe->is_src_dst)
 290     gid_address_copy (&a->lcl_eid, &fe->leid);
 291
 292   vnet_lisp_gpe_del_fwd_counters (a, feip[0]);
 293   vnet_lisp_gpe_add_del_fwd_entry (a, &sw_if_index);
 294
 295   /* delete entry in fwd table */
 296   hash_unset (lcm->fwd_entry_by_mapping_index, dst_map_index);
 297   vec_free (fe->locator_pairs);
 298   pool_put (lcm->fwd_entry_pool, fe);
 299 }
 300
 301 /**
 302  * Finds first remote locator with best (lowest) priority that has a local
 303  * peer locator with an underlying route to it.
 304  *
 305  */
 306 static u32
 307 get_locator_pairs (lisp_cp_main_t * lcm, mapping_t * lcl_map,
 308                    mapping_t * rmt_map, locator_pair_t ** locator_pairs)
 309 {
 310   u32 i, limitp = 0, li, found = 0, esi;
 311   locator_set_t *rmt_ls, *lcl_ls;
 312   ip_address_t _lcl_addr, *lcl_addr = &_lcl_addr;
 313   locator_t *lp, *rmt = 0;
 314   uword *checked = 0;
 315   locator_pair_t pair;
 316
 317   rmt_ls =
 318     pool_elt_at_index (lcm->locator_set_pool, rmt_map->locator_set_index);
 319   lcl_ls =
 320     pool_elt_at_index (lcm->locator_set_pool, lcl_map->locator_set_index);
 321
 322   if (!rmt_ls || vec_len (rmt_ls->locator_indices) == 0)
 323     return 0;
 324
 325   while (1)
 326     {
 327       rmt = 0;
 328
 329       /* find unvisited remote locator with best priority */
 330       for (i = 0; i < vec_len (rmt_ls->locator_indices); i++)
 331         {
 332           if (0 != hash_get (checked, i))
 333             continue;
 334
 335           li = vec_elt (rmt_ls->locator_indices, i);
 336           lp = pool_elt_at_index (lcm->locator_pool, li);
 337
 338           /* we don't support non-IP locators for now */
 339           if (gid_address_type (&lp->address) != GID_ADDR_IP_PREFIX)
 340             continue;
 341
 342           if ((found && lp->priority == limitp)
 343               || (!found && lp->priority >= limitp))
 344             {
 345               rmt = lp;
 346
 347               /* don't search for locators with lower priority and don't
 348                * check this locator again*/
 349               limitp = lp->priority;
 350               hash_set (checked, i, 1);
 351               break;
 352             }
 353         }
 354       /* check if a local locator with a route to remote locator exists */
 355       if (rmt != 0)
 356         {
 357           /* find egress sw_if_index for rmt locator */
 358           esi =
 359             ip_fib_get_egress_iface_for_dst (lcm,
 360                                              &gid_address_ip (&rmt->address));
 361           if ((u32) ~ 0 == esi)
 362             continue;
 363
 364           for (i = 0; i < vec_len (lcl_ls->locator_indices); i++)
 365             {
 366               li = vec_elt (lcl_ls->locator_indices, i);
 367               locator_t *sl = pool_elt_at_index (lcm->locator_pool, li);
 368
 369               /* found local locator with the needed sw_if_index */
 370               if (sl->sw_if_index == esi)
 371                 {
 372                   /* and it has an address */
 373                   if (0 == ip_interface_get_first_ip_address (lcm,
 374                                                               sl->sw_if_index,
 375                                                               gid_address_ip_version
 376                                                               (&rmt->address),
 377                                                               lcl_addr))
 378                     continue;
 379
 380                   clib_memset (&pair, 0, sizeof (pair));
 381                   ip_address_copy (&pair.rmt_loc,
 382                                    &gid_address_ip (&rmt->address));
 383                   ip_address_copy (&pair.lcl_loc, lcl_addr);
 384                   pair.weight = rmt->weight;
 385                   pair.priority = rmt->priority;
 386                   vec_add1 (locator_pairs[0], pair);
 387                   found = 1;
 388                 }
 389             }
 390         }
 391       else
 392         break;
 393     }
 394
 395   hash_free (checked);
 396   return found;
 397 }
 398
 399 static void
 400 gid_address_sd_to_flat (gid_address_t * dst, gid_address_t * src,
 401                         fid_address_t * fid)
 402 {
 403   ASSERT (GID_ADDR_SRC_DST == gid_address_type (src));
 404
 405   dst[0] = src[0];
 406
 407   switch (fid_addr_type (fid))
 408     {
 409     case FID_ADDR_IP_PREF:
 410       gid_address_type (dst) = GID_ADDR_IP_PREFIX;
 411       gid_address_ippref (dst) = fid_addr_ippref (fid);
 412       break;
 413     case FID_ADDR_MAC:
 414       gid_address_type (dst) = GID_ADDR_MAC;
 415       mac_copy (gid_address_mac (dst), fid_addr_mac (fid));
 416       break;
 417     default:
 418       clib_warning ("Unsupported fid type %d!", fid_addr_type (fid));
 419       break;
 420     }
 421 }
 422
 423 u8
 424 vnet_lisp_map_register_state_get (void)
 425 {
 426   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 427   return lcm->map_registering;
 428 }
 429
 430 u8
 431 vnet_lisp_rloc_probe_state_get (void)
 432 {
 433   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 434   return lcm->rloc_probing;
 435 }
 436
 437 static void
 438 dp_add_fwd_entry (lisp_cp_main_t * lcm, u32 src_map_index, u32 dst_map_index)
 439 {
 440   vnet_lisp_gpe_add_del_fwd_entry_args_t _a, *a = &_a;
 441   gid_address_t *rmt_eid, *lcl_eid;
 442   mapping_t *lcl_map, *rmt_map;
 443   u32 sw_if_index, **rmts, rmts_idx;
 444   uword *feip = 0, *dpid, *rmts_stored_idxp = 0;
 445   fwd_entry_t *fe;
 446   u8 type, is_src_dst = 0;
 447   int rv;
 448
 449   clib_memset (a, 0, sizeof (*a));
 450
 451   /* remove entry if it already exists */
 452   feip = hash_get (lcm->fwd_entry_by_mapping_index, dst_map_index);
 453   if (feip)
 454     dp_del_fwd_entry (lcm, dst_map_index);
 455
 456   /*
 457    * Determine local mapping and eid
 458    */
 459   if (lcm->flags & LISP_FLAG_PITR_MODE)
 460     {
 461       if (lcm->pitr_map_index != ~0)
 462         lcl_map = pool_elt_at_index (lcm->mapping_pool, lcm->pitr_map_index);
 463       else
 464         {
 465           clib_warning ("no PITR mapping configured!");
 466           return;
 467         }
 468     }
 469   else
 470     lcl_map = pool_elt_at_index (lcm->mapping_pool, src_map_index);
 471   lcl_eid = &lcl_map->eid;
 472
 473   /*
 474    * Determine remote mapping and eid
 475    */
 476   rmt_map = pool_elt_at_index (lcm->mapping_pool, dst_map_index);
 477   rmt_eid = &rmt_map->eid;
 478
 479   /*
 480    * Build and insert data plane forwarding entry
 481    */
 482   a->is_add = 1;
 483
 484   if (MR_MODE_SRC_DST == lcm->map_request_mode)
 485     {
 486       if (GID_ADDR_SRC_DST == gid_address_type (rmt_eid))
 487         {
 488           gid_address_sd_to_flat (&a->rmt_eid, rmt_eid,
 489                                   &gid_address_sd_dst (rmt_eid));
 490           gid_address_sd_to_flat (&a->lcl_eid, rmt_eid,
 491                                   &gid_address_sd_src (rmt_eid));
 492         }
 493       else
 494         {
 495           gid_address_copy (&a->rmt_eid, rmt_eid);
 496           gid_address_copy (&a->lcl_eid, lcl_eid);
 497         }
 498       is_src_dst = 1;
 499     }
 500   else
 501     gid_address_copy (&a->rmt_eid, rmt_eid);
 502
 503   a->vni = gid_address_vni (&a->rmt_eid);
 504   a->is_src_dst = is_src_dst;
 505
 506   /* get vrf or bd_index associated to vni */
 507   type = gid_address_type (&a->rmt_eid);
 508   if (GID_ADDR_IP_PREFIX == type)
 509     {
 510       dpid = hash_get (lcm->table_id_by_vni, a->vni);
 511       if (!dpid)
 512         {
 513           clib_warning ("vni %d not associated to a vrf!", a->vni);
 514           return;
 515         }
 516       a->table_id = dpid[0];
 517     }
 518   else if (GID_ADDR_MAC == type)
 519     {
 520       dpid = hash_get (lcm->bd_id_by_vni, a->vni);
 521       if (!dpid)
 522         {
 523           clib_warning ("vni %d not associated to a bridge domain !", a->vni);
 524           return;
 525         }
 526       a->bd_id = dpid[0];
 527     }
 528
 529   /* find best locator pair that 1) verifies LISP policy 2) are connected */
 530   rv = get_locator_pairs (lcm, lcl_map, rmt_map, &a->locator_pairs);
 531
 532   /* Either rmt mapping is negative or we can't find underlay path.
 533    * Try again with petr if configured */
 534   if (rv == 0 && (lcm->flags & LISP_FLAG_USE_PETR))
 535     {
 536       rmt_map = lisp_get_petr_mapping (lcm);
 537       rv = get_locator_pairs (lcm, lcl_map, rmt_map, &a->locator_pairs);
 538     }
 539
 540   /* negative entry */
 541   if (rv == 0)
 542     {
 543       a->is_negative = 1;
 544       a->action = rmt_map->action;
 545     }
 546
 547   rv = vnet_lisp_gpe_add_del_fwd_entry (a, &sw_if_index);
 548   if (rv)
 549     {
 550       if (a->locator_pairs)
 551         vec_free (a->locator_pairs);
 552       return;
 553     }
 554
 555   /* add tunnel to fwd entry table */
 556   pool_get (lcm->fwd_entry_pool, fe);
 557   vnet_lisp_gpe_add_fwd_counters (a, fe - lcm->fwd_entry_pool);
 558
 559   fe->locator_pairs = a->locator_pairs;
 560   gid_address_copy (&fe->reid, &a->rmt_eid);
 561
 562   if (is_src_dst)
 563     gid_address_copy (&fe->leid, &a->lcl_eid);
 564   else
 565     gid_address_copy (&fe->leid, lcl_eid);
 566
 567   fe->is_src_dst = is_src_dst;
 568   hash_set (lcm->fwd_entry_by_mapping_index, dst_map_index,
 569             fe - lcm->fwd_entry_pool);
 570
 571   /* Add rmt mapping to the vector of adjacent mappings to lcl mapping */
 572   rmts_stored_idxp =
 573     hash_get (lcm->lcl_to_rmt_adjs_by_lcl_idx, src_map_index);
 574   if (!rmts_stored_idxp)
 575     {
 576       pool_get (lcm->lcl_to_rmt_adjacencies, rmts);
 577       clib_memset (rmts, 0, sizeof (*rmts));
 578       rmts_idx = rmts - lcm->lcl_to_rmt_adjacencies;
 579       hash_set (lcm->lcl_to_rmt_adjs_by_lcl_idx, src_map_index, rmts_idx);
 580     }
 581   else
 582     {
 583       rmts_idx = (u32) (*rmts_stored_idxp);
 584       rmts = pool_elt_at_index (lcm->lcl_to_rmt_adjacencies, rmts_idx);
 585     }
 586   vec_add1 (rmts[0], dst_map_index);
 587 }
 588
 589 typedef struct
 590 {
 591   u32 si;
 592   u32 di;
 593 } fwd_entry_mt_arg_t;
 594
 595 static void *
 596 dp_add_fwd_entry_thread_fn (void *arg)
 597 {
 598   fwd_entry_mt_arg_t *a = arg;
 599   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 600   dp_add_fwd_entry (lcm, a->si, a->di);
 601   return 0;
 602 }
 603
 604 static int
 605 dp_add_fwd_entry_from_mt (u32 si, u32 di)
 606 {
 607   fwd_entry_mt_arg_t a;
 608
 609   clib_memset (&a, 0, sizeof (a));
 610   a.si = si;
 611   a.di = di;
 612
 613   vl_api_rpc_call_main_thread (dp_add_fwd_entry_thread_fn,
 614                                (u8 *) & a, sizeof (a));
 615   return 0;
 616 }
 617
 618 /**
 619  * Returns vector of adjacencies.
 620  *
 621  * The caller must free the vector returned by this function.
 622  *
 623  * @param vni virtual network identifier
 624  * @return vector of adjacencies
 625  */
 626 lisp_adjacency_t *
 627 vnet_lisp_adjacencies_get_by_vni (u32 vni)
 628 {
 629   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 630   fwd_entry_t *fwd;
 631   lisp_adjacency_t *adjs = 0, adj;
 632
 633   /* *INDENT-OFF* */
 634   pool_foreach(fwd, lcm->fwd_entry_pool,
 635   ({
 636     if (gid_address_vni (&fwd->reid) != vni)
 637       continue;
 638
 639     gid_address_copy (&adj.reid, &fwd->reid);
 640     gid_address_copy (&adj.leid, &fwd->leid);
 641     vec_add1 (adjs, adj);
 642   }));
 643   /* *INDENT-ON* */
 644
 645   return adjs;
 646 }
 647
 648 static lisp_msmr_t *
 649 get_map_server (ip_address_t * a)
 650 {
 651   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 652   lisp_msmr_t *m;
 653
 654   vec_foreach (m, lcm->map_servers)
 655   {
 656     if (!ip_address_cmp (&m->address, a))
 657       {
 658         return m;
 659       }
 660   }
 661   return 0;
 662 }
 663
 664 static lisp_msmr_t *
 665 get_map_resolver (ip_address_t * a)
 666 {
 667   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 668   lisp_msmr_t *m;
 669
 670   vec_foreach (m, lcm->map_resolvers)
 671   {
 672     if (!ip_address_cmp (&m->address, a))
 673       {
 674         return m;
 675       }
 676   }
 677   return 0;
 678 }
 679
 680 int
 681 vnet_lisp_add_del_map_server (ip_address_t * addr, u8 is_add)
 682 {
 683   u32 i;
 684   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 685   lisp_msmr_t _ms, *ms = &_ms;
 686
 687   if (vnet_lisp_enable_disable_status () == 0)
 688     {
 689       clib_warning ("LISP is disabled!");
 690       return VNET_API_ERROR_LISP_DISABLED;
 691     }
 692
 693   if (is_add)
 694     {
 695       if (get_map_server (addr))
 696         {
 697           clib_warning ("map-server %U already exists!", format_ip_address,
 698                         addr);
 699           return -1;
 700         }
 701
 702       clib_memset (ms, 0, sizeof (*ms));
 703       ip_address_copy (&ms->address, addr);
 704       vec_add1 (lcm->map_servers, ms[0]);
 705
 706       if (vec_len (lcm->map_servers) == 1)
 707         lcm->do_map_server_election = 1;
 708     }
 709   else
 710     {
 711       for (i = 0; i < vec_len (lcm->map_servers); i++)
 712         {
 713           ms = vec_elt_at_index (lcm->map_servers, i);
 714           if (!ip_address_cmp (&ms->address, addr))
 715             {
 716               if (!ip_address_cmp (&ms->address, &lcm->active_map_server))
 717                 lcm->do_map_server_election = 1;
 718
 719               vec_del1 (lcm->map_servers, i);
 720               break;
 721             }
 722         }
 723     }
 724
 725   return 0;
 726 }
 727
 728 /**
 729  * Add/remove mapping to/from map-cache. Overwriting not allowed.
 730  */
 731 int
 732 vnet_lisp_map_cache_add_del (vnet_lisp_add_del_mapping_args_t * a,
 733                              u32 * map_index_result)
 734 {
 735   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 736   u32 mi, *map_indexp, map_index, i;
 737   u32 **rmts = 0, *remote_idxp, rmts_itr, remote_idx;
 738   uword *rmts_idxp;
 739   mapping_t *m, *old_map;
 740   u32 **eid_indexes;
 741
 742   if (gid_address_type (&a->eid) == GID_ADDR_NSH)
 743     {
 744       if (gid_address_vni (&a->eid) != 0)
 745         {
 746           clib_warning ("Supported only default VNI for NSH!");
 747           return VNET_API_ERROR_INVALID_ARGUMENT;
 748         }
 749       if (gid_address_nsh_spi (&a->eid) > MAX_VALUE_U24)
 750         {
 751           clib_warning ("SPI is greater than 24bit!");
 752           return VNET_API_ERROR_INVALID_ARGUMENT;
 753         }
 754     }
 755
 756   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &a->eid);
 757   old_map = mi != ~0 ? pool_elt_at_index (lcm->mapping_pool, mi) : 0;
 758   if (a->is_add)
 759     {
 760       /* TODO check if overwriting and take appropriate actions */
 761       if (mi != GID_LOOKUP_MISS && !gid_address_cmp (&old_map->eid, &a->eid))
 762         {
 763           clib_warning ("eid %U found in the eid-table", format_gid_address,
 764                         &a->eid);
 765           return VNET_API_ERROR_VALUE_EXIST;
 766         }
 767
 768       pool_get (lcm->mapping_pool, m);
 769       gid_address_copy (&m->eid, &a->eid);
 770       m->locator_set_index = a->locator_set_index;
 771       m->ttl = a->ttl;
 772       m->action = a->action;
 773       m->local = a->local;
 774       m->is_static = a->is_static;
 775       m->key = vec_dup (a->key);
 776       m->key_id = a->key_id;
 777       m->authoritative = a->authoritative;
 778
 779       map_index = m - lcm->mapping_pool;
 780       gid_dictionary_add_del (&lcm->mapping_index_by_gid, &a->eid, map_index,
 781                               1);
 782
 783       if (pool_is_free_index (lcm->locator_set_pool, a->locator_set_index))
 784         {
 785           clib_warning ("Locator set with index %d doesn't exist",
 786                         a->locator_set_index);
 787           return VNET_API_ERROR_INVALID_VALUE;
 788         }
 789
 790       /* add eid to list of eids supported by locator-set */
 791       vec_validate (lcm->locator_set_to_eids, a->locator_set_index);
 792       eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids,
 793                                       a->locator_set_index);
 794       vec_add1 (eid_indexes[0], map_index);
 795
 796       if (a->local)
 797         {
 798           /* mark as local */
 799           vec_add1 (lcm->local_mappings_indexes, map_index);
 800         }
 801       map_index_result[0] = map_index;
 802     }
 803   else
 804     {
 805       if (mi == GID_LOOKUP_MISS)
 806         {
 807           clib_warning ("eid %U not found in the eid-table",
 808                         format_gid_address, &a->eid);
 809           return VNET_API_ERROR_INVALID_VALUE;
 810         }
 811
 812       /* clear locator-set to eids binding */
 813       eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids,
 814                                       a->locator_set_index);
 815       for (i = 0; i < vec_len (eid_indexes[0]); i++)
 816         {
 817           map_indexp = vec_elt_at_index (eid_indexes[0], i);
 818           if (map_indexp[0] == mi)
 819             break;
 820         }
 821       vec_del1 (eid_indexes[0], i);
 822
 823       /* remove local mark if needed */
 824       m = pool_elt_at_index (lcm->mapping_pool, mi);
 825       if (m->local)
 826         {
 827           /* Remove adjacencies associated with the local mapping */
 828           rmts_idxp = hash_get (lcm->lcl_to_rmt_adjs_by_lcl_idx, mi);
 829           if (rmts_idxp)
 830             {
 831               rmts =
 832                 pool_elt_at_index (lcm->lcl_to_rmt_adjacencies, rmts_idxp[0]);
 833               vec_foreach (remote_idxp, rmts[0])
 834               {
 835                 dp_del_fwd_entry (lcm, remote_idxp[0]);
 836               }
 837               vec_free (rmts[0]);
 838               pool_put (lcm->lcl_to_rmt_adjacencies, rmts);
 839               hash_unset (lcm->lcl_to_rmt_adjs_by_lcl_idx, mi);
 840             }
 841
 842           u32 k, *lm_indexp;
 843           for (k = 0; k < vec_len (lcm->local_mappings_indexes); k++)
 844             {
 845               lm_indexp = vec_elt_at_index (lcm->local_mappings_indexes, k);
 846               if (lm_indexp[0] == mi)
 847                 break;
 848             }
 849           vec_del1 (lcm->local_mappings_indexes, k);
 850         }
 851       else
 852         {
 853           /* Remove remote (if present) from the vectors of lcl-to-rmts
 854            * TODO: Address this in a more efficient way.
 855            */
 856           /* *INDENT-OFF* */
 857           pool_foreach (rmts, lcm->lcl_to_rmt_adjacencies,
 858           ({
 859             vec_foreach_index (rmts_itr, rmts[0])
 860             {
 861               remote_idx = vec_elt (rmts[0], rmts_itr);
 862               if (mi == remote_idx)
 863                 {
 864                   vec_del1 (rmts[0], rmts_itr);
 865                   break;
 866                 }
 867             }
 868           }));
 869           /* *INDENT-ON* */
 870         }
 871
 872       /* remove mapping from dictionary */
 873       gid_dictionary_add_del (&lcm->mapping_index_by_gid, &a->eid, 0, 0);
 874       gid_address_free (&m->eid);
 875       pool_put_index (lcm->mapping_pool, mi);
 876     }
 877
 878   return 0;
 879 }
 880
 881 /**
 882  *  Add/update/delete mapping to/in/from map-cache.
 883  */
 884 int
 885 vnet_lisp_add_del_local_mapping (vnet_lisp_add_del_mapping_args_t * a,
 886                                  u32 * map_index_result)
 887 {
 888   uword *dp_table = 0;
 889   u32 vni;
 890   u8 type;
 891
 892   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 893
 894   if (vnet_lisp_enable_disable_status () == 0)
 895     {
 896       clib_warning ("LISP is disabled!");
 897       return VNET_API_ERROR_LISP_DISABLED;
 898     }
 899
 900   vni = gid_address_vni (&a->eid);
 901   type = gid_address_type (&a->eid);
 902   if (GID_ADDR_IP_PREFIX == type)
 903     dp_table = hash_get (lcm->table_id_by_vni, vni);
 904   else if (GID_ADDR_MAC == type)
 905     dp_table = hash_get (lcm->bd_id_by_vni, vni);
 906
 907   if (!dp_table && GID_ADDR_NSH != type)
 908     {
 909       clib_warning ("vni %d not associated to a %s!", vni,
 910                     GID_ADDR_IP_PREFIX == type ? "vrf" : "bd");
 911       return VNET_API_ERROR_INVALID_VALUE;
 912     }
 913
 914   /* store/remove mapping from map-cache */
 915   return vnet_lisp_map_cache_add_del (a, map_index_result);
 916 }
 917
 918 static void
 919 add_l2_arp_bd (BVT (clib_bihash_kv) * kvp, void *arg)
 920 {
 921   u32 **ht = arg;
 922   u32 version = (u32) kvp->key[0];
 923   if (IP6 == version)
 924     return;
 925
 926   u32 bd = (u32) (kvp->key[0] >> 32);
 927   hash_set (ht[0], bd, 0);
 928 }
 929
 930 u32 *
 931 vnet_lisp_l2_arp_bds_get (void)
 932 {
 933   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 934   u32 *bds = 0;
 935
 936   gid_dict_foreach_l2_arp_ndp_entry (&lcm->mapping_index_by_gid,
 937                                      add_l2_arp_bd, &bds);
 938   return bds;
 939 }
 940
 941 static void
 942 add_ndp_bd (BVT (clib_bihash_kv) * kvp, void *arg)
 943 {
 944   u32 **ht = arg;
 945   u32 version = (u32) kvp->key[0];
 946   if (IP4 == version)
 947     return;
 948
 949   u32 bd = (u32) (kvp->key[0] >> 32);
 950   hash_set (ht[0], bd, 0);
 951 }
 952
 953 u32 *
 954 vnet_lisp_ndp_bds_get (void)
 955 {
 956   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 957   u32 *bds = 0;
 958
 959   gid_dict_foreach_l2_arp_ndp_entry (&lcm->mapping_index_by_gid,
 960                                      add_ndp_bd, &bds);
 961   return bds;
 962 }
 963
 964 typedef struct
 965 {
 966   void *vector;
 967   u32 bd;
 968 } lisp_add_l2_arp_ndp_args_t;
 969
 970 static void
 971 add_l2_arp_entry (BVT (clib_bihash_kv) * kvp, void *arg)
 972 {
 973   lisp_add_l2_arp_ndp_args_t *a = arg;
 974   lisp_api_l2_arp_entry_t **vector = a->vector, e;
 975
 976   u32 version = (u32) kvp->key[0];
 977   if (IP6 == version)
 978     return;
 979
 980   u32 bd = (u32) (kvp->key[0] >> 32);
 981
 982   if (bd == a->bd)
 983     {
 984       mac_copy (e.mac, (void *) &kvp->value);
 985       e.ip4 = (u32) kvp->key[1];
 986       vec_add1 (vector[0], e);
 987     }
 988 }
 989
 990 lisp_api_l2_arp_entry_t *
 991 vnet_lisp_l2_arp_entries_get_by_bd (u32 bd)
 992 {
 993   lisp_api_l2_arp_entry_t *entries = 0;
 994   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
 995   lisp_add_l2_arp_ndp_args_t a;
 996
 997   a.vector = &entries;
 998   a.bd = bd;
 999
1000   gid_dict_foreach_l2_arp_ndp_entry (&lcm->mapping_index_by_gid,
1001                                      add_l2_arp_entry, &a);
1002   return entries;
1003 }
1004
1005 static void
1006 add_ndp_entry (BVT (clib_bihash_kv) * kvp, void *arg)
1007 {
1008   lisp_add_l2_arp_ndp_args_t *a = arg;
1009   lisp_api_ndp_entry_t **vector = a->vector, e;
1010
1011   u32 version = (u32) kvp->key[0];
1012   if (IP4 == version)
1013     return;
1014
1015   u32 bd = (u32) (kvp->key[0] >> 32);
1016
1017   if (bd == a->bd)
1018     {
1019       mac_copy (e.mac, (void *) &kvp->value);
1020       clib_memcpy (e.ip6, &kvp->key[1], 16);
1021       vec_add1 (vector[0], e);
1022     }
1023 }
1024
1025 lisp_api_ndp_entry_t *
1026 vnet_lisp_ndp_entries_get_by_bd (u32 bd)
1027 {
1028   lisp_api_ndp_entry_t *entries = 0;
1029   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1030   lisp_add_l2_arp_ndp_args_t a;
1031
1032   a.vector = &entries;
1033   a.bd = bd;
1034
1035   gid_dict_foreach_l2_arp_ndp_entry (&lcm->mapping_index_by_gid,
1036                                      add_ndp_entry, &a);
1037   return entries;
1038 }
1039
1040 int
1041 vnet_lisp_add_del_l2_arp_ndp_entry (gid_address_t * key, u8 * mac, u8 is_add)
1042 {
1043   if (vnet_lisp_enable_disable_status () == 0)
1044     {
1045       clib_warning ("LISP is disabled!");
1046       return VNET_API_ERROR_LISP_DISABLED;
1047     }
1048
1049   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1050   int rc = 0;
1051
1052   u64 res = gid_dictionary_lookup (&lcm->mapping_index_by_gid, key);
1053   if (is_add)
1054     {
1055       if (res != GID_LOOKUP_MISS_L2)
1056         {
1057           clib_warning ("Entry %U exists in DB!", format_gid_address, key);
1058           return VNET_API_ERROR_ENTRY_ALREADY_EXISTS;
1059         }
1060       u64 val = mac_to_u64 (mac);
1061       gid_dictionary_add_del (&lcm->mapping_index_by_gid, key, val,
1062                               1 /* is_add */ );
1063     }
1064   else
1065     {
1066       if (res == GID_LOOKUP_MISS_L2)
1067         {
1068           clib_warning ("ONE entry %U not found - cannot delete!",
1069                         format_gid_address, key);
1070           return -1;
1071         }
1072       gid_dictionary_add_del (&lcm->mapping_index_by_gid, key, 0,
1073                               0 /* is_add */ );
1074     }
1075
1076   return rc;
1077 }
1078
1079 int
1080 vnet_lisp_eid_table_map (u32 vni, u32 dp_id, u8 is_l2, u8 is_add)
1081 {
1082   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1083   uword *dp_idp, *vnip, **dp_table_by_vni, **vni_by_dp_table;
1084
1085   if (vnet_lisp_enable_disable_status () == 0)
1086     {
1087       clib_warning ("LISP is disabled!");
1088       return VNET_API_ERROR_LISP_DISABLED;
1089     }
1090
1091   dp_table_by_vni = is_l2 ? &lcm->bd_id_by_vni : &lcm->table_id_by_vni;
1092   vni_by_dp_table = is_l2 ? &lcm->vni_by_bd_id : &lcm->vni_by_table_id;
1093
1094   if (!is_l2 && (vni == 0 || dp_id == 0))
1095     {
1096       clib_warning ("can't add/del default vni-vrf mapping!");
1097       return -1;
1098     }
1099
1100   dp_idp = hash_get (dp_table_by_vni[0], vni);
1101   vnip = hash_get (vni_by_dp_table[0], dp_id);
1102
1103   if (is_add)
1104     {
1105       if (dp_idp || vnip)
1106         {
1107           clib_warning ("vni %d or vrf %d already used in vrf/vni "
1108                         "mapping!", vni, dp_id);
1109           return -1;
1110         }
1111       hash_set (dp_table_by_vni[0], vni, dp_id);
1112       hash_set (vni_by_dp_table[0], dp_id, vni);
1113
1114       /* create dp iface */
1115       dp_add_del_iface (lcm, vni, is_l2, 1 /* is_add */ ,
1116                         1 /* with_default_route */ );
1117     }
1118   else
1119     {
1120       if (!dp_idp || !vnip)
1121         {
1122           clib_warning ("vni %d or vrf %d not used in any vrf/vni! "
1123                         "mapping!", vni, dp_id);
1124           return -1;
1125         }
1126       /* remove dp iface */
1127       dp_add_del_iface (lcm, vni, is_l2, 0 /* is_add */ , 0 /* unused */ );
1128
1129       hash_unset (dp_table_by_vni[0], vni);
1130       hash_unset (vni_by_dp_table[0], dp_id);
1131     }
1132   return 0;
1133
1134 }
1135
1136 /* return 0 if the two locator sets are identical 1 otherwise */
1137 static u8
1138 compare_locators (lisp_cp_main_t * lcm, u32 * old_ls_indexes,
1139                   locator_t * new_locators)
1140 {
1141   u32 i, old_li;
1142   locator_t *old_loc, *new_loc;
1143
1144   if (vec_len (old_ls_indexes) != vec_len (new_locators))
1145     return 1;
1146
1147   for (i = 0; i < vec_len (new_locators); i++)
1148     {
1149       old_li = vec_elt (old_ls_indexes, i);
1150       old_loc = pool_elt_at_index (lcm->locator_pool, old_li);
1151
1152       new_loc = vec_elt_at_index (new_locators, i);
1153
1154       if (locator_cmp (old_loc, new_loc))
1155         return 1;
1156     }
1157   return 0;
1158 }
1159
1160 typedef struct
1161 {
1162   u8 is_negative;
1163   void *lcm;
1164   gid_address_t *eids_to_be_deleted;
1165 } remove_mapping_args_t;
1166
1167 /**
1168  * Callback invoked when a sub-prefix is found
1169  */
1170 static void
1171 remove_mapping_if_needed (u32 mi, void *arg)
1172 {
1173   u8 delete = 0;
1174   remove_mapping_args_t *a = arg;
1175   lisp_cp_main_t *lcm = a->lcm;
1176   mapping_t *m;
1177   locator_set_t *ls;
1178
1179   m = pool_elt_at_index (lcm->mapping_pool, mi);
1180   if (!m)
1181     return;
1182
1183   ls = pool_elt_at_index (lcm->locator_set_pool, m->locator_set_index);
1184
1185   if (a->is_negative)
1186     {
1187       if (0 != vec_len (ls->locator_indices))
1188         delete = 1;
1189     }
1190   else
1191     {
1192       if (0 == vec_len (ls->locator_indices))
1193         delete = 1;
1194     }
1195
1196   if (delete)
1197     vec_add1 (a->eids_to_be_deleted, m->eid);
1198 }
1199
1200 /**
1201  * This function searches map cache and looks for IP prefixes that are subset
1202  * of the provided one. If such prefix is found depending on 'is_negative'
1203  * it does follows:
1204  *
1205  * 1) if is_negative is true and found prefix points to positive mapping,
1206  *    then the mapping is removed
1207  * 2) if is_negative is false and found prefix points to negative mapping,
1208  *    then the mapping is removed
1209  */
1210 static void
1211 remove_overlapping_sub_prefixes (lisp_cp_main_t * lcm, gid_address_t * eid,
1212                                  u8 is_negative)
1213 {
1214   gid_address_t *e;
1215   remove_mapping_args_t a;
1216
1217   clib_memset (&a, 0, sizeof (a));
1218
1219   /* do this only in src/dst mode ... */
1220   if (MR_MODE_SRC_DST != lcm->map_request_mode)
1221     return;
1222
1223   /* ... and  only for IP prefix */
1224   if (GID_ADDR_SRC_DST != gid_address_type (eid)
1225       || (FID_ADDR_IP_PREF != gid_address_sd_dst_type (eid)))
1226     return;
1227
1228   a.is_negative = is_negative;
1229   a.lcm = lcm;
1230
1231   gid_dict_foreach_subprefix (&lcm->mapping_index_by_gid, eid,
1232                               remove_mapping_if_needed, &a);
1233
1234   vec_foreach (e, a.eids_to_be_deleted)
1235   {
1236     vnet_lisp_add_del_adjacency_args_t _adj_args, *adj_args = &_adj_args;
1237
1238     clib_memset (adj_args, 0, sizeof (adj_args[0]));
1239     gid_address_copy (&adj_args->reid, e);
1240     adj_args->is_add = 0;
1241     if (vnet_lisp_add_del_adjacency (adj_args))
1242       clib_warning ("failed to del adjacency!");
1243
1244     vnet_lisp_del_mapping (e, NULL);
1245   }
1246
1247   vec_free (a.eids_to_be_deleted);
1248 }
1249
1250 static void
1251 mapping_delete_timer (lisp_cp_main_t * lcm, u32 mi)
1252 {
1253   timing_wheel_delete (&lcm->wheel, mi);
1254 }
1255
1256 static int
1257 is_local_ip (lisp_cp_main_t * lcm, ip_address_t * addr)
1258 {
1259   fib_node_index_t fei;
1260   fib_prefix_t prefix;
1261   fib_entry_flag_t flags;
1262
1263   ip_address_to_fib_prefix (addr, &prefix);
1264
1265   fei = fib_table_lookup (0, &prefix);
1266   flags = fib_entry_get_flags (fei);
1267   return (FIB_ENTRY_FLAG_LOCAL & flags);
1268 }
1269
1270 /**
1271  * Adds/updates mapping. Does not program forwarding.
1272  *
1273  * @param a parameters of the new mapping
1274  * @param rlocs vector of remote locators
1275  * @param res_map_index index of the newly created mapping
1276  * @param locators_changed indicator if locators were updated in the mapping
1277  * @return return code
1278  */
1279 int
1280 vnet_lisp_add_mapping (vnet_lisp_add_del_mapping_args_t * a,
1281                        locator_t * rlocs,
1282                        u32 * res_map_index, u8 * is_updated)
1283 {
1284   vnet_lisp_add_del_locator_set_args_t _ls_args, *ls_args = &_ls_args;
1285   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1286   u32 mi, ls_index = 0, dst_map_index;
1287   mapping_t *old_map;
1288   locator_t *loc;
1289
1290   if (vnet_lisp_enable_disable_status () == 0)
1291     {
1292       clib_warning ("LISP is disabled!");
1293       return VNET_API_ERROR_LISP_DISABLED;
1294     }
1295
1296   if (res_map_index)
1297     res_map_index[0] = ~0;
1298   if (is_updated)
1299     is_updated[0] = 0;
1300
1301   clib_memset (ls_args, 0, sizeof (ls_args[0]));
1302
1303   ls_args->locators = rlocs;
1304   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &a->eid);
1305   old_map = ((u32) ~ 0 != mi) ? pool_elt_at_index (lcm->mapping_pool, mi) : 0;
1306
1307   /* check if none of the locators match locally configured address */
1308   vec_foreach (loc, rlocs)
1309   {
1310     ip_prefix_t *p = &gid_address_ippref (&loc->address);
1311     if (is_local_ip (lcm, &ip_prefix_addr (p)))
1312       {
1313         clib_warning ("RLOC %U matches a local address!",
1314                       format_gid_address, &loc->address);
1315         return VNET_API_ERROR_LISP_RLOC_LOCAL;
1316       }
1317   }
1318
1319   /* overwrite: if mapping already exists, decide if locators should be
1320    * updated and be done */
1321   if (old_map && gid_address_cmp (&old_map->eid, &a->eid) == 0)
1322     {
1323       if (!a->is_static && (old_map->is_static || old_map->local))
1324         {
1325           /* do not overwrite local or static remote mappings */
1326           clib_warning ("mapping %U rejected due to collision with local "
1327                         "or static remote mapping!", format_gid_address,
1328                         &a->eid);
1329           return 0;
1330         }
1331
1332       locator_set_t *old_ls;
1333
1334       /* update mapping attributes */
1335       old_map->action = a->action;
1336       if (old_map->action != a->action && NULL != is_updated)
1337         is_updated[0] = 1;
1338
1339       old_map->authoritative = a->authoritative;
1340       old_map->ttl = a->ttl;
1341
1342       old_ls = pool_elt_at_index (lcm->locator_set_pool,
1343                                   old_map->locator_set_index);
1344       if (compare_locators (lcm, old_ls->locator_indices, ls_args->locators))
1345         {
1346           /* set locator-set index to overwrite */
1347           ls_args->is_add = 1;
1348           ls_args->index = old_map->locator_set_index;
1349           vnet_lisp_add_del_locator_set (ls_args, 0);
1350           if (is_updated)
1351             is_updated[0] = 1;
1352         }
1353       if (res_map_index)
1354         res_map_index[0] = mi;
1355     }
1356   /* new mapping */
1357   else
1358     {
1359       if (is_updated)
1360         is_updated[0] = 1;
1361       remove_overlapping_sub_prefixes (lcm, &a->eid, 0 == ls_args->locators);
1362
1363       ls_args->is_add = 1;
1364       ls_args->index = ~0;
1365
1366       vnet_lisp_add_del_locator_set (ls_args, &ls_index);
1367
1368       /* add mapping */
1369       a->is_add = 1;
1370       a->locator_set_index = ls_index;
1371       vnet_lisp_map_cache_add_del (a, &dst_map_index);
1372
1373       if (res_map_index)
1374         res_map_index[0] = dst_map_index;
1375     }
1376
1377   /* success */
1378   return 0;
1379 }
1380
1381 /**
1382  * Removes a mapping. Does not program forwarding.
1383  *
1384  * @param eid end-host identifier
1385  * @param res_map_index index of the removed mapping
1386  * @return return code
1387  */
1388 int
1389 vnet_lisp_del_mapping (gid_address_t * eid, u32 * res_map_index)
1390 {
1391   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1392   vnet_lisp_add_del_mapping_args_t _m_args, *m_args = &_m_args;
1393   vnet_lisp_add_del_locator_set_args_t _ls_args, *ls_args = &_ls_args;
1394   mapping_t *old_map;
1395   u32 mi;
1396
1397   clib_memset (ls_args, 0, sizeof (ls_args[0]));
1398   clib_memset (m_args, 0, sizeof (m_args[0]));
1399   if (res_map_index)
1400     res_map_index[0] = ~0;
1401
1402   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, eid);
1403   old_map = ((u32) ~ 0 != mi) ? pool_elt_at_index (lcm->mapping_pool, mi) : 0;
1404
1405   if (old_map == 0 || gid_address_cmp (&old_map->eid, eid) != 0)
1406     {
1407       clib_warning ("cannot delete mapping for eid %U",
1408                     format_gid_address, eid);
1409       return -1;
1410     }
1411
1412   m_args->is_add = 0;
1413   gid_address_copy (&m_args->eid, eid);
1414   m_args->locator_set_index = old_map->locator_set_index;
1415
1416   /* delete mapping associated from map-cache */
1417   vnet_lisp_map_cache_add_del (m_args, 0);
1418
1419   ls_args->is_add = 0;
1420   ls_args->index = old_map->locator_set_index;
1421
1422   /* delete locator set */
1423   vnet_lisp_add_del_locator_set (ls_args, 0);
1424
1425   /* delete timer associated to the mapping if any */
1426   if (old_map->timer_set)
1427     mapping_delete_timer (lcm, mi);
1428
1429   /* return old mapping index */
1430   if (res_map_index)
1431     res_map_index[0] = mi;
1432
1433   /* success */
1434   return 0;
1435 }
1436
1437 int
1438 vnet_lisp_clear_all_remote_adjacencies (void)
1439 {
1440   int rv = 0;
1441   u32 mi, *map_indices = 0, *map_indexp;
1442   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1443   vnet_lisp_add_del_mapping_args_t _dm_args, *dm_args = &_dm_args;
1444   vnet_lisp_add_del_locator_set_args_t _ls, *ls = &_ls;
1445
1446   /* *INDENT-OFF* */
1447   pool_foreach_index (mi, lcm->mapping_pool,
1448   ({
1449     vec_add1 (map_indices, mi);
1450   }));
1451   /* *INDENT-ON* */
1452
1453   vec_foreach (map_indexp, map_indices)
1454   {
1455     mapping_t *map = pool_elt_at_index (lcm->mapping_pool, map_indexp[0]);
1456     if (!map->local)
1457       {
1458         dp_del_fwd_entry (lcm, map_indexp[0]);
1459
1460         dm_args->is_add = 0;
1461         gid_address_copy (&dm_args->eid, &map->eid);
1462         dm_args->locator_set_index = map->locator_set_index;
1463
1464         /* delete mapping associated to fwd entry */
1465         vnet_lisp_map_cache_add_del (dm_args, 0);
1466
1467         ls->is_add = 0;
1468         ls->local = 0;
1469         ls->index = map->locator_set_index;
1470         /* delete locator set */
1471         rv = vnet_lisp_add_del_locator_set (ls, 0);
1472         if (rv != 0)
1473           goto cleanup;
1474       }
1475   }
1476
1477 cleanup:
1478   if (map_indices)
1479     vec_free (map_indices);
1480   return rv;
1481 }
1482
1483 /**
1484  * Adds adjacency or removes forwarding entry associated to remote mapping.
1485  * Note that adjacencies are not stored, they only result in forwarding entries
1486  * being created.
1487  */
1488 int
1489 vnet_lisp_add_del_adjacency (vnet_lisp_add_del_adjacency_args_t * a)
1490 {
1491   lisp_cp_main_t *lcm = &lisp_control_main;
1492   u32 local_mi, remote_mi = ~0;
1493
1494   if (vnet_lisp_enable_disable_status () == 0)
1495     {
1496       clib_warning ("LISP is disabled!");
1497       return VNET_API_ERROR_LISP_DISABLED;
1498     }
1499
1500   remote_mi = gid_dictionary_sd_lookup (&lcm->mapping_index_by_gid,
1501                                         &a->reid, &a->leid);
1502   if (GID_LOOKUP_MISS == remote_mi)
1503     {
1504       clib_warning ("Remote eid %U not found. Cannot add adjacency!",
1505                     format_gid_address, &a->reid);
1506
1507       return -1;
1508     }
1509
1510   if (a->is_add)
1511     {
1512       /* check if source eid has an associated mapping. If pitr mode is on,
1513        * just use the pitr's mapping */
1514       if (lcm->flags & LISP_FLAG_PITR_MODE)
1515         {
1516           if (lcm->pitr_map_index != ~0)
1517             {
1518               local_mi = lcm->pitr_map_index;
1519             }
1520           else
1521             {
1522               /* PITR mode is on, but no mapping is configured */
1523               return -1;
1524             }
1525         }
1526       else
1527         {
1528           if (gid_address_type (&a->reid) == GID_ADDR_NSH)
1529             {
1530               if (lcm->nsh_map_index == ~0)
1531                 local_mi = GID_LOOKUP_MISS;
1532               else
1533                 local_mi = lcm->nsh_map_index;
1534             }
1535           else
1536             {
1537               local_mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid,
1538                                                 &a->leid);
1539             }
1540         }
1541
1542       if (GID_LOOKUP_MISS == local_mi)
1543         {
1544           clib_warning ("Local eid %U not found. Cannot add adjacency!",
1545                         format_gid_address, &a->leid);
1546
1547           return -1;
1548         }
1549
1550       /* update forwarding */
1551       dp_add_fwd_entry (lcm, local_mi, remote_mi);
1552     }
1553   else
1554     dp_del_fwd_entry (lcm, remote_mi);
1555
1556   return 0;
1557 }
1558
1559 int
1560 vnet_lisp_set_map_request_mode (u8 mode)
1561 {
1562   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1563
1564   if (vnet_lisp_enable_disable_status () == 0)
1565     {
1566       clib_warning ("LISP is disabled!");
1567       return VNET_API_ERROR_LISP_DISABLED;
1568     }
1569
1570   if (mode >= _MR_MODE_MAX)
1571     {
1572       clib_warning ("Invalid LISP map request mode %d!", mode);
1573       return VNET_API_ERROR_INVALID_ARGUMENT;
1574     }
1575
1576   lcm->map_request_mode = mode;
1577   return 0;
1578 }
1579
1580 int
1581 vnet_lisp_nsh_set_locator_set (u8 * locator_set_name, u8 is_add)
1582 {
1583   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1584   lisp_gpe_main_t *lgm = vnet_lisp_gpe_get_main ();
1585   u32 locator_set_index = ~0;
1586   mapping_t *m;
1587   uword *p;
1588
1589   if (vnet_lisp_enable_disable_status () == 0)
1590     {
1591       clib_warning ("LISP is disabled!");
1592       return VNET_API_ERROR_LISP_DISABLED;
1593     }
1594
1595   if (is_add)
1596     {
1597       if (lcm->nsh_map_index == (u32) ~ 0)
1598         {
1599           p = hash_get_mem (lcm->locator_set_index_by_name, locator_set_name);
1600           if (!p)
1601             {
1602               clib_warning ("locator-set %v doesn't exist", locator_set_name);
1603               return -1;
1604             }
1605           locator_set_index = p[0];
1606
1607           pool_get (lcm->mapping_pool, m);
1608           clib_memset (m, 0, sizeof *m);
1609           m->locator_set_index = locator_set_index;
1610           m->local = 1;
1611           m->nsh_set = 1;
1612           lcm->nsh_map_index = m - lcm->mapping_pool;
1613
1614           if (~0 == vnet_lisp_gpe_add_nsh_iface (lgm))
1615             return -1;
1616         }
1617     }
1618   else
1619     {
1620       if (lcm->nsh_map_index != (u32) ~ 0)
1621         {
1622           /* remove NSH mapping */
1623           pool_put_index (lcm->mapping_pool, lcm->nsh_map_index);
1624           lcm->nsh_map_index = ~0;
1625           vnet_lisp_gpe_del_nsh_iface (lgm);
1626         }
1627     }
1628   return 0;
1629 }
1630
1631 int
1632 vnet_lisp_pitr_set_locator_set (u8 * locator_set_name, u8 is_add)
1633 {
1634   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1635   u32 locator_set_index = ~0;
1636   mapping_t *m;
1637   uword *p;
1638
1639   if (vnet_lisp_enable_disable_status () == 0)
1640     {
1641       clib_warning ("LISP is disabled!");
1642       return VNET_API_ERROR_LISP_DISABLED;
1643     }
1644
1645   p = hash_get_mem (lcm->locator_set_index_by_name, locator_set_name);
1646   if (!p)
1647     {
1648       clib_warning ("locator-set %v doesn't exist", locator_set_name);
1649       return -1;
1650     }
1651   locator_set_index = p[0];
1652
1653   if (is_add)
1654     {
1655       pool_get (lcm->mapping_pool, m);
1656       m->locator_set_index = locator_set_index;
1657       m->local = 1;
1658       m->pitr_set = 1;
1659       lcm->pitr_map_index = m - lcm->mapping_pool;
1660     }
1661   else
1662     {
1663       /* remove pitr mapping */
1664       pool_put_index (lcm->mapping_pool, lcm->pitr_map_index);
1665       lcm->pitr_map_index = ~0;
1666     }
1667   return 0;
1668 }
1669
1670 int
1671 vnet_lisp_map_register_fallback_threshold_set (u32 value)
1672 {
1673   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1674   if (0 == value)
1675     {
1676       return VNET_API_ERROR_INVALID_ARGUMENT;
1677     }
1678
1679   lcm->max_expired_map_registers = value;
1680   return 0;
1681 }
1682
1683 u32
1684 vnet_lisp_map_register_fallback_threshold_get (void)
1685 {
1686   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1687   return lcm->max_expired_map_registers;
1688 }
1689
1690 /**
1691  * Configure Proxy-ETR
1692  *
1693  * @param ip PETR's IP address
1694  * @param is_add Flag that indicates if this is an addition or removal
1695  *
1696  * return 0 on success
1697  */
1698 int
1699 vnet_lisp_use_petr (ip_address_t * ip, u8 is_add)
1700 {
1701   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1702   u32 ls_index = ~0;
1703   mapping_t *m;
1704   vnet_lisp_add_del_locator_set_args_t _ls_args, *ls_args = &_ls_args;
1705   locator_t loc;
1706
1707   if (vnet_lisp_enable_disable_status () == 0)
1708     {
1709       clib_warning ("LISP is disabled!");
1710       return VNET_API_ERROR_LISP_DISABLED;
1711     }
1712
1713   clib_memset (ls_args, 0, sizeof (*ls_args));
1714
1715   if (is_add)
1716     {
1717       /* Create dummy petr locator-set */
1718       clib_memset (&loc, 0, sizeof (loc));
1719       gid_address_from_ip (&loc.address, ip);
1720       loc.priority = 1;
1721       loc.state = loc.weight = 1;
1722       loc.local = 0;
1723
1724       ls_args->is_add = 1;
1725       ls_args->index = ~0;
1726       vec_add1 (ls_args->locators, loc);
1727       vnet_lisp_add_del_locator_set (ls_args, &ls_index);
1728
1729       /* Add petr mapping */
1730       pool_get (lcm->mapping_pool, m);
1731       m->locator_set_index = ls_index;
1732       lcm->petr_map_index = m - lcm->mapping_pool;
1733
1734       /* Enable use-petr */
1735       lcm->flags |= LISP_FLAG_USE_PETR;
1736     }
1737   else
1738     {
1739       m = pool_elt_at_index (lcm->mapping_pool, lcm->petr_map_index);
1740
1741       /* Remove petr locator */
1742       ls_args->is_add = 0;
1743       ls_args->index = m->locator_set_index;
1744       vnet_lisp_add_del_locator_set (ls_args, 0);
1745
1746       /* Remove petr mapping */
1747       pool_put_index (lcm->mapping_pool, lcm->petr_map_index);
1748
1749       /* Disable use-petr */
1750       lcm->flags &= ~LISP_FLAG_USE_PETR;
1751       lcm->petr_map_index = ~0;
1752     }
1753   return 0;
1754 }
1755
1756 /* cleans locator to locator-set data and removes locators not part of
1757  * any locator-set */
1758 static void
1759 clean_locator_to_locator_set (lisp_cp_main_t * lcm, u32 lsi)
1760 {
1761   u32 i, j, *loc_indexp, *ls_indexp, **ls_indexes, *to_be_deleted = 0;
1762   locator_set_t *ls = pool_elt_at_index (lcm->locator_set_pool, lsi);
1763   for (i = 0; i < vec_len (ls->locator_indices); i++)
1764     {
1765       loc_indexp = vec_elt_at_index (ls->locator_indices, i);
1766       ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets,
1767                                      loc_indexp[0]);
1768       for (j = 0; j < vec_len (ls_indexes[0]); j++)
1769         {
1770           ls_indexp = vec_elt_at_index (ls_indexes[0], j);
1771           if (ls_indexp[0] == lsi)
1772             break;
1773         }
1774
1775       /* delete index for removed locator-set */
1776       vec_del1 (ls_indexes[0], j);
1777
1778       /* delete locator if it's part of no locator-set */
1779       if (vec_len (ls_indexes[0]) == 0)
1780         {
1781           pool_put_index (lcm->locator_pool, loc_indexp[0]);
1782           vec_add1 (to_be_deleted, i);
1783         }
1784     }
1785
1786   if (to_be_deleted)
1787     {
1788       for (i = 0; i < vec_len (to_be_deleted); i++)
1789         {
1790           loc_indexp = vec_elt_at_index (to_be_deleted, i);
1791           vec_del1 (ls->locator_indices, loc_indexp[0]);
1792         }
1793       vec_free (to_be_deleted);
1794     }
1795 }
1796
1797 static inline uword *
1798 get_locator_set_index (vnet_lisp_add_del_locator_set_args_t * a, uword * p)
1799 {
1800   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1801
1802   ASSERT (a != NULL);
1803   ASSERT (p != NULL);
1804
1805   /* find locator-set */
1806   if (a->local)
1807     {
1808       ASSERT (a->name);
1809       p = hash_get_mem (lcm->locator_set_index_by_name, a->name);
1810     }
1811   else
1812     {
1813       *p = a->index;
1814     }
1815
1816   return p;
1817 }
1818
1819 static inline int
1820 is_locator_in_locator_set (lisp_cp_main_t * lcm, locator_set_t * ls,
1821                            locator_t * loc)
1822 {
1823   locator_t *itloc;
1824   u32 *locit;
1825
1826   ASSERT (ls != NULL);
1827   ASSERT (loc != NULL);
1828
1829   vec_foreach (locit, ls->locator_indices)
1830   {
1831     itloc = pool_elt_at_index (lcm->locator_pool, locit[0]);
1832     if ((ls->local && itloc->sw_if_index == loc->sw_if_index) ||
1833         (!ls->local && !gid_address_cmp (&itloc->address, &loc->address)))
1834       {
1835         clib_warning ("Duplicate locator");
1836         return VNET_API_ERROR_VALUE_EXIST;
1837       }
1838   }
1839
1840   return 0;
1841 }
1842
1843 static void
1844 update_adjacencies_by_map_index (lisp_cp_main_t * lcm,
1845                                  u32 mapping_index, u8 remove_only)
1846 {
1847   fwd_entry_t *fwd;
1848   mapping_t *map;
1849   uword *fei = 0, *rmts_idxp = 0;
1850   u32 **rmts = 0, *remote_idxp = 0, *rmts_copy = 0;
1851   vnet_lisp_add_del_adjacency_args_t _a, *a = &_a;
1852   clib_memset (a, 0, sizeof (*a));
1853
1854   map = pool_elt_at_index (lcm->mapping_pool, mapping_index);
1855
1856   if (map->local)
1857     {
1858       rmts_idxp = hash_get (lcm->lcl_to_rmt_adjs_by_lcl_idx, mapping_index);
1859       if (rmts_idxp)
1860         {
1861           rmts =
1862             pool_elt_at_index (lcm->lcl_to_rmt_adjacencies, rmts_idxp[0]);
1863           rmts_copy = vec_dup (rmts[0]);
1864
1865           vec_foreach (remote_idxp, rmts_copy)
1866           {
1867             fei = hash_get (lcm->fwd_entry_by_mapping_index, remote_idxp[0]);
1868             if (!fei)
1869               continue;
1870
1871             fwd = pool_elt_at_index (lcm->fwd_entry_pool, fei[0]);
1872             a->is_add = 0;
1873             gid_address_copy (&a->leid, &fwd->leid);
1874             gid_address_copy (&a->reid, &fwd->reid);
1875             vnet_lisp_add_del_adjacency (a);
1876
1877             if (!remove_only)
1878               {
1879                 a->is_add = 1;
1880                 vnet_lisp_add_del_adjacency (a);
1881               }
1882           }
1883           vec_free (rmts_copy);
1884         }
1885     }
1886   else
1887     {
1888       fei = hash_get (lcm->fwd_entry_by_mapping_index, mapping_index);
1889       if (!fei)
1890         return;
1891
1892       fwd = pool_elt_at_index (lcm->fwd_entry_pool, fei[0]);
1893       a->is_add = 0;
1894       gid_address_copy (&a->leid, &fwd->leid);
1895       gid_address_copy (&a->reid, &fwd->reid);
1896       vnet_lisp_add_del_adjacency (a);
1897
1898       if (!remove_only)
1899         {
1900           a->is_add = 1;
1901           vnet_lisp_add_del_adjacency (a);
1902         }
1903     }
1904 }
1905
1906 static void
1907 update_fwd_entries_by_locator_set (lisp_cp_main_t * lcm,
1908                                    u32 ls_index, u8 remove_only)
1909 {
1910   u32 i, *map_indexp;
1911   u32 **eid_indexes;
1912
1913   if (vec_len (lcm->locator_set_to_eids) <= ls_index)
1914     return;
1915
1916   eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids, ls_index);
1917
1918   for (i = 0; i < vec_len (eid_indexes[0]); i++)
1919     {
1920       map_indexp = vec_elt_at_index (eid_indexes[0], i);
1921       update_adjacencies_by_map_index (lcm, map_indexp[0], remove_only);
1922     }
1923 }
1924
1925 static inline void
1926 remove_locator_from_locator_set (locator_set_t * ls, u32 * locit,
1927                                  u32 ls_index, u32 loc_id)
1928 {
1929   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1930   u32 **ls_indexes = NULL;
1931
1932   ASSERT (ls != NULL);
1933   ASSERT (locit != NULL);
1934
1935   ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets, locit[0]);
1936   pool_put_index (lcm->locator_pool, locit[0]);
1937   vec_del1 (ls->locator_indices, loc_id);
1938   vec_del1 (ls_indexes[0], ls_index);
1939 }
1940
1941 int
1942 vnet_lisp_add_del_locator (vnet_lisp_add_del_locator_set_args_t * a,
1943                            locator_set_t * ls, u32 * ls_result)
1944 {
1945   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1946   locator_t *loc = NULL, *itloc = NULL;
1947   uword _p = (u32) ~ 0, *p = &_p;
1948   u32 loc_index = ~0, ls_index = ~0, *locit = NULL, **ls_indexes = NULL;
1949   u32 loc_id = ~0;
1950   int ret = 0;
1951
1952   ASSERT (a != NULL);
1953
1954   if (vnet_lisp_enable_disable_status () == 0)
1955     {
1956       clib_warning ("LISP is disabled!");
1957       return VNET_API_ERROR_LISP_DISABLED;
1958     }
1959
1960   p = get_locator_set_index (a, p);
1961   if (!p)
1962     {
1963       clib_warning ("locator-set %v doesn't exist", a->name);
1964       return VNET_API_ERROR_INVALID_ARGUMENT;
1965     }
1966
1967   if (ls == 0)
1968     {
1969       ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
1970       if (!ls)
1971         {
1972           clib_warning ("locator-set %d to be overwritten doesn't exist!",
1973                         p[0]);
1974           return VNET_API_ERROR_INVALID_ARGUMENT;
1975         }
1976     }
1977
1978   if (a->is_add)
1979     {
1980       if (ls_result)
1981         ls_result[0] = p[0];
1982
1983       /* allocate locators */
1984       vec_foreach (itloc, a->locators)
1985       {
1986         ret = is_locator_in_locator_set (lcm, ls, itloc);
1987         if (0 != ret)
1988           {
1989             return ret;
1990           }
1991
1992         pool_get (lcm->locator_pool, loc);
1993         loc[0] = itloc[0];
1994         loc_index = loc - lcm->locator_pool;
1995
1996         vec_add1 (ls->locator_indices, loc_index);
1997
1998         vec_validate (lcm->locator_to_locator_sets, loc_index);
1999         ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets,
2000                                        loc_index);
2001         vec_add1 (ls_indexes[0], p[0]);
2002       }
2003     }
2004   else
2005     {
2006       ls_index = p[0];
2007       u8 removed;
2008
2009       vec_foreach (itloc, a->locators)
2010       {
2011         removed = 0;
2012         loc_id = 0;
2013         vec_foreach (locit, ls->locator_indices)
2014         {
2015           loc = pool_elt_at_index (lcm->locator_pool, locit[0]);
2016
2017           if (loc->local && loc->sw_if_index == itloc->sw_if_index)
2018             {
2019               removed = 1;
2020               remove_locator_from_locator_set (ls, locit, ls_index, loc_id);
2021             }
2022           if (0 == loc->local &&
2023               !gid_address_cmp (&loc->address, &itloc->address))
2024             {
2025               removed = 1;
2026               remove_locator_from_locator_set (ls, locit, ls_index, loc_id);
2027             }
2028
2029           if (removed)
2030             {
2031               /* update fwd entries using this locator in DP */
2032               update_fwd_entries_by_locator_set (lcm, ls_index,
2033                                                  vec_len (ls->locator_indices)
2034                                                  == 0);
2035             }
2036
2037           loc_id++;
2038         }
2039       }
2040     }
2041
2042   return 0;
2043 }
2044
2045 int
2046 vnet_lisp_add_del_locator_set (vnet_lisp_add_del_locator_set_args_t * a,
2047                                u32 * ls_result)
2048 {
2049   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2050   locator_set_t *ls;
2051   uword _p = (u32) ~ 0, *p = &_p;
2052   u32 ls_index;
2053   u32 **eid_indexes;
2054   int ret = 0;
2055
2056   if (vnet_lisp_enable_disable_status () == 0)
2057     {
2058       clib_warning ("LISP is disabled!");
2059       return VNET_API_ERROR_LISP_DISABLED;
2060     }
2061
2062   if (a->is_add)
2063     {
2064       p = get_locator_set_index (a, p);
2065
2066       /* overwrite */
2067       if (p && p[0] != (u32) ~ 0)
2068         {
2069           ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
2070           if (!ls)
2071             {
2072               clib_warning ("locator-set %d to be overwritten doesn't exist!",
2073                             p[0]);
2074               return -1;
2075             }
2076
2077           /* clean locator to locator-set vectors and remove locators if
2078            * they're not part of another locator-set */
2079           clean_locator_to_locator_set (lcm, p[0]);
2080
2081           /* remove locator indices from locator set */
2082           vec_free (ls->locator_indices);
2083
2084           ls_index = p[0];
2085
2086           if (ls_result)
2087             ls_result[0] = p[0];
2088         }
2089       /* new locator-set */
2090       else
2091         {
2092           pool_get (lcm->locator_set_pool, ls);
2093           clib_memset (ls, 0, sizeof (*ls));
2094           ls_index = ls - lcm->locator_set_pool;
2095
2096           if (a->local)
2097             {
2098               ls->name = vec_dup (a->name);
2099
2100               if (!lcm->locator_set_index_by_name)
2101                 lcm->locator_set_index_by_name = hash_create_vec (
2102                                                                    /* size */
2103                                                                    0,
2104                                                                    sizeof
2105                                                                    (ls->name
2106                                                                     [0]),
2107                                                                    sizeof
2108                                                                    (uword));
2109               hash_set_mem (lcm->locator_set_index_by_name, ls->name,
2110                             ls_index);
2111
2112               /* mark as local locator-set */
2113               vec_add1 (lcm->local_locator_set_indexes, ls_index);
2114             }
2115           ls->local = a->local;
2116           if (ls_result)
2117             ls_result[0] = ls_index;
2118         }
2119
2120       ret = vnet_lisp_add_del_locator (a, ls, NULL);
2121       if (0 != ret)
2122         {
2123           return ret;
2124         }
2125     }
2126   else
2127     {
2128       p = get_locator_set_index (a, p);
2129       if (!p)
2130         {
2131           clib_warning ("locator-set %v doesn't exists", a->name);
2132           return -1;
2133         }
2134
2135       ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
2136       if (!ls)
2137         {
2138           clib_warning ("locator-set with index %d doesn't exists", p[0]);
2139           return -1;
2140         }
2141
2142       if (lcm->mreq_itr_rlocs == p[0])
2143         {
2144           clib_warning ("Can't delete the locator-set used to constrain "
2145                         "the itr-rlocs in map-requests!");
2146           return -1;
2147         }
2148
2149       if (vec_len (lcm->locator_set_to_eids) != 0)
2150         {
2151           eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids, p[0]);
2152           if (vec_len (eid_indexes[0]) != 0)
2153             {
2154               clib_warning
2155                 ("Can't delete a locator that supports a mapping!");
2156               return -1;
2157             }
2158         }
2159
2160       /* clean locator to locator-sets data */
2161       clean_locator_to_locator_set (lcm, p[0]);
2162
2163       if (ls->local)
2164         {
2165           u32 it, lsi;
2166
2167           vec_foreach_index (it, lcm->local_locator_set_indexes)
2168           {
2169             lsi = vec_elt (lcm->local_locator_set_indexes, it);
2170             if (lsi == p[0])
2171               {
2172                 vec_del1 (lcm->local_locator_set_indexes, it);
2173                 break;
2174               }
2175           }
2176           hash_unset_mem (lcm->locator_set_index_by_name, ls->name);
2177         }
2178       vec_free (ls->name);
2179       vec_free (ls->locator_indices);
2180       pool_put (lcm->locator_set_pool, ls);
2181     }
2182   return 0;
2183 }
2184
2185 int
2186 vnet_lisp_rloc_probe_enable_disable (u8 is_enable)
2187 {
2188   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2189
2190   lcm->rloc_probing = is_enable;
2191   return 0;
2192 }
2193
2194 int
2195 vnet_lisp_map_register_enable_disable (u8 is_enable)
2196 {
2197   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2198
2199   lcm->map_registering = is_enable;
2200   return 0;
2201 }
2202
2203 static void
2204 lisp_cp_register_dst_port (vlib_main_t * vm)
2205 {
2206   udp_register_dst_port (vm, UDP_DST_PORT_lisp_cp,
2207                          lisp_cp_input_node.index, 1 /* is_ip4 */ );
2208   udp_register_dst_port (vm, UDP_DST_PORT_lisp_cp6,
2209                          lisp_cp_input_node.index, 0 /* is_ip4 */ );
2210 }
2211
2212 static void
2213 lisp_cp_unregister_dst_port (vlib_main_t * vm)
2214 {
2215   udp_unregister_dst_port (vm, UDP_DST_PORT_lisp_cp, 0 /* is_ip4 */ );
2216   udp_unregister_dst_port (vm, UDP_DST_PORT_lisp_cp6, 1 /* is_ip4 */ );
2217 }
2218
2219 /**
2220  * lisp_cp_enable_l2_l3_ifaces
2221  *
2222  * Enable all l2 and l3 ifaces
2223  */
2224 static void
2225 lisp_cp_enable_l2_l3_ifaces (lisp_cp_main_t * lcm, u8 with_default_route)
2226 {
2227   u32 vni, dp_table;
2228
2229   /* *INDENT-OFF* */
2230   hash_foreach(vni, dp_table, lcm->table_id_by_vni, ({
2231     dp_add_del_iface(lcm, vni, /* is_l2 */ 0, /* is_add */1,
2232                      with_default_route);
2233   }));
2234   hash_foreach(vni, dp_table, lcm->bd_id_by_vni, ({
2235     dp_add_del_iface(lcm, vni, /* is_l2 */ 1, 1,
2236                      with_default_route);
2237   }));
2238   /* *INDENT-ON* */
2239 }
2240
2241 static void
2242 lisp_cp_disable_l2_l3_ifaces (lisp_cp_main_t * lcm)
2243 {
2244   u32 **rmts;
2245
2246   /* clear interface table */
2247   hash_free (lcm->fwd_entry_by_mapping_index);
2248   pool_free (lcm->fwd_entry_pool);
2249   /* Clear state tracking rmt-lcl fwd entries */
2250   /* *INDENT-OFF* */
2251   pool_foreach(rmts, lcm->lcl_to_rmt_adjacencies,
2252   {
2253     vec_free(rmts[0]);
2254   });
2255   /* *INDENT-ON* */
2256   hash_free (lcm->lcl_to_rmt_adjs_by_lcl_idx);
2257   pool_free (lcm->lcl_to_rmt_adjacencies);
2258 }
2259
2260 clib_error_t *
2261 vnet_lisp_enable_disable (u8 is_enable)
2262 {
2263   clib_error_t *error = 0;
2264   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2265   vnet_lisp_gpe_enable_disable_args_t _a, *a = &_a;
2266
2267   a->is_en = is_enable;
2268   error = vnet_lisp_gpe_enable_disable (a);
2269   if (error)
2270     {
2271       return clib_error_return (0, "failed to %s data-plane!",
2272                                 a->is_en ? "enable" : "disable");
2273     }
2274
2275   /* decide what to do based on mode */
2276
2277   if (lcm->flags & LISP_FLAG_XTR_MODE)
2278     {
2279       if (is_enable)
2280         {
2281           lisp_cp_register_dst_port (lcm->vlib_main);
2282           lisp_cp_enable_l2_l3_ifaces (lcm, 1 /* with_default_route */ );
2283         }
2284       else
2285         {
2286           lisp_cp_unregister_dst_port (lcm->vlib_main);
2287           lisp_cp_disable_l2_l3_ifaces (lcm);
2288         }
2289     }
2290
2291   if (lcm->flags & LISP_FLAG_PETR_MODE)
2292     {
2293       /* if in xTR mode, the LISP ports were already (un)registered above */
2294       if (!(lcm->flags & LISP_FLAG_XTR_MODE))
2295         {
2296           if (is_enable)
2297             lisp_cp_register_dst_port (lcm->vlib_main);
2298           else
2299             lisp_cp_unregister_dst_port (lcm->vlib_main);
2300         }
2301     }
2302
2303   if (lcm->flags & LISP_FLAG_PITR_MODE)
2304     {
2305       if (is_enable)
2306         {
2307           /* install interfaces, but no default routes */
2308           lisp_cp_enable_l2_l3_ifaces (lcm, 0 /* with_default_route */ );
2309         }
2310       else
2311         {
2312           lisp_cp_disable_l2_l3_ifaces (lcm);
2313         }
2314     }
2315
2316   /* update global flag */
2317   lcm->is_enabled = is_enable;
2318
2319   return 0;
2320 }
2321
2322 u8
2323 vnet_lisp_enable_disable_status (void)
2324 {
2325   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2326   return lcm->is_enabled;
2327 }
2328
2329 int
2330 vnet_lisp_add_del_map_resolver (vnet_lisp_add_del_map_resolver_args_t * a)
2331 {
2332   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2333   u32 i;
2334   lisp_msmr_t _mr, *mr = &_mr;
2335
2336   if (vnet_lisp_enable_disable_status () == 0)
2337     {
2338       clib_warning ("LISP is disabled!");
2339       return VNET_API_ERROR_LISP_DISABLED;
2340     }
2341
2342   if (a->is_add)
2343     {
2344
2345       if (get_map_resolver (&a->address))
2346         {
2347           clib_warning ("map-resolver %U already exists!", format_ip_address,
2348                         &a->address);
2349           return -1;
2350         }
2351
2352       clib_memset (mr, 0, sizeof (*mr));
2353       ip_address_copy (&mr->address, &a->address);
2354       vec_add1 (lcm->map_resolvers, *mr);
2355
2356       if (vec_len (lcm->map_resolvers) == 1)
2357         lcm->do_map_resolver_election = 1;
2358     }
2359   else
2360     {
2361       for (i = 0; i < vec_len (lcm->map_resolvers); i++)
2362         {
2363           mr = vec_elt_at_index (lcm->map_resolvers, i);
2364           if (!ip_address_cmp (&mr->address, &a->address))
2365             {
2366               if (!ip_address_cmp (&mr->address, &lcm->active_map_resolver))
2367                 lcm->do_map_resolver_election = 1;
2368
2369               vec_del1 (lcm->map_resolvers, i);
2370               break;
2371             }
2372         }
2373     }
2374   return 0;
2375 }
2376
2377 int
2378 vnet_lisp_map_register_set_ttl (u32 ttl)
2379 {
2380   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2381   lcm->map_register_ttl = ttl;
2382   return 0;
2383 }
2384
2385 u32
2386 vnet_lisp_map_register_get_ttl (void)
2387 {
2388   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2389   return lcm->map_register_ttl;
2390 }
2391
2392 int
2393 vnet_lisp_add_del_mreq_itr_rlocs (vnet_lisp_add_del_mreq_itr_rloc_args_t * a)
2394 {
2395   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2396   uword *p = 0;
2397
2398   if (vnet_lisp_enable_disable_status () == 0)
2399     {
2400       clib_warning ("LISP is disabled!");
2401       return VNET_API_ERROR_LISP_DISABLED;
2402     }
2403
2404   if (a->is_add)
2405     {
2406       p = hash_get_mem (lcm->locator_set_index_by_name, a->locator_set_name);
2407       if (!p)
2408         {
2409           clib_warning ("locator-set %v doesn't exist", a->locator_set_name);
2410           return VNET_API_ERROR_INVALID_ARGUMENT;
2411         }
2412
2413       lcm->mreq_itr_rlocs = p[0];
2414     }
2415   else
2416     {
2417       lcm->mreq_itr_rlocs = ~0;
2418     }
2419
2420   return 0;
2421 }
2422
2423 /* Statistics (not really errors) */
2424 #define foreach_lisp_cp_lookup_error           \
2425 _(DROP, "drop")                                \
2426 _(MAP_REQUESTS_SENT, "map-request sent")       \
2427 _(ARP_REPLY_TX, "ARP replies sent")            \
2428 _(NDP_NEIGHBOR_ADVERTISEMENT_TX,               \
2429   "neighbor advertisement sent")
2430
2431 static char *lisp_cp_lookup_error_strings[] = {
2432 #define _(sym,string) string,
2433   foreach_lisp_cp_lookup_error
2434 #undef _
2435 };
2436
2437 typedef enum
2438 {
2439 #define _(sym,str) LISP_CP_LOOKUP_ERROR_##sym,
2440   foreach_lisp_cp_lookup_error
2441 #undef _
2442     LISP_CP_LOOKUP_N_ERROR,
2443 } lisp_cp_lookup_error_t;
2444
2445 typedef enum
2446 {
2447   LISP_CP_LOOKUP_NEXT_DROP,
2448   LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX,
2449   LISP_CP_LOOKUP_N_NEXT,
2450 } lisp_cp_lookup_next_t;
2451
2452 typedef struct
2453 {
2454   gid_address_t dst_eid;
2455   ip_address_t map_resolver_ip;
2456 } lisp_cp_lookup_trace_t;
2457
2458 u8 *
2459 format_lisp_cp_lookup_trace (u8 * s, va_list * args)
2460 {
2461   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
2462   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
2463   lisp_cp_lookup_trace_t *t = va_arg (*args, lisp_cp_lookup_trace_t *);
2464
2465   s = format (s, "LISP-CP-LOOKUP: map-resolver: %U destination eid %U",
2466               format_ip_address, &t->map_resolver_ip, format_gid_address,
2467               &t->dst_eid);
2468   return s;
2469 }
2470
2471 int
2472 get_mr_and_local_iface_ip (lisp_cp_main_t * lcm, ip_address_t * mr_ip,
2473                            ip_address_t * sloc)
2474 {
2475   lisp_msmr_t *mrit;
2476   ip_address_t *a;
2477
2478   if (vec_len (lcm->map_resolvers) == 0)
2479     {
2480       clib_warning ("No map-resolver configured");
2481       return 0;
2482     }
2483
2484   /* find the first mr ip we have a route to and the ip of the
2485    * iface that has a route to it */
2486   vec_foreach (mrit, lcm->map_resolvers)
2487   {
2488     a = &mrit->address;
2489     if (0 != ip_fib_get_first_egress_ip_for_dst (lcm, a, sloc))
2490       {
2491         ip_address_copy (mr_ip, a);
2492
2493         /* also update globals */
2494         return 1;
2495       }
2496   }
2497
2498   clib_warning ("Can't find map-resolver and local interface ip!");
2499   return 0;
2500 }
2501
2502 static gid_address_t *
2503 build_itr_rloc_list (lisp_cp_main_t * lcm, locator_set_t * loc_set)
2504 {
2505   void *addr;
2506   u32 i;
2507   locator_t *loc;
2508   u32 *loc_indexp;
2509   ip_interface_address_t *ia = 0;
2510   gid_address_t gid_data, *gid = &gid_data;
2511   gid_address_t *rlocs = 0;
2512   ip_prefix_t *ippref = &gid_address_ippref (gid);
2513   ip_address_t *rloc = &ip_prefix_addr (ippref);
2514
2515   clib_memset (gid, 0, sizeof (gid[0]));
2516   gid_address_type (gid) = GID_ADDR_IP_PREFIX;
2517   for (i = 0; i < vec_len (loc_set->locator_indices); i++)
2518     {
2519       loc_indexp = vec_elt_at_index (loc_set->locator_indices, i);
2520       loc = pool_elt_at_index (lcm->locator_pool, loc_indexp[0]);
2521
2522       /* Add ipv4 locators first TODO sort them */
2523
2524       /* *INDENT-OFF* */
2525       foreach_ip_interface_address (&lcm->im4->lookup_main, ia,
2526                                     loc->sw_if_index, 1 /* unnumbered */,
2527       ({
2528         addr = ip_interface_address_get_address (&lcm->im4->lookup_main, ia);
2529         ip_address_set (rloc, addr, IP4);
2530         ip_prefix_len (ippref) = 32;
2531         ip_prefix_normalize (ippref);
2532         vec_add1 (rlocs, gid[0]);
2533       }));
2534
2535       /* Add ipv6 locators */
2536       foreach_ip_interface_address (&lcm->im6->lookup_main, ia,
2537                                     loc->sw_if_index, 1 /* unnumbered */,
2538       ({
2539         addr = ip_interface_address_get_address (&lcm->im6->lookup_main, ia);
2540         ip_address_set (rloc, addr, IP6);
2541         ip_prefix_len (ippref) = 128;
2542         ip_prefix_normalize (ippref);
2543         vec_add1 (rlocs, gid[0]);
2544       }));
2545       /* *INDENT-ON* */
2546
2547     }
2548   return rlocs;
2549 }
2550
2551 static vlib_buffer_t *
2552 build_map_request (lisp_cp_main_t * lcm, gid_address_t * deid,
2553                    ip_address_t * sloc, ip_address_t * rloc,
2554                    gid_address_t * itr_rlocs, u64 * nonce_res, u32 * bi_res)
2555 {
2556   vlib_buffer_t *b;
2557   u32 bi;
2558   vlib_main_t *vm = lcm->vlib_main;
2559
2560   if (vlib_buffer_alloc (vm, &bi, 1) != 1)
2561     {
2562       clib_warning ("Can't allocate buffer for Map-Request!");
2563       return 0;
2564     }
2565
2566   b = vlib_get_buffer (vm, bi);
2567
2568   /* leave some space for the encap headers */
2569   vlib_buffer_make_headroom (b, MAX_LISP_MSG_ENCAP_LEN);
2570
2571   /* put lisp msg */
2572   lisp_msg_put_mreq (lcm, b, NULL, deid, itr_rlocs, 0 /* smr invoked */ ,
2573                      1 /* rloc probe */ , nonce_res);
2574
2575   /* push outer ip header */
2576   pkt_push_udp_and_ip (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, sloc,
2577                        rloc, 1);
2578
2579   bi_res[0] = bi;
2580
2581   return b;
2582 }
2583
2584 static vlib_buffer_t *
2585 build_encapsulated_map_request (lisp_cp_main_t * lcm,
2586                                 gid_address_t * seid, gid_address_t * deid,
2587                                 locator_set_t * loc_set, ip_address_t * mr_ip,
2588                                 ip_address_t * sloc, u8 is_smr_invoked,
2589                                 u64 * nonce_res, u32 * bi_res)
2590 {
2591   vlib_buffer_t *b;
2592   u32 bi;
2593   gid_address_t *rlocs = 0;
2594   vlib_main_t *vm = lcm->vlib_main;
2595
2596   if (vlib_buffer_alloc (vm, &bi, 1) != 1)
2597     {
2598       clib_warning ("Can't allocate buffer for Map-Request!");
2599       return 0;
2600     }
2601
2602   b = vlib_get_buffer (vm, bi);
2603   b->flags = 0;
2604
2605   /* leave some space for the encap headers */
2606   vlib_buffer_make_headroom (b, MAX_LISP_MSG_ENCAP_LEN);
2607
2608   /* get rlocs */
2609   rlocs = build_itr_rloc_list (lcm, loc_set);
2610
2611   if (MR_MODE_SRC_DST == lcm->map_request_mode
2612       && GID_ADDR_SRC_DST != gid_address_type (deid))
2613     {
2614       gid_address_t sd;
2615       clib_memset (&sd, 0, sizeof (sd));
2616       build_src_dst (&sd, seid, deid);
2617       lisp_msg_put_mreq (lcm, b, seid, &sd, rlocs, is_smr_invoked,
2618                          0 /* rloc probe */ , nonce_res);
2619     }
2620   else
2621     {
2622       /* put lisp msg */
2623       lisp_msg_put_mreq (lcm, b, seid, deid, rlocs, is_smr_invoked,
2624                          0 /* rloc probe */ , nonce_res);
2625     }
2626
2627   /* push ecm: udp-ip-lisp */
2628   lisp_msg_push_ecm (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, seid, deid);
2629
2630   /* push outer ip header */
2631   pkt_push_udp_and_ip (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, sloc,
2632                        mr_ip, 1);
2633
2634   bi_res[0] = bi;
2635
2636   vec_free (rlocs);
2637   return b;
2638 }
2639
2640 static void
2641 reset_pending_mr_counters (pending_map_request_t * r)
2642 {
2643   r->time_to_expire = PENDING_MREQ_EXPIRATION_TIME;
2644   r->retries_num = 0;
2645 }
2646
2647 #define foreach_msmr \
2648   _(server) \
2649   _(resolver)
2650
2651 #define _(name) \
2652 static int                                                              \
2653 elect_map_ ## name (lisp_cp_main_t * lcm)                               \
2654 {                                                                       \
2655   lisp_msmr_t *mr;                                                      \
2656   vec_foreach (mr, lcm->map_ ## name ## s)                              \
2657   {                                                                     \
2658     if (!mr->is_down)                                                   \
2659       {                                                                 \
2660         ip_address_copy (&lcm->active_map_ ##name, &mr->address);       \
2661         lcm->do_map_ ## name ## _election = 0;                          \
2662         return 1;                                                       \
2663       }                                                                 \
2664   }                                                                     \
2665   return 0;                                                             \
2666 }
2667 foreach_msmr
2668 #undef _
2669   static void
2670 free_map_register_records (mapping_t * maps)
2671 {
2672   mapping_t *map;
2673   vec_foreach (map, maps) vec_free (map->locators);
2674
2675   vec_free (maps);
2676 }
2677
2678 static void
2679 add_locators (lisp_cp_main_t * lcm, mapping_t * m, u32 locator_set_index,
2680               ip_address_t * probed_loc)
2681 {
2682   u32 *li;
2683   locator_t *loc, new;
2684   ip_interface_address_t *ia = 0;
2685   void *addr;
2686   ip_address_t *new_ip = &gid_address_ip (&new.address);
2687
2688   m->locators = 0;
2689   locator_set_t *ls = pool_elt_at_index (lcm->locator_set_pool,
2690                                          locator_set_index);
2691   vec_foreach (li, ls->locator_indices)
2692   {
2693     loc = pool_elt_at_index (lcm->locator_pool, li[0]);
2694     new = loc[0];
2695     if (loc->local)
2696       {
2697           /* *INDENT-OFF* */
2698           foreach_ip_interface_address (&lcm->im4->lookup_main, ia,
2699                                         loc->sw_if_index, 1 /* unnumbered */,
2700           ({
2701             addr = ip_interface_address_get_address (&lcm->im4->lookup_main,
2702                                                      ia);
2703             ip_address_set (new_ip, addr, IP4);
2704           }));
2705
2706           /* Add ipv6 locators */
2707           foreach_ip_interface_address (&lcm->im6->lookup_main, ia,
2708                                         loc->sw_if_index, 1 /* unnumbered */,
2709           ({
2710             addr = ip_interface_address_get_address (&lcm->im6->lookup_main,
2711                                                      ia);
2712             ip_address_set (new_ip, addr, IP6);
2713           }));
2714           /* *INDENT-ON* */
2715
2716         if (probed_loc && ip_address_cmp (probed_loc, new_ip) == 0)
2717           new.probed = 1;
2718       }
2719     vec_add1 (m->locators, new);
2720   }
2721 }
2722
2723 static mapping_t *
2724 build_map_register_record_list (lisp_cp_main_t * lcm)
2725 {
2726   mapping_t *recs = 0, rec, *m;
2727
2728   /* *INDENT-OFF* */
2729   pool_foreach(m, lcm->mapping_pool,
2730   {
2731     /* for now build only local mappings */
2732     if (!m->local)
2733       continue;
2734
2735     rec = m[0];
2736     add_locators (lcm, &rec, m->locator_set_index, NULL);
2737     vec_add1 (recs, rec);
2738   });
2739   /* *INDENT-ON* */
2740
2741   return recs;
2742 }
2743
2744 static int
2745 update_map_register_auth_data (map_register_hdr_t * map_reg_hdr,
2746                                lisp_key_type_t key_id, u8 * key,
2747                                u16 auth_data_len, u32 msg_len)
2748 {
2749   MREG_KEY_ID (map_reg_hdr) = clib_host_to_net_u16 (key_id);
2750   MREG_AUTH_DATA_LEN (map_reg_hdr) = clib_host_to_net_u16 (auth_data_len);
2751
2752   unsigned char *result = HMAC (get_encrypt_fcn (key_id), key, vec_len (key),
2753                                 (unsigned char *) map_reg_hdr, msg_len, NULL,
2754                                 NULL);
2755   clib_memcpy (MREG_DATA (map_reg_hdr), result, auth_data_len);
2756
2757   return 0;
2758 }
2759
2760 static vlib_buffer_t *
2761 build_map_register (lisp_cp_main_t * lcm, ip_address_t * sloc,
2762                     ip_address_t * ms_ip, u64 * nonce_res, u8 want_map_notif,
2763                     mapping_t * records, lisp_key_type_t key_id, u8 * key,
2764                     u32 * bi_res)
2765 {
2766   void *map_reg_hdr;
2767   vlib_buffer_t *b;
2768   u32 bi, auth_data_len = 0, msg_len = 0;
2769   vlib_main_t *vm = lcm->vlib_main;
2770
2771   if (vlib_buffer_alloc (vm, &bi, 1) != 1)
2772     {
2773       clib_warning ("Can't allocate buffer for Map-Register!");
2774       return 0;
2775     }
2776
2777   b = vlib_get_buffer (vm, bi);
2778
2779   /* leave some space for the encap headers */
2780   vlib_buffer_make_headroom (b, MAX_LISP_MSG_ENCAP_LEN);
2781
2782   auth_data_len = auth_data_len_by_key_id (key_id);
2783   map_reg_hdr = lisp_msg_put_map_register (b, records, want_map_notif,
2784                                            auth_data_len, nonce_res,
2785                                            &msg_len);
2786
2787   update_map_register_auth_data (map_reg_hdr, key_id, key, auth_data_len,
2788                                  msg_len);
2789
2790   /* push outer ip header */
2791   pkt_push_udp_and_ip (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, sloc,
2792                        ms_ip, 1);
2793
2794   bi_res[0] = bi;
2795   return b;
2796 }
2797
2798 #define _(name) \
2799 static int                                                              \
2800 get_egress_map_ ##name## _ip (lisp_cp_main_t * lcm, ip_address_t * ip)  \
2801 {                                                                       \
2802   lisp_msmr_t *mr;                                                      \
2803   while (lcm->do_map_ ## name ## _election                              \
2804          | (0 == ip_fib_get_first_egress_ip_for_dst                     \
2805             (lcm, &lcm->active_map_ ##name, ip)))                       \
2806     {                                                                   \
2807       if (0 == elect_map_ ## name (lcm))                                \
2808         /* all map resolvers/servers are down */                        \
2809         {                                                               \
2810           /* restart MR/MS checking by marking all of them up */        \
2811           vec_foreach (mr, lcm->map_ ## name ## s) mr->is_down = 0;     \
2812           return -1;                                                    \
2813         }                                                               \
2814     }                                                                   \
2815   return 0;                                                             \
2816 }
2817
2818 foreach_msmr
2819 #undef _
2820 /* CP output statistics */
2821 #define foreach_lisp_cp_output_error                  \
2822 _(MAP_REGISTERS_SENT, "map-registers sent")           \
2823 _(MAP_REQUESTS_SENT, "map-requests sent")             \
2824 _(RLOC_PROBES_SENT, "rloc-probes sent")
2825 static char *lisp_cp_output_error_strings[] = {
2826 #define _(sym,string) string,
2827   foreach_lisp_cp_output_error
2828 #undef _
2829 };
2830
2831 typedef enum
2832 {
2833 #define _(sym,str) LISP_CP_OUTPUT_ERROR_##sym,
2834   foreach_lisp_cp_output_error
2835 #undef _
2836     LISP_CP_OUTPUT_N_ERROR,
2837 } lisp_cp_output_error_t;
2838
2839 static uword
2840 lisp_cp_output (vlib_main_t * vm, vlib_node_runtime_t * node,
2841                 vlib_frame_t * from_frame)
2842 {
2843   return 0;
2844 }
2845
2846 /* dummy node used only for statistics */
2847 /* *INDENT-OFF* */
2848 VLIB_REGISTER_NODE (lisp_cp_output_node) = {
2849   .function = lisp_cp_output,
2850   .name = "lisp-cp-output",
2851   .vector_size = sizeof (u32),
2852   .format_trace = format_lisp_cp_input_trace,
2853   .type = VLIB_NODE_TYPE_INTERNAL,
2854
2855   .n_errors = LISP_CP_OUTPUT_N_ERROR,
2856   .error_strings = lisp_cp_output_error_strings,
2857
2858   .n_next_nodes = LISP_CP_INPUT_N_NEXT,
2859
2860   .next_nodes = {
2861       [LISP_CP_INPUT_NEXT_DROP] = "error-drop",
2862   },
2863 };
2864 /* *INDENT-ON* */
2865
2866 static int
2867 send_rloc_probe (lisp_cp_main_t * lcm, gid_address_t * deid,
2868                  u32 local_locator_set_index, ip_address_t * sloc,
2869                  ip_address_t * rloc)
2870 {
2871   locator_set_t *ls;
2872   u32 bi;
2873   vlib_buffer_t *b;
2874   vlib_frame_t *f;
2875   u64 nonce = 0;
2876   u32 next_index, *to_next;
2877   gid_address_t *itr_rlocs;
2878
2879   ls = pool_elt_at_index (lcm->locator_set_pool, local_locator_set_index);
2880   itr_rlocs = build_itr_rloc_list (lcm, ls);
2881
2882   b = build_map_request (lcm, deid, sloc, rloc, itr_rlocs, &nonce, &bi);
2883   vec_free (itr_rlocs);
2884   if (!b)
2885     return -1;
2886
2887   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
2888
2889   next_index = (ip_addr_version (rloc) == IP4) ?
2890     ip4_lookup_node.index : ip6_lookup_node.index;
2891
2892   f = vlib_get_frame_to_node (lcm->vlib_main, next_index);
2893
2894   /* Enqueue the packet */
2895   to_next = vlib_frame_vector_args (f);
2896   to_next[0] = bi;
2897   f->n_vectors = 1;
2898   vlib_put_frame_to_node (lcm->vlib_main, next_index, f);
2899
2900   return 0;
2901 }
2902
2903 static int
2904 send_rloc_probes (lisp_cp_main_t * lcm)
2905 {
2906   u8 lprio = 0;
2907   mapping_t *lm;
2908   fwd_entry_t *e;
2909   locator_pair_t *lp;
2910   u32 si, rloc_probes_sent = 0;
2911
2912   /* *INDENT-OFF* */
2913   pool_foreach (e, lcm->fwd_entry_pool,
2914   {
2915     if (vec_len (e->locator_pairs) == 0)
2916       continue;
2917
2918     si = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &e->leid);
2919     if (~0 == si)
2920       {
2921         clib_warning ("internal error: cannot find local eid %U in "
2922                       "map-cache!", format_gid_address, &e->leid);
2923         continue;
2924       }
2925     lm = pool_elt_at_index (lcm->mapping_pool, si);
2926
2927     /* get the best (lowest) priority */
2928     lprio = e->locator_pairs[0].priority;
2929
2930     /* send rloc-probe for pair(s) with the best remote locator priority */
2931     vec_foreach (lp, e->locator_pairs)
2932       {
2933         if (lp->priority != lprio)
2934           break;
2935
2936         /* get first remote locator */
2937         send_rloc_probe (lcm, &e->reid, lm->locator_set_index, &lp->lcl_loc,
2938                          &lp->rmt_loc);
2939         rloc_probes_sent++;
2940       }
2941   });
2942   /* *INDENT-ON* */
2943
2944   vlib_node_increment_counter (vlib_get_main (), lisp_cp_output_node.index,
2945                                LISP_CP_OUTPUT_ERROR_RLOC_PROBES_SENT,
2946                                rloc_probes_sent);
2947   return 0;
2948 }
2949
2950 static int
2951 send_map_register (lisp_cp_main_t * lcm, u8 want_map_notif)
2952 {
2953   pending_map_register_t *pmr;
2954   u32 bi, map_registers_sent = 0;
2955   vlib_buffer_t *b;
2956   ip_address_t sloc;
2957   vlib_frame_t *f;
2958   u64 nonce = 0;
2959   u32 next_index, *to_next;
2960   mapping_t *records, *r, *group, *k;
2961
2962   if (get_egress_map_server_ip (lcm, &sloc) < 0)
2963     return -1;
2964
2965   records = build_map_register_record_list (lcm);
2966   if (!records)
2967     return -1;
2968
2969   vec_foreach (r, records)
2970   {
2971     u8 *key = r->key;
2972     u8 key_id = r->key_id;
2973
2974     if (!key)
2975       continue;                 /* no secret key -> map-register cannot be sent */
2976
2977     group = 0;
2978     vec_add1 (group, r[0]);
2979
2980     /* group mappings that share common key */
2981     for (k = r + 1; k < vec_end (records); k++)
2982       {
2983         if (k->key_id != r->key_id)
2984           continue;
2985
2986         if (vec_is_equal (k->key, r->key))
2987           {
2988             vec_add1 (group, k[0]);
2989             k->key = 0;         /* don't process this mapping again */
2990           }
2991       }
2992
2993     b = build_map_register (lcm, &sloc, &lcm->active_map_server, &nonce,
2994                             want_map_notif, group, key_id, key, &bi);
2995     vec_free (group);
2996     if (!b)
2997       continue;
2998
2999     vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
3000
3001     next_index = (ip_addr_version (&lcm->active_map_server) == IP4) ?
3002       ip4_lookup_node.index : ip6_lookup_node.index;
3003
3004     f = vlib_get_frame_to_node (lcm->vlib_main, next_index);
3005
3006     /* Enqueue the packet */
3007     to_next = vlib_frame_vector_args (f);
3008     to_next[0] = bi;
3009     f->n_vectors = 1;
3010     vlib_put_frame_to_node (lcm->vlib_main, next_index, f);
3011     map_registers_sent++;
3012
3013     pool_get (lcm->pending_map_registers_pool, pmr);
3014     clib_memset (pmr, 0, sizeof (*pmr));
3015     pmr->time_to_expire = PENDING_MREG_EXPIRATION_TIME;
3016     hash_set (lcm->map_register_messages_by_nonce, nonce,
3017               pmr - lcm->pending_map_registers_pool);
3018   }
3019   free_map_register_records (records);
3020
3021   vlib_node_increment_counter (vlib_get_main (), lisp_cp_output_node.index,
3022                                LISP_CP_OUTPUT_ERROR_MAP_REGISTERS_SENT,
3023                                map_registers_sent);
3024
3025   return 0;
3026 }
3027
3028 #define send_encapsulated_map_request(lcm, seid, deid, smr) \
3029   _send_encapsulated_map_request(lcm, seid, deid, smr, 0)
3030
3031 #define resend_encapsulated_map_request(lcm, seid, deid, smr) \
3032   _send_encapsulated_map_request(lcm, seid, deid, smr, 1)
3033
3034 static int
3035 _send_encapsulated_map_request (lisp_cp_main_t * lcm,
3036                                 gid_address_t * seid, gid_address_t * deid,
3037                                 u8 is_smr_invoked, u8 is_resend)
3038 {
3039   u32 next_index, bi = 0, *to_next, map_index;
3040   vlib_buffer_t *b;
3041   vlib_frame_t *f;
3042   u64 nonce = 0;
3043   locator_set_t *loc_set;
3044   mapping_t *map;
3045   pending_map_request_t *pmr, *duplicate_pmr = 0;
3046   ip_address_t sloc;
3047   u32 ls_index;
3048
3049   /* if there is already a pending request remember it */
3050
3051   /* *INDENT-OFF* */
3052   pool_foreach(pmr, lcm->pending_map_requests_pool,
3053   ({
3054     if (!gid_address_cmp (&pmr->src, seid)
3055         && !gid_address_cmp (&pmr->dst, deid))
3056       {
3057         duplicate_pmr = pmr;
3058         break;
3059       }
3060   }));
3061   /* *INDENT-ON* */
3062
3063   if (!is_resend && duplicate_pmr)
3064     {
3065       /* don't send the request if there is a pending map request already */
3066       return 0;
3067     }
3068
3069   u8 pitr_mode = lcm->flags & LISP_FLAG_PITR_MODE;
3070
3071   /* get locator-set for seid */
3072   if (!pitr_mode && gid_address_type (deid) != GID_ADDR_NSH)
3073     {
3074       map_index = gid_dictionary_lookup (&lcm->mapping_index_by_gid, seid);
3075       if (map_index == ~0)
3076         {
3077           clib_warning ("No local mapping found in eid-table for %U!",
3078                         format_gid_address, seid);
3079           return -1;
3080         }
3081
3082       map = pool_elt_at_index (lcm->mapping_pool, map_index);
3083
3084       if (!map->local)
3085         {
3086           clib_warning
3087             ("Mapping found for src eid %U is not marked as local!",
3088              format_gid_address, seid);
3089           return -1;
3090         }
3091       ls_index = map->locator_set_index;
3092     }
3093   else
3094     {
3095       if (pitr_mode)
3096         {
3097           if (lcm->pitr_map_index != ~0)
3098             {
3099               map =
3100                 pool_elt_at_index (lcm->mapping_pool, lcm->pitr_map_index);
3101               ls_index = map->locator_set_index;
3102             }
3103           else
3104             {
3105               return -1;
3106             }
3107         }
3108       else
3109         {
3110           if (lcm->nsh_map_index == (u32) ~ 0)
3111             {
3112               clib_warning ("No locator-set defined for NSH!");
3113               return -1;
3114             }
3115           else
3116             {
3117               map = pool_elt_at_index (lcm->mapping_pool, lcm->nsh_map_index);
3118               ls_index = map->locator_set_index;
3119             }
3120         }
3121     }
3122
3123   /* overwrite locator set if map-request itr-rlocs configured */
3124   if (~0 != lcm->mreq_itr_rlocs)
3125     {
3126       ls_index = lcm->mreq_itr_rlocs;
3127     }
3128
3129   loc_set = pool_elt_at_index (lcm->locator_set_pool, ls_index);
3130
3131   if (get_egress_map_resolver_ip (lcm, &sloc) < 0)
3132     {
3133       if (duplicate_pmr)
3134         duplicate_pmr->to_be_removed = 1;
3135       return -1;
3136     }
3137
3138   /* build the encapsulated map request */
3139   b = build_encapsulated_map_request (lcm, seid, deid, loc_set,
3140                                       &lcm->active_map_resolver,
3141                                       &sloc, is_smr_invoked, &nonce, &bi);
3142
3143   if (!b)
3144     return -1;
3145
3146   /* set fib index to default and lookup node */
3147   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
3148   next_index = (ip_addr_version (&lcm->active_map_resolver) == IP4) ?
3149     ip4_lookup_node.index : ip6_lookup_node.index;
3150
3151   f = vlib_get_frame_to_node (lcm->vlib_main, next_index);
3152
3153   /* Enqueue the packet */
3154   to_next = vlib_frame_vector_args (f);
3155   to_next[0] = bi;
3156   f->n_vectors = 1;
3157   vlib_put_frame_to_node (lcm->vlib_main, next_index, f);
3158
3159   vlib_node_increment_counter (vlib_get_main (), lisp_cp_output_node.index,
3160                                LISP_CP_OUTPUT_ERROR_MAP_REQUESTS_SENT, 1);
3161
3162   if (duplicate_pmr)
3163     /* if there is a pending request already update it */
3164     {
3165       if (clib_fifo_elts (duplicate_pmr->nonces) >= PENDING_MREQ_QUEUE_LEN)
3166         {
3167           /* remove the oldest nonce */
3168           u64 CLIB_UNUSED (tmp), *nonce_del;
3169           nonce_del = clib_fifo_head (duplicate_pmr->nonces);
3170           hash_unset (lcm->pending_map_requests_by_nonce, nonce_del[0]);
3171           clib_fifo_sub1 (duplicate_pmr->nonces, tmp);
3172         }
3173
3174       clib_fifo_add1 (duplicate_pmr->nonces, nonce);
3175       hash_set (lcm->pending_map_requests_by_nonce, nonce,
3176                 duplicate_pmr - lcm->pending_map_requests_pool);
3177     }
3178   else
3179     {
3180       /* add map-request to pending requests table */
3181       pool_get (lcm->pending_map_requests_pool, pmr);
3182       clib_memset (pmr, 0, sizeof (*pmr));
3183       gid_address_copy (&pmr->src, seid);
3184       gid_address_copy (&pmr->dst, deid);
3185       clib_fifo_add1 (pmr->nonces, nonce);
3186       pmr->is_smr_invoked = is_smr_invoked;
3187       reset_pending_mr_counters (pmr);
3188       hash_set (lcm->pending_map_requests_by_nonce, nonce,
3189                 pmr - lcm->pending_map_requests_pool);
3190     }
3191
3192   return 0;
3193 }
3194
3195 static void
3196 get_src_and_dst_ip (void *hdr, ip_address_t * src, ip_address_t * dst)
3197 {
3198   ip4_header_t *ip4 = hdr;
3199   ip6_header_t *ip6;
3200
3201   if ((ip4->ip_version_and_header_length & 0xF0) == 0x40)
3202     {
3203       ip_address_set (src, &ip4->src_address, IP4);
3204       ip_address_set (dst, &ip4->dst_address, IP4);
3205     }
3206   else
3207     {
3208       ip6 = hdr;
3209       ip_address_set (src, &ip6->src_address, IP6);
3210       ip_address_set (dst, &ip6->dst_address, IP6);
3211     }
3212 }
3213
3214 static u32
3215 lisp_get_vni_from_buffer_ip (lisp_cp_main_t * lcm, vlib_buffer_t * b,
3216                              u8 version)
3217 {
3218   uword *vnip;
3219   u32 vni = ~0, table_id = ~0;
3220
3221   table_id = fib_table_get_table_id_for_sw_if_index ((version ==
3222                                                       IP4 ? FIB_PROTOCOL_IP4 :
3223                                                       FIB_PROTOCOL_IP6),
3224                                                      vnet_buffer
3225                                                      (b)->sw_if_index
3226                                                      [VLIB_RX]);
3227
3228   vnip = hash_get (lcm->vni_by_table_id, table_id);
3229   if (vnip)
3230     vni = vnip[0];
3231   else
3232     clib_warning ("vrf %d is not mapped to any vni!", table_id);
3233
3234   return vni;
3235 }
3236
3237 always_inline u32
3238 lisp_get_bd_from_buffer_eth (vlib_buffer_t * b)
3239 {
3240   u32 sw_if_index0;
3241
3242   l2input_main_t *l2im = &l2input_main;
3243   l2_input_config_t *config;
3244   l2_bridge_domain_t *bd_config;
3245
3246   sw_if_index0 = vnet_buffer (b)->sw_if_index[VLIB_RX];
3247   config = vec_elt_at_index (l2im->configs, sw_if_index0);
3248   bd_config = vec_elt_at_index (l2im->bd_configs, config->bd_index);
3249
3250   return bd_config->bd_id;
3251 }
3252
3253 always_inline u32
3254 lisp_get_vni_from_buffer_eth (lisp_cp_main_t * lcm, vlib_buffer_t * b)
3255 {
3256   uword *vnip;
3257   u32 vni = ~0;
3258   u32 bd = lisp_get_bd_from_buffer_eth (b);
3259
3260   vnip = hash_get (lcm->vni_by_bd_id, bd);
3261   if (vnip)
3262     vni = vnip[0];
3263   else
3264     clib_warning ("bridge domain %d is not mapped to any vni!", bd);
3265
3266   return vni;
3267 }
3268
3269 void
3270 get_src_and_dst_eids_from_buffer (lisp_cp_main_t * lcm, vlib_buffer_t * b,
3271                                   gid_address_t * src, gid_address_t * dst,
3272                                   u16 type)
3273 {
3274   ethernet_header_t *eh;
3275   u32 vni = 0;
3276   icmp6_neighbor_discovery_ethernet_link_layer_address_option_t *opt;
3277
3278   clib_memset (src, 0, sizeof (*src));
3279   clib_memset (dst, 0, sizeof (*dst));
3280
3281   gid_address_type (dst) = GID_ADDR_NO_ADDRESS;
3282   gid_address_type (src) = GID_ADDR_NO_ADDRESS;
3283
3284   if (LISP_AFI_IP == type || LISP_AFI_IP6 == type)
3285     {
3286       ip4_header_t *ip;
3287       u8 version, preflen;
3288
3289       gid_address_type (src) = GID_ADDR_IP_PREFIX;
3290       gid_address_type (dst) = GID_ADDR_IP_PREFIX;
3291
3292       ip = vlib_buffer_get_current (b);
3293       get_src_and_dst_ip (ip, &gid_address_ip (src), &gid_address_ip (dst));
3294
3295       version = gid_address_ip_version (src);
3296       preflen = ip_address_max_len (version);
3297       gid_address_ippref_len (src) = preflen;
3298       gid_address_ippref_len (dst) = preflen;
3299
3300       vni = lisp_get_vni_from_buffer_ip (lcm, b, version);
3301       gid_address_vni (dst) = vni;
3302       gid_address_vni (src) = vni;
3303     }
3304   else if (LISP_AFI_MAC == type)
3305     {
3306       ethernet_arp_header_t *ah;
3307
3308       eh = vlib_buffer_get_current (b);
3309
3310       if (clib_net_to_host_u16 (eh->type) == ETHERNET_TYPE_ARP)
3311         {
3312           ah = (ethernet_arp_header_t *) (((u8 *) eh) + sizeof (*eh));
3313           gid_address_type (dst) = GID_ADDR_ARP;
3314
3315           if (clib_net_to_host_u16 (ah->opcode)
3316               != ETHERNET_ARP_OPCODE_request)
3317             {
3318               clib_memset (&gid_address_arp_ndp_ip (dst), 0,
3319                            sizeof (ip_address_t));
3320               ip_addr_version (&gid_address_arp_ndp_ip (dst)) = IP4;
3321               gid_address_arp_ndp_bd (dst) = ~0;
3322               return;
3323             }
3324
3325           gid_address_arp_bd (dst) = lisp_get_bd_from_buffer_eth (b);
3326           clib_memcpy (&gid_address_arp_ip4 (dst),
3327                        &ah->ip4_over_ethernet[1].ip4, 4);
3328         }
3329       else
3330         {
3331           if (clib_net_to_host_u16 (eh->type) == ETHERNET_TYPE_IP6)
3332             {
3333               ip6_header_t *ip;
3334               ip = (ip6_header_t *) (eh + 1);
3335
3336               if (IP_PROTOCOL_ICMP6 == ip->protocol)
3337                 {
3338                   icmp6_neighbor_solicitation_or_advertisement_header_t *ndh;
3339                   ndh = ip6_next_header (ip);
3340                   if (ndh->icmp.type == ICMP6_neighbor_solicitation)
3341                     {
3342                       gid_address_type (dst) = GID_ADDR_NDP;
3343
3344                       /* check that source link layer address option is present */
3345                       opt = (void *) (ndh + 1);
3346                       if ((opt->header.type !=
3347                            ICMP6_NEIGHBOR_DISCOVERY_OPTION_source_link_layer_address)
3348                           || (opt->header.n_data_u64s != 1))
3349                         {
3350                           clib_memset (&gid_address_arp_ndp_ip (dst), 0,
3351                                        sizeof (ip_address_t));
3352                           ip_addr_version (&gid_address_arp_ndp_ip (dst)) =
3353                             IP6;
3354                           gid_address_arp_ndp_bd (dst) = ~0;
3355                           gid_address_type (src) = GID_ADDR_NO_ADDRESS;
3356                           return;
3357                         }
3358
3359                       gid_address_ndp_bd (dst) =
3360                         lisp_get_bd_from_buffer_eth (b);
3361                       ip_address_set (&gid_address_arp_ndp_ip (dst),
3362                                       &ndh->target_address, IP6);
3363                       return;
3364                     }
3365                 }
3366             }
3367
3368           gid_address_type (src) = GID_ADDR_MAC;
3369           gid_address_type (dst) = GID_ADDR_MAC;
3370           mac_copy (&gid_address_mac (src), eh->src_address);
3371           mac_copy (&gid_address_mac (dst), eh->dst_address);
3372
3373           /* get vni */
3374           vni = lisp_get_vni_from_buffer_eth (lcm, b);
3375
3376           gid_address_vni (dst) = vni;
3377           gid_address_vni (src) = vni;
3378         }
3379     }
3380   else if (LISP_AFI_LCAF == type)
3381     {
3382       lisp_nsh_hdr_t *nh;
3383       eh = vlib_buffer_get_current (b);
3384
3385       if (clib_net_to_host_u16 (eh->type) == ETHERNET_TYPE_NSH)
3386         {
3387           nh = (lisp_nsh_hdr_t *) (((u8 *) eh) + sizeof (*eh));
3388           u32 spi = clib_net_to_host_u32 (nh->spi_si << 8);
3389           u8 si = (u8) clib_net_to_host_u32 (nh->spi_si);
3390           gid_address_nsh_spi (dst) = spi;
3391           gid_address_nsh_si (dst) = si;
3392
3393           gid_address_type (dst) = GID_ADDR_NSH;
3394           gid_address_type (src) = GID_ADDR_NSH;
3395         }
3396     }
3397 }
3398
3399 static uword
3400 lisp_cp_lookup_inline (vlib_main_t * vm,
3401                        vlib_node_runtime_t * node,
3402                        vlib_frame_t * from_frame, int overlay)
3403 {
3404   icmp6_neighbor_discovery_ethernet_link_layer_address_option_t *opt;
3405   u32 *from, *to_next, di, si;
3406   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3407   u32 next_index;
3408   uword n_left_from, n_left_to_next;
3409   vnet_main_t *vnm = vnet_get_main ();
3410
3411   from = vlib_frame_vector_args (from_frame);
3412   n_left_from = from_frame->n_vectors;
3413   next_index = node->cached_next_index;
3414
3415   while (n_left_from > 0)
3416     {
3417       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
3418
3419       while (n_left_from > 0 && n_left_to_next > 0)
3420         {
3421           u32 pi0, sw_if_index0, next0;
3422           u64 mac0;
3423           vlib_buffer_t *b0;
3424           gid_address_t src, dst;
3425           ethernet_arp_header_t *arp0;
3426           ethernet_header_t *eth0;
3427           vnet_hw_interface_t *hw_if0;
3428           ethernet_header_t *eh0;
3429           icmp6_neighbor_solicitation_or_advertisement_header_t *ndh;
3430           ip6_header_t *ip0;
3431
3432           pi0 = from[0];
3433           from += 1;
3434           n_left_from -= 1;
3435           to_next[0] = pi0;
3436           to_next += 1;
3437           n_left_to_next -= 1;
3438
3439           b0 = vlib_get_buffer (vm, pi0);
3440
3441           /* src/dst eid pair */
3442           get_src_and_dst_eids_from_buffer (lcm, b0, &src, &dst, overlay);
3443
3444           if (gid_address_type (&dst) == GID_ADDR_ARP)
3445             {
3446               mac0 = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &dst);
3447               if (GID_LOOKUP_MISS_L2 == mac0)
3448                 goto drop;
3449
3450               /* send ARP reply */
3451               sw_if_index0 = vnet_buffer (b0)->sw_if_index[VLIB_RX];
3452               vnet_buffer (b0)->sw_if_index[VLIB_TX] = sw_if_index0;
3453
3454               hw_if0 = vnet_get_sup_hw_interface (vnm, sw_if_index0);
3455
3456               eth0 = vlib_buffer_get_current (b0);
3457               arp0 = (ethernet_arp_header_t *) (((u8 *) eth0)
3458                                                 + sizeof (*eth0));
3459               arp0->opcode = clib_host_to_net_u16 (ETHERNET_ARP_OPCODE_reply);
3460               arp0->ip4_over_ethernet[1] = arp0->ip4_over_ethernet[0];
3461               mac_address_from_u64 (&arp0->ip4_over_ethernet[0].mac, mac0);
3462               clib_memcpy (&arp0->ip4_over_ethernet[0].ip4,
3463                            &gid_address_arp_ip4 (&dst), 4);
3464
3465               /* Hardware must be ethernet-like. */
3466               ASSERT (vec_len (hw_if0->hw_address) == 6);
3467
3468               clib_memcpy (eth0->dst_address, eth0->src_address, 6);
3469               clib_memcpy (eth0->src_address, hw_if0->hw_address, 6);
3470
3471               b0->error = node->errors[LISP_CP_LOOKUP_ERROR_ARP_REPLY_TX];
3472               next0 = LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX;
3473               goto enqueue;
3474             }
3475           else if (gid_address_type (&dst) == GID_ADDR_NDP)
3476             {
3477               mac0 = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &dst);
3478               if (GID_LOOKUP_MISS_L2 == mac0)
3479                 goto drop;
3480
3481               sw_if_index0 = vnet_buffer (b0)->sw_if_index[VLIB_RX];
3482               vnet_buffer (b0)->sw_if_index[VLIB_TX] = sw_if_index0;
3483
3484               eh0 = vlib_buffer_get_current (b0);
3485               ip0 = (ip6_header_t *) (eh0 + 1);
3486               ndh = ip6_next_header (ip0);
3487               int bogus_length;
3488               ip0->dst_address = ip0->src_address;
3489               ip0->src_address = ndh->target_address;
3490               ip0->hop_limit = 255;
3491               opt = (void *) (ndh + 1);
3492               opt->header.type =
3493                 ICMP6_NEIGHBOR_DISCOVERY_OPTION_target_link_layer_address;
3494               clib_memcpy (opt->ethernet_address, (u8 *) & mac0, 6);
3495               ndh->icmp.type = ICMP6_neighbor_advertisement;
3496               ndh->advertisement_flags = clib_host_to_net_u32
3497                 (ICMP6_NEIGHBOR_ADVERTISEMENT_FLAG_SOLICITED |
3498                  ICMP6_NEIGHBOR_ADVERTISEMENT_FLAG_OVERRIDE);
3499               ndh->icmp.checksum = 0;
3500               ndh->icmp.checksum =
3501                 ip6_tcp_udp_icmp_compute_checksum (vm, b0, ip0,
3502                                                    &bogus_length);
3503               clib_memcpy (eh0->dst_address, eh0->src_address, 6);
3504               clib_memcpy (eh0->src_address, (u8 *) & mac0, 6);
3505               b0->error =
3506                 node->errors
3507                 [LISP_CP_LOOKUP_ERROR_NDP_NEIGHBOR_ADVERTISEMENT_TX];
3508               next0 = LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX;
3509               goto enqueue;
3510             }
3511
3512           /* if we have remote mapping for destination already in map-cache
3513              add forwarding tunnel directly. If not send a map-request */
3514           di = gid_dictionary_sd_lookup (&lcm->mapping_index_by_gid, &dst,
3515                                          &src);
3516           if (~0 != di)
3517             {
3518               mapping_t *m = vec_elt_at_index (lcm->mapping_pool, di);
3519               /* send a map-request also in case of negative mapping entry
3520                  with corresponding action */
3521               if (m->action == LISP_SEND_MAP_REQUEST)
3522                 {
3523                   /* send map-request */
3524                   queue_map_request (&src, &dst, 0 /* smr_invoked */ ,
3525                                      0 /* is_resend */ );
3526                 }
3527               else
3528                 {
3529                   if (GID_ADDR_NSH != gid_address_type (&dst))
3530                     {
3531                       si = gid_dictionary_lookup (&lcm->mapping_index_by_gid,
3532                                                   &src);
3533                     }
3534                   else
3535                     si = lcm->nsh_map_index;
3536
3537                   if (~0 != si)
3538                     {
3539                       dp_add_fwd_entry_from_mt (si, di);
3540                     }
3541                 }
3542             }
3543           else
3544             {
3545               /* send map-request */
3546               queue_map_request (&src, &dst, 0 /* smr_invoked */ ,
3547                                  0 /* is_resend */ );
3548             }
3549
3550         drop:
3551           b0->error = node->errors[LISP_CP_LOOKUP_ERROR_DROP];
3552           next0 = LISP_CP_LOOKUP_NEXT_DROP;
3553         enqueue:
3554           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
3555             {
3556               lisp_cp_lookup_trace_t *tr = vlib_add_trace (vm, node, b0,
3557                                                            sizeof (*tr));
3558
3559               clib_memset (tr, 0, sizeof (*tr));
3560               gid_address_copy (&tr->dst_eid, &dst);
3561               ip_address_copy (&tr->map_resolver_ip,
3562                                &lcm->active_map_resolver);
3563             }
3564           gid_address_free (&dst);
3565           gid_address_free (&src);
3566           vlib_validate_buffer_enqueue_x1 (vm, node, next_index,
3567                                            to_next,
3568                                            n_left_to_next, pi0, next0);
3569         }
3570
3571       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
3572     }
3573   return from_frame->n_vectors;
3574 }
3575
3576 static uword
3577 lisp_cp_lookup_ip4 (vlib_main_t * vm,
3578                     vlib_node_runtime_t * node, vlib_frame_t * from_frame)
3579 {
3580   return (lisp_cp_lookup_inline (vm, node, from_frame, LISP_AFI_IP));
3581 }
3582
3583 static uword
3584 lisp_cp_lookup_ip6 (vlib_main_t * vm,
3585                     vlib_node_runtime_t * node, vlib_frame_t * from_frame)
3586 {
3587   return (lisp_cp_lookup_inline (vm, node, from_frame, LISP_AFI_IP6));
3588 }
3589
3590 static uword
3591 lisp_cp_lookup_l2 (vlib_main_t * vm,
3592                    vlib_node_runtime_t * node, vlib_frame_t * from_frame)
3593 {
3594   return (lisp_cp_lookup_inline (vm, node, from_frame, LISP_AFI_MAC));
3595 }
3596
3597 static uword
3598 lisp_cp_lookup_nsh (vlib_main_t * vm,
3599                     vlib_node_runtime_t * node, vlib_frame_t * from_frame)
3600 {
3601   /* TODO decide if NSH should be propagated as LCAF or not */
3602   return (lisp_cp_lookup_inline (vm, node, from_frame, LISP_AFI_LCAF));
3603 }
3604
3605 /* *INDENT-OFF* */
3606 VLIB_REGISTER_NODE (lisp_cp_lookup_ip4_node) = {
3607   .function = lisp_cp_lookup_ip4,
3608   .name = "lisp-cp-lookup-ip4",
3609   .vector_size = sizeof (u32),
3610   .format_trace = format_lisp_cp_lookup_trace,
3611   .type = VLIB_NODE_TYPE_INTERNAL,
3612
3613   .n_errors = LISP_CP_LOOKUP_N_ERROR,
3614   .error_strings = lisp_cp_lookup_error_strings,
3615
3616   .n_next_nodes = LISP_CP_LOOKUP_N_NEXT,
3617
3618   .next_nodes = {
3619       [LISP_CP_LOOKUP_NEXT_DROP] = "error-drop",
3620       [LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX] = "interface-output",
3621   },
3622 };
3623 /* *INDENT-ON* */
3624
3625 /* *INDENT-OFF* */
3626 VLIB_REGISTER_NODE (lisp_cp_lookup_ip6_node) = {
3627   .function = lisp_cp_lookup_ip6,
3628   .name = "lisp-cp-lookup-ip6",
3629   .vector_size = sizeof (u32),
3630   .format_trace = format_lisp_cp_lookup_trace,
3631   .type = VLIB_NODE_TYPE_INTERNAL,
3632
3633   .n_errors = LISP_CP_LOOKUP_N_ERROR,
3634   .error_strings = lisp_cp_lookup_error_strings,
3635
3636   .n_next_nodes = LISP_CP_LOOKUP_N_NEXT,
3637
3638   .next_nodes = {
3639       [LISP_CP_LOOKUP_NEXT_DROP] = "error-drop",
3640       [LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX] = "interface-output",
3641   },
3642 };
3643 /* *INDENT-ON* */
3644
3645 /* *INDENT-OFF* */
3646 VLIB_REGISTER_NODE (lisp_cp_lookup_l2_node) = {
3647   .function = lisp_cp_lookup_l2,
3648   .name = "lisp-cp-lookup-l2",
3649   .vector_size = sizeof (u32),
3650   .format_trace = format_lisp_cp_lookup_trace,
3651   .type = VLIB_NODE_TYPE_INTERNAL,
3652
3653   .n_errors = LISP_CP_LOOKUP_N_ERROR,
3654   .error_strings = lisp_cp_lookup_error_strings,
3655
3656   .n_next_nodes = LISP_CP_LOOKUP_N_NEXT,
3657
3658   .next_nodes = {
3659       [LISP_CP_LOOKUP_NEXT_DROP] = "error-drop",
3660       [LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX] = "interface-output",
3661   },
3662 };
3663 /* *INDENT-ON* */
3664
3665 /* *INDENT-OFF* */
3666 VLIB_REGISTER_NODE (lisp_cp_lookup_nsh_node) = {
3667   .function = lisp_cp_lookup_nsh,
3668   .name = "lisp-cp-lookup-nsh",
3669   .vector_size = sizeof (u32),
3670   .format_trace = format_lisp_cp_lookup_trace,
3671   .type = VLIB_NODE_TYPE_INTERNAL,
3672
3673   .n_errors = LISP_CP_LOOKUP_N_ERROR,
3674   .error_strings = lisp_cp_lookup_error_strings,
3675
3676   .n_next_nodes = LISP_CP_LOOKUP_N_NEXT,
3677
3678   .next_nodes = {
3679       [LISP_CP_LOOKUP_NEXT_DROP] = "error-drop",
3680       [LISP_CP_LOOKUP_NEXT_ARP_NDP_REPLY_TX] = "interface-output",
3681   },
3682 };
3683 /* *INDENT-ON* */
3684
3685 /* lisp_cp_input statistics */
3686 #define foreach_lisp_cp_input_error                               \
3687 _(DROP, "drop")                                                   \
3688 _(RLOC_PROBE_REQ_RECEIVED, "rloc-probe requests received")        \
3689 _(RLOC_PROBE_REP_RECEIVED, "rloc-probe replies received")         \
3690 _(MAP_NOTIFIES_RECEIVED, "map-notifies received")                 \
3691 _(MAP_REPLIES_RECEIVED, "map-replies received")
3692
3693 static char *lisp_cp_input_error_strings[] = {
3694 #define _(sym,string) string,
3695   foreach_lisp_cp_input_error
3696 #undef _
3697 };
3698
3699 typedef enum
3700 {
3701 #define _(sym,str) LISP_CP_INPUT_ERROR_##sym,
3702   foreach_lisp_cp_input_error
3703 #undef _
3704     LISP_CP_INPUT_N_ERROR,
3705 } lisp_cp_input_error_t;
3706
3707 typedef struct
3708 {
3709   gid_address_t dst_eid;
3710   ip4_address_t map_resolver_ip;
3711 } lisp_cp_input_trace_t;
3712
3713 u8 *
3714 format_lisp_cp_input_trace (u8 * s, va_list * args)
3715 {
3716   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
3717   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
3718   CLIB_UNUSED (lisp_cp_input_trace_t * t) =
3719     va_arg (*args, lisp_cp_input_trace_t *);
3720
3721   s = format (s, "LISP-CP-INPUT: TODO");
3722   return s;
3723 }
3724
3725 static void
3726 remove_expired_mapping (lisp_cp_main_t * lcm, u32 mi)
3727 {
3728   mapping_t *m;
3729   vnet_lisp_add_del_adjacency_args_t _adj_args, *adj_args = &_adj_args;
3730   clib_memset (adj_args, 0, sizeof (adj_args[0]));
3731
3732   m = pool_elt_at_index (lcm->mapping_pool, mi);
3733
3734   gid_address_copy (&adj_args->reid, &m->eid);
3735   adj_args->is_add = 0;
3736   if (vnet_lisp_add_del_adjacency (adj_args))
3737     clib_warning ("failed to del adjacency!");
3738
3739   vnet_lisp_del_mapping (&m->eid, NULL);
3740   mapping_delete_timer (lcm, mi);
3741 }
3742
3743 static void
3744 mapping_start_expiration_timer (lisp_cp_main_t * lcm, u32 mi,
3745                                 f64 expiration_time)
3746 {
3747   mapping_t *m;
3748   u64 now = clib_cpu_time_now ();
3749   u64 cpu_cps = lcm->vlib_main->clib_time.clocks_per_second;
3750   u64 exp_clock_time = now + expiration_time * cpu_cps;
3751
3752   m = pool_elt_at_index (lcm->mapping_pool, mi);
3753
3754   m->timer_set = 1;
3755   timing_wheel_insert (&lcm->wheel, exp_clock_time, mi);
3756 }
3757
3758 static void
3759 process_expired_mapping (lisp_cp_main_t * lcm, u32 mi)
3760 {
3761   int rv;
3762   vnet_lisp_gpe_add_del_fwd_entry_args_t _a, *a = &_a;
3763   mapping_t *m = pool_elt_at_index (lcm->mapping_pool, mi);
3764   uword *fei;
3765   fwd_entry_t *fe;
3766   vlib_counter_t c;
3767   u8 have_stats = 0;
3768
3769   if (m->delete_after_expiration)
3770     {
3771       remove_expired_mapping (lcm, mi);
3772       return;
3773     }
3774
3775   fei = hash_get (lcm->fwd_entry_by_mapping_index, mi);
3776   if (!fei)
3777     return;
3778
3779   fe = pool_elt_at_index (lcm->fwd_entry_pool, fei[0]);
3780
3781   clib_memset (a, 0, sizeof (*a));
3782   a->rmt_eid = fe->reid;
3783   if (fe->is_src_dst)
3784     a->lcl_eid = fe->leid;
3785   a->vni = gid_address_vni (&fe->reid);
3786
3787   rv = vnet_lisp_gpe_get_fwd_stats (a, &c);
3788   if (0 == rv)
3789     have_stats = 1;
3790
3791   if (m->almost_expired)
3792     {
3793       m->almost_expired = 0;    /* reset flag */
3794       if (have_stats)
3795         {
3796           if (m->packets != c.packets)
3797             {
3798               /* mapping is in use, re-fetch */
3799               map_request_args_t mr_args;
3800               clib_memset (&mr_args, 0, sizeof (mr_args));
3801               mr_args.seid = fe->leid;
3802               mr_args.deid = fe->reid;
3803
3804               send_map_request_thread_fn (&mr_args);
3805             }
3806           else
3807             remove_expired_mapping (lcm, mi);
3808         }
3809       else
3810         remove_expired_mapping (lcm, mi);
3811     }
3812   else
3813     {
3814       m->almost_expired = 1;
3815       mapping_start_expiration_timer (lcm, mi, TIME_UNTIL_REFETCH_OR_DELETE);
3816
3817       if (have_stats)
3818         /* save counter */
3819         m->packets = c.packets;
3820       else
3821         m->delete_after_expiration = 1;
3822     }
3823 }
3824
3825 static void
3826 map_records_arg_free (map_records_arg_t * a)
3827 {
3828   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3829   mapping_t *m;
3830   vec_foreach (m, a->mappings)
3831   {
3832     vec_free (m->locators);
3833     gid_address_free (&m->eid);
3834   }
3835   pool_put (lcm->map_records_args_pool[vlib_get_thread_index ()], a);
3836 }
3837
3838 void *
3839 process_map_reply (map_records_arg_t * a)
3840 {
3841   mapping_t *m;
3842   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3843   u32 dst_map_index = 0;
3844   pending_map_request_t *pmr;
3845   u64 *noncep;
3846   uword *pmr_index;
3847   u8 is_changed = 0;
3848
3849   if (a->is_rloc_probe)
3850     goto done;
3851
3852   /* Check pending requests table and nonce */
3853   pmr_index = hash_get (lcm->pending_map_requests_by_nonce, a->nonce);
3854   if (!pmr_index)
3855     {
3856       clib_warning ("No pending map-request entry with nonce %lu!", a->nonce);
3857       goto done;
3858     }
3859   pmr = pool_elt_at_index (lcm->pending_map_requests_pool, pmr_index[0]);
3860
3861   vec_foreach (m, a->mappings)
3862   {
3863     vnet_lisp_add_del_mapping_args_t _m_args, *m_args = &_m_args;
3864     clib_memset (m_args, 0, sizeof (m_args[0]));
3865     gid_address_copy (&m_args->eid, &m->eid);
3866     m_args->action = m->action;
3867     m_args->authoritative = m->authoritative;
3868     m_args->ttl = m->ttl;
3869     m_args->is_static = 0;
3870
3871     /* insert/update mappings cache */
3872     vnet_lisp_add_mapping (m_args, m->locators, &dst_map_index, &is_changed);
3873
3874     if (dst_map_index == (u32) ~ 0)
3875       continue;
3876
3877     if (is_changed)
3878       {
3879         /* try to program forwarding only if mapping saved or updated */
3880         vnet_lisp_add_del_adjacency_args_t _adj_args, *adj_args = &_adj_args;
3881         clib_memset (adj_args, 0, sizeof (adj_args[0]));
3882
3883         gid_address_copy (&adj_args->leid, &pmr->src);
3884         gid_address_copy (&adj_args->reid, &m->eid);
3885         adj_args->is_add = 1;
3886
3887         if (vnet_lisp_add_del_adjacency (adj_args))
3888           clib_warning ("failed to add adjacency!");
3889       }
3890
3891     if ((u32) ~ 0 != m->ttl)
3892       mapping_start_expiration_timer (lcm, dst_map_index,
3893                                       (m->ttl == 0) ? 0 : MAPPING_TIMEOUT);
3894   }
3895
3896   /* remove pending map request entry */
3897
3898   /* *INDENT-OFF* */
3899   clib_fifo_foreach (noncep, pmr->nonces, ({
3900     hash_unset(lcm->pending_map_requests_by_nonce, noncep[0]);
3901   }));
3902   /* *INDENT-ON* */
3903
3904   clib_fifo_free (pmr->nonces);
3905   pool_put (lcm->pending_map_requests_pool, pmr);
3906
3907 done:
3908   a->is_free = 1;
3909   return 0;
3910 }
3911
3912 static int
3913 is_auth_data_valid (map_notify_hdr_t * h, u32 msg_len,
3914                     lisp_key_type_t key_id, u8 * key)
3915 {
3916   u8 *auth_data = 0;
3917   u16 auth_data_len;
3918   int result;
3919
3920   auth_data_len = auth_data_len_by_key_id (key_id);
3921   if ((u16) ~ 0 == auth_data_len)
3922     {
3923       clib_warning ("invalid length for key_id %d!", key_id);
3924       return 0;
3925     }
3926
3927   /* save auth data */
3928   vec_validate (auth_data, auth_data_len - 1);
3929   clib_memcpy (auth_data, MNOTIFY_DATA (h), auth_data_len);
3930
3931   /* clear auth data */
3932   clib_memset (MNOTIFY_DATA (h), 0, auth_data_len);
3933
3934   /* get hash of the message */
3935   unsigned char *code = HMAC (get_encrypt_fcn (key_id), key, vec_len (key),
3936                               (unsigned char *) h, msg_len, NULL, NULL);
3937
3938   result = memcmp (code, auth_data, auth_data_len);
3939
3940   vec_free (auth_data);
3941
3942   return !result;
3943 }
3944
3945 static void
3946 process_map_notify (map_records_arg_t * a)
3947 {
3948   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3949   uword *pmr_index;
3950
3951   pmr_index = hash_get (lcm->map_register_messages_by_nonce, a->nonce);
3952   if (!pmr_index)
3953     {
3954       clib_warning ("No pending map-register entry with nonce %lu!",
3955                     a->nonce);
3956       return;
3957     }
3958
3959   a->is_free = 1;
3960   pool_put_index (lcm->pending_map_registers_pool, pmr_index[0]);
3961   hash_unset (lcm->map_register_messages_by_nonce, a->nonce);
3962
3963   /* reset map-notify counter */
3964   lcm->expired_map_registers = 0;
3965 }
3966
3967 static mapping_t *
3968 get_mapping (lisp_cp_main_t * lcm, gid_address_t * e)
3969 {
3970   u32 mi;
3971
3972   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, e);
3973   if (~0 == mi)
3974     {
3975       clib_warning ("eid %U not found in map-cache!", unformat_gid_address,
3976                     e);
3977       return 0;
3978     }
3979   return pool_elt_at_index (lcm->mapping_pool, mi);
3980 }
3981
3982 /**
3983  * When map-notify is received it is necessary that all EIDs in the record
3984  * list share common key. The key is then used to verify authentication
3985  * data in map-notify message.
3986  */
3987 static int
3988 map_record_integrity_check (lisp_cp_main_t * lcm, mapping_t * maps,
3989                             u32 key_id, u8 ** key_out)
3990 {
3991   u32 i, len = vec_len (maps);
3992   mapping_t *m;
3993
3994   /* get key of the first mapping */
3995   m = get_mapping (lcm, &maps[0].eid);
3996   if (!m || !m->key)
3997     return -1;
3998
3999   key_out[0] = m->key;
4000
4001   for (i = 1; i < len; i++)
4002     {
4003       m = get_mapping (lcm, &maps[i].eid);
4004       if (!m || !m->key)
4005         return -1;
4006
4007       if (key_id != m->key_id || vec_cmp (m->key, key_out[0]))
4008         {
4009           clib_warning ("keys does not match! %v, %v", key_out[0], m->key);
4010           return -1;
4011         }
4012     }
4013   return 0;
4014 }
4015
4016 static int
4017 parse_map_records (vlib_buffer_t * b, map_records_arg_t * a, u8 count)
4018 {
4019   locator_t *locators = 0;
4020   u32 i, len;
4021   gid_address_t deid;
4022   mapping_t m;
4023   locator_t *loc;
4024
4025   clib_memset (&m, 0, sizeof (m));
4026
4027   /* parse record eid */
4028   for (i = 0; i < count; i++)
4029     {
4030       locators = 0;
4031       len = lisp_msg_parse_mapping_record (b, &deid, &locators, NULL);
4032       if (len == ~0)
4033         {
4034           clib_warning ("Failed to parse mapping record!");
4035           vec_foreach (loc, locators) locator_free (loc);
4036           vec_free (locators);
4037           return -1;
4038         }
4039
4040       m.locators = locators;
4041       gid_address_copy (&m.eid, &deid);
4042       vec_add1 (a->mappings, m);
4043     }
4044
4045   return 0;
4046 }
4047
4048 static map_records_arg_t *
4049 map_record_args_get ()
4050 {
4051   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4052   map_records_arg_t *rec;
4053
4054   /* Cleanup first */
4055   /* *INDENT-OFF* */
4056   pool_foreach (rec, lcm->map_records_args_pool[vlib_get_thread_index()], ({
4057     if (rec->is_free)
4058       map_records_arg_free (rec);
4059   }));
4060   /* *INDENT-ON* */
4061
4062   pool_get (lcm->map_records_args_pool[vlib_get_thread_index ()], rec);
4063   return rec;
4064 }
4065
4066 static map_records_arg_t *
4067 parse_map_notify (vlib_buffer_t * b)
4068 {
4069   int rc = 0;
4070   map_notify_hdr_t *mnotif_hdr;
4071   lisp_key_type_t key_id;
4072   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4073   u8 *key = 0;
4074   gid_address_t deid;
4075   u16 auth_data_len = 0;
4076   u8 record_count;
4077   map_records_arg_t *a;
4078
4079   a = map_record_args_get ();
4080   clib_memset (a, 0, sizeof (*a));
4081   mnotif_hdr = vlib_buffer_get_current (b);
4082   vlib_buffer_pull (b, sizeof (*mnotif_hdr));
4083   clib_memset (&deid, 0, sizeof (deid));
4084
4085   a->nonce = MNOTIFY_NONCE (mnotif_hdr);
4086   key_id = clib_net_to_host_u16 (MNOTIFY_KEY_ID (mnotif_hdr));
4087   auth_data_len = auth_data_len_by_key_id (key_id);
4088
4089   /* advance buffer by authentication data */
4090   vlib_buffer_pull (b, auth_data_len);
4091
4092   record_count = MNOTIFY_REC_COUNT (mnotif_hdr);
4093   rc = parse_map_records (b, a, record_count);
4094   if (rc != 0)
4095     {
4096       map_records_arg_free (a);
4097       return 0;
4098     }
4099
4100   rc = map_record_integrity_check (lcm, a->mappings, key_id, &key);
4101   if (rc != 0)
4102     {
4103       map_records_arg_free (a);
4104       return 0;
4105     }
4106
4107   /* verify authentication data */
4108   if (!is_auth_data_valid (mnotif_hdr, vlib_buffer_get_tail (b)
4109                            - (u8 *) mnotif_hdr, key_id, key))
4110     {
4111       clib_warning ("Map-notify auth data verification failed for nonce "
4112                     "0x%lx!", a->nonce);
4113       map_records_arg_free (a);
4114       return 0;
4115     }
4116   return a;
4117 }
4118
4119 static vlib_buffer_t *
4120 build_map_reply (lisp_cp_main_t * lcm, ip_address_t * sloc,
4121                  ip_address_t * dst, u64 nonce, u8 probe_bit,
4122                  mapping_t * records, u16 dst_port, u32 * bi_res)
4123 {
4124   vlib_buffer_t *b;
4125   u32 bi;
4126   vlib_main_t *vm = lcm->vlib_main;
4127
4128   if (vlib_buffer_alloc (vm, &bi, 1) != 1)
4129     {
4130       clib_warning ("Can't allocate buffer for Map-Register!");
4131       return 0;
4132     }
4133
4134   b = vlib_get_buffer (vm, bi);
4135
4136   /* leave some space for the encap headers */
4137   vlib_buffer_make_headroom (b, MAX_LISP_MSG_ENCAP_LEN);
4138
4139   lisp_msg_put_map_reply (b, records, nonce, probe_bit);
4140
4141   /* push outer ip header */
4142   pkt_push_udp_and_ip (vm, b, LISP_CONTROL_PORT, dst_port, sloc, dst, 1);
4143
4144   bi_res[0] = bi;
4145   return b;
4146 }
4147
4148 static int
4149 send_map_reply (lisp_cp_main_t * lcm, u32 mi, ip_address_t * dst,
4150                 u8 probe_bit, u64 nonce, u16 dst_port,
4151                 ip_address_t * probed_loc)
4152 {
4153   ip_address_t src;
4154   u32 bi;
4155   vlib_buffer_t *b;
4156   vlib_frame_t *f;
4157   u32 next_index, *to_next;
4158   mapping_t *records = 0, *m;
4159
4160   m = pool_elt_at_index (lcm->mapping_pool, mi);
4161   if (!m)
4162     return -1;
4163
4164   vec_add1 (records, m[0]);
4165   add_locators (lcm, &records[0], m->locator_set_index, probed_loc);
4166   clib_memset (&src, 0, sizeof (src));
4167
4168   if (!ip_fib_get_first_egress_ip_for_dst (lcm, dst, &src))
4169     {
4170       clib_warning ("can't find interface address for %U", format_ip_address,
4171                     dst);
4172       return -1;
4173     }
4174
4175   b = build_map_reply (lcm, &src, dst, nonce, probe_bit, records, dst_port,
4176                        &bi);
4177   if (!b)
4178     return -1;
4179   free_map_register_records (records);
4180
4181   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
4182   next_index = (ip_addr_version (&lcm->active_map_resolver) == IP4) ?
4183     ip4_lookup_node.index : ip6_lookup_node.index;
4184
4185   f = vlib_get_frame_to_node (lcm->vlib_main, next_index);
4186
4187   /* Enqueue the packet */
4188   to_next = vlib_frame_vector_args (f);
4189   to_next[0] = bi;
4190   f->n_vectors = 1;
4191   vlib_put_frame_to_node (lcm->vlib_main, next_index, f);
4192   return 0;
4193 }
4194
4195 static void
4196 find_ip_header (vlib_buffer_t * b, u8 ** ip_hdr)
4197 {
4198   const i32 start = vnet_buffer (b)->l3_hdr_offset;
4199   if (start < 0 && start < -sizeof (b->pre_data))
4200     {
4201       *ip_hdr = 0;
4202       return;
4203     }
4204
4205   *ip_hdr = b->data + start;
4206   if ((u8 *) * ip_hdr > (u8 *) vlib_buffer_get_current (b))
4207     *ip_hdr = 0;
4208 }
4209
4210 void
4211 process_map_request (vlib_main_t * vm, vlib_node_runtime_t * node,
4212                      lisp_cp_main_t * lcm, vlib_buffer_t * b)
4213 {
4214   u8 *ip_hdr = 0;
4215   ip_address_t *dst_loc = 0, probed_loc, src_loc;
4216   mapping_t m;
4217   map_request_hdr_t *mreq_hdr;
4218   gid_address_t src, dst;
4219   u64 nonce;
4220   u32 i, len = 0, rloc_probe_recv = 0;
4221   gid_address_t *itr_rlocs = 0;
4222
4223   mreq_hdr = vlib_buffer_get_current (b);
4224   if (!MREQ_SMR (mreq_hdr) && !MREQ_RLOC_PROBE (mreq_hdr))
4225     {
4226       clib_warning
4227         ("Only SMR Map-Requests and RLOC probe supported for now!");
4228       return;
4229     }
4230
4231   vlib_buffer_pull (b, sizeof (*mreq_hdr));
4232   nonce = MREQ_NONCE (mreq_hdr);
4233
4234   /* parse src eid */
4235   len = lisp_msg_parse_addr (b, &src);
4236   if (len == ~0)
4237     return;
4238
4239   len = lisp_msg_parse_itr_rlocs (b, &itr_rlocs,
4240                                   MREQ_ITR_RLOC_COUNT (mreq_hdr) + 1);
4241   if (len == ~0)
4242     goto done;
4243
4244   /* parse eid records and send SMR-invoked map-requests */
4245   for (i = 0; i < MREQ_REC_COUNT (mreq_hdr); i++)
4246     {
4247       clib_memset (&dst, 0, sizeof (dst));
4248       len = lisp_msg_parse_eid_rec (b, &dst);
4249       if (len == ~0)
4250         {
4251           clib_warning ("Can't parse map-request EID-record");
4252           goto done;
4253         }
4254
4255       if (MREQ_SMR (mreq_hdr))
4256         {
4257           /* send SMR-invoked map-requests */
4258           queue_map_request (&dst, &src, 1 /* invoked */ , 0 /* resend */ );
4259         }
4260       else if (MREQ_RLOC_PROBE (mreq_hdr))
4261         {
4262           find_ip_header (b, &ip_hdr);
4263           if (!ip_hdr)
4264             {
4265               clib_warning ("Cannot find the IP header!");
4266               goto done;
4267             }
4268           rloc_probe_recv++;
4269           clib_memset (&m, 0, sizeof (m));
4270           u32 mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &dst);
4271
4272           // TODO: select best locator; for now use the first one
4273           dst_loc = &gid_address_ip (&itr_rlocs[0]);
4274
4275           /* get src/dst IP addresses */
4276           get_src_and_dst_ip (ip_hdr, &src_loc, &probed_loc);
4277
4278           // TODO get source port from buffer
4279           u16 src_port = LISP_CONTROL_PORT;
4280
4281           send_map_reply (lcm, mi, dst_loc, 1 /* probe-bit */ , nonce,
4282                           src_port, &probed_loc);
4283         }
4284     }
4285
4286 done:
4287   vlib_node_increment_counter (vm, node->node_index,
4288                                LISP_CP_INPUT_ERROR_RLOC_PROBE_REQ_RECEIVED,
4289                                rloc_probe_recv);
4290   vec_free (itr_rlocs);
4291 }
4292
4293 map_records_arg_t *
4294 parse_map_reply (vlib_buffer_t * b)
4295 {
4296   locator_t probed;
4297   gid_address_t deid;
4298   void *h;
4299   u32 i, len = 0;
4300   mapping_t m;
4301   map_reply_hdr_t *mrep_hdr;
4302   map_records_arg_t *a;
4303
4304   a = map_record_args_get ();
4305   clib_memset (a, 0, sizeof (*a));
4306
4307   locator_t *locators;
4308
4309   mrep_hdr = vlib_buffer_get_current (b);
4310   a->nonce = MREP_NONCE (mrep_hdr);
4311   a->is_rloc_probe = MREP_RLOC_PROBE (mrep_hdr);
4312   if (!vlib_buffer_has_space (b, sizeof (*mrep_hdr)))
4313     {
4314       clib_mem_free (a);
4315       return 0;
4316     }
4317   vlib_buffer_pull (b, sizeof (*mrep_hdr));
4318
4319   for (i = 0; i < MREP_REC_COUNT (mrep_hdr); i++)
4320     {
4321       clib_memset (&m, 0, sizeof (m));
4322       locators = 0;
4323       h = vlib_buffer_get_current (b);
4324
4325       m.ttl = clib_net_to_host_u32 (MAP_REC_TTL (h));
4326       m.action = MAP_REC_ACTION (h);
4327       m.authoritative = MAP_REC_AUTH (h);
4328
4329       len = lisp_msg_parse_mapping_record (b, &deid, &locators, &probed);
4330       if (len == ~0)
4331         {
4332           clib_warning ("Failed to parse mapping record!");
4333           map_records_arg_free (a);
4334           return 0;
4335         }
4336
4337       m.locators = locators;
4338       gid_address_copy (&m.eid, &deid);
4339       vec_add1 (a->mappings, m);
4340     }
4341   return a;
4342 }
4343
4344 static void
4345 queue_map_reply_for_processing (map_records_arg_t * a)
4346 {
4347   vl_api_rpc_call_main_thread (process_map_reply, (u8 *) a, sizeof (*a));
4348 }
4349
4350 static void
4351 queue_map_notify_for_processing (map_records_arg_t * a)
4352 {
4353   vl_api_rpc_call_main_thread (process_map_notify, (u8 *) a, sizeof (a[0]));
4354 }
4355
4356 static uword
4357 lisp_cp_input (vlib_main_t * vm, vlib_node_runtime_t * node,
4358                vlib_frame_t * from_frame)
4359 {
4360   u32 n_left_from, *from, *to_next_drop, rloc_probe_rep_recv = 0,
4361     map_notifies_recv = 0;
4362   lisp_msg_type_e type;
4363   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4364   map_records_arg_t *a;
4365
4366   from = vlib_frame_vector_args (from_frame);
4367   n_left_from = from_frame->n_vectors;
4368
4369
4370   while (n_left_from > 0)
4371     {
4372       u32 n_left_to_next_drop;
4373
4374       vlib_get_next_frame (vm, node, LISP_CP_INPUT_NEXT_DROP,
4375                            to_next_drop, n_left_to_next_drop);
4376       while (n_left_from > 0 && n_left_to_next_drop > 0)
4377         {
4378           u32 bi0;
4379           vlib_buffer_t *b0;
4380
4381           bi0 = from[0];
4382           from += 1;
4383           n_left_from -= 1;
4384           to_next_drop[0] = bi0;
4385           to_next_drop += 1;
4386           n_left_to_next_drop -= 1;
4387
4388           b0 = vlib_get_buffer (vm, bi0);
4389
4390           type = lisp_msg_type (vlib_buffer_get_current (b0));
4391           switch (type)
4392             {
4393             case LISP_MAP_REPLY:
4394               a = parse_map_reply (b0);
4395               if (a)
4396                 {
4397                   if (a->is_rloc_probe)
4398                     rloc_probe_rep_recv++;
4399                   queue_map_reply_for_processing (a);
4400                 }
4401               break;
4402             case LISP_MAP_REQUEST:
4403               process_map_request (vm, node, lcm, b0);
4404               break;
4405             case LISP_MAP_NOTIFY:
4406               a = parse_map_notify (b0);
4407               if (a)
4408                 {
4409                   map_notifies_recv++;
4410                   queue_map_notify_for_processing (a);
4411                 }
4412               break;
4413             default:
4414               clib_warning ("Unsupported LISP message type %d", type);
4415               break;
4416             }
4417
4418           b0->error = node->errors[LISP_CP_INPUT_ERROR_DROP];
4419
4420           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
4421             {
4422
4423             }
4424         }
4425
4426       vlib_put_next_frame (vm, node, LISP_CP_INPUT_NEXT_DROP,
4427                            n_left_to_next_drop);
4428     }
4429   vlib_node_increment_counter (vm, node->node_index,
4430                                LISP_CP_INPUT_ERROR_RLOC_PROBE_REP_RECEIVED,
4431                                rloc_probe_rep_recv);
4432   vlib_node_increment_counter (vm, node->node_index,
4433                                LISP_CP_INPUT_ERROR_MAP_NOTIFIES_RECEIVED,
4434                                map_notifies_recv);
4435   return from_frame->n_vectors;
4436 }
4437
4438 /* *INDENT-OFF* */
4439 VLIB_REGISTER_NODE (lisp_cp_input_node) = {
4440   .function = lisp_cp_input,
4441   .name = "lisp-cp-input",
4442   .vector_size = sizeof (u32),
4443   .format_trace = format_lisp_cp_input_trace,
4444   .type = VLIB_NODE_TYPE_INTERNAL,
4445
4446   .n_errors = LISP_CP_INPUT_N_ERROR,
4447   .error_strings = lisp_cp_input_error_strings,
4448
4449   .n_next_nodes = LISP_CP_INPUT_N_NEXT,
4450
4451   .next_nodes = {
4452       [LISP_CP_INPUT_NEXT_DROP] = "error-drop",
4453   },
4454 };
4455 /* *INDENT-ON* */
4456
4457 clib_error_t *
4458 lisp_cp_init (vlib_main_t * vm)
4459 {
4460   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4461   clib_error_t *error = 0;
4462   vlib_thread_main_t *vtm = vlib_get_thread_main ();
4463   u32 num_threads;
4464
4465   if ((error = vlib_call_init_function (vm, lisp_gpe_init)))
4466     return error;
4467
4468   lcm->im4 = &ip4_main;
4469   lcm->im6 = &ip6_main;
4470   lcm->vlib_main = vm;
4471   lcm->vnet_main = vnet_get_main ();
4472   lcm->mreq_itr_rlocs = ~0;
4473   lcm->flags = 0;
4474   lcm->pitr_map_index = ~0;
4475   lcm->petr_map_index = ~0;
4476   clib_memset (&lcm->active_map_resolver, 0,
4477                sizeof (lcm->active_map_resolver));
4478   clib_memset (&lcm->active_map_server, 0, sizeof (lcm->active_map_server));
4479
4480   gid_dictionary_init (&lcm->mapping_index_by_gid);
4481   lcm->do_map_resolver_election = 1;
4482   lcm->do_map_server_election = 1;
4483   lcm->map_request_mode = MR_MODE_DST_ONLY;
4484
4485   num_threads = 1 /* main thread */  + vtm->n_threads;
4486   vec_validate (lcm->map_records_args_pool, num_threads - 1);
4487
4488   /* default vrf mapped to vni 0 */
4489   hash_set (lcm->table_id_by_vni, 0, 0);
4490   hash_set (lcm->vni_by_table_id, 0, 0);
4491
4492   u64 now = clib_cpu_time_now ();
4493   timing_wheel_init (&lcm->wheel, now, vm->clib_time.clocks_per_second);
4494   lcm->nsh_map_index = ~0;
4495   lcm->map_register_ttl = MAP_REGISTER_DEFAULT_TTL;
4496   lcm->max_expired_map_registers = MAX_EXPIRED_MAP_REGISTERS_DEFAULT;
4497   lcm->expired_map_registers = 0;
4498   lcm->transport_protocol = LISP_TRANSPORT_PROTOCOL_UDP;
4499   lcm->flags |= LISP_FLAG_XTR_MODE;
4500   return 0;
4501 }
4502
4503 static int
4504 lisp_stats_api_fill (lisp_cp_main_t * lcm, lisp_gpe_main_t * lgm,
4505                      lisp_api_stats_t * stat, lisp_stats_key_t * key,
4506                      u32 stats_index)
4507 {
4508   vlib_counter_t v;
4509   vlib_combined_counter_main_t *cm = &lgm->counters;
4510   lisp_gpe_fwd_entry_key_t fwd_key;
4511   const lisp_gpe_tunnel_t *lgt;
4512   fwd_entry_t *fe;
4513
4514   clib_memset (stat, 0, sizeof (*stat));
4515   clib_memset (&fwd_key, 0, sizeof (fwd_key));
4516
4517   fe = pool_elt_at_index (lcm->fwd_entry_pool, key->fwd_entry_index);
4518   ASSERT (fe != 0);
4519
4520   gid_to_dp_address (&fe->reid, &stat->deid);
4521   gid_to_dp_address (&fe->leid, &stat->seid);
4522   stat->vni = gid_address_vni (&fe->reid);
4523
4524   lgt = lisp_gpe_tunnel_get (key->tunnel_index);
4525   stat->loc_rloc = lgt->key->lcl;
4526   stat->rmt_rloc = lgt->key->rmt;
4527
4528   vlib_get_combined_counter (cm, stats_index, &v);
4529   stat->counters = v;
4530   return 1;
4531 }
4532
4533 lisp_api_stats_t *
4534 vnet_lisp_get_stats (void)
4535 {
4536   lisp_gpe_main_t *lgm = vnet_lisp_gpe_get_main ();
4537   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4538   lisp_api_stats_t *stats = 0, stat;
4539   lisp_stats_key_t *key;
4540   u32 index;
4541
4542   /* *INDENT-OFF* */
4543   hash_foreach_mem (key, index, lgm->lisp_stats_index_by_key,
4544   {
4545     if (lisp_stats_api_fill (lcm, lgm, &stat, key, index))
4546       vec_add1 (stats, stat);
4547   });
4548   /* *INDENT-ON* */
4549
4550   return stats;
4551 }
4552
4553 static void *
4554 send_map_request_thread_fn (void *arg)
4555 {
4556   map_request_args_t *a = arg;
4557   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4558
4559   if (a->is_resend)
4560     resend_encapsulated_map_request (lcm, &a->seid, &a->deid, a->smr_invoked);
4561   else
4562     send_encapsulated_map_request (lcm, &a->seid, &a->deid, a->smr_invoked);
4563
4564   return 0;
4565 }
4566
4567 static int
4568 queue_map_request (gid_address_t * seid, gid_address_t * deid,
4569                    u8 smr_invoked, u8 is_resend)
4570 {
4571   map_request_args_t a;
4572
4573   a.is_resend = is_resend;
4574   gid_address_copy (&a.seid, seid);
4575   gid_address_copy (&a.deid, deid);
4576   a.smr_invoked = smr_invoked;
4577
4578   vl_api_rpc_call_main_thread (send_map_request_thread_fn,
4579                                (u8 *) & a, sizeof (a));
4580   return 0;
4581 }
4582
4583 /**
4584  * Take an action with a pending map request depending on expiration time
4585  * and re-try counters.
4586  */
4587 static void
4588 update_pending_request (pending_map_request_t * r, f64 dt)
4589 {
4590   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4591   lisp_msmr_t *mr;
4592
4593   if (r->time_to_expire - dt < 0)
4594     /* it's time to decide what to do with this pending request */
4595     {
4596       if (r->retries_num >= NUMBER_OF_RETRIES)
4597         /* too many retries -> assume current map resolver is not available */
4598         {
4599           mr = get_map_resolver (&lcm->active_map_resolver);
4600           if (!mr)
4601             {
4602               clib_warning ("Map resolver %U not found - probably deleted "
4603                             "by the user recently.", format_ip_address,
4604                             &lcm->active_map_resolver);
4605             }
4606           else
4607             {
4608               clib_warning ("map resolver %U is unreachable, ignoring",
4609                             format_ip_address, &lcm->active_map_resolver);
4610
4611               /* mark current map resolver unavailable so it won't be
4612                * selected next time */
4613               mr->is_down = 1;
4614               mr->last_update = vlib_time_now (lcm->vlib_main);
4615             }
4616
4617           reset_pending_mr_counters (r);
4618           elect_map_resolver (lcm);
4619
4620           /* try to find a next eligible map resolver and re-send */
4621           queue_map_request (&r->src, &r->dst, r->is_smr_invoked,
4622                              1 /* resend */ );
4623         }
4624       else
4625         {
4626           /* try again */
4627           queue_map_request (&r->src, &r->dst, r->is_smr_invoked,
4628                              1 /* resend */ );
4629           r->retries_num++;
4630           r->time_to_expire = PENDING_MREQ_EXPIRATION_TIME;
4631         }
4632     }
4633   else
4634     r->time_to_expire -= dt;
4635 }
4636
4637 static void
4638 remove_dead_pending_map_requests (lisp_cp_main_t * lcm)
4639 {
4640   u64 *nonce;
4641   pending_map_request_t *pmr;
4642   u32 *to_be_removed = 0, *pmr_index;
4643
4644   /* *INDENT-OFF* */
4645   pool_foreach (pmr, lcm->pending_map_requests_pool,
4646   ({
4647     if (pmr->to_be_removed)
4648       {
4649         clib_fifo_foreach (nonce, pmr->nonces, ({
4650           hash_unset (lcm->pending_map_requests_by_nonce, nonce[0]);
4651         }));
4652
4653         vec_add1 (to_be_removed, pmr - lcm->pending_map_requests_pool);
4654       }
4655   }));
4656   /* *INDENT-ON* */
4657
4658   vec_foreach (pmr_index, to_be_removed)
4659     pool_put_index (lcm->pending_map_requests_pool, pmr_index[0]);
4660
4661   vec_free (to_be_removed);
4662 }
4663
4664 static void
4665 update_rloc_probing (lisp_cp_main_t * lcm, f64 dt)
4666 {
4667   static f64 time_left = RLOC_PROBING_INTERVAL;
4668
4669   if (!lcm->is_enabled || !lcm->rloc_probing)
4670     return;
4671
4672   time_left -= dt;
4673   if (time_left <= 0)
4674     {
4675       time_left = RLOC_PROBING_INTERVAL;
4676       send_rloc_probes (lcm);
4677     }
4678 }
4679
4680 static int
4681 update_pending_map_register (pending_map_register_t * r, f64 dt, u8 * del_all)
4682 {
4683   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4684   lisp_msmr_t *ms;
4685   del_all[0] = 0;
4686
4687   r->time_to_expire -= dt;
4688
4689   if (r->time_to_expire < 0)
4690     {
4691       lcm->expired_map_registers++;
4692
4693       if (lcm->expired_map_registers >= lcm->max_expired_map_registers)
4694         {
4695           ms = get_map_server (&lcm->active_map_server);
4696           if (!ms)
4697             {
4698               clib_warning ("Map server %U not found - probably deleted "
4699                             "by the user recently.", format_ip_address,
4700                             &lcm->active_map_server);
4701             }
4702           else
4703             {
4704               clib_warning ("map server %U is unreachable, ignoring",
4705                             format_ip_address, &lcm->active_map_server);
4706
4707               /* mark current map server unavailable so it won't be
4708                * elected next time */
4709               ms->is_down = 1;
4710               ms->last_update = vlib_time_now (lcm->vlib_main);
4711             }
4712
4713           elect_map_server (lcm);
4714
4715           /* indication for deleting all pending map registers */
4716           del_all[0] = 1;
4717           lcm->expired_map_registers = 0;
4718           return 0;
4719         }
4720       else
4721         {
4722           /* delete pending map register */
4723           return 0;
4724         }
4725     }
4726   return 1;
4727 }
4728
4729 static void
4730 update_map_register (lisp_cp_main_t * lcm, f64 dt)
4731 {
4732   u32 *to_be_removed = 0, *pmr_index;
4733   static f64 time_left = QUICK_MAP_REGISTER_INTERVAL;
4734   static u64 mreg_sent_counter = 0;
4735
4736   pending_map_register_t *pmr;
4737   u8 del_all = 0;
4738
4739   if (!lcm->is_enabled || !lcm->map_registering)
4740     return;
4741
4742   /* *INDENT-OFF* */
4743   pool_foreach (pmr, lcm->pending_map_registers_pool,
4744   ({
4745     if (!update_pending_map_register (pmr, dt, &del_all))
4746     {
4747       if (del_all)
4748         break;
4749       vec_add1 (to_be_removed, pmr - lcm->pending_map_registers_pool);
4750     }
4751   }));
4752   /* *INDENT-ON* */
4753
4754   if (del_all)
4755     {
4756       /* delete all pending map register messages so they won't
4757        * trigger another map server election.. */
4758       pool_free (lcm->pending_map_registers_pool);
4759       hash_free (lcm->map_register_messages_by_nonce);
4760
4761       /* ..and trigger registration against next map server (if any) */
4762       time_left = 0;
4763     }
4764   else
4765     {
4766       vec_foreach (pmr_index, to_be_removed)
4767         pool_put_index (lcm->pending_map_registers_pool, pmr_index[0]);
4768     }
4769
4770   vec_free (to_be_removed);
4771
4772   time_left -= dt;
4773   if (time_left <= 0)
4774     {
4775       if (mreg_sent_counter >= QUICK_MAP_REGISTER_MSG_COUNT)
4776         time_left = MAP_REGISTER_INTERVAL;
4777       else
4778         {
4779           mreg_sent_counter++;
4780           time_left = QUICK_MAP_REGISTER_INTERVAL;
4781         }
4782       send_map_register (lcm, 1 /* want map notify */ );
4783     }
4784 }
4785
4786 static uword
4787 send_map_resolver_service (vlib_main_t * vm,
4788                            vlib_node_runtime_t * rt, vlib_frame_t * f)
4789 {
4790   u32 *expired = 0;
4791   f64 period = 2.0;
4792   pending_map_request_t *pmr;
4793   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4794
4795   while (1)
4796     {
4797       vlib_process_wait_for_event_or_clock (vm, period);
4798
4799       /* currently no signals are expected - just wait for clock */
4800       (void) vlib_process_get_events (vm, 0);
4801
4802       /* *INDENT-OFF* */
4803       pool_foreach (pmr, lcm->pending_map_requests_pool,
4804       ({
4805         if (!pmr->to_be_removed)
4806           update_pending_request (pmr, period);
4807       }));
4808       /* *INDENT-ON* */
4809
4810       remove_dead_pending_map_requests (lcm);
4811
4812       update_map_register (lcm, period);
4813       update_rloc_probing (lcm, period);
4814
4815       u64 now = clib_cpu_time_now ();
4816
4817       expired = timing_wheel_advance (&lcm->wheel, now, expired, 0);
4818       if (vec_len (expired) > 0)
4819         {
4820           u32 *mi = 0;
4821           vec_foreach (mi, expired)
4822           {
4823             process_expired_mapping (lcm, mi[0]);
4824           }
4825           _vec_len (expired) = 0;
4826         }
4827     }
4828
4829   /* unreachable */
4830   return 0;
4831 }
4832
4833 vnet_api_error_t
4834 vnet_lisp_stats_enable_disable (u8 enable)
4835 {
4836   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4837
4838   if (vnet_lisp_enable_disable_status () == 0)
4839     return VNET_API_ERROR_LISP_DISABLED;
4840
4841   if (enable)
4842     lcm->flags |= LISP_FLAG_STATS_ENABLED;
4843   else
4844     lcm->flags &= ~LISP_FLAG_STATS_ENABLED;
4845
4846   return 0;
4847 }
4848
4849 u8
4850 vnet_lisp_stats_enable_disable_state (void)
4851 {
4852   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4853
4854   if (vnet_lisp_enable_disable_status () == 0)
4855     return VNET_API_ERROR_LISP_DISABLED;
4856
4857   return lcm->flags & LISP_FLAG_STATS_ENABLED;
4858 }
4859
4860 /* *INDENT-OFF* */
4861 VLIB_REGISTER_NODE (lisp_retry_service_node,static) = {
4862     .function = send_map_resolver_service,
4863     .type = VLIB_NODE_TYPE_PROCESS,
4864     .name = "lisp-retry-service",
4865     .process_log2_n_stack_bytes = 16,
4866 };
4867 /* *INDENT-ON* */
4868
4869 u32
4870 vnet_lisp_set_transport_protocol (u8 protocol)
4871 {
4872   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4873
4874   if (protocol < LISP_TRANSPORT_PROTOCOL_UDP ||
4875       protocol > LISP_TRANSPORT_PROTOCOL_API)
4876     return VNET_API_ERROR_INVALID_ARGUMENT;
4877
4878   lcm->transport_protocol = protocol;
4879   return 0;
4880 }
4881
4882 lisp_transport_protocol_t
4883 vnet_lisp_get_transport_protocol (void)
4884 {
4885   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4886   return lcm->transport_protocol;
4887 }
4888
4889 int
4890 vnet_lisp_enable_disable_xtr_mode (u8 is_enabled)
4891 {
4892   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4893   u8 pitr_mode = lcm->flags & LISP_FLAG_PITR_MODE;
4894   u8 xtr_mode = lcm->flags & LISP_FLAG_XTR_MODE;
4895   u8 petr_mode = lcm->flags & LISP_FLAG_PETR_MODE;
4896
4897   if (pitr_mode && is_enabled)
4898     return VNET_API_ERROR_INVALID_ARGUMENT;
4899
4900   if (is_enabled && xtr_mode)
4901     return 0;
4902   if (!is_enabled && !xtr_mode)
4903     return 0;
4904
4905   if (is_enabled)
4906     {
4907       if (!petr_mode)
4908         {
4909           lisp_cp_register_dst_port (lcm->vlib_main);
4910         }
4911       lisp_cp_enable_l2_l3_ifaces (lcm, 1 /* with_default_route */ );
4912       lcm->flags |= LISP_FLAG_XTR_MODE;
4913     }
4914   else
4915     {
4916       if (!petr_mode)
4917         {
4918           lisp_cp_unregister_dst_port (lcm->vlib_main);
4919         }
4920       lisp_cp_disable_l2_l3_ifaces (lcm);
4921       lcm->flags &= ~LISP_FLAG_XTR_MODE;
4922     }
4923   return 0;
4924 }
4925
4926 int
4927 vnet_lisp_enable_disable_pitr_mode (u8 is_enabled)
4928 {
4929   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4930   u8 xtr_mode = lcm->flags & LISP_FLAG_XTR_MODE;
4931   u8 pitr_mode = lcm->flags & LISP_FLAG_PITR_MODE;
4932
4933   if (xtr_mode && is_enabled)
4934     return VNET_API_ERROR_INVALID_VALUE;
4935
4936   if (is_enabled && pitr_mode)
4937     return 0;
4938   if (!is_enabled && !pitr_mode)
4939     return 0;
4940
4941   if (is_enabled)
4942     {
4943       /* create iface, no default route */
4944       lisp_cp_enable_l2_l3_ifaces (lcm, 0 /* with_default_route */ );
4945       lcm->flags |= LISP_FLAG_PITR_MODE;
4946     }
4947   else
4948     {
4949       lisp_cp_disable_l2_l3_ifaces (lcm);
4950       lcm->flags &= ~LISP_FLAG_PITR_MODE;
4951     }
4952   return 0;
4953 }
4954
4955 int
4956 vnet_lisp_enable_disable_petr_mode (u8 is_enabled)
4957 {
4958   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4959   u8 xtr_mode = lcm->flags & LISP_FLAG_XTR_MODE;
4960   u8 petr_mode = lcm->flags & LISP_FLAG_PETR_MODE;
4961
4962   if (is_enabled && petr_mode)
4963     return 0;
4964   if (!is_enabled && !petr_mode)
4965     return 0;
4966
4967   if (is_enabled)
4968     {
4969       if (!xtr_mode)
4970         {
4971           lisp_cp_register_dst_port (lcm->vlib_main);
4972         }
4973       lcm->flags |= LISP_FLAG_PETR_MODE;
4974     }
4975   else
4976     {
4977       if (!xtr_mode)
4978         {
4979           lisp_cp_unregister_dst_port (lcm->vlib_main);
4980         }
4981       lcm->flags &= ~LISP_FLAG_PETR_MODE;
4982     }
4983   return 0;
4984 }
4985
4986 u8
4987 vnet_lisp_get_xtr_mode (void)
4988 {
4989   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4990   return (lcm->flags & LISP_FLAG_XTR_MODE);
4991 }
4992
4993 u8
4994 vnet_lisp_get_pitr_mode (void)
4995 {
4996   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
4997   return (lcm->flags & LISP_FLAG_PITR_MODE);
4998 }
4999
5000 u8
5001 vnet_lisp_get_petr_mode (void)
5002 {
5003   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
5004   return (lcm->flags & LISP_FLAG_PETR_MODE);
5005 }
5006
5007 VLIB_INIT_FUNCTION (lisp_cp_init);
5008
5009 /*
5010  * fd.io coding-style-patch-verification: ON
5011  *
5012  * Local Variables:
5013  * eval: (c-set-style "gnu")
5014  * End:
5015  */