Enable sched_trace log injection from userspacewip-gpu-cleanup

1 files changed, 73 insertions, 0 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 1b4b9d25dbdc..6a1095aa7725 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -310,6 +310,79 @@ asmlinkage long sys_null_call(cycles_t __user *ts)
         return ret;
 }
 
+
+asmlinkage long sys_sched_trace_event(int event, struct st_inject_args __user *__args)
+{
+        long retval = 0;
+        struct task_struct* t = current;
+
+        struct st_inject_args args;
+
+        if (is_realtime(t)) {
+                printk(KERN_WARNING "Only non-real-time tasks may inject sched_trace events.\n");
+                retval = -EINVAL;
+                goto out;
+        }
+
+        if (__args && copy_from_user(&args, __args, sizeof(args))) {
+                retval = -EFAULT;
+                goto out;
+        }
+
+        switch(event) {
+                /*************************************/
+                /* events that don't need parameters */
+                /*************************************/
+                case ST_INJECT_NAME:
+                        sched_trace_task_name(t);
+                        break;
+                case ST_INJECT_PARAM:
+                        /* presumes sporadic_task_ns() has already been called
+                         * and valid data has been initialized even if the calling
+                         * task is SCHED_NORMAL. */
+                        sched_trace_task_param(t);
+                        break;
+
+                /*******************************/
+                /* events that need parameters */
+                /*******************************/
+                case ST_INJECT_COMPLETION:
+                        if (!__args) {
+                                retval = -EINVAL;
+                                goto out;
+                        }
+
+                        /* slam in the data */
+                        t->rt_param.job_params.job_no = args.job_no;
+
+                        sched_trace_task_completion(t, 0);
+                        break;
+                case ST_INJECT_RELEASE:
+                        if (!__args) {
+                                retval = -EINVAL;
+                                goto out;
+                        }
+
+                        /* slam in the data */
+                        tsk_rt(t)->job_params.release = args.release;
+                        tsk_rt(t)->job_params.deadline = args.deadline;
+
+                        sched_trace_task_release(t);
+                        break;
+
+                /**********************/
+                /* unsupported events */
+                /**********************/
+                default:
+                        retval = -EINVAL;
+                        break;
+        }
+
+out:
+        return retval;
+}
+
+
 #if defined(CONFIG_LITMUS_NVIDIA) && defined(CONFIG_LITMUS_AFFINITY_LOCKING)
 void init_gpu_affinity_state(struct task_struct* p)
 {