Add library support for creating EDF-os tasks

2 files changed, 58 insertions, 0 deletions

diff --git a/src/litmus.c b/src/litmus.c
index c0dae95..b0416c4 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -107,6 +107,34 @@ int sporadic_task_ns_edffm(lt_t e, lt_t p, lt_t phase, int cpu,
         return set_rt_task_param(gettid(), &param);
 }
 
+int sporadic_task_ns_edfos(lt_t e, lt_t p, lt_t phase,
+                lt_t fracs[][2], task_class_t cls,
+                budget_policy_t budget_policy)
+{
+        struct rt_task param;
+        struct edfos_params os;
+        int ret, i;
+        os.migrat = 0;
+        os.first_cpu = UINT_MAX;
+        param.exec_cost = e;
+        param.period    = p;
+        for (i = 0; i < NR_CPUS; i++) {
+                if (fracs[i][0] > 0) {
+                        os.migrat++;
+                        if (os.first_cpu == UINT_MAX)
+                                os.first_cpu = i;
+                }
+                os.fraction[i][0] = fracs[i][0];
+                os.fraction[i][1] = fracs[i][1];
+        }
+        param.semi_part.os = os;
+        param.cls       = cls;
+        param.phase     = phase;
+        param.budget_policy = budget_policy;
+
+        return set_rt_task_param(gettid(), &param);
+}
+
 int sporadic_task_ns_npsf(lt_t e, lt_t p, lt_t phase,
                 int cpu, task_class_t cls, int npsf_id,
                 budget_policy_t budget_policy, int set_cpu_set)
diff --git a/src/task.c b/src/task.c
index daf95ca..e618d9e 100644
--- a/src/task.c
+++ b/src/task.c
@@ -71,6 +71,36 @@ int __create_rt_task_edffm(rt_fn_t rt_prog, void *arg, int cpu, int wcet,
                                 (rt_setup_fn_t) set_rt_task_param, &params);
 }
 
+int __create_rt_task_edfos(rt_fn_t rt_prog, void *arg, int wcet,
+                        int period, lt_t fracs[][2], task_class_t class)
+{
+        struct rt_task params;
+        struct edfos_params os;
+        int i;
+        params.period    = period;
+        params.exec_cost = wcet;
+        params.cls       = class;
+        params.phase     = 0;
+        /* enforce budget for tasks that might not use sleep_next_period() */
+        params.budget_policy = QUANTUM_ENFORCEMENT;
+
+        os.migrat = 0;
+        os.first_cpu = UINT_MAX;
+        for (i = 0; i < NR_CPUS; i++) {
+                if (fracs[i][0] > 0) {
+                        os.migrat++;
+                        if (os.first_cpu == UINT_MAX)
+                                os.first_cpu = i;
+                }
+                os.fraction[i][0] = fracs[i][0];
+                os.fraction[i][1] = fracs[i][1];
+        }
+        params.semi_part.os = os;
+
+        return __launch_rt_task(rt_prog, arg,
+                                (rt_setup_fn_t) set_rt_task_param, &params);
+}
+
 int __create_rt_task_npsf(rt_fn_t rt_prog, void *arg, int cpu, int wcet,
                 int period, int npsf_id, task_class_t class)
 {