1 files changed, 12 insertions, 0 deletions

diff --git a/include/litmus/budget.h b/include/litmus/budget.h
index ff18d89e8630..265f2b1e62b8 100644
--- a/include/litmus/budget.h
+++ b/include/litmus/budget.h
@@ -1,6 +1,11 @@
 #ifndef _LITMUS_BUDGET_H_
 #define _LITMUS_BUDGET_H_
 
+struct enforcement_timer {
+        struct hrtimer          timer;
+        int                     armed;
+};
+
 /**
  * update_enforcement_timer() - Update per-processor enforcement timer for
  * the next scheduled task.
@@ -11,6 +16,12 @@
 */
 void update_enforcement_timer(struct task_struct* t);
 
+void init_enforcement_timer(struct enforcement_timer *et);
+
+void arm_enforcement_timer(struct enforcement_timer* et, struct task_struct* t);
+
+void cancel_enforcement_timer(struct enforcement_timer* et);
+
 /* True if a task's server has progressed farther than the task
  * itself. This happens when budget enforcement has caused a task to be
  * booted off until the next period.
@@ -32,4 +43,5 @@ void server_release(struct task_struct *t);
  * so that we can write them to feather trace.
  */
 void task_release(struct task_struct *t);
+
 #endif