test/file-register.c

/*
 * 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 "liburing.h"

static int no_update = 0;

static void close_files(int *files, int nr_files, int add)
{
	char fname[32];
	int i;

	for (i = 0; i < nr_files; i++) {
		close(files[i]);
		if (!add)
			sprintf(fname, ".reg.%d", i);
		else
			sprintf(fname, ".add.%d", i);
		unlink(fname);
	}
	free(files);
}

static int *open_files(int nr_files, int extra, int add)
{
	char fname[32];
	int *files;
	int i;

	files = 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);
		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_replace_all(struct io_uring *ring)
{
	struct io_uring_files_update up;
	int *files;
	int ret, i;

	files = open_files(100, 0, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 100);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	up.fds = malloc(100 * sizeof(int));
	for (i = 0; i < 100; i++)
		up.fds[i] = -1;
	up.offset = 0;

	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES_UPDATE, &up, 100);
	if (ret != 100) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	close_files(files, 100, 0);
	return 0;
err:
	close_files(files, 100, 0);
	return 1;
}

static int test_replace(struct io_uring *ring)
{
	struct io_uring_files_update up;
	int *files;
	int ret;

	files = open_files(100, 0, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 100);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	up.fds = open_files(10, 0, 1);
	up.offset = 90;

	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES_UPDATE, &up, 10);
	if (ret != 10) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	close_files(files, 100, 0);
	close_files(up.fds, 10, 1);
	return 0;
err:
	close_files(files, 100, 0);
	close_files(up.fds, 10, 1);
	return 1;
}

static int test_removals(struct io_uring *ring)
{
	struct io_uring_files_update up;
	int *files;
	int ret, i;

	files = open_files(100, 0, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 100);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	up.fds = calloc(10, sizeof(int));
	for (i = 0; i < 10; i++)
		up.fds[i] = -1;
	up.offset = 50;

	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES_UPDATE, &up, 10);
	if (ret != 10) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	close_files(files, 100, 0);
	return 0;
err:
	close_files(files, 100, 0);
	return 1;
}

static int test_additions(struct io_uring *ring)
{
	struct io_uring_files_update up;
	int *files;
	int ret;

	files = open_files(100, 100, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 200);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	up.fds = open_files(2, 0, 1);
	up.offset = 100;
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES_UPDATE, &up, 2);
	if (ret != 2) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}

	close_files(files, 100, 0);
	close_files(up.fds, 2, 1);
	return 0;
err:
	close_files(files, 100, 0);
	close_files(up.fds, 2, 1);
	return 1;
}

static int test_sparse(struct io_uring *ring)
{
	int *files;
	int ret;

	files = open_files(100, 100, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 200);
	if (ret) {
		if (errno == EBADF) {
			printf("Sparse files not supported\n");
			no_update = 1;
			goto done;
		}
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}
	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret) {
		printf("ret=%d, errno=%d\n", ret, errno);
		goto err;
	}
done:
	close_files(files, 100, 0);
	return 0;
err:
	close_files(files, 100, 0);
	return 1;
}

static int test_basic_many(struct io_uring *ring)
{
	int *files;
	int ret;

	files = open_files(768, 0, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 768);
	if (ret)
		goto err;
	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret)
		goto err;
	close_files(files, 768, 0);
	return 0;
err:
	close_files(files, 768, 0);
	return 1;
}

static int test_basic(struct io_uring *ring)
{
	int *files;
	int ret;

	files = open_files(100, 0, 0);
	ret = io_uring_register(ring->ring_fd, IORING_REGISTER_FILES, files, 100);
	if (ret)
		goto err;
	ret = io_uring_register(ring->ring_fd, IORING_UNREGISTER_FILES, NULL, 0);
	if (ret)
		goto err;
	close_files(files, 100, 0);
	return 0;
err:
	close_files(files, 100, 0);
	return 1;
}

int main(int argc, char *argv[])
{
	struct io_uring ring;
	int ret;

	ret = io_uring_queue_init(8, &ring, 0);
	if (ret) {
		printf("ring setup failed\n");
		return 1;
	}

	ret = test_basic(&ring);
	if (ret) {
		printf("test_basic failed\n");
		return ret;
	}

	ret = test_basic_many(&ring);
	if (ret) {
		printf("test_basic_many failed\n");
		return ret;
	}

	ret = test_sparse(&ring);
	if (ret) {
		printf("test_sparse failed\n");
		return ret;
	}

	if (no_update)
		return 0;

	ret = test_additions(&ring);
	if (ret) {
		printf("test_additions failed\n");
		return ret;
	}

	ret = test_removals(&ring);
	if (ret) {
		printf("test_removals failed\n");
		return ret;
	}

	ret = test_replace(&ring);
	if (ret) {
		printf("test_replace failed\n");
		return ret;
	}

	ret = test_replace_all(&ring);
	if (ret) {
		printf("test_replace_all failed\n");
		return ret;
	}

	return 0;
}