backing-dev.c 27.3 KB
Newer Older
1 2 3

#include <linux/wait.h>
#include <linux/backing-dev.h>
4 5
#include <linux/kthread.h>
#include <linux/freezer.h>
6
#include <linux/fs.h>
7
#include <linux/pagemap.h>
8
#include <linux/mm.h>
9 10
#include <linux/sched.h>
#include <linux/module.h>
11 12
#include <linux/writeback.h>
#include <linux/device.h>
13
#include <trace/events/writeback.h>
14

15 16
static atomic_long_t bdi_seq = ATOMIC_LONG_INIT(0);

J
Jörn Engel 已提交
17 18
struct backing_dev_info noop_backing_dev_info = {
	.name		= "noop",
19
	.capabilities	= BDI_CAP_NO_ACCT_AND_WRITEBACK,
J
Jörn Engel 已提交
20
};
T
Tejun Heo 已提交
21
EXPORT_SYMBOL_GPL(noop_backing_dev_info);
J
Jörn Engel 已提交
22

23
static struct class *bdi_class;
24 25

/*
26
 * bdi_lock protects updates to bdi_list. bdi_list has RCU reader side
27 28
 * locking.
 */
29
DEFINE_SPINLOCK(bdi_lock);
30
LIST_HEAD(bdi_list);
31

32 33 34
/* bdi_wq serves all asynchronous writeback tasks */
struct workqueue_struct *bdi_wq;

35 36 37 38 39 40 41 42 43 44 45 46 47 48
#ifdef CONFIG_DEBUG_FS
#include <linux/debugfs.h>
#include <linux/seq_file.h>

static struct dentry *bdi_debug_root;

static void bdi_debug_init(void)
{
	bdi_debug_root = debugfs_create_dir("bdi", NULL);
}

static int bdi_debug_stats_show(struct seq_file *m, void *v)
{
	struct backing_dev_info *bdi = m->private;
C
Christoph Hellwig 已提交
49
	struct bdi_writeback *wb = &bdi->wb;
50 51
	unsigned long background_thresh;
	unsigned long dirty_thresh;
T
Tejun Heo 已提交
52
	unsigned long wb_thresh;
53
	unsigned long nr_dirty, nr_io, nr_more_io, nr_dirty_time;
54 55
	struct inode *inode;

56
	nr_dirty = nr_io = nr_more_io = nr_dirty_time = 0;
57
	spin_lock(&wb->list_lock);
58
	list_for_each_entry(inode, &wb->b_dirty, i_io_list)
C
Christoph Hellwig 已提交
59
		nr_dirty++;
60
	list_for_each_entry(inode, &wb->b_io, i_io_list)
C
Christoph Hellwig 已提交
61
		nr_io++;
62
	list_for_each_entry(inode, &wb->b_more_io, i_io_list)
C
Christoph Hellwig 已提交
63
		nr_more_io++;
64
	list_for_each_entry(inode, &wb->b_dirty_time, i_io_list)
65 66
		if (inode->i_state & I_DIRTY_TIME)
			nr_dirty_time++;
67
	spin_unlock(&wb->list_lock);
68

69
	global_dirty_limits(&background_thresh, &dirty_thresh);
T
Tejun Heo 已提交
70
	wb_thresh = wb_calc_thresh(wb, dirty_thresh);
71 72 73

#define K(x) ((x) << (PAGE_SHIFT - 10))
	seq_printf(m,
74 75 76 77 78
		   "BdiWriteback:       %10lu kB\n"
		   "BdiReclaimable:     %10lu kB\n"
		   "BdiDirtyThresh:     %10lu kB\n"
		   "DirtyThresh:        %10lu kB\n"
		   "BackgroundThresh:   %10lu kB\n"
79
		   "BdiDirtied:         %10lu kB\n"
80 81 82 83 84
		   "BdiWritten:         %10lu kB\n"
		   "BdiWriteBandwidth:  %10lu kBps\n"
		   "b_dirty:            %10lu\n"
		   "b_io:               %10lu\n"
		   "b_more_io:          %10lu\n"
85
		   "b_dirty_time:       %10lu\n"
86 87
		   "bdi_list:           %10u\n"
		   "state:              %10lx\n",
88 89
		   (unsigned long) K(wb_stat(wb, WB_WRITEBACK)),
		   (unsigned long) K(wb_stat(wb, WB_RECLAIMABLE)),
T
Tejun Heo 已提交
90
		   K(wb_thresh),
91 92
		   K(dirty_thresh),
		   K(background_thresh),
93 94
		   (unsigned long) K(wb_stat(wb, WB_DIRTIED)),
		   (unsigned long) K(wb_stat(wb, WB_WRITTEN)),
95
		   (unsigned long) K(wb->write_bandwidth),
96 97 98
		   nr_dirty,
		   nr_io,
		   nr_more_io,
99
		   nr_dirty_time,
100
		   !list_empty(&bdi->bdi_list), bdi->wb.state);
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 133 134 135 136 137 138 139 140 141 142
#undef K

	return 0;
}

static int bdi_debug_stats_open(struct inode *inode, struct file *file)
{
	return single_open(file, bdi_debug_stats_show, inode->i_private);
}

