tests: Refactor parallel test execution

1 files changed, 155 insertions, 0 deletions

diff --git a/cpukit/libmisc/testsupport/testparallel.c b/cpukit/libmisc/testsupport/testparallel.c
new file mode 100644
index 0000000000..681f769124
--- /dev/null
+++ b/cpukit/libmisc/testsupport/testparallel.c
@@ -0,0 +1,155 @@
+/*
+ * Copyright (c) 2013-2015 embedded brains GmbH.  All rights reserved.
+ *
+ *  embedded brains GmbH
+ *  Dornierstr. 4
+ *  82178 Puchheim
+ *  Germany
+ *  <rtems@embedded-brains.de>
+ *
+ * The license and distribution terms for this file may be
+ * found in the file LICENSE in this distribution or at
+ * http://www.rtems.org/license/LICENSE.
+ */
+
+
+#ifdef HAVE_CONFIG_H
+  #include "config.h"
+#endif
+
+#include <rtems/test.h>
+#include <rtems/score/assert.h>
+#include <rtems.h>
+
+static void stop_worker_timer(rtems_id timer_id, void *arg)
+{
+  rtems_test_parallel_context *ctx = arg;
+
+  _Atomic_Store_ulong(&ctx->stop, 1, ATOMIC_ORDER_RELAXED);
+}
+
+static void start_worker_stop_timer(
+  rtems_test_parallel_context *ctx,
+  rtems_interval duration
+)
+{
+  rtems_status_code sc;
+
+  _Atomic_Store_ulong(&ctx->stop, 0, ATOMIC_ORDER_RELEASE);
+
+  sc = rtems_timer_fire_after(
+    ctx->stop_worker_timer_id,
+    duration,
+    stop_worker_timer,
+    ctx
+  );
+  _Assert(sc == RTEMS_SUCCESSFUL);
+}
+
+static void run_tests(
+  rtems_test_parallel_context *ctx,
+  const rtems_test_parallel_job *jobs,
+  size_t job_count,
+  size_t worker_index
+)
+{
+  SMP_barrier_State bs = SMP_BARRIER_STATE_INITIALIZER;
+  size_t i;
+
+  for (i = 0; i < job_count; ++i) {
+    const rtems_test_parallel_job *job = &jobs[i];
+
+    if (rtems_test_parallel_is_master_worker(worker_index)) {
+      rtems_interval duration = (*job->init)(ctx, job->arg);
+
+      start_worker_stop_timer(ctx, duration);
+    }
+
+    _SMP_barrier_Wait(&ctx->barrier, &bs, ctx->worker_count);
+
+    (*job->body)(ctx, job->arg, worker_index);
+
+    _SMP_barrier_Wait(&ctx->barrier, &bs, ctx->worker_count);
+
+    if (rtems_test_parallel_is_master_worker(worker_index)) {
+      (*job->fini)(ctx, job->arg);
+    }
+  }
+}
+
+typedef struct {
+  rtems_test_parallel_context *ctx;
+  const rtems_test_parallel_job *jobs;
+  size_t job_count;
+  size_t worker_index;
+} worker_arg;
+
+static void worker_task(rtems_task_argument arg)
+{
+  worker_arg warg = *(worker_arg *) arg;
+  rtems_status_code sc;
+
+  sc = rtems_event_transient_send(warg.ctx->master_id);
+  _Assert(sc == RTEMS_SUCCESSFUL);
+
+  run_tests(warg.ctx, warg.jobs, warg.job_count, warg.worker_index);
+
+  rtems_task_delete(RTEMS_SELF);
+}
+
+void rtems_test_parallel(
+  rtems_test_parallel_context *ctx,
+  rtems_task_priority non_master_worker_priority,
+  const rtems_test_parallel_job *jobs,
+  size_t job_count
+)
+{
+  rtems_status_code sc;
+  size_t worker_index;
+
+  _Atomic_Init_ulong(&ctx->stop, 0);
+  _SMP_barrier_Control_initialize(&ctx->barrier);
+  ctx->worker_count = rtems_get_processor_count();
+  ctx->master_id = rtems_task_self();
+
+  sc = rtems_timer_create(
+    rtems_build_name('S', 'T', 'O', 'P'),
+    &ctx->stop_worker_timer_id
+  );
+  if (sc != RTEMS_SUCCESSFUL) {
+    rtems_fatal_error_occurred(sc);
+  }
+
+  for (worker_index = 1; worker_index < ctx->worker_count; ++worker_index) {
+    worker_arg warg = {
+      .ctx = ctx,
+      .jobs = jobs,
+      .job_count = job_count,
+      .worker_index = worker_index
+    };
+    rtems_id worker_id;
+
+    sc = rtems_task_create(
+      rtems_build_name('W', 'O', 'R', 'K'),
+      non_master_worker_priority,
+      RTEMS_MINIMUM_STACK_SIZE,
+      RTEMS_DEFAULT_MODES,
+      RTEMS_DEFAULT_ATTRIBUTES,
+      &worker_id
+    );
+    if (sc != RTEMS_SUCCESSFUL) {
+      rtems_fatal_error_occurred(sc);
+    }
+
+    sc = rtems_task_start(worker_id, worker_task, (rtems_task_argument) &warg);
+    _Assert(sc == RTEMS_SUCCESSFUL);
+
+    sc = rtems_event_transient_receive(RTEMS_WAIT, RTEMS_NO_TIMEOUT);
+    _Assert(sc == RTEMS_SUCCESSFUL);
+  }
+
+  run_tests(ctx, jobs, job_count, 0);
+
+  sc = rtems_timer_delete(ctx->stop_worker_timer_id);
+  _Assert(sc == RTEMS_SUCCESSFUL);
+}