about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--include/unistd.h1
-rw-r--r--src/thread/forkall.c66
2 files changed, 67 insertions, 0 deletions
diff --git a/include/unistd.h b/include/unistd.h
index a64d99b7..35cfda8c 100644
--- a/include/unistd.h
+++ b/include/unistd.h
@@ -147,6 +147,7 @@ int getdtablesize(void);
 #ifdef _GNU_SOURCE
 int brk(void *);
 void *sbrk(intptr_t);
+pid_t forkall(void);
 pid_t vfork(void);
 int vhangup(void);
 int chroot(const char *);
diff --git a/src/thread/forkall.c b/src/thread/forkall.c
new file mode 100644
index 00000000..403818ec
--- /dev/null
+++ b/src/thread/forkall.c
@@ -0,0 +1,66 @@
+#include "pthread_impl.h"
+#include <setjmp.h>
+
+struct thread {
+	struct thread *next;
+	pthread_t td;
+	jmp_buf jb;
+	void *tmp, *stack;
+};
+
+struct ctx {
+	struct thread *list;
+	pthread_t caller;
+	pid_t pid;
+	size_t cnt;
+	pthread_barrier_t barrier;
+};
+
+static void restart_thread(pthread_t self)
+{
+	struct thread *t = self->start_arg;
+	self->start_arg = t->tmp;
+	self->pid = getpid();
+	longjmp(t->jb, 1);
+}
+
+static void do_forkall(void *p)
+{
+	struct ctx *c = p, *volatile cv = c;
+	char tmpstack[2048];
+	struct thread *tp, t = {
+		.td = __pthread_self(),
+		.next = c->list,
+		.stack = tmpstack+1024
+	};
+
+	if (t.td != c->caller) {
+		c->cnt++;
+		t.tmp = t.td->start_arg;
+		t.td->start_arg = &t;
+		if (setjmp(t.jb)) {
+			c = cv;
+			if (c->pid) return;
+			pthread_barrier_wait(&c->barrier);
+			return;
+		}
+		c->list = &t;
+		__synccall_wait();
+		return;
+	}
+	c->pid = syscall(SYS_fork);
+	if (c->pid) return;
+
+	pthread_barrier_init(&c->barrier, 0, c->cnt);
+	for (tp=c->list; tp; tp=tp->next)
+		if (__uniclone(tp->stack, restart_thread, tp->td) < 0)
+			_exit(127);
+	pthread_barrier_wait(&c->barrier);
+}
+
+pid_t forkall()
+{
+	struct ctx c = { .caller = pthread_self() };
+	__synccall(do_forkall, &c);
+	return c.pid;
+}