xref: /qemu/migration/postcopy-ram.c (revision 8679638b0e231f97ad3456c681bf569ca7f7f6d5)
1eb59db53SDr. David Alan Gilbert /*
2eb59db53SDr. David Alan Gilbert  * Postcopy migration for RAM
3eb59db53SDr. David Alan Gilbert  *
4eb59db53SDr. David Alan Gilbert  * Copyright 2013-2015 Red Hat, Inc. and/or its affiliates
5eb59db53SDr. David Alan Gilbert  *
6eb59db53SDr. David Alan Gilbert  * Authors:
7eb59db53SDr. David Alan Gilbert  *  Dave Gilbert  <dgilbert@redhat.com>
8eb59db53SDr. David Alan Gilbert  *
9eb59db53SDr. David Alan Gilbert  * This work is licensed under the terms of the GNU GPL, version 2 or later.
10eb59db53SDr. David Alan Gilbert  * See the COPYING file in the top-level directory.
11eb59db53SDr. David Alan Gilbert  *
12eb59db53SDr. David Alan Gilbert  */
13eb59db53SDr. David Alan Gilbert 
14eb59db53SDr. David Alan Gilbert /*
15eb59db53SDr. David Alan Gilbert  * Postcopy is a migration technique where the execution flips from the
16eb59db53SDr. David Alan Gilbert  * source to the destination before all the data has been copied.
17eb59db53SDr. David Alan Gilbert  */
18eb59db53SDr. David Alan Gilbert 
191393a485SPeter Maydell #include "qemu/osdep.h"
20eb59db53SDr. David Alan Gilbert 
21eb59db53SDr. David Alan Gilbert #include "qemu-common.h"
22eb59db53SDr. David Alan Gilbert #include "migration/migration.h"
23eb59db53SDr. David Alan Gilbert #include "migration/postcopy-ram.h"
24eb59db53SDr. David Alan Gilbert #include "sysemu/sysemu.h"
25371ff5a3SDr. David Alan Gilbert #include "sysemu/balloon.h"
26eb59db53SDr. David Alan Gilbert #include "qemu/error-report.h"
27eb59db53SDr. David Alan Gilbert #include "trace.h"
28eb59db53SDr. David Alan Gilbert 
29e0b266f0SDr. David Alan Gilbert /* Arbitrary limit on size of each discard command,
30e0b266f0SDr. David Alan Gilbert  * keeps them around ~200 bytes
31e0b266f0SDr. David Alan Gilbert  */
32e0b266f0SDr. David Alan Gilbert #define MAX_DISCARDS_PER_COMMAND 12
33e0b266f0SDr. David Alan Gilbert 
34e0b266f0SDr. David Alan Gilbert struct PostcopyDiscardState {
35e0b266f0SDr. David Alan Gilbert     const char *ramblock_name;
36e0b266f0SDr. David Alan Gilbert     uint64_t offset; /* Bitmap entry for the 1st bit of this RAMBlock */
37e0b266f0SDr. David Alan Gilbert     uint16_t cur_entry;
38e0b266f0SDr. David Alan Gilbert     /*
39e0b266f0SDr. David Alan Gilbert      * Start and length of a discard range (bytes)
40e0b266f0SDr. David Alan Gilbert      */
41e0b266f0SDr. David Alan Gilbert     uint64_t start_list[MAX_DISCARDS_PER_COMMAND];
42e0b266f0SDr. David Alan Gilbert     uint64_t length_list[MAX_DISCARDS_PER_COMMAND];
43e0b266f0SDr. David Alan Gilbert     unsigned int nsentwords;
44e0b266f0SDr. David Alan Gilbert     unsigned int nsentcmds;
45e0b266f0SDr. David Alan Gilbert };
46e0b266f0SDr. David Alan Gilbert 
47eb59db53SDr. David Alan Gilbert /* Postcopy needs to detect accesses to pages that haven't yet been copied
48eb59db53SDr. David Alan Gilbert  * across, and efficiently map new pages in, the techniques for doing this
49eb59db53SDr. David Alan Gilbert  * are target OS specific.
50eb59db53SDr. David Alan Gilbert  */
51eb59db53SDr. David Alan Gilbert #if defined(__linux__)
52eb59db53SDr. David Alan Gilbert 
53c4faeed2SDr. David Alan Gilbert #include <poll.h>
54eb59db53SDr. David Alan Gilbert #include <sys/ioctl.h>
55eb59db53SDr. David Alan Gilbert #include <sys/syscall.h>
56eb59db53SDr. David Alan Gilbert #include <asm/types.h> /* for __u64 */
57eb59db53SDr. David Alan Gilbert #endif
58eb59db53SDr. David Alan Gilbert 
59d8b9d771SMatthew Fortune #if defined(__linux__) && defined(__NR_userfaultfd) && defined(CONFIG_EVENTFD)
60d8b9d771SMatthew Fortune #include <sys/eventfd.h>
61eb59db53SDr. David Alan Gilbert #include <linux/userfaultfd.h>
62eb59db53SDr. David Alan Gilbert 
63eb59db53SDr. David Alan Gilbert static bool ufd_version_check(int ufd)
64eb59db53SDr. David Alan Gilbert {
65eb59db53SDr. David Alan Gilbert     struct uffdio_api api_struct;
66eb59db53SDr. David Alan Gilbert     uint64_t ioctl_mask;
67eb59db53SDr. David Alan Gilbert 
68eb59db53SDr. David Alan Gilbert     api_struct.api = UFFD_API;
69eb59db53SDr. David Alan Gilbert     api_struct.features = 0;
70eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_API, &api_struct)) {
71eb59db53SDr. David Alan Gilbert         error_report("postcopy_ram_supported_by_host: UFFDIO_API failed: %s",
72eb59db53SDr. David Alan Gilbert                      strerror(errno));
73eb59db53SDr. David Alan Gilbert         return false;
74eb59db53SDr. David Alan Gilbert     }
75eb59db53SDr. David Alan Gilbert 
76eb59db53SDr. David Alan Gilbert     ioctl_mask = (__u64)1 << _UFFDIO_REGISTER |
77eb59db53SDr. David Alan Gilbert                  (__u64)1 << _UFFDIO_UNREGISTER;
78eb59db53SDr. David Alan Gilbert     if ((api_struct.ioctls & ioctl_mask) != ioctl_mask) {
79eb59db53SDr. David Alan Gilbert         error_report("Missing userfault features: %" PRIx64,
80eb59db53SDr. David Alan Gilbert                      (uint64_t)(~api_struct.ioctls & ioctl_mask));
81eb59db53SDr. David Alan Gilbert         return false;
82eb59db53SDr. David Alan Gilbert     }
83eb59db53SDr. David Alan Gilbert 
847e8cafb7SDr. David Alan Gilbert     if (getpagesize() != ram_pagesize_summary()) {
857e8cafb7SDr. David Alan Gilbert         bool have_hp = false;
867e8cafb7SDr. David Alan Gilbert         /* We've got a huge page */
877e8cafb7SDr. David Alan Gilbert #ifdef UFFD_FEATURE_MISSING_HUGETLBFS
887e8cafb7SDr. David Alan Gilbert         have_hp = api_struct.features & UFFD_FEATURE_MISSING_HUGETLBFS;
897e8cafb7SDr. David Alan Gilbert #endif
907e8cafb7SDr. David Alan Gilbert         if (!have_hp) {
917e8cafb7SDr. David Alan Gilbert             error_report("Userfault on this host does not support huge pages");
927e8cafb7SDr. David Alan Gilbert             return false;
937e8cafb7SDr. David Alan Gilbert         }
947e8cafb7SDr. David Alan Gilbert     }
95eb59db53SDr. David Alan Gilbert     return true;
96eb59db53SDr. David Alan Gilbert }
97eb59db53SDr. David Alan Gilbert 
98*8679638bSDr. David Alan Gilbert /* Callback from postcopy_ram_supported_by_host block iterator.
99*8679638bSDr. David Alan Gilbert  */
100*8679638bSDr. David Alan Gilbert static int test_range_shared(const char *block_name, void *host_addr,
101*8679638bSDr. David Alan Gilbert                              ram_addr_t offset, ram_addr_t length, void *opaque)
102*8679638bSDr. David Alan Gilbert {
103*8679638bSDr. David Alan Gilbert     if (qemu_ram_is_shared(qemu_ram_block_by_name(block_name))) {
104*8679638bSDr. David Alan Gilbert         error_report("Postcopy on shared RAM (%s) is not yet supported",
105*8679638bSDr. David Alan Gilbert                      block_name);
106*8679638bSDr. David Alan Gilbert         return 1;
107*8679638bSDr. David Alan Gilbert     }
108*8679638bSDr. David Alan Gilbert     return 0;
109*8679638bSDr. David Alan Gilbert }
110*8679638bSDr. David Alan Gilbert 
11158b7c17eSDr. David Alan Gilbert /*
11258b7c17eSDr. David Alan Gilbert  * Note: This has the side effect of munlock'ing all of RAM, that's
11358b7c17eSDr. David Alan Gilbert  * normally fine since if the postcopy succeeds it gets turned back on at the
11458b7c17eSDr. David Alan Gilbert  * end.
11558b7c17eSDr. David Alan Gilbert  */
116eb59db53SDr. David Alan Gilbert bool postcopy_ram_supported_by_host(void)
117eb59db53SDr. David Alan Gilbert {
118eb59db53SDr. David Alan Gilbert     long pagesize = getpagesize();
119eb59db53SDr. David Alan Gilbert     int ufd = -1;
120eb59db53SDr. David Alan Gilbert     bool ret = false; /* Error unless we change it */
121eb59db53SDr. David Alan Gilbert     void *testarea = NULL;
122eb59db53SDr. David Alan Gilbert     struct uffdio_register reg_struct;
123eb59db53SDr. David Alan Gilbert     struct uffdio_range range_struct;
124eb59db53SDr. David Alan Gilbert     uint64_t feature_mask;
125eb59db53SDr. David Alan Gilbert 
126eb59db53SDr. David Alan Gilbert     if ((1ul << qemu_target_page_bits()) > pagesize) {
127eb59db53SDr. David Alan Gilbert         error_report("Target page size bigger than host page size");
128eb59db53SDr. David Alan Gilbert         goto out;
129eb59db53SDr. David Alan Gilbert     }
130eb59db53SDr. David Alan Gilbert 
131eb59db53SDr. David Alan Gilbert     ufd = syscall(__NR_userfaultfd, O_CLOEXEC);
132eb59db53SDr. David Alan Gilbert     if (ufd == -1) {
133eb59db53SDr. David Alan Gilbert         error_report("%s: userfaultfd not available: %s", __func__,
134eb59db53SDr. David Alan Gilbert                      strerror(errno));
135eb59db53SDr. David Alan Gilbert         goto out;
136eb59db53SDr. David Alan Gilbert     }
137eb59db53SDr. David Alan Gilbert 
138eb59db53SDr. David Alan Gilbert     /* Version and features check */
139eb59db53SDr. David Alan Gilbert     if (!ufd_version_check(ufd)) {
140eb59db53SDr. David Alan Gilbert         goto out;
141eb59db53SDr. David Alan Gilbert     }
142eb59db53SDr. David Alan Gilbert 
143*8679638bSDr. David Alan Gilbert     /* We don't support postcopy with shared RAM yet */
144*8679638bSDr. David Alan Gilbert     if (qemu_ram_foreach_block(test_range_shared, NULL)) {
145*8679638bSDr. David Alan Gilbert         goto out;
146*8679638bSDr. David Alan Gilbert     }
147*8679638bSDr. David Alan Gilbert 
148eb59db53SDr. David Alan Gilbert     /*
14958b7c17eSDr. David Alan Gilbert      * userfault and mlock don't go together; we'll put it back later if
15058b7c17eSDr. David Alan Gilbert      * it was enabled.
15158b7c17eSDr. David Alan Gilbert      */
15258b7c17eSDr. David Alan Gilbert     if (munlockall()) {
15358b7c17eSDr. David Alan Gilbert         error_report("%s: munlockall: %s", __func__,  strerror(errno));
15458b7c17eSDr. David Alan Gilbert         return -1;
15558b7c17eSDr. David Alan Gilbert     }
15658b7c17eSDr. David Alan Gilbert 
15758b7c17eSDr. David Alan Gilbert     /*
158eb59db53SDr. David Alan Gilbert      *  We need to check that the ops we need are supported on anon memory
159eb59db53SDr. David Alan Gilbert      *  To do that we need to register a chunk and see the flags that
160eb59db53SDr. David Alan Gilbert      *  are returned.
161eb59db53SDr. David Alan Gilbert      */
162eb59db53SDr. David Alan Gilbert     testarea = mmap(NULL, pagesize, PROT_READ | PROT_WRITE, MAP_PRIVATE |
163eb59db53SDr. David Alan Gilbert                                     MAP_ANONYMOUS, -1, 0);
164eb59db53SDr. David Alan Gilbert     if (testarea == MAP_FAILED) {
165eb59db53SDr. David Alan Gilbert         error_report("%s: Failed to map test area: %s", __func__,
166eb59db53SDr. David Alan Gilbert                      strerror(errno));
167eb59db53SDr. David Alan Gilbert         goto out;
168eb59db53SDr. David Alan Gilbert     }
169eb59db53SDr. David Alan Gilbert     g_assert(((size_t)testarea & (pagesize-1)) == 0);
170eb59db53SDr. David Alan Gilbert 
171eb59db53SDr. David Alan Gilbert     reg_struct.range.start = (uintptr_t)testarea;
172eb59db53SDr. David Alan Gilbert     reg_struct.range.len = pagesize;
173eb59db53SDr. David Alan Gilbert     reg_struct.mode = UFFDIO_REGISTER_MODE_MISSING;
174eb59db53SDr. David Alan Gilbert 
175eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_REGISTER, &reg_struct)) {
176eb59db53SDr. David Alan Gilbert         error_report("%s userfault register: %s", __func__, strerror(errno));
177eb59db53SDr. David Alan Gilbert         goto out;
178eb59db53SDr. David Alan Gilbert     }
179eb59db53SDr. David Alan Gilbert 
180eb59db53SDr. David Alan Gilbert     range_struct.start = (uintptr_t)testarea;
181eb59db53SDr. David Alan Gilbert     range_struct.len = pagesize;
182eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_UNREGISTER, &range_struct)) {
183eb59db53SDr. David Alan Gilbert         error_report("%s userfault unregister: %s", __func__, strerror(errno));
184eb59db53SDr. David Alan Gilbert         goto out;
185eb59db53SDr. David Alan Gilbert     }
186eb59db53SDr. David Alan Gilbert 
187eb59db53SDr. David Alan Gilbert     feature_mask = (__u64)1 << _UFFDIO_WAKE |
188eb59db53SDr. David Alan Gilbert                    (__u64)1 << _UFFDIO_COPY |
189eb59db53SDr. David Alan Gilbert                    (__u64)1 << _UFFDIO_ZEROPAGE;
190eb59db53SDr. David Alan Gilbert     if ((reg_struct.ioctls & feature_mask) != feature_mask) {
191eb59db53SDr. David Alan Gilbert         error_report("Missing userfault map features: %" PRIx64,
192eb59db53SDr. David Alan Gilbert                      (uint64_t)(~reg_struct.ioctls & feature_mask));
193eb59db53SDr. David Alan Gilbert         goto out;
194eb59db53SDr. David Alan Gilbert     }
195eb59db53SDr. David Alan Gilbert 
196eb59db53SDr. David Alan Gilbert     /* Success! */
197eb59db53SDr. David Alan Gilbert     ret = true;
198eb59db53SDr. David Alan Gilbert out:
199eb59db53SDr. David Alan Gilbert     if (testarea) {
200eb59db53SDr. David Alan Gilbert         munmap(testarea, pagesize);
201eb59db53SDr. David Alan Gilbert     }
202eb59db53SDr. David Alan Gilbert     if (ufd != -1) {
203eb59db53SDr. David Alan Gilbert         close(ufd);
204eb59db53SDr. David Alan Gilbert     }
205eb59db53SDr. David Alan Gilbert     return ret;
206eb59db53SDr. David Alan Gilbert }
207eb59db53SDr. David Alan Gilbert 
2081caddf8aSDr. David Alan Gilbert /*
2091caddf8aSDr. David Alan Gilbert  * Setup an area of RAM so that it *can* be used for postcopy later; this
2101caddf8aSDr. David Alan Gilbert  * must be done right at the start prior to pre-copy.
2111caddf8aSDr. David Alan Gilbert  * opaque should be the MIS.
2121caddf8aSDr. David Alan Gilbert  */
2131caddf8aSDr. David Alan Gilbert static int init_range(const char *block_name, void *host_addr,
2141caddf8aSDr. David Alan Gilbert                       ram_addr_t offset, ram_addr_t length, void *opaque)
2151caddf8aSDr. David Alan Gilbert {
2161caddf8aSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
2171caddf8aSDr. David Alan Gilbert 
2181caddf8aSDr. David Alan Gilbert     trace_postcopy_init_range(block_name, host_addr, offset, length);
2191caddf8aSDr. David Alan Gilbert 
2201caddf8aSDr. David Alan Gilbert     /*
2211caddf8aSDr. David Alan Gilbert      * We need the whole of RAM to be truly empty for postcopy, so things
2221caddf8aSDr. David Alan Gilbert      * like ROMs and any data tables built during init must be zero'd
2231caddf8aSDr. David Alan Gilbert      * - we're going to get the copy from the source anyway.
2241caddf8aSDr. David Alan Gilbert      * (Precopy will just overwrite this data, so doesn't need the discard)
2251caddf8aSDr. David Alan Gilbert      */
226d3a5038cSDr. David Alan Gilbert     if (ram_discard_range(mis, block_name, 0, length)) {
2271caddf8aSDr. David Alan Gilbert         return -1;
2281caddf8aSDr. David Alan Gilbert     }
2291caddf8aSDr. David Alan Gilbert 
2301caddf8aSDr. David Alan Gilbert     return 0;
2311caddf8aSDr. David Alan Gilbert }
2321caddf8aSDr. David Alan Gilbert 
2331caddf8aSDr. David Alan Gilbert /*
2341caddf8aSDr. David Alan Gilbert  * At the end of migration, undo the effects of init_range
2351caddf8aSDr. David Alan Gilbert  * opaque should be the MIS.
2361caddf8aSDr. David Alan Gilbert  */
2371caddf8aSDr. David Alan Gilbert static int cleanup_range(const char *block_name, void *host_addr,
2381caddf8aSDr. David Alan Gilbert                         ram_addr_t offset, ram_addr_t length, void *opaque)
2391caddf8aSDr. David Alan Gilbert {
2401caddf8aSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
2411caddf8aSDr. David Alan Gilbert     struct uffdio_range range_struct;
2421caddf8aSDr. David Alan Gilbert     trace_postcopy_cleanup_range(block_name, host_addr, offset, length);
2431caddf8aSDr. David Alan Gilbert 
2441caddf8aSDr. David Alan Gilbert     /*
2451caddf8aSDr. David Alan Gilbert      * We turned off hugepage for the precopy stage with postcopy enabled
2461caddf8aSDr. David Alan Gilbert      * we can turn it back on now.
2471caddf8aSDr. David Alan Gilbert      */
2481d741439SDr. David Alan Gilbert     qemu_madvise(host_addr, length, QEMU_MADV_HUGEPAGE);
2491caddf8aSDr. David Alan Gilbert 
2501caddf8aSDr. David Alan Gilbert     /*
2511caddf8aSDr. David Alan Gilbert      * We can also turn off userfault now since we should have all the
2521caddf8aSDr. David Alan Gilbert      * pages.   It can be useful to leave it on to debug postcopy
2531caddf8aSDr. David Alan Gilbert      * if you're not sure it's always getting every page.
2541caddf8aSDr. David Alan Gilbert      */
2551caddf8aSDr. David Alan Gilbert     range_struct.start = (uintptr_t)host_addr;
2561caddf8aSDr. David Alan Gilbert     range_struct.len = length;
2571caddf8aSDr. David Alan Gilbert 
2581caddf8aSDr. David Alan Gilbert     if (ioctl(mis->userfault_fd, UFFDIO_UNREGISTER, &range_struct)) {
2591caddf8aSDr. David Alan Gilbert         error_report("%s: userfault unregister %s", __func__, strerror(errno));
2601caddf8aSDr. David Alan Gilbert 
2611caddf8aSDr. David Alan Gilbert         return -1;
2621caddf8aSDr. David Alan Gilbert     }
2631caddf8aSDr. David Alan Gilbert 
2641caddf8aSDr. David Alan Gilbert     return 0;
2651caddf8aSDr. David Alan Gilbert }
2661caddf8aSDr. David Alan Gilbert 
2671caddf8aSDr. David Alan Gilbert /*
2681caddf8aSDr. David Alan Gilbert  * Initialise postcopy-ram, setting the RAM to a state where we can go into
2691caddf8aSDr. David Alan Gilbert  * postcopy later; must be called prior to any precopy.
2701caddf8aSDr. David Alan Gilbert  * called from arch_init's similarly named ram_postcopy_incoming_init
2711caddf8aSDr. David Alan Gilbert  */
2721caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_init(MigrationIncomingState *mis, size_t ram_pages)
2731caddf8aSDr. David Alan Gilbert {
2741caddf8aSDr. David Alan Gilbert     if (qemu_ram_foreach_block(init_range, mis)) {
2751caddf8aSDr. David Alan Gilbert         return -1;
2761caddf8aSDr. David Alan Gilbert     }
2771caddf8aSDr. David Alan Gilbert 
2781caddf8aSDr. David Alan Gilbert     return 0;
2791caddf8aSDr. David Alan Gilbert }
2801caddf8aSDr. David Alan Gilbert 
2811caddf8aSDr. David Alan Gilbert /*
2821caddf8aSDr. David Alan Gilbert  * At the end of a migration where postcopy_ram_incoming_init was called.
2831caddf8aSDr. David Alan Gilbert  */
2841caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_cleanup(MigrationIncomingState *mis)
2851caddf8aSDr. David Alan Gilbert {
286c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_incoming_cleanup_entry();
287c4faeed2SDr. David Alan Gilbert 
288c4faeed2SDr. David Alan Gilbert     if (mis->have_fault_thread) {
289c4faeed2SDr. David Alan Gilbert         uint64_t tmp64;
290c4faeed2SDr. David Alan Gilbert 
2911caddf8aSDr. David Alan Gilbert         if (qemu_ram_foreach_block(cleanup_range, mis)) {
2921caddf8aSDr. David Alan Gilbert             return -1;
2931caddf8aSDr. David Alan Gilbert         }
294c4faeed2SDr. David Alan Gilbert         /*
295c4faeed2SDr. David Alan Gilbert          * Tell the fault_thread to exit, it's an eventfd that should
296c4faeed2SDr. David Alan Gilbert          * currently be at 0, we're going to increment it to 1
297c4faeed2SDr. David Alan Gilbert          */
298c4faeed2SDr. David Alan Gilbert         tmp64 = 1;
299c4faeed2SDr. David Alan Gilbert         if (write(mis->userfault_quit_fd, &tmp64, 8) == 8) {
300c4faeed2SDr. David Alan Gilbert             trace_postcopy_ram_incoming_cleanup_join();
301c4faeed2SDr. David Alan Gilbert             qemu_thread_join(&mis->fault_thread);
302c4faeed2SDr. David Alan Gilbert         } else {
303c4faeed2SDr. David Alan Gilbert             /* Not much we can do here, but may as well report it */
304c4faeed2SDr. David Alan Gilbert             error_report("%s: incrementing userfault_quit_fd: %s", __func__,
305c4faeed2SDr. David Alan Gilbert                          strerror(errno));
306c4faeed2SDr. David Alan Gilbert         }
307c4faeed2SDr. David Alan Gilbert         trace_postcopy_ram_incoming_cleanup_closeuf();
308c4faeed2SDr. David Alan Gilbert         close(mis->userfault_fd);
309c4faeed2SDr. David Alan Gilbert         close(mis->userfault_quit_fd);
310c4faeed2SDr. David Alan Gilbert         mis->have_fault_thread = false;
311c4faeed2SDr. David Alan Gilbert     }
312c4faeed2SDr. David Alan Gilbert 
313371ff5a3SDr. David Alan Gilbert     qemu_balloon_inhibit(false);
314371ff5a3SDr. David Alan Gilbert 
31558b7c17eSDr. David Alan Gilbert     if (enable_mlock) {
31658b7c17eSDr. David Alan Gilbert         if (os_mlock() < 0) {
31758b7c17eSDr. David Alan Gilbert             error_report("mlock: %s", strerror(errno));
31858b7c17eSDr. David Alan Gilbert             /*
31958b7c17eSDr. David Alan Gilbert              * It doesn't feel right to fail at this point, we have a valid
32058b7c17eSDr. David Alan Gilbert              * VM state.
32158b7c17eSDr. David Alan Gilbert              */
32258b7c17eSDr. David Alan Gilbert         }
32358b7c17eSDr. David Alan Gilbert     }
32458b7c17eSDr. David Alan Gilbert 
325c4faeed2SDr. David Alan Gilbert     postcopy_state_set(POSTCOPY_INCOMING_END);
326c4faeed2SDr. David Alan Gilbert     migrate_send_rp_shut(mis, qemu_file_get_error(mis->from_src_file) != 0);
3271caddf8aSDr. David Alan Gilbert 
328696ed9a9SDr. David Alan Gilbert     if (mis->postcopy_tmp_page) {
329df9ff5e1SDr. David Alan Gilbert         munmap(mis->postcopy_tmp_page, mis->largest_page_size);
330696ed9a9SDr. David Alan Gilbert         mis->postcopy_tmp_page = NULL;
331696ed9a9SDr. David Alan Gilbert     }
33241d84210SDr. David Alan Gilbert     if (mis->postcopy_tmp_zero_page) {
33341d84210SDr. David Alan Gilbert         munmap(mis->postcopy_tmp_zero_page, mis->largest_page_size);
33441d84210SDr. David Alan Gilbert         mis->postcopy_tmp_zero_page = NULL;
33541d84210SDr. David Alan Gilbert     }
336c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_incoming_cleanup_exit();
3371caddf8aSDr. David Alan Gilbert     return 0;
3381caddf8aSDr. David Alan Gilbert }
3391caddf8aSDr. David Alan Gilbert 
340f0a227adSDr. David Alan Gilbert /*
341f9527107SDr. David Alan Gilbert  * Disable huge pages on an area
342f9527107SDr. David Alan Gilbert  */
343f9527107SDr. David Alan Gilbert static int nhp_range(const char *block_name, void *host_addr,
344f9527107SDr. David Alan Gilbert                     ram_addr_t offset, ram_addr_t length, void *opaque)
345f9527107SDr. David Alan Gilbert {
346f9527107SDr. David Alan Gilbert     trace_postcopy_nhp_range(block_name, host_addr, offset, length);
347f9527107SDr. David Alan Gilbert 
348f9527107SDr. David Alan Gilbert     /*
349f9527107SDr. David Alan Gilbert      * Before we do discards we need to ensure those discards really
350f9527107SDr. David Alan Gilbert      * do delete areas of the page, even if THP thinks a hugepage would
351f9527107SDr. David Alan Gilbert      * be a good idea, so force hugepages off.
352f9527107SDr. David Alan Gilbert      */
3531d741439SDr. David Alan Gilbert     qemu_madvise(host_addr, length, QEMU_MADV_NOHUGEPAGE);
354f9527107SDr. David Alan Gilbert 
355f9527107SDr. David Alan Gilbert     return 0;
356f9527107SDr. David Alan Gilbert }
357f9527107SDr. David Alan Gilbert 
358f9527107SDr. David Alan Gilbert /*
359f9527107SDr. David Alan Gilbert  * Userfault requires us to mark RAM as NOHUGEPAGE prior to discard
360f9527107SDr. David Alan Gilbert  * however leaving it until after precopy means that most of the precopy
361f9527107SDr. David Alan Gilbert  * data is still THPd
362f9527107SDr. David Alan Gilbert  */
363f9527107SDr. David Alan Gilbert int postcopy_ram_prepare_discard(MigrationIncomingState *mis)
364f9527107SDr. David Alan Gilbert {
365f9527107SDr. David Alan Gilbert     if (qemu_ram_foreach_block(nhp_range, mis)) {
366f9527107SDr. David Alan Gilbert         return -1;
367f9527107SDr. David Alan Gilbert     }
368f9527107SDr. David Alan Gilbert 
369f9527107SDr. David Alan Gilbert     postcopy_state_set(POSTCOPY_INCOMING_DISCARD);
370f9527107SDr. David Alan Gilbert 
371f9527107SDr. David Alan Gilbert     return 0;
372f9527107SDr. David Alan Gilbert }
373f9527107SDr. David Alan Gilbert 
374f9527107SDr. David Alan Gilbert /*
375f0a227adSDr. David Alan Gilbert  * Mark the given area of RAM as requiring notification to unwritten areas
376f0a227adSDr. David Alan Gilbert  * Used as a  callback on qemu_ram_foreach_block.
377f0a227adSDr. David Alan Gilbert  *   host_addr: Base of area to mark
378f0a227adSDr. David Alan Gilbert  *   offset: Offset in the whole ram arena
379f0a227adSDr. David Alan Gilbert  *   length: Length of the section
380f0a227adSDr. David Alan Gilbert  *   opaque: MigrationIncomingState pointer
381f0a227adSDr. David Alan Gilbert  * Returns 0 on success
382f0a227adSDr. David Alan Gilbert  */
383f0a227adSDr. David Alan Gilbert static int ram_block_enable_notify(const char *block_name, void *host_addr,
384f0a227adSDr. David Alan Gilbert                                    ram_addr_t offset, ram_addr_t length,
385f0a227adSDr. David Alan Gilbert                                    void *opaque)
386f0a227adSDr. David Alan Gilbert {
387f0a227adSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
388f0a227adSDr. David Alan Gilbert     struct uffdio_register reg_struct;
389f0a227adSDr. David Alan Gilbert 
390f0a227adSDr. David Alan Gilbert     reg_struct.range.start = (uintptr_t)host_addr;
391f0a227adSDr. David Alan Gilbert     reg_struct.range.len = length;
392f0a227adSDr. David Alan Gilbert     reg_struct.mode = UFFDIO_REGISTER_MODE_MISSING;
393f0a227adSDr. David Alan Gilbert 
394f0a227adSDr. David Alan Gilbert     /* Now tell our userfault_fd that it's responsible for this area */
395f0a227adSDr. David Alan Gilbert     if (ioctl(mis->userfault_fd, UFFDIO_REGISTER, &reg_struct)) {
396f0a227adSDr. David Alan Gilbert         error_report("%s userfault register: %s", __func__, strerror(errno));
397f0a227adSDr. David Alan Gilbert         return -1;
398f0a227adSDr. David Alan Gilbert     }
399665414adSDr. David Alan Gilbert     if (!(reg_struct.ioctls & ((__u64)1 << _UFFDIO_COPY))) {
400665414adSDr. David Alan Gilbert         error_report("%s userfault: Region doesn't support COPY", __func__);
401665414adSDr. David Alan Gilbert         return -1;
402665414adSDr. David Alan Gilbert     }
403f0a227adSDr. David Alan Gilbert 
404f0a227adSDr. David Alan Gilbert     return 0;
405f0a227adSDr. David Alan Gilbert }
406f0a227adSDr. David Alan Gilbert 
407f0a227adSDr. David Alan Gilbert /*
408f0a227adSDr. David Alan Gilbert  * Handle faults detected by the USERFAULT markings
409f0a227adSDr. David Alan Gilbert  */
410f0a227adSDr. David Alan Gilbert static void *postcopy_ram_fault_thread(void *opaque)
411f0a227adSDr. David Alan Gilbert {
412f0a227adSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
413c4faeed2SDr. David Alan Gilbert     struct uffd_msg msg;
414c4faeed2SDr. David Alan Gilbert     int ret;
415c4faeed2SDr. David Alan Gilbert     RAMBlock *rb = NULL;
416c4faeed2SDr. David Alan Gilbert     RAMBlock *last_rb = NULL; /* last RAMBlock we sent part of */
417f0a227adSDr. David Alan Gilbert 
418c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_fault_thread_entry();
419f0a227adSDr. David Alan Gilbert     qemu_sem_post(&mis->fault_thread_sem);
420c4faeed2SDr. David Alan Gilbert 
421c4faeed2SDr. David Alan Gilbert     while (true) {
422c4faeed2SDr. David Alan Gilbert         ram_addr_t rb_offset;
423c4faeed2SDr. David Alan Gilbert         struct pollfd pfd[2];
424c4faeed2SDr. David Alan Gilbert 
425c4faeed2SDr. David Alan Gilbert         /*
426c4faeed2SDr. David Alan Gilbert          * We're mainly waiting for the kernel to give us a faulting HVA,
427c4faeed2SDr. David Alan Gilbert          * however we can be told to quit via userfault_quit_fd which is
428c4faeed2SDr. David Alan Gilbert          * an eventfd
429c4faeed2SDr. David Alan Gilbert          */
430c4faeed2SDr. David Alan Gilbert         pfd[0].fd = mis->userfault_fd;
431c4faeed2SDr. David Alan Gilbert         pfd[0].events = POLLIN;
432c4faeed2SDr. David Alan Gilbert         pfd[0].revents = 0;
433c4faeed2SDr. David Alan Gilbert         pfd[1].fd = mis->userfault_quit_fd;
434c4faeed2SDr. David Alan Gilbert         pfd[1].events = POLLIN; /* Waiting for eventfd to go positive */
435c4faeed2SDr. David Alan Gilbert         pfd[1].revents = 0;
436c4faeed2SDr. David Alan Gilbert 
437c4faeed2SDr. David Alan Gilbert         if (poll(pfd, 2, -1 /* Wait forever */) == -1) {
438c4faeed2SDr. David Alan Gilbert             error_report("%s: userfault poll: %s", __func__, strerror(errno));
439c4faeed2SDr. David Alan Gilbert             break;
440f0a227adSDr. David Alan Gilbert         }
441f0a227adSDr. David Alan Gilbert 
442c4faeed2SDr. David Alan Gilbert         if (pfd[1].revents) {
443c4faeed2SDr. David Alan Gilbert             trace_postcopy_ram_fault_thread_quit();
444c4faeed2SDr. David Alan Gilbert             break;
445c4faeed2SDr. David Alan Gilbert         }
446c4faeed2SDr. David Alan Gilbert 
447c4faeed2SDr. David Alan Gilbert         ret = read(mis->userfault_fd, &msg, sizeof(msg));
448c4faeed2SDr. David Alan Gilbert         if (ret != sizeof(msg)) {
449c4faeed2SDr. David Alan Gilbert             if (errno == EAGAIN) {
450c4faeed2SDr. David Alan Gilbert                 /*
451c4faeed2SDr. David Alan Gilbert                  * if a wake up happens on the other thread just after
452c4faeed2SDr. David Alan Gilbert                  * the poll, there is nothing to read.
453c4faeed2SDr. David Alan Gilbert                  */
454c4faeed2SDr. David Alan Gilbert                 continue;
455c4faeed2SDr. David Alan Gilbert             }
456c4faeed2SDr. David Alan Gilbert             if (ret < 0) {
457c4faeed2SDr. David Alan Gilbert                 error_report("%s: Failed to read full userfault message: %s",
458c4faeed2SDr. David Alan Gilbert                              __func__, strerror(errno));
459c4faeed2SDr. David Alan Gilbert                 break;
460c4faeed2SDr. David Alan Gilbert             } else {
461c4faeed2SDr. David Alan Gilbert                 error_report("%s: Read %d bytes from userfaultfd expected %zd",
462c4faeed2SDr. David Alan Gilbert                              __func__, ret, sizeof(msg));
463c4faeed2SDr. David Alan Gilbert                 break; /* Lost alignment, don't know what we'd read next */
464c4faeed2SDr. David Alan Gilbert             }
465c4faeed2SDr. David Alan Gilbert         }
466c4faeed2SDr. David Alan Gilbert         if (msg.event != UFFD_EVENT_PAGEFAULT) {
467c4faeed2SDr. David Alan Gilbert             error_report("%s: Read unexpected event %ud from userfaultfd",
468c4faeed2SDr. David Alan Gilbert                          __func__, msg.event);
469c4faeed2SDr. David Alan Gilbert             continue; /* It's not a page fault, shouldn't happen */
470c4faeed2SDr. David Alan Gilbert         }
471c4faeed2SDr. David Alan Gilbert 
472c4faeed2SDr. David Alan Gilbert         rb = qemu_ram_block_from_host(
473c4faeed2SDr. David Alan Gilbert                  (void *)(uintptr_t)msg.arg.pagefault.address,
474f615f396SPaolo Bonzini                  true, &rb_offset);
475c4faeed2SDr. David Alan Gilbert         if (!rb) {
476c4faeed2SDr. David Alan Gilbert             error_report("postcopy_ram_fault_thread: Fault outside guest: %"
477c4faeed2SDr. David Alan Gilbert                          PRIx64, (uint64_t)msg.arg.pagefault.address);
478c4faeed2SDr. David Alan Gilbert             break;
479c4faeed2SDr. David Alan Gilbert         }
480c4faeed2SDr. David Alan Gilbert 
481332847f0SDr. David Alan Gilbert         rb_offset &= ~(qemu_ram_pagesize(rb) - 1);
482c4faeed2SDr. David Alan Gilbert         trace_postcopy_ram_fault_thread_request(msg.arg.pagefault.address,
483c4faeed2SDr. David Alan Gilbert                                                 qemu_ram_get_idstr(rb),
484c4faeed2SDr. David Alan Gilbert                                                 rb_offset);
485c4faeed2SDr. David Alan Gilbert 
486c4faeed2SDr. David Alan Gilbert         /*
487c4faeed2SDr. David Alan Gilbert          * Send the request to the source - we want to request one
488c4faeed2SDr. David Alan Gilbert          * of our host page sizes (which is >= TPS)
489c4faeed2SDr. David Alan Gilbert          */
490c4faeed2SDr. David Alan Gilbert         if (rb != last_rb) {
491c4faeed2SDr. David Alan Gilbert             last_rb = rb;
492c4faeed2SDr. David Alan Gilbert             migrate_send_rp_req_pages(mis, qemu_ram_get_idstr(rb),
493332847f0SDr. David Alan Gilbert                                      rb_offset, qemu_ram_pagesize(rb));
494c4faeed2SDr. David Alan Gilbert         } else {
495c4faeed2SDr. David Alan Gilbert             /* Save some space */
496c4faeed2SDr. David Alan Gilbert             migrate_send_rp_req_pages(mis, NULL,
497332847f0SDr. David Alan Gilbert                                      rb_offset, qemu_ram_pagesize(rb));
498c4faeed2SDr. David Alan Gilbert         }
499c4faeed2SDr. David Alan Gilbert     }
500c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_fault_thread_exit();
501f0a227adSDr. David Alan Gilbert     return NULL;
502f0a227adSDr. David Alan Gilbert }
503f0a227adSDr. David Alan Gilbert 
504f0a227adSDr. David Alan Gilbert int postcopy_ram_enable_notify(MigrationIncomingState *mis)
505f0a227adSDr. David Alan Gilbert {
506c4faeed2SDr. David Alan Gilbert     /* Open the fd for the kernel to give us userfaults */
507c4faeed2SDr. David Alan Gilbert     mis->userfault_fd = syscall(__NR_userfaultfd, O_CLOEXEC | O_NONBLOCK);
508c4faeed2SDr. David Alan Gilbert     if (mis->userfault_fd == -1) {
509c4faeed2SDr. David Alan Gilbert         error_report("%s: Failed to open userfault fd: %s", __func__,
510c4faeed2SDr. David Alan Gilbert                      strerror(errno));
511c4faeed2SDr. David Alan Gilbert         return -1;
512c4faeed2SDr. David Alan Gilbert     }
513c4faeed2SDr. David Alan Gilbert 
514c4faeed2SDr. David Alan Gilbert     /*
515c4faeed2SDr. David Alan Gilbert      * Although the host check already tested the API, we need to
516c4faeed2SDr. David Alan Gilbert      * do the check again as an ABI handshake on the new fd.
517c4faeed2SDr. David Alan Gilbert      */
518c4faeed2SDr. David Alan Gilbert     if (!ufd_version_check(mis->userfault_fd)) {
519c4faeed2SDr. David Alan Gilbert         return -1;
520c4faeed2SDr. David Alan Gilbert     }
521c4faeed2SDr. David Alan Gilbert 
522c4faeed2SDr. David Alan Gilbert     /* Now an eventfd we use to tell the fault-thread to quit */
523c4faeed2SDr. David Alan Gilbert     mis->userfault_quit_fd = eventfd(0, EFD_CLOEXEC);
524c4faeed2SDr. David Alan Gilbert     if (mis->userfault_quit_fd == -1) {
525c4faeed2SDr. David Alan Gilbert         error_report("%s: Opening userfault_quit_fd: %s", __func__,
526c4faeed2SDr. David Alan Gilbert                      strerror(errno));
527c4faeed2SDr. David Alan Gilbert         close(mis->userfault_fd);
528c4faeed2SDr. David Alan Gilbert         return -1;
529c4faeed2SDr. David Alan Gilbert     }
530c4faeed2SDr. David Alan Gilbert 
531f0a227adSDr. David Alan Gilbert     qemu_sem_init(&mis->fault_thread_sem, 0);
532f0a227adSDr. David Alan Gilbert     qemu_thread_create(&mis->fault_thread, "postcopy/fault",
533f0a227adSDr. David Alan Gilbert                        postcopy_ram_fault_thread, mis, QEMU_THREAD_JOINABLE);
534f0a227adSDr. David Alan Gilbert     qemu_sem_wait(&mis->fault_thread_sem);
535f0a227adSDr. David Alan Gilbert     qemu_sem_destroy(&mis->fault_thread_sem);
536c4faeed2SDr. David Alan Gilbert     mis->have_fault_thread = true;
537f0a227adSDr. David Alan Gilbert 
538f0a227adSDr. David Alan Gilbert     /* Mark so that we get notified of accesses to unwritten areas */
539f0a227adSDr. David Alan Gilbert     if (qemu_ram_foreach_block(ram_block_enable_notify, mis)) {
540f0a227adSDr. David Alan Gilbert         return -1;
541f0a227adSDr. David Alan Gilbert     }
542f0a227adSDr. David Alan Gilbert 
543371ff5a3SDr. David Alan Gilbert     /*
544371ff5a3SDr. David Alan Gilbert      * Ballooning can mark pages as absent while we're postcopying
545371ff5a3SDr. David Alan Gilbert      * that would cause false userfaults.
546371ff5a3SDr. David Alan Gilbert      */
547371ff5a3SDr. David Alan Gilbert     qemu_balloon_inhibit(true);
548371ff5a3SDr. David Alan Gilbert 
549c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_enable_notify();
550c4faeed2SDr. David Alan Gilbert 
551f0a227adSDr. David Alan Gilbert     return 0;
552f0a227adSDr. David Alan Gilbert }
553f0a227adSDr. David Alan Gilbert 
554696ed9a9SDr. David Alan Gilbert /*
555696ed9a9SDr. David Alan Gilbert  * Place a host page (from) at (host) atomically
556696ed9a9SDr. David Alan Gilbert  * returns 0 on success
557696ed9a9SDr. David Alan Gilbert  */
558df9ff5e1SDr. David Alan Gilbert int postcopy_place_page(MigrationIncomingState *mis, void *host, void *from,
559df9ff5e1SDr. David Alan Gilbert                         size_t pagesize)
560696ed9a9SDr. David Alan Gilbert {
561696ed9a9SDr. David Alan Gilbert     struct uffdio_copy copy_struct;
562696ed9a9SDr. David Alan Gilbert 
563696ed9a9SDr. David Alan Gilbert     copy_struct.dst = (uint64_t)(uintptr_t)host;
564696ed9a9SDr. David Alan Gilbert     copy_struct.src = (uint64_t)(uintptr_t)from;
565df9ff5e1SDr. David Alan Gilbert     copy_struct.len = pagesize;
566696ed9a9SDr. David Alan Gilbert     copy_struct.mode = 0;
567696ed9a9SDr. David Alan Gilbert 
568696ed9a9SDr. David Alan Gilbert     /* copy also acks to the kernel waking the stalled thread up
569696ed9a9SDr. David Alan Gilbert      * TODO: We can inhibit that ack and only do it if it was requested
570696ed9a9SDr. David Alan Gilbert      * which would be slightly cheaper, but we'd have to be careful
571696ed9a9SDr. David Alan Gilbert      * of the order of updating our page state.
572696ed9a9SDr. David Alan Gilbert      */
573696ed9a9SDr. David Alan Gilbert     if (ioctl(mis->userfault_fd, UFFDIO_COPY, &copy_struct)) {
574696ed9a9SDr. David Alan Gilbert         int e = errno;
575df9ff5e1SDr. David Alan Gilbert         error_report("%s: %s copy host: %p from: %p (size: %zd)",
576df9ff5e1SDr. David Alan Gilbert                      __func__, strerror(e), host, from, pagesize);
577696ed9a9SDr. David Alan Gilbert 
578696ed9a9SDr. David Alan Gilbert         return -e;
579696ed9a9SDr. David Alan Gilbert     }
580696ed9a9SDr. David Alan Gilbert 
581696ed9a9SDr. David Alan Gilbert     trace_postcopy_place_page(host);
582696ed9a9SDr. David Alan Gilbert     return 0;
583696ed9a9SDr. David Alan Gilbert }
584696ed9a9SDr. David Alan Gilbert 
585696ed9a9SDr. David Alan Gilbert /*
586696ed9a9SDr. David Alan Gilbert  * Place a zero page at (host) atomically
587696ed9a9SDr. David Alan Gilbert  * returns 0 on success
588696ed9a9SDr. David Alan Gilbert  */
589df9ff5e1SDr. David Alan Gilbert int postcopy_place_page_zero(MigrationIncomingState *mis, void *host,
590df9ff5e1SDr. David Alan Gilbert                              size_t pagesize)
591696ed9a9SDr. David Alan Gilbert {
592df9ff5e1SDr. David Alan Gilbert     trace_postcopy_place_page_zero(host);
593696ed9a9SDr. David Alan Gilbert 
594df9ff5e1SDr. David Alan Gilbert     if (pagesize == getpagesize()) {
595df9ff5e1SDr. David Alan Gilbert         struct uffdio_zeropage zero_struct;
596696ed9a9SDr. David Alan Gilbert         zero_struct.range.start = (uint64_t)(uintptr_t)host;
597696ed9a9SDr. David Alan Gilbert         zero_struct.range.len = getpagesize();
598696ed9a9SDr. David Alan Gilbert         zero_struct.mode = 0;
599696ed9a9SDr. David Alan Gilbert 
600696ed9a9SDr. David Alan Gilbert         if (ioctl(mis->userfault_fd, UFFDIO_ZEROPAGE, &zero_struct)) {
601696ed9a9SDr. David Alan Gilbert             int e = errno;
602696ed9a9SDr. David Alan Gilbert             error_report("%s: %s zero host: %p",
603696ed9a9SDr. David Alan Gilbert                          __func__, strerror(e), host);
604696ed9a9SDr. David Alan Gilbert 
605696ed9a9SDr. David Alan Gilbert             return -e;
606696ed9a9SDr. David Alan Gilbert         }
607df9ff5e1SDr. David Alan Gilbert     } else {
60841d84210SDr. David Alan Gilbert         /* The kernel can't use UFFDIO_ZEROPAGE for hugepages */
60941d84210SDr. David Alan Gilbert         if (!mis->postcopy_tmp_zero_page) {
61041d84210SDr. David Alan Gilbert             mis->postcopy_tmp_zero_page = mmap(NULL, mis->largest_page_size,
61141d84210SDr. David Alan Gilbert                                                PROT_READ | PROT_WRITE,
61241d84210SDr. David Alan Gilbert                                                MAP_PRIVATE | MAP_ANONYMOUS,
61341d84210SDr. David Alan Gilbert                                                -1, 0);
61441d84210SDr. David Alan Gilbert             if (mis->postcopy_tmp_zero_page == MAP_FAILED) {
61541d84210SDr. David Alan Gilbert                 int e = errno;
61641d84210SDr. David Alan Gilbert                 mis->postcopy_tmp_zero_page = NULL;
61741d84210SDr. David Alan Gilbert                 error_report("%s: %s mapping large zero page",
61841d84210SDr. David Alan Gilbert                              __func__, strerror(e));
61941d84210SDr. David Alan Gilbert                 return -e;
62041d84210SDr. David Alan Gilbert             }
62141d84210SDr. David Alan Gilbert             memset(mis->postcopy_tmp_zero_page, '\0', mis->largest_page_size);
62241d84210SDr. David Alan Gilbert         }
62341d84210SDr. David Alan Gilbert         return postcopy_place_page(mis, host, mis->postcopy_tmp_zero_page,
62441d84210SDr. David Alan Gilbert                                    pagesize);
625df9ff5e1SDr. David Alan Gilbert     }
626696ed9a9SDr. David Alan Gilbert 
627696ed9a9SDr. David Alan Gilbert     return 0;
628696ed9a9SDr. David Alan Gilbert }
629696ed9a9SDr. David Alan Gilbert 
630696ed9a9SDr. David Alan Gilbert /*
631696ed9a9SDr. David Alan Gilbert  * Returns a target page of memory that can be mapped at a later point in time
632696ed9a9SDr. David Alan Gilbert  * using postcopy_place_page
633696ed9a9SDr. David Alan Gilbert  * The same address is used repeatedly, postcopy_place_page just takes the
634696ed9a9SDr. David Alan Gilbert  * backing page away.
635696ed9a9SDr. David Alan Gilbert  * Returns: Pointer to allocated page
636696ed9a9SDr. David Alan Gilbert  *
637696ed9a9SDr. David Alan Gilbert  */
638696ed9a9SDr. David Alan Gilbert void *postcopy_get_tmp_page(MigrationIncomingState *mis)
639696ed9a9SDr. David Alan Gilbert {
640696ed9a9SDr. David Alan Gilbert     if (!mis->postcopy_tmp_page) {
641df9ff5e1SDr. David Alan Gilbert         mis->postcopy_tmp_page = mmap(NULL, mis->largest_page_size,
642696ed9a9SDr. David Alan Gilbert                              PROT_READ | PROT_WRITE, MAP_PRIVATE |
643696ed9a9SDr. David Alan Gilbert                              MAP_ANONYMOUS, -1, 0);
6440e8b3cdfSEvgeny Yakovlev         if (mis->postcopy_tmp_page == MAP_FAILED) {
6450e8b3cdfSEvgeny Yakovlev             mis->postcopy_tmp_page = NULL;
646696ed9a9SDr. David Alan Gilbert             error_report("%s: %s", __func__, strerror(errno));
647696ed9a9SDr. David Alan Gilbert             return NULL;
648696ed9a9SDr. David Alan Gilbert         }
649696ed9a9SDr. David Alan Gilbert     }
650696ed9a9SDr. David Alan Gilbert 
651696ed9a9SDr. David Alan Gilbert     return mis->postcopy_tmp_page;
652696ed9a9SDr. David Alan Gilbert }
653696ed9a9SDr. David Alan Gilbert 
654eb59db53SDr. David Alan Gilbert #else
655eb59db53SDr. David Alan Gilbert /* No target OS support, stubs just fail */
656eb59db53SDr. David Alan Gilbert bool postcopy_ram_supported_by_host(void)
657eb59db53SDr. David Alan Gilbert {
658eb59db53SDr. David Alan Gilbert     error_report("%s: No OS support", __func__);
659eb59db53SDr. David Alan Gilbert     return false;
660eb59db53SDr. David Alan Gilbert }
661eb59db53SDr. David Alan Gilbert 
6621caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_init(MigrationIncomingState *mis, size_t ram_pages)
6631caddf8aSDr. David Alan Gilbert {
6641caddf8aSDr. David Alan Gilbert     error_report("postcopy_ram_incoming_init: No OS support");
6651caddf8aSDr. David Alan Gilbert     return -1;
6661caddf8aSDr. David Alan Gilbert }
6671caddf8aSDr. David Alan Gilbert 
6681caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_cleanup(MigrationIncomingState *mis)
6691caddf8aSDr. David Alan Gilbert {
6701caddf8aSDr. David Alan Gilbert     assert(0);
6711caddf8aSDr. David Alan Gilbert     return -1;
6721caddf8aSDr. David Alan Gilbert }
6731caddf8aSDr. David Alan Gilbert 
674f9527107SDr. David Alan Gilbert int postcopy_ram_prepare_discard(MigrationIncomingState *mis)
675f9527107SDr. David Alan Gilbert {
676f9527107SDr. David Alan Gilbert     assert(0);
677f9527107SDr. David Alan Gilbert     return -1;
678f9527107SDr. David Alan Gilbert }
679f9527107SDr. David Alan Gilbert 
680f0a227adSDr. David Alan Gilbert int postcopy_ram_enable_notify(MigrationIncomingState *mis)
681f0a227adSDr. David Alan Gilbert {
682f0a227adSDr. David Alan Gilbert     assert(0);
683f0a227adSDr. David Alan Gilbert     return -1;
684f0a227adSDr. David Alan Gilbert }
685696ed9a9SDr. David Alan Gilbert 
686df9ff5e1SDr. David Alan Gilbert int postcopy_place_page(MigrationIncomingState *mis, void *host, void *from,
687df9ff5e1SDr. David Alan Gilbert                         size_t pagesize)
688696ed9a9SDr. David Alan Gilbert {
689696ed9a9SDr. David Alan Gilbert     assert(0);
690696ed9a9SDr. David Alan Gilbert     return -1;
691696ed9a9SDr. David Alan Gilbert }
692696ed9a9SDr. David Alan Gilbert 
693df9ff5e1SDr. David Alan Gilbert int postcopy_place_page_zero(MigrationIncomingState *mis, void *host,
694df9ff5e1SDr. David Alan Gilbert                         size_t pagesize)
695696ed9a9SDr. David Alan Gilbert {
696696ed9a9SDr. David Alan Gilbert     assert(0);
697696ed9a9SDr. David Alan Gilbert     return -1;
698696ed9a9SDr. David Alan Gilbert }
699696ed9a9SDr. David Alan Gilbert 
700696ed9a9SDr. David Alan Gilbert void *postcopy_get_tmp_page(MigrationIncomingState *mis)
701696ed9a9SDr. David Alan Gilbert {
702696ed9a9SDr. David Alan Gilbert     assert(0);
703696ed9a9SDr. David Alan Gilbert     return NULL;
704696ed9a9SDr. David Alan Gilbert }
705696ed9a9SDr. David Alan Gilbert 
706eb59db53SDr. David Alan Gilbert #endif
707eb59db53SDr. David Alan Gilbert 
708e0b266f0SDr. David Alan Gilbert /* ------------------------------------------------------------------------- */
709e0b266f0SDr. David Alan Gilbert 
710e0b266f0SDr. David Alan Gilbert /**
711e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_init: Called at the start of each RAMBlock before
712e0b266f0SDr. David Alan Gilbert  *   asking to discard individual ranges.
713e0b266f0SDr. David Alan Gilbert  *
714e0b266f0SDr. David Alan Gilbert  * @ms: The current migration state.
715e0b266f0SDr. David Alan Gilbert  * @offset: the bitmap offset of the named RAMBlock in the migration
716e0b266f0SDr. David Alan Gilbert  *   bitmap.
717e0b266f0SDr. David Alan Gilbert  * @name: RAMBlock that discards will operate on.
718e0b266f0SDr. David Alan Gilbert  *
719e0b266f0SDr. David Alan Gilbert  * returns: a new PDS.
720e0b266f0SDr. David Alan Gilbert  */
721e0b266f0SDr. David Alan Gilbert PostcopyDiscardState *postcopy_discard_send_init(MigrationState *ms,
722e0b266f0SDr. David Alan Gilbert                                                  unsigned long offset,
723e0b266f0SDr. David Alan Gilbert                                                  const char *name)
724e0b266f0SDr. David Alan Gilbert {
725e0b266f0SDr. David Alan Gilbert     PostcopyDiscardState *res = g_malloc0(sizeof(PostcopyDiscardState));
726e0b266f0SDr. David Alan Gilbert 
727e0b266f0SDr. David Alan Gilbert     if (res) {
728e0b266f0SDr. David Alan Gilbert         res->ramblock_name = name;
729e0b266f0SDr. David Alan Gilbert         res->offset = offset;
730e0b266f0SDr. David Alan Gilbert     }
731e0b266f0SDr. David Alan Gilbert 
732e0b266f0SDr. David Alan Gilbert     return res;
733e0b266f0SDr. David Alan Gilbert }
734e0b266f0SDr. David Alan Gilbert 
735e0b266f0SDr. David Alan Gilbert /**
736e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_range: Called by the bitmap code for each chunk to
737e0b266f0SDr. David Alan Gilbert  *   discard. May send a discard message, may just leave it queued to
738e0b266f0SDr. David Alan Gilbert  *   be sent later.
739e0b266f0SDr. David Alan Gilbert  *
740e0b266f0SDr. David Alan Gilbert  * @ms: Current migration state.
741e0b266f0SDr. David Alan Gilbert  * @pds: Structure initialised by postcopy_discard_send_init().
742e0b266f0SDr. David Alan Gilbert  * @start,@length: a range of pages in the migration bitmap in the
743e0b266f0SDr. David Alan Gilbert  *   RAM block passed to postcopy_discard_send_init() (length=1 is one page)
744e0b266f0SDr. David Alan Gilbert  */
745e0b266f0SDr. David Alan Gilbert void postcopy_discard_send_range(MigrationState *ms, PostcopyDiscardState *pds,
746e0b266f0SDr. David Alan Gilbert                                 unsigned long start, unsigned long length)
747e0b266f0SDr. David Alan Gilbert {
748e0b266f0SDr. David Alan Gilbert     size_t tp_bits = qemu_target_page_bits();
749e0b266f0SDr. David Alan Gilbert     /* Convert to byte offsets within the RAM block */
750e0b266f0SDr. David Alan Gilbert     pds->start_list[pds->cur_entry] = (start - pds->offset) << tp_bits;
751e0b266f0SDr. David Alan Gilbert     pds->length_list[pds->cur_entry] = length << tp_bits;
752e0b266f0SDr. David Alan Gilbert     trace_postcopy_discard_send_range(pds->ramblock_name, start, length);
753e0b266f0SDr. David Alan Gilbert     pds->cur_entry++;
754e0b266f0SDr. David Alan Gilbert     pds->nsentwords++;
755e0b266f0SDr. David Alan Gilbert 
756e0b266f0SDr. David Alan Gilbert     if (pds->cur_entry == MAX_DISCARDS_PER_COMMAND) {
757e0b266f0SDr. David Alan Gilbert         /* Full set, ship it! */
75889a02a9fSzhanghailiang         qemu_savevm_send_postcopy_ram_discard(ms->to_dst_file,
75989a02a9fSzhanghailiang                                               pds->ramblock_name,
760e0b266f0SDr. David Alan Gilbert                                               pds->cur_entry,
761e0b266f0SDr. David Alan Gilbert                                               pds->start_list,
762e0b266f0SDr. David Alan Gilbert                                               pds->length_list);
763e0b266f0SDr. David Alan Gilbert         pds->nsentcmds++;
764e0b266f0SDr. David Alan Gilbert         pds->cur_entry = 0;
765e0b266f0SDr. David Alan Gilbert     }
766e0b266f0SDr. David Alan Gilbert }
767e0b266f0SDr. David Alan Gilbert 
768e0b266f0SDr. David Alan Gilbert /**
769e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_finish: Called at the end of each RAMBlock by the
770e0b266f0SDr. David Alan Gilbert  * bitmap code. Sends any outstanding discard messages, frees the PDS
771e0b266f0SDr. David Alan Gilbert  *
772e0b266f0SDr. David Alan Gilbert  * @ms: Current migration state.
773e0b266f0SDr. David Alan Gilbert  * @pds: Structure initialised by postcopy_discard_send_init().
774e0b266f0SDr. David Alan Gilbert  */
775e0b266f0SDr. David Alan Gilbert void postcopy_discard_send_finish(MigrationState *ms, PostcopyDiscardState *pds)
776e0b266f0SDr. David Alan Gilbert {
777e0b266f0SDr. David Alan Gilbert     /* Anything unsent? */
778e0b266f0SDr. David Alan Gilbert     if (pds->cur_entry) {
77989a02a9fSzhanghailiang         qemu_savevm_send_postcopy_ram_discard(ms->to_dst_file,
78089a02a9fSzhanghailiang                                               pds->ramblock_name,
781e0b266f0SDr. David Alan Gilbert                                               pds->cur_entry,
782e0b266f0SDr. David Alan Gilbert                                               pds->start_list,
783e0b266f0SDr. David Alan Gilbert                                               pds->length_list);
784e0b266f0SDr. David Alan Gilbert         pds->nsentcmds++;
785e0b266f0SDr. David Alan Gilbert     }
786e0b266f0SDr. David Alan Gilbert 
787e0b266f0SDr. David Alan Gilbert     trace_postcopy_discard_send_finish(pds->ramblock_name, pds->nsentwords,
788e0b266f0SDr. David Alan Gilbert                                        pds->nsentcmds);
789e0b266f0SDr. David Alan Gilbert 
790e0b266f0SDr. David Alan Gilbert     g_free(pds);
791e0b266f0SDr. David Alan Gilbert }
792