xref: /qemu/migration/postcopy-ram.c (revision 27dd21b461569f3b85a9db409f9117ae9d976548)
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"
20898ba906SDavid Hildenbrand #include "qemu/rcu.h"
21b85ea5faSPeter Maydell #include "qemu/madvise.h"
2251180423SJuan Quintela #include "exec/target_page.h"
236666c96aSJuan Quintela #include "migration.h"
2408a0aee1SJuan Quintela #include "qemu-file.h"
2520a519a0SJuan Quintela #include "savevm.h"
26be07b0acSJuan Quintela #include "postcopy-ram.h"
277b1e1a22SJuan Quintela #include "ram.h"
281693c64cSDr. David Alan Gilbert #include "qapi/error.h"
291693c64cSDr. David Alan Gilbert #include "qemu/notify.h"
30d4842052SMarkus Armbruster #include "qemu/rcu.h"
31eb59db53SDr. David Alan Gilbert #include "sysemu/sysemu.h"
32eb59db53SDr. David Alan Gilbert #include "qemu/error-report.h"
33eb59db53SDr. David Alan Gilbert #include "trace.h"
345cc8767dSLike Xu #include "hw/boards.h"
35898ba906SDavid Hildenbrand #include "exec/ramblock.h"
36eb59db53SDr. David Alan Gilbert 
37e0b266f0SDr. David Alan Gilbert /* Arbitrary limit on size of each discard command,
38e0b266f0SDr. David Alan Gilbert  * keeps them around ~200 bytes
39e0b266f0SDr. David Alan Gilbert  */
40e0b266f0SDr. David Alan Gilbert #define MAX_DISCARDS_PER_COMMAND 12
41e0b266f0SDr. David Alan Gilbert 
42e0b266f0SDr. David Alan Gilbert struct PostcopyDiscardState {
43e0b266f0SDr. David Alan Gilbert     const char *ramblock_name;
44e0b266f0SDr. David Alan Gilbert     uint16_t cur_entry;
45e0b266f0SDr. David Alan Gilbert     /*
46e0b266f0SDr. David Alan Gilbert      * Start and length of a discard range (bytes)
47e0b266f0SDr. David Alan Gilbert      */
48e0b266f0SDr. David Alan Gilbert     uint64_t start_list[MAX_DISCARDS_PER_COMMAND];
49e0b266f0SDr. David Alan Gilbert     uint64_t length_list[MAX_DISCARDS_PER_COMMAND];
50e0b266f0SDr. David Alan Gilbert     unsigned int nsentwords;
51e0b266f0SDr. David Alan Gilbert     unsigned int nsentcmds;
52e0b266f0SDr. David Alan Gilbert };
53e0b266f0SDr. David Alan Gilbert 
541693c64cSDr. David Alan Gilbert static NotifierWithReturnList postcopy_notifier_list;
551693c64cSDr. David Alan Gilbert 
561693c64cSDr. David Alan Gilbert void postcopy_infrastructure_init(void)
571693c64cSDr. David Alan Gilbert {
581693c64cSDr. David Alan Gilbert     notifier_with_return_list_init(&postcopy_notifier_list);
591693c64cSDr. David Alan Gilbert }
601693c64cSDr. David Alan Gilbert 
611693c64cSDr. David Alan Gilbert void postcopy_add_notifier(NotifierWithReturn *nn)
621693c64cSDr. David Alan Gilbert {
631693c64cSDr. David Alan Gilbert     notifier_with_return_list_add(&postcopy_notifier_list, nn);
641693c64cSDr. David Alan Gilbert }
651693c64cSDr. David Alan Gilbert 
661693c64cSDr. David Alan Gilbert void postcopy_remove_notifier(NotifierWithReturn *n)
671693c64cSDr. David Alan Gilbert {
681693c64cSDr. David Alan Gilbert     notifier_with_return_remove(n);
691693c64cSDr. David Alan Gilbert }
701693c64cSDr. David Alan Gilbert 
711693c64cSDr. David Alan Gilbert int postcopy_notify(enum PostcopyNotifyReason reason, Error **errp)
721693c64cSDr. David Alan Gilbert {
731693c64cSDr. David Alan Gilbert     struct PostcopyNotifyData pnd;
741693c64cSDr. David Alan Gilbert     pnd.reason = reason;
751693c64cSDr. David Alan Gilbert     pnd.errp = errp;
761693c64cSDr. David Alan Gilbert 
771693c64cSDr. David Alan Gilbert     return notifier_with_return_list_notify(&postcopy_notifier_list,
781693c64cSDr. David Alan Gilbert                                             &pnd);
791693c64cSDr. David Alan Gilbert }
801693c64cSDr. David Alan Gilbert 
81095c12a4SPeter Xu /*
82095c12a4SPeter Xu  * NOTE: this routine is not thread safe, we can't call it concurrently. But it
83095c12a4SPeter Xu  * should be good enough for migration's purposes.
84095c12a4SPeter Xu  */
85095c12a4SPeter Xu void postcopy_thread_create(MigrationIncomingState *mis,
86095c12a4SPeter Xu                             QemuThread *thread, const char *name,
87095c12a4SPeter Xu                             void *(*fn)(void *), int joinable)
88095c12a4SPeter Xu {
89095c12a4SPeter Xu     qemu_sem_init(&mis->thread_sync_sem, 0);
90095c12a4SPeter Xu     qemu_thread_create(thread, name, fn, mis, joinable);
91095c12a4SPeter Xu     qemu_sem_wait(&mis->thread_sync_sem);
92095c12a4SPeter Xu     qemu_sem_destroy(&mis->thread_sync_sem);
93095c12a4SPeter Xu }
94095c12a4SPeter Xu 
95eb59db53SDr. David Alan Gilbert /* Postcopy needs to detect accesses to pages that haven't yet been copied
96eb59db53SDr. David Alan Gilbert  * across, and efficiently map new pages in, the techniques for doing this
97eb59db53SDr. David Alan Gilbert  * are target OS specific.
98eb59db53SDr. David Alan Gilbert  */
99eb59db53SDr. David Alan Gilbert #if defined(__linux__)
100eb59db53SDr. David Alan Gilbert 
101c4faeed2SDr. David Alan Gilbert #include <poll.h>
102eb59db53SDr. David Alan Gilbert #include <sys/ioctl.h>
103eb59db53SDr. David Alan Gilbert #include <sys/syscall.h>
104eb59db53SDr. David Alan Gilbert #include <asm/types.h> /* for __u64 */
105eb59db53SDr. David Alan Gilbert #endif
106eb59db53SDr. David Alan Gilbert 
107d8b9d771SMatthew Fortune #if defined(__linux__) && defined(__NR_userfaultfd) && defined(CONFIG_EVENTFD)
108d8b9d771SMatthew Fortune #include <sys/eventfd.h>
109eb59db53SDr. David Alan Gilbert #include <linux/userfaultfd.h>
110eb59db53SDr. David Alan Gilbert 
1112a4c42f1SAlexey Perevalov typedef struct PostcopyBlocktimeContext {
1122a4c42f1SAlexey Perevalov     /* time when page fault initiated per vCPU */
1132a4c42f1SAlexey Perevalov     uint32_t *page_fault_vcpu_time;
1142a4c42f1SAlexey Perevalov     /* page address per vCPU */
1152a4c42f1SAlexey Perevalov     uintptr_t *vcpu_addr;
1162a4c42f1SAlexey Perevalov     uint32_t total_blocktime;
1172a4c42f1SAlexey Perevalov     /* blocktime per vCPU */
1182a4c42f1SAlexey Perevalov     uint32_t *vcpu_blocktime;
1192a4c42f1SAlexey Perevalov     /* point in time when last page fault was initiated */
1202a4c42f1SAlexey Perevalov     uint32_t last_begin;
1212a4c42f1SAlexey Perevalov     /* number of vCPU are suspended */
1222a4c42f1SAlexey Perevalov     int smp_cpus_down;
1232a4c42f1SAlexey Perevalov     uint64_t start_time;
1242a4c42f1SAlexey Perevalov 
1252a4c42f1SAlexey Perevalov     /*
1262a4c42f1SAlexey Perevalov      * Handler for exit event, necessary for
1272a4c42f1SAlexey Perevalov      * releasing whole blocktime_ctx
1282a4c42f1SAlexey Perevalov      */
1292a4c42f1SAlexey Perevalov     Notifier exit_notifier;
1302a4c42f1SAlexey Perevalov } PostcopyBlocktimeContext;
1312a4c42f1SAlexey Perevalov 
1322a4c42f1SAlexey Perevalov static void destroy_blocktime_context(struct PostcopyBlocktimeContext *ctx)
1332a4c42f1SAlexey Perevalov {
1342a4c42f1SAlexey Perevalov     g_free(ctx->page_fault_vcpu_time);
1352a4c42f1SAlexey Perevalov     g_free(ctx->vcpu_addr);
1362a4c42f1SAlexey Perevalov     g_free(ctx->vcpu_blocktime);
1372a4c42f1SAlexey Perevalov     g_free(ctx);
1382a4c42f1SAlexey Perevalov }
1392a4c42f1SAlexey Perevalov 
1402a4c42f1SAlexey Perevalov static void migration_exit_cb(Notifier *n, void *data)
1412a4c42f1SAlexey Perevalov {
1422a4c42f1SAlexey Perevalov     PostcopyBlocktimeContext *ctx = container_of(n, PostcopyBlocktimeContext,
1432a4c42f1SAlexey Perevalov                                                  exit_notifier);
1442a4c42f1SAlexey Perevalov     destroy_blocktime_context(ctx);
1452a4c42f1SAlexey Perevalov }
1462a4c42f1SAlexey Perevalov 
1472a4c42f1SAlexey Perevalov static struct PostcopyBlocktimeContext *blocktime_context_new(void)
1482a4c42f1SAlexey Perevalov {
1495cc8767dSLike Xu     MachineState *ms = MACHINE(qdev_get_machine());
1505cc8767dSLike Xu     unsigned int smp_cpus = ms->smp.cpus;
1512a4c42f1SAlexey Perevalov     PostcopyBlocktimeContext *ctx = g_new0(PostcopyBlocktimeContext, 1);
1522a4c42f1SAlexey Perevalov     ctx->page_fault_vcpu_time = g_new0(uint32_t, smp_cpus);
1532a4c42f1SAlexey Perevalov     ctx->vcpu_addr = g_new0(uintptr_t, smp_cpus);
1542a4c42f1SAlexey Perevalov     ctx->vcpu_blocktime = g_new0(uint32_t, smp_cpus);
1552a4c42f1SAlexey Perevalov 
1562a4c42f1SAlexey Perevalov     ctx->exit_notifier.notify = migration_exit_cb;
1572a4c42f1SAlexey Perevalov     ctx->start_time = qemu_clock_get_ms(QEMU_CLOCK_REALTIME);
1582a4c42f1SAlexey Perevalov     qemu_add_exit_notifier(&ctx->exit_notifier);
1592a4c42f1SAlexey Perevalov     return ctx;
1602a4c42f1SAlexey Perevalov }
161ca6011c2SAlexey Perevalov 
16265ace060SAlexey Perevalov static uint32List *get_vcpu_blocktime_list(PostcopyBlocktimeContext *ctx)
16365ace060SAlexey Perevalov {
1645cc8767dSLike Xu     MachineState *ms = MACHINE(qdev_get_machine());
16554aa3de7SEric Blake     uint32List *list = NULL;
16665ace060SAlexey Perevalov     int i;
16765ace060SAlexey Perevalov 
1685cc8767dSLike Xu     for (i = ms->smp.cpus - 1; i >= 0; i--) {
16954aa3de7SEric Blake         QAPI_LIST_PREPEND(list, ctx->vcpu_blocktime[i]);
17065ace060SAlexey Perevalov     }
17165ace060SAlexey Perevalov 
17265ace060SAlexey Perevalov     return list;
17365ace060SAlexey Perevalov }
17465ace060SAlexey Perevalov 
17565ace060SAlexey Perevalov /*
17665ace060SAlexey Perevalov  * This function just populates MigrationInfo from postcopy's
17765ace060SAlexey Perevalov  * blocktime context. It will not populate MigrationInfo,
17865ace060SAlexey Perevalov  * unless postcopy-blocktime capability was set.
17965ace060SAlexey Perevalov  *
18065ace060SAlexey Perevalov  * @info: pointer to MigrationInfo to populate
18165ace060SAlexey Perevalov  */
18265ace060SAlexey Perevalov void fill_destination_postcopy_migration_info(MigrationInfo *info)
18365ace060SAlexey Perevalov {
18465ace060SAlexey Perevalov     MigrationIncomingState *mis = migration_incoming_get_current();
18565ace060SAlexey Perevalov     PostcopyBlocktimeContext *bc = mis->blocktime_ctx;
18665ace060SAlexey Perevalov 
18765ace060SAlexey Perevalov     if (!bc) {
18865ace060SAlexey Perevalov         return;
18965ace060SAlexey Perevalov     }
19065ace060SAlexey Perevalov 
19165ace060SAlexey Perevalov     info->has_postcopy_blocktime = true;
19265ace060SAlexey Perevalov     info->postcopy_blocktime = bc->total_blocktime;
19365ace060SAlexey Perevalov     info->has_postcopy_vcpu_blocktime = true;
19465ace060SAlexey Perevalov     info->postcopy_vcpu_blocktime = get_vcpu_blocktime_list(bc);
19565ace060SAlexey Perevalov }
19665ace060SAlexey Perevalov 
19765ace060SAlexey Perevalov static uint32_t get_postcopy_total_blocktime(void)
19865ace060SAlexey Perevalov {
19965ace060SAlexey Perevalov     MigrationIncomingState *mis = migration_incoming_get_current();
20065ace060SAlexey Perevalov     PostcopyBlocktimeContext *bc = mis->blocktime_ctx;
20165ace060SAlexey Perevalov 
20265ace060SAlexey Perevalov     if (!bc) {
20365ace060SAlexey Perevalov         return 0;
20465ace060SAlexey Perevalov     }
20565ace060SAlexey Perevalov 
20665ace060SAlexey Perevalov     return bc->total_blocktime;
20765ace060SAlexey Perevalov }
20865ace060SAlexey Perevalov 
20954ae0886SAlexey Perevalov /**
21054ae0886SAlexey Perevalov  * receive_ufd_features: check userfault fd features, to request only supported
21154ae0886SAlexey Perevalov  * features in the future.
21254ae0886SAlexey Perevalov  *
21354ae0886SAlexey Perevalov  * Returns: true on success
21454ae0886SAlexey Perevalov  *
21554ae0886SAlexey Perevalov  * __NR_userfaultfd - should be checked before
21654ae0886SAlexey Perevalov  *  @features: out parameter will contain uffdio_api.features provided by kernel
21754ae0886SAlexey Perevalov  *              in case of success
21854ae0886SAlexey Perevalov  */
21954ae0886SAlexey Perevalov static bool receive_ufd_features(uint64_t *features)
22054ae0886SAlexey Perevalov {
22154ae0886SAlexey Perevalov     struct uffdio_api api_struct = {0};
22254ae0886SAlexey Perevalov     int ufd;
22354ae0886SAlexey Perevalov     bool ret = true;
22454ae0886SAlexey Perevalov 
22554ae0886SAlexey Perevalov     /* if we are here __NR_userfaultfd should exists */
22654ae0886SAlexey Perevalov     ufd = syscall(__NR_userfaultfd, O_CLOEXEC);
22754ae0886SAlexey Perevalov     if (ufd == -1) {
22854ae0886SAlexey Perevalov         error_report("%s: syscall __NR_userfaultfd failed: %s", __func__,
22954ae0886SAlexey Perevalov                      strerror(errno));
23054ae0886SAlexey Perevalov         return false;
23154ae0886SAlexey Perevalov     }
23254ae0886SAlexey Perevalov 
23354ae0886SAlexey Perevalov     /* ask features */
234eb59db53SDr. David Alan Gilbert     api_struct.api = UFFD_API;
235eb59db53SDr. David Alan Gilbert     api_struct.features = 0;
236eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_API, &api_struct)) {
2375553499fSAlexey Perevalov         error_report("%s: UFFDIO_API failed: %s", __func__,
238eb59db53SDr. David Alan Gilbert                      strerror(errno));
23954ae0886SAlexey Perevalov         ret = false;
24054ae0886SAlexey Perevalov         goto release_ufd;
24154ae0886SAlexey Perevalov     }
24254ae0886SAlexey Perevalov 
24354ae0886SAlexey Perevalov     *features = api_struct.features;
24454ae0886SAlexey Perevalov 
24554ae0886SAlexey Perevalov release_ufd:
24654ae0886SAlexey Perevalov     close(ufd);
24754ae0886SAlexey Perevalov     return ret;
24854ae0886SAlexey Perevalov }
24954ae0886SAlexey Perevalov 
25054ae0886SAlexey Perevalov /**
25154ae0886SAlexey Perevalov  * request_ufd_features: this function should be called only once on a newly
25254ae0886SAlexey Perevalov  * opened ufd, subsequent calls will lead to error.
25354ae0886SAlexey Perevalov  *
2543a4452d8Szhaolichang  * Returns: true on success
25554ae0886SAlexey Perevalov  *
25654ae0886SAlexey Perevalov  * @ufd: fd obtained from userfaultfd syscall
25754ae0886SAlexey Perevalov  * @features: bit mask see UFFD_API_FEATURES
25854ae0886SAlexey Perevalov  */
25954ae0886SAlexey Perevalov static bool request_ufd_features(int ufd, uint64_t features)
26054ae0886SAlexey Perevalov {
26154ae0886SAlexey Perevalov     struct uffdio_api api_struct = {0};
26254ae0886SAlexey Perevalov     uint64_t ioctl_mask;
26354ae0886SAlexey Perevalov 
26454ae0886SAlexey Perevalov     api_struct.api = UFFD_API;
26554ae0886SAlexey Perevalov     api_struct.features = features;
26654ae0886SAlexey Perevalov     if (ioctl(ufd, UFFDIO_API, &api_struct)) {
26754ae0886SAlexey Perevalov         error_report("%s failed: UFFDIO_API failed: %s", __func__,
26854ae0886SAlexey Perevalov                      strerror(errno));
269eb59db53SDr. David Alan Gilbert         return false;
270eb59db53SDr. David Alan Gilbert     }
271eb59db53SDr. David Alan Gilbert 
272eb59db53SDr. David Alan Gilbert     ioctl_mask = (__u64)1 << _UFFDIO_REGISTER |
273eb59db53SDr. David Alan Gilbert                  (__u64)1 << _UFFDIO_UNREGISTER;
274eb59db53SDr. David Alan Gilbert     if ((api_struct.ioctls & ioctl_mask) != ioctl_mask) {
275eb59db53SDr. David Alan Gilbert         error_report("Missing userfault features: %" PRIx64,
276eb59db53SDr. David Alan Gilbert                      (uint64_t)(~api_struct.ioctls & ioctl_mask));
277eb59db53SDr. David Alan Gilbert         return false;
278eb59db53SDr. David Alan Gilbert     }
279eb59db53SDr. David Alan Gilbert 
28054ae0886SAlexey Perevalov     return true;
28154ae0886SAlexey Perevalov }
28254ae0886SAlexey Perevalov 
28354ae0886SAlexey Perevalov static bool ufd_check_and_apply(int ufd, MigrationIncomingState *mis)
28454ae0886SAlexey Perevalov {
28554ae0886SAlexey Perevalov     uint64_t asked_features = 0;
28654ae0886SAlexey Perevalov     static uint64_t supported_features;
28754ae0886SAlexey Perevalov 
28854ae0886SAlexey Perevalov     /*
28954ae0886SAlexey Perevalov      * it's not possible to
29054ae0886SAlexey Perevalov      * request UFFD_API twice per one fd
29154ae0886SAlexey Perevalov      * userfault fd features is persistent
29254ae0886SAlexey Perevalov      */
29354ae0886SAlexey Perevalov     if (!supported_features) {
29454ae0886SAlexey Perevalov         if (!receive_ufd_features(&supported_features)) {
29554ae0886SAlexey Perevalov             error_report("%s failed", __func__);
29654ae0886SAlexey Perevalov             return false;
29754ae0886SAlexey Perevalov         }
29854ae0886SAlexey Perevalov     }
29954ae0886SAlexey Perevalov 
3002a4c42f1SAlexey Perevalov #ifdef UFFD_FEATURE_THREAD_ID
3012d1c37c6SPeter Xu     if (UFFD_FEATURE_THREAD_ID & supported_features) {
3022d1c37c6SPeter Xu         asked_features |= UFFD_FEATURE_THREAD_ID;
3032d1c37c6SPeter Xu         if (migrate_postcopy_blocktime()) {
3042a4c42f1SAlexey Perevalov             if (!mis->blocktime_ctx) {
3052a4c42f1SAlexey Perevalov                 mis->blocktime_ctx = blocktime_context_new();
3062a4c42f1SAlexey Perevalov             }
3072d1c37c6SPeter Xu         }
3082a4c42f1SAlexey Perevalov     }
3092a4c42f1SAlexey Perevalov #endif
3102a4c42f1SAlexey Perevalov 
31154ae0886SAlexey Perevalov     /*
31254ae0886SAlexey Perevalov      * request features, even if asked_features is 0, due to
31354ae0886SAlexey Perevalov      * kernel expects UFFD_API before UFFDIO_REGISTER, per
31454ae0886SAlexey Perevalov      * userfault file descriptor
31554ae0886SAlexey Perevalov      */
31654ae0886SAlexey Perevalov     if (!request_ufd_features(ufd, asked_features)) {
31754ae0886SAlexey Perevalov         error_report("%s failed: features %" PRIu64, __func__,
31854ae0886SAlexey Perevalov                      asked_features);
31954ae0886SAlexey Perevalov         return false;
32054ae0886SAlexey Perevalov     }
32154ae0886SAlexey Perevalov 
322038adc2fSWei Yang     if (qemu_real_host_page_size != ram_pagesize_summary()) {
3237e8cafb7SDr. David Alan Gilbert         bool have_hp = false;
3247e8cafb7SDr. David Alan Gilbert         /* We've got a huge page */
3257e8cafb7SDr. David Alan Gilbert #ifdef UFFD_FEATURE_MISSING_HUGETLBFS
32654ae0886SAlexey Perevalov         have_hp = supported_features & UFFD_FEATURE_MISSING_HUGETLBFS;
3277e8cafb7SDr. David Alan Gilbert #endif
3287e8cafb7SDr. David Alan Gilbert         if (!have_hp) {
3297e8cafb7SDr. David Alan Gilbert             error_report("Userfault on this host does not support huge pages");
3307e8cafb7SDr. David Alan Gilbert             return false;
3317e8cafb7SDr. David Alan Gilbert         }
3327e8cafb7SDr. David Alan Gilbert     }
333eb59db53SDr. David Alan Gilbert     return true;
334eb59db53SDr. David Alan Gilbert }
335eb59db53SDr. David Alan Gilbert 
3368679638bSDr. David Alan Gilbert /* Callback from postcopy_ram_supported_by_host block iterator.
3378679638bSDr. David Alan Gilbert  */
338754cb9c0SYury Kotov static int test_ramblock_postcopiable(RAMBlock *rb, void *opaque)
3398679638bSDr. David Alan Gilbert {
340754cb9c0SYury Kotov     const char *block_name = qemu_ram_get_idstr(rb);
341754cb9c0SYury Kotov     ram_addr_t length = qemu_ram_get_used_length(rb);
3425d214a92SDr. David Alan Gilbert     size_t pagesize = qemu_ram_pagesize(rb);
3435d214a92SDr. David Alan Gilbert 
3445d214a92SDr. David Alan Gilbert     if (length % pagesize) {
3455d214a92SDr. David Alan Gilbert         error_report("Postcopy requires RAM blocks to be a page size multiple,"
3465d214a92SDr. David Alan Gilbert                      " block %s is 0x" RAM_ADDR_FMT " bytes with a "
3475d214a92SDr. David Alan Gilbert                      "page size of 0x%zx", block_name, length, pagesize);
3485d214a92SDr. David Alan Gilbert         return 1;
3495d214a92SDr. David Alan Gilbert     }
3508679638bSDr. David Alan Gilbert     return 0;
3518679638bSDr. David Alan Gilbert }
3528679638bSDr. David Alan Gilbert 
35358b7c17eSDr. David Alan Gilbert /*
35458b7c17eSDr. David Alan Gilbert  * Note: This has the side effect of munlock'ing all of RAM, that's
35558b7c17eSDr. David Alan Gilbert  * normally fine since if the postcopy succeeds it gets turned back on at the
35658b7c17eSDr. David Alan Gilbert  * end.
35758b7c17eSDr. David Alan Gilbert  */
358d7651f15SAlexey Perevalov bool postcopy_ram_supported_by_host(MigrationIncomingState *mis)
359eb59db53SDr. David Alan Gilbert {
360038adc2fSWei Yang     long pagesize = qemu_real_host_page_size;
361eb59db53SDr. David Alan Gilbert     int ufd = -1;
362eb59db53SDr. David Alan Gilbert     bool ret = false; /* Error unless we change it */
363eb59db53SDr. David Alan Gilbert     void *testarea = NULL;
364eb59db53SDr. David Alan Gilbert     struct uffdio_register reg_struct;
365eb59db53SDr. David Alan Gilbert     struct uffdio_range range_struct;
366eb59db53SDr. David Alan Gilbert     uint64_t feature_mask;
3671693c64cSDr. David Alan Gilbert     Error *local_err = NULL;
368eb59db53SDr. David Alan Gilbert 
36920afaed9SJuan Quintela     if (qemu_target_page_size() > pagesize) {
370eb59db53SDr. David Alan Gilbert         error_report("Target page size bigger than host page size");
371eb59db53SDr. David Alan Gilbert         goto out;
372eb59db53SDr. David Alan Gilbert     }
373eb59db53SDr. David Alan Gilbert 
374eb59db53SDr. David Alan Gilbert     ufd = syscall(__NR_userfaultfd, O_CLOEXEC);
375eb59db53SDr. David Alan Gilbert     if (ufd == -1) {
376eb59db53SDr. David Alan Gilbert         error_report("%s: userfaultfd not available: %s", __func__,
377eb59db53SDr. David Alan Gilbert                      strerror(errno));
378eb59db53SDr. David Alan Gilbert         goto out;
379eb59db53SDr. David Alan Gilbert     }
380eb59db53SDr. David Alan Gilbert 
3811693c64cSDr. David Alan Gilbert     /* Give devices a chance to object */
3821693c64cSDr. David Alan Gilbert     if (postcopy_notify(POSTCOPY_NOTIFY_PROBE, &local_err)) {
3831693c64cSDr. David Alan Gilbert         error_report_err(local_err);
3841693c64cSDr. David Alan Gilbert         goto out;
3851693c64cSDr. David Alan Gilbert     }
3861693c64cSDr. David Alan Gilbert 
387eb59db53SDr. David Alan Gilbert     /* Version and features check */
38854ae0886SAlexey Perevalov     if (!ufd_check_and_apply(ufd, mis)) {
389eb59db53SDr. David Alan Gilbert         goto out;
390eb59db53SDr. David Alan Gilbert     }
391eb59db53SDr. David Alan Gilbert 
3928679638bSDr. David Alan Gilbert     /* We don't support postcopy with shared RAM yet */
393fbd162e6SYury Kotov     if (foreach_not_ignored_block(test_ramblock_postcopiable, NULL)) {
3948679638bSDr. David Alan Gilbert         goto out;
3958679638bSDr. David Alan Gilbert     }
3968679638bSDr. David Alan Gilbert 
397eb59db53SDr. David Alan Gilbert     /*
39858b7c17eSDr. David Alan Gilbert      * userfault and mlock don't go together; we'll put it back later if
39958b7c17eSDr. David Alan Gilbert      * it was enabled.
40058b7c17eSDr. David Alan Gilbert      */
40158b7c17eSDr. David Alan Gilbert     if (munlockall()) {
40258b7c17eSDr. David Alan Gilbert         error_report("%s: munlockall: %s", __func__,  strerror(errno));
403617a32f5SDr. David Alan Gilbert         goto out;
40458b7c17eSDr. David Alan Gilbert     }
40558b7c17eSDr. David Alan Gilbert 
40658b7c17eSDr. David Alan Gilbert     /*
407eb59db53SDr. David Alan Gilbert      *  We need to check that the ops we need are supported on anon memory
408eb59db53SDr. David Alan Gilbert      *  To do that we need to register a chunk and see the flags that
409eb59db53SDr. David Alan Gilbert      *  are returned.
410eb59db53SDr. David Alan Gilbert      */
411eb59db53SDr. David Alan Gilbert     testarea = mmap(NULL, pagesize, PROT_READ | PROT_WRITE, MAP_PRIVATE |
412eb59db53SDr. David Alan Gilbert                                     MAP_ANONYMOUS, -1, 0);
413eb59db53SDr. David Alan Gilbert     if (testarea == MAP_FAILED) {
414eb59db53SDr. David Alan Gilbert         error_report("%s: Failed to map test area: %s", __func__,
415eb59db53SDr. David Alan Gilbert                      strerror(errno));
416eb59db53SDr. David Alan Gilbert         goto out;
417eb59db53SDr. David Alan Gilbert     }
4187648297dSDavid Hildenbrand     g_assert(QEMU_PTR_IS_ALIGNED(testarea, pagesize));
419eb59db53SDr. David Alan Gilbert 
420eb59db53SDr. David Alan Gilbert     reg_struct.range.start = (uintptr_t)testarea;
421eb59db53SDr. David Alan Gilbert     reg_struct.range.len = pagesize;
422eb59db53SDr. David Alan Gilbert     reg_struct.mode = UFFDIO_REGISTER_MODE_MISSING;
423eb59db53SDr. David Alan Gilbert 
424eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_REGISTER, &reg_struct)) {
425eb59db53SDr. David Alan Gilbert         error_report("%s userfault register: %s", __func__, strerror(errno));
426eb59db53SDr. David Alan Gilbert         goto out;
427eb59db53SDr. David Alan Gilbert     }
428eb59db53SDr. David Alan Gilbert 
429eb59db53SDr. David Alan Gilbert     range_struct.start = (uintptr_t)testarea;
430eb59db53SDr. David Alan Gilbert     range_struct.len = pagesize;
431eb59db53SDr. David Alan Gilbert     if (ioctl(ufd, UFFDIO_UNREGISTER, &range_struct)) {
432eb59db53SDr. David Alan Gilbert         error_report("%s userfault unregister: %s", __func__, strerror(errno));
433eb59db53SDr. David Alan Gilbert         goto out;
434eb59db53SDr. David Alan Gilbert     }
435eb59db53SDr. David Alan Gilbert 
436eb59db53SDr. David Alan Gilbert     feature_mask = (__u64)1 << _UFFDIO_WAKE |
437eb59db53SDr. David Alan Gilbert                    (__u64)1 << _UFFDIO_COPY |
438eb59db53SDr. David Alan Gilbert                    (__u64)1 << _UFFDIO_ZEROPAGE;
439eb59db53SDr. David Alan Gilbert     if ((reg_struct.ioctls & feature_mask) != feature_mask) {
440eb59db53SDr. David Alan Gilbert         error_report("Missing userfault map features: %" PRIx64,
441eb59db53SDr. David Alan Gilbert                      (uint64_t)(~reg_struct.ioctls & feature_mask));
442eb59db53SDr. David Alan Gilbert         goto out;
443eb59db53SDr. David Alan Gilbert     }
444eb59db53SDr. David Alan Gilbert 
445eb59db53SDr. David Alan Gilbert     /* Success! */
446eb59db53SDr. David Alan Gilbert     ret = true;
447eb59db53SDr. David Alan Gilbert out:
448eb59db53SDr. David Alan Gilbert     if (testarea) {
449eb59db53SDr. David Alan Gilbert         munmap(testarea, pagesize);
450eb59db53SDr. David Alan Gilbert     }
451eb59db53SDr. David Alan Gilbert     if (ufd != -1) {
452eb59db53SDr. David Alan Gilbert         close(ufd);
453eb59db53SDr. David Alan Gilbert     }
454eb59db53SDr. David Alan Gilbert     return ret;
455eb59db53SDr. David Alan Gilbert }
456eb59db53SDr. David Alan Gilbert 
4571caddf8aSDr. David Alan Gilbert /*
4581caddf8aSDr. David Alan Gilbert  * Setup an area of RAM so that it *can* be used for postcopy later; this
4591caddf8aSDr. David Alan Gilbert  * must be done right at the start prior to pre-copy.
4601caddf8aSDr. David Alan Gilbert  * opaque should be the MIS.
4611caddf8aSDr. David Alan Gilbert  */
462754cb9c0SYury Kotov static int init_range(RAMBlock *rb, void *opaque)
4631caddf8aSDr. David Alan Gilbert {
464754cb9c0SYury Kotov     const char *block_name = qemu_ram_get_idstr(rb);
465754cb9c0SYury Kotov     void *host_addr = qemu_ram_get_host_addr(rb);
466754cb9c0SYury Kotov     ram_addr_t offset = qemu_ram_get_offset(rb);
467754cb9c0SYury Kotov     ram_addr_t length = qemu_ram_get_used_length(rb);
4681caddf8aSDr. David Alan Gilbert     trace_postcopy_init_range(block_name, host_addr, offset, length);
4691caddf8aSDr. David Alan Gilbert 
4701caddf8aSDr. David Alan Gilbert     /*
471898ba906SDavid Hildenbrand      * Save the used_length before running the guest. In case we have to
472898ba906SDavid Hildenbrand      * resize RAM blocks when syncing RAM block sizes from the source during
473898ba906SDavid Hildenbrand      * precopy, we'll update it manually via the ram block notifier.
474898ba906SDavid Hildenbrand      */
475898ba906SDavid Hildenbrand     rb->postcopy_length = length;
476898ba906SDavid Hildenbrand 
477898ba906SDavid Hildenbrand     /*
4781caddf8aSDr. David Alan Gilbert      * We need the whole of RAM to be truly empty for postcopy, so things
4791caddf8aSDr. David Alan Gilbert      * like ROMs and any data tables built during init must be zero'd
4801caddf8aSDr. David Alan Gilbert      * - we're going to get the copy from the source anyway.
4811caddf8aSDr. David Alan Gilbert      * (Precopy will just overwrite this data, so doesn't need the discard)
4821caddf8aSDr. David Alan Gilbert      */
483aaa2064cSJuan Quintela     if (ram_discard_range(block_name, 0, length)) {
4841caddf8aSDr. David Alan Gilbert         return -1;
4851caddf8aSDr. David Alan Gilbert     }
4861caddf8aSDr. David Alan Gilbert 
4871caddf8aSDr. David Alan Gilbert     return 0;
4881caddf8aSDr. David Alan Gilbert }
4891caddf8aSDr. David Alan Gilbert 
4901caddf8aSDr. David Alan Gilbert /*
4911caddf8aSDr. David Alan Gilbert  * At the end of migration, undo the effects of init_range
4921caddf8aSDr. David Alan Gilbert  * opaque should be the MIS.
4931caddf8aSDr. David Alan Gilbert  */
494754cb9c0SYury Kotov static int cleanup_range(RAMBlock *rb, void *opaque)
4951caddf8aSDr. David Alan Gilbert {
496754cb9c0SYury Kotov     const char *block_name = qemu_ram_get_idstr(rb);
497754cb9c0SYury Kotov     void *host_addr = qemu_ram_get_host_addr(rb);
498754cb9c0SYury Kotov     ram_addr_t offset = qemu_ram_get_offset(rb);
499898ba906SDavid Hildenbrand     ram_addr_t length = rb->postcopy_length;
5001caddf8aSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
5011caddf8aSDr. David Alan Gilbert     struct uffdio_range range_struct;
5021caddf8aSDr. David Alan Gilbert     trace_postcopy_cleanup_range(block_name, host_addr, offset, length);
5031caddf8aSDr. David Alan Gilbert 
5041caddf8aSDr. David Alan Gilbert     /*
5051caddf8aSDr. David Alan Gilbert      * We turned off hugepage for the precopy stage with postcopy enabled
5061caddf8aSDr. David Alan Gilbert      * we can turn it back on now.
5071caddf8aSDr. David Alan Gilbert      */
5081d741439SDr. David Alan Gilbert     qemu_madvise(host_addr, length, QEMU_MADV_HUGEPAGE);
5091caddf8aSDr. David Alan Gilbert 
5101caddf8aSDr. David Alan Gilbert     /*
5111caddf8aSDr. David Alan Gilbert      * We can also turn off userfault now since we should have all the
5121caddf8aSDr. David Alan Gilbert      * pages.   It can be useful to leave it on to debug postcopy
5131caddf8aSDr. David Alan Gilbert      * if you're not sure it's always getting every page.
5141caddf8aSDr. David Alan Gilbert      */
5151caddf8aSDr. David Alan Gilbert     range_struct.start = (uintptr_t)host_addr;
5161caddf8aSDr. David Alan Gilbert     range_struct.len = length;
5171caddf8aSDr. David Alan Gilbert 
5181caddf8aSDr. David Alan Gilbert     if (ioctl(mis->userfault_fd, UFFDIO_UNREGISTER, &range_struct)) {
5191caddf8aSDr. David Alan Gilbert         error_report("%s: userfault unregister %s", __func__, strerror(errno));
5201caddf8aSDr. David Alan Gilbert 
5211caddf8aSDr. David Alan Gilbert         return -1;
5221caddf8aSDr. David Alan Gilbert     }
5231caddf8aSDr. David Alan Gilbert 
5241caddf8aSDr. David Alan Gilbert     return 0;
5251caddf8aSDr. David Alan Gilbert }
5261caddf8aSDr. David Alan Gilbert 
5271caddf8aSDr. David Alan Gilbert /*
5281caddf8aSDr. David Alan Gilbert  * Initialise postcopy-ram, setting the RAM to a state where we can go into
5291caddf8aSDr. David Alan Gilbert  * postcopy later; must be called prior to any precopy.
5301caddf8aSDr. David Alan Gilbert  * called from arch_init's similarly named ram_postcopy_incoming_init
5311caddf8aSDr. David Alan Gilbert  */
532c136180cSDavid Hildenbrand int postcopy_ram_incoming_init(MigrationIncomingState *mis)
5331caddf8aSDr. David Alan Gilbert {
534fbd162e6SYury Kotov     if (foreach_not_ignored_block(init_range, NULL)) {
5351caddf8aSDr. David Alan Gilbert         return -1;
5361caddf8aSDr. David Alan Gilbert     }
5371caddf8aSDr. David Alan Gilbert 
5381caddf8aSDr. David Alan Gilbert     return 0;
5391caddf8aSDr. David Alan Gilbert }
5401caddf8aSDr. David Alan Gilbert 
541476ebf77SPeter Xu static void postcopy_temp_pages_cleanup(MigrationIncomingState *mis)
542476ebf77SPeter Xu {
54377dadc3fSPeter Xu     int i;
54477dadc3fSPeter Xu 
54577dadc3fSPeter Xu     if (mis->postcopy_tmp_pages) {
54677dadc3fSPeter Xu         for (i = 0; i < mis->postcopy_channels; i++) {
54777dadc3fSPeter Xu             if (mis->postcopy_tmp_pages[i].tmp_huge_page) {
54877dadc3fSPeter Xu                 munmap(mis->postcopy_tmp_pages[i].tmp_huge_page,
54977dadc3fSPeter Xu                        mis->largest_page_size);
55077dadc3fSPeter Xu                 mis->postcopy_tmp_pages[i].tmp_huge_page = NULL;
55177dadc3fSPeter Xu             }
55277dadc3fSPeter Xu         }
55377dadc3fSPeter Xu         g_free(mis->postcopy_tmp_pages);
55477dadc3fSPeter Xu         mis->postcopy_tmp_pages = NULL;
555476ebf77SPeter Xu     }
556476ebf77SPeter Xu 
557476ebf77SPeter Xu     if (mis->postcopy_tmp_zero_page) {
558476ebf77SPeter Xu         munmap(mis->postcopy_tmp_zero_page, mis->largest_page_size);
559476ebf77SPeter Xu         mis->postcopy_tmp_zero_page = NULL;
560476ebf77SPeter Xu     }
561476ebf77SPeter Xu }
562476ebf77SPeter Xu 
5631caddf8aSDr. David Alan Gilbert /*
5641caddf8aSDr. David Alan Gilbert  * At the end of a migration where postcopy_ram_incoming_init was called.
5651caddf8aSDr. David Alan Gilbert  */
5661caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_cleanup(MigrationIncomingState *mis)
5671caddf8aSDr. David Alan Gilbert {
568c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_incoming_cleanup_entry();
569c4faeed2SDr. David Alan Gilbert 
570c4faeed2SDr. David Alan Gilbert     if (mis->have_fault_thread) {
57146343570SDr. David Alan Gilbert         Error *local_err = NULL;
57246343570SDr. David Alan Gilbert 
57355d0fe82SIlya Maximets         /* Let the fault thread quit */
574d73415a3SStefan Hajnoczi         qatomic_set(&mis->fault_thread_quit, 1);
57555d0fe82SIlya Maximets         postcopy_fault_thread_notify(mis);
57655d0fe82SIlya Maximets         trace_postcopy_ram_incoming_cleanup_join();
57755d0fe82SIlya Maximets         qemu_thread_join(&mis->fault_thread);
57855d0fe82SIlya Maximets 
57946343570SDr. David Alan Gilbert         if (postcopy_notify(POSTCOPY_NOTIFY_INBOUND_END, &local_err)) {
58046343570SDr. David Alan Gilbert             error_report_err(local_err);
58146343570SDr. David Alan Gilbert             return -1;
58246343570SDr. David Alan Gilbert         }
58346343570SDr. David Alan Gilbert 
584fbd162e6SYury Kotov         if (foreach_not_ignored_block(cleanup_range, mis)) {
5851caddf8aSDr. David Alan Gilbert             return -1;
5861caddf8aSDr. David Alan Gilbert         }
5879ab7ef9bSPeter Xu 
588c4faeed2SDr. David Alan Gilbert         trace_postcopy_ram_incoming_cleanup_closeuf();
589c4faeed2SDr. David Alan Gilbert         close(mis->userfault_fd);
59064f615feSPeter Xu         close(mis->userfault_event_fd);
591c4faeed2SDr. David Alan Gilbert         mis->have_fault_thread = false;
592c4faeed2SDr. David Alan Gilbert     }
593c4faeed2SDr. David Alan Gilbert 
59458b7c17eSDr. David Alan Gilbert     if (enable_mlock) {
59558b7c17eSDr. David Alan Gilbert         if (os_mlock() < 0) {
59658b7c17eSDr. David Alan Gilbert             error_report("mlock: %s", strerror(errno));
59758b7c17eSDr. David Alan Gilbert             /*
59858b7c17eSDr. David Alan Gilbert              * It doesn't feel right to fail at this point, we have a valid
59958b7c17eSDr. David Alan Gilbert              * VM state.
60058b7c17eSDr. David Alan Gilbert              */
60158b7c17eSDr. David Alan Gilbert         }
60258b7c17eSDr. David Alan Gilbert     }
60358b7c17eSDr. David Alan Gilbert 
604476ebf77SPeter Xu     postcopy_temp_pages_cleanup(mis);
605476ebf77SPeter Xu 
60665ace060SAlexey Perevalov     trace_postcopy_ram_incoming_cleanup_blocktime(
60765ace060SAlexey Perevalov             get_postcopy_total_blocktime());
60865ace060SAlexey Perevalov 
609c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_incoming_cleanup_exit();
6101caddf8aSDr. David Alan Gilbert     return 0;
6111caddf8aSDr. David Alan Gilbert }
6121caddf8aSDr. David Alan Gilbert 
613f0a227adSDr. David Alan Gilbert /*
614f9527107SDr. David Alan Gilbert  * Disable huge pages on an area
615f9527107SDr. David Alan Gilbert  */
616754cb9c0SYury Kotov static int nhp_range(RAMBlock *rb, void *opaque)
617f9527107SDr. David Alan Gilbert {
618754cb9c0SYury Kotov     const char *block_name = qemu_ram_get_idstr(rb);
619754cb9c0SYury Kotov     void *host_addr = qemu_ram_get_host_addr(rb);
620754cb9c0SYury Kotov     ram_addr_t offset = qemu_ram_get_offset(rb);
621898ba906SDavid Hildenbrand     ram_addr_t length = rb->postcopy_length;
622f9527107SDr. David Alan Gilbert     trace_postcopy_nhp_range(block_name, host_addr, offset, length);
623f9527107SDr. David Alan Gilbert 
624f9527107SDr. David Alan Gilbert     /*
625f9527107SDr. David Alan Gilbert      * Before we do discards we need to ensure those discards really
626f9527107SDr. David Alan Gilbert      * do delete areas of the page, even if THP thinks a hugepage would
627f9527107SDr. David Alan Gilbert      * be a good idea, so force hugepages off.
628f9527107SDr. David Alan Gilbert      */
6291d741439SDr. David Alan Gilbert     qemu_madvise(host_addr, length, QEMU_MADV_NOHUGEPAGE);
630f9527107SDr. David Alan Gilbert 
631f9527107SDr. David Alan Gilbert     return 0;
632f9527107SDr. David Alan Gilbert }
633f9527107SDr. David Alan Gilbert 
634f9527107SDr. David Alan Gilbert /*
635f9527107SDr. David Alan Gilbert  * Userfault requires us to mark RAM as NOHUGEPAGE prior to discard
636f9527107SDr. David Alan Gilbert  * however leaving it until after precopy means that most of the precopy
637f9527107SDr. David Alan Gilbert  * data is still THPd
638f9527107SDr. David Alan Gilbert  */
639f9527107SDr. David Alan Gilbert int postcopy_ram_prepare_discard(MigrationIncomingState *mis)
640f9527107SDr. David Alan Gilbert {
641fbd162e6SYury Kotov     if (foreach_not_ignored_block(nhp_range, mis)) {
642f9527107SDr. David Alan Gilbert         return -1;
643f9527107SDr. David Alan Gilbert     }
644f9527107SDr. David Alan Gilbert 
645f9527107SDr. David Alan Gilbert     postcopy_state_set(POSTCOPY_INCOMING_DISCARD);
646f9527107SDr. David Alan Gilbert 
647f9527107SDr. David Alan Gilbert     return 0;
648f9527107SDr. David Alan Gilbert }
649f9527107SDr. David Alan Gilbert 
650f9527107SDr. David Alan Gilbert /*
651f0a227adSDr. David Alan Gilbert  * Mark the given area of RAM as requiring notification to unwritten areas
652fbd162e6SYury Kotov  * Used as a  callback on foreach_not_ignored_block.
653f0a227adSDr. David Alan Gilbert  *   host_addr: Base of area to mark
654f0a227adSDr. David Alan Gilbert  *   offset: Offset in the whole ram arena
655f0a227adSDr. David Alan Gilbert  *   length: Length of the section
656f0a227adSDr. David Alan Gilbert  *   opaque: MigrationIncomingState pointer
657f0a227adSDr. David Alan Gilbert  * Returns 0 on success
658f0a227adSDr. David Alan Gilbert  */
659754cb9c0SYury Kotov static int ram_block_enable_notify(RAMBlock *rb, void *opaque)
660f0a227adSDr. David Alan Gilbert {
661f0a227adSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
662f0a227adSDr. David Alan Gilbert     struct uffdio_register reg_struct;
663f0a227adSDr. David Alan Gilbert 
664754cb9c0SYury Kotov     reg_struct.range.start = (uintptr_t)qemu_ram_get_host_addr(rb);
665898ba906SDavid Hildenbrand     reg_struct.range.len = rb->postcopy_length;
666f0a227adSDr. David Alan Gilbert     reg_struct.mode = UFFDIO_REGISTER_MODE_MISSING;
667f0a227adSDr. David Alan Gilbert 
668f0a227adSDr. David Alan Gilbert     /* Now tell our userfault_fd that it's responsible for this area */
669f0a227adSDr. David Alan Gilbert     if (ioctl(mis->userfault_fd, UFFDIO_REGISTER, &reg_struct)) {
670f0a227adSDr. David Alan Gilbert         error_report("%s userfault register: %s", __func__, strerror(errno));
671f0a227adSDr. David Alan Gilbert         return -1;
672f0a227adSDr. David Alan Gilbert     }
673665414adSDr. David Alan Gilbert     if (!(reg_struct.ioctls & ((__u64)1 << _UFFDIO_COPY))) {
674665414adSDr. David Alan Gilbert         error_report("%s userfault: Region doesn't support COPY", __func__);
675665414adSDr. David Alan Gilbert         return -1;
676665414adSDr. David Alan Gilbert     }
6772ce16640SDr. David Alan Gilbert     if (reg_struct.ioctls & ((__u64)1 << _UFFDIO_ZEROPAGE)) {
6782ce16640SDr. David Alan Gilbert         qemu_ram_set_uf_zeroable(rb);
6792ce16640SDr. David Alan Gilbert     }
680f0a227adSDr. David Alan Gilbert 
681f0a227adSDr. David Alan Gilbert     return 0;
682f0a227adSDr. David Alan Gilbert }
683f0a227adSDr. David Alan Gilbert 
6845efc3564SDr. David Alan Gilbert int postcopy_wake_shared(struct PostCopyFD *pcfd,
6855efc3564SDr. David Alan Gilbert                          uint64_t client_addr,
6865efc3564SDr. David Alan Gilbert                          RAMBlock *rb)
6875efc3564SDr. David Alan Gilbert {
6885efc3564SDr. David Alan Gilbert     size_t pagesize = qemu_ram_pagesize(rb);
6895efc3564SDr. David Alan Gilbert     struct uffdio_range range;
6905efc3564SDr. David Alan Gilbert     int ret;
6915efc3564SDr. David Alan Gilbert     trace_postcopy_wake_shared(client_addr, qemu_ram_get_idstr(rb));
6927648297dSDavid Hildenbrand     range.start = ROUND_DOWN(client_addr, pagesize);
6935efc3564SDr. David Alan Gilbert     range.len = pagesize;
6945efc3564SDr. David Alan Gilbert     ret = ioctl(pcfd->fd, UFFDIO_WAKE, &range);
6955efc3564SDr. David Alan Gilbert     if (ret) {
6965efc3564SDr. David Alan Gilbert         error_report("%s: Failed to wake: %zx in %s (%s)",
6975efc3564SDr. David Alan Gilbert                      __func__, (size_t)client_addr, qemu_ram_get_idstr(rb),
6985efc3564SDr. David Alan Gilbert                      strerror(errno));
6995efc3564SDr. David Alan Gilbert     }
7005efc3564SDr. David Alan Gilbert     return ret;
7015efc3564SDr. David Alan Gilbert }
7025efc3564SDr. David Alan Gilbert 
7039470c5e0SDavid Hildenbrand static int postcopy_request_page(MigrationIncomingState *mis, RAMBlock *rb,
7049470c5e0SDavid Hildenbrand                                  ram_addr_t start, uint64_t haddr)
7059470c5e0SDavid Hildenbrand {
7069470c5e0SDavid Hildenbrand     void *aligned = (void *)(uintptr_t)ROUND_DOWN(haddr, qemu_ram_pagesize(rb));
7079470c5e0SDavid Hildenbrand 
7089470c5e0SDavid Hildenbrand     /*
7099470c5e0SDavid Hildenbrand      * Discarded pages (via RamDiscardManager) are never migrated. On unlikely
7109470c5e0SDavid Hildenbrand      * access, place a zeropage, which will also set the relevant bits in the
7119470c5e0SDavid Hildenbrand      * recv_bitmap accordingly, so we won't try placing a zeropage twice.
7129470c5e0SDavid Hildenbrand      *
7139470c5e0SDavid Hildenbrand      * Checking a single bit is sufficient to handle pagesize > TPS as either
7149470c5e0SDavid Hildenbrand      * all relevant bits are set or not.
7159470c5e0SDavid Hildenbrand      */
7169470c5e0SDavid Hildenbrand     assert(QEMU_IS_ALIGNED(start, qemu_ram_pagesize(rb)));
7179470c5e0SDavid Hildenbrand     if (ramblock_page_is_discarded(rb, start)) {
7189470c5e0SDavid Hildenbrand         bool received = ramblock_recv_bitmap_test_byte_offset(rb, start);
7199470c5e0SDavid Hildenbrand 
7209470c5e0SDavid Hildenbrand         return received ? 0 : postcopy_place_page_zero(mis, aligned, rb);
7219470c5e0SDavid Hildenbrand     }
7229470c5e0SDavid Hildenbrand 
7239470c5e0SDavid Hildenbrand     return migrate_send_rp_req_pages(mis, rb, start, haddr);
7249470c5e0SDavid Hildenbrand }
7259470c5e0SDavid Hildenbrand 
726f0a227adSDr. David Alan Gilbert /*
727096bf4c8SDr. David Alan Gilbert  * Callback from shared fault handlers to ask for a page,
728096bf4c8SDr. David Alan Gilbert  * the page must be specified by a RAMBlock and an offset in that rb
729096bf4c8SDr. David Alan Gilbert  * Note: Only for use by shared fault handlers (in fault thread)
730096bf4c8SDr. David Alan Gilbert  */
731096bf4c8SDr. David Alan Gilbert int postcopy_request_shared_page(struct PostCopyFD *pcfd, RAMBlock *rb,
732096bf4c8SDr. David Alan Gilbert                                  uint64_t client_addr, uint64_t rb_offset)
733096bf4c8SDr. David Alan Gilbert {
7347648297dSDavid Hildenbrand     uint64_t aligned_rbo = ROUND_DOWN(rb_offset, qemu_ram_pagesize(rb));
735096bf4c8SDr. David Alan Gilbert     MigrationIncomingState *mis = migration_incoming_get_current();
736096bf4c8SDr. David Alan Gilbert 
737096bf4c8SDr. David Alan Gilbert     trace_postcopy_request_shared_page(pcfd->idstr, qemu_ram_get_idstr(rb),
738096bf4c8SDr. David Alan Gilbert                                        rb_offset);
739dedfb4b2SDr. David Alan Gilbert     if (ramblock_recv_bitmap_test_byte_offset(rb, aligned_rbo)) {
740dedfb4b2SDr. David Alan Gilbert         trace_postcopy_request_shared_page_present(pcfd->idstr,
741dedfb4b2SDr. David Alan Gilbert                                         qemu_ram_get_idstr(rb), rb_offset);
742dedfb4b2SDr. David Alan Gilbert         return postcopy_wake_shared(pcfd, client_addr, rb);
743dedfb4b2SDr. David Alan Gilbert     }
7449470c5e0SDavid Hildenbrand     postcopy_request_page(mis, rb, aligned_rbo, client_addr);
745096bf4c8SDr. David Alan Gilbert     return 0;
746096bf4c8SDr. David Alan Gilbert }
747096bf4c8SDr. David Alan Gilbert 
748575b0b33SAlexey Perevalov static int get_mem_fault_cpu_index(uint32_t pid)
749575b0b33SAlexey Perevalov {
750575b0b33SAlexey Perevalov     CPUState *cpu_iter;
751575b0b33SAlexey Perevalov 
752575b0b33SAlexey Perevalov     CPU_FOREACH(cpu_iter) {
753575b0b33SAlexey Perevalov         if (cpu_iter->thread_id == pid) {
754575b0b33SAlexey Perevalov             trace_get_mem_fault_cpu_index(cpu_iter->cpu_index, pid);
755575b0b33SAlexey Perevalov             return cpu_iter->cpu_index;
756575b0b33SAlexey Perevalov         }
757575b0b33SAlexey Perevalov     }
758575b0b33SAlexey Perevalov     trace_get_mem_fault_cpu_index(-1, pid);
759575b0b33SAlexey Perevalov     return -1;
760575b0b33SAlexey Perevalov }
761575b0b33SAlexey Perevalov 
762575b0b33SAlexey Perevalov static uint32_t get_low_time_offset(PostcopyBlocktimeContext *dc)
763575b0b33SAlexey Perevalov {
764575b0b33SAlexey Perevalov     int64_t start_time_offset = qemu_clock_get_ms(QEMU_CLOCK_REALTIME) -
765575b0b33SAlexey Perevalov                                     dc->start_time;
766575b0b33SAlexey Perevalov     return start_time_offset < 1 ? 1 : start_time_offset & UINT32_MAX;
767575b0b33SAlexey Perevalov }
768575b0b33SAlexey Perevalov 
769575b0b33SAlexey Perevalov /*
770575b0b33SAlexey Perevalov  * This function is being called when pagefault occurs. It
771575b0b33SAlexey Perevalov  * tracks down vCPU blocking time.
772575b0b33SAlexey Perevalov  *
773575b0b33SAlexey Perevalov  * @addr: faulted host virtual address
774575b0b33SAlexey Perevalov  * @ptid: faulted process thread id
775575b0b33SAlexey Perevalov  * @rb: ramblock appropriate to addr
776575b0b33SAlexey Perevalov  */
777575b0b33SAlexey Perevalov static void mark_postcopy_blocktime_begin(uintptr_t addr, uint32_t ptid,
778575b0b33SAlexey Perevalov                                           RAMBlock *rb)
779575b0b33SAlexey Perevalov {
780575b0b33SAlexey Perevalov     int cpu, already_received;
781575b0b33SAlexey Perevalov     MigrationIncomingState *mis = migration_incoming_get_current();
782575b0b33SAlexey Perevalov     PostcopyBlocktimeContext *dc = mis->blocktime_ctx;
783575b0b33SAlexey Perevalov     uint32_t low_time_offset;
784575b0b33SAlexey Perevalov 
785575b0b33SAlexey Perevalov     if (!dc || ptid == 0) {
786575b0b33SAlexey Perevalov         return;
787575b0b33SAlexey Perevalov     }
788575b0b33SAlexey Perevalov     cpu = get_mem_fault_cpu_index(ptid);
789575b0b33SAlexey Perevalov     if (cpu < 0) {
790575b0b33SAlexey Perevalov         return;
791575b0b33SAlexey Perevalov     }
792575b0b33SAlexey Perevalov 
793575b0b33SAlexey Perevalov     low_time_offset = get_low_time_offset(dc);
794575b0b33SAlexey Perevalov     if (dc->vcpu_addr[cpu] == 0) {
795d73415a3SStefan Hajnoczi         qatomic_inc(&dc->smp_cpus_down);
796575b0b33SAlexey Perevalov     }
797575b0b33SAlexey Perevalov 
798d73415a3SStefan Hajnoczi     qatomic_xchg(&dc->last_begin, low_time_offset);
799d73415a3SStefan Hajnoczi     qatomic_xchg(&dc->page_fault_vcpu_time[cpu], low_time_offset);
800d73415a3SStefan Hajnoczi     qatomic_xchg(&dc->vcpu_addr[cpu], addr);
801575b0b33SAlexey Perevalov 
802da1725d3SWei Yang     /*
803da1725d3SWei Yang      * check it here, not at the beginning of the function,
804da1725d3SWei Yang      * due to, check could occur early than bitmap_set in
805da1725d3SWei Yang      * qemu_ufd_copy_ioctl
806da1725d3SWei Yang      */
807575b0b33SAlexey Perevalov     already_received = ramblock_recv_bitmap_test(rb, (void *)addr);
808575b0b33SAlexey Perevalov     if (already_received) {
809d73415a3SStefan Hajnoczi         qatomic_xchg(&dc->vcpu_addr[cpu], 0);
810d73415a3SStefan Hajnoczi         qatomic_xchg(&dc->page_fault_vcpu_time[cpu], 0);
811d73415a3SStefan Hajnoczi         qatomic_dec(&dc->smp_cpus_down);
812575b0b33SAlexey Perevalov     }
813575b0b33SAlexey Perevalov     trace_mark_postcopy_blocktime_begin(addr, dc, dc->page_fault_vcpu_time[cpu],
814575b0b33SAlexey Perevalov                                         cpu, already_received);
815575b0b33SAlexey Perevalov }
816575b0b33SAlexey Perevalov 
817575b0b33SAlexey Perevalov /*
818575b0b33SAlexey Perevalov  *  This function just provide calculated blocktime per cpu and trace it.
819575b0b33SAlexey Perevalov  *  Total blocktime is calculated in mark_postcopy_blocktime_end.
820575b0b33SAlexey Perevalov  *
821575b0b33SAlexey Perevalov  *
822575b0b33SAlexey Perevalov  * Assume we have 3 CPU
823575b0b33SAlexey Perevalov  *
824575b0b33SAlexey Perevalov  *      S1        E1           S1               E1
825575b0b33SAlexey Perevalov  * -----***********------------xxx***************------------------------> CPU1
826575b0b33SAlexey Perevalov  *
827575b0b33SAlexey Perevalov  *             S2                E2
828575b0b33SAlexey Perevalov  * ------------****************xxx---------------------------------------> CPU2
829575b0b33SAlexey Perevalov  *
830575b0b33SAlexey Perevalov  *                         S3            E3
831575b0b33SAlexey Perevalov  * ------------------------****xxx********-------------------------------> CPU3
832575b0b33SAlexey Perevalov  *
833575b0b33SAlexey Perevalov  * We have sequence S1,S2,E1,S3,S1,E2,E3,E1
834575b0b33SAlexey Perevalov  * S2,E1 - doesn't match condition due to sequence S1,S2,E1 doesn't include CPU3
835575b0b33SAlexey Perevalov  * S3,S1,E2 - sequence includes all CPUs, in this case overlap will be S1,E2 -
836575b0b33SAlexey Perevalov  *            it's a part of total blocktime.
837575b0b33SAlexey Perevalov  * S1 - here is last_begin
838575b0b33SAlexey Perevalov  * Legend of the picture is following:
839575b0b33SAlexey Perevalov  *              * - means blocktime per vCPU
840575b0b33SAlexey Perevalov  *              x - means overlapped blocktime (total blocktime)
841575b0b33SAlexey Perevalov  *
842575b0b33SAlexey Perevalov  * @addr: host virtual address
843575b0b33SAlexey Perevalov  */
844575b0b33SAlexey Perevalov static void mark_postcopy_blocktime_end(uintptr_t addr)
845575b0b33SAlexey Perevalov {
846575b0b33SAlexey Perevalov     MigrationIncomingState *mis = migration_incoming_get_current();
847575b0b33SAlexey Perevalov     PostcopyBlocktimeContext *dc = mis->blocktime_ctx;
8485cc8767dSLike Xu     MachineState *ms = MACHINE(qdev_get_machine());
8495cc8767dSLike Xu     unsigned int smp_cpus = ms->smp.cpus;
850575b0b33SAlexey Perevalov     int i, affected_cpu = 0;
851575b0b33SAlexey Perevalov     bool vcpu_total_blocktime = false;
852575b0b33SAlexey Perevalov     uint32_t read_vcpu_time, low_time_offset;
853575b0b33SAlexey Perevalov 
854575b0b33SAlexey Perevalov     if (!dc) {
855575b0b33SAlexey Perevalov         return;
856575b0b33SAlexey Perevalov     }
857575b0b33SAlexey Perevalov 
858575b0b33SAlexey Perevalov     low_time_offset = get_low_time_offset(dc);
859575b0b33SAlexey Perevalov     /* lookup cpu, to clear it,
8603a4452d8Szhaolichang      * that algorithm looks straightforward, but it's not
861575b0b33SAlexey Perevalov      * optimal, more optimal algorithm is keeping tree or hash
862575b0b33SAlexey Perevalov      * where key is address value is a list of  */
863575b0b33SAlexey Perevalov     for (i = 0; i < smp_cpus; i++) {
864575b0b33SAlexey Perevalov         uint32_t vcpu_blocktime = 0;
865575b0b33SAlexey Perevalov 
866d73415a3SStefan Hajnoczi         read_vcpu_time = qatomic_fetch_add(&dc->page_fault_vcpu_time[i], 0);
867d73415a3SStefan Hajnoczi         if (qatomic_fetch_add(&dc->vcpu_addr[i], 0) != addr ||
868575b0b33SAlexey Perevalov             read_vcpu_time == 0) {
869575b0b33SAlexey Perevalov             continue;
870575b0b33SAlexey Perevalov         }
871d73415a3SStefan Hajnoczi         qatomic_xchg(&dc->vcpu_addr[i], 0);
872575b0b33SAlexey Perevalov         vcpu_blocktime = low_time_offset - read_vcpu_time;
873575b0b33SAlexey Perevalov         affected_cpu += 1;
874575b0b33SAlexey Perevalov         /* we need to know is that mark_postcopy_end was due to
875575b0b33SAlexey Perevalov          * faulted page, another possible case it's prefetched
876575b0b33SAlexey Perevalov          * page and in that case we shouldn't be here */
877575b0b33SAlexey Perevalov         if (!vcpu_total_blocktime &&
878d73415a3SStefan Hajnoczi             qatomic_fetch_add(&dc->smp_cpus_down, 0) == smp_cpus) {
879575b0b33SAlexey Perevalov             vcpu_total_blocktime = true;
880575b0b33SAlexey Perevalov         }
881575b0b33SAlexey Perevalov         /* continue cycle, due to one page could affect several vCPUs */
882575b0b33SAlexey Perevalov         dc->vcpu_blocktime[i] += vcpu_blocktime;
883575b0b33SAlexey Perevalov     }
884575b0b33SAlexey Perevalov 
885d73415a3SStefan Hajnoczi     qatomic_sub(&dc->smp_cpus_down, affected_cpu);
886575b0b33SAlexey Perevalov     if (vcpu_total_blocktime) {
887d73415a3SStefan Hajnoczi         dc->total_blocktime += low_time_offset - qatomic_fetch_add(
888575b0b33SAlexey Perevalov                 &dc->last_begin, 0);
889575b0b33SAlexey Perevalov     }
890575b0b33SAlexey Perevalov     trace_mark_postcopy_blocktime_end(addr, dc, dc->total_blocktime,
891575b0b33SAlexey Perevalov                                       affected_cpu);
892575b0b33SAlexey Perevalov }
893575b0b33SAlexey Perevalov 
894*27dd21b4SPeter Xu static void postcopy_pause_fault_thread(MigrationIncomingState *mis)
8953a7804c3SPeter Xu {
8963a7804c3SPeter Xu     trace_postcopy_pause_fault_thread();
8973a7804c3SPeter Xu     qemu_sem_wait(&mis->postcopy_pause_sem_fault);
8983a7804c3SPeter Xu     trace_postcopy_pause_fault_thread_continued();
8993a7804c3SPeter Xu }
9003a7804c3SPeter Xu 
901096bf4c8SDr. David Alan Gilbert /*
902f0a227adSDr. David Alan Gilbert  * Handle faults detected by the USERFAULT markings
903f0a227adSDr. David Alan Gilbert  */
904f0a227adSDr. David Alan Gilbert static void *postcopy_ram_fault_thread(void *opaque)
905f0a227adSDr. David Alan Gilbert {
906f0a227adSDr. David Alan Gilbert     MigrationIncomingState *mis = opaque;
907c4faeed2SDr. David Alan Gilbert     struct uffd_msg msg;
908c4faeed2SDr. David Alan Gilbert     int ret;
90900fa4fc8SDr. David Alan Gilbert     size_t index;
910c4faeed2SDr. David Alan Gilbert     RAMBlock *rb = NULL;
911f0a227adSDr. David Alan Gilbert 
912c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_fault_thread_entry();
91374637e6fSLidong Chen     rcu_register_thread();
914096bf4c8SDr. David Alan Gilbert     mis->last_rb = NULL; /* last RAMBlock we sent part of */
915095c12a4SPeter Xu     qemu_sem_post(&mis->thread_sync_sem);
916c4faeed2SDr. David Alan Gilbert 
91700fa4fc8SDr. David Alan Gilbert     struct pollfd *pfd;
91800fa4fc8SDr. David Alan Gilbert     size_t pfd_len = 2 + mis->postcopy_remote_fds->len;
91900fa4fc8SDr. David Alan Gilbert 
92000fa4fc8SDr. David Alan Gilbert     pfd = g_new0(struct pollfd, pfd_len);
92100fa4fc8SDr. David Alan Gilbert 
92200fa4fc8SDr. David Alan Gilbert     pfd[0].fd = mis->userfault_fd;
92300fa4fc8SDr. David Alan Gilbert     pfd[0].events = POLLIN;
92400fa4fc8SDr. David Alan Gilbert     pfd[1].fd = mis->userfault_event_fd;
92500fa4fc8SDr. David Alan Gilbert     pfd[1].events = POLLIN; /* Waiting for eventfd to go positive */
92600fa4fc8SDr. David Alan Gilbert     trace_postcopy_ram_fault_thread_fds_core(pfd[0].fd, pfd[1].fd);
92700fa4fc8SDr. David Alan Gilbert     for (index = 0; index < mis->postcopy_remote_fds->len; index++) {
92800fa4fc8SDr. David Alan Gilbert         struct PostCopyFD *pcfd = &g_array_index(mis->postcopy_remote_fds,
92900fa4fc8SDr. David Alan Gilbert                                                  struct PostCopyFD, index);
93000fa4fc8SDr. David Alan Gilbert         pfd[2 + index].fd = pcfd->fd;
93100fa4fc8SDr. David Alan Gilbert         pfd[2 + index].events = POLLIN;
93200fa4fc8SDr. David Alan Gilbert         trace_postcopy_ram_fault_thread_fds_extra(2 + index, pcfd->idstr,
93300fa4fc8SDr. David Alan Gilbert                                                   pcfd->fd);
93400fa4fc8SDr. David Alan Gilbert     }
93500fa4fc8SDr. David Alan Gilbert 
936c4faeed2SDr. David Alan Gilbert     while (true) {
937c4faeed2SDr. David Alan Gilbert         ram_addr_t rb_offset;
93800fa4fc8SDr. David Alan Gilbert         int poll_result;
939c4faeed2SDr. David Alan Gilbert 
940c4faeed2SDr. David Alan Gilbert         /*
941c4faeed2SDr. David Alan Gilbert          * We're mainly waiting for the kernel to give us a faulting HVA,
942c4faeed2SDr. David Alan Gilbert          * however we can be told to quit via userfault_quit_fd which is
943c4faeed2SDr. David Alan Gilbert          * an eventfd
944c4faeed2SDr. David Alan Gilbert          */
945c4faeed2SDr. David Alan Gilbert 
94600fa4fc8SDr. David Alan Gilbert         poll_result = poll(pfd, pfd_len, -1 /* Wait forever */);
94700fa4fc8SDr. David Alan Gilbert         if (poll_result == -1) {
948c4faeed2SDr. David Alan Gilbert             error_report("%s: userfault poll: %s", __func__, strerror(errno));
949c4faeed2SDr. David Alan Gilbert             break;
950f0a227adSDr. David Alan Gilbert         }
951f0a227adSDr. David Alan Gilbert 
9523a7804c3SPeter Xu         if (!mis->to_src_file) {
9533a7804c3SPeter Xu             /*
9543a7804c3SPeter Xu              * Possibly someone tells us that the return path is
9553a7804c3SPeter Xu              * broken already using the event. We should hold until
9563a7804c3SPeter Xu              * the channel is rebuilt.
9573a7804c3SPeter Xu              */
958*27dd21b4SPeter Xu             postcopy_pause_fault_thread(mis);
9593a7804c3SPeter Xu         }
9603a7804c3SPeter Xu 
961c4faeed2SDr. David Alan Gilbert         if (pfd[1].revents) {
96264f615feSPeter Xu             uint64_t tmp64 = 0;
96364f615feSPeter Xu 
96464f615feSPeter Xu             /* Consume the signal */
96564f615feSPeter Xu             if (read(mis->userfault_event_fd, &tmp64, 8) != 8) {
96664f615feSPeter Xu                 /* Nothing obviously nicer than posting this error. */
96764f615feSPeter Xu                 error_report("%s: read() failed", __func__);
96864f615feSPeter Xu             }
96964f615feSPeter Xu 
970d73415a3SStefan Hajnoczi             if (qatomic_read(&mis->fault_thread_quit)) {
971c4faeed2SDr. David Alan Gilbert                 trace_postcopy_ram_fault_thread_quit();
972c4faeed2SDr. David Alan Gilbert                 break;
973c4faeed2SDr. David Alan Gilbert             }
97464f615feSPeter Xu         }
975c4faeed2SDr. David Alan Gilbert 
97600fa4fc8SDr. David Alan Gilbert         if (pfd[0].revents) {
97700fa4fc8SDr. David Alan Gilbert             poll_result--;
978c4faeed2SDr. David Alan Gilbert             ret = read(mis->userfault_fd, &msg, sizeof(msg));
979c4faeed2SDr. David Alan Gilbert             if (ret != sizeof(msg)) {
980c4faeed2SDr. David Alan Gilbert                 if (errno == EAGAIN) {
981c4faeed2SDr. David Alan Gilbert                     /*
982c4faeed2SDr. David Alan Gilbert                      * if a wake up happens on the other thread just after
983c4faeed2SDr. David Alan Gilbert                      * the poll, there is nothing to read.
984c4faeed2SDr. David Alan Gilbert                      */
985c4faeed2SDr. David Alan Gilbert                     continue;
986c4faeed2SDr. David Alan Gilbert                 }
987c4faeed2SDr. David Alan Gilbert                 if (ret < 0) {
98800fa4fc8SDr. David Alan Gilbert                     error_report("%s: Failed to read full userfault "
98900fa4fc8SDr. David Alan Gilbert                                  "message: %s",
990c4faeed2SDr. David Alan Gilbert                                  __func__, strerror(errno));
991c4faeed2SDr. David Alan Gilbert                     break;
992c4faeed2SDr. David Alan Gilbert                 } else {
99300fa4fc8SDr. David Alan Gilbert                     error_report("%s: Read %d bytes from userfaultfd "
99400fa4fc8SDr. David Alan Gilbert                                  "expected %zd",
995c4faeed2SDr. David Alan Gilbert                                  __func__, ret, sizeof(msg));
996c4faeed2SDr. David Alan Gilbert                     break; /* Lost alignment, don't know what we'd read next */
997c4faeed2SDr. David Alan Gilbert                 }
998c4faeed2SDr. David Alan Gilbert             }
999c4faeed2SDr. David Alan Gilbert             if (msg.event != UFFD_EVENT_PAGEFAULT) {
1000c4faeed2SDr. David Alan Gilbert                 error_report("%s: Read unexpected event %ud from userfaultfd",
1001c4faeed2SDr. David Alan Gilbert                              __func__, msg.event);
1002c4faeed2SDr. David Alan Gilbert                 continue; /* It's not a page fault, shouldn't happen */
1003c4faeed2SDr. David Alan Gilbert             }
1004c4faeed2SDr. David Alan Gilbert 
1005c4faeed2SDr. David Alan Gilbert             rb = qemu_ram_block_from_host(
1006c4faeed2SDr. David Alan Gilbert                      (void *)(uintptr_t)msg.arg.pagefault.address,
1007f615f396SPaolo Bonzini                      true, &rb_offset);
1008c4faeed2SDr. David Alan Gilbert             if (!rb) {
1009c4faeed2SDr. David Alan Gilbert                 error_report("postcopy_ram_fault_thread: Fault outside guest: %"
1010c4faeed2SDr. David Alan Gilbert                              PRIx64, (uint64_t)msg.arg.pagefault.address);
1011c4faeed2SDr. David Alan Gilbert                 break;
1012c4faeed2SDr. David Alan Gilbert             }
1013c4faeed2SDr. David Alan Gilbert 
10147648297dSDavid Hildenbrand             rb_offset = ROUND_DOWN(rb_offset, qemu_ram_pagesize(rb));
1015c4faeed2SDr. David Alan Gilbert             trace_postcopy_ram_fault_thread_request(msg.arg.pagefault.address,
1016c4faeed2SDr. David Alan Gilbert                                                 qemu_ram_get_idstr(rb),
1017575b0b33SAlexey Perevalov                                                 rb_offset,
1018575b0b33SAlexey Perevalov                                                 msg.arg.pagefault.feat.ptid);
1019575b0b33SAlexey Perevalov             mark_postcopy_blocktime_begin(
1020575b0b33SAlexey Perevalov                     (uintptr_t)(msg.arg.pagefault.address),
1021575b0b33SAlexey Perevalov                                 msg.arg.pagefault.feat.ptid, rb);
1022575b0b33SAlexey Perevalov 
10233a7804c3SPeter Xu retry:
1024c4faeed2SDr. David Alan Gilbert             /*
1025c4faeed2SDr. David Alan Gilbert              * Send the request to the source - we want to request one
1026c4faeed2SDr. David Alan Gilbert              * of our host page sizes (which is >= TPS)
1027c4faeed2SDr. David Alan Gilbert              */
10289470c5e0SDavid Hildenbrand             ret = postcopy_request_page(mis, rb, rb_offset,
10298f8bfffcSPeter Xu                                         msg.arg.pagefault.address);
10303a7804c3SPeter Xu             if (ret) {
10313a7804c3SPeter Xu                 /* May be network failure, try to wait for recovery */
1032*27dd21b4SPeter Xu                 postcopy_pause_fault_thread(mis);
10333a7804c3SPeter Xu                 goto retry;
1034c4faeed2SDr. David Alan Gilbert             }
1035c4faeed2SDr. David Alan Gilbert         }
103600fa4fc8SDr. David Alan Gilbert 
103700fa4fc8SDr. David Alan Gilbert         /* Now handle any requests from external processes on shared memory */
103800fa4fc8SDr. David Alan Gilbert         /* TODO: May need to handle devices deregistering during postcopy */
103900fa4fc8SDr. David Alan Gilbert         for (index = 2; index < pfd_len && poll_result; index++) {
104000fa4fc8SDr. David Alan Gilbert             if (pfd[index].revents) {
104100fa4fc8SDr. David Alan Gilbert                 struct PostCopyFD *pcfd =
104200fa4fc8SDr. David Alan Gilbert                     &g_array_index(mis->postcopy_remote_fds,
104300fa4fc8SDr. David Alan Gilbert                                    struct PostCopyFD, index - 2);
104400fa4fc8SDr. David Alan Gilbert 
104500fa4fc8SDr. David Alan Gilbert                 poll_result--;
104600fa4fc8SDr. David Alan Gilbert                 if (pfd[index].revents & POLLERR) {
104700fa4fc8SDr. David Alan Gilbert                     error_report("%s: POLLERR on poll %zd fd=%d",
104800fa4fc8SDr. David Alan Gilbert                                  __func__, index, pcfd->fd);
104900fa4fc8SDr. David Alan Gilbert                     pfd[index].events = 0;
105000fa4fc8SDr. David Alan Gilbert                     continue;
105100fa4fc8SDr. David Alan Gilbert                 }
105200fa4fc8SDr. David Alan Gilbert 
105300fa4fc8SDr. David Alan Gilbert                 ret = read(pcfd->fd, &msg, sizeof(msg));
105400fa4fc8SDr. David Alan Gilbert                 if (ret != sizeof(msg)) {
105500fa4fc8SDr. David Alan Gilbert                     if (errno == EAGAIN) {
105600fa4fc8SDr. David Alan Gilbert                         /*
105700fa4fc8SDr. David Alan Gilbert                          * if a wake up happens on the other thread just after
105800fa4fc8SDr. David Alan Gilbert                          * the poll, there is nothing to read.
105900fa4fc8SDr. David Alan Gilbert                          */
106000fa4fc8SDr. David Alan Gilbert                         continue;
106100fa4fc8SDr. David Alan Gilbert                     }
106200fa4fc8SDr. David Alan Gilbert                     if (ret < 0) {
106300fa4fc8SDr. David Alan Gilbert                         error_report("%s: Failed to read full userfault "
106400fa4fc8SDr. David Alan Gilbert                                      "message: %s (shared) revents=%d",
106500fa4fc8SDr. David Alan Gilbert                                      __func__, strerror(errno),
106600fa4fc8SDr. David Alan Gilbert                                      pfd[index].revents);
106700fa4fc8SDr. David Alan Gilbert                         /*TODO: Could just disable this sharer */
106800fa4fc8SDr. David Alan Gilbert                         break;
106900fa4fc8SDr. David Alan Gilbert                     } else {
107000fa4fc8SDr. David Alan Gilbert                         error_report("%s: Read %d bytes from userfaultfd "
107100fa4fc8SDr. David Alan Gilbert                                      "expected %zd (shared)",
107200fa4fc8SDr. David Alan Gilbert                                      __func__, ret, sizeof(msg));
107300fa4fc8SDr. David Alan Gilbert                         /*TODO: Could just disable this sharer */
107400fa4fc8SDr. David Alan Gilbert                         break; /*Lost alignment,don't know what we'd read next*/
107500fa4fc8SDr. David Alan Gilbert                     }
107600fa4fc8SDr. David Alan Gilbert                 }
107700fa4fc8SDr. David Alan Gilbert                 if (msg.event != UFFD_EVENT_PAGEFAULT) {
107800fa4fc8SDr. David Alan Gilbert                     error_report("%s: Read unexpected event %ud "
107900fa4fc8SDr. David Alan Gilbert                                  "from userfaultfd (shared)",
108000fa4fc8SDr. David Alan Gilbert                                  __func__, msg.event);
108100fa4fc8SDr. David Alan Gilbert                     continue; /* It's not a page fault, shouldn't happen */
108200fa4fc8SDr. David Alan Gilbert                 }
108300fa4fc8SDr. David Alan Gilbert                 /* Call the device handler registered with us */
108400fa4fc8SDr. David Alan Gilbert                 ret = pcfd->handler(pcfd, &msg);
108500fa4fc8SDr. David Alan Gilbert                 if (ret) {
108600fa4fc8SDr. David Alan Gilbert                     error_report("%s: Failed to resolve shared fault on %zd/%s",
108700fa4fc8SDr. David Alan Gilbert                                  __func__, index, pcfd->idstr);
108800fa4fc8SDr. David Alan Gilbert                     /* TODO: Fail? Disable this sharer? */
108900fa4fc8SDr. David Alan Gilbert                 }
109000fa4fc8SDr. David Alan Gilbert             }
109100fa4fc8SDr. David Alan Gilbert         }
109200fa4fc8SDr. David Alan Gilbert     }
109374637e6fSLidong Chen     rcu_unregister_thread();
1094c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_fault_thread_exit();
1095fc6008f3SMarc-André Lureau     g_free(pfd);
1096f0a227adSDr. David Alan Gilbert     return NULL;
1097f0a227adSDr. David Alan Gilbert }
1098f0a227adSDr. David Alan Gilbert 
1099476ebf77SPeter Xu static int postcopy_temp_pages_setup(MigrationIncomingState *mis)
1100476ebf77SPeter Xu {
110177dadc3fSPeter Xu     PostcopyTmpPage *tmp_page;
110277dadc3fSPeter Xu     int err, i, channels;
110377dadc3fSPeter Xu     void *temp_page;
1104476ebf77SPeter Xu 
110577dadc3fSPeter Xu     /* TODO: will be boosted when enable postcopy preemption */
110677dadc3fSPeter Xu     mis->postcopy_channels = 1;
110777dadc3fSPeter Xu 
110877dadc3fSPeter Xu     channels = mis->postcopy_channels;
110977dadc3fSPeter Xu     mis->postcopy_tmp_pages = g_malloc0_n(sizeof(PostcopyTmpPage), channels);
111077dadc3fSPeter Xu 
111177dadc3fSPeter Xu     for (i = 0; i < channels; i++) {
111277dadc3fSPeter Xu         tmp_page = &mis->postcopy_tmp_pages[i];
111377dadc3fSPeter Xu         temp_page = mmap(NULL, mis->largest_page_size, PROT_READ | PROT_WRITE,
1114476ebf77SPeter Xu                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
111577dadc3fSPeter Xu         if (temp_page == MAP_FAILED) {
1116476ebf77SPeter Xu             err = errno;
111777dadc3fSPeter Xu             error_report("%s: Failed to map postcopy_tmp_pages[%d]: %s",
111877dadc3fSPeter Xu                          __func__, i, strerror(err));
111977dadc3fSPeter Xu             /* Clean up will be done later */
1120476ebf77SPeter Xu             return -err;
1121476ebf77SPeter Xu         }
112277dadc3fSPeter Xu         tmp_page->tmp_huge_page = temp_page;
112377dadc3fSPeter Xu         /* Initialize default states for each tmp page */
112477dadc3fSPeter Xu         postcopy_temp_page_reset(tmp_page);
112577dadc3fSPeter Xu     }
1126476ebf77SPeter Xu 
1127476ebf77SPeter Xu     /*
1128476ebf77SPeter Xu      * Map large zero page when kernel can't use UFFDIO_ZEROPAGE for hugepages
1129476ebf77SPeter Xu      */
1130476ebf77SPeter Xu     mis->postcopy_tmp_zero_page = mmap(NULL, mis->largest_page_size,
1131476ebf77SPeter Xu                                        PROT_READ | PROT_WRITE,
1132476ebf77SPeter Xu                                        MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
1133476ebf77SPeter Xu     if (mis->postcopy_tmp_zero_page == MAP_FAILED) {
1134476ebf77SPeter Xu         err = errno;
1135476ebf77SPeter Xu         mis->postcopy_tmp_zero_page = NULL;
1136476ebf77SPeter Xu         error_report("%s: Failed to map large zero page %s",
1137476ebf77SPeter Xu                      __func__, strerror(err));
1138476ebf77SPeter Xu         return -err;
1139476ebf77SPeter Xu     }
1140476ebf77SPeter Xu 
1141476ebf77SPeter Xu     memset(mis->postcopy_tmp_zero_page, '\0', mis->largest_page_size);
1142476ebf77SPeter Xu 
1143476ebf77SPeter Xu     return 0;
1144476ebf77SPeter Xu }
1145476ebf77SPeter Xu 
11462a7eb148SWei Yang int postcopy_ram_incoming_setup(MigrationIncomingState *mis)
1147f0a227adSDr. David Alan Gilbert {
1148c4faeed2SDr. David Alan Gilbert     /* Open the fd for the kernel to give us userfaults */
1149c4faeed2SDr. David Alan Gilbert     mis->userfault_fd = syscall(__NR_userfaultfd, O_CLOEXEC | O_NONBLOCK);
1150c4faeed2SDr. David Alan Gilbert     if (mis->userfault_fd == -1) {
1151c4faeed2SDr. David Alan Gilbert         error_report("%s: Failed to open userfault fd: %s", __func__,
1152c4faeed2SDr. David Alan Gilbert                      strerror(errno));
1153c4faeed2SDr. David Alan Gilbert         return -1;
1154c4faeed2SDr. David Alan Gilbert     }
1155c4faeed2SDr. David Alan Gilbert 
1156c4faeed2SDr. David Alan Gilbert     /*
1157c4faeed2SDr. David Alan Gilbert      * Although the host check already tested the API, we need to
1158c4faeed2SDr. David Alan Gilbert      * do the check again as an ABI handshake on the new fd.
1159c4faeed2SDr. David Alan Gilbert      */
116054ae0886SAlexey Perevalov     if (!ufd_check_and_apply(mis->userfault_fd, mis)) {
1161c4faeed2SDr. David Alan Gilbert         return -1;
1162c4faeed2SDr. David Alan Gilbert     }
1163c4faeed2SDr. David Alan Gilbert 
1164c4faeed2SDr. David Alan Gilbert     /* Now an eventfd we use to tell the fault-thread to quit */
116564f615feSPeter Xu     mis->userfault_event_fd = eventfd(0, EFD_CLOEXEC);
116664f615feSPeter Xu     if (mis->userfault_event_fd == -1) {
116764f615feSPeter Xu         error_report("%s: Opening userfault_event_fd: %s", __func__,
1168c4faeed2SDr. David Alan Gilbert                      strerror(errno));
1169c4faeed2SDr. David Alan Gilbert         close(mis->userfault_fd);
1170c4faeed2SDr. David Alan Gilbert         return -1;
1171c4faeed2SDr. David Alan Gilbert     }
1172c4faeed2SDr. David Alan Gilbert 
1173095c12a4SPeter Xu     postcopy_thread_create(mis, &mis->fault_thread, "postcopy/fault",
1174095c12a4SPeter Xu                            postcopy_ram_fault_thread, QEMU_THREAD_JOINABLE);
1175c4faeed2SDr. David Alan Gilbert     mis->have_fault_thread = true;
1176f0a227adSDr. David Alan Gilbert 
1177f0a227adSDr. David Alan Gilbert     /* Mark so that we get notified of accesses to unwritten areas */
1178fbd162e6SYury Kotov     if (foreach_not_ignored_block(ram_block_enable_notify, mis)) {
117991b02dc7SFei Li         error_report("ram_block_enable_notify failed");
1180f0a227adSDr. David Alan Gilbert         return -1;
1181f0a227adSDr. David Alan Gilbert     }
1182f0a227adSDr. David Alan Gilbert 
1183476ebf77SPeter Xu     if (postcopy_temp_pages_setup(mis)) {
1184476ebf77SPeter Xu         /* Error dumped in the sub-function */
11853414322aSWei Yang         return -1;
11863414322aSWei Yang     }
11873414322aSWei Yang 
1188c4faeed2SDr. David Alan Gilbert     trace_postcopy_ram_enable_notify();
1189c4faeed2SDr. David Alan Gilbert 
1190f0a227adSDr. David Alan Gilbert     return 0;
1191f0a227adSDr. David Alan Gilbert }
1192f0a227adSDr. David Alan Gilbert 
1193eef621c4SPeter Xu static int qemu_ufd_copy_ioctl(MigrationIncomingState *mis, void *host_addr,
1194f9494614SAlexey Perevalov                                void *from_addr, uint64_t pagesize, RAMBlock *rb)
1195727b9d7eSAlexey Perevalov {
1196eef621c4SPeter Xu     int userfault_fd = mis->userfault_fd;
1197f9494614SAlexey Perevalov     int ret;
1198eef621c4SPeter Xu 
1199727b9d7eSAlexey Perevalov     if (from_addr) {
1200727b9d7eSAlexey Perevalov         struct uffdio_copy copy_struct;
1201727b9d7eSAlexey Perevalov         copy_struct.dst = (uint64_t)(uintptr_t)host_addr;
1202727b9d7eSAlexey Perevalov         copy_struct.src = (uint64_t)(uintptr_t)from_addr;
1203727b9d7eSAlexey Perevalov         copy_struct.len = pagesize;
1204727b9d7eSAlexey Perevalov         copy_struct.mode = 0;
1205f9494614SAlexey Perevalov         ret = ioctl(userfault_fd, UFFDIO_COPY, &copy_struct);
1206727b9d7eSAlexey Perevalov     } else {
1207727b9d7eSAlexey Perevalov         struct uffdio_zeropage zero_struct;
1208727b9d7eSAlexey Perevalov         zero_struct.range.start = (uint64_t)(uintptr_t)host_addr;
1209727b9d7eSAlexey Perevalov         zero_struct.range.len = pagesize;
1210727b9d7eSAlexey Perevalov         zero_struct.mode = 0;
1211f9494614SAlexey Perevalov         ret = ioctl(userfault_fd, UFFDIO_ZEROPAGE, &zero_struct);
1212727b9d7eSAlexey Perevalov     }
1213f9494614SAlexey Perevalov     if (!ret) {
12148f8bfffcSPeter Xu         qemu_mutex_lock(&mis->page_request_mutex);
1215f9494614SAlexey Perevalov         ramblock_recv_bitmap_set_range(rb, host_addr,
1216f9494614SAlexey Perevalov                                        pagesize / qemu_target_page_size());
12178f8bfffcSPeter Xu         /*
12188f8bfffcSPeter Xu          * If this page resolves a page fault for a previous recorded faulted
12198f8bfffcSPeter Xu          * address, take a special note to maintain the requested page list.
12208f8bfffcSPeter Xu          */
12218f8bfffcSPeter Xu         if (g_tree_lookup(mis->page_requested, host_addr)) {
12228f8bfffcSPeter Xu             g_tree_remove(mis->page_requested, host_addr);
12238f8bfffcSPeter Xu             mis->page_requested_count--;
12248f8bfffcSPeter Xu             trace_postcopy_page_req_del(host_addr, mis->page_requested_count);
12258f8bfffcSPeter Xu         }
12268f8bfffcSPeter Xu         qemu_mutex_unlock(&mis->page_request_mutex);
1227575b0b33SAlexey Perevalov         mark_postcopy_blocktime_end((uintptr_t)host_addr);
1228f9494614SAlexey Perevalov     }
1229f9494614SAlexey Perevalov     return ret;
1230727b9d7eSAlexey Perevalov }
1231727b9d7eSAlexey Perevalov 
1232d488b349SDr. David Alan Gilbert int postcopy_notify_shared_wake(RAMBlock *rb, uint64_t offset)
1233d488b349SDr. David Alan Gilbert {
1234d488b349SDr. David Alan Gilbert     int i;
1235d488b349SDr. David Alan Gilbert     MigrationIncomingState *mis = migration_incoming_get_current();
1236d488b349SDr. David Alan Gilbert     GArray *pcrfds = mis->postcopy_remote_fds;
1237d488b349SDr. David Alan Gilbert 
1238d488b349SDr. David Alan Gilbert     for (i = 0; i < pcrfds->len; i++) {
1239d488b349SDr. David Alan Gilbert         struct PostCopyFD *cur = &g_array_index(pcrfds, struct PostCopyFD, i);
1240d488b349SDr. David Alan Gilbert         int ret = cur->waker(cur, rb, offset);
1241d488b349SDr. David Alan Gilbert         if (ret) {
1242d488b349SDr. David Alan Gilbert             return ret;
1243d488b349SDr. David Alan Gilbert         }
1244d488b349SDr. David Alan Gilbert     }
1245d488b349SDr. David Alan Gilbert     return 0;
1246d488b349SDr. David Alan Gilbert }
1247d488b349SDr. David Alan Gilbert 
1248696ed9a9SDr. David Alan Gilbert /*
1249696ed9a9SDr. David Alan Gilbert  * Place a host page (from) at (host) atomically
1250696ed9a9SDr. David Alan Gilbert  * returns 0 on success
1251696ed9a9SDr. David Alan Gilbert  */
1252df9ff5e1SDr. David Alan Gilbert int postcopy_place_page(MigrationIncomingState *mis, void *host, void *from,
12538be4620bSAlexey Perevalov                         RAMBlock *rb)
1254696ed9a9SDr. David Alan Gilbert {
12558be4620bSAlexey Perevalov     size_t pagesize = qemu_ram_pagesize(rb);
1256696ed9a9SDr. David Alan Gilbert 
1257696ed9a9SDr. David Alan Gilbert     /* copy also acks to the kernel waking the stalled thread up
1258696ed9a9SDr. David Alan Gilbert      * TODO: We can inhibit that ack and only do it if it was requested
1259696ed9a9SDr. David Alan Gilbert      * which would be slightly cheaper, but we'd have to be careful
1260696ed9a9SDr. David Alan Gilbert      * of the order of updating our page state.
1261696ed9a9SDr. David Alan Gilbert      */
1262eef621c4SPeter Xu     if (qemu_ufd_copy_ioctl(mis, host, from, pagesize, rb)) {
1263696ed9a9SDr. David Alan Gilbert         int e = errno;
1264df9ff5e1SDr. David Alan Gilbert         error_report("%s: %s copy host: %p from: %p (size: %zd)",
1265df9ff5e1SDr. David Alan Gilbert                      __func__, strerror(e), host, from, pagesize);
1266696ed9a9SDr. David Alan Gilbert 
1267696ed9a9SDr. David Alan Gilbert         return -e;
1268696ed9a9SDr. David Alan Gilbert     }
1269696ed9a9SDr. David Alan Gilbert 
1270696ed9a9SDr. David Alan Gilbert     trace_postcopy_place_page(host);
1271dedfb4b2SDr. David Alan Gilbert     return postcopy_notify_shared_wake(rb,
1272dedfb4b2SDr. David Alan Gilbert                                        qemu_ram_block_host_offset(rb, host));
1273696ed9a9SDr. David Alan Gilbert }
1274696ed9a9SDr. David Alan Gilbert 
1275696ed9a9SDr. David Alan Gilbert /*
1276696ed9a9SDr. David Alan Gilbert  * Place a zero page at (host) atomically
1277696ed9a9SDr. David Alan Gilbert  * returns 0 on success
1278696ed9a9SDr. David Alan Gilbert  */
1279df9ff5e1SDr. David Alan Gilbert int postcopy_place_page_zero(MigrationIncomingState *mis, void *host,
12808be4620bSAlexey Perevalov                              RAMBlock *rb)
1281696ed9a9SDr. David Alan Gilbert {
12822ce16640SDr. David Alan Gilbert     size_t pagesize = qemu_ram_pagesize(rb);
1283df9ff5e1SDr. David Alan Gilbert     trace_postcopy_place_page_zero(host);
1284696ed9a9SDr. David Alan Gilbert 
12852ce16640SDr. David Alan Gilbert     /* Normal RAMBlocks can zero a page using UFFDIO_ZEROPAGE
12862ce16640SDr. David Alan Gilbert      * but it's not available for everything (e.g. hugetlbpages)
12872ce16640SDr. David Alan Gilbert      */
12882ce16640SDr. David Alan Gilbert     if (qemu_ram_is_uf_zeroable(rb)) {
1289eef621c4SPeter Xu         if (qemu_ufd_copy_ioctl(mis, host, NULL, pagesize, rb)) {
1290696ed9a9SDr. David Alan Gilbert             int e = errno;
1291696ed9a9SDr. David Alan Gilbert             error_report("%s: %s zero host: %p",
1292696ed9a9SDr. David Alan Gilbert                          __func__, strerror(e), host);
1293696ed9a9SDr. David Alan Gilbert 
1294696ed9a9SDr. David Alan Gilbert             return -e;
1295696ed9a9SDr. David Alan Gilbert         }
1296dedfb4b2SDr. David Alan Gilbert         return postcopy_notify_shared_wake(rb,
1297dedfb4b2SDr. David Alan Gilbert                                            qemu_ram_block_host_offset(rb,
1298dedfb4b2SDr. David Alan Gilbert                                                                       host));
1299df9ff5e1SDr. David Alan Gilbert     } else {
13006629890dSWei Yang         return postcopy_place_page(mis, host, mis->postcopy_tmp_zero_page, rb);
1301df9ff5e1SDr. David Alan Gilbert     }
1302696ed9a9SDr. David Alan Gilbert }
1303696ed9a9SDr. David Alan Gilbert 
1304eb59db53SDr. David Alan Gilbert #else
1305eb59db53SDr. David Alan Gilbert /* No target OS support, stubs just fail */
130665ace060SAlexey Perevalov void fill_destination_postcopy_migration_info(MigrationInfo *info)
130765ace060SAlexey Perevalov {
130865ace060SAlexey Perevalov }
130965ace060SAlexey Perevalov 
1310d7651f15SAlexey Perevalov bool postcopy_ram_supported_by_host(MigrationIncomingState *mis)
1311eb59db53SDr. David Alan Gilbert {
1312eb59db53SDr. David Alan Gilbert     error_report("%s: No OS support", __func__);
1313eb59db53SDr. David Alan Gilbert     return false;
1314eb59db53SDr. David Alan Gilbert }
1315eb59db53SDr. David Alan Gilbert 
1316c136180cSDavid Hildenbrand int postcopy_ram_incoming_init(MigrationIncomingState *mis)
13171caddf8aSDr. David Alan Gilbert {
13181caddf8aSDr. David Alan Gilbert     error_report("postcopy_ram_incoming_init: No OS support");
13191caddf8aSDr. David Alan Gilbert     return -1;
13201caddf8aSDr. David Alan Gilbert }
13211caddf8aSDr. David Alan Gilbert 
13221caddf8aSDr. David Alan Gilbert int postcopy_ram_incoming_cleanup(MigrationIncomingState *mis)
13231caddf8aSDr. David Alan Gilbert {
13241caddf8aSDr. David Alan Gilbert     assert(0);
13251caddf8aSDr. David Alan Gilbert     return -1;
13261caddf8aSDr. David Alan Gilbert }
13271caddf8aSDr. David Alan Gilbert 
1328f9527107SDr. David Alan Gilbert int postcopy_ram_prepare_discard(MigrationIncomingState *mis)
1329f9527107SDr. David Alan Gilbert {
1330f9527107SDr. David Alan Gilbert     assert(0);
1331f9527107SDr. David Alan Gilbert     return -1;
1332f9527107SDr. David Alan Gilbert }
1333f9527107SDr. David Alan Gilbert 
1334c188c539SMichael S. Tsirkin int postcopy_request_shared_page(struct PostCopyFD *pcfd, RAMBlock *rb,
1335c188c539SMichael S. Tsirkin                                  uint64_t client_addr, uint64_t rb_offset)
1336c188c539SMichael S. Tsirkin {
1337c188c539SMichael S. Tsirkin     assert(0);
1338c188c539SMichael S. Tsirkin     return -1;
1339c188c539SMichael S. Tsirkin }
1340c188c539SMichael S. Tsirkin 
13412a7eb148SWei Yang int postcopy_ram_incoming_setup(MigrationIncomingState *mis)
1342f0a227adSDr. David Alan Gilbert {
1343f0a227adSDr. David Alan Gilbert     assert(0);
1344f0a227adSDr. David Alan Gilbert     return -1;
1345f0a227adSDr. David Alan Gilbert }
1346696ed9a9SDr. David Alan Gilbert 
1347df9ff5e1SDr. David Alan Gilbert int postcopy_place_page(MigrationIncomingState *mis, void *host, void *from,
13488be4620bSAlexey Perevalov                         RAMBlock *rb)
1349696ed9a9SDr. David Alan Gilbert {
1350696ed9a9SDr. David Alan Gilbert     assert(0);
1351696ed9a9SDr. David Alan Gilbert     return -1;
1352696ed9a9SDr. David Alan Gilbert }
1353696ed9a9SDr. David Alan Gilbert 
1354df9ff5e1SDr. David Alan Gilbert int postcopy_place_page_zero(MigrationIncomingState *mis, void *host,
13558be4620bSAlexey Perevalov                         RAMBlock *rb)
1356696ed9a9SDr. David Alan Gilbert {
1357696ed9a9SDr. David Alan Gilbert     assert(0);
1358696ed9a9SDr. David Alan Gilbert     return -1;
1359696ed9a9SDr. David Alan Gilbert }
1360696ed9a9SDr. David Alan Gilbert 
13615efc3564SDr. David Alan Gilbert int postcopy_wake_shared(struct PostCopyFD *pcfd,
13625efc3564SDr. David Alan Gilbert                          uint64_t client_addr,
13635efc3564SDr. David Alan Gilbert                          RAMBlock *rb)
13645efc3564SDr. David Alan Gilbert {
13655efc3564SDr. David Alan Gilbert     assert(0);
13665efc3564SDr. David Alan Gilbert     return -1;
13675efc3564SDr. David Alan Gilbert }
1368eb59db53SDr. David Alan Gilbert #endif
1369eb59db53SDr. David Alan Gilbert 
1370e0b266f0SDr. David Alan Gilbert /* ------------------------------------------------------------------------- */
137177dadc3fSPeter Xu void postcopy_temp_page_reset(PostcopyTmpPage *tmp_page)
137277dadc3fSPeter Xu {
137377dadc3fSPeter Xu     tmp_page->target_pages = 0;
137477dadc3fSPeter Xu     tmp_page->host_addr = NULL;
137577dadc3fSPeter Xu     /*
137677dadc3fSPeter Xu      * This is set to true when reset, and cleared as long as we received any
137777dadc3fSPeter Xu      * of the non-zero small page within this huge page.
137877dadc3fSPeter Xu      */
137977dadc3fSPeter Xu     tmp_page->all_zero = true;
138077dadc3fSPeter Xu }
1381e0b266f0SDr. David Alan Gilbert 
13829ab7ef9bSPeter Xu void postcopy_fault_thread_notify(MigrationIncomingState *mis)
13839ab7ef9bSPeter Xu {
13849ab7ef9bSPeter Xu     uint64_t tmp64 = 1;
13859ab7ef9bSPeter Xu 
13869ab7ef9bSPeter Xu     /*
13879ab7ef9bSPeter Xu      * Wakeup the fault_thread.  It's an eventfd that should currently
13889ab7ef9bSPeter Xu      * be at 0, we're going to increment it to 1
13899ab7ef9bSPeter Xu      */
13909ab7ef9bSPeter Xu     if (write(mis->userfault_event_fd, &tmp64, 8) != 8) {
13919ab7ef9bSPeter Xu         /* Not much we can do here, but may as well report it */
13929ab7ef9bSPeter Xu         error_report("%s: incrementing failed: %s", __func__,
13939ab7ef9bSPeter Xu                      strerror(errno));
13949ab7ef9bSPeter Xu     }
13959ab7ef9bSPeter Xu }
13969ab7ef9bSPeter Xu 
1397e0b266f0SDr. David Alan Gilbert /**
1398e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_init: Called at the start of each RAMBlock before
1399e0b266f0SDr. David Alan Gilbert  *   asking to discard individual ranges.
1400e0b266f0SDr. David Alan Gilbert  *
1401e0b266f0SDr. David Alan Gilbert  * @ms: The current migration state.
1402810cf2bbSWei Yang  * @offset: the bitmap offset of the named RAMBlock in the migration bitmap.
1403e0b266f0SDr. David Alan Gilbert  * @name: RAMBlock that discards will operate on.
1404e0b266f0SDr. David Alan Gilbert  */
1405810cf2bbSWei Yang static PostcopyDiscardState pds = {0};
1406810cf2bbSWei Yang void postcopy_discard_send_init(MigrationState *ms, const char *name)
1407e0b266f0SDr. David Alan Gilbert {
1408810cf2bbSWei Yang     pds.ramblock_name = name;
1409810cf2bbSWei Yang     pds.cur_entry = 0;
1410810cf2bbSWei Yang     pds.nsentwords = 0;
1411810cf2bbSWei Yang     pds.nsentcmds = 0;
1412e0b266f0SDr. David Alan Gilbert }
1413e0b266f0SDr. David Alan Gilbert 
1414e0b266f0SDr. David Alan Gilbert /**
1415e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_range: Called by the bitmap code for each chunk to
1416e0b266f0SDr. David Alan Gilbert  *   discard. May send a discard message, may just leave it queued to
1417e0b266f0SDr. David Alan Gilbert  *   be sent later.
1418e0b266f0SDr. David Alan Gilbert  *
1419e0b266f0SDr. David Alan Gilbert  * @ms: Current migration state.
1420e0b266f0SDr. David Alan Gilbert  * @start,@length: a range of pages in the migration bitmap in the
1421e0b266f0SDr. David Alan Gilbert  *   RAM block passed to postcopy_discard_send_init() (length=1 is one page)
1422e0b266f0SDr. David Alan Gilbert  */
1423810cf2bbSWei Yang void postcopy_discard_send_range(MigrationState *ms, unsigned long start,
1424810cf2bbSWei Yang                                  unsigned long length)
1425e0b266f0SDr. David Alan Gilbert {
142620afaed9SJuan Quintela     size_t tp_size = qemu_target_page_size();
1427e0b266f0SDr. David Alan Gilbert     /* Convert to byte offsets within the RAM block */
1428810cf2bbSWei Yang     pds.start_list[pds.cur_entry] = start  * tp_size;
1429810cf2bbSWei Yang     pds.length_list[pds.cur_entry] = length * tp_size;
1430810cf2bbSWei Yang     trace_postcopy_discard_send_range(pds.ramblock_name, start, length);
1431810cf2bbSWei Yang     pds.cur_entry++;
1432810cf2bbSWei Yang     pds.nsentwords++;
1433e0b266f0SDr. David Alan Gilbert 
1434810cf2bbSWei Yang     if (pds.cur_entry == MAX_DISCARDS_PER_COMMAND) {
1435e0b266f0SDr. David Alan Gilbert         /* Full set, ship it! */
143689a02a9fSzhanghailiang         qemu_savevm_send_postcopy_ram_discard(ms->to_dst_file,
1437810cf2bbSWei Yang                                               pds.ramblock_name,
1438810cf2bbSWei Yang                                               pds.cur_entry,
1439810cf2bbSWei Yang                                               pds.start_list,
1440810cf2bbSWei Yang                                               pds.length_list);
1441810cf2bbSWei Yang         pds.nsentcmds++;
1442810cf2bbSWei Yang         pds.cur_entry = 0;
1443e0b266f0SDr. David Alan Gilbert     }
1444e0b266f0SDr. David Alan Gilbert }
1445e0b266f0SDr. David Alan Gilbert 
1446e0b266f0SDr. David Alan Gilbert /**
1447e0b266f0SDr. David Alan Gilbert  * postcopy_discard_send_finish: Called at the end of each RAMBlock by the
1448e0b266f0SDr. David Alan Gilbert  * bitmap code. Sends any outstanding discard messages, frees the PDS
1449e0b266f0SDr. David Alan Gilbert  *
1450e0b266f0SDr. David Alan Gilbert  * @ms: Current migration state.
1451e0b266f0SDr. David Alan Gilbert  */
1452810cf2bbSWei Yang void postcopy_discard_send_finish(MigrationState *ms)
1453e0b266f0SDr. David Alan Gilbert {
1454e0b266f0SDr. David Alan Gilbert     /* Anything unsent? */
1455810cf2bbSWei Yang     if (pds.cur_entry) {
145689a02a9fSzhanghailiang         qemu_savevm_send_postcopy_ram_discard(ms->to_dst_file,
1457810cf2bbSWei Yang                                               pds.ramblock_name,
1458810cf2bbSWei Yang                                               pds.cur_entry,
1459810cf2bbSWei Yang                                               pds.start_list,
1460810cf2bbSWei Yang                                               pds.length_list);
1461810cf2bbSWei Yang         pds.nsentcmds++;
1462e0b266f0SDr. David Alan Gilbert     }
1463e0b266f0SDr. David Alan Gilbert 
1464810cf2bbSWei Yang     trace_postcopy_discard_send_finish(pds.ramblock_name, pds.nsentwords,
1465810cf2bbSWei Yang                                        pds.nsentcmds);
1466e0b266f0SDr. David Alan Gilbert }
1467bac3b212SJuan Quintela 
1468bac3b212SJuan Quintela /*
1469bac3b212SJuan Quintela  * Current state of incoming postcopy; note this is not part of
1470bac3b212SJuan Quintela  * MigrationIncomingState since it's state is used during cleanup
1471bac3b212SJuan Quintela  * at the end as MIS is being freed.
1472bac3b212SJuan Quintela  */
1473bac3b212SJuan Quintela static PostcopyState incoming_postcopy_state;
1474bac3b212SJuan Quintela 
1475bac3b212SJuan Quintela PostcopyState  postcopy_state_get(void)
1476bac3b212SJuan Quintela {
1477d73415a3SStefan Hajnoczi     return qatomic_mb_read(&incoming_postcopy_state);
1478bac3b212SJuan Quintela }
1479bac3b212SJuan Quintela 
1480bac3b212SJuan Quintela /* Set the state and return the old state */
1481bac3b212SJuan Quintela PostcopyState postcopy_state_set(PostcopyState new_state)
1482bac3b212SJuan Quintela {
1483d73415a3SStefan Hajnoczi     return qatomic_xchg(&incoming_postcopy_state, new_state);
1484bac3b212SJuan Quintela }
148500fa4fc8SDr. David Alan Gilbert 
148600fa4fc8SDr. David Alan Gilbert /* Register a handler for external shared memory postcopy
148700fa4fc8SDr. David Alan Gilbert  * called on the destination.
148800fa4fc8SDr. David Alan Gilbert  */
148900fa4fc8SDr. David Alan Gilbert void postcopy_register_shared_ufd(struct PostCopyFD *pcfd)
149000fa4fc8SDr. David Alan Gilbert {
149100fa4fc8SDr. David Alan Gilbert     MigrationIncomingState *mis = migration_incoming_get_current();
149200fa4fc8SDr. David Alan Gilbert 
149300fa4fc8SDr. David Alan Gilbert     mis->postcopy_remote_fds = g_array_append_val(mis->postcopy_remote_fds,
149400fa4fc8SDr. David Alan Gilbert                                                   *pcfd);
149500fa4fc8SDr. David Alan Gilbert }
149600fa4fc8SDr. David Alan Gilbert 
149700fa4fc8SDr. David Alan Gilbert /* Unregister a handler for external shared memory postcopy
149800fa4fc8SDr. David Alan Gilbert  */
149900fa4fc8SDr. David Alan Gilbert void postcopy_unregister_shared_ufd(struct PostCopyFD *pcfd)
150000fa4fc8SDr. David Alan Gilbert {
150100fa4fc8SDr. David Alan Gilbert     guint i;
150200fa4fc8SDr. David Alan Gilbert     MigrationIncomingState *mis = migration_incoming_get_current();
150300fa4fc8SDr. David Alan Gilbert     GArray *pcrfds = mis->postcopy_remote_fds;
150400fa4fc8SDr. David Alan Gilbert 
150556559980SJuan Quintela     if (!pcrfds) {
150656559980SJuan Quintela         /* migration has already finished and freed the array */
150756559980SJuan Quintela         return;
150856559980SJuan Quintela     }
150900fa4fc8SDr. David Alan Gilbert     for (i = 0; i < pcrfds->len; i++) {
151000fa4fc8SDr. David Alan Gilbert         struct PostCopyFD *cur = &g_array_index(pcrfds, struct PostCopyFD, i);
151100fa4fc8SDr. David Alan Gilbert         if (cur->fd == pcfd->fd) {
151200fa4fc8SDr. David Alan Gilbert             mis->postcopy_remote_fds = g_array_remove_index(pcrfds, i);
151300fa4fc8SDr. David Alan Gilbert             return;
151400fa4fc8SDr. David Alan Gilbert         }
151500fa4fc8SDr. David Alan Gilbert     }
151600fa4fc8SDr. David Alan Gilbert }
1517