1 files changed, 60 insertions, 0 deletions

diff --git a/drivers/pci/pcie/portdrv_pci.c b/drivers/pci/pcie/portdrv_pci.c
index e0610bda1dea..3a7eefcb270a 100644
--- a/drivers/pci/pcie/portdrv_pci.c
+++ b/drivers/pci/pcie/portdrv_pci.c
@@ -11,6 +11,7 @@
 #include <linux/kernel.h>
 #include <linux/errno.h>
 #include <linux/pm.h>
+#include <linux/pm_runtime.h>
 #include <linux/init.h>
 #include <linux/pcieport_if.h>
 #include <linux/aer.h>
@@ -99,6 +100,51 @@ static int pcie_port_resume_noirq(struct device *dev)
         return 0;
 }
 
+#ifdef CONFIG_PM_RUNTIME
+struct d3cold_info {
+        bool no_d3cold;
+        unsigned int d3cold_delay;
+};
+
+static int pci_dev_d3cold_info(struct pci_dev *pdev, void *data)
+{
+        struct d3cold_info *info = data;
+
+        info->d3cold_delay = max_t(unsigned int, pdev->d3cold_delay,
+                                   info->d3cold_delay);
+        if (pdev->no_d3cold)
+                info->no_d3cold = true;
+        return 0;
+}
+
+static int pcie_port_runtime_suspend(struct device *dev)
+{
+        struct pci_dev *pdev = to_pci_dev(dev);
+        struct d3cold_info d3cold_info = {
+                .no_d3cold      = false,
+                .d3cold_delay   = PCI_PM_D3_WAIT,
+        };
+
+        /*
+         * If any subordinate device disable D3cold, we should not put
+         * the port into D3cold.  The D3cold delay of port should be
+         * the max of that of all subordinate devices.
+         */
+        pci_walk_bus(pdev->subordinate, pci_dev_d3cold_info, &d3cold_info);
+        pdev->no_d3cold = d3cold_info.no_d3cold;
+        pdev->d3cold_delay = d3cold_info.d3cold_delay;
+        return 0;
+}
+
+static int pcie_port_runtime_resume(struct device *dev)
+{
+        return 0;
+}
+#else
+#define pcie_port_runtime_suspend       NULL
+#define pcie_port_runtime_resume        NULL
+#endif
+
 static const struct dev_pm_ops pcie_portdrv_pm_ops = {
         .suspend        = pcie_port_device_suspend,
         .resume         = pcie_port_device_resume,
@@ -107,6 +153,8 @@ static const struct dev_pm_ops pcie_portdrv_pm_ops = {
         .poweroff       = pcie_port_device_suspend,
         .restore        = pcie_port_device_resume,
         .resume_noirq   = pcie_port_resume_noirq,
+        .runtime_suspend = pcie_port_runtime_suspend,
+        .runtime_resume = pcie_port_runtime_resume,
 };
 
 #define PCIE_PORTDRV_PM_OPS     (&pcie_portdrv_pm_ops)
@@ -117,6 +165,14 @@ static const struct dev_pm_ops pcie_portdrv_pm_ops = {
 #endif /* !PM */
 
 /*
+ * PCIe port runtime suspend is broken for some chipsets, so use a
+ * black list to disable runtime PM for these chipsets.
+ */
+static const struct pci_device_id port_runtime_pm_black_list[] = {
+        { /* end: all zeroes */ }
+};
+
+/*
  * pcie_portdrv_probe - Probe PCI-Express port devices
  * @dev: PCI-Express port device being probed
  *
@@ -144,12 +200,16 @@ static int __devinit pcie_portdrv_probe(struct pci_dev *dev,
                 return status;
 
         pci_save_state(dev);
+        if (!pci_match_id(port_runtime_pm_black_list, dev))
+                pm_runtime_put_noidle(&dev->dev);
 
         return 0;
 }
 
 static void pcie_portdrv_remove(struct pci_dev *dev)
 {
+        if (!pci_match_id(port_runtime_pm_black_list, dev))
+                pm_runtime_get_noresume(&dev->dev);
         pcie_port_device_remove(dev);
         pci_disable_device(dev);
 }