Added support for early releasing

2 files changed, 5 insertions, 1 deletions

diff --git a/include/litmus/reservations/gedf_reservation.h b/include/litmus/reservations/gedf_reservation.h
index 578b9cad5838..17965c89846c 100644
--- a/include/litmus/reservations/gedf_reservation.h
+++ b/include/litmus/reservations/gedf_reservation.h
@@ -15,6 +15,7 @@ struct gedf_reservation {
         lt_t cur_budget;
         int will_remove;
         int blocked;
+        int allow_early_release;
 };
 
 struct gedf_cpu_entry {
diff --git a/litmus/reservations/gedf_reservation.c b/litmus/reservations/gedf_reservation.c
index 1649eaf026d7..deb0c9e6e3ef 100644
--- a/litmus/reservations/gedf_reservation.c
+++ b/litmus/reservations/gedf_reservation.c
@@ -92,7 +92,8 @@ static void requeue(
         BUG_ON(!gedf_res);
         BUG_ON(is_queued_res(&gedf_res->res));
 
-        if (lt_before_eq(gedf_res->res.replenishment_time, litmus_clock()))
+        if (gedf_res->allow_early_release ||
+                        lt_before_eq(gedf_res->res.replenishment_time, litmus_clock()))
                 __add_ready_res(&gedf_env->domain, &gedf_res->res);
         else
                 __add_release_res(&gedf_env->domain, &gedf_res->res);
@@ -331,6 +332,7 @@ long alloc_gedf_task_reservation(
         gedf_task_res->task = task;
         gedf_task_res->gedf_res.res.priority = ULLONG_MAX - get_rt_relative_deadline(task);
         gedf_task_res->gedf_res.res.cur_budget = get_exec_cost(task);
+        gedf_task_res->gedf_res.allow_early_release = is_early_releasing(task);
 
         *_res = gedf_task_res;
         return 0;
@@ -353,6 +355,7 @@ long alloc_gedf_container_reservation(
         gedf_cont_res->max_budget = max_budget;
         gedf_cont_res->period = period;
         gedf_cont_res->relative_deadline = relative_deadline;
+        gedf_cont_res->gedf_res.allow_early_release = 0;
 
         *_res = gedf_cont_res;
         return 0;