4 files changed, 42 insertions, 24 deletions

diff --git a/arch/powerpc/kernel/vio.c b/arch/powerpc/kernel/vio.c
index fd631d4d160..eaf7f6992a2 100644
--- a/arch/powerpc/kernel/vio.c
+++ b/arch/powerpc/kernel/vio.c
@@ -49,11 +49,8 @@ static struct vio_dev vio_bus_device  = { /* fake "parent" device */
 };
 
 #ifdef CONFIG_PPC_ISERIES
-struct device *iSeries_vio_dev = &vio_bus_device.dev;
-EXPORT_SYMBOL(iSeries_vio_dev);
-
 static struct iommu_table veth_iommu_table;
-static struct iommu_table vio_iommu_table;
+struct iommu_table vio_iommu_table;
 
 static void __init iommu_vio_init(void)
 {
@@ -66,8 +63,6 @@ static void __init iommu_vio_init(void)
                 printk("Virtual Bus VETH TCE table failed.\n");
         if (!iommu_init_table(&vio_iommu_table, -1))
                 printk("Virtual Bus VIO TCE table failed.\n");
-        vio_bus_device.dev.archdata.dma_ops = &dma_iommu_ops;
-        vio_bus_device.dev.archdata.dma_data = &vio_iommu_table;
 }
 #else
 static void __init iommu_vio_init(void)
diff --git a/arch/powerpc/platforms/iseries/iommu.c b/arch/powerpc/platforms/iseries/iommu.c
index 3b6a9666c2c..3281f10bbd1 100644
--- a/arch/powerpc/platforms/iseries/iommu.c
+++ b/arch/powerpc/platforms/iseries/iommu.c
@@ -28,6 +28,7 @@
 #include <linux/dma-mapping.h>
 #include <linux/list.h>
 #include <linux/pci.h>
+#include <linux/module.h>
 
 #include <asm/iommu.h>
 #include <asm/tce.h>
@@ -36,6 +37,7 @@
 #include <asm/prom.h>
 #include <asm/pci-bridge.h>
 #include <asm/iseries/hv_call_xm.h>
+#include <asm/iseries/hv_call_event.h>
 #include <asm/iseries/iommu.h>
 
 static void tce_build_iSeries(struct iommu_table *tbl, long index, long npages,
@@ -189,6 +191,34 @@ void iommu_devnode_init_iSeries(struct pci_dev *pdev, struct device_node *dn)
 }
 #endif
 
