Linux Kernel Selftest development
 help / color / mirror / Atom feed
* [PATCH 1/1] tests: add pidfd poll tests
@ 2019-07-23 17:39 Suren Baghdasaryan
  2019-07-24 10:38 ` Christian Brauner
  0 siblings, 1 reply; 3+ messages in thread
From: Suren Baghdasaryan @ 2019-07-23 17:39 UTC (permalink / raw)
  To: surenb
  Cc: christian, arnd, ebiederm, keescook, joel, dancol, tglx, jannh,
	dhowells, mtk.manpages, luto, akpm, oleg, cyphar, torvalds, viro,
	linux-api, linux-kselftest, kernel-team

This adds testing for polling on pidfd of a process being killed. Test runs
10000 iterations by default to stress test pidfd polling functionality.
It accepts an optional command-line parameter to override the number or
iterations to run.
Specifically, it tests for:
- pidfd_open on a child process succeeds
- pidfd_send_signal on a child process succeeds
- polling on pidfd succeeds and returns exactly one event
- returned event is POLLIN
- event is received within 3 secs of the process being killed

10000 iterations was chosen because of the race condition being tested
which is not consistently reproducible but usually is revealed after less
than 2000 iterations.
Reveals race fixed by commit b191d6491be6 ("pidfd: fix a poll race when setting exit_state")

Signed-off-by: Suren Baghdasaryan <surenb@google.com>
---
 tools/testing/selftests/pidfd/.gitignore      |   1 +
 tools/testing/selftests/pidfd/Makefile        |   2 +-
 .../testing/selftests/pidfd/pidfd_poll_test.c | 137 ++++++++++++++++++
 3 files changed, 139 insertions(+), 1 deletion(-)
 create mode 100644 tools/testing/selftests/pidfd/pidfd_poll_test.c

diff --git a/tools/testing/selftests/pidfd/.gitignore b/tools/testing/selftests/pidfd/.gitignore
index 16d84d117bc0..a67896347d34 100644
--- a/tools/testing/selftests/pidfd/.gitignore
+++ b/tools/testing/selftests/pidfd/.gitignore
@@ -1,2 +1,3 @@
 pidfd_open_test
+pidfd_poll_test
 pidfd_test
diff --git a/tools/testing/selftests/pidfd/Makefile b/tools/testing/selftests/pidfd/Makefile
index 720b2d884b3c..ed58b7108d18 100644
--- a/tools/testing/selftests/pidfd/Makefile
+++ b/tools/testing/selftests/pidfd/Makefile
@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: GPL-2.0-only
 CFLAGS += -g -I../../../../usr/include/ -lpthread
 
-TEST_GEN_PROGS := pidfd_test pidfd_open_test
+TEST_GEN_PROGS := pidfd_test pidfd_open_test pidfd_poll_test
 
 include ../lib.mk
 
diff --git a/tools/testing/selftests/pidfd/pidfd_poll_test.c b/tools/testing/selftests/pidfd/pidfd_poll_test.c
new file mode 100644
index 000000000000..f2934aa070ae
--- /dev/null
+++ b/tools/testing/selftests/pidfd/pidfd_poll_test.c
@@ -0,0 +1,137 @@
+// SPDX-License-Identifier: GPL-2.0
+
+#define _GNU_SOURCE
+#include <errno.h>
+#include <linux/types.h>
+#include <linux/wait.h>
+#include <poll.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <syscall.h>
+#include <sys/wait.h>
+#include <unistd.h>
+
+#include "pidfd.h"
+#include "../kselftest.h"
+
+#define __NR_pidfd_send_signal 424
+#define __NR_pidfd_open 434
+
+static inline int sys_pidfd_open(pid_t pid, unsigned int flags)
+{
+	return syscall(__NR_pidfd_open, pid, flags);
+}
+
+static inline int sys_pidfd_send_signal(int pidfd, int sig, siginfo_t *info,
+					unsigned int flags)
+{
+	return syscall(__NR_pidfd_send_signal, pidfd, sig, info, flags);
+}
+
+static bool timeout;
+
+static void handle_alarm(int sig)
+{
+	timeout = true;
+}
+
+int main(int argc, char **argv)
+{
+	int ret = 1;
+	int pidfd = -1;
+	struct pollfd fds;
+	int iter, nevents;
+	int nr_iterations = 10000;
+
+	fds.events = POLLIN;
+	if (argc > 1) {
+		nr_iterations = atoi(argv[1]);
+		if (!nr_iterations) {
+			ksft_test_result_fail("invalid input parameter %s\n",
+				argv[1]);
+			return ksft_exit_fail();
+		}
+	}
+
+	ksft_print_msg("running pidfd poll test for %d iterations\n",
+		nr_iterations);
+
+	for (iter = 0; iter < nr_iterations; iter++) {
+		int child_pid = fork();
+
+		if (!child_pid) {
+			/* Child process just sleeps for a min */
+			sleep(60);
+			exit(0);
+		}
+
+		/* Parent kills the child and waits for its death */
+		if (child_pid < 0) {
+			if (errno == EAGAIN) {
+				iter--;
+				continue;
+			}
+			ksft_print_msg("%s - failed to fork a child process\n",
+				strerror(errno));
+		}
+		pidfd = sys_pidfd_open(child_pid, 0);
+		if (pidfd < 0) {
+			ksft_print_msg("%s - pidfd_open failed\n",
+				strerror(errno));
+			goto on_error;
+		}
+		/* Setup 3 sec alarm - plenty of time */
+		if (signal(SIGALRM, handle_alarm) == SIG_ERR) {
+			ksft_print_msg("%s - signal failed\n",
+				strerror(errno));
+			goto on_error;
+		}
+		alarm(3);
+		/* Send SIGKILL to the child */
+		if (sys_pidfd_send_signal(pidfd, SIGKILL, NULL, 0)) {
+			ksft_print_msg("%s - pidfd_send_signal failed\n",
+				strerror(errno));
+			goto on_error;
+		}
+		/* Wait for the death notification */
+		fds.fd = pidfd;
+		nevents = poll(&fds, 1, -1);
+		if (nevents < 0) {
+			ksft_print_msg("%s - poll failed\n",
+				strerror(errno));
+			goto on_error;
+		}
+		if (nevents != 1) {
+			ksft_print_msg("unexpected poll result: %d\n",
+				nevents);
+			goto on_error;
+		}
+		if (!(fds.revents & POLLIN)) {
+			ksft_print_msg(
+				"unexpected event type received: 0x%x\n",
+				fds.revents);
+			goto on_error;
+		}
+		if (timeout) {
+			ksft_print_msg("death notification wait timeout\n");
+			goto on_error;
+		}
+		close(pidfd);
+	}
+	ret = 0;
+
+on_error:
+	if (pidfd)
+		close(pidfd);
+
+	if (ret) {
+		ksft_test_result_fail("failed after %d retries\n", iter);
+		return ksft_exit_fail();
+	}
+
+	ksft_test_result_pass("pidfd poll test: pass\n");
+	return ksft_exit_pass();
+}
-- 
2.22.0.657.g960e92d24f-goog


^ permalink raw reply related	[flat|nested] 3+ messages in thread

end of thread, other threads:[~2019-07-24 17:04 UTC | newest]

Thread overview: 3+ messages (download: mbox.gz follow: Atom feed
-- links below jump to the message on this page --
2019-07-23 17:39 [PATCH 1/1] tests: add pidfd poll tests Suren Baghdasaryan
2019-07-24 10:38 ` Christian Brauner
2019-07-24 17:03   ` Suren Baghdasaryan

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox