1 files changed, 9 insertions, 0 deletions

diff --git a/kernel/fork.c b/kernel/fork.c
index 03c1eaaa6ef5..16b6ca38856e 100644
--- a/kernel/fork.c
+++ b/kernel/fork.c
@@ -85,6 +85,9 @@
 
 #include <trace/events/sched.h>
 
+#include <litmus/litmus.h>
+#include <litmus/sched_plugin.h>
+
 #define CREATE_TRACE_POINTS
 #include <trace/events/task.h>
 
@@ -253,6 +256,9 @@ void __put_task_struct(struct task_struct *tsk)
 
         task_numa_free(tsk);
         security_task_free(tsk);
+
+        exit_litmus(tsk);
+
         exit_creds(tsk);
         delayacct_tsk_free(tsk);
         put_signal_struct(tsk->signal);
@@ -355,6 +361,9 @@ static struct task_struct *dup_task_struct(struct task_struct *orig)
         tsk->seccomp.filter = NULL;
 #endif
 
+        /* Don't let the new task be a real-time task. */
+        litmus_fork(tsk);
+
         setup_thread_stack(tsk, orig);
         clear_user_return_notifier(tsk);
         clear_tsk_need_resched(tsk);