mirror of
https://github.com/torvalds/linux.git
synced 2026-05-25 23:52:08 +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>
111 lines
2.4 KiB
C
111 lines
2.4 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/*
|
|
* Copyright © 2021 Amazon.com, Inc. or its affiliates.
|
|
*/
|
|
|
|
#include "test_util.h"
|
|
#include "kvm_util.h"
|
|
#include "processor.h"
|
|
|
|
#include <stdint.h>
|
|
#include <time.h>
|
|
#include <sched.h>
|
|
#include <signal.h>
|
|
#include <pthread.h>
|
|
|
|
#define NR_TEST_VCPUS 20
|
|
|
|
static struct kvm_vm *vm;
|
|
pthread_spinlock_t create_lock;
|
|
|
|
#define TEST_TSC_KHZ 2345678UL
|
|
#define TEST_TSC_OFFSET 200000000
|
|
|
|
u64 tsc_sync;
|
|
static void guest_code(void)
|
|
{
|
|
u64 start_tsc, local_tsc, tmp;
|
|
|
|
start_tsc = rdtsc();
|
|
do {
|
|
tmp = READ_ONCE(tsc_sync);
|
|
local_tsc = rdtsc();
|
|
WRITE_ONCE(tsc_sync, local_tsc);
|
|
if (unlikely(local_tsc < tmp))
|
|
GUEST_SYNC_ARGS(0, local_tsc, tmp, 0, 0);
|
|
|
|
} while (local_tsc - start_tsc < 5000 * TEST_TSC_KHZ);
|
|
|
|
GUEST_DONE();
|
|
}
|
|
|
|
|
|
static void *run_vcpu(void *_cpu_nr)
|
|
{
|
|
unsigned long vcpu_id = (unsigned long)_cpu_nr;
|
|
unsigned long failures = 0;
|
|
static bool first_cpu_done;
|
|
struct kvm_vcpu *vcpu;
|
|
|
|
/* The kernel is fine, but vm_vcpu_add() needs locking */
|
|
pthread_spin_lock(&create_lock);
|
|
|
|
vcpu = vm_vcpu_add(vm, vcpu_id, guest_code);
|
|
|
|
if (!first_cpu_done) {
|
|
first_cpu_done = true;
|
|
vcpu_set_msr(vcpu, MSR_IA32_TSC, TEST_TSC_OFFSET);
|
|
}
|
|
|
|
pthread_spin_unlock(&create_lock);
|
|
|
|
for (;;) {
|
|
struct ucall uc;
|
|
|
|
vcpu_run(vcpu);
|
|
TEST_ASSERT_KVM_EXIT_REASON(vcpu, KVM_EXIT_IO);
|
|
|
|
switch (get_ucall(vcpu, &uc)) {
|
|
case UCALL_DONE:
|
|
goto out;
|
|
|
|
case UCALL_SYNC:
|
|
printf("Guest %d sync %lx %lx %ld\n", vcpu->id,
|
|
uc.args[2], uc.args[3], uc.args[2] - uc.args[3]);
|
|
failures++;
|
|
break;
|
|
|
|
default:
|
|
TEST_FAIL("Unknown ucall %lu", uc.cmd);
|
|
}
|
|
}
|
|
out:
|
|
return (void *)failures;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
TEST_REQUIRE(kvm_has_cap(KVM_CAP_VM_TSC_CONTROL));
|
|
|
|
vm = vm_create(NR_TEST_VCPUS);
|
|
vm_ioctl(vm, KVM_SET_TSC_KHZ, (void *) TEST_TSC_KHZ);
|
|
|
|
pthread_spin_init(&create_lock, PTHREAD_PROCESS_PRIVATE);
|
|
pthread_t cpu_threads[NR_TEST_VCPUS];
|
|
unsigned long cpu;
|
|
for (cpu = 0; cpu < NR_TEST_VCPUS; cpu++)
|
|
pthread_create(&cpu_threads[cpu], NULL, run_vcpu, (void *)cpu);
|
|
|
|
unsigned long failures = 0;
|
|
for (cpu = 0; cpu < NR_TEST_VCPUS; cpu++) {
|
|
void *this_cpu_failures;
|
|
pthread_join(cpu_threads[cpu], &this_cpu_failures);
|
|
failures += (unsigned long)this_cpu_failures;
|
|
}
|
|
|
|
TEST_ASSERT(!failures, "TSC sync failed");
|
|
pthread_spin_destroy(&create_lock);
|
|
kvm_vm_free(vm);
|
|
return 0;
|
|
}
|