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