static const struct file_operations bdi_debug_stats_fops = {
	.open		= bdi_debug_stats_open,
	.read		= seq_read,
	.llseek		= seq_lseek,
	.release	= single_release,
};

static void bdi_debug_register(struct backing_dev_info *bdi, const char *name)
{
	bdi->debug_dir = debugfs_create_dir(name, bdi_debug_root);
	bdi->debug_stats = debugfs_create_file("stats", 0444, bdi->debug_dir,
					       bdi, &bdi_debug_stats_fops);
}

static void bdi_debug_unregister(struct backing_dev_info *bdi)
{
	debugfs_remove(bdi->debug_stats);
	debugfs_remove(bdi->debug_dir);
}
#else
static inline void bdi_debug_init(void)
{
}
static inline void bdi_debug_register(struct backing_dev_info *bdi,
				      const char *name)
{
}
static inline void bdi_debug_unregister(struct backing_dev_info *bdi)
{
}
#endif

143 144 145 146 147 148
static ssize_t read_ahead_kb_store(struct device *dev,
				  struct device_attribute *attr,
				  const char *buf, size_t count)
{
	struct backing_dev_info *bdi = dev_get_drvdata(dev);
	unsigned long read_ahead_kb;
149
	ssize_t ret;
150

151 152 153 154 155 156 157
	ret = kstrtoul(buf, 10, &read_ahead_kb);
	if (ret < 0)
		return ret;

	bdi->ra_pages = read_ahead_kb >> (PAGE_SHIFT - 10);

	return count;
158 159 160 161 162 163 164 165 166 167 168
}

#define K(pages) ((pages) << (PAGE_SHIFT - 10))

#define BDI_SHOW(name, expr)						\
static ssize_t name##_show(struct device *dev,				\
			   struct device_attribute *attr, char *page)	\
{									\
	struct backing_dev_info *bdi = dev_get_drvdata(dev);		\
									\
	return snprintf(page, PAGE_SIZE-1, "%lld\n", (long long)expr);	\
169 170
}									\
static DEVICE_ATTR_RW(name);
171 172 173

BDI_SHOW(read_ahead_kb, K(bdi->ra_pages))

174 175 176 177 178
static ssize_t min_ratio_store(struct device *dev,
		struct device_attribute *attr, const char *buf, size_t count)
{
	struct backing_dev_info *bdi = dev_get_drvdata(dev);
	unsigned int ratio;
179 180 181 182 183 184 185 186 187
	ssize_t ret;

	ret = kstrtouint(buf, 10, &ratio);
	if (ret < 0)
		return ret;

	ret = bdi_set_min_ratio(bdi, ratio);
	if (!ret)
		ret = count;
188 189 190 191 192

	return ret;
}
BDI_SHOW(min_ratio, bdi->min_ratio)

193 194 195 196 197
static ssize_t max_ratio_store(struct device *dev,
		struct device_attribute *attr, const char *buf, size_t count)
{
	struct backing_dev_info *bdi = dev_get_drvdata(dev);
	unsigned int ratio;
198 199 200 201 202 203 204 205 206
	ssize_t ret;

	ret = kstrtouint(buf, 10, &ratio);
	if (ret < 0)
		return ret;

	ret = bdi_set_max_ratio(bdi, ratio);
	if (!ret)
		ret = count;
207 208 209 210 211

	return ret;
}
BDI_SHOW(max_ratio, bdi->max_ratio)

212 213 214 215 216 217 218 219 220
static ssize_t stable_pages_required_show(struct device *dev,
					  struct device_attribute *attr,
					  char *page)
{
	struct backing_dev_info *bdi = dev_get_drvdata(dev);

	return snprintf(page, PAGE_SIZE-1, "%d\n",
			bdi_cap_stable_pages_required(bdi) ? 1 : 0);
}
221 222 223 224 225 226 227 228
static DEVICE_ATTR_RO(stable_pages_required);

static struct attribute *bdi_dev_attrs[] = {
	&dev_attr_read_ahead_kb.attr,
	&dev_attr_min_ratio.attr,
	&dev_attr_max_ratio.attr,
	&dev_attr_stable_pages_required.attr,
	NULL,
229
};
230
ATTRIBUTE_GROUPS(bdi_dev);
231 232 233 234

static __init int bdi_class_init(void)
{
	bdi_class = class_create(THIS_MODULE, "bdi");
235 236 237
	if (IS_ERR(bdi_class))
		return PTR_ERR(bdi_class);

238
	bdi_class->dev_groups = bdi_dev_groups;
239
	bdi_debug_init();
240

241 242
	return 0;
}
243
postcore_initcall(bdi_class_init);
244

245 246 247 248
static int __init default_bdi_init(void)
{
	int err;

249
	bdi_wq = alloc_workqueue("writeback", WQ_MEM_RECLAIM | WQ_FREEZABLE |
250
					      WQ_UNBOUND | WQ_SYSFS, 0);
251 252 253
	if (!bdi_wq)
		return -ENOMEM;

254
	err = bdi_init(&noop_backing_dev_info);
255 256 257 258 259

	return err;
}
subsys_initcall(default_bdi_init);

