2 files changed, 18 insertions, 4 deletions

diff --git a/arch/mips/pci/pci-sb1250.c b/arch/mips/pci/pci-sb1250.c
index 42e4d2c800fa..2a09ad91ec8c 100644
--- a/arch/mips/pci/pci-sb1250.c
+++ b/arch/mips/pci/pci-sb1250.c
@@ -207,6 +207,7 @@ struct pci_controller sb1250_controller = {
 
 static int __init sb1250_pcibios_init(void)
 {
+        void __iomem *io_map_base;
         uint32_t cmdreg;
         uint64_t reg;
         extern int pci_probe_only;
@@ -253,12 +254,13 @@ static int __init sb1250_pcibios_init(void)
          * works correctly with most of Linux's drivers.
          * XXX ehs: Should this happen in PCI Device mode?
          */
-
-        set_io_port_base((unsigned long)
-                         ioremap(A_PHYS_LDTPCI_IO_MATCH_BYTES, 65536));
         isa_slot_offset = (unsigned long)
             ioremap(A_PHYS_LDTPCI_IO_MATCH_BYTES_32, 1024 * 1024);
 
+        io_map_base = ioremap(A_PHYS_LDTPCI_IO_MATCH_BYTES, 1024 * 1024);
+        sb1250_controller.io_map_base = io_map_base;
+        set_io_port_base((unsigned long)io_map_base);
+
 #ifdef CONFIG_SIBYTE_HAS_LDT
         /*
          * Also check the LDT bridge's enable, just in case we didn't
diff --git a/arch/mips/pci/pci-vr41xx.c b/arch/mips/pci/pci-vr41xx.c
index 33c4f683d067..d1e049b55f34 100644
--- a/arch/mips/pci/pci-vr41xx.c
+++ b/arch/mips/pci/pci-vr41xx.c
@@ -3,7 +3,7 @@
  *
  *  Copyright (C) 2001-2003 MontaVista Software Inc.
  *    Author: Yoichi Yuasa <yyuasa@mvista.com or source@mvista.com>
- *  Copyright (C) 2004-2005  Yoichi Yuasa <yoichi_yuasa@tripeaks.co.jp>
+ *  Copyright (C) 2004-2008  Yoichi Yuasa <yoichi_yuasa@tripeaks.co.jp>
  *  Copyright (C) 2004 by Ralf Baechle (ralf@linux-mips.org)
  *
  *  This program is free software; you can redistribute it and/or modify
@@ -300,6 +300,18 @@ static int __init vr41xx_pciu_init(void)
                 ioport_resource.end = IO_PORT_RESOURCE_END;
         }
 
+        if (setup->master_io) {
+                void __iomem *io_map_base;
+                struct resource *res = vr41xx_pci_controller.io_resource;
+                master = setup->master_io;
+                io_map_base = ioremap(master->bus_base_address,
+                                      res->end - res->start + 1);
+                if (!io_map_base)
+                        return -EBUSY;
+
+                vr41xx_pci_controller.io_map_base = (unsigned long)io_map_base;
+        }
+
         register_pci_controller(&vr41xx_pci_controller);
 
         return 0;