Merge tag 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/rmk/linux

[linux-block.git] / arch / arm / mm / tlb.c

diff --git a/arch/arm/mm/tlb.c b/arch/arm/mm/tlb.c
new file mode 100644 (file)
index 0000000..4235979
--- /dev/null
+++ b/arch/arm/mm/tlb.c
@@ -0,0 +1,84 @@
+// SPDX-License-Identifier: GPL-2.0-only
+// Copyright 2024 Google LLC
+// Author: Ard Biesheuvel <ardb@google.com>
+
+#include <linux/types.h>
+#include <asm/tlbflush.h>
+
+#ifdef CONFIG_CPU_TLB_V4WT
+void v4_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void v4_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns v4_tlb_fns __initconst = {
+       .flush_user_range       = v4_flush_user_tlb_range,
+       .flush_kern_range       = v4_flush_kern_tlb_range,
+       .tlb_flags              = v4_tlb_flags,
+};
+#endif
+
+#ifdef CONFIG_CPU_TLB_V4WB
+void v4wb_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void v4wb_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns v4wb_tlb_fns __initconst = {
+       .flush_user_range       = v4wb_flush_user_tlb_range,
+       .flush_kern_range       = v4wb_flush_kern_tlb_range,
+       .tlb_flags              = v4wb_tlb_flags,
+};
+#endif
+
+#if defined(CONFIG_CPU_TLB_V4WBI) || defined(CONFIG_CPU_TLB_FEROCEON)
+void v4wbi_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void v4wbi_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns v4wbi_tlb_fns __initconst = {
+       .flush_user_range       = v4wbi_flush_user_tlb_range,
+       .flush_kern_range       = v4wbi_flush_kern_tlb_range,
+       .tlb_flags              = v4wbi_tlb_flags,
+};
+#endif
+
+#ifdef CONFIG_CPU_TLB_V6
+void v6wbi_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void v6wbi_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns v6wbi_tlb_fns __initconst = {
+       .flush_user_range       = v6wbi_flush_user_tlb_range,
+       .flush_kern_range       = v6wbi_flush_kern_tlb_range,
+       .tlb_flags              = v6wbi_tlb_flags,
+};
+#endif
+
+#ifdef CONFIG_CPU_TLB_V7
+void v7wbi_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void v7wbi_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns v7wbi_tlb_fns __initconst = {
+       .flush_user_range       = v7wbi_flush_user_tlb_range,
+       .flush_kern_range       = v7wbi_flush_kern_tlb_range,
+       .tlb_flags              = IS_ENABLED(CONFIG_SMP) ? v7wbi_tlb_flags_smp
+                                                        : v7wbi_tlb_flags_up,
+};
+
+#ifdef CONFIG_SMP_ON_UP
+/* This will be run-time patched so the offset better be right */
+static_assert(offsetof(struct cpu_tlb_fns, tlb_flags) == 8);
+
+asm("  .pushsection    \".alt.smp.init\", \"a\"                \n" \
+    "  .align          2                                       \n" \
+    "  .long           v7wbi_tlb_fns + 8 - .                   \n" \
+    "  .long "         __stringify(v7wbi_tlb_flags_up) "       \n" \
+    "  .popsection                                             \n");
+#endif
+#endif
+
+#ifdef CONFIG_CPU_TLB_FA
+void fa_flush_user_tlb_range(unsigned long, unsigned long, struct vm_area_struct *);
+void fa_flush_kern_tlb_range(unsigned long, unsigned long);
+
+struct cpu_tlb_fns fa_tlb_fns __initconst = {
+       .flush_user_range       = fa_flush_user_tlb_range,
+       .flush_kern_range       = fa_flush_kern_tlb_range,
+       .tlb_flags              = fa_tlb_flags,
+};
+#endif