260
/*
261
 * This function is used when the first inode for this wb is marked dirty. It
262 263 264 265 266 267 268 269
 * wakes-up the corresponding bdi thread which should then take care of the
 * periodic background write-out of dirty inodes. Since the write-out would
 * starts only 'dirty_writeback_interval' centisecs from now anyway, we just
 * set up a timer which wakes the bdi thread up later.
 *
 * Note, we wouldn't bother setting up the timer, but this function is on the
 * fast-path (used by '__mark_inode_dirty()'), so we save few context switches
 * by delaying the wake-up.
270 271 272
 *
 * We have to be careful not to postpone flush work if it is scheduled for
 * earlier. Thus we use queue_delayed_work().
273
 */
274
void wb_wakeup_delayed(struct bdi_writeback *wb)
275 276 277 278
{
	unsigned long timeout;

	timeout = msecs_to_jiffies(dirty_writeback_interval * 10);
279 280 281 282
	spin_lock_bh(&wb->work_lock);
	if (test_bit(WB_registered, &wb->state))
		queue_delayed_work(bdi_wq, &wb->dwork, timeout);
	spin_unlock_bh(&wb->work_lock);
283 284
}

285
/*
286
 * Initial write bandwidth: 100 MB/s
287
 */
288
#define INIT_BW		(100 << (20 - PAGE_SHIFT))
289

T
Tejun Heo 已提交
290
static int wb_init(struct bdi_writeback *wb, struct backing_dev_info *bdi,
291
		   int blkcg_id, gfp_t gfp)
292
{
293
	int i, err;
294

295
	memset(wb, 0, sizeof(*wb));
296

297 298 299 300 301
	wb->bdi = bdi;
	wb->last_old_flush = jiffies;
	INIT_LIST_HEAD(&wb->b_dirty);
	INIT_LIST_HEAD(&wb->b_io);
	INIT_LIST_HEAD(&wb->b_more_io);
302
	INIT_LIST_HEAD(&wb->b_dirty_time);
303
	spin_lock_init(&wb->list_lock);
304

305 306 307 308 309
	wb->bw_time_stamp = jiffies;
	wb->balanced_dirty_ratelimit = INIT_BW;
	wb->dirty_ratelimit = INIT_BW;
	wb->write_bandwidth = INIT_BW;
	wb->avg_write_bandwidth = INIT_BW;
310

311 312 313
	spin_lock_init(&wb->work_lock);
	INIT_LIST_HEAD(&wb->work_list);
	INIT_DELAYED_WORK(&wb->dwork, wb_workfn);
314
	wb->dirty_sleep = jiffies;
315

316 317 318 319
	wb->congested = wb_congested_get_create(bdi, blkcg_id, gfp);
	if (!wb->congested)
		return -ENOMEM;

T
Tejun Heo 已提交
320
	err = fprop_local_init_percpu(&wb->completions, gfp);
321
	if (err)
322
		goto out_put_cong;
323

324
	for (i = 0; i < NR_WB_STAT_ITEMS; i++) {
T
Tejun Heo 已提交
325
		err = percpu_counter_init(&wb->stat[i], 0, gfp);
326 327
		if (err)
			goto out_destroy_stat;
328
	}
329

330
	return 0;
331 332

out_destroy_stat:
333
	while (i--)
334 335 336 337 338
		percpu_counter_destroy(&wb->stat[i]);
	fprop_local_destroy_percpu(&wb->completions);
out_put_cong:
	wb_congested_put(wb->congested);
	return err;
339 340
}

341 342 343
/*
 * Remove bdi from the global list and shutdown any threads we have running
 */
344
static void wb_shutdown(struct bdi_writeback *wb)
345
{
346
	/* Make sure nobody queues further work */
347 348 349
	spin_lock_bh(&wb->work_lock);
	if (!test_and_clear_bit(WB_registered, &wb->state)) {
		spin_unlock_bh(&wb->work_lock);
350
		return;
351
	}
352
	spin_unlock_bh(&wb->work_lock);
353 354

	/*
355 356 357
	 * Drain work list and shutdown the delayed_work.  !WB_registered
	 * tells wb_workfn() that @wb is dying and its work_list needs to
	 * be drained no matter what.
358
	 */
359 360 361 362 363
	mod_delayed_work(bdi_wq, &wb->dwork, 0);
	flush_delayed_work(&wb->dwork);
	WARN_ON(!list_empty(&wb->work_list));
}

364
static void wb_exit(struct bdi_writeback *wb)
365 366 367 368 369 370 371
{
	int i;

	WARN_ON(delayed_work_pending(&wb->dwork));

	for (i = 0; i < NR_WB_STAT_ITEMS; i++)
		percpu_counter_destroy(&wb->stat[i]);
372

373
	fprop_local_destroy_percpu(&wb->completions);
374
	wb_congested_put(wb->congested);
375
}
376

377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413
#ifdef CONFIG_CGROUP_WRITEBACK

#include <linux/memcontrol.h>

/*
 * cgwb_lock protects bdi->cgwb_tree, bdi->cgwb_congested_tree,
 * blkcg->cgwb_list, and memcg->cgwb_list.  bdi->cgwb_tree is also RCU
 * protected.  cgwb_release_wait is used to wait for the completion of cgwb
 * releases from bdi destruction path.
 */
static DEFINE_SPINLOCK(cgwb_lock);
static DECLARE_WAIT_QUEUE_HEAD(cgwb_release_wait);

