1 files changed, 8 insertions, 0 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 11ccaafd50de..26938acacafc 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -110,6 +110,14 @@ asmlinkage long sys_set_rt_task_param(pid_t pid, struct rt_task __user * param)
                        "because wcet > period\n", pid);
                 goto out_unlock;
         }
+        if (    tp.cls != RT_CLASS_HARD &&
+                tp.cls != RT_CLASS_SOFT &&
+                tp.cls != RT_CLASS_BEST_EFFORT)
+        {
+                printk(KERN_INFO "litmus: real-time task %d rejected "
+                                 "because its class is invalid\n");
+                goto out_unlock;
+        }
         if (tp.budget_policy != NO_ENFORCEMENT &&
             tp.budget_policy != QUANTUM_ENFORCEMENT &&
             tp.budget_policy != PRECISE_ENFORCEMENT)