linux/drivers/block/swim.c
<<
>>
Prefs
   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
  33struct 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
  46struct 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
  71struct 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 TWOMEG_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 MFM_MODE        0x175
 122#define SEEK_COMPLETE   0x176
 123#define ONEMEG_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
 159enum drive_location {
 160        INTERNAL_DRIVE = 0x02,
 161        EXTERNAL_DRIVE = 0x04,
 162};
 163
 164enum media_type {
 165        DD_MEDIA,
 166        HD_MEDIA,
 167};
 168
 169struct 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
 199enum motor_action {
 200        OFF,
 201        ON,
 202};
 203
 204enum head {
 205        LOWER_HEAD = 0,
 206        UPPER_HEAD = 1,
 207};
 208
 209#define FD_MAX_UNIT     2
 210
 211struct 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
 219extern int swim_read_sector_header(struct swim __iomem *base,
 220                                   struct sector_header *header);
 221extern int swim_read_sector_data(struct swim __iomem *base,
 222                                 unsigned char *data);
 223
 224static DEFINE_MUTEX(swim_mutex);
 225static 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
 250static 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;
 267is_iwm:
 268        local_irq_restore(flags);
 269        return 0;
 270}
 271
 272static 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
 281static 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
 297static 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
 310static 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
 322static 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
 343static 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
 359static 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
 369static 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
 387static 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
 409static 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
 428static 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
 445static 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
 459static 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
 496static 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
 528static 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
 548static 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
 582static 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
 589static 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
 607static 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                fs->type = swim_readbit(base, ONEMEG_MEDIA);
 616
 617                if (swim_track00(base))
 618                        printk(KERN_ERR
 619                                "SWIM: cannot move floppy head to track 0\n");
 620
 621                swim_track00(base);
 622
 623                get_floppy_geometry(fs, 0, &g);
 624                fs->total_secs = g->size;
 625                fs->secpercyl = g->head * g->sect;
 626                fs->secpertrack = g->sect;
 627                fs->track = 0;
 628        } else {
 629                fs->disk_in = 0;
 630        }
 631}
 632
 633static int floppy_open(struct block_device *bdev, fmode_t mode)
 634{
 635        struct floppy_state *fs = bdev->bd_disk->private_data;
 636        struct swim __iomem *base = fs->swd->base;
 637        int err;
 638
 639        if (fs->ref_count == -1 || (fs->ref_count && mode & FMODE_EXCL))
 640                return -EBUSY;
 641
 642        if (mode & FMODE_EXCL)
 643                fs->ref_count = -1;
 644        else
 645                fs->ref_count++;
 646
 647        swim_write(base, setup, S_IBM_DRIVE  | S_FCLK_DIV2);
 648        udelay(10);
 649        swim_drive(base, INTERNAL_DRIVE);
 650        swim_motor(base, ON);
 651        swim_action(base, SETMFM);
 652        if (fs->ejected)
 653                setup_medium(fs);
 654        if (!fs->disk_in) {
 655                err = -ENXIO;
 656                goto out;
 657        }
 658
 659        if (mode & FMODE_NDELAY)
 660                return 0;
 661
 662        if (mode & (FMODE_READ|FMODE_WRITE)) {
 663                check_disk_change(bdev);
 664                if ((mode & FMODE_WRITE) && fs->write_protected) {
 665                        err = -EROFS;
 666                        goto out;
 667                }
 668        }
 669        return 0;
 670out:
 671        if (fs->ref_count < 0)
 672                fs->ref_count = 0;
 673        else if (fs->ref_count > 0)
 674                --fs->ref_count;
 675
 676        if (fs->ref_count == 0)
 677                swim_motor(base, OFF);
 678        return err;
 679}
 680
 681static int floppy_unlocked_open(struct block_device *bdev, fmode_t mode)
 682{
 683        int ret;
 684
 685        mutex_lock(&swim_mutex);
 686        ret = floppy_open(bdev, mode);
 687        mutex_unlock(&swim_mutex);
 688
 689        return ret;
 690}
 691
 692static void floppy_release(struct gendisk *disk, fmode_t mode)
 693{
 694        struct floppy_state *fs = disk->private_data;
 695        struct swim __iomem *base = fs->swd->base;
 696
 697        mutex_lock(&swim_mutex);
 698        if (fs->ref_count < 0)
 699                fs->ref_count = 0;
 700        else if (fs->ref_count > 0)
 701                --fs->ref_count;
 702
 703        if (fs->ref_count == 0)
 704                swim_motor(base, OFF);
 705        mutex_unlock(&swim_mutex);
 706}
 707
 708static int floppy_ioctl(struct block_device *bdev, fmode_t mode,
 709                        unsigned int cmd, unsigned long param)
 710{
 711        struct floppy_state *fs = bdev->bd_disk->private_data;
 712        int err;
 713
 714        if ((cmd & 0x80) && !capable(CAP_SYS_ADMIN))
 715                        return -EPERM;
 716
 717        switch (cmd) {
 718        case FDEJECT:
 719                if (fs->ref_count != 1)
 720                        return -EBUSY;
 721                mutex_lock(&swim_mutex);
 722                err = floppy_eject(fs);
 723                mutex_unlock(&swim_mutex);
 724                return err;
 725
 726        case FDGETPRM:
 727                if (copy_to_user((void __user *) param, (void *) &floppy_type,
 728                                 sizeof(struct floppy_struct)))
 729                        return -EFAULT;
 730                break;
 731
 732        default:
 733                printk(KERN_DEBUG "SWIM floppy_ioctl: unknown cmd %d\n",
 734                       cmd);
 735                return -ENOSYS;
 736        }
 737        return 0;
 738}
 739
 740static int floppy_getgeo(struct block_device *bdev, struct hd_geometry *geo)
 741{
 742        struct floppy_state *fs = bdev->bd_disk->private_data;
 743        struct floppy_struct *g;
 744        int ret;
 745
 746        ret = get_floppy_geometry(fs, 0, &g);
 747        if (ret)
 748                return ret;
 749
 750        geo->heads = g->head;
 751        geo->sectors = g->sect;
 752        geo->cylinders = g->track;
 753
 754        return 0;
 755}
 756
 757static unsigned int floppy_check_events(struct gendisk *disk,
 758                                        unsigned int clearing)
 759{
 760        struct floppy_state *fs = disk->private_data;
 761
 762        return fs->ejected ? DISK_EVENT_MEDIA_CHANGE : 0;
 763}
 764
 765static int floppy_revalidate(struct gendisk *disk)
 766{
 767        struct floppy_state *fs = disk->private_data;
 768        struct swim __iomem *base = fs->swd->base;
 769
 770        swim_drive(base, fs->location);
 771
 772        if (fs->ejected)
 773                setup_medium(fs);
 774
 775        if (!fs->disk_in)
 776                swim_motor(base, OFF);
 777        else
 778                fs->ejected = 0;
 779
 780        return !fs->disk_in;
 781}
 782
 783static const struct block_device_operations floppy_fops = {
 784        .owner           = THIS_MODULE,
 785        .open            = floppy_unlocked_open,
 786        .release         = floppy_release,
 787        .ioctl           = floppy_ioctl,
 788        .getgeo          = floppy_getgeo,
 789        .check_events    = floppy_check_events,
 790        .revalidate_disk = floppy_revalidate,
 791};
 792
 793static struct kobject *floppy_find(dev_t dev, int *part, void *data)
 794{
 795        struct swim_priv *swd = data;
 796        int drive = (*part & 3);
 797
 798        if (drive > swd->floppy_count)
 799                return NULL;
 800
 801        *part = 0;
 802        return get_disk(swd->unit[drive].disk);
 803}
 804
 805static int swim_add_floppy(struct swim_priv *swd, enum drive_location location)
 806{
 807        struct floppy_state *fs = &swd->unit[swd->floppy_count];
 808        struct swim __iomem *base = swd->base;
 809
 810        fs->location = location;
 811
 812        swim_drive(base, location);
 813
 814        swim_motor(base, OFF);
 815
 816        if (swim_readbit(base, SINGLE_SIDED))
 817                fs->head_number = 1;
 818        else
 819                fs->head_number = 2;
 820        fs->ref_count = 0;
 821        fs->ejected = 1;
 822
 823        swd->floppy_count++;
 824
 825        return 0;
 826}
 827
 828static int swim_floppy_init(struct swim_priv *swd)
 829{
 830        int err;
 831        int drive;
 832        struct swim __iomem *base = swd->base;
 833
 834        /* scan floppy drives */
 835
 836        swim_drive(base, INTERNAL_DRIVE);
 837        if (swim_readbit(base, DRIVE_PRESENT))
 838                swim_add_floppy(swd, INTERNAL_DRIVE);
 839        swim_drive(base, EXTERNAL_DRIVE);
 840        if (swim_readbit(base, DRIVE_PRESENT))
 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                        put_disk(swd->unit[drive].disk);
 865                        goto exit_put_disks;
 866                }
 867                blk_queue_bounce_limit(swd->unit[drive].disk->queue,
 868                                BLK_BOUNCE_HIGH);
 869                swd->unit[drive].disk->queue->queuedata = swd;
 870                swd->unit[drive].swd = swd;
 871        }
 872
 873        for (drive = 0; drive < swd->floppy_count; drive++) {
 874                swd->unit[drive].disk->flags = GENHD_FL_REMOVABLE;
 875                swd->unit[drive].disk->major = FLOPPY_MAJOR;
 876                swd->unit[drive].disk->first_minor = drive;
 877                sprintf(swd->unit[drive].disk->disk_name, "fd%d", drive);
 878                swd->unit[drive].disk->fops = &floppy_fops;
 879                swd->unit[drive].disk->private_data = &swd->unit[drive];
 880                set_capacity(swd->unit[drive].disk, 2880);
 881                add_disk(swd->unit[drive].disk);
 882        }
 883
 884        blk_register_region(MKDEV(FLOPPY_MAJOR, 0), 256, THIS_MODULE,
 885                            floppy_find, NULL, swd);
 886
 887        return 0;
 888
 889exit_put_disks:
 890        unregister_blkdev(FLOPPY_MAJOR, "fd");
 891        while (drive--)
 892                put_disk(swd->unit[drive].disk);
 893        return err;
 894}
 895
 896static int swim_probe(struct platform_device *dev)
 897{
 898        struct resource *res;
 899        struct swim __iomem *swim_base;
 900        struct swim_priv *swd;
 901        int ret;
 902
 903        res = platform_get_resource(dev, IORESOURCE_MEM, 0);
 904        if (!res) {
 905                ret = -ENODEV;
 906                goto out;
 907        }
 908
 909        if (!request_mem_region(res->start, resource_size(res), CARDNAME)) {
 910                ret = -EBUSY;
 911                goto out;
 912        }
 913
 914        swim_base = ioremap(res->start, resource_size(res));
 915        if (!swim_base) {
 916                ret = -ENOMEM;
 917                goto out_release_io;
 918        }
 919
 920        /* probe device */
 921
 922        set_swim_mode(swim_base, 1);
 923        if (!get_swim_mode(swim_base)) {
 924                printk(KERN_INFO "SWIM device not found !\n");
 925                ret = -ENODEV;
 926                goto out_iounmap;
 927        }
 928
 929        /* set platform driver data */
 930
 931        swd = kzalloc(sizeof(struct swim_priv), GFP_KERNEL);
 932        if (!swd) {
 933                ret = -ENOMEM;
 934                goto out_iounmap;
 935        }
 936        platform_set_drvdata(dev, swd);
 937
 938        swd->base = swim_base;
 939
 940        ret = swim_floppy_init(swd);
 941        if (ret)
 942                goto out_kfree;
 943
 944        return 0;
 945
 946out_kfree:
 947        kfree(swd);
 948out_iounmap:
 949        iounmap(swim_base);
 950out_release_io:
 951        release_mem_region(res->start, resource_size(res));
 952out:
 953        return ret;
 954}
 955
 956static int swim_remove(struct platform_device *dev)
 957{
 958        struct swim_priv *swd = platform_get_drvdata(dev);
 959        int drive;
 960        struct resource *res;
 961
 962        blk_unregister_region(MKDEV(FLOPPY_MAJOR, 0), 256);
 963
 964        for (drive = 0; drive < swd->floppy_count; drive++) {
 965                del_gendisk(swd->unit[drive].disk);
 966                blk_cleanup_queue(swd->unit[drive].disk->queue);
 967                put_disk(swd->unit[drive].disk);
 968        }
 969
 970        unregister_blkdev(FLOPPY_MAJOR, "fd");
 971
 972        /* eject floppies */
 973
 974        for (drive = 0; drive < swd->floppy_count; drive++)
 975                floppy_eject(&swd->unit[drive]);
 976
 977        iounmap(swd->base);
 978
 979        res = platform_get_resource(dev, IORESOURCE_MEM, 0);
 980        if (res)
 981                release_mem_region(res->start, resource_size(res));
 982
 983        kfree(swd);
 984
 985        return 0;
 986}
 987
 988static struct platform_driver swim_driver = {
 989        .probe  = swim_probe,
 990        .remove = swim_remove,
 991        .driver   = {
 992                .name   = CARDNAME,
 993        },
 994};
 995
 996static int __init swim_init(void)
 997{
 998        printk(KERN_INFO "SWIM floppy driver %s\n", DRIVER_VERSION);
 999
1000        return platform_driver_register(&swim_driver);
1001}
1002module_init(swim_init);
1003
1004static void __exit swim_exit(void)
1005{
1006        platform_driver_unregister(&swim_driver);
1007}
1008module_exit(swim_exit);
1009
1010MODULE_DESCRIPTION("Driver for SWIM floppy controller");
1011MODULE_LICENSE("GPL");
1012MODULE_AUTHOR("Laurent Vivier <laurent@lvivier.info>");
1013MODULE_ALIAS_BLOCKDEV_MAJOR(FLOPPY_MAJOR);
1014