/**
 * wb_congested_get_create - get or create a wb_congested
 * @bdi: associated bdi
 * @blkcg_id: ID of the associated blkcg
 * @gfp: allocation mask
 *
 * Look up the wb_congested for @blkcg_id on @bdi.  If missing, create one.
 * The returned wb_congested has its reference count incremented.  Returns
 * NULL on failure.
 */
struct bdi_writeback_congested *
wb_congested_get_create(struct backing_dev_info *bdi, int blkcg_id, gfp_t gfp)
{
	struct bdi_writeback_congested *new_congested = NULL, *congested;
	struct rb_node **node, *parent;
	unsigned long flags;
retry:
	spin_lock_irqsave(&cgwb_lock, flags);

	node = &bdi->cgwb_congested_tree.rb_node;
	parent = NULL;

	while (*node != NULL) {
		parent = *node;
G
Geliang Tang 已提交
414 415
		congested = rb_entry(parent, struct bdi_writeback_congested,
				     rb_node);
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
		if (congested->blkcg_id < blkcg_id)
			node = &parent->rb_left;
		else if (congested->blkcg_id > blkcg_id)
			node = &parent->rb_right;
		else
			goto found;
	}

	if (new_congested) {
		/* !found and storage for new one already allocated, insert */
		congested = new_congested;
		new_congested = NULL;
		rb_link_node(&congested->rb_node, parent, node);
		rb_insert_color(&congested->rb_node, &bdi->cgwb_congested_tree);
		goto found;
	}

	spin_unlock_irqrestore(&cgwb_lock, flags);

	/* allocate storage for new one and retry */
	new_congested = kzalloc(sizeof(*new_congested), gfp);
	if (!new_congested)
		return NULL;

	atomic_set(&new_congested->refcnt, 0);
	new_congested->bdi = bdi;
	new_congested->blkcg_id = blkcg_id;
	goto retry;

found:
	atomic_inc(&congested->refcnt);
	spin_unlock_irqrestore(&cgwb_lock, flags);
	kfree(new_congested);
	return congested;
}

/**
 * wb_congested_put - put a wb_congested
 * @congested: wb_congested to put
 *
 * Put @congested and destroy it if the refcnt reaches zero.
 */
void wb_congested_put(struct bdi_writeback_congested *congested)
{
	unsigned long flags;

	local_irq_save(flags);
	if (!atomic_dec_and_lock(&congested->refcnt, &cgwb_lock)) {
		local_irq_restore(flags);
		return;
	}

468 469 470 471 472 473 474
	/* bdi might already have been destroyed leaving @congested unlinked */
	if (congested->bdi) {
		rb_erase(&congested->rb_node,
			 &congested->bdi->cgwb_congested_tree);
		congested->bdi = NULL;
	}

475 476 477 478 479 480 481 482 483 484
	spin_unlock_irqrestore(&cgwb_lock, flags);
	kfree(congested);
}

static void cgwb_release_workfn(struct work_struct *work)
{
	struct bdi_writeback *wb = container_of(work, struct bdi_writeback,
						release_work);
	struct backing_dev_info *bdi = wb->bdi;

485 486 487 488
	spin_lock_irq(&cgwb_lock);
	list_del_rcu(&wb->bdi_node);
	spin_unlock_irq(&cgwb_lock);

489 490 491 492 493
	wb_shutdown(wb);

	css_put(wb->memcg_css);
	css_put(wb->blkcg_css);

T
Tejun Heo 已提交
494
	fprop_local_destroy_percpu(&wb->memcg_completions);
495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531
	percpu_ref_exit(&wb->refcnt);
	wb_exit(wb);
	kfree_rcu(wb, rcu);

	if (atomic_dec_and_test(&bdi->usage_cnt))
		wake_up_all(&cgwb_release_wait);
}

static void cgwb_release(struct percpu_ref *refcnt)
{
	struct bdi_writeback *wb = container_of(refcnt, struct bdi_writeback,
						refcnt);
	schedule_work(&wb->release_work);
}

static void cgwb_kill(struct bdi_writeback *wb)
{
	lockdep_assert_held(&cgwb_lock);

	WARN_ON(!radix_tree_delete(&wb->bdi->cgwb_tree, wb->memcg_css->id));
	list_del(&wb->memcg_node);
	list_del(&wb->blkcg_node);
	percpu_ref_kill(&wb->refcnt);
}

static int cgwb_create(struct backing_dev_info *bdi,
		       struct cgroup_subsys_state *memcg_css, gfp_t gfp)
{
	struct mem_cgroup *memcg;
	struct cgroup_subsys_state *blkcg_css;
	struct blkcg *blkcg;
	struct list_head *memcg_cgwb_list, *blkcg_cgwb_list;
	struct bdi_writeback *wb;
	unsigned long flags;
	int ret = 0;

	memcg = mem_cgroup_from_css(memcg_css);
532
	blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
	blkcg = css_to_blkcg(blkcg_css);
	memcg_cgwb_list = mem_cgroup_cgwb_list(memcg);
	blkcg_cgwb_list = &blkcg->cgwb_list;

