Can disable nonpreemptivity

1 files changed, 16 insertions, 2 deletions

diff --git a/include/litmus/litmus.h b/include/litmus/litmus.h
index f0ddb89e68dd..9fcf1f45820f 100644
--- a/include/litmus/litmus.h
+++ b/include/litmus/litmus.h
@@ -169,7 +169,7 @@ static inline void request_exit_np(struct task_struct *t)
 
 static inline void make_np(struct task_struct *t)
 {
-        tsk_rt(t)->kernel_np++;
+        tsk_rt(t)->kernel_np = 1;
 }
 
 /* Caller should check if preemption is necessary when
@@ -177,7 +177,7 @@ static inline void make_np(struct task_struct *t)
  */
 static inline int take_np(struct task_struct *t)
 {
-        return --tsk_rt(t)->kernel_np;
+        return tsk_rt(t)->kernel_np = 0;
 }
 
 /* returns 0 if remote CPU needs an IPI to preempt, 1 if no IPI is required */
@@ -210,6 +210,20 @@ static inline int request_exit_np_atomic(struct task_struct *t)
 
 #else
 
+
+static inline void make_np(struct task_struct *t)
+{
+
+}
+
+/* Caller should check if preemption is necessary when
+ * the function return 0.
+ */
+static inline int take_np(struct task_struct *t)
+{
+        return 0;
+}
+
 static inline int is_kernel_np(struct task_struct* t)
 {
         return 0;