arm64: Treat all entry code as non-kprobe-able

Entry symbols are not kprobe safe. So blacklist them for kprobing.

Signed-off-by: Pratyush Anand <panand@redhat.com>
Signed-off-by: David A. Long <dave.long@linaro.org>
Acked-by: Masami Hiramatsu <mhiramat@kernel.org>
[catalin.marinas@arm.com: Do not include syscall wrappers in .entry.text]
Signed-off-by: Catalin Marinas <catalin.marinas@arm.com>

1 files changed, 26 insertions, 0 deletions

diff --git a/arch/arm64/kernel/probes/kprobes.c b/arch/arm64/kernel/probes/kprobes.c
index 44968019ce0d..0fe2b6578163 100644
--- a/arch/arm64/kernel/probes/kprobes.c
+++ b/arch/arm64/kernel/probes/kprobes.c
@@ -30,6 +30,7 @@
 #include <asm/insn.h>
 #include <asm/uaccess.h>
 #include <asm/irq.h>
+#include <asm-generic/sections.h>
 
 #include "decode-insn.h"
 
@@ -519,6 +520,31 @@ int __kprobes longjmp_break_handler(struct kprobe *p, struct pt_regs *regs)
         return 1;
 }
 
+bool arch_within_kprobe_blacklist(unsigned long addr)
+{
+        extern char __idmap_text_start[], __idmap_text_end[];
+        extern char __hyp_idmap_text_start[], __hyp_idmap_text_end[];
+
+        if ((addr >= (unsigned long)__kprobes_text_start &&
+            addr < (unsigned long)__kprobes_text_end) ||
+            (addr >= (unsigned long)__entry_text_start &&
+            addr < (unsigned long)__entry_text_end) ||
+            (addr >= (unsigned long)__idmap_text_start &&
+            addr < (unsigned long)__idmap_text_end) ||
+            !!search_exception_tables(addr))
+                return true;
+
+        if (!is_kernel_in_hyp_mode()) {
+                if ((addr >= (unsigned long)__hyp_text_start &&
+                    addr < (unsigned long)__hyp_text_end) ||
+                    (addr >= (unsigned long)__hyp_idmap_text_start &&
+                    addr < (unsigned long)__hyp_idmap_text_end))
+                        return true;
+        }
+
+        return false;
+}
+
 int __init arch_init_kprobes(void)
 {
         return 0;