+extern struct iommu_table vio_iommu_table;
+
+void *iseries_hv_alloc(size_t size, dma_addr_t *dma_handle, gfp_t flag)
+{
+        return iommu_alloc_coherent(&vio_iommu_table, size, dma_handle,
+                                DMA_32BIT_MASK, flag, -1);
+}
+EXPORT_SYMBOL_GPL(iseries_hv_alloc);
+
+void iseries_hv_free(size_t size, void *vaddr, dma_addr_t dma_handle)
+{
+        iommu_free_coherent(&vio_iommu_table, size, vaddr, dma_handle);
+}
+EXPORT_SYMBOL_GPL(iseries_hv_free);
+
+dma_addr_t iseries_hv_map(void *vaddr, size_t size,
+                        enum dma_data_direction direction)
+{
+        return iommu_map_single(&vio_iommu_table, vaddr, size,
+                                DMA_32BIT_MASK, direction);
+}
+
+void iseries_hv_unmap(dma_addr_t dma_handle, size_t size,
+                        enum dma_data_direction direction)
+{
+        iommu_unmap_single(&vio_iommu_table, dma_handle, size, direction);
+}
+
 void iommu_init_early_iSeries(void)
 {
         ppc_md.tce_build = tce_build_iSeries;
diff --git a/arch/powerpc/platforms/iseries/mf.c b/arch/powerpc/platforms/iseries/mf.c
index b1187d95e3b..c0f2433bc16 100644
--- a/arch/powerpc/platforms/iseries/mf.c
+++ b/arch/powerpc/platforms/iseries/mf.c
@@ -39,9 +39,9 @@
 #include <asm/paca.h>
 #include <asm/abs_addr.h>
 #include <asm/firmware.h>
-#include <asm/iseries/vio.h>
 #include <asm/iseries/mf.h>
 #include <asm/iseries/hv_lp_config.h>
+#include <asm/iseries/hv_lp_event.h>
 #include <asm/iseries/it_lp_queue.h>
 
 #include "setup.h"
@@ -870,8 +870,7 @@ static int proc_mf_dump_cmdline(char *page, char **start, off_t off,
         if ((off + count) > 256)
                 count = 256 - off;
 
-        dma_addr = dma_map_single(iSeries_vio_dev, page, off + count,
-                        DMA_FROM_DEVICE);
+        dma_addr = iseries_hv_map(page, off + count, DMA_FROM_DEVICE);
         if (dma_mapping_error(dma_addr))
                 return -ENOMEM;
         memset(page, 0, off + count);
@@ -883,8 +882,7 @@ static int proc_mf_dump_cmdline(char *page, char **start, off_t off,
         vsp_cmd.sub_data.kern.length = off + count;
         mb();
         rc = signal_vsp_instruction(&vsp_cmd);
-        dma_unmap_single(iSeries_vio_dev, dma_addr, off + count,
-                        DMA_FROM_DEVICE);
+        iseries_hv_unmap(dma_addr, off + count, DMA_FROM_DEVICE);
         if (rc)
                 return rc;
         if (vsp_cmd.result_code != 0)
@@ -919,8 +917,7 @@ static int mf_getVmlinuxChunk(char *buffer, int *size, int offset, u64 side)
         int len = *size;
         dma_addr_t dma_addr;
 
-        dma_addr = dma_map_single(iSeries_vio_dev, buffer, len,
-                        DMA_FROM_DEVICE);
+        dma_addr = iseries_hv_map(buffer, len, DMA_FROM_DEVICE);
         memset(buffer, 0, len);
         memset(&vsp_cmd, 0, sizeof(vsp_cmd));
         vsp_cmd.cmd = 32;
@@ -938,7 +935,7 @@ static int mf_getVmlinuxChunk(char *buffer, int *size, int offset, u64 side)
                         rc = -ENOMEM;
         }
 
-        dma_unmap_single(iSeries_vio_dev, dma_addr, len, DMA_FROM_DEVICE);
+        iseries_hv_unmap(dma_addr, len, DMA_FROM_DEVICE);
 
         return rc;
 }
@@ -1149,8 +1146,7 @@ static int proc_mf_change_cmdline(struct file *file, const char __user *buffer,
                 goto out;
 
         dma_addr = 0;
-        page = dma_alloc_coherent(iSeries_vio_dev, count, &dma_addr,
-                        GFP_ATOMIC);
+        page = iseries_hv_alloc(count, &dma_addr, GFP_ATOMIC);
         ret = -ENOMEM;
         if (page == NULL)
                 goto out;
@@ -1170,7 +1166,7 @@ static int proc_mf_change_cmdline(struct file *file, const char __user *buffer,
         ret = count;
 
 out_free:
-        dma_free_coherent(iSeries_vio_dev, count, page, dma_addr);
+        iseries_hv_free(count, page, dma_addr);
 out:
         return ret;
 }
@@ -1190,8 +1186,7 @@ static ssize_t proc_mf_change_vmlinux(struct file *file,
                 goto out;
 
         dma_addr = 0;
-        page = dma_alloc_coherent(iSeries_vio_dev, count, &dma_addr,
-                        GFP_ATOMIC);
+        page = iseries_hv_alloc(count, &dma_addr, GFP_ATOMIC);
         rc = -ENOMEM;
         if (page == NULL) {
                 printk(KERN_ERR "mf.c: couldn't allocate memory to set vmlinux chunk\n");
@@ -1219,7 +1214,7 @@ static ssize_t proc_mf_change_vmlinux(struct file *file,
         *ppos += count;
         rc = count;
 out_free:
-        dma_free_coherent(iSeries_vio_dev, count, page, dma_addr);
+        iseries_hv_free(count, page, dma_addr);
 out:
         return rc;
 }
diff --git a/arch/powerpc/platforms/iseries/viopath.c b/arch/powerpc/platforms/iseries/viopath.c
index 45f2fe39c87..df23331eb25 100644
--- a/arch/powerpc/platforms/iseries/viopath.c
+++ b/arch/powerpc/platforms/iseries/viopath.c
@@ -124,8 +124,7 @@ static int proc_viopath_show(struct seq_file *m, void *v)
         if (!buf)
                 return 0;
 
-        handle = dma_map_single(iSeries_vio_dev, buf, HW_PAGE_SIZE,
-                                DMA_FROM_DEVICE);
+        handle = iseries_hv_map(buf, HW_PAGE_SIZE, DMA_FROM_DEVICE);
 
         hvrc = HvCallEvent_signalLpEventFast(viopath_hostLp,
                         HvLpEvent_Type_VirtualIo,
@@ -146,8 +145,7 @@ static int proc_viopath_show(struct seq_file *m, void *v)
         buf[HW_PAGE_SIZE-1] = '\0';
         seq_printf(m, "%s", buf);
 
-        dma_unmap_single(iSeries_vio_dev, handle, HW_PAGE_SIZE,
-                         DMA_FROM_DEVICE);
+        iseries_hv_unmap(handle, HW_PAGE_SIZE, DMA_FROM_DEVICE);
         kfree(buf);
 
         seq_printf(m, "AVAILABLE_VETH=%x\n", vlanMap);