xref: /openbmc/linux/drivers/dma/mmp_tdma.c (revision 63705da3)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Driver For Marvell Two-channel DMA Engine
4  *
5  * Copyright: Marvell International Ltd.
6  */
7 
8 #include <linux/err.h>
9 #include <linux/module.h>
10 #include <linux/init.h>
11 #include <linux/types.h>
12 #include <linux/interrupt.h>
13 #include <linux/dma-mapping.h>
14 #include <linux/slab.h>
15 #include <linux/dmaengine.h>
16 #include <linux/platform_device.h>
17 #include <linux/device.h>
18 #include <linux/platform_data/dma-mmp_tdma.h>
19 #include <linux/of_device.h>
20 #include <linux/of_dma.h>
21 
22 #include "dmaengine.h"
23 
24 /*
25  * Two-Channel DMA registers
26  */
27 #define TDBCR		0x00	/* Byte Count */
28 #define TDSAR		0x10	/* Src Addr */
29 #define TDDAR		0x20	/* Dst Addr */
30 #define TDNDPR		0x30	/* Next Desc */
31 #define TDCR		0x40	/* Control */
32 #define TDCP		0x60	/* Priority*/
33 #define TDCDPR		0x70	/* Current Desc */
34 #define TDIMR		0x80	/* Int Mask */
35 #define TDISR		0xa0	/* Int Status */
36 
37 /* Two-Channel DMA Control Register */
38 #define TDCR_SSZ_8_BITS		(0x0 << 22)	/* Sample Size */
39 #define TDCR_SSZ_12_BITS	(0x1 << 22)
40 #define TDCR_SSZ_16_BITS	(0x2 << 22)
41 #define TDCR_SSZ_20_BITS	(0x3 << 22)
42 #define TDCR_SSZ_24_BITS	(0x4 << 22)
43 #define TDCR_SSZ_32_BITS	(0x5 << 22)
44 #define TDCR_SSZ_SHIFT		(0x1 << 22)
45 #define TDCR_SSZ_MASK		(0x7 << 22)
46 #define TDCR_SSPMOD		(0x1 << 21)	/* SSP MOD */
47 #define TDCR_ABR		(0x1 << 20)	/* Channel Abort */
48 #define TDCR_CDE		(0x1 << 17)	/* Close Desc Enable */
49 #define TDCR_PACKMOD		(0x1 << 16)	/* Pack Mode (ADMA Only) */
50 #define TDCR_CHANACT		(0x1 << 14)	/* Channel Active */
51 #define TDCR_FETCHND		(0x1 << 13)	/* Fetch Next Desc */
52 #define TDCR_CHANEN		(0x1 << 12)	/* Channel Enable */
53 #define TDCR_INTMODE		(0x1 << 10)	/* Interrupt Mode */
54 #define TDCR_CHAINMOD		(0x1 << 9)	/* Chain Mode */
55 #define TDCR_BURSTSZ_MSK	(0x7 << 6)	/* Burst Size */
56 #define TDCR_BURSTSZ_4B		(0x0 << 6)
57 #define TDCR_BURSTSZ_8B		(0x1 << 6)
58 #define TDCR_BURSTSZ_16B	(0x3 << 6)
59 #define TDCR_BURSTSZ_32B	(0x6 << 6)
60 #define TDCR_BURSTSZ_64B	(0x7 << 6)
61 #define TDCR_BURSTSZ_SQU_1B		(0x5 << 6)
62 #define TDCR_BURSTSZ_SQU_2B		(0x6 << 6)
63 #define TDCR_BURSTSZ_SQU_4B		(0x0 << 6)
64 #define TDCR_BURSTSZ_SQU_8B		(0x1 << 6)
65 #define TDCR_BURSTSZ_SQU_16B	(0x3 << 6)
66 #define TDCR_BURSTSZ_SQU_32B	(0x7 << 6)
67 #define TDCR_BURSTSZ_128B	(0x5 << 6)
68 #define TDCR_DSTDIR_MSK		(0x3 << 4)	/* Dst Direction */
69 #define TDCR_DSTDIR_ADDR_HOLD	(0x2 << 4)	/* Dst Addr Hold */
70 #define TDCR_DSTDIR_ADDR_INC	(0x0 << 4)	/* Dst Addr Increment */
71 #define TDCR_SRCDIR_MSK		(0x3 << 2)	/* Src Direction */
72 #define TDCR_SRCDIR_ADDR_HOLD	(0x2 << 2)	/* Src Addr Hold */
73 #define TDCR_SRCDIR_ADDR_INC	(0x0 << 2)	/* Src Addr Increment */
74 #define TDCR_DSTDESCCONT	(0x1 << 1)
75 #define TDCR_SRCDESTCONT	(0x1 << 0)
76 
77 /* Two-Channel DMA Int Mask Register */
78 #define TDIMR_COMP		(0x1 << 0)
79 
80 /* Two-Channel DMA Int Status Register */
81 #define TDISR_COMP		(0x1 << 0)
82 
83 /*
84  * Two-Channel DMA Descriptor Struct
85  * NOTE: desc's buf must be aligned to 16 bytes.
86  */
87 struct mmp_tdma_desc {
88 	u32 byte_cnt;
89 	u32 src_addr;
90 	u32 dst_addr;
91 	u32 nxt_desc;
92 };
93 
94 enum mmp_tdma_type {
95 	MMP_AUD_TDMA = 0,
96 	PXA910_SQU,
97 };
98 
99 #define TDMA_MAX_XFER_BYTES    SZ_64K
100 
101 struct mmp_tdma_chan {
102 	struct device			*dev;
103 	struct dma_chan			chan;
104 	struct dma_async_tx_descriptor	desc;
105 	struct tasklet_struct		tasklet;
106 
107 	struct mmp_tdma_desc		*desc_arr;
108 	dma_addr_t			desc_arr_phys;
109 	int				desc_num;
110 	enum dma_transfer_direction	dir;
111 	dma_addr_t			dev_addr;
112 	u32				burst_sz;
113 	enum dma_slave_buswidth		buswidth;
114 	enum dma_status			status;
115 	struct dma_slave_config		slave_config;
116 
117 	int				idx;
118 	enum mmp_tdma_type		type;
119 	int				irq;
120 	void __iomem			*reg_base;
121 
122 	size_t				buf_len;
123 	size_t				period_len;
124 	size_t				pos;
125 
126 	struct gen_pool			*pool;
127 };
128 
129 #define TDMA_CHANNEL_NUM 2
130 struct mmp_tdma_device {
131 	struct device			*dev;
132 	void __iomem			*base;
133 	struct dma_device		device;
134 	struct mmp_tdma_chan		*tdmac[TDMA_CHANNEL_NUM];
135 };
136 
137 #define to_mmp_tdma_chan(dchan) container_of(dchan, struct mmp_tdma_chan, chan)
138 
139 static int mmp_tdma_config_write(struct dma_chan *chan,
140 				 enum dma_transfer_direction dir,
141 				 struct dma_slave_config *dmaengine_cfg);
142 
143 static void mmp_tdma_chan_set_desc(struct mmp_tdma_chan *tdmac, dma_addr_t phys)
144 {
145 	writel(phys, tdmac->reg_base + TDNDPR);
146 	writel(readl(tdmac->reg_base + TDCR) | TDCR_FETCHND,
147 					tdmac->reg_base + TDCR);
148 }
149 
150 static void mmp_tdma_enable_irq(struct mmp_tdma_chan *tdmac, bool enable)
151 {
152 	if (enable)
153 		writel(TDIMR_COMP, tdmac->reg_base + TDIMR);
154 	else
155 		writel(0, tdmac->reg_base + TDIMR);
156 }
157 
158 static void mmp_tdma_enable_chan(struct mmp_tdma_chan *tdmac)
159 {
160 	/* enable dma chan */
161 	writel(readl(tdmac->reg_base + TDCR) | TDCR_CHANEN,
162 					tdmac->reg_base + TDCR);
163 	tdmac->status = DMA_IN_PROGRESS;
164 }
165 
166 static int mmp_tdma_disable_chan(struct dma_chan *chan)
167 {
168 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
169 	u32 tdcr;
170 
171 	tdcr = readl(tdmac->reg_base + TDCR);
172 	tdcr |= TDCR_ABR;
173 	tdcr &= ~TDCR_CHANEN;
174 	writel(tdcr, tdmac->reg_base + TDCR);
175 
176 	tdmac->status = DMA_COMPLETE;
177 
178 	return 0;
179 }
180 
181 static int mmp_tdma_resume_chan(struct dma_chan *chan)
182 {
183 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
184 
185 	writel(readl(tdmac->reg_base + TDCR) | TDCR_CHANEN,
186 					tdmac->reg_base + TDCR);
187 	tdmac->status = DMA_IN_PROGRESS;
188 
189 	return 0;
190 }
191 
192 static int mmp_tdma_pause_chan(struct dma_chan *chan)
193 {
194 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
195 
196 	writel(readl(tdmac->reg_base + TDCR) & ~TDCR_CHANEN,
197 					tdmac->reg_base + TDCR);
198 	tdmac->status = DMA_PAUSED;
199 
200 	return 0;
201 }
202 
203 static int mmp_tdma_config_chan(struct dma_chan *chan)
204 {
205 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
206 	unsigned int tdcr = 0;
207 
208 	mmp_tdma_disable_chan(chan);
209 
210 	if (tdmac->dir == DMA_MEM_TO_DEV)
211 		tdcr = TDCR_DSTDIR_ADDR_HOLD | TDCR_SRCDIR_ADDR_INC;
212 	else if (tdmac->dir == DMA_DEV_TO_MEM)
213 		tdcr = TDCR_SRCDIR_ADDR_HOLD | TDCR_DSTDIR_ADDR_INC;
214 
215 	if (tdmac->type == MMP_AUD_TDMA) {
216 		tdcr |= TDCR_PACKMOD;
217 
218 		switch (tdmac->burst_sz) {
219 		case 4:
220 			tdcr |= TDCR_BURSTSZ_4B;
221 			break;
222 		case 8:
223 			tdcr |= TDCR_BURSTSZ_8B;
224 			break;
225 		case 16:
226 			tdcr |= TDCR_BURSTSZ_16B;
227 			break;
228 		case 32:
229 			tdcr |= TDCR_BURSTSZ_32B;
230 			break;
231 		case 64:
232 			tdcr |= TDCR_BURSTSZ_64B;
233 			break;
234 		case 128:
235 			tdcr |= TDCR_BURSTSZ_128B;
236 			break;
237 		default:
238 			dev_err(tdmac->dev, "unknown burst size.\n");
239 			return -EINVAL;
240 		}
241 
242 		switch (tdmac->buswidth) {
243 		case DMA_SLAVE_BUSWIDTH_1_BYTE:
244 			tdcr |= TDCR_SSZ_8_BITS;
245 			break;
246 		case DMA_SLAVE_BUSWIDTH_2_BYTES:
247 			tdcr |= TDCR_SSZ_16_BITS;
248 			break;
249 		case DMA_SLAVE_BUSWIDTH_4_BYTES:
250 			tdcr |= TDCR_SSZ_32_BITS;
251 			break;
252 		default:
253 			dev_err(tdmac->dev, "unknown bus size.\n");
254 			return -EINVAL;
255 		}
256 	} else if (tdmac->type == PXA910_SQU) {
257 		tdcr |= TDCR_SSPMOD;
258 
259 		switch (tdmac->burst_sz) {
260 		case 1:
261 			tdcr |= TDCR_BURSTSZ_SQU_1B;
262 			break;
263 		case 2:
264 			tdcr |= TDCR_BURSTSZ_SQU_2B;
265 			break;
266 		case 4:
267 			tdcr |= TDCR_BURSTSZ_SQU_4B;
268 			break;
269 		case 8:
270 			tdcr |= TDCR_BURSTSZ_SQU_8B;
271 			break;
272 		case 16:
273 			tdcr |= TDCR_BURSTSZ_SQU_16B;
274 			break;
275 		case 32:
276 			tdcr |= TDCR_BURSTSZ_SQU_32B;
277 			break;
278 		default:
279 			dev_err(tdmac->dev, "unknown burst size.\n");
280 			return -EINVAL;
281 		}
282 	}
283 
284 	writel(tdcr, tdmac->reg_base + TDCR);
285 	return 0;
286 }
287 
288 static int mmp_tdma_clear_chan_irq(struct mmp_tdma_chan *tdmac)
289 {
290 	u32 reg = readl(tdmac->reg_base + TDISR);
291 
292 	if (reg & TDISR_COMP) {
293 		/* clear irq */
294 		reg &= ~TDISR_COMP;
295 		writel(reg, tdmac->reg_base + TDISR);
296 
297 		return 0;
298 	}
299 	return -EAGAIN;
300 }
301 
302 static size_t mmp_tdma_get_pos(struct mmp_tdma_chan *tdmac)
303 {
304 	size_t reg;
305 
306 	if (tdmac->idx == 0) {
307 		reg = __raw_readl(tdmac->reg_base + TDSAR);
308 		reg -= tdmac->desc_arr[0].src_addr;
309 	} else if (tdmac->idx == 1) {
310 		reg = __raw_readl(tdmac->reg_base + TDDAR);
311 		reg -= tdmac->desc_arr[0].dst_addr;
312 	} else
313 		return -EINVAL;
314 
315 	return reg;
316 }
317 
318 static irqreturn_t mmp_tdma_chan_handler(int irq, void *dev_id)
319 {
320 	struct mmp_tdma_chan *tdmac = dev_id;
321 
322 	if (mmp_tdma_clear_chan_irq(tdmac) == 0) {
323 		tasklet_schedule(&tdmac->tasklet);
324 		return IRQ_HANDLED;
325 	} else
326 		return IRQ_NONE;
327 }
328 
329 static irqreturn_t mmp_tdma_int_handler(int irq, void *dev_id)
330 {
331 	struct mmp_tdma_device *tdev = dev_id;
332 	int i, ret;
333 	int irq_num = 0;
334 
335 	for (i = 0; i < TDMA_CHANNEL_NUM; i++) {
336 		struct mmp_tdma_chan *tdmac = tdev->tdmac[i];
337 
338 		ret = mmp_tdma_chan_handler(irq, tdmac);
339 		if (ret == IRQ_HANDLED)
340 			irq_num++;
341 	}
342 
343 	if (irq_num)
344 		return IRQ_HANDLED;
345 	else
346 		return IRQ_NONE;
347 }
348 
349 static void dma_do_tasklet(struct tasklet_struct *t)
350 {
351 	struct mmp_tdma_chan *tdmac = from_tasklet(tdmac, t, tasklet);
352 
353 	dmaengine_desc_get_callback_invoke(&tdmac->desc, NULL);
354 }
355 
356 static void mmp_tdma_free_descriptor(struct mmp_tdma_chan *tdmac)
357 {
358 	struct gen_pool *gpool;
359 	int size = tdmac->desc_num * sizeof(struct mmp_tdma_desc);
360 
361 	gpool = tdmac->pool;
362 	if (gpool && tdmac->desc_arr)
363 		gen_pool_free(gpool, (unsigned long)tdmac->desc_arr,
364 				size);
365 	tdmac->desc_arr = NULL;
366 	if (tdmac->status == DMA_ERROR)
367 		tdmac->status = DMA_COMPLETE;
368 
369 	return;
370 }
371 
372 static dma_cookie_t mmp_tdma_tx_submit(struct dma_async_tx_descriptor *tx)
373 {
374 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(tx->chan);
375 
376 	mmp_tdma_chan_set_desc(tdmac, tdmac->desc_arr_phys);
377 
378 	return 0;
379 }
380 
381 static int mmp_tdma_alloc_chan_resources(struct dma_chan *chan)
382 {
383 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
384 	int ret;
385 
386 	dma_async_tx_descriptor_init(&tdmac->desc, chan);
387 	tdmac->desc.tx_submit = mmp_tdma_tx_submit;
388 
389 	if (tdmac->irq) {
390 		ret = devm_request_irq(tdmac->dev, tdmac->irq,
391 			mmp_tdma_chan_handler, 0, "tdma", tdmac);
392 		if (ret)
393 			return ret;
394 	}
395 	return 1;
396 }
397 
398 static void mmp_tdma_free_chan_resources(struct dma_chan *chan)
399 {
400 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
401 
402 	if (tdmac->irq)
403 		devm_free_irq(tdmac->dev, tdmac->irq, tdmac);
404 	mmp_tdma_free_descriptor(tdmac);
405 	return;
406 }
407 
408 static struct mmp_tdma_desc *mmp_tdma_alloc_descriptor(struct mmp_tdma_chan *tdmac)
409 {
410 	struct gen_pool *gpool;
411 	int size = tdmac->desc_num * sizeof(struct mmp_tdma_desc);
412 
413 	gpool = tdmac->pool;
414 	if (!gpool)
415 		return NULL;
416 
417 	tdmac->desc_arr = gen_pool_dma_alloc(gpool, size, &tdmac->desc_arr_phys);
418 
419 	return tdmac->desc_arr;
420 }
421 
422 static struct dma_async_tx_descriptor *mmp_tdma_prep_dma_cyclic(
423 		struct dma_chan *chan, dma_addr_t dma_addr, size_t buf_len,
424 		size_t period_len, enum dma_transfer_direction direction,
425 		unsigned long flags)
426 {
427 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
428 	struct mmp_tdma_desc *desc;
429 	int num_periods = buf_len / period_len;
430 	int i = 0, buf = 0;
431 
432 	if (!is_slave_direction(direction)) {
433 		dev_err(tdmac->dev, "unsupported transfer direction\n");
434 		return NULL;
435 	}
436 
437 	if (tdmac->status != DMA_COMPLETE) {
438 		dev_err(tdmac->dev, "controller busy");
439 		return NULL;
440 	}
441 
442 	if (period_len > TDMA_MAX_XFER_BYTES) {
443 		dev_err(tdmac->dev,
444 				"maximum period size exceeded: %zu > %d\n",
445 				period_len, TDMA_MAX_XFER_BYTES);
446 		goto err_out;
447 	}
448 
449 	tdmac->status = DMA_IN_PROGRESS;
450 	tdmac->desc_num = num_periods;
451 	desc = mmp_tdma_alloc_descriptor(tdmac);
452 	if (!desc)
453 		goto err_out;
454 
455 	if (mmp_tdma_config_write(chan, direction, &tdmac->slave_config))
456 		goto err_out;
457 
458 	while (buf < buf_len) {
459 		desc = &tdmac->desc_arr[i];
460 
461 		if (i + 1 == num_periods)
462 			desc->nxt_desc = tdmac->desc_arr_phys;
463 		else
464 			desc->nxt_desc = tdmac->desc_arr_phys +
465 				sizeof(*desc) * (i + 1);
466 
467 		if (direction == DMA_MEM_TO_DEV) {
468 			desc->src_addr = dma_addr;
469 			desc->dst_addr = tdmac->dev_addr;
470 		} else {
471 			desc->src_addr = tdmac->dev_addr;
472 			desc->dst_addr = dma_addr;
473 		}
474 		desc->byte_cnt = period_len;
475 		dma_addr += period_len;
476 		buf += period_len;
477 		i++;
478 	}
479 
480 	/* enable interrupt */
481 	if (flags & DMA_PREP_INTERRUPT)
482 		mmp_tdma_enable_irq(tdmac, true);
483 
484 	tdmac->buf_len = buf_len;
485 	tdmac->period_len = period_len;
486 	tdmac->pos = 0;
487 
488 	return &tdmac->desc;
489 
490 err_out:
491 	tdmac->status = DMA_ERROR;
492 	return NULL;
493 }
494 
495 static int mmp_tdma_terminate_all(struct dma_chan *chan)
496 {
497 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
498 
499 	mmp_tdma_disable_chan(chan);
500 	/* disable interrupt */
501 	mmp_tdma_enable_irq(tdmac, false);
502 
503 	return 0;
504 }
505 
506 static int mmp_tdma_config(struct dma_chan *chan,
507 			   struct dma_slave_config *dmaengine_cfg)
508 {
509 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
510 
511 	memcpy(&tdmac->slave_config, dmaengine_cfg, sizeof(*dmaengine_cfg));
512 
513 	return 0;
514 }
515 
516 static int mmp_tdma_config_write(struct dma_chan *chan,
517 				 enum dma_transfer_direction dir,
518 				 struct dma_slave_config *dmaengine_cfg)
519 {
520 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
521 
522 	if (dir == DMA_DEV_TO_MEM) {
523 		tdmac->dev_addr = dmaengine_cfg->src_addr;
524 		tdmac->burst_sz = dmaengine_cfg->src_maxburst;
525 		tdmac->buswidth = dmaengine_cfg->src_addr_width;
526 	} else {
527 		tdmac->dev_addr = dmaengine_cfg->dst_addr;
528 		tdmac->burst_sz = dmaengine_cfg->dst_maxburst;
529 		tdmac->buswidth = dmaengine_cfg->dst_addr_width;
530 	}
531 	tdmac->dir = dir;
532 
533 	return mmp_tdma_config_chan(chan);
534 }
535 
536 static enum dma_status mmp_tdma_tx_status(struct dma_chan *chan,
537 			dma_cookie_t cookie, struct dma_tx_state *txstate)
538 {
539 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
540 
541 	tdmac->pos = mmp_tdma_get_pos(tdmac);
542 	dma_set_tx_state(txstate, chan->completed_cookie, chan->cookie,
543 			 tdmac->buf_len - tdmac->pos);
544 
545 	return tdmac->status;
546 }
547 
548 static void mmp_tdma_issue_pending(struct dma_chan *chan)
549 {
550 	struct mmp_tdma_chan *tdmac = to_mmp_tdma_chan(chan);
551 
552 	mmp_tdma_enable_chan(tdmac);
553 }
554 
555 static int mmp_tdma_remove(struct platform_device *pdev)
556 {
557 	if (pdev->dev.of_node)
558 		of_dma_controller_free(pdev->dev.of_node);
559 
560 	return 0;
561 }
562 
563 static int mmp_tdma_chan_init(struct mmp_tdma_device *tdev,
564 					int idx, int irq,
565 					int type, struct gen_pool *pool)
566 {
567 	struct mmp_tdma_chan *tdmac;
568 
569 	if (idx >= TDMA_CHANNEL_NUM) {
570 		dev_err(tdev->dev, "too many channels for device!\n");
571 		return -EINVAL;
572 	}
573 
574 	/* alloc channel */
575 	tdmac = devm_kzalloc(tdev->dev, sizeof(*tdmac), GFP_KERNEL);
576 	if (!tdmac)
577 		return -ENOMEM;
578 
579 	if (irq)
580 		tdmac->irq = irq;
581 	tdmac->dev	   = tdev->dev;
582 	tdmac->chan.device = &tdev->device;
583 	tdmac->idx	   = idx;
584 	tdmac->type	   = type;
585 	tdmac->reg_base	   = tdev->base + idx * 4;
586 	tdmac->pool	   = pool;
587 	tdmac->status = DMA_COMPLETE;
588 	tdev->tdmac[tdmac->idx] = tdmac;
589 	tasklet_setup(&tdmac->tasklet, dma_do_tasklet);
590 
591 	/* add the channel to tdma_chan list */
592 	list_add_tail(&tdmac->chan.device_node,
593 			&tdev->device.channels);
594 	return 0;
595 }
596 
597 struct mmp_tdma_filter_param {
598 	unsigned int chan_id;
599 };
600 
601 static bool mmp_tdma_filter_fn(struct dma_chan *chan, void *fn_param)
602 {
603 	struct mmp_tdma_filter_param *param = fn_param;
604 
605 	if (chan->chan_id != param->chan_id)
606 		return false;
607 
608 	return true;
609 }
610 
611 static struct dma_chan *mmp_tdma_xlate(struct of_phandle_args *dma_spec,
612 			       struct of_dma *ofdma)
613 {
614 	struct mmp_tdma_device *tdev = ofdma->of_dma_data;
615 	dma_cap_mask_t mask = tdev->device.cap_mask;
616 	struct mmp_tdma_filter_param param;
617 
618 	if (dma_spec->args_count != 1)
619 		return NULL;
620 
621 	param.chan_id = dma_spec->args[0];
622 
623 	if (param.chan_id >= TDMA_CHANNEL_NUM)
624 		return NULL;
625 
626 	return __dma_request_channel(&mask, mmp_tdma_filter_fn, &param,
627 				     ofdma->of_node);
628 }
629 
630 static const struct of_device_id mmp_tdma_dt_ids[] = {
631 	{ .compatible = "marvell,adma-1.0", .data = (void *)MMP_AUD_TDMA},
632 	{ .compatible = "marvell,pxa910-squ", .data = (void *)PXA910_SQU},
633 	{}
634 };
635 MODULE_DEVICE_TABLE(of, mmp_tdma_dt_ids);
636 
637 static int mmp_tdma_probe(struct platform_device *pdev)
638 {
639 	enum mmp_tdma_type type;
640 	const struct of_device_id *of_id;
641 	struct mmp_tdma_device *tdev;
642 	struct resource *iores;
643 	int i, ret;
644 	int irq = 0, irq_num = 0;
645 	int chan_num = TDMA_CHANNEL_NUM;
646 	struct gen_pool *pool = NULL;
647 
648 	of_id = of_match_device(mmp_tdma_dt_ids, &pdev->dev);
649 	if (of_id)
650 		type = (enum mmp_tdma_type) of_id->data;
651 	else
652 		type = platform_get_device_id(pdev)->driver_data;
653 
654 	/* always have couple channels */
655 	tdev = devm_kzalloc(&pdev->dev, sizeof(*tdev), GFP_KERNEL);
656 	if (!tdev)
657 		return -ENOMEM;
658 
659 	tdev->dev = &pdev->dev;
660 
661 	for (i = 0; i < chan_num; i++) {
662 		if (platform_get_irq(pdev, i) > 0)
663 			irq_num++;
664 	}
665 
666 	iores = platform_get_resource(pdev, IORESOURCE_MEM, 0);
667 	tdev->base = devm_ioremap_resource(&pdev->dev, iores);
668 	if (IS_ERR(tdev->base))
669 		return PTR_ERR(tdev->base);
670 
671 	INIT_LIST_HEAD(&tdev->device.channels);
672 
673 	if (pdev->dev.of_node)
674 		pool = of_gen_pool_get(pdev->dev.of_node, "asram", 0);
675 	else
676 		pool = sram_get_gpool("asram");
677 	if (!pool) {
678 		dev_err(&pdev->dev, "asram pool not available\n");
679 		return -ENOMEM;
680 	}
681 
682 	if (irq_num != chan_num) {
683 		irq = platform_get_irq(pdev, 0);
684 		ret = devm_request_irq(&pdev->dev, irq,
685 			mmp_tdma_int_handler, IRQF_SHARED, "tdma", tdev);
686 		if (ret)
687 			return ret;
688 	}
689 
690 	/* initialize channel parameters */
691 	for (i = 0; i < chan_num; i++) {
692 		irq = (irq_num != chan_num) ? 0 : platform_get_irq(pdev, i);
693 		ret = mmp_tdma_chan_init(tdev, i, irq, type, pool);
694 		if (ret)
695 			return ret;
696 	}
697 
698 	dma_cap_set(DMA_SLAVE, tdev->device.cap_mask);
699 	dma_cap_set(DMA_CYCLIC, tdev->device.cap_mask);
700 	tdev->device.dev = &pdev->dev;
701 	tdev->device.device_alloc_chan_resources =
702 					mmp_tdma_alloc_chan_resources;
703 	tdev->device.device_free_chan_resources =
704 					mmp_tdma_free_chan_resources;
705 	tdev->device.device_prep_dma_cyclic = mmp_tdma_prep_dma_cyclic;
706 	tdev->device.device_tx_status = mmp_tdma_tx_status;
707 	tdev->device.device_issue_pending = mmp_tdma_issue_pending;
708 	tdev->device.device_config = mmp_tdma_config;
709 	tdev->device.device_pause = mmp_tdma_pause_chan;
710 	tdev->device.device_resume = mmp_tdma_resume_chan;
711 	tdev->device.device_terminate_all = mmp_tdma_terminate_all;
712 	tdev->device.copy_align = DMAENGINE_ALIGN_8_BYTES;
713 
714 	tdev->device.directions = BIT(DMA_DEV_TO_MEM) | BIT(DMA_MEM_TO_DEV);
715 	if (type == MMP_AUD_TDMA) {
716 		tdev->device.max_burst = SZ_128;
717 		tdev->device.src_addr_widths = BIT(DMA_SLAVE_BUSWIDTH_4_BYTES);
718 		tdev->device.dst_addr_widths = BIT(DMA_SLAVE_BUSWIDTH_4_BYTES);
719 	} else if (type == PXA910_SQU) {
720 		tdev->device.max_burst = SZ_32;
721 	}
722 	tdev->device.residue_granularity = DMA_RESIDUE_GRANULARITY_BURST;
723 	tdev->device.descriptor_reuse = true;
724 
725 	dma_set_mask(&pdev->dev, DMA_BIT_MASK(64));
726 	platform_set_drvdata(pdev, tdev);
727 
728 	ret = dmaenginem_async_device_register(&tdev->device);
729 	if (ret) {
730 		dev_err(tdev->device.dev, "unable to register\n");
731 		return ret;
732 	}
733 
734 	if (pdev->dev.of_node) {
735 		ret = of_dma_controller_register(pdev->dev.of_node,
736 							mmp_tdma_xlate, tdev);
737 		if (ret) {
738 			dev_err(tdev->device.dev,
739 				"failed to register controller\n");
740 			return ret;
741 		}
742 	}
743 
744 	dev_info(tdev->device.dev, "initialized\n");
745 	return 0;
746 }
747 
748 static const struct platform_device_id mmp_tdma_id_table[] = {
749 	{ "mmp-adma",	MMP_AUD_TDMA },
750 	{ "pxa910-squ",	PXA910_SQU },
751 	{ },
752 };
753 
754 static struct platform_driver mmp_tdma_driver = {
755 	.driver		= {
756 		.name	= "mmp-tdma",
757 		.of_match_table = mmp_tdma_dt_ids,
758 	},
759 	.id_table	= mmp_tdma_id_table,
760 	.probe		= mmp_tdma_probe,
761 	.remove		= mmp_tdma_remove,
762 };
763 
764 module_platform_driver(mmp_tdma_driver);
765 
766 MODULE_LICENSE("GPL");
767 MODULE_DESCRIPTION("MMP Two-Channel DMA Driver");
768 MODULE_ALIAS("platform:mmp-tdma");
769 MODULE_AUTHOR("Leo Yan <leoy@marvell.com>");
770 MODULE_AUTHOR("Zhangfei Gao <zhangfei.gao@marvell.com>");
771