| Kyle Swenson | 8d8f654 | 2021-03-15 11:02:55 -0600 | [diff] [blame^] | 1 | #include <linux/types.h> |
| 2 | #include <linux/init.h> | ||||
| 3 | #include <linux/slab.h> | ||||
| 4 | #include <linux/bootmem.h> | ||||
| 5 | #include <linux/string.h> | ||||
| 6 | #include <asm/setup.h> | ||||
| 7 | |||||
| 8 | |||||
| 9 | void * __init_refok zalloc_maybe_bootmem(size_t size, gfp_t mask) | ||||
| 10 | { | ||||
| 11 | void *p; | ||||
| 12 | |||||
| 13 | if (slab_is_available()) | ||||
| 14 | p = kzalloc(size, mask); | ||||
| 15 | else { | ||||
| 16 | p = memblock_virt_alloc(size, 0); | ||||
| 17 | } | ||||
| 18 | return p; | ||||
| 19 | } | ||||