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
19 #define BACKLOG 512
20
21 #define PORT 9100
22
23 struct io_uring ring;
24
25 struct __kernel_timespec ts = {
26 .tv_sec = 300,
27 .tv_nsec = 0,
28 };
29
add_timeout(struct io_uring * ring,int fd)30 static void add_timeout(struct io_uring *ring, int fd)
31 {
32 struct io_uring_sqe *sqe;
33
34 sqe = io_uring_get_sqe(ring);
35 io_uring_prep_timeout(sqe, &ts, 100, 0);
36 sqe->flags |= IOSQE_IO_LINK;
37 }
38
add_accept(struct io_uring * ring,int fd)39 static void add_accept(struct io_uring *ring, int fd)
40 {
41 struct io_uring_sqe *sqe;
42
43 sqe = io_uring_get_sqe(ring);
44 io_uring_prep_accept(sqe, fd, 0, 0, SOCK_NONBLOCK | SOCK_CLOEXEC);
45 sqe->flags |= IOSQE_IO_LINK;
46 }
47
setup_io_uring(void)48 static int setup_io_uring(void)
49 {
50 int ret;
51
52 ret = io_uring_queue_init(16, &ring, 0);
53 if (ret) {
54 fprintf(stderr, "Unable to setup io_uring: %s\n", strerror(-ret));
55 return 1;
56 }
57
58 return 0;
59 }
60
alarm_sig(int sig)61 static void alarm_sig(int sig)
62 {
63 exit(0);
64 }
65
main(int argc,char * argv[])66 int main(int argc, char *argv[])
67 {
68 struct sockaddr_in serv_addr;
69 struct io_uring_cqe *cqe;
70 int ret, sock_listen_fd;
71 const int val = 1;
72 int i;
73
74 if (argc > 1)
75 return 0;
76
77 sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
78 if (sock_listen_fd < 0) {
79 perror("socket");
80 return 1;
81 }
82
83 setsockopt(sock_listen_fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
84
85 memset(&serv_addr, 0, sizeof(serv_addr));
86 serv_addr.sin_family = AF_INET;
87 serv_addr.sin_addr.s_addr = INADDR_ANY;
88
89 for (i = 0; i < 100; i++) {
90 serv_addr.sin_port = htons(PORT + i);
91
92 ret = bind(sock_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
93 if (!ret)
94 break;
95 if (errno != EADDRINUSE) {
96 fprintf(stderr, "bind: %s\n", strerror(errno));
97 return 1;
98 }
99 if (i == 99) {
100 printf("Gave up on finding a port, skipping\n");
101 goto out;
102 }
103 }
104
105 if (listen(sock_listen_fd, BACKLOG) < 0) {
106 perror("Error listening on socket\n");
107 return 1;
108 }
109
110 if (setup_io_uring())
111 return 1;
112
113 add_timeout(&ring, sock_listen_fd);
114 add_accept(&ring, sock_listen_fd);
115
116 ret = io_uring_submit(&ring);
117 if (ret != 2) {
118 fprintf(stderr, "submit=%d\n", ret);
119 return 1;
120 }
121
122 signal(SIGALRM, alarm_sig);
123 alarm(1);
124
125 ret = io_uring_wait_cqe(&ring, &cqe);
126 if (ret) {
127 fprintf(stderr, "wait_cqe=%d\n", ret);
128 return 1;
129 }
130
131 out:
132 io_uring_queue_exit(&ring);
133 return 0;
134 }
135