genhd.c 26.7 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 57 58 59 60 61
/**
 * 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;

	if (unlikely(partno < 1 || partno > disk_max_parts(disk)))
		return NULL;
	rcu_read_lock();
	part = rcu_dereference(disk->__part[partno - 1]);
	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 88 89 90 91 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 119 120 121 122 123 124 125 126 127 128 129 130 131 132
	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;
	else
		piter->idx = 0;

	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;
		end = -1;
	} 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;

133
		get_device(part_to_dev(part));
134 135 136 137 138 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 185 186 187 188 189 190
		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;

	for (i = 0; i < disk_max_parts(disk); i++) {
		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 已提交
191 192 193 194 195 196 197 198
/*
 * Can be deleted altogether. Later.
 *
 */
static struct blk_major_name {
	struct blk_major_name *next;
	int major;
	char name[16];
199
} *major_names[BLKDEV_MAJOR_HASH_SIZE];
L
Linus Torvalds 已提交
200 201 202 203

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

207
#ifdef CONFIG_PROC_FS
208
void blkdev_show(struct seq_file *seqf, off_t offset)
209
{
210
	struct blk_major_name *dp;
211

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

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

226
	mutex_lock(&block_class_lock);
L
Linus Torvalds 已提交
227 228 229 230 231 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

	/* 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:
271
	mutex_unlock(&block_class_lock);
L
Linus Torvalds 已提交
272 273 274 275 276
	return ret;
}

EXPORT_SYMBOL(register_blkdev);

A
Akinobu Mita 已提交
277
void unregister_blkdev(unsigned int major, const char *name)
L
Linus Torvalds 已提交
278 279 280 281 282
{
	struct blk_major_name **n;
	struct blk_major_name *p = NULL;
	int index = major_to_index(major);

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

EXPORT_SYMBOL(unregister_blkdev);

static struct kobj_map *bdev_map;

301 302 303 304 305 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
/**
 * 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 已提交
333 334 335 336 337 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
/**
 * 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;
	}

374
	*devt = MKDEV(BLOCK_EXT_MAJOR, blk_mangle_minor(idx));
T
Tejun Heo 已提交
375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395
	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);
396
		idr_remove(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
T
Tejun Heo 已提交
397 398 399 400
		mutex_unlock(&ext_devt_mutex);
	}
}

401 402 403 404 405 406 407 408 409 410 411 412
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 已提交
413 414 415 416 417
/*
 * Register device numbers dev..(dev+range-1)
 * range must be nonzero
 * The hash chain is sorted on range, so that subranges can override.
 */
418
void blk_register_region(dev_t devt, unsigned long range, struct module *module,
L
Linus Torvalds 已提交
419 420 421
			 struct kobject *(*probe)(dev_t, int *, void *),
			 int (*lock)(dev_t, void *), void *data)
{
422
	kobj_map(bdev_map, devt, range, module, probe, lock, data);
L
Linus Torvalds 已提交
423 424 425 426
}

EXPORT_SYMBOL(blk_register_region);

427
void blk_unregister_region(dev_t devt, unsigned long range)
L
Linus Torvalds 已提交
428
{
429
	kobj_unmap(bdev_map, devt, range);
L
Linus Torvalds 已提交
430 431 432 433
}

EXPORT_SYMBOL(blk_unregister_region);

434
static struct kobject *exact_match(dev_t devt, int *partno, void *data)
L
Linus Torvalds 已提交
435 436
{
	struct gendisk *p = data;
437

438
	return &disk_to_dev(p)->kobj;
L
Linus Torvalds 已提交
439 440
}

441
static int exact_lock(dev_t devt, void *data)
L
Linus Torvalds 已提交
442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458
{
	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)
{
459
	struct backing_dev_info *bdi;
460
	int retval;
461

L
Linus Torvalds 已提交
462
	disk->flags |= GENHD_FL_UP;
463
	disk_to_dev(disk)->devt = MKDEV(disk->major, disk->first_minor);
464 465
	blk_register_region(disk_devt(disk), disk->minors, NULL,
			    exact_match, exact_lock, disk);
L
Linus Torvalds 已提交
466 467
	register_disk(disk);
	blk_register_queue(disk);
468 469

	bdi = &disk->queue->backing_dev_info;
470
	bdi_register_dev(bdi, disk_devt(disk));
471 472
	retval = sysfs_create_link(&disk_to_dev(disk)->kobj, &bdi->dev->kobj,
				   "bdi");
473
	WARN_ON(retval);
L
Linus Torvalds 已提交
474 475 476 477 478 479 480
}

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

void unlink_gendisk(struct gendisk *disk)
{
481
	sysfs_remove_link(&disk_to_dev(disk)->kobj, "bdi");
482
	bdi_unregister(&disk->queue->backing_dev_info);
L
Linus Torvalds 已提交
483
	blk_unregister_queue(disk);
484
	blk_unregister_region(disk_devt(disk), disk->minors);
L
Linus Torvalds 已提交
485 486 487 488
}

/**
 * get_gendisk - get partitioning information for a given device
489 490
 * @devt: device to get partitioning information for
 * @part: returned partition index
L
Linus Torvalds 已提交
491 492
 *
 * This function gets the structure containing partitioning
493
 * information for the given device @devt.
L
Linus Torvalds 已提交
494
 */
495
struct gendisk *get_gendisk(dev_t devt, int *partno)
L
Linus Torvalds 已提交
496
{
T
Tejun Heo 已提交
497 498 499 500 501 502 503 504 505 506 507 508
	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);
509
		part = idr_find(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
T
Tejun Heo 已提交
510 511 512 513 514 515
		if (part && get_disk(part_to_disk(part))) {
			*partno = part->partno;
			disk = part_to_disk(part);
		}
		mutex_unlock(&ext_devt_mutex);
	}
516

T
Tejun Heo 已提交
517
	return disk;
L
Linus Torvalds 已提交
518 519
}

