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 9void * __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} 20