blob: da22c84a8fed6c13421740ce02199560f3896a70 [file] [log] [blame]
Stephen Rothwell7b2c3c52007-09-17 14:08:06 +10001#include <linux/types.h>
2#include <linux/init.h>
3#include <linux/slab.h>
4#include <linux/bootmem.h>
Stephen Rothwell5669c3c2007-10-02 13:37:53 +10005#include <linux/string.h>
David Howellsae3a1972012-03-28 18:30:02 +01006#include <asm/setup.h>
Stephen Rothwell7b2c3c52007-09-17 14:08:06 +10007
Stephen Rothwell7b2c3c52007-09-17 14:08:06 +10008
Stephen Rothwell5669c3c2007-10-02 13:37:53 +10009void * __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}