[2/2] syscalls/io_uring02: Use IOSQE_ASYNC when available

Signed-off-by: Martin Doucha <mdoucha@suse.cz>
---
 .../kernel/syscalls/io_uring/io_uring02.c     | 70 ++++++++++++++++---
 1 file changed, 62 insertions(+), 8 deletions(-)

Hi Martin,

[ Cc: io_uring folks ]

thanks a lot for your fix. Working well, patchset merged.

Kind regards,
Petr

> Signed-off-by: Martin Doucha <mdoucha@suse.cz>
> ---
>  .../kernel/syscalls/io_uring/io_uring02.c     | 70 ++++++++++++++++---
>  1 file changed, 62 insertions(+), 8 deletions(-)

> diff --git a/testcases/kernel/syscalls/io_uring/io_uring02.c b/testcases/kernel/syscalls/io_uring/io_uring02.c
> index 08f4a1714..cd90fbdc3 100644
> --- a/testcases/kernel/syscalls/io_uring/io_uring02.c
> +++ b/testcases/kernel/syscalls/io_uring/io_uring02.c
> @@ -79,11 +79,11 @@ static void setup(void)
>  	SAFE_CHROOT(CHROOT_DIR);
>  }

> -static void run(void)
> +static void drain_fallback(void)
>  {
>  	uint32_t i, count, tail;
>  	int beef_found = 0;
> -	struct io_uring_sqe *sqe_ptr;
> +	struct io_uring_sqe *sqe_ptr = uring.sqr_entries;
>  	const struct io_uring_cqe *cqe_ptr;

>  	SAFE_SOCKETPAIR(AF_UNIX, SOCK_DGRAM, 0, sockpair);
> @@ -91,9 +91,6 @@ static void run(void)
>  		32+sizeof(buf));
>  	SAFE_FCNTL(sockpair[0], F_SETFL, O_NONBLOCK);

> -	SAFE_IO_URING_INIT(512, &params, &uring);
> -	sqe_ptr = uring.sqr_entries;
> -
>  	/* Add spam requests to force async processing of the real test */
>  	for (i = 0, tail = *uring.sqr_tail; i < 255; i++, tail++, sqe_ptr++) {
>  		memset(sqe_ptr, 0, sizeof(*sqe_ptr));
> @@ -150,7 +147,7 @@ static void run(void)
>  			tst_res(TFAIL | TTERRNO,
>  				"Write outside chroot failed unexpectedly");
>  		} else {
> -			tst_res(TPASS,
> +			tst_res(TPASS | TTERRNO,
>  				"Write outside chroot failed as expected");
>  		}
>  	}
> @@ -163,12 +160,69 @@ static void run(void)
>  	if (count)
>  		tst_res(TFAIL, "Wrong number of entries in completion queue");

> -	/* iteration cleanup */
> -	SAFE_IO_URING_CLOSE(&uring);
>  	SAFE_CLOSE(sockpair[0]);
>  	SAFE_CLOSE(sockpair[1]);
>  }

> +static void check_result(void)
> +{
> +	const struct io_uring_cqe *cqe_ptr;
> +
> +	cqe_ptr = uring.cqr_entries + (*uring.cqr_head & *uring.cqr_mask);
> +	++*uring.cqr_head;
> +	TST_ERR = -cqe_ptr->res;
> +
> +	if (cqe_ptr->user_data != BEEF_MARK) {
> +		tst_res(TFAIL, "Unexpected entry in completion queue");
> +		return;
> +	}
> +
> +	if (cqe_ptr->res == -EINVAL) {
> +		tst_res(TINFO, "IOSQE_ASYNC is not supported, using fallback");
> +		drain_fallback();
> +		return;
> +	}
> +
> +	tst_res(TINFO, "IOSQE_ASYNC is supported");
> +
> +	if (cqe_ptr->res >= 0) {
> +		tst_res(TFAIL, "Write outside chroot succeeded.");
> +		return;
> +	}
> +
> +	if (cqe_ptr->res != -ENOENT) {
> +		tst_res(TFAIL | TTERRNO,
> +			"Write outside chroot failed unexpectedly");
> +		return;
> +	}
> +
> +	tst_res(TPASS | TTERRNO, "Write outside chroot failed as expected");
> +}
> +
> +static void run(void)
> +{
> +	uint32_t tail;
> +	struct io_uring_sqe *sqe_ptr;
> +
> +	SAFE_IO_URING_INIT(512, &params, &uring);
> +	sqe_ptr = uring.sqr_entries;
> +	tail = *uring.sqr_tail;
> +
> +	memset(sqe_ptr, 0, sizeof(*sqe_ptr));
> +	sqe_ptr->opcode = IORING_OP_SENDMSG;
> +	sqe_ptr->flags = IOSQE_ASYNC;
> +	sqe_ptr->fd = sendsock;
> +	sqe_ptr->addr = (__u64)&beef_header;
> +	sqe_ptr->user_data = BEEF_MARK;
> +	uring.sqr_array[tail & *uring.sqr_mask] = 0;
> +	tail++;
> +
> +	__atomic_store(uring.sqr_tail, &tail, __ATOMIC_RELEASE);
> +	SAFE_IO_URING_ENTER(1, uring.fd, 1, 1, IORING_ENTER_GETEVENTS, NULL);
> +	check_result();
> +	SAFE_IO_URING_CLOSE(&uring);
> +}
> +
>  static void cleanup(void)
>  {
>  	if (uring.fd >= 0)