| xj | b04a402 | 2021-11-25 15:01:52 +0800 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
| 2 | #include <linux/types.h> | ||||
| 3 | #include <linux/init.h> | ||||
| 4 | #include <linux/slab.h> | ||||
| 5 | #include <linux/bootmem.h> | ||||
| 6 | #include <linux/string.h> | ||||
| 7 | #include <asm/setup.h> | ||||
| 8 | |||||
| 9 | |||||
| 10 | void * __ref zalloc_maybe_bootmem(size_t size, gfp_t mask) | ||||
| 11 | { | ||||
| 12 | void *p; | ||||
| 13 | |||||
| 14 | if (slab_is_available()) | ||||
| 15 | p = kzalloc(size, mask); | ||||
| 16 | else { | ||||
| 17 | p = memblock_virt_alloc(size, 0); | ||||
| 18 | } | ||||
| 19 | return p; | ||||
| 20 | } | ||||