mirror of
https://github.com/torvalds/linux.git
synced 2026-05-27 00:22:00 +02:00
26f8453288
Use u64 instead of uint64_t to make the KVM selftests code more concise and more similar to the kernel (since selftests are primarily developed by kernel developers). This commit was generated with the following command: git ls-files tools/testing/selftests/kvm | xargs sed -i 's/uint64_t/u64/g' Then by manually adjusting whitespace to make checkpatch.pl happy. Include <linux/types.h> in include/kvm_util_types.h, iinclude/test_util.h, and include/x86/pmu.h to pick up the tools-defined u64. Arguably, all headers (especially kvm_util_types.h) should have already been including stdint.h to get uint64_t from the libc headers, but the missing dependency only rears its head once KVM uses u64 instead of uint64_t. No functional change intended. Signed-off-by: David Matlack <dmatlack@google.com> [sean: rename pread_uint64() => pread_u64, expand on types.h include] Link: https://patch.msgid.link/20260420212004.3938325-5-seanjc@google.com Signed-off-by: Sean Christopherson <seanjc@google.com>
130 lines
3.1 KiB
C
130 lines
3.1 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/*
|
|
* x86-specific extensions to memstress.c.
|
|
*
|
|
* Copyright (C) 2022, Google, Inc.
|
|
*/
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <linux/bitmap.h>
|
|
#include <linux/bitops.h>
|
|
|
|
#include "test_util.h"
|
|
#include "kvm_util.h"
|
|
#include "memstress.h"
|
|
#include "processor.h"
|
|
#include "svm_util.h"
|
|
#include "vmx.h"
|
|
|
|
void memstress_l2_guest_code(u64 vcpu_id)
|
|
{
|
|
memstress_guest_code(vcpu_id);
|
|
vmcall();
|
|
}
|
|
|
|
extern char memstress_l2_guest_entry[];
|
|
__asm__(
|
|
"memstress_l2_guest_entry:"
|
|
" mov (%rsp), %rdi;"
|
|
" call memstress_l2_guest_code;"
|
|
" ud2;"
|
|
);
|
|
|
|
#define L2_GUEST_STACK_SIZE 64
|
|
|
|
static void l1_vmx_code(struct vmx_pages *vmx, u64 vcpu_id)
|
|
{
|
|
unsigned long l2_guest_stack[L2_GUEST_STACK_SIZE];
|
|
unsigned long *rsp;
|
|
|
|
GUEST_ASSERT(vmx->vmcs_gpa);
|
|
GUEST_ASSERT(prepare_for_vmx_operation(vmx));
|
|
GUEST_ASSERT(load_vmcs(vmx));
|
|
GUEST_ASSERT(ept_1g_pages_supported());
|
|
|
|
rsp = &l2_guest_stack[L2_GUEST_STACK_SIZE - 1];
|
|
*rsp = vcpu_id;
|
|
prepare_vmcs(vmx, memstress_l2_guest_entry, rsp);
|
|
|
|
GUEST_ASSERT(!vmlaunch());
|
|
GUEST_ASSERT_EQ(vmreadz(VM_EXIT_REASON), EXIT_REASON_VMCALL);
|
|
GUEST_DONE();
|
|
}
|
|
|
|
static void l1_svm_code(struct svm_test_data *svm, u64 vcpu_id)
|
|
{
|
|
unsigned long l2_guest_stack[L2_GUEST_STACK_SIZE];
|
|
unsigned long *rsp;
|
|
|
|
|
|
rsp = &l2_guest_stack[L2_GUEST_STACK_SIZE - 1];
|
|
*rsp = vcpu_id;
|
|
generic_svm_setup(svm, memstress_l2_guest_entry, rsp);
|
|
|
|
run_guest(svm->vmcb, svm->vmcb_gpa);
|
|
GUEST_ASSERT_EQ(svm->vmcb->control.exit_code, SVM_EXIT_VMMCALL);
|
|
GUEST_DONE();
|
|
}
|
|
|
|
|
|
static void memstress_l1_guest_code(void *data, u64 vcpu_id)
|
|
{
|
|
if (this_cpu_has(X86_FEATURE_VMX))
|
|
l1_vmx_code(data, vcpu_id);
|
|
else
|
|
l1_svm_code(data, vcpu_id);
|
|
}
|
|
|
|
u64 memstress_nested_pages(int nr_vcpus)
|
|
{
|
|
/*
|
|
* 513 page tables is enough to identity-map 256 TiB of L2 with 1G
|
|
* pages and 4-level paging, plus a few pages per-vCPU for data
|
|
* structures such as the VMCS.
|
|
*/
|
|
return 513 + 10 * nr_vcpus;
|
|
}
|
|
|
|
static void memstress_setup_ept_mappings(struct kvm_vm *vm)
|
|
{
|
|
u64 start, end;
|
|
|
|
/*
|
|
* Identity map the first 4G and the test region with 1G pages so that
|
|
* KVM can shadow the EPT12 with the maximum huge page size supported
|
|
* by the backing source.
|
|
*/
|
|
tdp_identity_map_1g(vm, 0, 0x100000000ULL);
|
|
|
|
start = align_down(memstress_args.gpa, PG_SIZE_1G);
|
|
end = align_up(memstress_args.gpa + memstress_args.size, PG_SIZE_1G);
|
|
tdp_identity_map_1g(vm, start, end - start);
|
|
}
|
|
|
|
void memstress_setup_nested(struct kvm_vm *vm, int nr_vcpus, struct kvm_vcpu *vcpus[])
|
|
{
|
|
struct kvm_regs regs;
|
|
gva_t nested_gva;
|
|
int vcpu_id;
|
|
|
|
TEST_REQUIRE(kvm_cpu_has_tdp());
|
|
|
|
vm_enable_tdp(vm);
|
|
memstress_setup_ept_mappings(vm);
|
|
for (vcpu_id = 0; vcpu_id < nr_vcpus; vcpu_id++) {
|
|
if (kvm_cpu_has(X86_FEATURE_VMX))
|
|
vcpu_alloc_vmx(vm, &nested_gva);
|
|
else
|
|
vcpu_alloc_svm(vm, &nested_gva);
|
|
|
|
/*
|
|
* Override the vCPU to run memstress_l1_guest_code() which will
|
|
* bounce it into L2 before calling memstress_guest_code().
|
|
*/
|
|
vcpu_regs_get(vcpus[vcpu_id], ®s);
|
|
regs.rip = (unsigned long) memstress_l1_guest_code;
|
|
vcpu_regs_set(vcpus[vcpu_id], ®s);
|
|
vcpu_args_set(vcpus[vcpu_id], 2, nested_gva, vcpu_id);
|
|
}
|
|
}
|