52 lines
1.3 KiB
C
52 lines
1.3 KiB
C
/* SPDX-License-Identifier: GPL-2.0-only */
|
|
#ifndef SELFTEST_KVM_UTIL_ARCH_H
|
|
#define SELFTEST_KVM_UTIL_ARCH_H
|
|
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
|
|
#include "kvm_util_types.h"
|
|
#include "test_util.h"
|
|
|
|
extern bool is_forced_emulation_enabled;
|
|
|
|
struct kvm_vm_arch {
|
|
vm_vaddr_t gdt;
|
|
vm_vaddr_t tss;
|
|
vm_vaddr_t idt;
|
|
|
|
uint64_t c_bit;
|
|
uint64_t s_bit;
|
|
int sev_fd;
|
|
bool is_pt_protected;
|
|
};
|
|
|
|
static inline bool __vm_arch_has_protected_memory(struct kvm_vm_arch *arch)
|
|
{
|
|
return arch->c_bit || arch->s_bit;
|
|
}
|
|
|
|
#define vm_arch_has_protected_memory(vm) \
|
|
__vm_arch_has_protected_memory(&(vm)->arch)
|
|
|
|
#define vcpu_arch_put_guest(mem, __val) \
|
|
do { \
|
|
const typeof(mem) val = (__val); \
|
|
\
|
|
if (!is_forced_emulation_enabled || guest_random_bool(&guest_rng)) { \
|
|
(mem) = val; \
|
|
} else if (guest_random_bool(&guest_rng)) { \
|
|
__asm__ __volatile__(KVM_FEP "mov %1, %0" \
|
|
: "+m" (mem) \
|
|
: "r" (val) : "memory"); \
|
|
} else { \
|
|
uint64_t __old = READ_ONCE(mem); \
|
|
\
|
|
__asm__ __volatile__(KVM_FEP LOCK_PREFIX "cmpxchg %[new], %[ptr]" \
|
|
: [ptr] "+m" (mem), [old] "+a" (__old) \
|
|
: [new]"r" (val) : "memory", "cc"); \
|
|
} \
|
|
} while (0)
|
|
|
|
#endif // SELFTEST_KVM_UTIL_ARCH_H
|