1 files changed, 32 insertions, 13 deletions

diff --git a/litmus/budget.c b/litmus/budget.c
index 75f4b5156947..7d94f7e61b27 100644
--- a/litmus/budget.c
+++ b/litmus/budget.c
@@ -5,6 +5,7 @@
 #include <litmus/litmus.h>
 #include <litmus/preempt.h>
 #include <litmus/budget.h>
+#include <litmus/sched_trace.h>
 
 struct enforcement_timer {
         /* The enforcement timer is used to accurately police
@@ -63,7 +64,7 @@ static void arm_enforcement_timer(struct enforcement_timer* et,
 
         /* Calling this when there is no budget left for the task
          * makes no sense, unless the task is non-preemptive. */
-        BUG_ON(budget_exhausted(t) && (!is_np(t)));
+        /* BUG_ON(budget_exhausted(t) && (!is_np(t))); */
 
         /* __hrtimer_start_range_ns() cancels the timer
          * anyway, so we don't have to check whether it is still armed */
@@ -109,20 +110,38 @@ static int __init init_budget_enforcement(void)
         return 0;
 }
 
+void task_release(struct task_struct *t)
+{
+        t->rt_param.job_params.real_release = t->rt_param.job_params.deadline;
+        t->rt_param.job_params.real_deadline += get_rt_period(t);
+        t->rt_param.job_params.job_no++;
+        TRACE_TASK(t, "Releasing task, rr=%llu rd=%llu\n",
+                   t->rt_param.job_params.real_release,
+                   t->rt_param.job_params.real_deadline);
+        sched_trace_task_release(t);
+}
+
+void server_release(struct task_struct *t)
+{
+        lt_t now = litmus_clock();
+        t->rt_param.job_params.exec_time = 0;
+        t->rt_param.job_params.release = t->rt_param.job_params.deadline;
+        t->rt_param.job_params.deadline +=  get_rt_period(t);
+        TRACE_TASK(t, "Releasing server, r=%llu d=%llu\n",
+                   t->rt_param.job_params.release,
+                   t->rt_param.job_params.deadline);
+        /* don't confuse linux */
+        t->rt.time_slice = 1;
+}
+
 void prepare_for_next_server(struct task_struct *t, int forced)
 {
-        if (!job_behind(t)) {
-                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;
-                /* update job sequence number */
-                t->rt_param.job_params.job_no++;
-        } else if (forced) {
-                t->rt_param.job_params.release = t->rt_param.job_params.deadline;
-                t->rt_param.job_params.exec_time = 0;
-        } else /* behind */{
-                t->rt_param.job_params.deadline += get_rt_period(t);
-                t->rt_param.job_params.job_no++;
+        if (forced || job_behind(t)) {
+                server_release(t);
+        }
+
+        if (!forced) {
+                task_release(t);
         }
 }