83 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * tools/testing/selftests/kvm/include/memstress.h4 *5 * Copyright (C) 2020, Google LLC.6 */7 8#ifndef SELFTEST_KVM_MEMSTRESS_H9#define SELFTEST_KVM_MEMSTRESS_H10 11#include <pthread.h>12 13#include "kvm_util.h"14 15/* Default guest test virtual memory offset */16#define DEFAULT_GUEST_TEST_MEM 0xc000000017 18#define DEFAULT_PER_VCPU_MEM_SIZE (1 << 30) /* 1G */19 20#define MEMSTRESS_MEM_SLOT_INDEX 121 22struct 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 32struct 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 56extern struct memstress_args memstress_args;57 58struct 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);62void memstress_destroy_vm(struct kvm_vm *vm);63 64void memstress_set_write_percent(struct kvm_vm *vm, uint32_t write_percent);65void memstress_set_random_access(struct kvm_vm *vm, bool random_access);66 67void memstress_start_vcpu_threads(int vcpus, void (*vcpu_fn)(struct memstress_vcpu_args *));68void memstress_join_vcpu_threads(int vcpus);69void memstress_guest_code(uint32_t vcpu_id);70 71uint64_t memstress_nested_pages(int nr_vcpus);72void memstress_setup_nested(struct kvm_vm *vm, int nr_vcpus, struct kvm_vcpu *vcpus[]);73 74void memstress_enable_dirty_logging(struct kvm_vm *vm, int slots);75void memstress_disable_dirty_logging(struct kvm_vm *vm, int slots);76void memstress_get_dirty_log(struct kvm_vm *vm, unsigned long *bitmaps[], int slots);77void memstress_clear_dirty_log(struct kvm_vm *vm, unsigned long *bitmaps[],78 int slots, uint64_t pages_per_slot);79unsigned long **memstress_alloc_bitmaps(int slots, uint64_t pages_per_slot);80void memstress_free_bitmaps(unsigned long *bitmaps[], int slots);81 82#endif /* SELFTEST_KVM_MEMSTRESS_H */83