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
15static 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
21static 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
28static 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
36static 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
43extern void pagetable_init(void);
44
45extern pgd_t *pgd_alloc(struct mm_struct *mm);
46
47#define __pte_free_tlb(tlb, pte, address)			\
48do {								\
49	pagetable_pte_dtor(page_ptdesc(pte));			\
50	tlb_remove_page_ptdesc((tlb), page_ptdesc(pte));	\
51} while (0)
52
53#ifndef __PAGETABLE_PMD_FOLDED
54
55static inline pmd_t *pmd_alloc_one(struct mm_struct *mm, unsigned long address)
56{
57	pmd_t *pmd;
58	struct ptdesc *ptdesc;
59
60	ptdesc = pagetable_alloc(GFP_KERNEL_ACCOUNT, 0);
61	if (!ptdesc)
62		return NULL;
63
64	if (!pagetable_pmd_ctor(ptdesc)) {
65		pagetable_free(ptdesc);
66		return NULL;
67	}
68
69	pmd = ptdesc_address(ptdesc);
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
80static inline pud_t *pud_alloc_one(struct mm_struct *mm, unsigned long address)
81{
82	pud_t *pud;
83	struct ptdesc *ptdesc = pagetable_alloc(GFP_KERNEL & ~__GFP_HIGHMEM, 0);
84
85	if (!ptdesc)
86		return NULL;
87	pagetable_pud_ctor(ptdesc);
88	pud = ptdesc_address(ptdesc);
89
90	pud_init(pud);
91	return pud;
92}
93
94#define __pud_free_tlb(tlb, x, addr)	pud_free((tlb)->mm, x)
95
96#endif /* __PAGETABLE_PUD_FOLDED */
97
98extern pte_t * __init populate_kernel_pte(unsigned long addr);
99#endif /* _ASM_PGALLOC_H */
100