drivers/dma/k3dma.c

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