NVMe: Fix memory leak on retried commands
[linux-2.6-block.git] / mm / backing-dev.c
CommitLineData
3fcfab16
AM
1
2#include <linux/wait.h>
3#include <linux/backing-dev.h>
03ba3782
JA
4#include <linux/kthread.h>
5#include <linux/freezer.h>
3fcfab16 6#include <linux/fs.h>
26160158 7#include <linux/pagemap.h>
03ba3782 8#include <linux/mm.h>
3fcfab16
AM
9#include <linux/sched.h>
10#include <linux/module.h>
cf0ca9fe
PZ
11#include <linux/writeback.h>
12#include <linux/device.h>
455b2864 13#include <trace/events/writeback.h>
cf0ca9fe 14
c3c53206
JA
15static atomic_long_t bdi_seq = ATOMIC_LONG_INIT(0);
16
5129a469
JE
17struct backing_dev_info noop_backing_dev_info = {
18 .name = "noop",
976e48f8 19 .capabilities = BDI_CAP_NO_ACCT_AND_WRITEBACK,
5129a469 20};
a212b105 21EXPORT_SYMBOL_GPL(noop_backing_dev_info);
5129a469 22
cf0ca9fe 23static struct class *bdi_class;
cfc4ba53
JA
24
25/*
181387da 26 * bdi_lock protects updates to bdi_list. bdi_list has RCU reader side
cfc4ba53
JA
27 * locking.
28 */
03ba3782 29DEFINE_SPINLOCK(bdi_lock);
66f3b8e2 30LIST_HEAD(bdi_list);
03ba3782 31
839a8e86
TH
32/* bdi_wq serves all asynchronous writeback tasks */
33struct workqueue_struct *bdi_wq;
34
76f1418b
MS
35#ifdef CONFIG_DEBUG_FS
36#include <linux/debugfs.h>
37#include <linux/seq_file.h>
38
39static struct dentry *bdi_debug_root;
40
41static void bdi_debug_init(void)
42{
43 bdi_debug_root = debugfs_create_dir("bdi", NULL);
44}
45
46static int bdi_debug_stats_show(struct seq_file *m, void *v)
47{
48 struct backing_dev_info *bdi = m->private;
c1955ce3 49 struct bdi_writeback *wb = &bdi->wb;
364aeb28
DR
50 unsigned long background_thresh;
51 unsigned long dirty_thresh;
0d960a38 52 unsigned long wb_thresh;
0ae45f63 53 unsigned long nr_dirty, nr_io, nr_more_io, nr_dirty_time;
f09b00d3
JA
54 struct inode *inode;
55
0ae45f63 56 nr_dirty = nr_io = nr_more_io = nr_dirty_time = 0;
f758eeab 57 spin_lock(&wb->list_lock);
c7f54084 58 list_for_each_entry(inode, &wb->b_dirty, i_io_list)
c1955ce3 59 nr_dirty++;
c7f54084 60 list_for_each_entry(inode, &wb->b_io, i_io_list)
c1955ce3 61 nr_io++;
c7f54084 62 list_for_each_entry(inode, &wb->b_more_io, i_io_list)
c1955ce3 63 nr_more_io++;
c7f54084 64 list_for_each_entry(inode, &wb->b_dirty_time, i_io_list)
0ae45f63
TT
65 if (inode->i_state & I_DIRTY_TIME)
66 nr_dirty_time++;
f758eeab 67 spin_unlock(&wb->list_lock);
76f1418b 68
16c4042f 69 global_dirty_limits(&background_thresh, &dirty_thresh);
0d960a38 70 wb_thresh = wb_calc_thresh(wb, dirty_thresh);
76f1418b
MS
71
72#define K(x) ((x) << (PAGE_SHIFT - 10))
73 seq_printf(m,
00821b00
WF
74 "BdiWriteback: %10lu kB\n"
75 "BdiReclaimable: %10lu kB\n"
76 "BdiDirtyThresh: %10lu kB\n"
77 "DirtyThresh: %10lu kB\n"
78 "BackgroundThresh: %10lu kB\n"
c8e28ce0 79 "BdiDirtied: %10lu kB\n"
00821b00
WF
80 "BdiWritten: %10lu kB\n"
81 "BdiWriteBandwidth: %10lu kBps\n"
82 "b_dirty: %10lu\n"
83 "b_io: %10lu\n"
84 "b_more_io: %10lu\n"
0ae45f63 85 "b_dirty_time: %10lu\n"
00821b00
WF
86 "bdi_list: %10u\n"
87 "state: %10lx\n",
93f78d88
TH
88 (unsigned long) K(wb_stat(wb, WB_WRITEBACK)),
89 (unsigned long) K(wb_stat(wb, WB_RECLAIMABLE)),
0d960a38 90 K(wb_thresh),
f7d2b1ec
JK
91 K(dirty_thresh),
92 K(background_thresh),
93f78d88
TH
93 (unsigned long) K(wb_stat(wb, WB_DIRTIED)),
94 (unsigned long) K(wb_stat(wb, WB_WRITTEN)),
a88a341a 95 (unsigned long) K(wb->write_bandwidth),
f7d2b1ec
JK
96 nr_dirty,
97 nr_io,
98 nr_more_io,
0ae45f63 99 nr_dirty_time,
4452226e 100 !list_empty(&bdi->bdi_list), bdi->wb.state);
76f1418b
MS
101#undef K
102
103 return 0;
104}
105
106static int bdi_debug_stats_open(struct inode *inode, struct file *file)
107{
108 return single_open(file, bdi_debug_stats_show, inode->i_private);
109}
110
111static const struct file_operations bdi_debug_stats_fops = {
112 .open = bdi_debug_stats_open,
113 .read = seq_read,
114 .llseek = seq_lseek,
115 .release = single_release,
116};
117
118static void bdi_debug_register(struct backing_dev_info *bdi, const char *name)
119{
120 bdi->debug_dir = debugfs_create_dir(name, bdi_debug_root);
121 bdi->debug_stats = debugfs_create_file("stats", 0444, bdi->debug_dir,
122 bdi, &bdi_debug_stats_fops);
123}
124
125static void bdi_debug_unregister(struct backing_dev_info *bdi)
126{
127 debugfs_remove(bdi->debug_stats);
128 debugfs_remove(bdi->debug_dir);
129}
130#else
131static inline void bdi_debug_init(void)
132{
133}
134static inline void bdi_debug_register(struct backing_dev_info *bdi,
135 const char *name)
136{
137}
138static inline void bdi_debug_unregister(struct backing_dev_info *bdi)
139{
140}
141#endif
142
cf0ca9fe
PZ
143static ssize_t read_ahead_kb_store(struct device *dev,
144 struct device_attribute *attr,
145 const char *buf, size_t count)
146{
147 struct backing_dev_info *bdi = dev_get_drvdata(dev);
cf0ca9fe 148 unsigned long read_ahead_kb;
7034ed13 149 ssize_t ret;
cf0ca9fe 150
7034ed13
NJ
151 ret = kstrtoul(buf, 10, &read_ahead_kb);
152 if (ret < 0)
153 return ret;
154
155 bdi->ra_pages = read_ahead_kb >> (PAGE_SHIFT - 10);
156
157 return count;
cf0ca9fe
PZ
158}
159
160#define K(pages) ((pages) << (PAGE_SHIFT - 10))
161
162#define BDI_SHOW(name, expr) \
163static ssize_t name##_show(struct device *dev, \
164 struct device_attribute *attr, char *page) \
165{ \
166 struct backing_dev_info *bdi = dev_get_drvdata(dev); \
167 \
168 return snprintf(page, PAGE_SIZE-1, "%lld\n", (long long)expr); \
d9e1241e
GKH
169} \
170static DEVICE_ATTR_RW(name);
cf0ca9fe
PZ
171
172BDI_SHOW(read_ahead_kb, K(bdi->ra_pages))
173
189d3c4a
PZ
174static ssize_t min_ratio_store(struct device *dev,
175 struct device_attribute *attr, const char *buf, size_t count)
176{
177 struct backing_dev_info *bdi = dev_get_drvdata(dev);
189d3c4a 178 unsigned int ratio;
7034ed13
NJ
179 ssize_t ret;
180
181 ret = kstrtouint(buf, 10, &ratio);
182 if (ret < 0)
183 return ret;
184
185 ret = bdi_set_min_ratio(bdi, ratio);
186 if (!ret)
187 ret = count;
189d3c4a 188
189d3c4a
PZ
189 return ret;
190}
191BDI_SHOW(min_ratio, bdi->min_ratio)
192
a42dde04
PZ
193static ssize_t max_ratio_store(struct device *dev,
194 struct device_attribute *attr, const char *buf, size_t count)
195{
196 struct backing_dev_info *bdi = dev_get_drvdata(dev);
a42dde04 197 unsigned int ratio;
7034ed13
NJ
198 ssize_t ret;
199
200 ret = kstrtouint(buf, 10, &ratio);
201 if (ret < 0)
202 return ret;
203
204 ret = bdi_set_max_ratio(bdi, ratio);
205 if (!ret)
206 ret = count;
a42dde04 207
a42dde04
PZ
208 return ret;
209}
210BDI_SHOW(max_ratio, bdi->max_ratio)
211
7d311cda
DW
212static ssize_t stable_pages_required_show(struct device *dev,
213 struct device_attribute *attr,
214 char *page)
215{
216 struct backing_dev_info *bdi = dev_get_drvdata(dev);
217
218 return snprintf(page, PAGE_SIZE-1, "%d\n",
219 bdi_cap_stable_pages_required(bdi) ? 1 : 0);
220}
d9e1241e
GKH
221static DEVICE_ATTR_RO(stable_pages_required);
222
223static struct attribute *bdi_dev_attrs[] = {
224 &dev_attr_read_ahead_kb.attr,
225 &dev_attr_min_ratio.attr,
226 &dev_attr_max_ratio.attr,
227 &dev_attr_stable_pages_required.attr,
228 NULL,
cf0ca9fe 229};
d9e1241e 230ATTRIBUTE_GROUPS(bdi_dev);
cf0ca9fe
PZ
231
232static __init int bdi_class_init(void)
233{
234 bdi_class = class_create(THIS_MODULE, "bdi");
14421453
AB
235 if (IS_ERR(bdi_class))
236 return PTR_ERR(bdi_class);
237
d9e1241e 238 bdi_class->dev_groups = bdi_dev_groups;
76f1418b 239 bdi_debug_init();
cf0ca9fe
PZ
240 return 0;
241}
76f1418b 242postcore_initcall(bdi_class_init);
cf0ca9fe 243
26160158
JA
244static int __init default_bdi_init(void)
245{
246 int err;
247
839a8e86 248 bdi_wq = alloc_workqueue("writeback", WQ_MEM_RECLAIM | WQ_FREEZABLE |
b5c872dd 249 WQ_UNBOUND | WQ_SYSFS, 0);
839a8e86
TH
250 if (!bdi_wq)
251 return -ENOMEM;
252
976e48f8 253 err = bdi_init(&noop_backing_dev_info);
26160158
JA
254
255 return err;
256}
257subsys_initcall(default_bdi_init);
258
6467716a 259/*
f0054bb1 260 * This function is used when the first inode for this wb is marked dirty. It
6467716a
AB
261 * wakes-up the corresponding bdi thread which should then take care of the
262 * periodic background write-out of dirty inodes. Since the write-out would
263 * starts only 'dirty_writeback_interval' centisecs from now anyway, we just
264 * set up a timer which wakes the bdi thread up later.
265 *
266 * Note, we wouldn't bother setting up the timer, but this function is on the
267 * fast-path (used by '__mark_inode_dirty()'), so we save few context switches
268 * by delaying the wake-up.
6ca738d6
DB
269 *
270 * We have to be careful not to postpone flush work if it is scheduled for
271 * earlier. Thus we use queue_delayed_work().
6467716a 272 */
f0054bb1 273void wb_wakeup_delayed(struct bdi_writeback *wb)
6467716a
AB
274{
275 unsigned long timeout;
276
277 timeout = msecs_to_jiffies(dirty_writeback_interval * 10);
f0054bb1
TH
278 spin_lock_bh(&wb->work_lock);
279 if (test_bit(WB_registered, &wb->state))
280 queue_delayed_work(bdi_wq, &wb->dwork, timeout);
281 spin_unlock_bh(&wb->work_lock);
03ba3782
JA
282}
283
cfc4ba53 284/*
a88a341a 285 * Initial write bandwidth: 100 MB/s
cfc4ba53 286 */
a88a341a 287#define INIT_BW (100 << (20 - PAGE_SHIFT))
cfc4ba53 288
8395cd9f 289static int wb_init(struct bdi_writeback *wb, struct backing_dev_info *bdi,
a13f35e8 290 int blkcg_id, gfp_t gfp)
cf0ca9fe 291{
93f78d88 292 int i, err;
cf0ca9fe 293
6467716a 294 memset(wb, 0, sizeof(*wb));
f1d0b063 295
6467716a
AB
296 wb->bdi = bdi;
297 wb->last_old_flush = jiffies;
298 INIT_LIST_HEAD(&wb->b_dirty);
299 INIT_LIST_HEAD(&wb->b_io);
300 INIT_LIST_HEAD(&wb->b_more_io);
0ae45f63 301 INIT_LIST_HEAD(&wb->b_dirty_time);
f758eeab 302 spin_lock_init(&wb->list_lock);
66f3b8e2 303
a88a341a
TH
304 wb->bw_time_stamp = jiffies;
305 wb->balanced_dirty_ratelimit = INIT_BW;
306 wb->dirty_ratelimit = INIT_BW;
307 wb->write_bandwidth = INIT_BW;
308 wb->avg_write_bandwidth = INIT_BW;
cf0ca9fe 309
f0054bb1
TH
310 spin_lock_init(&wb->work_lock);
311 INIT_LIST_HEAD(&wb->work_list);
312 INIT_DELAYED_WORK(&wb->dwork, wb_workfn);
c284de61 313
a13f35e8
TH
314 wb->congested = wb_congested_get_create(bdi, blkcg_id, gfp);
315 if (!wb->congested)
316 return -ENOMEM;
317
8395cd9f 318 err = fprop_local_init_percpu(&wb->completions, gfp);
a88a341a 319 if (err)
a13f35e8 320 goto out_put_cong;
c284de61 321
93f78d88 322 for (i = 0; i < NR_WB_STAT_ITEMS; i++) {
8395cd9f 323 err = percpu_counter_init(&wb->stat[i], 0, gfp);
a13f35e8
TH
324 if (err)
325 goto out_destroy_stat;
93f78d88 326 }
cf0ca9fe 327
93f78d88 328 return 0;
a13f35e8
TH
329
330out_destroy_stat:
331 while (--i)
332 percpu_counter_destroy(&wb->stat[i]);
333 fprop_local_destroy_percpu(&wb->completions);
334out_put_cong:
335 wb_congested_put(wb->congested);
336 return err;
cf0ca9fe 337}
cf0ca9fe 338
03ba3782
JA
339/*
340 * Remove bdi from the global list and shutdown any threads we have running
341 */
46100071 342static void wb_shutdown(struct bdi_writeback *wb)
66f3b8e2 343{
c4db59d3 344 /* Make sure nobody queues further work */
46100071
TH
345 spin_lock_bh(&wb->work_lock);
346 if (!test_and_clear_bit(WB_registered, &wb->state)) {
347 spin_unlock_bh(&wb->work_lock);
03ba3782 348 return;
c4db59d3 349 }
46100071 350 spin_unlock_bh(&wb->work_lock);
03ba3782
JA
351
352 /*
46100071
TH
353 * Drain work list and shutdown the delayed_work. !WB_registered
354 * tells wb_workfn() that @wb is dying and its work_list needs to
355 * be drained no matter what.
03ba3782 356 */
46100071
TH
357 mod_delayed_work(bdi_wq, &wb->dwork, 0);
358 flush_delayed_work(&wb->dwork);
359 WARN_ON(!list_empty(&wb->work_list));
360}
361
f0054bb1 362static void wb_exit(struct bdi_writeback *wb)
93f78d88
TH
363{
364 int i;
365
366 WARN_ON(delayed_work_pending(&wb->dwork));
367
368 for (i = 0; i < NR_WB_STAT_ITEMS; i++)
369 percpu_counter_destroy(&wb->stat[i]);
6467716a 370
a88a341a 371 fprop_local_destroy_percpu(&wb->completions);
a13f35e8 372 wb_congested_put(wb->congested);
a88a341a 373}
e98be2d5 374
52ebea74
TH
375#ifdef CONFIG_CGROUP_WRITEBACK
376
377#include <linux/memcontrol.h>
378
379/*
380 * cgwb_lock protects bdi->cgwb_tree, bdi->cgwb_congested_tree,
381 * blkcg->cgwb_list, and memcg->cgwb_list. bdi->cgwb_tree is also RCU
382 * protected. cgwb_release_wait is used to wait for the completion of cgwb
383 * releases from bdi destruction path.
384 */
385static DEFINE_SPINLOCK(cgwb_lock);
386static DECLARE_WAIT_QUEUE_HEAD(cgwb_release_wait);
387
388/**
389 * wb_congested_get_create - get or create a wb_congested
390 * @bdi: associated bdi
391 * @blkcg_id: ID of the associated blkcg
392 * @gfp: allocation mask
393 *
394 * Look up the wb_congested for @blkcg_id on @bdi. If missing, create one.
395 * The returned wb_congested has its reference count incremented. Returns
396 * NULL on failure.
397 */
398struct bdi_writeback_congested *
399wb_congested_get_create(struct backing_dev_info *bdi, int blkcg_id, gfp_t gfp)
400{
401 struct bdi_writeback_congested *new_congested = NULL, *congested;
402 struct rb_node **node, *parent;
403 unsigned long flags;
52ebea74
TH
404retry:
405 spin_lock_irqsave(&cgwb_lock, flags);
406
407 node = &bdi->cgwb_congested_tree.rb_node;
408 parent = NULL;
409
410 while (*node != NULL) {
411 parent = *node;
412 congested = container_of(parent, struct bdi_writeback_congested,
413 rb_node);
414 if (congested->blkcg_id < blkcg_id)
415 node = &parent->rb_left;
416 else if (congested->blkcg_id > blkcg_id)
417 node = &parent->rb_right;
418 else
419 goto found;
420 }
421
422 if (new_congested) {
423 /* !found and storage for new one already allocated, insert */
424 congested = new_congested;
425 new_congested = NULL;
426 rb_link_node(&congested->rb_node, parent, node);
427 rb_insert_color(&congested->rb_node, &bdi->cgwb_congested_tree);
52ebea74
TH
428 goto found;
429 }
430
431 spin_unlock_irqrestore(&cgwb_lock, flags);
432
433 /* allocate storage for new one and retry */
434 new_congested = kzalloc(sizeof(*new_congested), gfp);
435 if (!new_congested)
436 return NULL;
437
438 atomic_set(&new_congested->refcnt, 0);
439 new_congested->bdi = bdi;
440 new_congested->blkcg_id = blkcg_id;
441 goto retry;
442
443found:
444 atomic_inc(&congested->refcnt);
445 spin_unlock_irqrestore(&cgwb_lock, flags);
446 kfree(new_congested);
447 return congested;
448}
449
450/**
451 * wb_congested_put - put a wb_congested
452 * @congested: wb_congested to put
453 *
454 * Put @congested and destroy it if the refcnt reaches zero.
455 */
456void wb_congested_put(struct bdi_writeback_congested *congested)
457{
52ebea74
TH
458 unsigned long flags;
459
52ebea74
TH
460 local_irq_save(flags);
461 if (!atomic_dec_and_lock(&congested->refcnt, &cgwb_lock)) {
462 local_irq_restore(flags);
463 return;
464 }
465
a20135ff
TH
466 /* bdi might already have been destroyed leaving @congested unlinked */
467 if (congested->bdi) {
468 rb_erase(&congested->rb_node,
469 &congested->bdi->cgwb_congested_tree);
470 congested->bdi = NULL;
471 }
472
52ebea74
TH
473 spin_unlock_irqrestore(&cgwb_lock, flags);
474 kfree(congested);
52ebea74
TH
475}
476
477static void cgwb_release_workfn(struct work_struct *work)
478{
479 struct bdi_writeback *wb = container_of(work, struct bdi_writeback,
480 release_work);
481 struct backing_dev_info *bdi = wb->bdi;
482
b817525a
TH
483 spin_lock_irq(&cgwb_lock);
484 list_del_rcu(&wb->bdi_node);
485 spin_unlock_irq(&cgwb_lock);
486
52ebea74
TH
487 wb_shutdown(wb);
488
489 css_put(wb->memcg_css);
490 css_put(wb->blkcg_css);
52ebea74 491
841710aa 492 fprop_local_destroy_percpu(&wb->memcg_completions);
52ebea74
TH
493 percpu_ref_exit(&wb->refcnt);
494 wb_exit(wb);
495 kfree_rcu(wb, rcu);
496
497 if (atomic_dec_and_test(&bdi->usage_cnt))
498 wake_up_all(&cgwb_release_wait);
499}
500
501static void cgwb_release(struct percpu_ref *refcnt)
502{
503 struct bdi_writeback *wb = container_of(refcnt, struct bdi_writeback,
504 refcnt);
505 schedule_work(&wb->release_work);
506}
507
508static void cgwb_kill(struct bdi_writeback *wb)
509{
510 lockdep_assert_held(&cgwb_lock);
511
512 WARN_ON(!radix_tree_delete(&wb->bdi->cgwb_tree, wb->memcg_css->id));
513 list_del(&wb->memcg_node);
514 list_del(&wb->blkcg_node);
515 percpu_ref_kill(&wb->refcnt);
516}
517
518static int cgwb_create(struct backing_dev_info *bdi,
519 struct cgroup_subsys_state *memcg_css, gfp_t gfp)
520{
521 struct mem_cgroup *memcg;
522 struct cgroup_subsys_state *blkcg_css;
523 struct blkcg *blkcg;
524 struct list_head *memcg_cgwb_list, *blkcg_cgwb_list;
525 struct bdi_writeback *wb;
526 unsigned long flags;
527 int ret = 0;
528
529 memcg = mem_cgroup_from_css(memcg_css);
c165b3e3 530 blkcg_css = cgroup_get_e_css(memcg_css->cgroup, &io_cgrp_subsys);
52ebea74
TH
531 blkcg = css_to_blkcg(blkcg_css);
532 memcg_cgwb_list = mem_cgroup_cgwb_list(memcg);
533 blkcg_cgwb_list = &blkcg->cgwb_list;
534
535 /* look up again under lock and discard on blkcg mismatch */
536 spin_lock_irqsave(&cgwb_lock, flags);
537 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
538 if (wb && wb->blkcg_css != blkcg_css) {
539 cgwb_kill(wb);
540 wb = NULL;
541 }
542 spin_unlock_irqrestore(&cgwb_lock, flags);
543 if (wb)
544 goto out_put;
545
546 /* need to create a new one */
547 wb = kmalloc(sizeof(*wb), gfp);
548 if (!wb)
549 return -ENOMEM;
550
a13f35e8 551 ret = wb_init(wb, bdi, blkcg_css->id, gfp);
52ebea74
TH
552 if (ret)
553 goto err_free;
554
555 ret = percpu_ref_init(&wb->refcnt, cgwb_release, 0, gfp);
556 if (ret)
557 goto err_wb_exit;
558
841710aa
TH
559 ret = fprop_local_init_percpu(&wb->memcg_completions, gfp);
560 if (ret)
561 goto err_ref_exit;
562
52ebea74
TH
563 wb->memcg_css = memcg_css;
564 wb->blkcg_css = blkcg_css;
565 INIT_WORK(&wb->release_work, cgwb_release_workfn);
566 set_bit(WB_registered, &wb->state);
03ba3782
JA
567
568 /*
52ebea74
TH
569 * The root wb determines the registered state of the whole bdi and
570 * memcg_cgwb_list and blkcg_cgwb_list's next pointers indicate
571 * whether they're still online. Don't link @wb if any is dead.
572 * See wb_memcg_offline() and wb_blkcg_offline().
03ba3782 573 */
52ebea74
TH
574 ret = -ENODEV;
575 spin_lock_irqsave(&cgwb_lock, flags);
576 if (test_bit(WB_registered, &bdi->wb.state) &&
577 blkcg_cgwb_list->next && memcg_cgwb_list->next) {
578 /* we might have raced another instance of this function */
579 ret = radix_tree_insert(&bdi->cgwb_tree, memcg_css->id, wb);
580 if (!ret) {
581 atomic_inc(&bdi->usage_cnt);
b817525a 582 list_add_tail_rcu(&wb->bdi_node, &bdi->wb_list);
52ebea74
TH
583 list_add(&wb->memcg_node, memcg_cgwb_list);
584 list_add(&wb->blkcg_node, blkcg_cgwb_list);
585 css_get(memcg_css);
586 css_get(blkcg_css);
587 }
588 }
589 spin_unlock_irqrestore(&cgwb_lock, flags);
590 if (ret) {
591 if (ret == -EEXIST)
592 ret = 0;
a13f35e8 593 goto err_fprop_exit;
52ebea74
TH
594 }
595 goto out_put;
596
841710aa
TH
597err_fprop_exit:
598 fprop_local_destroy_percpu(&wb->memcg_completions);
52ebea74
TH
599err_ref_exit:
600 percpu_ref_exit(&wb->refcnt);
601err_wb_exit:
602 wb_exit(wb);
603err_free:
604 kfree(wb);
605out_put:
606 css_put(blkcg_css);
607 return ret;
66f3b8e2
JA
608}
609
52ebea74
TH
610/**
611 * wb_get_create - get wb for a given memcg, create if necessary
612 * @bdi: target bdi
613 * @memcg_css: cgroup_subsys_state of the target memcg (must have positive ref)
614 * @gfp: allocation mask to use
615 *
616 * Try to get the wb for @memcg_css on @bdi. If it doesn't exist, try to
617 * create one. The returned wb has its refcount incremented.
618 *
619 * This function uses css_get() on @memcg_css and thus expects its refcnt
620 * to be positive on invocation. IOW, rcu_read_lock() protection on
621 * @memcg_css isn't enough. try_get it before calling this function.
622 *
623 * A wb is keyed by its associated memcg. As blkcg implicitly enables
624 * memcg on the default hierarchy, memcg association is guaranteed to be
625 * more specific (equal or descendant to the associated blkcg) and thus can
626 * identify both the memcg and blkcg associations.
627 *
628 * Because the blkcg associated with a memcg may change as blkcg is enabled
629 * and disabled closer to root in the hierarchy, each wb keeps track of
630 * both the memcg and blkcg associated with it and verifies the blkcg on
631 * each lookup. On mismatch, the existing wb is discarded and a new one is
632 * created.
633 */
634struct bdi_writeback *wb_get_create(struct backing_dev_info *bdi,
635 struct cgroup_subsys_state *memcg_css,
636 gfp_t gfp)
6467716a 637{
52ebea74
TH
638 struct bdi_writeback *wb;
639
640 might_sleep_if(gfp & __GFP_WAIT);
641
642 if (!memcg_css->parent)
643 return &bdi->wb;
644
645 do {
646 rcu_read_lock();
647 wb = radix_tree_lookup(&bdi->cgwb_tree, memcg_css->id);
648 if (wb) {
649 struct cgroup_subsys_state *blkcg_css;
650
651 /* see whether the blkcg association has changed */
652 blkcg_css = cgroup_get_e_css(memcg_css->cgroup,
c165b3e3 653 &io_cgrp_subsys);
52ebea74
TH
654 if (unlikely(wb->blkcg_css != blkcg_css ||
655 !wb_tryget(wb)))
656 wb = NULL;
657 css_put(blkcg_css);
658 }
659 rcu_read_unlock();
660 } while (!wb && !cgwb_create(bdi, memcg_css, gfp));
661
662 return wb;
663}
6467716a 664
a13f35e8 665static int cgwb_bdi_init(struct backing_dev_info *bdi)
52ebea74 666{
a13f35e8
TH
667 int ret;
668
52ebea74
TH
669 INIT_RADIX_TREE(&bdi->cgwb_tree, GFP_ATOMIC);
670 bdi->cgwb_congested_tree = RB_ROOT;
671 atomic_set(&bdi->usage_cnt, 1);
a13f35e8
TH
672
673 ret = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
674 if (!ret) {
675 bdi->wb.memcg_css = mem_cgroup_root_css;
676 bdi->wb.blkcg_css = blkcg_root_css;
677 }
678 return ret;
6467716a
AB
679}
680
52ebea74
TH
681static void cgwb_bdi_destroy(struct backing_dev_info *bdi)
682{
683 struct radix_tree_iter iter;
a20135ff 684 struct bdi_writeback_congested *congested, *congested_n;
52ebea74
TH
685 void **slot;
686
687 WARN_ON(test_bit(WB_registered, &bdi->wb.state));
688
689 spin_lock_irq(&cgwb_lock);
a20135ff 690
52ebea74
TH
691 radix_tree_for_each_slot(slot, &bdi->cgwb_tree, &iter, 0)
692 cgwb_kill(*slot);
a20135ff
TH
693
694 rbtree_postorder_for_each_entry_safe(congested, congested_n,
695 &bdi->cgwb_congested_tree, rb_node) {
696 rb_erase(&congested->rb_node, &bdi->cgwb_congested_tree);
697 congested->bdi = NULL; /* mark @congested unlinked */
698 }
699
52ebea74
TH
700 spin_unlock_irq(&cgwb_lock);
701
702 /*
703 * All cgwb's and their congested states must be shutdown and
704 * released before returning. Drain the usage counter to wait for
705 * all cgwb's and cgwb_congested's ever created on @bdi.
706 */
707 atomic_dec(&bdi->usage_cnt);
708 wait_event(cgwb_release_wait, !atomic_read(&bdi->usage_cnt));
709}
710
711/**
712 * wb_memcg_offline - kill all wb's associated with a memcg being offlined
713 * @memcg: memcg being offlined
714 *
715 * Also prevents creation of any new wb's associated with @memcg.
e98be2d5 716 */
52ebea74
TH
717void wb_memcg_offline(struct mem_cgroup *memcg)
718{
719 LIST_HEAD(to_destroy);
720 struct list_head *memcg_cgwb_list = mem_cgroup_cgwb_list(memcg);
721 struct bdi_writeback *wb, *next;
722
723 spin_lock_irq(&cgwb_lock);
724 list_for_each_entry_safe(wb, next, memcg_cgwb_list, memcg_node)
725 cgwb_kill(wb);
726 memcg_cgwb_list->next = NULL; /* prevent new wb's */
727 spin_unlock_irq(&cgwb_lock);
728}
729
730/**
731 * wb_blkcg_offline - kill all wb's associated with a blkcg being offlined
732 * @blkcg: blkcg being offlined
733 *
734 * Also prevents creation of any new wb's associated with @blkcg.
735 */
736void wb_blkcg_offline(struct blkcg *blkcg)
737{
738 LIST_HEAD(to_destroy);
739 struct bdi_writeback *wb, *next;
740
741 spin_lock_irq(&cgwb_lock);
742 list_for_each_entry_safe(wb, next, &blkcg->cgwb_list, blkcg_node)
743 cgwb_kill(wb);
744 blkcg->cgwb_list.next = NULL; /* prevent new wb's */
745 spin_unlock_irq(&cgwb_lock);
746}
747
748#else /* CONFIG_CGROUP_WRITEBACK */
749
a13f35e8
TH
750static int cgwb_bdi_init(struct backing_dev_info *bdi)
751{
752 int err;
753
754 bdi->wb_congested = kzalloc(sizeof(*bdi->wb_congested), GFP_KERNEL);
755 if (!bdi->wb_congested)
756 return -ENOMEM;
757
758 err = wb_init(&bdi->wb, bdi, 1, GFP_KERNEL);
759 if (err) {
760 kfree(bdi->wb_congested);
761 return err;
762 }
763 return 0;
764}
765
52ebea74
TH
766static void cgwb_bdi_destroy(struct backing_dev_info *bdi) { }
767
768#endif /* CONFIG_CGROUP_WRITEBACK */
e98be2d5 769
b2e8fb6e
PZ
770int bdi_init(struct backing_dev_info *bdi)
771{
b817525a
TH
772 int ret;
773
cf0ca9fe
PZ
774 bdi->dev = NULL;
775
189d3c4a 776 bdi->min_ratio = 0;
a42dde04 777 bdi->max_ratio = 100;
eb608e3a 778 bdi->max_prop_frac = FPROP_FRAC_BASE;
66f3b8e2 779 INIT_LIST_HEAD(&bdi->bdi_list);
b817525a 780 INIT_LIST_HEAD(&bdi->wb_list);
cc395d7f 781 init_waitqueue_head(&bdi->wb_waitq);
03ba3782 782
b817525a
TH
783 ret = cgwb_bdi_init(bdi);
784
785 list_add_tail_rcu(&bdi->wb.bdi_node, &bdi->wb_list);
786
787 return ret;
b2e8fb6e
PZ
788}
789EXPORT_SYMBOL(bdi_init);
e98be2d5 790
46100071
TH
791int bdi_register(struct backing_dev_info *bdi, struct device *parent,
792 const char *fmt, ...)
793{
794 va_list args;
795 struct device *dev;
e98be2d5 796
46100071
TH
797 if (bdi->dev) /* The driver needs to use separate queues per device */
798 return 0;
e98be2d5 799
46100071
TH
800 va_start(args, fmt);
801 dev = device_create_vargs(bdi_class, parent, MKDEV(0, 0), bdi, fmt, args);
802 va_end(args);
803 if (IS_ERR(dev))
804 return PTR_ERR(dev);
04fbfdc1 805
46100071 806 bdi->dev = dev;
b2e8fb6e 807
46100071
TH
808 bdi_debug_register(bdi, dev_name(dev));
809 set_bit(WB_registered, &bdi->wb.state);
810
811 spin_lock_bh(&bdi_lock);
812 list_add_tail_rcu(&bdi->bdi_list, &bdi_list);
813 spin_unlock_bh(&bdi_lock);
814
815 trace_writeback_bdi_register(bdi);
816 return 0;
b2e8fb6e 817}
46100071 818EXPORT_SYMBOL(bdi_register);
b2e8fb6e 819
46100071 820int bdi_register_dev(struct backing_dev_info *bdi, dev_t dev)
b2e8fb6e 821{
46100071
TH
822 return bdi_register(bdi, NULL, "%u:%u", MAJOR(dev), MINOR(dev));
823}
824EXPORT_SYMBOL(bdi_register_dev);
825
826/*
827 * Remove bdi from bdi_list, and ensure that it is no longer visible
828 */
829static void bdi_remove_from_list(struct backing_dev_info *bdi)
830{
831 spin_lock_bh(&bdi_lock);
832 list_del_rcu(&bdi->bdi_list);
833 spin_unlock_bh(&bdi_lock);
b2e8fb6e 834
46100071
TH
835 synchronize_rcu_expedited();
836}
cf0ca9fe 837
b02176f3 838void bdi_unregister(struct backing_dev_info *bdi)
b2e8fb6e 839{
f0054bb1
TH
840 /* make sure nobody finds us on the bdi_list anymore */
841 bdi_remove_from_list(bdi);
842 wb_shutdown(&bdi->wb);
52ebea74 843 cgwb_bdi_destroy(bdi);
7a401a97 844
c4db59d3
CH
845 if (bdi->dev) {
846 bdi_debug_unregister(bdi);
847 device_unregister(bdi->dev);
848 bdi->dev = NULL;
849 }
b02176f3 850}
c4db59d3 851
b02176f3
TH
852void bdi_exit(struct backing_dev_info *bdi)
853{
854 WARN_ON_ONCE(bdi->dev);
f0054bb1 855 wb_exit(&bdi->wb);
b2e8fb6e 856}
b02176f3
TH
857
858void bdi_destroy(struct backing_dev_info *bdi)
859{
860 bdi_unregister(bdi);
861 bdi_exit(bdi);
862}
b2e8fb6e
PZ
863EXPORT_SYMBOL(bdi_destroy);
864
c3c53206
JA
865/*
866 * For use from filesystems to quickly init and register a bdi associated
867 * with dirty writeback
868 */
b4caecd4 869int bdi_setup_and_register(struct backing_dev_info *bdi, char *name)
c3c53206 870{
c3c53206
JA
871 int err;
872
873 bdi->name = name;
b4caecd4 874 bdi->capabilities = 0;
c3c53206
JA
875 err = bdi_init(bdi);
876 if (err)
877 return err;
878
02aa2a37
KC
879 err = bdi_register(bdi, NULL, "%.28s-%ld", name,
880 atomic_long_inc_return(&bdi_seq));
c3c53206
JA
881 if (err) {
882 bdi_destroy(bdi);
883 return err;
884 }
885
886 return 0;
887}
888EXPORT_SYMBOL(bdi_setup_and_register);
889
3fcfab16
AM
890static wait_queue_head_t congestion_wqh[2] = {
891 __WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[0]),
892 __WAIT_QUEUE_HEAD_INITIALIZER(congestion_wqh[1])
893 };
ec8a6f26 894static atomic_t nr_wb_congested[2];
3fcfab16 895
ec8a6f26 896void clear_wb_congested(struct bdi_writeback_congested *congested, int sync)
3fcfab16 897{
1faa16d2 898 wait_queue_head_t *wqh = &congestion_wqh[sync];
ec8a6f26 899 enum wb_state bit;
3fcfab16 900
4452226e 901 bit = sync ? WB_sync_congested : WB_async_congested;
ec8a6f26
TH
902 if (test_and_clear_bit(bit, &congested->state))
903 atomic_dec(&nr_wb_congested[sync]);
4e857c58 904 smp_mb__after_atomic();
3fcfab16
AM
905 if (waitqueue_active(wqh))
906 wake_up(wqh);
907}
ec8a6f26 908EXPORT_SYMBOL(clear_wb_congested);
3fcfab16 909
ec8a6f26 910void set_wb_congested(struct bdi_writeback_congested *congested, int sync)
3fcfab16 911{
4452226e 912 enum wb_state bit;
3fcfab16 913
4452226e 914 bit = sync ? WB_sync_congested : WB_async_congested;
ec8a6f26
TH
915 if (!test_and_set_bit(bit, &congested->state))
916 atomic_inc(&nr_wb_congested[sync]);
3fcfab16 917}
ec8a6f26 918EXPORT_SYMBOL(set_wb_congested);
3fcfab16
AM
919
920/**
921 * congestion_wait - wait for a backing_dev to become uncongested
8aa7e847 922 * @sync: SYNC or ASYNC IO
3fcfab16
AM
923 * @timeout: timeout in jiffies
924 *
925 * Waits for up to @timeout jiffies for a backing_dev (any backing_dev) to exit
926 * write congestion. If no backing_devs are congested then just wait for the
927 * next write to be completed.
928 */
8aa7e847 929long congestion_wait(int sync, long timeout)
3fcfab16
AM
930{
931 long ret;
52bb9198 932 unsigned long start = jiffies;
3fcfab16 933 DEFINE_WAIT(wait);
8aa7e847 934 wait_queue_head_t *wqh = &congestion_wqh[sync];
3fcfab16
AM
935
936 prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
937 ret = io_schedule_timeout(timeout);
938 finish_wait(wqh, &wait);
52bb9198
MG
939
940 trace_writeback_congestion_wait(jiffies_to_usecs(timeout),
941 jiffies_to_usecs(jiffies - start));
942
3fcfab16
AM
943 return ret;
944}
945EXPORT_SYMBOL(congestion_wait);
04fbfdc1 946
0e093d99
MG
947/**
948 * wait_iff_congested - Conditionally wait for a backing_dev to become uncongested or a zone to complete writes
949 * @zone: A zone to check if it is heavily congested
950 * @sync: SYNC or ASYNC IO
951 * @timeout: timeout in jiffies
952 *
953 * In the event of a congested backing_dev (any backing_dev) and the given
954 * @zone has experienced recent congestion, this waits for up to @timeout
955 * jiffies for either a BDI to exit congestion of the given @sync queue
956 * or a write to complete.
957 *
25985edc 958 * In the absence of zone congestion, cond_resched() is called to yield
0e093d99
MG
959 * the processor if necessary but otherwise does not sleep.
960 *
961 * The return value is 0 if the sleep is for the full timeout. Otherwise,
962 * it is the number of jiffies that were still remaining when the function
963 * returned. return_value == timeout implies the function did not sleep.
964 */
965long wait_iff_congested(struct zone *zone, int sync, long timeout)
966{
967 long ret;
968 unsigned long start = jiffies;
969 DEFINE_WAIT(wait);
970 wait_queue_head_t *wqh = &congestion_wqh[sync];
971
972 /*
973 * If there is no congestion, or heavy congestion is not being
974 * encountered in the current zone, yield if necessary instead
975 * of sleeping on the congestion queue
976 */
ec8a6f26 977 if (atomic_read(&nr_wb_congested[sync]) == 0 ||
57054651 978 !test_bit(ZONE_CONGESTED, &zone->flags)) {
0e093d99
MG
979 cond_resched();
980
981 /* In case we scheduled, work out time remaining */
982 ret = timeout - (jiffies - start);
983 if (ret < 0)
984 ret = 0;
985
986 goto out;
987 }
988
989 /* Sleep until uncongested or a write happens */
990 prepare_to_wait(wqh, &wait, TASK_UNINTERRUPTIBLE);
991 ret = io_schedule_timeout(timeout);
992 finish_wait(wqh, &wait);
993
994out:
995 trace_writeback_wait_iff_congested(jiffies_to_usecs(timeout),
996 jiffies_to_usecs(jiffies - start));
997
998 return ret;
999}
1000EXPORT_SYMBOL(wait_iff_congested);
3965c9ae
WL
1001
1002int pdflush_proc_obsolete(struct ctl_table *table, int write,
1003 void __user *buffer, size_t *lenp, loff_t *ppos)
1004{
1005 char kbuf[] = "0\n";
1006
4c3bffc2 1007 if (*ppos || *lenp < sizeof(kbuf)) {
3965c9ae
WL
1008 *lenp = 0;
1009 return 0;
1010 }
1011
1012 if (copy_to_user(buffer, kbuf, sizeof(kbuf)))
1013 return -EFAULT;
1014 printk_once(KERN_WARNING "%s exported in /proc is scheduled for removal\n",
1015 table->procname);
1016
1017 *lenp = 2;
1018 *ppos += *lenp;
1019 return 2;
1020}