1/*
2 * Support for OMAP DES and Triple DES HW acceleration.
3 *
4 * Copyright (c) 2013 Texas Instruments Incorporated
5 * Author: Joel Fernandes <joelf@ti.com>
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License version 2 as published
9 * by the Free Software Foundation.
10 *
11 */
12
13#define pr_fmt(fmt) "%s: " fmt, __func__
14
15#ifdef DEBUG
16#define prn(num) printk(#num "=%d\n", num)
17#define prx(num) printk(#num "=%x\n", num)
18#else
19#define prn(num) do { } while (0)
20#define prx(num)  do { } while (0)
21#endif
22
23#include <linux/err.h>
24#include <linux/module.h>
25#include <linux/init.h>
26#include <linux/errno.h>
27#include <linux/kernel.h>
28#include <linux/platform_device.h>
29#include <linux/scatterlist.h>
30#include <linux/dma-mapping.h>
31#include <linux/dmaengine.h>
32#include <linux/omap-dma.h>
33#include <linux/pm_runtime.h>
34#include <linux/of.h>
35#include <linux/of_device.h>
36#include <linux/of_address.h>
37#include <linux/io.h>
38#include <linux/crypto.h>
39#include <linux/interrupt.h>
40#include <crypto/scatterwalk.h>
41#include <crypto/des.h>
42
43#define DST_MAXBURST			2
44
45#define DES_BLOCK_WORDS		(DES_BLOCK_SIZE >> 2)
46
47#define _calc_walked(inout) (dd->inout##_walk.offset - dd->inout##_sg->offset)
48
49#define DES_REG_KEY(dd, x)		((dd)->pdata->key_ofs - \
50						((x ^ 0x01) * 0x04))
51
52#define DES_REG_IV(dd, x)		((dd)->pdata->iv_ofs + ((x) * 0x04))
53
54#define DES_REG_CTRL(dd)		((dd)->pdata->ctrl_ofs)
55#define DES_REG_CTRL_CBC		BIT(4)
56#define DES_REG_CTRL_TDES		BIT(3)
57#define DES_REG_CTRL_DIRECTION		BIT(2)
58#define DES_REG_CTRL_INPUT_READY	BIT(1)
59#define DES_REG_CTRL_OUTPUT_READY	BIT(0)
60
61#define DES_REG_DATA_N(dd, x)		((dd)->pdata->data_ofs + ((x) * 0x04))
62
63#define DES_REG_REV(dd)			((dd)->pdata->rev_ofs)
64
65#define DES_REG_MASK(dd)		((dd)->pdata->mask_ofs)
66
67#define DES_REG_LENGTH_N(x)		(0x24 + ((x) * 0x04))
68
69#define DES_REG_IRQ_STATUS(dd)         ((dd)->pdata->irq_status_ofs)
70#define DES_REG_IRQ_ENABLE(dd)         ((dd)->pdata->irq_enable_ofs)
71#define DES_REG_IRQ_DATA_IN            BIT(1)
72#define DES_REG_IRQ_DATA_OUT           BIT(2)
73
74#define FLAGS_MODE_MASK		0x000f
75#define FLAGS_ENCRYPT		BIT(0)
76#define FLAGS_CBC		BIT(1)
77#define FLAGS_INIT		BIT(4)
78#define FLAGS_BUSY		BIT(6)
79
80struct omap_des_ctx {
81	struct omap_des_dev *dd;
82
83	int		keylen;
84	u32		key[(3 * DES_KEY_SIZE) / sizeof(u32)];
85	unsigned long	flags;
86};
87
88struct omap_des_reqctx {
89	unsigned long mode;
90};
91
92#define OMAP_DES_QUEUE_LENGTH	1
93#define OMAP_DES_CACHE_SIZE	0
94
95struct omap_des_algs_info {
96	struct crypto_alg	*algs_list;
97	unsigned int		size;
98	unsigned int		registered;
99};
100
101struct omap_des_pdata {
102	struct omap_des_algs_info	*algs_info;
103	unsigned int	algs_info_size;
104
105	void		(*trigger)(struct omap_des_dev *dd, int length);
106
107	u32		key_ofs;
108	u32		iv_ofs;
109	u32		ctrl_ofs;
110	u32		data_ofs;
111	u32		rev_ofs;
112	u32		mask_ofs;
113	u32             irq_enable_ofs;
114	u32             irq_status_ofs;
115
116	u32		dma_enable_in;
117	u32		dma_enable_out;
118	u32		dma_start;
119
120	u32		major_mask;
121	u32		major_shift;
122	u32		minor_mask;
123	u32		minor_shift;
124};
125
126struct omap_des_dev {
127	struct list_head	list;
128	unsigned long		phys_base;
129	void __iomem		*io_base;
130	struct omap_des_ctx	*ctx;
131	struct device		*dev;
132	unsigned long		flags;
133	int			err;
134
135	/* spinlock used for queues */
136	spinlock_t		lock;
137	struct crypto_queue	queue;
138
139	struct tasklet_struct	done_task;
140	struct tasklet_struct	queue_task;
141
142	struct ablkcipher_request	*req;
143	/*
144	 * total is used by PIO mode for book keeping so introduce
145	 * variable total_save as need it to calc page_order
146	 */
147	size_t                          total;
148	size_t                          total_save;
149
150	struct scatterlist		*in_sg;
151	struct scatterlist		*out_sg;
152
153	/* Buffers for copying for unaligned cases */
154	struct scatterlist		in_sgl;
155	struct scatterlist		out_sgl;
156	struct scatterlist		*orig_out;
157	int				sgs_copied;
158
159	struct scatter_walk		in_walk;
160	struct scatter_walk		out_walk;
161	int			dma_in;
162	struct dma_chan		*dma_lch_in;
163	int			dma_out;
164	struct dma_chan		*dma_lch_out;
165	int			in_sg_len;
166	int			out_sg_len;
167	int			pio_only;
168	const struct omap_des_pdata	*pdata;
169};
170
171/* keep registered devices data here */
172static LIST_HEAD(dev_list);
173static DEFINE_SPINLOCK(list_lock);
174
175#ifdef DEBUG
176#define omap_des_read(dd, offset)                               \
177	({                                                              \
178	 int _read_ret;                                          \
179	 _read_ret = __raw_readl(dd->io_base + offset);          \
180	 pr_err("omap_des_read(" #offset "=%#x)= %#x\n",       \
181		 offset, _read_ret);                            \
182	 _read_ret;                                              \
183	 })
184#else
185static inline u32 omap_des_read(struct omap_des_dev *dd, u32 offset)
186{
187	return __raw_readl(dd->io_base + offset);
188}
189#endif
190
191#ifdef DEBUG
192#define omap_des_write(dd, offset, value)                               \
193	do {                                                            \
194		pr_err("omap_des_write(" #offset "=%#x) value=%#x\n", \
195				offset, value);                                \
196		__raw_writel(value, dd->io_base + offset);              \
197	} while (0)
198#else
199static inline void omap_des_write(struct omap_des_dev *dd, u32 offset,
200		u32 value)
201{
202	__raw_writel(value, dd->io_base + offset);
203}
204#endif
205
206static inline void omap_des_write_mask(struct omap_des_dev *dd, u32 offset,
207					u32 value, u32 mask)
208{
209	u32 val;
210
211	val = omap_des_read(dd, offset);
212	val &= ~mask;
213	val |= value;
214	omap_des_write(dd, offset, val);
215}
216
217static void omap_des_write_n(struct omap_des_dev *dd, u32 offset,
218					u32 *value, int count)
219{
220	for (; count--; value++, offset += 4)
221		omap_des_write(dd, offset, *value);
222}
223
224static int omap_des_hw_init(struct omap_des_dev *dd)
225{
226	int err;
227
228	/*
229	 * clocks are enabled when request starts and disabled when finished.
230	 * It may be long delays between requests.
231	 * Device might go to off mode to save power.
232	 */
233	err = pm_runtime_get_sync(dd->dev);
234	if (err < 0) {
235		pm_runtime_put_noidle(dd->dev);
236		dev_err(dd->dev, "%s: failed to get_sync(%d)\n", __func__, err);
237		return err;
238	}
239
240	if (!(dd->flags & FLAGS_INIT)) {
241		dd->flags |= FLAGS_INIT;
242		dd->err = 0;
243	}
244
245	return 0;
246}
247
248static int omap_des_write_ctrl(struct omap_des_dev *dd)
249{
250	unsigned int key32;
251	int i, err;
252	u32 val = 0, mask = 0;
253
254	err = omap_des_hw_init(dd);
255	if (err)
256		return err;
257
258	key32 = dd->ctx->keylen / sizeof(u32);
259
260	/* it seems a key should always be set even if it has not changed */
261	for (i = 0; i < key32; i++) {
262		omap_des_write(dd, DES_REG_KEY(dd, i),
263			       __le32_to_cpu(dd->ctx->key[i]));
264	}
265
266	if ((dd->flags & FLAGS_CBC) && dd->req->info)
267		omap_des_write_n(dd, DES_REG_IV(dd, 0), dd->req->info, 2);
268
269	if (dd->flags & FLAGS_CBC)
270		val |= DES_REG_CTRL_CBC;
271	if (dd->flags & FLAGS_ENCRYPT)
272		val |= DES_REG_CTRL_DIRECTION;
273	if (key32 == 6)
274		val |= DES_REG_CTRL_TDES;
275
276	mask |= DES_REG_CTRL_CBC | DES_REG_CTRL_DIRECTION | DES_REG_CTRL_TDES;
277
278	omap_des_write_mask(dd, DES_REG_CTRL(dd), val, mask);
279
280	return 0;
281}
282
283static void omap_des_dma_trigger_omap4(struct omap_des_dev *dd, int length)
284{
285	u32 mask, val;
286
287	omap_des_write(dd, DES_REG_LENGTH_N(0), length);
288
289	val = dd->pdata->dma_start;
290
291	if (dd->dma_lch_out != NULL)
292		val |= dd->pdata->dma_enable_out;
293	if (dd->dma_lch_in != NULL)
294		val |= dd->pdata->dma_enable_in;
295
296	mask = dd->pdata->dma_enable_out | dd->pdata->dma_enable_in |
297	       dd->pdata->dma_start;
298
299	omap_des_write_mask(dd, DES_REG_MASK(dd), val, mask);
300}
301
302static void omap_des_dma_stop(struct omap_des_dev *dd)
303{
304	u32 mask;
305
306	mask = dd->pdata->dma_enable_out | dd->pdata->dma_enable_in |
307	       dd->pdata->dma_start;
308
309	omap_des_write_mask(dd, DES_REG_MASK(dd), 0, mask);
310}
311
312static struct omap_des_dev *omap_des_find_dev(struct omap_des_ctx *ctx)
313{
314	struct omap_des_dev *dd = NULL, *tmp;
315
316	spin_lock_bh(&list_lock);
317	if (!ctx->dd) {
318		list_for_each_entry(tmp, &dev_list, list) {
319			/* FIXME: take fist available des core */
320			dd = tmp;
321			break;
322		}
323		ctx->dd = dd;
324	} else {
325		/* already found before */
326		dd = ctx->dd;
327	}
328	spin_unlock_bh(&list_lock);
329
330	return dd;
331}
332
333static void omap_des_dma_out_callback(void *data)
334{
335	struct omap_des_dev *dd = data;
336
337	/* dma_lch_out - completed */
338	tasklet_schedule(&dd->done_task);
339}
340
341static int omap_des_dma_init(struct omap_des_dev *dd)
342{
343	int err = -ENOMEM;
344	dma_cap_mask_t mask;
345
346	dd->dma_lch_out = NULL;
347	dd->dma_lch_in = NULL;
348
349	dma_cap_zero(mask);
350	dma_cap_set(DMA_SLAVE, mask);
351
352	dd->dma_lch_in = dma_request_slave_channel_compat(mask,
353							  omap_dma_filter_fn,
354							  &dd->dma_in,
355							  dd->dev, "rx");
356	if (!dd->dma_lch_in) {
357		dev_err(dd->dev, "Unable to request in DMA channel\n");
358		goto err_dma_in;
359	}
360
361	dd->dma_lch_out = dma_request_slave_channel_compat(mask,
362							   omap_dma_filter_fn,
363							   &dd->dma_out,
364							   dd->dev, "tx");
365	if (!dd->dma_lch_out) {
366		dev_err(dd->dev, "Unable to request out DMA channel\n");
367		goto err_dma_out;
368	}
369
370	return 0;
371
372err_dma_out:
373	dma_release_channel(dd->dma_lch_in);
374err_dma_in:
375	if (err)
376		pr_err("error: %d\n", err);
377	return err;
378}
379
380static void omap_des_dma_cleanup(struct omap_des_dev *dd)
381{
382	dma_release_channel(dd->dma_lch_out);
383	dma_release_channel(dd->dma_lch_in);
384}
385
386static void sg_copy_buf(void *buf, struct scatterlist *sg,
387			      unsigned int start, unsigned int nbytes, int out)
388{
389	struct scatter_walk walk;
390
391	if (!nbytes)
392		return;
393
394	scatterwalk_start(&walk, sg);
395	scatterwalk_advance(&walk, start);
396	scatterwalk_copychunks(buf, &walk, nbytes, out);
397	scatterwalk_done(&walk, out, 0);
398}
399
400static int omap_des_crypt_dma(struct crypto_tfm *tfm,
401		struct scatterlist *in_sg, struct scatterlist *out_sg,
402		int in_sg_len, int out_sg_len)
403{
404	struct omap_des_ctx *ctx = crypto_tfm_ctx(tfm);
405	struct omap_des_dev *dd = ctx->dd;
406	struct dma_async_tx_descriptor *tx_in, *tx_out;
407	struct dma_slave_config cfg;
408	int ret;
409
410	if (dd->pio_only) {
411		scatterwalk_start(&dd->in_walk, dd->in_sg);
412		scatterwalk_start(&dd->out_walk, dd->out_sg);
413
414		/* Enable DATAIN interrupt and let it take
415		   care of the rest */
416		omap_des_write(dd, DES_REG_IRQ_ENABLE(dd), 0x2);
417		return 0;
418	}
419
420	dma_sync_sg_for_device(dd->dev, dd->in_sg, in_sg_len, DMA_TO_DEVICE);
421
422	memset(&cfg, 0, sizeof(cfg));
423
424	cfg.src_addr = dd->phys_base + DES_REG_DATA_N(dd, 0);
425	cfg.dst_addr = dd->phys_base + DES_REG_DATA_N(dd, 0);
426	cfg.src_addr_width = DMA_SLAVE_BUSWIDTH_4_BYTES;
427	cfg.dst_addr_width = DMA_SLAVE_BUSWIDTH_4_BYTES;
428	cfg.src_maxburst = DST_MAXBURST;
429	cfg.dst_maxburst = DST_MAXBURST;
430
431	/* IN */
432	ret = dmaengine_slave_config(dd->dma_lch_in, &cfg);
433	if (ret) {
434		dev_err(dd->dev, "can't configure IN dmaengine slave: %d\n",
435			ret);
436		return ret;
437	}
438
439	tx_in = dmaengine_prep_slave_sg(dd->dma_lch_in, in_sg, in_sg_len,
440					DMA_MEM_TO_DEV,
441					DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
442	if (!tx_in) {
443		dev_err(dd->dev, "IN prep_slave_sg() failed\n");
444		return -EINVAL;
445	}
446
447	/* No callback necessary */
448	tx_in->callback_param = dd;
449
450	/* OUT */
451	ret = dmaengine_slave_config(dd->dma_lch_out, &cfg);
452	if (ret) {
453		dev_err(dd->dev, "can't configure OUT dmaengine slave: %d\n",
454			ret);
455		return ret;
456	}
457
458	tx_out = dmaengine_prep_slave_sg(dd->dma_lch_out, out_sg, out_sg_len,
459					DMA_DEV_TO_MEM,
460					DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
461	if (!tx_out) {
462		dev_err(dd->dev, "OUT prep_slave_sg() failed\n");
463		return -EINVAL;
464	}
465
466	tx_out->callback = omap_des_dma_out_callback;
467	tx_out->callback_param = dd;
468
469	dmaengine_submit(tx_in);
470	dmaengine_submit(tx_out);
471
472	dma_async_issue_pending(dd->dma_lch_in);
473	dma_async_issue_pending(dd->dma_lch_out);
474
475	/* start DMA */
476	dd->pdata->trigger(dd, dd->total);
477
478	return 0;
479}
480
481static int omap_des_crypt_dma_start(struct omap_des_dev *dd)
482{
483	struct crypto_tfm *tfm = crypto_ablkcipher_tfm(
484					crypto_ablkcipher_reqtfm(dd->req));
485	int err;
486
487	pr_debug("total: %d\n", dd->total);
488
489	if (!dd->pio_only) {
490		err = dma_map_sg(dd->dev, dd->in_sg, dd->in_sg_len,
491				 DMA_TO_DEVICE);
492		if (!err) {
493			dev_err(dd->dev, "dma_map_sg() error\n");
494			return -EINVAL;
495		}
496
497		err = dma_map_sg(dd->dev, dd->out_sg, dd->out_sg_len,
498				 DMA_FROM_DEVICE);
499		if (!err) {
500			dev_err(dd->dev, "dma_map_sg() error\n");
501			return -EINVAL;
502		}
503	}
504
505	err = omap_des_crypt_dma(tfm, dd->in_sg, dd->out_sg, dd->in_sg_len,
506				 dd->out_sg_len);
507	if (err && !dd->pio_only) {
508		dma_unmap_sg(dd->dev, dd->in_sg, dd->in_sg_len, DMA_TO_DEVICE);
509		dma_unmap_sg(dd->dev, dd->out_sg, dd->out_sg_len,
510			     DMA_FROM_DEVICE);
511	}
512
513	return err;
514}
515
516static void omap_des_finish_req(struct omap_des_dev *dd, int err)
517{
518	struct ablkcipher_request *req = dd->req;
519
520	pr_debug("err: %d\n", err);
521
522	pm_runtime_put(dd->dev);
523	dd->flags &= ~FLAGS_BUSY;
524
525	req->base.complete(&req->base, err);
526}
527
528static int omap_des_crypt_dma_stop(struct omap_des_dev *dd)
529{
530	int err = 0;
531
532	pr_debug("total: %d\n", dd->total);
533
534	omap_des_dma_stop(dd);
535
536	dmaengine_terminate_all(dd->dma_lch_in);
537	dmaengine_terminate_all(dd->dma_lch_out);
538
539	return err;
540}
541
542static int omap_des_copy_needed(struct scatterlist *sg)
543{
544	while (sg) {
545		if (!IS_ALIGNED(sg->offset, 4))
546			return -1;
547		if (!IS_ALIGNED(sg->length, DES_BLOCK_SIZE))
548			return -1;
549		sg = sg_next(sg);
550	}
551	return 0;
552}
553
554static int omap_des_copy_sgs(struct omap_des_dev *dd)
555{
556	void *buf_in, *buf_out;
557	int pages;
558
559	pages = dd->total >> PAGE_SHIFT;
560
561	if (dd->total & (PAGE_SIZE-1))
562		pages++;
563
564	BUG_ON(!pages);
565
566	buf_in = (void *)__get_free_pages(GFP_ATOMIC, pages);
567	buf_out = (void *)__get_free_pages(GFP_ATOMIC, pages);
568
569	if (!buf_in || !buf_out) {
570		pr_err("Couldn't allocated pages for unaligned cases.\n");
571		return -1;
572	}
573
574	dd->orig_out = dd->out_sg;
575
576	sg_copy_buf(buf_in, dd->in_sg, 0, dd->total, 0);
577
578	sg_init_table(&dd->in_sgl, 1);
579	sg_set_buf(&dd->in_sgl, buf_in, dd->total);
580	dd->in_sg = &dd->in_sgl;
581
582	sg_init_table(&dd->out_sgl, 1);
583	sg_set_buf(&dd->out_sgl, buf_out, dd->total);
584	dd->out_sg = &dd->out_sgl;
585
586	return 0;
587}
588
589static int omap_des_handle_queue(struct omap_des_dev *dd,
590			       struct ablkcipher_request *req)
591{
592	struct crypto_async_request *async_req, *backlog;
593	struct omap_des_ctx *ctx;
594	struct omap_des_reqctx *rctx;
595	unsigned long flags;
596	int err, ret = 0;
597
598	spin_lock_irqsave(&dd->lock, flags);
599	if (req)
600		ret = ablkcipher_enqueue_request(&dd->queue, req);
601	if (dd->flags & FLAGS_BUSY) {
602		spin_unlock_irqrestore(&dd->lock, flags);
603		return ret;
604	}
605	backlog = crypto_get_backlog(&dd->queue);
606	async_req = crypto_dequeue_request(&dd->queue);
607	if (async_req)
608		dd->flags |= FLAGS_BUSY;
609	spin_unlock_irqrestore(&dd->lock, flags);
610
611	if (!async_req)
612		return ret;
613
614	if (backlog)
615		backlog->complete(backlog, -EINPROGRESS);
616
617	req = ablkcipher_request_cast(async_req);
618
619	/* assign new request to device */
620	dd->req = req;
621	dd->total = req->nbytes;
622	dd->total_save = req->nbytes;
623	dd->in_sg = req->src;
624	dd->out_sg = req->dst;
625
626	if (omap_des_copy_needed(dd->in_sg) ||
627	    omap_des_copy_needed(dd->out_sg)) {
628		if (omap_des_copy_sgs(dd))
629			pr_err("Failed to copy SGs for unaligned cases\n");
630		dd->sgs_copied = 1;
631	} else {
632		dd->sgs_copied = 0;
633	}
634
635	dd->in_sg_len = scatterwalk_bytes_sglen(dd->in_sg, dd->total);
636	dd->out_sg_len = scatterwalk_bytes_sglen(dd->out_sg, dd->total);
637	BUG_ON(dd->in_sg_len < 0 || dd->out_sg_len < 0);
638
639	rctx = ablkcipher_request_ctx(req);
640	ctx = crypto_ablkcipher_ctx(crypto_ablkcipher_reqtfm(req));
641	rctx->mode &= FLAGS_MODE_MASK;
642	dd->flags = (dd->flags & ~FLAGS_MODE_MASK) | rctx->mode;
643
644	dd->ctx = ctx;
645	ctx->dd = dd;
646
647	err = omap_des_write_ctrl(dd);
648	if (!err)
649		err = omap_des_crypt_dma_start(dd);
650	if (err) {
651		/* des_task will not finish it, so do it here */
652		omap_des_finish_req(dd, err);
653		tasklet_schedule(&dd->queue_task);
654	}
655
656	return ret; /* return ret, which is enqueue return value */
657}
658
659static void omap_des_done_task(unsigned long data)
660{
661	struct omap_des_dev *dd = (struct omap_des_dev *)data;
662	void *buf_in, *buf_out;
663	int pages;
664
665	pr_debug("enter done_task\n");
666
667	if (!dd->pio_only) {
668		dma_sync_sg_for_device(dd->dev, dd->out_sg, dd->out_sg_len,
669				       DMA_FROM_DEVICE);
670		dma_unmap_sg(dd->dev, dd->in_sg, dd->in_sg_len, DMA_TO_DEVICE);
671		dma_unmap_sg(dd->dev, dd->out_sg, dd->out_sg_len,
672			     DMA_FROM_DEVICE);
673		omap_des_crypt_dma_stop(dd);
674	}
675
676	if (dd->sgs_copied) {
677		buf_in = sg_virt(&dd->in_sgl);
678		buf_out = sg_virt(&dd->out_sgl);
679
680		sg_copy_buf(buf_out, dd->orig_out, 0, dd->total_save, 1);
681
682		pages = get_order(dd->total_save);
683		free_pages((unsigned long)buf_in, pages);
684		free_pages((unsigned long)buf_out, pages);
685	}
686
687	omap_des_finish_req(dd, 0);
688	omap_des_handle_queue(dd, NULL);
689
690	pr_debug("exit\n");
691}
692
693static void omap_des_queue_task(unsigned long data)
694{
695	struct omap_des_dev *dd = (struct omap_des_dev *)data;
696
697	omap_des_handle_queue(dd, NULL);
698}
699
700static int omap_des_crypt(struct ablkcipher_request *req, unsigned long mode)
701{
702	struct omap_des_ctx *ctx = crypto_ablkcipher_ctx(
703			crypto_ablkcipher_reqtfm(req));
704	struct omap_des_reqctx *rctx = ablkcipher_request_ctx(req);
705	struct omap_des_dev *dd;
706
707	pr_debug("nbytes: %d, enc: %d, cbc: %d\n", req->nbytes,
708		 !!(mode & FLAGS_ENCRYPT),
709		 !!(mode & FLAGS_CBC));
710
711	if (!IS_ALIGNED(req->nbytes, DES_BLOCK_SIZE)) {
712		pr_err("request size is not exact amount of DES blocks\n");
713		return -EINVAL;
714	}
715
716	dd = omap_des_find_dev(ctx);
717	if (!dd)
718		return -ENODEV;
719
720	rctx->mode = mode;
721
722	return omap_des_handle_queue(dd, req);
723}
724
725/* ********************** ALG API ************************************ */
726
727static int omap_des_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
728			   unsigned int keylen)
729{
730	struct omap_des_ctx *ctx = crypto_ablkcipher_ctx(tfm);
731
732	if (keylen != DES_KEY_SIZE && keylen != (3*DES_KEY_SIZE))
733		return -EINVAL;
734
735	pr_debug("enter, keylen: %d\n", keylen);
736
737	memcpy(ctx->key, key, keylen);
738	ctx->keylen = keylen;
739
740	return 0;
741}
742
743static int omap_des_ecb_encrypt(struct ablkcipher_request *req)
744{
745	return omap_des_crypt(req, FLAGS_ENCRYPT);
746}
747
748static int omap_des_ecb_decrypt(struct ablkcipher_request *req)
749{
750	return omap_des_crypt(req, 0);
751}
752
753static int omap_des_cbc_encrypt(struct ablkcipher_request *req)
754{
755	return omap_des_crypt(req, FLAGS_ENCRYPT | FLAGS_CBC);
756}
757
758static int omap_des_cbc_decrypt(struct ablkcipher_request *req)
759{
760	return omap_des_crypt(req, FLAGS_CBC);
761}
762
763static int omap_des_cra_init(struct crypto_tfm *tfm)
764{
765	pr_debug("enter\n");
766
767	tfm->crt_ablkcipher.reqsize = sizeof(struct omap_des_reqctx);
768
769	return 0;
770}
771
772static void omap_des_cra_exit(struct crypto_tfm *tfm)
773{
774	pr_debug("enter\n");
775}
776
777/* ********************** ALGS ************************************ */
778
779static struct crypto_alg algs_ecb_cbc[] = {
780{
781	.cra_name		= "ecb(des)",
782	.cra_driver_name	= "ecb-des-omap",
783	.cra_priority		= 100,
784	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER |
785				  CRYPTO_ALG_KERN_DRIVER_ONLY |
786				  CRYPTO_ALG_ASYNC,
787	.cra_blocksize		= DES_BLOCK_SIZE,
788	.cra_ctxsize		= sizeof(struct omap_des_ctx),
789	.cra_alignmask		= 0,
790	.cra_type		= &crypto_ablkcipher_type,
791	.cra_module		= THIS_MODULE,
792	.cra_init		= omap_des_cra_init,
793	.cra_exit		= omap_des_cra_exit,
794	.cra_u.ablkcipher = {
795		.min_keysize	= DES_KEY_SIZE,
796		.max_keysize	= DES_KEY_SIZE,
797		.setkey		= omap_des_setkey,
798		.encrypt	= omap_des_ecb_encrypt,
799		.decrypt	= omap_des_ecb_decrypt,
800	}
801},
802{
803	.cra_name		= "cbc(des)",
804	.cra_driver_name	= "cbc-des-omap",
805	.cra_priority		= 100,
806	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER |
807				  CRYPTO_ALG_KERN_DRIVER_ONLY |
808				  CRYPTO_ALG_ASYNC,
809	.cra_blocksize		= DES_BLOCK_SIZE,
810	.cra_ctxsize		= sizeof(struct omap_des_ctx),
811	.cra_alignmask		= 0,
812	.cra_type		= &crypto_ablkcipher_type,
813	.cra_module		= THIS_MODULE,
814	.cra_init		= omap_des_cra_init,
815	.cra_exit		= omap_des_cra_exit,
816	.cra_u.ablkcipher = {
817		.min_keysize	= DES_KEY_SIZE,
818		.max_keysize	= DES_KEY_SIZE,
819		.ivsize		= DES_BLOCK_SIZE,
820		.setkey		= omap_des_setkey,
821		.encrypt	= omap_des_cbc_encrypt,
822		.decrypt	= omap_des_cbc_decrypt,
823	}
824},
825{
826	.cra_name		= "ecb(des3_ede)",
827	.cra_driver_name	= "ecb-des3-omap",
828	.cra_priority		= 100,
829	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER |
830				  CRYPTO_ALG_KERN_DRIVER_ONLY |
831				  CRYPTO_ALG_ASYNC,
832	.cra_blocksize		= DES_BLOCK_SIZE,
833	.cra_ctxsize		= sizeof(struct omap_des_ctx),
834	.cra_alignmask		= 0,
835	.cra_type		= &crypto_ablkcipher_type,
836	.cra_module		= THIS_MODULE,
837	.cra_init		= omap_des_cra_init,
838	.cra_exit		= omap_des_cra_exit,
839	.cra_u.ablkcipher = {
840		.min_keysize	= 3*DES_KEY_SIZE,
841		.max_keysize	= 3*DES_KEY_SIZE,
842		.setkey		= omap_des_setkey,
843		.encrypt	= omap_des_ecb_encrypt,
844		.decrypt	= omap_des_ecb_decrypt,
845	}
846},
847{
848	.cra_name		= "cbc(des3_ede)",
849	.cra_driver_name	= "cbc-des3-omap",
850	.cra_priority		= 100,
851	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER |
852				  CRYPTO_ALG_KERN_DRIVER_ONLY |
853				  CRYPTO_ALG_ASYNC,
854	.cra_blocksize		= DES_BLOCK_SIZE,
855	.cra_ctxsize		= sizeof(struct omap_des_ctx),
856	.cra_alignmask		= 0,
857	.cra_type		= &crypto_ablkcipher_type,
858	.cra_module		= THIS_MODULE,
859	.cra_init		= omap_des_cra_init,
860	.cra_exit		= omap_des_cra_exit,
861	.cra_u.ablkcipher = {
862		.min_keysize	= 3*DES_KEY_SIZE,
863		.max_keysize	= 3*DES_KEY_SIZE,
864		.ivsize		= DES_BLOCK_SIZE,
865		.setkey		= omap_des_setkey,
866		.encrypt	= omap_des_cbc_encrypt,
867		.decrypt	= omap_des_cbc_decrypt,
868	}
869}
870};
871
872static struct omap_des_algs_info omap_des_algs_info_ecb_cbc[] = {
873	{
874		.algs_list	= algs_ecb_cbc,
875		.size		= ARRAY_SIZE(algs_ecb_cbc),
876	},
877};
878
879#ifdef CONFIG_OF
880static const struct omap_des_pdata omap_des_pdata_omap4 = {
881	.algs_info	= omap_des_algs_info_ecb_cbc,
882	.algs_info_size	= ARRAY_SIZE(omap_des_algs_info_ecb_cbc),
883	.trigger	= omap_des_dma_trigger_omap4,
884	.key_ofs	= 0x14,
885	.iv_ofs		= 0x18,
886	.ctrl_ofs	= 0x20,
887	.data_ofs	= 0x28,
888	.rev_ofs	= 0x30,
889	.mask_ofs	= 0x34,
890	.irq_status_ofs = 0x3c,
891	.irq_enable_ofs = 0x40,
892	.dma_enable_in	= BIT(5),
893	.dma_enable_out	= BIT(6),
894	.major_mask	= 0x0700,
895	.major_shift	= 8,
896	.minor_mask	= 0x003f,
897	.minor_shift	= 0,
898};
899
900static irqreturn_t omap_des_irq(int irq, void *dev_id)
901{
902	struct omap_des_dev *dd = dev_id;
903	u32 status, i;
904	u32 *src, *dst;
905
906	status = omap_des_read(dd, DES_REG_IRQ_STATUS(dd));
907	if (status & DES_REG_IRQ_DATA_IN) {
908		omap_des_write(dd, DES_REG_IRQ_ENABLE(dd), 0x0);
909
910		BUG_ON(!dd->in_sg);
911
912		BUG_ON(_calc_walked(in) > dd->in_sg->length);
913
914		src = sg_virt(dd->in_sg) + _calc_walked(in);
915
916		for (i = 0; i < DES_BLOCK_WORDS; i++) {
917			omap_des_write(dd, DES_REG_DATA_N(dd, i), *src);
918
919			scatterwalk_advance(&dd->in_walk, 4);
920			if (dd->in_sg->length == _calc_walked(in)) {
921				dd->in_sg = sg_next(dd->in_sg);
922				if (dd->in_sg) {
923					scatterwalk_start(&dd->in_walk,
924							  dd->in_sg);
925					src = sg_virt(dd->in_sg) +
926					      _calc_walked(in);
927				}
928			} else {
929				src++;
930			}
931		}
932
933		/* Clear IRQ status */
934		status &= ~DES_REG_IRQ_DATA_IN;
935		omap_des_write(dd, DES_REG_IRQ_STATUS(dd), status);
936
937		/* Enable DATA_OUT interrupt */
938		omap_des_write(dd, DES_REG_IRQ_ENABLE(dd), 0x4);
939
940	} else if (status & DES_REG_IRQ_DATA_OUT) {
941		omap_des_write(dd, DES_REG_IRQ_ENABLE(dd), 0x0);
942
943		BUG_ON(!dd->out_sg);
944
945		BUG_ON(_calc_walked(out) > dd->out_sg->length);
946
947		dst = sg_virt(dd->out_sg) + _calc_walked(out);
948
949		for (i = 0; i < DES_BLOCK_WORDS; i++) {
950			*dst = omap_des_read(dd, DES_REG_DATA_N(dd, i));
951			scatterwalk_advance(&dd->out_walk, 4);
952			if (dd->out_sg->length == _calc_walked(out)) {
953				dd->out_sg = sg_next(dd->out_sg);
954				if (dd->out_sg) {
955					scatterwalk_start(&dd->out_walk,
956							  dd->out_sg);
957					dst = sg_virt(dd->out_sg) +
958					      _calc_walked(out);
959				}
960			} else {
961				dst++;
962			}
963		}
964
965		BUG_ON(dd->total < DES_BLOCK_SIZE);
966
967		dd->total -= DES_BLOCK_SIZE;
968
969		/* Clear IRQ status */
970		status &= ~DES_REG_IRQ_DATA_OUT;
971		omap_des_write(dd, DES_REG_IRQ_STATUS(dd), status);
972
973		if (!dd->total)
974			/* All bytes read! */
975			tasklet_schedule(&dd->done_task);
976		else
977			/* Enable DATA_IN interrupt for next block */
978			omap_des_write(dd, DES_REG_IRQ_ENABLE(dd), 0x2);
979	}
980
981	return IRQ_HANDLED;
982}
983
984static const struct of_device_id omap_des_of_match[] = {
985	{
986		.compatible	= "ti,omap4-des",
987		.data		= &omap_des_pdata_omap4,
988	},
989	{},
990};
991MODULE_DEVICE_TABLE(of, omap_des_of_match);
992
993static int omap_des_get_of(struct omap_des_dev *dd,
994		struct platform_device *pdev)
995{
996	const struct of_device_id *match;
997
998	match = of_match_device(of_match_ptr(omap_des_of_match), &pdev->dev);
999	if (!match) {
1000		dev_err(&pdev->dev, "no compatible OF match\n");
1001		return -EINVAL;
1002	}
1003
1004	dd->dma_out = -1; /* Dummy value that's unused */
1005	dd->dma_in = -1; /* Dummy value that's unused */
1006	dd->pdata = match->data;
1007
1008	return 0;
1009}
1010#else
1011static int omap_des_get_of(struct omap_des_dev *dd,
1012		struct device *dev)
1013{
1014	return -EINVAL;
1015}
1016#endif
1017
1018static int omap_des_get_pdev(struct omap_des_dev *dd,
1019		struct platform_device *pdev)
1020{
1021	struct device *dev = &pdev->dev;
1022	struct resource *r;
1023	int err = 0;
1024
1025	/* Get the DMA out channel */
1026	r = platform_get_resource(pdev, IORESOURCE_DMA, 0);
1027	if (!r) {
1028		dev_err(dev, "no DMA out resource info\n");
1029		err = -ENODEV;
1030		goto err;
1031	}
1032	dd->dma_out = r->start;
1033
1034	/* Get the DMA in channel */
1035	r = platform_get_resource(pdev, IORESOURCE_DMA, 1);
1036	if (!r) {
1037		dev_err(dev, "no DMA in resource info\n");
1038		err = -ENODEV;
1039		goto err;
1040	}
1041	dd->dma_in = r->start;
1042
1043	/* non-DT devices get pdata from pdev */
1044	dd->pdata = pdev->dev.platform_data;
1045
1046err:
1047	return err;
1048}
1049
1050static int omap_des_probe(struct platform_device *pdev)
1051{
1052	struct device *dev = &pdev->dev;
1053	struct omap_des_dev *dd;
1054	struct crypto_alg *algp;
1055	struct resource *res;
1056	int err = -ENOMEM, i, j, irq = -1;
1057	u32 reg;
1058
1059	dd = devm_kzalloc(dev, sizeof(struct omap_des_dev), GFP_KERNEL);
1060	if (dd == NULL) {
1061		dev_err(dev, "unable to alloc data struct.\n");
1062		goto err_data;
1063	}
1064	dd->dev = dev;
1065	platform_set_drvdata(pdev, dd);
1066
1067	spin_lock_init(&dd->lock);
1068	crypto_init_queue(&dd->queue, OMAP_DES_QUEUE_LENGTH);
1069
1070	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
1071	if (!res) {
1072		dev_err(dev, "no MEM resource info\n");
1073		goto err_res;
1074	}
1075
1076	err = (dev->of_node) ? omap_des_get_of(dd, pdev) :
1077			       omap_des_get_pdev(dd, pdev);
1078	if (err)
1079		goto err_res;
1080
1081	dd->io_base = devm_ioremap_resource(dev, res);
1082	if (IS_ERR(dd->io_base)) {
1083		err = PTR_ERR(dd->io_base);
1084		goto err_res;
1085	}
1086	dd->phys_base = res->start;
1087
1088	pm_runtime_enable(dev);
1089	err = pm_runtime_get_sync(dev);
1090	if (err < 0) {
1091		pm_runtime_put_noidle(dev);
1092		dev_err(dd->dev, "%s: failed to get_sync(%d)\n", __func__, err);
1093		goto err_get;
1094	}
1095
1096	omap_des_dma_stop(dd);
1097
1098	reg = omap_des_read(dd, DES_REG_REV(dd));
1099
1100	pm_runtime_put_sync(dev);
1101
1102	dev_info(dev, "OMAP DES hw accel rev: %u.%u\n",
1103		 (reg & dd->pdata->major_mask) >> dd->pdata->major_shift,
1104		 (reg & dd->pdata->minor_mask) >> dd->pdata->minor_shift);
1105
1106	tasklet_init(&dd->done_task, omap_des_done_task, (unsigned long)dd);
1107	tasklet_init(&dd->queue_task, omap_des_queue_task, (unsigned long)dd);
1108
1109	err = omap_des_dma_init(dd);
1110	if (err && DES_REG_IRQ_STATUS(dd) && DES_REG_IRQ_ENABLE(dd)) {
1111		dd->pio_only = 1;
1112
1113		irq = platform_get_irq(pdev, 0);
1114		if (irq < 0) {
1115			dev_err(dev, "can't get IRQ resource\n");
1116			goto err_irq;
1117		}
1118
1119		err = devm_request_irq(dev, irq, omap_des_irq, 0,
1120				dev_name(dev), dd);
1121		if (err) {
1122			dev_err(dev, "Unable to grab omap-des IRQ\n");
1123			goto err_irq;
1124		}
1125	}
1126
1127
1128	INIT_LIST_HEAD(&dd->list);
1129	spin_lock(&list_lock);
1130	list_add_tail(&dd->list, &dev_list);
1131	spin_unlock(&list_lock);
1132
1133	for (i = 0; i < dd->pdata->algs_info_size; i++) {
1134		for (j = 0; j < dd->pdata->algs_info[i].size; j++) {
1135			algp = &dd->pdata->algs_info[i].algs_list[j];
1136
1137			pr_debug("reg alg: %s\n", algp->cra_name);
1138			INIT_LIST_HEAD(&algp->cra_list);
1139
1140			err = crypto_register_alg(algp);
1141			if (err)
1142				goto err_algs;
1143
1144			dd->pdata->algs_info[i].registered++;
1145		}
1146	}
1147
1148	return 0;
1149err_algs:
1150	for (i = dd->pdata->algs_info_size - 1; i >= 0; i--)
1151		for (j = dd->pdata->algs_info[i].registered - 1; j >= 0; j--)
1152			crypto_unregister_alg(
1153					&dd->pdata->algs_info[i].algs_list[j]);
1154	if (!dd->pio_only)
1155		omap_des_dma_cleanup(dd);
1156err_irq:
1157	tasklet_kill(&dd->done_task);
1158	tasklet_kill(&dd->queue_task);
1159err_get:
1160	pm_runtime_disable(dev);
1161err_res:
1162	dd = NULL;
1163err_data:
1164	dev_err(dev, "initialization failed.\n");
1165	return err;
1166}
1167
1168static int omap_des_remove(struct platform_device *pdev)
1169{
1170	struct omap_des_dev *dd = platform_get_drvdata(pdev);
1171	int i, j;
1172
1173	if (!dd)
1174		return -ENODEV;
1175
1176	spin_lock(&list_lock);
1177	list_del(&dd->list);
1178	spin_unlock(&list_lock);
1179
1180	for (i = dd->pdata->algs_info_size - 1; i >= 0; i--)
1181		for (j = dd->pdata->algs_info[i].registered - 1; j >= 0; j--)
1182			crypto_unregister_alg(
1183					&dd->pdata->algs_info[i].algs_list[j]);
1184
1185	tasklet_kill(&dd->done_task);
1186	tasklet_kill(&dd->queue_task);
1187	omap_des_dma_cleanup(dd);
1188	pm_runtime_disable(dd->dev);
1189	dd = NULL;
1190
1191	return 0;
1192}
1193
1194#ifdef CONFIG_PM_SLEEP
1195static int omap_des_suspend(struct device *dev)
1196{
1197	pm_runtime_put_sync(dev);
1198	return 0;
1199}
1200
1201static int omap_des_resume(struct device *dev)
1202{
1203	int err;
1204
1205	err = pm_runtime_get_sync(dev);
1206	if (err < 0) {
1207		pm_runtime_put_noidle(dev);
1208		dev_err(dev, "%s: failed to get_sync(%d)\n", __func__, err);
1209		return err;
1210	}
1211	return 0;
1212}
1213#endif
1214
1215static SIMPLE_DEV_PM_OPS(omap_des_pm_ops, omap_des_suspend, omap_des_resume);
1216
1217static struct platform_driver omap_des_driver = {
1218	.probe	= omap_des_probe,
1219	.remove	= omap_des_remove,
1220	.driver	= {
1221		.name	= "omap-des",
1222		.pm	= &omap_des_pm_ops,
1223		.of_match_table	= of_match_ptr(omap_des_of_match),
1224	},
1225};
1226
1227module_platform_driver(omap_des_driver);
1228
1229MODULE_DESCRIPTION("OMAP DES hw acceleration support.");
1230MODULE_LICENSE("GPL v2");
1231MODULE_AUTHOR("Joel Fernandes <joelf@ti.com>");
1232