ARM: Orion: SATA: Add per channel clk/clkdev support.

[firefly-linux-kernel-4.4.55.git] / drivers / ata / sata_mv.c

diff --git a/drivers/ata/sata_mv.c b/drivers/ata/sata_mv.c
index 7336d4a7ab317c091b7b991a1355a712f58023ab..24712adf69dfb8a876ff98d2d39df1182de03942 100644 (file)
--- a/drivers/ata/sata_mv.c
+++ b/drivers/ata/sata_mv.c
@@ -553,6 +553,7 @@ struct mv_host_priv {
 
 #if defined(CONFIG_HAVE_CLK)
        struct clk              *clk;
+       struct clk              **port_clks;
 #endif
        /*
         * These consistent DMA memory pools give us guaranteed
@@ -4027,6 +4028,9 @@ static int mv_platform_probe(struct platform_device *pdev)
        struct resource *res;
        int n_ports = 0;
        int rc;
+#if defined(CONFIG_HAVE_CLK)
+       int port;
+#endif
 
        ata_print_version_once(&pdev->dev, DRV_VERSION);
 
@@ -4054,6 +4058,13 @@ static int mv_platform_probe(struct platform_device *pdev)
 
        if (!host || !hpriv)
                return -ENOMEM;
+#if defined(CONFIG_HAVE_CLK)
+       hpriv->port_clks = devm_kzalloc(&pdev->dev,
+                                       sizeof(struct clk *) * n_ports,
+                                       GFP_KERNEL);
+       if (!hpriv->port_clks)
+               return -ENOMEM;
+#endif
        host->private_data = hpriv;
        hpriv->n_ports = n_ports;
        hpriv->board_idx = chip_soc;
@@ -4066,9 +4077,17 @@ static int mv_platform_probe(struct platform_device *pdev)
 #if defined(CONFIG_HAVE_CLK)
        hpriv->clk = clk_get(&pdev->dev, NULL);
        if (IS_ERR(hpriv->clk))
-               dev_notice(&pdev->dev, "cannot get clkdev\n");
+               dev_notice(&pdev->dev, "cannot get optional clkdev\n");
        else
-               clk_enable(hpriv->clk);
+               clk_prepare_enable(hpriv->clk);
+
+       for (port = 0; port < n_ports; port++) {
+               char port_number[16];
+               sprintf(port_number, "%d", port);
+               hpriv->port_clks[port] = clk_get(&pdev->dev, port_number);
+               if (!IS_ERR(hpriv->port_clks[port]))
+                       clk_prepare_enable(hpriv->port_clks[port]);
+       }
 #endif
 
        /*
@@ -4098,9 +4117,15 @@ static int mv_platform_probe(struct platform_device *pdev)
 err:
 #if defined(CONFIG_HAVE_CLK)
        if (!IS_ERR(hpriv->clk)) {
-               clk_disable(hpriv->clk);
+               clk_disable_unprepare(hpriv->clk);
                clk_put(hpriv->clk);
        }
+       for (port = 0; port < n_ports; port++) {
+               if (!IS_ERR(hpriv->port_clks[port])) {
+                       clk_disable_unprepare(hpriv->port_clks[port]);
+                       clk_put(hpriv->port_clks[port]);
+               }
+       }
 #endif
 
        return rc;
@@ -4119,14 +4144,21 @@ static int __devexit mv_platform_remove(struct platform_device *pdev)
        struct ata_host *host = platform_get_drvdata(pdev);
 #if defined(CONFIG_HAVE_CLK)
        struct mv_host_priv *hpriv = host->private_data;
+       int port;
 #endif
        ata_host_detach(host);
 
 #if defined(CONFIG_HAVE_CLK)
        if (!IS_ERR(hpriv->clk)) {
-               clk_disable(hpriv->clk);
+               clk_disable_unprepare(hpriv->clk);
                clk_put(hpriv->clk);
        }
+       for (port = 0; port < host->n_ports; port++) {
+               if (!IS_ERR(hpriv->port_clks[port])) {
+                       clk_disable_unprepare(hpriv->port_clks[port]);
+                       clk_put(hpriv->port_clks[port]);
+               }
+       }
 #endif
        return 0;
 }