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