| /* |
| * Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| * Use of this source code is governed by the GPLv2 license. |
| * |
| * Test code for seccomp bpf. |
| */ |
| |
| #include <asm/siginfo.h> |
| #define __have_siginfo_t 1 |
| #define __have_sigval_t 1 |
| #define __have_sigevent_t 1 |
| |
| #include <errno.h> |
| #include <linux/filter.h> |
| #include <sys/prctl.h> |
| #include <sys/ptrace.h> |
| #include <sys/types.h> |
| #include <sys/user.h> |
| #include <linux/prctl.h> |
| #include <linux/ptrace.h> |
| #include <linux/seccomp.h> |
| #include <poll.h> |
| #include <pthread.h> |
| #include <semaphore.h> |
| #include <signal.h> |
| #include <stddef.h> |
| #include <stdbool.h> |
| #include <string.h> |
| #include <linux/elf.h> |
| #include <sys/uio.h> |
| |
| #define _GNU_SOURCE |
| #include <unistd.h> |
| #include <sys/syscall.h> |
| |
| #include "test_harness.h" |
| |
| #ifndef PR_SET_PTRACER |
| # define PR_SET_PTRACER 0x59616d61 |
| #endif |
| |
| #ifndef PR_SET_NO_NEW_PRIVS |
| #define PR_SET_NO_NEW_PRIVS 38 |
| #define PR_GET_NO_NEW_PRIVS 39 |
| #endif |
| |
| #ifndef PR_SECCOMP_EXT |
| #define PR_SECCOMP_EXT 43 |
| #endif |
| |
| #ifndef SECCOMP_EXT_ACT |
| #define SECCOMP_EXT_ACT 1 |
| #endif |
| |
| #ifndef SECCOMP_EXT_ACT_TSYNC |
| #define SECCOMP_EXT_ACT_TSYNC 1 |
| #endif |
| |
| #ifndef SECCOMP_MODE_STRICT |
| #define SECCOMP_MODE_STRICT 1 |
| #endif |
| |
| #ifndef SECCOMP_MODE_FILTER |
| #define SECCOMP_MODE_FILTER 2 |
| #endif |
| |
| #ifndef SECCOMP_RET_KILL |
| #define SECCOMP_RET_KILL 0x00000000U /* kill the task immediately */ |
| #define SECCOMP_RET_TRAP 0x00030000U /* disallow and force a SIGSYS */ |
| #define SECCOMP_RET_ERRNO 0x00050000U /* returns an errno */ |
| #define SECCOMP_RET_TRACE 0x7ff00000U /* pass to a tracer or disallow */ |
| #define SECCOMP_RET_ALLOW 0x7fff0000U /* allow */ |
| |
| /* Masks for the return value sections. */ |
| #define SECCOMP_RET_ACTION 0x7fff0000U |
| #define SECCOMP_RET_DATA 0x0000ffffU |
| |
| struct seccomp_data { |
| int nr; |
| __u32 arch; |
| __u64 instruction_pointer; |
| __u64 args[6]; |
| }; |
| #endif |
| |
| #if __BYTE_ORDER == __LITTLE_ENDIAN |
| #define syscall_arg(_n) (offsetof(struct seccomp_data, args[_n])) |
| #elif __BYTE_ORDER == __BIG_ENDIAN |
| #define syscall_arg(_n) (offsetof(struct seccomp_data, args[_n]) + sizeof(__u32)) |
| #else |
| #error "wut? Unknown __BYTE_ORDER?!" |
| #endif |
| |
| #define SIBLING_EXIT_UNKILLED 0xbadbeef |
| #define SIBLING_EXIT_FAILURE 0xbadface |
| #define SIBLING_EXIT_NEWPRIVS 0xbadfeed |
| |
| TEST(mode_strict_support) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_STRICT, NULL, NULL, NULL); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support CONFIG_SECCOMP"); |
| } |
| syscall(__NR_exit, 1); |
| } |
| |
| TEST_SIGNAL(mode_strict_cannot_call_prctl, SIGKILL) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_STRICT, NULL, NULL, NULL); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support CONFIG_SECCOMP"); |
| } |
| syscall(__NR_prctl, PR_SET_SECCOMP, SECCOMP_MODE_FILTER, |
| NULL, NULL, NULL); |
| EXPECT_FALSE(true) { |
| TH_LOG("Unreachable!"); |
| } |
| } |
| |
| /* Note! This doesn't test no new privs behavior */ |
| TEST(no_new_privs_support) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| EXPECT_EQ(0, ret) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| } |
| |
| /* Tests kernel support by checking for a copy_from_user() fault on * NULL. */ |
| TEST(mode_filter_support) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, NULL, 0, 0); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, NULL, NULL, NULL); |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(EFAULT, errno) { |
| TH_LOG("Kernel does not support CONFIG_SECCOMP_FILTER!"); |
| } |
| } |
| |
| TEST(mode_filter_without_nnp) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_GET_NO_NEW_PRIVS, 0, NULL, 0, 0); |
| ASSERT_LE(0, ret) { |
| TH_LOG("Expected 0 or unsupported for NO_NEW_PRIVS"); |
| } |
| errno = 0; |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| /* Succeeds with CAP_SYS_ADMIN, fails without */ |
| /* TODO(wad) check caps not euid */ |
| if (geteuid()) { |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(EACCES, errno); |
| } else { |
| EXPECT_EQ(0, ret); |
| } |
| } |
| |
| #define MAX_INSNS_PER_PATH 32768 |
| |
| TEST(filter_size_limits) |
| { |
| int i; |
| int count = BPF_MAXINSNS + 1; |
| struct sock_filter allow[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_filter *filter; |
| struct sock_fprog prog = { }; |
| long ret; |
| |
| filter = calloc(count, sizeof(*filter)); |
| ASSERT_NE(NULL, filter); |
| |
| for (i = 0; i < count; i++) |
| filter[i] = allow[0]; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| prog.filter = filter; |
| prog.len = count; |
| |
| /* Too many filter instructions in a single filter. */ |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| ASSERT_NE(0, ret) { |
| TH_LOG("Installing %d insn filter was allowed", prog.len); |
| } |
| |
| /* One less is okay, though. */ |
| prog.len -= 1; |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Installing %d insn filter wasn't allowed", prog.len); |
| } |
| } |
| |
| TEST(filter_chain_limits) |
| { |
| int i; |
| int count = BPF_MAXINSNS; |
| struct sock_filter allow[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_filter *filter; |
| struct sock_fprog prog = { }; |
| long ret; |
| |
| filter = calloc(count, sizeof(*filter)); |
| ASSERT_NE(NULL, filter); |
| |
| for (i = 0; i < count; i++) |
| filter[i] = allow[0]; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| prog.filter = filter; |
| prog.len = 1; |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| prog.len = count; |
| |
| /* Too many total filter instructions. */ |
| for (i = 0; i < MAX_INSNS_PER_PATH; i++) { |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| if (ret != 0) |
| break; |
| } |
| ASSERT_NE(0, ret) { |
| TH_LOG("Allowed %d %d-insn filters (total with penalties:%d)", |
| i, count, i * (count + 4)); |
| } |
| } |
| |
| TEST(mode_filter_cannot_move_to_strict) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_STRICT, NULL, 0, 0); |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(EINVAL, errno); |
| } |
| |
| |
| TEST(mode_filter_get_seccomp) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_GET_SECCOMP, 0, 0, 0, 0); |
| EXPECT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_GET_SECCOMP, 0, 0, 0, 0); |
| EXPECT_EQ(2, ret); |
| } |
| |
| |
| TEST(ALLOW_all) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| } |
| |
| TEST(empty_prog) |
| { |
| struct sock_filter filter[] = { |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(EINVAL, errno); |
| } |
| |
| TEST_SIGNAL(unknown_ret_is_kill_inside, SIGSYS) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, 0x10000000U), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| EXPECT_EQ(0, syscall(__NR_getpid)) { |
| TH_LOG("getpid() shouldn't ever return"); |
| } |
| } |
| |
| /* return code >= 0x80000000 is unused. */ |
| TEST_SIGNAL(unknown_ret_is_kill_above_allow, SIGSYS) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, 0x90000000U), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| EXPECT_EQ(0, syscall(__NR_getpid)) { |
| TH_LOG("getpid() shouldn't ever return"); |
| } |
| } |
| |
| TEST_SIGNAL(KILL_all, SIGSYS) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| } |
| |
| TEST_SIGNAL(KILL_one, SIGSYS) |
| { |
| 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, __NR_getpid, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_SIGNAL(KILL_one_arg_one, SIGSYS) |
| { |
| 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, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| /* Only both with lower 32-bit for now. */ |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, syscall_arg(0)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, 0x0C0FFEE, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| pid_t pid = getpid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(pid, syscall(__NR_getpid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid, 0x0C0FFEE)); |
| } |
| |
| TEST_SIGNAL(KILL_one_arg_six, SIGSYS) |
| { |
| 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, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| /* Only both with lower 32-bit for now. */ |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, syscall_arg(5)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, 0x0C0FFEE, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| pid_t pid = getpid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(pid, syscall(__NR_getpid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid, 1, 2, 3, 4, 5, 0x0C0FFEE)); |
| } |
| |
| /* TODO(wad) add 64-bit versus 32-bit arg tests. */ |
| TEST(arg_out_of_range) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, syscall_arg(6)), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(EINVAL, errno); |
| } |
| |
| TEST(ERRNO_valid) |
| { |
| 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, __NR_read, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ERRNO | E2BIG), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(-1, read(0, NULL, 0)); |
| EXPECT_EQ(E2BIG, errno); |
| } |
| |
| TEST(ERRNO_zero) |
| { |
| 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, __NR_read, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ERRNO | 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* "errno" of 0 is ok. */ |
| EXPECT_EQ(0, read(0, NULL, 0)); |
| } |
| |
| TEST(ERRNO_capped) |
| { |
| 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, __NR_read, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ERRNO | 4096), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| pid_t parent = getppid(); |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(-1, read(0, NULL, 0)); |
| EXPECT_EQ(4095, errno); |
| } |
| |
| FIXTURE_DATA(TRAP) { |
| struct sock_fprog prog; |
| }; |
| |
| FIXTURE_SETUP(TRAP) |
| { |
| 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, __NR_getpid, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRAP), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| |
| memset(&self->prog, 0, sizeof(self->prog)); |
| self->prog.filter = malloc(sizeof(filter)); |
| ASSERT_NE(NULL, self->prog.filter); |
| memcpy(self->prog.filter, filter, sizeof(filter)); |
| self->prog.len = (unsigned short)ARRAY_SIZE(filter); |
| } |
| |
| FIXTURE_TEARDOWN(TRAP) |
| { |
| if (self->prog.filter) |
| free(self->prog.filter); |
| } |
| |
| TEST_F_SIGNAL(TRAP, dfl, SIGSYS) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog); |
| ASSERT_EQ(0, ret); |
| syscall(__NR_getpid); |
| } |
| |
| /* Ensure that SIGSYS overrides SIG_IGN */ |
| TEST_F_SIGNAL(TRAP, ign, SIGSYS) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| signal(SIGSYS, SIG_IGN); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog); |
| ASSERT_EQ(0, ret); |
| syscall(__NR_getpid); |
| } |
| |
| static struct siginfo TRAP_info; |
| static volatile int TRAP_nr; |
| static void TRAP_action(int nr, siginfo_t *info, void *void_context) |
| { |
| memcpy(&TRAP_info, info, sizeof(TRAP_info)); |
| TRAP_nr = nr; |
| } |
| |
| TEST_F(TRAP, handler) |
| { |
| int ret, test; |
| struct sigaction act; |
| sigset_t mask; |
| |
| memset(&act, 0, sizeof(act)); |
| sigemptyset(&mask); |
| sigaddset(&mask, SIGSYS); |
| |
| act.sa_sigaction = &TRAP_action; |
| act.sa_flags = SA_SIGINFO; |
| ret = sigaction(SIGSYS, &act, NULL); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("sigaction failed"); |
| } |
| ret = sigprocmask(SIG_UNBLOCK, &mask, NULL); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("sigprocmask failed"); |
| } |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog); |
| ASSERT_EQ(0, ret); |
| TRAP_nr = 0; |
| memset(&TRAP_info, 0, sizeof(TRAP_info)); |
| /* Expect the registers to be rolled back. (nr = error) may vary |
| * based on arch. */ |
| ret = syscall(__NR_getpid); |
| /* Silence gcc warning about volatile. */ |
| test = TRAP_nr; |
| EXPECT_EQ(SIGSYS, test); |
| struct local_sigsys { |
| void *_call_addr; /* calling user insn */ |
| int _syscall; /* triggering system call number */ |
| unsigned int _arch; /* AUDIT_ARCH_* of syscall */ |
| } *sigsys = (struct local_sigsys *) |
| #ifdef si_syscall |
| &(TRAP_info.si_call_addr); |
| #else |
| &TRAP_info.si_pid; |
| #endif |
| EXPECT_EQ(__NR_getpid, sigsys->_syscall); |
| /* Make sure arch is non-zero. */ |
| EXPECT_NE(0, sigsys->_arch); |
| EXPECT_NE(0, (unsigned long)sigsys->_call_addr); |
| } |
| |
| FIXTURE_DATA(precedence) { |
| struct sock_fprog allow; |
| struct sock_fprog trace; |
| struct sock_fprog error; |
| struct sock_fprog trap; |
| struct sock_fprog kill; |
| }; |
| |
| FIXTURE_SETUP(precedence) |
| { |
| struct sock_filter allow_insns[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_filter trace_insns[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE), |
| }; |
| struct sock_filter error_insns[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ERRNO), |
| }; |
| struct sock_filter trap_insns[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRAP), |
| }; |
| struct sock_filter kill_insns[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_getpid, 1, 0), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| }; |
| |
| memset(self, 0, sizeof(*self)); |
| #define FILTER_ALLOC(_x) \ |
| self->_x.filter = malloc(sizeof(_x##_insns)); \ |
| ASSERT_NE(NULL, self->_x.filter); \ |
| memcpy(self->_x.filter, &_x##_insns, sizeof(_x##_insns)); \ |
| self->_x.len = (unsigned short)ARRAY_SIZE(_x##_insns) |
| FILTER_ALLOC(allow); |
| FILTER_ALLOC(trace); |
| FILTER_ALLOC(error); |
| FILTER_ALLOC(trap); |
| FILTER_ALLOC(kill); |
| } |
| |
| FIXTURE_TEARDOWN(precedence) |
| { |
| #define FILTER_FREE(_x) if (self->_x.filter) free(self->_x.filter) |
| FILTER_FREE(allow); |
| FILTER_FREE(trace); |
| FILTER_FREE(error); |
| FILTER_FREE(trap); |
| FILTER_FREE(kill); |
| } |
| |
| TEST_F(precedence, allow_ok) |
| { |
| pid_t parent, res = 0; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trap); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->kill); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| res = syscall(__NR_getppid); |
| EXPECT_EQ(parent, res); |
| } |
| |
| TEST_F_SIGNAL(precedence, kill_is_highest, SIGSYS) |
| { |
| pid_t parent, res = 0; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trap); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->kill); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| res = syscall(__NR_getppid); |
| EXPECT_EQ(parent, res); |
| /* getpid() should never return. */ |
| res = syscall(__NR_getpid); |
| EXPECT_EQ(0, res); |
| } |
| |
| TEST_F_SIGNAL(precedence, kill_is_highest_in_any_order, SIGSYS) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->kill); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trap); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_F_SIGNAL(precedence, trap_is_second, SIGSYS) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trap); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_F_SIGNAL(precedence, trap_is_second_in_any_order, SIGSYS) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trap); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* getpid() should never return. */ |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_F(precedence, errno_is_third) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_F(precedence, errno_is_third_in_any_order) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->error); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| EXPECT_EQ(0, syscall(__NR_getpid)); |
| } |
| |
| TEST_F(precedence, trace_is_fourth) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* No ptracer */ |
| EXPECT_EQ(-1, syscall(__NR_getpid)); |
| } |
| |
| TEST_F(precedence, trace_is_fourth_in_any_order) |
| { |
| pid_t parent; |
| long ret; |
| |
| parent = getppid(); |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->trace); |
| ASSERT_EQ(0, ret); |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->allow); |
| ASSERT_EQ(0, ret); |
| /* Should work just fine. */ |
| EXPECT_EQ(parent, syscall(__NR_getppid)); |
| /* No ptracer */ |
| EXPECT_EQ(-1, syscall(__NR_getpid)); |
| } |
| |
| #ifndef PTRACE_O_TRACESECCOMP |
| #define PTRACE_O_TRACESECCOMP 0x00000080 |
| #endif |
| |
| /* Catch the Ubuntu 12.04 value error. */ |
| #if PTRACE_EVENT_SECCOMP != 7 |
| #undef PTRACE_EVENT_SECCOMP |
| #endif |
| |
| #ifndef PTRACE_EVENT_SECCOMP |
| #define PTRACE_EVENT_SECCOMP 7 |
| #endif |
| |
| #define IS_SECCOMP_EVENT(status) ((status >> 16) == PTRACE_EVENT_SECCOMP) |
| bool tracer_running; |
| void tracer_stop(int sig) |
| { |
| tracer_running = false; |
| } |
| |
| typedef void tracer_func_t(struct __test_metadata *_metadata, |
| pid_t tracee, int status, void *args); |
| |
| void tracer(struct __test_metadata *_metadata, int fd, pid_t tracee, |
| tracer_func_t tracer_func, void *args) |
| { |
| int ret = -1; |
| struct sigaction action = { |
| .sa_handler = tracer_stop, |
| }; |
| |
| /* Allow external shutdown. */ |
| tracer_running = true; |
| ASSERT_EQ(0, sigaction(SIGUSR1, &action, NULL)); |
| |
| errno = 0; |
| while (ret == -1 && errno != EINVAL) |
| ret = ptrace(PTRACE_ATTACH, tracee, NULL, 0); |
| ASSERT_EQ(0, ret) { |
| kill(tracee, SIGKILL); |
| } |
| /* Wait for attach stop */ |
| wait(NULL); |
| |
| ret = ptrace(PTRACE_SETOPTIONS, tracee, NULL, PTRACE_O_TRACESECCOMP); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Failed to set PTRACE_O_TRACESECCOMP"); |
| kill(tracee, SIGKILL); |
| } |
| ptrace(PTRACE_CONT, tracee, NULL, 0); |
| |
| /* Unblock the tracee */ |
| ASSERT_EQ(1, write(fd, "A", 1)); |
| ASSERT_EQ(0, close(fd)); |
| |
| /* Run until we're shut down. Must assert to stop execution. */ |
| while (tracer_running) { |
| int status; |
| |
| if (wait(&status) != tracee) |
| continue; |
| if (WIFSIGNALED(status) || WIFEXITED(status)) |
| /* Child is dead. Time to go. */ |
| return; |
| |
| /* Make sure this is a seccomp event. */ |
| ASSERT_EQ(true, IS_SECCOMP_EVENT(status)); |
| |
| tracer_func(_metadata, tracee, status, args); |
| |
| ret = ptrace(PTRACE_CONT, tracee, NULL, NULL); |
| ASSERT_EQ(0, ret); |
| } |
| /* Directly report the status of our test harness results. */ |
| syscall(__NR_exit, _metadata->passed ? EXIT_SUCCESS : EXIT_FAILURE); |
| } |
| |
| /* Common tracer setup/teardown functions. */ |
| void cont_handler(int num) |
| { } |
| pid_t setup_trace_fixture(struct __test_metadata *_metadata, |
| tracer_func_t func, void *args) |
| { |
| char sync; |
| int pipefd[2]; |
| pid_t tracer_pid; |
| pid_t tracee = getpid(); |
| |
| /* Setup a pipe for clean synchronization. */ |
| ASSERT_EQ(0, pipe(pipefd)); |
| |
| /* Fork a child which we'll promote to tracer */ |
| tracer_pid = fork(); |
| ASSERT_LE(0, tracer_pid); |
| signal(SIGALRM, cont_handler); |
| if (tracer_pid == 0) { |
| close(pipefd[0]); |
| tracer(_metadata, pipefd[1], tracee, func, args); |
| syscall(__NR_exit, 0); |
| } |
| close(pipefd[1]); |
| prctl(PR_SET_PTRACER, tracer_pid, 0, 0, 0); |
| read(pipefd[0], &sync, 1); |
| close(pipefd[0]); |
| |
| return tracer_pid; |
| } |
| void teardown_trace_fixture(struct __test_metadata *_metadata, |
| pid_t tracer) |
| { |
| if (tracer) { |
| int status; |
| /* |
| * Extract the exit code from the other process and |
| * adopt it for ourselves in case its asserts failed. |
| */ |
| ASSERT_EQ(0, kill(tracer, SIGUSR1)); |
| ASSERT_EQ(tracer, waitpid(tracer, &status, 0)); |
| if (WEXITSTATUS(status)) |
| _metadata->passed = 0; |
| } |
| } |
| |
| /* "poke" tracer arguments and function. */ |
| struct tracer_args_poke_t { |
| unsigned long poke_addr; |
| }; |
| |
| void tracer_poke(struct __test_metadata *_metadata, pid_t tracee, int status, |
| void *args) |
| { |
| int ret; |
| unsigned long msg; |
| struct tracer_args_poke_t *info = (struct tracer_args_poke_t *)args; |
| |
| ret = ptrace(PTRACE_GETEVENTMSG, tracee, NULL, &msg); |
| EXPECT_EQ(0, ret); |
| /* If this fails, don't try to recover. */ |
| ASSERT_EQ(0x1001, msg) { |
| kill(tracee, SIGKILL); |
| } |
| /* |
| * Poke in the message. |
| * Registers are not touched to try to keep this relatively arch |
| * agnostic. |
| */ |
| ret = ptrace(PTRACE_POKEDATA, tracee, info->poke_addr, 0x1001); |
| EXPECT_EQ(0, ret); |
| } |
| |
| FIXTURE_DATA(TRACE_poke) { |
| struct sock_fprog prog; |
| pid_t tracer; |
| long poked; |
| struct tracer_args_poke_t tracer_args; |
| }; |
| |
| FIXTURE_SETUP(TRACE_poke) |
| { |
| 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, __NR_read, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE | 0x1001), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| |
| self->poked = 0; |
| memset(&self->prog, 0, sizeof(self->prog)); |
| self->prog.filter = malloc(sizeof(filter)); |
| ASSERT_NE(NULL, self->prog.filter); |
| memcpy(self->prog.filter, filter, sizeof(filter)); |
| self->prog.len = (unsigned short)ARRAY_SIZE(filter); |
| |
| /* Set up tracer args. */ |
| self->tracer_args.poke_addr = (unsigned long)&self->poked; |
| |
| /* Launch tracer. */ |
| self->tracer = setup_trace_fixture(_metadata, tracer_poke, |
| &self->tracer_args); |
| } |
| |
| FIXTURE_TEARDOWN(TRACE_poke) |
| { |
| teardown_trace_fixture(_metadata, self->tracer); |
| if (self->prog.filter) |
| free(self->prog.filter); |
| } |
| |
| TEST_F(TRACE_poke, read_has_side_effects) |
| { |
| ssize_t ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(0, self->poked); |
| ret = read(-1, NULL, 0); |
| EXPECT_EQ(-1, ret); |
| EXPECT_EQ(0x1001, self->poked); |
| } |
| |
| TEST_F(TRACE_poke, getpid_runs_normally) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| EXPECT_EQ(0, self->poked); |
| EXPECT_NE(0, syscall(__NR_getpid)); |
| EXPECT_EQ(0, self->poked); |
| } |
| |
| #if defined(__x86_64__) |
| # define ARCH_REGS struct user_regs_struct |
| # define SYSCALL_NUM orig_rax |
| # define SYSCALL_RET rax |
| #elif defined(__i386__) |
| # define ARCH_REGS struct user_regs_struct |
| # define SYSCALL_NUM orig_eax |
| # define SYSCALL_RET eax |
| #elif defined(__arm__) |
| # define ARCH_REGS struct pt_regs |
| # define SYSCALL_NUM ARM_r7 |
| # define SYSCALL_RET ARM_r0 |
| #elif defined(__aarch64__) |
| # define ARCH_REGS struct user_pt_regs |
| # define SYSCALL_NUM regs[8] |
| # define SYSCALL_RET regs[0] |
| #elif defined(__powerpc__) |
| # define ARCH_REGS struct pt_regs |
| # define SYSCALL_NUM gpr[0] |
| # define SYSCALL_RET gpr[3] |
| #else |
| # error "Do not know how to find your architecture's registers and syscalls" |
| #endif |
| |
| /* Architecture-specific syscall fetching routine. */ |
| int get_syscall(struct __test_metadata *_metadata, pid_t tracee) |
| { |
| struct iovec iov; |
| ARCH_REGS regs; |
| |
| iov.iov_base = ®s; |
| iov.iov_len = sizeof(regs); |
| EXPECT_EQ(0, ptrace(PTRACE_GETREGSET, tracee, NT_PRSTATUS, &iov)) { |
| TH_LOG("PTRACE_GETREGSET failed"); |
| return -1; |
| } |
| |
| return regs.SYSCALL_NUM; |
| } |
| |
| /* Architecture-specific syscall changing routine. */ |
| void change_syscall(struct __test_metadata *_metadata, |
| pid_t tracee, int syscall) |
| { |
| struct iovec iov; |
| int ret; |
| ARCH_REGS regs; |
| |
| iov.iov_base = ®s; |
| iov.iov_len = sizeof(regs); |
| ret = ptrace(PTRACE_GETREGSET, tracee, NT_PRSTATUS, &iov); |
| EXPECT_EQ(0, ret); |
| |
| #if defined(__x86_64__) || defined(__i386__) || defined(__aarch64__) || defined(__powerpc__) |
| { |
| regs.SYSCALL_NUM = syscall; |
| } |
| |
| #elif defined(__arm__) |
| # ifndef PTRACE_SET_SYSCALL |
| # define PTRACE_SET_SYSCALL 23 |
| # endif |
| { |
| ret = ptrace(PTRACE_SET_SYSCALL, tracee, NULL, syscall); |
| EXPECT_EQ(0, ret); |
| } |
| |
| #else |
| ASSERT_EQ(1, 0) { |
| TH_LOG("How is the syscall changed on this architecture?"); |
| } |
| #endif |
| |
| /* If syscall is skipped, change return value. */ |
| if (syscall == -1) |
| regs.SYSCALL_RET = 1; |
| |
| ret = ptrace(PTRACE_SETREGSET, tracee, NT_PRSTATUS, &iov); |
| EXPECT_EQ(0, ret); |
| } |
| |
| void tracer_syscall(struct __test_metadata *_metadata, pid_t tracee, |
| int status, void *args) |
| { |
| int ret; |
| unsigned long msg; |
| |
| /* Make sure we got the right message. */ |
| ret = ptrace(PTRACE_GETEVENTMSG, tracee, NULL, &msg); |
| EXPECT_EQ(0, ret); |
| |
| switch (msg) { |
| case 0x1002: |
| /* change getpid to getppid. */ |
| change_syscall(_metadata, tracee, __NR_getppid); |
| break; |
| case 0x1003: |
| /* skip gettid. */ |
| change_syscall(_metadata, tracee, -1); |
| break; |
| case 0x1004: |
| /* do nothing (allow getppid) */ |
| break; |
| default: |
| EXPECT_EQ(0, msg) { |
| TH_LOG("Unknown PTRACE_GETEVENTMSG: 0x%lx", msg); |
| kill(tracee, SIGKILL); |
| } |
| } |
| |
| } |
| |
| FIXTURE_DATA(TRACE_syscall) { |
| struct sock_fprog prog; |
| pid_t tracer, mytid, mypid, parent; |
| }; |
| |
| FIXTURE_SETUP(TRACE_syscall) |
| { |
| 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, __NR_getpid, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE | 0x1002), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_gettid, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE | 0x1003), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_getppid, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE | 0x1004), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| |
| memset(&self->prog, 0, sizeof(self->prog)); |
| self->prog.filter = malloc(sizeof(filter)); |
| ASSERT_NE(NULL, self->prog.filter); |
| memcpy(self->prog.filter, filter, sizeof(filter)); |
| self->prog.len = (unsigned short)ARRAY_SIZE(filter); |
| |
| /* Prepare some testable syscall results. */ |
| self->mytid = syscall(__NR_gettid); |
| ASSERT_GT(self->mytid, 0); |
| ASSERT_NE(self->mytid, 1) { |
| TH_LOG("Running this test as init is not supported. :)"); |
| } |
| |
| self->mypid = getpid(); |
| ASSERT_GT(self->mypid, 0); |
| ASSERT_EQ(self->mytid, self->mypid); |
| |
| self->parent = getppid(); |
| ASSERT_GT(self->parent, 0); |
| ASSERT_NE(self->parent, self->mypid); |
| |
| /* Launch tracer. */ |
| self->tracer = setup_trace_fixture(_metadata, tracer_syscall, NULL); |
| } |
| |
| FIXTURE_TEARDOWN(TRACE_syscall) |
| { |
| teardown_trace_fixture(_metadata, self->tracer); |
| if (self->prog.filter) |
| free(self->prog.filter); |
| } |
| |
| TEST_F(TRACE_syscall, syscall_allowed) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| /* getppid works as expected (no changes). */ |
| EXPECT_EQ(self->parent, syscall(__NR_getppid)); |
| EXPECT_NE(self->mypid, syscall(__NR_getppid)); |
| } |
| |
| TEST_F(TRACE_syscall, syscall_redirected) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| /* getpid has been redirected to getppid as expected. */ |
| EXPECT_EQ(self->parent, syscall(__NR_getpid)); |
| EXPECT_NE(self->mypid, syscall(__NR_getpid)); |
| } |
| |
| TEST_F(TRACE_syscall, syscall_dropped) |
| { |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &self->prog, 0, 0); |
| ASSERT_EQ(0, ret); |
| |
| /* gettid has been skipped and an altered return value stored. */ |
| EXPECT_EQ(1, syscall(__NR_gettid)); |
| EXPECT_NE(self->mytid, syscall(__NR_gettid)); |
| } |
| |
| #ifndef __NR_seccomp |
| # if defined(__i386__) |
| # define __NR_seccomp 354 |
| # elif defined(__x86_64__) |
| # define __NR_seccomp 317 |
| # elif defined(__arm__) |
| # define __NR_seccomp 383 |
| # elif defined(__aarch64__) |
| # define __NR_seccomp 277 |
| # elif defined(__powerpc__) |
| # define __NR_seccomp 358 |
| # else |
| # warning "seccomp syscall number unknown for this architecture" |
| # define __NR_seccomp 0xffff |
| # endif |
| #endif |
| |
| #ifndef SECCOMP_SET_MODE_STRICT |
| #define SECCOMP_SET_MODE_STRICT 0 |
| #endif |
| |
| #ifndef SECCOMP_SET_MODE_FILTER |
| #define SECCOMP_SET_MODE_FILTER 1 |
| #endif |
| |
| #ifndef SECCOMP_FLAG_FILTER_TSYNC |
| #define SECCOMP_FLAG_FILTER_TSYNC 1 |
| #endif |
| |
| #ifndef seccomp |
| int seccomp(unsigned int op, unsigned int flags, struct sock_fprog *filter) |
| { |
| errno = 0; |
| return syscall(__NR_seccomp, op, flags, filter); |
| } |
| #endif |
| |
| TEST(seccomp_syscall) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| /* Reject insane operation. */ |
| ret = seccomp(-1, 0, &prog); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Did not reject crazy op value!"); |
| } |
| |
| /* Reject strict with flags or pointer. */ |
| ret = seccomp(SECCOMP_SET_MODE_STRICT, -1, NULL); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Did not reject mode strict with flags!"); |
| } |
| ret = seccomp(SECCOMP_SET_MODE_STRICT, 0, &prog); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Did not reject mode strict with uargs!"); |
| } |
| |
| /* Reject insane args for filter. */ |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, -1, &prog); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Did not reject crazy filter flags!"); |
| } |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, NULL); |
| EXPECT_EQ(EFAULT, errno) { |
| TH_LOG("Did not reject NULL filter!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &prog); |
| EXPECT_EQ(0, errno) { |
| TH_LOG("Kernel does not support SECCOMP_SET_MODE_FILTER: %s", |
| strerror(errno)); |
| } |
| } |
| |
| TEST(seccomp_syscall_mode_lock) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, NULL, 0, 0); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &prog); |
| EXPECT_EQ(0, ret) { |
| TH_LOG("Could not install filter!"); |
| } |
| |
| /* Make sure neither entry point will switch to strict. */ |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_STRICT, 0, 0, 0); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Switched to mode strict!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_STRICT, 0, NULL); |
| EXPECT_EQ(EINVAL, errno) { |
| TH_LOG("Switched to mode strict!"); |
| } |
| } |
| |
| TEST(TSYNC_first) |
| { |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| long ret; |
| |
| ret = prctl(PR_SET_NO_NEW_PRIVS, 1, NULL, 0, 0); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &prog); |
| EXPECT_EQ(0, ret) { |
| TH_LOG("Could not install initial filter with TSYNC!"); |
| } |
| } |
| |
| #define TSYNC_SIBLINGS 2 |
| struct tsync_sibling { |
| pthread_t tid; |
| pid_t system_tid; |
| sem_t *started; |
| pthread_cond_t *cond; |
| pthread_mutex_t *mutex; |
| int diverge; |
| int num_waits; |
| struct sock_fprog *prog; |
| struct __test_metadata *metadata; |
| }; |
| |
| FIXTURE_DATA(TSYNC) { |
| struct sock_fprog root_prog, apply_prog; |
| struct tsync_sibling sibling[TSYNC_SIBLINGS]; |
| sem_t started; |
| pthread_cond_t cond; |
| pthread_mutex_t mutex; |
| int sibling_count; |
| }; |
| |
| FIXTURE_SETUP(TSYNC) |
| { |
| struct sock_filter root_filter[] = { |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_filter apply_filter[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_read, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| |
| memset(&self->root_prog, 0, sizeof(self->root_prog)); |
| memset(&self->apply_prog, 0, sizeof(self->apply_prog)); |
| memset(&self->sibling, 0, sizeof(self->sibling)); |
| self->root_prog.filter = malloc(sizeof(root_filter)); |
| ASSERT_NE(NULL, self->root_prog.filter); |
| memcpy(self->root_prog.filter, &root_filter, sizeof(root_filter)); |
| self->root_prog.len = (unsigned short)ARRAY_SIZE(root_filter); |
| |
| self->apply_prog.filter = malloc(sizeof(apply_filter)); |
| ASSERT_NE(NULL, self->apply_prog.filter); |
| memcpy(self->apply_prog.filter, &apply_filter, sizeof(apply_filter)); |
| self->apply_prog.len = (unsigned short)ARRAY_SIZE(apply_filter); |
| |
| self->sibling_count = 0; |
| pthread_mutex_init(&self->mutex, NULL); |
| pthread_cond_init(&self->cond, NULL); |
| sem_init(&self->started, 0, 0); |
| self->sibling[0].tid = 0; |
| self->sibling[0].cond = &self->cond; |
| self->sibling[0].started = &self->started; |
| self->sibling[0].mutex = &self->mutex; |
| self->sibling[0].diverge = 0; |
| self->sibling[0].num_waits = 1; |
| self->sibling[0].prog = &self->root_prog; |
| self->sibling[0].metadata = _metadata; |
| self->sibling[1].tid = 0; |
| self->sibling[1].cond = &self->cond; |
| self->sibling[1].started = &self->started; |
| self->sibling[1].mutex = &self->mutex; |
| self->sibling[1].diverge = 0; |
| self->sibling[1].prog = &self->root_prog; |
| self->sibling[1].num_waits = 1; |
| self->sibling[1].metadata = _metadata; |
| } |
| |
| FIXTURE_TEARDOWN(TSYNC) |
| { |
| int sib = 0; |
| |
| if (self->root_prog.filter) |
| free(self->root_prog.filter); |
| if (self->apply_prog.filter) |
| free(self->apply_prog.filter); |
| |
| for ( ; sib < self->sibling_count; ++sib) { |
| struct tsync_sibling *s = &self->sibling[sib]; |
| void *status; |
| |
| if (!s->tid) |
| continue; |
| if (pthread_kill(s->tid, 0)) { |
| pthread_cancel(s->tid); |
| pthread_join(s->tid, &status); |
| } |
| } |
| pthread_mutex_destroy(&self->mutex); |
| pthread_cond_destroy(&self->cond); |
| sem_destroy(&self->started); |
| } |
| |
| void *tsync_sibling(void *data) |
| { |
| long ret = 0; |
| struct tsync_sibling *me = data; |
| |
| me->system_tid = syscall(__NR_gettid); |
| |
| pthread_mutex_lock(me->mutex); |
| if (me->diverge) { |
| /* Just re-apply the root prog to fork the tree */ |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, |
| me->prog, 0, 0); |
| } |
| sem_post(me->started); |
| /* Return outside of started so parent notices failures. */ |
| if (ret) { |
| pthread_mutex_unlock(me->mutex); |
| return (void *)SIBLING_EXIT_FAILURE; |
| } |
| do { |
| pthread_cond_wait(me->cond, me->mutex); |
| me->num_waits = me->num_waits - 1; |
| } while (me->num_waits); |
| pthread_mutex_unlock(me->mutex); |
| |
| ret = prctl(PR_GET_NO_NEW_PRIVS, 0, 0, 0, 0); |
| if (!ret) |
| return (void *)SIBLING_EXIT_NEWPRIVS; |
| read(0, NULL, 0); |
| return (void *)SIBLING_EXIT_UNKILLED; |
| } |
| |
| void tsync_start_sibling(struct tsync_sibling *sibling) |
| { |
| pthread_create(&sibling->tid, NULL, tsync_sibling, (void *)sibling); |
| } |
| |
| TEST_F(TSYNC, siblings_fail_prctl) |
| { |
| long ret; |
| void *status; |
| 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, __NR_prctl, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ERRNO | EINVAL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| |
| ASSERT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| /* Check prctl failure detection by requesting sib 0 diverge. */ |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("setting filter failed"); |
| } |
| |
| self->sibling[0].diverge = 1; |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| /* Signal the threads to clean up*/ |
| pthread_mutex_lock(&self->mutex); |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| |
| /* Ensure diverging sibling failed to call prctl. */ |
| pthread_join(self->sibling[0].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_FAILURE, (long)status); |
| pthread_join(self->sibling[1].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_UNKILLED, (long)status); |
| } |
| |
| TEST_F(TSYNC, two_siblings_with_ancestor) |
| { |
| long ret; |
| void *status; |
| |
| ASSERT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &self->root_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support SECCOMP_SET_MODE_FILTER!"); |
| } |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Could install filter on all threads!"); |
| } |
| /* Tell the siblings to test the policy */ |
| pthread_mutex_lock(&self->mutex); |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| /* Ensure they are both killed and don't exit cleanly. */ |
| pthread_join(self->sibling[0].tid, &status); |
| EXPECT_EQ(0x0, (long)status); |
| pthread_join(self->sibling[1].tid, &status); |
| EXPECT_EQ(0x0, (long)status); |
| } |
| |
| TEST_F(TSYNC, two_sibling_want_nnp) |
| { |
| void *status; |
| |
| /* start siblings before any prctl() operations */ |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| /* Tell the siblings to test no policy */ |
| pthread_mutex_lock(&self->mutex); |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| |
| /* Ensure they are both upset about lacking nnp. */ |
| pthread_join(self->sibling[0].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_NEWPRIVS, (long)status); |
| pthread_join(self->sibling[1].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_NEWPRIVS, (long)status); |
| } |
| |
| TEST_F(TSYNC, two_siblings_with_no_filter) |
| { |
| long ret; |
| void *status; |
| |
| /* start siblings before any prctl() operations */ |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| ASSERT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Could install filter on all threads!"); |
| } |
| |
| /* Tell the siblings to test the policy */ |
| pthread_mutex_lock(&self->mutex); |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| |
| /* Ensure they are both killed and don't exit cleanly. */ |
| pthread_join(self->sibling[0].tid, &status); |
| EXPECT_EQ(0x0, (long)status); |
| pthread_join(self->sibling[1].tid, &status); |
| EXPECT_EQ(0x0, (long)status); |
| } |
| |
| TEST_F(TSYNC, two_siblings_with_one_divergence) |
| { |
| long ret; |
| void *status; |
| |
| ASSERT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &self->root_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support SECCOMP_SET_MODE_FILTER!"); |
| } |
| self->sibling[0].diverge = 1; |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(self->sibling[0].system_tid, ret) { |
| TH_LOG("Did not fail on diverged sibling."); |
| } |
| |
| /* Wake the threads */ |
| pthread_mutex_lock(&self->mutex); |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| |
| /* Ensure they are both unkilled. */ |
| pthread_join(self->sibling[0].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_UNKILLED, (long)status); |
| pthread_join(self->sibling[1].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_UNKILLED, (long)status); |
| } |
| |
| TEST_F(TSYNC, two_siblings_not_under_filter) |
| { |
| long ret, sib; |
| void *status; |
| |
| ASSERT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| /* |
| * Sibling 0 will have its own seccomp policy |
| * and Sibling 1 will not be under seccomp at |
| * all. Sibling 1 will enter seccomp and 0 |
| * will cause failure. |
| */ |
| self->sibling[0].diverge = 1; |
| tsync_start_sibling(&self->sibling[0]); |
| tsync_start_sibling(&self->sibling[1]); |
| |
| while (self->sibling_count < TSYNC_SIBLINGS) { |
| sem_wait(&self->started); |
| self->sibling_count++; |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, 0, &self->root_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Kernel does not support SECCOMP_SET_MODE_FILTER!"); |
| } |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(ret, self->sibling[0].system_tid) { |
| TH_LOG("Did not fail on diverged sibling."); |
| } |
| sib = 1; |
| if (ret == self->sibling[0].system_tid) |
| sib = 0; |
| |
| pthread_mutex_lock(&self->mutex); |
| |
| /* Increment the other siblings num_waits so we can clean up |
| * the one we just saw. |
| */ |
| self->sibling[!sib].num_waits += 1; |
| |
| /* Signal the thread to clean up*/ |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| pthread_join(self->sibling[sib].tid, &status); |
| EXPECT_EQ(SIBLING_EXIT_UNKILLED, (long)status); |
| /* Poll for actual task death. pthread_join doesn't guarantee it. */ |
| while (!kill(self->sibling[sib].system_tid, 0)) |
| sleep(0.1); |
| /* Switch to the remaining sibling */ |
| sib = !sib; |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(0, ret) { |
| TH_LOG("Expected the remaining sibling to sync"); |
| }; |
| |
| pthread_mutex_lock(&self->mutex); |
| |
| /* If remaining sibling didn't have a chance to wake up during |
| * the first broadcast, manually reduce the num_waits now. |
| */ |
| if (self->sibling[sib].num_waits > 1) |
| self->sibling[sib].num_waits = 1; |
| ASSERT_EQ(0, pthread_cond_broadcast(&self->cond)) { |
| TH_LOG("cond broadcast non-zero"); |
| } |
| pthread_mutex_unlock(&self->mutex); |
| pthread_join(self->sibling[sib].tid, &status); |
| EXPECT_EQ(0, (long)status); |
| /* Poll for actual task death. pthread_join doesn't guarantee it. */ |
| while (!kill(self->sibling[sib].system_tid, 0)) |
| sleep(0.1); |
| |
| ret = seccomp(SECCOMP_SET_MODE_FILTER, SECCOMP_FLAG_FILTER_TSYNC, |
| &self->apply_prog); |
| ASSERT_EQ(0, ret); /* just us chickens */ |
| } |
| |
| /* Make sure restarted syscalls are seen directly as "restart_syscall". */ |
| TEST(syscall_restart) |
| { |
| long ret; |
| unsigned long msg; |
| pid_t child_pid; |
| int pipefd[2]; |
| int status; |
| siginfo_t info = { }; |
| struct sock_filter filter[] = { |
| BPF_STMT(BPF_LD|BPF_W|BPF_ABS, |
| offsetof(struct seccomp_data, nr)), |
| |
| #ifdef __NR_sigreturn |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_sigreturn, 6, 0), |
| #endif |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_read, 5, 0), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_exit, 4, 0), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_rt_sigreturn, 3, 0), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_poll, 4, 0), |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_restart_syscall, 4, 0), |
| |
| /* Allow __NR_write for easy logging. */ |
| BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, __NR_write, 0, 1), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_ALLOW), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_KILL), |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE|0x100), /* poll */ |
| BPF_STMT(BPF_RET|BPF_K, SECCOMP_RET_TRACE|0x200), /* restart */ |
| }; |
| struct sock_fprog prog = { |
| .len = (unsigned short)ARRAY_SIZE(filter), |
| .filter = filter, |
| }; |
| |
| ASSERT_EQ(0, pipe(pipefd)); |
| |
| child_pid = fork(); |
| ASSERT_LE(0, child_pid); |
| if (child_pid == 0) { |
| /* Child uses EXPECT not ASSERT to deliver status correctly. */ |
| char buf = ' '; |
| struct pollfd fds = { |
| .fd = pipefd[0], |
| .events = POLLIN, |
| }; |
| |
| /* Attach parent as tracer and stop. */ |
| EXPECT_EQ(0, ptrace(PTRACE_TRACEME)); |
| EXPECT_EQ(0, raise(SIGSTOP)); |
| |
| EXPECT_EQ(0, close(pipefd[1])); |
| |
| EXPECT_EQ(0, prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0)) { |
| TH_LOG("Kernel does not support PR_SET_NO_NEW_PRIVS!"); |
| } |
| |
| ret = prctl(PR_SET_SECCOMP, SECCOMP_MODE_FILTER, &prog, 0, 0); |
| EXPECT_EQ(0, ret) { |
| TH_LOG("Failed to install filter!"); |
| } |
| |
| EXPECT_EQ(1, read(pipefd[0], &buf, 1)) { |
| TH_LOG("Failed to read() sync from parent"); |
| } |
| EXPECT_EQ('.', buf) { |
| TH_LOG("Failed to get sync data from read()"); |
| } |
| |
| /* Start poll to be interrupted. */ |
| errno = 0; |
| EXPECT_EQ(1, poll(&fds, 1, -1)) { |
| TH_LOG("Call to poll() failed (errno %d)", errno); |
| } |
| |
| /* Read final sync from parent. */ |
| EXPECT_EQ(1, read(pipefd[0], &buf, 1)) { |
| TH_LOG("Failed final read() from parent"); |
| } |
| EXPECT_EQ('!', buf) { |
| TH_LOG("Failed to get final data from read()"); |
| } |
| |
| /* Directly report the status of our test harness results. */ |
| syscall(__NR_exit, _metadata->passed ? EXIT_SUCCESS |
| : EXIT_FAILURE); |
| } |
| EXPECT_EQ(0, close(pipefd[0])); |
| |
| /* Attach to child, setup options, and release. */ |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| ASSERT_EQ(true, WIFSTOPPED(status)); |
| ASSERT_EQ(0, ptrace(PTRACE_SETOPTIONS, child_pid, NULL, |
| PTRACE_O_TRACESECCOMP)); |
| ASSERT_EQ(0, ptrace(PTRACE_CONT, child_pid, NULL, 0)); |
| ASSERT_EQ(1, write(pipefd[1], ".", 1)); |
| |
| /* Wait for poll() to start. */ |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| ASSERT_EQ(true, WIFSTOPPED(status)); |
| ASSERT_EQ(SIGTRAP, WSTOPSIG(status)); |
| ASSERT_EQ(PTRACE_EVENT_SECCOMP, (status >> 16)); |
| ASSERT_EQ(0, ptrace(PTRACE_GETEVENTMSG, child_pid, NULL, &msg)); |
| ASSERT_EQ(0x100, msg); |
| EXPECT_EQ(__NR_poll, get_syscall(_metadata, child_pid)); |
| |
| /* Might as well check siginfo for sanity while we're here. */ |
| ASSERT_EQ(0, ptrace(PTRACE_GETSIGINFO, child_pid, NULL, &info)); |
| ASSERT_EQ(SIGTRAP, info.si_signo); |
| ASSERT_EQ(SIGTRAP | (PTRACE_EVENT_SECCOMP << 8), info.si_code); |
| EXPECT_EQ(0, info.si_errno); |
| EXPECT_EQ(getuid(), info.si_uid); |
| /* Verify signal delivery came from child (seccomp-triggered). */ |
| EXPECT_EQ(child_pid, info.si_pid); |
| |
| /* Interrupt poll with SIGSTOP (which we'll need to handle). */ |
| ASSERT_EQ(0, kill(child_pid, SIGSTOP)); |
| ASSERT_EQ(0, ptrace(PTRACE_CONT, child_pid, NULL, 0)); |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| ASSERT_EQ(true, WIFSTOPPED(status)); |
| ASSERT_EQ(SIGSTOP, WSTOPSIG(status)); |
| /* Verify signal delivery came from parent now. */ |
| ASSERT_EQ(0, ptrace(PTRACE_GETSIGINFO, child_pid, NULL, &info)); |
| EXPECT_EQ(getpid(), info.si_pid); |
| |
| /* Restart poll with SIGCONT, which triggers restart_syscall. */ |
| ASSERT_EQ(0, kill(child_pid, SIGCONT)); |
| ASSERT_EQ(0, ptrace(PTRACE_CONT, child_pid, NULL, 0)); |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| ASSERT_EQ(true, WIFSTOPPED(status)); |
| ASSERT_EQ(SIGCONT, WSTOPSIG(status)); |
| ASSERT_EQ(0, ptrace(PTRACE_CONT, child_pid, NULL, 0)); |
| |
| /* Wait for restart_syscall() to start. */ |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| ASSERT_EQ(true, WIFSTOPPED(status)); |
| ASSERT_EQ(SIGTRAP, WSTOPSIG(status)); |
| ASSERT_EQ(PTRACE_EVENT_SECCOMP, (status >> 16)); |
| ASSERT_EQ(0, ptrace(PTRACE_GETEVENTMSG, child_pid, NULL, &msg)); |
| ASSERT_EQ(0x200, msg); |
| ret = get_syscall(_metadata, child_pid); |
| #if defined(__arm__) |
| /* FIXME: ARM does not expose true syscall in registers. */ |
| EXPECT_EQ(__NR_poll, ret); |
| #else |
| EXPECT_EQ(__NR_restart_syscall, ret); |
| #endif |
| |
| /* Write again to end poll. */ |
| ASSERT_EQ(0, ptrace(PTRACE_CONT, child_pid, NULL, 0)); |
| ASSERT_EQ(1, write(pipefd[1], "!", 1)); |
| EXPECT_EQ(0, close(pipefd[1])); |
| |
| ASSERT_EQ(child_pid, waitpid(child_pid, &status, 0)); |
| if (WIFSIGNALED(status) || WEXITSTATUS(status)) |
| _metadata->passed = 0; |
| } |
| |
| /* |
| * TODO: |
| * - add microbenchmarks |
| * - expand NNP testing |
| * - better arch-specific TRACE and TRAP handlers. |
| * - endianness checking when appropriate |
| * - 64-bit arg prodding |
| * - arch value testing (x86 modes especially) |
| * - ... |
| */ |
| |
| TEST_HARNESS_MAIN |