f48c6dd47e04d1910a3f74b85997e3e6089917a3
[pandora-kernel.git] / drivers / block / swim3.c
1 /*
2  * Driver for the SWIM3 (Super Woz Integrated Machine 3)
3  * floppy controller found on Power Macintoshes.
4  *
5  * Copyright (C) 1996 Paul Mackerras.
6  *
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * as published by the Free Software Foundation; either version
10  * 2 of the License, or (at your option) any later version.
11  */
12
13 /*
14  * TODO:
15  * handle 2 drives
16  * handle GCR disks
17  */
18
19 #include <linux/stddef.h>
20 #include <linux/kernel.h>
21 #include <linux/sched.h>
22 #include <linux/timer.h>
23 #include <linux/delay.h>
24 #include <linux/fd.h>
25 #include <linux/ioctl.h>
26 #include <linux/blkdev.h>
27 #include <linux/interrupt.h>
28 #include <linux/module.h>
29 #include <linux/spinlock.h>
30 #include <asm/io.h>
31 #include <asm/dbdma.h>
32 #include <asm/prom.h>
33 #include <asm/uaccess.h>
34 #include <asm/mediabay.h>
35 #include <asm/machdep.h>
36 #include <asm/pmac_feature.h>
37
38 static struct request_queue *swim3_queue;
39 static struct gendisk *disks[2];
40 static struct request *fd_req;
41
42 #define MAX_FLOPPIES    2
43
44 enum swim_state {
45         idle,
46         locating,
47         seeking,
48         settling,
49         do_transfer,
50         jogging,
51         available,
52         revalidating,
53         ejecting
54 };
55
56 #define REG(x)  unsigned char x; char x ## _pad[15];
57
58 /*
59  * The names for these registers mostly represent speculation on my part.
60  * It will be interesting to see how close they are to the names Apple uses.
61  */
62 struct swim3 {
63         REG(data);
64         REG(timer);             /* counts down at 1MHz */
65         REG(error);
66         REG(mode);
67         REG(select);            /* controls CA0, CA1, CA2 and LSTRB signals */
68         REG(setup);
69         REG(control);           /* writing bits clears them */
70         REG(status);            /* writing bits sets them in control */
71         REG(intr);
72         REG(nseek);             /* # tracks to seek */
73         REG(ctrack);            /* current track number */
74         REG(csect);             /* current sector number */
75         REG(gap3);              /* size of gap 3 in track format */
76         REG(sector);            /* sector # to read or write */
77         REG(nsect);             /* # sectors to read or write */
78         REG(intr_enable);
79 };
80
81 #define control_bic     control
82 #define control_bis     status
83
84 /* Bits in select register */
85 #define CA_MASK         7
86 #define LSTRB           8
87
88 /* Bits in control register */
89 #define DO_SEEK         0x80
90 #define FORMAT          0x40
91 #define SELECT          0x20
92 #define WRITE_SECTORS   0x10
93 #define DO_ACTION       0x08
94 #define DRIVE2_ENABLE   0x04
95 #define DRIVE_ENABLE    0x02
96 #define INTR_ENABLE     0x01
97
98 /* Bits in status register */
99 #define FIFO_1BYTE      0x80
100 #define FIFO_2BYTE      0x40
101 #define ERROR           0x20
102 #define DATA            0x08
103 #define RDDATA          0x04
104 #define INTR_PENDING    0x02
105 #define MARK_BYTE       0x01
106
107 /* Bits in intr and intr_enable registers */
108 #define ERROR_INTR      0x20
109 #define DATA_CHANGED    0x10
110 #define TRANSFER_DONE   0x08
111 #define SEEN_SECTOR     0x04
112 #define SEEK_DONE       0x02
113 #define TIMER_DONE      0x01
114
115 /* Bits in error register */
116 #define ERR_DATA_CRC    0x80
117 #define ERR_ADDR_CRC    0x40
118 #define ERR_OVERRUN     0x04
119 #define ERR_UNDERRUN    0x01
120
121 /* Bits in setup register */
122 #define S_SW_RESET      0x80
123 #define S_GCR_WRITE     0x40
124 #define S_IBM_DRIVE     0x20
125 #define S_TEST_MODE     0x10
126 #define S_FCLK_DIV2     0x08
127 #define S_GCR           0x04
128 #define S_COPY_PROT     0x02
129 #define S_INV_WDATA     0x01
130
131 /* Select values for swim3_action */
132 #define SEEK_POSITIVE   0
133 #define SEEK_NEGATIVE   4
134 #define STEP            1
135 #define MOTOR_ON        2
136 #define MOTOR_OFF       6
137 #define INDEX           3
138 #define EJECT           7
139 #define SETMFM          9
140 #define SETGCR          13
141
142 /* Select values for swim3_select and swim3_readbit */
143 #define STEP_DIR        0
144 #define STEPPING        1
145 #define MOTOR_ON        2
146 #define RELAX           3       /* also eject in progress */
147 #define READ_DATA_0     4
148 #define TWOMEG_DRIVE    5
149 #define SINGLE_SIDED    6       /* drive or diskette is 4MB type? */
150 #define DRIVE_PRESENT   7
151 #define DISK_IN         8
152 #define WRITE_PROT      9
153 #define TRACK_ZERO      10
154 #define TACHO           11
155 #define READ_DATA_1     12
156 #define MFM_MODE        13
157 #define SEEK_COMPLETE   14
158 #define ONEMEG_MEDIA    15
159
160 /* Definitions of values used in writing and formatting */
161 #define DATA_ESCAPE     0x99
162 #define GCR_SYNC_EXC    0x3f
163 #define GCR_SYNC_CONV   0x80
164 #define GCR_FIRST_MARK  0xd5
165 #define GCR_SECOND_MARK 0xaa
166 #define GCR_ADDR_MARK   "\xd5\xaa\x00"
167 #define GCR_DATA_MARK   "\xd5\xaa\x0b"
168 #define GCR_SLIP_BYTE   "\x27\xaa"
169 #define GCR_SELF_SYNC   "\x3f\xbf\x1e\x34\x3c\x3f"
170
171 #define DATA_99         "\x99\x99"
172 #define MFM_ADDR_MARK   "\x99\xa1\x99\xa1\x99\xa1\x99\xfe"
173 #define MFM_INDEX_MARK  "\x99\xc2\x99\xc2\x99\xc2\x99\xfc"
174 #define MFM_GAP_LEN     12
175
176 struct floppy_state {
177         enum swim_state state;
178         spinlock_t lock;
179         struct swim3 __iomem *swim3;    /* hardware registers */
180         struct dbdma_regs __iomem *dma; /* DMA controller registers */
181         int     swim3_intr;     /* interrupt number for SWIM3 */
182         int     dma_intr;       /* interrupt number for DMA channel */
183         int     cur_cyl;        /* cylinder head is on, or -1 */
184         int     cur_sector;     /* last sector we saw go past */
185         int     req_cyl;        /* the cylinder for the current r/w request */
186         int     head;           /* head number ditto */
187         int     req_sector;     /* sector number ditto */
188         int     scount;         /* # sectors we're transferring at present */
189         int     retries;
190         int     settle_time;
191         int     secpercyl;      /* disk geometry information */
192         int     secpertrack;
193         int     total_secs;
194         int     write_prot;     /* 1 if write-protected, 0 if not, -1 dunno */
195         struct dbdma_cmd *dma_cmd;
196         int     ref_count;
197         int     expect_cyl;
198         struct timer_list timeout;
199         int     timeout_pending;
200         int     ejected;
201         wait_queue_head_t wait;
202         int     wanted;
203         struct device_node*     media_bay; /* NULL when not in bay */
204         char    dbdma_cmd_space[5 * sizeof(struct dbdma_cmd)];
205 };
206
207 static struct floppy_state floppy_states[MAX_FLOPPIES];
208 static int floppy_count = 0;
209 static DEFINE_SPINLOCK(swim3_lock);
210
211 static unsigned short write_preamble[] = {
212         0x4e4e, 0x4e4e, 0x4e4e, 0x4e4e, 0x4e4e, /* gap field */
213         0, 0, 0, 0, 0, 0,                       /* sync field */
214         0x99a1, 0x99a1, 0x99a1, 0x99fb,         /* data address mark */
215         0x990f                                  /* no escape for 512 bytes */
216 };
217
218 static unsigned short write_postamble[] = {
219         0x9904,                                 /* insert CRC */
220         0x4e4e, 0x4e4e,
221         0x9908,                                 /* stop writing */
222         0, 0, 0, 0, 0, 0
223 };
224
225 static void swim3_select(struct floppy_state *fs, int sel);
226 static void swim3_action(struct floppy_state *fs, int action);
227 static int swim3_readbit(struct floppy_state *fs, int bit);
228 static void do_fd_request(struct request_queue * q);
229 static void start_request(struct floppy_state *fs);
230 static void set_timeout(struct floppy_state *fs, int nticks,
231                         void (*proc)(unsigned long));
232 static void scan_track(struct floppy_state *fs);
233 static void seek_track(struct floppy_state *fs, int n);
234 static void init_dma(struct dbdma_cmd *cp, int cmd, void *buf, int count);
235 static void setup_transfer(struct floppy_state *fs);
236 static void act(struct floppy_state *fs);
237 static void scan_timeout(unsigned long data);
238 static void seek_timeout(unsigned long data);
239 static void settle_timeout(unsigned long data);
240 static void xfer_timeout(unsigned long data);
241 static irqreturn_t swim3_interrupt(int irq, void *dev_id);
242 /*static void fd_dma_interrupt(int irq, void *dev_id);*/
243 static int grab_drive(struct floppy_state *fs, enum swim_state state,
244                       int interruptible);
245 static void release_drive(struct floppy_state *fs);
246 static int fd_eject(struct floppy_state *fs);
247 static int floppy_ioctl(struct block_device *bdev, fmode_t mode,
248                         unsigned int cmd, unsigned long param);
249 static int floppy_open(struct block_device *bdev, fmode_t mode);
250 static int floppy_release(struct gendisk *disk, fmode_t mode);
251 static int floppy_check_change(struct gendisk *disk);
252 static int floppy_revalidate(struct gendisk *disk);
253
254 static bool swim3_end_request(int err, unsigned int nr_bytes)
255 {
256         if (__blk_end_request(fd_req, err, nr_bytes))
257                 return true;
258
259         fd_req = NULL;
260         return false;
261 }
262
263 static bool swim3_end_request_cur(int err)
264 {
265         return swim3_end_request(err, blk_rq_cur_bytes(fd_req));
266 }
267
268 static void swim3_select(struct floppy_state *fs, int sel)
269 {
270         struct swim3 __iomem *sw = fs->swim3;
271
272         out_8(&sw->select, RELAX);
273         if (sel & 8)
274                 out_8(&sw->control_bis, SELECT);
275         else
276                 out_8(&sw->control_bic, SELECT);
277         out_8(&sw->select, sel & CA_MASK);
278 }
279
280 static void swim3_action(struct floppy_state *fs, int action)
281 {
282         struct swim3 __iomem *sw = fs->swim3;
283
284         swim3_select(fs, action);
285         udelay(1);
286         out_8(&sw->select, sw->select | LSTRB);
287         udelay(2);
288         out_8(&sw->select, sw->select & ~LSTRB);
289         udelay(1);
290 }
291
292 static int swim3_readbit(struct floppy_state *fs, int bit)
293 {
294         struct swim3 __iomem *sw = fs->swim3;
295         int stat;
296
297         swim3_select(fs, bit);
298         udelay(1);
299         stat = in_8(&sw->status);
300         return (stat & DATA) == 0;
301 }
302
303 static void do_fd_request(struct request_queue * q)
304 {
305         int i;
306         for(i=0;i<floppy_count;i++)
307         {
308 #ifdef CONFIG_PMAC_MEDIABAY
309                 if (floppy_states[i].media_bay &&
310                         check_media_bay(floppy_states[i].media_bay, MB_FD))
311                         continue;
312 #endif /* CONFIG_PMAC_MEDIABAY */
313                 start_request(&floppy_states[i]);
314         }
315 }
316
317 static void start_request(struct floppy_state *fs)
318 {
319         struct request *req;
320         unsigned long x;
321
322         if (fs->state == idle && fs->wanted) {
323                 fs->state = available;
324                 wake_up(&fs->wait);
325                 return;
326         }
327         while (fs->state == idle) {
328                 if (!fd_req) {
329                         fd_req = elv_next_request(swim3_queue);
330                         if (!fd_req)
331                                 break;
332                         blkdev_dequeue_request(fd_req);
333                 }
334                 req = fd_req;
335 #if 0
336                 printk("do_fd_req: dev=%s cmd=%d sec=%ld nr_sec=%u buf=%p\n",
337                        req->rq_disk->disk_name, req->cmd,
338                        (long)blk_rq_pos(req), blk_rq_sectors(req), req->buffer);
339                 printk("           errors=%d current_nr_sectors=%u\n",
340                        req->errors, blk_rq_cur_sectors(req));
341 #endif
342
343                 if (blk_rq_pos(req) >= fs->total_secs) {
344                         swim3_end_request_cur(-EIO);
345                         continue;
346                 }
347                 if (fs->ejected) {
348                         swim3_end_request_cur(-EIO);
349                         continue;
350                 }
351
352                 if (rq_data_dir(req) == WRITE) {
353                         if (fs->write_prot < 0)
354                                 fs->write_prot = swim3_readbit(fs, WRITE_PROT);
355                         if (fs->write_prot) {
356                                 swim3_end_request_cur(-EIO);
357                                 continue;
358                         }
359                 }
360
361                 /* Do not remove the cast. blk_rq_pos(req) is now a
362                  * sector_t and can be 64 bits, but it will never go
363                  * past 32 bits for this driver anyway, so we can
364                  * safely cast it down and not have to do a 64/32
365                  * division
366                  */
367                 fs->req_cyl = ((long)blk_rq_pos(req)) / fs->secpercyl;
368                 x = ((long)blk_rq_pos(req)) % fs->secpercyl;
369                 fs->head = x / fs->secpertrack;
370                 fs->req_sector = x % fs->secpertrack + 1;
371                 fd_req = req;
372                 fs->state = do_transfer;
373                 fs->retries = 0;
374
375                 act(fs);
376         }
377 }
378
379 static void set_timeout(struct floppy_state *fs, int nticks,
380                         void (*proc)(unsigned long))
381 {
382         unsigned long flags;
383
384         spin_lock_irqsave(&fs->lock, flags);
385         if (fs->timeout_pending)
386                 del_timer(&fs->timeout);
387         fs->timeout.expires = jiffies + nticks;
388         fs->timeout.function = proc;
389         fs->timeout.data = (unsigned long) fs;
390         add_timer(&fs->timeout);
391         fs->timeout_pending = 1;
392         spin_unlock_irqrestore(&fs->lock, flags);
393 }
394
395 static inline void scan_track(struct floppy_state *fs)
396 {
397         struct swim3 __iomem *sw = fs->swim3;
398
399         swim3_select(fs, READ_DATA_0);
400         in_8(&sw->intr);                /* clear SEEN_SECTOR bit */
401         in_8(&sw->error);
402         out_8(&sw->intr_enable, SEEN_SECTOR);
403         out_8(&sw->control_bis, DO_ACTION);
404         /* enable intr when track found */
405         set_timeout(fs, HZ, scan_timeout);      /* enable timeout */
406 }
407
408 static inline void seek_track(struct floppy_state *fs, int n)
409 {
410         struct swim3 __iomem *sw = fs->swim3;
411
412         if (n >= 0) {
413                 swim3_action(fs, SEEK_POSITIVE);
414                 sw->nseek = n;
415         } else {
416                 swim3_action(fs, SEEK_NEGATIVE);
417                 sw->nseek = -n;
418         }
419         fs->expect_cyl = (fs->cur_cyl >= 0)? fs->cur_cyl + n: -1;
420         swim3_select(fs, STEP);
421         in_8(&sw->error);
422         /* enable intr when seek finished */
423         out_8(&sw->intr_enable, SEEK_DONE);
424         out_8(&sw->control_bis, DO_SEEK);
425         set_timeout(fs, 3*HZ, seek_timeout);    /* enable timeout */
426         fs->settle_time = 0;
427 }
428
429 static inline void init_dma(struct dbdma_cmd *cp, int cmd,
430                             void *buf, int count)
431 {
432         st_le16(&cp->req_count, count);
433         st_le16(&cp->command, cmd);
434         st_le32(&cp->phy_addr, virt_to_bus(buf));
435         cp->xfer_status = 0;
436 }
437
438 static inline void setup_transfer(struct floppy_state *fs)
439 {
440         int n;
441         struct swim3 __iomem *sw = fs->swim3;
442         struct dbdma_cmd *cp = fs->dma_cmd;
443         struct dbdma_regs __iomem *dr = fs->dma;
444
445         if (blk_rq_cur_sectors(fd_req) <= 0) {
446                 printk(KERN_ERR "swim3: transfer 0 sectors?\n");
447                 return;
448         }
449         if (rq_data_dir(fd_req) == WRITE)
450                 n = 1;
451         else {
452                 n = fs->secpertrack - fs->req_sector + 1;
453                 if (n > blk_rq_cur_sectors(fd_req))
454                         n = blk_rq_cur_sectors(fd_req);
455         }
456         fs->scount = n;
457         swim3_select(fs, fs->head? READ_DATA_1: READ_DATA_0);
458         out_8(&sw->sector, fs->req_sector);
459         out_8(&sw->nsect, n);
460         out_8(&sw->gap3, 0);
461         out_le32(&dr->cmdptr, virt_to_bus(cp));
462         if (rq_data_dir(fd_req) == WRITE) {
463                 /* Set up 3 dma commands: write preamble, data, postamble */
464                 init_dma(cp, OUTPUT_MORE, write_preamble, sizeof(write_preamble));
465                 ++cp;
466                 init_dma(cp, OUTPUT_MORE, fd_req->buffer, 512);
467                 ++cp;
468                 init_dma(cp, OUTPUT_LAST, write_postamble, sizeof(write_postamble));
469         } else {
470                 init_dma(cp, INPUT_LAST, fd_req->buffer, n * 512);
471         }
472         ++cp;
473         out_le16(&cp->command, DBDMA_STOP);
474         out_8(&sw->control_bic, DO_ACTION | WRITE_SECTORS);
475         in_8(&sw->error);
476         out_8(&sw->control_bic, DO_ACTION | WRITE_SECTORS);
477         if (rq_data_dir(fd_req) == WRITE)
478                 out_8(&sw->control_bis, WRITE_SECTORS);
479         in_8(&sw->intr);
480         out_le32(&dr->control, (RUN << 16) | RUN);
481         /* enable intr when transfer complete */
482         out_8(&sw->intr_enable, TRANSFER_DONE);
483         out_8(&sw->control_bis, DO_ACTION);
484         set_timeout(fs, 2*HZ, xfer_timeout);    /* enable timeout */
485 }
486
487 static void act(struct floppy_state *fs)
488 {
489         for (;;) {
490                 switch (fs->state) {
491                 case idle:
492                         return;         /* XXX shouldn't get here */
493
494                 case locating:
495                         if (swim3_readbit(fs, TRACK_ZERO)) {
496                                 fs->cur_cyl = 0;
497                                 if (fs->req_cyl == 0)
498                                         fs->state = do_transfer;
499                                 else
500                                         fs->state = seeking;
501                                 break;
502                         }
503                         scan_track(fs);
504                         return;
505
506                 case seeking:
507                         if (fs->cur_cyl < 0) {
508                                 fs->expect_cyl = -1;
509                                 fs->state = locating;
510                                 break;
511                         }
512                         if (fs->req_cyl == fs->cur_cyl) {
513                                 printk("whoops, seeking 0\n");
514                                 fs->state = do_transfer;
515                                 break;
516                         }
517                         seek_track(fs, fs->req_cyl - fs->cur_cyl);
518                         return;
519
520                 case settling:
521                         /* check for SEEK_COMPLETE after 30ms */
522                         fs->settle_time = (HZ + 32) / 33;
523                         set_timeout(fs, fs->settle_time, settle_timeout);
524                         return;
525
526                 case do_transfer:
527                         if (fs->cur_cyl != fs->req_cyl) {
528                                 if (fs->retries > 5) {
529                                         swim3_end_request_cur(-EIO);
530                                         fs->state = idle;
531                                         return;
532                                 }
533                                 fs->state = seeking;
534                                 break;
535                         }
536                         setup_transfer(fs);
537                         return;
538
539                 case jogging:
540                         seek_track(fs, -5);
541                         return;
542
543                 default:
544                         printk(KERN_ERR"swim3: unknown state %d\n", fs->state);
545                         return;
546                 }
547         }
548 }
549
550 static void scan_timeout(unsigned long data)
551 {
552         struct floppy_state *fs = (struct floppy_state *) data;
553         struct swim3 __iomem *sw = fs->swim3;
554
555         fs->timeout_pending = 0;
556         out_8(&sw->control_bic, DO_ACTION | WRITE_SECTORS);
557         out_8(&sw->select, RELAX);
558         out_8(&sw->intr_enable, 0);
559         fs->cur_cyl = -1;
560         if (fs->retries > 5) {
561                 swim3_end_request_cur(-EIO);
562                 fs->state = idle;
563                 start_request(fs);
564         } else {
565                 fs->state = jogging;
566                 act(fs);
567         }
568 }
569
570 static void seek_timeout(unsigned long data)
571 {
572         struct floppy_state *fs = (struct floppy_state *) data;
573         struct swim3 __iomem *sw = fs->swim3;
574
575         fs->timeout_pending = 0;
576         out_8(&sw->control_bic, DO_SEEK);
577         out_8(&sw->select, RELAX);
578         out_8(&sw->intr_enable, 0);
579         printk(KERN_ERR "swim3: seek timeout\n");
580         swim3_end_request_cur(-EIO);
581         fs->state = idle;
582         start_request(fs);
583 }
584
585 static void settle_timeout(unsigned long data)
586 {
587         struct floppy_state *fs = (struct floppy_state *) data;
588         struct swim3 __iomem *sw = fs->swim3;
589
590         fs->timeout_pending = 0;
591         if (swim3_readbit(fs, SEEK_COMPLETE)) {
592                 out_8(&sw->select, RELAX);
593                 fs->state = locating;
594                 act(fs);
595                 return;
596         }
597         out_8(&sw->select, RELAX);
598         if (fs->settle_time < 2*HZ) {
599                 ++fs->settle_time;
600                 set_timeout(fs, 1, settle_timeout);
601                 return;
602         }
603         printk(KERN_ERR "swim3: seek settle timeout\n");
604         swim3_end_request_cur(-EIO);
605         fs->state = idle;
606         start_request(fs);
607 }
608
609 static void xfer_timeout(unsigned long data)
610 {
611         struct floppy_state *fs = (struct floppy_state *) data;
612         struct swim3 __iomem *sw = fs->swim3;
613         struct dbdma_regs __iomem *dr = fs->dma;
614         int n;
615
616         fs->timeout_pending = 0;
617         out_le32(&dr->control, RUN << 16);
618         /* We must wait a bit for dbdma to stop */
619         for (n = 0; (in_le32(&dr->status) & ACTIVE) && n < 1000; n++)
620                 udelay(1);
621         out_8(&sw->intr_enable, 0);
622         out_8(&sw->control_bic, WRITE_SECTORS | DO_ACTION);
623         out_8(&sw->select, RELAX);
624         printk(KERN_ERR "swim3: timeout %sing sector %ld\n",
625                (rq_data_dir(fd_req)==WRITE? "writ": "read"),
626                (long)blk_rq_pos(fd_req));
627         swim3_end_request_cur(-EIO);
628         fs->state = idle;
629         start_request(fs);
630 }
631
632 static irqreturn_t swim3_interrupt(int irq, void *dev_id)
633 {
634         struct floppy_state *fs = (struct floppy_state *) dev_id;
635         struct swim3 __iomem *sw = fs->swim3;
636         int intr, err, n;
637         int stat, resid;
638         struct dbdma_regs __iomem *dr;
639         struct dbdma_cmd *cp;
640
641         intr = in_8(&sw->intr);
642         err = (intr & ERROR_INTR)? in_8(&sw->error): 0;
643         if ((intr & ERROR_INTR) && fs->state != do_transfer)
644                 printk(KERN_ERR "swim3_interrupt, state=%d, dir=%x, intr=%x, err=%x\n",
645                        fs->state, rq_data_dir(fd_req), intr, err);
646         switch (fs->state) {
647         case locating:
648                 if (intr & SEEN_SECTOR) {
649                         out_8(&sw->control_bic, DO_ACTION | WRITE_SECTORS);
650                         out_8(&sw->select, RELAX);
651                         out_8(&sw->intr_enable, 0);
652                         del_timer(&fs->timeout);
653                         fs->timeout_pending = 0;
654                         if (sw->ctrack == 0xff) {
655                                 printk(KERN_ERR "swim3: seen sector but cyl=ff?\n");
656                                 fs->cur_cyl = -1;
657                                 if (fs->retries > 5) {
658                                         swim3_end_request_cur(-EIO);
659                                         fs->state = idle;
660                                         start_request(fs);
661                                 } else {
662                                         fs->state = jogging;
663                                         act(fs);
664                                 }
665                                 break;
666                         }
667                         fs->cur_cyl = sw->ctrack;
668                         fs->cur_sector = sw->csect;
669                         if (fs->expect_cyl != -1 && fs->expect_cyl != fs->cur_cyl)
670                                 printk(KERN_ERR "swim3: expected cyl %d, got %d\n",
671                                        fs->expect_cyl, fs->cur_cyl);
672                         fs->state = do_transfer;
673                         act(fs);
674                 }
675                 break;
676         case seeking:
677         case jogging:
678                 if (sw->nseek == 0) {
679                         out_8(&sw->control_bic, DO_SEEK);
680                         out_8(&sw->select, RELAX);
681                         out_8(&sw->intr_enable, 0);
682                         del_timer(&fs->timeout);
683                         fs->timeout_pending = 0;
684                         if (fs->state == seeking)
685                                 ++fs->retries;
686                         fs->state = settling;
687                         act(fs);
688                 }
689                 break;
690         case settling:
691                 out_8(&sw->intr_enable, 0);
692                 del_timer(&fs->timeout);
693                 fs->timeout_pending = 0;
694                 act(fs);
695                 break;
696         case do_transfer:
697                 if ((intr & (ERROR_INTR | TRANSFER_DONE)) == 0)
698                         break;
699                 out_8(&sw->intr_enable, 0);
700                 out_8(&sw->control_bic, WRITE_SECTORS | DO_ACTION);
701                 out_8(&sw->select, RELAX);
702                 del_timer(&fs->timeout);
703                 fs->timeout_pending = 0;
704                 dr = fs->dma;
705                 cp = fs->dma_cmd;
706                 if (rq_data_dir(fd_req) == WRITE)
707                         ++cp;
708                 /*
709                  * Check that the main data transfer has finished.
710                  * On writing, the swim3 sometimes doesn't use
711                  * up all the bytes of the postamble, so we can still
712                  * see DMA active here.  That doesn't matter as long
713                  * as all the sector data has been transferred.
714                  */
715                 if ((intr & ERROR_INTR) == 0 && cp->xfer_status == 0) {
716                         /* wait a little while for DMA to complete */
717                         for (n = 0; n < 100; ++n) {
718                                 if (cp->xfer_status != 0)
719                                         break;
720                                 udelay(1);
721                                 barrier();
722                         }
723                 }
724                 /* turn off DMA */
725                 out_le32(&dr->control, (RUN | PAUSE) << 16);
726                 stat = ld_le16(&cp->xfer_status);
727                 resid = ld_le16(&cp->res_count);
728                 if (intr & ERROR_INTR) {
729                         n = fs->scount - 1 - resid / 512;
730                         if (n > 0) {
731                                 blk_update_request(fd_req, 0, n << 9);
732                                 fs->req_sector += n;
733                         }
734                         if (fs->retries < 5) {
735                                 ++fs->retries;
736                                 act(fs);
737                         } else {
738                                 printk("swim3: error %sing block %ld (err=%x)\n",
739                                        rq_data_dir(fd_req) == WRITE? "writ": "read",
740                                        (long)blk_rq_pos(fd_req), err);
741                                 swim3_end_request_cur(-EIO);
742                                 fs->state = idle;
743                         }
744                 } else {
745                         if ((stat & ACTIVE) == 0 || resid != 0) {
746                                 /* musta been an error */
747                                 printk(KERN_ERR "swim3: fd dma: stat=%x resid=%d\n", stat, resid);
748                                 printk(KERN_ERR "  state=%d, dir=%x, intr=%x, err=%x\n",
749                                        fs->state, rq_data_dir(fd_req), intr, err);
750                                 swim3_end_request_cur(-EIO);
751                                 fs->state = idle;
752                                 start_request(fs);
753                                 break;
754                         }
755                         if (swim3_end_request(0, fs->scount << 9)) {
756                                 fs->req_sector += fs->scount;
757                                 if (fs->req_sector > fs->secpertrack) {
758                                         fs->req_sector -= fs->secpertrack;
759                                         if (++fs->head > 1) {
760                                                 fs->head = 0;
761                                                 ++fs->req_cyl;
762                                         }
763                                 }
764                                 act(fs);
765                         } else
766                                 fs->state = idle;
767                 }
768                 if (fs->state == idle)
769                         start_request(fs);
770                 break;
771         default:
772                 printk(KERN_ERR "swim3: don't know what to do in state %d\n", fs->state);
773         }
774         return IRQ_HANDLED;
775 }
776
777 /*
778 static void fd_dma_interrupt(int irq, void *dev_id)
779 {
780 }
781 */
782
783 static int grab_drive(struct floppy_state *fs, enum swim_state state,
784                       int interruptible)
785 {
786         unsigned long flags;
787
788         spin_lock_irqsave(&fs->lock, flags);
789         if (fs->state != idle) {
790                 ++fs->wanted;
791                 while (fs->state != available) {
792                         if (interruptible && signal_pending(current)) {
793                                 --fs->wanted;
794                                 spin_unlock_irqrestore(&fs->lock, flags);
795                                 return -EINTR;
796                         }
797                         interruptible_sleep_on(&fs->wait);
798                 }
799                 --fs->wanted;
800         }
801         fs->state = state;
802         spin_unlock_irqrestore(&fs->lock, flags);
803         return 0;
804 }
805
806 static void release_drive(struct floppy_state *fs)
807 {
808         unsigned long flags;
809
810         spin_lock_irqsave(&fs->lock, flags);
811         fs->state = idle;
812         start_request(fs);
813         spin_unlock_irqrestore(&fs->lock, flags);
814 }
815
816 static int fd_eject(struct floppy_state *fs)
817 {
818         int err, n;
819
820         err = grab_drive(fs, ejecting, 1);
821         if (err)
822                 return err;
823         swim3_action(fs, EJECT);
824         for (n = 20; n > 0; --n) {
825                 if (signal_pending(current)) {
826                         err = -EINTR;
827                         break;
828                 }
829                 swim3_select(fs, RELAX);
830                 schedule_timeout_interruptible(1);
831                 if (swim3_readbit(fs, DISK_IN) == 0)
832                         break;
833         }
834         swim3_select(fs, RELAX);
835         udelay(150);
836         fs->ejected = 1;
837         release_drive(fs);
838         return err;
839 }
840
841 static struct floppy_struct floppy_type =
842         { 2880,18,2,80,0,0x1B,0x00,0xCF,0x6C,NULL };    /*  7 1.44MB 3.5"   */
843
844 static int floppy_ioctl(struct block_device *bdev, fmode_t mode,
845                         unsigned int cmd, unsigned long param)
846 {
847         struct floppy_state *fs = bdev->bd_disk->private_data;
848         int err;
849                 
850         if ((cmd & 0x80) && !capable(CAP_SYS_ADMIN))
851                 return -EPERM;
852
853 #ifdef CONFIG_PMAC_MEDIABAY
854         if (fs->media_bay && check_media_bay(fs->media_bay, MB_FD))
855                 return -ENXIO;
856 #endif
857
858         switch (cmd) {
859         case FDEJECT:
860                 if (fs->ref_count != 1)
861                         return -EBUSY;
862                 err = fd_eject(fs);
863                 return err;
864         case FDGETPRM:
865                 if (copy_to_user((void __user *) param, &floppy_type,
866                                  sizeof(struct floppy_struct)))
867                         return -EFAULT;
868                 return 0;
869         }
870         return -ENOTTY;
871 }
872
873 static int floppy_open(struct block_device *bdev, fmode_t mode)
874 {
875         struct floppy_state *fs = bdev->bd_disk->private_data;
876         struct swim3 __iomem *sw = fs->swim3;
877         int n, err = 0;
878
879         if (fs->ref_count == 0) {
880 #ifdef CONFIG_PMAC_MEDIABAY
881                 if (fs->media_bay && check_media_bay(fs->media_bay, MB_FD))
882                         return -ENXIO;
883 #endif
884                 out_8(&sw->setup, S_IBM_DRIVE | S_FCLK_DIV2);
885                 out_8(&sw->control_bic, 0xff);
886                 out_8(&sw->mode, 0x95);
887                 udelay(10);
888                 out_8(&sw->intr_enable, 0);
889                 out_8(&sw->control_bis, DRIVE_ENABLE | INTR_ENABLE);
890                 swim3_action(fs, MOTOR_ON);
891                 fs->write_prot = -1;
892                 fs->cur_cyl = -1;
893                 for (n = 0; n < 2 * HZ; ++n) {
894                         if (n >= HZ/30 && swim3_readbit(fs, SEEK_COMPLETE))
895                                 break;
896                         if (signal_pending(current)) {
897                                 err = -EINTR;
898                                 break;
899                         }
900                         swim3_select(fs, RELAX);
901                         schedule_timeout_interruptible(1);
902                 }
903                 if (err == 0 && (swim3_readbit(fs, SEEK_COMPLETE) == 0
904                                  || swim3_readbit(fs, DISK_IN) == 0))
905                         err = -ENXIO;
906                 swim3_action(fs, SETMFM);
907                 swim3_select(fs, RELAX);
908
909         } else if (fs->ref_count == -1 || mode & FMODE_EXCL)
910                 return -EBUSY;
911
912         if (err == 0 && (mode & FMODE_NDELAY) == 0
913             && (mode & (FMODE_READ|FMODE_WRITE))) {
914                 check_disk_change(bdev);
915                 if (fs->ejected)
916                         err = -ENXIO;
917         }
918
919         if (err == 0 && (mode & FMODE_WRITE)) {
920                 if (fs->write_prot < 0)
921                         fs->write_prot = swim3_readbit(fs, WRITE_PROT);
922                 if (fs->write_prot)
923                         err = -EROFS;
924         }
925
926         if (err) {
927                 if (fs->ref_count == 0) {
928                         swim3_action(fs, MOTOR_OFF);
929                         out_8(&sw->control_bic, DRIVE_ENABLE | INTR_ENABLE);
930                         swim3_select(fs, RELAX);
931                 }
932                 return err;
933         }
934
935         if (mode & FMODE_EXCL)
936                 fs->ref_count = -1;
937         else
938                 ++fs->ref_count;
939
940         return 0;
941 }
942
943 static int floppy_release(struct gendisk *disk, fmode_t mode)
944 {
945         struct floppy_state *fs = disk->private_data;
946         struct swim3 __iomem *sw = fs->swim3;
947         if (fs->ref_count > 0 && --fs->ref_count == 0) {
948                 swim3_action(fs, MOTOR_OFF);
949                 out_8(&sw->control_bic, 0xff);
950                 swim3_select(fs, RELAX);
951         }
952         return 0;
953 }
954
955 static int floppy_check_change(struct gendisk *disk)
956 {
957         struct floppy_state *fs = disk->private_data;
958         return fs->ejected;
959 }
960
961 static int floppy_revalidate(struct gendisk *disk)
962 {
963         struct floppy_state *fs = disk->private_data;
964         struct swim3 __iomem *sw;
965         int ret, n;
966
967 #ifdef CONFIG_PMAC_MEDIABAY
968         if (fs->media_bay && check_media_bay(fs->media_bay, MB_FD))
969                 return -ENXIO;
970 #endif
971
972         sw = fs->swim3;
973         grab_drive(fs, revalidating, 0);
974         out_8(&sw->intr_enable, 0);
975         out_8(&sw->control_bis, DRIVE_ENABLE);
976         swim3_action(fs, MOTOR_ON);     /* necessary? */
977         fs->write_prot = -1;
978         fs->cur_cyl = -1;
979         mdelay(1);
980         for (n = HZ; n > 0; --n) {
981                 if (swim3_readbit(fs, SEEK_COMPLETE))
982                         break;
983                 if (signal_pending(current))
984                         break;
985                 swim3_select(fs, RELAX);
986                 schedule_timeout_interruptible(1);
987         }
988         ret = swim3_readbit(fs, SEEK_COMPLETE) == 0
989                 || swim3_readbit(fs, DISK_IN) == 0;
990         if (ret)
991                 swim3_action(fs, MOTOR_OFF);
992         else {
993                 fs->ejected = 0;
994                 swim3_action(fs, SETMFM);
995         }
996         swim3_select(fs, RELAX);
997
998         release_drive(fs);
999         return ret;
1000 }
1001
1002 static struct block_device_operations floppy_fops = {
1003         .open           = floppy_open,
1004         .release        = floppy_release,
1005         .locked_ioctl   = floppy_ioctl,
1006         .media_changed  = floppy_check_change,
1007         .revalidate_disk= floppy_revalidate,
1008 };
1009
1010 static int swim3_add_device(struct macio_dev *mdev, int index)
1011 {
1012         struct device_node *swim = mdev->ofdev.node;
1013         struct device_node *mediabay;
1014         struct floppy_state *fs = &floppy_states[index];
1015         int rc = -EBUSY;
1016
1017         /* Check & Request resources */
1018         if (macio_resource_count(mdev) < 2) {
1019                 printk(KERN_WARNING "ifd%d: no address for %s\n",
1020                        index, swim->full_name);
1021                 return -ENXIO;
1022         }
1023         if (macio_irq_count(mdev) < 2) {
1024                 printk(KERN_WARNING "fd%d: no intrs for device %s\n",
1025                         index, swim->full_name);
1026         }
1027         if (macio_request_resource(mdev, 0, "swim3 (mmio)")) {
1028                 printk(KERN_ERR "fd%d: can't request mmio resource for %s\n",
1029                        index, swim->full_name);
1030                 return -EBUSY;
1031         }
1032         if (macio_request_resource(mdev, 1, "swim3 (dma)")) {
1033                 printk(KERN_ERR "fd%d: can't request dma resource for %s\n",
1034                        index, swim->full_name);
1035                 macio_release_resource(mdev, 0);
1036                 return -EBUSY;
1037         }
1038         dev_set_drvdata(&mdev->ofdev.dev, fs);
1039
1040         mediabay = (strcasecmp(swim->parent->type, "media-bay") == 0) ?
1041                 swim->parent : NULL;
1042         if (mediabay == NULL)
1043                 pmac_call_feature(PMAC_FTR_SWIM3_ENABLE, swim, 0, 1);
1044         
1045         memset(fs, 0, sizeof(*fs));
1046         spin_lock_init(&fs->lock);
1047         fs->state = idle;
1048         fs->swim3 = (struct swim3 __iomem *)
1049                 ioremap(macio_resource_start(mdev, 0), 0x200);
1050         if (fs->swim3 == NULL) {
1051                 printk("fd%d: couldn't map registers for %s\n",
1052                        index, swim->full_name);
1053                 rc = -ENOMEM;
1054                 goto out_release;
1055         }
1056         fs->dma = (struct dbdma_regs __iomem *)
1057                 ioremap(macio_resource_start(mdev, 1), 0x200);
1058         if (fs->dma == NULL) {
1059                 printk("fd%d: couldn't map DMA for %s\n",
1060                        index, swim->full_name);
1061                 iounmap(fs->swim3);
1062                 rc = -ENOMEM;
1063                 goto out_release;
1064         }
1065         fs->swim3_intr = macio_irq(mdev, 0);
1066         fs->dma_intr = macio_irq(mdev, 1);;
1067         fs->cur_cyl = -1;
1068         fs->cur_sector = -1;
1069         fs->secpercyl = 36;
1070         fs->secpertrack = 18;
1071         fs->total_secs = 2880;
1072         fs->media_bay = mediabay;
1073         init_waitqueue_head(&fs->wait);
1074
1075         fs->dma_cmd = (struct dbdma_cmd *) DBDMA_ALIGN(fs->dbdma_cmd_space);
1076         memset(fs->dma_cmd, 0, 2 * sizeof(struct dbdma_cmd));
1077         st_le16(&fs->dma_cmd[1].command, DBDMA_STOP);
1078
1079         if (request_irq(fs->swim3_intr, swim3_interrupt, 0, "SWIM3", fs)) {
1080                 printk(KERN_ERR "fd%d: couldn't request irq %d for %s\n",
1081                        index, fs->swim3_intr, swim->full_name);
1082                 pmac_call_feature(PMAC_FTR_SWIM3_ENABLE, swim, 0, 0);
1083                 goto out_unmap;
1084                 return -EBUSY;
1085         }
1086 /*
1087         if (request_irq(fs->dma_intr, fd_dma_interrupt, 0, "SWIM3-dma", fs)) {
1088                 printk(KERN_ERR "Couldn't get irq %d for SWIM3 DMA",
1089                        fs->dma_intr);
1090                 return -EBUSY;
1091         }
1092 */
1093
1094         init_timer(&fs->timeout);
1095
1096         printk(KERN_INFO "fd%d: SWIM3 floppy controller %s\n", floppy_count,
1097                 mediabay ? "in media bay" : "");
1098
1099         return 0;
1100
1101  out_unmap:
1102         iounmap(fs->dma);
1103         iounmap(fs->swim3);
1104
1105  out_release:
1106         macio_release_resource(mdev, 0);
1107         macio_release_resource(mdev, 1);
1108
1109         return rc;
1110 }
1111
1112 static int __devinit swim3_attach(struct macio_dev *mdev, const struct of_device_id *match)
1113 {
1114         int i, rc;
1115         struct gendisk *disk;
1116
1117         /* Add the drive */
1118         rc = swim3_add_device(mdev, floppy_count);
1119         if (rc)
1120                 return rc;
1121
1122         /* Now create the queue if not there yet */
1123         if (swim3_queue == NULL) {
1124                 /* If we failed, there isn't much we can do as the driver is still
1125                  * too dumb to remove the device, just bail out
1126                  */
1127                 if (register_blkdev(FLOPPY_MAJOR, "fd"))
1128                         return 0;
1129                 swim3_queue = blk_init_queue(do_fd_request, &swim3_lock);
1130                 if (swim3_queue == NULL) {
1131                         unregister_blkdev(FLOPPY_MAJOR, "fd");
1132                         return 0;
1133                 }
1134         }
1135
1136         /* Now register that disk. Same comment about failure handling */
1137         i = floppy_count++;
1138         disk = disks[i] = alloc_disk(1);
1139         if (disk == NULL)
1140                 return 0;
1141
1142         disk->major = FLOPPY_MAJOR;
1143         disk->first_minor = i;
1144         disk->fops = &floppy_fops;
1145         disk->private_data = &floppy_states[i];
1146         disk->queue = swim3_queue;
1147         disk->flags |= GENHD_FL_REMOVABLE;
1148         sprintf(disk->disk_name, "fd%d", i);
1149         set_capacity(disk, 2880);
1150         add_disk(disk);
1151
1152         return 0;
1153 }
1154
1155 static struct of_device_id swim3_match[] =
1156 {
1157         {
1158         .name           = "swim3",
1159         },
1160         {
1161         .compatible     = "ohare-swim3"
1162         },
1163         {
1164         .compatible     = "swim3"
1165         },
1166 };
1167
1168 static struct macio_driver swim3_driver =
1169 {
1170         .name           = "swim3",
1171         .match_table    = swim3_match,
1172         .probe          = swim3_attach,
1173 #if 0
1174         .suspend        = swim3_suspend,
1175         .resume         = swim3_resume,
1176 #endif
1177 };
1178
1179
1180 int swim3_init(void)
1181 {
1182         macio_register_driver(&swim3_driver);
1183         return 0;
1184 }
1185
1186 module_init(swim3_init)
1187
1188 MODULE_LICENSE("GPL");
1189 MODULE_AUTHOR("Paul Mackerras");
1190 MODULE_ALIAS_BLOCKDEV_MAJOR(FLOPPY_MAJOR);