mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-10-31 16:38:12 +00:00
39b2e5cae4
The madv_populate selftest currently builds with a warning when the local installed headers (via the distribution) don't include MADV_POPULATE_READ and MADV_POPULATE_WRITE. The warning is correct, because the test cannot locate the necessary header. The reason is that the in-tree installed headers (usr/include) have a "linux" instead of a "sys" subdirectory. Including "linux/mman.h" instead of "sys/mman.h" doesn't work (e.g., mmap() and madvise() are not defined that way). The only thing that seems to work is including "linux/mman.h" in addition to "sys/mman.h". We can get rid of our availability check and simplify. Link: https://lkml.kernel.org/r/20211015165758.41374-1-david@redhat.com Signed-off-by: David Hildenbrand <david@redhat.com> Reported-by: Shuah Khan <skhan@linuxfoundation.org> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
329 lines
8 KiB
C
329 lines
8 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/*
|
|
* MADV_POPULATE_READ and MADV_POPULATE_WRITE tests
|
|
*
|
|
* Copyright 2021, Red Hat, Inc.
|
|
*
|
|
* Author(s): David Hildenbrand <david@redhat.com>
|
|
*/
|
|
#define _GNU_SOURCE
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <linux/mman.h>
|
|
#include <sys/mman.h>
|
|
|
|
#include "../kselftest.h"
|
|
|
|
/*
|
|
* For now, we're using 2 MiB of private anonymous memory for all tests.
|
|
*/
|
|
#define SIZE (2 * 1024 * 1024)
|
|
|
|
static size_t pagesize;
|
|
|
|
static uint64_t pagemap_get_entry(int fd, char *start)
|
|
{
|
|
const unsigned long pfn = (unsigned long)start / pagesize;
|
|
uint64_t entry;
|
|
int ret;
|
|
|
|
ret = pread(fd, &entry, sizeof(entry), pfn * sizeof(entry));
|
|
if (ret != sizeof(entry))
|
|
ksft_exit_fail_msg("reading pagemap failed\n");
|
|
return entry;
|
|
}
|
|
|
|
static bool pagemap_is_populated(int fd, char *start)
|
|
{
|
|
uint64_t entry = pagemap_get_entry(fd, start);
|
|
|
|
/* Present or swapped. */
|
|
return entry & 0xc000000000000000ull;
|
|
}
|
|
|
|
static bool pagemap_is_softdirty(int fd, char *start)
|
|
{
|
|
uint64_t entry = pagemap_get_entry(fd, start);
|
|
|
|
return entry & 0x0080000000000000ull;
|
|
}
|
|
|
|
static void sense_support(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
addr = mmap(0, pagesize, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (!addr)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
|
|
ret = madvise(addr, pagesize, MADV_POPULATE_READ);
|
|
if (ret)
|
|
ksft_exit_skip("MADV_POPULATE_READ is not available\n");
|
|
|
|
ret = madvise(addr, pagesize, MADV_POPULATE_WRITE);
|
|
if (ret)
|
|
ksft_exit_skip("MADV_POPULATE_WRITE is not available\n");
|
|
|
|
munmap(addr, pagesize);
|
|
}
|
|
|
|
static void test_prot_read(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_READ, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_READ);
|
|
ksft_test_result(!ret, "MADV_POPULATE_READ with PROT_READ\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_WRITE);
|
|
ksft_test_result(ret == -1 && errno == EINVAL,
|
|
"MADV_POPULATE_WRITE with PROT_READ\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
static void test_prot_write(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_WRITE, MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_READ);
|
|
ksft_test_result(ret == -1 && errno == EINVAL,
|
|
"MADV_POPULATE_READ with PROT_WRITE\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_WRITE);
|
|
ksft_test_result(!ret, "MADV_POPULATE_WRITE with PROT_WRITE\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
static void test_holes(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
ret = munmap(addr + pagesize, pagesize);
|
|
if (ret)
|
|
ksft_exit_fail_msg("munmap failed\n");
|
|
|
|
/* Hole in the middle */
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_READ);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_READ with holes in the middle\n");
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_WRITE);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_WRITE with holes in the middle\n");
|
|
|
|
/* Hole at end */
|
|
ret = madvise(addr, 2 * pagesize, MADV_POPULATE_READ);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_READ with holes at the end\n");
|
|
ret = madvise(addr, 2 * pagesize, MADV_POPULATE_WRITE);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_WRITE with holes at the end\n");
|
|
|
|
/* Hole at beginning */
|
|
ret = madvise(addr + pagesize, pagesize, MADV_POPULATE_READ);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_READ with holes at the beginning\n");
|
|
ret = madvise(addr + pagesize, pagesize, MADV_POPULATE_WRITE);
|
|
ksft_test_result(ret == -1 && errno == ENOMEM,
|
|
"MADV_POPULATE_WRITE with holes at the beginning\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
static bool range_is_populated(char *start, ssize_t size)
|
|
{
|
|
int fd = open("/proc/self/pagemap", O_RDONLY);
|
|
bool ret = true;
|
|
|
|
if (fd < 0)
|
|
ksft_exit_fail_msg("opening pagemap failed\n");
|
|
for (; size > 0 && ret; size -= pagesize, start += pagesize)
|
|
if (!pagemap_is_populated(fd, start))
|
|
ret = false;
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
static bool range_is_not_populated(char *start, ssize_t size)
|
|
{
|
|
int fd = open("/proc/self/pagemap", O_RDONLY);
|
|
bool ret = true;
|
|
|
|
if (fd < 0)
|
|
ksft_exit_fail_msg("opening pagemap failed\n");
|
|
for (; size > 0 && ret; size -= pagesize, start += pagesize)
|
|
if (pagemap_is_populated(fd, start))
|
|
ret = false;
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
static void test_populate_read(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
ksft_test_result(range_is_not_populated(addr, SIZE),
|
|
"range initially not populated\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_READ);
|
|
ksft_test_result(!ret, "MADV_POPULATE_READ\n");
|
|
ksft_test_result(range_is_populated(addr, SIZE),
|
|
"range is populated\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
static void test_populate_write(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
ksft_test_result(range_is_not_populated(addr, SIZE),
|
|
"range initially not populated\n");
|
|
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_WRITE);
|
|
ksft_test_result(!ret, "MADV_POPULATE_WRITE\n");
|
|
ksft_test_result(range_is_populated(addr, SIZE),
|
|
"range is populated\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
static bool range_is_softdirty(char *start, ssize_t size)
|
|
{
|
|
int fd = open("/proc/self/pagemap", O_RDONLY);
|
|
bool ret = true;
|
|
|
|
if (fd < 0)
|
|
ksft_exit_fail_msg("opening pagemap failed\n");
|
|
for (; size > 0 && ret; size -= pagesize, start += pagesize)
|
|
if (!pagemap_is_softdirty(fd, start))
|
|
ret = false;
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
static bool range_is_not_softdirty(char *start, ssize_t size)
|
|
{
|
|
int fd = open("/proc/self/pagemap", O_RDONLY);
|
|
bool ret = true;
|
|
|
|
if (fd < 0)
|
|
ksft_exit_fail_msg("opening pagemap failed\n");
|
|
for (; size > 0 && ret; size -= pagesize, start += pagesize)
|
|
if (pagemap_is_softdirty(fd, start))
|
|
ret = false;
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
static void clear_softdirty(void)
|
|
{
|
|
int fd = open("/proc/self/clear_refs", O_WRONLY);
|
|
const char *ctrl = "4";
|
|
int ret;
|
|
|
|
if (fd < 0)
|
|
ksft_exit_fail_msg("opening clear_refs failed\n");
|
|
ret = write(fd, ctrl, strlen(ctrl));
|
|
if (ret != strlen(ctrl))
|
|
ksft_exit_fail_msg("writing clear_refs failed\n");
|
|
close(fd);
|
|
}
|
|
|
|
static void test_softdirty(void)
|
|
{
|
|
char *addr;
|
|
int ret;
|
|
|
|
ksft_print_msg("[RUN] %s\n", __func__);
|
|
|
|
addr = mmap(0, SIZE, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE, 0, 0);
|
|
if (addr == MAP_FAILED)
|
|
ksft_exit_fail_msg("mmap failed\n");
|
|
|
|
/* Clear any softdirty bits. */
|
|
clear_softdirty();
|
|
ksft_test_result(range_is_not_softdirty(addr, SIZE),
|
|
"range is not softdirty\n");
|
|
|
|
/* Populating READ should set softdirty. */
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_READ);
|
|
ksft_test_result(!ret, "MADV_POPULATE_READ\n");
|
|
ksft_test_result(range_is_not_softdirty(addr, SIZE),
|
|
"range is not softdirty\n");
|
|
|
|
/* Populating WRITE should set softdirty. */
|
|
ret = madvise(addr, SIZE, MADV_POPULATE_WRITE);
|
|
ksft_test_result(!ret, "MADV_POPULATE_WRITE\n");
|
|
ksft_test_result(range_is_softdirty(addr, SIZE),
|
|
"range is softdirty\n");
|
|
|
|
munmap(addr, SIZE);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int err;
|
|
|
|
pagesize = getpagesize();
|
|
|
|
ksft_print_header();
|
|
ksft_set_plan(21);
|
|
|
|
sense_support();
|
|
test_prot_read();
|
|
test_prot_write();
|
|
test_holes();
|
|
test_populate_read();
|
|
test_populate_write();
|
|
test_softdirty();
|
|
|
|
err = ksft_get_fail_cnt();
|
|
if (err)
|
|
ksft_exit_fail_msg("%d out of %d tests failed\n",
|
|
err, ksft_test_num());
|
|
return ksft_exit_pass();
|
|
}
|