static void
try_lock_mutex_test(void)
{
int i, res;
ethr_tid tid;
res = ethr_mutex_init(&tlmt_mtx1);
ASSERT(res == 0);
res = ethr_mutex_init(&tlmt_mtx2);
ASSERT(res == 0);
res = ethr_cond_init(&tlmt_cnd2);
ASSERT(res == 0);
tlmt_mtx1_locked = 0;
tlmt_mtx1_do_unlock = 0;
res = ethr_thr_create(&tid, tlmt_thread, NULL, NULL);
ASSERT(res == 0);
ethr_mutex_lock(&tlmt_mtx2);
while (!tlmt_mtx1_locked) {
res = ethr_cond_wait(&tlmt_cnd2, &tlmt_mtx2);
ASSERT(res == 0 || res == EINTR);
}
ethr_mutex_unlock(&tlmt_mtx2);
for (i = 0; i < 10; i++) {
res = ethr_mutex_trylock(&tlmt_mtx1);
ASSERT(res == EBUSY);
}
ethr_mutex_lock(&tlmt_mtx2);
tlmt_mtx1_do_unlock = 1;
ethr_cond_signal(&tlmt_cnd2);
while (tlmt_mtx1_locked) {
res = ethr_cond_wait(&tlmt_cnd2, &tlmt_mtx2);
ASSERT(res == 0 || res == EINTR);
}
ethr_mutex_unlock(&tlmt_mtx2);
res = ethr_mutex_trylock(&tlmt_mtx1);
ASSERT(res == 0);
ethr_mutex_unlock(&tlmt_mtx1);
res = ethr_thr_join(tid, NULL);
ASSERT(res == 0);
res = ethr_mutex_destroy(&tlmt_mtx1);
ASSERT(res == 0);
res = ethr_mutex_destroy(&tlmt_mtx2);
ASSERT(res == 0);
res = ethr_cond_destroy(&tlmt_cnd2);
ASSERT(res == 0);
}
static void *
tlmt_thread(void *unused)
{
int res;
ethr_mutex_lock(&tlmt_mtx1);
ethr_mutex_lock(&tlmt_mtx2);
tlmt_mtx1_locked = 1;
ethr_cond_signal(&tlmt_cnd2);
while (!tlmt_mtx1_do_unlock) {
res = ethr_cond_wait(&tlmt_cnd2, &tlmt_mtx2);
ASSERT(res == 0 || res == EINTR);
}
ethr_mutex_unlock(&tlmt_mtx2);
ethr_mutex_unlock(&tlmt_mtx1);
ethr_mutex_lock(&tlmt_mtx2);
tlmt_mtx1_locked = 0;
ethr_cond_signal(&tlmt_cnd2);
ethr_mutex_unlock(&tlmt_mtx2);
return NULL;
}
static void *
bct_thread(void *unused)
{
int res;
ethr_mutex_lock(&bct_mutex);
while (!bct_done) {
bct_waiting++;
if (bct_waiting == BCT_THREADS)
ethr_cond_signal(&bct_cntrl_cond);
do {
res = ethr_cond_wait(&bct_cond, &bct_mutex);
} while (res == EINTR);
ASSERT(res == 0);
bct_woken++;
if (bct_woken == BCT_THREADS)
ethr_cond_signal(&bct_cntrl_cond);
}
ethr_mutex_unlock(&bct_mutex);
return NULL;
}
void
erl_drv_mutex_unlock(ErlDrvMutex *dmtx)
{
#ifdef USE_THREADS
if (!dmtx)
fatal_error(EINVAL, "erl_drv_mutex_unlock()");
ethr_mutex_unlock(&dmtx->mtx);
#endif
}
void
erl_drv_mutex_unlock(ErlDrvMutex *dmtx)
{
#ifdef USE_THREADS
int res = dmtx ? ethr_mutex_unlock(&dmtx->mtx) : EINVAL;
if (res != 0)
fatal_error(res, "erl_drv_mutex_unlock()");
#endif
}
static int
mtt_create_join_threads(void)
{
int no_tids = 100, ix = 0, res = 0, no_threads;
ethr_tid *tids;
mtt_terminate = 0;
tids = (ethr_tid *) malloc(sizeof(ethr_tid)*no_tids);
ASSERT(tids);
print_line("Beginning to create threads");
while (1) {
if (ix >= no_tids) {
no_tids += 100;
if (no_tids > MTT_HARD_LIMIT) {
print_line("Hit the hard limit on number of threads (%d)!",
MTT_HARD_LIMIT);
break;
}
tids = (ethr_tid *) realloc((void *)tids, sizeof(ethr_tid)*no_tids);
ASSERT(tids);
}
res = ethr_thr_create(&tids[ix], mtt_thread, NULL, NULL);
if (res != 0) {
break;
}
ix++;
}
no_threads = ix;
print_line("%d = ethr_thr_create()", res);
print_line("Number of created threads: %d", no_threads);
ethr_mutex_lock(&mtt_mutex);
mtt_terminate = 1;
ethr_cond_broadcast(&mtt_cond);
ethr_mutex_unlock(&mtt_mutex);
while (ix) {
res = ethr_thr_join(tids[--ix], NULL);
ASSERT(res == 0);
}
print_line("All created threads terminated");
free((void *) tids);
return no_threads;
}
static void *
ett_thread2(void *unused)
{
int res;
ethr_mutex_lock(&ett_mutex);
while (!ett_terminate) {
res = ethr_cond_wait(&ett_cond, &ett_mutex);
ASSERT(res == 0);
}
ethr_mutex_unlock(&ett_mutex);
return NULL;
}
static void *
dt_thread(void *unused)
{
ethr_mutex_lock(&dt_mutex);
dt_count++;
if (dt_count >= dt_limit)
ethr_cond_signal(&dt_cond);
ethr_mutex_unlock(&dt_mutex);
return NULL;
}
void *mtt_thread(void *unused)
{
int res;
ethr_mutex_lock(&mtt_mutex);
while (!mtt_terminate) {
res = ethr_cond_wait(&mtt_cond, &mtt_mutex);
ASSERT(res == 0 || res == EINTR);
}
ethr_mutex_unlock(&mtt_mutex);
return NULL;
}
void *
cwt_thread(void *unused)
{
int res;
ethr_mutex_lock(&cwt_mutex);
do {
res = ethr_cond_wait(&cwt_cond, &cwt_mutex);
} while (res == EINTR);
ASSERT(res == 0);
cwt_counter++;
ethr_mutex_unlock(&cwt_mutex);
return NULL;
}
void *
mt_thread(void *unused)
{
print_line("Aux thread tries to lock mutex");
ethr_mutex_lock(&mt_mutex);
print_line("Aux thread locked mutex");
ASSERT(mt_data == 0);
mt_data = 1;
print_line("Aux thread wrote");
print_line("Aux thread goes to sleep for 1 second");
do_sleep(1);
print_line("Aux thread woke up");
ASSERT(mt_data == 1);
ethr_mutex_unlock(&mt_mutex);
print_line("Aux thread unlocked mutex");
return NULL;
}
static void
detached_thread_test(void)
{
ethr_thr_opts thr_opts = ETHR_THR_OPTS_DEFAULT_INITER;
ethr_tid tid[DT_BATCH_SIZE];
int i, j, res;
res = ethr_mutex_init(&dt_mutex);
ASSERT(res == 0);
res = ethr_cond_init(&dt_cond);
ASSERT(res == 0);
thr_opts.detached = 1;
dt_count = 0;
dt_limit = 0;
for (i = 0; i < DT_THREADS/DT_BATCH_SIZE; i++) {
dt_limit += DT_BATCH_SIZE;
for (j = 0; j < DT_BATCH_SIZE; j++) {
res = ethr_thr_create(&tid[j], dt_thread, NULL, &thr_opts);
ASSERT(res == 0);
}
ethr_mutex_lock(&dt_mutex);
while (dt_count < dt_limit) {
res = ethr_cond_wait(&dt_cond, &dt_mutex);
ASSERT(res == 0 || res == EINTR);
}
ethr_mutex_unlock(&dt_mutex);
print_line("dt_count = %d", dt_count);
}
do_sleep(1);
}
static void
cond_wait_test(void)
{
ethr_tid tid1, tid2;
int res;
res = ethr_mutex_init(&cwt_mutex);
ASSERT(res == 0);
res = ethr_cond_init(&cwt_cond);
ASSERT(res == 0);
/* Wake with signal */
cwt_counter = 0;
res = ethr_thr_create(&tid1, cwt_thread, NULL, NULL);
ASSERT(res == 0);
res = ethr_thr_create(&tid2, cwt_thread, NULL, NULL);
ASSERT(res == 0);
do_sleep(1); /* Make sure threads waits on cond var */
ethr_mutex_lock(&cwt_mutex);
ethr_cond_signal(&cwt_cond); /* Wake one thread */
do_sleep(1); /* Make sure awakened thread waits on mutex */
ASSERT(cwt_counter == 0);
ethr_mutex_unlock(&cwt_mutex);
do_sleep(1); /* Let awakened thread proceed */
ethr_mutex_lock(&cwt_mutex);
ASSERT(cwt_counter == 1);
ethr_cond_signal(&cwt_cond); /* Wake the other thread */
do_sleep(1); /* Make sure awakened thread waits on mutex */
ASSERT(cwt_counter == 1);
ethr_mutex_unlock(&cwt_mutex);
do_sleep(1); /* Let awakened thread proceed */
ethr_mutex_lock(&cwt_mutex);
ASSERT(cwt_counter == 2);
ethr_mutex_unlock(&cwt_mutex);
res = ethr_thr_join(tid1, NULL);
ASSERT(res == 0);
res = ethr_thr_join(tid2, NULL);
ASSERT(res == 0);
/* Wake with broadcast */
cwt_counter = 0;
res = ethr_thr_create(&tid1, cwt_thread, NULL, NULL);
ASSERT(res == 0);
res = ethr_thr_create(&tid2, cwt_thread, NULL, NULL);
ASSERT(res == 0);
do_sleep(1); /* Make sure threads waits on cond var */
ethr_mutex_lock(&cwt_mutex);
ethr_cond_broadcast(&cwt_cond); /* Wake the threads */
do_sleep(1); /* Make sure awakened threads wait on mutex */
ASSERT(cwt_counter == 0);
ethr_mutex_unlock(&cwt_mutex);
do_sleep(1); /* Let awakened threads proceed */
ethr_mutex_lock(&cwt_mutex);
ASSERT(cwt_counter == 2);
ethr_mutex_unlock(&cwt_mutex);
res = ethr_thr_join(tid1, NULL);
ASSERT(res == 0);
res = ethr_thr_join(tid2, NULL);
ASSERT(res == 0);
res = ethr_mutex_destroy(&cwt_mutex);
ASSERT(res == 0);
res = ethr_cond_destroy(&cwt_cond);
ASSERT(res == 0);
}
static ERTS_INLINE void lcnt_unlock(void) {
ethr_mutex_unlock(&lcnt_data_lock);
}
static void
broadcast_test(void)
{
int res, i;
ethr_tid tid[BCT_THREADS];
res = ethr_mutex_init(&bct_mutex);
ASSERT(res == 0);
res = ethr_cond_init(&bct_cntrl_cond);
ASSERT(res == 0);
res = ethr_cond_init(&bct_cond);
ASSERT(res == 0);
for (i = 0; i < BCT_THREADS; i++) {
res = ethr_thr_create(&tid[i], bct_thread, NULL, NULL);
ASSERT(res == 0);
}
ethr_mutex_lock(&bct_mutex);
for (i = 0; i < BCT_NO_OF_WAITS; i++) {
while (bct_waiting != BCT_THREADS) {
res = ethr_cond_wait(&bct_cntrl_cond, &bct_mutex);
ASSERT(res == 0 || res == EINTR);
}
bct_waiting = 0;
bct_woken = 0;
/* Wake all threads */
ethr_cond_broadcast(&bct_cond);
while (bct_woken != BCT_THREADS) {
res = ethr_cond_wait(&bct_cntrl_cond, &bct_mutex);
ASSERT(res == 0 || res == EINTR);
}
}
bct_done = 1;
/* Wake all threads */
ethr_cond_broadcast(&bct_cond);
ethr_mutex_unlock(&bct_mutex);
for (i = 0; i < BCT_THREADS; i++) {
res = ethr_thr_join(tid[i], NULL);
ASSERT(res == 0);
}
res = ethr_mutex_destroy(&bct_mutex);
ASSERT(res == 0);
res = ethr_cond_destroy(&bct_cntrl_cond);
ASSERT(res == 0);
res = ethr_cond_destroy(&bct_cond);
ASSERT(res == 0);
}
static void
equal_tids_test(void)
{
int res, i;
res = ethr_mutex_init(&ett_mutex);
ASSERT(res == 0);
res = ethr_cond_init(&ett_cond);
ASSERT(res == 0);
ett_tids[0] = ethr_self();
res = ethr_thr_create(&ett_tids[1], ett_thread, (void *) &ett_tids[1], NULL);
ASSERT(res == 0);
ASSERT(ethr_equal_tids(ethr_self(), ett_tids[0]));
ASSERT(!ethr_equal_tids(ethr_self(), ett_tids[1]));
res = ethr_thr_join(ett_tids[1], NULL);
res = ethr_thr_create(&ett_tids[2], ett_thread, (void *) &ett_tids[2], NULL);
ASSERT(res == 0);
ASSERT(ethr_equal_tids(ethr_self(), ett_tids[0]));
ASSERT(!ethr_equal_tids(ethr_self(), ett_tids[1]));
ASSERT(!ethr_equal_tids(ethr_self(), ett_tids[2]));
#if 0
/* This fails on some linux platforms. Until we decides if a tid
* is allowed to be reused right away or not, we disable the test.
*/
ASSERT(!ethr_equal_tids(ett_tids[1], ett_tids[2]));
#endif
res = ethr_thr_join(ett_tids[2], NULL);
ASSERT(res == 0);
/* Second part of test */
ett_terminate = 0;
res = ethr_thr_create(&ett_tids[1], ett_thread2, NULL, NULL);
ASSERT(res == 0);
ASSERT(!ethr_equal_tids(ett_tids[0], ett_tids[1]));
for (i = 0; i < ETT_THREADS; i++) {
res = ethr_thr_create(&ett_tids[2], ett_thread, (void*)&ett_tids[2], NULL);
ASSERT(res == 0);
ASSERT(!ethr_equal_tids(ett_tids[0], ett_tids[2]));
ASSERT(!ethr_equal_tids(ett_tids[1], ett_tids[2]));
res = ethr_thr_join(ett_tids[2], NULL);
ASSERT(res == 0);
}
ethr_mutex_lock(&ett_mutex);
ett_terminate = 1;
ethr_cond_signal(&ett_cond);
ethr_mutex_unlock(&ett_mutex);
res = ethr_thr_join(ett_tids[1], NULL);
ASSERT(res == 0);
res = ethr_cond_destroy(&ett_cond);
ASSERT(res == 0);
res = ethr_mutex_destroy(&ett_mutex);
ASSERT(res == 0);
}
static void
mutex_test(void)
{
int res;
ethr_tid tid;
print_line("Trying to initialize mutex");
res = ethr_mutex_init(&mt_mutex);
ASSERT(res == 0);
print_line("Initialized mutex");
mt_data = 0;
print_line("Main thread tries to lock mutex");
ethr_mutex_lock(&mt_mutex);
print_line("Main thread locked mutex");
ASSERT(mt_data == 0);
print_line("Main thread about to create aux thread");
res = ethr_thr_create(&tid, mt_thread, NULL, NULL);
ASSERT(res == 0);
print_line("Main thread created aux thread");
print_line("Main thread goes to sleep for 1 second");
do_sleep(1);
print_line("Main thread woke up");
ASSERT(mt_data == 0);
ethr_mutex_unlock(&mt_mutex);
print_line("Main thread unlocked mutex");
print_line("Main thread goes to sleep for 1 second");
do_sleep(1);
print_line("Main thread woke up");
print_line("Main thread tries to lock mutex");
ethr_mutex_lock(&mt_mutex);
print_line("Main thread locked mutex");
ASSERT(mt_data == 1);
print_line("Main thread goes to sleep for 1 second");
do_sleep(1);
print_line("Main thread woke up");
ASSERT(mt_data == 1);
ethr_mutex_unlock(&mt_mutex);
print_line("Main thread unlocked mutex");
res = ethr_thr_join(tid, NULL);
ASSERT(res == 0);
print_line("Main thread joined aux thread");
res = ethr_mutex_destroy(&mt_mutex);
ASSERT(res == 0);
print_line("Main thread destroyed mutex");
}
