Since the pinned VMID space is not recycled, we need to make sure that
we release the vmid back into the pool when we are done with it.

Signed-off-by: Shameer Kolothum <[email protected]>
---
 arch/arm64/kvm/arm.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/arch/arm64/kvm/arm.c b/arch/arm64/kvm/arm.c
index 8955968be49f..d9900ffb88f4 100644
--- a/arch/arm64/kvm/arm.c
+++ b/arch/arm64/kvm/arm.c
@@ -181,8 +181,16 @@ void kvm_arch_destroy_vm(struct kvm *kvm)
        kvm_vgic_destroy(kvm);
 
        for (i = 0; i < KVM_MAX_VCPUS; ++i) {
-               if (kvm->vcpus[i]) {
-                       kvm_vcpu_destroy(kvm->vcpus[i]);
+               struct kvm_vcpu *vcpu = kvm->vcpus[i];
+
+               if (vcpu) {
+                       struct kvm_vmid *vmid = &vcpu->arch.hw_mmu->vmid;
+
+                       if (refcount_read(&vmid->pinned)) {
+                               ida_free(&kvm_pinned_vmids, vmid->vmid);
+                               refcount_set(&vmid->pinned, 0);
+                       }
+                       kvm_vcpu_destroy(vcpu);
                        kvm->vcpus[i] = NULL;
                }
        }
-- 
2.17.1

_______________________________________________
kvmarm mailing list
[email protected]
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to