genhd.c 26.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7
/*
 *  gendisk handling
 */

#include <linux/module.h>
#include <linux/fs.h>
#include <linux/genhd.h>
8
#include <linux/kdev_t.h>
L
Linus Torvalds 已提交
9 10 11 12 13 14 15 16
#include <linux/kernel.h>
#include <linux/blkdev.h>
#include <linux/init.h>
#include <linux/spinlock.h>
#include <linux/seq_file.h>
#include <linux/slab.h>
#include <linux/kmod.h>
#include <linux/kobj_map.h>
17
#include <linux/buffer_head.h>
18
#include <linux/mutex.h>
T
Tejun Heo 已提交
19
#include <linux/idr.h>
L
Linus Torvalds 已提交
20

21 22
#include "blk.h"

23 24 25 26
static DEFINE_MUTEX(block_class_lock);
#ifndef CONFIG_SYSFS_DEPRECATED
struct kobject *block_depr;
#endif
L
Linus Torvalds 已提交
27

T
Tejun Heo 已提交
28 29 30 31 32 33 34 35 36
/* for extended dynamic devt allocation, currently only one major is used */
#define MAX_EXT_DEVT		(1 << MINORBITS)

/* For extended devt allocation.  ext_devt_mutex prevents look up
 * results from going away underneath its user.
 */
static DEFINE_MUTEX(ext_devt_mutex);
static DEFINE_IDR(ext_devt_idr);

A
Adrian Bunk 已提交
37 38
static struct device_type disk_type;

39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
/**
 * disk_get_part - get partition
 * @disk: disk to look partition from
 * @partno: partition number
 *
 * Look for partition @partno from @disk.  If found, increment
 * reference count and return it.
 *
 * CONTEXT:
 * Don't care.
 *
 * RETURNS:
 * Pointer to the found partition on success, NULL if not found.
 */
struct hd_struct *disk_get_part(struct gendisk *disk, int partno)
{
	struct hd_struct *part;

T
Tejun Heo 已提交
57
	if (unlikely(partno < 0 || partno >= disk_max_parts(disk)))
58 59
		return NULL;
	rcu_read_lock();
T
Tejun Heo 已提交
60
	part = rcu_dereference(disk->__part[partno]);
61
	if (part)
62
		get_device(part_to_dev(part));
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
	rcu_read_unlock();

	return part;
}
EXPORT_SYMBOL_GPL(disk_get_part);

/**
 * disk_part_iter_init - initialize partition iterator
 * @piter: iterator to initialize
 * @disk: disk to iterate over
 * @flags: DISK_PITER_* flags
 *
 * Initialize @piter so that it iterates over partitions of @disk.
 *
 * CONTEXT:
 * Don't care.
 */
void disk_part_iter_init(struct disk_part_iter *piter, struct gendisk *disk,
			  unsigned int flags)
{
	piter->disk = disk;
	piter->part = NULL;

	if (flags & DISK_PITER_REVERSE)
		piter->idx = disk_max_parts(piter->disk) - 1;
T
Tejun Heo 已提交
88
	else if (flags & DISK_PITER_INCL_PART0)
89
		piter->idx = 0;
T
Tejun Heo 已提交
90 91
	else
		piter->idx = 1;
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118

	piter->flags = flags;
}
EXPORT_SYMBOL_GPL(disk_part_iter_init);

/**
 * disk_part_iter_next - proceed iterator to the next partition and return it
 * @piter: iterator of interest
 *
 * Proceed @piter to the next partition and return it.
 *
 * CONTEXT:
 * Don't care.
 */
struct hd_struct *disk_part_iter_next(struct disk_part_iter *piter)
{
	int inc, end;

	/* put the last partition */
	disk_put_part(piter->part);
	piter->part = NULL;

	rcu_read_lock();

	/* determine iteration parameters */
	if (piter->flags & DISK_PITER_REVERSE) {
		inc = -1;
T
Tejun Heo 已提交
119 120 121 122
		if (piter->flags & DISK_PITER_INCL_PART0)
			end = -1;
		else
			end = 0;
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
	} else {
		inc = 1;
		end = disk_max_parts(piter->disk);
	}

	/* iterate to the next partition */
	for (; piter->idx != end; piter->idx += inc) {
		struct hd_struct *part;

		part = rcu_dereference(piter->disk->__part[piter->idx]);
		if (!part)
			continue;
		if (!(piter->flags & DISK_PITER_INCL_EMPTY) && !part->nr_sects)
			continue;

138
		get_device(part_to_dev(part));
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
		piter->part = part;
		piter->idx += inc;
		break;
	}

	rcu_read_unlock();

	return piter->part;
}
EXPORT_SYMBOL_GPL(disk_part_iter_next);

/**
 * disk_part_iter_exit - finish up partition iteration
 * @piter: iter of interest
 *
 * Called when iteration is over.  Cleans up @piter.
 *
 * CONTEXT:
 * Don't care.
 */
void disk_part_iter_exit(struct disk_part_iter *piter)
{
	disk_put_part(piter->part);
	piter->part = NULL;
}
EXPORT_SYMBOL_GPL(disk_part_iter_exit);

