#include <linux/scatterlist.h>
 
 #include <asm/cacheflush.h>
+#include <asm/dma-iommu.h>
 
 #undef STATS
 
        if (buf == NULL) {
                dev_err(dev, "%s: unable to map unsafe buffer %p!\n",
                       __func__, ptr);
-               return DMA_ERROR_CODE;
+               return ARM_MAPPING_ERROR;
        }
 
        dev_dbg(dev, "%s: unsafe buffer %p (dma=%#x) mapped to %p (dma=%#x)\n",
 
        ret = needs_bounce(dev, dma_addr, size);
        if (ret < 0)
-               return DMA_ERROR_CODE;
+               return ARM_MAPPING_ERROR;
 
        if (ret == 0) {
                arm_dma_ops.sync_single_for_device(dev, dma_addr, size, dir);
 
        if (PageHighMem(page)) {
                dev_err(dev, "DMA buffer bouncing of HIGHMEM pages is not supported\n");
-               return DMA_ERROR_CODE;
+               return ARM_MAPPING_ERROR;
        }
 
        return map_single(dev, page_address(page) + offset, size, dir, attrs);
        return arm_dma_ops.set_dma_mask(dev, dma_mask);
 }
 
+static int dmabounce_mapping_error(struct device *dev, dma_addr_t dma_addr)
+{
+       return arm_dma_ops.mapping_error(dev, dma_addr);
+}
+
 static const struct dma_map_ops dmabounce_ops = {
        .alloc                  = arm_dma_alloc,
        .free                   = arm_dma_free,
        .sync_sg_for_cpu        = arm_dma_sync_sg_for_cpu,
        .sync_sg_for_device     = arm_dma_sync_sg_for_device,
        .set_dma_mask           = dmabounce_set_mask,
+       .mapping_error          = dmabounce_mapping_error,
 };
 
 static int dmabounce_init_pool(struct dmabounce_pool *pool, struct device *dev,
 
        __dma_page_cpu_to_dev(page, offset, size, dir);
 }
 
+static int arm_dma_mapping_error(struct device *dev, dma_addr_t dma_addr)
+{
+       return dma_addr == ARM_MAPPING_ERROR;
+}
+
 const struct dma_map_ops arm_dma_ops = {
        .alloc                  = arm_dma_alloc,
        .free                   = arm_dma_free,
        .sync_single_for_device = arm_dma_sync_single_for_device,
        .sync_sg_for_cpu        = arm_dma_sync_sg_for_cpu,
        .sync_sg_for_device     = arm_dma_sync_sg_for_device,
+       .mapping_error          = arm_dma_mapping_error,
 };
 EXPORT_SYMBOL(arm_dma_ops);
 
        .get_sgtable            = arm_dma_get_sgtable,
        .map_page               = arm_coherent_dma_map_page,
        .map_sg                 = arm_dma_map_sg,
+       .mapping_error          = arm_dma_mapping_error,
 };
 EXPORT_SYMBOL(arm_coherent_dma_ops);
 
        gfp &= ~(__GFP_COMP);
        args.gfp = gfp;
 
