genhd.c 60.1 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0
L
Linus Torvalds 已提交
2 3 4 5 6
/*
 *  gendisk handling
 */

#include <linux/module.h>
7
#include <linux/ctype.h>
L
Linus Torvalds 已提交
8 9
#include <linux/fs.h>
#include <linux/genhd.h>
10
#include <linux/kdev_t.h>
L
Linus Torvalds 已提交
11 12
#include <linux/kernel.h>
#include <linux/blkdev.h>
13
#include <linux/backing-dev.h>
L
Linus Torvalds 已提交
14 15
#include <linux/init.h>
#include <linux/spinlock.h>
16
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
17 18 19
#include <linux/seq_file.h>
#include <linux/slab.h>
#include <linux/kmod.h>
20
#include <linux/mutex.h>
T
Tejun Heo 已提交
21
#include <linux/idr.h>
22
#include <linux/log2.h>
23
#include <linux/pm_runtime.h>
24
#include <linux/badblocks.h>
L
Linus Torvalds 已提交
25

26 27
#include "blk.h"

28
static struct kobject *block_depr;
L
Linus Torvalds 已提交
29

30 31 32 33 34 35 36 37 38
struct bdev_map {
	struct bdev_map *next;
	dev_t dev;
	unsigned long range;
	struct module *owner;
	struct kobject *(*probe)(dev_t, int *, void *);
	int (*lock)(dev_t, void *);
	void *data;
} *bdev_map[255];
C
Christoph Hellwig 已提交
39
static DEFINE_MUTEX(bdev_map_lock);
40

T
Tejun Heo 已提交
41
/* for extended dynamic devt allocation, currently only one major is used */
42
#define NR_EXT_DEVT		(1 << MINORBITS)
T
Tejun Heo 已提交
43

44
/* For extended devt allocation.  ext_devt_lock prevents look up
T
Tejun Heo 已提交
45 46
 * results from going away underneath its user.
 */
47
static DEFINE_SPINLOCK(ext_devt_lock);
T
Tejun Heo 已提交
48 49
static DEFINE_IDR(ext_devt_idr);

D
Derek Basehore 已提交
50 51
static void disk_check_events(struct disk_events *ev,
			      unsigned int *clearing_ptr);
52
static void disk_alloc_events(struct gendisk *disk);
53 54 55 56
static void disk_add_events(struct gendisk *disk);
static void disk_del_events(struct gendisk *disk);
static void disk_release_events(struct gendisk *disk);

57 58 59 60
/*
 * Set disk capacity and notify if the size is not currently
 * zero and will not be set to zero
 */
61
bool set_capacity_revalidate_and_notify(struct gendisk *disk, sector_t size,
62
					bool update_bdev)
63 64 65 66
{
	sector_t capacity = get_capacity(disk);

	set_capacity(disk, size);
67 68
	if (update_bdev)
		revalidate_disk_size(disk, true);
69 70 71 72 73

	if (capacity != size && capacity != 0 && size != 0) {
		char *envp[] = { "RESIZE=1", NULL };

		kobject_uevent_env(&disk_to_dev(disk)->kobj, KOBJ_CHANGE, envp);
74
		return true;
75
	}
76 77

	return false;
78 79 80 81
}

EXPORT_SYMBOL_GPL(set_capacity_revalidate_and_notify);

82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
/*
 * Format the device name of the indicated disk into the supplied buffer and
 * return a pointer to that same buffer for convenience.
 */
char *disk_name(struct gendisk *hd, int partno, char *buf)
{
	if (!partno)
		snprintf(buf, BDEVNAME_SIZE, "%s", hd->disk_name);
	else if (isdigit(hd->disk_name[strlen(hd->disk_name)-1]))
		snprintf(buf, BDEVNAME_SIZE, "%sp%d", hd->disk_name, partno);
	else
		snprintf(buf, BDEVNAME_SIZE, "%s%d", hd->disk_name, partno);

	return buf;
}

const char *bdevname(struct block_device *bdev, char *buf)
{
100
	return disk_name(bdev->bd_disk, bdev->bd_partno, buf);
101 102
}
EXPORT_SYMBOL(bdevname);
103

104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
static void part_stat_read_all(struct hd_struct *part, struct disk_stats *stat)
{
	int cpu;

	memset(stat, 0, sizeof(struct disk_stats));
	for_each_possible_cpu(cpu) {
		struct disk_stats *ptr = per_cpu_ptr(part->dkstats, cpu);
		int group;

		for (group = 0; group < NR_STAT_GROUPS; group++) {
			stat->nsecs[group] += ptr->nsecs[group];
			stat->sectors[group] += ptr->sectors[group];
			stat->ios[group] += ptr->ios[group];
			stat->merges[group] += ptr->merges[group];
		}

		stat->io_ticks += ptr->io_ticks;
	}
}

124
static unsigned int part_in_flight(struct hd_struct *part)
125
{
126
	unsigned int inflight = 0;
127
	int cpu;
128

129
	for_each_possible_cpu(cpu) {
130 131
		inflight += part_stat_local_read_cpu(part, in_flight[0], cpu) +
			    part_stat_local_read_cpu(part, in_flight[1], cpu);
132
	}
133 134
	if ((int)inflight < 0)
		inflight = 0;
135

136
	return inflight;
137 138
}

139
static void part_in_flight_rw(struct hd_struct *part, unsigned int inflight[2])
140
{
141 142 143 144 145 146 147 148 149 150 151 152
	int cpu;

	inflight[0] = 0;
	inflight[1] = 0;
	for_each_possible_cpu(cpu) {
		inflight[0] += part_stat_local_read_cpu(part, in_flight[0], cpu);
		inflight[1] += part_stat_local_read_cpu(part, in_flight[1], cpu);
	}
	if ((int)inflight[0] < 0)
		inflight[0] = 0;
	if ((int)inflight[1] < 0)
		inflight[1] = 0;
153 154
}

155 156 157 158 159 160 161 162 163
struct hd_struct *__disk_get_part(struct gendisk *disk, int partno)
{
	struct disk_part_tbl *ptbl = rcu_dereference(disk->part_tbl);

	if (unlikely(partno < 0 || partno >= ptbl->len))
		return NULL;
	return rcu_dereference(ptbl->part[partno]);
}

164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
/**
 * 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)
{
180
	struct hd_struct *part;
T
Tejun Heo 已提交
181

182
	rcu_read_lock();
183 184 185
	part = __disk_get_part(disk, partno);
	if (part)
		get_device(part_to_dev(part));
186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
	rcu_read_unlock();

	return 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)
{
T
Tejun Heo 已提交
205 206 207 208 209
	struct disk_part_tbl *ptbl;

	rcu_read_lock();
	ptbl = rcu_dereference(disk->part_tbl);

210 211 212 213
	piter->disk = disk;
	piter->part = NULL;

	if (flags & DISK_PITER_REVERSE)
T
Tejun Heo 已提交
214
		piter->idx = ptbl->len - 1;
215
	else if (flags & (DISK_PITER_INCL_PART0 | DISK_PITER_INCL_EMPTY_PART0))
216
		piter->idx = 0;
T
Tejun Heo 已提交
217 218
	else
		piter->idx = 1;
219 220

	piter->flags = flags;
T
Tejun Heo 已提交
221 222

	rcu_read_unlock();
223 224 225 226 227 228 229 230 231 232 233 234 235 236
}
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)
{
T
Tejun Heo 已提交
237
	struct disk_part_tbl *ptbl;
238 239 240 241 242 243
	int inc, end;

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

T
Tejun Heo 已提交
244
	/* get part_tbl */
245
	rcu_read_lock();
T
Tejun Heo 已提交
246
	ptbl = rcu_dereference(piter->disk->part_tbl);
247 248 249 250

	/* determine iteration parameters */
	if (piter->flags & DISK_PITER_REVERSE) {
		inc = -1;
251 252
		if (piter->flags & (DISK_PITER_INCL_PART0 |
				    DISK_PITER_INCL_EMPTY_PART0))
T
Tejun Heo 已提交
253 254 255
			end = -1;
		else
			end = 0;
256 257
	} else {
		inc = 1;
T
Tejun Heo 已提交
258
		end = ptbl->len;
259 260 261 262 263 264
	}

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

T
Tejun Heo 已提交
265
		part = rcu_dereference(ptbl->part[piter->idx]);
266 267
		if (!part)
			continue;
268
		if (!part_nr_sects_read(part) &&
269 270 271
		    !(piter->flags & DISK_PITER_INCL_EMPTY) &&
		    !(piter->flags & DISK_PITER_INCL_EMPTY_PART0 &&
		      piter->idx == 0))
272 273
			continue;

274
		get_device(part_to_dev(part));
275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301
		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);

302 303 304
static inline int sector_in_part(struct hd_struct *part, sector_t sector)
{
	return part->start_sect <= sector &&
305
		sector < part->start_sect + part_nr_sects_read(part);
306 307
}

308 309 310 311 312 313 314 315 316
/**
 * 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:
317
 * RCU read locked.  The returned partition pointer is always valid
318 319
 * because its refcount is grabbed except for part0, which lifetime
 * is same with the disk.
320 321
 *
 * RETURNS:
T
Tejun Heo 已提交
322
 * Found partition on success, part0 is returned if no partition matches
323
 * or the matched partition is being deleted.
324 325 326
 */
struct hd_struct *disk_map_sector_rcu(struct gendisk *disk, sector_t sector)
{
T
Tejun Heo 已提交
327
	struct disk_part_tbl *ptbl;
328
	struct hd_struct *part;
329 330
	int i;

331
	rcu_read_lock();
T
Tejun Heo 已提交
332 333
	ptbl = rcu_dereference(disk->part_tbl);

334
	part = rcu_dereference(ptbl->last_lookup);
335
	if (part && sector_in_part(part, sector) && hd_struct_try_get(part))
336
		goto out_unlock;
337

T
Tejun Heo 已提交
338
	for (i = 1; i < ptbl->len; i++) {
339
		part = rcu_dereference(ptbl->part[i]);
340

341
		if (part && sector_in_part(part, sector)) {
342 343 344 345 346 347 348
			/*
			 * only live partition can be cached for lookup,
			 * so use-after-free on cached & deleting partition
			 * can be avoided
			 */
			if (!hd_struct_try_get(part))
				break;
349
			rcu_assign_pointer(ptbl->last_lookup, part);
350
			goto out_unlock;
351
		}
352
	}
353 354 355 356 357

	part = &disk->part0;
out_unlock:
	rcu_read_unlock();
	return part;
358 359
}

360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395
/**
 * disk_has_partitions
 * @disk: gendisk of interest
 *
 * Walk through the partition table and check if valid partition exists.
 *
 * CONTEXT:
 * Don't care.
 *
 * RETURNS:
 * True if the gendisk has at least one valid non-zero size partition.
 * Otherwise false.
 */
