xref: /openbmc/qemu/migration/dirtyrate.c (revision 69430111) 
1 /*
2  * Dirtyrate implement code
3  *
4  * Copyright (c) 2020 HUAWEI TECHNOLOGIES CO.,LTD.
5  *
6  * Authors:
7  *  Chuan Zheng <zhengchuan@huawei.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 
15 #include <zlib.h>
16 #include "qapi/error.h"
17 #include "cpu.h"
18 #include "qemu/config-file.h"
19 #include "exec/memory.h"
20 #include "exec/ramblock.h"
21 #include "exec/target_page.h"
22 #include "qemu/rcu_queue.h"
23 #include "qapi/qapi-commands-migration.h"
24 #include "migration.h"
25 #include "ram.h"
26 #include "trace.h"
27 #include "dirtyrate.h"
28 
29 static int CalculatingState = DIRTY_RATE_STATUS_UNSTARTED;
30 static struct DirtyRateStat DirtyStat;
31 
32 static int64_t set_sample_page_period(int64_t msec, int64_t initial_time)
33 {
34     int64_t current_time;
35 
36     current_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
37     if ((current_time - initial_time) >= msec) {
38         msec = current_time - initial_time;
39     } else {
40         g_usleep((msec + initial_time - current_time) * 1000);
41     }
42 
43     return msec;
44 }
45 
46 static bool is_sample_period_valid(int64_t sec)
47 {
48     if (sec < MIN_FETCH_DIRTYRATE_TIME_SEC ||
49         sec > MAX_FETCH_DIRTYRATE_TIME_SEC) {
50         return false;
51     }
52 
53     return true;
54 }
55 
56 static int dirtyrate_set_state(int *state, int old_state, int new_state)
57 {
58     assert(new_state < DIRTY_RATE_STATUS__MAX);
59     trace_dirtyrate_set_state(DirtyRateStatus_str(new_state));
60     if (qatomic_cmpxchg(state, old_state, new_state) == old_state) {
61         return 0;
62     } else {
63         return -1;
64     }
65 }
66 
67 static struct DirtyRateInfo *query_dirty_rate_info(void)
68 {
69     int64_t dirty_rate = DirtyStat.dirty_rate;
70     struct DirtyRateInfo *info = g_malloc0(sizeof(DirtyRateInfo));
71 
72     if (qatomic_read(&CalculatingState) == DIRTY_RATE_STATUS_MEASURED) {
73         info->has_dirty_rate = true;
74         info->dirty_rate = dirty_rate;
75     }
76 
77     info->status = CalculatingState;
78     info->start_time = DirtyStat.start_time;
79     info->calc_time = DirtyStat.calc_time;
80 
81     trace_query_dirty_rate_info(DirtyRateStatus_str(CalculatingState));
82 
83     return info;
84 }
85 
86 static void init_dirtyrate_stat(int64_t start_time, int64_t calc_time)
87 {
88     DirtyStat.total_dirty_samples = 0;
89     DirtyStat.total_sample_count = 0;
90     DirtyStat.total_block_mem_MB = 0;
91     DirtyStat.dirty_rate = -1;
92     DirtyStat.start_time = start_time;
93     DirtyStat.calc_time = calc_time;
94 }
95 
96 static void update_dirtyrate_stat(struct RamblockDirtyInfo *info)
97 {
98     DirtyStat.total_dirty_samples += info->sample_dirty_count;
99     DirtyStat.total_sample_count += info->sample_pages_count;
100     /* size of total pages in MB */
101     DirtyStat.total_block_mem_MB += (info->ramblock_pages *
102                                      TARGET_PAGE_SIZE) >> 20;
103 }
104 
105 static void update_dirtyrate(uint64_t msec)
106 {
107     uint64_t dirtyrate;
108     uint64_t total_dirty_samples = DirtyStat.total_dirty_samples;
109     uint64_t total_sample_count = DirtyStat.total_sample_count;
110     uint64_t total_block_mem_MB = DirtyStat.total_block_mem_MB;
111 
112     dirtyrate = total_dirty_samples * total_block_mem_MB *
113                 1000 / (total_sample_count * msec);
114 
115     DirtyStat.dirty_rate = dirtyrate;
116 }
117 
118 /*
119  * get hash result for the sampled memory with length of TARGET_PAGE_SIZE
120  * in ramblock, which starts from ramblock base address.
121  */
122 static uint32_t get_ramblock_vfn_hash(struct RamblockDirtyInfo *info,
123                                       uint64_t vfn)
124 {
125     uint32_t crc;
126 
127     crc = crc32(0, (info->ramblock_addr +
128                 vfn * TARGET_PAGE_SIZE), TARGET_PAGE_SIZE);
129 
130     trace_get_ramblock_vfn_hash(info->idstr, vfn, crc);
131     return crc;
132 }
133 
134 static bool save_ramblock_hash(struct RamblockDirtyInfo *info)
135 {
136     unsigned int sample_pages_count;
137     int i;
138     GRand *rand;
139 
140     sample_pages_count = info->sample_pages_count;
141 
142     /* ramblock size less than one page, return success to skip this ramblock */
143     if (unlikely(info->ramblock_pages == 0 || sample_pages_count == 0)) {
144         return true;
145     }
146 
147     info->hash_result = g_try_malloc0_n(sample_pages_count,
148                                         sizeof(uint32_t));
149     if (!info->hash_result) {
150         return false;
151     }
152 
153     info->sample_page_vfn = g_try_malloc0_n(sample_pages_count,
154                                             sizeof(uint64_t));
155     if (!info->sample_page_vfn) {
156         g_free(info->hash_result);
157         return false;
158     }
159 
160     rand  = g_rand_new();
161     for (i = 0; i < sample_pages_count; i++) {
162         info->sample_page_vfn[i] = g_rand_int_range(rand, 0,
163                                                     info->ramblock_pages - 1);
164         info->hash_result[i] = get_ramblock_vfn_hash(info,
165                                                      info->sample_page_vfn[i]);
166     }
167     g_rand_free(rand);
168 
169     return true;
170 }
171 
172 static void get_ramblock_dirty_info(RAMBlock *block,
173                                     struct RamblockDirtyInfo *info,
174                                     struct DirtyRateConfig *config)
175 {
176     uint64_t sample_pages_per_gigabytes = config->sample_pages_per_gigabytes;
177 
178     /* Right shift 30 bits to calc ramblock size in GB */
179     info->sample_pages_count = (qemu_ram_get_used_length(block) *
180                                 sample_pages_per_gigabytes) >> 30;
181     /* Right shift TARGET_PAGE_BITS to calc page count */
182     info->ramblock_pages = qemu_ram_get_used_length(block) >>
183                            TARGET_PAGE_BITS;
184     info->ramblock_addr = qemu_ram_get_host_addr(block);
185     strcpy(info->idstr, qemu_ram_get_idstr(block));
186 }
187 
188 static void free_ramblock_dirty_info(struct RamblockDirtyInfo *infos, int count)
189 {
190     int i;
191 
192     if (!infos) {
193         return;
194     }
195 
196     for (i = 0; i < count; i++) {
197         g_free(infos[i].sample_page_vfn);
198         g_free(infos[i].hash_result);
199     }
200     g_free(infos);
201 }
202 
203 static bool skip_sample_ramblock(RAMBlock *block)
204 {
205     /*
206      * Sample only blocks larger than MIN_RAMBLOCK_SIZE.
207      */
208     if (qemu_ram_get_used_length(block) < (MIN_RAMBLOCK_SIZE << 10)) {
209         trace_skip_sample_ramblock(block->idstr,
210                                    qemu_ram_get_used_length(block));
211         return true;
212     }
213 
214     return false;
215 }
216 
217 static bool record_ramblock_hash_info(struct RamblockDirtyInfo **block_dinfo,
218                                       struct DirtyRateConfig config,
219                                       int *block_count)
220 {
221     struct RamblockDirtyInfo *info = NULL;
222     struct RamblockDirtyInfo *dinfo = NULL;
223     RAMBlock *block = NULL;
224     int total_count = 0;
225     int index = 0;
226     bool ret = false;
227 
228     RAMBLOCK_FOREACH_MIGRATABLE(block) {
229         if (skip_sample_ramblock(block)) {
230             continue;
231         }
232         total_count++;
233     }
234 
235     dinfo = g_try_malloc0_n(total_count, sizeof(struct RamblockDirtyInfo));
236     if (dinfo == NULL) {
237         goto out;
238     }
239 
240     RAMBLOCK_FOREACH_MIGRATABLE(block) {
241         if (skip_sample_ramblock(block)) {
242             continue;
243         }
244         if (index >= total_count) {
245             break;
246         }
247         info = &dinfo[index];
248         get_ramblock_dirty_info(block, info, &config);
249         if (!save_ramblock_hash(info)) {
250             goto out;
251         }
252         index++;
253     }
254     ret = true;
255 
256 out:
257     *block_count = index;
258     *block_dinfo = dinfo;
259     return ret;
260 }
261 
262 static void calc_page_dirty_rate(struct RamblockDirtyInfo *info)
263 {
264     uint32_t crc;
265     int i;
266 
267     for (i = 0; i < info->sample_pages_count; i++) {
268         crc = get_ramblock_vfn_hash(info, info->sample_page_vfn[i]);
269         if (crc != info->hash_result[i]) {
270             trace_calc_page_dirty_rate(info->idstr, crc, info->hash_result[i]);
271             info->sample_dirty_count++;
272         }
273     }
274 }
275 
276 static struct RamblockDirtyInfo *
277 find_block_matched(RAMBlock *block, int count,
278                   struct RamblockDirtyInfo *infos)
279 {
280     int i;
281     struct RamblockDirtyInfo *matched;
282 
283     for (i = 0; i < count; i++) {
284         if (!strcmp(infos[i].idstr, qemu_ram_get_idstr(block))) {
285             break;
286         }
287     }
288 
289     if (i == count) {
290         return NULL;
291     }
292 
293     if (infos[i].ramblock_addr != qemu_ram_get_host_addr(block) ||
294         infos[i].ramblock_pages !=
295             (qemu_ram_get_used_length(block) >> TARGET_PAGE_BITS)) {
296         trace_find_page_matched(block->idstr);
297         return NULL;
298     }
299 
300     matched = &infos[i];
301 
302     return matched;
303 }
304 
305 static bool compare_page_hash_info(struct RamblockDirtyInfo *info,
306                                   int block_count)
307 {
308     struct RamblockDirtyInfo *block_dinfo = NULL;
309     RAMBlock *block = NULL;
310 
311     RAMBLOCK_FOREACH_MIGRATABLE(block) {
312         if (skip_sample_ramblock(block)) {
313             continue;
314         }
315         block_dinfo = find_block_matched(block, block_count, info);
316         if (block_dinfo == NULL) {
317             continue;
318         }
319         calc_page_dirty_rate(block_dinfo);
320         update_dirtyrate_stat(block_dinfo);
321     }
322 
323     if (DirtyStat.total_sample_count == 0) {
324         return false;
325     }
326 
327     return true;
328 }
329 
330 static void calculate_dirtyrate(struct DirtyRateConfig config)
331 {
332     struct RamblockDirtyInfo *block_dinfo = NULL;
333     int block_count = 0;
334     int64_t msec = 0;
335     int64_t initial_time;
336 
337     rcu_register_thread();
338     rcu_read_lock();
339     initial_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
340     if (!record_ramblock_hash_info(&block_dinfo, config, &block_count)) {
341         goto out;
342     }
343     rcu_read_unlock();
344 
345     msec = config.sample_period_seconds * 1000;
346     msec = set_sample_page_period(msec, initial_time);
347     DirtyStat.start_time = initial_time / 1000;
348     DirtyStat.calc_time = msec / 1000;
349 
350     rcu_read_lock();
351     if (!compare_page_hash_info(block_dinfo, block_count)) {
352         goto out;
353     }
354 
355     update_dirtyrate(msec);
356 
357 out:
358     rcu_read_unlock();
359     free_ramblock_dirty_info(block_dinfo, block_count);
360     rcu_unregister_thread();
361 }
362 
363 void *get_dirtyrate_thread(void *arg)
364 {
365     struct DirtyRateConfig config = *(struct DirtyRateConfig *)arg;
366     int ret;
367     int64_t start_time;
368     int64_t calc_time;
369 
370     ret = dirtyrate_set_state(&CalculatingState, DIRTY_RATE_STATUS_UNSTARTED,
371                               DIRTY_RATE_STATUS_MEASURING);
372     if (ret == -1) {
373         error_report("change dirtyrate state failed.");
374         return NULL;
375     }
376 
377     start_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME) / 1000;
378     calc_time = config.sample_period_seconds;
379     init_dirtyrate_stat(start_time, calc_time);
380 
381     calculate_dirtyrate(config);
382 
383     ret = dirtyrate_set_state(&CalculatingState, DIRTY_RATE_STATUS_MEASURING,
384                               DIRTY_RATE_STATUS_MEASURED);
385     if (ret == -1) {
386         error_report("change dirtyrate state failed.");
387     }
388     return NULL;
389 }
390 
391 void qmp_calc_dirty_rate(int64_t calc_time, Error **errp)
392 {
393     static struct DirtyRateConfig config;
394     QemuThread thread;
395     int ret;
396 
397     /*
398      * If the dirty rate is already being measured, don't attempt to start.
399      */
400     if (qatomic_read(&CalculatingState) == DIRTY_RATE_STATUS_MEASURING) {
401         error_setg(errp, "the dirty rate is already being measured.");
402         return;
403     }
404 
405     if (!is_sample_period_valid(calc_time)) {
406         error_setg(errp, "calc-time is out of range[%d, %d].",
407                          MIN_FETCH_DIRTYRATE_TIME_SEC,
408                          MAX_FETCH_DIRTYRATE_TIME_SEC);
409         return;
410     }
411 
412     /*
413      * Init calculation state as unstarted.
414      */
415     ret = dirtyrate_set_state(&CalculatingState, CalculatingState,
416                               DIRTY_RATE_STATUS_UNSTARTED);
417     if (ret == -1) {
418         error_setg(errp, "init dirty rate calculation state failed.");
419         return;
420     }
421 
422     config.sample_period_seconds = calc_time;
423     config.sample_pages_per_gigabytes = DIRTYRATE_DEFAULT_SAMPLE_PAGES;
424     qemu_thread_create(&thread, "get_dirtyrate", get_dirtyrate_thread,
425                        (void *)&config, QEMU_THREAD_DETACHED);
426 }
427 
428 struct DirtyRateInfo *qmp_query_dirty_rate(Error **errp)
429 {
430     return query_dirty_rate_info();
431 }
432