[PATCH 2/2] iommu/mediatek-v1: Add error handle for mtk_iommu_probe

[Yong Wu]

In the original code, we lack the error handle. This patch adds them.

Signed-off-by: Yong Wu <yong...@mediatek.com>
---
 drivers/iommu/mtk_iommu_v1.c | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/drivers/iommu/mtk_iommu_v1.c b/drivers/iommu/mtk_iommu_v1.c
index bed0bb9d63fd..8ba9a2ec5509 100644
--- a/drivers/iommu/mtk_iommu_v1.c
+++ b/drivers/iommu/mtk_iommu_v1.c
@@ -620,12 +620,26 @@ static int mtk_iommu_probe(struct platform_device *pdev)
 
        ret = iommu_device_register(&data->iommu);
        if (ret)
-               return ret;
+               goto out_sysfs_remove;
 
-       if (!iommu_present(&platform_bus_type))
-               bus_set_iommu(&platform_bus_type,  &mtk_iommu_ops);
+       if (!iommu_present(&platform_bus_type)) {
+               ret = bus_set_iommu(&platform_bus_type,  &mtk_iommu_ops);
+               if (ret)
+                       goto out_dev_unreg;
+       }
 
-       return component_master_add_with_match(dev, &mtk_iommu_com_ops, match);
+       ret = component_master_add_with_match(dev, &mtk_iommu_com_ops, match);
+       if (ret)
+               goto out_bus_set_null;
+       return ret;
+
+out_bus_set_null:
+       bus_set_iommu(&platform_bus_type, NULL);
+out_dev_unreg:
+       iommu_device_unregister(&data->iommu);
+out_sysfs_remove:
+       iommu_device_sysfs_remove(&data->iommu);
+       return ret;
 }
 
 static int mtk_iommu_remove(struct platform_device *pdev)
-- 
2.18.0