sched_trace: add task system release tracing

3 files changed, 16 insertions, 1 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 3d903c526d..51c98f3ea0 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -556,12 +556,12 @@ long litmus_admit_task(struct task_struct* tsk)
         spin_unlock_irqrestore(&task_transition_lock, flags);
 
         return retval;
-
 }
 
 void litmus_exit_task(struct task_struct* tsk)
 {
         if (is_realtime(tsk)) {
+                sched_trace_task_completion(tsk, 1);
                 litmus->task_exit(tsk);
                 BUG_ON(heap_node_in_heap(tsk_rt(tsk)->heap_node));
                 kmem_cache_free(heap_node_cache, tsk_rt(tsk)->heap_node);
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index fa71ef7bd0..32d1445967 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -179,3 +179,14 @@ feather_callback void do_sched_trace_task_resume(unsigned long id, unsigned long
                 put_record(rec);
         }
 }
+
+feather_callback void do_sched_trace_sys_release(unsigned long id, unsigned long _start)
+{
+        lt_t *start = (lt_t*) _start;
+        struct st_event_record* rec = get_record(ST_SYS_RELEASE, NULL);
+        if (rec) {
+                rec->data.sys_release.when    = now();
+                rec->data.sys_release.release = *start;
+                put_record(rec);
+        }
+}
diff --git a/litmus/sync.c b/litmus/sync.c
index c16f1dd2ab..8b0ebea49a 100644
--- a/litmus/sync.c
+++ b/litmus/sync.c
@@ -14,6 +14,8 @@
 #include <litmus/sched_plugin.h>
 #include <litmus/jobs.h>
 
+#include <litmus/sched_trace.h>
+
 static DECLARE_COMPLETION(ts_release);
 
 static long do_wait_for_ts_release(void)
@@ -43,12 +45,14 @@ static long do_release_ts(lt_t start)
         spin_lock_irqsave(&ts_release.wait.lock, flags);
         TRACE("<<<<<< synchronous task system release >>>>>>\n");
         
+        sched_trace_sys_release(&start);
         list_for_each(pos, &ts_release.wait.task_list) {
                 t = (struct task_struct*) list_entry(pos,
                                                      struct __wait_queue,
                                                      task_list)->private;
                 task_count++;
                 litmus->release_at(t, start + t->rt_param.task_params.phase);
+                sched_trace_task_release(t);
         }
 
         spin_unlock_irqrestore(&ts_release.wait.lock, flags);