1 files changed, 39 insertions, 19 deletions

diff --git a/drivers/dma/dw/core.c b/drivers/dma/dw/core.c
index cfdbb92aae1d..a27ded53ab4f 100644
--- a/drivers/dma/dw/core.c
+++ b/drivers/dma/dw/core.c
@@ -1493,6 +1493,13 @@ int dw_dma_probe(struct dw_dma_chip *chip, struct dw_dma_platform_data *pdata)
         dw->regs = chip->regs;
         chip->dw = dw;
 
+        dw->clk = devm_clk_get(chip->dev, "hclk");
+        if (IS_ERR(dw->clk))
+                return PTR_ERR(dw->clk);
+        err = clk_prepare_enable(dw->clk);
+        if (err)
+                return err;
+
         dw_params = dma_read_byaddr(chip->regs, DW_PARAMS);
         autocfg = dw_params >> DW_PARAMS_EN & 0x1;
 
@@ -1500,15 +1507,19 @@ int dw_dma_probe(struct dw_dma_chip *chip, struct dw_dma_platform_data *pdata)
 
         if (!pdata && autocfg) {
                 pdata = devm_kzalloc(chip->dev, sizeof(*pdata), GFP_KERNEL);
-                if (!pdata)
-                        return -ENOMEM;
+                if (!pdata) {
+                        err = -ENOMEM;
+                        goto err_pdata;
+                }
 
                 /* Fill platform data with the default values */
                 pdata->is_private = true;
                 pdata->chan_allocation_order = CHAN_ALLOCATION_ASCENDING;
                 pdata->chan_priority = CHAN_PRIORITY_ASCENDING;
-        } else if (!pdata || pdata->nr_channels > DW_DMA_MAX_NR_CHANNELS)
-                return -EINVAL;
+        } else if (!pdata || pdata->nr_channels > DW_DMA_MAX_NR_CHANNELS) {
+                err = -EINVAL;
+                goto err_pdata;
+        }
 
         if (autocfg)
                 nr_channels = (dw_params >> DW_PARAMS_NR_CHAN & 0x7) + 1;
@@ -1517,13 +1528,10 @@ int dw_dma_probe(struct dw_dma_chip *chip, struct dw_dma_platform_data *pdata)
 
         dw->chan = devm_kcalloc(chip->dev, nr_channels, sizeof(*dw->chan),
                                 GFP_KERNEL);
-        if (!dw->chan)
-                return -ENOMEM;
-
-        dw->clk = devm_clk_get(chip->dev, "hclk");
-        if (IS_ERR(dw->clk))
-                return PTR_ERR(dw->clk);
-        clk_prepare_enable(dw->clk);
+        if (!dw->chan) {
+                err = -ENOMEM;
+                goto err_pdata;
+        }
 
         /* Get hardware configuration parameters */
         if (autocfg) {
@@ -1548,21 +1556,22 @@ int dw_dma_probe(struct dw_dma_chip *chip, struct dw_dma_platform_data *pdata)
         /* Disable BLOCK interrupts as well */
         channel_clear_bit(dw, MASK.BLOCK, dw->all_chan_mask);
 
-        err = devm_request_irq(chip->dev, chip->irq, dw_dma_interrupt,
-                               IRQF_SHARED, "dw_dmac", dw);
-        if (err)
-                return err;
-
         /* Create a pool of consistent memory blocks for hardware descriptors */
         dw->desc_pool = dmam_pool_create("dw_dmac_desc_pool", chip->dev,
                                          sizeof(struct dw_desc), 4, 0);
         if (!dw->desc_pool) {
                 dev_err(chip->dev, "No memory for descriptors dma pool\n");
-                return -ENOMEM;
+                err = -ENOMEM;
+                goto err_pdata;
         }
 
         tasklet_init(&dw->tasklet, dw_dma_tasklet, (unsigned long)dw);
 
+        err = request_irq(chip->irq, dw_dma_interrupt, IRQF_SHARED,
+                          "dw_dmac", dw);
+        if (err)
+                goto err_pdata;
+
         INIT_LIST_HEAD(&dw->dma.channels);
         for (i = 0; i < nr_channels; i++) {
                 struct dw_dma_chan      *dwc = &dw->chan[i];
@@ -1650,12 +1659,20 @@ int dw_dma_probe(struct dw_dma_chip *chip, struct dw_dma_platform_data *pdata)
 
         dma_writel(dw, CFG, DW_CFG_DMA_EN);
 
+        err = dma_async_device_register(&dw->dma);
+        if (err)
+                goto err_dma_register;
+
         dev_info(chip->dev, "DesignWare DMA Controller, %d channels\n",
                  nr_channels);
 
-        dma_async_device_register(&dw->dma);
-
         return 0;
+
+err_dma_register:
+        free_irq(chip->irq, dw);
+err_pdata:
+        clk_disable_unprepare(dw->clk);
+        return err;
 }
 EXPORT_SYMBOL_GPL(dw_dma_probe);
 
@@ -1667,6 +1684,7 @@ int dw_dma_remove(struct dw_dma_chip *chip)
         dw_dma_off(dw);
         dma_async_device_unregister(&dw->dma);
 
+        free_irq(chip->irq, dw);
         tasklet_kill(&dw->tasklet);
 
         list_for_each_entry_safe(dwc, _dwc, &dw->dma.channels,
@@ -1675,6 +1693,8 @@ int dw_dma_remove(struct dw_dma_chip *chip)
                 channel_clear_bit(dw, CH_EN, dwc->mask);
         }
 
+        clk_disable_unprepare(dw->clk);
+
         return 0;
 }
 EXPORT_SYMBOL_GPL(dw_dma_remove);