Fix problem with tasks exhausting budget while unlinked

1 files changed, 6 insertions, 1 deletions

diff --git a/litmus/budget.c b/litmus/budget.c
index 310e9a3d4172..c3d0cb6c30a4 100644
--- a/litmus/budget.c
+++ b/litmus/budget.c
@@ -87,7 +87,12 @@ void update_enforcement_timer(struct task_struct* t)
         if (t && budget_precisely_enforced(t)) {
                 /* Make sure we call into the scheduler when this budget
                  * expires. */
-                arm_enforcement_timer(et, t);
+                if (budget_exhausted(t)) {
+                        on_enforcement_timeout(&et->timer);
+                }
+                else{
+                        arm_enforcement_timer(et, t);
+                }
         } else if (et->armed) {
                 /* Make sure we don't cause unnecessary interrupts. */
                 cancel_enforcement_timer(et);