drivers/ide/ide-floppy.c

   1 /*
   2  * IDE ATAPI floppy driver.
   3  *
   4  * Copyright (C) 1996-1999  Gadi Oxman <gadio@netvision.net.il>
   5  * Copyright (C) 2000-2002  Paul Bristow <paul@paulbristow.net>
   6  * Copyright (C) 2005       Bartlomiej Zolnierkiewicz
   7  *
   8  * This driver supports the following IDE floppy drives:
   9  *
  10  * LS-120/240 SuperDisk
  11  * Iomega Zip 100/250
  12  * Iomega PC Card Clik!/PocketZip
  13  *
  14  * For a historical changelog see
  15  * Documentation/ide/ChangeLog.ide-floppy.1996-2002
  16  */
  17
  18 #define DRV_NAME "ide-floppy"
  19
  20 #define IDEFLOPPY_VERSION "1.00"
  21
  22 #include <linux/module.h>
  23 #include <linux/types.h>
  24 #include <linux/string.h>
  25 #include <linux/kernel.h>
  26 #include <linux/delay.h>
  27 #include <linux/timer.h>
  28 #include <linux/mm.h>
  29 #include <linux/interrupt.h>
  30 #include <linux/major.h>
  31 #include <linux/errno.h>
  32 #include <linux/genhd.h>
  33 #include <linux/slab.h>
  34 #include <linux/cdrom.h>
  35 #include <linux/ide.h>
  36 #include <linux/hdreg.h>
  37 #include <linux/bitops.h>
  38 #include <linux/mutex.h>
  39 #include <linux/scatterlist.h>
  40
  41 #include <scsi/scsi_ioctl.h>
  42
  43 #include <asm/byteorder.h>
  44 #include <linux/irq.h>
  45 #include <linux/uaccess.h>
  46 #include <linux/io.h>
  47 #include <asm/unaligned.h>
  48
  49 #include "ide-floppy.h"
  50
  51 /* define to see debug info */
  52 #define IDEFLOPPY_DEBUG_LOG             0
  53
  54 /* #define IDEFLOPPY_DEBUG(fmt, args...) printk(KERN_INFO fmt, ## args) */
  55 #define IDEFLOPPY_DEBUG(fmt, args...)
  56
  57 #if IDEFLOPPY_DEBUG_LOG
  58 #define debug_log(fmt, args...) \
  59         printk(KERN_INFO "ide-floppy: " fmt, ## args)
  60 #else
  61 #define debug_log(fmt, args...) do {} while (0)
  62 #endif
  63
  64 /*
  65  * After each failed packet command we issue a request sense command and retry
  66  * the packet command IDEFLOPPY_MAX_PC_RETRIES times.
  67  */
  68 #define IDEFLOPPY_MAX_PC_RETRIES        3
  69
  70 /* format capacities descriptor codes */
  71 #define CAPACITY_INVALID        0x00
  72 #define CAPACITY_UNFORMATTED    0x01
  73 #define CAPACITY_CURRENT        0x02
  74 #define CAPACITY_NO_CARTRIDGE   0x03
  75
  76 /*
  77  * The following delay solves a problem with ATAPI Zip 100 drive where BSY bit
  78  * was apparently being deasserted before the unit was ready to receive data.
  79  */
  80 #define IDEFLOPPY_PC_DELAY      (HZ/20) /* default delay for ZIP 100 (50ms) */
  81
  82 /* Error code returned in rq->errors to the higher part of the driver. */
  83 #define IDEFLOPPY_ERROR_GENERAL         101
  84
  85 static DEFINE_MUTEX(idefloppy_ref_mutex);
  86
  87 static void idefloppy_cleanup_obj(struct kref *);
  88
  89 static struct ide_floppy_obj *ide_floppy_get(struct gendisk *disk)
  90 {
  91         struct ide_floppy_obj *floppy = NULL;
  92
  93         mutex_lock(&idefloppy_ref_mutex);
  94         floppy = ide_drv_g(disk, ide_floppy_obj);
  95         if (floppy) {
  96                 if (ide_device_get(floppy->drive))
  97                         floppy = NULL;
  98                 else
  99                         kref_get(&floppy->kref);
 100         }
 101         mutex_unlock(&idefloppy_ref_mutex);
 102         return floppy;
 103 }
 104
 105 static void ide_floppy_put(struct ide_floppy_obj *floppy)
 106 {
 107         ide_drive_t *drive = floppy->drive;
 108
 109         mutex_lock(&idefloppy_ref_mutex);
 110         kref_put(&floppy->kref, idefloppy_cleanup_obj);
 111         ide_device_put(drive);
 112         mutex_unlock(&idefloppy_ref_mutex);
 113 }
 114
 115 /*
 116  * Used to finish servicing a request. For read/write requests, we will call
 117  * ide_end_request to pass to the next buffer.
 118  */
 119 static int idefloppy_end_request(ide_drive_t *drive, int uptodate, int nsecs)
 120 {
 121         idefloppy_floppy_t *floppy = drive->driver_data;
 122         struct request *rq = HWGROUP(drive)->rq;
 123         int error;
 124
 125         debug_log("Reached %s\n", __func__);
 126
 127         switch (uptodate) {
 128         case 0: error = IDEFLOPPY_ERROR_GENERAL; break;
 129         case 1: error = 0; break;
 130         default: error = uptodate;
 131         }
 132         if (error)
 133                 floppy->failed_pc = NULL;
 134         /* Why does this happen? */
 135         if (!rq)
 136                 return 0;
 137         if (!blk_special_request(rq)) {
 138                 /* our real local end request function */
 139                 ide_end_request(drive, uptodate, nsecs);
 140                 return 0;
 141         }
 142         rq->errors = error;
 143         /* fixme: need to move this local also */
 144         ide_end_drive_cmd(drive, 0, 0);
 145         return 0;
 146 }
 147
 148 static void idefloppy_update_buffers(ide_drive_t *drive,
 149                                 struct ide_atapi_pc *pc)
 150 {
 151         struct request *rq = pc->rq;
 152         struct bio *bio = rq->bio;
 153
 154         while ((bio = rq->bio) != NULL)
 155                 idefloppy_end_request(drive, 1, 0);
 156 }
 157
 158 static void ide_floppy_callback(ide_drive_t *drive, int dsc)
 159 {
 160         idefloppy_floppy_t *floppy = drive->driver_data;
 161         struct ide_atapi_pc *pc = drive->pc;
 162         int uptodate = pc->error ? 0 : 1;
 163
 164         debug_log("Reached %s\n", __func__);
 165
 166         if (floppy->failed_pc == pc)
 167                 floppy->failed_pc = NULL;
 168
 169         if (pc->c[0] == GPCMD_READ_10 || pc->c[0] == GPCMD_WRITE_10 ||
 170             (pc->rq && blk_pc_request(pc->rq)))
 171                 uptodate = 1; /* FIXME */
 172         else if (pc->c[0] == GPCMD_REQUEST_SENSE) {
 173                 u8 *buf = pc->buf;
 174
 175                 if (!pc->error) {
 176                         floppy->sense_key = buf[2] & 0x0F;
 177                         floppy->asc = buf[12];
 178                         floppy->ascq = buf[13];
 179                         floppy->progress_indication = buf[15] & 0x80 ?
 180                                 (u16)get_unaligned((u16 *)&buf[16]) : 0x10000;
 181
 182                         if (floppy->failed_pc)
 183                                 debug_log("pc = %x, ", floppy->failed_pc->c[0]);
 184
 185                         debug_log("sense key = %x, asc = %x, ascq = %x\n",
 186                                   floppy->sense_key, floppy->asc, floppy->ascq);
 187                 } else
 188                         printk(KERN_ERR "Error in REQUEST SENSE itself - "
 189                                         "Aborting request!\n");
 190         }
 191
 192         idefloppy_end_request(drive, uptodate, 0);
 193 }
 194
 195 static void ide_floppy_report_error(idefloppy_floppy_t *floppy,
 196                                     struct ide_atapi_pc *pc)
 197 {
 198         /* supress error messages resulting from Medium not present */
 199         if (floppy->sense_key == 0x02 &&
 200             floppy->asc       == 0x3a &&
 201             floppy->ascq      == 0x00)
 202                 return;
 203
 204         printk(KERN_ERR "ide-floppy: %s: I/O error, pc = %2x, key = %2x, "
 205                         "asc = %2x, ascq = %2x\n",
 206                         floppy->drive->name, pc->c[0], floppy->sense_key,
 207                         floppy->asc, floppy->ascq);
 208
 209 }
 210
 211 static ide_startstop_t idefloppy_issue_pc(ide_drive_t *drive,
 212                 struct ide_atapi_pc *pc)
 213 {
 214         idefloppy_floppy_t *floppy = drive->driver_data;
 215
 216         if (floppy->failed_pc == NULL &&
 217             pc->c[0] != GPCMD_REQUEST_SENSE)
 218                 floppy->failed_pc = pc;
 219
 220         /* Set the current packet command */
 221         drive->pc = pc;
 222
 223         if (pc->retries > IDEFLOPPY_MAX_PC_RETRIES) {
 224                 if (!(pc->flags & PC_FLAG_SUPPRESS_ERROR))
 225                         ide_floppy_report_error(floppy, pc);
 226                 /* Giving up */
 227                 pc->error = IDEFLOPPY_ERROR_GENERAL;
 228
 229                 floppy->failed_pc = NULL;
 230                 drive->pc_callback(drive, 0);
 231                 return ide_stopped;
 232         }
 233
 234         debug_log("Retry number - %d\n", pc->retries);
 235
 236         pc->retries++;
 237
 238         return ide_issue_pc(drive, WAIT_FLOPPY_CMD, NULL);
 239 }
 240
 241 void ide_floppy_create_read_capacity_cmd(struct ide_atapi_pc *pc)
 242 {
 243         ide_init_pc(pc);
 244         pc->c[0] = GPCMD_READ_FORMAT_CAPACITIES;
 245         pc->c[7] = 255;
 246         pc->c[8] = 255;
 247         pc->req_xfer = 255;
 248 }
 249
 250 /* A mode sense command is used to "sense" floppy parameters. */
 251 void ide_floppy_create_mode_sense_cmd(struct ide_atapi_pc *pc, u8 page_code)
 252 {
 253         u16 length = 8; /* sizeof(Mode Parameter Header) = 8 Bytes */
 254
 255         ide_init_pc(pc);
 256         pc->c[0] = GPCMD_MODE_SENSE_10;
 257         pc->c[1] = 0;
 258         pc->c[2] = page_code;
 259
 260         switch (page_code) {
 261         case IDEFLOPPY_CAPABILITIES_PAGE:
 262                 length += 12;
 263                 break;
 264         case IDEFLOPPY_FLEXIBLE_DISK_PAGE:
 265                 length += 32;
 266                 break;
 267         default:
 268                 printk(KERN_ERR "ide-floppy: unsupported page code "
 269                                 "in create_mode_sense_cmd\n");
 270         }
 271         put_unaligned(cpu_to_be16(length), (u16 *) &pc->c[7]);
 272         pc->req_xfer = length;
 273 }
 274
 275 static void idefloppy_create_rw_cmd(idefloppy_floppy_t *floppy,
 276                                     struct ide_atapi_pc *pc, struct request *rq,
 277                                     unsigned long sector)
 278 {
 279         int block = sector / floppy->bs_factor;
 280         int blocks = rq->nr_sectors / floppy->bs_factor;
 281         int cmd = rq_data_dir(rq);
 282
 283         debug_log("create_rw10_cmd: block == %d, blocks == %d\n",
 284                 block, blocks);
 285
 286         ide_init_pc(pc);
 287         pc->c[0] = cmd == READ ? GPCMD_READ_10 : GPCMD_WRITE_10;
 288         put_unaligned(cpu_to_be16(blocks), (unsigned short *)&pc->c[7]);
 289         put_unaligned(cpu_to_be32(block), (unsigned int *) &pc->c[2]);
 290
 291         memcpy(rq->cmd, pc->c, 12);
 292
 293         pc->rq = rq;
 294         pc->b_count = 0;
 295         if (rq->cmd_flags & REQ_RW)
 296                 pc->flags |= PC_FLAG_WRITING;
 297         pc->buf = NULL;
 298         pc->req_xfer = pc->buf_size = blocks * floppy->block_size;
 299         pc->flags |= PC_FLAG_DMA_OK;
 300 }
 301
 302 static void idefloppy_blockpc_cmd(idefloppy_floppy_t *floppy,
 303                 struct ide_atapi_pc *pc, struct request *rq)
 304 {
 305         ide_init_pc(pc);
 306         memcpy(pc->c, rq->cmd, sizeof(pc->c));
 307         pc->rq = rq;
 308         pc->b_count = 0;
 309         if (rq->data_len && rq_data_dir(rq) == WRITE)
 310                 pc->flags |= PC_FLAG_WRITING;
 311         pc->buf = rq->data;
 312         if (rq->bio)
 313                 pc->flags |= PC_FLAG_DMA_OK;
 314         /*
 315          * possibly problematic, doesn't look like ide-floppy correctly
 316          * handled scattered requests if dma fails...
 317          */
 318         pc->req_xfer = pc->buf_size = rq->data_len;
 319 }
 320
 321 static ide_startstop_t idefloppy_do_request(ide_drive_t *drive,
 322                 struct request *rq, sector_t block_s)
 323 {
 324         idefloppy_floppy_t *floppy = drive->driver_data;
 325         ide_hwif_t *hwif = drive->hwif;
 326         struct ide_atapi_pc *pc;
 327         unsigned long block = (unsigned long)block_s;
 328
 329         debug_log("%s: dev: %s, cmd: 0x%x, cmd_type: %x, errors: %d\n",
 330                   __func__, rq->rq_disk ? rq->rq_disk->disk_name : "?",
 331                   rq->cmd[0], rq->cmd_type, rq->errors);
 332
 333         debug_log("%s: sector: %ld, nr_sectors: %ld, current_nr_sectors: %d\n",
 334                   __func__, (long)rq->sector, rq->nr_sectors,
 335                   rq->current_nr_sectors);
 336
 337         if (rq->errors >= ERROR_MAX) {
 338                 if (floppy->failed_pc)
 339                         ide_floppy_report_error(floppy, floppy->failed_pc);
 340                 else
 341                         printk(KERN_ERR "ide-floppy: %s: I/O error\n",
 342                                 drive->name);
 343                 idefloppy_end_request(drive, 0, 0);
 344                 return ide_stopped;
 345         }
 346         if (blk_fs_request(rq)) {
 347                 if (((long)rq->sector % floppy->bs_factor) ||
 348                     (rq->nr_sectors % floppy->bs_factor)) {
 349                         printk(KERN_ERR "%s: unsupported r/w request size\n",
 350                                         drive->name);
 351                         idefloppy_end_request(drive, 0, 0);
 352                         return ide_stopped;
 353                 }
 354                 pc = &floppy->queued_pc;
 355                 idefloppy_create_rw_cmd(floppy, pc, rq, block);
 356         } else if (blk_special_request(rq)) {
 357                 pc = (struct ide_atapi_pc *) rq->buffer;
 358         } else if (blk_pc_request(rq)) {
 359                 pc = &floppy->queued_pc;
 360                 idefloppy_blockpc_cmd(floppy, pc, rq);
 361         } else {
 362                 blk_dump_rq_flags(rq,
 363                         "ide-floppy: unsupported command in queue");
 364                 idefloppy_end_request(drive, 0, 0);
 365                 return ide_stopped;
 366         }
 367
 368         ide_init_sg_cmd(drive, rq);
 369         ide_map_sg(drive, rq);
 370
 371         pc->sg = hwif->sg_table;
 372         pc->sg_cnt = hwif->sg_nents;
 373
 374         pc->rq = rq;
 375
 376         return idefloppy_issue_pc(drive, pc);
 377 }
 378
 379 /*
 380  * Look at the flexible disk page parameters. We ignore the CHS capacity
 381  * parameters and use the LBA parameters instead.
 382  */
 383 static int ide_floppy_get_flexible_disk_page(ide_drive_t *drive)
 384 {
 385         idefloppy_floppy_t *floppy = drive->driver_data;
 386         struct gendisk *disk = floppy->disk;
 387         struct ide_atapi_pc pc;
 388         u8 *page;
 389         int capacity, lba_capacity;
 390         u16 transfer_rate, sector_size, cyls, rpm;
 391         u8 heads, sectors;
 392
 393         ide_floppy_create_mode_sense_cmd(&pc, IDEFLOPPY_FLEXIBLE_DISK_PAGE);
 394
 395         if (ide_queue_pc_tail(drive, disk, &pc)) {
 396                 printk(KERN_ERR "ide-floppy: Can't get flexible disk page"
 397                                 " parameters\n");
 398                 return 1;
 399         }
 400
 401         if (pc.buf[3] & 0x80)
 402                 drive->atapi_flags |= IDE_AFLAG_WP;
 403         else
 404                 drive->atapi_flags &= ~IDE_AFLAG_WP;
 405
 406         set_disk_ro(disk, !!(drive->atapi_flags & IDE_AFLAG_WP));
 407
 408         page = &pc.buf[8];
 409
 410         transfer_rate = be16_to_cpup((__be16 *)&pc.buf[8 + 2]);
 411         sector_size   = be16_to_cpup((__be16 *)&pc.buf[8 + 6]);
 412         cyls          = be16_to_cpup((__be16 *)&pc.buf[8 + 8]);
 413         rpm           = be16_to_cpup((__be16 *)&pc.buf[8 + 28]);
 414         heads         = pc.buf[8 + 4];
 415         sectors       = pc.buf[8 + 5];
 416
 417         capacity = cyls * heads * sectors * sector_size;
 418
 419         if (memcmp(page, &floppy->flexible_disk_page, 32))
 420                 printk(KERN_INFO "%s: %dkB, %d/%d/%d CHS, %d kBps, "
 421                                 "%d sector size, %d rpm\n",
 422                                 drive->name, capacity / 1024, cyls, heads,
 423                                 sectors, transfer_rate / 8, sector_size, rpm);
 424
 425         memcpy(&floppy->flexible_disk_page, page, 32);
 426         drive->bios_cyl = cyls;
 427         drive->bios_head = heads;
 428         drive->bios_sect = sectors;
 429         lba_capacity = floppy->blocks * floppy->block_size;
 430
 431         if (capacity < lba_capacity) {
 432                 printk(KERN_NOTICE "%s: The disk reports a capacity of %d "
 433                         "bytes, but the drive only handles %d\n",
 434                         drive->name, lba_capacity, capacity);
 435                 floppy->blocks = floppy->block_size ?
 436                         capacity / floppy->block_size : 0;
 437         }
 438         return 0;
 439 }
 440
 441 /*
 442  * Determine if a media is present in the floppy drive, and if so, its LBA
 443  * capacity.
 444  */
 445 static int ide_floppy_get_capacity(ide_drive_t *drive)
 446 {
 447         idefloppy_floppy_t *floppy = drive->driver_data;
 448         struct gendisk *disk = floppy->disk;
 449         struct ide_atapi_pc pc;
 450         u8 *cap_desc;
 451         u8 header_len, desc_cnt;
 452         int i, rc = 1, blocks, length;
 453
 454         drive->bios_cyl = 0;
 455         drive->bios_head = drive->bios_sect = 0;
 456         floppy->blocks = 0;
 457         floppy->bs_factor = 1;
 458         set_capacity(floppy->disk, 0);
 459
 460         ide_floppy_create_read_capacity_cmd(&pc);
 461         if (ide_queue_pc_tail(drive, disk, &pc)) {
 462                 printk(KERN_ERR "ide-floppy: Can't get floppy parameters\n");
 463                 return 1;
 464         }
 465         header_len = pc.buf[3];
 466         cap_desc = &pc.buf[4];
 467         desc_cnt = header_len / 8; /* capacity descriptor of 8 bytes */
 468
 469         for (i = 0; i < desc_cnt; i++) {
 470                 unsigned int desc_start = 4 + i*8;
 471
 472                 blocks = be32_to_cpup((__be32 *)&pc.buf[desc_start]);
 473                 length = be16_to_cpup((__be16 *)&pc.buf[desc_start + 6]);
 474
 475                 debug_log("Descriptor %d: %dkB, %d blocks, %d sector size\n",
 476                                 i, blocks * length / 1024, blocks, length);
 477
 478                 if (i)
 479                         continue;
 480                 /*
 481                  * the code below is valid only for the 1st descriptor, ie i=0
 482                  */
 483
 484                 switch (pc.buf[desc_start + 4] & 0x03) {
 485                 /* Clik! drive returns this instead of CAPACITY_CURRENT */
 486                 case CAPACITY_UNFORMATTED:
 487                         if (!(drive->atapi_flags & IDE_AFLAG_CLIK_DRIVE))
 488                                 /*
 489                                  * If it is not a clik drive, break out
 490                                  * (maintains previous driver behaviour)
 491                                  */
 492                                 break;
 493                 case CAPACITY_CURRENT:
 494                         /* Normal Zip/LS-120 disks */
 495                         if (memcmp(cap_desc, &floppy->cap_desc, 8))
 496                                 printk(KERN_INFO "%s: %dkB, %d blocks, %d "
 497                                         "sector size\n", drive->name,
 498                                         blocks * length / 1024, blocks, length);
 499                         memcpy(&floppy->cap_desc, cap_desc, 8);
 500
 501                         if (!length || length % 512) {
 502                                 printk(KERN_NOTICE "%s: %d bytes block size "
 503                                         "not supported\n", drive->name, length);
 504                         } else {
 505                                 floppy->blocks = blocks;
 506                                 floppy->block_size = length;
 507                                 floppy->bs_factor = length / 512;
 508                                 if (floppy->bs_factor != 1)
 509                                         printk(KERN_NOTICE "%s: warning: non "
 510                                                 "512 bytes block size not "
 511                                                 "fully supported\n",
 512                                                 drive->name);
 513                                 rc = 0;
 514                         }
 515                         break;
 516                 case CAPACITY_NO_CARTRIDGE:
 517                         /*
 518                          * This is a KERN_ERR so it appears on screen
 519                          * for the user to see
 520                          */
 521                         printk(KERN_ERR "%s: No disk in drive\n", drive->name);
 522                         break;
 523                 case CAPACITY_INVALID:
 524                         printk(KERN_ERR "%s: Invalid capacity for disk "
 525                                 "in drive\n", drive->name);
 526                         break;
 527                 }
 528                 debug_log("Descriptor 0 Code: %d\n",
 529                           pc.buf[desc_start + 4] & 0x03);
 530         }
 531
 532         /* Clik! disk does not support get_flexible_disk_page */
 533         if (!(drive->atapi_flags & IDE_AFLAG_CLIK_DRIVE))
 534                 (void) ide_floppy_get_flexible_disk_page(drive);
 535
 536         set_capacity(disk, floppy->blocks * floppy->bs_factor);
 537
 538         return rc;
 539 }
 540
 541 static sector_t idefloppy_capacity(ide_drive_t *drive)
 542 {
 543         idefloppy_floppy_t *floppy = drive->driver_data;
 544         unsigned long capacity = floppy->blocks * floppy->bs_factor;
 545
 546         return capacity;
 547 }
 548
 549 #ifdef CONFIG_IDE_PROC_FS
 550 ide_devset_rw_field(bios_cyl, bios_cyl);
 551 ide_devset_rw_field(bios_head, bios_head);
 552 ide_devset_rw_field(bios_sect, bios_sect);
 553 ide_devset_rw_field(ticks, pc_delay);
 554
 555 static const struct ide_proc_devset idefloppy_settings[] = {
 556         IDE_PROC_DEVSET(bios_cyl,  0, 1023),
 557         IDE_PROC_DEVSET(bios_head, 0,  255),
 558         IDE_PROC_DEVSET(bios_sect, 0,   63),
 559         IDE_PROC_DEVSET(ticks,     0,  255),
 560         { 0 },
 561 };
 562 #endif
 563
 564 static void idefloppy_setup(ide_drive_t *drive, idefloppy_floppy_t *floppy)
 565 {
 566         u16 *id = drive->id;
 567         u8 gcw[2];
 568
 569         *((u16 *)&gcw) = id[ATA_ID_CONFIG];
 570
 571         drive->pc_callback       = ide_floppy_callback;
 572         drive->pc_update_buffers = idefloppy_update_buffers;
 573         drive->pc_io_buffers     = ide_io_buffers;
 574
 575         if (((gcw[0] & 0x60) >> 5) == 1)
 576                 drive->atapi_flags |= IDE_AFLAG_DRQ_INTERRUPT;
 577         /*
 578          * We used to check revisions here. At this point however I'm giving up.
 579          * Just assume they are all broken, its easier.
 580          *
 581          * The actual reason for the workarounds was likely a driver bug after
 582          * all rather than a firmware bug, and the workaround below used to hide
 583          * it. It should be fixed as of version 1.9, but to be on the safe side
 584          * we'll leave the limitation below for the 2.2.x tree.
 585          */
 586         if (!strncmp((char *)&id[ATA_ID_PROD], "IOMEGA ZIP 100 ATAPI", 20)) {
 587                 drive->atapi_flags |= IDE_AFLAG_ZIP_DRIVE;
 588                 /* This value will be visible in the /proc/ide/hdx/settings */
 589                 drive->pc_delay = IDEFLOPPY_PC_DELAY;
 590                 blk_queue_max_sectors(drive->queue, 64);
 591         }
 592
 593         /*
 594          * Guess what? The IOMEGA Clik! drive also needs the above fix. It makes
 595          * nasty clicking noises without it, so please don't remove this.
 596          */
 597         if (strncmp((char *)&id[ATA_ID_PROD], "IOMEGA Clik!", 11) == 0) {
 598                 blk_queue_max_sectors(drive->queue, 64);
 599                 drive->atapi_flags |= IDE_AFLAG_CLIK_DRIVE;
 600                 /* IOMEGA Clik! drives do not support lock/unlock commands */
 601                 drive->atapi_flags |= IDE_AFLAG_NO_DOORLOCK;
 602         }
 603
 604         (void) ide_floppy_get_capacity(drive);
 605
 606         ide_proc_register_driver(drive, floppy->driver);
 607 }
 608
 609 static void ide_floppy_remove(ide_drive_t *drive)
 610 {
 611         idefloppy_floppy_t *floppy = drive->driver_data;
 612         struct gendisk *g = floppy->disk;
 613
 614         ide_proc_unregister_driver(drive, floppy->driver);
 615
 616         del_gendisk(g);
 617
 618         ide_floppy_put(floppy);
 619 }
 620
 621 static void idefloppy_cleanup_obj(struct kref *kref)
 622 {
 623         struct ide_floppy_obj *floppy = to_ide_drv(kref, ide_floppy_obj);
 624         ide_drive_t *drive = floppy->drive;
 625         struct gendisk *g = floppy->disk;
 626
 627         drive->driver_data = NULL;
 628         g->private_data = NULL;
 629         put_disk(g);
 630         kfree(floppy);
 631 }
 632
 633 #ifdef CONFIG_IDE_PROC_FS
 634 static int proc_idefloppy_read_capacity(char *page, char **start, off_t off,
 635                 int count, int *eof, void *data)
 636 {
 637         ide_drive_t*drive = (ide_drive_t *)data;
 638         int len;
 639
 640         len = sprintf(page, "%llu\n", (long long)idefloppy_capacity(drive));
 641         PROC_IDE_READ_RETURN(page, start, off, count, eof, len);
 642 }
 643
 644 static ide_proc_entry_t idefloppy_proc[] = {
 645         { "capacity",   S_IFREG|S_IRUGO, proc_idefloppy_read_capacity,  NULL },
 646         { "geometry",   S_IFREG|S_IRUGO, proc_ide_read_geometry,        NULL },
 647         { NULL, 0, NULL, NULL }
 648 };
 649 #endif  /* CONFIG_IDE_PROC_FS */
 650
 651 static int ide_floppy_probe(ide_drive_t *);
 652
 653 static ide_driver_t idefloppy_driver = {
 654         .gen_driver = {
 655                 .owner          = THIS_MODULE,
 656                 .name           = "ide-floppy",
 657                 .bus            = &ide_bus_type,
 658         },
 659         .probe                  = ide_floppy_probe,
 660         .remove                 = ide_floppy_remove,
 661         .version                = IDEFLOPPY_VERSION,
 662         .media                  = ide_floppy,
 663         .do_request             = idefloppy_do_request,
 664         .end_request            = idefloppy_end_request,
 665         .error                  = __ide_error,
 666 #ifdef CONFIG_IDE_PROC_FS
 667         .proc                   = idefloppy_proc,
 668         .settings               = idefloppy_settings,
 669 #endif
 670 };
 671
 672 static int idefloppy_open(struct inode *inode, struct file *filp)
 673 {
 674         struct gendisk *disk = inode->i_bdev->bd_disk;
 675         struct ide_floppy_obj *floppy;
 676         ide_drive_t *drive;
 677         int ret = 0;
 678
 679         debug_log("Reached %s\n", __func__);
 680
 681         floppy = ide_floppy_get(disk);
 682         if (!floppy)
 683                 return -ENXIO;
 684
 685         drive = floppy->drive;
 686
 687         floppy->openers++;
 688
 689         if (floppy->openers == 1) {
 690                 drive->atapi_flags &= ~IDE_AFLAG_FORMAT_IN_PROGRESS;
 691                 /* Just in case */
 692
 693                 if (ide_do_test_unit_ready(drive, disk))
 694                         ide_do_start_stop(drive, disk, 1);
 695
 696                 if (ide_floppy_get_capacity(drive)
 697                    && (filp->f_flags & O_NDELAY) == 0
 698                     /*
 699                      * Allow O_NDELAY to open a drive without a disk, or with an
 700                      * unreadable disk, so that we can get the format capacity
 701                      * of the drive or begin the format - Sam
 702                      */
 703                     ) {
 704                         ret = -EIO;
 705                         goto out_put_floppy;
 706                 }
 707
 708                 if ((drive->atapi_flags & IDE_AFLAG_WP) && (filp->f_mode & 2)) {
 709                         ret = -EROFS;
 710                         goto out_put_floppy;
 711                 }
 712
 713                 drive->atapi_flags |= IDE_AFLAG_MEDIA_CHANGED;
 714                 ide_set_media_lock(drive, disk, 1);
 715                 check_disk_change(inode->i_bdev);
 716         } else if (drive->atapi_flags & IDE_AFLAG_FORMAT_IN_PROGRESS) {
 717                 ret = -EBUSY;
 718                 goto out_put_floppy;
 719         }
 720         return 0;
 721
 722 out_put_floppy:
 723         floppy->openers--;
 724         ide_floppy_put(floppy);
 725         return ret;
 726 }
 727
 728 static int idefloppy_release(struct inode *inode, struct file *filp)
 729 {
 730         struct gendisk *disk = inode->i_bdev->bd_disk;
 731         struct ide_floppy_obj *floppy = ide_drv_g(disk, ide_floppy_obj);
 732         ide_drive_t *drive = floppy->drive;
 733
 734         debug_log("Reached %s\n", __func__);
 735
 736         if (floppy->openers == 1) {
 737                 ide_set_media_lock(drive, disk, 0);
 738                 drive->atapi_flags &= ~IDE_AFLAG_FORMAT_IN_PROGRESS;
 739         }
 740
 741         floppy->openers--;
 742
 743         ide_floppy_put(floppy);
 744
 745         return 0;
 746 }
 747
 748 static int idefloppy_getgeo(struct block_device *bdev, struct hd_geometry *geo)
 749 {
 750         struct ide_floppy_obj *floppy = ide_drv_g(bdev->bd_disk,
 751                                                      ide_floppy_obj);
 752         ide_drive_t *drive = floppy->drive;
 753
 754         geo->heads = drive->bios_head;
 755         geo->sectors = drive->bios_sect;
 756         geo->cylinders = (u16)drive->bios_cyl; /* truncate */
 757         return 0;
 758 }
 759
 760 static int ide_floppy_lockdoor(ide_drive_t *drive, struct ide_atapi_pc *pc,
 761                                unsigned long arg, unsigned int cmd)
 762 {
 763         idefloppy_floppy_t *floppy = drive->driver_data;
 764         struct gendisk *disk = floppy->disk;
 765         int prevent = (arg && cmd != CDROMEJECT) ? 1 : 0;
 766
 767         if (floppy->openers > 1)
 768                 return -EBUSY;
 769
 770         ide_set_media_lock(drive, disk, prevent);
 771
 772         if (cmd == CDROMEJECT)
 773                 ide_do_start_stop(drive, disk, 2);
 774
 775         return 0;
 776 }
 777
 778 static int idefloppy_ioctl(struct inode *inode, struct file *file,
 779                         unsigned int cmd, unsigned long arg)
 780 {
 781         struct block_device *bdev = inode->i_bdev;
 782         struct ide_floppy_obj *floppy = ide_drv_g(bdev->bd_disk,
 783                                                      ide_floppy_obj);
 784         ide_drive_t *drive = floppy->drive;
 785         struct ide_atapi_pc pc;
 786         void __user *argp = (void __user *)arg;
 787         int err;
 788
 789         if (cmd == CDROMEJECT || cmd == CDROM_LOCKDOOR)
 790                 return ide_floppy_lockdoor(drive, &pc, arg, cmd);
 791
 792         err = ide_floppy_format_ioctl(drive, file, cmd, argp);
 793         if (err != -ENOTTY)
 794                 return err;
 795
 796         /*
 797          * skip SCSI_IOCTL_SEND_COMMAND (deprecated)
 798          * and CDROM_SEND_PACKET (legacy) ioctls
 799          */
 800         if (cmd != CDROM_SEND_PACKET && cmd != SCSI_IOCTL_SEND_COMMAND)
 801                 err = scsi_cmd_ioctl(file, bdev->bd_disk->queue,
 802                                         bdev->bd_disk, cmd, argp);
 803
 804         if (err == -ENOTTY)
 805                 err = generic_ide_ioctl(drive, file, bdev, cmd, arg);
 806
 807         return err;
 808 }
 809
 810 static int idefloppy_media_changed(struct gendisk *disk)
 811 {
 812         struct ide_floppy_obj *floppy = ide_drv_g(disk, ide_floppy_obj);
 813         ide_drive_t *drive = floppy->drive;
 814         int ret;
 815
 816         /* do not scan partitions twice if this is a removable device */
 817         if (drive->attach) {
 818                 drive->attach = 0;
 819                 return 0;
 820         }
 821         ret = !!(drive->atapi_flags & IDE_AFLAG_MEDIA_CHANGED);
 822         drive->atapi_flags &= ~IDE_AFLAG_MEDIA_CHANGED;
 823         return ret;
 824 }
 825
 826 static int idefloppy_revalidate_disk(struct gendisk *disk)
 827 {
 828         struct ide_floppy_obj *floppy = ide_drv_g(disk, ide_floppy_obj);
 829         set_capacity(disk, idefloppy_capacity(floppy->drive));
 830         return 0;
 831 }
 832
 833 static struct block_device_operations idefloppy_ops = {
 834         .owner                  = THIS_MODULE,
 835         .open                   = idefloppy_open,
 836         .release                = idefloppy_release,
 837         .ioctl                  = idefloppy_ioctl,
 838         .getgeo                 = idefloppy_getgeo,
 839         .media_changed          = idefloppy_media_changed,
 840         .revalidate_disk        = idefloppy_revalidate_disk
 841 };
 842
 843 static int ide_floppy_probe(ide_drive_t *drive)
 844 {
 845         idefloppy_floppy_t *floppy;
 846         struct gendisk *g;
 847
 848         if (!strstr("ide-floppy", drive->driver_req))
 849                 goto failed;
 850
 851         if (drive->media != ide_floppy)
 852                 goto failed;
 853
 854         if (!ide_check_atapi_device(drive, DRV_NAME)) {
 855                 printk(KERN_ERR "ide-floppy: %s: not supported by this version"
 856                                 " of ide-floppy\n", drive->name);
 857                 goto failed;
 858         }
 859         floppy = kzalloc(sizeof(idefloppy_floppy_t), GFP_KERNEL);
 860         if (!floppy) {
 861                 printk(KERN_ERR "ide-floppy: %s: Can't allocate a floppy"
 862                                 " structure\n", drive->name);
 863                 goto failed;
 864         }
 865
 866         g = alloc_disk(1 << PARTN_BITS);
 867         if (!g)
 868                 goto out_free_floppy;
 869
 870         ide_init_disk(g, drive);
 871
 872         kref_init(&floppy->kref);
 873
 874         floppy->drive = drive;
 875         floppy->driver = &idefloppy_driver;
 876         floppy->disk = g;
 877
 878         g->private_data = &floppy->driver;
 879
 880         drive->driver_data = floppy;
 881
 882         idefloppy_setup(drive, floppy);
 883
 884         g->minors = 1 << PARTN_BITS;
 885         g->driverfs_dev = &drive->gendev;
 886         g->flags = drive->removable ? GENHD_FL_REMOVABLE : 0;
 887         g->fops = &idefloppy_ops;
 888         drive->attach = 1;
 889         add_disk(g);
 890         return 0;
 891
 892 out_free_floppy:
 893         kfree(floppy);
 894 failed:
 895         return -ENODEV;
 896 }
 897
 898 static void __exit idefloppy_exit(void)
 899 {
 900         driver_unregister(&idefloppy_driver.gen_driver);
 901 }
 902
 903 static int __init idefloppy_init(void)
 904 {
 905         printk("ide-floppy driver " IDEFLOPPY_VERSION "\n");
 906         return driver_register(&idefloppy_driver.gen_driver);
 907 }
 908
 909 MODULE_ALIAS("ide:*m-floppy*");
 910 MODULE_ALIAS("ide-floppy");
 911 module_init(idefloppy_init);
 912 module_exit(idefloppy_exit);
 913 MODULE_LICENSE("GPL");
 914 MODULE_DESCRIPTION("ATAPI FLOPPY Driver");
 915