drivers/crypto/sahara.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Cryptographic API.
   4  *
   5  * Support for SAHARA cryptographic accelerator.
   6  *
   7  * Copyright (c) 2014 Steffen Trumtrar <s.trumtrar@pengutronix.de>
   8  * Copyright (c) 2013 Vista Silicon S.L.
   9  * Author: Javier Martin <javier.martin@vista-silicon.com>
  10  *
  11  * Based on omap-aes.c and tegra-aes.c
  12  */
  13
  14 #include <crypto/aes.h>
  15 #include <crypto/internal/hash.h>
  16 #include <crypto/internal/skcipher.h>
  17 #include <crypto/scatterwalk.h>
  18 #include <crypto/sha1.h>
  19 #include <crypto/sha2.h>
  20
  21 #include <linux/clk.h>
  22 #include <linux/dma-mapping.h>
  23 #include <linux/interrupt.h>
  24 #include <linux/io.h>
  25 #include <linux/irq.h>
  26 #include <linux/kernel.h>
  27 #include <linux/kthread.h>
  28 #include <linux/module.h>
  29 #include <linux/mutex.h>
  30 #include <linux/of.h>
  31 #include <linux/of_device.h>
  32 #include <linux/platform_device.h>
  33
  34 #define SHA_BUFFER_LEN          PAGE_SIZE
  35 #define SAHARA_MAX_SHA_BLOCK_SIZE       SHA256_BLOCK_SIZE
  36
  37 #define SAHARA_NAME "sahara"
  38 #define SAHARA_VERSION_3        3
  39 #define SAHARA_VERSION_4        4
  40 #define SAHARA_TIMEOUT_MS       1000
  41 #define SAHARA_MAX_HW_DESC      2
  42 #define SAHARA_MAX_HW_LINK      20
  43
  44 #define FLAGS_MODE_MASK         0x000f
  45 #define FLAGS_ENCRYPT           BIT(0)
  46 #define FLAGS_CBC               BIT(1)
  47 #define FLAGS_NEW_KEY           BIT(3)
  48
  49 #define SAHARA_HDR_BASE                 0x00800000
  50 #define SAHARA_HDR_SKHA_ALG_AES 0
  51 #define SAHARA_HDR_SKHA_OP_ENC          (1 << 2)
  52 #define SAHARA_HDR_SKHA_MODE_ECB        (0 << 3)
  53 #define SAHARA_HDR_SKHA_MODE_CBC        (1 << 3)
  54 #define SAHARA_HDR_FORM_DATA            (5 << 16)
  55 #define SAHARA_HDR_FORM_KEY             (8 << 16)
  56 #define SAHARA_HDR_LLO                  (1 << 24)
  57 #define SAHARA_HDR_CHA_SKHA             (1 << 28)
  58 #define SAHARA_HDR_CHA_MDHA             (2 << 28)
  59 #define SAHARA_HDR_PARITY_BIT           (1 << 31)
  60
  61 #define SAHARA_HDR_MDHA_SET_MODE_MD_KEY 0x20880000
  62 #define SAHARA_HDR_MDHA_SET_MODE_HASH   0x208D0000
  63 #define SAHARA_HDR_MDHA_HASH            0xA0850000
  64 #define SAHARA_HDR_MDHA_STORE_DIGEST    0x20820000
  65 #define SAHARA_HDR_MDHA_ALG_SHA1        0
  66 #define SAHARA_HDR_MDHA_ALG_MD5         1
  67 #define SAHARA_HDR_MDHA_ALG_SHA256      2
  68 #define SAHARA_HDR_MDHA_ALG_SHA224      3
  69 #define SAHARA_HDR_MDHA_PDATA           (1 << 2)
  70 #define SAHARA_HDR_MDHA_HMAC            (1 << 3)
  71 #define SAHARA_HDR_MDHA_INIT            (1 << 5)
  72 #define SAHARA_HDR_MDHA_IPAD            (1 << 6)
  73 #define SAHARA_HDR_MDHA_OPAD            (1 << 7)
  74 #define SAHARA_HDR_MDHA_SWAP            (1 << 8)
  75 #define SAHARA_HDR_MDHA_MAC_FULL        (1 << 9)
  76 #define SAHARA_HDR_MDHA_SSL             (1 << 10)
  77
  78 /* SAHARA can only process one request at a time */
  79 #define SAHARA_QUEUE_LENGTH     1
  80
  81 #define SAHARA_REG_VERSION      0x00
  82 #define SAHARA_REG_DAR          0x04
  83 #define SAHARA_REG_CONTROL      0x08
  84 #define         SAHARA_CONTROL_SET_THROTTLE(x)  (((x) & 0xff) << 24)
  85 #define         SAHARA_CONTROL_SET_MAXBURST(x)  (((x) & 0xff) << 16)
  86 #define         SAHARA_CONTROL_RNG_AUTORSD      (1 << 7)
  87 #define         SAHARA_CONTROL_ENABLE_INT       (1 << 4)
  88 #define SAHARA_REG_CMD          0x0C
  89 #define         SAHARA_CMD_RESET                (1 << 0)
  90 #define         SAHARA_CMD_CLEAR_INT            (1 << 8)
  91 #define         SAHARA_CMD_CLEAR_ERR            (1 << 9)
  92 #define         SAHARA_CMD_SINGLE_STEP          (1 << 10)
  93 #define         SAHARA_CMD_MODE_BATCH           (1 << 16)
  94 #define         SAHARA_CMD_MODE_DEBUG           (1 << 18)
  95 #define SAHARA_REG_STATUS       0x10
  96 #define         SAHARA_STATUS_GET_STATE(x)      ((x) & 0x7)
  97 #define                 SAHARA_STATE_IDLE       0
  98 #define                 SAHARA_STATE_BUSY       1
  99 #define                 SAHARA_STATE_ERR        2
 100 #define                 SAHARA_STATE_FAULT      3
 101 #define                 SAHARA_STATE_COMPLETE   4
 102 #define                 SAHARA_STATE_COMP_FLAG  (1 << 2)
 103 #define         SAHARA_STATUS_DAR_FULL          (1 << 3)
 104 #define         SAHARA_STATUS_ERROR             (1 << 4)
 105 #define         SAHARA_STATUS_SECURE            (1 << 5)
 106 #define         SAHARA_STATUS_FAIL              (1 << 6)
 107 #define         SAHARA_STATUS_INIT              (1 << 7)
 108 #define         SAHARA_STATUS_RNG_RESEED        (1 << 8)
 109 #define         SAHARA_STATUS_ACTIVE_RNG        (1 << 9)
 110 #define         SAHARA_STATUS_ACTIVE_MDHA       (1 << 10)
 111 #define         SAHARA_STATUS_ACTIVE_SKHA       (1 << 11)
 112 #define         SAHARA_STATUS_MODE_BATCH        (1 << 16)
 113 #define         SAHARA_STATUS_MODE_DEDICATED    (1 << 17)
 114 #define         SAHARA_STATUS_MODE_DEBUG        (1 << 18)
 115 #define         SAHARA_STATUS_GET_ISTATE(x)     (((x) >> 24) & 0xff)
 116 #define SAHARA_REG_ERRSTATUS    0x14
 117 #define         SAHARA_ERRSTATUS_GET_SOURCE(x)  ((x) & 0xf)
 118 #define                 SAHARA_ERRSOURCE_CHA    14
 119 #define                 SAHARA_ERRSOURCE_DMA    15
 120 #define         SAHARA_ERRSTATUS_DMA_DIR        (1 << 8)
 121 #define         SAHARA_ERRSTATUS_GET_DMASZ(x)(((x) >> 9) & 0x3)
 122 #define         SAHARA_ERRSTATUS_GET_DMASRC(x) (((x) >> 13) & 0x7)
 123 #define         SAHARA_ERRSTATUS_GET_CHASRC(x)  (((x) >> 16) & 0xfff)
 124 #define         SAHARA_ERRSTATUS_GET_CHAERR(x)  (((x) >> 28) & 0x3)
 125 #define SAHARA_REG_FADDR        0x18
 126 #define SAHARA_REG_CDAR         0x1C
 127 #define SAHARA_REG_IDAR         0x20
 128
 129 struct sahara_hw_desc {
 130         u32     hdr;
 131         u32     len1;
 132         u32     p1;
 133         u32     len2;
 134         u32     p2;
 135         u32     next;
 136 };
 137
 138 struct sahara_hw_link {
 139         u32     len;
 140         u32     p;
 141         u32     next;
 142 };
 143
 144 struct sahara_ctx {
 145         unsigned long flags;
 146
 147         /* AES-specific context */
 148         int keylen;
 149         u8 key[AES_KEYSIZE_128];
 150         struct crypto_skcipher *fallback;
 151 };
 152
 153 struct sahara_aes_reqctx {
 154         unsigned long mode;
 155         struct skcipher_request fallback_req;   // keep at the end
 156 };
 157
 158 /*
 159  * struct sahara_sha_reqctx - private data per request
 160  * @buf: holds data for requests smaller than block_size
 161  * @rembuf: used to prepare one block_size-aligned request
 162  * @context: hw-specific context for request. Digest is extracted from this
 163  * @mode: specifies what type of hw-descriptor needs to be built
 164  * @digest_size: length of digest for this request
 165  * @context_size: length of hw-context for this request.
 166  *                Always digest_size + 4
 167  * @buf_cnt: number of bytes saved in buf
 168  * @sg_in_idx: number of hw links
 169  * @in_sg: scatterlist for input data
 170  * @in_sg_chain: scatterlists for chained input data
 171  * @total: total number of bytes for transfer
 172  * @last: is this the last block
 173  * @first: is this the first block
 174  * @active: inside a transfer
 175  */
 176 struct sahara_sha_reqctx {
 177         u8                      buf[SAHARA_MAX_SHA_BLOCK_SIZE];
 178         u8                      rembuf[SAHARA_MAX_SHA_BLOCK_SIZE];
 179         u8                      context[SHA256_DIGEST_SIZE + 4];
 180         unsigned int            mode;
 181         unsigned int            digest_size;
 182         unsigned int            context_size;
 183         unsigned int            buf_cnt;
 184         unsigned int            sg_in_idx;
 185         struct scatterlist      *in_sg;
 186         struct scatterlist      in_sg_chain[2];
 187         size_t                  total;
 188         unsigned int            last;
 189         unsigned int            first;
 190         unsigned int            active;
 191 };
 192
 193 struct sahara_dev {
 194         struct device           *device;
 195         unsigned int            version;
 196         void __iomem            *regs_base;
 197         struct clk              *clk_ipg;
 198         struct clk              *clk_ahb;
 199         struct mutex            queue_mutex;
 200         struct task_struct      *kthread;
 201         struct completion       dma_completion;
 202
 203         struct sahara_ctx       *ctx;
 204         struct crypto_queue     queue;
 205         unsigned long           flags;
 206
 207         struct sahara_hw_desc   *hw_desc[SAHARA_MAX_HW_DESC];
 208         dma_addr_t              hw_phys_desc[SAHARA_MAX_HW_DESC];
 209
 210         u8                      *key_base;
 211         dma_addr_t              key_phys_base;
 212
 213         u8                      *iv_base;
 214         dma_addr_t              iv_phys_base;
 215
 216         u8                      *context_base;
 217         dma_addr_t              context_phys_base;
 218
 219         struct sahara_hw_link   *hw_link[SAHARA_MAX_HW_LINK];
 220         dma_addr_t              hw_phys_link[SAHARA_MAX_HW_LINK];
 221
 222         size_t                  total;
 223         struct scatterlist      *in_sg;
 224         int             nb_in_sg;
 225         struct scatterlist      *out_sg;
 226         int             nb_out_sg;
 227
 228         u32                     error;
 229 };
 230
 231 static struct sahara_dev *dev_ptr;
 232
 233 static inline void sahara_write(struct sahara_dev *dev, u32 data, u32 reg)
 234 {
 235         writel(data, dev->regs_base + reg);
 236 }
 237
 238 static inline unsigned int sahara_read(struct sahara_dev *dev, u32 reg)
 239 {
 240         return readl(dev->regs_base + reg);
 241 }
 242
 243 static u32 sahara_aes_key_hdr(struct sahara_dev *dev)
 244 {
 245         u32 hdr = SAHARA_HDR_BASE | SAHARA_HDR_SKHA_ALG_AES |
 246                         SAHARA_HDR_FORM_KEY | SAHARA_HDR_LLO |
 247                         SAHARA_HDR_CHA_SKHA | SAHARA_HDR_PARITY_BIT;
 248
 249         if (dev->flags & FLAGS_CBC) {
 250                 hdr |= SAHARA_HDR_SKHA_MODE_CBC;
 251                 hdr ^= SAHARA_HDR_PARITY_BIT;
 252         }
 253
 254         if (dev->flags & FLAGS_ENCRYPT) {
 255                 hdr |= SAHARA_HDR_SKHA_OP_ENC;
 256                 hdr ^= SAHARA_HDR_PARITY_BIT;
 257         }
 258
 259         return hdr;
 260 }
 261
 262 static u32 sahara_aes_data_link_hdr(struct sahara_dev *dev)
 263 {
 264         return SAHARA_HDR_BASE | SAHARA_HDR_FORM_DATA |
 265                         SAHARA_HDR_CHA_SKHA | SAHARA_HDR_PARITY_BIT;
 266 }
 267
 268 static const char *sahara_err_src[16] = {
 269         "No error",
 270         "Header error",
 271         "Descriptor length error",
 272         "Descriptor length or pointer error",
 273         "Link length error",
 274         "Link pointer error",
 275         "Input buffer error",
 276         "Output buffer error",
 277         "Output buffer starvation",
 278         "Internal state fault",
 279         "General descriptor problem",
 280         "Reserved",
 281         "Descriptor address error",
 282         "Link address error",
 283         "CHA error",
 284         "DMA error"
 285 };
 286
 287 static const char *sahara_err_dmasize[4] = {
 288         "Byte transfer",
 289         "Half-word transfer",
 290         "Word transfer",
 291         "Reserved"
 292 };
 293
 294 static const char *sahara_err_dmasrc[8] = {
 295         "No error",
 296         "AHB bus error",
 297         "Internal IP bus error",
 298         "Parity error",
 299         "DMA crosses 256 byte boundary",
 300         "DMA is busy",
 301         "Reserved",
 302         "DMA HW error"
 303 };
 304
 305 static const char *sahara_cha_errsrc[12] = {
 306         "Input buffer non-empty",
 307         "Illegal address",
 308         "Illegal mode",
 309         "Illegal data size",
 310         "Illegal key size",
 311         "Write during processing",
 312         "CTX read during processing",
 313         "HW error",
 314         "Input buffer disabled/underflow",
 315         "Output buffer disabled/overflow",
 316         "DES key parity error",
 317         "Reserved"
 318 };
 319
 320 static const char *sahara_cha_err[4] = { "No error", "SKHA", "MDHA", "RNG" };
 321
 322 static void sahara_decode_error(struct sahara_dev *dev, unsigned int error)
 323 {
 324         u8 source = SAHARA_ERRSTATUS_GET_SOURCE(error);
 325         u16 chasrc = ffs(SAHARA_ERRSTATUS_GET_CHASRC(error));
 326
 327         dev_err(dev->device, "%s: Error Register = 0x%08x\n", __func__, error);
 328
 329         dev_err(dev->device, "  - %s.\n", sahara_err_src[source]);
 330
 331         if (source == SAHARA_ERRSOURCE_DMA) {
 332                 if (error & SAHARA_ERRSTATUS_DMA_DIR)
 333                         dev_err(dev->device, "          * DMA read.\n");
 334                 else
 335                         dev_err(dev->device, "          * DMA write.\n");
 336
 337                 dev_err(dev->device, "          * %s.\n",
 338                        sahara_err_dmasize[SAHARA_ERRSTATUS_GET_DMASZ(error)]);
 339                 dev_err(dev->device, "          * %s.\n",
 340                        sahara_err_dmasrc[SAHARA_ERRSTATUS_GET_DMASRC(error)]);
 341         } else if (source == SAHARA_ERRSOURCE_CHA) {
 342                 dev_err(dev->device, "          * %s.\n",
 343                         sahara_cha_errsrc[chasrc]);
 344                 dev_err(dev->device, "          * %s.\n",
 345                        sahara_cha_err[SAHARA_ERRSTATUS_GET_CHAERR(error)]);
 346         }
 347         dev_err(dev->device, "\n");
 348 }
 349
 350 static const char *sahara_state[4] = { "Idle", "Busy", "Error", "HW Fault" };
 351
 352 static void sahara_decode_status(struct sahara_dev *dev, unsigned int status)
 353 {
 354         u8 state;
 355
 356         if (!__is_defined(DEBUG))
 357                 return;
 358
 359         state = SAHARA_STATUS_GET_STATE(status);
 360
 361         dev_dbg(dev->device, "%s: Status Register = 0x%08x\n",
 362                 __func__, status);
 363
 364         dev_dbg(dev->device, "  - State = %d:\n", state);
 365         if (state & SAHARA_STATE_COMP_FLAG)
 366                 dev_dbg(dev->device, "          * Descriptor completed. IRQ pending.\n");
 367
 368         dev_dbg(dev->device, "          * %s.\n",
 369                sahara_state[state & ~SAHARA_STATE_COMP_FLAG]);
 370
 371         if (status & SAHARA_STATUS_DAR_FULL)
 372                 dev_dbg(dev->device, "  - DAR Full.\n");
 373         if (status & SAHARA_STATUS_ERROR)
 374                 dev_dbg(dev->device, "  - Error.\n");
 375         if (status & SAHARA_STATUS_SECURE)
 376                 dev_dbg(dev->device, "  - Secure.\n");
 377         if (status & SAHARA_STATUS_FAIL)
 378                 dev_dbg(dev->device, "  - Fail.\n");
 379         if (status & SAHARA_STATUS_RNG_RESEED)
 380                 dev_dbg(dev->device, "  - RNG Reseed Request.\n");
 381         if (status & SAHARA_STATUS_ACTIVE_RNG)
 382                 dev_dbg(dev->device, "  - RNG Active.\n");
 383         if (status & SAHARA_STATUS_ACTIVE_MDHA)
 384                 dev_dbg(dev->device, "  - MDHA Active.\n");
 385         if (status & SAHARA_STATUS_ACTIVE_SKHA)
 386                 dev_dbg(dev->device, "  - SKHA Active.\n");
 387
 388         if (status & SAHARA_STATUS_MODE_BATCH)
 389                 dev_dbg(dev->device, "  - Batch Mode.\n");
 390         else if (status & SAHARA_STATUS_MODE_DEDICATED)
 391                 dev_dbg(dev->device, "  - Dedicated Mode.\n");
 392         else if (status & SAHARA_STATUS_MODE_DEBUG)
 393                 dev_dbg(dev->device, "  - Debug Mode.\n");
 394
 395         dev_dbg(dev->device, "  - Internal state = 0x%02x\n",
 396                SAHARA_STATUS_GET_ISTATE(status));
 397
 398         dev_dbg(dev->device, "Current DAR: 0x%08x\n",
 399                 sahara_read(dev, SAHARA_REG_CDAR));
 400         dev_dbg(dev->device, "Initial DAR: 0x%08x\n\n",
 401                 sahara_read(dev, SAHARA_REG_IDAR));
 402 }
 403
 404 static void sahara_dump_descriptors(struct sahara_dev *dev)
 405 {
 406         int i;
 407
 408         if (!__is_defined(DEBUG))
 409                 return;
 410
 411         for (i = 0; i < SAHARA_MAX_HW_DESC; i++) {
 412                 dev_dbg(dev->device, "Descriptor (%d) (%pad):\n",
 413                         i, &dev->hw_phys_desc[i]);
 414                 dev_dbg(dev->device, "\thdr = 0x%08x\n", dev->hw_desc[i]->hdr);
 415                 dev_dbg(dev->device, "\tlen1 = %u\n", dev->hw_desc[i]->len1);
 416                 dev_dbg(dev->device, "\tp1 = 0x%08x\n", dev->hw_desc[i]->p1);
 417                 dev_dbg(dev->device, "\tlen2 = %u\n", dev->hw_desc[i]->len2);
 418                 dev_dbg(dev->device, "\tp2 = 0x%08x\n", dev->hw_desc[i]->p2);
 419                 dev_dbg(dev->device, "\tnext = 0x%08x\n",
 420                         dev->hw_desc[i]->next);
 421         }
 422         dev_dbg(dev->device, "\n");
 423 }
 424
 425 static void sahara_dump_links(struct sahara_dev *dev)
 426 {
 427         int i;
 428
 429         if (!__is_defined(DEBUG))
 430                 return;
 431
 432         for (i = 0; i < SAHARA_MAX_HW_LINK; i++) {
 433                 dev_dbg(dev->device, "Link (%d) (%pad):\n",
 434                         i, &dev->hw_phys_link[i]);
 435                 dev_dbg(dev->device, "\tlen = %u\n", dev->hw_link[i]->len);
 436                 dev_dbg(dev->device, "\tp = 0x%08x\n", dev->hw_link[i]->p);
 437                 dev_dbg(dev->device, "\tnext = 0x%08x\n",
 438                         dev->hw_link[i]->next);
 439         }
 440         dev_dbg(dev->device, "\n");
 441 }
 442
 443 static int sahara_hw_descriptor_create(struct sahara_dev *dev)
 444 {
 445         struct sahara_ctx *ctx = dev->ctx;
 446         struct scatterlist *sg;
 447         int ret;
 448         int i, j;
 449         int idx = 0;
 450
 451         /* Copy new key if necessary */
 452         if (ctx->flags & FLAGS_NEW_KEY) {
 453                 memcpy(dev->key_base, ctx->key, ctx->keylen);
 454                 ctx->flags &= ~FLAGS_NEW_KEY;
 455
 456                 if (dev->flags & FLAGS_CBC) {
 457                         dev->hw_desc[idx]->len1 = AES_BLOCK_SIZE;
 458                         dev->hw_desc[idx]->p1 = dev->iv_phys_base;
 459                 } else {
 460                         dev->hw_desc[idx]->len1 = 0;
 461                         dev->hw_desc[idx]->p1 = 0;
 462                 }
 463                 dev->hw_desc[idx]->len2 = ctx->keylen;
 464                 dev->hw_desc[idx]->p2 = dev->key_phys_base;
 465                 dev->hw_desc[idx]->next = dev->hw_phys_desc[1];
 466
 467                 dev->hw_desc[idx]->hdr = sahara_aes_key_hdr(dev);
 468
 469                 idx++;
 470         }
 471
 472         dev->nb_in_sg = sg_nents_for_len(dev->in_sg, dev->total);
 473         if (dev->nb_in_sg < 0) {
 474                 dev_err(dev->device, "Invalid numbers of src SG.\n");
 475                 return dev->nb_in_sg;
 476         }
 477         dev->nb_out_sg = sg_nents_for_len(dev->out_sg, dev->total);
 478         if (dev->nb_out_sg < 0) {
 479                 dev_err(dev->device, "Invalid numbers of dst SG.\n");
 480                 return dev->nb_out_sg;
 481         }
 482         if ((dev->nb_in_sg + dev->nb_out_sg) > SAHARA_MAX_HW_LINK) {
 483                 dev_err(dev->device, "not enough hw links (%d)\n",
 484                         dev->nb_in_sg + dev->nb_out_sg);
 485                 return -EINVAL;
 486         }
 487
 488         ret = dma_map_sg(dev->device, dev->in_sg, dev->nb_in_sg,
 489                          DMA_TO_DEVICE);
 490         if (ret != dev->nb_in_sg) {
 491                 dev_err(dev->device, "couldn't map in sg\n");
 492                 goto unmap_in;
 493         }
 494         ret = dma_map_sg(dev->device, dev->out_sg, dev->nb_out_sg,
 495                          DMA_FROM_DEVICE);
 496         if (ret != dev->nb_out_sg) {
 497                 dev_err(dev->device, "couldn't map out sg\n");
 498                 goto unmap_out;
 499         }
 500
 501         /* Create input links */
 502         dev->hw_desc[idx]->p1 = dev->hw_phys_link[0];
 503         sg = dev->in_sg;
 504         for (i = 0; i < dev->nb_in_sg; i++) {
 505                 dev->hw_link[i]->len = sg->length;
 506                 dev->hw_link[i]->p = sg->dma_address;
 507                 if (i == (dev->nb_in_sg - 1)) {
 508                         dev->hw_link[i]->next = 0;
 509                 } else {
 510                         dev->hw_link[i]->next = dev->hw_phys_link[i + 1];
 511                         sg = sg_next(sg);
 512                 }
 513         }
 514
 515         /* Create output links */
 516         dev->hw_desc[idx]->p2 = dev->hw_phys_link[i];
 517         sg = dev->out_sg;
 518         for (j = i; j < dev->nb_out_sg + i; j++) {
 519                 dev->hw_link[j]->len = sg->length;
 520                 dev->hw_link[j]->p = sg->dma_address;
 521                 if (j == (dev->nb_out_sg + i - 1)) {
 522                         dev->hw_link[j]->next = 0;
 523                 } else {
 524                         dev->hw_link[j]->next = dev->hw_phys_link[j + 1];
 525                         sg = sg_next(sg);
 526                 }
 527         }
 528
 529         /* Fill remaining fields of hw_desc[1] */
 530         dev->hw_desc[idx]->hdr = sahara_aes_data_link_hdr(dev);
 531         dev->hw_desc[idx]->len1 = dev->total;
 532         dev->hw_desc[idx]->len2 = dev->total;
 533         dev->hw_desc[idx]->next = 0;
 534
 535         sahara_dump_descriptors(dev);
 536         sahara_dump_links(dev);
 537
 538         sahara_write(dev, dev->hw_phys_desc[0], SAHARA_REG_DAR);
 539
 540         return 0;
 541
 542 unmap_out:
 543         dma_unmap_sg(dev->device, dev->out_sg, dev->nb_out_sg,
 544                 DMA_FROM_DEVICE);
 545 unmap_in:
 546         dma_unmap_sg(dev->device, dev->in_sg, dev->nb_in_sg,
 547                 DMA_TO_DEVICE);
 548
 549         return -EINVAL;
 550 }
 551
 552 static int sahara_aes_process(struct skcipher_request *req)
 553 {
 554         struct sahara_dev *dev = dev_ptr;
 555         struct sahara_ctx *ctx;
 556         struct sahara_aes_reqctx *rctx;
 557         int ret;
 558         unsigned long timeout;
 559
 560         /* Request is ready to be dispatched by the device */
 561         dev_dbg(dev->device,
 562                 "dispatch request (nbytes=%d, src=%p, dst=%p)\n",
 563                 req->cryptlen, req->src, req->dst);
 564
 565         /* assign new request to device */
 566         dev->total = req->cryptlen;
 567         dev->in_sg = req->src;
 568         dev->out_sg = req->dst;
 569
 570         rctx = skcipher_request_ctx(req);
 571         ctx = crypto_skcipher_ctx(crypto_skcipher_reqtfm(req));
 572         rctx->mode &= FLAGS_MODE_MASK;
 573         dev->flags = (dev->flags & ~FLAGS_MODE_MASK) | rctx->mode;
 574
 575         if ((dev->flags & FLAGS_CBC) && req->iv)
 576                 memcpy(dev->iv_base, req->iv, AES_KEYSIZE_128);
 577
 578         /* assign new context to device */
 579         dev->ctx = ctx;
 580
 581         reinit_completion(&dev->dma_completion);
 582
 583         ret = sahara_hw_descriptor_create(dev);
 584         if (ret)
 585                 return -EINVAL;
 586
 587         timeout = wait_for_completion_timeout(&dev->dma_completion,
 588                                 msecs_to_jiffies(SAHARA_TIMEOUT_MS));
 589         if (!timeout) {
 590                 dev_err(dev->device, "AES timeout\n");
 591                 return -ETIMEDOUT;
 592         }
 593
 594         dma_unmap_sg(dev->device, dev->out_sg, dev->nb_out_sg,
 595                 DMA_FROM_DEVICE);
 596         dma_unmap_sg(dev->device, dev->in_sg, dev->nb_in_sg,
 597                 DMA_TO_DEVICE);
 598
 599         return 0;
 600 }
 601
 602 static int sahara_aes_setkey(struct crypto_skcipher *tfm, const u8 *key,
 603                              unsigned int keylen)
 604 {
 605         struct sahara_ctx *ctx = crypto_skcipher_ctx(tfm);
 606
 607         ctx->keylen = keylen;
 608
 609         /* SAHARA only supports 128bit keys */
 610         if (keylen == AES_KEYSIZE_128) {
 611                 memcpy(ctx->key, key, keylen);
 612                 ctx->flags |= FLAGS_NEW_KEY;
 613                 return 0;
 614         }
 615
 616         if (keylen != AES_KEYSIZE_192 && keylen != AES_KEYSIZE_256)
 617                 return -EINVAL;
 618
 619         /*
 620          * The requested key size is not supported by HW, do a fallback.
 621          */
 622         crypto_skcipher_clear_flags(ctx->fallback, CRYPTO_TFM_REQ_MASK);
 623         crypto_skcipher_set_flags(ctx->fallback, tfm->base.crt_flags &
 624                                                  CRYPTO_TFM_REQ_MASK);
 625         return crypto_skcipher_setkey(ctx->fallback, key, keylen);
 626 }
 627
 628 static int sahara_aes_crypt(struct skcipher_request *req, unsigned long mode)
 629 {
 630         struct sahara_aes_reqctx *rctx = skcipher_request_ctx(req);
 631         struct sahara_dev *dev = dev_ptr;
 632         int err = 0;
 633
 634         dev_dbg(dev->device, "nbytes: %d, enc: %d, cbc: %d\n",
 635                 req->cryptlen, !!(mode & FLAGS_ENCRYPT), !!(mode & FLAGS_CBC));
 636
 637         if (!IS_ALIGNED(req->cryptlen, AES_BLOCK_SIZE)) {
 638                 dev_err(dev->device,
 639                         "request size is not exact amount of AES blocks\n");
 640                 return -EINVAL;
 641         }
 642
 643         rctx->mode = mode;
 644
 645         mutex_lock(&dev->queue_mutex);
 646         err = crypto_enqueue_request(&dev->queue, &req->base);
 647         mutex_unlock(&dev->queue_mutex);
 648
 649         wake_up_process(dev->kthread);
 650
 651         return err;
 652 }
 653
 654 static int sahara_aes_ecb_encrypt(struct skcipher_request *req)
 655 {
 656         struct sahara_aes_reqctx *rctx = skcipher_request_ctx(req);
 657         struct sahara_ctx *ctx = crypto_skcipher_ctx(
 658                 crypto_skcipher_reqtfm(req));
 659
 660         if (unlikely(ctx->keylen != AES_KEYSIZE_128)) {
 661                 skcipher_request_set_tfm(&rctx->fallback_req, ctx->fallback);
 662                 skcipher_request_set_callback(&rctx->fallback_req,
 663                                               req->base.flags,
 664                                               req->base.complete,
 665                                               req->base.data);
 666                 skcipher_request_set_crypt(&rctx->fallback_req, req->src,
 667                                            req->dst, req->cryptlen, req->iv);
 668                 return crypto_skcipher_encrypt(&rctx->fallback_req);
 669         }
 670
 671         return sahara_aes_crypt(req, FLAGS_ENCRYPT);
 672 }
 673
 674 static int sahara_aes_ecb_decrypt(struct skcipher_request *req)
 675 {
 676         struct sahara_aes_reqctx *rctx = skcipher_request_ctx(req);
 677         struct sahara_ctx *ctx = crypto_skcipher_ctx(
 678                 crypto_skcipher_reqtfm(req));
 679
 680         if (unlikely(ctx->keylen != AES_KEYSIZE_128)) {
 681                 skcipher_request_set_tfm(&rctx->fallback_req, ctx->fallback);
 682                 skcipher_request_set_callback(&rctx->fallback_req,
 683                                               req->base.flags,
 684                                               req->base.complete,
 685                                               req->base.data);
 686                 skcipher_request_set_crypt(&rctx->fallback_req, req->src,
 687                                            req->dst, req->cryptlen, req->iv);
 688                 return crypto_skcipher_decrypt(&rctx->fallback_req);
 689         }
 690
 691         return sahara_aes_crypt(req, 0);
 692 }
 693
 694 static int sahara_aes_cbc_encrypt(struct skcipher_request *req)
 695 {
 696         struct sahara_aes_reqctx *rctx = skcipher_request_ctx(req);
 697         struct sahara_ctx *ctx = crypto_skcipher_ctx(
 698                 crypto_skcipher_reqtfm(req));
 699
 700         if (unlikely(ctx->keylen != AES_KEYSIZE_128)) {
 701                 skcipher_request_set_tfm(&rctx->fallback_req, ctx->fallback);
 702                 skcipher_request_set_callback(&rctx->fallback_req,
 703                                               req->base.flags,
 704                                               req->base.complete,
 705                                               req->base.data);
 706                 skcipher_request_set_crypt(&rctx->fallback_req, req->src,
 707                                            req->dst, req->cryptlen, req->iv);
 708                 return crypto_skcipher_encrypt(&rctx->fallback_req);
 709         }
 710
 711         return sahara_aes_crypt(req, FLAGS_ENCRYPT | FLAGS_CBC);
 712 }
 713
 714 static int sahara_aes_cbc_decrypt(struct skcipher_request *req)
 715 {
 716         struct sahara_aes_reqctx *rctx = skcipher_request_ctx(req);
 717         struct sahara_ctx *ctx = crypto_skcipher_ctx(
 718                 crypto_skcipher_reqtfm(req));
 719
 720         if (unlikely(ctx->keylen != AES_KEYSIZE_128)) {
 721                 skcipher_request_set_tfm(&rctx->fallback_req, ctx->fallback);
 722                 skcipher_request_set_callback(&rctx->fallback_req,
 723                                               req->base.flags,
 724                                               req->base.complete,
 725                                               req->base.data);
 726                 skcipher_request_set_crypt(&rctx->fallback_req, req->src,
 727                                            req->dst, req->cryptlen, req->iv);
 728                 return crypto_skcipher_decrypt(&rctx->fallback_req);
 729         }
 730
 731         return sahara_aes_crypt(req, FLAGS_CBC);
 732 }
 733
 734 static int sahara_aes_init_tfm(struct crypto_skcipher *tfm)
 735 {
 736         const char *name = crypto_tfm_alg_name(&tfm->base);
 737         struct sahara_ctx *ctx = crypto_skcipher_ctx(tfm);
 738
 739         ctx->fallback = crypto_alloc_skcipher(name, 0,
 740                                               CRYPTO_ALG_NEED_FALLBACK);
 741         if (IS_ERR(ctx->fallback)) {
 742                 pr_err("Error allocating fallback algo %s\n", name);
 743                 return PTR_ERR(ctx->fallback);
 744         }
 745
 746         crypto_skcipher_set_reqsize(tfm, sizeof(struct sahara_aes_reqctx) +
 747                                          crypto_skcipher_reqsize(ctx->fallback));
 748
 749         return 0;
 750 }
 751
 752 static void sahara_aes_exit_tfm(struct crypto_skcipher *tfm)
 753 {
 754         struct sahara_ctx *ctx = crypto_skcipher_ctx(tfm);
 755
 756         crypto_free_skcipher(ctx->fallback);
 757 }
 758
 759 static u32 sahara_sha_init_hdr(struct sahara_dev *dev,
 760                               struct sahara_sha_reqctx *rctx)
 761 {
 762         u32 hdr = 0;
 763
 764         hdr = rctx->mode;
 765
 766         if (rctx->first) {
 767                 hdr |= SAHARA_HDR_MDHA_SET_MODE_HASH;
 768                 hdr |= SAHARA_HDR_MDHA_INIT;
 769         } else {
 770                 hdr |= SAHARA_HDR_MDHA_SET_MODE_MD_KEY;
 771         }
 772
 773         if (rctx->last)
 774                 hdr |= SAHARA_HDR_MDHA_PDATA;
 775
 776         if (hweight_long(hdr) % 2 == 0)
 777                 hdr |= SAHARA_HDR_PARITY_BIT;
 778
 779         return hdr;
 780 }
 781
 782 static int sahara_sha_hw_links_create(struct sahara_dev *dev,
 783                                        struct sahara_sha_reqctx *rctx,
 784                                        int start)
 785 {
 786         struct scatterlist *sg;
 787         unsigned int i;
 788         int ret;
 789
 790         dev->in_sg = rctx->in_sg;
 791
 792         dev->nb_in_sg = sg_nents_for_len(dev->in_sg, rctx->total);
 793         if (dev->nb_in_sg < 0) {
 794                 dev_err(dev->device, "Invalid numbers of src SG.\n");
 795                 return dev->nb_in_sg;
 796         }
 797         if ((dev->nb_in_sg) > SAHARA_MAX_HW_LINK) {
 798                 dev_err(dev->device, "not enough hw links (%d)\n",
 799                         dev->nb_in_sg + dev->nb_out_sg);
 800                 return -EINVAL;
 801         }
 802
 803         sg = dev->in_sg;
 804         ret = dma_map_sg(dev->device, dev->in_sg, dev->nb_in_sg, DMA_TO_DEVICE);
 805         if (!ret)
 806                 return -EFAULT;
 807
 808         for (i = start; i < dev->nb_in_sg + start; i++) {
 809                 dev->hw_link[i]->len = sg->length;
 810                 dev->hw_link[i]->p = sg->dma_address;
 811                 if (i == (dev->nb_in_sg + start - 1)) {
 812                         dev->hw_link[i]->next = 0;
 813                 } else {
 814                         dev->hw_link[i]->next = dev->hw_phys_link[i + 1];
 815                         sg = sg_next(sg);
 816                 }
 817         }
 818
 819         return i;
 820 }
 821
 822 static int sahara_sha_hw_data_descriptor_create(struct sahara_dev *dev,
 823                                                 struct sahara_sha_reqctx *rctx,
 824                                                 struct ahash_request *req,
 825                                                 int index)
 826 {
 827         unsigned result_len;
 828         int i = index;
 829
 830         if (rctx->first)
 831                 /* Create initial descriptor: #8*/
 832                 dev->hw_desc[index]->hdr = sahara_sha_init_hdr(dev, rctx);
 833         else
 834                 /* Create hash descriptor: #10. Must follow #6. */
 835                 dev->hw_desc[index]->hdr = SAHARA_HDR_MDHA_HASH;
 836
 837         dev->hw_desc[index]->len1 = rctx->total;
 838         if (dev->hw_desc[index]->len1 == 0) {
 839                 /* if len1 is 0, p1 must be 0, too */
 840                 dev->hw_desc[index]->p1 = 0;
 841                 rctx->sg_in_idx = 0;
 842         } else {
 843                 /* Create input links */
 844                 dev->hw_desc[index]->p1 = dev->hw_phys_link[index];
 845                 i = sahara_sha_hw_links_create(dev, rctx, index);
 846
 847                 rctx->sg_in_idx = index;
 848                 if (i < 0)
 849                         return i;
 850         }
 851
 852         dev->hw_desc[index]->p2 = dev->hw_phys_link[i];
 853
 854         /* Save the context for the next operation */
 855         result_len = rctx->context_size;
 856         dev->hw_link[i]->p = dev->context_phys_base;
 857
 858         dev->hw_link[i]->len = result_len;
 859         dev->hw_desc[index]->len2 = result_len;
 860
 861         dev->hw_link[i]->next = 0;
 862
 863         return 0;
 864 }
 865
 866 /*
 867  * Load descriptor aka #6
 868  *
 869  * To load a previously saved context back to the MDHA unit
 870  *
 871  * p1: Saved Context
 872  * p2: NULL
 873  *
 874  */
 875 static int sahara_sha_hw_context_descriptor_create(struct sahara_dev *dev,
 876                                                 struct sahara_sha_reqctx *rctx,
 877                                                 struct ahash_request *req,
 878                                                 int index)
 879 {
 880         dev->hw_desc[index]->hdr = sahara_sha_init_hdr(dev, rctx);
 881
 882         dev->hw_desc[index]->len1 = rctx->context_size;
 883         dev->hw_desc[index]->p1 = dev->hw_phys_link[index];
 884         dev->hw_desc[index]->len2 = 0;
 885         dev->hw_desc[index]->p2 = 0;
 886
 887         dev->hw_link[index]->len = rctx->context_size;
 888         dev->hw_link[index]->p = dev->context_phys_base;
 889         dev->hw_link[index]->next = 0;
 890
 891         return 0;
 892 }
 893
 894 static int sahara_walk_and_recalc(struct scatterlist *sg, unsigned int nbytes)
 895 {
 896         if (!sg || !sg->length)
 897                 return nbytes;
 898
 899         while (nbytes && sg) {
 900                 if (nbytes <= sg->length) {
 901                         sg->length = nbytes;
 902                         sg_mark_end(sg);
 903                         break;
 904                 }
 905                 nbytes -= sg->length;
 906                 sg = sg_next(sg);
 907         }
 908
 909         return nbytes;
 910 }
 911
 912 static int sahara_sha_prepare_request(struct ahash_request *req)
 913 {
 914         struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
 915         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
 916         unsigned int hash_later;
 917         unsigned int block_size;
 918         unsigned int len;
 919
 920         block_size = crypto_tfm_alg_blocksize(crypto_ahash_tfm(tfm));
 921
 922         /* append bytes from previous operation */
 923         len = rctx->buf_cnt + req->nbytes;
 924
 925         /* only the last transfer can be padded in hardware */
 926         if (!rctx->last && (len < block_size)) {
 927                 /* to few data, save for next operation */
 928                 scatterwalk_map_and_copy(rctx->buf + rctx->buf_cnt, req->src,
 929                                          0, req->nbytes, 0);
 930                 rctx->buf_cnt += req->nbytes;
 931
 932                 return 0;
 933         }
 934
 935         /* add data from previous operation first */
 936         if (rctx->buf_cnt)
 937                 memcpy(rctx->rembuf, rctx->buf, rctx->buf_cnt);
 938
 939         /* data must always be a multiple of block_size */
 940         hash_later = rctx->last ? 0 : len & (block_size - 1);
 941         if (hash_later) {
 942                 unsigned int offset = req->nbytes - hash_later;
 943                 /* Save remaining bytes for later use */
 944                 scatterwalk_map_and_copy(rctx->buf, req->src, offset,
 945                                         hash_later, 0);
 946         }
 947
 948         /* nbytes should now be multiple of blocksize */
 949         req->nbytes = req->nbytes - hash_later;
 950
 951         sahara_walk_and_recalc(req->src, req->nbytes);
 952
 953         /* have data from previous operation and current */
 954         if (rctx->buf_cnt && req->nbytes) {
 955                 sg_init_table(rctx->in_sg_chain, 2);
 956                 sg_set_buf(rctx->in_sg_chain, rctx->rembuf, rctx->buf_cnt);
 957
 958                 sg_chain(rctx->in_sg_chain, 2, req->src);
 959
 960                 rctx->total = req->nbytes + rctx->buf_cnt;
 961                 rctx->in_sg = rctx->in_sg_chain;
 962
 963                 req->src = rctx->in_sg_chain;
 964         /* only data from previous operation */
 965         } else if (rctx->buf_cnt) {
 966                 if (req->src)
 967                         rctx->in_sg = req->src;
 968                 else
 969                         rctx->in_sg = rctx->in_sg_chain;
 970                 /* buf was copied into rembuf above */
 971                 sg_init_one(rctx->in_sg, rctx->rembuf, rctx->buf_cnt);
 972                 rctx->total = rctx->buf_cnt;
 973         /* no data from previous operation */
 974         } else {
 975                 rctx->in_sg = req->src;
 976                 rctx->total = req->nbytes;
 977                 req->src = rctx->in_sg;
 978         }
 979
 980         /* on next call, we only have the remaining data in the buffer */
 981         rctx->buf_cnt = hash_later;
 982
 983         return -EINPROGRESS;
 984 }
 985
 986 static int sahara_sha_process(struct ahash_request *req)
 987 {
 988         struct sahara_dev *dev = dev_ptr;
 989         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
 990         int ret;
 991         unsigned long timeout;
 992
 993         ret = sahara_sha_prepare_request(req);
 994         if (!ret)
 995                 return ret;
 996
 997         if (rctx->first) {
 998                 sahara_sha_hw_data_descriptor_create(dev, rctx, req, 0);
 999                 dev->hw_desc[0]->next = 0;
1000                 rctx->first = 0;
1001         } else {
1002                 memcpy(dev->context_base, rctx->context, rctx->context_size);
1003
1004                 sahara_sha_hw_context_descriptor_create(dev, rctx, req, 0);
1005                 dev->hw_desc[0]->next = dev->hw_phys_desc[1];
1006                 sahara_sha_hw_data_descriptor_create(dev, rctx, req, 1);
1007                 dev->hw_desc[1]->next = 0;
1008         }
1009
1010         sahara_dump_descriptors(dev);
1011         sahara_dump_links(dev);
1012
1013         reinit_completion(&dev->dma_completion);
1014
1015         sahara_write(dev, dev->hw_phys_desc[0], SAHARA_REG_DAR);
1016
1017         timeout = wait_for_completion_timeout(&dev->dma_completion,
1018                                 msecs_to_jiffies(SAHARA_TIMEOUT_MS));
1019         if (!timeout) {
1020                 dev_err(dev->device, "SHA timeout\n");
1021                 return -ETIMEDOUT;
1022         }
1023
1024         if (rctx->sg_in_idx)
1025                 dma_unmap_sg(dev->device, dev->in_sg, dev->nb_in_sg,
1026                              DMA_TO_DEVICE);
1027
1028         memcpy(rctx->context, dev->context_base, rctx->context_size);
1029
1030         if (req->result)
1031                 memcpy(req->result, rctx->context, rctx->digest_size);
1032
1033         return 0;
1034 }
1035
1036 static int sahara_queue_manage(void *data)
1037 {
1038         struct sahara_dev *dev = (struct sahara_dev *)data;
1039         struct crypto_async_request *async_req;
1040         struct crypto_async_request *backlog;
1041         int ret = 0;
1042
1043         do {
1044                 __set_current_state(TASK_INTERRUPTIBLE);
1045
1046                 mutex_lock(&dev->queue_mutex);
1047                 backlog = crypto_get_backlog(&dev->queue);
1048                 async_req = crypto_dequeue_request(&dev->queue);
1049                 mutex_unlock(&dev->queue_mutex);
1050
1051                 if (backlog)
1052                         backlog->complete(backlog, -EINPROGRESS);
1053
1054                 if (async_req) {
1055                         if (crypto_tfm_alg_type(async_req->tfm) ==
1056                             CRYPTO_ALG_TYPE_AHASH) {
1057                                 struct ahash_request *req =
1058                                         ahash_request_cast(async_req);
1059
1060                                 ret = sahara_sha_process(req);
1061                         } else {
1062                                 struct skcipher_request *req =
1063                                         skcipher_request_cast(async_req);
1064
1065                                 ret = sahara_aes_process(req);
1066                         }
1067
1068                         async_req->complete(async_req, ret);
1069
1070                         continue;
1071                 }
1072
1073                 schedule();
1074         } while (!kthread_should_stop());
1075
1076         return 0;
1077 }
1078
1079 static int sahara_sha_enqueue(struct ahash_request *req, int last)
1080 {
1081         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
1082         struct sahara_dev *dev = dev_ptr;
1083         int ret;
1084
1085         if (!req->nbytes && !last)
1086                 return 0;
1087
1088         rctx->last = last;
1089
1090         if (!rctx->active) {
1091                 rctx->active = 1;
1092                 rctx->first = 1;
1093         }
1094
1095         mutex_lock(&dev->queue_mutex);
1096         ret = crypto_enqueue_request(&dev->queue, &req->base);
1097         mutex_unlock(&dev->queue_mutex);
1098
1099         wake_up_process(dev->kthread);
1100
1101         return ret;
1102 }
1103
1104 static int sahara_sha_init(struct ahash_request *req)
1105 {
1106         struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
1107         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
1108
1109         memset(rctx, 0, sizeof(*rctx));
1110
1111         switch (crypto_ahash_digestsize(tfm)) {
1112         case SHA1_DIGEST_SIZE:
1113                 rctx->mode |= SAHARA_HDR_MDHA_ALG_SHA1;
1114                 rctx->digest_size = SHA1_DIGEST_SIZE;
1115                 break;
1116         case SHA256_DIGEST_SIZE:
1117                 rctx->mode |= SAHARA_HDR_MDHA_ALG_SHA256;
1118                 rctx->digest_size = SHA256_DIGEST_SIZE;
1119                 break;
1120         default:
1121                 return -EINVAL;
1122         }
1123
1124         rctx->context_size = rctx->digest_size + 4;
1125         rctx->active = 0;
1126
1127         return 0;
1128 }
1129
1130 static int sahara_sha_update(struct ahash_request *req)
1131 {
1132         return sahara_sha_enqueue(req, 0);
1133 }
1134
1135 static int sahara_sha_final(struct ahash_request *req)
1136 {
1137         req->nbytes = 0;
1138         return sahara_sha_enqueue(req, 1);
1139 }
1140
1141 static int sahara_sha_finup(struct ahash_request *req)
1142 {
1143         return sahara_sha_enqueue(req, 1);
1144 }
1145
1146 static int sahara_sha_digest(struct ahash_request *req)
1147 {
1148         sahara_sha_init(req);
1149
1150         return sahara_sha_finup(req);
1151 }
1152
1153 static int sahara_sha_export(struct ahash_request *req, void *out)
1154 {
1155         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
1156
1157         memcpy(out, rctx, sizeof(struct sahara_sha_reqctx));
1158
1159         return 0;
1160 }
1161
1162 static int sahara_sha_import(struct ahash_request *req, const void *in)
1163 {
1164         struct sahara_sha_reqctx *rctx = ahash_request_ctx(req);
1165
1166         memcpy(rctx, in, sizeof(struct sahara_sha_reqctx));
1167
1168         return 0;
1169 }
1170
1171 static int sahara_sha_cra_init(struct crypto_tfm *tfm)
1172 {
1173         crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
1174                                  sizeof(struct sahara_sha_reqctx) +
1175                                  SHA_BUFFER_LEN + SHA256_BLOCK_SIZE);
1176
1177         return 0;
1178 }
1179
1180 static struct skcipher_alg aes_algs[] = {
1181 {
1182         .base.cra_name          = "ecb(aes)",
1183         .base.cra_driver_name   = "sahara-ecb-aes",
1184         .base.cra_priority      = 300,
1185         .base.cra_flags         = CRYPTO_ALG_ASYNC | CRYPTO_ALG_NEED_FALLBACK,
1186         .base.cra_blocksize     = AES_BLOCK_SIZE,
1187         .base.cra_ctxsize       = sizeof(struct sahara_ctx),
1188         .base.cra_alignmask     = 0x0,
1189         .base.cra_module        = THIS_MODULE,
1190
1191         .init                   = sahara_aes_init_tfm,
1192         .exit                   = sahara_aes_exit_tfm,
1193         .min_keysize            = AES_MIN_KEY_SIZE ,
1194         .max_keysize            = AES_MAX_KEY_SIZE,
1195         .setkey                 = sahara_aes_setkey,
1196         .encrypt                = sahara_aes_ecb_encrypt,
1197         .decrypt                = sahara_aes_ecb_decrypt,
1198 }, {
1199         .base.cra_name          = "cbc(aes)",
1200         .base.cra_driver_name   = "sahara-cbc-aes",
1201         .base.cra_priority      = 300,
1202         .base.cra_flags         = CRYPTO_ALG_ASYNC | CRYPTO_ALG_NEED_FALLBACK,
1203         .base.cra_blocksize     = AES_BLOCK_SIZE,
1204         .base.cra_ctxsize       = sizeof(struct sahara_ctx),
1205         .base.cra_alignmask     = 0x0,
1206         .base.cra_module        = THIS_MODULE,
1207
1208         .init                   = sahara_aes_init_tfm,
1209         .exit                   = sahara_aes_exit_tfm,
1210         .min_keysize            = AES_MIN_KEY_SIZE ,
1211         .max_keysize            = AES_MAX_KEY_SIZE,
1212         .ivsize                 = AES_BLOCK_SIZE,
1213         .setkey                 = sahara_aes_setkey,
1214         .encrypt                = sahara_aes_cbc_encrypt,
1215         .decrypt                = sahara_aes_cbc_decrypt,
1216 }
1217 };
1218
1219 static struct ahash_alg sha_v3_algs[] = {
1220 {
1221         .init           = sahara_sha_init,
1222         .update         = sahara_sha_update,
1223         .final          = sahara_sha_final,
1224         .finup          = sahara_sha_finup,
1225         .digest         = sahara_sha_digest,
1226         .export         = sahara_sha_export,
1227         .import         = sahara_sha_import,
1228         .halg.digestsize        = SHA1_DIGEST_SIZE,
1229         .halg.statesize         = sizeof(struct sahara_sha_reqctx),
1230         .halg.base      = {
1231                 .cra_name               = "sha1",
1232                 .cra_driver_name        = "sahara-sha1",
1233                 .cra_priority           = 300,
1234                 .cra_flags              = CRYPTO_ALG_ASYNC |
1235                                                 CRYPTO_ALG_NEED_FALLBACK,
1236                 .cra_blocksize          = SHA1_BLOCK_SIZE,
1237                 .cra_ctxsize            = sizeof(struct sahara_ctx),
1238                 .cra_alignmask          = 0,
1239                 .cra_module             = THIS_MODULE,
1240                 .cra_init               = sahara_sha_cra_init,
1241         }
1242 },
1243 };
1244
1245 static struct ahash_alg sha_v4_algs[] = {
1246 {
1247         .init           = sahara_sha_init,
1248         .update         = sahara_sha_update,
1249         .final          = sahara_sha_final,
1250         .finup          = sahara_sha_finup,
1251         .digest         = sahara_sha_digest,
1252         .export         = sahara_sha_export,
1253         .import         = sahara_sha_import,
1254         .halg.digestsize        = SHA256_DIGEST_SIZE,
1255         .halg.statesize         = sizeof(struct sahara_sha_reqctx),
1256         .halg.base      = {
1257                 .cra_name               = "sha256",
1258                 .cra_driver_name        = "sahara-sha256",
1259                 .cra_priority           = 300,
1260                 .cra_flags              = CRYPTO_ALG_ASYNC |
1261                                                 CRYPTO_ALG_NEED_FALLBACK,
1262                 .cra_blocksize          = SHA256_BLOCK_SIZE,
1263                 .cra_ctxsize            = sizeof(struct sahara_ctx),
1264                 .cra_alignmask          = 0,
1265                 .cra_module             = THIS_MODULE,
1266                 .cra_init               = sahara_sha_cra_init,
1267         }
1268 },
1269 };
1270
1271 static irqreturn_t sahara_irq_handler(int irq, void *data)
1272 {
1273         struct sahara_dev *dev = (struct sahara_dev *)data;
1274         unsigned int stat = sahara_read(dev, SAHARA_REG_STATUS);
1275         unsigned int err = sahara_read(dev, SAHARA_REG_ERRSTATUS);
1276
1277         sahara_write(dev, SAHARA_CMD_CLEAR_INT | SAHARA_CMD_CLEAR_ERR,
1278                      SAHARA_REG_CMD);
1279
1280         sahara_decode_status(dev, stat);
1281
1282         if (SAHARA_STATUS_GET_STATE(stat) == SAHARA_STATE_BUSY) {
1283                 return IRQ_NONE;
1284         } else if (SAHARA_STATUS_GET_STATE(stat) == SAHARA_STATE_COMPLETE) {
1285                 dev->error = 0;
1286         } else {
1287                 sahara_decode_error(dev, err);
1288                 dev->error = -EINVAL;
1289         }
1290
1291         complete(&dev->dma_completion);
1292
1293         return IRQ_HANDLED;
1294 }
1295
1296
1297 static int sahara_register_algs(struct sahara_dev *dev)
1298 {
1299         int err;
1300         unsigned int i, j, k, l;
1301
1302         for (i = 0; i < ARRAY_SIZE(aes_algs); i++) {
1303                 err = crypto_register_skcipher(&aes_algs[i]);
1304                 if (err)
1305                         goto err_aes_algs;
1306         }
1307
1308         for (k = 0; k < ARRAY_SIZE(sha_v3_algs); k++) {
1309                 err = crypto_register_ahash(&sha_v3_algs[k]);
1310                 if (err)
1311                         goto err_sha_v3_algs;
1312         }
1313
1314         if (dev->version > SAHARA_VERSION_3)
1315                 for (l = 0; l < ARRAY_SIZE(sha_v4_algs); l++) {
1316                         err = crypto_register_ahash(&sha_v4_algs[l]);
1317                         if (err)
1318                                 goto err_sha_v4_algs;
1319                 }
1320
1321         return 0;
1322
1323 err_sha_v4_algs:
1324         for (j = 0; j < l; j++)
1325                 crypto_unregister_ahash(&sha_v4_algs[j]);
1326
1327 err_sha_v3_algs:
1328         for (j = 0; j < k; j++)
1329                 crypto_unregister_ahash(&sha_v3_algs[j]);
1330
1331 err_aes_algs:
1332         for (j = 0; j < i; j++)
1333                 crypto_unregister_skcipher(&aes_algs[j]);
1334
1335         return err;
1336 }
1337
1338 static void sahara_unregister_algs(struct sahara_dev *dev)
1339 {
1340         unsigned int i;
1341
1342         for (i = 0; i < ARRAY_SIZE(aes_algs); i++)
1343                 crypto_unregister_skcipher(&aes_algs[i]);
1344
1345         for (i = 0; i < ARRAY_SIZE(sha_v3_algs); i++)
1346                 crypto_unregister_ahash(&sha_v3_algs[i]);
1347
1348         if (dev->version > SAHARA_VERSION_3)
1349                 for (i = 0; i < ARRAY_SIZE(sha_v4_algs); i++)
1350                         crypto_unregister_ahash(&sha_v4_algs[i]);
1351 }
1352
1353 static const struct of_device_id sahara_dt_ids[] = {
1354         { .compatible = "fsl,imx53-sahara" },
1355         { .compatible = "fsl,imx27-sahara" },
1356         { /* sentinel */ }
1357 };
1358 MODULE_DEVICE_TABLE(of, sahara_dt_ids);
1359
1360 static int sahara_probe(struct platform_device *pdev)
1361 {
1362         struct sahara_dev *dev;
1363         u32 version;
1364         int irq;
1365         int err;
1366         int i;
1367
1368         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
1369         if (!dev)
1370                 return -ENOMEM;
1371
1372         dev->device = &pdev->dev;
1373         platform_set_drvdata(pdev, dev);
1374
1375         /* Get the base address */
1376         dev->regs_base = devm_platform_ioremap_resource(pdev, 0);
1377         if (IS_ERR(dev->regs_base))
1378                 return PTR_ERR(dev->regs_base);
1379
1380         /* Get the IRQ */
1381         irq = platform_get_irq(pdev,  0);
1382         if (irq < 0)
1383                 return irq;
1384
1385         err = devm_request_irq(&pdev->dev, irq, sahara_irq_handler,
1386                                0, dev_name(&pdev->dev), dev);
1387         if (err) {
1388                 dev_err(&pdev->dev, "failed to request irq\n");
1389                 return err;
1390         }
1391
1392         /* clocks */
1393         dev->clk_ipg = devm_clk_get(&pdev->dev, "ipg");
1394         if (IS_ERR(dev->clk_ipg)) {
1395                 dev_err(&pdev->dev, "Could not get ipg clock\n");
1396                 return PTR_ERR(dev->clk_ipg);
1397         }
1398
1399         dev->clk_ahb = devm_clk_get(&pdev->dev, "ahb");
1400         if (IS_ERR(dev->clk_ahb)) {
1401                 dev_err(&pdev->dev, "Could not get ahb clock\n");
1402                 return PTR_ERR(dev->clk_ahb);
1403         }
1404
1405         /* Allocate HW descriptors */
1406         dev->hw_desc[0] = dmam_alloc_coherent(&pdev->dev,
1407                         SAHARA_MAX_HW_DESC * sizeof(struct sahara_hw_desc),
1408                         &dev->hw_phys_desc[0], GFP_KERNEL);
1409         if (!dev->hw_desc[0]) {
1410                 dev_err(&pdev->dev, "Could not allocate hw descriptors\n");
1411                 return -ENOMEM;
1412         }
1413         dev->hw_desc[1] = dev->hw_desc[0] + 1;
1414         dev->hw_phys_desc[1] = dev->hw_phys_desc[0] +
1415                                 sizeof(struct sahara_hw_desc);
1416
1417         /* Allocate space for iv and key */
1418         dev->key_base = dmam_alloc_coherent(&pdev->dev, 2 * AES_KEYSIZE_128,
1419                                 &dev->key_phys_base, GFP_KERNEL);
1420         if (!dev->key_base) {
1421                 dev_err(&pdev->dev, "Could not allocate memory for key\n");
1422                 return -ENOMEM;
1423         }
1424         dev->iv_base = dev->key_base + AES_KEYSIZE_128;
1425         dev->iv_phys_base = dev->key_phys_base + AES_KEYSIZE_128;
1426
1427         /* Allocate space for context: largest digest + message length field */
1428         dev->context_base = dmam_alloc_coherent(&pdev->dev,
1429                                         SHA256_DIGEST_SIZE + 4,
1430                                         &dev->context_phys_base, GFP_KERNEL);
1431         if (!dev->context_base) {
1432                 dev_err(&pdev->dev, "Could not allocate memory for MDHA context\n");
1433                 return -ENOMEM;
1434         }
1435
1436         /* Allocate space for HW links */
1437         dev->hw_link[0] = dmam_alloc_coherent(&pdev->dev,
1438                         SAHARA_MAX_HW_LINK * sizeof(struct sahara_hw_link),
1439                         &dev->hw_phys_link[0], GFP_KERNEL);
1440         if (!dev->hw_link[0]) {
1441                 dev_err(&pdev->dev, "Could not allocate hw links\n");
1442                 return -ENOMEM;
1443         }
1444         for (i = 1; i < SAHARA_MAX_HW_LINK; i++) {
1445                 dev->hw_phys_link[i] = dev->hw_phys_link[i - 1] +
1446                                         sizeof(struct sahara_hw_link);
1447                 dev->hw_link[i] = dev->hw_link[i - 1] + 1;
1448         }
1449
1450         crypto_init_queue(&dev->queue, SAHARA_QUEUE_LENGTH);
1451
1452         mutex_init(&dev->queue_mutex);
1453
1454         dev_ptr = dev;
1455
1456         dev->kthread = kthread_run(sahara_queue_manage, dev, "sahara_crypto");
1457         if (IS_ERR(dev->kthread)) {
1458                 return PTR_ERR(dev->kthread);
1459         }
1460
1461         init_completion(&dev->dma_completion);
1462
1463         err = clk_prepare_enable(dev->clk_ipg);
1464         if (err)
1465                 return err;
1466         err = clk_prepare_enable(dev->clk_ahb);
1467         if (err)
1468                 goto clk_ipg_disable;
1469
1470         version = sahara_read(dev, SAHARA_REG_VERSION);
1471         if (of_device_is_compatible(pdev->dev.of_node, "fsl,imx27-sahara")) {
1472                 if (version != SAHARA_VERSION_3)
1473                         err = -ENODEV;
1474         } else if (of_device_is_compatible(pdev->dev.of_node,
1475                         "fsl,imx53-sahara")) {
1476                 if (((version >> 8) & 0xff) != SAHARA_VERSION_4)
1477                         err = -ENODEV;
1478                 version = (version >> 8) & 0xff;
1479         }
1480         if (err == -ENODEV) {
1481                 dev_err(&pdev->dev, "SAHARA version %d not supported\n",
1482                                 version);
1483                 goto err_algs;
1484         }
1485
1486         dev->version = version;
1487
1488         sahara_write(dev, SAHARA_CMD_RESET | SAHARA_CMD_MODE_BATCH,
1489                      SAHARA_REG_CMD);
1490         sahara_write(dev, SAHARA_CONTROL_SET_THROTTLE(0) |
1491                         SAHARA_CONTROL_SET_MAXBURST(8) |
1492                         SAHARA_CONTROL_RNG_AUTORSD |
1493                         SAHARA_CONTROL_ENABLE_INT,
1494                         SAHARA_REG_CONTROL);
1495
1496         err = sahara_register_algs(dev);
1497         if (err)
1498                 goto err_algs;
1499
1500         dev_info(&pdev->dev, "SAHARA version %d initialized\n", version);
1501
1502         return 0;
1503
1504 err_algs:
1505         kthread_stop(dev->kthread);
1506         dev_ptr = NULL;
1507         clk_disable_unprepare(dev->clk_ahb);
1508 clk_ipg_disable:
1509         clk_disable_unprepare(dev->clk_ipg);
1510
1511         return err;
1512 }
1513
1514 static int sahara_remove(struct platform_device *pdev)
1515 {
1516         struct sahara_dev *dev = platform_get_drvdata(pdev);
1517
1518         kthread_stop(dev->kthread);
1519
1520         sahara_unregister_algs(dev);
1521
1522         clk_disable_unprepare(dev->clk_ipg);
1523         clk_disable_unprepare(dev->clk_ahb);
1524
1525         dev_ptr = NULL;
1526
1527         return 0;
1528 }
1529
1530 static struct platform_driver sahara_driver = {
1531         .probe          = sahara_probe,
1532         .remove         = sahara_remove,
1533         .driver         = {
1534                 .name   = SAHARA_NAME,
1535                 .of_match_table = sahara_dt_ids,
1536         },
1537 };
1538
1539 module_platform_driver(sahara_driver);
1540
1541 MODULE_LICENSE("GPL");
1542 MODULE_AUTHOR("Javier Martin <javier.martin@vista-silicon.com>");
1543 MODULE_AUTHOR("Steffen Trumtrar <s.trumtrar@pengutronix.de>");
1544 MODULE_DESCRIPTION("SAHARA2 HW crypto accelerator");