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