drivers/crypto/aesni_mb/rte_aesni_mb_pmd.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2015-2017 Intel Corporation
   3  */
   4
   5 #include <intel-ipsec-mb.h>
   6
   7 #include <rte_common.h>
   8 #include <rte_hexdump.h>
   9 #include <rte_cryptodev.h>
  10 #include <rte_cryptodev_pmd.h>
  11 #include <rte_bus_vdev.h>
  12 #include <rte_malloc.h>
  13 #include <rte_cpuflags.h>
  14 #include <rte_per_lcore.h>
  15
  16 #include "aesni_mb_pmd_private.h"
  17
  18 int aesni_mb_logtype_driver;
  19
  20 #define AES_CCM_DIGEST_MIN_LEN 4
  21 #define AES_CCM_DIGEST_MAX_LEN 16
  22 #define HMAC_MAX_BLOCK_SIZE 128
  23 static uint8_t cryptodev_driver_id;
  24
  25 /*
  26  * Needed to support CPU-CRYPTO API (rte_cryptodev_sym_cpu_crypto_process),
  27  * as we still use JOB based API even for synchronous processing.
  28  */
  29 static RTE_DEFINE_PER_LCORE(MB_MGR *, sync_mb_mgr);
  30
  31 typedef void (*hash_one_block_t)(const void *data, void *digest);
  32 typedef void (*aes_keyexp_t)(const void *key, void *enc_exp_keys, void *dec_exp_keys);
  33
  34 /**
  35  * Calculate the authentication pre-computes
  36  *
  37  * @param one_block_hash        Function pointer to calculate digest on ipad/opad
  38  * @param ipad                  Inner pad output byte array
  39  * @param opad                  Outer pad output byte array
  40  * @param hkey                  Authentication key
  41  * @param hkey_len              Authentication key length
  42  * @param blocksize             Block size of selected hash algo
  43  */
  44 static void
  45 calculate_auth_precomputes(hash_one_block_t one_block_hash,
  46                 uint8_t *ipad, uint8_t *opad,
  47                 const uint8_t *hkey, uint16_t hkey_len,
  48                 uint16_t blocksize)
  49 {
  50         unsigned i, length;
  51
  52         uint8_t ipad_buf[blocksize] __rte_aligned(16);
  53         uint8_t opad_buf[blocksize] __rte_aligned(16);
  54
  55         /* Setup inner and outer pads */
  56         memset(ipad_buf, HMAC_IPAD_VALUE, blocksize);
  57         memset(opad_buf, HMAC_OPAD_VALUE, blocksize);
  58
  59         /* XOR hash key with inner and outer pads */
  60         length = hkey_len > blocksize ? blocksize : hkey_len;
  61
  62         for (i = 0; i < length; i++) {
  63                 ipad_buf[i] ^= hkey[i];
  64                 opad_buf[i] ^= hkey[i];
  65         }
  66
  67         /* Compute partial hashes */
  68         (*one_block_hash)(ipad_buf, ipad);
  69         (*one_block_hash)(opad_buf, opad);
  70
  71         /* Clean up stack */
  72         memset(ipad_buf, 0, blocksize);
  73         memset(opad_buf, 0, blocksize);
  74 }
  75
  76 /** Get xform chain order */
  77 static enum aesni_mb_operation
  78 aesni_mb_get_chain_order(const struct rte_crypto_sym_xform *xform)
  79 {
  80         if (xform == NULL)
  81                 return AESNI_MB_OP_NOT_SUPPORTED;
  82
  83         if (xform->type == RTE_CRYPTO_SYM_XFORM_CIPHER) {
  84                 if (xform->next == NULL)
  85                         return AESNI_MB_OP_CIPHER_ONLY;
  86                 if (xform->next->type == RTE_CRYPTO_SYM_XFORM_AUTH)
  87                         return AESNI_MB_OP_CIPHER_HASH;
  88         }
  89
  90         if (xform->type == RTE_CRYPTO_SYM_XFORM_AUTH) {
  91                 if (xform->next == NULL)
  92                         return AESNI_MB_OP_HASH_ONLY;
  93                 if (xform->next->type == RTE_CRYPTO_SYM_XFORM_CIPHER)
  94                         return AESNI_MB_OP_HASH_CIPHER;
  95         }
  96 #if IMB_VERSION_NUM > IMB_VERSION(0, 52, 0)
  97         if (xform->type == RTE_CRYPTO_SYM_XFORM_AEAD) {
  98                 if (xform->aead.op == RTE_CRYPTO_AEAD_OP_ENCRYPT) {
  99                         /*
 100                          * CCM requires to hash first and cipher later
 101                          * when encrypting
 102                          */
 103                         if (xform->aead.algo == RTE_CRYPTO_AEAD_AES_CCM)
 104                                 return AESNI_MB_OP_AEAD_HASH_CIPHER;
 105                         else
 106                                 return AESNI_MB_OP_AEAD_CIPHER_HASH;
 107                 } else {
 108                         if (xform->aead.algo == RTE_CRYPTO_AEAD_AES_CCM)
 109                                 return AESNI_MB_OP_AEAD_CIPHER_HASH;
 110                         else
 111                                 return AESNI_MB_OP_AEAD_HASH_CIPHER;
 112                 }
 113         }
 114 #else
 115         if (xform->type == RTE_CRYPTO_SYM_XFORM_AEAD) {
 116                 if (xform->aead.algo == RTE_CRYPTO_AEAD_AES_CCM ||
 117                                 xform->aead.algo == RTE_CRYPTO_AEAD_AES_GCM) {
 118                         if (xform->aead.op == RTE_CRYPTO_AEAD_OP_ENCRYPT)
 119                                 return AESNI_MB_OP_AEAD_CIPHER_HASH;
 120                         else
 121                                 return AESNI_MB_OP_AEAD_HASH_CIPHER;
 122                 }
 123         }
 124 #endif
 125
 126         return AESNI_MB_OP_NOT_SUPPORTED;
 127 }
 128
 129 /** Set session authentication parameters */
 130 static int
 131 aesni_mb_set_session_auth_parameters(const MB_MGR *mb_mgr,
 132                 struct aesni_mb_session *sess,
 133                 const struct rte_crypto_sym_xform *xform)
 134 {
 135         hash_one_block_t hash_oneblock_fn = NULL;
 136         unsigned int key_larger_block_size = 0;
 137         uint8_t hashed_key[HMAC_MAX_BLOCK_SIZE] = { 0 };
 138         uint32_t auth_precompute = 1;
 139
 140         if (xform == NULL) {
 141                 sess->auth.algo = NULL_HASH;
 142                 return 0;
 143         }
 144
 145         if (xform->type != RTE_CRYPTO_SYM_XFORM_AUTH) {
 146                 AESNI_MB_LOG(ERR, "Crypto xform struct not of type auth");
 147                 return -1;
 148         }
 149
 150         /* Set the request digest size */
 151         sess->auth.req_digest_len = xform->auth.digest_length;
 152
 153         /* Select auth generate/verify */
 154         sess->auth.operation = xform->auth.op;
 155
 156         /* Set Authentication Parameters */
 157         if (xform->auth.algo == RTE_CRYPTO_AUTH_AES_XCBC_MAC) {
 158                 sess->auth.algo = AES_XCBC;
 159
 160                 uint16_t xcbc_mac_digest_len =
 161                         get_truncated_digest_byte_length(AES_XCBC);
 162                 if (sess->auth.req_digest_len != xcbc_mac_digest_len) {
 163                         AESNI_MB_LOG(ERR, "Invalid digest size\n");
 164                         return -EINVAL;
 165                 }
 166                 sess->auth.gen_digest_len = sess->auth.req_digest_len;
 167
 168                 IMB_AES_XCBC_KEYEXP(mb_mgr, xform->auth.key.data,
 169                                 sess->auth.xcbc.k1_expanded,
 170                                 sess->auth.xcbc.k2, sess->auth.xcbc.k3);
 171                 return 0;
 172         }
 173
 174         if (xform->auth.algo == RTE_CRYPTO_AUTH_AES_CMAC) {
 175                 uint32_t dust[4*15];
 176
 177                 sess->auth.algo = AES_CMAC;
 178
 179                 uint16_t cmac_digest_len = get_digest_byte_length(AES_CMAC);
 180
 181                 if (sess->auth.req_digest_len > cmac_digest_len) {
 182                         AESNI_MB_LOG(ERR, "Invalid digest size\n");
 183                         return -EINVAL;
 184                 }
 185                 /*
 186                  * Multi-buffer lib supports digest sizes from 4 to 16 bytes
 187                  * in version 0.50 and sizes of 12 and 16 bytes,
 188                  * in version 0.49.
 189                  * If size requested is different, generate the full digest
 190                  * (16 bytes) in a temporary location and then memcpy
 191                  * the requested number of bytes.
 192                  */
 193                 if (sess->auth.req_digest_len < 4)
 194                         sess->auth.gen_digest_len = cmac_digest_len;
 195                 else
 196                         sess->auth.gen_digest_len = sess->auth.req_digest_len;
 197
 198                 IMB_AES_KEYEXP_128(mb_mgr, xform->auth.key.data,
 199                                 sess->auth.cmac.expkey, dust);
 200                 IMB_AES_CMAC_SUBKEY_GEN_128(mb_mgr, sess->auth.cmac.expkey,
 201                                 sess->auth.cmac.skey1, sess->auth.cmac.skey2);
 202                 return 0;
 203         }
 204
 205         if (xform->auth.algo == RTE_CRYPTO_AUTH_AES_GMAC) {
 206                 if (xform->auth.op == RTE_CRYPTO_AUTH_OP_GENERATE) {
 207                         sess->cipher.direction = ENCRYPT;
 208                         sess->chain_order = CIPHER_HASH;
 209                 } else
 210                         sess->cipher.direction = DECRYPT;
 211
 212                 sess->auth.algo = AES_GMAC;
 213                 /*
 214                  * Multi-buffer lib supports 8, 12 and 16 bytes of digest.
 215                  * If size requested is different, generate the full digest
 216                  * (16 bytes) in a temporary location and then memcpy
 217                  * the requested number of bytes.
 218                  */
 219                 if (sess->auth.req_digest_len != 16 &&
 220                                 sess->auth.req_digest_len != 12 &&
 221                                 sess->auth.req_digest_len != 8) {
 222                         sess->auth.gen_digest_len = 16;
 223                 } else {
 224                         sess->auth.gen_digest_len = sess->auth.req_digest_len;
 225                 }
 226                 sess->iv.length = xform->auth.iv.length;
 227                 sess->iv.offset = xform->auth.iv.offset;
 228
 229                 switch (xform->auth.key.length) {
 230                 case AES_128_BYTES:
 231                         IMB_AES128_GCM_PRE(mb_mgr, xform->auth.key.data,
 232                                 &sess->cipher.gcm_key);
 233                         sess->cipher.key_length_in_bytes = AES_128_BYTES;
 234                         break;
 235                 case AES_192_BYTES:
 236                         IMB_AES192_GCM_PRE(mb_mgr, xform->auth.key.data,
 237                                 &sess->cipher.gcm_key);
 238                         sess->cipher.key_length_in_bytes = AES_192_BYTES;
 239                         break;
 240                 case AES_256_BYTES:
 241                         IMB_AES256_GCM_PRE(mb_mgr, xform->auth.key.data,
 242                                 &sess->cipher.gcm_key);
 243                         sess->cipher.key_length_in_bytes = AES_256_BYTES;
 244                         break;
 245                 default:
 246                         RTE_LOG(ERR, PMD, "failed to parse test type\n");
 247                         return -EINVAL;
 248                 }
 249
 250                 return 0;
 251         }
 252
 253         switch (xform->auth.algo) {
 254         case RTE_CRYPTO_AUTH_MD5_HMAC:
 255                 sess->auth.algo = MD5;
 256                 hash_oneblock_fn = mb_mgr->md5_one_block;
 257                 break;
 258         case RTE_CRYPTO_AUTH_SHA1_HMAC:
 259                 sess->auth.algo = SHA1;
 260                 hash_oneblock_fn = mb_mgr->sha1_one_block;
 261                 if (xform->auth.key.length > get_auth_algo_blocksize(SHA1)) {
 262                         IMB_SHA1(mb_mgr,
 263                                 xform->auth.key.data,
 264                                 xform->auth.key.length,
 265                                 hashed_key);
 266                         key_larger_block_size = 1;
 267                 }
 268                 break;
 269         case RTE_CRYPTO_AUTH_SHA1:
 270                 sess->auth.algo = PLAIN_SHA1;
 271                 auth_precompute = 0;
 272                 break;
 273         case RTE_CRYPTO_AUTH_SHA224_HMAC:
 274                 sess->auth.algo = SHA_224;
 275                 hash_oneblock_fn = mb_mgr->sha224_one_block;
 276                 if (xform->auth.key.length > get_auth_algo_blocksize(SHA_224)) {
 277                         IMB_SHA224(mb_mgr,
 278                                 xform->auth.key.data,
 279                                 xform->auth.key.length,
 280                                 hashed_key);
 281                         key_larger_block_size = 1;
 282                 }
 283                 break;
 284         case RTE_CRYPTO_AUTH_SHA224:
 285                 sess->auth.algo = PLAIN_SHA_224;
 286                 auth_precompute = 0;
 287                 break;
 288         case RTE_CRYPTO_AUTH_SHA256_HMAC:
 289                 sess->auth.algo = SHA_256;
 290                 hash_oneblock_fn = mb_mgr->sha256_one_block;
 291                 if (xform->auth.key.length > get_auth_algo_blocksize(SHA_256)) {
 292                         IMB_SHA256(mb_mgr,
 293                                 xform->auth.key.data,
 294                                 xform->auth.key.length,
 295                                 hashed_key);
 296                         key_larger_block_size = 1;
 297                 }
 298                 break;
 299         case RTE_CRYPTO_AUTH_SHA256:
 300                 sess->auth.algo = PLAIN_SHA_256;
 301                 auth_precompute = 0;
 302                 break;
 303         case RTE_CRYPTO_AUTH_SHA384_HMAC:
 304                 sess->auth.algo = SHA_384;
 305                 hash_oneblock_fn = mb_mgr->sha384_one_block;
 306                 if (xform->auth.key.length > get_auth_algo_blocksize(SHA_384)) {
 307                         IMB_SHA384(mb_mgr,
 308                                 xform->auth.key.data,
 309                                 xform->auth.key.length,
 310                                 hashed_key);
 311                         key_larger_block_size = 1;
 312                 }
 313                 break;
 314         case RTE_CRYPTO_AUTH_SHA384:
 315                 sess->auth.algo = PLAIN_SHA_384;
 316                 auth_precompute = 0;
 317                 break;
 318         case RTE_CRYPTO_AUTH_SHA512_HMAC:
 319                 sess->auth.algo = SHA_512;
 320                 hash_oneblock_fn = mb_mgr->sha512_one_block;
 321                 if (xform->auth.key.length > get_auth_algo_blocksize(SHA_512)) {
 322                         IMB_SHA512(mb_mgr,
 323                                 xform->auth.key.data,
 324                                 xform->auth.key.length,
 325                                 hashed_key);
 326                         key_larger_block_size = 1;
 327                 }
 328                 break;
 329         case RTE_CRYPTO_AUTH_SHA512:
 330                 sess->auth.algo = PLAIN_SHA_512;
 331                 auth_precompute = 0;
 332                 break;
 333         default:
 334                 AESNI_MB_LOG(ERR, "Unsupported authentication algorithm selection");
 335                 return -ENOTSUP;
 336         }
 337         uint16_t trunc_digest_size =
 338                         get_truncated_digest_byte_length(sess->auth.algo);
 339         uint16_t full_digest_size =
 340                         get_digest_byte_length(sess->auth.algo);
 341
 342         if (sess->auth.req_digest_len > full_digest_size ||
 343                         sess->auth.req_digest_len == 0) {
 344                 AESNI_MB_LOG(ERR, "Invalid digest size\n");
 345                 return -EINVAL;
 346         }
 347
 348         if (sess->auth.req_digest_len != trunc_digest_size &&
 349                         sess->auth.req_digest_len != full_digest_size)
 350                 sess->auth.gen_digest_len = full_digest_size;
 351         else
 352                 sess->auth.gen_digest_len = sess->auth.req_digest_len;
 353
 354         /* Plain SHA does not require precompute key */
 355         if (auth_precompute == 0)
 356                 return 0;
 357
 358         /* Calculate Authentication precomputes */
 359         if (key_larger_block_size) {
 360                 calculate_auth_precomputes(hash_oneblock_fn,
 361                         sess->auth.pads.inner, sess->auth.pads.outer,
 362                         hashed_key,
 363                         xform->auth.key.length,
 364                         get_auth_algo_blocksize(sess->auth.algo));
 365         } else {
 366                 calculate_auth_precomputes(hash_oneblock_fn,
 367                         sess->auth.pads.inner, sess->auth.pads.outer,
 368                         xform->auth.key.data,
 369                         xform->auth.key.length,
 370                         get_auth_algo_blocksize(sess->auth.algo));
 371         }
 372
 373         return 0;
 374 }
 375
 376 /** Set session cipher parameters */
 377 static int
 378 aesni_mb_set_session_cipher_parameters(const MB_MGR *mb_mgr,
 379                 struct aesni_mb_session *sess,
 380                 const struct rte_crypto_sym_xform *xform)
 381 {
 382         uint8_t is_aes = 0;
 383         uint8_t is_3DES = 0;
 384
 385         if (xform == NULL) {
 386                 sess->cipher.mode = NULL_CIPHER;
 387                 return 0;
 388         }
 389
 390         if (xform->type != RTE_CRYPTO_SYM_XFORM_CIPHER) {
 391                 AESNI_MB_LOG(ERR, "Crypto xform struct not of type cipher");
 392                 return -EINVAL;
 393         }
 394
 395         /* Select cipher direction */
 396         switch (xform->cipher.op) {
 397         case RTE_CRYPTO_CIPHER_OP_ENCRYPT:
 398                 sess->cipher.direction = ENCRYPT;
 399                 break;
 400         case RTE_CRYPTO_CIPHER_OP_DECRYPT:
 401                 sess->cipher.direction = DECRYPT;
 402                 break;
 403         default:
 404                 AESNI_MB_LOG(ERR, "Invalid cipher operation parameter");
 405                 return -EINVAL;
 406         }
 407
 408         /* Select cipher mode */
 409         switch (xform->cipher.algo) {
 410         case RTE_CRYPTO_CIPHER_AES_CBC:
 411                 sess->cipher.mode = CBC;
 412                 is_aes = 1;
 413                 break;
 414         case RTE_CRYPTO_CIPHER_AES_CTR:
 415                 sess->cipher.mode = CNTR;
 416                 is_aes = 1;
 417                 break;
 418         case RTE_CRYPTO_CIPHER_AES_DOCSISBPI:
 419                 sess->cipher.mode = DOCSIS_SEC_BPI;
 420                 is_aes = 1;
 421                 break;
 422         case RTE_CRYPTO_CIPHER_DES_CBC:
 423                 sess->cipher.mode = DES;
 424                 break;
 425         case RTE_CRYPTO_CIPHER_DES_DOCSISBPI:
 426                 sess->cipher.mode = DOCSIS_DES;
 427                 break;
 428         case RTE_CRYPTO_CIPHER_3DES_CBC:
 429                 sess->cipher.mode = DES3;
 430                 is_3DES = 1;
 431                 break;
 432         default:
 433                 AESNI_MB_LOG(ERR, "Unsupported cipher mode parameter");
 434                 return -ENOTSUP;
 435         }
 436
 437         /* Set IV parameters */
 438         sess->iv.offset = xform->cipher.iv.offset;
 439         sess->iv.length = xform->cipher.iv.length;
 440
 441         /* Check key length and choose key expansion function for AES */
 442         if (is_aes) {
 443                 switch (xform->cipher.key.length) {
 444                 case AES_128_BYTES:
 445                         sess->cipher.key_length_in_bytes = AES_128_BYTES;
 446                         IMB_AES_KEYEXP_128(mb_mgr, xform->cipher.key.data,
 447                                         sess->cipher.expanded_aes_keys.encode,
 448                                         sess->cipher.expanded_aes_keys.decode);
 449                         break;
 450                 case AES_192_BYTES:
 451                         sess->cipher.key_length_in_bytes = AES_192_BYTES;
 452                         IMB_AES_KEYEXP_192(mb_mgr, xform->cipher.key.data,
 453                                         sess->cipher.expanded_aes_keys.encode,
 454                                         sess->cipher.expanded_aes_keys.decode);
 455                         break;
 456                 case AES_256_BYTES:
 457                         sess->cipher.key_length_in_bytes = AES_256_BYTES;
 458                         IMB_AES_KEYEXP_256(mb_mgr, xform->cipher.key.data,
 459                                         sess->cipher.expanded_aes_keys.encode,
 460                                         sess->cipher.expanded_aes_keys.decode);
 461                         break;
 462                 default:
 463                         AESNI_MB_LOG(ERR, "Invalid cipher key length");
 464                         return -EINVAL;
 465                 }
 466         } else if (is_3DES) {
 467                 uint64_t *keys[3] = {sess->cipher.exp_3des_keys.key[0],
 468                                 sess->cipher.exp_3des_keys.key[1],
 469                                 sess->cipher.exp_3des_keys.key[2]};
 470
 471                 switch (xform->cipher.key.length) {
 472                 case  24:
 473                         IMB_DES_KEYSCHED(mb_mgr, keys[0],
 474                                         xform->cipher.key.data);
 475                         IMB_DES_KEYSCHED(mb_mgr, keys[1],
 476                                         xform->cipher.key.data + 8);
 477                         IMB_DES_KEYSCHED(mb_mgr, keys[2],
 478                                         xform->cipher.key.data + 16);
 479
 480                         /* Initialize keys - 24 bytes: [K1-K2-K3] */
 481                         sess->cipher.exp_3des_keys.ks_ptr[0] = keys[0];
 482                         sess->cipher.exp_3des_keys.ks_ptr[1] = keys[1];
 483                         sess->cipher.exp_3des_keys.ks_ptr[2] = keys[2];
 484                         break;
 485                 case 16:
 486                         IMB_DES_KEYSCHED(mb_mgr, keys[0],
 487                                         xform->cipher.key.data);
 488                         IMB_DES_KEYSCHED(mb_mgr, keys[1],
 489                                         xform->cipher.key.data + 8);
 490                         /* Initialize keys - 16 bytes: [K1=K1,K2=K2,K3=K1] */
 491                         sess->cipher.exp_3des_keys.ks_ptr[0] = keys[0];
 492                         sess->cipher.exp_3des_keys.ks_ptr[1] = keys[1];
 493                         sess->cipher.exp_3des_keys.ks_ptr[2] = keys[0];
 494                         break;
 495                 case 8:
 496                         IMB_DES_KEYSCHED(mb_mgr, keys[0],
 497                                         xform->cipher.key.data);
 498
 499                         /* Initialize keys - 8 bytes: [K1 = K2 = K3] */
 500                         sess->cipher.exp_3des_keys.ks_ptr[0] = keys[0];
 501                         sess->cipher.exp_3des_keys.ks_ptr[1] = keys[0];
 502                         sess->cipher.exp_3des_keys.ks_ptr[2] = keys[0];
 503                         break;
 504                 default:
 505                         AESNI_MB_LOG(ERR, "Invalid cipher key length");
 506                         return -EINVAL;
 507                 }
 508
 509                 sess->cipher.key_length_in_bytes = 24;
 510         } else {
 511                 if (xform->cipher.key.length != 8) {
 512                         AESNI_MB_LOG(ERR, "Invalid cipher key length");
 513                         return -EINVAL;
 514                 }
 515                 sess->cipher.key_length_in_bytes = 8;
 516
 517                 IMB_DES_KEYSCHED(mb_mgr,
 518                         (uint64_t *)sess->cipher.expanded_aes_keys.encode,
 519                                 xform->cipher.key.data);
 520                 IMB_DES_KEYSCHED(mb_mgr,
 521                         (uint64_t *)sess->cipher.expanded_aes_keys.decode,
 522                                 xform->cipher.key.data);
 523         }
 524
 525         return 0;
 526 }
 527
 528 static int
 529 aesni_mb_set_session_aead_parameters(const MB_MGR *mb_mgr,
 530                 struct aesni_mb_session *sess,
 531                 const struct rte_crypto_sym_xform *xform)
 532 {
 533         switch (xform->aead.op) {
 534         case RTE_CRYPTO_AEAD_OP_ENCRYPT:
 535                 sess->cipher.direction = ENCRYPT;
 536                 sess->auth.operation = RTE_CRYPTO_AUTH_OP_GENERATE;
 537                 break;
 538         case RTE_CRYPTO_AEAD_OP_DECRYPT:
 539                 sess->cipher.direction = DECRYPT;
 540                 sess->auth.operation = RTE_CRYPTO_AUTH_OP_VERIFY;
 541                 break;
 542         default:
 543                 AESNI_MB_LOG(ERR, "Invalid aead operation parameter");
 544                 return -EINVAL;
 545         }
 546
 547         switch (xform->aead.algo) {
 548         case RTE_CRYPTO_AEAD_AES_CCM:
 549                 sess->cipher.mode = CCM;
 550                 sess->auth.algo = AES_CCM;
 551
 552                 /* Check key length and choose key expansion function for AES */
 553                 switch (xform->aead.key.length) {
 554                 case AES_128_BYTES:
 555                         sess->cipher.key_length_in_bytes = AES_128_BYTES;
 556                         IMB_AES_KEYEXP_128(mb_mgr, xform->aead.key.data,
 557                                         sess->cipher.expanded_aes_keys.encode,
 558                                         sess->cipher.expanded_aes_keys.decode);
 559                         break;
 560                 default:
 561                         AESNI_MB_LOG(ERR, "Invalid cipher key length");
 562                         return -EINVAL;
 563                 }
 564
 565                 break;
 566
 567         case RTE_CRYPTO_AEAD_AES_GCM:
 568                 sess->cipher.mode = GCM;
 569                 sess->auth.algo = AES_GMAC;
 570
 571                 switch (xform->aead.key.length) {
 572                 case AES_128_BYTES:
 573                         sess->cipher.key_length_in_bytes = AES_128_BYTES;
 574                         IMB_AES128_GCM_PRE(mb_mgr, xform->aead.key.data,
 575                                 &sess->cipher.gcm_key);
 576                         break;
 577                 case AES_192_BYTES:
 578                         sess->cipher.key_length_in_bytes = AES_192_BYTES;
 579                         IMB_AES192_GCM_PRE(mb_mgr, xform->aead.key.data,
 580                                 &sess->cipher.gcm_key);
 581                         break;
 582                 case AES_256_BYTES:
 583                         sess->cipher.key_length_in_bytes = AES_256_BYTES;
 584                         IMB_AES256_GCM_PRE(mb_mgr, xform->aead.key.data,
 585                                 &sess->cipher.gcm_key);
 586                         break;
 587                 default:
 588                         AESNI_MB_LOG(ERR, "Invalid cipher key length");
 589                         return -EINVAL;
 590                 }
 591
 592                 break;
 593
 594         default:
 595                 AESNI_MB_LOG(ERR, "Unsupported aead mode parameter");
 596                 return -ENOTSUP;
 597         }
 598
 599         /* Set IV parameters */
 600         sess->iv.offset = xform->aead.iv.offset;
 601         sess->iv.length = xform->aead.iv.length;
 602
 603         sess->auth.req_digest_len = xform->aead.digest_length;
 604         /* CCM digests must be between 4 and 16 and an even number */
 605         if (sess->auth.req_digest_len < AES_CCM_DIGEST_MIN_LEN ||
 606                         sess->auth.req_digest_len > AES_CCM_DIGEST_MAX_LEN ||
 607                         (sess->auth.req_digest_len & 1) == 1) {
 608                 AESNI_MB_LOG(ERR, "Invalid digest size\n");
 609                 return -EINVAL;
 610         }
 611         sess->auth.gen_digest_len = sess->auth.req_digest_len;
 612
 613         return 0;
 614 }
 615
 616 /** Parse crypto xform chain and set private session parameters */
 617 int
 618 aesni_mb_set_session_parameters(const MB_MGR *mb_mgr,
 619                 struct aesni_mb_session *sess,
 620                 const struct rte_crypto_sym_xform *xform)
 621 {
 622         const struct rte_crypto_sym_xform *auth_xform = NULL;
 623         const struct rte_crypto_sym_xform *cipher_xform = NULL;
 624         const struct rte_crypto_sym_xform *aead_xform = NULL;
 625         int ret;
 626
 627         /* Select Crypto operation - hash then cipher / cipher then hash */
 628         switch (aesni_mb_get_chain_order(xform)) {
 629         case AESNI_MB_OP_HASH_CIPHER:
 630                 sess->chain_order = HASH_CIPHER;
 631                 auth_xform = xform;
 632                 cipher_xform = xform->next;
 633                 break;
 634         case AESNI_MB_OP_CIPHER_HASH:
 635                 sess->chain_order = CIPHER_HASH;
 636                 auth_xform = xform->next;
 637                 cipher_xform = xform;
 638                 break;
 639         case AESNI_MB_OP_HASH_ONLY:
 640                 sess->chain_order = HASH_CIPHER;
 641                 auth_xform = xform;
 642                 cipher_xform = NULL;
 643                 break;
 644         case AESNI_MB_OP_CIPHER_ONLY:
 645                 /*
 646                  * Multi buffer library operates only at two modes,
 647                  * CIPHER_HASH and HASH_CIPHER. When doing ciphering only,
 648                  * chain order depends on cipher operation: encryption is always
 649                  * the first operation and decryption the last one.
 650                  */
 651                 if (xform->cipher.op == RTE_CRYPTO_CIPHER_OP_ENCRYPT)
 652                         sess->chain_order = CIPHER_HASH;
 653                 else
 654                         sess->chain_order = HASH_CIPHER;
 655                 auth_xform = NULL;
 656                 cipher_xform = xform;
 657                 break;
 658         case AESNI_MB_OP_AEAD_CIPHER_HASH:
 659                 sess->chain_order = CIPHER_HASH;
 660                 sess->aead.aad_len = xform->aead.aad_length;
 661                 aead_xform = xform;
 662                 break;
 663         case AESNI_MB_OP_AEAD_HASH_CIPHER:
 664                 sess->chain_order = HASH_CIPHER;
 665                 sess->aead.aad_len = xform->aead.aad_length;
 666                 aead_xform = xform;
 667                 break;
 668         case AESNI_MB_OP_NOT_SUPPORTED:
 669         default:
 670                 AESNI_MB_LOG(ERR, "Unsupported operation chain order parameter");
 671                 return -ENOTSUP;
 672         }
 673
 674         /* Default IV length = 0 */
 675         sess->iv.length = 0;
 676
 677         ret = aesni_mb_set_session_auth_parameters(mb_mgr, sess, auth_xform);
 678         if (ret != 0) {
 679                 AESNI_MB_LOG(ERR, "Invalid/unsupported authentication parameters");
 680                 return ret;
 681         }
 682
 683         ret = aesni_mb_set_session_cipher_parameters(mb_mgr, sess,
 684                         cipher_xform);
 685         if (ret != 0) {
 686                 AESNI_MB_LOG(ERR, "Invalid/unsupported cipher parameters");
 687                 return ret;
 688         }
 689
 690         if (aead_xform) {
 691                 ret = aesni_mb_set_session_aead_parameters(mb_mgr, sess,
 692                                 aead_xform);
 693                 if (ret != 0) {
 694                         AESNI_MB_LOG(ERR, "Invalid/unsupported aead parameters");
 695                         return ret;
 696                 }
 697         }
 698
 699         return 0;
 700 }
 701
 702 /**
 703  * burst enqueue, place crypto operations on ingress queue for processing.
 704  *
 705  * @param __qp         Queue Pair to process
 706  * @param ops          Crypto operations for processing
 707  * @param nb_ops       Number of crypto operations for processing
 708  *
 709  * @return
 710  * - Number of crypto operations enqueued
 711  */
 712 static uint16_t
 713 aesni_mb_pmd_enqueue_burst(void *__qp, struct rte_crypto_op **ops,
 714                 uint16_t nb_ops)
 715 {
 716         struct aesni_mb_qp *qp = __qp;
 717
 718         unsigned int nb_enqueued;
 719
 720         nb_enqueued = rte_ring_enqueue_burst(qp->ingress_queue,
 721                         (void **)ops, nb_ops, NULL);
 722
 723         qp->stats.enqueued_count += nb_enqueued;
 724
 725         return nb_enqueued;
 726 }
 727
 728 /** Get multi buffer session */
 729 static inline struct aesni_mb_session *
 730 get_session(struct aesni_mb_qp *qp, struct rte_crypto_op *op)
 731 {
 732         struct aesni_mb_session *sess = NULL;
 733
 734         if (op->sess_type == RTE_CRYPTO_OP_WITH_SESSION) {
 735                 if (likely(op->sym->session != NULL))
 736                         sess = (struct aesni_mb_session *)
 737                                         get_sym_session_private_data(
 738                                         op->sym->session,
 739                                         cryptodev_driver_id);
 740         } else {
 741                 void *_sess = rte_cryptodev_sym_session_create(qp->sess_mp);
 742                 void *_sess_private_data = NULL;
 743
 744                 if (_sess == NULL)
 745                         return NULL;
 746
 747                 if (rte_mempool_get(qp->sess_mp_priv,
 748                                 (void **)&_sess_private_data))
 749                         return NULL;
 750
 751                 sess = (struct aesni_mb_session *)_sess_private_data;
 752
 753                 if (unlikely(aesni_mb_set_session_parameters(qp->mb_mgr,
 754                                 sess, op->sym->xform) != 0)) {
 755                         rte_mempool_put(qp->sess_mp, _sess);
 756                         rte_mempool_put(qp->sess_mp_priv, _sess_private_data);
 757                         sess = NULL;
 758                 }
 759                 op->sym->session = (struct rte_cryptodev_sym_session *)_sess;
 760                 set_sym_session_private_data(op->sym->session,
 761                                 cryptodev_driver_id, _sess_private_data);
 762         }
 763
 764         if (unlikely(sess == NULL))
 765                 op->status = RTE_CRYPTO_OP_STATUS_INVALID_SESSION;
 766
 767         return sess;
 768 }
 769
 770 static inline uint64_t
 771 auth_start_offset(struct rte_crypto_op *op, struct aesni_mb_session *session,
 772                 uint32_t oop)
 773 {
 774         struct rte_mbuf *m_src, *m_dst;
 775         uint8_t *p_src, *p_dst;
 776         uintptr_t u_src, u_dst;
 777         uint32_t cipher_end, auth_end;
 778
 779         /* Only cipher then hash needs special calculation. */
 780         if (!oop || session->chain_order != CIPHER_HASH)
 781                 return op->sym->auth.data.offset;
 782
 783         m_src = op->sym->m_src;
 784         m_dst = op->sym->m_dst;
 785
 786         p_src = rte_pktmbuf_mtod(m_src, uint8_t *);
 787         p_dst = rte_pktmbuf_mtod(m_dst, uint8_t *);
 788         u_src = (uintptr_t)p_src;
 789         u_dst = (uintptr_t)p_dst + op->sym->auth.data.offset;
 790
 791         /**
 792          * Copy the content between cipher offset and auth offset for generating
 793          * correct digest.
 794          */
 795         if (op->sym->cipher.data.offset > op->sym->auth.data.offset)
 796                 memcpy(p_dst + op->sym->auth.data.offset,
 797                                 p_src + op->sym->auth.data.offset,
 798                                 op->sym->cipher.data.offset -
 799                                 op->sym->auth.data.offset);
 800
 801         /**
 802          * Copy the content between (cipher offset + length) and (auth offset +
 803          * length) for generating correct digest
 804          */
 805         cipher_end = op->sym->cipher.data.offset + op->sym->cipher.data.length;
 806         auth_end = op->sym->auth.data.offset + op->sym->auth.data.length;
 807         if (cipher_end < auth_end)
 808                 memcpy(p_dst + cipher_end, p_src + cipher_end,
 809                                 auth_end - cipher_end);
 810
 811         /**
 812          * Since intel-ipsec-mb only supports positive values,
 813          * we need to deduct the correct offset between src and dst.
 814          */
 815
 816         return u_src < u_dst ? (u_dst - u_src) :
 817                         (UINT64_MAX - u_src + u_dst + 1);
 818 }
 819
 820 static inline void
 821 set_cpu_mb_job_params(JOB_AES_HMAC *job, struct aesni_mb_session *session,
 822                 union rte_crypto_sym_ofs sofs, void *buf, uint32_t len,
 823                 void *iv, void *aad, void *digest, void *udata)
 824 {
 825         /* Set crypto operation */
 826         job->chain_order = session->chain_order;
 827
 828         /* Set cipher parameters */
 829         job->cipher_direction = session->cipher.direction;
 830         job->cipher_mode = session->cipher.mode;
 831
 832         job->aes_key_len_in_bytes = session->cipher.key_length_in_bytes;
 833
 834         /* Set authentication parameters */
 835         job->hash_alg = session->auth.algo;
 836         job->iv = iv;
 837
 838         switch (job->hash_alg) {
 839         case AES_XCBC:
 840                 job->u.XCBC._k1_expanded = session->auth.xcbc.k1_expanded;
 841                 job->u.XCBC._k2 = session->auth.xcbc.k2;
 842                 job->u.XCBC._k3 = session->auth.xcbc.k3;
 843
 844                 job->aes_enc_key_expanded =
 845                                 session->cipher.expanded_aes_keys.encode;
 846                 job->aes_dec_key_expanded =
 847                                 session->cipher.expanded_aes_keys.decode;
 848                 break;
 849
 850         case AES_CCM:
 851                 job->u.CCM.aad = (uint8_t *)aad + 18;
 852                 job->u.CCM.aad_len_in_bytes = session->aead.aad_len;
 853                 job->aes_enc_key_expanded =
 854                                 session->cipher.expanded_aes_keys.encode;
 855                 job->aes_dec_key_expanded =
 856                                 session->cipher.expanded_aes_keys.decode;
 857                 job->iv++;
 858                 break;
 859
 860         case AES_CMAC:
 861                 job->u.CMAC._key_expanded = session->auth.cmac.expkey;
 862                 job->u.CMAC._skey1 = session->auth.cmac.skey1;
 863                 job->u.CMAC._skey2 = session->auth.cmac.skey2;
 864                 job->aes_enc_key_expanded =
 865                                 session->cipher.expanded_aes_keys.encode;
 866                 job->aes_dec_key_expanded =
 867                                 session->cipher.expanded_aes_keys.decode;
 868                 break;
 869
 870         case AES_GMAC:
 871                 if (session->cipher.mode == GCM) {
 872                         job->u.GCM.aad = aad;
 873                         job->u.GCM.aad_len_in_bytes = session->aead.aad_len;
 874                 } else {
 875                         /* For GMAC */
 876                         job->u.GCM.aad = buf;
 877                         job->u.GCM.aad_len_in_bytes = len;
 878                         job->cipher_mode = GCM;
 879                 }
 880                 job->aes_enc_key_expanded = &session->cipher.gcm_key;
 881                 job->aes_dec_key_expanded = &session->cipher.gcm_key;
 882                 break;
 883
 884         default:
 885                 job->u.HMAC._hashed_auth_key_xor_ipad =
 886                                 session->auth.pads.inner;
 887                 job->u.HMAC._hashed_auth_key_xor_opad =
 888                                 session->auth.pads.outer;
 889
 890                 if (job->cipher_mode == DES3) {
 891                         job->aes_enc_key_expanded =
 892                                 session->cipher.exp_3des_keys.ks_ptr;
 893                         job->aes_dec_key_expanded =
 894                                 session->cipher.exp_3des_keys.ks_ptr;
 895                 } else {
 896                         job->aes_enc_key_expanded =
 897                                 session->cipher.expanded_aes_keys.encode;
 898                         job->aes_dec_key_expanded =
 899                                 session->cipher.expanded_aes_keys.decode;
 900                 }
 901         }
 902
 903         /*
 904          * Multi-buffer library current only support returning a truncated
 905          * digest length as specified in the relevant IPsec RFCs
 906          */
 907
 908         /* Set digest location and length */
 909         job->auth_tag_output = digest;
 910         job->auth_tag_output_len_in_bytes = session->auth.gen_digest_len;
 911
 912         /* Set IV parameters */
 913         job->iv_len_in_bytes = session->iv.length;
 914
 915         /* Data Parameters */
 916         job->src = buf;
 917         job->dst = (uint8_t *)buf + sofs.ofs.cipher.head;
 918         job->cipher_start_src_offset_in_bytes = sofs.ofs.cipher.head;
 919         job->hash_start_src_offset_in_bytes = sofs.ofs.auth.head;
 920         if (job->hash_alg == AES_GMAC && session->cipher.mode != GCM) {
 921                 job->msg_len_to_hash_in_bytes = 0;
 922                 job->msg_len_to_cipher_in_bytes = 0;
 923         } else {
 924                 job->msg_len_to_hash_in_bytes = len - sofs.ofs.auth.head -
 925                         sofs.ofs.auth.tail;
 926                 job->msg_len_to_cipher_in_bytes = len - sofs.ofs.cipher.head -
 927                         sofs.ofs.cipher.tail;
 928         }
 929
 930         job->user_data = udata;
 931 }
 932
 933 /**
 934  * Process a crypto operation and complete a JOB_AES_HMAC job structure for
 935  * submission to the multi buffer library for processing.
 936  *
 937  * @param       qp      queue pair
 938  * @param       job     JOB_AES_HMAC structure to fill
 939  * @param       m       mbuf to process
 940  *
 941  * @return
 942  * - Completed JOB_AES_HMAC structure pointer on success
 943  * - NULL pointer if completion of JOB_AES_HMAC structure isn't possible
 944  */
 945 static inline int
 946 set_mb_job_params(JOB_AES_HMAC *job, struct aesni_mb_qp *qp,
 947                 struct rte_crypto_op *op, uint8_t *digest_idx)
 948 {
 949         struct rte_mbuf *m_src = op->sym->m_src, *m_dst;
 950         struct aesni_mb_session *session;
 951         uint32_t m_offset, oop;
 952
 953         session = get_session(qp, op);
 954         if (session == NULL) {
 955                 op->status = RTE_CRYPTO_OP_STATUS_INVALID_SESSION;
 956                 return -1;
 957         }
 958
 959         /* Set crypto operation */
 960         job->chain_order = session->chain_order;
 961
 962         /* Set cipher parameters */
 963         job->cipher_direction = session->cipher.direction;
 964         job->cipher_mode = session->cipher.mode;
 965
 966         job->aes_key_len_in_bytes = session->cipher.key_length_in_bytes;
 967
 968         /* Set authentication parameters */
 969         job->hash_alg = session->auth.algo;
 970
 971         switch (job->hash_alg) {
 972         case AES_XCBC:
 973                 job->u.XCBC._k1_expanded = session->auth.xcbc.k1_expanded;
 974                 job->u.XCBC._k2 = session->auth.xcbc.k2;
 975                 job->u.XCBC._k3 = session->auth.xcbc.k3;
 976
 977                 job->aes_enc_key_expanded =
 978                                 session->cipher.expanded_aes_keys.encode;
 979                 job->aes_dec_key_expanded =
 980                                 session->cipher.expanded_aes_keys.decode;
 981                 break;
 982
 983         case AES_CCM:
 984                 job->u.CCM.aad = op->sym->aead.aad.data + 18;
 985                 job->u.CCM.aad_len_in_bytes = session->aead.aad_len;
 986                 job->aes_enc_key_expanded =
 987                                 session->cipher.expanded_aes_keys.encode;
 988                 job->aes_dec_key_expanded =
 989                                 session->cipher.expanded_aes_keys.decode;
 990                 break;
 991
 992         case AES_CMAC:
 993                 job->u.CMAC._key_expanded = session->auth.cmac.expkey;
 994                 job->u.CMAC._skey1 = session->auth.cmac.skey1;
 995                 job->u.CMAC._skey2 = session->auth.cmac.skey2;
 996                 job->aes_enc_key_expanded =
 997                                 session->cipher.expanded_aes_keys.encode;
 998                 job->aes_dec_key_expanded =
 999                                 session->cipher.expanded_aes_keys.decode;
1000                 break;
1001
1002         case AES_GMAC:
1003                 if (session->cipher.mode == GCM) {
1004                         job->u.GCM.aad = op->sym->aead.aad.data;
1005                         job->u.GCM.aad_len_in_bytes = session->aead.aad_len;
1006                 } else {
1007                         /* For GMAC */
1008                         job->u.GCM.aad = rte_pktmbuf_mtod_offset(m_src,
1009                                         uint8_t *, op->sym->auth.data.offset);
1010                         job->u.GCM.aad_len_in_bytes = op->sym->auth.data.length;
1011                         job->cipher_mode = GCM;
1012                 }
1013                 job->aes_enc_key_expanded = &session->cipher.gcm_key;
1014                 job->aes_dec_key_expanded = &session->cipher.gcm_key;
1015                 break;
1016
1017         default:
1018                 job->u.HMAC._hashed_auth_key_xor_ipad = session->auth.pads.inner;
1019                 job->u.HMAC._hashed_auth_key_xor_opad = session->auth.pads.outer;
1020
1021                 if (job->cipher_mode == DES3) {
1022                         job->aes_enc_key_expanded =
1023                                 session->cipher.exp_3des_keys.ks_ptr;
1024                         job->aes_dec_key_expanded =
1025                                 session->cipher.exp_3des_keys.ks_ptr;
1026                 } else {
1027                         job->aes_enc_key_expanded =
1028                                 session->cipher.expanded_aes_keys.encode;
1029                         job->aes_dec_key_expanded =
1030                                 session->cipher.expanded_aes_keys.decode;
1031                 }
1032         }
1033
1034         if (!op->sym->m_dst) {
1035                 /* in-place operation */
1036                 m_dst = m_src;
1037                 oop = 0;
1038         } else if (op->sym->m_dst == op->sym->m_src) {
1039                 /* in-place operation */
1040                 m_dst = m_src;
1041                 oop = 0;
1042         } else {
1043                 /* out-of-place operation */
1044                 m_dst = op->sym->m_dst;
1045                 oop = 1;
1046         }
1047
1048         if (job->hash_alg == AES_CCM || (job->hash_alg == AES_GMAC &&
1049                         session->cipher.mode == GCM))
1050                 m_offset = op->sym->aead.data.offset;
1051         else
1052                 m_offset = op->sym->cipher.data.offset;
1053
1054         /* Set digest output location */
1055         if (job->hash_alg != NULL_HASH &&
1056                         session->auth.operation == RTE_CRYPTO_AUTH_OP_VERIFY) {
1057                 job->auth_tag_output = qp->temp_digests[*digest_idx];
1058                 *digest_idx = (*digest_idx + 1) % MAX_JOBS;
1059         } else {
1060                 if (job->hash_alg == AES_CCM || (job->hash_alg == AES_GMAC &&
1061                                 session->cipher.mode == GCM))
1062                         job->auth_tag_output = op->sym->aead.digest.data;
1063                 else
1064                         job->auth_tag_output = op->sym->auth.digest.data;
1065
1066                 if (session->auth.req_digest_len != session->auth.gen_digest_len) {
1067                         job->auth_tag_output = qp->temp_digests[*digest_idx];
1068                         *digest_idx = (*digest_idx + 1) % MAX_JOBS;
1069                 }
1070         }
1071         /*
1072          * Multi-buffer library current only support returning a truncated
1073          * digest length as specified in the relevant IPsec RFCs
1074          */
1075
1076         /* Set digest length */
1077         job->auth_tag_output_len_in_bytes = session->auth.gen_digest_len;
1078
1079         /* Set IV parameters */
1080         job->iv_len_in_bytes = session->iv.length;
1081
1082         /* Data Parameters */
1083         job->src = rte_pktmbuf_mtod(m_src, uint8_t *);
1084         job->dst = rte_pktmbuf_mtod_offset(m_dst, uint8_t *, m_offset);
1085
1086         switch (job->hash_alg) {
1087         case AES_CCM:
1088                 job->cipher_start_src_offset_in_bytes =
1089                                 op->sym->aead.data.offset;
1090                 job->msg_len_to_cipher_in_bytes = op->sym->aead.data.length;
1091                 job->hash_start_src_offset_in_bytes = op->sym->aead.data.offset;
1092                 job->msg_len_to_hash_in_bytes = op->sym->aead.data.length;
1093
1094                 job->iv = rte_crypto_op_ctod_offset(op, uint8_t *,
1095                         session->iv.offset + 1);
1096                 break;
1097
1098         case AES_GMAC:
1099                 if (session->cipher.mode == GCM) {
1100                         job->cipher_start_src_offset_in_bytes =
1101                                         op->sym->aead.data.offset;
1102                         job->hash_start_src_offset_in_bytes =
1103                                         op->sym->aead.data.offset;
1104                         job->msg_len_to_cipher_in_bytes =
1105                                         op->sym->aead.data.length;
1106                         job->msg_len_to_hash_in_bytes =
1107                                         op->sym->aead.data.length;
1108                 } else {
1109                         job->cipher_start_src_offset_in_bytes =
1110                                         op->sym->auth.data.offset;
1111                         job->hash_start_src_offset_in_bytes =
1112                                         op->sym->auth.data.offset;
1113                         job->msg_len_to_cipher_in_bytes = 0;
1114                         job->msg_len_to_hash_in_bytes = 0;
1115                 }
1116
1117                 job->iv = rte_crypto_op_ctod_offset(op, uint8_t *,
1118                                 session->iv.offset);
1119                 break;
1120
1121         default:
1122                 job->cipher_start_src_offset_in_bytes =
1123                                 op->sym->cipher.data.offset;
1124                 job->msg_len_to_cipher_in_bytes = op->sym->cipher.data.length;
1125
1126                 job->hash_start_src_offset_in_bytes = auth_start_offset(op,
1127                                 session, oop);
1128                 job->msg_len_to_hash_in_bytes = op->sym->auth.data.length;
1129
1130                 job->iv = rte_crypto_op_ctod_offset(op, uint8_t *,
1131                         session->iv.offset);
1132         }
1133
1134         /* Set user data to be crypto operation data struct */
1135         job->user_data = op;
1136
1137         return 0;
1138 }
1139
1140 static inline void
1141 verify_digest(JOB_AES_HMAC *job, void *digest, uint16_t len, uint8_t *status)
1142 {
1143         /* Verify digest if required */
1144         if (memcmp(job->auth_tag_output, digest, len) != 0)
1145                 *status = RTE_CRYPTO_OP_STATUS_AUTH_FAILED;
1146 }
1147
1148 static inline void
1149 generate_digest(JOB_AES_HMAC *job, struct rte_crypto_op *op,
1150                 struct aesni_mb_session *sess)
1151 {
1152         /* No extra copy needed */
1153         if (likely(sess->auth.req_digest_len == sess->auth.gen_digest_len))
1154                 return;
1155
1156         /*
1157          * This can only happen for HMAC, so only digest
1158          * for authentication algos is required
1159          */
1160         memcpy(op->sym->auth.digest.data, job->auth_tag_output,
1161                         sess->auth.req_digest_len);
1162 }
1163
1164 /**
1165  * Process a completed job and return rte_mbuf which job processed
1166  *
1167  * @param qp            Queue Pair to process
1168  * @param job   JOB_AES_HMAC job to process
1169  *
1170  * @return
1171  * - Returns processed crypto operation.
1172  * - Returns NULL on invalid job
1173  */
1174 static inline struct rte_crypto_op *
1175 post_process_mb_job(struct aesni_mb_qp *qp, JOB_AES_HMAC *job)
1176 {
1177         struct rte_crypto_op *op = (struct rte_crypto_op *)job->user_data;
1178         struct aesni_mb_session *sess = get_sym_session_private_data(
1179                                                         op->sym->session,
1180                                                         cryptodev_driver_id);
1181         if (unlikely(sess == NULL)) {
1182                 op->status = RTE_CRYPTO_OP_STATUS_INVALID_SESSION;
1183                 return op;
1184         }
1185
1186         if (likely(op->status == RTE_CRYPTO_OP_STATUS_NOT_PROCESSED)) {
1187                 switch (job->status) {
1188                 case STS_COMPLETED:
1189                         op->status = RTE_CRYPTO_OP_STATUS_SUCCESS;
1190
1191                         if (job->hash_alg == NULL_HASH)
1192                                 break;
1193
1194                         if (sess->auth.operation == RTE_CRYPTO_AUTH_OP_VERIFY) {
1195                                 if (job->hash_alg == AES_CCM ||
1196                                         (job->hash_alg == AES_GMAC &&
1197                                                 sess->cipher.mode == GCM))
1198                                         verify_digest(job,
1199                                                 op->sym->aead.digest.data,
1200                                                 sess->auth.req_digest_len,
1201                                                 &op->status);
1202                                 else
1203                                         verify_digest(job,
1204                                                 op->sym->auth.digest.data,
1205                                                 sess->auth.req_digest_len,
1206                                                 &op->status);
1207                         } else
1208                                 generate_digest(job, op, sess);
1209                         break;
1210                 default:
1211                         op->status = RTE_CRYPTO_OP_STATUS_ERROR;
1212                 }
1213         }
1214
1215         /* Free session if a session-less crypto op */
1216         if (op->sess_type == RTE_CRYPTO_OP_SESSIONLESS) {
1217                 memset(sess, 0, sizeof(struct aesni_mb_session));
1218                 memset(op->sym->session, 0,
1219                         rte_cryptodev_sym_get_existing_header_session_size(
1220                                 op->sym->session));
1221                 rte_mempool_put(qp->sess_mp_priv, sess);
1222                 rte_mempool_put(qp->sess_mp, op->sym->session);
1223                 op->sym->session = NULL;
1224         }
1225
1226         return op;
1227 }
1228
1229 static inline void
1230 post_process_mb_sync_job(JOB_AES_HMAC *job)
1231 {
1232         uint32_t *st;
1233
1234         st = job->user_data;
1235         st[0] = (job->status == STS_COMPLETED) ? 0 : EBADMSG;
1236 }
1237
1238 /**
1239  * Process a completed JOB_AES_HMAC job and keep processing jobs until
1240  * get_completed_job return NULL
1241  *
1242  * @param qp            Queue Pair to process
1243  * @param job           JOB_AES_HMAC job
1244  *
1245  * @return
1246  * - Number of processed jobs
1247  */
1248 static unsigned
1249 handle_completed_jobs(struct aesni_mb_qp *qp, JOB_AES_HMAC *job,
1250                 struct rte_crypto_op **ops, uint16_t nb_ops)
1251 {
1252         struct rte_crypto_op *op = NULL;
1253         unsigned processed_jobs = 0;
1254
1255         while (job != NULL) {
1256                 op = post_process_mb_job(qp, job);
1257
1258                 if (op) {
1259                         ops[processed_jobs++] = op;
1260                         qp->stats.dequeued_count++;
1261                 } else {
1262                         qp->stats.dequeue_err_count++;
1263                         break;
1264                 }
1265                 if (processed_jobs == nb_ops)
1266                         break;
1267
1268                 job = IMB_GET_COMPLETED_JOB(qp->mb_mgr);
1269         }
1270
1271         return processed_jobs;
1272 }
1273
1274 static inline uint32_t
1275 handle_completed_sync_jobs(JOB_AES_HMAC *job, MB_MGR *mb_mgr)
1276 {
1277         uint32_t i;
1278
1279         for (i = 0; job != NULL; i++, job = IMB_GET_COMPLETED_JOB(mb_mgr))
1280                 post_process_mb_sync_job(job);
1281
1282         return i;
1283 }
1284
1285 static inline uint32_t
1286 flush_mb_sync_mgr(MB_MGR *mb_mgr)
1287 {
1288         JOB_AES_HMAC *job;
1289
1290         job = IMB_FLUSH_JOB(mb_mgr);
1291         return handle_completed_sync_jobs(job, mb_mgr);
1292 }
1293
1294 static inline uint16_t
1295 flush_mb_mgr(struct aesni_mb_qp *qp, struct rte_crypto_op **ops,
1296                 uint16_t nb_ops)
1297 {
1298         int processed_ops = 0;
1299
1300         /* Flush the remaining jobs */
1301         JOB_AES_HMAC *job = IMB_FLUSH_JOB(qp->mb_mgr);
1302
1303         if (job)
1304                 processed_ops += handle_completed_jobs(qp, job,
1305                                 &ops[processed_ops], nb_ops - processed_ops);
1306
1307         return processed_ops;
1308 }
1309
1310 static inline JOB_AES_HMAC *
1311 set_job_null_op(JOB_AES_HMAC *job, struct rte_crypto_op *op)
1312 {
1313         job->chain_order = HASH_CIPHER;
1314         job->cipher_mode = NULL_CIPHER;
1315         job->hash_alg = NULL_HASH;
1316         job->cipher_direction = DECRYPT;
1317
1318         /* Set user data to be crypto operation data struct */
1319         job->user_data = op;
1320
1321         return job;
1322 }
1323
1324 static uint16_t
1325 aesni_mb_pmd_dequeue_burst(void *queue_pair, struct rte_crypto_op **ops,
1326                 uint16_t nb_ops)
1327 {
1328         struct aesni_mb_qp *qp = queue_pair;
1329
1330         struct rte_crypto_op *op;
1331         JOB_AES_HMAC *job;
1332
1333         int retval, processed_jobs = 0;
1334
1335         if (unlikely(nb_ops == 0))
1336                 return 0;
1337
1338         uint8_t digest_idx = qp->digest_idx;
1339         do {
1340                 /* Get next free mb job struct from mb manager */
1341                 job = IMB_GET_NEXT_JOB(qp->mb_mgr);
1342                 if (unlikely(job == NULL)) {
1343                         /* if no free mb job structs we need to flush mb_mgr */
1344                         processed_jobs += flush_mb_mgr(qp,
1345                                         &ops[processed_jobs],
1346                                         nb_ops - processed_jobs);
1347
1348                         if (nb_ops == processed_jobs)
1349                                 break;
1350
1351                         job = IMB_GET_NEXT_JOB(qp->mb_mgr);
1352                 }
1353
1354                 /*
1355                  * Get next operation to process from ingress queue.
1356                  * There is no need to return the job to the MB_MGR
1357                  * if there are no more operations to process, since the MB_MGR
1358                  * can use that pointer again in next get_next calls.
1359                  */
1360                 retval = rte_ring_dequeue(qp->ingress_queue, (void **)&op);
1361                 if (retval < 0)
1362                         break;
1363
1364                 retval = set_mb_job_params(job, qp, op, &digest_idx);
1365                 if (unlikely(retval != 0)) {
1366                         qp->stats.dequeue_err_count++;
1367                         set_job_null_op(job, op);
1368                 }
1369
1370                 /* Submit job to multi-buffer for processing */
1371 #ifdef RTE_LIBRTE_PMD_AESNI_MB_DEBUG
1372                 job = IMB_SUBMIT_JOB(qp->mb_mgr);
1373 #else
1374                 job = IMB_SUBMIT_JOB_NOCHECK(qp->mb_mgr);
1375 #endif
1376                 /*
1377                  * If submit returns a processed job then handle it,
1378                  * before submitting subsequent jobs
1379                  */
1380                 if (job)
1381                         processed_jobs += handle_completed_jobs(qp, job,
1382                                         &ops[processed_jobs],
1383                                         nb_ops - processed_jobs);
1384
1385         } while (processed_jobs < nb_ops);
1386
1387         qp->digest_idx = digest_idx;
1388
1389         if (processed_jobs < 1)
1390                 processed_jobs += flush_mb_mgr(qp,
1391                                 &ops[processed_jobs],
1392                                 nb_ops - processed_jobs);
1393
1394         return processed_jobs;
1395 }
1396
1397 static MB_MGR *
1398 alloc_init_mb_mgr(enum aesni_mb_vector_mode vector_mode)
1399 {
1400         MB_MGR *mb_mgr = alloc_mb_mgr(0);
1401         if (mb_mgr == NULL)
1402                 return NULL;
1403
1404         switch (vector_mode) {
1405         case RTE_AESNI_MB_SSE:
1406                 init_mb_mgr_sse(mb_mgr);
1407                 break;
1408         case RTE_AESNI_MB_AVX:
1409                 init_mb_mgr_avx(mb_mgr);
1410                 break;
1411         case RTE_AESNI_MB_AVX2:
1412                 init_mb_mgr_avx2(mb_mgr);
1413                 break;
1414         case RTE_AESNI_MB_AVX512:
1415                 init_mb_mgr_avx512(mb_mgr);
1416                 break;
1417         default:
1418                 AESNI_MB_LOG(ERR, "Unsupported vector mode %u\n", vector_mode);
1419                 free_mb_mgr(mb_mgr);
1420                 return NULL;
1421         }
1422
1423         return mb_mgr;
1424 }
1425
1426 static inline void
1427 aesni_mb_fill_error_code(struct rte_crypto_sym_vec *vec, int32_t err)
1428 {
1429         uint32_t i;
1430
1431         for (i = 0; i != vec->num; ++i)
1432                 vec->status[i] = err;
1433 }
1434
1435 static inline int
1436 check_crypto_sgl(union rte_crypto_sym_ofs so, const struct rte_crypto_sgl *sgl)
1437 {
1438         /* no multi-seg support with current AESNI-MB PMD */
1439         if (sgl->num != 1)
1440                 return ENOTSUP;
1441         else if (so.ofs.cipher.head + so.ofs.cipher.tail > sgl->vec[0].len)
1442                 return EINVAL;
1443         return 0;
1444 }
1445
1446 static inline JOB_AES_HMAC *
1447 submit_sync_job(MB_MGR *mb_mgr)
1448 {
1449 #ifdef RTE_LIBRTE_PMD_AESNI_MB_DEBUG
1450         return IMB_SUBMIT_JOB(mb_mgr);
1451 #else
1452         return IMB_SUBMIT_JOB_NOCHECK(mb_mgr);
1453 #endif
1454 }
1455
1456 static inline uint32_t
1457 generate_sync_dgst(struct rte_crypto_sym_vec *vec,
1458         const uint8_t dgst[][DIGEST_LENGTH_MAX], uint32_t len)
1459 {
1460         uint32_t i, k;
1461
1462         for (i = 0, k = 0; i != vec->num; i++) {
1463                 if (vec->status[i] == 0) {
1464                         memcpy(vec->digest[i], dgst[i], len);
1465                         k++;
1466                 }
1467         }
1468
1469         return k;
1470 }
1471
1472 static inline uint32_t
1473 verify_sync_dgst(struct rte_crypto_sym_vec *vec,
1474         const uint8_t dgst[][DIGEST_LENGTH_MAX], uint32_t len)
1475 {
1476         uint32_t i, k;
1477
1478         for (i = 0, k = 0; i != vec->num; i++) {
1479                 if (vec->status[i] == 0) {
1480                         if (memcmp(vec->digest[i], dgst[i], len) != 0)
1481                                 vec->status[i] = EBADMSG;
1482                         else
1483                                 k++;
1484                 }
1485         }
1486
1487         return k;
1488 }
1489
1490 uint32_t
1491 aesni_mb_cpu_crypto_process_bulk(struct rte_cryptodev *dev,
1492         struct rte_cryptodev_sym_session *sess, union rte_crypto_sym_ofs sofs,
1493         struct rte_crypto_sym_vec *vec)
1494 {
1495         int32_t ret;
1496         uint32_t i, j, k, len;
1497         void *buf;
1498         JOB_AES_HMAC *job;
1499         MB_MGR *mb_mgr;
1500         struct aesni_mb_private *priv;
1501         struct aesni_mb_session *s;
1502         uint8_t tmp_dgst[vec->num][DIGEST_LENGTH_MAX];
1503
1504         s = get_sym_session_private_data(sess, dev->driver_id);
1505         if (s == NULL) {
1506                 aesni_mb_fill_error_code(vec, EINVAL);
1507                 return 0;
1508         }
1509
1510         /* get per-thread MB MGR, create one if needed */
1511         mb_mgr = RTE_PER_LCORE(sync_mb_mgr);
1512         if (mb_mgr == NULL) {
1513
1514                 priv = dev->data->dev_private;
1515                 mb_mgr = alloc_init_mb_mgr(priv->vector_mode);
1516                 if (mb_mgr == NULL) {
1517                         aesni_mb_fill_error_code(vec, ENOMEM);
1518                         return 0;
1519                 }
1520                 RTE_PER_LCORE(sync_mb_mgr) = mb_mgr;
1521         }
1522
1523         for (i = 0, j = 0, k = 0; i != vec->num; i++) {
1524
1525
1526                 ret = check_crypto_sgl(sofs, vec->sgl + i);
1527                 if (ret != 0) {
1528                         vec->status[i] = ret;
1529                         continue;
1530                 }
1531
1532                 buf = vec->sgl[i].vec[0].base;
1533                 len = vec->sgl[i].vec[0].len;
1534
1535                 job = IMB_GET_NEXT_JOB(mb_mgr);
1536                 if (job == NULL) {
1537                         k += flush_mb_sync_mgr(mb_mgr);
1538                         job = IMB_GET_NEXT_JOB(mb_mgr);
1539                         RTE_ASSERT(job != NULL);
1540                 }
1541
1542                 /* Submit job for processing */
1543                 set_cpu_mb_job_params(job, s, sofs, buf, len,
1544                         vec->iv[i], vec->aad[i], tmp_dgst[i],
1545                         &vec->status[i]);
1546                 job = submit_sync_job(mb_mgr);
1547                 j++;
1548
1549                 /* handle completed jobs */
1550                 k += handle_completed_sync_jobs(job, mb_mgr);
1551         }
1552
1553         /* flush remaining jobs */
1554         while (k != j)
1555                 k += flush_mb_sync_mgr(mb_mgr);
1556
1557         /* finish processing for successful jobs: check/update digest */
1558         if (k != 0) {
1559                 if (s->auth.operation == RTE_CRYPTO_AUTH_OP_VERIFY)
1560                         k = verify_sync_dgst(vec,
1561                                 (const uint8_t (*)[DIGEST_LENGTH_MAX])tmp_dgst,
1562                                 s->auth.req_digest_len);
1563                 else
1564                         k = generate_sync_dgst(vec,
1565                                 (const uint8_t (*)[DIGEST_LENGTH_MAX])tmp_dgst,
1566                                 s->auth.req_digest_len);
1567         }
1568
1569         return k;
1570 }
1571
1572 static int cryptodev_aesni_mb_remove(struct rte_vdev_device *vdev);
1573
1574 static uint64_t
1575 vec_mode_to_flags(enum aesni_mb_vector_mode mode)
1576 {
1577         switch (mode) {
1578         case RTE_AESNI_MB_SSE:
1579                 return RTE_CRYPTODEV_FF_CPU_SSE;
1580         case RTE_AESNI_MB_AVX:
1581                 return RTE_CRYPTODEV_FF_CPU_AVX;
1582         case RTE_AESNI_MB_AVX2:
1583                 return RTE_CRYPTODEV_FF_CPU_AVX2;
1584         case RTE_AESNI_MB_AVX512:
1585                 return RTE_CRYPTODEV_FF_CPU_AVX512;
1586         default:
1587                 AESNI_MB_LOG(ERR, "Unsupported vector mode %u\n", mode);
1588                 return 0;
1589         }
1590 }
1591
1592 static int
1593 cryptodev_aesni_mb_create(const char *name,
1594                         struct rte_vdev_device *vdev,
1595                         struct rte_cryptodev_pmd_init_params *init_params)
1596 {
1597         struct rte_cryptodev *dev;
1598         struct aesni_mb_private *internals;
1599         enum aesni_mb_vector_mode vector_mode;
1600         MB_MGR *mb_mgr;
1601
1602         dev = rte_cryptodev_pmd_create(name, &vdev->device, init_params);
1603         if (dev == NULL) {
1604                 AESNI_MB_LOG(ERR, "failed to create cryptodev vdev");
1605                 return -ENODEV;
1606         }
1607
1608         /* Check CPU for supported vector instruction set */
1609         if (rte_cpu_get_flag_enabled(RTE_CPUFLAG_AVX512F))
1610                 vector_mode = RTE_AESNI_MB_AVX512;
1611         else if (rte_cpu_get_flag_enabled(RTE_CPUFLAG_AVX2))
1612                 vector_mode = RTE_AESNI_MB_AVX2;
1613         else if (rte_cpu_get_flag_enabled(RTE_CPUFLAG_AVX))
1614                 vector_mode = RTE_AESNI_MB_AVX;
1615         else
1616                 vector_mode = RTE_AESNI_MB_SSE;
1617
1618         dev->driver_id = cryptodev_driver_id;
1619         dev->dev_ops = rte_aesni_mb_pmd_ops;
1620
1621         /* register rx/tx burst functions for data path */
1622         dev->dequeue_burst = aesni_mb_pmd_dequeue_burst;
1623         dev->enqueue_burst = aesni_mb_pmd_enqueue_burst;
1624
1625         dev->feature_flags = RTE_CRYPTODEV_FF_SYMMETRIC_CRYPTO |
1626                         RTE_CRYPTODEV_FF_SYM_OPERATION_CHAINING |
1627                         RTE_CRYPTODEV_FF_OOP_LB_IN_LB_OUT |
1628                         RTE_CRYPTODEV_FF_SYM_CPU_CRYPTO |
1629                         RTE_CRYPTODEV_FF_SYM_SESSIONLESS;
1630
1631         /* Check CPU for support for AES instruction set */
1632         if (rte_cpu_get_flag_enabled(RTE_CPUFLAG_AES))
1633                 dev->feature_flags |= RTE_CRYPTODEV_FF_CPU_AESNI;
1634         else
1635                 AESNI_MB_LOG(WARNING, "AES instructions not supported by CPU");
1636
1637         dev->feature_flags |= vec_mode_to_flags(vector_mode);
1638
1639         mb_mgr = alloc_init_mb_mgr(vector_mode);
1640         if (mb_mgr == NULL) {
1641                 rte_cryptodev_pmd_destroy(dev);
1642                 return -ENOMEM;
1643         }
1644
1645         /* Set vector instructions mode supported */
1646         internals = dev->data->dev_private;
1647
1648         internals->vector_mode = vector_mode;
1649         internals->max_nb_queue_pairs = init_params->max_nb_queue_pairs;
1650         internals->mb_mgr = mb_mgr;
1651
1652         AESNI_MB_LOG(INFO, "IPSec Multi-buffer library version used: %s\n",
1653                         imb_get_version_str());
1654         return 0;
1655 }
1656
1657 static int
1658 cryptodev_aesni_mb_probe(struct rte_vdev_device *vdev)
1659 {
1660         struct rte_cryptodev_pmd_init_params init_params = {
1661                 "",
1662                 sizeof(struct aesni_mb_private),
1663                 rte_socket_id(),
1664                 RTE_CRYPTODEV_PMD_DEFAULT_MAX_NB_QUEUE_PAIRS
1665         };
1666         const char *name, *args;
1667         int retval;
1668
1669         name = rte_vdev_device_name(vdev);
1670         if (name == NULL)
1671                 return -EINVAL;
1672
1673         args = rte_vdev_device_args(vdev);
1674
1675         retval = rte_cryptodev_pmd_parse_input_args(&init_params, args);
1676         if (retval) {
1677                 AESNI_MB_LOG(ERR, "Failed to parse initialisation arguments[%s]",
1678                                 args);
1679                 return -EINVAL;
1680         }
1681
1682         return cryptodev_aesni_mb_create(name, vdev, &init_params);
1683 }
1684
1685 static int
1686 cryptodev_aesni_mb_remove(struct rte_vdev_device *vdev)
1687 {
1688         struct rte_cryptodev *cryptodev;
1689         struct aesni_mb_private *internals;
1690         const char *name;
1691
1692         name = rte_vdev_device_name(vdev);
1693         if (name == NULL)
1694                 return -EINVAL;
1695
1696         cryptodev = rte_cryptodev_pmd_get_named_dev(name);
1697         if (cryptodev == NULL)
1698                 return -ENODEV;
1699
1700         internals = cryptodev->data->dev_private;
1701
1702         free_mb_mgr(internals->mb_mgr);
1703         if (RTE_PER_LCORE(sync_mb_mgr)) {
1704                 free_mb_mgr(RTE_PER_LCORE(sync_mb_mgr));
1705                 RTE_PER_LCORE(sync_mb_mgr) = NULL;
1706         }
1707
1708         return rte_cryptodev_pmd_destroy(cryptodev);
1709 }
1710
1711 static struct rte_vdev_driver cryptodev_aesni_mb_pmd_drv = {
1712         .probe = cryptodev_aesni_mb_probe,
1713         .remove = cryptodev_aesni_mb_remove
1714 };
1715
1716 static struct cryptodev_driver aesni_mb_crypto_drv;
1717
1718 RTE_PMD_REGISTER_VDEV(CRYPTODEV_NAME_AESNI_MB_PMD, cryptodev_aesni_mb_pmd_drv);
1719 RTE_PMD_REGISTER_ALIAS(CRYPTODEV_NAME_AESNI_MB_PMD, cryptodev_aesni_mb_pmd);
1720 RTE_PMD_REGISTER_PARAM_STRING(CRYPTODEV_NAME_AESNI_MB_PMD,
1721         "max_nb_queue_pairs=<int> "
1722         "socket_id=<int>");
1723 RTE_PMD_REGISTER_CRYPTO_DRIVER(aesni_mb_crypto_drv,
1724                 cryptodev_aesni_mb_pmd_drv.driver,
1725                 cryptodev_driver_id);
1726
1727 RTE_INIT(aesni_mb_init_log)
1728 {
1729         aesni_mb_logtype_driver = rte_log_register("pmd.crypto.aesni_mb");
1730 }