[linux-2.6-block.git] / tools / testing / selftests / kvm / system_counter_offset_test.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2021, Google LLC.
 *
 * Tests for adjusting the system counter from userspace
 */
#include <asm/kvm_para.h>
#include <stdint.h>
#include <string.h>
#include <sys/stat.h>
#include <time.h>

#include "test_util.h"
#include "kvm_util.h"
#include "processor.h"

#ifdef __x86_64__

struct test_case {
	uint64_t tsc_offset;
};

static struct test_case test_cases[] = {
	{ 0 },
	{ 180 * NSEC_PER_SEC },
	{ -180 * NSEC_PER_SEC },
};

static void check_preconditions(struct kvm_vcpu *vcpu)
{
	__TEST_REQUIRE(!__vcpu_has_device_attr(vcpu, KVM_VCPU_TSC_CTRL,
					       KVM_VCPU_TSC_OFFSET),
		       "KVM_VCPU_TSC_OFFSET not supported; skipping test");
}

static void setup_system_counter(struct kvm_vcpu *vcpu, struct test_case *test)
{
	vcpu_device_attr_set(vcpu, KVM_VCPU_TSC_CTRL, KVM_VCPU_TSC_OFFSET,
			     &test->tsc_offset);
}

static uint64_t guest_read_system_counter(struct test_case *test)
{
	return rdtsc();
}

static uint64_t host_read_guest_system_counter(struct test_case *test)
{
	return rdtsc() + test->tsc_offset;
}

#else /* __x86_64__ */

#error test not implemented for this architecture!

#endif

#define GUEST_SYNC_CLOCK(__stage, __val)			\
		GUEST_SYNC_ARGS(__stage, __val, 0, 0, 0)

static void guest_main(void)
{
	int i;

	for (i = 0; i < ARRAY_SIZE(test_cases); i++) {
		struct test_case *test = &test_cases[i];

		GUEST_SYNC_CLOCK(i, guest_read_system_counter(test));
	}
}

static void handle_sync(struct ucall *uc, uint64_t start, uint64_t end)
{
	uint64_t obs = uc->args[2];

	TEST_ASSERT(start <= obs && obs <= end,
		    "unexpected system counter value: %"PRIu64" expected range: [%"PRIu64", %"PRIu64"]",
		    obs, start, end);

	pr_info("system counter value: %"PRIu64" expected range [%"PRIu64", %"PRIu64"]\n",
		obs, start, end);
}

static void handle_abort(struct ucall *uc)
{
	REPORT_GUEST_ASSERT(*uc);
}

static void enter_guest(struct kvm_vcpu *vcpu)
{
	uint64_t start, end;
	struct ucall uc;
	int i;

	for (i = 0; i < ARRAY_SIZE(test_cases); i++) {
		struct test_case *test = &test_cases[i];

		setup_system_counter(vcpu, test);
		start = host_read_guest_system_counter(test);
		vcpu_run(vcpu);
		end = host_read_guest_system_counter(test);

		switch (get_ucall(vcpu, &uc)) {
		case UCALL_SYNC:
			handle_sync(&uc, start, end);
			break;
		case UCALL_ABORT:
			handle_abort(&uc);
			return;
		default:
			TEST_ASSERT(0, "unhandled ucall %ld",
				    get_ucall(vcpu, &uc));
		}
	}
}

int main(void)
{
	struct kvm_vcpu *vcpu;
	struct kvm_vm *vm;

	vm = vm_create_with_one_vcpu(&vcpu, guest_main);
	check_preconditions(vcpu);

	enter_guest(vcpu);
	kvm_vm_free(vm);
}
Commit	Line	Data
3f9808ca OU	1	// SPDX-License-Identifier: GPL-2.0-only
	2	/*
	3	* Copyright (C) 2021, Google LLC.
	4	*
	5	* Tests for adjusting the system counter from userspace
	6	*/
	7	#include <asm/kvm_para.h>
	8	#include <stdint.h>
	9	#include <string.h>
	10	#include <sys/stat.h>
	11	#include <time.h>
	12
	13	#include "test_util.h"
	14	#include "kvm_util.h"
	15	#include "processor.h"
	16
