drivers/scsi/libata-core.c

   1 /*
   2  *  libata-core.c - helper library for ATA
   3  *
   4  *  Maintained by:  Jeff Garzik <jgarzik@pobox.com>
   5  *                  Please ALWAYS copy linux-ide@vger.kernel.org
   6  *                  on emails.
   7  *
   8  *  Copyright 2003-2004 Red Hat, Inc.  All rights reserved.
   9  *  Copyright 2003-2004 Jeff Garzik
  10  *
  11  *
  12  *  This program is free software; you can redistribute it and/or modify
  13  *  it under the terms of the GNU General Public License as published by
  14  *  the Free Software Foundation; either version 2, or (at your option)
  15  *  any later version.
  16  *
  17  *  This program is distributed in the hope that it will be useful,
  18  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  19  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20  *  GNU General Public License for more details.
  21  *
  22  *  You should have received a copy of the GNU General Public License
  23  *  along with this program; see the file COPYING.  If not, write to
  24  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  25  *
  26  *
  27  *  libata documentation is available via 'make {ps|pdf}docs',
  28  *  as Documentation/DocBook/libata.*
  29  *
  30  *  Hardware documentation available from http://www.t13.org/ and
  31  *  http://www.sata-io.org/
  32  *
  33  */
  34
  35 #include <linux/config.h>
  36 #include <linux/kernel.h>
  37 #include <linux/module.h>
  38 #include <linux/pci.h>
  39 #include <linux/init.h>
  40 #include <linux/list.h>
  41 #include <linux/mm.h>
  42 #include <linux/highmem.h>
  43 #include <linux/spinlock.h>
  44 #include <linux/blkdev.h>
  45 #include <linux/delay.h>
  46 #include <linux/timer.h>
  47 #include <linux/interrupt.h>
  48 #include <linux/completion.h>
  49 #include <linux/suspend.h>
  50 #include <linux/workqueue.h>
  51 #include <linux/jiffies.h>
  52 #include <linux/scatterlist.h>
  53 #include <scsi/scsi.h>
  54 #include "scsi_priv.h"
  55 #include <scsi/scsi_cmnd.h>
  56 #include <scsi/scsi_host.h>
  57 #include <linux/libata.h>
  58 #include <asm/io.h>
  59 #include <asm/semaphore.h>
  60 #include <asm/byteorder.h>
  61
  62 #include "libata.h"
  63
  64 static void ata_dev_reread_id(struct ata_port *ap, struct ata_device *dev);
  65 static void ata_dev_init_params(struct ata_port *ap, struct ata_device *dev);
  66 static void ata_set_mode(struct ata_port *ap);
  67 static void ata_dev_set_xfermode(struct ata_port *ap, struct ata_device *dev);
  68 static unsigned int ata_get_mode_mask(const struct ata_port *ap, int shift);
  69 static int fgb(u32 bitmap);
  70 static int ata_choose_xfer_mode(const struct ata_port *ap,
  71                                 u8 *xfer_mode_out,
  72                                 unsigned int *xfer_shift_out);
  73
  74 static unsigned int ata_unique_id = 1;
  75 static struct workqueue_struct *ata_wq;
  76
  77 int atapi_enabled = 0;
  78 module_param(atapi_enabled, int, 0444);
  79 MODULE_PARM_DESC(atapi_enabled, "Enable discovery of ATAPI devices (0=off, 1=on)");
  80
  81 MODULE_AUTHOR("Jeff Garzik");
  82 MODULE_DESCRIPTION("Library module for ATA devices");
  83 MODULE_LICENSE("GPL");
  84 MODULE_VERSION(DRV_VERSION);
  85
  86
  87 /**
  88  *      ata_tf_to_fis - Convert ATA taskfile to SATA FIS structure
  89  *      @tf: Taskfile to convert
  90  *      @fis: Buffer into which data will output
  91  *      @pmp: Port multiplier port
  92  *
  93  *      Converts a standard ATA taskfile to a Serial ATA
  94  *      FIS structure (Register - Host to Device).
  95  *
  96  *      LOCKING:
  97  *      Inherited from caller.
  98  */
  99
 100 void ata_tf_to_fis(const struct ata_taskfile *tf, u8 *fis, u8 pmp)
 101 {
 102         fis[0] = 0x27;  /* Register - Host to Device FIS */
 103         fis[1] = (pmp & 0xf) | (1 << 7); /* Port multiplier number,
 104                                             bit 7 indicates Command FIS */
 105         fis[2] = tf->command;
 106         fis[3] = tf->feature;
 107
 108         fis[4] = tf->lbal;
 109         fis[5] = tf->lbam;
 110         fis[6] = tf->lbah;
 111         fis[7] = tf->device;
 112
 113         fis[8] = tf->hob_lbal;
 114         fis[9] = tf->hob_lbam;
 115         fis[10] = tf->hob_lbah;
 116         fis[11] = tf->hob_feature;
 117
 118         fis[12] = tf->nsect;
 119         fis[13] = tf->hob_nsect;
 120         fis[14] = 0;
 121         fis[15] = tf->ctl;
 122
 123         fis[16] = 0;
 124         fis[17] = 0;
 125         fis[18] = 0;
 126         fis[19] = 0;
 127 }
 128
 129 /**
 130  *      ata_tf_from_fis - Convert SATA FIS to ATA taskfile
 131  *      @fis: Buffer from which data will be input
 132  *      @tf: Taskfile to output
 133  *
 134  *      Converts a serial ATA FIS structure to a standard ATA taskfile.
 135  *
 136  *      LOCKING:
 137  *      Inherited from caller.
 138  */
 139
 140 void ata_tf_from_fis(const u8 *fis, struct ata_taskfile *tf)
 141 {
 142         tf->command     = fis[2];       /* status */
 143         tf->feature     = fis[3];       /* error */
 144
 145         tf->lbal        = fis[4];
 146         tf->lbam        = fis[5];
 147         tf->lbah        = fis[6];
 148         tf->device      = fis[7];
 149
 150         tf->hob_lbal    = fis[8];
 151         tf->hob_lbam    = fis[9];
 152         tf->hob_lbah    = fis[10];
 153
 154         tf->nsect       = fis[12];
 155         tf->hob_nsect   = fis[13];
 156 }
 157
 158 static const u8 ata_rw_cmds[] = {
 159         /* pio multi */
 160         ATA_CMD_READ_MULTI,
 161         ATA_CMD_WRITE_MULTI,
 162         ATA_CMD_READ_MULTI_EXT,
 163         ATA_CMD_WRITE_MULTI_EXT,
 164         0,
 165         0,
 166         0,
 167         ATA_CMD_WRITE_MULTI_FUA_EXT,
 168         /* pio */
 169         ATA_CMD_PIO_READ,
 170         ATA_CMD_PIO_WRITE,
 171         ATA_CMD_PIO_READ_EXT,
 172         ATA_CMD_PIO_WRITE_EXT,
 173         0,
 174         0,
 175         0,
 176         0,
 177         /* dma */
 178         ATA_CMD_READ,
 179         ATA_CMD_WRITE,
 180         ATA_CMD_READ_EXT,
 181         ATA_CMD_WRITE_EXT,
 182         0,
 183         0,
 184         0,
 185         ATA_CMD_WRITE_FUA_EXT
 186 };
 187
 188 /**
 189  *      ata_rwcmd_protocol - set taskfile r/w commands and protocol
 190  *      @qc: command to examine and configure
 191  *
 192  *      Examine the device configuration and tf->flags to calculate
 193  *      the proper read/write commands and protocol to use.
 194  *
 195  *      LOCKING:
 196  *      caller.
 197  */
 198 int ata_rwcmd_protocol(struct ata_queued_cmd *qc)
 199 {
 200         struct ata_taskfile *tf = &qc->tf;
 201         struct ata_device *dev = qc->dev;
 202         u8 cmd;
 203
 204         int index, fua, lba48, write;
 205
 206         fua = (tf->flags & ATA_TFLAG_FUA) ? 4 : 0;
 207         lba48 = (tf->flags & ATA_TFLAG_LBA48) ? 2 : 0;
 208         write = (tf->flags & ATA_TFLAG_WRITE) ? 1 : 0;
 209
 210         if (dev->flags & ATA_DFLAG_PIO) {
 211                 tf->protocol = ATA_PROT_PIO;
 212                 index = dev->multi_count ? 0 : 8;
 213         } else if (lba48 && (qc->ap->flags & ATA_FLAG_PIO_LBA48)) {
 214                 /* Unable to use DMA due to host limitation */
 215                 tf->protocol = ATA_PROT_PIO;
 216                 index = dev->multi_count ? 0 : 4;
 217         } else {
 218                 tf->protocol = ATA_PROT_DMA;
 219                 index = 16;
 220         }
 221
 222         cmd = ata_rw_cmds[index + fua + lba48 + write];
 223         if (cmd) {
 224                 tf->command = cmd;
 225                 return 0;
 226         }
 227         return -1;
 228 }
 229
 230 static const char * const xfer_mode_str[] = {
 231         "UDMA/16",
 232         "UDMA/25",
 233         "UDMA/33",
 234         "UDMA/44",
 235         "UDMA/66",
 236         "UDMA/100",
 237         "UDMA/133",
 238         "UDMA7",
 239         "MWDMA0",
 240         "MWDMA1",
 241         "MWDMA2",
 242         "PIO0",
 243         "PIO1",
 244         "PIO2",
 245         "PIO3",
 246         "PIO4",
 247 };
 248
 249 /**
 250  *      ata_udma_string - convert UDMA bit offset to string
 251  *      @mask: mask of bits supported; only highest bit counts.
 252  *
 253  *      Determine string which represents the highest speed
 254  *      (highest bit in @udma_mask).
 255  *
 256  *      LOCKING:
 257  *      None.
 258  *
 259  *      RETURNS:
 260  *      Constant C string representing highest speed listed in
 261  *      @udma_mask, or the constant C string "<n/a>".
 262  */
 263
 264 static const char *ata_mode_string(unsigned int mask)
 265 {
 266         int i;
 267
 268         for (i = 7; i >= 0; i--)
 269                 if (mask & (1 << i))
 270                         goto out;
 271         for (i = ATA_SHIFT_MWDMA + 2; i >= ATA_SHIFT_MWDMA; i--)
 272                 if (mask & (1 << i))
 273                         goto out;
 274         for (i = ATA_SHIFT_PIO + 4; i >= ATA_SHIFT_PIO; i--)
 275                 if (mask & (1 << i))
 276                         goto out;
 277
 278         return "<n/a>";
 279
 280 out:
 281         return xfer_mode_str[i];
 282 }
 283
 284 /**
 285  *      ata_pio_devchk - PATA device presence detection
 286  *      @ap: ATA channel to examine
 287  *      @device: Device to examine (starting at zero)
 288  *
 289  *      This technique was originally described in
 290  *      Hale Landis's ATADRVR (www.ata-atapi.com), and
 291  *      later found its way into the ATA/ATAPI spec.
 292  *
 293  *      Write a pattern to the ATA shadow registers,
 294  *      and if a device is present, it will respond by
 295  *      correctly storing and echoing back the
 296  *      ATA shadow register contents.
 297  *
 298  *      LOCKING:
 299  *      caller.
 300  */
 301
 302 static unsigned int ata_pio_devchk(struct ata_port *ap,
 303                                    unsigned int device)
 304 {
 305         struct ata_ioports *ioaddr = &ap->ioaddr;
 306         u8 nsect, lbal;
 307
 308         ap->ops->dev_select(ap, device);
 309
 310         outb(0x55, ioaddr->nsect_addr);
 311         outb(0xaa, ioaddr->lbal_addr);
 312
 313         outb(0xaa, ioaddr->nsect_addr);
 314         outb(0x55, ioaddr->lbal_addr);
 315
 316         outb(0x55, ioaddr->nsect_addr);
 317         outb(0xaa, ioaddr->lbal_addr);
 318
 319         nsect = inb(ioaddr->nsect_addr);
 320         lbal = inb(ioaddr->lbal_addr);
 321
 322         if ((nsect == 0x55) && (lbal == 0xaa))
 323                 return 1;       /* we found a device */
 324
 325         return 0;               /* nothing found */
 326 }
 327
 328 /**
 329  *      ata_mmio_devchk - PATA device presence detection
 330  *      @ap: ATA channel to examine
 331  *      @device: Device to examine (starting at zero)
 332  *
 333  *      This technique was originally described in
 334  *      Hale Landis's ATADRVR (www.ata-atapi.com), and
 335  *      later found its way into the ATA/ATAPI spec.
 336  *
 337  *      Write a pattern to the ATA shadow registers,
 338  *      and if a device is present, it will respond by
 339  *      correctly storing and echoing back the
 340  *      ATA shadow register contents.
 341  *
 342  *      LOCKING:
 343  *      caller.
 344  */
 345
 346 static unsigned int ata_mmio_devchk(struct ata_port *ap,
 347                                     unsigned int device)
 348 {
 349         struct ata_ioports *ioaddr = &ap->ioaddr;
 350         u8 nsect, lbal;
 351
 352         ap->ops->dev_select(ap, device);
 353
 354         writeb(0x55, (void __iomem *) ioaddr->nsect_addr);
 355         writeb(0xaa, (void __iomem *) ioaddr->lbal_addr);
 356
 357         writeb(0xaa, (void __iomem *) ioaddr->nsect_addr);
 358         writeb(0x55, (void __iomem *) ioaddr->lbal_addr);
 359
 360         writeb(0x55, (void __iomem *) ioaddr->nsect_addr);
 361         writeb(0xaa, (void __iomem *) ioaddr->lbal_addr);
 362
 363         nsect = readb((void __iomem *) ioaddr->nsect_addr);
 364         lbal = readb((void __iomem *) ioaddr->lbal_addr);
 365
 366         if ((nsect == 0x55) && (lbal == 0xaa))
 367                 return 1;       /* we found a device */
 368
 369         return 0;               /* nothing found */
 370 }
 371
 372 /**
 373  *      ata_devchk - PATA device presence detection
 374  *      @ap: ATA channel to examine
 375  *      @device: Device to examine (starting at zero)
 376  *
 377  *      Dispatch ATA device presence detection, depending
 378  *      on whether we are using PIO or MMIO to talk to the
 379  *      ATA shadow registers.
 380  *
 381  *      LOCKING:
 382  *      caller.
 383  */
 384
 385 static unsigned int ata_devchk(struct ata_port *ap,
 386                                     unsigned int device)
 387 {
 388         if (ap->flags & ATA_FLAG_MMIO)
 389                 return ata_mmio_devchk(ap, device);
 390         return ata_pio_devchk(ap, device);
 391 }
 392
 393 /**
 394  *      ata_dev_classify - determine device type based on ATA-spec signature
 395  *      @tf: ATA taskfile register set for device to be identified
 396  *
 397  *      Determine from taskfile register contents whether a device is
 398  *      ATA or ATAPI, as per "Signature and persistence" section
 399  *      of ATA/PI spec (volume 1, sect 5.14).
 400  *
 401  *      LOCKING:
 402  *      None.
 403  *
 404  *      RETURNS:
 405  *      Device type, %ATA_DEV_ATA, %ATA_DEV_ATAPI, or %ATA_DEV_UNKNOWN
 406  *      the event of failure.
 407  */
 408
 409 unsigned int ata_dev_classify(const struct ata_taskfile *tf)
 410 {
 411         /* Apple's open source Darwin code hints that some devices only
 412          * put a proper signature into the LBA mid/high registers,
 413          * So, we only check those.  It's sufficient for uniqueness.
 414          */
 415
 416         if (((tf->lbam == 0) && (tf->lbah == 0)) ||
 417             ((tf->lbam == 0x3c) && (tf->lbah == 0xc3))) {
 418                 DPRINTK("found ATA device by sig\n");
 419                 return ATA_DEV_ATA;
 420         }
 421
 422         if (((tf->lbam == 0x14) && (tf->lbah == 0xeb)) ||
 423             ((tf->lbam == 0x69) && (tf->lbah == 0x96))) {
 424                 DPRINTK("found ATAPI device by sig\n");
 425                 return ATA_DEV_ATAPI;
 426         }
 427
 428         DPRINTK("unknown device\n");
 429         return ATA_DEV_UNKNOWN;
 430 }
 431
 432 /**
 433  *      ata_dev_try_classify - Parse returned ATA device signature
 434  *      @ap: ATA channel to examine
 435  *      @device: Device to examine (starting at zero)
 436  *      @r_err: Value of error register on completion
 437  *
 438  *      After an event -- SRST, E.D.D., or SATA COMRESET -- occurs,
 439  *      an ATA/ATAPI-defined set of values is placed in the ATA
 440  *      shadow registers, indicating the results of device detection
 441  *      and diagnostics.
 442  *
 443  *      Select the ATA device, and read the values from the ATA shadow
 444  *      registers.  Then parse according to the Error register value,
 445  *      and the spec-defined values examined by ata_dev_classify().
 446  *
 447  *      LOCKING:
 448  *      caller.
 449  *
 450  *      RETURNS:
 451  *      Device type - %ATA_DEV_ATA, %ATA_DEV_ATAPI or %ATA_DEV_NONE.
 452  */
 453
 454 static unsigned int
 455 ata_dev_try_classify(struct ata_port *ap, unsigned int device, u8 *r_err)
 456 {
 457         struct ata_taskfile tf;
 458         unsigned int class;
 459         u8 err;
 460
 461         ap->ops->dev_select(ap, device);
 462
 463         memset(&tf, 0, sizeof(tf));
 464
 465         ap->ops->tf_read(ap, &tf);
 466         err = tf.feature;
 467         if (r_err)
 468                 *r_err = err;
 469
 470         /* see if device passed diags */
 471         if (err == 1)
 472                 /* do nothing */ ;
 473         else if ((device == 0) && (err == 0x81))
 474                 /* do nothing */ ;
 475         else
 476                 return ATA_DEV_NONE;
 477
 478         /* determine if device is ATA or ATAPI */
 479         class = ata_dev_classify(&tf);
 480
 481         if (class == ATA_DEV_UNKNOWN)
 482                 return ATA_DEV_NONE;
 483         if ((class == ATA_DEV_ATA) && (ata_chk_status(ap) == 0))
 484                 return ATA_DEV_NONE;
 485         return class;
 486 }
 487
 488 /**
 489  *      ata_dev_id_string - Convert IDENTIFY DEVICE page into string
 490  *      @id: IDENTIFY DEVICE results we will examine
 491  *      @s: string into which data is output
 492  *      @ofs: offset into identify device page
 493  *      @len: length of string to return. must be an even number.
 494  *
 495  *      The strings in the IDENTIFY DEVICE page are broken up into
 496  *      16-bit chunks.  Run through the string, and output each
 497  *      8-bit chunk linearly, regardless of platform.
 498  *
 499  *      LOCKING:
 500  *      caller.
 501  */
 502
 503 void ata_dev_id_string(const u16 *id, unsigned char *s,
 504                        unsigned int ofs, unsigned int len)
 505 {
 506         unsigned int c;
 507
 508         while (len > 0) {
 509                 c = id[ofs] >> 8;
 510                 *s = c;
 511                 s++;
 512
 513                 c = id[ofs] & 0xff;
 514                 *s = c;
 515                 s++;
 516
 517                 ofs++;
 518                 len -= 2;
 519         }
 520 }
 521
 522 /**
 523  *      ata_dev_id_c_string - Convert IDENTIFY DEVICE page into C string
 524  *      @id: IDENTIFY DEVICE results we will examine
 525  *      @s: string into which data is output
 526  *      @ofs: offset into identify device page
 527  *      @len: length of string to return. must be an odd number.
 528  *
 529  *      This function is identical to ata_dev_id_string except that it
 530  *      trims trailing spaces and terminates the resulting string with
 531  *      null.  @len must be actual maximum length (even number) + 1.
 532  *
 533  *      LOCKING:
 534  *      caller.
 535  */
 536 void ata_dev_id_c_string(const u16 *id, unsigned char *s,
 537                          unsigned int ofs, unsigned int len)
 538 {
 539         unsigned char *p;
 540
 541         WARN_ON(!(len & 1));
 542
 543         ata_dev_id_string(id, s, ofs, len - 1);
 544
 545         p = s + strnlen(s, len - 1);
 546         while (p > s && p[-1] == ' ')
 547                 p--;
 548         *p = '\0';
 549 }
 550
 551 static u64 ata_id_n_sectors(const u16 *id)
 552 {
 553         if (ata_id_has_lba(id)) {
 554                 if (ata_id_has_lba48(id))
 555                         return ata_id_u64(id, 100);
 556                 else
 557                         return ata_id_u32(id, 60);
 558         } else {
 559                 if (ata_id_current_chs_valid(id))
 560                         return ata_id_u32(id, 57);
 561                 else
 562                         return id[1] * id[3] * id[6];
 563         }
 564 }
 565
 566 /**
 567  *      ata_noop_dev_select - Select device 0/1 on ATA bus
 568  *      @ap: ATA channel to manipulate
 569  *      @device: ATA device (numbered from zero) to select
 570  *
 571  *      This function performs no actual function.
 572  *
 573  *      May be used as the dev_select() entry in ata_port_operations.
 574  *
 575  *      LOCKING:
 576  *      caller.
 577  */
 578 void ata_noop_dev_select (struct ata_port *ap, unsigned int device)
 579 {
 580 }
 581
 582
 583 /**
 584  *      ata_std_dev_select - Select device 0/1 on ATA bus
 585  *      @ap: ATA channel to manipulate
 586  *      @device: ATA device (numbered from zero) to select
 587  *
 588  *      Use the method defined in the ATA specification to
 589  *      make either device 0, or device 1, active on the
 590  *      ATA channel.  Works with both PIO and MMIO.
 591  *
 592  *      May be used as the dev_select() entry in ata_port_operations.
 593  *
 594  *      LOCKING:
 595  *      caller.
 596  */
 597
 598 void ata_std_dev_select (struct ata_port *ap, unsigned int device)
 599 {
 600         u8 tmp;
 601
 602         if (device == 0)
 603                 tmp = ATA_DEVICE_OBS;
 604         else
 605                 tmp = ATA_DEVICE_OBS | ATA_DEV1;
 606
 607         if (ap->flags & ATA_FLAG_MMIO) {
 608                 writeb(tmp, (void __iomem *) ap->ioaddr.device_addr);
 609         } else {
 610                 outb(tmp, ap->ioaddr.device_addr);
 611         }
 612         ata_pause(ap);          /* needed; also flushes, for mmio */
 613 }
 614
 615 /**
 616  *      ata_dev_select - Select device 0/1 on ATA bus
 617  *      @ap: ATA channel to manipulate
 618  *      @device: ATA device (numbered from zero) to select
 619  *      @wait: non-zero to wait for Status register BSY bit to clear
 620  *      @can_sleep: non-zero if context allows sleeping
 621  *
 622  *      Use the method defined in the ATA specification to
 623  *      make either device 0, or device 1, active on the
 624  *      ATA channel.
 625  *
 626  *      This is a high-level version of ata_std_dev_select(),
 627  *      which additionally provides the services of inserting
 628  *      the proper pauses and status polling, where needed.
 629  *
 630  *      LOCKING:
 631  *      caller.
 632  */
 633
 634 void ata_dev_select(struct ata_port *ap, unsigned int device,
 635                            unsigned int wait, unsigned int can_sleep)
 636 {
 637         VPRINTK("ENTER, ata%u: device %u, wait %u\n",
 638                 ap->id, device, wait);
 639
 640         if (wait)
 641                 ata_wait_idle(ap);
 642
 643         ap->ops->dev_select(ap, device);
 644
 645         if (wait) {
 646                 if (can_sleep && ap->device[device].class == ATA_DEV_ATAPI)
 647                         msleep(150);
 648                 ata_wait_idle(ap);
 649         }
 650 }
 651
 652 /**
 653  *      ata_dump_id - IDENTIFY DEVICE info debugging output
 654  *      @id: IDENTIFY DEVICE page to dump
 655  *
 656  *      Dump selected 16-bit words from the given IDENTIFY DEVICE
 657  *      page.
 658  *
 659  *      LOCKING:
 660  *      caller.
 661  */
 662
 663 static inline void ata_dump_id(const u16 *id)
 664 {
 665         DPRINTK("49==0x%04x  "
 666                 "53==0x%04x  "
 667                 "63==0x%04x  "
 668                 "64==0x%04x  "
 669                 "75==0x%04x  \n",
 670                 id[49],
 671                 id[53],
 672                 id[63],
 673                 id[64],
 674                 id[75]);
 675         DPRINTK("80==0x%04x  "
 676                 "81==0x%04x  "
 677                 "82==0x%04x  "
 678                 "83==0x%04x  "
 679                 "84==0x%04x  \n",
 680                 id[80],
 681                 id[81],
 682                 id[82],
 683                 id[83],
 684                 id[84]);
 685         DPRINTK("88==0x%04x  "
 686                 "93==0x%04x\n",
 687                 id[88],
 688                 id[93]);
 689 }
 690
 691 /*
 692  *      Compute the PIO modes available for this device. This is not as
 693  *      trivial as it seems if we must consider early devices correctly.
 694  *
 695  *      FIXME: pre IDE drive timing (do we care ?).
 696  */
 697
 698 static unsigned int ata_pio_modes(const struct ata_device *adev)
 699 {
 700         u16 modes;
 701
 702         /* Usual case. Word 53 indicates word 64 is valid */
 703         if (adev->id[ATA_ID_FIELD_VALID] & (1 << 1)) {
 704                 modes = adev->id[ATA_ID_PIO_MODES] & 0x03;
 705                 modes <<= 3;
 706                 modes |= 0x7;
 707                 return modes;
 708         }
 709
 710         /* If word 64 isn't valid then Word 51 high byte holds the PIO timing
 711            number for the maximum. Turn it into a mask and return it */
 712         modes = (2 << ((adev->id[ATA_ID_OLD_PIO_MODES] >> 8) & 0xFF)) - 1 ;
 713         return modes;
 714         /* But wait.. there's more. Design your standards by committee and
 715            you too can get a free iordy field to process. However its the
 716            speeds not the modes that are supported... Note drivers using the
 717            timing API will get this right anyway */
 718 }
 719
 720 static inline void
 721 ata_queue_packet_task(struct ata_port *ap)
 722 {
 723         if (!(ap->flags & ATA_FLAG_FLUSH_PIO_TASK))
 724                 queue_work(ata_wq, &ap->packet_task);
 725 }
 726
 727 static inline void
 728 ata_queue_pio_task(struct ata_port *ap)
 729 {
 730         if (!(ap->flags & ATA_FLAG_FLUSH_PIO_TASK))
 731                 queue_work(ata_wq, &ap->pio_task);
 732 }
 733
 734 static inline void
 735 ata_queue_delayed_pio_task(struct ata_port *ap, unsigned long delay)
 736 {
 737         if (!(ap->flags & ATA_FLAG_FLUSH_PIO_TASK))
 738                 queue_delayed_work(ata_wq, &ap->pio_task, delay);
 739 }
 740
 741 /**
 742  *      ata_flush_pio_tasks - Flush pio_task and packet_task
 743  *      @ap: the target ata_port
 744  *
 745  *      After this function completes, pio_task and packet_task are
 746  *      guranteed not to be running or scheduled.
 747  *
 748  *      LOCKING:
 749  *      Kernel thread context (may sleep)
 750  */
 751
 752 static void ata_flush_pio_tasks(struct ata_port *ap)
 753 {
 754         int tmp = 0;
 755         unsigned long flags;
 756
 757         DPRINTK("ENTER\n");
 758
 759         spin_lock_irqsave(&ap->host_set->lock, flags);
 760         ap->flags |= ATA_FLAG_FLUSH_PIO_TASK;
 761         spin_unlock_irqrestore(&ap->host_set->lock, flags);
 762
 763         DPRINTK("flush #1\n");
 764         flush_workqueue(ata_wq);
 765
 766         /*
 767          * At this point, if a task is running, it's guaranteed to see
 768          * the FLUSH flag; thus, it will never queue pio tasks again.
 769          * Cancel and flush.
 770          */
 771         tmp |= cancel_delayed_work(&ap->pio_task);
 772         tmp |= cancel_delayed_work(&ap->packet_task);
 773         if (!tmp) {
 774                 DPRINTK("flush #2\n");
 775                 flush_workqueue(ata_wq);
 776         }
 777
 778         spin_lock_irqsave(&ap->host_set->lock, flags);
 779         ap->flags &= ~ATA_FLAG_FLUSH_PIO_TASK;
 780         spin_unlock_irqrestore(&ap->host_set->lock, flags);
 781
 782         DPRINTK("EXIT\n");
 783 }
 784
 785 void ata_qc_complete_internal(struct ata_queued_cmd *qc)
 786 {
 787         struct completion *waiting = qc->private_data;
 788
 789         qc->ap->ops->tf_read(qc->ap, &qc->tf);
 790         complete(waiting);
 791 }
 792
 793 /**
 794  *      ata_exec_internal - execute libata internal command
 795  *      @ap: Port to which the command is sent
 796  *      @dev: Device to which the command is sent
 797  *      @tf: Taskfile registers for the command and the result
 798  *      @dma_dir: Data tranfer direction of the command
 799  *      @buf: Data buffer of the command
 800  *      @buflen: Length of data buffer
 801  *
 802  *      Executes libata internal command with timeout.  @tf contains
 803  *      command on entry and result on return.  Timeout and error
 804  *      conditions are reported via return value.  No recovery action
 805  *      is taken after a command times out.  It's caller's duty to
 806  *      clean up after timeout.
 807  *
 808  *      LOCKING:
 809  *      None.  Should be called with kernel context, might sleep.
 810  */
 811
 812 static unsigned
 813 ata_exec_internal(struct ata_port *ap, struct ata_device *dev,
 814                   struct ata_taskfile *tf,
 815                   int dma_dir, void *buf, unsigned int buflen)
 816 {
 817         u8 command = tf->command;
 818         struct ata_queued_cmd *qc;
 819         DECLARE_COMPLETION(wait);
 820         unsigned long flags;
 821         unsigned int err_mask;
 822
 823         spin_lock_irqsave(&ap->host_set->lock, flags);
 824
 825         qc = ata_qc_new_init(ap, dev);
 826         BUG_ON(qc == NULL);
 827
 828         qc->tf = *tf;
 829         qc->dma_dir = dma_dir;
 830         if (dma_dir != DMA_NONE) {
 831                 ata_sg_init_one(qc, buf, buflen);
 832                 qc->nsect = buflen / ATA_SECT_SIZE;
 833         }
 834
 835         qc->private_data = &wait;
 836         qc->complete_fn = ata_qc_complete_internal;
 837
 838         qc->err_mask = ata_qc_issue(qc);
 839         if (qc->err_mask)
 840                 ata_qc_complete(qc);
 841
 842         spin_unlock_irqrestore(&ap->host_set->lock, flags);
 843
 844         if (!wait_for_completion_timeout(&wait, ATA_TMOUT_INTERNAL)) {
 845                 spin_lock_irqsave(&ap->host_set->lock, flags);
 846
 847                 /* We're racing with irq here.  If we lose, the
 848                  * following test prevents us from completing the qc
 849                  * again.  If completion irq occurs after here but
 850                  * before the caller cleans up, it will result in a
 851                  * spurious interrupt.  We can live with that.
 852                  */
 853                 if (qc->flags & ATA_QCFLAG_ACTIVE) {
 854                         qc->err_mask = AC_ERR_TIMEOUT;
 855                         ata_qc_complete(qc);
 856                         printk(KERN_WARNING "ata%u: qc timeout (cmd 0x%x)\n",
 857                                ap->id, command);
 858                 }
 859
 860                 spin_unlock_irqrestore(&ap->host_set->lock, flags);
 861         }
 862
 863         *tf = qc->tf;
 864         err_mask = qc->err_mask;
 865
 866         ata_qc_free(qc);
 867
 868         return err_mask;
 869 }
 870
 871 /**
 872  *      ata_pio_need_iordy      -       check if iordy needed
 873  *      @adev: ATA device
 874  *
 875  *      Check if the current speed of the device requires IORDY. Used
 876  *      by various controllers for chip configuration.
 877  */
 878
 879 unsigned int ata_pio_need_iordy(const struct ata_device *adev)
 880 {
 881         int pio;
 882         int speed = adev->pio_mode - XFER_PIO_0;
 883
 884         if (speed < 2)
 885                 return 0;
 886         if (speed > 2)
 887                 return 1;
 888
 889         /* If we have no drive specific rule, then PIO 2 is non IORDY */
 890
 891         if (adev->id[ATA_ID_FIELD_VALID] & 2) { /* EIDE */
 892                 pio = adev->id[ATA_ID_EIDE_PIO];
 893                 /* Is the speed faster than the drive allows non IORDY ? */
 894                 if (pio) {
 895                         /* This is cycle times not frequency - watch the logic! */
 896                         if (pio > 240)  /* PIO2 is 240nS per cycle */
 897                                 return 1;
 898                         return 0;
 899                 }
 900         }
 901         return 0;
 902 }
 903
 904 /**
 905  *      ata_dev_identify - obtain IDENTIFY x DEVICE page
 906  *      @ap: port on which device we wish to probe resides
 907  *      @device: device bus address, starting at zero
 908  *
 909  *      Following bus reset, we issue the IDENTIFY [PACKET] DEVICE
 910  *      command, and read back the 512-byte device information page.
 911  *      The device information page is fed to us via the standard
 912  *      PIO-IN protocol, but we hand-code it here. (TODO: investigate
 913  *      using standard PIO-IN paths)
 914  *
 915  *      After reading the device information page, we use several
 916  *      bits of information from it to initialize data structures
 917  *      that will be used during the lifetime of the ata_device.
 918  *      Other data from the info page is used to disqualify certain
 919  *      older ATA devices we do not wish to support.
 920  *
 921  *      LOCKING:
 922  *      Inherited from caller.  Some functions called by this function
 923  *      obtain the host_set lock.
 924  */
 925
 926 static void ata_dev_identify(struct ata_port *ap, unsigned int device)
 927 {
 928         struct ata_device *dev = &ap->device[device];
 929         unsigned int major_version;
 930         unsigned long xfer_modes;
 931         unsigned int using_edd;
 932         struct ata_taskfile tf;
 933         unsigned int err_mask;
 934         int i, rc;
 935
 936         if (!ata_dev_present(dev)) {
 937                 DPRINTK("ENTER/EXIT (host %u, dev %u) -- nodev\n",
 938                         ap->id, device);
 939                 return;
 940         }
 941
 942         if (ap->ops->probe_reset ||
 943             ap->flags & (ATA_FLAG_SRST | ATA_FLAG_SATA_RESET))
 944                 using_edd = 0;
 945         else
 946                 using_edd = 1;
 947
 948         DPRINTK("ENTER, host %u, dev %u\n", ap->id, device);
 949
 950         WARN_ON(dev->class != ATA_DEV_ATA && dev->class != ATA_DEV_ATAPI &&
 951                 dev->class != ATA_DEV_NONE);
 952
 953         ata_dev_select(ap, device, 1, 1); /* select device 0/1 */
 954
 955 retry:
 956         ata_tf_init(ap, &tf, device);
 957
 958         if (dev->class == ATA_DEV_ATA) {
 959                 tf.command = ATA_CMD_ID_ATA;
 960                 DPRINTK("do ATA identify\n");
 961         } else {
 962                 tf.command = ATA_CMD_ID_ATAPI;
 963                 DPRINTK("do ATAPI identify\n");
 964         }
 965
 966         tf.protocol = ATA_PROT_PIO;
 967
 968         err_mask = ata_exec_internal(ap, dev, &tf, DMA_FROM_DEVICE,
 969                                      dev->id, sizeof(dev->id));
 970
 971         if (err_mask) {
 972                 if (err_mask & ~AC_ERR_DEV)
 973                         goto err_out;
 974
 975                 /*
 976                  * arg!  EDD works for all test cases, but seems to return
 977                  * the ATA signature for some ATAPI devices.  Until the
 978                  * reason for this is found and fixed, we fix up the mess
 979                  * here.  If IDENTIFY DEVICE returns command aborted
 980                  * (as ATAPI devices do), then we issue an
 981                  * IDENTIFY PACKET DEVICE.
 982                  *
 983                  * ATA software reset (SRST, the default) does not appear
 984                  * to have this problem.
 985                  */
 986                 if ((using_edd) && (dev->class == ATA_DEV_ATA)) {
 987                         u8 err = tf.feature;
 988                         if (err & ATA_ABORTED) {
 989                                 dev->class = ATA_DEV_ATAPI;
 990                                 goto retry;
 991                         }
 992                 }
 993                 goto err_out;
 994         }
 995
 996         swap_buf_le16(dev->id, ATA_ID_WORDS);
 997
 998         /* print device capabilities */
 999         printk(KERN_DEBUG "ata%u: dev %u cfg "
1000                "49:%04x 82:%04x 83:%04x 84:%04x 85:%04x 86:%04x 87:%04x 88:%04x\n",
1001                ap->id, device, dev->id[49],
1002                dev->id[82], dev->id[83], dev->id[84],
1003                dev->id[85], dev->id[86], dev->id[87],
1004                dev->id[88]);
1005
1006         /*
1007          * common ATA, ATAPI feature tests
1008          */
1009
1010         /* we require DMA support (bits 8 of word 49) */
1011         if (!ata_id_has_dma(dev->id)) {
1012                 printk(KERN_DEBUG "ata%u: no dma\n", ap->id);
1013                 goto err_out_nosup;
1014         }
1015
1016         /* quick-n-dirty find max transfer mode; for printk only */
1017         xfer_modes = dev->id[ATA_ID_UDMA_MODES];
1018         if (!xfer_modes)
1019                 xfer_modes = (dev->id[ATA_ID_MWDMA_MODES]) << ATA_SHIFT_MWDMA;
1020         if (!xfer_modes)
1021                 xfer_modes = ata_pio_modes(dev);
1022
1023         ata_dump_id(dev->id);
1024
1025         /* ATA-specific feature tests */
1026         if (dev->class == ATA_DEV_ATA) {
1027                 dev->n_sectors = ata_id_n_sectors(dev->id);
1028
1029                 if (!ata_id_is_ata(dev->id))    /* sanity check */
1030                         goto err_out_nosup;
1031
1032                 /* get major version */
1033                 major_version = ata_id_major_version(dev->id);
1034
1035                 /*
1036                  * The exact sequence expected by certain pre-ATA4 drives is:
1037                  * SRST RESET
1038                  * IDENTIFY
1039                  * INITIALIZE DEVICE PARAMETERS
1040                  * anything else..
1041                  * Some drives were very specific about that exact sequence.
1042                  */
1043                 if (major_version < 4 || (!ata_id_has_lba(dev->id))) {
1044                         ata_dev_init_params(ap, dev);
1045
1046                         /* current CHS translation info (id[53-58]) might be
1047                          * changed. reread the identify device info.
1048                          */
1049                         ata_dev_reread_id(ap, dev);
1050                 }
1051
1052                 if (ata_id_has_lba(dev->id)) {
1053                         dev->flags |= ATA_DFLAG_LBA;
1054
1055                         if (ata_id_has_lba48(dev->id))
1056                                 dev->flags |= ATA_DFLAG_LBA48;
1057
1058                         /* print device info to dmesg */
1059                         printk(KERN_INFO "ata%u: dev %u ATA-%d, max %s, %Lu sectors:%s\n",
1060                                ap->id, device,
1061                                major_version,
1062                                ata_mode_string(xfer_modes),
1063                                (unsigned long long)dev->n_sectors,
1064                                dev->flags & ATA_DFLAG_LBA48 ? " LBA48" : " LBA");
1065                 } else {
1066                         /* CHS */
1067
1068                         /* Default translation */
1069                         dev->cylinders  = dev->id[1];
1070                         dev->heads      = dev->id[3];
1071                         dev->sectors    = dev->id[6];
1072
1073                         if (ata_id_current_chs_valid(dev->id)) {
1074                                 /* Current CHS translation is valid. */
1075                                 dev->cylinders = dev->id[54];
1076                                 dev->heads     = dev->id[55];
1077                                 dev->sectors   = dev->id[56];
1078                         }
1079
1080                         /* print device info to dmesg */
1081                         printk(KERN_INFO "ata%u: dev %u ATA-%d, max %s, %Lu sectors: CHS %d/%d/%d\n",
1082                                ap->id, device,
1083                                major_version,
1084                                ata_mode_string(xfer_modes),
1085                                (unsigned long long)dev->n_sectors,
1086                                (int)dev->cylinders, (int)dev->heads, (int)dev->sectors);
1087
1088                 }
1089
1090                 dev->cdb_len = 16;
1091         }
1092
1093         /* ATAPI-specific feature tests */
1094         else if (dev->class == ATA_DEV_ATAPI) {
1095                 if (ata_id_is_ata(dev->id))             /* sanity check */
1096                         goto err_out_nosup;
1097
1098                 rc = atapi_cdb_len(dev->id);
1099                 if ((rc < 12) || (rc > ATAPI_CDB_LEN)) {
1100                         printk(KERN_WARNING "ata%u: unsupported CDB len\n", ap->id);
1101                         goto err_out_nosup;
1102                 }
1103                 dev->cdb_len = (unsigned int) rc;
1104
1105                 /* print device info to dmesg */
1106                 printk(KERN_INFO "ata%u: dev %u ATAPI, max %s\n",
1107                        ap->id, device,
1108                        ata_mode_string(xfer_modes));
1109         }
1110
1111         ap->host->max_cmd_len = 0;
1112         for (i = 0; i < ATA_MAX_DEVICES; i++)
1113                 ap->host->max_cmd_len = max_t(unsigned int,
1114                                               ap->host->max_cmd_len,
1115                                               ap->device[i].cdb_len);
1116
1117         DPRINTK("EXIT, drv_stat = 0x%x\n", ata_chk_status(ap));
1118         return;
1119
1120 err_out_nosup:
1121         printk(KERN_WARNING "ata%u: dev %u not supported, ignoring\n",
1122                ap->id, device);
1123 err_out:
1124         dev->class++;   /* converts ATA_DEV_xxx into ATA_DEV_xxx_UNSUP */
1125         DPRINTK("EXIT, err\n");
1126 }
1127
1128
1129 static inline u8 ata_dev_knobble(const struct ata_port *ap,
1130                                  struct ata_device *dev)
1131 {
1132         return ((ap->cbl == ATA_CBL_SATA) && (!ata_id_is_sata(dev->id)));
1133 }
1134
1135 /**
1136  * ata_dev_config - Run device specific handlers & check for SATA->PATA bridges
1137  * @ap: Bus
1138  * @i:  Device
1139  *
1140  * LOCKING:
1141  */
1142
1143 void ata_dev_config(struct ata_port *ap, unsigned int i)
1144 {
1145         /* limit bridge transfers to udma5, 200 sectors */
1146         if (ata_dev_knobble(ap, &ap->device[i])) {
1147                 printk(KERN_INFO "ata%u(%u): applying bridge limits\n",
1148                        ap->id, i);
1149                 ap->udma_mask &= ATA_UDMA5;
1150                 ap->device[i].max_sectors = ATA_MAX_SECTORS;
1151         }
1152
1153         if (ap->ops->dev_config)
1154                 ap->ops->dev_config(ap, &ap->device[i]);
1155 }
1156
1157 /**
1158  *      ata_bus_probe - Reset and probe ATA bus
1159  *      @ap: Bus to probe
1160  *
1161  *      Master ATA bus probing function.  Initiates a hardware-dependent
1162  *      bus reset, then attempts to identify any devices found on
1163  *      the bus.
1164  *
1165  *      LOCKING:
1166  *      PCI/etc. bus probe sem.
1167  *
1168  *      RETURNS:
1169  *      Zero on success, non-zero on error.
1170  */
1171
1172 static int ata_bus_probe(struct ata_port *ap)
1173 {
1174         unsigned int i, found = 0;
1175
1176         if (ap->ops->probe_reset) {
1177                 unsigned int classes[ATA_MAX_DEVICES];
1178                 int rc;
1179
1180                 ata_port_probe(ap);
1181
1182                 rc = ap->ops->probe_reset(ap, classes);
1183                 if (rc == 0) {
1184                         for (i = 0; i < ATA_MAX_DEVICES; i++) {
1185                                 if (classes[i] == ATA_DEV_UNKNOWN)
1186                                         classes[i] = ATA_DEV_NONE;
1187                                 ap->device[i].class = classes[i];
1188                         }
1189                 } else {
1190                         printk(KERN_ERR "ata%u: probe reset failed, "
1191                                "disabling port\n", ap->id);
1192                         ata_port_disable(ap);
1193                 }
1194         } else
1195                 ap->ops->phy_reset(ap);
1196
1197         if (ap->flags & ATA_FLAG_PORT_DISABLED)
1198                 goto err_out;
1199
1200         for (i = 0; i < ATA_MAX_DEVICES; i++) {
1201                 ata_dev_identify(ap, i);
1202                 if (ata_dev_present(&ap->device[i])) {
1203                         found = 1;
1204                         ata_dev_config(ap,i);
1205                 }
1206         }
1207
1208         if ((!found) || (ap->flags & ATA_FLAG_PORT_DISABLED))
1209                 goto err_out_disable;
1210
1211         ata_set_mode(ap);
1212         if (ap->flags & ATA_FLAG_PORT_DISABLED)
1213                 goto err_out_disable;
1214
1215         return 0;
1216
1217 err_out_disable:
1218         ap->ops->port_disable(ap);
1219 err_out:
1220         return -1;
1221 }
1222
1223 /**
1224  *      ata_port_probe - Mark port as enabled
1225  *      @ap: Port for which we indicate enablement
1226  *
1227  *      Modify @ap data structure such that the system
1228  *      thinks that the entire port is enabled.
1229  *
1230  *      LOCKING: host_set lock, or some other form of
1231  *      serialization.
1232  */
1233
1234 void ata_port_probe(struct ata_port *ap)
1235 {
1236         ap->flags &= ~ATA_FLAG_PORT_DISABLED;
1237 }
1238
1239 /**
1240  *      sata_print_link_status - Print SATA link status
1241  *      @ap: SATA port to printk link status about
1242  *
1243  *      This function prints link speed and status of a SATA link.
1244  *
1245  *      LOCKING:
1246  *      None.
1247  */
1248 static void sata_print_link_status(struct ata_port *ap)
1249 {
1250         u32 sstatus, tmp;
1251         const char *speed;
1252
1253         if (!ap->ops->scr_read)
1254                 return;
1255
1256         sstatus = scr_read(ap, SCR_STATUS);
1257
1258         if (sata_dev_present(ap)) {
1259                 tmp = (sstatus >> 4) & 0xf;
1260                 if (tmp & (1 << 0))
1261                         speed = "1.5";
1262                 else if (tmp & (1 << 1))
1263                         speed = "3.0";
1264                 else
1265                         speed = "<unknown>";
1266                 printk(KERN_INFO "ata%u: SATA link up %s Gbps (SStatus %X)\n",
1267                        ap->id, speed, sstatus);
1268         } else {
1269                 printk(KERN_INFO "ata%u: SATA link down (SStatus %X)\n",
1270                        ap->id, sstatus);
1271         }
1272 }
1273
1274 /**
1275  *      __sata_phy_reset - Wake/reset a low-level SATA PHY
1276  *      @ap: SATA port associated with target SATA PHY.
1277  *
1278  *      This function issues commands to standard SATA Sxxx
1279  *      PHY registers, to wake up the phy (and device), and
1280  *      clear any reset condition.
1281  *
1282  *      LOCKING:
1283  *      PCI/etc. bus probe sem.
1284  *
1285  */
1286 void __sata_phy_reset(struct ata_port *ap)
1287 {
1288         u32 sstatus;
1289         unsigned long timeout = jiffies + (HZ * 5);
1290
1291         if (ap->flags & ATA_FLAG_SATA_RESET) {
1292                 /* issue phy wake/reset */
1293                 scr_write_flush(ap, SCR_CONTROL, 0x301);
1294                 /* Couldn't find anything in SATA I/II specs, but
1295                  * AHCI-1.1 10.4.2 says at least 1 ms. */
1296                 mdelay(1);
1297         }
1298         scr_write_flush(ap, SCR_CONTROL, 0x300); /* phy wake/clear reset */
1299
1300         /* wait for phy to become ready, if necessary */
1301         do {
1302                 msleep(200);
1303                 sstatus = scr_read(ap, SCR_STATUS);
1304                 if ((sstatus & 0xf) != 1)
1305                         break;
1306         } while (time_before(jiffies, timeout));
1307
1308         /* print link status */
1309         sata_print_link_status(ap);
1310
1311         /* TODO: phy layer with polling, timeouts, etc. */
1312         if (sata_dev_present(ap))
1313                 ata_port_probe(ap);
1314         else
1315                 ata_port_disable(ap);
1316
1317         if (ap->flags & ATA_FLAG_PORT_DISABLED)
1318                 return;
1319
1320         if (ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT)) {
1321                 ata_port_disable(ap);
1322                 return;
1323         }
1324
1325         ap->cbl = ATA_CBL_SATA;
1326 }
1327
1328 /**
1329  *      sata_phy_reset - Reset SATA bus.
1330  *      @ap: SATA port associated with target SATA PHY.
1331  *
1332  *      This function resets the SATA bus, and then probes
1333  *      the bus for devices.
1334  *
1335  *      LOCKING:
1336  *      PCI/etc. bus probe sem.
1337  *
1338  */
1339 void sata_phy_reset(struct ata_port *ap)
1340 {
1341         __sata_phy_reset(ap);
1342         if (ap->flags & ATA_FLAG_PORT_DISABLED)
1343                 return;
1344         ata_bus_reset(ap);
1345 }
1346
1347 /**
1348  *      ata_port_disable - Disable port.
1349  *      @ap: Port to be disabled.
1350  *
1351  *      Modify @ap data structure such that the system
1352  *      thinks that the entire port is disabled, and should
1353  *      never attempt to probe or communicate with devices
1354  *      on this port.
1355  *
1356  *      LOCKING: host_set lock, or some other form of
1357  *      serialization.
1358  */
1359
1360 void ata_port_disable(struct ata_port *ap)
1361 {
1362         ap->device[0].class = ATA_DEV_NONE;
1363         ap->device[1].class = ATA_DEV_NONE;
1364         ap->flags |= ATA_FLAG_PORT_DISABLED;
1365 }
1366
1367 /*
1368  * This mode timing computation functionality is ported over from
1369  * drivers/ide/ide-timing.h and was originally written by Vojtech Pavlik
1370  */
1371 /*
1372  * PIO 0-5, MWDMA 0-2 and UDMA 0-6 timings (in nanoseconds).
1373  * These were taken from ATA/ATAPI-6 standard, rev 0a, except
1374  * for PIO 5, which is a nonstandard extension and UDMA6, which
1375  * is currently supported only by Maxtor drives.
1376  */
1377
1378 static const struct ata_timing ata_timing[] = {
1379
1380         { XFER_UDMA_6,     0,   0,   0,   0,   0,   0,   0,  15 },
1381         { XFER_UDMA_5,     0,   0,   0,   0,   0,   0,   0,  20 },
1382         { XFER_UDMA_4,     0,   0,   0,   0,   0,   0,   0,  30 },
1383         { XFER_UDMA_3,     0,   0,   0,   0,   0,   0,   0,  45 },
1384
1385         { XFER_UDMA_2,     0,   0,   0,   0,   0,   0,   0,  60 },
1386         { XFER_UDMA_1,     0,   0,   0,   0,   0,   0,   0,  80 },
1387         { XFER_UDMA_0,     0,   0,   0,   0,   0,   0,   0, 120 },
1388
1389 /*      { XFER_UDMA_SLOW,  0,   0,   0,   0,   0,   0,   0, 150 }, */
1390
1391         { XFER_MW_DMA_2,  25,   0,   0,   0,  70,  25, 120,   0 },
1392         { XFER_MW_DMA_1,  45,   0,   0,   0,  80,  50, 150,   0 },
1393         { XFER_MW_DMA_0,  60,   0,   0,   0, 215, 215, 480,   0 },
1394
1395         { XFER_SW_DMA_2,  60,   0,   0,   0, 120, 120, 240,   0 },
1396         { XFER_SW_DMA_1,  90,   0,   0,   0, 240, 240, 480,   0 },
1397         { XFER_SW_DMA_0, 120,   0,   0,   0, 480, 480, 960,   0 },
1398
1399 /*      { XFER_PIO_5,     20,  50,  30, 100,  50,  30, 100,   0 }, */
1400         { XFER_PIO_4,     25,  70,  25, 120,  70,  25, 120,   0 },
1401         { XFER_PIO_3,     30,  80,  70, 180,  80,  70, 180,   0 },
1402
1403         { XFER_PIO_2,     30, 290,  40, 330, 100,  90, 240,   0 },
1404         { XFER_PIO_1,     50, 290,  93, 383, 125, 100, 383,   0 },
1405         { XFER_PIO_0,     70, 290, 240, 600, 165, 150, 600,   0 },
1406
1407 /*      { XFER_PIO_SLOW, 120, 290, 240, 960, 290, 240, 960,   0 }, */
1408
1409         { 0xFF }
1410 };
1411
1412 #define ENOUGH(v,unit)          (((v)-1)/(unit)+1)
1413 #define EZ(v,unit)              ((v)?ENOUGH(v,unit):0)
1414
1415 static void ata_timing_quantize(const struct ata_timing *t, struct ata_timing *q, int T, int UT)
1416 {
1417         q->setup   = EZ(t->setup   * 1000,  T);
1418         q->act8b   = EZ(t->act8b   * 1000,  T);
1419         q->rec8b   = EZ(t->rec8b   * 1000,  T);
1420         q->cyc8b   = EZ(t->cyc8b   * 1000,  T);
1421         q->active  = EZ(t->active  * 1000,  T);
1422         q->recover = EZ(t->recover * 1000,  T);
1423         q->cycle   = EZ(t->cycle   * 1000,  T);
1424         q->udma    = EZ(t->udma    * 1000, UT);
1425 }
1426
1427 void ata_timing_merge(const struct ata_timing *a, const struct ata_timing *b,
1428                       struct ata_timing *m, unsigned int what)
1429 {
1430         if (what & ATA_TIMING_SETUP  ) m->setup   = max(a->setup,   b->setup);
1431         if (what & ATA_TIMING_ACT8B  ) m->act8b   = max(a->act8b,   b->act8b);
1432         if (what & ATA_TIMING_REC8B  ) m->rec8b   = max(a->rec8b,   b->rec8b);
1433         if (what & ATA_TIMING_CYC8B  ) m->cyc8b   = max(a->cyc8b,   b->cyc8b);
1434         if (what & ATA_TIMING_ACTIVE ) m->active  = max(a->active,  b->active);
1435         if (what & ATA_TIMING_RECOVER) m->recover = max(a->recover, b->recover);
1436         if (what & ATA_TIMING_CYCLE  ) m->cycle   = max(a->cycle,   b->cycle);
1437         if (what & ATA_TIMING_UDMA   ) m->udma    = max(a->udma,    b->udma);
1438 }
1439
1440 static const struct ata_timing* ata_timing_find_mode(unsigned short speed)
1441 {
1442         const struct ata_timing *t;
1443
1444         for (t = ata_timing; t->mode != speed; t++)
1445                 if (t->mode == 0xFF)
1446                         return NULL;
1447         return t;
1448 }
1449
1450 int ata_timing_compute(struct ata_device *adev, unsigned short speed,
1451                        struct ata_timing *t, int T, int UT)
1452 {
1453         const struct ata_timing *s;
1454         struct ata_timing p;
1455
1456         /*
1457          * Find the mode.
1458          */
1459
1460         if (!(s = ata_timing_find_mode(speed)))
1461                 return -EINVAL;
1462
1463         memcpy(t, s, sizeof(*s));
1464
1465         /*
1466          * If the drive is an EIDE drive, it can tell us it needs extended
1467          * PIO/MW_DMA cycle timing.
1468          */
1469
1470         if (adev->id[ATA_ID_FIELD_VALID] & 2) { /* EIDE drive */
1471                 memset(&p, 0, sizeof(p));
1472                 if(speed >= XFER_PIO_0 && speed <= XFER_SW_DMA_0) {
1473                         if (speed <= XFER_PIO_2) p.cycle = p.cyc8b = adev->id[ATA_ID_EIDE_PIO];
1474                                             else p.cycle = p.cyc8b = adev->id[ATA_ID_EIDE_PIO_IORDY];
1475                 } else if(speed >= XFER_MW_DMA_0 && speed <= XFER_MW_DMA_2) {
1476                         p.cycle = adev->id[ATA_ID_EIDE_DMA_MIN];
1477                 }
1478                 ata_timing_merge(&p, t, t, ATA_TIMING_CYCLE | ATA_TIMING_CYC8B);
1479         }
1480
1481         /*
1482          * Convert the timing to bus clock counts.
1483          */
1484
1485         ata_timing_quantize(t, t, T, UT);
1486
1487         /*
1488          * Even in DMA/UDMA modes we still use PIO access for IDENTIFY,
1489          * S.M.A.R.T * and some other commands. We have to ensure that the
1490          * DMA cycle timing is slower/equal than the fastest PIO timing.
1491          */
1492
1493         if (speed > XFER_PIO_4) {
1494                 ata_timing_compute(adev, adev->pio_mode, &p, T, UT);
1495                 ata_timing_merge(&p, t, t, ATA_TIMING_ALL);
1496         }
1497
1498         /*
1499          * Lengthen active & recovery time so that cycle time is correct.
1500          */
1501
1502         if (t->act8b + t->rec8b < t->cyc8b) {
1503                 t->act8b += (t->cyc8b - (t->act8b + t->rec8b)) / 2;
1504                 t->rec8b = t->cyc8b - t->act8b;
1505         }
1506
1507         if (t->active + t->recover < t->cycle) {
1508                 t->active += (t->cycle - (t->active + t->recover)) / 2;
1509                 t->recover = t->cycle - t->active;
1510         }
1511
1512         return 0;
1513 }
1514
1515 static const struct {
1516         unsigned int shift;
1517         u8 base;
1518 } xfer_mode_classes[] = {
1519         { ATA_SHIFT_UDMA,       XFER_UDMA_0 },
1520         { ATA_SHIFT_MWDMA,      XFER_MW_DMA_0 },
1521         { ATA_SHIFT_PIO,        XFER_PIO_0 },
1522 };
1523
1524 static u8 base_from_shift(unsigned int shift)
1525 {
1526         int i;
1527
1528         for (i = 0; i < ARRAY_SIZE(xfer_mode_classes); i++)
1529                 if (xfer_mode_classes[i].shift == shift)
1530                         return xfer_mode_classes[i].base;
1531
1532         return 0xff;
1533 }
1534
1535 static void ata_dev_set_mode(struct ata_port *ap, struct ata_device *dev)
1536 {
1537         int ofs, idx;
1538         u8 base;
1539
1540         if (!ata_dev_present(dev) || (ap->flags & ATA_FLAG_PORT_DISABLED))
1541                 return;
1542
1543         if (dev->xfer_shift == ATA_SHIFT_PIO)
1544                 dev->flags |= ATA_DFLAG_PIO;
1545
1546         ata_dev_set_xfermode(ap, dev);
1547
1548         base = base_from_shift(dev->xfer_shift);
1549         ofs = dev->xfer_mode - base;
1550         idx = ofs + dev->xfer_shift;
1551         WARN_ON(idx >= ARRAY_SIZE(xfer_mode_str));
1552
1553         DPRINTK("idx=%d xfer_shift=%u, xfer_mode=0x%x, base=0x%x, offset=%d\n",
1554                 idx, dev->xfer_shift, (int)dev->xfer_mode, (int)base, ofs);
1555
1556         printk(KERN_INFO "ata%u: dev %u configured for %s\n",
1557                 ap->id, dev->devno, xfer_mode_str[idx]);
1558 }
1559
1560 static int ata_host_set_pio(struct ata_port *ap)
1561 {
1562         unsigned int mask;
1563         int x, i;
1564         u8 base, xfer_mode;
1565
1566         mask = ata_get_mode_mask(ap, ATA_SHIFT_PIO);
1567         x = fgb(mask);
1568         if (x < 0) {
1569                 printk(KERN_WARNING "ata%u: no PIO support\n", ap->id);
1570                 return -1;
1571         }
1572
1573         base = base_from_shift(ATA_SHIFT_PIO);
1574         xfer_mode = base + x;
1575
1576         DPRINTK("base 0x%x xfer_mode 0x%x mask 0x%x x %d\n",
1577                 (int)base, (int)xfer_mode, mask, x);
1578
1579         for (i = 0; i < ATA_MAX_DEVICES; i++) {
1580                 struct ata_device *dev = &ap->device[i];
1581                 if (ata_dev_present(dev)) {
1582                         dev->pio_mode = xfer_mode;
1583                         dev->xfer_mode = xfer_mode;
1584                         dev->xfer_shift = ATA_SHIFT_PIO;
1585                         if (ap->ops->set_piomode)
1586                                 ap->ops->set_piomode(ap, dev);
1587                 }
1588         }
1589
1590         return 0;
1591 }
1592
1593 static void ata_host_set_dma(struct ata_port *ap, u8 xfer_mode,
1594                             unsigned int xfer_shift)
1595 {
1596         int i;
1597
1598         for (i = 0; i < ATA_MAX_DEVICES; i++) {
1599                 struct ata_device *dev = &ap->device[i];
1600                 if (ata_dev_present(dev)) {
1601                         dev->dma_mode = xfer_mode;
1602                         dev->xfer_mode = xfer_mode;
1603                         dev->xfer_shift = xfer_shift;
1604                         if (ap->ops->set_dmamode)
1605                                 ap->ops->set_dmamode(ap, dev);
1606                 }
1607         }
1608 }
1609
1610 /**
1611  *      ata_set_mode - Program timings and issue SET FEATURES - XFER
1612  *      @ap: port on which timings will be programmed
1613  *
1614  *      Set ATA device disk transfer mode (PIO3, UDMA6, etc.).
1615  *
1616  *      LOCKING:
1617  *      PCI/etc. bus probe sem.
1618  */
1619 static void ata_set_mode(struct ata_port *ap)
1620 {
1621         unsigned int xfer_shift;
1622         u8 xfer_mode;
1623         int rc;
1624
1625         /* step 1: always set host PIO timings */
1626         rc = ata_host_set_pio(ap);
1627         if (rc)
1628                 goto err_out;
1629
1630         /* step 2: choose the best data xfer mode */
1631         xfer_mode = xfer_shift = 0;
1632         rc = ata_choose_xfer_mode(ap, &xfer_mode, &xfer_shift);
1633         if (rc)
1634                 goto err_out;
1635
1636         /* step 3: if that xfer mode isn't PIO, set host DMA timings */
1637         if (xfer_shift != ATA_SHIFT_PIO)
1638                 ata_host_set_dma(ap, xfer_mode, xfer_shift);
1639
1640         /* step 4: update devices' xfer mode */
1641         ata_dev_set_mode(ap, &ap->device[0]);
1642         ata_dev_set_mode(ap, &ap->device[1]);
1643
1644         if (ap->flags & ATA_FLAG_PORT_DISABLED)
1645                 return;
1646
1647         if (ap->ops->post_set_mode)
1648                 ap->ops->post_set_mode(ap);
1649
1650         return;
1651
1652 err_out:
1653         ata_port_disable(ap);
1654 }
1655
1656 /**
1657  *      ata_tf_to_host - issue ATA taskfile to host controller
1658  *      @ap: port to which command is being issued
1659  *      @tf: ATA taskfile register set
1660  *
1661  *      Issues ATA taskfile register set to ATA host controller,
1662  *      with proper synchronization with interrupt handler and
1663  *      other threads.
1664  *
1665  *      LOCKING:
1666  *      spin_lock_irqsave(host_set lock)
1667  */
1668
1669 static inline void ata_tf_to_host(struct ata_port *ap,
1670                                   const struct ata_taskfile *tf)
1671 {
1672         ap->ops->tf_load(ap, tf);
1673         ap->ops->exec_command(ap, tf);
1674 }
1675
1676 /**
1677  *      ata_busy_sleep - sleep until BSY clears, or timeout
1678  *      @ap: port containing status register to be polled
1679  *      @tmout_pat: impatience timeout
1680  *      @tmout: overall timeout
1681  *
1682  *      Sleep until ATA Status register bit BSY clears,
1683  *      or a timeout occurs.
1684  *
1685  *      LOCKING: None.
1686  */
1687
1688 unsigned int ata_busy_sleep (struct ata_port *ap,
1689                              unsigned long tmout_pat, unsigned long tmout)
1690 {
1691         unsigned long timer_start, timeout;
1692         u8 status;
1693
1694         status = ata_busy_wait(ap, ATA_BUSY, 300);
1695         timer_start = jiffies;
1696         timeout = timer_start + tmout_pat;
1697         while ((status & ATA_BUSY) && (time_before(jiffies, timeout))) {
1698                 msleep(50);
1699                 status = ata_busy_wait(ap, ATA_BUSY, 3);
1700         }
1701
1702         if (status & ATA_BUSY)
1703                 printk(KERN_WARNING "ata%u is slow to respond, "
1704                        "please be patient\n", ap->id);
1705
1706         timeout = timer_start + tmout;
1707         while ((status & ATA_BUSY) && (time_before(jiffies, timeout))) {
1708                 msleep(50);
1709                 status = ata_chk_status(ap);
1710         }
1711
1712         if (status & ATA_BUSY) {
1713                 printk(KERN_ERR "ata%u failed to respond (%lu secs)\n",
1714                        ap->id, tmout / HZ);
1715                 return 1;
1716         }
1717
1718         return 0;
1719 }
1720
1721 static void ata_bus_post_reset(struct ata_port *ap, unsigned int devmask)
1722 {
1723         struct ata_ioports *ioaddr = &ap->ioaddr;
1724         unsigned int dev0 = devmask & (1 << 0);
1725         unsigned int dev1 = devmask & (1 << 1);
1726         unsigned long timeout;
1727
1728         /* if device 0 was found in ata_devchk, wait for its
1729          * BSY bit to clear
1730          */
1731         if (dev0)
1732                 ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
1733
1734         /* if device 1 was found in ata_devchk, wait for
1735          * register access, then wait for BSY to clear
1736          */
1737         timeout = jiffies + ATA_TMOUT_BOOT;
1738         while (dev1) {
1739                 u8 nsect, lbal;
1740
1741                 ap->ops->dev_select(ap, 1);
1742                 if (ap->flags & ATA_FLAG_MMIO) {
1743                         nsect = readb((void __iomem *) ioaddr->nsect_addr);
1744                         lbal = readb((void __iomem *) ioaddr->lbal_addr);
1745                 } else {
1746                         nsect = inb(ioaddr->nsect_addr);
1747                         lbal = inb(ioaddr->lbal_addr);
1748                 }
1749                 if ((nsect == 1) && (lbal == 1))
1750                         break;
1751                 if (time_after(jiffies, timeout)) {
1752                         dev1 = 0;
1753                         break;
1754                 }
1755                 msleep(50);     /* give drive a breather */
1756         }
1757         if (dev1)
1758                 ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
1759
1760         /* is all this really necessary? */
1761         ap->ops->dev_select(ap, 0);
1762         if (dev1)
1763                 ap->ops->dev_select(ap, 1);
1764         if (dev0)
1765                 ap->ops->dev_select(ap, 0);
1766 }
1767
1768 /**
1769  *      ata_bus_edd - Issue EXECUTE DEVICE DIAGNOSTIC command.
1770  *      @ap: Port to reset and probe
1771  *
1772  *      Use the EXECUTE DEVICE DIAGNOSTIC command to reset and
1773  *      probe the bus.  Not often used these days.
1774  *
1775  *      LOCKING:
1776  *      PCI/etc. bus probe sem.
1777  *      Obtains host_set lock.
1778  *
1779  */
1780
1781 static unsigned int ata_bus_edd(struct ata_port *ap)
1782 {
1783         struct ata_taskfile tf;
1784         unsigned long flags;
1785
1786         /* set up execute-device-diag (bus reset) taskfile */
1787         /* also, take interrupts to a known state (disabled) */
1788         DPRINTK("execute-device-diag\n");
1789         ata_tf_init(ap, &tf, 0);
1790         tf.ctl |= ATA_NIEN;
1791         tf.command = ATA_CMD_EDD;
1792         tf.protocol = ATA_PROT_NODATA;
1793
1794         /* do bus reset */
1795         spin_lock_irqsave(&ap->host_set->lock, flags);
1796         ata_tf_to_host(ap, &tf);
1797         spin_unlock_irqrestore(&ap->host_set->lock, flags);
1798
1799         /* spec says at least 2ms.  but who knows with those
1800          * crazy ATAPI devices...
1801          */
1802         msleep(150);
1803
1804         return ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
1805 }
1806
1807 static unsigned int ata_bus_softreset(struct ata_port *ap,
1808                                       unsigned int devmask)
1809 {
1810         struct ata_ioports *ioaddr = &ap->ioaddr;
1811
1812         DPRINTK("ata%u: bus reset via SRST\n", ap->id);
1813
1814         /* software reset.  causes dev0 to be selected */
1815         if (ap->flags & ATA_FLAG_MMIO) {
1816                 writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
1817                 udelay(20);     /* FIXME: flush */
1818                 writeb(ap->ctl | ATA_SRST, (void __iomem *) ioaddr->ctl_addr);
1819                 udelay(20);     /* FIXME: flush */
1820                 writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
1821         } else {
1822                 outb(ap->ctl, ioaddr->ctl_addr);
1823                 udelay(10);
1824                 outb(ap->ctl | ATA_SRST, ioaddr->ctl_addr);
1825                 udelay(10);
1826                 outb(ap->ctl, ioaddr->ctl_addr);
1827         }
1828
1829         /* spec mandates ">= 2ms" before checking status.
1830          * We wait 150ms, because that was the magic delay used for
1831          * ATAPI devices in Hale Landis's ATADRVR, for the period of time
1832          * between when the ATA command register is written, and then
1833          * status is checked.  Because waiting for "a while" before
1834          * checking status is fine, post SRST, we perform this magic
1835          * delay here as well.
1836          */
1837         msleep(150);
1838
1839         ata_bus_post_reset(ap, devmask);
1840
1841         return 0;
1842 }
1843
1844 /**
1845  *      ata_bus_reset - reset host port and associated ATA channel
1846  *      @ap: port to reset
1847  *
1848  *      This is typically the first time we actually start issuing
1849  *      commands to the ATA channel.  We wait for BSY to clear, then
1850  *      issue EXECUTE DEVICE DIAGNOSTIC command, polling for its
1851  *      result.  Determine what devices, if any, are on the channel
1852  *      by looking at the device 0/1 error register.  Look at the signature
1853  *      stored in each device's taskfile registers, to determine if
1854  *      the device is ATA or ATAPI.
1855  *
1856  *      LOCKING:
1857  *      PCI/etc. bus probe sem.
1858  *      Obtains host_set lock.
1859  *
1860  *      SIDE EFFECTS:
1861  *      Sets ATA_FLAG_PORT_DISABLED if bus reset fails.
1862  */
1863
1864 void ata_bus_reset(struct ata_port *ap)
1865 {
1866         struct ata_ioports *ioaddr = &ap->ioaddr;
1867         unsigned int slave_possible = ap->flags & ATA_FLAG_SLAVE_POSS;
1868         u8 err;
1869         unsigned int dev0, dev1 = 0, rc = 0, devmask = 0;
1870
1871         DPRINTK("ENTER, host %u, port %u\n", ap->id, ap->port_no);
1872
1873         /* determine if device 0/1 are present */
1874         if (ap->flags & ATA_FLAG_SATA_RESET)
1875                 dev0 = 1;
1876         else {
1877                 dev0 = ata_devchk(ap, 0);
1878                 if (slave_possible)
1879                         dev1 = ata_devchk(ap, 1);
1880         }
1881
1882         if (dev0)
1883                 devmask |= (1 << 0);
1884         if (dev1)
1885                 devmask |= (1 << 1);
1886
1887         /* select device 0 again */
1888         ap->ops->dev_select(ap, 0);
1889
1890         /* issue bus reset */
1891         if (ap->flags & ATA_FLAG_SRST)
1892                 rc = ata_bus_softreset(ap, devmask);
1893         else if ((ap->flags & ATA_FLAG_SATA_RESET) == 0) {
1894                 /* set up device control */
1895                 if (ap->flags & ATA_FLAG_MMIO)
1896                         writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
1897                 else
1898                         outb(ap->ctl, ioaddr->ctl_addr);
1899                 rc = ata_bus_edd(ap);
1900         }
1901
1902         if (rc)
1903                 goto err_out;
1904
1905         /*
1906          * determine by signature whether we have ATA or ATAPI devices
1907          */
1908         ap->device[0].class = ata_dev_try_classify(ap, 0, &err);
1909         if ((slave_possible) && (err != 0x81))
1910                 ap->device[1].class = ata_dev_try_classify(ap, 1, &err);
1911
1912         /* re-enable interrupts */
1913         if (ap->ioaddr.ctl_addr)        /* FIXME: hack. create a hook instead */
1914                 ata_irq_on(ap);
1915
1916         /* is double-select really necessary? */
1917         if (ap->device[1].class != ATA_DEV_NONE)
1918                 ap->ops->dev_select(ap, 1);
1919         if (ap->device[0].class != ATA_DEV_NONE)
1920                 ap->ops->dev_select(ap, 0);
1921
1922         /* if no devices were detected, disable this port */
1923         if ((ap->device[0].class == ATA_DEV_NONE) &&
1924             (ap->device[1].class == ATA_DEV_NONE))
1925                 goto err_out;
1926
1927         if (ap->flags & (ATA_FLAG_SATA_RESET | ATA_FLAG_SRST)) {
1928                 /* set up device control for ATA_FLAG_SATA_RESET */
1929                 if (ap->flags & ATA_FLAG_MMIO)
1930                         writeb(ap->ctl, (void __iomem *) ioaddr->ctl_addr);
1931                 else
1932                         outb(ap->ctl, ioaddr->ctl_addr);
1933         }
1934
1935         DPRINTK("EXIT\n");
1936         return;
1937
1938 err_out:
1939         printk(KERN_ERR "ata%u: disabling port\n", ap->id);
1940         ap->ops->port_disable(ap);
1941
1942         DPRINTK("EXIT\n");
1943 }
1944
1945 static int sata_phy_resume(struct ata_port *ap)
1946 {
1947         unsigned long timeout = jiffies + (HZ * 5);
1948         u32 sstatus;
1949
1950         scr_write_flush(ap, SCR_CONTROL, 0x300);
1951
1952         /* Wait for phy to become ready, if necessary. */
1953         do {
1954                 msleep(200);
1955                 sstatus = scr_read(ap, SCR_STATUS);
1956                 if ((sstatus & 0xf) != 1)
1957                         return 0;
1958         } while (time_before(jiffies, timeout));
1959
1960         return -1;
1961 }
1962
1963 /**
1964  *      ata_std_probeinit - initialize probing
1965  *      @ap: port to be probed
1966  *
1967  *      @ap is about to be probed.  Initialize it.  This function is
1968  *      to be used as standard callback for ata_drive_probe_reset().
1969  *
1970  *      NOTE!!! Do not use this function as probeinit if a low level
1971  *      driver implements only hardreset.  Just pass NULL as probeinit
1972  *      in that case.  Using this function is probably okay but doing
1973  *      so makes reset sequence different from the original
1974  *      ->phy_reset implementation and Jeff nervous.  :-P
1975  */
1976 extern void ata_std_probeinit(struct ata_port *ap)
1977 {
1978         if (ap->flags & ATA_FLAG_SATA && ap->ops->scr_read) {
1979                 sata_phy_resume(ap);
1980                 if (sata_dev_present(ap))
1981                         ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT);
1982         }
1983 }
1984
1985 /**
1986  *      ata_std_softreset - reset host port via ATA SRST
1987  *      @ap: port to reset
1988  *      @verbose: fail verbosely
1989  *      @classes: resulting classes of attached devices
1990  *
1991  *      Reset host port using ATA SRST.  This function is to be used
1992  *      as standard callback for ata_drive_*_reset() functions.
1993  *
1994  *      LOCKING:
1995  *      Kernel thread context (may sleep)
1996  *
1997  *      RETURNS:
1998  *      0 on success, -errno otherwise.
1999  */
2000 int ata_std_softreset(struct ata_port *ap, int verbose, unsigned int *classes)
2001 {
2002         unsigned int slave_possible = ap->flags & ATA_FLAG_SLAVE_POSS;
2003         unsigned int devmask = 0, err_mask;
2004         u8 err;
2005
2006         DPRINTK("ENTER\n");
2007
2008         if (ap->ops->scr_read && !sata_dev_present(ap)) {
2009                 classes[0] = ATA_DEV_NONE;
2010                 goto out;
2011         }
2012
2013         /* determine if device 0/1 are present */
2014         if (ata_devchk(ap, 0))
2015                 devmask |= (1 << 0);
2016         if (slave_possible && ata_devchk(ap, 1))
2017                 devmask |= (1 << 1);
2018
2019         /* select device 0 again */
2020         ap->ops->dev_select(ap, 0);
2021
2022         /* issue bus reset */
2023         DPRINTK("about to softreset, devmask=%x\n", devmask);
2024         err_mask = ata_bus_softreset(ap, devmask);
2025         if (err_mask) {
2026                 if (verbose)
2027                         printk(KERN_ERR "ata%u: SRST failed (err_mask=0x%x)\n",
2028                                ap->id, err_mask);
2029                 else
2030                         DPRINTK("EXIT, softreset failed (err_mask=0x%x)\n",
2031                                 err_mask);
2032                 return -EIO;
2033         }
2034
2035         /* determine by signature whether we have ATA or ATAPI devices */
2036         classes[0] = ata_dev_try_classify(ap, 0, &err);
2037         if (slave_possible && err != 0x81)
2038                 classes[1] = ata_dev_try_classify(ap, 1, &err);
2039
2040  out:
2041         DPRINTK("EXIT, classes[0]=%u [1]=%u\n", classes[0], classes[1]);
2042         return 0;
2043 }
2044
2045 /**
2046  *      sata_std_hardreset - reset host port via SATA phy reset
2047  *      @ap: port to reset
2048  *      @verbose: fail verbosely
2049  *      @class: resulting class of attached device
2050  *
2051  *      SATA phy-reset host port using DET bits of SControl register.
2052  *      This function is to be used as standard callback for
2053  *      ata_drive_*_reset().
2054  *
2055  *      LOCKING:
2056  *      Kernel thread context (may sleep)
2057  *
2058  *      RETURNS:
2059  *      0 on success, -errno otherwise.
2060  */
2061 int sata_std_hardreset(struct ata_port *ap, int verbose, unsigned int *class)
2062 {
2063         DPRINTK("ENTER\n");
2064
2065         /* Issue phy wake/reset */
2066         scr_write_flush(ap, SCR_CONTROL, 0x301);
2067
2068         /*
2069          * Couldn't find anything in SATA I/II specs, but AHCI-1.1
2070          * 10.4.2 says at least 1 ms.
2071          */
2072         msleep(1);
2073
2074         /* Bring phy back */
2075         sata_phy_resume(ap);
2076
2077         /* TODO: phy layer with polling, timeouts, etc. */
2078         if (!sata_dev_present(ap)) {
2079                 *class = ATA_DEV_NONE;
2080                 DPRINTK("EXIT, link offline\n");
2081                 return 0;
2082         }
2083
2084         if (ata_busy_sleep(ap, ATA_TMOUT_BOOT_QUICK, ATA_TMOUT_BOOT)) {
2085                 if (verbose)
2086                         printk(KERN_ERR "ata%u: COMRESET failed "
2087                                "(device not ready)\n", ap->id);
2088                 else
2089                         DPRINTK("EXIT, device not ready\n");
2090                 return -EIO;
2091         }
2092
2093         ap->ops->dev_select(ap, 0);     /* probably unnecessary */
2094
2095         *class = ata_dev_try_classify(ap, 0, NULL);
2096
2097         DPRINTK("EXIT, class=%u\n", *class);
2098         return 0;
2099 }
2100
2101 /**
2102  *      ata_std_postreset - standard postreset callback
2103  *      @ap: the target ata_port
2104  *      @classes: classes of attached devices
2105  *
2106  *      This function is invoked after a successful reset.  Note that
2107  *      the device might have been reset more than once using
2108  *      different reset methods before postreset is invoked.
2109  *      postreset is also reponsible for setting cable type.
2110  *
2111  *      This function is to be used as standard callback for
2112  *      ata_drive_*_reset().
2113  *
2114  *      LOCKING:
2115  *      Kernel thread context (may sleep)
2116  */
2117 void ata_std_postreset(struct ata_port *ap, unsigned int *classes)
2118 {
2119         DPRINTK("ENTER\n");
2120
2121         /* set cable type */
2122         if (ap->cbl == ATA_CBL_NONE && ap->flags & ATA_FLAG_SATA)
2123                 ap->cbl = ATA_CBL_SATA;
2124
2125         /* print link status */
2126         if (ap->cbl == ATA_CBL_SATA)
2127                 sata_print_link_status(ap);
2128
2129         /* re-enable interrupts */
2130         if (ap->ioaddr.ctl_addr)        /* FIXME: hack. create a hook instead */
2131                 ata_irq_on(ap);
2132
2133         /* is double-select really necessary? */
2134         if (classes[0] != ATA_DEV_NONE)
2135                 ap->ops->dev_select(ap, 1);
2136         if (classes[1] != ATA_DEV_NONE)
2137                 ap->ops->dev_select(ap, 0);
2138
2139         /* bail out if no device is present */
2140         if (classes[0] == ATA_DEV_NONE && classes[1] == ATA_DEV_NONE) {
2141                 DPRINTK("EXIT, no device\n");
2142                 return;
2143         }
2144
2145         /* set up device control */
2146         if (ap->ioaddr.ctl_addr) {
2147                 if (ap->flags & ATA_FLAG_MMIO)
2148                         writeb(ap->ctl, (void __iomem *) ap->ioaddr.ctl_addr);
2149                 else
2150                         outb(ap->ctl, ap->ioaddr.ctl_addr);
2151         }
2152
2153         DPRINTK("EXIT\n");
2154 }
2155
2156 /**
2157  *      ata_std_probe_reset - standard probe reset method
2158  *      @ap: prot to perform probe-reset
2159  *      @classes: resulting classes of attached devices
2160  *
2161  *      The stock off-the-shelf ->probe_reset method.
2162  *
2163  *      LOCKING:
2164  *      Kernel thread context (may sleep)
2165  *
2166  *      RETURNS:
2167  *      0 on success, -errno otherwise.
2168  */
2169 int ata_std_probe_reset(struct ata_port *ap, unsigned int *classes)
2170 {
2171         ata_reset_fn_t hardreset;
2172
2173         hardreset = NULL;
2174         if (ap->flags & ATA_FLAG_SATA && ap->ops->scr_read)
2175                 hardreset = sata_std_hardreset;
2176
2177         return ata_drive_probe_reset(ap, ata_std_probeinit,
2178                                      ata_std_softreset, hardreset,
2179                                      ata_std_postreset, classes);
2180 }
2181
2182 static int do_probe_reset(struct ata_port *ap, ata_reset_fn_t reset,
2183                           ata_postreset_fn_t postreset,
2184                           unsigned int *classes)
2185 {
2186         int i, rc;
2187
2188         for (i = 0; i < ATA_MAX_DEVICES; i++)
2189                 classes[i] = ATA_DEV_UNKNOWN;
2190
2191         rc = reset(ap, 0, classes);
2192         if (rc)
2193                 return rc;
2194
2195         /* If any class isn't ATA_DEV_UNKNOWN, consider classification
2196          * is complete and convert all ATA_DEV_UNKNOWN to
2197          * ATA_DEV_NONE.
2198          */
2199         for (i = 0; i < ATA_MAX_DEVICES; i++)
2200                 if (classes[i] != ATA_DEV_UNKNOWN)
2201                         break;
2202
2203         if (i < ATA_MAX_DEVICES)
2204                 for (i = 0; i < ATA_MAX_DEVICES; i++)
2205                         if (classes[i] == ATA_DEV_UNKNOWN)
2206                                 classes[i] = ATA_DEV_NONE;
2207
2208         if (postreset)
2209                 postreset(ap, classes);
2210
2211         return classes[0] != ATA_DEV_UNKNOWN ? 0 : -ENODEV;
2212 }
2213
2214 /**
2215  *      ata_drive_probe_reset - Perform probe reset with given methods
2216  *      @ap: port to reset
2217  *      @probeinit: probeinit method (can be NULL)
2218  *      @softreset: softreset method (can be NULL)
2219  *      @hardreset: hardreset method (can be NULL)
2220  *      @postreset: postreset method (can be NULL)
2221  *      @classes: resulting classes of attached devices
2222  *
2223  *      Reset the specified port and classify attached devices using
2224  *      given methods.  This function prefers softreset but tries all
2225  *      possible reset sequences to reset and classify devices.  This
2226  *      function is intended to be used for constructing ->probe_reset
2227  *      callback by low level drivers.
2228  *
2229  *      Reset methods should follow the following rules.
2230  *
2231  *      - Return 0 on sucess, -errno on failure.
2232  *      - If classification is supported, fill classes[] with
2233  *        recognized class codes.
2234  *      - If classification is not supported, leave classes[] alone.
2235  *      - If verbose is non-zero, print error message on failure;
2236  *        otherwise, shut up.
2237  *
2238  *      LOCKING:
2239  *      Kernel thread context (may sleep)
2240  *
2241  *      RETURNS:
2242  *      0 on success, -EINVAL if no reset method is avaliable, -ENODEV
2243  *      if classification fails, and any error code from reset
2244  *      methods.
2245  */
2246 int ata_drive_probe_reset(struct ata_port *ap, ata_probeinit_fn_t probeinit,
2247                           ata_reset_fn_t softreset, ata_reset_fn_t hardreset,
2248                           ata_postreset_fn_t postreset, unsigned int *classes)
2249 {
2250         int rc = -EINVAL;
2251
2252         if (probeinit)
2253                 probeinit(ap);
2254
2255         if (softreset) {
2256                 rc = do_probe_reset(ap, softreset, postreset, classes);
2257                 if (rc == 0)
2258                         return 0;
2259         }
2260
2261         if (!hardreset)
2262                 return rc;
2263
2264         rc = do_probe_reset(ap, hardreset, postreset, classes);
2265         if (rc == 0 || rc != -ENODEV)
2266                 return rc;
2267
2268         if (softreset)
2269                 rc = do_probe_reset(ap, softreset, postreset, classes);
2270
2271         return rc;
2272 }
2273
2274 static void ata_pr_blacklisted(const struct ata_port *ap,
2275                                const struct ata_device *dev)
2276 {
2277         printk(KERN_WARNING "ata%u: dev %u is on DMA blacklist, disabling DMA\n",
2278                 ap->id, dev->devno);
2279 }
2280
2281 static const char * const ata_dma_blacklist [] = {
2282         "WDC AC11000H",
2283         "WDC AC22100H",
2284         "WDC AC32500H",
2285         "WDC AC33100H",
2286         "WDC AC31600H",
2287         "WDC AC32100H",
2288         "WDC AC23200L",
2289         "Compaq CRD-8241B",
2290         "CRD-8400B",
2291         "CRD-8480B",
2292         "CRD-8482B",
2293         "CRD-84",
2294         "SanDisk SDP3B",
2295         "SanDisk SDP3B-64",
2296         "SANYO CD-ROM CRD",
2297         "HITACHI CDR-8",
2298         "HITACHI CDR-8335",
2299         "HITACHI CDR-8435",
2300         "Toshiba CD-ROM XM-6202B",
2301         "TOSHIBA CD-ROM XM-1702BC",
2302         "CD-532E-A",
2303         "E-IDE CD-ROM CR-840",
2304         "CD-ROM Drive/F5A",
2305         "WPI CDD-820",
2306         "SAMSUNG CD-ROM SC-148C",
2307         "SAMSUNG CD-ROM SC",
2308         "SanDisk SDP3B-64",
2309         "ATAPI CD-ROM DRIVE 40X MAXIMUM",
2310         "_NEC DV5800A",
2311 };
2312
2313 static int ata_dma_blacklisted(const struct ata_device *dev)
2314 {
2315         unsigned char model_num[41];
2316         int i;
2317
2318         ata_dev_id_c_string(dev->id, model_num, ATA_ID_PROD_OFS,
2319                             sizeof(model_num));
2320
2321         for (i = 0; i < ARRAY_SIZE(ata_dma_blacklist); i++)
2322                 if (!strcmp(ata_dma_blacklist[i], model_num))
2323                         return 1;
2324
2325         return 0;
2326 }
2327
2328 static unsigned int ata_get_mode_mask(const struct ata_port *ap, int shift)
2329 {
2330         const struct ata_device *master, *slave;
2331         unsigned int mask;
2332
2333         master = &ap->device[0];
2334         slave = &ap->device[1];
2335
2336         WARN_ON(!ata_dev_present(master) && !ata_dev_present(slave));
2337
2338         if (shift == ATA_SHIFT_UDMA) {
2339                 mask = ap->udma_mask;
2340                 if (ata_dev_present(master)) {
2341                         mask &= (master->id[ATA_ID_UDMA_MODES] & 0xff);
2342                         if (ata_dma_blacklisted(master)) {
2343                                 mask = 0;
2344                                 ata_pr_blacklisted(ap, master);
2345                         }
2346                 }
2347                 if (ata_dev_present(slave)) {
2348                         mask &= (slave->id[ATA_ID_UDMA_MODES] & 0xff);
2349                         if (ata_dma_blacklisted(slave)) {
2350                                 mask = 0;
2351                                 ata_pr_blacklisted(ap, slave);
2352                         }
2353                 }
2354         }
2355         else if (shift == ATA_SHIFT_MWDMA) {
2356                 mask = ap->mwdma_mask;
2357                 if (ata_dev_present(master)) {
2358                         mask &= (master->id[ATA_ID_MWDMA_MODES] & 0x07);
2359                         if (ata_dma_blacklisted(master)) {
2360                                 mask = 0;
2361                                 ata_pr_blacklisted(ap, master);
2362                         }
2363                 }
2364                 if (ata_dev_present(slave)) {
2365                         mask &= (slave->id[ATA_ID_MWDMA_MODES] & 0x07);
2366                         if (ata_dma_blacklisted(slave)) {
2367                                 mask = 0;
2368                                 ata_pr_blacklisted(ap, slave);
2369                         }
2370                 }
2371         }
2372         else if (shift == ATA_SHIFT_PIO) {
2373                 mask = ap->pio_mask;
2374                 if (ata_dev_present(master)) {
2375                         /* spec doesn't return explicit support for
2376                          * PIO0-2, so we fake it
2377                          */
2378                         u16 tmp_mode = master->id[ATA_ID_PIO_MODES] & 0x03;
2379                         tmp_mode <<= 3;
2380                         tmp_mode |= 0x7;
2381                         mask &= tmp_mode;
2382                 }
2383                 if (ata_dev_present(slave)) {
2384                         /* spec doesn't return explicit support for
2385                          * PIO0-2, so we fake it
2386                          */
2387                         u16 tmp_mode = slave->id[ATA_ID_PIO_MODES] & 0x03;
2388                         tmp_mode <<= 3;
2389                         tmp_mode |= 0x7;
2390                         mask &= tmp_mode;
2391                 }
2392         }
2393         else {
2394                 mask = 0xffffffff; /* shut up compiler warning */
2395                 BUG();
2396         }
2397
2398         return mask;
2399 }
2400
2401 /* find greatest bit */
2402 static int fgb(u32 bitmap)
2403 {
2404         unsigned int i;
2405         int x = -1;
2406
2407         for (i = 0; i < 32; i++)
2408                 if (bitmap & (1 << i))
2409                         x = i;
2410
2411         return x;
2412 }
2413
2414 /**
2415  *      ata_choose_xfer_mode - attempt to find best transfer mode
2416  *      @ap: Port for which an xfer mode will be selected
2417  *      @xfer_mode_out: (output) SET FEATURES - XFER MODE code
2418  *      @xfer_shift_out: (output) bit shift that selects this mode
2419  *
2420  *      Based on host and device capabilities, determine the
2421  *      maximum transfer mode that is amenable to all.
2422  *
2423  *      LOCKING:
2424  *      PCI/etc. bus probe sem.
2425  *
2426  *      RETURNS:
2427  *      Zero on success, negative on error.
2428  */
2429
2430 static int ata_choose_xfer_mode(const struct ata_port *ap,
2431                                 u8 *xfer_mode_out,
2432                                 unsigned int *xfer_shift_out)
2433 {
2434         unsigned int mask, shift;
2435         int x, i;
2436
2437         for (i = 0; i < ARRAY_SIZE(xfer_mode_classes); i++) {
2438                 shift = xfer_mode_classes[i].shift;
2439                 mask = ata_get_mode_mask(ap, shift);
2440
2441                 x = fgb(mask);
2442                 if (x >= 0) {
2443                         *xfer_mode_out = xfer_mode_classes[i].base + x;
2444                         *xfer_shift_out = shift;
2445                         return 0;
2446                 }
2447         }
2448
2449         return -1;
2450 }
2451
2452 /**
2453  *      ata_dev_set_xfermode - Issue SET FEATURES - XFER MODE command
2454  *      @ap: Port associated with device @dev
2455  *      @dev: Device to which command will be sent
2456  *
2457  *      Issue SET FEATURES - XFER MODE command to device @dev
2458  *      on port @ap.
2459  *
2460  *      LOCKING:
2461  *      PCI/etc. bus probe sem.
2462  */
2463
2464 static void ata_dev_set_xfermode(struct ata_port *ap, struct ata_device *dev)
2465 {
2466         struct ata_taskfile tf;
2467
2468         /* set up set-features taskfile */
2469         DPRINTK("set features - xfer mode\n");
2470
2471         ata_tf_init(ap, &tf, dev->devno);
2472         tf.command = ATA_CMD_SET_FEATURES;
2473         tf.feature = SETFEATURES_XFER;
2474         tf.flags |= ATA_TFLAG_ISADDR | ATA_TFLAG_DEVICE;
2475         tf.protocol = ATA_PROT_NODATA;
2476         tf.nsect = dev->xfer_mode;
2477
2478         if (ata_exec_internal(ap, dev, &tf, DMA_NONE, NULL, 0)) {
2479                 printk(KERN_ERR "ata%u: failed to set xfermode, disabled\n",
2480                        ap->id);
2481                 ata_port_disable(ap);
2482         }
2483
2484         DPRINTK("EXIT\n");
2485 }
2486
2487 /**
2488  *      ata_dev_reread_id - Reread the device identify device info
2489  *      @ap: port where the device is
2490  *      @dev: device to reread the identify device info
2491  *
2492  *      LOCKING:
2493  */
2494
2495 static void ata_dev_reread_id(struct ata_port *ap, struct ata_device *dev)
2496 {
2497         struct ata_taskfile tf;
2498
2499         ata_tf_init(ap, &tf, dev->devno);
2500
2501         if (dev->class == ATA_DEV_ATA) {
2502                 tf.command = ATA_CMD_ID_ATA;
2503                 DPRINTK("do ATA identify\n");
2504         } else {
2505                 tf.command = ATA_CMD_ID_ATAPI;
2506                 DPRINTK("do ATAPI identify\n");
2507         }
2508
2509         tf.flags |= ATA_TFLAG_DEVICE;
2510         tf.protocol = ATA_PROT_PIO;
2511
2512         if (ata_exec_internal(ap, dev, &tf, DMA_FROM_DEVICE,
2513                               dev->id, sizeof(dev->id)))
2514                 goto err_out;
2515
2516         swap_buf_le16(dev->id, ATA_ID_WORDS);
2517
2518         ata_dump_id(dev->id);
2519
2520         DPRINTK("EXIT\n");
2521
2522         return;
2523 err_out:
2524         printk(KERN_ERR "ata%u: failed to reread ID, disabled\n", ap->id);
2525         ata_port_disable(ap);
2526 }
2527
2528 /**
2529  *      ata_dev_init_params - Issue INIT DEV PARAMS command
2530  *      @ap: Port associated with device @dev
2531  *      @dev: Device to which command will be sent
2532  *
2533  *      LOCKING:
2534  */
2535
2536 static void ata_dev_init_params(struct ata_port *ap, struct ata_device *dev)
2537 {
2538         struct ata_taskfile tf;
2539         u16 sectors = dev->id[6];
2540         u16 heads   = dev->id[3];
2541
2542         /* Number of sectors per track 1-255. Number of heads 1-16 */
2543         if (sectors < 1 || sectors > 255 || heads < 1 || heads > 16)
2544                 return;
2545
2546         /* set up init dev params taskfile */
2547         DPRINTK("init dev params \n");
2548
2549         ata_tf_init(ap, &tf, dev->devno);
2550         tf.command = ATA_CMD_INIT_DEV_PARAMS;
2551         tf.flags |= ATA_TFLAG_ISADDR | ATA_TFLAG_DEVICE;
2552         tf.protocol = ATA_PROT_NODATA;
2553         tf.nsect = sectors;
2554         tf.device |= (heads - 1) & 0x0f; /* max head = num. of heads - 1 */
2555
2556         if (ata_exec_internal(ap, dev, &tf, DMA_NONE, NULL, 0)) {
2557                 printk(KERN_ERR "ata%u: failed to init parameters, disabled\n",
2558                        ap->id);
2559                 ata_port_disable(ap);
2560         }
2561
2562         DPRINTK("EXIT\n");
2563 }
2564
2565 /**
2566  *      ata_sg_clean - Unmap DMA memory associated with command
2567  *      @qc: Command containing DMA memory to be released
2568  *
2569  *      Unmap all mapped DMA memory associated with this command.
2570  *
2571  *      LOCKING:
2572  *      spin_lock_irqsave(host_set lock)
2573  */
2574
2575 static void ata_sg_clean(struct ata_queued_cmd *qc)
2576 {
2577         struct ata_port *ap = qc->ap;
2578         struct scatterlist *sg = qc->__sg;
2579         int dir = qc->dma_dir;
2580         void *pad_buf = NULL;
2581
2582         WARN_ON(!(qc->flags & ATA_QCFLAG_DMAMAP));
2583         WARN_ON(sg == NULL);
2584
2585         if (qc->flags & ATA_QCFLAG_SINGLE)
2586                 WARN_ON(qc->n_elem != 1);
2587
2588         VPRINTK("unmapping %u sg elements\n", qc->n_elem);
2589
2590         /* if we padded the buffer out to 32-bit bound, and data
2591          * xfer direction is from-device, we must copy from the
2592          * pad buffer back into the supplied buffer
2593          */
2594         if (qc->pad_len && !(qc->tf.flags & ATA_TFLAG_WRITE))
2595                 pad_buf = ap->pad + (qc->tag * ATA_DMA_PAD_SZ);
2596
2597         if (qc->flags & ATA_QCFLAG_SG) {
2598                 if (qc->n_elem)
2599                         dma_unmap_sg(ap->host_set->dev, sg, qc->n_elem, dir);
2600                 /* restore last sg */
2601                 sg[qc->orig_n_elem - 1].length += qc->pad_len;
2602                 if (pad_buf) {
2603                         struct scatterlist *psg = &qc->pad_sgent;
2604                         void *addr = kmap_atomic(psg->page, KM_IRQ0);
2605                         memcpy(addr + psg->offset, pad_buf, qc->pad_len);
2606                         kunmap_atomic(addr, KM_IRQ0);
2607                 }
2608         } else {
2609                 if (sg_dma_len(&sg[0]) > 0)
2610                         dma_unmap_single(ap->host_set->dev,
2611                                 sg_dma_address(&sg[0]), sg_dma_len(&sg[0]),
2612                                 dir);
2613                 /* restore sg */
2614                 sg->length += qc->pad_len;
2615                 if (pad_buf)
2616                         memcpy(qc->buf_virt + sg->length - qc->pad_len,
2617                                pad_buf, qc->pad_len);
2618         }
2619
2620         qc->flags &= ~ATA_QCFLAG_DMAMAP;
2621         qc->__sg = NULL;
2622 }
2623
2624 /**
2625  *      ata_fill_sg - Fill PCI IDE PRD table
2626  *      @qc: Metadata associated with taskfile to be transferred
2627  *
2628  *      Fill PCI IDE PRD (scatter-gather) table with segments
2629  *      associated with the current disk command.
2630  *
2631  *      LOCKING:
2632  *      spin_lock_irqsave(host_set lock)
2633  *
2634  */
2635 static void ata_fill_sg(struct ata_queued_cmd *qc)
2636 {
2637         struct ata_port *ap = qc->ap;
2638         struct scatterlist *sg;
2639         unsigned int idx;
2640
2641         WARN_ON(qc->__sg == NULL);
2642         WARN_ON(qc->n_elem == 0);
2643
2644         idx = 0;
2645         ata_for_each_sg(sg, qc) {
2646                 u32 addr, offset;
2647                 u32 sg_len, len;
2648
2649                 /* determine if physical DMA addr spans 64K boundary.
2650                  * Note h/w doesn't support 64-bit, so we unconditionally
2651                  * truncate dma_addr_t to u32.
2652                  */
2653                 addr = (u32) sg_dma_address(sg);
2654                 sg_len = sg_dma_len(sg);
2655
2656                 while (sg_len) {
2657                         offset = addr & 0xffff;
2658                         len = sg_len;
2659                         if ((offset + sg_len) > 0x10000)
2660                                 len = 0x10000 - offset;
2661
2662                         ap->prd[idx].addr = cpu_to_le32(addr);
2663                         ap->prd[idx].flags_len = cpu_to_le32(len & 0xffff);
2664                         VPRINTK("PRD[%u] = (0x%X, 0x%X)\n", idx, addr, len);
2665
2666                         idx++;
2667                         sg_len -= len;
2668                         addr += len;
2669                 }
2670         }
2671
2672         if (idx)
2673                 ap->prd[idx - 1].flags_len |= cpu_to_le32(ATA_PRD_EOT);
2674 }
2675 /**
2676  *      ata_check_atapi_dma - Check whether ATAPI DMA can be supported
2677  *      @qc: Metadata associated with taskfile to check
2678  *
2679  *      Allow low-level driver to filter ATA PACKET commands, returning
2680  *      a status indicating whether or not it is OK to use DMA for the
2681  *      supplied PACKET command.
2682  *
2683  *      LOCKING:
2684  *      spin_lock_irqsave(host_set lock)
2685  *
2686  *      RETURNS: 0 when ATAPI DMA can be used
2687  *               nonzero otherwise
2688  */
2689 int ata_check_atapi_dma(struct ata_queued_cmd *qc)
2690 {
2691         struct ata_port *ap = qc->ap;
2692         int rc = 0; /* Assume ATAPI DMA is OK by default */
2693
2694         if (ap->ops->check_atapi_dma)
2695                 rc = ap->ops->check_atapi_dma(qc);
2696
2697         return rc;
2698 }
2699 /**
2700  *      ata_qc_prep - Prepare taskfile for submission
2701  *      @qc: Metadata associated with taskfile to be prepared
2702  *
2703  *      Prepare ATA taskfile for submission.
2704  *
2705  *      LOCKING:
2706  *      spin_lock_irqsave(host_set lock)
2707  */
2708 void ata_qc_prep(struct ata_queued_cmd *qc)
2709 {
2710         if (!(qc->flags & ATA_QCFLAG_DMAMAP))
2711                 return;
2712
2713         ata_fill_sg(qc);
2714 }
2715
2716 /**
2717  *      ata_sg_init_one - Associate command with memory buffer
2718  *      @qc: Command to be associated
2719  *      @buf: Memory buffer
2720  *      @buflen: Length of memory buffer, in bytes.
2721  *
2722  *      Initialize the data-related elements of queued_cmd @qc
2723  *      to point to a single memory buffer, @buf of byte length @buflen.
2724  *
2725  *      LOCKING:
2726  *      spin_lock_irqsave(host_set lock)
2727  */
2728
2729 void ata_sg_init_one(struct ata_queued_cmd *qc, void *buf, unsigned int buflen)
2730 {
2731         struct scatterlist *sg;
2732
2733         qc->flags |= ATA_QCFLAG_SINGLE;
2734
2735         memset(&qc->sgent, 0, sizeof(qc->sgent));
2736         qc->__sg = &qc->sgent;
2737         qc->n_elem = 1;
2738         qc->orig_n_elem = 1;
2739         qc->buf_virt = buf;
2740
2741         sg = qc->__sg;
2742         sg_init_one(sg, buf, buflen);
2743 }
2744
2745 /**
2746  *      ata_sg_init - Associate command with scatter-gather table.
2747  *      @qc: Command to be associated
2748  *      @sg: Scatter-gather table.
2749  *      @n_elem: Number of elements in s/g table.
2750  *
2751  *      Initialize the data-related elements of queued_cmd @qc
2752  *      to point to a scatter-gather table @sg, containing @n_elem
2753  *      elements.
2754  *
2755  *      LOCKING:
2756  *      spin_lock_irqsave(host_set lock)
2757  */
2758
2759 void ata_sg_init(struct ata_queued_cmd *qc, struct scatterlist *sg,
2760                  unsigned int n_elem)
2761 {
2762         qc->flags |= ATA_QCFLAG_SG;
2763         qc->__sg = sg;
2764         qc->n_elem = n_elem;
2765         qc->orig_n_elem = n_elem;
2766 }
2767
2768 /**
2769  *      ata_sg_setup_one - DMA-map the memory buffer associated with a command.
2770  *      @qc: Command with memory buffer to be mapped.
2771  *
2772  *      DMA-map the memory buffer associated with queued_cmd @qc.
2773  *
2774  *      LOCKING:
2775  *      spin_lock_irqsave(host_set lock)
2776  *
2777  *      RETURNS:
2778  *      Zero on success, negative on error.
2779  */
2780
2781 static int ata_sg_setup_one(struct ata_queued_cmd *qc)
2782 {
2783         struct ata_port *ap = qc->ap;
2784         int dir = qc->dma_dir;
2785         struct scatterlist *sg = qc->__sg;
2786         dma_addr_t dma_address;
2787
2788         /* we must lengthen transfers to end on a 32-bit boundary */
2789         qc->pad_len = sg->length & 3;
2790         if (qc->pad_len) {
2791                 void *pad_buf = ap->pad + (qc->tag * ATA_DMA_PAD_SZ);
2792                 struct scatterlist *psg = &qc->pad_sgent;
2793
2794                 WARN_ON(qc->dev->class != ATA_DEV_ATAPI);
2795
2796                 memset(pad_buf, 0, ATA_DMA_PAD_SZ);
2797
2798                 if (qc->tf.flags & ATA_TFLAG_WRITE)
2799                         memcpy(pad_buf, qc->buf_virt + sg->length - qc->pad_len,
2800                                qc->pad_len);
2801
2802                 sg_dma_address(psg) = ap->pad_dma + (qc->tag * ATA_DMA_PAD_SZ);
2803                 sg_dma_len(psg) = ATA_DMA_PAD_SZ;
2804                 /* trim sg */
2805                 sg->length -= qc->pad_len;
2806
2807                 DPRINTK("padding done, sg->length=%u pad_len=%u\n",
2808                         sg->length, qc->pad_len);
2809         }
2810
2811         if (!sg->length) {
2812                 sg_dma_address(sg) = 0;
2813                 goto skip_map;
2814         }
2815
2816         dma_address = dma_map_single(ap->host_set->dev, qc->buf_virt,
2817                                      sg->length, dir);
2818         if (dma_mapping_error(dma_address)) {
2819                 /* restore sg */
2820                 sg->length += qc->pad_len;
2821                 return -1;
2822         }
2823
2824         sg_dma_address(sg) = dma_address;
2825 skip_map:
2826         sg_dma_len(sg) = sg->length;
2827
2828         DPRINTK("mapped buffer of %d bytes for %s\n", sg_dma_len(sg),
2829                 qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read");
2830
2831         return 0;
2832 }
2833
2834 /**
2835  *      ata_sg_setup - DMA-map the scatter-gather table associated with a command.
2836  *      @qc: Command with scatter-gather table to be mapped.
2837  *
2838  *      DMA-map the scatter-gather table associated with queued_cmd @qc.
2839  *
2840  *      LOCKING:
2841  *      spin_lock_irqsave(host_set lock)
2842  *
2843  *      RETURNS:
2844  *      Zero on success, negative on error.
2845  *
2846  */
2847
2848 static int ata_sg_setup(struct ata_queued_cmd *qc)
2849 {
2850         struct ata_port *ap = qc->ap;
2851         struct scatterlist *sg = qc->__sg;
2852         struct scatterlist *lsg = &sg[qc->n_elem - 1];
2853         int n_elem, pre_n_elem, dir, trim_sg = 0;
2854
2855         VPRINTK("ENTER, ata%u\n", ap->id);
2856         WARN_ON(!(qc->flags & ATA_QCFLAG_SG));
2857
2858         /* we must lengthen transfers to end on a 32-bit boundary */
2859         qc->pad_len = lsg->length & 3;
2860         if (qc->pad_len) {
2861                 void *pad_buf = ap->pad + (qc->tag * ATA_DMA_PAD_SZ);
2862                 struct scatterlist *psg = &qc->pad_sgent;
2863                 unsigned int offset;
2864
2865                 WARN_ON(qc->dev->class != ATA_DEV_ATAPI);
2866
2867                 memset(pad_buf, 0, ATA_DMA_PAD_SZ);
2868
2869                 /*
2870                  * psg->page/offset are used to copy to-be-written
2871                  * data in this function or read data in ata_sg_clean.
2872                  */
2873                 offset = lsg->offset + lsg->length - qc->pad_len;
2874                 psg->page = nth_page(lsg->page, offset >> PAGE_SHIFT);
2875                 psg->offset = offset_in_page(offset);
2876
2877                 if (qc->tf.flags & ATA_TFLAG_WRITE) {
2878                         void *addr = kmap_atomic(psg->page, KM_IRQ0);
2879                         memcpy(pad_buf, addr + psg->offset, qc->pad_len);
2880                         kunmap_atomic(addr, KM_IRQ0);
2881                 }
2882
2883                 sg_dma_address(psg) = ap->pad_dma + (qc->tag * ATA_DMA_PAD_SZ);
2884                 sg_dma_len(psg) = ATA_DMA_PAD_SZ;
2885                 /* trim last sg */
2886                 lsg->length -= qc->pad_len;
2887                 if (lsg->length == 0)
2888                         trim_sg = 1;
2889
2890                 DPRINTK("padding done, sg[%d].length=%u pad_len=%u\n",
2891                         qc->n_elem - 1, lsg->length, qc->pad_len);
2892         }
2893
2894         pre_n_elem = qc->n_elem;
2895         if (trim_sg && pre_n_elem)
2896                 pre_n_elem--;
2897
2898         if (!pre_n_elem) {
2899                 n_elem = 0;
2900                 goto skip_map;
2901         }
2902
2903         dir = qc->dma_dir;
2904         n_elem = dma_map_sg(ap->host_set->dev, sg, pre_n_elem, dir);
2905         if (n_elem < 1) {
2906                 /* restore last sg */
2907                 lsg->length += qc->pad_len;
2908                 return -1;
2909         }
2910
2911         DPRINTK("%d sg elements mapped\n", n_elem);
2912
2913 skip_map:
2914         qc->n_elem = n_elem;
2915
2916         return 0;
2917 }
2918
2919 /**
2920  *      ata_poll_qc_complete - turn irq back on and finish qc
2921  *      @qc: Command to complete
2922  *      @err_mask: ATA status register content
2923  *
2924  *      LOCKING:
2925  *      None.  (grabs host lock)
2926  */
2927
2928 void ata_poll_qc_complete(struct ata_queued_cmd *qc)
2929 {
2930         struct ata_port *ap = qc->ap;
2931         unsigned long flags;
2932
2933         spin_lock_irqsave(&ap->host_set->lock, flags);
2934         ap->flags &= ~ATA_FLAG_NOINTR;
2935         ata_irq_on(ap);
2936         ata_qc_complete(qc);
2937         spin_unlock_irqrestore(&ap->host_set->lock, flags);
2938 }
2939
2940 /**
2941  *      ata_pio_poll - poll using PIO, depending on current state
2942  *      @ap: the target ata_port
2943  *
2944  *      LOCKING:
2945  *      None.  (executing in kernel thread context)
2946  *
2947  *      RETURNS:
2948  *      timeout value to use
2949  */
2950
2951 static unsigned long ata_pio_poll(struct ata_port *ap)
2952 {
2953         struct ata_queued_cmd *qc;
2954         u8 status;
2955         unsigned int poll_state = HSM_ST_UNKNOWN;
2956         unsigned int reg_state = HSM_ST_UNKNOWN;
2957
2958         qc = ata_qc_from_tag(ap, ap->active_tag);
2959         WARN_ON(qc == NULL);
2960
2961         switch (ap->hsm_task_state) {
2962         case HSM_ST:
2963         case HSM_ST_POLL:
2964                 poll_state = HSM_ST_POLL;
2965                 reg_state = HSM_ST;
2966                 break;
2967         case HSM_ST_LAST:
2968         case HSM_ST_LAST_POLL:
2969                 poll_state = HSM_ST_LAST_POLL;
2970                 reg_state = HSM_ST_LAST;
2971                 break;
2972         default:
2973                 BUG();
2974                 break;
2975         }
2976
2977         status = ata_chk_status(ap);
2978         if (status & ATA_BUSY) {
2979                 if (time_after(jiffies, ap->pio_task_timeout)) {
2980                         qc->err_mask |= AC_ERR_TIMEOUT;
2981                         ap->hsm_task_state = HSM_ST_TMOUT;
2982                         return 0;
2983                 }
2984                 ap->hsm_task_state = poll_state;
2985                 return ATA_SHORT_PAUSE;
2986         }
2987
2988         ap->hsm_task_state = reg_state;
2989         return 0;
2990 }
2991
2992 /**
2993  *      ata_pio_complete - check if drive is busy or idle
2994  *      @ap: the target ata_port
2995  *
2996  *      LOCKING:
2997  *      None.  (executing in kernel thread context)
2998  *
2999  *      RETURNS:
3000  *      Non-zero if qc completed, zero otherwise.
3001  */
3002
3003 static int ata_pio_complete (struct ata_port *ap)
3004 {
3005         struct ata_queued_cmd *qc;
3006         u8 drv_stat;
3007
3008         /*
3009          * This is purely heuristic.  This is a fast path.  Sometimes when
3010          * we enter, BSY will be cleared in a chk-status or two.  If not,
3011          * the drive is probably seeking or something.  Snooze for a couple
3012          * msecs, then chk-status again.  If still busy, fall back to
3013          * HSM_ST_POLL state.
3014          */
3015         drv_stat = ata_busy_wait(ap, ATA_BUSY, 10);
3016         if (drv_stat & ATA_BUSY) {
3017                 msleep(2);
3018                 drv_stat = ata_busy_wait(ap, ATA_BUSY, 10);
3019                 if (drv_stat & ATA_BUSY) {
3020                         ap->hsm_task_state = HSM_ST_LAST_POLL;
3021                         ap->pio_task_timeout = jiffies + ATA_TMOUT_PIO;
3022                         return 0;
3023                 }
3024         }
3025
3026         qc = ata_qc_from_tag(ap, ap->active_tag);
3027         WARN_ON(qc == NULL);
3028
3029         drv_stat = ata_wait_idle(ap);
3030         if (!ata_ok(drv_stat)) {
3031                 qc->err_mask |= __ac_err_mask(drv_stat);
3032                 ap->hsm_task_state = HSM_ST_ERR;
3033                 return 0;
3034         }
3035
3036         ap->hsm_task_state = HSM_ST_IDLE;
3037
3038         WARN_ON(qc->err_mask);
3039         ata_poll_qc_complete(qc);
3040
3041         /* another command may start at this point */
3042
3043         return 1;
3044 }
3045
3046
3047 /**
3048  *      swap_buf_le16 - swap halves of 16-bit words in place
3049  *      @buf:  Buffer to swap
3050  *      @buf_words:  Number of 16-bit words in buffer.
3051  *
3052  *      Swap halves of 16-bit words if needed to convert from
3053  *      little-endian byte order to native cpu byte order, or
3054  *      vice-versa.
3055  *
3056  *      LOCKING:
3057  *      Inherited from caller.
3058  */
3059 void swap_buf_le16(u16 *buf, unsigned int buf_words)
3060 {
3061 #ifdef __BIG_ENDIAN
3062         unsigned int i;
3063
3064         for (i = 0; i < buf_words; i++)
3065                 buf[i] = le16_to_cpu(buf[i]);
3066 #endif /* __BIG_ENDIAN */
3067 }
3068
3069 /**
3070  *      ata_mmio_data_xfer - Transfer data by MMIO
3071  *      @ap: port to read/write
3072  *      @buf: data buffer
3073  *      @buflen: buffer length
3074  *      @write_data: read/write
3075  *
3076  *      Transfer data from/to the device data register by MMIO.
3077  *
3078  *      LOCKING:
3079  *      Inherited from caller.
3080  */
3081
3082 static void ata_mmio_data_xfer(struct ata_port *ap, unsigned char *buf,
3083                                unsigned int buflen, int write_data)
3084 {
3085         unsigned int i;
3086         unsigned int words = buflen >> 1;
3087         u16 *buf16 = (u16 *) buf;
3088         void __iomem *mmio = (void __iomem *)ap->ioaddr.data_addr;
3089
3090         /* Transfer multiple of 2 bytes */
3091         if (write_data) {
3092                 for (i = 0; i < words; i++)
3093                         writew(le16_to_cpu(buf16[i]), mmio);
3094         } else {
3095                 for (i = 0; i < words; i++)
3096                         buf16[i] = cpu_to_le16(readw(mmio));
3097         }
3098
3099         /* Transfer trailing 1 byte, if any. */
3100         if (unlikely(buflen & 0x01)) {
3101                 u16 align_buf[1] = { 0 };
3102                 unsigned char *trailing_buf = buf + buflen - 1;
3103
3104                 if (write_data) {
3105                         memcpy(align_buf, trailing_buf, 1);
3106                         writew(le16_to_cpu(align_buf[0]), mmio);
3107                 } else {
3108                         align_buf[0] = cpu_to_le16(readw(mmio));
3109                         memcpy(trailing_buf, align_buf, 1);
3110                 }
3111         }
3112 }
3113
3114 /**
3115  *      ata_pio_data_xfer - Transfer data by PIO
3116  *      @ap: port to read/write
3117  *      @buf: data buffer
3118  *      @buflen: buffer length
3119  *      @write_data: read/write
3120  *
3121  *      Transfer data from/to the device data register by PIO.
3122  *
3123  *      LOCKING:
3124  *      Inherited from caller.
3125  */
3126
3127 static void ata_pio_data_xfer(struct ata_port *ap, unsigned char *buf,
3128                               unsigned int buflen, int write_data)
3129 {
3130         unsigned int words = buflen >> 1;
3131
3132         /* Transfer multiple of 2 bytes */
3133         if (write_data)
3134                 outsw(ap->ioaddr.data_addr, buf, words);
3135         else
3136                 insw(ap->ioaddr.data_addr, buf, words);
3137
3138         /* Transfer trailing 1 byte, if any. */
3139         if (unlikely(buflen & 0x01)) {
3140                 u16 align_buf[1] = { 0 };
3141                 unsigned char *trailing_buf = buf + buflen - 1;
3142
3143                 if (write_data) {
3144                         memcpy(align_buf, trailing_buf, 1);
3145                         outw(le16_to_cpu(align_buf[0]), ap->ioaddr.data_addr);
3146                 } else {
3147                         align_buf[0] = cpu_to_le16(inw(ap->ioaddr.data_addr));
3148                         memcpy(trailing_buf, align_buf, 1);
3149                 }
3150         }
3151 }
3152
3153 /**
3154  *      ata_data_xfer - Transfer data from/to the data register.
3155  *      @ap: port to read/write
3156  *      @buf: data buffer
3157  *      @buflen: buffer length
3158  *      @do_write: read/write
3159  *
3160  *      Transfer data from/to the device data register.
3161  *
3162  *      LOCKING:
3163  *      Inherited from caller.
3164  */
3165
3166 static void ata_data_xfer(struct ata_port *ap, unsigned char *buf,
3167                           unsigned int buflen, int do_write)
3168 {
3169         /* Make the crap hardware pay the costs not the good stuff */
3170         if (unlikely(ap->flags & ATA_FLAG_IRQ_MASK)) {
3171                 unsigned long flags;
3172                 local_irq_save(flags);
3173                 if (ap->flags & ATA_FLAG_MMIO)
3174                         ata_mmio_data_xfer(ap, buf, buflen, do_write);
3175                 else
3176                         ata_pio_data_xfer(ap, buf, buflen, do_write);
3177                 local_irq_restore(flags);
3178         } else {
3179                 if (ap->flags & ATA_FLAG_MMIO)
3180                         ata_mmio_data_xfer(ap, buf, buflen, do_write);
3181                 else
3182                         ata_pio_data_xfer(ap, buf, buflen, do_write);
3183         }
3184 }
3185
3186 /**
3187  *      ata_pio_sector - Transfer ATA_SECT_SIZE (512 bytes) of data.
3188  *      @qc: Command on going
3189  *
3190  *      Transfer ATA_SECT_SIZE of data from/to the ATA device.
3191  *
3192  *      LOCKING:
3193  *      Inherited from caller.
3194  */
3195
3196 static void ata_pio_sector(struct ata_queued_cmd *qc)
3197 {
3198         int do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
3199         struct scatterlist *sg = qc->__sg;
3200         struct ata_port *ap = qc->ap;
3201         struct page *page;
3202         unsigned int offset;
3203         unsigned char *buf;
3204
3205         if (qc->cursect == (qc->nsect - 1))
3206                 ap->hsm_task_state = HSM_ST_LAST;
3207
3208         page = sg[qc->cursg].page;
3209         offset = sg[qc->cursg].offset + qc->cursg_ofs * ATA_SECT_SIZE;
3210
3211         /* get the current page and offset */
3212         page = nth_page(page, (offset >> PAGE_SHIFT));
3213         offset %= PAGE_SIZE;
3214
3215         buf = kmap(page) + offset;
3216
3217         qc->cursect++;
3218         qc->cursg_ofs++;
3219
3220         if ((qc->cursg_ofs * ATA_SECT_SIZE) == (&sg[qc->cursg])->length) {
3221                 qc->cursg++;
3222                 qc->cursg_ofs = 0;
3223         }
3224
3225         DPRINTK("data %s\n", qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read");
3226
3227         /* do the actual data transfer */
3228         do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
3229         ata_data_xfer(ap, buf, ATA_SECT_SIZE, do_write);
3230
3231         kunmap(page);
3232 }
3233
3234 /**
3235  *      __atapi_pio_bytes - Transfer data from/to the ATAPI device.
3236  *      @qc: Command on going
3237  *      @bytes: number of bytes
3238  *
3239  *      Transfer Transfer data from/to the ATAPI device.
3240  *
3241  *      LOCKING:
3242  *      Inherited from caller.
3243  *
3244  */
3245
3246 static void __atapi_pio_bytes(struct ata_queued_cmd *qc, unsigned int bytes)
3247 {
3248         int do_write = (qc->tf.flags & ATA_TFLAG_WRITE);
3249         struct scatterlist *sg = qc->__sg;
3250         struct ata_port *ap = qc->ap;
3251         struct page *page;
3252         unsigned char *buf;
3253         unsigned int offset, count;
3254
3255         if (qc->curbytes + bytes >= qc->nbytes)
3256                 ap->hsm_task_state = HSM_ST_LAST;
3257
3258 next_sg:
3259         if (unlikely(qc->cursg >= qc->n_elem)) {
3260                 /*
3261                  * The end of qc->sg is reached and the device expects
3262                  * more data to transfer. In order not to overrun qc->sg
3263                  * and fulfill length specified in the byte count register,
3264                  *    - for read case, discard trailing data from the device
3265                  *    - for write case, padding zero data to the device
3266                  */
3267                 u16 pad_buf[1] = { 0 };
3268                 unsigned int words = bytes >> 1;
3269                 unsigned int i;
3270
3271                 if (words) /* warning if bytes > 1 */
3272                         printk(KERN_WARNING "ata%u: %u bytes trailing data\n",
3273                                ap->id, bytes);
3274
3275                 for (i = 0; i < words; i++)
3276                         ata_data_xfer(ap, (unsigned char*)pad_buf, 2, do_write);
3277
3278                 ap->hsm_task_state = HSM_ST_LAST;
3279                 return;
3280         }
3281
3282         sg = &qc->__sg[qc->cursg];
3283
3284         page = sg->page;
3285         offset = sg->offset + qc->cursg_ofs;
3286
3287         /* get the current page and offset */
3288         page = nth_page(page, (offset >> PAGE_SHIFT));
3289         offset %= PAGE_SIZE;
3290
3291         /* don't overrun current sg */
3292         count = min(sg->length - qc->cursg_ofs, bytes);
3293
3294         /* don't cross page boundaries */
3295         count = min(count, (unsigned int)PAGE_SIZE - offset);
3296
3297         buf = kmap(page) + offset;
3298
3299         bytes -= count;
3300         qc->curbytes += count;
3301         qc->cursg_ofs += count;
3302
3303         if (qc->cursg_ofs == sg->length) {
3304                 qc->cursg++;
3305                 qc->cursg_ofs = 0;
3306         }
3307
3308         DPRINTK("data %s\n", qc->tf.flags & ATA_TFLAG_WRITE ? "write" : "read");
3309
3310         /* do the actual data transfer */
3311         ata_data_xfer(ap, buf, count, do_write);
3312
3313         kunmap(page);
3314
3315         if (bytes)
3316                 goto next_sg;
3317 }
3318
3319 /**
3320  *      atapi_pio_bytes - Transfer data from/to the ATAPI device.
3321  *      @qc: Command on going
3322  *
3323  *      Transfer Transfer data from/to the ATAPI device.
3324  *
3325  *      LOCKING:
3326  *      Inherited from caller.
3327  */
3328
3329 static void atapi_pio_bytes(struct ata_queued_cmd *qc)
3330 {
3331         struct ata_port *ap = qc->ap;
3332         struct ata_device *dev = qc->dev;
3333         unsigned int ireason, bc_lo, bc_hi, bytes;
3334         int i_write, do_write = (qc->tf.flags & ATA_TFLAG_WRITE) ? 1 : 0;
3335
3336         ap->ops->tf_read(ap, &qc->tf);
3337         ireason = qc->tf.nsect;
3338         bc_lo = qc->tf.lbam;
3339         bc_hi = qc->tf.lbah;
3340         bytes = (bc_hi << 8) | bc_lo;
3341
3342         /* shall be cleared to zero, indicating xfer of data */
3343         if (ireason & (1 << 0))
3344                 goto err_out;
3345
3346         /* make sure transfer direction matches expected */
3347         i_write = ((ireason & (1 << 1)) == 0) ? 1 : 0;
3348         if (do_write != i_write)
3349                 goto err_out;
3350
3351         __atapi_pio_bytes(qc, bytes);
3352
3353         return;
3354
3355 err_out:
3356         printk(KERN_INFO "ata%u: dev %u: ATAPI check failed\n",
3357               ap->id, dev->devno);
3358         qc->err_mask |= AC_ERR_HSM;
3359         ap->hsm_task_state = HSM_ST_ERR;
3360 }
3361
3362 /**
3363  *      ata_pio_block - start PIO on a block
3364  *      @ap: the target ata_port
3365  *
3366  *      LOCKING:
3367  *      None.  (executing in kernel thread context)
3368  */
3369
3370 static void ata_pio_block(struct ata_port *ap)
3371 {
3372         struct ata_queued_cmd *qc;
3373         u8 status;
3374
3375         /*
3376          * This is purely heuristic.  This is a fast path.
3377          * Sometimes when we enter, BSY will be cleared in
3378          * a chk-status or two.  If not, the drive is probably seeking
3379          * or something.  Snooze for a couple msecs, then
3380          * chk-status again.  If still busy, fall back to
3381          * HSM_ST_POLL state.
3382          */
3383         status = ata_busy_wait(ap, ATA_BUSY, 5);
3384         if (status & ATA_BUSY) {
3385                 msleep(2);
3386                 status = ata_busy_wait(ap, ATA_BUSY, 10);
3387                 if (status & ATA_BUSY) {
3388                         ap->hsm_task_state = HSM_ST_POLL;
3389                         ap->pio_task_timeout = jiffies + ATA_TMOUT_PIO;
3390                         return;
3391                 }
3392         }
3393
3394         qc = ata_qc_from_tag(ap, ap->active_tag);
3395         WARN_ON(qc == NULL);
3396
3397         /* check error */
3398         if (status & (ATA_ERR | ATA_DF)) {
3399                 qc->err_mask |= AC_ERR_DEV;
3400                 ap->hsm_task_state = HSM_ST_ERR;
3401                 return;
3402         }
3403
3404         /* transfer data if any */
3405         if (is_atapi_taskfile(&qc->tf)) {
3406                 /* DRQ=0 means no more data to transfer */
3407                 if ((status & ATA_DRQ) == 0) {
3408                         ap->hsm_task_state = HSM_ST_LAST;
3409                         return;
3410                 }
3411
3412                 atapi_pio_bytes(qc);
3413         } else {
3414                 /* handle BSY=0, DRQ=0 as error */
3415                 if ((status & ATA_DRQ) == 0) {
3416                         qc->err_mask |= AC_ERR_HSM;
3417                         ap->hsm_task_state = HSM_ST_ERR;
3418                         return;
3419                 }
3420
3421                 ata_pio_sector(qc);
3422         }
3423 }
3424
3425 static void ata_pio_error(struct ata_port *ap)
3426 {
3427         struct ata_queued_cmd *qc;
3428
3429         printk(KERN_WARNING "ata%u: PIO error\n", ap->id);
3430
3431         qc = ata_qc_from_tag(ap, ap->active_tag);
3432         WARN_ON(qc == NULL);
3433
3434         /* make sure qc->err_mask is available to
3435          * know what's wrong and recover
3436          */
3437         WARN_ON(qc->err_mask == 0);
3438
3439         ap->hsm_task_state = HSM_ST_IDLE;
3440
3441         ata_poll_qc_complete(qc);
3442 }
3443
3444 static void ata_pio_task(void *_data)
3445 {
3446         struct ata_port *ap = _data;
3447         unsigned long timeout;
3448         int qc_completed;
3449
3450 fsm_start:
3451         timeout = 0;
3452         qc_completed = 0;
3453
3454         switch (ap->hsm_task_state) {
3455         case HSM_ST_IDLE:
3456                 return;
3457
3458         case HSM_ST:
3459                 ata_pio_block(ap);
3460                 break;
3461
3462         case HSM_ST_LAST:
3463                 qc_completed = ata_pio_complete(ap);
3464                 break;
3465
3466         case HSM_ST_POLL:
3467         case HSM_ST_LAST_POLL:
3468                 timeout = ata_pio_poll(ap);
3469                 break;
3470
3471         case HSM_ST_TMOUT:
3472         case HSM_ST_ERR:
3473                 ata_pio_error(ap);
3474                 return;
3475         }
3476
3477         if (timeout)
3478                 ata_queue_delayed_pio_task(ap, timeout);
3479         else if (!qc_completed)
3480                 goto fsm_start;
3481 }
3482
3483 /**
3484  *      ata_qc_timeout - Handle timeout of queued command
3485  *      @qc: Command that timed out
3486  *
3487  *      Some part of the kernel (currently, only the SCSI layer)
3488  *      has noticed that the active command on port @ap has not
3489  *      completed after a specified length of time.  Handle this
3490  *      condition by disabling DMA (if necessary) and completing
3491  *      transactions, with error if necessary.
3492  *
3493  *      This also handles the case of the "lost interrupt", where
3494  *      for some reason (possibly hardware bug, possibly driver bug)
3495  *      an interrupt was not delivered to the driver, even though the
3496  *      transaction completed successfully.
3497  *
3498  *      LOCKING:
3499  *      Inherited from SCSI layer (none, can sleep)
3500  */
3501
3502 static void ata_qc_timeout(struct ata_queued_cmd *qc)
3503 {
3504         struct ata_port *ap = qc->ap;
3505         struct ata_host_set *host_set = ap->host_set;
3506         u8 host_stat = 0, drv_stat;
3507         unsigned long flags;
3508
3509         DPRINTK("ENTER\n");
3510
3511         ata_flush_pio_tasks(ap);
3512         ap->hsm_task_state = HSM_ST_IDLE;
3513
3514         spin_lock_irqsave(&host_set->lock, flags);
3515
3516         switch (qc->tf.protocol) {
3517
3518         case ATA_PROT_DMA:
3519         case ATA_PROT_ATAPI_DMA:
3520                 host_stat = ap->ops->bmdma_status(ap);
3521
3522                 /* before we do anything else, clear DMA-Start bit */
3523                 ap->ops->bmdma_stop(qc);
3524
3525                 /* fall through */
3526
3527         default:
3528                 ata_altstatus(ap);
3529                 drv_stat = ata_chk_status(ap);
3530
3531                 /* ack bmdma irq events */
3532                 ap->ops->irq_clear(ap);
3533
3534                 printk(KERN_ERR "ata%u: command 0x%x timeout, stat 0x%x host_stat 0x%x\n",
3535                        ap->id, qc->tf.command, drv_stat, host_stat);
3536
3537                 /* complete taskfile transaction */
3538                 qc->err_mask |= ac_err_mask(drv_stat);
3539                 break;
3540         }
3541
3542         spin_unlock_irqrestore(&host_set->lock, flags);
3543
3544         ata_eh_qc_complete(qc);
3545
3546         DPRINTK("EXIT\n");
3547 }
3548
3549 /**
3550  *      ata_eng_timeout - Handle timeout of queued command
3551  *      @ap: Port on which timed-out command is active
3552  *
3553  *      Some part of the kernel (currently, only the SCSI layer)
3554  *      has noticed that the active command on port @ap has not
3555  *      completed after a specified length of time.  Handle this
3556  *      condition by disabling DMA (if necessary) and completing
3557  *      transactions, with error if necessary.
3558  *
3559  *      This also handles the case of the "lost interrupt", where
3560  *      for some reason (possibly hardware bug, possibly driver bug)
3561  *      an interrupt was not delivered to the driver, even though the
3562  *      transaction completed successfully.
3563  *
3564  *      LOCKING:
3565  *      Inherited from SCSI layer (none, can sleep)
3566  */
3567
3568 void ata_eng_timeout(struct ata_port *ap)
3569 {
3570         DPRINTK("ENTER\n");
3571
3572         ata_qc_timeout(ata_qc_from_tag(ap, ap->active_tag));
3573
3574         DPRINTK("EXIT\n");
3575 }
3576
3577 /**
3578  *      ata_qc_new - Request an available ATA command, for queueing
3579  *      @ap: Port associated with device @dev
3580  *      @dev: Device from whom we request an available command structure
3581  *
3582  *      LOCKING:
3583  *      None.
3584  */
3585
3586 static struct ata_queued_cmd *ata_qc_new(struct ata_port *ap)
3587 {
3588         struct ata_queued_cmd *qc = NULL;
3589         unsigned int i;
3590
3591         for (i = 0; i < ATA_MAX_QUEUE; i++)
3592                 if (!test_and_set_bit(i, &ap->qactive)) {
3593                         qc = ata_qc_from_tag(ap, i);
3594                         break;
3595                 }
3596
3597         if (qc)
3598                 qc->tag = i;
3599
3600         return qc;
3601 }
3602
3603 /**
3604  *      ata_qc_new_init - Request an available ATA command, and initialize it
3605  *      @ap: Port associated with device @dev
3606  *      @dev: Device from whom we request an available command structure
3607  *
3608  *      LOCKING:
3609  *      None.
3610  */
3611
3612 struct ata_queued_cmd *ata_qc_new_init(struct ata_port *ap,
3613                                       struct ata_device *dev)
3614 {
3615         struct ata_queued_cmd *qc;
3616
3617         qc = ata_qc_new(ap);
3618         if (qc) {
3619                 qc->scsicmd = NULL;
3620                 qc->ap = ap;
3621                 qc->dev = dev;
3622
3623                 ata_qc_reinit(qc);
3624         }
3625
3626         return qc;
3627 }
3628
3629 /**
3630  *      ata_qc_free - free unused ata_queued_cmd
3631  *      @qc: Command to complete
3632  *
3633  *      Designed to free unused ata_queued_cmd object
3634  *      in case something prevents using it.
3635  *
3636  *      LOCKING:
3637  *      spin_lock_irqsave(host_set lock)
3638  */
3639 void ata_qc_free(struct ata_queued_cmd *qc)
3640 {
3641         struct ata_port *ap = qc->ap;
3642         unsigned int tag;
3643
3644         WARN_ON(qc == NULL);    /* ata_qc_from_tag _might_ return NULL */
3645
3646         qc->flags = 0;
3647         tag = qc->tag;
3648         if (likely(ata_tag_valid(tag))) {
3649                 if (tag == ap->active_tag)
3650                         ap->active_tag = ATA_TAG_POISON;
3651                 qc->tag = ATA_TAG_POISON;
3652                 clear_bit(tag, &ap->qactive);
3653         }
3654 }
3655
3656 void __ata_qc_complete(struct ata_queued_cmd *qc)
3657 {
3658         WARN_ON(qc == NULL);    /* ata_qc_from_tag _might_ return NULL */
3659         WARN_ON(!(qc->flags & ATA_QCFLAG_ACTIVE));
3660
3661         if (likely(qc->flags & ATA_QCFLAG_DMAMAP))
3662                 ata_sg_clean(qc);
3663
3664         /* atapi: mark qc as inactive to prevent the interrupt handler
3665          * from completing the command twice later, before the error handler
3666          * is called. (when rc != 0 and atapi request sense is needed)
3667          */
3668         qc->flags &= ~ATA_QCFLAG_ACTIVE;
3669
3670         /* call completion callback */
3671         qc->complete_fn(qc);
3672 }
3673
3674 static inline int ata_should_dma_map(struct ata_queued_cmd *qc)
3675 {
3676         struct ata_port *ap = qc->ap;
3677
3678         switch (qc->tf.protocol) {
3679         case ATA_PROT_DMA:
3680         case ATA_PROT_ATAPI_DMA:
3681                 return 1;
3682
3683         case ATA_PROT_ATAPI:
3684         case ATA_PROT_PIO:
3685         case ATA_PROT_PIO_MULT:
3686                 if (ap->flags & ATA_FLAG_PIO_DMA)
3687                         return 1;
3688
3689                 /* fall through */
3690
3691         default:
3692                 return 0;
3693         }
3694
3695         /* never reached */
3696 }
3697
3698 /**
3699  *      ata_qc_issue - issue taskfile to device
3700  *      @qc: command to issue to device
3701  *
3702  *      Prepare an ATA command to submission to device.
3703  *      This includes mapping the data into a DMA-able
3704  *      area, filling in the S/G table, and finally
3705  *      writing the taskfile to hardware, starting the command.
3706  *
3707  *      LOCKING:
3708  *      spin_lock_irqsave(host_set lock)
3709  *
3710  *      RETURNS:
3711  *      Zero on success, AC_ERR_* mask on failure
3712  */
3713
3714 unsigned int ata_qc_issue(struct ata_queued_cmd *qc)
3715 {
3716         struct ata_port *ap = qc->ap;
3717
3718         if (ata_should_dma_map(qc)) {
3719                 if (qc->flags & ATA_QCFLAG_SG) {
3720                         if (ata_sg_setup(qc))
3721                                 goto sg_err;
3722                 } else if (qc->flags & ATA_QCFLAG_SINGLE) {
3723                         if (ata_sg_setup_one(qc))
3724                                 goto sg_err;
3725                 }
3726         } else {
3727                 qc->flags &= ~ATA_QCFLAG_DMAMAP;
3728         }
3729
3730         ap->ops->qc_prep(qc);
3731
3732         qc->ap->active_tag = qc->tag;
3733         qc->flags |= ATA_QCFLAG_ACTIVE;
3734
3735         return ap->ops->qc_issue(qc);
3736
3737 sg_err:
3738         qc->flags &= ~ATA_QCFLAG_DMAMAP;
3739         return AC_ERR_SYSTEM;
3740 }
3741
3742
3743 /**
3744  *      ata_qc_issue_prot - issue taskfile to device in proto-dependent manner
3745  *      @qc: command to issue to device
3746  *
3747  *      Using various libata functions and hooks, this function
3748  *      starts an ATA command.  ATA commands are grouped into
3749  *      classes called "protocols", and issuing each type of protocol
3750  *      is slightly different.
3751  *
3752  *      May be used as the qc_issue() entry in ata_port_operations.
3753  *
3754  *      LOCKING:
3755  *      spin_lock_irqsave(host_set lock)
3756  *
3757  *      RETURNS:
3758  *      Zero on success, AC_ERR_* mask on failure
3759  */
3760
3761 unsigned int ata_qc_issue_prot(struct ata_queued_cmd *qc)
3762 {
3763         struct ata_port *ap = qc->ap;
3764
3765         ata_dev_select(ap, qc->dev->devno, 1, 0);
3766
3767         switch (qc->tf.protocol) {
3768         case ATA_PROT_NODATA:
3769                 ata_tf_to_host(ap, &qc->tf);
3770                 break;
3771
3772         case ATA_PROT_DMA:
3773                 ap->ops->tf_load(ap, &qc->tf);   /* load tf registers */
3774                 ap->ops->bmdma_setup(qc);           /* set up bmdma */
3775                 ap->ops->bmdma_start(qc);           /* initiate bmdma */
3776                 break;
3777
3778         case ATA_PROT_PIO: /* load tf registers, initiate polling pio */
3779                 ata_qc_set_polling(qc);
3780                 ata_tf_to_host(ap, &qc->tf);
3781                 ap->hsm_task_state = HSM_ST;
3782                 ata_queue_pio_task(ap);
3783                 break;
3784
3785         case ATA_PROT_ATAPI:
3786                 ata_qc_set_polling(qc);
3787                 ata_tf_to_host(ap, &qc->tf);
3788                 ata_queue_packet_task(ap);
3789                 break;
3790
3791         case ATA_PROT_ATAPI_NODATA:
3792                 ap->flags |= ATA_FLAG_NOINTR;
3793                 ata_tf_to_host(ap, &qc->tf);
3794                 ata_queue_packet_task(ap);
3795                 break;
3796
3797         case ATA_PROT_ATAPI_DMA:
3798                 ap->flags |= ATA_FLAG_NOINTR;
3799                 ap->ops->tf_load(ap, &qc->tf);   /* load tf registers */
3800                 ap->ops->bmdma_setup(qc);           /* set up bmdma */
3801                 ata_queue_packet_task(ap);
3802                 break;
3803
3804         default:
3805                 WARN_ON(1);
3806                 return AC_ERR_SYSTEM;
3807         }
3808
3809         return 0;
3810 }
3811
3812 /**
3813  *      ata_bmdma_setup_mmio - Set up PCI IDE BMDMA transaction
3814  *      @qc: Info associated with this ATA transaction.
3815  *
3816  *      LOCKING:
3817  *      spin_lock_irqsave(host_set lock)
3818  */
3819
3820 static void ata_bmdma_setup_mmio (struct ata_queued_cmd *qc)
3821 {
3822         struct ata_port *ap = qc->ap;
3823         unsigned int rw = (qc->tf.flags & ATA_TFLAG_WRITE);
3824         u8 dmactl;
3825         void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
3826
3827         /* load PRD table addr. */
3828         mb();   /* make sure PRD table writes are visible to controller */
3829         writel(ap->prd_dma, mmio + ATA_DMA_TABLE_OFS);
3830
3831         /* specify data direction, triple-check start bit is clear */
3832         dmactl = readb(mmio + ATA_DMA_CMD);
3833         dmactl &= ~(ATA_DMA_WR | ATA_DMA_START);
3834         if (!rw)
3835                 dmactl |= ATA_DMA_WR;
3836         writeb(dmactl, mmio + ATA_DMA_CMD);
3837
3838         /* issue r/w command */
3839         ap->ops->exec_command(ap, &qc->tf);
3840 }
3841
3842 /**
3843  *      ata_bmdma_start_mmio - Start a PCI IDE BMDMA transaction
3844  *      @qc: Info associated with this ATA transaction.
3845  *
3846  *      LOCKING:
3847  *      spin_lock_irqsave(host_set lock)
3848  */
3849
3850 static void ata_bmdma_start_mmio (struct ata_queued_cmd *qc)
3851 {
3852         struct ata_port *ap = qc->ap;
3853         void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
3854         u8 dmactl;
3855
3856         /* start host DMA transaction */
3857         dmactl = readb(mmio + ATA_DMA_CMD);
3858         writeb(dmactl | ATA_DMA_START, mmio + ATA_DMA_CMD);
3859
3860         /* Strictly, one may wish to issue a readb() here, to
3861          * flush the mmio write.  However, control also passes
3862          * to the hardware at this point, and it will interrupt
3863          * us when we are to resume control.  So, in effect,
3864          * we don't care when the mmio write flushes.
3865          * Further, a read of the DMA status register _immediately_
3866          * following the write may not be what certain flaky hardware
3867          * is expected, so I think it is best to not add a readb()
3868          * without first all the MMIO ATA cards/mobos.
3869          * Or maybe I'm just being paranoid.
3870          */
3871 }
3872
3873 /**
3874  *      ata_bmdma_setup_pio - Set up PCI IDE BMDMA transaction (PIO)
3875  *      @qc: Info associated with this ATA transaction.
3876  *
3877  *      LOCKING:
3878  *      spin_lock_irqsave(host_set lock)
3879  */
3880
3881 static void ata_bmdma_setup_pio (struct ata_queued_cmd *qc)
3882 {
3883         struct ata_port *ap = qc->ap;
3884         unsigned int rw = (qc->tf.flags & ATA_TFLAG_WRITE);
3885         u8 dmactl;
3886
3887         /* load PRD table addr. */
3888         outl(ap->prd_dma, ap->ioaddr.bmdma_addr + ATA_DMA_TABLE_OFS);
3889
3890         /* specify data direction, triple-check start bit is clear */
3891         dmactl = inb(ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
3892         dmactl &= ~(ATA_DMA_WR | ATA_DMA_START);
3893         if (!rw)
3894                 dmactl |= ATA_DMA_WR;
3895         outb(dmactl, ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
3896
3897         /* issue r/w command */
3898         ap->ops->exec_command(ap, &qc->tf);
3899 }
3900
3901 /**
3902  *      ata_bmdma_start_pio - Start a PCI IDE BMDMA transaction (PIO)
3903  *      @qc: Info associated with this ATA transaction.
3904  *
3905  *      LOCKING:
3906  *      spin_lock_irqsave(host_set lock)
3907  */
3908
3909 static void ata_bmdma_start_pio (struct ata_queued_cmd *qc)
3910 {
3911         struct ata_port *ap = qc->ap;
3912         u8 dmactl;
3913
3914         /* start host DMA transaction */
3915         dmactl = inb(ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
3916         outb(dmactl | ATA_DMA_START,
3917              ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
3918 }
3919
3920
3921 /**
3922  *      ata_bmdma_start - Start a PCI IDE BMDMA transaction
3923  *      @qc: Info associated with this ATA transaction.
3924  *
3925  *      Writes the ATA_DMA_START flag to the DMA command register.
3926  *
3927  *      May be used as the bmdma_start() entry in ata_port_operations.
3928  *
3929  *      LOCKING:
3930  *      spin_lock_irqsave(host_set lock)
3931  */
3932 void ata_bmdma_start(struct ata_queued_cmd *qc)
3933 {
3934         if (qc->ap->flags & ATA_FLAG_MMIO)
3935                 ata_bmdma_start_mmio(qc);
3936         else
3937                 ata_bmdma_start_pio(qc);
3938 }
3939
3940
3941 /**
3942  *      ata_bmdma_setup - Set up PCI IDE BMDMA transaction
3943  *      @qc: Info associated with this ATA transaction.
3944  *
3945  *      Writes address of PRD table to device's PRD Table Address
3946  *      register, sets the DMA control register, and calls
3947  *      ops->exec_command() to start the transfer.
3948  *
3949  *      May be used as the bmdma_setup() entry in ata_port_operations.
3950  *
3951  *      LOCKING:
3952  *      spin_lock_irqsave(host_set lock)
3953  */
3954 void ata_bmdma_setup(struct ata_queued_cmd *qc)
3955 {
3956         if (qc->ap->flags & ATA_FLAG_MMIO)
3957                 ata_bmdma_setup_mmio(qc);
3958         else
3959                 ata_bmdma_setup_pio(qc);
3960 }
3961
3962
3963 /**
3964  *      ata_bmdma_irq_clear - Clear PCI IDE BMDMA interrupt.
3965  *      @ap: Port associated with this ATA transaction.
3966  *
3967  *      Clear interrupt and error flags in DMA status register.
3968  *
3969  *      May be used as the irq_clear() entry in ata_port_operations.
3970  *
3971  *      LOCKING:
3972  *      spin_lock_irqsave(host_set lock)
3973  */
3974
3975 void ata_bmdma_irq_clear(struct ata_port *ap)
3976 {
3977     if (ap->flags & ATA_FLAG_MMIO) {
3978         void __iomem *mmio = ((void __iomem *) ap->ioaddr.bmdma_addr) + ATA_DMA_STATUS;
3979         writeb(readb(mmio), mmio);
3980     } else {
3981         unsigned long addr = ap->ioaddr.bmdma_addr + ATA_DMA_STATUS;
3982         outb(inb(addr), addr);
3983     }
3984
3985 }
3986
3987
3988 /**
3989  *      ata_bmdma_status - Read PCI IDE BMDMA status
3990  *      @ap: Port associated with this ATA transaction.
3991  *
3992  *      Read and return BMDMA status register.
3993  *
3994  *      May be used as the bmdma_status() entry in ata_port_operations.
3995  *
3996  *      LOCKING:
3997  *      spin_lock_irqsave(host_set lock)
3998  */
3999
4000 u8 ata_bmdma_status(struct ata_port *ap)
4001 {
4002         u8 host_stat;
4003         if (ap->flags & ATA_FLAG_MMIO) {
4004                 void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
4005                 host_stat = readb(mmio + ATA_DMA_STATUS);
4006         } else
4007                 host_stat = inb(ap->ioaddr.bmdma_addr + ATA_DMA_STATUS);
4008         return host_stat;
4009 }
4010
4011
4012 /**
4013  *      ata_bmdma_stop - Stop PCI IDE BMDMA transfer
4014  *      @qc: Command we are ending DMA for
4015  *
4016  *      Clears the ATA_DMA_START flag in the dma control register
4017  *
4018  *      May be used as the bmdma_stop() entry in ata_port_operations.
4019  *
4020  *      LOCKING:
4021  *      spin_lock_irqsave(host_set lock)
4022  */
4023
4024 void ata_bmdma_stop(struct ata_queued_cmd *qc)
4025 {
4026         struct ata_port *ap = qc->ap;
4027         if (ap->flags & ATA_FLAG_MMIO) {
4028                 void __iomem *mmio = (void __iomem *) ap->ioaddr.bmdma_addr;
4029
4030                 /* clear start/stop bit */
4031                 writeb(readb(mmio + ATA_DMA_CMD) & ~ATA_DMA_START,
4032                         mmio + ATA_DMA_CMD);
4033         } else {
4034                 /* clear start/stop bit */
4035                 outb(inb(ap->ioaddr.bmdma_addr + ATA_DMA_CMD) & ~ATA_DMA_START,
4036                         ap->ioaddr.bmdma_addr + ATA_DMA_CMD);
4037         }
4038
4039         /* one-PIO-cycle guaranteed wait, per spec, for HDMA1:0 transition */
4040         ata_altstatus(ap);        /* dummy read */
4041 }
4042
4043 /**
4044  *      ata_host_intr - Handle host interrupt for given (port, task)
4045  *      @ap: Port on which interrupt arrived (possibly...)
4046  *      @qc: Taskfile currently active in engine
4047  *
4048  *      Handle host interrupt for given queued command.  Currently,
4049  *      only DMA interrupts are handled.  All other commands are
4050  *      handled via polling with interrupts disabled (nIEN bit).
4051  *
4052  *      LOCKING:
4053  *      spin_lock_irqsave(host_set lock)
4054  *
4055  *      RETURNS:
4056  *      One if interrupt was handled, zero if not (shared irq).
4057  */
4058
4059 inline unsigned int ata_host_intr (struct ata_port *ap,
4060                                    struct ata_queued_cmd *qc)
4061 {
4062         u8 status, host_stat;
4063
4064         switch (qc->tf.protocol) {
4065
4066         case ATA_PROT_DMA:
4067         case ATA_PROT_ATAPI_DMA:
4068         case ATA_PROT_ATAPI:
4069                 /* check status of DMA engine */
4070                 host_stat = ap->ops->bmdma_status(ap);
4071                 VPRINTK("ata%u: host_stat 0x%X\n", ap->id, host_stat);
4072
4073                 /* if it's not our irq... */
4074                 if (!(host_stat & ATA_DMA_INTR))
4075                         goto idle_irq;
4076
4077                 /* before we do anything else, clear DMA-Start bit */
4078                 ap->ops->bmdma_stop(qc);
4079
4080                 /* fall through */
4081
4082         case ATA_PROT_ATAPI_NODATA:
4083         case ATA_PROT_NODATA:
4084                 /* check altstatus */
4085                 status = ata_altstatus(ap);
4086                 if (status & ATA_BUSY)
4087                         goto idle_irq;
4088
4089                 /* check main status, clearing INTRQ */
4090                 status = ata_chk_status(ap);
4091                 if (unlikely(status & ATA_BUSY))
4092                         goto idle_irq;
4093                 DPRINTK("ata%u: protocol %d (dev_stat 0x%X)\n",
4094                         ap->id, qc->tf.protocol, status);
4095
4096                 /* ack bmdma irq events */
4097                 ap->ops->irq_clear(ap);
4098
4099                 /* complete taskfile transaction */
4100                 qc->err_mask |= ac_err_mask(status);
4101                 ata_qc_complete(qc);
4102                 break;
4103
4104         default:
4105                 goto idle_irq;
4106         }
4107
4108         return 1;       /* irq handled */
4109
4110 idle_irq:
4111         ap->stats.idle_irq++;
4112
4113 #ifdef ATA_IRQ_TRAP
4114         if ((ap->stats.idle_irq % 1000) == 0) {
4115                 handled = 1;
4116                 ata_irq_ack(ap, 0); /* debug trap */
4117                 printk(KERN_WARNING "ata%d: irq trap\n", ap->id);
4118         }
4119 #endif
4120         return 0;       /* irq not handled */
4121 }
4122
4123 /**
4124  *      ata_interrupt - Default ATA host interrupt handler
4125  *      @irq: irq line (unused)
4126  *      @dev_instance: pointer to our ata_host_set information structure
4127  *      @regs: unused
4128  *
4129  *      Default interrupt handler for PCI IDE devices.  Calls
4130  *      ata_host_intr() for each port that is not disabled.
4131  *
4132  *      LOCKING:
4133  *      Obtains host_set lock during operation.
4134  *
4135  *      RETURNS:
4136  *      IRQ_NONE or IRQ_HANDLED.
4137  */
4138
4139 irqreturn_t ata_interrupt (int irq, void *dev_instance, struct pt_regs *regs)
4140 {
4141         struct ata_host_set *host_set = dev_instance;
4142         unsigned int i;
4143         unsigned int handled = 0;
4144         unsigned long flags;
4145
4146         /* TODO: make _irqsave conditional on x86 PCI IDE legacy mode */
4147         spin_lock_irqsave(&host_set->lock, flags);
4148
4149         for (i = 0; i < host_set->n_ports; i++) {
4150                 struct ata_port *ap;
4151
4152                 ap = host_set->ports[i];
4153                 if (ap &&
4154                     !(ap->flags & (ATA_FLAG_PORT_DISABLED | ATA_FLAG_NOINTR))) {
4155                         struct ata_queued_cmd *qc;
4156
4157                         qc = ata_qc_from_tag(ap, ap->active_tag);
4158                         if (qc && (!(qc->tf.ctl & ATA_NIEN)) &&
4159                             (qc->flags & ATA_QCFLAG_ACTIVE))
4160                                 handled |= ata_host_intr(ap, qc);
4161                 }
4162         }
4163
4164         spin_unlock_irqrestore(&host_set->lock, flags);
4165
4166         return IRQ_RETVAL(handled);
4167 }
4168
4169 /**
4170  *      atapi_packet_task - Write CDB bytes to hardware
4171  *      @_data: Port to which ATAPI device is attached.
4172  *
4173  *      When device has indicated its readiness to accept
4174  *      a CDB, this function is called.  Send the CDB.
4175  *      If DMA is to be performed, exit immediately.
4176  *      Otherwise, we are in polling mode, so poll
4177  *      status under operation succeeds or fails.
4178  *
4179  *      LOCKING:
4180  *      Kernel thread context (may sleep)
4181  */
4182
4183 static void atapi_packet_task(void *_data)
4184 {
4185         struct ata_port *ap = _data;
4186         struct ata_queued_cmd *qc;
4187         u8 status;
4188
4189         qc = ata_qc_from_tag(ap, ap->active_tag);
4190         WARN_ON(qc == NULL);
4191         WARN_ON(!(qc->flags & ATA_QCFLAG_ACTIVE));
4192
4193         /* sleep-wait for BSY to clear */
4194         DPRINTK("busy wait\n");
4195         if (ata_busy_sleep(ap, ATA_TMOUT_CDB_QUICK, ATA_TMOUT_CDB)) {
4196                 qc->err_mask |= AC_ERR_TIMEOUT;
4197                 goto err_out;
4198         }
4199
4200         /* make sure DRQ is set */
4201         status = ata_chk_status(ap);
4202         if ((status & (ATA_BUSY | ATA_DRQ)) != ATA_DRQ) {
4203                 qc->err_mask |= AC_ERR_HSM;
4204                 goto err_out;
4205         }
4206
4207         /* send SCSI cdb */
4208         DPRINTK("send cdb\n");
4209         WARN_ON(qc->dev->cdb_len < 12);
4210
4211         if (qc->tf.protocol == ATA_PROT_ATAPI_DMA ||
4212             qc->tf.protocol == ATA_PROT_ATAPI_NODATA) {
4213                 unsigned long flags;
4214
4215                 /* Once we're done issuing command and kicking bmdma,
4216                  * irq handler takes over.  To not lose irq, we need
4217                  * to clear NOINTR flag before sending cdb, but
4218                  * interrupt handler shouldn't be invoked before we're
4219                  * finished.  Hence, the following locking.
4220                  */
4221                 spin_lock_irqsave(&ap->host_set->lock, flags);
4222                 ap->flags &= ~ATA_FLAG_NOINTR;
4223                 ata_data_xfer(ap, qc->cdb, qc->dev->cdb_len, 1);
4224                 if (qc->tf.protocol == ATA_PROT_ATAPI_DMA)
4225                         ap->ops->bmdma_start(qc);       /* initiate bmdma */
4226                 spin_unlock_irqrestore(&ap->host_set->lock, flags);
4227         } else {
4228                 ata_data_xfer(ap, qc->cdb, qc->dev->cdb_len, 1);
4229
4230                 /* PIO commands are handled by polling */
4231                 ap->hsm_task_state = HSM_ST;
4232                 ata_queue_pio_task(ap);
4233         }
4234
4235         return;
4236
4237 err_out:
4238         ata_poll_qc_complete(qc);
4239 }
4240
4241
4242 /*
4243  * Execute a 'simple' command, that only consists of the opcode 'cmd' itself,
4244  * without filling any other registers
4245  */
4246 static int ata_do_simple_cmd(struct ata_port *ap, struct ata_device *dev,
4247                              u8 cmd)
4248 {
4249         struct ata_taskfile tf;
4250         int err;
4251
4252         ata_tf_init(ap, &tf, dev->devno);
4253
4254         tf.command = cmd;
4255         tf.flags |= ATA_TFLAG_DEVICE;
4256         tf.protocol = ATA_PROT_NODATA;
4257
4258         err = ata_exec_internal(ap, dev, &tf, DMA_NONE, NULL, 0);
4259         if (err)
4260                 printk(KERN_ERR "%s: ata command failed: %d\n",
4261                                 __FUNCTION__, err);
4262
4263         return err;
4264 }
4265
4266 static int ata_flush_cache(struct ata_port *ap, struct ata_device *dev)
4267 {
4268         u8 cmd;
4269
4270         if (!ata_try_flush_cache(dev))
4271                 return 0;
4272
4273         if (ata_id_has_flush_ext(dev->id))
4274                 cmd = ATA_CMD_FLUSH_EXT;
4275         else
4276                 cmd = ATA_CMD_FLUSH;
4277
4278         return ata_do_simple_cmd(ap, dev, cmd);
4279 }
4280
4281 static int ata_standby_drive(struct ata_port *ap, struct ata_device *dev)
4282 {
4283         return ata_do_simple_cmd(ap, dev, ATA_CMD_STANDBYNOW1);
4284 }
4285
4286 static int ata_start_drive(struct ata_port *ap, struct ata_device *dev)
4287 {
4288         return ata_do_simple_cmd(ap, dev, ATA_CMD_IDLEIMMEDIATE);
4289 }
4290
4291 /**
4292  *      ata_device_resume - wakeup a previously suspended devices
4293  *      @ap: port the device is connected to
4294  *      @dev: the device to resume
4295  *
4296  *      Kick the drive back into action, by sending it an idle immediate
4297  *      command and making sure its transfer mode matches between drive
4298  *      and host.
4299  *
4300  */
4301 int ata_device_resume(struct ata_port *ap, struct ata_device *dev)
4302 {
4303         if (ap->flags & ATA_FLAG_SUSPENDED) {
4304                 ap->flags &= ~ATA_FLAG_SUSPENDED;
4305                 ata_set_mode(ap);
4306         }
4307         if (!ata_dev_present(dev))
4308                 return 0;
4309         if (dev->class == ATA_DEV_ATA)
4310                 ata_start_drive(ap, dev);
4311
4312         return 0;
4313 }
4314
4315 /**
4316  *      ata_device_suspend - prepare a device for suspend
4317  *      @ap: port the device is connected to
4318  *      @dev: the device to suspend
4319  *
4320  *      Flush the cache on the drive, if appropriate, then issue a
4321  *      standbynow command.
4322  */
4323 int ata_device_suspend(struct ata_port *ap, struct ata_device *dev)
4324 {
4325         if (!ata_dev_present(dev))
4326                 return 0;
4327         if (dev->class == ATA_DEV_ATA)
4328                 ata_flush_cache(ap, dev);
4329
4330         ata_standby_drive(ap, dev);
4331         ap->flags |= ATA_FLAG_SUSPENDED;
4332         return 0;
4333 }
4334
4335 /**
4336  *      ata_port_start - Set port up for dma.
4337  *      @ap: Port to initialize
4338  *
4339  *      Called just after data structures for each port are
4340  *      initialized.  Allocates space for PRD table.
4341  *
4342  *      May be used as the port_start() entry in ata_port_operations.
4343  *
4344  *      LOCKING:
4345  *      Inherited from caller.
4346  */
4347
4348 int ata_port_start (struct ata_port *ap)
4349 {
4350         struct device *dev = ap->host_set->dev;
4351         int rc;
4352
4353         ap->prd = dma_alloc_coherent(dev, ATA_PRD_TBL_SZ, &ap->prd_dma, GFP_KERNEL);
4354         if (!ap->prd)
4355                 return -ENOMEM;
4356
4357         rc = ata_pad_alloc(ap, dev);
4358         if (rc) {
4359                 dma_free_coherent(dev, ATA_PRD_TBL_SZ, ap->prd, ap->prd_dma);
4360                 return rc;
4361         }
4362
4363         DPRINTK("prd alloc, virt %p, dma %llx\n", ap->prd, (unsigned long long) ap->prd_dma);
4364
4365         return 0;
4366 }
4367
4368
4369 /**
4370  *      ata_port_stop - Undo ata_port_start()
4371  *      @ap: Port to shut down
4372  *
4373  *      Frees the PRD table.
4374  *
4375  *      May be used as the port_stop() entry in ata_port_operations.
4376  *
4377  *      LOCKING:
4378  *      Inherited from caller.
4379  */
4380
4381 void ata_port_stop (struct ata_port *ap)
4382 {
4383         struct device *dev = ap->host_set->dev;
4384
4385         dma_free_coherent(dev, ATA_PRD_TBL_SZ, ap->prd, ap->prd_dma);
4386         ata_pad_free(ap, dev);
4387 }
4388
4389 void ata_host_stop (struct ata_host_set *host_set)
4390 {
4391         if (host_set->mmio_base)
4392                 iounmap(host_set->mmio_base);
4393 }
4394
4395
4396 /**
4397  *      ata_host_remove - Unregister SCSI host structure with upper layers
4398  *      @ap: Port to unregister
4399  *      @do_unregister: 1 if we fully unregister, 0 to just stop the port
4400  *
4401  *      LOCKING:
4402  *      Inherited from caller.
4403  */
4404
4405 static void ata_host_remove(struct ata_port *ap, unsigned int do_unregister)
4406 {
4407         struct Scsi_Host *sh = ap->host;
4408
4409         DPRINTK("ENTER\n");
4410
4411         if (do_unregister)
4412                 scsi_remove_host(sh);
4413
4414         ap->ops->port_stop(ap);
4415 }
4416
4417 /**
4418  *      ata_host_init - Initialize an ata_port structure
4419  *      @ap: Structure to initialize
4420  *      @host: associated SCSI mid-layer structure
4421  *      @host_set: Collection of hosts to which @ap belongs
4422  *      @ent: Probe information provided by low-level driver
4423  *      @port_no: Port number associated with this ata_port
4424  *
4425  *      Initialize a new ata_port structure, and its associated
4426  *      scsi_host.
4427  *
4428  *      LOCKING:
4429  *      Inherited from caller.
4430  */
4431
4432 static void ata_host_init(struct ata_port *ap, struct Scsi_Host *host,
4433                           struct ata_host_set *host_set,
4434                           const struct ata_probe_ent *ent, unsigned int port_no)
4435 {
4436         unsigned int i;
4437
4438         host->max_id = 16;
4439         host->max_lun = 1;
4440         host->max_channel = 1;
4441         host->unique_id = ata_unique_id++;
4442         host->max_cmd_len = 12;
4443
4444         ap->flags = ATA_FLAG_PORT_DISABLED;
4445         ap->id = host->unique_id;
4446         ap->host = host;
4447         ap->ctl = ATA_DEVCTL_OBS;
4448         ap->host_set = host_set;
4449         ap->port_no = port_no;
4450         ap->hard_port_no =
4451                 ent->legacy_mode ? ent->hard_port_no : port_no;
4452         ap->pio_mask = ent->pio_mask;
4453         ap->mwdma_mask = ent->mwdma_mask;
4454         ap->udma_mask = ent->udma_mask;
4455         ap->flags |= ent->host_flags;
4456         ap->ops = ent->port_ops;
4457         ap->cbl = ATA_CBL_NONE;
4458         ap->active_tag = ATA_TAG_POISON;
4459         ap->last_ctl = 0xFF;
4460
4461         INIT_WORK(&ap->packet_task, atapi_packet_task, ap);
4462         INIT_WORK(&ap->pio_task, ata_pio_task, ap);
4463         INIT_LIST_HEAD(&ap->eh_done_q);
4464
4465         for (i = 0; i < ATA_MAX_DEVICES; i++)
4466                 ap->device[i].devno = i;
4467
4468 #ifdef ATA_IRQ_TRAP
4469         ap->stats.unhandled_irq = 1;
4470         ap->stats.idle_irq = 1;
4471 #endif
4472
4473         memcpy(&ap->ioaddr, &ent->port[port_no], sizeof(struct ata_ioports));
4474 }
4475
4476 /**
4477  *      ata_host_add - Attach low-level ATA driver to system
4478  *      @ent: Information provided by low-level driver
4479  *      @host_set: Collections of ports to which we add
4480  *      @port_no: Port number associated with this host
4481  *
4482  *      Attach low-level ATA driver to system.
4483  *
4484  *      LOCKING:
4485  *      PCI/etc. bus probe sem.
4486  *
4487  *      RETURNS:
4488  *      New ata_port on success, for NULL on error.
4489  */
4490
4491 static struct ata_port * ata_host_add(const struct ata_probe_ent *ent,
4492                                       struct ata_host_set *host_set,
4493                                       unsigned int port_no)
4494 {
4495         struct Scsi_Host *host;
4496         struct ata_port *ap;
4497         int rc;
4498
4499         DPRINTK("ENTER\n");
4500         host = scsi_host_alloc(ent->sht, sizeof(struct ata_port));
4501         if (!host)
4502                 return NULL;
4503
4504         ap = (struct ata_port *) &host->hostdata[0];
4505
4506         ata_host_init(ap, host, host_set, ent, port_no);
4507
4508         rc = ap->ops->port_start(ap);
4509         if (rc)
4510                 goto err_out;
4511
4512         return ap;
4513
4514 err_out:
4515         scsi_host_put(host);
4516         return NULL;
4517 }
4518
4519 /**
4520  *      ata_device_add - Register hardware device with ATA and SCSI layers
4521  *      @ent: Probe information describing hardware device to be registered
4522  *
4523  *      This function processes the information provided in the probe
4524  *      information struct @ent, allocates the necessary ATA and SCSI
4525  *      host information structures, initializes them, and registers
4526  *      everything with requisite kernel subsystems.
4527  *
4528  *      This function requests irqs, probes the ATA bus, and probes
4529  *      the SCSI bus.
4530  *
4531  *      LOCKING:
4532  *      PCI/etc. bus probe sem.
4533  *
4534  *      RETURNS:
4535  *      Number of ports registered.  Zero on error (no ports registered).
4536  */
4537
4538 int ata_device_add(const struct ata_probe_ent *ent)
4539 {
4540         unsigned int count = 0, i;
4541         struct device *dev = ent->dev;
4542         struct ata_host_set *host_set;
4543
4544         DPRINTK("ENTER\n");
4545         /* alloc a container for our list of ATA ports (buses) */
4546         host_set = kzalloc(sizeof(struct ata_host_set) +
4547                            (ent->n_ports * sizeof(void *)), GFP_KERNEL);
4548         if (!host_set)
4549                 return 0;
4550         spin_lock_init(&host_set->lock);
4551
4552         host_set->dev = dev;
4553         host_set->n_ports = ent->n_ports;
4554         host_set->irq = ent->irq;
4555         host_set->mmio_base = ent->mmio_base;
4556         host_set->private_data = ent->private_data;
4557         host_set->ops = ent->port_ops;
4558
4559         /* register each port bound to this device */
4560         for (i = 0; i < ent->n_ports; i++) {
4561                 struct ata_port *ap;
4562                 unsigned long xfer_mode_mask;
4563
4564                 ap = ata_host_add(ent, host_set, i);
4565                 if (!ap)
4566                         goto err_out;
4567
4568                 host_set->ports[i] = ap;
4569                 xfer_mode_mask =(ap->udma_mask << ATA_SHIFT_UDMA) |
4570                                 (ap->mwdma_mask << ATA_SHIFT_MWDMA) |
4571                                 (ap->pio_mask << ATA_SHIFT_PIO);
4572
4573                 /* print per-port info to dmesg */
4574                 printk(KERN_INFO "ata%u: %cATA max %s cmd 0x%lX ctl 0x%lX "
4575                                  "bmdma 0x%lX irq %lu\n",
4576                         ap->id,
4577                         ap->flags & ATA_FLAG_SATA ? 'S' : 'P',
4578                         ata_mode_string(xfer_mode_mask),
4579                         ap->ioaddr.cmd_addr,
4580                         ap->ioaddr.ctl_addr,
4581                         ap->ioaddr.bmdma_addr,
4582                         ent->irq);
4583
4584                 ata_chk_status(ap);
4585                 host_set->ops->irq_clear(ap);
4586                 count++;
4587         }
4588
4589         if (!count)
4590                 goto err_free_ret;
4591
4592         /* obtain irq, that is shared between channels */
4593         if (request_irq(ent->irq, ent->port_ops->irq_handler, ent->irq_flags,
4594                         DRV_NAME, host_set))
4595                 goto err_out;
4596
4597         /* perform each probe synchronously */
4598         DPRINTK("probe begin\n");
4599         for (i = 0; i < count; i++) {
4600                 struct ata_port *ap;
4601                 int rc;
4602
4603                 ap = host_set->ports[i];
4604
4605                 DPRINTK("ata%u: bus probe begin\n", ap->id);
4606                 rc = ata_bus_probe(ap);
4607                 DPRINTK("ata%u: bus probe end\n", ap->id);
4608
4609                 if (rc) {
4610                         /* FIXME: do something useful here?
4611                          * Current libata behavior will
4612                          * tear down everything when
4613                          * the module is removed
4614                          * or the h/w is unplugged.
4615                          */
4616                 }
4617
4618                 rc = scsi_add_host(ap->host, dev);
4619                 if (rc) {
4620                         printk(KERN_ERR "ata%u: scsi_add_host failed\n",
4621                                ap->id);
4622                         /* FIXME: do something useful here */
4623                         /* FIXME: handle unconditional calls to
4624                          * scsi_scan_host and ata_host_remove, below,
4625                          * at the very least
4626                          */
4627                 }
4628         }
4629
4630         /* probes are done, now scan each port's disk(s) */
4631         DPRINTK("host probe begin\n");
4632         for (i = 0; i < count; i++) {
4633                 struct ata_port *ap = host_set->ports[i];
4634
4635                 ata_scsi_scan_host(ap);
4636         }
4637
4638         dev_set_drvdata(dev, host_set);
4639
4640         VPRINTK("EXIT, returning %u\n", ent->n_ports);
4641         return ent->n_ports; /* success */
4642
4643 err_out:
4644         for (i = 0; i < count; i++) {
4645                 ata_host_remove(host_set->ports[i], 1);
4646                 scsi_host_put(host_set->ports[i]->host);
4647         }
4648 err_free_ret:
4649         kfree(host_set);
4650         VPRINTK("EXIT, returning 0\n");
4651         return 0;
4652 }
4653
4654 /**
4655  *      ata_host_set_remove - PCI layer callback for device removal
4656  *      @host_set: ATA host set that was removed
4657  *
4658  *      Unregister all objects associated with this host set. Free those
4659  *      objects.
4660  *
4661  *      LOCKING:
4662  *      Inherited from calling layer (may sleep).
4663  */
4664
4665 void ata_host_set_remove(struct ata_host_set *host_set)
4666 {
4667         struct ata_port *ap;
4668         unsigned int i;
4669
4670         for (i = 0; i < host_set->n_ports; i++) {
4671                 ap = host_set->ports[i];
4672                 scsi_remove_host(ap->host);
4673         }
4674
4675         free_irq(host_set->irq, host_set);
4676
4677         for (i = 0; i < host_set->n_ports; i++) {
4678                 ap = host_set->ports[i];
4679
4680                 ata_scsi_release(ap->host);
4681
4682                 if ((ap->flags & ATA_FLAG_NO_LEGACY) == 0) {
4683                         struct ata_ioports *ioaddr = &ap->ioaddr;
4684
4685                         if (ioaddr->cmd_addr == 0x1f0)
4686                                 release_region(0x1f0, 8);
4687                         else if (ioaddr->cmd_addr == 0x170)
4688                                 release_region(0x170, 8);
4689                 }
4690
4691                 scsi_host_put(ap->host);
4692         }
4693
4694         if (host_set->ops->host_stop)
4695                 host_set->ops->host_stop(host_set);
4696
4697         kfree(host_set);
4698 }
4699
4700 /**
4701  *      ata_scsi_release - SCSI layer callback hook for host unload
4702  *      @host: libata host to be unloaded
4703  *
4704  *      Performs all duties necessary to shut down a libata port...
4705  *      Kill port kthread, disable port, and release resources.
4706  *
4707  *      LOCKING:
4708  *      Inherited from SCSI layer.
4709  *
4710  *      RETURNS:
4711  *      One.
4712  */
4713
4714 int ata_scsi_release(struct Scsi_Host *host)
4715 {
4716         struct ata_port *ap = (struct ata_port *) &host->hostdata[0];
4717
4718         DPRINTK("ENTER\n");
4719
4720         ap->ops->port_disable(ap);
4721         ata_host_remove(ap, 0);
4722
4723         DPRINTK("EXIT\n");
4724         return 1;
4725 }
4726
4727 /**
4728  *      ata_std_ports - initialize ioaddr with standard port offsets.
4729  *      @ioaddr: IO address structure to be initialized
4730  *
4731  *      Utility function which initializes data_addr, error_addr,
4732  *      feature_addr, nsect_addr, lbal_addr, lbam_addr, lbah_addr,
4733  *      device_addr, status_addr, and command_addr to standard offsets
4734  *      relative to cmd_addr.
4735  *
4736  *      Does not set ctl_addr, altstatus_addr, bmdma_addr, or scr_addr.
4737  */
4738
4739 void ata_std_ports(struct ata_ioports *ioaddr)
4740 {
4741         ioaddr->data_addr = ioaddr->cmd_addr + ATA_REG_DATA;
4742         ioaddr->error_addr = ioaddr->cmd_addr + ATA_REG_ERR;
4743         ioaddr->feature_addr = ioaddr->cmd_addr + ATA_REG_FEATURE;
4744         ioaddr->nsect_addr = ioaddr->cmd_addr + ATA_REG_NSECT;
4745         ioaddr->lbal_addr = ioaddr->cmd_addr + ATA_REG_LBAL;
4746         ioaddr->lbam_addr = ioaddr->cmd_addr + ATA_REG_LBAM;
4747         ioaddr->lbah_addr = ioaddr->cmd_addr + ATA_REG_LBAH;
4748         ioaddr->device_addr = ioaddr->cmd_addr + ATA_REG_DEVICE;
4749         ioaddr->status_addr = ioaddr->cmd_addr + ATA_REG_STATUS;
4750         ioaddr->command_addr = ioaddr->cmd_addr + ATA_REG_CMD;
4751 }
4752
4753
4754 #ifdef CONFIG_PCI
4755
4756 void ata_pci_host_stop (struct ata_host_set *host_set)
4757 {
4758         struct pci_dev *pdev = to_pci_dev(host_set->dev);
4759
4760         pci_iounmap(pdev, host_set->mmio_base);
4761 }
4762
4763 /**
4764  *      ata_pci_remove_one - PCI layer callback for device removal
4765  *      @pdev: PCI device that was removed
4766  *
4767  *      PCI layer indicates to libata via this hook that
4768  *      hot-unplug or module unload event has occurred.
4769  *      Handle this by unregistering all objects associated
4770  *      with this PCI device.  Free those objects.  Then finally
4771  *      release PCI resources and disable device.
4772  *
4773  *      LOCKING:
4774  *      Inherited from PCI layer (may sleep).
4775  */
4776
4777 void ata_pci_remove_one (struct pci_dev *pdev)
4778 {
4779         struct device *dev = pci_dev_to_dev(pdev);
4780         struct ata_host_set *host_set = dev_get_drvdata(dev);
4781
4782         ata_host_set_remove(host_set);
4783         pci_release_regions(pdev);
4784         pci_disable_device(pdev);
4785         dev_set_drvdata(dev, NULL);
4786 }
4787
4788 /* move to PCI subsystem */
4789 int pci_test_config_bits(struct pci_dev *pdev, const struct pci_bits *bits)
4790 {
4791         unsigned long tmp = 0;
4792
4793         switch (bits->width) {
4794         case 1: {
4795                 u8 tmp8 = 0;
4796                 pci_read_config_byte(pdev, bits->reg, &tmp8);
4797                 tmp = tmp8;
4798                 break;
4799         }
4800         case 2: {
4801                 u16 tmp16 = 0;
4802                 pci_read_config_word(pdev, bits->reg, &tmp16);
4803                 tmp = tmp16;
4804                 break;
4805         }
4806         case 4: {
4807                 u32 tmp32 = 0;
4808                 pci_read_config_dword(pdev, bits->reg, &tmp32);
4809                 tmp = tmp32;
4810                 break;
4811         }
4812
4813         default:
4814                 return -EINVAL;
4815         }
4816
4817         tmp &= bits->mask;
4818
4819         return (tmp == bits->val) ? 1 : 0;
4820 }
4821
4822 int ata_pci_device_suspend(struct pci_dev *pdev, pm_message_t state)
4823 {
4824         pci_save_state(pdev);
4825         pci_disable_device(pdev);
4826         pci_set_power_state(pdev, PCI_D3hot);
4827         return 0;
4828 }
4829
4830 int ata_pci_device_resume(struct pci_dev *pdev)
4831 {
4832         pci_set_power_state(pdev, PCI_D0);
4833         pci_restore_state(pdev);
4834         pci_enable_device(pdev);
4835         pci_set_master(pdev);
4836         return 0;
4837 }
4838 #endif /* CONFIG_PCI */
4839
4840
4841 static int __init ata_init(void)
4842 {
4843         ata_wq = create_workqueue("ata");
4844         if (!ata_wq)
4845                 return -ENOMEM;
4846
4847         printk(KERN_DEBUG "libata version " DRV_VERSION " loaded.\n");
4848         return 0;
4849 }
4850
4851 static void __exit ata_exit(void)
4852 {
4853         destroy_workqueue(ata_wq);
4854 }
4855
4856 module_init(ata_init);
4857 module_exit(ata_exit);
4858
4859 static unsigned long ratelimit_time;
4860 static spinlock_t ata_ratelimit_lock = SPIN_LOCK_UNLOCKED;
4861
4862 int ata_ratelimit(void)
4863 {
4864         int rc;
4865         unsigned long flags;
4866
4867         spin_lock_irqsave(&ata_ratelimit_lock, flags);
4868
4869         if (time_after(jiffies, ratelimit_time)) {
4870                 rc = 1;
4871                 ratelimit_time = jiffies + (HZ/5);
4872         } else
4873                 rc = 0;
4874
4875         spin_unlock_irqrestore(&ata_ratelimit_lock, flags);
4876
4877         return rc;
4878 }
4879
4880 /*
4881  * libata is essentially a library of internal helper functions for
4882  * low-level ATA host controller drivers.  As such, the API/ABI is
4883  * likely to change as new drivers are added and updated.
4884  * Do not depend on ABI/API stability.
4885  */
4886
4887 EXPORT_SYMBOL_GPL(ata_std_bios_param);
4888 EXPORT_SYMBOL_GPL(ata_std_ports);
4889 EXPORT_SYMBOL_GPL(ata_device_add);
4890 EXPORT_SYMBOL_GPL(ata_host_set_remove);
4891 EXPORT_SYMBOL_GPL(ata_sg_init);
4892 EXPORT_SYMBOL_GPL(ata_sg_init_one);
4893 EXPORT_SYMBOL_GPL(__ata_qc_complete);
4894 EXPORT_SYMBOL_GPL(ata_qc_issue_prot);
4895 EXPORT_SYMBOL_GPL(ata_eng_timeout);
4896 EXPORT_SYMBOL_GPL(ata_tf_load);
4897 EXPORT_SYMBOL_GPL(ata_tf_read);
4898 EXPORT_SYMBOL_GPL(ata_noop_dev_select);
4899 EXPORT_SYMBOL_GPL(ata_std_dev_select);
4900 EXPORT_SYMBOL_GPL(ata_tf_to_fis);
4901 EXPORT_SYMBOL_GPL(ata_tf_from_fis);
4902 EXPORT_SYMBOL_GPL(ata_check_status);
4903 EXPORT_SYMBOL_GPL(ata_altstatus);
4904 EXPORT_SYMBOL_GPL(ata_exec_command);
4905 EXPORT_SYMBOL_GPL(ata_port_start);
4906 EXPORT_SYMBOL_GPL(ata_port_stop);
4907 EXPORT_SYMBOL_GPL(ata_host_stop);
4908 EXPORT_SYMBOL_GPL(ata_interrupt);
4909 EXPORT_SYMBOL_GPL(ata_qc_prep);
4910 EXPORT_SYMBOL_GPL(ata_bmdma_setup);
4911 EXPORT_SYMBOL_GPL(ata_bmdma_start);
4912 EXPORT_SYMBOL_GPL(ata_bmdma_irq_clear);
4913 EXPORT_SYMBOL_GPL(ata_bmdma_status);
4914 EXPORT_SYMBOL_GPL(ata_bmdma_stop);
4915 EXPORT_SYMBOL_GPL(ata_port_probe);
4916 EXPORT_SYMBOL_GPL(sata_phy_reset);
4917 EXPORT_SYMBOL_GPL(__sata_phy_reset);
4918 EXPORT_SYMBOL_GPL(ata_bus_reset);
4919 EXPORT_SYMBOL_GPL(ata_std_probeinit);
4920 EXPORT_SYMBOL_GPL(ata_std_softreset);
4921 EXPORT_SYMBOL_GPL(sata_std_hardreset);
4922 EXPORT_SYMBOL_GPL(ata_std_postreset);
4923 EXPORT_SYMBOL_GPL(ata_std_probe_reset);
4924 EXPORT_SYMBOL_GPL(ata_drive_probe_reset);
4925 EXPORT_SYMBOL_GPL(ata_port_disable);
4926 EXPORT_SYMBOL_GPL(ata_ratelimit);
4927 EXPORT_SYMBOL_GPL(ata_busy_sleep);
4928 EXPORT_SYMBOL_GPL(ata_scsi_ioctl);
4929 EXPORT_SYMBOL_GPL(ata_scsi_queuecmd);
4930 EXPORT_SYMBOL_GPL(ata_scsi_timed_out);
4931 EXPORT_SYMBOL_GPL(ata_scsi_error);
4932 EXPORT_SYMBOL_GPL(ata_scsi_slave_config);
4933 EXPORT_SYMBOL_GPL(ata_scsi_release);
4934 EXPORT_SYMBOL_GPL(ata_host_intr);
4935 EXPORT_SYMBOL_GPL(ata_dev_classify);
4936 EXPORT_SYMBOL_GPL(ata_dev_id_string);
4937 EXPORT_SYMBOL_GPL(ata_dev_id_c_string);
4938 EXPORT_SYMBOL_GPL(ata_dev_config);
4939 EXPORT_SYMBOL_GPL(ata_scsi_simulate);
4940 EXPORT_SYMBOL_GPL(ata_eh_qc_complete);
4941 EXPORT_SYMBOL_GPL(ata_eh_qc_retry);
4942
4943 EXPORT_SYMBOL_GPL(ata_pio_need_iordy);
4944 EXPORT_SYMBOL_GPL(ata_timing_compute);
4945 EXPORT_SYMBOL_GPL(ata_timing_merge);
4946
4947 #ifdef CONFIG_PCI
4948 EXPORT_SYMBOL_GPL(pci_test_config_bits);
4949 EXPORT_SYMBOL_GPL(ata_pci_host_stop);
4950 EXPORT_SYMBOL_GPL(ata_pci_init_native_mode);
4951 EXPORT_SYMBOL_GPL(ata_pci_init_one);
4952 EXPORT_SYMBOL_GPL(ata_pci_remove_one);
4953 EXPORT_SYMBOL_GPL(ata_pci_device_suspend);
4954 EXPORT_SYMBOL_GPL(ata_pci_device_resume);
4955 #endif /* CONFIG_PCI */
4956
4957 EXPORT_SYMBOL_GPL(ata_device_suspend);
4958 EXPORT_SYMBOL_GPL(ata_device_resume);
4959 EXPORT_SYMBOL_GPL(ata_scsi_device_suspend);
4960 EXPORT_SYMBOL_GPL(ata_scsi_device_resume);