1 files changed, 13 insertions, 2 deletions

diff --git a/include/litmus/locking.h b/include/litmus/locking.h
index 962ad5e6726a..08b06200b955 100644
--- a/include/litmus/locking.h
+++ b/include/litmus/locking.h
@@ -120,6 +120,12 @@ typedef lock_op_t lock_unlock_t;
 typedef int (*lock_open_t)(struct litmus_lock *l, void* __user arg);
 typedef void (*lock_free_t)(struct litmus_lock *l);
 
+#ifdef CONFIG_LITMUS_NESTED_LOCKING
+/* Assumes task's base-priority already updated to reflect new priority. */
+typedef void (*lock_budget_exhausted_t)(struct litmus_lock* l, struct task_struct* t);
+typedef void (*lock_omlp_virtual_unlock_t)(struct litmus_lock* l, struct task_struct* t);
+#endif
+
 struct litmus_lock_ops {
         /* Current task tries to obtain / drop a reference to a lock.
          * Optional methods, allowed by default. */
@@ -133,10 +139,12 @@ struct litmus_lock_ops {
         /* The lock is no longer being referenced (mandatory method). */
         lock_free_t deallocate;
 
-
 #ifdef CONFIG_LITMUS_NESTED_LOCKING
+        lock_budget_exhausted_t budget_exhausted;
+        lock_omlp_virtual_unlock_t omlp_virtual_unlock;
+
         void (*propagate_increase_inheritance)(struct litmus_lock* l, struct task_struct* t, raw_spinlock_t* to_unlock, unsigned long irqflags);
-        void (*propagate_decrease_inheritance)(struct litmus_lock* l, struct task_struct* t, raw_spinlock_t* to_unlock, unsigned long irqflags);
+        void (*propagate_decrease_inheritance)(struct litmus_lock* l, struct task_struct* t, raw_spinlock_t* to_unlock, unsigned long irqflags, int budget_triggered);
 #endif
 #ifdef CONFIG_LITMUS_DGL_SUPPORT
         raw_spinlock_t* (*get_dgl_spin_lock)(struct litmus_lock *l);
@@ -151,6 +159,9 @@ struct litmus_lock_ops {
 #endif
 
         /* all flags at the end */
+        unsigned int supports_budget_exhaustion:1;
+        unsigned int is_omlp_family:1;
+
 #ifdef CONFIG_LITMUS_NESTED_LOCKING
         unsigned int supports_nesting:1;
 #endif