track all live threads in an AS-safe, fully-consistent linked list

[musl] / src / thread / pthread_join.c

diff --git a/src/thread/pthread_join.c b/src/thread/pthread_join.c
index 86191f2..b8813e0 100644 (file)
--- a/src/thread/pthread_join.c
+++ b/src/thread/pthread_join.c
@@ -1,14 +1,39 @@
 #include "pthread_impl.h"
+#include <sys/mman.h>
 
-static void dummy(void *p)
+static void dummy1(pthread_t t)
 {
 }
+weak_alias(dummy1, __tl_sync);
 
-int pthread_join(pthread_t t, void **res)
+static int __pthread_timedjoin_np(pthread_t t, void **res, const struct timespec *at)
 {
-       int tmp;
-       while ((tmp = t->tid)) __timedwait(&t->tid, tmp, 0, 0, dummy, 0, 0);
+       int state, cs, r = 0;
+       __pthread_testcancel();
+       __pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
+       if (cs == PTHREAD_CANCEL_ENABLE) __pthread_setcancelstate(cs, 0);
+       while ((state = t->detach_state) && r != ETIMEDOUT && r != EINVAL) {
+               if (state >= DT_DETACHED) a_crash();
+               r = __timedwait_cp(&t->detach_state, state, CLOCK_REALTIME, at, 1);
+       }
+       __pthread_setcancelstate(cs, 0);
+       if (r == ETIMEDOUT || r == EINVAL) return r;
+       __tl_sync(t);
        if (res) *res = t->result;
-       if (t->map_base) munmap(t->map_base, t->map_size);
+       if (t->map_base) __munmap(t->map_base, t->map_size);
        return 0;
 }
+
+int __pthread_join(pthread_t t, void **res)
+{
+       return __pthread_timedjoin_np(t, res, 0);
+}
+
+static int __pthread_tryjoin_np(pthread_t t, void **res)
+{
+       return t->detach_state==DT_JOINABLE ? EBUSY : __pthread_join(t, res);
+}
+
+weak_alias(__pthread_tryjoin_np, pthread_tryjoin_np);
+weak_alias(__pthread_timedjoin_np, pthread_timedjoin_np);
+weak_alias(__pthread_join, pthread_join);