bool disk_has_partitions(struct gendisk *disk)
{
	struct disk_part_tbl *ptbl;
	int i;
	bool ret = false;

	rcu_read_lock();
	ptbl = rcu_dereference(disk->part_tbl);

	/* Iterate partitions skipping the whole device at index 0 */
	for (i = 1; i < ptbl->len; i++) {
		if (rcu_dereference(ptbl->part[i])) {
			ret = true;
			break;
		}
	}

	rcu_read_unlock();

	return ret;
}
EXPORT_SYMBOL_GPL(disk_has_partitions);

L
Linus Torvalds 已提交
396 397 398 399
/*
 * Can be deleted altogether. Later.
 *
 */
400
#define BLKDEV_MAJOR_HASH_SIZE 255
L
Linus Torvalds 已提交
401 402 403 404
static struct blk_major_name {
	struct blk_major_name *next;
	int major;
	char name[16];
405
} *major_names[BLKDEV_MAJOR_HASH_SIZE];
C
Christoph Hellwig 已提交
406
static DEFINE_MUTEX(major_names_lock);
L
Linus Torvalds 已提交
407 408

/* index in the above - for now: assume no multimajor ranges */
409
static inline int major_to_index(unsigned major)
L
Linus Torvalds 已提交
410
{
411
	return major % BLKDEV_MAJOR_HASH_SIZE;
412 413
}

414
#ifdef CONFIG_PROC_FS
415
void blkdev_show(struct seq_file *seqf, off_t offset)
416
{
417
	struct blk_major_name *dp;
418

C
Christoph Hellwig 已提交
419
	mutex_lock(&major_names_lock);
420 421
	for (dp = major_names[major_to_index(offset)]; dp; dp = dp->next)
		if (dp->major == offset)
422
			seq_printf(seqf, "%3d %s\n", dp->major, dp->name);
C
Christoph Hellwig 已提交
423
	mutex_unlock(&major_names_lock);
L
Linus Torvalds 已提交
424
}
425
#endif /* CONFIG_PROC_FS */
L
Linus Torvalds 已提交
426

427 428 429
/**
 * register_blkdev - register a new block device
 *
430 431
 * @major: the requested major device number [1..BLKDEV_MAJOR_MAX-1]. If
 *         @major = 0, try to allocate any unused major number.
432 433 434 435
 * @name: the name of the new block device as a zero terminated string
 *
 * The @name must be unique within the system.
 *
436 437
 * The return value depends on the @major input parameter:
 *
438 439
 *  - if a major device number was requested in range [1..BLKDEV_MAJOR_MAX-1]
 *    then the function returns zero on success, or a negative error code
440
 *  - if any unused major number was requested with @major = 0 parameter
441
 *    then the return value is the allocated major number in range
442 443 444 445
 *    [1..BLKDEV_MAJOR_MAX-1] or a negative error code otherwise
 *
 * See Documentation/admin-guide/devices.txt for the list of allocated
 * major numbers.
446
 */
L
Linus Torvalds 已提交
447 448 449 450 451
int register_blkdev(unsigned int major, const char *name)
{
	struct blk_major_name **n, *p;
	int index, ret = 0;

C
Christoph Hellwig 已提交
452
	mutex_lock(&major_names_lock);
L
Linus Torvalds 已提交
453 454 455 456 457 458 459 460 461

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

		if (index == 0) {
462 463
			printk("%s: failed to get major for %s\n",
			       __func__, name);
L
Linus Torvalds 已提交
464 465 466 467 468 469 470
			ret = -EBUSY;
			goto out;
		}
		major = index;
		ret = major;
	}

471
	if (major >= BLKDEV_MAJOR_MAX) {
472 473
		pr_err("%s: major requested (%u) is greater than the maximum (%u) for %s\n",
		       __func__, major, BLKDEV_MAJOR_MAX-1, name);
474 475 476 477 478

		ret = -EINVAL;
		goto out;
	}

L
Linus Torvalds 已提交
479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499
	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) {
500
		printk("register_blkdev: cannot get major %u for %s\n",
L
Linus Torvalds 已提交
501 502 503 504
		       major, name);
		kfree(p);
	}
out:
C
Christoph Hellwig 已提交
505
	mutex_unlock(&major_names_lock);
L
Linus Torvalds 已提交
506 507 508 509 510
	return ret;
}

EXPORT_SYMBOL(register_blkdev);

A
Akinobu Mita 已提交
511
void unregister_blkdev(unsigned int major, const char *name)
L
Linus Torvalds 已提交
512 513 514 515 516
{
	struct blk_major_name **n;
	struct blk_major_name *p = NULL;
	int index = major_to_index(major);

C
Christoph Hellwig 已提交
517
	mutex_lock(&major_names_lock);
L
Linus Torvalds 已提交
518 519 520
	for (n = &major_names[index]; *n; n = &(*n)->next)
		if ((*n)->major == major)
			break;
521 522 523
	if (!*n || strcmp((*n)->name, name)) {
		WARN_ON(1);
	} else {
L
Linus Torvalds 已提交
524 525 526
		p = *n;
		*n = p->next;
	}
C
Christoph Hellwig 已提交
527
	mutex_unlock(&major_names_lock);
L
Linus Torvalds 已提交
528 529 530 531 532
	kfree(p);
}

EXPORT_SYMBOL(unregister_blkdev);

533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
/**
 * 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 已提交
565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581
/**
 * blk_alloc_devt - allocate a dev_t for a partition
 * @part: partition to allocate dev_t for
 * @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);
T
Tejun Heo 已提交
582
	int idx;
T
Tejun Heo 已提交
583 584 585 586 587 588 589 590

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

	/* allocate ext devt */
591 592
	idr_preload(GFP_KERNEL);

593
	spin_lock_bh(&ext_devt_lock);
594
	idx = idr_alloc(&ext_devt_idr, part, 0, NR_EXT_DEVT, GFP_NOWAIT);
595
	spin_unlock_bh(&ext_devt_lock);
596 597

	idr_preload_end();
T
Tejun Heo 已提交
598 599
	if (idx < 0)
		return idx == -ENOSPC ? -EBUSY : idx;
T
Tejun Heo 已提交
600

601
	*devt = MKDEV(BLOCK_EXT_MAJOR, blk_mangle_minor(idx));
