1 /* 2 * Migration stats 3 * 4 * Copyright (c) 2012-2023 Red Hat Inc 5 * 6 * Authors: 7 * Juan Quintela <quintela@redhat.com> 8 * 9 * This work is licensed under the terms of the GNU GPL, version 2 or later. 10 * See the COPYING file in the top-level directory. 11 */ 12 13 #include "qemu/osdep.h" 14 #include "qemu/stats64.h" 15 #include "qemu-file.h" 16 #include "trace.h" 17 #include "migration-stats.h" 18 19 MigrationAtomicStats mig_stats; 20 21 bool migration_rate_exceeded(QEMUFile *f) 22 { 23 if (qemu_file_get_error(f)) { 24 return true; 25 } 26 27 uint64_t rate_limit_max = migration_rate_get(); 28 if (rate_limit_max == RATE_LIMIT_DISABLED) { 29 return false; 30 } 31 32 uint64_t rate_limit_start = stat64_get(&mig_stats.rate_limit_start); 33 uint64_t rate_limit_current = migration_transferred_bytes(f); 34 uint64_t rate_limit_used = rate_limit_current - rate_limit_start; 35 36 if (rate_limit_max > 0 && rate_limit_used > rate_limit_max) { 37 return true; 38 } 39 return false; 40 } 41 42 uint64_t migration_rate_get(void) 43 { 44 return stat64_get(&mig_stats.rate_limit_max); 45 } 46 47 #define XFER_LIMIT_RATIO (1000 / BUFFER_DELAY) 48 49 void migration_rate_set(uint64_t limit) 50 { 51 /* 52 * 'limit' is per second. But we check it each BUFFER_DELAY milliseconds. 53 */ 54 stat64_set(&mig_stats.rate_limit_max, limit / XFER_LIMIT_RATIO); 55 } 56 57 void migration_rate_reset(QEMUFile *f) 58 { 59 stat64_set(&mig_stats.rate_limit_start, migration_transferred_bytes(f)); 60 } 61 62 uint64_t migration_transferred_bytes(QEMUFile *f) 63 { 64 uint64_t multifd = stat64_get(&mig_stats.multifd_bytes); 65 uint64_t rdma = stat64_get(&mig_stats.rdma_bytes); 66 uint64_t qemu_file = qemu_file_transferred(f); 67 68 trace_migration_transferred_bytes(qemu_file, multifd, rdma); 69 return qemu_file + multifd + rdma; 70 } 71