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