1 files changed, 14 insertions, 1 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 2f780222d8e8..debfc45bb8e6 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -75,6 +75,7 @@ asmlinkage long sys_set_rt_task_param(pid_t pid, struct rt_task __user * param)
 {
         struct rt_task tp;
         struct task_struct *target;
+        int * lookup_table;
         int retval = -EINVAL;
 
         printk("Setting up rt task parameters for process %d.\n", pid);
@@ -124,7 +125,19 @@ asmlinkage long sys_set_rt_task_param(pid_t pid, struct rt_task __user * param)
                        pid, tp.budget_policy);
                 goto out_unlock;
         }
-
+        if (tp.lookup_len > 0) {
+                lookup_table = kmalloc(tp.lookup_len * sizeof(unsigned int),
+                                       GFP_KERNEL);
+                if (!lookup_table)
+                        goto out_unlock;
+                if (copy_from_user(lookup_table, tp.lookup_table, tp.lookup_len
+                                   * sizeof(lookup_table))) {
+                        retval = -EFAULT;
+                        kfree(lookup_table);
+                        goto out_unlock;
+                }
+                tp.lookup_table = lookup_table;
+        }
         target->rt_param.task_params = tp;
 
         retval = 0;