arm64: kprobes: Cleanup jprobe_return

jprobe_return seems to have aged badly. Comments referring to
non-existent behaviours, and a dangerous habit of messing
with registers without telling the compiler.

This patches applies the following remedies:
- Fix the comments to describe the actual behaviour
- Tidy up the asm sequence to directly assign the
  stack pointer without clobbering extra registers
- Mark the rest of the function as unreachable() so
  that the compiler knows that there is no need for
  an epilogue
- Stop making jprobe_return_break a global function
  (you really don't want to call that guy, and it isn't
  even a function).

Tested with tcp_probe.

Signed-off-by: Marc Zyngier <marc.zyngier@arm.com>
Signed-off-by: Catalin Marinas <catalin.marinas@arm.com>

diff --git a/arch/arm64/kernel/probes/kprobes.c b/arch/arm64/kernel/probes/kprobes.c
index 09f8ae98da5ad1f566b646d22e014d9c84a94522..973c15df52111d7ea55a9b385bdfa12c0b6f4d2e 100644 (file)
--- a/arch/arm64/kernel/probes/kprobes.c
+++ b/arch/arm64/kernel/probes/kprobes.c
@@ -34,8 +34,6 @@
 
 #include "decode-insn.h"
 
-void jprobe_return_break(void);
-
 DEFINE_PER_CPU(struct kprobe *, current_kprobe) = NULL;
 DEFINE_PER_CPU(struct kprobe_ctlblk, kprobe_ctlblk);
 
@@ -516,18 +514,17 @@ void __kprobes jprobe_return(void)
        /*
         * Jprobe handler return by entering break exception,
         * encoded same as kprobe, but with following conditions
-        * -a magic number in x0 to identify from rest of other kprobes.
+        * -a special PC to identify it from the other kprobes.
         * -restore stack addr to original saved pt_regs
         */
-       asm volatile ("ldr x0, [%0]\n\t"
-                     "mov sp, x0\n\t"
-                     ".globl jprobe_return_break\n\t"
-                     "jprobe_return_break:\n\t"
-                     "brk %1\n\t"
-                     :
-                     : "r"(&kcb->jprobe_saved_regs.sp),
-                     "I"(BRK64_ESR_KPROBES)
-                     : "memory");
+       asm volatile("                          mov sp, %0      \n"
+                    "jprobe_return_break:      brk %1          \n"
+                    :
+                    : "r" (kcb->jprobe_saved_regs.sp),
+                      "I" (BRK64_ESR_KPROBES)
+                    : "memory");
+
+       unreachable();
 }
 
 int __kprobes longjmp_break_handler(struct kprobe *p, struct pt_regs *regs)
@@ -536,6 +533,7 @@ int __kprobes longjmp_break_handler(struct kprobe *p, struct pt_regs *regs)
        long stack_addr = kcb->jprobe_saved_regs.sp;
        long orig_sp = kernel_stack_pointer(regs);
        struct jprobe *jp = container_of(p, struct jprobe, kp);
+       extern const char jprobe_return_break[];
 
        if (instruction_pointer(regs) != (u64) jprobe_return_break)
                return 0;