xref: /linux/tools/testing/selftests/futex/include/futex2test.h (revision 92329d578d60fe944e30da287ee28f5c154a5802)
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3  * Futex2 library addons for futex tests
4  *
5  * Copyright 2021 Collabora Ltd.
6  */
7 #include <linux/time_types.h>
8 #include <stdint.h>
9 
10 #define u64_to_ptr(x) ((void *)(uintptr_t)(x))
11 
12 #ifndef __NR_futex_waitv
13 #define __NR_futex_waitv 449
14 struct futex_waitv {
15 	__u64 val;
16 	__u64 uaddr;
17 	__u32 flags;
18 	__u32 __reserved;
19 };
20 #endif
21 
22 #ifndef __NR_futex_wake
23 #define __NR_futex_wake 454
24 #endif
25 
26 #ifndef __NR_futex_wait
27 #define __NR_futex_wait 455
28 #endif
29 
30 #ifndef FUTEX2_SIZE_U32
31 #define FUTEX2_SIZE_U32 0x02
32 #endif
33 
34 #ifndef FUTEX2_NUMA
35 #define FUTEX2_NUMA 0x04
36 #endif
37 
38 #ifndef FUTEX2_MPOL
39 #define FUTEX2_MPOL 0x08
40 #endif
41 
42 #ifndef FUTEX2_PRIVATE
43 #define FUTEX2_PRIVATE FUTEX_PRIVATE_FLAG
44 #endif
45 
46 #ifndef FUTEX2_NO_NODE
47 #define FUTEX_NO_NODE (-1)
48 #endif
49 
50 #ifndef FUTEX_32
51 #define FUTEX_32 FUTEX2_SIZE_U32
52 #endif
53 
54 struct futex32_numa {
55 	futex_t futex;
56 	futex_t numa;
57 };
58 
59 /**
60  * futex_waitv - Wait at multiple futexes, wake on any
61  * @waiters:    Array of waiters
62  * @nr_waiters: Length of waiters array
63  * @flags: Operation flags
64  * @timo:  Optional timeout for operation
65  */
futex_waitv(volatile struct futex_waitv * waiters,unsigned long nr_waiters,unsigned long flags,struct timespec * timo,clockid_t clockid)66 static inline int futex_waitv(volatile struct futex_waitv *waiters, unsigned long nr_waiters,
67 			      unsigned long flags, struct timespec *timo, clockid_t clockid)
68 {
69 		struct __kernel_timespec ts = {
70 			.tv_sec = timo->tv_sec,
71 			.tv_nsec = timo->tv_nsec,
72 		};
73 
74 		return syscall(__NR_futex_waitv, waiters, nr_waiters, flags, &ts, clockid);
75 }
76 
77 /*
78  * futex_wait() - block on uaddr with optional timeout
79  * @val:	Expected value
80  * @flags:	FUTEX2 flags
81  * @timeout:	Relative timeout
82  * @clockid:	Clock id for the timeout
83  */
futex2_wait(void * uaddr,long val,unsigned int flags,struct timespec * timeout,clockid_t clockid)84 static inline int futex2_wait(void *uaddr, long val, unsigned int flags,
85 			      struct timespec *timeout, clockid_t clockid)
86 {
87 	return syscall(__NR_futex_wait, uaddr, val, ~0U, flags, timeout, clockid);
88 }
89 
90 /*
91  * futex2_wake() - Wake a number of futexes
92  * @nr:		Number of threads to wake at most
93  * @flags:	FUTEX2 flags
94  */
futex2_wake(void * uaddr,int nr,unsigned int flags)95 static inline int futex2_wake(void *uaddr, int nr, unsigned int flags)
96 {
97 	return syscall(__NR_futex_wake, uaddr, ~0U, nr, flags);
98 }
99