drivers/crypto/allwinner/sun4i-ss/sun4i-ss-cipher.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * sun4i-ss-cipher.c - hardware cryptographic accelerator for Allwinner A20 SoC
   4  *
   5  * Copyright (C) 2013-2015 Corentin LABBE <clabbe.montjoie@gmail.com>
   6  *
   7  * This file add support for AES cipher with 128,192,256 bits
   8  * keysize in CBC and ECB mode.
   9  * Add support also for DES and 3DES in CBC and ECB mode.
  10  *
  11  * You could find the datasheet in Documentation/arm/sunxi.rst
  12  */
  13 #include "sun4i-ss.h"
  14
  15 static int noinline_for_stack sun4i_ss_opti_poll(struct skcipher_request *areq)
  16 {
  17         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
  18         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
  19         struct sun4i_ss_ctx *ss = op->ss;
  20         unsigned int ivsize = crypto_skcipher_ivsize(tfm);
  21         struct sun4i_cipher_req_ctx *ctx = skcipher_request_ctx(areq);
  22         u32 mode = ctx->mode;
  23         void *backup_iv = NULL;
  24         /* when activating SS, the default FIFO space is SS_RX_DEFAULT(32) */
  25         u32 rx_cnt = SS_RX_DEFAULT;
  26         u32 tx_cnt = 0;
  27         u32 spaces;
  28         u32 v;
  29         int err = 0;
  30         unsigned int i;
  31         unsigned int ileft = areq->cryptlen;
  32         unsigned int oleft = areq->cryptlen;
  33         unsigned int todo;
  34         unsigned long pi = 0, po = 0; /* progress for in and out */
  35         bool miter_err;
  36         struct sg_mapping_iter mi, mo;
  37         unsigned int oi, oo; /* offset for in and out */
  38         unsigned long flags;
  39         struct skcipher_alg *alg = crypto_skcipher_alg(tfm);
  40         struct sun4i_ss_alg_template *algt;
  41
  42         if (!areq->cryptlen)
  43                 return 0;
  44
  45         if (!areq->src || !areq->dst) {
  46                 dev_err_ratelimited(ss->dev, "ERROR: Some SGs are NULL\n");
  47                 return -EINVAL;
  48         }
  49
  50         if (areq->iv && ivsize > 0 && mode & SS_DECRYPTION) {
  51                 backup_iv = kzalloc(ivsize, GFP_KERNEL);
  52                 if (!backup_iv)
  53                         return -ENOMEM;
  54                 scatterwalk_map_and_copy(backup_iv, areq->src, areq->cryptlen - ivsize, ivsize, 0);
  55         }
  56
  57         if (IS_ENABLED(CONFIG_CRYPTO_DEV_SUN4I_SS_DEBUG)) {
  58                 algt = container_of(alg, struct sun4i_ss_alg_template, alg.crypto);
  59                 algt->stat_opti++;
  60                 algt->stat_bytes += areq->cryptlen;
  61         }
  62
  63         spin_lock_irqsave(&ss->slock, flags);
  64
  65         for (i = 0; i < op->keylen / 4; i++)
  66                 writesl(ss->base + SS_KEY0 + i * 4, &op->key[i], 1);
  67
  68         if (areq->iv) {
  69                 for (i = 0; i < 4 && i < ivsize / 4; i++) {
  70                         v = *(u32 *)(areq->iv + i * 4);
  71                         writesl(ss->base + SS_IV0 + i * 4, &v, 1);
  72                 }
  73         }
  74         writel(mode, ss->base + SS_CTL);
  75
  76
  77         ileft = areq->cryptlen / 4;
  78         oleft = areq->cryptlen / 4;
  79         oi = 0;
  80         oo = 0;
  81         do {
  82                 if (ileft) {
  83                         sg_miter_start(&mi, areq->src, sg_nents(areq->src),
  84                                         SG_MITER_FROM_SG | SG_MITER_ATOMIC);
  85                         if (pi)
  86                                 sg_miter_skip(&mi, pi);
  87                         miter_err = sg_miter_next(&mi);
  88                         if (!miter_err || !mi.addr) {
  89                                 dev_err_ratelimited(ss->dev, "ERROR: sg_miter return null\n");
  90                                 err = -EINVAL;
  91                                 goto release_ss;
  92                         }
  93                         todo = min(rx_cnt, ileft);
  94                         todo = min_t(size_t, todo, (mi.length - oi) / 4);
  95                         if (todo) {
  96                                 ileft -= todo;
  97                                 writesl(ss->base + SS_RXFIFO, mi.addr + oi, todo);
  98                                 oi += todo * 4;
  99                         }
 100                         if (oi == mi.length) {
 101                                 pi += mi.length;
 102                                 oi = 0;
 103                         }
 104                         sg_miter_stop(&mi);
 105                 }
 106
 107                 spaces = readl(ss->base + SS_FCSR);
 108                 rx_cnt = SS_RXFIFO_SPACES(spaces);
 109                 tx_cnt = SS_TXFIFO_SPACES(spaces);
 110
 111                 sg_miter_start(&mo, areq->dst, sg_nents(areq->dst),
 112                                SG_MITER_TO_SG | SG_MITER_ATOMIC);
 113                 if (po)
 114                         sg_miter_skip(&mo, po);
 115                 miter_err = sg_miter_next(&mo);
 116                 if (!miter_err || !mo.addr) {
 117                         dev_err_ratelimited(ss->dev, "ERROR: sg_miter return null\n");
 118                         err = -EINVAL;
 119                         goto release_ss;
 120                 }
 121                 todo = min(tx_cnt, oleft);
 122                 todo = min_t(size_t, todo, (mo.length - oo) / 4);
 123                 if (todo) {
 124                         oleft -= todo;
 125                         readsl(ss->base + SS_TXFIFO, mo.addr + oo, todo);
 126                         oo += todo * 4;
 127                 }
 128                 if (oo == mo.length) {
 129                         oo = 0;
 130                         po += mo.length;
 131                 }
 132                 sg_miter_stop(&mo);
 133         } while (oleft);
 134
 135         if (areq->iv) {
 136                 if (mode & SS_DECRYPTION) {
 137                         memcpy(areq->iv, backup_iv, ivsize);
 138                         kfree_sensitive(backup_iv);
 139                 } else {
 140                         scatterwalk_map_and_copy(areq->iv, areq->dst, areq->cryptlen - ivsize,
 141                                                  ivsize, 0);
 142                 }
 143         }
 144
 145 release_ss:
 146         writel(0, ss->base + SS_CTL);
 147         spin_unlock_irqrestore(&ss->slock, flags);
 148         return err;
 149 }
 150
 151 static int noinline_for_stack sun4i_ss_cipher_poll_fallback(struct skcipher_request *areq)
 152 {
 153         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 154         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 155         struct sun4i_cipher_req_ctx *ctx = skcipher_request_ctx(areq);
 156         int err;
 157         struct skcipher_alg *alg = crypto_skcipher_alg(tfm);
 158         struct sun4i_ss_alg_template *algt;
 159
 160         if (IS_ENABLED(CONFIG_CRYPTO_DEV_SUN4I_SS_DEBUG)) {
 161                 algt = container_of(alg, struct sun4i_ss_alg_template, alg.crypto);
 162                 algt->stat_fb++;
 163         }
 164
 165         skcipher_request_set_tfm(&ctx->fallback_req, op->fallback_tfm);
 166         skcipher_request_set_callback(&ctx->fallback_req, areq->base.flags,
 167                                       areq->base.complete, areq->base.data);
 168         skcipher_request_set_crypt(&ctx->fallback_req, areq->src, areq->dst,
 169                                    areq->cryptlen, areq->iv);
 170         if (ctx->mode & SS_DECRYPTION)
 171                 err = crypto_skcipher_decrypt(&ctx->fallback_req);
 172         else
 173                 err = crypto_skcipher_encrypt(&ctx->fallback_req);
 174
 175         return err;
 176 }
 177
 178 /* Generic function that support SG with size not multiple of 4 */
 179 static int sun4i_ss_cipher_poll(struct skcipher_request *areq)
 180 {
 181         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 182         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 183         struct sun4i_ss_ctx *ss = op->ss;
 184         int no_chunk = 1;
 185         struct scatterlist *in_sg = areq->src;
 186         struct scatterlist *out_sg = areq->dst;
 187         unsigned int ivsize = crypto_skcipher_ivsize(tfm);
 188         struct sun4i_cipher_req_ctx *ctx = skcipher_request_ctx(areq);
 189         struct skcipher_alg *alg = crypto_skcipher_alg(tfm);
 190         struct sun4i_ss_alg_template *algt;
 191         u32 mode = ctx->mode;
 192         /* when activating SS, the default FIFO space is SS_RX_DEFAULT(32) */
 193         u32 rx_cnt = SS_RX_DEFAULT;
 194         u32 tx_cnt = 0;
 195         u32 v;
 196         u32 spaces;
 197         int err = 0;
 198         unsigned int i;
 199         unsigned int ileft = areq->cryptlen;
 200         unsigned int oleft = areq->cryptlen;
 201         unsigned int todo;
 202         void *backup_iv = NULL;
 203         struct sg_mapping_iter mi, mo;
 204         unsigned long pi = 0, po = 0; /* progress for in and out */
 205         bool miter_err;
 206         unsigned int oi, oo;    /* offset for in and out */
 207         unsigned int ob = 0;    /* offset in buf */
 208         unsigned int obo = 0;   /* offset in bufo*/
 209         unsigned int obl = 0;   /* length of data in bufo */
 210         unsigned long flags;
 211         bool need_fallback = false;
 212
 213         if (!areq->cryptlen)
 214                 return 0;
 215
 216         if (!areq->src || !areq->dst) {
 217                 dev_err_ratelimited(ss->dev, "ERROR: Some SGs are NULL\n");
 218                 return -EINVAL;
 219         }
 220
 221         algt = container_of(alg, struct sun4i_ss_alg_template, alg.crypto);
 222         if (areq->cryptlen % algt->alg.crypto.base.cra_blocksize)
 223                 need_fallback = true;
 224
 225         /*
 226          * if we have only SGs with size multiple of 4,
 227          * we can use the SS optimized function
 228          */
 229         while (in_sg && no_chunk == 1) {
 230                 if ((in_sg->length | in_sg->offset) & 3u)
 231                         no_chunk = 0;
 232                 in_sg = sg_next(in_sg);
 233         }
 234         while (out_sg && no_chunk == 1) {
 235                 if ((out_sg->length | out_sg->offset) & 3u)
 236                         no_chunk = 0;
 237                 out_sg = sg_next(out_sg);
 238         }
 239
 240         if (no_chunk == 1 && !need_fallback)
 241                 return sun4i_ss_opti_poll(areq);
 242
 243         if (need_fallback)
 244                 return sun4i_ss_cipher_poll_fallback(areq);
 245
 246         if (areq->iv && ivsize > 0 && mode & SS_DECRYPTION) {
 247                 backup_iv = kzalloc(ivsize, GFP_KERNEL);
 248                 if (!backup_iv)
 249                         return -ENOMEM;
 250                 scatterwalk_map_and_copy(backup_iv, areq->src, areq->cryptlen - ivsize, ivsize, 0);
 251         }
 252
 253         if (IS_ENABLED(CONFIG_CRYPTO_DEV_SUN4I_SS_DEBUG)) {
 254                 algt->stat_req++;
 255                 algt->stat_bytes += areq->cryptlen;
 256         }
 257
 258         spin_lock_irqsave(&ss->slock, flags);
 259
 260         for (i = 0; i < op->keylen / 4; i++)
 261                 writesl(ss->base + SS_KEY0 + i * 4, &op->key[i], 1);
 262
 263         if (areq->iv) {
 264                 for (i = 0; i < 4 && i < ivsize / 4; i++) {
 265                         v = *(u32 *)(areq->iv + i * 4);
 266                         writesl(ss->base + SS_IV0 + i * 4, &v, 1);
 267                 }
 268         }
 269         writel(mode, ss->base + SS_CTL);
 270
 271         ileft = areq->cryptlen;
 272         oleft = areq->cryptlen;
 273         oi = 0;
 274         oo = 0;
 275
 276         while (oleft) {
 277                 if (ileft) {
 278                         sg_miter_start(&mi, areq->src, sg_nents(areq->src),
 279                                        SG_MITER_FROM_SG | SG_MITER_ATOMIC);
 280                         if (pi)
 281                                 sg_miter_skip(&mi, pi);
 282                         miter_err = sg_miter_next(&mi);
 283                         if (!miter_err || !mi.addr) {
 284                                 dev_err_ratelimited(ss->dev, "ERROR: sg_miter return null\n");
 285                                 err = -EINVAL;
 286                                 goto release_ss;
 287                         }
 288                         /*
 289                          * todo is the number of consecutive 4byte word that we
 290                          * can read from current SG
 291                          */
 292                         todo = min(rx_cnt, ileft / 4);
 293                         todo = min_t(size_t, todo, (mi.length - oi) / 4);
 294                         if (todo && !ob) {
 295                                 writesl(ss->base + SS_RXFIFO, mi.addr + oi,
 296                                         todo);
 297                                 ileft -= todo * 4;
 298                                 oi += todo * 4;
 299                         } else {
 300                                 /*
 301                                  * not enough consecutive bytes, so we need to
 302                                  * linearize in buf. todo is in bytes
 303                                  * After that copy, if we have a multiple of 4
 304                                  * we need to be able to write all buf in one
 305                                  * pass, so it is why we min() with rx_cnt
 306                                  */
 307                                 todo = min(rx_cnt * 4 - ob, ileft);
 308                                 todo = min_t(size_t, todo, mi.length - oi);
 309                                 memcpy(ss->buf + ob, mi.addr + oi, todo);
 310                                 ileft -= todo;
 311                                 oi += todo;
 312                                 ob += todo;
 313                                 if (!(ob % 4)) {
 314                                         writesl(ss->base + SS_RXFIFO, ss->buf,
 315                                                 ob / 4);
 316                                         ob = 0;
 317                                 }
 318                         }
 319                         if (oi == mi.length) {
 320                                 pi += mi.length;
 321                                 oi = 0;
 322                         }
 323                         sg_miter_stop(&mi);
 324                 }
 325
 326                 spaces = readl(ss->base + SS_FCSR);
 327                 rx_cnt = SS_RXFIFO_SPACES(spaces);
 328                 tx_cnt = SS_TXFIFO_SPACES(spaces);
 329
 330                 if (!tx_cnt)
 331                         continue;
 332                 sg_miter_start(&mo, areq->dst, sg_nents(areq->dst),
 333                                SG_MITER_TO_SG | SG_MITER_ATOMIC);
 334                 if (po)
 335                         sg_miter_skip(&mo, po);
 336                 miter_err = sg_miter_next(&mo);
 337                 if (!miter_err || !mo.addr) {
 338                         dev_err_ratelimited(ss->dev, "ERROR: sg_miter return null\n");
 339                         err = -EINVAL;
 340                         goto release_ss;
 341                 }
 342                 /* todo in 4bytes word */
 343                 todo = min(tx_cnt, oleft / 4);
 344                 todo = min_t(size_t, todo, (mo.length - oo) / 4);
 345
 346                 if (todo) {
 347                         readsl(ss->base + SS_TXFIFO, mo.addr + oo, todo);
 348                         oleft -= todo * 4;
 349                         oo += todo * 4;
 350                         if (oo == mo.length) {
 351                                 po += mo.length;
 352                                 oo = 0;
 353                         }
 354                 } else {
 355                         /*
 356                          * read obl bytes in bufo, we read at maximum for
 357                          * emptying the device
 358                          */
 359                         readsl(ss->base + SS_TXFIFO, ss->bufo, tx_cnt);
 360                         obl = tx_cnt * 4;
 361                         obo = 0;
 362                         do {
 363                                 /*
 364                                  * how many bytes we can copy ?
 365                                  * no more than remaining SG size
 366                                  * no more than remaining buffer
 367                                  * no need to test against oleft
 368                                  */
 369                                 todo = min_t(size_t,
 370                                              mo.length - oo, obl - obo);
 371                                 memcpy(mo.addr + oo, ss->bufo + obo, todo);
 372                                 oleft -= todo;
 373                                 obo += todo;
 374                                 oo += todo;
 375                                 if (oo == mo.length) {
 376                                         po += mo.length;
 377                                         sg_miter_next(&mo);
 378                                         oo = 0;
 379                                 }
 380                         } while (obo < obl);
 381                         /* bufo must be fully used here */
 382                 }
 383                 sg_miter_stop(&mo);
 384         }
 385         if (areq->iv) {
 386                 if (mode & SS_DECRYPTION) {
 387                         memcpy(areq->iv, backup_iv, ivsize);
 388                         kfree_sensitive(backup_iv);
 389                 } else {
 390                         scatterwalk_map_and_copy(areq->iv, areq->dst, areq->cryptlen - ivsize,
 391                                                  ivsize, 0);
 392                 }
 393         }
 394
 395 release_ss:
 396         writel(0, ss->base + SS_CTL);
 397         spin_unlock_irqrestore(&ss->slock, flags);
 398
 399         return err;
 400 }
 401
 402 /* CBC AES */
 403 int sun4i_ss_cbc_aes_encrypt(struct skcipher_request *areq)
 404 {
 405         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 406         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 407         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 408
 409         rctx->mode = SS_OP_AES | SS_CBC | SS_ENABLED | SS_ENCRYPTION |
 410                 op->keymode;
 411         return sun4i_ss_cipher_poll(areq);
 412 }
 413
 414 int sun4i_ss_cbc_aes_decrypt(struct skcipher_request *areq)
 415 {
 416         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 417         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 418         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 419
 420         rctx->mode = SS_OP_AES | SS_CBC | SS_ENABLED | SS_DECRYPTION |
 421                 op->keymode;
 422         return sun4i_ss_cipher_poll(areq);
 423 }
 424
 425 /* ECB AES */
 426 int sun4i_ss_ecb_aes_encrypt(struct skcipher_request *areq)
 427 {
 428         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 429         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 430         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 431
 432         rctx->mode = SS_OP_AES | SS_ECB | SS_ENABLED | SS_ENCRYPTION |
 433                 op->keymode;
 434         return sun4i_ss_cipher_poll(areq);
 435 }
 436
 437 int sun4i_ss_ecb_aes_decrypt(struct skcipher_request *areq)
 438 {
 439         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 440         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 441         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 442
 443         rctx->mode = SS_OP_AES | SS_ECB | SS_ENABLED | SS_DECRYPTION |
 444                 op->keymode;
 445         return sun4i_ss_cipher_poll(areq);
 446 }
 447
 448 /* CBC DES */
 449 int sun4i_ss_cbc_des_encrypt(struct skcipher_request *areq)
 450 {
 451         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 452         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 453         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 454
 455         rctx->mode = SS_OP_DES | SS_CBC | SS_ENABLED | SS_ENCRYPTION |
 456                 op->keymode;
 457         return sun4i_ss_cipher_poll(areq);
 458 }
 459
 460 int sun4i_ss_cbc_des_decrypt(struct skcipher_request *areq)
 461 {
 462         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 463         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 464         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 465
 466         rctx->mode = SS_OP_DES | SS_CBC | SS_ENABLED | SS_DECRYPTION |
 467                 op->keymode;
 468         return sun4i_ss_cipher_poll(areq);
 469 }
 470
 471 /* ECB DES */
 472 int sun4i_ss_ecb_des_encrypt(struct skcipher_request *areq)
 473 {
 474         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 475         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 476         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 477
 478         rctx->mode = SS_OP_DES | SS_ECB | SS_ENABLED | SS_ENCRYPTION |
 479                 op->keymode;
 480         return sun4i_ss_cipher_poll(areq);
 481 }
 482
 483 int sun4i_ss_ecb_des_decrypt(struct skcipher_request *areq)
 484 {
 485         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 486         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 487         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 488
 489         rctx->mode = SS_OP_DES | SS_ECB | SS_ENABLED | SS_DECRYPTION |
 490                 op->keymode;
 491         return sun4i_ss_cipher_poll(areq);
 492 }
 493
 494 /* CBC 3DES */
 495 int sun4i_ss_cbc_des3_encrypt(struct skcipher_request *areq)
 496 {
 497         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 498         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 499         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 500
 501         rctx->mode = SS_OP_3DES | SS_CBC | SS_ENABLED | SS_ENCRYPTION |
 502                 op->keymode;
 503         return sun4i_ss_cipher_poll(areq);
 504 }
 505
 506 int sun4i_ss_cbc_des3_decrypt(struct skcipher_request *areq)
 507 {
 508         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 509         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 510         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 511
 512         rctx->mode = SS_OP_3DES | SS_CBC | SS_ENABLED | SS_DECRYPTION |
 513                 op->keymode;
 514         return sun4i_ss_cipher_poll(areq);
 515 }
 516
 517 /* ECB 3DES */
 518 int sun4i_ss_ecb_des3_encrypt(struct skcipher_request *areq)
 519 {
 520         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 521         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 522         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 523
 524         rctx->mode = SS_OP_3DES | SS_ECB | SS_ENABLED | SS_ENCRYPTION |
 525                 op->keymode;
 526         return sun4i_ss_cipher_poll(areq);
 527 }
 528
 529 int sun4i_ss_ecb_des3_decrypt(struct skcipher_request *areq)
 530 {
 531         struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(areq);
 532         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 533         struct sun4i_cipher_req_ctx *rctx = skcipher_request_ctx(areq);
 534
 535         rctx->mode = SS_OP_3DES | SS_ECB | SS_ENABLED | SS_DECRYPTION |
 536                 op->keymode;
 537         return sun4i_ss_cipher_poll(areq);
 538 }
 539
 540 int sun4i_ss_cipher_init(struct crypto_tfm *tfm)
 541 {
 542         struct sun4i_tfm_ctx *op = crypto_tfm_ctx(tfm);
 543         struct sun4i_ss_alg_template *algt;
 544         const char *name = crypto_tfm_alg_name(tfm);
 545         int err;
 546
 547         memset(op, 0, sizeof(struct sun4i_tfm_ctx));
 548
 549         algt = container_of(tfm->__crt_alg, struct sun4i_ss_alg_template,
 550                             alg.crypto.base);
 551         op->ss = algt->ss;
 552
 553         op->fallback_tfm = crypto_alloc_skcipher(name, 0, CRYPTO_ALG_NEED_FALLBACK);
 554         if (IS_ERR(op->fallback_tfm)) {
 555                 dev_err(op->ss->dev, "ERROR: Cannot allocate fallback for %s %ld\n",
 556                         name, PTR_ERR(op->fallback_tfm));
 557                 return PTR_ERR(op->fallback_tfm);
 558         }
 559
 560         crypto_skcipher_set_reqsize(__crypto_skcipher_cast(tfm),
 561                                     sizeof(struct sun4i_cipher_req_ctx) +
 562                                     crypto_skcipher_reqsize(op->fallback_tfm));
 563
 564         err = pm_runtime_resume_and_get(op->ss->dev);
 565         if (err < 0)
 566                 goto error_pm;
 567
 568         return 0;
 569 error_pm:
 570         crypto_free_skcipher(op->fallback_tfm);
 571         return err;
 572 }
 573
 574 void sun4i_ss_cipher_exit(struct crypto_tfm *tfm)
 575 {
 576         struct sun4i_tfm_ctx *op = crypto_tfm_ctx(tfm);
 577
 578         crypto_free_skcipher(op->fallback_tfm);
 579         pm_runtime_put(op->ss->dev);
 580 }
 581
 582 /* check and set the AES key, prepare the mode to be used */
 583 int sun4i_ss_aes_setkey(struct crypto_skcipher *tfm, const u8 *key,
 584                         unsigned int keylen)
 585 {
 586         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 587         struct sun4i_ss_ctx *ss = op->ss;
 588
 589         switch (keylen) {
 590         case 128 / 8:
 591                 op->keymode = SS_AES_128BITS;
 592                 break;
 593         case 192 / 8:
 594                 op->keymode = SS_AES_192BITS;
 595                 break;
 596         case 256 / 8:
 597                 op->keymode = SS_AES_256BITS;
 598                 break;
 599         default:
 600                 dev_dbg(ss->dev, "ERROR: Invalid keylen %u\n", keylen);
 601                 return -EINVAL;
 602         }
 603         op->keylen = keylen;
 604         memcpy(op->key, key, keylen);
 605
 606         crypto_skcipher_clear_flags(op->fallback_tfm, CRYPTO_TFM_REQ_MASK);
 607         crypto_skcipher_set_flags(op->fallback_tfm, tfm->base.crt_flags & CRYPTO_TFM_REQ_MASK);
 608
 609         return crypto_skcipher_setkey(op->fallback_tfm, key, keylen);
 610 }
 611
 612 /* check and set the DES key, prepare the mode to be used */
 613 int sun4i_ss_des_setkey(struct crypto_skcipher *tfm, const u8 *key,
 614                         unsigned int keylen)
 615 {
 616         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 617         int err;
 618
 619         err = verify_skcipher_des_key(tfm, key);
 620         if (err)
 621                 return err;
 622
 623         op->keylen = keylen;
 624         memcpy(op->key, key, keylen);
 625
 626         crypto_skcipher_clear_flags(op->fallback_tfm, CRYPTO_TFM_REQ_MASK);
 627         crypto_skcipher_set_flags(op->fallback_tfm, tfm->base.crt_flags & CRYPTO_TFM_REQ_MASK);
 628
 629         return crypto_skcipher_setkey(op->fallback_tfm, key, keylen);
 630 }
 631
 632 /* check and set the 3DES key, prepare the mode to be used */
 633 int sun4i_ss_des3_setkey(struct crypto_skcipher *tfm, const u8 *key,
 634                          unsigned int keylen)
 635 {
 636         struct sun4i_tfm_ctx *op = crypto_skcipher_ctx(tfm);
 637         int err;
 638
 639         err = verify_skcipher_des3_key(tfm, key);
 640         if (err)
 641                 return err;
 642
 643         op->keylen = keylen;
 644         memcpy(op->key, key, keylen);
 645
 646         crypto_skcipher_clear_flags(op->fallback_tfm, CRYPTO_TFM_REQ_MASK);
 647         crypto_skcipher_set_flags(op->fallback_tfm, tfm->base.crt_flags & CRYPTO_TFM_REQ_MASK);
 648
 649         return crypto_skcipher_setkey(op->fallback_tfm, key, keylen);
 650 }