Track tasks average execution time, drop in TASK_EXIT record

1 files changed, 8 insertions, 5 deletions

diff --git a/litmus/jobs.c b/litmus/jobs.c
index 36e314625d86..4da53a8c4c43 100644
--- a/litmus/jobs.c
+++ b/litmus/jobs.c
@@ -9,12 +9,15 @@
 void prepare_for_next_period(struct task_struct *t)
 {
         BUG_ON(!t);
+#ifdef CONFIG_PLUGIN_COLOR
+        tsk_rt(t)->tot_exec_time += tsk_rt(t)->job_params.exec_time;
+#endif
         /* prepare next release */
-        t->rt_param.job_params.release   = t->rt_param.job_params.deadline;
-        t->rt_param.job_params.deadline += get_rt_period(t);
-        t->rt_param.job_params.exec_time = 0;
+        tsk_rt(t)->job_params.release   = tsk_rt(t)->job_params.deadline;
+        tsk_rt(t)->job_params.deadline += get_rt_period(t);
+        tsk_rt(t)->job_params.exec_time = 0;
         /* update job sequence number */
-        t->rt_param.job_params.job_no++;
+        tsk_rt(t)->job_params.job_no++;
 
         /* don't confuse Linux */
         t->rt.time_slice = 1;
@@ -22,7 +25,7 @@ void prepare_for_next_period(struct task_struct *t)
 
 void release_at(struct task_struct *t, lt_t start)
 {
-        t->rt_param.job_params.deadline = start;
+        tsk_rt(t)->job_params.deadline = start;
         prepare_for_next_period(t);
         set_rt_flags(t, RT_F_RUNNING);
 }