]>
Commit | Line | Data |
---|---|---|
a497ee34 | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
ea25da48 PV |
2 | /* |
3 | * cgroups support for the BFQ I/O scheduler. | |
ea25da48 PV |
4 | */ |
5 | #include <linux/module.h> | |
6 | #include <linux/slab.h> | |
7 | #include <linux/blkdev.h> | |
8 | #include <linux/cgroup.h> | |
ea25da48 PV |
9 | #include <linux/ktime.h> |
10 | #include <linux/rbtree.h> | |
11 | #include <linux/ioprio.h> | |
12 | #include <linux/sbitmap.h> | |
13 | #include <linux/delay.h> | |
14 | ||
2e9bc346 | 15 | #include "elevator.h" |
ea25da48 PV |
16 | #include "bfq-iosched.h" |
17 | ||
8060c47b | 18 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
c0ce79dc CH |
19 | static int bfq_stat_init(struct bfq_stat *stat, gfp_t gfp) |
20 | { | |
21 | int ret; | |
22 | ||
23 | ret = percpu_counter_init(&stat->cpu_cnt, 0, gfp); | |
24 | if (ret) | |
25 | return ret; | |
26 | ||
27 | atomic64_set(&stat->aux_cnt, 0); | |
28 | return 0; | |
29 | } | |
30 | ||
31 | static void bfq_stat_exit(struct bfq_stat *stat) | |
32 | { | |
33 | percpu_counter_destroy(&stat->cpu_cnt); | |
34 | } | |
35 | ||
36 | /** | |
37 | * bfq_stat_add - add a value to a bfq_stat | |
38 | * @stat: target bfq_stat | |
39 | * @val: value to add | |
40 | * | |
41 | * Add @val to @stat. The caller must ensure that IRQ on the same CPU | |
42 | * don't re-enter this function for the same counter. | |
43 | */ | |
44 | static inline void bfq_stat_add(struct bfq_stat *stat, uint64_t val) | |
45 | { | |
46 | percpu_counter_add_batch(&stat->cpu_cnt, val, BLKG_STAT_CPU_BATCH); | |
47 | } | |
48 | ||
49 | /** | |
50 | * bfq_stat_read - read the current value of a bfq_stat | |
51 | * @stat: bfq_stat to read | |
52 | */ | |
53 | static inline uint64_t bfq_stat_read(struct bfq_stat *stat) | |
54 | { | |
55 | return percpu_counter_sum_positive(&stat->cpu_cnt); | |
56 | } | |
57 | ||
58 | /** | |
59 | * bfq_stat_reset - reset a bfq_stat | |
60 | * @stat: bfq_stat to reset | |
61 | */ | |
62 | static inline void bfq_stat_reset(struct bfq_stat *stat) | |
63 | { | |
64 | percpu_counter_set(&stat->cpu_cnt, 0); | |
65 | atomic64_set(&stat->aux_cnt, 0); | |
66 | } | |
67 | ||
68 | /** | |
69 | * bfq_stat_add_aux - add a bfq_stat into another's aux count | |
70 | * @to: the destination bfq_stat | |
71 | * @from: the source | |
72 | * | |
73 | * Add @from's count including the aux one to @to's aux count. | |
74 | */ | |
75 | static inline void bfq_stat_add_aux(struct bfq_stat *to, | |
76 | struct bfq_stat *from) | |
77 | { | |
78 | atomic64_add(bfq_stat_read(from) + atomic64_read(&from->aux_cnt), | |
79 | &to->aux_cnt); | |
80 | } | |
81 | ||
c0ce79dc CH |
82 | /** |
83 | * blkg_prfill_stat - prfill callback for bfq_stat | |
84 | * @sf: seq_file to print to | |
85 | * @pd: policy private data of interest | |
86 | * @off: offset to the bfq_stat in @pd | |
87 | * | |
88 | * prfill callback for printing a bfq_stat. | |
89 | */ | |
90 | static u64 blkg_prfill_stat(struct seq_file *sf, struct blkg_policy_data *pd, | |
91 | int off) | |
92 | { | |
93 | return __blkg_prfill_u64(sf, pd, bfq_stat_read((void *)pd + off)); | |
94 | } | |
95 | ||
ea25da48 PV |
96 | /* bfqg stats flags */ |
97 | enum bfqg_stats_flags { | |
98 | BFQG_stats_waiting = 0, | |
99 | BFQG_stats_idling, | |
100 | BFQG_stats_empty, | |
101 | }; | |
102 | ||
103 | #define BFQG_FLAG_FNS(name) \ | |
104 | static void bfqg_stats_mark_##name(struct bfqg_stats *stats) \ | |
105 | { \ | |
106 | stats->flags |= (1 << BFQG_stats_##name); \ | |
107 | } \ | |
108 | static void bfqg_stats_clear_##name(struct bfqg_stats *stats) \ | |
109 | { \ | |
110 | stats->flags &= ~(1 << BFQG_stats_##name); \ | |
111 | } \ | |
112 | static int bfqg_stats_##name(struct bfqg_stats *stats) \ | |
113 | { \ | |
114 | return (stats->flags & (1 << BFQG_stats_##name)) != 0; \ | |
115 | } \ | |
116 | ||
117 | BFQG_FLAG_FNS(waiting) | |
118 | BFQG_FLAG_FNS(idling) | |
119 | BFQG_FLAG_FNS(empty) | |
120 | #undef BFQG_FLAG_FNS | |
121 | ||
8f9bebc3 | 122 | /* This should be called with the scheduler lock held. */ |
ea25da48 PV |
123 | static void bfqg_stats_update_group_wait_time(struct bfqg_stats *stats) |
124 | { | |
84c7afce | 125 | u64 now; |
ea25da48 PV |
126 | |
127 | if (!bfqg_stats_waiting(stats)) | |
128 | return; | |
129 | ||
84c7afce OS |
130 | now = ktime_get_ns(); |
131 | if (now > stats->start_group_wait_time) | |
c0ce79dc | 132 | bfq_stat_add(&stats->group_wait_time, |
ea25da48 PV |
133 | now - stats->start_group_wait_time); |
134 | bfqg_stats_clear_waiting(stats); | |
135 | } | |
136 | ||
8f9bebc3 | 137 | /* This should be called with the scheduler lock held. */ |
ea25da48 PV |
138 | static void bfqg_stats_set_start_group_wait_time(struct bfq_group *bfqg, |
139 | struct bfq_group *curr_bfqg) | |
140 | { | |
141 | struct bfqg_stats *stats = &bfqg->stats; | |
142 | ||
143 | if (bfqg_stats_waiting(stats)) | |
144 | return; | |
145 | if (bfqg == curr_bfqg) | |
146 | return; | |
84c7afce | 147 | stats->start_group_wait_time = ktime_get_ns(); |
ea25da48 PV |
148 | bfqg_stats_mark_waiting(stats); |
149 | } | |
150 | ||
8f9bebc3 | 151 | /* This should be called with the scheduler lock held. */ |
ea25da48 PV |
152 | static void bfqg_stats_end_empty_time(struct bfqg_stats *stats) |
153 | { | |
84c7afce | 154 | u64 now; |
ea25da48 PV |
155 | |
156 | if (!bfqg_stats_empty(stats)) | |
157 | return; | |
158 | ||
84c7afce OS |
159 | now = ktime_get_ns(); |
160 | if (now > stats->start_empty_time) | |
c0ce79dc | 161 | bfq_stat_add(&stats->empty_time, |
ea25da48 PV |
162 | now - stats->start_empty_time); |
163 | bfqg_stats_clear_empty(stats); | |
164 | } | |
165 | ||
166 | void bfqg_stats_update_dequeue(struct bfq_group *bfqg) | |
167 | { | |
c0ce79dc | 168 | bfq_stat_add(&bfqg->stats.dequeue, 1); |
ea25da48 PV |
169 | } |
170 | ||
171 | void bfqg_stats_set_start_empty_time(struct bfq_group *bfqg) | |
172 | { | |
173 | struct bfqg_stats *stats = &bfqg->stats; | |
174 | ||
175 | if (blkg_rwstat_total(&stats->queued)) | |
176 | return; | |
177 | ||
178 | /* | |
179 | * group is already marked empty. This can happen if bfqq got new | |
180 | * request in parent group and moved to this group while being added | |
181 | * to service tree. Just ignore the event and move on. | |
182 | */ | |
183 | if (bfqg_stats_empty(stats)) | |
184 | return; | |
185 | ||
84c7afce | 186 | stats->start_empty_time = ktime_get_ns(); |
ea25da48 PV |
187 | bfqg_stats_mark_empty(stats); |
188 | } | |
189 | ||
190 | void bfqg_stats_update_idle_time(struct bfq_group *bfqg) | |
191 | { | |
192 | struct bfqg_stats *stats = &bfqg->stats; | |
193 | ||
194 | if (bfqg_stats_idling(stats)) { | |
84c7afce | 195 | u64 now = ktime_get_ns(); |
ea25da48 | 196 | |
84c7afce | 197 | if (now > stats->start_idle_time) |
c0ce79dc | 198 | bfq_stat_add(&stats->idle_time, |
ea25da48 PV |
199 | now - stats->start_idle_time); |
200 | bfqg_stats_clear_idling(stats); | |
201 | } | |
202 | } | |
203 | ||
204 | void bfqg_stats_set_start_idle_time(struct bfq_group *bfqg) | |
205 | { | |
206 | struct bfqg_stats *stats = &bfqg->stats; | |
207 | ||
84c7afce | 208 | stats->start_idle_time = ktime_get_ns(); |
ea25da48 PV |
209 | bfqg_stats_mark_idling(stats); |
210 | } | |
211 | ||
212 | void bfqg_stats_update_avg_queue_size(struct bfq_group *bfqg) | |
213 | { | |
214 | struct bfqg_stats *stats = &bfqg->stats; | |
215 | ||
c0ce79dc | 216 | bfq_stat_add(&stats->avg_queue_size_sum, |
ea25da48 | 217 | blkg_rwstat_total(&stats->queued)); |
c0ce79dc | 218 | bfq_stat_add(&stats->avg_queue_size_samples, 1); |
ea25da48 PV |
219 | bfqg_stats_update_group_wait_time(stats); |
220 | } | |
221 | ||
a33801e8 LM |
222 | void bfqg_stats_update_io_add(struct bfq_group *bfqg, struct bfq_queue *bfqq, |
223 | unsigned int op) | |
224 | { | |
225 | blkg_rwstat_add(&bfqg->stats.queued, op, 1); | |
226 | bfqg_stats_end_empty_time(&bfqg->stats); | |
227 | if (!(bfqq == ((struct bfq_data *)bfqg->bfqd)->in_service_queue)) | |
228 | bfqg_stats_set_start_group_wait_time(bfqg, bfqq_group(bfqq)); | |
229 | } | |
230 | ||
231 | void bfqg_stats_update_io_remove(struct bfq_group *bfqg, unsigned int op) | |
232 | { | |
233 | blkg_rwstat_add(&bfqg->stats.queued, op, -1); | |
234 | } | |
235 | ||
236 | void bfqg_stats_update_io_merged(struct bfq_group *bfqg, unsigned int op) | |
237 | { | |
238 | blkg_rwstat_add(&bfqg->stats.merged, op, 1); | |
239 | } | |
240 | ||
84c7afce OS |
241 | void bfqg_stats_update_completion(struct bfq_group *bfqg, u64 start_time_ns, |
242 | u64 io_start_time_ns, unsigned int op) | |
a33801e8 LM |
243 | { |
244 | struct bfqg_stats *stats = &bfqg->stats; | |
84c7afce | 245 | u64 now = ktime_get_ns(); |
a33801e8 | 246 | |
84c7afce | 247 | if (now > io_start_time_ns) |
a33801e8 | 248 | blkg_rwstat_add(&stats->service_time, op, |
84c7afce OS |
249 | now - io_start_time_ns); |
250 | if (io_start_time_ns > start_time_ns) | |
a33801e8 | 251 | blkg_rwstat_add(&stats->wait_time, op, |
84c7afce | 252 | io_start_time_ns - start_time_ns); |
a33801e8 LM |
253 | } |
254 | ||
8060c47b | 255 | #else /* CONFIG_BFQ_CGROUP_DEBUG */ |
a33801e8 LM |
256 | |
257 | void bfqg_stats_update_io_add(struct bfq_group *bfqg, struct bfq_queue *bfqq, | |
258 | unsigned int op) { } | |
259 | void bfqg_stats_update_io_remove(struct bfq_group *bfqg, unsigned int op) { } | |
260 | void bfqg_stats_update_io_merged(struct bfq_group *bfqg, unsigned int op) { } | |
84c7afce OS |
261 | void bfqg_stats_update_completion(struct bfq_group *bfqg, u64 start_time_ns, |
262 | u64 io_start_time_ns, unsigned int op) { } | |
a33801e8 LM |
263 | void bfqg_stats_update_dequeue(struct bfq_group *bfqg) { } |
264 | void bfqg_stats_set_start_empty_time(struct bfq_group *bfqg) { } | |
265 | void bfqg_stats_update_idle_time(struct bfq_group *bfqg) { } | |
266 | void bfqg_stats_set_start_idle_time(struct bfq_group *bfqg) { } | |
267 | void bfqg_stats_update_avg_queue_size(struct bfq_group *bfqg) { } | |
268 | ||
8060c47b | 269 | #endif /* CONFIG_BFQ_CGROUP_DEBUG */ |
a33801e8 LM |
270 | |
271 | #ifdef CONFIG_BFQ_GROUP_IOSCHED | |
272 | ||
ea25da48 PV |
273 | /* |
274 | * blk-cgroup policy-related handlers | |
275 | * The following functions help in converting between blk-cgroup | |
276 | * internal structures and BFQ-specific structures. | |
277 | */ | |
278 | ||
279 | static struct bfq_group *pd_to_bfqg(struct blkg_policy_data *pd) | |
280 | { | |
281 | return pd ? container_of(pd, struct bfq_group, pd) : NULL; | |
282 | } | |
283 | ||
284 | struct blkcg_gq *bfqg_to_blkg(struct bfq_group *bfqg) | |
285 | { | |
286 | return pd_to_blkg(&bfqg->pd); | |
287 | } | |
288 | ||
289 | static struct bfq_group *blkg_to_bfqg(struct blkcg_gq *blkg) | |
290 | { | |
291 | return pd_to_bfqg(blkg_to_pd(blkg, &blkcg_policy_bfq)); | |
292 | } | |
293 | ||
294 | /* | |
295 | * bfq_group handlers | |
296 | * The following functions help in navigating the bfq_group hierarchy | |
297 | * by allowing to find the parent of a bfq_group or the bfq_group | |
298 | * associated to a bfq_queue. | |
299 | */ | |
300 | ||
301 | static struct bfq_group *bfqg_parent(struct bfq_group *bfqg) | |
302 | { | |
303 | struct blkcg_gq *pblkg = bfqg_to_blkg(bfqg)->parent; | |
304 | ||
305 | return pblkg ? blkg_to_bfqg(pblkg) : NULL; | |
306 | } | |
307 | ||
308 | struct bfq_group *bfqq_group(struct bfq_queue *bfqq) | |
309 | { | |
310 | struct bfq_entity *group_entity = bfqq->entity.parent; | |
311 | ||
312 | return group_entity ? container_of(group_entity, struct bfq_group, | |
313 | entity) : | |
314 | bfqq->bfqd->root_group; | |
315 | } | |
316 | ||
317 | /* | |
318 | * The following two functions handle get and put of a bfq_group by | |
319 | * wrapping the related blk-cgroup hooks. | |
320 | */ | |
321 | ||
322 | static void bfqg_get(struct bfq_group *bfqg) | |
323 | { | |
8f9bebc3 | 324 | bfqg->ref++; |
ea25da48 PV |
325 | } |
326 | ||
dfb79af5 | 327 | static void bfqg_put(struct bfq_group *bfqg) |
ea25da48 | 328 | { |
8f9bebc3 PV |
329 | bfqg->ref--; |
330 | ||
331 | if (bfqg->ref == 0) | |
332 | kfree(bfqg); | |
333 | } | |
334 | ||
2de791ab | 335 | static void bfqg_and_blkg_get(struct bfq_group *bfqg) |
8f9bebc3 PV |
336 | { |
337 | /* see comments in bfq_bic_update_cgroup for why refcounting bfqg */ | |
338 | bfqg_get(bfqg); | |
339 | ||
340 | blkg_get(bfqg_to_blkg(bfqg)); | |
341 | } | |
342 | ||
343 | void bfqg_and_blkg_put(struct bfq_group *bfqg) | |
344 | { | |
8f9bebc3 | 345 | blkg_put(bfqg_to_blkg(bfqg)); |
d5274b3c KK |
346 | |
347 | bfqg_put(bfqg); | |
ea25da48 PV |
348 | } |
349 | ||
fd41e603 TH |
350 | void bfqg_stats_update_legacy_io(struct request_queue *q, struct request *rq) |
351 | { | |
352 | struct bfq_group *bfqg = blkg_to_bfqg(rq->bio->bi_blkg); | |
353 | ||
08802ed6 HT |
354 | if (!bfqg) |
355 | return; | |
356 | ||
fd41e603 TH |
357 | blkg_rwstat_add(&bfqg->stats.bytes, rq->cmd_flags, blk_rq_bytes(rq)); |
358 | blkg_rwstat_add(&bfqg->stats.ios, rq->cmd_flags, 1); | |
359 | } | |
360 | ||
ea25da48 PV |
361 | /* @stats = 0 */ |
362 | static void bfqg_stats_reset(struct bfqg_stats *stats) | |
363 | { | |
8060c47b | 364 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
ea25da48 PV |
365 | /* queued stats shouldn't be cleared */ |
366 | blkg_rwstat_reset(&stats->merged); | |
367 | blkg_rwstat_reset(&stats->service_time); | |
368 | blkg_rwstat_reset(&stats->wait_time); | |
c0ce79dc CH |
369 | bfq_stat_reset(&stats->time); |
370 | bfq_stat_reset(&stats->avg_queue_size_sum); | |
371 | bfq_stat_reset(&stats->avg_queue_size_samples); | |
372 | bfq_stat_reset(&stats->dequeue); | |
373 | bfq_stat_reset(&stats->group_wait_time); | |
374 | bfq_stat_reset(&stats->idle_time); | |
375 | bfq_stat_reset(&stats->empty_time); | |
a33801e8 | 376 | #endif |
ea25da48 PV |
377 | } |
378 | ||
379 | /* @to += @from */ | |
380 | static void bfqg_stats_add_aux(struct bfqg_stats *to, struct bfqg_stats *from) | |
381 | { | |
382 | if (!to || !from) | |
383 | return; | |
384 | ||
8060c47b | 385 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
ea25da48 PV |
386 | /* queued stats shouldn't be cleared */ |
387 | blkg_rwstat_add_aux(&to->merged, &from->merged); | |
388 | blkg_rwstat_add_aux(&to->service_time, &from->service_time); | |
389 | blkg_rwstat_add_aux(&to->wait_time, &from->wait_time); | |
c0ce79dc CH |
390 | bfq_stat_add_aux(&from->time, &from->time); |
391 | bfq_stat_add_aux(&to->avg_queue_size_sum, &from->avg_queue_size_sum); | |
392 | bfq_stat_add_aux(&to->avg_queue_size_samples, | |
ea25da48 | 393 | &from->avg_queue_size_samples); |
c0ce79dc CH |
394 | bfq_stat_add_aux(&to->dequeue, &from->dequeue); |
395 | bfq_stat_add_aux(&to->group_wait_time, &from->group_wait_time); | |
396 | bfq_stat_add_aux(&to->idle_time, &from->idle_time); | |
397 | bfq_stat_add_aux(&to->empty_time, &from->empty_time); | |
a33801e8 | 398 | #endif |
ea25da48 PV |
399 | } |
400 | ||
401 | /* | |
402 | * Transfer @bfqg's stats to its parent's aux counts so that the ancestors' | |
403 | * recursive stats can still account for the amount used by this bfqg after | |
404 | * it's gone. | |
405 | */ | |
406 | static void bfqg_stats_xfer_dead(struct bfq_group *bfqg) | |
407 | { | |
408 | struct bfq_group *parent; | |
409 | ||
410 | if (!bfqg) /* root_group */ | |
411 | return; | |
412 | ||
413 | parent = bfqg_parent(bfqg); | |
414 | ||
0d945c1f | 415 | lockdep_assert_held(&bfqg_to_blkg(bfqg)->q->queue_lock); |
ea25da48 PV |
416 | |
417 | if (unlikely(!parent)) | |
418 | return; | |
419 | ||
420 | bfqg_stats_add_aux(&parent->stats, &bfqg->stats); | |
421 | bfqg_stats_reset(&bfqg->stats); | |
422 | } | |
423 | ||
424 | void bfq_init_entity(struct bfq_entity *entity, struct bfq_group *bfqg) | |
425 | { | |
426 | struct bfq_queue *bfqq = bfq_entity_to_bfqq(entity); | |
427 | ||
428 | entity->weight = entity->new_weight; | |
429 | entity->orig_weight = entity->new_weight; | |
430 | if (bfqq) { | |
431 | bfqq->ioprio = bfqq->new_ioprio; | |
432 | bfqq->ioprio_class = bfqq->new_ioprio_class; | |
8f9bebc3 PV |
433 | /* |
434 | * Make sure that bfqg and its associated blkg do not | |
435 | * disappear before entity. | |
436 | */ | |
437 | bfqg_and_blkg_get(bfqg); | |
ea25da48 PV |
438 | } |
439 | entity->parent = bfqg->my_entity; /* NULL for root group */ | |
440 | entity->sched_data = &bfqg->sched_data; | |
441 | } | |
442 | ||
443 | static void bfqg_stats_exit(struct bfqg_stats *stats) | |
444 | { | |
fd41e603 TH |
445 | blkg_rwstat_exit(&stats->bytes); |
446 | blkg_rwstat_exit(&stats->ios); | |
8060c47b | 447 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
ea25da48 PV |
448 | blkg_rwstat_exit(&stats->merged); |
449 | blkg_rwstat_exit(&stats->service_time); | |
450 | blkg_rwstat_exit(&stats->wait_time); | |
451 | blkg_rwstat_exit(&stats->queued); | |
c0ce79dc CH |
452 | bfq_stat_exit(&stats->time); |
453 | bfq_stat_exit(&stats->avg_queue_size_sum); | |
454 | bfq_stat_exit(&stats->avg_queue_size_samples); | |
455 | bfq_stat_exit(&stats->dequeue); | |
456 | bfq_stat_exit(&stats->group_wait_time); | |
457 | bfq_stat_exit(&stats->idle_time); | |
458 | bfq_stat_exit(&stats->empty_time); | |
a33801e8 | 459 | #endif |
ea25da48 PV |
460 | } |
461 | ||
462 | static int bfqg_stats_init(struct bfqg_stats *stats, gfp_t gfp) | |
463 | { | |
fd41e603 TH |
464 | if (blkg_rwstat_init(&stats->bytes, gfp) || |
465 | blkg_rwstat_init(&stats->ios, gfp)) | |
2fc428f6 | 466 | goto error; |
fd41e603 | 467 | |
8060c47b | 468 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
ea25da48 PV |
469 | if (blkg_rwstat_init(&stats->merged, gfp) || |
470 | blkg_rwstat_init(&stats->service_time, gfp) || | |
471 | blkg_rwstat_init(&stats->wait_time, gfp) || | |
472 | blkg_rwstat_init(&stats->queued, gfp) || | |
c0ce79dc CH |
473 | bfq_stat_init(&stats->time, gfp) || |
474 | bfq_stat_init(&stats->avg_queue_size_sum, gfp) || | |
475 | bfq_stat_init(&stats->avg_queue_size_samples, gfp) || | |
476 | bfq_stat_init(&stats->dequeue, gfp) || | |
477 | bfq_stat_init(&stats->group_wait_time, gfp) || | |
478 | bfq_stat_init(&stats->idle_time, gfp) || | |
2fc428f6 ZL |
479 | bfq_stat_init(&stats->empty_time, gfp)) |
480 | goto error; | |
a33801e8 | 481 | #endif |
ea25da48 PV |
482 | |
483 | return 0; | |
2fc428f6 ZL |
484 | |
485 | error: | |
486 | bfqg_stats_exit(stats); | |
487 | return -ENOMEM; | |
ea25da48 PV |
488 | } |
489 | ||
490 | static struct bfq_group_data *cpd_to_bfqgd(struct blkcg_policy_data *cpd) | |
491 | { | |
492 | return cpd ? container_of(cpd, struct bfq_group_data, pd) : NULL; | |
493 | } | |
494 | ||
495 | static struct bfq_group_data *blkcg_to_bfqgd(struct blkcg *blkcg) | |
496 | { | |
497 | return cpd_to_bfqgd(blkcg_to_cpd(blkcg, &blkcg_policy_bfq)); | |
498 | } | |
499 | ||
dfb79af5 | 500 | static struct blkcg_policy_data *bfq_cpd_alloc(gfp_t gfp) |
ea25da48 PV |
501 | { |
502 | struct bfq_group_data *bgd; | |
503 | ||
504 | bgd = kzalloc(sizeof(*bgd), gfp); | |
505 | if (!bgd) | |
506 | return NULL; | |
507 | return &bgd->pd; | |
508 | } | |
509 | ||
dfb79af5 | 510 | static void bfq_cpd_init(struct blkcg_policy_data *cpd) |
ea25da48 PV |
511 | { |
512 | struct bfq_group_data *d = cpd_to_bfqgd(cpd); | |
513 | ||
514 | d->weight = cgroup_subsys_on_dfl(io_cgrp_subsys) ? | |
515 | CGROUP_WEIGHT_DFL : BFQ_WEIGHT_LEGACY_DFL; | |
516 | } | |
517 | ||
dfb79af5 | 518 | static void bfq_cpd_free(struct blkcg_policy_data *cpd) |
ea25da48 PV |
519 | { |
520 | kfree(cpd_to_bfqgd(cpd)); | |
521 | } | |
522 | ||
cf09a8ee TH |
523 | static struct blkg_policy_data *bfq_pd_alloc(gfp_t gfp, struct request_queue *q, |
524 | struct blkcg *blkcg) | |
ea25da48 PV |
525 | { |
526 | struct bfq_group *bfqg; | |
527 | ||
cf09a8ee | 528 | bfqg = kzalloc_node(sizeof(*bfqg), gfp, q->node); |
ea25da48 PV |
529 | if (!bfqg) |
530 | return NULL; | |
531 | ||
532 | if (bfqg_stats_init(&bfqg->stats, gfp)) { | |
533 | kfree(bfqg); | |
534 | return NULL; | |
535 | } | |
536 | ||
8f9bebc3 PV |
537 | /* see comments in bfq_bic_update_cgroup for why refcounting */ |
538 | bfqg_get(bfqg); | |
ea25da48 PV |
539 | return &bfqg->pd; |
540 | } | |
541 | ||
dfb79af5 | 542 | static void bfq_pd_init(struct blkg_policy_data *pd) |
ea25da48 PV |
543 | { |
544 | struct blkcg_gq *blkg = pd_to_blkg(pd); | |
545 | struct bfq_group *bfqg = blkg_to_bfqg(blkg); | |
546 | struct bfq_data *bfqd = blkg->q->elevator->elevator_data; | |
547 | struct bfq_entity *entity = &bfqg->entity; | |
548 | struct bfq_group_data *d = blkcg_to_bfqgd(blkg->blkcg); | |
549 | ||
550 | entity->orig_weight = entity->weight = entity->new_weight = d->weight; | |
551 | entity->my_sched_data = &bfqg->sched_data; | |
430a67f9 PV |
552 | entity->last_bfqq_created = NULL; |
553 | ||
ea25da48 PV |
554 | bfqg->my_entity = entity; /* |
555 | * the root_group's will be set to NULL | |
556 | * in bfq_init_queue() | |
557 | */ | |
558 | bfqg->bfqd = bfqd; | |
559 | bfqg->active_entities = 0; | |
560 | bfqg->rq_pos_tree = RB_ROOT; | |
561 | } | |
562 | ||
dfb79af5 | 563 | static void bfq_pd_free(struct blkg_policy_data *pd) |
ea25da48 PV |
564 | { |
565 | struct bfq_group *bfqg = pd_to_bfqg(pd); | |
566 | ||
567 | bfqg_stats_exit(&bfqg->stats); | |
8f9bebc3 | 568 | bfqg_put(bfqg); |
ea25da48 PV |
569 | } |
570 | ||
dfb79af5 | 571 | static void bfq_pd_reset_stats(struct blkg_policy_data *pd) |
ea25da48 PV |
572 | { |
573 | struct bfq_group *bfqg = pd_to_bfqg(pd); | |
574 | ||
575 | bfqg_stats_reset(&bfqg->stats); | |
576 | } | |
577 | ||
578 | static void bfq_group_set_parent(struct bfq_group *bfqg, | |
579 | struct bfq_group *parent) | |
580 | { | |
581 | struct bfq_entity *entity; | |
582 | ||
583 | entity = &bfqg->entity; | |
584 | entity->parent = parent->my_entity; | |
585 | entity->sched_data = &parent->sched_data; | |
586 | } | |
587 | ||
588 | static struct bfq_group *bfq_lookup_bfqg(struct bfq_data *bfqd, | |
589 | struct blkcg *blkcg) | |
590 | { | |
591 | struct blkcg_gq *blkg; | |
592 | ||
593 | blkg = blkg_lookup(blkcg, bfqd->queue); | |
594 | if (likely(blkg)) | |
595 | return blkg_to_bfqg(blkg); | |
596 | return NULL; | |
597 | } | |
598 | ||
599 | struct bfq_group *bfq_find_set_group(struct bfq_data *bfqd, | |
600 | struct blkcg *blkcg) | |
601 | { | |
602 | struct bfq_group *bfqg, *parent; | |
603 | struct bfq_entity *entity; | |
604 | ||
605 | bfqg = bfq_lookup_bfqg(bfqd, blkcg); | |
606 | ||
607 | if (unlikely(!bfqg)) | |
608 | return NULL; | |
609 | ||
610 | /* | |
611 | * Update chain of bfq_groups as we might be handling a leaf group | |
612 | * which, along with some of its relatives, has not been hooked yet | |
613 | * to the private hierarchy of BFQ. | |
614 | */ | |
615 | entity = &bfqg->entity; | |
616 | for_each_entity(entity) { | |
14afc593 CN |
617 | struct bfq_group *curr_bfqg = container_of(entity, |
618 | struct bfq_group, entity); | |
619 | if (curr_bfqg != bfqd->root_group) { | |
620 | parent = bfqg_parent(curr_bfqg); | |
ea25da48 PV |
621 | if (!parent) |
622 | parent = bfqd->root_group; | |
14afc593 | 623 | bfq_group_set_parent(curr_bfqg, parent); |
ea25da48 PV |
624 | } |
625 | } | |
626 | ||
627 | return bfqg; | |
628 | } | |
629 | ||
630 | /** | |
631 | * bfq_bfqq_move - migrate @bfqq to @bfqg. | |
632 | * @bfqd: queue descriptor. | |
633 | * @bfqq: the queue to move. | |
634 | * @bfqg: the group to move to. | |
635 | * | |
636 | * Move @bfqq to @bfqg, deactivating it from its old group and reactivating | |
637 | * it on the new one. Avoid putting the entity on the old group idle tree. | |
638 | * | |
8f9bebc3 PV |
639 | * Must be called under the scheduler lock, to make sure that the blkg |
640 | * owning @bfqg does not disappear (see comments in | |
641 | * bfq_bic_update_cgroup on guaranteeing the consistency of blkg | |
642 | * objects). | |
ea25da48 PV |
643 | */ |
644 | void bfq_bfqq_move(struct bfq_data *bfqd, struct bfq_queue *bfqq, | |
645 | struct bfq_group *bfqg) | |
646 | { | |
647 | struct bfq_entity *entity = &bfqq->entity; | |
c5e4cb0f YK |
648 | struct bfq_group *old_parent = bfqq_group(bfqq); |
649 | ||
650 | /* | |
651 | * No point to move bfqq to the same group, which can happen when | |
652 | * root group is offlined | |
653 | */ | |
654 | if (old_parent == bfqg) | |
655 | return; | |
ea25da48 | 656 | |
8410f709 YK |
657 | /* |
658 | * oom_bfqq is not allowed to move, oom_bfqq will hold ref to root_group | |
659 | * until elevator exit. | |
660 | */ | |
661 | if (bfqq == &bfqd->oom_bfqq) | |
662 | return; | |
fd1bb3ae PV |
663 | /* |
664 | * Get extra reference to prevent bfqq from being freed in | |
665 | * next possible expire or deactivate. | |
666 | */ | |
667 | bfqq->ref++; | |
668 | ||
ea25da48 PV |
669 | /* If bfqq is empty, then bfq_bfqq_expire also invokes |
670 | * bfq_del_bfqq_busy, thereby removing bfqq and its entity | |
671 | * from data structures related to current group. Otherwise we | |
672 | * need to remove bfqq explicitly with bfq_deactivate_bfqq, as | |
673 | * we do below. | |
674 | */ | |
675 | if (bfqq == bfqd->in_service_queue) | |
676 | bfq_bfqq_expire(bfqd, bfqd->in_service_queue, | |
677 | false, BFQQE_PREEMPTED); | |
678 | ||
679 | if (bfq_bfqq_busy(bfqq)) | |
680 | bfq_deactivate_bfqq(bfqd, bfqq, false, false); | |
33a16a98 | 681 | else if (entity->on_st_or_in_serv) |
ea25da48 | 682 | bfq_put_idle_entity(bfq_entity_service_tree(entity), entity); |
c5e4cb0f | 683 | bfqg_and_blkg_put(old_parent); |
ea25da48 | 684 | |
d29bd414 PV |
685 | if (entity->parent && |
686 | entity->parent->last_bfqq_created == bfqq) | |
687 | entity->parent->last_bfqq_created = NULL; | |
688 | else if (bfqd->last_bfqq_created == bfqq) | |
689 | bfqd->last_bfqq_created = NULL; | |
690 | ||
ea25da48 PV |
691 | entity->parent = bfqg->my_entity; |
692 | entity->sched_data = &bfqg->sched_data; | |
8f9bebc3 PV |
693 | /* pin down bfqg and its associated blkg */ |
694 | bfqg_and_blkg_get(bfqg); | |
ea25da48 PV |
695 | |
696 | if (bfq_bfqq_busy(bfqq)) { | |
8cacc5ab PV |
697 | if (unlikely(!bfqd->nonrot_with_queueing)) |
698 | bfq_pos_tree_add_move(bfqd, bfqq); | |
ea25da48 PV |
699 | bfq_activate_bfqq(bfqd, bfqq); |
700 | } | |
701 | ||
702 | if (!bfqd->in_service_queue && !bfqd->rq_in_driver) | |
703 | bfq_schedule_dispatch(bfqd); | |
fd1bb3ae | 704 | /* release extra ref taken above, bfqq may happen to be freed now */ |
ecedd3d7 | 705 | bfq_put_queue(bfqq); |
ea25da48 PV |
706 | } |
707 | ||
708 | /** | |
709 | * __bfq_bic_change_cgroup - move @bic to @cgroup. | |
710 | * @bfqd: the queue descriptor. | |
711 | * @bic: the bic to move. | |
712 | * @blkcg: the blk-cgroup to move to. | |
713 | * | |
8f9bebc3 PV |
714 | * Move bic to blkcg, assuming that bfqd->lock is held; which makes |
715 | * sure that the reference to cgroup is valid across the call (see | |
716 | * comments in bfq_bic_update_cgroup on this issue) | |
ea25da48 PV |
717 | * |
718 | * NOTE: an alternative approach might have been to store the current | |
719 | * cgroup in bfqq and getting a reference to it, reducing the lookup | |
720 | * time here, at the price of slightly more complex code. | |
721 | */ | |
722 | static struct bfq_group *__bfq_bic_change_cgroup(struct bfq_data *bfqd, | |
723 | struct bfq_io_cq *bic, | |
724 | struct blkcg *blkcg) | |
725 | { | |
726 | struct bfq_queue *async_bfqq = bic_to_bfqq(bic, 0); | |
727 | struct bfq_queue *sync_bfqq = bic_to_bfqq(bic, 1); | |
728 | struct bfq_group *bfqg; | |
729 | struct bfq_entity *entity; | |
730 | ||
731 | bfqg = bfq_find_set_group(bfqd, blkcg); | |
732 | ||
733 | if (unlikely(!bfqg)) | |
734 | bfqg = bfqd->root_group; | |
735 | ||
736 | if (async_bfqq) { | |
737 | entity = &async_bfqq->entity; | |
738 | ||
739 | if (entity->sched_data != &bfqg->sched_data) { | |
740 | bic_set_bfqq(bic, NULL, 0); | |
c8997736 | 741 | bfq_release_process_ref(bfqd, async_bfqq); |
ea25da48 PV |
742 | } |
743 | } | |
744 | ||
745 | if (sync_bfqq) { | |
746 | entity = &sync_bfqq->entity; | |
747 | if (entity->sched_data != &bfqg->sched_data) | |
748 | bfq_bfqq_move(bfqd, sync_bfqq, bfqg); | |
749 | } | |
750 | ||
751 | return bfqg; | |
752 | } | |
753 | ||
754 | void bfq_bic_update_cgroup(struct bfq_io_cq *bic, struct bio *bio) | |
755 | { | |
756 | struct bfq_data *bfqd = bic_to_bfqd(bic); | |
757 | struct bfq_group *bfqg = NULL; | |
758 | uint64_t serial_nr; | |
759 | ||
760 | rcu_read_lock(); | |
0fe061b9 | 761 | serial_nr = __bio_blkcg(bio)->css.serial_nr; |
ea25da48 PV |
762 | |
763 | /* | |
764 | * Check whether blkcg has changed. The condition may trigger | |
765 | * spuriously on a newly created cic but there's no harm. | |
766 | */ | |
767 | if (unlikely(!bfqd) || likely(bic->blkcg_serial_nr == serial_nr)) | |
768 | goto out; | |
769 | ||
0fe061b9 | 770 | bfqg = __bfq_bic_change_cgroup(bfqd, bic, __bio_blkcg(bio)); |
8f9bebc3 PV |
771 | /* |
772 | * Update blkg_path for bfq_log_* functions. We cache this | |
773 | * path, and update it here, for the following | |
774 | * reasons. Operations on blkg objects in blk-cgroup are | |
775 | * protected with the request_queue lock, and not with the | |
776 | * lock that protects the instances of this scheduler | |
777 | * (bfqd->lock). This exposes BFQ to the following sort of | |
778 | * race. | |
779 | * | |
780 | * The blkg_lookup performed in bfq_get_queue, protected | |
781 | * through rcu, may happen to return the address of a copy of | |
782 | * the original blkg. If this is the case, then the | |
783 | * bfqg_and_blkg_get performed in bfq_get_queue, to pin down | |
784 | * the blkg, is useless: it does not prevent blk-cgroup code | |
785 | * from destroying both the original blkg and all objects | |
786 | * directly or indirectly referred by the copy of the | |
787 | * blkg. | |
788 | * | |
789 | * On the bright side, destroy operations on a blkg invoke, as | |
790 | * a first step, hooks of the scheduler associated with the | |
791 | * blkg. And these hooks are executed with bfqd->lock held for | |
792 | * BFQ. As a consequence, for any blkg associated with the | |
793 | * request queue this instance of the scheduler is attached | |
794 | * to, we are guaranteed that such a blkg is not destroyed, and | |
795 | * that all the pointers it contains are consistent, while we | |
796 | * are holding bfqd->lock. A blkg_lookup performed with | |
797 | * bfqd->lock held then returns a fully consistent blkg, which | |
798 | * remains consistent until this lock is held. | |
799 | * | |
800 | * Thanks to the last fact, and to the fact that: (1) bfqg has | |
801 | * been obtained through a blkg_lookup in the above | |
802 | * assignment, and (2) bfqd->lock is being held, here we can | |
803 | * safely use the policy data for the involved blkg (i.e., the | |
804 | * field bfqg->pd) to get to the blkg associated with bfqg, | |
805 | * and then we can safely use any field of blkg. After we | |
806 | * release bfqd->lock, even just getting blkg through this | |
807 | * bfqg may cause dangling references to be traversed, as | |
808 | * bfqg->pd may not exist any more. | |
809 | * | |
810 | * In view of the above facts, here we cache, in the bfqg, any | |
811 | * blkg data we may need for this bic, and for its associated | |
812 | * bfq_queue. As of now, we need to cache only the path of the | |
813 | * blkg, which is used in the bfq_log_* functions. | |
814 | * | |
815 | * Finally, note that bfqg itself needs to be protected from | |
816 | * destruction on the blkg_free of the original blkg (which | |
817 | * invokes bfq_pd_free). We use an additional private | |
818 | * refcounter for bfqg, to let it disappear only after no | |
819 | * bfq_queue refers to it any longer. | |
820 | */ | |
821 | blkg_path(bfqg_to_blkg(bfqg), bfqg->blkg_path, sizeof(bfqg->blkg_path)); | |
ea25da48 PV |
822 | bic->blkcg_serial_nr = serial_nr; |
823 | out: | |
824 | rcu_read_unlock(); | |
825 | } | |
826 | ||
827 | /** | |
828 | * bfq_flush_idle_tree - deactivate any entity on the idle tree of @st. | |
829 | * @st: the service tree being flushed. | |
830 | */ | |
831 | static void bfq_flush_idle_tree(struct bfq_service_tree *st) | |
832 | { | |
833 | struct bfq_entity *entity = st->first_idle; | |
834 | ||
835 | for (; entity ; entity = st->first_idle) | |
836 | __bfq_deactivate_entity(entity, false); | |
837 | } | |
838 | ||
839 | /** | |
840 | * bfq_reparent_leaf_entity - move leaf entity to the root_group. | |
841 | * @bfqd: the device data structure with the root group. | |
576682fa PV |
842 | * @entity: the entity to move, if entity is a leaf; or the parent entity |
843 | * of an active leaf entity to move, if entity is not a leaf. | |
ea25da48 PV |
844 | */ |
845 | static void bfq_reparent_leaf_entity(struct bfq_data *bfqd, | |
576682fa PV |
846 | struct bfq_entity *entity, |
847 | int ioprio_class) | |
ea25da48 | 848 | { |
576682fa PV |
849 | struct bfq_queue *bfqq; |
850 | struct bfq_entity *child_entity = entity; | |
851 | ||
852 | while (child_entity->my_sched_data) { /* leaf not reached yet */ | |
853 | struct bfq_sched_data *child_sd = child_entity->my_sched_data; | |
854 | struct bfq_service_tree *child_st = child_sd->service_tree + | |
855 | ioprio_class; | |
856 | struct rb_root *child_active = &child_st->active; | |
ea25da48 | 857 | |
576682fa PV |
858 | child_entity = bfq_entity_of(rb_first(child_active)); |
859 | ||
860 | if (!child_entity) | |
861 | child_entity = child_sd->in_service_entity; | |
862 | } | |
863 | ||
864 | bfqq = bfq_entity_to_bfqq(child_entity); | |
ea25da48 PV |
865 | bfq_bfqq_move(bfqd, bfqq, bfqd->root_group); |
866 | } | |
867 | ||
868 | /** | |
576682fa | 869 | * bfq_reparent_active_queues - move to the root group all active queues. |
ea25da48 PV |
870 | * @bfqd: the device data structure with the root group. |
871 | * @bfqg: the group to move from. | |
576682fa | 872 | * @st: the service tree to start the search from. |
ea25da48 | 873 | */ |
576682fa PV |
874 | static void bfq_reparent_active_queues(struct bfq_data *bfqd, |
875 | struct bfq_group *bfqg, | |
876 | struct bfq_service_tree *st, | |
877 | int ioprio_class) | |
ea25da48 PV |
878 | { |
879 | struct rb_root *active = &st->active; | |
576682fa | 880 | struct bfq_entity *entity; |
ea25da48 | 881 | |
576682fa PV |
882 | while ((entity = bfq_entity_of(rb_first(active)))) |
883 | bfq_reparent_leaf_entity(bfqd, entity, ioprio_class); | |
ea25da48 PV |
884 | |
885 | if (bfqg->sched_data.in_service_entity) | |
886 | bfq_reparent_leaf_entity(bfqd, | |
576682fa PV |
887 | bfqg->sched_data.in_service_entity, |
888 | ioprio_class); | |
ea25da48 PV |
889 | } |
890 | ||
891 | /** | |
892 | * bfq_pd_offline - deactivate the entity associated with @pd, | |
893 | * and reparent its children entities. | |
894 | * @pd: descriptor of the policy going offline. | |
895 | * | |
896 | * blkio already grabs the queue_lock for us, so no need to use | |
897 | * RCU-based magic | |
898 | */ | |
dfb79af5 | 899 | static void bfq_pd_offline(struct blkg_policy_data *pd) |
ea25da48 PV |
900 | { |
901 | struct bfq_service_tree *st; | |
902 | struct bfq_group *bfqg = pd_to_bfqg(pd); | |
903 | struct bfq_data *bfqd = bfqg->bfqd; | |
904 | struct bfq_entity *entity = bfqg->my_entity; | |
905 | unsigned long flags; | |
906 | int i; | |
907 | ||
52257ffb PV |
908 | spin_lock_irqsave(&bfqd->lock, flags); |
909 | ||
ea25da48 | 910 | if (!entity) /* root group */ |
52257ffb | 911 | goto put_async_queues; |
ea25da48 | 912 | |
ea25da48 PV |
913 | /* |
914 | * Empty all service_trees belonging to this group before | |
915 | * deactivating the group itself. | |
916 | */ | |
917 | for (i = 0; i < BFQ_IOPRIO_CLASSES; i++) { | |
918 | st = bfqg->sched_data.service_tree + i; | |
919 | ||
ea25da48 PV |
920 | /* |
921 | * It may happen that some queues are still active | |
922 | * (busy) upon group destruction (if the corresponding | |
923 | * processes have been forced to terminate). We move | |
924 | * all the leaf entities corresponding to these queues | |
925 | * to the root_group. | |
926 | * Also, it may happen that the group has an entity | |
927 | * in service, which is disconnected from the active | |
928 | * tree: it must be moved, too. | |
929 | * There is no need to put the sync queues, as the | |
930 | * scheduler has taken no reference. | |
931 | */ | |
576682fa | 932 | bfq_reparent_active_queues(bfqd, bfqg, st, i); |
4d38a87f PV |
933 | |
934 | /* | |
935 | * The idle tree may still contain bfq_queues | |
936 | * belonging to exited task because they never | |
937 | * migrated to a different cgroup from the one being | |
938 | * destroyed now. In addition, even | |
939 | * bfq_reparent_active_queues() may happen to add some | |
940 | * entities to the idle tree. It happens if, in some | |
941 | * of the calls to bfq_bfqq_move() performed by | |
942 | * bfq_reparent_active_queues(), the queue to move is | |
943 | * empty and gets expired. | |
944 | */ | |
945 | bfq_flush_idle_tree(st); | |
ea25da48 PV |
946 | } |
947 | ||
948 | __bfq_deactivate_entity(entity, false); | |
52257ffb PV |
949 | |
950 | put_async_queues: | |
ea25da48 PV |
951 | bfq_put_async_queues(bfqd, bfqg); |
952 | ||
953 | spin_unlock_irqrestore(&bfqd->lock, flags); | |
954 | /* | |
955 | * @blkg is going offline and will be ignored by | |
956 | * blkg_[rw]stat_recursive_sum(). Transfer stats to the parent so | |
957 | * that they don't get lost. If IOs complete after this point, the | |
958 | * stats for them will be lost. Oh well... | |
959 | */ | |
960 | bfqg_stats_xfer_dead(bfqg); | |
961 | } | |
962 | ||
963 | void bfq_end_wr_async(struct bfq_data *bfqd) | |
964 | { | |
965 | struct blkcg_gq *blkg; | |
966 | ||
967 | list_for_each_entry(blkg, &bfqd->queue->blkg_list, q_node) { | |
968 | struct bfq_group *bfqg = blkg_to_bfqg(blkg); | |
969 | ||
970 | bfq_end_wr_async_queues(bfqd, bfqg); | |
971 | } | |
972 | bfq_end_wr_async_queues(bfqd, bfqd->root_group); | |
973 | } | |
974 | ||
795fe54c | 975 | static int bfq_io_show_weight_legacy(struct seq_file *sf, void *v) |
ea25da48 PV |
976 | { |
977 | struct blkcg *blkcg = css_to_blkcg(seq_css(sf)); | |
978 | struct bfq_group_data *bfqgd = blkcg_to_bfqgd(blkcg); | |
979 | unsigned int val = 0; | |
980 | ||
981 | if (bfqgd) | |
982 | val = bfqgd->weight; | |
983 | ||
984 | seq_printf(sf, "%u\n", val); | |
985 | ||
986 | return 0; | |
987 | } | |
988 | ||
795fe54c FZ |
989 | static u64 bfqg_prfill_weight_device(struct seq_file *sf, |
990 | struct blkg_policy_data *pd, int off) | |
5ff047e3 | 991 | { |
795fe54c FZ |
992 | struct bfq_group *bfqg = pd_to_bfqg(pd); |
993 | ||
994 | if (!bfqg->entity.dev_weight) | |
995 | return 0; | |
996 | return __blkg_prfill_u64(sf, pd, bfqg->entity.dev_weight); | |
997 | } | |
998 | ||
999 | static int bfq_io_show_weight(struct seq_file *sf, void *v) | |
1000 | { | |
1001 | struct blkcg *blkcg = css_to_blkcg(seq_css(sf)); | |
1002 | struct bfq_group_data *bfqgd = blkcg_to_bfqgd(blkcg); | |
1003 | ||
1004 | seq_printf(sf, "default %u\n", bfqgd->weight); | |
1005 | blkcg_print_blkgs(sf, blkcg, bfqg_prfill_weight_device, | |
1006 | &blkcg_policy_bfq, 0, false); | |
1007 | return 0; | |
1008 | } | |
1009 | ||
1010 | static void bfq_group_set_weight(struct bfq_group *bfqg, u64 weight, u64 dev_weight) | |
1011 | { | |
1012 | weight = dev_weight ?: weight; | |
1013 | ||
1014 | bfqg->entity.dev_weight = dev_weight; | |
5ff047e3 FZ |
1015 | /* |
1016 | * Setting the prio_changed flag of the entity | |
1017 | * to 1 with new_weight == weight would re-set | |
1018 | * the value of the weight to its ioprio mapping. | |
1019 | * Set the flag only if necessary. | |
1020 | */ | |
1021 | if ((unsigned short)weight != bfqg->entity.new_weight) { | |
1022 | bfqg->entity.new_weight = (unsigned short)weight; | |
1023 | /* | |
1024 | * Make sure that the above new value has been | |
1025 | * stored in bfqg->entity.new_weight before | |
1026 | * setting the prio_changed flag. In fact, | |
1027 | * this flag may be read asynchronously (in | |
1028 | * critical sections protected by a different | |
1029 | * lock than that held here), and finding this | |
1030 | * flag set may cause the execution of the code | |
1031 | * for updating parameters whose value may | |
1032 | * depend also on bfqg->entity.new_weight (in | |
1033 | * __bfq_entity_update_weight_prio). | |
1034 | * This barrier makes sure that the new value | |
1035 | * of bfqg->entity.new_weight is correctly | |
1036 | * seen in that code. | |
1037 | */ | |
1038 | smp_wmb(); | |
1039 | bfqg->entity.prio_changed = 1; | |
1040 | } | |
1041 | } | |
1042 | ||
ea25da48 PV |
1043 | static int bfq_io_set_weight_legacy(struct cgroup_subsys_state *css, |
1044 | struct cftype *cftype, | |
1045 | u64 val) | |
1046 | { | |
1047 | struct blkcg *blkcg = css_to_blkcg(css); | |
1048 | struct bfq_group_data *bfqgd = blkcg_to_bfqgd(blkcg); | |
1049 | struct blkcg_gq *blkg; | |
1050 | int ret = -ERANGE; | |
1051 | ||
1052 | if (val < BFQ_MIN_WEIGHT || val > BFQ_MAX_WEIGHT) | |
1053 | return ret; | |
1054 | ||
1055 | ret = 0; | |
1056 | spin_lock_irq(&blkcg->lock); | |
1057 | bfqgd->weight = (unsigned short)val; | |
1058 | hlist_for_each_entry(blkg, &blkcg->blkg_list, blkcg_node) { | |
1059 | struct bfq_group *bfqg = blkg_to_bfqg(blkg); | |
1060 | ||
5ff047e3 | 1061 | if (bfqg) |
795fe54c | 1062 | bfq_group_set_weight(bfqg, val, 0); |
ea25da48 PV |
1063 | } |
1064 | spin_unlock_irq(&blkcg->lock); | |
1065 | ||
1066 | return ret; | |
1067 | } | |
1068 | ||
795fe54c FZ |
1069 | static ssize_t bfq_io_set_device_weight(struct kernfs_open_file *of, |
1070 | char *buf, size_t nbytes, | |
1071 | loff_t off) | |
ea25da48 | 1072 | { |
795fe54c FZ |
1073 | int ret; |
1074 | struct blkg_conf_ctx ctx; | |
1075 | struct blkcg *blkcg = css_to_blkcg(of_css(of)); | |
1076 | struct bfq_group *bfqg; | |
1077 | u64 v; | |
ea25da48 | 1078 | |
795fe54c | 1079 | ret = blkg_conf_prep(blkcg, &blkcg_policy_bfq, buf, &ctx); |
ea25da48 PV |
1080 | if (ret) |
1081 | return ret; | |
1082 | ||
795fe54c FZ |
1083 | if (sscanf(ctx.body, "%llu", &v) == 1) { |
1084 | /* require "default" on dfl */ | |
1085 | ret = -ERANGE; | |
1086 | if (!v) | |
1087 | goto out; | |
1088 | } else if (!strcmp(strim(ctx.body), "default")) { | |
1089 | v = 0; | |
1090 | } else { | |
1091 | ret = -EINVAL; | |
1092 | goto out; | |
1093 | } | |
1094 | ||
1095 | bfqg = blkg_to_bfqg(ctx.blkg); | |
1096 | ||
1097 | ret = -ERANGE; | |
1098 | if (!v || (v >= BFQ_MIN_WEIGHT && v <= BFQ_MAX_WEIGHT)) { | |
1099 | bfq_group_set_weight(bfqg, bfqg->entity.weight, v); | |
1100 | ret = 0; | |
1101 | } | |
1102 | out: | |
1103 | blkg_conf_finish(&ctx); | |
fc8ebd01 | 1104 | return ret ?: nbytes; |
ea25da48 PV |
1105 | } |
1106 | ||
795fe54c FZ |
1107 | static ssize_t bfq_io_set_weight(struct kernfs_open_file *of, |
1108 | char *buf, size_t nbytes, | |
1109 | loff_t off) | |
1110 | { | |
1111 | char *endp; | |
1112 | int ret; | |
1113 | u64 v; | |
1114 | ||
1115 | buf = strim(buf); | |
1116 | ||
1117 | /* "WEIGHT" or "default WEIGHT" sets the default weight */ | |
1118 | v = simple_strtoull(buf, &endp, 0); | |
1119 | if (*endp == '\0' || sscanf(buf, "default %llu", &v) == 1) { | |
1120 | ret = bfq_io_set_weight_legacy(of_css(of), NULL, v); | |
1121 | return ret ?: nbytes; | |
1122 | } | |
1123 | ||
1124 | return bfq_io_set_device_weight(of, buf, nbytes, off); | |
1125 | } | |
1126 | ||
a557f1c7 | 1127 | static int bfqg_print_rwstat(struct seq_file *sf, void *v) |
ea25da48 | 1128 | { |
a557f1c7 TH |
1129 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), blkg_prfill_rwstat, |
1130 | &blkcg_policy_bfq, seq_cft(sf)->private, true); | |
ea25da48 PV |
1131 | return 0; |
1132 | } | |
1133 | ||
a557f1c7 TH |
1134 | static u64 bfqg_prfill_rwstat_recursive(struct seq_file *sf, |
1135 | struct blkg_policy_data *pd, int off) | |
ea25da48 | 1136 | { |
a557f1c7 TH |
1137 | struct blkg_rwstat_sample sum; |
1138 | ||
1139 | blkg_rwstat_recursive_sum(pd_to_blkg(pd), &blkcg_policy_bfq, off, &sum); | |
1140 | return __blkg_prfill_rwstat(sf, pd, &sum); | |
1141 | } | |
1142 | ||
1143 | static int bfqg_print_rwstat_recursive(struct seq_file *sf, void *v) | |
1144 | { | |
1145 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), | |
1146 | bfqg_prfill_rwstat_recursive, &blkcg_policy_bfq, | |
1147 | seq_cft(sf)->private, true); | |
1148 | return 0; | |
1149 | } | |
1150 | ||
fd41e603 | 1151 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
a557f1c7 TH |
1152 | static int bfqg_print_stat(struct seq_file *sf, void *v) |
1153 | { | |
1154 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), blkg_prfill_stat, | |
1155 | &blkcg_policy_bfq, seq_cft(sf)->private, false); | |
ea25da48 PV |
1156 | return 0; |
1157 | } | |
1158 | ||
1159 | static u64 bfqg_prfill_stat_recursive(struct seq_file *sf, | |
1160 | struct blkg_policy_data *pd, int off) | |
1161 | { | |
d6258980 CH |
1162 | struct blkcg_gq *blkg = pd_to_blkg(pd); |
1163 | struct blkcg_gq *pos_blkg; | |
1164 | struct cgroup_subsys_state *pos_css; | |
1165 | u64 sum = 0; | |
1166 | ||
1167 | lockdep_assert_held(&blkg->q->queue_lock); | |
1168 | ||
1169 | rcu_read_lock(); | |
1170 | blkg_for_each_descendant_pre(pos_blkg, pos_css, blkg) { | |
1171 | struct bfq_stat *stat; | |
1172 | ||
1173 | if (!pos_blkg->online) | |
1174 | continue; | |
1175 | ||
1176 | stat = (void *)blkg_to_pd(pos_blkg, &blkcg_policy_bfq) + off; | |
1177 | sum += bfq_stat_read(stat) + atomic64_read(&stat->aux_cnt); | |
1178 | } | |
1179 | rcu_read_unlock(); | |
1180 | ||
ea25da48 PV |
1181 | return __blkg_prfill_u64(sf, pd, sum); |
1182 | } | |
1183 | ||
ea25da48 PV |
1184 | static int bfqg_print_stat_recursive(struct seq_file *sf, void *v) |
1185 | { | |
1186 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), | |
1187 | bfqg_prfill_stat_recursive, &blkcg_policy_bfq, | |
1188 | seq_cft(sf)->private, false); | |
1189 | return 0; | |
1190 | } | |
1191 | ||
ea25da48 PV |
1192 | static u64 bfqg_prfill_sectors(struct seq_file *sf, struct blkg_policy_data *pd, |
1193 | int off) | |
1194 | { | |
fd41e603 TH |
1195 | struct bfq_group *bfqg = blkg_to_bfqg(pd->blkg); |
1196 | u64 sum = blkg_rwstat_total(&bfqg->stats.bytes); | |
ea25da48 PV |
1197 | |
1198 | return __blkg_prfill_u64(sf, pd, sum >> 9); | |
1199 | } | |
1200 | ||
1201 | static int bfqg_print_stat_sectors(struct seq_file *sf, void *v) | |
1202 | { | |
1203 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), | |
1204 | bfqg_prfill_sectors, &blkcg_policy_bfq, 0, false); | |
1205 | return 0; | |
1206 | } | |
1207 | ||
1208 | static u64 bfqg_prfill_sectors_recursive(struct seq_file *sf, | |
1209 | struct blkg_policy_data *pd, int off) | |
1210 | { | |
7af6fd91 | 1211 | struct blkg_rwstat_sample tmp; |
5d0b6e48 | 1212 | |
fd41e603 TH |
1213 | blkg_rwstat_recursive_sum(pd->blkg, &blkcg_policy_bfq, |
1214 | offsetof(struct bfq_group, stats.bytes), &tmp); | |
ea25da48 | 1215 | |
7af6fd91 CH |
1216 | return __blkg_prfill_u64(sf, pd, |
1217 | (tmp.cnt[BLKG_RWSTAT_READ] + tmp.cnt[BLKG_RWSTAT_WRITE]) >> 9); | |
ea25da48 PV |
1218 | } |
1219 | ||
1220 | static int bfqg_print_stat_sectors_recursive(struct seq_file *sf, void *v) | |
1221 | { | |
1222 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), | |
1223 | bfqg_prfill_sectors_recursive, &blkcg_policy_bfq, 0, | |
1224 | false); | |
1225 | return 0; | |
1226 | } | |
1227 | ||
1228 | static u64 bfqg_prfill_avg_queue_size(struct seq_file *sf, | |
1229 | struct blkg_policy_data *pd, int off) | |
1230 | { | |
1231 | struct bfq_group *bfqg = pd_to_bfqg(pd); | |
c0ce79dc | 1232 | u64 samples = bfq_stat_read(&bfqg->stats.avg_queue_size_samples); |
ea25da48 PV |
1233 | u64 v = 0; |
1234 | ||
1235 | if (samples) { | |
c0ce79dc | 1236 | v = bfq_stat_read(&bfqg->stats.avg_queue_size_sum); |
ea25da48 PV |
1237 | v = div64_u64(v, samples); |
1238 | } | |
1239 | __blkg_prfill_u64(sf, pd, v); | |
1240 | return 0; | |
1241 | } | |
1242 | ||
1243 | /* print avg_queue_size */ | |
1244 | static int bfqg_print_avg_queue_size(struct seq_file *sf, void *v) | |
1245 | { | |
1246 | blkcg_print_blkgs(sf, css_to_blkcg(seq_css(sf)), | |
1247 | bfqg_prfill_avg_queue_size, &blkcg_policy_bfq, | |
1248 | 0, false); | |
1249 | return 0; | |
1250 | } | |
8060c47b | 1251 | #endif /* CONFIG_BFQ_CGROUP_DEBUG */ |
ea25da48 PV |
1252 | |
1253 | struct bfq_group *bfq_create_group_hierarchy(struct bfq_data *bfqd, int node) | |
1254 | { | |
1255 | int ret; | |
1256 | ||
1257 | ret = blkcg_activate_policy(bfqd->queue, &blkcg_policy_bfq); | |
1258 | if (ret) | |
1259 | return NULL; | |
1260 | ||
1261 | return blkg_to_bfqg(bfqd->queue->root_blkg); | |
1262 | } | |
1263 | ||
1264 | struct blkcg_policy blkcg_policy_bfq = { | |
1265 | .dfl_cftypes = bfq_blkg_files, | |
1266 | .legacy_cftypes = bfq_blkcg_legacy_files, | |
1267 | ||
1268 | .cpd_alloc_fn = bfq_cpd_alloc, | |
1269 | .cpd_init_fn = bfq_cpd_init, | |
1270 | .cpd_bind_fn = bfq_cpd_init, | |
1271 | .cpd_free_fn = bfq_cpd_free, | |
1272 | ||
1273 | .pd_alloc_fn = bfq_pd_alloc, | |
1274 | .pd_init_fn = bfq_pd_init, | |
1275 | .pd_offline_fn = bfq_pd_offline, | |
1276 | .pd_free_fn = bfq_pd_free, | |
1277 | .pd_reset_stats_fn = bfq_pd_reset_stats, | |
1278 | }; | |
1279 | ||
1280 | struct cftype bfq_blkcg_legacy_files[] = { | |
1281 | { | |
1282 | .name = "bfq.weight", | |
cf892988 | 1283 | .flags = CFTYPE_NOT_ON_ROOT, |
795fe54c | 1284 | .seq_show = bfq_io_show_weight_legacy, |
ea25da48 PV |
1285 | .write_u64 = bfq_io_set_weight_legacy, |
1286 | }, | |
795fe54c FZ |
1287 | { |
1288 | .name = "bfq.weight_device", | |
1289 | .flags = CFTYPE_NOT_ON_ROOT, | |
1290 | .seq_show = bfq_io_show_weight, | |
1291 | .write = bfq_io_set_weight, | |
1292 | }, | |
ea25da48 PV |
1293 | |
1294 | /* statistics, covers only the tasks in the bfqg */ | |
ea25da48 PV |
1295 | { |
1296 | .name = "bfq.io_service_bytes", | |
fd41e603 TH |
1297 | .private = offsetof(struct bfq_group, stats.bytes), |
1298 | .seq_show = bfqg_print_rwstat, | |
ea25da48 PV |
1299 | }, |
1300 | { | |
1301 | .name = "bfq.io_serviced", | |
fd41e603 TH |
1302 | .private = offsetof(struct bfq_group, stats.ios), |
1303 | .seq_show = bfqg_print_rwstat, | |
ea25da48 | 1304 | }, |
8060c47b | 1305 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
a33801e8 LM |
1306 | { |
1307 | .name = "bfq.time", | |
1308 | .private = offsetof(struct bfq_group, stats.time), | |
1309 | .seq_show = bfqg_print_stat, | |
1310 | }, | |
1311 | { | |
1312 | .name = "bfq.sectors", | |
1313 | .seq_show = bfqg_print_stat_sectors, | |
1314 | }, | |
ea25da48 PV |
1315 | { |
1316 | .name = "bfq.io_service_time", | |
1317 | .private = offsetof(struct bfq_group, stats.service_time), | |
1318 | .seq_show = bfqg_print_rwstat, | |
1319 | }, | |
1320 | { | |
1321 | .name = "bfq.io_wait_time", | |
1322 | .private = offsetof(struct bfq_group, stats.wait_time), | |
1323 | .seq_show = bfqg_print_rwstat, | |
1324 | }, | |
1325 | { | |
1326 | .name = "bfq.io_merged", | |
1327 | .private = offsetof(struct bfq_group, stats.merged), | |
1328 | .seq_show = bfqg_print_rwstat, | |
1329 | }, | |
1330 | { | |
1331 | .name = "bfq.io_queued", | |
1332 | .private = offsetof(struct bfq_group, stats.queued), | |
1333 | .seq_show = bfqg_print_rwstat, | |
1334 | }, | |
8060c47b | 1335 | #endif /* CONFIG_BFQ_CGROUP_DEBUG */ |
ea25da48 | 1336 | |
636b8fe8 | 1337 | /* the same statistics which cover the bfqg and its descendants */ |
ea25da48 PV |
1338 | { |
1339 | .name = "bfq.io_service_bytes_recursive", | |
fd41e603 TH |
1340 | .private = offsetof(struct bfq_group, stats.bytes), |
1341 | .seq_show = bfqg_print_rwstat_recursive, | |
ea25da48 PV |
1342 | }, |
1343 | { | |
1344 | .name = "bfq.io_serviced_recursive", | |
fd41e603 TH |
1345 | .private = offsetof(struct bfq_group, stats.ios), |
1346 | .seq_show = bfqg_print_rwstat_recursive, | |
ea25da48 | 1347 | }, |
8060c47b | 1348 | #ifdef CONFIG_BFQ_CGROUP_DEBUG |
a33801e8 LM |
1349 | { |
1350 | .name = "bfq.time_recursive", | |
1351 | .private = offsetof(struct bfq_group, stats.time), | |
1352 | .seq_show = bfqg_print_stat_recursive, | |
1353 | }, | |
1354 | { | |
1355 | .name = "bfq.sectors_recursive", | |
1356 | .seq_show = bfqg_print_stat_sectors_recursive, | |
1357 | }, | |
ea25da48 PV |
1358 | { |
1359 | .name = "bfq.io_service_time_recursive", | |
1360 | .private = offsetof(struct bfq_group, stats.service_time), | |
1361 | .seq_show = bfqg_print_rwstat_recursive, | |
1362 | }, | |
1363 | { | |
1364 | .name = "bfq.io_wait_time_recursive", | |
1365 | .private = offsetof(struct bfq_group, stats.wait_time), | |
1366 | .seq_show = bfqg_print_rwstat_recursive, | |
1367 | }, | |
1368 | { | |
1369 | .name = "bfq.io_merged_recursive", | |
1370 | .private = offsetof(struct bfq_group, stats.merged), | |
1371 | .seq_show = bfqg_print_rwstat_recursive, | |
1372 | }, | |
1373 | { | |
1374 | .name = "bfq.io_queued_recursive", | |
1375 | .private = offsetof(struct bfq_group, stats.queued), | |
1376 | .seq_show = bfqg_print_rwstat_recursive, | |
1377 | }, | |
1378 | { | |
1379 | .name = "bfq.avg_queue_size", | |
1380 | .seq_show = bfqg_print_avg_queue_size, | |
1381 | }, | |
1382 | { | |
1383 | .name = "bfq.group_wait_time", | |
1384 | .private = offsetof(struct bfq_group, stats.group_wait_time), | |
1385 | .seq_show = bfqg_print_stat, | |
1386 | }, | |
1387 | { | |
1388 | .name = "bfq.idle_time", | |
1389 | .private = offsetof(struct bfq_group, stats.idle_time), | |
1390 | .seq_show = bfqg_print_stat, | |
1391 | }, | |
1392 | { | |
1393 | .name = "bfq.empty_time", | |
1394 | .private = offsetof(struct bfq_group, stats.empty_time), | |
1395 | .seq_show = bfqg_print_stat, | |
1396 | }, | |
1397 | { | |
1398 | .name = "bfq.dequeue", | |
1399 | .private = offsetof(struct bfq_group, stats.dequeue), | |
1400 | .seq_show = bfqg_print_stat, | |
1401 | }, | |
8060c47b | 1402 | #endif /* CONFIG_BFQ_CGROUP_DEBUG */ |
ea25da48 PV |
1403 | { } /* terminate */ |
1404 | }; | |
1405 | ||
1406 | struct cftype bfq_blkg_files[] = { | |
1407 | { | |
1408 | .name = "bfq.weight", | |
cf892988 | 1409 | .flags = CFTYPE_NOT_ON_ROOT, |
ea25da48 PV |
1410 | .seq_show = bfq_io_show_weight, |
1411 | .write = bfq_io_set_weight, | |
1412 | }, | |
1413 | {} /* terminate */ | |
1414 | }; | |
1415 | ||
1416 | #else /* CONFIG_BFQ_GROUP_IOSCHED */ | |
1417 | ||
ea25da48 PV |
1418 | void bfq_bfqq_move(struct bfq_data *bfqd, struct bfq_queue *bfqq, |
1419 | struct bfq_group *bfqg) {} | |
1420 | ||
1421 | void bfq_init_entity(struct bfq_entity *entity, struct bfq_group *bfqg) | |
1422 | { | |
1423 | struct bfq_queue *bfqq = bfq_entity_to_bfqq(entity); | |
1424 | ||
1425 | entity->weight = entity->new_weight; | |
1426 | entity->orig_weight = entity->new_weight; | |
1427 | if (bfqq) { | |
1428 | bfqq->ioprio = bfqq->new_ioprio; | |
1429 | bfqq->ioprio_class = bfqq->new_ioprio_class; | |
1430 | } | |
1431 | entity->sched_data = &bfqg->sched_data; | |
1432 | } | |
1433 | ||
1434 | void bfq_bic_update_cgroup(struct bfq_io_cq *bic, struct bio *bio) {} | |
1435 | ||
1436 | void bfq_end_wr_async(struct bfq_data *bfqd) | |
1437 | { | |
1438 | bfq_end_wr_async_queues(bfqd, bfqd->root_group); | |
1439 | } | |
1440 | ||
1441 | struct bfq_group *bfq_find_set_group(struct bfq_data *bfqd, struct blkcg *blkcg) | |
1442 | { | |
1443 | return bfqd->root_group; | |
1444 | } | |
1445 | ||
1446 | struct bfq_group *bfqq_group(struct bfq_queue *bfqq) | |
1447 | { | |
1448 | return bfqq->bfqd->root_group; | |
1449 | } | |
1450 | ||
4d8340d0 PV |
1451 | void bfqg_and_blkg_get(struct bfq_group *bfqg) {} |
1452 | ||
1453 | void bfqg_and_blkg_put(struct bfq_group *bfqg) {} | |
1454 | ||
ea25da48 PV |
1455 | struct bfq_group *bfq_create_group_hierarchy(struct bfq_data *bfqd, int node) |
1456 | { | |
1457 | struct bfq_group *bfqg; | |
1458 | int i; | |
1459 | ||
1460 | bfqg = kmalloc_node(sizeof(*bfqg), GFP_KERNEL | __GFP_ZERO, node); | |
1461 | if (!bfqg) | |
1462 | return NULL; | |
1463 | ||
1464 | for (i = 0; i < BFQ_IOPRIO_CLASSES; i++) | |
1465 | bfqg->sched_data.service_tree[i] = BFQ_SERVICE_TREE_INIT; | |
1466 | ||
1467 | return bfqg; | |
1468 | } | |
1469 | #endif /* CONFIG_BFQ_GROUP_IOSCHED */ |