520 521 522 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)
{
	dev_t devt = MKDEV(0, 0);

	if (partno == 0)
		devt = disk_devt(disk);
	else {
540
		struct hd_struct *part;
541

542
		part = disk_get_part(disk, partno);
543 544
		if (part && part->nr_sects)
			devt = part_devt(part);
545
		disk_put_part(part);
546 547 548 549 550 551 552 553
	}

	if (likely(devt != MKDEV(0, 0)))
		return bdget(devt);
	return NULL;
}
EXPORT_SYMBOL(bdget_disk);

554 555 556 557 558 559 560
/*
 * 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)
{
561 562 563 564 565 566
	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);
567 568
		struct disk_part_iter piter;
		struct hd_struct *part;
569 570
		char name_buf[BDEVNAME_SIZE];
		char devt_buf[BDEVT_SIZE];
571 572 573 574 575 576 577 578 579 580 581 582 583 584

		/*
		 * 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.
		 */
585 586
		printk("%s %10llu %s",
		       bdevt_str(disk_devt(disk), devt_buf),
587
		       (unsigned long long)get_capacity(disk) >> 1,
588
		       disk_name(disk, 0, name_buf));
589 590 591 592 593 594 595 596
		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 */
597 598
		disk_part_iter_init(&piter, disk, 0);
		while ((part = disk_part_iter_next(&piter)))
599 600
			printk("  %s %10llu %s\n",
			       bdevt_str(part_devt(part), devt_buf),
601
			       (unsigned long long)part->nr_sects >> 1,
602
			       disk_name(disk, part->partno, name_buf));
603
		disk_part_iter_exit(&piter);
604 605
	}
	class_dev_iter_exit(&iter);
606 607
}

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

616 617 618 619 620 621 622 623 624 625 626 627 628
	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);
629 630
}

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

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

L
Linus Torvalds 已提交
640 641 642
	return NULL;
}

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

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

654
static void *show_partition_start(struct seq_file *seqf, loff_t *pos)
L
Linus Torvalds 已提交
655
{
656 657 658 659 660 661
	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 已提交
662 663
}

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

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

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

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

	return 0;
}

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


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

static int __init genhd_device_init(void)
{
714 715 716 717
	int error;

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

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

subsys_initcall(genhd_device_init);

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

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

740 741 742 743 744 745 746 747
static ssize_t disk_ext_range_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%d\n", disk_max_parts(disk) + 1);
}

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

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

K
Kay Sievers 已提交
757 758 759 760 761 762 763 764
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);
}

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

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

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

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

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