T
Tejun Heo 已提交
602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619
	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)
{
	if (devt == MKDEV(0, 0))
		return;

	if (MAJOR(devt) == BLOCK_EXT_MAJOR) {
620
		spin_lock_bh(&ext_devt_lock);
621
		idr_remove(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
622
		spin_unlock_bh(&ext_devt_lock);
T
Tejun Heo 已提交
623 624 625
	}
}

626 627
/*
 * We invalidate devt by assigning NULL pointer for devt in idr.
Y
Yufen Yu 已提交
628 629 630 631 632 633 634 635 636 637
 */
void blk_invalidate_devt(dev_t devt)
{
	if (MAJOR(devt) == BLOCK_EXT_MAJOR) {
		spin_lock_bh(&ext_devt_lock);
		idr_replace(&ext_devt_idr, NULL, blk_mangle_minor(MINOR(devt)));
		spin_unlock_bh(&ext_devt_lock);
	}
}

638 639 640 641 642 643 644 645 646 647 648 649
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 已提交
650 651 652 653 654
/*
 * Register device numbers dev..(dev+range-1)
 * range must be nonzero
 * The hash chain is sorted on range, so that subranges can override.
 */
655
void blk_register_region(dev_t devt, unsigned long range, struct module *module,
L
Linus Torvalds 已提交
656 657 658
			 struct kobject *(*probe)(dev_t, int *, void *),
			 int (*lock)(dev_t, void *), void *data)
{
659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676
	unsigned n = MAJOR(devt + range - 1) - MAJOR(devt) + 1;
	unsigned index = MAJOR(devt);
	unsigned i;
	struct bdev_map *p;

	n = min(n, 255u);
	p = kmalloc_array(n, sizeof(struct bdev_map), GFP_KERNEL);
	if (p == NULL)
		return;

	for (i = 0; i < n; i++, p++) {
		p->owner = module;
		p->probe = probe;
		p->lock = lock;
		p->dev = devt;
		p->range = range;
		p->data = data;
	}
L
Linus Torvalds 已提交
677

C
Christoph Hellwig 已提交
678
	mutex_lock(&bdev_map_lock);
679 680 681 682 683 684 685
	for (i = 0, p -= n; i < n; i++, p++, index++) {
		struct bdev_map **s = &bdev_map[index % 255];
		while (*s && (*s)->range < range)
			s = &(*s)->next;
		p->next = *s;
		*s = p;
	}
C
Christoph Hellwig 已提交
686
	mutex_unlock(&bdev_map_lock);
687
}
L
Linus Torvalds 已提交
688 689
EXPORT_SYMBOL(blk_register_region);

690
void blk_unregister_region(dev_t devt, unsigned long range)
L
Linus Torvalds 已提交
691
{
692 693 694 695
	unsigned n = MAJOR(devt + range - 1) - MAJOR(devt) + 1;
	unsigned index = MAJOR(devt);
	unsigned i;
	struct bdev_map *found = NULL;
L
Linus Torvalds 已提交
696

C
Christoph Hellwig 已提交
697
	mutex_lock(&bdev_map_lock);
698 699 700 701 702 703 704 705 706 707 708 709
	for (i = 0; i < min(n, 255u); i++, index++) {
		struct bdev_map **s;
		for (s = &bdev_map[index % 255]; *s; s = &(*s)->next) {
			struct bdev_map *p = *s;
			if (p->dev == devt && p->range == range) {
				*s = p->next;
				if (!found)
					found = p;
				break;
			}
		}
	}
C
Christoph Hellwig 已提交
710
	mutex_unlock(&bdev_map_lock);
711 712
	kfree(found);
}
L
Linus Torvalds 已提交
713 714
EXPORT_SYMBOL(blk_unregister_region);

715
static struct kobject *exact_match(dev_t devt, int *partno, void *data)
L
Linus Torvalds 已提交
716 717
{
	struct gendisk *p = data;
718

719
	return &disk_to_dev(p)->kobj;
L
Linus Torvalds 已提交
720 721
}

722
static int exact_lock(dev_t devt, void *data)
L
Linus Torvalds 已提交
723 724 725
{
	struct gendisk *p = data;

726
	if (!get_disk_and_module(p))
L
Linus Torvalds 已提交
727 728 729 730
		return -1;
	return 0;
}

731 732 733 734 735 736 737 738 739 740 741 742 743
static void disk_scan_partitions(struct gendisk *disk)
{
	struct block_device *bdev;

	if (!get_capacity(disk) || !disk_part_scan_enabled(disk))
		return;

	set_bit(GD_NEED_PART_SCAN, &disk->state);
	bdev = blkdev_get_by_dev(disk_devt(disk), FMODE_READ, NULL);
	if (!IS_ERR(bdev))
		blkdev_put(bdev, FMODE_READ);
}

744 745
static void register_disk(struct device *parent, struct gendisk *disk,
			  const struct attribute_group **groups)
746 747 748 749 750 751
{
	struct device *ddev = disk_to_dev(disk);
	struct disk_part_iter piter;
	struct hd_struct *part;
	int err;

752
	ddev->parent = parent;
753

754
	dev_set_name(ddev, "%s", disk->disk_name);
755 756 757 758

	/* delay uevents, until we scanned partition table */
	dev_set_uevent_suppress(ddev, 1);

759 760 761 762
	if (groups) {
		WARN_ON(ddev->groups);
		ddev->groups = groups;
	}
763 764 765 766 767 768 769 770 771 772
	if (device_add(ddev))
		return;
	if (!sysfs_deprecated) {
		err = sysfs_create_link(block_depr, &ddev->kobj,
					kobject_name(&ddev->kobj));
		if (err) {
			device_del(ddev);
			return;
		}
	}
773 774 775 776 777 778 779 780

	/*
	 * avoid probable deadlock caused by allocating memory with
	 * GFP_KERNEL in runtime_resume callback of its all ancestor
	 * devices
	 */
	pm_runtime_set_memalloc_noio(ddev, true);

781 782 783
	disk->part0.holder_dir = kobject_create_and_add("holders", &ddev->kobj);
	disk->slave_dir = kobject_create_and_add("slaves", &ddev->kobj);

784 785 786 787 788
	if (disk->flags & GENHD_FL_HIDDEN) {
		dev_set_uevent_suppress(ddev, 0);
		return;
	}

789
	disk_scan_partitions(disk);
790 791 792 793 794 795 796 797 798 799

	/* announce disk after possible partitions are created */
	dev_set_uevent_suppress(ddev, 0);
	kobject_uevent(&ddev->kobj, KOBJ_ADD);

	/* announce possible partitions */
	disk_part_iter_init(&piter, disk, 0);
	while ((part = disk_part_iter_next(&piter)))
		kobject_uevent(&part_to_dev(part)->kobj, KOBJ_ADD);
	disk_part_iter_exit(&piter);
800

801 802 803 804 805 806
	if (disk->queue->backing_dev_info->dev) {
		err = sysfs_create_link(&ddev->kobj,
			  &disk->queue->backing_dev_info->dev->kobj,
			  "bdi");
		WARN_ON(err);
	}
807 808
}

L
Linus Torvalds 已提交
809
/**
810
 * __device_add_disk - add disk information to kernel list
811
 * @parent: parent device for the disk
L
Linus Torvalds 已提交
812
 * @disk: per-device partitioning information
813
 * @groups: Additional per-device sysfs groups
814
 * @register_queue: register the queue if set to true
L
Linus Torvalds 已提交
815 816 817
 *
 * This function registers the partitioning information in @disk
 * with the kernel.
818 819
 *
 * FIXME: error handling
L
Linus Torvalds 已提交
820
 */
821
static void __device_add_disk(struct device *parent, struct gendisk *disk,
822
			      const struct attribute_group **groups,
823
			      bool register_queue)
L
Linus Torvalds 已提交
824
{
825
	dev_t devt;
826
	int retval;
827

828 829 830 831 832 833 834 835 836
	/*
	 * The disk queue should now be all set with enough information about
	 * the device for the elevator code to pick an adequate default
	 * elevator if one is needed, that is, for devices requesting queue
	 * registration.
	 */
	if (register_queue)
		elevator_init_mq(disk->queue);

837 838 839 840 841
	/* minors == 0 indicates to use ext devt from part0 and should
	 * be accompanied with EXT_DEVT flag.  Make sure all
	 * parameters make sense.
	 */
	WARN_ON(disk->minors && !(disk->major || disk->first_minor));
842 843
	WARN_ON(!disk->minors &&
		!(disk->flags & (GENHD_FL_EXT_DEVT | GENHD_FL_HIDDEN)));
844

L
Linus Torvalds 已提交
845
	disk->flags |= GENHD_FL_UP;
846 847 848 849 850 851 852 853 854

	retval = blk_alloc_devt(&disk->part0, &devt);
	if (retval) {
		WARN_ON(1);
		return;
	}
	disk->major = MAJOR(devt);
	disk->first_minor = MINOR(devt);

855 856
	disk_alloc_events(disk);

857 858 859 860 861 862 863 864
	if (disk->flags & GENHD_FL_HIDDEN) {
		/*
		 * Don't let hidden disks show up in /proc/partitions,
		 * and don't bother scanning for partitions either.
		 */
		disk->flags |= GENHD_FL_SUPPRESS_PARTITION_INFO;
		disk->flags |= GENHD_FL_NO_PART_SCAN;
	} else {
865 866
		struct backing_dev_info *bdi = disk->queue->backing_dev_info;
		struct device *dev = disk_to_dev(disk);
867 868
		int ret;

869
		/* Register BDI before referencing it from bdev */
870 871
		dev->devt = devt;
		ret = bdi_register(bdi, "%u:%u", MAJOR(devt), MINOR(devt));
872
		WARN_ON(ret);
873
		bdi_set_owner(bdi, dev);
874 875 876
		blk_register_region(disk_devt(disk), disk->minors, NULL,
				    exact_match, exact_lock, disk);
	}
877
	register_disk(parent, disk, groups);
878 879
	if (register_queue)
		blk_register_queue(disk);
880

881 882 883 884
	/*
	 * Take an extra ref on queue which will be put on disk_release()
	 * so that it sticks around as long as @disk is there.
	 */
T
Tejun Heo 已提交
885
	WARN_ON_ONCE(!blk_get_queue(disk->queue));
886

887
	disk_add_events(disk);
888
	blk_integrity_add(disk);
L
Linus Torvalds 已提交
889
}
890

891 892 893
void device_add_disk(struct device *parent, struct gendisk *disk,
		     const struct attribute_group **groups)

894
{
895
	__device_add_disk(parent, disk, groups, true);
896
}
897
EXPORT_SYMBOL(device_add_disk);
L
Linus Torvalds 已提交
898

899 900
void device_add_disk_no_queue_reg(struct device *parent, struct gendisk *disk)
{
901
	__device_add_disk(parent, disk, NULL, false);
902 903 904
}
EXPORT_SYMBOL(device_add_disk_no_queue_reg);

905 906 907 908 909 910 911 912 913 914
static void invalidate_partition(struct gendisk *disk, int partno)
{
	struct block_device *bdev;

	bdev = bdget_disk(disk, partno);
	if (!bdev)
		return;

	fsync_bdev(bdev);
	__invalidate_device(bdev, true);
915 916 917 918 919 920

	/*
	 * Unhash the bdev inode for this device so that it gets evicted as soon
	 * as last inode reference is dropped.
	 */
	remove_inode_hash(bdev->bd_inode);
921 922 923
	bdput(bdev);
}

924 925 926 927 928 929 930 931 932 933 934 935 936
/**
 * del_gendisk - remove the gendisk
 * @disk: the struct gendisk to remove
 *
 * Removes the gendisk and all its associated resources. This deletes the
 * partitions associated with the gendisk, and unregisters the associated
 * request_queue.
 *
 * This is the counter to the respective __device_add_disk() call.
 *
 * The final removal of the struct gendisk happens when its refcount reaches 0
 * with put_disk(), which should be called after del_gendisk(), if
 * __device_add_disk() was used.
937 938 939 940 941
 *
 * Drivers exist which depend on the release of the gendisk to be synchronous,
 * it should not be deferred.
 *
 * Context: can sleep
942
 */
943
void del_gendisk(struct gendisk *disk)
L
Linus Torvalds 已提交
944
{
945 946 947
	struct disk_part_iter piter;
	struct hd_struct *part;

948 949
	might_sleep();

950 951 952
	if (WARN_ON_ONCE(!disk->queue))
		return;

953
	blk_integrity_del(disk);
954 955
	disk_del_events(disk);

J
Jan Kara 已提交
956 957 958 959 960
	/*
	 * Block lookups of the disk until all bdevs are unhashed and the
	 * disk is marked as dead (GENHD_FL_UP cleared).
	 */
	down_write(&disk->lookup_sem);
961 962 963 964 965
	/* invalidate stuff */
	disk_part_iter_init(&piter, disk,
			     DISK_PITER_INCL_EMPTY | DISK_PITER_REVERSE);
	while ((part = disk_part_iter_next(&piter))) {
		invalidate_partition(disk, part->partno);
966
		delete_partition(part);
967 968 969 970 971 972
	}
	disk_part_iter_exit(&piter);

	invalidate_partition(disk, 0);
	set_capacity(disk, 0);
	disk->flags &= ~GENHD_FL_UP;
J
Jan Kara 已提交
973
	up_write(&disk->lookup_sem);
974

975
	if (!(disk->flags & GENHD_FL_HIDDEN)) {
976
		sysfs_remove_link(&disk_to_dev(disk)->kobj, "bdi");
977

978 979 980 981
		/*
		 * Unregister bdi before releasing device numbers (as they can
		 * get reused and we'd get clashes in sysfs).
		 */
982
		bdi_unregister(disk->queue->backing_dev_info);
983
	}
984

985 986
	blk_unregister_queue(disk);
	
987
	if (!(disk->flags & GENHD_FL_HIDDEN))
988
		blk_unregister_region(disk_devt(disk), disk->minors);
Y
Yufen Yu 已提交
989 990 991 992 993 994 995
	/*
	 * Remove gendisk pointer from idr so that it cannot be looked up
	 * while RCU period before freeing gendisk is running to prevent
	 * use-after-free issues. Note that the device number stays
	 * "in-use" until we really free the gendisk.
	 */
	blk_invalidate_devt(disk_devt(disk));
996 997 998 999 1000 1001 1002 1003

	kobject_put(disk->part0.holder_dir);
	kobject_put(disk->slave_dir);

	part_stat_set_all(&disk->part0, 0);
	disk->part0.stamp = 0;
	if (!sysfs_deprecated)
		sysfs_remove_link(block_depr, dev_name(disk_to_dev(disk)));
1004
	pm_runtime_set_memalloc_noio(disk_to_dev(disk), false);
1005
	device_del(disk_to_dev(disk));
L
Linus Torvalds 已提交
1006
}
1007
EXPORT_SYMBOL(del_gendisk);
L
Linus Torvalds 已提交
1008

1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033
/* sysfs access to bad-blocks list. */
static ssize_t disk_badblocks_show(struct device *dev,
					struct device_attribute *attr,
					char *page)
{
	struct gendisk *disk = dev_to_disk(dev);

	if (!disk->bb)
		return sprintf(page, "\n");

	return badblocks_show(disk->bb, page, 0);
}

static ssize_t disk_badblocks_store(struct device *dev,
					struct device_attribute *attr,
					const char *page, size_t len)
{
	struct gendisk *disk = dev_to_disk(dev);

	if (!disk->bb)
		return -ENXIO;

	return badblocks_store(disk->bb, page, len, 0);
}

1034 1035 1036 1037 1038 1039 1040
static struct gendisk *lookup_gendisk(dev_t dev, int *partno)
{
	struct kobject *kobj;
	struct bdev_map *p;
	unsigned long best = ~0UL;

retry:
C
Christoph Hellwig 已提交
1041
	mutex_lock(&bdev_map_lock);
1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061
	for (p = bdev_map[MAJOR(dev) % 255]; p; p = p->next) {
		struct kobject *(*probe)(dev_t, int *, void *);
		struct module *owner;
		void *data;

		if (p->dev > dev || p->dev + p->range - 1 < dev)
			continue;
		if (p->range - 1 >= best)
			break;
		if (!try_module_get(p->owner))
			continue;
		owner = p->owner;
		data = p->data;
		probe = p->probe;
		best = p->range - 1;
		*partno = dev - p->dev;
		if (p->lock && p->lock(dev, data) < 0) {
			module_put(owner);
			continue;
		}
C
Christoph Hellwig 已提交
1062
		mutex_unlock(&bdev_map_lock);
1063 1064 1065 1066 1067 1068 1069
		kobj = probe(dev, partno, data);
		/* Currently ->owner protects _only_ ->probe() itself. */
		module_put(owner);
		if (kobj)
			return dev_to_disk(kobj_to_dev(kobj));
		goto retry;
	}
C
Christoph Hellwig 已提交
1070
	mutex_unlock(&bdev_map_lock);
1071 1072 1073 1074
	return NULL;
}


L
Linus Torvalds 已提交
1075 1076
/**
 * get_gendisk - get partitioning information for a given device
1077
 * @devt: device to get partitioning information for
1078
 * @partno: returned partition index
L
Linus Torvalds 已提交
1079 1080
 *
 * This function gets the structure containing partitioning
1081
 * information for the given device @devt.
1082 1083
 *
 * Context: can sleep
L
Linus Torvalds 已提交
1084
 */
1085
struct gendisk *get_gendisk(dev_t devt, int *partno)
L
Linus Torvalds 已提交
1086
{
T
Tejun Heo 已提交
1087 1088
	struct gendisk *disk = NULL;

1089 1090
	might_sleep();

T
Tejun Heo 已提交
1091
	if (MAJOR(devt) != BLOCK_EXT_MAJOR) {
1092
		disk = lookup_gendisk(devt, partno);
T
Tejun Heo 已提交
1093 1094 1095
	} else {
		struct hd_struct *part;

1096
		spin_lock_bh(&ext_devt_lock);
1097
		part = idr_find(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
1098
		if (part && get_disk_and_module(part_to_disk(part))) {
T
Tejun Heo 已提交
1099 1100 1101
			*partno = part->partno;
			disk = part_to_disk(part);
		}
1102
		spin_unlock_bh(&ext_devt_lock);
T
Tejun Heo 已提交
1103
	}
1104

J
Jan Kara 已提交
1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115
	if (!disk)
		return NULL;

	/*
	 * Synchronize with del_gendisk() to not return disk that is being
	 * destroyed.
	 */
	down_read(&disk->lookup_sem);
	if (unlikely((disk->flags & GENHD_FL_HIDDEN) ||
		     !(disk->flags & GENHD_FL_UP))) {
		up_read(&disk->lookup_sem);
1116
		put_disk_and_module(disk);
1117
		disk = NULL;
J
Jan Kara 已提交
1118 1119
	} else {
		up_read(&disk->lookup_sem);
1120
	}
T
Tejun Heo 已提交
1121
	return disk;
L
Linus Torvalds 已提交
1122 1123
}

1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136
/**
 * 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.
 */
1137
struct block_device *bdget_disk(struct gendisk *disk, int partno)
1138
{
T
Tejun Heo 已提交
1139 1140
	struct hd_struct *part;
	struct block_device *bdev = NULL;
1141

T
Tejun Heo 已提交
1142
	part = disk_get_part(disk, partno);
1143
	if (part)
C
Christoph Hellwig 已提交
1144
		bdev = bdget_part(part);
T
Tejun Heo 已提交
1145
	disk_put_part(part);
1146

T
Tejun Heo 已提交
1147
	return bdev;
1148 1149 1150
}
EXPORT_SYMBOL(bdget_disk);

1151 1152 1153 1154 1155 1156 1157
/*
 * 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)
{
1158 1159 1160 1161 1162 1163
	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);
1164 1165
		struct disk_part_iter piter;
		struct hd_struct *part;
1166 1167
		char name_buf[BDEVNAME_SIZE];
		char devt_buf[BDEVT_SIZE];
1168 1169 1170

		/*
		 * Don't show empty devices or things that have been
L
Lucas De Marchi 已提交
1171
		 * suppressed
1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
		 */
		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.
		 */
T
Tejun Heo 已提交
1182 1183 1184
		disk_part_iter_init(&piter, disk, DISK_PITER_INCL_PART0);
		while ((part = disk_part_iter_next(&piter))) {
			bool is_part0 = part == &disk->part0;
1185

1186
			printk("%s%s %10llu %s %s", is_part0 ? "" : "  ",
1187
			       bdevt_str(part_devt(part), devt_buf),
1188 1189
			       (unsigned long long)part_nr_sects_read(part) >> 1
			       , disk_name(disk, part->partno, name_buf),
1190
			       part->info ? part->info->uuid : "");
T
Tejun Heo 已提交
1191
			if (is_part0) {
D
Dan Williams 已提交
1192
				if (dev->parent && dev->parent->driver)
T
Tejun Heo 已提交
1193
					printk(" driver: %s\n",
D
Dan Williams 已提交
1194
					      dev->parent->driver->name);
T
Tejun Heo 已提交
1195 1196 1197 1198 1199
				else
					printk(" (driver?)\n");
			} else
				printk("\n");
		}
1200
		disk_part_iter_exit(&piter);
1201 1202
	}
	class_dev_iter_exit(&iter);
1203 1204
}

