alloc.c 465 B

1234567891011121314151617181920212223
  1. // SPDX-License-Identifier: GPL-2.0
  2. #include <linux/types.h>
  3. #include <linux/init.h>
  4. #include <linux/slab.h>
  5. #include <linux/memblock.h>
  6. #include <linux/string.h>
  7. #include <asm/setup.h>
  8. void * __ref zalloc_maybe_bootmem(size_t size, gfp_t mask)
  9. {
  10. void *p;
  11. if (slab_is_available())
  12. p = kzalloc(size, mask);
  13. else {
  14. p = memblock_alloc(size, SMP_CACHE_BYTES);
  15. if (!p)
  16. panic("%s: Failed to allocate %zu bytes\n", __func__,
  17. size);
  18. }
  19. return p;
  20. }