/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
 */
#ifndef _ASM_PGALLOC_H
#define _ASM_PGALLOC_H

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

#define __HAVE_ARCH_PMD_ALLOC_ONE
#define __HAVE_ARCH_PUD_ALLOC_ONE
#define __HAVE_ARCH_PTE_ALLOC_ONE_KERNEL
#include <asm-generic/pgalloc.h>

static inline void pmd_populate_kernel(struct mm_struct *mm,
                                       pmd_t *pmd, pte_t *pte)
{
        set_pmd(pmd, __pmd((unsigned long)pte));
}

static inline void pmd_populate(struct mm_struct *mm, pmd_t *pmd, pgtable_t pte)
{
        set_pmd(pmd, __pmd((unsigned long)page_address(pte)));
}

#ifndef __PAGETABLE_PMD_FOLDED

static inline void pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd)
{
        set_pud(pud, __pud((unsigned long)pmd));
}
#endif

#ifndef __PAGETABLE_PUD_FOLDED

static inline void p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud)
{
        set_p4d(p4d, __p4d((unsigned long)pud));
}

#endif /* __PAGETABLE_PUD_FOLDED */

extern void pagetable_init(void);

extern pgd_t *pgd_alloc(struct mm_struct *mm);

static inline pte_t *pte_alloc_one_kernel(struct mm_struct *mm)
{
        pte_t *pte = __pte_alloc_one_kernel(mm);

        if (pte)
                kernel_pte_init(pte);

        return pte;
}

#define __pte_free_tlb(tlb, pte, address)       tlb_remove_ptdesc((tlb), page_ptdesc(pte))

#ifndef __PAGETABLE_PMD_FOLDED

static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
{
        pmd_t *pmd;
        struct ptdesc *ptdesc;

        ptdesc = pagetable_alloc(GFP_KERNEL_ACCOUNT, 0);
        if (!ptdesc)
                return NULL;

        if (!pagetable_pmd_ctor(mm, ptdesc)) {
                pagetable_free(ptdesc);
                return NULL;
        }

        pmd = ptdesc_address(ptdesc);
        pmd_init(pmd);
        return pmd;
}

#define __pmd_free_tlb(tlb, x, addr)    tlb_remove_ptdesc((tlb), virt_to_ptdesc(x))

#endif

#ifndef __PAGETABLE_PUD_FOLDED

static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
{
        pud_t *pud;
        struct ptdesc *ptdesc = pagetable_alloc(GFP_KERNEL, 0);

        if (!ptdesc)
                return NULL;
        pagetable_pud_ctor(ptdesc);
        pud = ptdesc_address(ptdesc);

        pud_init(pud);
        return pud;
}

#define __pud_free_tlb(tlb, x, addr)    tlb_remove_ptdesc((tlb), virt_to_ptdesc(x))

#endif /* __PAGETABLE_PUD_FOLDED */

extern pte_t * __init populate_kernel_pte(unsigned long addr);
#endif /* _ASM_PGALLOC_H */