lockdep: s390: connect the sysexit hook

Run the lockdep_sys_exit hook before returning to user space.

Reviewed-by: Martin Schwidefsky <schwidefsky@de.ibm.com>
Signed-off-by: Heiko Carstens <heiko.carstens@de.ibm.com>
Signed-off-by: Peter Zijlstra <a.p.zijlstra@chello.nl>
Signed-off-by: Ingo Molnar <mingo@elte.hu>

diff --git a/arch/s390/kernel/entry.S b/arch/s390/kernel/entry.S
index f3bceb165321e2bb3d57430a1b5fd8c90e7fbd9e..139ca153d5cc1dece7d4ac3679946d29a9baada9 100644 (file)
--- a/arch/s390/kernel/entry.S
+++ b/arch/s390/kernel/entry.S
@@ -68,9 +68,15 @@ STACK_SIZE  = 1 << STACK_SHIFT
        l       %r1,BASED(.Ltrace_irq_off)
        basr    %r14,%r1
        .endm
+
+       .macro  LOCKDEP_SYS_EXIT
+       l       %r1,BASED(.Llockdep_sys_exit)
+       basr    %r14,%r1
+       .endm
 #else
 #define TRACE_IRQS_ON
 #define TRACE_IRQS_OFF
+#define LOCKDEP_SYS_EXIT
 #endif
 
 /*
@@ -260,6 +266,7 @@ sysc_return:
        bno     BASED(sysc_leave)
        tm      __TI_flags+3(%r9),_TIF_WORK_SVC
        bnz     BASED(sysc_work)  # there is work to do (signals etc.)
+       LOCKDEP_SYS_EXIT
 sysc_leave:
        RESTORE_ALL __LC_RETURN_PSW,1
 
@@ -283,6 +290,7 @@ sysc_work:
        bo      BASED(sysc_restart)
        tm      __TI_flags+3(%r9),_TIF_SINGLE_STEP
        bo      BASED(sysc_singlestep)
+       LOCKDEP_SYS_EXIT
        b       BASED(sysc_leave)
 
 #
@@ -572,6 +580,7 @@ io_return:
 #endif
        tm      __TI_flags+3(%r9),_TIF_WORK_INT
        bnz     BASED(io_work)          # there is work to do (signals etc.)
+       LOCKDEP_SYS_EXIT
 io_leave:
        RESTORE_ALL __LC_RETURN_PSW,0
 io_done:
@@ -618,6 +627,7 @@ io_work_loop:
        bo      BASED(io_reschedule)
        tm      __TI_flags+3(%r9),(_TIF_SIGPENDING | _TIF_RESTORE_SIGMASK)
        bnz     BASED(io_sigpending)
+       LOCKDEP_SYS_EXIT
        b       BASED(io_leave)
 
 #
@@ -1040,6 +1050,8 @@ cleanup_io_leave_insn:
 .Ltrace_irq_on: .long  trace_hardirqs_on
 .Ltrace_irq_off:
                .long   trace_hardirqs_off
+.Llockdep_sys_exit:
+               .long   lockdep_sys_exit
 #endif
 .Lcritical_start:
                .long   __critical_start + 0x80000000

diff --git a/arch/s390/kernel/entry64.S b/arch/s390/kernel/entry64.S
index 9c0d5cc8269dd321a17da58a8046f2a20a2621ce..05e26d1fdf405dc21696b35aabb0217a6164d8d4 100644 (file)
--- a/arch/s390/kernel/entry64.S
+++ b/arch/s390/kernel/entry64.S
@@ -66,9 +66,14 @@ _TIF_WORK_INT = (_TIF_SIGPENDING | _TIF_RESTORE_SIGMASK | _TIF_NEED_RESCHED | \
        .macro  TRACE_IRQS_OFF
         brasl  %r14,trace_hardirqs_off
        .endm
+
+       .macro  LOCKDEP_SYS_EXIT
+        brasl  %r14,lockdep_sys_exit
+       .endm
 #else
 #define TRACE_IRQS_ON
 #define TRACE_IRQS_OFF
+#define LOCKDEP_SYS_EXIT
 #endif
 
        .macro  STORE_TIMER lc_offset
@@ -255,6 +260,7 @@ sysc_return:
        jno     sysc_leave
        tm      __TI_flags+7(%r9),_TIF_WORK_SVC
        jnz     sysc_work       # there is work to do (signals etc.)
+       LOCKDEP_SYS_EXIT
 sysc_leave:
        RESTORE_ALL __LC_RETURN_PSW,1
 
@@ -278,6 +284,7 @@ sysc_work:
        jo      sysc_restart
        tm      __TI_flags+7(%r9),_TIF_SINGLE_STEP
        jo      sysc_singlestep
+       LOCKDEP_SYS_EXIT
        j       sysc_leave
 
 #
@@ -558,6 +565,7 @@ io_return:
 #endif
        tm      __TI_flags+7(%r9),_TIF_WORK_INT
        jnz     io_work                 # there is work to do (signals etc.)
+       LOCKDEP_SYS_EXIT
 io_leave:
        RESTORE_ALL __LC_RETURN_PSW,0
 io_done:
@@ -605,6 +613,7 @@ io_work_loop:
        jo      io_reschedule
        tm      __TI_flags+7(%r9),(_TIF_SIGPENDING | _TIF_RESTORE_SIGMASK)
        jnz     io_sigpending
+       LOCKDEP_SYS_EXIT
        j       io_leave
 
 #