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