1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * tools/testing/selftests/kvm/include/perf_test_util.h
4  *
5  * Copyright (C) 2020, Google LLC.
6  */
7 
8 #ifndef SELFTEST_KVM_PERF_TEST_UTIL_H
9 #define SELFTEST_KVM_PERF_TEST_UTIL_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 PERF_TEST_MEM_SLOT_INDEX	1
21 
22 struct perf_test_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 	int vcpu_id;
29 };
30 
31 struct perf_test_args {
32 	struct kvm_vm *vm;
33 	uint64_t gpa;
34 	uint64_t guest_page_size;
35 	int wr_fract;
36 
37 	struct perf_test_vcpu_args vcpu_args[KVM_MAX_VCPUS];
38 };
39 
40 extern struct perf_test_args perf_test_args;
41 
42 struct kvm_vm *perf_test_create_vm(enum vm_guest_mode mode, int vcpus,
43 				   uint64_t vcpu_memory_bytes, int slots,
44 				   enum vm_mem_backing_src_type backing_src,
45 				   bool partition_vcpu_memory_access);
46 void perf_test_destroy_vm(struct kvm_vm *vm);
47 
48 void perf_test_set_wr_fract(struct kvm_vm *vm, int wr_fract);
49 
50 void perf_test_start_vcpu_threads(int vcpus, void (*vcpu_fn)(struct perf_test_vcpu_args *));
51 void perf_test_join_vcpu_threads(int vcpus);
52 
53 #endif /* SELFTEST_KVM_PERF_TEST_UTIL_H */
54