iommu/vt-d: fix resource leakage on error recovery path in iommu_init_domains()

Release allocated resources on error recovery path in function
iommu_init_domains().

Also improve printk messages in iommu_init_domains().

Acked-by: Yijing Wang <wangyijing@huawei.com>
Signed-off-by: Jiang Liu <jiang.liu@linux.intel.com>
Signed-off-by: Joerg Roedel <joro@8bytes.org>

1 files changed, 8 insertions, 4 deletions

diff --git a/drivers/iommu/intel-iommu.c b/drivers/iommu/intel-iommu.c
index e2d2cb3119b3..0cbf1dda0730 100644
--- a/drivers/iommu/intel-iommu.c
+++ b/drivers/iommu/intel-iommu.c
@@ -1255,8 +1255,8 @@ static int iommu_init_domains(struct intel_iommu *iommu)
         unsigned long nlongs;
 
         ndomains = cap_ndoms(iommu->cap);
-        pr_debug("IOMMU %d: Number of Domains supported <%ld>\n", iommu->seq_id,
-                        ndomains);
+        pr_debug("IOMMU%d: Number of Domains supported <%ld>\n",
+                 iommu->seq_id, ndomains);
         nlongs = BITS_TO_LONGS(ndomains);
 
         spin_lock_init(&iommu->lock);
@@ -1266,13 +1266,17 @@ static int iommu_init_domains(struct intel_iommu *iommu)
          */
         iommu->domain_ids = kcalloc(nlongs, sizeof(unsigned long), GFP_KERNEL);
         if (!iommu->domain_ids) {
-                printk(KERN_ERR "Allocating domain id array failed\n");
+                pr_err("IOMMU%d: allocating domain id array failed\n",
+                       iommu->seq_id);
                 return -ENOMEM;
         }
         iommu->domains = kcalloc(ndomains, sizeof(struct dmar_domain *),
                         GFP_KERNEL);
         if (!iommu->domains) {
-                printk(KERN_ERR "Allocating domain array failed\n");
+                pr_err("IOMMU%d: allocating domain array failed\n",
+                       iommu->seq_id);
+                kfree(iommu->domain_ids);
+                iommu->domain_ids = NULL;
                 return -ENOMEM;
         }