arch/loongarch/include/asm/pgalloc.h

   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         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
  55 static 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
  80 static 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         pud = ptdesc_address(ptdesc);
  88
  89         pud_init(pud);
  90         return pud;
  91 }
  92
  93 #define __pud_free_tlb(tlb, x, addr)    pud_free((tlb)->mm, x)
  94
  95 #endif /* __PAGETABLE_PUD_FOLDED */
  96
  97 #endif /* _ASM_PGALLOC_H */