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