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 */
83
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.