1 files changed, 5 insertions, 4 deletions

diff --git a/include/trace/events/sched.h b/include/trace/events/sched.h
index f6334782a593..959ff18b63b6 100644
--- a/include/trace/events/sched.h
+++ b/include/trace/events/sched.h
@@ -100,7 +100,7 @@ static inline long __trace_sched_switch_state(struct task_struct *p)
          * For all intents and purposes a preempted task is a running task.
          */
         if (task_thread_info(p)->preempt_count & PREEMPT_ACTIVE)
-                state = TASK_RUNNING;
+                state = TASK_RUNNING | TASK_STATE_MAX;
 #endif
 
         return state;
@@ -137,13 +137,14 @@ TRACE_EVENT(sched_switch,
                 __entry->next_prio      = next->prio;
         ),
 
-        TP_printk("prev_comm=%s prev_pid=%d prev_prio=%d prev_state=%s ==> next_comm=%s next_pid=%d next_prio=%d",
+        TP_printk("prev_comm=%s prev_pid=%d prev_prio=%d prev_state=%s%s ==> next_comm=%s next_pid=%d next_prio=%d",
                 __entry->prev_comm, __entry->prev_pid, __entry->prev_prio,
-                __entry->prev_state ?
-                  __print_flags(__entry->prev_state, "|",
+                __entry->prev_state & (TASK_STATE_MAX-1) ?
+                  __print_flags(__entry->prev_state & (TASK_STATE_MAX-1), "|",
                                 { 1, "S"} , { 2, "D" }, { 4, "T" }, { 8, "t" },
                                 { 16, "Z" }, { 32, "X" }, { 64, "x" },
                                 { 128, "W" }) : "R",
+                __entry->prev_state & TASK_STATE_MAX ? "+" : "",
                 __entry->next_comm, __entry->next_pid, __entry->next_prio)
 );