src/plugins/dpdk/ipsec/esp_encrypt.c

   1 /*
   2  * esp_encrypt.c : IPSec ESP encrypt node using DPDK Cryptodev
   3  *
   4  * Copyright (c) 2017 Intel and/or its affiliates.
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at:
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #include <vnet/vnet.h>
  19 #include <vnet/api_errno.h>
  20 #include <vnet/ip/ip.h>
  21
  22 #include <vnet/ipsec/ipsec.h>
  23 #include <vnet/ipsec/esp.h>
  24 #include <dpdk/ipsec/ipsec.h>
  25 #include <dpdk/device/dpdk.h>
  26 #include <dpdk/device/dpdk_priv.h>
  27
  28 #define foreach_esp_encrypt_next                   \
  29 _(DROP, "error-drop")                              \
  30 _(IP4_LOOKUP, "ip4-lookup")                        \
  31 _(IP6_LOOKUP, "ip6-lookup")                        \
  32 _(INTERFACE_OUTPUT, "interface-output")
  33
  34 #define _(v, s) ESP_ENCRYPT_NEXT_##v,
  35 typedef enum
  36 {
  37   foreach_esp_encrypt_next
  38 #undef _
  39     ESP_ENCRYPT_N_NEXT,
  40 } esp_encrypt_next_t;
  41
  42 #define foreach_esp_encrypt_error                   \
  43  _(RX_PKTS, "ESP pkts received")                    \
  44  _(SEQ_CYCLED, "Sequence number cycled")            \
  45  _(ENQ_FAIL, "Enqueue failed to crypto device")     \
  46  _(DISCARD, "Not enough crypto operations, discarding frame")  \
  47  _(SESSION, "Failed to get crypto session")         \
  48  _(NOSUP, "Cipher/Auth not supported")
  49
  50
  51 typedef enum
  52 {
  53 #define _(sym,str) ESP_ENCRYPT_ERROR_##sym,
  54   foreach_esp_encrypt_error
  55 #undef _
  56     ESP_ENCRYPT_N_ERROR,
  57 } esp_encrypt_error_t;
  58
  59 static char *esp_encrypt_error_strings[] = {
  60 #define _(sym,string) string,
  61   foreach_esp_encrypt_error
  62 #undef _
  63 };
  64
  65 vlib_node_registration_t dpdk_esp_encrypt_node;
  66
  67 typedef struct
  68 {
  69   ipsec_crypto_alg_t crypto_alg;
  70   ipsec_integ_alg_t integ_alg;
  71   u8 packet_data[64];
  72 } esp_encrypt_trace_t;
  73
  74 /* packet trace format function */
  75 static u8 *
  76 format_esp_encrypt_trace (u8 * s, va_list * args)
  77 {
  78   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
  79   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
  80   esp_encrypt_trace_t *t = va_arg (*args, esp_encrypt_trace_t *);
  81   ip4_header_t *ih4 = (ip4_header_t *) t->packet_data;
  82   u32 indent = format_get_indent (s), offset;
  83
  84   s = format (s, "cipher %U auth %U\n",
  85               format_ipsec_crypto_alg, t->crypto_alg,
  86               format_ipsec_integ_alg, t->integ_alg);
  87
  88   if ((ih4->ip_version_and_header_length & 0xF0) == 0x60)
  89     {
  90       s = format (s, "%U%U", format_white_space, indent,
  91                   format_ip6_header, ih4);
  92       offset = sizeof (ip6_header_t);
  93     }
  94   else
  95     {
  96       s = format (s, "%U%U", format_white_space, indent,
  97                   format_ip4_header, ih4);
  98       offset = ip4_header_bytes (ih4);
  99     }
 100
 101   s = format (s, "\n%U%U", format_white_space, indent,
 102               format_esp_header, t->packet_data + offset);
 103
 104   return s;
 105 }
 106
 107 static uword
 108 dpdk_esp_encrypt_node_fn (vlib_main_t * vm,
 109                           vlib_node_runtime_t * node,
 110                           vlib_frame_t * from_frame)
 111 {
 112   u32 n_left_from, *from, *to_next, next_index;
 113   ipsec_main_t *im = &ipsec_main;
 114   u32 thread_idx = vlib_get_thread_index ();
 115   dpdk_crypto_main_t *dcm = &dpdk_crypto_main;
 116   crypto_resource_t *res = 0;
 117   ipsec_sa_t *sa0 = 0;
 118   crypto_alg_t *cipher_alg = 0, *auth_alg = 0;
 119   struct rte_cryptodev_sym_session *session = 0;
 120   u32 ret, last_sa_index = ~0;
 121   u8 numa = rte_socket_id ();
 122   u8 is_aead = 0;
 123   crypto_worker_main_t *cwm =
 124     vec_elt_at_index (dcm->workers_main, thread_idx);
 125   struct rte_crypto_op **ops = cwm->ops;
 126
 127   from = vlib_frame_vector_args (from_frame);
 128   n_left_from = from_frame->n_vectors;
 129
 130   ret = crypto_alloc_ops (numa, ops, n_left_from);
 131   if (ret)
 132     {
 133       vlib_node_increment_counter (vm, dpdk_esp_encrypt_node.index,
 134                                    ESP_ENCRYPT_ERROR_DISCARD, 1);
 135       /* Discard whole frame */
 136       return n_left_from;
 137     }
 138
 139   next_index = ESP_ENCRYPT_NEXT_DROP;
 140
 141   while (n_left_from > 0)
 142     {
 143       u32 n_left_to_next;
 144
 145       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
 146
 147       while (n_left_from > 0 && n_left_to_next > 0)
 148         {
 149           clib_error_t *error;
 150           u32 bi0;
 151           vlib_buffer_t *b0 = 0;
 152           u32 sa_index0;
 153           ip4_and_esp_header_t *ih0, *oh0 = 0;
 154           ip6_and_esp_header_t *ih6_0, *oh6_0 = 0;
 155           esp_header_t *esp0;
 156           esp_footer_t *f0;
 157           u8 is_ipv6, next_hdr_type;
 158           u32 iv_size;
 159           u16 orig_sz;
 160           u8 trunc_size;
 161           struct rte_mbuf *mb0 = 0;
 162           struct rte_crypto_op *op;
 163           u16 res_idx;
 164
 165           bi0 = from[0];
 166           from += 1;
 167           n_left_from -= 1;
 168
 169           b0 = vlib_get_buffer (vm, bi0);
 170           ih0 = vlib_buffer_get_current (b0);
 171           mb0 = rte_mbuf_from_vlib_buffer (b0);
 172
 173           /* ih0/ih6_0 */
 174           CLIB_PREFETCH (ih0, sizeof (ih6_0[0]), LOAD);
 175           /* f0 */
 176           CLIB_PREFETCH (vlib_buffer_get_tail (b0), 20, STORE);
 177           /* mb0 */
 178           CLIB_PREFETCH (mb0, CLIB_CACHE_LINE_BYTES, STORE);
 179
 180           op = ops[0];
 181           ops += 1;
 182           ASSERT (op->status == RTE_CRYPTO_OP_STATUS_NOT_PROCESSED);
 183
 184           dpdk_op_priv_t *priv = crypto_op_get_priv (op);
 185
 186           u16 op_len =
 187             sizeof (op[0]) + sizeof (op[0].sym[0]) + sizeof (priv[0]);
 188           CLIB_PREFETCH (op, op_len, STORE);
 189
 190           sa_index0 = vnet_buffer (b0)->ipsec.sad_index;
 191
 192           if (sa_index0 != last_sa_index)
 193             {
 194               sa0 = pool_elt_at_index (im->sad, sa_index0);
 195
 196               cipher_alg =
 197                 vec_elt_at_index (dcm->cipher_algs, sa0->crypto_alg);
 198               auth_alg = vec_elt_at_index (dcm->auth_algs, sa0->integ_alg);
 199
 200               is_aead = (cipher_alg->type == RTE_CRYPTO_SYM_XFORM_AEAD);
 201
 202               if (is_aead)
 203                 auth_alg = cipher_alg;
 204
 205               res_idx = get_resource (cwm, sa0);
 206
 207               if (PREDICT_FALSE (res_idx == (u16) ~ 0))
 208                 {
 209                   clib_warning ("unsupported SA by thread index %u",
 210                                 thread_idx);
 211                   vlib_node_increment_counter (vm,
 212                                                dpdk_esp_encrypt_node.index,
 213                                                ESP_ENCRYPT_ERROR_NOSUP, 1);
 214                   to_next[0] = bi0;
 215                   to_next += 1;
 216                   n_left_to_next -= 1;
 217                   goto trace;
 218                 }
 219               res = vec_elt_at_index (dcm->resource, res_idx);
 220
 221               error = crypto_get_session (&session, sa_index0, res, cwm, 1);
 222               if (PREDICT_FALSE (error || !session))
 223                 {
 224                   clib_warning ("failed to get crypto session");
 225                   vlib_node_increment_counter (vm,
 226                                                dpdk_esp_encrypt_node.index,
 227                                                ESP_ENCRYPT_ERROR_SESSION, 1);
 228                   to_next[0] = bi0;
 229                   to_next += 1;
 230                   n_left_to_next -= 1;
 231                   goto trace;
 232                 }
 233
 234               last_sa_index = sa_index0;
 235             }
 236
 237           if (PREDICT_FALSE (esp_seq_advance (sa0)))
 238             {
 239               clib_warning ("sequence number counter has cycled SPI %u",
 240                             sa0->spi);
 241               vlib_node_increment_counter (vm, dpdk_esp_encrypt_node.index,
 242                                            ESP_ENCRYPT_ERROR_SEQ_CYCLED, 1);
 243               //TODO: rekey SA
 244               to_next[0] = bi0;
 245               to_next += 1;
 246               n_left_to_next -= 1;
 247               goto trace;
 248             }
 249
 250           orig_sz = b0->current_length;
 251
 252           /* TODO multi-seg support - total_length_not_including_first_buffer */
 253           sa0->total_data_size += b0->current_length;
 254
 255           res->ops[res->n_ops] = op;
 256           res->bi[res->n_ops] = bi0;
 257           res->n_ops += 1;
 258
 259           dpdk_gcm_cnt_blk *icb = &priv->cb;
 260
 261           crypto_set_icb (icb, sa0->salt, sa0->seq, sa0->seq_hi);
 262
 263           is_ipv6 = (ih0->ip4.ip_version_and_header_length & 0xF0) == 0x60;
 264
 265           iv_size = cipher_alg->iv_len;
 266           trunc_size = auth_alg->trunc_size;
 267
 268           if (sa0->is_tunnel)
 269             {
 270               if (!is_ipv6 && !sa0->is_tunnel_ip6)      /* ip4inip4 */
 271                 {
 272                   /* in tunnel mode send it back to FIB */
 273                   priv->next = DPDK_CRYPTO_INPUT_NEXT_IP4_LOOKUP;
 274                   u8 adv =
 275                     sizeof (ip4_header_t) + sizeof (esp_header_t) + iv_size;
 276                   vlib_buffer_advance (b0, -adv);
 277                   oh0 = vlib_buffer_get_current (b0);
 278                   next_hdr_type = IP_PROTOCOL_IP_IN_IP;
 279                   /*
 280                    * oh0->ip4.ip_version_and_header_length = 0x45;
 281                    * oh0->ip4.tos = ih0->ip4.tos;
 282                    * oh0->ip4.fragment_id = 0;
 283                    * oh0->ip4.flags_and_fragment_offset = 0;
 284                    */
 285                   oh0->ip4.checksum_data_64[0] =
 286                     clib_host_to_net_u64 (0x45ULL << 56);
 287                   /*
 288                    * oh0->ip4.ttl = 254;
 289                    * oh0->ip4.protocol = IP_PROTOCOL_IPSEC_ESP;
 290                    */
 291                   oh0->ip4.checksum_data_32[2] =
 292                     clib_host_to_net_u32 (0xfe320000);
 293
 294                   oh0->ip4.src_address.as_u32 =
 295                     sa0->tunnel_src_addr.ip4.as_u32;
 296                   oh0->ip4.dst_address.as_u32 =
 297                     sa0->tunnel_dst_addr.ip4.as_u32;
 298                   esp0 = &oh0->esp;
 299                   oh0->esp.spi = clib_host_to_net_u32 (sa0->spi);
 300                   oh0->esp.seq = clib_host_to_net_u32 (sa0->seq);
 301                 }
 302               else if (is_ipv6 && sa0->is_tunnel_ip6)   /* ip6inip6 */
 303                 {
 304                   /* in tunnel mode send it back to FIB */
 305                   priv->next = DPDK_CRYPTO_INPUT_NEXT_IP6_LOOKUP;
 306
 307                   u8 adv =
 308                     sizeof (ip6_header_t) + sizeof (esp_header_t) + iv_size;
 309                   vlib_buffer_advance (b0, -adv);
 310                   ih6_0 = (ip6_and_esp_header_t *) ih0;
 311                   oh6_0 = vlib_buffer_get_current (b0);
 312
 313                   next_hdr_type = IP_PROTOCOL_IPV6;
 314
 315                   oh6_0->ip6.ip_version_traffic_class_and_flow_label =
 316                     ih6_0->ip6.ip_version_traffic_class_and_flow_label;
 317
 318                   oh6_0->ip6.protocol = IP_PROTOCOL_IPSEC_ESP;
 319                   oh6_0->ip6.hop_limit = 254;
 320                   oh6_0->ip6.src_address.as_u64[0] =
 321                     sa0->tunnel_src_addr.ip6.as_u64[0];
 322                   oh6_0->ip6.src_address.as_u64[1] =
 323                     sa0->tunnel_src_addr.ip6.as_u64[1];
 324                   oh6_0->ip6.dst_address.as_u64[0] =
 325                     sa0->tunnel_dst_addr.ip6.as_u64[0];
 326                   oh6_0->ip6.dst_address.as_u64[1] =
 327                     sa0->tunnel_dst_addr.ip6.as_u64[1];
 328                   esp0 = &oh6_0->esp;
 329                   oh6_0->esp.spi = clib_host_to_net_u32 (sa0->spi);
 330                   oh6_0->esp.seq = clib_host_to_net_u32 (sa0->seq);
 331                 }
 332               else              /* unsupported ip4inip6, ip6inip4 */
 333                 {
 334                   vlib_node_increment_counter (vm,
 335                                                dpdk_esp_encrypt_node.index,
 336                                                ESP_ENCRYPT_ERROR_NOSUP, 1);
 337                   to_next[0] = bi0;
 338                   to_next += 1;
 339                   n_left_to_next -= 1;
 340                   goto trace;
 341                 }
 342               vnet_buffer (b0)->sw_if_index[VLIB_TX] = (u32) ~ 0;
 343             }
 344           else                  /* transport mode */
 345             {
 346               priv->next = DPDK_CRYPTO_INPUT_NEXT_INTERFACE_OUTPUT;
 347               u16 rewrite_len = vnet_buffer (b0)->ip.save_rewrite_length;
 348               u16 adv = sizeof (esp_header_t) + iv_size;
 349               vlib_buffer_advance (b0, -rewrite_len - adv);
 350               u8 *src = ((u8 *) ih0) - rewrite_len;
 351               u8 *dst = vlib_buffer_get_current (b0);
 352               oh0 = (ip4_and_esp_header_t *) (dst + rewrite_len);
 353
 354               if (is_ipv6)
 355                 {
 356                   orig_sz -= sizeof (ip6_header_t);
 357                   ih6_0 = (ip6_and_esp_header_t *) ih0;
 358                   next_hdr_type = ih6_0->ip6.protocol;
 359                   memmove (dst, src, rewrite_len + sizeof (ip6_header_t));
 360                   oh6_0 = (ip6_and_esp_header_t *) oh0;
 361                   oh6_0->ip6.protocol = IP_PROTOCOL_IPSEC_ESP;
 362                   esp0 = &oh6_0->esp;
 363                 }
 364               else              /* ipv4 */
 365                 {
 366                   orig_sz -= ip4_header_bytes (&ih0->ip4);
 367                   next_hdr_type = ih0->ip4.protocol;
 368                   memmove (dst, src,
 369                            rewrite_len + ip4_header_bytes (&ih0->ip4));
 370                   oh0->ip4.protocol = IP_PROTOCOL_IPSEC_ESP;
 371                   esp0 =
 372                     (esp_header_t *) (oh0 + ip4_header_bytes (&ih0->ip4));
 373                 }
 374               esp0->spi = clib_host_to_net_u32 (sa0->spi);
 375               esp0->seq = clib_host_to_net_u32 (sa0->seq);
 376             }
 377
 378           ASSERT (is_pow2 (cipher_alg->boundary));
 379           u16 mask = cipher_alg->boundary - 1;
 380           u16 pad_payload_len = ((orig_sz + 2) + mask) & ~mask;
 381           u8 pad_bytes = pad_payload_len - 2 - orig_sz;
 382
 383           u8 *padding =
 384             vlib_buffer_put_uninit (b0, pad_bytes + 2 + trunc_size);
 385
 386           if (pad_bytes)
 387             clib_memcpy (padding, pad_data, 16);
 388
 389           f0 = (esp_footer_t *) (padding + pad_bytes);
 390           f0->pad_length = pad_bytes;
 391           f0->next_header = next_hdr_type;
 392
 393           if (is_ipv6)
 394             {
 395               u16 len = b0->current_length - sizeof (ip6_header_t);
 396               oh6_0->ip6.payload_length = clib_host_to_net_u16 (len);
 397             }
 398           else
 399             {
 400               oh0->ip4.length = clib_host_to_net_u16 (b0->current_length);
 401               oh0->ip4.checksum = ip4_header_checksum (&oh0->ip4);
 402             }
 403
 404           vnet_buffer (b0)->sw_if_index[VLIB_RX] =
 405             vnet_buffer (b0)->sw_if_index[VLIB_RX];
 406           b0->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
 407
 408           /* mbuf packet starts at ESP header */
 409           mb0->data_len = vlib_buffer_get_tail (b0) - ((u8 *) esp0);
 410           mb0->pkt_len = vlib_buffer_get_tail (b0) - ((u8 *) esp0);
 411           mb0->data_off = ((void *) esp0) - mb0->buf_addr;
 412
 413           u32 cipher_off, cipher_len;
 414           u32 auth_len = 0, aad_size = 0;
 415           u32 *aad = NULL;
 416           u8 *digest = vlib_buffer_get_tail (b0) - trunc_size;
 417           u64 digest_paddr =
 418             mb0->buf_physaddr + digest - ((u8 *) mb0->buf_addr);
 419
 420           if (!is_aead && cipher_alg->alg == RTE_CRYPTO_CIPHER_AES_CBC)
 421             {
 422               cipher_off = sizeof (esp_header_t);
 423               cipher_len = iv_size + pad_payload_len;
 424             }
 425           else                  /* CTR/GCM */
 426             {
 427               u32 *esp_iv = (u32 *) (esp0 + 1);
 428               esp_iv[0] = sa0->seq;
 429               esp_iv[1] = sa0->seq_hi;
 430
 431               cipher_off = sizeof (esp_header_t) + iv_size;
 432               cipher_len = pad_payload_len;
 433
 434               iv_size = 12;     /* CTR/GCM IV size, not ESP IV size */
 435             }
 436
 437           if (is_aead)
 438             {
 439               aad = (u32 *) priv->aad;
 440               aad[0] = clib_host_to_net_u32 (sa0->spi);
 441               aad[1] = clib_host_to_net_u32 (sa0->seq);
 442
 443               if (sa0->use_esn)
 444                 {
 445                   aad[2] = clib_host_to_net_u32 (sa0->seq_hi);
 446                   aad_size = 12;
 447                 }
 448               else
 449                 aad_size = 8;
 450             }
 451           else
 452             {
 453               auth_len =
 454                 vlib_buffer_get_tail (b0) - ((u8 *) esp0) - trunc_size;
 455               if (sa0->use_esn)
 456                 {
 457                   *((u32 *) digest) = sa0->seq_hi;
 458                   auth_len += 4;
 459                 }
 460             }
 461
 462           crypto_op_setup (is_aead, mb0, op, session,
 463                            cipher_off, cipher_len, (u8 *) icb, iv_size,
 464                            0, auth_len, (u8 *) aad, aad_size,
 465                            digest, digest_paddr, trunc_size);
 466
 467         trace:
 468           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
 469             {
 470               esp_encrypt_trace_t *tr =
 471                 vlib_add_trace (vm, node, b0, sizeof (*tr));
 472               tr->crypto_alg = sa0->crypto_alg;
 473               tr->integ_alg = sa0->integ_alg;
 474               u8 *p = vlib_buffer_get_current (b0);
 475               if (!sa0->is_tunnel)
 476                 p += vnet_buffer (b0)->ip.save_rewrite_length;
 477               clib_memcpy (tr->packet_data, p, sizeof (tr->packet_data));
 478             }
 479         }
 480       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 481     }
 482   vlib_node_increment_counter (vm, dpdk_esp_encrypt_node.index,
 483                                ESP_ENCRYPT_ERROR_RX_PKTS,
 484                                from_frame->n_vectors);
 485
 486   crypto_enqueue_ops (vm, cwm, 1, dpdk_esp_encrypt_node.index,
 487                       ESP_ENCRYPT_ERROR_ENQ_FAIL, numa);
 488
 489   crypto_free_ops (numa, ops, cwm->ops + from_frame->n_vectors - ops);
 490
 491   return from_frame->n_vectors;
 492 }
 493
 494 /* *INDENT-OFF* */
 495 VLIB_REGISTER_NODE (dpdk_esp_encrypt_node) = {
 496   .function = dpdk_esp_encrypt_node_fn,
 497   .name = "dpdk-esp-encrypt",
 498   .flags = VLIB_NODE_FLAG_IS_OUTPUT,
 499   .vector_size = sizeof (u32),
 500   .format_trace = format_esp_encrypt_trace,
 501   .n_errors = ARRAY_LEN (esp_encrypt_error_strings),
 502   .error_strings = esp_encrypt_error_strings,
 503   .n_next_nodes = 1,
 504   .next_nodes =
 505     {
 506       [ESP_ENCRYPT_NEXT_DROP] = "error-drop",
 507     }
 508 };
 509 /* *INDENT-ON* */
 510
 511 VLIB_NODE_FUNCTION_MULTIARCH (dpdk_esp_encrypt_node, dpdk_esp_encrypt_node_fn)
 512 /*
 513  * fd.io coding-style-patch-verification: ON
 514  *
 515  * Local Variables:
 516  * eval: (c-set-style "gnu")
 517  * End:
 518  */