sched_trace: add sched_trace_last_suspension_as_completion()

New tracepoint for tracing the completion of sporadic jobs.

2 files changed, 24 insertions, 0 deletions

diff --git a/include/litmus/sched_trace.h b/include/litmus/sched_trace.h
index b4da7e39fa79..0532424fbee6 100644
--- a/include/litmus/sched_trace.h
+++ b/include/litmus/sched_trace.h
@@ -144,6 +144,9 @@ feather_callback void do_sched_trace_task_switch_away(unsigned long id,
 feather_callback void do_sched_trace_task_completion(unsigned long id,
                                                      struct task_struct* task,
                                                      unsigned long forced);
+feather_callback void do_sched_trace_last_suspension_as_completion(
+                                                unsigned long id,
+                                                struct task_struct* task);
 feather_callback void do_sched_trace_task_block(unsigned long id,
                                                 struct task_struct* task);
 feather_callback void do_sched_trace_task_resume(unsigned long id,
@@ -251,6 +254,12 @@ feather_callback void do_sched_trace_sys_release(unsigned long id,
                 trace_litmus_sys_release(when);                         \
         } while (0)
 
+#define sched_trace_last_suspension_as_completion(t)    \
+        do {                                                            \
+                SCHED_TRACE(SCHED_TRACE_BASE_ID + 11,                   \
+                                do_sched_trace_last_suspension_as_completion, t); \
+        } while (0)
+
 #define sched_trace_quantum_boundary() /* NOT IMPLEMENTED */
 
 #endif /* __KERNEL__ */
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index f623dc7a25eb..a6088f16bb08 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -194,6 +194,21 @@ feather_callback void do_sched_trace_task_completion(unsigned long id,
         }
 }
 
+feather_callback void do_sched_trace_last_suspension_as_completion(
+        unsigned long id,
+        unsigned long _task)
+{
+        struct task_struct *t = (struct task_struct*) _task;
+        struct st_event_record* rec = get_record(ST_COMPLETION, t);
+        if (rec) {
+                rec->data.completion.when
+                        = tsk_rt(t)->job_params.last_suspension;
+                rec->data.completion.forced = 0;
+                rec->data.completion.exec_time = get_exec_time(t);
+                put_record(rec);
+        }
+}
+
 feather_callback void do_sched_trace_task_block(unsigned long id,
                                                 unsigned long _task)
 {