mirror of
https://github.com/AuxXxilium/linux_dsm_epyc7002.git
synced 2024-11-24 09:40:58 +07:00
io_uring: file set registration should use interruptible waits
If an application attempts to register a set with unbounded requests pending, we can be stuck here forever if they don't complete. We can make this wait interruptible, and just abort if we get signaled. Signed-off-by: Jens Axboe <axboe@kernel.dk>
This commit is contained in:
parent
96fd84d83a
commit
c150368b49
@ -6524,8 +6524,13 @@ static int __io_uring_register(struct io_ring_ctx *ctx, unsigned opcode,
|
|||||||
* after we've killed the percpu ref.
|
* after we've killed the percpu ref.
|
||||||
*/
|
*/
|
||||||
mutex_unlock(&ctx->uring_lock);
|
mutex_unlock(&ctx->uring_lock);
|
||||||
wait_for_completion(&ctx->completions[0]);
|
ret = wait_for_completion_interruptible(&ctx->completions[0]);
|
||||||
mutex_lock(&ctx->uring_lock);
|
mutex_lock(&ctx->uring_lock);
|
||||||
|
if (ret) {
|
||||||
|
percpu_ref_resurrect(&ctx->refs);
|
||||||
|
ret = -EINTR;
|
||||||
|
goto out;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (opcode) {
|
switch (opcode) {
|
||||||
@ -6571,8 +6576,9 @@ static int __io_uring_register(struct io_ring_ctx *ctx, unsigned opcode,
|
|||||||
if (opcode != IORING_UNREGISTER_FILES &&
|
if (opcode != IORING_UNREGISTER_FILES &&
|
||||||
opcode != IORING_REGISTER_FILES_UPDATE) {
|
opcode != IORING_REGISTER_FILES_UPDATE) {
|
||||||
/* bring the ctx back to life */
|
/* bring the ctx back to life */
|
||||||
reinit_completion(&ctx->completions[0]);
|
|
||||||
percpu_ref_reinit(&ctx->refs);
|
percpu_ref_reinit(&ctx->refs);
|
||||||
|
out:
|
||||||
|
reinit_completion(&ctx->completions[0]);
|
||||||
}
|
}
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user