2 files changed, 11 insertions, 6 deletions

diff --git a/src/litmus.c b/src/litmus.c
index f71f337..2c4bc42 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -42,15 +42,16 @@ int be_migrate_to(int target_cpu)
 }
 
 int sporadic_task(lt_t e, lt_t p, lt_t phase,
-                  int cpu, task_class_t cls,
+                  int cpu, int criticality, task_class_t cls,
                   budget_policy_t budget_policy, int set_cpu_set)
 {
         return sporadic_task_ns(e * NS_PER_MS, p * NS_PER_MS, phase * NS_PER_MS,
-                                cpu, cls, budget_policy, set_cpu_set);
+                                cpu, criticality, cls, budget_policy,
+                                set_cpu_set);
 }
 
 int sporadic_task_ns(lt_t e, lt_t p, lt_t phase,
-                        int cpu, task_class_t cls,
+                        int cpu, int criticality, task_class_t cls,
                         budget_policy_t budget_policy, int set_cpu_set)
 {
         struct rt_task param;
@@ -58,6 +59,7 @@ int sporadic_task_ns(lt_t e, lt_t p, lt_t phase,
         param.exec_cost = e;
         param.period    = p;
         param.cpu       = cpu;
+        param.crit      = criticality;
         param.cls       = cls;
         param.phase     = phase;
         param.budget_policy = budget_policy;
diff --git a/src/task.c b/src/task.c
index 4d237bd..ece5f64 100644
--- a/src/task.c
+++ b/src/task.c
@@ -41,11 +41,12 @@ int __launch_rt_task(rt_fn_t rt_prog, void *rt_arg, rt_setup_fn_t setup,
 }
 
 int __create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, int wcet, int period,
-                     task_class_t class)
+                     int criticality, task_class_t class)
 {
         struct rt_task params;
         params.cpu       = cpu;
         params.period    = period;
+        params.crit      = criticality;
         params.exec_cost = wcet;
         params.cls       = class;
         params.phase     = 0;
@@ -56,8 +57,10 @@ int __create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, int wcet, int period,
                                 (rt_setup_fn_t) set_rt_task_param, &params);
 }
 
-int create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, int wcet, int period) {
-        return __create_rt_task(rt_prog, arg, cpu, wcet, period, RT_CLASS_HARD);
+int create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, int wcet, int period,
+                int criticality) {
+        return __create_rt_task(rt_prog, arg, cpu, wcet, period, criticality,
+                        RT_CLASS_HARD);
 }