drivers/block/swim.c

   1 /*
   2  * Driver for SWIM (Sander Woz Integrated Machine) floppy controller
   3  *
   4  * Copyright (C) 2004,2008 Laurent Vivier <Laurent@lvivier.info>
   5  *
   6  * based on Alastair Bridgewater SWIM analysis, 2001
   7  * based on SWIM3 driver (c) Paul Mackerras, 1996
   8  * based on netBSD IWM driver (c) 1997, 1998 Hauke Fath.
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License
  12  * as published by the Free Software Foundation; either version
  13  * 2 of the License, or (at your option) any later version.
  14  *
  15  * 2004-08-21 (lv) - Initial implementation
  16  * 2008-10-30 (lv) - Port to 2.6
  17  */
  18
  19 #include <linux/module.h>
  20 #include <linux/fd.h>
  21 #include <linux/slab.h>
  22 #include <linux/blkdev.h>
  23 #include <linux/mutex.h>
  24 #include <linux/hdreg.h>
  25 #include <linux/kernel.h>
  26 #include <linux/delay.h>
  27 #include <linux/platform_device.h>
  28
  29 #include <asm/mac_via.h>
  30
  31 #define CARDNAME "swim"
  32
  33 struct sector_header {
  34         unsigned char side;
  35         unsigned char track;
  36         unsigned char sector;
  37         unsigned char size;
  38         unsigned char crc0;
  39         unsigned char crc1;
  40 } __attribute__((packed));
  41
  42 #define DRIVER_VERSION "Version 0.2 (2008-10-30)"
  43
  44 #define REG(x)  unsigned char x, x ## _pad[0x200 - 1];
  45
  46 struct swim {
  47         REG(write_data)
  48         REG(write_mark)
  49         REG(write_CRC)
  50         REG(write_parameter)
  51         REG(write_phase)
  52         REG(write_setup)
  53         REG(write_mode0)
  54         REG(write_mode1)
  55
  56         REG(read_data)
  57         REG(read_mark)
  58         REG(read_error)
  59         REG(read_parameter)
  60         REG(read_phase)
  61         REG(read_setup)
  62         REG(read_status)
  63         REG(read_handshake)
  64 } __attribute__((packed));
  65
  66 #define swim_write(base, reg, v)        out_8(&(base)->write_##reg, (v))
  67 #define swim_read(base, reg)            in_8(&(base)->read_##reg)
  68
  69 /* IWM registers */
  70
  71 struct iwm {
  72         REG(ph0L)
  73         REG(ph0H)
  74         REG(ph1L)
  75         REG(ph1H)
  76         REG(ph2L)
  77         REG(ph2H)
  78         REG(ph3L)
  79         REG(ph3H)
  80         REG(mtrOff)
  81         REG(mtrOn)
  82         REG(intDrive)
  83         REG(extDrive)
  84         REG(q6L)
  85         REG(q6H)
  86         REG(q7L)
  87         REG(q7H)
  88 } __attribute__((packed));
  89
  90 #define iwm_write(base, reg, v)         out_8(&(base)->reg, (v))
  91 #define iwm_read(base, reg)             in_8(&(base)->reg)
  92
  93 /* bits in phase register */
  94
  95 #define SEEK_POSITIVE   0x070
  96 #define SEEK_NEGATIVE   0x074
  97 #define STEP            0x071
  98 #define MOTOR_ON        0x072
  99 #define MOTOR_OFF       0x076
 100 #define INDEX           0x073
 101 #define EJECT           0x077
 102 #define SETMFM          0x171
 103 #define SETGCR          0x175
 104
 105 #define RELAX           0x033
 106 #define LSTRB           0x008
 107
 108 #define CA_MASK         0x077
 109
 110 /* Select values for swim_select and swim_readbit */
 111
 112 #define READ_DATA_0     0x074
 113 #define ONEMEG_DRIVE    0x075
 114 #define SINGLE_SIDED    0x076
 115 #define DRIVE_PRESENT   0x077
 116 #define DISK_IN         0x170
 117 #define WRITE_PROT      0x171
 118 #define TRACK_ZERO      0x172
 119 #define TACHO           0x173
 120 #define READ_DATA_1     0x174
 121 #define GCR_MODE        0x175
 122 #define SEEK_COMPLETE   0x176
 123 #define TWOMEG_MEDIA    0x177
 124
 125 /* Bits in handshake register */
 126
 127 #define MARK_BYTE       0x01
 128 #define CRC_ZERO        0x02
 129 #define RDDATA          0x04
 130 #define SENSE           0x08
 131 #define MOTEN           0x10
 132 #define ERROR           0x20
 133 #define DAT2BYTE        0x40
 134 #define DAT1BYTE        0x80
 135
 136 /* bits in setup register */
 137
 138 #define S_INV_WDATA     0x01
 139 #define S_3_5_SELECT    0x02
 140 #define S_GCR           0x04
 141 #define S_FCLK_DIV2     0x08
 142 #define S_ERROR_CORR    0x10
 143 #define S_IBM_DRIVE     0x20
 144 #define S_GCR_WRITE     0x40
 145 #define S_TIMEOUT       0x80
 146
 147 /* bits in mode register */
 148
 149 #define CLFIFO          0x01
 150 #define ENBL1           0x02
 151 #define ENBL2           0x04
 152 #define ACTION          0x08
 153 #define WRITE_MODE      0x10
 154 #define HEDSEL          0x20
 155 #define MOTON           0x80
 156
 157 /*----------------------------------------------------------------------------*/
 158
 159 enum drive_location {
 160         INTERNAL_DRIVE = 0x02,
 161         EXTERNAL_DRIVE = 0x04,
 162 };
 163
 164 enum media_type {
 165         DD_MEDIA,
 166         HD_MEDIA,
 167 };
 168
 169 struct floppy_state {
 170
 171         /* physical properties */
 172
 173         enum drive_location location;   /* internal or external drive */
 174         int              head_number;   /* single- or double-sided drive */
 175
 176         /* media */
 177
 178         int              disk_in;
 179         int              ejected;
 180         enum media_type  type;
 181         int              write_protected;
 182
 183         int              total_secs;
 184         int              secpercyl;
 185         int              secpertrack;
 186
 187         /* in-use information */
 188
 189         int             track;
 190         int             ref_count;
 191
 192         struct gendisk *disk;
 193
 194         /* parent controller */
 195
 196         struct swim_priv *swd;
 197 };
 198
 199 enum motor_action {
 200         OFF,
 201         ON,
 202 };
 203
 204 enum head {
 205         LOWER_HEAD = 0,
 206         UPPER_HEAD = 1,
 207 };
 208
 209 #define FD_MAX_UNIT     2
 210
 211 struct swim_priv {
 212         struct swim __iomem *base;
 213         spinlock_t lock;
 214         int fdc_queue;
 215         int floppy_count;
 216         struct floppy_state unit[FD_MAX_UNIT];
 217 };
 218
 219 extern int swim_read_sector_header(struct swim __iomem *base,
 220                                    struct sector_header *header);
 221 extern int swim_read_sector_data(struct swim __iomem *base,
 222                                  unsigned char *data);
 223
 224 static DEFINE_MUTEX(swim_mutex);
 225 static inline void set_swim_mode(struct swim __iomem *base, int enable)
 226 {
 227         struct iwm __iomem *iwm_base;
 228         unsigned long flags;
 229
 230         if (!enable) {
 231                 swim_write(base, mode0, 0xf8);
 232                 return;
 233         }
 234
 235         iwm_base = (struct iwm __iomem *)base;
 236         local_irq_save(flags);
 237
 238         iwm_read(iwm_base, q7L);
 239         iwm_read(iwm_base, mtrOff);
 240         iwm_read(iwm_base, q6H);
 241
 242         iwm_write(iwm_base, q7H, 0x57);
 243         iwm_write(iwm_base, q7H, 0x17);
 244         iwm_write(iwm_base, q7H, 0x57);
 245         iwm_write(iwm_base, q7H, 0x57);
 246
 247         local_irq_restore(flags);
 248 }
 249
 250 static inline int get_swim_mode(struct swim __iomem *base)
 251 {
 252         unsigned long flags;
 253
 254         local_irq_save(flags);
 255
 256         swim_write(base, phase, 0xf5);
 257         if (swim_read(base, phase) != 0xf5)
 258                 goto is_iwm;
 259         swim_write(base, phase, 0xf6);
 260         if (swim_read(base, phase) != 0xf6)
 261                 goto is_iwm;
 262         swim_write(base, phase, 0xf7);
 263         if (swim_read(base, phase) != 0xf7)
 264                 goto is_iwm;
 265         local_irq_restore(flags);
 266         return 1;
 267 is_iwm:
 268         local_irq_restore(flags);
 269         return 0;
 270 }
 271
 272 static inline void swim_select(struct swim __iomem *base, int sel)
 273 {
 274         swim_write(base, phase, RELAX);
 275
 276         via1_set_head(sel & 0x100);
 277
 278         swim_write(base, phase, sel & CA_MASK);
 279 }
 280
 281 static inline void swim_action(struct swim __iomem *base, int action)
 282 {
 283         unsigned long flags;
 284
 285         local_irq_save(flags);
 286
 287         swim_select(base, action);
 288         udelay(1);
 289         swim_write(base, phase, (LSTRB<<4) | LSTRB);
 290         udelay(1);
 291         swim_write(base, phase, (LSTRB<<4) | ((~LSTRB) & 0x0F));
 292         udelay(1);
 293
 294         local_irq_restore(flags);
 295 }
 296
 297 static inline int swim_readbit(struct swim __iomem *base, int bit)
 298 {
 299         int stat;
 300
 301         swim_select(base, bit);
 302
 303         udelay(10);
 304
 305         stat = swim_read(base, handshake);
 306
 307         return (stat & SENSE) == 0;
 308 }
 309
 310 static inline void swim_drive(struct swim __iomem *base,
 311                               enum drive_location location)
 312 {
 313         if (location == INTERNAL_DRIVE) {
 314                 swim_write(base, mode0, EXTERNAL_DRIVE); /* clear drive 1 bit */
 315                 swim_write(base, mode1, INTERNAL_DRIVE); /* set drive 0 bit */
 316         } else if (location == EXTERNAL_DRIVE) {
 317                 swim_write(base, mode0, INTERNAL_DRIVE); /* clear drive 0 bit */
 318                 swim_write(base, mode1, EXTERNAL_DRIVE); /* set drive 1 bit */
 319         }
 320 }
 321
 322 static inline void swim_motor(struct swim __iomem *base,
 323                               enum motor_action action)
 324 {
 325         if (action == ON) {
 326                 int i;
 327
 328                 swim_action(base, MOTOR_ON);
 329
 330                 for (i = 0; i < 2*HZ; i++) {
 331                         swim_select(base, RELAX);
 332                         if (swim_readbit(base, MOTOR_ON))
 333                                 break;
 334                         current->state = TASK_INTERRUPTIBLE;
 335                         schedule_timeout(1);
 336                 }
 337         } else if (action == OFF) {
 338                 swim_action(base, MOTOR_OFF);
 339                 swim_select(base, RELAX);
 340         }
 341 }
 342
 343 static inline void swim_eject(struct swim __iomem *base)
 344 {
 345         int i;
 346
 347         swim_action(base, EJECT);
 348
 349         for (i = 0; i < 2*HZ; i++) {
 350                 swim_select(base, RELAX);
 351                 if (!swim_readbit(base, DISK_IN))
 352                         break;
 353                 current->state = TASK_INTERRUPTIBLE;
 354                 schedule_timeout(1);
 355         }
 356         swim_select(base, RELAX);
 357 }
 358
 359 static inline void swim_head(struct swim __iomem *base, enum head head)
 360 {
 361         /* wait drive is ready */
 362
 363         if (head == UPPER_HEAD)
 364                 swim_select(base, READ_DATA_1);
 365         else if (head == LOWER_HEAD)
 366                 swim_select(base, READ_DATA_0);
 367 }
 368
 369 static inline int swim_step(struct swim __iomem *base)
 370 {
 371         int wait;
 372
 373         swim_action(base, STEP);
 374
 375         for (wait = 0; wait < HZ; wait++) {
 376
 377                 current->state = TASK_INTERRUPTIBLE;
 378                 schedule_timeout(1);
 379
 380                 swim_select(base, RELAX);
 381                 if (!swim_readbit(base, STEP))
 382                         return 0;
 383         }
 384         return -1;
 385 }
 386
 387 static inline int swim_track00(struct swim __iomem *base)
 388 {
 389         int try;
 390
 391         swim_action(base, SEEK_NEGATIVE);
 392
 393         for (try = 0; try < 100; try++) {
 394
 395                 swim_select(base, RELAX);
 396                 if (swim_readbit(base, TRACK_ZERO))
 397                         break;
 398
 399                 if (swim_step(base))
 400                         return -1;
 401         }
 402
 403         if (swim_readbit(base, TRACK_ZERO))
 404                 return 0;
 405
 406         return -1;
 407 }
 408
 409 static inline int swim_seek(struct swim __iomem *base, int step)
 410 {
 411         if (step == 0)
 412                 return 0;
 413
 414         if (step < 0) {
 415                 swim_action(base, SEEK_NEGATIVE);
 416                 step = -step;
 417         } else
 418                 swim_action(base, SEEK_POSITIVE);
 419
 420         for ( ; step > 0; step--) {
 421                 if (swim_step(base))
 422                         return -1;
 423         }
 424
 425         return 0;
 426 }
 427
 428 static inline int swim_track(struct floppy_state *fs,  int track)
 429 {
 430         struct swim __iomem *base = fs->swd->base;
 431         int ret;
 432
 433         ret = swim_seek(base, track - fs->track);
 434
 435         if (ret == 0)
 436                 fs->track = track;
 437         else {
 438                 swim_track00(base);
 439                 fs->track = 0;
 440         }
 441
 442         return ret;
 443 }
 444
 445 static int floppy_eject(struct floppy_state *fs)
 446 {
 447         struct swim __iomem *base = fs->swd->base;
 448
 449         swim_drive(base, fs->location);
 450         swim_motor(base, OFF);
 451         swim_eject(base);
 452
 453         fs->disk_in = 0;
 454         fs->ejected = 1;
 455
 456         return 0;
 457 }
 458
 459 static inline int swim_read_sector(struct floppy_state *fs,
 460                                    int side, int track,
 461                                    int sector, unsigned char *buffer)
 462 {
 463         struct swim __iomem *base = fs->swd->base;
 464         unsigned long flags;
 465         struct sector_header header;
 466         int ret = -1;
 467         short i;
 468
 469         swim_track(fs, track);
 470
 471         swim_write(base, mode1, MOTON);
 472         swim_head(base, side);
 473         swim_write(base, mode0, side);
 474
 475         local_irq_save(flags);
 476         for (i = 0; i < 36; i++) {
 477                 ret = swim_read_sector_header(base, &header);
 478                 if (!ret && (header.sector == sector)) {
 479                         /* found */
 480
 481                         ret = swim_read_sector_data(base, buffer);
 482                         break;
 483                 }
 484         }
 485         local_irq_restore(flags);
 486
 487         swim_write(base, mode0, MOTON);
 488
 489         if ((header.side != side)  || (header.track != track) ||
 490              (header.sector != sector))
 491                 return 0;
 492
 493         return ret;
 494 }
 495
 496 static blk_status_t floppy_read_sectors(struct floppy_state *fs,
 497                                int req_sector, int sectors_nb,
 498                                unsigned char *buffer)
 499 {
 500         struct swim __iomem *base = fs->swd->base;
 501         int ret;
 502         int side, track, sector;
 503         int i, try;
 504
 505
 506         swim_drive(base, fs->location);
 507         for (i = req_sector; i < req_sector + sectors_nb; i++) {
 508                 int x;
 509                 track = i / fs->secpercyl;
 510                 x = i % fs->secpercyl;
 511                 side = x / fs->secpertrack;
 512                 sector = x % fs->secpertrack + 1;
 513
 514                 try = 5;
 515                 do {
 516                         ret = swim_read_sector(fs, side, track, sector,
 517                                                 buffer);
 518                         if (try-- == 0)
 519                                 return BLK_STS_IOERR;
 520                 } while (ret != 512);
 521
 522                 buffer += ret;
 523         }
 524
 525         return 0;
 526 }
 527
 528 static struct request *swim_next_request(struct swim_priv *swd)
 529 {
 530         struct request_queue *q;
 531         struct request *rq;
 532         int old_pos = swd->fdc_queue;
 533
 534         do {
 535                 q = swd->unit[swd->fdc_queue].disk->queue;
 536                 if (++swd->fdc_queue == swd->floppy_count)
 537                         swd->fdc_queue = 0;
 538                 if (q) {
 539                         rq = blk_fetch_request(q);
 540                         if (rq)
 541                                 return rq;
 542                 }
 543         } while (swd->fdc_queue != old_pos);
 544
 545         return NULL;
 546 }
 547
 548 static void do_fd_request(struct request_queue *q)
 549 {
 550         struct swim_priv *swd = q->queuedata;
 551         struct request *req;
 552         struct floppy_state *fs;
 553
 554         req = swim_next_request(swd);
 555         while (req) {
 556                 blk_status_t err = BLK_STS_IOERR;
 557
 558                 fs = req->rq_disk->private_data;
 559                 if (blk_rq_pos(req) >= fs->total_secs)
 560                         goto done;
 561                 if (!fs->disk_in)
 562                         goto done;
 563                 if (rq_data_dir(req) == WRITE && fs->write_protected)
 564                         goto done;
 565
 566                 switch (rq_data_dir(req)) {
 567                 case WRITE:
 568                         /* NOT IMPLEMENTED */
 569                         break;
 570                 case READ:
 571                         err = floppy_read_sectors(fs, blk_rq_pos(req),
 572                                                   blk_rq_cur_sectors(req),
 573                                                   bio_data(req->bio));
 574                         break;
 575                 }
 576         done:
 577                 if (!__blk_end_request_cur(req, err))
 578                         req = swim_next_request(swd);
 579         }
 580 }
 581
 582 static struct floppy_struct floppy_type[4] = {
 583         {    0,  0, 0,  0, 0, 0x00, 0x00, 0x00, 0x00, NULL }, /* no testing   */
 584         {  720,  9, 1, 80, 0, 0x2A, 0x02, 0xDF, 0x50, NULL }, /* 360KB SS 3.5"*/
 585         { 1440,  9, 2, 80, 0, 0x2A, 0x02, 0xDF, 0x50, NULL }, /* 720KB 3.5"   */
 586         { 2880, 18, 2, 80, 0, 0x1B, 0x00, 0xCF, 0x6C, NULL }, /* 1.44MB 3.5"  */
 587 };
 588
 589 static int get_floppy_geometry(struct floppy_state *fs, int type,
 590                                struct floppy_struct **g)
 591 {
 592         if (type >= ARRAY_SIZE(floppy_type))
 593                 return -EINVAL;
 594
 595         if (type)
 596                 *g = &floppy_type[type];
 597         else if (fs->type == HD_MEDIA) /* High-Density media */
 598                 *g = &floppy_type[3];
 599         else if (fs->head_number == 2) /* double-sided */
 600                 *g = &floppy_type[2];
 601         else
 602                 *g = &floppy_type[1];
 603
 604         return 0;
 605 }
 606
 607 static void setup_medium(struct floppy_state *fs)
 608 {
 609         struct swim __iomem *base = fs->swd->base;
 610
 611         if (swim_readbit(base, DISK_IN)) {
 612                 struct floppy_struct *g;
 613                 fs->disk_in = 1;
 614                 fs->write_protected = swim_readbit(base, WRITE_PROT);
 615
 616                 if (swim_track00(base))
 617                         printk(KERN_ERR
 618                                 "SWIM: cannot move floppy head to track 0\n");
 619
 620                 swim_track00(base);
 621
 622                 fs->type = swim_readbit(base, TWOMEG_MEDIA) ?
 623                         HD_MEDIA : DD_MEDIA;
 624                 fs->head_number = swim_readbit(base, SINGLE_SIDED) ? 1 : 2;
 625                 get_floppy_geometry(fs, 0, &g);
 626                 fs->total_secs = g->size;
 627                 fs->secpercyl = g->head * g->sect;
 628                 fs->secpertrack = g->sect;
 629                 fs->track = 0;
 630         } else {
 631                 fs->disk_in = 0;
 632         }
 633 }
 634
 635 static int floppy_open(struct block_device *bdev, fmode_t mode)
 636 {
 637         struct floppy_state *fs = bdev->bd_disk->private_data;
 638         struct swim __iomem *base = fs->swd->base;
 639         int err;
 640
 641         if (fs->ref_count == -1 || (fs->ref_count && mode & FMODE_EXCL))
 642                 return -EBUSY;
 643
 644         if (mode & FMODE_EXCL)
 645                 fs->ref_count = -1;
 646         else
 647                 fs->ref_count++;
 648
 649         swim_write(base, setup, S_IBM_DRIVE  | S_FCLK_DIV2);
 650         udelay(10);
 651         swim_drive(base, fs->location);
 652         swim_motor(base, ON);
 653         swim_action(base, SETMFM);
 654         if (fs->ejected)
 655                 setup_medium(fs);
 656         if (!fs->disk_in) {
 657                 err = -ENXIO;
 658                 goto out;
 659         }
 660
 661         set_capacity(fs->disk, fs->total_secs);
 662
 663         if (mode & FMODE_NDELAY)
 664                 return 0;
 665
 666         if (mode & (FMODE_READ|FMODE_WRITE)) {
 667                 check_disk_change(bdev);
 668                 if ((mode & FMODE_WRITE) && fs->write_protected) {
 669                         err = -EROFS;
 670                         goto out;
 671                 }
 672         }
 673         return 0;
 674 out:
 675         if (fs->ref_count < 0)
 676                 fs->ref_count = 0;
 677         else if (fs->ref_count > 0)
 678                 --fs->ref_count;
 679
 680         if (fs->ref_count == 0)
 681                 swim_motor(base, OFF);
 682         return err;
 683 }
 684
 685 static int floppy_unlocked_open(struct block_device *bdev, fmode_t mode)
 686 {
 687         int ret;
 688
 689         mutex_lock(&swim_mutex);
 690         ret = floppy_open(bdev, mode);
 691         mutex_unlock(&swim_mutex);
 692
 693         return ret;
 694 }
 695
 696 static void floppy_release(struct gendisk *disk, fmode_t mode)
 697 {
 698         struct floppy_state *fs = disk->private_data;
 699         struct swim __iomem *base = fs->swd->base;
 700
 701         mutex_lock(&swim_mutex);
 702         if (fs->ref_count < 0)
 703                 fs->ref_count = 0;
 704         else if (fs->ref_count > 0)
 705                 --fs->ref_count;
 706
 707         if (fs->ref_count == 0)
 708                 swim_motor(base, OFF);
 709         mutex_unlock(&swim_mutex);
 710 }
 711
 712 static int floppy_ioctl(struct block_device *bdev, fmode_t mode,
 713                         unsigned int cmd, unsigned long param)
 714 {
 715         struct floppy_state *fs = bdev->bd_disk->private_data;
 716         int err;
 717
 718         if ((cmd & 0x80) && !capable(CAP_SYS_ADMIN))
 719                         return -EPERM;
 720
 721         switch (cmd) {
 722         case FDEJECT:
 723                 if (fs->ref_count != 1)
 724                         return -EBUSY;
 725                 mutex_lock(&swim_mutex);
 726                 err = floppy_eject(fs);
 727                 mutex_unlock(&swim_mutex);
 728                 return err;
 729
 730         case FDGETPRM:
 731                 if (copy_to_user((void __user *) param, (void *) &floppy_type,
 732                                  sizeof(struct floppy_struct)))
 733                         return -EFAULT;
 734                 return 0;
 735         }
 736         return -ENOTTY;
 737 }
 738
 739 static int floppy_getgeo(struct block_device *bdev, struct hd_geometry *geo)
 740 {
 741         struct floppy_state *fs = bdev->bd_disk->private_data;
 742         struct floppy_struct *g;
 743         int ret;
 744
 745         ret = get_floppy_geometry(fs, 0, &g);
 746         if (ret)
 747                 return ret;
 748
 749         geo->heads = g->head;
 750         geo->sectors = g->sect;
 751         geo->cylinders = g->track;
 752
 753         return 0;
 754 }
 755
 756 static unsigned int floppy_check_events(struct gendisk *disk,
 757                                         unsigned int clearing)
 758 {
 759         struct floppy_state *fs = disk->private_data;
 760
 761         return fs->ejected ? DISK_EVENT_MEDIA_CHANGE : 0;
 762 }
 763
 764 static int floppy_revalidate(struct gendisk *disk)
 765 {
 766         struct floppy_state *fs = disk->private_data;
 767         struct swim __iomem *base = fs->swd->base;
 768
 769         swim_drive(base, fs->location);
 770
 771         if (fs->ejected)
 772                 setup_medium(fs);
 773
 774         if (!fs->disk_in)
 775                 swim_motor(base, OFF);
 776         else
 777                 fs->ejected = 0;
 778
 779         return !fs->disk_in;
 780 }
 781
 782 static const struct block_device_operations floppy_fops = {
 783         .owner           = THIS_MODULE,
 784         .open            = floppy_unlocked_open,
 785         .release         = floppy_release,
 786         .ioctl           = floppy_ioctl,
 787         .getgeo          = floppy_getgeo,
 788         .check_events    = floppy_check_events,
 789         .revalidate_disk = floppy_revalidate,
 790 };
 791
 792 static struct kobject *floppy_find(dev_t dev, int *part, void *data)
 793 {
 794         struct swim_priv *swd = data;
 795         int drive = (*part & 3);
 796
 797         if (drive >= swd->floppy_count)
 798                 return NULL;
 799
 800         *part = 0;
 801         return get_disk_and_module(swd->unit[drive].disk);
 802 }
 803
 804 static int swim_add_floppy(struct swim_priv *swd, enum drive_location location)
 805 {
 806         struct floppy_state *fs = &swd->unit[swd->floppy_count];
 807         struct swim __iomem *base = swd->base;
 808
 809         fs->location = location;
 810
 811         swim_drive(base, location);
 812
 813         swim_motor(base, OFF);
 814
 815         fs->type = HD_MEDIA;
 816         fs->head_number = 2;
 817
 818         fs->ref_count = 0;
 819         fs->ejected = 1;
 820
 821         swd->floppy_count++;
 822
 823         return 0;
 824 }
 825
 826 static int swim_floppy_init(struct swim_priv *swd)
 827 {
 828         int err;
 829         int drive;
 830         struct swim __iomem *base = swd->base;
 831
 832         /* scan floppy drives */
 833
 834         swim_drive(base, INTERNAL_DRIVE);
 835         if (swim_readbit(base, DRIVE_PRESENT) &&
 836             !swim_readbit(base, ONEMEG_DRIVE))
 837                 swim_add_floppy(swd, INTERNAL_DRIVE);
 838         swim_drive(base, EXTERNAL_DRIVE);
 839         if (swim_readbit(base, DRIVE_PRESENT) &&
 840             !swim_readbit(base, ONEMEG_DRIVE))
 841                 swim_add_floppy(swd, EXTERNAL_DRIVE);
 842
 843         /* register floppy drives */
 844
 845         err = register_blkdev(FLOPPY_MAJOR, "fd");
 846         if (err) {
 847                 printk(KERN_ERR "Unable to get major %d for SWIM floppy\n",
 848                        FLOPPY_MAJOR);
 849                 return -EBUSY;
 850         }
 851
 852         spin_lock_init(&swd->lock);
 853
 854         for (drive = 0; drive < swd->floppy_count; drive++) {
 855                 swd->unit[drive].disk = alloc_disk(1);
 856                 if (swd->unit[drive].disk == NULL) {
 857                         err = -ENOMEM;
 858                         goto exit_put_disks;
 859                 }
 860                 swd->unit[drive].disk->queue = blk_init_queue(do_fd_request,
 861                                                               &swd->lock);
 862                 if (!swd->unit[drive].disk->queue) {
 863                         err = -ENOMEM;
 864                         goto exit_put_disks;
 865                 }
 866                 blk_queue_bounce_limit(swd->unit[drive].disk->queue,
 867                                 BLK_BOUNCE_HIGH);
 868                 swd->unit[drive].disk->queue->queuedata = swd;
 869                 swd->unit[drive].swd = swd;
 870         }
 871
 872         for (drive = 0; drive < swd->floppy_count; drive++) {
 873                 swd->unit[drive].disk->flags = GENHD_FL_REMOVABLE;
 874                 swd->unit[drive].disk->major = FLOPPY_MAJOR;
 875                 swd->unit[drive].disk->first_minor = drive;
 876                 sprintf(swd->unit[drive].disk->disk_name, "fd%d", drive);
 877                 swd->unit[drive].disk->fops = &floppy_fops;
 878                 swd->unit[drive].disk->private_data = &swd->unit[drive];
 879                 set_capacity(swd->unit[drive].disk, 2880);
 880                 add_disk(swd->unit[drive].disk);
 881         }
 882
 883         blk_register_region(MKDEV(FLOPPY_MAJOR, 0), 256, THIS_MODULE,
 884                             floppy_find, NULL, swd);
 885
 886         return 0;
 887
 888 exit_put_disks:
 889         unregister_blkdev(FLOPPY_MAJOR, "fd");
 890         do {
 891                 struct gendisk *disk = swd->unit[drive].disk;
 892
 893                 if (disk) {
 894                         if (disk->queue) {
 895                                 blk_cleanup_queue(disk->queue);
 896                                 disk->queue = NULL;
 897                         }
 898                         put_disk(disk);
 899                 }
 900         } while (drive--);
 901         return err;
 902 }
 903
 904 static int swim_probe(struct platform_device *dev)
 905 {
 906         struct resource *res;
 907         struct swim __iomem *swim_base;
 908         struct swim_priv *swd;
 909         int ret;
 910
 911         res = platform_get_resource(dev, IORESOURCE_MEM, 0);
 912         if (!res) {
 913                 ret = -ENODEV;
 914                 goto out;
 915         }
 916
 917         if (!request_mem_region(res->start, resource_size(res), CARDNAME)) {
 918                 ret = -EBUSY;
 919                 goto out;
 920         }
 921
 922         swim_base = (struct swim __iomem *)res->start;
 923         if (!swim_base) {
 924                 ret = -ENOMEM;
 925                 goto out_release_io;
 926         }
 927
 928         /* probe device */
 929
 930         set_swim_mode(swim_base, 1);
 931         if (!get_swim_mode(swim_base)) {
 932                 printk(KERN_INFO "SWIM device not found !\n");
 933                 ret = -ENODEV;
 934                 goto out_release_io;
 935         }
 936
 937         /* set platform driver data */
 938
 939         swd = kzalloc(sizeof(struct swim_priv), GFP_KERNEL);
 940         if (!swd) {
 941                 ret = -ENOMEM;
 942                 goto out_release_io;
 943         }
 944         platform_set_drvdata(dev, swd);
 945
 946         swd->base = swim_base;
 947
 948         ret = swim_floppy_init(swd);
 949         if (ret)
 950                 goto out_kfree;
 951
 952         return 0;
 953
 954 out_kfree:
 955         kfree(swd);
 956 out_release_io:
 957         release_mem_region(res->start, resource_size(res));
 958 out:
 959         return ret;
 960 }
 961
 962 static int swim_remove(struct platform_device *dev)
 963 {
 964         struct swim_priv *swd = platform_get_drvdata(dev);
 965         int drive;
 966         struct resource *res;
 967
 968         blk_unregister_region(MKDEV(FLOPPY_MAJOR, 0), 256);
 969
 970         for (drive = 0; drive < swd->floppy_count; drive++) {
 971                 del_gendisk(swd->unit[drive].disk);
 972                 blk_cleanup_queue(swd->unit[drive].disk->queue);
 973                 put_disk(swd->unit[drive].disk);
 974         }
 975
 976         unregister_blkdev(FLOPPY_MAJOR, "fd");
 977
 978         /* eject floppies */
 979
 980         for (drive = 0; drive < swd->floppy_count; drive++)
 981                 floppy_eject(&swd->unit[drive]);
 982
 983         res = platform_get_resource(dev, IORESOURCE_MEM, 0);
 984         if (res)
 985                 release_mem_region(res->start, resource_size(res));
 986
 987         kfree(swd);
 988
 989         return 0;
 990 }
 991
 992 static struct platform_driver swim_driver = {
 993         .probe  = swim_probe,
 994         .remove = swim_remove,
 995         .driver   = {
 996                 .name   = CARDNAME,
 997         },
 998 };
 999
1000 static int __init swim_init(void)
1001 {
1002         printk(KERN_INFO "SWIM floppy driver %s\n", DRIVER_VERSION);
1003
1004         return platform_driver_register(&swim_driver);
1005 }
1006 module_init(swim_init);
1007
1008 static void __exit swim_exit(void)
1009 {
1010         platform_driver_unregister(&swim_driver);
1011 }
1012 module_exit(swim_exit);
1013
1014 MODULE_DESCRIPTION("Driver for SWIM floppy controller");
1015 MODULE_LICENSE("GPL");
1016 MODULE_AUTHOR("Laurent Vivier <laurent@lvivier.info>");
1017 MODULE_ALIAS_BLOCKDEV_MAJOR(FLOPPY_MAJOR);