diff mbox series

[1/1] io_uring/rsrc: fix rogue rsrc node grabbing

Message ID 1202ede2d7bb90136e3482b2b84aad9ed483e5d6.1680098433.git.asml.silence@gmail.com (mailing list archive)
State New
Headers show
Series [1/1] io_uring/rsrc: fix rogue rsrc node grabbing | expand

Commit Message

Pavel Begunkov March 29, 2023, 2:03 p.m. UTC
We should not be looking at ctx->rsrc_node and anyhow modifying the node
without holding uring_lock, grabbing references in such a way is not
safe either.

Cc: stable@vger.kernel.org
Fixes: 5106dd6e74ab6 ("io_uring: propagate issue_flags state down to file assignment")
Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
---
 io_uring/rsrc.h | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

Comments

Jens Axboe March 29, 2023, 3:24 p.m. UTC | #1
On Wed, 29 Mar 2023 15:03:43 +0100, Pavel Begunkov wrote:
> We should not be looking at ctx->rsrc_node and anyhow modifying the node
> without holding uring_lock, grabbing references in such a way is not
> safe either.
> 
> 

Applied, thanks!

[1/1] io_uring/rsrc: fix rogue rsrc node grabbing
      commit: 4ff0b50de8cabba055efe50bbcb7506c41a69835

Best regards,
diff mbox series

Patch

diff --git a/io_uring/rsrc.h b/io_uring/rsrc.h
index 2b8743645efc..f27f4975217d 100644
--- a/io_uring/rsrc.h
+++ b/io_uring/rsrc.h
@@ -144,15 +144,13 @@  static inline void io_req_set_rsrc_node(struct io_kiocb *req,
 					unsigned int issue_flags)
 {
 	if (!req->rsrc_node) {
-		req->rsrc_node = ctx->rsrc_node;
+		io_ring_submit_lock(ctx, issue_flags);
 
-		if (!(issue_flags & IO_URING_F_UNLOCKED)) {
-			lockdep_assert_held(&ctx->uring_lock);
+		lockdep_assert_held(&ctx->uring_lock);
 
-			io_charge_rsrc_node(ctx);
-		} else {
-			percpu_ref_get(&req->rsrc_node->refs);
-		}
+		req->rsrc_node = ctx->rsrc_node;
+		io_charge_rsrc_node(ctx);
+		io_ring_submit_unlock(ctx, issue_flags);
 	}
 }