	/* look up again under lock and discard on blkcg mismatch */
	spin_lock_irqsave(&cgwb_lock, flags);
	wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
	if (wb && wb->blkcg_css != blkcg_css) {
		cgwb_kill(wb);
		wb = NULL;
	}
	spin_unlock_irqrestore(&cgwb_lock, flags);
	if (wb)
		goto out_put;

	/* need to create a new one */
	wb = kmalloc(sizeof(*wb), gfp);
	if (!wb)
		return -ENOMEM;

553
	ret = wb_init(wb, bdi, blkcg_css->id, gfp);
554 555 556 557 558 559 560
	if (ret)
		goto err_free;

	ret = percpu_ref_init(&wb->refcnt, cgwb_release, 0, gfp);
	if (ret)
		goto err_wb_exit;

T
Tejun Heo 已提交
561 562 563 564
	ret = fprop_local_init_percpu(&wb->memcg_completions, gfp);
	if (ret)
		goto err_ref_exit;

565 566 567 568
	wb->memcg_css = memcg_css;
	wb->blkcg_css = blkcg_css;
	INIT_WORK(&wb->release_work, cgwb_release_workfn);
	set_bit(WB_registered, &wb->state);
569 570

	/*
571 572 573 574
	 * The root wb determines the registered state of the whole bdi and
	 * memcg_cgwb_list and blkcg_cgwb_list's next pointers indicate
	 * whether they're still online.  Don't link @wb if any is dead.
	 * See wb_memcg_offline() and wb_blkcg_offline().
575
	 */
576 577 578 579 580 581 582 583
	ret = -ENODEV;
	spin_lock_irqsave(&cgwb_lock, flags);
	if (test_bit(WB_registered, &bdi->wb.state) &&
	    blkcg_cgwb_list->next && memcg_cgwb_list->next) {
		/* we might have raced another instance of this function */
		ret = radix_tree_insert(&bdi->cgwb_tree, memcg_css->id, wb);
		if (!ret) {
			atomic_inc(&bdi->usage_cnt);
584
			list_add_tail_rcu(&wb->bdi_node, &bdi->wb_list);
585 586 587 588 589 590 591 592 593 594
			list_add(&wb->memcg_node, memcg_cgwb_list);
			list_add(&wb->blkcg_node, blkcg_cgwb_list);
			css_get(memcg_css);
			css_get(blkcg_css);
		}
	}
	spin_unlock_irqrestore(&cgwb_lock, flags);
	if (ret) {
		if (ret == -EEXIST)
			ret = 0;
595
		goto err_fprop_exit;
596 597 598
	}
	goto out_put;

T
Tejun Heo 已提交
599 600
err_fprop_exit:
	fprop_local_destroy_percpu(&wb->memcg_completions);
601 602 603 604 605 606 607 608 609
err_ref_exit:
	percpu_ref_exit(&wb->refcnt);
err_wb_exit:
	wb_exit(wb);
err_free:
	kfree(wb);
out_put:
	css_put(blkcg_css);
	return ret;
610 611
}

612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638
/**
 * wb_get_create - get wb for a given memcg, create if necessary
 * @bdi: target bdi
 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
 * @gfp: allocation mask to use
 *
 * Try to get the wb for @memcg_css on @bdi.  If it doesn't exist, try to
 * create one.  The returned wb has its refcount incremented.
 *
 * This function uses css_get() on @memcg_css and thus expects its refcnt
 * to be positive on invocation.  IOW, rcu_read_lock() protection on
 * @memcg_css isn't enough.  try_get it before calling this function.
 *
 * A wb is keyed by its associated memcg.  As blkcg implicitly enables
 * memcg on the default hierarchy, memcg association is guaranteed to be
 * more specific (equal or descendant to the associated blkcg) and thus can
 * identify both the memcg and blkcg associations.
 *
 * Because the blkcg associated with a memcg may change as blkcg is enabled
 * and disabled closer to root in the hierarchy, each wb keeps track of
 * both the memcg and blkcg associated with it and verifies the blkcg on
 * each lookup.  On mismatch, the existing wb is discarded and a new one is
 * created.
 */
struct bdi_writeback *wb_get_create(struct backing_dev_info *bdi,
				    struct cgroup_subsys_state *memcg_css,
				    gfp_t gfp)
639
{
640 641
	struct bdi_writeback *wb;

642
	might_sleep_if(gfpflags_allow_blocking(gfp));
643 644 645 646 647 648 649 650 651 652 653 654

	if (!memcg_css->parent)
		return &bdi->wb;

	do {
		rcu_read_lock();
		wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
		if (wb) {
			struct cgroup_subsys_state *blkcg_css;

			/* see whether the blkcg association has changed */
			blkcg_css = cgroup_get_e_css(memcg_css->cgroup,
655
						     &io_cgrp_subsys);
656 657 658 659 660 661 662 663 664 665
			if (unlikely(wb->blkcg_css != blkcg_css ||
				     !wb_tryget(wb)))
				wb = NULL;
			css_put(blkcg_css);
		}
		rcu_read_unlock();
	} while (!wb && !cgwb_create(bdi, memcg_css, gfp));

	return wb;
}
666

