Re: [PATCH v4 5/6] KVM: arm64: Add HDBSS fault handling and buffer flush

Tian Zheng <[email protected]> Tue, 4 Aug 2026 16:26:27 +0800
Newsgroups dev.linux.lists.kvmarm,org.infradead.lists.linux-arm-kernel,org.kernel.vger.kvm,org.kernel.vger.linux-kernel
Message-ID <[email protected]>

On 8/3/2026 6:43 PM, Leonardo Bras wrote:
> On Mon, Aug 03, 2026 at 11:15:45AM +0800, Tian Zheng wrote:
>>
>>
>> On 7/29/2026 11:30 PM, Leonardo Bras wrote:
>>> On Tue, Jul 28, 2026 at 03:52:05PM +0800, Tian Zheng wrote:
>>>>
>>>>
>>>> On 7/21/2026 10:18 PM, Leonardo Bras wrote:
>>>>> On Tue, Jul 21, 2026 at 04:53:08PM +0800, Inochi Amaoto wrote:
>>>>>> On Fri, Jul 17, 2026 at 04:44:24PM +0100, Leonardo Bras wrote:
>>>>>>> On Fri, Jul 17, 2026 at 02:51:12PM +0800, Tian Zheng wrote:
>>>>>>>>
>>>>>>>> On 7/14/2026 10:19 PM, Leonardo Bras wrote:
>>>>>>>>> On Tue, Jul 14, 2026 at 09:27:15PM +0800, Tian Zheng wrote:
>>>>>>>>>> On 7/14/2026 6:50 PM, Leonardo Bras wrote:
>>>>>>>>>>> On Tue, Jul 14, 2026 at 03:38:39PM +0800, Tian Zheng wrote:
>>>>>>>>>>>> On 7/13/2026 10:06 PM, Leonardo Bras wrote:
>>>>>>>>>>>>> On Thu, Jul 09, 2026 at 06:40:25PM +0800, Tian Zheng wrote:
>>>>>>>>>>>>>> From: eillon <[email protected]>
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> Add HDBSS fault handling for buffer full, external abort, and general
>>>>>>>>>>>>>> protection fault (GPF) events. When the HDBSS buffer becomes full,
>>>>>>>>>>>>>> the hardware traps to EL2 with an HDBSSF event, which is handled by
>>>>>>>>>>>>>> setting a flush request.
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> Add kvm_flush_hdbss_buffer() to consume HDBSS buffer entries and
>>>>>>>>>>>>>> propagate dirty information into the userspace-visible dirty bitmap.
>>>>>>>>>>>>>> Flush is triggered on vcpu_put, check_vcpu_requests, and
>>>>>>>>>>>>>> sync_dirty_log.
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> Add esr_iss2_is_hdbssf() helper for HDBSS fault detection in guest
>>>>>>>>>>>>>> abort handling.
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> Signed-off-by: Eillon <[email protected]>
>>>>>>>>>>>>>> Signed-off-by: Tian Zheng <[email protected]>
>>>>>>>>>>>>>> ---
>>>>>>>>>>>>>>        arch/arm64/include/asm/esr.h           |  5 +++
>>>>>>>>>>>>>>        arch/arm64/include/asm/kvm_dirty_bit.h | 11 +++++
>>>>>>>>>>>>>>        arch/arm64/include/asm/kvm_host.h      |  1 +
>>>>>>>>>>>>>>        arch/arm64/kvm/arm.c                   | 14 ++++++
>>>>>>>>>>>>>>        arch/arm64/kvm/dirty_bit.c             | 62 ++++++++++++++++++++++++++
>>>>>>>>>>>>>>        arch/arm64/kvm/mmu.c                   |  4 ++
>>>>>>>>>>>>>>        6 files changed, 97 insertions(+)
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> diff --git a/arch/arm64/include/asm/esr.h b/arch/arm64/include/asm/esr.h
>>>>>>>>>>>>>> index 81c17320a588..2e6b679b5908 100644
>>>>>>>>>>>>>> --- a/arch/arm64/include/asm/esr.h
>>>>>>>>>>>>>> +++ b/arch/arm64/include/asm/esr.h
>>>>>>>>>>>>>> @@ -437,6 +437,11 @@
>>>>>>>>>>>>>>        #ifndef __ASSEMBLER__
>>>>>>>>>>>>>>        #include <asm/types.h>
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> +static inline bool esr_iss2_is_hdbssf(unsigned long esr)
>>>>>>>>>>>>>> +{
>>>>>>>>>>>>>> +	return ESR_ELx_ISS2(esr) & ESR_ELx_HDBSSF;
>>>>>>>>>>>>> This will return a long, which will be casted as bool.
>>>>>>>>>>>>> In general, what I see in the kernel is something like:
>>>>>>>>>>>>>
>>>>>>>>>>>>> 	return !!(ESR_ELx_ISS2(esr) & ESR_ELx_HDBSSF)
>>>>>>>>>>>> ok!
>>>>>>>>>>>>
>>>>>>>>>>>>
>>>>>>>>>>>>>> +}
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>>        static inline unsigned long esr_brk_comment(unsigned long esr)
>>>>>>>>>>>>>>        {
>>>>>>>>>>>>>>        	return esr & ESR_ELx_BRK64_ISS_COMMENT_MASK;
>>>>>>>>>>>>>> diff --git a/arch/arm64/include/asm/kvm_dirty_bit.h b/arch/arm64/include/asm/kvm_dirty_bit.h
>>>>>>>>>>>>>> index 84b12f0a10af..4b28000e972f 100644
>>>>>>>>>>>>>> --- a/arch/arm64/include/asm/kvm_dirty_bit.h
>>>>>>>>>>>>>> +++ b/arch/arm64/include/asm/kvm_dirty_bit.h
>>>>>>>>>>>>>> @@ -10,7 +10,18 @@
>>>>>>>>>>>>>>        #include <asm/kvm_pgtable.h>
>>>>>>>>>>>>>>        #include <asm/sysreg.h>
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> +/* HDBSS entry field definitions */
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_VALID	BIT(0)
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_TTWL_SHIFT (1)
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_TTWL_MASK (GENMASK(3, 1))
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_TTWL(x) \
>>>>>>>>>>>>>> +	(((x) << HDBSS_ENTRY_TTWL_SHIFT) & HDBSS_ENTRY_TTWL_MASK)
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_TTWL_RESV HDBSS_ENTRY_TTWL(-4)
>>>>>>>>>>>>>> +#define HDBSS_ENTRY_IPA	GENMASK_ULL(55, 12)
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>>        int kvm_arm_vcpu_alloc_hdbss(struct kvm_vcpu *vcpu, unsigned int order);
>>>>>>>>>>>>>>        void kvm_arm_vcpu_free_hdbss(struct kvm_vcpu *vcpu);
>>>>>>>>>>>>>> +void kvm_flush_hdbss_buffer(struct kvm_vcpu *vcpu);
>>>>>>>>>>>>>> +int kvm_handle_hdbss_fault(struct kvm_vcpu *vcpu);
>>>>>>>>>>>>>>
>>>>>>>>>>>>>>        #endif /* __ARM64_KVM_DIRTY_BIT_H__ */
>>>>>>>>>>>>>> diff --git a/arch/arm64/include/asm/kvm_host.h b/arch/arm64/include/asm/kvm_host.h
>>>>>>>>>>>>>> index c41ec6d9c45a..cecfb884a64f 100644
>>>>>>>>>>>>>> --- a/arch/arm64/include/asm/kvm_host.h
>>>>>>>>>>>>>> +++ b/arch/arm64/include/asm/kvm_host.h
>>>>>>>>>>>>>> @@ -55,6 +55,7 @@
>>>>>>>>>>>>>>        #define KVM_REQ_GUEST_HYP_IRQ_PENDING	KVM_ARCH_REQ(9)
>>>>>>>>>>>>>>        #define KVM_REQ_MAP_L1_VNCR_EL2		KVM_ARCH_REQ(10)
>>>>>>>>>>>>>>        #define KVM_REQ_VGIC_PROCESS_UPDATE	KVM_ARCH_REQ(11)
>>>>>>>>>>>>>> +#define KVM_REQ_FLUSH_HDBSS			KVM_ARCH_REQ(12)
>>>>>>>>>>>>>>
>>>>>>>>>>>>>>        #define KVM_DIRTY_LOG_MANUAL_CAPS   (KVM_DIRTY_LOG_MANUAL_PROTECT_ENABLE | \
>>>>>>>>>>>>>>        				     KVM_DIRTY_LOG_INITIALLY_SET)
>>>>>>>>>>>>>> diff --git a/arch/arm64/kvm/arm.c b/arch/arm64/kvm/arm.c
>>>>>>>>>>>>>> index bf6688245d83..566953a4e23a 100644
>>>>>>>>>>>>>> --- a/arch/arm64/kvm/arm.c
>>>>>>>>>>>>>> +++ b/arch/arm64/kvm/arm.c
>>>>>>>>>>>>>> @@ -755,6 +755,9 @@ void kvm_arch_vcpu_put(struct kvm_vcpu *vcpu)
>>>>>>>>>>>>>>        		kvm_vcpu_put_hw_mmu(vcpu);
>>>>>>>>>>>>>>        	kvm_arm_vmid_clear_active();
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> +	if (vcpu->kvm->arch.enable_hdbss)
>>>>>>>>>>>>>> +		kvm_flush_hdbss_buffer(vcpu);
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>>        	vcpu_clear_on_unsupported_cpu(vcpu);
>>>>>>>>>>>>>>        	vcpu->cpu = -1;
>>>>>>>>>>>>>>        }
>>>>>>>>>>>>>> @@ -1157,6 +1160,9 @@ static int check_vcpu_requests(struct kvm_vcpu *vcpu)
>>>>>>>>>>>>>>        		if (kvm_dirty_ring_check_request(vcpu))
>>>>>>>>>>>>>>        			return 0;
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> +		if (kvm_check_request(KVM_REQ_FLUSH_HDBSS, vcpu))
>>>>>>>>>>>>>> +			kvm_flush_hdbss_buffer(vcpu);
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>>        		check_nested_vcpu_requests(vcpu);
>>>>>>>>>>>>>>        	}
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> @@ -1971,7 +1977,15 @@ long kvm_arch_vcpu_unlocked_ioctl(struct file *filp, unsigned int ioctl,
>>>>>>>>>>>>>>
>>>>>>>>>>>>>>        void kvm_arch_sync_dirty_log(struct kvm *kvm, struct kvm_memory_slot *memslot)
>>>>>>>>>>>>>>        {
>>>>>>>>>>>>>> +	/*
>>>>>>>>>>>>>> +	 * Flush all CPUs' dirty log buffers to the dirty_bitmap.  Called
>>>>>>>>>>>>>> +	 * before reporting dirty_bitmap to userspace. Send a request with
>>>>>>>>>>>>>> +	 * KVM_REQUEST_WAIT to flush buffer synchronously.
>>>>>>>>>>>>>> +	 */
>>>>>>>>>>>>>> +	if (!kvm->arch.enable_hdbss)
>>>>>>>>>>>>>> +		return;
>>>>>>>>>>>>>>
>>>>>>>>>>>>>> +	kvm_make_all_cpus_request(kvm, KVM_REQ_FLUSH_HDBSS | KVM_REQUEST_WAIT);
>>>>>>>>>>>>>>        }
>>>>>>>>>>>>>>
>>>>>>>>>>>>>>        static int kvm_vm_ioctl_set_device_addr(struct kvm *kvm,
>>>>>>>>>>>>>> diff --git a/arch/arm64/kvm/dirty_bit.c b/arch/arm64/kvm/dirty_bit.c
>>>>>>>>>>>>>> index 6c7a6ef66b5a..002366337637 100644
>>>>>>>>>>>>>> --- a/arch/arm64/kvm/dirty_bit.c
>>>>>>>>>>>>>> +++ b/arch/arm64/kvm/dirty_bit.c
>>>>>>>>>>>>>> @@ -50,3 +50,65 @@ void kvm_arm_vcpu_free_hdbss(struct kvm_vcpu *vcpu)
>>>>>>>>>>>>>>
>>>>>>>>>>>>>>        	vcpu->arch.hdbss.hdbssbr_el2 = 0;
>>>>>>>>>>>>>>        }
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +void kvm_flush_hdbss_buffer(struct kvm_vcpu *vcpu)
>>>>>>>>>>>>>> +{
>>>>>>>>>>>>>> +	int idx, curr_idx;
>>>>>>>>>>>>>> +	u64 *hdbss_buf;
>>>>>>>>>>>>>> +	struct kvm *kvm = vcpu->kvm;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +	if (!kvm->arch.enable_hdbss)
>>>>>>>>>>>>>> +		return;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +	curr_idx = HDBSSPROD_IDX(read_sysreg_s(SYS_HDBSSPROD_EL2));
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +	/* Do nothing if HDBSS buffer is empty or br_el2 is NULL */
>>>>>>>>>>>>>> +	if (curr_idx == 0 || vcpu->arch.hdbss.hdbssbr_el2 == 0)
>>>>>>>>>>>>>> +		return;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +	hdbss_buf = page_address(phys_to_page(vcpu->arch.hdbss.base_phys));
>>>>>>>>>>>>>> +	if (!hdbss_buf)
>>>>>>>>>>>>>> +		return;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +	guard(write_lock_irqsave)(&vcpu->kvm->mmu_lock);
>>>>>>>>>>>>>> +	for (idx = 0; idx < curr_idx; idx++) {
>>>>>>>>>>>>>> +		u64 gpa;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +		gpa = hdbss_buf[idx];
>>>>>>>>>>>>>> +		if (!(gpa & HDBSS_ENTRY_VALID))
>>>>>>>>>>>>>> +			continue;
>>>>>>>>>>>>>> +
>>>>>>>>>>>>>> +		gpa &= HDBSS_ENTRY_IPA;
>>>>>>>>>>>>>> +		kvm_vcpu_mark_page_dirty(vcpu, gpa >> PAGE_SHIFT);
>>>>>>>>>>>>> You mention that it does not support dirty-ring, but above function will
>>>>>>>>>>>>> mark the page as dirty in the dirty-ring :/
>>>>>>>>>>>>>
>>>>>>>>>>>> In kvm_arm_enable_hdbss_global(), we explicitly check and reject HDBSS
>>>>>>>>>>>> enablement if dirty-ring is active:
>>>>>>>>>>>>
>>>>>>>>>>>> ```
>>>>>>>>>>>> if (kvm->dirty_ring_size)
>>>>>>>>>>>>          return 0;
>>>>>>>>>>>> ```
>>>>>>>>>>>>
>>>>>>>>>>>> So when kvm_flush_hdbss_buffer() runs (which requires enable_hdbss = true),
>>>>>>>>>>>> we know for certain that
>>>>>>>>>>>>
>>>>>>>>>>>> kvm->dirty_ring_size == 0. Therefore, kvm_vcpu_mark_page_dirty() will always
>>>>>>>>>>>> take the dirty_bitmap path,
>>>>>>>>>>>>
>>>>>>>>>>>> never the dirty-ring path.
>>>>>>>>>>>>
>>>>>>>>>>>> That said, I'll add a comment in kvm_flush_hdbss_buffer() before dirty ring
>>>>>>>>>>>> mode is supported, to make this explicit:
>>>>>>>>>>>>
>>>>>>>>>>>> ```
>>>>>>>>>>>> /*
>>>>>>>>>>>>       * HDBSS is mutually exclusive with dirty-ring mode (see
>>>>>>>>>>>>       * kvm_arm_enable_hdbss_global()), so kvm_vcpu_mark_page_dirty()
>>>>>>>>>>>>       * will update the dirty_bitmap, not the dirty-ring.
>>>>>>>>>>>>       */
>>>>>>>>>>>> ```
>>>>>>>>>>>>
>>>>>>>>>>> Got it :)
>>>>>>>>>>>
>>>>>>>>>>> Out of curiosity: which issues have you found on supporting dirty-ring at
>>>>>>>>>>> this point?
>>>>>>>>>>>
>>>>>>>>>>> Thanks!
>>>>>>>>>>> Leo
>>>>>>>>>>
>>>>>>>>>> I haven't looked deeply into dirty-ring yet — my main concern is that if
>>>>>>>>>> both the dirty
>>>>>>>>>>
>>>>>>>>>> ring and HDBSS buffer fill up, the flush path might get blocked or
>>>>>>>>>> complicated.
>>>>>>>>>>
>>>>>>>>>> For now, I'm planning to match the HDBSS buffer size to the dirty ring size
>>>>>>>>>> in v5 and test it.
>>>>>>>>>>
>>>>>>>>>> Ideally, the two buffers would be the same size, and the entire dirty
>>>>>>>>>> tracking path would use
>>>>>>>>>>
>>>>>>>>> Ah, I see the point.
>>>>>>>>>
>>>>>>>>> IIRC, when dirty-ring gets full, the kernel returns to userspace with
>>>>>>>>> run->exit_reason == KVM_EXIT_DIRTY_RING_FULL, which will warn the VMM to
>>>>>>>>> drain the dirty-ring, and that makes space for us draining HDBSS to the
>>>>>>>>> dirty-ring again.
>>>>>>>>>
>>>>>>>>> The best way to achieve that, as I remember, is to always drain
>>>>>>>>> HDBSS as much as possible at guest_exitting. That will make more space to
>>>>>>>>> newer HDBSS entries, and we can get userspace to drain the dirty-ring
>>>>>>>>> earlier.
>>>>>>>>>
>>>>>>>>> I would say to even make HDBSS buffer half (entries) the dirty-ring. Then
>>>>>>>>> we can generally fully drain to the dirty-ring and even report ring full
>>>>>>>>> if the ring is above a given threshold percentage full.
>>>>>>>>>> HDBSS exclusively — no fallback to the legacy dirty bitmap path. If that
>>>>>>>>>> works, I think this approach should be fine.
>>>>>>>>>>
>>>>>>>>>> Let me know if you have any insights on dirty ring's full-buffer behavior —
>>>>>>>>>> that would be helpful.
>>>>>>>>>>
>>>>>>>>>>
>>>>>>>>> Will do!
>>>>>>>>>
>>>>>>>>> Thanks!
>>>>>>>>> Leo
>>>>>>>>
>>>>>>>>
>>>>>>>> Thanks for the insights — reusing PML's reservation mechanism makes sense.
>>>>>>>>
>>>>>>>> I think we can*keep the HDBSS buffer at 512 entries* (matching
>>>>>>>
>>>>>>> I recommend using a PAGESIZE (512 in 4k, but bigger in other sizes)
>>>>>>>
>>>>>>>> PML_LOG_NR_ENTRIES)
>>>>>>>>
>>>>>>>> for now, and *not expose any ioctl for userspace to configure it*. Since the
>>>>>>>> kernel
>>>>>>>>
>>>>>>>> auto-enables HDBSS, a userspace size knob would be confusing.
>>>>>>>>
>>>>>>>
>>>>>>> Humm, I am in favor of letting the user change it according to it's
>>>>>>> workload. Why would that be confusing?
>>>>>>>
>>>>>>> (Having a default size is useful just for enabling it to work without any
>>>>>>> change in current VMs)
>>>>>>>
>>>>>>>>
>>>>>>>> The reservation logic would be:
>>>>>>>>
>>>>>>>> - Implement kvm_cpu_dirty_log_size() on arm64 to return the HDBSS buffer
>>>>>>>> entry
>>>>>>>>
>>>>>>>> count (512, or 0 if HDBSS is not enabled)
>>>>>>>
>>>>>>> Why a get to log_size? does userspace need to know it's using HDBSS?
>>>>>>>
>>>>>>> Just a set should do, as VMMs can just try to set a value, and if it fails
>>>>>>> (IOCTL does not exist, or invalid value), then it can just go forward.
>>>>>>>
>>>>>>>>
>>>>>>>> - Reuse kvm_dirty_ring_get_rsvd_entries() to reserve space for one full
>>>>>>>> flush:
>>>>>>>>
>>>>>>>> KVM_DIRTY_RING_RSVD_ENTRIES + hdbss_entries
>>>>>>>>
>>>>>>>> - So soft_limit = dirty_ring_size - (KVM_DIRTY_RING_RSVD_ENTRIES +
>>>>>>>> hdbss_entries),
>>>>>>>>
>>>>>>>> guaranteeing a full flush always fits
>>>>>>>>
>>>>>>>> kvm_flush_hdbss_buffer() at guest_exit pushes via mark_page_dirty_in_slot()
>>>>>>>> ->
>>>>>>>>
>>>>>>>
>>>>>>> I think I get the point here: since we can have sw dirtying as well as
>>>>>>> HDBSS tracking, we may get to the point that we don't have enough space to
>>>>>>> flush hdbss -> dirty-ring, right?
>>>>>>>
>>>>>>>> kvm_dirty_ring_push() -> kvm_dirty_ring_soft_full, which triggers
>>>>>>>>
>>>>>>>> KVM_EXIT_DIRTY_RING_FULL when needed.
>>>>>>>>
>>>>>>>> If soft_limit is hit, KVM_REQ_DIRTY_RING_FULL is set and the next vcpu_run
>>>>>>>> exits
>>>>>>>>
>>>>>>>> to userspace for QEMU to drain the ring.
>>>>>>>
>>>>>>> So the software dirtying routine would be affected by the soft limit, but
>>>>>>> HADBSS exit would not. It means the dirty-ring would be effectively smaller
>>>>>>> than specified if we are having mostly sw dirtying.
>>>>>>>
>>>>>>> Did I get that right?
>>>>>>>
>>>>>>>>
>>>>>>>> *One more thing: *when userspace sets the dirty ring size via
>>>>>>>>
>>>>>>>> KVM_VM_IOCTL_ENABLE_DIRTY_LOG_RING, we already enforce that
>>>>>>>>
>>>>>>>> size >= kvm_dirty_ring_get_rsvd_entries(kvm) * sizeof(struct kvm_dirty_gfn)
>>>>>>>>
>>>>>>>> or size < PAGE_SIZE. With HDBSS, kvm_dirty_ring_get_rsvd_entries() will
>>>>>>>> include the
>>>>>>>>
>>>>>>>> HDBSS entries via kvm_cpu_dirty_log_size(), so the same check will
>>>>>>>> automatically guarantee
>>>>>>>>
>>>>>>>> the ring is large enough to accommodate the HDBSS buffer. No additional
>>>>>>>> validation is needed.
>>>>>>>>
>>>>>>>
>>>>>>
>>>>>
>>>>> Hi Inochi,
>>>>>
>>>>>> I have a small question on this, since HDBSS supports 2MiB buffer,
>>>>>> it  will has more entries than the dirty ring.
>>>>>
>>>>> The entries reserved for HDBSS are part of the dirty ring, but I get the
>>>>> point: having a dirty-ring with 64k entries and HDBSS with 256k entries
>>>>> (2MB/8) would be weird.
>>>>>
>>>>> For the dirty-ring I think the plan is to have HDBSS buffer to be a fraction
>>>>> of the configured dirty-ring size.
>>>>>
>>>>>> If we use
>>>>>> kvm_cpu_dirty_log_size() to reserve entries. It will always enter
>>>>>> soft limit if the HDBSS buffer is huge.
>>>>>>
>>>>>
>>>>> Yeah, but once HDBSS is active if we enable dirty_tracking, there should
>>>>> not be a lot of entries being added to the dirty ring that don't come from
>>>>> HDBSS, so it should be fine. If it happens, though, it will just exit guest
>>>>> and drain as it happens nowadays.
>>>>>
>>>>>
>>>>>> Since I think users set large buffer expect less context switch.
>>>>>> I think this may require some change on the dirty ring framework.
>>>>>> At least I think the soft limit check should be adjusted.
>>>>>>
>>>>>
>>>>> That may not be the case: the memory usage of the dirty-ring is
>>>>> (16 * dirty_ring_size * vcpus) in bytes.
>>>>>
>>>>> On a VM with 1024 vcpus, only the HDBSS buffer alone would be 2GB, plus 4GB
>>>>> of the reserved dirty-ring for that buffer.
>>>>>
>>>>> Which brings the discussion: maybe we should have a buffer management
>>>>> happen instead of using the reserving strategy: If the buffer does not fit
>>>>> in the dirty-ring, just move the remaining entries to the start, adjust the
>>>>> index register, and continue.
>>>>>
>>>>> (On a 2MB size it would be bad, as there could be many remaining entries)
>>>>>
>>>>> Alternatively, we can just keep guest_exitting until all HDBSS entries fit
>>>>> the dirty_ring, which introduce overhead, but should work as well.
>>>>>
>>>>> All options have some tradeoff, though.
>>>>>
>>>>> Thanks!
>>>>> Leo
>>>>>
>>>> Hi Leo, Inochi
>>>>
>>>> I agree with you — this is exactly why I mentioned earlier that I don't
>>>> want to expose a buffer size ioctl. It would just create a hidden
>>>> dependency on the dirty ring size, and the "right" value is really
>>>> something the kernel should figure out on its own, not something
>>>> userspace needs to care about.
>>>
>>> Yeap
>>>
>>>>
>>>> For v5, here's what I'm thinking:
>>>>
>>>> With dirty ring: auto-size the HDBSS buffer to half the ring size,
>>>> rounding down to whatever order the hardware supports.
>>>
>>> Just remember that someone can set a smaller dirty-ring than we can have
>>> in HDBSS, though.
>>>
>>> IIUC the minimum size we should have is PAGE_SIZE.
>>> That means 512 entries in 4kB pagesize, an 8K entries in 64kB pagesize.
>>>
>>> I mean, yes, maybe we could allocate a 64k page, and set SZ=4K in HDBSSBR,
>>> but that would be a waste of memory, right?
>>>
>>> OTOH, if we are using the approach of reserving in dirty-ring a size that
>>> fits the HDBSS buffer, we would be allocating 2 extra pages for every ring
>>> buffer.
>>>
>>>>
>>>> Without dirty ring (bitmap mode): just use a sensible default — like
>>>> 512 entries, or whatever size you think makes sense.
>>>>
>>>> And implement kvm_cpu_dirty_log_size() for the reservation logic.
>>>>
>>>> Does that sound reasonable? Or do you have a better idea for the bitmap
>>>> mode default?prevented
>>>>
>>>
>>> See above comment on using PAGE_SIZE.
>>> We would possibly be reserving a whole page for each HDBSS buffer anyway,
>>> so I think we should make use of it all at least on dirty-bitmap case as
>>> there are no extra per-vcpu allocations.
>>>
>>> Thanks!
>>> Leo
>>>
>>
>> Hi, Leo
>>
>> Thanks for the feedback — that makes sense. Here's my updated plan:
>>
>> For dirty-bitmap mode: allocate a full page for each HDBSS buffer. That
>> means 512 entries on 4KB pages, or 8192 entries on 64KB pages. We can
>> use it all.
> 
> Looks reasonable
> 
>>
>> For dirty-ring mode: the user's ability to configure the ring size is
>> already gated by kvm_dirty_ring_get_rsvd_entries():
>>
>> ```
>> kvm_vm_ioctl_enable_dirty_log_ring()
>>      -> kvm_dirty_ring_get_rsvd_entries()
>>          -> kvm_cpu_dirty_log_size()  /* returns HDBSS entry count */
>> ```
>>
>> Since we can make kvm_cpu_dirty_log_size() return at least PAGE_SIZE /
>> sizeof(u64) entries, the ring size check will naturally ensure that the
>> dirty ring is large enough for a full HDBSS flush. This way we avoid
>> wasting memory, and also guarantee that both the dirty ring and the
>> HDBSS buffer can be flushed properly.
> 
> Which strategy have you decided to go with?
> 
> Remember that in lazy-spliting we will have both sw dirty-bit logging and
> HDBSS dirty-bit logging, so just having dirty-ring size > hdbss buffer is
> no guarantee.
> 
> (But we could have the remaining buffer entries being moved to the start of
> the array in this case)
> 

Yes, you're right. And simply having ring_size > hdbss_buf_entries isn't
enough, we need the soft_limit check. The good thing is that the
existing infrastructure already handles this:

```
ring->soft_limit = ring->size - kvm_dirty_ring_get_rsvd_entries(kvm);
```

Once we implement kvm_cpu_dirty_log_size() on arm64 to return the HDBSS
entry count, the existing check in kvm_vm_ioctl_enable_dirty_log_ring()
will automatically enforce:

```
static int kvm_vm_ioctl_enable_dirty_log_ring(...)
{
	//...
	ring_size >= (KVM_DIRTY_RING_RSVD_ENTRIES + hdbss_buf_entries) *
	sizeof(struct kvm_dirty_gfn)
	//...
}
```

see 
https://lore.kernel.org/linux-arm-kernel/[email protected]/

>   
>>
>> On the ioctl: based on the analysis above, I still think we should not
>> expose an ioctl for userspace to configure the HDBSS buffer size. It
>> should be fully determined by the kernel. Exposing it would create an
>> unnecessary dependency on the dirty ring size and complicate the
>> implementation.
>>
> 
> The size of the HDBSS buffer is more affected by the workload than the
> kernel can predict.
> 

Yep, and the kernel seems to have no way to predict VM workload.

> Take dirty-ring as an example, as it behaves most likely to HDBSS. We need
> to let the user decide on the dirty-ring size, as the user can decide to
> pay extra memory for less guest_exits, or the opposite.
> 

Right. In dirty-ring mode, the ring size is already set by userspace, so
we don't need a separate ioctl for HDBSS. The kernel can just size the
HDBSS buffer based on the dirty ring size.

> But I agree that having the ioctl that have any effect on dirty-ring mode
> is redundant. Maybe we should advertise an ioctl for setting HDBSS buffer,
> but make it clear that only affects dirty-bitmap, such as 'manual protect'
> mode.
> 
> How does it sound?
> 
> Thanks!
> Leo
> 

That sounds great — only expose this ioctl for dirty-bitmap mode.

Thanks!
Tian