1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Based on a test case from Josef Grieb - test that we can exit without
4 * hanging if we have the task file table pinned by a request that is linked
5 * to another request that doesn't finish.
6 */
7 #include <errno.h>
8 #include <fcntl.h>
9 #include <netinet/in.h>
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <string.h>
13 #include <strings.h>
14 #include <sys/socket.h>
15 #include <unistd.h>
16 #include <poll.h>
17 #include "liburing.h"
18 #include "helpers.h"
19
20 #define BACKLOG 512
21
22 #define PORT 9100
23
24 static struct io_uring ring;
25
26 static struct __kernel_timespec ts = {
27 .tv_sec = 300,
28 .tv_nsec = 0,
29 };
30
add_timeout(struct io_uring * ring,int fd)31 static void add_timeout(struct io_uring *ring, int fd)
32 {
33 struct io_uring_sqe *sqe;
34
35 sqe = io_uring_get_sqe(ring);
36 io_uring_prep_timeout(sqe, &ts, 100, 0);
37 sqe->flags |= IOSQE_IO_LINK;
38 }
39
add_accept(struct io_uring * ring,int fd)40 static void add_accept(struct io_uring *ring, int fd)
41 {
42 struct io_uring_sqe *sqe;
43
44 sqe = io_uring_get_sqe(ring);
45 io_uring_prep_accept(sqe, fd, 0, 0, SOCK_NONBLOCK | SOCK_CLOEXEC);
46 sqe->flags |= IOSQE_IO_LINK;
47 }
48
setup_io_uring(void)49 static int setup_io_uring(void)
50 {
51 int ret;
52
53 ret = io_uring_queue_init(16, &ring, 0);
54 if (ret) {
55 fprintf(stderr, "Unable to setup io_uring: %s\n", strerror(-ret));
56 return 1;
57 }
58
59 return 0;
60 }
61
alarm_sig(int sig)62 static void alarm_sig(int sig)
63 {
64 exit(0);
65 }
66
main(int argc,char * argv[])67 int main(int argc, char *argv[])
68 {
69 struct sockaddr_in serv_addr;
70 struct io_uring_cqe *cqe;
71 int ret, sock_listen_fd;
72 const int val = 1;
73 int i;
74
75 if (argc > 1)
76 return T_EXIT_SKIP;
77
78 sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
79 if (sock_listen_fd < 0) {
80 perror("socket");
81 return T_EXIT_FAIL;
82 }
83
84 setsockopt(sock_listen_fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
85
86 memset(&serv_addr, 0, sizeof(serv_addr));
87 serv_addr.sin_family = AF_INET;
88 serv_addr.sin_addr.s_addr = INADDR_ANY;
89
90 for (i = 0; i < 100; i++) {
91 serv_addr.sin_port = htons(PORT + i);
92
93 ret = bind(sock_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
94 if (!ret)
95 break;
96 if (errno != EADDRINUSE) {
97 fprintf(stderr, "bind: %s\n", strerror(errno));
98 return T_EXIT_FAIL;
99 }
100 if (i == 99) {
101 printf("Gave up on finding a port, skipping\n");
102 goto skip;
103 }
104 }
105
106 if (listen(sock_listen_fd, BACKLOG) < 0) {
107 perror("Error listening on socket\n");
108 return T_EXIT_FAIL;
109 }
110
111 if (setup_io_uring())
112 return T_EXIT_FAIL;
113
114 add_timeout(&ring, sock_listen_fd);
115 add_accept(&ring, sock_listen_fd);
116
117 ret = io_uring_submit(&ring);
118 if (ret != 2) {
119 fprintf(stderr, "submit=%d\n", ret);
120 return T_EXIT_FAIL;
121 }
122
123 signal(SIGALRM, alarm_sig);
124 alarm(1);
125
126 ret = io_uring_wait_cqe(&ring, &cqe);
127 if (ret) {
128 fprintf(stderr, "wait_cqe=%d\n", ret);
129 return T_EXIT_FAIL;
130 }
131
132 io_uring_queue_exit(&ring);
133 return T_EXIT_PASS;
134 skip:
135 io_uring_queue_exit(&ring);
136 return T_EXIT_SKIP;
137 }
138