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

5 files changed, 70 insertions, 1 deletions

diff --git a/include/litmus/rt_param.h b/include/litmus/rt_param.h
index ec35c220abe4..2a4b99be636b 100644
--- a/include/litmus/rt_param.h
+++ b/include/litmus/rt_param.h
@@ -226,8 +226,11 @@ struct rt_param {
         /* Pointer to the page shared between userspace and kernel. */
         struct control_page * ctrl_page;
 
+#ifdef CONFIG_PLUGIN_COLOR
+        lt_t max_exec_time;
         struct color_ctrl_page *color_ctrl_page;
         struct dgl_group_req *req;
+#endif
 };
 
 /*      Possible RT flags       */
diff --git a/include/litmus/sched_trace.h b/include/litmus/sched_trace.h
index 0773aebcae16..8f61f3fa486b 100644
--- a/include/litmus/sched_trace.h
+++ b/include/litmus/sched_trace.h
@@ -81,6 +81,11 @@ struct st_sys_release_data {
         u64     release;
 };
 
+struct st_task_exit_data {
+        u64     max_exec_time;
+        u64     __unused;
+};
+
 #define DATA(x) struct st_ ## x ## _data x;
 
 typedef enum {
@@ -95,7 +100,8 @@ typedef enum {
         ST_BLOCK,
         ST_RESUME,
         ST_ACTION,
-        ST_SYS_RELEASE
+        ST_SYS_RELEASE,
+        ST_TASK_EXIT,
 } st_event_record_type_t;
 
 struct st_event_record {
@@ -114,6 +120,7 @@ struct st_event_record {
                 DATA(resume);
                 DATA(action);
                 DATA(sys_release);
+                DATA(task_exit);
         } data;
 };
 
@@ -155,6 +162,8 @@ feather_callback void do_sched_trace_action(unsigned long id,
                                             unsigned long action);
 feather_callback void do_sched_trace_sys_release(unsigned long id,
                                                  lt_t* start);
+feather_callback void do_sched_trace_task_exit(unsigned long id,
+                                                     struct task_struct* task);
 
 #endif
 
@@ -180,6 +189,7 @@ feather_callback void do_sched_trace_sys_release(unsigned long id,
 #define trace_litmus_task_block(t, i)
 #define trace_litmus_task_resume(t, i)
 #define trace_litmus_sys_release(start)
+#define trace_litmus_task_exit(t)
 
 #define trace_litmus_resource_acquire(t, i);
 #define trace_litmus_resource_release(t, i);
@@ -264,6 +274,13 @@ feather_callback void do_sched_trace_sys_release(unsigned long id,
                 trace_litmus_sys_release(when);                         \
         } while (0)
 
+#define sched_trace_task_exit(t)                                        \
+        do {                                                            \
+                SCHED_TRACE(SCHED_TRACE_BASE_ID + 11,                   \
+                        do_sched_trace_task_exit, t);                   \
+                trace_litmus_task_exit(t);                              \
+        } while (0)
+
 #define QT_START lt_t _qt_start = litmus_clock()
 #define QT_END \
         sched_trace_log_message("%d P%d      [%s@%s:%d]: Took %llu\n\n", \
diff --git a/include/trace/events/litmus.h b/include/trace/events/litmus.h
index 136a80db54a4..c767f4ffef69 100644
--- a/include/trace/events/litmus.h
+++ b/include/trace/events/litmus.h
@@ -284,6 +284,29 @@ TRACE_EVENT(litmus_sys_release,
 );
 
 /*
+ * Trace task exit
+ */
+TRACE_EVENT(litmus_task_exit,
+
+        TP_PROTO(struct task_struct *t),
+
+        TP_ARGS(t),
+
+        TP_STRUCT__entry(
+                __field( pid_t, pid )
+                __field( unsigned long long, max_exec_time )
+        ),
+
+        TP_fast_assign(
+                __entry->pid = t ? t->pid : 0;
+#ifdef CONFIG_PLUGIN_COLOR
+                __entry->max_exec_time = t ? t->rt_param.max_exec_time : 0;
+        ),
+
+        TP_printk("(%u) exit\n", __entry->pid)
+);
+
+/*
  * Containers
  */
 TRACE_EVENT(litmus_container_param,
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)