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
pmd_populate_kernel(struct mm_struct * mm,pmd_t * pmd,pte_t * pte)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
pmd_populate(struct mm_struct * mm,pmd_t * pmd,pgtable_t pte)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
pud_populate(struct mm_struct * mm,pud_t * pud,pmd_t * pmd)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
p4d_populate(struct mm_struct * mm,p4d_t * p4d,pud_t * pud)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 /*
46 * Initialize a new pmd table with invalid pointers.
47 */
48 extern void pmd_init(unsigned long page, unsigned long pagetable);
49
50 /*
51 * Initialize a new pgd / pmd table with invalid pointers.
52 */
53 extern void pgd_init(unsigned long page);
54 extern pgd_t *pgd_alloc(struct mm_struct *mm);
55
56 #define __pte_free_tlb(tlb, pte, address) \
57 do { \
58 pgtable_pte_page_dtor(pte); \
59 tlb_remove_page((tlb), pte); \
60 } while (0)
61
62 #ifndef __PAGETABLE_PMD_FOLDED
63
pmd_alloc_one(struct mm_struct * mm,unsigned long address)64 static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
65 {
66 pmd_t *pmd;
67 struct page *pg;
68
69 pg = alloc_page(GFP_KERNEL_ACCOUNT);
70 if (!pg)
71 return NULL;
72
73 if (!pgtable_pmd_page_ctor(pg)) {
74 __free_page(pg);
75 return NULL;
76 }
77
78 pmd = (pmd_t *)page_address(pg);
79 pmd_init((unsigned long)pmd, (unsigned long)invalid_pte_table);
80 return pmd;
81 }
82
83 #define __pmd_free_tlb(tlb, x, addr) pmd_free((tlb)->mm, x)
84
85 #endif
86
87 #ifndef __PAGETABLE_PUD_FOLDED
88
pud_alloc_one(struct mm_struct * mm,unsigned long address)89 static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
90 {
91 pud_t *pud;
92
93 pud = (pud_t *) __get_free_page(GFP_KERNEL);
94 if (pud)
95 pud_init((unsigned long)pud, (unsigned long)invalid_pmd_table);
96 return pud;
97 }
98
99 #define __pud_free_tlb(tlb, x, addr) pud_free((tlb)->mm, x)
100
101 #endif /* __PAGETABLE_PUD_FOLDED */
102
103 #endif /* _ASM_PGALLOC_H */
104