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 1213f0f509e93d5e2f787862ccc6804ab5ab1f46..11286ef44e3bd5fd1069087f8741e5e0ff3fdcf0 100644 (file)
--- a/src/process/fork.c
+++ b/src/process/fork.c
@@ -1,9 +1,36 @@
 #include <unistd.h>
+#include <string.h>
+#include <signal.h>
 #include "syscall.h"
+#include "libc.h"
+#include "pthread_impl.h"
 
-/* FIXME: add support for atfork stupidity */
+static void dummy(int x)
+{
+}
+
+weak_alias(dummy, __fork_handler);
 
 pid_t fork(void)
 {
-       return syscall0(__NR_fork);
+       pid_t ret;
+       sigset_t set;
+       __fork_handler(-1);
+       __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 = __syscall(SYS_gettid);
+               self->robust_list.off = 0;
+               self->robust_list.pending = 0;
+               self->next = self->prev = self;
+               libc.threads_minus_1 = 0;
+       }
+       __restore_sigs(&set);
+       __fork_handler(!ret);
+       return __syscall_ret(ret);
 }