Hook into fork(), exec(), and exit()

Allow LITMUS^RT to do some work when a process is created or
terminated.

3 files changed, 18 insertions, 1 deletions

diff --git a/kernel/exit.c b/kernel/exit.c
index 3076f3089919..89ea9c8a7305 100644
--- a/kernel/exit.c
+++ b/kernel/exit.c
@@ -60,6 +60,8 @@
 #include <asm/pgtable.h>
 #include <asm/mmu_context.h>
 
+extern void exit_od_table(struct task_struct *t);
+
 static void __unhash_process(struct task_struct *p, bool group_dead)
 {
         nr_threads--;
@@ -809,6 +811,8 @@ void __noreturn do_exit(long code)
                 tty_audit_exit();
         audit_free(tsk);
 
+        exit_od_table(tsk);
+
         tsk->exit_code = code;
         taskstats_exit(tsk, group_dead);
 
diff --git a/kernel/fork.c b/kernel/fork.c
index 59faac4de181..a8b40275b136 100644
--- a/kernel/fork.c
+++ b/kernel/fork.c
@@ -86,6 +86,9 @@
 
 #include <trace/events/sched.h>
 
+#include <litmus/litmus.h>
+#include <litmus/sched_plugin.h>
+
 #define CREATE_TRACE_POINTS
 #include <trace/events/task.h>
 
@@ -386,6 +389,9 @@ void __put_task_struct(struct task_struct *tsk)
         cgroup_free(tsk);
         task_numa_free(tsk);
         security_task_free(tsk);
+
+        exit_litmus(tsk);
+
         exit_creds(tsk);
         delayacct_tsk_free(tsk);
         put_signal_struct(tsk->signal);
diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index abdf10c424f0..d1a8feda6918 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -2415,6 +2415,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.
          */
@@ -2436,7 +2438,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)) {
@@ -2593,6 +2597,9 @@ void wake_up_new_task(struct task_struct *p)
         struct rq_flags rf;
         struct rq *rq;
 
+        if (is_realtime(p))
+                litmus->task_new(p, 1, 0);
+
         raw_spin_lock_irqsave(&p->pi_lock, rf.flags);
         p->state = TASK_RUNNING;
 #ifdef CONFIG_SMP