1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * Copyright (C) 2012 Regents of the University of California
4 */
5
6#ifndef _ASM_RISCV_TLB_H
7#define _ASM_RISCV_TLB_H
8
9struct mmu_gather;
10
11static void tlb_flush(struct mmu_gather *tlb);
12
13#ifdef CONFIG_MMU
14#include <linux/swap.h>
15
16/*
17 * While riscv platforms with riscv_ipi_for_rfence as true require an IPI to
18 * perform TLB shootdown, some platforms with riscv_ipi_for_rfence as false use
19 * SBI to perform TLB shootdown. To keep software pagetable walkers safe in this
20 * case we switch to RCU based table free (MMU_GATHER_RCU_TABLE_FREE). See the
21 * comment below 'ifdef CONFIG_MMU_GATHER_RCU_TABLE_FREE' in include/asm-generic/tlb.h
22 * for more details.
23 */
24static inline void __tlb_remove_table(void *table)
25{
26	free_page_and_swap_cache(table);
27}
28
29#endif /* CONFIG_MMU */
30
31#define tlb_flush tlb_flush
32#include <asm-generic/tlb.h>
33
34static inline void tlb_flush(struct mmu_gather *tlb)
35{
36#ifdef CONFIG_MMU
37	if (tlb->fullmm || tlb->need_flush_all || tlb->freed_tables)
38		flush_tlb_mm(tlb->mm);
39	else
40		flush_tlb_mm_range(tlb->mm, tlb->start, tlb->end,
41				   tlb_get_unmap_size(tlb));
42#endif
43}
44
45#endif /* _ASM_RISCV_TLB_H */
46