--- linux-2.5.46/include/linux/genhd.h 2002-11-04 14:30:08.000000000 -0800 +++ linux/include/linux/genhd.h 2002-11-06 04:19:13.000000000 -0800 @@ -107,6 +108,7 @@ int in_flight; unsigned long stamp, stamp_idle; unsigned time_in_queue; + void (*destructor)(struct gendisk *disk); }; /* drivers/block/ll_rw_blk.c */ @@ -283,8 +285,9 @@ extern void add_partition(struct gendisk *, int, sector_t, sector_t); extern void delete_partition(struct gendisk *, int); +extern void init_disk(struct gendisk *disk, int minors); extern struct gendisk *alloc_disk(int minors); extern struct gendisk *get_disk(struct gendisk *disk); extern void put_disk(struct gendisk *disk); --- linux-2.5.46/drivers/block/genhd.c 2002-11-04 14:30:27.000000000 -0800 +++ linux/drivers/block/genhd.c 2002-11-06 04:19:04.000000000 -0800 @@ -369,12 +366,16 @@ NULL, }; +static void disk_destructor(struct gendisk *disk) +{ + kfree(disk); +} + static void disk_release(struct kobject * kobj) { struct gendisk *disk = to_disk(kobj); - kfree(disk->random); - kfree(disk->part); - kfree(disk); + if (disk->destructor) + (*disk->destructor)(disk); } struct subsystem block_subsys = { @@ -384,28 +385,31 @@ .default_attrs = default_attrs, }; +/* init_disk assumes disk should is initialized to all zeroes, + except for disk->part, which points to the partitions structures, + which are also assumed to be initialized to all zeroes. */ +void init_disk(struct gendisk *disk, int minors) +{ + disk->minors = minors; + while (minors >>= 1) + disk->minor_shift++; + kobject_init(&disk->kobj); + disk->kobj.subsys = &block_subsys; + INIT_LIST_HEAD(&disk->full_list); + rand_initialize_disk(disk); +} + struct gendisk *alloc_disk(int minors) { - struct gendisk *disk = kmalloc(sizeof(struct gendisk), GFP_KERNEL); + int size = sizeof(struct gendisk) + + (minors - 1) * sizeof(struct hd_struct); + struct gendisk *disk = kmalloc(size, GFP_KERNEL); if (disk) { - memset(disk, 0, sizeof(struct gendisk)); - if (minors > 1) { - int size = (minors - 1) * sizeof(struct hd_struct); - disk->part = kmalloc(size, GFP_KERNEL); - if (!disk->part) { - kfree(disk); - return NULL; - } - memset(disk->part, 0, size); - } - disk->minors = minors; - while (minors >>= 1) - disk->minor_shift++; - kobject_init(&disk->kobj); - disk->kobj.subsys = &block_subsys; - INIT_LIST_HEAD(&disk->full_list); + memset(disk, 0, size); + disk->part = (struct hd_struct*) &disk[1]; + init_disk(disk, minors); + disk->destructor = disk_destructor; } - rand_initialize_disk(disk); return disk; } @@ -426,6 +430,7 @@ kobject_put(&disk->kobj); } +EXPORT_SYMBOL(init_disk); EXPORT_SYMBOL(alloc_disk); EXPORT_SYMBOL(get_disk); EXPORT_SYMBOL(put_disk);