max_vcpuid_cap_test.c

 1// SPDX-License-Identifier: GPL-2.0-only
 2/*
 3 * maximum APIC ID capability tests
 4 *
 5 * Copyright (C) 2022, Intel, Inc.
 6 *
 7 * Tests for getting/setting maximum APIC ID capability
 8 */
 9
10#include "kvm_util.h"
11
12#define MAX_VCPU_ID	2
13
14int main(int argc, char *argv[])
15{
16	struct kvm_vm *vm;
17	int ret;
18
19	vm = vm_create_barebones();
20
21	/* Get KVM_CAP_MAX_VCPU_ID cap supported in KVM */
22	ret = vm_check_cap(vm, KVM_CAP_MAX_VCPU_ID);
23
24	/* Try to set KVM_CAP_MAX_VCPU_ID beyond KVM cap */
25	ret = __vm_enable_cap(vm, KVM_CAP_MAX_VCPU_ID, ret + 1);
26	TEST_ASSERT(ret < 0,
27		    "Setting KVM_CAP_MAX_VCPU_ID beyond KVM cap should fail");
28
29	/* Test BOOT_CPU_ID interaction (MAX_VCPU_ID cannot be lower) */
30	if (kvm_has_cap(KVM_CAP_SET_BOOT_CPU_ID)) {
31		vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *)MAX_VCPU_ID);
32
33		/* Try setting KVM_CAP_MAX_VCPU_ID below BOOT_CPU_ID */
34		ret = __vm_enable_cap(vm, KVM_CAP_MAX_VCPU_ID, MAX_VCPU_ID - 1);
35		TEST_ASSERT(ret < 0,
36			    "Setting KVM_CAP_MAX_VCPU_ID below BOOT_CPU_ID should fail");
37	}
38
39	/* Set KVM_CAP_MAX_VCPU_ID */
40	vm_enable_cap(vm, KVM_CAP_MAX_VCPU_ID, MAX_VCPU_ID);
41
42	/* Try to set KVM_CAP_MAX_VCPU_ID again */
43	ret = __vm_enable_cap(vm, KVM_CAP_MAX_VCPU_ID, MAX_VCPU_ID + 1);
44	TEST_ASSERT(ret < 0,
45		    "Setting KVM_CAP_MAX_VCPU_ID multiple times should fail");
46
47	/* Create vCPU with id beyond KVM_CAP_MAX_VCPU_ID cap */
48	ret = __vm_ioctl(vm, KVM_CREATE_VCPU, (void *)MAX_VCPU_ID);
49	TEST_ASSERT(ret < 0, "Creating vCPU with ID > MAX_VCPU_ID should fail");
50
51	/* Create vCPU with bits 63:32 != 0, but an otherwise valid id */
52	ret = __vm_ioctl(vm, KVM_CREATE_VCPU, (void *)(1L << 32));
53	TEST_ASSERT(ret < 0, "Creating vCPU with ID[63:32] != 0 should fail");
54
55	/* Create vCPU with id within bounds */
56	ret = __vm_ioctl(vm, KVM_CREATE_VCPU, (void *)0);
57	TEST_ASSERT(ret >= 0, "Creating vCPU with ID 0 should succeed");
58
59	close(ret);
60	kvm_vm_free(vm);
61	return 0;
62}