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