667
static int cgwb_bdi_init(struct backing_dev_info *bdi)
668
{
669 670
	int ret;

671 672 673
	INIT_RADIX_TREE(&bdi->cgwb_tree, GFP_ATOMIC);
	bdi->cgwb_congested_tree = RB_ROOT;
	atomic_set(&bdi->usage_cnt, 1);
674 675 676

	ret = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
	if (!ret) {
677
		bdi->wb.memcg_css = &root_mem_cgroup->css;
678 679 680
		bdi->wb.blkcg_css = blkcg_root_css;
	}
	return ret;
681 682
}

683 684 685 686 687 688 689 690 691 692 693 694 695
static void cgwb_bdi_destroy(struct backing_dev_info *bdi)
{
	struct radix_tree_iter iter;
	void **slot;

	WARN_ON(test_bit(WB_registered, &bdi->wb.state));

	spin_lock_irq(&cgwb_lock);
	radix_tree_for_each_slot(slot, &bdi->cgwb_tree, &iter, 0)
		cgwb_kill(*slot);
	spin_unlock_irq(&cgwb_lock);

	/*
696 697
	 * All cgwb's must be shutdown and released before returning.  Drain
	 * the usage counter to wait for all cgwb's ever created on @bdi.
698 699 700
	 */
	atomic_dec(&bdi->usage_cnt);
	wait_event(cgwb_release_wait, !atomic_read(&bdi->usage_cnt));
J
Jan Kara 已提交
701 702 703 704 705
	/*
	 * Grab back our reference so that we hold it when @bdi gets
	 * re-registered.
	 */
	atomic_inc(&bdi->usage_cnt);
706 707 708 709 710 711 712
}

/**
 * wb_memcg_offline - kill all wb's associated with a memcg being offlined
 * @memcg: memcg being offlined
 *
 * Also prevents creation of any new wb's associated with @memcg.
713
 */
714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
void wb_memcg_offline(struct mem_cgroup *memcg)
{
	LIST_HEAD(to_destroy);
	struct list_head *memcg_cgwb_list = mem_cgroup_cgwb_list(memcg);
	struct bdi_writeback *wb, *next;

	spin_lock_irq(&cgwb_lock);
	list_for_each_entry_safe(wb, next, memcg_cgwb_list, memcg_node)
		cgwb_kill(wb);
	memcg_cgwb_list->next = NULL;	/* prevent new wb's */
	spin_unlock_irq(&cgwb_lock);
}

/**
 * wb_blkcg_offline - kill all wb's associated with a blkcg being offlined
 * @blkcg: blkcg being offlined
 *
 * Also prevents creation of any new wb's associated with @blkcg.
 */
void wb_blkcg_offline(struct blkcg *blkcg)
{
	LIST_HEAD(to_destroy);
	struct bdi_writeback *wb, *next;

	spin_lock_irq(&cgwb_lock);
	list_for_each_entry_safe(wb, next, &blkcg->cgwb_list, blkcg_node)
		cgwb_kill(wb);
	blkcg->cgwb_list.next = NULL;	/* prevent new wb's */
	spin_unlock_irq(&cgwb_lock);
}

745 746 747 748 749 750 751 752 753 754 755 756 757 758 759
static void cgwb_bdi_exit(struct backing_dev_info *bdi)
{
	struct rb_node *rbn;

	spin_lock_irq(&cgwb_lock);
	while ((rbn = rb_first(&bdi->cgwb_congested_tree))) {
		struct bdi_writeback_congested *congested =
			rb_entry(rbn, struct bdi_writeback_congested, rb_node);

		rb_erase(rbn, &bdi->cgwb_congested_tree);
		congested->bdi = NULL;	/* mark @congested unlinked */
	}
	spin_unlock_irq(&cgwb_lock);
}

760 761
#else	/* CONFIG_CGROUP_WRITEBACK */

762 763 764 765 766 767 768 769
static int cgwb_bdi_init(struct backing_dev_info *bdi)
{
	int err;

	bdi->wb_congested = kzalloc(sizeof(*bdi->wb_congested), GFP_KERNEL);
	if (!bdi->wb_congested)
		return -ENOMEM;

770 771
	atomic_set(&bdi->wb_congested->refcnt, 1);

772 773
	err = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
	if (err) {
774
		wb_congested_put(bdi->wb_congested);
775 776 777 778 779
		return err;
	}
	return 0;
}

780 781 782
static void cgwb_bdi_destroy(struct backing_dev_info *bdi) { }

static void cgwb_bdi_exit(struct backing_dev_info *bdi)
783 784 785
{
	wb_congested_put(bdi->wb_congested);
}
786 787

#endif	/* CONFIG_CGROUP_WRITEBACK */
788

789 790
int bdi_init(struct backing_dev_info *bdi)
{
791 792
	int ret;

793 794
	bdi->dev = NULL;

795
	kref_init(&bdi->refcnt);
796
	bdi->min_ratio = 0;
797
	bdi->max_ratio = 100;
798
	bdi->max_prop_frac = FPROP_FRAC_BASE;
799
	INIT_LIST_HEAD(&bdi->bdi_list);
800
	INIT_LIST_HEAD(&bdi->wb_list);
801
	init_waitqueue_head(&bdi->wb_waitq);
802

803 804 805 806 807
	ret = cgwb_bdi_init(bdi);

	list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);

	return ret;
808 809
}
EXPORT_SYMBOL(bdi_init);
810

