drivers/crypto/qce/aead.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2
   3 /*
   4  * Copyright (C) 2021, Linaro Limited. All rights reserved.
   5  */
   6 #include <linux/dma-mapping.h>
   7 #include <linux/interrupt.h>
   8 #include <crypto/gcm.h>
   9 #include <crypto/authenc.h>
  10 #include <crypto/internal/aead.h>
  11 #include <crypto/internal/des.h>
  12 #include <crypto/sha1.h>
  13 #include <crypto/sha2.h>
  14 #include <crypto/scatterwalk.h>
  15 #include "aead.h"
  16
  17 #define CCM_NONCE_ADATA_SHIFT           6
  18 #define CCM_NONCE_AUTHSIZE_SHIFT        3
  19 #define MAX_CCM_ADATA_HEADER_LEN        6
  20
  21 static LIST_HEAD(aead_algs);
  22
  23 static void qce_aead_done(void *data)
  24 {
  25         struct crypto_async_request *async_req = data;
  26         struct aead_request *req = aead_request_cast(async_req);
  27         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
  28         struct qce_aead_ctx *ctx = crypto_tfm_ctx(async_req->tfm);
  29         struct qce_alg_template *tmpl = to_aead_tmpl(crypto_aead_reqtfm(req));
  30         struct qce_device *qce = tmpl->qce;
  31         struct qce_result_dump *result_buf = qce->dma.result_buf;
  32         enum dma_data_direction dir_src, dir_dst;
  33         bool diff_dst;
  34         int error;
  35         u32 status;
  36         unsigned int totallen;
  37         unsigned char tag[SHA256_DIGEST_SIZE] = {0};
  38         int ret = 0;
  39
  40         diff_dst = (req->src != req->dst) ? true : false;
  41         dir_src = diff_dst ? DMA_TO_DEVICE : DMA_BIDIRECTIONAL;
  42         dir_dst = diff_dst ? DMA_FROM_DEVICE : DMA_BIDIRECTIONAL;
  43
  44         error = qce_dma_terminate_all(&qce->dma);
  45         if (error)
  46                 dev_dbg(qce->dev, "aead dma termination error (%d)\n",
  47                         error);
  48         if (diff_dst)
  49                 dma_unmap_sg(qce->dev, rctx->src_sg, rctx->src_nents, dir_src);
  50
  51         dma_unmap_sg(qce->dev, rctx->dst_sg, rctx->dst_nents, dir_dst);
  52
  53         if (IS_CCM(rctx->flags)) {
  54                 if (req->assoclen) {
  55                         sg_free_table(&rctx->src_tbl);
  56                         if (diff_dst)
  57                                 sg_free_table(&rctx->dst_tbl);
  58                 } else {
  59                         if (!(IS_DECRYPT(rctx->flags) && !diff_dst))
  60                                 sg_free_table(&rctx->dst_tbl);
  61                 }
  62         } else {
  63                 sg_free_table(&rctx->dst_tbl);
  64         }
  65
  66         error = qce_check_status(qce, &status);
  67         if (error < 0 && (error != -EBADMSG))
  68                 dev_err(qce->dev, "aead operation error (%x)\n", status);
  69
  70         if (IS_ENCRYPT(rctx->flags)) {
  71                 totallen = req->cryptlen + req->assoclen;
  72                 if (IS_CCM(rctx->flags))
  73                         scatterwalk_map_and_copy(rctx->ccmresult_buf, req->dst,
  74                                                  totallen, ctx->authsize, 1);
  75                 else
  76                         scatterwalk_map_and_copy(result_buf->auth_iv, req->dst,
  77                                                  totallen, ctx->authsize, 1);
  78
  79         } else if (!IS_CCM(rctx->flags)) {
  80                 totallen = req->cryptlen + req->assoclen - ctx->authsize;
  81                 scatterwalk_map_and_copy(tag, req->src, totallen, ctx->authsize, 0);
  82                 ret = memcmp(result_buf->auth_iv, tag, ctx->authsize);
  83                 if (ret) {
  84                         pr_err("Bad message error\n");
  85                          error = -EBADMSG;
  86                 }
  87         }
  88
  89         qce->async_req_done(qce, error);
  90 }
  91
  92 static struct scatterlist *
  93 qce_aead_prepare_result_buf(struct sg_table *tbl, struct aead_request *req)
  94 {
  95         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
  96         struct qce_alg_template *tmpl = to_aead_tmpl(crypto_aead_reqtfm(req));
  97         struct qce_device *qce = tmpl->qce;
  98
  99         sg_init_one(&rctx->result_sg, qce->dma.result_buf, QCE_RESULT_BUF_SZ);
 100         return qce_sgtable_add(tbl, &rctx->result_sg, QCE_RESULT_BUF_SZ);
 101 }
 102
 103 static struct scatterlist *
 104 qce_aead_prepare_ccm_result_buf(struct sg_table *tbl, struct aead_request *req)
 105 {
 106         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 107
 108         sg_init_one(&rctx->result_sg, rctx->ccmresult_buf, QCE_BAM_BURST_SIZE);
 109         return qce_sgtable_add(tbl, &rctx->result_sg, QCE_BAM_BURST_SIZE);
 110 }
 111
 112 static struct scatterlist *
 113 qce_aead_prepare_dst_buf(struct aead_request *req)
 114 {
 115         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 116         struct qce_alg_template *tmpl = to_aead_tmpl(crypto_aead_reqtfm(req));
 117         struct qce_device *qce = tmpl->qce;
 118         struct scatterlist *sg, *msg_sg, __sg[2];
 119         gfp_t gfp;
 120         unsigned int assoclen = req->assoclen;
 121         unsigned int totallen;
 122         int ret;
 123
 124         totallen = rctx->cryptlen + assoclen;
 125         rctx->dst_nents = sg_nents_for_len(req->dst, totallen);
 126         if (rctx->dst_nents < 0) {
 127                 dev_err(qce->dev, "Invalid numbers of dst SG.\n");
 128                 return ERR_PTR(-EINVAL);
 129         }
 130         if (IS_CCM(rctx->flags))
 131                 rctx->dst_nents += 2;
 132         else
 133                 rctx->dst_nents += 1;
 134
 135         gfp = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
 136                                                 GFP_KERNEL : GFP_ATOMIC;
 137         ret = sg_alloc_table(&rctx->dst_tbl, rctx->dst_nents, gfp);
 138         if (ret)
 139                 return ERR_PTR(ret);
 140
 141         if (IS_CCM(rctx->flags) && assoclen) {
 142                 /* Get the dst buffer */
 143                 msg_sg = scatterwalk_ffwd(__sg, req->dst, assoclen);
 144
 145                 sg = qce_sgtable_add(&rctx->dst_tbl, &rctx->adata_sg,
 146                                      rctx->assoclen);
 147                 if (IS_ERR(sg)) {
 148                         ret = PTR_ERR(sg);
 149                         goto dst_tbl_free;
 150                 }
 151                 /* dst buffer */
 152                 sg = qce_sgtable_add(&rctx->dst_tbl, msg_sg, rctx->cryptlen);
 153                 if (IS_ERR(sg)) {
 154                         ret = PTR_ERR(sg);
 155                         goto dst_tbl_free;
 156                 }
 157                 totallen = rctx->cryptlen + rctx->assoclen;
 158         } else {
 159                 if (totallen) {
 160                         sg = qce_sgtable_add(&rctx->dst_tbl, req->dst, totallen);
 161                         if (IS_ERR(sg))
 162                                 goto dst_tbl_free;
 163                 }
 164         }
 165         if (IS_CCM(rctx->flags))
 166                 sg = qce_aead_prepare_ccm_result_buf(&rctx->dst_tbl, req);
 167         else
 168                 sg = qce_aead_prepare_result_buf(&rctx->dst_tbl, req);
 169
 170         if (IS_ERR(sg))
 171                 goto dst_tbl_free;
 172
 173         sg_mark_end(sg);
 174         rctx->dst_sg = rctx->dst_tbl.sgl;
 175         rctx->dst_nents = sg_nents_for_len(rctx->dst_sg, totallen) + 1;
 176
 177         return sg;
 178
 179 dst_tbl_free:
 180         sg_free_table(&rctx->dst_tbl);
 181         return sg;
 182 }
 183
 184 static int
 185 qce_aead_ccm_prepare_buf_assoclen(struct aead_request *req)
 186 {
 187         struct scatterlist *sg, *msg_sg, __sg[2];
 188         struct crypto_aead *tfm = crypto_aead_reqtfm(req);
 189         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 190         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 191         unsigned int assoclen = rctx->assoclen;
 192         unsigned int adata_header_len, cryptlen, totallen;
 193         gfp_t gfp;
 194         bool diff_dst;
 195         int ret;
 196
 197         if (IS_DECRYPT(rctx->flags))
 198                 cryptlen = rctx->cryptlen + ctx->authsize;
 199         else
 200                 cryptlen = rctx->cryptlen;
 201         totallen = cryptlen + req->assoclen;
 202
 203         /* Get the msg */
 204         msg_sg = scatterwalk_ffwd(__sg, req->src, req->assoclen);
 205
 206         rctx->adata = kzalloc((ALIGN(assoclen, 16) + MAX_CCM_ADATA_HEADER_LEN) *
 207                                sizeof(unsigned char), GFP_ATOMIC);
 208         if (!rctx->adata)
 209                 return -ENOMEM;
 210
 211         /*
 212          * Format associated data (RFC3610 and NIST 800-38C)
 213          * Even though specification allows for AAD to be up to 2^64 - 1 bytes,
 214          * the assoclen field in aead_request is unsigned int and thus limits
 215          * the AAD to be up to 2^32 - 1 bytes. So we handle only two scenarios
 216          * while forming the header for AAD.
 217          */
 218         if (assoclen < 0xff00) {
 219                 adata_header_len = 2;
 220                 *(__be16 *)rctx->adata = cpu_to_be16(assoclen);
 221         } else {
 222                 adata_header_len = 6;
 223                 *(__be16 *)rctx->adata = cpu_to_be16(0xfffe);
 224                 *(__be32 *)(rctx->adata + 2) = cpu_to_be32(assoclen);
 225         }
 226
 227         /* Copy the associated data */
 228         if (sg_copy_to_buffer(req->src, sg_nents_for_len(req->src, assoclen),
 229                               rctx->adata + adata_header_len,
 230                               assoclen) != assoclen)
 231                 return -EINVAL;
 232
 233         /* Pad associated data to block size */
 234         rctx->assoclen = ALIGN(assoclen + adata_header_len, 16);
 235
 236         diff_dst = (req->src != req->dst) ? true : false;
 237
 238         if (diff_dst)
 239                 rctx->src_nents = sg_nents_for_len(req->src, totallen) + 1;
 240         else
 241                 rctx->src_nents = sg_nents_for_len(req->src, totallen) + 2;
 242
 243         gfp = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ? GFP_KERNEL : GFP_ATOMIC;
 244         ret = sg_alloc_table(&rctx->src_tbl, rctx->src_nents, gfp);
 245         if (ret)
 246                 return ret;
 247
 248         /* Associated Data */
 249         sg_init_one(&rctx->adata_sg, rctx->adata, rctx->assoclen);
 250         sg = qce_sgtable_add(&rctx->src_tbl, &rctx->adata_sg,
 251                              rctx->assoclen);
 252         if (IS_ERR(sg)) {
 253                 ret = PTR_ERR(sg);
 254                 goto err_free;
 255         }
 256         /* src msg */
 257         sg = qce_sgtable_add(&rctx->src_tbl, msg_sg, cryptlen);
 258         if (IS_ERR(sg)) {
 259                 ret = PTR_ERR(sg);
 260                 goto err_free;
 261         }
 262         if (!diff_dst) {
 263                 /*
 264                  * For decrypt, when src and dst buffers are same, there is already space
 265                  * in the buffer for padded 0's which is output in lieu of
 266                  * the MAC that is input. So skip the below.
 267                  */
 268                 if (!IS_DECRYPT(rctx->flags)) {
 269                         sg = qce_aead_prepare_ccm_result_buf(&rctx->src_tbl, req);
 270                         if (IS_ERR(sg)) {
 271                                 ret = PTR_ERR(sg);
 272                                 goto err_free;
 273                         }
 274                 }
 275         }
 276         sg_mark_end(sg);
 277         rctx->src_sg = rctx->src_tbl.sgl;
 278         totallen = cryptlen + rctx->assoclen;
 279         rctx->src_nents = sg_nents_for_len(rctx->src_sg, totallen);
 280
 281         if (diff_dst) {
 282                 sg = qce_aead_prepare_dst_buf(req);
 283                 if (IS_ERR(sg)) {
 284                         ret = PTR_ERR(sg);
 285                         goto err_free;
 286                 }
 287         } else {
 288                 if (IS_ENCRYPT(rctx->flags))
 289                         rctx->dst_nents = rctx->src_nents + 1;
 290                 else
 291                         rctx->dst_nents = rctx->src_nents;
 292                 rctx->dst_sg = rctx->src_sg;
 293         }
 294
 295         return 0;
 296 err_free:
 297         sg_free_table(&rctx->src_tbl);
 298         return ret;
 299 }
 300
 301 static int qce_aead_prepare_buf(struct aead_request *req)
 302 {
 303         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 304         struct qce_alg_template *tmpl = to_aead_tmpl(crypto_aead_reqtfm(req));
 305         struct qce_device *qce = tmpl->qce;
 306         struct scatterlist *sg;
 307         bool diff_dst = (req->src != req->dst) ? true : false;
 308         unsigned int totallen;
 309
 310         totallen = rctx->cryptlen + rctx->assoclen;
 311
 312         sg = qce_aead_prepare_dst_buf(req);
 313         if (IS_ERR(sg))
 314                 return PTR_ERR(sg);
 315         if (diff_dst) {
 316                 rctx->src_nents = sg_nents_for_len(req->src, totallen);
 317                 if (rctx->src_nents < 0) {
 318                         dev_err(qce->dev, "Invalid numbers of src SG.\n");
 319                         return -EINVAL;
 320                 }
 321                 rctx->src_sg = req->src;
 322         } else {
 323                 rctx->src_nents = rctx->dst_nents - 1;
 324                 rctx->src_sg = rctx->dst_sg;
 325         }
 326         return 0;
 327 }
 328
 329 static int qce_aead_ccm_prepare_buf(struct aead_request *req)
 330 {
 331         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 332         struct crypto_aead *tfm = crypto_aead_reqtfm(req);
 333         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 334         struct scatterlist *sg;
 335         bool diff_dst = (req->src != req->dst) ? true : false;
 336         unsigned int cryptlen;
 337
 338         if (rctx->assoclen)
 339                 return qce_aead_ccm_prepare_buf_assoclen(req);
 340
 341         if (IS_ENCRYPT(rctx->flags))
 342                 return qce_aead_prepare_buf(req);
 343
 344         cryptlen = rctx->cryptlen + ctx->authsize;
 345         if (diff_dst) {
 346                 rctx->src_nents = sg_nents_for_len(req->src, cryptlen);
 347                 rctx->src_sg = req->src;
 348                 sg = qce_aead_prepare_dst_buf(req);
 349                 if (IS_ERR(sg))
 350                         return PTR_ERR(sg);
 351         } else {
 352                 rctx->src_nents = sg_nents_for_len(req->src, cryptlen);
 353                 rctx->src_sg = req->src;
 354                 rctx->dst_nents = rctx->src_nents;
 355                 rctx->dst_sg = rctx->src_sg;
 356         }
 357
 358         return 0;
 359 }
 360
 361 static int qce_aead_create_ccm_nonce(struct qce_aead_reqctx *rctx, struct qce_aead_ctx *ctx)
 362 {
 363         unsigned int msglen_size, ivsize;
 364         u8 msg_len[4];
 365         int i;
 366
 367         if (!rctx || !rctx->iv)
 368                 return -EINVAL;
 369
 370         msglen_size = rctx->iv[0] + 1;
 371
 372         /* Verify that msg len size is valid */
 373         if (msglen_size < 2 || msglen_size > 8)
 374                 return -EINVAL;
 375
 376         ivsize = rctx->ivsize;
 377
 378         /*
 379          * Clear the msglen bytes in IV.
 380          * Else the h/w engine and nonce will use any stray value pending there.
 381          */
 382         if (!IS_CCM_RFC4309(rctx->flags)) {
 383                 for (i = 0; i < msglen_size; i++)
 384                         rctx->iv[ivsize - i - 1] = 0;
 385         }
 386
 387         /*
 388          * The crypto framework encodes cryptlen as unsigned int. Thus, even though
 389          * spec allows for upto 8 bytes to encode msg_len only 4 bytes are needed.
 390          */
 391         if (msglen_size > 4)
 392                 msglen_size = 4;
 393
 394         memcpy(&msg_len[0], &rctx->cryptlen, 4);
 395
 396         memcpy(&rctx->ccm_nonce[0], rctx->iv, rctx->ivsize);
 397         if (rctx->assoclen)
 398                 rctx->ccm_nonce[0] |= 1 << CCM_NONCE_ADATA_SHIFT;
 399         rctx->ccm_nonce[0] |= ((ctx->authsize - 2) / 2) <<
 400                                 CCM_NONCE_AUTHSIZE_SHIFT;
 401         for (i = 0; i < msglen_size; i++)
 402                 rctx->ccm_nonce[QCE_MAX_NONCE - i - 1] = msg_len[i];
 403
 404         return 0;
 405 }
 406
 407 static int
 408 qce_aead_async_req_handle(struct crypto_async_request *async_req)
 409 {
 410         struct aead_request *req = aead_request_cast(async_req);
 411         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 412         struct crypto_aead *tfm = crypto_aead_reqtfm(req);
 413         struct qce_aead_ctx *ctx = crypto_tfm_ctx(async_req->tfm);
 414         struct qce_alg_template *tmpl = to_aead_tmpl(crypto_aead_reqtfm(req));
 415         struct qce_device *qce = tmpl->qce;
 416         enum dma_data_direction dir_src, dir_dst;
 417         bool diff_dst;
 418         int dst_nents, src_nents, ret;
 419
 420         if (IS_CCM_RFC4309(rctx->flags)) {
 421                 memset(rctx->ccm_rfc4309_iv, 0, QCE_MAX_IV_SIZE);
 422                 rctx->ccm_rfc4309_iv[0] = 3;
 423                 memcpy(&rctx->ccm_rfc4309_iv[1], ctx->ccm4309_salt, QCE_CCM4309_SALT_SIZE);
 424                 memcpy(&rctx->ccm_rfc4309_iv[4], req->iv, 8);
 425                 rctx->iv = rctx->ccm_rfc4309_iv;
 426                 rctx->ivsize = AES_BLOCK_SIZE;
 427         } else {
 428                 rctx->iv = req->iv;
 429                 rctx->ivsize = crypto_aead_ivsize(tfm);
 430         }
 431         if (IS_CCM_RFC4309(rctx->flags))
 432                 rctx->assoclen = req->assoclen - 8;
 433         else
 434                 rctx->assoclen = req->assoclen;
 435
 436         diff_dst = (req->src != req->dst) ? true : false;
 437         dir_src = diff_dst ? DMA_TO_DEVICE : DMA_BIDIRECTIONAL;
 438         dir_dst = diff_dst ? DMA_FROM_DEVICE : DMA_BIDIRECTIONAL;
 439
 440         if (IS_CCM(rctx->flags)) {
 441                 ret = qce_aead_create_ccm_nonce(rctx, ctx);
 442                 if (ret)
 443                         return ret;
 444         }
 445         if (IS_CCM(rctx->flags))
 446                 ret = qce_aead_ccm_prepare_buf(req);
 447         else
 448                 ret = qce_aead_prepare_buf(req);
 449
 450         if (ret)
 451                 return ret;
 452         dst_nents = dma_map_sg(qce->dev, rctx->dst_sg, rctx->dst_nents, dir_dst);
 453         if (dst_nents < 0) {
 454                 ret = dst_nents;
 455                 goto error_free;
 456         }
 457
 458         if (diff_dst) {
 459                 src_nents = dma_map_sg(qce->dev, rctx->src_sg, rctx->src_nents, dir_src);
 460                 if (src_nents < 0) {
 461                         ret = src_nents;
 462                         goto error_unmap_dst;
 463                 }
 464         } else {
 465                 if (IS_CCM(rctx->flags) && IS_DECRYPT(rctx->flags))
 466                         src_nents = dst_nents;
 467                 else
 468                         src_nents = dst_nents - 1;
 469         }
 470
 471         ret = qce_dma_prep_sgs(&qce->dma, rctx->src_sg, src_nents, rctx->dst_sg, dst_nents,
 472                                qce_aead_done, async_req);
 473         if (ret)
 474                 goto error_unmap_src;
 475
 476         qce_dma_issue_pending(&qce->dma);
 477
 478         ret = qce_start(async_req, tmpl->crypto_alg_type);
 479         if (ret)
 480                 goto error_terminate;
 481
 482         return 0;
 483
 484 error_terminate:
 485         qce_dma_terminate_all(&qce->dma);
 486 error_unmap_src:
 487         if (diff_dst)
 488                 dma_unmap_sg(qce->dev, req->src, rctx->src_nents, dir_src);
 489 error_unmap_dst:
 490         dma_unmap_sg(qce->dev, rctx->dst_sg, rctx->dst_nents, dir_dst);
 491 error_free:
 492         if (IS_CCM(rctx->flags) && rctx->assoclen) {
 493                 sg_free_table(&rctx->src_tbl);
 494                 if (diff_dst)
 495                         sg_free_table(&rctx->dst_tbl);
 496         } else {
 497                 sg_free_table(&rctx->dst_tbl);
 498         }
 499         return ret;
 500 }
 501
 502 static int qce_aead_crypt(struct aead_request *req, int encrypt)
 503 {
 504         struct crypto_aead *tfm = crypto_aead_reqtfm(req);
 505         struct qce_aead_reqctx *rctx = aead_request_ctx(req);
 506         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 507         struct qce_alg_template *tmpl = to_aead_tmpl(tfm);
 508         unsigned int blocksize = crypto_aead_blocksize(tfm);
 509
 510         rctx->flags  = tmpl->alg_flags;
 511         rctx->flags |= encrypt ? QCE_ENCRYPT : QCE_DECRYPT;
 512
 513         if (encrypt)
 514                 rctx->cryptlen = req->cryptlen;
 515         else
 516                 rctx->cryptlen = req->cryptlen - ctx->authsize;
 517
 518         /* CE does not handle 0 length messages */
 519         if (!rctx->cryptlen) {
 520                 if (!(IS_CCM(rctx->flags) && IS_DECRYPT(rctx->flags)))
 521                         ctx->need_fallback = true;
 522         }
 523
 524         /* If fallback is needed, schedule and exit */
 525         if (ctx->need_fallback) {
 526                 /* Reset need_fallback in case the same ctx is used for another transaction */
 527                 ctx->need_fallback = false;
 528
 529                 aead_request_set_tfm(&rctx->fallback_req, ctx->fallback);
 530                 aead_request_set_callback(&rctx->fallback_req, req->base.flags,
 531                                           req->base.complete, req->base.data);
 532                 aead_request_set_crypt(&rctx->fallback_req, req->src,
 533                                        req->dst, req->cryptlen, req->iv);
 534                 aead_request_set_ad(&rctx->fallback_req, req->assoclen);
 535
 536                 return encrypt ? crypto_aead_encrypt(&rctx->fallback_req) :
 537                                  crypto_aead_decrypt(&rctx->fallback_req);
 538         }
 539
 540         /*
 541          * CBC algorithms require message lengths to be
 542          * multiples of block size.
 543          */
 544         if (IS_CBC(rctx->flags) && !IS_ALIGNED(rctx->cryptlen, blocksize))
 545                 return -EINVAL;
 546
 547         /* RFC4309 supported AAD size 16 bytes/20 bytes */
 548         if (IS_CCM_RFC4309(rctx->flags))
 549                 if (crypto_ipsec_check_assoclen(req->assoclen))
 550                         return -EINVAL;
 551
 552         return tmpl->qce->async_req_enqueue(tmpl->qce, &req->base);
 553 }
 554
 555 static int qce_aead_encrypt(struct aead_request *req)
 556 {
 557         return qce_aead_crypt(req, 1);
 558 }
 559
 560 static int qce_aead_decrypt(struct aead_request *req)
 561 {
 562         return qce_aead_crypt(req, 0);
 563 }
 564
 565 static int qce_aead_ccm_setkey(struct crypto_aead *tfm, const u8 *key,
 566                                unsigned int keylen)
 567 {
 568         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 569         unsigned long flags = to_aead_tmpl(tfm)->alg_flags;
 570
 571         if (IS_CCM_RFC4309(flags)) {
 572                 if (keylen < QCE_CCM4309_SALT_SIZE)
 573                         return -EINVAL;
 574                 keylen -= QCE_CCM4309_SALT_SIZE;
 575                 memcpy(ctx->ccm4309_salt, key + keylen, QCE_CCM4309_SALT_SIZE);
 576         }
 577
 578         if (keylen != AES_KEYSIZE_128 && keylen != AES_KEYSIZE_256 && keylen != AES_KEYSIZE_192)
 579                 return -EINVAL;
 580
 581         ctx->enc_keylen = keylen;
 582         ctx->auth_keylen = keylen;
 583
 584         memcpy(ctx->enc_key, key, keylen);
 585         memcpy(ctx->auth_key, key, keylen);
 586
 587         if (keylen == AES_KEYSIZE_192)
 588                 ctx->need_fallback = true;
 589
 590         return IS_CCM_RFC4309(flags) ?
 591                 crypto_aead_setkey(ctx->fallback, key, keylen + QCE_CCM4309_SALT_SIZE) :
 592                 crypto_aead_setkey(ctx->fallback, key, keylen);
 593 }
 594
 595 static int qce_aead_setkey(struct crypto_aead *tfm, const u8 *key, unsigned int keylen)
 596 {
 597         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 598         struct crypto_authenc_keys authenc_keys;
 599         unsigned long flags = to_aead_tmpl(tfm)->alg_flags;
 600         u32 _key[6];
 601         int err;
 602
 603         err = crypto_authenc_extractkeys(&authenc_keys, key, keylen);
 604         if (err)
 605                 return err;
 606
 607         if (authenc_keys.enckeylen > QCE_MAX_KEY_SIZE ||
 608             authenc_keys.authkeylen > QCE_MAX_KEY_SIZE)
 609                 return -EINVAL;
 610
 611         if (IS_DES(flags)) {
 612                 err = verify_aead_des_key(tfm, authenc_keys.enckey, authenc_keys.enckeylen);
 613                 if (err)
 614                         return err;
 615         } else if (IS_3DES(flags)) {
 616                 err = verify_aead_des3_key(tfm, authenc_keys.enckey, authenc_keys.enckeylen);
 617                 if (err)
 618                         return err;
 619                 /*
 620                  * The crypto engine does not support any two keys
 621                  * being the same for triple des algorithms. The
 622                  * verify_skcipher_des3_key does not check for all the
 623                  * below conditions. Schedule fallback in this case.
 624                  */
 625                 memcpy(_key, authenc_keys.enckey, DES3_EDE_KEY_SIZE);
 626                 if (!((_key[0] ^ _key[2]) | (_key[1] ^ _key[3])) ||
 627                     !((_key[2] ^ _key[4]) | (_key[3] ^ _key[5])) ||
 628                     !((_key[0] ^ _key[4]) | (_key[1] ^ _key[5])))
 629                         ctx->need_fallback = true;
 630         } else if (IS_AES(flags)) {
 631                 /* No random key sizes */
 632                 if (authenc_keys.enckeylen != AES_KEYSIZE_128 &&
 633                     authenc_keys.enckeylen != AES_KEYSIZE_192 &&
 634                     authenc_keys.enckeylen != AES_KEYSIZE_256)
 635                         return -EINVAL;
 636                 if (authenc_keys.enckeylen == AES_KEYSIZE_192)
 637                         ctx->need_fallback = true;
 638         }
 639
 640         ctx->enc_keylen = authenc_keys.enckeylen;
 641         ctx->auth_keylen = authenc_keys.authkeylen;
 642
 643         memcpy(ctx->enc_key, authenc_keys.enckey, authenc_keys.enckeylen);
 644
 645         memset(ctx->auth_key, 0, sizeof(ctx->auth_key));
 646         memcpy(ctx->auth_key, authenc_keys.authkey, authenc_keys.authkeylen);
 647
 648         return crypto_aead_setkey(ctx->fallback, key, keylen);
 649 }
 650
 651 static int qce_aead_setauthsize(struct crypto_aead *tfm, unsigned int authsize)
 652 {
 653         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 654         unsigned long flags = to_aead_tmpl(tfm)->alg_flags;
 655
 656         if (IS_CCM(flags)) {
 657                 if (authsize < 4 || authsize > 16 || authsize % 2)
 658                         return -EINVAL;
 659                 if (IS_CCM_RFC4309(flags) && (authsize < 8 || authsize % 4))
 660                         return -EINVAL;
 661         }
 662         ctx->authsize = authsize;
 663
 664         return crypto_aead_setauthsize(ctx->fallback, authsize);
 665 }
 666
 667 static int qce_aead_init(struct crypto_aead *tfm)
 668 {
 669         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 670
 671         ctx->need_fallback = false;
 672         ctx->fallback = crypto_alloc_aead(crypto_tfm_alg_name(&tfm->base),
 673                                           0, CRYPTO_ALG_NEED_FALLBACK);
 674
 675         if (IS_ERR(ctx->fallback))
 676                 return PTR_ERR(ctx->fallback);
 677
 678         crypto_aead_set_reqsize(tfm, sizeof(struct qce_aead_reqctx) +
 679                                 crypto_aead_reqsize(ctx->fallback));
 680         return 0;
 681 }
 682
 683 static void qce_aead_exit(struct crypto_aead *tfm)
 684 {
 685         struct qce_aead_ctx *ctx = crypto_aead_ctx(tfm);
 686
 687         crypto_free_aead(ctx->fallback);
 688 }
 689
 690 struct qce_aead_def {
 691         unsigned long flags;
 692         const char *name;
 693         const char *drv_name;
 694         unsigned int blocksize;
 695         unsigned int chunksize;
 696         unsigned int ivsize;
 697         unsigned int maxauthsize;
 698 };
 699
 700 static const struct qce_aead_def aead_def[] = {
 701         {
 702                 .flags          = QCE_ALG_DES | QCE_MODE_CBC | QCE_HASH_SHA1_HMAC,
 703                 .name           = "authenc(hmac(sha1),cbc(des))",
 704                 .drv_name       = "authenc-hmac-sha1-cbc-des-qce",
 705                 .blocksize      = DES_BLOCK_SIZE,
 706                 .ivsize         = DES_BLOCK_SIZE,
 707                 .maxauthsize    = SHA1_DIGEST_SIZE,
 708         },
 709         {
 710                 .flags          = QCE_ALG_3DES | QCE_MODE_CBC | QCE_HASH_SHA1_HMAC,
 711                 .name           = "authenc(hmac(sha1),cbc(des3_ede))",
 712                 .drv_name       = "authenc-hmac-sha1-cbc-3des-qce",
 713                 .blocksize      = DES3_EDE_BLOCK_SIZE,
 714                 .ivsize         = DES3_EDE_BLOCK_SIZE,
 715                 .maxauthsize    = SHA1_DIGEST_SIZE,
 716         },
 717         {
 718                 .flags          = QCE_ALG_DES | QCE_MODE_CBC | QCE_HASH_SHA256_HMAC,
 719                 .name           = "authenc(hmac(sha256),cbc(des))",
 720                 .drv_name       = "authenc-hmac-sha256-cbc-des-qce",
 721                 .blocksize      = DES_BLOCK_SIZE,
 722                 .ivsize         = DES_BLOCK_SIZE,
 723                 .maxauthsize    = SHA256_DIGEST_SIZE,
 724         },
 725         {
 726                 .flags          = QCE_ALG_3DES | QCE_MODE_CBC | QCE_HASH_SHA256_HMAC,
 727                 .name           = "authenc(hmac(sha256),cbc(des3_ede))",
 728                 .drv_name       = "authenc-hmac-sha256-cbc-3des-qce",
 729                 .blocksize      = DES3_EDE_BLOCK_SIZE,
 730                 .ivsize         = DES3_EDE_BLOCK_SIZE,
 731                 .maxauthsize    = SHA256_DIGEST_SIZE,
 732         },
 733         {
 734                 .flags          =  QCE_ALG_AES | QCE_MODE_CBC | QCE_HASH_SHA256_HMAC,
 735                 .name           = "authenc(hmac(sha256),cbc(aes))",
 736                 .drv_name       = "authenc-hmac-sha256-cbc-aes-qce",
 737                 .blocksize      = AES_BLOCK_SIZE,
 738                 .ivsize         = AES_BLOCK_SIZE,
 739                 .maxauthsize    = SHA256_DIGEST_SIZE,
 740         },
 741         {
 742                 .flags          =  QCE_ALG_AES | QCE_MODE_CCM,
 743                 .name           = "ccm(aes)",
 744                 .drv_name       = "ccm-aes-qce",
 745                 .blocksize      = 1,
 746                 .ivsize         = AES_BLOCK_SIZE,
 747                 .maxauthsize    = AES_BLOCK_SIZE,
 748         },
 749         {
 750                 .flags          =  QCE_ALG_AES | QCE_MODE_CCM | QCE_MODE_CCM_RFC4309,
 751                 .name           = "rfc4309(ccm(aes))",
 752                 .drv_name       = "rfc4309-ccm-aes-qce",
 753                 .blocksize      = 1,
 754                 .ivsize         = 8,
 755                 .maxauthsize    = AES_BLOCK_SIZE,
 756         },
 757 };
 758
 759 static int qce_aead_register_one(const struct qce_aead_def *def, struct qce_device *qce)
 760 {
 761         struct qce_alg_template *tmpl;
 762         struct aead_alg *alg;
 763         int ret;
 764
 765         tmpl = kzalloc(sizeof(*tmpl), GFP_KERNEL);
 766         if (!tmpl)
 767                 return -ENOMEM;
 768
 769         alg = &tmpl->alg.aead;
 770
 771         snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", def->name);
 772         snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
 773                  def->drv_name);
 774
 775         alg->base.cra_blocksize         = def->blocksize;
 776         alg->chunksize                  = def->chunksize;
 777         alg->ivsize                     = def->ivsize;
 778         alg->maxauthsize                = def->maxauthsize;
 779         if (IS_CCM(def->flags))
 780                 alg->setkey             = qce_aead_ccm_setkey;
 781         else
 782                 alg->setkey             = qce_aead_setkey;
 783         alg->setauthsize                = qce_aead_setauthsize;
 784         alg->encrypt                    = qce_aead_encrypt;
 785         alg->decrypt                    = qce_aead_decrypt;
 786         alg->init                       = qce_aead_init;
 787         alg->exit                       = qce_aead_exit;
 788
 789         alg->base.cra_priority          = 300;
 790         alg->base.cra_flags             = CRYPTO_ALG_ASYNC |
 791                                           CRYPTO_ALG_ALLOCATES_MEMORY |
 792                                           CRYPTO_ALG_KERN_DRIVER_ONLY |
 793                                           CRYPTO_ALG_NEED_FALLBACK;
 794         alg->base.cra_ctxsize           = sizeof(struct qce_aead_ctx);
 795         alg->base.cra_alignmask         = 0;
 796         alg->base.cra_module            = THIS_MODULE;
 797
 798         INIT_LIST_HEAD(&tmpl->entry);
 799         tmpl->crypto_alg_type = CRYPTO_ALG_TYPE_AEAD;
 800         tmpl->alg_flags = def->flags;
 801         tmpl->qce = qce;
 802
 803         ret = crypto_register_aead(alg);
 804         if (ret) {
 805                 kfree(tmpl);
 806                 dev_err(qce->dev, "%s registration failed\n", alg->base.cra_name);
 807                 return ret;
 808         }
 809
 810         list_add_tail(&tmpl->entry, &aead_algs);
 811         dev_dbg(qce->dev, "%s is registered\n", alg->base.cra_name);
 812         return 0;
 813 }
 814
 815 static void qce_aead_unregister(struct qce_device *qce)
 816 {
 817         struct qce_alg_template *tmpl, *n;
 818
 819         list_for_each_entry_safe(tmpl, n, &aead_algs, entry) {
 820                 crypto_unregister_aead(&tmpl->alg.aead);
 821                 list_del(&tmpl->entry);
 822                 kfree(tmpl);
 823         }
 824 }
 825
 826 static int qce_aead_register(struct qce_device *qce)
 827 {
 828         int ret, i;
 829
 830         for (i = 0; i < ARRAY_SIZE(aead_def); i++) {
 831                 ret = qce_aead_register_one(&aead_def[i], qce);
 832                 if (ret)
 833                         goto err;
 834         }
 835
 836         return 0;
 837 err:
 838         qce_aead_unregister(qce);
 839         return ret;
 840 }
 841
 842 const struct qce_algo_ops aead_ops = {
 843         .type = CRYPTO_ALG_TYPE_AEAD,
 844         .register_algs = qce_aead_register,
 845         .unregister_algs = qce_aead_unregister,
 846         .async_req_handle = qce_aead_async_req_handle,
 847 };