3 files changed, 23 insertions, 10 deletions

diff --git a/drivers/pci/pci-driver.c b/drivers/pci/pci-driver.c
index 840fdc5ba0d8..454853507b7e 100644
--- a/drivers/pci/pci-driver.c
+++ b/drivers/pci/pci-driver.c
@@ -267,11 +267,19 @@ static long local_pci_probe(void *_ddi)
         pm_runtime_get_sync(dev);
         pci_dev->driver = pci_drv;
         rc = pci_drv->probe(pci_dev, ddi->id);
-        if (rc) {
+        if (!rc)
+                return rc;
+        if (rc < 0) {
                 pci_dev->driver = NULL;
                 pm_runtime_put_sync(dev);
+                return rc;
         }
-        return rc;
+        /*
+         * Probe function should return < 0 for failure, 0 for success
+         * Treat values > 0 as success, but warn.
+         */
+        dev_warn(dev, "Driver probe function unexpectedly returned %d\n", rc);
+        return 0;
 }
 
 static int pci_call_probe(struct pci_driver *drv, struct pci_dev *dev,
diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
index 457f801fc49d..e485d92b3fa8 100644
--- a/drivers/pci/pci.c
+++ b/drivers/pci/pci.c
@@ -1155,8 +1155,12 @@ static void pci_enable_bridge(struct pci_dev *dev)
 
         pci_enable_bridge(dev->bus->self);
 
-        if (pci_is_enabled(dev))
+        if (pci_is_enabled(dev)) {
+                if (!dev->is_busmaster)
+                        pci_set_master(dev);
                 return;
+        }
+
         retval = pci_enable_device(dev);
         if (retval)
                 dev_err(&dev->dev, "Error enabling bridge (%d), continuing\n",
diff --git a/drivers/pci/pcie/portdrv_core.c b/drivers/pci/pcie/portdrv_core.c
index 31063ac30992..08d131f7815b 100644
--- a/drivers/pci/pcie/portdrv_core.c
+++ b/drivers/pci/pcie/portdrv_core.c
@@ -260,13 +260,14 @@ static int get_port_device_capability(struct pci_dev *dev)
         if (pcie_ports_disabled)
                 return 0;
 
-        err = pcie_port_platform_notify(dev, &cap_mask);
-        if (!pcie_ports_auto) {
-                cap_mask = PCIE_PORT_SERVICE_PME | PCIE_PORT_SERVICE_HP
-                                | PCIE_PORT_SERVICE_VC;
-                if (pci_aer_available())
-                        cap_mask |= PCIE_PORT_SERVICE_AER;
-        } else if (err) {
+        cap_mask = PCIE_PORT_SERVICE_PME | PCIE_PORT_SERVICE_HP
+                        | PCIE_PORT_SERVICE_VC;
+        if (pci_aer_available())
+                cap_mask |= PCIE_PORT_SERVICE_AER;
+
+        if (pcie_ports_auto) {
+                err = pcie_port_platform_notify(dev, &cap_mask);
+                if (err)
                         return 0;
         }