korg/linux
0
0
Fork 0
mirror of https://mirrors.bfsu.edu.cn/git/linux.git synced 2024-11-11 12:28:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

mm: use vmalloc_array and vcalloc for array allocations

Browse Source 
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>
This commit is contained in:
Paolo Bonzini 2022-03-08 05:02:21 -05:00
parent a8749a35c3
commit 3000f2e2fc
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mm/percpu-stats.c
View File

@ -144,7 +144,7 @@ alloc_buffer:
spin_unlock_irq(&pcpu_lock); spin_unlock_irq(&pcpu_lock);
/* there can be at most this many free and allocated fragments */ /* 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) if (!buffer)
return -ENOMEM; return -ENOMEM;

4
mm/swap_cgroup.c
View File

@ -167,14 +167,12 @@ unsigned short lookup_swap_cgroup_id(swp_entry_t ent)
int swap_cgroup_swapon(int type, unsigned long max_pages) int swap_cgroup_swapon(int type, unsigned long max_pages)
{ {
void *array; void *array;
unsigned long array_size;
unsigned long length; unsigned long length;
struct swap_cgroup_ctrl *ctrl; struct swap_cgroup_ctrl *ctrl;
length = DIV_ROUND_UP(max_pages, SC_PER_PAGE); 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) if (!array)
goto nomem; goto nomem;