#define __ASM_CSKY_REGDEF_H
 
 #define syscallid      r1
-#define r11_sig                r11
-
 #define regs_syscallid(regs) regs->regs[9]
 
 /*
 
 #define __ASM_CSKY_REGDEF_H
 
 #define syscallid      r7
-#define r11_sig                r11
-
 #define regs_syscallid(regs) regs->regs[3]
 
 /*
 
 #define __ASM_CSKY_PTRACE_H
 
 #include <uapi/asm/ptrace.h>
+#include <asm/traps.h>
+#include <linux/types.h>
 
 #ifndef __ASSEMBLY__
 
 #define instruction_pointer(regs) ((regs)->pc)
 #define profile_pc(regs) instruction_pointer(regs)
 
+static inline bool in_syscall(struct pt_regs const *regs)
+{
+       return ((regs->sr >> 16) & 0xff) == VEC_TRAP0;
+}
+
+static inline void forget_syscall(struct pt_regs *regs)
+{
+       regs->sr &= ~(0xff << 16);
+}
+
 static inline unsigned long regs_return_value(struct pt_regs *regs)
 {
        return regs->a0;
 
        mov     a0, sp
        movi    a1, \is_write
        jbsr    do_page_fault
-       movi    r11_sig, 0             /* r11 = 0, Not a syscall. */
        jmpi    ret_from_exception
 .endm
 
        bmaski  r10, THREAD_SHIFT
        andn    r9, r10
        ldw     r8, (r9, TINFO_FLAGS)
-       movi    r11_sig, 1
        ANDI_R3 r8, (_TIF_SYSCALL_TRACE | _TIF_SYSCALL_TRACEPOINT | _TIF_SYSCALL_AUDIT)
        cmpnei  r8, 0
        bf      3f
        /* If thread_info->flag is empty, RESTORE_ALL */
        cmpnei  r8, 0
        bf      1b
-       mov     a1, sp
        mov     a0, r8
-       mov     a2, r11_sig             /* syscall? */
-       btsti   r8, TIF_SIGPENDING      /* delivering a signal? */
-       /* prevent further restarts(set r11 = 0) */
-       clrt    r11_sig
+       mov     a1, sp
        jbsr    do_notify_resume        /* do signals */
        br      resume_userspace
 
        jmpi    schedule
 
 ENTRY(sys_rt_sigreturn)
-       movi    r11_sig, 0
        jmpi    do_rt_sigreturn
 
 ENTRY(csky_trap)
        SAVE_ALL EPC_KEEP
        psrset  ee
-       movi    r11_sig, 0             /* r11 = 0, Not a syscall. */
        mov     a0, sp                 /* Push Stack pointer arg */
        jbsr    trap_c                 /* Call C-level trap handler */
        jmpi    ret_from_exception
 ENTRY(csky_irq)
        SAVE_ALL EPC_KEEP
        psrset  ee
-       movi    r11_sig, 0              /* r11 = 0, Not a syscall. */
 
 #ifdef CONFIG_PREEMPT
        mov     r9, sp                  /* Get current stack  pointer */
 
  * that the kernel can handle, and then we build all the user-level signal
  * handling stack-frames in one go after that.
  */
-static void do_signal(struct pt_regs *regs, int syscall)
+static void do_signal(struct pt_regs *regs)
 {
        unsigned int retval = 0, continue_addr = 0, restart_addr = 0;
        struct ksignal ksig;
        /*
         * If we were from a system call, check for system call restarting...
         */
-       if (syscall) {
+       if (in_syscall(regs)) {
+               forget_syscall(regs);
+
                continue_addr = regs->pc;
 #if defined(__CSKYABIV2__)
                restart_addr = continue_addr - 4;
                restart_addr = continue_addr - 2;
 #endif
                retval = regs->a0;
-
                /*
                 * Prepare for system call restart.  We do this here so that a
                 * debugger will see the already changed.
        }
 
 no_signal:
-       if (syscall) {
+       if (in_syscall(regs)) {
+               forget_syscall(regs);
+
                /*
                 * Handle restarting a different system call.  As above,
                 * if a debugger has chosen to restart at a different PC,
 }
 
 asmlinkage void
-do_notify_resume(unsigned int thread_flags, struct pt_regs *regs, int syscall)
+do_notify_resume(unsigned int thread_flags, struct pt_regs *regs)
 {
        if (thread_flags & _TIF_SIGPENDING)
-               do_signal(regs, syscall);
+               do_signal(regs);
 
        if (thread_flags & _TIF_NOTIFY_RESUME) {
                clear_thread_flag(TIF_NOTIFY_RESUME);