/**
 * disk_map_sector_rcu - map sector to partition
 * @disk: gendisk of interest
 * @sector: sector to map
 *
 * Find out which partition @sector maps to on @disk.  This is
 * primarily used for stats accounting.
 *
 * CONTEXT:
 * RCU read locked.  The returned partition pointer is valid only
 * while preemption is disabled.
 *
 * RETURNS:
 * Found partition on success, NULL if there's no matching partition.
 */
struct hd_struct *disk_map_sector_rcu(struct gendisk *disk, sector_t sector)
{
	int i;

T
Tejun Heo 已提交
185
	for (i = 1; i < disk_max_parts(disk); i++) {
186 187 188 189 190 191 192 193 194 195
		struct hd_struct *part = rcu_dereference(disk->__part[i]);

		if (part && part->start_sect <= sector &&
		    sector < part->start_sect + part->nr_sects)
			return part;
	}
	return NULL;
}
EXPORT_SYMBOL_GPL(disk_map_sector_rcu);

L
Linus Torvalds 已提交
196 197 198 199 200 201 202 203
/*
 * Can be deleted altogether. Later.
 *
 */
static struct blk_major_name {
	struct blk_major_name *next;
	int major;
	char name[16];
204
} *major_names[BLKDEV_MAJOR_HASH_SIZE];
L
Linus Torvalds 已提交
205 206 207 208

/* index in the above - for now: assume no multimajor ranges */
static inline int major_to_index(int major)
{
209
	return major % BLKDEV_MAJOR_HASH_SIZE;
210 211
}

212
#ifdef CONFIG_PROC_FS
213
void blkdev_show(struct seq_file *seqf, off_t offset)
214
{
215
	struct blk_major_name *dp;
216

217
	if (offset < BLKDEV_MAJOR_HASH_SIZE) {
218
		mutex_lock(&block_class_lock);
219
		for (dp = major_names[offset]; dp; dp = dp->next)
220
			seq_printf(seqf, "%3d %s\n", dp->major, dp->name);
221
		mutex_unlock(&block_class_lock);
L
Linus Torvalds 已提交
222 223
	}
}
224
#endif /* CONFIG_PROC_FS */
L
Linus Torvalds 已提交
225 226 227 228 229 230

int register_blkdev(unsigned int major, const char *name)
{
	struct blk_major_name **n, *p;
	int index, ret = 0;

231
	mutex_lock(&block_class_lock);
L
Linus Torvalds 已提交
232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275

	/* temporary */
	if (major == 0) {
		for (index = ARRAY_SIZE(major_names)-1; index > 0; index--) {
			if (major_names[index] == NULL)
				break;
		}

		if (index == 0) {
			printk("register_blkdev: failed to get major for %s\n",
			       name);
			ret = -EBUSY;
			goto out;
		}
		major = index;
		ret = major;
	}

	p = kmalloc(sizeof(struct blk_major_name), GFP_KERNEL);
	if (p == NULL) {
		ret = -ENOMEM;
		goto out;
	}

	p->major = major;
	strlcpy(p->name, name, sizeof(p->name));
	p->next = NULL;
	index = major_to_index(major);

	for (n = &major_names[index]; *n; n = &(*n)->next) {
		if ((*n)->major == major)
			break;
	}
	if (!*n)
		*n = p;
	else
		ret = -EBUSY;

	if (ret < 0) {
		printk("register_blkdev: cannot get major %d for %s\n",
		       major, name);
		kfree(p);
	}
out:
276
	mutex_unlock(&block_class_lock);
L
Linus Torvalds 已提交
277 278 279 280 281
	return ret;
}

EXPORT_SYMBOL(register_blkdev);

A
Akinobu Mita 已提交
282
void unregister_blkdev(unsigned int major, const char *name)
L
Linus Torvalds 已提交
283 284 285 286 287
{
	struct blk_major_name **n;
	struct blk_major_name *p = NULL;
	int index = major_to_index(major);

288
	mutex_lock(&block_class_lock);
L
Linus Torvalds 已提交
289 290 291
	for (n = &major_names[index]; *n; n = &(*n)->next)
		if ((*n)->major == major)
			break;
292 293 294
	if (!*n || strcmp((*n)->name, name)) {
		WARN_ON(1);
	} else {
L
Linus Torvalds 已提交
295 296 297
		p = *n;
		*n = p->next;
	}
298
	mutex_unlock(&block_class_lock);
L
Linus Torvalds 已提交
299 300 301 302 303 304 305
	kfree(p);
}

EXPORT_SYMBOL(unregister_blkdev);

static struct kobj_map *bdev_map;

306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
/**
 * blk_mangle_minor - scatter minor numbers apart
 * @minor: minor number to mangle
 *
 * Scatter consecutively allocated @minor number apart if MANGLE_DEVT
 * is enabled.  Mangling twice gives the original value.
 *
 * RETURNS:
 * Mangled value.
 *
 * CONTEXT:
 * Don't care.
 */
static int blk_mangle_minor(int minor)
{
#ifdef CONFIG_DEBUG_BLOCK_EXT_DEVT
	int i;

	for (i = 0; i < MINORBITS / 2; i++) {
		int low = minor & (1 << i);
		int high = minor & (1 << (MINORBITS - 1 - i));
		int distance = MINORBITS - 1 - 2 * i;

		minor ^= low | high;	/* clear both bits */
		low <<= distance;	/* swap the positions */
		high >>= distance;
		minor |= low | high;	/* and set */
	}
#endif
	return minor;
}

