Skip to contentSkip to conversation

KVM: SVM: Update tracked pCPU on pCPU switch

PATCH Applied Yosry Ahmed 3 participants 3 messages Oct 13 – Nov 13

Subsystems (4)
lore
Share & tools
lore
#0 lore raw

PATCH v2 1/1 KVM: SVM: Update tracked pCPU on pCPU switch

When a pCPU switch is detected in pre_svm_run(), update the tracked pCPU and mark all fields dirty in both VMCBs (if nested is initialized), in addition to resetting the ASID generation in both VMCBs.

Commit bf260dc9e7b8 ("KVM: SVM: Trigger new ASID allocation in both VMCBs on pCPU switch") reset the ASID generation in both VMCBs on a pCPU switch, but only updated the tracked pCPU for the active VMCB to avoid unnecessarily dirtying the inactive VMCB. Leaving the inactive VMCB's tracked pCPU set to an old pCPU still allows an ASID allocated on one pCPU to be used on another:

- vCPU runs with both vmcb01 and vmcb02 on pCPU A (both track cpu=A).
- With vmcb01 active, vCPU migrates to pCPU B. vmcb01.cpu is updated to B, asid_generation is reset to 0 in both VMCBs, and a new ASID is allocated on pCPU B (setting vmcb01.asid_generation = gen_B). vmcb02.cpu remains A.
- vCPU switches to vmcb02 and migrates back to pCPU A before pre_svm_run(). Because vmcb02.cpu is still A, no pCPU switch is detected, leaving vmcb01 with cpu=B and asid_generation=gen_B. Because vmcb02.asid_generation is 0, a new ASID (asid_A) is allocated on pCPU A.
- vCPU switches back to vmcb01 and migrates to pCPU B. Neither a pCPU change nor a generation mismatch is detected for vmcb01, so VMRUN on pCPU B uses asid_A (allocated on pCPU A), which may collide with another vCPU on pCPU B.

Because the per-VMCB cpu tracking is shared between ASID management and per-pCPU hardware VMCB clean bits, also mark both VMCBs dirty when updating their tracked pCPU so that the inactive VMCB does not run with stale clean bits when later entered on the new pCPU.

Fixes bf260dc9e7b8 ("KVM: SVM: Trigger new ASID allocation in both VMCBs on pCPU switch")

---

v1 -> v2:
- Drop confusing comment [Jim].

---
 arch/x86/kvm/svm/svm.c | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)
arch/x86/kvm/svm/svm.c +11 -5
index 136069dfcd44f..95c61b7a470b8 100644
@@ -3880,7 +3880,8 @@ static int pre_svm_run(struct kvm_vcpu *vcpu)
 
 	/*
 	 * If the previous VMRUN of the VMCB occurred on a different physical
-	 * cpu, then mark the VMCB dirty as hardware's clean bits are per pCPU.
+	 * cpu, mark the VMCB dirty and reset the ASID generation, as hardware's
+	 * clean bits and ASIDs are per pCPU.
 	 *
 	 * Reset the ASID generation in both VMCBs. This will lead to assigning
 	 * a new ASID now, and then again when switching to the other VMCB.
@@ -3894,11 +3895,16 @@ static int pre_svm_run(struct kvm_vcpu *vcpu)
 	 *   another vCPU/VM.
 	 */
 	if (unlikely(svm->current_vmcb->cpu != vcpu->cpu)) {
-		vmcb_mark_all_dirty(svm->vmcb);
-		svm->current_vmcb->cpu = vcpu->cpu;
+		vmcb_mark_all_dirty(svm->vmcb01.ptr);
+		svm->vmcb01.cpu = vcpu->cpu;
 		svm->vmcb01.asid_generation = 0;
-		svm->nested.vmcb02.asid_generation = 0;
-        }
+
+		if (svm->nested.initialized) {
+			svm->nested.vmcb02.cpu = vcpu->cpu;
+			svm->nested.vmcb02.asid_generation = 0;
+			vmcb_mark_all_dirty(svm->nested.vmcb02.ptr);
+		}
+	}
 
 	if (is_sev_guest(vcpu))
 		return pre_sev_run(svm, vcpu->cpu);

base-commit: 6bd2905303c58679e303941b7d8ae8c074cd95ce

Signature · 1 line
2.56.0.385.gd3acb90ef8-goog
Reviewed-by on v2 1/1
#1 lore raw
Quoted 32 lines · Yosry Ahmed

When a pCPU switch is detected in pre_svm_run(), update the tracked pCPU and mark all fields dirty in both VMCBs (if nested is initialized), in addition to resetting the ASID generation in both VMCBs.

Commit bf260dc9e7b8 ("KVM: SVM: Trigger new ASID allocation in both VMCBs on pCPU switch") reset the ASID generation in both VMCBs on a pCPU switch, but only updated the tracked pCPU for the active VMCB to avoid unnecessarily dirtying the inactive VMCB. Leaving the inactive VMCB's tracked pCPU set to an old pCPU still allows an ASID allocated on one pCPU to be used on another:

- vCPU runs with both vmcb01 and vmcb02 on pCPU A (both track cpu=A).
- With vmcb01 active, vCPU migrates to pCPU B. vmcb01.cpu is updated to B, asid_generation is reset to 0 in both VMCBs, and a new ASID is allocated on pCPU B (setting vmcb01.asid_generation = gen_B). vmcb02.cpu remains A.
- vCPU switches to vmcb02 and migrates back to pCPU A before pre_svm_run(). Because vmcb02.cpu is still A, no pCPU switch is detected, leaving vmcb01 with cpu=B and asid_generation=gen_B. Because vmcb02.asid_generation is 0, a new ASID (asid_A) is allocated on pCPU A.
- vCPU switches back to vmcb01 and migrates to pCPU B. Neither a pCPU change nor a generation mismatch is detected for vmcb01, so VMRUN on pCPU B uses asid_A (allocated on pCPU A), which may collide with another vCPU on pCPU B.

Because the per-VMCB cpu tracking is shared between ASID management and per-pCPU hardware VMCB clean bits, also mark both VMCBs dirty when updating their tracked pCPU so that the inactive VMCB does not run with stale clean bits when later entered on the new pCPU.

Quoted 4 lines
Fixes bf260dc9e7b8 ("KVM: SVM: Trigger new ASID allocation in both VMCBs on pCPU switch")
maintainer Applied
#2 lore raw
In reply to Yosry Ahmed
Quoted 4 lines · Yosry Ahmed

When a pCPU switch is detected in pre_svm_run(), update the tracked pCPU and mark all fields dirty in both VMCBs (if nested is initialized), in addition to resetting the ASID generation in both VMCBs.

Quoted 8 lines

Commit bf260dc9e7b8 ("KVM: SVM: Trigger new ASID allocation in both VMCBs on pCPU switch") reset the ASID generation in both VMCBs on a pCPU switch, but only updated the tracked pCPU for the active VMCB to avoid unnecessarily dirtying the inactive VMCB. Leaving the inactive VMCB's tracked pCPU set to an old pCPU still allows an ASID allocated on one pCPU to be used on another:

[...]

Applied to kvm-x86 fixes, thanks!

[1/1] KVM: SVM: Update tracked pCPU on pCPU switch

      https://github.com/kvm-x86/linux/commit/d41766c45065

--
https://github.com/kvm-x86/linux/tree/next