mshv: use kmalloc_array in mshv_root_scheduler_init

Replace kmalloc() with kmalloc_array() to prevent potential
overflow, as recommended in Documentation/process/deprecated.rst.

No functional change.

Signed-off-by: Can Peng <pengcan@kylinos.cn>
Signed-off-by: Wei Liu <wei.liu@kernel.org>

diff --git a/drivers/hv/mshv_root_main.c b/drivers/hv/mshv_root_main.c
index bd1359eb58dd46d31d1ef3e5aed3e2382689fe73..146726cc4e9ba83025acebc81c0d6eb9302a27ab 100644 (file)
--- a/drivers/hv/mshv_root_main.c
+++ b/drivers/hv/mshv_root_main.c
@@ -2241,7 +2241,7 @@ static int mshv_root_scheduler_init(unsigned int cpu)
        outputarg = (void **)this_cpu_ptr(root_scheduler_output);
 
        /* Allocate two consecutive pages. One for input, one for output. */
-       p = kmalloc(2 * HV_HYP_PAGE_SIZE, GFP_KERNEL);
+       p = kmalloc_array(2, HV_HYP_PAGE_SIZE, GFP_KERNEL);
        if (!p)
                return -ENOMEM;