ata: ahci_tegra: suspend host if drive not present

Bug 200213053

Change-Id: I54f3322a13f839922a1a3da49d2eb24aee8b0ed0
Signed-off-by: Preetham Chandru R <pchandru@nvidia.com>
Reviewed-on: http://git-master/r/1173821
(cherry picked from commit b8c0276c9899693134394a130202120781404e16)

2 files changed, 32 insertions, 0 deletions

diff --git a/drivers/ata/tegra/ahci_tegra.c b/drivers/ata/tegra/ahci_tegra.c
index bbe777b2b..be7684126 100644
--- a/drivers/ata/tegra/ahci_tegra.c
+++ b/drivers/ata/tegra/ahci_tegra.c
@@ -38,6 +38,8 @@ static int tegra_ahci_suspend(struct device *dev);
 static int tegra_ahci_resume(struct device *dev);
 #endif
 
+static void tegra_ahci_shutdown(struct platform_device *pdev);
+
 static char * const tegra_rail_names[] = {};
 
 static const struct tegra_ahci_soc_data tegra_ahci_data = {
@@ -265,6 +267,34 @@ static int tegra_ahci_softreset(struct ata_link *link, unsigned int *class,
 
 }
 
+static void tegra_ahci_unbind(struct work_struct *work)
+{
+        struct tegra_ahci_priv *tegra =
+                        container_of(work, struct tegra_ahci_priv, work);
+        struct platform_device *pdev = tegra->pdev;
+
+        tegra_ahci_shutdown(pdev);
+        pm_runtime_put_sync(&pdev->dev);
+        pm_runtime_disable(&pdev->dev);
+        pdev->dev.driver->pm = NULL;
+        pdev->dev.driver->shutdown  = NULL;
+}
+
+static void tegra_ahci_error_handler(struct ata_port *ap)
+{
+        ahci_ops.error_handler(ap);
+        if (!ata_dev_enabled(ap->link.device)) {
+                if (!(ap->pflags & ATA_PFLAG_SUSPENDED)) {
+                        struct ata_host *host = ap->host;
+                        struct ahci_host_priv *hpriv =  host->private_data;
+                        struct tegra_ahci_priv *tegra = hpriv->plat_data;
+
+                        INIT_WORK(&tegra->work, tegra_ahci_unbind);
+                        schedule_work(&tegra->work);
+                }
+        }
+}
+
 static struct ata_port_operations ahci_tegra_port_ops = {
         .inherits       = &ahci_ops,
         .qc_issue       = tegra_ahci_qc_issue,
@@ -273,6 +303,7 @@ static struct ata_port_operations ahci_tegra_port_ops = {
         .port_resume    = tegra_ahci_port_resume,
         .hardreset      = tegra_ahci_hardreset,
         .softreset      = tegra_ahci_softreset,
+        .error_handler  = tegra_ahci_error_handler,
 };
 
 static struct ata_port_info ahci_tegra_port_info = {
diff --git a/drivers/ata/tegra/ahci_tegra.h b/drivers/ata/tegra/ahci_tegra.h
index da2c79731..c1ed476bf 100644
--- a/drivers/ata/tegra/ahci_tegra.h
+++ b/drivers/ata/tegra/ahci_tegra.h
@@ -208,6 +208,7 @@ struct tegra_ahci_priv {
         struct pinctrl_state       *devslp_active;
         struct pinctrl_state       *devslp_pullup;
         struct tegra_prod          *prod_list;
+        struct work_struct         work;
         int                        devslp_gpio;
         bool                       devslp_override;
         bool                       devslp_pinmux_override;