[CRIU] [PATCH 5/5] test: add a test for seccomp filter inheritance

[Tycho Andersen]

v2:
 * add another filter to the chain to test the inheritance chaining check
 * include zdtm.py desc file

Signed-off-by: Tycho Andersen <tycho.andersen at canonical.com>
---
 test/zdtm.sh                                       |   2 +
 test/zdtm/.gitignore                               |   1 +
 test/zdtm/live/static/Makefile                     |   1 +
 test/zdtm/live/static/seccomp_filter_inheritance.c | 177 +++++++++++++++++++++
 .../live/static/seccomp_filter_inheritance.desc    |   1 +
 5 files changed, 182 insertions(+)
 create mode 100644 test/zdtm/live/static/seccomp_filter_inheritance.c
 create mode 100644 test/zdtm/live/static/seccomp_filter_inheritance.desc

diff --git a/test/zdtm.sh b/test/zdtm.sh
index dea5218..81aacff 100755
--- a/test/zdtm.sh
+++ b/test/zdtm.sh
@@ -248,6 +248,7 @@ generate_test_list()
 	TEST_SECCOMP_FILTERS="
 		static/seccomp_filter
 		static/seccomp_filter_tsync
+		static/seccomp_filter_inheritance
 	"
 
 	$CRIU check -v0 --feature "mnt_id"
@@ -370,6 +371,7 @@ apparmor
 seccomp_strict
 seccomp_filter
 seccomp_filter_tsync
+seccomp_filter_inheritance
 different_creds
 inotify01
 ipc_namespace
diff --git a/test/zdtm/.gitignore b/test/zdtm/.gitignore
index ac853d7..166c26d 100644
--- a/test/zdtm/.gitignore
+++ b/test/zdtm/.gitignore
@@ -113,6 +113,7 @@
 /live/static/sched_prio00
 /live/static/seccomp_filter
 /live/static/seccomp_filter_tsync
+/live/static/seccomp_filter_inheritance
 /live/static/seccomp_strict
 /live/static/selfexe00
 /live/static/sem
diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index e2870dd..1747f81 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -130,6 +130,7 @@ TST_NOFILE	=				\
 		seccomp_strict			\
 		seccomp_filter			\
 		seccomp_filter_tsync			\
+		seccomp_filter_inheritance		\
 		different_creds			\
 		vsx				\
 		bridge				\
