LITMUS^RT: improve integration with fork()

1 files changed, 8 insertions, 1 deletions

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 13e6056cffbe..cec0e32b5c02 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -1941,6 +1941,8 @@ int sched_fork(unsigned long clone_flags, struct task_struct *p)
          */
         p->prio = current->normal_prio;
 
+        litmus_fork(p);
+
         /*
          * Revert to default priority/policy on fork if requested.
          */
@@ -1963,7 +1965,9 @@ int sched_fork(unsigned long clone_flags, struct task_struct *p)
                 p->sched_reset_on_fork = 0;
         }
 
-        if (dl_prio(p->prio)) {
+        if (is_realtime(p)) {
+                p->sched_class = &litmus_sched_class;
+        } else if (dl_prio(p->prio)) {
                 put_cpu();
                 return -EAGAIN;
         } else if (rt_prio(p->prio)) {
@@ -2114,6 +2118,9 @@ void wake_up_new_task(struct task_struct *p)
         unsigned long flags;
         struct rq *rq;
 
+        if (is_realtime(p))
+                litmus->task_new(p, 1, 0);
+
         raw_spin_lock_irqsave(&p->pi_lock, flags);
 #ifdef CONFIG_SMP
         /*