@nxtedition/rocksdb 5.2.36 → 5.2.39

package/deps/liburing/liburing/test/file-update.c

@@ -0,0 +1,173 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Description: run various file registration tests
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+
+#include "helpers.h"
+#include "liburing.h"
+
+static void close_files(int *files, int nr_files, int add)
+{
+	char fname[32];
+	int i;
+
+	for (i = 0; i < nr_files; i++) {
+		if (files)
+			close(files[i]);
+		if (!add)
+			sprintf(fname, ".reg.%d", i);
+		else
+			sprintf(fname, ".add.%d", i + add);
+		unlink(fname);
+	}
+	if (files)
+		free(files);
+}
+
+static int *open_files(int nr_files, int extra, int add)
+{
+	char fname[32];
+	int *files;
+	int i;
+
+	files = t_calloc(nr_files + extra, sizeof(int));
+
+	for (i = 0; i < nr_files; i++) {
+		if (!add)
+			sprintf(fname, ".reg.%d", i);
+		else
+			sprintf(fname, ".add.%d", i + add);
+		files[i] = open(fname, O_RDWR | O_CREAT, 0644);
+		if (files[i] < 0) {
+			perror("open");
+			free(files);
+			files = NULL;
+			break;
+		}
+	}
+	if (extra) {
+		for (i = nr_files; i < nr_files + extra; i++)
+			files[i] = -1;
+	}
+
+	return files;
+}
+
+static int test_update_multiring(struct io_uring *r1, struct io_uring *r2,
+				 struct io_uring *r3, int do_unreg)
+{
+	int *fds, *newfds;
+
+	fds = open_files(10, 0, 0);
+	newfds = open_files(10, 0, 1);
+
+	if (io_uring_register_files(r1, fds, 10) ||
+	    io_uring_register_files(r2, fds, 10) ||
+	    io_uring_register_files(r3, fds, 10)) {
+		fprintf(stderr, "%s: register files failed\n", __FUNCTION__);
+		goto err;
+	}
+
+	if (io_uring_register_files_update(r1, 0, newfds, 10) != 10 ||
+	    io_uring_register_files_update(r2, 0, newfds, 10) != 10 ||
+	    io_uring_register_files_update(r3, 0, newfds, 10) != 10) {
+		fprintf(stderr, "%s: update files failed\n", __FUNCTION__);
+		goto err;
+	}
+
+	if (!do_unreg)
+		goto done;
+
+	if (io_uring_unregister_files(r1) ||
+	    io_uring_unregister_files(r2) ||
+	    io_uring_unregister_files(r3)) {
+		fprintf(stderr, "%s: unregister files failed\n", __FUNCTION__);
+		goto err;
+	}
+
+done:
+	close_files(fds, 10, 0);
+	close_files(newfds, 10, 1);
+	return 0;
+err:
+	close_files(fds, 10, 0);
+	close_files(newfds, 10, 1);
+	return 1;
+}
+
+static int test_sqe_update(struct io_uring *ring)
+{
+	struct io_uring_sqe *sqe;
+	struct io_uring_cqe *cqe;
+	int *fds, i, ret;
+
+	fds = t_malloc(sizeof(int) * 10);
+	for (i = 0; i < 10; i++)
+		fds[i] = -1;
+
+	sqe = io_uring_get_sqe(ring);
+	io_uring_prep_files_update(sqe, fds, 10, 0);
+	ret = io_uring_submit(ring);
+	if (ret != 1) {
+		fprintf(stderr, "submit: %d\n", ret);
+		return 1;
+	}
+
+	ret = io_uring_wait_cqe(ring, &cqe);
+	if (ret) {
+		fprintf(stderr, "wait: %d\n", ret);
+		return 1;
+	}
+
+	ret = cqe->res;
+	io_uring_cqe_seen(ring, cqe);
+	free(fds);
+	if (ret == -EINVAL) {
+		fprintf(stdout, "IORING_OP_FILES_UPDATE not supported, skipping\n");
+		return 0;
+	}
+	return ret != 10;
+}
+
+int main(int argc, char *argv[])
+{
+	struct io_uring r1, r2, r3;
+	int ret;
+
+	if (argc > 1)
+		return 0;
+
+	if (io_uring_queue_init(8, &r1, 0) ||
+	    io_uring_queue_init(8, &r2, 0) ||
+	    io_uring_queue_init(8, &r3, 0)) {
+		fprintf(stderr, "ring setup failed\n");
+		return 1;
+	}
+
+	ret = test_update_multiring(&r1, &r2, &r3, 1);
+	if (ret) {
+		fprintf(stderr, "test_update_multiring w/unreg\n");
+		return ret;
+	}
+
+	ret = test_update_multiring(&r1, &r2, &r3, 0);
+	if (ret) {
+		fprintf(stderr, "test_update_multiring wo/unreg\n");
+		return ret;
+	}
+
+	ret = test_sqe_update(&r1);
+	if (ret) {
+		fprintf(stderr, "test_sqe_update failed\n");
+		return ret;
+	}
+
+	return 0;
+}

package/deps/liburing/liburing/test/files-exit-hang-poll.c

