| Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
| 2 | #include <linux/types.h> |
| 3 | #include <linux/init.h> |
| 4 | #include <linux/slab.h> |
| David Brazdil | 0f672f6 | 2019-12-10 10:32:29 +0000 | [diff] [blame^] | 5 | #include <linux/memblock.h> |
| Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 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 { |
| David Brazdil | 0f672f6 | 2019-12-10 10:32:29 +0000 | [diff] [blame^] | 17 | p = memblock_alloc(size, SMP_CACHE_BYTES); |
| 18 | if (!p) |
| 19 | panic("%s: Failed to allocate %zu bytes\n", __func__, |
| 20 | size); |
| Andrew Scull | b4b6d4a | 2019-01-02 15:54:55 +0000 | [diff] [blame] | 21 | } |
| 22 | return p; |
| 23 | } |