work in progress.

1 files changed, 7 insertions, 0 deletions

diff --git a/include/litmus/budget.h b/include/litmus/budget.h
index 72f04777e0b0..4f1bdd101a9e 100644
--- a/include/litmus/budget.h
+++ b/include/litmus/budget.h
@@ -58,6 +58,8 @@ typedef void (*preempt_t)(struct task_struct* t);
 typedef void (*sleep_t)(struct task_struct* t);
 typedef enum hrtimer_restart (*exhausted_t)(struct task_struct* t);
 typedef void (*exit_t)(struct task_struct* t);
+typedef void (*inherit_t)(struct task_struct* t, struct task_struct* prio_inh);
+typedef void (*disinherit_t)(struct task_struct* t, struct task_struct* prio_inh);
 
 struct budget_tracker_ops
 {
@@ -69,6 +71,9 @@ struct budget_tracker_ops
         exit_t                          on_exit;                /* task exiting rt mode */
 
         exhausted_t                     on_exhausted;   /* called by plugin::tick() or timer interrupt */
+
+        inherit_t                       on_inherit;
+        disinherit_t            on_disinherit;
 };
 
 struct budget_tracker
@@ -108,6 +113,8 @@ void sobliv_on_sleep(struct task_struct* t);
 /* Use the DRAIN_SIMPLE implementations */
 #define sobliv_on_preempt       simple_on_preempt
 #define sobliv_on_exit  simple_on_exit
+void sobliv_on_inherit(struct task_struct* t, struct task_struct* prio_inh);
+void sobliv_on_disinherit(struct task_struct* t, struct task_struct* prio_inh);
 
 
 void init_budget_tracker(struct budget_tracker* bt,