Skip to content

Commit

Permalink
io_uring: always lock __io_cqring_overflow_flush
Browse files Browse the repository at this point in the history
Conditional locking is never great, in case of
__io_cqring_overflow_flush(), which is a slow path, it's not justified.
Don't handle IOPOLL separately, always grab uring_lock for overflow
flushing.

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/162947df299aa12693ac4b305dacedab32ec7976.1712708261.git.asml.silence@gmail.com
Signed-off-by: Jens Axboe <axboe@kernel.dk>
  • Loading branch information
isilence authored and axboe committed Apr 15, 2024
1 parent 408024b commit 8d09a88
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions io_uring/io_uring.c
Original file line number Diff line number Diff line change
Expand Up @@ -673,6 +673,8 @@ static void io_cqring_overflow_kill(struct io_ring_ctx *ctx)
struct io_overflow_cqe *ocqe;
LIST_HEAD(list);

lockdep_assert_held(&ctx->uring_lock);

spin_lock(&ctx->completion_lock);
list_splice_init(&ctx->cq_overflow_list, &list);
clear_bit(IO_CHECK_CQ_OVERFLOW_BIT, &ctx->check_cq);
Expand All @@ -689,6 +691,8 @@ static void __io_cqring_overflow_flush(struct io_ring_ctx *ctx)
{
size_t cqe_size = sizeof(struct io_uring_cqe);

lockdep_assert_held(&ctx->uring_lock);

if (__io_cqring_events(ctx) == ctx->cq_entries)
return;

Expand Down Expand Up @@ -718,12 +722,9 @@ static void __io_cqring_overflow_flush(struct io_ring_ctx *ctx)

static void io_cqring_do_overflow_flush(struct io_ring_ctx *ctx)
{
/* iopoll syncs against uring_lock, not completion_lock */
if (ctx->flags & IORING_SETUP_IOPOLL)
mutex_lock(&ctx->uring_lock);
mutex_lock(&ctx->uring_lock);
__io_cqring_overflow_flush(ctx);
if (ctx->flags & IORING_SETUP_IOPOLL)
mutex_unlock(&ctx->uring_lock);
mutex_unlock(&ctx->uring_lock);
}

/* can be called by any task */
Expand Down Expand Up @@ -1522,6 +1523,8 @@ static int io_iopoll_check(struct io_ring_ctx *ctx, long min)
unsigned int nr_events = 0;
unsigned long check_cq;

lockdep_assert_held(&ctx->uring_lock);

if (!io_allowed_run_tw(ctx))
return -EEXIST;

Expand Down

0 comments on commit 8d09a88

Please sign in to comment.