[linux-2.6-block.git] / drivers / ide / ide-gd.c

#include <linux/module.h>
#include <linux/types.h>
#include <linux/string.h>
#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/genhd.h>
#include <linux/mutex.h>
#include <linux/ide.h>
#include <linux/hdreg.h>

#if !defined(CONFIG_DEBUG_BLOCK_EXT_DEVT)
#define IDE_DISK_MINORS		(1 << PARTN_BITS)
#else
#define IDE_DISK_MINORS		0
#endif

#include "ide-disk.h"
#include "ide-floppy.h"

#define IDE_GD_VERSION	"1.18"

/* module parameters */
static unsigned long debug_mask;
module_param(debug_mask, ulong, 0644);

static DEFINE_MUTEX(ide_disk_ref_mutex);

static void ide_disk_release(struct kref *);

static struct ide_disk_obj *ide_disk_get(struct gendisk *disk)
{
	struct ide_disk_obj *idkp = NULL;

	mutex_lock(&ide_disk_ref_mutex);
	idkp = ide_drv_g(disk, ide_disk_obj);
	if (idkp) {
		if (ide_device_get(idkp->drive))
			idkp = NULL;
		else
			kref_get(&idkp->kref);
	}
	mutex_unlock(&ide_disk_ref_mutex);
	return idkp;
}

static void ide_disk_put(struct ide_disk_obj *idkp)
{
	ide_drive_t *drive = idkp->drive;

	mutex_lock(&ide_disk_ref_mutex);
	kref_put(&idkp->kref, ide_disk_release);
	ide_device_put(drive);
	mutex_unlock(&ide_disk_ref_mutex);
}

sector_t ide_gd_capacity(ide_drive_t *drive)
{
	return drive->capacity64;
}

static int ide_gd_probe(ide_drive_t *);

static void ide_gd_remove(ide_drive_t *drive)
{
	struct ide_disk_obj *idkp = drive->driver_data;
	struct gendisk *g = idkp->disk;

	ide_proc_unregister_driver(drive, idkp->driver);

	del_gendisk(g);

	drive->disk_ops->flush(drive);

	ide_disk_put(idkp);
}

