drivers/block/swim.c

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