1 files changed, 7 insertions, 1 deletions

diff --git a/litmus/locking.c b/litmus/locking.c
index 0c1aa6aa40b7..4881ca119acf 100644
--- a/litmus/locking.c
+++ b/litmus/locking.c
@@ -1,3 +1,5 @@
+#include <linux/sched.h>
+#include <litmus/litmus.h>
 #include <litmus/fdso.h>
 
 #ifdef CONFIG_LITMUS_LOCKING
@@ -28,14 +30,18 @@ static inline struct litmus_lock* get_lock(struct od_table_entry* entry)
         return (struct litmus_lock*) entry->obj->obj;
 }
 
+atomic_t lock_id = ATOMIC_INIT(0);
+
 static  int create_generic_lock(void** obj_ref, obj_type_t type, void* __user arg)
 {
         struct litmus_lock* lock;
         int err;
 
         err = litmus->allocate_lock(&lock, type, arg);
-        if (err == 0)
+        if (err == 0) {
+                lock->id = atomic_add_return(1, &lock_id);
                 *obj_ref = lock;
+        }
         return err;
 }