Commit dca58c6a authored by Pavel Begunkov's avatar Pavel Begunkov Committed by Jens Axboe

io_uring: refactor __io_sqe_files_scm

__io_sqe_files_scm() is now called only from one place passing a single
file, so nr argument can be killed and __io_sqe_files_scm() simplified.
Signed-off-by: default avatarPavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/66b492bc66dc8356d45d64076bb31d677d11a7c9.1649334991.git.asml.silence@gmail.comSigned-off-by: default avatarJens Axboe <axboe@kernel.dk>
parent a03a2a20
...@@ -8609,12 +8609,12 @@ static struct io_sq_data *io_get_sq_data(struct io_uring_params *p, ...@@ -8609,12 +8609,12 @@ static struct io_sq_data *io_get_sq_data(struct io_uring_params *p,
* files because otherwise they can't form a loop and so are not interesting * files because otherwise they can't form a loop and so are not interesting
* for GC. * for GC.
*/ */
static int __io_sqe_files_scm(struct io_ring_ctx *ctx, int nr, int offset) static int __io_sqe_files_scm(struct io_ring_ctx *ctx, int offset)
{ {
struct file *file = io_file_from_index(ctx, offset);
struct sock *sk = ctx->ring_sock->sk; struct sock *sk = ctx->ring_sock->sk;
struct scm_fp_list *fpl; struct scm_fp_list *fpl;
struct sk_buff *skb; struct sk_buff *skb;
int i, nr_files;
fpl = kzalloc(sizeof(*fpl), GFP_KERNEL); fpl = kzalloc(sizeof(*fpl), GFP_KERNEL);
if (!fpl) if (!fpl)
...@@ -8628,39 +8628,17 @@ static int __io_sqe_files_scm(struct io_ring_ctx *ctx, int nr, int offset) ...@@ -8628,39 +8628,17 @@ static int __io_sqe_files_scm(struct io_ring_ctx *ctx, int nr, int offset)
skb->sk = sk; skb->sk = sk;
nr_files = 0;
fpl->user = get_uid(current_user()); fpl->user = get_uid(current_user());
for (i = 0; i < nr; i++) { fpl->fp[0] = get_file(file);
struct file *file = io_file_from_index(ctx, i + offset); unix_inflight(fpl->user, file);
if (!file || !io_file_need_scm(file)) fpl->max = SCM_MAX_FD;
continue; fpl->count = 1;
UNIXCB(skb).fp = fpl;
fpl->fp[nr_files] = get_file(file); skb->destructor = unix_destruct_scm;
unix_inflight(fpl->user, fpl->fp[nr_files]); refcount_add(skb->truesize, &sk->sk_wmem_alloc);
nr_files++; skb_queue_head(&sk->sk_receive_queue, skb);
} fput(file);
if (nr_files) {
fpl->max = SCM_MAX_FD;
fpl->count = nr_files;
UNIXCB(skb).fp = fpl;
skb->destructor = unix_destruct_scm;
refcount_add(skb->truesize, &sk->sk_wmem_alloc);
skb_queue_head(&sk->sk_receive_queue, skb);
for (i = 0; i < nr; i++) {
struct file *file = io_file_from_index(ctx, i + offset);
if (file && io_file_need_scm(file))
fput(file);
}
} else {
kfree_skb(skb);
free_uid(fpl->user);
kfree(fpl);
}
return 0; return 0;
} }
#endif #endif
...@@ -8904,7 +8882,7 @@ static int io_sqe_file_register(struct io_ring_ctx *ctx, struct file *file, ...@@ -8904,7 +8882,7 @@ static int io_sqe_file_register(struct io_ring_ctx *ctx, struct file *file,
return 0; return 0;
} }
return __io_sqe_files_scm(ctx, 1, index); return __io_sqe_files_scm(ctx, index);
#else #else
return 0; return 0;
#endif #endif
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment