text: /*
   * Freescale iMX PATA driver
   *
   * Copyright (C) 2011 Arnaud Patard <arnaud.patard@rtp-net.org>
   *
   * Based on pata_platform - Copyright (C) 2006 - 2007  Paul Mundt
   *
   * This file is subject to the terms and conditions of the GNU General Public
   * License.  See the file "COPYING" in the main directory of this archive
   * for more details.
   *
   * TODO:
   * - dmaengine support
   * - check if timing stuff needed
   */
  #include <linux/kernel.h>
  #include <linux/module.h>
  #include <linux/blkdev.h>
  #include <scsi/scsi_host.h>
  #include <linux/ata.h>
  #include <linux/libata.h>
  #include <linux/platform_device.h>
  #include <linux/clk.h>
  
  #define DRV_NAME "pata_imx"
  
  #define PATA_IMX_ATA_CONTROL		0x24
  #define PATA_IMX_ATA_CTRL_FIFO_RST_B	(1<<7)
  #define PATA_IMX_ATA_CTRL_ATA_RST_B	(1<<6)
  #define PATA_IMX_ATA_CTRL_IORDY_EN	(1<<0)
  #define PATA_IMX_ATA_INT_EN		0x2C
  #define PATA_IMX_ATA_INTR_ATA_INTRQ2	(1<<3)
  #define PATA_IMX_DRIVE_DATA		0xA0
  #define PATA_IMX_DRIVE_CONTROL		0xD8
  
  struct pata_imx_priv {
  	struct clk *clk;
  	/* timings/interrupt/control regs */
  	void __iomem *host_regs;
  	u32 ata_ctl;
  };
  
  static int pata_imx_set_mode(struct ata_link *link, struct ata_device **unused)
  {
  	struct ata_device *dev;
  	struct ata_port *ap = link->ap;
  	struct pata_imx_priv *priv = ap->host->private_data;
  	u32 val;
  
  	ata_for_each_dev(dev, link, ENABLED) {
  		dev->pio_mode = dev->xfer_mode = XFER_PIO_0;
  		dev->xfer_shift = ATA_SHIFT_PIO;
  		dev->flags |= ATA_DFLAG_PIO;
  
  		val = __raw_readl(priv->host_regs + PATA_IMX_ATA_CONTROL);
  		if (ata_pio_need_iordy(dev))
  			val |= PATA_IMX_ATA_CTRL_IORDY_EN;
  		else
  			val &= ~PATA_IMX_ATA_CTRL_IORDY_EN;
  		__raw_writel(val, priv->host_regs + PATA_IMX_ATA_CONTROL);
  
  		ata_dev_info(dev, "configured for PIO\n");
  	}
  	return 0;
  }
  
  static struct scsi_host_template pata_imx_sht = {
  	ATA_PIO_SHT(DRV_NAME),
  };
  
  static struct ata_port_operations pata_imx_port_ops = {
  	.inherits		= &ata_sff_port_ops,
  	.sff_data_xfer		= ata_sff_data_xfer_noirq,
  	.cable_detect		= ata_cable_unknown,
  	.set_mode		= pata_imx_set_mode,
  };
  
  static void pata_imx_setup_port(struct ata_ioports *ioaddr)
  {
  	/* Fixup the port shift for platforms that need it */
  	ioaddr->data_addr	= ioaddr->cmd_addr + (ATA_REG_DATA    << 2);
  	ioaddr->error_addr	= ioaddr->cmd_addr + (ATA_REG_ERR     << 2);
  	ioaddr->feature_addr	= ioaddr->cmd_addr + (ATA_REG_FEATURE << 2);
  	ioaddr->nsect_addr	= ioaddr->cmd_addr + (ATA_REG_NSECT   << 2);
  	ioaddr->lbal_addr	= ioaddr->cmd_addr + (ATA_REG_LBAL    << 2);
  	ioaddr->lbam_addr	= ioaddr->cmd_addr + (ATA_REG_LBAM    << 2);
  	ioaddr->lbah_addr	= ioaddr->cmd_addr + (ATA_REG_LBAH    << 2);
  	ioaddr->device_addr	= ioaddr->cmd_addr + (ATA_REG_DEVICE  << 2);
  	ioaddr->status_addr	= ioaddr->cmd_addr + (ATA_REG_STATUS  << 2);
  	ioaddr->command_addr	= ioaddr->cmd_addr + (ATA_REG_CMD     << 2);
  }
  
  static int pata_imx_probe(struct platform_device *pdev)
  {
  	struct ata_host *host;
  	struct ata_port *ap;
  	struct pata_imx_priv *priv;
  	int irq = 0;
  	struct resource *io_res;
  	int ret;
  
  	irq = platform_get_irq(pdev, 0);
  	if (irq < 0)
  		return irq;
  
  	priv = devm_kzalloc(&pdev->dev,
  				sizeof(struct pata_imx_priv), GFP_KERNEL);
  	if (!priv)
  		return -ENOMEM;
  
  	priv->clk = devm_clk_get(&pdev->dev, NULL);
  	if (IS_ERR(priv->clk)) {
  		dev_err(&pdev->dev, "Failed to get clock\n");
  		return PTR_ERR(priv->clk);
  	}
  
  	ret = clk_prepare_enable(priv->clk);
  	if (ret)
  		return ret;
  
  	host = ata_host_alloc(&pdev->dev, 1);
  	if (!host) {
  		ret = -ENOMEM;
  		goto err;
  	}
  
  	host->private_data = priv;
  	ap = host->ports[0];
  
  	ap->ops = &pata_imx_port_ops;
  	ap->pio_mask = ATA_PIO0;
  	ap->flags |= ATA_FLAG_SLAVE_POSS;
  
  	io_res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
  	priv->host_regs = devm_ioremap_resource(&pdev->dev, io_res);
  	if (IS_ERR(priv->host_regs)) {
  		ret = PTR_ERR(priv->host_regs);
  		goto err;
  	}
  
  	ap->ioaddr.cmd_addr = priv->host_regs + PATA_IMX_DRIVE_DATA;
  	ap->ioaddr.ctl_addr = priv->host_regs + PATA_IMX_DRIVE_CONTROL;
  
  	ap->ioaddr.altstatus_addr = ap->ioaddr.ctl_addr;
  
  	pata_imx_setup_port(&ap->ioaddr);
  
  	ata_port_desc(ap, "cmd 0x%llx ctl 0x%llx",
  		(unsigned long long)io_res->start + PATA_IMX_DRIVE_DATA,
  		(unsigned long long)io_res->start + PATA_IMX_DRIVE_CONTROL);
  
  	/* deassert resets */
  	__raw_writel(PATA_IMX_ATA_CTRL_FIFO_RST_B |
  			PATA_IMX_ATA_CTRL_ATA_RST_B,
  			priv->host_regs + PATA_IMX_ATA_CONTROL);
  	/* enable interrupts */
  	__raw_writel(PATA_IMX_ATA_INTR_ATA_INTRQ2,
  			priv->host_regs + PATA_IMX_ATA_INT_EN);
  
  	/* activate */
  	ret = ata_host_activate(host, irq, ata_sff_interrupt, 0,
  				&pata_imx_sht);
  
  	if (ret)
  		goto err;
  
  	return 0;
  err:
  	clk_disable_unprepare(priv->clk);
  
  	return ret;
  }
  
  static int pata_imx_remove(struct platform_device *pdev)
  {
  	struct ata_host *host = platform_get_drvdata(pdev);
  	struct pata_imx_priv *priv = host->private_data;
  
  	ata_host_detach(host);
  
  	__raw_writel(0, priv->host_regs + PATA_IMX_ATA_INT_EN);
  
  	clk_disable_unprepare(priv->clk);
  
  	return 0;
  }
  
  #ifdef CONFIG_PM_SLEEP
  static int pata_imx_suspend(struct device *dev)
  {
  	struct ata_host *host = dev_get_drvdata(dev);
  	struct pata_imx_priv *priv = host->private_data;
  	int ret;
  
  	ret = ata_host_suspend(host, PMSG_SUSPEND);
  	if (!ret) {
  		__raw_writel(0, priv->host_regs + PATA_IMX_ATA_INT_EN);
  		priv->ata_ctl =
  			__raw_readl(priv->host_regs + PATA_IMX_ATA_CONTROL);
  		clk_disable_unprepare(priv->clk);
  	}
  
  	return ret;
  }
  
  static int pata_imx_resume(struct device *dev)
  {
  	struct ata_host *host = dev_get_drvdata(dev);
  	struct pata_imx_priv *priv = host->private_data;
  
  	int ret = clk_prepare_enable(priv->clk);
  	if (ret)
  		return ret;
  
  	__raw_writel(priv->ata_ctl, priv->host_regs + PATA_IMX_ATA_CONTROL);
  
  	__raw_writel(PATA_IMX_ATA_INTR_ATA_INTRQ2,
  			priv->host_regs + PATA_IMX_ATA_INT_EN);
  
  	ata_host_resume(host);
  
  	return 0;
  }
  #endif
  
  static SIMPLE_DEV_PM_OPS(pata_imx_pm_ops, pata_imx_suspend, pata_imx_resume);
  
  static const struct of_device_id imx_pata_dt_ids[] = {
  	{
  		.compatible = "fsl,imx27-pata",
  	}, {
  		/* sentinel */
  	}
  };
  MODULE_DEVICE_TABLE(of, imx_pata_dt_ids);
  
  static struct platform_driver pata_imx_driver = {
  	.probe		= pata_imx_probe,
  	.remove		= pata_imx_remove,
  	.driver = {
  		.name		= DRV_NAME,
  		.of_match_table	= imx_pata_dt_ids,
  		.owner		= THIS_MODULE,
  		.pm		= &pata_imx_pm_ops,
  	},
  };
  
  module_platform_driver(pata_imx_driver);
  
  MODULE_AUTHOR("Arnaud Patard <arnaud.patard@rtp-net.org>");
  MODULE_DESCRIPTION("low-level driver for iMX PATA");
  MODULE_LICENSE("GPL");
  MODULE_ALIAS("platform:" DRV_NAME);
  