T
Tejun Heo 已提交
787 788 789
	cpu = disk_stat_lock();
	disk_round_stats(cpu, disk);
	disk_stat_unlock();
790
	return sprintf(buf,
791 792
		"%8lu %8lu %8llu %8u "
		"%8lu %8lu %8llu %8u "
L
Linus Torvalds 已提交
793 794
		"%8u %8u %8u"
		"\n",
795 796 797 798 799 800 801 802
		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 已提交
803 804 805 806 807
		disk->in_flight,
		jiffies_to_msecs(disk_stat_read(disk, io_ticks)),
		jiffies_to_msecs(disk_stat_read(disk, time_in_queue)));
}

808
#ifdef CONFIG_FAIL_MAKE_REQUEST
809 810 811 812 813 814 815
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);
}
816

817 818
static ssize_t disk_fail_store(struct device *dev,
			       struct device_attribute *attr,
819 820
			       const char *buf, size_t count)
{
821
	struct gendisk *disk = dev_to_disk(dev);
822 823 824 825 826 827 828 829 830 831 832 833 834 835
	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

836
static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
837
static DEVICE_ATTR(ext_range, S_IRUGO, disk_ext_range_show, NULL);
838
static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
K
Kay Sievers 已提交
839
static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
840 841 842
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);
843
#ifdef CONFIG_FAIL_MAKE_REQUEST
844 845
static struct device_attribute dev_attr_fail =
	__ATTR(make-it-fail, S_IRUGO|S_IWUSR, disk_fail_show, disk_fail_store);
846
#endif
847 848 849

static struct attribute *disk_attrs[] = {
	&dev_attr_range.attr,
850
	&dev_attr_ext_range.attr,
851
	&dev_attr_removable.attr,
K
Kay Sievers 已提交
852
	&dev_attr_ro.attr,
853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868
	&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 已提交
869 870
};

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

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

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

890
#ifdef CONFIG_PROC_FS
891 892 893 894 895 896 897 898
/*
 * 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 已提交
899 900
{
	struct gendisk *gp = v;
901 902
	struct disk_part_iter piter;
	struct hd_struct *hd;
L
Linus Torvalds 已提交
903
	char buf[BDEVNAME_SIZE];
T
Tejun Heo 已提交
904
	int cpu;
L
Linus Torvalds 已提交
905 906

	/*
907
	if (&disk_to_dev(gp)->kobj.entry == block_class.devices.next)
908
		seq_puts(seqf,	"major minor name"
L
Linus Torvalds 已提交
909 910 911 912 913
				"     rio rmerge rsect ruse wio wmerge "
				"wsect wuse running use aveq"
				"\n\n");
	*/
 
T
Tejun Heo 已提交
914 915 916
	cpu = disk_stat_lock();
	disk_round_stats(cpu, gp);
	disk_stat_unlock();
917
	seq_printf(seqf, "%4d %7d %s %lu %lu %llu %u %lu %lu %llu %u %u %u %u\n",
918 919
		MAJOR(disk_devt(gp)), MINOR(disk_devt(gp)),
		disk_name(gp, 0, buf),
920 921 922 923 924 925
		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 已提交
926 927 928 929 930
		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 */
931 932
	disk_part_iter_init(&piter, gp, 0);
	while ((hd = disk_part_iter_next(&piter))) {
T
Tejun Heo 已提交
933 934 935
		cpu = disk_stat_lock();
		part_round_stats(cpu, hd);
		disk_stat_unlock();
936
		seq_printf(seqf, "%4d %7d %s %lu %lu %llu "
937
			   "%u %lu %lu %llu %u %u %u %u\n",
938 939
			   MAJOR(part_devt(hd)), MINOR(part_devt(hd)),
			   disk_name(gp, hd->partno, buf),
940 941 942 943 944 945 946 947 948 949 950 951
			   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 已提交
952
	}
953
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
954 955 956 957
 
	return 0;
}

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

966 967 968 969 970 971 972 973 974 975
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.
	 */
976
	kobject_uevent_env(&disk_to_dev(gd)->kobj, KOBJ_CHANGE, envp);
977 978 979
	put_device(gd->driverfs_dev);
}

A
Adrian Bunk 已提交
980
#if 0
981 982 983 984 985 986
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 已提交
987
#endif  /*  0  */
988

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

