summaryrefslogtreecommitdiffstats
path: root/contrib/libs/liburing/test/files-exit-hang-poll.c
diff options
context:
space:
mode:
authorilnaz <[email protected]>2022-12-13 16:01:38 +0300
committerilnaz <[email protected]>2022-12-13 16:01:38 +0300
commitf2bea70bea01921ec43846224d100f2c70dd5719 (patch)
treeeead917572063b63adc1c9a76284c8fbd10f25a3 /contrib/libs/liburing/test/files-exit-hang-poll.c
parent1ab9ee3dfe0ab4023a3a57bf55de31dff3eac908 (diff)
Add cross-link
Diffstat (limited to 'contrib/libs/liburing/test/files-exit-hang-poll.c')
-rw-r--r--contrib/libs/liburing/test/files-exit-hang-poll.c115
1 files changed, 115 insertions, 0 deletions
diff --git a/contrib/libs/liburing/test/files-exit-hang-poll.c b/contrib/libs/liburing/test/files-exit-hang-poll.c
new file mode 100644
index 00000000000..2428a10283b
--- /dev/null
+++ b/contrib/libs/liburing/test/files-exit-hang-poll.c
@@ -0,0 +1,115 @@
+#include "../config-host.h"
+/* SPDX-License-Identifier: MIT */
+/*
+ * Based on a test case from Josef Grieb - test that we can exit without
+ * hanging if we have the task file table pinned by a request that is linked
+ * to another request that doesn't finish.
+ */
+#include <errno.h>
+#include <fcntl.h>
+#include <netinet/in.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <strings.h>
+#include <sys/socket.h>
+#include <unistd.h>
+#include <poll.h>
+#include "liburing.h"
+#include "helpers.h"
+
+#define BACKLOG 512
+
+static struct io_uring ring;
+
+static void add_poll(struct io_uring *ring, int fd)
+{
+ struct io_uring_sqe *sqe;
+
+ sqe = io_uring_get_sqe(ring);
+ io_uring_prep_poll_add(sqe, fd, POLLIN);
+ sqe->flags |= IOSQE_IO_LINK;
+}
+
+static void add_accept(struct io_uring *ring, int fd)
+{
+ struct io_uring_sqe *sqe;
+
+ sqe = io_uring_get_sqe(ring);
+ io_uring_prep_accept(sqe, fd, 0, 0, SOCK_NONBLOCK | SOCK_CLOEXEC);
+}
+
+static int setup_io_uring(void)
+{
+ int ret;
+
+ ret = io_uring_queue_init(16, &ring, 0);
+ if (ret) {
+ fprintf(stderr, "Unable to setup io_uring: %s\n", strerror(-ret));
+ return 1;
+ }
+
+ return 0;
+}
+
+static void alarm_sig(int sig)
+{
+ exit(0);
+}
+
+int main(int argc, char *argv[])
+{
+ struct sockaddr_in serv_addr;
+ struct io_uring_cqe *cqe;
+ int ret, sock_listen_fd;
+ const int val = 1;
+
+ if (argc > 1)
+ return T_EXIT_SKIP;
+
+ sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
+ if (sock_listen_fd < 0) {
+ perror("socket");
+ return T_EXIT_FAIL;
+ }
+
+ setsockopt(sock_listen_fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
+
+ memset(&serv_addr, 0, sizeof(serv_addr));
+ serv_addr.sin_family = AF_INET;
+ serv_addr.sin_addr.s_addr = INADDR_ANY;
+
+ if (t_bind_ephemeral_port(sock_listen_fd, &serv_addr)) {
+ perror("bind");
+ return T_EXIT_FAIL;
+ }
+
+ if (listen(sock_listen_fd, BACKLOG) < 0) {
+ perror("Error listening on socket\n");
+ return T_EXIT_FAIL;
+ }
+
+ if (setup_io_uring())
+ return T_EXIT_FAIL;
+
+ add_poll(&ring, sock_listen_fd);
+ add_accept(&ring, sock_listen_fd);
+
+ ret = io_uring_submit(&ring);
+ if (ret != 2) {
+ fprintf(stderr, "submit=%d\n", ret);
+ return T_EXIT_FAIL;
+ }
+
+ signal(SIGALRM, alarm_sig);
+ alarm(1);
+
+ ret = io_uring_wait_cqe(&ring, &cqe);
+ if (ret) {
+ fprintf(stderr, "wait_cqe=%d\n", ret);
+ return T_EXIT_FAIL;
+ }
+
+ io_uring_queue_exit(&ring);
+ return T_EXIT_PASS;
+}