3 files changed, 450 insertions, 0 deletions

diff --git a/bin/common.c b/bin/common.c
index 452b882..4fd24e1 100644
--- a/bin/common.c
+++ b/bin/common.c
@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <errno.h>
+#include <string.h>
 
 #include "common.h"
 
@@ -9,3 +10,121 @@ void bail_out(const char* msg)
         perror(msg);
         exit(-1 * errno);
 }
+
+/* EDF-WM helper functions to parse a custom text file format to "easily"
+ * launch tests with rtspin and rt_launch:
+ *
+ * Format for task:
+ *
+ * <task_id execution_cost period phase cpu slices_number> .
+ *
+ * If the task is split on multiple slices, slices_number is non 0
+ * and we scan a list of slice parameters up to slices_number:
+ *
+ * Format for slices:
+ *
+ * <task_id cpu deadline(from job release) budget offset> .
+ *
+ * The offset is the start time for the slice relative to the job release.
+ *
+ * Example:
+ * 14 2.26245771754 10 0 5 2
+ * 14 5 5.000000 1.497306 0.000000
+ * 14 7 10.000000 0.765152 5.000000
+ */
+
+#define fms_to_ns(x)    (lt_t)(((x) * __NS_PER_MS))
+/*
+ * <task_id, cpu, deadline (from job release), budget, offset> .
+ */
+int parse_edfwm_slice(FILE *ts, int slices_no, int task_id,
+                      struct rt_task *rt)
+{
+        int i, tid;
+        unsigned int cpu;
+        double deadline, budget, offset;
+
+        lt_t total_budget = 0;
+
+        struct edf_wm_params* wm = (struct edf_wm_params*) &rt->semi_part;
+
+        for (i = 0; i < slices_no; i++) {
+
+                if (fscanf(ts, "%d %u %lf %lf %lf\n", &tid, &cpu,
+                                &deadline, &budget, &offset) != EOF) {
+
+                        if (task_id != tid) {
+                                fprintf(stderr, "task_id %d != tid %d\n",
+                                                task_id, tid);
+                                return -1;
+                        }
+
+                        wm->slices[i].deadline = fms_to_ns(deadline);
+                        wm->slices[i].budget = fms_to_ns(budget);
+                        wm->slices[i].offset = fms_to_ns(offset);
+                        wm->slices[i].cpu    = cpu;
+
+                        printf("slice(tid, cpu, d, e, ph) = (%d, %u, %llu, %llu, %llu)\n",
+                                        tid, cpu, wm->slices[i].deadline,
+                                        wm->slices[i].budget, wm->slices[i].offset);
+
+                        if (wm->slices[i].budget < MIN_EDF_WM_SLICE_SIZE) {
+
+                                fprintf(stderr, "Slice %llu is too small\n",
+                                                wm->slices[i].budget);
+                                return -1;
+                        }
+
+                        total_budget += wm->slices[i].budget;
+                }
+
+                if (ferror(ts)) {
+                        fprintf(stderr, "Cannot read file\n");
+                        return -1;
+                }
+        }
+        wm->count = slices_no;
+        rt->exec_cost = total_budget;
+        printf("--- total %u slices ---\n", wm->count);
+        return 0;
+}
+
+/*
+ * <task_id, execution_cost, period, phase, cpu, slices_number> .
+ */
+int parse_edfwm_ts_file(FILE *ts, struct rt_task *rt)
+{
+        int task_id, ret = 1;
+        unsigned int cpu, sliceno;
+        double fwcet, fperiod, fphase;
+
+        ret = fscanf(ts, "%d %lf %lf %lf %d %d\n",
+                        &task_id, &fwcet, &fperiod, &fphase, &cpu, &sliceno);
+
+        if (ferror(ts))
+                goto err;
+
+        rt->exec_cost = fms_to_ns(fwcet);
+        rt->period = fms_to_ns(fperiod);
+        rt->phase = fms_to_ns(fphase);
+        rt->cpu = cpu;
+        rt->cls = RT_CLASS_HARD;
+        rt->budget_policy = PRECISE_ENFORCEMENT;
+
+        printf("(tid, wcet, period, ph, cpu, slices) = "
+                        "(%d, %llu, %llu, %llu, %u, %u)\n",
+                        task_id, rt->exec_cost, rt->period, rt->phase, cpu, sliceno);
+        if (sliceno > 0) {
+                memset(&rt->semi_part, 0, sizeof(struct edf_wm_params));
+                ret = parse_edfwm_slice(ts, sliceno, task_id, rt);
+                if (ret < 0)
+                        goto err;
+        }
+
+        return 0;
+
+err:
+        fprintf(stderr, "Error parsing file\n");
+        return -1;
+}
+
diff --git a/bin/rt_launch_edfwm.c b/bin/rt_launch_edfwm.c
new file mode 100644
index 0000000..9e8a322
--- /dev/null
+++ b/bin/rt_launch_edfwm.c
@@ -0,0 +1,98 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <limits.h>
+#include <signal.h>
+
+#include "litmus.h"
+#include "common.h"
+
+typedef struct {
+        int  wait;
+        char * exec_path;
+        char ** argv;
+} startup_info_t;
+
+
+int launch(void *task_info_p) {
+        startup_info_t *info = (startup_info_t*) task_info_p;
+        int ret;
+        if (info->wait) {
+                ret = wait_for_ts_release();
+                if (ret != 0)
+                        perror("wait_for_ts_release()");
+        }
+        ret = execvp(info->exec_path, info->argv);
+        perror("execv failed");
+        return ret;
+}
+
+void usage(char *error) {
+        fprintf(stderr, "%s\nUsage: rt_launch [-w] task_parameters_file "
+                        "program [arg1 arg2 ...]\n"
+                        "\t-w\tSynchronous release\n"
+                        "\tprogram to be launched\n",
+                        error);
+        exit(1);
+}
+
+#define OPTSTR "w"
+
+int main(int argc, char** argv)
+{
+        int ret;
+        int wait = 0;
+        int opt;
+        startup_info_t info;
+
+        struct rt_task rt;
+        FILE *file;
+
+        while ((opt = getopt(argc, argv, OPTSTR)) != -1) {
+                switch (opt) {
+                case 'w':
+                        wait = 1;
+                        break;
+                case ':':
+                        usage("Argument missing.");
+                        break;
+                case '?':
+                default:
+                        usage("Bad argument.");
+                        break;
+                }
+        }
+
+        signal(SIGUSR1, SIG_IGN);
+
+        if (argc - optind < 2)
+                usage("Arguments missing.");
+
+        if ((file = fopen(argv[optind + 0], "r")) == NULL) {
+                fprintf(stderr, "Cannot open %s\n", argv[1]);
+                return -1;
+        }
+
+        memset(&rt, 0, sizeof(struct rt_task));
+
+        if (parse_edfwm_ts_file(file, &rt) < 0)
+                bail_out("Could not parse file\n");
+
+        if (sporadic_task_ns_semi(&rt) < 0)
+                bail_out("could not setup rt task params");
+
+        fclose(file);
+
+        info.exec_path = argv[optind + 1];
+        info.argv      = argv + optind + 1;
+        info.wait      = wait;
+
+        ret = create_rt_task_semi(launch, &info, &rt);
+
+        if (ret < 0)
+                bail_out("could not create rt child process");
+
+        return 0;
+}
+
diff --git a/bin/rtspin_edfwm.c b/bin/rtspin_edfwm.c
new file mode 100644
index 0000000..21a5f3b
--- /dev/null
+++ b/bin/rtspin_edfwm.c
@@ -0,0 +1,233 @@
+#include <sys/time.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <time.h>
+#include <string.h>
+
+#include "litmus.h"
+#include "common.h"
+
+
+static double cputime()
+{
+        struct timespec ts;
+        int err;
+        err = clock_gettime(CLOCK_THREAD_CPUTIME_ID, &ts);
+        if (err != 0)
+                perror("clock_gettime");
+        return (ts.tv_sec + 1E-9 * ts.tv_nsec);
+}
+
+static double wctime()
+{
+        struct timeval tv;
+        gettimeofday(&tv, NULL);
+        return (tv.tv_sec + 1E-6 * tv.tv_usec);
+}
+
+void usage(char *error) {
+        fprintf(stderr, "Error: %s\n", error);
+        fprintf(stderr,
+                "Usage: rt_spin [-w] task_parameters_file duration\n"
+                "       rt_spin -l\n");
+        exit(1);
+}
+
+#define NUMS 4096
+static int num[NUMS];
+static double loop_length = 1.0;
+static char* progname;
+
+static int loop_once(void)
+{
+        int i, j = 0;
+        for (i = 0; i < NUMS; i++)
+                j += num[i]++;
+        return j;
+}
+
+static int loop_for(double exec_time)
+{
+        double t = 0;
+        int tmp = 0;
+/*      while (t + loop_length < exec_time) {
+                tmp += loop_once();
+                t += loop_length;
+        }
+*/
+        double start = cputime();
+        double now = cputime();
+        while (now + loop_length < start + exec_time) {
+                tmp += loop_once();
+                t += loop_length;
+                now = cputime();
+        }
+
+        return tmp;
+}
+
+static void fine_tune(double interval)
+{
+        double start, end, delta;
+
+        start = wctime();
+        loop_for(interval);
+        end = wctime();
+        delta = (end - start - interval) / interval;
+        if (delta != 0)
+                loop_length = loop_length / (1 - delta);
+}
+
+static void configure_loop(void)
+{
+        double start;
+
+        /* prime cache */
+        loop_once();
+        loop_once();
+        loop_once();
+
+        /* measure */
+        start = wctime();
+        loop_once(); /* hope we didn't get preempted  */
+        loop_length = wctime();
+        loop_length -= start;
+
+        /* fine tune */
+        fine_tune(0.1);
+        fine_tune(0.1);
+        fine_tune(0.1);
+}
+
+static void show_loop_length(void)
+{
+        printf("%s/%d: loop_length=%f (%ldus)\n",
+               progname, getpid(), loop_length,
+               (long) (loop_length * 1000000));
+}
+
+static void debug_delay_loop(void)
+{
+        double start, end, delay;
+        show_loop_length();
+        while (1) {
+                for (delay = 0.5; delay > 0.01; delay -= 0.01) {
+                        start = wctime();
+                        loop_for(delay);
+                        end = wctime();
+                        printf("%6.4fs: looped for %10.8fs, delta=%11.8fs, error=%7.4f%%\n",
+                               delay,
+                               end - start,
+                               end - start - delay,
+                               100 * (end - start - delay) / delay);
+                }
+        }
+}
+
+static int job(double exec_time)
+{
+        loop_for(exec_time);
+        sleep_next_period();
+        return 0;
+}
+
+#define OPTSTR "wld:v"
+
+int main(int argc, char** argv)
+{
+        int ret;
+
+        int opt;
+        int wait = 0;
+        int test_loop = 0;
+        int skip_config = 0;
+        int verbose = 0;
+        double wcet_ms;
+        double duration, start;
+
+        struct rt_task rt;
+        FILE *file;
+
+        progname = argv[0];
+
+        while ((opt = getopt(argc, argv, OPTSTR)) != -1) {
+                switch (opt) {
+                case 'w':
+                        wait = 1;
+                        break;
+                case 'l':
+                        test_loop = 1;
+                        break;
+                case 'd':
+                        /* manually configure delay per loop iteration
+                         * unit: microseconds */
+                        loop_length = atof(optarg) / 1000000;
+                        skip_config = 1;
+                        break;
+                case 'v':
+                        verbose = 1;
+                        break;
+                case ':':
+                        usage("Argument missing.");
+                        break;
+                case '?':
+                default:
+                        usage("Bad argument.");
+                        break;
+                }
+        }
+
+
+        if (!skip_config)
+                configure_loop();
+
+        if (test_loop) {
+                debug_delay_loop();
+                return 0;
+        }
+
+        if (argc - optind < 2)
+                usage("Arguments missing.");
+
+        if ((file = fopen(argv[optind + 0], "r")) == NULL) {
+                fprintf(stderr, "Cannot open %s\n", argv[1]);
+                return -1;
+        }
+        duration = atof(argv[optind + 1]);
+
+        memset(&rt, 0, sizeof(struct rt_task));
+
+        if (parse_edfwm_ts_file(file, &rt) < 0)
+                bail_out("Could not parse file\n");
+
+        if (sporadic_task_ns_semi(&rt) < 0)
+                bail_out("could not setup rt task params");
+
+        fclose(file);
+
+        if (verbose)
+                show_loop_length();
+
+        init_litmus();
+
+        ret = task_mode(LITMUS_RT_TASK);
+        if (ret != 0)
+                bail_out("could not become RT task");
+
+        if (wait) {
+                ret = wait_for_ts_release();
+                if (ret != 0)
+                        bail_out("wait_for_ts_release()");
+        }
+
+        wcet_ms = ((double) rt.exec_cost ) / __NS_PER_MS;
+        start = wctime();
+
+        while (start + duration > wctime()) {
+                job(wcet_ms * 0.0009); /* 90% wcet, in seconds */
+        }
+
+        return 0;
+}