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, ®_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, ®_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, ©_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