vserver 1.9.5.x5
[linux-2.6.git] / drivers / scsi / libata-core.c
index 7f2d11d..d74b7a4 100644 (file)
@@ -28,6 +28,7 @@
 #include <linux/pci.h>
 #include <linux/init.h>
 #include <linux/list.h>
+#include <linux/mm.h>
 #include <linux/highmem.h>
 #include <linux/spinlock.h>
 #include <linux/blkdev.h>
 #include <linux/workqueue.h>
 #include <scsi/scsi.h>
 #include "scsi.h"
+#include "scsi_priv.h"
 #include <scsi/scsi_host.h>
 #include <linux/libata.h>
 #include <asm/io.h>
 #include <asm/semaphore.h>
+#include <asm/byteorder.h>
 
 #include "libata.h"
 
 static unsigned int ata_busy_sleep (struct ata_port *ap,
                                    unsigned long tmout_pat,
                                    unsigned long tmout);
-static void __ata_dev_select (struct ata_port *ap, unsigned int device);
-static void ata_host_set_pio(struct ata_port *ap);
-static void ata_host_set_udma(struct ata_port *ap);
-static void ata_dev_set_pio(struct ata_port *ap, unsigned int device);
-static void ata_dev_set_udma(struct ata_port *ap, unsigned int device);
 static void ata_set_mode(struct ata_port *ap);
+static void ata_dev_set_xfermode(struct ata_port *ap, struct ata_device *dev);
+static unsigned int ata_get_mode_mask(struct ata_port *ap, int shift);
+static int fgb(u32 bitmap);
+static int ata_choose_xfer_mode(struct ata_port *ap,
+                               u8 *xfer_mode_out,
+                               unsigned int *xfer_shift_out);
+static int ata_qc_complete_noop(struct ata_queued_cmd *qc, u8 drv_stat);
+static void __ata_qc_complete(struct ata_queued_cmd *qc);
 
 static unsigned int ata_unique_id = 1;
 static struct workqueue_struct *ata_wq;
@@ -62,19 +68,20 @@ static struct workqueue_struct *ata_wq;
 MODULE_AUTHOR("Jeff Garzik");
 MODULE_DESCRIPTION("Library module for ATA devices");
 MODULE_LICENSE("GPL");
+MODULE_VERSION(DRV_VERSION);
 
 /**
- *     ata_tf_load_pio - send taskfile registers to host controller
+ *     ata_tf_load - send taskfile registers to host controller
  *     @ap: Port to which output is sent
  *     @tf: ATA taskfile register set
  *
- *     Outputs ATA taskfile to standard ATA host controller using PIO.
+ *     Outputs ATA taskfile to standard ATA host controller.
  *
  *     LOCKING:
  *     Inherited from caller.
  */
 
-void ata_tf_load_pio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_tf_load_pio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        struct ata_ioports *ioaddr = &ap->ioaddr;
        unsigned int is_addr = tf->flags & ATA_TFLAG_ISADDR;
@@ -132,23 +139,23 @@ void ata_tf_load_pio(struct ata_port *ap, struct ata_taskfile *tf)
  *     Inherited from caller.
  */
 
