teshsuite/mc/mcmini/simple_barrier_with_threads_ok.c

   1 #define _POSIX_C_SOURCE 200809L
   2
   3 #include <pthread.h>
   4 #include <stdlib.h>
   5 #include <stdio.h>
   6
   7 int DEBUG;
   8
   9 pthread_barrier_t barrier;
  10
  11 static void * thread_doit(void *t)
  12 {
  13     int *tid = (int*)t;
  14     if(DEBUG) {
  15         printf("Thread %d: Waiting at barrier\n", *tid);
  16     }
  17     pthread_barrier_wait(&barrier);
  18     if(DEBUG) {
  19         printf("Thread %d: Crossed barrier\n", *tid);
  20     }
  21     return NULL;
  22 }
  23
  24 int main(int argc, char* argv[])
  25 {
  26     if(argc < 3) {
  27         printf("Expected usage: %s THREAD_NUM DEBUG_FLAG\n", argv[0]);
  28         printf("DEBUG_FLAG: 0 - Don't display debug information, 1 - Display debug information\n");
  29         return -1;
  30     }
  31
  32     int THREAD_NUM = atoi(argv[1]);
  33     DEBUG = atoi(argv[2]);
  34
  35     pthread_t *threads = malloc(sizeof(pthread_t) * THREAD_NUM);
  36
  37     pthread_barrier_init(&barrier, NULL, THREAD_NUM);
  38
  39     int *tids = malloc(sizeof(int) * THREAD_NUM);
  40     for(int i = 0; i < THREAD_NUM; i++) {
  41         tids[i] = i;
  42         pthread_create(&threads[i], NULL, &thread_doit, &tids[i]);
  43     }
  44
  45     for(int i = 0; i < THREAD_NUM; i++) {
  46         pthread_join(threads[i], NULL);
  47     }
  48
  49     free(threads);
  50     free(tids);
  51     pthread_barrier_destroy(&barrier);
  52
  53     return 0;
  54 }