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 }
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 T_EXIT_SKIP;
76
77 sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
78 if (sock_listen_fd < 0) {
79 perror("socket");
80 return T_EXIT_FAIL;
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 T_EXIT_FAIL;
98 }
99 if (i == 99) {
100 printf("Gave up on finding a port, skipping\n");
101 goto skip;
102 }
103 }
104
105 if (listen(sock_listen_fd, BACKLOG) < 0) {
106 perror("Error listening on socket\n");
107 return T_EXIT_FAIL;
108 }
109
110 if (setup_io_uring())
111 return T_EXIT_FAIL;
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 T_EXIT_FAIL;
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 T_EXIT_FAIL;
129 }
130
131 io_uring_queue_exit(&ring);
132 return T_EXIT_PASS;
133 skip:
134 io_uring_queue_exit(&ring);
135 return T_EXIT_SKIP;
136 }
137