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); |
Omar Sandoval | 81380ca | 2017-04-07 08:56:26 -0600 | [diff] [blame] | 33 | bool blk_mq_dispatch_rq_list(struct request_queue *, struct list_head *); |
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 | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 57 | void blk_mq_insert_requests(struct blk_mq_hw_ctx *hctx, struct blk_mq_ctx *ctx, |
| 58 | struct list_head *list); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 59 | /* |
| 60 | * CPU hotplug helpers |
| 61 | */ |
Jens Axboe | 676141e | 2014-03-20 13:29:18 -0600 | [diff] [blame] | 62 | void blk_mq_enable_hotplug(void); |
| 63 | void blk_mq_disable_hotplug(void); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 64 | |
| 65 | /* |
| 66 | * CPU -> queue mappings |
| 67 | */ |
Jens Axboe | f14bbe7 | 2014-05-27 12:06:53 -0600 | [diff] [blame] | 68 | 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] | 69 | |
Christoph Hellwig | 7d7e0f9 | 2016-09-14 16:18:54 +0200 | [diff] [blame] | 70 | static inline struct blk_mq_hw_ctx *blk_mq_map_queue(struct request_queue *q, |
| 71 | int cpu) |
| 72 | { |
| 73 | return q->queue_hw_ctx[q->mq_map[cpu]]; |
| 74 | } |
| 75 | |
Jens Axboe | e93ecf6 | 2014-05-19 09:17:48 -0600 | [diff] [blame] | 76 | /* |
Jens Axboe | 67aec14 | 2014-05-30 08:25:36 -0600 | [diff] [blame] | 77 | * sysfs helpers |
| 78 | */ |
Ming Lei | 737f98c | 2017-02-22 18:13:59 +0800 | [diff] [blame] | 79 | extern void blk_mq_sysfs_init(struct request_queue *q); |
Ming Lei | 7ea5fe3 | 2017-02-22 18:14:00 +0800 | [diff] [blame] | 80 | extern void blk_mq_sysfs_deinit(struct request_queue *q); |
Bart Van Assche | 2d0364c | 2017-04-26 13:47:48 -0700 | [diff] [blame] | 81 | 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] | 82 | extern int blk_mq_sysfs_register(struct request_queue *q); |
| 83 | extern void blk_mq_sysfs_unregister(struct request_queue *q); |
Keith Busch | 868f2f0 | 2015-12-17 17:08:14 -0700 | [diff] [blame] | 84 | 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] | 85 | |
Christoph Hellwig | 9041583 | 2014-09-22 10:21:48 -0600 | [diff] [blame] | 86 | extern void blk_mq_rq_timed_out(struct request *req, bool reserved); |
| 87 | |
Ming Lei | e09aae7 | 2015-01-29 20:17:27 +0800 | [diff] [blame] | 88 | void blk_mq_release(struct request_queue *q); |
| 89 | |
Ming Lei | 1aecfe4 | 2014-06-01 00:43:36 +0800 | [diff] [blame] | 90 | static inline struct blk_mq_ctx *__blk_mq_get_ctx(struct request_queue *q, |
| 91 | unsigned int cpu) |
| 92 | { |
| 93 | return per_cpu_ptr(q->queue_ctx, cpu); |
| 94 | } |
| 95 | |
| 96 | /* |
| 97 | * This assumes per-cpu software queueing queues. They could be per-node |
| 98 | * as well, for instance. For now this is hardcoded as-is. Note that we don't |
| 99 | * care about preemption, since we know the ctx's are persistent. This does |
| 100 | * mean that we can't rely on ctx always matching the currently running CPU. |
| 101 | */ |
| 102 | static inline struct blk_mq_ctx *blk_mq_get_ctx(struct request_queue *q) |
| 103 | { |
| 104 | return __blk_mq_get_ctx(q, get_cpu()); |
| 105 | } |
| 106 | |
| 107 | static inline void blk_mq_put_ctx(struct blk_mq_ctx *ctx) |
| 108 | { |
| 109 | put_cpu(); |
| 110 | } |
| 111 | |
Ming Lei | cb96a42c | 2014-06-01 00:43:37 +0800 | [diff] [blame] | 112 | struct blk_mq_alloc_data { |
| 113 | /* input parameter */ |
| 114 | struct request_queue *q; |
Christoph Hellwig | 6f3b0e8 | 2015-11-26 09:13:05 +0100 | [diff] [blame] | 115 | unsigned int flags; |
Omar Sandoval | 229a9287 | 2017-04-14 00:59:59 -0700 | [diff] [blame] | 116 | unsigned int shallow_depth; |
Ming Lei | cb96a42c | 2014-06-01 00:43:37 +0800 | [diff] [blame] | 117 | |
| 118 | /* input & output parameter */ |
| 119 | struct blk_mq_ctx *ctx; |
| 120 | struct blk_mq_hw_ctx *hctx; |
| 121 | }; |
| 122 | |
Jens Axboe | 4941115 | 2017-01-13 08:09:05 -0700 | [diff] [blame] | 123 | static inline struct blk_mq_tags *blk_mq_tags_from_data(struct blk_mq_alloc_data *data) |
| 124 | { |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 125 | if (data->flags & BLK_MQ_REQ_INTERNAL) |
| 126 | return data->hctx->sched_tags; |
| 127 | |
Jens Axboe | 4941115 | 2017-01-13 08:09:05 -0700 | [diff] [blame] | 128 | return data->hctx->tags; |
| 129 | } |
| 130 | |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 131 | /* |
| 132 | * Internal helpers for request allocation/init/free |
| 133 | */ |
| 134 | void blk_mq_rq_ctx_init(struct request_queue *q, struct blk_mq_ctx *ctx, |
| 135 | struct request *rq, unsigned int op); |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 136 | void __blk_mq_finish_request(struct blk_mq_hw_ctx *hctx, struct blk_mq_ctx *ctx, |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 137 | struct request *rq); |
Jens Axboe | bd166ef | 2017-01-17 06:03:22 -0700 | [diff] [blame] | 138 | void blk_mq_finish_request(struct request *rq); |
Jens Axboe | 2c3ad66 | 2016-12-14 14:34:47 -0700 | [diff] [blame] | 139 | struct request *__blk_mq_alloc_request(struct blk_mq_alloc_data *data, |
| 140 | unsigned int op); |
| 141 | |
Bart Van Assche | 5d1b25c | 2016-10-28 17:19:15 -0700 | [diff] [blame] | 142 | static inline bool blk_mq_hctx_stopped(struct blk_mq_hw_ctx *hctx) |
| 143 | { |
| 144 | return test_bit(BLK_MQ_S_STOPPED, &hctx->state); |
| 145 | } |
| 146 | |
Ming Lei | 19c66e5 | 2014-12-03 19:38:04 +0800 | [diff] [blame] | 147 | static inline bool blk_mq_hw_queue_mapped(struct blk_mq_hw_ctx *hctx) |
| 148 | { |
| 149 | return hctx->nr_ctx && hctx->tags; |
| 150 | } |
| 151 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 152 | #endif |