2 files changed, 24 insertions, 4 deletions

diff --git a/include/litmus.h b/include/litmus.h
index a5d3e2e..7ee13dc 100644
--- a/include/litmus.h
+++ b/include/litmus.h
@@ -9,18 +9,21 @@ typedef int pid_t;	 /* PID of a task */
 /* obtain the PID of a thread */
 pid_t gettid(void);
 
+/* migrate to partition */
+int be_migrate_to(int target_cpu);
+
 int set_rt_task_param(pid_t pid, struct rt_task* param);
 int get_rt_task_param(pid_t pid, struct rt_task* param);
 
 /* setup helper */
 /* times are givin in ms */
 int sporadic_task(lt_t e, lt_t p, lt_t phase,
-                  int partition, task_class_t cls);
+                  int partition, task_class_t cls, int set_cpu_set);
 
 #define sporadic_global(e, p) \
-        sporadic_task(e, p, 0, 0, RT_CLASS_SOFT)
+        sporadic_task(e, p, 0, 0, RT_CLASS_SOFT, 0)
 #define sporadic_partitioned(e, p, cpu) \
-        sporadic_task(e, p, 0, cpu, RT_CLASS_SOFT)
+        sporadic_task(e, p, 0, cpu, RT_CLASS_SOFT, 1)
 
 /* file descriptor attached shared objects support */
 typedef enum  {
diff --git a/src/litmus.c b/src/litmus.c
index 4fe6572..8f03f4c 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -5,6 +5,8 @@
 #include <signal.h>
 #include <sys/mman.h>
 
+#include <sched.h> /* for cpu sets */
+
 #include "litmus.h"
 #include "internal.h"
 
@@ -29,15 +31,30 @@ task_class_t str2class(const char* str)
 
 #define NS_PER_MS 1000000
 
+/* only for best-effort execution: migrate to target_cpu */
+int be_migrate_to(int target_cpu)
+{
+        cpu_set_t cpu_set;
+
+        CPU_ZERO(&cpu_set);
+        CPU_SET(target_cpu, &cpu_set);
+        return sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set);
+}
+
 int sporadic_task(lt_t e, lt_t p, lt_t phase,
-                  int cpu, task_class_t cls)
+                  int cpu, task_class_t cls, int set_cpu_set)
 {
         struct rt_task param;
+        int ret;
         param.exec_cost = e * NS_PER_MS;
         param.period    = p * NS_PER_MS;
         param.cpu       = cpu;
         param.cls       = cls;
         param.phase     = phase;
+        if (set_cpu_set) {             
+                ret = be_migrate_to(cpu);
+                check("migrate to cpu");
+        }
         return set_rt_task_param(gettid(), &param);
 }