1 files changed, 7 insertions, 0 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 99714d06eed5..86ad5b375934 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -112,6 +112,13 @@ 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.budget_policy != NO_ENFORCEMENT &&
+                tp.budget_policy != QUANTUM_ENFORCEMENT)
+        {
+                printk(KERN_INFO "litmus: real-time task %d rejected "
+                        "because unsupported budget enforcement policy specified\n", pid);
+                goto out_unlock;
+        }
 
         target->rt_param.task_params = tp;