811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826
struct backing_dev_info *bdi_alloc_node(gfp_t gfp_mask, int node_id)
{
	struct backing_dev_info *bdi;

	bdi = kmalloc_node(sizeof(struct backing_dev_info),
			   gfp_mask | __GFP_ZERO, node_id);
	if (!bdi)
		return NULL;

	if (bdi_init(bdi)) {
		kfree(bdi);
		return NULL;
	}
	return bdi;
}

827 828 829 830 831
int bdi_register(struct backing_dev_info *bdi, struct device *parent,
		const char *fmt, ...)
{
	va_list args;
	struct device *dev;
832

833 834
	if (bdi->dev)	/* The driver needs to use separate queues per device */
		return 0;
835

836 837 838 839 840
	va_start(args, fmt);
	dev = device_create_vargs(bdi_class, parent, MKDEV(0, 0), bdi, fmt, args);
	va_end(args);
	if (IS_ERR(dev))
		return PTR_ERR(dev);
P
Peter Zijlstra 已提交
841

842
	bdi->dev = dev;
843

844 845 846 847 848 849 850 851 852
	bdi_debug_register(bdi, dev_name(dev));
	set_bit(WB_registered, &bdi->wb.state);

	spin_lock_bh(&bdi_lock);
	list_add_tail_rcu(&bdi->bdi_list, &bdi_list);
	spin_unlock_bh(&bdi_lock);

	trace_writeback_bdi_register(bdi);
	return 0;
853
}
854
EXPORT_SYMBOL(bdi_register);
855

856
int bdi_register_dev(struct backing_dev_info *bdi, dev_t dev)
857
{
858 859 860 861
	return bdi_register(bdi, NULL, "%u:%u", MAJOR(dev), MINOR(dev));
}
EXPORT_SYMBOL(bdi_register_dev);

862 863 864 865 866 867 868 869
int bdi_register_owner(struct backing_dev_info *bdi, struct device *owner)
{
	int rc;

	rc = bdi_register(bdi, NULL, "%u:%u", MAJOR(owner->devt),
			MINOR(owner->devt));
	if (rc)
		return rc;
J
Jan Kara 已提交
870 871
	/* Leaking owner reference... */
	WARN_ON(bdi->owner);
872 873 874 875 876 877
	bdi->owner = owner;
	get_device(owner);
	return 0;
}
EXPORT_SYMBOL(bdi_register_owner);

878 879 880 881 882 883 884 885
/*
 * Remove bdi from bdi_list, and ensure that it is no longer visible
 */
static void bdi_remove_from_list(struct backing_dev_info *bdi)
{
	spin_lock_bh(&bdi_lock);
	list_del_rcu(&bdi->bdi_list);
	spin_unlock_bh(&bdi_lock);
886

887 888
	synchronize_rcu_expedited();
}
889

890
void bdi_unregister(struct backing_dev_info *bdi)
891
{
892 893 894
	/* make sure nobody finds us on the bdi_list anymore */
	bdi_remove_from_list(bdi);
	wb_shutdown(&bdi->wb);
895
	cgwb_bdi_destroy(bdi);
896

897 898 899 900 901
	if (bdi->dev) {
		bdi_debug_unregister(bdi);
		device_unregister(bdi->dev);
		bdi->dev = NULL;
	}
902 903 904 905 906

	if (bdi->owner) {
		put_device(bdi->owner);
		bdi->owner = NULL;
	}
907
}
908

909
static void bdi_exit(struct backing_dev_info *bdi)
910 911
{
	WARN_ON_ONCE(bdi->dev);
912
	wb_exit(&bdi->wb);
913
	cgwb_bdi_exit(bdi);
914
}
915

916 917 918 919 920 921 922 923 924 925 926 927 928 929
static void release_bdi(struct kref *ref)
{
	struct backing_dev_info *bdi =
			container_of(ref, struct backing_dev_info, refcnt);

	bdi_exit(bdi);
	kfree(bdi);
}

void bdi_put(struct backing_dev_info *bdi)
{
	kref_put(&bdi->refcnt, release_bdi);
}

930 931 932 933 934
void bdi_destroy(struct backing_dev_info *bdi)
{
	bdi_unregister(bdi);
	bdi_exit(bdi);
}
935 936
EXPORT_SYMBOL(bdi_destroy);

937 938 939 940
/*
 * For use from filesystems to quickly init and register a bdi associated
 * with dirty writeback
 */
941
int bdi_setup_and_register(struct backing_dev_info *bdi, char *name)
942 943 944 945
{
	int err;

	bdi->name = name;
946
	bdi->capabilities = 0;
947 948 949 950
	err = bdi_init(bdi);
	if (err)
		return err;

951 952
	err = bdi_register(bdi, NULL, "%.28s-%ld", name,
			   atomic_long_inc_return(&bdi_seq));
953 954 955 956 957 958 959 960 961
	if (err) {
		bdi_destroy(bdi);
		return err;
	}

	return 0;
}
EXPORT_SYMBOL(bdi_setup_and_register);

962 963 964 965
static wait_queue_head_t congestion_wqh[2] = {
		__WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[0]),
		__WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[1])
	};
966
static atomic_t nr_wb_congested[2];
967

