io_uring: do init work in grab_env()

Place io_req_init_async() in io_req_work_grab_env() so it won't be
forgotten.

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
Signed-off-by: Jens Axboe <axboe@kernel.dk>
This commit is contained in:
Pavel Begunkov 2020-06-29 19:18:41 +03:00 committed by Jens Axboe
parent 351fd53595
commit edcdfcc149
1 changed files with 3 additions and 5 deletions

View File

@ -1105,6 +1105,8 @@ static inline void io_req_work_grab_env(struct io_kiocb *req)
{ {
const struct io_op_def *def = &io_op_defs[req->opcode]; const struct io_op_def *def = &io_op_defs[req->opcode];
io_req_init_async(req);
if (!req->work.mm && def->needs_mm) { if (!req->work.mm && def->needs_mm) {
mmgrab(current->mm); mmgrab(current->mm);
req->work.mm = current->mm; req->work.mm = current->mm;
@ -1161,9 +1163,7 @@ static inline void io_prep_async_work(struct io_kiocb *req,
req->work.flags |= IO_WQ_WORK_UNBOUND; req->work.flags |= IO_WQ_WORK_UNBOUND;
} }
io_req_init_async(req);
io_req_work_grab_env(req); io_req_work_grab_env(req);
*link = io_prep_linked_timeout(req); *link = io_prep_linked_timeout(req);
} }
@ -5254,10 +5254,8 @@ static int io_req_defer_prep(struct io_kiocb *req,
return ret; return ret;
} }
if (for_async || (req->flags & REQ_F_WORK_INITIALIZED)) { if (for_async || (req->flags & REQ_F_WORK_INITIALIZED))
io_req_init_async(req);
io_req_work_grab_env(req); io_req_work_grab_env(req);
}
switch (req->opcode) { switch (req->opcode) {
case IORING_OP_NOP: case IORING_OP_NOP: