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