* [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
@ 2022-06-08 11:12 Hao Xu
2022-06-08 11:27 ` Pavel Begunkov
2022-06-10 10:21 ` Pavel Begunkov
0 siblings, 2 replies; 9+ messages in thread
From: Hao Xu @ 2022-06-08 11:12 UTC (permalink / raw)
To: io-uring; +Cc: Jens Axboe, Pavel Begunkov
From: Hao Xu <[email protected]>
Add a new io_hash_bucket structure so that each bucket in cancel_hash
has separate spinlock. Use per entry lock for cancel_hash, this removes
some completion lock invocation and remove contension between different
cancel_hash entries.
Signed-off-by: Hao Xu <[email protected]>
---
v1->v2:
- Add per entry lock for poll/apoll task work code which was missed
in v1
- add an member in io_kiocb to track req's indice in cancel_hash
v2->v3:
- make struct io_hash_bucket align with cacheline to avoid cacheline
false sharing.
- re-calculate hash value when deleting an entry from cancel_hash.
(cannot leverage struct io_poll to store the indice since it's
already 64 Bytes)
io_uring/cancel.c | 14 +++++++--
io_uring/cancel.h | 6 ++++
io_uring/fdinfo.c | 9 ++++--
io_uring/io_uring.c | 8 +++--
io_uring/io_uring_types.h | 2 +-
io_uring/poll.c | 64 +++++++++++++++++++++------------------
6 files changed, 65 insertions(+), 38 deletions(-)
diff --git a/io_uring/cancel.c b/io_uring/cancel.c
index 83cceb52d82d..bced5d6b9294 100644
--- a/io_uring/cancel.c
+++ b/io_uring/cancel.c
@@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
if (!ret)
return 0;
- spin_lock(&ctx->completion_lock);
ret = io_poll_cancel(ctx, cd);
if (ret != -ENOENT)
goto out;
+ spin_lock(&ctx->completion_lock);
if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
ret = io_timeout_cancel(ctx, cd);
-out:
spin_unlock(&ctx->completion_lock);
+out:
return ret;
}
@@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
io_req_set_res(req, ret, 0);
return IOU_OK;
}
+
+inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
+{
+ unsigned int i;
+
+ for (i = 0; i < size; i++) {
+ spin_lock_init(&hash_table[i].lock);
+ INIT_HLIST_HEAD(&hash_table[i].list);
+ }
+}
diff --git a/io_uring/cancel.h b/io_uring/cancel.h
index 4f35d8696325..b57d6706f84d 100644
--- a/io_uring/cancel.h
+++ b/io_uring/cancel.h
@@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe);
int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
+inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size);
+
+struct io_hash_bucket {
+ spinlock_t lock;
+ struct hlist_head list;
+} ____cacheline_aligned_in_smp;
diff --git a/io_uring/fdinfo.c b/io_uring/fdinfo.c
index fcedde4b4b1e..f941c73f5502 100644
--- a/io_uring/fdinfo.c
+++ b/io_uring/fdinfo.c
@@ -13,6 +13,7 @@
#include "io_uring.h"
#include "sqpoll.h"
#include "fdinfo.h"
+#include "cancel.h"
#ifdef CONFIG_PROC_FS
static __cold int io_uring_show_cred(struct seq_file *m, unsigned int id,
@@ -157,17 +158,19 @@ static __cold void __io_uring_show_fdinfo(struct io_ring_ctx *ctx,
mutex_unlock(&ctx->uring_lock);
seq_puts(m, "PollList:\n");
- spin_lock(&ctx->completion_lock);
for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
- struct hlist_head *list = &ctx->cancel_hash[i];
+ struct io_hash_bucket *hb = &ctx->cancel_hash[i];
struct io_kiocb *req;
- hlist_for_each_entry(req, list, hash_node)
+ spin_lock(&hb->lock);
+ hlist_for_each_entry(req, &hb->list, hash_node)
seq_printf(m, " op=%d, task_works=%d\n", req->opcode,
task_work_pending(req->task));
+ spin_unlock(&hb->lock);
}
seq_puts(m, "CqOverflowList:\n");
+ spin_lock(&ctx->completion_lock);
list_for_each_entry(ocqe, &ctx->cq_overflow_list, list) {
struct io_uring_cqe *cqe = &ocqe->cqe;
diff --git a/io_uring/io_uring.c b/io_uring/io_uring.c
index 1572ebe3cff1..b67ab76b9e56 100644
--- a/io_uring/io_uring.c
+++ b/io_uring/io_uring.c
@@ -725,11 +725,13 @@ static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p)
if (hash_bits <= 0)
hash_bits = 1;
ctx->cancel_hash_bits = hash_bits;
- ctx->cancel_hash = kmalloc((1U << hash_bits) * sizeof(struct hlist_head),
- GFP_KERNEL);
+ ctx->cancel_hash =
+ kmalloc((1U << hash_bits) * sizeof(struct io_hash_bucket),
+ GFP_KERNEL);
if (!ctx->cancel_hash)
goto err;
- __hash_init(ctx->cancel_hash, 1U << hash_bits);
+
+ init_hash_table(ctx->cancel_hash, 1U << hash_bits);
ctx->dummy_ubuf = kzalloc(sizeof(*ctx->dummy_ubuf), GFP_KERNEL);
if (!ctx->dummy_ubuf)
diff --git a/io_uring/io_uring_types.h b/io_uring/io_uring_types.h
index 7c22cf35a7e2..b67843a5f3b6 100644
--- a/io_uring/io_uring_types.h
+++ b/io_uring/io_uring_types.h
@@ -230,7 +230,7 @@ struct io_ring_ctx {
* manipulate the list, hence no extra locking is needed there.
*/
struct io_wq_work_list iopoll_list;
- struct hlist_head *cancel_hash;
+ struct io_hash_bucket *cancel_hash;
unsigned cancel_hash_bits;
bool poll_multi_queue;
diff --git a/io_uring/poll.c b/io_uring/poll.c
index 0df5eca93b16..515f1727e3c6 100644
--- a/io_uring/poll.c
+++ b/io_uring/poll.c
@@ -19,6 +19,7 @@
#include "opdef.h"
#include "kbuf.h"
#include "poll.h"
+#include "cancel.h"
struct io_poll_update {
struct file *file;
@@ -73,10 +74,22 @@ static struct io_poll *io_poll_get_single(struct io_kiocb *req)
static void io_poll_req_insert(struct io_kiocb *req)
{
struct io_ring_ctx *ctx = req->ctx;
- struct hlist_head *list;
+ u32 index = hash_long(req->cqe.user_data, ctx->cancel_hash_bits);
+ struct io_hash_bucket *hb = &ctx->cancel_hash[index];
- list = &ctx->cancel_hash[hash_long(req->cqe.user_data, ctx->cancel_hash_bits)];
- hlist_add_head(&req->hash_node, list);
+ spin_lock(&hb->lock);
+ hlist_add_head(&req->hash_node, &hb->list);
+ spin_unlock(&hb->lock);
+}
+
+static void io_poll_req_delete(struct io_kiocb *req, struct io_ring_ctx *ctx)
+{
+ u32 index = hash_long(req->cqe.user_data, ctx->cancel_hash_bits);
+ spinlock_t *lock = &ctx->cancel_hash[index].lock;
+
+ spin_lock(lock);
+ hash_del(&req->hash_node);
+ spin_unlock(lock);
}
static void io_init_poll_iocb(struct io_poll *poll, __poll_t events,
@@ -220,8 +233,8 @@ static void io_poll_task_func(struct io_kiocb *req, bool *locked)
}
io_poll_remove_entries(req);
+ io_poll_req_delete(req, ctx);
spin_lock(&ctx->completion_lock);
- hash_del(&req->hash_node);
req->cqe.flags = 0;
__io_req_complete_post(req);
io_commit_cqring(ctx);
@@ -231,7 +244,6 @@ static void io_poll_task_func(struct io_kiocb *req, bool *locked)
static void io_apoll_task_func(struct io_kiocb *req, bool *locked)
{
- struct io_ring_ctx *ctx = req->ctx;
int ret;
ret = io_poll_check_events(req, locked);
@@ -239,9 +251,7 @@ static void io_apoll_task_func(struct io_kiocb *req, bool *locked)
return;
io_poll_remove_entries(req);
- spin_lock(&ctx->completion_lock);
- hash_del(&req->hash_node);
- spin_unlock(&ctx->completion_lock);
+ io_poll_req_delete(req, req->ctx);
if (!ret)
io_req_task_submit(req, locked);
@@ -435,9 +445,7 @@ static int __io_arm_poll_handler(struct io_kiocb *req,
return 0;
}
- spin_lock(&ctx->completion_lock);
io_poll_req_insert(req);
- spin_unlock(&ctx->completion_lock);
if (mask && (poll->events & EPOLLET)) {
/* can't multishot if failed, just queue the event we've got */
@@ -534,32 +542,32 @@ __cold bool io_poll_remove_all(struct io_ring_ctx *ctx, struct task_struct *tsk,
bool found = false;
int i;
- spin_lock(&ctx->completion_lock);
for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
- struct hlist_head *list;
+ struct io_hash_bucket *hb = &ctx->cancel_hash[i];
- list = &ctx->cancel_hash[i];
- hlist_for_each_entry_safe(req, tmp, list, hash_node) {
+ spin_lock(&hb->lock);
+ hlist_for_each_entry_safe(req, tmp, &hb->list, hash_node) {
if (io_match_task_safe(req, tsk, cancel_all)) {
hlist_del_init(&req->hash_node);
io_poll_cancel_req(req);
found = true;
}
}
+ spin_unlock(&hb->lock);
}
- spin_unlock(&ctx->completion_lock);
return found;
}
static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
struct io_cancel_data *cd)
- __must_hold(&ctx->completion_lock)
{
- struct hlist_head *list;
struct io_kiocb *req;
- list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)];
- hlist_for_each_entry(req, list, hash_node) {
+ u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
+ struct io_hash_bucket *hb = &ctx->cancel_hash[index];
+
+ spin_lock(&hb->lock);
+ hlist_for_each_entry(req, &hb->list, hash_node) {
if (cd->data != req->cqe.user_data)
continue;
if (poll_only && req->opcode != IORING_OP_POLL_ADD)
@@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
continue;
req->work.cancel_seq = cd->seq;
}
+ spin_unlock(&hb->lock);
return req;
}
+ spin_unlock(&hb->lock);
return NULL;
}
static struct io_kiocb *io_poll_file_find(struct io_ring_ctx *ctx,
struct io_cancel_data *cd)
- __must_hold(&ctx->completion_lock)
{
struct io_kiocb *req;
int i;
for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
- struct hlist_head *list;
+ struct io_hash_bucket *hb = &ctx->cancel_hash[i];
- list = &ctx->cancel_hash[i];
- hlist_for_each_entry(req, list, hash_node) {
+ spin_lock(&hb->lock);
+ hlist_for_each_entry(req, &hb->list, hash_node) {
if (!(cd->flags & IORING_ASYNC_CANCEL_ANY) &&
req->file != cd->file)
continue;
if (cd->seq == req->work.cancel_seq)
continue;
req->work.cancel_seq = cd->seq;
+ spin_unlock(&hb->lock);
return req;
}
+ spin_unlock(&hb->lock);
}
return NULL;
}
static bool io_poll_disarm(struct io_kiocb *req)
- __must_hold(&ctx->completion_lock)
{
if (!io_poll_get_ownership(req))
return false;
io_poll_remove_entries(req);
- hash_del(&req->hash_node);
+ io_poll_req_delete(req, req->ctx);
return true;
}
int io_poll_cancel(struct io_ring_ctx *ctx, struct io_cancel_data *cd)
- __must_hold(&ctx->completion_lock)
{
struct io_kiocb *req;
@@ -718,14 +727,11 @@ int io_poll_remove(struct io_kiocb *req, unsigned int issue_flags)
int ret2, ret = 0;
bool locked;
- spin_lock(&ctx->completion_lock);
preq = io_poll_find(ctx, true, &cd);
if (!preq || !io_poll_disarm(preq)) {
- spin_unlock(&ctx->completion_lock);
ret = preq ? -EALREADY : -ENOENT;
goto out;
}
- spin_unlock(&ctx->completion_lock);
if (poll_update->update_events || poll_update->update_user_data) {
/* only mask one event flags, keep behavior flags */
base-commit: d8271bf021438f468dab3cd84fe5279b5bbcead8
--
2.25.1
^ permalink raw reply related [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-08 11:12 [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock Hao Xu
@ 2022-06-08 11:27 ` Pavel Begunkov
2022-06-10 10:21 ` Pavel Begunkov
1 sibling, 0 replies; 9+ messages in thread
From: Pavel Begunkov @ 2022-06-08 11:27 UTC (permalink / raw)
To: Hao Xu, io-uring; +Cc: Jens Axboe
On 6/8/22 12:12, Hao Xu wrote:
> From: Hao Xu <[email protected]>
>
> Add a new io_hash_bucket structure so that each bucket in cancel_hash
> has separate spinlock. Use per entry lock for cancel_hash, this removes
> some completion lock invocation and remove contension between different
> cancel_hash entries.
Thanks for making the changes, the big picture looks good,
I'll review in a day or two
> Signed-off-by: Hao Xu <[email protected]>
> ---
>
> v1->v2:
> - Add per entry lock for poll/apoll task work code which was missed
> in v1
> - add an member in io_kiocb to track req's indice in cancel_hash
>
> v2->v3:
> - make struct io_hash_bucket align with cacheline to avoid cacheline
> false sharing.
> - re-calculate hash value when deleting an entry from cancel_hash.
> (cannot leverage struct io_poll to store the indice since it's
> already 64 Bytes)
>
> io_uring/cancel.c | 14 +++++++--
> io_uring/cancel.h | 6 ++++
> io_uring/fdinfo.c | 9 ++++--
> io_uring/io_uring.c | 8 +++--
> io_uring/io_uring_types.h | 2 +-
> io_uring/poll.c | 64 +++++++++++++++++++++------------------
> 6 files changed, 65 insertions(+), 38 deletions(-)
>
> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
> index 83cceb52d82d..bced5d6b9294 100644
> --- a/io_uring/cancel.c
> +++ b/io_uring/cancel.c
> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
> if (!ret)
> return 0;
>
> - spin_lock(&ctx->completion_lock);
> ret = io_poll_cancel(ctx, cd);
> if (ret != -ENOENT)
> goto out;
> + spin_lock(&ctx->completion_lock);
> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
> ret = io_timeout_cancel(ctx, cd);
> -out:
> spin_unlock(&ctx->completion_lock);
> +out:
> return ret;
> }
>
> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
> io_req_set_res(req, ret, 0);
> return IOU_OK;
> }
> +
> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
> +{
> + unsigned int i;
> +
> + for (i = 0; i < size; i++) {
> + spin_lock_init(&hash_table[i].lock);
> + INIT_HLIST_HEAD(&hash_table[i].list);
> + }
> +}
> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
> index 4f35d8696325..b57d6706f84d 100644
> --- a/io_uring/cancel.h
> +++ b/io_uring/cancel.h
> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe);
> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>
> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size);
> +
> +struct io_hash_bucket {
> + spinlock_t lock;
> + struct hlist_head list;
> +} ____cacheline_aligned_in_smp;
> diff --git a/io_uring/fdinfo.c b/io_uring/fdinfo.c
> index fcedde4b4b1e..f941c73f5502 100644
> --- a/io_uring/fdinfo.c
> +++ b/io_uring/fdinfo.c
> @@ -13,6 +13,7 @@
> #include "io_uring.h"
> #include "sqpoll.h"
> #include "fdinfo.h"
> +#include "cancel.h"
>
> #ifdef CONFIG_PROC_FS
> static __cold int io_uring_show_cred(struct seq_file *m, unsigned int id,
> @@ -157,17 +158,19 @@ static __cold void __io_uring_show_fdinfo(struct io_ring_ctx *ctx,
> mutex_unlock(&ctx->uring_lock);
>
> seq_puts(m, "PollList:\n");
> - spin_lock(&ctx->completion_lock);
> for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
> - struct hlist_head *list = &ctx->cancel_hash[i];
> + struct io_hash_bucket *hb = &ctx->cancel_hash[i];
> struct io_kiocb *req;
>
> - hlist_for_each_entry(req, list, hash_node)
> + spin_lock(&hb->lock);
> + hlist_for_each_entry(req, &hb->list, hash_node)
> seq_printf(m, " op=%d, task_works=%d\n", req->opcode,
> task_work_pending(req->task));
> + spin_unlock(&hb->lock);
> }
>
> seq_puts(m, "CqOverflowList:\n");
> + spin_lock(&ctx->completion_lock);
> list_for_each_entry(ocqe, &ctx->cq_overflow_list, list) {
> struct io_uring_cqe *cqe = &ocqe->cqe;
>
> diff --git a/io_uring/io_uring.c b/io_uring/io_uring.c
> index 1572ebe3cff1..b67ab76b9e56 100644
> --- a/io_uring/io_uring.c
> +++ b/io_uring/io_uring.c
> @@ -725,11 +725,13 @@ static __cold struct io_ring_ctx *io_ring_ctx_alloc(struct io_uring_params *p)
> if (hash_bits <= 0)
> hash_bits = 1;
> ctx->cancel_hash_bits = hash_bits;
> - ctx->cancel_hash = kmalloc((1U << hash_bits) * sizeof(struct hlist_head),
> - GFP_KERNEL);
> + ctx->cancel_hash =
> + kmalloc((1U << hash_bits) * sizeof(struct io_hash_bucket),
> + GFP_KERNEL);
> if (!ctx->cancel_hash)
> goto err;
> - __hash_init(ctx->cancel_hash, 1U << hash_bits);
> +
> + init_hash_table(ctx->cancel_hash, 1U << hash_bits);
>
> ctx->dummy_ubuf = kzalloc(sizeof(*ctx->dummy_ubuf), GFP_KERNEL);
> if (!ctx->dummy_ubuf)
> diff --git a/io_uring/io_uring_types.h b/io_uring/io_uring_types.h
> index 7c22cf35a7e2..b67843a5f3b6 100644
> --- a/io_uring/io_uring_types.h
> +++ b/io_uring/io_uring_types.h
> @@ -230,7 +230,7 @@ struct io_ring_ctx {
> * manipulate the list, hence no extra locking is needed there.
> */
> struct io_wq_work_list iopoll_list;
> - struct hlist_head *cancel_hash;
> + struct io_hash_bucket *cancel_hash;
> unsigned cancel_hash_bits;
> bool poll_multi_queue;
>
> diff --git a/io_uring/poll.c b/io_uring/poll.c
> index 0df5eca93b16..515f1727e3c6 100644
> --- a/io_uring/poll.c
> +++ b/io_uring/poll.c
> @@ -19,6 +19,7 @@
> #include "opdef.h"
> #include "kbuf.h"
> #include "poll.h"
> +#include "cancel.h"
>
> struct io_poll_update {
> struct file *file;
> @@ -73,10 +74,22 @@ static struct io_poll *io_poll_get_single(struct io_kiocb *req)
> static void io_poll_req_insert(struct io_kiocb *req)
> {
> struct io_ring_ctx *ctx = req->ctx;
> - struct hlist_head *list;
> + u32 index = hash_long(req->cqe.user_data, ctx->cancel_hash_bits);
> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
>
> - list = &ctx->cancel_hash[hash_long(req->cqe.user_data, ctx->cancel_hash_bits)];
> - hlist_add_head(&req->hash_node, list);
> + spin_lock(&hb->lock);
> + hlist_add_head(&req->hash_node, &hb->list);
> + spin_unlock(&hb->lock);
> +}
> +
> +static void io_poll_req_delete(struct io_kiocb *req, struct io_ring_ctx *ctx)
> +{
> + u32 index = hash_long(req->cqe.user_data, ctx->cancel_hash_bits);
> + spinlock_t *lock = &ctx->cancel_hash[index].lock;
> +
> + spin_lock(lock);
> + hash_del(&req->hash_node);
> + spin_unlock(lock);
> }
>
> static void io_init_poll_iocb(struct io_poll *poll, __poll_t events,
> @@ -220,8 +233,8 @@ static void io_poll_task_func(struct io_kiocb *req, bool *locked)
> }
>
> io_poll_remove_entries(req);
> + io_poll_req_delete(req, ctx);
> spin_lock(&ctx->completion_lock);
> - hash_del(&req->hash_node);
> req->cqe.flags = 0;
> __io_req_complete_post(req);
> io_commit_cqring(ctx);
> @@ -231,7 +244,6 @@ static void io_poll_task_func(struct io_kiocb *req, bool *locked)
>
> static void io_apoll_task_func(struct io_kiocb *req, bool *locked)
> {
> - struct io_ring_ctx *ctx = req->ctx;
> int ret;
>
> ret = io_poll_check_events(req, locked);
> @@ -239,9 +251,7 @@ static void io_apoll_task_func(struct io_kiocb *req, bool *locked)
> return;
>
> io_poll_remove_entries(req);
> - spin_lock(&ctx->completion_lock);
> - hash_del(&req->hash_node);
> - spin_unlock(&ctx->completion_lock);
> + io_poll_req_delete(req, req->ctx);
>
> if (!ret)
> io_req_task_submit(req, locked);
> @@ -435,9 +445,7 @@ static int __io_arm_poll_handler(struct io_kiocb *req,
> return 0;
> }
>
> - spin_lock(&ctx->completion_lock);
> io_poll_req_insert(req);
> - spin_unlock(&ctx->completion_lock);
>
> if (mask && (poll->events & EPOLLET)) {
> /* can't multishot if failed, just queue the event we've got */
> @@ -534,32 +542,32 @@ __cold bool io_poll_remove_all(struct io_ring_ctx *ctx, struct task_struct *tsk,
> bool found = false;
> int i;
>
> - spin_lock(&ctx->completion_lock);
> for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
> - struct hlist_head *list;
> + struct io_hash_bucket *hb = &ctx->cancel_hash[i];
>
> - list = &ctx->cancel_hash[i];
> - hlist_for_each_entry_safe(req, tmp, list, hash_node) {
> + spin_lock(&hb->lock);
> + hlist_for_each_entry_safe(req, tmp, &hb->list, hash_node) {
> if (io_match_task_safe(req, tsk, cancel_all)) {
> hlist_del_init(&req->hash_node);
> io_poll_cancel_req(req);
> found = true;
> }
> }
> + spin_unlock(&hb->lock);
> }
> - spin_unlock(&ctx->completion_lock);
> return found;
> }
>
> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
> struct io_cancel_data *cd)
> - __must_hold(&ctx->completion_lock)
> {
> - struct hlist_head *list;
> struct io_kiocb *req;
>
> - list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)];
> - hlist_for_each_entry(req, list, hash_node) {
> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
> +
> + spin_lock(&hb->lock);
> + hlist_for_each_entry(req, &hb->list, hash_node) {
> if (cd->data != req->cqe.user_data)
> continue;
> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
> continue;
> req->work.cancel_seq = cd->seq;
> }
> + spin_unlock(&hb->lock);
> return req;
> }
> + spin_unlock(&hb->lock);
> return NULL;
> }
>
> static struct io_kiocb *io_poll_file_find(struct io_ring_ctx *ctx,
> struct io_cancel_data *cd)
> - __must_hold(&ctx->completion_lock)
> {
> struct io_kiocb *req;
> int i;
>
> for (i = 0; i < (1U << ctx->cancel_hash_bits); i++) {
> - struct hlist_head *list;
> + struct io_hash_bucket *hb = &ctx->cancel_hash[i];
>
> - list = &ctx->cancel_hash[i];
> - hlist_for_each_entry(req, list, hash_node) {
> + spin_lock(&hb->lock);
> + hlist_for_each_entry(req, &hb->list, hash_node) {
> if (!(cd->flags & IORING_ASYNC_CANCEL_ANY) &&
> req->file != cd->file)
> continue;
> if (cd->seq == req->work.cancel_seq)
> continue;
> req->work.cancel_seq = cd->seq;
> + spin_unlock(&hb->lock);
> return req;
> }
> + spin_unlock(&hb->lock);
> }
> return NULL;
> }
>
> static bool io_poll_disarm(struct io_kiocb *req)
> - __must_hold(&ctx->completion_lock)
> {
> if (!io_poll_get_ownership(req))
> return false;
> io_poll_remove_entries(req);
> - hash_del(&req->hash_node);
> + io_poll_req_delete(req, req->ctx);
> return true;
> }
>
> int io_poll_cancel(struct io_ring_ctx *ctx, struct io_cancel_data *cd)
> - __must_hold(&ctx->completion_lock)
> {
> struct io_kiocb *req;
>
> @@ -718,14 +727,11 @@ int io_poll_remove(struct io_kiocb *req, unsigned int issue_flags)
> int ret2, ret = 0;
> bool locked;
>
> - spin_lock(&ctx->completion_lock);
> preq = io_poll_find(ctx, true, &cd);
> if (!preq || !io_poll_disarm(preq)) {
> - spin_unlock(&ctx->completion_lock);
> ret = preq ? -EALREADY : -ENOENT;
> goto out;
> }
> - spin_unlock(&ctx->completion_lock);
>
> if (poll_update->update_events || poll_update->update_user_data) {
> /* only mask one event flags, keep behavior flags */
>
> base-commit: d8271bf021438f468dab3cd84fe5279b5bbcead8
--
Pavel Begunkov
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-08 11:12 [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock Hao Xu
2022-06-08 11:27 ` Pavel Begunkov
@ 2022-06-10 10:21 ` Pavel Begunkov
2022-06-10 15:45 ` Hao Xu
2022-06-11 4:34 ` Hao Xu
1 sibling, 2 replies; 9+ messages in thread
From: Pavel Begunkov @ 2022-06-10 10:21 UTC (permalink / raw)
To: Hao Xu, io-uring; +Cc: Jens Axboe
On 6/8/22 12:12, Hao Xu wrote:
> From: Hao Xu <[email protected]>
>
> Add a new io_hash_bucket structure so that each bucket in cancel_hash
> has separate spinlock. Use per entry lock for cancel_hash, this removes
> some completion lock invocation and remove contension between different
> cancel_hash entries.
>
> Signed-off-by: Hao Xu <[email protected]>
> ---
>
> v1->v2:
> - Add per entry lock for poll/apoll task work code which was missed
> in v1
> - add an member in io_kiocb to track req's indice in cancel_hash
>
> v2->v3:
> - make struct io_hash_bucket align with cacheline to avoid cacheline
> false sharing.
> - re-calculate hash value when deleting an entry from cancel_hash.
> (cannot leverage struct io_poll to store the indice since it's
> already 64 Bytes)
>
> io_uring/cancel.c | 14 +++++++--
> io_uring/cancel.h | 6 ++++
> io_uring/fdinfo.c | 9 ++++--
> io_uring/io_uring.c | 8 +++--
> io_uring/io_uring_types.h | 2 +-
> io_uring/poll.c | 64 +++++++++++++++++++++------------------
> 6 files changed, 65 insertions(+), 38 deletions(-)
>
> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
> index 83cceb52d82d..bced5d6b9294 100644
> --- a/io_uring/cancel.c
> +++ b/io_uring/cancel.c
> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
> if (!ret)
> return 0;
>
> - spin_lock(&ctx->completion_lock);
> ret = io_poll_cancel(ctx, cd);
> if (ret != -ENOENT)
> goto out;
> + spin_lock(&ctx->completion_lock);
> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
> ret = io_timeout_cancel(ctx, cd);
> -out:
> spin_unlock(&ctx->completion_lock);
> +out:
> return ret;
> }
>
> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
> io_req_set_res(req, ret, 0);
> return IOU_OK;
> }
> +
> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
Not inline, it can break builds
> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
> index 4f35d8696325..b57d6706f84d 100644
> --- a/io_uring/cancel.h
> +++ b/io_uring/cancel.h
> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe);
> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>
> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size);
And this inline as well
> diff --git a/io_uring/poll.c b/io_uring/poll.c
> index 0df5eca93b16..515f1727e3c6 100644
> --- a/io_uring/poll.c
> +++ b/io_uring/poll.c
[...]
> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
> struct io_cancel_data *cd)
> - __must_hold(&ctx->completion_lock)
> {
> - struct hlist_head *list;
> struct io_kiocb *req;
>
> - list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)];
> - hlist_for_each_entry(req, list, hash_node) {
> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
> +
> + spin_lock(&hb->lock);
> + hlist_for_each_entry(req, &hb->list, hash_node) {
> if (cd->data != req->cqe.user_data)
> continue;
> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
> continue;
> req->work.cancel_seq = cd->seq;
> }
> + spin_unlock(&hb->lock);
The problem here is that after you unlock, nothing keeps the
request alive. Before it was more like
lock(completion_lock);
req = poll_find();
cancel_poll(req);
unlock(completion_lock);
and was relying on all of this happening under ->completion_lock.
Now following io_poll_disarm() and/or io_poll_cancel_req() race.
Same with io_poll_file_find().
Should be fine apart from this
--
Pavel Begunkov
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-10 10:21 ` Pavel Begunkov
@ 2022-06-10 15:45 ` Hao Xu
2022-06-10 16:10 ` Pavel Begunkov
2022-06-11 4:34 ` Hao Xu
1 sibling, 1 reply; 9+ messages in thread
From: Hao Xu @ 2022-06-10 15:45 UTC (permalink / raw)
To: Pavel Begunkov, io-uring; +Cc: Jens Axboe
On 6/10/22 18:21, Pavel Begunkov wrote:
> On 6/8/22 12:12, Hao Xu wrote:
>> From: Hao Xu <[email protected]>
>>
>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>> some completion lock invocation and remove contension between different
>> cancel_hash entries.
>>
>> Signed-off-by: Hao Xu <[email protected]>
>> ---
>>
>> v1->v2:
>> - Add per entry lock for poll/apoll task work code which was missed
>> in v1
>> - add an member in io_kiocb to track req's indice in cancel_hash
>>
>> v2->v3:
>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>> false sharing.
>> - re-calculate hash value when deleting an entry from cancel_hash.
>> (cannot leverage struct io_poll to store the indice since it's
>> already 64 Bytes)
>>
>> io_uring/cancel.c | 14 +++++++--
>> io_uring/cancel.h | 6 ++++
>> io_uring/fdinfo.c | 9 ++++--
>> io_uring/io_uring.c | 8 +++--
>> io_uring/io_uring_types.h | 2 +-
>> io_uring/poll.c | 64 +++++++++++++++++++++------------------
>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>
>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>> index 83cceb52d82d..bced5d6b9294 100644
>> --- a/io_uring/cancel.c
>> +++ b/io_uring/cancel.c
>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct
>> io_cancel_data *cd)
>> if (!ret)
>> return 0;
>> - spin_lock(&ctx->completion_lock);
>> ret = io_poll_cancel(ctx, cd);
>> if (ret != -ENOENT)
>> goto out;
>> + spin_lock(&ctx->completion_lock);
>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>> ret = io_timeout_cancel(ctx, cd);
>> -out:
>> spin_unlock(&ctx->completion_lock);
>> +out:
>> return ret;
>> }
>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req,
>> unsigned int issue_flags)
>> io_req_set_res(req, ret, 0);
>> return IOU_OK;
>> }
>> +
>> +inline void init_hash_table(struct io_hash_bucket *hash_table,
>> unsigned size)
>
> Not inline, it can break builds
>
>> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
>> index 4f35d8696325..b57d6706f84d 100644
>> --- a/io_uring/cancel.h
>> +++ b/io_uring/cancel.h
>> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const
>> struct io_uring_sqe *sqe);
>> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
>> +inline void init_hash_table(struct io_hash_bucket *hash_table,
>> unsigned size);
>
> And this inline as well
>
>> diff --git a/io_uring/poll.c b/io_uring/poll.c
>> index 0df5eca93b16..515f1727e3c6 100644
>> --- a/io_uring/poll.c
>> +++ b/io_uring/poll.c
> [...]
>> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool
>> poll_only,
>> struct io_cancel_data *cd)
>> - __must_hold(&ctx->completion_lock)
>> {
>> - struct hlist_head *list;
>> struct io_kiocb *req;
>> - list = &ctx->cancel_hash[hash_long(cd->data,
>> ctx->cancel_hash_bits)];
>> - hlist_for_each_entry(req, list, hash_node) {
>> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
>> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
>> +
>> + spin_lock(&hb->lock);
>> + hlist_for_each_entry(req, &hb->list, hash_node) {
>> if (cd->data != req->cqe.user_data)
>> continue;
>> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
>> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct
>> io_ring_ctx *ctx, bool poll_only,
>> continue;
>> req->work.cancel_seq = cd->seq;
>> }
>> + spin_unlock(&hb->lock);
>
> The problem here is that after you unlock, nothing keeps the
> request alive. Before it was more like
>
> lock(completion_lock);
> req = poll_find();
> cancel_poll(req);
> unlock(completion_lock);
>
> and was relying on all of this happening under ->completion_lock.
> Now following io_poll_disarm() and/or io_poll_cancel_req() race.
> Same with io_poll_file_find().
Looks we have to add completion_lock back for cancellation path.
>
> Should be fine apart from this
>
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-10 15:45 ` Hao Xu
@ 2022-06-10 16:10 ` Pavel Begunkov
2022-06-10 17:40 ` Hao Xu
0 siblings, 1 reply; 9+ messages in thread
From: Pavel Begunkov @ 2022-06-10 16:10 UTC (permalink / raw)
To: Hao Xu, io-uring; +Cc: Jens Axboe
On 6/10/22 16:45, Hao Xu wrote:
> On 6/10/22 18:21, Pavel Begunkov wrote:
>> On 6/8/22 12:12, Hao Xu wrote:
>>> From: Hao Xu <[email protected]>
>>>
>>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>>> some completion lock invocation and remove contension between different
>>> cancel_hash entries.
>>>
>>> Signed-off-by: Hao Xu <[email protected]>
>>> ---
>>>
>>> v1->v2:
>>> - Add per entry lock for poll/apoll task work code which was missed
>>> in v1
>>> - add an member in io_kiocb to track req's indice in cancel_hash
>>>
>>> v2->v3:
>>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>>> false sharing.
>>> - re-calculate hash value when deleting an entry from cancel_hash.
>>> (cannot leverage struct io_poll to store the indice since it's
>>> already 64 Bytes)
>>>
>>> io_uring/cancel.c | 14 +++++++--
>>> io_uring/cancel.h | 6 ++++
>>> io_uring/fdinfo.c | 9 ++++--
>>> io_uring/io_uring.c | 8 +++--
>>> io_uring/io_uring_types.h | 2 +-
>>> io_uring/poll.c | 64 +++++++++++++++++++++------------------
>>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>>
>>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>>> index 83cceb52d82d..bced5d6b9294 100644
>>> --- a/io_uring/cancel.c
>>> +++ b/io_uring/cancel.c
>>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
>>> if (!ret)
>>> return 0;
>>> - spin_lock(&ctx->completion_lock);
>>> ret = io_poll_cancel(ctx, cd);
>>> if (ret != -ENOENT)
>>> goto out;
>>> + spin_lock(&ctx->completion_lock);
>>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>>> ret = io_timeout_cancel(ctx, cd);
>>> -out:
>>> spin_unlock(&ctx->completion_lock);
>>> +out:
>>> return ret;
>>> }
>>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
>>> io_req_set_res(req, ret, 0);
>>> return IOU_OK;
>>> }
>>> +
>>> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
>>
>> Not inline, it can break builds
>>
>>> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
>>> index 4f35d8696325..b57d6706f84d 100644
>>> --- a/io_uring/cancel.h
>>> +++ b/io_uring/cancel.h
>>> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe);
>>> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>>> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
>>> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size);
>>
>> And this inline as well
>>
>>> diff --git a/io_uring/poll.c b/io_uring/poll.c
>>> index 0df5eca93b16..515f1727e3c6 100644
>>> --- a/io_uring/poll.c
>>> +++ b/io_uring/poll.c
>> [...]
>>> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
>>> struct io_cancel_data *cd)
>>> - __must_hold(&ctx->completion_lock)
>>> {
>>> - struct hlist_head *list;
>>> struct io_kiocb *req;
>>> - list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)];
>>> - hlist_for_each_entry(req, list, hash_node) {
>>> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
>>> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
>>> +
>>> + spin_lock(&hb->lock);
>>> + hlist_for_each_entry(req, &hb->list, hash_node) {
>>> if (cd->data != req->cqe.user_data)
>>> continue;
>>> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
>>> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
>>> continue;
>>> req->work.cancel_seq = cd->seq;
>>> }
>>> + spin_unlock(&hb->lock);
>>
>> The problem here is that after you unlock, nothing keeps the
>> request alive. Before it was more like
>>
>> lock(completion_lock);
>> req = poll_find();
>> cancel_poll(req);
>> unlock(completion_lock);
>>
>> and was relying on all of this happening under ->completion_lock.
>> Now following io_poll_disarm() and/or io_poll_cancel_req() race.
>> Same with io_poll_file_find().
>
> Looks we have to add completion_lock back for cancellation path.
It was relying on completion_lock only because it was guarding
the hashing, so now find+cancel should happen under the per
bucket spins, i.e.
lock(buckets[index].lock);
req = poll_find();
cancel_poll(req);
unlock(buckets[index].lock);
A bit trickier to code but doable.
--
Pavel Begunkov
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-10 16:10 ` Pavel Begunkov
@ 2022-06-10 17:40 ` Hao Xu
2022-06-10 17:45 ` Pavel Begunkov
0 siblings, 1 reply; 9+ messages in thread
From: Hao Xu @ 2022-06-10 17:40 UTC (permalink / raw)
To: Pavel Begunkov, io-uring; +Cc: Jens Axboe
On 6/11/22 00:10, Pavel Begunkov wrote:
> On 6/10/22 16:45, Hao Xu wrote:
>> On 6/10/22 18:21, Pavel Begunkov wrote:
>>> On 6/8/22 12:12, Hao Xu wrote:
>>>> From: Hao Xu <[email protected]>
>>>>
>>>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>>>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>>>> some completion lock invocation and remove contension between different
>>>> cancel_hash entries.
>>>>
>>>> Signed-off-by: Hao Xu <[email protected]>
>>>> ---
>>>>
>>>> v1->v2:
>>>> - Add per entry lock for poll/apoll task work code which was missed
>>>> in v1
>>>> - add an member in io_kiocb to track req's indice in cancel_hash
>>>>
>>>> v2->v3:
>>>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>>>> false sharing.
>>>> - re-calculate hash value when deleting an entry from cancel_hash.
>>>> (cannot leverage struct io_poll to store the indice since it's
>>>> already 64 Bytes)
>>>>
>>>> io_uring/cancel.c | 14 +++++++--
>>>> io_uring/cancel.h | 6 ++++
>>>> io_uring/fdinfo.c | 9 ++++--
>>>> io_uring/io_uring.c | 8 +++--
>>>> io_uring/io_uring_types.h | 2 +-
>>>> io_uring/poll.c | 64
>>>> +++++++++++++++++++++------------------
>>>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>>>
>>>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>>>> index 83cceb52d82d..bced5d6b9294 100644
>>>> --- a/io_uring/cancel.c
>>>> +++ b/io_uring/cancel.c
>>>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct
>>>> io_cancel_data *cd)
>>>> if (!ret)
>>>> return 0;
>>>> - spin_lock(&ctx->completion_lock);
>>>> ret = io_poll_cancel(ctx, cd);
>>>> if (ret != -ENOENT)
>>>> goto out;
>>>> + spin_lock(&ctx->completion_lock);
>>>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>>>> ret = io_timeout_cancel(ctx, cd);
>>>> -out:
>>>> spin_unlock(&ctx->completion_lock);
>>>> +out:
>>>> return ret;
>>>> }
>>>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req,
>>>> unsigned int issue_flags)
>>>> io_req_set_res(req, ret, 0);
>>>> return IOU_OK;
>>>> }
>>>> +
>>>> +inline void init_hash_table(struct io_hash_bucket *hash_table,
>>>> unsigned size)
>>>
>>> Not inline, it can break builds
>>>
>>>> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
>>>> index 4f35d8696325..b57d6706f84d 100644
>>>> --- a/io_uring/cancel.h
>>>> +++ b/io_uring/cancel.h
>>>> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const
>>>> struct io_uring_sqe *sqe);
>>>> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>>>> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
>>>> +inline void init_hash_table(struct io_hash_bucket *hash_table,
>>>> unsigned size);
>>>
>>> And this inline as well
>>>
>>>> diff --git a/io_uring/poll.c b/io_uring/poll.c
>>>> index 0df5eca93b16..515f1727e3c6 100644
>>>> --- a/io_uring/poll.c
>>>> +++ b/io_uring/poll.c
>>> [...]
>>>> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool
>>>> poll_only,
>>>> struct io_cancel_data *cd)
>>>> - __must_hold(&ctx->completion_lock)
>>>> {
>>>> - struct hlist_head *list;
>>>> struct io_kiocb *req;
>>>> - list = &ctx->cancel_hash[hash_long(cd->data,
>>>> ctx->cancel_hash_bits)];
>>>> - hlist_for_each_entry(req, list, hash_node) {
>>>> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
>>>> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
>>>> +
>>>> + spin_lock(&hb->lock);
>>>> + hlist_for_each_entry(req, &hb->list, hash_node) {
>>>> if (cd->data != req->cqe.user_data)
>>>> continue;
>>>> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
>>>> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct
>>>> io_ring_ctx *ctx, bool poll_only,
>>>> continue;
>>>> req->work.cancel_seq = cd->seq;
>>>> }
>>>> + spin_unlock(&hb->lock);
>>>
>>> The problem here is that after you unlock, nothing keeps the
>>> request alive. Before it was more like
>>>
>>> lock(completion_lock);
>>> req = poll_find();
>>> cancel_poll(req);
>>> unlock(completion_lock);
>>>
>>> and was relying on all of this happening under ->completion_lock.
>>> Now following io_poll_disarm() and/or io_poll_cancel_req() race.
>>> Same with io_poll_file_find().
>>
>> Looks we have to add completion_lock back for cancellation path.
>
> It was relying on completion_lock only because it was guarding
> the hashing, so now find+cancel should happen under the per
> bucket spins, i.e.
>
> lock(buckets[index].lock);
> req = poll_find();
> cancel_poll(req);
> unlock(buckets[index].lock);
>
> A bit trickier to code but doable.
Ah, seems I misunderstood your words, which I'm clear with now.
Yea, it's a bit odd. I'll think about this issue before taking this
solution tomorrow.
Btw, I saw a req->refcount set for poll_add, seems it is not necessary?
(I haven't check it carefully yet)
>
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-10 17:40 ` Hao Xu
@ 2022-06-10 17:45 ` Pavel Begunkov
0 siblings, 0 replies; 9+ messages in thread
From: Pavel Begunkov @ 2022-06-10 17:45 UTC (permalink / raw)
To: Hao Xu, io-uring; +Cc: Jens Axboe
On 6/10/22 18:40, Hao Xu wrote:
> On 6/11/22 00:10, Pavel Begunkov wrote:
>> On 6/10/22 16:45, Hao Xu wrote:
>>> On 6/10/22 18:21, Pavel Begunkov wrote:
>>>> On 6/8/22 12:12, Hao Xu wrote:
>>>>> From: Hao Xu <[email protected]>
>>>>>
>>>>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>>>>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>>>>> some completion lock invocation and remove contension between different
>>>>> cancel_hash entries.
>>>>>
>>>>> Signed-off-by: Hao Xu <[email protected]>
>>>>> ---
>>>>>
>>>>> v1->v2:
>>>>> - Add per entry lock for poll/apoll task work code which was missed
>>>>> in v1
>>>>> - add an member in io_kiocb to track req's indice in cancel_hash
>>>>>
>>>>> v2->v3:
>>>>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>>>>> false sharing.
>>>>> - re-calculate hash value when deleting an entry from cancel_hash.
>>>>> (cannot leverage struct io_poll to store the indice since it's
>>>>> already 64 Bytes)
>>>>>
>>>>> io_uring/cancel.c | 14 +++++++--
>>>>> io_uring/cancel.h | 6 ++++
>>>>> io_uring/fdinfo.c | 9 ++++--
>>>>> io_uring/io_uring.c | 8 +++--
>>>>> io_uring/io_uring_types.h | 2 +-
>>>>> io_uring/poll.c | 64 +++++++++++++++++++++------------------
>>>>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>>>>
>>>>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>>>>> index 83cceb52d82d..bced5d6b9294 100644
>>>>> --- a/io_uring/cancel.c
>>>>> +++ b/io_uring/cancel.c
>>>>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
>>>>> if (!ret)
>>>>> return 0;
>>>>> - spin_lock(&ctx->completion_lock);
>>>>> ret = io_poll_cancel(ctx, cd);
>>>>> if (ret != -ENOENT)
>>>>> goto out;
>>>>> + spin_lock(&ctx->completion_lock);
>>>>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>>>>> ret = io_timeout_cancel(ctx, cd);
>>>>> -out:
>>>>> spin_unlock(&ctx->completion_lock);
>>>>> +out:
>>>>> return ret;
>>>>> }
>>>>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
>>>>> io_req_set_res(req, ret, 0);
>>>>> return IOU_OK;
>>>>> }
>>>>> +
>>>>> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
>>>>
>>>> Not inline, it can break builds
>>>>
>>>>> diff --git a/io_uring/cancel.h b/io_uring/cancel.h
>>>>> index 4f35d8696325..b57d6706f84d 100644
>>>>> --- a/io_uring/cancel.h
>>>>> +++ b/io_uring/cancel.h
>>>>> @@ -4,3 +4,9 @@ int io_async_cancel_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe);
>>>>> int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags);
>>>>> int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd);
>>>>> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size);
>>>>
>>>> And this inline as well
>>>>
>>>>> diff --git a/io_uring/poll.c b/io_uring/poll.c
>>>>> index 0df5eca93b16..515f1727e3c6 100644
>>>>> --- a/io_uring/poll.c
>>>>> +++ b/io_uring/poll.c
>>>> [...]
>>>>> static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
>>>>> struct io_cancel_data *cd)
>>>>> - __must_hold(&ctx->completion_lock)
>>>>> {
>>>>> - struct hlist_head *list;
>>>>> struct io_kiocb *req;
>>>>> - list = &ctx->cancel_hash[hash_long(cd->data, ctx->cancel_hash_bits)];
>>>>> - hlist_for_each_entry(req, list, hash_node) {
>>>>> + u32 index = hash_long(cd->data, ctx->cancel_hash_bits);
>>>>> + struct io_hash_bucket *hb = &ctx->cancel_hash[index];
>>>>> +
>>>>> + spin_lock(&hb->lock);
>>>>> + hlist_for_each_entry(req, &hb->list, hash_node) {
>>>>> if (cd->data != req->cqe.user_data)
>>>>> continue;
>>>>> if (poll_only && req->opcode != IORING_OP_POLL_ADD)
>>>>> @@ -569,47 +577,48 @@ static struct io_kiocb *io_poll_find(struct io_ring_ctx *ctx, bool poll_only,
>>>>> continue;
>>>>> req->work.cancel_seq = cd->seq;
>>>>> }
>>>>> + spin_unlock(&hb->lock);
>>>>
>>>> The problem here is that after you unlock, nothing keeps the
>>>> request alive. Before it was more like
>>>>
>>>> lock(completion_lock);
>>>> req = poll_find();
>>>> cancel_poll(req);
>>>> unlock(completion_lock);
>>>>
>>>> and was relying on all of this happening under ->completion_lock.
>>>> Now following io_poll_disarm() and/or io_poll_cancel_req() race.
>>>> Same with io_poll_file_find().
>>>
>>> Looks we have to add completion_lock back for cancellation path.
>>
>> It was relying on completion_lock only because it was guarding
>> the hashing, so now find+cancel should happen under the per
>> bucket spins, i.e.
>>
>> lock(buckets[index].lock);
>> req = poll_find();
>> cancel_poll(req);
>> unlock(buckets[index].lock);
>>
>> A bit trickier to code but doable.
>
> Ah, seems I misunderstood your words, which I'm clear with now.
> Yea, it's a bit odd. I'll think about this issue before taking this
> solution tomorrow.
yeah, it is "a request won't be freed awhile hashed" kind of
synchronisation here.
> Btw, I saw a req->refcount set for poll_add, seems it is not necessary?
> (I haven't check it carefully yet)
In io_poll_add_prep()? Shouldn't be needed, I forgot to kill it
after adding ->poll_refs.
--
Pavel Begunkov
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-10 10:21 ` Pavel Begunkov
2022-06-10 15:45 ` Hao Xu
@ 2022-06-11 4:34 ` Hao Xu
2022-06-11 13:18 ` Pavel Begunkov
1 sibling, 1 reply; 9+ messages in thread
From: Hao Xu @ 2022-06-11 4:34 UTC (permalink / raw)
To: Pavel Begunkov, io-uring; +Cc: Jens Axboe
On 6/10/22 18:21, Pavel Begunkov wrote:
> On 6/8/22 12:12, Hao Xu wrote:
>> From: Hao Xu <[email protected]>
>>
>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>> some completion lock invocation and remove contension between different
>> cancel_hash entries.
>>
>> Signed-off-by: Hao Xu <[email protected]>
>> ---
>>
>> v1->v2:
>> - Add per entry lock for poll/apoll task work code which was missed
>> in v1
>> - add an member in io_kiocb to track req's indice in cancel_hash
>>
>> v2->v3:
>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>> false sharing.
>> - re-calculate hash value when deleting an entry from cancel_hash.
>> (cannot leverage struct io_poll to store the indice since it's
>> already 64 Bytes)
>>
>> io_uring/cancel.c | 14 +++++++--
>> io_uring/cancel.h | 6 ++++
>> io_uring/fdinfo.c | 9 ++++--
>> io_uring/io_uring.c | 8 +++--
>> io_uring/io_uring_types.h | 2 +-
>> io_uring/poll.c | 64 +++++++++++++++++++++------------------
>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>
>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>> index 83cceb52d82d..bced5d6b9294 100644
>> --- a/io_uring/cancel.c
>> +++ b/io_uring/cancel.c
>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct
>> io_cancel_data *cd)
>> if (!ret)
>> return 0;
>> - spin_lock(&ctx->completion_lock);
>> ret = io_poll_cancel(ctx, cd);
>> if (ret != -ENOENT)
>> goto out;
>> + spin_lock(&ctx->completion_lock);
>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>> ret = io_timeout_cancel(ctx, cd);
>> -out:
>> spin_unlock(&ctx->completion_lock);
>> +out:
>> return ret;
>> }
>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req,
>> unsigned int issue_flags)
>> io_req_set_res(req, ret, 0);
>> return IOU_OK;
>> }
>> +
>> +inline void init_hash_table(struct io_hash_bucket *hash_table,
>> unsigned size)
>
> Not inline, it can break builds
>
What do you mean? It's compiled well.
^ permalink raw reply [flat|nested] 9+ messages in thread
* Re: [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock
2022-06-11 4:34 ` Hao Xu
@ 2022-06-11 13:18 ` Pavel Begunkov
0 siblings, 0 replies; 9+ messages in thread
From: Pavel Begunkov @ 2022-06-11 13:18 UTC (permalink / raw)
To: Hao Xu, io-uring; +Cc: Jens Axboe
On 6/11/22 05:34, Hao Xu wrote:
> On 6/10/22 18:21, Pavel Begunkov wrote:
>> On 6/8/22 12:12, Hao Xu wrote:
>>> From: Hao Xu <[email protected]>
>>>
>>> Add a new io_hash_bucket structure so that each bucket in cancel_hash
>>> has separate spinlock. Use per entry lock for cancel_hash, this removes
>>> some completion lock invocation and remove contension between different
>>> cancel_hash entries.
>>>
>>> Signed-off-by: Hao Xu <[email protected]>
>>> ---
>>>
>>> v1->v2:
>>> - Add per entry lock for poll/apoll task work code which was missed
>>> in v1
>>> - add an member in io_kiocb to track req's indice in cancel_hash
>>>
>>> v2->v3:
>>> - make struct io_hash_bucket align with cacheline to avoid cacheline
>>> false sharing.
>>> - re-calculate hash value when deleting an entry from cancel_hash.
>>> (cannot leverage struct io_poll to store the indice since it's
>>> already 64 Bytes)
>>>
>>> io_uring/cancel.c | 14 +++++++--
>>> io_uring/cancel.h | 6 ++++
>>> io_uring/fdinfo.c | 9 ++++--
>>> io_uring/io_uring.c | 8 +++--
>>> io_uring/io_uring_types.h | 2 +-
>>> io_uring/poll.c | 64 +++++++++++++++++++++------------------
>>> 6 files changed, 65 insertions(+), 38 deletions(-)
>>>
>>> diff --git a/io_uring/cancel.c b/io_uring/cancel.c
>>> index 83cceb52d82d..bced5d6b9294 100644
>>> --- a/io_uring/cancel.c
>>> +++ b/io_uring/cancel.c
>>> @@ -93,14 +93,14 @@ int io_try_cancel(struct io_kiocb *req, struct io_cancel_data *cd)
>>> if (!ret)
>>> return 0;
>>> - spin_lock(&ctx->completion_lock);
>>> ret = io_poll_cancel(ctx, cd);
>>> if (ret != -ENOENT)
>>> goto out;
>>> + spin_lock(&ctx->completion_lock);
>>> if (!(cd->flags & IORING_ASYNC_CANCEL_FD))
>>> ret = io_timeout_cancel(ctx, cd);
>>> -out:
>>> spin_unlock(&ctx->completion_lock);
>>> +out:
>>> return ret;
>>> }
>>> @@ -192,3 +192,13 @@ int io_async_cancel(struct io_kiocb *req, unsigned int issue_flags)
>>> io_req_set_res(req, ret, 0);
>>> return IOU_OK;
>>> }
>>> +
>>> +inline void init_hash_table(struct io_hash_bucket *hash_table, unsigned size)
>>
>> Not inline, it can break builds
>>
>
> What do you mean? It's compiled well.
I might be wrong, but IIRC there could be linking issues. Anyway,
it's used only from another obj and the function is cold, so no reason
to have it inline.
--
Pavel Begunkov
^ permalink raw reply [flat|nested] 9+ messages in thread
end of thread, other threads:[~2022-06-11 13:18 UTC | newest]
Thread overview: 9+ messages (download: mbox.gz follow: Atom feed
-- links below jump to the message on this page --
2022-06-08 11:12 [PATCH v3] io_uring: switch cancel_hash to use per entry spinlock Hao Xu
2022-06-08 11:27 ` Pavel Begunkov
2022-06-10 10:21 ` Pavel Begunkov
2022-06-10 15:45 ` Hao Xu
2022-06-10 16:10 ` Pavel Begunkov
2022-06-10 17:40 ` Hao Xu
2022-06-10 17:45 ` Pavel Begunkov
2022-06-11 4:34 ` Hao Xu
2022-06-11 13:18 ` Pavel Begunkov
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox