Managed interrupts can not migrate affinity when their CPUs are offline. If the CPU is allowed to shutdown before they're returned, commands dispatched to managed queues won't be able to complete through their irq handlers. Wait in cpu hotplug handler until all inflight requests on the tags are completed or timeout. Wait once for each tags, so we can save time in case of shared tags. Based on the following patch from Keith, and use simple delay-spin instead. https://lore.kernel.org/linux-block/20190405215920.27085-1-keith.busch@xxxxxxxxx/ Cc: Christoph Hellwig <hch@xxxxxx> Cc: Bart Van Assche <bvanassche@xxxxxxx> Cc: Hannes Reinecke <hare@xxxxxxxx> Cc: John Garry <john.garry@xxxxxxxxxx> Cc: Keith Busch <keith.busch@xxxxxxxxx> Cc: Thomas Gleixner <tglx@xxxxxxxxxxxxx> Signed-off-by: Ming Lei <ming.lei@xxxxxxxxxx> --- block/blk-mq-tag.c | 2 +- block/blk-mq-tag.h | 5 ++++ block/blk-mq.c | 65 +++++++++++++++++++++++++++++++++++++++++++++- 3 files changed, 70 insertions(+), 2 deletions(-) diff --git a/block/blk-mq-tag.c b/block/blk-mq-tag.c index 7513c8eaabee..b24334f99c5d 100644 --- a/block/blk-mq-tag.c +++ b/block/blk-mq-tag.c @@ -332,7 +332,7 @@ static void bt_tags_for_each(struct blk_mq_tags *tags, struct sbitmap_queue *bt, * true to continue iterating tags, false to stop. * @priv: Will be passed as second argument to @fn. */ -static void blk_mq_all_tag_busy_iter(struct blk_mq_tags *tags, +void blk_mq_all_tag_busy_iter(struct blk_mq_tags *tags, busy_tag_iter_fn *fn, void *priv) { if (tags->nr_reserved_tags) diff --git a/block/blk-mq-tag.h b/block/blk-mq-tag.h index 61deab0b5a5a..f8de50485b42 100644 --- a/block/blk-mq-tag.h +++ b/block/blk-mq-tag.h @@ -19,6 +19,9 @@ struct blk_mq_tags { struct request **rqs; struct request **static_rqs; struct list_head page_list; + + #define BLK_MQ_TAGS_DRAINED 0 + unsigned long flags; }; @@ -35,6 +38,8 @@ extern int blk_mq_tag_update_depth(struct blk_mq_hw_ctx *hctx, extern void blk_mq_tag_wakeup_all(struct blk_mq_tags *tags, bool); void blk_mq_queue_tag_busy_iter(struct request_queue *q, busy_iter_fn *fn, void *priv); +void blk_mq_all_tag_busy_iter(struct blk_mq_tags *tags, + busy_tag_iter_fn *fn, void *priv); static inline struct sbq_wait_state *bt_wait_ptr(struct sbitmap_queue *bt, struct blk_mq_hw_ctx *hctx) diff --git a/block/blk-mq.c b/block/blk-mq.c index 08a6248d8536..d1d1b1a9628f 100644 --- a/block/blk-mq.c +++ b/block/blk-mq.c @@ -2214,6 +2214,60 @@ int blk_mq_alloc_rqs(struct blk_mq_tag_set *set, struct blk_mq_tags *tags, return -ENOMEM; } +static int blk_mq_hctx_notify_prepare(unsigned int cpu, struct hlist_node *node) +{ + struct blk_mq_hw_ctx *hctx; + struct blk_mq_tags *tags; + + hctx = hlist_entry_safe(node, struct blk_mq_hw_ctx, cpuhp_dead); + tags = hctx->tags; + + if (tags) + clear_bit(BLK_MQ_TAGS_DRAINED, &tags->flags); + + return 0; +} + +static bool blk_mq_count_inflight_rq(struct request *rq, void *data, + bool reserved) +{ + if (blk_mq_rq_state(rq) == MQ_RQ_IN_FLIGHT) + (*(unsigned long *)data)++; + + return true; +} + +static unsigned blk_mq_tags_inflight_rqs(struct blk_mq_tags *tags) +{ + unsigned long cnt = 0; + + blk_mq_all_tag_busy_iter(tags, blk_mq_count_inflight_rq, &cnt); + + return cnt; +} + +static void blk_mq_drain_inflight_rqs(struct blk_mq_tags *tags, int dead_cpu) +{ + unsigned long msecs_left = 1000 * 5; + + if (!tags) + return; + + if (test_and_set_bit(BLK_MQ_TAGS_DRAINED, &tags->flags)) + return; + + while (msecs_left > 0) { + if (!blk_mq_tags_inflight_rqs(tags)) + break; + msleep(5); + msecs_left -= 5; + } + + if (msecs_left > 0) + printk(KERN_WARNING "requests not completed from dead " + "CPU %d\n", dead_cpu); +} + /* * 'cpu' is going away. splice any existing rq_list entries from this * software queue to the hw queue dispatch list, and ensure that it @@ -2245,6 +2299,14 @@ static int blk_mq_hctx_notify_dead(unsigned int cpu, struct hlist_node *node) spin_unlock(&hctx->lock); blk_mq_run_hw_queue(hctx, true); + + /* + * Interrupt for this queue will be shutdown, so wait until all + * requests from this hw queue is done or timeout. + */ + if (cpumask_first_and(hctx->cpumask, cpu_online_mask) >= nr_cpu_ids) + blk_mq_drain_inflight_rqs(hctx->tags, cpu); + return 0; } @@ -3540,7 +3602,8 @@ EXPORT_SYMBOL(blk_mq_rq_cpu); static int __init blk_mq_init(void) { - cpuhp_setup_state_multi(CPUHP_BLK_MQ_DEAD, "block/mq:dead", NULL, + cpuhp_setup_state_multi(CPUHP_BLK_MQ_DEAD, "block/mq:dead", + blk_mq_hctx_notify_prepare, blk_mq_hctx_notify_dead); return 0; } -- 2.20.1
