block all signals during rsyscall
[musl] / src / thread / pthread_create.c
index d01be55..cdf36ff 100644 (file)
@@ -1,60 +1,65 @@
 #include "pthread_impl.h"
 
+static void dummy_1(pthread_t self)
+{
+}
+weak_alias(dummy_1, __pthread_tsd_run_dtors);
+weak_alias(dummy_1, __sigtimer_handler);
+
+#ifdef __pthread_unwind_next
+#undef __pthread_unwind_next
+#define __pthread_unwind_next __pthread_unwind_next_3
+#endif
+
 void __pthread_unwind_next(struct __ptcb *cb)
 {
-       int i, j, not_finished;
        pthread_t self;
 
        if (cb->__next) longjmp((void *)cb->__next->__jb, 1);
 
        self = pthread_self();
-       if (self->cancel) self->result = PTHREAD_CANCELLED;
-
-       if (!a_fetch_add(&libc.threads_minus_1, -1))
-               exit(0);
 
        LOCK(&self->exitlock);
 
-       not_finished = self->tsd_used;
-       for (j=0; not_finished && j<PTHREAD_DESTRUCTOR_ITERATIONS; j++) {
-               not_finished = 0;
-               for (i=0; i<PTHREAD_KEYS_MAX; i++) {
-                       if (self->tsd[i] && libc.tsd_keys[i]) {
-                               void *tmp = self->tsd[i];
-                               self->tsd[i] = 0;
-                               libc.tsd_keys[i](tmp);
-                               not_finished = 1;
-                       }
-               }
-       }
+       __pthread_tsd_run_dtors(self);
 
-       if (self->detached && self->map_base)
+       /* Mark this thread dead before decrementing count */
+       self->dead = 1;
+
+       if (!a_fetch_add(&libc.threads_minus_1, -1))
+               exit(0);
+
+       if (self->detached && self->map_base) {
+               syscall(__NR_rt_sigprocmask, SIG_BLOCK, (long)(uint64_t[1]){-1},0,8);
                __unmapself(self->map_base, self->map_size);
+       }
 
-       __syscall_exit(0);
+       syscall(SYS_exit, 0);
 }
 
 static void docancel(struct pthread *self)
 {
        struct __ptcb cb = { .__next = self->cancelbuf };
+       self->canceldisable = 1;
+       self->cancelasync = 0;
        __pthread_unwind_next(&cb);
 }
 
 static void cancel_handler(int sig, siginfo_t *si, void *ctx)
 {
        struct pthread *self = __pthread_self();
-       self->cancel = 1;
-       if (self->canceldisable || (!self->cancelasync && !self->cancelpoint))
-               return;
-       docancel(self);
+       if (si->si_code == SI_TIMER) __sigtimer_handler(self);
+       if (self->cancel && !self->canceldisable &&
+           (self->cancelasync || (self->cancelpoint==1 && PC_AT_SYS(ctx))))
+               docancel(self);
 }
 
 static void cancelpt(int x)
 {
        struct pthread *self = __pthread_self();
        if (self->canceldisable) return;
-       self->cancelpoint = x;
-       if (self->cancel) docancel(self);
+       if ((self->cancelpoint+=x)==1 && x>=0 && self->cancel)
+               docancel(self);
 }
 
 /* "rsyscall" is a mechanism by which a thread can synchronously force all
@@ -73,9 +78,18 @@ static struct {
 
 static void rsyscall_handler(int sig, siginfo_t *si, void *ctx)
 {
-       if (rs.cnt == libc.threads_minus_1) return;
+       struct pthread *self = __pthread_self();
+
+       if (!rs.hold || rs.cnt == libc.threads_minus_1) return;
+
+       /* Threads which have already decremented themselves from the
+        * thread count must not increment rs.cnt or otherwise act. */
+       if (self->dead) {
+               sigfillset(&((ucontext_t *)ctx)->uc_sigmask);
+               return;
+       }
 