T
Tejun Heo 已提交
338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378
/**
 * blk_alloc_devt - allocate a dev_t for a partition
 * @part: partition to allocate dev_t for
 * @gfp_mask: memory allocation flag
 * @devt: out parameter for resulting dev_t
 *
 * Allocate a dev_t for block device.
 *
 * RETURNS:
 * 0 on success, allocated dev_t is returned in *@devt.  -errno on
 * failure.
 *
 * CONTEXT:
 * Might sleep.
 */
int blk_alloc_devt(struct hd_struct *part, dev_t *devt)
{
	struct gendisk *disk = part_to_disk(part);
	int idx, rc;

	/* in consecutive minor range? */
	if (part->partno < disk->minors) {
		*devt = MKDEV(disk->major, disk->first_minor + part->partno);
		return 0;
	}

	/* allocate ext devt */
	do {
		if (!idr_pre_get(&ext_devt_idr, GFP_KERNEL))
			return -ENOMEM;
		rc = idr_get_new(&ext_devt_idr, part, &idx);
	} while (rc == -EAGAIN);

	if (rc)
		return rc;

	if (idx > MAX_EXT_DEVT) {
		idr_remove(&ext_devt_idr, idx);
		return -EBUSY;
	}

379
	*devt = MKDEV(BLOCK_EXT_MAJOR, blk_mangle_minor(idx));
T
Tejun Heo 已提交
380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
	return 0;
}

/**
 * blk_free_devt - free a dev_t
 * @devt: dev_t to free
 *
 * Free @devt which was allocated using blk_alloc_devt().
 *
 * CONTEXT:
 * Might sleep.
 */
