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
67struct atmel_tdes_caps {
68	bool	has_dma;
69	u32		has_cfb_3keys;
70};
71
72struct atmel_tdes_dev;
73
74struct 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
84struct atmel_tdes_reqctx {
85	unsigned long mode;
86};
87
88struct atmel_tdes_dma {
89	struct dma_chan			*chan;
90	struct dma_slave_config dma_conf;
91};
92
93struct 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
140struct atmel_tdes_drv {
141	struct list_head	dev_list;
142	spinlock_t		lock;
143};
144
145static 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
150static 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
181static inline u32 atmel_tdes_read(struct atmel_tdes_dev *dd, u32 offset)
182{
183	return readl_relaxed(dd->io_base + offset);
184}
185
186static 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
192static 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
199static 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
219static 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
236static 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
241static 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
253static 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
261static 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
320static 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
346static 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
379err_map_out:
380	dma_unmap_single(dd->dev, dd->dma_addr_in, dd->buflen,
381		DMA_TO_DEVICE);
382err_map_in:
383err_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
391static 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
401static 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
437static 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
506static 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
577static 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
588static 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
645static 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
671static 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
708static 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
720static 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
764err_dma_out:
765	dma_release_channel(dd->dma_lch_in.chan);
766err_dma_in:
767	dev_warn(dd->dev, "no DMA channel available\n");
768	return err;
769}
770
771static 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
777static 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
803static 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
829static int atmel_tdes_ecb_encrypt(struct ablkcipher_request *req)
830{
831	return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT);
832}
833
834static int atmel_tdes_ecb_decrypt(struct ablkcipher_request *req)
835{
836	return atmel_tdes_crypt(req, 0);
837}
838
839static int atmel_tdes_cbc_encrypt(struct ablkcipher_request *req)
840{
841	return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CBC);
842}
843
844static int atmel_tdes_cbc_decrypt(struct ablkcipher_request *req)
845{
846	return atmel_tdes_crypt(req, TDES_FLAGS_CBC);
847}
848static int atmel_tdes_cfb_encrypt(struct ablkcipher_request *req)
849{
850	return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_CFB);
851}
852
853static int atmel_tdes_cfb_decrypt(struct ablkcipher_request *req)
854{
855	return atmel_tdes_crypt(req, TDES_FLAGS_CFB);
856}
857
858static 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
864static int atmel_tdes_cfb8_decrypt(struct ablkcipher_request *req)
865{
866	return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB8);
867}
868
869static 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
875static int atmel_tdes_cfb16_decrypt(struct ablkcipher_request *req)
876{
877	return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB16);
878}
879
880static 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
886static int atmel_tdes_cfb32_decrypt(struct ablkcipher_request *req)
887{
888	return atmel_tdes_crypt(req, TDES_FLAGS_CFB | TDES_FLAGS_CFB32);
889}
890
891static int atmel_tdes_ofb_encrypt(struct ablkcipher_request *req)
892{
893	return atmel_tdes_crypt(req, TDES_FLAGS_ENCRYPT | TDES_FLAGS_OFB);
894}
895
896static int atmel_tdes_ofb_decrypt(struct ablkcipher_request *req)
897{
898	return atmel_tdes_crypt(req, TDES_FLAGS_OFB);
899}
900
901static 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
915static void atmel_tdes_cra_exit(struct crypto_tfm *tfm)
916{
917}
918
919static 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
1214static 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
1221static 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
1250static 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
1268static 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
1276static 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
1288err_tdes_algs:
1289	for (j = 0; j < i; j++)
1290		crypto_unregister_alg(&tdes_algs[j]);
1291
1292	return err;
1293}
1294
1295static 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)
1317static const struct of_device_id atmel_tdes_dt_ids[] = {
1318	{ .compatible = "atmel,at91sam9g46-tdes" },
1319	{ /* sentinel */ }
1320};
1321MODULE_DEVICE_TABLE(of, atmel_tdes_dt_ids);
1322
1323static 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 */
1350static inline struct crypto_platform_data *atmel_tdes_of_init(struct platform_device *pdev)
1351{
1352	return ERR_PTR(-EINVAL);
1353}
1354#endif
1355
1356static 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
1469err_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);
1475err_tdes_dma:
1476err_pdata:
1477	atmel_tdes_buff_cleanup(tdes_dd);
1478err_tdes_buff:
1479res_err:
1480	tasklet_kill(&tdes_dd->done_task);
1481	tasklet_kill(&tdes_dd->queue_task);
1482tdes_dd_err:
1483	dev_err(dev, "initialization failed.\n");
1484
1485	return err;
1486}
1487
1488static 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
1512static 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
1521module_platform_driver(atmel_tdes_driver);
1522
1523MODULE_DESCRIPTION("Atmel DES/TDES hw acceleration support.");
1524MODULE_LICENSE("GPL v2");
1525MODULE_AUTHOR("Nicolas Royer - Eukréa Electromatique");
1526