2 files changed, 35 insertions, 29 deletions

diff --git a/arch/sparc/mm/io-unit.c b/arch/sparc/mm/io-unit.c
index f167835db3df..1093514a5773 100644
--- a/arch/sparc/mm/io-unit.c
+++ b/arch/sparc/mm/io-unit.c
@@ -34,13 +34,18 @@
 #define IOPERM        (IOUPTE_CACHE | IOUPTE_WRITE | IOUPTE_VALID)
 #define MKIOPTE(phys) __iopte((((phys)>>4) & IOUPTE_PAGE) | IOPERM)
 
-void __init
-iounit_init(int sbi_node, int io_node, struct sbus_bus *sbus)
+void __init iounit_init(struct sbus_bus *sbus)
 {
-        iopte_t *xpt, *xptend;
+        struct device_node *dp = sbus->ofdev.node;
         struct iounit_struct *iounit;
-        struct linux_prom_registers iommu_promregs[PROMREG_MAX];
-        struct resource r;
+        iopte_t *xpt, *xptend;
+        struct of_device *op;
+
+        op = of_find_device_by_node(dp);
+        if (!op) {
+                prom_printf("SUN4D: Cannot find SBI of_device.\n");
+                prom_halt();
+        }
 
         iounit = kzalloc(sizeof(struct iounit_struct), GFP_ATOMIC);
         if (!iounit) {
@@ -55,18 +60,14 @@ iounit_init(int sbi_node, int io_node, struct sbus_bus *sbus)
         iounit->rotor[1] = IOUNIT_BMAP2_START;
         iounit->rotor[2] = IOUNIT_BMAPM_START;
 
-        xpt = NULL;
-        if(prom_getproperty(sbi_node, "reg", (void *) iommu_promregs,
-                            sizeof(iommu_promregs)) != -1) {
-                prom_apply_generic_ranges(io_node, 0, iommu_promregs, 3);
-                memset(&r, 0, sizeof(r));
-                r.flags = iommu_promregs[2].which_io;
-                r.start = iommu_promregs[2].phys_addr;
-                xpt = (iopte_t *) sbus_ioremap(&r, 0, PAGE_SIZE * 16, "XPT");
+        xpt = of_ioremap(&op->resource[2], 0, PAGE_SIZE * 16, "XPT");
+        if (!xpt) {
+                prom_printf("SUN4D: Cannot map External Page Table.");
+                prom_halt();
         }
-        if(!xpt) panic("Cannot map External Page Table.");
         
         sbus->ofdev.dev.archdata.iommu = iounit;
+        op->dev.archdata.iommu = iounit;
         iounit->page_table = xpt;
         spin_lock_init(&iounit->lock);
         
diff --git a/arch/sparc/mm/iommu.c b/arch/sparc/mm/iommu.c
index 4b934270f05e..a86c9f552fa1 100644
--- a/arch/sparc/mm/iommu.c
+++ b/arch/sparc/mm/iommu.c
@@ -55,30 +55,34 @@ static pgprot_t dvma_prot;		/* Consistent mapping pte flags */
 #define IOPERM        (IOPTE_CACHE | IOPTE_WRITE | IOPTE_VALID)
 #define MKIOPTE(pfn, perm) (((((pfn)<<8) & IOPTE_PAGE) | (perm)) & ~IOPTE_WAZ)
 
-void __init
-iommu_init(int iommund, struct sbus_bus *sbus)
+void __init iommu_init(struct device_node *parent, struct sbus_bus *sbus)
 {
-        unsigned int impl, vers;
-        unsigned long tmp;
+        struct of_device *parent_op, *op;
         struct iommu_struct *iommu;
-        struct linux_prom_registers iommu_promregs[PROMREG_MAX];
-        struct resource r;
+        unsigned int impl, vers;
         unsigned long *bitmap;
+        unsigned long tmp;
+
+        parent_op = of_find_device_by_node(parent);
+        if (!parent_op) {
+                prom_printf("Unable to find IOMMU of_device\n");
+                prom_halt();
+        }
+
+        op = of_find_device_by_node(sbus->ofdev.node);
+        if (!op) {
+                prom_printf("Unable to find SBUS of_device\n");
+                prom_halt();
+        }
 
         iommu = kmalloc(sizeof(struct iommu_struct), GFP_ATOMIC);
         if (!iommu) {
                 prom_printf("Unable to allocate iommu structure\n");
                 prom_halt();
         }
-        iommu->regs = NULL;
-        if (prom_getproperty(iommund, "reg", (void *) iommu_promregs,
-                         sizeof(iommu_promregs)) != -1) {
-                memset(&r, 0, sizeof(r));
-                r.flags = iommu_promregs[0].which_io;
-                r.start = iommu_promregs[0].phys_addr;
-                iommu->regs = (struct iommu_regs *)
-                        sbus_ioremap(&r, 0, PAGE_SIZE * 3, "iommu_regs");
-        }
+
+        iommu->regs = of_ioremap(&parent_op->resource[0], 0, PAGE_SIZE * 3,
+                                 "iommu_regs");
         if (!iommu->regs) {
                 prom_printf("Cannot map IOMMU registers\n");
                 prom_halt();
@@ -133,6 +137,7 @@ iommu_init(int iommund, struct sbus_bus *sbus)
             (int)(IOMMU_NPTES*sizeof(iopte_t)), (int)IOMMU_NPTES);
 
         sbus->ofdev.dev.archdata.iommu = iommu;
+        op->dev.archdata.iommu = iommu;
 }
 
 /* This begs to be btfixup-ed by srmmu. */