alloc.c 360 B

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