Implement OMLP-Family budget exhaustion hooks.

1 files changed, 11 insertions, 0 deletions

diff --git a/litmus/locking.c b/litmus/locking.c
index d347fdce4cb7..88905d4cff26 100644
--- a/litmus/locking.c
+++ b/litmus/locking.c
@@ -125,6 +125,11 @@ asmlinkage long sys_litmus_lock(int lock_od)
                         sched_trace_lock(current, l->ident, 1);
 
                         TRACE_CUR("Got lock %d\n", l->ident);
+
+                        if (tsk_rt(current)->outermost_lock == NULL) {
+                                TRACE_CUR("Lock %d is outermost lock.\n", l->ident);
+                                tsk_rt(current)->outermost_lock = l;
+                        }
                 }
         }
 
@@ -158,6 +163,12 @@ asmlinkage long sys_litmus_unlock(int lock_od)
                         sched_trace_lock(current, l->ident, 0);
 
                         TRACE_CUR("Unlocked %d\n", l->ident);
+
+                        if (tsk_rt(current)->outermost_lock == l) {
+                                TRACE_CUR("Lock %d assumed to be outermost lock.\n", l->ident);
+                                tsk_rt(current)->outermost_lock = NULL;
+                                WARN_ON(holds_locks(current));
+                        }
                 }
         }