L
Linus Torvalds 已提交
1205 1206
#ifdef CONFIG_PROC_FS
/* iterator */
1207
static void *disk_seqf_start(struct seq_file *seqf, loff_t *pos)
1208
{
1209 1210 1211
	loff_t skip = *pos;
	struct class_dev_iter *iter;
	struct device *dev;
1212

1213
	iter = kmalloc(sizeof(*iter), GFP_KERNEL);
1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
	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);
1226 1227
}

1228
static void *disk_seqf_next(struct seq_file *seqf, void *v, loff_t *pos)
L
Linus Torvalds 已提交
1229
{
1230
	struct device *dev;
L
Linus Torvalds 已提交
1231

1232 1233
	(*pos)++;
	dev = class_dev_iter_next(seqf->private);
1234
	if (dev)
1235
		return dev_to_disk(dev);
1236

L
Linus Torvalds 已提交
1237 1238 1239
	return NULL;
}

1240
static void disk_seqf_stop(struct seq_file *seqf, void *v)
1241
{
1242
	struct class_dev_iter *iter = seqf->private;
1243

1244 1245 1246 1247
	/* stop is called even after start failed :-( */
	if (iter) {
		class_dev_iter_exit(iter);
		kfree(iter);
1248
		seqf->private = NULL;
1249
	}
L
Linus Torvalds 已提交
1250 1251
}

1252
static void *show_partition_start(struct seq_file *seqf, loff_t *pos)
L
Linus Torvalds 已提交
1253
{
1254
	void *p;
1255 1256

	p = disk_seqf_start(seqf, pos);
1257
	if (!IS_ERR_OR_NULL(p) && !*pos)
1258 1259
		seq_puts(seqf, "major minor  #blocks  name\n\n");
	return p;
L
Linus Torvalds 已提交
1260 1261
}

1262
static int show_partition(struct seq_file *seqf, void *v)
L
Linus Torvalds 已提交
1263 1264
{
	struct gendisk *sgp = v;
1265 1266
	struct disk_part_iter piter;
	struct hd_struct *part;
L
Linus Torvalds 已提交
1267 1268 1269
	char buf[BDEVNAME_SIZE];

	/* Don't show non-partitionable removeable devices or empty devices */
T
Tejun Heo 已提交
1270
	if (!get_capacity(sgp) || (!disk_max_parts(sgp) &&
1271
				   (sgp->flags & GENHD_FL_REMOVABLE)))
L
Linus Torvalds 已提交
1272 1273 1274 1275 1276
		return 0;
	if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
		return 0;

	/* show the full disk and all non-0 size partitions of it */
T
Tejun Heo 已提交
1277
	disk_part_iter_init(&piter, sgp, DISK_PITER_INCL_PART0);
1278
	while ((part = disk_part_iter_next(&piter)))
1279
		seq_printf(seqf, "%4d  %7d %10llu %s\n",
1280
			   MAJOR(part_devt(part)), MINOR(part_devt(part)),
1281
			   (unsigned long long)part_nr_sects_read(part) >> 1,
1282
			   disk_name(sgp, part->partno, buf));
1283
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
1284 1285 1286 1287

	return 0;
}

1288
static const struct seq_operations partitions_op = {
1289 1290 1291
	.start	= show_partition_start,
	.next	= disk_seqf_next,
	.stop	= disk_seqf_stop,
1292
	.show	= show_partition
L
Linus Torvalds 已提交
1293 1294 1295 1296
};
#endif


1297
static struct kobject *base_probe(dev_t devt, int *partno, void *data)
L
Linus Torvalds 已提交
1298
{
1299
	if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
L
Linus Torvalds 已提交
1300
		/* Make old-style 2.4 aliases work */
1301
		request_module("block-major-%d", MAJOR(devt));
L
Linus Torvalds 已提交
1302 1303 1304
	return NULL;
}

1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320
static void bdev_map_init(void)
{
	struct bdev_map *base;
	int i;

	base = kzalloc(sizeof(*base), GFP_KERNEL);
	if (!base)
		panic("cannot allocate bdev_map");

	base->dev = 1;
	base->range = ~0 ;
	base->probe = base_probe;
	for (i = 0; i < 255; i++)
		bdev_map[i] = base;
}

L
Linus Torvalds 已提交
1321 1322
static int __init genhd_device_init(void)
{
1323 1324 1325 1326
	int error;

	block_class.dev_kobj = sysfs_dev_block_kobj;
	error = class_register(&block_class);
R
Roland McGrath 已提交
1327 1328
	if (unlikely(error))
		return error;
1329
	bdev_map_init();
L
Linus Torvalds 已提交
1330
	blk_dev_init();
1331

1332 1333
	register_blkdev(BLOCK_EXT_MAJOR, "blkext");

1334
	/* create top-level block dir */
1335 1336
	if (!sysfs_deprecated)
		block_depr = kobject_create_and_add("block", NULL);
1337
	return 0;
L
Linus Torvalds 已提交
1338 1339 1340 1341
}

