|
Packit |
6c4009 |
#include <pthread.h>
|
|
Packit |
6c4009 |
#include <stdbool.h>
|
|
Packit |
6c4009 |
#include <stdio.h>
|
|
Packit |
6c4009 |
#include <stdlib.h>
|
|
Packit |
6c4009 |
#include <string.h>
|
|
Packit |
6c4009 |
#include <unistd.h>
|
|
Packit |
6c4009 |
#include <atomic.h>
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static pthread_cond_t cond1 = PTHREAD_COND_INITIALIZER;
|
|
Packit |
6c4009 |
static pthread_mutex_t mut1 = PTHREAD_MUTEX_INITIALIZER;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static pthread_cond_t cond2 = PTHREAD_COND_INITIALIZER;
|
|
Packit |
6c4009 |
static pthread_mutex_t mut2 = PTHREAD_MUTEX_INITIALIZER;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static bool last_round;
|
|
Packit |
6c4009 |
static int ntogo;
|
|
Packit |
6c4009 |
static bool alldone;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void *
|
|
Packit |
6c4009 |
cons (void *arg)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
pthread_mutex_lock (&mut1);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
do
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
if (atomic_decrement_and_test (&ntogo))
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
pthread_mutex_lock (&mut2);
|
|
Packit |
6c4009 |
alldone = true;
|
|
Packit |
6c4009 |
pthread_cond_signal (&cond2);
|
|
Packit |
6c4009 |
pthread_mutex_unlock (&mut2);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
pthread_cond_wait (&cond1, &mut1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
while (! last_round);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
pthread_mutex_unlock (&mut1);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return NULL;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
int
|
|
Packit |
6c4009 |
main (int argc, char *argv[])
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
int opt;
|
|
Packit |
6c4009 |
int err;
|
|
Packit |
6c4009 |
int nthreads = 10;
|
|
Packit |
6c4009 |
int nrounds = 100;
|
|
Packit |
6c4009 |
bool keeplock = false;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
while ((opt = getopt (argc, argv, "n:r:k")) != -1)
|
|
Packit |
6c4009 |
switch (opt)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
case 'n':
|
|
Packit |
6c4009 |
nthreads = atol (optarg);
|
|
Packit |
6c4009 |
break;
|
|
Packit |
6c4009 |
case 'r':
|
|
Packit |
6c4009 |
nrounds = atol (optarg);
|
|
Packit |
6c4009 |
break;
|
|
Packit |
6c4009 |
case 'k':
|
|
Packit |
6c4009 |
keeplock = true;
|
|
Packit |
6c4009 |
break;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
ntogo = nthreads;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
pthread_t th[nthreads];
|
|
Packit |
6c4009 |
int i;
|
|
Packit |
6c4009 |
for (i = 0; __builtin_expect (i < nthreads, 1); ++i)
|
|
Packit |
6c4009 |
if (__glibc_unlikely ((err = pthread_create (&th[i], NULL, cons, (void *) (long) i)) != 0))
|
|
Packit |
6c4009 |
printf ("pthread_create: %s\n", strerror (err));
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (i = 0; __builtin_expect (i < nrounds, 1); ++i)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
pthread_mutex_lock (&mut2);
|
|
Packit |
6c4009 |
while (! alldone)
|
|
Packit |
6c4009 |
pthread_cond_wait (&cond2, &mut2);
|
|
Packit |
6c4009 |
pthread_mutex_unlock (&mut2);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
pthread_mutex_lock (&mut1);
|
|
Packit |
6c4009 |
if (! keeplock)
|
|
Packit |
6c4009 |
pthread_mutex_unlock (&mut1);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
ntogo = nthreads;
|
|
Packit |
6c4009 |
alldone = false;
|
|
Packit |
6c4009 |
if (i + 1 >= nrounds)
|
|
Packit |
6c4009 |
last_round = true;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
pthread_cond_broadcast (&cond1);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (keeplock)
|
|
Packit |
6c4009 |
pthread_mutex_unlock (&mut1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (i = 0; i < nthreads; ++i)
|
|
Packit |
6c4009 |
if ((err = pthread_join (th[i], NULL)) != 0)
|
|
Packit |
6c4009 |
printf ("pthread_create: %s\n", strerror (err));
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return 0;
|
|
Packit |
6c4009 |
}
|