Commit 40b99183 authored by Pavel Begunkov's avatar Pavel Begunkov Committed by Jens Axboe
Browse files

io_uring: deduplicate caches deallocation 



Add a function that frees all ring caches since we already have two
spots repeating the same thing and it's easy to miss it and change only
one of them.

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


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

io_uring/io_uring.c

+12 −12
Original line number Diff line number Diff line
@@ -282,6 +282,16 @@ static int io_alloc_hash_table(struct io_hash_table *table, unsigned bits) 
	return 0;

}



static void io_free_alloc_caches(struct io_ring_ctx *ctx)

{

	io_alloc_cache_free(&ctx->apoll_cache, kfree);

	io_alloc_cache_free(&ctx->netmsg_cache, io_netmsg_cache_free);

	io_alloc_cache_free(&ctx->rw_cache, io_rw_cache_free);

	io_alloc_cache_free(&ctx->uring_cache, kfree);

	io_alloc_cache_free(&ctx->msg_cache, kfree);

	io_futex_cache_free(ctx);

}



static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p)

{

	struct io_ring_ctx *ctx;
@@ -360,12 +370,7 @@ static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p) 
free_ref:

	percpu_ref_exit(&ctx->refs);

err:

	io_alloc_cache_free(&ctx->apoll_cache, kfree);

	io_alloc_cache_free(&ctx->netmsg_cache, io_netmsg_cache_free);

	io_alloc_cache_free(&ctx->rw_cache, io_rw_cache_free);

	io_alloc_cache_free(&ctx->uring_cache, kfree);

	io_alloc_cache_free(&ctx->msg_cache, kfree);

	io_futex_cache_free(ctx);

	io_free_alloc_caches(ctx);

	kvfree(ctx->cancel_table.hbs);

	xa_destroy(&ctx->io_bl_xa);

	kfree(ctx);
@@ -2702,12 +2707,7 @@ static __cold void io_ring_ctx_free(struct io_ring_ctx *ctx) 
	io_sqe_files_unregister(ctx);

	io_cqring_overflow_kill(ctx);

	io_eventfd_unregister(ctx);

	io_alloc_cache_free(&ctx->apoll_cache, kfree);

	io_alloc_cache_free(&ctx->netmsg_cache, io_netmsg_cache_free);

	io_alloc_cache_free(&ctx->rw_cache, io_rw_cache_free);

	io_alloc_cache_free(&ctx->uring_cache, kfree);

	io_alloc_cache_free(&ctx->msg_cache, kfree);

	io_futex_cache_free(ctx);

	io_free_alloc_caches(ctx);

	io_destroy_buffers(ctx);

	io_free_region(ctx, &ctx->param_region);

	mutex_unlock(&ctx->uring_lock);