1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
4  */
5 #ifndef _ASM_PGALLOC_H
6 #define _ASM_PGALLOC_H
7 
8 #include <linux/mm.h>
9 #include <linux/sched.h>
10 
11 #define __HAVE_ARCH_PMD_ALLOC_ONE
12 #define __HAVE_ARCH_PUD_ALLOC_ONE
13 #include <asm-generic/pgalloc.h>
14 
15 static inline void pmd_populate_kernel(struct mm_struct *mm,
16 				       pmd_t *pmd, pte_t *pte)
17 {
18 	set_pmd(pmd, __pmd((unsigned long)pte));
19 }
20 
21 static inline void pmd_populate(struct mm_struct *mm, pmd_t *pmd, pgtable_t pte)
22 {
23 	set_pmd(pmd, __pmd((unsigned long)page_address(pte)));
24 }
25 
26 #ifndef __PAGETABLE_PMD_FOLDED
27 
28 static inline void pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd)
29 {
30 	set_pud(pud, __pud((unsigned long)pmd));
31 }
32 #endif
33 
34 #ifndef __PAGETABLE_PUD_FOLDED
35 
36 static inline void p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud)
37 {
38 	set_p4d(p4d, __p4d((unsigned long)pud));
39 }
40 
41 #endif /* __PAGETABLE_PUD_FOLDED */
42 
43 extern void pagetable_init(void);
44 
45 extern pgd_t *pgd_alloc(struct mm_struct *mm);
46 
47 #define __pte_free_tlb(tlb, pte, address)			\
48 do {							\
49 	pgtable_pte_page_dtor(pte);			\
50 	tlb_remove_page((tlb), pte);			\
51 } while (0)
52 
53 #ifndef __PAGETABLE_PMD_FOLDED
54 
55 static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
56 {
57 	pmd_t *pmd;
58 	struct page *pg;
59 
60 	pg = alloc_page(GFP_KERNEL_ACCOUNT);
61 	if (!pg)
62 		return NULL;
63 
64 	if (!pgtable_pmd_page_ctor(pg)) {
65 		__free_page(pg);
66 		return NULL;
67 	}
68 
69 	pmd = (pmd_t *)page_address(pg);
70 	pmd_init(pmd);
71 	return pmd;
72 }
73 
74 #define __pmd_free_tlb(tlb, x, addr)	pmd_free((tlb)->mm, x)
75 
76 #endif
77 
78 #ifndef __PAGETABLE_PUD_FOLDED
79 
80 static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
81 {
82 	pud_t *pud;
83 
84 	pud = (pud_t *) __get_free_page(GFP_KERNEL);
85 	if (pud)
86 		pud_init(pud);
87 	return pud;
88 }
89 
90 #define __pud_free_tlb(tlb, x, addr)	pud_free((tlb)->mm, x)
91 
92 #endif /* __PAGETABLE_PUD_FOLDED */
93 
94 #endif /* _ASM_PGALLOC_H */
95