{
        struct pci_dev *dev;
 
+       pci_dev_lock(bus->self);
        list_for_each_entry(dev, &bus->devices, bus_list) {
-               pci_dev_lock(dev);
                if (dev->subordinate)
                        pci_bus_lock(dev->subordinate);
+               else
+                       pci_dev_lock(dev);
        }
 }
 
        list_for_each_entry(dev, &bus->devices, bus_list) {
                if (dev->subordinate)
                        pci_bus_unlock(dev->subordinate);
-               pci_dev_unlock(dev);
+               else
+                       pci_dev_unlock(dev);
        }
+       pci_dev_unlock(bus->self);
 }
 
 /* Return 1 on successful lock, 0 on contention */
 {
        struct pci_dev *dev;
 
+       if (!pci_dev_trylock(bus->self))
+               return 0;
+
        list_for_each_entry(dev, &bus->devices, bus_list) {
-               if (!pci_dev_trylock(dev))
-                       goto unlock;
                if (dev->subordinate) {
-                       if (!pci_bus_trylock(dev->subordinate)) {
-                               pci_dev_unlock(dev);
+                       if (!pci_bus_trylock(dev->subordinate))
                                goto unlock;
-                       }
-               }
+               } else if (!pci_dev_trylock(dev))
+                       goto unlock;
        }
        return 1;
 
        list_for_each_entry_continue_reverse(dev, &bus->devices, bus_list) {
                if (dev->subordinate)
                        pci_bus_unlock(dev->subordinate);
-               pci_dev_unlock(dev);
+               else
+                       pci_dev_unlock(dev);
        }
+       pci_dev_unlock(bus->self);
        return 0;
 }
 
        list_for_each_entry(dev, &slot->bus->devices, bus_list) {
                if (!dev->slot || dev->slot != slot)
                        continue;
-               pci_dev_lock(dev);
                if (dev->subordinate)
                        pci_bus_lock(dev->subordinate);
+               else
+                       pci_dev_lock(dev);
        }
 }
 
        list_for_each_entry(dev, &slot->bus->devices, bus_list) {
                if (!dev->slot || dev->slot != slot)
                        continue;
-               if (!pci_dev_trylock(dev))
-                       goto unlock;
                if (dev->subordinate) {
                        if (!pci_bus_trylock(dev->subordinate)) {
                                pci_dev_unlock(dev);
                                goto unlock;
                        }
-               }
+               } else if (!pci_dev_trylock(dev))
+                       goto unlock;
        }
        return 1;
 
                        continue;
                if (dev->subordinate)
                        pci_bus_unlock(dev->subordinate);
-               pci_dev_unlock(dev);
+               else
+                       pci_dev_unlock(dev);
        }
        return 0;
 }