-       if (syscall6(rs.nr, rs.arg[0], rs.arg[1], rs.arg[2],
+       if (syscall(rs.nr, rs.arg[0], rs.arg[1], rs.arg[2],
                rs.arg[3], rs.arg[4], rs.arg[5]) < 0 && !rs.err) rs.err=errno;
 
        a_inc(&rs.cnt);
@@ -103,9 +117,9 @@ static int rsyscall(int nr, long a, long b, long c, long d, long e, long f)
        rs.arg[0] = a; rs.arg[1] = b;
        rs.arg[2] = c; rs.arg[3] = d;
        rs.arg[4] = d; rs.arg[5] = f;
-       rs.hold = 1;
        rs.err = 0;
        rs.cnt = 0;
+       rs.hold = 1;
 
        /* Dispatch signals until all threads respond */
        for (i=libc.threads_minus_1; i; i--)
@@ -124,7 +138,7 @@ static int rsyscall(int nr, long a, long b, long c, long d, long e, long f)
        while((i=rs.cnt)) __wait(&rs.cnt, 0, i, 1);
 
        if (rs.err) errno = rs.err, ret = -1;
-       else ret = syscall6(nr, a, b, c, d, e, f);
+       else ret = syscall(nr, a, b, c, d, e, f);
 
        UNLOCK(&rs.lock);
        return ret;
@@ -134,26 +148,37 @@ static void init_threads()
 {
        struct sigaction sa = { .sa_flags = SA_SIGINFO | SA_RESTART };
        libc.lock = __lock;
+       libc.lockfile = __lockfile;
        libc.cancelpt = cancelpt;
        libc.rsyscall = rsyscall;
+
+       sigfillset(&sa.sa_mask);
+       sa.sa_sigaction = rsyscall_handler;
+       __libc_sigaction(SIGSYSCALL, &sa, 0);
+
+       sigemptyset(&sa.sa_mask);
        sa.sa_sigaction = cancel_handler;
        __libc_sigaction(SIGCANCEL, &sa, 0);
+
        sigaddset(&sa.sa_mask, SIGSYSCALL);
        sigaddset(&sa.sa_mask, SIGCANCEL);
-       sa.sa_sigaction = rsyscall_handler;
-       __libc_sigaction(SIGSYSCALL, &sa, 0);
-       sigprocmask(SIG_UNBLOCK, &sa.sa_mask, 0);
+       __libc_sigprocmask(SIG_UNBLOCK, &sa.sa_mask, 0);
 }
 
 static int start(void *p)
 {
        struct pthread *self = p;
+       if (self->unblock_cancel) {
+               sigset_t set;
+               sigemptyset(&set);
+               sigaddset(&set, SIGCANCEL);
+               __libc_sigprocmask(SIG_UNBLOCK, &set, 0);
+       }
        pthread_exit(self->start(self->start_arg));
        return 0;
 }
 
-#define CLONE_MAGIC 0x7d0f00
-int __clone(int (*)(void *), void *, int, void *, pid_t *, void *, pid_t *);
+int __uniclone(void *, int (*)(), void *);
 
 #define ROUND(x) (((x)+PAGE_SIZE-1)&-PAGE_SIZE)
 
@@ -174,12 +199,12 @@ int pthread_create(pthread_t *res, const pthread_attr_t *attr, void *(*entry)(vo
        if (!init && ++init) init_threads();
 
        if (!attr) attr = &default_attr;
-       guard = ROUND(attr->__guardsize + DEFAULT_GUARD_SIZE);
-       size = guard + ROUND(attr->__stacksize + DEFAULT_STACK_SIZE);
+       guard = ROUND(attr->_a_guardsize + DEFAULT_GUARD_SIZE);
+       size = guard + ROUND(attr->_a_stacksize + DEFAULT_STACK_SIZE);
        size += __pthread_tsd_size;
        map = mmap(0, size, PROT_READ|PROT_WRITE|PROT_EXEC, MAP_PRIVATE|MAP_ANON, -1, 0);
        if (!map) return EAGAIN;
-       mprotect(map, guard, PROT_NONE);
+       if (guard) mprotect(map, guard, PROT_NONE);
 
        tsd = map + size - __pthread_tsd_size;
        new = (void *)(tsd - sizeof *new - PAGE_SIZE%sizeof *new);
@@ -191,8 +216,10 @@ int pthread_create(pthread_t *res, const pthread_attr_t *attr, void *(*entry)(vo
        new->start_arg = arg;
        new->self = new;
        new->tsd = (void *)tsd;
-       new->detached = attr->__detach;
+       new->detached = attr->_a_detach;
        new->attr = *attr;
+       new->unblock_cancel = self->cancel;
+       new->result = PTHREAD_CANCELED;
        memcpy(new->tlsdesc, self->tlsdesc, sizeof new->tlsdesc);
        new->tlsdesc[1] = (uintptr_t)new;
        stack = (void *)((uintptr_t)new-1 & ~(uintptr_t)15);
@@ -203,8 +230,7 @@ int pthread_create(pthread_t *res, const pthread_attr_t *attr, void *(*entry)(vo
        while (rs.lock) __wait(&rs.lock, 0, 1, 1);
 
        a_inc(&libc.threads_minus_1);
-       ret = __clone(start, stack, CLONE_MAGIC, new,
-               &new->tid, &new->tlsdesc, &new->tid);
+       ret = __uniclone(stack, start, new);
 
        a_dec(&rs.blocks);
        if (rs.lock) __wake(&rs.blocks, 1, 1);