subsys_initcall(genhd_device_init);

1342 1343
static ssize_t disk_range_show(struct device *dev,
			       struct device_attribute *attr, char *buf)
L
Linus Torvalds 已提交
1344
{
1345
	struct gendisk *disk = dev_to_disk(dev);
L
Linus Torvalds 已提交
1346

1347
	return sprintf(buf, "%d\n", disk->minors);
L
Linus Torvalds 已提交
1348 1349
}

1350 1351 1352 1353 1354
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 已提交
1355
	return sprintf(buf, "%d\n", disk_max_parts(disk));
1356 1357
}

1358 1359
static ssize_t disk_removable_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
1360
{
1361
	struct gendisk *disk = dev_to_disk(dev);
1362

1363 1364
	return sprintf(buf, "%d\n",
		       (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
1365 1366
}

1367 1368 1369 1370 1371 1372 1373 1374 1375
static ssize_t disk_hidden_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_HIDDEN ? 1 : 0));
}

K
Kay Sievers 已提交
1376 1377 1378 1379 1380
static ssize_t disk_ro_show(struct device *dev,
				   struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

T
Tejun Heo 已提交
1381
	return sprintf(buf, "%d\n", get_disk_ro(disk) ? 1 : 0);
K
Kay Sievers 已提交
1382 1383
}

1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397
ssize_t part_size_show(struct device *dev,
		       struct device_attribute *attr, char *buf)
{
	struct hd_struct *p = dev_to_part(dev);

	return sprintf(buf, "%llu\n",
		(unsigned long long)part_nr_sects_read(p));
}

ssize_t part_stat_show(struct device *dev,
		       struct device_attribute *attr, char *buf)
{
	struct hd_struct *p = dev_to_part(dev);
	struct request_queue *q = part_to_disk(p)->queue;
1398
	struct disk_stats stat;
1399 1400
	unsigned int inflight;

1401
	part_stat_read_all(p, &stat);
1402 1403 1404
	if (queue_is_mq(q))
		inflight = blk_mq_in_flight(q, p);
	else
1405
		inflight = part_in_flight(p);
1406

1407 1408 1409 1410 1411 1412 1413
	return sprintf(buf,
		"%8lu %8lu %8llu %8u "
		"%8lu %8lu %8llu %8u "
		"%8u %8u %8u "
		"%8lu %8lu %8llu %8u "
		"%8lu %8u"
		"\n",
1414 1415 1416 1417 1418 1419 1420 1421
		stat.ios[STAT_READ],
		stat.merges[STAT_READ],
		(unsigned long long)stat.sectors[STAT_READ],
		(unsigned int)div_u64(stat.nsecs[STAT_READ], NSEC_PER_MSEC),
		stat.ios[STAT_WRITE],
		stat.merges[STAT_WRITE],
		(unsigned long long)stat.sectors[STAT_WRITE],
		(unsigned int)div_u64(stat.nsecs[STAT_WRITE], NSEC_PER_MSEC),
1422
		inflight,
1423
		jiffies_to_msecs(stat.io_ticks),
1424 1425 1426 1427 1428
		(unsigned int)div_u64(stat.nsecs[STAT_READ] +
				      stat.nsecs[STAT_WRITE] +
				      stat.nsecs[STAT_DISCARD] +
				      stat.nsecs[STAT_FLUSH],
						NSEC_PER_MSEC),
1429 1430 1431 1432 1433 1434
		stat.ios[STAT_DISCARD],
		stat.merges[STAT_DISCARD],
		(unsigned long long)stat.sectors[STAT_DISCARD],
		(unsigned int)div_u64(stat.nsecs[STAT_DISCARD], NSEC_PER_MSEC),
		stat.ios[STAT_FLUSH],
		(unsigned int)div_u64(stat.nsecs[STAT_FLUSH], NSEC_PER_MSEC));
1435 1436 1437 1438 1439 1440 1441 1442 1443
}

ssize_t part_inflight_show(struct device *dev, struct device_attribute *attr,
			   char *buf)
{
	struct hd_struct *p = dev_to_part(dev);
	struct request_queue *q = part_to_disk(p)->queue;
	unsigned int inflight[2];

1444 1445 1446
	if (queue_is_mq(q))
		blk_mq_in_flight_rw(q, p, inflight);
	else
1447
		part_in_flight_rw(p, inflight);
1448

1449 1450 1451
	return sprintf(buf, "%8u %8u\n", inflight[0], inflight[1]);
}

1452 1453
static ssize_t disk_capability_show(struct device *dev,
				    struct device_attribute *attr, char *buf)
1454
{
1455 1456 1457
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%x\n", disk->flags);
1458
}
1459

1460 1461 1462 1463 1464 1465 1466 1467 1468
static ssize_t disk_alignment_offset_show(struct device *dev,
					  struct device_attribute *attr,
					  char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

	return sprintf(buf, "%d\n", queue_alignment_offset(disk->queue));
}

1469 1470 1471 1472 1473 1474
static ssize_t disk_discard_alignment_show(struct device *dev,
					   struct device_attribute *attr,
					   char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

1475
	return sprintf(buf, "%d\n", queue_discard_alignment(disk->queue));
1476 1477
}

1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489
static DEVICE_ATTR(range, 0444, disk_range_show, NULL);
static DEVICE_ATTR(ext_range, 0444, disk_ext_range_show, NULL);
static DEVICE_ATTR(removable, 0444, disk_removable_show, NULL);
static DEVICE_ATTR(hidden, 0444, disk_hidden_show, NULL);
static DEVICE_ATTR(ro, 0444, disk_ro_show, NULL);
static DEVICE_ATTR(size, 0444, part_size_show, NULL);
static DEVICE_ATTR(alignment_offset, 0444, disk_alignment_offset_show, NULL);
static DEVICE_ATTR(discard_alignment, 0444, disk_discard_alignment_show, NULL);
static DEVICE_ATTR(capability, 0444, disk_capability_show, NULL);
static DEVICE_ATTR(stat, 0444, part_stat_show, NULL);
static DEVICE_ATTR(inflight, 0444, part_inflight_show, NULL);
static DEVICE_ATTR(badblocks, 0644, disk_badblocks_show, disk_badblocks_store);
1490

1491
#ifdef CONFIG_FAIL_MAKE_REQUEST
1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
ssize_t part_fail_show(struct device *dev,
		       struct device_attribute *attr, char *buf)
{
	struct hd_struct *p = dev_to_part(dev);

	return sprintf(buf, "%d\n", p->make_it_fail);
}

ssize_t part_fail_store(struct device *dev,
			struct device_attribute *attr,
			const char *buf, size_t count)
{
	struct hd_struct *p = dev_to_part(dev);
	int i;

	if (count > 0 && sscanf(buf, "%d", &i) > 0)
		p->make_it_fail = (i == 0) ? 0 : 1;

	return count;
}

1513
static struct device_attribute dev_attr_fail =
1514
	__ATTR(make-it-fail, 0644, part_fail_show, part_fail_store);
1515 1516
#endif /* CONFIG_FAIL_MAKE_REQUEST */

1517 1518
#ifdef CONFIG_FAIL_IO_TIMEOUT
static struct device_attribute dev_attr_fail_timeout =
1519
	__ATTR(io-timeout-fail, 0644, part_timeout_show, part_timeout_store);
1520
#endif
1521 1522 1523

static struct attribute *disk_attrs[] = {
	&dev_attr_range.attr,
1524
	&dev_attr_ext_range.attr,
1525
	&dev_attr_removable.attr,
1526
	&dev_attr_hidden.attr,
K
Kay Sievers 已提交
1527
	&dev_attr_ro.attr,
1528
	&dev_attr_size.attr,
1529
	&dev_attr_alignment_offset.attr,
1530
	&dev_attr_discard_alignment.attr,
1531 1532
	&dev_attr_capability.attr,
	&dev_attr_stat.attr,
1533
	&dev_attr_inflight.attr,
1534
	&dev_attr_badblocks.attr,
1535 1536
#ifdef CONFIG_FAIL_MAKE_REQUEST
	&dev_attr_fail.attr,
1537 1538 1539
#endif
#ifdef CONFIG_FAIL_IO_TIMEOUT
	&dev_attr_fail_timeout.attr,
1540 1541 1542 1543
#endif
	NULL
};

1544 1545 1546 1547 1548 1549 1550 1551 1552 1553
static umode_t disk_visible(struct kobject *kobj, struct attribute *a, int n)
{
	struct device *dev = container_of(kobj, typeof(*dev), kobj);
	struct gendisk *disk = dev_to_disk(dev);

	if (a == &dev_attr_badblocks.attr && !disk->bb)
		return 0;
	return a->mode;
}

1554 1555
static struct attribute_group disk_attr_group = {
	.attrs = disk_attrs,
1556
	.is_visible = disk_visible,
1557 1558
};

1559
static const struct attribute_group *disk_attr_groups[] = {
1560 1561
	&disk_attr_group,
	NULL
L
Linus Torvalds 已提交
1562 1563
};

T
Tejun Heo 已提交
1564 1565 1566 1567 1568 1569 1570 1571 1572
/**
 * disk_replace_part_tbl - replace disk->part_tbl in RCU-safe way
 * @disk: disk to replace part_tbl for
 * @new_ptbl: new part_tbl to install
 *
 * Replace disk->part_tbl with @new_ptbl in RCU-safe way.  The
 * original ptbl is freed using RCU callback.
 *
 * LOCKING:
1573
 * Matching bd_mutex locked or the caller is the only user of @disk.
T
Tejun Heo 已提交
1574 1575 1576 1577
 */
static void disk_replace_part_tbl(struct gendisk *disk,
				  struct disk_part_tbl *new_ptbl)
{
1578 1579
	struct disk_part_tbl *old_ptbl =
		rcu_dereference_protected(disk->part_tbl, 1);
T
Tejun Heo 已提交
1580 1581

	rcu_assign_pointer(disk->part_tbl, new_ptbl);
1582 1583 1584

	if (old_ptbl) {
		rcu_assign_pointer(old_ptbl->last_lookup, NULL);
1585
		kfree_rcu(old_ptbl, rcu_head);
1586
	}
T
Tejun Heo 已提交
1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597
}

/**
 * disk_expand_part_tbl - expand disk->part_tbl
 * @disk: disk to expand part_tbl for
 * @partno: expand such that this partno can fit in
 *
 * Expand disk->part_tbl such that @partno can fit in.  disk->part_tbl
 * uses RCU to allow unlocked dereferencing for stats and other stuff.
 *
 * LOCKING:
1598 1599
 * Matching bd_mutex locked or the caller is the only user of @disk.
 * Might sleep.
T
Tejun Heo 已提交
1600 1601 1602 1603 1604 1605
 *
 * RETURNS:
 * 0 on success, -errno on failure.
 */
