microblaze: Add {z,}alloc_maybe_bootmem functions (a84642a3) · Commits · 戴 / test

arch/microblaze/include/asm/system.h

+3 −0

Original line number	Diff line number	Diff line
		@@ -87,6 +87,9 @@ void free_initmem(void);
		extern char *klimit;
		extern void ret_from_fork(void);

		extern void *alloc_maybe_bootmem(size_t size, gfp_t mask);
		extern void *zalloc_maybe_bootmem(size_t size, gfp_t mask);

		#ifdef CONFIG_DEBUG_FS
		extern struct dentry *of_debugfs_root;
		#endif

+23 −0

Original line number	Diff line number	Diff line
		@@ -349,4 +349,27 @@ void __init *early_get_page(void)
		}
		return p;
		}

		void * __init_refok alloc_maybe_bootmem(size_t size, gfp_t mask)
		{
		if (mem_init_done)
		return kmalloc(size, mask);
		else
		return alloc_bootmem(size);
		}

		void * __init_refok zalloc_maybe_bootmem(size_t size, gfp_t mask)
		{
		void *p;

		if (mem_init_done)
		p = kzalloc(size, mask);
		else {
		p = alloc_bootmem(size);
		if (p)
		memset(p, 0, size);
		}
		return p;
		}

		#endif /* CONFIG_MMU */