blk-mq.h 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. #ifndef BLK_MQ_H
  2. #define BLK_MQ_H
  3. #include <linux/blkdev.h>
  4. struct blk_mq_tags;
  5. struct blk_mq_cpu_notifier {
  6. struct list_head list;
  7. void *data;
  8. int (*notify)(void *data, unsigned long action, unsigned int cpu);
  9. };
  10. struct blk_mq_ctxmap {
  11. unsigned int map_size;
  12. unsigned int bits_per_word;
  13. struct blk_align_bitmap *map;
  14. };
  15. struct blk_mq_hw_ctx {
  16. struct {
  17. spinlock_t lock;
  18. struct list_head dispatch;
  19. } ____cacheline_aligned_in_smp;
  20. unsigned long state; /* BLK_MQ_S_* flags */
  21. struct delayed_work run_work;
  22. struct delayed_work delay_work;
  23. cpumask_var_t cpumask;
  24. int next_cpu;
  25. int next_cpu_batch;
  26. unsigned long flags; /* BLK_MQ_F_* flags */
  27. struct request_queue *queue;
  28. unsigned int queue_num;
  29. void *driver_data;
  30. struct blk_mq_ctxmap ctx_map;
  31. unsigned int nr_ctx;
  32. struct blk_mq_ctx **ctxs;
  33. atomic_t wait_index;
  34. struct blk_mq_tags *tags;
  35. unsigned long queued;
  36. unsigned long run;
  37. #define BLK_MQ_MAX_DISPATCH_ORDER 10
  38. unsigned long dispatched[BLK_MQ_MAX_DISPATCH_ORDER];
  39. unsigned int numa_node;
  40. unsigned int cmd_size; /* per-request extra data */
  41. atomic_t nr_active;
  42. struct blk_mq_cpu_notifier cpu_notifier;
  43. struct kobject kobj;
  44. };
  45. struct blk_mq_tag_set {
  46. struct blk_mq_ops *ops;
  47. unsigned int nr_hw_queues;
  48. unsigned int queue_depth; /* max hw supported */
  49. unsigned int reserved_tags;
  50. unsigned int cmd_size; /* per-request extra data */
  51. int numa_node;
  52. unsigned int timeout;
  53. unsigned int flags; /* BLK_MQ_F_* */
  54. void *driver_data;
  55. struct blk_mq_tags **tags;
  56. struct mutex tag_list_lock;
  57. struct list_head tag_list;
  58. };
  59. typedef int (queue_rq_fn)(struct blk_mq_hw_ctx *, struct request *, bool);
  60. typedef struct blk_mq_hw_ctx *(map_queue_fn)(struct request_queue *, const int);
  61. typedef int (init_hctx_fn)(struct blk_mq_hw_ctx *, void *, unsigned int);
  62. typedef void (exit_hctx_fn)(struct blk_mq_hw_ctx *, unsigned int);
  63. typedef int (init_request_fn)(void *, struct request *, unsigned int,
  64. unsigned int, unsigned int);
  65. typedef void (exit_request_fn)(void *, struct request *, unsigned int,
  66. unsigned int);
  67. struct blk_mq_ops {
  68. /*
  69. * Queue request
  70. */
  71. queue_rq_fn *queue_rq;
  72. /*
  73. * Map to specific hardware queue
  74. */
  75. map_queue_fn *map_queue;
  76. /*
  77. * Called on request timeout
  78. */
  79. rq_timed_out_fn *timeout;
  80. softirq_done_fn *complete;
  81. /*
  82. * Called when the block layer side of a hardware queue has been
  83. * set up, allowing the driver to allocate/init matching structures.
  84. * Ditto for exit/teardown.
  85. */
  86. init_hctx_fn *init_hctx;
  87. exit_hctx_fn *exit_hctx;
  88. /*
  89. * Called for every command allocated by the block layer to allow
  90. * the driver to set up driver specific data.
  91. * Ditto for exit/teardown.
  92. */
  93. init_request_fn *init_request;
  94. exit_request_fn *exit_request;
  95. };
  96. enum {
  97. BLK_MQ_RQ_QUEUE_OK = 0, /* queued fine */
  98. BLK_MQ_RQ_QUEUE_BUSY = 1, /* requeue IO for later */
  99. BLK_MQ_RQ_QUEUE_ERROR = 2, /* end IO with error */
  100. BLK_MQ_F_SHOULD_MERGE = 1 << 0,
  101. BLK_MQ_F_TAG_SHARED = 1 << 1,
  102. BLK_MQ_F_SG_MERGE = 1 << 2,
  103. BLK_MQ_F_SYSFS_UP = 1 << 3,
  104. BLK_MQ_S_STOPPED = 0,
  105. BLK_MQ_S_TAG_ACTIVE = 1,
  106. BLK_MQ_MAX_DEPTH = 10240,
  107. BLK_MQ_CPU_WORK_BATCH = 8,
  108. };
  109. struct request_queue *blk_mq_init_queue(struct blk_mq_tag_set *);
  110. int blk_mq_register_disk(struct gendisk *);
  111. void blk_mq_unregister_disk(struct gendisk *);
  112. int blk_mq_alloc_tag_set(struct blk_mq_tag_set *set);
  113. void blk_mq_free_tag_set(struct blk_mq_tag_set *set);
  114. void blk_mq_flush_plug_list(struct blk_plug *plug, bool from_schedule);
  115. void blk_mq_insert_request(struct request *, bool, bool, bool);
  116. void blk_mq_run_queues(struct request_queue *q, bool async);
  117. void blk_mq_free_request(struct request *rq);
  118. bool blk_mq_can_queue(struct blk_mq_hw_ctx *);
  119. struct request *blk_mq_alloc_request(struct request_queue *q, int rw,
  120. gfp_t gfp, bool reserved);
  121. struct request *blk_mq_tag_to_rq(struct blk_mq_tags *tags, unsigned int tag);
  122. struct blk_mq_hw_ctx *blk_mq_map_queue(struct request_queue *, const int ctx_index);
  123. struct blk_mq_hw_ctx *blk_mq_alloc_single_hw_queue(struct blk_mq_tag_set *, unsigned int, int);
  124. void blk_mq_start_request(struct request *rq);
  125. void blk_mq_end_request(struct request *rq, int error);
  126. void __blk_mq_end_request(struct request *rq, int error);
  127. void blk_mq_requeue_request(struct request *rq);
  128. void blk_mq_add_to_requeue_list(struct request *rq, bool at_head);
  129. void blk_mq_kick_requeue_list(struct request_queue *q);
  130. void blk_mq_complete_request(struct request *rq);
  131. void blk_mq_stop_hw_queue(struct blk_mq_hw_ctx *hctx);
  132. void blk_mq_start_hw_queue(struct blk_mq_hw_ctx *hctx);
  133. void blk_mq_stop_hw_queues(struct request_queue *q);
  134. void blk_mq_start_hw_queues(struct request_queue *q);
  135. void blk_mq_start_stopped_hw_queues(struct request_queue *q, bool async);
  136. void blk_mq_delay_queue(struct blk_mq_hw_ctx *hctx, unsigned long msecs);
  137. void blk_mq_tag_busy_iter(struct blk_mq_tags *tags, void (*fn)(void *data, unsigned long *), void *data);
  138. /*
  139. * Driver command data is immediately after the request. So subtract request
  140. * size to get back to the original request.
  141. */
  142. static inline struct request *blk_mq_rq_from_pdu(void *pdu)
  143. {
  144. return pdu - sizeof(struct request);
  145. }
  146. static inline void *blk_mq_rq_to_pdu(struct request *rq)
  147. {
  148. return (void *) rq + sizeof(*rq);
  149. }
  150. #define queue_for_each_hw_ctx(q, hctx, i) \
  151. for ((i) = 0; (i) < (q)->nr_hw_queues && \
  152. ({ hctx = (q)->queue_hw_ctx[i]; 1; }); (i)++)
  153. #define queue_for_each_ctx(q, ctx, i) \
  154. for ((i) = 0; (i) < (q)->nr_queues && \
  155. ({ ctx = per_cpu_ptr((q)->queue_ctx, (i)); 1; }); (i)++)
  156. #define hctx_for_each_ctx(hctx, ctx, i) \
  157. for ((i) = 0; (i) < (hctx)->nr_ctx && \
  158. ({ ctx = (hctx)->ctxs[(i)]; 1; }); (i)++)
  159. #define blk_ctx_sum(q, sum) \
  160. ({ \
  161. struct blk_mq_ctx *__x; \
  162. unsigned int __ret = 0, __i; \
  163. \
  164. queue_for_each_ctx((q), __x, __i) \
  165. __ret += sum; \
  166. __ret; \
  167. })
  168. #endif