VPP-329: Add L2 support to lisp_eid_table_map_dump
[vpp.git] / vnet / 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.h>
21
22 typedef struct
23 {
24   u8 is_resend;
25   gid_address_t seid;
26   gid_address_t deid;
27   u8 smr_invoked;
28 } map_request_args_t;
29
30 static int
31 queue_map_request (gid_address_t * seid, gid_address_t * deid,
32                    u8 smr_invoked, u8 is_resend);
33
34 ip_interface_address_t *
35 ip_interface_get_first_interface_address (ip_lookup_main_t * lm,
36                                           u32 sw_if_index, u8 loop)
37 {
38   vnet_main_t *vnm = vnet_get_main ();
39   vnet_sw_interface_t *swif = vnet_get_sw_interface (vnm, sw_if_index);
40   if (loop && swif->flags & VNET_SW_INTERFACE_FLAG_UNNUMBERED)
41     sw_if_index = swif->unnumbered_sw_if_index;
42   u32 ia =
43     (vec_len ((lm)->if_address_pool_index_by_sw_if_index) > (sw_if_index)) ?
44     vec_elt ((lm)->if_address_pool_index_by_sw_if_index, (sw_if_index)) :
45     (u32) ~ 0;
46   return pool_elt_at_index ((lm)->if_address_pool, ia);
47 }
48
49 void *
50 ip_interface_get_first_address (ip_lookup_main_t * lm, u32 sw_if_index,
51                                 u8 version)
52 {
53   ip_interface_address_t *ia;
54
55   ia = ip_interface_get_first_interface_address (lm, sw_if_index, 1);
56   if (!ia)
57     return 0;
58   return ip_interface_address_get_address (lm, ia);
59 }
60
61 int
62 ip_interface_get_first_ip_address (lisp_cp_main_t * lcm, u32 sw_if_index,
63                                    u8 version, ip_address_t * result)
64 {
65   ip_lookup_main_t *lm;
66   void *addr;
67
68   lm = (version == IP4) ? &lcm->im4->lookup_main : &lcm->im6->lookup_main;
69   addr = ip_interface_get_first_address (lm, sw_if_index, version);
70   if (!addr)
71     return 0;
72
73   ip_address_set (result, addr, version);
74   return 1;
75 }
76
77 static u32
78 ip_fib_lookup_with_table (lisp_cp_main_t * lcm, u32 fib_index,
79                           ip_address_t * dst)
80 {
81   if (ip_addr_version (dst) == IP4)
82     return ip4_fib_lookup_with_table (lcm->im4, fib_index, &ip_addr_v4 (dst),
83                                       0);
84   else
85     return ip6_fib_lookup_with_table (lcm->im6, fib_index, &ip_addr_v6 (dst));
86 }
87
88 u32
89 ip_fib_get_egress_iface_for_dst_with_lm (lisp_cp_main_t * lcm,
90                                          ip_address_t * dst,
91                                          ip_lookup_main_t * lm)
92 {
93   u32 adj_index;
94   ip_adjacency_t *adj;
95
96   adj_index = ip_fib_lookup_with_table (lcm, 0, dst);
97   adj = ip_get_adjacency (lm, adj_index);
98
99   if (adj == 0)
100     return ~0;
101
102   /* we only want outgoing routes */
103   if (adj->lookup_next_index != IP_LOOKUP_NEXT_ARP
104       && adj->lookup_next_index != IP_LOOKUP_NEXT_REWRITE)
105     return ~0;
106
107   return adj->rewrite_header.sw_if_index;
108 }
109
110 /**
111  * Find the sw_if_index of the interface that would be used to egress towards
112  * dst.
113  */
114 u32
115 ip_fib_get_egress_iface_for_dst (lisp_cp_main_t * lcm, ip_address_t * dst)
116 {
117   ip_lookup_main_t *lm;
118
119   lm = ip_addr_version (dst) == IP4 ?
120     &lcm->im4->lookup_main : &lcm->im6->lookup_main;
121
122   return ip_fib_get_egress_iface_for_dst_with_lm (lcm, dst, lm);
123 }
124
125 /**
126  * Find first IP of the interface that would be used to egress towards dst.
127  * Returns 1 if the address is found 0 otherwise.
128  */
129 int
130 ip_fib_get_first_egress_ip_for_dst (lisp_cp_main_t * lcm, ip_address_t * dst,
131                                     ip_address_t * result)
132 {
133   u32 si;
134   ip_lookup_main_t *lm;
135   void *addr = 0;
136   u8 ipver;
137
138   ASSERT (result != 0);
139
140   ipver = ip_addr_version (dst);
141
142   lm = (ipver == IP4) ? &lcm->im4->lookup_main : &lcm->im6->lookup_main;
143   si = ip_fib_get_egress_iface_for_dst_with_lm (lcm, dst, lm);
144
145   if ((u32) ~ 0 == si)
146     return 0;
147
148   /* find the first ip address */
149   addr = ip_interface_get_first_address (lm, si, ipver);
150   if (0 == addr)
151     return 0;
152
153   ip_address_set (result, addr, ipver);
154   return 1;
155 }
156
157 static int
158 dp_add_del_iface (lisp_cp_main_t * lcm, u32 vni, u8 is_l2, u8 is_add)
159 {
160   uword *dp_table, *intf;
161   vnet_lisp_gpe_add_del_iface_args_t _ai, *ai = &_ai;
162
163   if (!is_l2)
164     {
165       dp_table = hash_get (lcm->table_id_by_vni, vni);
166
167       if (!dp_table)
168         {
169           clib_warning ("vni %d not associated to a vrf!", vni);
170           return VNET_API_ERROR_INVALID_VALUE;
171         }
172     }
173   else
174     {
175       dp_table = hash_get (lcm->bd_id_by_vni, vni);
176       if (!dp_table)
177         {
178           clib_warning ("vni %d not associated to a bridge domain!", vni);
179           return VNET_API_ERROR_INVALID_VALUE;
180         }
181     }
182
183   intf = hash_get (is_l2 ? lcm->l2_dp_intf_by_vni : lcm->dp_intf_by_vni, vni);
184
185   /* enable/disable data-plane interface */
186   if (is_add)
187     {
188       /* create interface */
189       if (!intf)
190         {
191           ai->is_add = 1;
192           ai->vni = vni;
193           ai->is_l2 = is_l2;
194           ai->dp_table = dp_table[0];
195
196           vnet_lisp_gpe_add_del_iface (ai, 0);
197
198           /* keep track of vnis for which interfaces have been created */
199           hash_set (lcm->dp_intf_by_vni, vni, 1);
200         }
201     }
202   else
203     {
204       if (intf == 0)
205         {
206           clib_warning ("interface for vni %d doesn't exist!", vni);
207           return VNET_API_ERROR_INVALID_VALUE;
208         }
209
210       ai->is_add = 0;
211       ai->vni = vni;
212       ai->dp_table = dp_table[0];
213       vnet_lisp_gpe_add_del_iface (ai, 0);
214       hash_unset (lcm->dp_intf_by_vni, vni);
215     }
216
217   return 0;
218 }
219
220 static void
221 dp_del_fwd_entry (lisp_cp_main_t * lcm, u32 src_map_index, u32 dst_map_index)
222 {
223   vnet_lisp_gpe_add_del_fwd_entry_args_t _a, *a = &_a;
224   fwd_entry_t *fe = 0;
225   uword *feip = 0;
226   memset (a, 0, sizeof (*a));
227
228   feip = hash_get (lcm->fwd_entry_by_mapping_index, dst_map_index);
229   if (!feip)
230     return;
231
232   fe = pool_elt_at_index (lcm->fwd_entry_pool, feip[0]);
233
234   /* delete dp fwd entry */
235   u32 sw_if_index;
236   a->is_add = 0;
237   a->locator_pairs = fe->locator_pairs;
238   a->vni = gid_address_vni (&a->rmt_eid);
239   gid_address_copy (&a->rmt_eid, &fe->deid);
240
241   vnet_lisp_gpe_add_del_fwd_entry (a, &sw_if_index);
242
243   /* delete entry in fwd table */
244   hash_unset (lcm->fwd_entry_by_mapping_index, dst_map_index);
245   vec_free (fe->locator_pairs);
246   pool_put (lcm->fwd_entry_pool, fe);
247 }
248
249 /**
250  * Finds first remote locator with best (lowest) priority that has a local
251  * peer locator with an underlying route to it.
252  *
253  */
254 static u32
255 get_locator_pairs (lisp_cp_main_t * lcm, mapping_t * lcl_map,
256                    mapping_t * rmt_map, locator_pair_t ** locator_pairs)
257 {
258   u32 i, limitp = 0, li, found = 0, esi;
259   locator_set_t *rmt_ls, *lcl_ls;
260   ip_address_t _lcl_addr, *lcl_addr = &_lcl_addr;
261   locator_t *lp, *rmt = 0;
262   uword *checked = 0;
263   locator_pair_t pair;
264
265   rmt_ls =
266     pool_elt_at_index (lcm->locator_set_pool, rmt_map->locator_set_index);
267   lcl_ls =
268     pool_elt_at_index (lcm->locator_set_pool, lcl_map->locator_set_index);
269
270   if (!rmt_ls || vec_len (rmt_ls->locator_indices) == 0)
271     return 0;
272
273   while (1)
274     {
275       rmt = 0;
276
277       /* find unvisited remote locator with best priority */
278       for (i = 0; i < vec_len (rmt_ls->locator_indices); i++)
279         {
280           if (0 != hash_get (checked, i))
281             continue;
282
283           li = vec_elt (rmt_ls->locator_indices, i);
284           lp = pool_elt_at_index (lcm->locator_pool, li);
285
286           /* we don't support non-IP locators for now */
287           if (gid_address_type (&lp->address) != GID_ADDR_IP_PREFIX)
288             continue;
289
290           if ((found && lp->priority == limitp)
291               || (!found && lp->priority >= limitp))
292             {
293               rmt = lp;
294
295               /* don't search for locators with lower priority and don't
296                * check this locator again*/
297               limitp = lp->priority;
298               hash_set (checked, i, 1);
299               break;
300             }
301         }
302       /* check if a local locator with a route to remote locator exists */
303       if (rmt != 0)
304         {
305           /* find egress sw_if_index for rmt locator */
306           esi =
307             ip_fib_get_egress_iface_for_dst (lcm,
308                                              &gid_address_ip (&rmt->address));
309           if ((u32) ~ 0 == esi)
310             continue;
311
312           for (i = 0; i < vec_len (lcl_ls->locator_indices); i++)
313             {
314               li = vec_elt (lcl_ls->locator_indices, i);
315               locator_t *sl = pool_elt_at_index (lcm->locator_pool, li);
316
317               /* found local locator with the needed sw_if_index */
318               if (sl->sw_if_index == esi)
319                 {
320                   /* and it has an address */
321                   if (0 == ip_interface_get_first_ip_address (lcm,
322                                                               sl->sw_if_index,
323                                                               gid_address_ip_version
324                                                               (&rmt->address),
325                                                               lcl_addr))
326                     continue;
327
328                   memset (&pair, 0, sizeof (pair));
329                   ip_address_copy (&pair.rmt_loc,
330                                    &gid_address_ip (&rmt->address));
331                   ip_address_copy (&pair.lcl_loc, lcl_addr);
332                   pair.weight = rmt->weight;
333                   vec_add1 (locator_pairs[0], pair);
334                   found = 1;
335                 }
336             }
337         }
338       else
339         break;
340     }
341
342   hash_free (checked);
343   return found;
344 }
345
346 static void
347 dp_add_fwd_entry (lisp_cp_main_t * lcm, u32 src_map_index, u32 dst_map_index)
348 {
349   vnet_lisp_gpe_add_del_fwd_entry_args_t _a, *a = &_a;
350   mapping_t *src_map, *dst_map;
351   u32 sw_if_index;
352   uword *feip = 0, *dpid;
353   fwd_entry_t *fe;
354   u8 type;
355
356   memset (a, 0, sizeof (*a));
357
358   /* remove entry if it already exists */
359   feip = hash_get (lcm->fwd_entry_by_mapping_index, dst_map_index);
360   if (feip)
361     dp_del_fwd_entry (lcm, src_map_index, dst_map_index);
362
363   src_map = pool_elt_at_index (lcm->mapping_pool, src_map_index);
364   dst_map = pool_elt_at_index (lcm->mapping_pool, dst_map_index);
365
366   /* insert data plane forwarding entry */
367   a->is_add = 1;
368
369   gid_address_copy (&a->rmt_eid, &dst_map->eid);
370   a->vni = gid_address_vni (&a->rmt_eid);
371
372   /* get vrf or bd_index associated to vni */
373   type = gid_address_type (&dst_map->eid);
374   if (GID_ADDR_IP_PREFIX == type)
375     {
376       dpid = hash_get (lcm->table_id_by_vni, a->vni);
377       if (!dpid)
378         {
379           clib_warning ("vni %d not associated to a vrf!", a->vni);
380           return;
381         }
382       a->table_id = dpid[0];
383     }
384   else if (GID_ADDR_MAC == type)
385     {
386       dpid = hash_get (lcm->bd_id_by_vni, a->vni);
387       if (!dpid)
388         {
389           clib_warning ("vni %d not associated to a bridge domain !", a->vni);
390           return;
391         }
392       a->bd_id = dpid[0];
393     }
394
395   /* find best locator pair that 1) verifies LISP policy 2) are connected */
396   if (0 == get_locator_pairs (lcm, src_map, dst_map, &a->locator_pairs))
397     {
398       /* negative entry */
399       a->is_negative = 1;
400       a->action = dst_map->action;
401     }
402
403   /* TODO remove */
404   u8 ipver = ip_prefix_version (&gid_address_ippref (&a->rmt_eid));
405   a->decap_next_index = (ipver == IP4) ?
406     LISP_GPE_INPUT_NEXT_IP4_INPUT : LISP_GPE_INPUT_NEXT_IP6_INPUT;
407
408   vnet_lisp_gpe_add_del_fwd_entry (a, &sw_if_index);
409
410   /* add tunnel to fwd entry table XXX check return value from DP insertion */
411   pool_get (lcm->fwd_entry_pool, fe);
412   fe->locator_pairs = a->locator_pairs;
413   gid_address_copy (&fe->deid, &a->rmt_eid);
414   hash_set (lcm->fwd_entry_by_mapping_index, dst_map_index,
415             fe - lcm->fwd_entry_pool);
416 }
417
418 /**
419  * Add/remove mapping to/from map-cache. Overwriting not allowed.
420  */
421 int
422 vnet_lisp_map_cache_add_del (vnet_lisp_add_del_mapping_args_t * a,
423                              u32 * map_index_result)
424 {
425   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
426   u32 mi, *map_indexp, map_index, i;
427   mapping_t *m, *old_map;
428   u32 **eid_indexes;
429
430   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &a->eid);
431   old_map = mi != ~0 ? pool_elt_at_index (lcm->mapping_pool, mi) : 0;
432   if (a->is_add)
433     {
434       /* TODO check if overwriting and take appropriate actions */
435       if (mi != GID_LOOKUP_MISS && !gid_address_cmp (&old_map->eid, &a->eid))
436         {
437           clib_warning ("eid %U found in the eid-table", format_gid_address,
438                         &a->eid);
439           return VNET_API_ERROR_VALUE_EXIST;
440         }
441
442       pool_get (lcm->mapping_pool, m);
443       gid_address_copy (&m->eid, &a->eid);
444       m->locator_set_index = a->locator_set_index;
445       m->ttl = a->ttl;
446       m->action = a->action;
447       m->local = a->local;
448       m->is_static = a->is_static;
449
450       map_index = m - lcm->mapping_pool;
451       gid_dictionary_add_del (&lcm->mapping_index_by_gid, &a->eid, map_index,
452                               1);
453
454       if (pool_is_free_index (lcm->locator_set_pool, a->locator_set_index))
455         {
456           clib_warning ("Locator set with index %d doesn't exist",
457                         a->locator_set_index);
458           return VNET_API_ERROR_INVALID_VALUE;
459         }
460
461       /* add eid to list of eids supported by locator-set */
462       vec_validate (lcm->locator_set_to_eids, a->locator_set_index);
463       eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids,
464                                       a->locator_set_index);
465       vec_add1 (eid_indexes[0], map_index);
466
467       if (a->local)
468         {
469           /* mark as local */
470           vec_add1 (lcm->local_mappings_indexes, map_index);
471         }
472       map_index_result[0] = map_index;
473     }
474   else
475     {
476       if (mi == GID_LOOKUP_MISS)
477         {
478           clib_warning ("eid %U not found in the eid-table",
479                         format_gid_address, &a->eid);
480           return VNET_API_ERROR_INVALID_VALUE;
481         }
482
483       /* clear locator-set to eids binding */
484       eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids,
485                                       a->locator_set_index);
486       for (i = 0; i < vec_len (eid_indexes[0]); i++)
487         {
488           map_indexp = vec_elt_at_index (eid_indexes[0], i);
489           if (map_indexp[0] == mi)
490             break;
491         }
492       vec_del1 (eid_indexes[0], i);
493
494       /* remove local mark if needed */
495       m = pool_elt_at_index (lcm->mapping_pool, mi);
496       if (m->local)
497         {
498           u32 k, *lm_indexp;
499           for (k = 0; k < vec_len (lcm->local_mappings_indexes); k++)
500             {
501               lm_indexp = vec_elt_at_index (lcm->local_mappings_indexes, k);
502               if (lm_indexp[0] == mi)
503                 break;
504             }
505           vec_del1 (lcm->local_mappings_indexes, k);
506         }
507
508       /* remove mapping from dictionary */
509       gid_dictionary_add_del (&lcm->mapping_index_by_gid, &a->eid, 0, 0);
510       gid_address_free (&m->eid);
511       pool_put_index (lcm->mapping_pool, mi);
512     }
513
514   return 0;
515 }
516
517 /**
518  *  Add/update/delete mapping to/in/from map-cache.
519  */
520 int
521 vnet_lisp_add_del_local_mapping (vnet_lisp_add_del_mapping_args_t * a,
522                                  u32 * map_index_result)
523 {
524   uword *dp_table = 0;
525   u32 vni;
526   u8 type;
527
528   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
529
530   if (vnet_lisp_enable_disable_status () == 0)
531     {
532       clib_warning ("LISP is disabled!");
533       return VNET_API_ERROR_LISP_DISABLED;
534     }
535
536   vni = gid_address_vni (&a->eid);
537   type = gid_address_type (&a->eid);
538   if (GID_ADDR_IP_PREFIX == type)
539     dp_table = hash_get (lcm->table_id_by_vni, vni);
540   else if (GID_ADDR_MAC == type)
541     dp_table = hash_get (lcm->bd_id_by_vni, vni);
542
543   if (!dp_table)
544     {
545       clib_warning ("vni %d not associated to a %s!", vni,
546                     GID_ADDR_IP_PREFIX == type ? "vrf" : "bd");
547       return VNET_API_ERROR_INVALID_VALUE;
548     }
549
550   /* store/remove mapping from map-cache */
551   return vnet_lisp_map_cache_add_del (a, map_index_result);
552 }
553
554 static clib_error_t *
555 lisp_add_del_local_eid_command_fn (vlib_main_t * vm, unformat_input_t * input,
556                                    vlib_cli_command_t * cmd)
557 {
558   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
559   unformat_input_t _line_input, *line_input = &_line_input;
560   u8 is_add = 1;
561   gid_address_t eid;
562   gid_address_t *eids = 0;
563   clib_error_t *error = 0;
564   u8 *locator_set_name = 0;
565   u32 locator_set_index = 0, map_index = 0;
566   uword *p;
567   vnet_lisp_add_del_mapping_args_t _a, *a = &_a;
568   int rv = 0;
569   u32 vni = 0;
570
571   memset (&eid, 0, sizeof (eid));
572   memset (a, 0, sizeof (*a));
573
574   /* Get a line of input. */
575   if (!unformat_user (input, unformat_line_input, line_input))
576     return 0;
577
578   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
579     {
580       if (unformat (line_input, "add"))
581         is_add = 1;
582       else if (unformat (line_input, "del"))
583         is_add = 0;
584       else if (unformat (line_input, "eid %U", unformat_gid_address, &eid))
585         ;
586       else if (unformat (line_input, "vni %d", &vni))
587         gid_address_vni (&eid) = vni;
588       else if (unformat (line_input, "locator-set %_%v%_", &locator_set_name))
589         {
590           p = hash_get_mem (lcm->locator_set_index_by_name, locator_set_name);
591           if (!p)
592             {
593               error = clib_error_return (0, "locator-set %s doesn't exist",
594                                          locator_set_name);
595               goto done;
596             }
597           locator_set_index = p[0];
598         }
599       else
600         {
601           error = unformat_parse_error (line_input);
602           goto done;
603         }
604     }
605   /* XXX treat batch configuration */
606
607   if (GID_ADDR_SRC_DST == gid_address_type (&eid))
608     {
609       error =
610         clib_error_return (0, "src/dst is not supported for local EIDs!");
611       goto done;
612     }
613
614   gid_address_copy (&a->eid, &eid);
615   a->is_add = is_add;
616   a->locator_set_index = locator_set_index;
617   a->local = 1;
618
619   rv = vnet_lisp_add_del_local_mapping (a, &map_index);
620   if (0 != rv)
621     {
622       error = clib_error_return (0, "failed to %s local mapping!",
623                                  is_add ? "add" : "delete");
624     }
625 done:
626   vec_free (eids);
627   if (locator_set_name)
628     vec_free (locator_set_name);
629   gid_address_free (&a->eid);
630   return error;
631 }
632
633 /* *INDENT-OFF* */
634 VLIB_CLI_COMMAND (lisp_add_del_local_eid_command) = {
635     .path = "lisp eid-table",
636     .short_help = "lisp eid-table add/del [vni <vni>] eid <eid> "
637       "locator-set <locator-set>",
638     .function = lisp_add_del_local_eid_command_fn,
639 };
640 /* *INDENT-ON* */
641
642 int
643 vnet_lisp_eid_table_map (u32 vni, u32 dp_id, u8 is_l2, u8 is_add)
644 {
645   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
646   uword *dp_idp, *vnip, **dp_table_by_vni, **vni_by_dp_table;
647
648   if (vnet_lisp_enable_disable_status () == 0)
649     {
650       clib_warning ("LISP is disabled!");
651       return -1;
652     }
653
654   dp_table_by_vni = is_l2 ? &lcm->bd_id_by_vni : &lcm->table_id_by_vni;
655   vni_by_dp_table = is_l2 ? &lcm->vni_by_bd_id : &lcm->vni_by_table_id;
656
657   if (!is_l2 && (vni == 0 || dp_id == 0))
658     {
659       clib_warning ("can't add/del default vni-vrf mapping!");
660       return -1;
661     }
662
663   dp_idp = hash_get (dp_table_by_vni[0], vni);
664   vnip = hash_get (vni_by_dp_table[0], dp_id);
665
666   if (is_add)
667     {
668       if (dp_idp || vnip)
669         {
670           clib_warning ("vni %d or vrf %d already used in vrf/vni "
671                         "mapping!", vni, dp_id);
672           return -1;
673         }
674       hash_set (dp_table_by_vni[0], vni, dp_id);
675       hash_set (vni_by_dp_table[0], dp_id, vni);
676
677       /* create dp iface */
678       dp_add_del_iface (lcm, vni, is_l2, 1);
679     }
680   else
681     {
682       if (!dp_idp || !vnip)
683         {
684           clib_warning ("vni %d or vrf %d not used in any vrf/vni! "
685                         "mapping!", vni, dp_id);
686           return -1;
687         }
688       hash_unset (dp_table_by_vni[0], vni);
689       hash_unset (vni_by_dp_table[0], dp_id);
690
691       /* remove dp iface */
692       dp_add_del_iface (lcm, vni, is_l2, 0);
693     }
694   return 0;
695
696 }
697
698 static clib_error_t *
699 lisp_eid_table_map_command_fn (vlib_main_t * vm,
700                                unformat_input_t * input,
701                                vlib_cli_command_t * cmd)
702 {
703   u8 is_add = 1, is_l2 = 0;
704   u32 vni = 0, dp_id = 0;
705   unformat_input_t _line_input, *line_input = &_line_input;
706
707   /* Get a line of input. */
708   if (!unformat_user (input, unformat_line_input, line_input))
709     return 0;
710
711   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
712     {
713       if (unformat (line_input, "del"))
714         is_add = 0;
715       else if (unformat (line_input, "vni %d", &vni))
716         ;
717       else if (unformat (line_input, "vrf %d", &dp_id))
718         ;
719       else if (unformat (line_input, "bd %d", &dp_id))
720         is_l2 = 1;
721       else
722         {
723           return unformat_parse_error (line_input);
724         }
725     }
726   vnet_lisp_eid_table_map (vni, dp_id, is_l2, is_add);
727   return 0;
728 }
729
730 /* *INDENT-OFF* */
731 VLIB_CLI_COMMAND (lisp_eid_table_map_command) = {
732     .path = "lisp eid-table map",
733     .short_help = "lisp eid-table map [del] vni <vni> vrf <vrf> | bd <bdi>",
734     .function = lisp_eid_table_map_command_fn,
735 };
736 /* *INDENT-ON* */
737
738 /* return 0 if the two locator sets are identical 1 otherwise */
739 static u8
740 compare_locators (lisp_cp_main_t * lcm, u32 * old_ls_indexes,
741                   locator_t * new_locators)
742 {
743   u32 i, old_li;
744   locator_t *old_loc, *new_loc;
745
746   if (vec_len (old_ls_indexes) != vec_len (new_locators))
747     return 1;
748
749   for (i = 0; i < vec_len (new_locators); i++)
750     {
751       old_li = vec_elt (old_ls_indexes, i);
752       old_loc = pool_elt_at_index (lcm->locator_pool, old_li);
753
754       new_loc = vec_elt_at_index (new_locators, i);
755
756       if (locator_cmp (old_loc, new_loc))
757         return 1;
758     }
759   return 0;
760 }
761
762 /**
763  * Adds/removes/updates mapping. Does not program forwarding.
764  *
765  * @param eid end-host identifier
766  * @param rlocs vector of remote locators
767  * @param action action for negative map-reply
768  * @param is_add add mapping if non-zero, delete otherwise
769  * @param res_map_index the map-index that was created/updated/removed. It is
770  *                      set to ~0 if no action is taken.
771  * @param is_static used for distinguishing between statically learned
772                     remote mappings and mappings obtained from MR
773  * @return return code
774  */
775 int
776 vnet_lisp_add_del_mapping (gid_address_t * eid, locator_t * rlocs, u8 action,
777                            u8 authoritative, u32 ttl, u8 is_add, u8 is_static,
778                            u32 * res_map_index)
779 {
780   vnet_lisp_add_del_mapping_args_t _m_args, *m_args = &_m_args;
781   vnet_lisp_add_del_locator_set_args_t _ls_args, *ls_args = &_ls_args;
782   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
783   u32 mi, ls_index = 0, dst_map_index;
784   mapping_t *old_map;
785
786   if (vnet_lisp_enable_disable_status () == 0)
787     {
788       clib_warning ("LISP is disabled!");
789       return VNET_API_ERROR_LISP_DISABLED;
790     }
791
792   if (res_map_index)
793     res_map_index[0] = ~0;
794
795   memset (m_args, 0, sizeof (m_args[0]));
796   memset (ls_args, 0, sizeof (ls_args[0]));
797
798   ls_args->locators = rlocs;
799
800   mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, eid);
801   old_map = ((u32) ~ 0 != mi) ? pool_elt_at_index (lcm->mapping_pool, mi) : 0;
802
803   if (is_add)
804     {
805       /* overwrite: if mapping already exists, decide if locators should be
806        * updated and be done */
807       if (old_map && gid_address_cmp (&old_map->eid, eid) == 0)
808         {
809           if (!is_static && (old_map->is_static || old_map->local))
810             {
811               /* do not overwrite local or static remote mappings */
812               clib_warning ("mapping %U rejected due to collision with local "
813                             "or static remote mapping!", format_gid_address,
814                             &eid);
815               return 0;
816             }
817
818           locator_set_t *old_ls;
819
820           /* update mapping attributes */
821           old_map->action = action;
822           old_map->authoritative = authoritative;
823           old_map->ttl = ttl;
824
825           old_ls = pool_elt_at_index (lcm->locator_set_pool,
826                                       old_map->locator_set_index);
827           if (compare_locators (lcm, old_ls->locator_indices,
828                                 ls_args->locators))
829             {
830               /* set locator-set index to overwrite */
831               ls_args->is_add = 1;
832               ls_args->index = old_map->locator_set_index;
833               vnet_lisp_add_del_locator_set (ls_args, 0);
834               if (res_map_index)
835                 res_map_index[0] = mi;
836             }
837         }
838       /* new mapping */
839       else
840         {
841           ls_args->is_add = 1;
842           ls_args->index = ~0;
843
844           vnet_lisp_add_del_locator_set (ls_args, &ls_index);
845
846           /* add mapping */
847           gid_address_copy (&m_args->eid, eid);
848           m_args->is_add = 1;
849           m_args->action = action;
850           m_args->locator_set_index = ls_index;
851           m_args->is_static = is_static;
852           vnet_lisp_map_cache_add_del (m_args, &dst_map_index);
853
854           if (res_map_index)
855             res_map_index[0] = dst_map_index;
856         }
857     }
858   else
859     {
860       if (old_map == 0 || gid_address_cmp (&old_map->eid, eid) != 0)
861         {
862           clib_warning ("cannot delete mapping for eid %U",
863                         format_gid_address, eid);
864           return -1;
865         }
866
867       m_args->is_add = 0;
868       gid_address_copy (&m_args->eid, eid);
869       m_args->locator_set_index = old_map->locator_set_index;
870
871       /* delete mapping associated from map-cache */
872       vnet_lisp_map_cache_add_del (m_args, 0);
873
874       ls_args->is_add = 0;
875       ls_args->index = old_map->locator_set_index;
876       /* delete locator set */
877       vnet_lisp_add_del_locator_set (ls_args, 0);
878
879       /* return old mapping index */
880       if (res_map_index)
881         res_map_index[0] = mi;
882     }
883
884   /* success */
885   return 0;
886 }
887
888 int
889 vnet_lisp_clear_all_remote_adjacencies (void)
890 {
891   int rv = 0;
892   u32 mi, *map_indices = 0, *map_indexp;
893   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
894   vnet_lisp_add_del_mapping_args_t _dm_args, *dm_args = &_dm_args;
895   vnet_lisp_add_del_locator_set_args_t _ls, *ls = &_ls;
896
897   /* *INDENT-OFF* */
898   pool_foreach_index (mi, lcm->mapping_pool,
899   ({
900     vec_add1 (map_indices, mi);
901   }));
902   /* *INDENT-ON* */
903
904   vec_foreach (map_indexp, map_indices)
905   {
906     mapping_t *map = pool_elt_at_index (lcm->mapping_pool, map_indexp[0]);
907     if (!map->local)
908       {
909         dp_del_fwd_entry (lcm, 0, map_indexp[0]);
910
911         dm_args->is_add = 0;
912         gid_address_copy (&dm_args->eid, &map->eid);
913         dm_args->locator_set_index = map->locator_set_index;
914
915         /* delete mapping associated to fwd entry */
916         vnet_lisp_map_cache_add_del (dm_args, 0);
917
918         ls->is_add = 0;
919         ls->local = 0;
920         ls->index = map->locator_set_index;
921         /* delete locator set */
922         rv = vnet_lisp_add_del_locator_set (ls, 0);
923         if (rv != 0)
924           goto cleanup;
925       }
926   }
927
928 cleanup:
929   if (map_indices)
930     vec_free (map_indices);
931   return rv;
932 }
933
934 /**
935  * Adds adjacency or removes forwarding entry associated to remote mapping.
936  * Note that adjacencies are not stored, they only result in forwarding entries
937  * being created.
938  */
939 int
940 lisp_add_del_adjacency (lisp_cp_main_t * lcm, gid_address_t * local_eid,
941                         gid_address_t * remote_eid, u8 is_add)
942 {
943   u32 local_mi, remote_mi = ~0;
944
945   if (vnet_lisp_enable_disable_status () == 0)
946     {
947       clib_warning ("LISP is disabled!");
948       return VNET_API_ERROR_LISP_DISABLED;
949     }
950
951   remote_mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, remote_eid);
952   if (GID_LOOKUP_MISS == remote_mi)
953     {
954       clib_warning ("Remote eid %U not found. Cannot add adjacency!",
955                     format_gid_address, remote_eid);
956
957       return -1;
958     }
959
960   if (is_add)
961     {
962       /* TODO 1) check if src/dst 2) once we have src/dst working, use it in
963        * delete*/
964
965       /* check if source eid has an associated mapping. If pitr mode is on,
966        * just use the pitr's mapping */
967       local_mi = lcm->lisp_pitr ? lcm->pitr_map_index :
968         gid_dictionary_lookup (&lcm->mapping_index_by_gid, local_eid);
969
970
971       if (GID_LOOKUP_MISS == local_mi)
972         {
973           clib_warning ("Local eid %U not found. Cannot add adjacency!",
974                         format_gid_address, local_eid);
975
976           return -1;
977         }
978
979       /* update forwarding */
980       dp_add_fwd_entry (lcm, local_mi, remote_mi);
981     }
982   else
983     dp_del_fwd_entry (lcm, 0, remote_mi);
984
985   return 0;
986 }
987
988 int
989 vnet_lisp_add_del_adjacency (vnet_lisp_add_del_adjacency_args_t * a)
990 {
991   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
992   return lisp_add_del_adjacency (lcm, &a->seid, &a->deid, a->is_add);
993 }
994
995 /**
996  * Handler for add/del remote mapping CLI.
997  *
998  * @param vm vlib context
999  * @param input input from user
1000  * @param cmd cmd
1001  * @return pointer to clib error structure
1002  */
1003 static clib_error_t *
1004 lisp_add_del_remote_mapping_command_fn (vlib_main_t * vm,
1005                                         unformat_input_t * input,
1006                                         vlib_cli_command_t * cmd)
1007 {
1008   clib_error_t *error = 0;
1009   unformat_input_t _line_input, *line_input = &_line_input;
1010   u8 is_add = 1, del_all = 0;
1011   locator_t rloc, *rlocs = 0, *curr_rloc = 0;
1012   gid_address_t eid;
1013   u8 eid_set = 0;
1014   u32 vni, action = ~0, p, w;
1015   int rv;
1016
1017   /* Get a line of input. */
1018   if (!unformat_user (input, unformat_line_input, line_input))
1019     return 0;
1020
1021   memset (&eid, 0, sizeof (eid));
1022   memset (&rloc, 0, sizeof (rloc));
1023
1024   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1025     {
1026       if (unformat (line_input, "del-all"))
1027         del_all = 1;
1028       else if (unformat (line_input, "del"))
1029         is_add = 0;
1030       else if (unformat (line_input, "add"))
1031         ;
1032       else if (unformat (line_input, "eid %U", unformat_gid_address, &eid))
1033         eid_set = 1;
1034       else if (unformat (line_input, "vni %u", &vni))
1035         {
1036           gid_address_vni (&eid) = vni;
1037         }
1038       else if (unformat (line_input, "p %d w %d", &p, &w))
1039         {
1040           if (!curr_rloc)
1041             {
1042               clib_warning
1043                 ("No RLOC configured for setting priority/weight!");
1044               goto done;
1045             }
1046           curr_rloc->priority = p;
1047           curr_rloc->weight = w;
1048         }
1049       else if (unformat (line_input, "rloc %U", unformat_ip_address,
1050                          &gid_address_ip (&rloc.address)))
1051         {
1052           vec_add1 (rlocs, rloc);
1053           curr_rloc = &rlocs[vec_len (rlocs) - 1];
1054         }
1055       else if (unformat (line_input, "action %U",
1056                          unformat_negative_mapping_action, &action))
1057         ;
1058       else
1059         {
1060           clib_warning ("parse error");
1061           goto done;
1062         }
1063     }
1064
1065   if (!eid_set)
1066     {
1067       clib_warning ("missing eid!");
1068       goto done;
1069     }
1070
1071   if (!del_all)
1072     {
1073       if (is_add && (~0 == action) && 0 == vec_len (rlocs))
1074         {
1075           clib_warning ("no action set for negative map-reply!");
1076           goto done;
1077         }
1078     }
1079   else
1080     {
1081       vnet_lisp_clear_all_remote_adjacencies ();
1082       goto done;
1083     }
1084
1085   /* TODO build src/dst with seid */
1086
1087   /* if it's a delete, clean forwarding */
1088   if (!is_add)
1089     {
1090       lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1091       rv = lisp_add_del_adjacency (lcm, 0, &eid, /* is_add */ 0);
1092       if (rv)
1093         {
1094           goto done;
1095         }
1096     }
1097
1098   /* add as static remote mapping, i.e., not authoritative and infinite
1099    * ttl */
1100   rv = vnet_lisp_add_del_mapping (&eid, rlocs, action, 0, ~0, is_add,
1101                                   1 /* is_static */ , 0);
1102
1103   if (rv)
1104     clib_warning ("failed to %s remote mapping!", is_add ? "add" : "delete");
1105
1106 done:
1107   vec_free (rlocs);
1108   unformat_free (line_input);
1109   return error;
1110 }
1111
1112 VLIB_CLI_COMMAND (lisp_add_del_remote_mapping_command) =
1113 {
1114 .path = "lisp remote-mapping",.short_help =
1115     "lisp remote-mapping add|del [del-all] vni <vni> "
1116     "eid <est-eid> [action <no-action|natively-forward|"
1117     "send-map-request|drop>] rloc <dst-locator> p <prio> w <weight> "
1118     "[rloc <dst-locator> ... ]",.function =
1119     lisp_add_del_remote_mapping_command_fn,};
1120
1121 /**
1122  * Handler for add/del adjacency CLI.
1123  */
1124 static clib_error_t *
1125 lisp_add_del_adjacency_command_fn (vlib_main_t * vm, unformat_input_t * input,
1126                                    vlib_cli_command_t * cmd)
1127 {
1128   clib_error_t *error = 0;
1129   unformat_input_t _line_input, *line_input = &_line_input;
1130   vnet_lisp_add_del_adjacency_args_t _a, *a = &_a;
1131   u8 is_add = 1;
1132   locator_t rloc, *rlocs = 0;
1133   ip_prefix_t *deid_ippref, *seid_ippref;
1134   gid_address_t seid, deid;
1135   u8 *dmac = gid_address_mac (&deid);
1136   u8 *smac = gid_address_mac (&seid);
1137   u8 deid_set = 0, seid_set = 0;
1138   u8 *s = 0;
1139   u32 vni, action = ~0;
1140   int rv;
1141
1142   /* Get a line of input. */
1143   if (!unformat_user (input, unformat_line_input, line_input))
1144     return 0;
1145
1146   memset (&deid, 0, sizeof (deid));
1147   memset (&seid, 0, sizeof (seid));
1148   memset (&rloc, 0, sizeof (rloc));
1149
1150   seid_ippref = &gid_address_ippref (&seid);
1151   deid_ippref = &gid_address_ippref (&deid);
1152
1153   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1154     {
1155       if (unformat (line_input, "del"))
1156         is_add = 0;
1157       else if (unformat (line_input, "add"))
1158         ;
1159       else if (unformat (line_input, "deid %U",
1160                          unformat_ip_prefix, deid_ippref))
1161         {
1162           gid_address_type (&deid) = GID_ADDR_IP_PREFIX;
1163           deid_set = 1;
1164         }
1165       else if (unformat (line_input, "deid %U", unformat_mac_address, dmac))
1166         {
1167           gid_address_type (&deid) = GID_ADDR_MAC;
1168           deid_set = 1;
1169         }
1170       else if (unformat (line_input, "vni %u", &vni))
1171         {
1172           gid_address_vni (&seid) = vni;
1173           gid_address_vni (&deid) = vni;
1174         }
1175       else if (unformat (line_input, "seid %U",
1176                          unformat_ip_prefix, seid_ippref))
1177         {
1178           gid_address_type (&seid) = GID_ADDR_IP_PREFIX;
1179           seid_set = 1;
1180         }
1181       else if (unformat (line_input, "seid %U", unformat_mac_address, smac))
1182         {
1183           gid_address_type (&seid) = GID_ADDR_MAC;
1184           seid_set = 1;
1185         }
1186       else
1187         {
1188           clib_warning ("parse error");
1189           goto done;
1190         }
1191     }
1192
1193   if (!deid_set)
1194     {
1195       clib_warning ("missing deid!");
1196       goto done;
1197     }
1198
1199   if (GID_ADDR_IP_PREFIX == gid_address_type (&deid))
1200     {
1201       /* if seid not set, make sure the ip version is the same as that
1202        * of the deid. This ensures the seid to be configured will be
1203        * either 0/0 or ::/0 */
1204       if (!seid_set)
1205         ip_prefix_version (seid_ippref) = ip_prefix_version (deid_ippref);
1206
1207       if (is_add &&
1208           (ip_prefix_version (deid_ippref)
1209            != ip_prefix_version (seid_ippref)))
1210         {
1211           clib_warning ("source and destination EIDs are not"
1212                         " in the same IP family!");
1213           goto done;
1214         }
1215     }
1216
1217   if (is_add && (~0 == action) && 0 == vec_len (rlocs))
1218     {
1219       clib_warning ("no action set for negative map-reply!");
1220       goto done;
1221     }
1222
1223   memset (a, 0, sizeof (a[0]));
1224   gid_address_copy (&a->seid, &deid);
1225   gid_address_copy (&a->deid, &seid);
1226
1227   a->is_add = is_add;
1228   rv = vnet_lisp_add_del_adjacency (a);
1229
1230   if (rv)
1231     clib_warning ("failed to %s adjacency!", is_add ? "add" : "delete");
1232
1233 done:
1234   unformat_free (line_input);
1235   if (s)
1236     vec_free (s);
1237   return error;
1238 }
1239
1240 /* *INDENT-OFF* */
1241 VLIB_CLI_COMMAND (lisp_add_del_adjacency_command) = {
1242     .path = "lisp adjacency",
1243     .short_help = "lisp adjacency add|del vni <vni>"
1244      "deid <dest-eid> seid <src-eid> [action <no-action|natively-forward|"
1245      "send-map-request|drop>] rloc <dst-locator> [rloc <dst-locator> ... ]",
1246     .function = lisp_add_del_adjacency_command_fn,
1247 };
1248 /* *INDENT-ON* */
1249
1250 static clib_error_t *
1251 lisp_show_map_resolvers_command_fn (vlib_main_t * vm,
1252                                     unformat_input_t * input,
1253                                     vlib_cli_command_t * cmd)
1254 {
1255   map_resolver_t *mr;
1256   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1257
1258   vec_foreach (mr, lcm->map_resolvers)
1259   {
1260     vlib_cli_output (vm, "%U", format_ip_address, &mr->address);
1261   }
1262   return 0;
1263 }
1264
1265 /* *INDENT-OFF* */
1266 VLIB_CLI_COMMAND (lisp_show_map_resolvers_command) = {
1267     .path = "show lisp map-resolvers",
1268     .short_help = "show lisp map-resolvers",
1269     .function = lisp_show_map_resolvers_command_fn,
1270 };
1271 /* *INDENT-ON* */
1272
1273 int
1274 vnet_lisp_pitr_set_locator_set (u8 * locator_set_name, u8 is_add)
1275 {
1276   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1277   u32 locator_set_index = ~0;
1278   mapping_t *m;
1279   uword *p;
1280
1281   if (vnet_lisp_enable_disable_status () == 0)
1282     {
1283       clib_warning ("LISP is disabled!");
1284       return VNET_API_ERROR_LISP_DISABLED;
1285     }
1286
1287   p = hash_get_mem (lcm->locator_set_index_by_name, locator_set_name);
1288   if (!p)
1289     {
1290       clib_warning ("locator-set %v doesn't exist", locator_set_name);
1291       return -1;
1292     }
1293   locator_set_index = p[0];
1294
1295   if (is_add)
1296     {
1297       pool_get (lcm->mapping_pool, m);
1298       m->locator_set_index = locator_set_index;
1299       m->local = 1;
1300       lcm->pitr_map_index = m - lcm->mapping_pool;
1301
1302       /* enable pitr mode */
1303       lcm->lisp_pitr = 1;
1304     }
1305   else
1306     {
1307       /* remove pitr mapping */
1308       pool_put_index (lcm->mapping_pool, lcm->pitr_map_index);
1309
1310       /* disable pitr mode */
1311       lcm->lisp_pitr = 0;
1312     }
1313   return 0;
1314 }
1315
1316 static clib_error_t *
1317 lisp_pitr_set_locator_set_command_fn (vlib_main_t * vm,
1318                                       unformat_input_t * input,
1319                                       vlib_cli_command_t * cmd)
1320 {
1321   u8 locator_name_set = 0;
1322   u8 *locator_set_name = 0;
1323   u8 is_add = 1;
1324   unformat_input_t _line_input, *line_input = &_line_input;
1325   clib_error_t *error = 0;
1326   int rv = 0;
1327
1328   /* Get a line of input. */
1329   if (!unformat_user (input, unformat_line_input, line_input))
1330     return 0;
1331
1332   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1333     {
1334       if (unformat (line_input, "ls %_%v%_", &locator_set_name))
1335         locator_name_set = 1;
1336       else if (unformat (line_input, "disable"))
1337         is_add = 0;
1338       else
1339         return clib_error_return (0, "parse error");
1340     }
1341
1342   if (!locator_name_set)
1343     {
1344       clib_warning ("No locator set specified!");
1345       goto done;
1346     }
1347   rv = vnet_lisp_pitr_set_locator_set (locator_set_name, is_add);
1348   if (0 != rv)
1349     {
1350       error = clib_error_return (0, "failed to %s pitr!",
1351                                  is_add ? "add" : "delete");
1352     }
1353
1354 done:
1355   if (locator_set_name)
1356     vec_free (locator_set_name);
1357   return error;
1358 }
1359
1360 /* *INDENT-OFF* */
1361 VLIB_CLI_COMMAND (lisp_pitr_set_locator_set_command) = {
1362     .path = "lisp pitr",
1363     .short_help = "lisp pitr [disable] ls <locator-set-name>",
1364     .function = lisp_pitr_set_locator_set_command_fn,
1365 };
1366 /* *INDENT-ON* */
1367
1368 static clib_error_t *
1369 lisp_show_pitr_command_fn (vlib_main_t * vm,
1370                            unformat_input_t * input, vlib_cli_command_t * cmd)
1371 {
1372   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1373   mapping_t *m;
1374   locator_set_t *ls;
1375   u8 *tmp_str = 0;
1376
1377   vlib_cli_output (vm, "%=20s%=16s",
1378                    "pitr", lcm->lisp_pitr ? "locator-set" : "");
1379
1380   if (!lcm->lisp_pitr)
1381     {
1382       vlib_cli_output (vm, "%=20s", "disable");
1383       return 0;
1384     }
1385
1386   if (~0 == lcm->pitr_map_index)
1387     {
1388       tmp_str = format (0, "N/A");
1389     }
1390   else
1391     {
1392       m = pool_elt_at_index (lcm->mapping_pool, lcm->pitr_map_index);
1393       if (~0 != m->locator_set_index)
1394         {
1395           ls =
1396             pool_elt_at_index (lcm->locator_set_pool, m->locator_set_index);
1397           tmp_str = format (0, "%s", ls->name);
1398         }
1399       else
1400         {
1401           tmp_str = format (0, "N/A");
1402         }
1403     }
1404   vec_add1 (tmp_str, 0);
1405
1406   vlib_cli_output (vm, "%=20s%=16s", "enable", tmp_str);
1407
1408   vec_free (tmp_str);
1409
1410   return 0;
1411 }
1412
1413 /* *INDENT-OFF* */
1414 VLIB_CLI_COMMAND (lisp_show_pitr_command) = {
1415     .path = "show lisp pitr",
1416     .short_help = "Show pitr",
1417     .function = lisp_show_pitr_command_fn,
1418 };
1419 /* *INDENT-ON* */
1420
1421 static u8 *
1422 format_eid_entry (u8 * s, va_list * args)
1423 {
1424   vnet_main_t *vnm = va_arg (*args, vnet_main_t *);
1425   lisp_cp_main_t *lcm = va_arg (*args, lisp_cp_main_t *);
1426   mapping_t *mapit = va_arg (*args, mapping_t *);
1427   locator_set_t *ls = va_arg (*args, locator_set_t *);
1428   gid_address_t *gid = &mapit->eid;
1429   u32 ttl = mapit->ttl;
1430   u8 aut = mapit->authoritative;
1431   u32 *loc_index;
1432   u8 first_line = 1;
1433   u8 *loc;
1434
1435   u8 *type = ls->local ? format (0, "local(%s)", ls->name)
1436     : format (0, "remote");
1437
1438   if (vec_len (ls->locator_indices) == 0)
1439     {
1440       s = format (s, "%-35U%-30s%-20u%-u", format_gid_address, gid,
1441                   type, ttl, aut);
1442     }
1443   else
1444     {
1445       vec_foreach (loc_index, ls->locator_indices)
1446       {
1447         locator_t *l = pool_elt_at_index (lcm->locator_pool, loc_index[0]);
1448         if (l->local)
1449           loc = format (0, "%U", format_vnet_sw_if_index_name, vnm,
1450                         l->sw_if_index);
1451         else
1452           loc = format (0, "%U", format_ip_address,
1453                         &gid_address_ip (&l->address));
1454
1455         if (first_line)
1456           {
1457             s = format (s, "%-35U%-20s%-30v%-20u%-u\n", format_gid_address,
1458                         gid, type, loc, ttl, aut);
1459             first_line = 0;
1460           }
1461         else
1462           s = format (s, "%55s%v\n", "", loc);
1463       }
1464     }
1465   return s;
1466 }
1467
1468 static clib_error_t *
1469 lisp_show_eid_table_command_fn (vlib_main_t * vm,
1470                                 unformat_input_t * input,
1471                                 vlib_cli_command_t * cmd)
1472 {
1473   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1474   mapping_t *mapit;
1475   unformat_input_t _line_input, *line_input = &_line_input;
1476   u32 mi;
1477   gid_address_t eid;
1478   u8 print_all = 1;
1479   u8 filter = 0;
1480
1481   memset (&eid, 0, sizeof (eid));
1482
1483   /* Get a line of input. */
1484   if (!unformat_user (input, unformat_line_input, line_input))
1485     return 0;
1486
1487   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1488     {
1489       if (unformat (line_input, "eid %U", unformat_gid_address, &eid))
1490         print_all = 0;
1491       else if (unformat (line_input, "local"))
1492         filter = 1;
1493       else if (unformat (line_input, "remote"))
1494         filter = 2;
1495       else
1496         return clib_error_return (0, "parse error: '%U'",
1497                                   format_unformat_error, line_input);
1498     }
1499
1500   vlib_cli_output (vm, "%-35s%-20s%-30s%-20s%-s",
1501                    "EID", "type", "locators", "ttl", "autoritative");
1502
1503   if (print_all)
1504     {
1505       /* *INDENT-OFF* */
1506       pool_foreach (mapit, lcm->mapping_pool,
1507       ({
1508         locator_set_t * ls = pool_elt_at_index (lcm->locator_set_pool,
1509                                                 mapit->locator_set_index);
1510         if (filter && !((1 == filter && ls->local) ||
1511           (2 == filter && !ls->local)))
1512           {
1513             continue;
1514           }
1515         vlib_cli_output (vm, "%U", format_eid_entry, lcm->vnet_main,
1516                          lcm, mapit, ls);
1517       }));
1518       /* *INDENT-ON* */
1519     }
1520   else
1521     {
1522       mi = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &eid);
1523       if ((u32) ~ 0 == mi)
1524         return 0;
1525
1526       mapit = pool_elt_at_index (lcm->mapping_pool, mi);
1527       locator_set_t *ls = pool_elt_at_index (lcm->locator_set_pool,
1528                                              mapit->locator_set_index);
1529
1530       if (filter && !((1 == filter && ls->local) ||
1531                       (2 == filter && !ls->local)))
1532         {
1533           return 0;
1534         }
1535
1536       vlib_cli_output (vm, "%U,", format_eid_entry, lcm->vnet_main,
1537                        lcm, mapit, ls);
1538     }
1539
1540   return 0;
1541 }
1542
1543 /* *INDENT-OFF* */
1544 VLIB_CLI_COMMAND (lisp_cp_show_eid_table_command) = {
1545     .path = "show lisp eid-table",
1546     .short_help = "Shows EID table",
1547     .function = lisp_show_eid_table_command_fn,
1548 };
1549 /* *INDENT-ON* */
1550
1551 /* cleans locator to locator-set data and removes locators not part of
1552  * any locator-set */
1553 static void
1554 clean_locator_to_locator_set (lisp_cp_main_t * lcm, u32 lsi)
1555 {
1556   u32 i, j, *loc_indexp, *ls_indexp, **ls_indexes, *to_be_deleted = 0;
1557   locator_set_t *ls = pool_elt_at_index (lcm->locator_set_pool, lsi);
1558   for (i = 0; i < vec_len (ls->locator_indices); i++)
1559     {
1560       loc_indexp = vec_elt_at_index (ls->locator_indices, i);
1561       ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets,
1562                                      loc_indexp[0]);
1563       for (j = 0; j < vec_len (ls_indexes[0]); j++)
1564         {
1565           ls_indexp = vec_elt_at_index (ls_indexes[0], j);
1566           if (ls_indexp[0] == lsi)
1567             break;
1568         }
1569
1570       /* delete index for removed locator-set */
1571       vec_del1 (ls_indexes[0], j);
1572
1573       /* delete locator if it's part of no locator-set */
1574       if (vec_len (ls_indexes[0]) == 0)
1575         {
1576           pool_put_index (lcm->locator_pool, loc_indexp[0]);
1577           vec_add1 (to_be_deleted, i);
1578         }
1579     }
1580
1581   if (to_be_deleted)
1582     {
1583       for (i = 0; i < vec_len (to_be_deleted); i++)
1584         {
1585           loc_indexp = vec_elt_at_index (to_be_deleted, i);
1586           vec_del1 (ls->locator_indices, loc_indexp[0]);
1587         }
1588       vec_free (to_be_deleted);
1589     }
1590 }
1591
1592 static inline uword *
1593 get_locator_set_index (vnet_lisp_add_del_locator_set_args_t * a, uword * p)
1594 {
1595   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1596
1597   ASSERT (a != NULL);
1598   ASSERT (p != NULL);
1599
1600   /* find locator-set */
1601   if (a->local)
1602     {
1603       p = hash_get_mem (lcm->locator_set_index_by_name, a->name);
1604     }
1605   else
1606     {
1607       *p = a->index;
1608     }
1609
1610   return p;
1611 }
1612
1613 static inline int
1614 is_locator_in_locator_set (lisp_cp_main_t * lcm, locator_set_t * ls,
1615                            locator_t * loc)
1616 {
1617   locator_t *itloc;
1618   u32 *locit;
1619
1620   ASSERT (ls != NULL);
1621   ASSERT (loc != NULL);
1622
1623   vec_foreach (locit, ls->locator_indices)
1624   {
1625     itloc = pool_elt_at_index (lcm->locator_pool, locit[0]);
1626     if ((ls->local && itloc->sw_if_index == loc->sw_if_index) ||
1627         (!ls->local && !gid_address_cmp (&itloc->address, &loc->address)))
1628       {
1629         clib_warning ("Duplicate locator");
1630         return VNET_API_ERROR_VALUE_EXIST;
1631       }
1632   }
1633
1634   return 0;
1635 }
1636
1637 static inline void
1638 remove_locator_from_locator_set (locator_set_t * ls, u32 * locit,
1639                                  u32 ls_index, u32 loc_id)
1640 {
1641   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1642   u32 **ls_indexes = NULL;
1643
1644   ASSERT (ls != NULL);
1645   ASSERT (locit != NULL);
1646
1647   ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets, locit[0]);
1648   pool_put_index (lcm->locator_pool, locit[0]);
1649   vec_del1 (ls->locator_indices, loc_id);
1650   vec_del1 (ls_indexes[0], ls_index);
1651 }
1652
1653 int
1654 vnet_lisp_add_del_locator (vnet_lisp_add_del_locator_set_args_t * a,
1655                            locator_set_t * ls, u32 * ls_result)
1656 {
1657   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1658   locator_t *loc = NULL, *itloc = NULL;
1659   uword _p = (u32) ~ 0, *p = &_p;
1660   u32 loc_index = ~0, ls_index = ~0, *locit = NULL, **ls_indexes = NULL;
1661   u32 loc_id = ~0;
1662   int ret = 0;
1663
1664   ASSERT (a != NULL);
1665
1666   if (vnet_lisp_enable_disable_status () == 0)
1667     {
1668       clib_warning ("LISP is disabled!");
1669       return VNET_API_ERROR_LISP_DISABLED;
1670     }
1671
1672   p = get_locator_set_index (a, p);
1673   if (!p)
1674     {
1675       clib_warning ("locator-set %v doesn't exist", a->name);
1676       return VNET_API_ERROR_INVALID_ARGUMENT;
1677     }
1678
1679   if (ls == 0)
1680     {
1681       ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
1682       if (!ls)
1683         {
1684           clib_warning ("locator-set %d to be overwritten doesn't exist!",
1685                         p[0]);
1686           return VNET_API_ERROR_INVALID_ARGUMENT;
1687         }
1688     }
1689
1690   if (a->is_add)
1691     {
1692       if (ls_result)
1693         ls_result[0] = p[0];
1694
1695       /* allocate locators */
1696       vec_foreach (itloc, a->locators)
1697       {
1698         ret = is_locator_in_locator_set (lcm, ls, itloc);
1699         if (0 != ret)
1700           {
1701             return ret;
1702           }
1703
1704         pool_get (lcm->locator_pool, loc);
1705         loc[0] = itloc[0];
1706         loc_index = loc - lcm->locator_pool;
1707
1708         vec_add1 (ls->locator_indices, loc_index);
1709
1710         vec_validate (lcm->locator_to_locator_sets, loc_index);
1711         ls_indexes = vec_elt_at_index (lcm->locator_to_locator_sets,
1712                                        loc_index);
1713         vec_add1 (ls_indexes[0], ls_index);
1714       }
1715     }
1716   else
1717     {
1718       ls_index = p[0];
1719
1720       itloc = a->locators;
1721       loc_id = 0;
1722       vec_foreach (locit, ls->locator_indices)
1723       {
1724         loc = pool_elt_at_index (lcm->locator_pool, locit[0]);
1725
1726         if (loc->local && loc->sw_if_index == itloc->sw_if_index)
1727           {
1728             remove_locator_from_locator_set (ls, locit, ls_index, loc_id);
1729           }
1730         if (0 == loc->local &&
1731             !gid_address_cmp (&loc->address, &itloc->address))
1732           {
1733             remove_locator_from_locator_set (ls, locit, ls_index, loc_id);
1734           }
1735
1736         loc_id++;
1737       }
1738     }
1739
1740   return 0;
1741 }
1742
1743 int
1744 vnet_lisp_add_del_locator_set (vnet_lisp_add_del_locator_set_args_t * a,
1745                                u32 * ls_result)
1746 {
1747   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1748   locator_set_t *ls;
1749   uword _p = (u32) ~ 0, *p = &_p;
1750   u32 ls_index;
1751   u32 **eid_indexes;
1752   int ret = 0;
1753
1754   if (vnet_lisp_enable_disable_status () == 0)
1755     {
1756       clib_warning ("LISP is disabled!");
1757       return VNET_API_ERROR_LISP_DISABLED;
1758     }
1759
1760   if (a->is_add)
1761     {
1762       p = get_locator_set_index (a, p);
1763
1764       /* overwrite */
1765       if (p && p[0] != (u32) ~ 0)
1766         {
1767           ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
1768           if (!ls)
1769             {
1770               clib_warning ("locator-set %d to be overwritten doesn't exist!",
1771                             p[0]);
1772               return -1;
1773             }
1774
1775           /* clean locator to locator-set vectors and remove locators if
1776            * they're not part of another locator-set */
1777           clean_locator_to_locator_set (lcm, p[0]);
1778
1779           /* remove locator indices from locator set */
1780           vec_free (ls->locator_indices);
1781
1782           ls_index = p[0];
1783
1784           if (ls_result)
1785             ls_result[0] = p[0];
1786         }
1787       /* new locator-set */
1788       else
1789         {
1790           pool_get (lcm->locator_set_pool, ls);
1791           memset (ls, 0, sizeof (*ls));
1792           ls_index = ls - lcm->locator_set_pool;
1793
1794           if (a->local)
1795             {
1796               ls->name = vec_dup (a->name);
1797
1798               if (!lcm->locator_set_index_by_name)
1799                 lcm->locator_set_index_by_name = hash_create_vec (
1800                                                                    /* size */
1801                                                                    0,
1802                                                                    sizeof
1803                                                                    (ls->name
1804                                                                     [0]),
1805                                                                    sizeof
1806                                                                    (uword));
1807               hash_set_mem (lcm->locator_set_index_by_name, ls->name,
1808                             ls_index);
1809
1810               /* mark as local locator-set */
1811               vec_add1 (lcm->local_locator_set_indexes, ls_index);
1812             }
1813           ls->local = a->local;
1814           if (ls_result)
1815             ls_result[0] = ls_index;
1816         }
1817
1818       ret = vnet_lisp_add_del_locator (a, ls, NULL);
1819       if (0 != ret)
1820         {
1821           return ret;
1822         }
1823     }
1824   else
1825     {
1826       p = get_locator_set_index (a, p);
1827       if (!p)
1828         {
1829           clib_warning ("locator-set %v doesn't exists", a->name);
1830           return -1;
1831         }
1832
1833       ls = pool_elt_at_index (lcm->locator_set_pool, p[0]);
1834       if (!ls)
1835         {
1836           clib_warning ("locator-set with index %d doesn't exists", p[0]);
1837           return -1;
1838         }
1839
1840       if (lcm->mreq_itr_rlocs == p[0])
1841         {
1842           clib_warning ("Can't delete the locator-set used to constrain "
1843                         "the itr-rlocs in map-requests!");
1844           return -1;
1845         }
1846
1847       if (vec_len (lcm->locator_set_to_eids) != 0)
1848         {
1849           eid_indexes = vec_elt_at_index (lcm->locator_set_to_eids, p[0]);
1850           if (vec_len (eid_indexes[0]) != 0)
1851             {
1852               clib_warning
1853                 ("Can't delete a locator that supports a mapping!");
1854               return -1;
1855             }
1856         }
1857
1858       /* clean locator to locator-sets data */
1859       clean_locator_to_locator_set (lcm, p[0]);
1860
1861       if (ls->local)
1862         {
1863           u32 it, lsi;
1864
1865           vec_foreach_index (it, lcm->local_locator_set_indexes)
1866           {
1867             lsi = vec_elt (lcm->local_locator_set_indexes, it);
1868             if (lsi == p[0])
1869               {
1870                 vec_del1 (lcm->local_locator_set_indexes, it);
1871                 break;
1872               }
1873           }
1874           hash_unset_mem (lcm->locator_set_index_by_name, ls->name);
1875         }
1876       vec_free (ls->name);
1877       vec_free (ls->locator_indices);
1878       pool_put (lcm->locator_set_pool, ls);
1879     }
1880   return 0;
1881 }
1882
1883 clib_error_t *
1884 vnet_lisp_enable_disable (u8 is_enable)
1885 {
1886   u32 vni, dp_table;
1887   clib_error_t *error = 0;
1888   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1889   vnet_lisp_gpe_enable_disable_args_t _a, *a = &_a;
1890
1891   a->is_en = is_enable;
1892   error = vnet_lisp_gpe_enable_disable (a);
1893   if (error)
1894     {
1895       return clib_error_return (0, "failed to %s data-plane!",
1896                                 a->is_en ? "enable" : "disable");
1897     }
1898
1899   if (is_enable)
1900     {
1901       /* enable all l2 and l3 ifaces */
1902
1903       /* *INDENT-OFF* */
1904       hash_foreach(vni, dp_table, lcm->table_id_by_vni, ({
1905         dp_add_del_iface(lcm, vni, 0, 1);
1906       }));
1907       hash_foreach(vni, dp_table, lcm->bd_id_by_vni, ({
1908         dp_add_del_iface(lcm, vni, /* is_l2 */ 1, 1);
1909       }));
1910       /* *INDENT-ON* */
1911     }
1912   else
1913     {
1914       /* clear interface table */
1915       hash_free (lcm->dp_intf_by_vni);
1916       hash_free (lcm->fwd_entry_by_mapping_index);
1917       pool_free (lcm->fwd_entry_pool);
1918     }
1919
1920   /* update global flag */
1921   lcm->is_enabled = is_enable;
1922
1923   return 0;
1924 }
1925
1926 static clib_error_t *
1927 lisp_enable_disable_command_fn (vlib_main_t * vm, unformat_input_t * input,
1928                                 vlib_cli_command_t * cmd)
1929 {
1930   unformat_input_t _line_input, *line_input = &_line_input;
1931   u8 is_enabled = 0;
1932   u8 is_set = 0;
1933
1934   /* Get a line of input. */
1935   if (!unformat_user (input, unformat_line_input, line_input))
1936     return 0;
1937
1938   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1939     {
1940       if (unformat (line_input, "enable"))
1941         {
1942           is_set = 1;
1943           is_enabled = 1;
1944         }
1945       else if (unformat (line_input, "disable"))
1946         is_set = 1;
1947       else
1948         {
1949           return clib_error_return (0, "parse error: '%U'",
1950                                     format_unformat_error, line_input);
1951         }
1952     }
1953
1954   if (!is_set)
1955     return clib_error_return (0, "state not set");
1956
1957   return vnet_lisp_enable_disable (is_enabled);
1958 }
1959
1960 /* *INDENT-OFF* */
1961 VLIB_CLI_COMMAND (lisp_cp_enable_disable_command) = {
1962     .path = "lisp",
1963     .short_help = "lisp [enable|disable]",
1964     .function = lisp_enable_disable_command_fn,
1965 };
1966 /* *INDENT-ON* */
1967
1968 u8
1969 vnet_lisp_enable_disable_status (void)
1970 {
1971   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1972   return lcm->is_enabled;
1973 }
1974
1975 static u8 *
1976 format_lisp_status (u8 * s, va_list * args)
1977 {
1978   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
1979   return format (s, "%s", lcm->is_enabled ? "enabled" : "disabled");
1980 }
1981
1982 static clib_error_t *
1983 lisp_show_status_command_fn (vlib_main_t * vm, unformat_input_t * input,
1984                              vlib_cli_command_t * cmd)
1985 {
1986   u8 *msg = 0;
1987   msg = format (msg, "feature: %U\ngpe: %U\n",
1988                 format_lisp_status, format_vnet_lisp_gpe_status);
1989   vlib_cli_output (vm, "%v", msg);
1990   vec_free (msg);
1991   return 0;
1992 }
1993
1994 /* *INDENT-OFF* */
1995 VLIB_CLI_COMMAND (lisp_show_status_command) = {
1996     .path = "show lisp status",
1997     .short_help = "show lisp status",
1998     .function = lisp_show_status_command_fn,
1999 };
2000 /* *INDENT-ON* */
2001
2002 static clib_error_t *
2003 lisp_show_eid_table_map_command_fn (vlib_main_t * vm,
2004                                     unformat_input_t * input,
2005                                     vlib_cli_command_t * cmd)
2006 {
2007   hash_pair_t *p;
2008   unformat_input_t _line_input, *line_input = &_line_input;
2009   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2010   uword *vni_table = 0;
2011   u8 is_l2 = 0;
2012
2013   /* Get a line of input. */
2014   if (!unformat_user (input, unformat_line_input, line_input))
2015     return 0;
2016
2017   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
2018     {
2019       if (unformat (line_input, "l2"))
2020         {
2021           vni_table = lcm->bd_id_by_vni;
2022           is_l2 = 1;
2023         }
2024       else if (unformat (line_input, "l3"))
2025         {
2026           vni_table = lcm->table_id_by_vni;
2027           is_l2 = 0;
2028         }
2029       else
2030         return clib_error_return (0, "parse error: '%U'",
2031                                   format_unformat_error, line_input);
2032     }
2033
2034   if (!vni_table)
2035     {
2036       vlib_cli_output (vm, "Error: expected l2|l3 param!\n");
2037       return 0;
2038     }
2039
2040   vlib_cli_output (vm, "%=10s%=10s", "VNI", is_l2 ? "BD" : "VRF");
2041
2042   /* *INDENT-OFF* */
2043   hash_foreach_pair (p, vni_table,
2044   ({
2045     vlib_cli_output (vm, "%=10d%=10d", p->key, p->value[0]);
2046   }));
2047   /* *INDENT-ON* */
2048
2049   return 0;
2050 }
2051
2052 /* *INDENT-OFF* */
2053 VLIB_CLI_COMMAND (lisp_show_eid_table_map_command) = {
2054     .path = "show lisp eid-table map",
2055     .short_help = "show lisp eid-table l2|l3",
2056     .function = lisp_show_eid_table_map_command_fn,
2057 };
2058 /* *INDENT-ON* */
2059
2060 static clib_error_t *
2061 lisp_add_del_locator_set_command_fn (vlib_main_t * vm,
2062                                      unformat_input_t * input,
2063                                      vlib_cli_command_t * cmd)
2064 {
2065   lisp_gpe_main_t *lgm = &lisp_gpe_main;
2066   vnet_main_t *vnm = lgm->vnet_main;
2067   unformat_input_t _line_input, *line_input = &_line_input;
2068   u8 is_add = 1;
2069   clib_error_t *error = 0;
2070   u8 *locator_set_name = 0;
2071   locator_t locator, *locators = 0;
2072   vnet_lisp_add_del_locator_set_args_t _a, *a = &_a;
2073   u32 ls_index = 0;
2074   int rv = 0;
2075
2076   memset (&locator, 0, sizeof (locator));
2077   memset (a, 0, sizeof (a[0]));
2078
2079   /* Get a line of input. */
2080   if (!unformat_user (input, unformat_line_input, line_input))
2081     return 0;
2082
2083   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
2084     {
2085       if (unformat (line_input, "add %_%v%_", &locator_set_name))
2086         is_add = 1;
2087       else if (unformat (line_input, "del %_%v%_", &locator_set_name))
2088         is_add = 0;
2089       else if (unformat (line_input, "iface %U p %d w %d",
2090                          unformat_vnet_sw_interface, vnm,
2091                          &locator.sw_if_index, &locator.priority,
2092                          &locator.weight))
2093         {
2094           locator.local = 1;
2095           vec_add1 (locators, locator);
2096         }
2097       else
2098         {
2099           error = unformat_parse_error (line_input);
2100           goto done;
2101         }
2102     }
2103
2104   a->name = locator_set_name;
2105   a->locators = locators;
2106   a->is_add = is_add;
2107   a->local = 1;
2108
2109   rv = vnet_lisp_add_del_locator_set (a, &ls_index);
2110   if (0 != rv)
2111     {
2112       error = clib_error_return (0, "failed to %s locator-set!",
2113                                  is_add ? "add" : "delete");
2114     }
2115
2116 done:
2117   vec_free (locators);
2118   if (locator_set_name)
2119     vec_free (locator_set_name);
2120   return error;
2121 }
2122
2123 /* *INDENT-OFF* */
2124 VLIB_CLI_COMMAND (lisp_cp_add_del_locator_set_command) = {
2125     .path = "lisp locator-set",
2126     .short_help = "lisp locator-set add/del <name> [iface <iface-name> "
2127         "p <priority> w <weight>]",
2128     .function = lisp_add_del_locator_set_command_fn,
2129 };
2130 /* *INDENT-ON* */
2131
2132 static clib_error_t *
2133 lisp_add_del_locator_in_set_command_fn (vlib_main_t * vm,
2134                                         unformat_input_t * input,
2135                                         vlib_cli_command_t * cmd)
2136 {
2137   lisp_gpe_main_t *lgm = &lisp_gpe_main;
2138   vnet_main_t *vnm = lgm->vnet_main;
2139   unformat_input_t _line_input, *line_input = &_line_input;
2140   u8 is_add = 1;
2141   clib_error_t *error = 0;
2142   u8 *locator_set_name = 0;
2143   u8 locator_set_name_set = 0;
2144   locator_t locator, *locators = 0;
2145   vnet_lisp_add_del_locator_set_args_t _a, *a = &_a;
2146   u32 ls_index = 0;
2147
2148   memset (&locator, 0, sizeof (locator));
2149   memset (a, 0, sizeof (a[0]));
2150
2151   /* Get a line of input. */
2152   if (!unformat_user (input, unformat_line_input, line_input))
2153     return 0;
2154
2155   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
2156     {
2157       if (unformat (line_input, "add"))
2158         is_add = 1;
2159       else if (unformat (line_input, "del"))
2160         is_add = 0;
2161       else if (unformat (line_input, "locator-set %_%v%_", &locator_set_name))
2162         locator_set_name_set = 1;
2163       else if (unformat (line_input, "iface %U p %d w %d",
2164                          unformat_vnet_sw_interface, vnm,
2165                          &locator.sw_if_index, &locator.priority,
2166                          &locator.weight))
2167         {
2168           locator.local = 1;
2169           vec_add1 (locators, locator);
2170         }
2171       else
2172         {
2173           error = unformat_parse_error (line_input);
2174           goto done;
2175         }
2176     }
2177
2178   if (!locator_set_name_set)
2179     {
2180       error = clib_error_return (0, "locator_set name not set!");
2181       goto done;
2182     }
2183
2184   a->name = locator_set_name;
2185   a->locators = locators;
2186   a->is_add = is_add;
2187   a->local = 1;
2188
2189   vnet_lisp_add_del_locator (a, 0, &ls_index);
2190
2191 done:
2192   vec_free (locators);
2193   vec_free (locator_set_name);
2194   return error;
2195 }
2196
2197 /* *INDENT-OFF* */
2198 VLIB_CLI_COMMAND (lisp_cp_add_del_locator_in_set_command) = {
2199     .path = "lisp locator",
2200     .short_help = "lisp locator add/del locator-set <name> iface <iface-name> "
2201                   "p <priority> w <weight>",
2202     .function = lisp_add_del_locator_in_set_command_fn,
2203 };
2204 /* *INDENT-ON* */
2205
2206 static clib_error_t *
2207 lisp_cp_show_locator_sets_command_fn (vlib_main_t * vm,
2208                                       unformat_input_t * input,
2209                                       vlib_cli_command_t * cmd)
2210 {
2211   locator_set_t *lsit;
2212   locator_t *loc;
2213   u32 *locit;
2214   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2215
2216   vlib_cli_output (vm, "%=20s%=16s%=16s%=16s", "Locator-set", "Locator",
2217                    "Priority", "Weight");
2218
2219   /* *INDENT-OFF* */
2220   pool_foreach (lsit, lcm->locator_set_pool,
2221   ({
2222     u8 * msg = 0;
2223     int next_line = 0;
2224     if (lsit->local)
2225       {
2226         msg = format (msg, "%=16v", lsit->name);
2227       }
2228     else
2229       {
2230         msg = format (msg, "%=16s", "remote");
2231       }
2232     vec_foreach (locit, lsit->locator_indices)
2233       {
2234         if (next_line)
2235           {
2236             msg = format (msg, "%16s", " ");
2237           }
2238         loc = pool_elt_at_index (lcm->locator_pool, locit[0]);
2239         if (loc->local)
2240           msg = format (msg, "%16d%16d%16d\n", loc->sw_if_index, loc->priority,
2241                         loc->weight);
2242         else
2243           msg = format (msg, "%16U%16d%16d\n", format_ip_address,
2244                         &gid_address_ip(&loc->address), loc->priority,
2245                         loc->weight);
2246         next_line = 1;
2247       }
2248     vlib_cli_output (vm, "%v", msg);
2249     vec_free (msg);
2250   }));
2251   /* *INDENT-ON* */
2252   return 0;
2253 }
2254
2255 /* *INDENT-OFF* */
2256 VLIB_CLI_COMMAND (lisp_cp_show_locator_sets_command) = {
2257     .path = "show lisp locator-set",
2258     .short_help = "Shows locator-sets",
2259     .function = lisp_cp_show_locator_sets_command_fn,
2260 };
2261 /* *INDENT-ON* */
2262
2263 static map_resolver_t *
2264 get_map_resolver (ip_address_t * a)
2265 {
2266   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2267   map_resolver_t *mr;
2268
2269   vec_foreach (mr, lcm->map_resolvers)
2270   {
2271     if (!ip_address_cmp (&mr->address, a))
2272       {
2273         return mr;
2274       }
2275   }
2276   return 0;
2277 }
2278
2279 int
2280 vnet_lisp_add_del_map_resolver (vnet_lisp_add_del_map_resolver_args_t * a)
2281 {
2282   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2283   u32 i;
2284   map_resolver_t _mr, *mr = &_mr;
2285
2286   if (vnet_lisp_enable_disable_status () == 0)
2287     {
2288       clib_warning ("LISP is disabled!");
2289       return VNET_API_ERROR_LISP_DISABLED;
2290     }
2291
2292   if (a->is_add)
2293     {
2294
2295       if (get_map_resolver (&a->address))
2296         {
2297           clib_warning ("map-resolver %U already exists!", format_ip_address,
2298                         &a->address);
2299           return -1;
2300         }
2301
2302       memset (mr, 0, sizeof (*mr));
2303       ip_address_copy (&mr->address, &a->address);
2304       vec_add1 (lcm->map_resolvers, *mr);
2305
2306       if (vec_len (lcm->map_resolvers) == 1)
2307         lcm->do_map_resolver_election = 1;
2308     }
2309   else
2310     {
2311       for (i = 0; i < vec_len (lcm->map_resolvers); i++)
2312         {
2313           mr = vec_elt_at_index (lcm->map_resolvers, i);
2314           if (!ip_address_cmp (&mr->address, &a->address))
2315             {
2316               if (!ip_address_cmp (&mr->address, &lcm->active_map_resolver))
2317                 lcm->do_map_resolver_election = 1;
2318
2319               vec_del1 (lcm->map_resolvers, i);
2320               break;
2321             }
2322         }
2323     }
2324   return 0;
2325 }
2326
2327 static clib_error_t *
2328 lisp_add_del_map_resolver_command_fn (vlib_main_t * vm,
2329                                       unformat_input_t * input,
2330                                       vlib_cli_command_t * cmd)
2331 {
2332   unformat_input_t _line_input, *line_input = &_line_input;
2333   u8 is_add = 1, addr_set = 0;
2334   ip_address_t ip_addr;
2335   clib_error_t *error = 0;
2336   int rv = 0;
2337   vnet_lisp_add_del_map_resolver_args_t _a, *a = &_a;
2338
2339   /* Get a line of input. */
2340   if (!unformat_user (input, unformat_line_input, line_input))
2341     return 0;
2342
2343   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
2344     {
2345       if (unformat (line_input, "add"))
2346         is_add = 1;
2347       else if (unformat (line_input, "del"))
2348         is_add = 0;
2349       else if (unformat (line_input, "%U", unformat_ip_address, &ip_addr))
2350         addr_set = 1;
2351       else
2352         {
2353           error = unformat_parse_error (line_input);
2354           goto done;
2355         }
2356     }
2357
2358   if (!addr_set)
2359     {
2360       error = clib_error_return (0, "Map-resolver address must be set!");
2361       goto done;
2362     }
2363
2364   a->is_add = is_add;
2365   a->address = ip_addr;
2366   rv = vnet_lisp_add_del_map_resolver (a);
2367   if (0 != rv)
2368     {
2369       error = clib_error_return (0, "failed to %s map-resolver!",
2370                                  is_add ? "add" : "delete");
2371     }
2372
2373 done:
2374   return error;
2375 }
2376
2377 /* *INDENT-OFF* */
2378 VLIB_CLI_COMMAND (lisp_add_del_map_resolver_command) = {
2379     .path = "lisp map-resolver",
2380     .short_help = "lisp map-resolver add/del <ip_address>",
2381     .function = lisp_add_del_map_resolver_command_fn,
2382 };
2383 /* *INDENT-ON* */
2384
2385 int
2386 vnet_lisp_add_del_mreq_itr_rlocs (vnet_lisp_add_del_mreq_itr_rloc_args_t * a)
2387 {
2388   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2389   uword *p = 0;
2390
2391   if (vnet_lisp_enable_disable_status () == 0)
2392     {
2393       clib_warning ("LISP is disabled!");
2394       return VNET_API_ERROR_LISP_DISABLED;
2395     }
2396
2397   if (a->is_add)
2398     {
2399       p = hash_get_mem (lcm->locator_set_index_by_name, a->locator_set_name);
2400       if (!p)
2401         {
2402           clib_warning ("locator-set %v doesn't exist", a->locator_set_name);
2403           return VNET_API_ERROR_INVALID_ARGUMENT;
2404         }
2405
2406       lcm->mreq_itr_rlocs = p[0];
2407     }
2408   else
2409     {
2410       lcm->mreq_itr_rlocs = ~0;
2411     }
2412
2413   return 0;
2414 }
2415
2416 static clib_error_t *
2417 lisp_add_del_mreq_itr_rlocs_command_fn (vlib_main_t * vm,
2418                                         unformat_input_t * input,
2419                                         vlib_cli_command_t * cmd)
2420 {
2421   unformat_input_t _line_input, *line_input = &_line_input;
2422   u8 is_add = 1;
2423   u8 *locator_set_name = 0;
2424   clib_error_t *error = 0;
2425   int rv = 0;
2426   vnet_lisp_add_del_mreq_itr_rloc_args_t _a, *a = &_a;
2427
2428   /* Get a line of input. */
2429   if (!unformat_user (input, unformat_line_input, line_input))
2430     return 0;
2431
2432   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
2433     {
2434       if (unformat (line_input, "del"))
2435         is_add = 0;
2436       else if (unformat (line_input, "add %s", &locator_set_name))
2437         is_add = 1;
2438       else
2439         {
2440           error = unformat_parse_error (line_input);
2441           goto done;
2442         }
2443     }
2444
2445   a->is_add = is_add;
2446   a->locator_set_name = locator_set_name;
2447   rv = vnet_lisp_add_del_mreq_itr_rlocs (a);
2448   if (0 != rv)
2449     {
2450       error = clib_error_return (0, "failed to %s map-request itr-rlocs!",
2451                                  is_add ? "add" : "delete");
2452     }
2453
2454   vec_free (locator_set_name);
2455
2456 done:
2457   return error;
2458
2459 }
2460
2461 /* *INDENT-OFF* */
2462 VLIB_CLI_COMMAND (lisp_add_del_map_request_command) = {
2463     .path = "lisp map-request itr-rlocs",
2464     .short_help = "lisp map-request itr-rlocs add/del <locator_set_name>",
2465     .function = lisp_add_del_mreq_itr_rlocs_command_fn,
2466 };
2467 /* *INDENT-ON* */
2468
2469 static clib_error_t *
2470 lisp_show_mreq_itr_rlocs_command_fn (vlib_main_t * vm,
2471                                      unformat_input_t * input,
2472                                      vlib_cli_command_t * cmd)
2473 {
2474   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2475   locator_set_t *loc_set;
2476
2477   vlib_cli_output (vm, "%=20s", "itr-rlocs");
2478
2479   if (~0 == lcm->mreq_itr_rlocs)
2480     {
2481       return 0;
2482     }
2483
2484   loc_set = pool_elt_at_index (lcm->locator_set_pool, lcm->mreq_itr_rlocs);
2485
2486   vlib_cli_output (vm, "%=20s", loc_set->name);
2487
2488   return 0;
2489 }
2490
2491 /* *INDENT-OFF* */
2492 VLIB_CLI_COMMAND (lisp_show_map_request_command) = {
2493     .path = "show lisp map-request itr-rlocs",
2494     .short_help = "Shows map-request itr-rlocs",
2495     .function = lisp_show_mreq_itr_rlocs_command_fn,
2496 };
2497 /* *INDENT-ON* */
2498
2499 /* Statistics (not really errors) */
2500 #define foreach_lisp_cp_lookup_error           \
2501 _(DROP, "drop")                                \
2502 _(MAP_REQUESTS_SENT, "map-request sent")
2503
2504 static char *lisp_cp_lookup_error_strings[] = {
2505 #define _(sym,string) string,
2506   foreach_lisp_cp_lookup_error
2507 #undef _
2508 };
2509
2510 typedef enum
2511 {
2512 #define _(sym,str) LISP_CP_LOOKUP_ERROR_##sym,
2513   foreach_lisp_cp_lookup_error
2514 #undef _
2515     LISP_CP_LOOKUP_N_ERROR,
2516 } lisp_cp_lookup_error_t;
2517
2518 typedef enum
2519 {
2520   LISP_CP_LOOKUP_NEXT_DROP,
2521   LISP_CP_LOOKUP_NEXT_IP4_LOOKUP,
2522   LISP_CP_LOOKUP_NEXT_IP6_LOOKUP,
2523   LISP_CP_LOOKUP_N_NEXT,
2524 } lisp_cp_lookup_next_t;
2525
2526 typedef struct
2527 {
2528   gid_address_t dst_eid;
2529   ip_address_t map_resolver_ip;
2530 } lisp_cp_lookup_trace_t;
2531
2532 u8 *
2533 format_lisp_cp_lookup_trace (u8 * s, va_list * args)
2534 {
2535   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
2536   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
2537   lisp_cp_lookup_trace_t *t = va_arg (*args, lisp_cp_lookup_trace_t *);
2538
2539   s = format (s, "LISP-CP-LOOKUP: map-resolver: %U destination eid %U",
2540               format_ip_address, &t->map_resolver_ip, format_gid_address,
2541               &t->dst_eid);
2542   return s;
2543 }
2544
2545 int
2546 get_mr_and_local_iface_ip (lisp_cp_main_t * lcm, ip_address_t * mr_ip,
2547                            ip_address_t * sloc)
2548 {
2549   map_resolver_t *mrit;
2550   ip_address_t *a;
2551
2552   if (vec_len (lcm->map_resolvers) == 0)
2553     {
2554       clib_warning ("No map-resolver configured");
2555       return 0;
2556     }
2557
2558   /* find the first mr ip we have a route to and the ip of the
2559    * iface that has a route to it */
2560   vec_foreach (mrit, lcm->map_resolvers)
2561   {
2562     a = &mrit->address;
2563     if (0 != ip_fib_get_first_egress_ip_for_dst (lcm, a, sloc))
2564       {
2565         ip_address_copy (mr_ip, a);
2566
2567         /* also update globals */
2568         return 1;
2569       }
2570   }
2571
2572   clib_warning ("Can't find map-resolver and local interface ip!");
2573   return 0;
2574 }
2575
2576 static gid_address_t *
2577 build_itr_rloc_list (lisp_cp_main_t * lcm, locator_set_t * loc_set)
2578 {
2579   void *addr;
2580   u32 i;
2581   locator_t *loc;
2582   u32 *loc_indexp;
2583   ip_interface_address_t *ia = 0;
2584   gid_address_t gid_data, *gid = &gid_data;
2585   gid_address_t *rlocs = 0;
2586   ip_prefix_t *ippref = &gid_address_ippref (gid);
2587   ip_address_t *rloc = &ip_prefix_addr (ippref);
2588
2589   memset (gid, 0, sizeof (gid[0]));
2590   gid_address_type (gid) = GID_ADDR_IP_PREFIX;
2591   for (i = 0; i < vec_len (loc_set->locator_indices); i++)
2592     {
2593       loc_indexp = vec_elt_at_index (loc_set->locator_indices, i);
2594       loc = pool_elt_at_index (lcm->locator_pool, loc_indexp[0]);
2595
2596       /* Add ipv4 locators first TODO sort them */
2597
2598       /* *INDENT-OFF* */
2599       foreach_ip_interface_address (&lcm->im4->lookup_main, ia,
2600                                     loc->sw_if_index, 1 /* unnumbered */,
2601       ({
2602         addr = ip_interface_address_get_address (&lcm->im4->lookup_main, ia);
2603         ip_address_set (rloc, addr, IP4);
2604         ip_prefix_len (ippref) = 32;
2605         ip_prefix_normalize (ippref);
2606         vec_add1 (rlocs, gid[0]);
2607       }));
2608
2609       /* Add ipv6 locators */
2610       foreach_ip_interface_address (&lcm->im6->lookup_main, ia,
2611                                     loc->sw_if_index, 1 /* unnumbered */,
2612       ({
2613         addr = ip_interface_address_get_address (&lcm->im6->lookup_main, ia);
2614         ip_address_set (rloc, addr, IP6);
2615         ip_prefix_len (ippref) = 128;
2616         ip_prefix_normalize (ippref);
2617         vec_add1 (rlocs, gid[0]);
2618       }));
2619       /* *INDENT-ON* */
2620
2621     }
2622   return rlocs;
2623 }
2624
2625 static vlib_buffer_t *
2626 build_encapsulated_map_request (lisp_cp_main_t * lcm,
2627                                 gid_address_t * seid, gid_address_t * deid,
2628                                 locator_set_t * loc_set, ip_address_t * mr_ip,
2629                                 ip_address_t * sloc, u8 is_smr_invoked,
2630                                 u64 * nonce_res, u32 * bi_res)
2631 {
2632   vlib_buffer_t *b;
2633   u32 bi;
2634   gid_address_t *rlocs = 0;
2635   vlib_main_t *vm = lcm->vlib_main;
2636
2637   if (vlib_buffer_alloc (vm, &bi, 1) != 1)
2638     {
2639       clib_warning ("Can't allocate buffer for Map-Request!");
2640       return 0;
2641     }
2642
2643   b = vlib_get_buffer (vm, bi);
2644
2645   /* leave some space for the encap headers */
2646   vlib_buffer_make_headroom (b, MAX_LISP_MSG_ENCAP_LEN);
2647
2648   /* get rlocs */
2649   rlocs = build_itr_rloc_list (lcm, loc_set);
2650
2651   /* put lisp msg */
2652   lisp_msg_put_mreq (lcm, b, seid, deid, rlocs, is_smr_invoked, nonce_res);
2653
2654   /* push ecm: udp-ip-lisp */
2655   lisp_msg_push_ecm (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, seid, deid);
2656
2657   /* push outer ip header */
2658   pkt_push_udp_and_ip (vm, b, LISP_CONTROL_PORT, LISP_CONTROL_PORT, sloc,
2659                        mr_ip);
2660
2661   bi_res[0] = bi;
2662
2663   vec_free (rlocs);
2664   return b;
2665 }
2666
2667 static void
2668 reset_pending_mr_counters (pending_map_request_t * r)
2669 {
2670   r->time_to_expire = PENDING_MREQ_EXPIRATION_TIME;
2671   r->retries_num = 0;
2672 }
2673
2674 static int
2675 elect_map_resolver (lisp_cp_main_t * lcm)
2676 {
2677   map_resolver_t *mr;
2678
2679   vec_foreach (mr, lcm->map_resolvers)
2680   {
2681     if (!mr->is_down)
2682       {
2683         ip_address_copy (&lcm->active_map_resolver, &mr->address);
2684         lcm->do_map_resolver_election = 0;
2685         return 1;
2686       }
2687   }
2688   return 0;
2689 }
2690
2691 #define send_encapsulated_map_request(lcm, seid, deid, smr) \
2692   _send_encapsulated_map_request(lcm, seid, deid, smr, 0)
2693
2694 #define resend_encapsulated_map_request(lcm, seid, deid, smr) \
2695   _send_encapsulated_map_request(lcm, seid, deid, smr, 1)
2696
2697 static int
2698 _send_encapsulated_map_request (lisp_cp_main_t * lcm,
2699                                 gid_address_t * seid, gid_address_t * deid,
2700                                 u8 is_smr_invoked, u8 is_resend)
2701 {
2702   map_resolver_t *mr;
2703   u32 next_index, bi = 0, *to_next, map_index;
2704   vlib_buffer_t *b;
2705   vlib_frame_t *f;
2706   u64 nonce = 0;
2707   locator_set_t *loc_set;
2708   mapping_t *map;
2709   pending_map_request_t *pmr, *duplicate_pmr = 0;
2710   ip_address_t sloc;
2711   u32 ls_index;
2712
2713   ASSERT (*lcm->pending_map_request_lock);
2714
2715   /* if there is already a pending request remember it */
2716
2717   /* *INDENT-OFF* */
2718   pool_foreach(pmr, lcm->pending_map_requests_pool,
2719   ({
2720     if (!gid_address_cmp (&pmr->src, seid)
2721         && !gid_address_cmp (&pmr->dst, deid))
2722       {
2723         duplicate_pmr = pmr;
2724         break;
2725       }
2726   }));
2727   /* *INDENT-ON* */
2728
2729   if (!is_resend && duplicate_pmr)
2730     {
2731       /* don't send the request if there is a pending map request already */
2732       return 0;
2733     }
2734
2735   /* get locator-set for seid */
2736   if (!lcm->lisp_pitr)
2737     {
2738       map_index = gid_dictionary_lookup (&lcm->mapping_index_by_gid, seid);
2739       if (map_index == ~0)
2740         {
2741           clib_warning ("No local mapping found in eid-table for %U!",
2742                         format_gid_address, seid);
2743           return -1;
2744         }
2745
2746       map = pool_elt_at_index (lcm->mapping_pool, map_index);
2747
2748       if (!map->local)
2749         {
2750           clib_warning
2751             ("Mapping found for src eid %U is not marked as local!",
2752              format_gid_address, seid);
2753           return -1;
2754         }
2755       ls_index = map->locator_set_index;
2756     }
2757   else
2758     {
2759       map_index = lcm->pitr_map_index;
2760       map = pool_elt_at_index (lcm->mapping_pool, lcm->pitr_map_index);
2761       ls_index = map->locator_set_index;
2762     }
2763
2764   /* overwrite locator set if map-request itr-rlocs configured */
2765   if (~0 != lcm->mreq_itr_rlocs)
2766     {
2767       ls_index = lcm->mreq_itr_rlocs;
2768     }
2769
2770   loc_set = pool_elt_at_index (lcm->locator_set_pool, ls_index);
2771
2772   while (lcm->do_map_resolver_election
2773          | (0 == ip_fib_get_first_egress_ip_for_dst (lcm,
2774                                                      &lcm->active_map_resolver,
2775                                                      &sloc)))
2776     {
2777       if (0 == elect_map_resolver (lcm))
2778         /* all Mrs are down */
2779         {
2780           if (duplicate_pmr)
2781             duplicate_pmr->to_be_removed = 1;
2782
2783           /* restart MR checking by marking all of them up */
2784           vec_foreach (mr, lcm->map_resolvers) mr->is_down = 0;
2785
2786           return -1;
2787         }
2788     }
2789
2790   /* build the encapsulated map request */
2791   b = build_encapsulated_map_request (lcm, seid, deid, loc_set,
2792                                       &lcm->active_map_resolver,
2793                                       &sloc, is_smr_invoked, &nonce, &bi);
2794
2795   if (!b)
2796     return -1;
2797
2798   /* set fib index to default and lookup node */
2799   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
2800   next_index = (ip_addr_version (&lcm->active_map_resolver) == IP4) ?
2801     ip4_lookup_node.index : ip6_lookup_node.index;
2802
2803   f = vlib_get_frame_to_node (lcm->vlib_main, next_index);
2804
2805   /* Enqueue the packet */
2806   to_next = vlib_frame_vector_args (f);
2807   to_next[0] = bi;
2808   f->n_vectors = 1;
2809   vlib_put_frame_to_node (lcm->vlib_main, next_index, f);
2810
2811   if (duplicate_pmr)
2812     /* if there is a pending request already update it */
2813     {
2814       if (clib_fifo_elts (duplicate_pmr->nonces) >= PENDING_MREQ_QUEUE_LEN)
2815         {
2816           /* remove the oldest nonce */
2817           u64 CLIB_UNUSED (tmp), *nonce_del;
2818           nonce_del = clib_fifo_head (duplicate_pmr->nonces);
2819           hash_unset (lcm->pending_map_requests_by_nonce, nonce_del[0]);
2820           clib_fifo_sub1 (duplicate_pmr->nonces, tmp);
2821         }
2822
2823       clib_fifo_add1 (duplicate_pmr->nonces, nonce);
2824       hash_set (lcm->pending_map_requests_by_nonce, nonce,
2825                 duplicate_pmr - lcm->pending_map_requests_pool);
2826     }
2827   else
2828     {
2829       /* add map-request to pending requests table */
2830       pool_get (lcm->pending_map_requests_pool, pmr);
2831       memset (pmr, 0, sizeof (*pmr));
2832       gid_address_copy (&pmr->src, seid);
2833       gid_address_copy (&pmr->dst, deid);
2834       clib_fifo_add1 (pmr->nonces, nonce);
2835       pmr->is_smr_invoked = is_smr_invoked;
2836       reset_pending_mr_counters (pmr);
2837       hash_set (lcm->pending_map_requests_by_nonce, nonce,
2838                 pmr - lcm->pending_map_requests_pool);
2839     }
2840
2841   return 0;
2842 }
2843
2844 static void
2845 get_src_and_dst_ip (void *hdr, ip_address_t * src, ip_address_t * dst)
2846 {
2847   ip4_header_t *ip4 = hdr;
2848   ip6_header_t *ip6;
2849
2850   if ((ip4->ip_version_and_header_length & 0xF0) == 0x40)
2851     {
2852       ip_address_set (src, &ip4->src_address, IP4);
2853       ip_address_set (dst, &ip4->dst_address, IP4);
2854     }
2855   else
2856     {
2857       ip6 = hdr;
2858       ip_address_set (src, &ip6->src_address, IP6);
2859       ip_address_set (dst, &ip6->dst_address, IP6);
2860     }
2861 }
2862
2863 static u32
2864 lisp_get_vni_from_buffer_ip (lisp_cp_main_t * lcm, vlib_buffer_t * b,
2865                              u8 version)
2866 {
2867   uword *vnip;
2868   u32 vni = ~0, table_id = ~0, fib_index;
2869
2870   if (version == IP4)
2871     {
2872       ip4_fib_t *fib;
2873       ip4_main_t *im4 = &ip4_main;
2874       fib_index = vec_elt (im4->fib_index_by_sw_if_index,
2875                            vnet_buffer (b)->sw_if_index[VLIB_RX]);
2876       fib = find_ip4_fib_by_table_index_or_id (im4, fib_index,
2877                                                IP4_ROUTE_FLAG_FIB_INDEX);
2878       table_id = fib->table_id;
2879     }
2880   else
2881     {
2882       ip6_fib_t *fib;
2883       ip6_main_t *im6 = &ip6_main;
2884       fib_index = vec_elt (im6->fib_index_by_sw_if_index,
2885                            vnet_buffer (b)->sw_if_index[VLIB_RX]);
2886       fib = find_ip6_fib_by_table_index_or_id (im6, fib_index,
2887                                                IP6_ROUTE_FLAG_FIB_INDEX);
2888       table_id = fib->table_id;
2889     }
2890
2891   vnip = hash_get (lcm->vni_by_table_id, table_id);
2892   if (vnip)
2893     vni = vnip[0];
2894   else
2895     clib_warning ("vrf %d is not mapped to any vni!", table_id);
2896
2897   return vni;
2898 }
2899
2900 always_inline u32
2901 lisp_get_vni_from_buffer_eth (lisp_cp_main_t * lcm, vlib_buffer_t * b)
2902 {
2903   uword *vnip;
2904   u32 vni = ~0;
2905   u32 sw_if_index0;
2906
2907   l2input_main_t *l2im = &l2input_main;
2908   l2_input_config_t *config;
2909   l2_bridge_domain_t *bd_config;
2910
2911   sw_if_index0 = vnet_buffer (b)->sw_if_index[VLIB_RX];
2912   config = vec_elt_at_index (l2im->configs, sw_if_index0);
2913   bd_config = vec_elt_at_index (l2im->bd_configs, config->bd_index);
2914
2915   vnip = hash_get (lcm->vni_by_bd_id, bd_config->bd_id);
2916   if (vnip)
2917     vni = vnip[0];
2918   else
2919     clib_warning ("bridge domain %d is not mapped to any vni!",
2920                   config->bd_index);
2921
2922   return vni;
2923 }
2924
2925 always_inline void
2926 get_src_and_dst_eids_from_buffer (lisp_cp_main_t * lcm, vlib_buffer_t * b,
2927                                   gid_address_t * src, gid_address_t * dst)
2928 {
2929   u32 vni = 0;
2930   u16 type;
2931
2932   memset (src, 0, sizeof (*src));
2933   memset (dst, 0, sizeof (*dst));
2934   type = vnet_buffer (b)->lisp.overlay_afi;
2935
2936   if (LISP_AFI_IP == type || LISP_AFI_IP6 == type)
2937     {
2938       ip4_header_t *ip;
2939       u8 version, preflen;
2940
2941       gid_address_type (src) = GID_ADDR_IP_PREFIX;
2942       gid_address_type (dst) = GID_ADDR_IP_PREFIX;
2943
2944       ip = vlib_buffer_get_current (b);
2945       get_src_and_dst_ip (ip, &gid_address_ip (src), &gid_address_ip (dst));
2946
2947       version = gid_address_ip_version (src);
2948       preflen = ip_address_max_len (version);
2949       gid_address_ippref_len (src) = preflen;
2950       gid_address_ippref_len (dst) = preflen;
2951
2952       vni = lisp_get_vni_from_buffer_ip (lcm, b, version);
2953       gid_address_vni (dst) = vni;
2954       gid_address_vni (src) = vni;
2955     }
2956   else if (LISP_AFI_MAC == type)
2957     {
2958       ethernet_header_t *eh;
2959
2960       eh = vlib_buffer_get_current (b);
2961
2962       gid_address_type (src) = GID_ADDR_MAC;
2963       gid_address_type (dst) = GID_ADDR_MAC;
2964       mac_copy (&gid_address_mac (src), eh->src_address);
2965       mac_copy (&gid_address_mac (dst), eh->dst_address);
2966
2967       /* get vni */
2968       vni = lisp_get_vni_from_buffer_eth (lcm, b);
2969
2970       gid_address_vni (dst) = vni;
2971       gid_address_vni (src) = vni;
2972     }
2973 }
2974
2975 static uword
2976 lisp_cp_lookup (vlib_main_t * vm, vlib_node_runtime_t * node,
2977                 vlib_frame_t * from_frame)
2978 {
2979   u32 *from, *to_next_drop, di, si;
2980   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
2981   u32 pkts_mapped = 0;
2982   uword n_left_from, n_left_to_next_drop;
2983
2984   from = vlib_frame_vector_args (from_frame);
2985   n_left_from = from_frame->n_vectors;
2986
2987   while (n_left_from > 0)
2988     {
2989       vlib_get_next_frame (vm, node, LISP_CP_LOOKUP_NEXT_DROP,
2990                            to_next_drop, n_left_to_next_drop);
2991
2992       while (n_left_from > 0 && n_left_to_next_drop > 0)
2993         {
2994           u32 pi0;
2995           vlib_buffer_t *b0;
2996           gid_address_t src, dst;
2997
2998           pi0 = from[0];
2999           from += 1;
3000           n_left_from -= 1;
3001           to_next_drop[0] = pi0;
3002           to_next_drop += 1;
3003           n_left_to_next_drop -= 1;
3004
3005           b0 = vlib_get_buffer (vm, pi0);
3006           b0->error = node->errors[LISP_CP_LOOKUP_ERROR_DROP];
3007
3008           /* src/dst eid pair */
3009           get_src_and_dst_eids_from_buffer (lcm, b0, &src, &dst);
3010
3011           /* if we have remote mapping for destination already in map-chache
3012              add forwarding tunnel directly. If not send a map-request */
3013           di = gid_dictionary_lookup (&lcm->mapping_index_by_gid, &dst);
3014           if (~0 != di)
3015             {
3016               mapping_t *m = vec_elt_at_index (lcm->mapping_pool, di);
3017               /* send a map-request also in case of negative mapping entry
3018                  with corresponding action */
3019               if (m->action == LISP_SEND_MAP_REQUEST)
3020                 {
3021                   /* send map-request */
3022                   queue_map_request (&src, &dst, 0 /* smr_invoked */ ,
3023                                      0 /* is_resend */ );
3024                   pkts_mapped++;
3025                 }
3026               else
3027                 {
3028                   si = gid_dictionary_lookup (&lcm->mapping_index_by_gid,
3029                                               &src);
3030                   if (~0 != si)
3031                     {
3032                       dp_add_fwd_entry (lcm, si, di);
3033                     }
3034                 }
3035             }
3036           else
3037             {
3038               /* send map-request */
3039               queue_map_request (&src, &dst, 0 /* smr_invoked */ ,
3040                                  0 /* is_resend */ );
3041               pkts_mapped++;
3042             }
3043
3044           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
3045             {
3046               lisp_cp_lookup_trace_t *tr = vlib_add_trace (vm, node, b0,
3047                                                            sizeof (*tr));
3048
3049               memset (tr, 0, sizeof (*tr));
3050               gid_address_copy (&tr->dst_eid, &dst);
3051               if (vec_len (lcm->map_resolvers) > 0)
3052                 {
3053                   clib_memcpy (&tr->map_resolver_ip,
3054                                vec_elt_at_index (lcm->map_resolvers, 0),
3055                                sizeof (ip_address_t));
3056                 }
3057             }
3058           gid_address_free (&dst);
3059           gid_address_free (&src);
3060         }
3061
3062       vlib_put_next_frame (vm, node, LISP_CP_LOOKUP_NEXT_DROP,
3063                            n_left_to_next_drop);
3064     }
3065   vlib_node_increment_counter (vm, node->node_index,
3066                                LISP_CP_LOOKUP_ERROR_MAP_REQUESTS_SENT,
3067                                pkts_mapped);
3068   return from_frame->n_vectors;
3069 }
3070
3071 /* *INDENT-OFF* */
3072 VLIB_REGISTER_NODE (lisp_cp_lookup_node) = {
3073   .function = lisp_cp_lookup,
3074   .name = "lisp-cp-lookup",
3075   .vector_size = sizeof (u32),
3076   .format_trace = format_lisp_cp_lookup_trace,
3077   .type = VLIB_NODE_TYPE_INTERNAL,
3078
3079   .n_errors = LISP_CP_LOOKUP_N_ERROR,
3080   .error_strings = lisp_cp_lookup_error_strings,
3081
3082   .n_next_nodes = LISP_CP_LOOKUP_N_NEXT,
3083
3084   .next_nodes = {
3085       [LISP_CP_LOOKUP_NEXT_DROP] = "error-drop",
3086       [LISP_CP_LOOKUP_NEXT_IP4_LOOKUP] = "ip4-lookup",
3087       [LISP_CP_LOOKUP_NEXT_IP6_LOOKUP] = "ip6-lookup",
3088   },
3089 };
3090 /* *INDENT-ON* */
3091
3092 /* lisp_cp_input statistics */
3093 #define foreach_lisp_cp_input_error                     \
3094 _(DROP, "drop")                                         \
3095 _(MAP_REPLIES_RECEIVED, "map-replies received")
3096
3097 static char *lisp_cp_input_error_strings[] = {
3098 #define _(sym,string) string,
3099   foreach_lisp_cp_input_error
3100 #undef _
3101 };
3102
3103 typedef enum
3104 {
3105 #define _(sym,str) LISP_CP_INPUT_ERROR_##sym,
3106   foreach_lisp_cp_input_error
3107 #undef _
3108     LISP_CP_INPUT_N_ERROR,
3109 } lisp_cp_input_error_t;
3110
3111 typedef enum
3112 {
3113   LISP_CP_INPUT_NEXT_DROP,
3114   LISP_CP_INPUT_N_NEXT,
3115 } lisp_cp_input_next_t;
3116
3117 typedef struct
3118 {
3119   gid_address_t dst_eid;
3120   ip4_address_t map_resolver_ip;
3121 } lisp_cp_input_trace_t;
3122
3123 u8 *
3124 format_lisp_cp_input_trace (u8 * s, va_list * args)
3125 {
3126   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
3127   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
3128   CLIB_UNUSED (lisp_cp_input_trace_t * t) =
3129     va_arg (*args, lisp_cp_input_trace_t *);
3130
3131   s = format (s, "LISP-CP-INPUT: TODO");
3132   return s;
3133 }
3134
3135 void *
3136 process_map_reply (void *arg)
3137 {
3138   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3139   vlib_buffer_t *b = arg;
3140   u32 len = 0, i, ttl, dst_map_index = 0;
3141   void *h;
3142   pending_map_request_t *pmr;
3143   locator_t probed;
3144   map_reply_hdr_t *mrep_hdr;
3145   u64 nonce, *noncep;
3146   gid_address_t deid;
3147   uword *pmr_index;
3148   u8 authoritative, action;
3149   locator_t *locators = 0, *loc;
3150
3151   mrep_hdr = vlib_buffer_get_current (b);
3152
3153   lisp_pending_map_request_lock (lcm);
3154
3155   /* Check pending requests table and nonce */
3156   nonce = MREP_NONCE (mrep_hdr);
3157   pmr_index = hash_get (lcm->pending_map_requests_by_nonce, nonce);
3158   if (!pmr_index)
3159     {
3160       clib_warning ("No pending map-request entry with nonce %lu!", nonce);
3161       goto done;
3162     }
3163   pmr = pool_elt_at_index (lcm->pending_map_requests_pool, pmr_index[0]);
3164
3165   vlib_buffer_pull (b, sizeof (*mrep_hdr));
3166
3167   for (i = 0; i < MREP_REC_COUNT (mrep_hdr); i++)
3168     {
3169       h = vlib_buffer_get_current (b);
3170       ttl = clib_net_to_host_u32 (MAP_REC_TTL (h));
3171       action = MAP_REC_ACTION (h);
3172       authoritative = MAP_REC_AUTH (h);
3173
3174       len = lisp_msg_parse_mapping_record (b, &deid, &locators, &probed);
3175       if (len == ~0)
3176         {
3177           clib_warning ("Failed to parse mapping record!");
3178           vec_foreach (loc, locators)
3179           {
3180             locator_free (loc);
3181           }
3182           vec_free (locators);
3183           goto done;
3184         }
3185
3186       /* insert/update mappings cache */
3187       vnet_lisp_add_del_mapping (&deid, locators, action, authoritative, ttl,
3188                                  1, 0 /* is_static */ , &dst_map_index);
3189
3190       /* try to program forwarding only if mapping saved or updated */
3191       if ((u32) ~ 0 != dst_map_index)
3192         lisp_add_del_adjacency (lcm, &pmr->src, &deid, 1);
3193
3194       vec_free (locators);
3195     }
3196
3197   /* remove pending map request entry */
3198
3199   /* *INDENT-OFF* */
3200   clib_fifo_foreach (noncep, pmr->nonces, ({
3201     hash_unset(lcm->pending_map_requests_by_nonce, noncep[0]);
3202   }));
3203   /* *INDENT-ON* */
3204
3205   clib_fifo_free (pmr->nonces);
3206   pool_put (lcm->pending_map_requests_pool, pmr);
3207
3208 done:
3209   lisp_pending_map_request_unlock (lcm);
3210   return 0;
3211 }
3212
3213 void
3214 process_map_request (vlib_main_t * vm, lisp_cp_main_t * lcm,
3215                      vlib_buffer_t * b)
3216 {
3217   map_request_hdr_t *mreq_hdr;
3218   gid_address_t src, dst;
3219 //  u64 nonce;
3220   u32 i, len = 0;
3221   gid_address_t *itr_rlocs = 0, *rloc;
3222
3223   mreq_hdr = vlib_buffer_get_current (b);
3224   vlib_buffer_pull (b, sizeof (*mreq_hdr));
3225
3226 //  nonce = MREQ_NONCE(mreq_hdr);
3227
3228   if (!MREQ_SMR (mreq_hdr))
3229     {
3230       clib_warning ("Only SMR Map-Requests supported for now!");
3231       return;
3232     }
3233
3234   /* parse src eid */
3235   len = lisp_msg_parse_addr (b, &src);
3236   if (len == ~0)
3237     return;
3238
3239   /* for now we don't do anything with the itr's rlocs */
3240   len =
3241     lisp_msg_parse_itr_rlocs (b, &itr_rlocs,
3242                               MREQ_ITR_RLOC_COUNT (mreq_hdr) + 1);
3243   if (len == ~0)
3244     return;
3245
3246   /* TODO: RLOCs are currently unused, so free them for now */
3247   vec_foreach (rloc, itr_rlocs)
3248   {
3249     gid_address_free (rloc);
3250   }
3251
3252   /* parse eid records and send SMR-invoked map-requests */
3253   for (i = 0; i < MREQ_REC_COUNT (mreq_hdr); i++)
3254     {
3255       memset (&dst, 0, sizeof (dst));
3256       len = lisp_msg_parse_eid_rec (b, &dst);
3257       if (len == ~0)
3258         {
3259           clib_warning ("Can't parse map-request EID-record");
3260           return;
3261         }
3262       /* send SMR-invoked map-requests */
3263       queue_map_request (&dst, &src, 1 /* invoked */ , 0 /* resend */ );
3264     }
3265 }
3266
3267 static void
3268 queue_map_reply (vlib_buffer_t * b)
3269 {
3270   vlib_buffer_t *a = clib_mem_alloc (sizeof (a[0]) + b->current_length);
3271
3272   clib_memcpy (a->data, b->data + b->current_data, b->current_length);
3273   a->current_length = b->current_length;
3274   a->current_data = 0;
3275
3276   vl_api_rpc_call_main_thread (process_map_reply, (u8 *) a, sizeof (a[0])
3277                                + a->current_length);
3278   clib_mem_free (a);
3279 }
3280
3281 static uword
3282 lisp_cp_input (vlib_main_t * vm, vlib_node_runtime_t * node,
3283                vlib_frame_t * from_frame)
3284 {
3285   u32 n_left_from, *from, *to_next_drop;
3286   lisp_msg_type_e type;
3287   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3288
3289   from = vlib_frame_vector_args (from_frame);
3290   n_left_from = from_frame->n_vectors;
3291
3292
3293   while (n_left_from > 0)
3294     {
3295       u32 n_left_to_next_drop;
3296
3297       vlib_get_next_frame (vm, node, LISP_CP_INPUT_NEXT_DROP,
3298                            to_next_drop, n_left_to_next_drop);
3299       while (n_left_from > 0 && n_left_to_next_drop > 0)
3300         {
3301           u32 bi0;
3302           vlib_buffer_t *b0;
3303
3304           bi0 = from[0];
3305           from += 1;
3306           n_left_from -= 1;
3307           to_next_drop[0] = bi0;
3308           to_next_drop += 1;
3309           n_left_to_next_drop -= 1;
3310
3311           b0 = vlib_get_buffer (vm, bi0);
3312
3313           type = lisp_msg_type (vlib_buffer_get_current (b0));
3314           switch (type)
3315             {
3316             case LISP_MAP_REPLY:
3317               queue_map_reply (b0);
3318               break;
3319             case LISP_MAP_REQUEST:
3320               process_map_request (vm, lcm, b0);
3321               break;
3322             default:
3323               clib_warning ("Unsupported LISP message type %d", type);
3324               break;
3325             }
3326
3327           b0->error = node->errors[LISP_CP_INPUT_ERROR_DROP];
3328
3329           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
3330             {
3331
3332             }
3333         }
3334
3335       vlib_put_next_frame (vm, node, LISP_CP_INPUT_NEXT_DROP,
3336                            n_left_to_next_drop);
3337     }
3338   return from_frame->n_vectors;
3339 }
3340
3341 /* *INDENT-OFF* */
3342 VLIB_REGISTER_NODE (lisp_cp_input_node) = {
3343   .function = lisp_cp_input,
3344   .name = "lisp-cp-input",
3345   .vector_size = sizeof (u32),
3346   .format_trace = format_lisp_cp_input_trace,
3347   .type = VLIB_NODE_TYPE_INTERNAL,
3348
3349   .n_errors = LISP_CP_INPUT_N_ERROR,
3350   .error_strings = lisp_cp_input_error_strings,
3351
3352   .n_next_nodes = LISP_CP_INPUT_N_NEXT,
3353
3354   .next_nodes = {
3355       [LISP_CP_INPUT_NEXT_DROP] = "error-drop",
3356   },
3357 };
3358 /* *INDENT-ON* */
3359
3360 clib_error_t *
3361 lisp_cp_init (vlib_main_t * vm)
3362 {
3363   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3364   clib_error_t *error = 0;
3365
3366   if ((error = vlib_call_init_function (vm, lisp_gpe_init)))
3367     return error;
3368
3369   lcm->im4 = &ip4_main;
3370   lcm->im6 = &ip6_main;
3371   lcm->vlib_main = vm;
3372   lcm->vnet_main = vnet_get_main ();
3373   lcm->mreq_itr_rlocs = ~0;
3374   lcm->lisp_pitr = 0;
3375
3376   lcm->pending_map_request_lock =
3377     clib_mem_alloc_aligned (CLIB_CACHE_LINE_BYTES, CLIB_CACHE_LINE_BYTES);
3378
3379   lcm->pending_map_request_lock[0] = 0;
3380   gid_dictionary_init (&lcm->mapping_index_by_gid);
3381   lcm->do_map_resolver_election = 1;
3382
3383   /* default vrf mapped to vni 0 */
3384   hash_set (lcm->table_id_by_vni, 0, 0);
3385   hash_set (lcm->vni_by_table_id, 0, 0);
3386
3387   udp_register_dst_port (vm, UDP_DST_PORT_lisp_cp,
3388                          lisp_cp_input_node.index, 1 /* is_ip4 */ );
3389   udp_register_dst_port (vm, UDP_DST_PORT_lisp_cp6,
3390                          lisp_cp_input_node.index, 0 /* is_ip4 */ );
3391
3392   return 0;
3393 }
3394
3395 static void *
3396 send_map_request_thread_fn (void *arg)
3397 {
3398   map_request_args_t *a = arg;
3399   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3400
3401   lisp_pending_map_request_lock (lcm);
3402
3403   if (a->is_resend)
3404     resend_encapsulated_map_request (lcm, &a->seid, &a->deid, a->smr_invoked);
3405   else
3406     send_encapsulated_map_request (lcm, &a->seid, &a->deid, a->smr_invoked);
3407
3408   lisp_pending_map_request_unlock (lcm);
3409
3410   return 0;
3411 }
3412
3413 static int
3414 queue_map_request (gid_address_t * seid, gid_address_t * deid,
3415                    u8 smr_invoked, u8 is_resend)
3416 {
3417   map_request_args_t a;
3418
3419   a.is_resend = is_resend;
3420   gid_address_copy (&a.seid, seid);
3421   gid_address_copy (&a.deid, deid);
3422   a.smr_invoked = smr_invoked;
3423
3424   vl_api_rpc_call_main_thread (send_map_request_thread_fn,
3425                                (u8 *) & a, sizeof (a));
3426   return 0;
3427 }
3428
3429 /**
3430  * Take an action with a pending map request depending on expiration time
3431  * and re-try counters.
3432  */
3433 static void
3434 update_pending_request (pending_map_request_t * r, f64 dt)
3435 {
3436   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3437   map_resolver_t *mr;
3438
3439   if (r->time_to_expire - dt < 0)
3440     /* it's time to decide what to do with this pending request */
3441     {
3442       if (r->retries_num >= NUMBER_OF_RETRIES)
3443         /* too many retries -> assume current map resolver is not available */
3444         {
3445           mr = get_map_resolver (&lcm->active_map_resolver);
3446           if (!mr)
3447             {
3448               clib_warning ("Map resolver %U not found - probably deleted "
3449                             "by the user recently.", format_ip_address,
3450                             &lcm->active_map_resolver);
3451             }
3452           else
3453             {
3454               clib_warning ("map resolver %U is unreachable, ignoring",
3455                             format_ip_address, &lcm->active_map_resolver);
3456
3457               /* mark current map resolver unavailable so it won't be
3458                * selected next time */
3459               mr->is_down = 1;
3460               mr->last_update = vlib_time_now (lcm->vlib_main);
3461             }
3462
3463           reset_pending_mr_counters (r);
3464           elect_map_resolver (lcm);
3465
3466           /* try to find a next eligible map resolver and re-send */
3467           queue_map_request (&r->src, &r->dst, r->is_smr_invoked,
3468                              1 /* resend */ );
3469         }
3470       else
3471         {
3472           /* try again */
3473           queue_map_request (&r->src, &r->dst, r->is_smr_invoked,
3474                              1 /* resend */ );
3475           r->retries_num++;
3476           r->time_to_expire = PENDING_MREQ_EXPIRATION_TIME;
3477         }
3478     }
3479   else
3480     r->time_to_expire -= dt;
3481 }
3482
3483 static void
3484 remove_dead_pending_map_requests (lisp_cp_main_t * lcm)
3485 {
3486   u64 *nonce;
3487   pending_map_request_t *pmr;
3488   u32 *to_be_removed = 0, *pmr_index;
3489
3490   ASSERT (*lcm->pending_map_request_lock);
3491
3492   /* *INDENT-OFF* */
3493   pool_foreach (pmr, lcm->pending_map_requests_pool,
3494   ({
3495     if (pmr->to_be_removed)
3496       {
3497         clib_fifo_foreach (nonce, pmr->nonces, ({
3498           hash_unset (lcm->pending_map_requests_by_nonce, nonce[0]);
3499         }));
3500
3501         vec_add1 (to_be_removed, pmr - lcm->pending_map_requests_pool);
3502       }
3503   }));
3504   /* *INDENT-ON* */
3505
3506   vec_foreach (pmr_index, to_be_removed)
3507     pool_put_index (lcm->pending_map_requests_by_nonce, pmr_index[0]);
3508
3509   vec_free (to_be_removed);
3510 }
3511
3512 static uword
3513 send_map_resolver_service (vlib_main_t * vm,
3514                            vlib_node_runtime_t * rt, vlib_frame_t * f)
3515 {
3516   f64 period = 2.0;
3517   pending_map_request_t *pmr;
3518   lisp_cp_main_t *lcm = vnet_lisp_cp_get_main ();
3519
3520   while (1)
3521     {
3522       vlib_process_wait_for_event_or_clock (vm, period);
3523
3524       /* currently no signals are expected - just wait for clock */
3525       (void) vlib_process_get_events (vm, 0);
3526
3527       lisp_pending_map_request_lock (lcm);
3528
3529       /* *INDENT-OFF* */
3530       pool_foreach (pmr, lcm->pending_map_requests_pool,
3531       ({
3532         if (!pmr->to_be_removed)
3533           update_pending_request (pmr, period);
3534       }));
3535       /* *INDENT-ON* */
3536
3537       remove_dead_pending_map_requests (lcm);
3538
3539       lisp_pending_map_request_unlock (lcm);
3540     }
3541
3542   /* unreachable */
3543   return 0;
3544 }
3545
3546 /* *INDENT-OFF* */
3547 VLIB_REGISTER_NODE (lisp_retry_service_node,static) = {
3548     .function = send_map_resolver_service,
3549     .type = VLIB_NODE_TYPE_PROCESS,
3550     .name = "lisp-retry-service",
3551     .process_log2_n_stack_bytes = 16,
3552 };
3553 /* *INDENT-ON* */
3554
3555 VLIB_INIT_FUNCTION (lisp_cp_init);
3556
3557 /*
3558  * fd.io coding-style-patch-verification: ON
3559  *
3560  * Local Variables:
3561  * eval: (c-set-style "gnu")
3562  * End:
3563  */