xref: /openbmc/linux/drivers/dma/k3dma.c (revision a8fe58ce)
1 /*
2  * Copyright (c) 2013 Linaro Ltd.
3  * Copyright (c) 2013 Hisilicon Limited.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  */
9 #include <linux/sched.h>
10 #include <linux/device.h>
11 #include <linux/dmaengine.h>
12 #include <linux/init.h>
13 #include <linux/interrupt.h>
14 #include <linux/kernel.h>
15 #include <linux/module.h>
16 #include <linux/platform_device.h>
17 #include <linux/slab.h>
18 #include <linux/spinlock.h>
19 #include <linux/of_device.h>
20 #include <linux/of.h>
21 #include <linux/clk.h>
22 #include <linux/of_dma.h>
23 
24 #include "virt-dma.h"
25 
26 #define DRIVER_NAME		"k3-dma"
27 #define DMA_MAX_SIZE		0x1ffc
28 
29 #define INT_STAT		0x00
30 #define INT_TC1			0x04
31 #define INT_ERR1		0x0c
32 #define INT_ERR2		0x10
33 #define INT_TC1_MASK		0x18
34 #define INT_ERR1_MASK		0x20
35 #define INT_ERR2_MASK		0x24
36 #define INT_TC1_RAW		0x600
37 #define INT_ERR1_RAW		0x608
38 #define INT_ERR2_RAW		0x610
39 #define CH_PRI			0x688
40 #define CH_STAT			0x690
41 #define CX_CUR_CNT		0x704
42 #define CX_LLI			0x800
43 #define CX_CNT			0x810
44 #define CX_SRC			0x814
45 #define CX_DST			0x818
46 #define CX_CFG			0x81c
47 #define AXI_CFG			0x820
48 #define AXI_CFG_DEFAULT		0x201201
49 
50 #define CX_LLI_CHAIN_EN		0x2
51 #define CX_CFG_EN		0x1
52 #define CX_CFG_MEM2PER		(0x1 << 2)
53 #define CX_CFG_PER2MEM		(0x2 << 2)
54 #define CX_CFG_SRCINCR		(0x1 << 31)
55 #define CX_CFG_DSTINCR		(0x1 << 30)
56 
57 struct k3_desc_hw {
58 	u32 lli;
59 	u32 reserved[3];
60 	u32 count;
61 	u32 saddr;
62 	u32 daddr;
63 	u32 config;
64 } __aligned(32);
65 
66 struct k3_dma_desc_sw {
67 	struct virt_dma_desc	vd;
68 	dma_addr_t		desc_hw_lli;
69 	size_t			desc_num;
70 	size_t			size;
71 	struct k3_desc_hw	desc_hw[0];
72 };
73 
74 struct k3_dma_phy;
75 
76 struct k3_dma_chan {
77 	u32			ccfg;
78 	struct virt_dma_chan	vc;
79 	struct k3_dma_phy	*phy;
80 	struct list_head	node;
81 	enum dma_transfer_direction dir;
82 	dma_addr_t		dev_addr;
83 	enum dma_status		status;
84 };
85 
86 struct k3_dma_phy {
87 	u32			idx;
88 	void __iomem		*base;
89 	struct k3_dma_chan	*vchan;
90 	struct k3_dma_desc_sw	*ds_run;
91 	struct k3_dma_desc_sw	*ds_done;
92 };
93 
94 struct k3_dma_dev {
95 	struct dma_device	slave;
96 	void __iomem		*base;
97 	struct tasklet_struct	task;
98 	spinlock_t		lock;
99 	struct list_head	chan_pending;
100 	struct k3_dma_phy	*phy;
101 	struct k3_dma_chan	*chans;
102 	struct clk		*clk;
103 	u32			dma_channels;
104 	u32			dma_requests;
105 };
106 
107 #define to_k3_dma(dmadev) container_of(dmadev, struct k3_dma_dev, slave)
108 
109 static struct k3_dma_chan *to_k3_chan(struct dma_chan *chan)
110 {
111 	return container_of(chan, struct k3_dma_chan, vc.chan);
112 }
113 
114 static void k3_dma_pause_dma(struct k3_dma_phy *phy, bool on)
115 {
116 	u32 val = 0;
117 
118 	if (on) {
119 		val = readl_relaxed(phy->base + CX_CFG);
120 		val |= CX_CFG_EN;
121 		writel_relaxed(val, phy->base + CX_CFG);
122 	} else {
123 		val = readl_relaxed(phy->base + CX_CFG);
124 		val &= ~CX_CFG_EN;
125 		writel_relaxed(val, phy->base + CX_CFG);
126 	}
127 }
128 
129 static void k3_dma_terminate_chan(struct k3_dma_phy *phy, struct k3_dma_dev *d)
130 {
131 	u32 val = 0;
132 
133 	k3_dma_pause_dma(phy, false);
134 
135 	val = 0x1 << phy->idx;
136 	writel_relaxed(val, d->base + INT_TC1_RAW);
137 	writel_relaxed(val, d->base + INT_ERR1_RAW);
138 	writel_relaxed(val, d->base + INT_ERR2_RAW);
139 }
140 
141 static void k3_dma_set_desc(struct k3_dma_phy *phy, struct k3_desc_hw *hw)
142 {
143 	writel_relaxed(hw->lli, phy->base + CX_LLI);
144 	writel_relaxed(hw->count, phy->base + CX_CNT);
145 	writel_relaxed(hw->saddr, phy->base + CX_SRC);
146 	writel_relaxed(hw->daddr, phy->base + CX_DST);
147 	writel_relaxed(AXI_CFG_DEFAULT, phy->base + AXI_CFG);
148 	writel_relaxed(hw->config, phy->base + CX_CFG);
149 }
150 
151 static u32 k3_dma_get_curr_cnt(struct k3_dma_dev *d, struct k3_dma_phy *phy)
152 {
153 	u32 cnt = 0;
154 
155 	cnt = readl_relaxed(d->base + CX_CUR_CNT + phy->idx * 0x10);
156 	cnt &= 0xffff;
157 	return cnt;
158 }
159 
160 static u32 k3_dma_get_curr_lli(struct k3_dma_phy *phy)
161 {
162 	return readl_relaxed(phy->base + CX_LLI);
163 }
164 
165 static u32 k3_dma_get_chan_stat(struct k3_dma_dev *d)
166 {
167 	return readl_relaxed(d->base + CH_STAT);
168 }
169 
170 static void k3_dma_enable_dma(struct k3_dma_dev *d, bool on)
171 {
172 	if (on) {
173 		/* set same priority */
174 		writel_relaxed(0x0, d->base + CH_PRI);
175 
176 		/* unmask irq */
177 		writel_relaxed(0xffff, d->base + INT_TC1_MASK);
178 		writel_relaxed(0xffff, d->base + INT_ERR1_MASK);
179 		writel_relaxed(0xffff, d->base + INT_ERR2_MASK);
180 	} else {
181 		/* mask irq */
182 		writel_relaxed(0x0, d->base + INT_TC1_MASK);
183 		writel_relaxed(0x0, d->base + INT_ERR1_MASK);
184 		writel_relaxed(0x0, d->base + INT_ERR2_MASK);
185 	}
186 }
187 
188 static irqreturn_t k3_dma_int_handler(int irq, void *dev_id)
189 {
190 	struct k3_dma_dev *d = (struct k3_dma_dev *)dev_id;
191 	struct k3_dma_phy *p;
192 	struct k3_dma_chan *c;
193 	u32 stat = readl_relaxed(d->base + INT_STAT);
194 	u32 tc1  = readl_relaxed(d->base + INT_TC1);
195 	u32 err1 = readl_relaxed(d->base + INT_ERR1);
196 	u32 err2 = readl_relaxed(d->base + INT_ERR2);
197 	u32 i, irq_chan = 0;
198 
199 	while (stat) {
200 		i = __ffs(stat);
201 		stat &= (stat - 1);
202 		if (likely(tc1 & BIT(i))) {
203 			p = &d->phy[i];
204 			c = p->vchan;
205 			if (c) {
206 				unsigned long flags;
207 
208 				spin_lock_irqsave(&c->vc.lock, flags);
209 				vchan_cookie_complete(&p->ds_run->vd);
210 				p->ds_done = p->ds_run;
211 				spin_unlock_irqrestore(&c->vc.lock, flags);
212 			}
213 			irq_chan |= BIT(i);
214 		}
215 		if (unlikely((err1 & BIT(i)) || (err2 & BIT(i))))
216 			dev_warn(d->slave.dev, "DMA ERR\n");
217 	}
218 
219 	writel_relaxed(irq_chan, d->base + INT_TC1_RAW);
220 	writel_relaxed(err1, d->base + INT_ERR1_RAW);
221 	writel_relaxed(err2, d->base + INT_ERR2_RAW);
222 
223 	if (irq_chan) {
224 		tasklet_schedule(&d->task);
225 		return IRQ_HANDLED;
226 	} else
227 		return IRQ_NONE;
228 }
229 
230 static int k3_dma_start_txd(struct k3_dma_chan *c)
231 {
232 	struct k3_dma_dev *d = to_k3_dma(c->vc.chan.device);
233 	struct virt_dma_desc *vd = vchan_next_desc(&c->vc);
234 
235 	if (!c->phy)
236 		return -EAGAIN;
237 
238 	if (BIT(c->phy->idx) & k3_dma_get_chan_stat(d))
239 		return -EAGAIN;
240 
241 	if (vd) {
242 		struct k3_dma_desc_sw *ds =
243 			container_of(vd, struct k3_dma_desc_sw, vd);
244 		/*
245 		 * fetch and remove request from vc->desc_issued
246 		 * so vc->desc_issued only contains desc pending
247 		 */
248 		list_del(&ds->vd.node);
249 		c->phy->ds_run = ds;
250 		c->phy->ds_done = NULL;
251 		/* start dma */
252 		k3_dma_set_desc(c->phy, &ds->desc_hw[0]);
253 		return 0;
254 	}
255 	c->phy->ds_done = NULL;
256 	c->phy->ds_run = NULL;
257 	return -EAGAIN;
258 }
259 
260 static void k3_dma_tasklet(unsigned long arg)
261 {
262 	struct k3_dma_dev *d = (struct k3_dma_dev *)arg;
263 	struct k3_dma_phy *p;
264 	struct k3_dma_chan *c, *cn;
265 	unsigned pch, pch_alloc = 0;
266 
267 	/* check new dma request of running channel in vc->desc_issued */
268 	list_for_each_entry_safe(c, cn, &d->slave.channels, vc.chan.device_node) {
269 		spin_lock_irq(&c->vc.lock);
270 		p = c->phy;
271 		if (p && p->ds_done) {
272 			if (k3_dma_start_txd(c)) {
273 				/* No current txd associated with this channel */
274 				dev_dbg(d->slave.dev, "pchan %u: free\n", p->idx);
275 				/* Mark this channel free */
276 				c->phy = NULL;
277 				p->vchan = NULL;
278 			}
279 		}
280 		spin_unlock_irq(&c->vc.lock);
281 	}
282 
283 	/* check new channel request in d->chan_pending */
284 	spin_lock_irq(&d->lock);
285 	for (pch = 0; pch < d->dma_channels; pch++) {
286 		p = &d->phy[pch];
287 
288 		if (p->vchan == NULL && !list_empty(&d->chan_pending)) {
289 			c = list_first_entry(&d->chan_pending,
290 				struct k3_dma_chan, node);
291 			/* remove from d->chan_pending */
292 			list_del_init(&c->node);
293 			pch_alloc |= 1 << pch;
294 			/* Mark this channel allocated */
295 			p->vchan = c;
296 			c->phy = p;
297 			dev_dbg(d->slave.dev, "pchan %u: alloc vchan %p\n", pch, &c->vc);
298 		}
299 	}
300 	spin_unlock_irq(&d->lock);
301 
302 	for (pch = 0; pch < d->dma_channels; pch++) {
303 		if (pch_alloc & (1 << pch)) {
304 			p = &d->phy[pch];
305 			c = p->vchan;
306 			if (c) {
307 				spin_lock_irq(&c->vc.lock);
308 				k3_dma_start_txd(c);
309 				spin_unlock_irq(&c->vc.lock);
310 			}
311 		}
312 	}
313 }
314 
315 static void k3_dma_free_chan_resources(struct dma_chan *chan)
316 {
317 	struct k3_dma_chan *c = to_k3_chan(chan);
318 	struct k3_dma_dev *d = to_k3_dma(chan->device);
319 	unsigned long flags;
320 
321 	spin_lock_irqsave(&d->lock, flags);
322 	list_del_init(&c->node);
323 	spin_unlock_irqrestore(&d->lock, flags);
324 
325 	vchan_free_chan_resources(&c->vc);
326 	c->ccfg = 0;
327 }
328 
329 static enum dma_status k3_dma_tx_status(struct dma_chan *chan,
330 	dma_cookie_t cookie, struct dma_tx_state *state)
331 {
332 	struct k3_dma_chan *c = to_k3_chan(chan);
333 	struct k3_dma_dev *d = to_k3_dma(chan->device);
334 	struct k3_dma_phy *p;
335 	struct virt_dma_desc *vd;
336 	unsigned long flags;
337 	enum dma_status ret;
338 	size_t bytes = 0;
339 
340 	ret = dma_cookie_status(&c->vc.chan, cookie, state);
341 	if (ret == DMA_COMPLETE)
342 		return ret;
343 
344 	spin_lock_irqsave(&c->vc.lock, flags);
345 	p = c->phy;
346 	ret = c->status;
347 
348 	/*
349 	 * If the cookie is on our issue queue, then the residue is
350 	 * its total size.
351 	 */
352 	vd = vchan_find_desc(&c->vc, cookie);
353 	if (vd) {
354 		bytes = container_of(vd, struct k3_dma_desc_sw, vd)->size;
355 	} else if ((!p) || (!p->ds_run)) {
356 		bytes = 0;
357 	} else {
358 		struct k3_dma_desc_sw *ds = p->ds_run;
359 		u32 clli = 0, index = 0;
360 
361 		bytes = k3_dma_get_curr_cnt(d, p);
362 		clli = k3_dma_get_curr_lli(p);
363 		index = (clli - ds->desc_hw_lli) / sizeof(struct k3_desc_hw);
364 		for (; index < ds->desc_num; index++) {
365 			bytes += ds->desc_hw[index].count;
366 			/* end of lli */
367 			if (!ds->desc_hw[index].lli)
368 				break;
369 		}
370 	}
371 	spin_unlock_irqrestore(&c->vc.lock, flags);
372 	dma_set_residue(state, bytes);
373 	return ret;
374 }
375 
376 static void k3_dma_issue_pending(struct dma_chan *chan)
377 {
378 	struct k3_dma_chan *c = to_k3_chan(chan);
379 	struct k3_dma_dev *d = to_k3_dma(chan->device);
380 	unsigned long flags;
381 
382 	spin_lock_irqsave(&c->vc.lock, flags);
383 	/* add request to vc->desc_issued */
384 	if (vchan_issue_pending(&c->vc)) {
385 		spin_lock(&d->lock);
386 		if (!c->phy) {
387 			if (list_empty(&c->node)) {
388 				/* if new channel, add chan_pending */
389 				list_add_tail(&c->node, &d->chan_pending);
390 				/* check in tasklet */
391 				tasklet_schedule(&d->task);
392 				dev_dbg(d->slave.dev, "vchan %p: issued\n", &c->vc);
393 			}
394 		}
395 		spin_unlock(&d->lock);
396 	} else
397 		dev_dbg(d->slave.dev, "vchan %p: nothing to issue\n", &c->vc);
398 	spin_unlock_irqrestore(&c->vc.lock, flags);
399 }
400 
401 static void k3_dma_fill_desc(struct k3_dma_desc_sw *ds, dma_addr_t dst,
402 			dma_addr_t src, size_t len, u32 num, u32 ccfg)
403 {
404 	if ((num + 1) < ds->desc_num)
405 		ds->desc_hw[num].lli = ds->desc_hw_lli + (num + 1) *
406 			sizeof(struct k3_desc_hw);
407 	ds->desc_hw[num].lli |= CX_LLI_CHAIN_EN;
408 	ds->desc_hw[num].count = len;
409 	ds->desc_hw[num].saddr = src;
410 	ds->desc_hw[num].daddr = dst;
411 	ds->desc_hw[num].config = ccfg;
412 }
413 
414 static struct dma_async_tx_descriptor *k3_dma_prep_memcpy(
415 	struct dma_chan *chan,	dma_addr_t dst, dma_addr_t src,
416 	size_t len, unsigned long flags)
417 {
418 	struct k3_dma_chan *c = to_k3_chan(chan);
419 	struct k3_dma_desc_sw *ds;
420 	size_t copy = 0;
421 	int num = 0;
422 
423 	if (!len)
424 		return NULL;
425 
426 	num = DIV_ROUND_UP(len, DMA_MAX_SIZE);
427 	ds = kzalloc(sizeof(*ds) + num * sizeof(ds->desc_hw[0]), GFP_ATOMIC);
428 	if (!ds) {
429 		dev_dbg(chan->device->dev, "vchan %p: kzalloc fail\n", &c->vc);
430 		return NULL;
431 	}
432 	ds->desc_hw_lli = __virt_to_phys((unsigned long)&ds->desc_hw[0]);
433 	ds->size = len;
434 	ds->desc_num = num;
435 	num = 0;
436 
437 	if (!c->ccfg) {
438 		/* default is memtomem, without calling device_config */
439 		c->ccfg = CX_CFG_SRCINCR | CX_CFG_DSTINCR | CX_CFG_EN;
440 		c->ccfg |= (0xf << 20) | (0xf << 24);	/* burst = 16 */
441 		c->ccfg |= (0x3 << 12) | (0x3 << 16);	/* width = 64 bit */
442 	}
443 
444 	do {
445 		copy = min_t(size_t, len, DMA_MAX_SIZE);
446 		k3_dma_fill_desc(ds, dst, src, copy, num++, c->ccfg);
447 
448 		if (c->dir == DMA_MEM_TO_DEV) {
449 			src += copy;
450 		} else if (c->dir == DMA_DEV_TO_MEM) {
451 			dst += copy;
452 		} else {
453 			src += copy;
454 			dst += copy;
455 		}
456 		len -= copy;
457 	} while (len);
458 
459 	ds->desc_hw[num-1].lli = 0;	/* end of link */
460 	return vchan_tx_prep(&c->vc, &ds->vd, flags);
461 }
462 
463 static struct dma_async_tx_descriptor *k3_dma_prep_slave_sg(
464 	struct dma_chan *chan, struct scatterlist *sgl, unsigned int sglen,
465 	enum dma_transfer_direction dir, unsigned long flags, void *context)
466 {
467 	struct k3_dma_chan *c = to_k3_chan(chan);
468 	struct k3_dma_desc_sw *ds;
469 	size_t len, avail, total = 0;
470 	struct scatterlist *sg;
471 	dma_addr_t addr, src = 0, dst = 0;
472 	int num = sglen, i;
473 
474 	if (sgl == NULL)
475 		return NULL;
476 
477 	for_each_sg(sgl, sg, sglen, i) {
478 		avail = sg_dma_len(sg);
479 		if (avail > DMA_MAX_SIZE)
480 			num += DIV_ROUND_UP(avail, DMA_MAX_SIZE) - 1;
481 	}
482 
483 	ds = kzalloc(sizeof(*ds) + num * sizeof(ds->desc_hw[0]), GFP_ATOMIC);
484 	if (!ds) {
485 		dev_dbg(chan->device->dev, "vchan %p: kzalloc fail\n", &c->vc);
486 		return NULL;
487 	}
488 	ds->desc_hw_lli = __virt_to_phys((unsigned long)&ds->desc_hw[0]);
489 	ds->desc_num = num;
490 	num = 0;
491 
492 	for_each_sg(sgl, sg, sglen, i) {
493 		addr = sg_dma_address(sg);
494 		avail = sg_dma_len(sg);
495 		total += avail;
496 
497 		do {
498 			len = min_t(size_t, avail, DMA_MAX_SIZE);
499 
500 			if (dir == DMA_MEM_TO_DEV) {
501 				src = addr;
502 				dst = c->dev_addr;
503 			} else if (dir == DMA_DEV_TO_MEM) {
504 				src = c->dev_addr;
505 				dst = addr;
506 			}
507 
508 			k3_dma_fill_desc(ds, dst, src, len, num++, c->ccfg);
509 
510 			addr += len;
511 			avail -= len;
512 		} while (avail);
513 	}
514 
515 	ds->desc_hw[num-1].lli = 0;	/* end of link */
516 	ds->size = total;
517 	return vchan_tx_prep(&c->vc, &ds->vd, flags);
518 }
519 
520 static int k3_dma_config(struct dma_chan *chan,
521 			 struct dma_slave_config *cfg)
522 {
523 	struct k3_dma_chan *c = to_k3_chan(chan);
524 	u32 maxburst = 0, val = 0;
525 	enum dma_slave_buswidth width = DMA_SLAVE_BUSWIDTH_UNDEFINED;
526 
527 	if (cfg == NULL)
528 		return -EINVAL;
529 	c->dir = cfg->direction;
530 	if (c->dir == DMA_DEV_TO_MEM) {
531 		c->ccfg = CX_CFG_DSTINCR;
532 		c->dev_addr = cfg->src_addr;
533 		maxburst = cfg->src_maxburst;
534 		width = cfg->src_addr_width;
535 	} else if (c->dir == DMA_MEM_TO_DEV) {
536 		c->ccfg = CX_CFG_SRCINCR;
537 		c->dev_addr = cfg->dst_addr;
538 		maxburst = cfg->dst_maxburst;
539 		width = cfg->dst_addr_width;
540 	}
541 	switch (width) {
542 	case DMA_SLAVE_BUSWIDTH_1_BYTE:
543 	case DMA_SLAVE_BUSWIDTH_2_BYTES:
544 	case DMA_SLAVE_BUSWIDTH_4_BYTES:
545 	case DMA_SLAVE_BUSWIDTH_8_BYTES:
546 		val =  __ffs(width);
547 		break;
548 	default:
549 		val = 3;
550 		break;
551 	}
552 	c->ccfg |= (val << 12) | (val << 16);
553 
554 	if ((maxburst == 0) || (maxburst > 16))
555 		val = 16;
556 	else
557 		val = maxburst - 1;
558 	c->ccfg |= (val << 20) | (val << 24);
559 	c->ccfg |= CX_CFG_MEM2PER | CX_CFG_EN;
560 
561 	/* specific request line */
562 	c->ccfg |= c->vc.chan.chan_id << 4;
563 
564 	return 0;
565 }
566 
567 static int k3_dma_terminate_all(struct dma_chan *chan)
568 {
569 	struct k3_dma_chan *c = to_k3_chan(chan);
570 	struct k3_dma_dev *d = to_k3_dma(chan->device);
571 	struct k3_dma_phy *p = c->phy;
572 	unsigned long flags;
573 	LIST_HEAD(head);
574 
575 	dev_dbg(d->slave.dev, "vchan %p: terminate all\n", &c->vc);
576 
577 	/* Prevent this channel being scheduled */
578 	spin_lock(&d->lock);
579 	list_del_init(&c->node);
580 	spin_unlock(&d->lock);
581 
582 	/* Clear the tx descriptor lists */
583 	spin_lock_irqsave(&c->vc.lock, flags);
584 	vchan_get_all_descriptors(&c->vc, &head);
585 	if (p) {
586 		/* vchan is assigned to a pchan - stop the channel */
587 		k3_dma_terminate_chan(p, d);
588 		c->phy = NULL;
589 		p->vchan = NULL;
590 		p->ds_run = p->ds_done = NULL;
591 	}
592 	spin_unlock_irqrestore(&c->vc.lock, flags);
593 	vchan_dma_desc_free_list(&c->vc, &head);
594 
595 	return 0;
596 }
597 
598 static int k3_dma_transfer_pause(struct dma_chan *chan)
599 {
600 	struct k3_dma_chan *c = to_k3_chan(chan);
601 	struct k3_dma_dev *d = to_k3_dma(chan->device);
602 	struct k3_dma_phy *p = c->phy;
603 
604 	dev_dbg(d->slave.dev, "vchan %p: pause\n", &c->vc);
605 	if (c->status == DMA_IN_PROGRESS) {
606 		c->status = DMA_PAUSED;
607 		if (p) {
608 			k3_dma_pause_dma(p, false);
609 		} else {
610 			spin_lock(&d->lock);
611 			list_del_init(&c->node);
612 			spin_unlock(&d->lock);
613 		}
614 	}
615 
616 	return 0;
617 }
618 
619 static int k3_dma_transfer_resume(struct dma_chan *chan)
620 {
621 	struct k3_dma_chan *c = to_k3_chan(chan);
622 	struct k3_dma_dev *d = to_k3_dma(chan->device);
623 	struct k3_dma_phy *p = c->phy;
624 	unsigned long flags;
625 
626 	dev_dbg(d->slave.dev, "vchan %p: resume\n", &c->vc);
627 	spin_lock_irqsave(&c->vc.lock, flags);
628 	if (c->status == DMA_PAUSED) {
629 		c->status = DMA_IN_PROGRESS;
630 		if (p) {
631 			k3_dma_pause_dma(p, true);
632 		} else if (!list_empty(&c->vc.desc_issued)) {
633 			spin_lock(&d->lock);
634 			list_add_tail(&c->node, &d->chan_pending);
635 			spin_unlock(&d->lock);
636 		}
637 	}
638 	spin_unlock_irqrestore(&c->vc.lock, flags);
639 
640 	return 0;
641 }
642 
643 static void k3_dma_free_desc(struct virt_dma_desc *vd)
644 {
645 	struct k3_dma_desc_sw *ds =
646 		container_of(vd, struct k3_dma_desc_sw, vd);
647 
648 	kfree(ds);
649 }
650 
651 static const struct of_device_id k3_pdma_dt_ids[] = {
652 	{ .compatible = "hisilicon,k3-dma-1.0", },
653 	{}
654 };
655 MODULE_DEVICE_TABLE(of, k3_pdma_dt_ids);
656 
657 static struct dma_chan *k3_of_dma_simple_xlate(struct of_phandle_args *dma_spec,
658 						struct of_dma *ofdma)
659 {
660 	struct k3_dma_dev *d = ofdma->of_dma_data;
661 	unsigned int request = dma_spec->args[0];
662 
663 	if (request > d->dma_requests)
664 		return NULL;
665 
666 	return dma_get_slave_channel(&(d->chans[request].vc.chan));
667 }
668 
669 static int k3_dma_probe(struct platform_device *op)
670 {
671 	struct k3_dma_dev *d;
672 	const struct of_device_id *of_id;
673 	struct resource *iores;
674 	int i, ret, irq = 0;
675 
676 	iores = platform_get_resource(op, IORESOURCE_MEM, 0);
677 	if (!iores)
678 		return -EINVAL;
679 
680 	d = devm_kzalloc(&op->dev, sizeof(*d), GFP_KERNEL);
681 	if (!d)
682 		return -ENOMEM;
683 
684 	d->base = devm_ioremap_resource(&op->dev, iores);
685 	if (IS_ERR(d->base))
686 		return PTR_ERR(d->base);
687 
688 	of_id = of_match_device(k3_pdma_dt_ids, &op->dev);
689 	if (of_id) {
690 		of_property_read_u32((&op->dev)->of_node,
691 				"dma-channels", &d->dma_channels);
692 		of_property_read_u32((&op->dev)->of_node,
693 				"dma-requests", &d->dma_requests);
694 	}
695 
696 	d->clk = devm_clk_get(&op->dev, NULL);
697 	if (IS_ERR(d->clk)) {
698 		dev_err(&op->dev, "no dma clk\n");
699 		return PTR_ERR(d->clk);
700 	}
701 
702 	irq = platform_get_irq(op, 0);
703 	ret = devm_request_irq(&op->dev, irq,
704 			k3_dma_int_handler, 0, DRIVER_NAME, d);
705 	if (ret)
706 		return ret;
707 
708 	/* init phy channel */
709 	d->phy = devm_kzalloc(&op->dev,
710 		d->dma_channels * sizeof(struct k3_dma_phy), GFP_KERNEL);
711 	if (d->phy == NULL)
712 		return -ENOMEM;
713 
714 	for (i = 0; i < d->dma_channels; i++) {
715 		struct k3_dma_phy *p = &d->phy[i];
716 
717 		p->idx = i;
718 		p->base = d->base + i * 0x40;
719 	}
720 
721 	INIT_LIST_HEAD(&d->slave.channels);
722 	dma_cap_set(DMA_SLAVE, d->slave.cap_mask);
723 	dma_cap_set(DMA_MEMCPY, d->slave.cap_mask);
724 	d->slave.dev = &op->dev;
725 	d->slave.device_free_chan_resources = k3_dma_free_chan_resources;
726 	d->slave.device_tx_status = k3_dma_tx_status;
727 	d->slave.device_prep_dma_memcpy = k3_dma_prep_memcpy;
728 	d->slave.device_prep_slave_sg = k3_dma_prep_slave_sg;
729 	d->slave.device_issue_pending = k3_dma_issue_pending;
730 	d->slave.device_config = k3_dma_config;
731 	d->slave.device_pause = k3_dma_transfer_pause;
732 	d->slave.device_resume = k3_dma_transfer_resume;
733 	d->slave.device_terminate_all = k3_dma_terminate_all;
734 	d->slave.copy_align = DMAENGINE_ALIGN_8_BYTES;
735 
736 	/* init virtual channel */
737 	d->chans = devm_kzalloc(&op->dev,
738 		d->dma_requests * sizeof(struct k3_dma_chan), GFP_KERNEL);
739 	if (d->chans == NULL)
740 		return -ENOMEM;
741 
742 	for (i = 0; i < d->dma_requests; i++) {
743 		struct k3_dma_chan *c = &d->chans[i];
744 
745 		c->status = DMA_IN_PROGRESS;
746 		INIT_LIST_HEAD(&c->node);
747 		c->vc.desc_free = k3_dma_free_desc;
748 		vchan_init(&c->vc, &d->slave);
749 	}
750 
751 	/* Enable clock before accessing registers */
752 	ret = clk_prepare_enable(d->clk);
753 	if (ret < 0) {
754 		dev_err(&op->dev, "clk_prepare_enable failed: %d\n", ret);
755 		return ret;
756 	}
757 
758 	k3_dma_enable_dma(d, true);
759 
760 	ret = dma_async_device_register(&d->slave);
761 	if (ret)
762 		return ret;
763 
764 	ret = of_dma_controller_register((&op->dev)->of_node,
765 					k3_of_dma_simple_xlate, d);
766 	if (ret)
767 		goto of_dma_register_fail;
768 
769 	spin_lock_init(&d->lock);
770 	INIT_LIST_HEAD(&d->chan_pending);
771 	tasklet_init(&d->task, k3_dma_tasklet, (unsigned long)d);
772 	platform_set_drvdata(op, d);
773 	dev_info(&op->dev, "initialized\n");
774 
775 	return 0;
776 
777 of_dma_register_fail:
778 	dma_async_device_unregister(&d->slave);
779 	return ret;
780 }
781 
782 static int k3_dma_remove(struct platform_device *op)
783 {
784 	struct k3_dma_chan *c, *cn;
785 	struct k3_dma_dev *d = platform_get_drvdata(op);
786 
787 	dma_async_device_unregister(&d->slave);
788 	of_dma_controller_free((&op->dev)->of_node);
789 
790 	list_for_each_entry_safe(c, cn, &d->slave.channels, vc.chan.device_node) {
791 		list_del(&c->vc.chan.device_node);
792 		tasklet_kill(&c->vc.task);
793 	}
794 	tasklet_kill(&d->task);
795 	clk_disable_unprepare(d->clk);
796 	return 0;
797 }
798 
799 #ifdef CONFIG_PM_SLEEP
800 static int k3_dma_suspend_dev(struct device *dev)
801 {
802 	struct k3_dma_dev *d = dev_get_drvdata(dev);
803 	u32 stat = 0;
804 
805 	stat = k3_dma_get_chan_stat(d);
806 	if (stat) {
807 		dev_warn(d->slave.dev,
808 			"chan %d is running fail to suspend\n", stat);
809 		return -1;
810 	}
811 	k3_dma_enable_dma(d, false);
812 	clk_disable_unprepare(d->clk);
813 	return 0;
814 }
815 
816 static int k3_dma_resume_dev(struct device *dev)
817 {
818 	struct k3_dma_dev *d = dev_get_drvdata(dev);
819 	int ret = 0;
820 
821 	ret = clk_prepare_enable(d->clk);
822 	if (ret < 0) {
823 		dev_err(d->slave.dev, "clk_prepare_enable failed: %d\n", ret);
824 		return ret;
825 	}
826 	k3_dma_enable_dma(d, true);
827 	return 0;
828 }
829 #endif
830 
831 static SIMPLE_DEV_PM_OPS(k3_dma_pmops, k3_dma_suspend_dev, k3_dma_resume_dev);
832 
833 static struct platform_driver k3_pdma_driver = {
834 	.driver		= {
835 		.name	= DRIVER_NAME,
836 		.pm	= &k3_dma_pmops,
837 		.of_match_table = k3_pdma_dt_ids,
838 	},
839 	.probe		= k3_dma_probe,
840 	.remove		= k3_dma_remove,
841 };
842 
843 module_platform_driver(k3_pdma_driver);
844 
845 MODULE_DESCRIPTION("Hisilicon k3 DMA Driver");
846 MODULE_ALIAS("platform:k3dma");
847 MODULE_LICENSE("GPL v2");
848