Added exit and tardy hooks.

5 files changed, 10 insertions, 7 deletions

diff --git a/include/litmus/litmus.h b/include/litmus/litmus.h
index 9282f3a8f28f..6a1a59da6b52 100644
--- a/include/litmus/litmus.h
+++ b/include/litmus/litmus.h
@@ -60,7 +60,8 @@ void litmus_exit_task(struct task_struct *tsk);
 #define get_release_policy(t) (tsk_rt(t)->task_params.release_policy)
 
 /* job_param macros */
-#define get_exec_time(t)    (tsk_rt(t)->job_params.exec_time)
+#define get_job_no(t)           (tsk_rt(t)->job_params.job_no)
+#define get_exec_time(t)        (tsk_rt(t)->job_params.exec_time)
 #define get_deadline(t)         (tsk_rt(t)->job_params.deadline)
 #define get_release(t)          (tsk_rt(t)->job_params.release)
 #define get_lateness(t)         (tsk_rt(t)->job_params.lateness)
diff --git a/include/litmus/sched_trace.h b/include/litmus/sched_trace.h
index bf6c4cf4316c..2f992789affb 100644
--- a/include/litmus/sched_trace.h
+++ b/include/litmus/sched_trace.h
@@ -308,7 +308,6 @@ feather_callback void do_sched_trace_task_tardy(unsigned long id,
         do {                                                            \
                 SCHED_TRACE(SCHED_TRACE_BASE_ID + 12,                   \
                         do_sched_trace_task_tardy, t);                  \
-                trace_litmus_task_tardy(t);                             \
         } while (0)
 
 #define QT_START lt_t _qt_start = litmus_clock()
diff --git a/litmus/litmus.c b/litmus/litmus.c
index dc94be71bfb6..4bbac9786243 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -384,6 +384,9 @@ void litmus_exit_task(struct task_struct* tsk)
 {
         if (is_realtime(tsk)) {
                 sched_trace_task_completion(tsk, 1);
+                sched_trace_task_exit(tsk);
+                sched_trace_task_tardy(tsk);
+                printk(KERN_WARNING "Swear I'm doing \n");
 
                 litmus->task_exit(tsk);
 
diff --git a/litmus/preempt.c b/litmus/preempt.c
index 6be2f26728b8..a202d70c627e 100644
--- a/litmus/preempt.c
+++ b/litmus/preempt.c
@@ -27,10 +27,10 @@ void sched_state_will_schedule(struct task_struct* tsk)
                         set_sched_state(PICKED_WRONG_TASK);
                 else
                         set_sched_state(WILL_SCHEDULE);
-        } else
-                /* Litmus tasks should never be subject to a remote
-                 * set_tsk_need_resched(). */
-                BUG_ON(is_realtime(tsk));
+        } /* else */
+          /*    /\* Litmus tasks should never be subject to a remote */
+          /*     * set_tsk_need_resched(). *\/ */
+          /*    BUG_ON(is_realtime(tsk)); */
 #ifdef CONFIG_PREEMPT_STATE_TRACE
         TRACE_TASK(tsk, "set_tsk_need_resched() ret:%p\n",
                    __builtin_return_address(0));
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 7578acd666f3..3c42dfedac18 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -234,7 +234,7 @@ feather_callback void do_sched_trace_task_exit(unsigned long id,
 {
         struct task_struct *t = (struct task_struct*) _task;
         const lt_t max_exec_time = tsk_rt(t)->max_exec_time;
-        const lt_t avg_exec_time = tsk_rt(t)->tot_exec_time / (get_rt_job(t) - 1);
+        const lt_t avg_exec_time = tsk_rt(t)->tot_exec_time / (get_job_no(t) - 1);
 
         struct st_event_record *rec = get_record(ST_TASK_EXIT, t);
         if (rec) {