Merge git://git.kernel.org/pub/scm/linux/kernel/git/sfrench/cifs-2.6
[linux-2.6] / drivers / ide / ide-disk.c
1 /*
2  *  linux/drivers/ide/ide-disk.c        Version 1.18    Mar 05, 2003
3  *
4  *  Copyright (C) 1994-1998  Linus Torvalds & authors (see below)
5  *  Copyright (C) 1998-2002  Linux ATA Development
6  *                              Andre Hedrick <andre@linux-ide.org>
7  *  Copyright (C) 2003       Red Hat <alan@redhat.com>
8  */
9
10 /*
11  *  Mostly written by Mark Lord <mlord@pobox.com>
12  *                and Gadi Oxman <gadio@netvision.net.il>
13  *                and Andre Hedrick <andre@linux-ide.org>
14  *
15  * This is the IDE/ATA disk driver, as evolved from hd.c and ide.c.
16  */
17
18 #define IDEDISK_VERSION "1.18"
19
20 //#define DEBUG
21
22 #include <linux/module.h>
23 #include <linux/types.h>
24 #include <linux/string.h>
25 #include <linux/kernel.h>
26 #include <linux/timer.h>
27 #include <linux/mm.h>
28 #include <linux/interrupt.h>
29 #include <linux/major.h>
30 #include <linux/errno.h>
31 #include <linux/genhd.h>
32 #include <linux/slab.h>
33 #include <linux/delay.h>
34 #include <linux/mutex.h>
35 #include <linux/leds.h>
36
37 #define _IDE_DISK
38
39 #include <linux/ide.h>
40
41 #include <asm/byteorder.h>
42 #include <asm/irq.h>
43 #include <asm/uaccess.h>
44 #include <asm/io.h>
45 #include <asm/div64.h>
46
47 struct ide_disk_obj {
48         ide_drive_t     *drive;
49         ide_driver_t    *driver;
50         struct gendisk  *disk;
51         struct kref     kref;
52         unsigned int    openers;        /* protected by BKL for now */
53 };
54
55 static DEFINE_MUTEX(idedisk_ref_mutex);
56
57 #define to_ide_disk(obj) container_of(obj, struct ide_disk_obj, kref)
58
59 #define ide_disk_g(disk) \
60         container_of((disk)->private_data, struct ide_disk_obj, driver)
61
62 static struct ide_disk_obj *ide_disk_get(struct gendisk *disk)
63 {
64         struct ide_disk_obj *idkp = NULL;
65
66         mutex_lock(&idedisk_ref_mutex);
67         idkp = ide_disk_g(disk);
68         if (idkp)
69                 kref_get(&idkp->kref);
70         mutex_unlock(&idedisk_ref_mutex);
71         return idkp;
72 }
73
74 static void ide_disk_release(struct kref *);
75
76 static void ide_disk_put(struct ide_disk_obj *idkp)
77 {
78         mutex_lock(&idedisk_ref_mutex);
79         kref_put(&idkp->kref, ide_disk_release);
80         mutex_unlock(&idedisk_ref_mutex);
81 }
82
83 /*
84  * lba_capacity_is_ok() performs a sanity check on the claimed "lba_capacity"
85  * value for this drive (from its reported identification information).
86  *
87  * Returns:     1 if lba_capacity looks sensible
88  *              0 otherwise
89  *
90  * It is called only once for each drive.
91  */
92 static int lba_capacity_is_ok (struct hd_driveid *id)
93 {
94         unsigned long lba_sects, chs_sects, head, tail;
95
96         /* No non-LBA info .. so valid! */
97         if (id->cyls == 0)
98                 return 1;
99
100         /*
101          * The ATA spec tells large drives to return
102          * C/H/S = 16383/16/63 independent of their size.
103          * Some drives can be jumpered to use 15 heads instead of 16.
104          * Some drives can be jumpered to use 4092 cyls instead of 16383.
105          */
106         if ((id->cyls == 16383
107              || (id->cyls == 4092 && id->cur_cyls == 16383)) &&
108             id->sectors == 63 &&
109             (id->heads == 15 || id->heads == 16) &&
110             (id->lba_capacity >= 16383*63*id->heads))
111                 return 1;
112
113         lba_sects   = id->lba_capacity;
114         chs_sects   = id->cyls * id->heads * id->sectors;
115
116         /* perform a rough sanity check on lba_sects:  within 10% is OK */
117         if ((lba_sects - chs_sects) < chs_sects/10)
118                 return 1;
119
120         /* some drives have the word order reversed */
121         head = ((lba_sects >> 16) & 0xffff);
122         tail = (lba_sects & 0xffff);
123         lba_sects = (head | (tail << 16));
124         if ((lba_sects - chs_sects) < chs_sects/10) {
125                 id->lba_capacity = lba_sects;
126                 return 1;       /* lba_capacity is (now) good */
127         }
128
129         return 0;       /* lba_capacity value may be bad */
130 }
131
132 static const u8 ide_rw_cmds[] = {
133         WIN_MULTREAD,
134         WIN_MULTWRITE,
135         WIN_MULTREAD_EXT,
136         WIN_MULTWRITE_EXT,
137         WIN_READ,
138         WIN_WRITE,
139         WIN_READ_EXT,
140         WIN_WRITE_EXT,
141         WIN_READDMA,
142         WIN_WRITEDMA,
143         WIN_READDMA_EXT,
144         WIN_WRITEDMA_EXT,
145 };
146
147 static const u8 ide_data_phases[] = {
148         TASKFILE_MULTI_IN,
149         TASKFILE_MULTI_OUT,
150         TASKFILE_IN,
151         TASKFILE_OUT,
152         TASKFILE_IN_DMA,
153         TASKFILE_OUT_DMA,
154 };
155
156 static void ide_tf_set_cmd(ide_drive_t *drive, ide_task_t *task, u8 dma)
157 {
158         u8 index, lba48, write;
159
160         lba48 = (task->tf_flags & IDE_TFLAG_LBA48) ? 2 : 0;
161         write = (task->tf_flags & IDE_TFLAG_WRITE) ? 1 : 0;
162
163         if (dma)
164                 index = drive->vdma ? 4 : 8;
165         else
166                 index = drive->mult_count ? 0 : 4;
167
168         task->tf.command = ide_rw_cmds[index + lba48 + write];
169
170         if (dma)
171                 index = 8; /* fixup index */
172
173         task->data_phase = ide_data_phases[index / 2 + write];
174 }
175
176 /*
177  * __ide_do_rw_disk() issues READ and WRITE commands to a disk,
178  * using LBA if supported, or CHS otherwise, to address sectors.
179  */
180 static ide_startstop_t __ide_do_rw_disk(ide_drive_t *drive, struct request *rq, sector_t block)
181 {
182         ide_hwif_t *hwif        = HWIF(drive);
183         unsigned int dma        = drive->using_dma;
184         u16 nsectors            = (u16)rq->nr_sectors;
185         u8 lba48                = (drive->addressing == 1) ? 1 : 0;
186         ide_task_t              task;
187         struct ide_taskfile     *tf = &task.tf;
188         ide_startstop_t         rc;
189
190         if ((hwif->host_flags & IDE_HFLAG_NO_LBA48_DMA) && lba48 && dma) {
191                 if (block + rq->nr_sectors > 1ULL << 28)
192                         dma = 0;
193                 else
194                         lba48 = 0;
195         }
196
197         if (!dma) {
198                 ide_init_sg_cmd(drive, rq);
199                 ide_map_sg(drive, rq);
200         }
201
202         memset(&task, 0, sizeof(task));
203         task.tf_flags = IDE_TFLAG_NO_SELECT_MASK;  /* FIXME? */
204         task.tf_flags |= (IDE_TFLAG_TF | IDE_TFLAG_DEVICE);
205
206         if (drive->select.b.lba) {
207                 if (lba48) {
208                         pr_debug("%s: LBA=0x%012llx\n", drive->name,
209                                         (unsigned long long)block);
210
211                         tf->hob_nsect = (nsectors >> 8) & 0xff;
212                         tf->hob_lbal  = (u8)(block >> 24);
213                         if (sizeof(block) != 4) {
214                                 tf->hob_lbam = (u8)((u64)block >> 32);
215                                 tf->hob_lbah = (u8)((u64)block >> 40);
216                         }
217
218                         tf->nsect  = nsectors & 0xff;
219                         tf->lbal   = (u8) block;
220                         tf->lbam   = (u8)(block >>  8);
221                         tf->lbah   = (u8)(block >> 16);
222
223                         task.tf_flags |= (IDE_TFLAG_LBA48 | IDE_TFLAG_HOB);
224                 } else {
225                         tf->nsect  = nsectors & 0xff;
226                         tf->lbal   = block;
227                         tf->lbam   = block >>= 8;
228                         tf->lbah   = block >>= 8;
229                         tf->device = (block >> 8) & 0xf;
230                 }
231         } else {
232                 unsigned int sect,head,cyl,track;
233                 track = (int)block / drive->sect;
234                 sect  = (int)block % drive->sect + 1;
235                 head  = track % drive->head;
236                 cyl   = track / drive->head;
237
238                 pr_debug("%s: CHS=%u/%u/%u\n", drive->name, cyl, head, sect);
239
240                 tf->nsect  = nsectors & 0xff;
241                 tf->lbal   = sect;
242                 tf->lbam   = cyl;
243                 tf->lbah   = cyl >> 8;
244                 tf->device = head;
245         }
246
247         if (rq_data_dir(rq))
248                 task.tf_flags |= IDE_TFLAG_WRITE;
249
250         ide_tf_set_cmd(drive, &task, dma);
251         if (!dma)
252                 hwif->data_phase = task.data_phase;
253         task.rq = rq;
254
255         rc = do_rw_taskfile(drive, &task);
256
257         if (rc == ide_stopped && dma) {
258                 /* fallback to PIO */
259                 task.tf_flags |= IDE_TFLAG_DMA_PIO_FALLBACK;
260                 ide_tf_set_cmd(drive, &task, 0);
261                 hwif->data_phase = task.data_phase;
262                 ide_init_sg_cmd(drive, rq);
263                 rc = do_rw_taskfile(drive, &task);
264         }
265
266         return rc;
267 }
268
269 /*
270  * 268435455  == 137439 MB or 28bit limit
271  * 320173056  == 163929 MB or 48bit addressing
272  * 1073741822 == 549756 MB or 48bit addressing fake drive
273  */
274
275 static ide_startstop_t ide_do_rw_disk (ide_drive_t *drive, struct request *rq, sector_t block)
276 {
277         ide_hwif_t *hwif = HWIF(drive);
278
279         BUG_ON(drive->blocked);
280
281         if (!blk_fs_request(rq)) {
282                 blk_dump_rq_flags(rq, "ide_do_rw_disk - bad command");
283                 ide_end_request(drive, 0, 0);
284                 return ide_stopped;
285         }
286
287         ledtrig_ide_activity();
288
289         pr_debug("%s: %sing: block=%llu, sectors=%lu, buffer=0x%08lx\n",
290                  drive->name, rq_data_dir(rq) == READ ? "read" : "writ",
291                  (unsigned long long)block, rq->nr_sectors,
292                  (unsigned long)rq->buffer);
293
294         if (hwif->rw_disk)
295                 hwif->rw_disk(drive, rq);
296
297         return __ide_do_rw_disk(drive, rq, block);
298 }
299
300 /*
301  * Queries for true maximum capacity of the drive.
302  * Returns maximum LBA address (> 0) of the drive, 0 if failed.
303  */
304 static u64 idedisk_read_native_max_address(ide_drive_t *drive, int lba48)
305 {
306         ide_task_t args;
307         struct ide_taskfile *tf = &args.tf;
308         u64 addr = 0;
309
310         /* Create IDE/ATA command request structure */
311         memset(&args, 0, sizeof(ide_task_t));
312         if (lba48)
313                 tf->command = WIN_READ_NATIVE_MAX_EXT;
314         else
315                 tf->command = WIN_READ_NATIVE_MAX;
316         tf->device  = ATA_LBA;
317         args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
318         if (lba48)
319                 args.tf_flags |= (IDE_TFLAG_LBA48 | IDE_TFLAG_HOB);
320         /* submit command request */
321         ide_no_data_taskfile(drive, &args);
322
323         /* if OK, compute maximum address value */
324         if ((tf->status & 0x01) == 0)
325                 addr = ide_get_lba_addr(tf, lba48) + 1;
326
327         return addr;
328 }
329
330 /*
331  * Sets maximum virtual LBA address of the drive.
332  * Returns new maximum virtual LBA address (> 0) or 0 on failure.
333  */
334 static u64 idedisk_set_max_address(ide_drive_t *drive, u64 addr_req, int lba48)
335 {
336         ide_task_t args;
337         struct ide_taskfile *tf = &args.tf;
338         u64 addr_set = 0;
339
340         addr_req--;
341         /* Create IDE/ATA command request structure */
342         memset(&args, 0, sizeof(ide_task_t));
343         tf->lbal     = (addr_req >>  0) & 0xff;
344         tf->lbam     = (addr_req >>= 8) & 0xff;
345         tf->lbah     = (addr_req >>= 8) & 0xff;
346         if (lba48) {
347                 tf->hob_lbal = (addr_req >>= 8) & 0xff;
348                 tf->hob_lbam = (addr_req >>= 8) & 0xff;
349                 tf->hob_lbah = (addr_req >>= 8) & 0xff;
350                 tf->command  = WIN_SET_MAX_EXT;
351         } else {
352                 tf->device   = (addr_req >>= 8) & 0x0f;
353                 tf->command  = WIN_SET_MAX;
354         }
355         tf->device |= ATA_LBA;
356         args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
357         if (lba48)
358                 args.tf_flags |= (IDE_TFLAG_LBA48 | IDE_TFLAG_HOB);
359         /* submit command request */
360         ide_no_data_taskfile(drive, &args);
361         /* if OK, compute maximum address value */
362         if ((tf->status & 0x01) == 0)
363                 addr_set = ide_get_lba_addr(tf, lba48) + 1;
364
365         return addr_set;
366 }
367
368 static unsigned long long sectors_to_MB(unsigned long long n)
369 {
370         n <<= 9;                /* make it bytes */
371         do_div(n, 1000000);     /* make it MB */
372         return n;
373 }
374
375 /*
376  * Bits 10 of command_set_1 and cfs_enable_1 must be equal,
377  * so on non-buggy drives we need test only one.
378  * However, we should also check whether these fields are valid.
379  */
380 static inline int idedisk_supports_hpa(const struct hd_driveid *id)
381 {
382         return (id->command_set_1 & 0x0400) && (id->cfs_enable_1 & 0x0400);
383 }
384
385 /*
386  * The same here.
387  */
388 static inline int idedisk_supports_lba48(const struct hd_driveid *id)
389 {
390         return (id->command_set_2 & 0x0400) && (id->cfs_enable_2 & 0x0400)
391                && id->lba_capacity_2;
392 }
393
394 /*
395  * Some disks report total number of sectors instead of
396  * maximum sector address.  We list them here.
397  */
398 static const struct drive_list_entry hpa_list[] = {
399         { "ST340823A",  NULL },
400         { "ST320413A",  NULL },
401         { NULL,         NULL }
402 };
403
404 static void idedisk_check_hpa(ide_drive_t *drive)
405 {
406         unsigned long long capacity, set_max;
407         int lba48 = idedisk_supports_lba48(drive->id);
408
409         capacity = drive->capacity64;
410
411         set_max = idedisk_read_native_max_address(drive, lba48);
412
413         if (ide_in_drive_list(drive->id, hpa_list)) {
414                 /*
415                  * Since we are inclusive wrt to firmware revisions do this
416                  * extra check and apply the workaround only when needed.
417                  */
418                 if (set_max == capacity + 1)
419                         set_max--;
420         }
421
422         if (set_max <= capacity)
423                 return;
424
425         printk(KERN_INFO "%s: Host Protected Area detected.\n"
426                          "\tcurrent capacity is %llu sectors (%llu MB)\n"
427                          "\tnative  capacity is %llu sectors (%llu MB)\n",
428                          drive->name,
429                          capacity, sectors_to_MB(capacity),
430                          set_max, sectors_to_MB(set_max));
431
432         set_max = idedisk_set_max_address(drive, set_max, lba48);
433
434         if (set_max) {
435                 drive->capacity64 = set_max;
436                 printk(KERN_INFO "%s: Host Protected Area disabled.\n",
437                                  drive->name);
438         }
439 }
440
441 /*
442  * Compute drive->capacity, the full capacity of the drive
443  * Called with drive->id != NULL.
444  *
445  * To compute capacity, this uses either of
446  *
447  *    1. CHS value set by user       (whatever user sets will be trusted)
448  *    2. LBA value from target drive (require new ATA feature)
449  *    3. LBA value from system BIOS  (new one is OK, old one may break)
450  *    4. CHS value from system BIOS  (traditional style)
451  *
452  * in above order (i.e., if value of higher priority is available,
453  * reset will be ignored).
454  */
455 static void init_idedisk_capacity (ide_drive_t  *drive)
456 {
457         struct hd_driveid *id = drive->id;
458         /*
459          * If this drive supports the Host Protected Area feature set,
460          * then we may need to change our opinion about the drive's capacity.
461          */
462         int hpa = idedisk_supports_hpa(id);
463
464         if (idedisk_supports_lba48(id)) {
465                 /* drive speaks 48-bit LBA */
466                 drive->select.b.lba = 1;
467                 drive->capacity64 = id->lba_capacity_2;
468                 if (hpa)
469                         idedisk_check_hpa(drive);
470         } else if ((id->capability & 2) && lba_capacity_is_ok(id)) {
471                 /* drive speaks 28-bit LBA */
472                 drive->select.b.lba = 1;
473                 drive->capacity64 = id->lba_capacity;
474                 if (hpa)
475                         idedisk_check_hpa(drive);
476         } else {
477                 /* drive speaks boring old 28-bit CHS */
478                 drive->capacity64 = drive->cyl * drive->head * drive->sect;
479         }
480 }
481
482 static sector_t idedisk_capacity (ide_drive_t *drive)
483 {
484         return drive->capacity64 - drive->sect0;
485 }
486
487 #ifdef CONFIG_IDE_PROC_FS
488 static int smart_enable(ide_drive_t *drive)
489 {
490         ide_task_t args;
491         struct ide_taskfile *tf = &args.tf;
492
493         memset(&args, 0, sizeof(ide_task_t));
494         tf->feature = SMART_ENABLE;
495         tf->lbam    = SMART_LCYL_PASS;
496         tf->lbah    = SMART_HCYL_PASS;
497         tf->command = WIN_SMART;
498         args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
499         return ide_no_data_taskfile(drive, &args);
500 }
501
502 static int get_smart_data(ide_drive_t *drive, u8 *buf, u8 sub_cmd)
503 {
504         ide_task_t args;
505         struct ide_taskfile *tf = &args.tf;
506
507         memset(&args, 0, sizeof(ide_task_t));
508         tf->feature = sub_cmd;
509         tf->nsect   = 0x01;
510         tf->lbam    = SMART_LCYL_PASS;
511         tf->lbah    = SMART_HCYL_PASS;
512         tf->command = WIN_SMART;
513         args.tf_flags   = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
514         args.data_phase = TASKFILE_IN;
515         (void) smart_enable(drive);
516         return ide_raw_taskfile(drive, &args, buf, 1);
517 }
518
519 static int proc_idedisk_read_cache
520         (char *page, char **start, off_t off, int count, int *eof, void *data)
521 {
522         ide_drive_t     *drive = (ide_drive_t *) data;
523         char            *out = page;
524         int             len;
525
526         if (drive->id_read)
527                 len = sprintf(out,"%i\n", drive->id->buf_size / 2);
528         else
529                 len = sprintf(out,"(none)\n");
530         PROC_IDE_READ_RETURN(page,start,off,count,eof,len);
531 }
532
533 static int proc_idedisk_read_capacity
534         (char *page, char **start, off_t off, int count, int *eof, void *data)
535 {
536         ide_drive_t*drive = (ide_drive_t *)data;
537         int len;
538
539         len = sprintf(page,"%llu\n", (long long)idedisk_capacity(drive));
540         PROC_IDE_READ_RETURN(page,start,off,count,eof,len);
541 }
542
543 static int proc_idedisk_read_smart_thresholds
544         (char *page, char **start, off_t off, int count, int *eof, void *data)
545 {
546         ide_drive_t     *drive = (ide_drive_t *)data;
547         int             len = 0, i = 0;
548
549         if (get_smart_data(drive, page, SMART_READ_THRESHOLDS) == 0) {
550                 unsigned short *val = (unsigned short *) page;
551                 char *out = ((char *)val) + (SECTOR_WORDS * 4);
552                 page = out;
553                 do {
554                         out += sprintf(out, "%04x%c", le16_to_cpu(*val), (++i & 7) ? ' ' : '\n');
555                         val += 1;
556                 } while (i < (SECTOR_WORDS * 2));
557                 len = out - page;
558         }
559         PROC_IDE_READ_RETURN(page,start,off,count,eof,len);
560 }
561
562 static int proc_idedisk_read_smart_values
563         (char *page, char **start, off_t off, int count, int *eof, void *data)
564 {
565         ide_drive_t     *drive = (ide_drive_t *)data;
566         int             len = 0, i = 0;
567
568         if (get_smart_data(drive, page, SMART_READ_VALUES) == 0) {
569                 unsigned short *val = (unsigned short *) page;
570                 char *out = ((char *)val) + (SECTOR_WORDS * 4);
571                 page = out;
572                 do {
573                         out += sprintf(out, "%04x%c", le16_to_cpu(*val), (++i & 7) ? ' ' : '\n');
574                         val += 1;
575                 } while (i < (SECTOR_WORDS * 2));
576                 len = out - page;
577         }
578         PROC_IDE_READ_RETURN(page,start,off,count,eof,len);
579 }
580
581 static ide_proc_entry_t idedisk_proc[] = {
582         { "cache",              S_IFREG|S_IRUGO,        proc_idedisk_read_cache,                NULL },
583         { "capacity",           S_IFREG|S_IRUGO,        proc_idedisk_read_capacity,             NULL },
584         { "geometry",           S_IFREG|S_IRUGO,        proc_ide_read_geometry,                 NULL },
585         { "smart_values",       S_IFREG|S_IRUSR,        proc_idedisk_read_smart_values,         NULL },
586         { "smart_thresholds",   S_IFREG|S_IRUSR,        proc_idedisk_read_smart_thresholds,     NULL },
587         { NULL, 0, NULL, NULL }
588 };
589 #endif  /* CONFIG_IDE_PROC_FS */
590
591 static void idedisk_prepare_flush(struct request_queue *q, struct request *rq)
592 {
593         ide_drive_t *drive = q->queuedata;
594         ide_task_t task;
595
596         memset(&task, 0, sizeof(task));
597         if (ide_id_has_flush_cache_ext(drive->id) &&
598             (drive->capacity64 >= (1UL << 28)))
599                 task.tf.command = WIN_FLUSH_CACHE_EXT;
600         else
601                 task.tf.command = WIN_FLUSH_CACHE;
602         task.tf_flags   = IDE_TFLAG_OUT_TF | IDE_TFLAG_OUT_DEVICE;
603         task.data_phase = TASKFILE_NO_DATA;
604
605         rq->cmd_type = REQ_TYPE_ATA_TASKFILE;
606         rq->cmd_flags |= REQ_SOFTBARRIER;
607         rq->special = &task;
608 }
609
610 /*
611  * This is tightly woven into the driver->do_special can not touch.
612  * DON'T do it again until a total personality rewrite is committed.
613  */
614 static int set_multcount(ide_drive_t *drive, int arg)
615 {
616         struct request rq;
617
618         if (arg < 0 || arg > drive->id->max_multsect)
619                 return -EINVAL;
620
621         if (drive->special.b.set_multmode)
622                 return -EBUSY;
623
624         ide_init_drive_cmd (&rq);
625         rq.cmd_type = REQ_TYPE_ATA_TASKFILE;
626
627         drive->mult_req = arg;
628         drive->special.b.set_multmode = 1;
629         (void) ide_do_drive_cmd (drive, &rq, ide_wait);
630         return (drive->mult_count == arg) ? 0 : -EIO;
631 }
632
633 static int set_nowerr(ide_drive_t *drive, int arg)
634 {
635         if (arg < 0 || arg > 1)
636                 return -EINVAL;
637
638         if (ide_spin_wait_hwgroup(drive))
639                 return -EBUSY;
640         drive->nowerr = arg;
641         drive->bad_wstat = arg ? BAD_R_STAT : BAD_W_STAT;
642         spin_unlock_irq(&ide_lock);
643         return 0;
644 }
645
646 static void update_ordered(ide_drive_t *drive)
647 {
648         struct hd_driveid *id = drive->id;
649         unsigned ordered = QUEUE_ORDERED_NONE;
650         prepare_flush_fn *prep_fn = NULL;
651
652         if (drive->wcache) {
653                 unsigned long long capacity;
654                 int barrier;
655                 /*
656                  * We must avoid issuing commands a drive does not
657                  * understand or we may crash it. We check flush cache
658                  * is supported. We also check we have the LBA48 flush
659                  * cache if the drive capacity is too large. By this
660                  * time we have trimmed the drive capacity if LBA48 is
661                  * not available so we don't need to recheck that.
662                  */
663                 capacity = idedisk_capacity(drive);
664                 barrier = ide_id_has_flush_cache(id) && !drive->noflush &&
665                         (drive->addressing == 0 || capacity <= (1ULL << 28) ||
666                          ide_id_has_flush_cache_ext(id));
667
668                 printk(KERN_INFO "%s: cache flushes %ssupported\n",
669                        drive->name, barrier ? "" : "not ");
670
671                 if (barrier) {
672                         ordered = QUEUE_ORDERED_DRAIN_FLUSH;
673                         prep_fn = idedisk_prepare_flush;
674                 }
675         } else
676                 ordered = QUEUE_ORDERED_DRAIN;
677
678         blk_queue_ordered(drive->queue, ordered, prep_fn);
679 }
680
681 static int write_cache(ide_drive_t *drive, int arg)
682 {
683         ide_task_t args;
684         int err = 1;
685
686         if (arg < 0 || arg > 1)
687                 return -EINVAL;
688
689         if (ide_id_has_flush_cache(drive->id)) {
690                 memset(&args, 0, sizeof(ide_task_t));
691                 args.tf.feature = arg ?
692                         SETFEATURES_EN_WCACHE : SETFEATURES_DIS_WCACHE;
693                 args.tf.command = WIN_SETFEATURES;
694                 args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
695                 err = ide_no_data_taskfile(drive, &args);
696                 if (err == 0)
697                         drive->wcache = arg;
698         }
699
700         update_ordered(drive);
701
702         return err;
703 }
704
705 static int do_idedisk_flushcache (ide_drive_t *drive)
706 {
707         ide_task_t args;
708
709         memset(&args, 0, sizeof(ide_task_t));
710         if (ide_id_has_flush_cache_ext(drive->id))
711                 args.tf.command = WIN_FLUSH_CACHE_EXT;
712         else
713                 args.tf.command = WIN_FLUSH_CACHE;
714         args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
715         return ide_no_data_taskfile(drive, &args);
716 }
717
718 static int set_acoustic (ide_drive_t *drive, int arg)
719 {
720         ide_task_t args;
721
722         if (arg < 0 || arg > 254)
723                 return -EINVAL;
724
725         memset(&args, 0, sizeof(ide_task_t));
726         args.tf.feature = arg ? SETFEATURES_EN_AAM : SETFEATURES_DIS_AAM;
727         args.tf.nsect   = arg;
728         args.tf.command = WIN_SETFEATURES;
729         args.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
730         ide_no_data_taskfile(drive, &args);
731         drive->acoustic = arg;
732         return 0;
733 }
734
735 /*
736  * drive->addressing:
737  *      0: 28-bit
738  *      1: 48-bit
739  *      2: 48-bit capable doing 28-bit
740  */
741 static int set_lba_addressing(ide_drive_t *drive, int arg)
742 {
743         if (arg < 0 || arg > 2)
744                 return -EINVAL;
745
746         drive->addressing =  0;
747
748         if (drive->hwif->host_flags & IDE_HFLAG_NO_LBA48)
749                 return 0;
750
751         if (!idedisk_supports_lba48(drive->id))
752                 return -EIO;
753         drive->addressing = arg;
754         return 0;
755 }
756
757 #ifdef CONFIG_IDE_PROC_FS
758 static void idedisk_add_settings(ide_drive_t *drive)
759 {
760         struct hd_driveid *id = drive->id;
761
762         ide_add_setting(drive,  "bios_cyl",     SETTING_RW,     TYPE_INT,       0,      65535,                  1,      1,      &drive->bios_cyl,       NULL);
763         ide_add_setting(drive,  "bios_head",    SETTING_RW,     TYPE_BYTE,      0,      255,                    1,      1,      &drive->bios_head,      NULL);
764         ide_add_setting(drive,  "bios_sect",    SETTING_RW,     TYPE_BYTE,      0,      63,                     1,      1,      &drive->bios_sect,      NULL);
765         ide_add_setting(drive,  "address",      SETTING_RW,     TYPE_BYTE,      0,      2,                      1,      1,      &drive->addressing,     set_lba_addressing);
766         ide_add_setting(drive,  "multcount",    SETTING_RW,     TYPE_BYTE,      0,      id->max_multsect,       1,      1,      &drive->mult_count,     set_multcount);
767         ide_add_setting(drive,  "nowerr",       SETTING_RW,     TYPE_BYTE,      0,      1,                      1,      1,      &drive->nowerr,         set_nowerr);
768         ide_add_setting(drive,  "lun",          SETTING_RW,     TYPE_INT,       0,      7,                      1,      1,      &drive->lun,            NULL);
769         ide_add_setting(drive,  "wcache",       SETTING_RW,     TYPE_BYTE,      0,      1,                      1,      1,      &drive->wcache,         write_cache);
770         ide_add_setting(drive,  "acoustic",     SETTING_RW,     TYPE_BYTE,      0,      254,                    1,      1,      &drive->acoustic,       set_acoustic);
771         ide_add_setting(drive,  "failures",     SETTING_RW,     TYPE_INT,       0,      65535,                  1,      1,      &drive->failures,       NULL);
772         ide_add_setting(drive,  "max_failures", SETTING_RW,     TYPE_INT,       0,      65535,                  1,      1,      &drive->max_failures,   NULL);
773 }
774 #else
775 static inline void idedisk_add_settings(ide_drive_t *drive) { ; }
776 #endif
777
778 static void idedisk_setup (ide_drive_t *drive)
779 {
780         ide_hwif_t *hwif = drive->hwif;
781         struct hd_driveid *id = drive->id;
782         unsigned long long capacity;
783
784         idedisk_add_settings(drive);
785
786         if (drive->id_read == 0)
787                 return;
788
789         if (drive->removable) {
790                 /*
791                  * Removable disks (eg. SYQUEST); ignore 'WD' drives 
792                  */
793                 if (id->model[0] != 'W' || id->model[1] != 'D') {
794                         drive->doorlocking = 1;
795                 }
796         }
797
798         (void)set_lba_addressing(drive, 1);
799
800         if (drive->addressing == 1) {
801                 int max_s = 2048;
802
803                 if (max_s > hwif->rqsize)
804                         max_s = hwif->rqsize;
805
806                 blk_queue_max_sectors(drive->queue, max_s);
807         }
808
809         printk(KERN_INFO "%s: max request size: %dKiB\n", drive->name, drive->queue->max_sectors / 2);
810
811         /* calculate drive capacity, and select LBA if possible */
812         init_idedisk_capacity (drive);
813
814         /* limit drive capacity to 137GB if LBA48 cannot be used */
815         if (drive->addressing == 0 && drive->capacity64 > 1ULL << 28) {
816                 printk(KERN_WARNING "%s: cannot use LBA48 - full capacity "
817                        "%llu sectors (%llu MB)\n",
818                        drive->name, (unsigned long long)drive->capacity64,
819                        sectors_to_MB(drive->capacity64));
820                 drive->capacity64 = 1ULL << 28;
821         }
822
823         if ((hwif->host_flags & IDE_HFLAG_NO_LBA48_DMA) && drive->addressing) {
824                 if (drive->capacity64 > 1ULL << 28) {
825                         printk(KERN_INFO "%s: cannot use LBA48 DMA - PIO mode will"
826                                          " be used for accessing sectors > %u\n",
827                                          drive->name, 1 << 28);
828                 } else
829                         drive->addressing = 0;
830         }
831
832         /*
833          * if possible, give fdisk access to more of the drive,
834          * by correcting bios_cyls:
835          */
836         capacity = idedisk_capacity (drive);
837         if (!drive->forced_geom) {
838
839                 if (idedisk_supports_lba48(drive->id)) {
840                         /* compatibility */
841                         drive->bios_sect = 63;
842                         drive->bios_head = 255;
843                 }
844
845                 if (drive->bios_sect && drive->bios_head) {
846                         unsigned int cap0 = capacity; /* truncate to 32 bits */
847                         unsigned int cylsz, cyl;
848
849                         if (cap0 != capacity)
850                                 drive->bios_cyl = 65535;
851                         else {
852                                 cylsz = drive->bios_sect * drive->bios_head;
853                                 cyl = cap0 / cylsz;
854                                 if (cyl > 65535)
855                                         cyl = 65535;
856                                 if (cyl > drive->bios_cyl)
857                                         drive->bios_cyl = cyl;
858                         }
859                 }
860         }
861         printk(KERN_INFO "%s: %llu sectors (%llu MB)",
862                          drive->name, capacity, sectors_to_MB(capacity));
863
864         /* Only print cache size when it was specified */
865         if (id->buf_size)
866                 printk (" w/%dKiB Cache", id->buf_size/2);
867
868         printk(KERN_CONT ", CHS=%d/%d/%d\n",
869                          drive->bios_cyl, drive->bios_head, drive->bios_sect);
870
871         /* write cache enabled? */
872         if ((id->csfo & 1) || (id->cfs_enable_1 & (1 << 5)))
873                 drive->wcache = 1;
874
875         write_cache(drive, 1);
876 }
877
878 static void ide_cacheflush_p(ide_drive_t *drive)
879 {
880         if (!drive->wcache || !ide_id_has_flush_cache(drive->id))
881                 return;
882
883         if (do_idedisk_flushcache(drive))
884                 printk(KERN_INFO "%s: wcache flush failed!\n", drive->name);
885 }
886
887 static void ide_disk_remove(ide_drive_t *drive)
888 {
889         struct ide_disk_obj *idkp = drive->driver_data;
890         struct gendisk *g = idkp->disk;
891
892         ide_proc_unregister_driver(drive, idkp->driver);
893
894         del_gendisk(g);
895
896         ide_cacheflush_p(drive);
897
898         ide_disk_put(idkp);
899 }
900
901 static void ide_disk_release(struct kref *kref)
902 {
903         struct ide_disk_obj *idkp = to_ide_disk(kref);
904         ide_drive_t *drive = idkp->drive;
905         struct gendisk *g = idkp->disk;
906
907         drive->driver_data = NULL;
908         g->private_data = NULL;
909         put_disk(g);
910         kfree(idkp);
911 }
912
913 static int ide_disk_probe(ide_drive_t *drive);
914
915 /*
916  * On HPA drives the capacity needs to be
917  * reinitilized on resume otherwise the disk
918  * can not be used and a hard reset is required
919  */
920 static void ide_disk_resume(ide_drive_t *drive)
921 {
922         if (idedisk_supports_hpa(drive->id))
923                 init_idedisk_capacity(drive);
924 }
925
926 static void ide_device_shutdown(ide_drive_t *drive)
927 {
928 #ifdef  CONFIG_ALPHA
929         /* On Alpha, halt(8) doesn't actually turn the machine off,
930            it puts you into the sort of firmware monitor. Typically,
931            it's used to boot another kernel image, so it's not much
932            different from reboot(8). Therefore, we don't need to
933            spin down the disk in this case, especially since Alpha
934            firmware doesn't handle disks in standby mode properly.
935            On the other hand, it's reasonably safe to turn the power
936            off when the shutdown process reaches the firmware prompt,
937            as the firmware initialization takes rather long time -
938            at least 10 seconds, which should be sufficient for
939            the disk to expire its write cache. */
940         if (system_state != SYSTEM_POWER_OFF) {
941 #else
942         if (system_state == SYSTEM_RESTART) {
943 #endif
944                 ide_cacheflush_p(drive);
945                 return;
946         }
947
948         printk("Shutdown: %s\n", drive->name);
949         drive->gendev.bus->suspend(&drive->gendev, PMSG_SUSPEND);
950 }
951
952 static ide_driver_t idedisk_driver = {
953         .gen_driver = {
954                 .owner          = THIS_MODULE,
955                 .name           = "ide-disk",
956                 .bus            = &ide_bus_type,
957         },
958         .probe                  = ide_disk_probe,
959         .remove                 = ide_disk_remove,
960         .resume                 = ide_disk_resume,
961         .shutdown               = ide_device_shutdown,
962         .version                = IDEDISK_VERSION,
963         .media                  = ide_disk,
964         .supports_dsc_overlap   = 0,
965         .do_request             = ide_do_rw_disk,
966         .end_request            = ide_end_request,
967         .error                  = __ide_error,
968         .abort                  = __ide_abort,
969 #ifdef CONFIG_IDE_PROC_FS
970         .proc                   = idedisk_proc,
971 #endif
972 };
973
974 static int idedisk_set_doorlock(ide_drive_t *drive, int on)
975 {
976         ide_task_t task;
977
978         memset(&task, 0, sizeof(task));
979         task.tf.command = on ? WIN_DOORLOCK : WIN_DOORUNLOCK;
980         task.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
981
982         return ide_no_data_taskfile(drive, &task);
983 }
984
985 static int idedisk_open(struct inode *inode, struct file *filp)
986 {
987         struct gendisk *disk = inode->i_bdev->bd_disk;
988         struct ide_disk_obj *idkp;
989         ide_drive_t *drive;
990
991         if (!(idkp = ide_disk_get(disk)))
992                 return -ENXIO;
993
994         drive = idkp->drive;
995
996         idkp->openers++;
997
998         if (drive->removable && idkp->openers == 1) {
999                 check_disk_change(inode->i_bdev);
1000                 /*
1001                  * Ignore the return code from door_lock,
1002                  * since the open() has already succeeded,
1003                  * and the door_lock is irrelevant at this point.
1004                  */
1005                 if (drive->doorlocking && idedisk_set_doorlock(drive, 1))
1006                         drive->doorlocking = 0;
1007         }
1008         return 0;
1009 }
1010
1011 static int idedisk_release(struct inode *inode, struct file *filp)
1012 {
1013         struct gendisk *disk = inode->i_bdev->bd_disk;
1014         struct ide_disk_obj *idkp = ide_disk_g(disk);
1015         ide_drive_t *drive = idkp->drive;
1016
1017         if (idkp->openers == 1)
1018                 ide_cacheflush_p(drive);
1019
1020         if (drive->removable && idkp->openers == 1) {
1021                 if (drive->doorlocking && idedisk_set_doorlock(drive, 0))
1022                         drive->doorlocking = 0;
1023         }
1024
1025         idkp->openers--;
1026
1027         ide_disk_put(idkp);
1028
1029         return 0;
1030 }
1031
1032 static int idedisk_getgeo(struct block_device *bdev, struct hd_geometry *geo)
1033 {
1034         struct ide_disk_obj *idkp = ide_disk_g(bdev->bd_disk);
1035         ide_drive_t *drive = idkp->drive;
1036
1037         geo->heads = drive->bios_head;
1038         geo->sectors = drive->bios_sect;
1039         geo->cylinders = (u16)drive->bios_cyl; /* truncate */
1040         return 0;
1041 }
1042
1043 static int idedisk_ioctl(struct inode *inode, struct file *file,
1044                         unsigned int cmd, unsigned long arg)
1045 {
1046         unsigned long flags;
1047         struct block_device *bdev = inode->i_bdev;
1048         struct ide_disk_obj *idkp = ide_disk_g(bdev->bd_disk);
1049         ide_drive_t *drive = idkp->drive;
1050         int err, (*setfunc)(ide_drive_t *, int);
1051         u8 *val;
1052
1053         switch (cmd) {
1054         case HDIO_GET_ADDRESS:   val = &drive->addressing;      goto read_val;
1055         case HDIO_GET_MULTCOUNT: val = &drive->mult_count;      goto read_val;
1056         case HDIO_GET_NOWERR:    val = &drive->nowerr;          goto read_val;
1057         case HDIO_GET_WCACHE:    val = &drive->wcache;          goto read_val;
1058         case HDIO_GET_ACOUSTIC:  val = &drive->acoustic;        goto read_val;
1059         case HDIO_SET_ADDRESS:   setfunc = set_lba_addressing;  goto set_val;
1060         case HDIO_SET_MULTCOUNT: setfunc = set_multcount;       goto set_val;
1061         case HDIO_SET_NOWERR:    setfunc = set_nowerr;          goto set_val;
1062         case HDIO_SET_WCACHE:    setfunc = write_cache;         goto set_val;
1063         case HDIO_SET_ACOUSTIC:  setfunc = set_acoustic;        goto set_val;
1064         }
1065
1066         return generic_ide_ioctl(drive, file, bdev, cmd, arg);
1067
1068 read_val:
1069         mutex_lock(&ide_setting_mtx);
1070         spin_lock_irqsave(&ide_lock, flags);
1071         err = *val;
1072         spin_unlock_irqrestore(&ide_lock, flags);
1073         mutex_unlock(&ide_setting_mtx);
1074         return err >= 0 ? put_user(err, (long __user *)arg) : err;
1075
1076 set_val:
1077         if (bdev != bdev->bd_contains)
1078                 err = -EINVAL;
1079         else {
1080                 if (!capable(CAP_SYS_ADMIN))
1081                         err = -EACCES;
1082                 else {
1083                         mutex_lock(&ide_setting_mtx);
1084                         err = setfunc(drive, arg);
1085                         mutex_unlock(&ide_setting_mtx);
1086                 }
1087         }
1088         return err;
1089 }
1090
1091 static int idedisk_media_changed(struct gendisk *disk)
1092 {
1093         struct ide_disk_obj *idkp = ide_disk_g(disk);
1094         ide_drive_t *drive = idkp->drive;
1095
1096         /* do not scan partitions twice if this is a removable device */
1097         if (drive->attach) {
1098                 drive->attach = 0;
1099                 return 0;
1100         }
1101         /* if removable, always assume it was changed */
1102         return drive->removable;
1103 }
1104
1105 static int idedisk_revalidate_disk(struct gendisk *disk)
1106 {
1107         struct ide_disk_obj *idkp = ide_disk_g(disk);
1108         set_capacity(disk, idedisk_capacity(idkp->drive));
1109         return 0;
1110 }
1111
1112 static struct block_device_operations idedisk_ops = {
1113         .owner          = THIS_MODULE,
1114         .open           = idedisk_open,
1115         .release        = idedisk_release,
1116         .ioctl          = idedisk_ioctl,
1117         .getgeo         = idedisk_getgeo,
1118         .media_changed  = idedisk_media_changed,
1119         .revalidate_disk= idedisk_revalidate_disk
1120 };
1121
1122 MODULE_DESCRIPTION("ATA DISK Driver");
1123
1124 static int ide_disk_probe(ide_drive_t *drive)
1125 {
1126         struct ide_disk_obj *idkp;
1127         struct gendisk *g;
1128
1129         /* strstr("foo", "") is non-NULL */
1130         if (!strstr("ide-disk", drive->driver_req))
1131                 goto failed;
1132         if (!drive->present)
1133                 goto failed;
1134         if (drive->media != ide_disk)
1135                 goto failed;
1136
1137         idkp = kzalloc(sizeof(*idkp), GFP_KERNEL);
1138         if (!idkp)
1139                 goto failed;
1140
1141         g = alloc_disk_node(1 << PARTN_BITS,
1142                         hwif_to_node(drive->hwif));
1143         if (!g)
1144                 goto out_free_idkp;
1145
1146         ide_init_disk(g, drive);
1147
1148         ide_proc_register_driver(drive, &idedisk_driver);
1149
1150         kref_init(&idkp->kref);
1151
1152         idkp->drive = drive;
1153         idkp->driver = &idedisk_driver;
1154         idkp->disk = g;
1155
1156         g->private_data = &idkp->driver;
1157
1158         drive->driver_data = idkp;
1159
1160         idedisk_setup(drive);
1161         if ((!drive->head || drive->head > 16) && !drive->select.b.lba) {
1162                 printk(KERN_ERR "%s: INVALID GEOMETRY: %d PHYSICAL HEADS?\n",
1163                         drive->name, drive->head);
1164                 drive->attach = 0;
1165         } else
1166                 drive->attach = 1;
1167
1168         g->minors = 1 << PARTN_BITS;
1169         g->driverfs_dev = &drive->gendev;
1170         g->flags = drive->removable ? GENHD_FL_REMOVABLE : 0;
1171         set_capacity(g, idedisk_capacity(drive));
1172         g->fops = &idedisk_ops;
1173         add_disk(g);
1174         return 0;
1175
1176 out_free_idkp:
1177         kfree(idkp);
1178 failed:
1179         return -ENODEV;
1180 }
1181
1182 static void __exit idedisk_exit (void)
1183 {
1184         driver_unregister(&idedisk_driver.gen_driver);
1185 }
1186
1187 static int __init idedisk_init(void)
1188 {
1189         return driver_register(&idedisk_driver.gen_driver);
1190 }
1191
1192 MODULE_ALIAS("ide:*m-disk*");
1193 module_init(idedisk_init);
1194 module_exit(idedisk_exit);
1195 MODULE_LICENSE("GPL");