1 /* SPDX-License-Identifier: GPL-2.0 */
3 * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
9 #include <linux/sched.h>
11 #define __HAVE_ARCH_PMD_ALLOC_ONE
12 #define __HAVE_ARCH_PUD_ALLOC_ONE
13 #include <asm-generic/pgalloc.h>
15 static inline void pmd_populate_kernel(struct mm_struct *mm,
16 pmd_t *pmd, pte_t *pte)
18 set_pmd(pmd, __pmd((unsigned long)pte));
21 static inline void pmd_populate(struct mm_struct *mm, pmd_t *pmd, pgtable_t pte)
23 set_pmd(pmd, __pmd((unsigned long)page_address(pte)));
26 #ifndef __PAGETABLE_PMD_FOLDED
28 static inline void pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd)
30 set_pud(pud, __pud((unsigned long)pmd));
34 #ifndef __PAGETABLE_PUD_FOLDED
36 static inline void p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud)
38 set_p4d(p4d, __p4d((unsigned long)pud));
41 #endif /* __PAGETABLE_PUD_FOLDED */
43 extern void pagetable_init(void);
45 extern pgd_t *pgd_alloc(struct mm_struct *mm);
47 #define __pte_free_tlb(tlb, pte, address) \
49 pagetable_pte_dtor(page_ptdesc(pte)); \
50 tlb_remove_page_ptdesc((tlb), page_ptdesc(pte)); \
53 #ifndef __PAGETABLE_PMD_FOLDED
55 static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
58 struct ptdesc *ptdesc;
60 ptdesc = pagetable_alloc(GFP_KERNEL_ACCOUNT, 0);
64 if (!pagetable_pmd_ctor(ptdesc)) {
65 pagetable_free(ptdesc);
69 pmd = ptdesc_address(ptdesc);
74 #define __pmd_free_tlb(tlb, x, addr) pmd_free((tlb)->mm, x)
78 #ifndef __PAGETABLE_PUD_FOLDED
80 static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
83 struct ptdesc *ptdesc = pagetable_alloc(GFP_KERNEL & ~__GFP_HIGHMEM, 0);
87 pagetable_pud_ctor(ptdesc);
88 pud = ptdesc_address(ptdesc);
94 #define __pud_free_tlb(tlb, x, addr) pud_free((tlb)->mm, x)
96 #endif /* __PAGETABLE_PUD_FOLDED */
98 extern pte_t * __init populate_kernel_pte(unsigned long addr);
99 #endif /* _ASM_PGALLOC_H */