| 1 | /* SPDX-License-Identifier: GPL-2.0-only */ |
| 2 | /* |
| 3 | * Copyright (C) 2009 Chen Liqin <liqin.chen@sunplusct.com> |
| 4 | * Copyright (C) 2012 Regents of the University of California |
| 5 | */ |
| 6 | |
| 7 | #ifndef _ASM_RISCV_PGALLOC_H |
| 8 | #define _ASM_RISCV_PGALLOC_H |
| 9 | |
| 10 | #include <linux/mm.h> |
| 11 | #include <asm/sbi.h> |
| 12 | #include <asm/tlb.h> |
| 13 | |
| 14 | #ifdef CONFIG_MMU |
| 15 | #define __HAVE_ARCH_PUD_FREE |
| 16 | #include <asm-generic/pgalloc.h> |
| 17 | |
| 18 | static inline void pmd_populate_kernel(struct mm_struct *mm, |
| 19 | pmd_t *pmd, pte_t *pte) |
| 20 | { |
| 21 | unsigned long pfn = virt_to_pfn(pte); |
| 22 | |
| 23 | set_pmd(pmd, __pmd((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 24 | } |
| 25 | |
| 26 | static inline void pmd_populate(struct mm_struct *mm, |
| 27 | pmd_t *pmd, pgtable_t pte) |
| 28 | { |
| 29 | unsigned long pfn = virt_to_pfn(page_address(pte)); |
| 30 | |
| 31 | set_pmd(pmd, __pmd((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 32 | } |
| 33 | |
| 34 | #ifndef __PAGETABLE_PMD_FOLDED |
| 35 | static inline void pud_populate(struct mm_struct *mm, pud_t *pud, pmd_t *pmd) |
| 36 | { |
| 37 | unsigned long pfn = virt_to_pfn(pmd); |
| 38 | |
| 39 | set_pud(pud, __pud((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 40 | } |
| 41 | |
| 42 | static inline void p4d_populate(struct mm_struct *mm, p4d_t *p4d, pud_t *pud) |
| 43 | { |
| 44 | if (pgtable_l4_enabled) { |
| 45 | unsigned long pfn = virt_to_pfn(pud); |
| 46 | |
| 47 | set_p4d(p4d, __p4d((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | static inline void p4d_populate_safe(struct mm_struct *mm, p4d_t *p4d, |
| 52 | pud_t *pud) |
| 53 | { |
| 54 | if (pgtable_l4_enabled) { |
| 55 | unsigned long pfn = virt_to_pfn(pud); |
| 56 | |
| 57 | set_p4d_safe(p4d, |
| 58 | __p4d((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 59 | } |
| 60 | } |
| 61 | |
| 62 | static inline void pgd_populate(struct mm_struct *mm, pgd_t *pgd, p4d_t *p4d) |
| 63 | { |
| 64 | if (pgtable_l5_enabled) { |
| 65 | unsigned long pfn = virt_to_pfn(p4d); |
| 66 | |
| 67 | set_pgd(pgd, __pgd((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 68 | } |
| 69 | } |
| 70 | |
| 71 | static inline void pgd_populate_safe(struct mm_struct *mm, pgd_t *pgd, |
| 72 | p4d_t *p4d) |
| 73 | { |
| 74 | if (pgtable_l5_enabled) { |
| 75 | unsigned long pfn = virt_to_pfn(p4d); |
| 76 | |
| 77 | set_pgd_safe(pgd, |
| 78 | __pgd((pfn << _PAGE_PFN_SHIFT) | _PAGE_TABLE)); |
| 79 | } |
| 80 | } |
| 81 | |
| 82 | #define pud_free pud_free |
| 83 | static inline void pud_free(struct mm_struct *mm, pud_t *pud) |
| 84 | { |
| 85 | if (pgtable_l4_enabled) |
| 86 | __pud_free(mm, pud); |
| 87 | } |
| 88 | |
| 89 | static inline void __pud_free_tlb(struct mmu_gather *tlb, pud_t *pud, |
| 90 | unsigned long addr) |
| 91 | { |
| 92 | if (pgtable_l4_enabled) |
| 93 | tlb_remove_ptdesc(tlb, virt_to_ptdesc(pud)); |
| 94 | } |
| 95 | |
| 96 | static inline void __p4d_free_tlb(struct mmu_gather *tlb, p4d_t *p4d, |
| 97 | unsigned long addr) |
| 98 | { |
| 99 | if (pgtable_l5_enabled) |
| 100 | tlb_remove_ptdesc(tlb, virt_to_ptdesc(p4d)); |
| 101 | } |
| 102 | #endif /* __PAGETABLE_PMD_FOLDED */ |
| 103 | |
| 104 | static inline void sync_kernel_mappings(pgd_t *pgd) |
| 105 | { |
| 106 | memcpy(pgd + USER_PTRS_PER_PGD, |
| 107 | init_mm.pgd + USER_PTRS_PER_PGD, |
| 108 | (PTRS_PER_PGD - USER_PTRS_PER_PGD) * sizeof(pgd_t)); |
| 109 | } |
| 110 | |
| 111 | static inline pgd_t *pgd_alloc(struct mm_struct *mm) |
| 112 | { |
| 113 | pgd_t *pgd; |
| 114 | |
| 115 | pgd = __pgd_alloc(mm, 0); |
| 116 | if (likely(pgd != NULL)) { |
| 117 | /* Copy kernel mappings */ |
| 118 | sync_kernel_mappings(pgd); |
| 119 | } |
| 120 | return pgd; |
| 121 | } |
| 122 | |
| 123 | #ifndef __PAGETABLE_PMD_FOLDED |
| 124 | |
| 125 | static inline void __pmd_free_tlb(struct mmu_gather *tlb, pmd_t *pmd, |
| 126 | unsigned long addr) |
| 127 | { |
| 128 | tlb_remove_ptdesc(tlb, virt_to_ptdesc(pmd)); |
| 129 | } |
| 130 | |
| 131 | #endif /* __PAGETABLE_PMD_FOLDED */ |
| 132 | |
| 133 | static inline void __pte_free_tlb(struct mmu_gather *tlb, pgtable_t pte, |
| 134 | unsigned long addr) |
| 135 | { |
| 136 | tlb_remove_ptdesc(tlb, page_ptdesc(pte)); |
| 137 | } |
| 138 | #endif /* CONFIG_MMU */ |
| 139 | |
| 140 | #endif /* _ASM_RISCV_PGALLOC_H */ |