int disk_expand_part_tbl(struct gendisk *disk, int partno)
{
1606 1607
	struct disk_part_tbl *old_ptbl =
		rcu_dereference_protected(disk->part_tbl, 1);
T
Tejun Heo 已提交
1608 1609
	struct disk_part_tbl *new_ptbl;
	int len = old_ptbl ? old_ptbl->len : 0;
1610 1611 1612 1613 1614 1615 1616 1617 1618
	int i, target;

	/*
	 * check for int overflow, since we can get here from blkpg_ioctl()
	 * with a user passed 'partno'.
	 */
	target = partno + 1;
	if (target < 0)
		return -EINVAL;
T
Tejun Heo 已提交
1619 1620 1621 1622 1623 1624 1625 1626

	/* disk_max_parts() is zero during initialization, ignore if so */
	if (disk_max_parts(disk) && target > disk_max_parts(disk))
		return -EINVAL;

	if (target <= len)
		return 0;

1627 1628
	new_ptbl = kzalloc_node(struct_size(new_ptbl, part, target), GFP_KERNEL,
				disk->node_id);
T
Tejun Heo 已提交
1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640
	if (!new_ptbl)
		return -ENOMEM;

	new_ptbl->len = target;

	for (i = 0; i < len; i++)
		rcu_assign_pointer(new_ptbl->part[i], old_ptbl->part[i]);

	disk_replace_part_tbl(disk, new_ptbl);
	return 0;
}

1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656
/**
 * disk_release - releases all allocated resources of the gendisk
 * @dev: the device representing this disk
 *
 * This function releases all allocated resources of the gendisk.
 *
 * The struct gendisk refcount is incremented with get_gendisk() or
 * get_disk_and_module(), and its refcount is decremented with
 * put_disk_and_module() or put_disk(). Once the refcount reaches 0 this
 * function is called.
 *
 * Drivers which used __device_add_disk() have a gendisk with a request_queue
 * assigned. Since the request_queue sits on top of the gendisk for these
 * drivers we also call blk_put_queue() for them, and we expect the
 * request_queue refcount to reach 0 at this point, and so the request_queue
 * will also be freed prior to the disk.
1657 1658
 *
 * Context: can sleep
1659
 */
1660
static void disk_release(struct device *dev)
L
Linus Torvalds 已提交
1661
{
1662 1663
	struct gendisk *disk = dev_to_disk(dev);

1664 1665
	might_sleep();

1666
	blk_free_devt(dev->devt);
1667
	disk_release_events(disk);
L
Linus Torvalds 已提交
1668
	kfree(disk->random);
T
Tejun Heo 已提交
1669
	disk_replace_part_tbl(disk, NULL);
1670
	hd_free_part(&disk->part0);
1671 1672
	if (disk->queue)
		blk_put_queue(disk->queue);
L
Linus Torvalds 已提交
1673 1674
	kfree(disk);
}
1675 1676
struct class block_class = {
	.name		= "block",
L
Linus Torvalds 已提交
1677 1678
};

1679
static char *block_devnode(struct device *dev, umode_t *mode,
1680
			   kuid_t *uid, kgid_t *gid)
1681 1682 1683
{
	struct gendisk *disk = dev_to_disk(dev);

1684 1685
	if (disk->fops->devnode)
		return disk->fops->devnode(disk, mode);
1686 1687 1688
	return NULL;
}

1689
const struct device_type disk_type = {
1690 1691 1692
	.name		= "disk",
	.groups		= disk_attr_groups,
	.release	= disk_release,
1693
	.devnode	= block_devnode,
L
Linus Torvalds 已提交
1694 1695
};

1696
#ifdef CONFIG_PROC_FS
1697 1698 1699 1700 1701 1702 1703 1704
/*
 * 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 已提交
1705 1706
{
	struct gendisk *gp = v;
1707 1708
	struct disk_part_iter piter;
	struct hd_struct *hd;
L
Linus Torvalds 已提交
1709
	char buf[BDEVNAME_SIZE];
1710
	unsigned int inflight;
1711
	struct disk_stats stat;
L
Linus Torvalds 已提交
1712 1713

	/*
1714
	if (&disk_to_dev(gp)->kobj.entry == block_class.devices.next)
1715
		seq_puts(seqf,	"major minor name"
L
Linus Torvalds 已提交
1716 1717 1718 1719
				"     rio rmerge rsect ruse wio wmerge "
				"wsect wuse running use aveq"
				"\n\n");
	*/
1720

1721
	disk_part_iter_init(&piter, gp, DISK_PITER_INCL_EMPTY_PART0);
1722
	while ((hd = disk_part_iter_next(&piter))) {
1723
		part_stat_read_all(hd, &stat);
1724 1725 1726
		if (queue_is_mq(gp->queue))
			inflight = blk_mq_in_flight(gp->queue, hd);
		else
1727
			inflight = part_in_flight(hd);
1728

1729 1730 1731 1732
		seq_printf(seqf, "%4d %7d %s "
			   "%lu %lu %lu %u "
			   "%lu %lu %lu %u "
			   "%u %u %u "
1733 1734 1735
			   "%lu %lu %lu %u "
			   "%lu %u"
			   "\n",
1736 1737
			   MAJOR(part_devt(hd)), MINOR(part_devt(hd)),
			   disk_name(gp, hd->partno, buf),
1738 1739 1740 1741 1742 1743 1744 1745 1746 1747
			   stat.ios[STAT_READ],
			   stat.merges[STAT_READ],
			   stat.sectors[STAT_READ],
			   (unsigned int)div_u64(stat.nsecs[STAT_READ],
							NSEC_PER_MSEC),
			   stat.ios[STAT_WRITE],
			   stat.merges[STAT_WRITE],
			   stat.sectors[STAT_WRITE],
			   (unsigned int)div_u64(stat.nsecs[STAT_WRITE],
							NSEC_PER_MSEC),
1748
			   inflight,
1749
			   jiffies_to_msecs(stat.io_ticks),
1750 1751 1752 1753 1754
			   (unsigned int)div_u64(stat.nsecs[STAT_READ] +
						 stat.nsecs[STAT_WRITE] +
						 stat.nsecs[STAT_DISCARD] +
						 stat.nsecs[STAT_FLUSH],
							NSEC_PER_MSEC),
1755 1756 1757 1758 1759 1760 1761 1762
			   stat.ios[STAT_DISCARD],
			   stat.merges[STAT_DISCARD],
			   stat.sectors[STAT_DISCARD],
			   (unsigned int)div_u64(stat.nsecs[STAT_DISCARD],
						 NSEC_PER_MSEC),
			   stat.ios[STAT_FLUSH],
			   (unsigned int)div_u64(stat.nsecs[STAT_FLUSH],
						 NSEC_PER_MSEC)
1763
			);
L
Linus Torvalds 已提交
1764
	}
1765
	disk_part_iter_exit(&piter);
1766

L
Linus Torvalds 已提交
1767 1768 1769
	return 0;
}

1770
static const struct seq_operations diskstats_op = {
1771 1772 1773
	.start	= disk_seqf_start,
	.next	= disk_seqf_next,
	.stop	= disk_seqf_stop,
L
Linus Torvalds 已提交
1774 1775
	.show	= diskstats_show
};
1776 1777 1778

static int __init proc_genhd_init(void)
{
1779 1780
	proc_create_seq("diskstats", 0, NULL, &diskstats_op);
	proc_create_seq("partitions", 0, NULL, &partitions_op);
1781 1782 1783
	return 0;
}
module_init(proc_genhd_init);
1784
#endif /* CONFIG_PROC_FS */
L
Linus Torvalds 已提交
1785

1786
dev_t blk_lookup_devt(const char *name, int partno)
1787
{
1788 1789 1790
	dev_t devt = MKDEV(0, 0);
	struct class_dev_iter iter;
	struct device *dev;
1791

1792 1793
	class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
	while ((dev = class_dev_iter_next(&iter))) {
1794
		struct gendisk *disk = dev_to_disk(dev);
T
Tejun Heo 已提交
1795
		struct hd_struct *part;
1796

1797
		if (strcmp(dev_name(dev), name))
1798 1799
			continue;

1800 1801 1802 1803 1804 1805 1806 1807
		if (partno < disk->minors) {
			/* We need to return the right devno, even
			 * if the partition doesn't exist yet.
			 */
			devt = MKDEV(MAJOR(dev->devt),
				     MINOR(dev->devt) + partno);
			break;
		}
T
Tejun Heo 已提交
1808
		part = disk_get_part(disk, partno);
1809
		if (part) {
1810
			devt = part_devt(part);
1811
			disk_put_part(part);
T
Tejun Heo 已提交
1812
			break;
1813
		}
T
Tejun Heo 已提交
1814
		disk_put_part(part);
1815
	}
1816
	class_dev_iter_exit(&iter);
1817 1818 1819
	return devt;
}

1820
struct gendisk *__alloc_disk_node(int minors, int node_id)
1821 1822
{
	struct gendisk *disk;
1823
	struct disk_part_tbl *ptbl;
1824

1825 1826
	if (minors > DISK_MAX_PARTS) {
		printk(KERN_ERR
R
Randy Dunlap 已提交
1827
			"block: can't allocate more than %d partitions\n",
1828 1829 1830
			DISK_MAX_PARTS);
		minors = DISK_MAX_PARTS;
	}
1831

1832
	disk = kzalloc_node(sizeof(struct gendisk), GFP_KERNEL, node_id);
1833 1834
	if (!disk)
		return NULL;
1835

1836 1837 1838
	disk->part0.dkstats = alloc_percpu(struct disk_stats);
	if (!disk->part0.dkstats)
		goto out_free_disk;
T
Tejun Heo 已提交
1839

1840 1841 1842 1843 1844
	init_rwsem(&disk->lookup_sem);
	disk->node_id = node_id;
	if (disk_expand_part_tbl(disk, 0)) {
		free_percpu(disk->part0.dkstats);
		goto out_free_disk;
L
Linus Torvalds 已提交
1845
	}
1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867

	ptbl = rcu_dereference_protected(disk->part_tbl, 1);
	rcu_assign_pointer(ptbl->part[0], &disk->part0);

	/*
	 * set_capacity() and get_capacity() currently don't use
	 * seqcounter to read/update the part0->nr_sects. Still init
	 * the counter as we can read the sectors in IO submission
	 * patch using seqence counters.
	 *
	 * TODO: Ideally set_capacity() and get_capacity() should be
	 * converted to make use of bd_mutex and sequence counters.
	 */
	hd_sects_seq_init(&disk->part0);
	if (hd_ref_init(&disk->part0))
		goto out_free_part0;

	disk->minors = minors;
	rand_initialize_disk(disk);
	disk_to_dev(disk)->class = &block_class;
	disk_to_dev(disk)->type = &disk_type;
	device_initialize(disk_to_dev(disk));
L
Linus Torvalds 已提交
1868
	return disk;
1869 1870 1871 1872 1873 1874

out_free_part0:
	hd_free_part(&disk->part0);
out_free_disk:
	kfree(disk);
	return NULL;
L
Linus Torvalds 已提交
1875
}
1876
EXPORT_SYMBOL(__alloc_disk_node);
L
Linus Torvalds 已提交
1877

