drivers/crypto/marvell/hash.c

   1 /*
   2  * Hash algorithms supported by the CESA: MD5, SHA1 and SHA256.
   3  *
   4  * Author: Boris Brezillon <boris.brezillon@free-electrons.com>
   5  * Author: Arnaud Ebalard <arno@natisbad.org>
   6  *
   7  * This work is based on an initial version written by
   8  * Sebastian Andrzej Siewior < sebastian at breakpoint dot cc >
   9  *
  10  * This program is free software; you can redistribute it and/or modify it
  11  * under the terms of the GNU General Public License version 2 as published
  12  * by the Free Software Foundation.
  13  */
  14
  15 #include <crypto/md5.h>
  16 #include <crypto/sha.h>
  17
  18 #include "cesa.h"
  19
  20 struct mv_cesa_ahash_dma_iter {
  21         struct mv_cesa_dma_iter base;
  22         struct mv_cesa_sg_dma_iter src;
  23 };
  24
  25 static inline void
  26 mv_cesa_ahash_req_iter_init(struct mv_cesa_ahash_dma_iter *iter,
  27                             struct ahash_request *req)
  28 {
  29         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
  30         unsigned int len = req->nbytes;
  31
  32         if (!creq->last_req)
  33                 len = (len + creq->cache_ptr) & ~CESA_HASH_BLOCK_SIZE_MSK;
  34
  35         mv_cesa_req_dma_iter_init(&iter->base, len);
  36         mv_cesa_sg_dma_iter_init(&iter->src, req->src, DMA_TO_DEVICE);
  37         iter->src.op_offset = creq->cache_ptr;
  38 }
  39
  40 static inline bool
  41 mv_cesa_ahash_req_iter_next_op(struct mv_cesa_ahash_dma_iter *iter)
  42 {
  43         iter->src.op_offset = 0;
  44
  45         return mv_cesa_req_dma_iter_next_op(&iter->base);
  46 }
  47
  48 static inline int mv_cesa_ahash_dma_alloc_cache(struct mv_cesa_ahash_req *creq,
  49                                                 gfp_t flags)
  50 {
  51         struct mv_cesa_ahash_dma_req *dreq = &creq->req.dma;
  52
  53         creq->cache = dma_pool_alloc(cesa_dev->dma->cache_pool, flags,
  54                                      &dreq->cache_dma);
  55         if (!creq->cache)
  56                 return -ENOMEM;
  57
  58         return 0;
  59 }
  60
  61 static inline int mv_cesa_ahash_std_alloc_cache(struct mv_cesa_ahash_req *creq,
  62                                                 gfp_t flags)
  63 {
  64         creq->cache = kzalloc(CESA_MAX_HASH_BLOCK_SIZE, flags);
  65         if (!creq->cache)
  66                 return -ENOMEM;
  67
  68         return 0;
  69 }
  70
  71 static int mv_cesa_ahash_alloc_cache(struct ahash_request *req)
  72 {
  73         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
  74         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
  75                       GFP_KERNEL : GFP_ATOMIC;
  76         int ret;
  77
  78         if (creq->cache)
  79                 return 0;
  80
  81         if (creq->req.base.type == CESA_DMA_REQ)
  82                 ret = mv_cesa_ahash_dma_alloc_cache(creq, flags);
  83         else
  84                 ret = mv_cesa_ahash_std_alloc_cache(creq, flags);
  85
  86         return ret;
  87 }
  88
  89 static inline void mv_cesa_ahash_dma_free_cache(struct mv_cesa_ahash_req *creq)
  90 {
  91         dma_pool_free(cesa_dev->dma->cache_pool, creq->cache,
  92                       creq->req.dma.cache_dma);
  93 }
  94
  95 static inline void mv_cesa_ahash_std_free_cache(struct mv_cesa_ahash_req *creq)
  96 {
  97         kfree(creq->cache);
  98 }
  99
 100 static void mv_cesa_ahash_free_cache(struct mv_cesa_ahash_req *creq)
 101 {
 102         if (!creq->cache)
 103                 return;
 104
 105         if (creq->req.base.type == CESA_DMA_REQ)
 106                 mv_cesa_ahash_dma_free_cache(creq);
 107         else
 108                 mv_cesa_ahash_std_free_cache(creq);
 109
 110         creq->cache = NULL;
 111 }
 112
 113 static int mv_cesa_ahash_dma_alloc_padding(struct mv_cesa_ahash_dma_req *req,
 114                                            gfp_t flags)
 115 {
 116         if (req->padding)
 117                 return 0;
 118
 119         req->padding = dma_pool_alloc(cesa_dev->dma->padding_pool, flags,
 120                                       &req->padding_dma);
 121         if (!req->padding)
 122                 return -ENOMEM;
 123
 124         return 0;
 125 }
 126
 127 static void mv_cesa_ahash_dma_free_padding(struct mv_cesa_ahash_dma_req *req)
 128 {
 129         if (!req->padding)
 130                 return;
 131
 132         dma_pool_free(cesa_dev->dma->padding_pool, req->padding,
 133                       req->padding_dma);
 134         req->padding = NULL;
 135 }
 136
 137 static inline void mv_cesa_ahash_dma_last_cleanup(struct ahash_request *req)
 138 {
 139         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 140
 141         mv_cesa_ahash_dma_free_padding(&creq->req.dma);
 142 }
 143
 144 static inline void mv_cesa_ahash_dma_cleanup(struct ahash_request *req)
 145 {
 146         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 147
 148         dma_unmap_sg(cesa_dev->dev, req->src, creq->src_nents, DMA_TO_DEVICE);
 149         mv_cesa_dma_cleanup(&creq->req.dma.base);
 150 }
 151
 152 static inline void mv_cesa_ahash_cleanup(struct ahash_request *req)
 153 {
 154         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 155
 156         if (creq->req.base.type == CESA_DMA_REQ)
 157                 mv_cesa_ahash_dma_cleanup(req);
 158 }
 159
 160 static void mv_cesa_ahash_last_cleanup(struct ahash_request *req)
 161 {
 162         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 163
 164         mv_cesa_ahash_free_cache(creq);
 165
 166         if (creq->req.base.type == CESA_DMA_REQ)
 167                 mv_cesa_ahash_dma_last_cleanup(req);
 168 }
 169
 170 static int mv_cesa_ahash_pad_len(struct mv_cesa_ahash_req *creq)
 171 {
 172         unsigned int index, padlen;
 173
 174         index = creq->len & CESA_HASH_BLOCK_SIZE_MSK;
 175         padlen = (index < 56) ? (56 - index) : (64 + 56 - index);
 176
 177         return padlen;
 178 }
 179
 180 static int mv_cesa_ahash_pad_req(struct mv_cesa_ahash_req *creq, u8 *buf)
 181 {
 182         __be64 bits = cpu_to_be64(creq->len << 3);
 183         unsigned int index, padlen;
 184
 185         buf[0] = 0x80;
 186         /* Pad out to 56 mod 64 */
 187         index = creq->len & CESA_HASH_BLOCK_SIZE_MSK;
 188         padlen = mv_cesa_ahash_pad_len(creq);
 189         memset(buf + 1, 0, padlen - 1);
 190         memcpy(buf + padlen, &bits, sizeof(bits));
 191
 192         return padlen + 8;
 193 }
 194
 195 static void mv_cesa_ahash_std_step(struct ahash_request *req)
 196 {
 197         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 198         struct mv_cesa_ahash_std_req *sreq = &creq->req.std;
 199         struct mv_cesa_engine *engine = sreq->base.engine;
 200         struct mv_cesa_op_ctx *op;
 201         unsigned int new_cache_ptr = 0;
 202         u32 frag_mode;
 203         size_t  len;
 204
 205         if (creq->cache_ptr)
 206                 memcpy(engine->sram + CESA_SA_DATA_SRAM_OFFSET, creq->cache,
 207                        creq->cache_ptr);
 208
 209         len = min_t(size_t, req->nbytes + creq->cache_ptr - sreq->offset,
 210                     CESA_SA_SRAM_PAYLOAD_SIZE);
 211
 212         if (!creq->last_req) {
 213                 new_cache_ptr = len & CESA_HASH_BLOCK_SIZE_MSK;
 214                 len &= ~CESA_HASH_BLOCK_SIZE_MSK;
 215         }
 216
 217         if (len - creq->cache_ptr)
 218                 sreq->offset += sg_pcopy_to_buffer(req->src, creq->src_nents,
 219                                                    engine->sram +
 220                                                    CESA_SA_DATA_SRAM_OFFSET +
 221                                                    creq->cache_ptr,
 222                                                    len - creq->cache_ptr,
 223                                                    sreq->offset);
 224
 225         op = &creq->op_tmpl;
 226
 227         frag_mode = mv_cesa_get_op_cfg(op) & CESA_SA_DESC_CFG_FRAG_MSK;
 228
 229         if (creq->last_req && sreq->offset == req->nbytes &&
 230             creq->len <= CESA_SA_DESC_MAC_SRC_TOTAL_LEN_MAX) {
 231                 if (frag_mode == CESA_SA_DESC_CFG_FIRST_FRAG)
 232                         frag_mode = CESA_SA_DESC_CFG_NOT_FRAG;
 233                 else if (frag_mode == CESA_SA_DESC_CFG_MID_FRAG)
 234                         frag_mode = CESA_SA_DESC_CFG_LAST_FRAG;
 235         }
 236
 237         if (frag_mode == CESA_SA_DESC_CFG_NOT_FRAG ||
 238             frag_mode == CESA_SA_DESC_CFG_LAST_FRAG) {
 239                 if (len &&
 240                     creq->len <= CESA_SA_DESC_MAC_SRC_TOTAL_LEN_MAX) {
 241                         mv_cesa_set_mac_op_total_len(op, creq->len);
 242                 } else {
 243                         int trailerlen = mv_cesa_ahash_pad_len(creq) + 8;
 244
 245                         if (len + trailerlen > CESA_SA_SRAM_PAYLOAD_SIZE) {
 246                                 len &= CESA_HASH_BLOCK_SIZE_MSK;
 247                                 new_cache_ptr = 64 - trailerlen;
 248                                 memcpy(creq->cache,
 249                                        engine->sram +
 250                                        CESA_SA_DATA_SRAM_OFFSET + len,
 251                                        new_cache_ptr);
 252                         } else {
 253                                 len += mv_cesa_ahash_pad_req(creq,
 254                                                 engine->sram + len +
 255                                                 CESA_SA_DATA_SRAM_OFFSET);
 256                         }
 257
 258                         if (frag_mode == CESA_SA_DESC_CFG_LAST_FRAG)
 259                                 frag_mode = CESA_SA_DESC_CFG_MID_FRAG;
 260                         else
 261                                 frag_mode = CESA_SA_DESC_CFG_FIRST_FRAG;
 262                 }
 263         }
 264
 265         mv_cesa_set_mac_op_frag_len(op, len);
 266         mv_cesa_update_op_cfg(op, frag_mode, CESA_SA_DESC_CFG_FRAG_MSK);
 267
 268         /* FIXME: only update enc_len field */
 269         memcpy(engine->sram, op, sizeof(*op));
 270
 271         if (frag_mode == CESA_SA_DESC_CFG_FIRST_FRAG)
 272                 mv_cesa_update_op_cfg(op, CESA_SA_DESC_CFG_MID_FRAG,
 273                                       CESA_SA_DESC_CFG_FRAG_MSK);
 274
 275         creq->cache_ptr = new_cache_ptr;
 276
 277         mv_cesa_set_int_mask(engine, CESA_SA_INT_ACCEL0_DONE);
 278         writel(CESA_SA_CFG_PARA_DIS, engine->regs + CESA_SA_CFG);
 279         writel(CESA_SA_CMD_EN_CESA_SA_ACCL0, engine->regs + CESA_SA_CMD);
 280 }
 281
 282 static int mv_cesa_ahash_std_process(struct ahash_request *req, u32 status)
 283 {
 284         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 285         struct mv_cesa_ahash_std_req *sreq = &creq->req.std;
 286
 287         if (sreq->offset < (req->nbytes - creq->cache_ptr))
 288                 return -EINPROGRESS;
 289
 290         return 0;
 291 }
 292
 293 static inline void mv_cesa_ahash_dma_prepare(struct ahash_request *req)
 294 {
 295         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 296         struct mv_cesa_tdma_req *dreq = &creq->req.dma.base;
 297
 298         mv_cesa_dma_prepare(dreq, dreq->base.engine);
 299 }
 300
 301 static void mv_cesa_ahash_std_prepare(struct ahash_request *req)
 302 {
 303         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 304         struct mv_cesa_ahash_std_req *sreq = &creq->req.std;
 305         struct mv_cesa_engine *engine = sreq->base.engine;
 306
 307         sreq->offset = 0;
 308         mv_cesa_adjust_op(engine, &creq->op_tmpl);
 309         memcpy(engine->sram, &creq->op_tmpl, sizeof(creq->op_tmpl));
 310 }
 311
 312 static void mv_cesa_ahash_step(struct crypto_async_request *req)
 313 {
 314         struct ahash_request *ahashreq = ahash_request_cast(req);
 315         struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
 316
 317         if (creq->req.base.type == CESA_DMA_REQ)
 318                 mv_cesa_dma_step(&creq->req.dma.base);
 319         else
 320                 mv_cesa_ahash_std_step(ahashreq);
 321 }
 322
 323 static int mv_cesa_ahash_process(struct crypto_async_request *req, u32 status)
 324 {
 325         struct ahash_request *ahashreq = ahash_request_cast(req);
 326         struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
 327         struct mv_cesa_engine *engine = creq->req.base.engine;
 328         unsigned int digsize;
 329         int ret, i;
 330
 331         if (creq->req.base.type == CESA_DMA_REQ)
 332                 ret = mv_cesa_dma_process(&creq->req.dma.base, status);
 333         else
 334                 ret = mv_cesa_ahash_std_process(ahashreq, status);
 335
 336         if (ret == -EINPROGRESS)
 337                 return ret;
 338
 339         digsize = crypto_ahash_digestsize(crypto_ahash_reqtfm(ahashreq));
 340         for (i = 0; i < digsize / 4; i++)
 341                 creq->state[i] = readl(engine->regs + CESA_IVDIG(i));
 342
 343         if (creq->cache_ptr)
 344                 sg_pcopy_to_buffer(ahashreq->src, creq->src_nents,
 345                                    creq->cache,
 346                                    creq->cache_ptr,
 347                                    ahashreq->nbytes - creq->cache_ptr);
 348
 349         if (creq->last_req) {
 350                 for (i = 0; i < digsize / 4; i++) {
 351                         /*
 352                          * Hardware provides MD5 digest in a different
 353                          * endianness than SHA-1 and SHA-256 ones.
 354                          */
 355                         if (digsize == MD5_DIGEST_SIZE)
 356                                 creq->state[i] = cpu_to_le32(creq->state[i]);
 357                         else
 358                                 creq->state[i] = cpu_to_be32(creq->state[i]);
 359                 }
 360
 361                 memcpy(ahashreq->result, creq->state, digsize);
 362         }
 363
 364         return ret;
 365 }
 366
 367 static void mv_cesa_ahash_prepare(struct crypto_async_request *req,
 368                                   struct mv_cesa_engine *engine)
 369 {
 370         struct ahash_request *ahashreq = ahash_request_cast(req);
 371         struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
 372         unsigned int digsize;
 373         int i;
 374
 375         creq->req.base.engine = engine;
 376
 377         if (creq->req.base.type == CESA_DMA_REQ)
 378                 mv_cesa_ahash_dma_prepare(ahashreq);
 379         else
 380                 mv_cesa_ahash_std_prepare(ahashreq);
 381
 382         digsize = crypto_ahash_digestsize(crypto_ahash_reqtfm(ahashreq));
 383         for (i = 0; i < digsize / 4; i++)
 384                 writel(creq->state[i],
 385                        engine->regs + CESA_IVDIG(i));
 386 }
 387
 388 static void mv_cesa_ahash_req_cleanup(struct crypto_async_request *req)
 389 {
 390         struct ahash_request *ahashreq = ahash_request_cast(req);
 391         struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
 392
 393         if (creq->last_req)
 394                 mv_cesa_ahash_last_cleanup(ahashreq);
 395
 396         mv_cesa_ahash_cleanup(ahashreq);
 397 }
 398
 399 static const struct mv_cesa_req_ops mv_cesa_ahash_req_ops = {
 400         .step = mv_cesa_ahash_step,
 401         .process = mv_cesa_ahash_process,
 402         .prepare = mv_cesa_ahash_prepare,
 403         .cleanup = mv_cesa_ahash_req_cleanup,
 404 };
 405
 406 static int mv_cesa_ahash_init(struct ahash_request *req,
 407                               struct mv_cesa_op_ctx *tmpl)
 408 {
 409         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 410
 411         memset(creq, 0, sizeof(*creq));
 412         mv_cesa_update_op_cfg(tmpl,
 413                               CESA_SA_DESC_CFG_OP_MAC_ONLY |
 414                               CESA_SA_DESC_CFG_FIRST_FRAG,
 415                               CESA_SA_DESC_CFG_OP_MSK |
 416                               CESA_SA_DESC_CFG_FRAG_MSK);
 417         mv_cesa_set_mac_op_total_len(tmpl, 0);
 418         mv_cesa_set_mac_op_frag_len(tmpl, 0);
 419         creq->op_tmpl = *tmpl;
 420         creq->len = 0;
 421
 422         return 0;
 423 }
 424
 425 static inline int mv_cesa_ahash_cra_init(struct crypto_tfm *tfm)
 426 {
 427         struct mv_cesa_hash_ctx *ctx = crypto_tfm_ctx(tfm);
 428
 429         ctx->base.ops = &mv_cesa_ahash_req_ops;
 430
 431         crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
 432                                  sizeof(struct mv_cesa_ahash_req));
 433         return 0;
 434 }
 435
 436 static int mv_cesa_ahash_cache_req(struct ahash_request *req, bool *cached)
 437 {
 438         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 439         int ret;
 440
 441         if (((creq->cache_ptr + req->nbytes) & CESA_HASH_BLOCK_SIZE_MSK) &&
 442             !creq->last_req) {
 443                 ret = mv_cesa_ahash_alloc_cache(req);
 444                 if (ret)
 445                         return ret;
 446         }
 447
 448         if (creq->cache_ptr + req->nbytes < 64 && !creq->last_req) {
 449                 *cached = true;
 450
 451                 if (!req->nbytes)
 452                         return 0;
 453
 454                 sg_pcopy_to_buffer(req->src, creq->src_nents,
 455                                    creq->cache + creq->cache_ptr,
 456                                    req->nbytes, 0);
 457
 458                 creq->cache_ptr += req->nbytes;
 459         }
 460
 461         return 0;
 462 }
 463
 464 static struct mv_cesa_op_ctx *
 465 mv_cesa_ahash_dma_add_cache(struct mv_cesa_tdma_chain *chain,
 466                             struct mv_cesa_ahash_dma_iter *dma_iter,
 467                             struct mv_cesa_ahash_req *creq,
 468                             gfp_t flags)
 469 {
 470         struct mv_cesa_ahash_dma_req *ahashdreq = &creq->req.dma;
 471         struct mv_cesa_op_ctx *op = NULL;
 472         int ret;
 473
 474         if (!creq->cache_ptr)
 475                 return NULL;
 476
 477         ret = mv_cesa_dma_add_data_transfer(chain,
 478                                             CESA_SA_DATA_SRAM_OFFSET,
 479                                             ahashdreq->cache_dma,
 480                                             creq->cache_ptr,
 481                                             CESA_TDMA_DST_IN_SRAM,
 482                                             flags);
 483         if (ret)
 484                 return ERR_PTR(ret);
 485
 486         if (!dma_iter->base.op_len) {
 487                 op = mv_cesa_dma_add_op(chain, &creq->op_tmpl, false, flags);
 488                 if (IS_ERR(op))
 489                         return op;
 490
 491                 mv_cesa_set_mac_op_frag_len(op, creq->cache_ptr);
 492
 493                 /* Add dummy desc to launch crypto operation */
 494                 ret = mv_cesa_dma_add_dummy_launch(chain, flags);
 495                 if (ret)
 496                         return ERR_PTR(ret);
 497         }
 498
 499         return op;
 500 }
 501
 502 static struct mv_cesa_op_ctx *
 503 mv_cesa_ahash_dma_add_data(struct mv_cesa_tdma_chain *chain,
 504                            struct mv_cesa_ahash_dma_iter *dma_iter,
 505                            struct mv_cesa_ahash_req *creq,
 506                            gfp_t flags)
 507 {
 508         struct mv_cesa_op_ctx *op;
 509         int ret;
 510
 511         op = mv_cesa_dma_add_op(chain, &creq->op_tmpl, false, flags);
 512         if (IS_ERR(op))
 513                 return op;
 514
 515         mv_cesa_set_mac_op_frag_len(op, dma_iter->base.op_len);
 516
 517         if ((mv_cesa_get_op_cfg(&creq->op_tmpl) & CESA_SA_DESC_CFG_FRAG_MSK) ==
 518             CESA_SA_DESC_CFG_FIRST_FRAG)
 519                 mv_cesa_update_op_cfg(&creq->op_tmpl,
 520                                       CESA_SA_DESC_CFG_MID_FRAG,
 521                                       CESA_SA_DESC_CFG_FRAG_MSK);
 522
 523         /* Add input transfers */
 524         ret = mv_cesa_dma_add_op_transfers(chain, &dma_iter->base,
 525                                            &dma_iter->src, flags);
 526         if (ret)
 527                 return ERR_PTR(ret);
 528
 529         /* Add dummy desc to launch crypto operation */
 530         ret = mv_cesa_dma_add_dummy_launch(chain, flags);
 531         if (ret)
 532                 return ERR_PTR(ret);
 533
 534         return op;
 535 }
 536
 537 static struct mv_cesa_op_ctx *
 538 mv_cesa_ahash_dma_last_req(struct mv_cesa_tdma_chain *chain,
 539                            struct mv_cesa_ahash_dma_iter *dma_iter,
 540                            struct mv_cesa_ahash_req *creq,
 541                            struct mv_cesa_op_ctx *op,
 542                            gfp_t flags)
 543 {
 544         struct mv_cesa_ahash_dma_req *ahashdreq = &creq->req.dma;
 545         unsigned int len, trailerlen, padoff = 0;
 546         int ret;
 547
 548         if (!creq->last_req)
 549                 return op;
 550
 551         if (op && creq->len <= CESA_SA_DESC_MAC_SRC_TOTAL_LEN_MAX) {
 552                 u32 frag = CESA_SA_DESC_CFG_NOT_FRAG;
 553
 554                 if ((mv_cesa_get_op_cfg(op) & CESA_SA_DESC_CFG_FRAG_MSK) !=
 555                     CESA_SA_DESC_CFG_FIRST_FRAG)
 556                         frag = CESA_SA_DESC_CFG_LAST_FRAG;
 557
 558                 mv_cesa_update_op_cfg(op, frag, CESA_SA_DESC_CFG_FRAG_MSK);
 559
 560                 return op;
 561         }
 562
 563         ret = mv_cesa_ahash_dma_alloc_padding(ahashdreq, flags);
 564         if (ret)
 565                 return ERR_PTR(ret);
 566
 567         trailerlen = mv_cesa_ahash_pad_req(creq, ahashdreq->padding);
 568
 569         if (op) {
 570                 len = min(CESA_SA_SRAM_PAYLOAD_SIZE - dma_iter->base.op_len,
 571                           trailerlen);
 572                 if (len) {
 573                         ret = mv_cesa_dma_add_data_transfer(chain,
 574                                                 CESA_SA_DATA_SRAM_OFFSET +
 575                                                 dma_iter->base.op_len,
 576                                                 ahashdreq->padding_dma,
 577                                                 len, CESA_TDMA_DST_IN_SRAM,
 578                                                 flags);
 579                         if (ret)
 580                                 return ERR_PTR(ret);
 581
 582                         mv_cesa_update_op_cfg(op, CESA_SA_DESC_CFG_MID_FRAG,
 583                                               CESA_SA_DESC_CFG_FRAG_MSK);
 584                         mv_cesa_set_mac_op_frag_len(op,
 585                                         dma_iter->base.op_len + len);
 586                         padoff += len;
 587                 }
 588         }
 589
 590         if (padoff >= trailerlen)
 591                 return op;
 592
 593         if ((mv_cesa_get_op_cfg(&creq->op_tmpl) & CESA_SA_DESC_CFG_FRAG_MSK) !=
 594             CESA_SA_DESC_CFG_FIRST_FRAG)
 595                 mv_cesa_update_op_cfg(&creq->op_tmpl,
 596                                       CESA_SA_DESC_CFG_MID_FRAG,
 597                                       CESA_SA_DESC_CFG_FRAG_MSK);
 598
 599         op = mv_cesa_dma_add_op(chain, &creq->op_tmpl, false, flags);
 600         if (IS_ERR(op))
 601                 return op;
 602
 603         mv_cesa_set_mac_op_frag_len(op, trailerlen - padoff);
 604
 605         ret = mv_cesa_dma_add_data_transfer(chain,
 606                                             CESA_SA_DATA_SRAM_OFFSET,
 607                                             ahashdreq->padding_dma +
 608                                             padoff,
 609                                             trailerlen - padoff,
 610                                             CESA_TDMA_DST_IN_SRAM,
 611                                             flags);
 612         if (ret)
 613                 return ERR_PTR(ret);
 614
 615         /* Add dummy desc to launch crypto operation */
 616         ret = mv_cesa_dma_add_dummy_launch(chain, flags);
 617         if (ret)
 618                 return ERR_PTR(ret);
 619
 620         return op;
 621 }
 622
 623 static int mv_cesa_ahash_dma_req_init(struct ahash_request *req)
 624 {
 625         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 626         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
 627                       GFP_KERNEL : GFP_ATOMIC;
 628         struct mv_cesa_ahash_dma_req *ahashdreq = &creq->req.dma;
 629         struct mv_cesa_tdma_req *dreq = &ahashdreq->base;
 630         struct mv_cesa_tdma_chain chain;
 631         struct mv_cesa_ahash_dma_iter iter;
 632         struct mv_cesa_op_ctx *op = NULL;
 633         int ret;
 634
 635         dreq->chain.first = NULL;
 636         dreq->chain.last = NULL;
 637
 638         if (creq->src_nents) {
 639                 ret = dma_map_sg(cesa_dev->dev, req->src, creq->src_nents,
 640                                  DMA_TO_DEVICE);
 641                 if (!ret) {
 642                         ret = -ENOMEM;
 643                         goto err;
 644                 }
 645         }
 646
 647         mv_cesa_tdma_desc_iter_init(&chain);
 648         mv_cesa_ahash_req_iter_init(&iter, req);
 649
 650         op = mv_cesa_ahash_dma_add_cache(&chain, &iter,
 651                                          creq, flags);
 652         if (IS_ERR(op)) {
 653                 ret = PTR_ERR(op);
 654                 goto err_free_tdma;
 655         }
 656
 657         do {
 658                 if (!iter.base.op_len)
 659                         break;
 660
 661                 op = mv_cesa_ahash_dma_add_data(&chain, &iter,
 662                                                 creq, flags);
 663                 if (IS_ERR(op)) {
 664                         ret = PTR_ERR(op);
 665                         goto err_free_tdma;
 666                 }
 667         } while (mv_cesa_ahash_req_iter_next_op(&iter));
 668
 669         op = mv_cesa_ahash_dma_last_req(&chain, &iter, creq, op, flags);
 670         if (IS_ERR(op)) {
 671                 ret = PTR_ERR(op);
 672                 goto err_free_tdma;
 673         }
 674
 675         if (op) {
 676                 /* Add dummy desc to wait for crypto operation end */
 677                 ret = mv_cesa_dma_add_dummy_end(&chain, flags);
 678                 if (ret)
 679                         goto err_free_tdma;
 680         }
 681
 682         if (!creq->last_req)
 683                 creq->cache_ptr = req->nbytes + creq->cache_ptr -
 684                                   iter.base.len;
 685         else
 686                 creq->cache_ptr = 0;
 687
 688         dreq->chain = chain;
 689
 690         return 0;
 691
 692 err_free_tdma:
 693         mv_cesa_dma_cleanup(dreq);
 694         dma_unmap_sg(cesa_dev->dev, req->src, creq->src_nents, DMA_TO_DEVICE);
 695
 696 err:
 697         mv_cesa_ahash_last_cleanup(req);
 698
 699         return ret;
 700 }
 701
 702 static int mv_cesa_ahash_req_init(struct ahash_request *req, bool *cached)
 703 {
 704         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 705         int ret;
 706
 707         if (cesa_dev->caps->has_tdma)
 708                 creq->req.base.type = CESA_DMA_REQ;
 709         else
 710                 creq->req.base.type = CESA_STD_REQ;
 711
 712         creq->src_nents = sg_nents_for_len(req->src, req->nbytes);
 713
 714         ret = mv_cesa_ahash_cache_req(req, cached);
 715         if (ret)
 716                 return ret;
 717
 718         if (*cached)
 719                 return 0;
 720
 721         if (creq->req.base.type == CESA_DMA_REQ)
 722                 ret = mv_cesa_ahash_dma_req_init(req);
 723
 724         return ret;
 725 }
 726
 727 static int mv_cesa_ahash_update(struct ahash_request *req)
 728 {
 729         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 730         bool cached = false;
 731         int ret;
 732
 733         creq->len += req->nbytes;
 734         ret = mv_cesa_ahash_req_init(req, &cached);
 735         if (ret)
 736                 return ret;
 737
 738         if (cached)
 739                 return 0;
 740
 741         ret = mv_cesa_queue_req(&req->base);
 742         if (ret && ret != -EINPROGRESS) {
 743                 mv_cesa_ahash_cleanup(req);
 744                 return ret;
 745         }
 746
 747         return ret;
 748 }
 749
 750 static int mv_cesa_ahash_final(struct ahash_request *req)
 751 {
 752         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 753         struct mv_cesa_op_ctx *tmpl = &creq->op_tmpl;
 754         bool cached = false;
 755         int ret;
 756
 757         mv_cesa_set_mac_op_total_len(tmpl, creq->len);
 758         creq->last_req = true;
 759         req->nbytes = 0;
 760
 761         ret = mv_cesa_ahash_req_init(req, &cached);
 762         if (ret)
 763                 return ret;
 764
 765         if (cached)
 766                 return 0;
 767
 768         ret = mv_cesa_queue_req(&req->base);
 769         if (ret && ret != -EINPROGRESS)
 770                 mv_cesa_ahash_cleanup(req);
 771
 772         return ret;
 773 }
 774
 775 static int mv_cesa_ahash_finup(struct ahash_request *req)
 776 {
 777         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 778         struct mv_cesa_op_ctx *tmpl = &creq->op_tmpl;
 779         bool cached = false;
 780         int ret;
 781
 782         creq->len += req->nbytes;
 783         mv_cesa_set_mac_op_total_len(tmpl, creq->len);
 784         creq->last_req = true;
 785
 786         ret = mv_cesa_ahash_req_init(req, &cached);
 787         if (ret)
 788                 return ret;
 789
 790         if (cached)
 791                 return 0;
 792
 793         ret = mv_cesa_queue_req(&req->base);
 794         if (ret && ret != -EINPROGRESS)
 795                 mv_cesa_ahash_cleanup(req);
 796
 797         return ret;
 798 }
 799
 800 static int mv_cesa_md5_init(struct ahash_request *req)
 801 {
 802         struct mv_cesa_op_ctx tmpl;
 803
 804         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_MD5);
 805
 806         mv_cesa_ahash_init(req, &tmpl);
 807
 808         return 0;
 809 }
 810
 811 static int mv_cesa_md5_export(struct ahash_request *req, void *out)
 812 {
 813         struct md5_state *out_state = out;
 814         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
 815         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 816         unsigned int digsize = crypto_ahash_digestsize(ahash);
 817
 818         out_state->byte_count = creq->len;
 819         memcpy(out_state->hash, creq->state, digsize);
 820         memset(out_state->block, 0, sizeof(out_state->block));
 821         if (creq->cache)
 822                 memcpy(out_state->block, creq->cache, creq->cache_ptr);
 823
 824         return 0;
 825 }
 826
 827 static int mv_cesa_md5_import(struct ahash_request *req, const void *in)
 828 {
 829         const struct md5_state *in_state = in;
 830         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
 831         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 832         unsigned int digsize = crypto_ahash_digestsize(ahash);
 833         unsigned int cache_ptr;
 834         int ret;
 835
 836         ret = crypto_ahash_init(req);
 837         if (ret)
 838                 return ret;
 839
 840         if (in_state->byte_count >= sizeof(in_state->block))
 841                 mv_cesa_update_op_cfg(&creq->op_tmpl,
 842                                       CESA_SA_DESC_CFG_MID_FRAG,
 843                                       CESA_SA_DESC_CFG_FRAG_MSK);
 844
 845         creq->len = in_state->byte_count;
 846         memcpy(creq->state, in_state->hash, digsize);
 847         creq->cache_ptr = 0;
 848
 849         cache_ptr = creq->len % sizeof(in_state->block);
 850         if (!cache_ptr)
 851                 return 0;
 852
 853         ret = mv_cesa_ahash_alloc_cache(req);
 854         if (ret)
 855                 return ret;
 856
 857         memcpy(creq->cache, in_state->block, cache_ptr);
 858         creq->cache_ptr = cache_ptr;
 859
 860         return 0;
 861 }
 862
 863 static int mv_cesa_md5_digest(struct ahash_request *req)
 864 {
 865         int ret;
 866
 867         ret = mv_cesa_md5_init(req);
 868         if (ret)
 869                 return ret;
 870
 871         return mv_cesa_ahash_finup(req);
 872 }
 873
 874 struct ahash_alg mv_md5_alg = {
 875         .init = mv_cesa_md5_init,
 876         .update = mv_cesa_ahash_update,
 877         .final = mv_cesa_ahash_final,
 878         .finup = mv_cesa_ahash_finup,
 879         .digest = mv_cesa_md5_digest,
 880         .export = mv_cesa_md5_export,
 881         .import = mv_cesa_md5_import,
 882         .halg = {
 883                 .digestsize = MD5_DIGEST_SIZE,
 884                 .statesize = sizeof(struct md5_state),
 885                 .base = {
 886                         .cra_name = "md5",
 887                         .cra_driver_name = "mv-md5",
 888                         .cra_priority = 300,
 889                         .cra_flags = CRYPTO_ALG_ASYNC |
 890                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
 891                         .cra_blocksize = MD5_HMAC_BLOCK_SIZE,
 892                         .cra_ctxsize = sizeof(struct mv_cesa_hash_ctx),
 893                         .cra_init = mv_cesa_ahash_cra_init,
 894                         .cra_module = THIS_MODULE,
 895                  }
 896         }
 897 };
 898
 899 static int mv_cesa_sha1_init(struct ahash_request *req)
 900 {
 901         struct mv_cesa_op_ctx tmpl;
 902
 903         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_SHA1);
 904
 905         mv_cesa_ahash_init(req, &tmpl);
 906
 907         return 0;
 908 }
 909
 910 static int mv_cesa_sha1_export(struct ahash_request *req, void *out)
 911 {
 912         struct sha1_state *out_state = out;
 913         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
 914         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 915         unsigned int digsize = crypto_ahash_digestsize(ahash);
 916
 917         out_state->count = creq->len;
 918         memcpy(out_state->state, creq->state, digsize);
 919         memset(out_state->buffer, 0, sizeof(out_state->buffer));
 920         if (creq->cache)
 921                 memcpy(out_state->buffer, creq->cache, creq->cache_ptr);
 922
 923         return 0;
 924 }
 925
 926 static int mv_cesa_sha1_import(struct ahash_request *req, const void *in)
 927 {
 928         const struct sha1_state *in_state = in;
 929         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
 930         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 931         unsigned int digsize = crypto_ahash_digestsize(ahash);
 932         unsigned int cache_ptr;
 933         int ret;
 934
 935         ret = crypto_ahash_init(req);
 936         if (ret)
 937                 return ret;
 938
 939         if (in_state->count >= SHA1_BLOCK_SIZE)
 940                 mv_cesa_update_op_cfg(&creq->op_tmpl,
 941                                       CESA_SA_DESC_CFG_MID_FRAG,
 942                                       CESA_SA_DESC_CFG_FRAG_MSK);
 943
 944         creq->len = in_state->count;
 945         memcpy(creq->state, in_state->state, digsize);
 946         creq->cache_ptr = 0;
 947
 948         cache_ptr = creq->len % SHA1_BLOCK_SIZE;
 949         if (!cache_ptr)
 950                 return 0;
 951
 952         ret = mv_cesa_ahash_alloc_cache(req);
 953         if (ret)
 954                 return ret;
 955
 956         memcpy(creq->cache, in_state->buffer, cache_ptr);
 957         creq->cache_ptr = cache_ptr;
 958
 959         return 0;
 960 }
 961
 962 static int mv_cesa_sha1_digest(struct ahash_request *req)
 963 {
 964         int ret;
 965
 966         ret = mv_cesa_sha1_init(req);
 967         if (ret)
 968                 return ret;
 969
 970         return mv_cesa_ahash_finup(req);
 971 }
 972
 973 struct ahash_alg mv_sha1_alg = {
 974         .init = mv_cesa_sha1_init,
 975         .update = mv_cesa_ahash_update,
 976         .final = mv_cesa_ahash_final,
 977         .finup = mv_cesa_ahash_finup,
 978         .digest = mv_cesa_sha1_digest,
 979         .export = mv_cesa_sha1_export,
 980         .import = mv_cesa_sha1_import,
 981         .halg = {
 982                 .digestsize = SHA1_DIGEST_SIZE,
 983                 .statesize = sizeof(struct sha1_state),
 984                 .base = {
 985                         .cra_name = "sha1",
 986                         .cra_driver_name = "mv-sha1",
 987                         .cra_priority = 300,
 988                         .cra_flags = CRYPTO_ALG_ASYNC |
 989                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
 990                         .cra_blocksize = SHA1_BLOCK_SIZE,
 991                         .cra_ctxsize = sizeof(struct mv_cesa_hash_ctx),
 992                         .cra_init = mv_cesa_ahash_cra_init,
 993                         .cra_module = THIS_MODULE,
 994                  }
 995         }
 996 };
 997
 998 static int mv_cesa_sha256_init(struct ahash_request *req)
 999 {
1000         struct mv_cesa_op_ctx tmpl;
1001
1002         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_SHA256);
1003
1004         mv_cesa_ahash_init(req, &tmpl);
1005
1006         return 0;
1007 }
1008
1009 static int mv_cesa_sha256_digest(struct ahash_request *req)
1010 {
1011         int ret;
1012
1013         ret = mv_cesa_sha256_init(req);
1014         if (ret)
1015                 return ret;
1016
1017         return mv_cesa_ahash_finup(req);
1018 }
1019
1020 static int mv_cesa_sha256_export(struct ahash_request *req, void *out)
1021 {
1022         struct sha256_state *out_state = out;
1023         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1024         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
1025         unsigned int ds = crypto_ahash_digestsize(ahash);
1026
1027         out_state->count = creq->len;
1028         memcpy(out_state->state, creq->state, ds);
1029         memset(out_state->buf, 0, sizeof(out_state->buf));
1030         if (creq->cache)
1031                 memcpy(out_state->buf, creq->cache, creq->cache_ptr);
1032
1033         return 0;
1034 }
1035
1036 static int mv_cesa_sha256_import(struct ahash_request *req, const void *in)
1037 {
1038         const struct sha256_state *in_state = in;
1039         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1040         struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
1041         unsigned int digsize = crypto_ahash_digestsize(ahash);
1042         unsigned int cache_ptr;
1043         int ret;
1044
1045         ret = crypto_ahash_init(req);
1046         if (ret)
1047                 return ret;
1048
1049         if (in_state->count >= SHA256_BLOCK_SIZE)
1050                 mv_cesa_update_op_cfg(&creq->op_tmpl,
1051                                       CESA_SA_DESC_CFG_MID_FRAG,
1052                                       CESA_SA_DESC_CFG_FRAG_MSK);
1053
1054         creq->len = in_state->count;
1055         memcpy(creq->state, in_state->state, digsize);
1056         creq->cache_ptr = 0;
1057
1058         cache_ptr = creq->len % SHA256_BLOCK_SIZE;
1059         if (!cache_ptr)
1060                 return 0;
1061
1062         ret = mv_cesa_ahash_alloc_cache(req);
1063         if (ret)
1064                 return ret;
1065
1066         memcpy(creq->cache, in_state->buf, cache_ptr);
1067         creq->cache_ptr = cache_ptr;
1068
1069         return 0;
1070 }
1071
1072 struct ahash_alg mv_sha256_alg = {
1073         .init = mv_cesa_sha256_init,
1074         .update = mv_cesa_ahash_update,
1075         .final = mv_cesa_ahash_final,
1076         .finup = mv_cesa_ahash_finup,
1077         .digest = mv_cesa_sha256_digest,
1078         .export = mv_cesa_sha256_export,
1079         .import = mv_cesa_sha256_import,
1080         .halg = {
1081                 .digestsize = SHA256_DIGEST_SIZE,
1082                 .statesize = sizeof(struct sha256_state),
1083                 .base = {
1084                         .cra_name = "sha256",
1085                         .cra_driver_name = "mv-sha256",
1086                         .cra_priority = 300,
1087                         .cra_flags = CRYPTO_ALG_ASYNC |
1088                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
1089                         .cra_blocksize = SHA256_BLOCK_SIZE,
1090                         .cra_ctxsize = sizeof(struct mv_cesa_hash_ctx),
1091                         .cra_init = mv_cesa_ahash_cra_init,
1092                         .cra_module = THIS_MODULE,
1093                  }
1094         }
1095 };
1096
1097 struct mv_cesa_ahash_result {
1098         struct completion completion;
1099         int error;
1100 };
1101
1102 static void mv_cesa_hmac_ahash_complete(struct crypto_async_request *req,
1103                                         int error)
1104 {
1105         struct mv_cesa_ahash_result *result = req->data;
1106
1107         if (error == -EINPROGRESS)
1108                 return;
1109
1110         result->error = error;
1111         complete(&result->completion);
1112 }
1113
1114 static int mv_cesa_ahmac_iv_state_init(struct ahash_request *req, u8 *pad,
1115                                        void *state, unsigned int blocksize)
1116 {
1117         struct mv_cesa_ahash_result result;
1118         struct scatterlist sg;
1119         int ret;
1120
1121         ahash_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
1122                                    mv_cesa_hmac_ahash_complete, &result);
1123         sg_init_one(&sg, pad, blocksize);
1124         ahash_request_set_crypt(req, &sg, pad, blocksize);
1125         init_completion(&result.completion);
1126
1127         ret = crypto_ahash_init(req);
1128         if (ret)
1129                 return ret;
1130
1131         ret = crypto_ahash_update(req);
1132         if (ret && ret != -EINPROGRESS)
1133                 return ret;
1134
1135         wait_for_completion_interruptible(&result.completion);
1136         if (result.error)
1137                 return result.error;
1138
1139         ret = crypto_ahash_export(req, state);
1140         if (ret)
1141                 return ret;
1142
1143         return 0;
1144 }
1145
1146 static int mv_cesa_ahmac_pad_init(struct ahash_request *req,
1147                                   const u8 *key, unsigned int keylen,
1148                                   u8 *ipad, u8 *opad,
1149                                   unsigned int blocksize)
1150 {
1151         struct mv_cesa_ahash_result result;
1152         struct scatterlist sg;
1153         int ret;
1154         int i;
1155
1156         if (keylen <= blocksize) {
1157                 memcpy(ipad, key, keylen);
1158         } else {
1159                 u8 *keydup = kmemdup(key, keylen, GFP_KERNEL);
1160
1161                 if (!keydup)
1162                         return -ENOMEM;
1163
1164                 ahash_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
1165                                            mv_cesa_hmac_ahash_complete,
1166                                            &result);
1167                 sg_init_one(&sg, keydup, keylen);
1168                 ahash_request_set_crypt(req, &sg, ipad, keylen);
1169                 init_completion(&result.completion);
1170
1171                 ret = crypto_ahash_digest(req);
1172                 if (ret == -EINPROGRESS) {
1173                         wait_for_completion_interruptible(&result.completion);
1174                         ret = result.error;
1175                 }
1176
1177                 /* Set the memory region to 0 to avoid any leak. */
1178                 memset(keydup, 0, keylen);
1179                 kfree(keydup);
1180
1181                 if (ret)
1182                         return ret;
1183
1184                 keylen = crypto_ahash_digestsize(crypto_ahash_reqtfm(req));
1185         }
1186
1187         memset(ipad + keylen, 0, blocksize - keylen);
1188         memcpy(opad, ipad, blocksize);
1189
1190         for (i = 0; i < blocksize; i++) {
1191                 ipad[i] ^= 0x36;
1192                 opad[i] ^= 0x5c;
1193         }
1194
1195         return 0;
1196 }
1197
1198 static int mv_cesa_ahmac_setkey(const char *hash_alg_name,
1199                                 const u8 *key, unsigned int keylen,
1200                                 void *istate, void *ostate)
1201 {
1202         struct ahash_request *req;
1203         struct crypto_ahash *tfm;
1204         unsigned int blocksize;
1205         u8 *ipad = NULL;
1206         u8 *opad;
1207         int ret;
1208
1209         tfm = crypto_alloc_ahash(hash_alg_name, CRYPTO_ALG_TYPE_AHASH,
1210                                  CRYPTO_ALG_TYPE_AHASH_MASK);
1211         if (IS_ERR(tfm))
1212                 return PTR_ERR(tfm);
1213
1214         req = ahash_request_alloc(tfm, GFP_KERNEL);
1215         if (!req) {
1216                 ret = -ENOMEM;
1217                 goto free_ahash;
1218         }
1219
1220         crypto_ahash_clear_flags(tfm, ~0);
1221
1222         blocksize = crypto_tfm_alg_blocksize(crypto_ahash_tfm(tfm));
1223
1224         ipad = kzalloc(2 * blocksize, GFP_KERNEL);
1225         if (!ipad) {
1226                 ret = -ENOMEM;
1227                 goto free_req;
1228         }
1229
1230         opad = ipad + blocksize;
1231
1232         ret = mv_cesa_ahmac_pad_init(req, key, keylen, ipad, opad, blocksize);
1233         if (ret)
1234                 goto free_ipad;
1235
1236         ret = mv_cesa_ahmac_iv_state_init(req, ipad, istate, blocksize);
1237         if (ret)
1238                 goto free_ipad;
1239
1240         ret = mv_cesa_ahmac_iv_state_init(req, opad, ostate, blocksize);
1241
1242 free_ipad:
1243         kfree(ipad);
1244 free_req:
1245         ahash_request_free(req);
1246 free_ahash:
1247         crypto_free_ahash(tfm);
1248
1249         return ret;
1250 }
1251
1252 static int mv_cesa_ahmac_cra_init(struct crypto_tfm *tfm)
1253 {
1254         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(tfm);
1255
1256         ctx->base.ops = &mv_cesa_ahash_req_ops;
1257
1258         crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
1259                                  sizeof(struct mv_cesa_ahash_req));
1260         return 0;
1261 }
1262
1263 static int mv_cesa_ahmac_md5_init(struct ahash_request *req)
1264 {
1265         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
1266         struct mv_cesa_op_ctx tmpl;
1267
1268         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_HMAC_MD5);
1269         memcpy(tmpl.ctx.hash.iv, ctx->iv, sizeof(ctx->iv));
1270
1271         mv_cesa_ahash_init(req, &tmpl);
1272
1273         return 0;
1274 }
1275
1276 static int mv_cesa_ahmac_md5_setkey(struct crypto_ahash *tfm, const u8 *key,
1277                                     unsigned int keylen)
1278 {
1279         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(crypto_ahash_tfm(tfm));
1280         struct md5_state istate, ostate;
1281         int ret, i;
1282
1283         ret = mv_cesa_ahmac_setkey("mv-md5", key, keylen, &istate, &ostate);
1284         if (ret)
1285                 return ret;
1286
1287         for (i = 0; i < ARRAY_SIZE(istate.hash); i++)
1288                 ctx->iv[i] = be32_to_cpu(istate.hash[i]);
1289
1290         for (i = 0; i < ARRAY_SIZE(ostate.hash); i++)
1291                 ctx->iv[i + 8] = be32_to_cpu(ostate.hash[i]);
1292
1293         return 0;
1294 }
1295
1296 static int mv_cesa_ahmac_md5_digest(struct ahash_request *req)
1297 {
1298         int ret;
1299
1300         ret = mv_cesa_ahmac_md5_init(req);
1301         if (ret)
1302                 return ret;
1303
1304         return mv_cesa_ahash_finup(req);
1305 }
1306
1307 struct ahash_alg mv_ahmac_md5_alg = {
1308         .init = mv_cesa_ahmac_md5_init,
1309         .update = mv_cesa_ahash_update,
1310         .final = mv_cesa_ahash_final,
1311         .finup = mv_cesa_ahash_finup,
1312         .digest = mv_cesa_ahmac_md5_digest,
1313         .setkey = mv_cesa_ahmac_md5_setkey,
1314         .export = mv_cesa_md5_export,
1315         .import = mv_cesa_md5_import,
1316         .halg = {
1317                 .digestsize = MD5_DIGEST_SIZE,
1318                 .statesize = sizeof(struct md5_state),
1319                 .base = {
1320                         .cra_name = "hmac(md5)",
1321                         .cra_driver_name = "mv-hmac-md5",
1322                         .cra_priority = 300,
1323                         .cra_flags = CRYPTO_ALG_ASYNC |
1324                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
1325                         .cra_blocksize = MD5_HMAC_BLOCK_SIZE,
1326                         .cra_ctxsize = sizeof(struct mv_cesa_hmac_ctx),
1327                         .cra_init = mv_cesa_ahmac_cra_init,
1328                         .cra_module = THIS_MODULE,
1329                  }
1330         }
1331 };
1332
1333 static int mv_cesa_ahmac_sha1_init(struct ahash_request *req)
1334 {
1335         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
1336         struct mv_cesa_op_ctx tmpl;
1337
1338         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_HMAC_SHA1);
1339         memcpy(tmpl.ctx.hash.iv, ctx->iv, sizeof(ctx->iv));
1340
1341         mv_cesa_ahash_init(req, &tmpl);
1342
1343         return 0;
1344 }
1345
1346 static int mv_cesa_ahmac_sha1_setkey(struct crypto_ahash *tfm, const u8 *key,
1347                                      unsigned int keylen)
1348 {
1349         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(crypto_ahash_tfm(tfm));
1350         struct sha1_state istate, ostate;
1351         int ret, i;
1352
1353         ret = mv_cesa_ahmac_setkey("mv-sha1", key, keylen, &istate, &ostate);
1354         if (ret)
1355                 return ret;
1356
1357         for (i = 0; i < ARRAY_SIZE(istate.state); i++)
1358                 ctx->iv[i] = be32_to_cpu(istate.state[i]);
1359
1360         for (i = 0; i < ARRAY_SIZE(ostate.state); i++)
1361                 ctx->iv[i + 8] = be32_to_cpu(ostate.state[i]);
1362
1363         return 0;
1364 }
1365
1366 static int mv_cesa_ahmac_sha1_digest(struct ahash_request *req)
1367 {
1368         int ret;
1369
1370         ret = mv_cesa_ahmac_sha1_init(req);
1371         if (ret)
1372                 return ret;
1373
1374         return mv_cesa_ahash_finup(req);
1375 }
1376
1377 struct ahash_alg mv_ahmac_sha1_alg = {
1378         .init = mv_cesa_ahmac_sha1_init,
1379         .update = mv_cesa_ahash_update,
1380         .final = mv_cesa_ahash_final,
1381         .finup = mv_cesa_ahash_finup,
1382         .digest = mv_cesa_ahmac_sha1_digest,
1383         .setkey = mv_cesa_ahmac_sha1_setkey,
1384         .export = mv_cesa_sha1_export,
1385         .import = mv_cesa_sha1_import,
1386         .halg = {
1387                 .digestsize = SHA1_DIGEST_SIZE,
1388                 .statesize = sizeof(struct sha1_state),
1389                 .base = {
1390                         .cra_name = "hmac(sha1)",
1391                         .cra_driver_name = "mv-hmac-sha1",
1392                         .cra_priority = 300,
1393                         .cra_flags = CRYPTO_ALG_ASYNC |
1394                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
1395                         .cra_blocksize = SHA1_BLOCK_SIZE,
1396                         .cra_ctxsize = sizeof(struct mv_cesa_hmac_ctx),
1397                         .cra_init = mv_cesa_ahmac_cra_init,
1398                         .cra_module = THIS_MODULE,
1399                  }
1400         }
1401 };
1402
1403 static int mv_cesa_ahmac_sha256_setkey(struct crypto_ahash *tfm, const u8 *key,
1404                                        unsigned int keylen)
1405 {
1406         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(crypto_ahash_tfm(tfm));
1407         struct sha256_state istate, ostate;
1408         int ret, i;
1409
1410         ret = mv_cesa_ahmac_setkey("mv-sha256", key, keylen, &istate, &ostate);
1411         if (ret)
1412                 return ret;
1413
1414         for (i = 0; i < ARRAY_SIZE(istate.state); i++)
1415                 ctx->iv[i] = be32_to_cpu(istate.state[i]);
1416
1417         for (i = 0; i < ARRAY_SIZE(ostate.state); i++)
1418                 ctx->iv[i + 8] = be32_to_cpu(ostate.state[i]);
1419
1420         return 0;
1421 }
1422
1423 static int mv_cesa_ahmac_sha256_init(struct ahash_request *req)
1424 {
1425         struct mv_cesa_hmac_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
1426         struct mv_cesa_op_ctx tmpl;
1427
1428         mv_cesa_set_op_cfg(&tmpl, CESA_SA_DESC_CFG_MACM_HMAC_SHA256);
1429         memcpy(tmpl.ctx.hash.iv, ctx->iv, sizeof(ctx->iv));
1430
1431         mv_cesa_ahash_init(req, &tmpl);
1432
1433         return 0;
1434 }
1435
1436 static int mv_cesa_ahmac_sha256_digest(struct ahash_request *req)
1437 {
1438         int ret;
1439
1440         ret = mv_cesa_ahmac_sha256_init(req);
1441         if (ret)
1442                 return ret;
1443
1444         return mv_cesa_ahash_finup(req);
1445 }
1446
1447 struct ahash_alg mv_ahmac_sha256_alg = {
1448         .init = mv_cesa_ahmac_sha256_init,
1449         .update = mv_cesa_ahash_update,
1450         .final = mv_cesa_ahash_final,
1451         .finup = mv_cesa_ahash_finup,
1452         .digest = mv_cesa_ahmac_sha256_digest,
1453         .setkey = mv_cesa_ahmac_sha256_setkey,
1454         .export = mv_cesa_sha256_export,
1455         .import = mv_cesa_sha256_import,
1456         .halg = {
1457                 .digestsize = SHA256_DIGEST_SIZE,
1458                 .statesize = sizeof(struct sha256_state),
1459                 .base = {
1460                         .cra_name = "hmac(sha256)",
1461                         .cra_driver_name = "mv-hmac-sha256",
1462                         .cra_priority = 300,
1463                         .cra_flags = CRYPTO_ALG_ASYNC |
1464                                      CRYPTO_ALG_KERN_DRIVER_ONLY,
1465                         .cra_blocksize = SHA256_BLOCK_SIZE,
1466                         .cra_ctxsize = sizeof(struct mv_cesa_hmac_ctx),
1467                         .cra_init = mv_cesa_ahmac_cra_init,
1468                         .cra_module = THIS_MODULE,
1469                  }
1470         }
1471 };