-       *handle = DMA_ERROR_CODE;
+       *handle = ARM_MAPPING_ERROR;
        allowblock = gfpflags_allow_blocking(gfp);
        cma = allowblock ? dev_get_cma_area(dev) : false;
 
        if (i == mapping->nr_bitmaps) {
                if (extend_iommu_mapping(mapping)) {
                        spin_unlock_irqrestore(&mapping->lock, flags);
-                       return DMA_ERROR_CODE;
+                       return ARM_MAPPING_ERROR;
                }
 
                start = bitmap_find_next_zero_area(mapping->bitmaps[i],
 
                if (start > mapping->bits) {
                        spin_unlock_irqrestore(&mapping->lock, flags);
-                       return DMA_ERROR_CODE;
+                       return ARM_MAPPING_ERROR;
                }
 
                bitmap_set(mapping->bitmaps[i], start, count);
        int i;
 
        dma_addr = __alloc_iova(mapping, size);
-       if (dma_addr == DMA_ERROR_CODE)
+       if (dma_addr == ARM_MAPPING_ERROR)
                return dma_addr;
 
        iova = dma_addr;
 fail:
        iommu_unmap(mapping->domain, dma_addr, iova-dma_addr);
        __free_iova(mapping, dma_addr, size);
-       return DMA_ERROR_CODE;
+       return ARM_MAPPING_ERROR;
 }
 
 static int __iommu_remove_mapping(struct device *dev, dma_addr_t iova, size_t size)
                return NULL;
 
        *handle = __iommu_create_mapping(dev, &page, size, attrs);
-       if (*handle == DMA_ERROR_CODE)
+       if (*handle == ARM_MAPPING_ERROR)
                goto err_mapping;
 
        return addr;
        struct page **pages;
        void *addr = NULL;
 
-       *handle = DMA_ERROR_CODE;
+       *handle = ARM_MAPPING_ERROR;
        size = PAGE_ALIGN(size);
 
        if (coherent_flag  == COHERENT || !gfpflags_allow_blocking(gfp))
                return NULL;
 
        *handle = __iommu_create_mapping(dev, pages, size, attrs);
-       if (*handle == DMA_ERROR_CODE)
+       if (*handle == ARM_MAPPING_ERROR)
                goto err_buffer;
 
        if (attrs & DMA_ATTR_NO_KERNEL_MAPPING)
        int prot;
 
        size = PAGE_ALIGN(size);
-       *handle = DMA_ERROR_CODE;
+       *handle = ARM_MAPPING_ERROR;
 
        iova_base = iova = __alloc_iova(mapping, size);
-       if (iova == DMA_ERROR_CODE)
+       if (iova == ARM_MAPPING_ERROR)
                return -ENOMEM;
 
        for (count = 0, s = sg; count < (size >> PAGE_SHIFT); s = sg_next(s)) {
        for (i = 1; i < nents; i++) {
                s = sg_next(s);
 
-               s->dma_address = DMA_ERROR_CODE;
+               s->dma_address = ARM_MAPPING_ERROR;
                s->dma_length = 0;
 
                if (s->offset || (size & ~PAGE_MASK) || size + s->length > max) {
        int ret, prot, len = PAGE_ALIGN(size + offset);
 
        dma_addr = __alloc_iova(mapping, len);
-       if (dma_addr == DMA_ERROR_CODE)
+       if (dma_addr == ARM_MAPPING_ERROR)
                return dma_addr;
 
        prot = __dma_info_to_prot(dir, attrs);
        return dma_addr + offset;
 fail:
        __free_iova(mapping, dma_addr, len);
-       return DMA_ERROR_CODE;
+       return ARM_MAPPING_ERROR;
 }
 
 /**
        size_t len = PAGE_ALIGN(size + offset);
 
        dma_addr = __alloc_iova(mapping, len);
-       if (dma_addr == DMA_ERROR_CODE)
+       if (dma_addr == ARM_MAPPING_ERROR)
                return dma_addr;
 
        prot = __dma_info_to_prot(dir, attrs) | IOMMU_MMIO;
        return dma_addr + offset;
 fail:
        __free_iova(mapping, dma_addr, len);
-       return DMA_ERROR_CODE;
+       return ARM_MAPPING_ERROR;
 }
 
 /**
 
        .map_resource           = arm_iommu_map_resource,
        .unmap_resource         = arm_iommu_unmap_resource,
+
+       .mapping_error          = arm_dma_mapping_error,
 };
 
 const struct dma_map_ops iommu_coherent_ops = {
 
        .map_resource   = arm_iommu_map_resource,
        .unmap_resource = arm_iommu_unmap_resource,
+
+       .mapping_error          = arm_dma_mapping_error,
 };
 
 /**