cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

clone3_selftests.h (1888B)

      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 CLONE_INTO_CGROUP
     19#define CLONE_INTO_CGROUP 0x200000000ULL /* Clone into a specific cgroup given the right permissions. */
     20#endif
     21
     22#ifndef __NR_clone3
     23#define __NR_clone3 -1
     24#endif
     25
     26struct __clone_args {
     27	__aligned_u64 flags;
     28	__aligned_u64 pidfd;
     29	__aligned_u64 child_tid;
     30	__aligned_u64 parent_tid;
     31	__aligned_u64 exit_signal;
     32	__aligned_u64 stack;
     33	__aligned_u64 stack_size;
     34	__aligned_u64 tls;
     35#ifndef CLONE_ARGS_SIZE_VER0
     36#define CLONE_ARGS_SIZE_VER0 64	/* sizeof first published struct */
     37#endif
     38	__aligned_u64 set_tid;
     39	__aligned_u64 set_tid_size;
     40#ifndef CLONE_ARGS_SIZE_VER1
     41#define CLONE_ARGS_SIZE_VER1 80	/* sizeof second published struct */
     42#endif
     43	__aligned_u64 cgroup;
     44#ifndef CLONE_ARGS_SIZE_VER2
     45#define CLONE_ARGS_SIZE_VER2 88	/* sizeof third published struct */
     46#endif
     47};
     48
     49static pid_t sys_clone3(struct __clone_args *args, size_t size)
     50{
     51	fflush(stdout);
     52	fflush(stderr);
     53	return syscall(__NR_clone3, args, size);
     54}
     55
     56static inline void test_clone3_supported(void)
     57{
     58	pid_t pid;
     59	struct __clone_args args = {};
     60
     61	if (__NR_clone3 < 0)
     62		ksft_exit_skip("clone3() syscall is not supported\n");
     63
     64	/* Set to something that will always cause EINVAL. */
     65	args.exit_signal = -1;
     66	pid = sys_clone3(&args, sizeof(args));
     67	if (!pid)
     68		exit(EXIT_SUCCESS);
     69
     70	if (pid > 0) {
     71		wait(NULL);
     72		ksft_exit_fail_msg(
     73			"Managed to create child process with invalid exit_signal\n");
     74	}
     75
     76	if (errno == ENOSYS)
     77		ksft_exit_skip("clone3() syscall is not supported\n");
     78
     79	ksft_print_msg("clone3() syscall supported\n");
     80}
     81
     82#endif /* _CLONE3_SELFTESTS_H */