Implement arm64 version of kvm_arch_dirty_log_clear() making use of FEAT_HACDBS. It works by transversing the dirty-bitmap and converting the set bits into HDBSS entries in a 64-page blocks granularity. The resulting HDBSS array is then fed to the HACDBS mechanism that walks the pagetable marking writable-dirty pages as writable-clean. In case of error, rewrite all unprocessed entries, including the faulting one, to the dirty-bitmap and fall back to generic software cleaning. In case of the options to "manual protect + init set" are enabled, do the hugepage splitting in the same fashion as the generic software cleaning, i.e. in 64-page blocks. For that, remove the static qualifier from kvm_mmu_split_huge_pages() and make the function available on kvm_host.h. Signed-off-by: Leonardo Bras --- arch/arm64/include/asm/kvm_dirty_bit.h | 24 ++++ include/linux/kvm_host.h | 3 + arch/arm64/kvm/dirty_bit.c | 163 +++++++++++++++++++++++++ arch/arm64/kvm/mmu.c | 4 +- 4 files changed, 192 insertions(+), 2 deletions(-) diff --git a/arch/arm64/include/asm/kvm_dirty_bit.h b/arch/arm64/include/asm/kvm_dirty_bit.h index dd16438f0651..d368a23a84b3 100644 --- a/arch/arm64/include/asm/kvm_dirty_bit.h +++ b/arch/arm64/include/asm/kvm_dirty_bit.h @@ -2,11 +2,35 @@ /* * Copyright (C) 2026 ARM Ltd. * Author: Leonardo Bras */ #ifndef __ARM64_KVM_DIRTY_BIT_H__ #define __ARM64_KVM_DIRTY_BIT_H__ #include +int __kvm_arch_dirty_log_clear(struct kvm *kvm, + struct kvm_memory_slot *memslot, + struct kvm_clear_dirty_log *log, + unsigned long *bitmap, + bool *flush); + +static inline bool kvm_arch_dirty_clear_enabled(struct kvm *kvm) +{ + return system_supports_hacdbs() && kvm->arch.mmu.pgt && + (kvm->arch.mmu.pgt->flags & KVM_PGTABLE_S2_DBM); +} + +static inline int kvm_arch_dirty_log_clear(struct kvm *kvm, + struct kvm_memory_slot *memslot, + struct kvm_clear_dirty_log *log, + unsigned long *bitmap, + bool *flush) +{ + if (!kvm_arch_dirty_clear_enabled(kvm)) + return -EPERM; + + return __kvm_arch_dirty_log_clear(kvm, memslot, log, bitmap, flush); +} + #endif /* __ARM64_KVM_DIRTY_BIT_H__ */ diff --git a/include/linux/kvm_host.h b/include/linux/kvm_host.h index ab8cfaec82d3..7ea6ed7ce203 100644 --- a/include/linux/kvm_host.h +++ b/include/linux/kvm_host.h @@ -1662,20 +1662,23 @@ void kvm_arch_disable_virtualization_cpu(void); bool kvm_vcpu_has_events(struct kvm_vcpu *vcpu); int kvm_arch_vcpu_runnable(struct kvm_vcpu *vcpu); bool kvm_arch_vcpu_in_kernel(struct kvm_vcpu *vcpu); int kvm_arch_vcpu_should_kick(struct kvm_vcpu *vcpu); bool kvm_arch_dy_runnable(struct kvm_vcpu *vcpu); bool kvm_arch_dy_has_pending_interrupt(struct kvm_vcpu *vcpu); bool kvm_arch_vcpu_preempted_in_kernel(struct kvm_vcpu *vcpu); void kvm_arch_pre_destroy_vm(struct kvm *kvm); void kvm_arch_create_vm_debugfs(struct kvm *kvm); +int kvm_mmu_split_huge_pages(struct kvm *kvm, phys_addr_t addr, + phys_addr_t end); + #ifndef __KVM_HAVE_ARCH_VM_ALLOC /* * All architectures that want to use vzalloc currently also * need their own kvm_arch_alloc_vm implementation. */ static inline struct kvm *kvm_arch_alloc_vm(void) { return kzalloc_obj(struct kvm, GFP_KERNEL_ACCOUNT); } #endif diff --git a/arch/arm64/kvm/dirty_bit.c b/arch/arm64/kvm/dirty_bit.c index 33826bbd16d9..5a6266205295 100644 --- a/arch/arm64/kvm/dirty_bit.c +++ b/arch/arm64/kvm/dirty_bit.c @@ -183,10 +183,173 @@ static int dirty_bit_clear(struct kvm *kvm, u64 *hw_entries, int size) * later in generic dirty-cleaning code already performs a DSB before * doing the TLBI. */ preempt_disable(); preempt_notifier_unregister(&save.notifier); preempt_enable(); return ret; } + +static inline void hdbss_to_bitmap(u64 *hdbss_array, int start, int end, + unsigned long *dirty_bitmap, + unsigned long long offset) +{ + u64 w = (gpa_to_gfn(hdbss_array[start]) - offset) / BITS_PER_LONG; + u64 mask = 0; + int idx = start; + + /* + * Entries should be in order, since they were extracted from + * the dirty-bitmap, so batching the atomic writes is efficient. + */ + + do { + u64 entry = (gpa_to_gfn(hdbss_array[idx]) - offset); + + if (entry / BITS_PER_LONG == w) { + mask |= BIT(entry % BITS_PER_LONG); + } else { + atomic_long_or(mask, (atomic_long_t *)&dirty_bitmap[w]); + w = entry / BITS_PER_LONG; + mask = BIT(entry % BITS_PER_LONG); + } + } while (++idx < end); + atomic_long_or(mask, (atomic_long_t *)&dirty_bitmap[w]); +} + +static inline int mask_to_hdbss(unsigned long *mask, u64 *hw_entries, const gfn_t offset, + u64 ttwl, int idx, int entries_sz) +{ + while (idx < entries_sz) { + int j = __ffs(*mask); + u64 a = gfn_to_gpa(offset + j); + + hw_entries[idx++] = (a & HDBSS_ENTRY_IPA) | + ttwl | + HDBSS_ENTRY_VALID; + + *mask &= ~BIT(j); + if (!*mask) + break; + } + + return idx; +} + +int __kvm_arch_dirty_log_clear(struct kvm *kvm, + struct kvm_memory_slot *memslot, + struct kvm_clear_dirty_log *log, + unsigned long *bitmap, + bool *flush) +{ + int ret = 0; + int idx = 0; + unsigned long *dirty_bitmap = memslot->dirty_bitmap; + u64 *hw_entries; + const int entries_sz = PAGE_SIZE / sizeof(*hw_entries); + u64 ttwl; + u64 start, end; + gfn_t base_gfn; + + hw_entries = kmalloc_objs(u64, entries_sz, GFP_KERNEL); + if (!hw_entries) + return -ENOMEM; + + ttwl = HDBSS_ENTRY_TTWL(KVM_PGTABLE_LAST_LEVEL); + + if (log) { + start = log->first_page / BITS_PER_LONG; + end = start + DIV_ROUND_UP(log->num_pages, BITS_PER_LONG); + base_gfn = memslot->base_gfn + log->first_page % BITS_PER_LONG; + } else { + start = 0; + end = kvm_dirty_bitmap_bytes(memslot) / sizeof(long); + base_gfn = memslot->base_gfn; + } + + write_lock(&kvm->mmu_lock); + + for (unsigned long i = start; i < end; i++) { + unsigned long mask; + gfn_t offset; + atomic_long_t *p; + + if (log) { /* Clean only what is in the input bitmap */ + mask = bitmap[i - start]; + if (!mask) + continue; + + p = (atomic_long_t *)&dirty_bitmap[i]; + mask &= atomic_long_fetch_andnot(mask, p); + } else { /* Clean everything, save to output bitmap */ + if (!dirty_bitmap[i]) + continue; + + mask = xchg(&dirty_bitmap[i], 0); + bitmap[i] = mask; + } + + if (!mask) + continue; + + offset = base_gfn + i * BITS_PER_LONG; + + if (kvm_dirty_log_manual_protect_and_init_set(kvm)) + kvm_mmu_split_huge_pages(kvm, + gfn_to_gpa(offset + __ffs(mask)), + gfn_to_gpa(offset + __fls(mask) + 1)); + + do { + idx = mask_to_hdbss(&mask, hw_entries, offset, ttwl, idx, entries_sz); + if (idx >= entries_sz) { + ret = dirty_bit_clear(kvm, hw_entries, idx); + *flush = *flush || ret > 0; + if (ret != idx) { + /* Restore mask as before processing */ + atomic_long_or(mask, (atomic_long_t *)&dirty_bitmap[i]); + end = i + 1; + goto out_err; + } + idx = 0; + } + } while (mask); + } + + if (idx != 0) { + ret = dirty_bit_clear(kvm, hw_entries, idx); + *flush = *flush || ret > 0; + } +out_err: + if (unlikely(idx > 0 && ret != idx)) { + /* + * In case there is an error and not all entries in HACDBS get + * cleaned, we have to mark the dirty bits back in the bitmap, + * as that will be used by the software routine. + */ + + if (ret < idx) { + if (!log) { + /* + * We have to restore everything, cleaned up to + * now, because software routine uses xchg, and + * something may have been dirtied in the meantime + */ + for (int i = start; i < end; i++) + atomic_long_or(bitmap[i], + (atomic_long_t *)&dirty_bitmap[i]); + } else { + /* Restore only what have not been cleaned yet */ + hdbss_to_bitmap(hw_entries, ret, idx, dirty_bitmap, + memslot->base_gfn); + } + } + + ret = -EAGAIN; + } + + write_unlock(&kvm->mmu_lock); + kfree(hw_entries); + + return ret; +} diff --git a/arch/arm64/kvm/mmu.c b/arch/arm64/kvm/mmu.c index e086c01a9325..2f9d90c35668 100644 --- a/arch/arm64/kvm/mmu.c +++ b/arch/arm64/kvm/mmu.c @@ -110,22 +110,22 @@ static bool need_split_memcache_topup_or_resched(struct kvm *kvm) if (need_resched() || rwlock_needbreak(&kvm->mmu_lock)) return true; chunk_size = kvm->arch.mmu.split_page_chunk_size; min = kvm_mmu_split_nr_page_tables(chunk_size); cache = &kvm->arch.mmu.split_page_cache; return kvm_mmu_memory_cache_nr_free_objects(cache) < min; } -static int kvm_mmu_split_huge_pages(struct kvm *kvm, phys_addr_t addr, - phys_addr_t end) +int kvm_mmu_split_huge_pages(struct kvm *kvm, phys_addr_t addr, + phys_addr_t end) { struct kvm_mmu_memory_cache *cache; struct kvm_pgtable *pgt; int ret, cache_capacity; u64 next, chunk_size; lockdep_assert_held_write(&kvm->mmu_lock); chunk_size = kvm->arch.mmu.split_page_chunk_size; cache_capacity = kvm_mmu_split_nr_page_tables(chunk_size); -- 2.55.0