overhaul cancellation to fix resource leaks and dangerous behavior with signals

[musl] / src / process / fork.c

diff --git a/src/process/fork.c b/src/process/fork.c
index 1213f0f..012b7ca 100644 (file)
--- a/src/process/fork.c
+++ b/src/process/fork.c
@@ -1,9 +1,19 @@
 #include <unistd.h>
 #include "syscall.h"
-
-/* FIXME: add support for atfork stupidity */
+#include "libc.h"
+#include "pthread_impl.h"
 
 pid_t fork(void)
 {
-       return syscall0(__NR_fork);
+       pid_t ret;
+       if (libc.fork_handler) libc.fork_handler(-1);
+       ret = syscall(SYS_fork);
+       if (libc.lock && !ret) {
+               pthread_t self = __pthread_self();
+               self->pid = syscall(SYS_getpid);
+               self->tid = syscall(SYS_gettid);
+               libc.threads_minus_1 = 0;
+       }
+       if (libc.fork_handler) libc.fork_handler(!ret);
+       return ret;
 }