Loading...
1// SPDX-License-Identifier: GPL-2.0-only
2
3#include <linux/wait.h>
4#include <linux/rbtree.h>
5#include <linux/backing-dev.h>
6#include <linux/kthread.h>
7#include <linux/freezer.h>
8#include <linux/fs.h>
9#include <linux/pagemap.h>
10#include <linux/mm.h>
11#include <linux/sched.h>
12#include <linux/module.h>
13#include <linux/writeback.h>
14#include <linux/device.h>
15#include <trace/events/writeback.h>
16
17struct backing_dev_info noop_backing_dev_info = {
18 .name = "noop",
19 .capabilities = BDI_CAP_NO_ACCT_AND_WRITEBACK,
20};
21EXPORT_SYMBOL_GPL(noop_backing_dev_info);
22
23static struct class *bdi_class;
24
25/*
26 * bdi_lock protects bdi_tree and updates to bdi_list. bdi_list has RCU
27 * reader side locking.
28 */
29DEFINE_SPINLOCK(bdi_lock);
30static u64 bdi_id_cursor;
31static struct rb_root bdi_tree = RB_ROOT;
32LIST_HEAD(bdi_list);
33
34/* bdi_wq serves all asynchronous writeback tasks */
35struct workqueue_struct *bdi_wq;
36
37#ifdef CONFIG_DEBUG_FS
38#include <linux/debugfs.h>
39#include <linux/seq_file.h>
40
41static struct dentry *bdi_debug_root;
42
43static void bdi_debug_init(void)
44{
45 bdi_debug_root = debugfs_create_dir("bdi", NULL);
46}
47
48static int bdi_debug_stats_show(struct seq_file *m, void *v)
49{
50 struct backing_dev_info *bdi = m->private;
51 struct bdi_writeback *wb = &bdi->wb;
52 unsigned long background_thresh;
53 unsigned long dirty_thresh;
54 unsigned long wb_thresh;
55 unsigned long nr_dirty, nr_io, nr_more_io, nr_dirty_time;
56 struct inode *inode;
57
58 nr_dirty = nr_io = nr_more_io = nr_dirty_time = 0;
59 spin_lock(&wb->list_lock);
60 list_for_each_entry(inode, &wb->b_dirty, i_io_list)
61 nr_dirty++;
62 list_for_each_entry(inode, &wb->b_io, i_io_list)
63 nr_io++;
64 list_for_each_entry(inode, &wb->b_more_io, i_io_list)
65 nr_more_io++;
66 list_for_each_entry(inode, &wb->b_dirty_time, i_io_list)
67 if (inode->i_state & I_DIRTY_TIME)
68 nr_dirty_time++;
69 spin_unlock(&wb->list_lock);
70
71 global_dirty_limits(&background_thresh, &dirty_thresh);
72 wb_thresh = wb_calc_thresh(wb, dirty_thresh);
73
74#define K(x) ((x) << (PAGE_SHIFT - 10))
75 seq_printf(m,
76 "BdiWriteback: %10lu kB\n"
77 "BdiReclaimable: %10lu kB\n"
78 "BdiDirtyThresh: %10lu kB\n"
79 "DirtyThresh: %10lu kB\n"
80 "BackgroundThresh: %10lu kB\n"
81 "BdiDirtied: %10lu kB\n"
82 "BdiWritten: %10lu kB\n"
83 "BdiWriteBandwidth: %10lu kBps\n"
84 "b_dirty: %10lu\n"
85 "b_io: %10lu\n"
86 "b_more_io: %10lu\n"
87 "b_dirty_time: %10lu\n"
88 "bdi_list: %10u\n"
89 "state: %10lx\n",
90 (unsigned long) K(wb_stat(wb, WB_WRITEBACK)),
91 (unsigned long) K(wb_stat(wb, WB_RECLAIMABLE)),
92 K(wb_thresh),
93 K(dirty_thresh),
94 K(background_thresh),
95 (unsigned long) K(wb_stat(wb, WB_DIRTIED)),
96 (unsigned long) K(wb_stat(wb, WB_WRITTEN)),
97 (unsigned long) K(wb->write_bandwidth),
98 nr_dirty,
99 nr_io,
100 nr_more_io,
101 nr_dirty_time,
102 !list_empty(&bdi->bdi_list), bdi->wb.state);
103#undef K
104
105 return 0;
106}
107DEFINE_SHOW_ATTRIBUTE(bdi_debug_stats);
108
109static void bdi_debug_register(struct backing_dev_info *bdi, const char *name)
110{
111 bdi->debug_dir = debugfs_create_dir(name, bdi_debug_root);
112
113 debugfs_create_file("stats", 0444, bdi->debug_dir, bdi,
114 &bdi_debug_stats_fops);
115}
116
117static void bdi_debug_unregister(struct backing_dev_info *bdi)
118{
119 debugfs_remove_recursive(bdi->debug_dir);
120}
121#else
122static inline void bdi_debug_init(void)
123{
124}
125static inline void bdi_debug_register(struct backing_dev_info *bdi,
126 const char *name)
127{
128}
129static inline void bdi_debug_unregister(struct backing_dev_info *bdi)
130{
131}
132#endif
133
134static ssize_t read_ahead_kb_store(struct device *dev,
135 struct device_attribute *attr,
136 const char *buf, size_t count)
137{
138 struct backing_dev_info *bdi = dev_get_drvdata(dev);
139 unsigned long read_ahead_kb;
140 ssize_t ret;
141
142 ret = kstrtoul(buf, 10, &read_ahead_kb);
143 if (ret < 0)
144 return ret;
145
146 bdi->ra_pages = read_ahead_kb >> (PAGE_SHIFT - 10);
147
148 return count;
149}
150
151#define K(pages) ((pages) << (PAGE_SHIFT - 10))
152
153#define BDI_SHOW(name, expr) \
154static ssize_t name##_show(struct device *dev, \
155 struct device_attribute *attr, char *page) \
156{ \
157 struct backing_dev_info *bdi = dev_get_drvdata(dev); \
158 \
159 return snprintf(page, PAGE_SIZE-1, "%lld\n", (long long)expr); \
160} \
161static DEVICE_ATTR_RW(name);
162
163BDI_SHOW(read_ahead_kb, K(bdi->ra_pages))
164
165static ssize_t min_ratio_store(struct device *dev,
166 struct device_attribute *attr, const char *buf, size_t count)
167{
168 struct backing_dev_info *bdi = dev_get_drvdata(dev);
169 unsigned int ratio;
170 ssize_t ret;
171
172 ret = kstrtouint(buf, 10, &ratio);
173 if (ret < 0)
174 return ret;
175
176 ret = bdi_set_min_ratio(bdi, ratio);
177 if (!ret)
178 ret = count;
179
180 return ret;
181}
182BDI_SHOW(min_ratio, bdi->min_ratio)
183
184static ssize_t max_ratio_store(struct device *dev,
185 struct device_attribute *attr, const char *buf, size_t count)
186{
187 struct backing_dev_info *bdi = dev_get_drvdata(dev);
188 unsigned int ratio;
189 ssize_t ret;
190
191 ret = kstrtouint(buf, 10, &ratio);
192 if (ret < 0)
193 return ret;
194
195 ret = bdi_set_max_ratio(bdi, ratio);
196 if (!ret)
197 ret = count;
198
199 return ret;
200}
201BDI_SHOW(max_ratio, bdi->max_ratio)
202
203static ssize_t stable_pages_required_show(struct device *dev,
204 struct device_attribute *attr,
205 char *page)
206{
207 struct backing_dev_info *bdi = dev_get_drvdata(dev);
208
209 return snprintf(page, PAGE_SIZE-1, "%d\n",
210 bdi_cap_stable_pages_required(bdi) ? 1 : 0);
211}
212static DEVICE_ATTR_RO(stable_pages_required);
213
214static struct attribute *bdi_dev_attrs[] = {
215 &dev_attr_read_ahead_kb.attr,
216 &dev_attr_min_ratio.attr,
217 &dev_attr_max_ratio.attr,
218 &dev_attr_stable_pages_required.attr,
219 NULL,
220};
221ATTRIBUTE_GROUPS(bdi_dev);
222
223static __init int bdi_class_init(void)
224{
225 bdi_class = class_create(THIS_MODULE, "bdi");
226 if (IS_ERR(bdi_class))
227 return PTR_ERR(bdi_class);
228
229 bdi_class->dev_groups = bdi_dev_groups;
230 bdi_debug_init();
231
232 return 0;
233}
234postcore_initcall(bdi_class_init);
235
236static int bdi_init(struct backing_dev_info *bdi);
237
238static int __init default_bdi_init(void)
239{
240 int err;
241
242 bdi_wq = alloc_workqueue("writeback", WQ_MEM_RECLAIM | WQ_UNBOUND |
243 WQ_SYSFS, 0);
244 if (!bdi_wq)
245 return -ENOMEM;
246
247 err = bdi_init(&noop_backing_dev_info);
248
249 return err;
250}
251subsys_initcall(default_bdi_init);
252
253/*
254 * This function is used when the first inode for this wb is marked dirty. It
255 * wakes-up the corresponding bdi thread which should then take care of the
256 * periodic background write-out of dirty inodes. Since the write-out would
257 * starts only 'dirty_writeback_interval' centisecs from now anyway, we just
258 * set up a timer which wakes the bdi thread up later.
259 *
260 * Note, we wouldn't bother setting up the timer, but this function is on the
261 * fast-path (used by '__mark_inode_dirty()'), so we save few context switches
262 * by delaying the wake-up.
263 *
264 * We have to be careful not to postpone flush work if it is scheduled for
265 * earlier. Thus we use queue_delayed_work().
266 */
267void wb_wakeup_delayed(struct bdi_writeback *wb)
268{
269 unsigned long timeout;
270
271 timeout = msecs_to_jiffies(dirty_writeback_interval * 10);
272 spin_lock_bh(&wb->work_lock);
273 if (test_bit(WB_registered, &wb->state))
274 queue_delayed_work(bdi_wq, &wb->dwork, timeout);
275 spin_unlock_bh(&wb->work_lock);
276}
277
278/*
279 * Initial write bandwidth: 100 MB/s
280 */
281#define INIT_BW (100 << (20 - PAGE_SHIFT))
282
283static int wb_init(struct bdi_writeback *wb, struct backing_dev_info *bdi,
284 int blkcg_id, gfp_t gfp)
285{
286 int i, err;
287
288 memset(wb, 0, sizeof(*wb));
289
290 if (wb != &bdi->wb)
291 bdi_get(bdi);
292 wb->bdi = bdi;
293 wb->last_old_flush = jiffies;
294 INIT_LIST_HEAD(&wb->b_dirty);
295 INIT_LIST_HEAD(&wb->b_io);
296 INIT_LIST_HEAD(&wb->b_more_io);
297 INIT_LIST_HEAD(&wb->b_dirty_time);
298 spin_lock_init(&wb->list_lock);
299
300 wb->bw_time_stamp = jiffies;
301 wb->balanced_dirty_ratelimit = INIT_BW;
302 wb->dirty_ratelimit = INIT_BW;
303 wb->write_bandwidth = INIT_BW;
304 wb->avg_write_bandwidth = INIT_BW;
305
306 spin_lock_init(&wb->work_lock);
307 INIT_LIST_HEAD(&wb->work_list);
308 INIT_DELAYED_WORK(&wb->dwork, wb_workfn);
309 wb->dirty_sleep = jiffies;
310
311 wb->congested = wb_congested_get_create(bdi, blkcg_id, gfp);
312 if (!wb->congested) {
313 err = -ENOMEM;
314 goto out_put_bdi;
315 }
316
317 err = fprop_local_init_percpu(&wb->completions, gfp);
318 if (err)
319 goto out_put_cong;
320
321 for (i = 0; i < NR_WB_STAT_ITEMS; i++) {
322 err = percpu_counter_init(&wb->stat[i], 0, gfp);
323 if (err)
324 goto out_destroy_stat;
325 }
326
327 return 0;
328
329out_destroy_stat:
330 while (i--)
331 percpu_counter_destroy(&wb->stat[i]);
332 fprop_local_destroy_percpu(&wb->completions);
333out_put_cong:
334 wb_congested_put(wb->congested);
335out_put_bdi:
336 if (wb != &bdi->wb)
337 bdi_put(bdi);
338 return err;
339}
340
341static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb);
342
343/*
344 * Remove bdi from the global list and shutdown any threads we have running
345 */
346static void wb_shutdown(struct bdi_writeback *wb)
347{
348 /* Make sure nobody queues further work */
349 spin_lock_bh(&wb->work_lock);
350 if (!test_and_clear_bit(WB_registered, &wb->state)) {
351 spin_unlock_bh(&wb->work_lock);
352 return;
353 }
354 spin_unlock_bh(&wb->work_lock);
355
356 cgwb_remove_from_bdi_list(wb);
357 /*
358 * Drain work list and shutdown the delayed_work. !WB_registered
359 * tells wb_workfn() that @wb is dying and its work_list needs to
360 * be drained no matter what.
361 */
362 mod_delayed_work(bdi_wq, &wb->dwork, 0);
363 flush_delayed_work(&wb->dwork);
364 WARN_ON(!list_empty(&wb->work_list));
365}
366
367static void wb_exit(struct bdi_writeback *wb)
368{
369 int i;
370
371 WARN_ON(delayed_work_pending(&wb->dwork));
372
373 for (i = 0; i < NR_WB_STAT_ITEMS; i++)
374 percpu_counter_destroy(&wb->stat[i]);
375
376 fprop_local_destroy_percpu(&wb->completions);
377 wb_congested_put(wb->congested);
378 if (wb != &wb->bdi->wb)
379 bdi_put(wb->bdi);
380}
381
382#ifdef CONFIG_CGROUP_WRITEBACK
383
384#include <linux/memcontrol.h>
385
386/*
387 * cgwb_lock protects bdi->cgwb_tree, bdi->cgwb_congested_tree,
388 * blkcg->cgwb_list, and memcg->cgwb_list. bdi->cgwb_tree is also RCU
389 * protected.
390 */
391static DEFINE_SPINLOCK(cgwb_lock);
392static struct workqueue_struct *cgwb_release_wq;
393
394/**
395 * wb_congested_get_create - get or create a wb_congested
396 * @bdi: associated bdi
397 * @blkcg_id: ID of the associated blkcg
398 * @gfp: allocation mask
399 *
400 * Look up the wb_congested for @blkcg_id on @bdi. If missing, create one.
401 * The returned wb_congested has its reference count incremented. Returns
402 * NULL on failure.
403 */
404struct bdi_writeback_congested *
405wb_congested_get_create(struct backing_dev_info *bdi, int blkcg_id, gfp_t gfp)
406{
407 struct bdi_writeback_congested *new_congested = NULL, *congested;
408 struct rb_node **node, *parent;
409 unsigned long flags;
410retry:
411 spin_lock_irqsave(&cgwb_lock, flags);
412
413 node = &bdi->cgwb_congested_tree.rb_node;
414 parent = NULL;
415
416 while (*node != NULL) {
417 parent = *node;
418 congested = rb_entry(parent, struct bdi_writeback_congested,
419 rb_node);
420 if (congested->blkcg_id < blkcg_id)
421 node = &parent->rb_left;
422 else if (congested->blkcg_id > blkcg_id)
423 node = &parent->rb_right;
424 else
425 goto found;
426 }
427
428 if (new_congested) {
429 /* !found and storage for new one already allocated, insert */
430 congested = new_congested;
431 rb_link_node(&congested->rb_node, parent, node);
432 rb_insert_color(&congested->rb_node, &bdi->cgwb_congested_tree);
433 spin_unlock_irqrestore(&cgwb_lock, flags);
434 return congested;
435 }
436
437 spin_unlock_irqrestore(&cgwb_lock, flags);
438
439 /* allocate storage for new one and retry */
440 new_congested = kzalloc(sizeof(*new_congested), gfp);
441 if (!new_congested)
442 return NULL;
443
444 refcount_set(&new_congested->refcnt, 1);
445 new_congested->__bdi = bdi;
446 new_congested->blkcg_id = blkcg_id;
447 goto retry;
448
449found:
450 refcount_inc(&congested->refcnt);
451 spin_unlock_irqrestore(&cgwb_lock, flags);
452 kfree(new_congested);
453 return congested;
454}
455
456/**
457 * wb_congested_put - put a wb_congested
458 * @congested: wb_congested to put
459 *
460 * Put @congested and destroy it if the refcnt reaches zero.
461 */
462void wb_congested_put(struct bdi_writeback_congested *congested)
463{
464 unsigned long flags;
465
466 if (!refcount_dec_and_lock_irqsave(&congested->refcnt, &cgwb_lock, &flags))
467 return;
468
469 /* bdi might already have been destroyed leaving @congested unlinked */
470 if (congested->__bdi) {
471 rb_erase(&congested->rb_node,
472 &congested->__bdi->cgwb_congested_tree);
473 congested->__bdi = NULL;
474 }
475
476 spin_unlock_irqrestore(&cgwb_lock, flags);
477 kfree(congested);
478}
479
480static void cgwb_release_workfn(struct work_struct *work)
481{
482 struct bdi_writeback *wb = container_of(work, struct bdi_writeback,
483 release_work);
484 struct blkcg *blkcg = css_to_blkcg(wb->blkcg_css);
485
486 mutex_lock(&wb->bdi->cgwb_release_mutex);
487 wb_shutdown(wb);
488
489 css_put(wb->memcg_css);
490 css_put(wb->blkcg_css);
491 mutex_unlock(&wb->bdi->cgwb_release_mutex);
492
493 /* triggers blkg destruction if cgwb_refcnt becomes zero */
494 blkcg_cgwb_put(blkcg);
495
496 fprop_local_destroy_percpu(&wb->memcg_completions);
497 percpu_ref_exit(&wb->refcnt);
498 wb_exit(wb);
499 kfree_rcu(wb, rcu);
500}
501
502static void cgwb_release(struct percpu_ref *refcnt)
503{
504 struct bdi_writeback *wb = container_of(refcnt, struct bdi_writeback,
505 refcnt);
506 queue_work(cgwb_release_wq, &wb->release_work);
507}
508
509static void cgwb_kill(struct bdi_writeback *wb)
510{
511 lockdep_assert_held(&cgwb_lock);
512
513 WARN_ON(!radix_tree_delete(&wb->bdi->cgwb_tree, wb->memcg_css->id));
514 list_del(&wb->memcg_node);
515 list_del(&wb->blkcg_node);
516 percpu_ref_kill(&wb->refcnt);
517}
518
519static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb)
520{
521 spin_lock_irq(&cgwb_lock);
522 list_del_rcu(&wb->bdi_node);
523 spin_unlock_irq(&cgwb_lock);
524}
525
526static int cgwb_create(struct backing_dev_info *bdi,
527 struct cgroup_subsys_state *memcg_css, gfp_t gfp)
528{
529 struct mem_cgroup *memcg;
530 struct cgroup_subsys_state *blkcg_css;
531 struct blkcg *blkcg;
532 struct list_head *memcg_cgwb_list, *blkcg_cgwb_list;
533 struct bdi_writeback *wb;
534 unsigned long flags;
535 int ret = 0;
536
537 memcg = mem_cgroup_from_css(memcg_css);
538 blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
539 blkcg = css_to_blkcg(blkcg_css);
540 memcg_cgwb_list = &memcg->cgwb_list;
541 blkcg_cgwb_list = &blkcg->cgwb_list;
542
543 /* look up again under lock and discard on blkcg mismatch */
544 spin_lock_irqsave(&cgwb_lock, flags);
545 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
546 if (wb && wb->blkcg_css != blkcg_css) {
547 cgwb_kill(wb);
548 wb = NULL;
549 }
550 spin_unlock_irqrestore(&cgwb_lock, flags);
551 if (wb)
552 goto out_put;
553
554 /* need to create a new one */
555 wb = kmalloc(sizeof(*wb), gfp);
556 if (!wb) {
557 ret = -ENOMEM;
558 goto out_put;
559 }
560
561 ret = wb_init(wb, bdi, blkcg_css->id, gfp);
562 if (ret)
563 goto err_free;
564
565 ret = percpu_ref_init(&wb->refcnt, cgwb_release, 0, gfp);
566 if (ret)
567 goto err_wb_exit;
568
569 ret = fprop_local_init_percpu(&wb->memcg_completions, gfp);
570 if (ret)
571 goto err_ref_exit;
572
573 wb->memcg_css = memcg_css;
574 wb->blkcg_css = blkcg_css;
575 INIT_WORK(&wb->release_work, cgwb_release_workfn);
576 set_bit(WB_registered, &wb->state);
577
578 /*
579 * The root wb determines the registered state of the whole bdi and
580 * memcg_cgwb_list and blkcg_cgwb_list's next pointers indicate
581 * whether they're still online. Don't link @wb if any is dead.
582 * See wb_memcg_offline() and wb_blkcg_offline().
583 */
584 ret = -ENODEV;
585 spin_lock_irqsave(&cgwb_lock, flags);
586 if (test_bit(WB_registered, &bdi->wb.state) &&
587 blkcg_cgwb_list->next && memcg_cgwb_list->next) {
588 /* we might have raced another instance of this function */
589 ret = radix_tree_insert(&bdi->cgwb_tree, memcg_css->id, wb);
590 if (!ret) {
591 list_add_tail_rcu(&wb->bdi_node, &bdi->wb_list);
592 list_add(&wb->memcg_node, memcg_cgwb_list);
593 list_add(&wb->blkcg_node, blkcg_cgwb_list);
594 blkcg_cgwb_get(blkcg);
595 css_get(memcg_css);
596 css_get(blkcg_css);
597 }
598 }
599 spin_unlock_irqrestore(&cgwb_lock, flags);
600 if (ret) {
601 if (ret == -EEXIST)
602 ret = 0;
603 goto err_fprop_exit;
604 }
605 goto out_put;
606
607err_fprop_exit:
608 fprop_local_destroy_percpu(&wb->memcg_completions);
609err_ref_exit:
610 percpu_ref_exit(&wb->refcnt);
611err_wb_exit:
612 wb_exit(wb);
613err_free:
614 kfree(wb);
615out_put:
616 css_put(blkcg_css);
617 return ret;
618}
619
620/**
621 * wb_get_lookup - get wb for a given memcg
622 * @bdi: target bdi
623 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
624 *
625 * Try to get the wb for @memcg_css on @bdi. The returned wb has its
626 * refcount incremented.
627 *
628 * This function uses css_get() on @memcg_css and thus expects its refcnt
629 * to be positive on invocation. IOW, rcu_read_lock() protection on
630 * @memcg_css isn't enough. try_get it before calling this function.
631 *
632 * A wb is keyed by its associated memcg. As blkcg implicitly enables
633 * memcg on the default hierarchy, memcg association is guaranteed to be
634 * more specific (equal or descendant to the associated blkcg) and thus can
635 * identify both the memcg and blkcg associations.
636 *
637 * Because the blkcg associated with a memcg may change as blkcg is enabled
638 * and disabled closer to root in the hierarchy, each wb keeps track of
639 * both the memcg and blkcg associated with it and verifies the blkcg on
640 * each lookup. On mismatch, the existing wb is discarded and a new one is
641 * created.
642 */
643struct bdi_writeback *wb_get_lookup(struct backing_dev_info *bdi,
644 struct cgroup_subsys_state *memcg_css)
645{
646 struct bdi_writeback *wb;
647
648 if (!memcg_css->parent)
649 return &bdi->wb;
650
651 rcu_read_lock();
652 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
653 if (wb) {
654 struct cgroup_subsys_state *blkcg_css;
655
656 /* see whether the blkcg association has changed */
657 blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
658 if (unlikely(wb->blkcg_css != blkcg_css || !wb_tryget(wb)))
659 wb = NULL;
660 css_put(blkcg_css);
661 }
662 rcu_read_unlock();
663
664 return wb;
665}
666
667/**
668 * wb_get_create - get wb for a given memcg, create if necessary
669 * @bdi: target bdi
670 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
671 * @gfp: allocation mask to use
672 *
673 * Try to get the wb for @memcg_css on @bdi. If it doesn't exist, try to
674 * create one. See wb_get_lookup() for more details.
675 */
676struct bdi_writeback *wb_get_create(struct backing_dev_info *bdi,
677 struct cgroup_subsys_state *memcg_css,
678 gfp_t gfp)
679{
680 struct bdi_writeback *wb;
681
682 might_sleep_if(gfpflags_allow_blocking(gfp));
683
684 if (!memcg_css->parent)
685 return &bdi->wb;
686
687 do {
688 wb = wb_get_lookup(bdi, memcg_css);
689 } while (!wb && !cgwb_create(bdi, memcg_css, gfp));
690
691 return wb;
692}
693
694static int cgwb_bdi_init(struct backing_dev_info *bdi)
695{
696 int ret;
697
698 INIT_RADIX_TREE(&bdi->cgwb_tree, GFP_ATOMIC);
699 bdi->cgwb_congested_tree = RB_ROOT;
700 mutex_init(&bdi->cgwb_release_mutex);
701 init_rwsem(&bdi->wb_switch_rwsem);
702
703 ret = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
704 if (!ret) {
705 bdi->wb.memcg_css = &root_mem_cgroup->css;
706 bdi->wb.blkcg_css = blkcg_root_css;
707 }
708 return ret;
709}
710
711static void cgwb_bdi_unregister(struct backing_dev_info *bdi)
712{
713 struct radix_tree_iter iter;
714 void **slot;
715 struct bdi_writeback *wb;
716
717 WARN_ON(test_bit(WB_registered, &bdi->wb.state));
718
719 spin_lock_irq(&cgwb_lock);
720 radix_tree_for_each_slot(slot, &bdi->cgwb_tree, &iter, 0)
721 cgwb_kill(*slot);
722 spin_unlock_irq(&cgwb_lock);
723
724 mutex_lock(&bdi->cgwb_release_mutex);
725 spin_lock_irq(&cgwb_lock);
726 while (!list_empty(&bdi->wb_list)) {
727 wb = list_first_entry(&bdi->wb_list, struct bdi_writeback,
728 bdi_node);
729 spin_unlock_irq(&cgwb_lock);
730 wb_shutdown(wb);
731 spin_lock_irq(&cgwb_lock);
732 }
733 spin_unlock_irq(&cgwb_lock);
734 mutex_unlock(&bdi->cgwb_release_mutex);
735}
736
737/**
738 * wb_memcg_offline - kill all wb's associated with a memcg being offlined
739 * @memcg: memcg being offlined
740 *
741 * Also prevents creation of any new wb's associated with @memcg.
742 */
743void wb_memcg_offline(struct mem_cgroup *memcg)
744{
745 struct list_head *memcg_cgwb_list = &memcg->cgwb_list;
746 struct bdi_writeback *wb, *next;
747
748 spin_lock_irq(&cgwb_lock);
749 list_for_each_entry_safe(wb, next, memcg_cgwb_list, memcg_node)
750 cgwb_kill(wb);
751 memcg_cgwb_list->next = NULL; /* prevent new wb's */
752 spin_unlock_irq(&cgwb_lock);
753}
754
755/**
756 * wb_blkcg_offline - kill all wb's associated with a blkcg being offlined
757 * @blkcg: blkcg being offlined
758 *
759 * Also prevents creation of any new wb's associated with @blkcg.
760 */
761void wb_blkcg_offline(struct blkcg *blkcg)
762{
763 struct bdi_writeback *wb, *next;
764
765 spin_lock_irq(&cgwb_lock);
766 list_for_each_entry_safe(wb, next, &blkcg->cgwb_list, blkcg_node)
767 cgwb_kill(wb);
768 blkcg->cgwb_list.next = NULL; /* prevent new wb's */
769 spin_unlock_irq(&cgwb_lock);
770}
771
772static void cgwb_bdi_exit(struct backing_dev_info *bdi)
773{
774 struct rb_node *rbn;
775
776 spin_lock_irq(&cgwb_lock);
777 while ((rbn = rb_first(&bdi->cgwb_congested_tree))) {
778 struct bdi_writeback_congested *congested =
779 rb_entry(rbn, struct bdi_writeback_congested, rb_node);
780
781 rb_erase(rbn, &bdi->cgwb_congested_tree);
782 congested->__bdi = NULL; /* mark @congested unlinked */
783 }
784 spin_unlock_irq(&cgwb_lock);
785}
786
787static void cgwb_bdi_register(struct backing_dev_info *bdi)
788{
789 spin_lock_irq(&cgwb_lock);
790 list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);
791 spin_unlock_irq(&cgwb_lock);
792}
793
794static int __init cgwb_init(void)
795{
796 /*
797 * There can be many concurrent release work items overwhelming
798 * system_wq. Put them in a separate wq and limit concurrency.
799 * There's no point in executing many of these in parallel.
800 */
801 cgwb_release_wq = alloc_workqueue("cgwb_release", 0, 1);
802 if (!cgwb_release_wq)
803 return -ENOMEM;
804
805 return 0;
806}
807subsys_initcall(cgwb_init);
808
809#else /* CONFIG_CGROUP_WRITEBACK */
810
811static int cgwb_bdi_init(struct backing_dev_info *bdi)
812{
813 int err;
814
815 bdi->wb_congested = kzalloc(sizeof(*bdi->wb_congested), GFP_KERNEL);
816 if (!bdi->wb_congested)
817 return -ENOMEM;
818
819 refcount_set(&bdi->wb_congested->refcnt, 1);
820
821 err = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
822 if (err) {
823 wb_congested_put(bdi->wb_congested);
824 return err;
825 }
826 return 0;
827}
828
829static void cgwb_bdi_unregister(struct backing_dev_info *bdi) { }
830
831static void cgwb_bdi_exit(struct backing_dev_info *bdi)
832{
833 wb_congested_put(bdi->wb_congested);
834}
835
836static void cgwb_bdi_register(struct backing_dev_info *bdi)
837{
838 list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);
839}
840
841static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb)
842{
843 list_del_rcu(&wb->bdi_node);
844}
845
846#endif /* CONFIG_CGROUP_WRITEBACK */
847
848static int bdi_init(struct backing_dev_info *bdi)
849{
850 int ret;
851
852 bdi->dev = NULL;
853
854 kref_init(&bdi->refcnt);
855 bdi->min_ratio = 0;
856 bdi->max_ratio = 100;
857 bdi->max_prop_frac = FPROP_FRAC_BASE;
858 INIT_LIST_HEAD(&bdi->bdi_list);
859 INIT_LIST_HEAD(&bdi->wb_list);
860 init_waitqueue_head(&bdi->wb_waitq);
861
862 ret = cgwb_bdi_init(bdi);
863
864 return ret;
865}
866
867struct backing_dev_info *bdi_alloc_node(gfp_t gfp_mask, int node_id)
868{
869 struct backing_dev_info *bdi;
870
871 bdi = kmalloc_node(sizeof(struct backing_dev_info),
872 gfp_mask | __GFP_ZERO, node_id);
873 if (!bdi)
874 return NULL;
875
876 if (bdi_init(bdi)) {
877 kfree(bdi);
878 return NULL;
879 }
880 return bdi;
881}
882EXPORT_SYMBOL(bdi_alloc_node);
883
884static struct rb_node **bdi_lookup_rb_node(u64 id, struct rb_node **parentp)
885{
886 struct rb_node **p = &bdi_tree.rb_node;
887 struct rb_node *parent = NULL;
888 struct backing_dev_info *bdi;
889
890 lockdep_assert_held(&bdi_lock);
891
892 while (*p) {
893 parent = *p;
894 bdi = rb_entry(parent, struct backing_dev_info, rb_node);
895
896 if (bdi->id > id)
897 p = &(*p)->rb_left;
898 else if (bdi->id < id)
899 p = &(*p)->rb_right;
900 else
901 break;
902 }
903
904 if (parentp)
905 *parentp = parent;
906 return p;
907}
908
909/**
910 * bdi_get_by_id - lookup and get bdi from its id
911 * @id: bdi id to lookup
912 *
913 * Find bdi matching @id and get it. Returns NULL if the matching bdi
914 * doesn't exist or is already unregistered.
915 */
916struct backing_dev_info *bdi_get_by_id(u64 id)
917{
918 struct backing_dev_info *bdi = NULL;
919 struct rb_node **p;
920
921 spin_lock_bh(&bdi_lock);
922 p = bdi_lookup_rb_node(id, NULL);
923 if (*p) {
924 bdi = rb_entry(*p, struct backing_dev_info, rb_node);
925 bdi_get(bdi);
926 }
927 spin_unlock_bh(&bdi_lock);
928
929 return bdi;
930}
931
932int bdi_register_va(struct backing_dev_info *bdi, const char *fmt, va_list args)
933{
934 struct device *dev;
935 struct rb_node *parent, **p;
936
937 if (bdi->dev) /* The driver needs to use separate queues per device */
938 return 0;
939
940 dev = device_create_vargs(bdi_class, NULL, MKDEV(0, 0), bdi, fmt, args);
941 if (IS_ERR(dev))
942 return PTR_ERR(dev);
943
944 cgwb_bdi_register(bdi);
945 bdi->dev = dev;
946
947 bdi_debug_register(bdi, dev_name(dev));
948 set_bit(WB_registered, &bdi->wb.state);
949
950 spin_lock_bh(&bdi_lock);
951
952 bdi->id = ++bdi_id_cursor;
953
954 p = bdi_lookup_rb_node(bdi->id, &parent);
955 rb_link_node(&bdi->rb_node, parent, p);
956 rb_insert_color(&bdi->rb_node, &bdi_tree);
957
958 list_add_tail_rcu(&bdi->bdi_list, &bdi_list);
959
960 spin_unlock_bh(&bdi_lock);
961
962 trace_writeback_bdi_register(bdi);
963 return 0;
964}
965EXPORT_SYMBOL(bdi_register_va);
966
967int bdi_register(struct backing_dev_info *bdi, const char *fmt, ...)
968{
969 va_list args;
970 int ret;
971
972 va_start(args, fmt);
973 ret = bdi_register_va(bdi, fmt, args);
974 va_end(args);
975 return ret;
976}
977EXPORT_SYMBOL(bdi_register);
978
979int bdi_register_owner(struct backing_dev_info *bdi, struct device *owner)
980{
981 int rc;
982
983 rc = bdi_register(bdi, "%u:%u", MAJOR(owner->devt), MINOR(owner->devt));
984 if (rc)
985 return rc;
986 /* Leaking owner reference... */
987 WARN_ON(bdi->owner);
988 bdi->owner = owner;
989 get_device(owner);
990 return 0;
991}
992EXPORT_SYMBOL(bdi_register_owner);
993
994/*
995 * Remove bdi from bdi_list, and ensure that it is no longer visible
996 */
997static void bdi_remove_from_list(struct backing_dev_info *bdi)
998{
999 spin_lock_bh(&bdi_lock);
1000 rb_erase(&bdi->rb_node, &bdi_tree);
1001 list_del_rcu(&bdi->bdi_list);
1002 spin_unlock_bh(&bdi_lock);
1003
1004 synchronize_rcu_expedited();
1005}
1006
1007void bdi_unregister(struct backing_dev_info *bdi)
1008{
1009 /* make sure nobody finds us on the bdi_list anymore */
1010 bdi_remove_from_list(bdi);
1011 wb_shutdown(&bdi->wb);
1012 cgwb_bdi_unregister(bdi);
1013
1014 if (bdi->dev) {
1015 bdi_debug_unregister(bdi);
1016 device_unregister(bdi->dev);
1017 bdi->dev = NULL;
1018 }
1019
1020 if (bdi->owner) {
1021 put_device(bdi->owner);
1022 bdi->owner = NULL;
1023 }
1024}
1025
1026static void release_bdi(struct kref *ref)
1027{
1028 struct backing_dev_info *bdi =
1029 container_of(ref, struct backing_dev_info, refcnt);
1030
1031 if (test_bit(WB_registered, &bdi->wb.state))
1032 bdi_unregister(bdi);
1033 WARN_ON_ONCE(bdi->dev);
1034 wb_exit(&bdi->wb);
1035 cgwb_bdi_exit(bdi);
1036 kfree(bdi);
1037}
1038
1039void bdi_put(struct backing_dev_info *bdi)
1040{
1041 kref_put(&bdi->refcnt, release_bdi);
1042}
1043EXPORT_SYMBOL(bdi_put);
1044
1045static wait_queue_head_t congestion_wqh[2] = {
1046 __WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[0]),
1047 __WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[1])
1048 };
1049static atomic_t nr_wb_congested[2];
1050
1051void clear_wb_congested(struct bdi_writeback_congested *congested, int sync)
1052{
1053 wait_queue_head_t *wqh = &congestion_wqh[sync];
1054 enum wb_congested_state bit;
1055
1056 bit = sync ? WB_sync_congested : WB_async_congested;
1057 if (test_and_clear_bit(bit, &congested->state))
1058 atomic_dec(&nr_wb_congested[sync]);
1059 smp_mb__after_atomic();
1060 if (waitqueue_active(wqh))
1061 wake_up(wqh);
1062}
1063EXPORT_SYMBOL(clear_wb_congested);
1064
1065void set_wb_congested(struct bdi_writeback_congested *congested, int sync)
1066{
1067 enum wb_congested_state bit;
1068
1069 bit = sync ? WB_sync_congested : WB_async_congested;
1070 if (!test_and_set_bit(bit, &congested->state))
1071 atomic_inc(&nr_wb_congested[sync]);
1072}
1073EXPORT_SYMBOL(set_wb_congested);
1074
1075/**
1076 * congestion_wait - wait for a backing_dev to become uncongested
1077 * @sync: SYNC or ASYNC IO
1078 * @timeout: timeout in jiffies
1079 *
1080 * Waits for up to @timeout jiffies for a backing_dev (any backing_dev) to exit
1081 * write congestion. If no backing_devs are congested then just wait for the
1082 * next write to be completed.
1083 */
1084long congestion_wait(int sync, long timeout)
1085{
1086 long ret;
1087 unsigned long start = jiffies;
1088 DEFINE_WAIT(wait);
1089 wait_queue_head_t *wqh = &congestion_wqh[sync];
1090
1091 prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
1092 ret = io_schedule_timeout(timeout);
1093 finish_wait(wqh, &wait);
1094
1095 trace_writeback_congestion_wait(jiffies_to_usecs(timeout),
1096 jiffies_to_usecs(jiffies - start));
1097
1098 return ret;
1099}
1100EXPORT_SYMBOL(congestion_wait);
1101
1102/**
1103 * wait_iff_congested - Conditionally wait for a backing_dev to become uncongested or a pgdat to complete writes
1104 * @sync: SYNC or ASYNC IO
1105 * @timeout: timeout in jiffies
1106 *
1107 * In the event of a congested backing_dev (any backing_dev) this waits
1108 * for up to @timeout jiffies for either a BDI to exit congestion of the
1109 * given @sync queue or a write to complete.
1110 *
1111 * The return value is 0 if the sleep is for the full timeout. Otherwise,
1112 * it is the number of jiffies that were still remaining when the function
1113 * returned. return_value == timeout implies the function did not sleep.
1114 */
1115long wait_iff_congested(int sync, long timeout)
1116{
1117 long ret;
1118 unsigned long start = jiffies;
1119 DEFINE_WAIT(wait);
1120 wait_queue_head_t *wqh = &congestion_wqh[sync];
1121
1122 /*
1123 * If there is no congestion, yield if necessary instead
1124 * of sleeping on the congestion queue
1125 */
1126 if (atomic_read(&nr_wb_congested[sync]) == 0) {
1127 cond_resched();
1128
1129 /* In case we scheduled, work out time remaining */
1130 ret = timeout - (jiffies - start);
1131 if (ret < 0)
1132 ret = 0;
1133
1134 goto out;
1135 }
1136
1137 /* Sleep until uncongested or a write happens */
1138 prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
1139 ret = io_schedule_timeout(timeout);
1140 finish_wait(wqh, &wait);
1141
1142out:
1143 trace_writeback_wait_iff_congested(jiffies_to_usecs(timeout),
1144 jiffies_to_usecs(jiffies - start));
1145
1146 return ret;
1147}
1148EXPORT_SYMBOL(wait_iff_congested);
1// SPDX-License-Identifier: GPL-2.0-only
2
3#include <linux/blkdev.h>
4#include <linux/wait.h>
5#include <linux/rbtree.h>
6#include <linux/kthread.h>
7#include <linux/backing-dev.h>
8#include <linux/blk-cgroup.h>
9#include <linux/freezer.h>
10#include <linux/fs.h>
11#include <linux/pagemap.h>
12#include <linux/mm.h>
13#include <linux/sched/mm.h>
14#include <linux/sched.h>
15#include <linux/module.h>
16#include <linux/writeback.h>
17#include <linux/device.h>
18#include <trace/events/writeback.h>
19#include "internal.h"
20
21struct backing_dev_info noop_backing_dev_info;
22EXPORT_SYMBOL_GPL(noop_backing_dev_info);
23
24static const char *bdi_unknown_name = "(unknown)";
25
26/*
27 * bdi_lock protects bdi_tree and updates to bdi_list. bdi_list has RCU
28 * reader side locking.
29 */
30DEFINE_SPINLOCK(bdi_lock);
31static u64 bdi_id_cursor;
32static struct rb_root bdi_tree = RB_ROOT;
33LIST_HEAD(bdi_list);
34
35/* bdi_wq serves all asynchronous writeback tasks */
36struct workqueue_struct *bdi_wq;
37
38#ifdef CONFIG_DEBUG_FS
39#include <linux/debugfs.h>
40#include <linux/seq_file.h>
41
42static struct dentry *bdi_debug_root;
43
44static void bdi_debug_init(void)
45{
46 bdi_debug_root = debugfs_create_dir("bdi", NULL);
47}
48
49static int bdi_debug_stats_show(struct seq_file *m, void *v)
50{
51 struct backing_dev_info *bdi = m->private;
52 struct bdi_writeback *wb = &bdi->wb;
53 unsigned long background_thresh;
54 unsigned long dirty_thresh;
55 unsigned long wb_thresh;
56 unsigned long nr_dirty, nr_io, nr_more_io, nr_dirty_time;
57 struct inode *inode;
58
59 nr_dirty = nr_io = nr_more_io = nr_dirty_time = 0;
60 spin_lock(&wb->list_lock);
61 list_for_each_entry(inode, &wb->b_dirty, i_io_list)
62 nr_dirty++;
63 list_for_each_entry(inode, &wb->b_io, i_io_list)
64 nr_io++;
65 list_for_each_entry(inode, &wb->b_more_io, i_io_list)
66 nr_more_io++;
67 list_for_each_entry(inode, &wb->b_dirty_time, i_io_list)
68 if (inode->i_state & I_DIRTY_TIME)
69 nr_dirty_time++;
70 spin_unlock(&wb->list_lock);
71
72 global_dirty_limits(&background_thresh, &dirty_thresh);
73 wb_thresh = wb_calc_thresh(wb, dirty_thresh);
74
75 seq_printf(m,
76 "BdiWriteback: %10lu kB\n"
77 "BdiReclaimable: %10lu kB\n"
78 "BdiDirtyThresh: %10lu kB\n"
79 "DirtyThresh: %10lu kB\n"
80 "BackgroundThresh: %10lu kB\n"
81 "BdiDirtied: %10lu kB\n"
82 "BdiWritten: %10lu kB\n"
83 "BdiWriteBandwidth: %10lu kBps\n"
84 "b_dirty: %10lu\n"
85 "b_io: %10lu\n"
86 "b_more_io: %10lu\n"
87 "b_dirty_time: %10lu\n"
88 "bdi_list: %10u\n"
89 "state: %10lx\n",
90 (unsigned long) K(wb_stat(wb, WB_WRITEBACK)),
91 (unsigned long) K(wb_stat(wb, WB_RECLAIMABLE)),
92 K(wb_thresh),
93 K(dirty_thresh),
94 K(background_thresh),
95 (unsigned long) K(wb_stat(wb, WB_DIRTIED)),
96 (unsigned long) K(wb_stat(wb, WB_WRITTEN)),
97 (unsigned long) K(wb->write_bandwidth),
98 nr_dirty,
99 nr_io,
100 nr_more_io,
101 nr_dirty_time,
102 !list_empty(&bdi->bdi_list), bdi->wb.state);
103
104 return 0;
105}
106DEFINE_SHOW_ATTRIBUTE(bdi_debug_stats);
107
108static void bdi_debug_register(struct backing_dev_info *bdi, const char *name)
109{
110 bdi->debug_dir = debugfs_create_dir(name, bdi_debug_root);
111
112 debugfs_create_file("stats", 0444, bdi->debug_dir, bdi,
113 &bdi_debug_stats_fops);
114}
115
116static void bdi_debug_unregister(struct backing_dev_info *bdi)
117{
118 debugfs_remove_recursive(bdi->debug_dir);
119}
120#else
121static inline void bdi_debug_init(void)
122{
123}
124static inline void bdi_debug_register(struct backing_dev_info *bdi,
125 const char *name)
126{
127}
128static inline void bdi_debug_unregister(struct backing_dev_info *bdi)
129{
130}
131#endif
132
133static ssize_t read_ahead_kb_store(struct device *dev,
134 struct device_attribute *attr,
135 const char *buf, size_t count)
136{
137 struct backing_dev_info *bdi = dev_get_drvdata(dev);
138 unsigned long read_ahead_kb;
139 ssize_t ret;
140
141 ret = kstrtoul(buf, 10, &read_ahead_kb);
142 if (ret < 0)
143 return ret;
144
145 bdi->ra_pages = read_ahead_kb >> (PAGE_SHIFT - 10);
146
147 return count;
148}
149
150#define BDI_SHOW(name, expr) \
151static ssize_t name##_show(struct device *dev, \
152 struct device_attribute *attr, char *buf) \
153{ \
154 struct backing_dev_info *bdi = dev_get_drvdata(dev); \
155 \
156 return sysfs_emit(buf, "%lld\n", (long long)expr); \
157} \
158static DEVICE_ATTR_RW(name);
159
160BDI_SHOW(read_ahead_kb, K(bdi->ra_pages))
161
162static ssize_t min_ratio_store(struct device *dev,
163 struct device_attribute *attr, const char *buf, size_t count)
164{
165 struct backing_dev_info *bdi = dev_get_drvdata(dev);
166 unsigned int ratio;
167 ssize_t ret;
168
169 ret = kstrtouint(buf, 10, &ratio);
170 if (ret < 0)
171 return ret;
172
173 ret = bdi_set_min_ratio(bdi, ratio);
174 if (!ret)
175 ret = count;
176
177 return ret;
178}
179BDI_SHOW(min_ratio, bdi->min_ratio / BDI_RATIO_SCALE)
180
181static ssize_t min_ratio_fine_store(struct device *dev,
182 struct device_attribute *attr, const char *buf, size_t count)
183{
184 struct backing_dev_info *bdi = dev_get_drvdata(dev);
185 unsigned int ratio;
186 ssize_t ret;
187
188 ret = kstrtouint(buf, 10, &ratio);
189 if (ret < 0)
190 return ret;
191
192 ret = bdi_set_min_ratio_no_scale(bdi, ratio);
193 if (!ret)
194 ret = count;
195
196 return ret;
197}
198BDI_SHOW(min_ratio_fine, bdi->min_ratio)
199
200static ssize_t max_ratio_store(struct device *dev,
201 struct device_attribute *attr, const char *buf, size_t count)
202{
203 struct backing_dev_info *bdi = dev_get_drvdata(dev);
204 unsigned int ratio;
205 ssize_t ret;
206
207 ret = kstrtouint(buf, 10, &ratio);
208 if (ret < 0)
209 return ret;
210
211 ret = bdi_set_max_ratio(bdi, ratio);
212 if (!ret)
213 ret = count;
214
215 return ret;
216}
217BDI_SHOW(max_ratio, bdi->max_ratio / BDI_RATIO_SCALE)
218
219static ssize_t max_ratio_fine_store(struct device *dev,
220 struct device_attribute *attr, const char *buf, size_t count)
221{
222 struct backing_dev_info *bdi = dev_get_drvdata(dev);
223 unsigned int ratio;
224 ssize_t ret;
225
226 ret = kstrtouint(buf, 10, &ratio);
227 if (ret < 0)
228 return ret;
229
230 ret = bdi_set_max_ratio_no_scale(bdi, ratio);
231 if (!ret)
232 ret = count;
233
234 return ret;
235}
236BDI_SHOW(max_ratio_fine, bdi->max_ratio)
237
238static ssize_t min_bytes_show(struct device *dev,
239 struct device_attribute *attr,
240 char *buf)
241{
242 struct backing_dev_info *bdi = dev_get_drvdata(dev);
243
244 return sysfs_emit(buf, "%llu\n", bdi_get_min_bytes(bdi));
245}
246
247static ssize_t min_bytes_store(struct device *dev,
248 struct device_attribute *attr, const char *buf, size_t count)
249{
250 struct backing_dev_info *bdi = dev_get_drvdata(dev);
251 u64 bytes;
252 ssize_t ret;
253
254 ret = kstrtoull(buf, 10, &bytes);
255 if (ret < 0)
256 return ret;
257
258 ret = bdi_set_min_bytes(bdi, bytes);
259 if (!ret)
260 ret = count;
261
262 return ret;
263}
264static DEVICE_ATTR_RW(min_bytes);
265
266static ssize_t max_bytes_show(struct device *dev,
267 struct device_attribute *attr,
268 char *buf)
269{
270 struct backing_dev_info *bdi = dev_get_drvdata(dev);
271
272 return sysfs_emit(buf, "%llu\n", bdi_get_max_bytes(bdi));
273}
274
275static ssize_t max_bytes_store(struct device *dev,
276 struct device_attribute *attr, const char *buf, size_t count)
277{
278 struct backing_dev_info *bdi = dev_get_drvdata(dev);
279 u64 bytes;
280 ssize_t ret;
281
282 ret = kstrtoull(buf, 10, &bytes);
283 if (ret < 0)
284 return ret;
285
286 ret = bdi_set_max_bytes(bdi, bytes);
287 if (!ret)
288 ret = count;
289
290 return ret;
291}
292static DEVICE_ATTR_RW(max_bytes);
293
294static ssize_t stable_pages_required_show(struct device *dev,
295 struct device_attribute *attr,
296 char *buf)
297{
298 dev_warn_once(dev,
299 "the stable_pages_required attribute has been removed. Use the stable_writes queue attribute instead.\n");
300 return sysfs_emit(buf, "%d\n", 0);
301}
302static DEVICE_ATTR_RO(stable_pages_required);
303
304static ssize_t strict_limit_store(struct device *dev,
305 struct device_attribute *attr, const char *buf, size_t count)
306{
307 struct backing_dev_info *bdi = dev_get_drvdata(dev);
308 unsigned int strict_limit;
309 ssize_t ret;
310
311 ret = kstrtouint(buf, 10, &strict_limit);
312 if (ret < 0)
313 return ret;
314
315 ret = bdi_set_strict_limit(bdi, strict_limit);
316 if (!ret)
317 ret = count;
318
319 return ret;
320}
321
322static ssize_t strict_limit_show(struct device *dev,
323 struct device_attribute *attr, char *buf)
324{
325 struct backing_dev_info *bdi = dev_get_drvdata(dev);
326
327 return sysfs_emit(buf, "%d\n",
328 !!(bdi->capabilities & BDI_CAP_STRICTLIMIT));
329}
330static DEVICE_ATTR_RW(strict_limit);
331
332static struct attribute *bdi_dev_attrs[] = {
333 &dev_attr_read_ahead_kb.attr,
334 &dev_attr_min_ratio.attr,
335 &dev_attr_min_ratio_fine.attr,
336 &dev_attr_max_ratio.attr,
337 &dev_attr_max_ratio_fine.attr,
338 &dev_attr_min_bytes.attr,
339 &dev_attr_max_bytes.attr,
340 &dev_attr_stable_pages_required.attr,
341 &dev_attr_strict_limit.attr,
342 NULL,
343};
344ATTRIBUTE_GROUPS(bdi_dev);
345
346static const struct class bdi_class = {
347 .name = "bdi",
348 .dev_groups = bdi_dev_groups,
349};
350
351static __init int bdi_class_init(void)
352{
353 int ret;
354
355 ret = class_register(&bdi_class);
356 if (ret)
357 return ret;
358
359 bdi_debug_init();
360
361 return 0;
362}
363postcore_initcall(bdi_class_init);
364
365static int __init default_bdi_init(void)
366{
367 bdi_wq = alloc_workqueue("writeback", WQ_MEM_RECLAIM | WQ_UNBOUND |
368 WQ_SYSFS, 0);
369 if (!bdi_wq)
370 return -ENOMEM;
371 return 0;
372}
373subsys_initcall(default_bdi_init);
374
375/*
376 * This function is used when the first inode for this wb is marked dirty. It
377 * wakes-up the corresponding bdi thread which should then take care of the
378 * periodic background write-out of dirty inodes. Since the write-out would
379 * starts only 'dirty_writeback_interval' centisecs from now anyway, we just
380 * set up a timer which wakes the bdi thread up later.
381 *
382 * Note, we wouldn't bother setting up the timer, but this function is on the
383 * fast-path (used by '__mark_inode_dirty()'), so we save few context switches
384 * by delaying the wake-up.
385 *
386 * We have to be careful not to postpone flush work if it is scheduled for
387 * earlier. Thus we use queue_delayed_work().
388 */
389void wb_wakeup_delayed(struct bdi_writeback *wb)
390{
391 unsigned long timeout;
392
393 timeout = msecs_to_jiffies(dirty_writeback_interval * 10);
394 spin_lock_irq(&wb->work_lock);
395 if (test_bit(WB_registered, &wb->state))
396 queue_delayed_work(bdi_wq, &wb->dwork, timeout);
397 spin_unlock_irq(&wb->work_lock);
398}
399
400static void wb_update_bandwidth_workfn(struct work_struct *work)
401{
402 struct bdi_writeback *wb = container_of(to_delayed_work(work),
403 struct bdi_writeback, bw_dwork);
404
405 wb_update_bandwidth(wb);
406}
407
408/*
409 * Initial write bandwidth: 100 MB/s
410 */
411#define INIT_BW (100 << (20 - PAGE_SHIFT))
412
413static int wb_init(struct bdi_writeback *wb, struct backing_dev_info *bdi,
414 gfp_t gfp)
415{
416 int i, err;
417
418 memset(wb, 0, sizeof(*wb));
419
420 wb->bdi = bdi;
421 wb->last_old_flush = jiffies;
422 INIT_LIST_HEAD(&wb->b_dirty);
423 INIT_LIST_HEAD(&wb->b_io);
424 INIT_LIST_HEAD(&wb->b_more_io);
425 INIT_LIST_HEAD(&wb->b_dirty_time);
426 spin_lock_init(&wb->list_lock);
427
428 atomic_set(&wb->writeback_inodes, 0);
429 wb->bw_time_stamp = jiffies;
430 wb->balanced_dirty_ratelimit = INIT_BW;
431 wb->dirty_ratelimit = INIT_BW;
432 wb->write_bandwidth = INIT_BW;
433 wb->avg_write_bandwidth = INIT_BW;
434
435 spin_lock_init(&wb->work_lock);
436 INIT_LIST_HEAD(&wb->work_list);
437 INIT_DELAYED_WORK(&wb->dwork, wb_workfn);
438 INIT_DELAYED_WORK(&wb->bw_dwork, wb_update_bandwidth_workfn);
439
440 err = fprop_local_init_percpu(&wb->completions, gfp);
441 if (err)
442 return err;
443
444 for (i = 0; i < NR_WB_STAT_ITEMS; i++) {
445 err = percpu_counter_init(&wb->stat[i], 0, gfp);
446 if (err)
447 goto out_destroy_stat;
448 }
449
450 return 0;
451
452out_destroy_stat:
453 while (i--)
454 percpu_counter_destroy(&wb->stat[i]);
455 fprop_local_destroy_percpu(&wb->completions);
456 return err;
457}
458
459static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb);
460
461/*
462 * Remove bdi from the global list and shutdown any threads we have running
463 */
464static void wb_shutdown(struct bdi_writeback *wb)
465{
466 /* Make sure nobody queues further work */
467 spin_lock_irq(&wb->work_lock);
468 if (!test_and_clear_bit(WB_registered, &wb->state)) {
469 spin_unlock_irq(&wb->work_lock);
470 return;
471 }
472 spin_unlock_irq(&wb->work_lock);
473
474 cgwb_remove_from_bdi_list(wb);
475 /*
476 * Drain work list and shutdown the delayed_work. !WB_registered
477 * tells wb_workfn() that @wb is dying and its work_list needs to
478 * be drained no matter what.
479 */
480 mod_delayed_work(bdi_wq, &wb->dwork, 0);
481 flush_delayed_work(&wb->dwork);
482 WARN_ON(!list_empty(&wb->work_list));
483 flush_delayed_work(&wb->bw_dwork);
484}
485
486static void wb_exit(struct bdi_writeback *wb)
487{
488 int i;
489
490 WARN_ON(delayed_work_pending(&wb->dwork));
491
492 for (i = 0; i < NR_WB_STAT_ITEMS; i++)
493 percpu_counter_destroy(&wb->stat[i]);
494
495 fprop_local_destroy_percpu(&wb->completions);
496}
497
498#ifdef CONFIG_CGROUP_WRITEBACK
499
500#include <linux/memcontrol.h>
501
502/*
503 * cgwb_lock protects bdi->cgwb_tree, blkcg->cgwb_list, offline_cgwbs and
504 * memcg->cgwb_list. bdi->cgwb_tree is also RCU protected.
505 */
506static DEFINE_SPINLOCK(cgwb_lock);
507static struct workqueue_struct *cgwb_release_wq;
508
509static LIST_HEAD(offline_cgwbs);
510static void cleanup_offline_cgwbs_workfn(struct work_struct *work);
511static DECLARE_WORK(cleanup_offline_cgwbs_work, cleanup_offline_cgwbs_workfn);
512
513static void cgwb_free_rcu(struct rcu_head *rcu_head)
514{
515 struct bdi_writeback *wb = container_of(rcu_head,
516 struct bdi_writeback, rcu);
517
518 percpu_ref_exit(&wb->refcnt);
519 kfree(wb);
520}
521
522static void cgwb_release_workfn(struct work_struct *work)
523{
524 struct bdi_writeback *wb = container_of(work, struct bdi_writeback,
525 release_work);
526 struct backing_dev_info *bdi = wb->bdi;
527
528 mutex_lock(&wb->bdi->cgwb_release_mutex);
529 wb_shutdown(wb);
530
531 css_put(wb->memcg_css);
532 css_put(wb->blkcg_css);
533 mutex_unlock(&wb->bdi->cgwb_release_mutex);
534
535 /* triggers blkg destruction if no online users left */
536 blkcg_unpin_online(wb->blkcg_css);
537
538 fprop_local_destroy_percpu(&wb->memcg_completions);
539
540 spin_lock_irq(&cgwb_lock);
541 list_del(&wb->offline_node);
542 spin_unlock_irq(&cgwb_lock);
543
544 wb_exit(wb);
545 bdi_put(bdi);
546 WARN_ON_ONCE(!list_empty(&wb->b_attached));
547 call_rcu(&wb->rcu, cgwb_free_rcu);
548}
549
550static void cgwb_release(struct percpu_ref *refcnt)
551{
552 struct bdi_writeback *wb = container_of(refcnt, struct bdi_writeback,
553 refcnt);
554 queue_work(cgwb_release_wq, &wb->release_work);
555}
556
557static void cgwb_kill(struct bdi_writeback *wb)
558{
559 lockdep_assert_held(&cgwb_lock);
560
561 WARN_ON(!radix_tree_delete(&wb->bdi->cgwb_tree, wb->memcg_css->id));
562 list_del(&wb->memcg_node);
563 list_del(&wb->blkcg_node);
564 list_add(&wb->offline_node, &offline_cgwbs);
565 percpu_ref_kill(&wb->refcnt);
566}
567
568static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb)
569{
570 spin_lock_irq(&cgwb_lock);
571 list_del_rcu(&wb->bdi_node);
572 spin_unlock_irq(&cgwb_lock);
573}
574
575static int cgwb_create(struct backing_dev_info *bdi,
576 struct cgroup_subsys_state *memcg_css, gfp_t gfp)
577{
578 struct mem_cgroup *memcg;
579 struct cgroup_subsys_state *blkcg_css;
580 struct list_head *memcg_cgwb_list, *blkcg_cgwb_list;
581 struct bdi_writeback *wb;
582 unsigned long flags;
583 int ret = 0;
584
585 memcg = mem_cgroup_from_css(memcg_css);
586 blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
587 memcg_cgwb_list = &memcg->cgwb_list;
588 blkcg_cgwb_list = blkcg_get_cgwb_list(blkcg_css);
589
590 /* look up again under lock and discard on blkcg mismatch */
591 spin_lock_irqsave(&cgwb_lock, flags);
592 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
593 if (wb && wb->blkcg_css != blkcg_css) {
594 cgwb_kill(wb);
595 wb = NULL;
596 }
597 spin_unlock_irqrestore(&cgwb_lock, flags);
598 if (wb)
599 goto out_put;
600
601 /* need to create a new one */
602 wb = kmalloc(sizeof(*wb), gfp);
603 if (!wb) {
604 ret = -ENOMEM;
605 goto out_put;
606 }
607
608 ret = wb_init(wb, bdi, gfp);
609 if (ret)
610 goto err_free;
611
612 ret = percpu_ref_init(&wb->refcnt, cgwb_release, 0, gfp);
613 if (ret)
614 goto err_wb_exit;
615
616 ret = fprop_local_init_percpu(&wb->memcg_completions, gfp);
617 if (ret)
618 goto err_ref_exit;
619
620 wb->memcg_css = memcg_css;
621 wb->blkcg_css = blkcg_css;
622 INIT_LIST_HEAD(&wb->b_attached);
623 INIT_WORK(&wb->release_work, cgwb_release_workfn);
624 set_bit(WB_registered, &wb->state);
625 bdi_get(bdi);
626
627 /*
628 * The root wb determines the registered state of the whole bdi and
629 * memcg_cgwb_list and blkcg_cgwb_list's next pointers indicate
630 * whether they're still online. Don't link @wb if any is dead.
631 * See wb_memcg_offline() and wb_blkcg_offline().
632 */
633 ret = -ENODEV;
634 spin_lock_irqsave(&cgwb_lock, flags);
635 if (test_bit(WB_registered, &bdi->wb.state) &&
636 blkcg_cgwb_list->next && memcg_cgwb_list->next) {
637 /* we might have raced another instance of this function */
638 ret = radix_tree_insert(&bdi->cgwb_tree, memcg_css->id, wb);
639 if (!ret) {
640 list_add_tail_rcu(&wb->bdi_node, &bdi->wb_list);
641 list_add(&wb->memcg_node, memcg_cgwb_list);
642 list_add(&wb->blkcg_node, blkcg_cgwb_list);
643 blkcg_pin_online(blkcg_css);
644 css_get(memcg_css);
645 css_get(blkcg_css);
646 }
647 }
648 spin_unlock_irqrestore(&cgwb_lock, flags);
649 if (ret) {
650 if (ret == -EEXIST)
651 ret = 0;
652 goto err_fprop_exit;
653 }
654 goto out_put;
655
656err_fprop_exit:
657 bdi_put(bdi);
658 fprop_local_destroy_percpu(&wb->memcg_completions);
659err_ref_exit:
660 percpu_ref_exit(&wb->refcnt);
661err_wb_exit:
662 wb_exit(wb);
663err_free:
664 kfree(wb);
665out_put:
666 css_put(blkcg_css);
667 return ret;
668}
669
670/**
671 * wb_get_lookup - get wb for a given memcg
672 * @bdi: target bdi
673 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
674 *
675 * Try to get the wb for @memcg_css on @bdi. The returned wb has its
676 * refcount incremented.
677 *
678 * This function uses css_get() on @memcg_css and thus expects its refcnt
679 * to be positive on invocation. IOW, rcu_read_lock() protection on
680 * @memcg_css isn't enough. try_get it before calling this function.
681 *
682 * A wb is keyed by its associated memcg. As blkcg implicitly enables
683 * memcg on the default hierarchy, memcg association is guaranteed to be
684 * more specific (equal or descendant to the associated blkcg) and thus can
685 * identify both the memcg and blkcg associations.
686 *
687 * Because the blkcg associated with a memcg may change as blkcg is enabled
688 * and disabled closer to root in the hierarchy, each wb keeps track of
689 * both the memcg and blkcg associated with it and verifies the blkcg on
690 * each lookup. On mismatch, the existing wb is discarded and a new one is
691 * created.
692 */
693struct bdi_writeback *wb_get_lookup(struct backing_dev_info *bdi,
694 struct cgroup_subsys_state *memcg_css)
695{
696 struct bdi_writeback *wb;
697
698 if (!memcg_css->parent)
699 return &bdi->wb;
700
701 rcu_read_lock();
702 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
703 if (wb) {
704 struct cgroup_subsys_state *blkcg_css;
705
706 /* see whether the blkcg association has changed */
707 blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
708 if (unlikely(wb->blkcg_css != blkcg_css || !wb_tryget(wb)))
709 wb = NULL;
710 css_put(blkcg_css);
711 }
712 rcu_read_unlock();
713
714 return wb;
715}
716
717/**
718 * wb_get_create - get wb for a given memcg, create if necessary
719 * @bdi: target bdi
720 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
721 * @gfp: allocation mask to use
722 *
723 * Try to get the wb for @memcg_css on @bdi. If it doesn't exist, try to
724 * create one. See wb_get_lookup() for more details.
725 */
726struct bdi_writeback *wb_get_create(struct backing_dev_info *bdi,
727 struct cgroup_subsys_state *memcg_css,
728 gfp_t gfp)
729{
730 struct bdi_writeback *wb;
731
732 might_alloc(gfp);
733
734 do {
735 wb = wb_get_lookup(bdi, memcg_css);
736 } while (!wb && !cgwb_create(bdi, memcg_css, gfp));
737
738 return wb;
739}
740
741static int cgwb_bdi_init(struct backing_dev_info *bdi)
742{
743 int ret;
744
745 INIT_RADIX_TREE(&bdi->cgwb_tree, GFP_ATOMIC);
746 mutex_init(&bdi->cgwb_release_mutex);
747 init_rwsem(&bdi->wb_switch_rwsem);
748
749 ret = wb_init(&bdi->wb, bdi, GFP_KERNEL);
750 if (!ret) {
751 bdi->wb.memcg_css = &root_mem_cgroup->css;
752 bdi->wb.blkcg_css = blkcg_root_css;
753 }
754 return ret;
755}
756
757static void cgwb_bdi_unregister(struct backing_dev_info *bdi)
758{
759 struct radix_tree_iter iter;
760 void **slot;
761 struct bdi_writeback *wb;
762
763 WARN_ON(test_bit(WB_registered, &bdi->wb.state));
764
765 spin_lock_irq(&cgwb_lock);
766 radix_tree_for_each_slot(slot, &bdi->cgwb_tree, &iter, 0)
767 cgwb_kill(*slot);
768 spin_unlock_irq(&cgwb_lock);
769
770 mutex_lock(&bdi->cgwb_release_mutex);
771 spin_lock_irq(&cgwb_lock);
772 while (!list_empty(&bdi->wb_list)) {
773 wb = list_first_entry(&bdi->wb_list, struct bdi_writeback,
774 bdi_node);
775 spin_unlock_irq(&cgwb_lock);
776 wb_shutdown(wb);
777 spin_lock_irq(&cgwb_lock);
778 }
779 spin_unlock_irq(&cgwb_lock);
780 mutex_unlock(&bdi->cgwb_release_mutex);
781}
782
783/*
784 * cleanup_offline_cgwbs_workfn - try to release dying cgwbs
785 *
786 * Try to release dying cgwbs by switching attached inodes to the nearest
787 * living ancestor's writeback. Processed wbs are placed at the end
788 * of the list to guarantee the forward progress.
789 */
790static void cleanup_offline_cgwbs_workfn(struct work_struct *work)
791{
792 struct bdi_writeback *wb;
793 LIST_HEAD(processed);
794
795 spin_lock_irq(&cgwb_lock);
796
797 while (!list_empty(&offline_cgwbs)) {
798 wb = list_first_entry(&offline_cgwbs, struct bdi_writeback,
799 offline_node);
800 list_move(&wb->offline_node, &processed);
801
802 /*
803 * If wb is dirty, cleaning up the writeback by switching
804 * attached inodes will result in an effective removal of any
805 * bandwidth restrictions, which isn't the goal. Instead,
806 * it can be postponed until the next time, when all io
807 * will be likely completed. If in the meantime some inodes
808 * will get re-dirtied, they should be eventually switched to
809 * a new cgwb.
810 */
811 if (wb_has_dirty_io(wb))
812 continue;
813
814 if (!wb_tryget(wb))
815 continue;
816
817 spin_unlock_irq(&cgwb_lock);
818 while (cleanup_offline_cgwb(wb))
819 cond_resched();
820 spin_lock_irq(&cgwb_lock);
821
822 wb_put(wb);
823 }
824
825 if (!list_empty(&processed))
826 list_splice_tail(&processed, &offline_cgwbs);
827
828 spin_unlock_irq(&cgwb_lock);
829}
830
831/**
832 * wb_memcg_offline - kill all wb's associated with a memcg being offlined
833 * @memcg: memcg being offlined
834 *
835 * Also prevents creation of any new wb's associated with @memcg.
836 */
837void wb_memcg_offline(struct mem_cgroup *memcg)
838{
839 struct list_head *memcg_cgwb_list = &memcg->cgwb_list;
840 struct bdi_writeback *wb, *next;
841
842 spin_lock_irq(&cgwb_lock);
843 list_for_each_entry_safe(wb, next, memcg_cgwb_list, memcg_node)
844 cgwb_kill(wb);
845 memcg_cgwb_list->next = NULL; /* prevent new wb's */
846 spin_unlock_irq(&cgwb_lock);
847
848 queue_work(system_unbound_wq, &cleanup_offline_cgwbs_work);
849}
850
851/**
852 * wb_blkcg_offline - kill all wb's associated with a blkcg being offlined
853 * @css: blkcg being offlined
854 *
855 * Also prevents creation of any new wb's associated with @blkcg.
856 */
857void wb_blkcg_offline(struct cgroup_subsys_state *css)
858{
859 struct bdi_writeback *wb, *next;
860 struct list_head *list = blkcg_get_cgwb_list(css);
861
862 spin_lock_irq(&cgwb_lock);
863 list_for_each_entry_safe(wb, next, list, blkcg_node)
864 cgwb_kill(wb);
865 list->next = NULL; /* prevent new wb's */
866 spin_unlock_irq(&cgwb_lock);
867}
868
869static void cgwb_bdi_register(struct backing_dev_info *bdi)
870{
871 spin_lock_irq(&cgwb_lock);
872 list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);
873 spin_unlock_irq(&cgwb_lock);
874}
875
876static int __init cgwb_init(void)
877{
878 /*
879 * There can be many concurrent release work items overwhelming
880 * system_wq. Put them in a separate wq and limit concurrency.
881 * There's no point in executing many of these in parallel.
882 */
883 cgwb_release_wq = alloc_workqueue("cgwb_release", 0, 1);
884 if (!cgwb_release_wq)
885 return -ENOMEM;
886
887 return 0;
888}
889subsys_initcall(cgwb_init);
890
891#else /* CONFIG_CGROUP_WRITEBACK */
892
893static int cgwb_bdi_init(struct backing_dev_info *bdi)
894{
895 return wb_init(&bdi->wb, bdi, GFP_KERNEL);
896}
897
898static void cgwb_bdi_unregister(struct backing_dev_info *bdi) { }
899
900static void cgwb_bdi_register(struct backing_dev_info *bdi)
901{
902 list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);
903}
904
905static void cgwb_remove_from_bdi_list(struct bdi_writeback *wb)
906{
907 list_del_rcu(&wb->bdi_node);
908}
909
910#endif /* CONFIG_CGROUP_WRITEBACK */
911
912int bdi_init(struct backing_dev_info *bdi)
913{
914 bdi->dev = NULL;
915
916 kref_init(&bdi->refcnt);
917 bdi->min_ratio = 0;
918 bdi->max_ratio = 100 * BDI_RATIO_SCALE;
919 bdi->max_prop_frac = FPROP_FRAC_BASE;
920 INIT_LIST_HEAD(&bdi->bdi_list);
921 INIT_LIST_HEAD(&bdi->wb_list);
922 init_waitqueue_head(&bdi->wb_waitq);
923 bdi->last_bdp_sleep = jiffies;
924
925 return cgwb_bdi_init(bdi);
926}
927
928struct backing_dev_info *bdi_alloc(int node_id)
929{
930 struct backing_dev_info *bdi;
931
932 bdi = kzalloc_node(sizeof(*bdi), GFP_KERNEL, node_id);
933 if (!bdi)
934 return NULL;
935
936 if (bdi_init(bdi)) {
937 kfree(bdi);
938 return NULL;
939 }
940 bdi->capabilities = BDI_CAP_WRITEBACK | BDI_CAP_WRITEBACK_ACCT;
941 bdi->ra_pages = VM_READAHEAD_PAGES;
942 bdi->io_pages = VM_READAHEAD_PAGES;
943 timer_setup(&bdi->laptop_mode_wb_timer, laptop_mode_timer_fn, 0);
944 return bdi;
945}
946EXPORT_SYMBOL(bdi_alloc);
947
948static struct rb_node **bdi_lookup_rb_node(u64 id, struct rb_node **parentp)
949{
950 struct rb_node **p = &bdi_tree.rb_node;
951 struct rb_node *parent = NULL;
952 struct backing_dev_info *bdi;
953
954 lockdep_assert_held(&bdi_lock);
955
956 while (*p) {
957 parent = *p;
958 bdi = rb_entry(parent, struct backing_dev_info, rb_node);
959
960 if (bdi->id > id)
961 p = &(*p)->rb_left;
962 else if (bdi->id < id)
963 p = &(*p)->rb_right;
964 else
965 break;
966 }
967
968 if (parentp)
969 *parentp = parent;
970 return p;
971}
972
973/**
974 * bdi_get_by_id - lookup and get bdi from its id
975 * @id: bdi id to lookup
976 *
977 * Find bdi matching @id and get it. Returns NULL if the matching bdi
978 * doesn't exist or is already unregistered.
979 */
980struct backing_dev_info *bdi_get_by_id(u64 id)
981{
982 struct backing_dev_info *bdi = NULL;
983 struct rb_node **p;
984
985 spin_lock_bh(&bdi_lock);
986 p = bdi_lookup_rb_node(id, NULL);
987 if (*p) {
988 bdi = rb_entry(*p, struct backing_dev_info, rb_node);
989 bdi_get(bdi);
990 }
991 spin_unlock_bh(&bdi_lock);
992
993 return bdi;
994}
995
996int bdi_register_va(struct backing_dev_info *bdi, const char *fmt, va_list args)
997{
998 struct device *dev;
999 struct rb_node *parent, **p;
1000
1001 if (bdi->dev) /* The driver needs to use separate queues per device */
1002 return 0;
1003
1004 vsnprintf(bdi->dev_name, sizeof(bdi->dev_name), fmt, args);
1005 dev = device_create(&bdi_class, NULL, MKDEV(0, 0), bdi, bdi->dev_name);
1006 if (IS_ERR(dev))
1007 return PTR_ERR(dev);
1008
1009 cgwb_bdi_register(bdi);
1010 bdi->dev = dev;
1011
1012 bdi_debug_register(bdi, dev_name(dev));
1013 set_bit(WB_registered, &bdi->wb.state);
1014
1015 spin_lock_bh(&bdi_lock);
1016
1017 bdi->id = ++bdi_id_cursor;
1018
1019 p = bdi_lookup_rb_node(bdi->id, &parent);
1020 rb_link_node(&bdi->rb_node, parent, p);
1021 rb_insert_color(&bdi->rb_node, &bdi_tree);
1022
1023 list_add_tail_rcu(&bdi->bdi_list, &bdi_list);
1024
1025 spin_unlock_bh(&bdi_lock);
1026
1027 trace_writeback_bdi_register(bdi);
1028 return 0;
1029}
1030
1031int bdi_register(struct backing_dev_info *bdi, const char *fmt, ...)
1032{
1033 va_list args;
1034 int ret;
1035
1036 va_start(args, fmt);
1037 ret = bdi_register_va(bdi, fmt, args);
1038 va_end(args);
1039 return ret;
1040}
1041EXPORT_SYMBOL(bdi_register);
1042
1043void bdi_set_owner(struct backing_dev_info *bdi, struct device *owner)
1044{
1045 WARN_ON_ONCE(bdi->owner);
1046 bdi->owner = owner;
1047 get_device(owner);
1048}
1049
1050/*
1051 * Remove bdi from bdi_list, and ensure that it is no longer visible
1052 */
1053static void bdi_remove_from_list(struct backing_dev_info *bdi)
1054{
1055 spin_lock_bh(&bdi_lock);
1056 rb_erase(&bdi->rb_node, &bdi_tree);
1057 list_del_rcu(&bdi->bdi_list);
1058 spin_unlock_bh(&bdi_lock);
1059
1060 synchronize_rcu_expedited();
1061}
1062
1063void bdi_unregister(struct backing_dev_info *bdi)
1064{
1065 del_timer_sync(&bdi->laptop_mode_wb_timer);
1066
1067 /* make sure nobody finds us on the bdi_list anymore */
1068 bdi_remove_from_list(bdi);
1069 wb_shutdown(&bdi->wb);
1070 cgwb_bdi_unregister(bdi);
1071
1072 /*
1073 * If this BDI's min ratio has been set, use bdi_set_min_ratio() to
1074 * update the global bdi_min_ratio.
1075 */
1076 if (bdi->min_ratio)
1077 bdi_set_min_ratio(bdi, 0);
1078
1079 if (bdi->dev) {
1080 bdi_debug_unregister(bdi);
1081 device_unregister(bdi->dev);
1082 bdi->dev = NULL;
1083 }
1084
1085 if (bdi->owner) {
1086 put_device(bdi->owner);
1087 bdi->owner = NULL;
1088 }
1089}
1090EXPORT_SYMBOL(bdi_unregister);
1091
1092static void release_bdi(struct kref *ref)
1093{
1094 struct backing_dev_info *bdi =
1095 container_of(ref, struct backing_dev_info, refcnt);
1096
1097 WARN_ON_ONCE(test_bit(WB_registered, &bdi->wb.state));
1098 WARN_ON_ONCE(bdi->dev);
1099 wb_exit(&bdi->wb);
1100 kfree(bdi);
1101}
1102
1103void bdi_put(struct backing_dev_info *bdi)
1104{
1105 kref_put(&bdi->refcnt, release_bdi);
1106}
1107EXPORT_SYMBOL(bdi_put);
1108
1109struct backing_dev_info *inode_to_bdi(struct inode *inode)
1110{
1111 struct super_block *sb;
1112
1113 if (!inode)
1114 return &noop_backing_dev_info;
1115
1116 sb = inode->i_sb;
1117#ifdef CONFIG_BLOCK
1118 if (sb_is_blkdev_sb(sb))
1119 return I_BDEV(inode)->bd_disk->bdi;
1120#endif
1121 return sb->s_bdi;
1122}
1123EXPORT_SYMBOL(inode_to_bdi);
1124
1125const char *bdi_dev_name(struct backing_dev_info *bdi)
1126{
1127 if (!bdi || !bdi->dev)
1128 return bdi_unknown_name;
1129 return bdi->dev_name;
1130}
1131EXPORT_SYMBOL_GPL(bdi_dev_name);