Litmus core: add budget_remaining() helper

Returns the amount of budget left for a given job.

1 files changed, 8 insertions, 0 deletions

diff --git a/include/litmus/litmus.h b/include/litmus/litmus.h
index 674528eece14..ba6c5381e151 100644
--- a/include/litmus/litmus.h
+++ b/include/litmus/litmus.h
@@ -85,6 +85,14 @@ inline static int budget_exhausted(struct task_struct* t)
         return get_exec_time(t) >= get_exec_cost(t);
 }
 
+inline static lt_t budget_remaining(struct task_struct* t)
+{
+        /* This overflows if called when there is no budget remaining. */
+        BUG_ON(budget_exhausted(t));
+
+        return get_exec_cost(t) - get_exec_time(t);
+}
+
 #define budget_enforced(t) (tsk_rt(t)->task_params.budget_policy != NO_ENFORCEMENT)