New thread infrastructure -- not used anywhere yet, but tested.

1 files changed, 150 insertions, 0 deletions

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 <garrett@damore.org>
+//
+// 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