Add task_exit event that records max exec time under color plugin.

2 files changed, 26 insertions, 0 deletions

diff --git a/litmus/sched_color.c b/litmus/sched_color.c
index 29516e948749..0448983e13ce 100644
--- a/litmus/sched_color.c
+++ b/litmus/sched_color.c
@@ -205,6 +205,7 @@ static void link(struct rt_server *server, struct task_struct *task)
 static void job_completion(struct rt_server *server)
 {
         struct task_struct *t = server->linked;
+        lt_t et;
 
         TRACE_TASK(t, "Job completed\n");
         if (is_server(t))
@@ -212,6 +213,13 @@ static void job_completion(struct rt_server *server)
         else
                 sched_trace_task_completion(t, 0);
 
+        if (1 < get_rt_job(t)) {
+                /* our releases happen at the second job */
+                et = get_exec_time(t);
+                if (et > tsk_rt(t)->max_exec_time)
+                        tsk_rt(t)->max_exec_time = et;
+        }
+
         unlink(server);
         set_rt_flags(t, RT_F_SLEEP);
         prepare_for_next_period(t);
@@ -525,6 +533,8 @@ static void color_task_new(struct task_struct *t, int on_rq, int running)
         dgl_group_req_init(&group_lock, req);
         tsk_rt(t)->req = req;
 
+        tsk_rt(t)->max_exec_time = 0;
+
         /* Fill request */
         if (tsk_rt(t)->color_ctrl_page) {
                 colors = tsk_rt(t)->color_ctrl_page->colors;
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 5ef8d09ab41f..636ea038f403 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -226,6 +226,22 @@ feather_callback void do_sched_trace_sys_release(unsigned long id,
         }
 }
 
+feather_callback void do_sched_trace_task_exit(unsigned long id,
+                                                 unsigned long _task)
+{
+        struct task_struct *t = (struct task_struct*) _task;
+#ifdef CONFIG_PLUGIN_COLOR
+        const lt_t max_exec_time = tsk_rt(t)->max_exec_time;
+#else
+        const lt_t max_exec_time = 0;
+#endif
+        struct st_event_record *rec = get_record(ST_TASK_EXIT, t);
+        if (rec) {
+                rec->data.task_exit.max_exec_time = max_exec_time;
+                put_record(rec);
+        }
+}
+
 feather_callback void do_sched_trace_action(unsigned long id,
                                             unsigned long _task,
                                             unsigned long action)