1 files changed, 3 insertions, 3 deletions

diff --git a/kernel/power/swap.c b/kernel/power/swap.c
index 1efcb5b0c3ed..c2bcf97d24c8 100644
--- a/kernel/power/swap.c
+++ b/kernel/power/swap.c
@@ -698,7 +698,7 @@ static int save_image_lzo(struct swap_map_handle *handle,
                 goto out_clean;
         }
 
-        data = vmalloc(sizeof(*data) * nr_threads);
+        data = vmalloc(array_size(nr_threads, sizeof(*data)));
         if (!data) {
                 pr_err("Failed to allocate LZO data\n");
                 ret = -ENOMEM;
@@ -1183,14 +1183,14 @@ static int load_image_lzo(struct swap_map_handle *handle,
         nr_threads = num_online_cpus() - 1;
         nr_threads = clamp_val(nr_threads, 1, LZO_THREADS);
 
-        page = vmalloc(sizeof(*page) * LZO_MAX_RD_PAGES);
+        page = vmalloc(array_size(LZO_MAX_RD_PAGES, sizeof(*page)));
         if (!page) {
                 pr_err("Failed to allocate LZO page\n");
                 ret = -ENOMEM;
                 goto out_clean;
         }
 
-        data = vmalloc(sizeof(*data) * nr_threads);
+        data = vmalloc(array_size(nr_threads, sizeof(*data)));
         if (!data) {
                 pr_err("Failed to allocate LZO data\n");
                 ret = -ENOMEM;