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