1 /* 2 * QEMU block throttling group infrastructure 3 * 4 * Copyright (C) Nodalink, EURL. 2014 5 * Copyright (C) Igalia, S.L. 2015 6 * 7 * Authors: 8 * Benoît Canet <benoit.canet@nodalink.com> 9 * Alberto Garcia <berto@igalia.com> 10 * 11 * This program is free software; you can redistribute it and/or 12 * modify it under the terms of the GNU General Public License as 13 * published by the Free Software Foundation; either version 2 or 14 * (at your option) version 3 of the License. 15 * 16 * This program is distributed in the hope that it will be useful, 17 * but WITHOUT ANY WARRANTY; without even the implied warranty of 18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 19 * GNU General Public License for more details. 20 * 21 * You should have received a copy of the GNU General Public License 22 * along with this program; if not, see <http://www.gnu.org/licenses/>. 23 */ 24 25 #include "qemu/osdep.h" 26 #include "sysemu/block-backend.h" 27 #include "block/throttle-groups.h" 28 #include "qemu/throttle-options.h" 29 #include "qemu/queue.h" 30 #include "qemu/thread.h" 31 #include "sysemu/qtest.h" 32 #include "qapi/error.h" 33 #include "qapi-visit.h" 34 #include "qom/object.h" 35 #include "qom/object_interfaces.h" 36 37 static void throttle_group_obj_init(Object *obj); 38 static void throttle_group_obj_complete(UserCreatable *obj, Error **errp); 39 40 /* The ThrottleGroup structure (with its ThrottleState) is shared 41 * among different ThrottleGroupMembers and it's independent from 42 * AioContext, so in order to use it from different threads it needs 43 * its own locking. 44 * 45 * This locking is however handled internally in this file, so it's 46 * transparent to outside users. 47 * 48 * The whole ThrottleGroup structure is private and invisible to 49 * outside users, that only use it through its ThrottleState. 50 * 51 * In addition to the ThrottleGroup structure, ThrottleGroupMember has 52 * fields that need to be accessed by other members of the group and 53 * therefore also need to be protected by this lock. Once a 54 * ThrottleGroupMember is registered in a group those fields can be accessed 55 * by other threads any time. 56 * 57 * Again, all this is handled internally and is mostly transparent to 58 * the outside. The 'throttle_timers' field however has an additional 59 * constraint because it may be temporarily invalid (see for example 60 * blk_set_aio_context()). Therefore in this file a thread will 61 * access some other ThrottleGroupMember's timers only after verifying that 62 * that ThrottleGroupMember has throttled requests in the queue. 63 */ 64 typedef struct ThrottleGroup { 65 Object parent_obj; 66 67 /* refuse individual property change if initialization is complete */ 68 bool is_initialized; 69 char *name; /* This is constant during the lifetime of the group */ 70 71 QemuMutex lock; /* This lock protects the following four fields */ 72 ThrottleState ts; 73 QLIST_HEAD(, ThrottleGroupMember) head; 74 ThrottleGroupMember *tokens[2]; 75 bool any_timer_armed[2]; 76 QEMUClockType clock_type; 77 78 /* This field is protected by the global QEMU mutex */ 79 QTAILQ_ENTRY(ThrottleGroup) list; 80 } ThrottleGroup; 81 82 /* This is protected by the global QEMU mutex */ 83 static QTAILQ_HEAD(, ThrottleGroup) throttle_groups = 84 QTAILQ_HEAD_INITIALIZER(throttle_groups); 85 86 87 /* This function reads throttle_groups and must be called under the global 88 * mutex. 89 */ 90 static ThrottleGroup *throttle_group_by_name(const char *name) 91 { 92 ThrottleGroup *iter; 93 94 /* Look for an existing group with that name */ 95 QTAILQ_FOREACH(iter, &throttle_groups, list) { 96 if (!g_strcmp0(name, iter->name)) { 97 return iter; 98 } 99 } 100 101 return NULL; 102 } 103 104 /* This function reads throttle_groups and must be called under the global 105 * mutex. 106 */ 107 bool throttle_group_exists(const char *name) 108 { 109 return throttle_group_by_name(name) != NULL; 110 } 111 112 /* Increments the reference count of a ThrottleGroup given its name. 113 * 114 * If no ThrottleGroup is found with the given name a new one is 115 * created. 116 * 117 * This function edits throttle_groups and must be called under the global 118 * mutex. 119 * 120 * @name: the name of the ThrottleGroup 121 * @ret: the ThrottleState member of the ThrottleGroup 122 */ 123 ThrottleState *throttle_group_incref(const char *name) 124 { 125 ThrottleGroup *tg = NULL; 126 127 /* Look for an existing group with that name */ 128 tg = throttle_group_by_name(name); 129 130 if (tg) { 131 object_ref(OBJECT(tg)); 132 } else { 133 /* Create a new one if not found */ 134 /* new ThrottleGroup obj will have a refcnt = 1 */ 135 tg = THROTTLE_GROUP(object_new(TYPE_THROTTLE_GROUP)); 136 tg->name = g_strdup(name); 137 throttle_group_obj_complete(USER_CREATABLE(tg), &error_abort); 138 } 139 140 return &tg->ts; 141 } 142 143 /* Decrease the reference count of a ThrottleGroup. 144 * 145 * When the reference count reaches zero the ThrottleGroup is 146 * destroyed. 147 * 148 * This function edits throttle_groups and must be called under the global 149 * mutex. 150 * 151 * @ts: The ThrottleGroup to unref, given by its ThrottleState member 152 */ 153 void throttle_group_unref(ThrottleState *ts) 154 { 155 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 156 object_unref(OBJECT(tg)); 157 } 158 159 /* Get the name from a ThrottleGroupMember's group. The name (and the pointer) 160 * is guaranteed to remain constant during the lifetime of the group. 161 * 162 * @tgm: a ThrottleGroupMember 163 * @ret: the name of the group. 164 */ 165 const char *throttle_group_get_name(ThrottleGroupMember *tgm) 166 { 167 ThrottleGroup *tg = container_of(tgm->throttle_state, ThrottleGroup, ts); 168 return tg->name; 169 } 170 171 /* Return the next ThrottleGroupMember in the round-robin sequence, simulating 172 * a circular list. 173 * 174 * This assumes that tg->lock is held. 175 * 176 * @tgm: the current ThrottleGroupMember 177 * @ret: the next ThrottleGroupMember in the sequence 178 */ 179 static ThrottleGroupMember *throttle_group_next_tgm(ThrottleGroupMember *tgm) 180 { 181 ThrottleState *ts = tgm->throttle_state; 182 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 183 ThrottleGroupMember *next = QLIST_NEXT(tgm, round_robin); 184 185 if (!next) { 186 next = QLIST_FIRST(&tg->head); 187 } 188 189 return next; 190 } 191 192 /* 193 * Return whether a ThrottleGroupMember has pending requests. 194 * 195 * This assumes that tg->lock is held. 196 * 197 * @tgm: the ThrottleGroupMember 198 * @is_write: the type of operation (read/write) 199 * @ret: whether the ThrottleGroupMember has pending requests. 200 */ 201 static inline bool tgm_has_pending_reqs(ThrottleGroupMember *tgm, 202 bool is_write) 203 { 204 return tgm->pending_reqs[is_write]; 205 } 206 207 /* Return the next ThrottleGroupMember in the round-robin sequence with pending 208 * I/O requests. 209 * 210 * This assumes that tg->lock is held. 211 * 212 * @tgm: the current ThrottleGroupMember 213 * @is_write: the type of operation (read/write) 214 * @ret: the next ThrottleGroupMember with pending requests, or tgm if 215 * there is none. 216 */ 217 static ThrottleGroupMember *next_throttle_token(ThrottleGroupMember *tgm, 218 bool is_write) 219 { 220 ThrottleState *ts = tgm->throttle_state; 221 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 222 ThrottleGroupMember *token, *start; 223 224 start = token = tg->tokens[is_write]; 225 226 /* get next bs round in round robin style */ 227 token = throttle_group_next_tgm(token); 228 while (token != start && !tgm_has_pending_reqs(token, is_write)) { 229 token = throttle_group_next_tgm(token); 230 } 231 232 /* If no IO are queued for scheduling on the next round robin token 233 * then decide the token is the current tgm because chances are 234 * the current tgm got the current request queued. 235 */ 236 if (token == start && !tgm_has_pending_reqs(token, is_write)) { 237 token = tgm; 238 } 239 240 /* Either we return the original TGM, or one with pending requests */ 241 assert(token == tgm || tgm_has_pending_reqs(token, is_write)); 242 243 return token; 244 } 245 246 /* Check if the next I/O request for a ThrottleGroupMember needs to be 247 * throttled or not. If there's no timer set in this group, set one and update 248 * the token accordingly. 249 * 250 * This assumes that tg->lock is held. 251 * 252 * @tgm: the current ThrottleGroupMember 253 * @is_write: the type of operation (read/write) 254 * @ret: whether the I/O request needs to be throttled or not 255 */ 256 static bool throttle_group_schedule_timer(ThrottleGroupMember *tgm, 257 bool is_write) 258 { 259 ThrottleState *ts = tgm->throttle_state; 260 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 261 ThrottleTimers *tt = &tgm->throttle_timers; 262 bool must_wait; 263 264 if (atomic_read(&tgm->io_limits_disabled)) { 265 return false; 266 } 267 268 /* Check if any of the timers in this group is already armed */ 269 if (tg->any_timer_armed[is_write]) { 270 return true; 271 } 272 273 must_wait = throttle_schedule_timer(ts, tt, is_write); 274 275 /* If a timer just got armed, set tgm as the current token */ 276 if (must_wait) { 277 tg->tokens[is_write] = tgm; 278 tg->any_timer_armed[is_write] = true; 279 } 280 281 return must_wait; 282 } 283 284 /* Start the next pending I/O request for a ThrottleGroupMember. Return whether 285 * any request was actually pending. 286 * 287 * @tgm: the current ThrottleGroupMember 288 * @is_write: the type of operation (read/write) 289 */ 290 static bool coroutine_fn throttle_group_co_restart_queue(ThrottleGroupMember *tgm, 291 bool is_write) 292 { 293 bool ret; 294 295 qemu_co_mutex_lock(&tgm->throttled_reqs_lock); 296 ret = qemu_co_queue_next(&tgm->throttled_reqs[is_write]); 297 qemu_co_mutex_unlock(&tgm->throttled_reqs_lock); 298 299 return ret; 300 } 301 302 /* Look for the next pending I/O request and schedule it. 303 * 304 * This assumes that tg->lock is held. 305 * 306 * @tgm: the current ThrottleGroupMember 307 * @is_write: the type of operation (read/write) 308 */ 309 static void schedule_next_request(ThrottleGroupMember *tgm, bool is_write) 310 { 311 ThrottleState *ts = tgm->throttle_state; 312 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 313 bool must_wait; 314 ThrottleGroupMember *token; 315 316 /* Check if there's any pending request to schedule next */ 317 token = next_throttle_token(tgm, is_write); 318 if (!tgm_has_pending_reqs(token, is_write)) { 319 return; 320 } 321 322 /* Set a timer for the request if it needs to be throttled */ 323 must_wait = throttle_group_schedule_timer(token, is_write); 324 325 /* If it doesn't have to wait, queue it for immediate execution */ 326 if (!must_wait) { 327 /* Give preference to requests from the current tgm */ 328 if (qemu_in_coroutine() && 329 throttle_group_co_restart_queue(tgm, is_write)) { 330 token = tgm; 331 } else { 332 ThrottleTimers *tt = &token->throttle_timers; 333 int64_t now = qemu_clock_get_ns(tg->clock_type); 334 timer_mod(tt->timers[is_write], now); 335 tg->any_timer_armed[is_write] = true; 336 } 337 tg->tokens[is_write] = token; 338 } 339 } 340 341 /* Check if an I/O request needs to be throttled, wait and set a timer 342 * if necessary, and schedule the next request using a round robin 343 * algorithm. 344 * 345 * @tgm: the current ThrottleGroupMember 346 * @bytes: the number of bytes for this I/O 347 * @is_write: the type of operation (read/write) 348 */ 349 void coroutine_fn throttle_group_co_io_limits_intercept(ThrottleGroupMember *tgm, 350 unsigned int bytes, 351 bool is_write) 352 { 353 bool must_wait; 354 ThrottleGroupMember *token; 355 ThrottleGroup *tg = container_of(tgm->throttle_state, ThrottleGroup, ts); 356 qemu_mutex_lock(&tg->lock); 357 358 /* First we check if this I/O has to be throttled. */ 359 token = next_throttle_token(tgm, is_write); 360 must_wait = throttle_group_schedule_timer(token, is_write); 361 362 /* Wait if there's a timer set or queued requests of this type */ 363 if (must_wait || tgm->pending_reqs[is_write]) { 364 tgm->pending_reqs[is_write]++; 365 qemu_mutex_unlock(&tg->lock); 366 qemu_co_mutex_lock(&tgm->throttled_reqs_lock); 367 qemu_co_queue_wait(&tgm->throttled_reqs[is_write], 368 &tgm->throttled_reqs_lock); 369 qemu_co_mutex_unlock(&tgm->throttled_reqs_lock); 370 qemu_mutex_lock(&tg->lock); 371 tgm->pending_reqs[is_write]--; 372 } 373 374 /* The I/O will be executed, so do the accounting */ 375 throttle_account(tgm->throttle_state, is_write, bytes); 376 377 /* Schedule the next request */ 378 schedule_next_request(tgm, is_write); 379 380 qemu_mutex_unlock(&tg->lock); 381 } 382 383 typedef struct { 384 ThrottleGroupMember *tgm; 385 bool is_write; 386 } RestartData; 387 388 static void coroutine_fn throttle_group_restart_queue_entry(void *opaque) 389 { 390 RestartData *data = opaque; 391 ThrottleGroupMember *tgm = data->tgm; 392 ThrottleState *ts = tgm->throttle_state; 393 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 394 bool is_write = data->is_write; 395 bool empty_queue; 396 397 empty_queue = !throttle_group_co_restart_queue(tgm, is_write); 398 399 /* If the request queue was empty then we have to take care of 400 * scheduling the next one */ 401 if (empty_queue) { 402 qemu_mutex_lock(&tg->lock); 403 schedule_next_request(tgm, is_write); 404 qemu_mutex_unlock(&tg->lock); 405 } 406 407 g_free(data); 408 } 409 410 static void throttle_group_restart_queue(ThrottleGroupMember *tgm, bool is_write) 411 { 412 Coroutine *co; 413 RestartData *rd = g_new0(RestartData, 1); 414 415 rd->tgm = tgm; 416 rd->is_write = is_write; 417 418 co = qemu_coroutine_create(throttle_group_restart_queue_entry, rd); 419 aio_co_enter(tgm->aio_context, co); 420 } 421 422 void throttle_group_restart_tgm(ThrottleGroupMember *tgm) 423 { 424 if (tgm->throttle_state) { 425 throttle_group_restart_queue(tgm, 0); 426 throttle_group_restart_queue(tgm, 1); 427 } 428 } 429 430 /* Update the throttle configuration for a particular group. Similar 431 * to throttle_config(), but guarantees atomicity within the 432 * throttling group. 433 * 434 * @tgm: a ThrottleGroupMember that is a member of the group 435 * @cfg: the configuration to set 436 */ 437 void throttle_group_config(ThrottleGroupMember *tgm, ThrottleConfig *cfg) 438 { 439 ThrottleState *ts = tgm->throttle_state; 440 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 441 qemu_mutex_lock(&tg->lock); 442 throttle_config(ts, tg->clock_type, cfg); 443 qemu_mutex_unlock(&tg->lock); 444 445 throttle_group_restart_tgm(tgm); 446 } 447 448 /* Get the throttle configuration from a particular group. Similar to 449 * throttle_get_config(), but guarantees atomicity within the 450 * throttling group. 451 * 452 * @tgm: a ThrottleGroupMember that is a member of the group 453 * @cfg: the configuration will be written here 454 */ 455 void throttle_group_get_config(ThrottleGroupMember *tgm, ThrottleConfig *cfg) 456 { 457 ThrottleState *ts = tgm->throttle_state; 458 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 459 qemu_mutex_lock(&tg->lock); 460 throttle_get_config(ts, cfg); 461 qemu_mutex_unlock(&tg->lock); 462 } 463 464 /* ThrottleTimers callback. This wakes up a request that was waiting 465 * because it had been throttled. 466 * 467 * @tgm: the ThrottleGroupMember whose request had been throttled 468 * @is_write: the type of operation (read/write) 469 */ 470 static void timer_cb(ThrottleGroupMember *tgm, bool is_write) 471 { 472 ThrottleState *ts = tgm->throttle_state; 473 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 474 475 /* The timer has just been fired, so we can update the flag */ 476 qemu_mutex_lock(&tg->lock); 477 tg->any_timer_armed[is_write] = false; 478 qemu_mutex_unlock(&tg->lock); 479 480 /* Run the request that was waiting for this timer */ 481 throttle_group_restart_queue(tgm, is_write); 482 } 483 484 static void read_timer_cb(void *opaque) 485 { 486 timer_cb(opaque, false); 487 } 488 489 static void write_timer_cb(void *opaque) 490 { 491 timer_cb(opaque, true); 492 } 493 494 /* Register a ThrottleGroupMember from the throttling group, also initializing 495 * its timers and updating its throttle_state pointer to point to it. If a 496 * throttling group with that name does not exist yet, it will be created. 497 * 498 * This function edits throttle_groups and must be called under the global 499 * mutex. 500 * 501 * @tgm: the ThrottleGroupMember to insert 502 * @groupname: the name of the group 503 * @ctx: the AioContext to use 504 */ 505 void throttle_group_register_tgm(ThrottleGroupMember *tgm, 506 const char *groupname, 507 AioContext *ctx) 508 { 509 int i; 510 ThrottleState *ts = throttle_group_incref(groupname); 511 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 512 513 tgm->throttle_state = ts; 514 tgm->aio_context = ctx; 515 516 qemu_mutex_lock(&tg->lock); 517 /* If the ThrottleGroup is new set this ThrottleGroupMember as the token */ 518 for (i = 0; i < 2; i++) { 519 if (!tg->tokens[i]) { 520 tg->tokens[i] = tgm; 521 } 522 } 523 524 QLIST_INSERT_HEAD(&tg->head, tgm, round_robin); 525 526 throttle_timers_init(&tgm->throttle_timers, 527 tgm->aio_context, 528 tg->clock_type, 529 read_timer_cb, 530 write_timer_cb, 531 tgm); 532 qemu_co_mutex_init(&tgm->throttled_reqs_lock); 533 qemu_co_queue_init(&tgm->throttled_reqs[0]); 534 qemu_co_queue_init(&tgm->throttled_reqs[1]); 535 536 qemu_mutex_unlock(&tg->lock); 537 } 538 539 /* Unregister a ThrottleGroupMember from its group, removing it from the list, 540 * destroying the timers and setting the throttle_state pointer to NULL. 541 * 542 * The ThrottleGroupMember must not have pending throttled requests, so the 543 * caller has to drain them first. 544 * 545 * The group will be destroyed if it's empty after this operation. 546 * 547 * @tgm the ThrottleGroupMember to remove 548 */ 549 void throttle_group_unregister_tgm(ThrottleGroupMember *tgm) 550 { 551 ThrottleState *ts = tgm->throttle_state; 552 ThrottleGroup *tg = container_of(ts, ThrottleGroup, ts); 553 ThrottleGroupMember *token; 554 int i; 555 556 if (!ts) { 557 /* Discard already unregistered tgm */ 558 return; 559 } 560 561 assert(tgm->pending_reqs[0] == 0 && tgm->pending_reqs[1] == 0); 562 assert(qemu_co_queue_empty(&tgm->throttled_reqs[0])); 563 assert(qemu_co_queue_empty(&tgm->throttled_reqs[1])); 564 565 qemu_mutex_lock(&tg->lock); 566 for (i = 0; i < 2; i++) { 567 if (tg->tokens[i] == tgm) { 568 token = throttle_group_next_tgm(tgm); 569 /* Take care of the case where this is the last tgm in the group */ 570 if (token == tgm) { 571 token = NULL; 572 } 573 tg->tokens[i] = token; 574 } 575 } 576 577 /* remove the current tgm from the list */ 578 QLIST_REMOVE(tgm, round_robin); 579 throttle_timers_destroy(&tgm->throttle_timers); 580 qemu_mutex_unlock(&tg->lock); 581 582 throttle_group_unref(&tg->ts); 583 tgm->throttle_state = NULL; 584 } 585 586 void throttle_group_attach_aio_context(ThrottleGroupMember *tgm, 587 AioContext *new_context) 588 { 589 ThrottleTimers *tt = &tgm->throttle_timers; 590 throttle_timers_attach_aio_context(tt, new_context); 591 tgm->aio_context = new_context; 592 } 593 594 void throttle_group_detach_aio_context(ThrottleGroupMember *tgm) 595 { 596 ThrottleTimers *tt = &tgm->throttle_timers; 597 598 /* Requests must have been drained */ 599 assert(tgm->pending_reqs[0] == 0 && tgm->pending_reqs[1] == 0); 600 assert(qemu_co_queue_empty(&tgm->throttled_reqs[0])); 601 assert(qemu_co_queue_empty(&tgm->throttled_reqs[1])); 602 603 throttle_timers_detach_aio_context(tt); 604 tgm->aio_context = NULL; 605 } 606 607 #undef THROTTLE_OPT_PREFIX 608 #define THROTTLE_OPT_PREFIX "x-" 609 610 /* Helper struct and array for QOM property setter/getter */ 611 typedef struct { 612 const char *name; 613 BucketType type; 614 enum { 615 AVG, 616 MAX, 617 BURST_LENGTH, 618 IOPS_SIZE, 619 } category; 620 } ThrottleParamInfo; 621 622 static ThrottleParamInfo properties[] = { 623 { 624 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_TOTAL, 625 THROTTLE_OPS_TOTAL, AVG, 626 }, 627 { 628 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_TOTAL_MAX, 629 THROTTLE_OPS_TOTAL, MAX, 630 }, 631 { 632 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_TOTAL_MAX_LENGTH, 633 THROTTLE_OPS_TOTAL, BURST_LENGTH, 634 }, 635 { 636 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_READ, 637 THROTTLE_OPS_READ, AVG, 638 }, 639 { 640 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_READ_MAX, 641 THROTTLE_OPS_READ, MAX, 642 }, 643 { 644 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_READ_MAX_LENGTH, 645 THROTTLE_OPS_READ, BURST_LENGTH, 646 }, 647 { 648 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_WRITE, 649 THROTTLE_OPS_WRITE, AVG, 650 }, 651 { 652 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_WRITE_MAX, 653 THROTTLE_OPS_WRITE, MAX, 654 }, 655 { 656 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_WRITE_MAX_LENGTH, 657 THROTTLE_OPS_WRITE, BURST_LENGTH, 658 }, 659 { 660 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_TOTAL, 661 THROTTLE_BPS_TOTAL, AVG, 662 }, 663 { 664 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_TOTAL_MAX, 665 THROTTLE_BPS_TOTAL, MAX, 666 }, 667 { 668 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_TOTAL_MAX_LENGTH, 669 THROTTLE_BPS_TOTAL, BURST_LENGTH, 670 }, 671 { 672 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_READ, 673 THROTTLE_BPS_READ, AVG, 674 }, 675 { 676 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_READ_MAX, 677 THROTTLE_BPS_READ, MAX, 678 }, 679 { 680 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_READ_MAX_LENGTH, 681 THROTTLE_BPS_READ, BURST_LENGTH, 682 }, 683 { 684 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_WRITE, 685 THROTTLE_BPS_WRITE, AVG, 686 }, 687 { 688 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_WRITE_MAX, 689 THROTTLE_BPS_WRITE, MAX, 690 }, 691 { 692 THROTTLE_OPT_PREFIX QEMU_OPT_BPS_WRITE_MAX_LENGTH, 693 THROTTLE_BPS_WRITE, BURST_LENGTH, 694 }, 695 { 696 THROTTLE_OPT_PREFIX QEMU_OPT_IOPS_SIZE, 697 0, IOPS_SIZE, 698 } 699 }; 700 701 /* This function edits throttle_groups and must be called under the global 702 * mutex */ 703 static void throttle_group_obj_init(Object *obj) 704 { 705 ThrottleGroup *tg = THROTTLE_GROUP(obj); 706 707 tg->clock_type = QEMU_CLOCK_REALTIME; 708 if (qtest_enabled()) { 709 /* For testing block IO throttling only */ 710 tg->clock_type = QEMU_CLOCK_VIRTUAL; 711 } 712 tg->is_initialized = false; 713 qemu_mutex_init(&tg->lock); 714 throttle_init(&tg->ts); 715 QLIST_INIT(&tg->head); 716 } 717 718 /* This function edits throttle_groups and must be called under the global 719 * mutex */ 720 static void throttle_group_obj_complete(UserCreatable *obj, Error **errp) 721 { 722 ThrottleGroup *tg = THROTTLE_GROUP(obj); 723 ThrottleConfig cfg; 724 725 /* set group name to object id if it exists */ 726 if (!tg->name && tg->parent_obj.parent) { 727 tg->name = object_get_canonical_path_component(OBJECT(obj)); 728 } 729 /* We must have a group name at this point */ 730 assert(tg->name); 731 732 /* error if name is duplicate */ 733 if (throttle_group_exists(tg->name)) { 734 error_setg(errp, "A group with this name already exists"); 735 return; 736 } 737 738 /* check validity */ 739 throttle_get_config(&tg->ts, &cfg); 740 if (!throttle_is_valid(&cfg, errp)) { 741 return; 742 } 743 throttle_config(&tg->ts, tg->clock_type, &cfg); 744 QTAILQ_INSERT_TAIL(&throttle_groups, tg, list); 745 tg->is_initialized = true; 746 } 747 748 /* This function edits throttle_groups and must be called under the global 749 * mutex */ 750 static void throttle_group_obj_finalize(Object *obj) 751 { 752 ThrottleGroup *tg = THROTTLE_GROUP(obj); 753 if (tg->is_initialized) { 754 QTAILQ_REMOVE(&throttle_groups, tg, list); 755 } 756 qemu_mutex_destroy(&tg->lock); 757 g_free(tg->name); 758 } 759 760 static void throttle_group_set(Object *obj, Visitor *v, const char * name, 761 void *opaque, Error **errp) 762 763 { 764 ThrottleGroup *tg = THROTTLE_GROUP(obj); 765 ThrottleConfig *cfg; 766 ThrottleParamInfo *info = opaque; 767 Error *local_err = NULL; 768 int64_t value; 769 770 /* If we have finished initialization, don't accept individual property 771 * changes through QOM. Throttle configuration limits must be set in one 772 * transaction, as certain combinations are invalid. 773 */ 774 if (tg->is_initialized) { 775 error_setg(&local_err, "Property cannot be set after initialization"); 776 goto ret; 777 } 778 779 visit_type_int64(v, name, &value, &local_err); 780 if (local_err) { 781 goto ret; 782 } 783 if (value < 0) { 784 error_setg(&local_err, "Property values cannot be negative"); 785 goto ret; 786 } 787 788 cfg = &tg->ts.cfg; 789 switch (info->category) { 790 case AVG: 791 cfg->buckets[info->type].avg = value; 792 break; 793 case MAX: 794 cfg->buckets[info->type].max = value; 795 break; 796 case BURST_LENGTH: 797 if (value > UINT_MAX) { 798 error_setg(&local_err, "%s value must be in the" 799 "range [0, %u]", info->name, UINT_MAX); 800 goto ret; 801 } 802 cfg->buckets[info->type].burst_length = value; 803 break; 804 case IOPS_SIZE: 805 cfg->op_size = value; 806 break; 807 } 808 809 ret: 810 error_propagate(errp, local_err); 811 return; 812 813 } 814 815 static void throttle_group_get(Object *obj, Visitor *v, const char *name, 816 void *opaque, Error **errp) 817 { 818 ThrottleGroup *tg = THROTTLE_GROUP(obj); 819 ThrottleConfig cfg; 820 ThrottleParamInfo *info = opaque; 821 int64_t value; 822 823 throttle_get_config(&tg->ts, &cfg); 824 switch (info->category) { 825 case AVG: 826 value = cfg.buckets[info->type].avg; 827 break; 828 case MAX: 829 value = cfg.buckets[info->type].max; 830 break; 831 case BURST_LENGTH: 832 value = cfg.buckets[info->type].burst_length; 833 break; 834 case IOPS_SIZE: 835 value = cfg.op_size; 836 break; 837 } 838 839 visit_type_int64(v, name, &value, errp); 840 } 841 842 static void throttle_group_set_limits(Object *obj, Visitor *v, 843 const char *name, void *opaque, 844 Error **errp) 845 846 { 847 ThrottleGroup *tg = THROTTLE_GROUP(obj); 848 ThrottleConfig cfg; 849 ThrottleLimits arg = { 0 }; 850 ThrottleLimits *argp = &arg; 851 Error *local_err = NULL; 852 853 visit_type_ThrottleLimits(v, name, &argp, &local_err); 854 if (local_err) { 855 goto ret; 856 } 857 qemu_mutex_lock(&tg->lock); 858 throttle_get_config(&tg->ts, &cfg); 859 throttle_limits_to_config(argp, &cfg, &local_err); 860 if (local_err) { 861 goto unlock; 862 } 863 throttle_config(&tg->ts, tg->clock_type, &cfg); 864 865 unlock: 866 qemu_mutex_unlock(&tg->lock); 867 ret: 868 error_propagate(errp, local_err); 869 return; 870 } 871 872 static void throttle_group_get_limits(Object *obj, Visitor *v, 873 const char *name, void *opaque, 874 Error **errp) 875 { 876 ThrottleGroup *tg = THROTTLE_GROUP(obj); 877 ThrottleConfig cfg; 878 ThrottleLimits arg = { 0 }; 879 ThrottleLimits *argp = &arg; 880 881 qemu_mutex_lock(&tg->lock); 882 throttle_get_config(&tg->ts, &cfg); 883 qemu_mutex_unlock(&tg->lock); 884 885 throttle_config_to_limits(&cfg, argp); 886 887 visit_type_ThrottleLimits(v, name, &argp, errp); 888 } 889 890 static bool throttle_group_can_be_deleted(UserCreatable *uc) 891 { 892 return OBJECT(uc)->ref == 1; 893 } 894 895 static void throttle_group_obj_class_init(ObjectClass *klass, void *class_data) 896 { 897 size_t i = 0; 898 UserCreatableClass *ucc = USER_CREATABLE_CLASS(klass); 899 900 ucc->complete = throttle_group_obj_complete; 901 ucc->can_be_deleted = throttle_group_can_be_deleted; 902 903 /* individual properties */ 904 for (i = 0; i < sizeof(properties) / sizeof(ThrottleParamInfo); i++) { 905 object_class_property_add(klass, 906 properties[i].name, 907 "int", 908 throttle_group_get, 909 throttle_group_set, 910 NULL, &properties[i], 911 &error_abort); 912 } 913 914 /* ThrottleLimits */ 915 object_class_property_add(klass, 916 "limits", "ThrottleLimits", 917 throttle_group_get_limits, 918 throttle_group_set_limits, 919 NULL, NULL, 920 &error_abort); 921 } 922 923 static const TypeInfo throttle_group_info = { 924 .name = TYPE_THROTTLE_GROUP, 925 .parent = TYPE_OBJECT, 926 .class_init = throttle_group_obj_class_init, 927 .instance_size = sizeof(ThrottleGroup), 928 .instance_init = throttle_group_obj_init, 929 .instance_finalize = throttle_group_obj_finalize, 930 .interfaces = (InterfaceInfo[]) { 931 { TYPE_USER_CREATABLE }, 932 { } 933 }, 934 }; 935 936 static void throttle_groups_init(void) 937 { 938 type_register_static(&throttle_group_info); 939 } 940 941 type_init(throttle_groups_init); 942