KVM: riscv: Update G-stage PTE permissions atomically

When a fault hits an existing G-stage leaf with the same PFN, KVM only
needs to update the PTE permissions. This path will be used by read-side
fault handling, so it must not overwrite a concurrent PTE update.

Use the cmpxchg helper when relaxing permissions on an existing leaf,
following the same concurrency model used by x86 for atomic SPTE
permission updates. Retry if another CPU changed the PTE first, and use
cpu_relax() while spinning.

Signed-off-by: Jinyu Tang <tjytimi@163.com>
Reviewed-by: Anup Patel <anup@brainfault.org>
Link: https://lore.kernel.org/r/20260517153427.94889-5-tjytimi@163.com
Signed-off-by: Anup Patel <anup@brainfault.org>
This commit is contained in:
Jinyu Tang 2026-05-17 23:34:26 +08:00 committed by Anup Patel
parent 7dd416fdd3
commit d7a26a0ba7

View File

@ -182,17 +182,22 @@ int kvm_riscv_gstage_set_pte(struct kvm_gstage *gstage,
static void kvm_riscv_gstage_update_pte_prot(struct kvm_gstage *gstage, u32 level,
gpa_t addr, pte_t *ptep, pgprot_t prot)
{
pte_t new_pte;
pte_t old_pte, new_pte;
if (pgprot_val(pte_pgprot(ptep_get(ptep))) == pgprot_val(prot))
return;
for (;;) {
old_pte = ptep_get(ptep);
if (pgprot_val(pte_pgprot(old_pte)) == pgprot_val(prot))
return;
new_pte = pfn_pte(pte_pfn(ptep_get(ptep)), prot);
new_pte = pte_mkdirty(new_pte);
new_pte = pfn_pte(pte_pfn(old_pte), prot);
new_pte = pte_mkdirty(new_pte);
set_pte(ptep, new_pte);
if (kvm_riscv_gstage_try_update_pte(gstage, level, addr, ptep,
old_pte, new_pte))
return;
gstage_tlb_flush(gstage, level, addr);
cpu_relax();
}
}
int kvm_riscv_gstage_map_page(struct kvm_gstage *gstage,