mirror of
https://mirrors.bfsu.edu.cn/git/linux.git
synced 2024-12-28 21:45:01 +08:00
io_uring: add missing lock in io_get_file_fixed
io_get_file_fixed will access io_uring's context. Lock it if it is invoked unlocked (eg via io-wq) to avoid a race condition with fixed files getting unregistered. No single upstream patch exists for this issue, it was fixed as part of the file assignment changes that went into the 5.18 cycle. Signed-off-by: Jheng, Bing-Jhong Billy <billy@starlabs.sg> Signed-off-by: Jens Axboe <axboe@kernel.dk> Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
This commit is contained in:
parent
7735809333
commit
cf7f9cd500
@ -1095,7 +1095,8 @@ static int __io_register_rsrc_update(struct io_ring_ctx *ctx, unsigned type,
|
||||
unsigned nr_args);
|
||||
static void io_clean_op(struct io_kiocb *req);
|
||||
static struct file *io_file_get(struct io_ring_ctx *ctx,
|
||||
struct io_kiocb *req, int fd, bool fixed);
|
||||
struct io_kiocb *req, int fd, bool fixed,
|
||||
unsigned int issue_flags);
|
||||
static void __io_queue_sqe(struct io_kiocb *req);
|
||||
static void io_rsrc_put_work(struct work_struct *work);
|
||||
|
||||
@ -4121,7 +4122,7 @@ static int io_tee(struct io_kiocb *req, unsigned int issue_flags)
|
||||
return -EAGAIN;
|
||||
|
||||
in = io_file_get(req->ctx, req, sp->splice_fd_in,
|
||||
(sp->flags & SPLICE_F_FD_IN_FIXED));
|
||||
(sp->flags & SPLICE_F_FD_IN_FIXED), issue_flags);
|
||||
if (!in) {
|
||||
ret = -EBADF;
|
||||
goto done;
|
||||
@ -4161,7 +4162,7 @@ static int io_splice(struct io_kiocb *req, unsigned int issue_flags)
|
||||
return -EAGAIN;
|
||||
|
||||
in = io_file_get(req->ctx, req, sp->splice_fd_in,
|
||||
(sp->flags & SPLICE_F_FD_IN_FIXED));
|
||||
(sp->flags & SPLICE_F_FD_IN_FIXED), issue_flags);
|
||||
if (!in) {
|
||||
ret = -EBADF;
|
||||
goto done;
|
||||
@ -7047,13 +7048,16 @@ static void io_fixed_file_set(struct io_fixed_file *file_slot, struct file *file
|
||||
}
|
||||
|
||||
static inline struct file *io_file_get_fixed(struct io_ring_ctx *ctx,
|
||||
struct io_kiocb *req, int fd)
|
||||
struct io_kiocb *req, int fd,
|
||||
unsigned int issue_flags)
|
||||
{
|
||||
struct file *file;
|
||||
struct file *file = NULL;
|
||||
unsigned long file_ptr;
|
||||
|
||||
io_ring_submit_lock(ctx, !(issue_flags & IO_URING_F_NONBLOCK));
|
||||
|
||||
if (unlikely((unsigned int)fd >= ctx->nr_user_files))
|
||||
return NULL;
|
||||
goto out;
|
||||
fd = array_index_nospec(fd, ctx->nr_user_files);
|
||||
file_ptr = io_fixed_file_slot(&ctx->file_table, fd)->file_ptr;
|
||||
file = (struct file *) (file_ptr & FFS_MASK);
|
||||
@ -7061,6 +7065,8 @@ static inline struct file *io_file_get_fixed(struct io_ring_ctx *ctx,
|
||||
/* mask in overlapping REQ_F and FFS bits */
|
||||
req->flags |= (file_ptr << REQ_F_NOWAIT_READ_BIT);
|
||||
io_req_set_rsrc_node(req);
|
||||
out:
|
||||
io_ring_submit_unlock(ctx, !(issue_flags & IO_URING_F_NONBLOCK));
|
||||
return file;
|
||||
}
|
||||
|
||||
@ -7078,10 +7084,11 @@ static struct file *io_file_get_normal(struct io_ring_ctx *ctx,
|
||||
}
|
||||
|
||||
static inline struct file *io_file_get(struct io_ring_ctx *ctx,
|
||||
struct io_kiocb *req, int fd, bool fixed)
|
||||
struct io_kiocb *req, int fd, bool fixed,
|
||||
unsigned int issue_flags)
|
||||
{
|
||||
if (fixed)
|
||||
return io_file_get_fixed(ctx, req, fd);
|
||||
return io_file_get_fixed(ctx, req, fd, issue_flags);
|
||||
else
|
||||
return io_file_get_normal(ctx, req, fd);
|
||||
}
|
||||
@ -7303,7 +7310,7 @@ static int io_init_req(struct io_ring_ctx *ctx, struct io_kiocb *req,
|
||||
|
||||
if (io_op_defs[req->opcode].needs_file) {
|
||||
req->file = io_file_get(ctx, req, READ_ONCE(sqe->fd),
|
||||
(sqe_flags & IOSQE_FIXED_FILE));
|
||||
(sqe_flags & IOSQE_FIXED_FILE), 0);
|
||||
if (unlikely(!req->file))
|
||||
ret = -EBADF;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user