2 files changed, 21 insertions, 0 deletions

diff --git a/drivers/dma/sh/shdma-base.c b/drivers/dma/sh/shdma-base.c
index 27f5c781fd73..f4cd946d259d 100644
--- a/drivers/dma/sh/shdma-base.c
+++ b/drivers/dma/sh/shdma-base.c
@@ -483,6 +483,7 @@ static struct shdma_desc *shdma_add_desc(struct shdma_chan *schan,
         new->mark = DESC_PREPARED;
         new->async_tx.flags = flags;
         new->direction = direction;
+        new->partial = 0;
 
         *len -= copy_size;
         if (direction == DMA_MEM_TO_MEM || direction == DMA_MEM_TO_DEV)
@@ -644,6 +645,14 @@ static int shdma_control(struct dma_chan *chan, enum dma_ctrl_cmd cmd,
         case DMA_TERMINATE_ALL:
                 spin_lock_irqsave(&schan->chan_lock, flags);
                 ops->halt_channel(schan);
+
+                if (ops->get_partial && !list_empty(&schan->ld_queue)) {
+                        /* Record partial transfer */
+                        struct shdma_desc *desc = list_first_entry(&schan->ld_queue,
+                                                struct shdma_desc, node);
+                        desc->partial = ops->get_partial(schan, desc);
+                }
+
                 spin_unlock_irqrestore(&schan->chan_lock, flags);
 
                 shdma_chan_ld_cleanup(schan, true);
diff --git a/drivers/dma/sh/shdma.c b/drivers/dma/sh/shdma.c
index 027c9be97654..f41bcc5267fd 100644
--- a/drivers/dma/sh/shdma.c
+++ b/drivers/dma/sh/shdma.c
@@ -381,6 +381,17 @@ static bool sh_dmae_chan_irq(struct shdma_chan *schan, int irq)
         return true;
 }
 
+static size_t sh_dmae_get_partial(struct shdma_chan *schan,
+                                  struct shdma_desc *sdesc)
+{
+        struct sh_dmae_chan *sh_chan = container_of(schan, struct sh_dmae_chan,
+                                                    shdma_chan);
+        struct sh_dmae_desc *sh_desc = container_of(sdesc,
+                                        struct sh_dmae_desc, shdma_desc);
+        return (sh_desc->hw.tcr - sh_dmae_readl(sh_chan, TCR)) <<
+                sh_chan->xmit_shift;
+}
+
 /* Called from error IRQ or NMI */
 static bool sh_dmae_reset(struct sh_dmae_device *shdev)
 {
@@ -632,6 +643,7 @@ static const struct shdma_ops sh_dmae_shdma_ops = {
         .start_xfer = sh_dmae_start_xfer,
         .embedded_desc = sh_dmae_embedded_desc,
         .chan_irq = sh_dmae_chan_irq,
+        .get_partial = sh_dmae_get_partial,
 };
 
 static int __devinit sh_dmae_probe(struct platform_device *pdev)