Commit 0396ad37 authored by Pavel Begunkov's avatar Pavel Begunkov Committed by Jens Axboe
Browse files

io_uring: cap cached iovec/bvec size 



Bvecs can be large, put an arbitrary limit on the max vector size it
can cache.

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


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

io_uring/net.c

+3 −0
Original line number Diff line number Diff line
@@ -152,6 +152,9 @@ static void io_netmsg_recycle(struct io_kiocb *req, unsigned int issue_flags) 


	/* Let normal cleanup path reap it if we fail adding to the cache */

	io_alloc_cache_vec_kasan(&hdr->vec);

	if (hdr->vec.nr > IO_VEC_CACHE_SOFT_CAP)

		io_vec_free(&hdr->vec);



	if (io_alloc_cache_put(&req->ctx->netmsg_cache, hdr)) {

		req->async_data = NULL;

		req->flags &= ~REQ_F_ASYNC_DATA;

io_uring/rsrc.h

+2 −0
Original line number Diff line number Diff line
@@ -5,6 +5,8 @@ 
#include <linux/io_uring_types.h>

#include <linux/lockdep.h>



#define IO_VEC_CACHE_SOFT_CAP		256



enum {

	IORING_RSRC_FILE		= 0,

	IORING_RSRC_BUFFER		= 1,

io_uring/rw.c

+3 −0
Original line number Diff line number Diff line
@@ -150,6 +150,9 @@ static void io_rw_recycle(struct io_kiocb *req, unsigned int issue_flags) 
		return;



	io_alloc_cache_vec_kasan(&rw->vec);

	if (rw->vec.nr > IO_VEC_CACHE_SOFT_CAP)

		io_vec_free(&rw->vec);



	if (io_alloc_cache_put(&req->ctx->rw_cache, rw)) {

		req->async_data = NULL;

		req->flags &= ~REQ_F_ASYNC_DATA;