1 // SPDX-License-Identifier: GPL-2.0
2 
3 /* Based on Christian Brauner's clone3() example */
4 
5 #define _GNU_SOURCE
6 #include <errno.h>
7 #include <inttypes.h>
8 #include <linux/types.h>
9 #include <linux/sched.h>
10 #include <stdint.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <sys/syscall.h>
14 #include <sys/types.h>
15 #include <sys/un.h>
16 #include <sys/wait.h>
17 #include <unistd.h>
18 #include <sched.h>
19 
20 #include "../kselftest.h"
21 #include "clone3_selftests.h"
22 
23 enum test_mode {
24 	CLONE3_ARGS_NO_TEST,
25 	CLONE3_ARGS_ALL_0,
26 	CLONE3_ARGS_INVAL_EXIT_SIGNAL_BIG,
27 	CLONE3_ARGS_INVAL_EXIT_SIGNAL_NEG,
28 	CLONE3_ARGS_INVAL_EXIT_SIGNAL_CSIG,
29 	CLONE3_ARGS_INVAL_EXIT_SIGNAL_NSIG,
30 };
31 
32 static int call_clone3(uint64_t flags, size_t size, enum test_mode test_mode)
33 {
34 	struct __clone_args args = {
35 		.flags = flags,
36 		.exit_signal = SIGCHLD,
37 	};
38 
39 	struct clone_args_extended {
40 		struct __clone_args args;
41 		__aligned_u64 excess_space[2];
42 	} args_ext;
43 
44 	pid_t pid = -1;
45 	int status;
46 
47 	memset(&args_ext, 0, sizeof(args_ext));
48 	if (size > sizeof(struct __clone_args))
49 		args_ext.excess_space[1] = 1;
50 
51 	if (size == 0)
52 		size = sizeof(struct __clone_args);
53 
54 	switch (test_mode) {
55 	case CLONE3_ARGS_NO_TEST:
56 		/*
57 		 * Uses default 'flags' and 'SIGCHLD'
58 		 * assignment.
59 		 */
60 		break;
61 	case CLONE3_ARGS_ALL_0:
62 		args.flags = 0;
63 		args.exit_signal = 0;
64 		break;
65 	case CLONE3_ARGS_INVAL_EXIT_SIGNAL_BIG:
66 		args.exit_signal = 0xbadc0ded00000000ULL;
67 		break;
68 	case CLONE3_ARGS_INVAL_EXIT_SIGNAL_NEG:
69 		args.exit_signal = 0x0000000080000000ULL;
70 		break;
71 	case CLONE3_ARGS_INVAL_EXIT_SIGNAL_CSIG:
72 		args.exit_signal = 0x0000000000000100ULL;
73 		break;
74 	case CLONE3_ARGS_INVAL_EXIT_SIGNAL_NSIG:
75 		args.exit_signal = 0x00000000000000f0ULL;
76 		break;
77 	}
78 
79 	memcpy(&args_ext.args, &args, sizeof(struct __clone_args));
80 
81 	pid = sys_clone3((struct __clone_args *)&args_ext, size);
82 	if (pid < 0) {
83 		ksft_print_msg("%s - Failed to create new process\n",
84 				strerror(errno));
85 		return -errno;
86 	}
87 
88 	if (pid == 0) {
89 		ksft_print_msg("I am the child, my PID is %d\n", getpid());
90 		_exit(EXIT_SUCCESS);
91 	}
92 
93 	ksft_print_msg("I am the parent (%d). My child's pid is %d\n",
94 			getpid(), pid);
95 
96 	if (waitpid(-1, &status, __WALL) < 0) {
97 		ksft_print_msg("Child returned %s\n", strerror(errno));
98 		return -errno;
99 	}
100 	if (WEXITSTATUS(status))
101 		return WEXITSTATUS(status);
102 
103 	return 0;
104 }
105 
106 static void test_clone3(uint64_t flags, size_t size, int expected,
107 		       enum test_mode test_mode)
108 {
109 	int ret;
110 
111 	ksft_print_msg(
112 		"[%d] Trying clone3() with flags %#" PRIx64 " (size %zu)\n",
113 		getpid(), flags, size);
114 	ret = call_clone3(flags, size, test_mode);
115 	ksft_print_msg("[%d] clone3() with flags says: %d expected %d\n",
116 			getpid(), ret, expected);
117 	if (ret != expected)
118 		ksft_test_result_fail(
119 			"[%d] Result (%d) is different than expected (%d)\n",
120 			getpid(), ret, expected);
121 	else
122 		ksft_test_result_pass(
123 			"[%d] Result (%d) matches expectation (%d)\n",
124 			getpid(), ret, expected);
125 }
126 
127 int main(int argc, char *argv[])
128 {
129 	pid_t pid;
130 
131 	uid_t uid = getuid();
132 
133 	ksft_print_header();
134 	ksft_set_plan(17);
135 	test_clone3_supported();
136 
137 	/* Just a simple clone3() should return 0.*/
138 	test_clone3(0, 0, 0, CLONE3_ARGS_NO_TEST);
139 
140 	/* Do a clone3() in a new PID NS.*/
141 	if (uid == 0)
142 		test_clone3(CLONE_NEWPID, 0, 0, CLONE3_ARGS_NO_TEST);
143 	else
144 		ksft_test_result_skip("Skipping clone3() with CLONE_NEWPID\n");
145 
146 	/* Do a clone3() with CLONE_ARGS_SIZE_VER0. */
147 	test_clone3(0, CLONE_ARGS_SIZE_VER0, 0, CLONE3_ARGS_NO_TEST);
148 
149 	/* Do a clone3() with CLONE_ARGS_SIZE_VER0 - 8 */
150 	test_clone3(0, CLONE_ARGS_SIZE_VER0 - 8, -EINVAL, CLONE3_ARGS_NO_TEST);
151 
152 	/* Do a clone3() with sizeof(struct clone_args) + 8 */
153 	test_clone3(0, sizeof(struct __clone_args) + 8, 0, CLONE3_ARGS_NO_TEST);
154 
155 	/* Do a clone3() with exit_signal having highest 32 bits non-zero */
156 	test_clone3(0, 0, -EINVAL, CLONE3_ARGS_INVAL_EXIT_SIGNAL_BIG);
157 
158 	/* Do a clone3() with negative 32-bit exit_signal */
159 	test_clone3(0, 0, -EINVAL, CLONE3_ARGS_INVAL_EXIT_SIGNAL_NEG);
160 
161 	/* Do a clone3() with exit_signal not fitting into CSIGNAL mask */
162 	test_clone3(0, 0, -EINVAL, CLONE3_ARGS_INVAL_EXIT_SIGNAL_CSIG);
163 
164 	/* Do a clone3() with NSIG < exit_signal < CSIG */
165 	test_clone3(0, 0, -EINVAL, CLONE3_ARGS_INVAL_EXIT_SIGNAL_NSIG);
166 
167 	test_clone3(0, sizeof(struct __clone_args) + 8, 0, CLONE3_ARGS_ALL_0);
168 
169 	test_clone3(0, sizeof(struct __clone_args) + 16, -E2BIG,
170 			CLONE3_ARGS_ALL_0);
171 
172 	test_clone3(0, sizeof(struct __clone_args) * 2, -E2BIG,
173 			CLONE3_ARGS_ALL_0);
174 
175 	/* Do a clone3() with > page size */
176 	test_clone3(0, getpagesize() + 8, -E2BIG, CLONE3_ARGS_NO_TEST);
177 
178 	/* Do a clone3() with CLONE_ARGS_SIZE_VER0 in a new PID NS. */
179 	if (uid == 0)
180 		test_clone3(CLONE_NEWPID, CLONE_ARGS_SIZE_VER0, 0,
181 				CLONE3_ARGS_NO_TEST);
182 	else
183 		ksft_test_result_skip("Skipping clone3() with CLONE_NEWPID\n");
184 
185 	/* Do a clone3() with CLONE_ARGS_SIZE_VER0 - 8 in a new PID NS */
186 	test_clone3(CLONE_NEWPID, CLONE_ARGS_SIZE_VER0 - 8, -EINVAL,
187 			CLONE3_ARGS_NO_TEST);
188 
189 	/* Do a clone3() with sizeof(struct clone_args) + 8 in a new PID NS */
190 	if (uid == 0)
191 		test_clone3(CLONE_NEWPID, sizeof(struct __clone_args) + 8, 0,
192 				CLONE3_ARGS_NO_TEST);
193 	else
194 		ksft_test_result_skip("Skipping clone3() with CLONE_NEWPID\n");
195 
196 	/* Do a clone3() with > page size in a new PID NS */
197 	test_clone3(CLONE_NEWPID, getpagesize() + 8, -E2BIG,
198 			CLONE3_ARGS_NO_TEST);
199 
200 	return !ksft_get_fail_cnt() ? ksft_exit_pass() : ksft_exit_fail();
201 }
202