From f0bf079dca01acb6dfc07af03226a0cae7a90010 Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Sun, 1 Jan 2017 13:32:32 -0800 Subject: New thread infrastructure -- not used anywhere yet, but tested. --- src/core/thread.c | 150 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 150 insertions(+) create mode 100644 src/core/thread.c (limited to 'src/core/thread.c') diff --git a/src/core/thread.c b/src/core/thread.c new file mode 100644 index 00000000..4f714341 --- /dev/null +++ b/src/core/thread.c @@ -0,0 +1,150 @@ +// +// Copyright 2016 Garrett D'Amore +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include "core/nng_impl.h" + +int +nni_mtx_init(nni_mtx *mtx) +{ + return (nni_plat_mtx_init(&mtx->mtx)); +} + +void +nni_mtx_fini(nni_mtx *mtx) +{ + nni_plat_mtx_fini(&mtx->mtx); +} + +void +nni_mtx_lock(nni_mtx *mtx) +{ + nni_plat_mtx_lock(&mtx->mtx); +} + +void +nni_mtx_unlock(nni_mtx *mtx) +{ + nni_plat_mtx_unlock(&mtx->mtx); +} + +int +nni_mtx_trylock(nni_mtx *mtx) +{ + return (nni_plat_mtx_trylock(&mtx->mtx)); +} + +int +nni_cv_init(nni_cv *cv, nni_mtx *mtx) +{ + return (nni_plat_cv_init(&cv->cv, &mtx->mtx)); +} + +void +nni_cv_fini(nni_cv *cv) +{ + nni_plat_cv_fini(&cv->cv); +} + +void +nni_cv_wait(nni_cv *cv) +{ + nni_plat_cv_wait(&cv->cv); +} + +int +nni_cv_until(nni_cv *cv, nni_time until) +{ + // Some special cases for times. Catching these here means that + // platforms can assume a valid time is presented to them. + if (until == NNI_TIME_NEVER) { + nni_plat_cv_wait(&cv->cv); + return (0); + } + if (until == NNI_TIME_ZERO) { + return (NNG_EAGAIN); + } + + return (nni_plat_cv_until(&cv->cv, until)); +} + +void +nni_cv_wake(nni_cv *cv) +{ + return (nni_plat_cv_wake(&cv->cv)); +} + +static void +nni_thr_wrap(void *arg) +{ + nni_thr *thr = arg; + int stop; + + nni_plat_mtx_lock(&thr->mtx); + while (((stop = thr->stop) == 0) && (thr->start == 0)) { + nni_plat_cv_wait(&thr->cv); + } + nni_plat_mtx_unlock(&thr->mtx); + if (!stop) { + thr->fn(thr->arg); + } + nni_plat_mtx_lock(&thr->mtx); + thr->done = 1; + nni_plat_cv_wake(&thr->cv); + nni_plat_mtx_unlock(&thr->mtx); +} + +int +nni_thr_init(nni_thr *thr, nni_thr_func fn, void *arg) +{ + int rv; + + thr->done = 0; + thr->start = 0; + thr->stop = 0; + thr->fn = fn; + thr->arg = arg; + + if ((rv = nni_plat_mtx_init(&thr->mtx)) != 0) { + return (rv); + } + if ((rv = nni_plat_cv_init(&thr->cv, &thr->mtx)) != 0) { + nni_plat_mtx_fini(&thr->mtx); + return (rv); + } + if ((rv = nni_plat_thr_init(&thr->thr, nni_thr_wrap, thr)) != 0) { + nni_plat_cv_fini(&thr->cv); + nni_plat_mtx_fini(&thr->mtx); + return (rv); + } + return (0); +} + +void +nni_thr_run(nni_thr *thr) +{ + nni_plat_mtx_lock(&thr->mtx); + thr->start = 1; + nni_plat_cv_wake(&thr->cv); + nni_plat_mtx_unlock(&thr->mtx); +} + +void +nni_thr_fini(nni_thr *thr) +{ + nni_plat_mtx_lock(&thr->mtx); + thr->stop = 1; + nni_plat_cv_wake(&thr->cv); + while (!thr->done) { + nni_plat_cv_wait(&thr->cv); + } + nni_plat_mtx_unlock(&thr->mtx); + nni_plat_thr_fini(&thr->thr); + nni_plat_cv_fini(&thr->cv); + nni_plat_mtx_fini(&thr->mtx); +} \ No newline at end of file -- cgit v1.2.3-70-g09d2