drivers/ata/sata_nv.c

   1 /*
   2  *  sata_nv.c - NVIDIA nForce SATA
   3  *
   4  *  Copyright 2004 NVIDIA Corp.  All rights reserved.
   5  *  Copyright 2004 Andrew Chew
   6  *
   7  *
   8  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 2, or (at your option)
  11  *  any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; see the file COPYING.  If not, write to
  20  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  21  *
  22  *
  23  *  libata documentation is available via 'make {ps|pdf}docs',
  24  *  as Documentation/DocBook/libata.*
  25  *
  26  *  No hardware documentation available outside of NVIDIA.
  27  *  This driver programs the NVIDIA SATA controller in a similar
  28  *  fashion as with other PCI IDE BMDMA controllers, with a few
  29  *  NV-specific details such as register offsets, SATA phy location,
  30  *  hotplug info, etc.
  31  *
  32  *  CK804/MCP04 controllers support an alternate programming interface
  33  *  similar to the ADMA specification (with some modifications).
  34  *  This allows the use of NCQ. Non-DMA-mapped ATA commands are still
  35  *  sent through the legacy interface.
  36  *
  37  */
  38
  39 #include <linux/kernel.h>
  40 #include <linux/module.h>
  41 #include <linux/gfp.h>
  42 #include <linux/pci.h>
  43 #include <linux/init.h>
  44 #include <linux/blkdev.h>
  45 #include <linux/delay.h>
  46 #include <linux/interrupt.h>
  47 #include <linux/device.h>
  48 #include <scsi/scsi_host.h>
  49 #include <scsi/scsi_device.h>
  50 #include <linux/libata.h>
  51
  52 #define DRV_NAME                        "sata_nv"
  53 #define DRV_VERSION                     "3.5"
  54
  55 #define NV_ADMA_DMA_BOUNDARY            0xffffffffUL
  56
  57 enum {
  58         NV_MMIO_BAR                     = 5,
  59
  60         NV_PORTS                        = 2,
  61         NV_PIO_MASK                     = ATA_PIO4,
  62         NV_MWDMA_MASK                   = ATA_MWDMA2,
  63         NV_UDMA_MASK                    = ATA_UDMA6,
  64         NV_PORT0_SCR_REG_OFFSET         = 0x00,
  65         NV_PORT1_SCR_REG_OFFSET         = 0x40,
  66
  67         /* INT_STATUS/ENABLE */
  68         NV_INT_STATUS                   = 0x10,
  69         NV_INT_ENABLE                   = 0x11,
  70         NV_INT_STATUS_CK804             = 0x440,
  71         NV_INT_ENABLE_CK804             = 0x441,
  72
  73         /* INT_STATUS/ENABLE bits */
  74         NV_INT_DEV                      = 0x01,
  75         NV_INT_PM                       = 0x02,
  76         NV_INT_ADDED                    = 0x04,
  77         NV_INT_REMOVED                  = 0x08,
  78
  79         NV_INT_PORT_SHIFT               = 4,    /* each port occupies 4 bits */
  80
  81         NV_INT_ALL                      = 0x0f,
  82         NV_INT_MASK                     = NV_INT_DEV |
  83                                           NV_INT_ADDED | NV_INT_REMOVED,
  84
  85         /* INT_CONFIG */
  86         NV_INT_CONFIG                   = 0x12,
  87         NV_INT_CONFIG_METHD             = 0x01, // 0 = INT, 1 = SMI
  88
  89         // For PCI config register 20
  90         NV_MCP_SATA_CFG_20              = 0x50,
  91         NV_MCP_SATA_CFG_20_SATA_SPACE_EN = 0x04,
  92         NV_MCP_SATA_CFG_20_PORT0_EN     = (1 << 17),
  93         NV_MCP_SATA_CFG_20_PORT1_EN     = (1 << 16),
  94         NV_MCP_SATA_CFG_20_PORT0_PWB_EN = (1 << 14),
  95         NV_MCP_SATA_CFG_20_PORT1_PWB_EN = (1 << 12),
  96
  97         NV_ADMA_MAX_CPBS                = 32,
  98         NV_ADMA_CPB_SZ                  = 128,
  99         NV_ADMA_APRD_SZ                 = 16,
 100         NV_ADMA_SGTBL_LEN               = (1024 - NV_ADMA_CPB_SZ) /
 101                                            NV_ADMA_APRD_SZ,
 102         NV_ADMA_SGTBL_TOTAL_LEN         = NV_ADMA_SGTBL_LEN + 5,
 103         NV_ADMA_SGTBL_SZ                = NV_ADMA_SGTBL_LEN * NV_ADMA_APRD_SZ,
 104         NV_ADMA_PORT_PRIV_DMA_SZ        = NV_ADMA_MAX_CPBS *
 105                                            (NV_ADMA_CPB_SZ + NV_ADMA_SGTBL_SZ),
 106
 107         /* BAR5 offset to ADMA general registers */
 108         NV_ADMA_GEN                     = 0x400,
 109         NV_ADMA_GEN_CTL                 = 0x00,
 110         NV_ADMA_NOTIFIER_CLEAR          = 0x30,
 111
 112         /* BAR5 offset to ADMA ports */
 113         NV_ADMA_PORT                    = 0x480,
 114
 115         /* size of ADMA port register space  */
 116         NV_ADMA_PORT_SIZE               = 0x100,
 117
 118         /* ADMA port registers */
 119         NV_ADMA_CTL                     = 0x40,
 120         NV_ADMA_CPB_COUNT               = 0x42,
 121         NV_ADMA_NEXT_CPB_IDX            = 0x43,
 122         NV_ADMA_STAT                    = 0x44,
 123         NV_ADMA_CPB_BASE_LOW            = 0x48,
 124         NV_ADMA_CPB_BASE_HIGH           = 0x4C,
 125         NV_ADMA_APPEND                  = 0x50,
 126         NV_ADMA_NOTIFIER                = 0x68,
 127         NV_ADMA_NOTIFIER_ERROR          = 0x6C,
 128
 129         /* NV_ADMA_CTL register bits */
 130         NV_ADMA_CTL_HOTPLUG_IEN         = (1 << 0),
 131         NV_ADMA_CTL_CHANNEL_RESET       = (1 << 5),
 132         NV_ADMA_CTL_GO                  = (1 << 7),
 133         NV_ADMA_CTL_AIEN                = (1 << 8),
 134         NV_ADMA_CTL_READ_NON_COHERENT   = (1 << 11),
 135         NV_ADMA_CTL_WRITE_NON_COHERENT  = (1 << 12),
 136
 137         /* CPB response flag bits */
 138         NV_CPB_RESP_DONE                = (1 << 0),
 139         NV_CPB_RESP_ATA_ERR             = (1 << 3),
 140         NV_CPB_RESP_CMD_ERR             = (1 << 4),
 141         NV_CPB_RESP_CPB_ERR             = (1 << 7),
 142
 143         /* CPB control flag bits */
 144         NV_CPB_CTL_CPB_VALID            = (1 << 0),
 145         NV_CPB_CTL_QUEUE                = (1 << 1),
 146         NV_CPB_CTL_APRD_VALID           = (1 << 2),
 147         NV_CPB_CTL_IEN                  = (1 << 3),
 148         NV_CPB_CTL_FPDMA                = (1 << 4),
 149
 150         /* APRD flags */
 151         NV_APRD_WRITE                   = (1 << 1),
 152         NV_APRD_END                     = (1 << 2),
 153         NV_APRD_CONT                    = (1 << 3),
 154
 155         /* NV_ADMA_STAT flags */
 156         NV_ADMA_STAT_TIMEOUT            = (1 << 0),
 157         NV_ADMA_STAT_HOTUNPLUG          = (1 << 1),
 158         NV_ADMA_STAT_HOTPLUG            = (1 << 2),
 159         NV_ADMA_STAT_CPBERR             = (1 << 4),
 160         NV_ADMA_STAT_SERROR             = (1 << 5),
 161         NV_ADMA_STAT_CMD_COMPLETE       = (1 << 6),
 162         NV_ADMA_STAT_IDLE               = (1 << 8),
 163         NV_ADMA_STAT_LEGACY             = (1 << 9),
 164         NV_ADMA_STAT_STOPPED            = (1 << 10),
 165         NV_ADMA_STAT_DONE               = (1 << 12),
 166         NV_ADMA_STAT_ERR                = NV_ADMA_STAT_CPBERR |
 167                                           NV_ADMA_STAT_TIMEOUT,
 168
 169         /* port flags */
 170         NV_ADMA_PORT_REGISTER_MODE      = (1 << 0),
 171         NV_ADMA_ATAPI_SETUP_COMPLETE    = (1 << 1),
 172
 173         /* MCP55 reg offset */
 174         NV_CTL_MCP55                    = 0x400,
 175         NV_INT_STATUS_MCP55             = 0x440,
 176         NV_INT_ENABLE_MCP55             = 0x444,
 177         NV_NCQ_REG_MCP55                = 0x448,
 178
 179         /* MCP55 */
 180         NV_INT_ALL_MCP55                = 0xffff,
 181         NV_INT_PORT_SHIFT_MCP55         = 16,   /* each port occupies 16 bits */
 182         NV_INT_MASK_MCP55               = NV_INT_ALL_MCP55 & 0xfffd,
 183
 184         /* SWNCQ ENABLE BITS*/
 185         NV_CTL_PRI_SWNCQ                = 0x02,
 186         NV_CTL_SEC_SWNCQ                = 0x04,
 187
 188         /* SW NCQ status bits*/
 189         NV_SWNCQ_IRQ_DEV                = (1 << 0),
 190         NV_SWNCQ_IRQ_PM                 = (1 << 1),
 191         NV_SWNCQ_IRQ_ADDED              = (1 << 2),
 192         NV_SWNCQ_IRQ_REMOVED            = (1 << 3),
 193
 194         NV_SWNCQ_IRQ_BACKOUT            = (1 << 4),
 195         NV_SWNCQ_IRQ_SDBFIS             = (1 << 5),
 196         NV_SWNCQ_IRQ_DHREGFIS           = (1 << 6),
 197         NV_SWNCQ_IRQ_DMASETUP           = (1 << 7),
 198
 199         NV_SWNCQ_IRQ_HOTPLUG            = NV_SWNCQ_IRQ_ADDED |
 200                                           NV_SWNCQ_IRQ_REMOVED,
 201
 202 };
 203
 204 /* ADMA Physical Region Descriptor - one SG segment */
 205 struct nv_adma_prd {
 206         __le64                  addr;
 207         __le32                  len;
 208         u8                      flags;
 209         u8                      packet_len;
 210         __le16                  reserved;
 211 };
 212
 213 enum nv_adma_regbits {
 214         CMDEND  = (1 << 15),            /* end of command list */
 215         WNB     = (1 << 14),            /* wait-not-BSY */
 216         IGN     = (1 << 13),            /* ignore this entry */
 217         CS1n    = (1 << (4 + 8)),       /* std. PATA signals follow... */
 218         DA2     = (1 << (2 + 8)),
 219         DA1     = (1 << (1 + 8)),
 220         DA0     = (1 << (0 + 8)),
 221 };
 222
 223 /* ADMA Command Parameter Block
 224    The first 5 SG segments are stored inside the Command Parameter Block itself.
 225    If there are more than 5 segments the remainder are stored in a separate
 226    memory area indicated by next_aprd. */
 227 struct nv_adma_cpb {
 228         u8                      resp_flags;    /* 0 */
 229         u8                      reserved1;     /* 1 */
 230         u8                      ctl_flags;     /* 2 */
 231         /* len is length of taskfile in 64 bit words */
 232         u8                      len;            /* 3  */
 233         u8                      tag;           /* 4 */
 234         u8                      next_cpb_idx;  /* 5 */
 235         __le16                  reserved2;     /* 6-7 */
 236         __le16                  tf[12];        /* 8-31 */
 237         struct nv_adma_prd      aprd[5];       /* 32-111 */
 238         __le64                  next_aprd;     /* 112-119 */
 239         __le64                  reserved3;     /* 120-127 */
 240 };
 241
 242
 243 struct nv_adma_port_priv {
 244         struct nv_adma_cpb      *cpb;
 245         dma_addr_t              cpb_dma;
 246         struct nv_adma_prd      *aprd;
 247         dma_addr_t              aprd_dma;
 248         void __iomem            *ctl_block;
 249         void __iomem            *gen_block;
 250         void __iomem            *notifier_clear_block;
 251         u64                     adma_dma_mask;
 252         u8                      flags;
 253         int                     last_issue_ncq;
 254 };
 255
 256 struct nv_host_priv {
 257         unsigned long           type;
 258 };
 259
 260 struct defer_queue {
 261         u32             defer_bits;
 262         unsigned int    head;
 263         unsigned int    tail;
 264         unsigned int    tag[ATA_MAX_QUEUE];
 265 };
 266
 267 enum ncq_saw_flag_list {
 268         ncq_saw_d2h     = (1U << 0),
 269         ncq_saw_dmas    = (1U << 1),
 270         ncq_saw_sdb     = (1U << 2),
 271         ncq_saw_backout = (1U << 3),
 272 };
 273
 274 struct nv_swncq_port_priv {
 275         struct ata_bmdma_prd *prd;       /* our SG list */
 276         dma_addr_t      prd_dma; /* and its DMA mapping */
 277         void __iomem    *sactive_block;
 278         void __iomem    *irq_block;
 279         void __iomem    *tag_block;
 280         u32             qc_active;
 281
 282         unsigned int    last_issue_tag;
 283
 284         /* fifo circular queue to store deferral command */
 285         struct defer_queue defer_queue;
 286
 287         /* for NCQ interrupt analysis */
 288         u32             dhfis_bits;
 289         u32             dmafis_bits;
 290         u32             sdbfis_bits;
 291
 292         unsigned int    ncq_flags;
 293 };
 294
 295
 296 #define NV_ADMA_CHECK_INTR(GCTL, PORT) ((GCTL) & (1 << (19 + (12 * (PORT)))))
 297
 298 static int nv_init_one(struct pci_dev *pdev, const struct pci_device_id *ent);
 299 #ifdef CONFIG_PM
 300 static int nv_pci_device_resume(struct pci_dev *pdev);
 301 #endif
 302 static void nv_ck804_host_stop(struct ata_host *host);
 303 static irqreturn_t nv_generic_interrupt(int irq, void *dev_instance);
 304 static irqreturn_t nv_nf2_interrupt(int irq, void *dev_instance);
 305 static irqreturn_t nv_ck804_interrupt(int irq, void *dev_instance);
 306 static int nv_scr_read(struct ata_link *link, unsigned int sc_reg, u32 *val);
 307 static int nv_scr_write(struct ata_link *link, unsigned int sc_reg, u32 val);
 308
 309 static int nv_hardreset(struct ata_link *link, unsigned int *class,
 310                         unsigned long deadline);
 311 static void nv_nf2_freeze(struct ata_port *ap);
 312 static void nv_nf2_thaw(struct ata_port *ap);
 313 static void nv_ck804_freeze(struct ata_port *ap);
 314 static void nv_ck804_thaw(struct ata_port *ap);
 315 static int nv_adma_slave_config(struct scsi_device *sdev);
 316 static int nv_adma_check_atapi_dma(struct ata_queued_cmd *qc);
 317 static void nv_adma_qc_prep(struct ata_queued_cmd *qc);
 318 static unsigned int nv_adma_qc_issue(struct ata_queued_cmd *qc);
 319 static irqreturn_t nv_adma_interrupt(int irq, void *dev_instance);
 320 static void nv_adma_irq_clear(struct ata_port *ap);
 321 static int nv_adma_port_start(struct ata_port *ap);
 322 static void nv_adma_port_stop(struct ata_port *ap);
 323 #ifdef CONFIG_PM
 324 static int nv_adma_port_suspend(struct ata_port *ap, pm_message_t mesg);
 325 static int nv_adma_port_resume(struct ata_port *ap);
 326 #endif
 327 static void nv_adma_freeze(struct ata_port *ap);
 328 static void nv_adma_thaw(struct ata_port *ap);
 329 static void nv_adma_error_handler(struct ata_port *ap);
 330 static void nv_adma_host_stop(struct ata_host *host);
 331 static void nv_adma_post_internal_cmd(struct ata_queued_cmd *qc);
 332 static void nv_adma_tf_read(struct ata_port *ap, struct ata_taskfile *tf);
 333
 334 static void nv_mcp55_thaw(struct ata_port *ap);
 335 static void nv_mcp55_freeze(struct ata_port *ap);
 336 static void nv_swncq_error_handler(struct ata_port *ap);
 337 static int nv_swncq_slave_config(struct scsi_device *sdev);
 338 static int nv_swncq_port_start(struct ata_port *ap);
 339 static void nv_swncq_qc_prep(struct ata_queued_cmd *qc);
 340 static void nv_swncq_fill_sg(struct ata_queued_cmd *qc);
 341 static unsigned int nv_swncq_qc_issue(struct ata_queued_cmd *qc);
 342 static void nv_swncq_irq_clear(struct ata_port *ap, u16 fis);
 343 static irqreturn_t nv_swncq_interrupt(int irq, void *dev_instance);
 344 #ifdef CONFIG_PM
 345 static int nv_swncq_port_suspend(struct ata_port *ap, pm_message_t mesg);
 346 static int nv_swncq_port_resume(struct ata_port *ap);
 347 #endif
 348
 349 enum nv_host_type
 350 {
 351         GENERIC,
 352         NFORCE2,
 353         NFORCE3 = NFORCE2,      /* NF2 == NF3 as far as sata_nv is concerned */
 354         CK804,
 355         ADMA,
 356         MCP5x,
 357         SWNCQ,
 358 };
 359
 360 static const struct pci_device_id nv_pci_tbl[] = {
 361         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE2S_SATA), NFORCE2 },
 362         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE3S_SATA), NFORCE3 },
 363         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE3S_SATA2), NFORCE3 },
 364         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_CK804_SATA), CK804 },
 365         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_CK804_SATA2), CK804 },
 366         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP04_SATA), CK804 },
 367         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP04_SATA2), CK804 },
 368         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP51_SATA), MCP5x },
 369         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP51_SATA2), MCP5x },
 370         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP55_SATA), MCP5x },
 371         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP55_SATA2), MCP5x },
 372         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP61_SATA), GENERIC },
 373         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP61_SATA2), GENERIC },
 374         { PCI_VDEVICE(NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE_MCP61_SATA3), GENERIC },
 375
 376         { } /* terminate list */
 377 };
 378
 379 static struct pci_driver nv_pci_driver = {
 380         .name                   = DRV_NAME,
 381         .id_table               = nv_pci_tbl,
 382         .probe                  = nv_init_one,
 383 #ifdef CONFIG_PM
 384         .suspend                = ata_pci_device_suspend,
 385         .resume                 = nv_pci_device_resume,
 386 #endif
 387         .remove                 = ata_pci_remove_one,
 388 };
 389
 390 static struct scsi_host_template nv_sht = {
 391         ATA_BMDMA_SHT(DRV_NAME),
 392 };
 393
 394 static struct scsi_host_template nv_adma_sht = {
 395         ATA_NCQ_SHT(DRV_NAME),
 396         .can_queue              = NV_ADMA_MAX_CPBS,
 397         .sg_tablesize           = NV_ADMA_SGTBL_TOTAL_LEN,
 398         .dma_boundary           = NV_ADMA_DMA_BOUNDARY,
 399         .slave_configure        = nv_adma_slave_config,
 400 };
 401
 402 static struct scsi_host_template nv_swncq_sht = {
 403         ATA_NCQ_SHT(DRV_NAME),
 404         .can_queue              = ATA_MAX_QUEUE,
 405         .sg_tablesize           = LIBATA_MAX_PRD,
 406         .dma_boundary           = ATA_DMA_BOUNDARY,
 407         .slave_configure        = nv_swncq_slave_config,
 408 };
 409
 410 /*
 411  * NV SATA controllers have various different problems with hardreset
 412  * protocol depending on the specific controller and device.
 413  *
 414  * GENERIC:
 415  *
 416  *  bko11195 reports that link doesn't come online after hardreset on
 417  *  generic nv's and there have been several other similar reports on
 418  *  linux-ide.
 419  *
 420  *  bko12351#c23 reports that warmplug on MCP61 doesn't work with
 421  *  softreset.
 422  *
 423  * NF2/3:
 424  *
 425  *  bko3352 reports nf2/3 controllers can't determine device signature
 426  *  reliably after hardreset.  The following thread reports detection
 427  *  failure on cold boot with the standard debouncing timing.
 428  *
 429  *  http://thread.gmane.org/gmane.linux.ide/34098
 430  *
 431  *  bko12176 reports that hardreset fails to bring up the link during
 432  *  boot on nf2.
 433  *
 434  * CK804:
 435  *
 436  *  For initial probing after boot and hot plugging, hardreset mostly
 437  *  works fine on CK804 but curiously, reprobing on the initial port
 438  *  by rescanning or rmmod/insmod fails to acquire the initial D2H Reg
 439  *  FIS in somewhat undeterministic way.
 440  *
 441  * SWNCQ:
 442  *
 443  *  bko12351 reports that when SWNCQ is enabled, for hotplug to work,
 444  *  hardreset should be used and hardreset can't report proper
 445  *  signature, which suggests that mcp5x is closer to nf2 as long as
 446  *  reset quirkiness is concerned.
 447  *
 448  *  bko12703 reports that boot probing fails for intel SSD with
 449  *  hardreset.  Link fails to come online.  Softreset works fine.
 450  *
 451  * The failures are varied but the following patterns seem true for
 452  * all flavors.
 453  *
 454  * - Softreset during boot always works.
 455  *
 456  * - Hardreset during boot sometimes fails to bring up the link on
 457  *   certain comibnations and device signature acquisition is
 458  *   unreliable.
 459  *
 460  * - Hardreset is often necessary after hotplug.
 461  *
 462  * So, preferring softreset for boot probing and error handling (as
 463  * hardreset might bring down the link) but using hardreset for
 464  * post-boot probing should work around the above issues in most
 465  * cases.  Define nv_hardreset() which only kicks in for post-boot
 466  * probing and use it for all variants.
 467  */
 468 static struct ata_port_operations nv_generic_ops = {
 469         .inherits               = &ata_bmdma_port_ops,
 470         .lost_interrupt         = ATA_OP_NULL,
 471         .scr_read               = nv_scr_read,
 472         .scr_write              = nv_scr_write,
 473         .hardreset              = nv_hardreset,
 474 };
 475
 476 static struct ata_port_operations nv_nf2_ops = {
 477         .inherits               = &nv_generic_ops,
 478         .freeze                 = nv_nf2_freeze,
 479         .thaw                   = nv_nf2_thaw,
 480 };
 481
 482 static struct ata_port_operations nv_ck804_ops = {
 483         .inherits               = &nv_generic_ops,
 484         .freeze                 = nv_ck804_freeze,
 485         .thaw                   = nv_ck804_thaw,
 486         .host_stop              = nv_ck804_host_stop,
 487 };
 488
 489 static struct ata_port_operations nv_adma_ops = {
 490         .inherits               = &nv_ck804_ops,
 491
 492         .check_atapi_dma        = nv_adma_check_atapi_dma,
 493         .sff_tf_read            = nv_adma_tf_read,
 494         .qc_defer               = ata_std_qc_defer,
 495         .qc_prep                = nv_adma_qc_prep,
 496         .qc_issue               = nv_adma_qc_issue,
 497         .sff_irq_clear          = nv_adma_irq_clear,
 498
 499         .freeze                 = nv_adma_freeze,
 500         .thaw                   = nv_adma_thaw,
 501         .error_handler          = nv_adma_error_handler,
 502         .post_internal_cmd      = nv_adma_post_internal_cmd,
 503
 504         .port_start             = nv_adma_port_start,
 505         .port_stop              = nv_adma_port_stop,
 506 #ifdef CONFIG_PM
 507         .port_suspend           = nv_adma_port_suspend,
 508         .port_resume            = nv_adma_port_resume,
 509 #endif
 510         .host_stop              = nv_adma_host_stop,
 511 };
 512
 513 static struct ata_port_operations nv_swncq_ops = {
 514         .inherits               = &nv_generic_ops,
 515
 516         .qc_defer               = ata_std_qc_defer,
 517         .qc_prep                = nv_swncq_qc_prep,
 518         .qc_issue               = nv_swncq_qc_issue,
 519
 520         .freeze                 = nv_mcp55_freeze,
 521         .thaw                   = nv_mcp55_thaw,
 522         .error_handler          = nv_swncq_error_handler,
 523
 524 #ifdef CONFIG_PM
 525         .port_suspend           = nv_swncq_port_suspend,
 526         .port_resume            = nv_swncq_port_resume,
 527 #endif
 528         .port_start             = nv_swncq_port_start,
 529 };
 530
 531 struct nv_pi_priv {
 532         irq_handler_t                   irq_handler;
 533         struct scsi_host_template       *sht;
 534 };
 535
 536 #define NV_PI_PRIV(_irq_handler, _sht) \
 537         &(struct nv_pi_priv){ .irq_handler = _irq_handler, .sht = _sht }
 538
 539 static const struct ata_port_info nv_port_info[] = {
 540         /* generic */
 541         {
 542                 .flags          = ATA_FLAG_SATA,
 543                 .pio_mask       = NV_PIO_MASK,
 544                 .mwdma_mask     = NV_MWDMA_MASK,
 545                 .udma_mask      = NV_UDMA_MASK,
 546                 .port_ops       = &nv_generic_ops,
 547                 .private_data   = NV_PI_PRIV(nv_generic_interrupt, &nv_sht),
 548         },
 549         /* nforce2/3 */
 550         {
 551                 .flags          = ATA_FLAG_SATA,
 552                 .pio_mask       = NV_PIO_MASK,
 553                 .mwdma_mask     = NV_MWDMA_MASK,
 554                 .udma_mask      = NV_UDMA_MASK,
 555                 .port_ops       = &nv_nf2_ops,
 556                 .private_data   = NV_PI_PRIV(nv_nf2_interrupt, &nv_sht),
 557         },
 558         /* ck804 */
 559         {
 560                 .flags          = ATA_FLAG_SATA,
 561                 .pio_mask       = NV_PIO_MASK,
 562                 .mwdma_mask     = NV_MWDMA_MASK,
 563                 .udma_mask      = NV_UDMA_MASK,
 564                 .port_ops       = &nv_ck804_ops,
 565                 .private_data   = NV_PI_PRIV(nv_ck804_interrupt, &nv_sht),
 566         },
 567         /* ADMA */
 568         {
 569                 .flags          = ATA_FLAG_SATA | ATA_FLAG_NCQ,
 570                 .pio_mask       = NV_PIO_MASK,
 571                 .mwdma_mask     = NV_MWDMA_MASK,
 572                 .udma_mask      = NV_UDMA_MASK,
 573                 .port_ops       = &nv_adma_ops,
 574                 .private_data   = NV_PI_PRIV(nv_adma_interrupt, &nv_adma_sht),
 575         },
 576         /* MCP5x */
 577         {
 578                 .flags          = ATA_FLAG_SATA,
 579                 .pio_mask       = NV_PIO_MASK,
 580                 .mwdma_mask     = NV_MWDMA_MASK,
 581                 .udma_mask      = NV_UDMA_MASK,
 582                 .port_ops       = &nv_generic_ops,
 583                 .private_data   = NV_PI_PRIV(nv_generic_interrupt, &nv_sht),
 584         },
 585         /* SWNCQ */
 586         {
 587                 .flags          = ATA_FLAG_SATA | ATA_FLAG_NCQ,
 588                 .pio_mask       = NV_PIO_MASK,
 589                 .mwdma_mask     = NV_MWDMA_MASK,
 590                 .udma_mask      = NV_UDMA_MASK,
 591                 .port_ops       = &nv_swncq_ops,
 592                 .private_data   = NV_PI_PRIV(nv_swncq_interrupt, &nv_swncq_sht),
 593         },
 594 };
 595
 596 MODULE_AUTHOR("NVIDIA");
 597 MODULE_DESCRIPTION("low-level driver for NVIDIA nForce SATA controller");
 598 MODULE_LICENSE("GPL");
 599 MODULE_DEVICE_TABLE(pci, nv_pci_tbl);
 600 MODULE_VERSION(DRV_VERSION);
 601
 602 static bool adma_enabled;
 603 static bool swncq_enabled = 1;
 604 static bool msi_enabled;
 605
 606 static void nv_adma_register_mode(struct ata_port *ap)
 607 {
 608         struct nv_adma_port_priv *pp = ap->private_data;
 609         void __iomem *mmio = pp->ctl_block;
 610         u16 tmp, status;
 611         int count = 0;
 612
 613         if (pp->flags & NV_ADMA_PORT_REGISTER_MODE)
 614                 return;
 615
 616         status = readw(mmio + NV_ADMA_STAT);
 617         while (!(status & NV_ADMA_STAT_IDLE) && count < 20) {
 618                 ndelay(50);
 619                 status = readw(mmio + NV_ADMA_STAT);
 620                 count++;
 621         }
 622         if (count == 20)
 623                 ata_port_warn(ap, "timeout waiting for ADMA IDLE, stat=0x%hx\n",
 624                               status);
 625
 626         tmp = readw(mmio + NV_ADMA_CTL);
 627         writew(tmp & ~NV_ADMA_CTL_GO, mmio + NV_ADMA_CTL);
 628
 629         count = 0;
 630         status = readw(mmio + NV_ADMA_STAT);
 631         while (!(status & NV_ADMA_STAT_LEGACY) && count < 20) {
 632                 ndelay(50);
 633                 status = readw(mmio + NV_ADMA_STAT);
 634                 count++;
 635         }
 636         if (count == 20)
 637                 ata_port_warn(ap,
 638                               "timeout waiting for ADMA LEGACY, stat=0x%hx\n",
 639                               status);
 640
 641         pp->flags |= NV_ADMA_PORT_REGISTER_MODE;
 642 }
 643
 644 static void nv_adma_mode(struct ata_port *ap)
 645 {
 646         struct nv_adma_port_priv *pp = ap->private_data;
 647         void __iomem *mmio = pp->ctl_block;
 648         u16 tmp, status;
 649         int count = 0;
 650
 651         if (!(pp->flags & NV_ADMA_PORT_REGISTER_MODE))
 652                 return;
 653
 654         WARN_ON(pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE);
 655
 656         tmp = readw(mmio + NV_ADMA_CTL);
 657         writew(tmp | NV_ADMA_CTL_GO, mmio + NV_ADMA_CTL);
 658
 659         status = readw(mmio + NV_ADMA_STAT);
 660         while (((status & NV_ADMA_STAT_LEGACY) ||
 661               !(status & NV_ADMA_STAT_IDLE)) && count < 20) {
 662                 ndelay(50);
 663                 status = readw(mmio + NV_ADMA_STAT);
 664                 count++;
 665         }
 666         if (count == 20)
 667                 ata_port_warn(ap,
 668                         "timeout waiting for ADMA LEGACY clear and IDLE, stat=0x%hx\n",
 669                         status);
 670
 671         pp->flags &= ~NV_ADMA_PORT_REGISTER_MODE;
 672 }
 673
 674 static int nv_adma_slave_config(struct scsi_device *sdev)
 675 {
 676         struct ata_port *ap = ata_shost_to_port(sdev->host);
 677         struct nv_adma_port_priv *pp = ap->private_data;
 678         struct nv_adma_port_priv *port0, *port1;
 679         struct scsi_device *sdev0, *sdev1;
 680         struct pci_dev *pdev = to_pci_dev(ap->host->dev);
 681         unsigned long segment_boundary, flags;
 682         unsigned short sg_tablesize;
 683         int rc;
 684         int adma_enable;
 685         u32 current_reg, new_reg, config_mask;
 686
 687         rc = ata_scsi_slave_config(sdev);
 688
 689         if (sdev->id >= ATA_MAX_DEVICES || sdev->channel || sdev->lun)
 690                 /* Not a proper libata device, ignore */
 691                 return rc;
 692
 693         spin_lock_irqsave(ap->lock, flags);
 694
 695         if (ap->link.device[sdev->id].class == ATA_DEV_ATAPI) {
 696                 /*
 697                  * NVIDIA reports that ADMA mode does not support ATAPI commands.
 698                  * Therefore ATAPI commands are sent through the legacy interface.
 699                  * However, the legacy interface only supports 32-bit DMA.
 700                  * Restrict DMA parameters as required by the legacy interface
 701                  * when an ATAPI device is connected.
 702                  */
 703                 segment_boundary = ATA_DMA_BOUNDARY;
 704                 /* Subtract 1 since an extra entry may be needed for padding, see
 705                    libata-scsi.c */
 706                 sg_tablesize = LIBATA_MAX_PRD - 1;
 707
 708                 /* Since the legacy DMA engine is in use, we need to disable ADMA
 709                    on the port. */
 710                 adma_enable = 0;
 711                 nv_adma_register_mode(ap);
 712         } else {
 713                 segment_boundary = NV_ADMA_DMA_BOUNDARY;
 714                 sg_tablesize = NV_ADMA_SGTBL_TOTAL_LEN;
 715                 adma_enable = 1;
 716         }
 717
 718         pci_read_config_dword(pdev, NV_MCP_SATA_CFG_20, &current_reg);
 719
 720         if (ap->port_no == 1)
 721                 config_mask = NV_MCP_SATA_CFG_20_PORT1_EN |
 722                               NV_MCP_SATA_CFG_20_PORT1_PWB_EN;
 723         else
 724                 config_mask = NV_MCP_SATA_CFG_20_PORT0_EN |
 725                               NV_MCP_SATA_CFG_20_PORT0_PWB_EN;
 726
 727         if (adma_enable) {
 728                 new_reg = current_reg | config_mask;
 729                 pp->flags &= ~NV_ADMA_ATAPI_SETUP_COMPLETE;
 730         } else {
 731                 new_reg = current_reg & ~config_mask;
 732                 pp->flags |= NV_ADMA_ATAPI_SETUP_COMPLETE;
 733         }
 734
 735         if (current_reg != new_reg)
 736                 pci_write_config_dword(pdev, NV_MCP_SATA_CFG_20, new_reg);
 737
 738         port0 = ap->host->ports[0]->private_data;
 739         port1 = ap->host->ports[1]->private_data;
 740         sdev0 = ap->host->ports[0]->link.device[0].sdev;
 741         sdev1 = ap->host->ports[1]->link.device[0].sdev;
 742         if ((port0->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) ||
 743             (port1->flags & NV_ADMA_ATAPI_SETUP_COMPLETE)) {
 744                 /** We have to set the DMA mask to 32-bit if either port is in
 745                     ATAPI mode, since they are on the same PCI device which is
 746                     used for DMA mapping. If we set the mask we also need to set
 747                     the bounce limit on both ports to ensure that the block
 748                     layer doesn't feed addresses that cause DMA mapping to
 749                     choke. If either SCSI device is not allocated yet, it's OK
 750                     since that port will discover its correct setting when it
 751                     does get allocated.
 752                     Note: Setting 32-bit mask should not fail. */
 753                 if (sdev0)
 754                         blk_queue_bounce_limit(sdev0->request_queue,
 755                                                ATA_DMA_MASK);
 756                 if (sdev1)
 757                         blk_queue_bounce_limit(sdev1->request_queue,
 758                                                ATA_DMA_MASK);
 759
 760                 pci_set_dma_mask(pdev, ATA_DMA_MASK);
 761         } else {
 762                 /** This shouldn't fail as it was set to this value before */
 763                 pci_set_dma_mask(pdev, pp->adma_dma_mask);
 764                 if (sdev0)
 765                         blk_queue_bounce_limit(sdev0->request_queue,
 766                                                pp->adma_dma_mask);
 767                 if (sdev1)
 768                         blk_queue_bounce_limit(sdev1->request_queue,
 769                                                pp->adma_dma_mask);
 770         }
 771
 772         blk_queue_segment_boundary(sdev->request_queue, segment_boundary);
 773         blk_queue_max_segments(sdev->request_queue, sg_tablesize);
 774         ata_port_info(ap,
 775                       "DMA mask 0x%llX, segment boundary 0x%lX, hw segs %hu\n",
 776                       (unsigned long long)*ap->host->dev->dma_mask,
 777                       segment_boundary, sg_tablesize);
 778
 779         spin_unlock_irqrestore(ap->lock, flags);
 780
 781         return rc;
 782 }
 783
 784 static int nv_adma_check_atapi_dma(struct ata_queued_cmd *qc)
 785 {
 786         struct nv_adma_port_priv *pp = qc->ap->private_data;
 787         return !(pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE);
 788 }
 789
 790 static void nv_adma_tf_read(struct ata_port *ap, struct ata_taskfile *tf)
 791 {
 792         /* Other than when internal or pass-through commands are executed,
 793            the only time this function will be called in ADMA mode will be
 794            if a command fails. In the failure case we don't care about going
 795            into register mode with ADMA commands pending, as the commands will
 796            all shortly be aborted anyway. We assume that NCQ commands are not
 797            issued via passthrough, which is the only way that switching into
 798            ADMA mode could abort outstanding commands. */
 799         nv_adma_register_mode(ap);
 800
 801         ata_sff_tf_read(ap, tf);
 802 }
 803
 804 static unsigned int nv_adma_tf_to_cpb(struct ata_taskfile *tf, __le16 *cpb)
 805 {
 806         unsigned int idx = 0;
 807
 808         if (tf->flags & ATA_TFLAG_ISADDR) {
 809                 if (tf->flags & ATA_TFLAG_LBA48) {
 810                         cpb[idx++] = cpu_to_le16((ATA_REG_ERR   << 8) | tf->hob_feature | WNB);
 811                         cpb[idx++] = cpu_to_le16((ATA_REG_NSECT << 8) | tf->hob_nsect);
 812                         cpb[idx++] = cpu_to_le16((ATA_REG_LBAL  << 8) | tf->hob_lbal);
 813                         cpb[idx++] = cpu_to_le16((ATA_REG_LBAM  << 8) | tf->hob_lbam);
 814                         cpb[idx++] = cpu_to_le16((ATA_REG_LBAH  << 8) | tf->hob_lbah);
 815                         cpb[idx++] = cpu_to_le16((ATA_REG_ERR    << 8) | tf->feature);
 816                 } else
 817                         cpb[idx++] = cpu_to_le16((ATA_REG_ERR    << 8) | tf->feature | WNB);
 818
 819                 cpb[idx++] = cpu_to_le16((ATA_REG_NSECT  << 8) | tf->nsect);
 820                 cpb[idx++] = cpu_to_le16((ATA_REG_LBAL   << 8) | tf->lbal);
 821                 cpb[idx++] = cpu_to_le16((ATA_REG_LBAM   << 8) | tf->lbam);
 822                 cpb[idx++] = cpu_to_le16((ATA_REG_LBAH   << 8) | tf->lbah);
 823         }
 824
 825         if (tf->flags & ATA_TFLAG_DEVICE)
 826                 cpb[idx++] = cpu_to_le16((ATA_REG_DEVICE << 8) | tf->device);
 827
 828         cpb[idx++] = cpu_to_le16((ATA_REG_CMD    << 8) | tf->command | CMDEND);
 829
 830         while (idx < 12)
 831                 cpb[idx++] = cpu_to_le16(IGN);
 832
 833         return idx;
 834 }
 835
 836 static int nv_adma_check_cpb(struct ata_port *ap, int cpb_num, int force_err)
 837 {
 838         struct nv_adma_port_priv *pp = ap->private_data;
 839         u8 flags = pp->cpb[cpb_num].resp_flags;
 840
 841         VPRINTK("CPB %d, flags=0x%x\n", cpb_num, flags);
 842
 843         if (unlikely((force_err ||
 844                      flags & (NV_CPB_RESP_ATA_ERR |
 845                               NV_CPB_RESP_CMD_ERR |
 846                               NV_CPB_RESP_CPB_ERR)))) {
 847                 struct ata_eh_info *ehi = &ap->link.eh_info;
 848                 int freeze = 0;
 849
 850                 ata_ehi_clear_desc(ehi);
 851                 __ata_ehi_push_desc(ehi, "CPB resp_flags 0x%x: ", flags);
 852                 if (flags & NV_CPB_RESP_ATA_ERR) {
 853                         ata_ehi_push_desc(ehi, "ATA error");
 854                         ehi->err_mask |= AC_ERR_DEV;
 855                 } else if (flags & NV_CPB_RESP_CMD_ERR) {
 856                         ata_ehi_push_desc(ehi, "CMD error");
 857                         ehi->err_mask |= AC_ERR_DEV;
 858                 } else if (flags & NV_CPB_RESP_CPB_ERR) {
 859                         ata_ehi_push_desc(ehi, "CPB error");
 860                         ehi->err_mask |= AC_ERR_SYSTEM;
 861                         freeze = 1;
 862                 } else {
 863                         /* notifier error, but no error in CPB flags? */
 864                         ata_ehi_push_desc(ehi, "unknown");
 865                         ehi->err_mask |= AC_ERR_OTHER;
 866                         freeze = 1;
 867                 }
 868                 /* Kill all commands. EH will determine what actually failed. */
 869                 if (freeze)
 870                         ata_port_freeze(ap);
 871                 else
 872                         ata_port_abort(ap);
 873                 return -1;
 874         }
 875
 876         if (likely(flags & NV_CPB_RESP_DONE))
 877                 return 1;
 878         return 0;
 879 }
 880
 881 static int nv_host_intr(struct ata_port *ap, u8 irq_stat)
 882 {
 883         struct ata_queued_cmd *qc = ata_qc_from_tag(ap, ap->link.active_tag);
 884
 885         /* freeze if hotplugged */
 886         if (unlikely(irq_stat & (NV_INT_ADDED | NV_INT_REMOVED))) {
 887                 ata_port_freeze(ap);
 888                 return 1;
 889         }
 890
 891         /* bail out if not our interrupt */
 892         if (!(irq_stat & NV_INT_DEV))
 893                 return 0;
 894
 895         /* DEV interrupt w/ no active qc? */
 896         if (unlikely(!qc || (qc->tf.flags & ATA_TFLAG_POLLING))) {
 897                 ata_sff_check_status(ap);
 898                 return 1;
 899         }
 900
 901         /* handle interrupt */
 902         return ata_bmdma_port_intr(ap, qc);
 903 }
 904
 905 static irqreturn_t nv_adma_interrupt(int irq, void *dev_instance)
 906 {
 907         struct ata_host *host = dev_instance;
 908         int i, handled = 0;
 909         u32 notifier_clears[2];
 910
 911         spin_lock(&host->lock);
 912
 913         for (i = 0; i < host->n_ports; i++) {
 914                 struct ata_port *ap = host->ports[i];
 915                 struct nv_adma_port_priv *pp = ap->private_data;
 916                 void __iomem *mmio = pp->ctl_block;
 917                 u16 status;
 918                 u32 gen_ctl;
 919                 u32 notifier, notifier_error;
 920
 921                 notifier_clears[i] = 0;
 922
 923                 /* if ADMA is disabled, use standard ata interrupt handler */
 924                 if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) {
 925                         u8 irq_stat = readb(host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_CK804)
 926                                 >> (NV_INT_PORT_SHIFT * i);
 927                         handled += nv_host_intr(ap, irq_stat);
 928                         continue;
 929                 }
 930
 931                 /* if in ATA register mode, check for standard interrupts */
 932                 if (pp->flags & NV_ADMA_PORT_REGISTER_MODE) {
 933                         u8 irq_stat = readb(host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_CK804)
 934                                 >> (NV_INT_PORT_SHIFT * i);
 935                         if (ata_tag_valid(ap->link.active_tag))
 936                                 /** NV_INT_DEV indication seems unreliable
 937                                     at times at least in ADMA mode. Force it
 938                                     on always when a command is active, to
 939                                     prevent losing interrupts. */
 940                                 irq_stat |= NV_INT_DEV;
 941                         handled += nv_host_intr(ap, irq_stat);
 942                 }
 943
 944                 notifier = readl(mmio + NV_ADMA_NOTIFIER);
 945                 notifier_error = readl(mmio + NV_ADMA_NOTIFIER_ERROR);
 946                 notifier_clears[i] = notifier | notifier_error;
 947
 948                 gen_ctl = readl(pp->gen_block + NV_ADMA_GEN_CTL);
 949
 950                 if (!NV_ADMA_CHECK_INTR(gen_ctl, ap->port_no) && !notifier &&
 951                     !notifier_error)
 952                         /* Nothing to do */
 953                         continue;
 954
 955                 status = readw(mmio + NV_ADMA_STAT);
 956
 957                 /*
 958                  * Clear status. Ensure the controller sees the
 959                  * clearing before we start looking at any of the CPB
 960                  * statuses, so that any CPB completions after this
 961                  * point in the handler will raise another interrupt.
 962                  */
 963                 writew(status, mmio + NV_ADMA_STAT);
 964                 readw(mmio + NV_ADMA_STAT); /* flush posted write */
 965                 rmb();
 966
 967                 handled++; /* irq handled if we got here */
 968
 969                 /* freeze if hotplugged or controller error */
 970                 if (unlikely(status & (NV_ADMA_STAT_HOTPLUG |
 971                                        NV_ADMA_STAT_HOTUNPLUG |
 972                                        NV_ADMA_STAT_TIMEOUT |
 973                                        NV_ADMA_STAT_SERROR))) {
 974                         struct ata_eh_info *ehi = &ap->link.eh_info;
 975
 976                         ata_ehi_clear_desc(ehi);
 977                         __ata_ehi_push_desc(ehi, "ADMA status 0x%08x: ", status);
 978                         if (status & NV_ADMA_STAT_TIMEOUT) {
 979                                 ehi->err_mask |= AC_ERR_SYSTEM;
 980                                 ata_ehi_push_desc(ehi, "timeout");
 981                         } else if (status & NV_ADMA_STAT_HOTPLUG) {
 982                                 ata_ehi_hotplugged(ehi);
 983                                 ata_ehi_push_desc(ehi, "hotplug");
 984                         } else if (status & NV_ADMA_STAT_HOTUNPLUG) {
 985                                 ata_ehi_hotplugged(ehi);
 986                                 ata_ehi_push_desc(ehi, "hot unplug");
 987                         } else if (status & NV_ADMA_STAT_SERROR) {
 988                                 /* let EH analyze SError and figure out cause */
 989                                 ata_ehi_push_desc(ehi, "SError");
 990                         } else
 991                                 ata_ehi_push_desc(ehi, "unknown");
 992                         ata_port_freeze(ap);
 993                         continue;
 994                 }
 995
 996                 if (status & (NV_ADMA_STAT_DONE |
 997                               NV_ADMA_STAT_CPBERR |
 998                               NV_ADMA_STAT_CMD_COMPLETE)) {
 999                         u32 check_commands = notifier_clears[i];
1000                         u32 done_mask = 0;
1001                         int pos, rc;
1002
1003                         if (status & NV_ADMA_STAT_CPBERR) {
1004                                 /* check all active commands */
1005                                 if (ata_tag_valid(ap->link.active_tag))
1006                                         check_commands = 1 <<
1007                                                 ap->link.active_tag;
1008                                 else
1009                                         check_commands = ap->link.sactive;
1010                         }
1011
1012                         /* check CPBs for completed commands */
1013                         while ((pos = ffs(check_commands))) {
1014                                 pos--;
1015                                 rc = nv_adma_check_cpb(ap, pos,
1016                                                 notifier_error & (1 << pos));
1017                                 if (rc > 0)
1018                                         done_mask |= 1 << pos;
1019                                 else if (unlikely(rc < 0))
1020                                         check_commands = 0;
1021                                 check_commands &= ~(1 << pos);
1022                         }
1023                         ata_qc_complete_multiple(ap, ap->qc_active ^ done_mask);
1024                 }
1025         }
1026
1027         if (notifier_clears[0] || notifier_clears[1]) {
1028                 /* Note: Both notifier clear registers must be written
1029                    if either is set, even if one is zero, according to NVIDIA. */
1030                 struct nv_adma_port_priv *pp = host->ports[0]->private_data;
1031                 writel(notifier_clears[0], pp->notifier_clear_block);
1032                 pp = host->ports[1]->private_data;
1033                 writel(notifier_clears[1], pp->notifier_clear_block);
1034         }
1035
1036         spin_unlock(&host->lock);
1037
1038         return IRQ_RETVAL(handled);
1039 }
1040
1041 static void nv_adma_freeze(struct ata_port *ap)
1042 {
1043         struct nv_adma_port_priv *pp = ap->private_data;
1044         void __iomem *mmio = pp->ctl_block;
1045         u16 tmp;
1046
1047         nv_ck804_freeze(ap);
1048
1049         if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE)
1050                 return;
1051
1052         /* clear any outstanding CK804 notifications */
1053         writeb(NV_INT_ALL << (ap->port_no * NV_INT_PORT_SHIFT),
1054                 ap->host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_CK804);
1055
1056         /* Disable interrupt */
1057         tmp = readw(mmio + NV_ADMA_CTL);
1058         writew(tmp & ~(NV_ADMA_CTL_AIEN | NV_ADMA_CTL_HOTPLUG_IEN),
1059                 mmio + NV_ADMA_CTL);
1060         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1061 }
1062
1063 static void nv_adma_thaw(struct ata_port *ap)
1064 {
1065         struct nv_adma_port_priv *pp = ap->private_data;
1066         void __iomem *mmio = pp->ctl_block;
1067         u16 tmp;
1068
1069         nv_ck804_thaw(ap);
1070
1071         if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE)
1072                 return;
1073
1074         /* Enable interrupt */
1075         tmp = readw(mmio + NV_ADMA_CTL);
1076         writew(tmp | (NV_ADMA_CTL_AIEN | NV_ADMA_CTL_HOTPLUG_IEN),
1077                 mmio + NV_ADMA_CTL);
1078         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1079 }
1080
1081 static void nv_adma_irq_clear(struct ata_port *ap)
1082 {
1083         struct nv_adma_port_priv *pp = ap->private_data;
1084         void __iomem *mmio = pp->ctl_block;
1085         u32 notifier_clears[2];
1086
1087         if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) {
1088                 ata_bmdma_irq_clear(ap);
1089                 return;
1090         }
1091
1092         /* clear any outstanding CK804 notifications */
1093         writeb(NV_INT_ALL << (ap->port_no * NV_INT_PORT_SHIFT),
1094                 ap->host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_CK804);
1095
1096         /* clear ADMA status */
1097         writew(0xffff, mmio + NV_ADMA_STAT);
1098
1099         /* clear notifiers - note both ports need to be written with
1100            something even though we are only clearing on one */
1101         if (ap->port_no == 0) {
1102                 notifier_clears[0] = 0xFFFFFFFF;
1103                 notifier_clears[1] = 0;
1104         } else {
1105                 notifier_clears[0] = 0;
1106                 notifier_clears[1] = 0xFFFFFFFF;
1107         }
1108         pp = ap->host->ports[0]->private_data;
1109         writel(notifier_clears[0], pp->notifier_clear_block);
1110         pp = ap->host->ports[1]->private_data;
1111         writel(notifier_clears[1], pp->notifier_clear_block);
1112 }
1113
1114 static void nv_adma_post_internal_cmd(struct ata_queued_cmd *qc)
1115 {
1116         struct nv_adma_port_priv *pp = qc->ap->private_data;
1117
1118         if (pp->flags & NV_ADMA_PORT_REGISTER_MODE)
1119                 ata_bmdma_post_internal_cmd(qc);
1120 }
1121
1122 static int nv_adma_port_start(struct ata_port *ap)
1123 {
1124         struct device *dev = ap->host->dev;
1125         struct nv_adma_port_priv *pp;
1126         int rc;
1127         void *mem;
1128         dma_addr_t mem_dma;
1129         void __iomem *mmio;
1130         struct pci_dev *pdev = to_pci_dev(dev);
1131         u16 tmp;
1132
1133         VPRINTK("ENTER\n");
1134
1135         /* Ensure DMA mask is set to 32-bit before allocating legacy PRD and
1136            pad buffers */
1137         rc = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));
1138         if (rc)
1139                 return rc;
1140         rc = pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(32));
1141         if (rc)
1142                 return rc;
1143
1144         /* we might fallback to bmdma, allocate bmdma resources */
1145         rc = ata_bmdma_port_start(ap);
1146         if (rc)
1147                 return rc;
1148
1149         pp = devm_kzalloc(dev, sizeof(*pp), GFP_KERNEL);
1150         if (!pp)
1151                 return -ENOMEM;
1152
1153         mmio = ap->host->iomap[NV_MMIO_BAR] + NV_ADMA_PORT +
1154                ap->port_no * NV_ADMA_PORT_SIZE;
1155         pp->ctl_block = mmio;
1156         pp->gen_block = ap->host->iomap[NV_MMIO_BAR] + NV_ADMA_GEN;
1157         pp->notifier_clear_block = pp->gen_block +
1158                NV_ADMA_NOTIFIER_CLEAR + (4 * ap->port_no);
1159
1160         /* Now that the legacy PRD and padding buffer are allocated we can
1161            safely raise the DMA mask to allocate the CPB/APRD table.
1162            These are allowed to fail since we store the value that ends up
1163            being used to set as the bounce limit in slave_config later if
1164            needed. */
1165         pci_set_dma_mask(pdev, DMA_BIT_MASK(64));
1166         pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(64));
1167         pp->adma_dma_mask = *dev->dma_mask;
1168
1169         mem = dmam_alloc_coherent(dev, NV_ADMA_PORT_PRIV_DMA_SZ,
1170                                   &mem_dma, GFP_KERNEL);
1171         if (!mem)
1172                 return -ENOMEM;
1173         memset(mem, 0, NV_ADMA_PORT_PRIV_DMA_SZ);
1174
1175         /*
1176          * First item in chunk of DMA memory:
1177          * 128-byte command parameter block (CPB)
1178          * one for each command tag
1179          */
1180         pp->cpb     = mem;
1181         pp->cpb_dma = mem_dma;
1182
1183         writel(mem_dma & 0xFFFFFFFF,    mmio + NV_ADMA_CPB_BASE_LOW);
1184         writel((mem_dma >> 16) >> 16,   mmio + NV_ADMA_CPB_BASE_HIGH);
1185
1186         mem     += NV_ADMA_MAX_CPBS * NV_ADMA_CPB_SZ;
1187         mem_dma += NV_ADMA_MAX_CPBS * NV_ADMA_CPB_SZ;
1188
1189         /*
1190          * Second item: block of ADMA_SGTBL_LEN s/g entries
1191          */
1192         pp->aprd = mem;
1193         pp->aprd_dma = mem_dma;
1194
1195         ap->private_data = pp;
1196
1197         /* clear any outstanding interrupt conditions */
1198         writew(0xffff, mmio + NV_ADMA_STAT);
1199
1200         /* initialize port variables */
1201         pp->flags = NV_ADMA_PORT_REGISTER_MODE;
1202
1203         /* clear CPB fetch count */
1204         writew(0, mmio + NV_ADMA_CPB_COUNT);
1205
1206         /* clear GO for register mode, enable interrupt */
1207         tmp = readw(mmio + NV_ADMA_CTL);
1208         writew((tmp & ~NV_ADMA_CTL_GO) | NV_ADMA_CTL_AIEN |
1209                 NV_ADMA_CTL_HOTPLUG_IEN, mmio + NV_ADMA_CTL);
1210
1211         tmp = readw(mmio + NV_ADMA_CTL);
1212         writew(tmp | NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1213         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1214         udelay(1);
1215         writew(tmp & ~NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1216         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1217
1218         return 0;
1219 }
1220
1221 static void nv_adma_port_stop(struct ata_port *ap)
1222 {
1223         struct nv_adma_port_priv *pp = ap->private_data;
1224         void __iomem *mmio = pp->ctl_block;
1225
1226         VPRINTK("ENTER\n");
1227         writew(0, mmio + NV_ADMA_CTL);
1228 }
1229
1230 #ifdef CONFIG_PM
1231 static int nv_adma_port_suspend(struct ata_port *ap, pm_message_t mesg)
1232 {
1233         struct nv_adma_port_priv *pp = ap->private_data;
1234         void __iomem *mmio = pp->ctl_block;
1235
1236         /* Go to register mode - clears GO */
1237         nv_adma_register_mode(ap);
1238
1239         /* clear CPB fetch count */
1240         writew(0, mmio + NV_ADMA_CPB_COUNT);
1241
1242         /* disable interrupt, shut down port */
1243         writew(0, mmio + NV_ADMA_CTL);
1244
1245         return 0;
1246 }
1247
1248 static int nv_adma_port_resume(struct ata_port *ap)
1249 {
1250         struct nv_adma_port_priv *pp = ap->private_data;
1251         void __iomem *mmio = pp->ctl_block;
1252         u16 tmp;
1253
1254         /* set CPB block location */
1255         writel(pp->cpb_dma & 0xFFFFFFFF,        mmio + NV_ADMA_CPB_BASE_LOW);
1256         writel((pp->cpb_dma >> 16) >> 16,       mmio + NV_ADMA_CPB_BASE_HIGH);
1257
1258         /* clear any outstanding interrupt conditions */
1259         writew(0xffff, mmio + NV_ADMA_STAT);
1260
1261         /* initialize port variables */
1262         pp->flags |= NV_ADMA_PORT_REGISTER_MODE;
1263
1264         /* clear CPB fetch count */
1265         writew(0, mmio + NV_ADMA_CPB_COUNT);
1266
1267         /* clear GO for register mode, enable interrupt */
1268         tmp = readw(mmio + NV_ADMA_CTL);
1269         writew((tmp & ~NV_ADMA_CTL_GO) | NV_ADMA_CTL_AIEN |
1270                 NV_ADMA_CTL_HOTPLUG_IEN, mmio + NV_ADMA_CTL);
1271
1272         tmp = readw(mmio + NV_ADMA_CTL);
1273         writew(tmp | NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1274         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1275         udelay(1);
1276         writew(tmp & ~NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1277         readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1278
1279         return 0;
1280 }
1281 #endif
1282
1283 static void nv_adma_setup_port(struct ata_port *ap)
1284 {
1285         void __iomem *mmio = ap->host->iomap[NV_MMIO_BAR];
1286         struct ata_ioports *ioport = &ap->ioaddr;
1287
1288         VPRINTK("ENTER\n");
1289
1290         mmio += NV_ADMA_PORT + ap->port_no * NV_ADMA_PORT_SIZE;
1291
1292         ioport->cmd_addr        = mmio;
1293         ioport->data_addr       = mmio + (ATA_REG_DATA * 4);
1294         ioport->error_addr      =
1295         ioport->feature_addr    = mmio + (ATA_REG_ERR * 4);
1296         ioport->nsect_addr      = mmio + (ATA_REG_NSECT * 4);
1297         ioport->lbal_addr       = mmio + (ATA_REG_LBAL * 4);
1298         ioport->lbam_addr       = mmio + (ATA_REG_LBAM * 4);
1299         ioport->lbah_addr       = mmio + (ATA_REG_LBAH * 4);
1300         ioport->device_addr     = mmio + (ATA_REG_DEVICE * 4);
1301         ioport->status_addr     =
1302         ioport->command_addr    = mmio + (ATA_REG_STATUS * 4);
1303         ioport->altstatus_addr  =
1304         ioport->ctl_addr        = mmio + 0x20;
1305 }
1306
1307 static int nv_adma_host_init(struct ata_host *host)
1308 {
1309         struct pci_dev *pdev = to_pci_dev(host->dev);
1310         unsigned int i;
1311         u32 tmp32;
1312
1313         VPRINTK("ENTER\n");
1314
1315         /* enable ADMA on the ports */
1316         pci_read_config_dword(pdev, NV_MCP_SATA_CFG_20, &tmp32);
1317         tmp32 |= NV_MCP_SATA_CFG_20_PORT0_EN |
1318                  NV_MCP_SATA_CFG_20_PORT0_PWB_EN |
1319                  NV_MCP_SATA_CFG_20_PORT1_EN |
1320                  NV_MCP_SATA_CFG_20_PORT1_PWB_EN;
1321
1322         pci_write_config_dword(pdev, NV_MCP_SATA_CFG_20, tmp32);
1323
1324         for (i = 0; i < host->n_ports; i++)
1325                 nv_adma_setup_port(host->ports[i]);
1326
1327         return 0;
1328 }
1329
1330 static void nv_adma_fill_aprd(struct ata_queued_cmd *qc,
1331                               struct scatterlist *sg,
1332                               int idx,
1333                               struct nv_adma_prd *aprd)
1334 {
1335         u8 flags = 0;
1336         if (qc->tf.flags & ATA_TFLAG_WRITE)
1337                 flags |= NV_APRD_WRITE;
1338         if (idx == qc->n_elem - 1)
1339                 flags |= NV_APRD_END;
1340         else if (idx != 4)
1341                 flags |= NV_APRD_CONT;
1342
1343         aprd->addr  = cpu_to_le64(((u64)sg_dma_address(sg)));
1344         aprd->len   = cpu_to_le32(((u32)sg_dma_len(sg))); /* len in bytes */
1345         aprd->flags = flags;
1346         aprd->packet_len = 0;
1347 }
1348
1349 static void nv_adma_fill_sg(struct ata_queued_cmd *qc, struct nv_adma_cpb *cpb)
1350 {
1351         struct nv_adma_port_priv *pp = qc->ap->private_data;
1352         struct nv_adma_prd *aprd;
1353         struct scatterlist *sg;
1354         unsigned int si;
1355
1356         VPRINTK("ENTER\n");
1357
1358         for_each_sg(qc->sg, sg, qc->n_elem, si) {
1359                 aprd = (si < 5) ? &cpb->aprd[si] :
1360                                &pp->aprd[NV_ADMA_SGTBL_LEN * qc->tag + (si-5)];
1361                 nv_adma_fill_aprd(qc, sg, si, aprd);
1362         }
1363         if (si > 5)
1364                 cpb->next_aprd = cpu_to_le64(((u64)(pp->aprd_dma + NV_ADMA_SGTBL_SZ * qc->tag)));
1365         else
1366                 cpb->next_aprd = cpu_to_le64(0);
1367 }
1368
1369 static int nv_adma_use_reg_mode(struct ata_queued_cmd *qc)
1370 {
1371         struct nv_adma_port_priv *pp = qc->ap->private_data;
1372
1373         /* ADMA engine can only be used for non-ATAPI DMA commands,
1374            or interrupt-driven no-data commands. */
1375         if ((pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) ||
1376            (qc->tf.flags & ATA_TFLAG_POLLING))
1377                 return 1;
1378
1379         if ((qc->flags & ATA_QCFLAG_DMAMAP) ||
1380            (qc->tf.protocol == ATA_PROT_NODATA))
1381                 return 0;
1382
1383         return 1;
1384 }
1385
1386 static void nv_adma_qc_prep(struct ata_queued_cmd *qc)
1387 {
1388         struct nv_adma_port_priv *pp = qc->ap->private_data;
1389         struct nv_adma_cpb *cpb = &pp->cpb[qc->tag];
1390         u8 ctl_flags = NV_CPB_CTL_CPB_VALID |
1391                        NV_CPB_CTL_IEN;
1392
1393         if (nv_adma_use_reg_mode(qc)) {
1394                 BUG_ON(!(pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) &&
1395                         (qc->flags & ATA_QCFLAG_DMAMAP));
1396                 nv_adma_register_mode(qc->ap);
1397                 ata_bmdma_qc_prep(qc);
1398                 return;
1399         }
1400
1401         cpb->resp_flags = NV_CPB_RESP_DONE;
1402         wmb();
1403         cpb->ctl_flags = 0;
1404         wmb();
1405
1406         cpb->len                = 3;
1407         cpb->tag                = qc->tag;
1408         cpb->next_cpb_idx       = 0;
1409
1410         /* turn on NCQ flags for NCQ commands */
1411         if (qc->tf.protocol == ATA_PROT_NCQ)
1412                 ctl_flags |= NV_CPB_CTL_QUEUE | NV_CPB_CTL_FPDMA;
1413
1414         VPRINTK("qc->flags = 0x%lx\n", qc->flags);
1415
1416         nv_adma_tf_to_cpb(&qc->tf, cpb->tf);
1417
1418         if (qc->flags & ATA_QCFLAG_DMAMAP) {
1419                 nv_adma_fill_sg(qc, cpb);
1420                 ctl_flags |= NV_CPB_CTL_APRD_VALID;
1421         } else
1422                 memset(&cpb->aprd[0], 0, sizeof(struct nv_adma_prd) * 5);
1423
1424         /* Be paranoid and don't let the device see NV_CPB_CTL_CPB_VALID
1425            until we are finished filling in all of the contents */
1426         wmb();
1427         cpb->ctl_flags = ctl_flags;
1428         wmb();
1429         cpb->resp_flags = 0;
1430 }
1431
1432 static unsigned int nv_adma_qc_issue(struct ata_queued_cmd *qc)
1433 {
1434         struct nv_adma_port_priv *pp = qc->ap->private_data;
1435         void __iomem *mmio = pp->ctl_block;
1436         int curr_ncq = (qc->tf.protocol == ATA_PROT_NCQ);
1437
1438         VPRINTK("ENTER\n");
1439
1440         /* We can't handle result taskfile with NCQ commands, since
1441            retrieving the taskfile switches us out of ADMA mode and would abort
1442            existing commands. */
1443         if (unlikely(qc->tf.protocol == ATA_PROT_NCQ &&
1444                      (qc->flags & ATA_QCFLAG_RESULT_TF))) {
1445                 ata_dev_err(qc->dev, "NCQ w/ RESULT_TF not allowed\n");
1446                 return AC_ERR_SYSTEM;
1447         }
1448
1449         if (nv_adma_use_reg_mode(qc)) {
1450                 /* use ATA register mode */
1451                 VPRINTK("using ATA register mode: 0x%lx\n", qc->flags);
1452                 BUG_ON(!(pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE) &&
1453                         (qc->flags & ATA_QCFLAG_DMAMAP));
1454                 nv_adma_register_mode(qc->ap);
1455                 return ata_bmdma_qc_issue(qc);
1456         } else
1457                 nv_adma_mode(qc->ap);
1458
1459         /* write append register, command tag in lower 8 bits
1460            and (number of cpbs to append -1) in top 8 bits */
1461         wmb();
1462
1463         if (curr_ncq != pp->last_issue_ncq) {
1464                 /* Seems to need some delay before switching between NCQ and
1465                    non-NCQ commands, else we get command timeouts and such. */
1466                 udelay(20);
1467                 pp->last_issue_ncq = curr_ncq;
1468         }
1469
1470         writew(qc->tag, mmio + NV_ADMA_APPEND);
1471
1472         DPRINTK("Issued tag %u\n", qc->tag);
1473
1474         return 0;
1475 }
1476
1477 static irqreturn_t nv_generic_interrupt(int irq, void *dev_instance)
1478 {
1479         struct ata_host *host = dev_instance;
1480         unsigned int i;
1481         unsigned int handled = 0;
1482         unsigned long flags;
1483
1484         spin_lock_irqsave(&host->lock, flags);
1485
1486         for (i = 0; i < host->n_ports; i++) {
1487                 struct ata_port *ap = host->ports[i];
1488                 struct ata_queued_cmd *qc;
1489
1490                 qc = ata_qc_from_tag(ap, ap->link.active_tag);
1491                 if (qc && (!(qc->tf.flags & ATA_TFLAG_POLLING))) {
1492                         handled += ata_bmdma_port_intr(ap, qc);
1493                 } else {
1494                         /*
1495                          * No request pending?  Clear interrupt status
1496                          * anyway, in case there's one pending.
1497                          */
1498                         ap->ops->sff_check_status(ap);
1499                 }
1500         }
1501
1502         spin_unlock_irqrestore(&host->lock, flags);
1503
1504         return IRQ_RETVAL(handled);
1505 }
1506
1507 static irqreturn_t nv_do_interrupt(struct ata_host *host, u8 irq_stat)
1508 {
1509         int i, handled = 0;
1510
1511         for (i = 0; i < host->n_ports; i++) {
1512                 handled += nv_host_intr(host->ports[i], irq_stat);
1513                 irq_stat >>= NV_INT_PORT_SHIFT;
1514         }
1515
1516         return IRQ_RETVAL(handled);
1517 }
1518
1519 static irqreturn_t nv_nf2_interrupt(int irq, void *dev_instance)
1520 {
1521         struct ata_host *host = dev_instance;
1522         u8 irq_stat;
1523         irqreturn_t ret;
1524
1525         spin_lock(&host->lock);
1526         irq_stat = ioread8(host->ports[0]->ioaddr.scr_addr + NV_INT_STATUS);
1527         ret = nv_do_interrupt(host, irq_stat);
1528         spin_unlock(&host->lock);
1529
1530         return ret;
1531 }
1532
1533 static irqreturn_t nv_ck804_interrupt(int irq, void *dev_instance)
1534 {
1535         struct ata_host *host = dev_instance;
1536         u8 irq_stat;
1537         irqreturn_t ret;
1538
1539         spin_lock(&host->lock);
1540         irq_stat = readb(host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_CK804);
1541         ret = nv_do_interrupt(host, irq_stat);
1542         spin_unlock(&host->lock);
1543
1544         return ret;
1545 }
1546
1547 static int nv_scr_read(struct ata_link *link, unsigned int sc_reg, u32 *val)
1548 {
1549         if (sc_reg > SCR_CONTROL)
1550                 return -EINVAL;
1551
1552         *val = ioread32(link->ap->ioaddr.scr_addr + (sc_reg * 4));
1553         return 0;
1554 }
1555
1556 static int nv_scr_write(struct ata_link *link, unsigned int sc_reg, u32 val)
1557 {
1558         if (sc_reg > SCR_CONTROL)
1559                 return -EINVAL;
1560
1561         iowrite32(val, link->ap->ioaddr.scr_addr + (sc_reg * 4));
1562         return 0;
1563 }
1564
1565 static int nv_hardreset(struct ata_link *link, unsigned int *class,
1566                         unsigned long deadline)
1567 {
1568         struct ata_eh_context *ehc = &link->eh_context;
1569
1570         /* Do hardreset iff it's post-boot probing, please read the
1571          * comment above port ops for details.
1572          */
1573         if (!(link->ap->pflags & ATA_PFLAG_LOADING) &&
1574             !ata_dev_enabled(link->device))
1575                 sata_link_hardreset(link, sata_deb_timing_hotplug, deadline,
1576                                     NULL, NULL);
1577         else {
1578                 const unsigned long *timing = sata_ehc_deb_timing(ehc);
1579                 int rc;
1580
1581                 if (!(ehc->i.flags & ATA_EHI_QUIET))
1582                         ata_link_info(link,
1583                                       "nv: skipping hardreset on occupied port\n");
1584
1585                 /* make sure the link is online */
1586                 rc = sata_link_resume(link, timing, deadline);
1587                 /* whine about phy resume failure but proceed */
1588                 if (rc && rc != -EOPNOTSUPP)
1589                         ata_link_warn(link, "failed to resume link (errno=%d)\n",
1590                                       rc);
1591         }
1592
1593         /* device signature acquisition is unreliable */
1594         return -EAGAIN;
1595 }
1596
1597 static void nv_nf2_freeze(struct ata_port *ap)
1598 {
1599         void __iomem *scr_addr = ap->host->ports[0]->ioaddr.scr_addr;
1600         int shift = ap->port_no * NV_INT_PORT_SHIFT;
1601         u8 mask;
1602
1603         mask = ioread8(scr_addr + NV_INT_ENABLE);
1604         mask &= ~(NV_INT_ALL << shift);
1605         iowrite8(mask, scr_addr + NV_INT_ENABLE);
1606 }
1607
1608 static void nv_nf2_thaw(struct ata_port *ap)
1609 {
1610         void __iomem *scr_addr = ap->host->ports[0]->ioaddr.scr_addr;
1611         int shift = ap->port_no * NV_INT_PORT_SHIFT;
1612         u8 mask;
1613
1614         iowrite8(NV_INT_ALL << shift, scr_addr + NV_INT_STATUS);
1615
1616         mask = ioread8(scr_addr + NV_INT_ENABLE);
1617         mask |= (NV_INT_MASK << shift);
1618         iowrite8(mask, scr_addr + NV_INT_ENABLE);
1619 }
1620
1621 static void nv_ck804_freeze(struct ata_port *ap)
1622 {
1623         void __iomem *mmio_base = ap->host->iomap[NV_MMIO_BAR];
1624         int shift = ap->port_no * NV_INT_PORT_SHIFT;
1625         u8 mask;
1626
1627         mask = readb(mmio_base + NV_INT_ENABLE_CK804);
1628         mask &= ~(NV_INT_ALL << shift);
1629         writeb(mask, mmio_base + NV_INT_ENABLE_CK804);
1630 }
1631
1632 static void nv_ck804_thaw(struct ata_port *ap)
1633 {
1634         void __iomem *mmio_base = ap->host->iomap[NV_MMIO_BAR];
1635         int shift = ap->port_no * NV_INT_PORT_SHIFT;
1636         u8 mask;
1637
1638         writeb(NV_INT_ALL << shift, mmio_base + NV_INT_STATUS_CK804);
1639
1640         mask = readb(mmio_base + NV_INT_ENABLE_CK804);
1641         mask |= (NV_INT_MASK << shift);
1642         writeb(mask, mmio_base + NV_INT_ENABLE_CK804);
1643 }
1644
1645 static void nv_mcp55_freeze(struct ata_port *ap)
1646 {
1647         void __iomem *mmio_base = ap->host->iomap[NV_MMIO_BAR];
1648         int shift = ap->port_no * NV_INT_PORT_SHIFT_MCP55;
1649         u32 mask;
1650
1651         writel(NV_INT_ALL_MCP55 << shift, mmio_base + NV_INT_STATUS_MCP55);
1652
1653         mask = readl(mmio_base + NV_INT_ENABLE_MCP55);
1654         mask &= ~(NV_INT_ALL_MCP55 << shift);
1655         writel(mask, mmio_base + NV_INT_ENABLE_MCP55);
1656 }
1657
1658 static void nv_mcp55_thaw(struct ata_port *ap)
1659 {
1660         void __iomem *mmio_base = ap->host->iomap[NV_MMIO_BAR];
1661         int shift = ap->port_no * NV_INT_PORT_SHIFT_MCP55;
1662         u32 mask;
1663
1664         writel(NV_INT_ALL_MCP55 << shift, mmio_base + NV_INT_STATUS_MCP55);
1665
1666         mask = readl(mmio_base + NV_INT_ENABLE_MCP55);
1667         mask |= (NV_INT_MASK_MCP55 << shift);
1668         writel(mask, mmio_base + NV_INT_ENABLE_MCP55);
1669 }
1670
1671 static void nv_adma_error_handler(struct ata_port *ap)
1672 {
1673         struct nv_adma_port_priv *pp = ap->private_data;
1674         if (!(pp->flags & NV_ADMA_PORT_REGISTER_MODE)) {
1675                 void __iomem *mmio = pp->ctl_block;
1676                 int i;
1677                 u16 tmp;
1678
1679                 if (ata_tag_valid(ap->link.active_tag) || ap->link.sactive) {
1680                         u32 notifier = readl(mmio + NV_ADMA_NOTIFIER);
1681                         u32 notifier_error = readl(mmio + NV_ADMA_NOTIFIER_ERROR);
1682                         u32 gen_ctl = readl(pp->gen_block + NV_ADMA_GEN_CTL);
1683                         u32 status = readw(mmio + NV_ADMA_STAT);
1684                         u8 cpb_count = readb(mmio + NV_ADMA_CPB_COUNT);
1685                         u8 next_cpb_idx = readb(mmio + NV_ADMA_NEXT_CPB_IDX);
1686
1687                         ata_port_err(ap,
1688                                 "EH in ADMA mode, notifier 0x%X "
1689                                 "notifier_error 0x%X gen_ctl 0x%X status 0x%X "
1690                                 "next cpb count 0x%X next cpb idx 0x%x\n",
1691                                 notifier, notifier_error, gen_ctl, status,
1692                                 cpb_count, next_cpb_idx);
1693
1694                         for (i = 0; i < NV_ADMA_MAX_CPBS; i++) {
1695                                 struct nv_adma_cpb *cpb = &pp->cpb[i];
1696                                 if ((ata_tag_valid(ap->link.active_tag) && i == ap->link.active_tag) ||
1697                                     ap->link.sactive & (1 << i))
1698                                         ata_port_err(ap,
1699                                                 "CPB %d: ctl_flags 0x%x, resp_flags 0x%x\n",
1700                                                 i, cpb->ctl_flags, cpb->resp_flags);
1701                         }
1702                 }
1703
1704                 /* Push us back into port register mode for error handling. */
1705                 nv_adma_register_mode(ap);
1706
1707                 /* Mark all of the CPBs as invalid to prevent them from
1708                    being executed */
1709                 for (i = 0; i < NV_ADMA_MAX_CPBS; i++)
1710                         pp->cpb[i].ctl_flags &= ~NV_CPB_CTL_CPB_VALID;
1711
1712                 /* clear CPB fetch count */
1713                 writew(0, mmio + NV_ADMA_CPB_COUNT);
1714
1715                 /* Reset channel */
1716                 tmp = readw(mmio + NV_ADMA_CTL);
1717                 writew(tmp | NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1718                 readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1719                 udelay(1);
1720                 writew(tmp & ~NV_ADMA_CTL_CHANNEL_RESET, mmio + NV_ADMA_CTL);
1721                 readw(mmio + NV_ADMA_CTL);      /* flush posted write */
1722         }
1723
1724         ata_bmdma_error_handler(ap);
1725 }
1726
1727 static void nv_swncq_qc_to_dq(struct ata_port *ap, struct ata_queued_cmd *qc)
1728 {
1729         struct nv_swncq_port_priv *pp = ap->private_data;
1730         struct defer_queue *dq = &pp->defer_queue;
1731
1732         /* queue is full */
1733         WARN_ON(dq->tail - dq->head == ATA_MAX_QUEUE);
1734         dq->defer_bits |= (1 << qc->tag);
1735         dq->tag[dq->tail++ & (ATA_MAX_QUEUE - 1)] = qc->tag;
1736 }
1737
1738 static struct ata_queued_cmd *nv_swncq_qc_from_dq(struct ata_port *ap)
1739 {
1740         struct nv_swncq_port_priv *pp = ap->private_data;
1741         struct defer_queue *dq = &pp->defer_queue;
1742         unsigned int tag;
1743
1744         if (dq->head == dq->tail)       /* null queue */
1745                 return NULL;
1746
1747         tag = dq->tag[dq->head & (ATA_MAX_QUEUE - 1)];
1748         dq->tag[dq->head++ & (ATA_MAX_QUEUE - 1)] = ATA_TAG_POISON;
1749         WARN_ON(!(dq->defer_bits & (1 << tag)));
1750         dq->defer_bits &= ~(1 << tag);
1751
1752         return ata_qc_from_tag(ap, tag);
1753 }
1754
1755 static void nv_swncq_fis_reinit(struct ata_port *ap)
1756 {
1757         struct nv_swncq_port_priv *pp = ap->private_data;
1758
1759         pp->dhfis_bits = 0;
1760         pp->dmafis_bits = 0;
1761         pp->sdbfis_bits = 0;
1762         pp->ncq_flags = 0;
1763 }
1764
1765 static void nv_swncq_pp_reinit(struct ata_port *ap)
1766 {
1767         struct nv_swncq_port_priv *pp = ap->private_data;
1768         struct defer_queue *dq = &pp->defer_queue;
1769
1770         dq->head = 0;
1771         dq->tail = 0;
1772         dq->defer_bits = 0;
1773         pp->qc_active = 0;
1774         pp->last_issue_tag = ATA_TAG_POISON;
1775         nv_swncq_fis_reinit(ap);
1776 }
1777
1778 static void nv_swncq_irq_clear(struct ata_port *ap, u16 fis)
1779 {
1780         struct nv_swncq_port_priv *pp = ap->private_data;
1781
1782         writew(fis, pp->irq_block);
1783 }
1784
1785 static void __ata_bmdma_stop(struct ata_port *ap)
1786 {
1787         struct ata_queued_cmd qc;
1788
1789         qc.ap = ap;
1790         ata_bmdma_stop(&qc);
1791 }
1792
1793 static void nv_swncq_ncq_stop(struct ata_port *ap)
1794 {
1795         struct nv_swncq_port_priv *pp = ap->private_data;
1796         unsigned int i;
1797         u32 sactive;
1798         u32 done_mask;
1799
1800         ata_port_err(ap, "EH in SWNCQ mode,QC:qc_active 0x%X sactive 0x%X\n",
1801                      ap->qc_active, ap->link.sactive);
1802         ata_port_err(ap,
1803                 "SWNCQ:qc_active 0x%X defer_bits 0x%X last_issue_tag 0x%x\n  "
1804                 "dhfis 0x%X dmafis 0x%X sdbfis 0x%X\n",
1805                 pp->qc_active, pp->defer_queue.defer_bits, pp->last_issue_tag,
1806                 pp->dhfis_bits, pp->dmafis_bits, pp->sdbfis_bits);
1807
1808         ata_port_err(ap, "ATA_REG 0x%X ERR_REG 0x%X\n",
1809                      ap->ops->sff_check_status(ap),
1810                      ioread8(ap->ioaddr.error_addr));
1811
1812         sactive = readl(pp->sactive_block);
1813         done_mask = pp->qc_active ^ sactive;
1814
1815         ata_port_err(ap, "tag : dhfis dmafis sdbfis sactive\n");
1816         for (i = 0; i < ATA_MAX_QUEUE; i++) {
1817                 u8 err = 0;
1818                 if (pp->qc_active & (1 << i))
1819                         err = 0;
1820                 else if (done_mask & (1 << i))
1821                         err = 1;
1822                 else
1823                         continue;
1824
1825                 ata_port_err(ap,
1826                              "tag 0x%x: %01x %01x %01x %01x %s\n", i,
1827                              (pp->dhfis_bits >> i) & 0x1,
1828                              (pp->dmafis_bits >> i) & 0x1,
1829                              (pp->sdbfis_bits >> i) & 0x1,
1830                              (sactive >> i) & 0x1,
1831                              (err ? "error! tag doesn't exit" : " "));
1832         }
1833
1834         nv_swncq_pp_reinit(ap);
1835         ap->ops->sff_irq_clear(ap);
1836         __ata_bmdma_stop(ap);
1837         nv_swncq_irq_clear(ap, 0xffff);
1838 }
1839
1840 static void nv_swncq_error_handler(struct ata_port *ap)
1841 {
1842         struct ata_eh_context *ehc = &ap->link.eh_context;
1843
1844         if (ap->link.sactive) {
1845                 nv_swncq_ncq_stop(ap);
1846                 ehc->i.action |= ATA_EH_RESET;
1847         }
1848
1849         ata_bmdma_error_handler(ap);
1850 }
1851
1852 #ifdef CONFIG_PM
1853 static int nv_swncq_port_suspend(struct ata_port *ap, pm_message_t mesg)
1854 {
1855         void __iomem *mmio = ap->host->iomap[NV_MMIO_BAR];
1856         u32 tmp;
1857
1858         /* clear irq */
1859         writel(~0, mmio + NV_INT_STATUS_MCP55);
1860
1861         /* disable irq */
1862         writel(0, mmio + NV_INT_ENABLE_MCP55);
1863
1864         /* disable swncq */
1865         tmp = readl(mmio + NV_CTL_MCP55);
1866         tmp &= ~(NV_CTL_PRI_SWNCQ | NV_CTL_SEC_SWNCQ);
1867         writel(tmp, mmio + NV_CTL_MCP55);
1868
1869         return 0;
1870 }
1871
1872 static int nv_swncq_port_resume(struct ata_port *ap)
1873 {
1874         void __iomem *mmio = ap->host->iomap[NV_MMIO_BAR];
1875         u32 tmp;
1876
1877         /* clear irq */
1878         writel(~0, mmio + NV_INT_STATUS_MCP55);
1879
1880         /* enable irq */
1881         writel(0x00fd00fd, mmio + NV_INT_ENABLE_MCP55);
1882
1883         /* enable swncq */
1884         tmp = readl(mmio + NV_CTL_MCP55);
1885         writel(tmp | NV_CTL_PRI_SWNCQ | NV_CTL_SEC_SWNCQ, mmio + NV_CTL_MCP55);
1886
1887         return 0;
1888 }
1889 #endif
1890
1891 static void nv_swncq_host_init(struct ata_host *host)
1892 {
1893         u32 tmp;
1894         void __iomem *mmio = host->iomap[NV_MMIO_BAR];
1895         struct pci_dev *pdev = to_pci_dev(host->dev);
1896         u8 regval;
1897
1898         /* disable  ECO 398 */
1899         pci_read_config_byte(pdev, 0x7f, &regval);
1900         regval &= ~(1 << 7);
1901         pci_write_config_byte(pdev, 0x7f, regval);
1902
1903         /* enable swncq */
1904         tmp = readl(mmio + NV_CTL_MCP55);
1905         VPRINTK("HOST_CTL:0x%X\n", tmp);
1906         writel(tmp | NV_CTL_PRI_SWNCQ | NV_CTL_SEC_SWNCQ, mmio + NV_CTL_MCP55);
1907
1908         /* enable irq intr */
1909         tmp = readl(mmio + NV_INT_ENABLE_MCP55);
1910         VPRINTK("HOST_ENABLE:0x%X\n", tmp);
1911         writel(tmp | 0x00fd00fd, mmio + NV_INT_ENABLE_MCP55);
1912
1913         /*  clear port irq */
1914         writel(~0x0, mmio + NV_INT_STATUS_MCP55);
1915 }
1916
1917 static int nv_swncq_slave_config(struct scsi_device *sdev)
1918 {
1919         struct ata_port *ap = ata_shost_to_port(sdev->host);
1920         struct pci_dev *pdev = to_pci_dev(ap->host->dev);
1921         struct ata_device *dev;
1922         int rc;
1923         u8 rev;
1924         u8 check_maxtor = 0;
1925         unsigned char model_num[ATA_ID_PROD_LEN + 1];
1926
1927         rc = ata_scsi_slave_config(sdev);
1928         if (sdev->id >= ATA_MAX_DEVICES || sdev->channel || sdev->lun)
1929                 /* Not a proper libata device, ignore */
1930                 return rc;
1931
1932         dev = &ap->link.device[sdev->id];
1933         if (!(ap->flags & ATA_FLAG_NCQ) || dev->class == ATA_DEV_ATAPI)
1934                 return rc;
1935
1936         /* if MCP51 and Maxtor, then disable ncq */
1937         if (pdev->device == PCI_DEVICE_ID_NVIDIA_NFORCE_MCP51_SATA ||
1938                 pdev->device == PCI_DEVICE_ID_NVIDIA_NFORCE_MCP51_SATA2)
1939                 check_maxtor = 1;
1940
1941         /* if MCP55 and rev <= a2 and Maxtor, then disable ncq */
1942         if (pdev->device == PCI_DEVICE_ID_NVIDIA_NFORCE_MCP55_SATA ||
1943                 pdev->device == PCI_DEVICE_ID_NVIDIA_NFORCE_MCP55_SATA2) {
1944                 pci_read_config_byte(pdev, 0x8, &rev);
1945                 if (rev <= 0xa2)
1946                         check_maxtor = 1;
1947         }
1948
1949         if (!check_maxtor)
1950                 return rc;
1951
1952         ata_id_c_string(dev->id, model_num, ATA_ID_PROD, sizeof(model_num));
1953
1954         if (strncmp(model_num, "Maxtor", 6) == 0) {
1955                 ata_scsi_change_queue_depth(sdev, 1, SCSI_QDEPTH_DEFAULT);
1956                 ata_dev_notice(dev, "Disabling SWNCQ mode (depth %x)\n",
1957                                sdev->queue_depth);
1958         }
1959
1960         return rc;
1961 }
1962
1963 static int nv_swncq_port_start(struct ata_port *ap)
1964 {
1965         struct device *dev = ap->host->dev;
1966         void __iomem *mmio = ap->host->iomap[NV_MMIO_BAR];
1967         struct nv_swncq_port_priv *pp;
1968         int rc;
1969
1970         /* we might fallback to bmdma, allocate bmdma resources */
1971         rc = ata_bmdma_port_start(ap);
1972         if (rc)
1973                 return rc;
1974
1975         pp = devm_kzalloc(dev, sizeof(*pp), GFP_KERNEL);
1976         if (!pp)
1977                 return -ENOMEM;
1978
1979         pp->prd = dmam_alloc_coherent(dev, ATA_PRD_TBL_SZ * ATA_MAX_QUEUE,
1980                                       &pp->prd_dma, GFP_KERNEL);
1981         if (!pp->prd)
1982                 return -ENOMEM;
1983         memset(pp->prd, 0, ATA_PRD_TBL_SZ * ATA_MAX_QUEUE);
1984
1985         ap->private_data = pp;
1986         pp->sactive_block = ap->ioaddr.scr_addr + 4 * SCR_ACTIVE;
1987         pp->irq_block = mmio + NV_INT_STATUS_MCP55 + ap->port_no * 2;
1988         pp->tag_block = mmio + NV_NCQ_REG_MCP55 + ap->port_no * 2;
1989
1990         return 0;
1991 }
1992
1993 static void nv_swncq_qc_prep(struct ata_queued_cmd *qc)
1994 {
1995         if (qc->tf.protocol != ATA_PROT_NCQ) {
1996                 ata_bmdma_qc_prep(qc);
1997                 return;
1998         }
1999
2000         if (!(qc->flags & ATA_QCFLAG_DMAMAP))
2001                 return;
2002
2003         nv_swncq_fill_sg(qc);
2004 }
2005
2006 static void nv_swncq_fill_sg(struct ata_queued_cmd *qc)
2007 {
2008         struct ata_port *ap = qc->ap;
2009         struct scatterlist *sg;
2010         struct nv_swncq_port_priv *pp = ap->private_data;
2011         struct ata_bmdma_prd *prd;
2012         unsigned int si, idx;
2013
2014         prd = pp->prd + ATA_MAX_PRD * qc->tag;
2015
2016         idx = 0;
2017         for_each_sg(qc->sg, sg, qc->n_elem, si) {
2018                 u32 addr, offset;
2019                 u32 sg_len, len;
2020
2021                 addr = (u32)sg_dma_address(sg);
2022                 sg_len = sg_dma_len(sg);
2023
2024                 while (sg_len) {
2025                         offset = addr & 0xffff;
2026                         len = sg_len;
2027                         if ((offset + sg_len) > 0x10000)
2028                                 len = 0x10000 - offset;
2029
2030                         prd[idx].addr = cpu_to_le32(addr);
2031                         prd[idx].flags_len = cpu_to_le32(len & 0xffff);
2032
2033                         idx++;
2034                         sg_len -= len;
2035                         addr += len;
2036                 }
2037         }
2038
2039         prd[idx - 1].flags_len |= cpu_to_le32(ATA_PRD_EOT);
2040 }
2041
2042 static unsigned int nv_swncq_issue_atacmd(struct ata_port *ap,
2043                                           struct ata_queued_cmd *qc)
2044 {
2045         struct nv_swncq_port_priv *pp = ap->private_data;
2046
2047         if (qc == NULL)
2048                 return 0;
2049
2050         DPRINTK("Enter\n");
2051
2052         writel((1 << qc->tag), pp->sactive_block);
2053         pp->last_issue_tag = qc->tag;
2054         pp->dhfis_bits &= ~(1 << qc->tag);
2055         pp->dmafis_bits &= ~(1 << qc->tag);
2056         pp->qc_active |= (0x1 << qc->tag);
2057
2058         ap->ops->sff_tf_load(ap, &qc->tf);       /* load tf registers */
2059         ap->ops->sff_exec_command(ap, &qc->tf);
2060
2061         DPRINTK("Issued tag %u\n", qc->tag);
2062
2063         return 0;
2064 }
2065
2066 static unsigned int nv_swncq_qc_issue(struct ata_queued_cmd *qc)
2067 {
2068         struct ata_port *ap = qc->ap;
2069         struct nv_swncq_port_priv *pp = ap->private_data;
2070
2071         if (qc->tf.protocol != ATA_PROT_NCQ)
2072                 return ata_bmdma_qc_issue(qc);
2073
2074         DPRINTK("Enter\n");
2075
2076         if (!pp->qc_active)
2077                 nv_swncq_issue_atacmd(ap, qc);
2078         else
2079                 nv_swncq_qc_to_dq(ap, qc);      /* add qc to defer queue */
2080
2081         return 0;
2082 }
2083
2084 static void nv_swncq_hotplug(struct ata_port *ap, u32 fis)
2085 {
2086         u32 serror;
2087         struct ata_eh_info *ehi = &ap->link.eh_info;
2088
2089         ata_ehi_clear_desc(ehi);
2090
2091         /* AHCI needs SError cleared; otherwise, it might lock up */
2092         sata_scr_read(&ap->link, SCR_ERROR, &serror);
2093         sata_scr_write(&ap->link, SCR_ERROR, serror);
2094
2095         /* analyze @irq_stat */
2096         if (fis & NV_SWNCQ_IRQ_ADDED)
2097                 ata_ehi_push_desc(ehi, "hot plug");
2098         else if (fis & NV_SWNCQ_IRQ_REMOVED)
2099                 ata_ehi_push_desc(ehi, "hot unplug");
2100
2101         ata_ehi_hotplugged(ehi);
2102
2103         /* okay, let's hand over to EH */
2104         ehi->serror |= serror;
2105
2106         ata_port_freeze(ap);
2107 }
2108
2109 static int nv_swncq_sdbfis(struct ata_port *ap)
2110 {
2111         struct ata_queued_cmd *qc;
2112         struct nv_swncq_port_priv *pp = ap->private_data;
2113         struct ata_eh_info *ehi = &ap->link.eh_info;
2114         u32 sactive;
2115         u32 done_mask;
2116         u8 host_stat;
2117         u8 lack_dhfis = 0;
2118
2119         host_stat = ap->ops->bmdma_status(ap);
2120         if (unlikely(host_stat & ATA_DMA_ERR)) {
2121                 /* error when transferring data to/from memory */
2122                 ata_ehi_clear_desc(ehi);
2123                 ata_ehi_push_desc(ehi, "BMDMA stat 0x%x", host_stat);
2124                 ehi->err_mask |= AC_ERR_HOST_BUS;
2125                 ehi->action |= ATA_EH_RESET;
2126                 return -EINVAL;
2127         }
2128
2129         ap->ops->sff_irq_clear(ap);
2130         __ata_bmdma_stop(ap);
2131
2132         sactive = readl(pp->sactive_block);
2133         done_mask = pp->qc_active ^ sactive;
2134
2135         pp->qc_active &= ~done_mask;
2136         pp->dhfis_bits &= ~done_mask;
2137         pp->dmafis_bits &= ~done_mask;
2138         pp->sdbfis_bits |= done_mask;
2139         ata_qc_complete_multiple(ap, ap->qc_active ^ done_mask);
2140
2141         if (!ap->qc_active) {
2142                 DPRINTK("over\n");
2143                 nv_swncq_pp_reinit(ap);
2144                 return 0;
2145         }
2146
2147         if (pp->qc_active & pp->dhfis_bits)
2148                 return 0;
2149
2150         if ((pp->ncq_flags & ncq_saw_backout) ||
2151             (pp->qc_active ^ pp->dhfis_bits))
2152                 /* if the controller can't get a device to host register FIS,
2153                  * The driver needs to reissue the new command.
2154                  */
2155                 lack_dhfis = 1;
2156
2157         DPRINTK("id 0x%x QC: qc_active 0x%x,"
2158                 "SWNCQ:qc_active 0x%X defer_bits %X "
2159                 "dhfis 0x%X dmafis 0x%X last_issue_tag %x\n",
2160                 ap->print_id, ap->qc_active, pp->qc_active,
2161                 pp->defer_queue.defer_bits, pp->dhfis_bits,
2162                 pp->dmafis_bits, pp->last_issue_tag);
2163
2164         nv_swncq_fis_reinit(ap);
2165
2166         if (lack_dhfis) {
2167                 qc = ata_qc_from_tag(ap, pp->last_issue_tag);
2168                 nv_swncq_issue_atacmd(ap, qc);
2169                 return 0;
2170         }
2171
2172         if (pp->defer_queue.defer_bits) {
2173                 /* send deferral queue command */
2174                 qc = nv_swncq_qc_from_dq(ap);
2175                 WARN_ON(qc == NULL);
2176                 nv_swncq_issue_atacmd(ap, qc);
2177         }
2178
2179         return 0;
2180 }
2181
2182 static inline u32 nv_swncq_tag(struct ata_port *ap)
2183 {
2184         struct nv_swncq_port_priv *pp = ap->private_data;
2185         u32 tag;
2186
2187         tag = readb(pp->tag_block) >> 2;
2188         return (tag & 0x1f);
2189 }
2190
2191 static void nv_swncq_dmafis(struct ata_port *ap)
2192 {
2193         struct ata_queued_cmd *qc;
2194         unsigned int rw;
2195         u8 dmactl;
2196         u32 tag;
2197         struct nv_swncq_port_priv *pp = ap->private_data;
2198
2199         __ata_bmdma_stop(ap);
2200         tag = nv_swncq_tag(ap);
2201
2202         DPRINTK("dma setup tag 0x%x\n", tag);
2203         qc = ata_qc_from_tag(ap, tag);
2204
2205         if (unlikely(!qc))
2206                 return;
2207
2208         rw = qc->tf.flags & ATA_TFLAG_WRITE;
2209
2210         /* load PRD table addr. */
2211         iowrite32(pp->prd_dma + ATA_PRD_TBL_SZ * qc->tag,
2212                   ap->ioaddr.bmdma_addr + ATA_DMA_TABLE_OFS);
2213
2214         /* specify data direction, triple-check start bit is clear */
2215         dmactl = ioread8(ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
2216         dmactl &= ~ATA_DMA_WR;
2217         if (!rw)
2218                 dmactl |= ATA_DMA_WR;
2219
2220         iowrite8(dmactl | ATA_DMA_START, ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
2221 }
2222
2223 static void nv_swncq_host_interrupt(struct ata_port *ap, u16 fis)
2224 {
2225         struct nv_swncq_port_priv *pp = ap->private_data;
2226         struct ata_queued_cmd *qc;
2227         struct ata_eh_info *ehi = &ap->link.eh_info;
2228         u32 serror;
2229         u8 ata_stat;
2230
2231         ata_stat = ap->ops->sff_check_status(ap);
2232         nv_swncq_irq_clear(ap, fis);
2233         if (!fis)
2234                 return;
2235
2236         if (ap->pflags & ATA_PFLAG_FROZEN)
2237                 return;
2238
2239         if (fis & NV_SWNCQ_IRQ_HOTPLUG) {
2240                 nv_swncq_hotplug(ap, fis);
2241                 return;
2242         }
2243
2244         if (!pp->qc_active)
2245                 return;
2246
2247         if (ap->ops->scr_read(&ap->link, SCR_ERROR, &serror))
2248                 return;
2249         ap->ops->scr_write(&ap->link, SCR_ERROR, serror);
2250
2251         if (ata_stat & ATA_ERR) {
2252                 ata_ehi_clear_desc(ehi);
2253                 ata_ehi_push_desc(ehi, "Ata error. fis:0x%X", fis);
2254                 ehi->err_mask |= AC_ERR_DEV;
2255                 ehi->serror |= serror;
2256                 ehi->action |= ATA_EH_RESET;
2257                 ata_port_freeze(ap);
2258                 return;
2259         }
2260
2261         if (fis & NV_SWNCQ_IRQ_BACKOUT) {
2262                 /* If the IRQ is backout, driver must issue
2263                  * the new command again some time later.
2264                  */
2265                 pp->ncq_flags |= ncq_saw_backout;
2266         }
2267
2268         if (fis & NV_SWNCQ_IRQ_SDBFIS) {
2269                 pp->ncq_flags |= ncq_saw_sdb;
2270                 DPRINTK("id 0x%x SWNCQ: qc_active 0x%X "
2271                         "dhfis 0x%X dmafis 0x%X sactive 0x%X\n",
2272                         ap->print_id, pp->qc_active, pp->dhfis_bits,
2273                         pp->dmafis_bits, readl(pp->sactive_block));
2274                 if (nv_swncq_sdbfis(ap) < 0)
2275                         goto irq_error;
2276         }
2277
2278         if (fis & NV_SWNCQ_IRQ_DHREGFIS) {
2279                 /* The interrupt indicates the new command
2280                  * was transmitted correctly to the drive.
2281                  */
2282                 pp->dhfis_bits |= (0x1 << pp->last_issue_tag);
2283                 pp->ncq_flags |= ncq_saw_d2h;
2284                 if (pp->ncq_flags & (ncq_saw_sdb | ncq_saw_backout)) {
2285                         ata_ehi_push_desc(ehi, "illegal fis transaction");
2286                         ehi->err_mask |= AC_ERR_HSM;
2287                         ehi->action |= ATA_EH_RESET;
2288                         goto irq_error;
2289                 }
2290
2291                 if (!(fis & NV_SWNCQ_IRQ_DMASETUP) &&
2292                     !(pp->ncq_flags & ncq_saw_dmas)) {
2293                         ata_stat = ap->ops->sff_check_status(ap);
2294                         if (ata_stat & ATA_BUSY)
2295                                 goto irq_exit;
2296
2297                         if (pp->defer_queue.defer_bits) {
2298                                 DPRINTK("send next command\n");
2299                                 qc = nv_swncq_qc_from_dq(ap);
2300                                 nv_swncq_issue_atacmd(ap, qc);
2301                         }
2302                 }
2303         }
2304
2305         if (fis & NV_SWNCQ_IRQ_DMASETUP) {
2306                 /* program the dma controller with appropriate PRD buffers
2307                  * and start the DMA transfer for requested command.
2308                  */
2309                 pp->dmafis_bits |= (0x1 << nv_swncq_tag(ap));
2310                 pp->ncq_flags |= ncq_saw_dmas;
2311                 nv_swncq_dmafis(ap);
2312         }
2313
2314 irq_exit:
2315         return;
2316 irq_error:
2317         ata_ehi_push_desc(ehi, "fis:0x%x", fis);
2318         ata_port_freeze(ap);
2319         return;
2320 }
2321
2322 static irqreturn_t nv_swncq_interrupt(int irq, void *dev_instance)
2323 {
2324         struct ata_host *host = dev_instance;
2325         unsigned int i;
2326         unsigned int handled = 0;
2327         unsigned long flags;
2328         u32 irq_stat;
2329
2330         spin_lock_irqsave(&host->lock, flags);
2331
2332         irq_stat = readl(host->iomap[NV_MMIO_BAR] + NV_INT_STATUS_MCP55);
2333
2334         for (i = 0; i < host->n_ports; i++) {
2335                 struct ata_port *ap = host->ports[i];
2336
2337                 if (ap->link.sactive) {
2338                         nv_swncq_host_interrupt(ap, (u16)irq_stat);
2339                         handled = 1;
2340                 } else {
2341                         if (irq_stat)   /* reserve Hotplug */
2342                                 nv_swncq_irq_clear(ap, 0xfff0);
2343
2344                         handled += nv_host_intr(ap, (u8)irq_stat);
2345                 }
2346                 irq_stat >>= NV_INT_PORT_SHIFT_MCP55;
2347         }
2348
2349         spin_unlock_irqrestore(&host->lock, flags);
2350
2351         return IRQ_RETVAL(handled);
2352 }
2353
2354 static int nv_init_one(struct pci_dev *pdev, const struct pci_device_id *ent)
2355 {
2356         const struct ata_port_info *ppi[] = { NULL, NULL };
2357         struct nv_pi_priv *ipriv;
2358         struct ata_host *host;
2359         struct nv_host_priv *hpriv;
2360         int rc;
2361         u32 bar;
2362         void __iomem *base;
2363         unsigned long type = ent->driver_data;
2364
2365         // Make sure this is a SATA controller by counting the number of bars
2366         // (NVIDIA SATA controllers will always have six bars).  Otherwise,
2367         // it's an IDE controller and we ignore it.
2368         for (bar = 0; bar < 6; bar++)
2369                 if (pci_resource_start(pdev, bar) == 0)
2370                         return -ENODEV;
2371
2372         ata_print_version_once(&pdev->dev, DRV_VERSION);
2373
2374         rc = pcim_enable_device(pdev);
2375         if (rc)
2376                 return rc;
2377
2378         /* determine type and allocate host */
2379         if (type == CK804 && adma_enabled) {
2380                 dev_notice(&pdev->dev, "Using ADMA mode\n");
2381                 type = ADMA;
2382         } else if (type == MCP5x && swncq_enabled) {
2383                 dev_notice(&pdev->dev, "Using SWNCQ mode\n");
2384                 type = SWNCQ;
2385         }
2386
2387         ppi[0] = &nv_port_info[type];
2388         ipriv = ppi[0]->private_data;
2389         rc = ata_pci_bmdma_prepare_host(pdev, ppi, &host);
2390         if (rc)
2391                 return rc;
2392
2393         hpriv = devm_kzalloc(&pdev->dev, sizeof(*hpriv), GFP_KERNEL);
2394         if (!hpriv)
2395                 return -ENOMEM;
2396         hpriv->type = type;
2397         host->private_data = hpriv;
2398
2399         /* request and iomap NV_MMIO_BAR */
2400         rc = pcim_iomap_regions(pdev, 1 << NV_MMIO_BAR, DRV_NAME);
2401         if (rc)
2402                 return rc;
2403
2404         /* configure SCR access */
2405         base = host->iomap[NV_MMIO_BAR];
2406         host->ports[0]->ioaddr.scr_addr = base + NV_PORT0_SCR_REG_OFFSET;
2407         host->ports[1]->ioaddr.scr_addr = base + NV_PORT1_SCR_REG_OFFSET;
2408
2409         /* enable SATA space for CK804 */
2410         if (type >= CK804) {
2411                 u8 regval;
2412
2413                 pci_read_config_byte(pdev, NV_MCP_SATA_CFG_20, &regval);
2414                 regval |= NV_MCP_SATA_CFG_20_SATA_SPACE_EN;
2415                 pci_write_config_byte(pdev, NV_MCP_SATA_CFG_20, regval);
2416         }
2417
2418         /* init ADMA */
2419         if (type == ADMA) {
2420                 rc = nv_adma_host_init(host);
2421                 if (rc)
2422                         return rc;
2423         } else if (type == SWNCQ)
2424                 nv_swncq_host_init(host);
2425
2426         if (msi_enabled) {
2427                 dev_notice(&pdev->dev, "Using MSI\n");
2428                 pci_enable_msi(pdev);
2429         }
2430
2431         pci_set_master(pdev);
2432         return ata_pci_sff_activate_host(host, ipriv->irq_handler, ipriv->sht);
2433 }
2434
2435 #ifdef CONFIG_PM
2436 static int nv_pci_device_resume(struct pci_dev *pdev)
2437 {
2438         struct ata_host *host = dev_get_drvdata(&pdev->dev);
2439         struct nv_host_priv *hpriv = host->private_data;
2440         int rc;
2441
2442         rc = ata_pci_device_do_resume(pdev);
2443         if (rc)
2444                 return rc;
2445
2446         if (pdev->dev.power.power_state.event == PM_EVENT_SUSPEND) {
2447                 if (hpriv->type >= CK804) {
2448                         u8 regval;
2449
2450                         pci_read_config_byte(pdev, NV_MCP_SATA_CFG_20, &regval);
2451                         regval |= NV_MCP_SATA_CFG_20_SATA_SPACE_EN;
2452                         pci_write_config_byte(pdev, NV_MCP_SATA_CFG_20, regval);
2453                 }
2454                 if (hpriv->type == ADMA) {
2455                         u32 tmp32;
2456                         struct nv_adma_port_priv *pp;
2457                         /* enable/disable ADMA on the ports appropriately */
2458                         pci_read_config_dword(pdev, NV_MCP_SATA_CFG_20, &tmp32);
2459
2460                         pp = host->ports[0]->private_data;
2461                         if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE)
2462                                 tmp32 &= ~(NV_MCP_SATA_CFG_20_PORT0_EN |
2463                                            NV_MCP_SATA_CFG_20_PORT0_PWB_EN);
2464                         else
2465                                 tmp32 |=  (NV_MCP_SATA_CFG_20_PORT0_EN |
2466                                            NV_MCP_SATA_CFG_20_PORT0_PWB_EN);
2467                         pp = host->ports[1]->private_data;
2468                         if (pp->flags & NV_ADMA_ATAPI_SETUP_COMPLETE)
2469                                 tmp32 &= ~(NV_MCP_SATA_CFG_20_PORT1_EN |
2470                                            NV_MCP_SATA_CFG_20_PORT1_PWB_EN);
2471                         else
2472                                 tmp32 |=  (NV_MCP_SATA_CFG_20_PORT1_EN |
2473                                            NV_MCP_SATA_CFG_20_PORT1_PWB_EN);
2474
2475                         pci_write_config_dword(pdev, NV_MCP_SATA_CFG_20, tmp32);
2476                 }
2477         }
2478
2479         ata_host_resume(host);
2480
2481         return 0;
2482 }
2483 #endif
2484
2485 static void nv_ck804_host_stop(struct ata_host *host)
2486 {
2487         struct pci_dev *pdev = to_pci_dev(host->dev);
2488         u8 regval;
2489
2490         /* disable SATA space for CK804 */
2491         pci_read_config_byte(pdev, NV_MCP_SATA_CFG_20, &regval);
2492         regval &= ~NV_MCP_SATA_CFG_20_SATA_SPACE_EN;
2493         pci_write_config_byte(pdev, NV_MCP_SATA_CFG_20, regval);
2494 }
2495
2496 static void nv_adma_host_stop(struct ata_host *host)
2497 {
2498         struct pci_dev *pdev = to_pci_dev(host->dev);
2499         u32 tmp32;
2500
2501         /* disable ADMA on the ports */
2502         pci_read_config_dword(pdev, NV_MCP_SATA_CFG_20, &tmp32);
2503         tmp32 &= ~(NV_MCP_SATA_CFG_20_PORT0_EN |
2504                    NV_MCP_SATA_CFG_20_PORT0_PWB_EN |
2505                    NV_MCP_SATA_CFG_20_PORT1_EN |
2506                    NV_MCP_SATA_CFG_20_PORT1_PWB_EN);
2507
2508         pci_write_config_dword(pdev, NV_MCP_SATA_CFG_20, tmp32);
2509
2510         nv_ck804_host_stop(host);
2511 }
2512
2513 module_pci_driver(nv_pci_driver);
2514
2515 module_param_named(adma, adma_enabled, bool, 0444);
2516 MODULE_PARM_DESC(adma, "Enable use of ADMA (Default: false)");
2517 module_param_named(swncq, swncq_enabled, bool, 0444);
2518 MODULE_PARM_DESC(swncq, "Enable use of SWNCQ (Default: true)");
2519 module_param_named(msi, msi_enabled, bool, 0444);
2520 MODULE_PARM_DESC(msi, "Enable use of MSI (Default: false)");