-void ata_tf_load_mmio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_tf_load_mmio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        struct ata_ioports *ioaddr = &ap->ioaddr;
        unsigned int is_addr = tf->flags & ATA_TFLAG_ISADDR;
 
        if (tf->ctl != ap->last_ctl) {
-               writeb(tf->ctl, ap->ioaddr.ctl_addr);
+               writeb(tf->ctl, (void __iomem *) ap->ioaddr.ctl_addr);
                ap->last_ctl = tf->ctl;
                ata_wait_idle(ap);
        }
 
        if (is_addr && (tf->flags & ATA_TFLAG_LBA48)) {
-               writeb(tf->hob_feature, (void *) ioaddr->feature_addr);
-               writeb(tf->hob_nsect, (void *) ioaddr->nsect_addr);
-               writeb(tf->hob_lbal, (void *) ioaddr->lbal_addr);
-               writeb(tf->hob_lbam, (void *) ioaddr->lbam_addr);
-               writeb(tf->hob_lbah, (void *) ioaddr->lbah_addr);
+               writeb(tf->hob_feature, (void __iomem *) ioaddr->feature_addr);
+               writeb(tf->hob_nsect, (void __iomem *) ioaddr->nsect_addr);
+               writeb(tf->hob_lbal, (void __iomem *) ioaddr->lbal_addr);
+               writeb(tf->hob_lbam, (void __iomem *) ioaddr->lbam_addr);
+               writeb(tf->hob_lbah, (void __iomem *) ioaddr->lbah_addr);
                VPRINTK("hob: feat 0x%X nsect 0x%X, lba 0x%X 0x%X 0x%X\n",
                        tf->hob_feature,
                        tf->hob_nsect,
@@ -158,11 +165,11 @@ void ata_tf_load_mmio(struct ata_port *ap, struct ata_taskfile *tf)
        }
 
        if (is_addr) {
-               writeb(tf->feature, (void *) ioaddr->feature_addr);
-               writeb(tf->nsect, (void *) ioaddr->nsect_addr);
-               writeb(tf->lbal, (void *) ioaddr->lbal_addr);
-               writeb(tf->lbam, (void *) ioaddr->lbam_addr);
-               writeb(tf->lbah, (void *) ioaddr->lbah_addr);
+               writeb(tf->feature, (void __iomem *) ioaddr->feature_addr);
+               writeb(tf->nsect, (void __iomem *) ioaddr->nsect_addr);
+               writeb(tf->lbal, (void __iomem *) ioaddr->lbal_addr);
+               writeb(tf->lbam, (void __iomem *) ioaddr->lbam_addr);
+               writeb(tf->lbah, (void __iomem *) ioaddr->lbah_addr);
                VPRINTK("feat 0x%X nsect 0x%X lba 0x%X 0x%X 0x%X\n",
                        tf->feature,
                        tf->nsect,
@@ -172,26 +179,34 @@ void ata_tf_load_mmio(struct ata_port *ap, struct ata_taskfile *tf)
        }
 
        if (tf->flags & ATA_TFLAG_DEVICE) {
-               writeb(tf->device, (void *) ioaddr->device_addr);
+               writeb(tf->device, (void __iomem *) ioaddr->device_addr);
                VPRINTK("device 0x%X\n", tf->device);
        }
 
        ata_wait_idle(ap);
 }
 
+void ata_tf_load(struct ata_port *ap, struct ata_taskfile *tf)
+{
+       if (ap->flags & ATA_FLAG_MMIO)
+               ata_tf_load_mmio(ap, tf);
+       else
+               ata_tf_load_pio(ap, tf);
+}
+
 /**
- *     ata_exec_command_pio - issue ATA command to host controller
+ *     ata_exec_command - issue ATA command to host controller
  *     @ap: port to which command is being issued
  *     @tf: ATA taskfile register set
  *
- *     Issues PIO write to ATA command register, with proper
+ *     Issues PIO/MMIO write to ATA command register, with proper
  *     synchronization with interrupt handler / other threads.
  *
  *     LOCKING:
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_exec_command_pio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_exec_command_pio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        DPRINTK("ata%u: cmd 0x%X\n", ap->id, tf->command);
 
@@ -212,20 +227,28 @@ void ata_exec_command_pio(struct ata_port *ap, struct ata_taskfile *tf)
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_exec_command_mmio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_exec_command_mmio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        DPRINTK("ata%u: cmd 0x%X\n", ap->id, tf->command);
 
-               writeb(tf->command, (void *) ap->ioaddr.command_addr);
+               writeb(tf->command, (void __iomem *) ap->ioaddr.command_addr);
        ata_pause(ap);
 }
 
+void ata_exec_command(struct ata_port *ap, struct ata_taskfile *tf)
+{
+       if (ap->flags & ATA_FLAG_MMIO)
+               ata_exec_command_mmio(ap, tf);
+       else
+               ata_exec_command_pio(ap, tf);
+}
+
 /**
  *     ata_exec - issue ATA command to host controller
  *     @ap: port to which command is being issued
  *     @tf: ATA taskfile register set
  *
- *     Issues PIO write to ATA command register, with proper
+ *     Issues PIO/MMIO write to ATA command register, with proper
  *     synchronization with interrupt handler / other threads.
  *
  *     LOCKING:
@@ -248,7 +271,7 @@ static inline void ata_exec(struct ata_port *ap, struct ata_taskfile *tf)
  *     @tf: ATA taskfile register set
  *
  *     Issues ATA taskfile register set to ATA host controller,
- *     via PIO, with proper synchronization with interrupt handler and
+ *     with proper synchronization with interrupt handler and
  *     other threads.
  *
  *     LOCKING:
@@ -268,7 +291,7 @@ static void ata_tf_to_host(struct ata_port *ap, struct ata_taskfile *tf)
  *     @tf: ATA taskfile register set
  *
  *     Issues ATA taskfile register set to ATA host controller,
- *     via PIO, with proper synchronization with interrupt handler and
+ *     with proper synchronization with interrupt handler and
  *     other threads.
  *
  *     LOCKING:
@@ -282,18 +305,18 @@ void ata_tf_to_host_nolock(struct ata_port *ap, struct ata_taskfile *tf)
 }
 
 /**
- *     ata_tf_read_pio - input device's ATA taskfile shadow registers
+ *     ata_tf_read - input device's ATA taskfile shadow registers
  *     @ap: Port from which input is read
  *     @tf: ATA taskfile register set for storing input
  *
  *     Reads ATA taskfile registers for currently-selected device
- *     into @tf via PIO.
+ *     into @tf.
  *
  *     LOCKING:
  *     Inherited from caller.
  */
 
-void ata_tf_read_pio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_tf_read_pio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        struct ata_ioports *ioaddr = &ap->ioaddr;
 
@@ -325,38 +348,46 @@ void ata_tf_read_pio(struct ata_port *ap, struct ata_taskfile *tf)
  *     Inherited from caller.
  */
 
-void ata_tf_read_mmio(struct ata_port *ap, struct ata_taskfile *tf)
+static void ata_tf_read_mmio(struct ata_port *ap, struct ata_taskfile *tf)
 {
        struct ata_ioports *ioaddr = &ap->ioaddr;
 
-       tf->nsect = readb((void *)ioaddr->nsect_addr);
-       tf->lbal = readb((void *)ioaddr->lbal_addr);
-       tf->lbam = readb((void *)ioaddr->lbam_addr);
-       tf->lbah = readb((void *)ioaddr->lbah_addr);
-       tf->device = readb((void *)ioaddr->device_addr);
+       tf->nsect = readb((void __iomem *)ioaddr->nsect_addr);
+       tf->lbal = readb((void __iomem *)ioaddr->lbal_addr);
+       tf->lbam = readb((void __iomem *)ioaddr->lbam_addr);
+       tf->lbah = readb((void __iomem *)ioaddr->lbah_addr);
+       tf->device = readb((void __iomem *)ioaddr->device_addr);
 
        if (tf->flags & ATA_TFLAG_LBA48) {
-               writeb(tf->ctl | ATA_HOB, ap->ioaddr.ctl_addr);
-               tf->hob_feature = readb((void *)ioaddr->error_addr);
-               tf->hob_nsect = readb((void *)ioaddr->nsect_addr);
-               tf->hob_lbal = readb((void *)ioaddr->lbal_addr);
-               tf->hob_lbam = readb((void *)ioaddr->lbam_addr);
-               tf->hob_lbah = readb((void *)ioaddr->lbah_addr);
+               writeb(tf->ctl | ATA_HOB, (void __iomem *) ap->ioaddr.ctl_addr);
+               tf->hob_feature = readb((void __iomem *)ioaddr->error_addr);
+               tf->hob_nsect = readb((void __iomem *)ioaddr->nsect_addr);
+               tf->hob_lbal = readb((void __iomem *)ioaddr->lbal_addr);
+               tf->hob_lbam = readb((void __iomem *)ioaddr->lbam_addr);
+               tf->hob_lbah = readb((void __iomem *)ioaddr->lbah_addr);
        }
 }
 
+void ata_tf_read(struct ata_port *ap, struct ata_taskfile *tf)
+{
+       if (ap->flags & ATA_FLAG_MMIO)
+               ata_tf_read_mmio(ap, tf);
+       else
+               ata_tf_read_pio(ap, tf);
+}
+
 /**
  *     ata_check_status_pio - Read device status reg & clear interrupt
  *     @ap: port where the device is
  *
  *     Reads ATA taskfile status register for currently-selected device
- *     via PIO and return it's value. This also clears pending interrupts
+ *     and return it's value. This also clears pending interrupts
  *      from this device
  *
  *     LOCKING:
  *     Inherited from caller.
  */
-u8 ata_check_status_pio(struct ata_port *ap)
+static u8 ata_check_status_pio(struct ata_port *ap)
 {
        return inb(ap->ioaddr.status_addr);
 }
@@ -372,9 +403,37 @@ u8 ata_check_status_pio(struct ata_port *ap)
  *     LOCKING:
  *     Inherited from caller.
  */
-u8 ata_check_status_mmio(struct ata_port *ap)
+static u8 ata_check_status_mmio(struct ata_port *ap)
+{
+               return readb((void __iomem *) ap->ioaddr.status_addr);
+}
+
+u8 ata_check_status(struct ata_port *ap)
+{
+       if (ap->flags & ATA_FLAG_MMIO)
+               return ata_check_status_mmio(ap);
+       return ata_check_status_pio(ap);
+}
+
+u8 ata_altstatus(struct ata_port *ap)
+{
+       if (ap->ops->check_altstatus)
+               return ap->ops->check_altstatus(ap);
+
+       if (ap->flags & ATA_FLAG_MMIO)
+               return readb((void __iomem *)ap->ioaddr.altstatus_addr);
+       return inb(ap->ioaddr.altstatus_addr);
+}
+
+u8 ata_chk_err(struct ata_port *ap)
 {
-               return readb((void *) ap->ioaddr.status_addr);
+       if (ap->ops->check_err)
+               return ap->ops->check_err(ap);
+
+       if (ap->flags & ATA_FLAG_MMIO) {
+               return readb((void __iomem *) ap->ioaddr.error_addr);
+       }
+       return inb(ap->ioaddr.error_addr);
 }
 
 /**
@@ -524,7 +583,7 @@ static void ata_dev_set_protocol(struct ata_device *dev)
        dev->write_cmd = (cmd >> 8) & 0xff;
 }
 
-static const char * udma_str[] = {
+static const char * xfer_mode_str[] = {
        "UDMA/16",
        "UDMA/25",
        "UDMA/33",
@@ -533,11 +592,19 @@ static const char * udma_str[] = {
        "UDMA/100",
        "UDMA/133",
        "UDMA7",
+       "MWDMA0",
+       "MWDMA1",
+       "MWDMA2",
+       "PIO0",
+       "PIO1",
+       "PIO2",
+       "PIO3",
+       "PIO4",
 };
 
 /**
  *     ata_udma_string - convert UDMA bit offset to string
- *     @udma_mask: mask of bits supported; only highest bit counts.
+ *     @mask: mask of bits supported; only highest bit counts.
  *
  *     Determine string which represents the highest speed
  *     (highest bit in @udma_mask).
@@ -550,16 +617,24 @@ static const char * udma_str[] = {
  *     @udma_mask, or the constant C string "<n/a>".
  */
 
-static const char *ata_udma_string(unsigned int udma_mask)
+static const char *ata_mode_string(unsigned int mask)
 {
        int i;
 
-       for (i = 7; i >= 0; i--) {
-               if (udma_mask & (1 << i))
-                       return udma_str[i];
-       }
+       for (i = 7; i >= 0; i--)
+               if (mask & (1 << i))
+                       goto out;
+       for (i = ATA_SHIFT_MWDMA + 2; i >= ATA_SHIFT_MWDMA; i--)
+               if (mask & (1 << i))
+                       goto out;
+       for (i = ATA_SHIFT_PIO + 4; i >= ATA_SHIFT_PIO; i--)
+               if (mask & (1 << i))
+                       goto out;
 
        return "<n/a>";
+
+out:
+       return xfer_mode_str[i];
 }
 
 /**
@@ -586,7 +661,7 @@ static unsigned int ata_pio_devchk(struct ata_port *ap,
        struct ata_ioports *ioaddr = &ap->ioaddr;
        u8 nsect, lbal;
 
-       __ata_dev_select(ap, device);
+       ap->ops->dev_select(ap, device);
 
        outb(0x55, ioaddr->nsect_addr);
        outb(0xaa, ioaddr->lbal_addr);
@@ -630,19 +705,19 @@ static unsigned int ata_mmio_devchk(struct ata_port *ap,
        struct ata_ioports *ioaddr = &ap->ioaddr;
        u8 nsect, lbal;
 
-       __ata_dev_select(ap, device);
+       ap->ops->dev_select(ap, device);
 
-       writeb(0x55, (void *) ioaddr->nsect_addr);
-       writeb(0xaa, (void *) ioaddr->lbal_addr);
+       writeb(0x55, (void __iomem *) ioaddr->nsect_addr);
+       writeb(0xaa, (void __iomem *) ioaddr->lbal_addr);
 
-       writeb(0xaa, (void *) ioaddr->nsect_addr);
-       writeb(0x55, (void *) ioaddr->lbal_addr);
+       writeb(0xaa, (void __iomem *) ioaddr->nsect_addr);
+       writeb(0x55, (void __iomem *) ioaddr->lbal_addr);
 
-       writeb(0x55, (void *) ioaddr->nsect_addr);
-       writeb(0xaa, (void *) ioaddr->lbal_addr);
+       writeb(0x55, (void __iomem *) ioaddr->nsect_addr);
+       writeb(0xaa, (void __iomem *) ioaddr->lbal_addr);
 
-       nsect = readb((void *) ioaddr->nsect_addr);
-       lbal = readb((void *) ioaddr->lbal_addr);
+       nsect = readb((void __iomem *) ioaddr->nsect_addr);
+       lbal = readb((void __iomem *) ioaddr->lbal_addr);
 
        if ((nsect == 0x55) && (lbal == 0xaa))
                return 1;       /* we found a device */
@@ -651,7 +726,7 @@ static unsigned int ata_mmio_devchk(struct ata_port *ap,
 }
 
 /**
- *     ata_dev_devchk - PATA device presence detection
+ *     ata_devchk - PATA device presence detection
  *     @ap: ATA channel to examine
  *     @device: Device to examine (starting at zero)
  *
@@ -663,7 +738,7 @@ static unsigned int ata_mmio_devchk(struct ata_port *ap,
  *     caller.
  */
 
-static unsigned int ata_dev_devchk(struct ata_port *ap,
+static unsigned int ata_devchk(struct ata_port *ap,
                                    unsigned int device)
 {
        if (ap->flags & ATA_FLAG_MMIO)
@@ -687,7 +762,7 @@ static unsigned int ata_dev_devchk(struct ata_port *ap,
  *     the event of failure.
  */
 
-static unsigned int ata_dev_classify(struct ata_taskfile *tf)
+unsigned int ata_dev_classify(struct ata_taskfile *tf)
 {
        /* Apple's open source Darwin code hints that some devices only
         * put a proper signature into the LBA mid/high registers,
@@ -735,7 +810,7 @@ static u8 ata_dev_try_classify(struct ata_port *ap, unsigned int device)
        unsigned int class;
        u8 err;
 
-       __ata_dev_select(ap, device);
+       ap->ops->dev_select(ap, device);
 
        memset(&tf, 0, sizeof(tf));
 
@@ -766,7 +841,7 @@ static u8 ata_dev_try_classify(struct ata_port *ap, unsigned int device)
 
 /**
  *     ata_dev_id_string - Convert IDENTIFY DEVICE page into string
- *     @dev: Device whose IDENTIFY DEVICE results we will examine
+ *     @id: IDENTIFY DEVICE results we will examine
  *     @s: string into which data is output
  *     @ofs: offset into identify device page
  *     @len: length of string to return. must be an even number.
@@ -779,17 +854,17 @@ static u8 ata_dev_try_classify(struct ata_port *ap, unsigned int device)
  *     caller.
  */
 
-void ata_dev_id_string(struct ata_device *dev, unsigned char *s,
+void ata_dev_id_string(u16 *id, unsigned char *s,
                       unsigned int ofs, unsigned int len)
 {
        unsigned int c;
 
        while (len > 0) {
-               c = dev->id[ofs] >> 8;
+               c = id[ofs] >> 8;
                *s = c;
                s++;
 
-               c = dev->id[ofs] & 0xff;
+               c = id[ofs] & 0xff;
                *s = c;
                s++;
 
@@ -798,8 +873,12 @@ void ata_dev_id_string(struct ata_device *dev, unsigned char *s,
        }
 }
 
+void ata_noop_dev_select (struct ata_port *ap, unsigned int device)
+{
+}
+
 /**
- *     __ata_dev_select - Select device 0/1 on ATA bus
+ *     ata_std_dev_select - Select device 0/1 on ATA bus
  *     @ap: ATA channel to manipulate
  *     @device: ATA device (numbered from zero) to select
  *
@@ -811,7 +890,7 @@ void ata_dev_id_string(struct ata_device *dev, unsigned char *s,
  *     caller.
  */
 
-static void __ata_dev_select (struct ata_port *ap, unsigned int device)
+void ata_std_dev_select (struct ata_port *ap, unsigned int device)
 {
        u8 tmp;
 
@@ -821,7 +900,7 @@ static void __ata_dev_select (struct ata_port *ap, unsigned int device)
                tmp = ATA_DEVICE_OBS | ATA_DEV1;
 
        if (ap->flags & ATA_FLAG_MMIO) {
-               writeb(tmp, (void *) ap->ioaddr.device_addr);
+               writeb(tmp, (void __iomem *) ap->ioaddr.device_addr);
        } else {
                outb(tmp, ap->ioaddr.device_addr);
        }
@@ -839,7 +918,7 @@ static void __ata_dev_select (struct ata_port *ap, unsigned int device)
  *     make either device 0, or device 1, active on the
  *     ATA channel.
  *
- *     This is a high-level version of __ata_dev_select(),
+ *     This is a high-level version of ata_std_dev_select(),
  *     which additionally provides the services of inserting
  *     the proper pauses and status polling, where needed.
  *
@@ -856,7 +935,7 @@ void ata_dev_select(struct ata_port *ap, unsigned int device,
        if (wait)
                ata_wait_idle(ap);
 
-       __ata_dev_select(ap, device);
+       ap->ops->dev_select(ap, device);
 
        if (wait) {
                if (can_sleep && ap->device[device].class == ATA_DEV_ATAPI)
@@ -930,10 +1009,14 @@ static void ata_dev_identify(struct ata_port *ap, unsigned int device)
 {
        struct ata_device *dev = &ap->device[device];
        unsigned int i;
-       u16 tmp, udma_modes;
+       u16 tmp;
+       unsigned long xfer_modes;
        u8 status;
-       struct ata_taskfile tf;
        unsigned int using_edd;
+       DECLARE_COMPLETION(wait);
+       struct ata_queued_cmd *qc;
+       unsigned long flags;
+       int rc;
 
        if (!ata_dev_present(dev)) {
                DPRINTK("ENTER/EXIT (host %u, dev %u) -- nodev\n",
@@ -953,27 +1036,34 @@ static void ata_dev_identify(struct ata_port *ap, unsigned int device)
 
        ata_dev_select(ap, device, 1, 1); /* select device 0/1 */
 
-retry:
-       ata_tf_init(ap, &tf, device);
-       tf.ctl |= ATA_NIEN;
-       tf.protocol = ATA_PROT_PIO;
+       qc = ata_qc_new_init(ap, dev);
+       BUG_ON(qc == NULL);
 
+       ata_sg_init_one(qc, dev->id, sizeof(dev->id));
+       qc->dma_dir = DMA_FROM_DEVICE;
+       qc->tf.protocol = ATA_PROT_PIO;
+       qc->nsect = 1;
+
+retry:
        if (dev->class == ATA_DEV_ATA) {
-               tf.command = ATA_CMD_ID_ATA;
+               qc->tf.command = ATA_CMD_ID_ATA;
                DPRINTK("do ATA identify\n");
        } else {
-               tf.command = ATA_CMD_ID_ATAPI;
+               qc->tf.command = ATA_CMD_ID_ATAPI;
                DPRINTK("do ATAPI identify\n");
        }
 
-       ata_tf_to_host(ap, &tf);
+       qc->waiting = &wait;
+       qc->complete_fn = ata_qc_complete_noop;
 
-       /* crazy ATAPI devices... */
-       if (dev->class == ATA_DEV_ATAPI)
-               msleep(150);
+       spin_lock_irqsave(&ap->host_set->lock, flags);
+       rc = ata_qc_issue(qc);
+       spin_unlock_irqrestore(&ap->host_set->lock, flags);
 
-       if (ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT))
+       if (rc)
                goto err_out;
+       else
+               wait_for_completion(&wait);
 
        status = ata_chk_status(ap);
        if (status & ATA_ERR) {
@@ -988,44 +1078,21 @@ retry:
                 * ATA software reset (SRST, the default) does not appear
                 * to have this problem.
                 */
-               if ((using_edd) && (tf.command == ATA_CMD_ID_ATA)) {
+               if ((using_edd) && (qc->tf.command == ATA_CMD_ID_ATA)) {
                        u8 err = ata_chk_err(ap);
                        if (err & ATA_ABORTED) {
                                dev->class = ATA_DEV_ATAPI;
+                               qc->cursg = 0;
+                               qc->cursg_ofs = 0;
+                               qc->cursect = 0;
+                               qc->nsect = 1;
                                goto retry;
                        }
                }
                goto err_out;
        }
 
-       /* make sure we have BSY=0, DRQ=1 */
-       if ((status & ATA_DRQ) == 0) {
-               printk(KERN_WARNING "ata%u: dev %u (ATA%s?) not returning id page (0x%x)\n",
-                      ap->id, device,
-                      dev->class == ATA_DEV_ATA ? "" : "PI",
-                      status);
-               goto err_out;
-       }
-
-       /* read IDENTIFY [X] DEVICE page */
-       if (ap->flags & ATA_FLAG_MMIO) {
-               for (i = 0; i < ATA_ID_WORDS; i++)
-                       dev->id[i] = readw((void *)ap->ioaddr.data_addr);
-       } else
-               for (i = 0; i < ATA_ID_WORDS; i++)
-                       dev->id[i] = inw(ap->ioaddr.data_addr);
-
-       /* wait for host_idle */
-       status = ata_wait_idle(ap);
-       if (status & (ATA_BUSY | ATA_DRQ)) {
-               printk(KERN_WARNING "ata%u: dev %u (ATA%s?) error after id page (0x%x)\n",
-                      ap->id, device,
-                      dev->class == ATA_DEV_ATA ? "" : "PI",
-                      status);
-               goto err_out;
-       }
-
-       ata_irq_on(ap); /* re-enable interrupts */
+       swap_buf_le16(dev->id, ATA_ID_WORDS);
 
        /* print device capabilities */
        printk(KERN_DEBUG "ata%u: dev %u cfg "
@@ -1040,24 +1107,25 @@ retry:
         */
 
        /* we require LBA and DMA support (bits 8 & 9 of word 49) */
-       if (!ata_id_has_dma(dev) || !ata_id_has_lba(dev)) {
+       if (!ata_id_has_dma(dev->id) || !ata_id_has_lba(dev->id)) {
                printk(KERN_DEBUG "ata%u: no dma/lba\n", ap->id);
                goto err_out_nosup;
        }
 
-       /* we require UDMA support */
-       udma_modes =
-       tmp = dev->id[ATA_ID_UDMA_MODES];
-       if ((tmp & 0xff) == 0) {
-               printk(KERN_DEBUG "ata%u: no udma\n", ap->id);
-               goto err_out_nosup;
+       /* quick-n-dirty find max transfer mode; for printk only */
+       xfer_modes = dev->id[ATA_ID_UDMA_MODES];
+       if (!xfer_modes)
+               xfer_modes = (dev->id[ATA_ID_MWDMA_MODES]) << ATA_SHIFT_MWDMA;
+       if (!xfer_modes) {
+               xfer_modes = (dev->id[ATA_ID_PIO_MODES]) << (ATA_SHIFT_PIO + 3);
+               xfer_modes |= (0x7 << ATA_SHIFT_PIO);
        }
 
        ata_dump_id(dev);
 
        /* ATA-specific feature tests */
        if (dev->class == ATA_DEV_ATA) {
-               if (!ata_id_is_ata(dev))        /* sanity check */
+               if (!ata_id_is_ata(dev->id))    /* sanity check */
                        goto err_out_nosup;
 
                tmp = dev->id[ATA_ID_MAJOR_VER];
@@ -1071,11 +1139,11 @@ retry:
                        goto err_out_nosup;
                }
 
-               if (ata_id_has_lba48(dev)) {
+               if (ata_id_has_lba48(dev->id)) {
                        dev->flags |= ATA_DFLAG_LBA48;
-                       dev->n_sectors = ata_id_u64(dev, 100);
+                       dev->n_sectors = ata_id_u64(dev->id, 100);
                } else {
-                       dev->n_sectors = ata_id_u32(dev, 60);
+                       dev->n_sectors = ata_id_u32(dev->id, 60);
                }
 
                ap->host->max_cmd_len = 16;
@@ -1083,25 +1151,28 @@ retry:
                /* print device info to dmesg */
                printk(KERN_INFO "ata%u: dev %u ATA, max %s, %Lu sectors:%s\n",
                       ap->id, device,
-                      ata_udma_string(udma_modes),
+                      ata_mode_string(xfer_modes),
                       (unsigned long long)dev->n_sectors,
                       dev->flags & ATA_DFLAG_LBA48 ? " lba48" : "");
        }
 
        /* ATAPI-specific feature tests */
        else {
-               if (ata_id_is_ata(dev))         /* sanity check */
+               if (ata_id_is_ata(dev->id))             /* sanity check */
                        goto err_out_nosup;
 
-               /* see if 16-byte commands supported */
-               tmp = dev->id[0] & 0x3;
-               if (tmp == 1)
-                       ap->host->max_cmd_len = 16;
+               rc = atapi_cdb_len(dev->id);
+               if ((rc < 12) || (rc > ATAPI_CDB_LEN)) {
+                       printk(KERN_WARNING "ata%u: unsupported CDB len\n", ap->id);
+                       goto err_out_nosup;
+               }
+               ap->cdb_len = (unsigned int) rc;
+               ap->host->max_cmd_len = (unsigned char) ap->cdb_len;
 
                /* print device info to dmesg */
                printk(KERN_INFO "ata%u: dev %u ATAPI, max %s\n",
                       ap->id, device,
-                      ata_udma_string(udma_modes));
+                      ata_mode_string(xfer_modes));
        }
 
        DPRINTK("EXIT, drv_stat = 0x%x\n", ata_chk_status(ap));
@@ -1111,7 +1182,6 @@ err_out_nosup:
        printk(KERN_WARNING "ata%u: dev %u not supported, ignoring\n",
               ap->id, device);
 err_out:
-       ata_irq_on(ap); /* re-enable interrupts */
        dev->class++;   /* converts ATA_DEV_xxx into ATA_DEV_xxx_UNSUP */
        DPRINTK("EXIT, err\n");
 }
@@ -1171,13 +1241,13 @@ void ata_port_probe(struct ata_port *ap)
 }
 
 /**
- *     sata_phy_reset -
+ *     __sata_phy_reset -
  *     @ap:
  *
  *     LOCKING:
  *
  */
-void sata_phy_reset(struct ata_port *ap)
+void __sata_phy_reset(struct ata_port *ap)
 {
        u32 sstatus;
        unsigned long timeout = jiffies + (HZ * 5);
@@ -1215,6 +1285,21 @@ void sata_phy_reset(struct ata_port *ap)
                return;
        }
 
+       ap->cbl = ATA_CBL_SATA;
+}
+
+/**
+ *     __sata_phy_reset -
+ *     @ap:
+ *
+ *     LOCKING:
+ *
+ */
+void sata_phy_reset(struct ata_port *ap)
+{
+       __sata_phy_reset(ap);
+       if (ap->flags & ATA_FLAG_PORT_DISABLED)
+               return;
        ata_bus_reset(ap);
 }
 
@@ -1232,6 +1317,101 @@ void ata_port_disable(struct ata_port *ap)
        ap->flags |= ATA_FLAG_PORT_DISABLED;
 }
 
+static struct {
+       unsigned int shift;
+       u8 base;
+} xfer_mode_classes[] = {
+       { ATA_SHIFT_UDMA,       XFER_UDMA_0 },
+       { ATA_SHIFT_MWDMA,      XFER_MW_DMA_0 },
+       { ATA_SHIFT_PIO,        XFER_PIO_0 },
+};
+
+static inline u8 base_from_shift(unsigned int shift)
+{
+       int i;
+
+       for (i = 0; i < ARRAY_SIZE(xfer_mode_classes); i++)
+               if (xfer_mode_classes[i].shift == shift)
+                       return xfer_mode_classes[i].base;
+
+       return 0xff;
+}
+
+static void ata_dev_set_mode(struct ata_port *ap, struct ata_device *dev)
+{
+       int ofs, idx;
+       u8 base;
+
+       if (!ata_dev_present(dev) || (ap->flags & ATA_FLAG_PORT_DISABLED))
+               return;
+
+       if (dev->xfer_shift == ATA_SHIFT_PIO)
+               dev->flags |= ATA_DFLAG_PIO;
+
+       ata_dev_set_xfermode(ap, dev);
+
+       base = base_from_shift(dev->xfer_shift);
+       ofs = dev->xfer_mode - base;
+       idx = ofs + dev->xfer_shift;
+       WARN_ON(idx >= ARRAY_SIZE(xfer_mode_str));
+
+       DPRINTK("idx=%d xfer_shift=%u, xfer_mode=0x%x, base=0x%x, offset=%d\n",
+               idx, dev->xfer_shift, (int)dev->xfer_mode, (int)base, ofs);
+
+       printk(KERN_INFO "ata%u: dev %u configured for %s\n",
+               ap->id, dev->devno, xfer_mode_str[idx]);
+}
+
+static int ata_host_set_pio(struct ata_port *ap)
+{
+       unsigned int mask;
+       int x, i;
+       u8 base, xfer_mode;
+
+       mask = ata_get_mode_mask(ap, ATA_SHIFT_PIO);
+       x = fgb(mask);
+       if (x < 0) {
+               printk(KERN_WARNING "ata%u: no PIO support\n", ap->id);
+               return -1;
+       }
+
+       base = base_from_shift(ATA_SHIFT_PIO);
+       xfer_mode = base + x;
+
+       DPRINTK("base 0x%x xfer_mode 0x%x mask 0x%x x %d\n",
+               (int)base, (int)xfer_mode, mask, x);
+
+       for (i = 0; i < ATA_MAX_DEVICES; i++) {
+               struct ata_device *dev = &ap->device[i];
+               if (ata_dev_present(dev)) {
+                       dev->pio_mode = xfer_mode;
+                       dev->xfer_mode = xfer_mode;
+                       dev->xfer_shift = ATA_SHIFT_PIO;
+                       if (ap->ops->set_piomode)
+                               ap->ops->set_piomode(ap, dev);
+               }
+       }
+
+       return 0;
+}
+
+static void ata_host_set_dma(struct ata_port *ap, u8 xfer_mode,
+                           unsigned int xfer_shift)
+{
+       int i;
+
+       for (i = 0; i < ATA_MAX_DEVICES; i++) {
+               struct ata_device *dev = &ap->device[i];
+               if (ata_dev_present(dev)) {
+                       dev->dma_mode = xfer_mode;
+                       dev->xfer_mode = xfer_mode;
+                       dev->xfer_shift = xfer_shift;
+                       if (ap->ops->set_dmamode)
+                               ap->ops->set_dmamode(ap, dev);
+               }
+       }
+}
+
 /**
  *     ata_set_mode - Program timings and issue SET FEATURES - XFER
  *     @ap: port on which timings will be programmed
@@ -1241,29 +1421,28 @@ void ata_port_disable(struct ata_port *ap)
  */
 static void ata_set_mode(struct ata_port *ap)
 {
-       unsigned int force_pio, i;
+       unsigned int i, xfer_shift;
+       u8 xfer_mode;
+       int rc;
 
-       ata_host_set_pio(ap);
-       if (ap->flags & ATA_FLAG_PORT_DISABLED)
-               return;
+       /* step 1: always set host PIO timings */
+       rc = ata_host_set_pio(ap);
+       if (rc)
+               goto err_out;
 
-       ata_host_set_udma(ap);
-       if (ap->flags & ATA_FLAG_PORT_DISABLED)
-               return;
+       /* step 2: choose the best data xfer mode */
+       xfer_mode = xfer_shift = 0;
+       rc = ata_choose_xfer_mode(ap, &xfer_mode, &xfer_shift);
+       if (rc)
+               goto err_out;
 
-#ifdef ATA_FORCE_PIO
-       force_pio = 1;
-#else
-       force_pio = 0;
-#endif
+       /* step 3: if that xfer mode isn't PIO, set host DMA timings */
+       if (xfer_shift != ATA_SHIFT_PIO)
+               ata_host_set_dma(ap, xfer_mode, xfer_shift);
 
-       if (force_pio) {
-               ata_dev_set_pio(ap, 0);
-               ata_dev_set_pio(ap, 1);
-       } else {
-               ata_dev_set_udma(ap, 0);
-               ata_dev_set_udma(ap, 1);
-       }
+       /* step 4: update devices' xfer mode */
+       ata_dev_set_mode(ap, &ap->device[0]);
+       ata_dev_set_mode(ap, &ap->device[1]);
 
        if (ap->flags & ATA_FLAG_PORT_DISABLED)
                return;
@@ -1275,6 +1454,11 @@ static void ata_set_mode(struct ata_port *ap)
                struct ata_device *dev = &ap->device[i];
                ata_dev_set_protocol(dev);
        }
+
+       return;
+
+err_out:
+       ata_port_disable(ap);
 }
 
 /**
@@ -1328,23 +1512,23 @@ static void ata_bus_post_reset(struct ata_port *ap, unsigned int devmask)
        unsigned int dev1 = devmask & (1 << 1);
        unsigned long timeout;
 
-       /* if device 0 was found in ata_dev_devchk, wait for its
+       /* if device 0 was found in ata_devchk, wait for its
         * BSY bit to clear
         */
        if (dev0)
                ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
 
-       /* if device 1 was found in ata_dev_devchk, wait for
+       /* if device 1 was found in ata_devchk, wait for
         * register access, then wait for BSY to clear
         */
        timeout = jiffies + ATA_TMOUT_BOOT;
        while (dev1) {
                u8 nsect, lbal;
 
-               __ata_dev_select(ap, 1);
+               ap->ops->dev_select(ap, 1);
                if (ap->flags & ATA_FLAG_MMIO) {
-                       nsect = readb((void *) ioaddr->nsect_addr);
-                       lbal = readb((void *) ioaddr->lbal_addr);
+                       nsect = readb((void __iomem *) ioaddr->nsect_addr);
+                       lbal = readb((void __iomem *) ioaddr->lbal_addr);
                } else {
                        nsect = inb(ioaddr->nsect_addr);
                        lbal = inb(ioaddr->lbal_addr);
@@ -1361,11 +1545,11 @@ static void ata_bus_post_reset(struct ata_port *ap, unsigned int devmask)
                ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
 
        /* is all this really necessary? */
-       __ata_dev_select(ap, 0);
+       ap->ops->dev_select(ap, 0);
        if (dev1)
-               __ata_dev_select(ap, 1);
+               ap->ops->dev_select(ap, 1);
        if (dev0)
-               __ata_dev_select(ap, 0);
+               ap->ops->dev_select(ap, 0);
 }
 
 /**
@@ -1408,11 +1592,11 @@ static unsigned int ata_bus_softreset(struct ata_port *ap,
 
        /* software reset.  causes dev0 to be selected */
        if (ap->flags & ATA_FLAG_MMIO) {
-               writeb(ap->ctl, ioaddr->ctl_addr);
+               writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
                udelay(20);     /* FIXME: flush */
-               writeb(ap->ctl | ATA_SRST, ioaddr->ctl_addr);
+               writeb(ap->ctl | ATA_SRST, (void __iomem *) ioaddr->ctl_addr);
                udelay(20);     /* FIXME: flush */
-               writeb(ap->ctl, ioaddr->ctl_addr);
+               writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
        } else {
                outb(ap->ctl, ioaddr->ctl_addr);
                udelay(10);
@@ -1469,9 +1653,9 @@ void ata_bus_reset(struct ata_port *ap)
        if (ap->flags & ATA_FLAG_SATA_RESET)
                dev0 = 1;
        else {
-               dev0 = ata_dev_devchk(ap, 0);
+               dev0 = ata_devchk(ap, 0);
                if (slave_possible)
-                       dev1 = ata_dev_devchk(ap, 1);
+                       dev1 = ata_devchk(ap, 1);
        }
 
        if (dev0)
@@ -1480,7 +1664,7 @@ void ata_bus_reset(struct ata_port *ap)
                devmask |= (1 << 1);
 
        /* select device 0 again */
-       __ata_dev_select(ap, 0);
+       ap->ops->dev_select(ap, 0);
 
        /* issue bus reset */
        if (ap->flags & ATA_FLAG_SRST)
@@ -1488,7 +1672,7 @@ void ata_bus_reset(struct ata_port *ap)
        else if ((ap->flags & ATA_FLAG_SATA_RESET) == 0) {
                /* set up device control */
                if (ap->flags & ATA_FLAG_MMIO)
-                       writeb(ap->ctl, ioaddr->ctl_addr);
+                       writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
                else
                        outb(ap->ctl, ioaddr->ctl_addr);
                rc = ata_bus_edd(ap);
@@ -1505,13 +1689,14 @@ void ata_bus_reset(struct ata_port *ap)
                ata_dev_try_classify(ap, 1);
 
        /* re-enable interrupts */
-       ata_irq_on(ap);
+       if (ap->ioaddr.ctl_addr)        /* FIXME: hack. create a hook instead */
+               ata_irq_on(ap);
 
        /* is double-select really necessary? */
        if (ap->device[1].class != ATA_DEV_NONE)
-               __ata_dev_select(ap, 1);
+               ap->ops->dev_select(ap, 1);
        if (ap->device[0].class != ATA_DEV_NONE)
-               __ata_dev_select(ap, 0);
+               ap->ops->dev_select(ap, 0);
 
        /* if no devices were detected, disable this port */
        if ((ap->device[0].class == ATA_DEV_NONE) &&
@@ -1521,7 +1706,7 @@ void ata_bus_reset(struct ata_port *ap)
        if (ap->flags & (ATA_FLAG_SATA_RESET | ATA_FLAG_SRST)) {
                /* set up device control for ATA_FLAG_SATA_RESET */
                if (ap->flags & ATA_FLAG_MMIO)
-                       writeb(ap->ctl, ioaddr->ctl_addr);
+                       writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
                else
                        outb(ap->ctl, ioaddr->ctl_addr);
        }
@@ -1536,208 +1721,229 @@ err_out:
        DPRINTK("EXIT\n");
 }
 
-/**
- *     ata_host_set_pio -
- *     @ap:
- *
- *     LOCKING:
- */
+static void ata_pr_blacklisted(struct ata_port *ap, struct ata_device *dev)
+{
+       printk(KERN_WARNING "ata%u: dev %u is on DMA blacklist, disabling DMA\n",
+               ap->id, dev->devno);
+}
+
+static const char * ata_dma_blacklist [] = {
+       "WDC AC11000H",
+       "WDC AC22100H",
+       "WDC AC32500H",
+       "WDC AC33100H",
+       "WDC AC31600H",
+       "WDC AC32100H",
+       "WDC AC23200L",
+       "Compaq CRD-8241B",
+       "CRD-8400B",
+       "CRD-8480B",
+       "CRD-8482B",
+       "CRD-84",
+       "SanDisk SDP3B",
+       "SanDisk SDP3B-64",
+       "SANYO CD-ROM CRD",
+       "HITACHI CDR-8",
+       "HITACHI CDR-8335",
+       "HITACHI CDR-8435",
+       "Toshiba CD-ROM XM-6202B",
+       "CD-532E-A",
+       "E-IDE CD-ROM CR-840",
+       "CD-ROM Drive/F5A",
+       "WPI CDD-820",
+       "SAMSUNG CD-ROM SC-148C",
+       "SAMSUNG CD-ROM SC",
+       "SanDisk SDP3B-64",
+       "SAMSUNG CD-ROM SN-124",
+       "ATAPI CD-ROM DRIVE 40X MAXIMUM",
+       "_NEC DV5800A",
+};
 
-static void ata_host_set_pio(struct ata_port *ap)
+static int ata_dma_blacklisted(struct ata_port *ap, struct ata_device *dev)
 {
-       struct ata_device *master, *slave;
-       unsigned int pio, i;
-       u16 mask;
-
-       master = &ap->device[0];
-       slave = &ap->device[1];
-
-       assert (ata_dev_present(master) || ata_dev_present(slave));
+       unsigned char model_num[40];
+       char *s;
+       unsigned int len;
+       int i;
 
-       mask = ap->pio_mask;
-       if (ata_dev_present(master))
-               mask &= (master->id[ATA_ID_PIO_MODES] & 0x03);
-       if (ata_dev_present(slave))
-               mask &= (slave->id[ATA_ID_PIO_MODES] & 0x03);
+       ata_dev_id_string(dev->id, model_num, ATA_ID_PROD_OFS,
+                         sizeof(model_num));
+       s = &model_num[0];
+       len = strnlen(s, sizeof(model_num));
 
-       /* require pio mode 3 or 4 support for host and all devices */
-       if (mask == 0) {
-               printk(KERN_WARNING "ata%u: no PIO3/4 support, ignoring\n",
-                      ap->id);
-               goto err_out;
+       /* ATAPI specifies that empty space is blank-filled; remove blanks */
+       while ((len > 0) && (s[len - 1] == ' ')) {
+               len--;
+               s[len] = 0;
        }
 
-       pio = (mask & ATA_ID_PIO4) ? 4 : 3;
-       for (i = 0; i < ATA_MAX_DEVICES; i++)
-               if (ata_dev_present(&ap->device[i])) {
-                       ap->device[i].pio_mode = (pio == 3) ?
-                               XFER_PIO_3 : XFER_PIO_4;
-                       if (ap->ops->set_piomode)
-                               ap->ops->set_piomode(ap, &ap->device[i], pio);
-               }
-
-       return;
+       for (i = 0; i < ARRAY_SIZE(ata_dma_blacklist); i++)
+               if (!strncmp(ata_dma_blacklist[i], s, len))
+                       return 1;
 
-err_out:
-       ap->ops->port_disable(ap);
+       return 0;
 }
 
-/**
- *     ata_host_set_udma -
- *     @ap:
- *
- *     LOCKING:
- */
-
-static void ata_host_set_udma(struct ata_port *ap)
+static unsigned int ata_get_mode_mask(struct ata_port *ap, int shift)
 {
        struct ata_device *master, *slave;
-       u16 mask;
-       unsigned int i, j;
-       int udma_mode = -1;
+       unsigned int mask;
 
        master = &ap->device[0];
        slave = &ap->device[1];
 
        assert (ata_dev_present(master) || ata_dev_present(slave));
-       assert ((ap->flags & ATA_FLAG_PORT_DISABLED) == 0);
-
-       DPRINTK("udma masks: host 0x%X, master 0x%X, slave 0x%X\n",
-               ap->udma_mask,
-               (!ata_dev_present(master)) ? 0xff :
-                       (master->id[ATA_ID_UDMA_MODES] & 0xff),
-               (!ata_dev_present(slave)) ? 0xff :
-                       (slave->id[ATA_ID_UDMA_MODES] & 0xff));
-
-       mask = ap->udma_mask;
-       if (ata_dev_present(master))
-               mask &= (master->id[ATA_ID_UDMA_MODES] & 0xff);
-       if (ata_dev_present(slave))
-               mask &= (slave->id[ATA_ID_UDMA_MODES] & 0xff);
-
-       i = XFER_UDMA_7;
-       while (i >= XFER_UDMA_0) {
-               j = i - XFER_UDMA_0;
-               DPRINTK("mask 0x%X i 0x%X j %u\n", mask, i, j);
-               if (mask & (1 << j)) {
-                       udma_mode = i;
-                       break;
-               }
 
-               i--;
+       if (shift == ATA_SHIFT_UDMA) {
+               mask = ap->udma_mask;
+               if (ata_dev_present(master)) {
+                       mask &= (master->id[ATA_ID_UDMA_MODES] & 0xff);
+                       if (ata_dma_blacklisted(ap, master)) {
+                               mask = 0;
+                               ata_pr_blacklisted(ap, master);
+                       }
+               }
+               if (ata_dev_present(slave)) {
+                       mask &= (slave->id[ATA_ID_UDMA_MODES] & 0xff);
+                       if (ata_dma_blacklisted(ap, slave)) {
+                               mask = 0;
+                               ata_pr_blacklisted(ap, slave);
+                       }
+               }
        }
-
-       /* require udma for host and all attached devices */
-       if (udma_mode < 0) {
-               printk(KERN_WARNING "ata%u: no UltraDMA support, ignoring\n",
-                      ap->id);
-               goto err_out;
+       else if (shift == ATA_SHIFT_MWDMA) {
+               mask = ap->mwdma_mask;
+               if (ata_dev_present(master)) {
+                       mask &= (master->id[ATA_ID_MWDMA_MODES] & 0x07);
+                       if (ata_dma_blacklisted(ap, master)) {
+                               mask = 0;
+                               ata_pr_blacklisted(ap, master);
+                       }
+               }
+               if (ata_dev_present(slave)) {
+                       mask &= (slave->id[ATA_ID_MWDMA_MODES] & 0x07);
+                       if (ata_dma_blacklisted(ap, slave)) {
+                               mask = 0;
+                               ata_pr_blacklisted(ap, slave);
+                       }
+               }
        }
-
-       for (i = 0; i < ATA_MAX_DEVICES; i++)
-               if (ata_dev_present(&ap->device[i])) {
-                       ap->device[i].udma_mode = udma_mode;
-                       if (ap->ops->set_udmamode)
-                               ap->ops->set_udmamode(ap, &ap->device[i],
-                                                     udma_mode);
+       else if (shift == ATA_SHIFT_PIO) {
+               mask = ap->pio_mask;
+               if (ata_dev_present(master)) {
+                       /* spec doesn't return explicit support for
+                        * PIO0-2, so we fake it
+                        */
+                       u16 tmp_mode = master->id[ATA_ID_PIO_MODES] & 0x03;
+                       tmp_mode <<= 3;
+                       tmp_mode |= 0x7;
+                       mask &= tmp_mode;
                }
+               if (ata_dev_present(slave)) {
+                       /* spec doesn't return explicit support for
+                        * PIO0-2, so we fake it
+                        */
+                       u16 tmp_mode = slave->id[ATA_ID_PIO_MODES] & 0x03;
+                       tmp_mode <<= 3;
+                       tmp_mode |= 0x7;
+                       mask &= tmp_mode;
+               }
+       }
+       else {
+               mask = 0xffffffff; /* shut up compiler warning */
+               BUG();
+       }
 
-       return;
-
-err_out:
-       ap->ops->port_disable(ap);
+       return mask;
 }
 
-/**
- *     ata_dev_set_xfermode - Issue SET FEATURES - XFER MODE command
- *     @ap: Port associated with device @dev
- *     @dev: Device to which command will be sent
- *
- *     LOCKING:
- */
-
-static void ata_dev_set_xfermode(struct ata_port *ap, struct ata_device *dev)
+/* find greatest bit */
+static int fgb(u32 bitmap)
 {
-       struct ata_taskfile tf;
-
-       /* set up set-features taskfile */
-       DPRINTK("set features - xfer mode\n");
-       ata_tf_init(ap, &tf, dev->devno);
-       tf.ctl |= ATA_NIEN;
-       tf.command = ATA_CMD_SET_FEATURES;
-       tf.feature = SETFEATURES_XFER;
-       tf.flags |= ATA_TFLAG_ISADDR | ATA_TFLAG_DEVICE;
-       tf.protocol = ATA_PROT_NODATA;
-       if (dev->flags & ATA_DFLAG_PIO)
-               tf.nsect = dev->pio_mode;
-       else
-               tf.nsect = dev->udma_mode;
-
-       /* do bus reset */
-       ata_tf_to_host(ap, &tf);
-
-       /* crazy ATAPI devices... */
-       if (dev->class == ATA_DEV_ATAPI)
-               msleep(150);
-
-       ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
-
-       ata_irq_on(ap); /* re-enable interrupts */
+       unsigned int i;
+       int x = -1;
 
-       ata_wait_idle(ap);
+       for (i = 0; i < 32; i++)
+               if (bitmap & (1 << i))
+                       x = i;
 
-       DPRINTK("EXIT\n");
+       return x;
 }
 
 /**
- *     ata_dev_set_udma - Set ATA device's transfer mode to Ultra DMA
- *     @ap: Port associated with device @dev
- *     @device: Device whose mode will be set
+ *     ata_choose_xfer_mode - attempt to find best transfer mode
+ *     @ap: Port for which an xfer mode will be selected
+ *     @xfer_mode_out: (output) SET FEATURES - XFER MODE code
+ *     @xfer_shift_out: (output) bit shift that selects this mode
  *
  *     LOCKING:
+ *
+ *     RETURNS:
+ *     Zero on success, negative on error.
  */
 
-static void ata_dev_set_udma(struct ata_port *ap, unsigned int device)
+static int ata_choose_xfer_mode(struct ata_port *ap,
+                               u8 *xfer_mode_out,
+                               unsigned int *xfer_shift_out)
 {
-       struct ata_device *dev = &ap->device[device];
+       unsigned int mask, shift;
+       int x, i;
 
-       if (!ata_dev_present(dev) || (ap->flags & ATA_FLAG_PORT_DISABLED))
-               return;
+       for (i = 0; i < ARRAY_SIZE(xfer_mode_classes); i++) {
+               shift = xfer_mode_classes[i].shift;
+               mask = ata_get_mode_mask(ap, shift);
 
-       ata_dev_set_xfermode(ap, dev);
+               x = fgb(mask);
+               if (x >= 0) {
+                       *xfer_mode_out = xfer_mode_classes[i].base + x;
+                       *xfer_shift_out = shift;
+                       return 0;
+               }
+       }
 
-       assert((dev->udma_mode >= XFER_UDMA_0) &&
-              (dev->udma_mode <= XFER_UDMA_7));
-       printk(KERN_INFO "ata%u: dev %u configured for %s\n",
-              ap->id, device,
-              udma_str[dev->udma_mode - XFER_UDMA_0]);
+       return -1;
 }
 
 /**
- *     ata_dev_set_pio - Set ATA device's transfer mode to PIO
+ *     ata_dev_set_xfermode - Issue SET FEATURES - XFER MODE command
  *     @ap: Port associated with device @dev
- *     @device: Device whose mode will be set
+ *     @dev: Device to which command will be sent
  *
  *     LOCKING:
  */
 
-static void ata_dev_set_pio(struct ata_port *ap, unsigned int device)
+static void ata_dev_set_xfermode(struct ata_port *ap, struct ata_device *dev)
 {
-       struct ata_device *dev = &ap->device[device];
+       DECLARE_COMPLETION(wait);
+       struct ata_queued_cmd *qc;
+       int rc;
+       unsigned long flags;
 
-       if (!ata_dev_present(dev) || (ap->flags & ATA_FLAG_PORT_DISABLED))
-               return;
+       /* set up set-features taskfile */
+       DPRINTK("set features - xfer mode\n");
 
-       /* force PIO mode */
-       dev->flags |= ATA_DFLAG_PIO;
+       qc = ata_qc_new_init(ap, dev);
+       BUG_ON(qc == NULL);
 
-       ata_dev_set_xfermode(ap, dev);
+       qc->tf.command = ATA_CMD_SET_FEATURES;
+       qc->tf.feature = SETFEATURES_XFER;
+       qc->tf.flags |= ATA_TFLAG_ISADDR | ATA_TFLAG_DEVICE;
+       qc->tf.protocol = ATA_PROT_NODATA;
+       qc->tf.nsect = dev->xfer_mode;
+
+       qc->waiting = &wait;
+       qc->complete_fn = ata_qc_complete_noop;
+
+       spin_lock_irqsave(&ap->host_set->lock, flags);
+       rc = ata_qc_issue(qc);
+       spin_unlock_irqrestore(&ap->host_set->lock, flags);
+
+       if (rc)
+               ata_port_disable(ap);
+       else
+               wait_for_completion(&wait);
 
-       assert((dev->pio_mode >= XFER_PIO_3) &&
-              (dev->pio_mode <= XFER_PIO_4));
-       printk(KERN_INFO "ata%u: dev %u configured for PIO%c\n",
-              ap->id, device,
-              dev->pio_mode == 3 ? '3' : '4');
+       DPRINTK("EXIT\n");
 }
 
 /**
@@ -1751,7 +1957,7 @@ static void ata_sg_clean(struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
        struct scatterlist *sg = qc->sg;
-       int dir = qc->pci_dma_dir;
+       int dir = qc->dma_dir;
 
        assert(qc->flags & ATA_QCFLAG_DMAMAP);
        assert(sg != NULL);
@@ -1762,9 +1968,9 @@ static void ata_sg_clean(struct ata_queued_cmd *qc)
        DPRINTK("unmapping %u sg elements\n", qc->n_elem);
 
        if (qc->flags & ATA_QCFLAG_SG)
-               pci_unmap_sg(ap->host_set->pdev, sg, qc->n_elem, dir);
+               dma_unmap_sg(ap->host_set->dev, sg, qc->n_elem, dir);
        else
-               pci_unmap_single(ap->host_set->pdev, sg_dma_address(&sg[0]),
+               dma_unmap_single(ap->host_set->dev, sg_dma_address(&sg[0]),
                                 sg_dma_len(&sg[0]), dir);
 
        qc->flags &= ~ATA_QCFLAG_DMAMAP;
@@ -1789,7 +1995,7 @@ static void ata_fill_sg(struct ata_queued_cmd *qc)
 
        idx = 0;
        for (nelem = qc->n_elem; nelem; nelem--,sg++) {
-               u32 addr, boundary;
+               u32 addr, offset;
                u32 sg_len, len;
 
                /* determine if physical DMA addr spans 64K boundary.
@@ -1800,10 +2006,10 @@ static void ata_fill_sg(struct ata_queued_cmd *qc)
                sg_len = sg_dma_len(sg);
 
                while (sg_len) {
-                       boundary = (addr & ~0xffff) + (0xffff + 1);
+                       offset = addr & 0xffff;
                        len = sg_len;
-                       if ((addr + sg_len) > boundary)
-                               len = boundary - addr;
+                       if ((offset + sg_len) > 0x10000)
+                               len = 0x10000 - offset;
 
                        ap->prd[idx].addr = cpu_to_le32(addr);
                        ap->prd[idx].flags_len = cpu_to_le32(len & 0xffff);
@@ -1818,7 +2024,24 @@ static void ata_fill_sg(struct ata_queued_cmd *qc)
        if (idx)
                ap->prd[idx - 1].flags_len |= cpu_to_le32(ATA_PRD_EOT);
 }
-
+/**
+ *     ata_check_atapi_dma - Check whether ATAPI DMA can be supported
+ *     @qc: Metadata associated with taskfile to check
+ *
+ *     LOCKING:
+ *     RETURNS: 0 when ATAPI DMA can be used
+ *               nonzero otherwise
+ */
+int ata_check_atapi_dma(struct ata_queued_cmd *qc)
+{
+       struct ata_port *ap = qc->ap;
+       int rc = 0; /* Assume ATAPI DMA is OK by default */
+
+       if (ap->ops->check_atapi_dma)
+               rc = ap->ops->check_atapi_dma(qc);
+
+       return rc;
+}
 /**
  *     ata_qc_prep - Prepare taskfile for submission
  *     @qc: Metadata associated with taskfile to be prepared
@@ -1849,8 +2072,6 @@ void ata_sg_init_one(struct ata_queued_cmd *qc, void *buf, unsigned int buflen)
        sg->page = virt_to_page(buf);
        sg->offset = (unsigned long) buf & ~PAGE_MASK;
        sg_dma_len(sg) = buflen;
-
-       WARN_ON(buflen > PAGE_SIZE);
 }
 
 void ata_sg_init(struct ata_queued_cmd *qc, struct scatterlist *sg,
@@ -1875,13 +2096,13 @@ void ata_sg_init(struct ata_queued_cmd *qc, struct scatterlist *sg,
 static int ata_sg_setup_one(struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
-       int dir = qc->pci_dma_dir;
+       int dir = qc->dma_dir;
        struct scatterlist *sg = qc->sg;
        dma_addr_t dma_address;
 
-       dma_address = pci_map_single(ap->host_set->pdev, qc->buf_virt,
+       dma_address = dma_map_single(ap->host_set->dev, qc->buf_virt,
                                     sg_dma_len(sg), dir);
-       if (pci_dma_mapping_error(dma_address))
+       if (dma_mapping_error(dma_address))
                return -1;
 
        sg_dma_address(sg) = dma_address;
@@ -1912,8 +2133,8 @@ static int ata_sg_setup(struct ata_queued_cmd *qc)
        VPRINTK("ENTER, ata%u\n", ap->id);
        assert(qc->flags & ATA_QCFLAG_SG);
 
-       dir = qc->pci_dma_dir;
-       n_elem = pci_map_sg(ap->host_set->pdev, sg, qc->n_elem, dir);
+       dir = qc->dma_dir;
+       n_elem = dma_map_sg(ap->host_set->dev, sg, qc->n_elem, dir);
        if (n_elem < 1)
                return -1;
 
@@ -2003,7 +2224,7 @@ static void ata_pio_complete (struct ata_port *ap)
        }
 
        drv_stat = ata_wait_idle(ap);
-       if (drv_stat & (ATA_BUSY | ATA_DRQ)) {
+       if (!ata_ok(drv_stat)) {
                ap->pio_task_state = PIO_ST_ERR;
                return;
        }
@@ -2018,6 +2239,176 @@ static void ata_pio_complete (struct ata_port *ap)
        ata_qc_complete(qc, drv_stat);
 }
 
+void swap_buf_le16(u16 *buf, unsigned int buf_words)
+{
+#ifdef __BIG_ENDIAN
+       unsigned int i;
+
+       for (i = 0; i < buf_words; i++)
+               buf[i] = le16_to_cpu(buf[i]);
+#endif /* __BIG_ENDIAN */
+}
+
+static void ata_mmio_data_xfer(struct ata_port *ap, unsigned char *buf,
+                              unsigned int buflen, int write_data)
+{
+       unsigned int i;
+       unsigned int words = buflen >> 1;
+       u16 *buf16 = (u16 *) buf;
+       void __iomem *mmio = (void __iomem *)ap->ioaddr.data_addr;
+
+       if (write_data) {
+               for (i = 0; i < words; i++)
+                       writew(le16_to_cpu(buf16[i]), mmio);
+       } else {
+               for (i = 0; i < words; i++)
+                       buf16[i] = cpu_to_le16(readw(mmio));
+       }
+}
+
+static void ata_pio_data_xfer(struct ata_port *ap, unsigned char *buf,
+                             unsigned int buflen, int write_data)
+{
+       unsigned int dwords = buflen >> 1;
+
+       if (write_data)
+               outsw(ap->ioaddr.data_addr, buf, dwords);
+       else
+               insw(ap->ioaddr.data_addr, buf, dwords);
+}
+
+static void ata_data_xfer(struct ata_port *ap, unsigned char *buf,
+                         unsigned int buflen, int do_write)
+{
+       if (ap->flags & ATA_FLAG_MMIO)
+               ata_mmio_data_xfer(ap, buf, buflen, do_write);
+       else
+               ata_pio_data_xfer(ap, buf, buflen, do_write);
+}
+
+static void ata_pio_sector(struct ata_queued_cmd *qc)
+{
+       int do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
+       struct scatterlist *sg = qc->sg;
+       struct ata_port *ap = qc->ap;
+       struct page *page;
+       unsigned int offset;
+       unsigned char *buf;
+
+       if (qc->cursect == (qc->nsect - 1))
+               ap->pio_task_state = PIO_ST_LAST;
+
+       page = sg[qc->cursg].page;
+       offset = sg[qc->cursg].offset + qc->cursg_ofs * ATA_SECT_SIZE;
+
+       /* get the current page and offset */
+       page = nth_page(page, (offset >> PAGE_SHIFT));
+       offset %= PAGE_SIZE;
+
+       buf = kmap(page) + offset;
+
+       qc->cursect++;
+       qc->cursg_ofs++;
+
+       if ((qc->cursg_ofs * ATA_SECT_SIZE) == sg_dma_len(&sg[qc->cursg])) {
+               qc->cursg++;
+               qc->cursg_ofs = 0;
+       }
+
+       DPRINTK("data %s\n", qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read");
+
+       /* do the actual data transfer */
+       do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
+       ata_data_xfer(ap, buf, ATA_SECT_SIZE, do_write);
+
+       kunmap(page);
+}
+
+static void __atapi_pio_bytes(struct ata_queued_cmd *qc, unsigned int bytes)
+{
+       int do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
+       struct scatterlist *sg = qc->sg;
+       struct ata_port *ap = qc->ap;
+       struct page *page;
+       unsigned char *buf;
+       unsigned int offset, count;
+
+       if (qc->curbytes == qc->nbytes - bytes)
+               ap->pio_task_state = PIO_ST_LAST;
+
+next_sg:
+       sg = &qc->sg[qc->cursg];
+
+next_page:
+       page = sg->page;
+       offset = sg->offset + qc->cursg_ofs;
+
+       /* get the current page and offset */
+       page = nth_page(page, (offset >> PAGE_SHIFT));
+       offset %= PAGE_SIZE;
+
+       count = min(sg_dma_len(sg) - qc->cursg_ofs, bytes);
+
+       /* don't cross page boundaries */
+       count = min(count, (unsigned int)PAGE_SIZE - offset);
+
+       buf = kmap(page) + offset;
+
+       bytes -= count;
+       qc->curbytes += count;
+       qc->cursg_ofs += count;
+
+       if (qc->cursg_ofs == sg_dma_len(sg)) {
+               qc->cursg++;
+               qc->cursg_ofs = 0;
+       }
+
+       DPRINTK("data %s\n", qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read");
+
+       /* do the actual data transfer */
+       ata_data_xfer(ap, buf, count, do_write);
+
+       kunmap(page);
+
+       if (bytes) {
+               if (qc->cursg_ofs < sg_dma_len(sg))
+                       goto next_page;
+               goto next_sg;
+       }
+}
+
+static void atapi_pio_bytes(struct ata_queued_cmd *qc)
+{
+       struct ata_port *ap = qc->ap;
+       struct ata_device *dev = qc->dev;
+       unsigned int ireason, bc_lo, bc_hi, bytes;
+       int i_write, do_write = (qc->tf.flags & ATA_TFLAG_WRITE) ? 1 : 0;
+
+       ap->ops->tf_read(ap, &qc->tf);
+       ireason = qc->tf.nsect;
+       bc_lo = qc->tf.lbam;
+       bc_hi = qc->tf.lbah;
+       bytes = (bc_hi << 8) | bc_lo;
+
+       /* shall be cleared to zero, indicating xfer of data */
+       if (ireason & (1 << 0))
+               goto err_out;
+
+       /* make sure transfer direction matches expected */
+       i_write = ((ireason & (1 << 1)) == 0) ? 1 : 0;
+       if (do_write != i_write)
+               goto err_out;
+
+       __atapi_pio_bytes(qc, bytes);
+
+       return;
+
+err_out:
+       printk(KERN_INFO "ata%u: dev %u: ATAPI check failed\n",
+             ap->id, dev->devno);
+       ap->pio_task_state = PIO_ST_ERR;
+}
+
 /**
  *     ata_pio_sector -
  *     @ap:
@@ -2025,12 +2416,9 @@ static void ata_pio_complete (struct ata_port *ap)
  *     LOCKING:
  */
 
-static void ata_pio_sector(struct ata_port *ap)
+static void ata_pio_block(struct ata_port *ap)
 {
        struct ata_queued_cmd *qc;
-       struct scatterlist *sg;
-       struct page *page;
-       unsigned char *buf;
        u8 status;
 
        /*
@@ -2052,45 +2440,49 @@ static void ata_pio_sector(struct ata_port *ap)
                }
        }
 
-       /* handle BSY=0, DRQ=0 as error */
-       if ((status & ATA_DRQ) == 0) {
-               ap->pio_task_state = PIO_ST_ERR;
-               return;
-       }
-
        qc = ata_qc_from_tag(ap, ap->active_tag);
        assert(qc != NULL);
 
-       sg = qc->sg;
+       if (is_atapi_taskfile(&qc->tf)) {
+               /* no more data to transfer or unsupported ATAPI command */
+               if ((status & ATA_DRQ) == 0) {
+                       ap->pio_task_state = PIO_ST_IDLE;
 
-       if (qc->cursect == (qc->nsect - 1))
-               ap->pio_task_state = PIO_ST_LAST;
-
-       page = sg[qc->cursg].page;
-       buf = kmap(page) +
-             sg[qc->cursg].offset + (qc->cursg_ofs * ATA_SECT_SIZE);
+                       ata_irq_on(ap);
 
-       qc->cursect++;
-       qc->cursg_ofs++;
+                       ata_qc_complete(qc, status);
+                       return;
+               }
 
-       if (qc->flags & ATA_QCFLAG_SG)
-               if ((qc->cursg_ofs * ATA_SECT_SIZE) == sg_dma_len(&sg[qc->cursg])) {
-                       qc->cursg++;
-                       qc->cursg_ofs = 0;
+               atapi_pio_bytes(qc);
+       } else {
+               /* handle BSY=0, DRQ=0 as error */
+               if ((status & ATA_DRQ) == 0) {
+                       ap->pio_task_state = PIO_ST_ERR;
+                       return;
                }
 
-       DPRINTK("data %s, drv_stat 0x%X\n",
-               qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read",
-               status);
+               ata_pio_sector(qc);
+       }
+}
 
-       /* do the actual data transfer */
-       /* FIXME: mmio-ize */
-       if (qc->tf.flags & ATA_TFLAG_WRITE)
-               outsl(ap->ioaddr.data_addr, buf, ATA_SECT_DWORDS);
-       else
-               insl(ap->ioaddr.data_addr, buf, ATA_SECT_DWORDS);
+static void ata_pio_error(struct ata_port *ap)
+{
+       struct ata_queued_cmd *qc;
+       u8 drv_stat;
 
-       kunmap(page);
+       qc = ata_qc_from_tag(ap, ap->active_tag);
+       assert(qc != NULL);
+
+       drv_stat = ata_chk_status(ap);
+       printk(KERN_WARNING "ata%u: PIO error, drv_stat 0x%x\n",
+              ap->id, drv_stat);
+
+       ap->pio_task_state = PIO_ST_IDLE;
+
+       ata_irq_on(ap);
+
+       ata_qc_complete(qc, drv_stat | ATA_ERR);
 }
 
 static void ata_pio_task(void *_data)
@@ -2099,8 +2491,11 @@ static void ata_pio_task(void *_data)
        unsigned long timeout = 0;
 
        switch (ap->pio_task_state) {
+       case PIO_ST_IDLE:
+               return;
+
        case PIO_ST:
-               ata_pio_sector(ap);
+               ata_pio_block(ap);
                break;
 
        case PIO_ST_LAST:
@@ -2113,27 +2508,62 @@ static void ata_pio_task(void *_data)
                break;
 
        case PIO_ST_TMOUT:
-               printk(KERN_ERR "ata%d: FIXME: PIO_ST_TMOUT\n", /* FIXME */
-                      ap->id);
-               timeout = 11 * HZ;
-               break;
-
        case PIO_ST_ERR:
-               printk(KERN_ERR "ata%d: FIXME: PIO_ST_ERR\n", /* FIXME */
-                      ap->id);
-               timeout = 11 * HZ;
-               break;
+               ata_pio_error(ap);
+               return;
        }
 
-       if ((ap->pio_task_state != PIO_ST_IDLE) &&
-           (ap->pio_task_state != PIO_ST_TMOUT) &&
-           (ap->pio_task_state != PIO_ST_ERR)) {
-               if (timeout)
-                       queue_delayed_work(ata_wq, &ap->pio_task,
-                                          timeout);
-               else
-                       queue_work(ata_wq, &ap->pio_task);
-       }
+       if (timeout)
+               queue_delayed_work(ata_wq, &ap->pio_task,
+                                  timeout);
+       else
+               queue_work(ata_wq, &ap->pio_task);
+}
+
+static void atapi_request_sense(struct ata_port *ap, struct ata_device *dev,
+                               struct scsi_cmnd *cmd)
+{
+       DECLARE_COMPLETION(wait);
+       struct ata_queued_cmd *qc;
+       unsigned long flags;
+       int rc;
+
+       DPRINTK("ATAPI request sense\n");
+
+       qc = ata_qc_new_init(ap, dev);
+       BUG_ON(qc == NULL);
+
+       /* FIXME: is this needed? */
+       memset(cmd->sense_buffer, 0, sizeof(cmd->sense_buffer));
+
+       ata_sg_init_one(qc, cmd->sense_buffer, sizeof(cmd->sense_buffer));
+       qc->dma_dir = DMA_FROM_DEVICE;
+
+       memset(&qc->cdb, 0, sizeof(ap->cdb_len));
+       qc->cdb[0] = REQUEST_SENSE;
+       qc->cdb[4] = SCSI_SENSE_BUFFERSIZE;
+
+       qc->tf.flags |= ATA_TFLAG_ISADDR | ATA_TFLAG_DEVICE;
+       qc->tf.command = ATA_CMD_PACKET;
+
+       qc->tf.protocol = ATA_PROT_ATAPI;
+       qc->tf.lbam = (8 * 1024) & 0xff;
+       qc->tf.lbah = (8 * 1024) >> 8;
+       qc->nbytes = SCSI_SENSE_BUFFERSIZE;
+
+       qc->waiting = &wait;
+       qc->complete_fn = ata_qc_complete_noop;
+
+       spin_lock_irqsave(&ap->host_set->lock, flags);
+       rc = ata_qc_issue(qc);
+       spin_unlock_irqrestore(&ap->host_set->lock, flags);
+
+       if (rc)
+               ata_port_disable(ap);
+       else
+               wait_for_completion(&wait);
+
+       DPRINTK("EXIT\n");
 }
 
 /**
@@ -2157,10 +2587,29 @@ static void ata_pio_task(void *_data)
 static void ata_qc_timeout(struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
+       struct ata_device *dev = qc->dev;
        u8 host_stat = 0, drv_stat;
 
        DPRINTK("ENTER\n");
 
+       /* FIXME: doesn't this conflict with timeout handling? */
+       if (qc->dev->class == ATA_DEV_ATAPI && qc->scsicmd) {
+               struct scsi_cmnd *cmd = qc->scsicmd;
+
+               if (!scsi_eh_eflags_chk(cmd, SCSI_EH_CANCEL_CMD)) {
+
+                       /* finish completing original command */
+                       __ata_qc_complete(qc);
+
+                       atapi_request_sense(ap, dev, cmd);
+
+                       cmd->result = (CHECK_CONDITION << 1) | (DID_OK << 16);
+                       scsi_finish_command(cmd);
+
+                       goto out;
+               }
+       }
+
        /* hack alert!  We cannot use the supplied completion
         * function from inside the ->eh_strategy_handler() thread.
         * libata is the only user of ->eh_strategy_handler() in
@@ -2173,20 +2622,19 @@ static void ata_qc_timeout(struct ata_queued_cmd *qc)
 
        case ATA_PROT_DMA:
        case ATA_PROT_ATAPI_DMA:
-               host_stat = ata_bmdma_status(ap);
+               host_stat = ap->ops->bmdma_status(ap);
 
                /* before we do anything else, clear DMA-Start bit */
-               ata_bmdma_stop(ap);
+               ap->ops->bmdma_stop(ap);
 
                /* fall through */
 
-       case ATA_PROT_NODATA:
        default:
                ata_altstatus(ap);
                drv_stat = ata_chk_status(ap);
 
                /* ack bmdma irq events */
-               ata_bmdma_ack_irq(ap);
+               ap->ops->irq_clear(ap);
 
                printk(KERN_ERR "ata%u: command 0x%x timeout, stat 0x%x host_stat 0x%x\n",
                       ap->id, qc->tf.command, drv_stat, host_stat);
@@ -2195,7 +2643,7 @@ static void ata_qc_timeout(struct ata_queued_cmd *qc)
                ata_qc_complete(qc, drv_stat);
                break;
        }
-
+out:
        DPRINTK("EXIT\n");
 }
 
@@ -2284,11 +2732,10 @@ struct ata_queued_cmd *ata_qc_new_init(struct ata_port *ap,
                qc->dev = dev;
                qc->cursect = qc->cursg = qc->cursg_ofs = 0;
                qc->nsect = 0;
+               qc->nbytes = qc->curbytes = 0;
 
                ata_tf_init(ap, &qc->tf, dev->devno);
 
-               if (likely((dev->flags & ATA_DFLAG_PIO) == 0))
-                       qc->flags |= ATA_QCFLAG_DMA;
                if (dev->flags & ATA_DFLAG_LBA48)
                        qc->tf.flags |= ATA_TFLAG_LBA48;
        }
@@ -2296,6 +2743,53 @@ struct ata_queued_cmd *ata_qc_new_init(struct ata_port *ap,
        return qc;
 }
 
+static int ata_qc_complete_noop(struct ata_queued_cmd *qc, u8 drv_stat)
+{
+       return 0;
+}
+
+static void __ata_qc_complete(struct ata_queued_cmd *qc)
+{
+       struct ata_port *ap = qc->ap;
+       unsigned int tag, do_clear = 0;
+
+       qc->flags = 0;
+       tag = qc->tag;
+       if (likely(ata_tag_valid(tag))) {
+               if (tag == ap->active_tag)
+                       ap->active_tag = ATA_TAG_POISON;
+               qc->tag = ATA_TAG_POISON;
+               do_clear = 1;
+       }
+
+       if (qc->waiting) {
+               struct completion *waiting = qc->waiting;
+               qc->waiting = NULL;
+               complete(waiting);
+       }
+
+       if (likely(do_clear))
+               clear_bit(tag, &ap->qactive);
+}
+
+/**
+ *     ata_qc_free - free unused ata_queued_cmd
+ *     @qc: Command to complete
+ *
+ *     Designed to free unused ata_queued_cmd object
+ *     in case something prevents using it.
+ *
+ *     LOCKING:
+ *
+ */
+void ata_qc_free(struct ata_queued_cmd *qc)
+{
+       assert(qc != NULL);     /* ata_qc_from_tag _might_ return NULL */
+       assert(qc->waiting == NULL);    /* nothing should be waiting */
+
+       __ata_qc_complete(qc);
+}
+
 /**
  *     ata_qc_complete - Complete an active ATA command
  *     @qc: Command to complete
@@ -2307,8 +2801,6 @@ struct ata_queued_cmd *ata_qc_new_init(struct ata_port *ap,
 
 void ata_qc_complete(struct ata_queued_cmd *qc, u8 drv_stat)
 {
-       struct ata_port *ap = qc->ap;
-       unsigned int tag, do_clear = 0;
        int rc;
 
        assert(qc != NULL);     /* ata_qc_from_tag _might_ return NULL */
@@ -2326,20 +2818,33 @@ void ata_qc_complete(struct ata_queued_cmd *qc, u8 drv_stat)
        if (rc != 0)
                return;
 
-       qc->flags = 0;
-       tag = qc->tag;
-       if (likely(ata_tag_valid(tag))) {
-               if (tag == ap->active_tag)
-                       ap->active_tag = ATA_TAG_POISON;
-               qc->tag = ATA_TAG_POISON;
-               do_clear = 1;
-       }
+       __ata_qc_complete(qc);
 
-       if (qc->waiting)
-               complete(qc->waiting);
+       VPRINTK("EXIT\n");
+}
 
-       if (likely(do_clear))
-               clear_bit(tag, &ap->qactive);
+static inline int ata_should_dma_map(struct ata_queued_cmd *qc)
+{
+       struct ata_port *ap = qc->ap;
+
+       switch (qc->tf.protocol) {
+       case ATA_PROT_DMA:
+       case ATA_PROT_ATAPI_DMA:
+               return 1;
+
+       case ATA_PROT_ATAPI:
+       case ATA_PROT_PIO:
+       case ATA_PROT_PIO_MULT:
+               if (ap->flags & ATA_FLAG_PIO_DMA)
+                       return 1;
+
+               /* fall through */
+
+       default:
+               return 0;
+       }
+
+       /* never reached */
 }
 
 /**
@@ -2362,12 +2867,16 @@ int ata_qc_issue(struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
 
-       if (qc->flags & ATA_QCFLAG_SG) {
-               if (ata_sg_setup(qc))
-                       goto err_out;
-       } else if (qc->flags & ATA_QCFLAG_SINGLE) {
-               if (ata_sg_setup_one(qc))
-                       goto err_out;
+       if (ata_should_dma_map(qc)) {
+               if (qc->flags & ATA_QCFLAG_SG) {
+                       if (ata_sg_setup(qc))
+                               goto err_out;
+               } else if (qc->flags & ATA_QCFLAG_SINGLE) {
+                       if (ata_sg_setup_one(qc))
+                               goto err_out;
+               }
+       } else {
+               qc->flags &= ~ATA_QCFLAG_DMAMAP;
        }
 
        ap->ops->qc_prep(qc);
@@ -2422,6 +2931,12 @@ int ata_qc_issue_prot(struct ata_queued_cmd *qc)
                break;
 
        case ATA_PROT_ATAPI:
+               ata_qc_set_polling(qc);
+               ata_tf_to_host_nolock(ap, &qc->tf);
+               queue_work(ata_wq, &ap->packet_task);
+               break;
+
+       case ATA_PROT_ATAPI_NODATA:
                ata_tf_to_host_nolock(ap, &qc->tf);
                queue_work(ata_wq, &ap->packet_task);
                break;
@@ -2441,19 +2956,19 @@ int ata_qc_issue_prot(struct ata_queued_cmd *qc)
 }
 
 /**
- *     ata_bmdma_setup_mmio - Set up PCI IDE BMDMA transaction (MMIO)
+ *     ata_bmdma_setup - Set up PCI IDE BMDMA transaction
  *     @qc: Info associated with this ATA transaction.
  *
  *     LOCKING:
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_bmdma_setup_mmio (struct ata_queued_cmd *qc)
+static void ata_bmdma_setup_mmio (struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
        unsigned int rw = (qc->tf.flags & ATA_TFLAG_WRITE);
        u8 dmactl;
-       void *mmio = (void *) ap->ioaddr.bmdma_addr;
+       void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
 
        /* load PRD table addr. */
        mb();   /* make sure PRD table writes are visible to controller */
@@ -2471,17 +2986,17 @@ void ata_bmdma_setup_mmio (struct ata_queued_cmd *qc)
 }
 
 /**
- *     ata_bmdma_start_mmio - Start a PCI IDE BMDMA transaction (MMIO)
+ *     ata_bmdma_start - Start a PCI IDE BMDMA transaction
  *     @qc: Info associated with this ATA transaction.
  *
  *     LOCKING:
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_bmdma_start_mmio (struct ata_queued_cmd *qc)
+static void ata_bmdma_start_mmio (struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
-       void *mmio = (void *) ap->ioaddr.bmdma_addr;
+       void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
        u8 dmactl;
 
        /* start host DMA transaction */
@@ -2509,7 +3024,7 @@ void ata_bmdma_start_mmio (struct ata_queued_cmd *qc)
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_bmdma_setup_pio (struct ata_queued_cmd *qc)
+static void ata_bmdma_setup_pio (struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
        unsigned int rw = (qc->tf.flags & ATA_TFLAG_WRITE);
@@ -2537,7 +3052,7 @@ void ata_bmdma_setup_pio (struct ata_queued_cmd *qc)
  *     spin_lock_irqsave(host_set lock)
  */
 
-void ata_bmdma_start_pio (struct ata_queued_cmd *qc)
+static void ata_bmdma_start_pio (struct ata_queued_cmd *qc)
 {
        struct ata_port *ap = qc->ap;
        u8 dmactl;
@@ -2548,9 +3063,61 @@ void ata_bmdma_start_pio (struct ata_queued_cmd *qc)
             ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
 }
 
+void ata_bmdma_start(struct ata_queued_cmd *qc)
+{
+       if (qc->ap->flags & ATA_FLAG_MMIO)
+               ata_bmdma_start_mmio(qc);
+       else
+               ata_bmdma_start_pio(qc);
+}
+
+void ata_bmdma_setup(struct ata_queued_cmd *qc)
+{
+       if (qc->ap->flags & ATA_FLAG_MMIO)
+               ata_bmdma_setup_mmio(qc);
+       else
+               ata_bmdma_setup_pio(qc);
+}
+
 void ata_bmdma_irq_clear(struct ata_port *ap)
 {
-       ata_bmdma_ack_irq(ap);
+    if (ap->flags & ATA_FLAG_MMIO) {
+        void __iomem *mmio = ((void __iomem *) ap->ioaddr.bmdma_addr) + ATA_DMA_STATUS;
+        writeb(readb(mmio), mmio);
+    } else {
+        unsigned long addr = ap->ioaddr.bmdma_addr + ATA_DMA_STATUS;
+        outb(inb(addr), addr);
+    }
+
+}
+
+u8 ata_bmdma_status(struct ata_port *ap)
+{
+       u8 host_stat;
+       if (ap->flags & ATA_FLAG_MMIO) {
+               void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
+               host_stat = readb(mmio + ATA_DMA_STATUS);
+       } else
+       host_stat = inb(ap->ioaddr.bmdma_addr + ATA_DMA_STATUS);
+       return host_stat;
+}
+
+void ata_bmdma_stop(struct ata_port *ap)
+{
+       if (ap->flags & ATA_FLAG_MMIO) {
+               void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
+
+               /* clear start/stop bit */
+               writeb(readb(mmio + ATA_DMA_CMD) & ~ATA_DMA_START,
+                       mmio + ATA_DMA_CMD);
+       } else {
+               /* clear start/stop bit */
+               outb(inb(ap->ioaddr.bmdma_addr + ATA_DMA_CMD) & ~ATA_DMA_START,
+                       ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
+       }
+
+       /* one-PIO-cycle guaranteed wait, per spec, for HDMA1:0 transition */
+       ata_altstatus(ap);        /* dummy read */
 }
 
 /**
@@ -2580,18 +3147,19 @@ inline unsigned int ata_host_intr (struct ata_port *ap,
        case ATA_PROT_ATAPI_DMA:
        case ATA_PROT_ATAPI:
                /* check status of DMA engine */
-               host_stat = ata_bmdma_status(ap);
-               VPRINTK("BUS_DMA (host_stat 0x%X)\n", host_stat);
+               host_stat = ap->ops->bmdma_status(ap);
+               VPRINTK("ata%u: host_stat 0x%X\n", ap->id, host_stat);
 
                /* if it's not our irq... */
                if (!(host_stat & ATA_DMA_INTR))
                        goto idle_irq;
 
                /* before we do anything else, clear DMA-Start bit */
-               ata_bmdma_stop(ap);
+               ap->ops->bmdma_stop(ap);
 
                /* fall through */
 
+       case ATA_PROT_ATAPI_NODATA:
        case ATA_PROT_NODATA:
                /* check altstatus */
                status = ata_altstatus(ap);
@@ -2602,10 +3170,11 @@ inline unsigned int ata_host_intr (struct ata_port *ap,
                status = ata_chk_status(ap);
                if (unlikely(status & ATA_BUSY))
                        goto idle_irq;
-               DPRINTK("BUS_NODATA (dev_stat 0x%X)\n", status);
+               DPRINTK("ata%u: protocol %d (dev_stat 0x%X)\n",
+                       ap->id, qc->tf.protocol, status);
 
                /* ack bmdma irq events */
-               ata_bmdma_ack_irq(ap);
+               ap->ops->irq_clear(ap);
 
                /* complete taskfile transaction */
                ata_qc_complete(qc, status);
@@ -2701,21 +3270,20 @@ static void atapi_packet_task(void *_data)
 
        /* make sure DRQ is set */
        status = ata_chk_status(ap);
-       if ((status & ATA_DRQ) == 0)
+       if ((status & (ATA_BUSY | ATA_DRQ)) != ATA_DRQ)
                goto err_out;
 
        /* send SCSI cdb */
-       /* FIXME: mmio-ize */
        DPRINTK("send cdb\n");
-       outsl(ap->ioaddr.data_addr,
-             qc->scsicmd->cmnd, ap->host->max_cmd_len / 4);
+       assert(ap->cdb_len >= 12);
+       ata_data_xfer(ap, qc->cdb, ap->cdb_len, 1);
 
        /* if we are DMA'ing, irq handler takes over from here */
        if (qc->tf.protocol == ATA_PROT_ATAPI_DMA)
                ap->ops->bmdma_start(qc);           /* initiate bmdma */
 
        /* non-data commands are also handled via irq */
-       else if (qc->scsicmd->sc_data_direction == SCSI_DATA_NONE) {
+       else if (qc->tf.protocol == ATA_PROT_ATAPI_NODATA) {
                /* do nothing */
        }
 
@@ -2733,9 +3301,9 @@ err_out:
 
 int ata_port_start (struct ata_port *ap)
 {
-       struct pci_dev *pdev = ap->host_set->pdev;
+       struct device *dev = ap->host_set->dev;
 
-       ap->prd = pci_alloc_consistent(pdev, ATA_PRD_TBL_SZ, &ap->prd_dma);
+       ap->prd = dma_alloc_coherent(dev, ATA_PRD_TBL_SZ, &ap->prd_dma, GFP_KERNEL);
        if (!ap->prd)
                return -ENOMEM;
 
@@ -2746,9 +3314,9 @@ int ata_port_start (struct ata_port *ap)
 
 void ata_port_stop (struct ata_port *ap)
 {
-       struct pci_dev *pdev = ap->host_set->pdev;
+       struct device *dev = ap->host_set->dev;
 
-       pci_free_consistent(pdev, ATA_PRD_TBL_SZ, ap->prd, ap->prd_dma);
+       dma_free_coherent(dev, ATA_PRD_TBL_SZ, ap->prd, ap->prd_dma);
 }
 
 /**
@@ -2794,7 +3362,7 @@ static void ata_host_init(struct ata_port *ap, struct Scsi_Host *host,
        host->max_channel = 1;
        host->unique_id = ata_unique_id++;
        host->max_cmd_len = 12;
-       scsi_set_device(host, &ent->pdev->dev);
+       scsi_set_device(host, ent->dev);
        scsi_assign_lock(host, &host_set->lock);
 
        ap->flags = ATA_FLAG_PORT_DISABLED;
@@ -2803,12 +3371,14 @@ static void ata_host_init(struct ata_port *ap, struct Scsi_Host *host,
        ap->ctl = ATA_DEVCTL_OBS;
        ap->host_set = host_set;
        ap->port_no = port_no;
+       ap->hard_port_no =
+               ent->legacy_mode ? ent->hard_port_no : port_no;
        ap->pio_mask = ent->pio_mask;
+       ap->mwdma_mask = ent->mwdma_mask;
        ap->udma_mask = ent->udma_mask;
        ap->flags |= ent->host_flags;
        ap->ops = ent->port_ops;
        ap->cbl = ATA_CBL_NONE;
-       ap->device[0].flags = ATA_DFLAG_MASTER;
        ap->active_tag = ATA_TAG_POISON;
        ap->last_ctl = 0xFF;
 
@@ -2879,7 +3449,7 @@ err_out:
 int ata_device_add(struct ata_probe_ent *ent)
 {
        unsigned int count = 0, i;
-       struct pci_dev *pdev = ent->pdev;
+       struct device *dev = ent->dev;
        struct ata_host_set *host_set;
 
        DPRINTK("ENTER\n");
@@ -2891,7 +3461,7 @@ int ata_device_add(struct ata_probe_ent *ent)
        memset(host_set, 0, sizeof(struct ata_host_set) + (ent->n_ports * sizeof(void *)));
        spin_lock_init(&host_set->lock);
 
-       host_set->pdev = pdev;
+       host_set->dev = dev;
        host_set->n_ports = ent->n_ports;
        host_set->irq = ent->irq;
        host_set->mmio_base = ent->mmio_base;
@@ -2901,19 +3471,23 @@ int ata_device_add(struct ata_probe_ent *ent)
        /* register each port bound to this device */
        for (i = 0; i < ent->n_ports; i++) {
                struct ata_port *ap;
+               unsigned long xfer_mode_mask;
 
                ap = ata_host_add(ent, host_set, i);
                if (!ap)
                        goto err_out;
 
                host_set->ports[i] = ap;
+               xfer_mode_mask =(ap->udma_mask << ATA_SHIFT_UDMA) |
+                               (ap->mwdma_mask << ATA_SHIFT_MWDMA) |
+                               (ap->pio_mask << ATA_SHIFT_PIO);
 
                /* print per-port info to dmesg */
                printk(KERN_INFO "ata%u: %cATA max %s cmd 0x%lX ctl 0x%lX "
                                 "bmdma 0x%lX irq %lu\n",
                        ap->id,
                        ap->flags & ATA_FLAG_SATA ? 'S' : 'P',
-                       ata_udma_string(ent->udma_mask),
+                       ata_mode_string(xfer_mode_mask),
                        ap->ioaddr.cmd_addr,
                        ap->ioaddr.ctl_addr,
                        ap->ioaddr.bmdma_addr,
@@ -2955,7 +3529,7 @@ int ata_device_add(struct ata_probe_ent *ent)
                         */
                }
 
-               rc = scsi_add_host(ap->host, &pdev->dev);
+               rc = scsi_add_host(ap->host, dev);
                if (rc) {
                        printk(KERN_ERR "ata%u: scsi_add_host failed\n",
                               ap->id);
@@ -2975,7 +3549,7 @@ int ata_device_add(struct ata_probe_ent *ent)
                scsi_scan_host(ap->host);
        }
 
-       pci_set_drvdata(pdev, host_set);
+       dev_set_drvdata(dev, host_set);
 
        VPRINTK("EXIT, returning %u\n", ent->n_ports);
        return ent->n_ports; /* success */
@@ -3035,6 +3609,108 @@ void ata_std_ports(struct ata_ioports *ioaddr)
        ioaddr->command_addr = ioaddr->cmd_addr + ATA_REG_CMD;
 }
 
+static struct ata_probe_ent *
+ata_probe_ent_alloc(struct device *dev, struct ata_port_info *port)
+{
+       struct ata_probe_ent *probe_ent;
+
+       probe_ent = kmalloc(sizeof(*probe_ent), GFP_KERNEL);
+       if (!probe_ent) {
+               printk(KERN_ERR DRV_NAME "(%s): out of memory\n",
+                      kobject_name(&(dev->kobj)));
+               return NULL;
+       }
+
+       memset(probe_ent, 0, sizeof(*probe_ent));
+
+       INIT_LIST_HEAD(&probe_ent->node);
+       probe_ent->dev = dev;
+
+       probe_ent->sht = port->sht;
+       probe_ent->host_flags = port->host_flags;
+       probe_ent->pio_mask = port->pio_mask;
+       probe_ent->mwdma_mask = port->mwdma_mask;
+       probe_ent->udma_mask = port->udma_mask;
+       probe_ent->port_ops = port->port_ops;
+
+       return probe_ent;
+}
+
+#ifdef CONFIG_PCI
+struct ata_probe_ent *
+ata_pci_init_native_mode(struct pci_dev *pdev, struct ata_port_info **port)
+{
+       struct ata_probe_ent *probe_ent =
+               ata_probe_ent_alloc(pci_dev_to_dev(pdev), port[0]);
+       if (!probe_ent)
+               return NULL;
+
+       probe_ent->n_ports = 2;
+       probe_ent->irq = pdev->irq;
+       probe_ent->irq_flags = SA_SHIRQ;
+
+       probe_ent->port[0].cmd_addr = pci_resource_start(pdev, 0);
+       probe_ent->port[0].altstatus_addr =
+       probe_ent->port[0].ctl_addr =
+               pci_resource_start(pdev, 1) | ATA_PCI_CTL_OFS;
+       probe_ent->port[0].bmdma_addr = pci_resource_start(pdev, 4);
+
+       probe_ent->port[1].cmd_addr = pci_resource_start(pdev, 2);
+       probe_ent->port[1].altstatus_addr =
+       probe_ent->port[1].ctl_addr =
+               pci_resource_start(pdev, 3) | ATA_PCI_CTL_OFS;
+       probe_ent->port[1].bmdma_addr = pci_resource_start(pdev, 4) + 8;
+
+       ata_std_ports(&probe_ent->port[0]);
+       ata_std_ports(&probe_ent->port[1]);
+
+       return probe_ent;
+}
+
+static struct ata_probe_ent *
+ata_pci_init_legacy_mode(struct pci_dev *pdev, struct ata_port_info **port,
+    struct ata_probe_ent **ppe2)
+{
+       struct ata_probe_ent *probe_ent, *probe_ent2;
+
+       probe_ent = ata_probe_ent_alloc(pci_dev_to_dev(pdev), port[0]);
+       if (!probe_ent)
+               return NULL;
+       probe_ent2 = ata_probe_ent_alloc(pci_dev_to_dev(pdev), port[1]);
+       if (!probe_ent2) {
+               kfree(probe_ent);
+               return NULL;
+       }
+
+       probe_ent->n_ports = 1;
+       probe_ent->irq = 14;
+
+       probe_ent->hard_port_no = 0;
+       probe_ent->legacy_mode = 1;
+
+       probe_ent2->n_ports = 1;
+       probe_ent2->irq = 15;
+
+       probe_ent2->hard_port_no = 1;
+       probe_ent2->legacy_mode = 1;
+
+       probe_ent->port[0].cmd_addr = 0x1f0;
+       probe_ent->port[0].altstatus_addr =
+       probe_ent->port[0].ctl_addr = 0x3f6;
+       probe_ent->port[0].bmdma_addr = pci_resource_start(pdev, 4);
+
+       probe_ent2->port[0].cmd_addr = 0x170;
+       probe_ent2->port[0].altstatus_addr =
+       probe_ent2->port[0].ctl_addr = 0x376;
+       probe_ent2->port[0].bmdma_addr = pci_resource_start(pdev, 4)+8;
+
+       ata_std_ports(&probe_ent->port[0]);
+       ata_std_ports(&probe_ent2->port[0]);
+
+       *ppe2 = probe_ent2;
+       return probe_ent;
+}
+
 /**
  *     ata_pci_init_one - Initialize/register PCI IDE host controller
  *     @pdev: Controller to be initialized
@@ -3052,20 +3728,22 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                      unsigned int n_ports)
 {
        struct ata_probe_ent *probe_ent, *probe_ent2 = NULL;
-       struct ata_port_info *port0, *port1;
+       struct ata_port_info *port[2];
        u8 tmp8, mask;
        unsigned int legacy_mode = 0;
+       int disable_dev_on_err = 1;
        int rc;
 
        DPRINTK("ENTER\n");
 
-       port0 = port_info[0];
+       port[0] = port_info[0];
        if (n_ports > 1)
-               port1 = port_info[1];
+               port[1] = port_info[1];
        else
-               port1 = port0;
+               port[1] = port[0];
 
-       if ((port0->host_flags & ATA_FLAG_NO_LEGACY) == 0) {
+       if ((port[0]->host_flags & ATA_FLAG_NO_LEGACY) == 0
+           && (pdev->class >> 8) == PCI_CLASS_STORAGE_IDE) {
                /* TODO: support transitioning to native mode? */
                pci_read_config_byte(pdev, PCI_CLASS_PROG, &tmp8);
                mask = (1 << 2) | (1 << 0);
@@ -3084,8 +3762,10 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                return rc;
 
        rc = pci_request_regions(pdev, DRV_NAME);
-       if (rc)
+       if (rc) {
+               disable_dev_on_err = 0;
                goto err_out;
+       }
 
        if (legacy_mode) {
                if (!request_region(0x1f0, 8, "libata")) {
@@ -3095,8 +3775,10 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                        conflict = ____request_resource(&ioport_resource, &res);
                        if (!strcmp(conflict->name, "libata"))
                                legacy_mode |= (1 << 0);
-                       else
+                       else {
+                               disable_dev_on_err = 0;
                                printk(KERN_WARNING "ata: 0x1f0 IDE port busy\n");
+                       }
                } else
                        legacy_mode |= (1 << 0);
 
@@ -3107,8 +3789,10 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                        conflict = ____request_resource(&ioport_resource, &res);
                        if (!strcmp(conflict->name, "libata"))
                                legacy_mode |= (1 << 1);
-                       else
+                       else {
+                               disable_dev_on_err = 0;
                                printk(KERN_WARNING "ata: 0x170 IDE port busy\n");
+                       }
                } else
                        legacy_mode |= (1 << 1);
        }
@@ -3126,75 +3810,15 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
        if (rc)
                goto err_out_regions;
 
-       probe_ent = kmalloc(sizeof(*probe_ent), GFP_KERNEL);
+       if (legacy_mode) {
+               probe_ent = ata_pci_init_legacy_mode(pdev, port, &probe_ent2);
+       } else
+               probe_ent = ata_pci_init_native_mode(pdev, port);
        if (!probe_ent) {
                rc = -ENOMEM;
                goto err_out_regions;
        }
 
-       memset(probe_ent, 0, sizeof(*probe_ent));
-       probe_ent->pdev = pdev;
-       INIT_LIST_HEAD(&probe_ent->node);
-
-       if (legacy_mode) {
-               probe_ent2 = kmalloc(sizeof(*probe_ent), GFP_KERNEL);
-               if (!probe_ent2) {
-                       rc = -ENOMEM;
-                       goto err_out_free_ent;
-               }
-
-               memset(probe_ent2, 0, sizeof(*probe_ent));
-               probe_ent2->pdev = pdev;
-               INIT_LIST_HEAD(&probe_ent2->node);
-       }
-
-       probe_ent->port[0].bmdma_addr = pci_resource_start(pdev, 4);
-       probe_ent->sht = port0->sht;
-       probe_ent->host_flags = port0->host_flags;
-       probe_ent->pio_mask = port0->pio_mask;
-       probe_ent->udma_mask = port0->udma_mask;
-       probe_ent->port_ops = port0->port_ops;
-
-       if (legacy_mode) {
-               probe_ent->port[0].cmd_addr = 0x1f0;
-               probe_ent->port[0].altstatus_addr =
-               probe_ent->port[0].ctl_addr = 0x3f6;
-               probe_ent->n_ports = 1;
-               probe_ent->irq = 14;
-               ata_std_ports(&probe_ent->port[0]);
-
-               probe_ent2->port[0].cmd_addr = 0x170;
-               probe_ent2->port[0].altstatus_addr =
-               probe_ent2->port[0].ctl_addr = 0x376;
-               probe_ent2->port[0].bmdma_addr = pci_resource_start(pdev, 4)+8;
-               probe_ent2->n_ports = 1;
-               probe_ent2->irq = 15;
-               ata_std_ports(&probe_ent2->port[0]);
-
-               probe_ent2->sht = port1->sht;
-               probe_ent2->host_flags = port1->host_flags;
-               probe_ent2->pio_mask = port1->pio_mask;
-               probe_ent2->udma_mask = port1->udma_mask;
-               probe_ent2->port_ops = port1->port_ops;
-       } else {
-               probe_ent->port[0].cmd_addr = pci_resource_start(pdev, 0);
-               ata_std_ports(&probe_ent->port[0]);
-               probe_ent->port[0].altstatus_addr =
-               probe_ent->port[0].ctl_addr =
-                       pci_resource_start(pdev, 1) | ATA_PCI_CTL_OFS;
-
-               probe_ent->port[1].cmd_addr = pci_resource_start(pdev, 2);
-               ata_std_ports(&probe_ent->port[1]);
-               probe_ent->port[1].altstatus_addr =
-               probe_ent->port[1].ctl_addr =
-                       pci_resource_start(pdev, 3) | ATA_PCI_CTL_OFS;
-               probe_ent->port[1].bmdma_addr = pci_resource_start(pdev, 4) + 8;
-
-               probe_ent->n_ports = 2;
-               probe_ent->irq = pdev->irq;
-               probe_ent->irq_flags = SA_SHIRQ;
-       }
-
        pci_set_master(pdev);
 
        /* FIXME: check ata_device_add return */
@@ -3203,17 +3827,14 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                        ata_device_add(probe_ent);
                if (legacy_mode & (1 << 1))
                        ata_device_add(probe_ent2);
-               kfree(probe_ent2);
-       } else {
+       } else
                ata_device_add(probe_ent);
-               assert(probe_ent2 == NULL);
-       }
+
        kfree(probe_ent);
+       kfree(probe_ent2);
 
        return 0;
 
-err_out_free_ent:
-       kfree(probe_ent);
 err_out_regions:
        if (legacy_mode & (1 << 0))
                release_region(0x1f0, 8);
@@ -3221,7 +3842,8 @@ err_out_regions:
                release_region(0x170, 8);
        pci_release_regions(pdev);
 err_out:
-       pci_disable_device(pdev);
+       if (disable_dev_on_err)
+               pci_disable_device(pdev);
        return rc;
 }
 
@@ -3241,7 +3863,8 @@ err_out:
 
 void ata_pci_remove_one (struct pci_dev *pdev)
 {
-       struct ata_host_set *host_set = pci_get_drvdata(pdev);
+       struct device *dev = pci_dev_to_dev(pdev);
+       struct ata_host_set *host_set = dev_get_drvdata(dev);
        struct ata_port *ap;
        unsigned int i;
 
@@ -3282,7 +3905,7 @@ void ata_pci_remove_one (struct pci_dev *pdev)
 
        kfree(host_set);
        pci_disable_device(pdev);
-       pci_set_drvdata(pdev, NULL);
+       dev_set_drvdata(dev, NULL);
 }
 
 /* move to PCI subsystem */
@@ -3318,6 +3941,7 @@ int pci_test_config_bits(struct pci_dev *pdev, struct pci_bits *bits)
 
        return (tmp == bits->val) ? 1 : 0;
 }
+#endif /* CONFIG_PCI */
 
 
 /**
@@ -3354,7 +3978,6 @@ module_exit(ata_exit);
  * Do not depend on ABI/API stability.
  */
 
-EXPORT_SYMBOL_GPL(pci_test_config_bits);
 EXPORT_SYMBOL_GPL(ata_std_bios_param);
 EXPORT_SYMBOL_GPL(ata_std_ports);
 EXPORT_SYMBOL_GPL(ata_device_add);
@@ -3363,34 +3986,43 @@ EXPORT_SYMBOL_GPL(ata_sg_init_one);
 EXPORT_SYMBOL_GPL(ata_qc_complete);
 EXPORT_SYMBOL_GPL(ata_qc_issue_prot);
 EXPORT_SYMBOL_GPL(ata_eng_timeout);
-EXPORT_SYMBOL_GPL(ata_tf_load_pio);
-EXPORT_SYMBOL_GPL(ata_tf_load_mmio);
-EXPORT_SYMBOL_GPL(ata_tf_read_pio);
-EXPORT_SYMBOL_GPL(ata_tf_read_mmio);
+EXPORT_SYMBOL_GPL(ata_tf_load);
+EXPORT_SYMBOL_GPL(ata_tf_read);
+EXPORT_SYMBOL_GPL(ata_noop_dev_select);
+EXPORT_SYMBOL_GPL(ata_std_dev_select);
 EXPORT_SYMBOL_GPL(ata_tf_to_fis);
 EXPORT_SYMBOL_GPL(ata_tf_from_fis);
-EXPORT_SYMBOL_GPL(ata_check_status_pio);
-EXPORT_SYMBOL_GPL(ata_check_status_mmio);
-EXPORT_SYMBOL_GPL(ata_exec_command_pio);
-EXPORT_SYMBOL_GPL(ata_exec_command_mmio);
+EXPORT_SYMBOL_GPL(ata_check_status);
+EXPORT_SYMBOL_GPL(ata_altstatus);
+EXPORT_SYMBOL_GPL(ata_chk_err);
+EXPORT_SYMBOL_GPL(ata_exec_command);
 EXPORT_SYMBOL_GPL(ata_port_start);
 EXPORT_SYMBOL_GPL(ata_port_stop);
 EXPORT_SYMBOL_GPL(ata_interrupt);
 EXPORT_SYMBOL_GPL(ata_qc_prep);
-EXPORT_SYMBOL_GPL(ata_bmdma_setup_pio);
-EXPORT_SYMBOL_GPL(ata_bmdma_start_pio);
-EXPORT_SYMBOL_GPL(ata_bmdma_setup_mmio);
-EXPORT_SYMBOL_GPL(ata_bmdma_start_mmio);
+EXPORT_SYMBOL_GPL(ata_bmdma_setup);
+EXPORT_SYMBOL_GPL(ata_bmdma_start);
 EXPORT_SYMBOL_GPL(ata_bmdma_irq_clear);
+EXPORT_SYMBOL_GPL(ata_bmdma_status);
+EXPORT_SYMBOL_GPL(ata_bmdma_stop);
 EXPORT_SYMBOL_GPL(ata_port_probe);
 EXPORT_SYMBOL_GPL(sata_phy_reset);
+EXPORT_SYMBOL_GPL(__sata_phy_reset);
 EXPORT_SYMBOL_GPL(ata_bus_reset);
 EXPORT_SYMBOL_GPL(ata_port_disable);
-EXPORT_SYMBOL_GPL(ata_pci_init_one);
-EXPORT_SYMBOL_GPL(ata_pci_remove_one);
+EXPORT_SYMBOL_GPL(ata_scsi_ioctl);
 EXPORT_SYMBOL_GPL(ata_scsi_queuecmd);
 EXPORT_SYMBOL_GPL(ata_scsi_error);
 EXPORT_SYMBOL_GPL(ata_scsi_slave_config);
 EXPORT_SYMBOL_GPL(ata_scsi_release);
 EXPORT_SYMBOL_GPL(ata_host_intr);
+EXPORT_SYMBOL_GPL(ata_dev_classify);
 EXPORT_SYMBOL_GPL(ata_dev_id_string);
+EXPORT_SYMBOL_GPL(ata_scsi_simulate);
+
+#ifdef CONFIG_PCI
+EXPORT_SYMBOL_GPL(pci_test_config_bits);
+EXPORT_SYMBOL_GPL(ata_pci_init_native_mode);
+EXPORT_SYMBOL_GPL(ata_pci_init_one);
+EXPORT_SYMBOL_GPL(ata_pci_remove_one);
+#endif /* CONFIG_PCI */