SAFE public projects git trees. - safe/jmp/linux-2.6/blob - drivers/scsi/ahci.c

   1 /*
   2  *  ahci.c - AHCI SATA support
   3  *
   4  *  Maintained by:  Jeff Garzik <jgarzik@pobox.com>
   5  *                  Please ALWAYS copy linux-ide@vger.kernel.org
   6  *                  on emails.
   7  *
   8  *  Copyright 2004-2005 Red Hat, Inc.
   9  *
  10  *
  11  *  This program is free software; you can redistribute it and/or modify
  12  *  it under the terms of the GNU General Public License as published by
  13  *  the Free Software Foundation; either version 2, or (at your option)
  14  *  any later version.
  15  *
  16  *  This program is distributed in the hope that it will be useful,
  17  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  *  GNU General Public License for more details.
  20  *
  21  *  You should have received a copy of the GNU General Public License
  22  *  along with this program; see the file COPYING.  If not, write to
  23  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  24  *
  25  *
  26  * libata documentation is available via 'make {ps|pdf}docs',
  27  * as Documentation/DocBook/libata.*
  28  *
  29  * AHCI hardware documentation:
  30  * http://www.intel.com/technology/serialata/pdf/rev1_0.pdf
  31  * http://www.intel.com/technology/serialata/pdf/rev1_1.pdf
  32  *
  33  */
  34
  35 #include <linux/kernel.h>
  36 #include <linux/module.h>
  37 #include <linux/pci.h>
  38 #include <linux/init.h>
  39 #include <linux/blkdev.h>
  40 #include <linux/delay.h>
  41 #include <linux/interrupt.h>
  42 #include <linux/sched.h>
  43 #include <linux/dma-mapping.h>
  44 #include <linux/device.h>
  45 #include <scsi/scsi_host.h>
  46 #include <scsi/scsi_cmnd.h>
  47 #include <linux/libata.h>
  48 #include <asm/io.h>
  49
  50 #define DRV_NAME        "ahci"
  51 #define DRV_VERSION     "1.2"
  52
  53
  54 enum {
  55         AHCI_PCI_BAR            = 5,
  56         AHCI_MAX_SG             = 168, /* hardware max is 64K */
  57         AHCI_DMA_BOUNDARY       = 0xffffffff,
  58         AHCI_USE_CLUSTERING     = 0,
  59         AHCI_CMD_SLOT_SZ        = 32 * 32,
  60         AHCI_RX_FIS_SZ          = 256,
  61         AHCI_CMD_TBL_HDR        = 0x80,
  62         AHCI_CMD_TBL_CDB        = 0x40,
  63         AHCI_CMD_TBL_SZ         = AHCI_CMD_TBL_HDR + (AHCI_MAX_SG * 16),
  64         AHCI_PORT_PRIV_DMA_SZ   = AHCI_CMD_SLOT_SZ + AHCI_CMD_TBL_SZ +
  65                                   AHCI_RX_FIS_SZ,
  66         AHCI_IRQ_ON_SG          = (1 << 31),
  67         AHCI_CMD_ATAPI          = (1 << 5),
  68         AHCI_CMD_WRITE          = (1 << 6),
  69         AHCI_CMD_RESET          = (1 << 8),
  70         AHCI_CMD_CLR_BUSY       = (1 << 10),
  71
  72         RX_FIS_D2H_REG          = 0x40, /* offset of D2H Register FIS data */
  73
  74         board_ahci              = 0,
  75
  76         /* global controller registers */
  77         HOST_CAP                = 0x00, /* host capabilities */
  78         HOST_CTL                = 0x04, /* global host control */
  79         HOST_IRQ_STAT           = 0x08, /* interrupt status */
  80         HOST_PORTS_IMPL         = 0x0c, /* bitmap of implemented ports */
  81         HOST_VERSION            = 0x10, /* AHCI spec. version compliancy */
  82
  83         /* HOST_CTL bits */
  84         HOST_RESET              = (1 << 0),  /* reset controller; self-clear */
  85         HOST_IRQ_EN             = (1 << 1),  /* global IRQ enable */
  86         HOST_AHCI_EN            = (1 << 31), /* AHCI enabled */
  87
  88         /* HOST_CAP bits */
  89         HOST_CAP_64             = (1 << 31), /* PCI DAC (64-bit DMA) support */
  90         HOST_CAP_CLO            = (1 << 24), /* Command List Override support */
  91
  92         /* registers for each SATA port */
  93         PORT_LST_ADDR           = 0x00, /* command list DMA addr */
  94         PORT_LST_ADDR_HI        = 0x04, /* command list DMA addr hi */
  95         PORT_FIS_ADDR           = 0x08, /* FIS rx buf addr */
  96         PORT_FIS_ADDR_HI        = 0x0c, /* FIS rx buf addr hi */
  97         PORT_IRQ_STAT           = 0x10, /* interrupt status */
  98         PORT_IRQ_MASK           = 0x14, /* interrupt enable/disable mask */
  99         PORT_CMD                = 0x18, /* port command */
 100         PORT_TFDATA             = 0x20, /* taskfile data */
 101         PORT_SIG                = 0x24, /* device TF signature */
 102         PORT_CMD_ISSUE          = 0x38, /* command issue */
 103         PORT_SCR                = 0x28, /* SATA phy register block */
 104         PORT_SCR_STAT           = 0x28, /* SATA phy register: SStatus */
 105         PORT_SCR_CTL            = 0x2c, /* SATA phy register: SControl */
 106         PORT_SCR_ERR            = 0x30, /* SATA phy register: SError */
 107         PORT_SCR_ACT            = 0x34, /* SATA phy register: SActive */
 108
 109         /* PORT_IRQ_{STAT,MASK} bits */
 110         PORT_IRQ_COLD_PRES      = (1 << 31), /* cold presence detect */
 111         PORT_IRQ_TF_ERR         = (1 << 30), /* task file error */
 112         PORT_IRQ_HBUS_ERR       = (1 << 29), /* host bus fatal error */
 113         PORT_IRQ_HBUS_DATA_ERR  = (1 << 28), /* host bus data error */
 114         PORT_IRQ_IF_ERR         = (1 << 27), /* interface fatal error */
 115         PORT_IRQ_IF_NONFATAL    = (1 << 26), /* interface non-fatal error */
 116         PORT_IRQ_OVERFLOW       = (1 << 24), /* xfer exhausted available S/G */
 117         PORT_IRQ_BAD_PMP        = (1 << 23), /* incorrect port multiplier */
 118
 119         PORT_IRQ_PHYRDY         = (1 << 22), /* PhyRdy changed */
 120         PORT_IRQ_DEV_ILCK       = (1 << 7), /* device interlock */
 121         PORT_IRQ_CONNECT        = (1 << 6), /* port connect change status */
 122         PORT_IRQ_SG_DONE        = (1 << 5), /* descriptor processed */
 123         PORT_IRQ_UNK_FIS        = (1 << 4), /* unknown FIS rx'd */
 124         PORT_IRQ_SDB_FIS        = (1 << 3), /* Set Device Bits FIS rx'd */
 125         PORT_IRQ_DMAS_FIS       = (1 << 2), /* DMA Setup FIS rx'd */
 126         PORT_IRQ_PIOS_FIS       = (1 << 1), /* PIO Setup FIS rx'd */
 127         PORT_IRQ_D2H_REG_FIS    = (1 << 0), /* D2H Register FIS rx'd */
 128
 129         PORT_IRQ_FATAL          = PORT_IRQ_TF_ERR |
 130                                   PORT_IRQ_HBUS_ERR |
 131                                   PORT_IRQ_HBUS_DATA_ERR |
 132                                   PORT_IRQ_IF_ERR,
 133         DEF_PORT_IRQ            = PORT_IRQ_FATAL | PORT_IRQ_PHYRDY |
 134                                   PORT_IRQ_CONNECT | PORT_IRQ_SG_DONE |
 135                                   PORT_IRQ_UNK_FIS | PORT_IRQ_SDB_FIS |
 136                                   PORT_IRQ_DMAS_FIS | PORT_IRQ_PIOS_FIS |
 137                                   PORT_IRQ_D2H_REG_FIS,
 138
 139         /* PORT_CMD bits */
 140         PORT_CMD_ATAPI          = (1 << 24), /* Device is ATAPI */
 141         PORT_CMD_LIST_ON        = (1 << 15), /* cmd list DMA engine running */
 142         PORT_CMD_FIS_ON         = (1 << 14), /* FIS DMA engine running */
 143         PORT_CMD_FIS_RX         = (1 << 4), /* Enable FIS receive DMA engine */
 144         PORT_CMD_CLO            = (1 << 3), /* Command list override */
 145         PORT_CMD_POWER_ON       = (1 << 2), /* Power up device */
 146         PORT_CMD_SPIN_UP        = (1 << 1), /* Spin up device */
 147         PORT_CMD_START          = (1 << 0), /* Enable port DMA engine */
 148
 149         PORT_CMD_ICC_ACTIVE     = (0x1 << 28), /* Put i/f in active state */
 150         PORT_CMD_ICC_PARTIAL    = (0x2 << 28), /* Put i/f in partial state */
 151         PORT_CMD_ICC_SLUMBER    = (0x6 << 28), /* Put i/f in slumber state */
 152
 153         /* hpriv->flags bits */
 154         AHCI_FLAG_MSI           = (1 << 0),
 155 };
 156
 157 struct ahci_cmd_hdr {
 158         u32                     opts;
 159         u32                     status;
 160         u32                     tbl_addr;
 161         u32                     tbl_addr_hi;
 162         u32                     reserved[4];
 163 };
 164
 165 struct ahci_sg {
 166         u32                     addr;
 167         u32                     addr_hi;
 168         u32                     reserved;
 169         u32                     flags_size;
 170 };
 171
 172 struct ahci_host_priv {
 173         unsigned long           flags;
 174         u32                     cap;    /* cache of HOST_CAP register */
 175         u32                     port_map; /* cache of HOST_PORTS_IMPL reg */
 176 };
 177
 178 struct ahci_port_priv {
 179         struct ahci_cmd_hdr     *cmd_slot;
 180         dma_addr_t              cmd_slot_dma;
 181         void                    *cmd_tbl;
 182         dma_addr_t              cmd_tbl_dma;
 183         struct ahci_sg          *cmd_tbl_sg;
 184         void                    *rx_fis;
 185         dma_addr_t              rx_fis_dma;
 186 };
 187
 188 static u32 ahci_scr_read (struct ata_port *ap, unsigned int sc_reg);
 189 static void ahci_scr_write (struct ata_port *ap, unsigned int sc_reg, u32 val);
 190 static int ahci_init_one (struct pci_dev *pdev, const struct pci_device_id *ent);
 191 static unsigned int ahci_qc_issue(struct ata_queued_cmd *qc);
 192 static irqreturn_t ahci_interrupt (int irq, void *dev_instance, struct pt_regs *regs);
 193 static int ahci_probe_reset(struct ata_port *ap, unsigned int *classes);
 194 static void ahci_irq_clear(struct ata_port *ap);
 195 static void ahci_eng_timeout(struct ata_port *ap);
 196 static int ahci_port_start(struct ata_port *ap);
 197 static void ahci_port_stop(struct ata_port *ap);
 198 static void ahci_tf_read(struct ata_port *ap, struct ata_taskfile *tf);
 199 static void ahci_qc_prep(struct ata_queued_cmd *qc);
 200 static u8 ahci_check_status(struct ata_port *ap);
 201 static inline int ahci_host_intr(struct ata_port *ap, struct ata_queued_cmd *qc);
 202 static void ahci_remove_one (struct pci_dev *pdev);
 203
 204 static struct scsi_host_template ahci_sht = {
 205         .module                 = THIS_MODULE,
 206         .name                   = DRV_NAME,
 207         .ioctl                  = ata_scsi_ioctl,
 208         .queuecommand           = ata_scsi_queuecmd,
 209         .eh_timed_out           = ata_scsi_timed_out,
 210         .eh_strategy_handler    = ata_scsi_error,
 211         .can_queue              = ATA_DEF_QUEUE,
 212         .this_id                = ATA_SHT_THIS_ID,
 213         .sg_tablesize           = AHCI_MAX_SG,
 214         .max_sectors            = ATA_MAX_SECTORS,
 215         .cmd_per_lun            = ATA_SHT_CMD_PER_LUN,
 216         .emulated               = ATA_SHT_EMULATED,
 217         .use_clustering         = AHCI_USE_CLUSTERING,
 218         .proc_name              = DRV_NAME,
 219         .dma_boundary           = AHCI_DMA_BOUNDARY,
 220         .slave_configure        = ata_scsi_slave_config,
 221         .bios_param             = ata_std_bios_param,
 222 };
 223
 224 static const struct ata_port_operations ahci_ops = {
 225         .port_disable           = ata_port_disable,
 226
 227         .check_status           = ahci_check_status,
 228         .check_altstatus        = ahci_check_status,
 229         .dev_select             = ata_noop_dev_select,
 230
 231         .tf_read                = ahci_tf_read,
 232
 233         .probe_reset            = ahci_probe_reset,
 234
 235         .qc_prep                = ahci_qc_prep,
 236         .qc_issue               = ahci_qc_issue,
 237
 238         .eng_timeout            = ahci_eng_timeout,
 239
 240         .irq_handler            = ahci_interrupt,
 241         .irq_clear              = ahci_irq_clear,
 242
 243         .scr_read               = ahci_scr_read,
 244         .scr_write              = ahci_scr_write,
 245
 246         .port_start             = ahci_port_start,
 247         .port_stop              = ahci_port_stop,
 248 };
 249
 250 static const struct ata_port_info ahci_port_info[] = {
 251         /* board_ahci */
 252         {
 253                 .sht            = &ahci_sht,
 254                 .host_flags     = ATA_FLAG_SATA | ATA_FLAG_NO_LEGACY |
 255                                   ATA_FLAG_MMIO | ATA_FLAG_PIO_DMA,
 256                 .pio_mask       = 0x1f, /* pio0-4 */
 257                 .udma_mask      = 0x7f, /* udma0-6 ; FIXME */
 258                 .port_ops       = &ahci_ops,
 259         },
 260 };
 261
 262 static const struct pci_device_id ahci_pci_tbl[] = {
 263         { PCI_VENDOR_ID_INTEL, 0x2652, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 264           board_ahci }, /* ICH6 */
 265         { PCI_VENDOR_ID_INTEL, 0x2653, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 266           board_ahci }, /* ICH6M */
 267         { PCI_VENDOR_ID_INTEL, 0x27c1, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 268           board_ahci }, /* ICH7 */
 269         { PCI_VENDOR_ID_INTEL, 0x27c5, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 270           board_ahci }, /* ICH7M */
 271         { PCI_VENDOR_ID_INTEL, 0x27c3, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 272           board_ahci }, /* ICH7R */
 273         { PCI_VENDOR_ID_AL, 0x5288, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 274           board_ahci }, /* ULi M5288 */
 275         { PCI_VENDOR_ID_INTEL, 0x2681, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 276           board_ahci }, /* ESB2 */
 277         { PCI_VENDOR_ID_INTEL, 0x2682, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 278           board_ahci }, /* ESB2 */
 279         { PCI_VENDOR_ID_INTEL, 0x2683, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 280           board_ahci }, /* ESB2 */
 281         { PCI_VENDOR_ID_INTEL, 0x27c6, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 282           board_ahci }, /* ICH7-M DH */
 283         { PCI_VENDOR_ID_INTEL, 0x2821, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 284           board_ahci }, /* ICH8 */
 285         { PCI_VENDOR_ID_INTEL, 0x2822, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 286           board_ahci }, /* ICH8 */
 287         { PCI_VENDOR_ID_INTEL, 0x2824, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 288           board_ahci }, /* ICH8 */
 289         { PCI_VENDOR_ID_INTEL, 0x2829, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 290           board_ahci }, /* ICH8M */
 291         { PCI_VENDOR_ID_INTEL, 0x282a, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 292           board_ahci }, /* ICH8M */
 293         { 0x197b, 0x2360, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 294           board_ahci }, /* JMicron JMB360 */
 295         { 0x197b, 0x2363, PCI_ANY_ID, PCI_ANY_ID, 0, 0,
 296           board_ahci }, /* JMicron JMB363 */
 297         { }     /* terminate list */
 298 };
 299
 300
 301 static struct pci_driver ahci_pci_driver = {
 302         .name                   = DRV_NAME,
 303         .id_table               = ahci_pci_tbl,
 304         .probe                  = ahci_init_one,
 305         .remove                 = ahci_remove_one,
 306 };
 307
 308
 309 static inline unsigned long ahci_port_base_ul (unsigned long base, unsigned int port)
 310 {
 311         return base + 0x100 + (port * 0x80);
 312 }
 313
 314 static inline void __iomem *ahci_port_base (void __iomem *base, unsigned int port)
 315 {
 316         return (void __iomem *) ahci_port_base_ul((unsigned long)base, port);
 317 }
 318
 319 static int ahci_port_start(struct ata_port *ap)
 320 {
 321         struct device *dev = ap->host_set->dev;
 322         struct ahci_host_priv *hpriv = ap->host_set->private_data;
 323         struct ahci_port_priv *pp;
 324         void __iomem *mmio = ap->host_set->mmio_base;
 325         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 326         void *mem;
 327         dma_addr_t mem_dma;
 328         int rc;
 329
 330         pp = kmalloc(sizeof(*pp), GFP_KERNEL);
 331         if (!pp)
 332                 return -ENOMEM;
 333         memset(pp, 0, sizeof(*pp));
 334
 335         rc = ata_pad_alloc(ap, dev);
 336         if (rc) {
 337                 kfree(pp);
 338                 return rc;
 339         }
 340
 341         mem = dma_alloc_coherent(dev, AHCI_PORT_PRIV_DMA_SZ, &mem_dma, GFP_KERNEL);
 342         if (!mem) {
 343                 ata_pad_free(ap, dev);
 344                 kfree(pp);
 345                 return -ENOMEM;
 346         }
 347         memset(mem, 0, AHCI_PORT_PRIV_DMA_SZ);
 348
 349         /*
 350          * First item in chunk of DMA memory: 32-slot command table,
 351          * 32 bytes each in size
 352          */
 353         pp->cmd_slot = mem;
 354         pp->cmd_slot_dma = mem_dma;
 355
 356         mem += AHCI_CMD_SLOT_SZ;
 357         mem_dma += AHCI_CMD_SLOT_SZ;
 358
 359         /*
 360          * Second item: Received-FIS area
 361          */
 362         pp->rx_fis = mem;
 363         pp->rx_fis_dma = mem_dma;
 364
 365         mem += AHCI_RX_FIS_SZ;
 366         mem_dma += AHCI_RX_FIS_SZ;
 367
 368         /*
 369          * Third item: data area for storing a single command
 370          * and its scatter-gather table
 371          */
 372         pp->cmd_tbl = mem;
 373         pp->cmd_tbl_dma = mem_dma;
 374
 375         pp->cmd_tbl_sg = mem + AHCI_CMD_TBL_HDR;
 376
 377         ap->private_data = pp;
 378
 379         if (hpriv->cap & HOST_CAP_64)
 380                 writel((pp->cmd_slot_dma >> 16) >> 16, port_mmio + PORT_LST_ADDR_HI);
 381         writel(pp->cmd_slot_dma & 0xffffffff, port_mmio + PORT_LST_ADDR);
 382         readl(port_mmio + PORT_LST_ADDR); /* flush */
 383
 384         if (hpriv->cap & HOST_CAP_64)
 385                 writel((pp->rx_fis_dma >> 16) >> 16, port_mmio + PORT_FIS_ADDR_HI);
 386         writel(pp->rx_fis_dma & 0xffffffff, port_mmio + PORT_FIS_ADDR);
 387         readl(port_mmio + PORT_FIS_ADDR); /* flush */
 388
 389         writel(PORT_CMD_ICC_ACTIVE | PORT_CMD_FIS_RX |
 390                PORT_CMD_POWER_ON | PORT_CMD_SPIN_UP |
 391                PORT_CMD_START, port_mmio + PORT_CMD);
 392         readl(port_mmio + PORT_CMD); /* flush */
 393
 394         return 0;
 395 }
 396
 397
 398 static void ahci_port_stop(struct ata_port *ap)
 399 {
 400         struct device *dev = ap->host_set->dev;
 401         struct ahci_port_priv *pp = ap->private_data;
 402         void __iomem *mmio = ap->host_set->mmio_base;
 403         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 404         u32 tmp;
 405
 406         tmp = readl(port_mmio + PORT_CMD);
 407         tmp &= ~(PORT_CMD_START | PORT_CMD_FIS_RX);
 408         writel(tmp, port_mmio + PORT_CMD);
 409         readl(port_mmio + PORT_CMD); /* flush */
 410
 411         /* spec says 500 msecs for each PORT_CMD_{START,FIS_RX} bit, so
 412          * this is slightly incorrect.
 413          */
 414         msleep(500);
 415
 416         ap->private_data = NULL;
 417         dma_free_coherent(dev, AHCI_PORT_PRIV_DMA_SZ,
 418                           pp->cmd_slot, pp->cmd_slot_dma);
 419         ata_pad_free(ap, dev);
 420         kfree(pp);
 421 }
 422
 423 static u32 ahci_scr_read (struct ata_port *ap, unsigned int sc_reg_in)
 424 {
 425         unsigned int sc_reg;
 426
 427         switch (sc_reg_in) {
 428         case SCR_STATUS:        sc_reg = 0; break;
 429         case SCR_CONTROL:       sc_reg = 1; break;
 430         case SCR_ERROR:         sc_reg = 2; break;
 431         case SCR_ACTIVE:        sc_reg = 3; break;
 432         default:
 433                 return 0xffffffffU;
 434         }
 435
 436         return readl((void __iomem *) ap->ioaddr.scr_addr + (sc_reg * 4));
 437 }
 438
 439
 440 static void ahci_scr_write (struct ata_port *ap, unsigned int sc_reg_in,
 441                                u32 val)
 442 {
 443         unsigned int sc_reg;
 444
 445         switch (sc_reg_in) {
 446         case SCR_STATUS:        sc_reg = 0; break;
 447         case SCR_CONTROL:       sc_reg = 1; break;
 448         case SCR_ERROR:         sc_reg = 2; break;
 449         case SCR_ACTIVE:        sc_reg = 3; break;
 450         default:
 451                 return;
 452         }
 453
 454         writel(val, (void __iomem *) ap->ioaddr.scr_addr + (sc_reg * 4));
 455 }
 456
 457 static int ahci_stop_engine(struct ata_port *ap)
 458 {
 459         void __iomem *mmio = ap->host_set->mmio_base;
 460         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 461         int work;
 462         u32 tmp;
 463
 464         tmp = readl(port_mmio + PORT_CMD);
 465         tmp &= ~PORT_CMD_START;
 466         writel(tmp, port_mmio + PORT_CMD);
 467
 468         /* wait for engine to stop.  TODO: this could be
 469          * as long as 500 msec
 470          */
 471         work = 1000;
 472         while (work-- > 0) {
 473                 tmp = readl(port_mmio + PORT_CMD);
 474                 if ((tmp & PORT_CMD_LIST_ON) == 0)
 475                         return 0;
 476                 udelay(10);
 477         }
 478
 479         return -EIO;
 480 }
 481
 482 static void ahci_start_engine(struct ata_port *ap)
 483 {
 484         void __iomem *mmio = ap->host_set->mmio_base;
 485         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 486         u32 tmp;
 487
 488         tmp = readl(port_mmio + PORT_CMD);
 489         tmp |= PORT_CMD_START;
 490         writel(tmp, port_mmio + PORT_CMD);
 491         readl(port_mmio + PORT_CMD); /* flush */
 492 }
 493
 494 static unsigned int ahci_dev_classify(struct ata_port *ap)
 495 {
 496         void __iomem *port_mmio = (void __iomem *) ap->ioaddr.cmd_addr;
 497         struct ata_taskfile tf;
 498         u32 tmp;
 499
 500         tmp = readl(port_mmio + PORT_SIG);
 501         tf.lbah         = (tmp >> 24)   & 0xff;
 502         tf.lbam         = (tmp >> 16)   & 0xff;
 503         tf.lbal         = (tmp >> 8)    & 0xff;
 504         tf.nsect        = (tmp)         & 0xff;
 505
 506         return ata_dev_classify(&tf);
 507 }
 508
 509 static void ahci_fill_cmd_slot(struct ahci_port_priv *pp, u32 opts)
 510 {
 511         pp->cmd_slot[0].opts = cpu_to_le32(opts);
 512         pp->cmd_slot[0].status = 0;
 513         pp->cmd_slot[0].tbl_addr = cpu_to_le32(pp->cmd_tbl_dma & 0xffffffff);
 514         pp->cmd_slot[0].tbl_addr_hi = cpu_to_le32((pp->cmd_tbl_dma >> 16) >> 16);
 515 }
 516
 517 static int ahci_hardreset(struct ata_port *ap, int verbose, unsigned int *class)
 518 {
 519         int rc;
 520
 521         DPRINTK("ENTER\n");
 522
 523         ahci_stop_engine(ap);
 524         rc = sata_std_hardreset(ap, verbose, class);
 525         ahci_start_engine(ap);
 526
 527         if (rc == 0)
 528                 *class = ahci_dev_classify(ap);
 529         if (*class == ATA_DEV_UNKNOWN)
 530                 *class = ATA_DEV_NONE;
 531
 532         DPRINTK("EXIT, rc=%d, class=%u\n", rc, *class);
 533         return rc;
 534 }
 535
 536 static void ahci_postreset(struct ata_port *ap, unsigned int *class)
 537 {
 538         void __iomem *port_mmio = (void __iomem *) ap->ioaddr.cmd_addr;
 539         u32 new_tmp, tmp;
 540
 541         ata_std_postreset(ap, class);
 542
 543         /* Make sure port's ATAPI bit is set appropriately */
 544         new_tmp = tmp = readl(port_mmio + PORT_CMD);
 545         if (*class == ATA_DEV_ATAPI)
 546                 new_tmp |= PORT_CMD_ATAPI;
 547         else
 548                 new_tmp &= ~PORT_CMD_ATAPI;
 549         if (new_tmp != tmp) {
 550                 writel(new_tmp, port_mmio + PORT_CMD);
 551                 readl(port_mmio + PORT_CMD); /* flush */
 552         }
 553 }
 554
 555 static int ahci_probe_reset(struct ata_port *ap, unsigned int *classes)
 556 {
 557         return ata_drive_probe_reset(ap, NULL, NULL, ahci_hardreset,
 558                                      ahci_postreset, classes);
 559 }
 560
 561 static u8 ahci_check_status(struct ata_port *ap)
 562 {
 563         void __iomem *mmio = (void __iomem *) ap->ioaddr.cmd_addr;
 564
 565         return readl(mmio + PORT_TFDATA) & 0xFF;
 566 }
 567
 568 static void ahci_tf_read(struct ata_port *ap, struct ata_taskfile *tf)
 569 {
 570         struct ahci_port_priv *pp = ap->private_data;
 571         u8 *d2h_fis = pp->rx_fis + RX_FIS_D2H_REG;
 572
 573         ata_tf_from_fis(d2h_fis, tf);
 574 }
 575
 576 static unsigned int ahci_fill_sg(struct ata_queued_cmd *qc)
 577 {
 578         struct ahci_port_priv *pp = qc->ap->private_data;
 579         struct scatterlist *sg;
 580         struct ahci_sg *ahci_sg;
 581         unsigned int n_sg = 0;
 582
 583         VPRINTK("ENTER\n");
 584
 585         /*
 586          * Next, the S/G list.
 587          */
 588         ahci_sg = pp->cmd_tbl_sg;
 589         ata_for_each_sg(sg, qc) {
 590                 dma_addr_t addr = sg_dma_address(sg);
 591                 u32 sg_len = sg_dma_len(sg);
 592
 593                 ahci_sg->addr = cpu_to_le32(addr & 0xffffffff);
 594                 ahci_sg->addr_hi = cpu_to_le32((addr >> 16) >> 16);
 595                 ahci_sg->flags_size = cpu_to_le32(sg_len - 1);
 596
 597                 ahci_sg++;
 598                 n_sg++;
 599         }
 600
 601         return n_sg;
 602 }
 603
 604 static void ahci_qc_prep(struct ata_queued_cmd *qc)
 605 {
 606         struct ata_port *ap = qc->ap;
 607         struct ahci_port_priv *pp = ap->private_data;
 608         int is_atapi = is_atapi_taskfile(&qc->tf);
 609         u32 opts;
 610         const u32 cmd_fis_len = 5; /* five dwords */
 611         unsigned int n_elem;
 612
 613         /*
 614          * Fill in command table information.  First, the header,
 615          * a SATA Register - Host to Device command FIS.
 616          */
 617         ata_tf_to_fis(&qc->tf, pp->cmd_tbl, 0);
 618         if (is_atapi) {
 619                 memset(pp->cmd_tbl + AHCI_CMD_TBL_CDB, 0, 32);
 620                 memcpy(pp->cmd_tbl + AHCI_CMD_TBL_CDB, qc->cdb,
 621                        qc->dev->cdb_len);
 622         }
 623
 624         n_elem = 0;
 625         if (qc->flags & ATA_QCFLAG_DMAMAP)
 626                 n_elem = ahci_fill_sg(qc);
 627
 628         /*
 629          * Fill in command slot information.
 630          */
 631         opts = cmd_fis_len | n_elem << 16;
 632         if (qc->tf.flags & ATA_TFLAG_WRITE)
 633                 opts |= AHCI_CMD_WRITE;
 634         if (is_atapi)
 635                 opts |= AHCI_CMD_ATAPI;
 636
 637         ahci_fill_cmd_slot(pp, opts);
 638 }
 639
 640 static void ahci_restart_port(struct ata_port *ap, u32 irq_stat)
 641 {
 642         void __iomem *mmio = ap->host_set->mmio_base;
 643         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 644         u32 tmp;
 645
 646         if ((ap->device[0].class != ATA_DEV_ATAPI) ||
 647             ((irq_stat & PORT_IRQ_TF_ERR) == 0))
 648                 printk(KERN_WARNING "ata%u: port reset, "
 649                        "p_is %x is %x pis %x cmd %x tf %x ss %x se %x\n",
 650                         ap->id,
 651                         irq_stat,
 652                         readl(mmio + HOST_IRQ_STAT),
 653                         readl(port_mmio + PORT_IRQ_STAT),
 654                         readl(port_mmio + PORT_CMD),
 655                         readl(port_mmio + PORT_TFDATA),
 656                         readl(port_mmio + PORT_SCR_STAT),
 657                         readl(port_mmio + PORT_SCR_ERR));
 658
 659         /* stop DMA */
 660         ahci_stop_engine(ap);
 661
 662         /* clear SATA phy error, if any */
 663         tmp = readl(port_mmio + PORT_SCR_ERR);
 664         writel(tmp, port_mmio + PORT_SCR_ERR);
 665
 666         /* if DRQ/BSY is set, device needs to be reset.
 667          * if so, issue COMRESET
 668          */
 669         tmp = readl(port_mmio + PORT_TFDATA);
 670         if (tmp & (ATA_BUSY | ATA_DRQ)) {
 671                 writel(0x301, port_mmio + PORT_SCR_CTL);
 672                 readl(port_mmio + PORT_SCR_CTL); /* flush */
 673                 udelay(10);
 674                 writel(0x300, port_mmio + PORT_SCR_CTL);
 675                 readl(port_mmio + PORT_SCR_CTL); /* flush */
 676         }
 677
 678         /* re-start DMA */
 679         ahci_start_engine(ap);
 680 }
 681
 682 static void ahci_eng_timeout(struct ata_port *ap)
 683 {
 684         struct ata_host_set *host_set = ap->host_set;
 685         void __iomem *mmio = host_set->mmio_base;
 686         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 687         struct ata_queued_cmd *qc;
 688         unsigned long flags;
 689
 690         printk(KERN_WARNING "ata%u: handling error/timeout\n", ap->id);
 691
 692         spin_lock_irqsave(&host_set->lock, flags);
 693
 694         ahci_restart_port(ap, readl(port_mmio + PORT_IRQ_STAT));
 695         qc = ata_qc_from_tag(ap, ap->active_tag);
 696         qc->err_mask |= AC_ERR_TIMEOUT;
 697
 698         spin_unlock_irqrestore(&host_set->lock, flags);
 699
 700         ata_eh_qc_complete(qc);
 701 }
 702
 703 static inline int ahci_host_intr(struct ata_port *ap, struct ata_queued_cmd *qc)
 704 {
 705         void __iomem *mmio = ap->host_set->mmio_base;
 706         void __iomem *port_mmio = ahci_port_base(mmio, ap->port_no);
 707         u32 status, serr, ci;
 708
 709         serr = readl(port_mmio + PORT_SCR_ERR);
 710         writel(serr, port_mmio + PORT_SCR_ERR);
 711
 712         status = readl(port_mmio + PORT_IRQ_STAT);
 713         writel(status, port_mmio + PORT_IRQ_STAT);
 714
 715         ci = readl(port_mmio + PORT_CMD_ISSUE);
 716         if (likely((ci & 0x1) == 0)) {
 717                 if (qc) {
 718                         WARN_ON(qc->err_mask);
 719                         ata_qc_complete(qc);
 720                         qc = NULL;
 721                 }
 722         }
 723
 724         if (status & PORT_IRQ_FATAL) {
 725                 unsigned int err_mask;
 726                 if (status & PORT_IRQ_TF_ERR)
 727                         err_mask = AC_ERR_DEV;
 728                 else if (status & PORT_IRQ_IF_ERR)
 729                         err_mask = AC_ERR_ATA_BUS;
 730                 else
 731                         err_mask = AC_ERR_HOST_BUS;
 732
 733                 /* command processing has stopped due to error; restart */
 734                 ahci_restart_port(ap, status);
 735
 736                 if (qc) {
 737                         qc->err_mask |= err_mask;
 738                         ata_qc_complete(qc);
 739                 }
 740         }
 741
 742         return 1;
 743 }
 744
 745 static void ahci_irq_clear(struct ata_port *ap)
 746 {
 747         /* TODO */
 748 }
 749
 750 static irqreturn_t ahci_interrupt (int irq, void *dev_instance, struct pt_regs *regs)
 751 {
 752         struct ata_host_set *host_set = dev_instance;
 753         struct ahci_host_priv *hpriv;
 754         unsigned int i, handled = 0;
 755         void __iomem *mmio;
 756         u32 irq_stat, irq_ack = 0;
 757
 758         VPRINTK("ENTER\n");
 759
 760         hpriv = host_set->private_data;
 761         mmio = host_set->mmio_base;
 762
 763         /* sigh.  0xffffffff is a valid return from h/w */
 764         irq_stat = readl(mmio + HOST_IRQ_STAT);
 765         irq_stat &= hpriv->port_map;
 766         if (!irq_stat)
 767                 return IRQ_NONE;
 768
 769         spin_lock(&host_set->lock);
 770
 771         for (i = 0; i < host_set->n_ports; i++) {
 772                 struct ata_port *ap;
 773
 774                 if (!(irq_stat & (1 << i)))
 775                         continue;
 776
 777                 ap = host_set->ports[i];
 778                 if (ap) {
 779                         struct ata_queued_cmd *qc;
 780                         qc = ata_qc_from_tag(ap, ap->active_tag);
 781                         if (!ahci_host_intr(ap, qc))
 782                                 if (ata_ratelimit()) {
 783                                         struct pci_dev *pdev =
 784                                                 to_pci_dev(ap->host_set->dev);
 785                                         dev_printk(KERN_WARNING, &pdev->dev,
 786                                           "unhandled interrupt on port %u\n",
 787                                           i);
 788                                 }
 789
 790                         VPRINTK("port %u\n", i);
 791                 } else {
 792                         VPRINTK("port %u (no irq)\n", i);
 793                         if (ata_ratelimit()) {
 794                                 struct pci_dev *pdev =
 795                                         to_pci_dev(ap->host_set->dev);
 796                                 dev_printk(KERN_WARNING, &pdev->dev,
 797                                         "interrupt on disabled port %u\n", i);
 798                         }
 799                 }
 800
 801                 irq_ack |= (1 << i);
 802         }
 803
 804         if (irq_ack) {
 805                 writel(irq_ack, mmio + HOST_IRQ_STAT);
 806                 handled = 1;
 807         }
 808
 809         spin_unlock(&host_set->lock);
 810
 811         VPRINTK("EXIT\n");
 812
 813         return IRQ_RETVAL(handled);
 814 }
 815
 816 static unsigned int ahci_qc_issue(struct ata_queued_cmd *qc)
 817 {
 818         struct ata_port *ap = qc->ap;
 819         void __iomem *port_mmio = (void __iomem *) ap->ioaddr.cmd_addr;
 820
 821         writel(1, port_mmio + PORT_CMD_ISSUE);
 822         readl(port_mmio + PORT_CMD_ISSUE);      /* flush */
 823
 824         return 0;
 825 }
 826
 827 static void ahci_setup_port(struct ata_ioports *port, unsigned long base,
 828                             unsigned int port_idx)
 829 {
 830         VPRINTK("ENTER, base==0x%lx, port_idx %u\n", base, port_idx);
 831         base = ahci_port_base_ul(base, port_idx);
 832         VPRINTK("base now==0x%lx\n", base);
 833
 834         port->cmd_addr          = base;
 835         port->scr_addr          = base + PORT_SCR;
 836
 837         VPRINTK("EXIT\n");
 838 }
 839
 840 static int ahci_host_init(struct ata_probe_ent *probe_ent)
 841 {
 842         struct ahci_host_priv *hpriv = probe_ent->private_data;
 843         struct pci_dev *pdev = to_pci_dev(probe_ent->dev);
 844         void __iomem *mmio = probe_ent->mmio_base;
 845         u32 tmp, cap_save;
 846         unsigned int i, j, using_dac;
 847         int rc;
 848         void __iomem *port_mmio;
 849
 850         cap_save = readl(mmio + HOST_CAP);
 851         cap_save &= ( (1<<28) | (1<<17) );
 852         cap_save |= (1 << 27);
 853
 854         /* global controller reset */
 855         tmp = readl(mmio + HOST_CTL);
 856         if ((tmp & HOST_RESET) == 0) {
 857                 writel(tmp | HOST_RESET, mmio + HOST_CTL);
 858                 readl(mmio + HOST_CTL); /* flush */
 859         }
 860
 861         /* reset must complete within 1 second, or
 862          * the hardware should be considered fried.
 863          */
 864         ssleep(1);
 865
 866         tmp = readl(mmio + HOST_CTL);
 867         if (tmp & HOST_RESET) {
 868                 dev_printk(KERN_ERR, &pdev->dev,
 869                            "controller reset failed (0x%x)\n", tmp);
 870                 return -EIO;
 871         }
 872
 873         writel(HOST_AHCI_EN, mmio + HOST_CTL);
 874         (void) readl(mmio + HOST_CTL);  /* flush */
 875         writel(cap_save, mmio + HOST_CAP);
 876         writel(0xf, mmio + HOST_PORTS_IMPL);
 877         (void) readl(mmio + HOST_PORTS_IMPL);   /* flush */
 878
 879         if (pdev->vendor == PCI_VENDOR_ID_INTEL) {
 880                 u16 tmp16;
 881
 882                 pci_read_config_word(pdev, 0x92, &tmp16);
 883                 tmp16 |= 0xf;
 884                 pci_write_config_word(pdev, 0x92, tmp16);
 885         }
 886
 887         hpriv->cap = readl(mmio + HOST_CAP);
 888         hpriv->port_map = readl(mmio + HOST_PORTS_IMPL);
 889         probe_ent->n_ports = (hpriv->cap & 0x1f) + 1;
 890
 891         VPRINTK("cap 0x%x  port_map 0x%x  n_ports %d\n",
 892                 hpriv->cap, hpriv->port_map, probe_ent->n_ports);
 893
 894         using_dac = hpriv->cap & HOST_CAP_64;
 895         if (using_dac &&
 896             !pci_set_dma_mask(pdev, DMA_64BIT_MASK)) {
 897                 rc = pci_set_consistent_dma_mask(pdev, DMA_64BIT_MASK);
 898                 if (rc) {
 899                         rc = pci_set_consistent_dma_mask(pdev, DMA_32BIT_MASK);
 900                         if (rc) {
 901                                 dev_printk(KERN_ERR, &pdev->dev,
 902                                            "64-bit DMA enable failed\n");
 903                                 return rc;
 904                         }
 905                 }
 906         } else {
 907                 rc = pci_set_dma_mask(pdev, DMA_32BIT_MASK);
 908                 if (rc) {
 909                         dev_printk(KERN_ERR, &pdev->dev,
 910                                    "32-bit DMA enable failed\n");
 911                         return rc;
 912                 }
 913                 rc = pci_set_consistent_dma_mask(pdev, DMA_32BIT_MASK);
 914                 if (rc) {
 915                         dev_printk(KERN_ERR, &pdev->dev,
 916                                    "32-bit consistent DMA enable failed\n");
 917                         return rc;
 918                 }
 919         }
 920
 921         for (i = 0; i < probe_ent->n_ports; i++) {
 922 #if 0 /* BIOSen initialize this incorrectly */
 923                 if (!(hpriv->port_map & (1 << i)))
 924                         continue;
 925 #endif
 926
 927                 port_mmio = ahci_port_base(mmio, i);
 928                 VPRINTK("mmio %p  port_mmio %p\n", mmio, port_mmio);
 929
 930                 ahci_setup_port(&probe_ent->port[i],
 931                                 (unsigned long) mmio, i);
 932
 933                 /* make sure port is not active */
 934                 tmp = readl(port_mmio + PORT_CMD);
 935                 VPRINTK("PORT_CMD 0x%x\n", tmp);
 936                 if (tmp & (PORT_CMD_LIST_ON | PORT_CMD_FIS_ON |
 937                            PORT_CMD_FIS_RX | PORT_CMD_START)) {
 938                         tmp &= ~(PORT_CMD_LIST_ON | PORT_CMD_FIS_ON |
 939                                  PORT_CMD_FIS_RX | PORT_CMD_START);
 940                         writel(tmp, port_mmio + PORT_CMD);
 941                         readl(port_mmio + PORT_CMD); /* flush */
 942
 943                         /* spec says 500 msecs for each bit, so
 944                          * this is slightly incorrect.
 945                          */
 946                         msleep(500);
 947                 }
 948
 949                 writel(PORT_CMD_SPIN_UP, port_mmio + PORT_CMD);
 950
 951                 j = 0;
 952                 while (j < 100) {
 953                         msleep(10);
 954                         tmp = readl(port_mmio + PORT_SCR_STAT);
 955                         if ((tmp & 0xf) == 0x3)
 956                                 break;
 957                         j++;
 958                 }
 959
 960                 tmp = readl(port_mmio + PORT_SCR_ERR);
 961                 VPRINTK("PORT_SCR_ERR 0x%x\n", tmp);
 962                 writel(tmp, port_mmio + PORT_SCR_ERR);
 963
 964                 /* ack any pending irq events for this port */
 965                 tmp = readl(port_mmio + PORT_IRQ_STAT);
 966                 VPRINTK("PORT_IRQ_STAT 0x%x\n", tmp);
 967                 if (tmp)
 968                         writel(tmp, port_mmio + PORT_IRQ_STAT);
 969
 970                 writel(1 << i, mmio + HOST_IRQ_STAT);
 971
 972                 /* set irq mask (enables interrupts) */
 973                 writel(DEF_PORT_IRQ, port_mmio + PORT_IRQ_MASK);
 974         }
 975
 976         tmp = readl(mmio + HOST_CTL);
 977         VPRINTK("HOST_CTL 0x%x\n", tmp);
 978         writel(tmp | HOST_IRQ_EN, mmio + HOST_CTL);
 979         tmp = readl(mmio + HOST_CTL);
 980         VPRINTK("HOST_CTL 0x%x\n", tmp);
 981
 982         pci_set_master(pdev);
 983
 984         return 0;
 985 }
 986
 987 static void ahci_print_info(struct ata_probe_ent *probe_ent)
 988 {
 989         struct ahci_host_priv *hpriv = probe_ent->private_data;
 990         struct pci_dev *pdev = to_pci_dev(probe_ent->dev);
 991         void __iomem *mmio = probe_ent->mmio_base;
 992         u32 vers, cap, impl, speed;
 993         const char *speed_s;
 994         u16 cc;
 995         const char *scc_s;
 996
 997         vers = readl(mmio + HOST_VERSION);
 998         cap = hpriv->cap;
 999         impl = hpriv->port_map;
1000
1001         speed = (cap >> 20) & 0xf;
1002         if (speed == 1)
1003                 speed_s = "1.5";
1004         else if (speed == 2)
1005                 speed_s = "3";
1006         else
1007                 speed_s = "?";
1008
1009         pci_read_config_word(pdev, 0x0a, &cc);
1010         if (cc == 0x0101)
1011                 scc_s = "IDE";
1012         else if (cc == 0x0106)
1013                 scc_s = "SATA";
1014         else if (cc == 0x0104)
1015                 scc_s = "RAID";
1016         else
1017                 scc_s = "unknown";
1018
1019         dev_printk(KERN_INFO, &pdev->dev,
1020                 "AHCI %02x%02x.%02x%02x "
1021                 "%u slots %u ports %s Gbps 0x%x impl %s mode\n"
1022                 ,
1023
1024                 (vers >> 24) & 0xff,
1025                 (vers >> 16) & 0xff,
1026                 (vers >> 8) & 0xff,
1027                 vers & 0xff,
1028
1029                 ((cap >> 8) & 0x1f) + 1,
1030                 (cap & 0x1f) + 1,
1031                 speed_s,
1032                 impl,
1033                 scc_s);
1034
1035         dev_printk(KERN_INFO, &pdev->dev,
1036                 "flags: "
1037                 "%s%s%s%s%s%s"
1038                 "%s%s%s%s%s%s%s\n"
1039                 ,
1040
1041                 cap & (1 << 31) ? "64bit " : "",
1042                 cap & (1 << 30) ? "ncq " : "",
1043                 cap & (1 << 28) ? "ilck " : "",
1044                 cap & (1 << 27) ? "stag " : "",
1045                 cap & (1 << 26) ? "pm " : "",
1046                 cap & (1 << 25) ? "led " : "",
1047
1048                 cap & (1 << 24) ? "clo " : "",
1049                 cap & (1 << 19) ? "nz " : "",
1050                 cap & (1 << 18) ? "only " : "",
1051                 cap & (1 << 17) ? "pmp " : "",
1052                 cap & (1 << 15) ? "pio " : "",
1053                 cap & (1 << 14) ? "slum " : "",
1054                 cap & (1 << 13) ? "part " : ""
1055                 );
1056 }
1057
1058 static int ahci_init_one (struct pci_dev *pdev, const struct pci_device_id *ent)
1059 {
1060         static int printed_version;
1061         struct ata_probe_ent *probe_ent = NULL;
1062         struct ahci_host_priv *hpriv;
1063         unsigned long base;
1064         void __iomem *mmio_base;
1065         unsigned int board_idx = (unsigned int) ent->driver_data;
1066         int have_msi, pci_dev_busy = 0;
1067         int rc;
1068
1069         VPRINTK("ENTER\n");
1070
1071         if (!printed_version++)
1072                 dev_printk(KERN_DEBUG, &pdev->dev, "version " DRV_VERSION "\n");
1073
1074         rc = pci_enable_device(pdev);
1075         if (rc)
1076                 return rc;
1077
1078         rc = pci_request_regions(pdev, DRV_NAME);
1079         if (rc) {
1080                 pci_dev_busy = 1;
1081                 goto err_out;
1082         }
1083
1084         if (pci_enable_msi(pdev) == 0)
1085                 have_msi = 1;
1086         else {
1087                 pci_intx(pdev, 1);
1088                 have_msi = 0;
1089         }
1090
1091         probe_ent = kmalloc(sizeof(*probe_ent), GFP_KERNEL);
1092         if (probe_ent == NULL) {
1093                 rc = -ENOMEM;
1094                 goto err_out_msi;
1095         }
1096
1097         memset(probe_ent, 0, sizeof(*probe_ent));
1098         probe_ent->dev = pci_dev_to_dev(pdev);
1099         INIT_LIST_HEAD(&probe_ent->node);
1100
1101         mmio_base = pci_iomap(pdev, AHCI_PCI_BAR, 0);
1102         if (mmio_base == NULL) {
1103                 rc = -ENOMEM;
1104                 goto err_out_free_ent;
1105         }
1106         base = (unsigned long) mmio_base;
1107
1108         hpriv = kmalloc(sizeof(*hpriv), GFP_KERNEL);
1109         if (!hpriv) {
1110                 rc = -ENOMEM;
1111                 goto err_out_iounmap;
1112         }
1113         memset(hpriv, 0, sizeof(*hpriv));
1114
1115         probe_ent->sht          = ahci_port_info[board_idx].sht;
1116         probe_ent->host_flags   = ahci_port_info[board_idx].host_flags;
1117         probe_ent->pio_mask     = ahci_port_info[board_idx].pio_mask;
1118         probe_ent->udma_mask    = ahci_port_info[board_idx].udma_mask;
1119         probe_ent->port_ops     = ahci_port_info[board_idx].port_ops;
1120
1121         probe_ent->irq = pdev->irq;
1122         probe_ent->irq_flags = SA_SHIRQ;
1123         probe_ent->mmio_base = mmio_base;
1124         probe_ent->private_data = hpriv;
1125
1126         if (have_msi)
1127                 hpriv->flags |= AHCI_FLAG_MSI;
1128
1129         /* JMicron-specific fixup: make sure we're in AHCI mode */
1130         if (pdev->vendor == 0x197b)
1131                 pci_write_config_byte(pdev, 0x41, 0xa1);
1132
1133         /* initialize adapter */
1134         rc = ahci_host_init(probe_ent);
1135         if (rc)
1136                 goto err_out_hpriv;
1137
1138         ahci_print_info(probe_ent);
1139
1140         /* FIXME: check ata_device_add return value */
1141         ata_device_add(probe_ent);
1142         kfree(probe_ent);
1143
1144         return 0;
1145
1146 err_out_hpriv:
1147         kfree(hpriv);
1148 err_out_iounmap:
1149         pci_iounmap(pdev, mmio_base);
1150 err_out_free_ent:
1151         kfree(probe_ent);
1152 err_out_msi:
1153         if (have_msi)
1154                 pci_disable_msi(pdev);
1155         else
1156                 pci_intx(pdev, 0);
1157         pci_release_regions(pdev);
1158 err_out:
1159         if (!pci_dev_busy)
1160                 pci_disable_device(pdev);
1161         return rc;
1162 }
1163
1164 static void ahci_remove_one (struct pci_dev *pdev)
1165 {
1166         struct device *dev = pci_dev_to_dev(pdev);
1167         struct ata_host_set *host_set = dev_get_drvdata(dev);
1168         struct ahci_host_priv *hpriv = host_set->private_data;
1169         struct ata_port *ap;
1170         unsigned int i;
1171         int have_msi;
1172
1173         for (i = 0; i < host_set->n_ports; i++) {
1174                 ap = host_set->ports[i];
1175
1176                 scsi_remove_host(ap->host);
1177         }
1178
1179         have_msi = hpriv->flags & AHCI_FLAG_MSI;
1180         free_irq(host_set->irq, host_set);
1181
1182         for (i = 0; i < host_set->n_ports; i++) {
1183                 ap = host_set->ports[i];
1184
1185                 ata_scsi_release(ap->host);
1186                 scsi_host_put(ap->host);
1187         }
1188
1189         kfree(hpriv);
1190         pci_iounmap(pdev, host_set->mmio_base);
1191         kfree(host_set);
1192
1193         if (have_msi)
1194                 pci_disable_msi(pdev);
1195         else
1196                 pci_intx(pdev, 0);
1197         pci_release_regions(pdev);
1198         pci_disable_device(pdev);
1199         dev_set_drvdata(dev, NULL);
1200 }
1201
1202 static int __init ahci_init(void)
1203 {
1204         return pci_module_init(&ahci_pci_driver);
1205 }
1206
1207 static void __exit ahci_exit(void)
1208 {
1209         pci_unregister_driver(&ahci_pci_driver);
1210 }
1211
1212
1213 MODULE_AUTHOR("Jeff Garzik");
1214 MODULE_DESCRIPTION("AHCI SATA low-level driver");
1215 MODULE_LICENSE("GPL");
1216 MODULE_DEVICE_TABLE(pci, ahci_pci_tbl);
1217 MODULE_VERSION(DRV_VERSION);
1218
1219 module_init(ahci_init);
1220 module_exit(ahci_exit);