Now that we have a wrapper for the sysreg accesses, let's use that
consistently.

Signed-off-by: Marc Zyngier <[email protected]>
---
 arch/arm64/kvm/handle_exit.c | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/arch/arm64/kvm/handle_exit.c b/arch/arm64/kvm/handle_exit.c
index 10f13bd2e319..9678e2389c2f 100644
--- a/arch/arm64/kvm/handle_exit.c
+++ b/arch/arm64/kvm/handle_exit.c
@@ -217,10 +217,11 @@ static int handle_sve(struct kvm_vcpu *vcpu, struct 
kvm_run *run)
        return 1;
 }
 
-#define __ptrauth_save_key(regs, key)                                          
\
+
+#define __ptrauth_save_key(ctxt, key)                                          
\
 ({                                                                             
\
-       regs[key ## KEYLO_EL1] = read_sysreg_s(SYS_ ## key ## KEYLO_EL1);       
\
-       regs[key ## KEYHI_EL1] = read_sysreg_s(SYS_ ## key ## KEYHI_EL1);       
\
+       ctxt_sys_reg(ctxt, key ## KEYLO_EL1) = read_sysreg_s(SYS_ ## key ## 
KEYLO_EL1); \
+       ctxt_sys_reg(ctxt, key ## KEYHI_EL1) = read_sysreg_s(SYS_ ## key ## 
KEYHI_EL1); \
 })
 
 /*
@@ -234,11 +235,11 @@ void kvm_arm_vcpu_ptrauth_trap(struct kvm_vcpu *vcpu)
        if (vcpu_has_ptrauth(vcpu)) {
                vcpu_ptrauth_enable(vcpu);
                ctxt = vcpu->arch.host_cpu_context;
-               __ptrauth_save_key(ctxt->sys_regs, APIA);
-               __ptrauth_save_key(ctxt->sys_regs, APIB);
-               __ptrauth_save_key(ctxt->sys_regs, APDA);
-               __ptrauth_save_key(ctxt->sys_regs, APDB);
-               __ptrauth_save_key(ctxt->sys_regs, APGA);
+               __ptrauth_save_key(ctxt, APIA);
+               __ptrauth_save_key(ctxt, APIB);
+               __ptrauth_save_key(ctxt, APDA);
+               __ptrauth_save_key(ctxt, APDB);
+               __ptrauth_save_key(ctxt, APGA);
        } else {
                kvm_inject_undefined(vcpu);
        }
-- 
2.20.1

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

Reply via email to