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

#include <linux/pgtable.h>
#include <asm/pgalloc.h>

/*
 * {pgd,p4d}_populate_kernel() are defined as macros to allow
 * compile-time optimization based on the configured page table levels.
 * Without this, linking may fail because callers (e.g., KASAN) may rely
 * on calls to these functions being optimized away when passing symbols
 * that exist only for certain page table levels.
 */
#define pgd_populate_kernel(addr, pgd, p4d)                \
    do {                                \
        __pgd_populate_kernel(&init_mm, pgd, p4d);        \
        if (ARCH_PAGE_TABLE_SYNC_MASK & PGTBL_PGD_MODIFIED)    \
            arch_sync_kernel_mappings(addr, addr);        \
    } while (0)

#define p4d_populate_kernel(addr, p4d, pud)                \
    do {                                \
        __p4d_populate_kernel(&init_mm, p4d, pud);        \
        if (ARCH_PAGE_TABLE_SYNC_MASK & PGTBL_P4D_MODIFIED)    \
            arch_sync_kernel_mappings(addr, addr);        \
    } while (0)

#endif /* _LINUX_PGALLOC_H */