src/plugins/crypto_ipsecmb/ipsecmb.c

   1 /*
   2  * ipsecmb.c - Intel IPSec Multi-buffer library Crypto Engine
   3  *
   4  * Copyright (c) 2019 Cisco Systemss
   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 <fcntl.h>
  19
  20 #include <intel-ipsec-mb.h>
  21
  22 #include <vnet/vnet.h>
  23 #include <vnet/plugin/plugin.h>
  24 #include <vpp/app/version.h>
  25 #include <vnet/crypto/crypto.h>
  26 #include <vppinfra/cpu.h>
  27
  28 #define HMAC_MAX_BLOCK_SIZE SHA_512_BLOCK_SIZE
  29 #define EXPANDED_KEY_N_BYTES (16 * 15)
  30
  31 typedef struct
  32 {
  33   MB_MGR *mgr;
  34   __m128i cbc_iv;
  35 } ipsecmb_per_thread_data_t;
  36
  37 typedef struct
  38 {
  39   u16 data_size;
  40   u8 block_size;
  41   aes_gcm_pre_t aes_gcm_pre;
  42   keyexp_t keyexp;
  43   hash_one_block_t hash_one_block;
  44   hash_fn_t hash_fn;
  45 } ipsecmb_alg_data_t;
  46
  47 typedef struct ipsecmb_main_t_
  48 {
  49   ipsecmb_per_thread_data_t *per_thread_data;
  50   ipsecmb_alg_data_t alg_data[VNET_CRYPTO_N_ALGS];
  51   void **key_data;
  52 } ipsecmb_main_t;
  53
  54 typedef struct
  55 {
  56   u8 enc_key_exp[EXPANDED_KEY_N_BYTES];
  57   u8 dec_key_exp[EXPANDED_KEY_N_BYTES];
  58 } ipsecmb_aes_cbc_key_data_t;
  59
  60 static ipsecmb_main_t ipsecmb_main = { };
  61
  62 /*
  63  * (Alg, JOB_HASH_ALG, fn, block-size-bytes, hash-size-bytes, digest-size-bytes)
  64  */
  65 #define foreach_ipsecmb_hmac_op                                \
  66   _(SHA1,   SHA1,    sha1,   64,  20, 20)                      \
  67   _(SHA224, SHA_224, sha224, 64,  32, 28)                      \
  68   _(SHA256, SHA_256, sha256, 64,  32, 32)                      \
  69   _(SHA384, SHA_384, sha384, 128, 64, 48)                      \
  70   _(SHA512, SHA_512, sha512, 128, 64, 64)
  71
  72 /*
  73  * (Alg, key-len-bits)
  74  */
  75 #define foreach_ipsecmb_cbc_cipher_op                          \
  76   _(AES_128_CBC, 128)                                          \
  77   _(AES_192_CBC, 192)                                          \
  78   _(AES_256_CBC, 256)
  79
  80 /*
  81  * (Alg, key-len-bytes, iv-len-bytes)
  82  */
  83 #define foreach_ipsecmb_gcm_cipher_op                          \
  84   _(AES_128_GCM, 128)                                          \
  85   _(AES_192_GCM, 192)                                          \
  86   _(AES_256_GCM, 256)
  87
  88 always_inline void
  89 ipsecmb_retire_hmac_job (JOB_AES_HMAC * job, u32 * n_fail, u32 digest_size)
  90 {
  91   vnet_crypto_op_t *op = job->user_data;
  92   u32 len = op->digest_len ? op->digest_len : digest_size;
  93
  94   if (STS_COMPLETED != job->status)
  95     {
  96       op->status = VNET_CRYPTO_OP_STATUS_FAIL_BAD_HMAC;
  97       *n_fail = *n_fail + 1;
  98       return;
  99     }
 100
 101   if (op->flags & VNET_CRYPTO_OP_FLAG_HMAC_CHECK)
 102     {
 103       if ((memcmp (op->digest, job->auth_tag_output, len)))
 104         {
 105           *n_fail = *n_fail + 1;
 106           op->status = VNET_CRYPTO_OP_STATUS_FAIL_BAD_HMAC;
 107           return;
 108         }
 109     }
 110   else if (len == digest_size)
 111     clib_memcpy_fast (op->digest, job->auth_tag_output, digest_size);
 112   else
 113     clib_memcpy_fast (op->digest, job->auth_tag_output, len);
 114
 115   op->status = VNET_CRYPTO_OP_STATUS_COMPLETED;
 116 }
 117
 118 static_always_inline u32
 119 ipsecmb_ops_hmac_inline (vlib_main_t * vm, vnet_crypto_op_t * ops[],
 120                          u32 n_ops, u32 block_size, u32 hash_size,
 121                          u32 digest_size, JOB_HASH_ALG alg)
 122 {
 123   ipsecmb_main_t *imbm = &ipsecmb_main;
 124   ipsecmb_per_thread_data_t *ptd = vec_elt_at_index (imbm->per_thread_data,
 125                                                      vm->thread_index);
 126   JOB_AES_HMAC *job;
 127   u32 i, n_fail = 0;
 128   u8 scratch[n_ops][digest_size];
 129
 130   /*
 131    * queue all the jobs first ...
 132    */
 133   for (i = 0; i < n_ops; i++)
 134     {
 135       vnet_crypto_op_t *op = ops[i];
 136       u8 *kd = (u8 *) imbm->key_data[op->key_index];
 137
 138       job = IMB_GET_NEXT_JOB (ptd->mgr);
 139
 140       job->src = op->src;
 141       job->hash_start_src_offset_in_bytes = 0;
 142       job->msg_len_to_hash_in_bytes = op->len;
 143       job->hash_alg = alg;
 144       job->auth_tag_output_len_in_bytes = digest_size;
 145       job->auth_tag_output = scratch[i];
 146
 147       job->cipher_mode = NULL_CIPHER;
 148       job->cipher_direction = DECRYPT;
 149       job->chain_order = HASH_CIPHER;
 150
 151       job->u.HMAC._hashed_auth_key_xor_ipad = kd;
 152       job->u.HMAC._hashed_auth_key_xor_opad = kd + hash_size;
 153       job->user_data = op;
 154
 155       job = IMB_SUBMIT_JOB (ptd->mgr);
 156
 157       if (job)
 158         ipsecmb_retire_hmac_job (job, &n_fail, digest_size);
 159     }
 160
 161   while ((job = IMB_FLUSH_JOB (ptd->mgr)))
 162     ipsecmb_retire_hmac_job (job, &n_fail, digest_size);
 163
 164   return n_ops - n_fail;
 165 }
 166
 167 #define _(a, b, c, d, e, f)                                             \
 168 static_always_inline u32                                                \
 169 ipsecmb_ops_hmac_##a (vlib_main_t * vm,                                 \
 170                       vnet_crypto_op_t * ops[],                         \
 171                       u32 n_ops)                                        \
 172 { return ipsecmb_ops_hmac_inline (vm, ops, n_ops, d, e, f, b); }        \
 173
 174 foreach_ipsecmb_hmac_op;
 175 #undef _
 176
 177 always_inline void
 178 ipsecmb_retire_cipher_job (JOB_AES_HMAC * job, u32 * n_fail)
 179 {
 180   vnet_crypto_op_t *op = job->user_data;
 181
 182   if (STS_COMPLETED != job->status)
 183     {
 184       op->status = VNET_CRYPTO_OP_STATUS_FAIL_BAD_HMAC;
 185       *n_fail = *n_fail + 1;
 186     }
 187   else
 188     op->status = VNET_CRYPTO_OP_STATUS_COMPLETED;
 189 }
 190
 191 static_always_inline u32
 192 ipsecmb_ops_cbc_cipher_inline (vlib_main_t * vm, vnet_crypto_op_t * ops[],
 193                                u32 n_ops, u32 key_len,
 194                                JOB_CIPHER_DIRECTION direction)
 195 {
 196   ipsecmb_main_t *imbm = &ipsecmb_main;
 197   ipsecmb_per_thread_data_t *ptd = vec_elt_at_index (imbm->per_thread_data,
 198                                                      vm->thread_index);
 199   JOB_AES_HMAC *job;
 200   u32 i, n_fail = 0;
 201
 202   for (i = 0; i < n_ops; i++)
 203     {
 204       ipsecmb_aes_cbc_key_data_t *kd;
 205       vnet_crypto_op_t *op = ops[i];
 206       kd = (ipsecmb_aes_cbc_key_data_t *) imbm->key_data[op->key_index];
 207       __m128i iv;
 208
 209       job = IMB_GET_NEXT_JOB (ptd->mgr);
 210
 211       job->src = op->src;
 212       job->dst = op->dst;
 213       job->msg_len_to_cipher_in_bytes = op->len;
 214       job->cipher_start_src_offset_in_bytes = 0;
 215
 216       job->hash_alg = NULL_HASH;
 217       job->cipher_mode = CBC;
 218       job->cipher_direction = direction;
 219       job->chain_order = (direction == ENCRYPT ? CIPHER_HASH : HASH_CIPHER);
 220
 221       if ((direction == ENCRYPT) && (op->flags & VNET_CRYPTO_OP_FLAG_INIT_IV))
 222         {
 223           iv = ptd->cbc_iv;
 224           _mm_storeu_si128 ((__m128i *) op->iv, iv);
 225           ptd->cbc_iv = _mm_aesenc_si128 (iv, iv);
 226         }
 227
 228       job->aes_key_len_in_bytes = key_len / 8;
 229       job->aes_enc_key_expanded = kd->enc_key_exp;
 230       job->aes_dec_key_expanded = kd->dec_key_exp;
 231       job->iv = op->iv;
 232       job->iv_len_in_bytes = AES_BLOCK_SIZE;
 233
 234       job->user_data = op;
 235
 236       job = IMB_SUBMIT_JOB (ptd->mgr);
 237
 238       if (job)
 239         ipsecmb_retire_cipher_job (job, &n_fail);
 240     }
 241
 242   while ((job = IMB_FLUSH_JOB (ptd->mgr)))
 243     ipsecmb_retire_cipher_job (job, &n_fail);
 244
 245   return n_ops - n_fail;
 246 }
 247
 248 #define _(a, b)                                                              \
 249 static_always_inline u32                                                     \
 250 ipsecmb_ops_cbc_cipher_enc_##a (vlib_main_t * vm,                            \
 251                                 vnet_crypto_op_t * ops[],                    \
 252                                 u32 n_ops)                                   \
 253 { return ipsecmb_ops_cbc_cipher_inline (vm, ops, n_ops, b, ENCRYPT); }       \
 254                                                                              \
 255 static_always_inline u32                                                     \
 256 ipsecmb_ops_cbc_cipher_dec_##a (vlib_main_t * vm,                            \
 257                                 vnet_crypto_op_t * ops[],                    \
 258                                 u32 n_ops)                                   \
 259 { return ipsecmb_ops_cbc_cipher_inline (vm, ops, n_ops, b, DECRYPT); }       \
 260
 261 foreach_ipsecmb_cbc_cipher_op;
 262 #undef _
 263
 264 always_inline void
 265 ipsecmb_retire_gcm_cipher_job (JOB_AES_HMAC * job,
 266                                u32 * n_fail, JOB_CIPHER_DIRECTION direction)
 267 {
 268   vnet_crypto_op_t *op = job->user_data;
 269
 270   if (STS_COMPLETED != job->status)
 271     {
 272       op->status = VNET_CRYPTO_OP_STATUS_FAIL_BAD_HMAC;
 273       *n_fail = *n_fail + 1;
 274       return;
 275     }
 276   else
 277     op->status = VNET_CRYPTO_OP_STATUS_COMPLETED;
 278
 279   if (DECRYPT == direction)
 280     {
 281       if ((memcmp (op->tag, job->auth_tag_output, op->tag_len)))
 282         {
 283           *n_fail = *n_fail + 1;
 284           op->status = VNET_CRYPTO_OP_STATUS_FAIL_BAD_HMAC;
 285         }
 286     }
 287 }
 288
 289 static_always_inline u32
 290 ipsecmb_ops_gcm_cipher_inline (vlib_main_t * vm, vnet_crypto_op_t * ops[],
 291                                u32 n_ops, u32 key_len,
 292                                JOB_CIPHER_DIRECTION direction)
 293 {
 294   ipsecmb_main_t *imbm = &ipsecmb_main;
 295   ipsecmb_per_thread_data_t *ptd = vec_elt_at_index (imbm->per_thread_data,
 296                                                      vm->thread_index);
 297   JOB_AES_HMAC *job;
 298   u32 i, n_fail = 0;
 299   u8 scratch[n_ops][64];
 300
 301   /*
 302    * queue all the jobs first ...
 303    */
 304   for (i = 0; i < n_ops; i++)
 305     {
 306       struct gcm_key_data *kd;
 307       vnet_crypto_op_t *op = ops[i];
 308       kd = (struct gcm_key_data *) imbm->key_data[op->key_index];
 309
 310       job = IMB_GET_NEXT_JOB (ptd->mgr);
 311
 312       job->src = op->src;
 313       job->dst = op->dst;
 314       job->msg_len_to_cipher_in_bytes = op->len;
 315       job->cipher_start_src_offset_in_bytes = 0;
 316
 317       job->hash_alg = AES_GMAC;
 318       job->cipher_mode = GCM;
 319       job->cipher_direction = direction;
 320       job->chain_order = (direction == ENCRYPT ? CIPHER_HASH : HASH_CIPHER);
 321
 322       job->iv = op->iv;
 323       job->aes_key_len_in_bytes = key_len / 8;
 324       job->aes_enc_key_expanded = kd;
 325       job->aes_dec_key_expanded = kd;
 326       job->iv_len_in_bytes = 12;
 327
 328       job->u.GCM.aad = op->aad;
 329       job->u.GCM.aad_len_in_bytes = op->aad_len;
 330       job->auth_tag_output_len_in_bytes = op->tag_len;
 331       if (DECRYPT == direction)
 332         job->auth_tag_output = scratch[i];
 333       else
 334         job->auth_tag_output = op->tag;
 335       job->user_data = op;
 336
 337       job = IMB_SUBMIT_JOB (ptd->mgr);
 338
 339       if (job)
 340         ipsecmb_retire_gcm_cipher_job (job, &n_fail, direction);
 341     }
 342
 343   while ((job = IMB_FLUSH_JOB (ptd->mgr)))
 344     ipsecmb_retire_gcm_cipher_job (job, &n_fail, direction);
 345
 346   return n_ops - n_fail;
 347 }
 348
 349 #define _(a, b)                                                              \
 350 static_always_inline u32                                                     \
 351 ipsecmb_ops_gcm_cipher_enc_##a (vlib_main_t * vm, vnet_crypto_op_t * ops[],  \
 352                                 u32 n_ops)                                   \
 353 { return ipsecmb_ops_gcm_cipher_inline (vm, ops, n_ops, b, ENCRYPT); }       \
 354                                                                              \
 355 static_always_inline u32                                                     \
 356 ipsecmb_ops_gcm_cipher_dec_##a (vlib_main_t * vm, vnet_crypto_op_t * ops[],  \
 357                                 u32 n_ops)                                   \
 358 { return ipsecmb_ops_gcm_cipher_inline (vm, ops, n_ops, b, DECRYPT); }       \
 359
 360 foreach_ipsecmb_gcm_cipher_op;
 361 #undef _
 362
 363 clib_error_t *
 364 crypto_ipsecmb_iv_init (ipsecmb_main_t * imbm)
 365 {
 366   ipsecmb_per_thread_data_t *ptd;
 367   clib_error_t *err = 0;
 368   int fd;
 369
 370   if ((fd = open ("/dev/urandom", O_RDONLY)) < 0)
 371     return clib_error_return_unix (0, "failed to open '/dev/urandom'");
 372
 373   vec_foreach (ptd, imbm->per_thread_data)
 374   {
 375     if (read (fd, &ptd->cbc_iv, sizeof (ptd->cbc_iv)) != sizeof (ptd->cbc_iv))
 376       {
 377         err = clib_error_return_unix (0, "'/dev/urandom' read failure");
 378         close (fd);
 379         return (err);
 380       }
 381   }
 382
 383   close (fd);
 384   return (NULL);
 385 }
 386
 387 static void
 388 crypto_ipsecmb_key_handler (vlib_main_t * vm, vnet_crypto_key_op_t kop,
 389                             vnet_crypto_key_index_t idx)
 390 {
 391   ipsecmb_main_t *imbm = &ipsecmb_main;
 392   vnet_crypto_key_t *key = vnet_crypto_get_key (idx);
 393   ipsecmb_alg_data_t *ad = imbm->alg_data + key->alg;
 394   u32 i;
 395   void *kd;
 396
 397   if (kop == VNET_CRYPTO_KEY_OP_DEL)
 398     {
 399       if (idx >= vec_len (imbm->key_data))
 400         return;
 401
 402       if (imbm->key_data[idx] == 0)
 403         return;
 404
 405       clib_mem_free_s (imbm->key_data[idx]);
 406       imbm->key_data[idx] = 0;
 407       return;
 408     }
 409
 410   if (ad->data_size == 0)
 411     return;
 412
 413   vec_validate_aligned (imbm->key_data, idx, CLIB_CACHE_LINE_BYTES);
 414
 415   if (kop == VNET_CRYPTO_KEY_OP_MODIFY && imbm->key_data[idx])
 416     {
 417       clib_mem_free_s (imbm->key_data[idx]);
 418     }
 419
 420   kd = imbm->key_data[idx] = clib_mem_alloc_aligned (ad->data_size,
 421                                                      CLIB_CACHE_LINE_BYTES);
 422
 423   /* AES CBC key expansion */
 424   if (ad->keyexp)
 425     {
 426       ad->keyexp (key->data, ((ipsecmb_aes_cbc_key_data_t *) kd)->enc_key_exp,
 427                   ((ipsecmb_aes_cbc_key_data_t *) kd)->dec_key_exp);
 428       return;
 429     }
 430
 431   /* AES GCM */
 432   if (ad->aes_gcm_pre)
 433     {
 434       ad->aes_gcm_pre (key->data, (struct gcm_key_data *) kd);
 435       return;
 436     }
 437
 438   /* HMAC */
 439   if (ad->hash_one_block)
 440     {
 441       const int block_qw = HMAC_MAX_BLOCK_SIZE / sizeof (u64);
 442       u64 pad[block_qw], key_hash[block_qw];
 443
 444       clib_memset_u8 (key_hash, 0, HMAC_MAX_BLOCK_SIZE);
 445       if (vec_len (key->data) <= ad->block_size)
 446         clib_memcpy_fast (key_hash, key->data, vec_len (key->data));
 447       else
 448         ad->hash_fn (key->data, vec_len (key->data), key_hash);
 449
 450       for (i = 0; i < block_qw; i++)
 451         pad[i] = key_hash[i] ^ 0x3636363636363636;
 452       ad->hash_one_block (pad, kd);
 453
 454       for (i = 0; i < block_qw; i++)
 455         pad[i] = key_hash[i] ^ 0x5c5c5c5c5c5c5c5c;
 456       ad->hash_one_block (pad, ((u8 *) kd) + (ad->data_size / 2));
 457
 458       return;
 459     }
 460 }
 461
 462 static clib_error_t *
 463 crypto_ipsecmb_init (vlib_main_t * vm)
 464 {
 465   ipsecmb_main_t *imbm = &ipsecmb_main;
 466   ipsecmb_alg_data_t *ad;
 467   ipsecmb_per_thread_data_t *ptd;
 468   vlib_thread_main_t *tm = vlib_get_thread_main ();
 469   clib_error_t *error;
 470   MB_MGR *m = 0;
 471   u32 eidx;
 472   u8 *name;
 473
 474   if (!clib_cpu_supports_aes ())
 475     return 0;
 476
 477   /*
 478    * A priority that is better than OpenSSL but worse than VPP natvie
 479    */
 480   name = format (0, "Intel(R) Multi-Buffer Crypto for IPsec Library %s%c",
 481                  IMB_VERSION_STR, 0);
 482   eidx = vnet_crypto_register_engine (vm, "ipsecmb", 80, (char *) name);
 483
 484   vec_validate (imbm->per_thread_data, tm->n_vlib_mains - 1);
 485
 486   /* *INDENT-OFF* */
 487   vec_foreach (ptd, imbm->per_thread_data)
 488     {
 489         ptd->mgr = alloc_mb_mgr (0);
 490         if (clib_cpu_supports_avx512f ())
 491           init_mb_mgr_avx512 (ptd->mgr);
 492         else if (clib_cpu_supports_avx2 ())
 493           init_mb_mgr_avx2 (ptd->mgr);
 494         else
 495           init_mb_mgr_sse (ptd->mgr);
 496
 497         if (ptd == imbm->per_thread_data)
 498           m = ptd->mgr;
 499     }
 500   /* *INDENT-ON* */
 501
 502   if (clib_cpu_supports_x86_aes () && (error = crypto_ipsecmb_iv_init (imbm)))
 503     return (error);
 504
 505 #define _(a, b, c, d, e, f)                                              \
 506   vnet_crypto_register_ops_handler (vm, eidx, VNET_CRYPTO_OP_##a##_HMAC, \
 507                                     ipsecmb_ops_hmac_##a);               \
 508   ad = imbm->alg_data + VNET_CRYPTO_ALG_HMAC_##a;                        \
 509   ad->block_size = d;                                                    \
 510   ad->data_size = e * 2;                                                 \
 511   ad->hash_one_block = m-> c##_one_block;                                \
 512   ad->hash_fn = m-> c;                                                   \
 513
 514   foreach_ipsecmb_hmac_op;
 515 #undef _
 516 #define _(a, b)                                                         \
 517   vnet_crypto_register_ops_handler (vm, eidx, VNET_CRYPTO_OP_##a##_ENC, \
 518                                     ipsecmb_ops_cbc_cipher_enc_##a);    \
 519   vnet_crypto_register_ops_handler (vm, eidx, VNET_CRYPTO_OP_##a##_DEC, \
 520                                     ipsecmb_ops_cbc_cipher_dec_##a);    \
 521   ad = imbm->alg_data + VNET_CRYPTO_ALG_##a;                            \
 522   ad->data_size = sizeof (ipsecmb_aes_cbc_key_data_t);                  \
 523   ad->keyexp = m->keyexp_##b;                                           \
 524
 525   foreach_ipsecmb_cbc_cipher_op;
 526 #undef _
 527 #define _(a, b)                                                         \
 528   vnet_crypto_register_ops_handler (vm, eidx, VNET_CRYPTO_OP_##a##_ENC, \
 529                                     ipsecmb_ops_gcm_cipher_enc_##a);    \
 530   vnet_crypto_register_ops_handler (vm, eidx, VNET_CRYPTO_OP_##a##_DEC, \
 531                                     ipsecmb_ops_gcm_cipher_dec_##a);    \
 532   ad = imbm->alg_data + VNET_CRYPTO_ALG_##a;                            \
 533   ad->data_size = sizeof (struct gcm_key_data);                         \
 534   ad->aes_gcm_pre = m->gcm##b##_pre;                                    \
 535
 536   foreach_ipsecmb_gcm_cipher_op;
 537 #undef _
 538
 539   vnet_crypto_register_key_handler (vm, eidx, crypto_ipsecmb_key_handler);
 540   return (NULL);
 541 }
 542
 543 /* *INDENT-OFF* */
 544 VLIB_INIT_FUNCTION (crypto_ipsecmb_init) =
 545 {
 546   .runs_after = VLIB_INITS ("vnet_crypto_init"),
 547 };
 548 /* *INDENT-ON* */
 549
 550 /* *INDENT-OFF* */
 551 VLIB_PLUGIN_REGISTER () =
 552 {
 553   .version = VPP_BUILD_VER,
 554   .description = "Intel IPSEC Multi-buffer Crypto Engine",
 555 };
 556 /* *INDENT-ON* */
 557
 558 /*
 559  * fd.io coding-style-patch-verification: ON
 560  *
 561  * Local Variables:
 562  * eval: (c-set-style "gnu")
 563  * End:
 564  */