From: Oliver Upton <oliver.up...@linux.dev>

Use ARRAY_SIZE() instead of an open-coded version.

Signed-off-by: Oliver Upton <oliver.up...@linux.dev>
Signed-off-by: Marc Zyngier <m...@kernel.org>
---
 arch/arm64/kvm/hyp/nvhe/stacktrace.c | 3 +--
 arch/arm64/kvm/stacktrace.c          | 6 ++++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/arch/arm64/kvm/hyp/nvhe/stacktrace.c 
b/arch/arm64/kvm/hyp/nvhe/stacktrace.c
index acbe272ecb32..58f645ad66bc 100644
--- a/arch/arm64/kvm/hyp/nvhe/stacktrace.c
+++ b/arch/arm64/kvm/hyp/nvhe/stacktrace.c
@@ -103,14 +103,13 @@ static void notrace unwind(struct unwind_state *state,
 static bool pkvm_save_backtrace_entry(void *arg, unsigned long where)
 {
        unsigned long *stacktrace = this_cpu_ptr(pkvm_stacktrace);
-       int size = NVHE_STACKTRACE_SIZE / sizeof(long);
        int *idx = (int *)arg;
 
        /*
         * Need 2 free slots: 1 for current entry and 1 for the
         * delimiter.
         */
-       if (*idx > size - 2)
+       if (*idx > ARRAY_SIZE(pkvm_stacktrace) - 2)
                return false;
 
        stacktrace[*idx] = where;
diff --git a/arch/arm64/kvm/stacktrace.c b/arch/arm64/kvm/stacktrace.c
index 417665854f86..949d19d603fb 100644
--- a/arch/arm64/kvm/stacktrace.c
+++ b/arch/arm64/kvm/stacktrace.c
@@ -187,11 +187,13 @@ static void pkvm_dump_backtrace(unsigned long hyp_offset)
 {
        unsigned long *stacktrace
                = (unsigned long *) this_cpu_ptr_nvhe_sym(pkvm_stacktrace);
-       int i, size = NVHE_STACKTRACE_SIZE / sizeof(long);
+       int i;
 
        kvm_nvhe_dump_backtrace_start();
        /* The saved stacktrace is terminated by a null entry */
-       for (i = 0; i < size && stacktrace[i]; i++)
+       for (i = 0;
+            i < ARRAY_SIZE(kvm_nvhe_sym(pkvm_stacktrace)) && stacktrace[i];
+            i++)
                kvm_nvhe_dump_backtrace_entry((void *)hyp_offset, 
stacktrace[i]);
        kvm_nvhe_dump_backtrace_end();
 }
-- 
2.34.1

_______________________________________________
kvmarm mailing list
kvmarm@lists.cs.columbia.edu
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to