libpthread-137.1.1.tar.gz

[apple/libpthread.git] / tests / atfork.c

diff --git a/tests/atfork.c b/tests/atfork.c
new file mode 100644 (file)
index 0000000..7b760c0
--- /dev/null
+++ b/tests/atfork.c
@@ -0,0 +1,75 @@
+#include <errno.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <os/assumes.h>
+#include <sys/wait.h>
+
+#define DECL_ATFORK(x) \
+static void prepare_##x(void) { \
+       printf("%d: %s\n", getpid(), __FUNCTION__); \
+} \
+static void parent_##x(void) { \
+       printf("%d: %s\n", getpid(), __FUNCTION__); \
+} \
+static void child_##x(void) { \
+       printf("%d: %s\n", getpid(), __FUNCTION__); \
+}
+
+#define ATFORK(x) \
+os_assumes_zero(pthread_atfork(prepare_##x, parent_##x, child_##x));
+
+DECL_ATFORK(1);
+DECL_ATFORK(2);
+DECL_ATFORK(3);
+DECL_ATFORK(4);
+DECL_ATFORK(5);
+DECL_ATFORK(6);
+DECL_ATFORK(7);
+DECL_ATFORK(8);
+DECL_ATFORK(9);
+DECL_ATFORK(10);
+DECL_ATFORK(11);
+DECL_ATFORK(12);
+DECL_ATFORK(13);
+DECL_ATFORK(14);
+DECL_ATFORK(15);
+DECL_ATFORK(16);
+DECL_ATFORK(17);
+DECL_ATFORK(18);
+DECL_ATFORK(19);
+
+int main(int argc, char *argv[]) {
+       ATFORK(1);
+       ATFORK(2);
+       ATFORK(3);
+       ATFORK(4);
+       ATFORK(5);
+       ATFORK(6);
+       ATFORK(7);
+       ATFORK(8);
+       ATFORK(9);
+       ATFORK(10);
+       ATFORK(11);
+       ATFORK(12);
+       ATFORK(13);
+       ATFORK(14);
+       ATFORK(15);
+       ATFORK(16);
+       ATFORK(17);
+       ATFORK(18);
+       ATFORK(19);
+
+       pid_t pid = fork();
+       if (pid == 0) {
+               pid = fork(); 
+       }
+       if (pid == -1) {
+               posix_assumes_zero(pid);
+       } else if (pid > 0) {
+               int status;
+               posix_assumes_zero(waitpid(pid, &status, 0));
+               posix_assumes_zero(WEXITSTATUS(status));
+       }
+       return 0;
+}