tools/testing/selftests/kvm/include/memstress.h

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * tools/testing/selftests/kvm/include/memstress.h
   4  *
   5  * Copyright (C) 2020, Google LLC.
   6  */
   7
   8 #ifndef SELFTEST_KVM_MEMSTRESS_H
   9 #define SELFTEST_KVM_MEMSTRESS_H
  10
  11 #include <pthread.h>
  12
  13 #include "kvm_util.h"
  14
  15 /* Default guest test virtual memory offset */
  16 #define DEFAULT_GUEST_TEST_MEM          0xc0000000
  17
  18 #define DEFAULT_PER_VCPU_MEM_SIZE       (1 << 30) /* 1G */
  19
  20 #define MEMSTRESS_MEM_SLOT_INDEX        1
  21
  22 struct memstress_vcpu_args {
  23         uint64_t gpa;
  24         uint64_t gva;
  25         uint64_t pages;
  26
  27         /* Only used by the host userspace part of the vCPU thread */
  28         struct kvm_vcpu *vcpu;
  29         int vcpu_idx;
  30 };
  31
  32 struct memstress_args {
  33         struct kvm_vm *vm;
  34         /* The starting address and size of the guest test region. */
  35         uint64_t gpa;
  36         uint64_t size;
  37         uint64_t guest_page_size;
  38         uint32_t random_seed;
  39         uint32_t write_percent;
  40
  41         /* Run vCPUs in L2 instead of L1, if the architecture supports it. */
  42         bool nested;
  43         /* Randomize which pages are accessed by the guest. */
  44         bool random_access;
  45         /* True if all vCPUs are pinned to pCPUs */
  46         bool pin_vcpus;
  47         /* The vCPU=>pCPU pinning map. Only valid if pin_vcpus is true. */
  48         uint32_t vcpu_to_pcpu[KVM_MAX_VCPUS];
  49
  50         /* Test is done, stop running vCPUs. */
  51         bool stop_vcpus;
  52
  53         struct memstress_vcpu_args vcpu_args[KVM_MAX_VCPUS];
  54 };
  55
  56 extern struct memstress_args memstress_args;
  57
  58 struct kvm_vm *memstress_create_vm(enum vm_guest_mode mode, int nr_vcpus,
  59                                    uint64_t vcpu_memory_bytes, int slots,
  60                                    enum vm_mem_backing_src_type backing_src,
  61                                    bool partition_vcpu_memory_access);
  62 void memstress_destroy_vm(struct kvm_vm *vm);
  63
  64 void memstress_set_write_percent(struct kvm_vm *vm, uint32_t write_percent);
  65 void memstress_set_random_access(struct kvm_vm *vm, bool random_access);
  66
  67 void memstress_start_vcpu_threads(int vcpus, void (*vcpu_fn)(struct memstress_vcpu_args *));
  68 void memstress_join_vcpu_threads(int vcpus);
  69 void memstress_guest_code(uint32_t vcpu_id);
  70
  71 uint64_t memstress_nested_pages(int nr_vcpus);
  72 void memstress_setup_nested(struct kvm_vm *vm, int nr_vcpus, struct kvm_vcpu *vcpus[]);
  73
  74 void memstress_enable_dirty_logging(struct kvm_vm *vm, int slots);
  75 void memstress_disable_dirty_logging(struct kvm_vm *vm, int slots);
  76 void memstress_get_dirty_log(struct kvm_vm *vm, unsigned long *bitmaps[], int slots);
  77 void memstress_clear_dirty_log(struct kvm_vm *vm, unsigned long *bitmaps[],
  78                                int slots, uint64_t pages_per_slot);
  79 unsigned long **memstress_alloc_bitmaps(int slots, uint64_t pages_per_slot);
  80 void memstress_free_bitmaps(unsigned long *bitmaps[], int slots);
  81
  82 #endif /* SELFTEST_KVM_MEMSTRESS_H */