53 lines · c
1/* SPDX-License-Identifier: GPL-2.0 */2/*3 * KVM userfaultfd util4 *5 * Copyright (C) 2018, Red Hat, Inc.6 * Copyright (C) 2019-2022 Google LLC7 */8#include <inttypes.h>9#include <time.h>10#include <pthread.h>11#include <linux/userfaultfd.h>12 13#include "test_util.h"14 15typedef int (*uffd_handler_t)(int uffd_mode, int uffd, struct uffd_msg *msg);16 17struct uffd_reader_args {18 int uffd_mode;19 int uffd;20 useconds_t delay;21 uffd_handler_t handler;22 /* Holds the read end of the pipe for killing the reader. */23 int pipe;24};25 26struct uffd_desc {27 int uffd;28 uint64_t num_readers;29 /* Holds the write ends of the pipes for killing the readers. */30 int *pipefds;31 pthread_t *readers;32 struct uffd_reader_args *reader_args;33};34 35struct uffd_desc *uffd_setup_demand_paging(int uffd_mode, useconds_t delay,36 void *hva, uint64_t len,37 uint64_t num_readers,38 uffd_handler_t handler);39 40void uffd_stop_demand_paging(struct uffd_desc *uffd);41 42#ifdef PRINT_PER_PAGE_UPDATES43#define PER_PAGE_DEBUG(...) printf(__VA_ARGS__)44#else45#define PER_PAGE_DEBUG(...) _no_printf(__VA_ARGS__)46#endif47 48#ifdef PRINT_PER_VCPU_UPDATES49#define PER_VCPU_DEBUG(...) printf(__VA_ARGS__)50#else51#define PER_VCPU_DEBUG(...) _no_printf(__VA_ARGS__)52#endif53