blkcg: delete unused APIs
[linux-block.git] / include / linux / blk-cgroup.h
CommitLineData
31e4c28d
VG
1#ifndef _BLK_CGROUP_H
2#define _BLK_CGROUP_H
3/*
4 * Common Block IO controller cgroup interface
5 *
6 * Based on ideas and code from CFQ, CFS and BFQ:
7 * Copyright (C) 2003 Jens Axboe <axboe@kernel.dk>
8 *
9 * Copyright (C) 2008 Fabio Checconi <fabio@gandalf.sssup.it>
10 * Paolo Valente <paolo.valente@unimore.it>
11 *
12 * Copyright (C) 2009 Vivek Goyal <vgoyal@redhat.com>
13 * Nauman Rafique <nauman@google.com>
14 */
15
16#include <linux/cgroup.h>
24bdb8ef 17#include <linux/percpu_counter.h>
829fdb50 18#include <linux/seq_file.h>
a637120e 19#include <linux/radix-tree.h>
a051661c 20#include <linux/blkdev.h>
a5049a8a 21#include <linux/atomic.h>
31e4c28d 22
24bdb8ef
TH
23/* percpu_counter batch for blkg_[rw]stats, per-cpu drift doesn't matter */
24#define BLKG_STAT_CPU_BATCH (INT_MAX / 2)
25
9355aede
VG
26/* Max limits for throttle policy */
27#define THROTL_IOPS_MAX UINT_MAX
28
f48ec1d7
TH
29#ifdef CONFIG_BLK_CGROUP
30
edcb0722
TH
31enum blkg_rwstat_type {
32 BLKG_RWSTAT_READ,
33 BLKG_RWSTAT_WRITE,
34 BLKG_RWSTAT_SYNC,
35 BLKG_RWSTAT_ASYNC,
36
37 BLKG_RWSTAT_NR,
38 BLKG_RWSTAT_TOTAL = BLKG_RWSTAT_NR,
303a3acb
DS
39};
40
a637120e
TH
41struct blkcg_gq;
42
3c798398 43struct blkcg {
36558c8a
TH
44 struct cgroup_subsys_state css;
45 spinlock_t lock;
a637120e
TH
46
47 struct radix_tree_root blkg_tree;
55679c8d 48 struct blkcg_gq __rcu *blkg_hint;
36558c8a 49 struct hlist_head blkg_list;
9a9e8a26 50
81437648 51 struct blkcg_policy_data *cpd[BLKCG_MAX_POLS];
52ebea74 52
7876f930 53 struct list_head all_blkcgs_node;
52ebea74
TH
54#ifdef CONFIG_CGROUP_WRITEBACK
55 struct list_head cgwb_list;
56#endif
31e4c28d
VG
57};
58
e6269c44
TH
59/*
60 * blkg_[rw]stat->aux_cnt is excluded for local stats but included for
24bdb8ef
TH
61 * recursive. Used to carry stats of dead children, and, for blkg_rwstat,
62 * to carry result values from read and sum operations.
e6269c44 63 */
edcb0722 64struct blkg_stat {
24bdb8ef 65 struct percpu_counter cpu_cnt;
e6269c44 66 atomic64_t aux_cnt;
edcb0722
TH
67};
68
69struct blkg_rwstat {
24bdb8ef 70 struct percpu_counter cpu_cnt[BLKG_RWSTAT_NR];
e6269c44 71 atomic64_t aux_cnt[BLKG_RWSTAT_NR];
edcb0722
TH
72};
73
f95a04af
TH
74/*
75 * A blkcg_gq (blkg) is association between a block cgroup (blkcg) and a
76 * request_queue (q). This is used by blkcg policies which need to track
77 * information per blkcg - q pair.
78 *
001bea73
TH
79 * There can be multiple active blkcg policies and each blkg:policy pair is
80 * represented by a blkg_policy_data which is allocated and freed by each
81 * policy's pd_alloc/free_fn() methods. A policy can allocate private data
82 * area by allocating larger data structure which embeds blkg_policy_data
83 * at the beginning.
f95a04af 84 */
0381411e 85struct blkg_policy_data {
b276a876 86 /* the blkg and policy id this per-policy data belongs to */
3c798398 87 struct blkcg_gq *blkg;
b276a876 88 int plid;
0381411e
TH
89};
90
e48453c3 91/*
e4a9bde9
TH
92 * Policies that need to keep per-blkcg data which is independent from any
93 * request_queue associated to it should implement cpd_alloc/free_fn()
94 * methods. A policy can allocate private data area by allocating larger
95 * data structure which embeds blkcg_policy_data at the beginning.
96 * cpd_init() is invoked to let each policy handle per-blkcg data.
e48453c3
AA
97 */
98struct blkcg_policy_data {
81437648
TH
99 /* the blkcg and policy id this per-policy data belongs to */
100 struct blkcg *blkcg;
e48453c3 101 int plid;
e48453c3
AA
102};
103
3c798398
TH
104/* association between a blk cgroup and a request queue */
105struct blkcg_gq {
c875f4d0 106 /* Pointer to the associated request_queue */
36558c8a
TH
107 struct request_queue *q;
108 struct list_head q_node;
109 struct hlist_node blkcg_node;
3c798398 110 struct blkcg *blkcg;
3c547865 111
ce7acfea
TH
112 /*
113 * Each blkg gets congested separately and the congestion state is
114 * propagated to the matching bdi_writeback_congested.
115 */
116 struct bdi_writeback_congested *wb_congested;
117
3c547865
TH
118 /* all non-root blkcg_gq's are guaranteed to have access to parent */
119 struct blkcg_gq *parent;
120
a051661c
TH
121 /* request allocation list for this blkcg-q pair */
122 struct request_list rl;
3c547865 123
1adaf3dd 124 /* reference count */
a5049a8a 125 atomic_t refcnt;
22084190 126
f427d909
TH
127 /* is this blkg online? protected by both blkcg and q locks */
128 bool online;
129
77ea7338
TH
130 struct blkg_rwstat stat_bytes;
131 struct blkg_rwstat stat_ios;
132
36558c8a 133 struct blkg_policy_data *pd[BLKCG_MAX_POLS];
1adaf3dd 134
36558c8a 135 struct rcu_head rcu_head;
31e4c28d
VG
136};
137
e4a9bde9 138typedef struct blkcg_policy_data *(blkcg_pol_alloc_cpd_fn)(gfp_t gfp);
81437648 139typedef void (blkcg_pol_init_cpd_fn)(struct blkcg_policy_data *cpd);
e4a9bde9 140typedef void (blkcg_pol_free_cpd_fn)(struct blkcg_policy_data *cpd);
69d7fde5 141typedef void (blkcg_pol_bind_cpd_fn)(struct blkcg_policy_data *cpd);
001bea73 142typedef struct blkg_policy_data *(blkcg_pol_alloc_pd_fn)(gfp_t gfp, int node);
a9520cd6
TH
143typedef void (blkcg_pol_init_pd_fn)(struct blkg_policy_data *pd);
144typedef void (blkcg_pol_online_pd_fn)(struct blkg_policy_data *pd);
145typedef void (blkcg_pol_offline_pd_fn)(struct blkg_policy_data *pd);
001bea73 146typedef void (blkcg_pol_free_pd_fn)(struct blkg_policy_data *pd);
a9520cd6 147typedef void (blkcg_pol_reset_pd_stats_fn)(struct blkg_policy_data *pd);
3e252066 148
3c798398 149struct blkcg_policy {
36558c8a 150 int plid;
36558c8a 151 /* cgroup files for the policy */
2ee867dc 152 struct cftype *dfl_cftypes;
880f50e2 153 struct cftype *legacy_cftypes;
f9fcc2d3
TH
154
155 /* operations */
e4a9bde9 156 blkcg_pol_alloc_cpd_fn *cpd_alloc_fn;
e48453c3 157 blkcg_pol_init_cpd_fn *cpd_init_fn;
e4a9bde9 158 blkcg_pol_free_cpd_fn *cpd_free_fn;
69d7fde5 159 blkcg_pol_bind_cpd_fn *cpd_bind_fn;
e4a9bde9 160
001bea73 161 blkcg_pol_alloc_pd_fn *pd_alloc_fn;
f9fcc2d3 162 blkcg_pol_init_pd_fn *pd_init_fn;
f427d909
TH
163 blkcg_pol_online_pd_fn *pd_online_fn;
164 blkcg_pol_offline_pd_fn *pd_offline_fn;
001bea73 165 blkcg_pol_free_pd_fn *pd_free_fn;
f9fcc2d3 166 blkcg_pol_reset_pd_stats_fn *pd_reset_stats_fn;
3e252066
VG
167};
168
3c798398 169extern struct blkcg blkcg_root;
496d5e75 170extern struct cgroup_subsys_state * const blkcg_root_css;
36558c8a 171
24f29046
TH
172struct blkcg_gq *blkg_lookup_slowpath(struct blkcg *blkcg,
173 struct request_queue *q, bool update_hint);
3c798398
TH
174struct blkcg_gq *blkg_lookup_create(struct blkcg *blkcg,
175 struct request_queue *q);
36558c8a
TH
176int blkcg_init_queue(struct request_queue *q);
177void blkcg_drain_queue(struct request_queue *q);
178void blkcg_exit_queue(struct request_queue *q);
5efd6113 179
3e252066 180/* Blkio controller policy registration */
d5bf0291 181int blkcg_policy_register(struct blkcg_policy *pol);
3c798398 182void blkcg_policy_unregister(struct blkcg_policy *pol);
36558c8a 183int blkcg_activate_policy(struct request_queue *q,
3c798398 184 const struct blkcg_policy *pol);
36558c8a 185void blkcg_deactivate_policy(struct request_queue *q,
3c798398 186 const struct blkcg_policy *pol);
3e252066 187
dd165eb3 188const char *blkg_dev_name(struct blkcg_gq *blkg);
3c798398 189void blkcg_print_blkgs(struct seq_file *sf, struct blkcg *blkcg,
f95a04af
TH
190 u64 (*prfill)(struct seq_file *,
191 struct blkg_policy_data *, int),
3c798398 192 const struct blkcg_policy *pol, int data,
ec399347 193 bool show_total);
f95a04af
TH
194u64 __blkg_prfill_u64(struct seq_file *sf, struct blkg_policy_data *pd, u64 v);
195u64 __blkg_prfill_rwstat(struct seq_file *sf, struct blkg_policy_data *pd,
829fdb50 196 const struct blkg_rwstat *rwstat);
f95a04af
TH
197u64 blkg_prfill_stat(struct seq_file *sf, struct blkg_policy_data *pd, int off);
198u64 blkg_prfill_rwstat(struct seq_file *sf, struct blkg_policy_data *pd,
199 int off);
77ea7338
TH
200int blkg_print_stat_bytes(struct seq_file *sf, void *v);
201int blkg_print_stat_ios(struct seq_file *sf, void *v);
202int blkg_print_stat_bytes_recursive(struct seq_file *sf, void *v);
203int blkg_print_stat_ios_recursive(struct seq_file *sf, void *v);
829fdb50 204
f12c74ca
TH
205u64 blkg_stat_recursive_sum(struct blkcg_gq *blkg,
206 struct blkcg_policy *pol, int off);
207struct blkg_rwstat blkg_rwstat_recursive_sum(struct blkcg_gq *blkg,
208 struct blkcg_policy *pol, int off);
16b3de66 209
829fdb50 210struct blkg_conf_ctx {
36558c8a 211 struct gendisk *disk;
3c798398 212 struct blkcg_gq *blkg;
36aa9e5f 213 char *body;
829fdb50
TH
214};
215
3c798398 216int blkg_conf_prep(struct blkcg *blkcg, const struct blkcg_policy *pol,
36aa9e5f 217 char *input, struct blkg_conf_ctx *ctx);
829fdb50
TH
218void blkg_conf_finish(struct blkg_conf_ctx *ctx);
219
220
a7c6d554
TH
221static inline struct blkcg *css_to_blkcg(struct cgroup_subsys_state *css)
222{
223 return css ? container_of(css, struct blkcg, css) : NULL;
224}
225
b1208b56
TH
226static inline struct blkcg *task_blkcg(struct task_struct *tsk)
227{
c165b3e3 228 return css_to_blkcg(task_css(tsk, io_cgrp_id));
b1208b56
TH
229}
230
231static inline struct blkcg *bio_blkcg(struct bio *bio)
232{
233 if (bio && bio->bi_css)
a7c6d554 234 return css_to_blkcg(bio->bi_css);
b1208b56
TH
235 return task_blkcg(current);
236}
237
3c547865
TH
238/**
239 * blkcg_parent - get the parent of a blkcg
240 * @blkcg: blkcg of interest
241 *
242 * Return the parent blkcg of @blkcg. Can be called anytime.
243 */
244static inline struct blkcg *blkcg_parent(struct blkcg *blkcg)
245{
5c9d535b 246 return css_to_blkcg(blkcg->css.parent);
3c547865
TH
247}
248
24f29046
TH
249/**
250 * __blkg_lookup - internal version of blkg_lookup()
251 * @blkcg: blkcg of interest
252 * @q: request_queue of interest
253 * @update_hint: whether to update lookup hint with the result or not
254 *
255 * This is internal version and shouldn't be used by policy
256 * implementations. Looks up blkgs for the @blkcg - @q pair regardless of
257 * @q's bypass state. If @update_hint is %true, the caller should be
258 * holding @q->queue_lock and lookup hint is updated on success.
259 */
260static inline struct blkcg_gq *__blkg_lookup(struct blkcg *blkcg,
261 struct request_queue *q,
262 bool update_hint)
263{
264 struct blkcg_gq *blkg;
265
85b6bc9d
TH
266 if (blkcg == &blkcg_root)
267 return q->root_blkg;
268
24f29046
TH
269 blkg = rcu_dereference(blkcg->blkg_hint);
270 if (blkg && blkg->q == q)
271 return blkg;
272
273 return blkg_lookup_slowpath(blkcg, q, update_hint);
274}
275
276/**
277 * blkg_lookup - lookup blkg for the specified blkcg - q pair
278 * @blkcg: blkcg of interest
279 * @q: request_queue of interest
280 *
281 * Lookup blkg for the @blkcg - @q pair. This function should be called
282 * under RCU read lock and is guaranteed to return %NULL if @q is bypassing
283 * - see blk_queue_bypass_start() for details.
284 */
285static inline struct blkcg_gq *blkg_lookup(struct blkcg *blkcg,
286 struct request_queue *q)
287{
288 WARN_ON_ONCE(!rcu_read_lock_held());
289
290 if (unlikely(blk_queue_bypass(q)))
291 return NULL;
292 return __blkg_lookup(blkcg, q, false);
293}
294
0381411e
TH
295/**
296 * blkg_to_pdata - get policy private data
297 * @blkg: blkg of interest
298 * @pol: policy of interest
299 *
300 * Return pointer to private data associated with the @blkg-@pol pair.
301 */
f95a04af
TH
302static inline struct blkg_policy_data *blkg_to_pd(struct blkcg_gq *blkg,
303 struct blkcg_policy *pol)
0381411e 304{
f95a04af 305 return blkg ? blkg->pd[pol->plid] : NULL;
0381411e
TH
306}
307
e48453c3
AA
308static inline struct blkcg_policy_data *blkcg_to_cpd(struct blkcg *blkcg,
309 struct blkcg_policy *pol)
310{
81437648 311 return blkcg ? blkcg->cpd[pol->plid] : NULL;
e48453c3
AA
312}
313
0381411e
TH
314/**
315 * pdata_to_blkg - get blkg associated with policy private data
f95a04af 316 * @pd: policy private data of interest
0381411e 317 *
f95a04af 318 * @pd is policy private data. Determine the blkg it's associated with.
0381411e 319 */
f95a04af 320static inline struct blkcg_gq *pd_to_blkg(struct blkg_policy_data *pd)
0381411e 321{
f95a04af 322 return pd ? pd->blkg : NULL;
0381411e
TH
323}
324
81437648
TH
325static inline struct blkcg *cpd_to_blkcg(struct blkcg_policy_data *cpd)
326{
327 return cpd ? cpd->blkcg : NULL;
328}
329
54e7ed12
TH
330/**
331 * blkg_path - format cgroup path of blkg
332 * @blkg: blkg of interest
333 * @buf: target buffer
334 * @buflen: target buffer length
335 *
336 * Format the path of the cgroup of @blkg into @buf.
337 */
3c798398 338static inline int blkg_path(struct blkcg_gq *blkg, char *buf, int buflen)
afc24d49 339{
4c737b41 340 return cgroup_path(blkg->blkcg->css.cgroup, buf, buflen);
afc24d49
VG
341}
342
1adaf3dd
TH
343/**
344 * blkg_get - get a blkg reference
345 * @blkg: blkg to get
346 *
a5049a8a 347 * The caller should be holding an existing reference.
1adaf3dd 348 */
3c798398 349static inline void blkg_get(struct blkcg_gq *blkg)
1adaf3dd 350{
a5049a8a
TH
351 WARN_ON_ONCE(atomic_read(&blkg->refcnt) <= 0);
352 atomic_inc(&blkg->refcnt);
1adaf3dd
TH
353}
354
2a4fd070 355void __blkg_release_rcu(struct rcu_head *rcu);
1adaf3dd
TH
356
357/**
358 * blkg_put - put a blkg reference
359 * @blkg: blkg to put
1adaf3dd 360 */
3c798398 361static inline void blkg_put(struct blkcg_gq *blkg)
1adaf3dd 362{
a5049a8a
TH
363 WARN_ON_ONCE(atomic_read(&blkg->refcnt) <= 0);
364 if (atomic_dec_and_test(&blkg->refcnt))
2a4fd070 365 call_rcu(&blkg->rcu_head, __blkg_release_rcu);
1adaf3dd
TH
366}
367
dd4a4ffc
TH
368/**
369 * blkg_for_each_descendant_pre - pre-order walk of a blkg's descendants
370 * @d_blkg: loop cursor pointing to the current descendant
492eb21b 371 * @pos_css: used for iteration
dd4a4ffc
TH
372 * @p_blkg: target blkg to walk descendants of
373 *
374 * Walk @c_blkg through the descendants of @p_blkg. Must be used with RCU
375 * read locked. If called under either blkcg or queue lock, the iteration
376 * is guaranteed to include all and only online blkgs. The caller may
492eb21b 377 * update @pos_css by calling css_rightmost_descendant() to skip subtree.
bd8815a6 378 * @p_blkg is included in the iteration and the first node to be visited.
dd4a4ffc 379 */
492eb21b
TH
380#define blkg_for_each_descendant_pre(d_blkg, pos_css, p_blkg) \
381 css_for_each_descendant_pre((pos_css), &(p_blkg)->blkcg->css) \
382 if (((d_blkg) = __blkg_lookup(css_to_blkcg(pos_css), \
dd4a4ffc
TH
383 (p_blkg)->q, false)))
384
aa539cb3
TH
385/**
386 * blkg_for_each_descendant_post - post-order walk of a blkg's descendants
387 * @d_blkg: loop cursor pointing to the current descendant
492eb21b 388 * @pos_css: used for iteration
aa539cb3
TH
389 * @p_blkg: target blkg to walk descendants of
390 *
391 * Similar to blkg_for_each_descendant_pre() but performs post-order
bd8815a6
TH
392 * traversal instead. Synchronization rules are the same. @p_blkg is
393 * included in the iteration and the last node to be visited.
aa539cb3 394 */
492eb21b
TH
395#define blkg_for_each_descendant_post(d_blkg, pos_css, p_blkg) \
396 css_for_each_descendant_post((pos_css), &(p_blkg)->blkcg->css) \
397 if (((d_blkg) = __blkg_lookup(css_to_blkcg(pos_css), \
aa539cb3
TH
398 (p_blkg)->q, false)))
399
a051661c
TH
400/**
401 * blk_get_rl - get request_list to use
402 * @q: request_queue of interest
403 * @bio: bio which will be attached to the allocated request (may be %NULL)
404 *
405 * The caller wants to allocate a request from @q to use for @bio. Find
406 * the request_list to use and obtain a reference on it. Should be called
407 * under queue_lock. This function is guaranteed to return non-%NULL
408 * request_list.
409 */
410static inline struct request_list *blk_get_rl(struct request_queue *q,
411 struct bio *bio)
412{
413 struct blkcg *blkcg;
414 struct blkcg_gq *blkg;
415
416 rcu_read_lock();
417
418 blkcg = bio_blkcg(bio);
419
420 /* bypass blkg lookup and use @q->root_rl directly for root */
421 if (blkcg == &blkcg_root)
422 goto root_rl;
423
424 /*
425 * Try to use blkg->rl. blkg lookup may fail under memory pressure
426 * or if either the blkcg or queue is going away. Fall back to
427 * root_rl in such cases.
428 */
ae118896
TH
429 blkg = blkg_lookup(blkcg, q);
430 if (unlikely(!blkg))
a051661c
TH
431 goto root_rl;
432
433 blkg_get(blkg);
434 rcu_read_unlock();
435 return &blkg->rl;
436root_rl:
437 rcu_read_unlock();
438 return &q->root_rl;
439}
440
441/**
442 * blk_put_rl - put request_list
443 * @rl: request_list to put
444 *
445 * Put the reference acquired by blk_get_rl(). Should be called under
446 * queue_lock.
447 */
448static inline void blk_put_rl(struct request_list *rl)
449{
401efbf8 450 if (rl->blkg->blkcg != &blkcg_root)
a051661c
TH
451 blkg_put(rl->blkg);
452}
453
454/**
455 * blk_rq_set_rl - associate a request with a request_list
456 * @rq: request of interest
457 * @rl: target request_list
458 *
459 * Associate @rq with @rl so that accounting and freeing can know the
460 * request_list @rq came from.
461 */
462static inline void blk_rq_set_rl(struct request *rq, struct request_list *rl)
463{
464 rq->rl = rl;
465}
466
467/**
468 * blk_rq_rl - return the request_list a request came from
469 * @rq: request of interest
470 *
471 * Return the request_list @rq is allocated from.
472 */
473static inline struct request_list *blk_rq_rl(struct request *rq)
474{
475 return rq->rl;
476}
477
478struct request_list *__blk_queue_next_rl(struct request_list *rl,
479 struct request_queue *q);
480/**
481 * blk_queue_for_each_rl - iterate through all request_lists of a request_queue
482 *
483 * Should be used under queue_lock.
484 */
485#define blk_queue_for_each_rl(rl, q) \
486 for ((rl) = &(q)->root_rl; (rl); (rl) = __blk_queue_next_rl((rl), (q)))
487
24bdb8ef 488static inline int blkg_stat_init(struct blkg_stat *stat, gfp_t gfp)
90d3839b 489{
24bdb8ef
TH
490 int ret;
491
492 ret = percpu_counter_init(&stat->cpu_cnt, 0, gfp);
493 if (ret)
494 return ret;
495
e6269c44 496 atomic64_set(&stat->aux_cnt, 0);
24bdb8ef
TH
497 return 0;
498}
499
500static inline void blkg_stat_exit(struct blkg_stat *stat)
501{
502 percpu_counter_destroy(&stat->cpu_cnt);
90d3839b
PZ
503}
504
edcb0722
TH
505/**
506 * blkg_stat_add - add a value to a blkg_stat
507 * @stat: target blkg_stat
508 * @val: value to add
509 *
24bdb8ef
TH
510 * Add @val to @stat. The caller must ensure that IRQ on the same CPU
511 * don't re-enter this function for the same counter.
edcb0722
TH
512 */
513static inline void blkg_stat_add(struct blkg_stat *stat, uint64_t val)
514{
104b4e51 515 percpu_counter_add_batch(&stat->cpu_cnt, val, BLKG_STAT_CPU_BATCH);
edcb0722
TH
516}
517
518/**
519 * blkg_stat_read - read the current value of a blkg_stat
520 * @stat: blkg_stat to read
edcb0722
TH
521 */
522static inline uint64_t blkg_stat_read(struct blkg_stat *stat)
523{
24bdb8ef 524 return percpu_counter_sum_positive(&stat->cpu_cnt);
edcb0722
TH
525}
526
527/**
528 * blkg_stat_reset - reset a blkg_stat
529 * @stat: blkg_stat to reset
530 */
531static inline void blkg_stat_reset(struct blkg_stat *stat)
532{
24bdb8ef 533 percpu_counter_set(&stat->cpu_cnt, 0);
e6269c44 534 atomic64_set(&stat->aux_cnt, 0);
edcb0722
TH
535}
536
16b3de66 537/**
e6269c44 538 * blkg_stat_add_aux - add a blkg_stat into another's aux count
16b3de66
TH
539 * @to: the destination blkg_stat
540 * @from: the source
541 *
e6269c44 542 * Add @from's count including the aux one to @to's aux count.
16b3de66 543 */
e6269c44
TH
544static inline void blkg_stat_add_aux(struct blkg_stat *to,
545 struct blkg_stat *from)
16b3de66 546{
e6269c44
TH
547 atomic64_add(blkg_stat_read(from) + atomic64_read(&from->aux_cnt),
548 &to->aux_cnt);
16b3de66
TH
549}
550
24bdb8ef 551static inline int blkg_rwstat_init(struct blkg_rwstat *rwstat, gfp_t gfp)
90d3839b 552{
24bdb8ef
TH
553 int i, ret;
554
555 for (i = 0; i < BLKG_RWSTAT_NR; i++) {
556 ret = percpu_counter_init(&rwstat->cpu_cnt[i], 0, gfp);
557 if (ret) {
558 while (--i >= 0)
559 percpu_counter_destroy(&rwstat->cpu_cnt[i]);
560 return ret;
561 }
562 atomic64_set(&rwstat->aux_cnt[i], 0);
563 }
564 return 0;
565}
e6269c44 566
24bdb8ef
TH
567static inline void blkg_rwstat_exit(struct blkg_rwstat *rwstat)
568{
569 int i;
e6269c44
TH
570
571 for (i = 0; i < BLKG_RWSTAT_NR; i++)
24bdb8ef 572 percpu_counter_destroy(&rwstat->cpu_cnt[i]);
90d3839b
PZ
573}
574
edcb0722
TH
575/**
576 * blkg_rwstat_add - add a value to a blkg_rwstat
577 * @rwstat: target blkg_rwstat
ef295ecf 578 * @op: REQ_OP and flags
edcb0722
TH
579 * @val: value to add
580 *
581 * Add @val to @rwstat. The counters are chosen according to @rw. The
582 * caller is responsible for synchronizing calls to this function.
583 */
584static inline void blkg_rwstat_add(struct blkg_rwstat *rwstat,
ef295ecf 585 unsigned int op, uint64_t val)
edcb0722 586{
24bdb8ef 587 struct percpu_counter *cnt;
edcb0722 588
63a4cc24 589 if (op_is_write(op))
24bdb8ef 590 cnt = &rwstat->cpu_cnt[BLKG_RWSTAT_WRITE];
edcb0722 591 else
24bdb8ef
TH
592 cnt = &rwstat->cpu_cnt[BLKG_RWSTAT_READ];
593
104b4e51 594 percpu_counter_add_batch(cnt, val, BLKG_STAT_CPU_BATCH);
24bdb8ef 595
d71d9ae1 596 if (op_is_sync(op))
24bdb8ef 597 cnt = &rwstat->cpu_cnt[BLKG_RWSTAT_SYNC];
edcb0722 598 else
24bdb8ef 599 cnt = &rwstat->cpu_cnt[BLKG_RWSTAT_ASYNC];
edcb0722 600
104b4e51 601 percpu_counter_add_batch(cnt, val, BLKG_STAT_CPU_BATCH);
edcb0722
TH
602}
603
604/**
605 * blkg_rwstat_read - read the current values of a blkg_rwstat
606 * @rwstat: blkg_rwstat to read
607 *
24bdb8ef 608 * Read the current snapshot of @rwstat and return it in the aux counts.
edcb0722 609 */
c94bed89 610static inline struct blkg_rwstat blkg_rwstat_read(struct blkg_rwstat *rwstat)
edcb0722 611{
24bdb8ef
TH
612 struct blkg_rwstat result;
613 int i;
edcb0722 614
24bdb8ef
TH
615 for (i = 0; i < BLKG_RWSTAT_NR; i++)
616 atomic64_set(&result.aux_cnt[i],
617 percpu_counter_sum_positive(&rwstat->cpu_cnt[i]));
618 return result;
edcb0722
TH
619}
620
621/**
4d5e80a7 622 * blkg_rwstat_total - read the total count of a blkg_rwstat
edcb0722
TH
623 * @rwstat: blkg_rwstat to read
624 *
625 * Return the total count of @rwstat regardless of the IO direction. This
626 * function can be called without synchronization and takes care of u64
627 * atomicity.
628 */
4d5e80a7 629static inline uint64_t blkg_rwstat_total(struct blkg_rwstat *rwstat)
edcb0722
TH
630{
631 struct blkg_rwstat tmp = blkg_rwstat_read(rwstat);
632
24bdb8ef
TH
633 return atomic64_read(&tmp.aux_cnt[BLKG_RWSTAT_READ]) +
634 atomic64_read(&tmp.aux_cnt[BLKG_RWSTAT_WRITE]);
edcb0722
TH
635}
636
637/**
638 * blkg_rwstat_reset - reset a blkg_rwstat
639 * @rwstat: blkg_rwstat to reset
640 */
641static inline void blkg_rwstat_reset(struct blkg_rwstat *rwstat)
642{
e6269c44
TH
643 int i;
644
24bdb8ef
TH
645 for (i = 0; i < BLKG_RWSTAT_NR; i++) {
646 percpu_counter_set(&rwstat->cpu_cnt[i], 0);
e6269c44 647 atomic64_set(&rwstat->aux_cnt[i], 0);
24bdb8ef 648 }
edcb0722
TH
649}
650
16b3de66 651/**
e6269c44 652 * blkg_rwstat_add_aux - add a blkg_rwstat into another's aux count
16b3de66
TH
653 * @to: the destination blkg_rwstat
654 * @from: the source
655 *
e6269c44 656 * Add @from's count including the aux one to @to's aux count.
16b3de66 657 */
e6269c44
TH
658static inline void blkg_rwstat_add_aux(struct blkg_rwstat *to,
659 struct blkg_rwstat *from)
16b3de66
TH
660{
661 struct blkg_rwstat v = blkg_rwstat_read(from);
662 int i;
663
16b3de66 664 for (i = 0; i < BLKG_RWSTAT_NR; i++)
24bdb8ef
TH
665 atomic64_add(atomic64_read(&v.aux_cnt[i]) +
666 atomic64_read(&from->aux_cnt[i]),
e6269c44 667 &to->aux_cnt[i]);
16b3de66
TH
668}
669
ae118896
TH
670#ifdef CONFIG_BLK_DEV_THROTTLING
671extern bool blk_throtl_bio(struct request_queue *q, struct blkcg_gq *blkg,
672 struct bio *bio);
673#else
674static inline bool blk_throtl_bio(struct request_queue *q, struct blkcg_gq *blkg,
675 struct bio *bio) { return false; }
676#endif
677
678static inline bool blkcg_bio_issue_check(struct request_queue *q,
679 struct bio *bio)
680{
681 struct blkcg *blkcg;
682 struct blkcg_gq *blkg;
683 bool throtl = false;
684
685 rcu_read_lock();
686 blkcg = bio_blkcg(bio);
687
007cc56b
SL
688 /* associate blkcg if bio hasn't attached one */
689 bio_associate_blkcg(bio, &blkcg->css);
690
ae118896
TH
691 blkg = blkg_lookup(blkcg, q);
692 if (unlikely(!blkg)) {
693 spin_lock_irq(q->queue_lock);
694 blkg = blkg_lookup_create(blkcg, q);
695 if (IS_ERR(blkg))
696 blkg = NULL;
697 spin_unlock_irq(q->queue_lock);
698 }
699
700 throtl = blk_throtl_bio(q, blkg, bio);
701
77ea7338
TH
702 if (!throtl) {
703 blkg = blkg ?: q->root_blkg;
ef295ecf 704 blkg_rwstat_add(&blkg->stat_bytes, bio->bi_opf,
77ea7338 705 bio->bi_iter.bi_size);
ef295ecf 706 blkg_rwstat_add(&blkg->stat_ios, bio->bi_opf, 1);
77ea7338
TH
707 }
708
ae118896
TH
709 rcu_read_unlock();
710 return !throtl;
711}
712
36558c8a
TH
713#else /* CONFIG_BLK_CGROUP */
714
efa7d1c7
TH
715struct blkcg {
716};
2f5ea477 717
f95a04af
TH
718struct blkg_policy_data {
719};
720
e48453c3
AA
721struct blkcg_policy_data {
722};
723
3c798398 724struct blkcg_gq {
2f5ea477
JA
725};
726
3c798398 727struct blkcg_policy {
3e252066
VG
728};
729
496d5e75
TH
730#define blkcg_root_css ((struct cgroup_subsys_state *)ERR_PTR(-EINVAL))
731
efa7d1c7
TH
732#ifdef CONFIG_BLOCK
733
3c798398 734static inline struct blkcg_gq *blkg_lookup(struct blkcg *blkcg, void *key) { return NULL; }
5efd6113
TH
735static inline int blkcg_init_queue(struct request_queue *q) { return 0; }
736static inline void blkcg_drain_queue(struct request_queue *q) { }
737static inline void blkcg_exit_queue(struct request_queue *q) { }
d5bf0291 738static inline int blkcg_policy_register(struct blkcg_policy *pol) { return 0; }
3c798398 739static inline void blkcg_policy_unregister(struct blkcg_policy *pol) { }
a2b1693b 740static inline int blkcg_activate_policy(struct request_queue *q,
3c798398 741 const struct blkcg_policy *pol) { return 0; }
a2b1693b 742static inline void blkcg_deactivate_policy(struct request_queue *q,
3c798398
TH
743 const struct blkcg_policy *pol) { }
744
b1208b56 745static inline struct blkcg *bio_blkcg(struct bio *bio) { return NULL; }
a051661c 746
f95a04af
TH
747static inline struct blkg_policy_data *blkg_to_pd(struct blkcg_gq *blkg,
748 struct blkcg_policy *pol) { return NULL; }
749static inline struct blkcg_gq *pd_to_blkg(struct blkg_policy_data *pd) { return NULL; }
3c798398
TH
750static inline char *blkg_path(struct blkcg_gq *blkg) { return NULL; }
751static inline void blkg_get(struct blkcg_gq *blkg) { }
752static inline void blkg_put(struct blkcg_gq *blkg) { }
afc24d49 753
a051661c
TH
754static inline struct request_list *blk_get_rl(struct request_queue *q,
755 struct bio *bio) { return &q->root_rl; }
756static inline void blk_put_rl(struct request_list *rl) { }
757static inline void blk_rq_set_rl(struct request *rq, struct request_list *rl) { }
758static inline struct request_list *blk_rq_rl(struct request *rq) { return &rq->q->root_rl; }
759
ae118896
TH
760static inline bool blkcg_bio_issue_check(struct request_queue *q,
761 struct bio *bio) { return true; }
762
a051661c
TH
763#define blk_queue_for_each_rl(rl, q) \
764 for ((rl) = &(q)->root_rl; (rl); (rl) = NULL)
765
efa7d1c7 766#endif /* CONFIG_BLOCK */
36558c8a
TH
767#endif /* CONFIG_BLK_CGROUP */
768#endif /* _BLK_CGROUP_H */