tools/testing/selftests/kvm/x86/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
  14 int 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 }