/* SPDX-License-Identifier: MIT */
/*
 * Description: Single depth submit+wait poll hang test
 *
 */
#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>

#include "liburing.h"

#define BLOCKS	4096

int main(int argc, char *argv[])
{
	struct io_uring ring;
	struct io_uring_sqe *sqe;
	struct io_uring_cqe *cqe;
	struct iovec iov;
	char buf[32];
	off_t offset;
	unsigned blocks;
	int ret, fd;

	if (posix_memalign(&iov.iov_base, 4096, 4096)) {
		fprintf(stderr, "memalign failed\n");
		return 1;
	}
	iov.iov_len = 4096;

	ret = io_uring_queue_init(1, &ring, IORING_SETUP_IOPOLL);
	if (ret) {
		fprintf(stderr, "ring setup failed\n");
		return 1;

	}

	sprintf(buf, "./XXXXXX");
	fd = mkostemp(buf, O_WRONLY | O_DIRECT | O_CREAT);
	if (fd < 0) {
		perror("mkostemp");
		return 1;
	}

	offset = 0;
	blocks = BLOCKS;
	do {
		sqe = io_uring_get_sqe(&ring);
		if (!sqe) {
			fprintf(stderr, "get sqe failed\n");
			goto err;
		}
		io_uring_prep_writev(sqe, fd, &iov, 1, offset);
		ret = io_uring_submit_and_wait(&ring, 1);
		if (ret < 0) {
			fprintf(stderr, "submit_and_wait: %d\n", ret);
			goto err;
		}
		ret = io_uring_wait_cqe(&ring, &cqe);
		if (ret < 0) {
			fprintf(stderr, "wait completion: %d\n", ret);
			goto err;
		}
		if (cqe->res != 4096) {
			if (cqe->res == -EOPNOTSUPP)
				goto skipped;
			goto err;
		}
		io_uring_cqe_seen(&ring, cqe);
		offset += 4096;
	} while (--blocks);
		
	close(fd);
	unlink(buf);
	return 0;
err:
	close(fd);
	unlink(buf);
	return 1;
skipped:
	fprintf(stderr, "Polling not supported in current dir, test skipped\n");
	close(fd);
	unlink(buf);
	return 0;
}