Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1 | /* |
| 2 | * Common Block IO controller cgroup interface |
| 3 | * |
| 4 | * Based on ideas and code from CFQ, CFS and BFQ: |
| 5 | * Copyright (C) 2003 Jens Axboe <axboe@kernel.dk> |
| 6 | * |
| 7 | * Copyright (C) 2008 Fabio Checconi <fabio@gandalf.sssup.it> |
| 8 | * Paolo Valente <paolo.valente@unimore.it> |
| 9 | * |
| 10 | * Copyright (C) 2009 Vivek Goyal <vgoyal@redhat.com> |
| 11 | * Nauman Rafique <nauman@google.com> |
| 12 | */ |
| 13 | #include <linux/ioprio.h> |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 14 | #include <linux/seq_file.h> |
| 15 | #include <linux/kdev_t.h> |
Vivek Goyal | 9d6a986 | 2009-12-04 10:36:41 -0500 | [diff] [blame] | 16 | #include <linux/module.h> |
Stephen Rothwell | accee78 | 2009-12-07 19:29:39 +1100 | [diff] [blame] | 17 | #include <linux/err.h> |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 18 | #include <linux/blkdev.h> |
Tejun Heo | 5a0e3ad | 2010-03-24 17:04:11 +0900 | [diff] [blame] | 19 | #include <linux/slab.h> |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 20 | #include <linux/genhd.h> |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 21 | #include <linux/delay.h> |
Tejun Heo | 9a9e8a2 | 2012-03-19 15:10:56 -0700 | [diff] [blame] | 22 | #include <linux/atomic.h> |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 23 | #include "blk-cgroup.h" |
Tejun Heo | 5efd611 | 2012-03-05 13:15:12 -0800 | [diff] [blame] | 24 | #include "blk.h" |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 25 | |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 26 | #define MAX_KEY_LEN 100 |
| 27 | |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 28 | static DEFINE_SPINLOCK(blkio_list_lock); |
| 29 | static LIST_HEAD(blkio_list); |
Vivek Goyal | b1c3576 | 2009-12-03 12:59:47 -0500 | [diff] [blame] | 30 | |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 31 | static DEFINE_MUTEX(all_q_mutex); |
| 32 | static LIST_HEAD(all_q_list); |
| 33 | |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 34 | /* List of groups pending per cpu stats allocation */ |
| 35 | static DEFINE_SPINLOCK(alloc_list_lock); |
| 36 | static LIST_HEAD(alloc_list); |
| 37 | |
| 38 | static void blkio_stat_alloc_fn(struct work_struct *); |
| 39 | static DECLARE_DELAYED_WORK(blkio_stat_alloc_work, blkio_stat_alloc_fn); |
| 40 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 41 | struct blkio_cgroup blkio_root_cgroup = { .weight = 2*BLKIO_WEIGHT_DEFAULT }; |
Vivek Goyal | 9d6a986 | 2009-12-04 10:36:41 -0500 | [diff] [blame] | 42 | EXPORT_SYMBOL_GPL(blkio_root_cgroup); |
| 43 | |
Tejun Heo | 035d10b | 2012-03-05 13:15:04 -0800 | [diff] [blame] | 44 | static struct blkio_policy_type *blkio_policy[BLKIO_NR_POLICIES]; |
| 45 | |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 46 | /* for encoding cft->private value on file */ |
| 47 | #define BLKIOFILE_PRIVATE(x, val) (((x) << 16) | (val)) |
| 48 | /* What policy owns the file, proportional or throttle */ |
| 49 | #define BLKIOFILE_POLICY(val) (((val) >> 16) & 0xffff) |
| 50 | #define BLKIOFILE_ATTR(val) ((val) & 0xffff) |
| 51 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 52 | struct blkio_cgroup *cgroup_to_blkio_cgroup(struct cgroup *cgroup) |
| 53 | { |
| 54 | return container_of(cgroup_subsys_state(cgroup, blkio_subsys_id), |
| 55 | struct blkio_cgroup, css); |
| 56 | } |
Vivek Goyal | 9d6a986 | 2009-12-04 10:36:41 -0500 | [diff] [blame] | 57 | EXPORT_SYMBOL_GPL(cgroup_to_blkio_cgroup); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 58 | |
Tejun Heo | 4f85cb9 | 2012-03-05 13:15:28 -0800 | [diff] [blame] | 59 | static struct blkio_cgroup *task_blkio_cgroup(struct task_struct *tsk) |
Vivek Goyal | 70087dc | 2011-05-16 15:24:08 +0200 | [diff] [blame] | 60 | { |
| 61 | return container_of(task_subsys_state(tsk, blkio_subsys_id), |
| 62 | struct blkio_cgroup, css); |
| 63 | } |
Tejun Heo | 4f85cb9 | 2012-03-05 13:15:28 -0800 | [diff] [blame] | 64 | |
| 65 | struct blkio_cgroup *bio_blkio_cgroup(struct bio *bio) |
| 66 | { |
| 67 | if (bio && bio->bi_css) |
| 68 | return container_of(bio->bi_css, struct blkio_cgroup, css); |
| 69 | return task_blkio_cgroup(current); |
| 70 | } |
| 71 | EXPORT_SYMBOL_GPL(bio_blkio_cgroup); |
Vivek Goyal | 70087dc | 2011-05-16 15:24:08 +0200 | [diff] [blame] | 72 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 73 | static inline void blkio_update_group_weight(struct blkio_group *blkg, |
| 74 | int plid, unsigned int weight) |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 75 | { |
| 76 | struct blkio_policy_type *blkiop; |
| 77 | |
| 78 | list_for_each_entry(blkiop, &blkio_list, list) { |
| 79 | /* If this policy does not own the blkg, do not send updates */ |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 80 | if (blkiop->plid != plid) |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 81 | continue; |
| 82 | if (blkiop->ops.blkio_update_group_weight_fn) |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 83 | blkiop->ops.blkio_update_group_weight_fn(blkg->q, |
Vivek Goyal | fe07143 | 2010-10-01 14:49:49 +0200 | [diff] [blame] | 84 | blkg, weight); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 85 | } |
| 86 | } |
| 87 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 88 | static inline void blkio_update_group_bps(struct blkio_group *blkg, int plid, |
| 89 | u64 bps, int fileid) |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 90 | { |
| 91 | struct blkio_policy_type *blkiop; |
| 92 | |
| 93 | list_for_each_entry(blkiop, &blkio_list, list) { |
| 94 | |
| 95 | /* If this policy does not own the blkg, do not send updates */ |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 96 | if (blkiop->plid != plid) |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 97 | continue; |
| 98 | |
| 99 | if (fileid == BLKIO_THROTL_read_bps_device |
| 100 | && blkiop->ops.blkio_update_group_read_bps_fn) |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 101 | blkiop->ops.blkio_update_group_read_bps_fn(blkg->q, |
Vivek Goyal | fe07143 | 2010-10-01 14:49:49 +0200 | [diff] [blame] | 102 | blkg, bps); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 103 | |
| 104 | if (fileid == BLKIO_THROTL_write_bps_device |
| 105 | && blkiop->ops.blkio_update_group_write_bps_fn) |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 106 | blkiop->ops.blkio_update_group_write_bps_fn(blkg->q, |
Vivek Goyal | fe07143 | 2010-10-01 14:49:49 +0200 | [diff] [blame] | 107 | blkg, bps); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 108 | } |
| 109 | } |
| 110 | |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 111 | static inline void blkio_update_group_iops(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 112 | int plid, unsigned int iops, |
| 113 | int fileid) |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 114 | { |
| 115 | struct blkio_policy_type *blkiop; |
| 116 | |
| 117 | list_for_each_entry(blkiop, &blkio_list, list) { |
| 118 | |
| 119 | /* If this policy does not own the blkg, do not send updates */ |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 120 | if (blkiop->plid != plid) |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 121 | continue; |
| 122 | |
| 123 | if (fileid == BLKIO_THROTL_read_iops_device |
| 124 | && blkiop->ops.blkio_update_group_read_iops_fn) |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 125 | blkiop->ops.blkio_update_group_read_iops_fn(blkg->q, |
Vivek Goyal | fe07143 | 2010-10-01 14:49:49 +0200 | [diff] [blame] | 126 | blkg, iops); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 127 | |
| 128 | if (fileid == BLKIO_THROTL_write_iops_device |
| 129 | && blkiop->ops.blkio_update_group_write_iops_fn) |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 130 | blkiop->ops.blkio_update_group_write_iops_fn(blkg->q, |
Vivek Goyal | fe07143 | 2010-10-01 14:49:49 +0200 | [diff] [blame] | 131 | blkg,iops); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 132 | } |
| 133 | } |
| 134 | |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 135 | /* |
| 136 | * Add to the appropriate stat variable depending on the request type. |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 137 | * This should be called with queue_lock held. |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 138 | */ |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 139 | static void blkio_add_stat(uint64_t *stat, uint64_t add, bool direction, |
| 140 | bool sync) |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 141 | { |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 142 | if (direction) |
| 143 | stat[BLKIO_STAT_WRITE] += add; |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 144 | else |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 145 | stat[BLKIO_STAT_READ] += add; |
| 146 | if (sync) |
| 147 | stat[BLKIO_STAT_SYNC] += add; |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 148 | else |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 149 | stat[BLKIO_STAT_ASYNC] += add; |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 150 | } |
| 151 | |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 152 | /* |
| 153 | * Decrements the appropriate stat variable if non-zero depending on the |
| 154 | * request type. Panics on value being zero. |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 155 | * This should be called with the queue_lock held. |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 156 | */ |
| 157 | static void blkio_check_and_dec_stat(uint64_t *stat, bool direction, bool sync) |
| 158 | { |
| 159 | if (direction) { |
| 160 | BUG_ON(stat[BLKIO_STAT_WRITE] == 0); |
| 161 | stat[BLKIO_STAT_WRITE]--; |
| 162 | } else { |
| 163 | BUG_ON(stat[BLKIO_STAT_READ] == 0); |
| 164 | stat[BLKIO_STAT_READ]--; |
| 165 | } |
| 166 | if (sync) { |
| 167 | BUG_ON(stat[BLKIO_STAT_SYNC] == 0); |
| 168 | stat[BLKIO_STAT_SYNC]--; |
| 169 | } else { |
| 170 | BUG_ON(stat[BLKIO_STAT_ASYNC] == 0); |
| 171 | stat[BLKIO_STAT_ASYNC]--; |
| 172 | } |
| 173 | } |
| 174 | |
| 175 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 176 | /* This should be called with the queue_lock held. */ |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 177 | static void blkio_set_start_group_wait_time(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 178 | struct blkio_policy_type *pol, |
| 179 | struct blkio_group *curr_blkg) |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 180 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 181 | struct blkg_policy_data *pd = blkg->pd[pol->plid]; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 182 | |
| 183 | if (blkio_blkg_waiting(&pd->stats)) |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 184 | return; |
| 185 | if (blkg == curr_blkg) |
| 186 | return; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 187 | pd->stats.start_group_wait_time = sched_clock(); |
| 188 | blkio_mark_blkg_waiting(&pd->stats); |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 189 | } |
| 190 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 191 | /* This should be called with the queue_lock held. */ |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 192 | static void blkio_update_group_wait_time(struct blkio_group_stats *stats) |
| 193 | { |
| 194 | unsigned long long now; |
| 195 | |
| 196 | if (!blkio_blkg_waiting(stats)) |
| 197 | return; |
| 198 | |
| 199 | now = sched_clock(); |
| 200 | if (time_after64(now, stats->start_group_wait_time)) |
| 201 | stats->group_wait_time += now - stats->start_group_wait_time; |
| 202 | blkio_clear_blkg_waiting(stats); |
| 203 | } |
| 204 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 205 | /* This should be called with the queue_lock held. */ |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 206 | static void blkio_end_empty_time(struct blkio_group_stats *stats) |
| 207 | { |
| 208 | unsigned long long now; |
| 209 | |
| 210 | if (!blkio_blkg_empty(stats)) |
| 211 | return; |
| 212 | |
| 213 | now = sched_clock(); |
| 214 | if (time_after64(now, stats->start_empty_time)) |
| 215 | stats->empty_time += now - stats->start_empty_time; |
| 216 | blkio_clear_blkg_empty(stats); |
| 217 | } |
| 218 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 219 | void blkiocg_update_set_idle_time_stats(struct blkio_group *blkg, |
| 220 | struct blkio_policy_type *pol) |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 221 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 222 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 223 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 224 | lockdep_assert_held(blkg->q->queue_lock); |
| 225 | BUG_ON(blkio_blkg_idling(stats)); |
| 226 | |
| 227 | stats->start_idle_time = sched_clock(); |
| 228 | blkio_mark_blkg_idling(stats); |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 229 | } |
| 230 | EXPORT_SYMBOL_GPL(blkiocg_update_set_idle_time_stats); |
| 231 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 232 | void blkiocg_update_idle_time_stats(struct blkio_group *blkg, |
| 233 | struct blkio_policy_type *pol) |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 234 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 235 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 236 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 237 | lockdep_assert_held(blkg->q->queue_lock); |
| 238 | |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 239 | if (blkio_blkg_idling(stats)) { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 240 | unsigned long long now = sched_clock(); |
| 241 | |
| 242 | if (time_after64(now, stats->start_idle_time)) { |
| 243 | u64_stats_update_begin(&stats->syncp); |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 244 | stats->idle_time += now - stats->start_idle_time; |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 245 | u64_stats_update_end(&stats->syncp); |
| 246 | } |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 247 | blkio_clear_blkg_idling(stats); |
| 248 | } |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 249 | } |
| 250 | EXPORT_SYMBOL_GPL(blkiocg_update_idle_time_stats); |
| 251 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 252 | void blkiocg_update_avg_queue_size_stats(struct blkio_group *blkg, |
| 253 | struct blkio_policy_type *pol) |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 254 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 255 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 256 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 257 | lockdep_assert_held(blkg->q->queue_lock); |
| 258 | |
| 259 | u64_stats_update_begin(&stats->syncp); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 260 | stats->avg_queue_size_sum += |
| 261 | stats->stat_arr[BLKIO_STAT_QUEUED][BLKIO_STAT_READ] + |
| 262 | stats->stat_arr[BLKIO_STAT_QUEUED][BLKIO_STAT_WRITE]; |
| 263 | stats->avg_queue_size_samples++; |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 264 | blkio_update_group_wait_time(stats); |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 265 | u64_stats_update_end(&stats->syncp); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 266 | } |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 267 | EXPORT_SYMBOL_GPL(blkiocg_update_avg_queue_size_stats); |
| 268 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 269 | void blkiocg_set_start_empty_time(struct blkio_group *blkg, |
| 270 | struct blkio_policy_type *pol) |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 271 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 272 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 273 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 274 | lockdep_assert_held(blkg->q->queue_lock); |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 275 | |
| 276 | if (stats->stat_arr[BLKIO_STAT_QUEUED][BLKIO_STAT_READ] || |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 277 | stats->stat_arr[BLKIO_STAT_QUEUED][BLKIO_STAT_WRITE]) |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 278 | return; |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 279 | |
| 280 | /* |
Vivek Goyal | e5ff082 | 2010-04-26 19:25:11 +0200 | [diff] [blame] | 281 | * group is already marked empty. This can happen if cfqq got new |
| 282 | * request in parent group and moved to this group while being added |
| 283 | * to service tree. Just ignore the event and move on. |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 284 | */ |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 285 | if (blkio_blkg_empty(stats)) |
Vivek Goyal | e5ff082 | 2010-04-26 19:25:11 +0200 | [diff] [blame] | 286 | return; |
Vivek Goyal | e5ff082 | 2010-04-26 19:25:11 +0200 | [diff] [blame] | 287 | |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 288 | stats->start_empty_time = sched_clock(); |
| 289 | blkio_mark_blkg_empty(stats); |
Divyesh Shah | 28baf44 | 2010-04-14 11:22:38 +0200 | [diff] [blame] | 290 | } |
| 291 | EXPORT_SYMBOL_GPL(blkiocg_set_start_empty_time); |
| 292 | |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 293 | void blkiocg_update_dequeue_stats(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 294 | struct blkio_policy_type *pol, |
| 295 | unsigned long dequeue) |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 296 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 297 | struct blkg_policy_data *pd = blkg->pd[pol->plid]; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 298 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 299 | lockdep_assert_held(blkg->q->queue_lock); |
| 300 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 301 | pd->stats.dequeue += dequeue; |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 302 | } |
| 303 | EXPORT_SYMBOL_GPL(blkiocg_update_dequeue_stats); |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 304 | #else |
| 305 | static inline void blkio_set_start_group_wait_time(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 306 | struct blkio_policy_type *pol, |
| 307 | struct blkio_group *curr_blkg) { } |
| 308 | static inline void blkio_end_empty_time(struct blkio_group_stats *stats) { } |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 309 | #endif |
| 310 | |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 311 | void blkiocg_update_io_add_stats(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 312 | struct blkio_policy_type *pol, |
| 313 | struct blkio_group *curr_blkg, bool direction, |
| 314 | bool sync) |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 315 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 316 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 317 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 318 | lockdep_assert_held(blkg->q->queue_lock); |
| 319 | |
| 320 | u64_stats_update_begin(&stats->syncp); |
| 321 | blkio_add_stat(stats->stat_arr[BLKIO_STAT_QUEUED], 1, direction, sync); |
| 322 | blkio_end_empty_time(stats); |
| 323 | u64_stats_update_end(&stats->syncp); |
| 324 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 325 | blkio_set_start_group_wait_time(blkg, pol, curr_blkg); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 326 | } |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 327 | EXPORT_SYMBOL_GPL(blkiocg_update_io_add_stats); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 328 | |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 329 | void blkiocg_update_io_remove_stats(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 330 | struct blkio_policy_type *pol, |
| 331 | bool direction, bool sync) |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 332 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 333 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 334 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 335 | lockdep_assert_held(blkg->q->queue_lock); |
| 336 | |
| 337 | u64_stats_update_begin(&stats->syncp); |
| 338 | blkio_check_and_dec_stat(stats->stat_arr[BLKIO_STAT_QUEUED], direction, |
| 339 | sync); |
| 340 | u64_stats_update_end(&stats->syncp); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 341 | } |
Divyesh Shah | a11cdaa | 2010-04-13 19:59:17 +0200 | [diff] [blame] | 342 | EXPORT_SYMBOL_GPL(blkiocg_update_io_remove_stats); |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 343 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 344 | void blkiocg_update_timeslice_used(struct blkio_group *blkg, |
| 345 | struct blkio_policy_type *pol, |
| 346 | unsigned long time, |
| 347 | unsigned long unaccounted_time) |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 348 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 349 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 350 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 351 | lockdep_assert_held(blkg->q->queue_lock); |
| 352 | |
| 353 | u64_stats_update_begin(&stats->syncp); |
| 354 | stats->time += time; |
Vivek Goyal | a23e686 | 2011-05-19 15:38:20 -0400 | [diff] [blame] | 355 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 356 | stats->unaccounted_time += unaccounted_time; |
Vivek Goyal | a23e686 | 2011-05-19 15:38:20 -0400 | [diff] [blame] | 357 | #endif |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 358 | u64_stats_update_end(&stats->syncp); |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 359 | } |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 360 | EXPORT_SYMBOL_GPL(blkiocg_update_timeslice_used); |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 361 | |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 362 | /* |
| 363 | * should be called under rcu read lock or queue lock to make sure blkg pointer |
| 364 | * is valid. |
| 365 | */ |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 366 | void blkiocg_update_dispatch_stats(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 367 | struct blkio_policy_type *pol, |
| 368 | uint64_t bytes, bool direction, bool sync) |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 369 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 370 | struct blkg_policy_data *pd = blkg->pd[pol->plid]; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 371 | struct blkio_group_stats_cpu *stats_cpu; |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 372 | unsigned long flags; |
| 373 | |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 374 | /* If per cpu stats are not allocated yet, don't do any accounting. */ |
| 375 | if (pd->stats_cpu == NULL) |
| 376 | return; |
| 377 | |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 378 | /* |
| 379 | * Disabling interrupts to provide mutual exclusion between two |
| 380 | * writes on same cpu. It probably is not needed for 64bit. Not |
| 381 | * optimizing that case yet. |
| 382 | */ |
| 383 | local_irq_save(flags); |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 384 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 385 | stats_cpu = this_cpu_ptr(pd->stats_cpu); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 386 | |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 387 | u64_stats_update_begin(&stats_cpu->syncp); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 388 | stats_cpu->sectors += bytes >> 9; |
| 389 | blkio_add_stat(stats_cpu->stat_arr_cpu[BLKIO_STAT_CPU_SERVICED], |
| 390 | 1, direction, sync); |
| 391 | blkio_add_stat(stats_cpu->stat_arr_cpu[BLKIO_STAT_CPU_SERVICE_BYTES], |
| 392 | bytes, direction, sync); |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 393 | u64_stats_update_end(&stats_cpu->syncp); |
| 394 | local_irq_restore(flags); |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 395 | } |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 396 | EXPORT_SYMBOL_GPL(blkiocg_update_dispatch_stats); |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 397 | |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 398 | void blkiocg_update_completion_stats(struct blkio_group *blkg, |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 399 | struct blkio_policy_type *pol, |
| 400 | uint64_t start_time, |
| 401 | uint64_t io_start_time, bool direction, |
| 402 | bool sync) |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 403 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 404 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 405 | unsigned long long now = sched_clock(); |
| 406 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 407 | lockdep_assert_held(blkg->q->queue_lock); |
| 408 | |
| 409 | u64_stats_update_begin(&stats->syncp); |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 410 | if (time_after64(now, io_start_time)) |
| 411 | blkio_add_stat(stats->stat_arr[BLKIO_STAT_SERVICE_TIME], |
| 412 | now - io_start_time, direction, sync); |
| 413 | if (time_after64(io_start_time, start_time)) |
| 414 | blkio_add_stat(stats->stat_arr[BLKIO_STAT_WAIT_TIME], |
| 415 | io_start_time - start_time, direction, sync); |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 416 | u64_stats_update_end(&stats->syncp); |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 417 | } |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 418 | EXPORT_SYMBOL_GPL(blkiocg_update_completion_stats); |
Divyesh Shah | 9195291 | 2010-04-01 15:01:41 -0700 | [diff] [blame] | 419 | |
Vivek Goyal | 317389a | 2011-05-23 10:02:19 +0200 | [diff] [blame] | 420 | /* Merged stats are per cpu. */ |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 421 | void blkiocg_update_io_merged_stats(struct blkio_group *blkg, |
| 422 | struct blkio_policy_type *pol, |
| 423 | bool direction, bool sync) |
Divyesh Shah | 812d402 | 2010-04-08 21:14:23 -0700 | [diff] [blame] | 424 | { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 425 | struct blkio_group_stats *stats = &blkg->pd[pol->plid]->stats; |
Divyesh Shah | 812d402 | 2010-04-08 21:14:23 -0700 | [diff] [blame] | 426 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 427 | lockdep_assert_held(blkg->q->queue_lock); |
| 428 | |
| 429 | u64_stats_update_begin(&stats->syncp); |
Tejun Heo | 5fe224d | 2012-03-08 10:53:57 -0800 | [diff] [blame] | 430 | blkio_add_stat(stats->stat_arr[BLKIO_STAT_MERGED], 1, direction, sync); |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 431 | u64_stats_update_end(&stats->syncp); |
Divyesh Shah | 812d402 | 2010-04-08 21:14:23 -0700 | [diff] [blame] | 432 | } |
| 433 | EXPORT_SYMBOL_GPL(blkiocg_update_io_merged_stats); |
| 434 | |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 435 | /* |
| 436 | * Worker for allocating per cpu stat for blk groups. This is scheduled on |
| 437 | * the system_nrt_wq once there are some groups on the alloc_list waiting |
| 438 | * for allocation. |
| 439 | */ |
| 440 | static void blkio_stat_alloc_fn(struct work_struct *work) |
| 441 | { |
| 442 | static void *pcpu_stats[BLKIO_NR_POLICIES]; |
| 443 | struct delayed_work *dwork = to_delayed_work(work); |
| 444 | struct blkio_group *blkg; |
| 445 | int i; |
| 446 | bool empty = false; |
| 447 | |
| 448 | alloc_stats: |
| 449 | for (i = 0; i < BLKIO_NR_POLICIES; i++) { |
| 450 | if (pcpu_stats[i] != NULL) |
| 451 | continue; |
| 452 | |
| 453 | pcpu_stats[i] = alloc_percpu(struct blkio_group_stats_cpu); |
| 454 | |
| 455 | /* Allocation failed. Try again after some time. */ |
| 456 | if (pcpu_stats[i] == NULL) { |
| 457 | queue_delayed_work(system_nrt_wq, dwork, |
| 458 | msecs_to_jiffies(10)); |
| 459 | return; |
| 460 | } |
| 461 | } |
| 462 | |
| 463 | spin_lock_irq(&blkio_list_lock); |
| 464 | spin_lock(&alloc_list_lock); |
| 465 | |
| 466 | /* cgroup got deleted or queue exited. */ |
| 467 | if (!list_empty(&alloc_list)) { |
| 468 | blkg = list_first_entry(&alloc_list, struct blkio_group, |
| 469 | alloc_node); |
| 470 | for (i = 0; i < BLKIO_NR_POLICIES; i++) { |
| 471 | struct blkg_policy_data *pd = blkg->pd[i]; |
| 472 | |
| 473 | if (blkio_policy[i] && pd && !pd->stats_cpu) |
| 474 | swap(pd->stats_cpu, pcpu_stats[i]); |
| 475 | } |
| 476 | |
| 477 | list_del_init(&blkg->alloc_node); |
| 478 | } |
| 479 | |
| 480 | empty = list_empty(&alloc_list); |
| 481 | |
| 482 | spin_unlock(&alloc_list_lock); |
| 483 | spin_unlock_irq(&blkio_list_lock); |
| 484 | |
| 485 | if (!empty) |
| 486 | goto alloc_stats; |
| 487 | } |
| 488 | |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 489 | /** |
| 490 | * blkg_free - free a blkg |
| 491 | * @blkg: blkg to free |
| 492 | * |
| 493 | * Free @blkg which may be partially allocated. |
| 494 | */ |
| 495 | static void blkg_free(struct blkio_group *blkg) |
| 496 | { |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 497 | int i; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 498 | |
| 499 | if (!blkg) |
| 500 | return; |
| 501 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 502 | for (i = 0; i < BLKIO_NR_POLICIES; i++) { |
| 503 | struct blkg_policy_data *pd = blkg->pd[i]; |
| 504 | |
| 505 | if (pd) { |
| 506 | free_percpu(pd->stats_cpu); |
| 507 | kfree(pd); |
| 508 | } |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 509 | } |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 510 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 511 | kfree(blkg); |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 512 | } |
| 513 | |
| 514 | /** |
| 515 | * blkg_alloc - allocate a blkg |
| 516 | * @blkcg: block cgroup the new blkg is associated with |
| 517 | * @q: request_queue the new blkg is associated with |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 518 | * |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 519 | * Allocate a new blkg assocating @blkcg and @q. |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 520 | */ |
| 521 | static struct blkio_group *blkg_alloc(struct blkio_cgroup *blkcg, |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 522 | struct request_queue *q) |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 523 | { |
| 524 | struct blkio_group *blkg; |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 525 | int i; |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 526 | |
| 527 | /* alloc and init base part */ |
| 528 | blkg = kzalloc_node(sizeof(*blkg), GFP_ATOMIC, q->node); |
| 529 | if (!blkg) |
| 530 | return NULL; |
| 531 | |
Tejun Heo | c875f4d | 2012-03-05 13:15:22 -0800 | [diff] [blame] | 532 | blkg->q = q; |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 533 | INIT_LIST_HEAD(&blkg->q_node); |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 534 | INIT_LIST_HEAD(&blkg->alloc_node); |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 535 | blkg->blkcg = blkcg; |
Tejun Heo | 1adaf3d | 2012-03-05 13:15:15 -0800 | [diff] [blame] | 536 | blkg->refcnt = 1; |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 537 | cgroup_path(blkcg->css.cgroup, blkg->path, sizeof(blkg->path)); |
| 538 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 539 | for (i = 0; i < BLKIO_NR_POLICIES; i++) { |
| 540 | struct blkio_policy_type *pol = blkio_policy[i]; |
| 541 | struct blkg_policy_data *pd; |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 542 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 543 | if (!pol) |
| 544 | continue; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 545 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 546 | /* alloc per-policy data and attach it to blkg */ |
| 547 | pd = kzalloc_node(sizeof(*pd) + pol->pdata_size, GFP_ATOMIC, |
| 548 | q->node); |
| 549 | if (!pd) { |
| 550 | blkg_free(blkg); |
| 551 | return NULL; |
| 552 | } |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 553 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 554 | blkg->pd[i] = pd; |
| 555 | pd->blkg = blkg; |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 556 | } |
| 557 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 558 | /* invoke per-policy init */ |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 559 | for (i = 0; i < BLKIO_NR_POLICIES; i++) { |
| 560 | struct blkio_policy_type *pol = blkio_policy[i]; |
| 561 | |
| 562 | if (pol) |
| 563 | pol->ops.blkio_init_group_fn(blkg); |
| 564 | } |
| 565 | |
Tejun Heo | 0381411 | 2012-03-05 13:15:14 -0800 | [diff] [blame] | 566 | return blkg; |
| 567 | } |
| 568 | |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 569 | struct blkio_group *blkg_lookup_create(struct blkio_cgroup *blkcg, |
| 570 | struct request_queue *q, |
| 571 | enum blkio_policy_id plid, |
| 572 | bool for_root) |
| 573 | __releases(q->queue_lock) __acquires(q->queue_lock) |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 574 | { |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 575 | struct blkio_group *blkg; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 576 | |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 577 | WARN_ON_ONCE(!rcu_read_lock_held()); |
| 578 | lockdep_assert_held(q->queue_lock); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 579 | |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 580 | /* |
| 581 | * This could be the first entry point of blkcg implementation and |
| 582 | * we shouldn't allow anything to go through for a bypassing queue. |
| 583 | * The following can be removed if blkg lookup is guaranteed to |
| 584 | * fail on a bypassing queue. |
| 585 | */ |
| 586 | if (unlikely(blk_queue_bypass(q)) && !for_root) |
| 587 | return ERR_PTR(blk_queue_dead(q) ? -EINVAL : -EBUSY); |
| 588 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 589 | blkg = blkg_lookup(blkcg, q); |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 590 | if (blkg) |
| 591 | return blkg; |
| 592 | |
Tejun Heo | 7ee9c56 | 2012-03-05 13:15:11 -0800 | [diff] [blame] | 593 | /* blkg holds a reference to blkcg */ |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 594 | if (!css_tryget(&blkcg->css)) |
| 595 | return ERR_PTR(-EINVAL); |
| 596 | |
| 597 | /* |
| 598 | * Allocate and initialize. |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 599 | */ |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 600 | blkg = blkg_alloc(blkcg, q); |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 601 | |
| 602 | /* did alloc fail? */ |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 603 | if (unlikely(!blkg)) { |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 604 | blkg = ERR_PTR(-ENOMEM); |
| 605 | goto out; |
| 606 | } |
| 607 | |
| 608 | /* insert */ |
| 609 | spin_lock(&blkcg->lock); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 610 | hlist_add_head_rcu(&blkg->blkcg_node, &blkcg->blkg_list); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 611 | list_add(&blkg->q_node, &q->blkg_list); |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 612 | spin_unlock(&blkcg->lock); |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 613 | |
| 614 | spin_lock(&alloc_list_lock); |
| 615 | list_add(&blkg->alloc_node, &alloc_list); |
| 616 | /* Queue per cpu stat allocation from worker thread. */ |
| 617 | queue_delayed_work(system_nrt_wq, &blkio_stat_alloc_work, 0); |
| 618 | spin_unlock(&alloc_list_lock); |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 619 | out: |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 620 | return blkg; |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 621 | } |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 622 | EXPORT_SYMBOL_GPL(blkg_lookup_create); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 623 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 624 | /* called under rcu_read_lock(). */ |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 625 | struct blkio_group *blkg_lookup(struct blkio_cgroup *blkcg, |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 626 | struct request_queue *q) |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 627 | { |
| 628 | struct blkio_group *blkg; |
| 629 | struct hlist_node *n; |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 630 | |
Tejun Heo | ca32aef | 2012-03-05 13:15:03 -0800 | [diff] [blame] | 631 | hlist_for_each_entry_rcu(blkg, n, &blkcg->blkg_list, blkcg_node) |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 632 | if (blkg->q == q) |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 633 | return blkg; |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 634 | return NULL; |
| 635 | } |
Tejun Heo | cd1604f | 2012-03-05 13:15:06 -0800 | [diff] [blame] | 636 | EXPORT_SYMBOL_GPL(blkg_lookup); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 637 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 638 | static void blkg_destroy(struct blkio_group *blkg) |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 639 | { |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 640 | struct request_queue *q = blkg->q; |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 641 | struct blkio_cgroup *blkcg = blkg->blkcg; |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 642 | |
| 643 | lockdep_assert_held(q->queue_lock); |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 644 | lockdep_assert_held(&blkcg->lock); |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 645 | |
| 646 | /* Something wrong if we are trying to remove same group twice */ |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 647 | WARN_ON_ONCE(list_empty(&blkg->q_node)); |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 648 | WARN_ON_ONCE(hlist_unhashed(&blkg->blkcg_node)); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 649 | list_del_init(&blkg->q_node); |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 650 | hlist_del_init_rcu(&blkg->blkcg_node); |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 651 | |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 652 | spin_lock(&alloc_list_lock); |
| 653 | list_del_init(&blkg->alloc_node); |
| 654 | spin_unlock(&alloc_list_lock); |
| 655 | |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 656 | /* |
| 657 | * Put the reference taken at the time of creation so that when all |
| 658 | * queues are gone, group can be destroyed. |
| 659 | */ |
| 660 | blkg_put(blkg); |
| 661 | } |
| 662 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 663 | /* |
| 664 | * XXX: This updates blkg policy data in-place for root blkg, which is |
| 665 | * necessary across elevator switch and policy registration as root blkgs |
| 666 | * aren't shot down. This broken and racy implementation is temporary. |
| 667 | * Eventually, blkg shoot down will be replaced by proper in-place update. |
| 668 | */ |
| 669 | void update_root_blkg_pd(struct request_queue *q, enum blkio_policy_id plid) |
| 670 | { |
| 671 | struct blkio_policy_type *pol = blkio_policy[plid]; |
| 672 | struct blkio_group *blkg = blkg_lookup(&blkio_root_cgroup, q); |
| 673 | struct blkg_policy_data *pd; |
| 674 | |
| 675 | if (!blkg) |
| 676 | return; |
| 677 | |
| 678 | kfree(blkg->pd[plid]); |
| 679 | blkg->pd[plid] = NULL; |
| 680 | |
| 681 | if (!pol) |
| 682 | return; |
| 683 | |
| 684 | pd = kzalloc(sizeof(*pd) + pol->pdata_size, GFP_KERNEL); |
| 685 | WARN_ON_ONCE(!pd); |
| 686 | |
| 687 | pd->stats_cpu = alloc_percpu(struct blkio_group_stats_cpu); |
| 688 | WARN_ON_ONCE(!pd->stats_cpu); |
| 689 | |
| 690 | blkg->pd[plid] = pd; |
| 691 | pd->blkg = blkg; |
| 692 | pol->ops.blkio_init_group_fn(blkg); |
| 693 | } |
| 694 | EXPORT_SYMBOL_GPL(update_root_blkg_pd); |
| 695 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 696 | /** |
| 697 | * blkg_destroy_all - destroy all blkgs associated with a request_queue |
| 698 | * @q: request_queue of interest |
| 699 | * @destroy_root: whether to destroy root blkg or not |
| 700 | * |
| 701 | * Destroy blkgs associated with @q. If @destroy_root is %true, all are |
| 702 | * destroyed; otherwise, root blkg is left alone. |
| 703 | */ |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 704 | void blkg_destroy_all(struct request_queue *q, bool destroy_root) |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 705 | { |
| 706 | struct blkio_group *blkg, *n; |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 707 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 708 | spin_lock_irq(q->queue_lock); |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 709 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 710 | list_for_each_entry_safe(blkg, n, &q->blkg_list, q_node) { |
| 711 | struct blkio_cgroup *blkcg = blkg->blkcg; |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 712 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 713 | /* skip root? */ |
| 714 | if (!destroy_root && blkg->blkcg == &blkio_root_cgroup) |
| 715 | continue; |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 716 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 717 | spin_lock(&blkcg->lock); |
| 718 | blkg_destroy(blkg); |
| 719 | spin_unlock(&blkcg->lock); |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 720 | } |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 721 | |
| 722 | spin_unlock_irq(q->queue_lock); |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 723 | } |
Tejun Heo | 03aa264 | 2012-03-05 13:15:19 -0800 | [diff] [blame] | 724 | EXPORT_SYMBOL_GPL(blkg_destroy_all); |
Tejun Heo | 72e06c2 | 2012-03-05 13:15:00 -0800 | [diff] [blame] | 725 | |
Tejun Heo | 1adaf3d | 2012-03-05 13:15:15 -0800 | [diff] [blame] | 726 | static void blkg_rcu_free(struct rcu_head *rcu_head) |
| 727 | { |
| 728 | blkg_free(container_of(rcu_head, struct blkio_group, rcu_head)); |
| 729 | } |
| 730 | |
| 731 | void __blkg_release(struct blkio_group *blkg) |
| 732 | { |
| 733 | /* release the extra blkcg reference this blkg has been holding */ |
| 734 | css_put(&blkg->blkcg->css); |
| 735 | |
| 736 | /* |
| 737 | * A group is freed in rcu manner. But having an rcu lock does not |
| 738 | * mean that one can access all the fields of blkg and assume these |
| 739 | * are valid. For example, don't try to follow throtl_data and |
| 740 | * request queue links. |
| 741 | * |
| 742 | * Having a reference to blkg under an rcu allows acess to only |
| 743 | * values local to groups like group stats and group rate limits |
| 744 | */ |
| 745 | call_rcu(&blkg->rcu_head, blkg_rcu_free); |
| 746 | } |
| 747 | EXPORT_SYMBOL_GPL(__blkg_release); |
| 748 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 749 | static void blkio_reset_stats_cpu(struct blkio_group *blkg, int plid) |
Vivek Goyal | f0bdc8c | 2011-05-19 15:38:30 -0400 | [diff] [blame] | 750 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 751 | struct blkg_policy_data *pd = blkg->pd[plid]; |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 752 | int cpu; |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 753 | |
| 754 | if (pd->stats_cpu == NULL) |
| 755 | return; |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 756 | |
| 757 | for_each_possible_cpu(cpu) { |
| 758 | struct blkio_group_stats_cpu *sc = |
| 759 | per_cpu_ptr(pd->stats_cpu, cpu); |
| 760 | |
| 761 | sc->sectors = 0; |
| 762 | memset(sc->stat_arr_cpu, 0, sizeof(sc->stat_arr_cpu)); |
Vivek Goyal | f0bdc8c | 2011-05-19 15:38:30 -0400 | [diff] [blame] | 763 | } |
| 764 | } |
| 765 | |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 766 | static int |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 767 | blkiocg_reset_stats(struct cgroup *cgroup, struct cftype *cftype, u64 val) |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 768 | { |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 769 | struct blkio_cgroup *blkcg = cgroup_to_blkio_cgroup(cgroup); |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 770 | struct blkio_group *blkg; |
| 771 | struct hlist_node *n; |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 772 | int i; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 773 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 774 | spin_lock(&blkio_list_lock); |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 775 | spin_lock_irq(&blkcg->lock); |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 776 | |
| 777 | /* |
| 778 | * Note that stat reset is racy - it doesn't synchronize against |
| 779 | * stat updates. This is a debug feature which shouldn't exist |
| 780 | * anyway. If you get hit by a race, retry. |
| 781 | */ |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 782 | hlist_for_each_entry(blkg, n, &blkcg->blkg_list, blkcg_node) { |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 783 | struct blkio_policy_type *pol; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 784 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 785 | list_for_each_entry(pol, &blkio_list, list) { |
| 786 | struct blkg_policy_data *pd = blkg->pd[pol->plid]; |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 787 | struct blkio_group_stats *stats = &pd->stats; |
Vivek Goyal | f0bdc8c | 2011-05-19 15:38:30 -0400 | [diff] [blame] | 788 | |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 789 | /* queued stats shouldn't be cleared */ |
| 790 | for (i = 0; i < ARRAY_SIZE(stats->stat_arr); i++) |
| 791 | if (i != BLKIO_STAT_QUEUED) |
| 792 | memset(stats->stat_arr[i], 0, |
| 793 | sizeof(stats->stat_arr[i])); |
| 794 | stats->time = 0; |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 795 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Tejun Heo | 997a026 | 2012-03-08 10:53:58 -0800 | [diff] [blame] | 796 | memset((void *)stats + BLKG_STATS_DEBUG_CLEAR_START, 0, |
| 797 | BLKG_STATS_DEBUG_CLEAR_SIZE); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 798 | #endif |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 799 | blkio_reset_stats_cpu(blkg, pol->plid); |
| 800 | } |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 801 | } |
Vivek Goyal | f0bdc8c | 2011-05-19 15:38:30 -0400 | [diff] [blame] | 802 | |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 803 | spin_unlock_irq(&blkcg->lock); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 804 | spin_unlock(&blkio_list_lock); |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 805 | return 0; |
| 806 | } |
| 807 | |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 808 | static void blkio_get_key_name(enum stat_sub_type type, const char *dname, |
| 809 | char *str, int chars_left, bool diskname_only) |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 810 | { |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 811 | snprintf(str, chars_left, "%s", dname); |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 812 | chars_left -= strlen(str); |
| 813 | if (chars_left <= 0) { |
| 814 | printk(KERN_WARNING |
| 815 | "Possibly incorrect cgroup stat display format"); |
| 816 | return; |
| 817 | } |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 818 | if (diskname_only) |
| 819 | return; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 820 | switch (type) { |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 821 | case BLKIO_STAT_READ: |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 822 | strlcat(str, " Read", chars_left); |
| 823 | break; |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 824 | case BLKIO_STAT_WRITE: |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 825 | strlcat(str, " Write", chars_left); |
| 826 | break; |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 827 | case BLKIO_STAT_SYNC: |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 828 | strlcat(str, " Sync", chars_left); |
| 829 | break; |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 830 | case BLKIO_STAT_ASYNC: |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 831 | strlcat(str, " Async", chars_left); |
| 832 | break; |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 833 | case BLKIO_STAT_TOTAL: |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 834 | strlcat(str, " Total", chars_left); |
| 835 | break; |
| 836 | default: |
| 837 | strlcat(str, " Invalid", chars_left); |
| 838 | } |
| 839 | } |
| 840 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 841 | static uint64_t blkio_read_stat_cpu(struct blkio_group *blkg, int plid, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 842 | enum stat_type_cpu type, enum stat_sub_type sub_type) |
| 843 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 844 | struct blkg_policy_data *pd = blkg->pd[plid]; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 845 | int cpu; |
| 846 | struct blkio_group_stats_cpu *stats_cpu; |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 847 | u64 val = 0, tval; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 848 | |
Vivek Goyal | 1cd9e03 | 2012-03-08 10:53:56 -0800 | [diff] [blame] | 849 | if (pd->stats_cpu == NULL) |
| 850 | return val; |
| 851 | |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 852 | for_each_possible_cpu(cpu) { |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 853 | unsigned int start; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 854 | stats_cpu = per_cpu_ptr(pd->stats_cpu, cpu); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 855 | |
Vivek Goyal | 575969a | 2011-05-19 15:38:29 -0400 | [diff] [blame] | 856 | do { |
| 857 | start = u64_stats_fetch_begin(&stats_cpu->syncp); |
| 858 | if (type == BLKIO_STAT_CPU_SECTORS) |
| 859 | tval = stats_cpu->sectors; |
| 860 | else |
| 861 | tval = stats_cpu->stat_arr_cpu[type][sub_type]; |
| 862 | } while(u64_stats_fetch_retry(&stats_cpu->syncp, start)); |
| 863 | |
| 864 | val += tval; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 865 | } |
| 866 | |
| 867 | return val; |
| 868 | } |
| 869 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 870 | static uint64_t blkio_get_stat_cpu(struct blkio_group *blkg, int plid, |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 871 | struct cgroup_map_cb *cb, const char *dname, |
| 872 | enum stat_type_cpu type) |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 873 | { |
| 874 | uint64_t disk_total, val; |
| 875 | char key_str[MAX_KEY_LEN]; |
| 876 | enum stat_sub_type sub_type; |
| 877 | |
| 878 | if (type == BLKIO_STAT_CPU_SECTORS) { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 879 | val = blkio_read_stat_cpu(blkg, plid, type, 0); |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 880 | blkio_get_key_name(0, dname, key_str, MAX_KEY_LEN, true); |
| 881 | cb->fill(cb, key_str, val); |
| 882 | return val; |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 883 | } |
| 884 | |
| 885 | for (sub_type = BLKIO_STAT_READ; sub_type < BLKIO_STAT_TOTAL; |
| 886 | sub_type++) { |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 887 | blkio_get_key_name(sub_type, dname, key_str, MAX_KEY_LEN, |
| 888 | false); |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 889 | val = blkio_read_stat_cpu(blkg, plid, type, sub_type); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 890 | cb->fill(cb, key_str, val); |
| 891 | } |
| 892 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 893 | disk_total = blkio_read_stat_cpu(blkg, plid, type, BLKIO_STAT_READ) + |
| 894 | blkio_read_stat_cpu(blkg, plid, type, BLKIO_STAT_WRITE); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 895 | |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 896 | blkio_get_key_name(BLKIO_STAT_TOTAL, dname, key_str, MAX_KEY_LEN, |
| 897 | false); |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 898 | cb->fill(cb, key_str, disk_total); |
| 899 | return disk_total; |
| 900 | } |
| 901 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 902 | static uint64_t blkio_get_stat(struct blkio_group *blkg, int plid, |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 903 | struct cgroup_map_cb *cb, const char *dname, |
| 904 | enum stat_type type) |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 905 | { |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 906 | struct blkio_group_stats *stats = &blkg->pd[plid]->stats; |
| 907 | uint64_t v = 0, disk_total = 0; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 908 | char key_str[MAX_KEY_LEN]; |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 909 | unsigned int sync_start; |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 910 | int st; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 911 | |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 912 | if (type >= BLKIO_STAT_ARR_NR) { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 913 | do { |
| 914 | sync_start = u64_stats_fetch_begin(&stats->syncp); |
| 915 | switch (type) { |
| 916 | case BLKIO_STAT_TIME: |
| 917 | v = stats->time; |
| 918 | break; |
Justin TerAvest | 9026e52 | 2011-03-22 21:26:54 +0100 | [diff] [blame] | 919 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 920 | case BLKIO_STAT_UNACCOUNTED_TIME: |
| 921 | v = stats->unaccounted_time; |
| 922 | break; |
| 923 | case BLKIO_STAT_AVG_QUEUE_SIZE: { |
| 924 | uint64_t samples = stats->avg_queue_size_samples; |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 925 | |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 926 | if (samples) { |
| 927 | v = stats->avg_queue_size_sum; |
| 928 | do_div(v, samples); |
| 929 | } |
| 930 | break; |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 931 | } |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 932 | case BLKIO_STAT_IDLE_TIME: |
| 933 | v = stats->idle_time; |
| 934 | break; |
| 935 | case BLKIO_STAT_EMPTY_TIME: |
| 936 | v = stats->empty_time; |
| 937 | break; |
| 938 | case BLKIO_STAT_DEQUEUE: |
| 939 | v = stats->dequeue; |
| 940 | break; |
| 941 | case BLKIO_STAT_GROUP_WAIT_TIME: |
| 942 | v = stats->group_wait_time; |
| 943 | break; |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 944 | #endif |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 945 | default: |
| 946 | WARN_ON_ONCE(1); |
| 947 | } |
| 948 | } while (u64_stats_fetch_retry(&stats->syncp, sync_start)); |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 949 | |
| 950 | blkio_get_key_name(0, dname, key_str, MAX_KEY_LEN, true); |
| 951 | cb->fill(cb, key_str, v); |
| 952 | return v; |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 953 | } |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 954 | |
| 955 | for (st = BLKIO_STAT_READ; st < BLKIO_STAT_TOTAL; st++) { |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 956 | do { |
| 957 | sync_start = u64_stats_fetch_begin(&stats->syncp); |
| 958 | v = stats->stat_arr[type][st]; |
| 959 | } while (u64_stats_fetch_retry(&stats->syncp, sync_start)); |
Tejun Heo | c4c76a0 | 2012-03-08 10:53:59 -0800 | [diff] [blame] | 960 | |
| 961 | blkio_get_key_name(st, dname, key_str, MAX_KEY_LEN, false); |
| 962 | cb->fill(cb, key_str, v); |
| 963 | if (st == BLKIO_STAT_READ || st == BLKIO_STAT_WRITE) |
| 964 | disk_total += v; |
| 965 | } |
| 966 | |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 967 | blkio_get_key_name(BLKIO_STAT_TOTAL, dname, key_str, MAX_KEY_LEN, |
| 968 | false); |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 969 | cb->fill(cb, key_str, disk_total); |
| 970 | return disk_total; |
| 971 | } |
| 972 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 973 | static int blkio_policy_parse_and_set(char *buf, enum blkio_policy_id plid, |
| 974 | int fileid, struct blkio_cgroup *blkcg) |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 975 | { |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 976 | struct gendisk *disk = NULL; |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 977 | struct blkio_group *blkg = NULL; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 978 | struct blkg_policy_data *pd; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 979 | char *s[4], *p, *major_s = NULL, *minor_s = NULL; |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 980 | unsigned long major, minor; |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 981 | int i = 0, ret = -EINVAL; |
| 982 | int part; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 983 | dev_t dev; |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 984 | u64 temp; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 985 | |
| 986 | memset(s, 0, sizeof(s)); |
| 987 | |
| 988 | while ((p = strsep(&buf, " ")) != NULL) { |
| 989 | if (!*p) |
| 990 | continue; |
| 991 | |
| 992 | s[i++] = p; |
| 993 | |
| 994 | /* Prevent from inputing too many things */ |
| 995 | if (i == 3) |
| 996 | break; |
| 997 | } |
| 998 | |
| 999 | if (i != 2) |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1000 | goto out; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1001 | |
| 1002 | p = strsep(&s[0], ":"); |
| 1003 | if (p != NULL) |
| 1004 | major_s = p; |
| 1005 | else |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1006 | goto out; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1007 | |
| 1008 | minor_s = s[0]; |
| 1009 | if (!minor_s) |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1010 | goto out; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1011 | |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1012 | if (strict_strtoul(major_s, 10, &major)) |
| 1013 | goto out; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1014 | |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1015 | if (strict_strtoul(minor_s, 10, &minor)) |
| 1016 | goto out; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1017 | |
| 1018 | dev = MKDEV(major, minor); |
| 1019 | |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1020 | if (strict_strtoull(s[1], 10, &temp)) |
| 1021 | goto out; |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 1022 | |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1023 | disk = get_gendisk(dev, &part); |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1024 | if (!disk || part) |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1025 | goto out; |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1026 | |
| 1027 | rcu_read_lock(); |
| 1028 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1029 | spin_lock_irq(disk->queue->queue_lock); |
| 1030 | blkg = blkg_lookup_create(blkcg, disk->queue, plid, false); |
| 1031 | spin_unlock_irq(disk->queue->queue_lock); |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1032 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1033 | if (IS_ERR(blkg)) { |
| 1034 | ret = PTR_ERR(blkg); |
| 1035 | goto out_unlock; |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 1036 | } |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1037 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1038 | pd = blkg->pd[plid]; |
| 1039 | |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1040 | switch (plid) { |
| 1041 | case BLKIO_POLICY_PROP: |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 1042 | if ((temp < BLKIO_WEIGHT_MIN && temp > 0) || |
| 1043 | temp > BLKIO_WEIGHT_MAX) |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1044 | goto out_unlock; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1045 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1046 | pd->conf.weight = temp; |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1047 | blkio_update_group_weight(blkg, plid, temp ?: blkcg->weight); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1048 | break; |
| 1049 | case BLKIO_POLICY_THROTL: |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1050 | switch(fileid) { |
| 1051 | case BLKIO_THROTL_read_bps_device: |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1052 | pd->conf.bps[READ] = temp; |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1053 | blkio_update_group_bps(blkg, plid, temp ?: -1, fileid); |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1054 | break; |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1055 | case BLKIO_THROTL_write_bps_device: |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1056 | pd->conf.bps[WRITE] = temp; |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1057 | blkio_update_group_bps(blkg, plid, temp ?: -1, fileid); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1058 | break; |
| 1059 | case BLKIO_THROTL_read_iops_device: |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1060 | if (temp > THROTL_IOPS_MAX) |
| 1061 | goto out_unlock; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1062 | pd->conf.iops[READ] = temp; |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1063 | blkio_update_group_iops(blkg, plid, temp ?: -1, fileid); |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1064 | break; |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1065 | case BLKIO_THROTL_write_iops_device: |
Wanlong Gao | d11bb44 | 2011-09-21 10:22:10 +0200 | [diff] [blame] | 1066 | if (temp > THROTL_IOPS_MAX) |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1067 | goto out_unlock; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1068 | pd->conf.iops[WRITE] = temp; |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1069 | blkio_update_group_iops(blkg, plid, temp ?: -1, fileid); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1070 | break; |
| 1071 | } |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1072 | break; |
| 1073 | default: |
| 1074 | BUG(); |
| 1075 | } |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1076 | ret = 0; |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1077 | out_unlock: |
| 1078 | rcu_read_unlock(); |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1079 | out: |
| 1080 | put_disk(disk); |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1081 | |
| 1082 | /* |
| 1083 | * If queue was bypassing, we should retry. Do so after a short |
| 1084 | * msleep(). It isn't strictly necessary but queue can be |
| 1085 | * bypassing for some time and it's always nice to avoid busy |
| 1086 | * looping. |
| 1087 | */ |
| 1088 | if (ret == -EBUSY) { |
| 1089 | msleep(10); |
| 1090 | return restart_syscall(); |
| 1091 | } |
Tejun Heo | ece8424 | 2011-10-19 14:31:15 +0200 | [diff] [blame] | 1092 | return ret; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1093 | } |
| 1094 | |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1095 | static int blkiocg_file_write(struct cgroup *cgrp, struct cftype *cft, |
| 1096 | const char *buffer) |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1097 | { |
| 1098 | int ret = 0; |
| 1099 | char *buf; |
Tejun Heo | e56da7e | 2012-03-05 13:15:07 -0800 | [diff] [blame] | 1100 | struct blkio_cgroup *blkcg = cgroup_to_blkio_cgroup(cgrp); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1101 | enum blkio_policy_id plid = BLKIOFILE_POLICY(cft->private); |
| 1102 | int fileid = BLKIOFILE_ATTR(cft->private); |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1103 | |
| 1104 | buf = kstrdup(buffer, GFP_KERNEL); |
| 1105 | if (!buf) |
| 1106 | return -ENOMEM; |
| 1107 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1108 | ret = blkio_policy_parse_and_set(buf, plid, fileid, blkcg); |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1109 | kfree(buf); |
| 1110 | return ret; |
| 1111 | } |
| 1112 | |
Vivek Goyal | 92616b5 | 2012-03-05 13:15:10 -0800 | [diff] [blame] | 1113 | static const char *blkg_dev_name(struct blkio_group *blkg) |
| 1114 | { |
| 1115 | /* some drivers (floppy) instantiate a queue w/o disk registered */ |
| 1116 | if (blkg->q->backing_dev_info.dev) |
| 1117 | return dev_name(blkg->q->backing_dev_info.dev); |
| 1118 | return NULL; |
| 1119 | } |
| 1120 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1121 | static void blkio_print_group_conf(struct cftype *cft, struct blkio_group *blkg, |
| 1122 | struct seq_file *m) |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1123 | { |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1124 | int plid = BLKIOFILE_POLICY(cft->private); |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1125 | int fileid = BLKIOFILE_ATTR(cft->private); |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1126 | struct blkg_policy_data *pd = blkg->pd[plid]; |
| 1127 | const char *dname = blkg_dev_name(blkg); |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1128 | int rw = WRITE; |
| 1129 | |
Vivek Goyal | 92616b5 | 2012-03-05 13:15:10 -0800 | [diff] [blame] | 1130 | if (!dname) |
| 1131 | return; |
| 1132 | |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1133 | switch (plid) { |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1134 | case BLKIO_POLICY_PROP: |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1135 | if (pd->conf.weight) |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 1136 | seq_printf(m, "%s\t%u\n", |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1137 | dname, pd->conf.weight); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1138 | break; |
| 1139 | case BLKIO_POLICY_THROTL: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1140 | switch (fileid) { |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1141 | case BLKIO_THROTL_read_bps_device: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1142 | rw = READ; |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1143 | case BLKIO_THROTL_write_bps_device: |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1144 | if (pd->conf.bps[rw]) |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 1145 | seq_printf(m, "%s\t%llu\n", |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1146 | dname, pd->conf.bps[rw]); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1147 | break; |
| 1148 | case BLKIO_THROTL_read_iops_device: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1149 | rw = READ; |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1150 | case BLKIO_THROTL_write_iops_device: |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1151 | if (pd->conf.iops[rw]) |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 1152 | seq_printf(m, "%s\t%u\n", |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1153 | dname, pd->conf.iops[rw]); |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1154 | break; |
| 1155 | } |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1156 | break; |
| 1157 | default: |
| 1158 | BUG(); |
| 1159 | } |
| 1160 | } |
| 1161 | |
| 1162 | /* cgroup files which read their data from policy nodes end up here */ |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1163 | static void blkio_read_conf(struct cftype *cft, struct blkio_cgroup *blkcg, |
| 1164 | struct seq_file *m) |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1165 | { |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1166 | struct blkio_group *blkg; |
| 1167 | struct hlist_node *n; |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1168 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1169 | spin_lock_irq(&blkcg->lock); |
| 1170 | hlist_for_each_entry(blkg, n, &blkcg->blkg_list, blkcg_node) |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1171 | blkio_print_group_conf(cft, blkg, m); |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1172 | spin_unlock_irq(&blkcg->lock); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1173 | } |
| 1174 | |
| 1175 | static int blkiocg_file_read(struct cgroup *cgrp, struct cftype *cft, |
| 1176 | struct seq_file *m) |
| 1177 | { |
| 1178 | struct blkio_cgroup *blkcg; |
| 1179 | enum blkio_policy_id plid = BLKIOFILE_POLICY(cft->private); |
| 1180 | int name = BLKIOFILE_ATTR(cft->private); |
| 1181 | |
| 1182 | blkcg = cgroup_to_blkio_cgroup(cgrp); |
| 1183 | |
| 1184 | switch(plid) { |
| 1185 | case BLKIO_POLICY_PROP: |
| 1186 | switch(name) { |
| 1187 | case BLKIO_PROP_weight_device: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1188 | blkio_read_conf(cft, blkcg, m); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1189 | return 0; |
| 1190 | default: |
| 1191 | BUG(); |
| 1192 | } |
| 1193 | break; |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1194 | case BLKIO_POLICY_THROTL: |
| 1195 | switch(name){ |
| 1196 | case BLKIO_THROTL_read_bps_device: |
| 1197 | case BLKIO_THROTL_write_bps_device: |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1198 | case BLKIO_THROTL_read_iops_device: |
| 1199 | case BLKIO_THROTL_write_iops_device: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1200 | blkio_read_conf(cft, blkcg, m); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1201 | return 0; |
| 1202 | default: |
| 1203 | BUG(); |
| 1204 | } |
| 1205 | break; |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1206 | default: |
| 1207 | BUG(); |
| 1208 | } |
| 1209 | |
| 1210 | return 0; |
| 1211 | } |
| 1212 | |
| 1213 | static int blkio_read_blkg_stats(struct blkio_cgroup *blkcg, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1214 | struct cftype *cft, struct cgroup_map_cb *cb, |
| 1215 | enum stat_type type, bool show_total, bool pcpu) |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1216 | { |
| 1217 | struct blkio_group *blkg; |
| 1218 | struct hlist_node *n; |
| 1219 | uint64_t cgroup_total = 0; |
| 1220 | |
Tejun Heo | c875f4d | 2012-03-05 13:15:22 -0800 | [diff] [blame] | 1221 | spin_lock_irq(&blkcg->lock); |
| 1222 | |
| 1223 | hlist_for_each_entry(blkg, n, &blkcg->blkg_list, blkcg_node) { |
Vivek Goyal | 92616b5 | 2012-03-05 13:15:10 -0800 | [diff] [blame] | 1224 | const char *dname = blkg_dev_name(blkg); |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1225 | int plid = BLKIOFILE_POLICY(cft->private); |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 1226 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1227 | if (!dname) |
Tejun Heo | 7a4dd28 | 2012-03-05 13:15:09 -0800 | [diff] [blame] | 1228 | continue; |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 1229 | if (pcpu) |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1230 | cgroup_total += blkio_get_stat_cpu(blkg, plid, |
| 1231 | cb, dname, type); |
Tejun Heo | edf1b87 | 2012-03-08 10:54:00 -0800 | [diff] [blame] | 1232 | else |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1233 | cgroup_total += blkio_get_stat(blkg, plid, |
| 1234 | cb, dname, type); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1235 | } |
| 1236 | if (show_total) |
| 1237 | cb->fill(cb, "Total", cgroup_total); |
Tejun Heo | c875f4d | 2012-03-05 13:15:22 -0800 | [diff] [blame] | 1238 | |
| 1239 | spin_unlock_irq(&blkcg->lock); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1240 | return 0; |
| 1241 | } |
| 1242 | |
| 1243 | /* All map kind of cgroup file get serviced by this function */ |
| 1244 | static int blkiocg_file_read_map(struct cgroup *cgrp, struct cftype *cft, |
| 1245 | struct cgroup_map_cb *cb) |
| 1246 | { |
| 1247 | struct blkio_cgroup *blkcg; |
| 1248 | enum blkio_policy_id plid = BLKIOFILE_POLICY(cft->private); |
| 1249 | int name = BLKIOFILE_ATTR(cft->private); |
| 1250 | |
| 1251 | blkcg = cgroup_to_blkio_cgroup(cgrp); |
| 1252 | |
| 1253 | switch(plid) { |
| 1254 | case BLKIO_POLICY_PROP: |
| 1255 | switch(name) { |
| 1256 | case BLKIO_PROP_time: |
| 1257 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1258 | BLKIO_STAT_TIME, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1259 | case BLKIO_PROP_sectors: |
| 1260 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1261 | BLKIO_STAT_CPU_SECTORS, 0, 1); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1262 | case BLKIO_PROP_io_service_bytes: |
| 1263 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1264 | BLKIO_STAT_CPU_SERVICE_BYTES, 1, 1); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1265 | case BLKIO_PROP_io_serviced: |
| 1266 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1267 | BLKIO_STAT_CPU_SERVICED, 1, 1); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1268 | case BLKIO_PROP_io_service_time: |
| 1269 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1270 | BLKIO_STAT_SERVICE_TIME, 1, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1271 | case BLKIO_PROP_io_wait_time: |
| 1272 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1273 | BLKIO_STAT_WAIT_TIME, 1, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1274 | case BLKIO_PROP_io_merged: |
| 1275 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Tejun Heo | 5fe224d | 2012-03-08 10:53:57 -0800 | [diff] [blame] | 1276 | BLKIO_STAT_MERGED, 1, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1277 | case BLKIO_PROP_io_queued: |
| 1278 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1279 | BLKIO_STAT_QUEUED, 1, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1280 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Justin TerAvest | 9026e52 | 2011-03-22 21:26:54 +0100 | [diff] [blame] | 1281 | case BLKIO_PROP_unaccounted_time: |
| 1282 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1283 | BLKIO_STAT_UNACCOUNTED_TIME, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1284 | case BLKIO_PROP_dequeue: |
| 1285 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1286 | BLKIO_STAT_DEQUEUE, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1287 | case BLKIO_PROP_avg_queue_size: |
| 1288 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1289 | BLKIO_STAT_AVG_QUEUE_SIZE, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1290 | case BLKIO_PROP_group_wait_time: |
| 1291 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1292 | BLKIO_STAT_GROUP_WAIT_TIME, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1293 | case BLKIO_PROP_idle_time: |
| 1294 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1295 | BLKIO_STAT_IDLE_TIME, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1296 | case BLKIO_PROP_empty_time: |
| 1297 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1298 | BLKIO_STAT_EMPTY_TIME, 0, 0); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1299 | #endif |
| 1300 | default: |
| 1301 | BUG(); |
| 1302 | } |
| 1303 | break; |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1304 | case BLKIO_POLICY_THROTL: |
| 1305 | switch(name){ |
| 1306 | case BLKIO_THROTL_io_service_bytes: |
| 1307 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1308 | BLKIO_STAT_CPU_SERVICE_BYTES, 1, 1); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1309 | case BLKIO_THROTL_io_serviced: |
| 1310 | return blkio_read_blkg_stats(blkcg, cft, cb, |
Vivek Goyal | 5624a4e | 2011-05-19 15:38:28 -0400 | [diff] [blame] | 1311 | BLKIO_STAT_CPU_SERVICED, 1, 1); |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1312 | default: |
| 1313 | BUG(); |
| 1314 | } |
| 1315 | break; |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1316 | default: |
| 1317 | BUG(); |
| 1318 | } |
| 1319 | |
| 1320 | return 0; |
| 1321 | } |
| 1322 | |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1323 | static int blkio_weight_write(struct blkio_cgroup *blkcg, int plid, u64 val) |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1324 | { |
| 1325 | struct blkio_group *blkg; |
| 1326 | struct hlist_node *n; |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1327 | |
| 1328 | if (val < BLKIO_WEIGHT_MIN || val > BLKIO_WEIGHT_MAX) |
| 1329 | return -EINVAL; |
| 1330 | |
| 1331 | spin_lock(&blkio_list_lock); |
| 1332 | spin_lock_irq(&blkcg->lock); |
| 1333 | blkcg->weight = (unsigned int)val; |
| 1334 | |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1335 | hlist_for_each_entry(blkg, n, &blkcg->blkg_list, blkcg_node) { |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1336 | struct blkg_policy_data *pd = blkg->pd[plid]; |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1337 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1338 | if (!pd->conf.weight) |
Tejun Heo | c176826 | 2012-03-05 13:15:17 -0800 | [diff] [blame] | 1339 | blkio_update_group_weight(blkg, plid, blkcg->weight); |
Tejun Heo | 549d3aa | 2012-03-05 13:15:16 -0800 | [diff] [blame] | 1340 | } |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1341 | |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1342 | spin_unlock_irq(&blkcg->lock); |
| 1343 | spin_unlock(&blkio_list_lock); |
| 1344 | return 0; |
| 1345 | } |
| 1346 | |
| 1347 | static u64 blkiocg_file_read_u64 (struct cgroup *cgrp, struct cftype *cft) { |
| 1348 | struct blkio_cgroup *blkcg; |
| 1349 | enum blkio_policy_id plid = BLKIOFILE_POLICY(cft->private); |
| 1350 | int name = BLKIOFILE_ATTR(cft->private); |
| 1351 | |
| 1352 | blkcg = cgroup_to_blkio_cgroup(cgrp); |
| 1353 | |
| 1354 | switch(plid) { |
| 1355 | case BLKIO_POLICY_PROP: |
| 1356 | switch(name) { |
| 1357 | case BLKIO_PROP_weight: |
| 1358 | return (u64)blkcg->weight; |
| 1359 | } |
| 1360 | break; |
| 1361 | default: |
| 1362 | BUG(); |
| 1363 | } |
| 1364 | return 0; |
| 1365 | } |
| 1366 | |
| 1367 | static int |
| 1368 | blkiocg_file_write_u64(struct cgroup *cgrp, struct cftype *cft, u64 val) |
| 1369 | { |
| 1370 | struct blkio_cgroup *blkcg; |
| 1371 | enum blkio_policy_id plid = BLKIOFILE_POLICY(cft->private); |
| 1372 | int name = BLKIOFILE_ATTR(cft->private); |
| 1373 | |
| 1374 | blkcg = cgroup_to_blkio_cgroup(cgrp); |
| 1375 | |
| 1376 | switch(plid) { |
| 1377 | case BLKIO_POLICY_PROP: |
| 1378 | switch(name) { |
| 1379 | case BLKIO_PROP_weight: |
Tejun Heo | 4bfd482 | 2012-03-05 13:15:08 -0800 | [diff] [blame] | 1380 | return blkio_weight_write(blkcg, plid, val); |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1381 | } |
| 1382 | break; |
| 1383 | default: |
| 1384 | BUG(); |
| 1385 | } |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1386 | |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1387 | return 0; |
| 1388 | } |
| 1389 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1390 | struct cftype blkio_files[] = { |
| 1391 | { |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1392 | .name = "weight_device", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1393 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1394 | BLKIO_PROP_weight_device), |
| 1395 | .read_seq_string = blkiocg_file_read, |
| 1396 | .write_string = blkiocg_file_write, |
Gui Jianfeng | 34d0f17 | 2010-04-13 16:05:49 +0800 | [diff] [blame] | 1397 | .max_write_len = 256, |
| 1398 | }, |
| 1399 | { |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1400 | .name = "weight", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1401 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1402 | BLKIO_PROP_weight), |
| 1403 | .read_u64 = blkiocg_file_read_u64, |
| 1404 | .write_u64 = blkiocg_file_write_u64, |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1405 | }, |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1406 | { |
| 1407 | .name = "time", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1408 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1409 | BLKIO_PROP_time), |
| 1410 | .read_map = blkiocg_file_read_map, |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1411 | }, |
| 1412 | { |
| 1413 | .name = "sectors", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1414 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1415 | BLKIO_PROP_sectors), |
| 1416 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 1417 | }, |
| 1418 | { |
| 1419 | .name = "io_service_bytes", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1420 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1421 | BLKIO_PROP_io_service_bytes), |
| 1422 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 1423 | }, |
| 1424 | { |
| 1425 | .name = "io_serviced", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1426 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1427 | BLKIO_PROP_io_serviced), |
| 1428 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 1429 | }, |
| 1430 | { |
| 1431 | .name = "io_service_time", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1432 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1433 | BLKIO_PROP_io_service_time), |
| 1434 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 303a3ac | 2010-04-01 15:01:24 -0700 | [diff] [blame] | 1435 | }, |
| 1436 | { |
| 1437 | .name = "io_wait_time", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1438 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1439 | BLKIO_PROP_io_wait_time), |
| 1440 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 1441 | }, |
| 1442 | { |
Divyesh Shah | 812d402 | 2010-04-08 21:14:23 -0700 | [diff] [blame] | 1443 | .name = "io_merged", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1444 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1445 | BLKIO_PROP_io_merged), |
| 1446 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 812d402 | 2010-04-08 21:14:23 -0700 | [diff] [blame] | 1447 | }, |
| 1448 | { |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 1449 | .name = "io_queued", |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1450 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1451 | BLKIO_PROP_io_queued), |
| 1452 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 1453 | }, |
| 1454 | { |
Divyesh Shah | 84c124d | 2010-04-09 08:31:19 +0200 | [diff] [blame] | 1455 | .name = "reset_stats", |
| 1456 | .write_u64 = blkiocg_reset_stats, |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1457 | }, |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1458 | #ifdef CONFIG_BLK_DEV_THROTTLING |
| 1459 | { |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1460 | .name = "throttle.read_bps_device", |
| 1461 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1462 | BLKIO_THROTL_read_bps_device), |
| 1463 | .read_seq_string = blkiocg_file_read, |
| 1464 | .write_string = blkiocg_file_write, |
| 1465 | .max_write_len = 256, |
| 1466 | }, |
| 1467 | |
| 1468 | { |
| 1469 | .name = "throttle.write_bps_device", |
| 1470 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1471 | BLKIO_THROTL_write_bps_device), |
| 1472 | .read_seq_string = blkiocg_file_read, |
| 1473 | .write_string = blkiocg_file_write, |
| 1474 | .max_write_len = 256, |
| 1475 | }, |
Vivek Goyal | 7702e8f | 2010-09-15 17:06:36 -0400 | [diff] [blame] | 1476 | |
| 1477 | { |
| 1478 | .name = "throttle.read_iops_device", |
| 1479 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1480 | BLKIO_THROTL_read_iops_device), |
| 1481 | .read_seq_string = blkiocg_file_read, |
| 1482 | .write_string = blkiocg_file_write, |
| 1483 | .max_write_len = 256, |
| 1484 | }, |
| 1485 | |
| 1486 | { |
| 1487 | .name = "throttle.write_iops_device", |
| 1488 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1489 | BLKIO_THROTL_write_iops_device), |
| 1490 | .read_seq_string = blkiocg_file_read, |
| 1491 | .write_string = blkiocg_file_write, |
| 1492 | .max_write_len = 256, |
| 1493 | }, |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1494 | { |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1495 | .name = "throttle.io_service_bytes", |
| 1496 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1497 | BLKIO_THROTL_io_service_bytes), |
| 1498 | .read_map = blkiocg_file_read_map, |
| 1499 | }, |
| 1500 | { |
Vivek Goyal | 4c9eefa | 2010-09-15 17:06:34 -0400 | [diff] [blame] | 1501 | .name = "throttle.io_serviced", |
| 1502 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_THROTL, |
| 1503 | BLKIO_THROTL_io_serviced), |
| 1504 | .read_map = blkiocg_file_read_map, |
| 1505 | }, |
Vivek Goyal | 13f9825 | 2010-10-01 14:49:41 +0200 | [diff] [blame] | 1506 | #endif /* CONFIG_BLK_DEV_THROTTLING */ |
| 1507 | |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1508 | #ifdef CONFIG_DEBUG_BLK_CGROUP |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 1509 | { |
| 1510 | .name = "avg_queue_size", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1511 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1512 | BLKIO_PROP_avg_queue_size), |
| 1513 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 1514 | }, |
| 1515 | { |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 1516 | .name = "group_wait_time", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1517 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1518 | BLKIO_PROP_group_wait_time), |
| 1519 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 1520 | }, |
| 1521 | { |
| 1522 | .name = "idle_time", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1523 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1524 | BLKIO_PROP_idle_time), |
| 1525 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 1526 | }, |
| 1527 | { |
| 1528 | .name = "empty_time", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1529 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1530 | BLKIO_PROP_empty_time), |
| 1531 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | 812df48 | 2010-04-08 21:15:35 -0700 | [diff] [blame] | 1532 | }, |
| 1533 | { |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1534 | .name = "dequeue", |
Vivek Goyal | 062a644 | 2010-09-15 17:06:33 -0400 | [diff] [blame] | 1535 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1536 | BLKIO_PROP_dequeue), |
| 1537 | .read_map = blkiocg_file_read_map, |
Divyesh Shah | cdc1184 | 2010-04-08 21:15:10 -0700 | [diff] [blame] | 1538 | }, |
Justin TerAvest | 9026e52 | 2011-03-22 21:26:54 +0100 | [diff] [blame] | 1539 | { |
| 1540 | .name = "unaccounted_time", |
| 1541 | .private = BLKIOFILE_PRIVATE(BLKIO_POLICY_PROP, |
| 1542 | BLKIO_PROP_unaccounted_time), |
| 1543 | .read_map = blkiocg_file_read_map, |
| 1544 | }, |
Vivek Goyal | 2208419 | 2009-12-03 12:59:49 -0500 | [diff] [blame] | 1545 | #endif |
Tejun Heo | 4baf6e3 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1546 | { } /* terminate */ |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1547 | }; |
| 1548 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1549 | /** |
| 1550 | * blkiocg_pre_destroy - cgroup pre_destroy callback |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1551 | * @cgroup: cgroup of interest |
| 1552 | * |
| 1553 | * This function is called when @cgroup is about to go away and responsible |
| 1554 | * for shooting down all blkgs associated with @cgroup. blkgs should be |
| 1555 | * removed while holding both q and blkcg locks. As blkcg lock is nested |
| 1556 | * inside q lock, this function performs reverse double lock dancing. |
| 1557 | * |
| 1558 | * This is the blkcg counterpart of ioc_release_fn(). |
| 1559 | */ |
Tejun Heo | 959d851 | 2012-04-01 12:30:01 -0700 | [diff] [blame^] | 1560 | static int blkiocg_pre_destroy(struct cgroup *cgroup) |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1561 | { |
| 1562 | struct blkio_cgroup *blkcg = cgroup_to_blkio_cgroup(cgroup); |
| 1563 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1564 | spin_lock_irq(&blkcg->lock); |
Tejun Heo | 7ee9c56 | 2012-03-05 13:15:11 -0800 | [diff] [blame] | 1565 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1566 | while (!hlist_empty(&blkcg->blkg_list)) { |
| 1567 | struct blkio_group *blkg = hlist_entry(blkcg->blkg_list.first, |
| 1568 | struct blkio_group, blkcg_node); |
Tejun Heo | c875f4d | 2012-03-05 13:15:22 -0800 | [diff] [blame] | 1569 | struct request_queue *q = blkg->q; |
Vivek Goyal | b1c3576 | 2009-12-03 12:59:47 -0500 | [diff] [blame] | 1570 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1571 | if (spin_trylock(q->queue_lock)) { |
| 1572 | blkg_destroy(blkg); |
| 1573 | spin_unlock(q->queue_lock); |
| 1574 | } else { |
| 1575 | spin_unlock_irq(&blkcg->lock); |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1576 | cpu_relax(); |
Dan Carpenter | a556793 | 2012-03-29 20:57:08 +0200 | [diff] [blame] | 1577 | spin_lock_irq(&blkcg->lock); |
Jens Axboe | 0f3942a | 2010-05-03 14:28:55 +0200 | [diff] [blame] | 1578 | } |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1579 | } |
Jens Axboe | 0f3942a | 2010-05-03 14:28:55 +0200 | [diff] [blame] | 1580 | |
Tejun Heo | 9f13ef6 | 2012-03-05 13:15:21 -0800 | [diff] [blame] | 1581 | spin_unlock_irq(&blkcg->lock); |
Tejun Heo | 7ee9c56 | 2012-03-05 13:15:11 -0800 | [diff] [blame] | 1582 | return 0; |
| 1583 | } |
| 1584 | |
Li Zefan | 761b3ef | 2012-01-31 13:47:36 +0800 | [diff] [blame] | 1585 | static void blkiocg_destroy(struct cgroup *cgroup) |
Tejun Heo | 7ee9c56 | 2012-03-05 13:15:11 -0800 | [diff] [blame] | 1586 | { |
| 1587 | struct blkio_cgroup *blkcg = cgroup_to_blkio_cgroup(cgroup); |
| 1588 | |
Ben Blum | 67523c4 | 2010-03-10 15:22:11 -0800 | [diff] [blame] | 1589 | if (blkcg != &blkio_root_cgroup) |
| 1590 | kfree(blkcg); |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1591 | } |
| 1592 | |
Li Zefan | 761b3ef | 2012-01-31 13:47:36 +0800 | [diff] [blame] | 1593 | static struct cgroup_subsys_state *blkiocg_create(struct cgroup *cgroup) |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1594 | { |
Tejun Heo | 9a9e8a2 | 2012-03-19 15:10:56 -0700 | [diff] [blame] | 1595 | static atomic64_t id_seq = ATOMIC64_INIT(0); |
Li Zefan | 0341509 | 2010-05-07 08:57:00 +0200 | [diff] [blame] | 1596 | struct blkio_cgroup *blkcg; |
| 1597 | struct cgroup *parent = cgroup->parent; |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1598 | |
Li Zefan | 0341509 | 2010-05-07 08:57:00 +0200 | [diff] [blame] | 1599 | if (!parent) { |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1600 | blkcg = &blkio_root_cgroup; |
| 1601 | goto done; |
| 1602 | } |
| 1603 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1604 | blkcg = kzalloc(sizeof(*blkcg), GFP_KERNEL); |
| 1605 | if (!blkcg) |
| 1606 | return ERR_PTR(-ENOMEM); |
| 1607 | |
| 1608 | blkcg->weight = BLKIO_WEIGHT_DEFAULT; |
Tejun Heo | 9a9e8a2 | 2012-03-19 15:10:56 -0700 | [diff] [blame] | 1609 | blkcg->id = atomic64_inc_return(&id_seq); /* root is 0, start from 1 */ |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1610 | done: |
| 1611 | spin_lock_init(&blkcg->lock); |
| 1612 | INIT_HLIST_HEAD(&blkcg->blkg_list); |
| 1613 | |
| 1614 | return &blkcg->css; |
| 1615 | } |
| 1616 | |
Tejun Heo | 5efd611 | 2012-03-05 13:15:12 -0800 | [diff] [blame] | 1617 | /** |
| 1618 | * blkcg_init_queue - initialize blkcg part of request queue |
| 1619 | * @q: request_queue to initialize |
| 1620 | * |
| 1621 | * Called from blk_alloc_queue_node(). Responsible for initializing blkcg |
| 1622 | * part of new request_queue @q. |
| 1623 | * |
| 1624 | * RETURNS: |
| 1625 | * 0 on success, -errno on failure. |
| 1626 | */ |
| 1627 | int blkcg_init_queue(struct request_queue *q) |
| 1628 | { |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1629 | int ret; |
| 1630 | |
Tejun Heo | 5efd611 | 2012-03-05 13:15:12 -0800 | [diff] [blame] | 1631 | might_sleep(); |
| 1632 | |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1633 | ret = blk_throtl_init(q); |
| 1634 | if (ret) |
| 1635 | return ret; |
| 1636 | |
| 1637 | mutex_lock(&all_q_mutex); |
| 1638 | INIT_LIST_HEAD(&q->all_q_node); |
| 1639 | list_add_tail(&q->all_q_node, &all_q_list); |
| 1640 | mutex_unlock(&all_q_mutex); |
| 1641 | |
| 1642 | return 0; |
Tejun Heo | 5efd611 | 2012-03-05 13:15:12 -0800 | [diff] [blame] | 1643 | } |
| 1644 | |
| 1645 | /** |
| 1646 | * blkcg_drain_queue - drain blkcg part of request_queue |
| 1647 | * @q: request_queue to drain |
| 1648 | * |
| 1649 | * Called from blk_drain_queue(). Responsible for draining blkcg part. |
| 1650 | */ |
| 1651 | void blkcg_drain_queue(struct request_queue *q) |
| 1652 | { |
| 1653 | lockdep_assert_held(q->queue_lock); |
| 1654 | |
| 1655 | blk_throtl_drain(q); |
| 1656 | } |
| 1657 | |
| 1658 | /** |
| 1659 | * blkcg_exit_queue - exit and release blkcg part of request_queue |
| 1660 | * @q: request_queue being released |
| 1661 | * |
| 1662 | * Called from blk_release_queue(). Responsible for exiting blkcg part. |
| 1663 | */ |
| 1664 | void blkcg_exit_queue(struct request_queue *q) |
| 1665 | { |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1666 | mutex_lock(&all_q_mutex); |
| 1667 | list_del_init(&q->all_q_node); |
| 1668 | mutex_unlock(&all_q_mutex); |
| 1669 | |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1670 | blkg_destroy_all(q, true); |
| 1671 | |
Tejun Heo | 5efd611 | 2012-03-05 13:15:12 -0800 | [diff] [blame] | 1672 | blk_throtl_exit(q); |
| 1673 | } |
| 1674 | |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1675 | /* |
| 1676 | * We cannot support shared io contexts, as we have no mean to support |
| 1677 | * two tasks with the same ioc in two different groups without major rework |
| 1678 | * of the main cic data structures. For now we allow a task to change |
| 1679 | * its cgroup only if it's the only owner of its ioc. |
| 1680 | */ |
Li Zefan | 761b3ef | 2012-01-31 13:47:36 +0800 | [diff] [blame] | 1681 | static int blkiocg_can_attach(struct cgroup *cgrp, struct cgroup_taskset *tset) |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1682 | { |
Tejun Heo | bb9d97b | 2011-12-12 18:12:21 -0800 | [diff] [blame] | 1683 | struct task_struct *task; |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1684 | struct io_context *ioc; |
| 1685 | int ret = 0; |
| 1686 | |
| 1687 | /* task_lock() is needed to avoid races with exit_io_context() */ |
Tejun Heo | bb9d97b | 2011-12-12 18:12:21 -0800 | [diff] [blame] | 1688 | cgroup_taskset_for_each(task, cgrp, tset) { |
| 1689 | task_lock(task); |
| 1690 | ioc = task->io_context; |
| 1691 | if (ioc && atomic_read(&ioc->nr_tasks) > 1) |
| 1692 | ret = -EINVAL; |
| 1693 | task_unlock(task); |
| 1694 | if (ret) |
| 1695 | break; |
| 1696 | } |
Vivek Goyal | 31e4c28 | 2009-12-03 12:59:42 -0500 | [diff] [blame] | 1697 | return ret; |
| 1698 | } |
| 1699 | |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1700 | static void blkcg_bypass_start(void) |
| 1701 | __acquires(&all_q_mutex) |
| 1702 | { |
| 1703 | struct request_queue *q; |
| 1704 | |
| 1705 | mutex_lock(&all_q_mutex); |
| 1706 | |
| 1707 | list_for_each_entry(q, &all_q_list, all_q_node) { |
| 1708 | blk_queue_bypass_start(q); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1709 | blkg_destroy_all(q, false); |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1710 | } |
| 1711 | } |
| 1712 | |
| 1713 | static void blkcg_bypass_end(void) |
| 1714 | __releases(&all_q_mutex) |
| 1715 | { |
| 1716 | struct request_queue *q; |
| 1717 | |
| 1718 | list_for_each_entry(q, &all_q_list, all_q_node) |
| 1719 | blk_queue_bypass_end(q); |
| 1720 | |
| 1721 | mutex_unlock(&all_q_mutex); |
| 1722 | } |
| 1723 | |
Tejun Heo | 676f7c8 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1724 | struct cgroup_subsys blkio_subsys = { |
| 1725 | .name = "blkio", |
| 1726 | .create = blkiocg_create, |
| 1727 | .can_attach = blkiocg_can_attach, |
Tejun Heo | 959d851 | 2012-04-01 12:30:01 -0700 | [diff] [blame^] | 1728 | .pre_destroy = blkiocg_pre_destroy, |
Tejun Heo | 676f7c8 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1729 | .destroy = blkiocg_destroy, |
Tejun Heo | 676f7c8 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1730 | .subsys_id = blkio_subsys_id, |
Tejun Heo | 4baf6e3 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1731 | .base_cftypes = blkio_files, |
Tejun Heo | 676f7c8 | 2012-04-01 12:09:55 -0700 | [diff] [blame] | 1732 | .module = THIS_MODULE, |
| 1733 | }; |
| 1734 | EXPORT_SYMBOL_GPL(blkio_subsys); |
| 1735 | |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1736 | void blkio_policy_register(struct blkio_policy_type *blkiop) |
| 1737 | { |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1738 | struct request_queue *q; |
| 1739 | |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1740 | blkcg_bypass_start(); |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1741 | spin_lock(&blkio_list_lock); |
Tejun Heo | 035d10b | 2012-03-05 13:15:04 -0800 | [diff] [blame] | 1742 | |
| 1743 | BUG_ON(blkio_policy[blkiop->plid]); |
| 1744 | blkio_policy[blkiop->plid] = blkiop; |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1745 | list_add_tail(&blkiop->list, &blkio_list); |
Tejun Heo | 035d10b | 2012-03-05 13:15:04 -0800 | [diff] [blame] | 1746 | |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1747 | spin_unlock(&blkio_list_lock); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1748 | list_for_each_entry(q, &all_q_list, all_q_node) |
| 1749 | update_root_blkg_pd(q, blkiop->plid); |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1750 | blkcg_bypass_end(); |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1751 | } |
| 1752 | EXPORT_SYMBOL_GPL(blkio_policy_register); |
| 1753 | |
| 1754 | void blkio_policy_unregister(struct blkio_policy_type *blkiop) |
| 1755 | { |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1756 | struct request_queue *q; |
| 1757 | |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1758 | blkcg_bypass_start(); |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1759 | spin_lock(&blkio_list_lock); |
Tejun Heo | 035d10b | 2012-03-05 13:15:04 -0800 | [diff] [blame] | 1760 | |
| 1761 | BUG_ON(blkio_policy[blkiop->plid] != blkiop); |
| 1762 | blkio_policy[blkiop->plid] = NULL; |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1763 | list_del_init(&blkiop->list); |
Tejun Heo | 035d10b | 2012-03-05 13:15:04 -0800 | [diff] [blame] | 1764 | |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1765 | spin_unlock(&blkio_list_lock); |
Tejun Heo | e8989fa | 2012-03-05 13:15:20 -0800 | [diff] [blame] | 1766 | list_for_each_entry(q, &all_q_list, all_q_node) |
| 1767 | update_root_blkg_pd(q, blkiop->plid); |
Tejun Heo | 923adde | 2012-03-05 13:15:13 -0800 | [diff] [blame] | 1768 | blkcg_bypass_end(); |
Vivek Goyal | 3e25206 | 2009-12-04 10:36:42 -0500 | [diff] [blame] | 1769 | } |
| 1770 | EXPORT_SYMBOL_GPL(blkio_policy_unregister); |