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