1 // SPDX-License-Identifier: GPL-2.0 2 3 /* 4 * Based on Christian Brauner's clone3() example. 5 * These tests are assuming to be running in the host's 6 * PID namespace. 7 */ 8 9 /* capabilities related code based on selftests/bpf/test_verifier.c */ 10 11 #define _GNU_SOURCE 12 #include <errno.h> 13 #include <linux/types.h> 14 #include <linux/sched.h> 15 #include <stdio.h> 16 #include <stdlib.h> 17 #include <stdbool.h> 18 #include <sys/capability.h> 19 #include <sys/prctl.h> 20 #include <sys/syscall.h> 21 #include <sys/types.h> 22 #include <sys/un.h> 23 #include <sys/wait.h> 24 #include <unistd.h> 25 #include <sched.h> 26 27 #include "../kselftest_harness.h" 28 #include "clone3_selftests.h" 29 30 #ifndef MAX_PID_NS_LEVEL 31 #define MAX_PID_NS_LEVEL 32 32 #endif 33 34 static void child_exit(int ret) 35 { 36 fflush(stdout); 37 fflush(stderr); 38 _exit(ret); 39 } 40 41 static int call_clone3_set_tid(struct __test_metadata *_metadata, 42 pid_t *set_tid, size_t set_tid_size) 43 { 44 int status; 45 pid_t pid = -1; 46 47 struct clone_args args = { 48 .exit_signal = SIGCHLD, 49 .set_tid = ptr_to_u64(set_tid), 50 .set_tid_size = set_tid_size, 51 }; 52 53 pid = sys_clone3(&args, sizeof(struct clone_args)); 54 if (pid < 0) { 55 TH_LOG("%s - Failed to create new process", strerror(errno)); 56 return -errno; 57 } 58 59 if (pid == 0) { 60 int ret; 61 char tmp = 0; 62 63 TH_LOG("I am the child, my PID is %d (expected %d)", getpid(), set_tid[0]); 64 65 if (set_tid[0] != getpid()) 66 child_exit(EXIT_FAILURE); 67 child_exit(EXIT_SUCCESS); 68 } 69 70 TH_LOG("I am the parent (%d). My child's pid is %d", getpid(), pid); 71 72 if (waitpid(pid, &status, 0) < 0) { 73 TH_LOG("Child returned %s", strerror(errno)); 74 return -errno; 75 } 76 77 if (!WIFEXITED(status)) 78 return -1; 79 80 return WEXITSTATUS(status); 81 } 82 83 static int test_clone3_set_tid(struct __test_metadata *_metadata, 84 pid_t *set_tid, size_t set_tid_size) 85 { 86 int ret; 87 88 TH_LOG("[%d] Trying clone3() with CLONE_SET_TID to %d", getpid(), set_tid[0]); 89 ret = call_clone3_set_tid(_metadata, set_tid, set_tid_size); 90 TH_LOG("[%d] clone3() with CLONE_SET_TID %d says:%d", getpid(), set_tid[0], ret); 91 return ret; 92 } 93 94 struct libcap { 95 struct __user_cap_header_struct hdr; 96 struct __user_cap_data_struct data[2]; 97 }; 98 99 static int set_capability(void) 100 { 101 cap_value_t cap_values[] = { CAP_SETUID, CAP_SETGID }; 102 struct libcap *cap; 103 int ret = -1; 104 cap_t caps; 105 106 caps = cap_get_proc(); 107 if (!caps) { 108 perror("cap_get_proc"); 109 return -1; 110 } 111 112 /* Drop all capabilities */ 113 if (cap_clear(caps)) { 114 perror("cap_clear"); 115 goto out; 116 } 117 118 cap_set_flag(caps, CAP_EFFECTIVE, 2, cap_values, CAP_SET); 119 cap_set_flag(caps, CAP_PERMITTED, 2, cap_values, CAP_SET); 120 121 cap = (struct libcap *) caps; 122 123 /* 40 -> CAP_CHECKPOINT_RESTORE */ 124 cap->data[1].effective |= 1 << (40 - 32); 125 cap->data[1].permitted |= 1 << (40 - 32); 126 127 if (cap_set_proc(caps)) { 128 perror("cap_set_proc"); 129 goto out; 130 } 131 ret = 0; 132 out: 133 if (cap_free(caps)) 134 perror("cap_free"); 135 return ret; 136 } 137 138 TEST(clone3_cap_checkpoint_restore) 139 { 140 pid_t pid; 141 int status; 142 int ret = 0; 143 pid_t set_tid[1]; 144 145 test_clone3_supported(); 146 147 EXPECT_EQ(getuid(), 0) 148 XFAIL(return, "Skipping all tests as non-root\n"); 149 150 memset(&set_tid, 0, sizeof(set_tid)); 151 152 /* Find the current active PID */ 153 pid = fork(); 154 if (pid == 0) { 155 TH_LOG("Child has PID %d", getpid()); 156 child_exit(EXIT_SUCCESS); 157 } 158 ASSERT_GT(waitpid(pid, &status, 0), 0) 159 TH_LOG("Waiting for child %d failed", pid); 160 161 /* After the child has finished, its PID should be free. */ 162 set_tid[0] = pid; 163 164 ASSERT_EQ(set_capability(), 0) 165 TH_LOG("Could not set CAP_CHECKPOINT_RESTORE"); 166 167 ASSERT_EQ(prctl(PR_SET_KEEPCAPS, 1, 0, 0, 0), 0); 168 169 EXPECT_EQ(setgid(65534), 0) 170 TH_LOG("Failed to setgid(65534)"); 171 ASSERT_EQ(setuid(65534), 0); 172 173 set_tid[0] = pid; 174 /* This would fail without CAP_CHECKPOINT_RESTORE */ 175 ASSERT_EQ(test_clone3_set_tid(_metadata, set_tid, 1), -EPERM); 176 ASSERT_EQ(set_capability(), 0) 177 TH_LOG("Could not set CAP_CHECKPOINT_RESTORE"); 178 /* This should work as we have CAP_CHECKPOINT_RESTORE as non-root */ 179 ASSERT_EQ(test_clone3_set_tid(_metadata, set_tid, 1), 0); 180 } 181 182 TEST_HARNESS_MAIN 183