3f9808ca OU	17	#ifdef __x86_64__
	18
	19	struct test_case {
	20	uint64_t tsc_offset;
	21	};
	22
	23	static struct test_case test_cases[] = {
	24	{ 0 },
	25	{ 180 * NSEC_PER_SEC },
	26	{ -180 * NSEC_PER_SEC },
	27	};
	28
10f0b222	29	static void check_preconditions(struct kvm_vcpu *vcpu)
3f9808ca	30	{
7ed397d1 SC	31	__TEST_REQUIRE(!__vcpu_has_device_attr(vcpu, KVM_VCPU_TSC_CTRL,
	32	KVM_VCPU_TSC_OFFSET),
	33	"KVM_VCPU_TSC_OFFSET not supported; skipping test");
3f9808ca OU	34	}
3f9808ca OU	35
10f0b222	36	static void setup_system_counter(struct kvm_vcpu vcpu, struct test_case test)
3f9808ca	37	{
768e9a61 SC	38	vcpu_device_attr_set(vcpu, KVM_VCPU_TSC_CTRL, KVM_VCPU_TSC_OFFSET,
768e9a61 SC	39	&test->tsc_offset);
3f9808ca OU	40	}
	41
	42	static uint64_t guest_read_system_counter(struct test_case *test)
	43	{
	44	return rdtsc();
	45	}
	46
	47	static uint64_t host_read_guest_system_counter(struct test_case *test)
	48	{
	49	return rdtsc() + test->tsc_offset;
	50	}
	51
	52	#else /* __x86_64__ */
	53
	54	#error test not implemented for this architecture!
	55
	56	#endif
	57
	58	#define GUEST_SYNC_CLOCK(__stage, __val) \
	59	GUEST_SYNC_ARGS(__stage, __val, 0, 0, 0)
	60
	61	static void guest_main(void)
	62	{
	63	int i;
	64
	65	for (i = 0; i < ARRAY_SIZE(test_cases); i++) {
	66	struct test_case *test = &test_cases[i];
	67
	68	GUEST_SYNC_CLOCK(i, guest_read_system_counter(test));
	69	}
	70	}
	71
	72	static void handle_sync(struct ucall *uc, uint64_t start, uint64_t end)
	73	{
	74	uint64_t obs = uc->args[2];
	75
	76	TEST_ASSERT(start <= obs && obs <= end,
	77	"unexpected system counter value: %"PRIu64" expected range: [%"PRIu64", %"PRIu64"]",
	78	obs, start, end);
	79
	80	pr_info("system counter value: %"PRIu64" expected range [%"PRIu64", %"PRIu64"]\n",
	81	obs, start, end);
	82	}
	83
	84	static void handle_abort(struct ucall *uc)
	85	{
594a1c27	86	REPORT_GUEST_ASSERT(*uc);
3f9808ca OU	87	}
3f9808ca OU	88
10f0b222	89	static void enter_guest(struct kvm_vcpu *vcpu)
3f9808ca OU	90	{
	91	uint64_t start, end;
	92	struct ucall uc;
	93	int i;
	94
	95	for (i = 0; i < ARRAY_SIZE(test_cases); i++) {
	96	struct test_case *test = &test_cases[i];
	97
10f0b222	98	setup_system_counter(vcpu, test);
3f9808ca	99	start = host_read_guest_system_counter(test);
768e9a61	100	vcpu_run(vcpu);
3f9808ca OU	101	end = host_read_guest_system_counter(test);
3f9808ca OU	102
768e9a61	103	switch (get_ucall(vcpu, &uc)) {
3f9808ca OU	104	case UCALL_SYNC:
	105	handle_sync(&uc, start, end);
	106	break;
	107	case UCALL_ABORT:
	108	handle_abort(&uc);
	109	return;
	110	default:
250e138d	111	TEST_ASSERT(0, "unhandled ucall %ld",
768e9a61	112	get_ucall(vcpu, &uc));
3f9808ca OU	113	}
	114	}
	115	}
	116
	117	int main(void)
	118	{
10f0b222	119	struct kvm_vcpu *vcpu;
3f9808ca OU	120	struct kvm_vm *vm;
3f9808ca OU	121
10f0b222 SC	122	vm = vm_create_with_one_vcpu(&vcpu, guest_main);
10f0b222 SC	123	check_preconditions(vcpu);
3f9808ca	124
10f0b222	125	enter_guest(vcpu);
3f9808ca OU	126	kvm_vm_free(vm);
3f9808ca OU	127	}