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