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
 7#include <asm/system.h>
 8
 9void * __init_refok zalloc_maybe_bootmem(size_t size, gfp_t mask)
10{
11	void *p;
12
13	if (mem_init_done)
14		p = kzalloc(size, mask);
15	else {
16		p = alloc_bootmem(size);
17		if (p)
18			memset(p, 0, size);
19	}
20	return p;
21}