diff --git a/test/zdtm/live/static/seccomp_filter_inheritance.c b/test/zdtm/live/static/seccomp_filter_inheritance.c
new file mode 100644
index 0000000..e35c3e3
--- /dev/null
+++ b/test/zdtm/live/static/seccomp_filter_inheritance.c
@@ -0,0 +1,177 @@
+#include <unistd.h>
+#include <stdbool.h>
+#include <signal.h>
+#include <stddef.h>
+#include <sys/prctl.h>
+#include <sys/ptrace.h>
+#include <linux/seccomp.h>
+#include <linux/filter.h>
+#include <linux/limits.h>
+#include <linux/bpf.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/wait.h>
+#include <sys/syscall.h>
+#include "zdtmtst.h"
+
+const char *test_doc	= "Check that SECCOMP_MODE_FILTER is restored";
+const char *test_author	= "Tycho Andersen <tycho.andersen at canonical.com>";
+
+int get_seccomp_mode(pid_t pid)
+{
+	FILE *f;
+	char buf[PATH_MAX];
+
+	sprintf(buf, "/proc/%d/status", pid);
+	f = fopen(buf, "r+");
+	if (!f) {
+		pr_perror("fopen failed");
+		return -1;
+	}
+
+	while (NULL != fgets(buf, sizeof(buf), f)) {
+		int mode;
+
+		if (sscanf(buf, "Seccomp:\t%d", &mode) != 1)
+			continue;
+
+		fclose(f);
+		return mode;
+	}
+	fclose(f);
+
+	return -1;
+}
+
+int filter_syscall(int syscall_nr)
+{
+	struct sock_filter filter[] = {
+		BPF_STMT(BPF_LD+BPF_W+BPF_ABS, offsetof(struct seccomp_data, nr)),
+		BPF_JUMP(BPF_JMP+BPF_JEQ+BPF_K, syscall_nr, 0, 1),
+		BPF_STMT(BPF_RET+BPF_K, SECCOMP_RET_KILL),
+		BPF_STMT(BPF_RET+BPF_K, SECCOMP_RET_ALLOW),
+	};
+
+	struct sock_fprog bpf_prog = {
+		.len = (unsigned short)(sizeof(filter)/sizeof(filter[0])),
+		.filter = filter,
+	};
+
+	if (prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &bpf_prog) < 0) {
+		pr_perror("prctl failed");
+		return -1;
+	}
+
+	return 0;
+}
+
+int main(int argc, char ** argv)
+{
+	pid_t pid;
+	int mode, status;
+	int sk_pair[2], sk, ret;
+	char c = 'K';
+
+	test_init(argc, argv);
+
+	if (socketpair(PF_LOCAL, SOCK_SEQPACKET, 0, sk_pair)) {
+		pr_perror("socketpair");
+		return -1;
+	}
+
+	pid = fork();
+	if (pid < 0) {
+		pr_perror("fork");
+		return -1;
+	}
+
+	if (pid == 0) {
+
+		pid_t pid2;
+
+		sk = sk_pair[1];
+		close(sk_pair[0]);
+
+		if (filter_syscall(__NR_ptrace) < 0)
+			_exit(1);
+
+		if (filter_syscall(__NR_fstat) < 0)
+			_exit(1);
+
+		zdtm_seccomp = 1;
+		test_msg("SECCOMP_MODE_FILTER is enabled\n");
+
+		pid2 = fork();
+		if (pid2 < 0)
+			_exit(1);
+
+		if (!pid2) {
+
+			if (write(sk, &c, 1) != 1) {
+				pr_perror("write");
+				_exit(1);
+			}
+
+			if (read(sk, &c, 1) != 1) {
+				pr_perror("read");
+				_exit(1);
+			}
+
+			/* We expect to be killed by our policy above. */
+			ptrace(PTRACE_TRACEME);
+			_exit(1);
+		}
+
+		if (waitpid(pid2, &status, 0) != pid2) {
+			pr_perror("waitpid");
+			_exit(1);
+		}
+
+		if (WTERMSIG(status) != SIGSYS) {
+			pr_perror("expected SIGSYS, got %d\n", WTERMSIG(status));
+			_exit(1);
+		}
+
+		_exit(0);
+	}
+
+	sk = sk_pair[0];
+	close(sk_pair[1]);
+
+	if ((ret = read(sk, &c, 1)) != 1) {
+		pr_perror("read %d", ret);
+		goto err;
+	}
+
+	test_daemon();
+	test_waitsig();
+
+	mode = get_seccomp_mode(pid);
+	if (write(sk, &c, 1) != 1) {
+		pr_perror("write");
+		goto err;
+	}
+
+	if (mode != SECCOMP_MODE_FILTER) {
+		fail("seccomp mode mismatch %d\n", mode);
+		return 1;
+	}
+
+	if (waitpid(pid, &status, 0) != pid) {
+		pr_perror("waitpid");
+		_exit(1);
+	}
+
+	if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
+		fail("bad exit status");
+		return 1;
+	}
+
+	pass();
+
+	return 0;
+err:
+	kill(pid, SIGKILL);
+	return 1;
+}
diff --git a/test/zdtm/live/static/seccomp_filter_inheritance.desc b/test/zdtm/live/static/seccomp_filter_inheritance.desc
new file mode 100644
index 0000000..14dd961
--- /dev/null
+++ b/test/zdtm/live/static/seccomp_filter_inheritance.desc
@@ -0,0 +1 @@
+{'flags': 'suid', 'feature': 'seccomp_filters'}
-- 
2.5.0