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

[oweals/musl.git] / src / process / fork.c

diff --git a/src/process/fork.c b/src/process/fork.c
index a8bdbe0b03833cd763a0ad661de90d6d8d295dd2..11286ef44e3bd5fd1069087f8741e5e0ff3fdcf0 100644 (file)
--- a/src/process/fork.c
+++ b/src/process/fork.c
@@ -1,4 +1,6 @@
 #include <unistd.h>
+#include <string.h>
+#include <signal.h>
 #include "syscall.h"
 #include "libc.h"
 #include "pthread_impl.h"
@@ -12,15 +14,23 @@ weak_alias(dummy, __fork_handler);
 pid_t fork(void)
 {
        pid_t ret;
+       sigset_t set;
        __fork_handler(-1);
-       ret = syscall(SYS_fork);
-       if (libc.main_thread && !ret) {
+       __block_all_sigs(&set);
+#ifdef SYS_fork
+       ret = __syscall(SYS_fork);
+#else
+       ret = __syscall(SYS_clone, SIGCHLD, 0);
+#endif
+       if (!ret) {
                pthread_t self = __pthread_self();
-               self->tid = self->pid = syscall(SYS_getpid);
-               memset(&self->robust_list, 0, sizeof self->robust_list);
+               self->tid = __syscall(SYS_gettid);
+               self->robust_list.off = 0;
+               self->robust_list.pending = 0;
+               self->next = self->prev = self;
                libc.threads_minus_1 = 0;
-               libc.main_thread = self;
        }
+       __restore_sigs(&set);
        __fork_handler(!ret);
-       return ret;
+       return __syscall_ret(ret);
 }