1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * Copyright (C) 2020 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 #define pmd_pgtable(pmd) pmd_page(pmd)
26
27 #ifndef __PAGETABLE_PMD_FOLDED
28
pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd)29 static inline void pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd)
30 {
31 set_pud(pud, __pud((unsigned long)pmd));
32 }
33 #endif
34
35 #ifndef __PAGETABLE_PUD_FOLDED
36
p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud)37 static inline void p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud)
38 {
39 set_p4d(p4d, __p4d((unsigned long)pud));
40 }
41
42 #endif /* __PAGETABLE_PUD_FOLDED */
43
44 extern void pagetable_init(void);
45
46 /*
47 * Initialize a new pmd table with invalid pointers.
48 */
49 extern void pmd_init(unsigned long page, unsigned long pagetable);
50
51 /*
52 * Initialize a new pgd / pmd table with invalid pointers.
53 */
54 extern void pgd_init(unsigned long page);
55 extern pgd_t *pgd_alloc(struct mm_struct *mm);
56
57 #define __pte_free_tlb(tlb,pte,address) \
58 do { \
59 pgtable_pte_page_dtor(pte); \
60 tlb_remove_page((tlb), pte); \
61 } while (0)
62
63 #ifndef __PAGETABLE_PMD_FOLDED
64
pmd_alloc_one(struct mm_struct *mm, unsigned long address)65 static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
66 {
67 pmd_t *pmd;
68 struct page *pg;
69
70 pg = alloc_pages(GFP_KERNEL_ACCOUNT, PMD_ORDER);
71 if (!pg)
72 return NULL;
73
74 if (!pgtable_pmd_page_ctor(pg)) {
75 __free_pages(pg, PMD_ORDER);
76 return NULL;
77 }
78
79 pmd = (pmd_t *)page_address(pg);
80 pmd_init((unsigned long)pmd, (unsigned long)invalid_pte_table);
81 return pmd;
82 }
83
84 #define __pmd_free_tlb(tlb, x, addr) pmd_free((tlb)->mm, x)
85
86 #endif
87
88 #ifndef __PAGETABLE_PUD_FOLDED
89
pud_alloc_one(struct mm_struct *mm, unsigned long address)90 static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
91 {
92 pud_t *pud;
93
94 pud = (pud_t *) __get_free_pages(GFP_KERNEL, PUD_ORDER);
95 if (pud)
96 pud_init((unsigned long)pud, (unsigned long)invalid_pmd_table);
97 return pud;
98 }
99
100 #define __pud_free_tlb(tlb, x, addr) pud_free((tlb)->mm, x)
101
102 #endif /* __PAGETABLE_PUD_FOLDED */
103
104 extern pte_t * __init populate_kernel_pte(unsigned long addr);
105 #endif /* _ASM_PGALLOC_H */
106