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