Commit 2f71d238 authored by Pavel Begunkov's avatar Pavel Begunkov Committed by Jens Axboe
Browse files

io_uring/mock: allow to choose FMODE_NOWAIT 



Add an option to choose whether the file supports FMODE_NOWAIT, that
changes the execution path io_uring request takes.

Signed-off-by: default avatarPavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/1e532565b05a05b23589d237c24ee1a3d90c2fd9.1750599274.git.asml.silence@gmail.com


Signed-off-by: default avatarJens Axboe <axboe@kernel.dk>
parent d1aa0346

include/uapi/linux/io_uring/mock_file.h

+5 −0
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ 
enum {

	IORING_MOCK_FEAT_CMD_COPY,

	IORING_MOCK_FEAT_RW_ZERO,

	IORING_MOCK_FEAT_RW_NOWAIT,



	IORING_MOCK_FEAT_END,

};
@@ -15,6 +16,10 @@ struct io_uring_mock_probe { 
	__u64		__resv[9];

};



enum {

	IORING_MOCK_CREATE_F_SUPPORT_NOWAIT			= 1,

};



struct io_uring_mock_create {

	__u32		out_fd;

	__u32		flags;

io_uring/mock_file.c

+7 −1
Original line number Diff line number Diff line
@@ -131,6 +131,8 @@ static const struct file_operations io_mock_fops = { 
	.llseek		= io_mock_llseek,

};



#define IO_VALID_CREATE_FLAGS (IORING_MOCK_CREATE_F_SUPPORT_NOWAIT)



static int io_create_mock_file(struct io_uring_cmd *cmd, unsigned int issue_flags)

{

	const struct io_uring_sqe *sqe = cmd->sqe;
@@ -157,7 +159,9 @@ static int io_create_mock_file(struct io_uring_cmd *cmd, unsigned int issue_flag 
	memset(&mc, 0, sizeof(mc));

	if (copy_from_user(&mc, uarg, uarg_size))

		return -EFAULT;

	if (!mem_is_zero(mc.__resv, sizeof(mc.__resv)) || mc.flags)

	if (!mem_is_zero(mc.__resv, sizeof(mc.__resv)))

		return -EINVAL;

	if (mc.flags & ~IO_VALID_CREATE_FLAGS)

		return -EINVAL;

	if (mc.file_size > SZ_1G)

		return -EINVAL;
@@ -180,6 +184,8 @@ static int io_create_mock_file(struct io_uring_cmd *cmd, unsigned int issue_flag 
	file->f_mode |= FMODE_READ | FMODE_CAN_READ |

			FMODE_WRITE | FMODE_CAN_WRITE |

			FMODE_LSEEK;

	if (mc.flags & IORING_MOCK_CREATE_F_SUPPORT_NOWAIT)

		file->f_mode |= FMODE_NOWAIT;



	mc.out_fd = fd;

	if (copy_to_user(uarg, &mc, uarg_size)) {