#include "amd_iommu_types.h"
 #include "irq_remapping.h"
 
+#define AMD_IOMMU_MAPPING_ERROR        0
+
 #define CMD_SET_TYPE(cmd, t) ((cmd)->data[1] |= ((t) << 28))
 
 #define LOOP_TIMEOUT   100000
        paddr &= PAGE_MASK;
 
        address = dma_ops_alloc_iova(dev, dma_dom, pages, dma_mask);
-       if (address == DMA_ERROR_CODE)
+       if (address == AMD_IOMMU_MAPPING_ERROR)
                goto out;
 
        prot = dir2prot(direction);
 
        dma_ops_free_iova(dma_dom, address, pages);
 
-       return DMA_ERROR_CODE;
+       return AMD_IOMMU_MAPPING_ERROR;
 }
 
 /*
        if (PTR_ERR(domain) == -EINVAL)
                return (dma_addr_t)paddr;
        else if (IS_ERR(domain))
-               return DMA_ERROR_CODE;
+               return AMD_IOMMU_MAPPING_ERROR;
 
        dma_mask = *dev->dma_mask;
        dma_dom = to_dma_ops_domain(domain);
        npages = sg_num_pages(dev, sglist, nelems);
 
        address = dma_ops_alloc_iova(dev, dma_dom, npages, dma_mask);
-       if (address == DMA_ERROR_CODE)
+       if (address == AMD_IOMMU_MAPPING_ERROR)
                goto out_err;
 
        prot = dir2prot(direction);
        *dma_addr = __map_single(dev, dma_dom, page_to_phys(page),
                                 size, DMA_BIDIRECTIONAL, dma_mask);
 
-       if (*dma_addr == DMA_ERROR_CODE)
+       if (*dma_addr == AMD_IOMMU_MAPPING_ERROR)
                goto out_free;
 
        return page_address(page);
        return check_device(dev);
 }
 
+static int amd_iommu_mapping_error(struct device *dev, dma_addr_t dma_addr)
+{
+       return dma_addr == AMD_IOMMU_MAPPING_ERROR;
+}
+
 static const struct dma_map_ops amd_iommu_dma_ops = {
        .alloc          = alloc_coherent,
        .free           = free_coherent,
        .map_sg         = map_sg,
        .unmap_sg       = unmap_sg,
        .dma_supported  = amd_iommu_dma_supported,
+       .mapping_error  = amd_iommu_mapping_error,
 };
 
 static int init_reserved_iova_ranges(void)