Add a migration test to guest_memfd_test, run for the MMAP | INIT_SHARED configuration on systems with at least two NUMA nodes (skipped otherwise).
Migrate every folio from node 0 to node 1 with move_pages(2) and check both the resulting node and the data. Migrate them back and re-check the data. Signed-off-by: Shivank Garg <[email protected]> --- tools/testing/selftests/kvm/guest_memfd_test.c | 77 ++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) diff --git a/tools/testing/selftests/kvm/guest_memfd_test.c b/tools/testing/selftests/kvm/guest_memfd_test.c index 832ef4dfb99faa4411af847d21eb426c34342434..04931d3add46cb117fe5b093ed48f838cb124542 100644 --- a/tools/testing/selftests/kvm/guest_memfd_test.c +++ b/tools/testing/selftests/kvm/guest_memfd_test.c @@ -76,6 +76,82 @@ static void test_mmap_supported(int fd, size_t total_size) kvm_munmap(mem, total_size); } +/* + * Each page is filled with a distinct byte (its index). Check every byte that + * data is intact after migration. + */ +static void verify_page(const char *page, int page_idx, size_t size, + const char *when) +{ + char expected = (char)(page_idx & 0xff); + size_t off; + + for (off = 0; off < size; off++) + TEST_ASSERT(page[off] == expected, + "Page %d corrupted at offset %zu %s", page_idx, off, when); +} + +static void test_migrate_folio(int fd, size_t total_size) +{ + const unsigned long nodemask_0 = 1; /* nid: 0 */ + unsigned long maxnode = BITS_PER_TYPE(nodemask_0); + int page_count = total_size / page_size; + void **addr; + int *status, *nodes; + char *mem; + int i; + + if (!is_multi_numa_node_system()) + return; + + mem = kvm_mmap(total_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd); + + addr = calloc(page_count, sizeof(*addr)); + status = calloc(page_count, sizeof(*status)); + nodes = calloc(page_count, sizeof(*nodes)); + TEST_ASSERT(addr && status && nodes, "Failed to allocate page arrays"); + + /* Allocate all folios on node 0 and fill each with a known pattern. */ + kvm_mbind(mem, total_size, MPOL_BIND, &nodemask_0, maxnode, 0); + for (i = 0; i < page_count; i++) { + memset(mem + i * page_size, (char)(i & 0xff), page_size); + addr[i] = mem + i * page_size; + } + + kvm_move_pages(0, page_count, addr, NULL, status, 0); + for (i = 0; i < page_count; i++) + TEST_ASSERT(status[i] == 0, "Page %d should be on node 0", i); + + /* Migrate node 0 -> 1, then check both the location and the data. */ + for (i = 0; i < page_count; i++) + nodes[i] = 1; + kvm_move_pages(0, page_count, addr, nodes, status, MPOL_MF_MOVE); + + kvm_move_pages(0, page_count, addr, NULL, status, 0); + for (i = 0; i < page_count; i++) + TEST_ASSERT(status[i] == 1, + "Page %d should be on node 1 after migration", i); + for (i = 0; i < page_count; i++) + verify_page(mem + i * page_size, i, page_size, "after migration"); + + /* Migrate back node 1 -> 0, then re-check the location and the data. */ + for (i = 0; i < page_count; i++) + nodes[i] = 0; + kvm_move_pages(0, page_count, addr, nodes, status, MPOL_MF_MOVE); + + kvm_move_pages(0, page_count, addr, NULL, status, 0); + for (i = 0; i < page_count; i++) + TEST_ASSERT(status[i] == 0, + "Page %d should be on node 0 after round-trip", i); + for (i = 0; i < page_count; i++) + verify_page(mem + i * page_size, i, page_size, "after round-trip"); + + free(addr); + free(status); + free(nodes); + kvm_munmap(mem, total_size); +} + static void test_mbind(int fd, size_t total_size) { const unsigned long nodemask_0 = 1; /* nid: 0 */ @@ -434,6 +510,7 @@ static void __test_guest_memfd(struct kvm_vm *vm, u64 flags) gmem_test(fault_overflow, vm, flags); gmem_test(numa_allocation, vm, flags); __gmem_test(collapse, vm, flags, pmd_size); + gmem_test(migrate_folio, vm, flags); } else { gmem_test(fault_private, vm, flags); } -- 2.43.0
