{
        struct mtk_iommu_data *data = dev_iommu_priv_get(dev);
        struct mtk_iommu_domain *dom = to_mtk_domain(domain);
+       struct dma_iommu_mapping *mtk_mapping;
        int ret;
 
-       if (!data)
-               return -ENODEV;
+       /* Only allow the domain created internally. */
+       mtk_mapping = data->dev->archdata.iommu;
+       if (mtk_mapping->domain != domain)
+               return 0;
 
        if (!data->m4u_dom) {
                data->m4u_dom = dom;
 {
        struct mtk_iommu_data *data = dev_iommu_priv_get(dev);
 
-       if (!data)
-               return;
-
        mtk_iommu_config(data, dev, false);
 }
 
        return 0;
 }
 
+static int mtk_iommu_def_domain_type(struct device *dev)
+{
+       return IOMMU_DOMAIN_UNMANAGED;
+}
+
 static struct iommu_device *mtk_iommu_probe_device(struct device *dev)
 {
        struct iommu_fwspec *fwspec = dev_iommu_fwspec_get(dev);
        .probe_device   = mtk_iommu_probe_device,
        .probe_finalize = mtk_iommu_probe_finalize,
        .release_device = mtk_iommu_release_device,
+       .def_domain_type = mtk_iommu_def_domain_type,
        .device_group   = generic_device_group,
        .pgsize_bitmap  = ~0UL << MT2701_IOMMU_PAGE_SHIFT,
 };