created generic domain interface for tasks

1 files changed, 62 insertions, 3 deletions

diff --git a/litmus/rt_domain.c b/litmus/rt_domain.c
index fd0e6c235e1..f781a2c61fc 100644
--- a/litmus/rt_domain.c
+++ b/litmus/rt_domain.c
@@ -14,13 +14,11 @@
 #include <litmus/litmus.h>
 #include <litmus/sched_plugin.h>
 #include <litmus/sched_trace.h>
-
 #include <litmus/rt_domain.h>
-
 #include <litmus/trace.h>
-
 #include <litmus/bheap.h>
 
+
 /* Uncomment when debugging timer races... */
 #if 0
 #define VTRACE_TASK TRACE_TASK
@@ -354,3 +352,64 @@ void __add_release(rt_domain_t* rt, struct task_struct *task)
         TS_SCHED2_END(task);
 }
 
+/******************************************************************************
+ * domain_t wrapper
+ ******************************************************************************/
+
+/* pd_requeue - calls underlying rt_domain add methods.
+ * If the task is not yet released, it is inserted into the rt_domain
+ * ready queue. Otherwise, it is queued for release.
+ *
+ * Assumes the caller already holds dom->lock.
+ */
+static void pd_requeue(domain_t *dom, struct task_struct *task)
+{
+        rt_domain_t *domain = (rt_domain_t*)dom->data;
+        BUG_ON(!task || !is_realtime(task));
+        BUG_ON(is_queued(task));
+
+        if (is_released(task, litmus_clock())) {
+                __add_ready(domain, task);
+        } else {
+                /* task has to wait for next release */
+                add_release(domain, task);
+        }
+}
+
+/* pd_take_ready - removes and returns the next ready task from the rt_domain
+ *
+ * Assumes the caller already holds dom->lock.
+ */
+static struct task_struct* pd_take_ready(domain_t *dom)
+{
+        return __take_ready((rt_domain_t*)dom->data);
+}
+
+/* pd_peek_ready - returns the head of the rt_domain ready queue
+ *
+ * Assumes the caller already holds dom->lock.
+ */
+static struct task_struct* pd_peek_ready(domain_t *dom)
+{
+        return __next_ready((rt_domain_t*)dom->data);
+}
+
+/* pd_domain_init - create a generic domain wrapper for an rt_domain
+ */
+void pd_domain_init(domain_t *dom, bheap_prio_t order,
+                    check_resched_needed_t check,
+                    release_jobs_t release,
+                    preempt_needed_t preempt_needed)
+{
+        rt_domain_t *domain = kmalloc(sizeof(rt_domain_t), GFP_ATOMIC);
+
+        rt_domain_init(domain, order, check, release);
+        domain_init(dom, &domain->ready_lock,
+                    pd_requeue, pd_peek_ready, pd_take_ready,
+                    preempt_needed);
+}
+
+void pd_domain_free(domain_t *dom)
+{
+        kfree(dom->data);
+}