1 files changed, 10 insertions, 1 deletions

diff --git a/drivers/mmc/host/cqhci.c b/drivers/mmc/host/cqhci.c
index bd9a59bc0e31..a8af682a9182 100644
--- a/drivers/mmc/host/cqhci.c
+++ b/drivers/mmc/host/cqhci.c
@@ -217,12 +217,21 @@ static int cqhci_host_alloc_tdl(struct cqhci_host *cq_host)
                                                  cq_host->desc_size,
                                                  &cq_host->desc_dma_base,
                                                  GFP_KERNEL);
+        if (!cq_host->desc_base)
+                return -ENOMEM;
+
         cq_host->trans_desc_base = dmam_alloc_coherent(mmc_dev(cq_host->mmc),
                                               cq_host->data_size,
                                               &cq_host->trans_desc_dma_base,
                                               GFP_KERNEL);
-        if (!cq_host->desc_base || !cq_host->trans_desc_base)
+        if (!cq_host->trans_desc_base) {
+                dmam_free_coherent(mmc_dev(cq_host->mmc), cq_host->desc_size,
+                                   cq_host->desc_base,
+                                   cq_host->desc_dma_base);
+                cq_host->desc_base = NULL;
+                cq_host->desc_dma_base = 0;
                 return -ENOMEM;
+        }
 
         pr_debug("%s: cqhci: desc-base: 0x%p trans-base: 0x%p\n desc_dma 0x%llx trans_dma: 0x%llx\n",
                  mmc_hostname(cq_host->mmc), cq_host->desc_base, cq_host->trans_desc_base,