1878 1879
/**
 * get_disk_and_module - increments the gendisk and gendisk fops module refcount
1880
 * @disk: the struct gendisk to increment the refcount for
1881 1882 1883
 *
 * This increments the refcount for the struct gendisk, and the gendisk's
 * fops module owner.
1884 1885
 *
 * Context: Any context.
1886
 */
1887
struct kobject *get_disk_and_module(struct gendisk *disk)
L
Linus Torvalds 已提交
1888 1889 1890 1891 1892 1893 1894 1895 1896
{
	struct module *owner;
	struct kobject *kobj;

	if (!disk->fops)
		return NULL;
	owner = disk->fops->owner;
	if (owner && !try_module_get(owner))
		return NULL;
J
Jan Kara 已提交
1897
	kobj = kobject_get_unless_zero(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1898 1899 1900 1901 1902 1903 1904
	if (kobj == NULL) {
		module_put(owner);
		return NULL;
	}
	return kobj;

}
1905
EXPORT_SYMBOL(get_disk_and_module);
L
Linus Torvalds 已提交
1906

1907 1908
/**
 * put_disk - decrements the gendisk refcount
1909
 * @disk: the struct gendisk to decrement the refcount for
1910 1911 1912
 *
 * This decrements the refcount for the struct gendisk. When this reaches 0
 * we'll have disk_release() called.
1913 1914 1915
 *
 * Context: Any context, but the last reference must not be dropped from
 *          atomic context.
1916
 */
L
Linus Torvalds 已提交
1917 1918 1919
void put_disk(struct gendisk *disk)
{
	if (disk)
1920
		kobject_put(&disk_to_dev(disk)->kobj);
L
Linus Torvalds 已提交
1921 1922 1923
}
EXPORT_SYMBOL(put_disk);

1924 1925
/**
 * put_disk_and_module - decrements the module and gendisk refcount
1926
 * @disk: the struct gendisk to decrement the refcount for
1927
 *
1928 1929
 * This is a counterpart of get_disk_and_module() and thus also of
 * get_gendisk().
1930 1931 1932
 *
 * Context: Any context, but the last reference must not be dropped from
 *          atomic context.
1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944
 */
void put_disk_and_module(struct gendisk *disk)
{
	if (disk) {
		struct module *owner = disk->fops->owner;

		put_disk(disk);
		module_put(owner);
	}
}
EXPORT_SYMBOL(put_disk_and_module);

1945 1946 1947 1948 1949 1950 1951 1952 1953 1954
static void set_disk_ro_uevent(struct gendisk *gd, int ro)
{
	char event[] = "DISK_RO=1";
	char *envp[] = { event, NULL };

	if (!ro)
		event[8] = '0';
	kobject_uevent_env(&disk_to_dev(gd)->kobj, KOBJ_CHANGE, envp);
}

L
Linus Torvalds 已提交
1955 1956
void set_disk_ro(struct gendisk *disk, int flag)
{
1957 1958 1959
	struct disk_part_iter piter;
	struct hd_struct *part;

1960 1961 1962 1963 1964 1965
	if (disk->part0.policy != flag) {
		set_disk_ro_uevent(disk, flag);
		disk->part0.policy = flag;
	}

	disk_part_iter_init(&piter, disk, DISK_PITER_INCL_EMPTY);
1966 1967 1968
	while ((part = disk_part_iter_next(&piter)))
		part->policy = flag;
	disk_part_iter_exit(&piter);
L
Linus Torvalds 已提交
1969 1970 1971 1972 1973 1974 1975 1976
}

EXPORT_SYMBOL(set_disk_ro);

int bdev_read_only(struct block_device *bdev)
{
	if (!bdev)
		return 0;
T
Tejun Heo 已提交
1977
	return bdev->bd_part->policy;
L
Linus Torvalds 已提交
1978 1979 1980 1981
}

EXPORT_SYMBOL(bdev_read_only);

1982 1983 1984 1985 1986 1987 1988 1989
/*
 * Disk events - monitor disk events like media change and eject request.
 */
struct disk_events {
	struct list_head	node;		/* all disk_event's */
	struct gendisk		*disk;		/* the associated disk */
	spinlock_t		lock;

1990
	struct mutex		block_mutex;	/* protects blocking */
1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013
	int			block;		/* event blocking depth */
	unsigned int		pending;	/* events already sent out */
	unsigned int		clearing;	/* events being cleared */

	long			poll_msecs;	/* interval, -1 for default */
	struct delayed_work	dwork;
};

static const char *disk_events_strs[] = {
	[ilog2(DISK_EVENT_MEDIA_CHANGE)]	= "media_change",
	[ilog2(DISK_EVENT_EJECT_REQUEST)]	= "eject_request",
};

static char *disk_uevents[] = {
	[ilog2(DISK_EVENT_MEDIA_CHANGE)]	= "DISK_MEDIA_CHANGE=1",
	[ilog2(DISK_EVENT_EJECT_REQUEST)]	= "DISK_EJECT_REQUEST=1",
};

/* list of all disk_events */
static DEFINE_MUTEX(disk_events_mutex);
static LIST_HEAD(disk_events);

/* disable in-kernel polling by default */
2014
static unsigned long disk_events_dfl_poll_msecs;
2015 2016 2017 2018 2019 2020 2021 2022

static unsigned long disk_events_poll_jiffies(struct gendisk *disk)
{
	struct disk_events *ev = disk->ev;
	long intv_msecs = 0;

	/*
	 * If device-specific poll interval is set, always use it.  If
2023
	 * the default is being used, poll if the POLL flag is set.
2024 2025 2026
	 */
	if (ev->poll_msecs >= 0)
		intv_msecs = ev->poll_msecs;
2027
	else if (disk->event_flags & DISK_EVENT_FLAG_POLL)
2028 2029 2030 2031 2032
		intv_msecs = disk_events_dfl_poll_msecs;

	return msecs_to_jiffies(intv_msecs);
}

2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048
/**
 * disk_block_events - block and flush disk event checking
 * @disk: disk to block events for
 *
 * On return from this function, it is guaranteed that event checking
 * isn't in progress and won't happen until unblocked by
 * disk_unblock_events().  Events blocking is counted and the actual
 * unblocking happens after the matching number of unblocks are done.
 *
 * Note that this intentionally does not block event checking from
 * disk_clear_events().
 *
 * CONTEXT:
 * Might sleep.
 */
void disk_block_events(struct gendisk *disk)
2049 2050 2051 2052 2053
{
	struct disk_events *ev = disk->ev;
	unsigned long flags;
	bool cancel;

2054 2055 2056
	if (!ev)
		return;

2057 2058 2059 2060 2061 2062
	/*
	 * Outer mutex ensures that the first blocker completes canceling
	 * the event work before further blockers are allowed to finish.
	 */
	mutex_lock(&ev->block_mutex);

2063 2064 2065 2066
	spin_lock_irqsave(&ev->lock, flags);
	cancel = !ev->block++;
	spin_unlock_irqrestore(&ev->lock, flags);

2067 2068
	if (cancel)
		cancel_delayed_work_sync(&disk->ev->dwork);
2069 2070

	mutex_unlock(&ev->block_mutex);
2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088
}

static void __disk_unblock_events(struct gendisk *disk, bool check_now)
{
	struct disk_events *ev = disk->ev;
	unsigned long intv;
	unsigned long flags;

	spin_lock_irqsave(&ev->lock, flags);

	if (WARN_ON_ONCE(ev->block <= 0))
		goto out_unlock;

	if (--ev->block)
		goto out_unlock;

	intv = disk_events_poll_jiffies(disk);
	if (check_now)
2089 2090
		queue_delayed_work(system_freezable_power_efficient_wq,
				&ev->dwork, 0);
2091
	else if (intv)
2092 2093
		queue_delayed_work(system_freezable_power_efficient_wq,
				&ev->dwork, intv);
2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110
out_unlock:
	spin_unlock_irqrestore(&ev->lock, flags);
}

/**
 * disk_unblock_events - unblock disk event checking
 * @disk: disk to unblock events for
 *
 * Undo disk_block_events().  When the block count reaches zero, it
 * starts events polling if configured.
 *
 * CONTEXT:
 * Don't care.  Safe to call from irq context.
 */
void disk_unblock_events(struct gendisk *disk)
{
	if (disk->ev)
2111
		__disk_unblock_events(disk, false);
2112 2113 2114
}

/**
2115 2116 2117
 * disk_flush_events - schedule immediate event checking and flushing
 * @disk: disk to check and flush events for
 * @mask: events to flush
2118
 *
2119 2120 2121
 * Schedule immediate event checking on @disk if not blocked.  Events in
 * @mask are scheduled to be cleared from the driver.  Note that this
 * doesn't clear the events from @disk->ev.
2122 2123
 *
 * CONTEXT:
2124
 * If @mask is non-zero must be called with bdev->bd_mutex held.
2125
 */
2126
void disk_flush_events(struct gendisk *disk, unsigned int mask)
2127
{
2128 2129 2130 2131 2132
	struct disk_events *ev = disk->ev;

	if (!ev)
		return;

2133 2134
	spin_lock_irq(&ev->lock);
	ev->clearing |= mask;
2135
	if (!ev->block)
2136 2137
		mod_delayed_work(system_freezable_power_efficient_wq,
				&ev->dwork, 0);
2138
	spin_unlock_irq(&ev->lock);
2139 2140 2141 2142 2143
}

/**
 * disk_clear_events - synchronously check, clear and return pending events
 * @disk: disk to fetch and clear events from
2144
 * @mask: mask of events to be fetched and cleared
2145 2146 2147 2148 2149 2150 2151
 *
 * Disk events are synchronously checked and pending events in @mask
 * are cleared and returned.  This ignores the block count.
 *
 * CONTEXT:
 * Might sleep.
 */
2152
static unsigned int disk_clear_events(struct gendisk *disk, unsigned int mask)
2153 2154 2155
{
	struct disk_events *ev = disk->ev;
	unsigned int pending;
D
Derek Basehore 已提交
2156
	unsigned int clearing = mask;
2157

2158
	if (!ev)
2159 2160
		return 0;

D
Derek Basehore 已提交
2161 2162 2163 2164 2165 2166 2167
	disk_block_events(disk);

	/*
	 * store the union of mask and ev->clearing on the stack so that the
	 * race with disk_flush_events does not cause ambiguity (ev->clearing
	 * can still be modified even if events are blocked).
	 */
2168
	spin_lock_irq(&ev->lock);
D
Derek Basehore 已提交
2169 2170
	clearing |= ev->clearing;
	ev->clearing = 0;
2171 2172
	spin_unlock_irq(&ev->lock);

D
Derek Basehore 已提交
2173
	disk_check_events(ev, &clearing);
2174
	/*
D
Derek Basehore 已提交
2175 2176
	 * if ev->clearing is not 0, the disk_flush_events got called in the
	 * middle of this function, so we want to run the workfn without delay.
2177
	 */
D
Derek Basehore 已提交
2178
	__disk_unblock_events(disk, ev->clearing ? true : false);
2179 2180 2181 2182 2183 2184

	/* then, fetch and clear pending events */
	spin_lock_irq(&ev->lock);
	pending = ev->pending & mask;
	ev->pending &= ~mask;
	spin_unlock_irq(&ev->lock);
D
Derek Basehore 已提交
2185
	WARN_ON_ONCE(clearing & mask);
2186 2187 2188 2189

	return pending;
}

2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211
/**
 * bdev_check_media_change - check if a removable media has been changed
 * @bdev: block device to check
 *
 * Check whether a removable media has been changed, and attempt to free all
 * dentries and inodes and invalidates all block device page cache entries in
 * that case.
 *
 * Returns %true if the block device changed, or %false if not.
 */
bool bdev_check_media_change(struct block_device *bdev)
{
	unsigned int events;

	events = disk_clear_events(bdev->bd_disk, DISK_EVENT_MEDIA_CHANGE |
				   DISK_EVENT_EJECT_REQUEST);
	if (!(events & DISK_EVENT_MEDIA_CHANGE))
		return false;

	if (__invalidate_device(bdev, true))
		pr_warn("VFS: busy inodes on changed media %s\n",
			bdev->bd_disk->disk_name);
2212
	set_bit(GD_NEED_PART_SCAN, &bdev->bd_disk->state);
2213 2214 2215 2216
	return true;
}
EXPORT_SYMBOL(bdev_check_media_change);

D
Derek Basehore 已提交
2217 2218 2219 2220
/*
 * Separate this part out so that a different pointer for clearing_ptr can be
 * passed in for disk_clear_events.
 */
2221 2222 2223 2224
static void disk_events_workfn(struct work_struct *work)
{
	struct delayed_work *dwork = to_delayed_work(work);
	struct disk_events *ev = container_of(dwork, struct disk_events, dwork);
D
Derek Basehore 已提交
2225 2226 2227 2228 2229 2230 2231

	disk_check_events(ev, &ev->clearing);
}

static void disk_check_events(struct disk_events *ev,
			      unsigned int *clearing_ptr)
{
2232 2233
	struct gendisk *disk = ev->disk;
	char *envp[ARRAY_SIZE(disk_uevents) + 1] = { };
D
Derek Basehore 已提交
2234
	unsigned int clearing = *clearing_ptr;
2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246
	unsigned int events;
	unsigned long intv;
	int nr_events = 0, i;

	/* check events */
	events = disk->fops->check_events(disk, clearing);

	/* accumulate pending events and schedule next poll if necessary */
	spin_lock_irq(&ev->lock);

	events &= ~ev->pending;
	ev->pending |= events;
D
Derek Basehore 已提交
2247
	*clearing_ptr &= ~clearing;
2248 2249 2250

	intv = disk_events_poll_jiffies(disk);
	if (!ev->block && intv)
2251 2252
		queue_delayed_work(system_freezable_power_efficient_wq,
				&ev->dwork, intv);
2253 2254 2255

	spin_unlock_irq(&ev->lock);

2256 2257
	/*
	 * Tell userland about new events.  Only the events listed in
2258 2259 2260
	 * @disk->events are reported, and only if DISK_EVENT_FLAG_UEVENT
	 * is set. Otherwise, events are processed internally but never
	 * get reported to userland.
2261
	 */
2262
	for (i = 0; i < ARRAY_SIZE(disk_uevents); i++)
2263 2264
		if ((events & disk->events & (1 << i)) &&
		    (disk->event_flags & DISK_EVENT_FLAG_UEVENT))
2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276
			envp[nr_events++] = disk_uevents[i];

	if (nr_events)
		kobject_uevent_env(&disk_to_dev(disk)->kobj, KOBJ_CHANGE, envp);
}

/*
 * A disk events enabled device has the following sysfs nodes under
 * its /sys/block/X/ directory.
 *
 * events		: list of all supported events
 * events_async		: list of events which can be detected w/o polling
2277
 *			  (always empty, only for backwards compatibility)
2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301
 * events_poll_msecs	: polling interval, 0: disable, -1: system default
 */
static ssize_t __disk_events_show(unsigned int events, char *buf)
{
	const char *delim = "";
	ssize_t pos = 0;
	int i;

	for (i = 0; i < ARRAY_SIZE(disk_events_strs); i++)
		if (events & (1 << i)) {
			pos += sprintf(buf + pos, "%s%s",
				       delim, disk_events_strs[i]);
			delim = " ";
		}
	if (pos)
		pos += sprintf(buf + pos, "\n");
	return pos;
}

static ssize_t disk_events_show(struct device *dev,
				struct device_attribute *attr, char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

2302 2303 2304
	if (!(disk->event_flags & DISK_EVENT_FLAG_UEVENT))
		return 0;

2305 2306 2307 2308 2309 2310
	return __disk_events_show(disk->events, buf);
}

static ssize_t disk_events_async_show(struct device *dev,
				      struct device_attribute *attr, char *buf)
{
2311
	return 0;
2312 2313 2314 2315 2316 2317 2318 2319
}

static ssize_t disk_events_poll_msecs_show(struct device *dev,
					   struct device_attribute *attr,
					   char *buf)
{
	struct gendisk *disk = dev_to_disk(dev);

2320 2321 2322
	if (!disk->ev)
		return sprintf(buf, "-1\n");

2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338
	return sprintf(buf, "%ld\n", disk->ev->poll_msecs);
}

static ssize_t disk_events_poll_msecs_store(struct device *dev,
					    struct device_attribute *attr,
					    const char *buf, size_t count)
{
	struct gendisk *disk = dev_to_disk(dev);
	long intv;

	if (!count || !sscanf(buf, "%ld", &intv))
		return -EINVAL;

	if (intv < 0 && intv != -1)
		return -EINVAL;

2339 2340 2341
	if (!disk->ev)
		return -ENODEV;

2342
	disk_block_events(disk);
2343 2344 2345 2346 2347 2348
	disk->ev->poll_msecs = intv;
	__disk_unblock_events(disk, true);

	return count;
}

2349 2350 2351
static const DEVICE_ATTR(events, 0444, disk_events_show, NULL);
static const DEVICE_ATTR(events_async, 0444, disk_events_async_show, NULL);
static const DEVICE_ATTR(events_poll_msecs, 0644,
2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
			 disk_events_poll_msecs_show,
			 disk_events_poll_msecs_store);

static const struct attribute *disk_events_attrs[] = {
	&dev_attr_events.attr,
	&dev_attr_events_async.attr,
	&dev_attr_events_poll_msecs.attr,
	NULL,
};

/*
 * The default polling interval can be specified by the kernel
 * parameter block.events_dfl_poll_msecs which defaults to 0
 * (disable).  This can also be modified runtime by writing to
2366
 * /sys/module/block/parameters/events_dfl_poll_msecs.
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380
 */
static int disk_events_set_dfl_poll_msecs(const char *val,
					  const struct kernel_param *kp)
{
	struct disk_events *ev;
	int ret;

	ret = param_set_ulong(val, kp);
	if (ret < 0)
		return ret;

	mutex_lock(&disk_events_mutex);

	list_for_each_entry(ev, &disk_events, node)
2381
		disk_flush_events(ev->disk, 0);
2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399

	mutex_unlock(&disk_events_mutex);

	return 0;
}

static const struct kernel_param_ops disk_events_dfl_poll_msecs_param_ops = {
	.set	= disk_events_set_dfl_poll_msecs,
	.get	= param_get_ulong,
};

#undef MODULE_PARAM_PREFIX
#define MODULE_PARAM_PREFIX	"block."

module_param_cb(events_dfl_poll_msecs, &disk_events_dfl_poll_msecs_param_ops,
		&disk_events_dfl_poll_msecs, 0644);

/*
2400
 * disk_{alloc|add|del|release}_events - initialize and destroy disk_events.
2401
 */
2402
static void disk_alloc_events(struct gendisk *disk)
2403 2404 2405
{
	struct disk_events *ev;

2406
	if (!disk->fops->check_events || !disk->events)
2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417
		return;

	ev = kzalloc(sizeof(*ev), GFP_KERNEL);
	if (!ev) {
		pr_warn("%s: failed to initialize events\n", disk->disk_name);
		return;
	}

	INIT_LIST_HEAD(&ev->node);
	ev->disk = disk;
	spin_lock_init(&ev->lock);
2418
	mutex_init(&ev->block_mutex);
2419 2420 2421 2422
	ev->block = 1;
	ev->poll_msecs = -1;
	INIT_DELAYED_WORK(&ev->dwork, disk_events_workfn);

2423 2424 2425 2426 2427 2428 2429 2430 2431 2432
	disk->ev = ev;
}

static void disk_add_events(struct gendisk *disk)
{
	/* FIXME: error handling */
	if (sysfs_create_files(&disk_to_dev(disk)->kobj, disk_events_attrs) < 0)
		pr_warn("%s: failed to create sysfs files for events\n",
			disk->disk_name);

2433 2434 2435
	if (!disk->ev)
		return;

2436
	mutex_lock(&disk_events_mutex);
2437
	list_add_tail(&disk->ev->node, &disk_events);
2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448
	mutex_unlock(&disk_events_mutex);

	/*
	 * Block count is initialized to 1 and the following initial
	 * unblock kicks it into action.
	 */
	__disk_unblock_events(disk, true);
}

static void disk_del_events(struct gendisk *disk)
{
2449 2450
	if (disk->ev) {
		disk_block_events(disk);
2451

2452 2453 2454 2455
		mutex_lock(&disk_events_mutex);
		list_del_init(&disk->ev->node);
		mutex_unlock(&disk_events_mutex);
	}
2456 2457 2458 2459 2460 2461 2462 2463 2464 2465

	sysfs_remove_files(&disk_to_dev(disk)->kobj, disk_events_attrs);
}

static void disk_release_events(struct gendisk *disk)
{
	/* the block count should be 1 from disk_del_events() */
	WARN_ON_ONCE(disk->ev && disk->ev->block != 1);
	kfree(disk->ev);
}