]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Throttle infrastructure tests | |
3 | * | |
4 | * Copyright Nodalink, EURL. 2013-2014 | |
5 | * Copyright Igalia, S.L. 2015 | |
6 | * | |
7 | * Authors: | |
8 | * BenoƮt Canet <benoit.canet@nodalink.com> | |
9 | * Alberto Garcia <berto@igalia.com> | |
10 | * | |
11 | * This work is licensed under the terms of the GNU LGPL, version 2 or later. | |
12 | * See the COPYING.LIB file in the top-level directory. | |
13 | */ | |
14 | ||
15 | #include "qemu/osdep.h" | |
16 | #include <math.h> | |
17 | #include "block/aio.h" | |
18 | #include "qapi/error.h" | |
19 | #include "qemu/throttle.h" | |
20 | #include "qemu/error-report.h" | |
21 | #include "qemu/main-loop.h" | |
22 | #include "qemu/module.h" | |
23 | #include "block/throttle-groups.h" | |
24 | #include "sysemu/block-backend.h" | |
25 | ||
26 | static AioContext *ctx; | |
27 | static LeakyBucket bkt; | |
28 | static ThrottleConfig cfg; | |
29 | static ThrottleGroupMember tgm; | |
30 | static ThrottleState ts; | |
31 | static ThrottleTimers *tt; | |
32 | ||
33 | /* useful function */ | |
34 | static bool double_cmp(double x, double y) | |
35 | { | |
36 | return fabsl(x - y) < 1e-6; | |
37 | } | |
38 | ||
39 | /* tests for single bucket operations */ | |
40 | static void test_leak_bucket(void) | |
41 | { | |
42 | throttle_config_init(&cfg); | |
43 | bkt = cfg.buckets[THROTTLE_BPS_TOTAL]; | |
44 | ||
45 | /* set initial value */ | |
46 | bkt.avg = 150; | |
47 | bkt.max = 15; | |
48 | bkt.level = 1.5; | |
49 | ||
50 | /* leak an op work of time */ | |
51 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 150); | |
52 | g_assert(bkt.avg == 150); | |
53 | g_assert(bkt.max == 15); | |
54 | g_assert(double_cmp(bkt.level, 0.5)); | |
55 | ||
56 | /* leak again emptying the bucket */ | |
57 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 150); | |
58 | g_assert(bkt.avg == 150); | |
59 | g_assert(bkt.max == 15); | |
60 | g_assert(double_cmp(bkt.level, 0)); | |
61 | ||
62 | /* check that the bucket level won't go lower */ | |
63 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 150); | |
64 | g_assert(bkt.avg == 150); | |
65 | g_assert(bkt.max == 15); | |
66 | g_assert(double_cmp(bkt.level, 0)); | |
67 | ||
68 | /* check that burst_level leaks correctly */ | |
69 | bkt.burst_level = 6; | |
70 | bkt.max = 250; | |
71 | bkt.burst_length = 2; /* otherwise burst_level will not leak */ | |
72 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 100); | |
73 | g_assert(double_cmp(bkt.burst_level, 3.5)); | |
74 | ||
75 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 100); | |
76 | g_assert(double_cmp(bkt.burst_level, 1)); | |
77 | ||
78 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 100); | |
79 | g_assert(double_cmp(bkt.burst_level, 0)); | |
80 | ||
81 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 100); | |
82 | g_assert(double_cmp(bkt.burst_level, 0)); | |
83 | } | |
84 | ||
85 | static void test_compute_wait(void) | |
86 | { | |
87 | unsigned i; | |
88 | int64_t wait; | |
89 | int64_t result; | |
90 | ||
91 | throttle_config_init(&cfg); | |
92 | bkt = cfg.buckets[THROTTLE_BPS_TOTAL]; | |
93 | ||
94 | /* no operation limit set */ | |
95 | bkt.avg = 0; | |
96 | bkt.max = 15; | |
97 | bkt.level = 1.5; | |
98 | wait = throttle_compute_wait(&bkt); | |
99 | g_assert(!wait); | |
100 | ||
101 | /* zero delta */ | |
102 | bkt.avg = 150; | |
103 | bkt.max = 15; | |
104 | bkt.level = 15; | |
105 | wait = throttle_compute_wait(&bkt); | |
106 | g_assert(!wait); | |
107 | ||
108 | /* below zero delta */ | |
109 | bkt.avg = 150; | |
110 | bkt.max = 15; | |
111 | bkt.level = 9; | |
112 | wait = throttle_compute_wait(&bkt); | |
113 | g_assert(!wait); | |
114 | ||
115 | /* half an operation above max */ | |
116 | bkt.avg = 150; | |
117 | bkt.max = 15; | |
118 | bkt.level = 15.5; | |
119 | wait = throttle_compute_wait(&bkt); | |
120 | /* time required to do half an operation */ | |
121 | result = (int64_t) NANOSECONDS_PER_SECOND / 150 / 2; | |
122 | g_assert(wait == result); | |
123 | ||
124 | /* Perform I/O for 2.2 seconds at a rate of bkt.max */ | |
125 | bkt.burst_length = 2; | |
126 | bkt.level = 0; | |
127 | bkt.avg = 10; | |
128 | bkt.max = 200; | |
129 | for (i = 0; i < 22; i++) { | |
130 | double units = bkt.max / 10; | |
131 | bkt.level += units; | |
132 | bkt.burst_level += units; | |
133 | throttle_leak_bucket(&bkt, NANOSECONDS_PER_SECOND / 10); | |
134 | wait = throttle_compute_wait(&bkt); | |
135 | g_assert(double_cmp(bkt.burst_level, 0)); | |
136 | g_assert(double_cmp(bkt.level, (i + 1) * (bkt.max - bkt.avg) / 10)); | |
137 | /* We can do bursts for the 2 seconds we have configured in | |
138 | * burst_length. We have 100 extra miliseconds of burst | |
139 | * because bkt.level has been leaking during this time. | |
140 | * After that, we have to wait. */ | |
141 | result = i < 21 ? 0 : 1.8 * NANOSECONDS_PER_SECOND; | |
142 | g_assert(wait == result); | |
143 | } | |
144 | } | |
145 | ||
146 | /* functions to test ThrottleState initialization/destroy methods */ | |
147 | static void read_timer_cb(void *opaque) | |
148 | { | |
149 | } | |
150 | ||
151 | static void write_timer_cb(void *opaque) | |
152 | { | |
153 | } | |
154 | ||
155 | static void test_init(void) | |
156 | { | |
157 | int i; | |
158 | ||
159 | tt = &tgm.throttle_timers; | |
160 | ||
161 | /* fill the structures with crap */ | |
162 | memset(&ts, 1, sizeof(ts)); | |
163 | memset(tt, 1, sizeof(*tt)); | |
164 | ||
165 | /* init structures */ | |
166 | throttle_init(&ts); | |
167 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
168 | read_timer_cb, write_timer_cb, &ts); | |
169 | ||
170 | /* check initialized fields */ | |
171 | g_assert(tt->clock_type == QEMU_CLOCK_VIRTUAL); | |
172 | g_assert(tt->timers[0]); | |
173 | g_assert(tt->timers[1]); | |
174 | ||
175 | /* check other fields where cleared */ | |
176 | g_assert(!ts.previous_leak); | |
177 | g_assert(!ts.cfg.op_size); | |
178 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
179 | g_assert(!ts.cfg.buckets[i].avg); | |
180 | g_assert(!ts.cfg.buckets[i].max); | |
181 | g_assert(!ts.cfg.buckets[i].level); | |
182 | } | |
183 | ||
184 | throttle_timers_destroy(tt); | |
185 | } | |
186 | ||
187 | static void test_destroy(void) | |
188 | { | |
189 | int i; | |
190 | throttle_init(&ts); | |
191 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
192 | read_timer_cb, write_timer_cb, &ts); | |
193 | throttle_timers_destroy(tt); | |
194 | for (i = 0; i < 2; i++) { | |
195 | g_assert(!tt->timers[i]); | |
196 | } | |
197 | } | |
198 | ||
199 | /* function to test throttle_config and throttle_get_config */ | |
200 | static void test_config_functions(void) | |
201 | { | |
202 | int i; | |
203 | ThrottleConfig orig_cfg, final_cfg; | |
204 | ||
205 | orig_cfg.buckets[THROTTLE_BPS_TOTAL].avg = 153; | |
206 | orig_cfg.buckets[THROTTLE_BPS_READ].avg = 56; | |
207 | orig_cfg.buckets[THROTTLE_BPS_WRITE].avg = 1; | |
208 | ||
209 | orig_cfg.buckets[THROTTLE_OPS_TOTAL].avg = 150; | |
210 | orig_cfg.buckets[THROTTLE_OPS_READ].avg = 69; | |
211 | orig_cfg.buckets[THROTTLE_OPS_WRITE].avg = 23; | |
212 | ||
213 | orig_cfg.buckets[THROTTLE_BPS_TOTAL].max = 0; | |
214 | orig_cfg.buckets[THROTTLE_BPS_READ].max = 56; | |
215 | orig_cfg.buckets[THROTTLE_BPS_WRITE].max = 120; | |
216 | ||
217 | orig_cfg.buckets[THROTTLE_OPS_TOTAL].max = 150; | |
218 | orig_cfg.buckets[THROTTLE_OPS_READ].max = 400; | |
219 | orig_cfg.buckets[THROTTLE_OPS_WRITE].max = 500; | |
220 | ||
221 | orig_cfg.buckets[THROTTLE_BPS_TOTAL].level = 45; | |
222 | orig_cfg.buckets[THROTTLE_BPS_READ].level = 65; | |
223 | orig_cfg.buckets[THROTTLE_BPS_WRITE].level = 23; | |
224 | ||
225 | orig_cfg.buckets[THROTTLE_OPS_TOTAL].level = 1; | |
226 | orig_cfg.buckets[THROTTLE_OPS_READ].level = 90; | |
227 | orig_cfg.buckets[THROTTLE_OPS_WRITE].level = 75; | |
228 | ||
229 | orig_cfg.op_size = 1; | |
230 | ||
231 | throttle_init(&ts); | |
232 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
233 | read_timer_cb, write_timer_cb, &ts); | |
234 | /* structure reset by throttle_init previous_leak should be null */ | |
235 | g_assert(!ts.previous_leak); | |
236 | throttle_config(&ts, QEMU_CLOCK_VIRTUAL, &orig_cfg); | |
237 | ||
238 | /* has previous leak been initialized by throttle_config ? */ | |
239 | g_assert(ts.previous_leak); | |
240 | ||
241 | /* get back the fixed configuration */ | |
242 | throttle_get_config(&ts, &final_cfg); | |
243 | ||
244 | throttle_timers_destroy(tt); | |
245 | ||
246 | g_assert(final_cfg.buckets[THROTTLE_BPS_TOTAL].avg == 153); | |
247 | g_assert(final_cfg.buckets[THROTTLE_BPS_READ].avg == 56); | |
248 | g_assert(final_cfg.buckets[THROTTLE_BPS_WRITE].avg == 1); | |
249 | ||
250 | g_assert(final_cfg.buckets[THROTTLE_OPS_TOTAL].avg == 150); | |
251 | g_assert(final_cfg.buckets[THROTTLE_OPS_READ].avg == 69); | |
252 | g_assert(final_cfg.buckets[THROTTLE_OPS_WRITE].avg == 23); | |
253 | ||
254 | g_assert(final_cfg.buckets[THROTTLE_BPS_TOTAL].max == 0); | |
255 | g_assert(final_cfg.buckets[THROTTLE_BPS_READ].max == 56); | |
256 | g_assert(final_cfg.buckets[THROTTLE_BPS_WRITE].max == 120); | |
257 | ||
258 | g_assert(final_cfg.buckets[THROTTLE_OPS_TOTAL].max == 150); | |
259 | g_assert(final_cfg.buckets[THROTTLE_OPS_READ].max == 400); | |
260 | g_assert(final_cfg.buckets[THROTTLE_OPS_WRITE].max == 500); | |
261 | ||
262 | g_assert(final_cfg.op_size == 1); | |
263 | ||
264 | /* check bucket have been cleared */ | |
265 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
266 | g_assert(!final_cfg.buckets[i].level); | |
267 | } | |
268 | } | |
269 | ||
270 | /* functions to test is throttle is enabled by a config */ | |
271 | static void set_cfg_value(bool is_max, int index, int value) | |
272 | { | |
273 | if (is_max) { | |
274 | cfg.buckets[index].max = value; | |
275 | /* If max is set, avg should never be 0 */ | |
276 | cfg.buckets[index].avg = MAX(cfg.buckets[index].avg, 1); | |
277 | } else { | |
278 | cfg.buckets[index].avg = value; | |
279 | } | |
280 | } | |
281 | ||
282 | static void test_enabled(void) | |
283 | { | |
284 | int i; | |
285 | ||
286 | throttle_config_init(&cfg); | |
287 | g_assert(!throttle_enabled(&cfg)); | |
288 | ||
289 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
290 | throttle_config_init(&cfg); | |
291 | set_cfg_value(false, i, 150); | |
292 | g_assert(throttle_is_valid(&cfg, NULL)); | |
293 | g_assert(throttle_enabled(&cfg)); | |
294 | } | |
295 | ||
296 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
297 | throttle_config_init(&cfg); | |
298 | set_cfg_value(false, i, -150); | |
299 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
300 | } | |
301 | } | |
302 | ||
303 | /* tests functions for throttle_conflicting */ | |
304 | ||
305 | static void test_conflicts_for_one_set(bool is_max, | |
306 | int total, | |
307 | int read, | |
308 | int write) | |
309 | { | |
310 | throttle_config_init(&cfg); | |
311 | g_assert(throttle_is_valid(&cfg, NULL)); | |
312 | ||
313 | set_cfg_value(is_max, total, 1); | |
314 | set_cfg_value(is_max, read, 1); | |
315 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
316 | ||
317 | throttle_config_init(&cfg); | |
318 | set_cfg_value(is_max, total, 1); | |
319 | set_cfg_value(is_max, write, 1); | |
320 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
321 | ||
322 | throttle_config_init(&cfg); | |
323 | set_cfg_value(is_max, total, 1); | |
324 | set_cfg_value(is_max, read, 1); | |
325 | set_cfg_value(is_max, write, 1); | |
326 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
327 | ||
328 | throttle_config_init(&cfg); | |
329 | set_cfg_value(is_max, total, 1); | |
330 | g_assert(throttle_is_valid(&cfg, NULL)); | |
331 | ||
332 | throttle_config_init(&cfg); | |
333 | set_cfg_value(is_max, read, 1); | |
334 | set_cfg_value(is_max, write, 1); | |
335 | g_assert(throttle_is_valid(&cfg, NULL)); | |
336 | } | |
337 | ||
338 | static void test_conflicting_config(void) | |
339 | { | |
340 | /* bps average conflicts */ | |
341 | test_conflicts_for_one_set(false, | |
342 | THROTTLE_BPS_TOTAL, | |
343 | THROTTLE_BPS_READ, | |
344 | THROTTLE_BPS_WRITE); | |
345 | ||
346 | /* ops average conflicts */ | |
347 | test_conflicts_for_one_set(false, | |
348 | THROTTLE_OPS_TOTAL, | |
349 | THROTTLE_OPS_READ, | |
350 | THROTTLE_OPS_WRITE); | |
351 | ||
352 | /* bps average conflicts */ | |
353 | test_conflicts_for_one_set(true, | |
354 | THROTTLE_BPS_TOTAL, | |
355 | THROTTLE_BPS_READ, | |
356 | THROTTLE_BPS_WRITE); | |
357 | /* ops average conflicts */ | |
358 | test_conflicts_for_one_set(true, | |
359 | THROTTLE_OPS_TOTAL, | |
360 | THROTTLE_OPS_READ, | |
361 | THROTTLE_OPS_WRITE); | |
362 | } | |
363 | /* functions to test the throttle_is_valid function */ | |
364 | static void test_is_valid_for_value(int value, bool should_be_valid) | |
365 | { | |
366 | int is_max, index; | |
367 | for (is_max = 0; is_max < 2; is_max++) { | |
368 | for (index = 0; index < BUCKETS_COUNT; index++) { | |
369 | throttle_config_init(&cfg); | |
370 | set_cfg_value(is_max, index, value); | |
371 | g_assert(throttle_is_valid(&cfg, NULL) == should_be_valid); | |
372 | } | |
373 | } | |
374 | } | |
375 | ||
376 | static void test_is_valid(void) | |
377 | { | |
378 | /* negative number are invalid */ | |
379 | test_is_valid_for_value(-1, false); | |
380 | /* zero are valids */ | |
381 | test_is_valid_for_value(0, true); | |
382 | /* positives numers are valids */ | |
383 | test_is_valid_for_value(1, true); | |
384 | } | |
385 | ||
386 | static void test_ranges(void) | |
387 | { | |
388 | int i; | |
389 | ||
390 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
391 | LeakyBucket *b = &cfg.buckets[i]; | |
392 | throttle_config_init(&cfg); | |
393 | ||
394 | /* avg = 0 means throttling is disabled, but the config is valid */ | |
395 | b->avg = 0; | |
396 | g_assert(throttle_is_valid(&cfg, NULL)); | |
397 | g_assert(!throttle_enabled(&cfg)); | |
398 | ||
399 | /* These are valid configurations (values <= THROTTLE_VALUE_MAX) */ | |
400 | b->avg = 1; | |
401 | g_assert(throttle_is_valid(&cfg, NULL)); | |
402 | ||
403 | b->avg = THROTTLE_VALUE_MAX; | |
404 | g_assert(throttle_is_valid(&cfg, NULL)); | |
405 | ||
406 | b->avg = THROTTLE_VALUE_MAX; | |
407 | b->max = THROTTLE_VALUE_MAX; | |
408 | g_assert(throttle_is_valid(&cfg, NULL)); | |
409 | ||
410 | /* Values over THROTTLE_VALUE_MAX are not allowed */ | |
411 | b->avg = THROTTLE_VALUE_MAX + 1; | |
412 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
413 | ||
414 | b->avg = THROTTLE_VALUE_MAX; | |
415 | b->max = THROTTLE_VALUE_MAX + 1; | |
416 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
417 | ||
418 | /* burst_length must be between 1 and THROTTLE_VALUE_MAX */ | |
419 | b->avg = 1; | |
420 | b->max = 1; | |
421 | b->burst_length = 0; | |
422 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
423 | ||
424 | b->avg = 1; | |
425 | b->max = 1; | |
426 | b->burst_length = 1; | |
427 | g_assert(throttle_is_valid(&cfg, NULL)); | |
428 | ||
429 | b->avg = 1; | |
430 | b->max = 1; | |
431 | b->burst_length = THROTTLE_VALUE_MAX; | |
432 | g_assert(throttle_is_valid(&cfg, NULL)); | |
433 | ||
434 | b->avg = 1; | |
435 | b->max = 1; | |
436 | b->burst_length = THROTTLE_VALUE_MAX + 1; | |
437 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
438 | ||
439 | /* burst_length * max cannot exceed THROTTLE_VALUE_MAX */ | |
440 | b->avg = 1; | |
441 | b->max = 2; | |
442 | b->burst_length = THROTTLE_VALUE_MAX / 2; | |
443 | g_assert(throttle_is_valid(&cfg, NULL)); | |
444 | ||
445 | b->avg = 1; | |
446 | b->max = 3; | |
447 | b->burst_length = THROTTLE_VALUE_MAX / 2; | |
448 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
449 | ||
450 | b->avg = 1; | |
451 | b->max = THROTTLE_VALUE_MAX; | |
452 | b->burst_length = 1; | |
453 | g_assert(throttle_is_valid(&cfg, NULL)); | |
454 | ||
455 | b->avg = 1; | |
456 | b->max = THROTTLE_VALUE_MAX; | |
457 | b->burst_length = 2; | |
458 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
459 | } | |
460 | } | |
461 | ||
462 | static void test_max_is_missing_limit(void) | |
463 | { | |
464 | int i; | |
465 | ||
466 | for (i = 0; i < BUCKETS_COUNT; i++) { | |
467 | throttle_config_init(&cfg); | |
468 | cfg.buckets[i].max = 100; | |
469 | cfg.buckets[i].avg = 0; | |
470 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
471 | ||
472 | cfg.buckets[i].max = 0; | |
473 | cfg.buckets[i].avg = 0; | |
474 | g_assert(throttle_is_valid(&cfg, NULL)); | |
475 | ||
476 | cfg.buckets[i].max = 0; | |
477 | cfg.buckets[i].avg = 100; | |
478 | g_assert(throttle_is_valid(&cfg, NULL)); | |
479 | ||
480 | cfg.buckets[i].max = 30; | |
481 | cfg.buckets[i].avg = 100; | |
482 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
483 | ||
484 | cfg.buckets[i].max = 100; | |
485 | cfg.buckets[i].avg = 100; | |
486 | g_assert(throttle_is_valid(&cfg, NULL)); | |
487 | } | |
488 | } | |
489 | ||
490 | static void test_iops_size_is_missing_limit(void) | |
491 | { | |
492 | /* A total/read/write iops limit is required */ | |
493 | throttle_config_init(&cfg); | |
494 | cfg.op_size = 4096; | |
495 | g_assert(!throttle_is_valid(&cfg, NULL)); | |
496 | } | |
497 | ||
498 | static void test_have_timer(void) | |
499 | { | |
500 | /* zero structures */ | |
501 | memset(&ts, 0, sizeof(ts)); | |
502 | memset(tt, 0, sizeof(*tt)); | |
503 | ||
504 | /* no timer set should return false */ | |
505 | g_assert(!throttle_timers_are_initialized(tt)); | |
506 | ||
507 | /* init structures */ | |
508 | throttle_init(&ts); | |
509 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
510 | read_timer_cb, write_timer_cb, &ts); | |
511 | ||
512 | /* timer set by init should return true */ | |
513 | g_assert(throttle_timers_are_initialized(tt)); | |
514 | ||
515 | throttle_timers_destroy(tt); | |
516 | } | |
517 | ||
518 | static void test_detach_attach(void) | |
519 | { | |
520 | /* zero structures */ | |
521 | memset(&ts, 0, sizeof(ts)); | |
522 | memset(tt, 0, sizeof(*tt)); | |
523 | ||
524 | /* init the structure */ | |
525 | throttle_init(&ts); | |
526 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
527 | read_timer_cb, write_timer_cb, &ts); | |
528 | ||
529 | /* timer set by init should return true */ | |
530 | g_assert(throttle_timers_are_initialized(tt)); | |
531 | ||
532 | /* timer should no longer exist after detaching */ | |
533 | throttle_timers_detach_aio_context(tt); | |
534 | g_assert(!throttle_timers_are_initialized(tt)); | |
535 | ||
536 | /* timer should exist again after attaching */ | |
537 | throttle_timers_attach_aio_context(tt, ctx); | |
538 | g_assert(throttle_timers_are_initialized(tt)); | |
539 | ||
540 | throttle_timers_destroy(tt); | |
541 | } | |
542 | ||
543 | static bool do_test_accounting(bool is_ops, /* are we testing bps or ops */ | |
544 | int size, /* size of the operation to do */ | |
545 | double avg, /* io limit */ | |
546 | uint64_t op_size, /* ideal size of an io */ | |
547 | double total_result, | |
548 | double read_result, | |
549 | double write_result) | |
550 | { | |
551 | BucketType to_test[2][3] = { { THROTTLE_BPS_TOTAL, | |
552 | THROTTLE_BPS_READ, | |
553 | THROTTLE_BPS_WRITE, }, | |
554 | { THROTTLE_OPS_TOTAL, | |
555 | THROTTLE_OPS_READ, | |
556 | THROTTLE_OPS_WRITE, } }; | |
557 | ThrottleConfig cfg; | |
558 | BucketType index; | |
559 | int i; | |
560 | ||
561 | throttle_config_init(&cfg); | |
562 | ||
563 | for (i = 0; i < 3; i++) { | |
564 | BucketType index = to_test[is_ops][i]; | |
565 | cfg.buckets[index].avg = avg; | |
566 | } | |
567 | ||
568 | cfg.op_size = op_size; | |
569 | ||
570 | throttle_init(&ts); | |
571 | throttle_timers_init(tt, ctx, QEMU_CLOCK_VIRTUAL, | |
572 | read_timer_cb, write_timer_cb, &ts); | |
573 | throttle_config(&ts, QEMU_CLOCK_VIRTUAL, &cfg); | |
574 | ||
575 | /* account a read */ | |
576 | throttle_account(&ts, false, size); | |
577 | /* account a write */ | |
578 | throttle_account(&ts, true, size); | |
579 | ||
580 | /* check total result */ | |
581 | index = to_test[is_ops][0]; | |
582 | if (!double_cmp(ts.cfg.buckets[index].level, total_result)) { | |
583 | return false; | |
584 | } | |
585 | ||
586 | /* check read result */ | |
587 | index = to_test[is_ops][1]; | |
588 | if (!double_cmp(ts.cfg.buckets[index].level, read_result)) { | |
589 | return false; | |
590 | } | |
591 | ||
592 | /* check write result */ | |
593 | index = to_test[is_ops][2]; | |
594 | if (!double_cmp(ts.cfg.buckets[index].level, write_result)) { | |
595 | return false; | |
596 | } | |
597 | ||
598 | throttle_timers_destroy(tt); | |
599 | ||
600 | return true; | |
601 | } | |
602 | ||
603 | static void test_accounting(void) | |
604 | { | |
605 | /* tests for bps */ | |
606 | ||
607 | /* op of size 1 */ | |
608 | g_assert(do_test_accounting(false, | |
609 | 1 * 512, | |
610 | 150, | |
611 | 0, | |
612 | 1024, | |
613 | 512, | |
614 | 512)); | |
615 | ||
616 | /* op of size 2 */ | |
617 | g_assert(do_test_accounting(false, | |
618 | 2 * 512, | |
619 | 150, | |
620 | 0, | |
621 | 2048, | |
622 | 1024, | |
623 | 1024)); | |
624 | ||
625 | /* op of size 2 and orthogonal parameter change */ | |
626 | g_assert(do_test_accounting(false, | |
627 | 2 * 512, | |
628 | 150, | |
629 | 17, | |
630 | 2048, | |
631 | 1024, | |
632 | 1024)); | |
633 | ||
634 | ||
635 | /* tests for ops */ | |
636 | ||
637 | /* op of size 1 */ | |
638 | g_assert(do_test_accounting(true, | |
639 | 1 * 512, | |
640 | 150, | |
641 | 0, | |
642 | 2, | |
643 | 1, | |
644 | 1)); | |
645 | ||
646 | /* op of size 2 */ | |
647 | g_assert(do_test_accounting(true, | |
648 | 2 * 512, | |
649 | 150, | |
650 | 0, | |
651 | 2, | |
652 | 1, | |
653 | 1)); | |
654 | ||
655 | /* jumbo op accounting fragmentation : size 64 with op size of 13 units */ | |
656 | g_assert(do_test_accounting(true, | |
657 | 64 * 512, | |
658 | 150, | |
659 | 13 * 512, | |
660 | (64.0 * 2) / 13, | |
661 | (64.0 / 13), | |
662 | (64.0 / 13))); | |
663 | ||
664 | /* same with orthogonal parameters changes */ | |
665 | g_assert(do_test_accounting(true, | |
666 | 64 * 512, | |
667 | 300, | |
668 | 13 * 512, | |
669 | (64.0 * 2) / 13, | |
670 | (64.0 / 13), | |
671 | (64.0 / 13))); | |
672 | } | |
673 | ||
674 | static void test_groups(void) | |
675 | { | |
676 | ThrottleConfig cfg1, cfg2; | |
677 | BlockBackend *blk1, *blk2, *blk3; | |
678 | BlockBackendPublic *blkp1, *blkp2, *blkp3; | |
679 | ThrottleGroupMember *tgm1, *tgm2, *tgm3; | |
680 | ||
681 | /* No actual I/O is performed on these devices */ | |
682 | blk1 = blk_new(qemu_get_aio_context(), 0, BLK_PERM_ALL); | |
683 | blk2 = blk_new(qemu_get_aio_context(), 0, BLK_PERM_ALL); | |
684 | blk3 = blk_new(qemu_get_aio_context(), 0, BLK_PERM_ALL); | |
685 | ||
686 | blkp1 = blk_get_public(blk1); | |
687 | blkp2 = blk_get_public(blk2); | |
688 | blkp3 = blk_get_public(blk3); | |
689 | ||
690 | tgm1 = &blkp1->throttle_group_member; | |
691 | tgm2 = &blkp2->throttle_group_member; | |
692 | tgm3 = &blkp3->throttle_group_member; | |
693 | ||
694 | g_assert(tgm1->throttle_state == NULL); | |
695 | g_assert(tgm2->throttle_state == NULL); | |
696 | g_assert(tgm3->throttle_state == NULL); | |
697 | ||
698 | throttle_group_register_tgm(tgm1, "bar", blk_get_aio_context(blk1)); | |
699 | throttle_group_register_tgm(tgm2, "foo", blk_get_aio_context(blk2)); | |
700 | throttle_group_register_tgm(tgm3, "bar", blk_get_aio_context(blk3)); | |
701 | ||
702 | g_assert(tgm1->throttle_state != NULL); | |
703 | g_assert(tgm2->throttle_state != NULL); | |
704 | g_assert(tgm3->throttle_state != NULL); | |
705 | ||
706 | g_assert(!strcmp(throttle_group_get_name(tgm1), "bar")); | |
707 | g_assert(!strcmp(throttle_group_get_name(tgm2), "foo")); | |
708 | g_assert(tgm1->throttle_state == tgm3->throttle_state); | |
709 | ||
710 | /* Setting the config of a group member affects the whole group */ | |
711 | throttle_config_init(&cfg1); | |
712 | cfg1.buckets[THROTTLE_BPS_READ].avg = 500000; | |
713 | cfg1.buckets[THROTTLE_BPS_WRITE].avg = 285000; | |
714 | cfg1.buckets[THROTTLE_OPS_READ].avg = 20000; | |
715 | cfg1.buckets[THROTTLE_OPS_WRITE].avg = 12000; | |
716 | throttle_group_config(tgm1, &cfg1); | |
717 | ||
718 | throttle_group_get_config(tgm1, &cfg1); | |
719 | throttle_group_get_config(tgm3, &cfg2); | |
720 | g_assert(!memcmp(&cfg1, &cfg2, sizeof(cfg1))); | |
721 | ||
722 | cfg2.buckets[THROTTLE_BPS_READ].avg = 4547; | |
723 | cfg2.buckets[THROTTLE_BPS_WRITE].avg = 1349; | |
724 | cfg2.buckets[THROTTLE_OPS_READ].avg = 123; | |
725 | cfg2.buckets[THROTTLE_OPS_WRITE].avg = 86; | |
726 | throttle_group_config(tgm3, &cfg1); | |
727 | ||
728 | throttle_group_get_config(tgm1, &cfg1); | |
729 | throttle_group_get_config(tgm3, &cfg2); | |
730 | g_assert(!memcmp(&cfg1, &cfg2, sizeof(cfg1))); | |
731 | ||
732 | throttle_group_unregister_tgm(tgm1); | |
733 | throttle_group_unregister_tgm(tgm2); | |
734 | throttle_group_unregister_tgm(tgm3); | |
735 | ||
736 | g_assert(tgm1->throttle_state == NULL); | |
737 | g_assert(tgm2->throttle_state == NULL); | |
738 | g_assert(tgm3->throttle_state == NULL); | |
739 | } | |
740 | ||
741 | int main(int argc, char **argv) | |
742 | { | |
743 | qemu_init_main_loop(&error_fatal); | |
744 | ctx = qemu_get_aio_context(); | |
745 | bdrv_init(); | |
746 | module_call_init(MODULE_INIT_QOM); | |
747 | ||
748 | do {} while (g_main_context_iteration(NULL, false)); | |
749 | ||
750 | /* tests in the same order as the header function declarations */ | |
751 | g_test_init(&argc, &argv, NULL); | |
752 | g_test_add_func("/throttle/leak_bucket", test_leak_bucket); | |
753 | g_test_add_func("/throttle/compute_wait", test_compute_wait); | |
754 | g_test_add_func("/throttle/init", test_init); | |
755 | g_test_add_func("/throttle/destroy", test_destroy); | |
756 | g_test_add_func("/throttle/have_timer", test_have_timer); | |
757 | g_test_add_func("/throttle/detach_attach", test_detach_attach); | |
758 | g_test_add_func("/throttle/config/enabled", test_enabled); | |
759 | g_test_add_func("/throttle/config/conflicting", test_conflicting_config); | |
760 | g_test_add_func("/throttle/config/is_valid", test_is_valid); | |
761 | g_test_add_func("/throttle/config/ranges", test_ranges); | |
762 | g_test_add_func("/throttle/config/max", test_max_is_missing_limit); | |
763 | g_test_add_func("/throttle/config/iops_size", | |
764 | test_iops_size_is_missing_limit); | |
765 | g_test_add_func("/throttle/config_functions", test_config_functions); | |
766 | g_test_add_func("/throttle/accounting", test_accounting); | |
767 | g_test_add_func("/throttle/groups", test_groups); | |
768 | return g_test_run(); | |
769 | } | |
770 |