1 /* SPDX-License-Identifier: GPL-2.0-only */ 2 /* 3 * Based on arch/arm/include/asm/tlb.h 4 * 5 * Copyright (C) 2002 Russell King 6 * Copyright (C) 2012 ARM Ltd. 7 */ 8 #ifndef __ASM_TLB_H 9 #define __ASM_TLB_H 10 11 #include <linux/pagemap.h> 12 #include <linux/swap.h> 13 14 static inline void __tlb_remove_table(void *_table) 15 { 16 free_page_and_swap_cache((struct page *)_table); 17 } 18 19 #define tlb_flush tlb_flush 20 static void tlb_flush(struct mmu_gather *tlb); 21 22 #include <asm-generic/tlb.h> 23 24 /* 25 * get the tlbi levels in arm64. Default value is 0 if more than one 26 * of cleared_* is set or neither is set. 27 * Arm64 doesn't support p4ds now. 28 */ 29 static inline int tlb_get_level(struct mmu_gather *tlb) 30 { 31 if (tlb->cleared_ptes && !(tlb->cleared_pmds || 32 tlb->cleared_puds || 33 tlb->cleared_p4ds)) 34 return 3; 35 36 if (tlb->cleared_pmds && !(tlb->cleared_ptes || 37 tlb->cleared_puds || 38 tlb->cleared_p4ds)) 39 return 2; 40 41 if (tlb->cleared_puds && !(tlb->cleared_ptes || 42 tlb->cleared_pmds || 43 tlb->cleared_p4ds)) 44 return 1; 45 46 return 0; 47 } 48 49 static inline void tlb_flush(struct mmu_gather *tlb) 50 { 51 struct vm_area_struct vma = TLB_FLUSH_VMA(tlb->mm, 0); 52 bool last_level = !tlb->freed_tables; 53 unsigned long stride = tlb_get_unmap_size(tlb); 54 int tlb_level = tlb_get_level(tlb); 55 56 /* 57 * If we're tearing down the address space then we only care about 58 * invalidating the walk-cache, since the ASID allocator won't 59 * reallocate our ASID without invalidating the entire TLB. 60 */ 61 if (tlb->fullmm) { 62 if (!last_level) 63 flush_tlb_mm(tlb->mm); 64 return; 65 } 66 67 __flush_tlb_range(&vma, tlb->start, tlb->end, stride, 68 last_level, tlb_level); 69 } 70 71 static inline void __pte_free_tlb(struct mmu_gather *tlb, pgtable_t pte, 72 unsigned long addr) 73 { 74 pgtable_pte_page_dtor(pte); 75 tlb_remove_table(tlb, pte); 76 } 77 78 #if CONFIG_PGTABLE_LEVELS > 2 79 static inline void __pmd_free_tlb(struct mmu_gather *tlb, pmd_t *pmdp, 80 unsigned long addr) 81 { 82 struct page *page = virt_to_page(pmdp); 83 84 pgtable_pmd_page_dtor(page); 85 tlb_remove_table(tlb, page); 86 } 87 #endif 88 89 #if CONFIG_PGTABLE_LEVELS > 3 90 static inline void __pud_free_tlb(struct mmu_gather *tlb, pud_t *pudp, 91 unsigned long addr) 92 { 93 tlb_remove_table(tlb, virt_to_page(pudp)); 94 } 95 #endif 96 97 #endif 98