^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) /* SPDX-License-Identifier: GPL-2.0 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #ifndef __PIDFD_H
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #define __PIDFD_H
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) #define _GNU_SOURCE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) #include <fcntl.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include <sched.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <signal.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) #include <syscall.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include <sys/mount.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #include <sys/types.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) #include <sys/wait.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) #include "../kselftest.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) #ifndef P_PIDFD
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) #define P_PIDFD 3
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) #ifndef CLONE_NEWTIME
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) #define CLONE_NEWTIME 0x00000080
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) #ifndef CLONE_PIDFD
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) #define CLONE_PIDFD 0x00001000
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) #ifndef __NR_pidfd_open
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) #define __NR_pidfd_open -1
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) #ifndef __NR_pidfd_send_signal
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) #define __NR_pidfd_send_signal -1
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) #ifndef __NR_clone3
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) #define __NR_clone3 -1
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) #ifndef __NR_pidfd_getfd
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) #define __NR_pidfd_getfd -1
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) #ifndef PIDFD_NONBLOCK
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) #define PIDFD_NONBLOCK O_NONBLOCK
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) * The kernel reserves 300 pids via RESERVED_PIDS in kernel/pid.c
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) * That means, when it wraps around any pid < 300 will be skipped.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) * So we need to use a pid > 300 in order to test recycling.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) #define PID_RECYCLE 1000
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) * Define a few custom error codes for the child process to clearly indicate
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) * what is happening. This way we can tell the difference between a system
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) * error, a test error, etc.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) #define PIDFD_PASS 0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) #define PIDFD_FAIL 1
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) #define PIDFD_ERROR 2
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) #define PIDFD_SKIP 3
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) #define PIDFD_XFAIL 4
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) static inline int wait_for_pid(pid_t pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) int status, ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) again:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) ret = waitpid(pid, &status, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) if (ret == -1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) if (errno == EINTR)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) goto again;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81) ksft_print_msg("waitpid returned -1, errno=%d\n", errno);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) if (!WIFEXITED(status)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86) ksft_print_msg(
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) "waitpid !WIFEXITED, WIFSIGNALED=%d, WTERMSIG=%d\n",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88) WIFSIGNALED(status), WTERMSIG(status));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89) return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) ret = WEXITSTATUS(status);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93) ksft_print_msg("waitpid WEXITSTATUS=%d\n", ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) static inline int sys_pidfd_open(pid_t pid, unsigned int flags)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 99) return syscall(__NR_pidfd_open, pid, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) static inline int sys_pidfd_send_signal(int pidfd, int sig, siginfo_t *info,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) unsigned int flags)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) return syscall(__NR_pidfd_send_signal, pidfd, sig, info, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) static inline int sys_pidfd_getfd(int pidfd, int fd, int flags)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) return syscall(__NR_pidfd_getfd, pidfd, fd, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) static inline int sys_memfd_create(const char *name, unsigned int flags)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) return syscall(__NR_memfd_create, name, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) #endif /* __PIDFD_H */