995 996
	class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
	while ((dev = class_dev_iter_next(&iter))) {
997 998
		struct gendisk *disk = dev_to_disk(dev);

999 1000 1001 1002 1003 1004 1005 1006
		if (strcmp(dev->bus_id, name))
			continue;
		if (partno < 0 || partno > disk_max_parts(disk))
			continue;

		if (partno == 0)
			devt = disk_devt(disk);
		else {
1007
			struct hd_struct *part;
1008

1009 1010 1011
			part = disk_get_part(disk, partno);
			if (!part || !part->nr_sects) {
				disk_put_part(part);
1012
				continue;
1013
			}
1014 1015

			devt = part_devt(part);
1016
			disk_put_part(part);
1017
		}
1018
		break;
1019
	}
1020
	class_dev_iter_exit(&iter);
1021 1022 1023 1024
	return devt;
}
EXPORT_SYMBOL(blk_lookup_devt);

L
Linus Torvalds 已提交
1025 1026
struct gendisk *alloc_disk(int minors)
{
1027 1028 1029 1030
	return alloc_disk_node(minors, -1);
}

struct gendisk *alloc_disk_node(int minors, int node_id)
T
Tejun Heo 已提交
1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
{
	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)
1041 1042 1043
{
	struct gendisk *disk;

1044 1045
	disk = kmalloc_node(sizeof(struct gendisk),
				GFP_KERNEL | __GFP_ZERO, node_id);
L
Linus Torvalds 已提交
1046
	if (disk) {
T
Tejun Heo 已提交
1047 1048
		int tot_minors = minors + ext_minors;

L
Linus Torvalds 已提交
1049 1050 1051 1052
		if (!init_disk_stats(disk)) {
			kfree(disk);
			return NULL;
		}
T
Tejun Heo 已提交
1053 1054
		if (tot_minors > 1) {
			int size = (tot_minors - 1) * sizeof(struct hd_struct *);
1055
			disk->__part = kmalloc_node(size,
1056
				GFP_KERNEL | __GFP_ZERO, node_id);
1057
			if (!disk->__part) {
1058
				free_disk_stats(disk);
L
Linus Torvalds 已提交
1059 1060 1061 1062 1063
				kfree(disk);
				return NULL;
			}
		}
		disk->minors = minors;
T
Tejun Heo 已提交
1064
		disk->ext_minors = ext_minors;
L
Linus Torvalds 已提交
1065
		rand_initialize_disk(disk);
1066 1067 1068
		disk_to_dev(disk)->class = &block_class;
		disk_to_dev(disk)->type = &disk_type;
		device_initialize(disk_to_dev(disk));
1069 1070
		INIT_WORK(&disk->async_notify,
			media_change_notify_thread);
L
Linus Torvalds 已提交
1071 1072 1073 1074 1075
	}
	return disk;
}

EXPORT_SYMBOL(alloc_disk);
1076
EXPORT_SYMBOL(alloc_disk_node);
T
Tejun Heo 已提交
1077 1078
EXPORT_SYMBOL(alloc_disk_ext);
EXPORT_SYMBOL(alloc_disk_ext_node);
L
Linus Torvalds 已提交
1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089

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;
1090
	kobj = kobject_get(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
	if (kobj == NULL) {
		module_put(owner);
		return NULL;
	}
	return kobj;

}

EXPORT_SYMBOL(get_disk);

void put_disk(struct gendisk *disk)
{
	if (disk)
1104
		kobject_put(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120
}

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)
{
1121 1122 1123
	struct disk_part_iter piter;
	struct hd_struct *part;

L
Linus Torvalds 已提交
1124
	disk->policy = flag;
1125 1126 1127 1128
	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 已提交
1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144
}

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);

1145
int invalidate_partition(struct gendisk *disk, int partno)
L
Linus Torvalds 已提交
1146 1147
{
	int res = 0;
1148
	struct block_device *bdev = bdget_disk(disk, partno);
L
Linus Torvalds 已提交
1149
	if (bdev) {
1150 1151
		fsync_bdev(bdev);
		res = __invalidate_device(bdev);
L
Linus Torvalds 已提交
1152 1153 1154 1155 1156 1157
		bdput(bdev);
	}
	return res;
}

EXPORT_SYMBOL(invalidate_partition);