/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __LINUX_BOOTMEM_INFO_H
#define __LINUX_BOOTMEM_INFO_H

#include <linux/mm.h>
#include <linux/kmemleak.h>

/*
 * Types for free bootmem stored in the low bits of page->private.
 */
enum bootmem_type {
	MEMORY_HOTPLUG_MIN_BOOTMEM_TYPE = 1,
	SECTION_INFO = MEMORY_HOTPLUG_MIN_BOOTMEM_TYPE,
	MIX_SECTION_INFO,
	NODE_INFO,
	MEMORY_HOTPLUG_MAX_BOOTMEM_TYPE = NODE_INFO,
};

#ifdef CONFIG_HAVE_BOOTMEM_INFO_NODE
void __init register_page_bootmem_info_node(struct pglist_data *pgdat);
void register_page_bootmem_memmap(unsigned long section_nr, struct page *map,
				  unsigned long nr_pages);

void get_page_bootmem(unsigned long info, struct page *page,
		enum bootmem_type type);
void put_page_bootmem(struct page *page);

static inline enum bootmem_type bootmem_type(const struct page *page)
{
	return (unsigned long)page->private & 0xf;
}

static inline unsigned long bootmem_info(const struct page *page)
{
	return (unsigned long)page->private >> 4;
}

/*
 * Any memory allocated via the memblock allocator and not via the
 * buddy will be marked reserved already in the memmap. For those
 * pages, we can call this function to free it to buddy allocator.
 */
static inline void free_bootmem_page(struct page *page)
{
	enum bootmem_type type = bootmem_type(page);

	/*
	 * The reserve_bootmem_region sets the reserved flag on bootmem
	 * pages.
	 */
	VM_BUG_ON_PAGE(page_ref_count(page) != 2, page);

	if (type == SECTION_INFO || type == MIX_SECTION_INFO)
		put_page_bootmem(page);
	else
		VM_BUG_ON_PAGE(1, page);
}
#else
static inline void register_page_bootmem_info_node(struct pglist_data *pgdat)
{
}

static inline void register_page_bootmem_memmap(unsigned long section_nr,
		struct page *map, unsigned long nr_pages)
{
}

static inline void put_page_bootmem(struct page *page)
{
}

static inline enum bootmem_type bootmem_type(const struct page *page)
{
	return SECTION_INFO;
}

static inline unsigned long bootmem_info(const struct page *page)
{
	return 0;
}

static inline void get_page_bootmem(unsigned long info, struct page *page,
				    enum bootmem_type type)
{
}

static inline void free_bootmem_page(struct page *page)
{
	kmemleak_free_part_phys(PFN_PHYS(page_to_pfn(page)), PAGE_SIZE);
	free_reserved_page(page);
}
#endif

#endif /* __LINUX_BOOTMEM_INFO_H */
