55 lines
1.2 KiB
C
55 lines
1.2 KiB
C
#include <pthread.h>
|
|
#include <inttypes.h>
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdatomic.h>
|
|
|
|
#include "lock.h"
|
|
|
|
#define RUNS (4096 * 256)
|
|
#define THREADS 4
|
|
|
|
static struct lock_t lock;
|
|
|
|
static int leader[RUNS] = { 0 };
|
|
static atomic_int nb_leaders[RUNS] = { 0 }; // used to check correctness
|
|
|
|
void* elect(void* _tid) {
|
|
intptr_t tid = (intptr_t) _tid;
|
|
for (int r = 0; r < RUNS; r++) {
|
|
lock_acquire(&lock);
|
|
if (leader[r] == 0) {
|
|
leader[r] = tid;
|
|
atomic_fetch_add(&nb_leaders[r], 1); // used to check correctness
|
|
}
|
|
lock_release(&lock);
|
|
// **Correct**, a single thread tries to become leader at a time.
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
lock_init(&lock);
|
|
pthread_t handlers[THREADS];
|
|
|
|
for (intptr_t i = 0; i < THREADS; i++) {
|
|
int res = pthread_create(&handlers[i], NULL, elect, (void*)(i + 1));
|
|
assert(!res);
|
|
}
|
|
|
|
for (int i = 0; i < THREADS; i++) {
|
|
int res = pthread_join(handlers[i], NULL);
|
|
assert(!res);
|
|
}
|
|
|
|
int r = 0;
|
|
for (; r < RUNS; r++) {
|
|
if (nb_leaders[r] != 1) {
|
|
printf("Leader election for round %d failed.\n", r);
|
|
break;
|
|
}
|
|
}
|
|
if (r == RUNS) {
|
|
printf("Looks correct to me! :)\n");
|
|
}
|
|
}
|