Added support for choices in budget policy enforcement.wip-budget

NO_ENFORCEMENT - A job may execute beyond its declared execution time.
  Jobs notify the kernel that they are complete via liblitmus's
  sleep_next_period()
QUANTUM_ENFORCEMENT - The kernel terminates a job if its actual execution
  time exceeds the declared execution time.
PRECISE_ENFORCEMENT - Hook declared, but not yet implemented.  Plan to
  support this policy through hrtimers.  Error thrown if specified.

4 files changed, 19 insertions, 6 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index e43596a5104c..8b8f2c171a5e 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;
 
diff --git a/litmus/sched_cedf.c b/litmus/sched_cedf.c
index da44b451c9ad..76706a29245e 100644
--- a/litmus/sched_cedf.c
+++ b/litmus/sched_cedf.c
@@ -320,7 +320,7 @@ static noinline void job_completion(struct task_struct *t, int forced)
  */
 static void cedf_tick(struct task_struct* t)
 {
-        if (is_realtime(t) && budget_exhausted(t)) {
+        if (is_realtime(t) && budget_enforced(t) && budget_exhausted(t)) {
                 if (!is_np(t)) {
                         /* np tasks will be preempted when they become
                          * preemptable again
@@ -378,7 +378,9 @@ static struct task_struct* cedf_schedule(struct task_struct * prev)
         /* (0) Determine state */
         exists      = entry->scheduled != NULL;
         blocks      = exists && !is_running(entry->scheduled);
-        out_of_time = exists && budget_exhausted(entry->scheduled);
+        out_of_time = exists &&
+                                  budget_enforced(entry->scheduled) &&
+                                  budget_exhausted(entry->scheduled);
         np          = exists && is_np(entry->scheduled);
         sleep       = exists && get_rt_flags(entry->scheduled) == RT_F_SLEEP;
         preempt     = entry->scheduled != entry->linked;
diff --git a/litmus/sched_gsn_edf.c b/litmus/sched_gsn_edf.c
index b9310dd6f75c..6137c74729cb 100644
--- a/litmus/sched_gsn_edf.c
+++ b/litmus/sched_gsn_edf.c
@@ -336,7 +336,7 @@ static noinline void job_completion(struct task_struct *t, int forced)
  */
 static void gsnedf_tick(struct task_struct* t)
 {
-        if (is_realtime(t) && budget_exhausted(t)) {
+        if (is_realtime(t) && budget_enforced(t) && budget_exhausted(t)) {
                 if (!is_np(t)) {
                         /* np tasks will be preempted when they become
                          * preemptable again
@@ -399,7 +399,9 @@ static struct task_struct* gsnedf_schedule(struct task_struct * prev)
         /* (0) Determine state */
         exists      = entry->scheduled != NULL;
         blocks      = exists && !is_running(entry->scheduled);
-        out_of_time = exists && budget_exhausted(entry->scheduled);
+        out_of_time = exists &&
+                                  budget_enforced(entry->scheduled) &&
+                                  budget_exhausted(entry->scheduled);
         np          = exists && is_np(entry->scheduled);
         sleep       = exists && get_rt_flags(entry->scheduled) == RT_F_SLEEP;
         preempt     = entry->scheduled != entry->linked;
diff --git a/litmus/sched_psn_edf.c b/litmus/sched_psn_edf.c
index 7f71ecfaaaae..af0b30cb8b89 100644
--- a/litmus/sched_psn_edf.c
+++ b/litmus/sched_psn_edf.c
@@ -107,7 +107,7 @@ static void psnedf_tick(struct task_struct *t)
          */
         BUG_ON(is_realtime(t) && t != pedf->scheduled);
 
-        if (is_realtime(t) && budget_exhausted(t)) {
+        if (is_realtime(t) && budget_enforced(t) && budget_exhausted(t)) {
                 if (!is_np(t)) {
                         set_tsk_need_resched(t);
                         TRACE("psnedf_scheduler_tick: "
@@ -143,7 +143,9 @@ static struct task_struct* psnedf_schedule(struct task_struct * prev)
         /* (0) Determine state */
         exists      = pedf->scheduled != NULL;
         blocks      = exists && !is_running(pedf->scheduled);
-        out_of_time = exists && budget_exhausted(pedf->scheduled);
+        out_of_time = exists &&
+                                  budget_enforced(pedf->scheduled) &&
+                                  budget_exhausted(pedf->scheduled);
         np          = exists && is_np(pedf->scheduled);
         sleep       = exists && get_rt_flags(pedf->scheduled) == RT_F_SLEEP;
         preempt     = edf_preemption_needed(edf, prev);