1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright Collabora Ltd., 2021
4  *
5  * futex cmp requeue test by André Almeida <andrealmeid@collabora.com>
6  */
7 
8 #include <pthread.h>
9 #include <limits.h>
10 #include "logging.h"
11 #include "futextest.h"
12 
13 #define TEST_NAME "futex-requeue"
14 #define timeout_ns  30000000
15 #define WAKE_WAIT_US 10000
16 
17 volatile futex_t *f1;
18 
19 void usage(char *prog)
20 {
21 	printf("Usage: %s\n", prog);
22 	printf("  -c	Use color\n");
23 	printf("  -h	Display this help message\n");
24 	printf("  -v L	Verbosity level: %d=QUIET %d=CRITICAL %d=INFO\n",
25 	       VQUIET, VCRITICAL, VINFO);
26 }
27 
28 void *waiterfn(void *arg)
29 {
30 	struct timespec to;
31 
32 	to.tv_sec = 0;
33 	to.tv_nsec = timeout_ns;
34 
35 	if (futex_wait(f1, *f1, &to, 0))
36 		printf("waiter failed errno %d\n", errno);
37 
38 	return NULL;
39 }
40 
41 int main(int argc, char *argv[])
42 {
43 	pthread_t waiter[10];
44 	int res, ret = RET_PASS;
45 	int c, i;
46 	volatile futex_t _f1 = 0;
47 	volatile futex_t f2 = 0;
48 
49 	f1 = &_f1;
50 
51 	while ((c = getopt(argc, argv, "cht:v:")) != -1) {
52 		switch (c) {
53 		case 'c':
54 			log_color(1);
55 			break;
56 		case 'h':
57 			usage(basename(argv[0]));
58 			exit(0);
59 		case 'v':
60 			log_verbosity(atoi(optarg));
61 			break;
62 		default:
63 			usage(basename(argv[0]));
64 			exit(1);
65 		}
66 	}
67 
68 	ksft_print_header();
69 	ksft_set_plan(2);
70 	ksft_print_msg("%s: Test futex_requeue\n",
71 		       basename(argv[0]));
72 
73 	/*
74 	 * Requeue a waiter from f1 to f2, and wake f2.
75 	 */
76 	if (pthread_create(&waiter[0], NULL, waiterfn, NULL))
77 		error("pthread_create failed\n", errno);
78 
79 	usleep(WAKE_WAIT_US);
80 
81 	info("Requeuing 1 futex from f1 to f2\n");
82 	res = futex_cmp_requeue(f1, 0, &f2, 0, 1, 0);
83 	if (res != 1) {
84 		ksft_test_result_fail("futex_requeue simple returned: %d %s\n",
85 				      res ? errno : res,
86 				      res ? strerror(errno) : "");
87 		ret = RET_FAIL;
88 	}
89 
90 
91 	info("Waking 1 futex at f2\n");
92 	res = futex_wake(&f2, 1, 0);
93 	if (res != 1) {
94 		ksft_test_result_fail("futex_requeue simple returned: %d %s\n",
95 				      res ? errno : res,
96 				      res ? strerror(errno) : "");
97 		ret = RET_FAIL;
98 	} else {
99 		ksft_test_result_pass("futex_requeue simple succeeds\n");
100 	}
101 
102 
103 	/*
104 	 * Create 10 waiters at f1. At futex_requeue, wake 3 and requeue 7.
105 	 * At futex_wake, wake INT_MAX (should be exactly 7).
106 	 */
107 	for (i = 0; i < 10; i++) {
108 		if (pthread_create(&waiter[i], NULL, waiterfn, NULL))
109 			error("pthread_create failed\n", errno);
110 	}
111 
112 	usleep(WAKE_WAIT_US);
113 
114 	info("Waking 3 futexes at f1 and requeuing 7 futexes from f1 to f2\n");
115 	res = futex_cmp_requeue(f1, 0, &f2, 3, 7, 0);
116 	if (res != 10) {
117 		ksft_test_result_fail("futex_requeue many returned: %d %s\n",
118 				      res ? errno : res,
119 				      res ? strerror(errno) : "");
120 		ret = RET_FAIL;
121 	}
122 
123 	info("Waking INT_MAX futexes at f2\n");
124 	res = futex_wake(&f2, INT_MAX, 0);
125 	if (res != 7) {
126 		ksft_test_result_fail("futex_requeue many returned: %d %s\n",
127 				      res ? errno : res,
128 				      res ? strerror(errno) : "");
129 		ret = RET_FAIL;
130 	} else {
131 		ksft_test_result_pass("futex_requeue many succeeds\n");
132 	}
133 
134 	ksft_print_cnts();
135 	return ret;
136 }
137