xref: /openbmc/qemu/migration/migration-stats.c (revision 0743f41f) 
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 
migration_rate_exceeded(QEMUFile * f)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();
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 
migration_rate_get(void)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 
migration_rate_set(uint64_t limit)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 
migration_rate_reset(void)57 void migration_rate_reset(void)
58 {
59     stat64_set(&mig_stats.rate_limit_start, migration_transferred_bytes());
60 }
61 
migration_transferred_bytes(void)62 uint64_t migration_transferred_bytes(void)
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 = stat64_get(&mig_stats.qemu_file_transferred);
67 
68     trace_migration_transferred_bytes(qemu_file, multifd, rdma);
69     return qemu_file + multifd + rdma;
70 }
71