tdp_mmu.h

  1// SPDX-License-Identifier: GPL-2.0
  2
  3#ifndef __KVM_X86_MMU_TDP_MMU_H
  4#define __KVM_X86_MMU_TDP_MMU_H
  5
  6#include <linux/kvm_host.h>
  7
  8#include "spte.h"
  9
 10void kvm_mmu_init_tdp_mmu(struct kvm *kvm);
 11void kvm_mmu_uninit_tdp_mmu(struct kvm *kvm);
 12
 13void kvm_tdp_mmu_alloc_root(struct kvm_vcpu *vcpu, bool private);
 14
 15__must_check static inline bool kvm_tdp_mmu_get_root(struct kvm_mmu_page *root)
 16{
 17	return refcount_inc_not_zero(&root->tdp_mmu_root_count);
 18}
 19
 20void kvm_tdp_mmu_put_root(struct kvm *kvm, struct kvm_mmu_page *root);
 21
 22enum kvm_tdp_mmu_root_types {
 23	KVM_INVALID_ROOTS = BIT(0),
 24	KVM_DIRECT_ROOTS = BIT(1),
 25	KVM_MIRROR_ROOTS = BIT(2),
 26	KVM_VALID_ROOTS = KVM_DIRECT_ROOTS | KVM_MIRROR_ROOTS,
 27	KVM_ALL_ROOTS = KVM_VALID_ROOTS | KVM_INVALID_ROOTS,
 28};
 29
 30static inline enum kvm_tdp_mmu_root_types kvm_gfn_range_filter_to_root_types(struct kvm *kvm,
 31							     enum kvm_gfn_range_filter process)
 32{
 33	enum kvm_tdp_mmu_root_types ret = 0;
 34
 35	if (!kvm_has_mirrored_tdp(kvm))
 36		return KVM_DIRECT_ROOTS;
 37
 38	if (process & KVM_FILTER_PRIVATE)
 39		ret |= KVM_MIRROR_ROOTS;
 40	if (process & KVM_FILTER_SHARED)
 41		ret |= KVM_DIRECT_ROOTS;
 42
 43	WARN_ON_ONCE(!ret);
 44
 45	return ret;
 46}
 47
 48static inline struct kvm_mmu_page *tdp_mmu_get_root_for_fault(struct kvm_vcpu *vcpu,
 49							      struct kvm_page_fault *fault)
 50{
 51	if (unlikely(!kvm_is_addr_direct(vcpu->kvm, fault->addr)))
 52		return root_to_sp(vcpu->arch.mmu->mirror_root_hpa);
 53
 54	return root_to_sp(vcpu->arch.mmu->root.hpa);
 55}
 56
 57static inline struct kvm_mmu_page *tdp_mmu_get_root(struct kvm_vcpu *vcpu,
 58						    enum kvm_tdp_mmu_root_types type)
 59{
 60	if (unlikely(type == KVM_MIRROR_ROOTS))
 61		return root_to_sp(vcpu->arch.mmu->mirror_root_hpa);
 62
 63	return root_to_sp(vcpu->arch.mmu->root.hpa);
 64}
 65
 66bool kvm_tdp_mmu_zap_leafs(struct kvm *kvm, gfn_t start, gfn_t end, bool flush);
 67bool kvm_tdp_mmu_zap_possible_nx_huge_page(struct kvm *kvm,
 68					   struct kvm_mmu_page *sp);
 69void kvm_tdp_mmu_zap_all(struct kvm *kvm);
 70void kvm_tdp_mmu_invalidate_roots(struct kvm *kvm,
 71				  enum kvm_tdp_mmu_root_types root_types);
 72void kvm_tdp_mmu_zap_invalidated_roots(struct kvm *kvm, bool shared);
 73
 74int kvm_tdp_mmu_map(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault);
 75
 76bool kvm_tdp_mmu_unmap_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range,
 77				 bool flush);
 78bool kvm_tdp_mmu_age_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range);
 79bool kvm_tdp_mmu_test_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range);
 80
 81bool kvm_tdp_mmu_wrprot_slot(struct kvm *kvm,
 82			     const struct kvm_memory_slot *slot, int min_level);
 83void kvm_tdp_mmu_clear_dirty_slot(struct kvm *kvm,
 84				  const struct kvm_memory_slot *slot);
 85void kvm_tdp_mmu_clear_dirty_pt_masked(struct kvm *kvm,
 86				       struct kvm_memory_slot *slot,
 87				       gfn_t gfn, unsigned long mask,
 88				       bool wrprot);
 89void kvm_tdp_mmu_recover_huge_pages(struct kvm *kvm,
 90				    const struct kvm_memory_slot *slot);
 91
 92bool kvm_tdp_mmu_write_protect_gfn(struct kvm *kvm,
 93				   struct kvm_memory_slot *slot, gfn_t gfn,
 94				   int min_level);
 95
 96void kvm_tdp_mmu_try_split_huge_pages(struct kvm *kvm,
 97				      const struct kvm_memory_slot *slot,
 98				      gfn_t start, gfn_t end,
 99				      int target_level, bool shared);
100
101static inline void kvm_tdp_mmu_walk_lockless_begin(void)
102{
103	rcu_read_lock();
104}
105
106static inline void kvm_tdp_mmu_walk_lockless_end(void)
107{
108	rcu_read_unlock();
109}
110
111int kvm_tdp_mmu_get_walk(struct kvm_vcpu *vcpu, u64 addr, u64 *sptes,
112			 int *root_level);
113u64 *kvm_tdp_mmu_fast_pf_get_last_sptep(struct kvm_vcpu *vcpu, gfn_t gfn,
114					u64 *spte);
115
116#ifdef CONFIG_X86_64
117static inline bool is_tdp_mmu_page(struct kvm_mmu_page *sp) { return sp->tdp_mmu_page; }
118#else
119static inline bool is_tdp_mmu_page(struct kvm_mmu_page *sp) { return false; }
120#endif
121
122#endif /* __KVM_X86_MMU_TDP_MMU_H */