Use kcalloc() in main_func() to gain built-in overflow protection, making
memory allocation safer when calculating allocation size compared to
explicit multiplication.

Signed-off-by: Qianfeng Rong <rongqianf...@vivo.com>
---
 kernel/rcu/refscale.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kernel/rcu/refscale.c b/kernel/rcu/refscale.c
index df646e0694a8..5840fac06feb 100644
--- a/kernel/rcu/refscale.c
+++ b/kernel/rcu/refscale.c
@@ -1021,7 +1021,7 @@ static int main_func(void *arg)
        set_user_nice(current, MAX_NICE);
 
        VERBOSE_SCALEOUT("main_func task started");
-       result_avg = kzalloc(nruns * sizeof(*result_avg), GFP_KERNEL);
+       result_avg = kcalloc(nruns, sizeof(*result_avg), GFP_KERNEL);
        buf = kzalloc(800 + 64, GFP_KERNEL);
        if (!result_avg || !buf) {
                SCALEOUT_ERRSTRING("out of memory");
-- 
2.34.1


Reply via email to