Properly clean up allocated resources on the drivers probe failure and
remove unneeded checks.

Signed-off-by: Dmitry Osipenko <dig...@gmail.com>
---
 drivers/iommu/tegra-gart.c | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/drivers/iommu/tegra-gart.c b/drivers/iommu/tegra-gart.c
index 9a427cb35340..def570bc88c4 100644
--- a/drivers/iommu/tegra-gart.c
+++ b/drivers/iommu/tegra-gart.c
@@ -429,9 +429,6 @@ static int tegra_gart_probe(struct platform_device *pdev)
        struct device *dev = &pdev->dev;
        int ret;
 
-       if (gart_handle)
-               return -EIO;
-
        BUILD_BUG_ON(PAGE_SHIFT != GART_PAGE_SHIFT);
 
        /* the GART memory aperture is required */
@@ -466,8 +463,7 @@ static int tegra_gart_probe(struct platform_device *pdev)
        ret = iommu_device_register(&gart->iommu);
        if (ret) {
                dev_err(dev, "Failed to register IOMMU\n");
-               iommu_device_sysfs_remove(&gart->iommu);
-               return ret;
+               goto remove_sysfs;
        }
 
        gart->dev = &pdev->dev;
@@ -483,7 +479,8 @@ static int tegra_gart_probe(struct platform_device *pdev)
        gart->savedata = vmalloc(array_size(sizeof(u32), gart->page_count));
        if (!gart->savedata) {
                dev_err(dev, "failed to allocate context save area\n");
-               return -ENOMEM;
+               ret = -ENOMEM;
+               goto unregister_iommu;
        }
 
        platform_set_drvdata(pdev, gart);
@@ -493,6 +490,13 @@ static int tegra_gart_probe(struct platform_device *pdev)
        tegra_mc_register_gart(&gart->mc_gart_handle);
 
        return 0;
+
+unregister_iommu:
+       iommu_device_unregister(&gart->iommu);
+remove_sysfs:
+       iommu_device_sysfs_remove(&gart->iommu);
+
+       return ret;
 }
 
 static const struct dev_pm_ops tegra_gart_pm_ops = {
-- 
2.18.0

_______________________________________________
iommu mailing list
iommu@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/iommu

Reply via email to