3 files changed, 20 insertions, 7 deletions

diff --git a/arch/powerpc/kernel/eeh_driver.c b/arch/powerpc/kernel/eeh_driver.c
index 36bed5a12750..c17f90d0f73c 100644
--- a/arch/powerpc/kernel/eeh_driver.c
+++ b/arch/powerpc/kernel/eeh_driver.c
@@ -369,7 +369,9 @@ static void *eeh_rmv_device(void *data, void *userdata)
         edev->mode |= EEH_DEV_DISCONNECTED;
         (*removed)++;
 
+        pci_lock_rescan_remove();
         pci_stop_and_remove_bus_device(dev);
+        pci_unlock_rescan_remove();
 
         return NULL;
 }
@@ -416,10 +418,13 @@ static int eeh_reset_device(struct eeh_pe *pe, struct pci_bus *bus)
          * into pcibios_add_pci_devices().
          */
         eeh_pe_state_mark(pe, EEH_PE_KEEP);
-        if (bus)
+        if (bus) {
+                pci_lock_rescan_remove();
                 pcibios_remove_pci_devices(bus);
-        else if (frozen_bus)
+                pci_unlock_rescan_remove();
+        } else if (frozen_bus) {
                 eeh_pe_dev_traverse(pe, eeh_rmv_device, &removed);
+        }
 
         /* Reset the pci controller. (Asserts RST#; resets config space).
          * Reconfigure bridges and devices. Don't try to bring the system
@@ -429,6 +434,8 @@ static int eeh_reset_device(struct eeh_pe *pe, struct pci_bus *bus)
         if (rc)
                 return rc;
 
+        pci_lock_rescan_remove();
+
         /* Restore PE */
         eeh_ops->configure_bridge(pe);
         eeh_pe_restore_bars(pe);
@@ -462,6 +469,7 @@ static int eeh_reset_device(struct eeh_pe *pe, struct pci_bus *bus)
         pe->tstamp = tstamp;
         pe->freeze_count = cnt;
 
+        pci_unlock_rescan_remove();
         return 0;
 }
 
@@ -618,8 +626,11 @@ perm_error:
         eeh_pe_dev_traverse(pe, eeh_report_failure, NULL);
 
         /* Shut down the device drivers for good. */
-        if (frozen_bus)
+        if (frozen_bus) {
+                pci_lock_rescan_remove();
                 pcibios_remove_pci_devices(frozen_bus);
+                pci_unlock_rescan_remove();
+        }
 }
 
 static void eeh_handle_special_event(void)
@@ -692,6 +703,7 @@ static void eeh_handle_special_event(void)
         if (rc == 2 || rc == 1)
                 eeh_handle_normal_event(pe);
         else {
+                pci_lock_rescan_remove();
                 list_for_each_entry_safe(hose, tmp,
                         &hose_list, list_node) {
                         phb_pe = eeh_phb_pe_get(hose);
@@ -703,6 +715,7 @@ static void eeh_handle_special_event(void)
                         eeh_pe_dev_traverse(pe, eeh_report_failure, NULL);
                         pcibios_remove_pci_devices(bus);
                 }
+                pci_unlock_rescan_remove();
         }
 }
 
diff --git a/arch/powerpc/kernel/pci-common.c b/arch/powerpc/kernel/pci-common.c
index a1e3e40ca3fd..d9476c1fc959 100644
--- a/arch/powerpc/kernel/pci-common.c
+++ b/arch/powerpc/kernel/pci-common.c
@@ -835,7 +835,7 @@ static void pcibios_fixup_resources(struct pci_dev *dev)
                  * at 0 as unset as well, except if PCI_PROBE_ONLY is also set
                  * since in that case, we don't want to re-assign anything
                  */
-                pcibios_resource_to_bus(dev, &reg, res);
+                pcibios_resource_to_bus(dev->bus, &reg, res);
                 if (pci_has_flag(PCI_REASSIGN_ALL_RSRC) ||
                     (reg.start == 0 && !pci_has_flag(PCI_PROBE_ONLY))) {
                         /* Only print message if not re-assigning */
@@ -886,7 +886,7 @@ static int pcibios_uninitialized_bridge_resource(struct pci_bus *bus,
 
         /* Job is a bit different between memory and IO */
         if (res->flags & IORESOURCE_MEM) {
-                pcibios_resource_to_bus(dev, &region, res);
+                pcibios_resource_to_bus(dev->bus, &region, res);
 
                 /* If the BAR is non-0 then it's probably been initialized */
                 if (region.start != 0)
diff --git a/arch/powerpc/kernel/pci_of_scan.c b/arch/powerpc/kernel/pci_of_scan.c
index ac0b034f9ae0..83c26d829991 100644
--- a/arch/powerpc/kernel/pci_of_scan.c
+++ b/arch/powerpc/kernel/pci_of_scan.c
@@ -111,7 +111,7 @@ static void of_pci_parse_addrs(struct device_node *node, struct pci_dev *dev)
                 res->name = pci_name(dev);
                 region.start = base;
                 region.end = base + size - 1;
-                pcibios_bus_to_resource(dev, res, &region);
+                pcibios_bus_to_resource(dev->bus, res, &region);
         }
 }
 
@@ -280,7 +280,7 @@ void of_scan_pci_bridge(struct pci_dev *dev)
                 res->flags = flags;
                 region.start = of_read_number(&ranges[1], 2);
                 region.end = region.start + size - 1;
-                pcibios_bus_to_resource(dev, res, &region);
+                pcibios_bus_to_resource(dev->bus, res, &region);
         }
         sprintf(bus->name, "PCI Bus %04x:%02x", pci_domain_nr(bus),
                 bus->number);