968
void clear_wb_congested(struct bdi_writeback_congested *congested, int sync)
969
{
970
	wait_queue_head_t *wqh = &congestion_wqh[sync];
971
	enum wb_congested_state bit;
972

973
	bit = sync ? WB_sync_congested : WB_async_congested;
974 975
	if (test_and_clear_bit(bit, &congested->state))
		atomic_dec(&nr_wb_congested[sync]);
976
	smp_mb__after_atomic();
977 978 979
	if (waitqueue_active(wqh))
		wake_up(wqh);
}
980
EXPORT_SYMBOL(clear_wb_congested);
981

982
void set_wb_congested(struct bdi_writeback_congested *congested, int sync)
983
{
984
	enum wb_congested_state bit;
985

986
	bit = sync ? WB_sync_congested : WB_async_congested;
987 988
	if (!test_and_set_bit(bit, &congested->state))
		atomic_inc(&nr_wb_congested[sync]);
989
}
990
EXPORT_SYMBOL(set_wb_congested);
991 992 993

/**
 * congestion_wait - wait for a backing_dev to become uncongested
994
 * @sync: SYNC or ASYNC IO
995 996 997 998 999 1000
 * @timeout: timeout in jiffies
 *
 * Waits for up to @timeout jiffies for a backing_dev (any backing_dev) to exit
 * write congestion.  If no backing_devs are congested then just wait for the
 * next write to be completed.
 */
1001
long congestion_wait(int sync, long timeout)
1002 1003
{
	long ret;
1004
	unsigned long start = jiffies;
1005
	DEFINE_WAIT(wait);
1006
	wait_queue_head_t *wqh = &congestion_wqh[sync];
1007 1008 1009 1010

	prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
	ret = io_schedule_timeout(timeout);
	finish_wait(wqh, &wait);
1011 1012 1013 1014

	trace_writeback_congestion_wait(jiffies_to_usecs(timeout),
					jiffies_to_usecs(jiffies - start));

1015 1016 1017
	return ret;
}
EXPORT_SYMBOL(congestion_wait);
P
Peter Zijlstra 已提交
1018

1019
/**
M
Mel Gorman 已提交
1020 1021
 * wait_iff_congested - Conditionally wait for a backing_dev to become uncongested or a pgdat to complete writes
 * @pgdat: A pgdat to check if it is heavily congested
1022 1023 1024 1025
 * @sync: SYNC or ASYNC IO
 * @timeout: timeout in jiffies
 *
 * In the event of a congested backing_dev (any backing_dev) and the given
M
Mel Gorman 已提交
1026
 * @pgdat has experienced recent congestion, this waits for up to @timeout
1027 1028 1029
 * jiffies for either a BDI to exit congestion of the given @sync queue
 * or a write to complete.
 *
M
Mel Gorman 已提交
1030
 * In the absence of pgdat congestion, cond_resched() is called to yield
1031
 * the processor if necessary but otherwise does not sleep.
1032 1033 1034 1035 1036
 *
 * The return value is 0 if the sleep is for the full timeout. Otherwise,
 * it is the number of jiffies that were still remaining when the function
 * returned. return_value == timeout implies the function did not sleep.
 */
M
Mel Gorman 已提交
1037
long wait_iff_congested(struct pglist_data *pgdat, int sync, long timeout)
1038 1039 1040 1041 1042 1043 1044 1045
{
	long ret;
	unsigned long start = jiffies;
	DEFINE_WAIT(wait);
	wait_queue_head_t *wqh = &congestion_wqh[sync];

	/*
	 * If there is no congestion, or heavy congestion is not being
M
Mel Gorman 已提交
1046
	 * encountered in the current pgdat, yield if necessary instead
1047 1048
	 * of sleeping on the congestion queue
	 */
1049
	if (atomic_read(&nr_wb_congested[sync]) == 0 ||
M
Mel Gorman 已提交
1050
	    !test_bit(PGDAT_CONGESTED, &pgdat->flags)) {
1051
		cond_resched();
M
Mel Gorman 已提交
1052

1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
		/* In case we scheduled, work out time remaining */
		ret = timeout - (jiffies - start);
		if (ret < 0)
			ret = 0;

		goto out;
	}

	/* Sleep until uncongested or a write happens */
	prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
	ret = io_schedule_timeout(timeout);
	finish_wait(wqh, &wait);

out:
	trace_writeback_wait_iff_congested(jiffies_to_usecs(timeout),
					jiffies_to_usecs(jiffies - start));

	return ret;
}
EXPORT_SYMBOL(wait_iff_congested);
1073 1074 1075 1076 1077 1078

int pdflush_proc_obsolete(struct ctl_table *table, int write,
			void __user *buffer, size_t *lenp, loff_t *ppos)
{
	char kbuf[] = "0\n";

1079
	if (*ppos || *lenp < sizeof(kbuf)) {
1080 1081 1082 1083 1084 1085
		*lenp = 0;
		return 0;
	}

	if (copy_to_user(buffer, kbuf, sizeof(kbuf)))
		return -EFAULT;
1086 1087
	pr_warn_once("%s exported in /proc is scheduled for removal\n",
		     table->procname);
1088 1089 1090 1091 1092

	*lenp = 2;
	*ppos += *lenp;
	return 2;
}