xref: /qemu/migration/dirtyrate.c (revision 7afa08cd8fd2e5016658c39f6eb327e92edf0612)
14240dceeSChuan Zheng /*
24240dceeSChuan Zheng  * Dirtyrate implement code
34240dceeSChuan Zheng  *
44240dceeSChuan Zheng  * Copyright (c) 2020 HUAWEI TECHNOLOGIES CO.,LTD.
54240dceeSChuan Zheng  *
64240dceeSChuan Zheng  * Authors:
74240dceeSChuan Zheng  *  Chuan Zheng <zhengchuan@huawei.com>
84240dceeSChuan Zheng  *
94240dceeSChuan Zheng  * This work is licensed under the terms of the GNU GPL, version 2 or later.
104240dceeSChuan Zheng  * See the COPYING file in the top-level directory.
114240dceeSChuan Zheng  */
124240dceeSChuan Zheng 
134240dceeSChuan Zheng #include "qemu/osdep.h"
14662770afSMarc-André Lureau #include <zlib.h>
154240dceeSChuan Zheng #include "qapi/error.h"
164240dceeSChuan Zheng #include "cpu.h"
174240dceeSChuan Zheng #include "exec/ramblock.h"
184240dceeSChuan Zheng #include "qemu/rcu_queue.h"
194240dceeSChuan Zheng #include "qapi/qapi-commands-migration.h"
203ded54b1SChuan Zheng #include "ram.h"
213c0b5dffSChuan Zheng #include "trace.h"
224240dceeSChuan Zheng #include "dirtyrate.h"
234240dceeSChuan Zheng 
247df3aa30SChuan Zheng static int CalculatingState = DIRTY_RATE_STATUS_UNSTARTED;
25c9a58d71SChuan Zheng static struct DirtyRateStat DirtyStat;
267df3aa30SChuan Zheng 
27eca58224SChuan Zheng static int64_t set_sample_page_period(int64_t msec, int64_t initial_time)
28eca58224SChuan Zheng {
29eca58224SChuan Zheng     int64_t current_time;
30eca58224SChuan Zheng 
31eca58224SChuan Zheng     current_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
32eca58224SChuan Zheng     if ((current_time - initial_time) >= msec) {
33eca58224SChuan Zheng         msec = current_time - initial_time;
34eca58224SChuan Zheng     } else {
35eca58224SChuan Zheng         g_usleep((msec + initial_time - current_time) * 1000);
36eca58224SChuan Zheng     }
37eca58224SChuan Zheng 
38eca58224SChuan Zheng     return msec;
39eca58224SChuan Zheng }
40eca58224SChuan Zheng 
41eca58224SChuan Zheng static bool is_sample_period_valid(int64_t sec)
42eca58224SChuan Zheng {
43eca58224SChuan Zheng     if (sec < MIN_FETCH_DIRTYRATE_TIME_SEC ||
44eca58224SChuan Zheng         sec > MAX_FETCH_DIRTYRATE_TIME_SEC) {
45eca58224SChuan Zheng         return false;
46eca58224SChuan Zheng     }
47eca58224SChuan Zheng 
48eca58224SChuan Zheng     return true;
49eca58224SChuan Zheng }
50eca58224SChuan Zheng 
51*7afa08cdSHyman Huang(黄勇) static bool is_sample_pages_valid(int64_t pages)
52*7afa08cdSHyman Huang(黄勇) {
53*7afa08cdSHyman Huang(黄勇)     return pages >= MIN_SAMPLE_PAGE_COUNT &&
54*7afa08cdSHyman Huang(黄勇)            pages <= MAX_SAMPLE_PAGE_COUNT;
55*7afa08cdSHyman Huang(黄勇) }
56*7afa08cdSHyman Huang(黄勇) 
577df3aa30SChuan Zheng static int dirtyrate_set_state(int *state, int old_state, int new_state)
587df3aa30SChuan Zheng {
597df3aa30SChuan Zheng     assert(new_state < DIRTY_RATE_STATUS__MAX);
603c0b5dffSChuan Zheng     trace_dirtyrate_set_state(DirtyRateStatus_str(new_state));
617df3aa30SChuan Zheng     if (qatomic_cmpxchg(state, old_state, new_state) == old_state) {
627df3aa30SChuan Zheng         return 0;
637df3aa30SChuan Zheng     } else {
647df3aa30SChuan Zheng         return -1;
657df3aa30SChuan Zheng     }
667df3aa30SChuan Zheng }
677df3aa30SChuan Zheng 
684c437254SChuan Zheng static struct DirtyRateInfo *query_dirty_rate_info(void)
694c437254SChuan Zheng {
704c437254SChuan Zheng     int64_t dirty_rate = DirtyStat.dirty_rate;
714c437254SChuan Zheng     struct DirtyRateInfo *info = g_malloc0(sizeof(DirtyRateInfo));
724c437254SChuan Zheng 
734c437254SChuan Zheng     if (qatomic_read(&CalculatingState) == DIRTY_RATE_STATUS_MEASURED) {
74b1a859cfSChuan Zheng         info->has_dirty_rate = true;
754c437254SChuan Zheng         info->dirty_rate = dirty_rate;
764c437254SChuan Zheng     }
774c437254SChuan Zheng 
784c437254SChuan Zheng     info->status = CalculatingState;
794c437254SChuan Zheng     info->start_time = DirtyStat.start_time;
804c437254SChuan Zheng     info->calc_time = DirtyStat.calc_time;
81*7afa08cdSHyman Huang(黄勇)     info->sample_pages = DirtyStat.sample_pages;
824c437254SChuan Zheng 
833c0b5dffSChuan Zheng     trace_query_dirty_rate_info(DirtyRateStatus_str(CalculatingState));
843c0b5dffSChuan Zheng 
854c437254SChuan Zheng     return info;
864c437254SChuan Zheng }
874c437254SChuan Zheng 
88*7afa08cdSHyman Huang(黄勇) static void init_dirtyrate_stat(int64_t start_time, int64_t calc_time,
89*7afa08cdSHyman Huang(黄勇)                                 uint64_t sample_pages)
90c9a58d71SChuan Zheng {
91c9a58d71SChuan Zheng     DirtyStat.total_dirty_samples = 0;
92c9a58d71SChuan Zheng     DirtyStat.total_sample_count = 0;
93c9a58d71SChuan Zheng     DirtyStat.total_block_mem_MB = 0;
94c9a58d71SChuan Zheng     DirtyStat.dirty_rate = -1;
95aa84b506SChuan Zheng     DirtyStat.start_time = start_time;
96aa84b506SChuan Zheng     DirtyStat.calc_time = calc_time;
97*7afa08cdSHyman Huang(黄勇)     DirtyStat.sample_pages = sample_pages;
98c9a58d71SChuan Zheng }
99c9a58d71SChuan Zheng 
100c9a58d71SChuan Zheng static void update_dirtyrate_stat(struct RamblockDirtyInfo *info)
101c9a58d71SChuan Zheng {
102c9a58d71SChuan Zheng     DirtyStat.total_dirty_samples += info->sample_dirty_count;
103c9a58d71SChuan Zheng     DirtyStat.total_sample_count += info->sample_pages_count;
104c9a58d71SChuan Zheng     /* size of total pages in MB */
105c9a58d71SChuan Zheng     DirtyStat.total_block_mem_MB += (info->ramblock_pages *
106c9a58d71SChuan Zheng                                      TARGET_PAGE_SIZE) >> 20;
107c9a58d71SChuan Zheng }
108c9a58d71SChuan Zheng 
109c9a58d71SChuan Zheng static void update_dirtyrate(uint64_t msec)
110c9a58d71SChuan Zheng {
111c9a58d71SChuan Zheng     uint64_t dirtyrate;
112c9a58d71SChuan Zheng     uint64_t total_dirty_samples = DirtyStat.total_dirty_samples;
113c9a58d71SChuan Zheng     uint64_t total_sample_count = DirtyStat.total_sample_count;
114c9a58d71SChuan Zheng     uint64_t total_block_mem_MB = DirtyStat.total_block_mem_MB;
115c9a58d71SChuan Zheng 
116c9a58d71SChuan Zheng     dirtyrate = total_dirty_samples * total_block_mem_MB *
117c9a58d71SChuan Zheng                 1000 / (total_sample_count * msec);
118c9a58d71SChuan Zheng 
119c9a58d71SChuan Zheng     DirtyStat.dirty_rate = dirtyrate;
120c9a58d71SChuan Zheng }
1217df3aa30SChuan Zheng 
122ba0e519fSChuan Zheng /*
123ba0e519fSChuan Zheng  * get hash result for the sampled memory with length of TARGET_PAGE_SIZE
124ba0e519fSChuan Zheng  * in ramblock, which starts from ramblock base address.
125ba0e519fSChuan Zheng  */
126ba0e519fSChuan Zheng static uint32_t get_ramblock_vfn_hash(struct RamblockDirtyInfo *info,
127ba0e519fSChuan Zheng                                       uint64_t vfn)
128ba0e519fSChuan Zheng {
129ba0e519fSChuan Zheng     uint32_t crc;
130ba0e519fSChuan Zheng 
131ba0e519fSChuan Zheng     crc = crc32(0, (info->ramblock_addr +
132ba0e519fSChuan Zheng                 vfn * TARGET_PAGE_SIZE), TARGET_PAGE_SIZE);
133ba0e519fSChuan Zheng 
1343c0b5dffSChuan Zheng     trace_get_ramblock_vfn_hash(info->idstr, vfn, crc);
135ba0e519fSChuan Zheng     return crc;
136ba0e519fSChuan Zheng }
137ba0e519fSChuan Zheng 
138ba0e519fSChuan Zheng static bool save_ramblock_hash(struct RamblockDirtyInfo *info)
139ba0e519fSChuan Zheng {
140ba0e519fSChuan Zheng     unsigned int sample_pages_count;
141ba0e519fSChuan Zheng     int i;
142ba0e519fSChuan Zheng     GRand *rand;
143ba0e519fSChuan Zheng 
144ba0e519fSChuan Zheng     sample_pages_count = info->sample_pages_count;
145ba0e519fSChuan Zheng 
146ba0e519fSChuan Zheng     /* ramblock size less than one page, return success to skip this ramblock */
147ba0e519fSChuan Zheng     if (unlikely(info->ramblock_pages == 0 || sample_pages_count == 0)) {
148ba0e519fSChuan Zheng         return true;
149ba0e519fSChuan Zheng     }
150ba0e519fSChuan Zheng 
151ba0e519fSChuan Zheng     info->hash_result = g_try_malloc0_n(sample_pages_count,
152ba0e519fSChuan Zheng                                         sizeof(uint32_t));
153ba0e519fSChuan Zheng     if (!info->hash_result) {
154ba0e519fSChuan Zheng         return false;
155ba0e519fSChuan Zheng     }
156ba0e519fSChuan Zheng 
157ba0e519fSChuan Zheng     info->sample_page_vfn = g_try_malloc0_n(sample_pages_count,
158ba0e519fSChuan Zheng                                             sizeof(uint64_t));
159ba0e519fSChuan Zheng     if (!info->sample_page_vfn) {
160ba0e519fSChuan Zheng         g_free(info->hash_result);
161ba0e519fSChuan Zheng         return false;
162ba0e519fSChuan Zheng     }
163ba0e519fSChuan Zheng 
164ba0e519fSChuan Zheng     rand  = g_rand_new();
165ba0e519fSChuan Zheng     for (i = 0; i < sample_pages_count; i++) {
166ba0e519fSChuan Zheng         info->sample_page_vfn[i] = g_rand_int_range(rand, 0,
167ba0e519fSChuan Zheng                                                     info->ramblock_pages - 1);
168ba0e519fSChuan Zheng         info->hash_result[i] = get_ramblock_vfn_hash(info,
169ba0e519fSChuan Zheng                                                      info->sample_page_vfn[i]);
170ba0e519fSChuan Zheng     }
171ba0e519fSChuan Zheng     g_rand_free(rand);
172ba0e519fSChuan Zheng 
173ba0e519fSChuan Zheng     return true;
174ba0e519fSChuan Zheng }
175ba0e519fSChuan Zheng 
176ba0e519fSChuan Zheng static void get_ramblock_dirty_info(RAMBlock *block,
177ba0e519fSChuan Zheng                                     struct RamblockDirtyInfo *info,
178ba0e519fSChuan Zheng                                     struct DirtyRateConfig *config)
179ba0e519fSChuan Zheng {
180ba0e519fSChuan Zheng     uint64_t sample_pages_per_gigabytes = config->sample_pages_per_gigabytes;
181ba0e519fSChuan Zheng 
182ba0e519fSChuan Zheng     /* Right shift 30 bits to calc ramblock size in GB */
183ba0e519fSChuan Zheng     info->sample_pages_count = (qemu_ram_get_used_length(block) *
184ba0e519fSChuan Zheng                                 sample_pages_per_gigabytes) >> 30;
185ba0e519fSChuan Zheng     /* Right shift TARGET_PAGE_BITS to calc page count */
186ba0e519fSChuan Zheng     info->ramblock_pages = qemu_ram_get_used_length(block) >>
187ba0e519fSChuan Zheng                            TARGET_PAGE_BITS;
188ba0e519fSChuan Zheng     info->ramblock_addr = qemu_ram_get_host_addr(block);
189ba0e519fSChuan Zheng     strcpy(info->idstr, qemu_ram_get_idstr(block));
190ba0e519fSChuan Zheng }
191ba0e519fSChuan Zheng 
192cf0bbb49SChuan Zheng static void free_ramblock_dirty_info(struct RamblockDirtyInfo *infos, int count)
193cf0bbb49SChuan Zheng {
194cf0bbb49SChuan Zheng     int i;
195cf0bbb49SChuan Zheng 
196cf0bbb49SChuan Zheng     if (!infos) {
197cf0bbb49SChuan Zheng         return;
198cf0bbb49SChuan Zheng     }
199cf0bbb49SChuan Zheng 
200cf0bbb49SChuan Zheng     for (i = 0; i < count; i++) {
201cf0bbb49SChuan Zheng         g_free(infos[i].sample_page_vfn);
202cf0bbb49SChuan Zheng         g_free(infos[i].hash_result);
203cf0bbb49SChuan Zheng     }
204cf0bbb49SChuan Zheng     g_free(infos);
205cf0bbb49SChuan Zheng }
206cf0bbb49SChuan Zheng 
207f82583cdSChuan Zheng static bool skip_sample_ramblock(RAMBlock *block)
208f82583cdSChuan Zheng {
209f82583cdSChuan Zheng     /*
210f82583cdSChuan Zheng      * Sample only blocks larger than MIN_RAMBLOCK_SIZE.
211f82583cdSChuan Zheng      */
212f82583cdSChuan Zheng     if (qemu_ram_get_used_length(block) < (MIN_RAMBLOCK_SIZE << 10)) {
2133c0b5dffSChuan Zheng         trace_skip_sample_ramblock(block->idstr,
2143c0b5dffSChuan Zheng                                    qemu_ram_get_used_length(block));
215f82583cdSChuan Zheng         return true;
216f82583cdSChuan Zheng     }
217f82583cdSChuan Zheng 
218f82583cdSChuan Zheng     return false;
219f82583cdSChuan Zheng }
220f82583cdSChuan Zheng 
221ba0e519fSChuan Zheng static bool record_ramblock_hash_info(struct RamblockDirtyInfo **block_dinfo,
222ba0e519fSChuan Zheng                                       struct DirtyRateConfig config,
223ba0e519fSChuan Zheng                                       int *block_count)
224ba0e519fSChuan Zheng {
225ba0e519fSChuan Zheng     struct RamblockDirtyInfo *info = NULL;
226ba0e519fSChuan Zheng     struct RamblockDirtyInfo *dinfo = NULL;
227ba0e519fSChuan Zheng     RAMBlock *block = NULL;
228ba0e519fSChuan Zheng     int total_count = 0;
229ba0e519fSChuan Zheng     int index = 0;
230ba0e519fSChuan Zheng     bool ret = false;
231ba0e519fSChuan Zheng 
232ba0e519fSChuan Zheng     RAMBLOCK_FOREACH_MIGRATABLE(block) {
233f82583cdSChuan Zheng         if (skip_sample_ramblock(block)) {
234f82583cdSChuan Zheng             continue;
235f82583cdSChuan Zheng         }
236ba0e519fSChuan Zheng         total_count++;
237ba0e519fSChuan Zheng     }
238ba0e519fSChuan Zheng 
239ba0e519fSChuan Zheng     dinfo = g_try_malloc0_n(total_count, sizeof(struct RamblockDirtyInfo));
240ba0e519fSChuan Zheng     if (dinfo == NULL) {
241ba0e519fSChuan Zheng         goto out;
242ba0e519fSChuan Zheng     }
243ba0e519fSChuan Zheng 
244ba0e519fSChuan Zheng     RAMBLOCK_FOREACH_MIGRATABLE(block) {
245f82583cdSChuan Zheng         if (skip_sample_ramblock(block)) {
246f82583cdSChuan Zheng             continue;
247f82583cdSChuan Zheng         }
248ba0e519fSChuan Zheng         if (index >= total_count) {
249ba0e519fSChuan Zheng             break;
250ba0e519fSChuan Zheng         }
251ba0e519fSChuan Zheng         info = &dinfo[index];
252ba0e519fSChuan Zheng         get_ramblock_dirty_info(block, info, &config);
253ba0e519fSChuan Zheng         if (!save_ramblock_hash(info)) {
254ba0e519fSChuan Zheng             goto out;
255ba0e519fSChuan Zheng         }
256ba0e519fSChuan Zheng         index++;
257ba0e519fSChuan Zheng     }
258ba0e519fSChuan Zheng     ret = true;
259ba0e519fSChuan Zheng 
260ba0e519fSChuan Zheng out:
261ba0e519fSChuan Zheng     *block_count = index;
262ba0e519fSChuan Zheng     *block_dinfo = dinfo;
263ba0e519fSChuan Zheng     return ret;
264ba0e519fSChuan Zheng }
265ba0e519fSChuan Zheng 
2669c04387bSChuan Zheng static void calc_page_dirty_rate(struct RamblockDirtyInfo *info)
2679c04387bSChuan Zheng {
2689c04387bSChuan Zheng     uint32_t crc;
2699c04387bSChuan Zheng     int i;
2709c04387bSChuan Zheng 
2719c04387bSChuan Zheng     for (i = 0; i < info->sample_pages_count; i++) {
2729c04387bSChuan Zheng         crc = get_ramblock_vfn_hash(info, info->sample_page_vfn[i]);
2739c04387bSChuan Zheng         if (crc != info->hash_result[i]) {
2743c0b5dffSChuan Zheng             trace_calc_page_dirty_rate(info->idstr, crc, info->hash_result[i]);
2759c04387bSChuan Zheng             info->sample_dirty_count++;
2769c04387bSChuan Zheng         }
2779c04387bSChuan Zheng     }
2789c04387bSChuan Zheng }
2799c04387bSChuan Zheng 
2809c04387bSChuan Zheng static struct RamblockDirtyInfo *
2819c04387bSChuan Zheng find_block_matched(RAMBlock *block, int count,
2829c04387bSChuan Zheng                   struct RamblockDirtyInfo *infos)
2839c04387bSChuan Zheng {
2849c04387bSChuan Zheng     int i;
2859c04387bSChuan Zheng     struct RamblockDirtyInfo *matched;
2869c04387bSChuan Zheng 
2879c04387bSChuan Zheng     for (i = 0; i < count; i++) {
2889c04387bSChuan Zheng         if (!strcmp(infos[i].idstr, qemu_ram_get_idstr(block))) {
2899c04387bSChuan Zheng             break;
2909c04387bSChuan Zheng         }
2919c04387bSChuan Zheng     }
2929c04387bSChuan Zheng 
2939c04387bSChuan Zheng     if (i == count) {
2949c04387bSChuan Zheng         return NULL;
2959c04387bSChuan Zheng     }
2969c04387bSChuan Zheng 
2979c04387bSChuan Zheng     if (infos[i].ramblock_addr != qemu_ram_get_host_addr(block) ||
2989c04387bSChuan Zheng         infos[i].ramblock_pages !=
2999c04387bSChuan Zheng             (qemu_ram_get_used_length(block) >> TARGET_PAGE_BITS)) {
3003c0b5dffSChuan Zheng         trace_find_page_matched(block->idstr);
3019c04387bSChuan Zheng         return NULL;
3029c04387bSChuan Zheng     }
3039c04387bSChuan Zheng 
3049c04387bSChuan Zheng     matched = &infos[i];
3059c04387bSChuan Zheng 
3069c04387bSChuan Zheng     return matched;
3079c04387bSChuan Zheng }
3089c04387bSChuan Zheng 
3099c04387bSChuan Zheng static bool compare_page_hash_info(struct RamblockDirtyInfo *info,
3109c04387bSChuan Zheng                                   int block_count)
3119c04387bSChuan Zheng {
3129c04387bSChuan Zheng     struct RamblockDirtyInfo *block_dinfo = NULL;
3139c04387bSChuan Zheng     RAMBlock *block = NULL;
3149c04387bSChuan Zheng 
3159c04387bSChuan Zheng     RAMBLOCK_FOREACH_MIGRATABLE(block) {
316f82583cdSChuan Zheng         if (skip_sample_ramblock(block)) {
317f82583cdSChuan Zheng             continue;
318f82583cdSChuan Zheng         }
3199c04387bSChuan Zheng         block_dinfo = find_block_matched(block, block_count, info);
3209c04387bSChuan Zheng         if (block_dinfo == NULL) {
3219c04387bSChuan Zheng             continue;
3229c04387bSChuan Zheng         }
3239c04387bSChuan Zheng         calc_page_dirty_rate(block_dinfo);
3249c04387bSChuan Zheng         update_dirtyrate_stat(block_dinfo);
3259c04387bSChuan Zheng     }
3269c04387bSChuan Zheng 
3279c04387bSChuan Zheng     if (DirtyStat.total_sample_count == 0) {
3289c04387bSChuan Zheng         return false;
3299c04387bSChuan Zheng     }
3309c04387bSChuan Zheng 
3319c04387bSChuan Zheng     return true;
3329c04387bSChuan Zheng }
3339c04387bSChuan Zheng 
3344240dceeSChuan Zheng static void calculate_dirtyrate(struct DirtyRateConfig config)
3354240dceeSChuan Zheng {
336cf0bbb49SChuan Zheng     struct RamblockDirtyInfo *block_dinfo = NULL;
337cf0bbb49SChuan Zheng     int block_count = 0;
338cf0bbb49SChuan Zheng     int64_t msec = 0;
339cf0bbb49SChuan Zheng     int64_t initial_time;
340cf0bbb49SChuan Zheng 
341cf0bbb49SChuan Zheng     rcu_register_thread();
342cf0bbb49SChuan Zheng     rcu_read_lock();
343cf0bbb49SChuan Zheng     initial_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
344cf0bbb49SChuan Zheng     if (!record_ramblock_hash_info(&block_dinfo, config, &block_count)) {
345cf0bbb49SChuan Zheng         goto out;
346cf0bbb49SChuan Zheng     }
347cf0bbb49SChuan Zheng     rcu_read_unlock();
348cf0bbb49SChuan Zheng 
349cf0bbb49SChuan Zheng     msec = config.sample_period_seconds * 1000;
350cf0bbb49SChuan Zheng     msec = set_sample_page_period(msec, initial_time);
3514c437254SChuan Zheng     DirtyStat.start_time = initial_time / 1000;
3524c437254SChuan Zheng     DirtyStat.calc_time = msec / 1000;
353cf0bbb49SChuan Zheng 
354cf0bbb49SChuan Zheng     rcu_read_lock();
355cf0bbb49SChuan Zheng     if (!compare_page_hash_info(block_dinfo, block_count)) {
356cf0bbb49SChuan Zheng         goto out;
357cf0bbb49SChuan Zheng     }
358cf0bbb49SChuan Zheng 
359cf0bbb49SChuan Zheng     update_dirtyrate(msec);
360cf0bbb49SChuan Zheng 
361cf0bbb49SChuan Zheng out:
362cf0bbb49SChuan Zheng     rcu_read_unlock();
363cf0bbb49SChuan Zheng     free_ramblock_dirty_info(block_dinfo, block_count);
364cf0bbb49SChuan Zheng     rcu_unregister_thread();
3654240dceeSChuan Zheng }
3664240dceeSChuan Zheng 
3674240dceeSChuan Zheng void *get_dirtyrate_thread(void *arg)
3684240dceeSChuan Zheng {
3694240dceeSChuan Zheng     struct DirtyRateConfig config = *(struct DirtyRateConfig *)arg;
3707df3aa30SChuan Zheng     int ret;
371aa84b506SChuan Zheng     int64_t start_time;
372aa84b506SChuan Zheng     int64_t calc_time;
373*7afa08cdSHyman Huang(黄勇)     uint64_t sample_pages;
3747df3aa30SChuan Zheng 
3757df3aa30SChuan Zheng     ret = dirtyrate_set_state(&CalculatingState, DIRTY_RATE_STATUS_UNSTARTED,
3767df3aa30SChuan Zheng                               DIRTY_RATE_STATUS_MEASURING);
3777df3aa30SChuan Zheng     if (ret == -1) {
3787df3aa30SChuan Zheng         error_report("change dirtyrate state failed.");
3797df3aa30SChuan Zheng         return NULL;
3807df3aa30SChuan Zheng     }
3814240dceeSChuan Zheng 
382aa84b506SChuan Zheng     start_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME) / 1000;
383aa84b506SChuan Zheng     calc_time = config.sample_period_seconds;
384*7afa08cdSHyman Huang(黄勇)     sample_pages = config.sample_pages_per_gigabytes;
385*7afa08cdSHyman Huang(黄勇)     init_dirtyrate_stat(start_time, calc_time, sample_pages);
386aa84b506SChuan Zheng 
3874240dceeSChuan Zheng     calculate_dirtyrate(config);
3884240dceeSChuan Zheng 
3897df3aa30SChuan Zheng     ret = dirtyrate_set_state(&CalculatingState, DIRTY_RATE_STATUS_MEASURING,
3907df3aa30SChuan Zheng                               DIRTY_RATE_STATUS_MEASURED);
3917df3aa30SChuan Zheng     if (ret == -1) {
3927df3aa30SChuan Zheng         error_report("change dirtyrate state failed.");
3937df3aa30SChuan Zheng     }
3944240dceeSChuan Zheng     return NULL;
3954240dceeSChuan Zheng }
3964c437254SChuan Zheng 
397*7afa08cdSHyman Huang(黄勇) void qmp_calc_dirty_rate(int64_t calc_time, bool has_sample_pages,
398*7afa08cdSHyman Huang(黄勇)                          int64_t sample_pages, Error **errp)
3994c437254SChuan Zheng {
4004c437254SChuan Zheng     static struct DirtyRateConfig config;
4014c437254SChuan Zheng     QemuThread thread;
4024c437254SChuan Zheng     int ret;
4034c437254SChuan Zheng 
4044c437254SChuan Zheng     /*
4054c437254SChuan Zheng      * If the dirty rate is already being measured, don't attempt to start.
4064c437254SChuan Zheng      */
4074c437254SChuan Zheng     if (qatomic_read(&CalculatingState) == DIRTY_RATE_STATUS_MEASURING) {
4084c437254SChuan Zheng         error_setg(errp, "the dirty rate is already being measured.");
4094c437254SChuan Zheng         return;
4104c437254SChuan Zheng     }
4114c437254SChuan Zheng 
4124c437254SChuan Zheng     if (!is_sample_period_valid(calc_time)) {
4134c437254SChuan Zheng         error_setg(errp, "calc-time is out of range[%d, %d].",
4144c437254SChuan Zheng                          MIN_FETCH_DIRTYRATE_TIME_SEC,
4154c437254SChuan Zheng                          MAX_FETCH_DIRTYRATE_TIME_SEC);
4164c437254SChuan Zheng         return;
4174c437254SChuan Zheng     }
4184c437254SChuan Zheng 
419*7afa08cdSHyman Huang(黄勇)     if (has_sample_pages) {
420*7afa08cdSHyman Huang(黄勇)         if (!is_sample_pages_valid(sample_pages)) {
421*7afa08cdSHyman Huang(黄勇)             error_setg(errp, "sample-pages is out of range[%d, %d].",
422*7afa08cdSHyman Huang(黄勇)                             MIN_SAMPLE_PAGE_COUNT,
423*7afa08cdSHyman Huang(黄勇)                             MAX_SAMPLE_PAGE_COUNT);
424*7afa08cdSHyman Huang(黄勇)             return;
425*7afa08cdSHyman Huang(黄勇)         }
426*7afa08cdSHyman Huang(黄勇)     } else {
427*7afa08cdSHyman Huang(黄勇)         sample_pages = DIRTYRATE_DEFAULT_SAMPLE_PAGES;
428*7afa08cdSHyman Huang(黄勇)     }
429*7afa08cdSHyman Huang(黄勇) 
4304c437254SChuan Zheng     /*
4314c437254SChuan Zheng      * Init calculation state as unstarted.
4324c437254SChuan Zheng      */
4334c437254SChuan Zheng     ret = dirtyrate_set_state(&CalculatingState, CalculatingState,
4344c437254SChuan Zheng                               DIRTY_RATE_STATUS_UNSTARTED);
4354c437254SChuan Zheng     if (ret == -1) {
4364c437254SChuan Zheng         error_setg(errp, "init dirty rate calculation state failed.");
4374c437254SChuan Zheng         return;
4384c437254SChuan Zheng     }
4394c437254SChuan Zheng 
4404c437254SChuan Zheng     config.sample_period_seconds = calc_time;
441*7afa08cdSHyman Huang(黄勇)     config.sample_pages_per_gigabytes = sample_pages;
4424c437254SChuan Zheng     qemu_thread_create(&thread, "get_dirtyrate", get_dirtyrate_thread,
4434c437254SChuan Zheng                        (void *)&config, QEMU_THREAD_DETACHED);
4444c437254SChuan Zheng }
4454c437254SChuan Zheng 
4464c437254SChuan Zheng struct DirtyRateInfo *qmp_query_dirty_rate(Error **errp)
4474c437254SChuan Zheng {
4484c437254SChuan Zheng     return query_dirty_rate_info();
4494c437254SChuan Zheng }
450