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