1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/pci/host/pci-hyperv.c b/drivers/pci/host/pci-hyperv.c
index 39fafda4deff..a1b3c1957d43 100644
--- a/drivers/pci/host/pci-hyperv.c
+++ b/drivers/pci/host/pci-hyperv.c
@@ -1209,9 +1209,11 @@ static int create_root_hv_pci_bus(struct hv_pcibus_device *hbus)
         hbus->pci_bus->msi = &hbus->msi_chip;
         hbus->pci_bus->msi->dev = &hbus->hdev->device;
 
+        pci_lock_rescan_remove();
         pci_scan_child_bus(hbus->pci_bus);
         pci_bus_assign_resources(hbus->pci_bus);
         pci_bus_add_devices(hbus->pci_bus);
+        pci_unlock_rescan_remove();
         hbus->state = hv_pcibus_installed;
         return 0;
 }
@@ -1612,8 +1614,10 @@ static void hv_eject_device_work(struct work_struct *work)
         pdev = pci_get_domain_bus_and_slot(hpdev->hbus->sysdata.domain, 0,
                                            wslot);
         if (pdev) {
+                pci_lock_rescan_remove();
                 pci_stop_and_remove_bus_device(pdev);
                 pci_dev_put(pdev);
+                pci_unlock_rescan_remove();
         }
 
         spin_lock_irqsave(&hpdev->hbus->device_list_lock, flags);