1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef BLK_INTERNAL_H 3 #define BLK_INTERNAL_H 4 5 #include <linux/idr.h> 6 #include <linux/blk-mq.h> 7 #include <linux/part_stat.h> 8 #include <linux/blk-crypto.h> 9 #include <xen/xen.h> 10 #include "blk-crypto-internal.h" 11 #include "blk-mq.h" 12 #include "blk-mq-sched.h" 13 14 /* Max future timer expiry for timeouts */ 15 #define BLK_MAX_TIMEOUT (5 * HZ) 16 17 extern struct dentry *blk_debugfs_root; 18 19 struct blk_flush_queue { 20 unsigned int flush_pending_idx:1; 21 unsigned int flush_running_idx:1; 22 blk_status_t rq_status; 23 unsigned long flush_pending_since; 24 struct list_head flush_queue[2]; 25 struct list_head flush_data_in_flight; 26 struct request *flush_rq; 27 28 struct lock_class_key key; 29 spinlock_t mq_flush_lock; 30 }; 31 32 extern struct kmem_cache *blk_requestq_cachep; 33 extern struct kobj_type blk_queue_ktype; 34 extern struct ida blk_queue_ida; 35 36 static inline struct blk_flush_queue * 37 blk_get_flush_queue(struct request_queue *q, struct blk_mq_ctx *ctx) 38 { 39 return blk_mq_map_queue(q, REQ_OP_FLUSH, ctx)->fq; 40 } 41 42 static inline void __blk_get_queue(struct request_queue *q) 43 { 44 kobject_get(&q->kobj); 45 } 46 47 static inline bool 48 is_flush_rq(struct request *req, struct blk_mq_hw_ctx *hctx) 49 { 50 return hctx->fq->flush_rq == req; 51 } 52 53 struct blk_flush_queue *blk_alloc_flush_queue(int node, int cmd_size, 54 gfp_t flags); 55 void blk_free_flush_queue(struct blk_flush_queue *q); 56 57 void blk_freeze_queue(struct request_queue *q); 58 59 static inline bool biovec_phys_mergeable(struct request_queue *q, 60 struct bio_vec *vec1, struct bio_vec *vec2) 61 { 62 unsigned long mask = queue_segment_boundary(q); 63 phys_addr_t addr1 = page_to_phys(vec1->bv_page) + vec1->bv_offset; 64 phys_addr_t addr2 = page_to_phys(vec2->bv_page) + vec2->bv_offset; 65 66 if (addr1 + vec1->bv_len != addr2) 67 return false; 68 if (xen_domain() && !xen_biovec_phys_mergeable(vec1, vec2->bv_page)) 69 return false; 70 if ((addr1 | mask) != ((addr2 + vec2->bv_len - 1) | mask)) 71 return false; 72 return true; 73 } 74 75 static inline bool __bvec_gap_to_prev(struct request_queue *q, 76 struct bio_vec *bprv, unsigned int offset) 77 { 78 return (offset & queue_virt_boundary(q)) || 79 ((bprv->bv_offset + bprv->bv_len) & queue_virt_boundary(q)); 80 } 81 82 /* 83 * Check if adding a bio_vec after bprv with offset would create a gap in 84 * the SG list. Most drivers don't care about this, but some do. 85 */ 86 static inline bool bvec_gap_to_prev(struct request_queue *q, 87 struct bio_vec *bprv, unsigned int offset) 88 { 89 if (!queue_virt_boundary(q)) 90 return false; 91 return __bvec_gap_to_prev(q, bprv, offset); 92 } 93 94 static inline void blk_rq_bio_prep(struct request *rq, struct bio *bio, 95 unsigned int nr_segs) 96 { 97 rq->nr_phys_segments = nr_segs; 98 rq->__data_len = bio->bi_iter.bi_size; 99 rq->bio = rq->biotail = bio; 100 rq->ioprio = bio_prio(bio); 101 102 if (bio->bi_disk) 103 rq->rq_disk = bio->bi_disk; 104 } 105 106 #ifdef CONFIG_BLK_DEV_INTEGRITY 107 void blk_flush_integrity(void); 108 bool __bio_integrity_endio(struct bio *); 109 void bio_integrity_free(struct bio *bio); 110 static inline bool bio_integrity_endio(struct bio *bio) 111 { 112 if (bio_integrity(bio)) 113 return __bio_integrity_endio(bio); 114 return true; 115 } 116 117 static inline bool integrity_req_gap_back_merge(struct request *req, 118 struct bio *next) 119 { 120 struct bio_integrity_payload *bip = bio_integrity(req->bio); 121 struct bio_integrity_payload *bip_next = bio_integrity(next); 122 123 return bvec_gap_to_prev(req->q, &bip->bip_vec[bip->bip_vcnt - 1], 124 bip_next->bip_vec[0].bv_offset); 125 } 126 127 static inline bool integrity_req_gap_front_merge(struct request *req, 128 struct bio *bio) 129 { 130 struct bio_integrity_payload *bip = bio_integrity(bio); 131 struct bio_integrity_payload *bip_next = bio_integrity(req->bio); 132 133 return bvec_gap_to_prev(req->q, &bip->bip_vec[bip->bip_vcnt - 1], 134 bip_next->bip_vec[0].bv_offset); 135 } 136 137 void blk_integrity_add(struct gendisk *); 138 void blk_integrity_del(struct gendisk *); 139 #else /* CONFIG_BLK_DEV_INTEGRITY */ 140 static inline bool integrity_req_gap_back_merge(struct request *req, 141 struct bio *next) 142 { 143 return false; 144 } 145 static inline bool integrity_req_gap_front_merge(struct request *req, 146 struct bio *bio) 147 { 148 return false; 149 } 150 151 static inline void blk_flush_integrity(void) 152 { 153 } 154 static inline bool bio_integrity_endio(struct bio *bio) 155 { 156 return true; 157 } 158 static inline void bio_integrity_free(struct bio *bio) 159 { 160 } 161 static inline void blk_integrity_add(struct gendisk *disk) 162 { 163 } 164 static inline void blk_integrity_del(struct gendisk *disk) 165 { 166 } 167 #endif /* CONFIG_BLK_DEV_INTEGRITY */ 168 169 unsigned long blk_rq_timeout(unsigned long timeout); 170 void blk_add_timer(struct request *req); 171 172 bool bio_attempt_front_merge(struct request *req, struct bio *bio, 173 unsigned int nr_segs); 174 bool bio_attempt_back_merge(struct request *req, struct bio *bio, 175 unsigned int nr_segs); 176 bool bio_attempt_discard_merge(struct request_queue *q, struct request *req, 177 struct bio *bio); 178 bool blk_attempt_plug_merge(struct request_queue *q, struct bio *bio, 179 unsigned int nr_segs, struct request **same_queue_rq); 180 181 void blk_account_io_start(struct request *req); 182 void blk_account_io_done(struct request *req, u64 now); 183 184 /* 185 * Internal elevator interface 186 */ 187 #define ELV_ON_HASH(rq) ((rq)->rq_flags & RQF_HASHED) 188 189 void blk_insert_flush(struct request *rq); 190 191 void elevator_init_mq(struct request_queue *q); 192 int elevator_switch_mq(struct request_queue *q, 193 struct elevator_type *new_e); 194 void __elevator_exit(struct request_queue *, struct elevator_queue *); 195 int elv_register_queue(struct request_queue *q, bool uevent); 196 void elv_unregister_queue(struct request_queue *q); 197 198 static inline void elevator_exit(struct request_queue *q, 199 struct elevator_queue *e) 200 { 201 lockdep_assert_held(&q->sysfs_lock); 202 203 blk_mq_sched_free_requests(q); 204 __elevator_exit(q, e); 205 } 206 207 struct hd_struct *__disk_get_part(struct gendisk *disk, int partno); 208 209 ssize_t part_size_show(struct device *dev, struct device_attribute *attr, 210 char *buf); 211 ssize_t part_stat_show(struct device *dev, struct device_attribute *attr, 212 char *buf); 213 ssize_t part_inflight_show(struct device *dev, struct device_attribute *attr, 214 char *buf); 215 ssize_t part_fail_show(struct device *dev, struct device_attribute *attr, 216 char *buf); 217 ssize_t part_fail_store(struct device *dev, struct device_attribute *attr, 218 const char *buf, size_t count); 219 ssize_t part_timeout_show(struct device *, struct device_attribute *, char *); 220 ssize_t part_timeout_store(struct device *, struct device_attribute *, 221 const char *, size_t); 222 223 void __blk_queue_split(struct bio **bio, unsigned int *nr_segs); 224 int ll_back_merge_fn(struct request *req, struct bio *bio, 225 unsigned int nr_segs); 226 int ll_front_merge_fn(struct request *req, struct bio *bio, 227 unsigned int nr_segs); 228 struct request *attempt_back_merge(struct request_queue *q, struct request *rq); 229 struct request *attempt_front_merge(struct request_queue *q, struct request *rq); 230 int blk_attempt_req_merge(struct request_queue *q, struct request *rq, 231 struct request *next); 232 unsigned int blk_recalc_rq_segments(struct request *rq); 233 void blk_rq_set_mixed_merge(struct request *rq); 234 bool blk_rq_merge_ok(struct request *rq, struct bio *bio); 235 enum elv_merge blk_try_merge(struct request *rq, struct bio *bio); 236 237 int blk_dev_init(void); 238 239 /* 240 * Contribute to IO statistics IFF: 241 * 242 * a) it's attached to a gendisk, and 243 * b) the queue had IO stats enabled when this request was started 244 */ 245 static inline bool blk_do_io_stat(struct request *rq) 246 { 247 return rq->rq_disk && (rq->rq_flags & RQF_IO_STAT); 248 } 249 250 static inline void req_set_nomerge(struct request_queue *q, struct request *req) 251 { 252 req->cmd_flags |= REQ_NOMERGE; 253 if (req == q->last_merge) 254 q->last_merge = NULL; 255 } 256 257 /* 258 * The max size one bio can handle is UINT_MAX becasue bvec_iter.bi_size 259 * is defined as 'unsigned int', meantime it has to aligned to with logical 260 * block size which is the minimum accepted unit by hardware. 261 */ 262 static inline unsigned int bio_allowed_max_sectors(struct request_queue *q) 263 { 264 return round_down(UINT_MAX, queue_logical_block_size(q)) >> 9; 265 } 266 267 /* 268 * Internal io_context interface 269 */ 270 void get_io_context(struct io_context *ioc); 271 struct io_cq *ioc_lookup_icq(struct io_context *ioc, struct request_queue *q); 272 struct io_cq *ioc_create_icq(struct io_context *ioc, struct request_queue *q, 273 gfp_t gfp_mask); 274 void ioc_clear_queue(struct request_queue *q); 275 276 int create_task_io_context(struct task_struct *task, gfp_t gfp_mask, int node); 277 278 /* 279 * Internal throttling interface 280 */ 281 #ifdef CONFIG_BLK_DEV_THROTTLING 282 extern int blk_throtl_init(struct request_queue *q); 283 extern void blk_throtl_exit(struct request_queue *q); 284 extern void blk_throtl_register_queue(struct request_queue *q); 285 bool blk_throtl_bio(struct bio *bio); 286 #else /* CONFIG_BLK_DEV_THROTTLING */ 287 static inline int blk_throtl_init(struct request_queue *q) { return 0; } 288 static inline void blk_throtl_exit(struct request_queue *q) { } 289 static inline void blk_throtl_register_queue(struct request_queue *q) { } 290 static inline bool blk_throtl_bio(struct bio *bio) { return false; } 291 #endif /* CONFIG_BLK_DEV_THROTTLING */ 292 #ifdef CONFIG_BLK_DEV_THROTTLING_LOW 293 extern ssize_t blk_throtl_sample_time_show(struct request_queue *q, char *page); 294 extern ssize_t blk_throtl_sample_time_store(struct request_queue *q, 295 const char *page, size_t count); 296 extern void blk_throtl_bio_endio(struct bio *bio); 297 extern void blk_throtl_stat_add(struct request *rq, u64 time); 298 #else 299 static inline void blk_throtl_bio_endio(struct bio *bio) { } 300 static inline void blk_throtl_stat_add(struct request *rq, u64 time) { } 301 #endif 302 303 #ifdef CONFIG_BOUNCE 304 extern int init_emergency_isa_pool(void); 305 extern void blk_queue_bounce(struct request_queue *q, struct bio **bio); 306 #else 307 static inline int init_emergency_isa_pool(void) 308 { 309 return 0; 310 } 311 static inline void blk_queue_bounce(struct request_queue *q, struct bio **bio) 312 { 313 } 314 #endif /* CONFIG_BOUNCE */ 315 316 #ifdef CONFIG_BLK_CGROUP_IOLATENCY 317 extern int blk_iolatency_init(struct request_queue *q); 318 #else 319 static inline int blk_iolatency_init(struct request_queue *q) { return 0; } 320 #endif 321 322 struct bio *blk_next_bio(struct bio *bio, unsigned int nr_pages, gfp_t gfp); 323 324 #ifdef CONFIG_BLK_DEV_ZONED 325 void blk_queue_free_zone_bitmaps(struct request_queue *q); 326 #else 327 static inline void blk_queue_free_zone_bitmaps(struct request_queue *q) {} 328 #endif 329 330 struct hd_struct *disk_map_sector_rcu(struct gendisk *disk, sector_t sector); 331 332 int blk_alloc_devt(struct hd_struct *part, dev_t *devt); 333 void blk_free_devt(dev_t devt); 334 void blk_invalidate_devt(dev_t devt); 335 char *disk_name(struct gendisk *hd, int partno, char *buf); 336 #define ADDPART_FLAG_NONE 0 337 #define ADDPART_FLAG_RAID 1 338 #define ADDPART_FLAG_WHOLEDISK 2 339 void delete_partition(struct gendisk *disk, struct hd_struct *part); 340 int bdev_add_partition(struct block_device *bdev, int partno, 341 sector_t start, sector_t length); 342 int bdev_del_partition(struct block_device *bdev, int partno); 343 int bdev_resize_partition(struct block_device *bdev, int partno, 344 sector_t start, sector_t length); 345 int disk_expand_part_tbl(struct gendisk *disk, int target); 346 int hd_ref_init(struct hd_struct *part); 347 348 /* no need to get/put refcount of part0 */ 349 static inline int hd_struct_try_get(struct hd_struct *part) 350 { 351 if (part->partno) 352 return percpu_ref_tryget_live(&part->ref); 353 return 1; 354 } 355 356 static inline void hd_struct_put(struct hd_struct *part) 357 { 358 if (part->partno) 359 percpu_ref_put(&part->ref); 360 } 361 362 static inline void hd_free_part(struct hd_struct *part) 363 { 364 free_percpu(part->dkstats); 365 kfree(part->info); 366 percpu_ref_exit(&part->ref); 367 } 368 369 /* 370 * Any access of part->nr_sects which is not protected by partition 371 * bd_mutex or gendisk bdev bd_mutex, should be done using this 372 * accessor function. 373 * 374 * Code written along the lines of i_size_read() and i_size_write(). 375 * CONFIG_PREEMPTION case optimizes the case of UP kernel with preemption 376 * on. 377 */ 378 static inline sector_t part_nr_sects_read(struct hd_struct *part) 379 { 380 #if BITS_PER_LONG==32 && defined(CONFIG_SMP) 381 sector_t nr_sects; 382 unsigned seq; 383 do { 384 seq = read_seqcount_begin(&part->nr_sects_seq); 385 nr_sects = part->nr_sects; 386 } while (read_seqcount_retry(&part->nr_sects_seq, seq)); 387 return nr_sects; 388 #elif BITS_PER_LONG==32 && defined(CONFIG_PREEMPTION) 389 sector_t nr_sects; 390 391 preempt_disable(); 392 nr_sects = part->nr_sects; 393 preempt_enable(); 394 return nr_sects; 395 #else 396 return part->nr_sects; 397 #endif 398 } 399 400 /* 401 * Should be called with mutex lock held (typically bd_mutex) of partition 402 * to provide mutual exlusion among writers otherwise seqcount might be 403 * left in wrong state leaving the readers spinning infinitely. 404 */ 405 static inline void part_nr_sects_write(struct hd_struct *part, sector_t size) 406 { 407 #if BITS_PER_LONG==32 && defined(CONFIG_SMP) 408 preempt_disable(); 409 write_seqcount_begin(&part->nr_sects_seq); 410 part->nr_sects = size; 411 write_seqcount_end(&part->nr_sects_seq); 412 preempt_enable(); 413 #elif BITS_PER_LONG==32 && defined(CONFIG_PREEMPTION) 414 preempt_disable(); 415 part->nr_sects = size; 416 preempt_enable(); 417 #else 418 part->nr_sects = size; 419 #endif 420 } 421 422 int bio_add_hw_page(struct request_queue *q, struct bio *bio, 423 struct page *page, unsigned int len, unsigned int offset, 424 unsigned int max_sectors, bool *same_page); 425 426 #endif /* BLK_INTERNAL_H */ 427