blk-cgroup.h 22 KB

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