x86/smpboot: Init apic mapping before usage
[cascardo/linux.git] / drivers / dma / tegra210-adma.c
1 /*
2  * ADMA driver for Nvidia's Tegra210 ADMA controller.
3  *
4  * Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms and conditions of the GNU General Public License,
8  * version 2, as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope it will be useful, but WITHOUT
11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
13  * more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
17  */
18
19 #include <linux/clk.h>
20 #include <linux/iopoll.h>
21 #include <linux/module.h>
22 #include <linux/of_device.h>
23 #include <linux/of_dma.h>
24 #include <linux/of_irq.h>
25 #include <linux/pm_clock.h>
26 #include <linux/pm_runtime.h>
27 #include <linux/slab.h>
28
29 #include "virt-dma.h"
30
31 #define ADMA_CH_CMD                                     0x00
32 #define ADMA_CH_STATUS                                  0x0c
33 #define ADMA_CH_STATUS_XFER_EN                          BIT(0)
34
35 #define ADMA_CH_INT_STATUS                              0x10
36 #define ADMA_CH_INT_STATUS_XFER_DONE                    BIT(0)
37
38 #define ADMA_CH_INT_CLEAR                               0x1c
39 #define ADMA_CH_CTRL                                    0x24
40 #define ADMA_CH_CTRL_TX_REQ(val)                        (((val) & 0xf) << 28)
41 #define ADMA_CH_CTRL_TX_REQ_MAX                         10
42 #define ADMA_CH_CTRL_RX_REQ(val)                        (((val) & 0xf) << 24)
43 #define ADMA_CH_CTRL_RX_REQ_MAX                         10
44 #define ADMA_CH_CTRL_DIR(val)                           (((val) & 0xf) << 12)
45 #define ADMA_CH_CTRL_DIR_AHUB2MEM                       2
46 #define ADMA_CH_CTRL_DIR_MEM2AHUB                       4
47 #define ADMA_CH_CTRL_MODE_CONTINUOUS                    (2 << 8)
48 #define ADMA_CH_CTRL_FLOWCTRL_EN                        BIT(1)
49
50 #define ADMA_CH_CONFIG                                  0x28
51 #define ADMA_CH_CONFIG_SRC_BUF(val)                     (((val) & 0x7) << 28)
52 #define ADMA_CH_CONFIG_TRG_BUF(val)                     (((val) & 0x7) << 24)
53 #define ADMA_CH_CONFIG_BURST_SIZE(val)                  (((val) & 0x7) << 20)
54 #define ADMA_CH_CONFIG_BURST_16                         5
55 #define ADMA_CH_CONFIG_WEIGHT_FOR_WRR(val)              ((val) & 0xf)
56 #define ADMA_CH_CONFIG_MAX_BUFS                         8
57
58 #define ADMA_CH_FIFO_CTRL                               0x2c
59 #define ADMA_CH_FIFO_CTRL_OVRFW_THRES(val)              (((val) & 0xf) << 24)
60 #define ADMA_CH_FIFO_CTRL_STARV_THRES(val)              (((val) & 0xf) << 16)
61 #define ADMA_CH_FIFO_CTRL_TX_SIZE(val)                  (((val) & 0xf) << 8)
62 #define ADMA_CH_FIFO_CTRL_RX_SIZE(val)                  ((val) & 0xf)
63
64 #define ADMA_CH_LOWER_SRC_ADDR                          0x34
65 #define ADMA_CH_LOWER_TRG_ADDR                          0x3c
66 #define ADMA_CH_TC                                      0x44
67 #define ADMA_CH_TC_COUNT_MASK                           0x3ffffffc
68
69 #define ADMA_CH_XFER_STATUS                             0x54
70 #define ADMA_CH_XFER_STATUS_COUNT_MASK                  0xffff
71
72 #define ADMA_GLOBAL_CMD                                 0xc00
73 #define ADMA_GLOBAL_SOFT_RESET                          0xc04
74 #define ADMA_GLOBAL_INT_CLEAR                           0xc20
75 #define ADMA_GLOBAL_CTRL                                0xc24
76
77 #define ADMA_CH_REG_OFFSET(a)                           (a * 0x80)
78
79 #define ADMA_CH_FIFO_CTRL_DEFAULT       (ADMA_CH_FIFO_CTRL_OVRFW_THRES(1) | \
80                                          ADMA_CH_FIFO_CTRL_STARV_THRES(1) | \
81                                          ADMA_CH_FIFO_CTRL_TX_SIZE(3)     | \
82                                          ADMA_CH_FIFO_CTRL_RX_SIZE(3))
83 struct tegra_adma;
84
85 /*
86  * struct tegra_adma_chip_data - Tegra chip specific data
87  * @nr_channels: Number of DMA channels available.
88  */
89 struct tegra_adma_chip_data {
90         int nr_channels;
91 };
92
93 /*
94  * struct tegra_adma_chan_regs - Tegra ADMA channel registers
95  */
96 struct tegra_adma_chan_regs {
97         unsigned int ctrl;
98         unsigned int config;
99         unsigned int src_addr;
100         unsigned int trg_addr;
101         unsigned int fifo_ctrl;
102         unsigned int tc;
103 };
104
105 /*
106  * struct tegra_adma_desc - Tegra ADMA descriptor to manage transfer requests.
107  */
108 struct tegra_adma_desc {
109         struct virt_dma_desc            vd;
110         struct tegra_adma_chan_regs     ch_regs;
111         size_t                          buf_len;
112         size_t                          period_len;
113         size_t                          num_periods;
114 };
115
116 /*
117  * struct tegra_adma_chan - Tegra ADMA channel information
118  */
119 struct tegra_adma_chan {
120         struct virt_dma_chan            vc;
121         struct tegra_adma_desc          *desc;
122         struct tegra_adma               *tdma;
123         int                             irq;
124         void __iomem                    *chan_addr;
125
126         /* Slave channel configuration info */
127         struct dma_slave_config         sconfig;
128         enum dma_transfer_direction     sreq_dir;
129         unsigned int                    sreq_index;
130         bool                            sreq_reserved;
131
132         /* Transfer count and position info */
133         unsigned int                    tx_buf_count;
134         unsigned int                    tx_buf_pos;
135 };
136
137 /*
138  * struct tegra_adma - Tegra ADMA controller information
139  */
140 struct tegra_adma {
141         struct dma_device               dma_dev;
142         struct device                   *dev;
143         void __iomem                    *base_addr;
144         unsigned int                    nr_channels;
145         unsigned long                   rx_requests_reserved;
146         unsigned long                   tx_requests_reserved;
147
148         /* Used to store global command register state when suspending */
149         unsigned int                    global_cmd;
150
151         /* Last member of the structure */
152         struct tegra_adma_chan          channels[0];
153 };
154
155 static inline void tdma_write(struct tegra_adma *tdma, u32 reg, u32 val)
156 {
157         writel(val, tdma->base_addr + reg);
158 }
159
160 static inline u32 tdma_read(struct tegra_adma *tdma, u32 reg)
161 {
162         return readl(tdma->base_addr + reg);
163 }
164
165 static inline void tdma_ch_write(struct tegra_adma_chan *tdc, u32 reg, u32 val)
166 {
167         writel(val, tdc->chan_addr + reg);
168 }
169
170 static inline u32 tdma_ch_read(struct tegra_adma_chan *tdc, u32 reg)
171 {
172         return readl(tdc->chan_addr + reg);
173 }
174
175 static inline struct tegra_adma_chan *to_tegra_adma_chan(struct dma_chan *dc)
176 {
177         return container_of(dc, struct tegra_adma_chan, vc.chan);
178 }
179
180 static inline struct tegra_adma_desc *to_tegra_adma_desc(
181                 struct dma_async_tx_descriptor *td)
182 {
183         return container_of(td, struct tegra_adma_desc, vd.tx);
184 }
185
186 static inline struct device *tdc2dev(struct tegra_adma_chan *tdc)
187 {
188         return tdc->tdma->dev;
189 }
190
191 static void tegra_adma_desc_free(struct virt_dma_desc *vd)
192 {
193         kfree(container_of(vd, struct tegra_adma_desc, vd));
194 }
195
196 static int tegra_adma_slave_config(struct dma_chan *dc,
197                                    struct dma_slave_config *sconfig)
198 {
199         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
200
201         memcpy(&tdc->sconfig, sconfig, sizeof(*sconfig));
202
203         return 0;
204 }
205
206 static int tegra_adma_init(struct tegra_adma *tdma)
207 {
208         u32 status;
209         int ret;
210
211         /* Clear any interrupts */
212         tdma_write(tdma, ADMA_GLOBAL_INT_CLEAR, 0x1);
213
214         /* Assert soft reset */
215         tdma_write(tdma, ADMA_GLOBAL_SOFT_RESET, 0x1);
216
217         /* Wait for reset to clear */
218         ret = readx_poll_timeout(readl,
219                                  tdma->base_addr + ADMA_GLOBAL_SOFT_RESET,
220                                  status, status == 0, 20, 10000);
221         if (ret)
222                 return ret;
223
224         /* Enable global ADMA registers */
225         tdma_write(tdma, ADMA_GLOBAL_CMD, 1);
226
227         return 0;
228 }
229
230 static int tegra_adma_request_alloc(struct tegra_adma_chan *tdc,
231                                     enum dma_transfer_direction direction)
232 {
233         struct tegra_adma *tdma = tdc->tdma;
234         unsigned int sreq_index = tdc->sreq_index;
235
236         if (tdc->sreq_reserved)
237                 return tdc->sreq_dir == direction ? 0 : -EINVAL;
238
239         switch (direction) {
240         case DMA_MEM_TO_DEV:
241                 if (sreq_index > ADMA_CH_CTRL_TX_REQ_MAX) {
242                         dev_err(tdma->dev, "invalid DMA request\n");
243                         return -EINVAL;
244                 }
245
246                 if (test_and_set_bit(sreq_index, &tdma->tx_requests_reserved)) {
247                         dev_err(tdma->dev, "DMA request reserved\n");
248                         return -EINVAL;
249                 }
250                 break;
251
252         case DMA_DEV_TO_MEM:
253                 if (sreq_index > ADMA_CH_CTRL_RX_REQ_MAX) {
254                         dev_err(tdma->dev, "invalid DMA request\n");
255                         return -EINVAL;
256                 }
257
258                 if (test_and_set_bit(sreq_index, &tdma->rx_requests_reserved)) {
259                         dev_err(tdma->dev, "DMA request reserved\n");
260                         return -EINVAL;
261                 }
262                 break;
263
264         default:
265                 dev_WARN(tdma->dev, "channel %s has invalid transfer type\n",
266                          dma_chan_name(&tdc->vc.chan));
267                 return -EINVAL;
268         }
269
270         tdc->sreq_dir = direction;
271         tdc->sreq_reserved = true;
272
273         return 0;
274 }
275
276 static void tegra_adma_request_free(struct tegra_adma_chan *tdc)
277 {
278         struct tegra_adma *tdma = tdc->tdma;
279
280         if (!tdc->sreq_reserved)
281                 return;
282
283         switch (tdc->sreq_dir) {
284         case DMA_MEM_TO_DEV:
285                 clear_bit(tdc->sreq_index, &tdma->tx_requests_reserved);
286                 break;
287
288         case DMA_DEV_TO_MEM:
289                 clear_bit(tdc->sreq_index, &tdma->rx_requests_reserved);
290                 break;
291
292         default:
293                 dev_WARN(tdma->dev, "channel %s has invalid transfer type\n",
294                          dma_chan_name(&tdc->vc.chan));
295                 return;
296         }
297
298         tdc->sreq_reserved = false;
299 }
300
301 static u32 tegra_adma_irq_status(struct tegra_adma_chan *tdc)
302 {
303         u32 status = tdma_ch_read(tdc, ADMA_CH_INT_STATUS);
304
305         return status & ADMA_CH_INT_STATUS_XFER_DONE;
306 }
307
308 static u32 tegra_adma_irq_clear(struct tegra_adma_chan *tdc)
309 {
310         u32 status = tegra_adma_irq_status(tdc);
311
312         if (status)
313                 tdma_ch_write(tdc, ADMA_CH_INT_CLEAR, status);
314
315         return status;
316 }
317
318 static void tegra_adma_stop(struct tegra_adma_chan *tdc)
319 {
320         unsigned int status;
321
322         /* Disable ADMA */
323         tdma_ch_write(tdc, ADMA_CH_CMD, 0);
324
325         /* Clear interrupt status */
326         tegra_adma_irq_clear(tdc);
327
328         if (readx_poll_timeout_atomic(readl, tdc->chan_addr + ADMA_CH_STATUS,
329                         status, !(status & ADMA_CH_STATUS_XFER_EN),
330                         20, 10000)) {
331                 dev_err(tdc2dev(tdc), "unable to stop DMA channel\n");
332                 return;
333         }
334
335         kfree(tdc->desc);
336         tdc->desc = NULL;
337 }
338
339 static void tegra_adma_start(struct tegra_adma_chan *tdc)
340 {
341         struct virt_dma_desc *vd = vchan_next_desc(&tdc->vc);
342         struct tegra_adma_chan_regs *ch_regs;
343         struct tegra_adma_desc *desc;
344
345         if (!vd)
346                 return;
347
348         list_del(&vd->node);
349
350         desc = to_tegra_adma_desc(&vd->tx);
351
352         if (!desc) {
353                 dev_warn(tdc2dev(tdc), "unable to start DMA, no descriptor\n");
354                 return;
355         }
356
357         ch_regs = &desc->ch_regs;
358
359         tdc->tx_buf_pos = 0;
360         tdc->tx_buf_count = 0;
361         tdma_ch_write(tdc, ADMA_CH_TC, ch_regs->tc);
362         tdma_ch_write(tdc, ADMA_CH_CTRL, ch_regs->ctrl);
363         tdma_ch_write(tdc, ADMA_CH_LOWER_SRC_ADDR, ch_regs->src_addr);
364         tdma_ch_write(tdc, ADMA_CH_LOWER_TRG_ADDR, ch_regs->trg_addr);
365         tdma_ch_write(tdc, ADMA_CH_FIFO_CTRL, ch_regs->fifo_ctrl);
366         tdma_ch_write(tdc, ADMA_CH_CONFIG, ch_regs->config);
367
368         /* Start ADMA */
369         tdma_ch_write(tdc, ADMA_CH_CMD, 1);
370
371         tdc->desc = desc;
372 }
373
374 static unsigned int tegra_adma_get_residue(struct tegra_adma_chan *tdc)
375 {
376         struct tegra_adma_desc *desc = tdc->desc;
377         unsigned int max = ADMA_CH_XFER_STATUS_COUNT_MASK + 1;
378         unsigned int pos = tdma_ch_read(tdc, ADMA_CH_XFER_STATUS);
379         unsigned int periods_remaining;
380
381         /*
382          * Handle wrap around of buffer count register
383          */
384         if (pos < tdc->tx_buf_pos)
385                 tdc->tx_buf_count += pos + (max - tdc->tx_buf_pos);
386         else
387                 tdc->tx_buf_count += pos - tdc->tx_buf_pos;
388
389         periods_remaining = tdc->tx_buf_count % desc->num_periods;
390         tdc->tx_buf_pos = pos;
391
392         return desc->buf_len - (periods_remaining * desc->period_len);
393 }
394
395 static irqreturn_t tegra_adma_isr(int irq, void *dev_id)
396 {
397         struct tegra_adma_chan *tdc = dev_id;
398         unsigned long status;
399         unsigned long flags;
400
401         spin_lock_irqsave(&tdc->vc.lock, flags);
402
403         status = tegra_adma_irq_clear(tdc);
404         if (status == 0 || !tdc->desc) {
405                 spin_unlock_irqrestore(&tdc->vc.lock, flags);
406                 return IRQ_NONE;
407         }
408
409         vchan_cyclic_callback(&tdc->desc->vd);
410
411         spin_unlock_irqrestore(&tdc->vc.lock, flags);
412
413         return IRQ_HANDLED;
414 }
415
416 static void tegra_adma_issue_pending(struct dma_chan *dc)
417 {
418         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
419         unsigned long flags;
420
421         spin_lock_irqsave(&tdc->vc.lock, flags);
422
423         if (vchan_issue_pending(&tdc->vc)) {
424                 if (!tdc->desc)
425                         tegra_adma_start(tdc);
426         }
427
428         spin_unlock_irqrestore(&tdc->vc.lock, flags);
429 }
430
431 static int tegra_adma_terminate_all(struct dma_chan *dc)
432 {
433         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
434         unsigned long flags;
435         LIST_HEAD(head);
436
437         spin_lock_irqsave(&tdc->vc.lock, flags);
438
439         if (tdc->desc)
440                 tegra_adma_stop(tdc);
441
442         tegra_adma_request_free(tdc);
443         vchan_get_all_descriptors(&tdc->vc, &head);
444         spin_unlock_irqrestore(&tdc->vc.lock, flags);
445         vchan_dma_desc_free_list(&tdc->vc, &head);
446
447         return 0;
448 }
449
450 static enum dma_status tegra_adma_tx_status(struct dma_chan *dc,
451                                             dma_cookie_t cookie,
452                                             struct dma_tx_state *txstate)
453 {
454         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
455         struct tegra_adma_desc *desc;
456         struct virt_dma_desc *vd;
457         enum dma_status ret;
458         unsigned long flags;
459         unsigned int residual;
460
461         ret = dma_cookie_status(dc, cookie, txstate);
462         if (ret == DMA_COMPLETE || !txstate)
463                 return ret;
464
465         spin_lock_irqsave(&tdc->vc.lock, flags);
466
467         vd = vchan_find_desc(&tdc->vc, cookie);
468         if (vd) {
469                 desc = to_tegra_adma_desc(&vd->tx);
470                 residual = desc->ch_regs.tc;
471         } else if (tdc->desc && tdc->desc->vd.tx.cookie == cookie) {
472                 residual = tegra_adma_get_residue(tdc);
473         } else {
474                 residual = 0;
475         }
476
477         spin_unlock_irqrestore(&tdc->vc.lock, flags);
478
479         dma_set_residue(txstate, residual);
480
481         return ret;
482 }
483
484 static int tegra_adma_set_xfer_params(struct tegra_adma_chan *tdc,
485                                       struct tegra_adma_desc *desc,
486                                       dma_addr_t buf_addr,
487                                       enum dma_transfer_direction direction)
488 {
489         struct tegra_adma_chan_regs *ch_regs = &desc->ch_regs;
490         unsigned int burst_size, adma_dir;
491
492         if (desc->num_periods > ADMA_CH_CONFIG_MAX_BUFS)
493                 return -EINVAL;
494
495         switch (direction) {
496         case DMA_MEM_TO_DEV:
497                 adma_dir = ADMA_CH_CTRL_DIR_MEM2AHUB;
498                 burst_size = fls(tdc->sconfig.dst_maxburst);
499                 ch_regs->config = ADMA_CH_CONFIG_SRC_BUF(desc->num_periods - 1);
500                 ch_regs->ctrl = ADMA_CH_CTRL_TX_REQ(tdc->sreq_index);
501                 ch_regs->src_addr = buf_addr;
502                 break;
503
504         case DMA_DEV_TO_MEM:
505                 adma_dir = ADMA_CH_CTRL_DIR_AHUB2MEM;
506                 burst_size = fls(tdc->sconfig.src_maxburst);
507                 ch_regs->config = ADMA_CH_CONFIG_TRG_BUF(desc->num_periods - 1);
508                 ch_regs->ctrl = ADMA_CH_CTRL_RX_REQ(tdc->sreq_index);
509                 ch_regs->trg_addr = buf_addr;
510                 break;
511
512         default:
513                 dev_err(tdc2dev(tdc), "DMA direction is not supported\n");
514                 return -EINVAL;
515         }
516
517         if (!burst_size || burst_size > ADMA_CH_CONFIG_BURST_16)
518                 burst_size = ADMA_CH_CONFIG_BURST_16;
519
520         ch_regs->ctrl |= ADMA_CH_CTRL_DIR(adma_dir) |
521                          ADMA_CH_CTRL_MODE_CONTINUOUS |
522                          ADMA_CH_CTRL_FLOWCTRL_EN;
523         ch_regs->config |= ADMA_CH_CONFIG_BURST_SIZE(burst_size);
524         ch_regs->config |= ADMA_CH_CONFIG_WEIGHT_FOR_WRR(1);
525         ch_regs->fifo_ctrl = ADMA_CH_FIFO_CTRL_DEFAULT;
526         ch_regs->tc = desc->period_len & ADMA_CH_TC_COUNT_MASK;
527
528         return tegra_adma_request_alloc(tdc, direction);
529 }
530
531 static struct dma_async_tx_descriptor *tegra_adma_prep_dma_cyclic(
532         struct dma_chan *dc, dma_addr_t buf_addr, size_t buf_len,
533         size_t period_len, enum dma_transfer_direction direction,
534         unsigned long flags)
535 {
536         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
537         struct tegra_adma_desc *desc = NULL;
538
539         if (!buf_len || !period_len || period_len > ADMA_CH_TC_COUNT_MASK) {
540                 dev_err(tdc2dev(tdc), "invalid buffer/period len\n");
541                 return NULL;
542         }
543
544         if (buf_len % period_len) {
545                 dev_err(tdc2dev(tdc), "buf_len not a multiple of period_len\n");
546                 return NULL;
547         }
548
549         if (!IS_ALIGNED(buf_addr, 4)) {
550                 dev_err(tdc2dev(tdc), "invalid buffer alignment\n");
551                 return NULL;
552         }
553
554         desc = kzalloc(sizeof(*desc), GFP_NOWAIT);
555         if (!desc)
556                 return NULL;
557
558         desc->buf_len = buf_len;
559         desc->period_len = period_len;
560         desc->num_periods = buf_len / period_len;
561
562         if (tegra_adma_set_xfer_params(tdc, desc, buf_addr, direction)) {
563                 kfree(desc);
564                 return NULL;
565         }
566
567         return vchan_tx_prep(&tdc->vc, &desc->vd, flags);
568 }
569
570 static int tegra_adma_alloc_chan_resources(struct dma_chan *dc)
571 {
572         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
573         int ret;
574
575         ret = request_irq(tdc->irq, tegra_adma_isr, 0, dma_chan_name(dc), tdc);
576         if (ret) {
577                 dev_err(tdc2dev(tdc), "failed to get interrupt for %s\n",
578                         dma_chan_name(dc));
579                 return ret;
580         }
581
582         ret = pm_runtime_get_sync(tdc2dev(tdc));
583         if (ret < 0) {
584                 free_irq(tdc->irq, tdc);
585                 return ret;
586         }
587
588         dma_cookie_init(&tdc->vc.chan);
589
590         return 0;
591 }
592
593 static void tegra_adma_free_chan_resources(struct dma_chan *dc)
594 {
595         struct tegra_adma_chan *tdc = to_tegra_adma_chan(dc);
596
597         tegra_adma_terminate_all(dc);
598         vchan_free_chan_resources(&tdc->vc);
599         tasklet_kill(&tdc->vc.task);
600         free_irq(tdc->irq, tdc);
601         pm_runtime_put(tdc2dev(tdc));
602
603         tdc->sreq_index = 0;
604         tdc->sreq_dir = DMA_TRANS_NONE;
605 }
606
607 static struct dma_chan *tegra_dma_of_xlate(struct of_phandle_args *dma_spec,
608                                            struct of_dma *ofdma)
609 {
610         struct tegra_adma *tdma = ofdma->of_dma_data;
611         struct tegra_adma_chan *tdc;
612         struct dma_chan *chan;
613         unsigned int sreq_index;
614
615         if (dma_spec->args_count != 1)
616                 return NULL;
617
618         sreq_index = dma_spec->args[0];
619
620         if (sreq_index == 0) {
621                 dev_err(tdma->dev, "DMA request must not be 0\n");
622                 return NULL;
623         }
624
625         chan = dma_get_any_slave_channel(&tdma->dma_dev);
626         if (!chan)
627                 return NULL;
628
629         tdc = to_tegra_adma_chan(chan);
630         tdc->sreq_index = sreq_index;
631
632         return chan;
633 }
634
635 static int tegra_adma_runtime_suspend(struct device *dev)
636 {
637         struct tegra_adma *tdma = dev_get_drvdata(dev);
638
639         tdma->global_cmd = tdma_read(tdma, ADMA_GLOBAL_CMD);
640
641         return pm_clk_suspend(dev);
642 }
643
644 static int tegra_adma_runtime_resume(struct device *dev)
645 {
646         struct tegra_adma *tdma = dev_get_drvdata(dev);
647         int ret;
648
649         ret = pm_clk_resume(dev);
650         if (ret)
651                 return ret;
652
653         tdma_write(tdma, ADMA_GLOBAL_CMD, tdma->global_cmd);
654
655         return 0;
656 }
657
658 static const struct tegra_adma_chip_data tegra210_chip_data = {
659         .nr_channels = 22,
660 };
661
662 static const struct of_device_id tegra_adma_of_match[] = {
663         { .compatible = "nvidia,tegra210-adma", .data = &tegra210_chip_data },
664         { },
665 };
666 MODULE_DEVICE_TABLE(of, tegra_adma_of_match);
667
668 static int tegra_adma_probe(struct platform_device *pdev)
669 {
670         const struct tegra_adma_chip_data *cdata;
671         struct tegra_adma *tdma;
672         struct resource *res;
673         int ret, i;
674
675         cdata = of_device_get_match_data(&pdev->dev);
676         if (!cdata) {
677                 dev_err(&pdev->dev, "device match data not found\n");
678                 return -ENODEV;
679         }
680
681         tdma = devm_kzalloc(&pdev->dev, sizeof(*tdma) + cdata->nr_channels *
682                             sizeof(struct tegra_adma_chan), GFP_KERNEL);
683         if (!tdma)
684                 return -ENOMEM;
685
686         tdma->dev = &pdev->dev;
687         tdma->nr_channels = cdata->nr_channels;
688         platform_set_drvdata(pdev, tdma);
689
690         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
691         tdma->base_addr = devm_ioremap_resource(&pdev->dev, res);
692         if (IS_ERR(tdma->base_addr))
693                 return PTR_ERR(tdma->base_addr);
694
695         ret = pm_clk_create(&pdev->dev);
696         if (ret)
697                 return ret;
698
699         ret = of_pm_clk_add_clk(&pdev->dev, "d_audio");
700         if (ret)
701                 goto clk_destroy;
702
703         pm_runtime_enable(&pdev->dev);
704
705         ret = pm_runtime_get_sync(&pdev->dev);
706         if (ret < 0)
707                 goto rpm_disable;
708
709         ret = tegra_adma_init(tdma);
710         if (ret)
711                 goto rpm_put;
712
713         INIT_LIST_HEAD(&tdma->dma_dev.channels);
714         for (i = 0; i < tdma->nr_channels; i++) {
715                 struct tegra_adma_chan *tdc = &tdma->channels[i];
716
717                 tdc->chan_addr = tdma->base_addr + ADMA_CH_REG_OFFSET(i);
718
719                 tdc->irq = of_irq_get(pdev->dev.of_node, i);
720                 if (tdc->irq < 0) {
721                         ret = tdc->irq;
722                         goto irq_dispose;
723                 }
724
725                 vchan_init(&tdc->vc, &tdma->dma_dev);
726                 tdc->vc.desc_free = tegra_adma_desc_free;
727                 tdc->tdma = tdma;
728         }
729
730         dma_cap_set(DMA_SLAVE, tdma->dma_dev.cap_mask);
731         dma_cap_set(DMA_PRIVATE, tdma->dma_dev.cap_mask);
732         dma_cap_set(DMA_CYCLIC, tdma->dma_dev.cap_mask);
733
734         tdma->dma_dev.dev = &pdev->dev;
735         tdma->dma_dev.device_alloc_chan_resources =
736                                         tegra_adma_alloc_chan_resources;
737         tdma->dma_dev.device_free_chan_resources =
738                                         tegra_adma_free_chan_resources;
739         tdma->dma_dev.device_issue_pending = tegra_adma_issue_pending;
740         tdma->dma_dev.device_prep_dma_cyclic = tegra_adma_prep_dma_cyclic;
741         tdma->dma_dev.device_config = tegra_adma_slave_config;
742         tdma->dma_dev.device_tx_status = tegra_adma_tx_status;
743         tdma->dma_dev.device_terminate_all = tegra_adma_terminate_all;
744         tdma->dma_dev.src_addr_widths = BIT(DMA_SLAVE_BUSWIDTH_4_BYTES);
745         tdma->dma_dev.dst_addr_widths = BIT(DMA_SLAVE_BUSWIDTH_4_BYTES);
746         tdma->dma_dev.directions = BIT(DMA_DEV_TO_MEM) | BIT(DMA_MEM_TO_DEV);
747         tdma->dma_dev.residue_granularity = DMA_RESIDUE_GRANULARITY_SEGMENT;
748
749         ret = dma_async_device_register(&tdma->dma_dev);
750         if (ret < 0) {
751                 dev_err(&pdev->dev, "ADMA registration failed: %d\n", ret);
752                 goto irq_dispose;
753         }
754
755         ret = of_dma_controller_register(pdev->dev.of_node,
756                                          tegra_dma_of_xlate, tdma);
757         if (ret < 0) {
758                 dev_err(&pdev->dev, "ADMA OF registration failed %d\n", ret);
759                 goto dma_remove;
760         }
761
762         pm_runtime_put(&pdev->dev);
763
764         dev_info(&pdev->dev, "Tegra210 ADMA driver registered %d channels\n",
765                  tdma->nr_channels);
766
767         return 0;
768
769 dma_remove:
770         dma_async_device_unregister(&tdma->dma_dev);
771 irq_dispose:
772         while (--i >= 0)
773                 irq_dispose_mapping(tdma->channels[i].irq);
774 rpm_put:
775         pm_runtime_put_sync(&pdev->dev);
776 rpm_disable:
777         pm_runtime_disable(&pdev->dev);
778 clk_destroy:
779         pm_clk_destroy(&pdev->dev);
780
781         return ret;
782 }
783
784 static int tegra_adma_remove(struct platform_device *pdev)
785 {
786         struct tegra_adma *tdma = platform_get_drvdata(pdev);
787         int i;
788
789         dma_async_device_unregister(&tdma->dma_dev);
790
791         for (i = 0; i < tdma->nr_channels; ++i)
792                 irq_dispose_mapping(tdma->channels[i].irq);
793
794         pm_runtime_put_sync(&pdev->dev);
795         pm_runtime_disable(&pdev->dev);
796         pm_clk_destroy(&pdev->dev);
797
798         return 0;
799 }
800
801 #ifdef CONFIG_PM_SLEEP
802 static int tegra_adma_pm_suspend(struct device *dev)
803 {
804         return pm_runtime_suspended(dev) == false;
805 }
806 #endif
807
808 static const struct dev_pm_ops tegra_adma_dev_pm_ops = {
809         SET_RUNTIME_PM_OPS(tegra_adma_runtime_suspend,
810                            tegra_adma_runtime_resume, NULL)
811         SET_SYSTEM_SLEEP_PM_OPS(tegra_adma_pm_suspend, NULL)
812 };
813
814 static struct platform_driver tegra_admac_driver = {
815         .driver = {
816                 .name   = "tegra-adma",
817                 .pm     = &tegra_adma_dev_pm_ops,
818                 .of_match_table = tegra_adma_of_match,
819         },
820         .probe          = tegra_adma_probe,
821         .remove         = tegra_adma_remove,
822 };
823
824 module_platform_driver(tegra_admac_driver);
825
826 MODULE_ALIAS("platform:tegra210-adma");
827 MODULE_DESCRIPTION("NVIDIA Tegra ADMA driver");
828 MODULE_AUTHOR("Dara Ramesh <dramesh@nvidia.com>");
829 MODULE_AUTHOR("Jon Hunter <jonathanh@nvidia.com>");
830 MODULE_LICENSE("GPL v2");