Christoph Hellwig | 3dcf60bc | 2019-04-30 14:42:43 -0400 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 2 | /* |
| 3 | * Functions related to setting various queue properties from drivers |
| 4 | */ |
| 5 | #include <linux/kernel.h> |
| 6 | #include <linux/module.h> |
| 7 | #include <linux/bio.h> |
| 8 | #include <linux/blkdev.h> |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 9 | #include <linux/blk-mq.h> |
Clark Williams | cf4aebc2 | 2013-02-07 09:46:59 -0600 | [diff] [blame] | 10 | #include <linux/sched/sysctl.h> |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 11 | |
| 12 | #include "blk.h" |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 13 | #include "blk-mq-sched.h" |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 14 | |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 15 | /** |
| 16 | * blk_end_sync_rq - executes a completion event on a request |
| 17 | * @rq: request to complete |
Randy Dunlap | 710027a | 2008-08-19 20:13:11 +0200 | [diff] [blame] | 18 | * @error: end I/O status of the request |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 19 | */ |
Christoph Hellwig | 2a842ac | 2017-06-03 09:38:04 +0200 | [diff] [blame] | 20 | static void blk_end_sync_rq(struct request *rq, blk_status_t error) |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 21 | { |
| 22 | struct completion *waiting = rq->end_io_data; |
| 23 | |
Keith Busch | fb9b16e | 2021-06-10 14:44:36 -0700 | [diff] [blame] | 24 | rq->end_io_data = (void *)(uintptr_t)error; |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 25 | |
| 26 | /* |
| 27 | * complete last, if this is a stack request the process (and thus |
| 28 | * the rq pointer) could be invalid right after this complete() |
| 29 | */ |
| 30 | complete(waiting); |
| 31 | } |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 32 | |
| 33 | /** |
Guoqing Jiang | 8eeed0b | 2021-01-25 05:49:57 +0100 | [diff] [blame] | 34 | * blk_execute_rq_nowait - insert a request to I/O scheduler for execution |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 35 | * @bd_disk: matching gendisk |
| 36 | * @rq: request to insert |
| 37 | * @at_head: insert request at head or tail of queue |
| 38 | * @done: I/O completion handler |
| 39 | * |
| 40 | * Description: |
Randy Dunlap | 710027a | 2008-08-19 20:13:11 +0200 | [diff] [blame] | 41 | * Insert a fully prepared request at the back of the I/O scheduler queue |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 42 | * for execution. Don't wait for completion. |
Muthukumar Ratty | e81ca6f | 2012-06-29 15:31:49 +0000 | [diff] [blame] | 43 | * |
| 44 | * Note: |
| 45 | * This function will invoke @done directly if the queue is dead. |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 46 | */ |
Guoqing Jiang | 8eeed0b | 2021-01-25 05:49:57 +0100 | [diff] [blame] | 47 | void blk_execute_rq_nowait(struct gendisk *bd_disk, struct request *rq, |
| 48 | int at_head, rq_end_io_fn *done) |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 49 | { |
Tejun Heo | 8ba6143 | 2011-12-14 00:33:37 +0100 | [diff] [blame] | 50 | WARN_ON(irqs_disabled()); |
Christoph Hellwig | 57292b5 | 2017-01-31 16:57:29 +0100 | [diff] [blame] | 51 | WARN_ON(!blk_rq_is_passthrough(rq)); |
James Bottomley | bfe159a | 2011-07-07 15:45:40 -0500 | [diff] [blame] | 52 | |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 53 | rq->rq_disk = bd_disk; |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 54 | rq->end_io = done; |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 55 | |
Konstantin Khlebnikov | b5af37a | 2020-05-27 07:24:16 +0200 | [diff] [blame] | 56 | blk_account_io_start(rq); |
Logan Gunthorpe | 48d9b0d | 2019-10-10 17:36:26 -0600 | [diff] [blame] | 57 | |
Ming Lei | 43a5e4e | 2013-12-26 21:31:35 +0800 | [diff] [blame] | 58 | /* |
| 59 | * don't check dying flag for MQ because the request won't |
Bart Van Assche | 68bdf1a | 2016-07-19 08:18:06 -0700 | [diff] [blame] | 60 | * be reused after dying flag is set |
Ming Lei | 43a5e4e | 2013-12-26 21:31:35 +0800 | [diff] [blame] | 61 | */ |
Jens Axboe | a1ce35f | 2018-10-29 10:23:51 -0600 | [diff] [blame] | 62 | blk_mq_sched_insert_request(rq, at_head, true, false); |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 63 | } |
| 64 | EXPORT_SYMBOL_GPL(blk_execute_rq_nowait); |
| 65 | |
Keith Busch | c01b5a8 | 2021-06-10 14:44:34 -0700 | [diff] [blame] | 66 | static bool blk_rq_is_poll(struct request *rq) |
| 67 | { |
| 68 | return rq->mq_hctx && rq->mq_hctx->type == HCTX_TYPE_POLL; |
| 69 | } |
| 70 | |
| 71 | static void blk_rq_poll_completion(struct request *rq, struct completion *wait) |
| 72 | { |
| 73 | do { |
| 74 | blk_poll(rq->q, request_to_qc_t(rq->mq_hctx, rq), true); |
| 75 | cond_resched(); |
| 76 | } while (!completion_done(wait)); |
| 77 | } |
| 78 | |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 79 | /** |
| 80 | * blk_execute_rq - insert a request into queue for execution |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 81 | * @bd_disk: matching gendisk |
| 82 | * @rq: request to insert |
| 83 | * @at_head: insert request at head or tail of queue |
| 84 | * |
| 85 | * Description: |
Randy Dunlap | 710027a | 2008-08-19 20:13:11 +0200 | [diff] [blame] | 86 | * Insert a fully prepared request at the back of the I/O scheduler queue |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 87 | * for execution and wait for completion. |
Keith Busch | fb9b16e | 2021-06-10 14:44:36 -0700 | [diff] [blame] | 88 | * Return: The blk_status_t result provided to blk_mq_end_request(). |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 89 | */ |
Keith Busch | fb9b16e | 2021-06-10 14:44:36 -0700 | [diff] [blame] | 90 | blk_status_t blk_execute_rq(struct gendisk *bd_disk, struct request *rq, int at_head) |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 91 | { |
| 92 | DECLARE_COMPLETION_ONSTACK(wait); |
Mark Lord | 4b19776 | 2010-09-24 09:51:13 -0400 | [diff] [blame] | 93 | unsigned long hang_check; |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 94 | |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 95 | rq->end_io_data = &wait; |
Guoqing Jiang | 8eeed0b | 2021-01-25 05:49:57 +0100 | [diff] [blame] | 96 | blk_execute_rq_nowait(bd_disk, rq, at_head, blk_end_sync_rq); |
Mark Lord | 4b19776 | 2010-09-24 09:51:13 -0400 | [diff] [blame] | 97 | |
| 98 | /* Prevent hang_check timer from firing at us during very long I/O */ |
| 99 | hang_check = sysctl_hung_task_timeout_secs; |
Keith Busch | c01b5a8 | 2021-06-10 14:44:34 -0700 | [diff] [blame] | 100 | |
| 101 | if (blk_rq_is_poll(rq)) |
| 102 | blk_rq_poll_completion(rq, &wait); |
| 103 | else if (hang_check) |
Vladimir Davydov | 5577022 | 2013-02-14 18:19:59 +0400 | [diff] [blame] | 104 | while (!wait_for_completion_io_timeout(&wait, hang_check * (HZ/2))); |
Mark Lord | 4b19776 | 2010-09-24 09:51:13 -0400 | [diff] [blame] | 105 | else |
Vladimir Davydov | 5577022 | 2013-02-14 18:19:59 +0400 | [diff] [blame] | 106 | wait_for_completion_io(&wait); |
Keith Busch | fb9b16e | 2021-06-10 14:44:36 -0700 | [diff] [blame] | 107 | |
| 108 | return (blk_status_t)(uintptr_t)rq->end_io_data; |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 109 | } |
Jens Axboe | 86db1e2 | 2008-01-29 14:53:40 +0100 | [diff] [blame] | 110 | EXPORT_SYMBOL(blk_execute_rq); |