162306a36Sopenharmony_ci/* SPDX-License-Identifier: GPL-2.0 */
262306a36Sopenharmony_ci#ifndef __KVM_X86_PAGE_TRACK_H
362306a36Sopenharmony_ci#define __KVM_X86_PAGE_TRACK_H
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include <linux/kvm_host.h>
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <asm/kvm_page_track.h>
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci
1062306a36Sopenharmony_cibool kvm_page_track_write_tracking_enabled(struct kvm *kvm);
1162306a36Sopenharmony_ciint kvm_page_track_write_tracking_alloc(struct kvm_memory_slot *slot);
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_civoid kvm_page_track_free_memslot(struct kvm_memory_slot *slot);
1462306a36Sopenharmony_ciint kvm_page_track_create_memslot(struct kvm *kvm,
1562306a36Sopenharmony_ci				  struct kvm_memory_slot *slot,
1662306a36Sopenharmony_ci				  unsigned long npages);
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_civoid __kvm_write_track_add_gfn(struct kvm *kvm, struct kvm_memory_slot *slot,
1962306a36Sopenharmony_ci			       gfn_t gfn);
2062306a36Sopenharmony_civoid __kvm_write_track_remove_gfn(struct kvm *kvm,
2162306a36Sopenharmony_ci				  struct kvm_memory_slot *slot, gfn_t gfn);
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_cibool kvm_gfn_is_write_tracked(struct kvm *kvm,
2462306a36Sopenharmony_ci			      const struct kvm_memory_slot *slot, gfn_t gfn);
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci#ifdef CONFIG_KVM_EXTERNAL_WRITE_TRACKING
2762306a36Sopenharmony_ciint kvm_page_track_init(struct kvm *kvm);
2862306a36Sopenharmony_civoid kvm_page_track_cleanup(struct kvm *kvm);
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_civoid __kvm_page_track_write(struct kvm *kvm, gpa_t gpa, const u8 *new, int bytes);
3162306a36Sopenharmony_civoid kvm_page_track_delete_slot(struct kvm *kvm, struct kvm_memory_slot *slot);
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_cistatic inline bool kvm_page_track_has_external_user(struct kvm *kvm)
3462306a36Sopenharmony_ci{
3562306a36Sopenharmony_ci	return !hlist_empty(&kvm->arch.track_notifier_head.track_notifier_list);
3662306a36Sopenharmony_ci}
3762306a36Sopenharmony_ci#else
3862306a36Sopenharmony_cistatic inline int kvm_page_track_init(struct kvm *kvm) { return 0; }
3962306a36Sopenharmony_cistatic inline void kvm_page_track_cleanup(struct kvm *kvm) { }
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_cistatic inline void __kvm_page_track_write(struct kvm *kvm, gpa_t gpa,
4262306a36Sopenharmony_ci					  const u8 *new, int bytes) { }
4362306a36Sopenharmony_cistatic inline void kvm_page_track_delete_slot(struct kvm *kvm,
4462306a36Sopenharmony_ci					      struct kvm_memory_slot *slot) { }
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistatic inline bool kvm_page_track_has_external_user(struct kvm *kvm) { return false; }
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci#endif /* CONFIG_KVM_EXTERNAL_WRITE_TRACKING */
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_cistatic inline void kvm_page_track_write(struct kvm_vcpu *vcpu, gpa_t gpa,
5162306a36Sopenharmony_ci					const u8 *new, int bytes)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	__kvm_page_track_write(vcpu->kvm, gpa, new, bytes);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	kvm_mmu_track_write(vcpu, gpa, new, bytes);
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci#endif /* __KVM_X86_PAGE_TRACK_H */
59