#include <linux/pci.h>
 #include <asm/pci_dma.h>
 
+#define S390_MAPPING_ERROR             (~(dma_addr_t) 0x0)
+
 static struct kmem_cache *dma_region_table_cache;
 static struct kmem_cache *dma_page_table_cache;
 static int s390_iommu_strict;
 
 out_error:
        spin_unlock_irqrestore(&zdev->iommu_bitmap_lock, flags);
-       return DMA_ERROR_CODE;
+       return S390_MAPPING_ERROR;
 }
 
 static void dma_free_address(struct device *dev, dma_addr_t dma_addr, int size)
        /* This rounds up number of pages based on size and offset */
        nr_pages = iommu_num_pages(pa, size, PAGE_SIZE);
        dma_addr = dma_alloc_address(dev, nr_pages);
-       if (dma_addr == DMA_ERROR_CODE) {
+       if (dma_addr == S390_MAPPING_ERROR) {
                ret = -ENOSPC;
                goto out_err;
        }
 out_err:
        zpci_err("map error:\n");
        zpci_err_dma(ret, pa);
-       return DMA_ERROR_CODE;
+       return S390_MAPPING_ERROR;
 }
 
 static void s390_dma_unmap_pages(struct device *dev, dma_addr_t dma_addr,
        int ret;
 
        dma_addr_base = dma_alloc_address(dev, nr_pages);
-       if (dma_addr_base == DMA_ERROR_CODE)
+       if (dma_addr_base == S390_MAPPING_ERROR)
                return -ENOMEM;
 
        dma_addr = dma_addr_base;
        for (i = 1; i < nr_elements; i++) {
                s = sg_next(s);
 
-               s->dma_address = DMA_ERROR_CODE;
+               s->dma_address = S390_MAPPING_ERROR;
                s->dma_length = 0;
 
                if (s->offset || (size & ~PAGE_MASK) ||
                s->dma_length = 0;
        }
 }
+       
+static int s390_mapping_error(struct device *dev, dma_addr_t dma_addr)
+{
+       return dma_addr == S390_MAPPING_ERROR;
+}
 
 int zpci_dma_init_device(struct zpci_dev *zdev)
 {
        .unmap_sg       = s390_dma_unmap_sg,
        .map_page       = s390_dma_map_pages,
        .unmap_page     = s390_dma_unmap_pages,
+       .mapping_error  = s390_mapping_error,
        /* if we support direct DMA this must be conditional */
        .is_phys        = 0,
        /* dma_supported is unconditionally true without a callback */