drivers/crypto/atmel-tdes.c

   1 /*
   2  * Cryptographic API.
   3  *
   4  * Support for ATMEL DES/TDES HW acceleration.
   5  *
   6  * Copyright (c) 2012 Eukréa Electromatique - ATMEL
   7  * Author: Nicolas Royer <nicolas@eukrea.com>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as published
  11  * by the Free Software Foundation.
  12  *
  13  * Some ideas are from omap-aes.c drivers.
  14  */
  15
  16
  17 #include <linux/kernel.h>
  18 #include <linux/module.h>
  19 #include <linux/slab.h>
  20 #include <linux/err.h>
  21 #include <linux/clk.h>
  22 #include <linux/io.h>
  23 #include <linux/hw_random.h>
  24 #include <linux/platform_device.h>
  25
  26 #include <linux/device.h>
  27 #include <linux/init.h>
  28 #include <linux/errno.h>
  29 #include <linux/interrupt.h>
  30 #include <linux/irq.h>
  31 #include <linux/scatterlist.h>
  32 #include <linux/dma-mapping.h>
  33 #include <linux/of_device.h>
  34 #include <linux/delay.h>
  35 #include <linux/crypto.h>
  36 #include <linux/cryptohash.h>
  37 #include <crypto/scatterwalk.h>
  38 #include <crypto/algapi.h>
  39 #include <crypto/des.h>
  40 #include <crypto/hash.h>
  41 #include <crypto/internal/hash.h>
  42 #include <linux/platform_data/crypto-atmel.h>
  43 #include "atmel-tdes-regs.h"
  44
  45 /* TDES flags  */
  46 #define TDES_FLAGS_MODE_MASK            0x00ff
  47 #define TDES_FLAGS_ENCRYPT      BIT(0)
  48 #define TDES_FLAGS_CBC          BIT(1)
  49 #define TDES_FLAGS_CFB          BIT(2)
  50 #define TDES_FLAGS_CFB8         BIT(3)
  51 #define TDES_FLAGS_CFB16        BIT(4)
  52 #define TDES_FLAGS_CFB32        BIT(5)
  53 #define TDES_FLAGS_CFB64        BIT(6)
  54 #define TDES_FLAGS_OFB          BIT(7)
  55
  56 #define TDES_FLAGS_INIT         BIT(16)
  57 #define TDES_FLAGS_FAST         BIT(17)
  58 #define TDES_FLAGS_BUSY         BIT(18)
  59 #define TDES_FLAGS_DMA          BIT(19)
  60
  61 #define ATMEL_TDES_QUEUE_LENGTH 50
  62
  63 #define CFB8_BLOCK_SIZE         1
  64 #define CFB16_BLOCK_SIZE        2
  65 #define CFB32_BLOCK_SIZE        4
  66
  67 struct atmel_tdes_caps {
  68         bool    has_dma;
  69         u32             has_cfb_3keys;
  70 };
  71
  72 struct atmel_tdes_dev;
  73
  74 struct atmel_tdes_ctx {
  75         struct atmel_tdes_dev *dd;
  76
  77         int             keylen;
  78         u32             key[3*DES_KEY_SIZE / sizeof(u32)];
  79         unsigned long   flags;
  80
  81         u16             block_size;
  82 };
  83
  84 struct atmel_tdes_reqctx {
  85         unsigned long mode;
  86 };
  87
  88 struct atmel_tdes_dma {
  89         struct dma_chan                 *chan;
  90         struct dma_slave_config dma_conf;
  91 };
  92
  93 struct atmel_tdes_dev {
  94         struct list_head        list;
  95         unsigned long           phys_base;
  96         void __iomem            *io_base;
  97
  98         struct atmel_tdes_ctx   *ctx;
  99         struct device           *dev;
 100         struct clk                      *iclk;
 101         int                                     irq;
 102
 103         unsigned long           flags;
 104         int                     err;
 105
 106         spinlock_t              lock;
 107         struct crypto_queue     queue;
 108
 109         struct tasklet_struct   done_task;
 110         struct tasklet_struct   queue_task;
 111
 112         struct ablkcipher_request       *req;
 113         size_t                          total;
 114
 115         struct scatterlist      *in_sg;
 116         unsigned int            nb_in_sg;
 117         size_t                          in_offset;
 118         struct scatterlist      *out_sg;
 119         unsigned int            nb_out_sg;
 120         size_t                          out_offset;
 121
 122         size_t  buflen;
 123         size_t  dma_size;
 124
 125         void    *buf_in;
 126         int             dma_in;
 127         dma_addr_t      dma_addr_in;
 128         struct atmel_tdes_dma   dma_lch_in;
 129
 130         void    *buf_out;
 131         int             dma_out;
 132         dma_addr_t      dma_addr_out;
 133         struct atmel_tdes_dma   dma_lch_out;
 134
 135         struct atmel_tdes_caps  caps;
 136
 137         u32     hw_version;
 138 };
 139
 140 struct atmel_tdes_drv {
 141         struct list_head        dev_list;
 142         spinlock_t              lock;
 143 };
 144
 145 static struct atmel_tdes_drv atmel_tdes = {
 146         .dev_list = LIST_HEAD_INIT(atmel_tdes.dev_list),
 147         .lock = __SPIN_LOCK_UNLOCKED(atmel_tdes.lock),
 148 };
 149
 150 static int atmel_tdes_sg_copy(struct scatterlist **sg, size_t *offset,
 151                         void *buf, size_t buflen, size_t total, int out)
 152 {
 153         unsigned int count, off = 0;
 154
 155         while (buflen && total) {
 156                 count = min((*sg)->length - *offset, total);
 157                 count = min(count, buflen);
 158
 159                 if (!count)
 160                         return off;
 161
 162                 scatterwalk_map_and_copy(buf + off, *sg, *offset, count, out);
 163
 164                 off += count;
 165                 buflen -= count;
 166                 *offset += count;
 167                 total -= count;
 168
 169                 if (*offset == (*sg)->length) {
 170                         *sg = sg_next(*sg);
 171                         if (*sg)
 172                                 *offset = 0;
 173                         else
 174                                 total = 0;
 175                 }
 176         }
 177
 178         return off;
 179 }
 180
 181 static inline u32 atmel_tdes_read(struct atmel_tdes_dev *dd, u32 offset)
 182 {
 183         return readl_relaxed(dd->io_base + offset);
 184 }
 185
 186 static inline void atmel_tdes_write(struct atmel_tdes_dev *dd,
 187                                         u32 offset, u32 value)
 188 {
 189         writel_relaxed(value, dd->io_base + offset);
 190 }
 191
 192 static void atmel_tdes_write_n(struct atmel_tdes_dev *dd, u32 offset,
 193                                         u32 *value, int count)
 194 {
 195         for (; count--; value++, offset += 4)
 196                 atmel_tdes_write(dd, offset, *value);
 197 }
 198
 199 static struct atmel_tdes_dev *atmel_tdes_find_dev(struct atmel_tdes_ctx *ctx)
 200 {
 201         struct atmel_tdes_dev *tdes_dd = NULL;
 202         struct atmel_tdes_dev *tmp;
 203
 204         spin_lock_bh(&atmel_tdes.lock);
 205         if (!ctx->dd) {
 206                 list_for_each_entry(tmp, &atmel_tdes.dev_list, list) {
 207                         tdes_dd = tmp;
 208                         break;
 209                 }
 210                 ctx->dd = tdes_dd;
 211         } else {
 212                 tdes_dd = ctx->dd;
 213         }
 214         spin_unlock_bh(&atmel_tdes.lock);
 215
 216         return tdes_dd;
 217 }
 218
 219 static int atmel_tdes_hw_init(struct atmel_tdes_dev *dd)
 220 {
 221         int err;
 222
 223         err = clk_prepare_enable(dd->iclk);
 224         if (err)
 225                 return err;
 226
 227         if (!(dd->flags & TDES_FLAGS_INIT)) {
 228                 atmel_tdes_write(dd, TDES_CR, TDES_CR_SWRST);
 229                 dd->flags |= TDES_FLAGS_INIT;
 230                 dd->err = 0;
 231         }
 232
 233         return 0;
 234 }
 235
 236 static inline unsigned int atmel_tdes_get_version(struct atmel_tdes_dev *dd)
 237 {
 238         return atmel_tdes_read(dd, TDES_HW_VERSION) & 0x00000fff;
 239 }
 240
 241 static void atmel_tdes_hw_version_init(struct atmel_tdes_dev *dd)
 242 {
 243         atmel_tdes_hw_init(dd);
 244
 245         dd->hw_version = atmel_tdes_get_version(dd);
 246
 247         dev_info(dd->dev,
 248                         "version: 0x%x\n", dd->hw_version);
 249
 250         clk_disable_unprepare(dd->iclk);
 251 }
 252
 253 static void atmel_tdes_dma_callback(void *data)
 254 {
 255         struct atmel_tdes_dev *dd = data;
 256
 257         /* dma_lch_out - completed */
 258         tasklet_schedule(&dd->done_task);
 259 }
 260
 261 static int atmel_tdes_write_ctrl(struct atmel_tdes_dev *dd)
 262 {
 263         int err;
 264         u32 valcr = 0, valmr = TDES_MR_SMOD_PDC;
 265
 266         err = atmel_tdes_hw_init(dd);
 267
 268         if (err)
 269                 return err;
 270
 271         if (!dd->caps.has_dma)
 272                 atmel_tdes_write(dd, TDES_PTCR,
 273                         TDES_PTCR_TXTDIS | TDES_PTCR_RXTDIS);
 274
 275         /* MR register must be set before IV registers */
 276         if (dd->ctx->keylen > (DES_KEY_SIZE << 1)) {
 277                 valmr |= TDES_MR_KEYMOD_3KEY;
 278                 valmr |= TDES_MR_TDESMOD_TDES;
 279         } else if (dd->ctx->keylen > DES_KEY_SIZE) {
 280                 valmr |= TDES_MR_KEYMOD_2KEY;
 281                 valmr |= TDES_MR_TDESMOD_TDES;
 282         } else {
 283                 valmr |= TDES_MR_TDESMOD_DES;
 284         }
 285
 286         if (dd->flags & TDES_FLAGS_CBC) {
 287                 valmr |= TDES_MR_OPMOD_CBC;
 288         } else if (dd->flags & TDES_FLAGS_CFB) {
 289                 valmr |= TDES_MR_OPMOD_CFB;
 290
 291                 if (dd->flags & TDES_FLAGS_CFB8)
 292                         valmr |= TDES_MR_CFBS_8b;
 293                 else if (dd->flags & TDES_FLAGS_CFB16)
 294                         valmr |= TDES_MR_CFBS_16b;
 295                 else if (dd->flags & TDES_FLAGS_CFB32)
 296                         valmr |= TDES_MR_CFBS_32b;
 297                 else if (dd->flags & TDES_FLAGS_CFB64)
 298                         valmr |= TDES_MR_CFBS_64b;
 299         } else if (dd->flags & TDES_FLAGS_OFB) {
 300                 valmr |= TDES_MR_OPMOD_OFB;
 301         }
 302
 303         if ((dd->flags & TDES_FLAGS_ENCRYPT) || (dd->flags & TDES_FLAGS_OFB))
 304                 valmr |= TDES_MR_CYPHER_ENC;
 305
 306         atmel_tdes_write(dd, TDES_CR, valcr);
 307         atmel_tdes_write(dd, TDES_MR, valmr);
 308
 309         atmel_tdes_write_n(dd, TDES_KEY1W1R, dd->ctx->key,
 310                                                 dd->ctx->keylen >> 2);
 311
 312         if (((dd->flags & TDES_FLAGS_CBC) || (dd->flags & TDES_FLAGS_CFB) ||
 313                 (dd->flags & TDES_FLAGS_OFB)) && dd->req->info) {
 314                 atmel_tdes_write_n(dd, TDES_IV1R, dd->req->info, 2);
 315         }
 316
 317         return 0;
 318 }
 319
 320 static int atmel_tdes_crypt_pdc_stop(struct atmel_tdes_dev *dd)
 321 {
 322         int err = 0;
 323         size_t count;
 324
 325         atmel_tdes_write(dd, TDES_PTCR, TDES_PTCR_TXTDIS|TDES_PTCR_RXTDIS);
 326
 327         if (dd->flags & TDES_FLAGS_FAST) {
 328                 dma_unmap_sg(dd->dev, dd->out_sg, 1, DMA_FROM_DEVICE);
 329                 dma_unmap_sg(dd->dev, dd->in_sg, 1, DMA_TO_DEVICE);
 330         } else {
 331                 dma_sync_single_for_device(dd->dev, dd->dma_addr_out,
 332                                            dd->dma_size, DMA_FROM_DEVICE);
 333
 334                 /* copy data */
 335                 count = atmel_tdes_sg_copy(&dd->out_sg, &dd->out_offset,
 336                                 dd->buf_out, dd->buflen, dd->dma_size, 1);
 337                 if (count != dd->dma_size) {
 338                         err = -EINVAL;
 339                         pr_err("not all data converted: %u\n", count);
 340                 }
 341         }
 342
 343         return err;
 344 }
 345
 346 static int atmel_tdes_buff_init(struct atmel_tdes_dev *dd)
 347 {
 348         int err = -ENOMEM;
 349
 350         dd->buf_in = (void *)__get_free_pages(GFP_KERNEL, 0);
 351         dd->buf_out = (void *)__get_free_pages(GFP_KERNEL, 0);
 352         dd->buflen = PAGE_SIZE;
 353         dd->buflen &= ~(DES_BLOCK_SIZE - 1);
 354
 355         if (!dd->buf_in || !dd->buf_out) {
 356                 dev_err(dd->dev, "unable to alloc pages.\n");
 357                 goto err_alloc;
 358         }
 359
 360         /* MAP here */
 361         dd->dma_addr_in = dma_map_single(dd->dev, dd->buf_in,
 362                                         dd->buflen, DMA_TO_DEVICE);
 363         if (dma_mapping_error(dd->dev, dd->dma_addr_in)) {
 364                 dev_err(dd->dev, "dma %d bytes error\n", dd->buflen);
 365                 err = -EINVAL;
 366                 goto err_map_in;
 367         }
 368
 369         dd->dma_addr_out = dma_map_single(dd->dev, dd->buf_out,
 370                                         dd->buflen, DMA_FROM_DEVICE);
 371         if (dma_mapping_error(dd->dev, dd->dma_addr_out)) {
 372                 dev_err(dd->dev, "dma %d bytes error\n", dd->buflen);
 373                 err = -EINVAL;
 374                 goto err_map_out;
 375         }
 376
 377         return 0;
 378
 379 err_map_out:
 380         dma_unmap_single(dd->dev, dd->dma_addr_in, dd->buflen,
 381                 DMA_TO_DEVICE);
 382 err_map_in:
 383 err_alloc:
 384         free_page((unsigned long)dd->buf_out);
 385         free_page((unsigned long)dd->buf_in);
 386         if (err)
 387                 pr_err("error: %d\n", err);
 388         return err;
 389 }
 390
 391 static void atmel_tdes_buff_cleanup(struct atmel_tdes_dev *dd)
 392 {
 393         dma_unmap_single(dd->dev, dd->dma_addr_out, dd->buflen,
 394                          DMA_FROM_DEVICE);
 395         dma_unmap_single(dd->dev, dd->dma_addr_in, dd->buflen,
 396                 DMA_TO_DEVICE);
 397         free_page((unsigned long)dd->buf_out);
 398         free_page((unsigned long)dd->buf_in);
 399 }
 400
 401 static int atmel_tdes_crypt_pdc(struct crypto_tfm *tfm, dma_addr_t dma_addr_in,
 402                                dma_addr_t dma_addr_out, int length)
 403 {
 404         struct atmel_tdes_ctx *ctx = crypto_tfm_ctx(tfm);
 405         struct atmel_tdes_dev *dd = ctx->dd;
 406         int len32;
 407
 408         dd->dma_size = length;
 409
 410         if (!(dd->flags & TDES_FLAGS_FAST)) {
 411                 dma_sync_single_for_device(dd->dev, dma_addr_in, length,
 412                                            DMA_TO_DEVICE);
 413         }
 414
 415         if ((dd->flags & TDES_FLAGS_CFB) && (dd->flags & TDES_FLAGS_CFB8))
 416                 len32 = DIV_ROUND_UP(length, sizeof(u8));
 417         else if ((dd->flags & TDES_FLAGS_CFB) && (dd->flags & TDES_FLAGS_CFB16))
 418                 len32 = DIV_ROUND_UP(length, sizeof(u16));
 419         else
 420                 len32 = DIV_ROUND_UP(length, sizeof(u32));
 421
 422         atmel_tdes_write(dd, TDES_PTCR, TDES_PTCR_TXTDIS|TDES_PTCR_RXTDIS);
 423         atmel_tdes_write(dd, TDES_TPR, dma_addr_in);
 424         atmel_tdes_write(dd, TDES_TCR, len32);
 425         atmel_tdes_write(dd, TDES_RPR, dma_addr_out);
 426         atmel_tdes_write(dd, TDES_RCR, len32);
 427
 428         /* Enable Interrupt */
 429         atmel_tdes_write(dd, TDES_IER, TDES_INT_ENDRX);
 430
 431         /* Start DMA transfer */
 432         atmel_tdes_write(dd, TDES_PTCR, TDES_PTCR_TXTEN | TDES_PTCR_RXTEN);
 433
 434         return 0;
 435 }
 436
 437 static int atmel_tdes_crypt_dma(struct crypto_tfm *tfm, dma_addr_t dma_addr_in,
 438                                dma_addr_t dma_addr_out, int length)
 439 {
 440         struct atmel_tdes_ctx *ctx = crypto_tfm_ctx(tfm);
 441         struct atmel_tdes_dev *dd = ctx->dd;
 442         struct scatterlist sg[2];
 443         struct dma_async_tx_descriptor  *in_desc, *out_desc;
 444
 445         dd->dma_size = length;
 446
 447         if (!(dd->flags & TDES_FLAGS_FAST)) {
 448                 dma_sync_single_for_device(dd->dev, dma_addr_in, length,
 449                                            DMA_TO_DEVICE);
 450         }
 451
 452         if (dd->flags & TDES_FLAGS_CFB8) {
 453                 dd->dma_lch_in.dma_conf.dst_addr_width =
 454                         DMA_SLAVE_BUSWIDTH_1_BYTE;
 455                 dd->dma_lch_out.dma_conf.src_addr_width =
 456                         DMA_SLAVE_BUSWIDTH_1_BYTE;
 457         } else if (dd->flags & TDES_FLAGS_CFB16) {
 458                 dd->dma_lch_in.dma_conf.dst_addr_width =
 459                         DMA_SLAVE_BUSWIDTH_2_BYTES;
 460                 dd->dma_lch_out.dma_conf.src_addr_width =
 461                         DMA_SLAVE_BUSWIDTH_2_BYTES;
 462         } else {
 463                 dd->dma_lch_in.dma_conf.dst_addr_width =
 464                         DMA_SLAVE_BUSWIDTH_4_BYTES;
 465                 dd->dma_lch_out.dma_conf.src_addr_width =
 466                         DMA_SLAVE_BUSWIDTH_4_BYTES;
 467         }
 468
 469         dmaengine_slave_config(dd->dma_lch_in.chan, &dd->dma_lch_in.dma_conf);
 470         dmaengine_slave_config(dd->dma_lch_out.chan, &dd->dma_lch_out.dma_conf);
 471
 472         dd->flags |= TDES_FLAGS_DMA;
 473
 474         sg_init_table(&sg[0], 1);
 475         sg_dma_address(&sg[0]) = dma_addr_in;
 476         sg_dma_len(&sg[0]) = length;
 477
 478         sg_init_table(&sg[1], 1);
 479         sg_dma_address(&sg[1]) = dma_addr_out;
 480         sg_dma_len(&sg[1]) = length;
 481
 482         in_desc = dmaengine_prep_slave_sg(dd->dma_lch_in.chan, &sg[0],
 483                                 1, DMA_MEM_TO_DEV,
 484                                 DMA_PREP_INTERRUPT  |  DMA_CTRL_ACK);
 485         if (!in_desc)
 486                 return -EINVAL;
 487
 488         out_desc = dmaengine_prep_slave_sg(dd->dma_lch_out.chan, &sg[1],
 489                                 1, DMA_DEV_TO_MEM,
 490                                 DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
 491         if (!out_desc)
 492                 return -EINVAL;
 493
 494         out_desc->callback = atmel_tdes_dma_callback;
 495         out_desc->callback_param = dd;
 496
 497         dmaengine_submit(out_desc);
 498         dma_async_issue_pending(dd->dma_lch_out.chan);
 499
 500         dmaengine_submit(in_desc);
 501         dma_async_issue_pending(dd->dma_lch_in.chan);
 502
 503         return 0;
 504 }
 505
 506 static int atmel_tdes_crypt_start(struct atmel_tdes_dev *dd)
 507 {
 508         struct crypto_tfm *tfm = crypto_ablkcipher_tfm(
 509                                         crypto_ablkcipher_reqtfm(dd->req));
 510         int err, fast = 0, in, out;
 511         size_t count;
 512         dma_addr_t addr_in, addr_out;
 513
 514         if ((!dd->in_offset) && (!dd->out_offset)) {
 515                 /* check for alignment */
 516                 in = IS_ALIGNED((u32)dd->in_sg->offset, sizeof(u32)) &&
 517                         IS_ALIGNED(dd->in_sg->length, dd->ctx->block_size);
 518                 out = IS_ALIGNED((u32)dd->out_sg->offset, sizeof(u32)) &&
 519                         IS_ALIGNED(dd->out_sg->length, dd->ctx->block_size);
 520                 fast = in && out;
 521
 522                 if (sg_dma_len(dd->in_sg) != sg_dma_len(dd->out_sg))
 523                         fast = 0;
 524         }
 525
 526
 527         if (fast)  {
 528                 count = min(dd->total, sg_dma_len(dd->in_sg));
 529                 count = min(count, sg_dma_len(dd->out_sg));
 530
 531                 err = dma_map_sg(dd->dev, dd->in_sg, 1, DMA_TO_DEVICE);
 532                 if (!err) {
 533                         dev_err(dd->dev, "dma_map_sg() error\n");
 534                         return -EINVAL;
 535                 }
 536
 537                 err = dma_map_sg(dd->dev, dd->out_sg, 1,
 538                                 DMA_FROM_DEVICE);
 539                 if (!err) {
 540                         dev_err(dd->dev, "dma_map_sg() error\n");
 541                         dma_unmap_sg(dd->dev, dd->in_sg, 1,
 542                                 DMA_TO_DEVICE);
 543                         return -EINVAL;
 544                 }
 545
 546                 addr_in = sg_dma_address(dd->in_sg);
 547                 addr_out = sg_dma_address(dd->out_sg);
 548
 549                 dd->flags |= TDES_FLAGS_FAST;
 550
 551         } else {
 552                 /* use cache buffers */
 553                 count = atmel_tdes_sg_copy(&dd->in_sg, &dd->in_offset,
 554                                 dd->buf_in, dd->buflen, dd->total, 0);
 555
 556                 addr_in = dd->dma_addr_in;
 557                 addr_out = dd->dma_addr_out;
 558
 559                 dd->flags &= ~TDES_FLAGS_FAST;
 560         }
 561
 562         dd->total -= count;
 563
 564         if (dd->caps.has_dma)
 565                 err = atmel_tdes_crypt_dma(tfm, addr_in, addr_out, count);
 566         else
 567                 err = atmel_tdes_crypt_pdc(tfm, addr_in, addr_out, count);
 568
 569         if (err && (dd->flags & TDES_FLAGS_FAST)) {
 570                 dma_unmap_sg(dd->dev, dd->in_sg, 1, DMA_TO_DEVICE);
 571                 dma_unmap_sg(dd->dev, dd->out_sg, 1, DMA_TO_DEVICE);
 572         }
 573
 574         return err;
 575 }
 576
 577 static void atmel_tdes_finish_req(struct atmel_tdes_dev *dd, int err)
 578 {
 579         struct ablkcipher_request *req = dd->req;
 580
 581         clk_disable_unprepare(dd->iclk);
 582
 583         dd->flags &= ~TDES_FLAGS_BUSY;
 584
 585         req->base.complete(&req->base, err);
 586 }
 587
 588 static int atmel_tdes_handle_queue(struct atmel_tdes_dev *dd,
 589                                struct ablkcipher_request *req)
 590 {
 591         struct crypto_async_request *async_req, *backlog;
 592         struct atmel_tdes_ctx *ctx;
 593         struct atmel_tdes_reqctx *rctx;
 594         unsigned long flags;
 595         int err, ret = 0;
 596
 597         spin_lock_irqsave(&dd->lock, flags);
 598         if (req)
 599                 ret = ablkcipher_enqueue_request(&dd->queue, req);
 600         if (dd->flags & TDES_FLAGS_BUSY) {
 601                 spin_unlock_irqrestore(&dd->lock, flags);
 602                 return ret;
 603         }
 604         backlog = crypto_get_backlog(&dd->queue);
 605         async_req = crypto_dequeue_request(&dd->queue);
 606         if (async_req)
 607                 dd->flags |= TDES_FLAGS_BUSY;
 608         spin_unlock_irqrestore(&dd->lock, flags);
 609
 610         if (!async_req)
 611                 return ret;
 612
 613         if (backlog)
 614                 backlog->complete(backlog, -EINPROGRESS);
 615
 616         req = ablkcipher_request_cast(async_req);
 617
 618         /* assign new request to device */
 619         dd->req = req;
 620         dd->total = req->nbytes;
 621         dd->in_offset = 0;
 622         dd->in_sg = req->src;
 623         dd->out_offset = 0;
 624         dd->out_sg = req->dst;
 625
 626         rctx = ablkcipher_request_ctx(req);
 627         ctx = crypto_ablkcipher_ctx(crypto_ablkcipher_reqtfm(req));
 628         rctx->mode &= TDES_FLAGS_MODE_MASK;
 629         dd->flags = (dd->flags & ~TDES_FLAGS_MODE_MASK) | rctx->mode;
 630         dd->ctx = ctx;
 631         ctx->dd = dd;
 632
 633         err = atmel_tdes_write_ctrl(dd);
 634         if (!err)
 635                 err = atmel_tdes_crypt_start(dd);
 636         if (err) {
 637                 /* des_task will not finish it, so do it here */
 638                 atmel_tdes_finish_req(dd, err);
 639                 tasklet_schedule(&dd->queue_task);
 640         }
 641
 642         return ret;
 643 }
 644
 645 static int atmel_tdes_crypt_dma_stop(struct atmel_tdes_dev *dd)
 646 {
 647         int err = -EINVAL;
 648         size_t count;
 649
 650         if (dd->flags & TDES_FLAGS_DMA) {
 651                 err = 0;
 652                 if  (dd->flags & TDES_FLAGS_FAST) {
 653                         dma_unmap_sg(dd->dev, dd->out_sg, 1, DMA_FROM_DEVICE);
 654                         dma_unmap_sg(dd->dev, dd->in_sg, 1, DMA_TO_DEVICE);
 655                 } else {
 656                         dma_sync_single_for_device(dd->dev, dd->dma_addr_out,
 657                                 dd->dma_size, DMA_FROM_DEVICE);
 658
 659                         /* copy data */
 660                         count = atmel_tdes_sg_copy(&dd->out_sg, &dd->out_offset,
 661                                 dd->buf_out, dd->buflen, dd->dma_size, 1);
 662                         if (count != dd->dma_size) {
 663                                 err = -EINVAL;
 664                                 pr_err("not all data converted: %u\n", count);
 665                         }
 666                 }
 667         }
 668         return err;
 669 }
 670
 671 static int atmel_tdes_crypt(struct ablkcipher_request *req, unsigned long mode)
 672 {
 673         struct atmel_tdes_ctx *ctx = crypto_ablkcipher_ctx(
 674                         crypto_ablkcipher_reqtfm(req));
 675         struct atmel_tdes_reqctx *rctx = ablkcipher_request_ctx(req);
 676
 677         if (mode & TDES_FLAGS_CFB8) {
 678                 if (!IS_ALIGNED(req->nbytes, CFB8_BLOCK_SIZE)) {
 679                         pr_err("request size is not exact amount of CFB8 blocks\n");
 680                         return -EINVAL;
 681                 }
 682                 ctx->block_size = CFB8_BLOCK_SIZE;
 683         } else if (mode & TDES_FLAGS_CFB16) {
 684                 if (!IS_ALIGNED(req->nbytes, CFB16_BLOCK_SIZE)) {
 685                         pr_err("request size is not exact amount of CFB16 blocks\n");
 686                         return -EINVAL;
 687                 }
 688                 ctx->block_size = CFB16_BLOCK_SIZE;
 689         } else if (mode & TDES_FLAGS_CFB32) {
 690                 if (!IS_ALIGNED(req->nbytes, CFB32_BLOCK_SIZE)) {
 691                         pr_err("request size is not exact amount of CFB32 blocks\n");
 692                         return -EINVAL;
 693                 }
 694                 ctx->block_size = CFB32_BLOCK_SIZE;
 695         } else {
 696                 if (!IS_ALIGNED(req->nbytes, DES_BLOCK_SIZE)) {
 697                         pr_err("request size is not exact amount of DES blocks\n");
 698                         return -EINVAL;
 699                 }
 700                 ctx->block_size = DES_BLOCK_SIZE;
 701         }
 702
 703         rctx->mode = mode;
 704
 705         return atmel_tdes_handle_queue(ctx->dd, req);
 706 }
 707
 708 static bool atmel_tdes_filter(struct dma_chan *chan, void *slave)
 709 {
 710         struct at_dma_slave     *sl = slave;
 711
 712         if (sl && sl->dma_dev == chan->device->dev) {
 713                 chan->private = sl;
 714                 return true;
 715         } else {
 716                 return false;
 717         }
 718 }
 719
 720 static int atmel_tdes_dma_init(struct atmel_tdes_dev *dd,
 721                         struct crypto_platform_data *pdata)
 722 {
 723         int err = -ENOMEM;
 724         dma_cap_mask_t mask;
 725
 726         dma_cap_zero(mask);
 727         dma_cap_set(DMA_SLAVE, mask);
 728
 729         /* Try to grab 2 DMA channels */
 730         dd->dma_lch_in.chan = dma_request_slave_channel_compat(mask,
 731                         atmel_tdes_filter, &pdata->dma_slave->rxdata, dd->dev, "tx");
 732         if (!dd->dma_lch_in.chan)
 733                 goto err_dma_in;
 734
 735         dd->dma_lch_in.dma_conf.direction = DMA_MEM_TO_DEV;
 736         dd->dma_lch_in.dma_conf.dst_addr = dd->phys_base +
 737                 TDES_IDATA1R;
 738         dd->dma_lch_in.dma_conf.src_maxburst = 1;
 739         dd->dma_lch_in.dma_conf.src_addr_width =
 740                 DMA_SLAVE_BUSWIDTH_4_BYTES;
 741         dd->dma_lch_in.dma_conf.dst_maxburst = 1;
 742         dd->dma_lch_in.dma_conf.dst_addr_width =
 743                 DMA_SLAVE_BUSWIDTH_4_BYTES;
 744         dd->dma_lch_in.dma_conf.device_fc = false;
 745
 746         dd->dma_lch_out.chan = dma_request_slave_channel_compat(mask,
 747                         atmel_tdes_filter, &pdata->dma_slave->txdata, dd->dev, "rx");
 748         if (!dd->dma_lch_out.chan)
 749                 goto err_dma_out;
 750
 751         dd->dma_lch_out.dma_conf.direction = DMA_DEV_TO_MEM;
 752         dd->dma_lch_out.dma_conf.src_addr = dd->phys_base +
 753                 TDES_ODATA1R;
 754         dd->dma_lch_out.dma_conf.src_maxburst = 1;
 755         dd->dma_lch_out.dma_conf.src_addr_width =
 756                 DMA_SLAVE_BUSWIDTH_4_BYTES;
 757         dd->dma_lch_out.dma_conf.dst_maxburst = 1;
 758         dd->dma_lch_out.dma_conf.dst_addr_width =
 759                 DMA_SLAVE_BUSWIDTH_4_BYTES;
 760         dd->dma_lch_out.dma_conf.device_fc = false;
 761
 762         return 0;
 763
 764 err_dma_out:
 765         dma_release_channel(dd->dma_lch_in.chan);
 766 err_dma_in:
 767         dev_warn(dd->dev, "no DMA channel available\n");
 768         return err;
 769 }
 770
 771 static void atmel_tdes_dma_cleanup(struct atmel_tdes_dev *dd)
 772 {
 773         dma_release_channel(dd->dma_lch_in.chan);
 774         dma_release_channel(dd->dma_lch_out.chan);
 775 }
 776
 777 static int atmel_des_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
 778                            unsigned int keylen)
 779 {
 780         u32 tmp[DES_EXPKEY_WORDS];
 781         int err;
 782         struct crypto_tfm *ctfm = crypto_ablkcipher_tfm(tfm);
 783
 784         struct atmel_tdes_ctx *ctx = crypto_ablkcipher_ctx(tfm);
 785
 786         if (keylen != DES_KEY_SIZE) {
 787                 crypto_ablkcipher_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
 788                 return -EINVAL;
 789         }
 790
 791         err = des_ekey(tmp, key);
 792         if (err == 0 && (ctfm->crt_flags & CRYPTO_TFM_REQ_WEAK_KEY)) {
 793                 ctfm->crt_flags |= CRYPTO_TFM_RES_WEAK_KEY;
 794                 return -EINVAL;
 795         }
 796
 797         memcpy(ctx->key, key, keylen);
 798         ctx->keylen = keylen;
 799
 800         return 0;
 801 }
 802
 803 static int atmel_tdes_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
 804                            unsigned int keylen)
 805 {
 806         struct atmel_tdes_ctx *ctx = crypto_ablkcipher_ctx(tfm);
 807         const char *alg_name;
 808
 809         alg_name = crypto_tfm_alg_name(crypto_ablkcipher_tfm(tfm));
 810
 811         /*
 812          * HW bug in cfb 3-keys mode.
 813          */
 814         if (!ctx->dd->caps.has_cfb_3keys && strstr(alg_name, "cfb")
 815                         && (keylen != 2*DES_KEY_SIZE)) {
 816                 crypto_ablkcipher_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
 817                 return -EINVAL;
 818         } else if ((keylen != 2*DES_KEY_SIZE) && (keylen != 3*DES_KEY_SIZE)) {
 819                 crypto_ablkcipher_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
 820                 return -EINVAL;
 821         }
 822
 823         memcpy(ctx->key, key, keylen);
 824         ctx->keylen = keylen;
 825
 826         return 0;
 827 }
 828
 829 static int atmel_tdes_ecb_encrypt(struct ablkcipher_request *req)
 830 {
 831         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT);
 832 }
 833
 834 static int atmel_tdes_ecb_decrypt(struct ablkcipher_request *req)
 835 {
 836         return atmel_tdes_crypt(req, 0);
 837 }
 838
 839 static int atmel_tdes_cbc_encrypt(struct ablkcipher_request *req)
 840 {
 841         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CBC);
 842 }
 843
 844 static int atmel_tdes_cbc_decrypt(struct ablkcipher_request *req)
 845 {
 846         return atmel_tdes_crypt(req, TDES_FLAGS_CBC);
 847 }
 848 static int atmel_tdes_cfb_encrypt(struct ablkcipher_request *req)
 849 {
 850         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CFB);
 851 }
 852
 853 static int atmel_tdes_cfb_decrypt(struct ablkcipher_request *req)
 854 {
 855         return atmel_tdes_crypt(req, TDES_FLAGS_CFB);
 856 }
 857
 858 static int atmel_tdes_cfb8_encrypt(struct ablkcipher_request *req)
 859 {
 860         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CFB |
 861                                                 TDES_FLAGS_CFB8);
 862 }
 863
 864 static int atmel_tdes_cfb8_decrypt(struct ablkcipher_request *req)
 865 {
 866         return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB8);
 867 }
 868
 869 static int atmel_tdes_cfb16_encrypt(struct ablkcipher_request *req)
 870 {
 871         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CFB |
 872                                                 TDES_FLAGS_CFB16);
 873 }
 874
 875 static int atmel_tdes_cfb16_decrypt(struct ablkcipher_request *req)
 876 {
 877         return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB16);
 878 }
 879
 880 static int atmel_tdes_cfb32_encrypt(struct ablkcipher_request *req)
 881 {
 882         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CFB |
 883                                                 TDES_FLAGS_CFB32);
 884 }
 885
 886 static int atmel_tdes_cfb32_decrypt(struct ablkcipher_request *req)
 887 {
 888         return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB32);
 889 }
 890
 891 static int atmel_tdes_ofb_encrypt(struct ablkcipher_request *req)
 892 {
 893         return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_OFB);
 894 }
 895
 896 static int atmel_tdes_ofb_decrypt(struct ablkcipher_request *req)
 897 {
 898         return atmel_tdes_crypt(req, TDES_FLAGS_OFB);
 899 }
 900
 901 static int atmel_tdes_cra_init(struct crypto_tfm *tfm)
 902 {
 903         struct atmel_tdes_ctx *ctx = crypto_tfm_ctx(tfm);
 904         struct atmel_tdes_dev *dd;
 905
 906         tfm->crt_ablkcipher.reqsize = sizeof(struct atmel_tdes_reqctx);
 907
 908         dd = atmel_tdes_find_dev(ctx);
 909         if (!dd)
 910                 return -ENODEV;
 911
 912         return 0;
 913 }
 914
 915 static void atmel_tdes_cra_exit(struct crypto_tfm *tfm)
 916 {
 917 }
 918
 919 static struct crypto_alg tdes_algs[] = {
 920 {
 921         .cra_name               = "ecb(des)",
 922         .cra_driver_name        = "atmel-ecb-des",
 923         .cra_priority           = 100,
 924         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 925         .cra_blocksize          = DES_BLOCK_SIZE,
 926         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
 927         .cra_alignmask          = 0x7,
 928         .cra_type               = &crypto_ablkcipher_type,
 929         .cra_module             = THIS_MODULE,
 930         .cra_init               = atmel_tdes_cra_init,
 931         .cra_exit               = atmel_tdes_cra_exit,
 932         .cra_u.ablkcipher = {
 933                 .min_keysize    = DES_KEY_SIZE,
 934                 .max_keysize    = DES_KEY_SIZE,
 935                 .setkey         = atmel_des_setkey,
 936                 .encrypt        = atmel_tdes_ecb_encrypt,
 937                 .decrypt        = atmel_tdes_ecb_decrypt,
 938         }
 939 },
 940 {
 941         .cra_name               = "cbc(des)",
 942         .cra_driver_name        = "atmel-cbc-des",
 943         .cra_priority           = 100,
 944         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 945         .cra_blocksize          = DES_BLOCK_SIZE,
 946         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
 947         .cra_alignmask          = 0x7,
 948         .cra_type               = &crypto_ablkcipher_type,
 949         .cra_module             = THIS_MODULE,
 950         .cra_init               = atmel_tdes_cra_init,
 951         .cra_exit               = atmel_tdes_cra_exit,
 952         .cra_u.ablkcipher = {
 953                 .min_keysize    = DES_KEY_SIZE,
 954                 .max_keysize    = DES_KEY_SIZE,
 955                 .ivsize         = DES_BLOCK_SIZE,
 956                 .setkey         = atmel_des_setkey,
 957                 .encrypt        = atmel_tdes_cbc_encrypt,
 958                 .decrypt        = atmel_tdes_cbc_decrypt,
 959         }
 960 },
 961 {
 962         .cra_name               = "cfb(des)",
 963         .cra_driver_name        = "atmel-cfb-des",
 964         .cra_priority           = 100,
 965         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 966         .cra_blocksize          = DES_BLOCK_SIZE,
 967         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
 968         .cra_alignmask          = 0x7,
 969         .cra_type               = &crypto_ablkcipher_type,
 970         .cra_module             = THIS_MODULE,
 971         .cra_init               = atmel_tdes_cra_init,
 972         .cra_exit               = atmel_tdes_cra_exit,
 973         .cra_u.ablkcipher = {
 974                 .min_keysize    = DES_KEY_SIZE,
 975                 .max_keysize    = DES_KEY_SIZE,
 976                 .ivsize         = DES_BLOCK_SIZE,
 977                 .setkey         = atmel_des_setkey,
 978                 .encrypt        = atmel_tdes_cfb_encrypt,
 979                 .decrypt        = atmel_tdes_cfb_decrypt,
 980         }
 981 },
 982 {
 983         .cra_name               = "cfb8(des)",
 984         .cra_driver_name        = "atmel-cfb8-des",
 985         .cra_priority           = 100,
 986         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 987         .cra_blocksize          = CFB8_BLOCK_SIZE,
 988         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
 989         .cra_alignmask          = 0,
 990         .cra_type               = &crypto_ablkcipher_type,
 991         .cra_module             = THIS_MODULE,
 992         .cra_init               = atmel_tdes_cra_init,
 993         .cra_exit               = atmel_tdes_cra_exit,
 994         .cra_u.ablkcipher = {
 995                 .min_keysize    = DES_KEY_SIZE,
 996                 .max_keysize    = DES_KEY_SIZE,
 997                 .ivsize         = DES_BLOCK_SIZE,
 998                 .setkey         = atmel_des_setkey,
 999                 .encrypt        = atmel_tdes_cfb8_encrypt,
1000                 .decrypt        = atmel_tdes_cfb8_decrypt,
1001         }
1002 },
1003 {
1004         .cra_name               = "cfb16(des)",
1005         .cra_driver_name        = "atmel-cfb16-des",
1006         .cra_priority           = 100,
1007         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1008         .cra_blocksize          = CFB16_BLOCK_SIZE,
1009         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1010         .cra_alignmask          = 0x1,
1011         .cra_type               = &crypto_ablkcipher_type,
1012         .cra_module             = THIS_MODULE,
1013         .cra_init               = atmel_tdes_cra_init,
1014         .cra_exit               = atmel_tdes_cra_exit,
1015         .cra_u.ablkcipher = {
1016                 .min_keysize    = DES_KEY_SIZE,
1017                 .max_keysize    = DES_KEY_SIZE,
1018                 .ivsize         = DES_BLOCK_SIZE,
1019                 .setkey         = atmel_des_setkey,
1020                 .encrypt        = atmel_tdes_cfb16_encrypt,
1021                 .decrypt        = atmel_tdes_cfb16_decrypt,
1022         }
1023 },
1024 {
1025         .cra_name               = "cfb32(des)",
1026         .cra_driver_name        = "atmel-cfb32-des",
1027         .cra_priority           = 100,
1028         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1029         .cra_blocksize          = CFB32_BLOCK_SIZE,
1030         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1031         .cra_alignmask          = 0x3,
1032         .cra_type               = &crypto_ablkcipher_type,
1033         .cra_module             = THIS_MODULE,
1034         .cra_init               = atmel_tdes_cra_init,
1035         .cra_exit               = atmel_tdes_cra_exit,
1036         .cra_u.ablkcipher = {
1037                 .min_keysize    = DES_KEY_SIZE,
1038                 .max_keysize    = DES_KEY_SIZE,
1039                 .ivsize         = DES_BLOCK_SIZE,
1040                 .setkey         = atmel_des_setkey,
1041                 .encrypt        = atmel_tdes_cfb32_encrypt,
1042                 .decrypt        = atmel_tdes_cfb32_decrypt,
1043         }
1044 },
1045 {
1046         .cra_name               = "ofb(des)",
1047         .cra_driver_name        = "atmel-ofb-des",
1048         .cra_priority           = 100,
1049         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1050         .cra_blocksize          = DES_BLOCK_SIZE,
1051         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1052         .cra_alignmask          = 0x7,
1053         .cra_type               = &crypto_ablkcipher_type,
1054         .cra_module             = THIS_MODULE,
1055         .cra_init               = atmel_tdes_cra_init,
1056         .cra_exit               = atmel_tdes_cra_exit,
1057         .cra_u.ablkcipher = {
1058                 .min_keysize    = DES_KEY_SIZE,
1059                 .max_keysize    = DES_KEY_SIZE,
1060                 .ivsize         = DES_BLOCK_SIZE,
1061                 .setkey         = atmel_des_setkey,
1062                 .encrypt        = atmel_tdes_ofb_encrypt,
1063                 .decrypt        = atmel_tdes_ofb_decrypt,
1064         }
1065 },
1066 {
1067         .cra_name               = "ecb(des3_ede)",
1068         .cra_driver_name        = "atmel-ecb-tdes",
1069         .cra_priority           = 100,
1070         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1071         .cra_blocksize          = DES_BLOCK_SIZE,
1072         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1073         .cra_alignmask          = 0x7,
1074         .cra_type               = &crypto_ablkcipher_type,
1075         .cra_module             = THIS_MODULE,
1076         .cra_init               = atmel_tdes_cra_init,
1077         .cra_exit               = atmel_tdes_cra_exit,
1078         .cra_u.ablkcipher = {
1079                 .min_keysize    = 2 * DES_KEY_SIZE,
1080                 .max_keysize    = 3 * DES_KEY_SIZE,
1081                 .setkey         = atmel_tdes_setkey,
1082                 .encrypt        = atmel_tdes_ecb_encrypt,
1083                 .decrypt        = atmel_tdes_ecb_decrypt,
1084         }
1085 },
1086 {
1087         .cra_name               = "cbc(des3_ede)",
1088         .cra_driver_name        = "atmel-cbc-tdes",
1089         .cra_priority           = 100,
1090         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1091         .cra_blocksize          = DES_BLOCK_SIZE,
1092         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1093         .cra_alignmask          = 0x7,
1094         .cra_type               = &crypto_ablkcipher_type,
1095         .cra_module             = THIS_MODULE,
1096         .cra_init               = atmel_tdes_cra_init,
1097         .cra_exit               = atmel_tdes_cra_exit,
1098         .cra_u.ablkcipher = {
1099                 .min_keysize    = 2*DES_KEY_SIZE,
1100                 .max_keysize    = 3*DES_KEY_SIZE,
1101                 .ivsize         = DES_BLOCK_SIZE,
1102                 .setkey         = atmel_tdes_setkey,
1103                 .encrypt        = atmel_tdes_cbc_encrypt,
1104                 .decrypt        = atmel_tdes_cbc_decrypt,
1105         }
1106 },
1107 {
1108         .cra_name               = "cfb(des3_ede)",
1109         .cra_driver_name        = "atmel-cfb-tdes",
1110         .cra_priority           = 100,
1111         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1112         .cra_blocksize          = DES_BLOCK_SIZE,
1113         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1114         .cra_alignmask          = 0x7,
1115         .cra_type               = &crypto_ablkcipher_type,
1116         .cra_module             = THIS_MODULE,
1117         .cra_init               = atmel_tdes_cra_init,
1118         .cra_exit               = atmel_tdes_cra_exit,
1119         .cra_u.ablkcipher = {
1120                 .min_keysize    = 2*DES_KEY_SIZE,
1121                 .max_keysize    = 2*DES_KEY_SIZE,
1122                 .ivsize         = DES_BLOCK_SIZE,
1123                 .setkey         = atmel_tdes_setkey,
1124                 .encrypt        = atmel_tdes_cfb_encrypt,
1125                 .decrypt        = atmel_tdes_cfb_decrypt,
1126         }
1127 },
1128 {
1129         .cra_name               = "cfb8(des3_ede)",
1130         .cra_driver_name        = "atmel-cfb8-tdes",
1131         .cra_priority           = 100,
1132         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1133         .cra_blocksize          = CFB8_BLOCK_SIZE,
1134         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1135         .cra_alignmask          = 0,
1136         .cra_type               = &crypto_ablkcipher_type,
1137         .cra_module             = THIS_MODULE,
1138         .cra_init               = atmel_tdes_cra_init,
1139         .cra_exit               = atmel_tdes_cra_exit,
1140         .cra_u.ablkcipher = {
1141                 .min_keysize    = 2*DES_KEY_SIZE,
1142                 .max_keysize    = 2*DES_KEY_SIZE,
1143                 .ivsize         = DES_BLOCK_SIZE,
1144                 .setkey         = atmel_tdes_setkey,
1145                 .encrypt        = atmel_tdes_cfb8_encrypt,
1146                 .decrypt        = atmel_tdes_cfb8_decrypt,
1147         }
1148 },
1149 {
1150         .cra_name               = "cfb16(des3_ede)",
1151         .cra_driver_name        = "atmel-cfb16-tdes",
1152         .cra_priority           = 100,
1153         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1154         .cra_blocksize          = CFB16_BLOCK_SIZE,
1155         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1156         .cra_alignmask          = 0x1,
1157         .cra_type               = &crypto_ablkcipher_type,
1158         .cra_module             = THIS_MODULE,
1159         .cra_init               = atmel_tdes_cra_init,
1160         .cra_exit               = atmel_tdes_cra_exit,
1161         .cra_u.ablkcipher = {
1162                 .min_keysize    = 2*DES_KEY_SIZE,
1163                 .max_keysize    = 2*DES_KEY_SIZE,
1164                 .ivsize         = DES_BLOCK_SIZE,
1165                 .setkey         = atmel_tdes_setkey,
1166                 .encrypt        = atmel_tdes_cfb16_encrypt,
1167                 .decrypt        = atmel_tdes_cfb16_decrypt,
1168         }
1169 },
1170 {
1171         .cra_name               = "cfb32(des3_ede)",
1172         .cra_driver_name        = "atmel-cfb32-tdes",
1173         .cra_priority           = 100,
1174         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1175         .cra_blocksize          = CFB32_BLOCK_SIZE,
1176         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1177         .cra_alignmask          = 0x3,
1178         .cra_type               = &crypto_ablkcipher_type,
1179         .cra_module             = THIS_MODULE,
1180         .cra_init               = atmel_tdes_cra_init,
1181         .cra_exit               = atmel_tdes_cra_exit,
1182         .cra_u.ablkcipher = {
1183                 .min_keysize    = 2*DES_KEY_SIZE,
1184                 .max_keysize    = 2*DES_KEY_SIZE,
1185                 .ivsize         = DES_BLOCK_SIZE,
1186                 .setkey         = atmel_tdes_setkey,
1187                 .encrypt        = atmel_tdes_cfb32_encrypt,
1188                 .decrypt        = atmel_tdes_cfb32_decrypt,
1189         }
1190 },
1191 {
1192         .cra_name               = "ofb(des3_ede)",
1193         .cra_driver_name        = "atmel-ofb-tdes",
1194         .cra_priority           = 100,
1195         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
1196         .cra_blocksize          = DES_BLOCK_SIZE,
1197         .cra_ctxsize            = sizeof(struct atmel_tdes_ctx),
1198         .cra_alignmask          = 0x7,
1199         .cra_type               = &crypto_ablkcipher_type,
1200         .cra_module             = THIS_MODULE,
1201         .cra_init               = atmel_tdes_cra_init,
1202         .cra_exit               = atmel_tdes_cra_exit,
1203         .cra_u.ablkcipher = {
1204                 .min_keysize    = 2*DES_KEY_SIZE,
1205                 .max_keysize    = 3*DES_KEY_SIZE,
1206                 .ivsize         = DES_BLOCK_SIZE,
1207                 .setkey         = atmel_tdes_setkey,
1208                 .encrypt        = atmel_tdes_ofb_encrypt,
1209                 .decrypt        = atmel_tdes_ofb_decrypt,
1210         }
1211 },
1212 };
1213
1214 static void atmel_tdes_queue_task(unsigned long data)
1215 {
1216         struct atmel_tdes_dev *dd = (struct atmel_tdes_dev *)data;
1217
1218         atmel_tdes_handle_queue(dd, NULL);
1219 }
1220
1221 static void atmel_tdes_done_task(unsigned long data)
1222 {
1223         struct atmel_tdes_dev *dd = (struct atmel_tdes_dev *) data;
1224         int err;
1225
1226         if (!(dd->flags & TDES_FLAGS_DMA))
1227                 err = atmel_tdes_crypt_pdc_stop(dd);
1228         else
1229                 err = atmel_tdes_crypt_dma_stop(dd);
1230
1231         err = dd->err ? : err;
1232
1233         if (dd->total && !err) {
1234                 if (dd->flags & TDES_FLAGS_FAST) {
1235                         dd->in_sg = sg_next(dd->in_sg);
1236                         dd->out_sg = sg_next(dd->out_sg);
1237                         if (!dd->in_sg || !dd->out_sg)
1238                                 err = -EINVAL;
1239                 }
1240                 if (!err)
1241                         err = atmel_tdes_crypt_start(dd);
1242                 if (!err)
1243                         return; /* DMA started. Not fininishing. */
1244         }
1245
1246         atmel_tdes_finish_req(dd, err);
1247         atmel_tdes_handle_queue(dd, NULL);
1248 }
1249
1250 static irqreturn_t atmel_tdes_irq(int irq, void *dev_id)
1251 {
1252         struct atmel_tdes_dev *tdes_dd = dev_id;
1253         u32 reg;
1254
1255         reg = atmel_tdes_read(tdes_dd, TDES_ISR);
1256         if (reg & atmel_tdes_read(tdes_dd, TDES_IMR)) {
1257                 atmel_tdes_write(tdes_dd, TDES_IDR, reg);
1258                 if (TDES_FLAGS_BUSY & tdes_dd->flags)
1259                         tasklet_schedule(&tdes_dd->done_task);
1260                 else
1261                         dev_warn(tdes_dd->dev, "TDES interrupt when no active requests.\n");
1262                 return IRQ_HANDLED;
1263         }
1264
1265         return IRQ_NONE;
1266 }
1267
1268 static void atmel_tdes_unregister_algs(struct atmel_tdes_dev *dd)
1269 {
1270         int i;
1271
1272         for (i = 0; i < ARRAY_SIZE(tdes_algs); i++)
1273                 crypto_unregister_alg(&tdes_algs[i]);
1274 }
1275
1276 static int atmel_tdes_register_algs(struct atmel_tdes_dev *dd)
1277 {
1278         int err, i, j;
1279
1280         for (i = 0; i < ARRAY_SIZE(tdes_algs); i++) {
1281                 err = crypto_register_alg(&tdes_algs[i]);
1282                 if (err)
1283                         goto err_tdes_algs;
1284         }
1285
1286         return 0;
1287
1288 err_tdes_algs:
1289         for (j = 0; j < i; j++)
1290                 crypto_unregister_alg(&tdes_algs[j]);
1291
1292         return err;
1293 }
1294
1295 static void atmel_tdes_get_cap(struct atmel_tdes_dev *dd)
1296 {
1297
1298         dd->caps.has_dma = 0;
1299         dd->caps.has_cfb_3keys = 0;
1300
1301         /* keep only major version number */
1302         switch (dd->hw_version & 0xf00) {
1303         case 0x700:
1304                 dd->caps.has_dma = 1;
1305                 dd->caps.has_cfb_3keys = 1;
1306                 break;
1307         case 0x600:
1308                 break;
1309         default:
1310                 dev_warn(dd->dev,
1311                                 "Unmanaged tdes version, set minimum capabilities\n");
1312                 break;
1313         }
1314 }
1315
1316 #if defined(CONFIG_OF)
1317 static const struct of_device_id atmel_tdes_dt_ids[] = {
1318         { .compatible = "atmel,at91sam9g46-tdes" },
1319         { /* sentinel */ }
1320 };
1321 MODULE_DEVICE_TABLE(of, atmel_tdes_dt_ids);
1322
1323 static struct crypto_platform_data *atmel_tdes_of_init(struct platform_device *pdev)
1324 {
1325         struct device_node *np = pdev->dev.of_node;
1326         struct crypto_platform_data *pdata;
1327
1328         if (!np) {
1329                 dev_err(&pdev->dev, "device node not found\n");
1330                 return ERR_PTR(-EINVAL);
1331         }
1332
1333         pdata = devm_kzalloc(&pdev->dev, sizeof(*pdata), GFP_KERNEL);
1334         if (!pdata) {
1335                 dev_err(&pdev->dev, "could not allocate memory for pdata\n");
1336                 return ERR_PTR(-ENOMEM);
1337         }
1338
1339         pdata->dma_slave = devm_kzalloc(&pdev->dev,
1340                                         sizeof(*(pdata->dma_slave)),
1341                                         GFP_KERNEL);
1342         if (!pdata->dma_slave) {
1343                 dev_err(&pdev->dev, "could not allocate memory for dma_slave\n");
1344                 return ERR_PTR(-ENOMEM);
1345         }
1346
1347         return pdata;
1348 }
1349 #else /* CONFIG_OF */
1350 static inline struct crypto_platform_data *atmel_tdes_of_init(struct platform_device *pdev)
1351 {
1352         return ERR_PTR(-EINVAL);
1353 }
1354 #endif
1355
1356 static int atmel_tdes_probe(struct platform_device *pdev)
1357 {
1358         struct atmel_tdes_dev *tdes_dd;
1359         struct crypto_platform_data     *pdata;
1360         struct device *dev = &pdev->dev;
1361         struct resource *tdes_res;
1362         int err;
1363
1364         tdes_dd = devm_kmalloc(&pdev->dev, sizeof(*tdes_dd), GFP_KERNEL);
1365         if (tdes_dd == NULL) {
1366                 dev_err(dev, "unable to alloc data struct.\n");
1367                 err = -ENOMEM;
1368                 goto tdes_dd_err;
1369         }
1370
1371         tdes_dd->dev = dev;
1372
1373         platform_set_drvdata(pdev, tdes_dd);
1374
1375         INIT_LIST_HEAD(&tdes_dd->list);
1376         spin_lock_init(&tdes_dd->lock);
1377
1378         tasklet_init(&tdes_dd->done_task, atmel_tdes_done_task,
1379                                         (unsigned long)tdes_dd);
1380         tasklet_init(&tdes_dd->queue_task, atmel_tdes_queue_task,
1381                                         (unsigned long)tdes_dd);
1382
1383         crypto_init_queue(&tdes_dd->queue, ATMEL_TDES_QUEUE_LENGTH);
1384
1385         tdes_dd->irq = -1;
1386
1387         /* Get the base address */
1388         tdes_res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
1389         if (!tdes_res) {
1390                 dev_err(dev, "no MEM resource info\n");
1391                 err = -ENODEV;
1392                 goto res_err;
1393         }
1394         tdes_dd->phys_base = tdes_res->start;
1395
1396         /* Get the IRQ */
1397         tdes_dd->irq = platform_get_irq(pdev,  0);
1398         if (tdes_dd->irq < 0) {
1399                 dev_err(dev, "no IRQ resource info\n");
1400                 err = tdes_dd->irq;
1401                 goto res_err;
1402         }
1403
1404         err = devm_request_irq(&pdev->dev, tdes_dd->irq, atmel_tdes_irq,
1405                                IRQF_SHARED, "atmel-tdes", tdes_dd);
1406         if (err) {
1407                 dev_err(dev, "unable to request tdes irq.\n");
1408                 goto res_err;
1409         }
1410
1411         /* Initializing the clock */
1412         tdes_dd->iclk = devm_clk_get(&pdev->dev, "tdes_clk");
1413         if (IS_ERR(tdes_dd->iclk)) {
1414                 dev_err(dev, "clock initialization failed.\n");
1415                 err = PTR_ERR(tdes_dd->iclk);
1416                 goto res_err;
1417         }
1418
1419         tdes_dd->io_base = devm_ioremap_resource(&pdev->dev, tdes_res);
1420         if (!tdes_dd->io_base) {
1421                 dev_err(dev, "can't ioremap\n");
1422                 err = -ENOMEM;
1423                 goto res_err;
1424         }
1425
1426         atmel_tdes_hw_version_init(tdes_dd);
1427
1428         atmel_tdes_get_cap(tdes_dd);
1429
1430         err = atmel_tdes_buff_init(tdes_dd);
1431         if (err)
1432                 goto err_tdes_buff;
1433
1434         if (tdes_dd->caps.has_dma) {
1435                 pdata = pdev->dev.platform_data;
1436                 if (!pdata) {
1437                         pdata = atmel_tdes_of_init(pdev);
1438                         if (IS_ERR(pdata)) {
1439                                 dev_err(&pdev->dev, "platform data not available\n");
1440                                 err = PTR_ERR(pdata);
1441                                 goto err_pdata;
1442                         }
1443                 }
1444                 if (!pdata->dma_slave) {
1445                         err = -ENXIO;
1446                         goto err_pdata;
1447                 }
1448                 err = atmel_tdes_dma_init(tdes_dd, pdata);
1449                 if (err)
1450                         goto err_tdes_dma;
1451
1452                 dev_info(dev, "using %s, %s for DMA transfers\n",
1453                                 dma_chan_name(tdes_dd->dma_lch_in.chan),
1454                                 dma_chan_name(tdes_dd->dma_lch_out.chan));
1455         }
1456
1457         spin_lock(&atmel_tdes.lock);
1458         list_add_tail(&tdes_dd->list, &atmel_tdes.dev_list);
1459         spin_unlock(&atmel_tdes.lock);
1460
1461         err = atmel_tdes_register_algs(tdes_dd);
1462         if (err)
1463                 goto err_algs;
1464
1465         dev_info(dev, "Atmel DES/TDES\n");
1466
1467         return 0;
1468
1469 err_algs:
1470         spin_lock(&atmel_tdes.lock);
1471         list_del(&tdes_dd->list);
1472         spin_unlock(&atmel_tdes.lock);
1473         if (tdes_dd->caps.has_dma)
1474                 atmel_tdes_dma_cleanup(tdes_dd);
1475 err_tdes_dma:
1476 err_pdata:
1477         atmel_tdes_buff_cleanup(tdes_dd);
1478 err_tdes_buff:
1479 res_err:
1480         tasklet_kill(&tdes_dd->done_task);
1481         tasklet_kill(&tdes_dd->queue_task);
1482 tdes_dd_err:
1483         dev_err(dev, "initialization failed.\n");
1484
1485         return err;
1486 }
1487
1488 static int atmel_tdes_remove(struct platform_device *pdev)
1489 {
1490         static struct atmel_tdes_dev *tdes_dd;
1491
1492         tdes_dd = platform_get_drvdata(pdev);
1493         if (!tdes_dd)
1494                 return -ENODEV;
1495         spin_lock(&atmel_tdes.lock);
1496         list_del(&tdes_dd->list);
1497         spin_unlock(&atmel_tdes.lock);
1498
1499         atmel_tdes_unregister_algs(tdes_dd);
1500
1501         tasklet_kill(&tdes_dd->done_task);
1502         tasklet_kill(&tdes_dd->queue_task);
1503
1504         if (tdes_dd->caps.has_dma)
1505                 atmel_tdes_dma_cleanup(tdes_dd);
1506
1507         atmel_tdes_buff_cleanup(tdes_dd);
1508
1509         return 0;
1510 }
1511
1512 static struct platform_driver atmel_tdes_driver = {
1513         .probe          = atmel_tdes_probe,
1514         .remove         = atmel_tdes_remove,
1515         .driver         = {
1516                 .name   = "atmel_tdes",
1517                 .of_match_table = of_match_ptr(atmel_tdes_dt_ids),
1518         },
1519 };
1520
1521 module_platform_driver(atmel_tdes_driver);
1522
1523 MODULE_DESCRIPTION("Atmel DES/TDES hw acceleration support.");
1524 MODULE_LICENSE("GPL v2");
1525 MODULE_AUTHOR("Nicolas Royer - Eukréa Electromatique");