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