d559565487e681eb99d0d9ecb384439aec2cac0d
[vpp.git] / src / plugins / ikev2 / ikev2.c
1 /*
2  * Copyright (c) 2015 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 <vlib/vlib.h>
17 #include <vlib/unix/plugin.h>
18 #include <vlibmemory/api.h>
19 #include <vpp/app/version.h>
20 #include <vnet/vnet.h>
21 #include <vppinfra/error.h>
22 #include <vppinfra/random.h>
23 #include <vnet/udp/udp.h>
24 #include <vnet/ipsec/ipsec.h>
25 #include <vnet/ipsec/ipsec_tun.h>
26 #include <vnet/ipip/ipip.h>
27 #include <plugins/ikev2/ikev2.h>
28 #include <plugins/ikev2/ikev2_priv.h>
29 #include <plugins/dns/dns.h>
30 #include <openssl/sha.h>
31 #include <vnet/ipsec/ipsec_punt.h>
32 #include <plugins/ikev2/ikev2.api_enum.h>
33
34 #define IKEV2_LIVENESS_RETRIES 3
35 #define IKEV2_LIVENESS_PERIOD_CHECK 30
36
37 ikev2_main_t ikev2_main;
38
39 static int ikev2_delete_tunnel_interface (vnet_main_t * vnm,
40                                           ikev2_sa_t * sa,
41                                           ikev2_child_sa_t * child);
42
43 #define ikev2_set_state(sa, v, ...) do { \
44     (sa)->state = v; \
45     ikev2_elog_sa_state("ispi %lx SA state changed to " #v __VA_ARGS__, sa->ispi); \
46   } while(0);
47
48 typedef struct
49 {
50   u32 next_index;
51   u32 sw_if_index;
52 } ikev2_trace_t;
53
54 static u8 *
55 format_ikev2_trace (u8 * s, va_list * args)
56 {
57   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
58   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
59   ikev2_trace_t *t = va_arg (*args, ikev2_trace_t *);
60
61   s = format (s, "ikev2: sw_if_index %d, next index %d",
62               t->sw_if_index, t->next_index);
63   return s;
64 }
65
66 #define IKEV2_GENERATE_SA_INIT_OK_str ""
67 #define IKEV2_GENERATE_SA_INIT_OK_ERR_NO_DH_STR \
68   "no DH group configured for IKE proposals!"
69 #define IKEV2_GENERATE_SA_INIT_OK_ERR_UNSUPP_STR \
70   "DH group not supported!"
71
72 typedef enum
73 {
74   IKEV2_GENERATE_SA_INIT_OK,
75   IKEV2_GENERATE_SA_INIT_ERR_NO_DH,
76   IKEV2_GENERATE_SA_INIT_ERR_UNSUPPORTED_DH,
77 } ikev2_generate_sa_error_t;
78
79 static u8 *
80 format_ikev2_gen_sa_error (u8 * s, va_list * args)
81 {
82   ikev2_generate_sa_error_t e = va_arg (*args, ikev2_generate_sa_error_t);
83   switch (e)
84     {
85     case IKEV2_GENERATE_SA_INIT_OK:
86       break;
87     case IKEV2_GENERATE_SA_INIT_ERR_NO_DH:
88       s = format (s, IKEV2_GENERATE_SA_INIT_OK_ERR_NO_DH_STR);
89       break;
90     case IKEV2_GENERATE_SA_INIT_ERR_UNSUPPORTED_DH:
91       s = format (s, IKEV2_GENERATE_SA_INIT_OK_ERR_UNSUPP_STR);
92       break;
93     }
94   return s;
95 }
96
97 typedef enum
98 {
99   IKEV2_NEXT_IP4_LOOKUP,
100   IKEV2_NEXT_IP4_ERROR_DROP,
101   IKEV2_IP4_N_NEXT,
102 } ikev2_ip4_next_t;
103
104 typedef enum
105 {
106   IKEV2_NEXT_IP6_LOOKUP,
107   IKEV2_NEXT_IP6_ERROR_DROP,
108   IKEV2_IP6_N_NEXT,
109 } ikev2_ip6_next_t;
110
111 typedef u32 ikev2_non_esp_marker;
112
113 static u16
114 ikev2_get_port (ikev2_sa_t *sa)
115 {
116   return ikev2_natt_active (sa) ? IKEV2_PORT_NATT : IKEV2_PORT;
117 }
118
119 static int
120 ikev2_insert_non_esp_marker (ike_header_t *ike, int len)
121 {
122   memmove ((u8 *) ike + sizeof (ikev2_non_esp_marker), ike, len);
123   clib_memset (ike, 0, sizeof (ikev2_non_esp_marker));
124   return len + sizeof (ikev2_non_esp_marker);
125 }
126
127 static ikev2_sa_transform_t *
128 ikev2_find_transform_data (ikev2_sa_transform_t * t)
129 {
130   ikev2_main_t *km = &ikev2_main;
131   ikev2_sa_transform_t *td;
132
133   vec_foreach (td, km->supported_transforms)
134   {
135     if (td->type != t->type)
136       continue;
137
138     if (td->transform_id != t->transform_id)
139       continue;
140
141     if (td->type == IKEV2_TRANSFORM_TYPE_ENCR)
142       {
143         if (vec_len (t->attrs) != 4 || t->attrs[0] != 0x80
144             || t->attrs[1] != 14)
145           continue;
146
147         if (((t->attrs[2] << 8 | t->attrs[3]) / 8) != td->key_len)
148           continue;
149       }
150     return td;
151   }
152   return 0;
153 }
154
155 static ikev2_sa_proposal_t *
156 ikev2_select_proposal (ikev2_sa_proposal_t * proposals,
157                        ikev2_protocol_id_t prot_id)
158 {
159   ikev2_sa_proposal_t *rv = 0;
160   ikev2_sa_proposal_t *proposal;
161   ikev2_sa_transform_t *transform, *new_t;
162   u8 mandatory_bitmap, optional_bitmap;
163
164   if (prot_id == IKEV2_PROTOCOL_IKE)
165     {
166       mandatory_bitmap = (1 << IKEV2_TRANSFORM_TYPE_ENCR) |
167         (1 << IKEV2_TRANSFORM_TYPE_PRF) | (1 << IKEV2_TRANSFORM_TYPE_DH);
168       optional_bitmap = mandatory_bitmap | (1 << IKEV2_TRANSFORM_TYPE_INTEG);
169     }
170   else if (prot_id == IKEV2_PROTOCOL_ESP)
171     {
172       mandatory_bitmap = (1 << IKEV2_TRANSFORM_TYPE_ENCR) |
173         (1 << IKEV2_TRANSFORM_TYPE_ESN);
174       optional_bitmap = mandatory_bitmap |
175         (1 << IKEV2_TRANSFORM_TYPE_INTEG) | (1 << IKEV2_TRANSFORM_TYPE_DH);
176     }
177   else if (prot_id == IKEV2_PROTOCOL_AH)
178     {
179       mandatory_bitmap = (1 << IKEV2_TRANSFORM_TYPE_INTEG) |
180         (1 << IKEV2_TRANSFORM_TYPE_ESN);
181       optional_bitmap = mandatory_bitmap | (1 << IKEV2_TRANSFORM_TYPE_DH);
182     }
183   else
184     return 0;
185
186   vec_add2 (rv, proposal, 1);
187
188   vec_foreach (proposal, proposals)
189   {
190     u8 bitmap = 0;
191     if (proposal->protocol_id != prot_id)
192       continue;
193
194     vec_foreach (transform, proposal->transforms)
195     {
196       if ((1 << transform->type) & bitmap)
197         continue;
198
199       if (ikev2_find_transform_data (transform))
200         {
201           bitmap |= 1 << transform->type;
202           vec_add2 (rv->transforms, new_t, 1);
203           clib_memcpy_fast (new_t, transform, sizeof (*new_t));
204           new_t->attrs = vec_dup (transform->attrs);
205         }
206     }
207
208     if ((bitmap & mandatory_bitmap) == mandatory_bitmap &&
209         (bitmap & ~optional_bitmap) == 0)
210       {
211         rv->proposal_num = proposal->proposal_num;
212         rv->protocol_id = proposal->protocol_id;
213         RAND_bytes ((u8 *) & rv->spi, sizeof (rv->spi));
214         if (rv->protocol_id != IKEV2_PROTOCOL_IKE)
215           rv->spi &= 0xffffffff;
216         goto done;
217       }
218     else
219       {
220         vec_free (rv->transforms);
221       }
222   }
223
224   vec_free (rv);
225 done:
226   return rv;
227 }
228
229 ikev2_sa_transform_t *
230 ikev2_sa_get_td_for_type (ikev2_sa_proposal_t * p,
231                           ikev2_transform_type_t type)
232 {
233   ikev2_sa_transform_t *t;
234
235   if (!p)
236     return 0;
237
238   vec_foreach (t, p->transforms)
239   {
240     if (t->type == type)
241       return ikev2_find_transform_data (t);
242   }
243   return 0;
244 }
245
246 ikev2_child_sa_t *
247 ikev2_sa_get_child (ikev2_sa_t * sa, u32 spi, ikev2_protocol_id_t prot_id,
248                     int by_initiator)
249 {
250   ikev2_child_sa_t *c;
251   vec_foreach (c, sa->childs)
252   {
253     ikev2_sa_proposal_t *proposal =
254       by_initiator ? &c->i_proposals[0] : &c->r_proposals[0];
255     if (proposal && proposal->spi == spi && proposal->protocol_id == prot_id)
256       return c;
257   }
258
259   return 0;
260 }
261
262 void
263 ikev2_sa_free_proposal_vector (ikev2_sa_proposal_t ** v)
264 {
265   ikev2_sa_proposal_t *p;
266   ikev2_sa_transform_t *t;
267
268   if (!*v)
269     return;
270
271   vec_foreach (p, *v)
272   {
273     vec_foreach (t, p->transforms)
274     {
275       vec_free (t->attrs);
276     }
277     vec_free (p->transforms);
278   }
279   vec_free (*v);
280 }
281
282 static void
283 ikev2_sa_free_child_sa (ikev2_child_sa_t * c)
284 {
285   ikev2_sa_free_proposal_vector (&c->r_proposals);
286   ikev2_sa_free_proposal_vector (&c->i_proposals);
287   vec_free (c->sk_ai);
288   vec_free (c->sk_ar);
289   vec_free (c->sk_ei);
290   vec_free (c->sk_er);
291   vec_free (c->tsi);
292   vec_free (c->tsr);
293 }
294
295 static void
296 ikev2_sa_free_all_child_sa (ikev2_child_sa_t ** childs)
297 {
298   ikev2_child_sa_t *c;
299   vec_foreach (c, *childs) ikev2_sa_free_child_sa (c);
300
301   vec_free (*childs);
302 }
303
304 static void
305 ikev2_sa_del_child_sa (ikev2_sa_t * sa, ikev2_child_sa_t * child)
306 {
307   ikev2_sa_free_child_sa (child);
308   vec_del1 (sa->childs, child - sa->childs);
309 }
310
311 static void
312 ikev2_sa_free_all_vec (ikev2_sa_t * sa)
313 {
314   vec_free (sa->i_nonce);
315   vec_free (sa->r_nonce);
316
317   vec_free (sa->dh_shared_key);
318   vec_free (sa->dh_private_key);
319   vec_free (sa->i_dh_data);
320   vec_free (sa->r_dh_data);
321
322   ikev2_sa_free_proposal_vector (&sa->r_proposals);
323   ikev2_sa_free_proposal_vector (&sa->i_proposals);
324
325   vec_free (sa->sk_d);
326   vec_free (sa->sk_ai);
327   vec_free (sa->sk_ar);
328   vec_free (sa->sk_ei);
329   vec_free (sa->sk_er);
330   vec_free (sa->sk_pi);
331   vec_free (sa->sk_pr);
332
333   vec_free (sa->i_id.data);
334   vec_free (sa->r_id.data);
335
336   vec_free (sa->i_auth.data);
337   if (sa->i_auth.key)
338     EVP_PKEY_free (sa->i_auth.key);
339   vec_free (sa->r_auth.data);
340   if (sa->r_auth.key)
341     EVP_PKEY_free (sa->r_auth.key);
342
343   vec_free (sa->del);
344
345   vec_free (sa->rekey);
346
347   vec_free (sa->last_sa_init_req_packet_data);
348   vec_free (sa->last_sa_init_res_packet_data);
349
350   vec_free (sa->last_res_packet_data);
351
352   ikev2_sa_free_all_child_sa (&sa->childs);
353 }
354
355 static void
356 ikev2_delete_sa (ikev2_main_per_thread_data_t * ptd, ikev2_sa_t * sa)
357 {
358   uword *p;
359
360   ikev2_sa_free_all_vec (sa);
361
362   p = hash_get (ptd->sa_by_rspi, sa->rspi);
363   if (p)
364     {
365       hash_unset (ptd->sa_by_rspi, sa->rspi);
366       pool_put (ptd->sas, sa);
367     }
368 }
369
370 static ikev2_generate_sa_error_t
371 ikev2_generate_sa_init_data (ikev2_sa_t * sa)
372 {
373   ikev2_sa_transform_t *t = 0, *t2;
374   ikev2_main_t *km = &ikev2_main;
375
376   if (sa->dh_group == IKEV2_TRANSFORM_DH_TYPE_NONE)
377     return IKEV2_GENERATE_SA_INIT_ERR_NO_DH;
378
379   /* check if received DH group is on our list of supported groups */
380   vec_foreach (t2, km->supported_transforms)
381   {
382     if (t2->type == IKEV2_TRANSFORM_TYPE_DH && sa->dh_group == t2->dh_type)
383       {
384         t = t2;
385         break;
386       }
387   }
388
389   if (!t)
390     {
391       sa->dh_group = IKEV2_TRANSFORM_DH_TYPE_NONE;
392       return IKEV2_GENERATE_SA_INIT_ERR_UNSUPPORTED_DH;
393     }
394
395   if (sa->is_initiator)
396     {
397       /* generate rspi */
398       RAND_bytes ((u8 *) & sa->ispi, 8);
399
400       /* generate nonce */
401       sa->i_nonce = vec_new (u8, IKEV2_NONCE_SIZE);
402       RAND_bytes ((u8 *) sa->i_nonce, IKEV2_NONCE_SIZE);
403     }
404   else
405     {
406       /* generate rspi */
407       RAND_bytes ((u8 *) & sa->rspi, 8);
408
409       /* generate nonce */
410       sa->r_nonce = vec_new (u8, vec_len (sa->i_nonce));
411       RAND_bytes ((u8 *) sa->r_nonce, vec_len (sa->i_nonce));
412     }
413
414   /* generate dh keys */
415   ikev2_generate_dh (sa, t);
416
417   return IKEV2_GENERATE_SA_INIT_OK;
418 }
419
420 static void
421 ikev2_complete_sa_data (ikev2_sa_t * sa, ikev2_sa_t * sai)
422 {
423   ikev2_sa_transform_t *t = 0, *t2;
424   ikev2_main_t *km = &ikev2_main;
425
426   /*move some data to the new SA */
427 #define _(A) ({void* __tmp__ = (A); (A) = 0; __tmp__;})
428   sa->i_nonce = _(sai->i_nonce);
429   sa->i_dh_data = _(sai->i_dh_data);
430   sa->dh_private_key = _(sai->dh_private_key);
431   ip_address_copy (&sa->iaddr, &sai->iaddr);
432   ip_address_copy (&sa->raddr, &sai->raddr);
433   sa->is_initiator = sai->is_initiator;
434   sa->i_id.type = sai->i_id.type;
435   sa->r_id.type = sai->r_id.type;
436   sa->profile_index = sai->profile_index;
437   sa->tun_itf = sai->tun_itf;
438   sa->is_tun_itf_set = sai->is_tun_itf_set;
439   if (sai->natt_state == IKEV2_NATT_DISABLED)
440     sa->natt_state = IKEV2_NATT_DISABLED;
441   sa->i_id.data = _(sai->i_id.data);
442   sa->r_id.data = _(sai->r_id.data);
443   sa->i_auth.method = sai->i_auth.method;
444   sa->i_auth.hex = sai->i_auth.hex;
445   sa->i_auth.data = _(sai->i_auth.data);
446   sa->i_auth.key = _(sai->i_auth.key);
447   sa->last_sa_init_req_packet_data = _(sai->last_sa_init_req_packet_data);
448   sa->last_init_msg_id = sai->last_init_msg_id;
449   sa->childs = _(sai->childs);
450   sa->udp_encap = sai->udp_encap;
451   sa->ipsec_over_udp_port = sai->ipsec_over_udp_port;
452   sa->dst_port = sai->dst_port;
453   sa->sw_if_index = sai->sw_if_index;
454 #undef _
455
456
457   if (sa->dh_group == IKEV2_TRANSFORM_DH_TYPE_NONE)
458     {
459       return;
460     }
461
462   /* check if received DH group is on our list of supported groups */
463   vec_foreach (t2, km->supported_transforms)
464   {
465     if (t2->type == IKEV2_TRANSFORM_TYPE_DH && sa->dh_group == t2->dh_type)
466       {
467         t = t2;
468         break;
469       }
470   }
471
472   if (!t)
473     {
474       sa->dh_group = IKEV2_TRANSFORM_DH_TYPE_NONE;
475       return;
476     }
477
478
479   /* generate dh keys */
480   ikev2_complete_dh (sa, t);
481
482 }
483
484 static void
485 ikev2_calc_keys_internal (ikev2_sa_t *sa, u8 *skeyseed)
486 {
487   u8 *tmp;
488   /* calculate SKEYSEED = prf(Ni | Nr, g^ir) */
489   u8 *s = 0;
490   u16 integ_key_len = 0, salt_len = 0;
491   ikev2_sa_transform_t *tr_encr, *tr_prf, *tr_integ;
492   tr_encr =
493     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_ENCR);
494   tr_prf =
495     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
496   tr_integ =
497     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_INTEG);
498
499   if (tr_integ)
500     integ_key_len = tr_integ->key_len;
501   else
502     salt_len = sizeof (u32);
503
504   vec_append (s, sa->i_nonce);
505   vec_append (s, sa->r_nonce);
506
507   /* Calculate S = Ni | Nr | SPIi | SPIr */
508   u64 *spi;
509   vec_add2 (s, tmp, 2 * sizeof (*spi));
510   spi = (u64 *) tmp;
511   spi[0] = clib_host_to_net_u64 (sa->ispi);
512   spi[1] = clib_host_to_net_u64 (sa->rspi);
513
514   /* calculate PRFplus */
515   u8 *keymat;
516   int len = tr_prf->key_trunc + /* SK_d */
517     integ_key_len * 2 +         /* SK_ai, SK_ar */
518     tr_encr->key_len * 2 +      /* SK_ei, SK_er */
519     tr_prf->key_len * 2 +       /* SK_pi, SK_pr */
520     salt_len * 2;
521
522   keymat = ikev2_calc_prfplus (tr_prf, skeyseed, s, len);
523   vec_free (s);
524
525   int pos = 0;
526
527   /* SK_d */
528   sa->sk_d = vec_new (u8, tr_prf->key_trunc);
529   clib_memcpy_fast (sa->sk_d, keymat + pos, tr_prf->key_trunc);
530   pos += tr_prf->key_trunc;
531
532   if (integ_key_len)
533     {
534       /* SK_ai */
535       sa->sk_ai = vec_new (u8, integ_key_len);
536       clib_memcpy_fast (sa->sk_ai, keymat + pos, integ_key_len);
537       pos += integ_key_len;
538
539       /* SK_ar */
540       sa->sk_ar = vec_new (u8, integ_key_len);
541       clib_memcpy_fast (sa->sk_ar, keymat + pos, integ_key_len);
542       pos += integ_key_len;
543     }
544
545   /* SK_ei */
546   sa->sk_ei = vec_new (u8, tr_encr->key_len + salt_len);
547   clib_memcpy_fast (sa->sk_ei, keymat + pos, tr_encr->key_len + salt_len);
548   pos += tr_encr->key_len + salt_len;
549
550   /* SK_er */
551   sa->sk_er = vec_new (u8, tr_encr->key_len + salt_len);
552   clib_memcpy_fast (sa->sk_er, keymat + pos, tr_encr->key_len + salt_len);
553   pos += tr_encr->key_len + salt_len;
554
555   /* SK_pi */
556   sa->sk_pi = vec_new (u8, tr_prf->key_len);
557   clib_memcpy_fast (sa->sk_pi, keymat + pos, tr_prf->key_len);
558   pos += tr_prf->key_len;
559
560   /* SK_pr */
561   sa->sk_pr = vec_new (u8, tr_prf->key_len);
562   clib_memcpy_fast (sa->sk_pr, keymat + pos, tr_prf->key_len);
563   pos += tr_prf->key_len;
564
565   vec_free (keymat);
566   sa->keys_generated = 1;
567 }
568
569 static void
570 ikev2_calc_keys_rekey (ikev2_sa_t *sa_new, ikev2_sa_t *sa_old)
571 {
572   u8 *s = 0, *skeyseed = 0;
573   ikev2_sa_transform_t *tr_prf =
574     ikev2_sa_get_td_for_type (sa_old->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
575
576   vec_append (s, sa_new->dh_shared_key);
577   vec_append (s, sa_new->i_nonce);
578   vec_append (s, sa_new->r_nonce);
579   skeyseed = ikev2_calc_prf (tr_prf, sa_old->sk_d, s);
580
581   ikev2_calc_keys_internal (sa_new, skeyseed);
582
583   vec_free (skeyseed);
584   vec_free (s);
585 }
586
587 static void
588 ikev2_calc_keys (ikev2_sa_t *sa)
589 {
590   u8 *s = 0, *skeyseed = 0;
591   ikev2_sa_transform_t *tr_prf =
592     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
593
594   vec_append (s, sa->i_nonce);
595   vec_append (s, sa->r_nonce);
596   skeyseed = ikev2_calc_prf (tr_prf, s, sa->dh_shared_key);
597
598   ikev2_calc_keys_internal (sa, skeyseed);
599
600   vec_free (skeyseed);
601   vec_free (s);
602 }
603
604 static void
605 ikev2_calc_child_keys (ikev2_sa_t *sa, ikev2_child_sa_t *child, u8 kex)
606 {
607   u8 *s = 0;
608   u16 integ_key_len = 0;
609   u8 salt_len = 0;
610
611   ikev2_sa_transform_t *tr_prf, *ctr_encr, *ctr_integ;
612   tr_prf =
613     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
614   ctr_encr =
615     ikev2_sa_get_td_for_type (child->r_proposals, IKEV2_TRANSFORM_TYPE_ENCR);
616   ctr_integ =
617     ikev2_sa_get_td_for_type (child->r_proposals, IKEV2_TRANSFORM_TYPE_INTEG);
618
619   if (ctr_integ)
620     integ_key_len = ctr_integ->key_len;
621   else
622     salt_len = sizeof (u32);
623
624   if (kex)
625     vec_append (s, sa->dh_shared_key);
626   vec_append (s, sa->i_nonce);
627   vec_append (s, sa->r_nonce);
628   /* calculate PRFplus */
629   u8 *keymat;
630   int len = ctr_encr->key_len * 2 + integ_key_len * 2 + salt_len * 2;
631
632   keymat = ikev2_calc_prfplus (tr_prf, sa->sk_d, s, len);
633
634   int pos = 0;
635
636   /* SK_ei */
637   child->sk_ei = vec_new (u8, ctr_encr->key_len);
638   clib_memcpy_fast (child->sk_ei, keymat + pos, ctr_encr->key_len);
639   pos += ctr_encr->key_len;
640
641   if (ctr_integ)
642     {
643       /* SK_ai */
644       child->sk_ai = vec_new (u8, ctr_integ->key_len);
645       clib_memcpy_fast (child->sk_ai, keymat + pos, ctr_integ->key_len);
646       pos += ctr_integ->key_len;
647     }
648   else
649     {
650       clib_memcpy (&child->salt_ei, keymat + pos, salt_len);
651       pos += salt_len;
652     }
653
654   /* SK_er */
655   child->sk_er = vec_new (u8, ctr_encr->key_len);
656   clib_memcpy_fast (child->sk_er, keymat + pos, ctr_encr->key_len);
657   pos += ctr_encr->key_len;
658
659   if (ctr_integ)
660     {
661       /* SK_ar */
662       child->sk_ar = vec_new (u8, integ_key_len);
663       clib_memcpy_fast (child->sk_ar, keymat + pos, integ_key_len);
664       pos += integ_key_len;
665     }
666   else
667     {
668       clib_memcpy (&child->salt_er, keymat + pos, salt_len);
669       pos += salt_len;
670     }
671
672   ASSERT (pos == len);
673
674   vec_free (keymat);
675 }
676
677 static u8 *
678 ikev2_compute_nat_sha1 (u64 ispi, u64 rspi, ip_address_t *ia, u16 port)
679 {
680   const u32 max_buf_size =
681     sizeof (ispi) + sizeof (rspi) + sizeof (ip6_address_t) + sizeof (u16);
682   u8 buf[max_buf_size];
683   u8 *res = vec_new (u8, 20);
684
685   clib_memcpy_fast (&buf[0], &ispi, sizeof (ispi));
686   clib_memcpy_fast (&buf[8], &rspi, sizeof (rspi));
687   clib_memcpy_fast (&buf[8 + 8], ip_addr_bytes (ia), ip_address_size (ia));
688   clib_memcpy_fast (&buf[8 + 8 + ip_address_size (ia)], &port, sizeof (port));
689   SHA1 (buf, 2 * sizeof (ispi) + sizeof (port) + ip_address_size (ia), res);
690   return res;
691 }
692
693 static int
694 ikev2_parse_ke_payload (const void *p, u32 rlen, ikev2_sa_t * sa,
695                         u8 ** ke_data)
696 {
697   const ike_ke_payload_header_t *ke = p;
698   u16 plen = clib_net_to_host_u16 (ke->length);
699   ASSERT (plen >= sizeof (*ke) && plen <= rlen);
700   if (sizeof (*ke) > rlen)
701     {
702       ikev2_elog_error ("KE: packet too small");
703       return 0;
704     }
705
706   sa->dh_group = clib_net_to_host_u16 (ke->dh_group);
707   vec_reset_length (ke_data[0]);
708   vec_add (ke_data[0], ke->payload, plen - sizeof (*ke));
709   return 1;
710 }
711
712 static int
713 ikev2_parse_nonce_payload (const void *p, u32 rlen, const u8 **nonce)
714 {
715   const ike_payload_header_t *ikep = p;
716   u16 plen = clib_net_to_host_u16 (ikep->length);
717   ASSERT (plen >= sizeof (*ikep) && plen <= rlen);
718   int len = plen - sizeof (*ikep);
719   ASSERT (len >= 16 && len <= 256);
720   if (PREDICT_FALSE (len < 16 || len > 256))
721     {
722       ikev2_elog_error ("NONCE: bad size");
723       return 0;
724     }
725   *nonce = ikep->payload;
726   return len;
727 }
728
729 static int
730 ikev2_check_payload_length (const ike_payload_header_t * ikep, int rlen,
731                             u16 * plen)
732 {
733   if (sizeof (*ikep) > rlen)
734     {
735       ikev2_elog_error ("payload: packet too small");
736       return 0;
737     }
738   *plen = clib_net_to_host_u16 (ikep->length);
739   if (*plen < sizeof (*ikep) || *plen > rlen)
740     {
741       ikev2_elog_error ("payload: bad size");
742       return 0;
743     }
744   return 1;
745 }
746
747 static int
748 ikev2_process_sa_init_req (vlib_main_t *vm, ikev2_sa_t *sa, ike_header_t *ike,
749                            udp_header_t *udp, u32 len, u32 sw_if_index)
750 {
751   int p = 0;
752   u8 payload = ike->nextpayload;
753   ike_payload_header_t *ikep;
754   u16 plen;
755
756   ikev2_elog_exchange ("ispi %lx rspi %lx IKE_INIT request received "
757                        "from ", clib_net_to_host_u64 (ike->ispi),
758                        clib_net_to_host_u64 (ike->rspi),
759                        ip_addr_v4 (&sa->iaddr).as_u32,
760                        ip_addr_version (&sa->iaddr) == AF_IP4);
761
762   sa->ispi = clib_net_to_host_u64 (ike->ispi);
763   sa->sw_if_index = sw_if_index;
764
765   /* store whole IKE payload - needed for PSK auth */
766   vec_reset_length (sa->last_sa_init_req_packet_data);
767   vec_add (sa->last_sa_init_req_packet_data, ike, len);
768
769   if (len < sizeof (*ike))
770     {
771       ikev2_elog_error ("IKE_INIT request too small");
772       return 0;
773     }
774
775   len -= sizeof (*ike);
776   while (p < len && payload != IKEV2_PAYLOAD_NONE)
777     {
778       ikep = (ike_payload_header_t *) & ike->payload[p];
779       int current_length = len - p;
780       if (!ikev2_check_payload_length (ikep, current_length, &plen))
781         return 0;
782
783       if (payload == IKEV2_PAYLOAD_SA)
784         {
785           ikev2_sa_free_proposal_vector (&sa->i_proposals);
786           sa->i_proposals = ikev2_parse_sa_payload (ikep, current_length);
787         }
788       else if (payload == IKEV2_PAYLOAD_KE)
789         {
790           if (!ikev2_parse_ke_payload (ikep, current_length, sa,
791                                        &sa->i_dh_data))
792             return 0;
793         }
794       else if (payload == IKEV2_PAYLOAD_NONCE)
795         {
796           const u8 *nonce;
797           int nonce_len;
798           vec_reset_length (sa->i_nonce);
799           if ((nonce_len = ikev2_parse_nonce_payload (ikep, current_length,
800                                                       &nonce)) <= 0)
801             return 0;
802           vec_add (sa->i_nonce, nonce, nonce_len);
803         }
804       else if (payload == IKEV2_PAYLOAD_NOTIFY)
805         {
806           ikev2_notify_t *n =
807             ikev2_parse_notify_payload (ikep, current_length);
808           if (n->msg_type == IKEV2_NOTIFY_MSG_NAT_DETECTION_SOURCE_IP)
809             {
810               u8 *src_sha = ikev2_compute_nat_sha1 (ike->ispi, 0, &sa->iaddr,
811                                                     udp->src_port);
812               if (clib_memcmp (src_sha, n->data, vec_len (src_sha)))
813                 {
814                   if (sa->natt_state == IKEV2_NATT_ENABLED)
815                     sa->natt_state = IKEV2_NATT_ACTIVE;
816                   ikev2_elog_uint (IKEV2_LOG_DEBUG, "ispi %lx initiator"
817                                    " behind NAT", sa->ispi);
818                 }
819               vec_free (src_sha);
820             }
821           else if (n->msg_type ==
822                    IKEV2_NOTIFY_MSG_NAT_DETECTION_DESTINATION_IP)
823             {
824               u8 *dst_sha = ikev2_compute_nat_sha1 (ike->ispi, 0, &sa->raddr,
825                                                     udp->dst_port);
826               if (clib_memcmp (dst_sha, n->data, vec_len (dst_sha)))
827                 {
828                   if (sa->natt_state == IKEV2_NATT_ENABLED)
829                     sa->natt_state = IKEV2_NATT_ACTIVE;
830                   ikev2_elog_uint (IKEV2_LOG_DEBUG, "ispi %lx responder"
831                                    " (self) behind NAT", sa->ispi);
832                 }
833               vec_free (dst_sha);
834             }
835           vec_free (n);
836         }
837       else if (payload == IKEV2_PAYLOAD_VENDOR)
838         {
839           ikev2_parse_vendor_payload (ikep);
840         }
841       else
842         {
843           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
844                            payload);
845           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
846             {
847               ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
848               sa->unsupported_cp = payload;
849               return 0;
850             }
851         }
852
853       payload = ikep->nextpayload;
854       p += plen;
855     }
856
857   ikev2_set_state (sa, IKEV2_STATE_SA_INIT);
858   return 1;
859 }
860
861 static void
862 ikev2_process_sa_init_resp (vlib_main_t * vm,
863                             ikev2_sa_t * sa, ike_header_t * ike,
864                             udp_header_t * udp, u32 len)
865 {
866   int p = 0;
867   u8 payload = ike->nextpayload;
868   ike_payload_header_t *ikep;
869   u16 plen;
870
871   sa->ispi = clib_net_to_host_u64 (ike->ispi);
872   sa->rspi = clib_net_to_host_u64 (ike->rspi);
873
874   ikev2_elog_exchange ("ispi %lx rspi %lx IKE_INIT response received "
875                        "from ", sa->ispi, sa->rspi,
876                        ip_addr_v4 (&sa->raddr).as_u32,
877                        ip_addr_version (&sa->raddr) == AF_IP4);
878
879   /* store whole IKE payload - needed for PSK auth */
880   vec_reset_length (sa->last_sa_init_res_packet_data);
881   vec_add (sa->last_sa_init_res_packet_data, ike, len);
882
883   if (sizeof (*ike) > len)
884     {
885       ikev2_elog_error ("IKE_INIT response too small");
886       return;
887     }
888
889   len -= sizeof (*ike);
890   while (p < len && payload != IKEV2_PAYLOAD_NONE)
891     {
892       int current_length = len - p;
893       ikep = (ike_payload_header_t *) & ike->payload[p];
894       if (!ikev2_check_payload_length (ikep, current_length, &plen))
895         return;
896
897       if (payload == IKEV2_PAYLOAD_SA)
898         {
899           ikev2_sa_free_proposal_vector (&sa->r_proposals);
900           sa->r_proposals = ikev2_parse_sa_payload (ikep, current_length);
901           if (sa->r_proposals)
902             {
903               ikev2_set_state (sa, IKEV2_STATE_SA_INIT);
904               ike->msgid =
905                 clib_host_to_net_u32 (clib_net_to_host_u32 (ike->msgid) + 1);
906             }
907         }
908       else if (payload == IKEV2_PAYLOAD_KE)
909         {
910           if (!ikev2_parse_ke_payload (ikep, current_length, sa,
911                                        &sa->r_dh_data))
912             return;
913         }
914       else if (payload == IKEV2_PAYLOAD_NONCE)
915         {
916           const u8 *nonce;
917           int nonce_len;
918           vec_reset_length (sa->r_nonce);
919           if ((nonce_len = ikev2_parse_nonce_payload (ikep, current_length,
920                                                       &nonce)) <= 0)
921             return;
922           vec_add (sa->r_nonce, nonce, nonce_len);
923         }
924       else if (payload == IKEV2_PAYLOAD_NOTIFY)
925         {
926           ikev2_notify_t *n =
927             ikev2_parse_notify_payload (ikep, current_length);
928           if (n->msg_type == IKEV2_NOTIFY_MSG_NAT_DETECTION_SOURCE_IP)
929             {
930               u8 *src_sha = ikev2_compute_nat_sha1 (ike->ispi, ike->rspi,
931                                                     &sa->raddr,
932                                                     udp->src_port);
933               if (clib_memcmp (src_sha, n->data, vec_len (src_sha)))
934                 {
935                   ikev2_elog_uint (IKEV2_LOG_DEBUG, "ispi %lx responder"
936                                    " behind NAT, unsupported", sa->ispi);
937                 }
938               vec_free (src_sha);
939             }
940           else if (n->msg_type ==
941                    IKEV2_NOTIFY_MSG_NAT_DETECTION_DESTINATION_IP)
942             {
943               u8 *dst_sha = ikev2_compute_nat_sha1 (ike->ispi, ike->rspi,
944                                                     &sa->iaddr,
945                                                     udp->dst_port);
946               if (clib_memcmp (dst_sha, n->data, vec_len (dst_sha)))
947                 {
948                   if (sa->natt_state == IKEV2_NATT_ENABLED)
949                     sa->natt_state = IKEV2_NATT_ACTIVE;
950                   ikev2_elog_uint (IKEV2_LOG_DEBUG, "ispi %lx initiator"
951                                    " (self) behind NAT", sa->ispi);
952                 }
953               vec_free (dst_sha);
954             }
955           vec_free (n);
956         }
957       else if (payload == IKEV2_PAYLOAD_VENDOR)
958         {
959           ikev2_parse_vendor_payload (ikep);
960         }
961       else
962         {
963           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
964                            payload);
965           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
966             {
967               ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
968               sa->unsupported_cp = payload;
969               return;
970             }
971         }
972
973       payload = ikep->nextpayload;
974       p += plen;
975     }
976 }
977
978 static u8 *
979 ikev2_decrypt_sk_payload (ikev2_sa_t * sa, ike_header_t * ike,
980                           u8 * payload, u32 rlen, u32 * out_len)
981 {
982   ikev2_main_per_thread_data_t *ptd = ikev2_get_per_thread_data ();
983   int p = 0;
984   u8 last_payload = 0, *hmac = 0, *plaintext = 0;
985   ike_payload_header_t *ikep = 0;
986   u16 plen = 0;
987   u32 dlen = 0;
988   ikev2_sa_transform_t *tr_integ;
989   ikev2_sa_transform_t *tr_encr;
990   tr_integ =
991     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_INTEG);
992   tr_encr =
993     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_ENCR);
994   int is_aead = tr_encr->encr_type == IKEV2_TRANSFORM_ENCR_TYPE_AES_GCM_16;
995
996   if (((!sa->sk_ar || !sa->sk_ai) && !is_aead) || (!sa->sk_ei || !sa->sk_er))
997     return 0;
998
999   if (rlen <= sizeof (*ike))
1000     return 0;
1001
1002   int len = rlen - sizeof (*ike);
1003   while (p < len &&
1004          *payload != IKEV2_PAYLOAD_NONE && last_payload != IKEV2_PAYLOAD_SK)
1005     {
1006       ikep = (ike_payload_header_t *) & ike->payload[p];
1007       int current_length = len - p;
1008       if (!ikev2_check_payload_length (ikep, current_length, &plen))
1009         return 0;
1010
1011       if (*payload == IKEV2_PAYLOAD_SK)
1012         {
1013           last_payload = *payload;
1014         }
1015       else
1016         {
1017           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
1018                            *payload);
1019           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
1020             {
1021               sa->unsupported_cp = *payload;
1022               return 0;
1023             }
1024         }
1025
1026       *payload = ikep->nextpayload;
1027       p += plen;
1028     }
1029
1030   if (last_payload != IKEV2_PAYLOAD_SK)
1031     {
1032       ikev2_elog_error ("Last payload must be SK");
1033       return 0;
1034     }
1035
1036   if (is_aead)
1037     {
1038       if (plen < sizeof (*ikep) + IKEV2_GCM_ICV_SIZE)
1039         return 0;
1040
1041       plen -= sizeof (*ikep) + IKEV2_GCM_ICV_SIZE;
1042       u8 *aad = (u8 *) ike;
1043       u32 aad_len = ikep->payload - aad;
1044       u8 *tag = ikep->payload + plen;
1045
1046       int rc = ikev2_decrypt_aead_data (ptd, sa, tr_encr, ikep->payload,
1047                                         plen, aad, aad_len, tag, &dlen);
1048       if (rc)
1049         {
1050           *out_len = dlen;
1051           plaintext = ikep->payload + IKEV2_GCM_IV_SIZE;
1052         }
1053     }
1054   else
1055     {
1056       if (rlen < tr_integ->key_trunc)
1057         return 0;
1058
1059       hmac =
1060         ikev2_calc_integr (tr_integ, sa->is_initiator ? sa->sk_ar : sa->sk_ai,
1061                            (u8 *) ike, rlen - tr_integ->key_trunc);
1062
1063       if (plen < sizeof (*ikep) + tr_integ->key_trunc)
1064         return 0;
1065
1066       plen = plen - sizeof (*ikep) - tr_integ->key_trunc;
1067
1068       if (clib_memcmp (hmac, &ikep->payload[plen], tr_integ->key_trunc))
1069         {
1070           ikev2_elog_error ("message integrity check failed");
1071           vec_free (hmac);
1072           return 0;
1073         }
1074       vec_free (hmac);
1075
1076       int rc = ikev2_decrypt_data (ptd, sa, tr_encr, ikep->payload, plen,
1077                                    &dlen);
1078       if (rc)
1079         {
1080           *out_len = dlen;
1081           plaintext = ikep->payload + tr_encr->block_size;
1082         }
1083     }
1084
1085   return plaintext;
1086 }
1087
1088 static int
1089 ikev2_is_id_equal (const ikev2_id_t *i1, const ikev2_id_t *i2)
1090 {
1091   if (i1->type != i2->type)
1092     return 0;
1093
1094   if (vec_len (i1->data) != vec_len (i2->data))
1095     return 0;
1096
1097   if (clib_memcmp (i1->data, i2->data, vec_len (i1->data)))
1098     return 0;
1099
1100   return 1;
1101 }
1102
1103 static void
1104 ikev2_initial_contact_cleanup_internal (ikev2_main_per_thread_data_t * ptd,
1105                                         ikev2_sa_t * sa)
1106 {
1107   ikev2_main_t *km = &ikev2_main;
1108   ikev2_sa_t *tmp;
1109   u32 i, *delete = 0;
1110   ikev2_child_sa_t *c;
1111
1112   /* find old IKE SAs with the same authenticated identity */
1113   pool_foreach (tmp, ptd->sas)  {
1114     if (!ikev2_is_id_equal (&tmp->i_id, &sa->i_id)
1115         || !ikev2_is_id_equal(&tmp->r_id, &sa->r_id))
1116       continue;
1117
1118     if (sa->rspi != tmp->rspi)
1119       vec_add1(delete, tmp - ptd->sas);
1120   }
1121
1122   for (i = 0; i < vec_len (delete); i++)
1123     {
1124       tmp = pool_elt_at_index (ptd->sas, delete[i]);
1125       vec_foreach (c, tmp->childs)
1126       {
1127         ikev2_delete_tunnel_interface (km->vnet_main, tmp, c);
1128       }
1129       ikev2_delete_sa (ptd, tmp);
1130     }
1131
1132   vec_free (delete);
1133   sa->initial_contact = 0;
1134 }
1135
1136 static void
1137 ikev2_initial_contact_cleanup (ikev2_main_per_thread_data_t * ptd,
1138                                ikev2_sa_t * sa)
1139 {
1140   ikev2_main_t *km = &ikev2_main;
1141
1142   if (!sa->initial_contact)
1143     return;
1144
1145   if (ptd)
1146     {
1147       ikev2_initial_contact_cleanup_internal (ptd, sa);
1148     }
1149   else
1150     {
1151       vec_foreach (ptd, km->per_thread_data)
1152         ikev2_initial_contact_cleanup_internal (ptd, sa);
1153     }
1154   sa->initial_contact = 0;
1155 }
1156
1157 static int
1158 ikev2_parse_id_payload (const void *p, u16 rlen, ikev2_id_t * sa_id)
1159 {
1160   const ike_id_payload_header_t *id = p;
1161   u16 plen = clib_net_to_host_u16 (id->length);
1162   if (plen < sizeof (*id) || plen > rlen)
1163     return 0;
1164
1165   sa_id->type = id->id_type;
1166   vec_reset_length (sa_id->data);
1167   vec_add (sa_id->data, id->payload, plen - sizeof (*id));
1168
1169   return 1;
1170 }
1171
1172 static int
1173 ikev2_parse_auth_payload (const void *p, u32 rlen, ikev2_auth_t * a)
1174 {
1175   const ike_auth_payload_header_t *ah = p;
1176   u16 plen = clib_net_to_host_u16 (ah->length);
1177
1178   a->method = ah->auth_method;
1179   vec_reset_length (a->data);
1180   vec_add (a->data, ah->payload, plen - sizeof (*ah));
1181   return 1;
1182 }
1183
1184 static int
1185 ikev2_process_auth_req (vlib_main_t * vm, ikev2_sa_t * sa,
1186                         ike_header_t * ike, u32 len)
1187 {
1188   int p = 0;
1189   ikev2_child_sa_t *first_child_sa;
1190   u8 payload = ike->nextpayload;
1191   u8 *plaintext = 0;
1192   ike_payload_header_t *ikep;
1193   u16 plen;
1194   u32 dlen = 0;
1195
1196   ikev2_elog_exchange ("ispi %lx rspi %lx EXCHANGE_IKE_AUTH received "
1197                        "from ", clib_host_to_net_u64 (ike->ispi),
1198                        clib_host_to_net_u64 (ike->rspi),
1199                        sa->is_initiator ?
1200                        ip_addr_v4 (&sa->raddr).as_u32 :
1201                        ip_addr_v4 (&sa->iaddr).as_u32,
1202                        ip_addr_version (&sa->raddr) == AF_IP4);
1203
1204   ikev2_calc_keys (sa);
1205
1206   plaintext = ikev2_decrypt_sk_payload (sa, ike, &payload, len, &dlen);
1207
1208   if (!plaintext)
1209     {
1210       if (sa->unsupported_cp)
1211         {
1212           ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
1213           return 0;
1214         }
1215       goto malformed;
1216     }
1217
1218   /* select or create 1st child SA */
1219   if (sa->is_initiator)
1220     {
1221       first_child_sa = &sa->childs[0];
1222     }
1223   else
1224     {
1225       ikev2_sa_free_all_child_sa (&sa->childs);
1226       vec_add2 (sa->childs, first_child_sa, 1);
1227     }
1228
1229
1230   /* process encrypted payload */
1231   while (p < dlen && payload != IKEV2_PAYLOAD_NONE)
1232     {
1233       ikep = (ike_payload_header_t *) & plaintext[p];
1234       int current_length = dlen - p;
1235       if (!ikev2_check_payload_length (ikep, current_length, &plen))
1236         goto malformed;
1237
1238       if (payload == IKEV2_PAYLOAD_SA)  /* 33 */
1239         {
1240           if (sa->is_initiator)
1241             {
1242               ikev2_sa_free_proposal_vector (&first_child_sa->r_proposals);
1243               first_child_sa->r_proposals = ikev2_parse_sa_payload (ikep,
1244                                                                     current_length);
1245             }
1246           else
1247             {
1248               ikev2_sa_free_proposal_vector (&first_child_sa->i_proposals);
1249               first_child_sa->i_proposals = ikev2_parse_sa_payload (ikep,
1250                                                                     current_length);
1251             }
1252         }
1253       else if (payload == IKEV2_PAYLOAD_IDI)    /* 35 */
1254         {
1255           if (!ikev2_parse_id_payload (ikep, current_length, &sa->i_id))
1256             goto malformed;
1257         }
1258       else if (payload == IKEV2_PAYLOAD_IDR)    /* 36 */
1259         {
1260           if (!ikev2_parse_id_payload (ikep, current_length, &sa->r_id))
1261             goto malformed;
1262         }
1263       else if (payload == IKEV2_PAYLOAD_AUTH)   /* 39 */
1264         {
1265           if (sa->is_initiator)
1266             {
1267               if (!ikev2_parse_auth_payload (ikep, current_length,
1268                                              &sa->r_auth))
1269                 goto malformed;
1270             }
1271           else
1272             {
1273               if (!ikev2_parse_auth_payload (ikep, current_length,
1274                                              &sa->i_auth))
1275                 goto malformed;
1276             }
1277         }
1278       else if (payload == IKEV2_PAYLOAD_NOTIFY) /* 41 */
1279         {
1280           ikev2_notify_t *n =
1281             ikev2_parse_notify_payload (ikep, current_length);
1282           if (n->msg_type == IKEV2_NOTIFY_MSG_INITIAL_CONTACT)
1283             {
1284               sa->initial_contact = 1;
1285             }
1286           vec_free (n);
1287         }
1288       else if (payload == IKEV2_PAYLOAD_VENDOR) /* 43 */
1289         {
1290           ikev2_parse_vendor_payload (ikep);
1291         }
1292       else if (payload == IKEV2_PAYLOAD_TSI)    /* 44 */
1293         {
1294           vec_free (first_child_sa->tsi);
1295           first_child_sa->tsi = ikev2_parse_ts_payload (ikep, current_length);
1296         }
1297       else if (payload == IKEV2_PAYLOAD_TSR)    /* 45 */
1298         {
1299           vec_free (first_child_sa->tsr);
1300           first_child_sa->tsr = ikev2_parse_ts_payload (ikep, current_length);
1301         }
1302       else
1303         {
1304           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
1305                            payload);
1306
1307           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
1308             {
1309               ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
1310               sa->unsupported_cp = payload;
1311               return 0;
1312             }
1313         }
1314
1315       payload = ikep->nextpayload;
1316       p += plen;
1317     }
1318
1319   return 1;
1320
1321 malformed:
1322   ikev2_set_state (sa, IKEV2_STATE_DELETED, ": malformed IKE_AUTH");
1323   return 0;
1324 }
1325
1326 static int
1327 ikev2_process_informational_req (vlib_main_t * vm,
1328                                  ikev2_sa_t * sa, ike_header_t * ike, u32 len)
1329 {
1330   int p = 0;
1331   u8 payload = ike->nextpayload;
1332   u8 *plaintext = 0;
1333   ike_payload_header_t *ikep;
1334   u32 dlen = 0;
1335   ikev2_notify_t *n = 0;
1336
1337   sa->liveness_retries = 0;
1338   ikev2_elog_exchange ("ispi %lx rspi %lx INFORMATIONAL received "
1339                        "from ", clib_host_to_net_u64 (ike->ispi),
1340                        clib_host_to_net_u64 (ike->rspi),
1341                        ip_addr_v4 (&sa->iaddr).as_u32,
1342                        ip_addr_version (&sa->iaddr) == AF_IP4);
1343
1344   plaintext = ikev2_decrypt_sk_payload (sa, ike, &payload, len, &dlen);
1345
1346   if (!plaintext)
1347     return 0;
1348
1349   /* process encrypted payload */
1350   p = 0;
1351   while (p < dlen && payload != IKEV2_PAYLOAD_NONE)
1352     {
1353       u32 current_length = dlen - p;
1354       if (p + sizeof (*ikep) > dlen)
1355         return 0;
1356
1357       ikep = (ike_payload_header_t *) & plaintext[p];
1358       u16 plen = clib_net_to_host_u16 (ikep->length);
1359
1360       if (plen < sizeof (*ikep) || plen > current_length)
1361         return 0;
1362
1363       if (payload == IKEV2_PAYLOAD_NOTIFY)      /* 41 */
1364         {
1365           n = ikev2_parse_notify_payload (ikep, current_length);
1366           if (!n)
1367             return 0;
1368           if (n->msg_type == IKEV2_NOTIFY_MSG_AUTHENTICATION_FAILED)
1369             ikev2_set_state (sa, IKEV2_STATE_AUTH_FAILED);
1370           vec_free (n);
1371         }
1372       else if (payload == IKEV2_PAYLOAD_DELETE) /* 42 */
1373         {
1374           sa->del = ikev2_parse_delete_payload (ikep, current_length);
1375         }
1376       else if (payload == IKEV2_PAYLOAD_VENDOR) /* 43 */
1377         {
1378           ikev2_parse_vendor_payload (ikep);
1379         }
1380       else
1381         {
1382           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
1383                            payload);
1384           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
1385             {
1386               sa->unsupported_cp = payload;
1387               return 0;
1388             }
1389         }
1390       payload = ikep->nextpayload;
1391       p += plen;
1392     }
1393   return 1;
1394 }
1395
1396 static int
1397 ikev2_process_create_child_sa_rekey (ikev2_sa_t *sa, ikev2_sa_t *sar,
1398                                      ikev2_rekey_t *rekey,
1399                                      ikev2_sa_proposal_t *proposal,
1400                                      ikev2_ts_t *tsi, ikev2_ts_t *tsr,
1401                                      const u8 *nonce, int nonce_len)
1402 {
1403   ikev2_sa_transform_t *tr;
1404
1405   rekey->i_proposal = proposal;
1406   rekey->r_proposal = ikev2_select_proposal (proposal, IKEV2_PROTOCOL_ESP);
1407
1408   if (sar->dh_group)
1409     {
1410       tr =
1411         ikev2_sa_get_td_for_type (rekey->r_proposal, IKEV2_TRANSFORM_TYPE_DH);
1412
1413       if (!tr || tr->dh_type != sar->dh_group)
1414         {
1415           rekey->notify_type = IKEV2_NOTIFY_MSG_INVALID_KE_PAYLOAD;
1416           ikev2_sa_free_proposal_vector (&rekey->r_proposal);
1417           return 0;
1418         }
1419
1420       vec_free (sa->dh_shared_key);
1421       vec_free (sa->dh_private_key);
1422       vec_free (sa->i_dh_data);
1423       vec_free (sa->r_dh_data);
1424
1425       sa->dh_group = sar->dh_group;
1426       sa->i_dh_data = sar->i_dh_data;
1427       sar->i_dh_data = 0;
1428
1429       ikev2_generate_dh (sa, tr);
1430       rekey->kex = 1;
1431     }
1432
1433   vec_reset_length (sa->i_nonce);
1434   vec_add (sa->i_nonce, nonce, nonce_len);
1435
1436   vec_validate (sa->r_nonce, nonce_len - 1);
1437   RAND_bytes ((u8 *) sa->r_nonce, nonce_len);
1438
1439   rekey->tsi = tsi;
1440   rekey->tsr = tsr;
1441
1442   return 1;
1443 }
1444
1445 static void
1446 ikev2_complete_sa_rekey (ikev2_sa_t *sa_new, ikev2_sa_t *sa_old,
1447                          ikev2_sa_rekey_t *sa_rekey)
1448 {
1449   sa_new->del = 0;
1450   sa_new->rekey = 0;
1451   sa_new->new_child = 0;
1452   sa_new->sa_rekey = 0;
1453   sa_new->last_sa_init_req_packet_data = 0;
1454   sa_new->last_sa_init_res_packet_data = 0;
1455   sa_new->last_msg_id = ~0;
1456   sa_new->last_res_packet_data = 0;
1457   sa_new->last_init_msg_id = 0;
1458   clib_memset (&sa_new->stats, 0, sizeof (sa_new->stats));
1459
1460   sa_new->ispi = sa_rekey->ispi;
1461   sa_new->rspi = sa_rekey->rspi;
1462   sa_new->i_nonce = sa_rekey->i_nonce;
1463   sa_new->r_nonce = sa_rekey->r_nonce;
1464   sa_new->dh_group = sa_rekey->dh_group;
1465   sa_new->dh_shared_key = sa_rekey->dh_shared_key;
1466   sa_new->dh_private_key = sa_rekey->dh_private_key;
1467   sa_new->i_dh_data = sa_rekey->i_dh_data;
1468   sa_new->r_dh_data = sa_rekey->r_dh_data;
1469   sa_new->i_proposals = sa_rekey->i_proposals;
1470   sa_new->r_proposals = sa_rekey->r_proposals;
1471
1472   sa_new->sk_d = 0;
1473   sa_new->sk_ai = 0;
1474   sa_new->sk_ar = 0;
1475   sa_new->sk_ei = 0;
1476   sa_new->sk_er = 0;
1477   sa_new->sk_pi = 0;
1478   sa_new->sk_pr = 0;
1479   ikev2_calc_keys_rekey (sa_new, sa_old);
1480
1481   sa_new->i_auth.data = vec_dup (sa_old->i_auth.data);
1482   sa_new->i_auth.key = sa_old->i_auth.key;
1483   if (sa_new->i_auth.key)
1484     EVP_PKEY_up_ref (sa_new->i_auth.key);
1485
1486   sa_new->r_auth.data = vec_dup (sa_old->r_auth.data);
1487   sa_new->r_auth.key = sa_old->r_auth.key;
1488   if (sa_new->r_auth.key)
1489     EVP_PKEY_up_ref (sa_new->r_auth.key);
1490
1491   sa_new->i_id.data = vec_dup (sa_old->i_id.data);
1492   sa_new->r_id.data = vec_dup (sa_old->r_id.data);
1493
1494   sa_old->is_tun_itf_set = 0;
1495   sa_old->tun_itf = ~0;
1496   sa_old->old_id_expiration = 0;
1497   sa_old->current_remote_id_mask = 0;
1498   sa_old->old_remote_id = 0;
1499   sa_old->old_remote_id_present = 0;
1500   sa_old->childs = 0;
1501   sa_old->sw_if_index = ~0;
1502 }
1503
1504 static void
1505 ikev2_process_sa_rekey (ikev2_sa_t *sa_new, ikev2_sa_t *sa_old,
1506                         ikev2_sa_rekey_t *sa_rekey)
1507 {
1508   ikev2_sa_transform_t *tr;
1509
1510   if (ikev2_generate_sa_init_data (sa_new) != IKEV2_GENERATE_SA_INIT_OK)
1511     {
1512       sa_rekey->notify_type = IKEV2_NOTIFY_MSG_INVALID_KE_PAYLOAD;
1513       return;
1514     }
1515
1516   sa_new->r_proposals =
1517     ikev2_select_proposal (sa_new->i_proposals, IKEV2_PROTOCOL_IKE);
1518
1519   tr = ikev2_sa_get_td_for_type (sa_new->r_proposals, IKEV2_TRANSFORM_TYPE_DH);
1520   if (!tr || tr->dh_type != sa_new->dh_group)
1521     {
1522       sa_rekey->notify_type = IKEV2_NOTIFY_MSG_INVALID_KE_PAYLOAD;
1523       return;
1524     }
1525
1526   sa_rekey->notify_type = 0;
1527   sa_rekey->ispi = sa_new->i_proposals[0].spi;
1528   sa_rekey->rspi = sa_new->r_proposals[0].spi;
1529   sa_rekey->i_nonce = sa_new->i_nonce;
1530   sa_rekey->r_nonce = sa_new->r_nonce;
1531   sa_rekey->dh_group = sa_new->dh_group;
1532   sa_rekey->dh_shared_key = sa_new->dh_shared_key;
1533   sa_rekey->dh_private_key = sa_new->dh_private_key;
1534   sa_rekey->i_dh_data = sa_new->i_dh_data;
1535   sa_rekey->r_dh_data = sa_new->r_dh_data;
1536   sa_rekey->i_proposals = sa_new->i_proposals;
1537   sa_rekey->r_proposals = sa_new->r_proposals;
1538
1539   sa_new->i_nonce = 0;
1540   sa_new->r_nonce = 0;
1541   sa_new->dh_shared_key = 0;
1542   sa_new->dh_private_key = 0;
1543   sa_new->i_dh_data = 0;
1544   sa_new->r_dh_data = 0;
1545   sa_new->i_proposals = 0;
1546   sa_new->r_proposals = 0;
1547 }
1548
1549 static int
1550 ikev2_process_create_child_sa_req (vlib_main_t * vm,
1551                                    ikev2_sa_t * sa, ike_header_t * ike,
1552                                    u32 len)
1553 {
1554   int p = 0;
1555   u8 payload = ike->nextpayload;
1556   u8 *plaintext = 0;
1557   ikev2_rekey_t *rekey;
1558   ike_payload_header_t *ikep;
1559   ikev2_notify_t *n = 0;
1560   ikev2_ts_t *tsi = 0;
1561   ikev2_ts_t *tsr = 0;
1562   ikev2_sa_proposal_t *proposal = 0;
1563   ikev2_child_sa_t *child_sa;
1564   u32 dlen = 0, src;
1565   u16 plen;
1566   const u8 *nonce = 0;
1567   int nonce_len = 0;
1568   ikev2_sa_t sar;
1569
1570   clib_memset (&sar, 0, sizeof (sar));
1571
1572   if (sa->is_initiator)
1573     src = ip_addr_v4 (&sa->raddr).as_u32;
1574   else
1575     src = ip_addr_v4 (&sa->iaddr).as_u32;
1576
1577   ikev2_elog_exchange ("ispi %lx rspi %lx CREATE_CHILD_SA received from",
1578                        clib_host_to_net_u64 (ike->ispi),
1579                        clib_host_to_net_u64 (ike->rspi), src,
1580                        ip_addr_version (&sa->raddr) == AF_IP4);
1581
1582   plaintext = ikev2_decrypt_sk_payload (sa, ike, &payload, len, &dlen);
1583
1584   if (!plaintext)
1585     goto cleanup_and_exit;
1586
1587   /* process encrypted payload */
1588   p = 0;
1589   while (payload != IKEV2_PAYLOAD_NONE)
1590     {
1591       ikep = (ike_payload_header_t *) & plaintext[p];
1592       int current_length = dlen - p;
1593       if (!ikev2_check_payload_length (ikep, current_length, &plen))
1594         goto cleanup_and_exit;
1595
1596       if (payload == IKEV2_PAYLOAD_SA)
1597         {
1598           proposal = ikev2_parse_sa_payload (ikep, current_length);
1599         }
1600       else if (payload == IKEV2_PAYLOAD_KE)
1601         {
1602           if (!ikev2_parse_ke_payload (ikep, current_length, &sar,
1603                                        &sar.i_dh_data))
1604             goto cleanup_and_exit;
1605         }
1606       else if (payload == IKEV2_PAYLOAD_NOTIFY)
1607         {
1608           ikev2_notify_t *n0;
1609           n0 = ikev2_parse_notify_payload (ikep, current_length);
1610           if (n0->msg_type == IKEV2_NOTIFY_MSG_REKEY_SA)
1611             {
1612               vec_free (n);
1613               n = n0;
1614             }
1615           else
1616             vec_free (n0);
1617         }
1618       else if (payload == IKEV2_PAYLOAD_DELETE)
1619         {
1620           sa->del = ikev2_parse_delete_payload (ikep, current_length);
1621         }
1622       else if (payload == IKEV2_PAYLOAD_VENDOR)
1623         {
1624           ikev2_parse_vendor_payload (ikep);
1625         }
1626       else if (payload == IKEV2_PAYLOAD_NONCE)
1627         {
1628           nonce_len = ikev2_parse_nonce_payload (ikep, current_length, &nonce);
1629           if (nonce_len <= 0)
1630             goto cleanup_and_exit;
1631         }
1632       else if (payload == IKEV2_PAYLOAD_TSI)
1633         {
1634           tsi = ikev2_parse_ts_payload (ikep, current_length);
1635         }
1636       else if (payload == IKEV2_PAYLOAD_TSR)
1637         {
1638           tsr = ikev2_parse_ts_payload (ikep, current_length);
1639         }
1640       else
1641         {
1642           ikev2_elog_uint (IKEV2_LOG_ERROR, "Unknown payload! type=%d",
1643                            payload);
1644           if (ikep->flags & IKEV2_PAYLOAD_FLAG_CRITICAL)
1645             {
1646               sa->unsupported_cp = payload;
1647               goto cleanup_and_exit;
1648             }
1649         }
1650       payload = ikep->nextpayload;
1651       p += plen;
1652     }
1653
1654   if (!proposal || !nonce ||
1655       (proposal->protocol_id != IKEV2_PROTOCOL_ESP &&
1656        proposal->protocol_id != IKEV2_PROTOCOL_IKE))
1657     goto cleanup_and_exit;
1658
1659   if (sa->is_initiator)
1660     {
1661       rekey = sa->rekey;
1662       if (vec_len (rekey) == 0)
1663         goto cleanup_and_exit;
1664       rekey->notify_type = 0;
1665       rekey->protocol_id = proposal->protocol_id;
1666       rekey->i_proposal =
1667         ikev2_select_proposal (proposal, IKEV2_PROTOCOL_ESP);
1668       rekey->i_proposal->spi = rekey->spi;
1669       rekey->r_proposal = proposal;
1670       rekey->tsi = tsi;
1671       rekey->tsr = tsr;
1672       /* update Nr */
1673       vec_reset_length (sa->r_nonce);
1674       vec_add (sa->r_nonce, nonce, nonce_len);
1675       child_sa = ikev2_sa_get_child (sa, rekey->ispi, IKEV2_PROTOCOL_ESP, 1);
1676       if (child_sa)
1677         {
1678           child_sa->rekey_retries = 0;
1679         }
1680     }
1681   else
1682     {
1683       if (n)
1684         {
1685           child_sa = ikev2_sa_get_child (sa, n->spi, n->protocol_id, 1);
1686           if (!child_sa)
1687             {
1688               ikev2_elog_uint (IKEV2_LOG_ERROR, "child SA spi %lx not found",
1689                                n->spi);
1690               goto cleanup_and_exit;
1691             }
1692           vec_add2 (sa->rekey, rekey, 1);
1693           rekey->notify_type = 0;
1694           rekey->kex = 0;
1695           rekey->protocol_id = n->protocol_id;
1696           rekey->spi = n->spi;
1697           if (sa->old_remote_id_present)
1698             {
1699               rekey->notify_type = IKEV2_NOTIFY_MSG_TEMPORARY_FAILURE;
1700               vec_free (proposal);
1701               vec_free (tsr);
1702               vec_free (tsi);
1703             }
1704           else if (!ikev2_process_create_child_sa_rekey (
1705                      sa, &sar, rekey, proposal, tsi, tsr, nonce, nonce_len))
1706             {
1707               vec_free (proposal);
1708               vec_free (tsr);
1709               vec_free (tsi);
1710             }
1711         }
1712       else if (proposal[0].protocol_id == IKEV2_PROTOCOL_IKE)
1713         {
1714           ikev2_sa_rekey_t *sa_rekey;
1715           if (tsi || tsr)
1716             goto cleanup_and_exit;
1717           sar.i_proposals = proposal;
1718           vec_add (sar.i_nonce, nonce, nonce_len);
1719           vec_add2 (sa->sa_rekey, sa_rekey, 1);
1720           ikev2_process_sa_rekey (&sar, sa, sa_rekey);
1721         }
1722       else
1723         {
1724           /* create new child SA */
1725           vec_add2 (sa->new_child, rekey, 1);
1726           rekey->notify_type = 0;
1727           rekey->kex = 0;
1728           if (!ikev2_process_create_child_sa_rekey (
1729                 sa, &sar, rekey, proposal, tsi, tsr, nonce, nonce_len))
1730             {
1731               vec_free (proposal);
1732               vec_free (tsr);
1733               vec_free (tsi);
1734             }
1735         }
1736     }
1737   vec_free (n);
1738   ikev2_sa_free_all_vec (&sar);
1739   return 1;
1740
1741 cleanup_and_exit:
1742   vec_free (n);
1743   vec_free (proposal);
1744   vec_free (tsr);
1745   vec_free (tsi);
1746   ikev2_sa_free_all_vec (&sar);
1747   return 0;
1748 }
1749
1750 static u8 *
1751 ikev2_sa_generate_authmsg (ikev2_sa_t * sa, int is_responder)
1752 {
1753   u8 *authmsg = 0;
1754   u8 *data;
1755   u8 *nonce;
1756   ikev2_id_t *id;
1757   u8 *key;
1758   u8 *packet_data;
1759   ikev2_sa_transform_t *tr_prf;
1760
1761   tr_prf =
1762     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
1763
1764   if (is_responder)
1765     {
1766       id = &sa->r_id;
1767       key = sa->sk_pr;
1768       nonce = sa->i_nonce;
1769       packet_data = sa->last_sa_init_res_packet_data;
1770     }
1771   else
1772     {
1773       id = &sa->i_id;
1774       key = sa->sk_pi;
1775       nonce = sa->r_nonce;
1776       packet_data = sa->last_sa_init_req_packet_data;
1777     }
1778
1779   data = vec_new (u8, 4);
1780   data[0] = id->type;
1781   vec_append (data, id->data);
1782
1783   u8 *id_hash = ikev2_calc_prf (tr_prf, key, data);
1784   vec_append (authmsg, packet_data);
1785   vec_append (authmsg, nonce);
1786   vec_append (authmsg, id_hash);
1787   vec_free (id_hash);
1788   vec_free (data);
1789
1790   return authmsg;
1791 }
1792
1793 static int
1794 ikev2_match_profile (const ikev2_profile_t *p, const ikev2_id_t *id_loc,
1795                      const ikev2_id_t *id_rem, int is_initiator)
1796 {
1797   /* on the initiator, IDi is always present and must match
1798    * however on the responder, IDr (which is our local id) is optional */
1799   if ((is_initiator || id_loc->type != 0) &&
1800       !ikev2_is_id_equal (&p->loc_id, id_loc))
1801     return 0;
1802
1803   /* on the initiator, we might not have configured a specific remote id
1804    * however on the responder, the remote id should always be configured */
1805   if ((!is_initiator || p->rem_id.type != 0) &&
1806       !ikev2_is_id_equal (&p->rem_id, id_rem))
1807     return 0;
1808
1809   return 1;
1810 }
1811
1812 static int
1813 ikev2_ts_cmp (ikev2_ts_t * ts1, ikev2_ts_t * ts2)
1814 {
1815   if (ts1->ts_type == ts2->ts_type && ts1->protocol_id == ts2->protocol_id &&
1816       ts1->start_port == ts2->start_port && ts1->end_port == ts2->end_port &&
1817       !ip_address_cmp (&ts1->start_addr, &ts2->start_addr) &&
1818       !ip_address_cmp (&ts1->end_addr, &ts2->end_addr))
1819     return 1;
1820
1821   return 0;
1822 }
1823
1824 static void
1825 ikev2_sa_match_ts (ikev2_sa_t * sa)
1826 {
1827   ikev2_main_t *km = &ikev2_main;
1828   ikev2_profile_t *p;
1829   ikev2_ts_t *ts, *p_tsi, *p_tsr, *tsi = 0, *tsr = 0;
1830   ikev2_id_t *id_rem, *id_loc;
1831
1832   pool_foreach (p, km->profiles)  {
1833
1834     if (sa->is_initiator)
1835       {
1836         p_tsi = &p->loc_ts;
1837         p_tsr = &p->rem_ts;
1838         id_rem = &sa->r_id;
1839         id_loc = &sa->i_id;
1840       }
1841     else
1842       {
1843         p_tsi = &p->rem_ts;
1844         p_tsr = &p->loc_ts;
1845         id_rem = &sa->i_id;
1846         id_loc = &sa->r_id;
1847       }
1848
1849     if (!ikev2_match_profile (p, id_loc, id_rem, sa->is_initiator))
1850       continue;
1851
1852     sa->profile_index = p - km->profiles;
1853
1854     vec_foreach(ts, sa->childs[0].tsi)
1855       {
1856         if (ikev2_ts_cmp(p_tsi, ts))
1857           {
1858             vec_add1 (tsi, ts[0]);
1859             break;
1860           }
1861       }
1862
1863     vec_foreach(ts, sa->childs[0].tsr)
1864       {
1865         if (ikev2_ts_cmp(p_tsr, ts))
1866           {
1867             vec_add1 (tsr, ts[0]);
1868             break;
1869           }
1870       }
1871
1872     break;
1873   }
1874
1875   if (tsi && tsr)
1876     {
1877       vec_free (sa->childs[0].tsi);
1878       vec_free (sa->childs[0].tsr);
1879       sa->childs[0].tsi = tsi;
1880       sa->childs[0].tsr = tsr;
1881     }
1882   else
1883     {
1884       vec_free (tsi);
1885       vec_free (tsr);
1886       ikev2_set_state (sa, IKEV2_STATE_TS_UNACCEPTABLE);
1887     }
1888 }
1889
1890 static ikev2_profile_t *
1891 ikev2_select_profile (ikev2_main_t *km, ikev2_sa_t *sa,
1892                       ikev2_sa_transform_t *tr_prf, u8 *key_pad)
1893 {
1894   ikev2_profile_t *ret = 0, *p;
1895   ikev2_id_t *id_rem, *id_loc;
1896   ikev2_auth_t *sa_auth;
1897   u8 *authmsg, *psk = 0, *auth = 0;
1898
1899   authmsg = ikev2_sa_generate_authmsg (sa, sa->is_initiator);
1900
1901   if (sa->is_initiator)
1902     {
1903       id_rem = &sa->r_id;
1904       id_loc = &sa->i_id;
1905       sa_auth = &sa->r_auth;
1906     }
1907   else
1908     {
1909       id_rem = &sa->i_id;
1910       id_loc = &sa->r_id;
1911       sa_auth = &sa->i_auth;
1912     }
1913
1914   pool_foreach (p, km->profiles)
1915     {
1916       if (!ikev2_match_profile (p, id_loc, id_rem, sa->is_initiator))
1917         continue;
1918
1919       if (sa_auth->method == IKEV2_AUTH_METHOD_SHARED_KEY_MIC)
1920         {
1921           if (!p->auth.data ||
1922               p->auth.method != IKEV2_AUTH_METHOD_SHARED_KEY_MIC)
1923             continue;
1924
1925           psk = ikev2_calc_prf (tr_prf, p->auth.data, key_pad);
1926           auth = ikev2_calc_prf (tr_prf, psk, authmsg);
1927
1928           if (!clib_memcmp (auth, sa_auth->data, vec_len (sa_auth->data)))
1929             {
1930               ikev2_set_state (sa, IKEV2_STATE_AUTHENTICATED);
1931               vec_free (auth);
1932               ret = p;
1933               break;
1934             }
1935           else
1936             {
1937               ikev2_elog_uint (IKEV2_LOG_ERROR,
1938                                "shared key mismatch! ispi %lx", sa->ispi);
1939             }
1940         }
1941       else if (sa_auth->method == IKEV2_AUTH_METHOD_RSA_SIG)
1942         {
1943           if (p->auth.method != IKEV2_AUTH_METHOD_RSA_SIG)
1944             continue;
1945
1946           if (ikev2_verify_sign (p->auth.key, sa_auth->data, authmsg) == 1)
1947             {
1948               ikev2_set_state (sa, IKEV2_STATE_AUTHENTICATED);
1949               ret = p;
1950               break;
1951             }
1952           else
1953             {
1954               ikev2_elog_uint (IKEV2_LOG_ERROR,
1955                                "cert verification failed! ispi %lx", sa->ispi);
1956             }
1957         }
1958     }
1959   vec_free (authmsg);
1960   return ret;
1961 }
1962
1963 static void
1964 ikev2_sa_auth (ikev2_sa_t *sa)
1965 {
1966   ikev2_main_t *km = &ikev2_main;
1967   ikev2_profile_t *sel_p = 0;
1968   ikev2_sa_transform_t *tr_prf;
1969   u8 *psk, *authmsg, *key_pad;
1970
1971   tr_prf =
1972     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
1973
1974   /* only shared key and rsa signature */
1975   if (!(sa->i_auth.method == IKEV2_AUTH_METHOD_SHARED_KEY_MIC ||
1976         sa->i_auth.method == IKEV2_AUTH_METHOD_RSA_SIG))
1977     {
1978       ikev2_elog_uint (IKEV2_LOG_ERROR, "unsupported authentication method %u",
1979                        sa->i_auth.method);
1980       ikev2_set_state (sa, IKEV2_STATE_AUTH_FAILED);
1981       return;
1982     }
1983
1984   key_pad = format (0, "%s", IKEV2_KEY_PAD);
1985   sel_p = ikev2_select_profile (km, sa, tr_prf, key_pad);
1986
1987   if (sel_p)
1988     {
1989       ASSERT (sa->state == IKEV2_STATE_AUTHENTICATED);
1990       sa->udp_encap = sel_p->udp_encap;
1991       sa->ipsec_over_udp_port = sel_p->ipsec_over_udp_port;
1992
1993       if (!sa->is_initiator)
1994         {
1995           vec_free (sa->r_id.data);
1996           sa->r_id.data = vec_dup (sel_p->loc_id.data);
1997           sa->r_id.type = sel_p->loc_id.type;
1998           sa->i_id.data = vec_dup (sel_p->rem_id.data);
1999           sa->i_id.type = sel_p->rem_id.type;
2000
2001           /* generate our auth data */
2002           authmsg = ikev2_sa_generate_authmsg (sa, 1);
2003           if (sel_p->auth.method == IKEV2_AUTH_METHOD_SHARED_KEY_MIC)
2004             {
2005               vec_free (sa->r_auth.data);
2006               psk = ikev2_calc_prf (tr_prf, sel_p->auth.data, key_pad);
2007               sa->r_auth.data = ikev2_calc_prf (tr_prf, psk, authmsg);
2008               sa->r_auth.method = IKEV2_AUTH_METHOD_SHARED_KEY_MIC;
2009               vec_free (psk);
2010             }
2011           else if (sel_p->auth.method == IKEV2_AUTH_METHOD_RSA_SIG)
2012             {
2013               vec_free (sa->r_auth.data);
2014               sa->r_auth.data = ikev2_calc_sign (km->pkey, authmsg);
2015               sa->r_auth.method = IKEV2_AUTH_METHOD_RSA_SIG;
2016             }
2017           vec_free (authmsg);
2018
2019           /* select transforms for 1st child sa */
2020           ikev2_sa_free_proposal_vector (&sa->childs[0].r_proposals);
2021           sa->childs[0].r_proposals =
2022             ikev2_select_proposal (sa->childs[0].i_proposals,
2023                                    IKEV2_PROTOCOL_ESP);
2024
2025           if (~0 != sel_p->tun_itf)
2026             {
2027               sa->is_tun_itf_set = 1;
2028               sa->tun_itf = sel_p->tun_itf;
2029             }
2030         }
2031     }
2032   else
2033     {
2034       ikev2_elog_uint (IKEV2_LOG_ERROR, "authentication failed, no matching "
2035                        "profile found! ispi %lx", sa->ispi);
2036       ikev2_set_state (sa, IKEV2_STATE_AUTH_FAILED);
2037     }
2038   vec_free (key_pad);
2039 }
2040
2041 static void
2042 ikev2_sa_auth_init (ikev2_sa_t * sa)
2043 {
2044   ikev2_main_t *km = &ikev2_main;
2045   u8 *authmsg, *key_pad, *psk = 0;
2046   ikev2_sa_transform_t *tr_prf;
2047
2048   tr_prf =
2049     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_PRF);
2050
2051   /* only shared key and rsa signature */
2052   if (!(sa->i_auth.method == IKEV2_AUTH_METHOD_SHARED_KEY_MIC ||
2053         sa->i_auth.method == IKEV2_AUTH_METHOD_RSA_SIG))
2054     {
2055       ikev2_elog_uint (IKEV2_LOG_ERROR,
2056                        "unsupported authentication method %u",
2057                        sa->i_auth.method);
2058       ikev2_set_state (sa, IKEV2_STATE_AUTH_FAILED);
2059       return;
2060     }
2061
2062   authmsg = ikev2_sa_generate_authmsg (sa, 0);
2063
2064   if (sa->i_auth.method == IKEV2_AUTH_METHOD_SHARED_KEY_MIC)
2065     {
2066       key_pad = format (0, "%s", IKEV2_KEY_PAD);
2067       psk = ikev2_calc_prf (tr_prf, sa->i_auth.data, key_pad);
2068       vec_free (sa->i_auth.data);
2069       sa->i_auth.data = ikev2_calc_prf (tr_prf, psk, authmsg);
2070       sa->i_auth.method = IKEV2_AUTH_METHOD_SHARED_KEY_MIC;
2071       vec_free (psk);
2072       vec_free (key_pad);
2073     }
2074   else if (sa->i_auth.method == IKEV2_AUTH_METHOD_RSA_SIG)
2075     {
2076       vec_free (sa->i_auth.data);
2077       sa->i_auth.data = ikev2_calc_sign (km->pkey, authmsg);
2078       sa->i_auth.method = IKEV2_AUTH_METHOD_RSA_SIG;
2079     }
2080   vec_free (authmsg);
2081 }
2082
2083 static u32
2084 ikev2_mk_local_sa_id (u32 sai, u32 ci, u32 ti)
2085 {
2086   return (0x80000000 | (ti << 24) | (sai << 12) | ci);
2087 }
2088
2089 static u32
2090 ikev2_mk_remote_sa_id (u32 sai, u32 ci, u32 ti)
2091 {
2092   return (0xc0000000 | (ti << 24) | (sai << 12) | ci);
2093 }
2094
2095 typedef struct
2096 {
2097   u32 sw_if_index;
2098   u32 salt_local;
2099   u32 salt_remote;
2100   u32 local_sa_id;
2101   u32 remote_sa_id;
2102   ipsec_sa_flags_t flags;
2103   u32 local_spi;
2104   u32 remote_spi;
2105   ipsec_crypto_alg_t encr_type;
2106   ipsec_integ_alg_t integ_type;
2107   ip_address_t local_ip;
2108   ip_address_t remote_ip;
2109   ipsec_key_t loc_ckey, rem_ckey, loc_ikey, rem_ikey;
2110   u8 is_rekey;
2111   u32 old_remote_sa_id;
2112   u16 ipsec_over_udp_port;
2113   u16 src_port;
2114   u16 dst_port;
2115 } ikev2_add_ipsec_tunnel_args_t;
2116
2117 static void
2118 ikev2_add_tunnel_from_main (ikev2_add_ipsec_tunnel_args_t * a)
2119 {
2120   ikev2_main_t *km = &ikev2_main;
2121   u32 sw_if_index;
2122   int rv = 0;
2123   tunnel_t tun_in = {
2124     .t_flags = TUNNEL_FLAG_NONE,
2125     .t_encap_decap_flags = TUNNEL_ENCAP_DECAP_FLAG_NONE,
2126     .t_dscp = 0,
2127     .t_mode = TUNNEL_MODE_P2P,
2128     .t_table_id = 0,
2129     .t_hop_limit = 255,
2130     .t_src = a->remote_ip,
2131     .t_dst = a->local_ip,
2132   };
2133   tunnel_t tun_out = {
2134     .t_flags = TUNNEL_FLAG_NONE,
2135     .t_encap_decap_flags = TUNNEL_ENCAP_DECAP_FLAG_NONE,
2136     .t_dscp = 0,
2137     .t_mode = TUNNEL_MODE_P2P,
2138     .t_table_id = 0,
2139     .t_hop_limit = 255,
2140     .t_src = a->local_ip,
2141     .t_dst = a->remote_ip,
2142   };
2143
2144   if (~0 == a->sw_if_index)
2145     {
2146       /* no tunnel associated with the SA/profile - create a new one */
2147       rv = ipip_add_tunnel (IPIP_TRANSPORT_IP4, ~0, &ip_addr_46 (&a->local_ip),
2148                             &ip_addr_46 (&a->remote_ip), 0,
2149                             TUNNEL_ENCAP_DECAP_FLAG_NONE, IP_DSCP_CS0,
2150                             TUNNEL_MODE_P2P, &sw_if_index);
2151
2152       if (rv == VNET_API_ERROR_IF_ALREADY_EXISTS)
2153         {
2154           if (hash_get (km->sw_if_indices, sw_if_index))
2155             /* interface is managed by IKE; proceed with updating SAs */
2156             rv = 0;
2157         }
2158       hash_set1 (km->sw_if_indices, sw_if_index);
2159     }
2160   else
2161     {
2162       sw_if_index = a->sw_if_index;
2163       vnet_sw_interface_admin_up (vnet_get_main (), sw_if_index);
2164     }
2165
2166   if (rv)
2167     {
2168       ikev2_elog_uint (IKEV2_LOG_ERROR,
2169                        "installing ipip tunnel failed! local spi: %x",
2170                        a->local_spi);
2171       return;
2172     }
2173
2174   u32 *sas_in = NULL;
2175   vec_add1 (sas_in, a->remote_sa_id);
2176   if (a->is_rekey)
2177     {
2178       ipsec_tun_protect_del (sw_if_index, NULL);
2179
2180       /* replace local SA immediately */
2181       ipsec_sa_unlock_id (a->local_sa_id);
2182
2183       /* keep the old sa */
2184       vec_add1 (sas_in, a->old_remote_sa_id);
2185     }
2186
2187   rv = ipsec_sa_add_and_lock (a->local_sa_id, a->local_spi, IPSEC_PROTOCOL_ESP,
2188                               a->encr_type, &a->loc_ckey, a->integ_type,
2189                               &a->loc_ikey, a->flags, a->salt_local,
2190                               a->src_port, a->dst_port, 0, &tun_out, NULL);
2191   if (rv)
2192     goto err0;
2193
2194   rv = ipsec_sa_add_and_lock (
2195     a->remote_sa_id, a->remote_spi, IPSEC_PROTOCOL_ESP, a->encr_type,
2196     &a->rem_ckey, a->integ_type, &a->rem_ikey,
2197     (a->flags | IPSEC_SA_FLAG_IS_INBOUND), a->salt_remote,
2198     a->ipsec_over_udp_port, a->ipsec_over_udp_port, 0, &tun_in, NULL);
2199   if (rv)
2200     goto err1;
2201
2202   rv = ipsec_tun_protect_update (sw_if_index, NULL, a->local_sa_id, sas_in);
2203   if (rv)
2204     goto err2;
2205
2206   return;
2207
2208 err2:
2209   ipsec_sa_unlock_id (a->remote_sa_id);
2210 err1:
2211   ipsec_sa_unlock_id (a->local_sa_id);
2212 err0:
2213   vec_free (sas_in);
2214 }
2215
2216 static int
2217 ikev2_create_tunnel_interface (vlib_main_t *vm, ikev2_sa_t *sa,
2218                                ikev2_child_sa_t *child, u32 sa_index,
2219                                u32 child_index, u8 is_rekey, u8 kex)
2220 {
2221   u32 thread_index = vlib_get_thread_index ();
2222   ikev2_main_t *km = &ikev2_main;
2223   ipsec_crypto_alg_t encr_type;
2224   ipsec_integ_alg_t integ_type;
2225   ikev2_profile_t *p = 0;
2226   ikev2_sa_transform_t *tr;
2227   ikev2_sa_proposal_t *proposals;
2228   u8 is_aead = 0;
2229   ikev2_add_ipsec_tunnel_args_t a;
2230
2231   clib_memset (&a, 0, sizeof (a));
2232
2233   if (!child->r_proposals)
2234     {
2235       ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2236       return 1;
2237     }
2238
2239   if (sa->is_initiator)
2240     {
2241       ip_address_copy (&a.local_ip, &sa->iaddr);
2242       ip_address_copy (&a.remote_ip, &sa->raddr);
2243       proposals = child->r_proposals;
2244       a.local_spi = child->r_proposals[0].spi;
2245       a.remote_spi = child->i_proposals[0].spi;
2246     }
2247   else
2248     {
2249       ip_address_copy (&a.local_ip, &sa->raddr);
2250       ip_address_copy (&a.remote_ip, &sa->iaddr);
2251       proposals = child->i_proposals;
2252       a.local_spi = child->i_proposals[0].spi;
2253       a.remote_spi = child->r_proposals[0].spi;
2254     }
2255
2256   a.flags = IPSEC_SA_FLAG_USE_ANTI_REPLAY;
2257   if (sa->udp_encap)
2258     {
2259       a.flags |= IPSEC_SA_FLAG_IS_TUNNEL;
2260       a.flags |= IPSEC_SA_FLAG_UDP_ENCAP;
2261     }
2262   if (ikev2_natt_active (sa))
2263     a.flags |= IPSEC_SA_FLAG_UDP_ENCAP;
2264   a.is_rekey = is_rekey;
2265
2266   tr = ikev2_sa_get_td_for_type (proposals, IKEV2_TRANSFORM_TYPE_ESN);
2267   if (tr && tr->esn_type)
2268     a.flags |= IPSEC_SA_FLAG_USE_ESN;
2269
2270   tr = ikev2_sa_get_td_for_type (proposals, IKEV2_TRANSFORM_TYPE_ENCR);
2271   if (tr)
2272     {
2273       if (tr->encr_type == IKEV2_TRANSFORM_ENCR_TYPE_AES_CBC && tr->key_len)
2274         {
2275           switch (tr->key_len)
2276             {
2277             case 16:
2278               encr_type = IPSEC_CRYPTO_ALG_AES_CBC_128;
2279               break;
2280             case 24:
2281               encr_type = IPSEC_CRYPTO_ALG_AES_CBC_192;
2282               break;
2283             case 32:
2284               encr_type = IPSEC_CRYPTO_ALG_AES_CBC_256;
2285               break;
2286             default:
2287               ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2288               return 1;
2289               break;
2290             }
2291         }
2292       else if (tr->encr_type == IKEV2_TRANSFORM_ENCR_TYPE_AES_GCM_16
2293                && tr->key_len)
2294         {
2295           switch (tr->key_len)
2296             {
2297             case 16:
2298               encr_type = IPSEC_CRYPTO_ALG_AES_GCM_128;
2299               break;
2300             case 24:
2301               encr_type = IPSEC_CRYPTO_ALG_AES_GCM_192;
2302               break;
2303             case 32:
2304               encr_type = IPSEC_CRYPTO_ALG_AES_GCM_256;
2305               break;
2306             default:
2307               ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2308               return 1;
2309               break;
2310             }
2311           is_aead = 1;
2312         }
2313       else
2314         {
2315           ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2316           return 1;
2317         }
2318     }
2319   else
2320     {
2321       ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2322       return 1;
2323     }
2324   a.encr_type = encr_type;
2325
2326   if (!is_aead)
2327     {
2328       tr = ikev2_sa_get_td_for_type (proposals, IKEV2_TRANSFORM_TYPE_INTEG);
2329       if (tr)
2330         {
2331           switch (tr->integ_type)
2332             {
2333             case IKEV2_TRANSFORM_INTEG_TYPE_AUTH_HMAC_SHA2_256_128:
2334               integ_type = IPSEC_INTEG_ALG_SHA_256_128;
2335               break;
2336             case IKEV2_TRANSFORM_INTEG_TYPE_AUTH_HMAC_SHA2_384_192:
2337               integ_type = IPSEC_INTEG_ALG_SHA_384_192;
2338               break;
2339             case IKEV2_TRANSFORM_INTEG_TYPE_AUTH_HMAC_SHA2_512_256:
2340               integ_type = IPSEC_INTEG_ALG_SHA_512_256;
2341               break;
2342             case IKEV2_TRANSFORM_INTEG_TYPE_AUTH_HMAC_SHA1_96:
2343               integ_type = IPSEC_INTEG_ALG_SHA1_96;
2344               break;
2345             default:
2346               ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2347               return 1;
2348             }
2349         }
2350       else
2351         {
2352           ikev2_set_state (sa, IKEV2_STATE_NO_PROPOSAL_CHOSEN);
2353           return 1;
2354         }
2355     }
2356   else
2357     {
2358       integ_type = IPSEC_INTEG_ALG_NONE;
2359     }
2360
2361   a.integ_type = integ_type;
2362   ikev2_calc_child_keys (sa, child, kex);
2363
2364   if (sa->is_initiator)
2365     {
2366       ipsec_mk_key (&a.loc_ikey, child->sk_ai, vec_len (child->sk_ai));
2367       ipsec_mk_key (&a.rem_ikey, child->sk_ar, vec_len (child->sk_ar));
2368       ipsec_mk_key (&a.loc_ckey, child->sk_ei, vec_len (child->sk_ei));
2369       ipsec_mk_key (&a.rem_ckey, child->sk_er, vec_len (child->sk_er));
2370       if (is_aead)
2371         {
2372           a.salt_remote = child->salt_er;
2373           a.salt_local = child->salt_ei;
2374         }
2375       a.dst_port = a.src_port = sa->ipsec_over_udp_port;
2376     }
2377   else
2378     {
2379       ipsec_mk_key (&a.loc_ikey, child->sk_ar, vec_len (child->sk_ar));
2380       ipsec_mk_key (&a.rem_ikey, child->sk_ai, vec_len (child->sk_ai));
2381       ipsec_mk_key (&a.loc_ckey, child->sk_er, vec_len (child->sk_er));
2382       ipsec_mk_key (&a.rem_ckey, child->sk_ei, vec_len (child->sk_ei));
2383       if (is_aead)
2384         {
2385           a.salt_remote = child->salt_ei;
2386           a.salt_local = child->salt_er;
2387         }
2388       a.dst_port =
2389         ikev2_natt_active (sa) ? sa->dst_port : sa->ipsec_over_udp_port;
2390       a.src_port = sa->ipsec_over_udp_port;
2391     }
2392
2393   if (sa->is_initiator && sa->profile_index != ~0)
2394     p = pool_elt_at_index (km->profiles, sa->profile_index);
2395
2396   if (p && p->lifetime)
2397     {
2398       child->time_to_expiration = vlib_time_now (vm) + p->lifetime;
2399       if (p->lifetime_jitter)
2400         {
2401           // This is not much better than rand(3), which Coverity warns
2402           // is unsuitable for security applications; random_u32 is
2403           // however fast. If this perturbance to the expiration time
2404           // needs to use a better RNG then we may need to use something
2405           // like /dev/urandom which has significant overhead.
2406           u32 rnd = (u32) (vlib_time_now (vm) * 1e6);
2407           rnd = random_u32 (&rnd);
2408
2409           child->time_to_expiration += 1 + (rnd % p->lifetime_jitter);
2410         }
2411     }
2412
2413   if (thread_index & 0xffffffc0)
2414     ikev2_elog_error ("error: thread index exceeds max range 0x3f!");
2415
2416   if (child_index & 0xfffff000 || sa_index & 0xfffff000)
2417     ikev2_elog_error ("error: sa/child index exceeds max range 0xfff!");
2418
2419   child->local_sa_id =
2420     a.local_sa_id =
2421     ikev2_mk_local_sa_id (sa_index, child_index, thread_index);
2422
2423   u32 remote_sa_id = ikev2_mk_remote_sa_id (sa_index, child_index,
2424                                             thread_index);
2425
2426   if (is_rekey)
2427     {
2428       /* create a new remote SA ID to keep the old SA for a bit longer
2429        * so the peer has some time to swap their SAs */
2430
2431       /* use most significat bit of child index part in id */
2432       u32 mask = 0x800;
2433       if (sa->current_remote_id_mask)
2434         {
2435           sa->old_remote_id = a.old_remote_sa_id = remote_sa_id | mask;
2436           sa->current_remote_id_mask = 0;
2437         }
2438       else
2439         {
2440           sa->old_remote_id = a.old_remote_sa_id = remote_sa_id;
2441           sa->current_remote_id_mask = mask;
2442           remote_sa_id |= mask;
2443         }
2444       sa->old_id_expiration = 3.0;
2445       sa->old_remote_id_present = 1;
2446     }
2447
2448   child->remote_sa_id = a.remote_sa_id = remote_sa_id;
2449
2450   a.sw_if_index = (sa->is_tun_itf_set ? sa->tun_itf : ~0);
2451   a.ipsec_over_udp_port = sa->ipsec_over_udp_port;
2452
2453   vl_api_rpc_call_main_thread (ikev2_add_tunnel_from_main,
2454                                (u8 *) & a, sizeof (a));
2455   return 0;
2456 }
2457
2458 typedef struct
2459 {
2460   ip46_address_t local_ip;
2461   ip46_address_t remote_ip;
2462   u32 remote_sa_id;
2463   u32 local_sa_id;
2464   u32 sw_if_index;
2465 } ikev2_del_ipsec_tunnel_args_t;
2466
2467 static u32
2468 ikev2_flip_alternate_sa_bit (u32 id)
2469 {
2470   u32 mask = 0x800;
2471   if (mask & id)
2472     return id & ~mask;
2473   return id | mask;
2474 }
2475
2476 static void
2477 ikev2_del_tunnel_from_main (ikev2_del_ipsec_tunnel_args_t * a)
2478 {
2479   ikev2_main_t *km = &ikev2_main;
2480   ipip_tunnel_t *ipip = NULL;
2481   u32 sw_if_index;
2482
2483   if (~0 == a->sw_if_index)
2484     {
2485     ipip_tunnel_key_t key = {
2486       .src = a->local_ip,
2487       .dst = a->remote_ip,
2488       .transport = IPIP_TRANSPORT_IP4,
2489       .fib_index = 0,
2490     };
2491
2492       ipip = ipip_tunnel_db_find (&key);
2493
2494       if (ipip)
2495         {
2496           sw_if_index = ipip->sw_if_index;
2497           hash_unset (km->sw_if_indices, ipip->sw_if_index);
2498         }
2499       else
2500         sw_if_index = ~0;
2501     }
2502   else
2503     {
2504       sw_if_index = a->sw_if_index;
2505       vnet_sw_interface_admin_down (vnet_get_main (), sw_if_index);
2506     }
2507
2508   if (~0 != sw_if_index)
2509     ipsec_tun_protect_del (sw_if_index, NULL);
2510
2511   ipsec_sa_unlock_id (a->remote_sa_id);
2512   ipsec_sa_unlock_id (a->local_sa_id);
2513   ipsec_sa_unlock_id (ikev2_flip_alternate_sa_bit (a->remote_sa_id));
2514
2515   if (ipip)
2516     ipip_del_tunnel (ipip->sw_if_index);
2517 }
2518
2519 static int
2520 ikev2_delete_tunnel_interface (vnet_main_t * vnm, ikev2_sa_t * sa,
2521                                ikev2_child_sa_t * child)
2522 {
2523   ikev2_del_ipsec_tunnel_args_t a;
2524
2525   clib_memset (&a, 0, sizeof (a));
2526
2527   if (sa->is_initiator)
2528     {
2529       ip_address_to_46 (&sa->iaddr, &a.local_ip);
2530       ip_address_to_46 (&sa->raddr, &a.remote_ip);
2531     }
2532   else
2533     {
2534       ip_address_to_46 (&sa->raddr, &a.local_ip);
2535       ip_address_to_46 (&sa->iaddr, &a.remote_ip);
2536     }
2537
2538   a.remote_sa_id = child->remote_sa_id;
2539   a.local_sa_id = child->local_sa_id;
2540   a.sw_if_index = (sa->is_tun_itf_set ? sa->tun_itf : ~0);
2541
2542   vl_api_rpc_call_main_thread (ikev2_del_tunnel_from_main, (u8 *) & a,
2543                                sizeof (a));
2544   return 0;
2545 }
2546
2547 static void
2548 ikev2_add_invalid_ke_payload (ikev2_sa_t *sa, ikev2_payload_chain_t *chain)
2549 {
2550   u8 *data = vec_new (u8, 2);
2551   ikev2_sa_transform_t *tr_dh =
2552     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_DH);
2553   ASSERT (tr_dh && tr_dh->dh_type);
2554   data[0] = (tr_dh->dh_type >> 8) & 0xff;
2555   data[1] = (tr_dh->dh_type) & 0xff;
2556   ikev2_payload_add_notify (chain, IKEV2_NOTIFY_MSG_INVALID_KE_PAYLOAD, data);
2557   vec_free (data);
2558 }
2559
2560 static void
2561 ikev2_add_notify_payload (ikev2_sa_t *sa, ikev2_payload_chain_t *chain,
2562                           u16 notify_type)
2563 {
2564   if (notify_type == IKEV2_NOTIFY_MSG_INVALID_KE_PAYLOAD)
2565     ikev2_add_invalid_ke_payload (sa, chain);
2566   else
2567     ikev2_payload_add_notify (chain, notify_type, 0);
2568 }
2569
2570 static void
2571 ikev2_add_create_child_resp (ikev2_sa_t *sa, ikev2_rekey_t *rekey,
2572                              ikev2_payload_chain_t *chain)
2573 {
2574   if (rekey->notify_type)
2575     {
2576       ikev2_add_notify_payload (sa, chain, rekey->notify_type);
2577       return;
2578     }
2579
2580   ikev2_payload_add_sa (chain, rekey->r_proposal, 0);
2581   ikev2_payload_add_nonce (chain, sa->r_nonce);
2582   if (rekey->kex)
2583     ikev2_payload_add_ke (chain, sa->dh_group, sa->r_dh_data);
2584   ikev2_payload_add_ts (chain, rekey->tsi, IKEV2_PAYLOAD_TSI);
2585   ikev2_payload_add_ts (chain, rekey->tsr, IKEV2_PAYLOAD_TSR);
2586 }
2587
2588 static u32
2589 ikev2_generate_message (vlib_buffer_t *b, ikev2_sa_t *sa, ike_header_t *ike,
2590                         void *user, udp_header_t *udp, ikev2_stats_t *stats)
2591 {
2592   ikev2_main_t *km = &ikev2_main;
2593   u16 buffer_data_size = vlib_buffer_get_default_data_size (km->vlib_main);
2594   v8 *integ = 0;
2595   ike_payload_header_t *ph;
2596   u16 plen;
2597   u32 tlen = 0;
2598
2599   ikev2_sa_transform_t *tr_encr, *tr_integ;
2600   tr_encr =
2601     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_ENCR);
2602   tr_integ =
2603     ikev2_sa_get_td_for_type (sa->r_proposals, IKEV2_TRANSFORM_TYPE_INTEG);
2604
2605   ikev2_payload_chain_t *chain = 0;
2606   ikev2_payload_new_chain (chain);
2607
2608   if (ike->exchange == IKEV2_EXCHANGE_SA_INIT)
2609     {
2610       if (sa->r_proposals == 0)
2611         {
2612           ikev2_payload_add_notify (chain,
2613                                     IKEV2_NOTIFY_MSG_NO_PROPOSAL_CHOSEN, 0);
2614           ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
2615         }
2616       else if (sa->dh_group == IKEV2_TRANSFORM_DH_TYPE_NONE)
2617         {
2618           ikev2_add_invalid_ke_payload (sa, chain);
2619           ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
2620         }
2621       else if (sa->state == IKEV2_STATE_NOTIFY_AND_DELETE)
2622         {
2623           u8 *data = vec_new (u8, 1);
2624
2625           data[0] = sa->unsupported_cp;
2626           ikev2_payload_add_notify (chain,
2627                                     IKEV2_NOTIFY_MSG_UNSUPPORTED_CRITICAL_PAYLOAD,
2628                                     data);
2629           vec_free (data);
2630         }
2631       else
2632         {
2633           ASSERT (udp);
2634
2635           ike->rspi = clib_host_to_net_u64 (sa->rspi);
2636           ikev2_payload_add_sa (chain, sa->r_proposals, 0);
2637           ikev2_payload_add_ke (chain, sa->dh_group, sa->r_dh_data);
2638           ikev2_payload_add_nonce (chain, sa->r_nonce);
2639
2640           u8 *nat_detection_sha1 =
2641             ikev2_compute_nat_sha1 (clib_host_to_net_u64 (sa->ispi),
2642                                     clib_host_to_net_u64 (sa->rspi),
2643                                     &sa->raddr, udp->dst_port);
2644           ikev2_payload_add_notify (chain,
2645                                     IKEV2_NOTIFY_MSG_NAT_DETECTION_SOURCE_IP,
2646                                     nat_detection_sha1);
2647           vec_free (nat_detection_sha1);
2648           nat_detection_sha1 =
2649             ikev2_compute_nat_sha1 (clib_host_to_net_u64 (sa->ispi),
2650                                     clib_host_to_net_u64 (sa->rspi),
2651                                     &sa->iaddr, udp->src_port);
2652           ikev2_payload_add_notify (chain,
2653                                     IKEV2_NOTIFY_MSG_NAT_DETECTION_DESTINATION_IP,
2654                                     nat_detection_sha1);
2655           vec_free (nat_detection_sha1);
2656         }
2657     }
2658   else if (ike->exchange == IKEV2_EXCHANGE_IKE_AUTH)
2659     {
2660       if (sa->state == IKEV2_STATE_AUTHENTICATED)
2661         {
2662           ikev2_payload_add_id (chain, &sa->r_id, IKEV2_PAYLOAD_IDR);
2663           ikev2_payload_add_auth (chain, &sa->r_auth);
2664           ikev2_payload_add_sa (chain, sa->childs[0].r_proposals, 0);
2665           ikev2_payload_add_ts (chain, sa->childs[0].tsi, IKEV2_PAYLOAD_TSI);
2666           ikev2_payload_add_ts (chain, sa->childs[0].tsr, IKEV2_PAYLOAD_TSR);
2667         }
2668       else if (sa->state == IKEV2_STATE_AUTH_FAILED)
2669         {
2670           ikev2_payload_add_notify (chain,
2671                                     IKEV2_NOTIFY_MSG_AUTHENTICATION_FAILED,
2672                                     0);
2673           ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
2674         }
2675       else if (sa->state == IKEV2_STATE_TS_UNACCEPTABLE)
2676         {
2677           ikev2_payload_add_notify (chain, IKEV2_NOTIFY_MSG_TS_UNACCEPTABLE,
2678                                     0);
2679           ikev2_payload_add_id (chain, &sa->r_id, IKEV2_PAYLOAD_IDR);
2680           ikev2_payload_add_auth (chain, &sa->r_auth);
2681         }
2682       else if (sa->state == IKEV2_STATE_NO_PROPOSAL_CHOSEN)
2683         {
2684           ikev2_payload_add_notify (chain,
2685                                     IKEV2_NOTIFY_MSG_NO_PROPOSAL_CHOSEN, 0);
2686           ikev2_payload_add_id (chain, &sa->r_id, IKEV2_PAYLOAD_IDR);
2687           ikev2_payload_add_auth (chain, &sa->r_auth);
2688           ikev2_payload_add_ts (chain, sa->childs[0].tsi, IKEV2_PAYLOAD_TSI);
2689           ikev2_payload_add_ts (chain, sa->childs[0].tsr, IKEV2_PAYLOAD_TSR);
2690         }
2691       else if (sa->state == IKEV2_STATE_NOTIFY_AND_DELETE)
2692         {
2693           u8 *data = vec_new (u8, 1);
2694
2695           data[0] = sa->unsupported_cp;
2696           ikev2_payload_add_notify (chain,
2697                                     IKEV2_NOTIFY_MSG_UNSUPPORTED_CRITICAL_PAYLOAD,
2698                                     data);
2699           vec_free (data);
2700         }
2701       else if (sa->state == IKEV2_STATE_SA_INIT)
2702         {
2703           ikev2_payload_add_id (chain, &sa->i_id, IKEV2_PAYLOAD_IDI);
2704           /* IDr is optional when sending INIT from the initiator */
2705           ASSERT (sa->r_id.type != 0 || sa->is_initiator);
2706           if (sa->r_id.type != 0)
2707             ikev2_payload_add_id (chain, &sa->r_id, IKEV2_PAYLOAD_IDR);
2708           ikev2_payload_add_auth (chain, &sa->i_auth);
2709           ikev2_payload_add_sa (chain, sa->childs[0].i_proposals, 0);
2710           ikev2_payload_add_ts (chain, sa->childs[0].tsi, IKEV2_PAYLOAD_TSI);
2711           ikev2_payload_add_ts (chain, sa->childs[0].tsr, IKEV2_PAYLOAD_TSR);
2712           ikev2_payload_add_notify (chain, IKEV2_NOTIFY_MSG_INITIAL_CONTACT,
2713                                     0);
2714         }
2715       else
2716         {
2717           ikev2_set_state (sa, IKEV2_STATE_DELETED, ": unexpected IKE_AUTH");
2718           goto done;
2719         }
2720     }
2721   else if (ike->exchange == IKEV2_EXCHANGE_INFORMATIONAL)
2722     {
2723       /* if pending delete */
2724       if (sa->del)
2725         {
2726           if (sa->del[0].protocol_id == IKEV2_PROTOCOL_IKE)
2727             {
2728               if (ike_hdr_is_request (ike))
2729                 ikev2_payload_add_delete (chain, sa->del);
2730
2731               /* The response to a request that deletes the IKE SA is an empty
2732                  INFORMATIONAL response. */
2733               ikev2_set_state (sa, IKEV2_STATE_NOTIFY_AND_DELETE);
2734             }
2735           /* The response to a request that deletes ESP or AH SAs will contain
2736              delete payloads for the paired SAs going in the other direction. */
2737           else
2738             {
2739               ikev2_payload_add_delete (chain, sa->del);
2740             }
2741           vec_free (sa->del);
2742           sa->del = 0;
2743         }
2744       /* received N(AUTHENTICATION_FAILED) */
2745       else if (sa->state == IKEV2_STATE_AUTH_FAILED)
2746         {
2747           ikev2_set_state (sa, IKEV2_STATE_DELETED, ": auth failed");
2748           goto done;
2749         }
2750       /* received unsupported critical payload */
2751       else if (sa->unsupported_cp)
2752         {
2753           u8 *data = vec_new (u8, 1);
2754
2755           data[0] = sa->unsupported_cp;
2756           ikev2_payload_add_notify (chain,
2757                                     IKEV2_NOTIFY_MSG_UNSUPPORTED_CRITICAL_PAYLOAD,
2758                                     data);
2759           vec_free (data);
2760           sa->unsupported_cp = 0;
2761         }
2762       else
2763         /* else send empty response */
2764         {
2765           if (ike_hdr_is_response (ike))
2766             {
2767               ASSERT (stats != 0);
2768               stats->n_keepalives++;
2769               sa->stats.n_keepalives++;
2770             }
2771         }
2772     }
2773   else if (ike->exchange == IKEV2_EXCHANGE_CREATE_CHILD_SA)
2774     {
2775       if (sa->is_initiator)
2776         {
2777
2778           ikev2_sa_proposal_t *proposals = (ikev2_sa_proposal_t *) user;
2779           ikev2_notify_t notify;
2780           u8 *data = vec_new (u8, 4);
2781           clib_memset (&notify, 0, sizeof (notify));
2782           notify.protocol_id = IKEV2_PROTOCOL_ESP;
2783           notify.spi = sa->childs[0].i_proposals->spi;
2784           *(u32 *) data = clib_host_to_net_u32 (notify.spi);
2785
2786           ikev2_payload_add_sa (chain, proposals, 0);
2787           ikev2_payload_add_nonce (chain, sa->i_nonce);
2788           ikev2_payload_add_ts (chain, sa->childs[0].tsi, IKEV2_PAYLOAD_TSI);
2789           ikev2_payload_add_ts (chain, sa->childs[0].tsr, IKEV2_PAYLOAD_TSR);
2790           ikev2_payload_add_notify_2 (chain, IKEV2_NOTIFY_MSG_REKEY_SA, data,
2791                                       &notify);
2792
2793           vec_free (data);
2794         }
2795       else if (vec_len (sa->rekey) > 0)
2796         {
2797           ikev2_add_create_child_resp (sa, &sa->rekey[0], chain);
2798           vec_del1 (sa->rekey, 0);
2799         }
2800       else if (vec_len (sa->new_child) > 0)
2801         {
2802           ikev2_add_create_child_resp (sa, &sa->new_child[0], chain);
2803           vec_del1 (sa->new_child, 0);
2804         }
2805       else if (vec_len (sa->sa_rekey) > 0)
2806         {
2807           if (sa->sa_rekey[0].notify_type)
2808             ikev2_add_notify_payload (sa, chain, sa->sa_rekey[0].notify_type);
2809           else
2810             {
2811               ikev2_payload_add_sa (chain, sa->sa_rekey[0].r_proposals, 1);
2812               ikev2_payload_add_nonce (chain, sa->sa_rekey[0].r_nonce);
2813               ikev2_payload_add_ke (chain, sa->sa_rekey[0].dh_group,
2814                                     sa->sa_rekey[0].r_dh_data);
2815             }
2816           vec_del1 (sa->sa_rekey, 0);
2817         }
2818       else if (sa->unsupported_cp)
2819         {
2820           u8 *data = vec_new (u8, 1);
2821
2822           data[0] = sa->unsupported_cp;
2823           ikev2_payload_add_notify (
2824             chain, IKEV2_NOTIFY_MSG_UNSUPPORTED_CRITICAL_PAYLOAD, data);
2825           vec_free (data);
2826           sa->unsupported_cp = 0;
2827         }
2828       else
2829         {
2830           ikev2_payload_add_notify (chain, IKEV2_NOTIFY_MSG_NO_ADDITIONAL_SAS,
2831                                     0);
2832         }
2833     }
2834
2835   /* IKEv2 header */
2836   ike->version = IKE_VERSION_2;
2837   ike->nextpayload = IKEV2_PAYLOAD_SK;
2838   tlen = sizeof (*ike);
2839
2840   if (sa->is_initiator)
2841     ike->flags |= IKEV2_HDR_FLAG_INITIATOR;
2842
2843   if (ike->exchange == IKEV2_EXCHANGE_SA_INIT)
2844     {
2845       tlen += vec_len (chain->data);
2846       ike->nextpayload = chain->first_payload_type;
2847       ike->length = clib_host_to_net_u32 (tlen);
2848
2849       if (tlen + b->current_length + b->current_data > buffer_data_size)
2850         {
2851           tlen = ~0;
2852           goto done;
2853         }
2854
2855       clib_memcpy_fast (ike->payload, chain->data, vec_len (chain->data));
2856
2857       /* store whole IKE payload - needed for PSK auth */
2858       vec_reset_length (sa->last_sa_init_res_packet_data);
2859       vec_add (sa->last_sa_init_res_packet_data, ike, tlen);
2860     }
2861   else
2862     {
2863       ikev2_main_per_thread_data_t *ptd = ikev2_get_per_thread_data ();
2864       ikev2_payload_chain_add_padding (chain, tr_encr->block_size);
2865
2866       /* SK payload */
2867       plen = sizeof (*ph);
2868       ph = (ike_payload_header_t *) & ike->payload[0];
2869       ph->nextpayload = chain->first_payload_type;
2870       ph->flags = 0;
2871       int is_aead =
2872         tr_encr->encr_type == IKEV2_TRANSFORM_ENCR_TYPE_AES_GCM_16;
2873       int iv_len = is_aead ? IKEV2_GCM_IV_SIZE : tr_encr->block_size;
2874       plen += vec_len (chain->data) + iv_len;
2875
2876       /* add space for hmac/tag */
2877       if (tr_integ)
2878         plen += tr_integ->key_trunc;
2879       else
2880         plen += IKEV2_GCM_ICV_SIZE;
2881       tlen += plen;
2882
2883       if (tlen + b->current_length + b->current_data > buffer_data_size)
2884         {
2885           tlen = ~0;
2886           goto done;
2887         }
2888
2889       /* payload and total length */
2890       ph->length = clib_host_to_net_u16 (plen);
2891       ike->length = clib_host_to_net_u32 (tlen);
2892
2893       if (is_aead)
2894         {
2895           if (!ikev2_encrypt_aead_data (ptd, sa, tr_encr, chain->data,
2896                                         ph->payload, (u8 *) ike,
2897                                         sizeof (*ike) + sizeof (*ph),
2898                                         ph->payload + plen - sizeof (*ph) -
2899                                         IKEV2_GCM_ICV_SIZE))
2900             {
2901               tlen = ~0;
2902               goto done;
2903             }
2904         }
2905       else
2906         {
2907           if (!ikev2_encrypt_data
2908               (ptd, sa, tr_encr, chain->data, ph->payload))
2909             {
2910               tlen = ~0;
2911               goto done;
2912             }
2913           integ =
2914             ikev2_calc_integr (tr_integ,
2915                                sa->is_initiator ? sa->sk_ai : sa->sk_ar,
2916                                (u8 *) ike, tlen - tr_integ->key_trunc);
2917           clib_memcpy_fast (ike->payload + tlen - tr_integ->key_trunc -
2918                             sizeof (*ike), integ, tr_integ->key_trunc);
2919         }
2920
2921       /* store whole IKE payload - needed for retransmit */
2922       vec_reset_length (sa->last_res_packet_data);
2923       vec_add (sa->last_res_packet_data, ike, tlen);
2924     }
2925
2926 done:
2927   ikev2_payload_destroy_chain (chain);
2928   vec_free (integ);
2929   return tlen;
2930 }
2931
2932 static u32
2933 ikev2_retransmit_sa_init_one (ikev2_sa_t * sa, ike_header_t * ike,
2934                               ip_address_t iaddr, ip_address_t raddr,
2935                               u32 rlen)
2936 {
2937   int p = 0;
2938   ike_header_t *tmp;
2939   u8 payload = ike->nextpayload;
2940
2941   if (sa->ispi != clib_net_to_host_u64 (ike->ispi) ||
2942       ip_address_cmp (&sa->iaddr, &iaddr) ||
2943       ip_address_cmp (&sa->raddr, &raddr))
2944     {
2945       return 0;
2946     }
2947
2948   while (p < rlen && payload != IKEV2_PAYLOAD_NONE)
2949     {
2950       ike_payload_header_t *ikep = (ike_payload_header_t *) & ike->payload[p];
2951       u32 plen = clib_net_to_host_u16 (ikep->length);
2952
2953       if (plen < sizeof (ike_payload_header_t))
2954         return ~0;
2955
2956       if (payload == IKEV2_PAYLOAD_NONCE &&
2957           !clib_memcmp (sa->i_nonce, ikep->payload, plen - sizeof (*ikep)))
2958         {
2959           /* req is retransmit */
2960           if (sa->state == IKEV2_STATE_SA_INIT)
2961             {
2962               sa->stats.n_init_retransmit++;
2963               tmp = (ike_header_t *) sa->last_sa_init_res_packet_data;
2964               u32 slen = clib_net_to_host_u32 (tmp->length);
2965               ike->ispi = tmp->ispi;
2966               ike->rspi = tmp->rspi;
2967               ike->nextpayload = tmp->nextpayload;
2968               ike->version = tmp->version;
2969               ike->exchange = tmp->exchange;
2970               ike->flags = tmp->flags;
2971               ike->msgid = tmp->msgid;
2972               ike->length = tmp->length;
2973               clib_memcpy_fast (ike->payload, tmp->payload,
2974                                 slen - sizeof (*ike));
2975               ikev2_elog_uint_peers (IKEV2_LOG_DEBUG,
2976                                      "ispi %lx IKE_SA_INIT retransmit "
2977                                      "from %d.%d.%d.%d to %d.%d.%d.%d",
2978                                      ike->ispi,
2979                                      ip_addr_v4 (&raddr).as_u32,
2980                                      ip_addr_v4 (&iaddr).as_u32);
2981               return slen;
2982             }
2983           /* else ignore req */
2984           else
2985             {
2986               ikev2_elog_uint_peers (IKEV2_LOG_DEBUG,
2987                                      "ispi %lx IKE_SA_INIT ignore "
2988                                      "from %d.%d.%d.%d to %d.%d.%d.%d",
2989                                      ike->ispi,
2990                                      ip_addr_v4 (&raddr).as_u32,
2991                                      ip_addr_v4 (&iaddr).as_u32);
2992               return ~0;
2993             }
2994         }
2995       payload = ikep->nextpayload;
2996       p += plen;
2997     }
2998
2999   return 0;
3000 }
3001
3002 static u32
3003 ikev2_retransmit_sa_init (ike_header_t * ike, ip_address_t iaddr,
3004                           ip_address_t raddr, u32 rlen)
3005 {
3006   ikev2_sa_t *sa;
3007   u32 res;
3008   ikev2_main_per_thread_data_t *ptd = ikev2_get_per_thread_data ();
3009
3010   pool_foreach (sa, ptd->sas)  {
3011     res = ikev2_retransmit_sa_init_one (sa, ike, iaddr, raddr, rlen);
3012     if (res)
3013       return res;
3014   }
3015
3016   /* req is not retransmit */
3017   return 0;
3018 }
3019
3020 static u32
3021 ikev2_retransmit_resp (ikev2_sa_t * sa, ike_header_t * ike)
3022 {
3023   if (ike_hdr_is_response (ike))
3024     return 0;
3025
3026   u32 msg_id = clib_net_to_host_u32 (ike->msgid);
3027
3028   /* new req */
3029   if (msg_id > sa->last_msg_id || sa->last_msg_id == ~0)
3030     {
3031       sa->last_msg_id = msg_id;
3032       return 0;
3033     }
3034
3035   /* retransmitted req */
3036   if (msg_id == sa->last_msg_id)
3037     {
3038       sa->stats.n_retransmit++;
3039       ike_header_t *tmp = (ike_header_t *) sa->last_res_packet_data;
3040       u32 slen = clib_net_to_host_u32 (tmp->length);
3041       ike->ispi = tmp->ispi;
3042       ike->rspi = tmp->rspi;
3043       ike->nextpayload = tmp->nextpayload;
3044       ike->version = tmp->version;
3045       ike->exchange = tmp->exchange;
3046       ike->flags = tmp->flags;
3047       ike->msgid = tmp->msgid;
3048       ike->length = tmp->length;
3049       clib_memcpy_fast (ike->payload, tmp->payload, slen - sizeof (*ike));
3050       ikev2_elog_uint_peers (IKEV2_LOG_DEBUG, "IKE retransmit msgid %d",
3051                              msg_id, ip_addr_v4 (&sa->raddr).as_u32,
3052                              ip_addr_v4 (&sa->iaddr).as_u32);
3053       return slen;
3054     }
3055
3056   /* old req ignore */
3057   ikev2_elog_uint_peers (IKEV2_LOG_DEBUG, "IKE req ignore msgid %d",
3058                          msg_id, ip_addr_v4 (&sa->raddr).as_u32,
3059                          ip_addr_v4 (&sa->iaddr).as_u32);
3060   return ~0;
3061 }
3062
3063 static void
3064 ikev2_init_sa (vlib_main_t * vm, ikev2_sa_t * sa)
3065 {
3066   ikev2_main_t *km = &ikev2_main;
3067   sa->liveness_period_check = vlib_time_now (vm) + km->liveness_period;
3068   sa->profile_index = ~0;
3069 }
3070
3071 static void
3072 ikev2_del_sa_init_from_main (u64 * ispi)
3073 {
3074   ikev2_main_t *km = &ikev2_main;
3075   uword *p = hash_get (km->sa_by_ispi, *ispi);
3076   if (p)
3077     {
3078       ikev2_sa_t *sai = pool_elt_at_index (km->sais, p[0]);
3079       hash_unset (km->sa_by_ispi, sai->ispi);
3080       ikev2_sa_free_all_vec (sai);
3081       pool_put (km->sais, sai);
3082     }
3083 }
3084
3085 static void
3086 ikev2_del_sa_init (u64 ispi)
3087 {
3088   vl_api_rpc_call_main_thread (ikev2_del_sa_init_from_main, (u8 *) & ispi,
3089                                sizeof (ispi));
3090 }
3091
3092 static void
3093 ikev2_rewrite_v6_addrs (ikev2_sa_t *sa, ip6_header_t *ih)
3094 {
3095   if (sa->is_initiator)
3096     {
3097       ip_address_copy_addr (&ih->dst_address, &sa->raddr);
3098       ip_address_copy_addr (&ih->src_address, &sa->iaddr);
3099     }
3100   else
3101     {
3102       ip_address_copy_addr (&ih->dst_address, &sa->iaddr);
3103       ip_address_copy_addr (&ih->src_address, &sa->raddr);
3104     }
3105 }
3106
3107 static void
3108 ikev2_rewrite_v4_addrs (ikev2_sa_t *sa, ip4_header_t *ih)
3109 {
3110   if (sa->is_initiator)
3111     {
3112       ip_address_copy_addr (&ih->dst_address, &sa->raddr);
3113       ip_address_copy_addr (&ih->src_address, &sa->iaddr);
3114     }
3115   else
3116     {
3117       ip_address_copy_addr (&ih->dst_address, &sa->iaddr);
3118       ip_address_copy_addr (&ih->src_address, &sa->raddr);
3119     }
3120 }
3121
3122 static void
3123 ikev2_set_ip_address (ikev2_sa_t *sa, const void *iaddr, const void *raddr,
3124                       const ip_address_family_t af)
3125 {
3126   ip_address_set (&sa->raddr, raddr, af);
3127   ip_address_set (&sa->iaddr, iaddr, af);
3128 }
3129
3130 static void
3131 ikev2_elog_uint_peers_addr (u32 exchange, ip4_header_t * ip4,
3132                             ip6_header_t * ip6, u8 is_ip4)
3133 {
3134   u32 src, dst;
3135   if (is_ip4)
3136     {
3137       src = ip4->src_address.as_u32;
3138       dst = ip4->dst_address.as_u32;
3139     }
3140   else
3141     {
3142       src = ip6->src_address.as_u32[3];
3143       dst = ip6->dst_address.as_u32[3];
3144     }
3145   ikev2_elog_uint_peers (IKEV2_LOG_WARNING, "IKEv2 exchange %d "
3146                          "received from %d.%d.%d.%d to %d.%d.%d.%d",
3147                          exchange, src, dst);
3148 }
3149
3150 static void
3151 ikev2_generate_sa_init_data_and_log (ikev2_sa_t * sa)
3152 {
3153   ikev2_generate_sa_error_t rc = ikev2_generate_sa_init_data (sa);
3154
3155   if (PREDICT_TRUE (rc == IKEV2_GENERATE_SA_INIT_OK))
3156     return;
3157
3158   if (rc == IKEV2_GENERATE_SA_INIT_ERR_NO_DH)
3159     ikev2_elog_error (IKEV2_GENERATE_SA_INIT_OK_ERR_NO_DH_STR);
3160   else if (rc == IKEV2_GENERATE_SA_INIT_ERR_UNSUPPORTED_DH)
3161     ikev2_elog_error (IKEV2_GENERATE_SA_INIT_OK_ERR_UNSUPP_STR);
3162 }
3163
3164 static void
3165 ikev2_update_stats (vlib_main_t *vm, u32 node_index, ikev2_stats_t *s)
3166 {
3167   vlib_node_increment_counter (vm, node_index, IKEV2_ERROR_KEEPALIVE,
3168                                s->n_keepalives);
3169   vlib_node_increment_counter (vm, node_index, IKEV2_ERROR_REKEY_REQ,
3170                                s->n_rekey_req);
3171   vlib_node_increment_counter (vm, node_index, IKEV2_ERROR_INIT_SA_REQ,
3172                                s->n_sa_init_req);
3173   vlib_node_increment_counter (vm, node_index, IKEV2_ERROR_IKE_AUTH_REQ,
3174                                s->n_sa_auth_req);
3175 }
3176
3177 static uword
3178 ikev2_node_internal (vlib_main_t *vm, vlib_node_runtime_t *node,
3179                      vlib_frame_t *frame, u8 is_ip4, u8 natt)
3180 {
3181   u32 n_left = frame->n_vectors, *from;
3182   ikev2_main_t *km = &ikev2_main;
3183   vlib_buffer_t *bufs[VLIB_FRAME_SIZE], **b;
3184   u16 nexts[VLIB_FRAME_SIZE], *next = nexts;
3185   ikev2_main_per_thread_data_t *ptd = ikev2_get_per_thread_data ();
3186   ikev2_stats_t _stats, *stats = &_stats;
3187   int res;
3188
3189   /* no NAT traversal for ip6 */
3190   ASSERT (!natt || is_ip4);
3191
3192   clib_memset_u16 (stats, 0, sizeof (stats[0]) / sizeof (u16));
3193   from = vlib_frame_vector_args (frame);
3194   vlib_get_buffers (vm, from, bufs, n_left);
3195   b = bufs;
3196
3197   while (n_left > 0)
3198     {
3199       vlib_buffer_t *b0 = b[0];
3200       next[0] = is_ip4 ? IKEV2_NEXT_IP4_ERROR_DROP
3201         : IKEV2_NEXT_IP6_ERROR_DROP;
3202       ip4_header_t *ip40 = 0;
3203       ip6_header_t *ip60 = 0;
3204       udp_header_t *udp0;
3205       ike_header_t *ike0;
3206       ikev2_sa_t *sa0 = 0;
3207       ikev2_sa_t sa;            /* temporary store for SA */
3208       u32 rlen, slen = 0;
3209       int ip_hdr_sz = 0;
3210       int is_req = 0;
3211
3212       if (natt)
3213         {
3214           u8 *ptr = vlib_buffer_get_current (b0);
3215           ip40 = (ip4_header_t *) ptr;
3216           ptr += sizeof (*ip40);
3217           udp0 = (udp_header_t *) ptr;
3218           ptr += sizeof (*udp0);
3219           ike0 = (ike_header_t *) ptr;
3220           ip_hdr_sz = sizeof (*ip40);
3221         }
3222       else
3223         {
3224           u8 *ipx_hdr = b0->data + vnet_buffer (b0)->l3_hdr_offset;
3225           ike0 = vlib_buffer_get_current (b0);
3226           vlib_buffer_advance (b0, -sizeof (*udp0));
3227           udp0 = vlib_buffer_get_current (b0);
3228
3229           if (is_ip4)
3230             {
3231               ip40 = (ip4_header_t *) ipx_hdr;
3232               ip_hdr_sz = sizeof (*ip40);
3233             }
3234           else
3235             {
3236               ip60 = (ip6_header_t *) ipx_hdr;
3237               ip_hdr_sz = sizeof (*ip60);
3238             }
3239           vlib_buffer_advance (b0, -ip_hdr_sz);
3240         }
3241
3242       rlen = b0->current_length - ip_hdr_sz - sizeof (*udp0);
3243
3244       /* check for non-esp marker */
3245       if (natt)
3246         {
3247           ASSERT (*((u32 *) ike0) == 0);
3248           ike0 =
3249             (ike_header_t *) ((u8 *) ike0 + sizeof (ikev2_non_esp_marker));
3250           rlen -= sizeof (ikev2_non_esp_marker);
3251         }
3252
3253       if (clib_net_to_host_u32 (ike0->length) != rlen)
3254         {
3255           vlib_node_increment_counter (vm, node->node_index,
3256                                        IKEV2_ERROR_BAD_LENGTH, 1);
3257           goto dispatch0;
3258         }
3259
3260       if (ike0->version != IKE_VERSION_2)
3261         {
3262           vlib_node_increment_counter (vm, node->node_index,
3263                                        IKEV2_ERROR_NOT_IKEV2, 1);
3264           goto dispatch0;
3265         }
3266
3267       if (ike0->exchange == IKEV2_EXCHANGE_SA_INIT)
3268         {
3269           sa0 = &sa;
3270           clib_memset (sa0, 0, sizeof (*sa0));
3271
3272           if (ike_hdr_is_initiator (ike0))
3273             {
3274               sa0->stats.n_sa_init_req++;
3275               stats->n_sa_init_req++;
3276               if (ike0->rspi == 0)
3277                 {
3278                   if (is_ip4)
3279                     ikev2_set_ip_address (sa0, &ip40->src_address,
3280                                           &ip40->dst_address, AF_IP4);
3281                   else
3282                     ikev2_set_ip_address (sa0, &ip60->src_address,
3283                                           &ip60->dst_address, AF_IP6);
3284
3285                   sa0->dst_port = clib_net_to_host_u16 (udp0->src_port);
3286
3287                   slen =
3288                     ikev2_retransmit_sa_init (ike0, sa0->iaddr,
3289                                               sa0->raddr, rlen);
3290                   if (slen)
3291                     {
3292                       vlib_node_increment_counter (vm, node->node_index,
3293                                                    ~0 ==
3294                                                    slen ?
3295                                                    IKEV2_ERROR_IKE_SA_INIT_IGNORE
3296                                                    :
3297                                                    IKEV2_ERROR_IKE_SA_INIT_RETRANSMIT,
3298                                                    1);
3299                       goto dispatch0;
3300                     }
3301
3302                   res = ikev2_process_sa_init_req (
3303                     vm, sa0, ike0, udp0, rlen,
3304                     vnet_buffer (b0)->sw_if_index[VLIB_RX]);
3305                   if (!res)
3306                     vlib_node_increment_counter (vm, node->node_index,
3307                                                  IKEV2_ERROR_MALFORMED_PACKET,
3308                                                  1);
3309
3310                   if (sa0->state == IKEV2_STATE_SA_INIT)
3311                     {
3312                       ikev2_sa_free_proposal_vector (&sa0->r_proposals);
3313                       sa0->r_proposals =
3314                         ikev2_select_proposal (sa0->i_proposals,
3315                                                IKEV2_PROTOCOL_IKE);
3316                       ikev2_generate_sa_init_data_and_log (sa0);
3317                     }
3318
3319                   if (sa0->state == IKEV2_STATE_SA_INIT
3320                       || sa0->state == IKEV2_STATE_NOTIFY_AND_DELETE)
3321                     {
3322                       ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3323                       slen =
3324                         ikev2_generate_message (b0, sa0, ike0, 0, udp0, stats);
3325                       if (~0 == slen)
3326                         vlib_node_increment_counter (vm, node->node_index,
3327                                                      IKEV2_ERROR_NO_BUFF_SPACE,
3328                                                      1);
3329                     }
3330
3331                   if (sa0->state == IKEV2_STATE_SA_INIT)
3332                     {
3333                       /* add SA to the pool */
3334                       pool_get (ptd->sas, sa0);
3335                       clib_memcpy_fast (sa0, &sa, sizeof (*sa0));
3336                       ikev2_init_sa (vm, sa0);
3337                       hash_set (ptd->sa_by_rspi, sa0->rspi, sa0 - ptd->sas);
3338                     }
3339                   else
3340                     {
3341                       ikev2_sa_free_all_vec (sa0);
3342                     }
3343                 }
3344             }
3345           else                  //received sa_init without initiator flag
3346             {
3347               if (is_ip4)
3348                 ikev2_set_ip_address (sa0, &ip40->dst_address,
3349                                       &ip40->src_address, AF_IP4);
3350               else
3351                 ikev2_set_ip_address (sa0, &ip60->dst_address,
3352                                       &ip60->src_address, AF_IP6);
3353
3354               ikev2_process_sa_init_resp (vm, sa0, ike0, udp0, rlen);
3355
3356               if (sa0->state == IKEV2_STATE_SA_INIT)
3357                 {
3358                   is_req = 1;
3359                   ike0->exchange = IKEV2_EXCHANGE_IKE_AUTH;
3360                   uword *p = hash_get (km->sa_by_ispi, sa0->ispi);
3361                   if (p)
3362                     {
3363                       ikev2_sa_t *sai = pool_elt_at_index (km->sais, p[0]);
3364
3365                       if (clib_atomic_bool_cmp_and_swap
3366                           (&sai->init_response_received, 0, 1))
3367                         {
3368                           ikev2_complete_sa_data (sa0, sai);
3369                           ikev2_calc_keys (sa0);
3370                           ikev2_sa_auth_init (sa0);
3371                           ike0->flags = IKEV2_HDR_FLAG_INITIATOR;
3372                           ike0->msgid =
3373                             clib_net_to_host_u32 (sai->last_init_msg_id);
3374                           sa0->last_init_msg_id = sai->last_init_msg_id + 1;
3375                           slen = ikev2_generate_message (b0, sa0, ike0, 0,
3376                                                          udp0, stats);
3377                           if (~0 == slen)
3378                             vlib_node_increment_counter (vm,
3379                                                          node->node_index,
3380                                                          IKEV2_ERROR_NO_BUFF_SPACE,
3381                                                          1);
3382                         }
3383                       else
3384                         {
3385                           /* we've already processed sa-init response */
3386                           sa0->state = IKEV2_STATE_UNKNOWN;
3387                         }
3388                     }
3389                 }
3390
3391               if (sa0->state == IKEV2_STATE_SA_INIT)
3392                 {
3393                   /* add SA to the pool */
3394                   pool_get (ptd->sas, sa0);
3395                   clib_memcpy_fast (sa0, &sa, sizeof (*sa0));
3396                   hash_set (ptd->sa_by_rspi, sa0->rspi, sa0 - ptd->sas);
3397                 }
3398               else
3399                 {
3400                   ikev2_sa_free_all_vec (sa0);
3401                 }
3402             }
3403         }
3404       else if (ike0->exchange == IKEV2_EXCHANGE_IKE_AUTH)
3405         {
3406           uword *p;
3407           p = hash_get (ptd->sa_by_rspi, clib_net_to_host_u64 (ike0->rspi));
3408           if (p)
3409             {
3410               sa0 = pool_elt_at_index (ptd->sas, p[0]);
3411               slen = ikev2_retransmit_resp (sa0, ike0);
3412               if (slen)
3413                 {
3414                   vlib_node_increment_counter (vm, node->node_index,
3415                                                ~0 ==
3416                                                slen ?
3417                                                IKEV2_ERROR_IKE_REQ_IGNORE
3418                                                :
3419                                                IKEV2_ERROR_IKE_REQ_RETRANSMIT,
3420                                                1);
3421                   goto dispatch0;
3422                 }
3423
3424               sa0->dst_port = clib_net_to_host_u16 (udp0->src_port);
3425               res = ikev2_process_auth_req (vm, sa0, ike0, rlen);
3426               if (res)
3427                 ikev2_sa_auth (sa0);
3428               else
3429                 vlib_node_increment_counter (vm, node->node_index,
3430                                              IKEV2_ERROR_MALFORMED_PACKET, 1);
3431               if (sa0->state == IKEV2_STATE_AUTHENTICATED)
3432                 {
3433                   ikev2_initial_contact_cleanup (ptd, sa0);
3434                   p = hash_get (ptd->sa_by_rspi,
3435                                 clib_net_to_host_u64 (ike0->rspi));
3436                   ikev2_sa_match_ts (sa0);
3437                   if (sa0->state != IKEV2_STATE_TS_UNACCEPTABLE)
3438                     ikev2_create_tunnel_interface (vm, sa0, &sa0->childs[0],
3439                                                    p[0], 0, 0, 0);
3440                 }
3441
3442               if (sa0->is_initiator)
3443                 {
3444                   sa0->last_msg_id = ~0;
3445                   ikev2_del_sa_init (sa0->ispi);
3446                 }
3447               else
3448                 {
3449                   sa0->stats.n_sa_auth_req++;
3450                   stats->n_sa_auth_req++;
3451                   ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3452                   slen =
3453                     ikev2_generate_message (b0, sa0, ike0, 0, udp0, stats);
3454                   if (~0 == slen)
3455                     vlib_node_increment_counter (vm, node->node_index,
3456                                                  IKEV2_ERROR_NO_BUFF_SPACE,
3457                                                  1);
3458                 }
3459             }
3460         }
3461       else if (ike0->exchange == IKEV2_EXCHANGE_INFORMATIONAL)
3462         {
3463           uword *p;
3464           p = hash_get (ptd->sa_by_rspi, clib_net_to_host_u64 (ike0->rspi));
3465           if (p)
3466             {
3467               sa0 = pool_elt_at_index (ptd->sas, p[0]);
3468               slen = ikev2_retransmit_resp (sa0, ike0);
3469               if (slen)
3470                 {
3471                   vlib_node_increment_counter (vm, node->node_index,
3472                                                ~0 ==
3473                                                slen ?
3474                                                IKEV2_ERROR_IKE_REQ_IGNORE
3475                                                :
3476                                                IKEV2_ERROR_IKE_REQ_RETRANSMIT,
3477                                                1);
3478                   goto dispatch0;
3479                 }
3480
3481               res = ikev2_process_informational_req (vm, sa0, ike0, rlen);
3482               if (!res)
3483                 {
3484                   vlib_node_increment_counter (vm, node->node_index,
3485                                                IKEV2_ERROR_MALFORMED_PACKET,
3486                                                1);
3487                   slen = ~0;
3488                   goto dispatch0;
3489                 }
3490
3491               if (sa0->del)
3492                 {
3493                   if (sa0->del[0].protocol_id != IKEV2_PROTOCOL_IKE)
3494                     {
3495                       ikev2_delete_t *d, *tmp, *resp = 0;
3496                       vec_foreach (d, sa0->del)
3497                       {
3498                         ikev2_child_sa_t *ch_sa;
3499                         ch_sa = ikev2_sa_get_child (sa0, d->spi,
3500                                                     d->protocol_id,
3501                                                     !sa0->is_initiator);
3502                         if (ch_sa)
3503                           {
3504                             ikev2_delete_tunnel_interface (km->vnet_main,
3505                                                            sa0, ch_sa);
3506                             if (!sa0->is_initiator)
3507                               {
3508                                 vec_add2 (resp, tmp, 1);
3509                                 tmp->protocol_id = d->protocol_id;
3510                                 tmp->spi = ch_sa->r_proposals[0].spi;
3511                               }
3512                             ikev2_sa_del_child_sa (sa0, ch_sa);
3513                           }
3514                       }
3515                       if (!sa0->is_initiator)
3516                         {
3517                           vec_free (sa0->del);
3518                           sa0->del = resp;
3519                         }
3520                     }
3521                 }
3522               if (ike_hdr_is_request (ike0))
3523                 {
3524                   ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3525                   slen =
3526                     ikev2_generate_message (b0, sa0, ike0, 0, udp0, stats);
3527                   if (~0 == slen)
3528                     vlib_node_increment_counter (vm, node->node_index,
3529                                                  IKEV2_ERROR_NO_BUFF_SPACE,
3530                                                  1);
3531                 }
3532             }
3533         }
3534       else if (ike0->exchange == IKEV2_EXCHANGE_CREATE_CHILD_SA)
3535         {
3536           uword *p;
3537           p = hash_get (ptd->sa_by_rspi, clib_net_to_host_u64 (ike0->rspi));
3538           if (p)
3539             {
3540               sa0 = pool_elt_at_index (ptd->sas, p[0]);
3541               slen = ikev2_retransmit_resp (sa0, ike0);
3542               if (slen)
3543                 {
3544                   vlib_node_increment_counter (vm, node->node_index,
3545                                                ~0 ==
3546                                                slen ?
3547                                                IKEV2_ERROR_IKE_REQ_IGNORE
3548                                                :
3549                                                IKEV2_ERROR_IKE_REQ_RETRANSMIT,
3550                                                1);
3551                   goto dispatch0;
3552                 }
3553
3554               res = ikev2_process_create_child_sa_req (vm, sa0, ike0, rlen);
3555               if (!res)
3556                 {
3557                   vlib_node_increment_counter (vm, node->node_index,
3558                                                IKEV2_ERROR_MALFORMED_PACKET,
3559                                                1);
3560                   slen = ~0;
3561                   goto dispatch0;
3562                 }
3563
3564               if (vec_len (sa0->rekey) > 0)
3565                 {
3566                   if (!sa0->rekey[0].notify_type &&
3567                       sa0->rekey[0].protocol_id != IKEV2_PROTOCOL_IKE)
3568                     {
3569                       if (vec_len (sa0->childs) > 0)
3570                         ikev2_sa_free_all_child_sa (&sa0->childs);
3571                       ikev2_child_sa_t *child;
3572                       vec_add2 (sa0->childs, child, 1);
3573                       clib_memset (child, 0, sizeof (*child));
3574                       child->r_proposals = sa0->rekey[0].r_proposal;
3575                       child->i_proposals = sa0->rekey[0].i_proposal;
3576                       child->tsi = sa0->rekey[0].tsi;
3577                       child->tsr = sa0->rekey[0].tsr;
3578                       ikev2_create_tunnel_interface (vm, sa0, child, p[0],
3579                                                      child - sa0->childs, 1,
3580                                                      sa0->rekey[0].kex);
3581                     }
3582                   if (ike_hdr_is_response (ike0))
3583                     {
3584                       vec_free (sa0->rekey);
3585                     }
3586                   else
3587                     {
3588                       stats->n_rekey_req++;
3589                       sa0->stats.n_rekey_req++;
3590                       ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3591                       slen =
3592                         ikev2_generate_message (b0, sa0, ike0, 0, udp0, stats);
3593                       if (~0 == slen)
3594                         vlib_node_increment_counter (vm, node->node_index,
3595                                                      IKEV2_ERROR_NO_BUFF_SPACE,
3596                                                      1);
3597                     }
3598                 }
3599               else if (vec_len (sa0->new_child) > 0)
3600                 {
3601                   ikev2_child_sa_t *c;
3602                   vec_add2 (sa0->childs, c, 1);
3603                   memset (c, 0, sizeof (*c));
3604                   c->r_proposals = sa0->new_child[0].r_proposal;
3605                   c->i_proposals = sa0->new_child[0].i_proposal;
3606                   c->tsi = sa0->new_child[0].tsi;
3607                   c->tsr = sa0->new_child[0].tsr;
3608                   ikev2_create_tunnel_interface (vm, sa0, c, p[0],
3609                                                  c - sa0->childs, 0,
3610                                                  sa0->new_child[0].kex);
3611                   if (ike_hdr_is_request (ike0))
3612                     {
3613                       ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3614                       slen =
3615                         ikev2_generate_message (b0, sa0, ike0, 0, udp0, 0);
3616                       if (~0 == slen)
3617                         vlib_node_increment_counter (
3618                           vm, node->node_index, IKEV2_ERROR_NO_BUFF_SPACE, 1);
3619                     }
3620                 }
3621               else if (vec_len (sa0->sa_rekey) > 0)
3622                 {
3623                   if (!sa0->sa_rekey[0].notify_type)
3624                     {
3625                       ikev2_sa_t *sar, *tmp = 0;
3626                       pool_get (ptd->sas, tmp);
3627                       sa0 = pool_elt_at_index (ptd->sas, p[0]);
3628                       /* swap old/new SAs to keep index and inherit IPsec SA */
3629                       clib_memcpy_fast (tmp, sa0, sizeof (*tmp));
3630                       sar = sa0;
3631                       sa0 = tmp;
3632                       hash_set (ptd->sa_by_rspi, sa0->rspi, sa0 - ptd->sas);
3633                       p = hash_get (ptd->sa_by_rspi, sa0->rspi);
3634                       ikev2_complete_sa_rekey (sar, sa0, &sa0->sa_rekey[0]);
3635                       hash_set (ptd->sa_by_rspi, sar->rspi, sar - ptd->sas);
3636                     }
3637                   if (ike_hdr_is_response (ike0))
3638                     {
3639                       vec_free (sa0->sa_rekey);
3640                     }
3641                   else
3642                     {
3643                       stats->n_rekey_req++;
3644                       sa0->stats.n_rekey_req++;
3645                       ike0->flags = IKEV2_HDR_FLAG_RESPONSE;
3646                       slen =
3647                         ikev2_generate_message (b0, sa0, ike0, 0, udp0, stats);
3648                       if (~0 == slen)
3649                         vlib_node_increment_counter (
3650                           vm, node->node_index, IKEV2_ERROR_NO_BUFF_SPACE, 1);
3651                     }
3652                 }
3653             }
3654         }
3655       else
3656         {
3657           ikev2_elog_uint_peers_addr (ike0->exchange, ip40, ip60, is_ip4);
3658         }
3659
3660     dispatch0:
3661       /* if we are sending packet back, rewrite headers */
3662       if (slen && ~0 != slen)
3663         {
3664           if (is_ip4)
3665             {
3666               next[0] = IKEV2_NEXT_IP4_LOOKUP;
3667               ikev2_rewrite_v4_addrs (sa0, ip40);
3668             }
3669           else
3670             {
3671               next[0] = IKEV2_NEXT_IP6_LOOKUP;
3672               ikev2_rewrite_v6_addrs (sa0, ip60);
3673             }
3674
3675           if (is_req)
3676             {
3677               udp0->dst_port = udp0->src_port =
3678                 clib_net_to_host_u16 (ikev2_get_port (sa0));
3679
3680               if (udp0->dst_port == clib_net_to_host_u16 (IKEV2_PORT_NATT)
3681                   && ikev2_natt_active (sa0))
3682                 {
3683                   if (!natt)
3684                     slen = ikev2_insert_non_esp_marker (ike0, slen);
3685                 }
3686             }
3687           else
3688             {
3689               if (natt)
3690                 slen += sizeof (ikev2_non_esp_marker);
3691
3692               u16 tp = udp0->dst_port;
3693               udp0->dst_port = udp0->src_port;
3694               udp0->src_port = tp;
3695             }
3696
3697           udp0->length = clib_host_to_net_u16 (slen + sizeof (udp_header_t));
3698           udp0->checksum = 0;
3699           b0->current_length = slen + ip_hdr_sz + sizeof (udp_header_t);
3700           if (is_ip4)
3701             {
3702               ip40->length = clib_host_to_net_u16 (b0->current_length);
3703               ip40->checksum = ip4_header_checksum (ip40);
3704             }
3705           else
3706             {
3707               ip60->payload_length =
3708                 clib_host_to_net_u16 (b0->current_length - sizeof (*ip60));
3709             }
3710         }
3711       /* delete sa */
3712       if (sa0 && (sa0->state == IKEV2_STATE_DELETED ||
3713                   sa0->state == IKEV2_STATE_NOTIFY_AND_DELETE))
3714         {
3715           ikev2_child_sa_t *c;
3716
3717           vec_foreach (c, sa0->childs)
3718             ikev2_delete_tunnel_interface (km->vnet_main, sa0, c);
3719
3720           ikev2_delete_sa (ptd, sa0);
3721         }
3722       if (PREDICT_FALSE ((node->flags & VLIB_NODE_FLAG_TRACE)
3723                          && (b0->flags & VLIB_BUFFER_IS_TRACED)))
3724         {
3725
3726           ikev2_trace_t *t = vlib_add_trace (vm, node, b0, sizeof (*t));
3727           t->sw_if_index = vnet_buffer (b0)->sw_if_index[VLIB_RX];
3728           t->next_index = next[0];
3729         }
3730       n_left -= 1;
3731       next += 1;
3732       b += 1;
3733     }
3734
3735   ikev2_update_stats (vm, node->node_index, stats);
3736   vlib_node_increment_counter (vm, node->node_index,
3737                                IKEV2_ERROR_PROCESSED, frame->n_vectors);
3738   vlib_buffer_enqueue_to_next (vm, node, from, nexts, frame->n_vectors);
3739   return frame->n_vectors;
3740 }
3741
3742 static uword
3743 ikev2_ip4 (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
3744 {
3745   return ikev2_node_internal (vm, node, frame, 1 /* is_ip4 */, 0);
3746 }
3747
3748 static uword
3749 ikev2_ip4_natt (vlib_main_t *vm, vlib_node_runtime_t *node,
3750                 vlib_frame_t *frame)
3751 {
3752   return ikev2_node_internal (vm, node, frame, 1 /* is_ip4 */, 1 /* natt */);
3753 }
3754
3755 static uword
3756 ikev2_ip6 (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
3757 {
3758   return ikev2_node_internal (vm, node, frame, 0 /* is_ip4 */, 0);
3759 }
3760
3761 VLIB_REGISTER_NODE (ikev2_node_ip4,static) = {
3762   .function = ikev2_ip4,
3763   .name = "ikev2-ip4",
3764   .vector_size = sizeof (u32),
3765   .format_trace = format_ikev2_trace,
3766   .type = VLIB_NODE_TYPE_INTERNAL,
3767
3768   .n_errors = IKEV2_N_ERROR,
3769   .error_counters = ikev2_error_counters,
3770
3771   .n_next_nodes = IKEV2_IP4_N_NEXT,
3772   .next_nodes = {
3773     [IKEV2_NEXT_IP4_LOOKUP] = "ip4-lookup",
3774     [IKEV2_NEXT_IP4_ERROR_DROP] = "error-drop",
3775   },
3776 };
3777
3778 VLIB_REGISTER_NODE (ikev2_node_ip4_natt,static) = {
3779   .function = ikev2_ip4_natt,
3780   .name = "ikev2-ip4-natt",
3781   .vector_size = sizeof (u32),
3782   .format_trace = format_ikev2_trace,
3783   .type = VLIB_NODE_TYPE_INTERNAL,
3784
3785   .n_errors = IKEV2_N_ERROR,
3786   .error_counters = ikev2_error_counters,
3787
3788   .n_next_nodes = IKEV2_IP4_N_NEXT,
3789   .next_nodes = {
3790     [IKEV2_NEXT_IP4_LOOKUP] = "ip4-lookup",
3791     [IKEV2_NEXT_IP4_ERROR_DROP] = "error-drop",
3792   },
3793 };
3794
3795 VLIB_REGISTER_NODE (ikev2_node_ip6,static) = {
3796   .function = ikev2_ip6,
3797   .name = "ikev2-ip6",
3798   .vector_size = sizeof (u32),
3799   .format_trace = format_ikev2_trace,
3800   .type = VLIB_NODE_TYPE_INTERNAL,
3801
3802   .n_errors = IKEV2_N_ERROR,
3803   .error_counters = ikev2_error_counters,
3804
3805   .n_next_nodes = IKEV2_IP6_N_NEXT,
3806   .next_nodes = {
3807     [IKEV2_NEXT_IP6_LOOKUP] = "ip6-lookup",
3808     [IKEV2_NEXT_IP6_ERROR_DROP] = "error-drop",
3809   },
3810 };
3811
3812 // set ikev2 proposals when vpp is used as initiator
3813 static clib_error_t *
3814 ikev2_set_initiator_proposals (vlib_main_t * vm, ikev2_sa_t * sa,
3815                                ikev2_transforms_set * ts,
3816                                ikev2_sa_proposal_t ** proposals, int is_ike)
3817 {
3818   clib_error_t *r;
3819   ikev2_main_t *km = &ikev2_main;
3820   ikev2_sa_proposal_t *proposal;
3821   vec_add2 (*proposals, proposal, 1);
3822   ikev2_sa_transform_t *td;
3823   int error;
3824
3825   /* Encryption */
3826   error = 1;
3827   vec_foreach (td, km->supported_transforms)
3828   {
3829     if (td->type == IKEV2_TRANSFORM_TYPE_ENCR
3830         && td->encr_type == ts->crypto_alg
3831         && td->key_len == ts->crypto_key_size / 8)
3832       {
3833         u16 attr[2];
3834         attr[0] = clib_host_to_net_u16 (14 | (1 << 15));
3835         attr[1] = clib_host_to_net_u16 (td->key_len << 3);
3836         vec_add (td->attrs, (u8 *) attr, 4);
3837         vec_add1 (proposal->transforms, *td);
3838         td->attrs = 0;
3839
3840         error = 0;
3841         break;
3842       }
3843   }
3844   if (error)
3845     {
3846       r = clib_error_return (0, "Unsupported algorithm");
3847       return r;
3848     }
3849
3850   if (IKEV2_TRANSFORM_INTEG_TYPE_NONE != ts->integ_alg)
3851     {
3852       /* Integrity */
3853       error = 1;
3854       vec_foreach (td, km->supported_transforms)
3855       {
3856         if (td->type == IKEV2_TRANSFORM_TYPE_INTEG
3857             && td->integ_type == ts->integ_alg)
3858           {
3859             vec_add1 (proposal->transforms, *td);
3860             error = 0;
3861             break;
3862           }
3863       }
3864       if (error)
3865         {
3866           ikev2_elog_error
3867             ("Didn't find any supported algorithm for IKEV2_TRANSFORM_TYPE_INTEG");
3868           r = clib_error_return (0, "Unsupported algorithm");
3869           return r;
3870         }
3871     }
3872
3873   /* PRF */
3874   if (is_ike)
3875     {
3876       error = 1;
3877       vec_foreach (td, km->supported_transforms)
3878       {
3879         if (td->type == IKEV2_TRANSFORM_TYPE_PRF
3880             && td->prf_type == IKEV2_TRANSFORM_PRF_TYPE_PRF_HMAC_SHA2_256)
3881           {
3882             vec_add1 (proposal->transforms, *td);
3883             error = 0;
3884             break;
3885           }
3886       }
3887       if (error)
3888         {
3889           r = clib_error_return (0, "Unsupported algorithm");
3890           return r;
3891         }
3892     }
3893
3894   /* DH */
3895   if (is_ike)
3896     {
3897       error = 1;
3898       vec_foreach (td, km->supported_transforms)
3899       {
3900         if (td->type == IKEV2_TRANSFORM_TYPE_DH && td->dh_type == ts->dh_type)
3901           {
3902             vec_add1 (proposal->transforms, *td);
3903             if (is_ike)
3904               {
3905                 sa->dh_group = td->dh_type;
3906               }
3907             error = 0;
3908             break;
3909           }
3910       }
3911       if (error)
3912         {
3913           r = clib_error_return (0, "Unsupported algorithm");
3914           return r;
3915         }
3916     }
3917
3918   if (!is_ike)
3919     {
3920       error = 1;
3921       vec_foreach (td, km->supported_transforms)
3922       {
3923         if (td->type == IKEV2_TRANSFORM_TYPE_ESN)
3924           {
3925             vec_add1 (proposal->transforms, *td);
3926             error = 0;
3927           }
3928       }
3929       if (error)
3930         {
3931           r = clib_error_return (0, "Unsupported algorithm");
3932           return r;
3933         }
3934     }
3935
3936
3937   return 0;
3938 }
3939
3940 static ikev2_profile_t *
3941 ikev2_profile_index_by_name (u8 * name)
3942 {
3943   ikev2_main_t *km = &ikev2_main;
3944   uword *p;
3945
3946   p = mhash_get (&km->profile_index_by_name, name);
3947   if (!p)
3948     return 0;
3949
3950   return pool_elt_at_index (km->profiles, p[0]);
3951 }
3952
3953
3954 static void
3955 ikev2_send_ike (vlib_main_t * vm, ip_address_t * src, ip_address_t * dst,
3956                 u32 bi0, u32 len, u16 src_port, u16 dst_port, u32 sw_if_index)
3957 {
3958   ip4_header_t *ip40;
3959   ip6_header_t *ip60;
3960   udp_header_t *udp0;
3961   vlib_buffer_t *b0;
3962   vlib_frame_t *f;
3963   u32 *to_next;
3964
3965   b0 = vlib_get_buffer (vm, bi0);
3966   vlib_buffer_advance (b0, -sizeof (udp_header_t));
3967   udp0 = vlib_buffer_get_current (b0);
3968   udp0->dst_port = clib_host_to_net_u16 (dst_port);
3969   udp0->src_port = clib_host_to_net_u16 (src_port);
3970   udp0->length = clib_host_to_net_u16 (len + sizeof (udp_header_t));
3971   udp0->checksum = 0;
3972
3973   if (ip_addr_version (dst) == AF_IP4)
3974     {
3975       vlib_buffer_advance (b0, -sizeof (ip4_header_t));
3976       ip40 = vlib_buffer_get_current (b0);
3977       ip40->ip_version_and_header_length = 0x45;
3978       ip40->tos = 0;
3979       ip40->fragment_id = 0;
3980       ip40->flags_and_fragment_offset = 0;
3981       ip40->ttl = 0xff;
3982       ip40->protocol = IP_PROTOCOL_UDP;
3983       ip40->dst_address.as_u32 = ip_addr_v4 (dst).as_u32;
3984       ip40->src_address.as_u32 = ip_addr_v4 (src).as_u32;
3985       b0->current_length =
3986         len + sizeof (ip4_header_t) + sizeof (udp_header_t);
3987       ip40->length = clib_host_to_net_u16 (b0->current_length);
3988       ip40->checksum = ip4_header_checksum (ip40);
3989     }
3990   else
3991     {
3992       vlib_buffer_advance (b0, -sizeof (ip6_header_t));
3993       ip60 = vlib_buffer_get_current (b0);
3994
3995       b0->current_length = len + sizeof (*ip60) + sizeof (udp_header_t);
3996       ip60->ip_version_traffic_class_and_flow_label =
3997         clib_host_to_net_u32 (0x6 << 28);
3998       ip60->payload_length =
3999         clib_host_to_net_u16 (b0->current_length - sizeof (*ip60));
4000       ip60->protocol = IP_PROTOCOL_UDP;
4001       ip60->hop_limit = 0xff;
4002       clib_memcpy_fast (ip60->src_address.as_u8, ip_addr_v6 (src).as_u8,
4003                         sizeof (ip60->src_address));
4004       clib_memcpy_fast (ip60->dst_address.as_u8, ip_addr_v6 (dst).as_u8,
4005                         sizeof (ip60->src_address));
4006     }
4007
4008   b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
4009   vnet_buffer (b0)->sw_if_index[VLIB_RX] = sw_if_index;
4010   vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0;
4011
4012   u32 next_index = (ip_addr_version (dst) == AF_IP4) ?
4013     ip4_lookup_node.index : ip6_lookup_node.index;
4014
4015   /* send the request */
4016   f = vlib_get_frame_to_node (vm, next_index);
4017   to_next = vlib_frame_vector_args (f);
4018   to_next[0] = bi0;
4019   f->n_vectors = 1;
4020   vlib_put_frame_to_node (vm, next_index, f);
4021
4022 }
4023
4024 static u32
4025 ikev2_get_new_ike_header_buff (vlib_main_t * vm, vlib_buffer_t ** b)
4026 {
4027   u32 bi0;
4028   if (vlib_buffer_alloc (vm, &bi0, 1) != 1)
4029     {
4030       *b = 0;
4031       return 0;
4032     }
4033   *b = vlib_get_buffer (vm, bi0);
4034   return bi0;
4035 }
4036
4037 clib_error_t *
4038 ikev2_set_local_key (vlib_main_t * vm, u8 * file)
4039 {
4040   ikev2_main_t *km = &ikev2_main;
4041
4042   if (km->pkey)
4043     EVP_PKEY_free (km->pkey);
4044   km->pkey = ikev2_load_key_file (file);
4045   if (km->pkey == NULL)
4046     return clib_error_return (0, "load key '%s' failed", file);
4047
4048   return 0;
4049 }
4050
4051 static vnet_api_error_t
4052 ikev2_register_udp_port (ikev2_profile_t *p, u16 port)
4053 {
4054   ipsec_register_udp_port (port, 0 /* is_ip4 */);
4055   ipsec_register_udp_port (port, 1 /* is_ip4 */);
4056   p->ipsec_over_udp_port = port;
4057   return 0;
4058 }
4059
4060 static void
4061 ikev2_unregister_udp_port (ikev2_profile_t *p)
4062 {
4063   if (p->ipsec_over_udp_port == IPSEC_UDP_PORT_NONE)
4064     return;
4065
4066   ipsec_unregister_udp_port (p->ipsec_over_udp_port, 0 /* is_ip4 */);
4067   ipsec_unregister_udp_port (p->ipsec_over_udp_port, 1 /* is_ip4 */);
4068   p->ipsec_over_udp_port = IPSEC_UDP_PORT_NONE;
4069 }
4070
4071 static void
4072 ikev2_initiate_delete_ike_sa_internal (vlib_main_t * vm,
4073                                        ikev2_main_per_thread_data_t * tkm,
4074                                        ikev2_sa_t * sa, u8 send_notification)
4075 {
4076   ikev2_main_t *km = &ikev2_main;
4077   ip_address_t *src, *dst;
4078   vlib_buffer_t *b0;
4079   ikev2_child_sa_t *c;
4080
4081   /* Create the Initiator notification for IKE SA removal */
4082   ike_header_t *ike0;
4083   u32 bi0 = 0;
4084   int len;
4085
4086   vec_resize (sa->del, 1);
4087   sa->del->protocol_id = IKEV2_PROTOCOL_IKE;
4088   sa->del->spi = sa->ispi;
4089
4090   if (send_notification)
4091     {
4092       bi0 = ikev2_get_new_ike_header_buff (vm, &b0);
4093       if (!bi0)
4094         {
4095           ikev2_log_error ("buffer alloc failure");
4096           goto delete_sa;
4097         }
4098
4099       ike0 = vlib_buffer_get_current (b0);
4100       ike0->exchange = IKEV2_EXCHANGE_INFORMATIONAL;
4101       ike0->ispi = clib_host_to_net_u64 (sa->ispi);
4102       ike0->rspi = clib_host_to_net_u64 (sa->rspi);
4103       ike0->flags = 0;
4104       ike0->msgid = clib_host_to_net_u32 (sa->last_init_msg_id);
4105       sa->last_init_msg_id += 1;
4106       len = ikev2_generate_message (b0, sa, ike0, 0, 0, 0);
4107       if (~0 == len)
4108         return;
4109
4110       if (ikev2_natt_active (sa))
4111         len = ikev2_insert_non_esp_marker (ike0, len);
4112
4113       if (sa->is_initiator)
4114         {
4115           src = &sa->iaddr;
4116           dst = &sa->raddr;
4117         }
4118       else
4119         {
4120           dst = &sa->iaddr;
4121           src = &sa->raddr;
4122         }
4123
4124       ikev2_send_ike (vm, src, dst, bi0, len,
4125                       ikev2_get_port (sa), sa->dst_port, 0);
4126     }
4127
4128 delete_sa:
4129   /* delete local SA */
4130   vec_foreach (c, sa->childs)
4131     ikev2_delete_tunnel_interface (km->vnet_main, sa, c);
4132
4133   u64 rspi = sa->rspi;
4134   ikev2_sa_free_all_vec (sa);
4135   uword *p = hash_get (tkm->sa_by_rspi, rspi);
4136   if (p)
4137     {
4138       hash_unset (tkm->sa_by_rspi, rspi);
4139       pool_put (tkm->sas, sa);
4140     }
4141 }
4142
4143 static void
4144 ikev2_cleanup_profile_sessions (ikev2_main_t * km, ikev2_profile_t * p)
4145 {
4146   ikev2_main_per_thread_data_t *tkm;
4147   ikev2_sa_t *sa;
4148   u32 pi = p - km->profiles;
4149   u32 *sai;
4150   u32 *del_sai = 0;
4151
4152   pool_foreach (sa, km->sais)  {
4153     if (pi == sa->profile_index)
4154       vec_add1 (del_sai, sa - km->sais);
4155   }
4156
4157   vec_foreach (sai, del_sai)
4158   {
4159     sa = pool_elt_at_index (km->sais, sai[0]);
4160     ikev2_sa_free_all_vec (sa);
4161     hash_unset (km->sa_by_ispi, sa->ispi);
4162     pool_put (km->sais, sa);
4163   }
4164   vec_reset_length (del_sai);
4165
4166   vec_foreach (tkm, km->per_thread_data)
4167   {
4168     pool_foreach (sa, tkm->sas)  {
4169       if (sa->profile_index != ~0 && pi == sa->profile_index)
4170         vec_add1 (del_sai, sa - tkm->sas);
4171     }
4172
4173     vec_foreach (sai, del_sai)
4174     {
4175       sa = pool_elt_at_index (tkm->sas, sai[0]);
4176       ikev2_initiate_delete_ike_sa_internal (km->vlib_main, tkm, sa, 1);
4177     }
4178
4179     vec_reset_length (del_sai);
4180   }
4181
4182   vec_free (del_sai);
4183 }
4184
4185 static void
4186 ikev2_profile_responder_free (ikev2_responder_t *r)
4187 {
4188   vec_free (r->hostname);
4189 }
4190
4191 static void
4192 ikev2_profile_free (ikev2_profile_t * p)
4193 {
4194   vec_free (p->name);
4195
4196   vec_free (p->auth.data);
4197   if (p->auth.key)
4198     EVP_PKEY_free (p->auth.key);
4199
4200   ikev2_profile_responder_free (&p->responder);
4201
4202   vec_free (p->loc_id.data);
4203   vec_free (p->rem_id.data);
4204 }
4205
4206 static void
4207 ikev2_bind (vlib_main_t *vm, ikev2_main_t *km)
4208 {
4209   if (0 == km->bind_refcount)
4210     {
4211       udp_register_dst_port (vm, IKEV2_PORT, ikev2_node_ip4.index, 1);
4212       udp_register_dst_port (vm, IKEV2_PORT, ikev2_node_ip6.index, 0);
4213       udp_register_dst_port (vm, IKEV2_PORT_NATT, ikev2_node_ip4.index, 1);
4214       udp_register_dst_port (vm, IKEV2_PORT_NATT, ikev2_node_ip6.index, 0);
4215
4216       vlib_punt_register (km->punt_hdl,
4217                           ipsec_punt_reason[IPSEC_PUNT_IP4_SPI_UDP_0],
4218                           "ikev2-ip4-natt");
4219     }
4220
4221   km->bind_refcount++;
4222 }
4223
4224 static void
4225 ikev2_unbind (vlib_main_t *vm, ikev2_main_t *km)
4226 {
4227   km->bind_refcount--;
4228   if (0 == km->bind_refcount)
4229     {
4230       vlib_punt_unregister (km->punt_hdl,
4231                             ipsec_punt_reason[IPSEC_PUNT_IP4_SPI_UDP_0],
4232                             "ikev2-ip4-natt");
4233
4234       udp_unregister_dst_port (vm, IKEV2_PORT_NATT, 0);
4235       udp_unregister_dst_port (vm, IKEV2_PORT_NATT, 1);
4236       udp_unregister_dst_port (vm, IKEV2_PORT, 0);
4237       udp_unregister_dst_port (vm, IKEV2_PORT, 1);
4238     }
4239 }
4240
4241 static void ikev2_lazy_init (ikev2_main_t *km);
4242
4243 clib_error_t *
4244 ikev2_add_del_profile (vlib_main_t * vm, u8 * name, int is_add)
4245 {
4246   ikev2_main_t *km = &ikev2_main;
4247   ikev2_profile_t *p;
4248
4249   ikev2_lazy_init (km);
4250
4251   if (is_add)
4252     {
4253       if (ikev2_profile_index_by_name (name))
4254         return clib_error_return (0, "policy %v already exists", name);
4255
4256       pool_get (km->profiles, p);
4257       clib_memset (p, 0, sizeof (*p));
4258       p->name = vec_dup (name);
4259       p->ipsec_over_udp_port = IPSEC_UDP_PORT_NONE;
4260       p->responder.sw_if_index = ~0;
4261       p->tun_itf = ~0;
4262       uword index = p - km->profiles;
4263       mhash_set_mem (&km->profile_index_by_name, name, &index, 0);
4264
4265       ikev2_bind (vm, km);
4266     }
4267   else
4268     {
4269       p = ikev2_profile_index_by_name (name);
4270       if (!p)
4271         return clib_error_return (0, "policy %v does not exists", name);
4272
4273       ikev2_unbind (vm, km);
4274
4275       ikev2_unregister_udp_port (p);
4276       ikev2_cleanup_profile_sessions (km, p);
4277
4278       ikev2_profile_free (p);
4279       pool_put (km->profiles, p);
4280       mhash_unset (&km->profile_index_by_name, name, 0);
4281     }
4282   return 0;
4283 }
4284
4285 clib_error_t *
4286 ikev2_set_profile_auth (vlib_main_t * vm, u8 * name, u8 auth_method,
4287                         u8 * auth_data, u8 data_hex_format)
4288 {
4289   ikev2_profile_t *p;
4290   clib_error_t *r;
4291
4292   p = ikev2_profile_index_by_name (name);
4293
4294   if (!p)
4295     {
4296       r = clib_error_return (0, "unknown profile %v", name);
4297       return r;
4298     }
4299
4300   if (p->auth.key)
4301     EVP_PKEY_free (p->auth.key);
4302   vec_free (p->auth.data);
4303
4304   p->auth.method = auth_method;
4305   p->auth.data = vec_dup (auth_data);
4306   p->auth.hex = data_hex_format;
4307
4308   if (auth_method == IKEV2_AUTH_METHOD_RSA_SIG)
4309     {
4310       vec_add1 (p->auth.data, 0);
4311       p->auth.key = ikev2_load_cert_file (p->auth.data);
4312       if (p->auth.key == NULL)
4313         return clib_error_return (0, "load cert '%s' failed", p->auth.data);
4314     }
4315
4316   return 0;
4317 }
4318
4319 static int
4320 ikev2_is_id_supported (u8 id_type)
4321 {
4322   return (id_type == IKEV2_ID_TYPE_ID_IPV4_ADDR ||
4323           id_type == IKEV2_ID_TYPE_ID_IPV6_ADDR ||
4324           id_type == IKEV2_ID_TYPE_ID_RFC822_ADDR ||
4325           id_type == IKEV2_ID_TYPE_ID_FQDN);
4326 }
4327
4328 clib_error_t *
4329 ikev2_set_profile_id (vlib_main_t * vm, u8 * name, u8 id_type, u8 * data,
4330                       int is_local)
4331 {
4332   ikev2_profile_t *p;
4333   clib_error_t *r;
4334
4335   if (!ikev2_is_id_supported (id_type))
4336     {
4337       r = clib_error_return (0, "unsupported identity type %U",
4338                              format_ikev2_id_type, id_type);
4339       return r;
4340     }
4341
4342   p = ikev2_profile_index_by_name (name);
4343
4344   if (!p)
4345     {
4346       r = clib_error_return (0, "unknown profile %v", name);
4347       return r;
4348     }
4349
4350   if (is_local)
4351     {
4352       vec_free (p->loc_id.data);
4353       p->loc_id.type = id_type;
4354       p->loc_id.data = vec_dup (data);
4355     }
4356   else
4357     {
4358       vec_free (p->rem_id.data);
4359       p->rem_id.type = id_type;
4360       p->rem_id.data = vec_dup (data);
4361     }
4362
4363   return 0;
4364 }
4365
4366 static void
4367 ikev2_set_ts_type (ikev2_ts_t *ts, const ip_address_t *addr)
4368 {
4369   if (ip_addr_version (addr) == AF_IP4)
4370     ts->ts_type = TS_IPV4_ADDR_RANGE;
4371   else
4372     ts->ts_type = TS_IPV6_ADDR_RANGE;
4373 }
4374
4375 static void
4376 ikev2_set_ts_addrs (ikev2_ts_t *ts, const ip_address_t *start,
4377                     const ip_address_t *end)
4378 {
4379   ip_address_copy (&ts->start_addr, start);
4380   ip_address_copy (&ts->end_addr, end);
4381 }
4382
4383 clib_error_t *
4384 ikev2_set_profile_ts (vlib_main_t * vm, u8 * name, u8 protocol_id,
4385                       u16 start_port, u16 end_port, ip_address_t start_addr,
4386                       ip_address_t end_addr, int is_local)
4387 {
4388   ikev2_profile_t *p;
4389   clib_error_t *r;
4390
4391   p = ikev2_profile_index_by_name (name);
4392
4393   if (!p)
4394     {
4395       r = clib_error_return (0, "unknown profile %v", name);
4396       return r;
4397     }
4398
4399   if (ip_addr_version (&start_addr) != ip_addr_version (&end_addr))
4400     return clib_error_return (0, "IP address version mismatch!");
4401
4402   if (is_local)
4403     {
4404       ikev2_set_ts_addrs (&p->loc_ts, &start_addr, &end_addr);
4405       p->loc_ts.start_port = start_port;
4406       p->loc_ts.end_port = end_port;
4407       p->loc_ts.protocol_id = protocol_id;
4408       ikev2_set_ts_type (&p->loc_ts, &start_addr);
4409     }
4410   else
4411     {
4412       ikev2_set_ts_addrs (&p->rem_ts, &start_addr, &end_addr);
4413       p->rem_ts.start_port = start_port;
4414       p->rem_ts.end_port = end_port;
4415       p->rem_ts.protocol_id = protocol_id;
4416       ikev2_set_ts_type (&p->rem_ts, &start_addr);
4417     }
4418
4419   return 0;
4420 }
4421
4422 clib_error_t *
4423 ikev2_set_profile_responder_hostname (vlib_main_t *vm, u8 *name, u8 *hostname,
4424                                       u32 sw_if_index)
4425 {
4426   ikev2_profile_t *p;
4427   clib_error_t *r;
4428
4429   p = ikev2_profile_index_by_name (name);
4430
4431   if (!p)
4432     {
4433       r = clib_error_return (0, "unknown profile %v", name);
4434       return r;
4435     }
4436
4437   p->responder.is_resolved = 0;
4438   p->responder.sw_if_index = sw_if_index;
4439   p->responder.hostname = vec_dup (hostname);
4440
4441   return 0;
4442 }
4443
4444 clib_error_t *
4445 ikev2_set_profile_responder (vlib_main_t * vm, u8 * name,
4446                              u32 sw_if_index, ip_address_t addr)
4447 {
4448   ikev2_profile_t *p;
4449   clib_error_t *r;
4450
4451   p = ikev2_profile_index_by_name (name);
4452
4453   if (!p)
4454     {
4455       r = clib_error_return (0, "unknown profile %v", name);
4456       return r;
4457     }
4458
4459   p->responder.is_resolved = 1;
4460   p->responder.sw_if_index = sw_if_index;
4461   ip_address_copy (&p->responder.addr, &addr);
4462
4463   return 0;
4464 }
4465
4466 clib_error_t *
4467 ikev2_set_profile_ike_transforms (vlib_main_t * vm, u8 * name,
4468                                   ikev2_transform_encr_type_t crypto_alg,
4469                                   ikev2_transform_integ_type_t integ_alg,
4470                                   ikev2_transform_dh_type_t dh_type,
4471                                   u32 crypto_key_size)
4472 {
4473   ikev2_profile_t *p;
4474
4475   p = ikev2_profile_index_by_name (name);
4476   if (!p)
4477     return clib_error_return (0, "unknown profile %v", name);
4478
4479   if ((IKEV2_TRANSFORM_INTEG_TYPE_NONE != integ_alg) +
4480         (IKEV2_TRANSFORM_ENCR_TYPE_AES_GCM_16 == crypto_alg) !=
4481       1)
4482     return clib_error_return (0, "invalid cipher + integrity algorithm");
4483
4484   p->ike_ts.crypto_alg = crypto_alg;
4485   p->ike_ts.integ_alg = integ_alg;
4486   p->ike_ts.dh_type = dh_type;
4487   p->ike_ts.crypto_key_size = crypto_key_size;
4488   return 0;
4489 }
4490
4491 clib_error_t *
4492 ikev2_set_profile_esp_transforms (vlib_main_t * vm, u8 * name,
4493                                   ikev2_transform_encr_type_t crypto_alg,
4494                                   ikev2_transform_integ_type_t integ_alg,
4495                                   u32 crypto_key_size)
4496 {
4497   ikev2_profile_t *p;
4498   clib_error_t *r;
4499
4500   p = ikev2_profile_index_by_name (name);
4501
4502   if (!p)
4503     {
4504       r = clib_error_return (0, "unknown profile %v", name);
4505       return r;
4506     }
4507
4508   p->esp_ts.crypto_alg = crypto_alg;
4509   p->esp_ts.integ_alg = integ_alg;
4510   p->esp_ts.crypto_key_size = crypto_key_size;
4511   return 0;
4512 }
4513
4514 clib_error_t *
4515 ikev2_set_profile_tunnel_interface (vlib_main_t * vm,
4516                                     u8 * name, u32 sw_if_index)
4517 {
4518   ikev2_profile_t *p;
4519   clib_error_t *r;
4520
4521   p = ikev2_profile_index_by_name (name);
4522
4523   if (!p)
4524     {
4525       r = clib_error_return (0, "unknown profile %v", name);
4526       return r;
4527     }
4528
4529   p->tun_itf = sw_if_index;
4530
4531   return 0;
4532 }
4533
4534 vnet_api_error_t
4535 ikev2_set_profile_ipsec_udp_port (vlib_main_t * vm, u8 * name, u16 port,
4536                                   u8 is_set)
4537 {
4538   ikev2_profile_t *p = ikev2_profile_index_by_name (name);
4539   vnet_api_error_t rv = 0;
4540
4541   if (!p)
4542     return VNET_API_ERROR_INVALID_VALUE;
4543
4544   if (is_set)
4545     {
4546       if (p->ipsec_over_udp_port != IPSEC_UDP_PORT_NONE)
4547         return VNET_API_ERROR_VALUE_EXIST;
4548
4549       rv = ikev2_register_udp_port (p, port);
4550     }
4551   else
4552     {
4553       if (p->ipsec_over_udp_port == IPSEC_UDP_PORT_NONE)
4554         return VNET_API_ERROR_INVALID_VALUE;
4555
4556       ikev2_unregister_udp_port (p);
4557     }
4558   return rv;
4559 }
4560
4561 clib_error_t *
4562 ikev2_set_profile_udp_encap (vlib_main_t * vm, u8 * name)
4563 {
4564   ikev2_profile_t *p = ikev2_profile_index_by_name (name);
4565   clib_error_t *r;
4566
4567   if (!p)
4568     {
4569       r = clib_error_return (0, "unknown profile %v", name);
4570       return r;
4571     }
4572
4573   p->udp_encap = 1;
4574   return 0;
4575 }
4576
4577 clib_error_t *
4578 ikev2_set_profile_sa_lifetime (vlib_main_t * vm, u8 * name,
4579                                u64 lifetime, u32 jitter, u32 handover,
4580                                u64 maxdata)
4581 {
4582   ikev2_profile_t *p;
4583   clib_error_t *r;
4584
4585   p = ikev2_profile_index_by_name (name);
4586
4587   if (!p)
4588     {
4589       r = clib_error_return (0, "unknown profile %v", name);
4590       return r;
4591     }
4592
4593   p->lifetime = lifetime;
4594   p->lifetime_jitter = jitter;
4595   p->handover = handover;
4596   p->lifetime_maxdata = maxdata;
4597   return 0;
4598 }
4599
4600 static int
4601 ikev2_get_if_address (u32 sw_if_index, ip_address_family_t af,
4602                       ip_address_t * out_addr)
4603 {
4604   ip4_address_t *if_ip4;
4605   ip6_address_t *if_ip6;
4606
4607   if (af == AF_IP4)
4608     {
4609       if_ip4 = ip4_interface_first_address (&ip4_main, sw_if_index, 0);
4610       if (if_ip4)
4611         {
4612           ip_address_set (out_addr, if_ip4, AF_IP4);
4613           return 1;
4614         }
4615     }
4616   else
4617     {
4618       if_ip6 = ip6_interface_first_address (&ip6_main, sw_if_index);
4619       if (if_ip6)
4620         {
4621           ip_address_set (out_addr, if_ip6, AF_IP6);
4622           return 1;
4623         }
4624     }
4625   return 0;
4626 }
4627
4628 static clib_error_t *
4629 ikev2_resolve_responder_hostname (vlib_main_t *vm, ikev2_responder_t *r)
4630 {
4631   ikev2_main_t *km = &ikev2_main;
4632   dns_cache_entry_t *ep = 0;
4633   dns_pending_request_t _t0, *t0 = &_t0;
4634   dns_resolve_name_t _rn, *rn = &_rn;
4635   u8 *name;
4636   int rv;
4637
4638   if (!km->dns_resolve_name_ptr)
4639     return clib_error_return (0, "cannot load symbols from dns plugin");
4640
4641   t0->request_type = DNS_API_PENDING_NAME_TO_IP;
4642   /* VPP main curse: IKEv2 uses only non-NULL terminated vectors internally
4643    * whereas DNS resolver expects a NULL-terminated C-string */
4644   name = vec_dup (r->hostname);
4645   vec_terminate_c_string (name);
4646   rv = ((__typeof__ (dns_resolve_name) *) km->dns_resolve_name_ptr) (name, &ep,
4647                                                                      t0, rn);
4648   vec_free (name);
4649   if (rv < 0)
4650     return clib_error_return (0, "dns lookup failure");
4651
4652   if (ep == 0)
4653     return 0;
4654
4655   ip_address_copy (&r->addr, &rn->address);
4656   r->is_resolved = 1;
4657   return 0;
4658 }
4659
4660 clib_error_t *
4661 ikev2_initiate_sa_init (vlib_main_t * vm, u8 * name)
4662 {
4663   ikev2_profile_t *p;
4664   clib_error_t *r;
4665   ikev2_main_t *km = &ikev2_main;
4666   vlib_buffer_t *b0;
4667   ike_header_t *ike0;
4668   u32 bi0 = 0;
4669   int len = sizeof (ike_header_t), valid_ip = 0;
4670   ip_address_t src_if_ip = ip_address_initializer;
4671
4672   p = ikev2_profile_index_by_name (name);
4673
4674   if (!p)
4675     {
4676       r = clib_error_return (0, "unknown profile %v", name);
4677       return r;
4678     }
4679
4680   if (p->responder.sw_if_index == ~0 ||
4681       (ip_address_is_zero (&p->responder.addr) &&
4682        vec_len (p->responder.hostname) == 0))
4683     {
4684       r = clib_error_return (0, "responder not set for profile %v", name);
4685       return r;
4686     }
4687
4688   if (!p->responder.is_resolved)
4689     {
4690       /* try to resolve using dns plugin
4691        * success does not mean we have resolved the name */
4692       r = ikev2_resolve_responder_hostname (vm, &p->responder);
4693       if (r)
4694         return r;
4695     }
4696
4697   if (p->responder.is_resolved &&
4698       ikev2_get_if_address (p->responder.sw_if_index,
4699                             ip_addr_version (&p->responder.addr), &src_if_ip))
4700     {
4701       valid_ip = 1;
4702     }
4703
4704   /* Prepare the SA and the IKE payload */
4705   ikev2_sa_t sa;
4706   clib_memset (&sa, 0, sizeof (ikev2_sa_t));
4707   ikev2_payload_chain_t *chain = 0;
4708   ikev2_payload_new_chain (chain);
4709
4710   /* Build the IKE proposal payload */
4711   ikev2_sa_proposal_t *proposals = 0;
4712   ikev2_set_initiator_proposals (vm, &sa, &p->ike_ts, &proposals, 1);
4713   proposals[0].proposal_num = 1;
4714   proposals[0].protocol_id = IKEV2_PROTOCOL_IKE;
4715
4716   /* Add and then cleanup proposal data */
4717   ikev2_payload_add_sa (chain, proposals, 0);
4718   ikev2_sa_free_proposal_vector (&proposals);
4719
4720   sa.is_initiator = 1;
4721   sa.profile_index = p - km->profiles;
4722   sa.state = IKEV2_STATE_SA_INIT;
4723   sa.tun_itf = p->tun_itf;
4724   sa.udp_encap = p->udp_encap;
4725   if (p->natt_disabled)
4726     sa.natt_state = IKEV2_NATT_DISABLED;
4727   sa.ipsec_over_udp_port = p->ipsec_over_udp_port;
4728   sa.is_tun_itf_set = 1;
4729   sa.initial_contact = 1;
4730   sa.dst_port = IKEV2_PORT;
4731
4732   ikev2_generate_sa_error_t rc = ikev2_generate_sa_init_data (&sa);
4733   if (rc != IKEV2_GENERATE_SA_INIT_OK)
4734     {
4735       ikev2_sa_free_all_vec (&sa);
4736       ikev2_payload_destroy_chain (chain);
4737       return clib_error_return (0, "%U", format_ikev2_gen_sa_error, rc);
4738     }
4739
4740   ikev2_payload_add_ke (chain, sa.dh_group, sa.i_dh_data);
4741   ikev2_payload_add_nonce (chain, sa.i_nonce);
4742
4743   /* Build the child SA proposal */
4744   vec_resize (sa.childs, 1);
4745   ikev2_set_initiator_proposals (vm, &sa, &p->esp_ts,
4746                                  &sa.childs[0].i_proposals, 0);
4747   sa.childs[0].i_proposals[0].proposal_num = 1;
4748   sa.childs[0].i_proposals[0].protocol_id = IKEV2_PROTOCOL_ESP;
4749   RAND_bytes ((u8 *) & sa.childs[0].i_proposals[0].spi,
4750               sizeof (sa.childs[0].i_proposals[0].spi));
4751   sa.childs[0].i_proposals[0].spi &= 0xffffffff;
4752
4753   /* Add NAT detection notification messages (mandatory) */
4754   u8 *nat_detection_sha1 = ikev2_compute_nat_sha1 (
4755     clib_host_to_net_u64 (sa.ispi), clib_host_to_net_u64 (sa.rspi), &src_if_ip,
4756     clib_host_to_net_u16 (IKEV2_PORT));
4757
4758   ikev2_payload_add_notify (chain, IKEV2_NOTIFY_MSG_NAT_DETECTION_SOURCE_IP,
4759                             nat_detection_sha1);
4760   vec_free (nat_detection_sha1);
4761   nat_detection_sha1 =
4762     ikev2_compute_nat_sha1 (clib_host_to_net_u64 (sa.ispi),
4763                             clib_host_to_net_u64 (sa.rspi),
4764                             &p->responder.addr,
4765                             clib_host_to_net_u16 (sa.dst_port));
4766   ikev2_payload_add_notify (chain,
4767                             IKEV2_NOTIFY_MSG_NAT_DETECTION_DESTINATION_IP,
4768                             nat_detection_sha1);
4769   vec_free (nat_detection_sha1);
4770
4771   u8 *sig_hash_algo = vec_new (u8, 8);
4772   u64 tmpsig = clib_host_to_net_u64 (0x0001000200030004);
4773   clib_memcpy_fast (sig_hash_algo, &tmpsig, sizeof (tmpsig));
4774   ikev2_payload_add_notify (chain,
4775                             IKEV2_NOTIFY_MSG_SIGNATURE_HASH_ALGORITHMS,
4776                             sig_hash_algo);
4777   vec_free (sig_hash_algo);
4778
4779   bi0 = ikev2_get_new_ike_header_buff (vm, &b0);
4780   if (!bi0)
4781     {
4782       ikev2_sa_free_all_vec (&sa);
4783       ikev2_payload_destroy_chain (chain);
4784       char *errmsg = "buffer alloc failure";
4785       ikev2_log_error (errmsg);
4786       return clib_error_return (0, errmsg);
4787     }
4788   ike0 = vlib_buffer_get_current (b0);
4789
4790   /* Buffer update and boilerplate */
4791   len += vec_len (chain->data);
4792   ike0->nextpayload = chain->first_payload_type;
4793   ike0->length = clib_host_to_net_u32 (len);
4794   clib_memcpy_fast (ike0->payload, chain->data, vec_len (chain->data));
4795   ikev2_payload_destroy_chain (chain);
4796
4797   ike0->version = IKE_VERSION_2;
4798   ike0->flags = IKEV2_HDR_FLAG_INITIATOR;
4799   ike0->exchange = IKEV2_EXCHANGE_SA_INIT;
4800   ike0->ispi = clib_host_to_net_u64 (sa.ispi);
4801   ike0->rspi = 0;
4802   ike0->msgid = 0;
4803   sa.last_init_msg_id += 1;
4804
4805   /* store whole IKE payload - needed for PSK auth */
4806   vec_reset_length (sa.last_sa_init_req_packet_data);
4807   vec_add (sa.last_sa_init_req_packet_data, ike0, len);
4808
4809   /* add data to the SA then add it to the pool */
4810   ip_address_copy (&sa.iaddr, &src_if_ip);
4811   ip_address_copy (&sa.raddr, &p->responder.addr);
4812   sa.i_id.type = p->loc_id.type;
4813   sa.i_id.data = vec_dup (p->loc_id.data);
4814   sa.r_id.type = p->rem_id.type;
4815   sa.r_id.data = vec_dup (p->rem_id.data);
4816   sa.i_auth.method = p->auth.method;
4817   sa.i_auth.hex = p->auth.hex;
4818   sa.i_auth.data = vec_dup (p->auth.data);
4819   sa.sw_if_index = p->responder.sw_if_index;
4820   vec_add (sa.childs[0].tsi, &p->loc_ts, 1);
4821   vec_add (sa.childs[0].tsr, &p->rem_ts, 1);
4822
4823   ikev2_initial_contact_cleanup (0, &sa);
4824
4825   /* add SA to the pool */
4826   ikev2_sa_t *sa0 = 0;
4827   pool_get (km->sais, sa0);
4828   clib_memcpy_fast (sa0, &sa, sizeof (*sa0));
4829   hash_set (km->sa_by_ispi, sa0->ispi, sa0 - km->sais);
4830
4831   if (valid_ip)
4832     {
4833       ikev2_send_ike (vm, &src_if_ip, &p->responder.addr, bi0, len, IKEV2_PORT,
4834                       sa.dst_port, sa.sw_if_index);
4835
4836       ikev2_elog_exchange
4837         ("ispi %lx rspi %lx IKEV2_EXCHANGE_SA_INIT sent to ",
4838          clib_host_to_net_u64 (sa0->ispi), 0,
4839          ip_addr_v4 (&p->responder.addr).as_u32,
4840          ip_addr_version (&p->responder.addr) == AF_IP4);
4841     }
4842
4843   return 0;
4844 }
4845
4846 static void
4847 ikev2_delete_child_sa_internal (vlib_main_t * vm, ikev2_sa_t * sa,
4848                                 ikev2_child_sa_t * csa)
4849 {
4850   /* Create the Initiator notification for child SA removal */
4851   ikev2_main_t *km = &ikev2_main;
4852   ike_header_t *ike0;
4853   u32 bi0 = 0;
4854   vlib_buffer_t *b0;
4855   int len;
4856
4857   bi0 = ikev2_get_new_ike_header_buff (vm, &b0);
4858   if (!bi0)
4859     {
4860       ikev2_log_error ("buffer alloc failure");
4861       return;
4862     }
4863
4864   ike0 = vlib_buffer_get_current (b0);
4865   ike0->exchange = IKEV2_EXCHANGE_INFORMATIONAL;
4866   ike0->ispi = clib_host_to_net_u64 (sa->ispi);
4867   ike0->rspi = clib_host_to_net_u64 (sa->rspi);
4868   ike0->flags = 0;
4869   vec_resize (sa->del, 1);
4870   sa->del->protocol_id = IKEV2_PROTOCOL_ESP;
4871   sa->del->spi = csa->i_proposals->spi;
4872   ike0->msgid = clib_host_to_net_u32 (sa->last_init_msg_id);
4873   sa->last_init_msg_id += 1;
4874   len = ikev2_generate_message (b0, sa, ike0, 0, 0, 0);
4875   if (~0 == len)
4876     return;
4877
4878   if (ikev2_natt_active (sa))
4879     len = ikev2_insert_non_esp_marker (ike0, len);
4880   ikev2_send_ike (vm, &sa->iaddr, &sa->raddr, bi0, len,
4881                   ikev2_get_port (sa), sa->dst_port, sa->sw_if_index);
4882
4883   /* delete local child SA */
4884   ikev2_delete_tunnel_interface (km->vnet_main, sa, csa);
4885   ikev2_sa_del_child_sa (sa, csa);
4886 }
4887
4888 clib_error_t *
4889 ikev2_initiate_delete_child_sa (vlib_main_t * vm, u32 ispi)
4890 {
4891   clib_error_t *r;
4892   ikev2_main_t *km = &ikev2_main;
4893   ikev2_main_per_thread_data_t *tkm;
4894   ikev2_sa_t *fsa = 0;
4895   ikev2_child_sa_t *fchild = 0;
4896
4897   /* Search for the child SA */
4898   vec_foreach (tkm, km->per_thread_data)
4899   {
4900     ikev2_sa_t *sa;
4901     if (fchild)
4902       break;
4903     pool_foreach (sa, tkm->sas)  {
4904       fchild = ikev2_sa_get_child(sa, ispi, IKEV2_PROTOCOL_ESP, 1);
4905       if (fchild)
4906         {
4907           fsa = sa;
4908           break;
4909         }
4910     }
4911   }
4912
4913   if (!fchild || !fsa)
4914     {
4915       r = clib_error_return (0, "Child SA not found");
4916       return r;
4917     }
4918   else
4919     {
4920       ikev2_delete_child_sa_internal (vm, fsa, fchild);
4921     }
4922
4923   return 0;
4924 }
4925
4926 clib_error_t *
4927 ikev2_initiate_delete_ike_sa (vlib_main_t * vm, u64 ispi)
4928 {
4929   clib_error_t *r;
4930   ikev2_main_t *km = &ikev2_main;
4931   ikev2_main_per_thread_data_t *tkm;
4932   ikev2_sa_t *fsa = 0;
4933   ikev2_main_per_thread_data_t *ftkm = 0;
4934
4935   /* Search for the IKE SA */
4936   vec_foreach (tkm, km->per_thread_data)
4937   {
4938     ikev2_sa_t *sa;
4939     if (fsa)
4940       break;
4941     pool_foreach (sa, tkm->sas)  {
4942       if (sa->ispi == ispi)
4943         {
4944           fsa = sa;
4945           ftkm = tkm;
4946           break;
4947         }
4948     }
4949   }
4950
4951   if (!fsa)
4952     {
4953       r = clib_error_return (0, "IKE SA not found");
4954       return r;
4955     }
4956
4957   ikev2_initiate_delete_ike_sa_internal (vm, ftkm, fsa, 1);
4958   return 0;
4959 }
4960
4961 static void
4962 ikev2_rekey_child_sa_internal (vlib_main_t * vm, ikev2_sa_t * sa,
4963                                ikev2_child_sa_t * csa)
4964 {
4965   /* Create the Initiator request for create child SA */
4966   ike_header_t *ike0;
4967   vlib_buffer_t *b0;
4968   u32 bi0 = 0;
4969   int len;
4970
4971   bi0 = ikev2_get_new_ike_header_buff (vm, &b0);
4972   if (!bi0)
4973     {
4974       ikev2_log_error ("buffer alloc failure");
4975       return;
4976     }
4977
4978   ike0 = vlib_buffer_get_current (b0);
4979   ike0->version = IKE_VERSION_2;
4980   ike0->flags = IKEV2_HDR_FLAG_INITIATOR;
4981   ike0->exchange = IKEV2_EXCHANGE_CREATE_CHILD_SA;
4982   ike0->ispi = clib_host_to_net_u64 (sa->ispi);
4983   ike0->rspi = clib_host_to_net_u64 (sa->rspi);
4984   ike0->msgid = clib_host_to_net_u32 (sa->last_init_msg_id);
4985   sa->last_init_msg_id += 1;
4986
4987   ikev2_rekey_t *rekey;
4988   vec_reset_length (sa->rekey);
4989   vec_add2 (sa->rekey, rekey, 1);
4990   rekey->kex = 0;
4991   ikev2_sa_proposal_t *proposals = vec_dup (csa->i_proposals);
4992
4993   /*need new ispi */
4994   RAND_bytes ((u8 *) & proposals[0].spi, sizeof (proposals[0].spi));
4995   proposals[0].spi &= 0xffffffff;
4996   rekey->spi = proposals[0].spi;
4997   rekey->ispi = csa->i_proposals->spi;
4998   len = ikev2_generate_message (b0, sa, ike0, proposals, 0, 0);
4999   if (~0 == len)
5000     return;
5001
5002   if (ikev2_natt_active (sa))
5003     len = ikev2_insert_non_esp_marker (ike0, len);
5004   ikev2_send_ike (vm, &sa->iaddr, &sa->raddr, bi0, len,
5005                   ikev2_get_port (sa), ikev2_get_port (sa), sa->sw_if_index);
5006   vec_free (proposals);
5007 }
5008
5009 clib_error_t *
5010 ikev2_initiate_rekey_child_sa (vlib_main_t * vm, u32 ispi)
5011 {
5012   clib_error_t *r;
5013   ikev2_main_t *km = &ikev2_main;
5014   ikev2_main_per_thread_data_t *tkm;
5015   ikev2_sa_t *fsa = 0;
5016   ikev2_child_sa_t *fchild = 0;
5017
5018   /* Search for the child SA */
5019   vec_foreach (tkm, km->per_thread_data)
5020   {
5021     ikev2_sa_t *sa;
5022     if (fchild)
5023       break;
5024     pool_foreach (sa, tkm->sas)  {
5025       fchild = ikev2_sa_get_child(sa, ispi, IKEV2_PROTOCOL_ESP, 1);
5026       if (fchild)
5027         {
5028           fsa = sa;
5029           break;
5030         }
5031     }
5032   }
5033
5034   if (!fchild || !fsa)
5035     {
5036       r = clib_error_return (0, "Child SA not found");
5037       return r;
5038     }
5039   else
5040     {
5041       ikev2_rekey_child_sa_internal (vm, fsa, fchild);
5042     }
5043
5044   return 0;
5045 }
5046
5047 static int
5048 ikev2_sa_sw_if_match (ikev2_sa_t * sa, u32 sw_if_index)
5049 {
5050   return (sa->sw_if_index == sw_if_index) && sa->is_initiator;
5051 }
5052
5053 static void
5054 ikev2_sa_del (ikev2_profile_t * p, u32 sw_if_index)
5055 {
5056   u64 *ispi, *ispi_vec = 0;
5057   ikev2_sa_t *sa, **sap, **sa_vec = 0;
5058   ikev2_main_t *km = &ikev2_main;
5059   ikev2_main_per_thread_data_t *tkm;
5060   p->responder.sw_if_index = ~0;
5061
5062   vec_foreach (tkm, km->per_thread_data)
5063   {
5064     pool_foreach (sa, tkm->sas)  {
5065       if (ikev2_sa_sw_if_match (sa, sw_if_index))
5066         vec_add1 (sa_vec, sa);
5067     }
5068
5069     vec_foreach (sap, sa_vec)
5070     {
5071       ikev2_initiate_delete_ike_sa_internal (km->vlib_main, tkm, *sap, 0);
5072     }
5073     vec_reset_length (sa_vec);
5074   }
5075   vec_free (sa_vec);
5076
5077   pool_foreach (sa, km->sais)  {
5078     if (ikev2_sa_sw_if_match (sa, sw_if_index))
5079       vec_add1 (ispi_vec, sa->ispi);
5080   }
5081
5082   vec_foreach (ispi, ispi_vec)
5083   {
5084     ikev2_del_sa_init_from_main (ispi);
5085   }
5086
5087   vec_free (ispi_vec);
5088 }
5089
5090 static clib_error_t *
5091 ikev2_sw_interface_add_del (vnet_main_t * vnm, u32 sw_if_index, u32 is_add)
5092 {
5093   ikev2_main_t *km = &ikev2_main;
5094   ikev2_profile_t *p;
5095
5096   if (is_add)
5097     return 0;
5098
5099   pool_foreach (p, km->profiles)  {
5100     if (p->responder.sw_if_index == sw_if_index)
5101       ikev2_sa_del (p, sw_if_index);
5102   }
5103
5104   return 0;
5105 }
5106
5107 VNET_SW_INTERFACE_ADD_DEL_FUNCTION (ikev2_sw_interface_add_del);
5108
5109 clib_error_t *
5110 ikev2_init (vlib_main_t * vm)
5111 {
5112   ikev2_main_t *km = &ikev2_main;
5113
5114   clib_memset (km, 0, sizeof (ikev2_main_t));
5115
5116   km->log_level = IKEV2_LOG_ERROR;
5117   km->log_class = vlib_log_register_class ("ikev2", 0);
5118
5119   km->vnet_main = vnet_get_main ();
5120   km->vlib_main = vm;
5121
5122   km->liveness_period = IKEV2_LIVENESS_PERIOD_CHECK;
5123   km->liveness_max_retries = IKEV2_LIVENESS_RETRIES;
5124
5125   return 0;
5126 }
5127
5128 VLIB_INIT_FUNCTION (ikev2_init) = {
5129   .runs_after = VLIB_INITS ("ipsec_init", "ipsec_punt_init"),
5130 };
5131
5132 static u8
5133 ikev2_mngr_process_child_sa (ikev2_sa_t * sa, ikev2_child_sa_t * csa,
5134                              u8 del_old_ids)
5135 {
5136   ikev2_main_t *km = &ikev2_main;
5137   ikev2_profile_t *p = 0;
5138   vlib_main_t *vm = km->vlib_main;
5139   f64 now = vlib_time_now (vm);
5140   u8 res = 0;
5141
5142   if (sa->profile_index != ~0)
5143     p = pool_elt_at_index (km->profiles, sa->profile_index);
5144
5145   if (sa->is_initiator && p && csa->time_to_expiration
5146       && now > csa->time_to_expiration)
5147     {
5148       if (!csa->is_expired || csa->rekey_retries > 0)
5149         {
5150           ikev2_rekey_child_sa_internal (vm, sa, csa);
5151           csa->time_to_expiration = now + p->handover;
5152           csa->is_expired = 1;
5153           if (csa->rekey_retries == 0)
5154             {
5155               csa->rekey_retries = 5;
5156             }
5157           else if (csa->rekey_retries > 0)
5158             {
5159               csa->rekey_retries--;
5160               ikev2_log_debug ("Rekeying Child SA 0x%x, retries left %d",
5161                                csa->i_proposals->spi, csa->rekey_retries);
5162               if (csa->rekey_retries == 0)
5163                 {
5164                   csa->rekey_retries = -1;
5165                 }
5166             }
5167           res |= 1;
5168         }
5169       else
5170         {
5171           csa->time_to_expiration = 0;
5172           ikev2_delete_child_sa_internal (vm, sa, csa);
5173           res |= 1;
5174           return res;
5175         }
5176     }
5177
5178   if (del_old_ids)
5179     {
5180       ipip_tunnel_t *ipip = NULL;
5181       u32 sw_if_index = sa->is_tun_itf_set ? sa->tun_itf : ~0;
5182       if (~0 == sw_if_index)
5183         {
5184           ip46_address_t local_ip;
5185           ip46_address_t remote_ip;
5186           if (sa->is_initiator)
5187             {
5188               local_ip = to_ip46 (ip_addr_version (&sa->iaddr),
5189                                   ip_addr_bytes (&sa->iaddr));
5190               remote_ip = to_ip46 (ip_addr_version (&sa->raddr),
5191                                    ip_addr_bytes (&sa->raddr));
5192             }
5193           else
5194             {
5195               local_ip = to_ip46 (ip_addr_version (&sa->raddr),
5196                                   ip_addr_bytes (&sa->raddr));
5197               remote_ip = to_ip46 (ip_addr_version (&sa->iaddr),
5198                                    ip_addr_bytes (&sa->iaddr));
5199             }
5200
5201        ipip_tunnel_key_t key = {
5202          .src = local_ip,
5203          .dst = remote_ip,
5204          .transport = IPIP_TRANSPORT_IP4,
5205          .fib_index = 0,
5206        };
5207
5208           ipip = ipip_tunnel_db_find (&key);
5209
5210           if (ipip)
5211             sw_if_index = ipip->sw_if_index;
5212           else
5213             return res;
5214         }
5215
5216       u32 *sas_in = NULL;
5217       vec_add1 (sas_in, csa->remote_sa_id);
5218       vlib_worker_thread_barrier_sync (vm);
5219       int rv = ipsec_tun_protect_update (sw_if_index, NULL,
5220                                          csa->local_sa_id, sas_in);
5221       if (rv)
5222         vec_free (sas_in);
5223       ipsec_sa_unlock_id (ikev2_flip_alternate_sa_bit (csa->remote_sa_id));
5224       vlib_worker_thread_barrier_release (vm);
5225     }
5226
5227   return res;
5228 }
5229
5230 int
5231 ikev2_set_log_level (ikev2_log_level_t log_level)
5232 {
5233   ikev2_main_t *km = &ikev2_main;
5234
5235   if (log_level >= IKEV2_LOG_MAX)
5236     {
5237       ikev2_log_error ("unknown logging level %d", log_level);
5238       return -1;
5239     }
5240
5241   km->log_level = log_level;
5242   return 0;
5243 }
5244
5245 clib_error_t *
5246 ikev2_set_liveness_params (u32 period, u32 max_retries)
5247 {
5248   ikev2_main_t *km = &ikev2_main;
5249
5250   if (period == 0 || max_retries == 0)
5251     return clib_error_return (0, "invalid args");
5252
5253   km->liveness_period = period;
5254   km->liveness_max_retries = max_retries;
5255   return 0;
5256 }
5257
5258 clib_error_t *
5259 ikev2_profile_natt_disable (u8 * name)
5260 {
5261   ikev2_profile_t *p = ikev2_profile_index_by_name (name);
5262   if (!p)
5263     return clib_error_return (0, "unknown profile %v", name);
5264
5265   p->natt_disabled = 1;
5266   return 0;
5267 }
5268
5269 static void
5270 ikev2_mngr_process_ipsec_sa (ipsec_sa_t * ipsec_sa)
5271 {
5272   ikev2_main_t *km = &ikev2_main;
5273   vlib_main_t *vm = km->vlib_main;
5274   ikev2_main_per_thread_data_t *tkm;
5275   ikev2_sa_t *fsa = 0;
5276   ikev2_profile_t *p = 0;
5277   ikev2_child_sa_t *fchild = 0;
5278   f64 now = vlib_time_now (vm);
5279   vlib_counter_t counts;
5280
5281   /* Search for the SA and child SA */
5282   vec_foreach (tkm, km->per_thread_data)
5283   {
5284     ikev2_sa_t *sa;
5285     if (fchild)
5286       break;
5287     pool_foreach (sa, tkm->sas)  {
5288       fchild = ikev2_sa_get_child(sa, ipsec_sa->spi, IKEV2_PROTOCOL_ESP, 1);
5289       if (fchild)
5290         {
5291           fsa = sa;
5292           break;
5293         }
5294     }
5295   }
5296   vlib_get_combined_counter (&ipsec_sa_counters,
5297                              ipsec_sa->stat_index, &counts);
5298
5299   if (fsa && fsa->profile_index != ~0 && fsa->is_initiator)
5300     p = pool_elt_at_index (km->profiles, fsa->profile_index);
5301
5302   if (fchild && p && p->lifetime_maxdata)
5303     {
5304       if (!fchild->is_expired && counts.bytes > p->lifetime_maxdata)
5305         {
5306           fchild->time_to_expiration = now;
5307         }
5308     }
5309 }
5310
5311 static void
5312 ikev2_process_pending_sa_init_one (vlib_main_t *vm, ikev2_main_t *km,
5313                                    ikev2_sa_t *sa)
5314 {
5315   ikev2_profile_t *p;
5316   u32 bi0;
5317   u8 *nat_sha, *np;
5318   p = pool_elt_at_index (km->profiles, sa->profile_index);
5319
5320   if (!p->responder.is_resolved)
5321     {
5322       clib_error_t *r = ikev2_resolve_responder_hostname (vm, &p->responder);
5323       if (r)
5324         {
5325           clib_error_free (r);
5326           return;
5327         }
5328
5329       if (!p->responder.is_resolved)
5330         return;
5331
5332       ip_address_copy (&sa->raddr, &p->responder.addr);
5333
5334       /* update nat detection destination hash */
5335       np = ikev2_find_ike_notify_payload (
5336         (ike_header_t *) sa->last_sa_init_req_packet_data,
5337         IKEV2_NOTIFY_MSG_NAT_DETECTION_DESTINATION_IP);
5338       if (np)
5339         {
5340           nat_sha = ikev2_compute_nat_sha1 (
5341             clib_host_to_net_u64 (sa->ispi), clib_host_to_net_u64 (sa->rspi),
5342             &sa->raddr, clib_host_to_net_u16 (sa->dst_port));
5343           clib_memcpy_fast (np, nat_sha, vec_len (nat_sha));
5344           vec_free (nat_sha);
5345         }
5346     }
5347
5348   if (ip_address_is_zero (&sa->iaddr))
5349     {
5350       if (!ikev2_get_if_address (p->responder.sw_if_index,
5351                                  ip_addr_version (&p->responder.addr),
5352                                  &sa->iaddr))
5353         return;
5354
5355       /* update NAT detection payload */
5356       np =
5357         ikev2_find_ike_notify_payload
5358         ((ike_header_t *) sa->last_sa_init_req_packet_data,
5359          IKEV2_NOTIFY_MSG_NAT_DETECTION_SOURCE_IP);
5360       if (np)
5361         {
5362           nat_sha =
5363             ikev2_compute_nat_sha1 (clib_host_to_net_u64 (sa->ispi),
5364                                     clib_host_to_net_u64 (sa->rspi),
5365                                     &sa->iaddr,
5366                                     clib_host_to_net_u16 (IKEV2_PORT));
5367           clib_memcpy_fast (np, nat_sha, vec_len (nat_sha));
5368           vec_free (nat_sha);
5369         }
5370     }
5371
5372   if (vlib_buffer_alloc (km->vlib_main, &bi0, 1) != 1)
5373     return;
5374
5375   vlib_buffer_t *b = vlib_get_buffer (km->vlib_main, bi0);
5376   clib_memcpy_fast (vlib_buffer_get_current (b),
5377                     sa->last_sa_init_req_packet_data,
5378                     vec_len (sa->last_sa_init_req_packet_data));
5379
5380   ikev2_send_ike (km->vlib_main, &sa->iaddr, &sa->raddr, bi0,
5381                   vec_len (sa->last_sa_init_req_packet_data),
5382                   ikev2_get_port (sa), IKEV2_PORT, sa->sw_if_index);
5383 }
5384
5385 static void
5386 ikev2_process_pending_sa_init (vlib_main_t *vm, ikev2_main_t *km)
5387 {
5388   u32 sai;
5389   u64 ispi;
5390   ikev2_sa_t *sa;
5391
5392   hash_foreach (ispi, sai, km->sa_by_ispi,
5393   ({
5394     sa = pool_elt_at_index (km->sais, sai);
5395     if (sa->init_response_received)
5396       continue;
5397
5398     ikev2_process_pending_sa_init_one (vm, km, sa);
5399   }));
5400 }
5401
5402 static void
5403 ikev2_send_informational_request (ikev2_sa_t * sa)
5404 {
5405   ikev2_main_t *km = &ikev2_main;
5406   ip_address_t *src, *dst;
5407   ike_header_t *ike0;
5408   vlib_buffer_t *b0;
5409   u32 bi0 = 0;
5410   u16 dp;
5411   int len;
5412
5413   bi0 = ikev2_get_new_ike_header_buff (km->vlib_main, &b0);
5414   if (!bi0)
5415     {
5416       ikev2_log_error ("buffer alloc failure");
5417       return;
5418     }
5419
5420   ike0 = vlib_buffer_get_current (b0);
5421   ike0->exchange = IKEV2_EXCHANGE_INFORMATIONAL;
5422   ike0->ispi = clib_host_to_net_u64 (sa->ispi);
5423   ike0->rspi = clib_host_to_net_u64 (sa->rspi);
5424   ike0->msgid = clib_host_to_net_u32 (sa->last_init_msg_id);
5425   ike0->flags = 0;
5426   sa->last_init_msg_id += 1;
5427   len = ikev2_generate_message (b0, sa, ike0, 0, 0, 0);
5428   if (~0 == len)
5429     return;
5430
5431   if (ikev2_natt_active (sa))
5432     len = ikev2_insert_non_esp_marker (ike0, len);
5433
5434   if (sa->is_initiator)
5435     {
5436       src = &sa->iaddr;
5437       dst = &sa->raddr;
5438     }
5439   else
5440     {
5441       dst = &sa->iaddr;
5442       src = &sa->raddr;
5443     }
5444
5445   dp = sa->dst_port ? sa->dst_port : ikev2_get_port (sa);
5446   ikev2_send_ike (km->vlib_main, src, dst, bi0, len, ikev2_get_port (sa), dp,
5447                   sa->sw_if_index);
5448 }
5449
5450 void
5451 ikev2_disable_dpd (void)
5452 {
5453   ikev2_main_t *km = &ikev2_main;
5454   km->dpd_disabled = 1;
5455 }
5456
5457 static int
5458 ikev2_mngr_process_responder_sas (ikev2_sa_t *sa)
5459 {
5460   ikev2_main_t *km = &ikev2_main;
5461   vlib_main_t *vm = km->vlib_main;
5462
5463   if (!sa->keys_generated)
5464     return 0;
5465
5466   if (sa->liveness_retries >= km->liveness_max_retries)
5467     return 1;
5468
5469   f64 now = vlib_time_now (vm);
5470
5471   if (sa->liveness_period_check < now)
5472     {
5473       sa->liveness_retries++;
5474       sa->liveness_period_check = now + km->liveness_period;
5475       ikev2_send_informational_request (sa);
5476     }
5477   return 0;
5478 }
5479
5480 static uword
5481 ikev2_mngr_process_fn (vlib_main_t * vm, vlib_node_runtime_t * rt,
5482                        vlib_frame_t * f)
5483 {
5484   ikev2_main_t *km = &ikev2_main;
5485   ikev2_profile_t *p;
5486   ikev2_child_sa_t *c;
5487   u32 *sai;
5488
5489   /* lazy init will wake it up */
5490   vlib_process_wait_for_event (vm);
5491
5492   while (1)
5493     {
5494       vlib_process_wait_for_event_or_clock (vm, 2);
5495       vlib_process_get_events (vm, NULL);
5496
5497       /* process ike child sas */
5498       ikev2_main_per_thread_data_t *tkm;
5499       vec_foreach (tkm, km->per_thread_data)
5500       {
5501         ikev2_sa_t *sa;
5502         u32 *to_be_deleted = 0;
5503
5504         pool_foreach (sa, tkm->sas)  {
5505           ikev2_child_sa_t *c;
5506           u8 del_old_ids = 0;
5507
5508           if (sa->state == IKEV2_STATE_SA_INIT)
5509             {
5510               if (vec_len (sa->childs) > 0)
5511                 vec_add1 (to_be_deleted, sa - tkm->sas);
5512             }
5513           else if (sa->state != IKEV2_STATE_AUTHENTICATED)
5514             continue;
5515
5516           if (sa->old_remote_id_present && 0 > sa->old_id_expiration)
5517             {
5518               sa->old_remote_id_present = 0;
5519               del_old_ids = 1;
5520             }
5521           else
5522             sa->old_id_expiration -= 1;
5523
5524           vec_foreach (c, sa->childs)
5525             ikev2_mngr_process_child_sa (sa, c, del_old_ids);
5526
5527           if (!km->dpd_disabled && ikev2_mngr_process_responder_sas (sa))
5528             vec_add1 (to_be_deleted, sa - tkm->sas);
5529           }
5530
5531         vec_foreach (sai, to_be_deleted)
5532         {
5533           sa = pool_elt_at_index (tkm->sas, sai[0]);
5534           const u32 profile_index = sa->profile_index;
5535           const int reinitiate = (sa->is_initiator && profile_index != ~0);
5536           vec_foreach (c, sa->childs)
5537           {
5538             ikev2_delete_tunnel_interface (km->vnet_main, sa, c);
5539             ikev2_sa_del_child_sa (sa, c);
5540           }
5541           ikev2_sa_free_all_vec (sa);
5542           hash_unset (tkm->sa_by_rspi, sa->rspi);
5543           pool_put (tkm->sas, sa);
5544
5545           if (reinitiate)
5546             {
5547               p = pool_elt_at_index (km->profiles, profile_index);
5548               if (p)
5549                 {
5550                   clib_error_t *e = ikev2_initiate_sa_init (vm, p->name);
5551                   if (e)
5552                     {
5553                       ikev2_log_error ("%U", format_clib_error, e);
5554                       clib_error_free (e);
5555                     }
5556                 }
5557             }
5558         }
5559         vec_free (to_be_deleted);
5560       }
5561
5562       /* process ipsec sas */
5563       ipsec_sa_t *sa;
5564       pool_foreach (sa, ipsec_sa_pool)
5565         {
5566           ikev2_mngr_process_ipsec_sa (sa);
5567         }
5568
5569       ikev2_process_pending_sa_init (vm, km);
5570     }
5571   return 0;
5572 }
5573
5574 VLIB_REGISTER_NODE (ikev2_mngr_process_node, static) = {
5575     .function = ikev2_mngr_process_fn,
5576     .type = VLIB_NODE_TYPE_PROCESS,
5577     .name =
5578     "ikev2-manager-process",
5579 };
5580
5581 static void
5582 ikev2_lazy_init (ikev2_main_t *km)
5583 {
5584   vlib_thread_main_t *tm = vlib_get_thread_main ();
5585   int thread_id;
5586
5587   if (km->lazy_init_done)
5588     return;
5589
5590   ikev2_crypto_init (km);
5591
5592   mhash_init_vec_string (&km->profile_index_by_name, sizeof (uword));
5593
5594   vec_validate_aligned (km->per_thread_data, tm->n_vlib_mains - 1,
5595                         CLIB_CACHE_LINE_BYTES);
5596   for (thread_id = 0; thread_id < tm->n_vlib_mains; thread_id++)
5597     {
5598       ikev2_main_per_thread_data_t *ptd =
5599         vec_elt_at_index (km->per_thread_data, thread_id);
5600
5601       ptd->sa_by_rspi = hash_create (0, sizeof (uword));
5602
5603 #if OPENSSL_VERSION_NUMBER >= 0x10100000L
5604       ptd->evp_ctx = EVP_CIPHER_CTX_new ();
5605       ptd->hmac_ctx = HMAC_CTX_new ();
5606 #else
5607       EVP_CIPHER_CTX_init (&ptd->_evp_ctx);
5608       ptd->evp_ctx = &ptd->_evp_ctx;
5609       HMAC_CTX_init (&(ptd->_hmac_ctx));
5610       ptd->hmac_ctx = &ptd->_hmac_ctx;
5611 #endif
5612     }
5613
5614   km->sa_by_ispi = hash_create (0, sizeof (uword));
5615   km->sw_if_indices = hash_create (0, 0);
5616
5617   km->punt_hdl = vlib_punt_client_register ("ikev2");
5618
5619   km->dns_resolve_name_ptr =
5620     vlib_get_plugin_symbol ("dns_plugin.so", "dns_resolve_name");
5621   if (!km->dns_resolve_name_ptr)
5622     ikev2_log_error ("cannot load symbols from dns plugin");
5623
5624   /* wake up ikev2 process */
5625   vlib_process_signal_event (vlib_get_first_main (),
5626                              ikev2_mngr_process_node.index, 0, 0);
5627
5628   km->lazy_init_done = 1;
5629 }
5630
5631 VLIB_PLUGIN_REGISTER () = {
5632     .version = VPP_BUILD_VER,
5633     .description = "Internet Key Exchange (IKEv2) Protocol",
5634 };
5635
5636 /*
5637  * fd.io coding-style-patch-verification: ON
5638  *
5639  * Local Variables:
5640  * eval: (c-set-style "gnu")
5641  * End:
5642  */