1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
#include <linux/types.h> #include <linux/init.h> #include <linux/slab.h> #include <linux/bootmem.h> #include <linux/string.h> #include <asm/setup.h> void * __init_refok zalloc_maybe_bootmem(size_t size, gfp_t mask) { void *p; if (slab_is_available()) p = kzalloc(size, mask); else { p = memblock_virt_alloc(size, 0); } return p; }