ata: add/use ata_taskfile::{error|status} fields
[platform/kernel/linux-rpi.git] / drivers / ata / pata_macio.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Libata based driver for Apple "macio" family of PATA controllers
4  *
5  * Copyright 2008/2009 Benjamin Herrenschmidt, IBM Corp
6  *                     <benh@kernel.crashing.org>
7  *
8  * Some bits and pieces from drivers/ide/ppc/pmac.c
9  *
10  */
11
12 #undef DEBUG
13 #undef DEBUG_DMA
14
15 #include <linux/kernel.h>
16 #include <linux/module.h>
17 #include <linux/init.h>
18 #include <linux/blkdev.h>
19 #include <linux/ata.h>
20 #include <linux/libata.h>
21 #include <linux/adb.h>
22 #include <linux/pmu.h>
23 #include <linux/scatterlist.h>
24 #include <linux/of.h>
25 #include <linux/gfp.h>
26 #include <linux/pci.h>
27
28 #include <scsi/scsi.h>
29 #include <scsi/scsi_host.h>
30 #include <scsi/scsi_device.h>
31
32 #include <asm/macio.h>
33 #include <asm/io.h>
34 #include <asm/dbdma.h>
35 #include <asm/machdep.h>
36 #include <asm/pmac_feature.h>
37 #include <asm/mediabay.h>
38
39 #ifdef DEBUG_DMA
40 #define dev_dbgdma(dev, format, arg...)         \
41         dev_printk(KERN_DEBUG , dev , format , ## arg)
42 #else
43 #define dev_dbgdma(dev, format, arg...)         \
44         ({ if (0) dev_printk(KERN_DEBUG, dev, format, ##arg); 0; })
45 #endif
46
47 #define DRV_NAME        "pata_macio"
48 #define DRV_VERSION     "0.9"
49
50 /* Models of macio ATA controller */
51 enum {
52         controller_ohare,       /* OHare based */
53         controller_heathrow,    /* Heathrow/Paddington */
54         controller_kl_ata3,     /* KeyLargo ATA-3 */
55         controller_kl_ata4,     /* KeyLargo ATA-4 */
56         controller_un_ata6,     /* UniNorth2 ATA-6 */
57         controller_k2_ata6,     /* K2 ATA-6 */
58         controller_sh_ata6,     /* Shasta ATA-6 */
59 };
60
61 static const char* macio_ata_names[] = {
62         "OHare ATA",            /* OHare based */
63         "Heathrow ATA",         /* Heathrow/Paddington */
64         "KeyLargo ATA-3",       /* KeyLargo ATA-3 (MDMA only) */
65         "KeyLargo ATA-4",       /* KeyLargo ATA-4 (UDMA/66) */
66         "UniNorth ATA-6",       /* UniNorth2 ATA-6 (UDMA/100) */
67         "K2 ATA-6",             /* K2 ATA-6 (UDMA/100) */
68         "Shasta ATA-6",         /* Shasta ATA-6 (UDMA/133) */
69 };
70
71 /*
72  * Extra registers, both 32-bit little-endian
73  */
74 #define IDE_TIMING_CONFIG       0x200
75 #define IDE_INTERRUPT           0x300
76
77 /* Kauai (U2) ATA has different register setup */
78 #define IDE_KAUAI_PIO_CONFIG    0x200
79 #define IDE_KAUAI_ULTRA_CONFIG  0x210
80 #define IDE_KAUAI_POLL_CONFIG   0x220
81
82 /*
83  * Timing configuration register definitions
84  */
85
86 /* Number of IDE_SYSCLK_NS ticks, argument is in nanoseconds */
87 #define SYSCLK_TICKS(t)         (((t) + IDE_SYSCLK_NS - 1) / IDE_SYSCLK_NS)
88 #define SYSCLK_TICKS_66(t)      (((t) + IDE_SYSCLK_66_NS - 1) / IDE_SYSCLK_66_NS)
89 #define IDE_SYSCLK_NS           30      /* 33Mhz cell */
90 #define IDE_SYSCLK_66_NS        15      /* 66Mhz cell */
91
92 /* 133Mhz cell, found in shasta.
93  * See comments about 100 Mhz Uninorth 2...
94  * Note that PIO_MASK and MDMA_MASK seem to overlap, that's just
95  * weird and I don't now why .. at this stage
96  */
97 #define TR_133_PIOREG_PIO_MASK          0xff000fff
98 #define TR_133_PIOREG_MDMA_MASK         0x00fff800
99 #define TR_133_UDMAREG_UDMA_MASK        0x0003ffff
100 #define TR_133_UDMAREG_UDMA_EN          0x00000001
101
102 /* 100Mhz cell, found in Uninorth 2 and K2. It appears as a pci device
103  * (106b/0033) on uninorth or K2 internal PCI bus and it's clock is
104  * controlled like gem or fw. It appears to be an evolution of keylargo
105  * ATA4 with a timing register extended to 2x32bits registers (one
106  * for PIO & MWDMA and one for UDMA, and a similar DBDMA channel.
107  * It has it's own local feature control register as well.
108  *
109  * After scratching my mind over the timing values, at least for PIO
110  * and MDMA, I think I've figured the format of the timing register,
111  * though I use pre-calculated tables for UDMA as usual...
112  */
113 #define TR_100_PIO_ADDRSETUP_MASK       0xff000000 /* Size of field unknown */
114 #define TR_100_PIO_ADDRSETUP_SHIFT      24
115 #define TR_100_MDMA_MASK                0x00fff000
116 #define TR_100_MDMA_RECOVERY_MASK       0x00fc0000
117 #define TR_100_MDMA_RECOVERY_SHIFT      18
118 #define TR_100_MDMA_ACCESS_MASK         0x0003f000
119 #define TR_100_MDMA_ACCESS_SHIFT        12
120 #define TR_100_PIO_MASK                 0xff000fff
121 #define TR_100_PIO_RECOVERY_MASK        0x00000fc0
122 #define TR_100_PIO_RECOVERY_SHIFT       6
123 #define TR_100_PIO_ACCESS_MASK          0x0000003f
124 #define TR_100_PIO_ACCESS_SHIFT         0
125
126 #define TR_100_UDMAREG_UDMA_MASK        0x0000ffff
127 #define TR_100_UDMAREG_UDMA_EN          0x00000001
128
129
130 /* 66Mhz cell, found in KeyLargo. Can do ultra mode 0 to 2 on
131  * 40 connector cable and to 4 on 80 connector one.
132  * Clock unit is 15ns (66Mhz)
133  *
134  * 3 Values can be programmed:
135  *  - Write data setup, which appears to match the cycle time. They
136  *    also call it DIOW setup.
137  *  - Ready to pause time (from spec)
138  *  - Address setup. That one is weird. I don't see where exactly
139  *    it fits in UDMA cycles, I got it's name from an obscure piece
140  *    of commented out code in Darwin. They leave it to 0, we do as
141  *    well, despite a comment that would lead to think it has a
142  *    min value of 45ns.
143  * Apple also add 60ns to the write data setup (or cycle time ?) on
144  * reads.
145  */
146 #define TR_66_UDMA_MASK                 0xfff00000
147 #define TR_66_UDMA_EN                   0x00100000 /* Enable Ultra mode for DMA */
148 #define TR_66_PIO_ADDRSETUP_MASK        0xe0000000 /* Address setup */
149 #define TR_66_PIO_ADDRSETUP_SHIFT       29
150 #define TR_66_UDMA_RDY2PAUS_MASK        0x1e000000 /* Ready 2 pause time */
151 #define TR_66_UDMA_RDY2PAUS_SHIFT       25
152 #define TR_66_UDMA_WRDATASETUP_MASK     0x01e00000 /* Write data setup time */
153 #define TR_66_UDMA_WRDATASETUP_SHIFT    21
154 #define TR_66_MDMA_MASK                 0x000ffc00
155 #define TR_66_MDMA_RECOVERY_MASK        0x000f8000
156 #define TR_66_MDMA_RECOVERY_SHIFT       15
157 #define TR_66_MDMA_ACCESS_MASK          0x00007c00
158 #define TR_66_MDMA_ACCESS_SHIFT         10
159 #define TR_66_PIO_MASK                  0xe00003ff
160 #define TR_66_PIO_RECOVERY_MASK         0x000003e0
161 #define TR_66_PIO_RECOVERY_SHIFT        5
162 #define TR_66_PIO_ACCESS_MASK           0x0000001f
163 #define TR_66_PIO_ACCESS_SHIFT          0
164
165 /* 33Mhz cell, found in OHare, Heathrow (& Paddington) and KeyLargo
166  * Can do pio & mdma modes, clock unit is 30ns (33Mhz)
167  *
168  * The access time and recovery time can be programmed. Some older
169  * Darwin code base limit OHare to 150ns cycle time. I decided to do
170  * the same here fore safety against broken old hardware ;)
171  * The HalfTick bit, when set, adds half a clock (15ns) to the access
172  * time and removes one from recovery. It's not supported on KeyLargo
173  * implementation afaik. The E bit appears to be set for PIO mode 0 and
174  * is used to reach long timings used in this mode.
175  */
176 #define TR_33_MDMA_MASK                 0x003ff800
177 #define TR_33_MDMA_RECOVERY_MASK        0x001f0000
178 #define TR_33_MDMA_RECOVERY_SHIFT       16
179 #define TR_33_MDMA_ACCESS_MASK          0x0000f800
180 #define TR_33_MDMA_ACCESS_SHIFT         11
181 #define TR_33_MDMA_HALFTICK             0x00200000
182 #define TR_33_PIO_MASK                  0x000007ff
183 #define TR_33_PIO_E                     0x00000400
184 #define TR_33_PIO_RECOVERY_MASK         0x000003e0
185 #define TR_33_PIO_RECOVERY_SHIFT        5
186 #define TR_33_PIO_ACCESS_MASK           0x0000001f
187 #define TR_33_PIO_ACCESS_SHIFT          0
188
189 /*
190  * Interrupt register definitions. Only present on newer cells
191  * (Keylargo and later afaik) so we don't use it.
192  */
193 #define IDE_INTR_DMA                    0x80000000
194 #define IDE_INTR_DEVICE                 0x40000000
195
196 /*
197  * FCR Register on Kauai. Not sure what bit 0x4 is  ...
198  */
199 #define KAUAI_FCR_UATA_MAGIC            0x00000004
200 #define KAUAI_FCR_UATA_RESET_N          0x00000002
201 #define KAUAI_FCR_UATA_ENABLE           0x00000001
202
203
204 /* Allow up to 256 DBDMA commands per xfer */
205 #define MAX_DCMDS               256
206
207 /* Don't let a DMA segment go all the way to 64K */
208 #define MAX_DBDMA_SEG           0xff00
209
210
211 /*
212  * Wait 1s for disk to answer on IDE bus after a hard reset
213  * of the device (via GPIO/FCR).
214  *
215  * Some devices seem to "pollute" the bus even after dropping
216  * the BSY bit (typically some combo drives slave on the UDMA
217  * bus) after a hard reset. Since we hard reset all drives on
218  * KeyLargo ATA66, we have to keep that delay around. I may end
219  * up not hard resetting anymore on these and keep the delay only
220  * for older interfaces instead (we have to reset when coming
221  * from MacOS...) --BenH.
222  */
223 #define IDE_WAKEUP_DELAY_MS     1000
224
225 struct pata_macio_timing;
226
227 struct pata_macio_priv {
228         int                             kind;
229         int                             aapl_bus_id;
230         int                             mediabay : 1;
231         struct device_node              *node;
232         struct macio_dev                *mdev;
233         struct pci_dev                  *pdev;
234         struct device                   *dev;
235         int                             irq;
236         u32                             treg[2][2];
237         void __iomem                    *tfregs;
238         void __iomem                    *kauai_fcr;
239         struct dbdma_cmd *              dma_table_cpu;
240         dma_addr_t                      dma_table_dma;
241         struct ata_host                 *host;
242         const struct pata_macio_timing  *timings;
243 };
244
245 /* Previous variants of this driver used to calculate timings
246  * for various variants of the chip and use tables for others.
247  *
248  * Not only was this confusing, but in addition, it isn't clear
249  * whether our calculation code was correct. It didn't entirely
250  * match the darwin code and whatever documentation I could find
251  * on these cells
252  *
253  * I decided to entirely rely on a table instead for this version
254  * of the driver. Also, because I don't really care about derated
255  * modes and really old HW other than making it work, I'm not going
256  * to calculate / snoop timing values for something else than the
257  * standard modes.
258  */
259 struct pata_macio_timing {
260         int     mode;
261         u32     reg1;   /* Bits to set in first timing reg */
262         u32     reg2;   /* Bits to set in second timing reg */
263 };
264
265 static const struct pata_macio_timing pata_macio_ohare_timings[] = {
266         { XFER_PIO_0,           0x00000526,     0, },
267         { XFER_PIO_1,           0x00000085,     0, },
268         { XFER_PIO_2,           0x00000025,     0, },
269         { XFER_PIO_3,           0x00000025,     0, },
270         { XFER_PIO_4,           0x00000025,     0, },
271         { XFER_MW_DMA_0,        0x00074000,     0, },
272         { XFER_MW_DMA_1,        0x00221000,     0, },
273         { XFER_MW_DMA_2,        0x00211000,     0, },
274         { -1, 0, 0 }
275 };
276
277 static const struct pata_macio_timing pata_macio_heathrow_timings[] = {
278         { XFER_PIO_0,           0x00000526,     0, },
279         { XFER_PIO_1,           0x00000085,     0, },
280         { XFER_PIO_2,           0x00000025,     0, },
281         { XFER_PIO_3,           0x00000025,     0, },
282         { XFER_PIO_4,           0x00000025,     0, },
283         { XFER_MW_DMA_0,        0x00074000,     0, },
284         { XFER_MW_DMA_1,        0x00221000,     0, },
285         { XFER_MW_DMA_2,        0x00211000,     0, },
286         { -1, 0, 0 }
287 };
288
289 static const struct pata_macio_timing pata_macio_kl33_timings[] = {
290         { XFER_PIO_0,           0x00000526,     0, },
291         { XFER_PIO_1,           0x00000085,     0, },
292         { XFER_PIO_2,           0x00000025,     0, },
293         { XFER_PIO_3,           0x00000025,     0, },
294         { XFER_PIO_4,           0x00000025,     0, },
295         { XFER_MW_DMA_0,        0x00084000,     0, },
296         { XFER_MW_DMA_1,        0x00021800,     0, },
297         { XFER_MW_DMA_2,        0x00011800,     0, },
298         { -1, 0, 0 }
299 };
300
301 static const struct pata_macio_timing pata_macio_kl66_timings[] = {
302         { XFER_PIO_0,           0x0000038c,     0, },
303         { XFER_PIO_1,           0x0000020a,     0, },
304         { XFER_PIO_2,           0x00000127,     0, },
305         { XFER_PIO_3,           0x000000c6,     0, },
306         { XFER_PIO_4,           0x00000065,     0, },
307         { XFER_MW_DMA_0,        0x00084000,     0, },
308         { XFER_MW_DMA_1,        0x00029800,     0, },
309         { XFER_MW_DMA_2,        0x00019400,     0, },
310         { XFER_UDMA_0,          0x19100000,     0, },
311         { XFER_UDMA_1,          0x14d00000,     0, },
312         { XFER_UDMA_2,          0x10900000,     0, },
313         { XFER_UDMA_3,          0x0c700000,     0, },
314         { XFER_UDMA_4,          0x0c500000,     0, },
315         { -1, 0, 0 }
316 };
317
318 static const struct pata_macio_timing pata_macio_kauai_timings[] = {
319         { XFER_PIO_0,           0x08000a92,     0, },
320         { XFER_PIO_1,           0x0800060f,     0, },
321         { XFER_PIO_2,           0x0800038b,     0, },
322         { XFER_PIO_3,           0x05000249,     0, },
323         { XFER_PIO_4,           0x04000148,     0, },
324         { XFER_MW_DMA_0,        0x00618000,     0, },
325         { XFER_MW_DMA_1,        0x00209000,     0, },
326         { XFER_MW_DMA_2,        0x00148000,     0, },
327         { XFER_UDMA_0,                   0,     0x000070c1, },
328         { XFER_UDMA_1,                   0,     0x00005d81, },
329         { XFER_UDMA_2,                   0,     0x00004a61, },
330         { XFER_UDMA_3,                   0,     0x00003a51, },
331         { XFER_UDMA_4,                   0,     0x00002a31, },
332         { XFER_UDMA_5,                   0,     0x00002921, },
333         { -1, 0, 0 }
334 };
335
336 static const struct pata_macio_timing pata_macio_shasta_timings[] = {
337         { XFER_PIO_0,           0x0a000c97,     0, },
338         { XFER_PIO_1,           0x07000712,     0, },
339         { XFER_PIO_2,           0x040003cd,     0, },
340         { XFER_PIO_3,           0x0500028b,     0, },
341         { XFER_PIO_4,           0x0400010a,     0, },
342         { XFER_MW_DMA_0,        0x00820800,     0, },
343         { XFER_MW_DMA_1,        0x0028b000,     0, },
344         { XFER_MW_DMA_2,        0x001ca000,     0, },
345         { XFER_UDMA_0,                   0,     0x00035901, },
346         { XFER_UDMA_1,                   0,     0x000348b1, },
347         { XFER_UDMA_2,                   0,     0x00033881, },
348         { XFER_UDMA_3,                   0,     0x00033861, },
349         { XFER_UDMA_4,                   0,     0x00033841, },
350         { XFER_UDMA_5,                   0,     0x00033031, },
351         { XFER_UDMA_6,                   0,     0x00033021, },
352         { -1, 0, 0 }
353 };
354
355 static const struct pata_macio_timing *pata_macio_find_timing(
356                                             struct pata_macio_priv *priv,
357                                             int mode)
358 {
359         int i;
360
361         for (i = 0; priv->timings[i].mode > 0; i++) {
362                 if (priv->timings[i].mode == mode)
363                         return &priv->timings[i];
364         }
365         return NULL;
366 }
367
368
369 static void pata_macio_apply_timings(struct ata_port *ap, unsigned int device)
370 {
371         struct pata_macio_priv *priv = ap->private_data;
372         void __iomem *rbase = ap->ioaddr.cmd_addr;
373
374         if (priv->kind == controller_sh_ata6 ||
375             priv->kind == controller_un_ata6 ||
376             priv->kind == controller_k2_ata6) {
377                 writel(priv->treg[device][0], rbase + IDE_KAUAI_PIO_CONFIG);
378                 writel(priv->treg[device][1], rbase + IDE_KAUAI_ULTRA_CONFIG);
379         } else
380                 writel(priv->treg[device][0], rbase + IDE_TIMING_CONFIG);
381 }
382
383 static void pata_macio_dev_select(struct ata_port *ap, unsigned int device)
384 {
385         ata_sff_dev_select(ap, device);
386
387         /* Apply timings */
388         pata_macio_apply_timings(ap, device);
389 }
390
391 static void pata_macio_set_timings(struct ata_port *ap,
392                                    struct ata_device *adev)
393 {
394         struct pata_macio_priv *priv = ap->private_data;
395         const struct pata_macio_timing *t;
396
397         dev_dbg(priv->dev, "Set timings: DEV=%d,PIO=0x%x (%s),DMA=0x%x (%s)\n",
398                 adev->devno,
399                 adev->pio_mode,
400                 ata_mode_string(ata_xfer_mode2mask(adev->pio_mode)),
401                 adev->dma_mode,
402                 ata_mode_string(ata_xfer_mode2mask(adev->dma_mode)));
403
404         /* First clear timings */
405         priv->treg[adev->devno][0] = priv->treg[adev->devno][1] = 0;
406
407         /* Now get the PIO timings */
408         t = pata_macio_find_timing(priv, adev->pio_mode);
409         if (t == NULL) {
410                 dev_warn(priv->dev, "Invalid PIO timing requested: 0x%x\n",
411                          adev->pio_mode);
412                 t = pata_macio_find_timing(priv, XFER_PIO_0);
413         }
414         BUG_ON(t == NULL);
415
416         /* PIO timings only ever use the first treg */
417         priv->treg[adev->devno][0] |= t->reg1;
418
419         /* Now get DMA timings */
420         t = pata_macio_find_timing(priv, adev->dma_mode);
421         if (t == NULL || (t->reg1 == 0 && t->reg2 == 0)) {
422                 dev_dbg(priv->dev, "DMA timing not set yet, using MW_DMA_0\n");
423                 t = pata_macio_find_timing(priv, XFER_MW_DMA_0);
424         }
425         BUG_ON(t == NULL);
426
427         /* DMA timings can use both tregs */
428         priv->treg[adev->devno][0] |= t->reg1;
429         priv->treg[adev->devno][1] |= t->reg2;
430
431         dev_dbg(priv->dev, " -> %08x %08x\n",
432                 priv->treg[adev->devno][0],
433                 priv->treg[adev->devno][1]);
434
435         /* Apply to hardware */
436         pata_macio_apply_timings(ap, adev->devno);
437 }
438
439 /*
440  * Blast some well known "safe" values to the timing registers at init or
441  * wakeup from sleep time, before we do real calculation
442  */
443 static void pata_macio_default_timings(struct pata_macio_priv *priv)
444 {
445         unsigned int value, value2 = 0;
446
447         switch(priv->kind) {
448                 case controller_sh_ata6:
449                         value = 0x0a820c97;
450                         value2 = 0x00033031;
451                         break;
452                 case controller_un_ata6:
453                 case controller_k2_ata6:
454                         value = 0x08618a92;
455                         value2 = 0x00002921;
456                         break;
457                 case controller_kl_ata4:
458                         value = 0x0008438c;
459                         break;
460                 case controller_kl_ata3:
461                         value = 0x00084526;
462                         break;
463                 case controller_heathrow:
464                 case controller_ohare:
465                 default:
466                         value = 0x00074526;
467                         break;
468         }
469         priv->treg[0][0] = priv->treg[1][0] = value;
470         priv->treg[0][1] = priv->treg[1][1] = value2;
471 }
472
473 static int pata_macio_cable_detect(struct ata_port *ap)
474 {
475         struct pata_macio_priv *priv = ap->private_data;
476
477         /* Get cable type from device-tree */
478         if (priv->kind == controller_kl_ata4 ||
479             priv->kind == controller_un_ata6 ||
480             priv->kind == controller_k2_ata6 ||
481             priv->kind == controller_sh_ata6) {
482                 const char* cable = of_get_property(priv->node, "cable-type",
483                                                     NULL);
484                 struct device_node *root = of_find_node_by_path("/");
485                 const char *model = of_get_property(root, "model", NULL);
486
487                 of_node_put(root);
488
489                 if (cable && !strncmp(cable, "80-", 3)) {
490                         /* Some drives fail to detect 80c cable in PowerBook
491                          * These machine use proprietary short IDE cable
492                          * anyway
493                          */
494                         if (!strncmp(model, "PowerBook", 9))
495                                 return ATA_CBL_PATA40_SHORT;
496                         else
497                                 return ATA_CBL_PATA80;
498                 }
499         }
500
501         /* G5's seem to have incorrect cable type in device-tree.
502          * Let's assume they always have a 80 conductor cable, this seem to
503          * be always the case unless the user mucked around
504          */
505         if (of_device_is_compatible(priv->node, "K2-UATA") ||
506             of_device_is_compatible(priv->node, "shasta-ata"))
507                 return ATA_CBL_PATA80;
508
509         /* Anything else is 40 connectors */
510         return ATA_CBL_PATA40;
511 }
512
513 static enum ata_completion_errors pata_macio_qc_prep(struct ata_queued_cmd *qc)
514 {
515         unsigned int write = (qc->tf.flags & ATA_TFLAG_WRITE);
516         struct ata_port *ap = qc->ap;
517         struct pata_macio_priv *priv = ap->private_data;
518         struct scatterlist *sg;
519         struct dbdma_cmd *table;
520         unsigned int si, pi;
521
522         dev_dbgdma(priv->dev, "%s: qc %p flags %lx, write %d dev %d\n",
523                    __func__, qc, qc->flags, write, qc->dev->devno);
524
525         if (!(qc->flags & ATA_QCFLAG_DMAMAP))
526                 return AC_ERR_OK;
527
528         table = (struct dbdma_cmd *) priv->dma_table_cpu;
529
530         pi = 0;
531         for_each_sg(qc->sg, sg, qc->n_elem, si) {
532                 u32 addr, sg_len, len;
533
534                 /* determine if physical DMA addr spans 64K boundary.
535                  * Note h/w doesn't support 64-bit, so we unconditionally
536                  * truncate dma_addr_t to u32.
537                  */
538                 addr = (u32) sg_dma_address(sg);
539                 sg_len = sg_dma_len(sg);
540
541                 while (sg_len) {
542                         /* table overflow should never happen */
543                         BUG_ON (pi++ >= MAX_DCMDS);
544
545                         len = (sg_len < MAX_DBDMA_SEG) ? sg_len : MAX_DBDMA_SEG;
546                         table->command = cpu_to_le16(write ? OUTPUT_MORE: INPUT_MORE);
547                         table->req_count = cpu_to_le16(len);
548                         table->phy_addr = cpu_to_le32(addr);
549                         table->cmd_dep = 0;
550                         table->xfer_status = 0;
551                         table->res_count = 0;
552                         addr += len;
553                         sg_len -= len;
554                         ++table;
555                 }
556         }
557
558         /* Should never happen according to Tejun */
559         BUG_ON(!pi);
560
561         /* Convert the last command to an input/output */
562         table--;
563         table->command = cpu_to_le16(write ? OUTPUT_LAST: INPUT_LAST);
564         table++;
565
566         /* Add the stop command to the end of the list */
567         memset(table, 0, sizeof(struct dbdma_cmd));
568         table->command = cpu_to_le16(DBDMA_STOP);
569
570         dev_dbgdma(priv->dev, "%s: %d DMA list entries\n", __func__, pi);
571
572         return AC_ERR_OK;
573 }
574
575
576 static void pata_macio_freeze(struct ata_port *ap)
577 {
578         struct dbdma_regs __iomem *dma_regs = ap->ioaddr.bmdma_addr;
579
580         if (dma_regs) {
581                 unsigned int timeout = 1000000;
582
583                 /* Make sure DMA controller is stopped */
584                 writel((RUN|PAUSE|FLUSH|WAKE|DEAD) << 16, &dma_regs->control);
585                 while (--timeout && (readl(&dma_regs->status) & RUN))
586                         udelay(1);
587         }
588
589         ata_sff_freeze(ap);
590 }
591
592
593 static void pata_macio_bmdma_setup(struct ata_queued_cmd *qc)
594 {
595         struct ata_port *ap = qc->ap;
596         struct pata_macio_priv *priv = ap->private_data;
597         struct dbdma_regs __iomem *dma_regs = ap->ioaddr.bmdma_addr;
598         int dev = qc->dev->devno;
599
600         dev_dbgdma(priv->dev, "%s: qc %p\n", __func__, qc);
601
602         /* Make sure DMA commands updates are visible */
603         writel(priv->dma_table_dma, &dma_regs->cmdptr);
604
605         /* On KeyLargo 66Mhz cell, we need to add 60ns to wrDataSetup on
606          * UDMA reads
607          */
608         if (priv->kind == controller_kl_ata4 &&
609             (priv->treg[dev][0] & TR_66_UDMA_EN)) {
610                 void __iomem *rbase = ap->ioaddr.cmd_addr;
611                 u32 reg = priv->treg[dev][0];
612
613                 if (!(qc->tf.flags & ATA_TFLAG_WRITE))
614                         reg += 0x00800000;
615                 writel(reg, rbase + IDE_TIMING_CONFIG);
616         }
617
618         /* issue r/w command */
619         ap->ops->sff_exec_command(ap, &qc->tf);
620 }
621
622 static void pata_macio_bmdma_start(struct ata_queued_cmd *qc)
623 {
624         struct ata_port *ap = qc->ap;
625         struct pata_macio_priv *priv = ap->private_data;
626         struct dbdma_regs __iomem *dma_regs = ap->ioaddr.bmdma_addr;
627
628         dev_dbgdma(priv->dev, "%s: qc %p\n", __func__, qc);
629
630         writel((RUN << 16) | RUN, &dma_regs->control);
631         /* Make sure it gets to the controller right now */
632         (void)readl(&dma_regs->control);
633 }
634
635 static void pata_macio_bmdma_stop(struct ata_queued_cmd *qc)
636 {
637         struct ata_port *ap = qc->ap;
638         struct pata_macio_priv *priv = ap->private_data;
639         struct dbdma_regs __iomem *dma_regs = ap->ioaddr.bmdma_addr;
640         unsigned int timeout = 1000000;
641
642         dev_dbgdma(priv->dev, "%s: qc %p\n", __func__, qc);
643
644         /* Stop the DMA engine and wait for it to full halt */
645         writel (((RUN|WAKE|DEAD) << 16), &dma_regs->control);
646         while (--timeout && (readl(&dma_regs->status) & RUN))
647                 udelay(1);
648 }
649
650 static u8 pata_macio_bmdma_status(struct ata_port *ap)
651 {
652         struct pata_macio_priv *priv = ap->private_data;
653         struct dbdma_regs __iomem *dma_regs = ap->ioaddr.bmdma_addr;
654         u32 dstat, rstat = ATA_DMA_INTR;
655         unsigned long timeout = 0;
656
657         dstat = readl(&dma_regs->status);
658
659         dev_dbgdma(priv->dev, "%s: dstat=%x\n", __func__, dstat);
660
661         /* We have two things to deal with here:
662          *
663          * - The dbdma won't stop if the command was started
664          * but completed with an error without transferring all
665          * datas. This happens when bad blocks are met during
666          * a multi-block transfer.
667          *
668          * - The dbdma fifo hasn't yet finished flushing to
669          * to system memory when the disk interrupt occurs.
670          *
671          */
672
673         /* First check for errors */
674         if ((dstat & (RUN|DEAD)) != RUN)
675                 rstat |= ATA_DMA_ERR;
676
677         /* If ACTIVE is cleared, the STOP command has been hit and
678          * the transfer is complete. If not, we have to flush the
679          * channel.
680          */
681         if ((dstat & ACTIVE) == 0)
682                 return rstat;
683
684         dev_dbgdma(priv->dev, "%s: DMA still active, flushing...\n", __func__);
685
686         /* If dbdma didn't execute the STOP command yet, the
687          * active bit is still set. We consider that we aren't
688          * sharing interrupts (which is hopefully the case with
689          * those controllers) and so we just try to flush the
690          * channel for pending data in the fifo
691          */
692         udelay(1);
693         writel((FLUSH << 16) | FLUSH, &dma_regs->control);
694         for (;;) {
695                 udelay(1);
696                 dstat = readl(&dma_regs->status);
697                 if ((dstat & FLUSH) == 0)
698                         break;
699                 if (++timeout > 1000) {
700                         dev_warn(priv->dev, "timeout flushing DMA\n");
701                         rstat |= ATA_DMA_ERR;
702                         break;
703                 }
704         }
705         return rstat;
706 }
707
708 /* port_start is when we allocate the DMA command list */
709 static int pata_macio_port_start(struct ata_port *ap)
710 {
711         struct pata_macio_priv *priv = ap->private_data;
712
713         if (ap->ioaddr.bmdma_addr == NULL)
714                 return 0;
715
716         /* Allocate space for the DBDMA commands.
717          *
718          * The +2 is +1 for the stop command and +1 to allow for
719          * aligning the start address to a multiple of 16 bytes.
720          */
721         priv->dma_table_cpu =
722                 dmam_alloc_coherent(priv->dev,
723                                     (MAX_DCMDS + 2) * sizeof(struct dbdma_cmd),
724                                     &priv->dma_table_dma, GFP_KERNEL);
725         if (priv->dma_table_cpu == NULL) {
726                 dev_err(priv->dev, "Unable to allocate DMA command list\n");
727                 ap->ioaddr.bmdma_addr = NULL;
728                 ap->mwdma_mask = 0;
729                 ap->udma_mask = 0;
730         }
731         return 0;
732 }
733
734 static void pata_macio_irq_clear(struct ata_port *ap)
735 {
736         struct pata_macio_priv *priv = ap->private_data;
737
738         /* Nothing to do here */
739
740         dev_dbgdma(priv->dev, "%s\n", __func__);
741 }
742
743 static void pata_macio_reset_hw(struct pata_macio_priv *priv, int resume)
744 {
745         dev_dbg(priv->dev, "Enabling & resetting... \n");
746
747         if (priv->mediabay)
748                 return;
749
750         if (priv->kind == controller_ohare && !resume) {
751                 /* The code below is having trouble on some ohare machines
752                  * (timing related ?). Until I can put my hand on one of these
753                  * units, I keep the old way
754                  */
755                 ppc_md.feature_call(PMAC_FTR_IDE_ENABLE, priv->node, 0, 1);
756         } else {
757                 int rc;
758
759                 /* Reset and enable controller */
760                 rc = ppc_md.feature_call(PMAC_FTR_IDE_RESET,
761                                          priv->node, priv->aapl_bus_id, 1);
762                 ppc_md.feature_call(PMAC_FTR_IDE_ENABLE,
763                                     priv->node, priv->aapl_bus_id, 1);
764                 msleep(10);
765                 /* Only bother waiting if there's a reset control */
766                 if (rc == 0) {
767                         ppc_md.feature_call(PMAC_FTR_IDE_RESET,
768                                             priv->node, priv->aapl_bus_id, 0);
769                         msleep(IDE_WAKEUP_DELAY_MS);
770                 }
771         }
772
773         /* If resuming a PCI device, restore the config space here */
774         if (priv->pdev && resume) {
775                 int rc;
776
777                 pci_restore_state(priv->pdev);
778                 rc = pcim_enable_device(priv->pdev);
779                 if (rc)
780                         dev_err(&priv->pdev->dev,
781                                 "Failed to enable device after resume (%d)\n",
782                                 rc);
783                 else
784                         pci_set_master(priv->pdev);
785         }
786
787         /* On Kauai, initialize the FCR. We don't perform a reset, doesn't really
788          * seem necessary and speeds up the boot process
789          */
790         if (priv->kauai_fcr)
791                 writel(KAUAI_FCR_UATA_MAGIC |
792                        KAUAI_FCR_UATA_RESET_N |
793                        KAUAI_FCR_UATA_ENABLE, priv->kauai_fcr);
794 }
795
796 /* Hook the standard slave config to fixup some HW related alignment
797  * restrictions
798  */
799 static int pata_macio_slave_config(struct scsi_device *sdev)
800 {
801         struct ata_port *ap = ata_shost_to_port(sdev->host);
802         struct pata_macio_priv *priv = ap->private_data;
803         struct ata_device *dev;
804         u16 cmd;
805         int rc;
806
807         /* First call original */
808         rc = ata_scsi_slave_config(sdev);
809         if (rc)
810                 return rc;
811
812         /* This is lifted from sata_nv */
813         dev = &ap->link.device[sdev->id];
814
815         /* OHare has issues with non cache aligned DMA on some chipsets */
816         if (priv->kind == controller_ohare) {
817                 blk_queue_update_dma_alignment(sdev->request_queue, 31);
818                 blk_queue_update_dma_pad(sdev->request_queue, 31);
819
820                 /* Tell the world about it */
821                 ata_dev_info(dev, "OHare alignment limits applied\n");
822                 return 0;
823         }
824
825         /* We only have issues with ATAPI */
826         if (dev->class != ATA_DEV_ATAPI)
827                 return 0;
828
829         /* Shasta and K2 seem to have "issues" with reads ... */
830         if (priv->kind == controller_sh_ata6 || priv->kind == controller_k2_ata6) {
831                 /* Allright these are bad, apply restrictions */
832                 blk_queue_update_dma_alignment(sdev->request_queue, 15);
833                 blk_queue_update_dma_pad(sdev->request_queue, 15);
834
835                 /* We enable MWI and hack cache line size directly here, this
836                  * is specific to this chipset and not normal values, we happen
837                  * to somewhat know what we are doing here (which is basically
838                  * to do the same Apple does and pray they did not get it wrong :-)
839                  */
840                 BUG_ON(!priv->pdev);
841                 pci_write_config_byte(priv->pdev, PCI_CACHE_LINE_SIZE, 0x08);
842                 pci_read_config_word(priv->pdev, PCI_COMMAND, &cmd);
843                 pci_write_config_word(priv->pdev, PCI_COMMAND,
844                                       cmd | PCI_COMMAND_INVALIDATE);
845
846                 /* Tell the world about it */
847                 ata_dev_info(dev, "K2/Shasta alignment limits applied\n");
848         }
849
850         return 0;
851 }
852
853 #ifdef CONFIG_PM_SLEEP
854 static int pata_macio_do_suspend(struct pata_macio_priv *priv, pm_message_t mesg)
855 {
856         int rc;
857
858         /* First, core libata suspend to do most of the work */
859         rc = ata_host_suspend(priv->host, mesg);
860         if (rc)
861                 return rc;
862
863         /* Restore to default timings */
864         pata_macio_default_timings(priv);
865
866         /* Mask interrupt. Not strictly necessary but old driver did
867          * it and I'd rather not change that here */
868         disable_irq(priv->irq);
869
870         /* The media bay will handle itself just fine */
871         if (priv->mediabay)
872                 return 0;
873
874         /* Kauai has bus control FCRs directly here */
875         if (priv->kauai_fcr) {
876                 u32 fcr = readl(priv->kauai_fcr);
877                 fcr &= ~(KAUAI_FCR_UATA_RESET_N | KAUAI_FCR_UATA_ENABLE);
878                 writel(fcr, priv->kauai_fcr);
879         }
880
881         /* For PCI, save state and disable DMA. No need to call
882          * pci_set_power_state(), the HW doesn't do D states that
883          * way, the platform code will take care of suspending the
884          * ASIC properly
885          */
886         if (priv->pdev) {
887                 pci_save_state(priv->pdev);
888                 pci_disable_device(priv->pdev);
889         }
890
891         /* Disable the bus on older machines and the cell on kauai */
892         ppc_md.feature_call(PMAC_FTR_IDE_ENABLE, priv->node,
893                             priv->aapl_bus_id, 0);
894
895         return 0;
896 }
897
898 static int pata_macio_do_resume(struct pata_macio_priv *priv)
899 {
900         /* Reset and re-enable the HW */
901         pata_macio_reset_hw(priv, 1);
902
903         /* Sanitize drive timings */
904         pata_macio_apply_timings(priv->host->ports[0], 0);
905
906         /* We want our IRQ back ! */
907         enable_irq(priv->irq);
908
909         /* Let the libata core take it from there */
910         ata_host_resume(priv->host);
911
912         return 0;
913 }
914 #endif /* CONFIG_PM_SLEEP */
915
916 static struct scsi_host_template pata_macio_sht = {
917         __ATA_BASE_SHT(DRV_NAME),
918         .sg_tablesize           = MAX_DCMDS,
919         /* We may not need that strict one */
920         .dma_boundary           = ATA_DMA_BOUNDARY,
921         /* Not sure what the real max is but we know it's less than 64K, let's
922          * use 64K minus 256
923          */
924         .max_segment_size       = MAX_DBDMA_SEG,
925         .slave_configure        = pata_macio_slave_config,
926         .sdev_attrs             = ata_common_sdev_attrs,
927         .can_queue              = ATA_DEF_QUEUE,
928         .tag_alloc_policy       = BLK_TAG_ALLOC_RR,
929 };
930
931 static struct ata_port_operations pata_macio_ops = {
932         .inherits               = &ata_bmdma_port_ops,
933
934         .freeze                 = pata_macio_freeze,
935         .set_piomode            = pata_macio_set_timings,
936         .set_dmamode            = pata_macio_set_timings,
937         .cable_detect           = pata_macio_cable_detect,
938         .sff_dev_select         = pata_macio_dev_select,
939         .qc_prep                = pata_macio_qc_prep,
940         .bmdma_setup            = pata_macio_bmdma_setup,
941         .bmdma_start            = pata_macio_bmdma_start,
942         .bmdma_stop             = pata_macio_bmdma_stop,
943         .bmdma_status           = pata_macio_bmdma_status,
944         .port_start             = pata_macio_port_start,
945         .sff_irq_clear          = pata_macio_irq_clear,
946 };
947
948 static void pata_macio_invariants(struct pata_macio_priv *priv)
949 {
950         const int *bidp;
951
952         /* Identify the type of controller */
953         if (of_device_is_compatible(priv->node, "shasta-ata")) {
954                 priv->kind = controller_sh_ata6;
955                 priv->timings = pata_macio_shasta_timings;
956         } else if (of_device_is_compatible(priv->node, "kauai-ata")) {
957                 priv->kind = controller_un_ata6;
958                 priv->timings = pata_macio_kauai_timings;
959         } else if (of_device_is_compatible(priv->node, "K2-UATA")) {
960                 priv->kind = controller_k2_ata6;
961                 priv->timings = pata_macio_kauai_timings;
962         } else if (of_device_is_compatible(priv->node, "keylargo-ata")) {
963                 if (of_node_name_eq(priv->node, "ata-4")) {
964                         priv->kind = controller_kl_ata4;
965                         priv->timings = pata_macio_kl66_timings;
966                 } else {
967                         priv->kind = controller_kl_ata3;
968                         priv->timings = pata_macio_kl33_timings;
969                 }
970         } else if (of_device_is_compatible(priv->node, "heathrow-ata")) {
971                 priv->kind = controller_heathrow;
972                 priv->timings = pata_macio_heathrow_timings;
973         } else {
974                 priv->kind = controller_ohare;
975                 priv->timings = pata_macio_ohare_timings;
976         }
977
978         /* XXX FIXME --- setup priv->mediabay here */
979
980         /* Get Apple bus ID (for clock and ASIC control) */
981         bidp = of_get_property(priv->node, "AAPL,bus-id", NULL);
982         priv->aapl_bus_id =  bidp ? *bidp : 0;
983
984         /* Fixup missing Apple bus ID in case of media-bay */
985         if (priv->mediabay && !bidp)
986                 priv->aapl_bus_id = 1;
987 }
988
989 static void pata_macio_setup_ios(struct ata_ioports *ioaddr,
990                                  void __iomem * base, void __iomem * dma)
991 {
992         /* cmd_addr is the base of regs for that port */
993         ioaddr->cmd_addr        = base;
994
995         /* taskfile registers */
996         ioaddr->data_addr       = base + (ATA_REG_DATA    << 4);
997         ioaddr->error_addr      = base + (ATA_REG_ERR     << 4);
998         ioaddr->feature_addr    = base + (ATA_REG_FEATURE << 4);
999         ioaddr->nsect_addr      = base + (ATA_REG_NSECT   << 4);
1000         ioaddr->lbal_addr       = base + (ATA_REG_LBAL    << 4);
1001         ioaddr->lbam_addr       = base + (ATA_REG_LBAM    << 4);
1002         ioaddr->lbah_addr       = base + (ATA_REG_LBAH    << 4);
1003         ioaddr->device_addr     = base + (ATA_REG_DEVICE  << 4);
1004         ioaddr->status_addr     = base + (ATA_REG_STATUS  << 4);
1005         ioaddr->command_addr    = base + (ATA_REG_CMD     << 4);
1006         ioaddr->altstatus_addr  = base + 0x160;
1007         ioaddr->ctl_addr        = base + 0x160;
1008         ioaddr->bmdma_addr      = dma;
1009 }
1010
1011 static void pmac_macio_calc_timing_masks(struct pata_macio_priv *priv,
1012                                          struct ata_port_info *pinfo)
1013 {
1014         int i = 0;
1015
1016         pinfo->pio_mask         = 0;
1017         pinfo->mwdma_mask       = 0;
1018         pinfo->udma_mask        = 0;
1019
1020         while (priv->timings[i].mode > 0) {
1021                 unsigned int mask = 1U << (priv->timings[i].mode & 0x0f);
1022                 switch(priv->timings[i].mode & 0xf0) {
1023                 case 0x00: /* PIO */
1024                         pinfo->pio_mask |= (mask >> 8);
1025                         break;
1026                 case 0x20: /* MWDMA */
1027                         pinfo->mwdma_mask |= mask;
1028                         break;
1029                 case 0x40: /* UDMA */
1030                         pinfo->udma_mask |= mask;
1031                         break;
1032                 }
1033                 i++;
1034         }
1035         dev_dbg(priv->dev, "Supported masks: PIO=%lx, MWDMA=%lx, UDMA=%lx\n",
1036                 pinfo->pio_mask, pinfo->mwdma_mask, pinfo->udma_mask);
1037 }
1038
1039 static int pata_macio_common_init(struct pata_macio_priv *priv,
1040                                   resource_size_t tfregs,
1041                                   resource_size_t dmaregs,
1042                                   resource_size_t fcregs,
1043                                   unsigned long irq)
1044 {
1045         struct ata_port_info            pinfo;
1046         const struct ata_port_info      *ppi[] = { &pinfo, NULL };
1047         void __iomem                    *dma_regs = NULL;
1048
1049         /* Fill up privates with various invariants collected from the
1050          * device-tree
1051          */
1052         pata_macio_invariants(priv);
1053
1054         /* Make sure we have sane initial timings in the cache */
1055         pata_macio_default_timings(priv);
1056
1057         /* Allocate libata host for 1 port */
1058         memset(&pinfo, 0, sizeof(struct ata_port_info));
1059         pmac_macio_calc_timing_masks(priv, &pinfo);
1060         pinfo.flags             = ATA_FLAG_SLAVE_POSS;
1061         pinfo.port_ops          = &pata_macio_ops;
1062         pinfo.private_data      = priv;
1063
1064         priv->host = ata_host_alloc_pinfo(priv->dev, ppi, 1);
1065         if (priv->host == NULL) {
1066                 dev_err(priv->dev, "Failed to allocate ATA port structure\n");
1067                 return -ENOMEM;
1068         }
1069
1070         /* Setup the private data in host too */
1071         priv->host->private_data = priv;
1072
1073         /* Map base registers */
1074         priv->tfregs = devm_ioremap(priv->dev, tfregs, 0x100);
1075         if (priv->tfregs == NULL) {
1076                 dev_err(priv->dev, "Failed to map ATA ports\n");
1077                 return -ENOMEM;
1078         }
1079         priv->host->iomap = &priv->tfregs;
1080
1081         /* Map DMA regs */
1082         if (dmaregs != 0) {
1083                 dma_regs = devm_ioremap(priv->dev, dmaregs,
1084                                         sizeof(struct dbdma_regs));
1085                 if (dma_regs == NULL)
1086                         dev_warn(priv->dev, "Failed to map ATA DMA registers\n");
1087         }
1088
1089         /* If chip has local feature control, map those regs too */
1090         if (fcregs != 0) {
1091                 priv->kauai_fcr = devm_ioremap(priv->dev, fcregs, 4);
1092                 if (priv->kauai_fcr == NULL) {
1093                         dev_err(priv->dev, "Failed to map ATA FCR register\n");
1094                         return -ENOMEM;
1095                 }
1096         }
1097
1098         /* Setup port data structure */
1099         pata_macio_setup_ios(&priv->host->ports[0]->ioaddr,
1100                              priv->tfregs, dma_regs);
1101         priv->host->ports[0]->private_data = priv;
1102
1103         /* hard-reset the controller */
1104         pata_macio_reset_hw(priv, 0);
1105         pata_macio_apply_timings(priv->host->ports[0], 0);
1106
1107         /* Enable bus master if necessary */
1108         if (priv->pdev && dma_regs)
1109                 pci_set_master(priv->pdev);
1110
1111         dev_info(priv->dev, "Activating pata-macio chipset %s, Apple bus ID %d\n",
1112                  macio_ata_names[priv->kind], priv->aapl_bus_id);
1113
1114         /* Start it up */
1115         priv->irq = irq;
1116         return ata_host_activate(priv->host, irq, ata_bmdma_interrupt, 0,
1117                                  &pata_macio_sht);
1118 }
1119
1120 static int pata_macio_attach(struct macio_dev *mdev,
1121                              const struct of_device_id *match)
1122 {
1123         struct pata_macio_priv  *priv;
1124         resource_size_t         tfregs, dmaregs = 0;
1125         unsigned long           irq;
1126         int                     rc;
1127
1128         /* Check for broken device-trees */
1129         if (macio_resource_count(mdev) == 0) {
1130                 dev_err(&mdev->ofdev.dev,
1131                         "No addresses for controller\n");
1132                 return -ENXIO;
1133         }
1134
1135         /* Enable managed resources */
1136         macio_enable_devres(mdev);
1137
1138         /* Allocate and init private data structure */
1139         priv = devm_kzalloc(&mdev->ofdev.dev,
1140                             sizeof(struct pata_macio_priv), GFP_KERNEL);
1141         if (!priv)
1142                 return -ENOMEM;
1143
1144         priv->node = of_node_get(mdev->ofdev.dev.of_node);
1145         priv->mdev = mdev;
1146         priv->dev = &mdev->ofdev.dev;
1147
1148         /* Request memory resource for taskfile registers */
1149         if (macio_request_resource(mdev, 0, "pata-macio")) {
1150                 dev_err(&mdev->ofdev.dev,
1151                         "Cannot obtain taskfile resource\n");
1152                 return -EBUSY;
1153         }
1154         tfregs = macio_resource_start(mdev, 0);
1155
1156         /* Request resources for DMA registers if any */
1157         if (macio_resource_count(mdev) >= 2) {
1158                 if (macio_request_resource(mdev, 1, "pata-macio-dma"))
1159                         dev_err(&mdev->ofdev.dev,
1160                                 "Cannot obtain DMA resource\n");
1161                 else
1162                         dmaregs = macio_resource_start(mdev, 1);
1163         }
1164
1165         /*
1166          * Fixup missing IRQ for some old implementations with broken
1167          * device-trees.
1168          *
1169          * This is a bit bogus, it should be fixed in the device-tree itself,
1170          * via the existing macio fixups, based on the type of interrupt
1171          * controller in the machine. However, I have no test HW for this case,
1172          * and this trick works well enough on those old machines...
1173          */
1174         if (macio_irq_count(mdev) == 0) {
1175                 dev_warn(&mdev->ofdev.dev,
1176                          "No interrupts for controller, using 13\n");
1177                 irq = irq_create_mapping(NULL, 13);
1178         } else
1179                 irq = macio_irq(mdev, 0);
1180
1181         /* Prevvent media bay callbacks until fully registered */
1182         lock_media_bay(priv->mdev->media_bay);
1183
1184         /* Get register addresses and call common initialization */
1185         rc = pata_macio_common_init(priv,
1186                                     tfregs,             /* Taskfile regs */
1187                                     dmaregs,            /* DBDMA regs */
1188                                     0,                  /* Feature control */
1189                                     irq);
1190         unlock_media_bay(priv->mdev->media_bay);
1191
1192         return rc;
1193 }
1194
1195 static int pata_macio_detach(struct macio_dev *mdev)
1196 {
1197         struct ata_host *host = macio_get_drvdata(mdev);
1198         struct pata_macio_priv *priv = host->private_data;
1199
1200         lock_media_bay(priv->mdev->media_bay);
1201
1202         /* Make sure the mediabay callback doesn't try to access
1203          * dead stuff
1204          */
1205         priv->host->private_data = NULL;
1206
1207         ata_host_detach(host);
1208
1209         unlock_media_bay(priv->mdev->media_bay);
1210
1211         return 0;
1212 }
1213
1214 #ifdef CONFIG_PM_SLEEP
1215 static int pata_macio_suspend(struct macio_dev *mdev, pm_message_t mesg)
1216 {
1217         struct ata_host *host = macio_get_drvdata(mdev);
1218
1219         return pata_macio_do_suspend(host->private_data, mesg);
1220 }
1221
1222 static int pata_macio_resume(struct macio_dev *mdev)
1223 {
1224         struct ata_host *host = macio_get_drvdata(mdev);
1225
1226         return pata_macio_do_resume(host->private_data);
1227 }
1228 #endif /* CONFIG_PM_SLEEP */
1229
1230 #ifdef CONFIG_PMAC_MEDIABAY
1231 static void pata_macio_mb_event(struct macio_dev* mdev, int mb_state)
1232 {
1233         struct ata_host *host = macio_get_drvdata(mdev);
1234         struct ata_port *ap;
1235         struct ata_eh_info *ehi;
1236         struct ata_device *dev;
1237         unsigned long flags;
1238
1239         if (!host || !host->private_data)
1240                 return;
1241         ap = host->ports[0];
1242         spin_lock_irqsave(ap->lock, flags);
1243         ehi = &ap->link.eh_info;
1244         if (mb_state == MB_CD) {
1245                 ata_ehi_push_desc(ehi, "mediabay plug");
1246                 ata_ehi_hotplugged(ehi);
1247                 ata_port_freeze(ap);
1248         } else {
1249                 ata_ehi_push_desc(ehi, "mediabay unplug");
1250                 ata_for_each_dev(dev, &ap->link, ALL)
1251                         dev->flags |= ATA_DFLAG_DETACH;
1252                 ata_port_abort(ap);
1253         }
1254         spin_unlock_irqrestore(ap->lock, flags);
1255
1256 }
1257 #endif /* CONFIG_PMAC_MEDIABAY */
1258
1259
1260 static int pata_macio_pci_attach(struct pci_dev *pdev,
1261                                  const struct pci_device_id *id)
1262 {
1263         struct pata_macio_priv  *priv;
1264         struct device_node      *np;
1265         resource_size_t         rbase;
1266
1267         /* We cannot use a MacIO controller without its OF device node */
1268         np = pci_device_to_OF_node(pdev);
1269         if (np == NULL) {
1270                 dev_err(&pdev->dev,
1271                         "Cannot find OF device node for controller\n");
1272                 return -ENODEV;
1273         }
1274
1275         /* Check that it can be enabled */
1276         if (pcim_enable_device(pdev)) {
1277                 dev_err(&pdev->dev,
1278                         "Cannot enable controller PCI device\n");
1279                 return -ENXIO;
1280         }
1281
1282         /* Allocate and init private data structure */
1283         priv = devm_kzalloc(&pdev->dev,
1284                             sizeof(struct pata_macio_priv), GFP_KERNEL);
1285         if (!priv)
1286                 return -ENOMEM;
1287
1288         priv->node = of_node_get(np);
1289         priv->pdev = pdev;
1290         priv->dev = &pdev->dev;
1291
1292         /* Get MMIO regions */
1293         if (pci_request_regions(pdev, "pata-macio")) {
1294                 dev_err(&pdev->dev,
1295                         "Cannot obtain PCI resources\n");
1296                 return -EBUSY;
1297         }
1298
1299         /* Get register addresses and call common initialization */
1300         rbase = pci_resource_start(pdev, 0);
1301         if (pata_macio_common_init(priv,
1302                                    rbase + 0x2000,      /* Taskfile regs */
1303                                    rbase + 0x1000,      /* DBDMA regs */
1304                                    rbase,               /* Feature control */
1305                                    pdev->irq))
1306                 return -ENXIO;
1307
1308         return 0;
1309 }
1310
1311 static void pata_macio_pci_detach(struct pci_dev *pdev)
1312 {
1313         struct ata_host *host = pci_get_drvdata(pdev);
1314
1315         ata_host_detach(host);
1316 }
1317
1318 #ifdef CONFIG_PM_SLEEP
1319 static int pata_macio_pci_suspend(struct pci_dev *pdev, pm_message_t mesg)
1320 {
1321         struct ata_host *host = pci_get_drvdata(pdev);
1322
1323         return pata_macio_do_suspend(host->private_data, mesg);
1324 }
1325
1326 static int pata_macio_pci_resume(struct pci_dev *pdev)
1327 {
1328         struct ata_host *host = pci_get_drvdata(pdev);
1329
1330         return pata_macio_do_resume(host->private_data);
1331 }
1332 #endif /* CONFIG_PM_SLEEP */
1333
1334 static const struct of_device_id pata_macio_match[] =
1335 {
1336         {
1337         .name           = "IDE",
1338         },
1339         {
1340         .name           = "ATA",
1341         },
1342         {
1343         .type           = "ide",
1344         },
1345         {
1346         .type           = "ata",
1347         },
1348         {},
1349 };
1350 MODULE_DEVICE_TABLE(of, pata_macio_match);
1351
1352 static struct macio_driver pata_macio_driver =
1353 {
1354         .driver = {
1355                 .name           = "pata-macio",
1356                 .owner          = THIS_MODULE,
1357                 .of_match_table = pata_macio_match,
1358         },
1359         .probe          = pata_macio_attach,
1360         .remove         = pata_macio_detach,
1361 #ifdef CONFIG_PM_SLEEP
1362         .suspend        = pata_macio_suspend,
1363         .resume         = pata_macio_resume,
1364 #endif
1365 #ifdef CONFIG_PMAC_MEDIABAY
1366         .mediabay_event = pata_macio_mb_event,
1367 #endif
1368 };
1369
1370 static const struct pci_device_id pata_macio_pci_match[] = {
1371         { PCI_VDEVICE(APPLE, PCI_DEVICE_ID_APPLE_UNI_N_ATA),    0 },
1372         { PCI_VDEVICE(APPLE, PCI_DEVICE_ID_APPLE_IPID_ATA100),  0 },
1373         { PCI_VDEVICE(APPLE, PCI_DEVICE_ID_APPLE_K2_ATA100),    0 },
1374         { PCI_VDEVICE(APPLE, PCI_DEVICE_ID_APPLE_SH_ATA),       0 },
1375         { PCI_VDEVICE(APPLE, PCI_DEVICE_ID_APPLE_IPID2_ATA),    0 },
1376         {},
1377 };
1378
1379 static struct pci_driver pata_macio_pci_driver = {
1380         .name           = "pata-pci-macio",
1381         .id_table       = pata_macio_pci_match,
1382         .probe          = pata_macio_pci_attach,
1383         .remove         = pata_macio_pci_detach,
1384 #ifdef CONFIG_PM_SLEEP
1385         .suspend        = pata_macio_pci_suspend,
1386         .resume         = pata_macio_pci_resume,
1387 #endif
1388         .driver = {
1389                 .owner          = THIS_MODULE,
1390         },
1391 };
1392 MODULE_DEVICE_TABLE(pci, pata_macio_pci_match);
1393
1394
1395 static int __init pata_macio_init(void)
1396 {
1397         int rc;
1398
1399         if (!machine_is(powermac))
1400                 return -ENODEV;
1401
1402         rc = pci_register_driver(&pata_macio_pci_driver);
1403         if (rc)
1404                 return rc;
1405         rc = macio_register_driver(&pata_macio_driver);
1406         if (rc) {
1407                 pci_unregister_driver(&pata_macio_pci_driver);
1408                 return rc;
1409         }
1410         return 0;
1411 }
1412
1413 static void __exit pata_macio_exit(void)
1414 {
1415         macio_unregister_driver(&pata_macio_driver);
1416         pci_unregister_driver(&pata_macio_pci_driver);
1417 }
1418
1419 module_init(pata_macio_init);
1420 module_exit(pata_macio_exit);
1421
1422 MODULE_AUTHOR("Benjamin Herrenschmidt");
1423 MODULE_DESCRIPTION("Apple MacIO PATA driver");
1424 MODULE_LICENSE("GPL");
1425 MODULE_VERSION(DRV_VERSION);