drivers/crypto/marvell/cesa/hash.c

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