Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net-2.6
[pandora-kernel.git] / drivers / macintosh / mediabay.c
1 /*
2  * Driver for the media bay on the PowerBook 3400 and 2400.
3  *
4  * Copyright (C) 1998 Paul Mackerras.
5  *
6  * Various evolutions by Benjamin Herrenschmidt & Henry Worth
7  *
8  *  This program is free software; you can redistribute it and/or
9  *  modify it under the terms of the GNU General Public License
10  *  as published by the Free Software Foundation; either version
11  *  2 of the License, or (at your option) any later version.
12  */
13 #include <linux/types.h>
14 #include <linux/errno.h>
15 #include <linux/kernel.h>
16 #include <linux/delay.h>
17 #include <linux/sched.h>
18 #include <linux/timer.h>
19 #include <linux/hdreg.h>
20 #include <linux/stddef.h>
21 #include <linux/init.h>
22 #include <linux/ide.h>
23 #include <linux/kthread.h>
24 #include <asm/prom.h>
25 #include <asm/pgtable.h>
26 #include <asm/io.h>
27 #include <asm/machdep.h>
28 #include <asm/pmac_feature.h>
29 #include <asm/mediabay.h>
30 #include <asm/sections.h>
31 #include <asm/ohare.h>
32 #include <asm/heathrow.h>
33 #include <asm/keylargo.h>
34 #include <linux/adb.h>
35 #include <linux/pmu.h>
36
37
38 #define MB_DEBUG
39
40 #ifdef MB_DEBUG
41 #define MBDBG(fmt, arg...)      printk(KERN_INFO fmt , ## arg)
42 #else
43 #define MBDBG(fmt, arg...)      do { } while (0)
44 #endif
45
46 #define MB_FCR32(bay, r)        ((bay)->base + ((r) >> 2))
47 #define MB_FCR8(bay, r)         (((volatile u8 __iomem *)((bay)->base)) + (r))
48
49 #define MB_IN32(bay,r)          (in_le32(MB_FCR32(bay,r)))
50 #define MB_OUT32(bay,r,v)       (out_le32(MB_FCR32(bay,r), (v)))
51 #define MB_BIS(bay,r,v)         (MB_OUT32((bay), (r), MB_IN32((bay), r) | (v)))
52 #define MB_BIC(bay,r,v)         (MB_OUT32((bay), (r), MB_IN32((bay), r) & ~(v)))
53 #define MB_IN8(bay,r)           (in_8(MB_FCR8(bay,r)))
54 #define MB_OUT8(bay,r,v)        (out_8(MB_FCR8(bay,r), (v)))
55
56 struct media_bay_info;
57
58 struct mb_ops {
59         char*   name;
60         void    (*init)(struct media_bay_info *bay);
61         u8      (*content)(struct media_bay_info *bay);
62         void    (*power)(struct media_bay_info *bay, int on_off);
63         int     (*setup_bus)(struct media_bay_info *bay, u8 device_id);
64         void    (*un_reset)(struct media_bay_info *bay);
65         void    (*un_reset_ide)(struct media_bay_info *bay);
66 };
67
68 struct media_bay_info {
69         u32 __iomem                     *base;
70         int                             content_id;
71         int                             state;
72         int                             last_value;
73         int                             value_count;
74         int                             timer;
75         struct macio_dev                *mdev;
76         struct mb_ops*                  ops;
77         int                             index;
78         int                             cached_gpio;
79         int                             sleeping;
80         struct semaphore                lock;
81 #ifdef CONFIG_BLK_DEV_IDE_PMAC
82         ide_hwif_t                      *cd_port;
83         void __iomem                    *cd_base;
84         int                             cd_irq;
85         int                             cd_retry;
86 #endif
87 #if defined(CONFIG_BLK_DEV_IDE_PMAC)
88         int                             cd_index;
89 #endif
90 };
91
92 #define MAX_BAYS        2
93
94 static struct media_bay_info media_bays[MAX_BAYS];
95 int media_bay_count = 0;
96
97 #ifdef CONFIG_BLK_DEV_IDE_PMAC
98 /* check the busy bit in the media-bay ide interface
99    (assumes the media-bay contains an ide device) */
100 #define MB_IDE_READY(i) ((readb(media_bays[i].cd_base + 0x70) & 0x80) == 0)
101 #endif
102
103 /*
104  * Wait that number of ms between each step in normal polling mode
105  */
106 #define MB_POLL_DELAY   25
107
108 /*
109  * Consider the media-bay ID value stable if it is the same for
110  * this number of milliseconds
111  */
112 #define MB_STABLE_DELAY 100
113
114 /* Wait after powering up the media bay this delay in ms
115  * timeout bumped for some powerbooks
116  */
117 #define MB_POWER_DELAY  200
118
119 /*
120  * Hold the media-bay reset signal true for this many ticks
121  * after a device is inserted before releasing it.
122  */
123 #define MB_RESET_DELAY  50
124
125 /*
126  * Wait this long after the reset signal is released and before doing
127  * further operations. After this delay, the IDE reset signal is released
128  * too for an IDE device
129  */
130 #define MB_SETUP_DELAY  100
131
132 /*
133  * Wait this many ticks after an IDE device (e.g. CD-ROM) is inserted
134  * (or until the device is ready) before waiting for busy bit to disappear
135  */
136 #define MB_IDE_WAIT     1000
137
138 /*
139  * Timeout waiting for busy bit of an IDE device to go down
140  */
141 #define MB_IDE_TIMEOUT  5000
142
143 /*
144  * Max retries of the full power up/down sequence for an IDE device
145  */
146 #define MAX_CD_RETRIES  3
147
148 /*
149  * States of a media bay
150  */
151 enum {
152         mb_empty = 0,           /* Idle */
153         mb_powering_up,         /* power bit set, waiting MB_POWER_DELAY */
154         mb_enabling_bay,        /* enable bits set, waiting MB_RESET_DELAY */
155         mb_resetting,           /* reset bit unset, waiting MB_SETUP_DELAY */
156         mb_ide_resetting,       /* IDE reset bit unser, waiting MB_IDE_WAIT */
157         mb_ide_waiting,         /* Waiting for BUSY bit to go away until MB_IDE_TIMEOUT */
158         mb_up,                  /* Media bay full */
159         mb_powering_down        /* Powering down (avoid too fast down/up) */
160 };
161
162 #define MB_POWER_SOUND          0x08
163 #define MB_POWER_FLOPPY         0x04
164 #define MB_POWER_ATA            0x02
165 #define MB_POWER_PCI            0x01
166 #define MB_POWER_OFF            0x00
167
168 /*
169  * Functions for polling content of media bay
170  */
171  
172 static u8
173 ohare_mb_content(struct media_bay_info *bay)
174 {
175         return (MB_IN32(bay, OHARE_MBCR) >> 12) & 7;
176 }
177
178 static u8
179 heathrow_mb_content(struct media_bay_info *bay)
180 {
181         return (MB_IN32(bay, HEATHROW_MBCR) >> 12) & 7;
182 }
183
184 static u8
185 keylargo_mb_content(struct media_bay_info *bay)
186 {
187         int new_gpio;
188
189         new_gpio = MB_IN8(bay, KL_GPIO_MEDIABAY_IRQ) & KEYLARGO_GPIO_INPUT_DATA;
190         if (new_gpio) {
191                 bay->cached_gpio = new_gpio;
192                 return MB_NO;
193         } else if (bay->cached_gpio != new_gpio) {
194                 MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_ENABLE);
195                 (void)MB_IN32(bay, KEYLARGO_MBCR);
196                 udelay(5);
197                 MB_BIC(bay, KEYLARGO_MBCR, 0x0000000F);
198                 (void)MB_IN32(bay, KEYLARGO_MBCR);
199                 udelay(5);
200                 bay->cached_gpio = new_gpio;
201         }
202         return (MB_IN32(bay, KEYLARGO_MBCR) >> 4) & 7;
203 }
204
205 /*
206  * Functions for powering up/down the bay, puts the bay device
207  * into reset state as well
208  */
209
210 static void
211 ohare_mb_power(struct media_bay_info* bay, int on_off)
212 {
213         if (on_off) {
214                 /* Power up device, assert it's reset line */
215                 MB_BIC(bay, OHARE_FCR, OH_BAY_RESET_N);
216                 MB_BIC(bay, OHARE_FCR, OH_BAY_POWER_N);
217         } else {
218                 /* Disable all devices */
219                 MB_BIC(bay, OHARE_FCR, OH_BAY_DEV_MASK);
220                 MB_BIC(bay, OHARE_FCR, OH_FLOPPY_ENABLE);
221                 /* Cut power from bay, release reset line */
222                 MB_BIS(bay, OHARE_FCR, OH_BAY_POWER_N);
223                 MB_BIS(bay, OHARE_FCR, OH_BAY_RESET_N);
224                 MB_BIS(bay, OHARE_FCR, OH_IDE1_RESET_N);
225         }
226         MB_BIC(bay, OHARE_MBCR, 0x00000F00);
227 }
228
229 static void
230 heathrow_mb_power(struct media_bay_info* bay, int on_off)
231 {
232         if (on_off) {
233                 /* Power up device, assert it's reset line */
234                 MB_BIC(bay, HEATHROW_FCR, HRW_BAY_RESET_N);
235                 MB_BIC(bay, HEATHROW_FCR, HRW_BAY_POWER_N);
236         } else {
237                 /* Disable all devices */
238                 MB_BIC(bay, HEATHROW_FCR, HRW_BAY_DEV_MASK);
239                 MB_BIC(bay, HEATHROW_FCR, HRW_SWIM_ENABLE);
240                 /* Cut power from bay, release reset line */
241                 MB_BIS(bay, HEATHROW_FCR, HRW_BAY_POWER_N);
242                 MB_BIS(bay, HEATHROW_FCR, HRW_BAY_RESET_N);
243                 MB_BIS(bay, HEATHROW_FCR, HRW_IDE1_RESET_N);
244         }
245         MB_BIC(bay, HEATHROW_MBCR, 0x00000F00);
246 }
247
248 static void
249 keylargo_mb_power(struct media_bay_info* bay, int on_off)
250 {
251         if (on_off) {
252                 /* Power up device, assert it's reset line */
253                 MB_BIC(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_RESET);
254                 MB_BIC(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_POWER);
255         } else {
256                 /* Disable all devices */
257                 MB_BIC(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_MASK);
258                 MB_BIC(bay, KEYLARGO_FCR1, KL1_EIDE0_ENABLE);
259                 /* Cut power from bay, release reset line */
260                 MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_POWER);
261                 MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_RESET);
262                 MB_BIS(bay, KEYLARGO_FCR1, KL1_EIDE0_RESET_N);
263         }
264         MB_BIC(bay, KEYLARGO_MBCR, 0x0000000F);
265 }
266
267 /*
268  * Functions for configuring the media bay for a given type of device,
269  * enable the related busses
270  */
271
272 static int
273 ohare_mb_setup_bus(struct media_bay_info* bay, u8 device_id)
274 {
275         switch(device_id) {
276                 case MB_FD:
277                 case MB_FD1:
278                         MB_BIS(bay, OHARE_FCR, OH_BAY_FLOPPY_ENABLE);
279                         MB_BIS(bay, OHARE_FCR, OH_FLOPPY_ENABLE);
280                         return 0;
281                 case MB_CD:
282                         MB_BIC(bay, OHARE_FCR, OH_IDE1_RESET_N);
283                         MB_BIS(bay, OHARE_FCR, OH_BAY_IDE_ENABLE);
284                         return 0;
285                 case MB_PCI:
286                         MB_BIS(bay, OHARE_FCR, OH_BAY_PCI_ENABLE);
287                         return 0;
288         }
289         return -ENODEV;
290 }
291
292 static int
293 heathrow_mb_setup_bus(struct media_bay_info* bay, u8 device_id)
294 {
295         switch(device_id) {
296                 case MB_FD:
297                 case MB_FD1:
298                         MB_BIS(bay, HEATHROW_FCR, HRW_BAY_FLOPPY_ENABLE);
299                         MB_BIS(bay, HEATHROW_FCR, HRW_SWIM_ENABLE);
300                         return 0;
301                 case MB_CD:
302                         MB_BIC(bay, HEATHROW_FCR, HRW_IDE1_RESET_N);
303                         MB_BIS(bay, HEATHROW_FCR, HRW_BAY_IDE_ENABLE);
304                         return 0;
305                 case MB_PCI:
306                         MB_BIS(bay, HEATHROW_FCR, HRW_BAY_PCI_ENABLE);
307                         return 0;
308         }
309         return -ENODEV;
310 }
311
312 static int
313 keylargo_mb_setup_bus(struct media_bay_info* bay, u8 device_id)
314 {
315         switch(device_id) {
316                 case MB_CD:
317                         MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_IDE_ENABLE);
318                         MB_BIC(bay, KEYLARGO_FCR1, KL1_EIDE0_RESET_N);
319                         MB_BIS(bay, KEYLARGO_FCR1, KL1_EIDE0_ENABLE);
320                         return 0;
321                 case MB_PCI:
322                         MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_PCI_ENABLE);
323                         return 0;
324                 case MB_SOUND:
325                         MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_SOUND_ENABLE);
326                         return 0;
327         }
328         return -ENODEV;
329 }
330
331 /*
332  * Functions for tweaking resets
333  */
334
335 static void
336 ohare_mb_un_reset(struct media_bay_info* bay)
337 {
338         MB_BIS(bay, OHARE_FCR, OH_BAY_RESET_N);
339 }
340
341 static void keylargo_mb_init(struct media_bay_info *bay)
342 {
343         MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_ENABLE);
344 }
345
346 static void heathrow_mb_un_reset(struct media_bay_info* bay)
347 {
348         MB_BIS(bay, HEATHROW_FCR, HRW_BAY_RESET_N);
349 }
350
351 static void keylargo_mb_un_reset(struct media_bay_info* bay)
352 {
353         MB_BIS(bay, KEYLARGO_MBCR, KL_MBCR_MB0_DEV_RESET);
354 }
355
356 static void ohare_mb_un_reset_ide(struct media_bay_info* bay)
357 {
358         MB_BIS(bay, OHARE_FCR, OH_IDE1_RESET_N);
359 }
360
361 static void heathrow_mb_un_reset_ide(struct media_bay_info* bay)
362 {
363         MB_BIS(bay, HEATHROW_FCR, HRW_IDE1_RESET_N);
364 }
365
366 static void keylargo_mb_un_reset_ide(struct media_bay_info* bay)
367 {
368         MB_BIS(bay, KEYLARGO_FCR1, KL1_EIDE0_RESET_N);
369 }
370
371 static inline void set_mb_power(struct media_bay_info* bay, int onoff)
372 {
373         /* Power up up and assert the bay reset line */
374         if (onoff) {
375                 bay->ops->power(bay, 1);
376                 bay->state = mb_powering_up;
377                 MBDBG("mediabay%d: powering up\n", bay->index);
378         } else { 
379                 /* Make sure everything is powered down & disabled */
380                 bay->ops->power(bay, 0);
381                 bay->state = mb_powering_down;
382                 MBDBG("mediabay%d: powering down\n", bay->index);
383         }
384         bay->timer = msecs_to_jiffies(MB_POWER_DELAY);
385 }
386
387 static void poll_media_bay(struct media_bay_info* bay)
388 {
389         int id = bay->ops->content(bay);
390
391         if (id == bay->last_value) {
392                 if (id != bay->content_id) {
393                         bay->value_count += msecs_to_jiffies(MB_POLL_DELAY);
394                         if (bay->value_count >= msecs_to_jiffies(MB_STABLE_DELAY)) {
395                                 /* If the device type changes without going thru
396                                  * "MB_NO", we force a pass by "MB_NO" to make sure
397                                  * things are properly reset
398                                  */
399                                 if ((id != MB_NO) && (bay->content_id != MB_NO)) {
400                                         id = MB_NO;
401                                         MBDBG("mediabay%d: forcing MB_NO\n", bay->index);
402                                 }
403                                 MBDBG("mediabay%d: switching to %d\n", bay->index, id);
404                                 set_mb_power(bay, id != MB_NO);
405                                 bay->content_id = id;
406                                 if (id == MB_NO) {
407 #ifdef CONFIG_BLK_DEV_IDE_PMAC
408                                         bay->cd_retry = 0;
409 #endif
410                                         printk(KERN_INFO "media bay %d is empty\n", bay->index);
411                                 }
412                         }
413                 }
414         } else {
415                 bay->last_value = id;
416                 bay->value_count = 0;
417         }
418 }
419
420 #ifdef CONFIG_BLK_DEV_IDE_PMAC
421 int check_media_bay(struct device_node *which_bay, int what)
422 {
423         int     i;
424
425         for (i=0; i<media_bay_count; i++)
426                 if (media_bays[i].mdev && which_bay == media_bays[i].mdev->ofdev.node) {
427                         if ((what == media_bays[i].content_id) && media_bays[i].state == mb_up)
428                                 return 0;
429                         media_bays[i].cd_index = -1;
430                         return -EINVAL;
431                 }
432         return -ENODEV;
433 }
434 EXPORT_SYMBOL(check_media_bay);
435
436 int check_media_bay_by_base(unsigned long base, int what)
437 {
438         int     i;
439
440         for (i=0; i<media_bay_count; i++)
441                 if (media_bays[i].mdev && base == (unsigned long) media_bays[i].cd_base) {
442                         if ((what == media_bays[i].content_id) && media_bays[i].state == mb_up)
443                                 return 0;
444                         media_bays[i].cd_index = -1;
445                         return -EINVAL;
446                 } 
447
448         return -ENODEV;
449 }
450
451 int media_bay_set_ide_infos(struct device_node* which_bay, unsigned long base,
452                             int irq, ide_hwif_t *hwif)
453 {
454         int     i;
455
456         for (i=0; i<media_bay_count; i++) {
457                 struct media_bay_info* bay = &media_bays[i];
458
459                 if (bay->mdev && which_bay == bay->mdev->ofdev.node) {
460                         int timeout = 5000, index = hwif->index;
461                         
462                         down(&bay->lock);
463
464                         bay->cd_port    = hwif;
465                         bay->cd_base    = (void __iomem *) base;
466                         bay->cd_irq     = irq;
467
468                         if ((MB_CD != bay->content_id) || bay->state != mb_up) {
469                                 up(&bay->lock);
470                                 return 0;
471                         }
472                         printk(KERN_DEBUG "Registered ide%d for media bay %d\n", index, i);
473                         do {
474                                 if (MB_IDE_READY(i)) {
475                                         bay->cd_index   = index;
476                                         up(&bay->lock);
477                                         return 0;
478                                 }
479                                 mdelay(1);
480                         } while(--timeout);
481                         printk(KERN_DEBUG "Timeount waiting IDE in bay %d\n", i);
482                         up(&bay->lock);
483                         return -ENODEV;
484                 }
485         }
486
487         return -ENODEV;
488 }
489 #endif /* CONFIG_BLK_DEV_IDE_PMAC */
490
491 static void media_bay_step(int i)
492 {
493         struct media_bay_info* bay = &media_bays[i];
494
495         /* We don't poll when powering down */
496         if (bay->state != mb_powering_down)
497             poll_media_bay(bay);
498
499         /* If timer expired or polling IDE busy, run state machine */
500         if ((bay->state != mb_ide_waiting) && (bay->timer != 0)) {
501                 bay->timer -= msecs_to_jiffies(MB_POLL_DELAY);
502                 if (bay->timer > 0)
503                         return;
504                 bay->timer = 0;
505         }
506
507         switch(bay->state) {
508         case mb_powering_up:
509                 if (bay->ops->setup_bus(bay, bay->last_value) < 0) {
510                         MBDBG("mediabay%d: device not supported (kind:%d)\n", i, bay->content_id);
511                         set_mb_power(bay, 0);
512                         break;
513                 }
514                 bay->timer = msecs_to_jiffies(MB_RESET_DELAY);
515                 bay->state = mb_enabling_bay;
516                 MBDBG("mediabay%d: enabling (kind:%d)\n", i, bay->content_id);
517                 break;
518         case mb_enabling_bay:
519                 bay->ops->un_reset(bay);
520                 bay->timer = msecs_to_jiffies(MB_SETUP_DELAY);
521                 bay->state = mb_resetting;
522                 MBDBG("mediabay%d: waiting reset (kind:%d)\n", i, bay->content_id);
523                 break;
524         case mb_resetting:
525                 if (bay->content_id != MB_CD) {
526                         MBDBG("mediabay%d: bay is up (kind:%d)\n", i, bay->content_id);
527                         bay->state = mb_up;
528                         break;
529                 }
530 #ifdef CONFIG_BLK_DEV_IDE_PMAC
531                 MBDBG("mediabay%d: waiting IDE reset (kind:%d)\n", i, bay->content_id);
532                 bay->ops->un_reset_ide(bay);
533                 bay->timer = msecs_to_jiffies(MB_IDE_WAIT);
534                 bay->state = mb_ide_resetting;
535 #else
536                 printk(KERN_DEBUG "media-bay %d is ide (not compiled in kernel)\n", i);
537                 set_mb_power(bay, 0);
538 #endif /* CONFIG_BLK_DEV_IDE_PMAC */
539                 break;
540 #ifdef CONFIG_BLK_DEV_IDE_PMAC
541         case mb_ide_resetting:
542                 bay->timer = msecs_to_jiffies(MB_IDE_TIMEOUT);
543                 bay->state = mb_ide_waiting;
544                 MBDBG("mediabay%d: waiting IDE ready (kind:%d)\n", i, bay->content_id);
545                 break;
546         case mb_ide_waiting:
547                 if (bay->cd_base == NULL) {
548                         bay->timer = 0;
549                         bay->state = mb_up;
550                         MBDBG("mediabay%d: up before IDE init\n", i);
551                         break;
552                 } else if (MB_IDE_READY(i)) {
553                         bay->timer = 0;
554                         bay->state = mb_up;
555                         if (bay->cd_index < 0) {
556                                 printk("mediabay %d, registering IDE...\n", i);
557                                 pmu_suspend();
558                                 ide_port_scan(bay->cd_port);
559                                 if (bay->cd_port->present)
560                                         bay->cd_index = bay->cd_port->index;
561                                 pmu_resume();
562                         }
563                         if (bay->cd_index == -1) {
564                                 /* We eventually do a retry */
565                                 bay->cd_retry++;
566                                 printk("IDE register error\n");
567                                 set_mb_power(bay, 0);
568                         } else {
569                                 printk(KERN_DEBUG "media-bay %d is ide%d\n", i, bay->cd_index);
570                                 MBDBG("mediabay %d IDE ready\n", i);
571                         }
572                         break;
573                 } else if (bay->timer > 0)
574                         bay->timer -= msecs_to_jiffies(MB_POLL_DELAY);
575                 if (bay->timer <= 0) {
576                         printk("\nIDE Timeout in bay %d !, IDE state is: 0x%02x\n",
577                                i, readb(bay->cd_base + 0x70));
578                         MBDBG("mediabay%d: nIDE Timeout !\n", i);
579                         set_mb_power(bay, 0);
580                         bay->timer = 0;
581                 }
582                 break;
583 #endif /* CONFIG_BLK_DEV_IDE_PMAC */
584         case mb_powering_down:
585                 bay->state = mb_empty;
586 #ifdef CONFIG_BLK_DEV_IDE_PMAC
587                 if (bay->cd_index >= 0) {
588                         printk(KERN_DEBUG "Unregistering mb %d ide, index:%d\n", i,
589                                bay->cd_index);
590                         ide_port_unregister_devices(bay->cd_port);
591                         bay->cd_index = -1;
592                 }
593                 if (bay->cd_retry) {
594                         if (bay->cd_retry > MAX_CD_RETRIES) {
595                                 /* Should add an error sound (sort of beep in dmasound) */
596                                 printk("\nmedia-bay %d, IDE device badly inserted or unrecognised\n", i);
597                         } else {
598                                 /* Force a new power down/up sequence */
599                                 bay->content_id = MB_NO;
600                         }
601                 }
602 #endif /* CONFIG_BLK_DEV_IDE_PMAC */
603                 MBDBG("mediabay%d: end of power down\n", i);
604                 break;
605         }
606 }
607
608 /*
609  * This procedure runs as a kernel thread to poll the media bay
610  * once each tick and register and unregister the IDE interface
611  * with the IDE driver.  It needs to be a thread because
612  * ide_register can't be called from interrupt context.
613  */
614 static int media_bay_task(void *x)
615 {
616         int     i;
617
618         while (!kthread_should_stop()) {
619                 for (i = 0; i < media_bay_count; ++i) {
620                         down(&media_bays[i].lock);
621                         if (!media_bays[i].sleeping)
622                                 media_bay_step(i);
623                         up(&media_bays[i].lock);
624                 }
625
626                 msleep_interruptible(MB_POLL_DELAY);
627         }
628         return 0;
629 }
630
631 static int __devinit media_bay_attach(struct macio_dev *mdev, const struct of_device_id *match)
632 {
633         struct media_bay_info* bay;
634         u32 __iomem *regbase;
635         struct device_node *ofnode;
636         unsigned long base;
637         int i;
638
639         ofnode = mdev->ofdev.node;
640
641         if (macio_resource_count(mdev) < 1)
642                 return -ENODEV;
643         if (macio_request_resources(mdev, "media-bay"))
644                 return -EBUSY;
645         /* Media bay registers are located at the beginning of the
646          * mac-io chip, for now, we trick and align down the first
647          * resource passed in
648          */
649         base = macio_resource_start(mdev, 0) & 0xffff0000u;
650         regbase = (u32 __iomem *)ioremap(base, 0x100);
651         if (regbase == NULL) {
652                 macio_release_resources(mdev);
653                 return -ENOMEM;
654         }
655         
656         i = media_bay_count++;
657         bay = &media_bays[i];
658         bay->mdev = mdev;
659         bay->base = regbase;
660         bay->index = i;
661         bay->ops = match->data;
662         bay->sleeping = 0;
663         init_MUTEX(&bay->lock);
664
665         /* Init HW probing */
666         if (bay->ops->init)
667                 bay->ops->init(bay);
668
669         printk(KERN_INFO "mediabay%d: Registered %s media-bay\n", i, bay->ops->name);
670
671         /* Force an immediate detect */
672         set_mb_power(bay, 0);
673         msleep(MB_POWER_DELAY);
674         bay->content_id = MB_NO;
675         bay->last_value = bay->ops->content(bay);
676         bay->value_count = msecs_to_jiffies(MB_STABLE_DELAY);
677         bay->state = mb_empty;
678         do {
679                 msleep(MB_POLL_DELAY);
680                 media_bay_step(i);
681         } while((bay->state != mb_empty) &&
682                 (bay->state != mb_up));
683
684         /* Mark us ready by filling our mdev data */
685         macio_set_drvdata(mdev, bay);
686
687         /* Startup kernel thread */
688         if (i == 0)
689                 kthread_run(media_bay_task, NULL, "media-bay");
690
691         return 0;
692
693 }
694
695 static int media_bay_suspend(struct macio_dev *mdev, pm_message_t state)
696 {
697         struct media_bay_info   *bay = macio_get_drvdata(mdev);
698
699         if (state.event != mdev->ofdev.dev.power.power_state.event
700             && (state.event & PM_EVENT_SLEEP)) {
701                 down(&bay->lock);
702                 bay->sleeping = 1;
703                 set_mb_power(bay, 0);
704                 up(&bay->lock);
705                 msleep(MB_POLL_DELAY);
706                 mdev->ofdev.dev.power.power_state = state;
707         }
708         return 0;
709 }
710
711 static int media_bay_resume(struct macio_dev *mdev)
712 {
713         struct media_bay_info   *bay = macio_get_drvdata(mdev);
714
715         if (mdev->ofdev.dev.power.power_state.event != PM_EVENT_ON) {
716                 mdev->ofdev.dev.power.power_state = PMSG_ON;
717
718                 /* We re-enable the bay using it's previous content
719                    only if it did not change. Note those bozo timings,
720                    they seem to help the 3400 get it right.
721                  */
722                 /* Force MB power to 0 */
723                 down(&bay->lock);
724                 set_mb_power(bay, 0);
725                 msleep(MB_POWER_DELAY);
726                 if (bay->ops->content(bay) != bay->content_id) {
727                         printk("mediabay%d: content changed during sleep...\n", bay->index);
728                         up(&bay->lock);
729                         return 0;
730                 }
731                 set_mb_power(bay, 1);
732                 bay->last_value = bay->content_id;
733                 bay->value_count = msecs_to_jiffies(MB_STABLE_DELAY);
734                 bay->timer = msecs_to_jiffies(MB_POWER_DELAY);
735 #ifdef CONFIG_BLK_DEV_IDE_PMAC
736                 bay->cd_retry = 0;
737 #endif
738                 do {
739                         msleep(MB_POLL_DELAY);
740                         media_bay_step(bay->index);
741                 } while((bay->state != mb_empty) &&
742                         (bay->state != mb_up));
743                 bay->sleeping = 0;
744                 up(&bay->lock);
745         }
746         return 0;
747 }
748
749
750 /* Definitions of "ops" structures.
751  */
752 static struct mb_ops ohare_mb_ops = {
753         .name           = "Ohare",
754         .content        = ohare_mb_content,
755         .power          = ohare_mb_power,
756         .setup_bus      = ohare_mb_setup_bus,
757         .un_reset       = ohare_mb_un_reset,
758         .un_reset_ide   = ohare_mb_un_reset_ide,
759 };
760
761 static struct mb_ops heathrow_mb_ops = {
762         .name           = "Heathrow",
763         .content        = heathrow_mb_content,
764         .power          = heathrow_mb_power,
765         .setup_bus      = heathrow_mb_setup_bus,
766         .un_reset       = heathrow_mb_un_reset,
767         .un_reset_ide   = heathrow_mb_un_reset_ide,
768 };
769
770 static struct mb_ops keylargo_mb_ops = {
771         .name           = "KeyLargo",
772         .init           = keylargo_mb_init,
773         .content        = keylargo_mb_content,
774         .power          = keylargo_mb_power,
775         .setup_bus      = keylargo_mb_setup_bus,
776         .un_reset       = keylargo_mb_un_reset,
777         .un_reset_ide   = keylargo_mb_un_reset_ide,
778 };
779
780 /*
781  * It seems that the bit for the media-bay interrupt in the IRQ_LEVEL
782  * register is always set when there is something in the media bay.
783  * This causes problems for the interrupt code if we attach an interrupt
784  * handler to the media-bay interrupt, because it tends to go into
785  * an infinite loop calling the media bay interrupt handler.
786  * Therefore we do it all by polling the media bay once each tick.
787  */
788
789 static struct of_device_id media_bay_match[] =
790 {
791         {
792         .name           = "media-bay",
793         .compatible     = "keylargo-media-bay",
794         .data           = &keylargo_mb_ops,
795         },
796         {
797         .name           = "media-bay",
798         .compatible     = "heathrow-media-bay",
799         .data           = &heathrow_mb_ops,
800         },
801         {
802         .name           = "media-bay",
803         .compatible     = "ohare-media-bay",
804         .data           = &ohare_mb_ops,
805         },
806         {},
807 };
808
809 static struct macio_driver media_bay_driver =
810 {
811         .name           = "media-bay",
812         .match_table    = media_bay_match,
813         .probe          = media_bay_attach,
814         .suspend        = media_bay_suspend,
815         .resume         = media_bay_resume
816 };
817
818 static int __init media_bay_init(void)
819 {
820         int i;
821
822         for (i=0; i<MAX_BAYS; i++) {
823                 memset((char *)&media_bays[i], 0, sizeof(struct media_bay_info));
824                 media_bays[i].content_id        = -1;
825 #ifdef CONFIG_BLK_DEV_IDE_PMAC
826                 media_bays[i].cd_index          = -1;
827 #endif
828         }
829         if (!machine_is(powermac))
830                 return 0;
831
832         macio_register_driver(&media_bay_driver);       
833
834         return 0;
835 }
836
837 device_initcall(media_bay_init);