void blk_free_devt(dev_t devt)
{
	might_sleep();

	if (devt == MKDEV(0, 0))
		return;

	if (MAJOR(devt) == BLOCK_EXT_MAJOR) {
		mutex_lock(&ext_devt_mutex);
401
		idr_remove(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
T
Tejun Heo 已提交
402 403 404 405
		mutex_unlock(&ext_devt_mutex);
	}
}

406 407 408 409 410 411 412 413 414 415 416 417
static char *bdevt_str(dev_t devt, char *buf)
{
	if (MAJOR(devt) <= 0xff && MINOR(devt) <= 0xff) {
		char tbuf[BDEVT_SIZE];
		snprintf(tbuf, BDEVT_SIZE, "%02x%02x", MAJOR(devt), MINOR(devt));
		snprintf(buf, BDEVT_SIZE, "%-9s", tbuf);
	} else
		snprintf(buf, BDEVT_SIZE, "%03x:%05x", MAJOR(devt), MINOR(devt));

	return buf;
}

L
Linus Torvalds 已提交
418 419 420 421 422
/*
 * Register device numbers dev..(dev+range-1)
 * range must be nonzero
 * The hash chain is sorted on range, so that subranges can override.
 */
423
void blk_register_region(dev_t devt, unsigned long range, struct module *module,
L
Linus Torvalds 已提交
424 425 426
			 struct kobject *(*probe)(dev_t, int *, void *),
			 int (*lock)(dev_t, void *), void *data)
{
427
	kobj_map(bdev_map, devt, range, module, probe, lock, data);
L
Linus Torvalds 已提交
428 429 430 431
}

EXPORT_SYMBOL(blk_register_region);

432
void blk_unregister_region(dev_t devt, unsigned long range)
L
Linus Torvalds 已提交
433
{
434
	kobj_unmap(bdev_map, devt, range);
L
Linus Torvalds 已提交
435 436 437 438
}

EXPORT_SYMBOL(blk_unregister_region);

439
static struct kobject *exact_match(dev_t devt, int *partno, void *data)
L
Linus Torvalds 已提交
440 441
{
	struct gendisk *p = data;
442

443
	return &disk_to_dev(p)->kobj;
L
Linus Torvalds 已提交
444 445
}

446
static int exact_lock(dev_t devt, void *data)
L
Linus Torvalds 已提交
447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463
{
	struct gendisk *p = data;

	if (!get_disk(p))
		return -1;
	return 0;
}

/**
 * add_disk - add partitioning information to kernel list
 * @disk: per-device partitioning information
 *
 * This function registers the partitioning information in @disk
 * with the kernel.
 */
void add_disk(struct gendisk *disk)
{
464
	struct backing_dev_info *bdi;
465
	int retval;
466

L
Linus Torvalds 已提交
467
	disk->flags |= GENHD_FL_UP;
468
	disk_to_dev(disk)->devt = MKDEV(disk->major, disk->first_minor);
469 470
	blk_register_region(disk_devt(disk), disk->minors, NULL,
			    exact_match, exact_lock, disk);
L
Linus Torvalds 已提交
471 472
	register_disk(disk);
	blk_register_queue(disk);
473 474

	bdi = &disk->queue->backing_dev_info;
475
	bdi_register_dev(bdi, disk_devt(disk));
476 477
	retval = sysfs_create_link(&disk_to_dev(disk)->kobj, &bdi->dev->kobj,
				   "bdi");
478
	WARN_ON(retval);
L
Linus Torvalds 已提交
479 480 481 482 483 484 485
}

EXPORT_SYMBOL(add_disk);
EXPORT_SYMBOL(del_gendisk);	/* in partitions/check.c */

void unlink_gendisk(struct gendisk *disk)
{
486
	sysfs_remove_link(&disk_to_dev(disk)->kobj, "bdi");
487
	bdi_unregister(&disk->queue->backing_dev_info);
L
Linus Torvalds 已提交
488
	blk_unregister_queue(disk);
489
	blk_unregister_region(disk_devt(disk), disk->minors);
L
Linus Torvalds 已提交
490 491 492 493
}

/**
 * get_gendisk - get partitioning information for a given device
494 495
 * @devt: device to get partitioning information for
 * @part: returned partition index
L
Linus Torvalds 已提交
496 497
 *
 * This function gets the structure containing partitioning
498
 * information for the given device @devt.
L
Linus Torvalds 已提交
499
 */
500
struct gendisk *get_gendisk(dev_t devt, int *partno)
L
Linus Torvalds 已提交
501
{
T
Tejun Heo 已提交
502 503 504 505 506 507 508 509 510 511 512 513
	struct gendisk *disk = NULL;

	if (MAJOR(devt) != BLOCK_EXT_MAJOR) {
		struct kobject *kobj;

		kobj = kobj_lookup(bdev_map, devt, partno);
		if (kobj)
			disk = dev_to_disk(kobj_to_dev(kobj));
	} else {
		struct hd_struct *part;

		mutex_lock(&ext_devt_mutex);
514
		part = idr_find(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
T
Tejun Heo 已提交
515 516 517 518 519 520
		if (part && get_disk(part_to_disk(part))) {
			*partno = part->partno;
			disk = part_to_disk(part);
		}
		mutex_unlock(&ext_devt_mutex);
	}
521

T
Tejun Heo 已提交
522
	return disk;
L
Linus Torvalds 已提交
523 524
}

525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
/**
 * bdget_disk - do bdget() by gendisk and partition number
 * @disk: gendisk of interest
 * @partno: partition number
 *
 * Find partition @partno from @disk, do bdget() on it.
 *
 * CONTEXT:
 * Don't care.
 *
 * RETURNS:
 * Resulting block_device on success, NULL on failure.
 */
extern struct block_device *bdget_disk(struct gendisk *disk, int partno)
{
T
Tejun Heo 已提交
540 541
	struct hd_struct *part;
	struct block_device *bdev = NULL;
542

T
Tejun Heo 已提交
543 544 545 546
	part = disk_get_part(disk, partno);
	if (part && (part->nr_sects || partno == 0))
		bdev = bdget(part_devt(part));
	disk_put_part(part);
547

T
Tejun Heo 已提交
548
	return bdev;
549 550 551
}
EXPORT_SYMBOL(bdget_disk);

552 553 554 555 556 557 558
/*
 * print a full list of all partitions - intended for places where the root
 * filesystem can't be mounted and thus to give the victim some idea of what
 * went wrong
 */
void __init printk_all_partitions(void)
{
559 560 561 562 563 564
	struct class_dev_iter iter;
	struct device *dev;

	class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
	while ((dev = class_dev_iter_next(&iter))) {
		struct gendisk *disk = dev_to_disk(dev);
565 566
		struct disk_part_iter piter;
		struct hd_struct *part;
567 568
		char name_buf[BDEVNAME_SIZE];
		char devt_buf[BDEVT_SIZE];
569 570 571 572 573 574 575 576 577 578 579 580 581 582

		/*
		 * Don't show empty devices or things that have been
		 * surpressed
		 */
		if (get_capacity(disk) == 0 ||
		    (disk->flags & GENHD_FL_SUPPRESS_PARTITION_INFO))
			continue;

		/*
		 * Note, unlike /proc/partitions, I am showing the
		 * numbers in hex - the same format as the root=
		 * option takes.
		 */
583 584
		printk("%s %10llu %s",
		       bdevt_str(disk_devt(disk), devt_buf),
585
		       (unsigned long long)get_capacity(disk) >> 1,
586
		       disk_name(disk, 0, name_buf));
587 588 589 590 591 592 593 594
		if (disk->driverfs_dev != NULL &&
		    disk->driverfs_dev->driver != NULL)
			printk(" driver: %s\n",
			       disk->driverfs_dev->driver->name);
		else
			printk(" (driver?)\n");

		/* now show the partitions */
595 596
		disk_part_iter_init(&piter, disk, 0);
		while ((part = disk_part_iter_next(&piter)))
597 598
			printk("  %s %10llu %s\n",
			       bdevt_str(part_devt(part), devt_buf),
599
			       (unsigned long long)part->nr_sects >> 1,
600
			       disk_name(disk, part->partno, name_buf));
601
		disk_part_iter_exit(&piter);
602 603
	}
	class_dev_iter_exit(&iter);
604 605
}

L
Linus Torvalds 已提交
606 607
#ifdef CONFIG_PROC_FS
/* iterator */
608
static void *disk_seqf_start(struct seq_file *seqf, loff_t *pos)
609
{
610 611 612
	loff_t skip = *pos;
	struct class_dev_iter *iter;
	struct device *dev;
613

614 615 616 617 618 619 620 621 622 623 624 625 626
	iter = kmalloc(GFP_KERNEL, sizeof(*iter));
	if (!iter)
		return ERR_PTR(-ENOMEM);

	seqf->private = iter;
	class_dev_iter_init(iter, &block_class, NULL, &disk_type);
	do {
		dev = class_dev_iter_next(iter);
		if (!dev)
			return NULL;
	} while (skip--);

	return dev_to_disk(dev);
627 628
}

629
static void *disk_seqf_next(struct seq_file *seqf, void *v, loff_t *pos)
L
Linus Torvalds 已提交
630
{
631
	struct device *dev;
L
Linus Torvalds 已提交
632

633 634
	(*pos)++;
	dev = class_dev_iter_next(seqf->private);
635
	if (dev)
636
		return dev_to_disk(dev);
637

L
Linus Torvalds 已提交
638 639 640
	return NULL;
}

641
static void disk_seqf_stop(struct seq_file *seqf, void *v)
642
{
643
	struct class_dev_iter *iter = seqf->private;
644

645 646 647 648
	/* stop is called even after start failed :-( */
	if (iter) {
		class_dev_iter_exit(iter);
		kfree(iter);
649
	}
L
Linus Torvalds 已提交
650 651
}

652
static void *show_partition_start(struct seq_file *seqf, loff_t *pos)
L
Linus Torvalds 已提交
653
{
654 655 656 657 658 659
	static void *p;

	p = disk_seqf_start(seqf, pos);
	if (!IS_ERR(p) && p)
		seq_puts(seqf, "major minor  #blocks  name\n\n");
	return p;
L
Linus Torvalds 已提交
660 661
}

662
static int show_partition(struct seq_file *seqf, void *v)
L
Linus Torvalds 已提交
663 664
{
	struct gendisk *sgp = v;
665 666
	struct disk_part_iter piter;
	struct hd_struct *part;
L
Linus Torvalds 已提交
667 668 669
	char buf[BDEVNAME_SIZE];

	/* Don't show non-partitionable removeable devices or empty devices */
T
Tejun Heo 已提交
670
	if (!get_capacity(sgp) || (!disk_partitionable(sgp) &&
671
				   (sgp->flags & GENHD_FL_REMOVABLE)))
L
Linus Torvalds 已提交
672 673 674 675 676
		return 0;
	if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
		return 0;

	/* show the full disk and all non-0 size partitions of it */
677
	seq_printf(seqf, "%4d  %7d %10llu %s\n",
678
		MAJOR(disk_devt(sgp)), MINOR(disk_devt(sgp)),
L
Linus Torvalds 已提交
679 680
		(unsigned long long)get_capacity(sgp) >> 1,
		disk_name(sgp, 0, buf));
681 682 683

	disk_part_iter_init(&piter, sgp, 0);
	while ((part = disk_part_iter_next(&piter)))
684
		seq_printf(seqf, "%4d  %7d %10llu %s\n",
685 686 687
			   MAJOR(part_devt(part)), MINOR(part_devt(part)),
			   (unsigned long long)part->nr_sects >> 1,
			   disk_name(sgp, part->partno, buf));
688
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
689 690 691 692

	return 0;
}

693
const struct seq_operations partitions_op = {
694 695 696
	.start	= show_partition_start,
	.next	= disk_seqf_next,
	.stop	= disk_seqf_stop,
697
	.show	= show_partition
L
Linus Torvalds 已提交
698 699 700 701
};
#endif


702
static struct kobject *base_probe(dev_t devt, int *partno, void *data)
L
Linus Torvalds 已提交
703
{
704
	if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
L
Linus Torvalds 已提交
705
		/* Make old-style 2.4 aliases work */
706
		request_module("block-major-%d", MAJOR(devt));
L
Linus Torvalds 已提交
707 708 709 710 711
	return NULL;
}

static int __init genhd_device_init(void)
{
712 713 714 715
	int error;

	block_class.dev_kobj = sysfs_dev_block_kobj;
	error = class_register(&block_class);
R
Roland McGrath 已提交
716 717
	if (unlikely(error))
		return error;
718
	bdev_map = kobj_map_init(base_probe, &block_class_lock);
L
Linus Torvalds 已提交
719
	blk_dev_init();
720 721 722 723 724

#ifndef CONFIG_SYSFS_DEPRECATED
	/* create top-level block dir */
	block_depr = kobject_create_and_add("block", NULL);
#endif
725
	return 0;
L
Linus Torvalds 已提交
726 727 728 729
}

subsys_initcall(genhd_device_init);

730 731
static ssize_t disk_range_show(struct device *dev,
			       struct device_attribute *attr, char *buf)
L
Linus Torvalds 已提交
732
{
733
	struct gendisk *disk = dev_to_disk(dev);
L
Linus Torvalds 已提交
734

735
	return sprintf(buf, "%d\n", disk->minors);
L
Linus Torvalds 已提交
736 737
}

738 739 740 741 742
static ssize_t disk_ext_range_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

T
Tejun Heo 已提交
743
	return sprintf(buf, "%d\n", disk_max_parts(disk));
744 745
}

746 747
static ssize_t disk_removable_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
748
{
749
	struct gendisk *disk = dev_to_disk(dev);
750

751 752
	return sprintf(buf, "%d\n",
		       (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
753 754
}

K
Kay Sievers 已提交
755 756 757 758 759 760 761 762
static ssize_t disk_ro_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%d\n", disk->policy ? 1 : 0);
}

763 764
static ssize_t disk_size_show(struct device *dev,
			      struct device_attribute *attr, char *buf)
765
{
766
	struct gendisk *disk = dev_to_disk(dev);
L
Linus Torvalds 已提交
767

768
	return sprintf(buf, "%llu\n", (unsigned long long)get_capacity(disk));
L
Linus Torvalds 已提交
769
}
770 771 772

static ssize_t disk_capability_show(struct device *dev,
				    struct device_attribute *attr, char *buf)
773
{
774 775 776
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%x\n", disk->flags);
777
}
778 779 780

static ssize_t disk_stat_show(struct device *dev,
			      struct device_attribute *attr, char *buf)
L
Linus Torvalds 已提交
781
{
782
	struct gendisk *disk = dev_to_disk(dev);
T
Tejun Heo 已提交
783
	int cpu;
784

T
Tejun Heo 已提交
785 786 787
	cpu = disk_stat_lock();
	disk_round_stats(cpu, disk);
	disk_stat_unlock();
788
	return sprintf(buf,
789 790
		"%8lu %8lu %8llu %8u "
		"%8lu %8lu %8llu %8u "
L
Linus Torvalds 已提交
791 792
		"%8u %8u %8u"
		"\n",
793 794 795 796 797 798 799 800
		disk_stat_read(disk, ios[READ]),
		disk_stat_read(disk, merges[READ]),
		(unsigned long long)disk_stat_read(disk, sectors[READ]),
		jiffies_to_msecs(disk_stat_read(disk, ticks[READ])),
		disk_stat_read(disk, ios[WRITE]),
		disk_stat_read(disk, merges[WRITE]),
		(unsigned long long)disk_stat_read(disk, sectors[WRITE]),
		jiffies_to_msecs(disk_stat_read(disk, ticks[WRITE])),
L
Linus Torvalds 已提交
801 802 803 804 805
		disk->in_flight,
		jiffies_to_msecs(disk_stat_read(disk, io_ticks)),
		jiffies_to_msecs(disk_stat_read(disk, time_in_queue)));
}

806
#ifdef CONFIG_FAIL_MAKE_REQUEST
807 808 809 810 811 812 813
static ssize_t disk_fail_show(struct device *dev,
			      struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%d\n", disk->flags & GENHD_FL_FAIL ? 1 : 0);
}
814

815 816
static ssize_t disk_fail_store(struct device *dev,
			       struct device_attribute *attr,
817 818
			       const char *buf, size_t count)
{
819
	struct gendisk *disk = dev_to_disk(dev);
820 821 822 823 824 825 826 827 828 829 830 831 832 833
	int i;

	if (count > 0 && sscanf(buf, "%d", &i) > 0) {
		if (i == 0)
			disk->flags &= ~GENHD_FL_FAIL;
		else
			disk->flags |= GENHD_FL_FAIL;
	}

	return count;
}

#endif

834
static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
835
static DEVICE_ATTR(ext_range, S_IRUGO, disk_ext_range_show, NULL);
836
static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
K
Kay Sievers 已提交
837
static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
838 839 840
static DEVICE_ATTR(size, S_IRUGO, disk_size_show, NULL);
static DEVICE_ATTR(capability, S_IRUGO, disk_capability_show, NULL);
static DEVICE_ATTR(stat, S_IRUGO, disk_stat_show, NULL);
841
#ifdef CONFIG_FAIL_MAKE_REQUEST
842 843
static struct device_attribute dev_attr_fail =
	__ATTR(make-it-fail, S_IRUGO|S_IWUSR, disk_fail_show, disk_fail_store);
844
#endif
845 846 847

static struct attribute *disk_attrs[] = {
	&dev_attr_range.attr,
848
	&dev_attr_ext_range.attr,
849
	&dev_attr_removable.attr,
K
Kay Sievers 已提交
850
	&dev_attr_ro.attr,
851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866
	&dev_attr_size.attr,
	&dev_attr_capability.attr,
	&dev_attr_stat.attr,
#ifdef CONFIG_FAIL_MAKE_REQUEST
	&dev_attr_fail.attr,
#endif
	NULL
};

static struct attribute_group disk_attr_group = {
	.attrs = disk_attrs,
};

static struct attribute_group *disk_attr_groups[] = {
	&disk_attr_group,
	NULL
L
Linus Torvalds 已提交
867 868
};

869
static void disk_release(struct device *dev)
L
Linus Torvalds 已提交
870
{
871 872
	struct gendisk *disk = dev_to_disk(dev);

L
Linus Torvalds 已提交
873
	kfree(disk->random);
874
	kfree(disk->__part);
L
Linus Torvalds 已提交
875 876 877
	free_disk_stats(disk);
	kfree(disk);
}
878 879
struct class block_class = {
	.name		= "block",
L
Linus Torvalds 已提交
880 881
};

A
Adrian Bunk 已提交
882
static struct device_type disk_type = {
883 884 885
	.name		= "disk",
	.groups		= disk_attr_groups,
	.release	= disk_release,
L
Linus Torvalds 已提交
886 887
};

888
#ifdef CONFIG_PROC_FS
889 890 891 892 893 894 895 896
/*
 * aggregate disk stat collector.  Uses the same stats that the sysfs
 * entries do, above, but makes them available through one seq_file.
 *
 * The output looks suspiciously like /proc/partitions with a bunch of
 * extra fields.
 */
static int diskstats_show(struct seq_file *seqf, void *v)
L
Linus Torvalds 已提交
897 898
{
	struct gendisk *gp = v;
899 900
	struct disk_part_iter piter;
	struct hd_struct *hd;
L
Linus Torvalds 已提交
901
	char buf[BDEVNAME_SIZE];
T
Tejun Heo 已提交
902
	int cpu;
L
Linus Torvalds 已提交
903 904

	/*
905
	if (&disk_to_dev(gp)->kobj.entry == block_class.devices.next)
906
		seq_puts(seqf,	"major minor name"
L
Linus Torvalds 已提交
907 908 909 910 911
				"     rio rmerge rsect ruse wio wmerge "
				"wsect wuse running use aveq"
				"\n\n");
	*/
 
T
Tejun Heo 已提交
912 913 914
	cpu = disk_stat_lock();
	disk_round_stats(cpu, gp);
	disk_stat_unlock();
915
	seq_printf(seqf, "%4d %7d %s %lu %lu %llu %u %lu %lu %llu %u %u %u %u\n",
916 917
		MAJOR(disk_devt(gp)), MINOR(disk_devt(gp)),
		disk_name(gp, 0, buf),
918 919 920 921 922 923
		disk_stat_read(gp, ios[0]), disk_stat_read(gp, merges[0]),
		(unsigned long long)disk_stat_read(gp, sectors[0]),
		jiffies_to_msecs(disk_stat_read(gp, ticks[0])),
		disk_stat_read(gp, ios[1]), disk_stat_read(gp, merges[1]),
		(unsigned long long)disk_stat_read(gp, sectors[1]),
		jiffies_to_msecs(disk_stat_read(gp, ticks[1])),
L
Linus Torvalds 已提交
924 925 926 927 928
		gp->in_flight,
		jiffies_to_msecs(disk_stat_read(gp, io_ticks)),
		jiffies_to_msecs(disk_stat_read(gp, time_in_queue)));

	/* now show all non-0 size partitions of it */
929 930
	disk_part_iter_init(&piter, gp, 0);
	while ((hd = disk_part_iter_next(&piter))) {
T
Tejun Heo 已提交
931 932 933
		cpu = disk_stat_lock();
		part_round_stats(cpu, hd);
		disk_stat_unlock();
934
		seq_printf(seqf, "%4d %7d %s %lu %lu %llu "
935
			   "%u %lu %lu %llu %u %u %u %u\n",
936 937
			   MAJOR(part_devt(hd)), MINOR(part_devt(hd)),
			   disk_name(gp, hd->partno, buf),
938 939 940 941 942 943 944 945 946 947 948 949
			   part_stat_read(hd, ios[0]),
			   part_stat_read(hd, merges[0]),
			   (unsigned long long)part_stat_read(hd, sectors[0]),
			   jiffies_to_msecs(part_stat_read(hd, ticks[0])),
			   part_stat_read(hd, ios[1]),
			   part_stat_read(hd, merges[1]),
			   (unsigned long long)part_stat_read(hd, sectors[1]),
			   jiffies_to_msecs(part_stat_read(hd, ticks[1])),
			   hd->in_flight,
			   jiffies_to_msecs(part_stat_read(hd, io_ticks)),
			   jiffies_to_msecs(part_stat_read(hd, time_in_queue))
			);
L
Linus Torvalds 已提交
950
	}
951
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
952 953 954 955
 
	return 0;
}

956
const struct seq_operations diskstats_op = {
957 958 959
	.start	= disk_seqf_start,
	.next	= disk_seqf_next,
	.stop	= disk_seqf_stop,
L
Linus Torvalds 已提交
960 961
	.show	= diskstats_show
};
962
#endif /* CONFIG_PROC_FS */
L
Linus Torvalds 已提交
963

964 965 966 967 968 969 970 971 972 973
static void media_change_notify_thread(struct work_struct *work)
{
	struct gendisk *gd = container_of(work, struct gendisk, async_notify);
	char event[] = "MEDIA_CHANGE=1";
	char *envp[] = { event, NULL };

	/*
	 * set enviroment vars to indicate which event this is for
	 * so that user space will know to go check the media status.
	 */
974
	kobject_uevent_env(&disk_to_dev(gd)->kobj, KOBJ_CHANGE, envp);
975 976 977
	put_device(gd->driverfs_dev);
}

A
Adrian Bunk 已提交
978
#if 0
979 980 981 982 983 984
void genhd_media_change_notify(struct gendisk *disk)
{
	get_device(disk->driverfs_dev);
	schedule_work(&disk->async_notify);
}
EXPORT_SYMBOL_GPL(genhd_media_change_notify);
A
Adrian Bunk 已提交
985
#endif  /*  0  */
986

987
dev_t blk_lookup_devt(const char *name, int partno)
988
{
989 990 991
	dev_t devt = MKDEV(0, 0);
	struct class_dev_iter iter;
	struct device *dev;
992

993 994
	class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
	while ((dev = class_dev_iter_next(&iter))) {
995
		struct gendisk *disk = dev_to_disk(dev);
T
Tejun Heo 已提交
996
		struct hd_struct *part;
997

998 999 1000
		if (strcmp(dev->bus_id, name))
			continue;

T
Tejun Heo 已提交
1001 1002
		part = disk_get_part(disk, partno);
		if (part && (part->nr_sects || partno == 0)) {
1003
			devt = part_devt(part);
1004
			disk_put_part(part);
T
Tejun Heo 已提交
1005
			break;
1006
		}
T
Tejun Heo 已提交
1007
		disk_put_part(part);
1008
	}
1009
	class_dev_iter_exit(&iter);
1010 1011 1012 1013
	return devt;
}
EXPORT_SYMBOL(blk_lookup_devt);

L
Linus Torvalds 已提交
1014 1015
struct gendisk *alloc_disk(int minors)
{
1016 1017 1018 1019
	return alloc_disk_node(minors, -1);
}

struct gendisk *alloc_disk_node(int minors, int node_id)
T
Tejun Heo 已提交
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
{
	return alloc_disk_ext_node(minors, 0, node_id);
}

struct gendisk *alloc_disk_ext(int minors, int ext_minors)
{
	return alloc_disk_ext_node(minors, ext_minors, -1);
}

struct gendisk *alloc_disk_ext_node(int minors, int ext_minors, int node_id)
1030 1031 1032
{
	struct gendisk *disk;

1033 1034
	disk = kmalloc_node(sizeof(struct gendisk),
				GFP_KERNEL | __GFP_ZERO, node_id);
L
Linus Torvalds 已提交
1035
	if (disk) {
T
Tejun Heo 已提交
1036
		int tot_minors = minors + ext_minors;
T
Tejun Heo 已提交
1037
		int size = tot_minors * sizeof(struct hd_struct *);
T
Tejun Heo 已提交
1038

L
Linus Torvalds 已提交
1039 1040 1041 1042
		if (!init_disk_stats(disk)) {
			kfree(disk);
			return NULL;
		}
T
Tejun Heo 已提交
1043 1044 1045 1046 1047 1048 1049

		disk->__part = kmalloc_node(size, GFP_KERNEL | __GFP_ZERO,
					    node_id);
		if (!disk->__part) {
			free_disk_stats(disk);
			kfree(disk);
			return NULL;
L
Linus Torvalds 已提交
1050
		}
T
Tejun Heo 已提交
1051 1052
		disk->__part[0] = &disk->part0;

L
Linus Torvalds 已提交
1053
		disk->minors = minors;
T
Tejun Heo 已提交
1054
		disk->ext_minors = ext_minors;
L
Linus Torvalds 已提交
1055
		rand_initialize_disk(disk);
1056 1057 1058
		disk_to_dev(disk)->class = &block_class;
		disk_to_dev(disk)->type = &disk_type;
		device_initialize(disk_to_dev(disk));
1059 1060
		INIT_WORK(&disk->async_notify,
			media_change_notify_thread);
L
Linus Torvalds 已提交
1061 1062 1063 1064 1065
	}
	return disk;
}

EXPORT_SYMBOL(alloc_disk);
1066
EXPORT_SYMBOL(alloc_disk_node);
T
Tejun Heo 已提交
1067 1068
EXPORT_SYMBOL(alloc_disk_ext);
EXPORT_SYMBOL(alloc_disk_ext_node);
L
Linus Torvalds 已提交
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079

struct kobject *get_disk(struct gendisk *disk)
{
	struct module *owner;
	struct kobject *kobj;

	if (!disk->fops)
		return NULL;
	owner = disk->fops->owner;
	if (owner && !try_module_get(owner))
		return NULL;
1080
	kobj = kobject_get(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093
	if (kobj == NULL) {
		module_put(owner);
		return NULL;
	}
	return kobj;

}

EXPORT_SYMBOL(get_disk);

void put_disk(struct gendisk *disk)
{
	if (disk)
1094
		kobject_put(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
}

EXPORT_SYMBOL(put_disk);

void set_device_ro(struct block_device *bdev, int flag)
{
	if (bdev->bd_contains != bdev)
		bdev->bd_part->policy = flag;
	else
		bdev->bd_disk->policy = flag;
}

EXPORT_SYMBOL(set_device_ro);

void set_disk_ro(struct gendisk *disk, int flag)
{
1111 1112 1113
	struct disk_part_iter piter;
	struct hd_struct *part;

L
Linus Torvalds 已提交
1114
	disk->policy = flag;
1115 1116 1117 1118
	disk_part_iter_init(&piter, disk, DISK_PITER_INCL_EMPTY);
	while ((part = disk_part_iter_next(&piter)))
		part->policy = flag;
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134
}

EXPORT_SYMBOL(set_disk_ro);

int bdev_read_only(struct block_device *bdev)
{
	if (!bdev)
		return 0;
	else if (bdev->bd_contains != bdev)
		return bdev->bd_part->policy;
	else
		return bdev->bd_disk->policy;
}

EXPORT_SYMBOL(bdev_read_only);

1135
int invalidate_partition(struct gendisk *disk, int partno)
L
Linus Torvalds 已提交
1136 1137
{
	int res = 0;
1138
	struct block_device *bdev = bdget_disk(disk, partno);
L
Linus Torvalds 已提交
1139
	if (bdev) {
1140 1141
		fsync_bdev(bdev);
		res = __invalidate_device(bdev);
L
Linus Torvalds 已提交
1142 1143 1144 1145 1146 1147
		bdput(bdev);
	}
	return res;
}

EXPORT_SYMBOL(invalidate_partition);