d246f6c82546e89046ca85a9423aec03a68da71c
[vpp.git] / src / vnet / map / map.c
1 /*
2  * map.c : MAP support
3  *
4  * Copyright (c) 2015 Cisco and/or its affiliates.
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 #include <vnet/fib/fib_table.h>
19 #include <vnet/fib/ip6_fib.h>
20 #include <vnet/adj/adj.h>
21 #include <vnet/map/map_dpo.h>
22 #include <vppinfra/crc32.h>
23
24 #include "map.h"
25
26 /*
27  * This code supports the following MAP modes:
28  *
29  * Algorithmic Shared IPv4 address (ea_bits_len > 0):
30  *   ea_bits_len + ip4_prefix > 32
31  *   psid_length > 0, ip6_prefix < 64, ip4_prefix <= 32
32  * Algorithmic Full IPv4 address (ea_bits_len > 0):
33  *   ea_bits_len + ip4_prefix = 32
34  *   psid_length = 0, ip6_prefix < 64, ip4_prefix <= 32
35  * Algorithmic IPv4 prefix (ea_bits_len > 0):
36  *   ea_bits_len + ip4_prefix < 32
37  *   psid_length = 0, ip6_prefix < 64, ip4_prefix <= 32
38  *
39  * Independent Shared IPv4 address (ea_bits_len = 0):
40  *   ip4_prefix = 32
41  *   psid_length > 0
42  *   Rule IPv6 address = 128, Rule PSID Set
43  * Independent Full IPv4 address (ea_bits_len = 0):
44  *   ip4_prefix = 32
45  *   psid_length = 0, ip6_prefix = 128
46  * Independent IPv4 prefix (ea_bits_len = 0):
47  *   ip4_prefix < 32
48  *   psid_length = 0, ip6_prefix = 128
49  *
50  */
51
52 /*
53  * This code supports MAP-T:
54  *
55  * With DMR prefix length equal to 96.
56  *
57  */
58
59
60
61 int
62 map_create_domain (ip4_address_t * ip4_prefix,
63                    u8 ip4_prefix_len,
64                    ip6_address_t * ip6_prefix,
65                    u8 ip6_prefix_len,
66                    ip6_address_t * ip6_src,
67                    u8 ip6_src_len,
68                    u8 ea_bits_len,
69                    u8 psid_offset,
70                    u8 psid_length, u32 * map_domain_index, u16 mtu, u8 flags)
71 {
72   u8 suffix_len, suffix_shift;
73   map_main_t *mm = &map_main;
74   dpo_id_t dpo_v4 = DPO_INVALID;
75   dpo_id_t dpo_v6 = DPO_INVALID;
76   map_domain_t *d;
77
78   /* Sanity check on the src prefix length */
79   if (flags & MAP_DOMAIN_TRANSLATION)
80     {
81       if (ip6_src_len != 96)
82         {
83           clib_warning ("MAP-T only supports ip6_src_len = 96 for now.");
84           return -1;
85         }
86     }
87   else
88     {
89       if (ip6_src_len != 128)
90         {
91           clib_warning
92             ("MAP-E requires a BR address, not a prefix (ip6_src_len should "
93              "be 128).");
94           return -1;
95         }
96     }
97
98   /* How many, and which bits to grab from the IPv4 DA */
99   if (ip4_prefix_len + ea_bits_len < 32)
100     {
101       flags |= MAP_DOMAIN_PREFIX;
102       suffix_shift = 32 - ip4_prefix_len - ea_bits_len;
103       suffix_len = ea_bits_len;
104     }
105   else
106     {
107       suffix_shift = 0;
108       suffix_len = 32 - ip4_prefix_len;
109     }
110
111   /* EA bits must be within the first 64 bits */
112   if (ea_bits_len > 0 && ((ip6_prefix_len + ea_bits_len) > 64 ||
113                           ip6_prefix_len + suffix_len + psid_length > 64))
114     {
115       clib_warning
116         ("Embedded Address bits must be within the first 64 bits of "
117          "the IPv6 prefix");
118       return -1;
119     }
120
121   /* Get domain index */
122   pool_get_aligned (mm->domains, d, CLIB_CACHE_LINE_BYTES);
123   memset (d, 0, sizeof (*d));
124   *map_domain_index = d - mm->domains;
125
126   /* Init domain struct */
127   d->ip4_prefix.as_u32 = ip4_prefix->as_u32;
128   d->ip4_prefix_len = ip4_prefix_len;
129   d->ip6_prefix = *ip6_prefix;
130   d->ip6_prefix_len = ip6_prefix_len;
131   d->ip6_src = *ip6_src;
132   d->ip6_src_len = ip6_src_len;
133   d->ea_bits_len = ea_bits_len;
134   d->psid_offset = psid_offset;
135   d->psid_length = psid_length;
136   d->mtu = mtu;
137   d->flags = flags;
138   d->suffix_shift = suffix_shift;
139   d->suffix_mask = (1 << suffix_len) - 1;
140
141   d->psid_shift = 16 - psid_length - psid_offset;
142   d->psid_mask = (1 << d->psid_length) - 1;
143   d->ea_shift = 64 - ip6_prefix_len - suffix_len - d->psid_length;
144
145   /* MAP data-plane object */
146   if (d->flags & MAP_DOMAIN_TRANSLATION)
147     map_t_dpo_create (DPO_PROTO_IP4, *map_domain_index, &dpo_v4);
148   else
149     map_dpo_create (DPO_PROTO_IP4, *map_domain_index, &dpo_v4);
150
151   /* Create ip4 route */
152   fib_prefix_t pfx = {
153     .fp_proto = FIB_PROTOCOL_IP4,
154     .fp_len = d->ip4_prefix_len,
155     .fp_addr = {
156                 .ip4 = d->ip4_prefix,
157                 }
158     ,
159   };
160   fib_table_entry_special_dpo_add (0, &pfx,
161                                    FIB_SOURCE_MAP,
162                                    FIB_ENTRY_FLAG_EXCLUSIVE, &dpo_v4);
163   dpo_reset (&dpo_v4);
164
165   /*
166    * construct a DPO to use the v6 domain
167    */
168   if (d->flags & MAP_DOMAIN_TRANSLATION)
169     map_t_dpo_create (DPO_PROTO_IP6, *map_domain_index, &dpo_v6);
170   else
171     map_dpo_create (DPO_PROTO_IP6, *map_domain_index, &dpo_v6);
172
173   /*
174    * Multiple MAP domains may share same source IPv6 TEP. Which is just dandy.
175    * We are not tracking the sharing. So a v4 lookup to find the correct
176    * domain post decap/trnaslate is always done
177    *
178    * Create ip6 route. This is a reference counted add. If the prefix
179    * already exists and is MAP sourced, it is now MAP source n+1 times
180    * and will need to be removed n+1 times.
181    */
182   fib_prefix_t pfx6 = {
183     .fp_proto = FIB_PROTOCOL_IP6,
184     .fp_len = d->ip6_src_len,
185     .fp_addr.ip6 = d->ip6_src,
186   };
187
188   fib_table_entry_special_dpo_add (0, &pfx6,
189                                    FIB_SOURCE_MAP,
190                                    FIB_ENTRY_FLAG_EXCLUSIVE, &dpo_v6);
191   dpo_reset (&dpo_v6);
192
193   /* Validate packet/byte counters */
194   map_domain_counter_lock (mm);
195   int i;
196   for (i = 0; i < vec_len (mm->simple_domain_counters); i++)
197     {
198       vlib_validate_simple_counter (&mm->simple_domain_counters[i],
199                                     *map_domain_index);
200       vlib_zero_simple_counter (&mm->simple_domain_counters[i],
201                                 *map_domain_index);
202     }
203   for (i = 0; i < vec_len (mm->domain_counters); i++)
204     {
205       vlib_validate_combined_counter (&mm->domain_counters[i],
206                                       *map_domain_index);
207       vlib_zero_combined_counter (&mm->domain_counters[i], *map_domain_index);
208     }
209   map_domain_counter_unlock (mm);
210
211   return 0;
212 }
213
214 /*
215  * map_delete_domain
216  */
217 int
218 map_delete_domain (u32 map_domain_index)
219 {
220   map_main_t *mm = &map_main;
221   map_domain_t *d;
222
223   if (pool_is_free_index (mm->domains, map_domain_index))
224     {
225       clib_warning ("MAP domain delete: domain does not exist: %d",
226                     map_domain_index);
227       return -1;
228     }
229
230   d = pool_elt_at_index (mm->domains, map_domain_index);
231
232   fib_prefix_t pfx = {
233     .fp_proto = FIB_PROTOCOL_IP4,
234     .fp_len = d->ip4_prefix_len,
235     .fp_addr = {
236                 .ip4 = d->ip4_prefix,
237                 }
238     ,
239   };
240   fib_table_entry_special_remove (0, &pfx, FIB_SOURCE_MAP);
241
242   fib_prefix_t pfx6 = {
243     .fp_proto = FIB_PROTOCOL_IP6,
244     .fp_len = d->ip6_src_len,
245     .fp_addr = {
246                 .ip6 = d->ip6_src,
247                 }
248     ,
249   };
250   fib_table_entry_special_remove (0, &pfx6, FIB_SOURCE_MAP);
251
252   /* Deleting rules */
253   if (d->rules)
254     clib_mem_free (d->rules);
255
256   pool_put (mm->domains, d);
257
258   return 0;
259 }
260
261 int
262 map_add_del_psid (u32 map_domain_index, u16 psid, ip6_address_t * tep,
263                   u8 is_add)
264 {
265   map_domain_t *d;
266   map_main_t *mm = &map_main;
267
268   if (pool_is_free_index (mm->domains, map_domain_index))
269     {
270       clib_warning ("MAP rule: domain does not exist: %d", map_domain_index);
271       return -1;
272     }
273   d = pool_elt_at_index (mm->domains, map_domain_index);
274
275   /* Rules are only used in 1:1 independent case */
276   if (d->ea_bits_len > 0)
277     return (-1);
278
279   if (!d->rules)
280     {
281       u32 l = (0x1 << d->psid_length) * sizeof (ip6_address_t);
282       d->rules = clib_mem_alloc_aligned (l, CLIB_CACHE_LINE_BYTES);
283       if (!d->rules)
284         return -1;
285       memset (d->rules, 0, l);
286     }
287
288   if (psid >= (0x1 << d->psid_length))
289     {
290       clib_warning ("MAP rule: PSID outside bounds: %d [%d]", psid,
291                     0x1 << d->psid_length);
292       return -1;
293     }
294
295   if (is_add)
296     {
297       d->rules[psid] = *tep;
298     }
299   else
300     {
301       memset (&d->rules[psid], 0, sizeof (ip6_address_t));
302     }
303   return 0;
304 }
305
306 #ifdef MAP_SKIP_IP6_LOOKUP
307 /**
308  * Pre-resolvd per-protocol global next-hops
309  */
310 map_main_pre_resolved_t pre_resolved[FIB_PROTOCOL_MAX];
311
312 static void
313 map_pre_resolve_init (map_main_pre_resolved_t * pr)
314 {
315   pr->fei = FIB_NODE_INDEX_INVALID;
316   fib_node_init (&pr->node, FIB_NODE_TYPE_MAP_E);
317 }
318
319 static u8 *
320 format_map_pre_resolve (u8 * s, va_list ap)
321 {
322   map_main_pre_resolved_t *pr = va_arg (ap, map_main_pre_resolved_t *);
323
324   if (FIB_NODE_INDEX_INVALID != pr->fei)
325     {
326       fib_prefix_t pfx;
327
328       fib_entry_get_prefix (pr->fei, &pfx);
329
330       return (format (s, "%U (%u)",
331                       format_ip46_address, &pfx.fp_addr, IP46_TYPE_ANY,
332                       pr->dpo.dpoi_index));
333     }
334   else
335     {
336       return (format (s, "un-set"));
337     }
338 }
339
340
341 /**
342  * Function definition to inform the FIB node that its last lock has gone.
343  */
344 static void
345 map_last_lock_gone (fib_node_t * node)
346 {
347   /*
348    * The MAP is a root of the graph. As such
349    * it never has children and thus is never locked.
350    */
351   ASSERT (0);
352 }
353
354 static map_main_pre_resolved_t *
355 map_from_fib_node (fib_node_t * node)
356 {
357 #if (CLIB_DEBUG > 0)
358   ASSERT (FIB_NODE_TYPE_MAP_E == node->fn_type);
359 #endif
360   return ((map_main_pre_resolved_t *)
361           (((char *) node) -
362            STRUCT_OFFSET_OF (map_main_pre_resolved_t, node)));
363 }
364
365 static void
366 map_stack (map_main_pre_resolved_t * pr)
367 {
368   const dpo_id_t *dpo;
369
370   dpo = fib_entry_contribute_ip_forwarding (pr->fei);
371
372   dpo_copy (&pr->dpo, dpo);
373 }
374
375 /**
376  * Function definition to backwalk a FIB node
377  */
378 static fib_node_back_walk_rc_t
379 map_back_walk (fib_node_t * node, fib_node_back_walk_ctx_t * ctx)
380 {
381   map_stack (map_from_fib_node (node));
382
383   return (FIB_NODE_BACK_WALK_CONTINUE);
384 }
385
386 /**
387  * Function definition to get a FIB node from its index
388  */
389 static fib_node_t *
390 map_fib_node_get (fib_node_index_t index)
391 {
392   return (&pre_resolved[index].node);
393 }
394
395 /*
396  * Virtual function table registered by MPLS GRE tunnels
397  * for participation in the FIB object graph.
398  */
399 const static fib_node_vft_t map_vft = {
400   .fnv_get = map_fib_node_get,
401   .fnv_last_lock = map_last_lock_gone,
402   .fnv_back_walk = map_back_walk,
403 };
404
405 static void
406 map_fib_resolve (map_main_pre_resolved_t * pr,
407                  fib_protocol_t proto, u8 len, const ip46_address_t * addr)
408 {
409   fib_prefix_t pfx = {
410     .fp_proto = proto,
411     .fp_len = len,
412     .fp_addr = *addr,
413   };
414
415   pr->fei = fib_table_entry_special_add (0,     // default fib
416                                          &pfx,
417                                          FIB_SOURCE_RR, FIB_ENTRY_FLAG_NONE);
418   pr->sibling = fib_entry_child_add (pr->fei, FIB_NODE_TYPE_MAP_E, proto);
419   map_stack (pr);
420 }
421
422 static void
423 map_fib_unresolve (map_main_pre_resolved_t * pr,
424                    fib_protocol_t proto, u8 len, const ip46_address_t * addr)
425 {
426   fib_prefix_t pfx = {
427     .fp_proto = proto,
428     .fp_len = len,
429     .fp_addr = *addr,
430   };
431
432   fib_entry_child_remove (pr->fei, pr->sibling);
433
434   fib_table_entry_special_remove (0,    // default fib
435                                   &pfx, FIB_SOURCE_RR);
436   dpo_reset (&pr->dpo);
437
438   pr->fei = FIB_NODE_INDEX_INVALID;
439   pr->sibling = FIB_NODE_INDEX_INVALID;
440 }
441
442 static void
443 map_pre_resolve (ip4_address_t * ip4, ip6_address_t * ip6, int is_del)
444 {
445   if (ip6 && (ip6->as_u64[0] != 0 || ip6->as_u64[1] != 0))
446     {
447       ip46_address_t addr = {
448         .ip6 = *ip6,
449       };
450       if (is_del)
451         map_fib_unresolve (&pre_resolved[FIB_PROTOCOL_IP6],
452                            FIB_PROTOCOL_IP6, 128, &addr);
453       else
454         map_fib_resolve (&pre_resolved[FIB_PROTOCOL_IP6],
455                          FIB_PROTOCOL_IP6, 128, &addr);
456     }
457   if (ip4 && (ip4->as_u32 != 0))
458     {
459       ip46_address_t addr = {
460         .ip4 = *ip4,
461       };
462       if (is_del)
463         map_fib_unresolve (&pre_resolved[FIB_PROTOCOL_IP4],
464                            FIB_PROTOCOL_IP4, 32, &addr);
465       else
466         map_fib_resolve (&pre_resolved[FIB_PROTOCOL_IP4],
467                          FIB_PROTOCOL_IP4, 32, &addr);
468     }
469 }
470 #endif
471
472 static clib_error_t *
473 map_security_check_command_fn (vlib_main_t * vm,
474                                unformat_input_t * input,
475                                vlib_cli_command_t * cmd)
476 {
477   unformat_input_t _line_input, *line_input = &_line_input;
478   map_main_t *mm = &map_main;
479   clib_error_t *error = NULL;
480
481   /* Get a line of input. */
482   if (!unformat_user (input, unformat_line_input, line_input))
483     return 0;
484
485   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
486     {
487       if (unformat (line_input, "off"))
488         mm->sec_check = false;
489       else if (unformat (line_input, "on"))
490         mm->sec_check = true;
491       else
492         {
493           error = clib_error_return (0, "unknown input `%U'",
494                                      format_unformat_error, line_input);
495           goto done;
496         }
497     }
498
499 done:
500   unformat_free (line_input);
501
502   return error;
503 }
504
505 static clib_error_t *
506 map_security_check_frag_command_fn (vlib_main_t * vm,
507                                     unformat_input_t * input,
508                                     vlib_cli_command_t * cmd)
509 {
510   unformat_input_t _line_input, *line_input = &_line_input;
511   map_main_t *mm = &map_main;
512   clib_error_t *error = NULL;
513
514   /* Get a line of input. */
515   if (!unformat_user (input, unformat_line_input, line_input))
516     return 0;
517
518   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
519     {
520       if (unformat (line_input, "off"))
521         mm->sec_check_frag = false;
522       else if (unformat (line_input, "on"))
523         mm->sec_check_frag = true;
524       else
525         {
526           error = clib_error_return (0, "unknown input `%U'",
527                                      format_unformat_error, line_input);
528           goto done;
529         }
530     }
531
532 done:
533   unformat_free (line_input);
534
535   return error;
536 }
537
538 static clib_error_t *
539 map_add_domain_command_fn (vlib_main_t * vm,
540                            unformat_input_t * input, vlib_cli_command_t * cmd)
541 {
542   unformat_input_t _line_input, *line_input = &_line_input;
543   ip4_address_t ip4_prefix;
544   ip6_address_t ip6_prefix;
545   ip6_address_t ip6_src;
546   u32 ip6_prefix_len = 0, ip4_prefix_len = 0, map_domain_index, ip6_src_len;
547   u32 num_m_args = 0;
548   /* Optional arguments */
549   u32 ea_bits_len = 0, psid_offset = 0, psid_length = 0;
550   u32 mtu = 0;
551   u8 flags = 0;
552   ip6_src_len = 128;
553   clib_error_t *error = NULL;
554
555   /* Get a line of input. */
556   if (!unformat_user (input, unformat_line_input, line_input))
557     return 0;
558
559   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
560     {
561       if (unformat
562           (line_input, "ip4-pfx %U/%d", unformat_ip4_address, &ip4_prefix,
563            &ip4_prefix_len))
564         num_m_args++;
565       else
566         if (unformat
567             (line_input, "ip6-pfx %U/%d", unformat_ip6_address, &ip6_prefix,
568              &ip6_prefix_len))
569         num_m_args++;
570       else
571         if (unformat
572             (line_input, "ip6-src %U/%d", unformat_ip6_address, &ip6_src,
573              &ip6_src_len))
574         num_m_args++;
575       else
576         if (unformat
577             (line_input, "ip6-src %U", unformat_ip6_address, &ip6_src))
578         num_m_args++;
579       else if (unformat (line_input, "ea-bits-len %d", &ea_bits_len))
580         num_m_args++;
581       else if (unformat (line_input, "psid-offset %d", &psid_offset))
582         num_m_args++;
583       else if (unformat (line_input, "psid-len %d", &psid_length))
584         num_m_args++;
585       else if (unformat (line_input, "mtu %d", &mtu))
586         num_m_args++;
587       else if (unformat (line_input, "map-t"))
588         flags |= MAP_DOMAIN_TRANSLATION;
589       else
590         {
591           error = clib_error_return (0, "unknown input `%U'",
592                                      format_unformat_error, line_input);
593           goto done;
594         }
595     }
596
597   if (num_m_args < 3)
598     {
599       error = clib_error_return (0, "mandatory argument(s) missing");
600       goto done;
601     }
602
603   map_create_domain (&ip4_prefix, ip4_prefix_len,
604                      &ip6_prefix, ip6_prefix_len, &ip6_src, ip6_src_len,
605                      ea_bits_len, psid_offset, psid_length, &map_domain_index,
606                      mtu, flags);
607
608 done:
609   unformat_free (line_input);
610
611   return error;
612 }
613
614 static clib_error_t *
615 map_del_domain_command_fn (vlib_main_t * vm,
616                            unformat_input_t * input, vlib_cli_command_t * cmd)
617 {
618   unformat_input_t _line_input, *line_input = &_line_input;
619   u32 num_m_args = 0;
620   u32 map_domain_index;
621   clib_error_t *error = NULL;
622
623   /* Get a line of input. */
624   if (!unformat_user (input, unformat_line_input, line_input))
625     return 0;
626
627   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
628     {
629       if (unformat (line_input, "index %d", &map_domain_index))
630         num_m_args++;
631       else
632         {
633           error = clib_error_return (0, "unknown input `%U'",
634                                      format_unformat_error, line_input);
635           goto done;
636         }
637     }
638
639   if (num_m_args != 1)
640     {
641       error = clib_error_return (0, "mandatory argument(s) missing");
642       goto done;
643     }
644
645   map_delete_domain (map_domain_index);
646
647 done:
648   unformat_free (line_input);
649
650   return error;
651 }
652
653 static clib_error_t *
654 map_add_rule_command_fn (vlib_main_t * vm,
655                          unformat_input_t * input, vlib_cli_command_t * cmd)
656 {
657   unformat_input_t _line_input, *line_input = &_line_input;
658   ip6_address_t tep;
659   u32 num_m_args = 0;
660   u32 psid = 0, map_domain_index;
661   clib_error_t *error = NULL;
662
663   /* Get a line of input. */
664   if (!unformat_user (input, unformat_line_input, line_input))
665     return 0;
666
667   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
668     {
669       if (unformat (line_input, "index %d", &map_domain_index))
670         num_m_args++;
671       else if (unformat (line_input, "psid %d", &psid))
672         num_m_args++;
673       else
674         if (unformat (line_input, "ip6-dst %U", unformat_ip6_address, &tep))
675         num_m_args++;
676       else
677         {
678           error = clib_error_return (0, "unknown input `%U'",
679                                      format_unformat_error, line_input);
680           goto done;
681         }
682     }
683
684   if (num_m_args != 3)
685     {
686       error = clib_error_return (0, "mandatory argument(s) missing");
687       goto done;
688     }
689
690   if (map_add_del_psid (map_domain_index, psid, &tep, 1) != 0)
691     {
692       error = clib_error_return (0, "Failing to add Mapping Rule");
693       goto done;
694     }
695
696 done:
697   unformat_free (line_input);
698
699   return error;
700 }
701
702 #if MAP_SKIP_IP6_LOOKUP
703 static clib_error_t *
704 map_pre_resolve_command_fn (vlib_main_t * vm,
705                             unformat_input_t * input,
706                             vlib_cli_command_t * cmd)
707 {
708   unformat_input_t _line_input, *line_input = &_line_input;
709   ip4_address_t ip4nh, *p_v4 = NULL;
710   ip6_address_t ip6nh, *p_v6 = NULL;
711   clib_error_t *error = NULL;
712   int is_del = 0;
713
714   memset (&ip4nh, 0, sizeof (ip4nh));
715   memset (&ip6nh, 0, sizeof (ip6nh));
716
717   /* Get a line of input. */
718   if (!unformat_user (input, unformat_line_input, line_input))
719     return 0;
720
721   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
722     {
723       if (unformat (line_input, "ip4-nh %U", unformat_ip4_address, &ip4nh))
724         p_v4 = &ip4nh;
725       else
726         if (unformat (line_input, "ip6-nh %U", unformat_ip6_address, &ip6nh))
727         p_v6 = &ip6nh;
728       else if (unformat (line_input, "del"))
729         is_del = 1;
730       else
731         {
732           error = clib_error_return (0, "unknown input `%U'",
733                                      format_unformat_error, line_input);
734           goto done;
735         }
736     }
737
738   map_pre_resolve (p_v4, p_v6, is_del);
739
740 done:
741   unformat_free (line_input);
742
743   return error;
744 }
745 #endif
746
747 static clib_error_t *
748 map_icmp_relay_source_address_command_fn (vlib_main_t * vm,
749                                           unformat_input_t * input,
750                                           vlib_cli_command_t * cmd)
751 {
752   unformat_input_t _line_input, *line_input = &_line_input;
753   ip4_address_t icmp_src_address;
754   map_main_t *mm = &map_main;
755   clib_error_t *error = NULL;
756
757   mm->icmp4_src_address.as_u32 = 0;
758
759   /* Get a line of input. */
760   if (!unformat_user (input, unformat_line_input, line_input))
761     return 0;
762
763   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
764     {
765       if (unformat
766           (line_input, "%U", unformat_ip4_address, &icmp_src_address))
767         mm->icmp4_src_address = icmp_src_address;
768       else
769         {
770           error = clib_error_return (0, "unknown input `%U'",
771                                      format_unformat_error, line_input);
772           goto done;
773         }
774     }
775
776 done:
777   unformat_free (line_input);
778
779   return error;
780 }
781
782 static clib_error_t *
783 map_icmp_unreachables_command_fn (vlib_main_t * vm,
784                                   unformat_input_t * input,
785                                   vlib_cli_command_t * cmd)
786 {
787   unformat_input_t _line_input, *line_input = &_line_input;
788   map_main_t *mm = &map_main;
789   int num_m_args = 0;
790   clib_error_t *error = NULL;
791
792   /* Get a line of input. */
793   if (!unformat_user (input, unformat_line_input, line_input))
794     return 0;
795
796   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
797     {
798       num_m_args++;
799       if (unformat (line_input, "on"))
800         mm->icmp6_enabled = true;
801       else if (unformat (line_input, "off"))
802         mm->icmp6_enabled = false;
803       else
804         {
805           error = clib_error_return (0, "unknown input `%U'",
806                                      format_unformat_error, line_input);
807           goto done;
808         }
809     }
810
811
812   if (num_m_args != 1)
813     error = clib_error_return (0, "mandatory argument(s) missing");
814
815 done:
816   unformat_free (line_input);
817
818   return error;
819 }
820
821 static clib_error_t *
822 map_fragment_command_fn (vlib_main_t * vm,
823                          unformat_input_t * input, vlib_cli_command_t * cmd)
824 {
825   unformat_input_t _line_input, *line_input = &_line_input;
826   map_main_t *mm = &map_main;
827   clib_error_t *error = NULL;
828
829   /* Get a line of input. */
830   if (!unformat_user (input, unformat_line_input, line_input))
831     return 0;
832
833   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
834     {
835       if (unformat (line_input, "inner"))
836         mm->frag_inner = true;
837       else if (unformat (line_input, "outer"))
838         mm->frag_inner = false;
839       else
840         {
841           error = clib_error_return (0, "unknown input `%U'",
842                                      format_unformat_error, line_input);
843           goto done;
844         }
845     }
846
847 done:
848   unformat_free (line_input);
849
850   return error;
851 }
852
853 static clib_error_t *
854 map_fragment_df_command_fn (vlib_main_t * vm,
855                             unformat_input_t * input,
856                             vlib_cli_command_t * cmd)
857 {
858   unformat_input_t _line_input, *line_input = &_line_input;
859   map_main_t *mm = &map_main;
860   clib_error_t *error = NULL;
861
862   /* Get a line of input. */
863   if (!unformat_user (input, unformat_line_input, line_input))
864     return 0;
865
866   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
867     {
868       if (unformat (line_input, "on"))
869         mm->frag_ignore_df = true;
870       else if (unformat (line_input, "off"))
871         mm->frag_ignore_df = false;
872       else
873         {
874           error = clib_error_return (0, "unknown input `%U'",
875                                      format_unformat_error, line_input);
876           goto done;
877         }
878     }
879
880 done:
881   unformat_free (line_input);
882
883   return error;
884 }
885
886 static clib_error_t *
887 map_traffic_class_command_fn (vlib_main_t * vm,
888                               unformat_input_t * input,
889                               vlib_cli_command_t * cmd)
890 {
891   unformat_input_t _line_input, *line_input = &_line_input;
892   map_main_t *mm = &map_main;
893   u32 tc = 0;
894   clib_error_t *error = NULL;
895
896   mm->tc_copy = false;
897
898   /* Get a line of input. */
899   if (!unformat_user (input, unformat_line_input, line_input))
900     return 0;
901
902   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
903     {
904       if (unformat (line_input, "copy"))
905         mm->tc_copy = true;
906       else if (unformat (line_input, "%x", &tc))
907         mm->tc = tc & 0xff;
908       else
909         {
910           error = clib_error_return (0, "unknown input `%U'",
911                                      format_unformat_error, line_input);
912           goto done;
913         }
914     }
915
916 done:
917   unformat_free (line_input);
918
919   return error;
920 }
921
922 static u8 *
923 format_map_domain (u8 * s, va_list * args)
924 {
925   map_domain_t *d = va_arg (*args, map_domain_t *);
926   bool counters = va_arg (*args, int);
927   map_main_t *mm = &map_main;
928   ip6_address_t ip6_prefix;
929
930   if (d->rules)
931     memset (&ip6_prefix, 0, sizeof (ip6_prefix));
932   else
933     ip6_prefix = d->ip6_prefix;
934
935   s = format (s,
936               "[%d] ip4-pfx %U/%d ip6-pfx %U/%d ip6-src %U/%d ea_bits_len %d psid-offset %d psid-len %d mtu %d %s",
937               d - mm->domains,
938               format_ip4_address, &d->ip4_prefix, d->ip4_prefix_len,
939               format_ip6_address, &ip6_prefix, d->ip6_prefix_len,
940               format_ip6_address, &d->ip6_src, d->ip6_src_len,
941               d->ea_bits_len, d->psid_offset, d->psid_length, d->mtu,
942               (d->flags & MAP_DOMAIN_TRANSLATION) ? "map-t" : "");
943
944   if (counters)
945     {
946       map_domain_counter_lock (mm);
947       vlib_counter_t v;
948       vlib_get_combined_counter (&mm->domain_counters[MAP_DOMAIN_COUNTER_TX],
949                                  d - mm->domains, &v);
950       s = format (s, "  TX: %lld/%lld", v.packets, v.bytes);
951       vlib_get_combined_counter (&mm->domain_counters[MAP_DOMAIN_COUNTER_RX],
952                                  d - mm->domains, &v);
953       s = format (s, "  RX: %lld/%lld", v.packets, v.bytes);
954       map_domain_counter_unlock (mm);
955     }
956   s = format (s, "\n");
957
958   if (d->rules)
959     {
960       int i;
961       ip6_address_t dst;
962       for (i = 0; i < (0x1 << d->psid_length); i++)
963         {
964           dst = d->rules[i];
965           if (dst.as_u64[0] == 0 && dst.as_u64[1] == 0)
966             continue;
967           s = format (s,
968                       " rule psid: %d ip6-dst %U\n", i, format_ip6_address,
969                       &dst);
970         }
971     }
972   return s;
973 }
974
975 static u8 *
976 format_map_ip4_reass (u8 * s, va_list * args)
977 {
978   map_main_t *mm = &map_main;
979   map_ip4_reass_t *r = va_arg (*args, map_ip4_reass_t *);
980   map_ip4_reass_key_t *k = &r->key;
981   f64 now = vlib_time_now (mm->vlib_main);
982   f64 lifetime = (((f64) mm->ip4_reass_conf_lifetime_ms) / 1000);
983   f64 dt = (r->ts + lifetime > now) ? (r->ts + lifetime - now) : -1;
984   s = format (s,
985               "ip4-reass src=%U  dst=%U  protocol=%d  identifier=%d  port=%d  lifetime=%.3lf\n",
986               format_ip4_address, &k->src.as_u8, format_ip4_address,
987               &k->dst.as_u8, k->protocol,
988               clib_net_to_host_u16 (k->fragment_id),
989               (r->port >= 0) ? clib_net_to_host_u16 (r->port) : -1, dt);
990   return s;
991 }
992
993 static u8 *
994 format_map_ip6_reass (u8 * s, va_list * args)
995 {
996   map_main_t *mm = &map_main;
997   map_ip6_reass_t *r = va_arg (*args, map_ip6_reass_t *);
998   map_ip6_reass_key_t *k = &r->key;
999   f64 now = vlib_time_now (mm->vlib_main);
1000   f64 lifetime = (((f64) mm->ip6_reass_conf_lifetime_ms) / 1000);
1001   f64 dt = (r->ts + lifetime > now) ? (r->ts + lifetime - now) : -1;
1002   s = format (s,
1003               "ip6-reass src=%U  dst=%U  protocol=%d  identifier=%d  lifetime=%.3lf\n",
1004               format_ip6_address, &k->src.as_u8, format_ip6_address,
1005               &k->dst.as_u8, k->protocol,
1006               clib_net_to_host_u32 (k->fragment_id), dt);
1007   return s;
1008 }
1009
1010 static clib_error_t *
1011 show_map_domain_command_fn (vlib_main_t * vm, unformat_input_t * input,
1012                             vlib_cli_command_t * cmd)
1013 {
1014   unformat_input_t _line_input, *line_input = &_line_input;
1015   map_main_t *mm = &map_main;
1016   map_domain_t *d;
1017   bool counters = false;
1018   u32 map_domain_index = ~0;
1019   clib_error_t *error = NULL;
1020
1021   /* Get a line of input. */
1022   if (!unformat_user (input, unformat_line_input, line_input))
1023     return 0;
1024
1025   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1026     {
1027       if (unformat (line_input, "counters"))
1028         counters = true;
1029       else if (unformat (line_input, "index %d", &map_domain_index))
1030         ;
1031       else
1032         {
1033           error = clib_error_return (0, "unknown input `%U'",
1034                                      format_unformat_error, line_input);
1035           goto done;
1036         }
1037     }
1038
1039   if (pool_elts (mm->domains) == 0)
1040     vlib_cli_output (vm, "No MAP domains are configured...");
1041
1042   if (map_domain_index == ~0)
1043     {
1044     /* *INDENT-OFF* */
1045     pool_foreach(d, mm->domains, ({vlib_cli_output(vm, "%U", format_map_domain, d, counters);}));
1046     /* *INDENT-ON* */
1047     }
1048   else
1049     {
1050       if (pool_is_free_index (mm->domains, map_domain_index))
1051         {
1052           error = clib_error_return (0, "MAP domain does not exists %d",
1053                                      map_domain_index);
1054           goto done;
1055         }
1056
1057       d = pool_elt_at_index (mm->domains, map_domain_index);
1058       vlib_cli_output (vm, "%U", format_map_domain, d, counters);
1059     }
1060
1061 done:
1062   unformat_free (line_input);
1063
1064   return error;
1065 }
1066
1067 static clib_error_t *
1068 show_map_fragments_command_fn (vlib_main_t * vm, unformat_input_t * input,
1069                                vlib_cli_command_t * cmd)
1070 {
1071   map_main_t *mm = &map_main;
1072   map_ip4_reass_t *f4;
1073   map_ip6_reass_t *f6;
1074
1075   /* *INDENT-OFF* */
1076   pool_foreach(f4, mm->ip4_reass_pool, ({vlib_cli_output (vm, "%U", format_map_ip4_reass, f4);}));
1077   /* *INDENT-ON* */
1078   /* *INDENT-OFF* */
1079   pool_foreach(f6, mm->ip6_reass_pool, ({vlib_cli_output (vm, "%U", format_map_ip6_reass, f6);}));
1080   /* *INDENT-ON* */
1081   return (0);
1082 }
1083
1084 u64
1085 map_error_counter_get (u32 node_index, map_error_t map_error)
1086 {
1087   vlib_main_t *vm = vlib_get_main ();
1088   vlib_node_runtime_t *error_node = vlib_node_get_runtime (vm, node_index);
1089   vlib_error_main_t *em = &vm->error_main;
1090   vlib_error_t e = error_node->errors[map_error];
1091   vlib_node_t *n = vlib_get_node (vm, node_index);
1092   u32 ci;
1093
1094   ci = vlib_error_get_code (e);
1095   ASSERT (ci < n->n_errors);
1096   ci += n->error_heap_index;
1097
1098   return (em->counters[ci]);
1099 }
1100
1101 static clib_error_t *
1102 show_map_stats_command_fn (vlib_main_t * vm, unformat_input_t * input,
1103                            vlib_cli_command_t * cmd)
1104 {
1105   map_main_t *mm = &map_main;
1106   map_domain_t *d;
1107   int domains = 0, rules = 0, domaincount = 0, rulecount = 0;
1108   if (pool_elts (mm->domains) == 0)
1109     vlib_cli_output (vm, "No MAP domains are configured...");
1110
1111   /* *INDENT-OFF* */
1112   pool_foreach(d, mm->domains, ({
1113     if (d->rules) {
1114       rulecount+= 0x1 << d->psid_length;
1115       rules += sizeof(ip6_address_t) * 0x1 << d->psid_length;
1116     }
1117     domains += sizeof(*d);
1118     domaincount++;
1119   }));
1120   /* *INDENT-ON* */
1121
1122   vlib_cli_output (vm, "MAP domains structure: %d\n", sizeof (map_domain_t));
1123   vlib_cli_output (vm, "MAP domains: %d (%d bytes)\n", domaincount, domains);
1124   vlib_cli_output (vm, "MAP rules: %d (%d bytes)\n", rulecount, rules);
1125   vlib_cli_output (vm, "Total: %d bytes)\n", rules + domains);
1126
1127 #if MAP_SKIP_IP6_LOOKUP
1128   vlib_cli_output (vm,
1129                    "MAP pre-resolve: IP6 next-hop: %U, IP4 next-hop: %U\n",
1130                    format_map_pre_resolve, &pre_resolved[FIB_PROTOCOL_IP6],
1131                    format_map_pre_resolve, &pre_resolved[FIB_PROTOCOL_IP4]);
1132
1133 #endif
1134
1135   if (mm->tc_copy)
1136     vlib_cli_output (vm, "MAP traffic-class: copy");
1137   else
1138     vlib_cli_output (vm, "MAP traffic-class: %x", mm->tc);
1139
1140   vlib_cli_output (vm,
1141                    "MAP IPv6 inbound security check: %s, fragmented packet security check: %s",
1142                    mm->sec_check ? "enabled" : "disabled",
1143                    mm->sec_check_frag ? "enabled" : "disabled");
1144
1145   vlib_cli_output (vm, "ICMP-relay IPv4 source address: %U\n",
1146                    format_ip4_address, &mm->icmp4_src_address);
1147   vlib_cli_output (vm, "ICMP6 unreachables sent for unmatched packets: %s\n",
1148                    mm->icmp6_enabled ? "enabled" : "disabled");
1149   vlib_cli_output (vm, "Inner fragmentation: %s\n",
1150                    mm->frag_inner ? "enabled" : "disabled");
1151   vlib_cli_output (vm, "Fragment packets regardless of DF flag: %s\n",
1152                    mm->frag_ignore_df ? "enabled" : "disabled");
1153
1154   /*
1155    * Counters
1156    */
1157   vlib_combined_counter_main_t *cm = mm->domain_counters;
1158   u64 total_pkts[MAP_N_DOMAIN_COUNTER];
1159   u64 total_bytes[MAP_N_DOMAIN_COUNTER];
1160   int which, i;
1161   vlib_counter_t v;
1162
1163   memset (total_pkts, 0, sizeof (total_pkts));
1164   memset (total_bytes, 0, sizeof (total_bytes));
1165
1166   map_domain_counter_lock (mm);
1167   vec_foreach (cm, mm->domain_counters)
1168   {
1169     which = cm - mm->domain_counters;
1170
1171     for (i = 0; i < vlib_combined_counter_n_counters (cm); i++)
1172       {
1173         vlib_get_combined_counter (cm, i, &v);
1174         total_pkts[which] += v.packets;
1175         total_bytes[which] += v.bytes;
1176       }
1177   }
1178   map_domain_counter_unlock (mm);
1179
1180   vlib_cli_output (vm, "Encapsulated packets: %lld bytes: %lld\n",
1181                    total_pkts[MAP_DOMAIN_COUNTER_TX],
1182                    total_bytes[MAP_DOMAIN_COUNTER_TX]);
1183   vlib_cli_output (vm, "Decapsulated packets: %lld bytes: %lld\n",
1184                    total_pkts[MAP_DOMAIN_COUNTER_RX],
1185                    total_bytes[MAP_DOMAIN_COUNTER_RX]);
1186
1187   vlib_cli_output (vm, "ICMP relayed packets: %d\n",
1188                    vlib_get_simple_counter (&mm->icmp_relayed, 0));
1189
1190   return 0;
1191 }
1192
1193 static clib_error_t *
1194 map_params_reass_command_fn (vlib_main_t * vm, unformat_input_t * input,
1195                              vlib_cli_command_t * cmd)
1196 {
1197   unformat_input_t _line_input, *line_input = &_line_input;
1198   u32 lifetime = ~0;
1199   f64 ht_ratio = (MAP_IP4_REASS_CONF_HT_RATIO_MAX + 1);
1200   u32 pool_size = ~0;
1201   u64 buffers = ~(0ull);
1202   u8 ip4 = 0, ip6 = 0;
1203
1204   if (!unformat_user (input, unformat_line_input, line_input))
1205     return 0;
1206
1207   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
1208     {
1209       if (unformat (line_input, "lifetime %u", &lifetime))
1210         ;
1211       else if (unformat (line_input, "ht-ratio %lf", &ht_ratio))
1212         ;
1213       else if (unformat (line_input, "pool-size %u", &pool_size))
1214         ;
1215       else if (unformat (line_input, "buffers %llu", &buffers))
1216         ;
1217       else if (unformat (line_input, "ip4"))
1218         ip4 = 1;
1219       else if (unformat (line_input, "ip6"))
1220         ip6 = 1;
1221       else
1222         {
1223           unformat_free (line_input);
1224           return clib_error_return (0, "invalid input");
1225         }
1226     }
1227   unformat_free (line_input);
1228
1229   if (!ip4 && !ip6)
1230     return clib_error_return (0, "must specify ip4 and/or ip6");
1231
1232   if (ip4)
1233     {
1234       if (pool_size != ~0 && pool_size > MAP_IP4_REASS_CONF_POOL_SIZE_MAX)
1235         return clib_error_return (0, "invalid ip4-reass pool-size ( > %d)",
1236                                   MAP_IP4_REASS_CONF_POOL_SIZE_MAX);
1237       if (ht_ratio != (MAP_IP4_REASS_CONF_HT_RATIO_MAX + 1)
1238           && ht_ratio > MAP_IP4_REASS_CONF_HT_RATIO_MAX)
1239         return clib_error_return (0, "invalid ip4-reass ht-ratio ( > %d)",
1240                                   MAP_IP4_REASS_CONF_HT_RATIO_MAX);
1241       if (lifetime != ~0 && lifetime > MAP_IP4_REASS_CONF_LIFETIME_MAX)
1242         return clib_error_return (0, "invalid ip4-reass lifetime ( > %d)",
1243                                   MAP_IP4_REASS_CONF_LIFETIME_MAX);
1244       if (buffers != ~(0ull) && buffers > MAP_IP4_REASS_CONF_BUFFERS_MAX)
1245         return clib_error_return (0, "invalid ip4-reass buffers ( > %ld)",
1246                                   MAP_IP4_REASS_CONF_BUFFERS_MAX);
1247     }
1248
1249   if (ip6)
1250     {
1251       if (pool_size != ~0 && pool_size > MAP_IP6_REASS_CONF_POOL_SIZE_MAX)
1252         return clib_error_return (0, "invalid ip6-reass pool-size ( > %d)",
1253                                   MAP_IP6_REASS_CONF_POOL_SIZE_MAX);
1254       if (ht_ratio != (MAP_IP4_REASS_CONF_HT_RATIO_MAX + 1)
1255           && ht_ratio > MAP_IP6_REASS_CONF_HT_RATIO_MAX)
1256         return clib_error_return (0, "invalid ip6-reass ht-log2len ( > %d)",
1257                                   MAP_IP6_REASS_CONF_HT_RATIO_MAX);
1258       if (lifetime != ~0 && lifetime > MAP_IP6_REASS_CONF_LIFETIME_MAX)
1259         return clib_error_return (0, "invalid ip6-reass lifetime ( > %d)",
1260                                   MAP_IP6_REASS_CONF_LIFETIME_MAX);
1261       if (buffers != ~(0ull) && buffers > MAP_IP6_REASS_CONF_BUFFERS_MAX)
1262         return clib_error_return (0, "invalid ip6-reass buffers ( > %ld)",
1263                                   MAP_IP6_REASS_CONF_BUFFERS_MAX);
1264     }
1265
1266   if (ip4)
1267     {
1268       u32 reass = 0, packets = 0;
1269       if (pool_size != ~0)
1270         {
1271           if (map_ip4_reass_conf_pool_size (pool_size, &reass, &packets))
1272             {
1273               vlib_cli_output (vm, "Could not set ip4-reass pool-size");
1274             }
1275           else
1276             {
1277               vlib_cli_output (vm,
1278                                "Setting ip4-reass pool-size (destroyed-reassembly=%u , dropped-fragments=%u)",
1279                                reass, packets);
1280             }
1281         }
1282       if (ht_ratio != (MAP_IP4_REASS_CONF_HT_RATIO_MAX + 1))
1283         {
1284           if (map_ip4_reass_conf_ht_ratio (ht_ratio, &reass, &packets))
1285             {
1286               vlib_cli_output (vm, "Could not set ip4-reass ht-log2len");
1287             }
1288           else
1289             {
1290               vlib_cli_output (vm,
1291                                "Setting ip4-reass ht-log2len (destroyed-reassembly=%u , dropped-fragments=%u)",
1292                                reass, packets);
1293             }
1294         }
1295       if (lifetime != ~0)
1296         {
1297           if (map_ip4_reass_conf_lifetime (lifetime))
1298             vlib_cli_output (vm, "Could not set ip4-reass lifetime");
1299           else
1300             vlib_cli_output (vm, "Setting ip4-reass lifetime");
1301         }
1302       if (buffers != ~(0ull))
1303         {
1304           if (map_ip4_reass_conf_buffers (buffers))
1305             vlib_cli_output (vm, "Could not set ip4-reass buffers");
1306           else
1307             vlib_cli_output (vm, "Setting ip4-reass buffers");
1308         }
1309
1310       if (map_main.ip4_reass_conf_buffers >
1311           map_main.ip4_reass_conf_pool_size *
1312           MAP_IP4_REASS_MAX_FRAGMENTS_PER_REASSEMBLY)
1313         {
1314           vlib_cli_output (vm,
1315                            "Note: 'ip4-reass buffers' > pool-size * max-fragments-per-reassembly.");
1316         }
1317     }
1318
1319   if (ip6)
1320     {
1321       u32 reass = 0, packets = 0;
1322       if (pool_size != ~0)
1323         {
1324           if (map_ip6_reass_conf_pool_size (pool_size, &reass, &packets))
1325             {
1326               vlib_cli_output (vm, "Could not set ip6-reass pool-size");
1327             }
1328           else
1329             {
1330               vlib_cli_output (vm,
1331                                "Setting ip6-reass pool-size (destroyed-reassembly=%u , dropped-fragments=%u)",
1332                                reass, packets);
1333             }
1334         }
1335       if (ht_ratio != (MAP_IP4_REASS_CONF_HT_RATIO_MAX + 1))
1336         {
1337           if (map_ip6_reass_conf_ht_ratio (ht_ratio, &reass, &packets))
1338             {
1339               vlib_cli_output (vm, "Could not set ip6-reass ht-log2len");
1340             }
1341           else
1342             {
1343               vlib_cli_output (vm,
1344                                "Setting ip6-reass ht-log2len (destroyed-reassembly=%u , dropped-fragments=%u)",
1345                                reass, packets);
1346             }
1347         }
1348       if (lifetime != ~0)
1349         {
1350           if (map_ip6_reass_conf_lifetime (lifetime))
1351             vlib_cli_output (vm, "Could not set ip6-reass lifetime");
1352           else
1353             vlib_cli_output (vm, "Setting ip6-reass lifetime");
1354         }
1355       if (buffers != ~(0ull))
1356         {
1357           if (map_ip6_reass_conf_buffers (buffers))
1358             vlib_cli_output (vm, "Could not set ip6-reass buffers");
1359           else
1360             vlib_cli_output (vm, "Setting ip6-reass buffers");
1361         }
1362
1363       if (map_main.ip6_reass_conf_buffers >
1364           map_main.ip6_reass_conf_pool_size *
1365           MAP_IP6_REASS_MAX_FRAGMENTS_PER_REASSEMBLY)
1366         {
1367           vlib_cli_output (vm,
1368                            "Note: 'ip6-reass buffers' > pool-size * max-fragments-per-reassembly.");
1369         }
1370     }
1371
1372   return 0;
1373 }
1374
1375
1376 /*
1377  * packet trace format function
1378  */
1379 u8 *
1380 format_map_trace (u8 * s, va_list * args)
1381 {
1382   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
1383   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
1384   map_trace_t *t = va_arg (*args, map_trace_t *);
1385   u32 map_domain_index = t->map_domain_index;
1386   u16 port = t->port;
1387
1388   s =
1389     format (s, "MAP domain index: %d L4 port: %u", map_domain_index,
1390             clib_net_to_host_u16 (port));
1391
1392   return s;
1393 }
1394
1395 static_always_inline map_ip4_reass_t *
1396 map_ip4_reass_lookup (map_ip4_reass_key_t * k, u32 bucket, f64 now)
1397 {
1398   map_main_t *mm = &map_main;
1399   u32 ri = mm->ip4_reass_hash_table[bucket];
1400   while (ri != MAP_REASS_INDEX_NONE)
1401     {
1402       map_ip4_reass_t *r = pool_elt_at_index (mm->ip4_reass_pool, ri);
1403       if (r->key.as_u64[0] == k->as_u64[0] &&
1404           r->key.as_u64[1] == k->as_u64[1] &&
1405           now < r->ts + (((f64) mm->ip4_reass_conf_lifetime_ms) / 1000))
1406         {
1407           return r;
1408         }
1409       ri = r->bucket_next;
1410     }
1411   return NULL;
1412 }
1413
1414 #define map_ip4_reass_pool_index(r) (r - map_main.ip4_reass_pool)
1415
1416 void
1417 map_ip4_reass_free (map_ip4_reass_t * r, u32 ** pi_to_drop)
1418 {
1419   map_main_t *mm = &map_main;
1420   map_ip4_reass_get_fragments (r, pi_to_drop);
1421
1422   // Unlink in hash bucket
1423   map_ip4_reass_t *r2 = NULL;
1424   u32 r2i = mm->ip4_reass_hash_table[r->bucket];
1425   while (r2i != map_ip4_reass_pool_index (r))
1426     {
1427       ASSERT (r2i != MAP_REASS_INDEX_NONE);
1428       r2 = pool_elt_at_index (mm->ip4_reass_pool, r2i);
1429       r2i = r2->bucket_next;
1430     }
1431   if (r2)
1432     {
1433       r2->bucket_next = r->bucket_next;
1434     }
1435   else
1436     {
1437       mm->ip4_reass_hash_table[r->bucket] = r->bucket_next;
1438     }
1439
1440   // Unlink in list
1441   if (r->fifo_next == map_ip4_reass_pool_index (r))
1442     {
1443       mm->ip4_reass_fifo_last = MAP_REASS_INDEX_NONE;
1444     }
1445   else
1446     {
1447       if (mm->ip4_reass_fifo_last == map_ip4_reass_pool_index (r))
1448         mm->ip4_reass_fifo_last = r->fifo_prev;
1449       pool_elt_at_index (mm->ip4_reass_pool, r->fifo_prev)->fifo_next =
1450         r->fifo_next;
1451       pool_elt_at_index (mm->ip4_reass_pool, r->fifo_next)->fifo_prev =
1452         r->fifo_prev;
1453     }
1454
1455   pool_put (mm->ip4_reass_pool, r);
1456   mm->ip4_reass_allocated--;
1457 }
1458
1459 map_ip4_reass_t *
1460 map_ip4_reass_get (u32 src, u32 dst, u16 fragment_id,
1461                    u8 protocol, u32 ** pi_to_drop)
1462 {
1463   map_ip4_reass_t *r;
1464   map_main_t *mm = &map_main;
1465   map_ip4_reass_key_t k = {.src.data_u32 = src,
1466     .dst.data_u32 = dst,
1467     .fragment_id = fragment_id,
1468     .protocol = protocol
1469   };
1470
1471   u32 h = 0;
1472 #ifdef clib_crc32c_uses_intrinsics
1473   h = clib_crc32c ((u8 *) k.as_u32, 16);
1474 #else
1475   u64 tmp = k.as_u32[0] ^ k.as_u32[1] ^ k.as_u32[2] ^ k.as_u32[3];
1476   h = clib_xxhash (tmp);
1477 #endif
1478   h = h >> (32 - mm->ip4_reass_ht_log2len);
1479
1480   f64 now = vlib_time_now (mm->vlib_main);
1481
1482   //Cache garbage collection
1483   while (mm->ip4_reass_fifo_last != MAP_REASS_INDEX_NONE)
1484     {
1485       map_ip4_reass_t *last =
1486         pool_elt_at_index (mm->ip4_reass_pool, mm->ip4_reass_fifo_last);
1487       if (last->ts + (((f64) mm->ip4_reass_conf_lifetime_ms) / 1000) < now)
1488         map_ip4_reass_free (last, pi_to_drop);
1489       else
1490         break;
1491     }
1492
1493   if ((r = map_ip4_reass_lookup (&k, h, now)))
1494     return r;
1495
1496   if (mm->ip4_reass_allocated >= mm->ip4_reass_conf_pool_size)
1497     return NULL;
1498
1499   pool_get (mm->ip4_reass_pool, r);
1500   mm->ip4_reass_allocated++;
1501   int i;
1502   for (i = 0; i < MAP_IP4_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1503     r->fragments[i] = ~0;
1504
1505   u32 ri = map_ip4_reass_pool_index (r);
1506
1507   //Link in new bucket
1508   r->bucket = h;
1509   r->bucket_next = mm->ip4_reass_hash_table[h];
1510   mm->ip4_reass_hash_table[h] = ri;
1511
1512   //Link in fifo
1513   if (mm->ip4_reass_fifo_last != MAP_REASS_INDEX_NONE)
1514     {
1515       r->fifo_next =
1516         pool_elt_at_index (mm->ip4_reass_pool,
1517                            mm->ip4_reass_fifo_last)->fifo_next;
1518       r->fifo_prev = mm->ip4_reass_fifo_last;
1519       pool_elt_at_index (mm->ip4_reass_pool, r->fifo_prev)->fifo_next = ri;
1520       pool_elt_at_index (mm->ip4_reass_pool, r->fifo_next)->fifo_prev = ri;
1521     }
1522   else
1523     {
1524       r->fifo_next = r->fifo_prev = ri;
1525       mm->ip4_reass_fifo_last = ri;
1526     }
1527
1528   //Set other fields
1529   r->ts = now;
1530   r->key = k;
1531   r->port = -1;
1532 #ifdef MAP_IP4_REASS_COUNT_BYTES
1533   r->expected_total = 0xffff;
1534   r->forwarded = 0;
1535 #endif
1536
1537   return r;
1538 }
1539
1540 int
1541 map_ip4_reass_add_fragment (map_ip4_reass_t * r, u32 pi)
1542 {
1543   if (map_main.ip4_reass_buffered_counter >= map_main.ip4_reass_conf_buffers)
1544     return -1;
1545
1546   int i;
1547   for (i = 0; i < MAP_IP4_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1548     if (r->fragments[i] == ~0)
1549       {
1550         r->fragments[i] = pi;
1551         map_main.ip4_reass_buffered_counter++;
1552         return 0;
1553       }
1554   return -1;
1555 }
1556
1557 static_always_inline map_ip6_reass_t *
1558 map_ip6_reass_lookup (map_ip6_reass_key_t * k, u32 bucket, f64 now)
1559 {
1560   map_main_t *mm = &map_main;
1561   u32 ri = mm->ip6_reass_hash_table[bucket];
1562   while (ri != MAP_REASS_INDEX_NONE)
1563     {
1564       map_ip6_reass_t *r = pool_elt_at_index (mm->ip6_reass_pool, ri);
1565       if (now < r->ts + (((f64) mm->ip6_reass_conf_lifetime_ms) / 1000) &&
1566           r->key.as_u64[0] == k->as_u64[0] &&
1567           r->key.as_u64[1] == k->as_u64[1] &&
1568           r->key.as_u64[2] == k->as_u64[2] &&
1569           r->key.as_u64[3] == k->as_u64[3] &&
1570           r->key.as_u64[4] == k->as_u64[4])
1571         return r;
1572       ri = r->bucket_next;
1573     }
1574   return NULL;
1575 }
1576
1577 #define map_ip6_reass_pool_index(r) (r - map_main.ip6_reass_pool)
1578
1579 void
1580 map_ip6_reass_free (map_ip6_reass_t * r, u32 ** pi_to_drop)
1581 {
1582   map_main_t *mm = &map_main;
1583   int i;
1584   for (i = 0; i < MAP_IP6_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1585     if (r->fragments[i].pi != ~0)
1586       {
1587         vec_add1 (*pi_to_drop, r->fragments[i].pi);
1588         r->fragments[i].pi = ~0;
1589         map_main.ip6_reass_buffered_counter--;
1590       }
1591
1592   // Unlink in hash bucket
1593   map_ip6_reass_t *r2 = NULL;
1594   u32 r2i = mm->ip6_reass_hash_table[r->bucket];
1595   while (r2i != map_ip6_reass_pool_index (r))
1596     {
1597       ASSERT (r2i != MAP_REASS_INDEX_NONE);
1598       r2 = pool_elt_at_index (mm->ip6_reass_pool, r2i);
1599       r2i = r2->bucket_next;
1600     }
1601   if (r2)
1602     {
1603       r2->bucket_next = r->bucket_next;
1604     }
1605   else
1606     {
1607       mm->ip6_reass_hash_table[r->bucket] = r->bucket_next;
1608     }
1609
1610   // Unlink in list
1611   if (r->fifo_next == map_ip6_reass_pool_index (r))
1612     {
1613       //Single element in the list, list is now empty
1614       mm->ip6_reass_fifo_last = MAP_REASS_INDEX_NONE;
1615     }
1616   else
1617     {
1618       if (mm->ip6_reass_fifo_last == map_ip6_reass_pool_index (r))      //First element
1619         mm->ip6_reass_fifo_last = r->fifo_prev;
1620       pool_elt_at_index (mm->ip6_reass_pool, r->fifo_prev)->fifo_next =
1621         r->fifo_next;
1622       pool_elt_at_index (mm->ip6_reass_pool, r->fifo_next)->fifo_prev =
1623         r->fifo_prev;
1624     }
1625
1626   // Free from pool if necessary
1627   pool_put (mm->ip6_reass_pool, r);
1628   mm->ip6_reass_allocated--;
1629 }
1630
1631 map_ip6_reass_t *
1632 map_ip6_reass_get (ip6_address_t * src, ip6_address_t * dst, u32 fragment_id,
1633                    u8 protocol, u32 ** pi_to_drop)
1634 {
1635   map_ip6_reass_t *r;
1636   map_main_t *mm = &map_main;
1637   map_ip6_reass_key_t k = {
1638     .src = *src,
1639     .dst = *dst,
1640     .fragment_id = fragment_id,
1641     .protocol = protocol
1642   };
1643
1644   u32 h = 0;
1645   int i;
1646
1647 #ifdef clib_crc32c_uses_intrinsics
1648   h = clib_crc32c ((u8 *) k.as_u32, 40);
1649 #else
1650   u64 tmp =
1651     k.as_u64[0] ^ k.as_u64[1] ^ k.as_u64[2] ^ k.as_u64[3] ^ k.as_u64[4];
1652   h = clib_xxhash (tmp);
1653 #endif
1654
1655   h = h >> (32 - mm->ip6_reass_ht_log2len);
1656
1657   f64 now = vlib_time_now (mm->vlib_main);
1658
1659   //Cache garbage collection
1660   while (mm->ip6_reass_fifo_last != MAP_REASS_INDEX_NONE)
1661     {
1662       map_ip6_reass_t *last =
1663         pool_elt_at_index (mm->ip6_reass_pool, mm->ip6_reass_fifo_last);
1664       if (last->ts + (((f64) mm->ip6_reass_conf_lifetime_ms) / 1000) < now)
1665         map_ip6_reass_free (last, pi_to_drop);
1666       else
1667         break;
1668     }
1669
1670   if ((r = map_ip6_reass_lookup (&k, h, now)))
1671     return r;
1672
1673   if (mm->ip6_reass_allocated >= mm->ip6_reass_conf_pool_size)
1674     return NULL;
1675
1676   pool_get (mm->ip6_reass_pool, r);
1677   mm->ip6_reass_allocated++;
1678   for (i = 0; i < MAP_IP6_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1679     {
1680       r->fragments[i].pi = ~0;
1681       r->fragments[i].next_data_len = 0;
1682       r->fragments[i].next_data_offset = 0;
1683     }
1684
1685   u32 ri = map_ip6_reass_pool_index (r);
1686
1687   //Link in new bucket
1688   r->bucket = h;
1689   r->bucket_next = mm->ip6_reass_hash_table[h];
1690   mm->ip6_reass_hash_table[h] = ri;
1691
1692   //Link in fifo
1693   if (mm->ip6_reass_fifo_last != MAP_REASS_INDEX_NONE)
1694     {
1695       r->fifo_next =
1696         pool_elt_at_index (mm->ip6_reass_pool,
1697                            mm->ip6_reass_fifo_last)->fifo_next;
1698       r->fifo_prev = mm->ip6_reass_fifo_last;
1699       pool_elt_at_index (mm->ip6_reass_pool, r->fifo_prev)->fifo_next = ri;
1700       pool_elt_at_index (mm->ip6_reass_pool, r->fifo_next)->fifo_prev = ri;
1701     }
1702   else
1703     {
1704       r->fifo_next = r->fifo_prev = ri;
1705       mm->ip6_reass_fifo_last = ri;
1706     }
1707
1708   //Set other fields
1709   r->ts = now;
1710   r->key = k;
1711   r->ip4_header.ip_version_and_header_length = 0;
1712 #ifdef MAP_IP6_REASS_COUNT_BYTES
1713   r->expected_total = 0xffff;
1714   r->forwarded = 0;
1715 #endif
1716   return r;
1717 }
1718
1719 int
1720 map_ip6_reass_add_fragment (map_ip6_reass_t * r, u32 pi,
1721                             u16 data_offset, u16 next_data_offset,
1722                             u8 * data_start, u16 data_len)
1723 {
1724   map_ip6_fragment_t *f = NULL, *prev_f = NULL;
1725   u16 copied_len = (data_len > 20) ? 20 : data_len;
1726
1727   if (map_main.ip6_reass_buffered_counter >= map_main.ip6_reass_conf_buffers)
1728     return -1;
1729
1730   //Lookup for fragments for the current buffer
1731   //and the one before that
1732   int i;
1733   for (i = 0; i < MAP_IP6_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1734     {
1735       if (data_offset && r->fragments[i].next_data_offset == data_offset)
1736         {
1737           prev_f = &r->fragments[i];    // This is buffer for previous packet
1738         }
1739       else if (r->fragments[i].next_data_offset == next_data_offset)
1740         {
1741           f = &r->fragments[i]; // This is a buffer for the current packet
1742         }
1743       else if (r->fragments[i].next_data_offset == 0)
1744         {                       //Available
1745           if (f == NULL)
1746             f = &r->fragments[i];
1747           else if (prev_f == NULL)
1748             prev_f = &r->fragments[i];
1749         }
1750     }
1751
1752   if (!f || f->pi != ~0)
1753     return -1;
1754
1755   if (data_offset)
1756     {
1757       if (!prev_f)
1758         return -1;
1759
1760       clib_memcpy (prev_f->next_data, data_start, copied_len);
1761       prev_f->next_data_len = copied_len;
1762       prev_f->next_data_offset = data_offset;
1763     }
1764   else
1765     {
1766       if (((ip4_header_t *) data_start)->ip_version_and_header_length != 0x45)
1767         return -1;
1768
1769       if (r->ip4_header.ip_version_and_header_length == 0)
1770         clib_memcpy (&r->ip4_header, data_start, sizeof (ip4_header_t));
1771     }
1772
1773   if (data_len > 20)
1774     {
1775       f->next_data_offset = next_data_offset;
1776       f->pi = pi;
1777       map_main.ip6_reass_buffered_counter++;
1778     }
1779   return 0;
1780 }
1781
1782 void
1783 map_ip4_reass_reinit (u32 * trashed_reass, u32 * dropped_packets)
1784 {
1785   map_main_t *mm = &map_main;
1786   int i;
1787
1788   if (dropped_packets)
1789     *dropped_packets = mm->ip4_reass_buffered_counter;
1790   if (trashed_reass)
1791     *trashed_reass = mm->ip4_reass_allocated;
1792   if (mm->ip4_reass_fifo_last != MAP_REASS_INDEX_NONE)
1793     {
1794       u16 ri = mm->ip4_reass_fifo_last;
1795       do
1796         {
1797           map_ip4_reass_t *r = pool_elt_at_index (mm->ip4_reass_pool, ri);
1798           for (i = 0; i < MAP_IP4_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1799             if (r->fragments[i] != ~0)
1800               map_ip4_drop_pi (r->fragments[i]);
1801
1802           ri = r->fifo_next;
1803           pool_put (mm->ip4_reass_pool, r);
1804         }
1805       while (ri != mm->ip4_reass_fifo_last);
1806     }
1807
1808   vec_free (mm->ip4_reass_hash_table);
1809   vec_resize (mm->ip4_reass_hash_table, 1 << mm->ip4_reass_ht_log2len);
1810   for (i = 0; i < (1 << mm->ip4_reass_ht_log2len); i++)
1811     mm->ip4_reass_hash_table[i] = MAP_REASS_INDEX_NONE;
1812   pool_free (mm->ip4_reass_pool);
1813   pool_alloc (mm->ip4_reass_pool, mm->ip4_reass_conf_pool_size);
1814
1815   mm->ip4_reass_allocated = 0;
1816   mm->ip4_reass_fifo_last = MAP_REASS_INDEX_NONE;
1817   mm->ip4_reass_buffered_counter = 0;
1818 }
1819
1820 u8
1821 map_get_ht_log2len (f32 ht_ratio, u16 pool_size)
1822 {
1823   u32 desired_size = (u32) (pool_size * ht_ratio);
1824   u8 i;
1825   for (i = 1; i < 31; i++)
1826     if ((1 << i) >= desired_size)
1827       return i;
1828   return 4;
1829 }
1830
1831 int
1832 map_ip4_reass_conf_ht_ratio (f32 ht_ratio, u32 * trashed_reass,
1833                              u32 * dropped_packets)
1834 {
1835   map_main_t *mm = &map_main;
1836   if (ht_ratio > MAP_IP4_REASS_CONF_HT_RATIO_MAX)
1837     return -1;
1838
1839   map_ip4_reass_lock ();
1840   mm->ip4_reass_conf_ht_ratio = ht_ratio;
1841   mm->ip4_reass_ht_log2len =
1842     map_get_ht_log2len (ht_ratio, mm->ip4_reass_conf_pool_size);
1843   map_ip4_reass_reinit (trashed_reass, dropped_packets);
1844   map_ip4_reass_unlock ();
1845   return 0;
1846 }
1847
1848 int
1849 map_ip4_reass_conf_pool_size (u16 pool_size, u32 * trashed_reass,
1850                               u32 * dropped_packets)
1851 {
1852   map_main_t *mm = &map_main;
1853   if (pool_size > MAP_IP4_REASS_CONF_POOL_SIZE_MAX)
1854     return -1;
1855
1856   map_ip4_reass_lock ();
1857   mm->ip4_reass_conf_pool_size = pool_size;
1858   map_ip4_reass_reinit (trashed_reass, dropped_packets);
1859   map_ip4_reass_unlock ();
1860   return 0;
1861 }
1862
1863 int
1864 map_ip4_reass_conf_lifetime (u16 lifetime_ms)
1865 {
1866   map_main.ip4_reass_conf_lifetime_ms = lifetime_ms;
1867   return 0;
1868 }
1869
1870 int
1871 map_ip4_reass_conf_buffers (u32 buffers)
1872 {
1873   map_main.ip4_reass_conf_buffers = buffers;
1874   return 0;
1875 }
1876
1877 void
1878 map_ip6_reass_reinit (u32 * trashed_reass, u32 * dropped_packets)
1879 {
1880   map_main_t *mm = &map_main;
1881   if (dropped_packets)
1882     *dropped_packets = mm->ip6_reass_buffered_counter;
1883   if (trashed_reass)
1884     *trashed_reass = mm->ip6_reass_allocated;
1885   int i;
1886   if (mm->ip6_reass_fifo_last != MAP_REASS_INDEX_NONE)
1887     {
1888       u16 ri = mm->ip6_reass_fifo_last;
1889       do
1890         {
1891           map_ip6_reass_t *r = pool_elt_at_index (mm->ip6_reass_pool, ri);
1892           for (i = 0; i < MAP_IP6_REASS_MAX_FRAGMENTS_PER_REASSEMBLY; i++)
1893             if (r->fragments[i].pi != ~0)
1894               map_ip6_drop_pi (r->fragments[i].pi);
1895
1896           ri = r->fifo_next;
1897           pool_put (mm->ip6_reass_pool, r);
1898         }
1899       while (ri != mm->ip6_reass_fifo_last);
1900       mm->ip6_reass_fifo_last = MAP_REASS_INDEX_NONE;
1901     }
1902
1903   vec_free (mm->ip6_reass_hash_table);
1904   vec_resize (mm->ip6_reass_hash_table, 1 << mm->ip6_reass_ht_log2len);
1905   for (i = 0; i < (1 << mm->ip6_reass_ht_log2len); i++)
1906     mm->ip6_reass_hash_table[i] = MAP_REASS_INDEX_NONE;
1907   pool_free (mm->ip6_reass_pool);
1908   pool_alloc (mm->ip6_reass_pool, mm->ip4_reass_conf_pool_size);
1909
1910   mm->ip6_reass_allocated = 0;
1911   mm->ip6_reass_buffered_counter = 0;
1912 }
1913
1914 int
1915 map_ip6_reass_conf_ht_ratio (f32 ht_ratio, u32 * trashed_reass,
1916                              u32 * dropped_packets)
1917 {
1918   map_main_t *mm = &map_main;
1919   if (ht_ratio > MAP_IP6_REASS_CONF_HT_RATIO_MAX)
1920     return -1;
1921
1922   map_ip6_reass_lock ();
1923   mm->ip6_reass_conf_ht_ratio = ht_ratio;
1924   mm->ip6_reass_ht_log2len =
1925     map_get_ht_log2len (ht_ratio, mm->ip6_reass_conf_pool_size);
1926   map_ip6_reass_reinit (trashed_reass, dropped_packets);
1927   map_ip6_reass_unlock ();
1928   return 0;
1929 }
1930
1931 int
1932 map_ip6_reass_conf_pool_size (u16 pool_size, u32 * trashed_reass,
1933                               u32 * dropped_packets)
1934 {
1935   map_main_t *mm = &map_main;
1936   if (pool_size > MAP_IP6_REASS_CONF_POOL_SIZE_MAX)
1937     return -1;
1938
1939   map_ip6_reass_lock ();
1940   mm->ip6_reass_conf_pool_size = pool_size;
1941   map_ip6_reass_reinit (trashed_reass, dropped_packets);
1942   map_ip6_reass_unlock ();
1943   return 0;
1944 }
1945
1946 int
1947 map_ip6_reass_conf_lifetime (u16 lifetime_ms)
1948 {
1949   map_main.ip6_reass_conf_lifetime_ms = lifetime_ms;
1950   return 0;
1951 }
1952
1953 int
1954 map_ip6_reass_conf_buffers (u32 buffers)
1955 {
1956   map_main.ip6_reass_conf_buffers = buffers;
1957   return 0;
1958 }
1959
1960 /* *INDENT-OFF* */
1961
1962 /*?
1963  * Configure MAP reassembly behaviour
1964  *
1965  * @cliexpar
1966  * @cliexstart{map params reassembly}
1967  * @cliexend
1968  ?*/
1969 VLIB_CLI_COMMAND(map_ip4_reass_lifetime_command, static) = {
1970   .path = "map params reassembly",
1971   .short_help = "map params reassembly [ip4 | ip6] [lifetime <lifetime-ms>] "
1972                 "[pool-size <pool-size>] [buffers <buffers>] "
1973                 "[ht-ratio <ht-ratio>]",
1974   .function = map_params_reass_command_fn,
1975 };
1976
1977 /*?
1978  * Set or copy the IP TOS/Traffic Class field
1979  *
1980  * @cliexpar
1981  * @cliexstart{map params traffic-class}
1982  *
1983  * This command is used to set the traffic-class field in translated
1984  * or encapsulated packets. If copy is specifed (the default) then the
1985  * traffic-class/TOS field is copied from the original packet to the
1986  * translated / encapsulating header.
1987  * @cliexend
1988  ?*/
1989 VLIB_CLI_COMMAND(map_traffic_class_command, static) = {
1990   .path = "map params traffic-class",
1991   .short_help = "map params traffic-class {0x0-0xff | copy}",
1992   .function = map_traffic_class_command_fn,
1993 };
1994
1995 /*?
1996  * Bypass IP4/IP6 lookup
1997  *
1998  * @cliexpar
1999  * @cliexstart{map params pre-resolve}
2000  *
2001  * Bypass a second FIB lookup of the translated or encapsulated
2002  * packet, and forward the packet directly to the specified
2003  * next-hop. This optimization trades forwarding flexibility for
2004  * performance.
2005  * @cliexend
2006  ?*/
2007 VLIB_CLI_COMMAND(map_pre_resolve_command, static) = {
2008   .path = "map params pre-resolve",
2009   .short_help = " map params pre-resolve {ip4-nh <address>} "
2010                 "| {ip6-nh <address>}",
2011   .function = map_pre_resolve_command_fn,
2012 };
2013
2014 /*?
2015  * Enable or disable the MAP-E inbound security check
2016  *
2017  * @cliexpar
2018  * @cliexstart{map params security-check}
2019  *
2020  * By default, a decapsulated packet's IPv4 source address will be
2021  * verified against the outer header's IPv6 source address. Disabling
2022  * this feature will allow IPv4 source address spoofing.
2023  * @cliexend
2024  ?*/
2025 VLIB_CLI_COMMAND(map_security_check_command, static) = {
2026   .path = "map params security-check",
2027   .short_help = "map params security-check on|off",
2028   .function = map_security_check_command_fn,
2029 };
2030
2031 /*?
2032  * Specifiy the IPv4 source address used for relayed ICMP error messages
2033  *
2034  * @cliexpar
2035  * @cliexstart{map params icmp source-address}
2036  *
2037  * This command specifies which IPv4 source address (must be local to
2038  * the system), that is used for relayed received IPv6 ICMP error
2039  * messages.
2040  * @cliexend
2041  ?*/
2042 VLIB_CLI_COMMAND(map_icmp_relay_source_address_command, static) = {
2043   .path = "map params icmp source-address",
2044   .short_help = "map params icmp source-address <ip4-address>",
2045   .function = map_icmp_relay_source_address_command_fn,
2046 };
2047
2048 /*?
2049  * Send IPv6 ICMP unreachables
2050  *
2051  * @cliexpar
2052  * @cliexstart{map params icmp6 unreachables}
2053  *
2054  * Send IPv6 ICMP unreachable messages back if security check fails or
2055  * no MAP domain exists.
2056  * @cliexend
2057  ?*/
2058 VLIB_CLI_COMMAND(map_icmp_unreachables_command, static) = {
2059   .path = "map params icmp6 unreachables",
2060   .short_help = "map params icmp6 unreachables {on|off}",
2061   .function = map_icmp_unreachables_command_fn,
2062 };
2063
2064 /*?
2065  * Configure MAP fragmentation behaviour
2066  *
2067  * @cliexpar
2068  * @cliexstart{map params fragment}
2069  * @cliexend
2070  ?*/
2071 VLIB_CLI_COMMAND(map_fragment_command, static) = {
2072   .path = "map params fragment",
2073   .short_help = "map params fragment inner|outer",
2074   .function = map_fragment_command_fn,
2075 };
2076
2077 /*?
2078  * Ignore the IPv4 Don't fragment bit
2079  *
2080  * @cliexpar
2081  * @cliexstart{map params fragment ignore-df}
2082  *
2083  * Allows fragmentation of the IPv4 packet even if the DF bit is
2084  * set. The choice between inner or outer fragmentation of tunnel
2085  * packets is complicated. The benefit of inner fragmentation is that
2086  * the ultimate endpoint must reassemble, instead of the tunnel
2087  * endpoint.
2088  * @cliexend
2089  ?*/
2090 VLIB_CLI_COMMAND(map_fragment_df_command, static) = {
2091   .path = "map params fragment ignore-df",
2092   .short_help = "map params fragment ignore-df on|off",
2093   .function = map_fragment_df_command_fn,
2094 };
2095
2096 /*?
2097  * Specifiy if the inbound security check should be done on fragments
2098  *
2099  * @cliexpar
2100  * @cliexstart{map params security-check fragments}
2101  *
2102  * Typically the inbound on-decapsulation security check is only done
2103  * on the first packet. The packet that contains the L4
2104  * information. While a security check on every fragment is possible,
2105  * it has a cost. State must be created on the first fragment.
2106  * @cliexend
2107  ?*/
2108 VLIB_CLI_COMMAND(map_security_check_frag_command, static) = {
2109   .path = "map params security-check fragments",
2110   .short_help = "map params security-check fragments on|off",
2111   .function = map_security_check_frag_command_fn,
2112 };
2113
2114 /*?
2115  * Add MAP domain
2116  *
2117  * @cliexpar
2118  * @cliexstart{map add domain}
2119  * @cliexend
2120  ?*/
2121 VLIB_CLI_COMMAND(map_add_domain_command, static) = {
2122   .path = "map add domain",
2123   .short_help = "map add domain ip4-pfx <ip4-pfx> ip6-pfx <ip6-pfx> "
2124       "ip6-src <ip6-pfx> ea-bits-len <n> psid-offset <n> psid-len <n> "
2125       "[map-t] [mtu <mtu>]",
2126   .function = map_add_domain_command_fn,
2127 };
2128
2129 /*?
2130  * Add MAP rule to a domain
2131  *
2132  * @cliexpar
2133  * @cliexstart{map add rule}
2134  * @cliexend
2135  ?*/
2136 VLIB_CLI_COMMAND(map_add_rule_command, static) = {
2137   .path = "map add rule",
2138   .short_help = "map add rule index <domain> psid <psid> ip6-dst <ip6-addr>",
2139   .function = map_add_rule_command_fn,
2140 };
2141
2142 /*?
2143  * Delete MAP domain
2144  *
2145  * @cliexpar
2146  * @cliexstart{map del domain}
2147  * @cliexend
2148  ?*/
2149 VLIB_CLI_COMMAND(map_del_command, static) = {
2150   .path = "map del domain",
2151   .short_help = "map del domain index <domain>",
2152   .function = map_del_domain_command_fn,
2153 };
2154
2155 /*?
2156  * Show MAP domains
2157  *
2158  * @cliexpar
2159  * @cliexstart{show map domain}
2160  * @cliexend
2161  ?*/
2162 VLIB_CLI_COMMAND(show_map_domain_command, static) = {
2163   .path = "show map domain",
2164   .short_help = "show map domain index <n> [counters]",
2165   .function = show_map_domain_command_fn,
2166 };
2167
2168 /*?
2169  * Show MAP statistics
2170  *
2171  * @cliexpar
2172  * @cliexstart{show map stats}
2173  * @cliexend
2174  ?*/
2175 VLIB_CLI_COMMAND(show_map_stats_command, static) = {
2176   .path = "show map stats",
2177   .short_help = "show map stats",
2178   .function = show_map_stats_command_fn,
2179 };
2180
2181 /*?
2182  * Show MAP fragmentation information
2183  *
2184  * @cliexpar
2185  * @cliexstart{show map fragments}
2186  * @cliexend
2187  ?*/
2188 VLIB_CLI_COMMAND(show_map_fragments_command, static) = {
2189   .path = "show map fragments",
2190   .short_help = "show map fragments",
2191   .function = show_map_fragments_command_fn,
2192 };
2193 /* *INDENT-ON* */
2194
2195 /*
2196  * map_init
2197  */
2198 clib_error_t *
2199 map_init (vlib_main_t * vm)
2200 {
2201   map_main_t *mm = &map_main;
2202   mm->vnet_main = vnet_get_main ();
2203   mm->vlib_main = vm;
2204
2205 #ifdef MAP_SKIP_IP6_LOOKUP
2206   fib_protocol_t proto;
2207
2208   FOR_EACH_FIB_PROTOCOL (proto)
2209   {
2210     map_pre_resolve_init (&pre_resolved[proto]);
2211   }
2212 #endif
2213
2214   /* traffic class */
2215   mm->tc = 0;
2216   mm->tc_copy = true;
2217
2218   /* Inbound security check */
2219   mm->sec_check = true;
2220   mm->sec_check_frag = false;
2221
2222   /* ICMP6 Type 1, Code 5 for security check failure */
2223   mm->icmp6_enabled = false;
2224
2225   /* Inner or outer fragmentation */
2226   mm->frag_inner = false;
2227   mm->frag_ignore_df = false;
2228
2229   vec_validate (mm->domain_counters, MAP_N_DOMAIN_COUNTER - 1);
2230   mm->domain_counters[MAP_DOMAIN_COUNTER_RX].name = "rx";
2231   mm->domain_counters[MAP_DOMAIN_COUNTER_TX].name = "tx";
2232
2233   vlib_validate_simple_counter (&mm->icmp_relayed, 0);
2234   vlib_zero_simple_counter (&mm->icmp_relayed, 0);
2235
2236   /* IP4 virtual reassembly */
2237   mm->ip4_reass_hash_table = 0;
2238   mm->ip4_reass_pool = 0;
2239   mm->ip4_reass_lock =
2240     clib_mem_alloc_aligned (CLIB_CACHE_LINE_BYTES, CLIB_CACHE_LINE_BYTES);
2241   mm->ip4_reass_conf_ht_ratio = MAP_IP4_REASS_HT_RATIO_DEFAULT;
2242   mm->ip4_reass_conf_lifetime_ms = MAP_IP4_REASS_LIFETIME_DEFAULT;
2243   mm->ip4_reass_conf_pool_size = MAP_IP4_REASS_POOL_SIZE_DEFAULT;
2244   mm->ip4_reass_conf_buffers = MAP_IP4_REASS_BUFFERS_DEFAULT;
2245   mm->ip4_reass_ht_log2len =
2246     map_get_ht_log2len (mm->ip4_reass_conf_ht_ratio,
2247                         mm->ip4_reass_conf_pool_size);
2248   mm->ip4_reass_fifo_last = MAP_REASS_INDEX_NONE;
2249   map_ip4_reass_reinit (NULL, NULL);
2250
2251   /* IP6 virtual reassembly */
2252   mm->ip6_reass_hash_table = 0;
2253   mm->ip6_reass_pool = 0;
2254   mm->ip6_reass_lock =
2255     clib_mem_alloc_aligned (CLIB_CACHE_LINE_BYTES, CLIB_CACHE_LINE_BYTES);
2256   mm->ip6_reass_conf_ht_ratio = MAP_IP6_REASS_HT_RATIO_DEFAULT;
2257   mm->ip6_reass_conf_lifetime_ms = MAP_IP6_REASS_LIFETIME_DEFAULT;
2258   mm->ip6_reass_conf_pool_size = MAP_IP6_REASS_POOL_SIZE_DEFAULT;
2259   mm->ip6_reass_conf_buffers = MAP_IP6_REASS_BUFFERS_DEFAULT;
2260   mm->ip6_reass_ht_log2len =
2261     map_get_ht_log2len (mm->ip6_reass_conf_ht_ratio,
2262                         mm->ip6_reass_conf_pool_size);
2263   mm->ip6_reass_fifo_last = MAP_REASS_INDEX_NONE;
2264   map_ip6_reass_reinit (NULL, NULL);
2265
2266 #ifdef MAP_SKIP_IP6_LOOKUP
2267   fib_node_register_type (FIB_NODE_TYPE_MAP_E, &map_vft);
2268 #endif
2269   map_dpo_module_init ();
2270
2271   return 0;
2272 }
2273
2274 VLIB_INIT_FUNCTION (map_init);
2275
2276 /*
2277  * fd.io coding-style-patch-verification: ON
2278  *
2279  * Local Variables:
2280  * eval: (c-set-style "gnu")
2281  * End:
2282  */