+// SPDX-FileCopyrightText: 2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
/*
- * test_urcu_fork.c
- *
* Userspace RCU library - test program (fork)
- *
- * Copyright February 2012 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-#define _GNU_SOURCE
-#include "config.h"
#include <stdio.h>
#include <pthread.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <unistd.h>
#include <stdio.h>
-#include <assert.h>
#include <sched.h>
#include <errno.h>
+#include <urcu/assert.h>
#include <urcu/arch.h>
#include <urcu/tls-compat.h>
#endif
#include <urcu.h>
+#include "tap.h"
+
/* We generate children 3 levels deep */
#define FORK_DEPTH 3
/* Each generation spawns 10 children */
#define NR_FORK 10
+#define NR_TESTS NR_FORK
+
static int fork_generation;
+/*
+ * Only print diagnostic for top level parent process, else the console
+ * has trouble formatting the tap output.
+ */
+#define diag_gen0(...) \
+ do { \
+ if (!fork_generation) \
+ diag(__VA_ARGS__); \
+ } while (0)
+
struct test_node {
int somedata;
struct rcu_head head;
{
struct test_node *node;
- fprintf(stderr, "rcu callback invoked in pid: %d\n",
- (int) getpid());
+ diag_gen0("rcu callback invoked in pid: %d", (int) getpid());
node = caa_container_of(head, struct test_node, head);
free(node);
}
rcu_read_lock();
rcu_read_unlock();
- node = malloc(sizeof(*node));
- assert(node);
+ node = (struct test_node *) malloc(sizeof(*node));
+ urcu_posix_assert(node);
call_rcu(&node->head, cb);
{
pid_t pid;
- fprintf(stderr, "%s parent pid: %d, before fork\n",
+ diag_gen0("%s parent pid: %d, before fork",
execname, (int) getpid());
call_rcu_before_fork();
if (pid == 0) {
/* child */
fork_generation++;
+ tap_disable();
call_rcu_after_fork_child();
- fprintf(stderr, "%s child pid: %d, after fork\n",
+ diag_gen0("%s child pid: %d, after fork",
execname, (int) getpid());
test_rcu();
- fprintf(stderr, "%s child pid: %d, after rcu test\n",
+ diag_gen0("%s child pid: %d, after rcu test",
execname, (int) getpid());
if (fork_generation >= FORK_DEPTH)
exit(EXIT_SUCCESS);
/* parent */
call_rcu_after_fork_parent();
- fprintf(stderr, "%s parent pid: %d, after fork\n",
+ diag_gen0("%s parent pid: %d, after fork",
execname, (int) getpid());
test_rcu();
- fprintf(stderr, "%s parent pid: %d, after rcu test\n",
+ diag_gen0("%s parent pid: %d, after rcu test",
execname, (int) getpid());
for (;;) {
pid = wait(&status);
if (pid < 0) {
- perror("wait");
+ if (!fork_generation)
+ perror("wait");
return -1;
}
if (WIFEXITED(status)) {
- fprintf(stderr, "child %u exited normally with status %u\n",
+ diag_gen0("child %u exited normally with status %u",
pid, WEXITSTATUS(status));
if (WEXITSTATUS(status))
return -1;
break;
} else if (WIFSIGNALED(status)) {
- fprintf(stderr, "child %u was terminated by signal %u\n",
+ diag_gen0("child %u was terminated by signal %u",
pid, WTERMSIG(status));
return -1;
} else {
}
return 1;
} else {
- perror("fork");
+ if (!fork_generation)
+ perror("fork");
return -1;
}
}
-int main(int argc, char **argv)
+int main(int argc __attribute__((unused)), char **argv)
{
unsigned int i;
+ plan_tests(NR_TESTS);
+
#if 0
/* pthread_atfork does not work with malloc/free in callbacks */
ret = pthread_atfork(call_rcu_before_fork,
test_rcu();
synchronize_rcu();
ret = do_fork(argv[0]);
- if (ret == 0) /* child */
+ if (!fork_generation) {
+ ok(ret >= 0, "child status %d", ret);
+ }
+ if (ret == 0) { /* child */
goto restart;
- else if (ret < 0)
+ } else if (ret < 0) {
goto error;
- /* else parent, continue. */
+ } else {
+ /* else parent, continue. */
+ }
+ }
+ if (!fork_generation) {
+ return exit_status();
+ } else {
+ exit(EXIT_SUCCESS);
}
- exit(EXIT_SUCCESS);
error:
- exit(EXIT_FAILURE);
+ if (!fork_generation) {
+ return exit_status();
+ } else {
+ exit(EXIT_FAILURE);
+ }
}