Just like removal.

Because We could add new bus under the bridges...

Signed-off-by: Yinghai Lu <[email protected]>
---
 drivers/pci/pci-sysfs.c |   43 +++++++++++++++++++++++++++++++------------
 1 files changed, 31 insertions(+), 12 deletions(-)

diff --git a/drivers/pci/pci-sysfs.c b/drivers/pci/pci-sysfs.c
index 60cb06b..7ebc764 100644
--- a/drivers/pci/pci-sysfs.c
+++ b/drivers/pci/pci-sysfs.c
@@ -327,21 +327,31 @@ dev_rescan_store(struct device *dev, struct 
device_attribute *attr,
        return count;
 }
 
+static void bridge_rescan_callback(struct device *dev)
+{
+       struct pci_dev *pdev = to_pci_dev(dev);
+
+       mutex_lock(&pci_remove_rescan_mutex);
+       pci_rescan_bus_bridge_resize(pdev);
+       mutex_unlock(&pci_remove_rescan_mutex);
+}
+
 static ssize_t
 dev_bridge_rescan_store(struct device *dev, struct device_attribute *attr,
                 const char *buf, size_t count)
 {
+       int ret = 0;
        unsigned long val;
-       struct pci_dev *pdev = to_pci_dev(dev);
 
        if (kstrtoul(buf, 0, &val) < 0)
                return -EINVAL;
 
-       if (val) {
-               mutex_lock(&pci_remove_rescan_mutex);
-               pci_rescan_bus_bridge_resize(pdev);
-               mutex_unlock(&pci_remove_rescan_mutex);
-       }
+       if (val)
+               ret = device_schedule_callback(dev, bridge_rescan_callback);
+
+       if (ret)
+               count = ret;
+
        return count;
 }
 
@@ -377,21 +387,30 @@ remove_store(struct device *dev, struct device_attribute 
*dummy,
        return count;
 }
 
+static void bus_rescan_callback(struct device *dev)
+{
+       struct pci_bus *bus = to_pci_bus(dev);
+
+       mutex_lock(&pci_remove_rescan_mutex);
+       pci_rescan_bus(bus);
+       mutex_unlock(&pci_remove_rescan_mutex);
+}
 static ssize_t
 dev_bus_rescan_store(struct device *dev, struct device_attribute *attr,
                 const char *buf, size_t count)
 {
+       int ret = 0;
        unsigned long val;
-       struct pci_bus *bus = to_pci_bus(dev);
 
        if (strict_strtoul(buf, 0, &val) < 0)
                return -EINVAL;
 
-       if (val) {
-               mutex_lock(&pci_remove_rescan_mutex);
-               pci_rescan_bus(bus);
-               mutex_unlock(&pci_remove_rescan_mutex);
-       }
+       if (val)
+               ret = device_schedule_callback(dev, bus_rescan_callback);
+
+       if (ret)
+               count = ret;
+
        return count;
 }
 
-- 
1.7.7

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [email protected]
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to