1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Driver for the Loongson LS2X APB DMA Controller
4 *
5 * Copyright (C) 2017-2023 Loongson Corporation
6 */
7
8#include <linux/clk.h>
9#include <linux/dma-mapping.h>
10#include <linux/dmapool.h>
11#include <linux/interrupt.h>
12#include <linux/io.h>
13#include <linux/io-64-nonatomic-lo-hi.h>
14#include <linux/module.h>
15#include <linux/of.h>
16#include <linux/of_dma.h>
17#include <linux/platform_device.h>
18#include <linux/slab.h>
19
20#include "dmaengine.h"
21#include "virt-dma.h"
22
23/* Global Configuration Register */
24#define LDMA_ORDER_ERG		0x0
25
26/* Bitfield definitions */
27
28/* Bitfields in Global Configuration Register */
29#define LDMA_64BIT_EN		BIT(0) /* 1: 64 bit support */
30#define LDMA_UNCOHERENT_EN	BIT(1) /* 0: cache, 1: uncache */
31#define LDMA_ASK_VALID		BIT(2)
32#define LDMA_START		BIT(3) /* DMA start operation */
33#define LDMA_STOP		BIT(4) /* DMA stop operation */
34#define LDMA_CONFIG_MASK	GENMASK(4, 0) /* DMA controller config bits mask */
35
36/* Bitfields in ndesc_addr field of HW decriptor */
37#define LDMA_DESC_EN		BIT(0) /*1: The next descriptor is valid */
38#define LDMA_DESC_ADDR_LOW	GENMASK(31, 1)
39
40/* Bitfields in cmd field of HW decriptor */
41#define LDMA_INT		BIT(1) /* Enable DMA interrupts */
42#define LDMA_DATA_DIRECTION	BIT(12) /* 1: write to device, 0: read from device */
43
44#define LDMA_SLAVE_BUSWIDTHS	(BIT(DMA_SLAVE_BUSWIDTH_4_BYTES) | \
45				 BIT(DMA_SLAVE_BUSWIDTH_8_BYTES))
46
47#define LDMA_MAX_TRANS_LEN	U32_MAX
48
49/*--  descriptors  -----------------------------------------------------*/
50
51/*
52 * struct ls2x_dma_hw_desc - DMA HW descriptor
53 * @ndesc_addr: the next descriptor low address.
54 * @mem_addr: memory low address.
55 * @apb_addr: device buffer address.
56 * @len: length of a piece of carried content, in words.
57 * @step_len: length between two moved memory data blocks.
58 * @step_times: number of blocks to be carried in a single DMA operation.
59 * @cmd: descriptor command or state.
60 * @stats: DMA status.
61 * @high_ndesc_addr: the next descriptor high address.
62 * @high_mem_addr: memory high address.
63 * @reserved: reserved
64 */
65struct ls2x_dma_hw_desc {
66	u32 ndesc_addr;
67	u32 mem_addr;
68	u32 apb_addr;
69	u32 len;
70	u32 step_len;
71	u32 step_times;
72	u32 cmd;
73	u32 stats;
74	u32 high_ndesc_addr;
75	u32 high_mem_addr;
76	u32 reserved[2];
77} __packed;
78
79/*
80 * struct ls2x_dma_sg - ls2x dma scatter gather entry
81 * @hw: the pointer to DMA HW descriptor.
82 * @llp: physical address of the DMA HW descriptor.
83 * @phys: destination or source address(mem).
84 * @len: number of Bytes to read.
85 */
86struct ls2x_dma_sg {
87	struct ls2x_dma_hw_desc	*hw;
88	dma_addr_t		llp;
89	dma_addr_t		phys;
90	u32			len;
91};
92
93/*
94 * struct ls2x_dma_desc - software descriptor
95 * @vdesc: pointer to the virtual dma descriptor.
96 * @cyclic: flag to dma cyclic
97 * @burst_size: burst size of transaction, in words.
98 * @desc_num: number of sg entries.
99 * @direction: transfer direction, to or from device.
100 * @status: dma controller status.
101 * @sg: array of sgs.
102 */
103struct ls2x_dma_desc {
104	struct virt_dma_desc		vdesc;
105	bool				cyclic;
106	size_t				burst_size;
107	u32				desc_num;
108	enum dma_transfer_direction	direction;
109	enum dma_status			status;
110	struct ls2x_dma_sg		sg[] __counted_by(desc_num);
111};
112
113/*--  Channels  --------------------------------------------------------*/
114
115/*
116 * struct ls2x_dma_chan - internal representation of an LS2X APB DMA channel
117 * @vchan: virtual dma channel entry.
118 * @desc: pointer to the ls2x sw dma descriptor.
119 * @pool: hw desc table
120 * @irq: irq line
121 * @sconfig: configuration for slave transfers, passed via .device_config
122 */
123struct ls2x_dma_chan {
124	struct virt_dma_chan	vchan;
125	struct ls2x_dma_desc	*desc;
126	void			*pool;
127	int			irq;
128	struct dma_slave_config	sconfig;
129};
130
131/*--  Controller  ------------------------------------------------------*/
132
133/*
134 * struct ls2x_dma_priv - LS2X APB DMAC specific information
135 * @ddev: dmaengine dma_device object members
136 * @dma_clk: DMAC clock source
137 * @regs: memory mapped register base
138 * @lchan: channel to store ls2x_dma_chan structures
139 */
140struct ls2x_dma_priv {
141	struct dma_device	ddev;
142	struct clk		*dma_clk;
143	void __iomem		*regs;
144	struct ls2x_dma_chan	lchan;
145};
146
147/*--  Helper functions  ------------------------------------------------*/
148
149static inline struct ls2x_dma_desc *to_ldma_desc(struct virt_dma_desc *vdesc)
150{
151	return container_of(vdesc, struct ls2x_dma_desc, vdesc);
152}
153
154static inline struct ls2x_dma_chan *to_ldma_chan(struct dma_chan *chan)
155{
156	return container_of(chan, struct ls2x_dma_chan, vchan.chan);
157}
158
159static inline struct ls2x_dma_priv *to_ldma_priv(struct dma_device *ddev)
160{
161	return container_of(ddev, struct ls2x_dma_priv, ddev);
162}
163
164static struct device *chan2dev(struct dma_chan *chan)
165{
166	return &chan->dev->device;
167}
168
169static void ls2x_dma_desc_free(struct virt_dma_desc *vdesc)
170{
171	struct ls2x_dma_chan *lchan = to_ldma_chan(vdesc->tx.chan);
172	struct ls2x_dma_desc *desc = to_ldma_desc(vdesc);
173	int i;
174
175	for (i = 0; i < desc->desc_num; i++) {
176		if (desc->sg[i].hw)
177			dma_pool_free(lchan->pool, desc->sg[i].hw,
178				      desc->sg[i].llp);
179	}
180
181	kfree(desc);
182}
183
184static void ls2x_dma_write_cmd(struct ls2x_dma_chan *lchan, bool cmd)
185{
186	struct ls2x_dma_priv *priv = to_ldma_priv(lchan->vchan.chan.device);
187	u64 val;
188
189	val = lo_hi_readq(priv->regs + LDMA_ORDER_ERG) & ~LDMA_CONFIG_MASK;
190	val |= LDMA_64BIT_EN | cmd;
191	lo_hi_writeq(val, priv->regs + LDMA_ORDER_ERG);
192}
193
194static void ls2x_dma_start_transfer(struct ls2x_dma_chan *lchan)
195{
196	struct ls2x_dma_priv *priv = to_ldma_priv(lchan->vchan.chan.device);
197	struct ls2x_dma_sg *ldma_sg;
198	struct virt_dma_desc *vdesc;
199	u64 val;
200
201	/* Get the next descriptor */
202	vdesc = vchan_next_desc(&lchan->vchan);
203	if (!vdesc) {
204		lchan->desc = NULL;
205		return;
206	}
207
208	list_del(&vdesc->node);
209	lchan->desc = to_ldma_desc(vdesc);
210	ldma_sg = &lchan->desc->sg[0];
211
212	/* Start DMA */
213	lo_hi_writeq(0, priv->regs + LDMA_ORDER_ERG);
214	val = (ldma_sg->llp & ~LDMA_CONFIG_MASK) | LDMA_64BIT_EN | LDMA_START;
215	lo_hi_writeq(val, priv->regs + LDMA_ORDER_ERG);
216}
217
218static size_t ls2x_dmac_detect_burst(struct ls2x_dma_chan *lchan)
219{
220	u32 maxburst, buswidth;
221
222	/* Reject definitely invalid configurations */
223	if ((lchan->sconfig.src_addr_width & LDMA_SLAVE_BUSWIDTHS) &&
224	    (lchan->sconfig.dst_addr_width & LDMA_SLAVE_BUSWIDTHS))
225		return 0;
226
227	if (lchan->sconfig.direction == DMA_MEM_TO_DEV) {
228		maxburst = lchan->sconfig.dst_maxburst;
229		buswidth = lchan->sconfig.dst_addr_width;
230	} else {
231		maxburst = lchan->sconfig.src_maxburst;
232		buswidth = lchan->sconfig.src_addr_width;
233	}
234
235	/* If maxburst is zero, fallback to LDMA_MAX_TRANS_LEN */
236	return maxburst ? (maxburst * buswidth) >> 2 : LDMA_MAX_TRANS_LEN;
237}
238
239static void ls2x_dma_fill_desc(struct ls2x_dma_chan *lchan, u32 sg_index,
240			       struct ls2x_dma_desc *desc)
241{
242	struct ls2x_dma_sg *ldma_sg = &desc->sg[sg_index];
243	u32 num_segments, segment_size;
244
245	if (desc->direction == DMA_MEM_TO_DEV) {
246		ldma_sg->hw->cmd = LDMA_INT | LDMA_DATA_DIRECTION;
247		ldma_sg->hw->apb_addr = lchan->sconfig.dst_addr;
248	} else {
249		ldma_sg->hw->cmd = LDMA_INT;
250		ldma_sg->hw->apb_addr = lchan->sconfig.src_addr;
251	}
252
253	ldma_sg->hw->mem_addr = lower_32_bits(ldma_sg->phys);
254	ldma_sg->hw->high_mem_addr = upper_32_bits(ldma_sg->phys);
255
256	/* Split into multiple equally sized segments if necessary */
257	num_segments = DIV_ROUND_UP((ldma_sg->len + 3) >> 2, desc->burst_size);
258	segment_size = DIV_ROUND_UP((ldma_sg->len + 3) >> 2, num_segments);
259
260	/* Word count register takes input in words */
261	ldma_sg->hw->len = segment_size;
262	ldma_sg->hw->step_times = num_segments;
263	ldma_sg->hw->step_len = 0;
264
265	/* lets make a link list */
266	if (sg_index) {
267		desc->sg[sg_index - 1].hw->ndesc_addr = ldma_sg->llp | LDMA_DESC_EN;
268		desc->sg[sg_index - 1].hw->high_ndesc_addr = upper_32_bits(ldma_sg->llp);
269	}
270}
271
272/*--  DMA Engine API  --------------------------------------------------*/
273
274/*
275 * ls2x_dma_alloc_chan_resources - allocate resources for DMA channel
276 * @chan: allocate descriptor resources for this channel
277 *
278 * return - the number of allocated descriptors
279 */
280static int ls2x_dma_alloc_chan_resources(struct dma_chan *chan)
281{
282	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
283
284	/* Create a pool of consistent memory blocks for hardware descriptors */
285	lchan->pool = dma_pool_create(dev_name(chan2dev(chan)),
286				      chan->device->dev, PAGE_SIZE,
287				      __alignof__(struct ls2x_dma_hw_desc), 0);
288	if (!lchan->pool) {
289		dev_err(chan2dev(chan), "No memory for descriptors\n");
290		return -ENOMEM;
291	}
292
293	return 1;
294}
295
296/*
297 * ls2x_dma_free_chan_resources - free all channel resources
298 * @chan: DMA channel
299 */
300static void ls2x_dma_free_chan_resources(struct dma_chan *chan)
301{
302	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
303
304	vchan_free_chan_resources(to_virt_chan(chan));
305	dma_pool_destroy(lchan->pool);
306	lchan->pool = NULL;
307}
308
309/*
310 * ls2x_dma_prep_slave_sg - prepare descriptors for a DMA_SLAVE transaction
311 * @chan: DMA channel
312 * @sgl: scatterlist to transfer to/from
313 * @sg_len: number of entries in @scatterlist
314 * @direction: DMA direction
315 * @flags: tx descriptor status flags
316 * @context: transaction context (ignored)
317 *
318 * Return: Async transaction descriptor on success and NULL on failure
319 */
320static struct dma_async_tx_descriptor *
321ls2x_dma_prep_slave_sg(struct dma_chan *chan, struct scatterlist *sgl,
322		       u32 sg_len, enum dma_transfer_direction direction,
323		       unsigned long flags, void *context)
324{
325	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
326	struct ls2x_dma_desc *desc;
327	struct scatterlist *sg;
328	size_t burst_size;
329	int i;
330
331	if (unlikely(!sg_len || !is_slave_direction(direction)))
332		return NULL;
333
334	burst_size = ls2x_dmac_detect_burst(lchan);
335	if (!burst_size)
336		return NULL;
337
338	desc = kzalloc(struct_size(desc, sg, sg_len), GFP_NOWAIT);
339	if (!desc)
340		return NULL;
341
342	desc->desc_num = sg_len;
343	desc->direction = direction;
344	desc->burst_size = burst_size;
345
346	for_each_sg(sgl, sg, sg_len, i) {
347		struct ls2x_dma_sg *ldma_sg = &desc->sg[i];
348
349		/* Allocate DMA capable memory for hardware descriptor */
350		ldma_sg->hw = dma_pool_alloc(lchan->pool, GFP_NOWAIT, &ldma_sg->llp);
351		if (!ldma_sg->hw) {
352			desc->desc_num = i;
353			ls2x_dma_desc_free(&desc->vdesc);
354			return NULL;
355		}
356
357		ldma_sg->phys = sg_dma_address(sg);
358		ldma_sg->len = sg_dma_len(sg);
359
360		ls2x_dma_fill_desc(lchan, i, desc);
361	}
362
363	/* Setting the last descriptor enable bit */
364	desc->sg[sg_len - 1].hw->ndesc_addr &= ~LDMA_DESC_EN;
365	desc->status = DMA_IN_PROGRESS;
366
367	return vchan_tx_prep(&lchan->vchan, &desc->vdesc, flags);
368}
369
370/*
371 * ls2x_dma_prep_dma_cyclic - prepare the cyclic DMA transfer
372 * @chan: the DMA channel to prepare
373 * @buf_addr: physical DMA address where the buffer starts
374 * @buf_len: total number of bytes for the entire buffer
375 * @period_len: number of bytes for each period
376 * @direction: transfer direction, to or from device
377 * @flags: tx descriptor status flags
378 *
379 * Return: Async transaction descriptor on success and NULL on failure
380 */
381static struct dma_async_tx_descriptor *
382ls2x_dma_prep_dma_cyclic(struct dma_chan *chan, dma_addr_t buf_addr, size_t buf_len,
383			 size_t period_len, enum dma_transfer_direction direction,
384			 unsigned long flags)
385{
386	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
387	struct ls2x_dma_desc *desc;
388	size_t burst_size;
389	u32 num_periods;
390	int i;
391
392	if (unlikely(!buf_len || !period_len))
393		return NULL;
394
395	if (unlikely(!is_slave_direction(direction)))
396		return NULL;
397
398	burst_size = ls2x_dmac_detect_burst(lchan);
399	if (!burst_size)
400		return NULL;
401
402	num_periods = buf_len / period_len;
403	desc = kzalloc(struct_size(desc, sg, num_periods), GFP_NOWAIT);
404	if (!desc)
405		return NULL;
406
407	desc->desc_num = num_periods;
408	desc->direction = direction;
409	desc->burst_size = burst_size;
410
411	/* Build cyclic linked list */
412	for (i = 0; i < num_periods; i++) {
413		struct ls2x_dma_sg *ldma_sg = &desc->sg[i];
414
415		/* Allocate DMA capable memory for hardware descriptor */
416		ldma_sg->hw = dma_pool_alloc(lchan->pool, GFP_NOWAIT, &ldma_sg->llp);
417		if (!ldma_sg->hw) {
418			desc->desc_num = i;
419			ls2x_dma_desc_free(&desc->vdesc);
420			return NULL;
421		}
422
423		ldma_sg->phys = buf_addr + period_len * i;
424		ldma_sg->len = period_len;
425
426		ls2x_dma_fill_desc(lchan, i, desc);
427	}
428
429	/* Lets make a cyclic list */
430	desc->sg[num_periods - 1].hw->ndesc_addr = desc->sg[0].llp | LDMA_DESC_EN;
431	desc->sg[num_periods - 1].hw->high_ndesc_addr = upper_32_bits(desc->sg[0].llp);
432	desc->cyclic = true;
433	desc->status = DMA_IN_PROGRESS;
434
435	return vchan_tx_prep(&lchan->vchan, &desc->vdesc, flags);
436}
437
438/*
439 * ls2x_slave_config - set slave configuration for channel
440 * @chan: dma channel
441 * @cfg: slave configuration
442 *
443 * Sets slave configuration for channel
444 */
445static int ls2x_dma_slave_config(struct dma_chan *chan,
446				 struct dma_slave_config *config)
447{
448	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
449
450	memcpy(&lchan->sconfig, config, sizeof(*config));
451	return 0;
452}
453
454/*
455 * ls2x_dma_issue_pending - push pending transactions to the hardware
456 * @chan: channel
457 *
458 * When this function is called, all pending transactions are pushed to the
459 * hardware and executed.
460 */
461static void ls2x_dma_issue_pending(struct dma_chan *chan)
462{
463	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
464	unsigned long flags;
465
466	spin_lock_irqsave(&lchan->vchan.lock, flags);
467	if (vchan_issue_pending(&lchan->vchan) && !lchan->desc)
468		ls2x_dma_start_transfer(lchan);
469	spin_unlock_irqrestore(&lchan->vchan.lock, flags);
470}
471
472/*
473 * ls2x_dma_terminate_all - terminate all transactions
474 * @chan: channel
475 *
476 * Stops all DMA transactions.
477 */
478static int ls2x_dma_terminate_all(struct dma_chan *chan)
479{
480	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
481	unsigned long flags;
482	LIST_HEAD(head);
483
484	spin_lock_irqsave(&lchan->vchan.lock, flags);
485	/* Setting stop cmd */
486	ls2x_dma_write_cmd(lchan, LDMA_STOP);
487	if (lchan->desc) {
488		vchan_terminate_vdesc(&lchan->desc->vdesc);
489		lchan->desc = NULL;
490	}
491
492	vchan_get_all_descriptors(&lchan->vchan, &head);
493	spin_unlock_irqrestore(&lchan->vchan.lock, flags);
494
495	vchan_dma_desc_free_list(&lchan->vchan, &head);
496	return 0;
497}
498
499/*
500 * ls2x_dma_synchronize - Synchronizes the termination of transfers to the
501 * current context.
502 * @chan: channel
503 */
504static void ls2x_dma_synchronize(struct dma_chan *chan)
505{
506	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
507
508	vchan_synchronize(&lchan->vchan);
509}
510
511static int ls2x_dma_pause(struct dma_chan *chan)
512{
513	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
514	unsigned long flags;
515
516	spin_lock_irqsave(&lchan->vchan.lock, flags);
517	if (lchan->desc && lchan->desc->status == DMA_IN_PROGRESS) {
518		ls2x_dma_write_cmd(lchan, LDMA_STOP);
519		lchan->desc->status = DMA_PAUSED;
520	}
521	spin_unlock_irqrestore(&lchan->vchan.lock, flags);
522
523	return 0;
524}
525
526static int ls2x_dma_resume(struct dma_chan *chan)
527{
528	struct ls2x_dma_chan *lchan = to_ldma_chan(chan);
529	unsigned long flags;
530
531	spin_lock_irqsave(&lchan->vchan.lock, flags);
532	if (lchan->desc && lchan->desc->status == DMA_PAUSED) {
533		lchan->desc->status = DMA_IN_PROGRESS;
534		ls2x_dma_write_cmd(lchan, LDMA_START);
535	}
536	spin_unlock_irqrestore(&lchan->vchan.lock, flags);
537
538	return 0;
539}
540
541/*
542 * ls2x_dma_isr - LS2X DMA Interrupt handler
543 * @irq: IRQ number
544 * @dev_id: Pointer to ls2x_dma_chan
545 *
546 * Return: IRQ_HANDLED/IRQ_NONE
547 */
548static irqreturn_t ls2x_dma_isr(int irq, void *dev_id)
549{
550	struct ls2x_dma_chan *lchan = dev_id;
551	struct ls2x_dma_desc *desc;
552
553	spin_lock(&lchan->vchan.lock);
554	desc = lchan->desc;
555	if (desc) {
556		if (desc->cyclic) {
557			vchan_cyclic_callback(&desc->vdesc);
558		} else {
559			desc->status = DMA_COMPLETE;
560			vchan_cookie_complete(&desc->vdesc);
561			ls2x_dma_start_transfer(lchan);
562		}
563
564		/* ls2x_dma_start_transfer() updates lchan->desc */
565		if (!lchan->desc)
566			ls2x_dma_write_cmd(lchan, LDMA_STOP);
567	}
568	spin_unlock(&lchan->vchan.lock);
569
570	return IRQ_HANDLED;
571}
572
573static int ls2x_dma_chan_init(struct platform_device *pdev,
574			      struct ls2x_dma_priv *priv)
575{
576	struct ls2x_dma_chan *lchan = &priv->lchan;
577	struct device *dev = &pdev->dev;
578	int ret;
579
580	lchan->irq = platform_get_irq(pdev, 0);
581	if (lchan->irq < 0)
582		return lchan->irq;
583
584	ret = devm_request_irq(dev, lchan->irq, ls2x_dma_isr, IRQF_TRIGGER_RISING,
585			       dev_name(&pdev->dev), lchan);
586	if (ret)
587		return ret;
588
589	/* Initialize channels related values */
590	INIT_LIST_HEAD(&priv->ddev.channels);
591	lchan->vchan.desc_free = ls2x_dma_desc_free;
592	vchan_init(&lchan->vchan, &priv->ddev);
593
594	return 0;
595}
596
597/*
598 * ls2x_dma_probe - Driver probe function
599 * @pdev: Pointer to the platform_device structure
600 *
601 * Return: '0' on success and failure value on error
602 */
603static int ls2x_dma_probe(struct platform_device *pdev)
604{
605	struct device *dev = &pdev->dev;
606	struct ls2x_dma_priv *priv;
607	struct dma_device *ddev;
608	int ret;
609
610	priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
611	if (!priv)
612		return -ENOMEM;
613
614	priv->regs = devm_platform_ioremap_resource(pdev, 0);
615	if (IS_ERR(priv->regs))
616		return dev_err_probe(dev, PTR_ERR(priv->regs),
617				     "devm_platform_ioremap_resource failed.\n");
618
619	priv->dma_clk = devm_clk_get(&pdev->dev, NULL);
620	if (IS_ERR(priv->dma_clk))
621		return dev_err_probe(dev, PTR_ERR(priv->dma_clk), "devm_clk_get failed.\n");
622
623	ret = clk_prepare_enable(priv->dma_clk);
624	if (ret)
625		return dev_err_probe(dev, ret, "clk_prepare_enable failed.\n");
626
627	ret = ls2x_dma_chan_init(pdev, priv);
628	if (ret)
629		goto disable_clk;
630
631	ddev = &priv->ddev;
632	ddev->dev = dev;
633	dma_cap_zero(ddev->cap_mask);
634	dma_cap_set(DMA_SLAVE, ddev->cap_mask);
635	dma_cap_set(DMA_CYCLIC, ddev->cap_mask);
636
637	ddev->device_alloc_chan_resources = ls2x_dma_alloc_chan_resources;
638	ddev->device_free_chan_resources = ls2x_dma_free_chan_resources;
639	ddev->device_tx_status = dma_cookie_status;
640	ddev->device_issue_pending = ls2x_dma_issue_pending;
641	ddev->device_prep_slave_sg = ls2x_dma_prep_slave_sg;
642	ddev->device_prep_dma_cyclic = ls2x_dma_prep_dma_cyclic;
643	ddev->device_config = ls2x_dma_slave_config;
644	ddev->device_terminate_all = ls2x_dma_terminate_all;
645	ddev->device_synchronize = ls2x_dma_synchronize;
646	ddev->device_pause = ls2x_dma_pause;
647	ddev->device_resume = ls2x_dma_resume;
648
649	ddev->src_addr_widths = LDMA_SLAVE_BUSWIDTHS;
650	ddev->dst_addr_widths = LDMA_SLAVE_BUSWIDTHS;
651	ddev->directions = BIT(DMA_DEV_TO_MEM) | BIT(DMA_MEM_TO_DEV);
652
653	ret = dma_async_device_register(&priv->ddev);
654	if (ret < 0)
655		goto disable_clk;
656
657	ret = of_dma_controller_register(dev->of_node, of_dma_xlate_by_chan_id, priv);
658	if (ret < 0)
659		goto unregister_dmac;
660
661	platform_set_drvdata(pdev, priv);
662
663	dev_info(dev, "Loongson LS2X APB DMA driver registered successfully.\n");
664	return 0;
665
666unregister_dmac:
667	dma_async_device_unregister(&priv->ddev);
668disable_clk:
669	clk_disable_unprepare(priv->dma_clk);
670
671	return ret;
672}
673
674/*
675 * ls2x_dma_remove - Driver remove function
676 * @pdev: Pointer to the platform_device structure
677 */
678static void ls2x_dma_remove(struct platform_device *pdev)
679{
680	struct ls2x_dma_priv *priv = platform_get_drvdata(pdev);
681
682	of_dma_controller_free(pdev->dev.of_node);
683	dma_async_device_unregister(&priv->ddev);
684	clk_disable_unprepare(priv->dma_clk);
685}
686
687static const struct of_device_id ls2x_dma_of_match_table[] = {
688	{ .compatible = "loongson,ls2k1000-apbdma" },
689	{ /* sentinel */ }
690};
691MODULE_DEVICE_TABLE(of, ls2x_dma_of_match_table);
692
693static struct platform_driver ls2x_dmac_driver = {
694	.probe		= ls2x_dma_probe,
695	.remove_new	= ls2x_dma_remove,
696	.driver = {
697		.name	= "ls2x-apbdma",
698		.of_match_table	= ls2x_dma_of_match_table,
699	},
700};
701module_platform_driver(ls2x_dmac_driver);
702
703MODULE_DESCRIPTION("Loongson LS2X APB DMA Controller driver");
704MODULE_AUTHOR("Loongson Technology Corporation Limited");
705MODULE_LICENSE("GPL");
706