1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: Test that io_uring_submit_and_wait_timeout() returns the
4 * right value (submit count) and that it doesn't end up waiting twice.
5 *
6 */
7 #include <errno.h>
8 #include <stdio.h>
9 #include <unistd.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <fcntl.h>
13 #include <sys/time.h>
14
15 #include "liburing.h"
16 #include "test.h"
17
mtime_since(const struct timeval * s,const struct timeval * e)18 static unsigned long long mtime_since(const struct timeval *s,
19 const struct timeval *e)
20 {
21 long long sec, usec;
22
23 sec = e->tv_sec - s->tv_sec;
24 usec = (e->tv_usec - s->tv_usec);
25 if (sec > 0 && usec < 0) {
26 sec--;
27 usec += 1000000;
28 }
29
30 sec *= 1000;
31 usec /= 1000;
32 return sec + usec;
33 }
34
mtime_since_now(struct timeval * tv)35 static unsigned long long mtime_since_now(struct timeval *tv)
36 {
37 struct timeval end;
38
39 gettimeofday(&end, NULL);
40 return mtime_since(tv, &end);
41 }
42
test(struct io_uring * ring)43 static int test(struct io_uring *ring)
44 {
45 struct io_uring_cqe *cqe;
46 struct io_uring_sqe *sqe;
47 struct __kernel_timespec ts;
48 struct timeval tv;
49 int ret, i;
50
51 for (i = 0; i < 1; i++) {
52 sqe = io_uring_get_sqe(ring);
53 if (!sqe) {
54 fprintf(stderr, "get sqe failed at %d\n", i);
55 goto err;
56 }
57 io_uring_prep_nop(sqe);
58 }
59
60 ts.tv_sec = 1;
61 ts.tv_nsec = 0;
62 gettimeofday(&tv, NULL);
63 ret = io_uring_submit_and_wait_timeout(ring, &cqe, 2, &ts, NULL);
64 if (ret < 0) {
65 fprintf(stderr, "submit_and_wait_timeout: %d\n", ret);
66 goto err;
67 }
68 ret = mtime_since_now(&tv);
69 /* allow some slack, should be around 1s */
70 if (ret > 1200) {
71 fprintf(stderr, "wait took too long: %d\n", ret);
72 goto err;
73 }
74 return 0;
75 err:
76 return 1;
77 }
78
test_ring(void)79 static int test_ring(void)
80 {
81 struct io_uring ring;
82 struct io_uring_params p = { };
83 int ret;
84
85 p.flags = 0;
86 ret = io_uring_queue_init_params(8, &ring, &p);
87 if (ret) {
88 fprintf(stderr, "ring setup failed: %d\n", ret);
89 return 1;
90 }
91
92 ret = test(&ring);
93 if (ret) {
94 fprintf(stderr, "test failed\n");
95 goto err;
96 }
97 err:
98 io_uring_queue_exit(&ring);
99 return ret;
100 }
101
main(int argc,char * argv[])102 int main(int argc, char *argv[])
103 {
104 if (argc > 1)
105 return 0;
106
107 return test_ring();
108 }
109