static void ide_disk_release(struct kref *kref)
{
	struct ide_disk_obj *idkp = to_ide_drv(kref, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;
	struct gendisk *g = idkp->disk;

	drive->disk_ops = NULL;
	drive->driver_data = NULL;
	g->private_data = NULL;
	put_disk(g);
	kfree(idkp);
}

/*
 * On HPA drives the capacity needs to be
 * reinitilized on resume otherwise the disk
 * can not be used and a hard reset is required
 */
static void ide_gd_resume(ide_drive_t *drive)
{
	if (ata_id_hpa_enabled(drive->id))
		(void)drive->disk_ops->get_capacity(drive);
}

static void ide_gd_shutdown(ide_drive_t *drive)
{
#ifdef	CONFIG_ALPHA
	/* On Alpha, halt(8) doesn't actually turn the machine off,
	   it puts you into the sort of firmware monitor. Typically,
	   it's used to boot another kernel image, so it's not much
	   different from reboot(8). Therefore, we don't need to
	   spin down the disk in this case, especially since Alpha
	   firmware doesn't handle disks in standby mode properly.
	   On the other hand, it's reasonably safe to turn the power
	   off when the shutdown process reaches the firmware prompt,
	   as the firmware initialization takes rather long time -
	   at least 10 seconds, which should be sufficient for
	   the disk to expire its write cache. */
	if (system_state != SYSTEM_POWER_OFF) {
#else
	if (system_state == SYSTEM_RESTART) {
#endif
		drive->disk_ops->flush(drive);
		return;
	}

	printk(KERN_INFO "Shutdown: %s\n", drive->name);

	drive->gendev.bus->suspend(&drive->gendev, PMSG_SUSPEND);
}

#ifdef CONFIG_IDE_PROC_FS
static ide_proc_entry_t *ide_disk_proc_entries(ide_drive_t *drive)
{
	return (drive->media == ide_disk) ? ide_disk_proc : ide_floppy_proc;
}

static const struct ide_proc_devset *ide_disk_proc_devsets(ide_drive_t *drive)
{
	return (drive->media == ide_disk) ? ide_disk_settings
					  : ide_floppy_settings;
}
#endif

static ide_startstop_t ide_gd_do_request(ide_drive_t *drive,
					 struct request *rq, sector_t sector)
{
	return drive->disk_ops->do_request(drive, rq, sector);
}

static int ide_gd_end_request(ide_drive_t *drive, int uptodate, int nrsecs)
{
	return drive->disk_ops->end_request(drive, uptodate, nrsecs);
}

static ide_driver_t ide_gd_driver = {
	.gen_driver = {
		.owner		= THIS_MODULE,
		.name		= "ide-gd",
		.bus		= &ide_bus_type,
	},
	.probe			= ide_gd_probe,
	.remove			= ide_gd_remove,
	.resume			= ide_gd_resume,
	.shutdown		= ide_gd_shutdown,
	.version		= IDE_GD_VERSION,
	.do_request		= ide_gd_do_request,
	.end_request		= ide_gd_end_request,
	.error			= __ide_error,
#ifdef CONFIG_IDE_PROC_FS
	.proc_entries		= ide_disk_proc_entries,
	.proc_devsets		= ide_disk_proc_devsets,
#endif
};

static int ide_gd_open(struct block_device *bdev, fmode_t mode)
{
	struct gendisk *disk = bdev->bd_disk;
	struct ide_disk_obj *idkp;
	ide_drive_t *drive;
	int ret = 0;

	idkp = ide_disk_get(disk);
	if (idkp == NULL)
		return -ENXIO;

	drive = idkp->drive;

	ide_debug_log(IDE_DBG_FUNC, "Call %s\n", __func__);

	idkp->openers++;

	if ((drive->dev_flags & IDE_DFLAG_REMOVABLE) && idkp->openers == 1) {
		drive->dev_flags &= ~IDE_DFLAG_FORMAT_IN_PROGRESS;
		/* Just in case */

		ret = drive->disk_ops->init_media(drive, disk);

		/*
		 * Allow O_NDELAY to open a drive without a disk, or with an
		 * unreadable disk, so that we can get the format capacity
		 * of the drive or begin the format - Sam
		 */
		if (ret && (mode & FMODE_NDELAY) == 0) {
			ret = -EIO;
			goto out_put_idkp;
		}

		if ((drive->dev_flags & IDE_DFLAG_WP) && (mode & FMODE_WRITE)) {
			ret = -EROFS;
			goto out_put_idkp;
		}

		/*
		 * Ignore the return code from door_lock,
		 * since the open() has already succeeded,
		 * and the door_lock is irrelevant at this point.
		 */
		drive->disk_ops->set_doorlock(drive, disk, 1);
		drive->dev_flags |= IDE_DFLAG_MEDIA_CHANGED;
		check_disk_change(bdev);
	} else if (drive->dev_flags & IDE_DFLAG_FORMAT_IN_PROGRESS) {
		ret = -EBUSY;
		goto out_put_idkp;
	}
	return 0;

out_put_idkp:
	idkp->openers--;
	ide_disk_put(idkp);
	return ret;
}

static int ide_gd_release(struct gendisk *disk, fmode_t mode)
{
	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;

	ide_debug_log(IDE_DBG_FUNC, "Call %s\n", __func__);

	if (idkp->openers == 1)
		drive->disk_ops->flush(drive);

	if ((drive->dev_flags & IDE_DFLAG_REMOVABLE) && idkp->openers == 1) {
		drive->disk_ops->set_doorlock(drive, disk, 0);
		drive->dev_flags &= ~IDE_DFLAG_FORMAT_IN_PROGRESS;
	}

	idkp->openers--;

	ide_disk_put(idkp);

	return 0;
}

static int ide_gd_getgeo(struct block_device *bdev, struct hd_geometry *geo)
{
	struct ide_disk_obj *idkp = ide_drv_g(bdev->bd_disk, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;

	geo->heads = drive->bios_head;
	geo->sectors = drive->bios_sect;
	geo->cylinders = (u16)drive->bios_cyl; /* truncate */
	return 0;
}

static int ide_gd_media_changed(struct gendisk *disk)
{
	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;
	int ret;

	/* do not scan partitions twice if this is a removable device */
	if (drive->dev_flags & IDE_DFLAG_ATTACH) {
		drive->dev_flags &= ~IDE_DFLAG_ATTACH;
		return 0;
	}

	ret = !!(drive->dev_flags & IDE_DFLAG_MEDIA_CHANGED);
	drive->dev_flags &= ~IDE_DFLAG_MEDIA_CHANGED;

	return ret;
}

static int ide_gd_revalidate_disk(struct gendisk *disk)
{
	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;

	if (ide_gd_media_changed(disk))
		drive->disk_ops->get_capacity(drive);

	set_capacity(disk, ide_gd_capacity(drive));
	return 0;
}

static int ide_gd_ioctl(struct block_device *bdev, fmode_t mode,
			     unsigned int cmd, unsigned long arg)
{
	struct ide_disk_obj *idkp = ide_drv_g(bdev->bd_disk, ide_disk_obj);
	ide_drive_t *drive = idkp->drive;

	return drive->disk_ops->ioctl(drive, bdev, mode, cmd, arg);
}

static struct block_device_operations ide_gd_ops = {
	.owner			= THIS_MODULE,
	.open			= ide_gd_open,
	.release		= ide_gd_release,
	.locked_ioctl		= ide_gd_ioctl,
	.getgeo			= ide_gd_getgeo,
	.media_changed		= ide_gd_media_changed,
	.revalidate_disk	= ide_gd_revalidate_disk
};

static int ide_gd_probe(ide_drive_t *drive)
{
	const struct ide_disk_ops *disk_ops = NULL;
	struct ide_disk_obj *idkp;
	struct gendisk *g;

	/* strstr("foo", "") is non-NULL */
	if (!strstr("ide-gd", drive->driver_req))
		goto failed;

#ifdef CONFIG_IDE_GD_ATA
	if (drive->media == ide_disk)
		disk_ops = &ide_ata_disk_ops;
#endif
#ifdef CONFIG_IDE_GD_ATAPI
	if (drive->media == ide_floppy)
		disk_ops = &ide_atapi_disk_ops;
#endif
	if (disk_ops == NULL)
		goto failed;

	if (disk_ops->check(drive, DRV_NAME) == 0) {
		printk(KERN_ERR PFX "%s: not supported by this driver\n",
			drive->name);
		goto failed;
	}

	idkp = kzalloc(sizeof(*idkp), GFP_KERNEL);
	if (!idkp) {
		printk(KERN_ERR PFX "%s: can't allocate a disk structure\n",
			drive->name);
		goto failed;
	}

	g = alloc_disk_node(IDE_DISK_MINORS, hwif_to_node(drive->hwif));
	if (!g)
		goto out_free_idkp;

	ide_init_disk(g, drive);

	kref_init(&idkp->kref);

	idkp->drive = drive;
	idkp->driver = &ide_gd_driver;
	idkp->disk = g;

	g->private_data = &idkp->driver;

	drive->driver_data = idkp;
	drive->debug_mask = debug_mask;
	drive->disk_ops = disk_ops;

	disk_ops->setup(drive);

	set_capacity(g, ide_gd_capacity(drive));

	g->minors = IDE_DISK_MINORS;
	g->driverfs_dev = &drive->gendev;
	g->flags |= GENHD_FL_EXT_DEVT;
	if (drive->dev_flags & IDE_DFLAG_REMOVABLE)
		g->flags = GENHD_FL_REMOVABLE;
	g->fops = &ide_gd_ops;
	add_disk(g);
	return 0;

out_free_idkp:
	kfree(idkp);
failed:
	return -ENODEV;
}

static int __init ide_gd_init(void)
{
	printk(KERN_INFO DRV_NAME " driver " IDE_GD_VERSION "\n");
	return driver_register(&ide_gd_driver.gen_driver);
}

static void __exit ide_gd_exit(void)
{
	driver_unregister(&ide_gd_driver.gen_driver);
}

MODULE_ALIAS("ide:*m-disk*");
MODULE_ALIAS("ide-disk");
MODULE_ALIAS("ide:*m-floppy*");
MODULE_ALIAS("ide-floppy");
module_init(ide_gd_init);
module_exit(ide_gd_exit);
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("generic ATA/ATAPI disk driver");
Commit	Line	Data
5fef0e5c BZ	1	#include <linux/module.h>
	2	#include <linux/types.h>
	3	#include <linux/string.h>
	4	#include <linux/kernel.h>
	5	#include <linux/errno.h>
	6	#include <linux/genhd.h>
	7	#include <linux/mutex.h>
	8	#include <linux/ide.h>
	9	#include <linux/hdreg.h>
	10
	11	#if !defined(CONFIG_DEBUG_BLOCK_EXT_DEVT)
	12	#define IDE_DISK_MINORS (1 << PARTN_BITS)
	13	#else
	14	#define IDE_DISK_MINORS 0
	15	#endif
	16
	17	#include "ide-disk.h"
806f80a6	18	#include "ide-floppy.h"
5fef0e5c BZ	19
	20	#define IDE_GD_VERSION "1.18"
	21
806f80a6 BZ	22	/* module parameters */
	23	static unsigned long debug_mask;
	24	module_param(debug_mask, ulong, 0644);
	25
5fef0e5c BZ	26	static DEFINE_MUTEX(ide_disk_ref_mutex);
	27
	28	static void ide_disk_release(struct kref *);
	29
	30	static struct ide_disk_obj ide_disk_get(struct gendisk disk)
	31	{
	32	struct ide_disk_obj *idkp = NULL;
	33
	34	mutex_lock(&ide_disk_ref_mutex);
	35	idkp = ide_drv_g(disk, ide_disk_obj);
	36	if (idkp) {
	37	if (ide_device_get(idkp->drive))
	38	idkp = NULL;
	39	else
	40	kref_get(&idkp->kref);
	41	}
	42	mutex_unlock(&ide_disk_ref_mutex);
	43	return idkp;
	44	}
	45
	46	static void ide_disk_put(struct ide_disk_obj *idkp)
	47	{
	48	ide_drive_t *drive = idkp->drive;
	49
	50	mutex_lock(&ide_disk_ref_mutex);
	51	kref_put(&idkp->kref, ide_disk_release);
	52	ide_device_put(drive);
	53	mutex_unlock(&ide_disk_ref_mutex);
	54	}
	55
	56	sector_t ide_gd_capacity(ide_drive_t *drive)
	57	{
	58	return drive->capacity64;
	59	}
	60
	61	static int ide_gd_probe(ide_drive_t *);
	62
	63	static void ide_gd_remove(ide_drive_t *drive)
	64	{
	65	struct ide_disk_obj *idkp = drive->driver_data;
	66	struct gendisk *g = idkp->disk;
	67
	68	ide_proc_unregister_driver(drive, idkp->driver);
	69
	70	del_gendisk(g);
	71
806f80a6	72	drive->disk_ops->flush(drive);
5fef0e5c BZ	73
	74	ide_disk_put(idkp);
	75	}
	76
	77	static void ide_disk_release(struct kref *kref)
	78	{
	79	struct ide_disk_obj *idkp = to_ide_drv(kref, ide_disk_obj);
	80	ide_drive_t *drive = idkp->drive;
	81	struct gendisk *g = idkp->disk;
	82
806f80a6	83	drive->disk_ops = NULL;
5fef0e5c BZ	84	drive->driver_data = NULL;
	85	g->private_data = NULL;
	86	put_disk(g);
	87	kfree(idkp);
	88	}
	89
	90	/*
	91	* On HPA drives the capacity needs to be
	92	* reinitilized on resume otherwise the disk
	93	* can not be used and a hard reset is required
	94	*/
	95	static void ide_gd_resume(ide_drive_t *drive)
	96	{
	97	if (ata_id_hpa_enabled(drive->id))
806f80a6	98	(void)drive->disk_ops->get_capacity(drive);
5fef0e5c BZ	99	}
	100
	101	static void ide_gd_shutdown(ide_drive_t *drive)
	102	{
	103	#ifdef CONFIG_ALPHA
	104	/* On Alpha, halt(8) doesn't actually turn the machine off,
	105	it puts you into the sort of firmware monitor. Typically,
	106	it's used to boot another kernel image, so it's not much
	107	different from reboot(8). Therefore, we don't need to
	108	spin down the disk in this case, especially since Alpha
	109	firmware doesn't handle disks in standby mode properly.
	110	On the other hand, it's reasonably safe to turn the power
	111	off when the shutdown process reaches the firmware prompt,
	112	as the firmware initialization takes rather long time -
	113	at least 10 seconds, which should be sufficient for
	114	the disk to expire its write cache. */
	115	if (system_state != SYSTEM_POWER_OFF) {
	116	#else
	117	if (system_state == SYSTEM_RESTART) {
	118	#endif
806f80a6	119	drive->disk_ops->flush(drive);
5fef0e5c BZ	120	return;
	121	}
	122
	123	printk(KERN_INFO "Shutdown: %s\n", drive->name);
	124
	125	drive->gendev.bus->suspend(&drive->gendev, PMSG_SUSPEND);
	126	}
	127
79cb3803 BZ	128	#ifdef CONFIG_IDE_PROC_FS
	129	static ide_proc_entry_t ide_disk_proc_entries(ide_drive_t drive)
	130	{
806f80a6	131	return (drive->media == ide_disk) ? ide_disk_proc : ide_floppy_proc;
79cb3803 BZ	132	}
	133
	134	static const struct ide_proc_devset ide_disk_proc_devsets(ide_drive_t drive)
	135	{
806f80a6 BZ	136	return (drive->media == ide_disk) ? ide_disk_settings
806f80a6 BZ	137	: ide_floppy_settings;
79cb3803 BZ	138	}
	139	#endif
	140
806f80a6 BZ	141	static ide_startstop_t ide_gd_do_request(ide_drive_t *drive,
	142	struct request *rq, sector_t sector)
	143	{
	144	return drive->disk_ops->do_request(drive, rq, sector);
	145	}
	146
	147	static int ide_gd_end_request(ide_drive_t *drive, int uptodate, int nrsecs)
	148	{
	149	return drive->disk_ops->end_request(drive, uptodate, nrsecs);
	150	}
	151
5fef0e5c BZ	152	static ide_driver_t ide_gd_driver = {
	153	.gen_driver = {
	154	.owner = THIS_MODULE,
806f80a6	155	.name = "ide-gd",
5fef0e5c BZ	156	.bus = &ide_bus_type,
	157	},
	158	.probe = ide_gd_probe,
	159	.remove = ide_gd_remove,
	160	.resume = ide_gd_resume,
	161	.shutdown = ide_gd_shutdown,
	162	.version = IDE_GD_VERSION,
806f80a6 BZ	163	.do_request = ide_gd_do_request,
806f80a6 BZ	164	.end_request = ide_gd_end_request,
5fef0e5c BZ	165	.error = __ide_error,
5fef0e5c BZ	166	#ifdef CONFIG_IDE_PROC_FS
79cb3803 BZ	167	.proc_entries = ide_disk_proc_entries,
79cb3803 BZ	168	.proc_devsets = ide_disk_proc_devsets,
5fef0e5c BZ	169	#endif
	170	};
	171
b2f21e05	172	static int ide_gd_open(struct block_device *bdev, fmode_t mode)
5fef0e5c	173	{
b2f21e05	174	struct gendisk *disk = bdev->bd_disk;
5fef0e5c BZ	175	struct ide_disk_obj *idkp;
5fef0e5c BZ	176	ide_drive_t *drive;
806f80a6	177	int ret = 0;
5fef0e5c BZ	178
	179	idkp = ide_disk_get(disk);
	180	if (idkp == NULL)
	181	return -ENXIO;
	182
	183	drive = idkp->drive;
	184
806f80a6 BZ	185	ide_debug_log(IDE_DBG_FUNC, "Call %s\n", __func__);
806f80a6 BZ	186
5fef0e5c BZ	187	idkp->openers++;
	188
	189	if ((drive->dev_flags & IDE_DFLAG_REMOVABLE) && idkp->openers == 1) {
806f80a6 BZ	190	drive->dev_flags &= ~IDE_DFLAG_FORMAT_IN_PROGRESS;
	191	/* Just in case */
	192
	193	ret = drive->disk_ops->init_media(drive, disk);
	194
	195	/*
	196	* Allow O_NDELAY to open a drive without a disk, or with an
	197	* unreadable disk, so that we can get the format capacity
	198	* of the drive or begin the format - Sam
	199	*/
b2f21e05	200	if (ret && (mode & FMODE_NDELAY) == 0) {
806f80a6 BZ	201	ret = -EIO;
	202	goto out_put_idkp;
	203	}
	204
b2f21e05	205	if ((drive->dev_flags & IDE_DFLAG_WP) && (mode & FMODE_WRITE)) {
806f80a6 BZ	206	ret = -EROFS;
	207	goto out_put_idkp;
	208	}
	209
5fef0e5c BZ	210	/*
	211	* Ignore the return code from door_lock,
	212	* since the open() has already succeeded,
	213	* and the door_lock is irrelevant at this point.
	214	*/
806f80a6	215	drive->disk_ops->set_doorlock(drive, disk, 1);
cedd120c	216	drive->dev_flags \|= IDE_DFLAG_MEDIA_CHANGED;
b2f21e05	217	check_disk_change(bdev);
806f80a6 BZ	218	} else if (drive->dev_flags & IDE_DFLAG_FORMAT_IN_PROGRESS) {
	219	ret = -EBUSY;
	220	goto out_put_idkp;
5fef0e5c BZ	221	}
5fef0e5c BZ	222	return 0;
806f80a6 BZ	223
	224	out_put_idkp:
	225	idkp->openers--;
	226	ide_disk_put(idkp);
	227	return ret;
5fef0e5c BZ	228	}
5fef0e5c BZ	229
b2f21e05	230	static int ide_gd_release(struct gendisk *disk, fmode_t mode)
5fef0e5c	231	{
5fef0e5c BZ	232	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
	233	ide_drive_t *drive = idkp->drive;
	234
806f80a6 BZ	235	ide_debug_log(IDE_DBG_FUNC, "Call %s\n", __func__);
806f80a6 BZ	236
5fef0e5c	237	if (idkp->openers == 1)
806f80a6	238	drive->disk_ops->flush(drive);
5fef0e5c	239
806f80a6 BZ	240	if ((drive->dev_flags & IDE_DFLAG_REMOVABLE) && idkp->openers == 1) {
	241	drive->disk_ops->set_doorlock(drive, disk, 0);
	242	drive->dev_flags &= ~IDE_DFLAG_FORMAT_IN_PROGRESS;
	243	}
5fef0e5c BZ	244
	245	idkp->openers--;
	246
	247	ide_disk_put(idkp);
	248
	249	return 0;
	250	}
	251
	252	static int ide_gd_getgeo(struct block_device bdev, struct hd_geometry geo)
	253	{
	254	struct ide_disk_obj *idkp = ide_drv_g(bdev->bd_disk, ide_disk_obj);
	255	ide_drive_t *drive = idkp->drive;
	256
	257	geo->heads = drive->bios_head;
	258	geo->sectors = drive->bios_sect;
	259	geo->cylinders = (u16)drive->bios_cyl; /* truncate */
	260	return 0;
	261	}
	262
	263	static int ide_gd_media_changed(struct gendisk *disk)
	264	{
	265	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
	266	ide_drive_t *drive = idkp->drive;
cedd120c	267	int ret;
5fef0e5c BZ	268
	269	/* do not scan partitions twice if this is a removable device */
	270	if (drive->dev_flags & IDE_DFLAG_ATTACH) {
	271	drive->dev_flags &= ~IDE_DFLAG_ATTACH;
	272	return 0;
	273	}
	274
cedd120c BZ	275	ret = !!(drive->dev_flags & IDE_DFLAG_MEDIA_CHANGED);
	276	drive->dev_flags &= ~IDE_DFLAG_MEDIA_CHANGED;
	277
	278	return ret;
5fef0e5c BZ	279	}
	280
	281	static int ide_gd_revalidate_disk(struct gendisk *disk)
	282	{
	283	struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
52ebb438 BP	284	ide_drive_t *drive = idkp->drive;
	285
	286	if (ide_gd_media_changed(disk))
	287	drive->disk_ops->get_capacity(drive);
	288
	289	set_capacity(disk, ide_gd_capacity(drive));
5fef0e5c BZ	290	return 0;
	291	}
	292
b2f21e05	293	static int ide_gd_ioctl(struct block_device *bdev, fmode_t mode,
806f80a6 BZ	294	unsigned int cmd, unsigned long arg)
806f80a6 BZ	295	{
806f80a6 BZ	296	struct ide_disk_obj *idkp = ide_drv_g(bdev->bd_disk, ide_disk_obj);
	297	ide_drive_t *drive = idkp->drive;
	298
b2f21e05	299	return drive->disk_ops->ioctl(drive, bdev, mode, cmd, arg);
806f80a6 BZ	300	}
806f80a6 BZ	301
5fef0e5c BZ	302	static struct block_device_operations ide_gd_ops = {
5fef0e5c BZ	303	.owner = THIS_MODULE,
b2f21e05 AV	304	.open = ide_gd_open,
	305	.release = ide_gd_release,
	306	.locked_ioctl = ide_gd_ioctl,
5fef0e5c BZ	307	.getgeo = ide_gd_getgeo,
	308	.media_changed = ide_gd_media_changed,
	309	.revalidate_disk = ide_gd_revalidate_disk
	310	};
	311
	312	static int ide_gd_probe(ide_drive_t *drive)
	313	{
806f80a6	314	const struct ide_disk_ops *disk_ops = NULL;
5fef0e5c BZ	315	struct ide_disk_obj *idkp;
	316	struct gendisk *g;
	317
	318	/* strstr("foo", "") is non-NULL */
806f80a6 BZ	319	if (!strstr("ide-gd", drive->driver_req))
	320	goto failed;
	321
	322	#ifdef CONFIG_IDE_GD_ATA
	323	if (drive->media == ide_disk)
	324	disk_ops = &ide_ata_disk_ops;
	325	#endif
	326	#ifdef CONFIG_IDE_GD_ATAPI
	327	if (drive->media == ide_floppy)
	328	disk_ops = &ide_atapi_disk_ops;
	329	#endif
	330	if (disk_ops == NULL)
5fef0e5c BZ	331	goto failed;
5fef0e5c BZ	332
806f80a6 BZ	333	if (disk_ops->check(drive, DRV_NAME) == 0) {
	334	printk(KERN_ERR PFX "%s: not supported by this driver\n",
	335	drive->name);
5fef0e5c	336	goto failed;
806f80a6	337	}
5fef0e5c BZ	338
5fef0e5c BZ	339	idkp = kzalloc(sizeof(*idkp), GFP_KERNEL);
806f80a6 BZ	340	if (!idkp) {
	341	printk(KERN_ERR PFX "%s: can't allocate a disk structure\n",
	342	drive->name);
5fef0e5c	343	goto failed;
806f80a6	344	}
5fef0e5c BZ	345
	346	g = alloc_disk_node(IDE_DISK_MINORS, hwif_to_node(drive->hwif));
	347	if (!g)
	348	goto out_free_idkp;
	349
	350	ide_init_disk(g, drive);
	351
	352	kref_init(&idkp->kref);
	353
	354	idkp->drive = drive;
	355	idkp->driver = &ide_gd_driver;
	356	idkp->disk = g;
	357
	358	g->private_data = &idkp->driver;
	359
	360	drive->driver_data = idkp;
806f80a6 BZ	361	drive->debug_mask = debug_mask;
806f80a6 BZ	362	drive->disk_ops = disk_ops;
5fef0e5c	363
806f80a6	364	disk_ops->setup(drive);
5fef0e5c BZ	365
	366	set_capacity(g, ide_gd_capacity(drive));
	367
	368	g->minors = IDE_DISK_MINORS;
	369	g->driverfs_dev = &drive->gendev;
	370	g->flags \|= GENHD_FL_EXT_DEVT;
	371	if (drive->dev_flags & IDE_DFLAG_REMOVABLE)
	372	g->flags = GENHD_FL_REMOVABLE;
	373	g->fops = &ide_gd_ops;
	374	add_disk(g);
	375	return 0;
	376
	377	out_free_idkp:
	378	kfree(idkp);
	379	failed:
	380	return -ENODEV;
	381	}
	382
	383	static int __init ide_gd_init(void)
	384	{
806f80a6	385	printk(KERN_INFO DRV_NAME " driver " IDE_GD_VERSION "\n");
5fef0e5c BZ	386	return driver_register(&ide_gd_driver.gen_driver);
	387	}
	388
	389	static void __exit ide_gd_exit(void)
	390	{
	391	driver_unregister(&ide_gd_driver.gen_driver);
	392	}
	393
	394	MODULE_ALIAS("ide:m-disk");
	395	MODULE_ALIAS("ide-disk");
806f80a6 BZ	396	MODULE_ALIAS("ide:m-floppy");
806f80a6 BZ	397	MODULE_ALIAS("ide-floppy");
5fef0e5c BZ	398	module_init(ide_gd_init);
	399	module_exit(ide_gd_exit);
	400	MODULE_LICENSE("GPL");
806f80a6	401	MODULE_DESCRIPTION("generic ATA/ATAPI disk driver");