Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 1 | #ifndef INT_BLK_MQ_H |
| 2 | #define INT_BLK_MQ_H |
| 3 | |
Jens Axboe | cf43e6b | 2016-11-07 21:32:37 -0700 | [diff] [blame] | 4 | #include "blk-stat.h" |
| 5 | |
Christoph Hellwig | 24d2f90 | 2014-04-15 14:14:00 -0600 | [diff] [blame] | 6 | struct blk_mq_tag_set; |
| 7 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 8 | struct blk_mq_ctx { |
| 9 | struct { |
| 10 | spinlock_t lock; |
| 11 | struct list_head rq_list; |
| 12 | } ____cacheline_aligned_in_smp; |
| 13 | |
| 14 | unsigned int cpu; |
| 15 | unsigned int index_hw; |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 16 | |
| 17 | /* incremented at dispatch time */ |
| 18 | unsigned long rq_dispatched[2]; |
| 19 | unsigned long rq_merged; |
| 20 | |
| 21 | /* incremented at completion time */ |
| 22 | unsigned long ____cacheline_aligned_in_smp rq_completed[2]; |
| 23 | |
| 24 | struct request_queue *queue; |
| 25 | struct kobject kobj; |
Jens Axboe | 4bb659b | 2014-05-09 09:36:49 -0600 | [diff] [blame] | 26 | } ____cacheline_aligned_in_smp; |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 27 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 28 | void blk_mq_run_hw_queue(struct blk_mq_hw_ctx *hctx, bool async); |
Tejun Heo | 780db20 | 2014-07-01 10:31:13 -0600 | [diff] [blame] | 29 | void blk_mq_freeze_queue(struct request_queue *q); |
Ming Lei | 3edcc0c | 2013-12-26 21:31:38 +0800 | [diff] [blame] | 30 | void blk_mq_free_queue(struct request_queue *q); |
Jens Axboe | e3a2b3f | 2014-05-20 11:49:02 -0600 | [diff] [blame] | 31 | int blk_mq_update_nr_requests(struct request_queue *q, unsigned int nr); |
Jens Axboe | aed3ea9 | 2014-12-22 14:04:42 -0700 | [diff] [blame] | 32 | void blk_mq_wake_waiters(struct request_queue *q); |
Ming Lei | de14829 | 2017-10-14 17:22:29 +0800 | [diff] [blame^] | 33 | bool blk_mq_dispatch_rq_list(struct request_queue *, struct list_head *, bool); |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 34 | void blk_mq_flush_busy_ctxs(struct blk_mq_hw_ctx *hctx, struct list_head *list); |
Jens Axboe | 50e1dab | 2017-01-26 14:42:34 -0700 | [diff] [blame] | 35 | bool blk_mq_hctx_has_pending(struct blk_mq_hw_ctx *hctx); |
Jens Axboe | bd6737f | 2017-01-27 01:00:47 -0700 | [diff] [blame] | 36 | bool blk_mq_get_driver_tag(struct request *rq, struct blk_mq_hw_ctx **hctx, |
| 37 | bool wait); |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 38 | |
| 39 | /* |
| 40 | * Internal helpers for allocating/freeing the request map |
| 41 | */ |
Jens Axboe | cc71a6f | 2017-01-11 14:29:56 -0700 | [diff] [blame] | 42 | void blk_mq_free_rqs(struct blk_mq_tag_set *set, struct blk_mq_tags *tags, |
| 43 | unsigned int hctx_idx); |
| 44 | void blk_mq_free_rq_map(struct blk_mq_tags *tags); |
| 45 | struct blk_mq_tags *blk_mq_alloc_rq_map(struct blk_mq_tag_set *set, |
| 46 | unsigned int hctx_idx, |
| 47 | unsigned int nr_tags, |
| 48 | unsigned int reserved_tags); |
| 49 | int blk_mq_alloc_rqs(struct blk_mq_tag_set *set, struct blk_mq_tags *tags, |
| 50 | unsigned int hctx_idx, unsigned int depth); |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 51 | |
| 52 | /* |
| 53 | * Internal helpers for request insertion into sw queues |
| 54 | */ |
| 55 | void __blk_mq_insert_request(struct blk_mq_hw_ctx *hctx, struct request *rq, |
| 56 | bool at_head); |
Jens Axboe | 157f377 | 2017-09-11 16:43:57 -0600 | [diff] [blame] | 57 | void blk_mq_request_bypass_insert(struct request *rq); |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 58 | void blk_mq_insert_requests(struct blk_mq_hw_ctx *hctx, struct blk_mq_ctx *ctx, |
| 59 | struct list_head *list); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 60 | |
| 61 | /* |
| 62 | * CPU -> queue mappings |
| 63 | */ |
Jens Axboe | f14bbe7 | 2014-05-27 12:06:53 -0600 | [diff] [blame] | 64 | extern int blk_mq_hw_queue_to_node(unsigned int *map, unsigned int); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 65 | |
Christoph Hellwig | 7d7e0f9 | 2016-09-14 16:18:54 +0200 | [diff] [blame] | 66 | static inline struct blk_mq_hw_ctx *blk_mq_map_queue(struct request_queue *q, |
| 67 | int cpu) |
| 68 | { |
| 69 | return q->queue_hw_ctx[q->mq_map[cpu]]; |
| 70 | } |
| 71 | |
Jens Axboe | e93ecf6 | 2014-05-19 09:17:48 -0600 | [diff] [blame] | 72 | /* |
Jens Axboe | 67aec14 | 2014-05-30 08:25:36 -0600 | [diff] [blame] | 73 | * sysfs helpers |
| 74 | */ |
Ming Lei | 737f98c | 2017-02-22 18:13:59 +0800 | [diff] [blame] | 75 | extern void blk_mq_sysfs_init(struct request_queue *q); |
Ming Lei | 7ea5fe3 | 2017-02-22 18:14:00 +0800 | [diff] [blame] | 76 | extern void blk_mq_sysfs_deinit(struct request_queue *q); |
Bart Van Assche | 2d0364c | 2017-04-26 13:47:48 -0700 | [diff] [blame] | 77 | extern int __blk_mq_register_dev(struct device *dev, struct request_queue *q); |
Jens Axboe | 67aec14 | 2014-05-30 08:25:36 -0600 | [diff] [blame] | 78 | extern int blk_mq_sysfs_register(struct request_queue *q); |
| 79 | extern void blk_mq_sysfs_unregister(struct request_queue *q); |
Keith Busch | 868f2f0 | 2015-12-17 17:08:14 -0700 | [diff] [blame] | 80 | extern void blk_mq_hctx_kobj_init(struct blk_mq_hw_ctx *hctx); |
Jens Axboe | 67aec14 | 2014-05-30 08:25:36 -0600 | [diff] [blame] | 81 | |
Christoph Hellwig | 9041583 | 2014-09-22 10:21:48 -0600 | [diff] [blame] | 82 | extern void blk_mq_rq_timed_out(struct request *req, bool reserved); |
| 83 | |
Ming Lei | e09aae7e | 2015-01-29 20:17:27 +0800 | [diff] [blame] | 84 | void blk_mq_release(struct request_queue *q); |
| 85 | |
Ming Lei | 1aecfe4 | 2014-06-01 00:43:36 +0800 | [diff] [blame] | 86 | static inline struct blk_mq_ctx *__blk_mq_get_ctx(struct request_queue *q, |
| 87 | unsigned int cpu) |
| 88 | { |
| 89 | return per_cpu_ptr(q->queue_ctx, cpu); |
| 90 | } |
| 91 | |
| 92 | /* |
| 93 | * This assumes per-cpu software queueing queues. They could be per-node |
| 94 | * as well, for instance. For now this is hardcoded as-is. Note that we don't |
| 95 | * care about preemption, since we know the ctx's are persistent. This does |
| 96 | * mean that we can't rely on ctx always matching the currently running CPU. |
| 97 | */ |
| 98 | static inline struct blk_mq_ctx *blk_mq_get_ctx(struct request_queue *q) |
| 99 | { |
| 100 | return __blk_mq_get_ctx(q, get_cpu()); |
| 101 | } |
| 102 | |
| 103 | static inline void blk_mq_put_ctx(struct blk_mq_ctx *ctx) |
| 104 | { |
| 105 | put_cpu(); |
| 106 | } |
| 107 | |
Ming Lei | cb96a42 | 2014-06-01 00:43:37 +0800 | [diff] [blame] | 108 | struct blk_mq_alloc_data { |
| 109 | /* input parameter */ |
| 110 | struct request_queue *q; |
Christoph Hellwig | 6f3b0e8 | 2015-11-26 09:13:05 +0100 | [diff] [blame] | 111 | unsigned int flags; |
Omar Sandoval | 229a9287 | 2017-04-14 00:59:59 -0700 | [diff] [blame] | 112 | unsigned int shallow_depth; |
Ming Lei | cb96a42 | 2014-06-01 00:43:37 +0800 | [diff] [blame] | 113 | |
| 114 | /* input & output parameter */ |
| 115 | struct blk_mq_ctx *ctx; |
| 116 | struct blk_mq_hw_ctx *hctx; |
| 117 | }; |
| 118 | |
Jens Axboe | 4941115 | 2017-01-13 08:09:05 -0700 | [diff] [blame] | 119 | static inline struct blk_mq_tags *blk_mq_tags_from_data(struct blk_mq_alloc_data *data) |
| 120 | { |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 121 | if (data->flags & BLK_MQ_REQ_INTERNAL) |
| 122 | return data->hctx->sched_tags; |
| 123 | |
Jens Axboe | 4941115 | 2017-01-13 08:09:05 -0700 | [diff] [blame] | 124 | return data->hctx->tags; |
| 125 | } |
| 126 | |
Bart Van Assche | 5d1b25c | 2016-10-28 17:19:15 -0700 | [diff] [blame] | 127 | static inline bool blk_mq_hctx_stopped(struct blk_mq_hw_ctx *hctx) |
| 128 | { |
| 129 | return test_bit(BLK_MQ_S_STOPPED, &hctx->state); |
| 130 | } |
| 131 | |
Ming Lei | 19c66e5 | 2014-12-03 19:38:04 +0800 | [diff] [blame] | 132 | static inline bool blk_mq_hw_queue_mapped(struct blk_mq_hw_ctx *hctx) |
| 133 | { |
| 134 | return hctx->nr_ctx && hctx->tags; |
| 135 | } |
| 136 | |
Jens Axboe | f299b7c | 2017-08-08 17:51:45 -0600 | [diff] [blame] | 137 | void blk_mq_in_flight(struct request_queue *q, struct hd_struct *part, |
| 138 | unsigned int inflight[2]); |
| 139 | |
Ming Lei | de14829 | 2017-10-14 17:22:29 +0800 | [diff] [blame^] | 140 | static inline void blk_mq_put_dispatch_budget(struct blk_mq_hw_ctx *hctx) |
| 141 | { |
| 142 | struct request_queue *q = hctx->queue; |
| 143 | |
| 144 | if (q->mq_ops->put_budget) |
| 145 | q->mq_ops->put_budget(hctx); |
| 146 | } |
| 147 | |
| 148 | static inline blk_status_t blk_mq_get_dispatch_budget( |
| 149 | struct blk_mq_hw_ctx *hctx) |
| 150 | { |
| 151 | struct request_queue *q = hctx->queue; |
| 152 | |
| 153 | if (q->mq_ops->get_budget) |
| 154 | return q->mq_ops->get_budget(hctx); |
| 155 | return BLK_STS_OK; |
| 156 | } |
| 157 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 158 | #endif |