ydb/contrib/libs/liburing/test/nop-all-sizes.c

#include "../config-host.h"
/* SPDX-License-Identifier: MIT */
/*
 * Description: exercise full filling of SQ and CQ ring
 *
 */
#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>

#include "liburing.h"

#define MAX_ENTRIES	32768

static int fill_nops(struct io_uring *ring)
{
	struct io_uring_sqe *sqe;
	int filled = 0;

	do {
		sqe = io_uring_get_sqe(ring);
		if (!sqe)
			break;

		io_uring_prep_nop(sqe);
		filled++;
	} while (1);

	return filled;
}

static int test_nops(struct io_uring *ring)
{
	struct io_uring_cqe *cqe;
	int ret, nr, total = 0, i;

	nr = fill_nops(ring);

	ret = io_uring_submit(ring);
	if (ret != nr) {
		fprintf(stderr, "submit %d, wanted %d\n", ret, nr);
		goto err;
	}
	total += ret;

	nr = fill_nops(ring);

	ret = io_uring_submit(ring);
	if (ret != nr) {
		fprintf(stderr, "submit %d, wanted %d\n", ret, nr);
		goto err;
	}
	total += ret;

	for (i = 0; i < total; i++) {
		ret = io_uring_wait_cqe(ring, &cqe);
		if (ret < 0) {
			fprintf(stderr, "wait completion %d\n", ret);
			goto err;
		}

		io_uring_cqe_seen(ring, cqe);
	}
	return 0;
err:
	return 1;
}

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

	if (argc > 1)
		return 0;

	depth = 1;
	while (depth <= MAX_ENTRIES) {
		ret = io_uring_queue_init(depth, &ring, 0);
		if (ret) {
			if (ret == -ENOMEM)
				break;
			fprintf(stderr, "ring setup failed: %d\n", ret);
			return 1;
		}

		ret = test_nops(&ring);
		if (ret) {
			fprintf(stderr, "test_single_nop failed\n");
			return ret;
		}
		depth <<= 1;
		io_uring_queue_exit(&ring);
	}

	return 0;
}