Quelle  cr4_cpuid_sync_test.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * CR4 and CPUID sync test
 *
 * Copyright 2018, Red Hat, Inc. and/or its affiliates.
 *
 * Author:
 *   Wei Huang <wei@redhat.com>
 */

#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>

#include "test_util.h"

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

#define MAGIC_HYPERCALL_PORT 0x80

static void guest_code(void)
{
 u32 regs[4] = {
  [KVM_CPUID_EAX] = X86_FEATURE_OSXSAVE.function,
  [KVM_CPUID_ECX] = X86_FEATURE_OSXSAVE.index,
 };

 /* CR4.OSXSAVE should be enabled by default (for selftests vCPUs). */
 GUEST_ASSERT(get_cr4() & X86_CR4_OSXSAVE);

 /* verify CR4.OSXSAVE == CPUID.OSXSAVE */
 GUEST_ASSERT(this_cpu_has(X86_FEATURE_OSXSAVE));

 /*
  * Notify hypervisor to clear CR4.0SXSAVE, do CPUID and save output,
  * and then restore CR4.  Do this all in  assembly to ensure no AVX
  * instructions are executed while OSXSAVE=0.
 */
 asm volatile (
  "out %%al, $" __stringify(MAGIC_HYPERCALL_PORT) "\n\t"
  "cpuid\n\t"
  "mov %%rdi, %%cr4\n\t"
  : "+a" (regs[KVM_CPUID_EAX]),
    "=b" (regs[KVM_CPUID_EBX]),
    "+c" (regs[KVM_CPUID_ECX]),
    "=d" (regs[KVM_CPUID_EDX])
  : "D" (get_cr4())
 );

 /* Verify KVM cleared OSXSAVE in CPUID when it was cleared in CR4. */
 GUEST_ASSERT(!(regs[X86_FEATURE_OSXSAVE.reg] & BIT(X86_FEATURE_OSXSAVE.bit)));

 /* Verify restoring CR4 also restored OSXSAVE in CPUID. */
 GUEST_ASSERT(this_cpu_has(X86_FEATURE_OSXSAVE));

 GUEST_DONE();
}

int main(int argc, char *argv[])
{
 struct kvm_vcpu *vcpu;
 struct kvm_vm *vm;
 struct kvm_sregs sregs;
 struct ucall uc;

 TEST_REQUIRE(kvm_cpu_has(X86_FEATURE_XSAVE));

 vm = vm_create_with_one_vcpu(&vcpu, guest_code);

 while (1) {
  vcpu_run(vcpu);
  TEST_ASSERT_KVM_EXIT_REASON(vcpu, KVM_EXIT_IO);

  if (vcpu->run->io.port == MAGIC_HYPERCALL_PORT &&
      vcpu->run->io.direction == KVM_EXIT_IO_OUT) {
   /* emulate hypervisor clearing CR4.OSXSAVE */
   vcpu_sregs_get(vcpu, &sregs);
   sregs.cr4 &= ~X86_CR4_OSXSAVE;
   vcpu_sregs_set(vcpu, &sregs);
   continue;
  }

  switch (get_ucall(vcpu, &uc)) {
  case UCALL_ABORT:
   REPORT_GUEST_ASSERT(uc);
   break;
  case UCALL_DONE:
   goto done;
  default:
   TEST_FAIL("Unknown ucall %lu", uc.cmd);
  }
 }

done:
 kvm_vm_free(vm);
 return 0;
}