1 // SPDX-License-Identifier: GPL-2.0 1 // SPDX-License-Identifier: GPL-2.0
2 2
3 #include <linux/log2.h> 3 #include <linux/log2.h>
4 #include <linux/slab.h> 4 #include <linux/slab.h>
5 #include <linux/vmalloc.h> 5 #include <linux/vmalloc.h>
6 #include "darray.h" 6 #include "darray.h"
7 7
8 int __bch2_darray_resize_noprof(darray_char *d 8 int __bch2_darray_resize_noprof(darray_char *d, size_t element_size, size_t new_size, gfp_t gfp)
9 { 9 {
10 if (new_size > d->size) { 10 if (new_size > d->size) {
11 new_size = roundup_pow_of_two( 11 new_size = roundup_pow_of_two(new_size);
12 12
13 /* 13 /*
14 * This is a workaround: kvmal 14 * This is a workaround: kvmalloc() doesn't support > INT_MAX
15 * allocations, but vmalloc() 15 * allocations, but vmalloc() does.
16 * The limit needs to be lifte 16 * The limit needs to be lifted from kvmalloc, and when it does
17 * we'll go back to just using 17 * we'll go back to just using that.
18 */ 18 */
19 size_t bytes; 19 size_t bytes;
20 if (unlikely(check_mul_overflo 20 if (unlikely(check_mul_overflow(new_size, element_size, &bytes)))
21 return -ENOMEM; 21 return -ENOMEM;
22 22
23 void *data = likely(bytes < IN 23 void *data = likely(bytes < INT_MAX)
24 ? kvmalloc_noprof(byte 24 ? kvmalloc_noprof(bytes, gfp)
25 : vmalloc_noprof(bytes 25 : vmalloc_noprof(bytes);
26 if (!data) 26 if (!data)
27 return -ENOMEM; 27 return -ENOMEM;
28 28
29 if (d->size) 29 if (d->size)
30 memcpy(data, d->data, 30 memcpy(data, d->data, d->size * element_size);
31 if (d->data != d->preallocated 31 if (d->data != d->preallocated)
32 kvfree(d->data); 32 kvfree(d->data);
33 d->data = data; 33 d->data = data;
34 d->size = new_size; 34 d->size = new_size;
35 } 35 }
36 36
37 return 0; 37 return 0;
38 } 38 }
39 39
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.