1 files changed, 38 insertions, 0 deletions

diff --git a/drivers/pci/host/pci-mvebu.c b/drivers/pci/host/pci-mvebu.c
index 1309cfbaa719..1ab863551920 100644
--- a/drivers/pci/host/pci-mvebu.c
+++ b/drivers/pci/host/pci-mvebu.c
@@ -129,6 +129,7 @@ struct mvebu_pcie_port {
         size_t memwin_size;
         phys_addr_t iowin_base;
         size_t iowin_size;
+        u32 saved_pcie_stat;
 };
 
 static inline void mvebu_writel(struct mvebu_pcie_port *port, u32 val, u32 reg)
@@ -899,6 +900,35 @@ static void mvebu_pcie_msi_enable(struct mvebu_pcie *pcie)
                 pcie->msi->dev = &pcie->pdev->dev;
 }
 
+static int mvebu_pcie_suspend(struct device *dev)
+{
+        struct mvebu_pcie *pcie;
+        int i;
+
+        pcie = dev_get_drvdata(dev);
+        for (i = 0; i < pcie->nports; i++) {
+                struct mvebu_pcie_port *port = pcie->ports + i;
+                port->saved_pcie_stat = mvebu_readl(port, PCIE_STAT_OFF);
+        }
+
+        return 0;
+}
+
+static int mvebu_pcie_resume(struct device *dev)
+{
+        struct mvebu_pcie *pcie;
+        int i;
+
+        pcie = dev_get_drvdata(dev);
+        for (i = 0; i < pcie->nports; i++) {
+                struct mvebu_pcie_port *port = pcie->ports + i;
+                mvebu_writel(port, port->saved_pcie_stat, PCIE_STAT_OFF);
+                mvebu_pcie_setup_hw(port);
+        }
+
+        return 0;
+}
+
 static int mvebu_pcie_probe(struct platform_device *pdev)
 {
         struct mvebu_pcie *pcie;
@@ -1056,6 +1086,8 @@ static int mvebu_pcie_probe(struct platform_device *pdev)
         mvebu_pcie_msi_enable(pcie);
         mvebu_pcie_enable(pcie);
 
+        platform_set_drvdata(pdev, pcie);
+
         return 0;
 }
 
@@ -1068,12 +1100,18 @@ static const struct of_device_id mvebu_pcie_of_match_table[] = {
 };
 MODULE_DEVICE_TABLE(of, mvebu_pcie_of_match_table);
 
+static struct dev_pm_ops mvebu_pcie_pm_ops = {
+        .suspend_noirq = mvebu_pcie_suspend,
+        .resume_noirq = mvebu_pcie_resume,
+};
+
 static struct platform_driver mvebu_pcie_driver = {
         .driver = {
                 .name = "mvebu-pcie",
                 .of_match_table = mvebu_pcie_of_match_table,
                 /* driver unloading/unbinding currently not supported */
                 .suppress_bind_attrs = true,
+                .pm = &mvebu_pcie_pm_ops,
         },
         .probe = mvebu_pcie_probe,
 };