Implement pthread_barrier calls in sthread, and test them in McMini

[simgrid.git] / teshsuite / mc / mcmini / simple_barrier_with_threads_ok.c

diff --git a/teshsuite/mc/mcmini/simple_barrier_with_threads_ok.c b/teshsuite/mc/mcmini/simple_barrier_with_threads_ok.c
new file mode 100644 (file)
index 0000000..0e38451
--- /dev/null
+++ b/teshsuite/mc/mcmini/simple_barrier_with_threads_ok.c
@@ -0,0 +1,54 @@
+#define _POSIX_C_SOURCE 200809L
+
+#include <pthread.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+int DEBUG;
+
+pthread_barrier_t barrier;
+
+static void * thread_doit(void *t)
+{
+    int *tid = (int*)t;
+    if(DEBUG) {
+        printf("Thread %d: Waiting at barrier\n", *tid);
+    }
+    pthread_barrier_wait(&barrier);
+    if(DEBUG) {
+        printf("Thread %d: Crossed barrier\n", *tid);
+    }
+    return NULL;
+}
+
+int main(int argc, char* argv[])
+{
+    if(argc < 3) {
+        printf("Expected usage: %s THREAD_NUM DEBUG_FLAG\n", argv[0]);
+        printf("DEBUG_FLAG: 0 - Don't display debug information, 1 - Display debug information\n");
+        return -1;
+    }
+
+    int THREAD_NUM = atoi(argv[1]);
+    DEBUG = atoi(argv[2]);
+
+    pthread_t *threads = malloc(sizeof(pthread_t) * THREAD_NUM);
+
+    pthread_barrier_init(&barrier, NULL, THREAD_NUM);
+
+    int *tids = malloc(sizeof(int) * THREAD_NUM);
+    for(int i = 0; i < THREAD_NUM; i++) {
+        tids[i] = i;
+        pthread_create(&threads[i], NULL, &thread_doit, &tids[i]);
+    }
+
+    for(int i = 0; i < THREAD_NUM; i++) {
+        pthread_join(threads[i], NULL);
+    }
+
+    free(threads);
+    free(tids);
+    pthread_barrier_destroy(&barrier);
+    
+    return 0;
+}