clone3_selftests.h

 1/* SPDX-License-Identifier: GPL-2.0 */
 2
 3#ifndef _CLONE3_SELFTESTS_H
 4#define _CLONE3_SELFTESTS_H
 5
 6#define _GNU_SOURCE
 7#include <sched.h>
 8#include <linux/sched.h>
 9#include <linux/types.h>
10#include <stdint.h>
11#include <syscall.h>
12#include <sys/wait.h>
13
14#include "kselftest.h"
15
16#define ptr_to_u64(ptr) ((__u64)((uintptr_t)(ptr)))
17
18#ifndef __NR_clone3
19#define __NR_clone3 435
20#endif
21
22struct __clone_args {
23	__aligned_u64 flags;
24	__aligned_u64 pidfd;
25	__aligned_u64 child_tid;
26	__aligned_u64 parent_tid;
27	__aligned_u64 exit_signal;
28	__aligned_u64 stack;
29	__aligned_u64 stack_size;
30	__aligned_u64 tls;
31	__aligned_u64 set_tid;
32	__aligned_u64 set_tid_size;
33	__aligned_u64 cgroup;
34};
35
36static pid_t sys_clone3(struct __clone_args *args, size_t size)
37{
38	fflush(stdout);
39	fflush(stderr);
40	return syscall(__NR_clone3, args, size);
41}
42
43static inline void test_clone3_supported(void)
44{
45	pid_t pid;
46	struct __clone_args args = {};
47
48	if (__NR_clone3 < 0)
49		ksft_exit_skip("clone3() syscall is not supported\n");
50
51	/* Set to something that will always cause EINVAL. */
52	args.exit_signal = -1;
53	pid = sys_clone3(&args, sizeof(args));
54	if (!pid)
55		exit(EXIT_SUCCESS);
56
57	if (pid > 0) {
58		wait(NULL);
59		ksft_exit_fail_msg(
60			"Managed to create child process with invalid exit_signal\n");
61	}
62
63	if (errno == ENOSYS)
64		ksft_exit_skip("clone3() syscall is not supported\n");
65
66	ksft_print_msg("clone3() syscall supported\n");
67}
68
69#endif /* _CLONE3_SELFTESTS_H */