mm: use vmalloc_array and vcalloc for array allocations

Instead of using array_size or just a multiply, use a function that
takes care of both the multiplication and the overflow checks.

Acked-by: Michal Hocko <mhocko@suse.com>
Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>

diff --git a/mm/percpu-stats.c b/mm/percpu-stats.c
index c6bd092ff7a35ea91a3f2f8046775115876d5954..dd3590dfc23d2c6012409499e8fd7bfae75a6e8a 100644 (file)
--- a/mm/percpu-stats.c
+++ b/mm/percpu-stats.c
@@ -144,7 +144,7 @@ alloc_buffer:
        spin_unlock_irq(&pcpu_lock);
 
        /* there can be at most this many free and allocated fragments */
-       buffer = vmalloc(array_size(sizeof(int), (2 * max_nr_alloc + 1)));
+       buffer = vmalloc_array(2 * max_nr_alloc + 1, sizeof(int));
        if (!buffer)
                return -ENOMEM;
 

diff --git a/mm/swap_cgroup.c b/mm/swap_cgroup.c
index 7f34343c075a1cfbcad053aaa1d25bab7d548bf0..5a9442979a1859227e18914678f7d0e4162c503a 100644 (file)
--- a/mm/swap_cgroup.c
+++ b/mm/swap_cgroup.c
@@ -167,14 +167,12 @@ unsigned short lookup_swap_cgroup_id(swp_entry_t ent)
 int swap_cgroup_swapon(int type, unsigned long max_pages)
 {
        void *array;
-       unsigned long array_size;
        unsigned long length;
        struct swap_cgroup_ctrl *ctrl;
 
        length = DIV_ROUND_UP(max_pages, SC_PER_PAGE);
-       array_size = length * sizeof(void *);
 
-       array = vzalloc(array_size);
+       array = vcalloc(length, sizeof(void *));
        if (!array)
                goto nomem;