@@ -0,0 +1,128 @@
+/* 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"
+
+#define BACKLOG 512
+
+#define PORT 9100
+
+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;
+	int i;
+
+	if (argc > 1)
+		return 0;
+
+	sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
+	if (sock_listen_fd < 0) {
+		perror("socket");
+		return 1;
+	}
+
+	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;
+
+	for (i = 0; i < 100; i++) {
+		serv_addr.sin_port = htons(PORT + i);
+
+		ret = bind(sock_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
+		if (!ret)
+			break;
+		if (errno != EADDRINUSE) {
+			fprintf(stderr, "bind: %s\n", strerror(errno));
+			return 1;
+		}
+		if (i == 99) {
+			printf("Gave up on finding a port, skipping\n");
+			goto out;
+		}
+	}
+
+	if (listen(sock_listen_fd, BACKLOG) < 0) {
+		perror("Error listening on socket\n");
+		return 1;
+	}
+
+	if (setup_io_uring())
+		return 1;
+
+	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 1;
+	}
+
+	signal(SIGALRM, alarm_sig);
+	alarm(1);
+
+	ret = io_uring_wait_cqe(&ring, &cqe);
+	if (ret) {
+		fprintf(stderr, "wait_cqe=%d\n", ret);
+		return 1;
+	}
+
+out:
+	io_uring_queue_exit(&ring);
+	return 0;
+}

package/deps/liburing/liburing/test/files-exit-hang-timeout.c

@@ -0,0 +1,134 @@
+/* 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"
+
+#define BACKLOG 512
+
+#define PORT 9100
+
+struct io_uring ring;
+
+struct __kernel_timespec ts = {
+	.tv_sec		= 300,
+	.tv_nsec	= 0,
+};
+
+static void add_timeout(struct io_uring *ring, int fd)
+{
+	struct io_uring_sqe *sqe;
+
+	sqe = io_uring_get_sqe(ring);
+	io_uring_prep_timeout(sqe, &ts, 100, 0);
+	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);
+	sqe->flags |= IOSQE_IO_LINK;
+}
+
+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;
+	int i;
+
+	if (argc > 1)
+		return 0;
+
+	sock_listen_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
+	if (sock_listen_fd < 0) {
+		perror("socket");
+		return 1;
+	}
+
+	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;
+
+	for (i = 0; i < 100; i++) {
+		serv_addr.sin_port = htons(PORT + i);
+
+		ret = bind(sock_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
+		if (!ret)
+			break;
+		if (errno != EADDRINUSE) {
+			fprintf(stderr, "bind: %s\n", strerror(errno));
+			return 1;
+		}
+		if (i == 99) {
+			printf("Gave up on finding a port, skipping\n");
+			goto out;
+		}
+	}
+
+	if (listen(sock_listen_fd, BACKLOG) < 0) {
+		perror("Error listening on socket\n");
+		return 1;
+	}
+
+	if (setup_io_uring())
+		return 1;
+
+	add_timeout(&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 1;
+	}
+
+	signal(SIGALRM, alarm_sig);
+	alarm(1);
+
+	ret = io_uring_wait_cqe(&ring, &cqe);
+	if (ret) {
+		fprintf(stderr, "wait_cqe=%d\n", ret);
+		return 1;
+	}
+
+out:
+	io_uring_queue_exit(&ring);
+	return 0;
+}

package/deps/liburing/liburing/test/fixed-link.c

@@ -0,0 +1,90 @@
+/* SPDX-License-Identifier: MIT */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+#include <sys/types.h>
+
+#include "helpers.h"
+#include "liburing.h"
+
+#define IOVECS_LEN 2
+
+int main(int argc, char *argv[])
+{
+	struct iovec iovecs[IOVECS_LEN];
+	struct io_uring ring;
+	int i, fd, ret;
+
+	if (argc > 1)
+		return 0;
+
+	fd = open("/dev/zero", O_RDONLY);
+	if (fd < 0) {
+		fprintf(stderr, "Failed to open /dev/zero\n");
+		return 1;
+	}
+
+	if (io_uring_queue_init(32, &ring, 0) < 0) {
+		fprintf(stderr, "Faild to init io_uring\n");
+		close(fd);
+		return 1;
+	}
+
+	for (i = 0; i < IOVECS_LEN; ++i) {
+		iovecs[i].iov_base = t_malloc(64);
+		iovecs[i].iov_len = 64;
+	};
+
+	ret = io_uring_register_buffers(&ring, iovecs, IOVECS_LEN);
+	if (ret) {
+		fprintf(stderr, "Failed to register buffers\n");
+		return 1;
+	}
+
+	for (i = 0; i < IOVECS_LEN; ++i) {
+		struct io_uring_sqe *sqe = io_uring_get_sqe(&ring);
+		const char *str = "#include <errno.h>";
+
+		iovecs[i].iov_len = strlen(str);
+		io_uring_prep_read_fixed(sqe, fd, iovecs[i].iov_base, strlen(str), 0, i);
+		if (i == 0)
+			io_uring_sqe_set_flags(sqe, IOSQE_IO_LINK);
+		io_uring_sqe_set_data(sqe, (void *)str);
+	}
+
+	ret = io_uring_submit_and_wait(&ring, IOVECS_LEN);
+	if (ret < 0) {
+		fprintf(stderr, "Failed to submit IO\n");
+		return 1;
+	} else if (ret < 2) {
+		fprintf(stderr, "Submitted %d, wanted %d\n", ret, IOVECS_LEN);
+		return 1;
+	}
+
+	for (i = 0; i < IOVECS_LEN; i++) {
+		struct io_uring_cqe *cqe;
+
+		ret = io_uring_wait_cqe(&ring, &cqe);
+		if (ret) {
+			fprintf(stderr, "wait_cqe=%d\n", ret);
+			return 1;
+		}
+		if (cqe->res != iovecs[i].iov_len) {
+			fprintf(stderr, "read: wanted %ld, got %d\n",
+					(long) iovecs[i].iov_len, cqe->res);
+			return 1;
+		}
+		io_uring_cqe_seen(&ring, cqe);
+	}
+
+	close(fd);
+	io_uring_queue_exit(&ring);
+
+	for (i = 0; i < IOVECS_LEN; ++i)
+		free(iovecs[i].iov_base);
+
+	return 0;
+}