[PATCH 02/27] RISC-V: KVM: Fix error code returned for Ssaia ONE_REG

[Anup Patel]

Return -ENOENT for Ssaia ONE_REG when Ssaia is not enabled
for a VCPU.

This will make Ssaia ONE_REG error codes consistent with
other ONE_REG interfaces of KVM RISC-V.

Fixes: 2a88f38cd58d ("RISC-V: KVM: return ENOENT in *_one_reg() when reg is 
unknown")
Signed-off-by: Anup Patel <anup.pa...@oss.qualcomm.com>
---
 arch/riscv/kvm/aia.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/arch/riscv/kvm/aia.c b/arch/riscv/kvm/aia.c
index dad318185660..31baea9f0589 100644
--- a/arch/riscv/kvm/aia.c
+++ b/arch/riscv/kvm/aia.c
@@ -183,6 +183,8 @@ int kvm_riscv_vcpu_aia_get_csr(struct kvm_vcpu *vcpu,
 {
        struct kvm_vcpu_aia_csr *csr = &vcpu->arch.aia_context.guest_csr;
 
+       if (!riscv_isa_extension_available(vcpu->arch.isa, SSAIA))
+               return -ENOENT;
        if (reg_num >= sizeof(struct kvm_riscv_aia_csr) / sizeof(unsigned long))
                return -ENOENT;
 
@@ -199,6 +201,8 @@ int kvm_riscv_vcpu_aia_set_csr(struct kvm_vcpu *vcpu,
 {
        struct kvm_vcpu_aia_csr *csr = &vcpu->arch.aia_context.guest_csr;
 
+       if (!riscv_isa_extension_available(vcpu->arch.isa, SSAIA))
+               return -ENOENT;
        if (reg_num >= sizeof(struct kvm_riscv_aia_csr) / sizeof(unsigned long))
                return -ENOENT;
 
-- 
2.43.0