LoongArch: KVM: Make function kvm_own_lbt() robust

Add the flag KVM_LARCH_LBT checking in function kvm_own_lbt(), so that
it can be called safely rather than duplicated enabling again.

Cc: stable@vger.kernel.org
Signed-off-by: Bibo Mao <maobibo@loongson.cn>
Signed-off-by: Huacai Chen <chenhuacai@loongson.cn>

diff --git a/arch/loongarch/kvm/vcpu.c b/arch/loongarch/kvm/vcpu.c
index d1b8c50941ca233a8fc4519a8276d6bccd916f24..ce478151466c00d3f0a363affb3137bd265625df 100644 (file)
--- a/arch/loongarch/kvm/vcpu.c
+++ b/arch/loongarch/kvm/vcpu.c
@@ -1283,9 +1283,11 @@ int kvm_own_lbt(struct kvm_vcpu *vcpu)
                return -EINVAL;
 
        preempt_disable();
-       set_csr_euen(CSR_EUEN_LBTEN);
-       _restore_lbt(&vcpu->arch.lbt);
-       vcpu->arch.aux_inuse |= KVM_LARCH_LBT;
+       if (!(vcpu->arch.aux_inuse & KVM_LARCH_LBT)) {
+               set_csr_euen(CSR_EUEN_LBTEN);
+               _restore_lbt(&vcpu->arch.lbt);
+               vcpu->arch.aux_inuse |= KVM_LARCH_LBT;
+       }
        preempt_enable();
 
        return 0;