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 
10 void kvm_mmu_init_tdp_mmu(struct kvm *kvm);
11 void kvm_mmu_uninit_tdp_mmu(struct kvm *kvm);
12 
13 int kvm_tdp_mmu_alloc_root(struct kvm_vcpu *vcpu);
14 
kvm_tdp_mmu_get_root(struct kvm_mmu_page * root)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 
20 void kvm_tdp_mmu_put_root(struct kvm *kvm, struct kvm_mmu_page *root);
21 
22 bool kvm_tdp_mmu_zap_leafs(struct kvm *kvm, gfn_t start, gfn_t end, bool flush);
23 bool kvm_tdp_mmu_zap_sp(struct kvm *kvm, struct kvm_mmu_page *sp);
24 void kvm_tdp_mmu_zap_all(struct kvm *kvm);
25 void kvm_tdp_mmu_invalidate_all_roots(struct kvm *kvm);
26 void kvm_tdp_mmu_zap_invalidated_roots(struct kvm *kvm);
27 
28 int kvm_tdp_mmu_map(struct kvm_vcpu *vcpu, struct kvm_page_fault *fault);
29 
30 bool kvm_tdp_mmu_unmap_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range,
31 				 bool flush);
32 bool kvm_tdp_mmu_age_gfn_range(struct kvm *kvm, struct kvm_gfn_range *range);
33 bool kvm_tdp_mmu_test_age_gfn(struct kvm *kvm, struct kvm_gfn_range *range);
34 
35 bool kvm_tdp_mmu_wrprot_slot(struct kvm *kvm,
36 			     const struct kvm_memory_slot *slot, int min_level);
37 bool kvm_tdp_mmu_clear_dirty_slot(struct kvm *kvm,
38 				  const struct kvm_memory_slot *slot);
39 void kvm_tdp_mmu_clear_dirty_pt_masked(struct kvm *kvm,
40 				       struct kvm_memory_slot *slot,
41 				       gfn_t gfn, unsigned long mask,
42 				       bool wrprot);
43 void kvm_tdp_mmu_zap_collapsible_sptes(struct kvm *kvm,
44 				       const struct kvm_memory_slot *slot);
45 
46 bool kvm_tdp_mmu_write_protect_gfn(struct kvm *kvm,
47 				   struct kvm_memory_slot *slot, gfn_t gfn,
48 				   int min_level);
49 
50 void kvm_tdp_mmu_try_split_huge_pages(struct kvm *kvm,
51 				      const struct kvm_memory_slot *slot,
52 				      gfn_t start, gfn_t end,
53 				      int target_level, bool shared);
54 
kvm_tdp_mmu_walk_lockless_begin(void)55 static inline void kvm_tdp_mmu_walk_lockless_begin(void)
56 {
57 	rcu_read_lock();
58 }
59 
kvm_tdp_mmu_walk_lockless_end(void)60 static inline void kvm_tdp_mmu_walk_lockless_end(void)
61 {
62 	rcu_read_unlock();
63 }
64 
65 int kvm_tdp_mmu_get_walk(struct kvm_vcpu *vcpu, u64 addr, u64 *sptes,
66 			 int *root_level);
67 u64 *kvm_tdp_mmu_fast_pf_get_last_sptep(struct kvm_vcpu *vcpu, gfn_t gfn,
68 					u64 *spte);
69 
70 #ifdef CONFIG_X86_64
is_tdp_mmu_page(struct kvm_mmu_page * sp)71 static inline bool is_tdp_mmu_page(struct kvm_mmu_page *sp) { return sp->tdp_mmu_page; }
72 #else
is_tdp_mmu_page(struct kvm_mmu_page * sp)73 static inline bool is_tdp_mmu_page(struct kvm_mmu_page *sp) { return false; }
74 #endif
75 
76 #endif /* __KVM_X86_MMU_TDP_MMU_H */
77