3 files changed, 26 insertions, 38 deletions

diff --git a/arch/powerpc/kernel/signal.c b/arch/powerpc/kernel/signal.c
index a2dc75793bd5..3b997118df50 100644
--- a/arch/powerpc/kernel/signal.c
+++ b/arch/powerpc/kernel/signal.c
@@ -158,10 +158,8 @@ static int do_signal(struct pt_regs *regs)
 
 void do_notify_resume(struct pt_regs *regs, unsigned long thread_info_flags)
 {
-        if (thread_info_flags & _TIF_UPROBE) {
-                clear_thread_flag(TIF_UPROBE);
+        if (thread_info_flags & _TIF_UPROBE)
                 uprobe_notify_resume(regs);
-        }
 
         if (thread_info_flags & _TIF_SIGPENDING)
                 do_signal(regs);
diff --git a/arch/powerpc/kernel/uprobes.c b/arch/powerpc/kernel/uprobes.c
index d2d46d1014f8..bc77834dbf43 100644
--- a/arch/powerpc/kernel/uprobes.c
+++ b/arch/powerpc/kernel/uprobes.c
@@ -64,6 +64,8 @@ int arch_uprobe_pre_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
         autask->saved_trap_nr = current->thread.trap_nr;
         current->thread.trap_nr = UPROBE_TRAP_NR;
         regs->nip = current->utask->xol_vaddr;
+
+        user_enable_single_step(current);
         return 0;
 }
 
@@ -119,6 +121,8 @@ int arch_uprobe_post_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
          * to be executed.
          */
         regs->nip = utask->vaddr + MAX_UINSN_BYTES;
+
+        user_disable_single_step(current);
         return 0;
 }
 
@@ -162,6 +166,8 @@ void arch_uprobe_abort_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
 
         current->thread.trap_nr = utask->autask.saved_trap_nr;
         instruction_pointer_set(regs, utask->vaddr);
+
+        user_disable_single_step(current);
 }
 
 /*
diff --git a/arch/x86/kernel/uprobes.c b/arch/x86/kernel/uprobes.c
index aafa5557b396..c71025b67462 100644
--- a/arch/x86/kernel/uprobes.c
+++ b/arch/x86/kernel/uprobes.c
@@ -478,6 +478,11 @@ int arch_uprobe_pre_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
         regs->ip = current->utask->xol_vaddr;
         pre_xol_rip_insn(auprobe, regs, autask);
 
+        autask->saved_tf = !!(regs->flags & X86_EFLAGS_TF);
+        regs->flags |= X86_EFLAGS_TF;
+        if (test_tsk_thread_flag(current, TIF_BLOCKSTEP))
+                set_task_blockstep(current, false);
+
         return 0;
 }
 
@@ -603,6 +608,16 @@ int arch_uprobe_post_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
         if (auprobe->fixups & UPROBE_FIX_CALL)
                 result = adjust_ret_addr(regs->sp, correction);
 
+        /*
+         * arch_uprobe_pre_xol() doesn't save the state of TIF_BLOCKSTEP
+         * so we can get an extra SIGTRAP if we do not clear TF. We need
+         * to examine the opcode to make it right.
+         */
+        if (utask->autask.saved_tf)
+                send_sig(SIGTRAP, current, 0);
+        else if (!(auprobe->fixups & UPROBE_FIX_SETF))
+                regs->flags &= ~X86_EFLAGS_TF;
+
         return result;
 }
 
@@ -647,6 +662,10 @@ void arch_uprobe_abort_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
         current->thread.trap_nr = utask->autask.saved_trap_nr;
         handle_riprel_post_xol(auprobe, regs, NULL);
         instruction_pointer_set(regs, utask->vaddr);
+
+        /* clear TF if it was set by us in arch_uprobe_pre_xol() */
+        if (!utask->autask.saved_tf)
+                regs->flags &= ~X86_EFLAGS_TF;
 }
 
 /*
@@ -676,38 +695,3 @@ bool arch_uprobe_skip_sstep(struct arch_uprobe *auprobe, struct pt_regs *regs)
                 send_sig(SIGTRAP, current, 0);
         return ret;
 }
-
-void arch_uprobe_enable_step(struct arch_uprobe *auprobe)
-{
-        struct task_struct *task = current;
-        struct arch_uprobe_task *autask = &task->utask->autask;
-        struct pt_regs *regs = task_pt_regs(task);
-
-        autask->saved_tf = !!(regs->flags & X86_EFLAGS_TF);
-
-        regs->flags |= X86_EFLAGS_TF;
-        if (test_tsk_thread_flag(task, TIF_BLOCKSTEP))
-                set_task_blockstep(task, false);
-}
-
-void arch_uprobe_disable_step(struct arch_uprobe *auprobe)
-{
-        struct task_struct *task = current;
-        struct arch_uprobe_task *autask = &task->utask->autask;
-        bool trapped = (task->utask->state == UTASK_SSTEP_TRAPPED);
-        struct pt_regs *regs = task_pt_regs(task);
-        /*
-         * The state of TIF_BLOCKSTEP was not saved so we can get an extra
-         * SIGTRAP if we do not clear TF. We need to examine the opcode to
-         * make it right.
-         */
-        if (unlikely(trapped)) {
-                if (!autask->saved_tf)
-                        regs->flags &= ~X86_EFLAGS_TF;
-        } else {
-                if (autask->saved_tf)
-                        send_sig(SIGTRAP, task, 0);
-                else if (!(auprobe->fixups & UPROBE_FIX_SETF))
-                        regs->flags &= ~X86_EFLAGS_TF;
-        }
-}