blk-core.c 47.8 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0
L
Linus Torvalds 已提交
2 3 4 5 6
/*
 * Copyright (C) 1991, 1992 Linus Torvalds
 * Copyright (C) 1994,      Karl Keyte: Added support for disk statistics
 * Elevator latency, (C) 2000  Andrea Arcangeli <andrea@suse.de> SuSE
 * Queue request tables / lock, selectable elevator, Jens Axboe <axboe@suse.de>
7 8
 * kernel-doc documentation started by NeilBrown <neilb@cse.unsw.edu.au>
 *	-  July2000
L
Linus Torvalds 已提交
9 10 11 12 13 14 15 16 17 18 19
 * bio rewrite, highmem i/o, etc, Jens Axboe <axboe@suse.de> - may 2001
 */

/*
 * This handles all read/write requests to block devices
 */
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/backing-dev.h>
#include <linux/bio.h>
#include <linux/blkdev.h>
20
#include <linux/blk-mq.h>
L
Linus Torvalds 已提交
21 22 23 24 25 26 27 28 29
#include <linux/highmem.h>
#include <linux/mm.h>
#include <linux/kernel_stat.h>
#include <linux/string.h>
#include <linux/init.h>
#include <linux/completion.h>
#include <linux/slab.h>
#include <linux/swap.h>
#include <linux/writeback.h>
30
#include <linux/task_io_accounting_ops.h>
31
#include <linux/fault-inject.h>
32
#include <linux/list_sort.h>
T
Tejun Heo 已提交
33
#include <linux/delay.h>
34
#include <linux/ratelimit.h>
L
Lin Ming 已提交
35
#include <linux/pm_runtime.h>
36
#include <linux/blk-cgroup.h>
37
#include <linux/debugfs.h>
38
#include <linux/bpf.h>
39 40 41

#define CREATE_TRACE_POINTS
#include <trace/events/block.h>
L
Linus Torvalds 已提交
42

43
#include "blk.h"
44
#include "blk-mq.h"
45
#include "blk-mq-sched.h"
46
#include "blk-pm.h"
47
#include "blk-rq-qos.h"
48

49 50 51 52
#ifdef CONFIG_DEBUG_FS
struct dentry *blk_debugfs_root;
#endif

53
EXPORT_TRACEPOINT_SYMBOL_GPL(block_bio_remap);
54
EXPORT_TRACEPOINT_SYMBOL_GPL(block_rq_remap);
55
EXPORT_TRACEPOINT_SYMBOL_GPL(block_bio_complete);
K
Keith Busch 已提交
56
EXPORT_TRACEPOINT_SYMBOL_GPL(block_split);
N
NeilBrown 已提交
57
EXPORT_TRACEPOINT_SYMBOL_GPL(block_unplug);
58

59 60
DEFINE_IDA(blk_queue_ida);

L
Linus Torvalds 已提交
61 62 63
/*
 * For queue allocation
 */
64
struct kmem_cache *blk_requestq_cachep;
L
Linus Torvalds 已提交
65 66 67 68

/*
 * Controlling structure to kblockd
 */
69
static struct workqueue_struct *kblockd_workqueue;
L
Linus Torvalds 已提交
70

71 72 73 74 75 76 77
/**
 * blk_queue_flag_set - atomically set a queue flag
 * @flag: flag to be set
 * @q: request queue
 */
void blk_queue_flag_set(unsigned int flag, struct request_queue *q)
{
78
	set_bit(flag, &q->queue_flags);
79 80 81 82 83 84 85 86 87 88
}
EXPORT_SYMBOL(blk_queue_flag_set);

/**
 * blk_queue_flag_clear - atomically clear a queue flag
 * @flag: flag to be cleared
 * @q: request queue
 */
void blk_queue_flag_clear(unsigned int flag, struct request_queue *q)
{
89
	clear_bit(flag, &q->queue_flags);
90 91 92 93 94 95 96 97 98 99 100 101 102
}
EXPORT_SYMBOL(blk_queue_flag_clear);

/**
 * blk_queue_flag_test_and_set - atomically test and set a queue flag
 * @flag: flag to be set
 * @q: request queue
 *
 * Returns the previous value of @flag - 0 if the flag was not set and 1 if
 * the flag was already set.
 */
bool blk_queue_flag_test_and_set(unsigned int flag, struct request_queue *q)
{
103
	return test_and_set_bit(flag, &q->queue_flags);
104 105 106
}
EXPORT_SYMBOL_GPL(blk_queue_flag_test_and_set);

107
void blk_rq_init(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
108
{
109 110
	memset(rq, 0, sizeof(*rq));

L
Linus Torvalds 已提交
111
	INIT_LIST_HEAD(&rq->queuelist);
J
Jens Axboe 已提交
112
	rq->q = q;
113
	rq->__sector = (sector_t) -1;
114 115
	INIT_HLIST_NODE(&rq->hash);
	RB_CLEAR_NODE(&rq->rb_node);
J
Jens Axboe 已提交
116
	rq->tag = -1;
117
	rq->internal_tag = -1;
118
	rq->start_time_ns = ktime_get_ns();
119
	rq->part = NULL;
120
	refcount_set(&rq->ref, 1);
L
Linus Torvalds 已提交
121
}
122
EXPORT_SYMBOL(blk_rq_init);
L
Linus Torvalds 已提交
123

124 125 126 127 128 129 130 131
#define REQ_OP_NAME(name) [REQ_OP_##name] = #name
static const char *const blk_op_name[] = {
	REQ_OP_NAME(READ),
	REQ_OP_NAME(WRITE),
	REQ_OP_NAME(FLUSH),
	REQ_OP_NAME(DISCARD),
	REQ_OP_NAME(SECURE_ERASE),
	REQ_OP_NAME(ZONE_RESET),
132
	REQ_OP_NAME(ZONE_RESET_ALL),
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
	REQ_OP_NAME(WRITE_SAME),
	REQ_OP_NAME(WRITE_ZEROES),
	REQ_OP_NAME(SCSI_IN),
	REQ_OP_NAME(SCSI_OUT),
	REQ_OP_NAME(DRV_IN),
	REQ_OP_NAME(DRV_OUT),
};
#undef REQ_OP_NAME

/**
 * blk_op_str - Return string XXX in the REQ_OP_XXX.
 * @op: REQ_OP_XXX.
 *
 * Description: Centralize block layer function to convert REQ_OP_XXX into
 * string format. Useful in the debugging and tracing bio or request. For
 * invalid REQ_OP_XXX it returns string "UNKNOWN".
 */
inline const char *blk_op_str(unsigned int op)
{
	const char *op_str = "UNKNOWN";

	if (op < ARRAY_SIZE(blk_op_name) && blk_op_name[op])
		op_str = blk_op_name[op];

	return op_str;
}
EXPORT_SYMBOL_GPL(blk_op_str);

161 162 163 164 165 166 167 168 169 170 171 172 173 174
static const struct {
	int		errno;
	const char	*name;
} blk_errors[] = {
	[BLK_STS_OK]		= { 0,		"" },
	[BLK_STS_NOTSUPP]	= { -EOPNOTSUPP, "operation not supported" },
	[BLK_STS_TIMEOUT]	= { -ETIMEDOUT,	"timeout" },
	[BLK_STS_NOSPC]		= { -ENOSPC,	"critical space allocation" },
	[BLK_STS_TRANSPORT]	= { -ENOLINK,	"recoverable transport" },
	[BLK_STS_TARGET]	= { -EREMOTEIO,	"critical target" },
	[BLK_STS_NEXUS]		= { -EBADE,	"critical nexus" },
	[BLK_STS_MEDIUM]	= { -ENODATA,	"critical medium" },
	[BLK_STS_PROTECTION]	= { -EILSEQ,	"protection" },
	[BLK_STS_RESOURCE]	= { -ENOMEM,	"kernel resource" },
175
	[BLK_STS_DEV_RESOURCE]	= { -EBUSY,	"device resource" },
176
	[BLK_STS_AGAIN]		= { -EAGAIN,	"nonblocking retry" },
177

178 179 180
	/* device mapper special case, should not leak out: */
	[BLK_STS_DM_REQUEUE]	= { -EREMCHG, "dm internal retry" },

181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
	/* everything else not covered above: */
	[BLK_STS_IOERR]		= { -EIO,	"I/O" },
};

blk_status_t errno_to_blk_status(int errno)
{
	int i;

	for (i = 0; i < ARRAY_SIZE(blk_errors); i++) {
		if (blk_errors[i].errno == errno)
			return (__force blk_status_t)i;
	}

	return BLK_STS_IOERR;
}
EXPORT_SYMBOL_GPL(errno_to_blk_status);

int blk_status_to_errno(blk_status_t status)
{
	int idx = (__force int)status;

202
	if (WARN_ON_ONCE(idx >= ARRAY_SIZE(blk_errors)))
203 204 205 206 207
		return -EIO;
	return blk_errors[idx].errno;
}
EXPORT_SYMBOL_GPL(blk_status_to_errno);

208 209
static void print_req_error(struct request *req, blk_status_t status,
		const char *caller)
210 211 212
{
	int idx = (__force int)status;

213
	if (WARN_ON_ONCE(idx >= ARRAY_SIZE(blk_errors)))
214 215
		return;

216
	printk_ratelimited(KERN_ERR
217 218
		"%s: %s error, dev %s, sector %llu op 0x%x:(%s) flags 0x%x "
		"phys_seg %u prio class %u\n",
219
		caller, blk_errors[idx].name,
220 221 222 223 224
		req->rq_disk ? req->rq_disk->disk_name : "?",
		blk_rq_pos(req), req_op(req), blk_op_str(req_op(req)),
		req->cmd_flags & ~REQ_OP_MASK,
		req->nr_phys_segments,
		IOPRIO_PRIO_CLASS(req->ioprio));
225 226
}

N
NeilBrown 已提交
227
static void req_bio_endio(struct request *rq, struct bio *bio,
228
			  unsigned int nbytes, blk_status_t error)
L
Linus Torvalds 已提交
229
{
230
	if (error)
231
		bio->bi_status = error;
232

233
	if (unlikely(rq->rq_flags & RQF_QUIET))
234
		bio_set_flag(bio, BIO_QUIET);
235

236
	bio_advance(bio, nbytes);
237

T
Tejun Heo 已提交
238
	/* don't actually finish bio if it's part of flush sequence */
239
	if (bio->bi_iter.bi_size == 0 && !(rq->rq_flags & RQF_FLUSH_SEQ))
240
		bio_endio(bio);
L
Linus Torvalds 已提交
241 242 243 244
}

void blk_dump_rq_flags(struct request *rq, char *msg)
{
245 246
	printk(KERN_INFO "%s: dev %s: flags=%llx\n", msg,
		rq->rq_disk ? rq->rq_disk->disk_name : "?",
J
Jens Axboe 已提交
247
		(unsigned long long) rq->cmd_flags);
L
Linus Torvalds 已提交
248

249 250 251
	printk(KERN_INFO "  sector %llu, nr/cnr %u/%u\n",
	       (unsigned long long)blk_rq_pos(rq),
	       blk_rq_sectors(rq), blk_rq_cur_sectors(rq));
252 253
	printk(KERN_INFO "  bio %p, biotail %p, len %u\n",
	       rq->bio, rq->biotail, blk_rq_bytes(rq));
L
Linus Torvalds 已提交
254 255 256 257 258 259 260 261 262 263 264 265
}
EXPORT_SYMBOL(blk_dump_rq_flags);

/**
 * blk_sync_queue - cancel any pending callbacks on a queue
 * @q: the queue
 *
 * Description:
 *     The block layer may perform asynchronous callback activity
 *     on a queue, such as calling the unplug function after a timeout.
 *     A block device may call blk_sync_queue to ensure that any
 *     such activity is cancelled, thus allowing it to release resources
266
 *     that the callbacks might use. The caller must already have made sure
L
Linus Torvalds 已提交
267 268 269
 *     that its ->make_request_fn will not re-add plugging prior to calling
 *     this function.
 *
270
 *     This function does not cancel any asynchronous activity arising
271
 *     out of elevator or throttling code. That would require elevator_exit()
272
 *     and blkcg_exit_queue() to be called with queue lock initialized.
273
 *
L
Linus Torvalds 已提交
274 275 276
 */
void blk_sync_queue(struct request_queue *q)
{
277
	del_timer_sync(&q->timeout);
278
	cancel_work_sync(&q->timeout_work);
L
Linus Torvalds 已提交
279 280 281
}
EXPORT_SYMBOL(blk_sync_queue);

282
/**
283
 * blk_set_pm_only - increment pm_only counter
284 285
 * @q: request queue pointer
 */
286
void blk_set_pm_only(struct request_queue *q)
287
{
288
	atomic_inc(&q->pm_only);
289
}
290
EXPORT_SYMBOL_GPL(blk_set_pm_only);
291

292
void blk_clear_pm_only(struct request_queue *q)
293
{
294 295 296 297 298 299
	int pm_only;

	pm_only = atomic_dec_return(&q->pm_only);
	WARN_ON_ONCE(pm_only < 0);
	if (pm_only == 0)
		wake_up_all(&q->mq_freeze_wq);
300
}
301
EXPORT_SYMBOL_GPL(blk_clear_pm_only);
302

303
void blk_put_queue(struct request_queue *q)
304 305 306
{
	kobject_put(&q->kobj);
}
J
Jens Axboe 已提交
307
EXPORT_SYMBOL(blk_put_queue);
308

309 310
void blk_set_queue_dying(struct request_queue *q)
{
311
	blk_queue_flag_set(QUEUE_FLAG_DYING, q);
312

313 314 315 316 317 318 319
	/*
	 * When queue DYING flag is set, we need to block new req
	 * entering queue, so we call blk_freeze_queue_start() to
	 * prevent I/O from crossing blk_queue_enter().
	 */
	blk_freeze_queue_start(q);

J
Jens Axboe 已提交
320
	if (queue_is_mq(q))
321
		blk_mq_wake_waiters(q);
322 323 324

	/* Make blk_queue_enter() reexamine the DYING flag. */
	wake_up_all(&q->mq_freeze_wq);
325 326 327
}
EXPORT_SYMBOL_GPL(blk_set_queue_dying);

328 329 330 331
/**
 * blk_cleanup_queue - shutdown a request queue
 * @q: request queue to shutdown
 *
332 333
 * Mark @q DYING, drain all pending requests, mark @q DEAD, destroy and
 * put it.  All future requests will be failed immediately with -ENODEV.
334
 */
335
void blk_cleanup_queue(struct request_queue *q)
336
{
B
Bart Van Assche 已提交
337
	/* mark @q DYING, no new request or merges will be allowed afterwards */
338
	mutex_lock(&q->sysfs_lock);
339
	blk_set_queue_dying(q);
340

341 342 343
	blk_queue_flag_set(QUEUE_FLAG_NOMERGES, q);
	blk_queue_flag_set(QUEUE_FLAG_NOXMERGES, q);
	blk_queue_flag_set(QUEUE_FLAG_DYING, q);
344 345
	mutex_unlock(&q->sysfs_lock);

346 347
	/*
	 * Drain all requests queued before DYING marking. Set DEAD flag to
348 349
	 * prevent that blk_mq_run_hw_queues() accesses the hardware queues
	 * after draining finished.
350
	 */
351
	blk_freeze_queue(q);
352 353 354

	rq_qos_exit(q);

355
	blk_queue_flag_set(QUEUE_FLAG_DEAD, q);
356

357 358 359
	/* for synchronous bio-based driver finish in-flight integrity i/o */
	blk_flush_integrity();

360
	/* @q won't process any more request, flush async actions */
361
	del_timer_sync(&q->backing_dev_info->laptop_mode_wb_timer);
362 363
	blk_sync_queue(q);

J
Jens Axboe 已提交
364
	if (queue_is_mq(q))
365
		blk_mq_exit_queue(q);
J
Jens Axboe 已提交
366

367 368 369 370 371 372 373 374 375 376 377 378 379
	/*
	 * In theory, request pool of sched_tags belongs to request queue.
	 * However, the current implementation requires tag_set for freeing
	 * requests, so free the pool now.
	 *
	 * Queue has become frozen, there can't be any in-queue requests, so
	 * it is safe to free requests now.
	 */
	mutex_lock(&q->sysfs_lock);
	if (q->elevator)
		blk_mq_sched_free_requests(q);
	mutex_unlock(&q->sysfs_lock);

380
	percpu_ref_exit(&q->q_usage_counter);
B
Bart Van Assche 已提交
381

382
	/* @q is and will stay empty, shutdown and put */
383 384
	blk_put_queue(q);
}
L
Linus Torvalds 已提交
385 386
EXPORT_SYMBOL(blk_cleanup_queue);

387
struct request_queue *blk_alloc_queue(gfp_t gfp_mask)
L
Linus Torvalds 已提交
388
{
389
	return blk_alloc_queue_node(gfp_mask, NUMA_NO_NODE);
390 391
}
EXPORT_SYMBOL(blk_alloc_queue);
L
Linus Torvalds 已提交
392

393 394 395 396 397
/**
 * blk_queue_enter() - try to increase q->q_usage_counter
 * @q: request queue pointer
 * @flags: BLK_MQ_REQ_NOWAIT and/or BLK_MQ_REQ_PREEMPT
 */
398
int blk_queue_enter(struct request_queue *q, blk_mq_req_flags_t flags)
399
{
400
	const bool pm = flags & BLK_MQ_REQ_PREEMPT;
401

402
	while (true) {
403
		bool success = false;
404

405
		rcu_read_lock();
406 407
		if (percpu_ref_tryget_live(&q->q_usage_counter)) {
			/*
408 409 410
			 * The code that increments the pm_only counter is
			 * responsible for ensuring that that counter is
			 * globally visible before the queue is unfrozen.
411
			 */
412
			if (pm || !blk_queue_pm_only(q)) {
413 414 415 416 417
				success = true;
			} else {
				percpu_ref_put(&q->q_usage_counter);
			}
		}
418
		rcu_read_unlock();
419 420

		if (success)
421 422
			return 0;

423
		if (flags & BLK_MQ_REQ_NOWAIT)
424 425
			return -EBUSY;

426
		/*
427
		 * read pair of barrier in blk_freeze_queue_start(),
428
		 * we need to order reading __PERCPU_REF_DEAD flag of
429 430 431
		 * .q_usage_counter and reading .mq_freeze_depth or
		 * queue dying flag, otherwise the following wait may
		 * never return if the two reads are reordered.
432 433 434
		 */
		smp_rmb();

435
		wait_event(q->mq_freeze_wq,
436
			   (!q->mq_freeze_depth &&
437 438
			    (pm || (blk_pm_request_resume(q),
				    !blk_queue_pm_only(q)))) ||
439
			   blk_queue_dying(q));
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457
		if (blk_queue_dying(q))
			return -ENODEV;
	}
}

void blk_queue_exit(struct request_queue *q)
{
	percpu_ref_put(&q->q_usage_counter);
}

static void blk_queue_usage_counter_release(struct percpu_ref *ref)
{
	struct request_queue *q =
		container_of(ref, struct request_queue, q_usage_counter);

	wake_up_all(&q->mq_freeze_wq);
}

458
static void blk_rq_timed_out_timer(struct timer_list *t)
459
{
460
	struct request_queue *q = from_timer(q, t, timeout);
461 462 463 464

	kblockd_schedule_work(&q->timeout_work);
}

465 466 467 468
static void blk_timeout_work(struct work_struct *work)
{
}

469 470 471 472 473
/**
 * blk_alloc_queue_node - allocate a request queue
 * @gfp_mask: memory allocation flags
 * @node_id: NUMA node to allocate memory from
 */
474
struct request_queue *blk_alloc_queue_node(gfp_t gfp_mask, int node_id)
475
{
476
	struct request_queue *q;
477
	int ret;
478

479
	q = kmem_cache_alloc_node(blk_requestq_cachep,
480
				gfp_mask | __GFP_ZERO, node_id);
L
Linus Torvalds 已提交
481 482 483
	if (!q)
		return NULL;

484 485 486
	INIT_LIST_HEAD(&q->queue_head);
	q->last_merge = NULL;

487
	q->id = ida_simple_get(&blk_queue_ida, 0, 0, gfp_mask);
488
	if (q->id < 0)
489
		goto fail_q;
490

491 492
	ret = bioset_init(&q->bio_split, BIO_POOL_SIZE, 0, BIOSET_NEED_BVECS);
	if (ret)
493 494
		goto fail_id;

495 496 497 498
	q->backing_dev_info = bdi_alloc_node(gfp_mask, node_id);
	if (!q->backing_dev_info)
		goto fail_split;

499 500 501 502
	q->stats = blk_alloc_queue_stats();
	if (!q->stats)
		goto fail_stats;

503
	q->backing_dev_info->ra_pages = VM_READAHEAD_PAGES;
504 505
	q->backing_dev_info->capabilities = BDI_CAP_CGROUP_WRITEBACK;
	q->backing_dev_info->name = "block";
506
	q->node = node_id;
507

508 509 510
	timer_setup(&q->backing_dev_info->laptop_mode_wb_timer,
		    laptop_mode_timer_fn, 0);
	timer_setup(&q->timeout, blk_rq_timed_out_timer, 0);
511
	INIT_WORK(&q->timeout_work, blk_timeout_work);
512
	INIT_LIST_HEAD(&q->icq_list);
513
#ifdef CONFIG_BLK_CGROUP
514
	INIT_LIST_HEAD(&q->blkg_list);
515
#endif
516

517
	kobject_init(&q->kobj, &blk_queue_ktype);
L
Linus Torvalds 已提交
518

519 520 521
#ifdef CONFIG_BLK_DEV_IO_TRACE
	mutex_init(&q->blk_trace_mutex);
#endif
522
	mutex_init(&q->sysfs_lock);
523
	spin_lock_init(&q->queue_lock);
524

525
	init_waitqueue_head(&q->mq_freeze_wq);
526
	mutex_init(&q->mq_freeze_lock);
527

528 529 530 531 532 533 534
	/*
	 * Init percpu_ref in atomic mode so that it's faster to shutdown.
	 * See blk_register_queue() for details.
	 */
	if (percpu_ref_init(&q->q_usage_counter,
				blk_queue_usage_counter_release,
				PERCPU_REF_INIT_ATOMIC, GFP_KERNEL))
535
		goto fail_bdi;
536

537 538 539
	if (blkcg_init_queue(q))
		goto fail_ref;

L
Linus Torvalds 已提交
540
	return q;
541

542 543
fail_ref:
	percpu_ref_exit(&q->q_usage_counter);
544
fail_bdi:
545 546
	blk_free_queue_stats(q->stats);
fail_stats:
547
	bdi_put(q->backing_dev_info);
548
fail_split:
549
	bioset_exit(&q->bio_split);
550 551 552 553 554
fail_id:
	ida_simple_remove(&blk_queue_ida, q->id);
fail_q:
	kmem_cache_free(blk_requestq_cachep, q);
	return NULL;
L
Linus Torvalds 已提交
555
}
556
EXPORT_SYMBOL(blk_alloc_queue_node);
L
Linus Torvalds 已提交
557

T
Tejun Heo 已提交
558
bool blk_get_queue(struct request_queue *q)
L
Linus Torvalds 已提交
559
{
B
Bart Van Assche 已提交
560
	if (likely(!blk_queue_dying(q))) {
T
Tejun Heo 已提交
561 562
		__blk_get_queue(q);
		return true;
L
Linus Torvalds 已提交
563 564
	}

T
Tejun Heo 已提交
565
	return false;
L
Linus Torvalds 已提交
566
}
J
Jens Axboe 已提交
567
EXPORT_SYMBOL(blk_get_queue);
L
Linus Torvalds 已提交
568

J
Jens Axboe 已提交
569 570 571 572 573
/**
 * blk_get_request - allocate a request
 * @q: request queue to allocate a request for
 * @op: operation (REQ_OP_*) and REQ_* flags, e.g. REQ_SYNC.
 * @flags: BLK_MQ_REQ_* flags, e.g. BLK_MQ_REQ_NOWAIT.
L
Linus Torvalds 已提交
574
 */
J
Jens Axboe 已提交
575 576
struct request *blk_get_request(struct request_queue *q, unsigned int op,
				blk_mq_req_flags_t flags)
L
Linus Torvalds 已提交
577
{
J
Jens Axboe 已提交
578
	struct request *req;
L
Linus Torvalds 已提交
579

J
Jens Axboe 已提交
580 581
	WARN_ON_ONCE(op & REQ_NOWAIT);
	WARN_ON_ONCE(flags & ~(BLK_MQ_REQ_NOWAIT | BLK_MQ_REQ_PREEMPT));
L
Linus Torvalds 已提交
582

J
Jens Axboe 已提交
583 584 585
	req = blk_mq_alloc_request(q, op, flags);
	if (!IS_ERR(req) && q->mq_ops->initialize_rq_fn)
		q->mq_ops->initialize_rq_fn(req);
L
Linus Torvalds 已提交
586

J
Jens Axboe 已提交
587
	return req;
L
Linus Torvalds 已提交
588
}
J
Jens Axboe 已提交
589
EXPORT_SYMBOL(blk_get_request);
L
Linus Torvalds 已提交
590 591 592

void blk_put_request(struct request *req)
{
J
Jens Axboe 已提交
593
	blk_mq_free_request(req);
L
Linus Torvalds 已提交
594 595 596
}
EXPORT_SYMBOL(blk_put_request);

597 598
bool bio_attempt_back_merge(struct request *req, struct bio *bio,
		unsigned int nr_segs)
599
{
J
Jens Axboe 已提交
600
	const int ff = bio->bi_opf & REQ_FAILFAST_MASK;
601

602
	if (!ll_back_merge_fn(req, bio, nr_segs))
603 604
		return false;

605
	trace_block_bio_backmerge(req->q, req, bio);
606 607 608 609 610 611

	if ((req->cmd_flags & REQ_FAILFAST_MASK) != ff)
		blk_rq_set_mixed_merge(req);

	req->biotail->bi_next = bio;
	req->biotail = bio;
612
	req->__data_len += bio->bi_iter.bi_size;
613

614
	blk_account_io_start(req, false);
615 616 617
	return true;
}

618 619
bool bio_attempt_front_merge(struct request *req, struct bio *bio,
		unsigned int nr_segs)
620
{
J
Jens Axboe 已提交
621
	const int ff = bio->bi_opf & REQ_FAILFAST_MASK;
622

623
	if (!ll_front_merge_fn(req, bio, nr_segs))
624 625
		return false;

626
	trace_block_bio_frontmerge(req->q, req, bio);
627 628 629 630 631 632 633

	if ((req->cmd_flags & REQ_FAILFAST_MASK) != ff)
		blk_rq_set_mixed_merge(req);

	bio->bi_next = req->bio;
	req->bio = bio;

634 635
	req->__sector = bio->bi_iter.bi_sector;
	req->__data_len += bio->bi_iter.bi_size;
636

637
	blk_account_io_start(req, false);
638 639 640
	return true;
}

641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663
bool bio_attempt_discard_merge(struct request_queue *q, struct request *req,
		struct bio *bio)
{
	unsigned short segments = blk_rq_nr_discard_segments(req);

	if (segments >= queue_max_discard_segments(q))
		goto no_merge;
	if (blk_rq_sectors(req) + bio_sectors(bio) >
	    blk_rq_get_max_sectors(req, blk_rq_pos(req)))
		goto no_merge;

	req->biotail->bi_next = bio;
	req->biotail = bio;
	req->__data_len += bio->bi_iter.bi_size;
	req->nr_phys_segments = segments + 1;

	blk_account_io_start(req, false);
	return true;
no_merge:
	req_set_nomerge(q, req);
	return false;
}

664
/**
665
 * blk_attempt_plug_merge - try to merge with %current's plugged list
666 667
 * @q: request_queue new bio is being queued at
 * @bio: new bio being queued
668
 * @nr_segs: number of segments in @bio
669 670 671
 * @same_queue_rq: pointer to &struct request that gets filled in when
 * another request associated with @q is found on the plug list
 * (optional, may be %NULL)
672 673 674 675 676
 *
 * Determine whether @bio being queued on @q can be merged with a request
 * on %current's plugged list.  Returns %true if merge was successful,
 * otherwise %false.
 *
677 678 679 680 681 682
 * Plugging coalesces IOs from the same issuer for the same purpose without
 * going through @q->queue_lock.  As such it's more of an issuing mechanism
 * than scheduling, and the request, while may have elvpriv data, is not
 * added on the elevator at this point.  In addition, we don't have
 * reliable access to the elevator outside queue lock.  Only check basic
 * merging parameters without querying the elevator.
683 684
 *
 * Caller must ensure !blk_queue_nomerges(q) beforehand.
685
 */
686
bool blk_attempt_plug_merge(struct request_queue *q, struct bio *bio,
687
		unsigned int nr_segs, struct request **same_queue_rq)
688 689 690
{
	struct blk_plug *plug;
	struct request *rq;
S
Shaohua Li 已提交
691
	struct list_head *plug_list;
692

693
	plug = blk_mq_plug(q, bio);
694
	if (!plug)
695
		return false;
696

J
Jens Axboe 已提交
697
	plug_list = &plug->mq_list;
S
Shaohua Li 已提交
698 699

	list_for_each_entry_reverse(rq, plug_list, queuelist) {
700
		bool merged = false;
701

702
		if (rq->q == q && same_queue_rq) {
703 704 705 706 707
			/*
			 * Only blk-mq multiple hardware queues case checks the
			 * rq in the same queue, there should be only one such
			 * rq in a queue
			 **/
708
			*same_queue_rq = rq;
709
		}
710

711
		if (rq->q != q || !blk_rq_merge_ok(rq, bio))
712 713
			continue;

714 715
		switch (blk_try_merge(rq, bio)) {
		case ELEVATOR_BACK_MERGE:
716
			merged = bio_attempt_back_merge(rq, bio, nr_segs);
717 718
			break;
		case ELEVATOR_FRONT_MERGE:
719
			merged = bio_attempt_front_merge(rq, bio, nr_segs);
720
			break;
721 722 723
		case ELEVATOR_DISCARD_MERGE:
			merged = bio_attempt_discard_merge(q, rq, bio);
			break;
724 725
		default:
			break;
726
		}
727 728 729

		if (merged)
			return true;
730
	}
731 732

	return false;
733 734
}

735
static void handle_bad_sector(struct bio *bio, sector_t maxsector)
L
Linus Torvalds 已提交
736 737 738 739
{
	char b[BDEVNAME_SIZE];

	printk(KERN_INFO "attempt to access beyond end of device\n");
740
	printk(KERN_INFO "%s: rw=%d, want=%Lu, limit=%Lu\n",
741
			bio_devname(bio, b), bio->bi_opf,
K
Kent Overstreet 已提交
742
			(unsigned long long)bio_end_sector(bio),
743
			(long long)maxsector);
L
Linus Torvalds 已提交
744 745
}

746 747 748 749 750 751 752 753 754 755
#ifdef CONFIG_FAIL_MAKE_REQUEST

static DECLARE_FAULT_ATTR(fail_make_request);

static int __init setup_fail_make_request(char *str)
{
	return setup_fault_attr(&fail_make_request, str);
}
__setup("fail_make_request=", setup_fail_make_request);

756
static bool should_fail_request(struct hd_struct *part, unsigned int bytes)
757
{
758
	return part->make_it_fail && should_fail(&fail_make_request, bytes);
759 760 761 762
}

static int __init fail_make_request_debugfs(void)
{
763 764 765
	struct dentry *dir = fault_create_debugfs_attr("fail_make_request",
						NULL, &fail_make_request);

766
	return PTR_ERR_OR_ZERO(dir);
767 768 769 770 771 772
}

late_initcall(fail_make_request_debugfs);

#else /* CONFIG_FAIL_MAKE_REQUEST */

773 774
static inline bool should_fail_request(struct hd_struct *part,
					unsigned int bytes)
775
{
776
	return false;
777 778 779 780
}

#endif /* CONFIG_FAIL_MAKE_REQUEST */

781 782
static inline bool bio_check_ro(struct bio *bio, struct hd_struct *part)
{
783 784
	const int op = bio_op(bio);

785
	if (part->policy && op_is_write(op)) {
786 787
		char b[BDEVNAME_SIZE];

788 789 790
		if (op_is_flush(bio->bi_opf) && !bio_sectors(bio))
			return false;

791
		WARN_ONCE(1,
792 793 794
		       "generic_make_request: Trying to write "
			"to read-only block-device %s (partno %d)\n",
			bio_devname(bio, b), part->partno);
795 796
		/* Older lvm-tools actually trigger this */
		return false;
797 798 799 800 801
	}

	return false;
}

802 803 804 805 806 807 808 809
static noinline int should_fail_bio(struct bio *bio)
{
	if (should_fail_request(&bio->bi_disk->part0, bio->bi_iter.bi_size))
		return -EIO;
	return 0;
}
ALLOW_ERROR_INJECTION(should_fail_bio, ERRNO);

810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827
/*
 * Check whether this bio extends beyond the end of the device or partition.
 * This may well happen - the kernel calls bread() without checking the size of
 * the device, e.g., when mounting a file system.
 */
static inline int bio_check_eod(struct bio *bio, sector_t maxsector)
{
	unsigned int nr_sectors = bio_sectors(bio);

	if (nr_sectors && maxsector &&
	    (nr_sectors > maxsector ||
	     bio->bi_iter.bi_sector > maxsector - nr_sectors)) {
		handle_bad_sector(bio, maxsector);
		return -EIO;
	}
	return 0;
}

828 829 830 831 832 833
/*
 * Remap block n of partition p to block n+start(p) of the disk.
 */
static inline int blk_partition_remap(struct bio *bio)
{
	struct hd_struct *p;
834
	int ret = -EIO;
835

836 837
	rcu_read_lock();
	p = __disk_get_part(bio->bi_disk, bio->bi_partno);
838 839 840 841 842
	if (unlikely(!p))
		goto out;
	if (unlikely(should_fail_request(p, bio->bi_iter.bi_size)))
		goto out;
	if (unlikely(bio_check_ro(bio, p)))
843 844
		goto out;

845 846 847 848
	/*
	 * Zone reset does not include bi_size so bio_sectors() is always 0.
	 * Include a test for the reset op code and perform the remap if needed.
	 */
849 850 851 852 853 854 855
	if (bio_sectors(bio) || bio_op(bio) == REQ_OP_ZONE_RESET) {
		if (bio_check_eod(bio, part_nr_sects_read(p)))
			goto out;
		bio->bi_iter.bi_sector += p->start_sect;
		trace_block_bio_remap(bio->bi_disk->queue, bio, part_devt(p),
				      bio->bi_iter.bi_sector - p->start_sect);
	}
856
	bio->bi_partno = 0;
857
	ret = 0;
858 859
out:
	rcu_read_unlock();
860 861 862
	return ret;
}

863 864
static noinline_for_stack bool
generic_make_request_checks(struct bio *bio)
L
Linus Torvalds 已提交
865
{
866
	struct request_queue *q;
867
	int nr_sectors = bio_sectors(bio);
868
	blk_status_t status = BLK_STS_IOERR;
869
	char b[BDEVNAME_SIZE];
L
Linus Torvalds 已提交
870 871 872

	might_sleep();

873
	q = bio->bi_disk->queue;
874 875 876 877
	if (unlikely(!q)) {
		printk(KERN_ERR
		       "generic_make_request: Trying to access "
			"nonexistent block-device %s (%Lu)\n",
878
			bio_devname(bio, b), (long long)bio->bi_iter.bi_sector);
879 880
		goto end_io;
	}
881

882 883 884 885
	/*
	 * For a REQ_NOWAIT based request, return -EOPNOTSUPP
	 * if queue is not a request based queue.
	 */
J
Jens Axboe 已提交
886
	if ((bio->bi_opf & REQ_NOWAIT) && !queue_is_mq(q))
887 888
		goto not_supported;

889
	if (should_fail_bio(bio))
890
		goto end_io;
891

892 893
	if (bio->bi_partno) {
		if (unlikely(blk_partition_remap(bio)))
894 895
			goto end_io;
	} else {
896 897 898
		if (unlikely(bio_check_ro(bio, &bio->bi_disk->part0)))
			goto end_io;
		if (unlikely(bio_check_eod(bio, get_capacity(bio->bi_disk))))
899 900
			goto end_io;
	}
901

902 903 904 905 906
	/*
	 * Filter flush bio's early so that make_request based
	 * drivers without flush support don't have to worry
	 * about them.
	 */
907
	if (op_is_flush(bio->bi_opf) &&
J
Jens Axboe 已提交
908
	    !test_bit(QUEUE_FLAG_WC, &q->queue_flags)) {
J
Jens Axboe 已提交
909
		bio->bi_opf &= ~(REQ_PREFLUSH | REQ_FUA);
910
		if (!nr_sectors) {
911
			status = BLK_STS_OK;
912 913
			goto end_io;
		}
914
	}
915

916 917 918
	if (!test_bit(QUEUE_FLAG_POLL, &q->queue_flags))
		bio->bi_opf &= ~REQ_HIPRI;

919 920 921 922 923 924 925 926 927 928
	switch (bio_op(bio)) {
	case REQ_OP_DISCARD:
		if (!blk_queue_discard(q))
			goto not_supported;
		break;
	case REQ_OP_SECURE_ERASE:
		if (!blk_queue_secure_erase(q))
			goto not_supported;
		break;
	case REQ_OP_WRITE_SAME:
929
		if (!q->limits.max_write_same_sectors)
930
			goto not_supported;
931
		break;
932
	case REQ_OP_ZONE_RESET:
933
		if (!blk_queue_is_zoned(q))
934
			goto not_supported;
935
		break;
936 937 938 939
	case REQ_OP_ZONE_RESET_ALL:
		if (!blk_queue_is_zoned(q) || !blk_queue_zone_resetall(q))
			goto not_supported;
		break;
940
	case REQ_OP_WRITE_ZEROES:
941
		if (!q->limits.max_write_zeroes_sectors)
942 943
			goto not_supported;
		break;
944 945
	default:
		break;
946
	}
947

T
Tejun Heo 已提交
948 949 950 951 952 953 954 955
	/*
	 * Various block parts want %current->io_context and lazy ioc
	 * allocation ends up trading a lot of pain for a small amount of
	 * memory.  Just allocate it upfront.  This may fail and block
	 * layer knows how to live with it.
	 */
	create_io_context(GFP_ATOMIC, q->node);

956 957
	if (!blkcg_bio_issue_check(q, bio))
		return false;
958

N
NeilBrown 已提交
959 960 961 962 963 964 965
	if (!bio_flagged(bio, BIO_TRACE_COMPLETION)) {
		trace_block_bio_queue(q, bio);
		/* Now that enqueuing has been traced, we need to trace
		 * completion as well.
		 */
		bio_set_flag(bio, BIO_TRACE_COMPLETION);
	}
966
	return true;
967

968
not_supported:
969
	status = BLK_STS_NOTSUPP;
970
end_io:
971
	bio->bi_status = status;
972
	bio_endio(bio);
973
	return false;
L
Linus Torvalds 已提交
974 975
}

976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998
/**
 * generic_make_request - hand a buffer to its device driver for I/O
 * @bio:  The bio describing the location in memory and on the device.
 *
 * generic_make_request() is used to make I/O requests of block
 * devices. It is passed a &struct bio, which describes the I/O that needs
 * to be done.
 *
 * generic_make_request() does not return any status.  The
 * success/failure status of the request, along with notification of
 * completion, is delivered asynchronously through the bio->bi_end_io
 * function described (one day) else where.
 *
 * The caller of generic_make_request must make sure that bi_io_vec
 * are set to describe the memory buffer, and that bi_dev and bi_sector are
 * set to describe the device address, and the
 * bi_end_io and optionally bi_private are set to describe how
 * completion notification should be signaled.
 *
 * generic_make_request and the drivers it calls may use bi_next if this
 * bio happens to be merged with someone else, and may resubmit the bio to
 * a lower device by calling into generic_make_request recursively, which
 * means the bio should NOT be touched after the call to ->make_request_fn.
999
 */
1000
blk_qc_t generic_make_request(struct bio *bio)
1001
{
1002 1003 1004 1005 1006 1007 1008 1009
	/*
	 * bio_list_on_stack[0] contains bios submitted by the current
	 * make_request_fn.
	 * bio_list_on_stack[1] contains bios that were submitted before
	 * the current make_request_fn, but that haven't been processed
	 * yet.
	 */
	struct bio_list bio_list_on_stack[2];
1010
	blk_qc_t ret = BLK_QC_T_NONE;
1011

1012
	if (!generic_make_request_checks(bio))
1013
		goto out;
1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024

	/*
	 * We only want one ->make_request_fn to be active at a time, else
	 * stack usage with stacked devices could be a problem.  So use
	 * current->bio_list to keep a list of requests submited by a
	 * make_request_fn function.  current->bio_list is also used as a
	 * flag to say if generic_make_request is currently active in this
	 * task or not.  If it is NULL, then no make_request is active.  If
	 * it is non-NULL, then a make_request is active, and new requests
	 * should be added at the tail
	 */
1025
	if (current->bio_list) {
1026
		bio_list_add(&current->bio_list[0], bio);
1027
		goto out;
1028
	}
1029

1030 1031 1032 1033 1034
	/* following loop may be a bit non-obvious, and so deserves some
	 * explanation.
	 * Before entering the loop, bio->bi_next is NULL (as all callers
	 * ensure that) so we have a list with a single bio.
	 * We pretend that we have just taken it off a longer list, so
1035 1036
	 * we assign bio_list to a pointer to the bio_list_on_stack,
	 * thus initialising the bio_list of new bios to be
1037
	 * added.  ->make_request() may indeed add some more bios
1038 1039 1040
	 * through a recursive call to generic_make_request.  If it
	 * did, we find a non-NULL value in bio_list and re-enter the loop
	 * from the top.  In this case we really did just take the bio
1041
	 * of the top of the list (no pretending) and so remove it from
1042
	 * bio_list, and call into ->make_request() again.
1043 1044
	 */
	BUG_ON(bio->bi_next);
1045 1046
	bio_list_init(&bio_list_on_stack[0]);
	current->bio_list = bio_list_on_stack;
1047
	do {
1048 1049 1050
		struct request_queue *q = bio->bi_disk->queue;
		blk_mq_req_flags_t flags = bio->bi_opf & REQ_NOWAIT ?
			BLK_MQ_REQ_NOWAIT : 0;
1051

1052
		if (likely(blk_queue_enter(q, flags) == 0)) {
1053 1054 1055
			struct bio_list lower, same;

			/* Create a fresh bio_list for all subordinate requests */
1056 1057
			bio_list_on_stack[1] = bio_list_on_stack[0];
			bio_list_init(&bio_list_on_stack[0]);
1058
			ret = q->make_request_fn(q, bio);
1059

1060 1061
			blk_queue_exit(q);

1062 1063 1064 1065 1066
			/* sort new bios into those for a lower level
			 * and those for the same level
			 */
			bio_list_init(&lower);
			bio_list_init(&same);
1067
			while ((bio = bio_list_pop(&bio_list_on_stack[0])) != NULL)
1068
				if (q == bio->bi_disk->queue)
1069 1070 1071 1072
					bio_list_add(&same, bio);
				else
					bio_list_add(&lower, bio);
			/* now assemble so we handle the lowest level first */
1073 1074 1075
			bio_list_merge(&bio_list_on_stack[0], &lower);
			bio_list_merge(&bio_list_on_stack[0], &same);
			bio_list_merge(&bio_list_on_stack[0], &bio_list_on_stack[1]);
1076
		} else {
1077 1078 1079 1080 1081
			if (unlikely(!blk_queue_dying(q) &&
					(bio->bi_opf & REQ_NOWAIT)))
				bio_wouldblock_error(bio);
			else
				bio_io_error(bio);
1082
		}
1083
		bio = bio_list_pop(&bio_list_on_stack[0]);
1084
	} while (bio);
1085
	current->bio_list = NULL; /* deactivate */
1086 1087 1088

out:
	return ret;
1089
}
L
Linus Torvalds 已提交
1090 1091
EXPORT_SYMBOL(generic_make_request);

1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
/**
 * direct_make_request - hand a buffer directly to its device driver for I/O
 * @bio:  The bio describing the location in memory and on the device.
 *
 * This function behaves like generic_make_request(), but does not protect
 * against recursion.  Must only be used if the called driver is known
 * to not call generic_make_request (or direct_make_request) again from
 * its make_request function.  (Calling direct_make_request again from
 * a workqueue is perfectly fine as that doesn't recurse).
 */
blk_qc_t direct_make_request(struct bio *bio)
{
	struct request_queue *q = bio->bi_disk->queue;
	bool nowait = bio->bi_opf & REQ_NOWAIT;
	blk_qc_t ret;

	if (!generic_make_request_checks(bio))
		return BLK_QC_T_NONE;

1111
	if (unlikely(blk_queue_enter(q, nowait ? BLK_MQ_REQ_NOWAIT : 0))) {
1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125
		if (nowait && !blk_queue_dying(q))
			bio->bi_status = BLK_STS_AGAIN;
		else
			bio->bi_status = BLK_STS_IOERR;
		bio_endio(bio);
		return BLK_QC_T_NONE;
	}

	ret = q->make_request_fn(q, bio);
	blk_queue_exit(q);
	return ret;
}
EXPORT_SYMBOL_GPL(direct_make_request);

L
Linus Torvalds 已提交
1126
/**
1127
 * submit_bio - submit a bio to the block device layer for I/O
L
Linus Torvalds 已提交
1128 1129 1130 1131
 * @bio: The &struct bio which describes the I/O
 *
 * submit_bio() is very similar in purpose to generic_make_request(), and
 * uses that function to do most of the work. Both are fairly rough
1132
 * interfaces; @bio must be presetup and ready for I/O.
L
Linus Torvalds 已提交
1133 1134
 *
 */
1135
blk_qc_t submit_bio(struct bio *bio)
L
Linus Torvalds 已提交
1136
{
T
Tejun Heo 已提交
1137 1138 1139
	if (blkcg_punt_bio_submit(bio))
		return BLK_QC_T_NONE;

1140 1141 1142 1143
	/*
	 * If it's a regular read/write or a barrier with data attached,
	 * go through the normal accounting stuff before submission.
	 */
1144
	if (bio_has_data(bio)) {
1145 1146
		unsigned int count;

1147
		if (unlikely(bio_op(bio) == REQ_OP_WRITE_SAME))
1148
			count = queue_logical_block_size(bio->bi_disk->queue) >> 9;
1149 1150 1151
		else
			count = bio_sectors(bio);

1152
		if (op_is_write(bio_op(bio))) {
1153 1154
			count_vm_events(PGPGOUT, count);
		} else {
1155
			task_io_account_read(bio->bi_iter.bi_size);
1156 1157 1158 1159 1160
			count_vm_events(PGPGIN, count);
		}

		if (unlikely(block_dump)) {
			char b[BDEVNAME_SIZE];
1161
			printk(KERN_DEBUG "%s(%d): %s block %Lu on %s (%u sectors)\n",
1162
			current->comm, task_pid_nr(current),
1163
				op_is_write(bio_op(bio)) ? "WRITE" : "READ",
1164
				(unsigned long long)bio->bi_iter.bi_sector,
1165
				bio_devname(bio, b), count);
1166
		}
L
Linus Torvalds 已提交
1167 1168
	}

1169
	return generic_make_request(bio);
L
Linus Torvalds 已提交
1170 1171 1172
}
EXPORT_SYMBOL(submit_bio);

1173
/**
1174 1175
 * blk_cloned_rq_check_limits - Helper function to check a cloned request
 *                              for new the queue limits
1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186
 * @q:  the queue
 * @rq: the request being checked
 *
 * Description:
 *    @rq may have been made based on weaker limitations of upper-level queues
 *    in request stacking drivers, and it may violate the limitation of @q.
 *    Since the block layer and the underlying device driver trust @rq
 *    after it is inserted to @q, it should be checked against @q before
 *    the insertion using this generic function.
 *
 *    Request stacking drivers like request-based dm may change the queue
1187 1188
 *    limits when retrying requests on other queues. Those requests need
 *    to be checked against the new queue limits again during dispatch.
1189
 */
1190 1191
static int blk_cloned_rq_check_limits(struct request_queue *q,
				      struct request *rq)
1192
{
1193
	if (blk_rq_sectors(rq) > blk_queue_get_max_sectors(q, req_op(rq))) {
1194 1195 1196
		printk(KERN_ERR "%s: over max size limit. (%u > %u)\n",
			__func__, blk_rq_sectors(rq),
			blk_queue_get_max_sectors(q, req_op(rq)));
1197 1198 1199 1200 1201 1202 1203 1204 1205
		return -EIO;
	}

	/*
	 * queue's settings related to segment counting like q->bounce_pfn
	 * may differ from that of other stacking queues.
	 * Recalculate it to check the request correctly on this queue's
	 * limitation.
	 */
1206
	rq->nr_phys_segments = blk_recalc_rq_segments(rq);
1207
	if (rq->nr_phys_segments > queue_max_segments(q)) {
1208 1209
		printk(KERN_ERR "%s: over max segments limit. (%hu > %hu)\n",
			__func__, rq->nr_phys_segments, queue_max_segments(q));
1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220
		return -EIO;
	}

	return 0;
}

/**
 * blk_insert_cloned_request - Helper for stacking drivers to submit a request
 * @q:  the queue to submit the request
 * @rq: the request being queued
 */
1221
blk_status_t blk_insert_cloned_request(struct request_queue *q, struct request *rq)
1222
{
1223
	if (blk_cloned_rq_check_limits(q, rq))
1224
		return BLK_STS_IOERR;
1225

1226 1227
	if (rq->rq_disk &&
	    should_fail_request(&rq->rq_disk->part0, blk_rq_bytes(rq)))
1228
		return BLK_STS_IOERR;
1229

J
Jens Axboe 已提交
1230 1231
	if (blk_queue_io_stat(q))
		blk_account_io_start(rq, true);
1232 1233

	/*
J
Jens Axboe 已提交
1234 1235 1236
	 * Since we have a scheduler attached on the top device,
	 * bypass a potential scheduler on the bottom device for
	 * insert.
1237
	 */
1238
	return blk_mq_request_issue_directly(rq, true);
1239 1240 1241
}
EXPORT_SYMBOL_GPL(blk_insert_cloned_request);

1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
/**
 * blk_rq_err_bytes - determine number of bytes till the next failure boundary
 * @rq: request to examine
 *
 * Description:
 *     A request could be merge of IOs which require different failure
 *     handling.  This function determines the number of bytes which
 *     can be failed from the beginning of the request without
 *     crossing into area which need to be retried further.
 *
 * Return:
 *     The number of bytes to fail.
 */
unsigned int blk_rq_err_bytes(const struct request *rq)
{
	unsigned int ff = rq->cmd_flags & REQ_FAILFAST_MASK;
	unsigned int bytes = 0;
	struct bio *bio;

1261
	if (!(rq->rq_flags & RQF_MIXED_MERGE))
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
		return blk_rq_bytes(rq);

	/*
	 * Currently the only 'mixing' which can happen is between
	 * different fastfail types.  We can safely fail portions
	 * which have all the failfast bits that the first one has -
	 * the ones which are at least as eager to fail as the first
	 * one.
	 */
	for (bio = rq->bio; bio; bio = bio->bi_next) {
J
Jens Axboe 已提交
1272
		if ((bio->bi_opf & ff) != ff)
1273
			break;
1274
		bytes += bio->bi_iter.bi_size;
1275 1276 1277 1278 1279 1280 1281 1282
	}

	/* this could lead to infinite loop */
	BUG_ON(blk_rq_bytes(rq) && !bytes);
	return bytes;
}
EXPORT_SYMBOL_GPL(blk_rq_err_bytes);

1283
void blk_account_io_completion(struct request *req, unsigned int bytes)
1284
{
1285
	if (blk_do_io_stat(req)) {
1286
		const int sgrp = op_stat_group(req_op(req));
1287 1288
		struct hd_struct *part;

1289
		part_stat_lock();
1290
		part = req->part;
1291
		part_stat_add(part, sectors[sgrp], bytes >> 9);
1292 1293 1294 1295
		part_stat_unlock();
	}
}

1296
void blk_account_io_done(struct request *req, u64 now)
1297 1298
{
	/*
1299 1300 1301
	 * Account IO completion.  flush_rq isn't accounted as a
	 * normal IO on queueing nor completion.  Accounting the
	 * containing request is enough.
1302
	 */
1303
	if (blk_do_io_stat(req) && !(req->rq_flags & RQF_FLUSH_SEQ)) {
1304
		const int sgrp = op_stat_group(req_op(req));
1305 1306
		struct hd_struct *part;

1307
		part_stat_lock();
1308
		part = req->part;
1309

1310
		update_io_ticks(part, jiffies);
1311 1312
		part_stat_inc(part, ios[sgrp]);
		part_stat_add(part, nsecs[sgrp], now - req->start_time_ns);
1313
		part_stat_add(part, time_in_queue, nsecs_to_jiffies64(now - req->start_time_ns));
1314
		part_dec_in_flight(req->q, part, rq_data_dir(req));
1315

1316
		hd_struct_put(part);
1317 1318 1319 1320
		part_stat_unlock();
	}
}

1321 1322 1323 1324 1325 1326 1327 1328
void blk_account_io_start(struct request *rq, bool new_io)
{
	struct hd_struct *part;
	int rw = rq_data_dir(rq);

	if (!blk_do_io_stat(rq))
		return;

1329
	part_stat_lock();
1330 1331 1332

	if (!new_io) {
		part = rq->part;
1333
		part_stat_inc(part, merges[rw]);
1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347
	} else {
		part = disk_map_sector_rcu(rq->rq_disk, blk_rq_pos(rq));
		if (!hd_struct_try_get(part)) {
			/*
			 * The partition is already being removed,
			 * the request will be accounted on the disk only
			 *
			 * We take a reference on disk->part0 although that
			 * partition will never be deleted, so we can treat
			 * it as any other partition.
			 */
			part = &rq->rq_disk->part0;
			hd_struct_get(part);
		}
1348
		part_inc_in_flight(rq->q, part, rw);
1349 1350 1351
		rq->part = part;
	}

1352 1353
	update_io_ticks(part, jiffies);

1354 1355 1356
	part_stat_unlock();
}

1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377
/*
 * Steal bios from a request and add them to a bio list.
 * The request must not have been partially completed before.
 */
void blk_steal_bios(struct bio_list *list, struct request *rq)
{
	if (rq->bio) {
		if (list->tail)
			list->tail->bi_next = rq->bio;
		else
			list->head = rq->bio;
		list->tail = rq->biotail;

		rq->bio = NULL;
		rq->biotail = NULL;
	}

	rq->__data_len = 0;
}
EXPORT_SYMBOL_GPL(blk_steal_bios);

1378
/**
1379
 * blk_update_request - Special helper function for request stacking drivers
1380
 * @req:      the request being processed
1381
 * @error:    block status code
1382
 * @nr_bytes: number of bytes to complete @req
1383 1384
 *
 * Description:
1385 1386 1387
 *     Ends I/O on a number of bytes attached to @req, but doesn't complete
 *     the request structure even if @req doesn't have leftover.
 *     If @req has leftover, sets it up for the next range of segments.
1388 1389 1390
 *
 *     This special helper function is only for request stacking drivers
 *     (e.g. request-based dm) so that they can handle partial completion.
1391
 *     Actual device drivers should use blk_mq_end_request instead.
1392 1393 1394
 *
 *     Passing the result of blk_rq_bytes() as @nr_bytes guarantees
 *     %false return from this function.
1395
 *
1396 1397 1398 1399
 * Note:
 *	The RQF_SPECIAL_PAYLOAD flag is ignored on purpose in both
 *	blk_rq_bytes() and in blk_update_request().
 *
1400
 * Return:
1401 1402
 *     %false - this request doesn't have any more data
 *     %true  - this request has more data
1403
 **/
1404 1405
bool blk_update_request(struct request *req, blk_status_t error,
		unsigned int nr_bytes)
L
Linus Torvalds 已提交
1406
{
1407
	int total_bytes;
L
Linus Torvalds 已提交
1408

1409
	trace_block_rq_complete(req, blk_status_to_errno(error), nr_bytes);
1410

1411 1412 1413
	if (!req->bio)
		return false;

1414 1415
	if (unlikely(error && !blk_rq_is_passthrough(req) &&
		     !(req->rq_flags & RQF_QUIET)))
1416
		print_req_error(req, error, __func__);
L
Linus Torvalds 已提交
1417

1418
	blk_account_io_completion(req, nr_bytes);
1419

1420 1421 1422
	total_bytes = 0;
	while (req->bio) {
		struct bio *bio = req->bio;
1423
		unsigned bio_bytes = min(bio->bi_iter.bi_size, nr_bytes);
L
Linus Torvalds 已提交
1424

1425
		if (bio_bytes == bio->bi_iter.bi_size)
L
Linus Torvalds 已提交
1426 1427
			req->bio = bio->bi_next;

N
NeilBrown 已提交
1428 1429
		/* Completion has already been traced */
		bio_clear_flag(bio, BIO_TRACE_COMPLETION);
1430
		req_bio_endio(req, bio, bio_bytes, error);
L
Linus Torvalds 已提交
1431

1432 1433
		total_bytes += bio_bytes;
		nr_bytes -= bio_bytes;
L
Linus Torvalds 已提交
1434

1435 1436
		if (!nr_bytes)
			break;
L
Linus Torvalds 已提交
1437 1438 1439 1440 1441
	}

	/*
	 * completely done
	 */
1442 1443 1444 1445 1446 1447
	if (!req->bio) {
		/*
		 * Reset counters so that the request stacking driver
		 * can find how many bytes remain in the request
		 * later.
		 */
1448
		req->__data_len = 0;
1449 1450
		return false;
	}
L
Linus Torvalds 已提交
1451

1452
	req->__data_len -= total_bytes;
1453 1454

	/* update sector only for requests with clear definition of sector */
1455
	if (!blk_rq_is_passthrough(req))
1456
		req->__sector += total_bytes >> 9;
1457

1458
	/* mixed attributes always follow the first bio */
1459
	if (req->rq_flags & RQF_MIXED_MERGE) {
1460
		req->cmd_flags &= ~REQ_FAILFAST_MASK;
J
Jens Axboe 已提交
1461
		req->cmd_flags |= req->bio->bi_opf & REQ_FAILFAST_MASK;
1462 1463
	}

1464 1465 1466 1467 1468 1469 1470 1471 1472
	if (!(req->rq_flags & RQF_SPECIAL_PAYLOAD)) {
		/*
		 * If total number of sectors is less than the first segment
		 * size, something has gone terribly wrong.
		 */
		if (blk_rq_bytes(req) < blk_rq_cur_bytes(req)) {
			blk_dump_rq_flags(req, "request botched");
			req->__data_len = blk_rq_cur_bytes(req);
		}
1473

1474
		/* recalculate the number of segments */
1475
		req->nr_phys_segments = blk_recalc_rq_segments(req);
1476
	}
1477

1478
	return true;
L
Linus Torvalds 已提交
1479
}
1480
EXPORT_SYMBOL_GPL(blk_update_request);
L
Linus Torvalds 已提交
1481

1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492
#if ARCH_IMPLEMENTS_FLUSH_DCACHE_PAGE
/**
 * rq_flush_dcache_pages - Helper function to flush all pages in a request
 * @rq: the request to be flushed
 *
 * Description:
 *     Flush all pages in @rq.
 */
void rq_flush_dcache_pages(struct request *rq)
{
	struct req_iterator iter;
1493
	struct bio_vec bvec;
1494 1495

	rq_for_each_segment(bvec, rq, iter)
1496
		flush_dcache_page(bvec.bv_page);
1497 1498 1499 1500
}
EXPORT_SYMBOL_GPL(rq_flush_dcache_pages);
#endif

1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521
/**
 * blk_lld_busy - Check if underlying low-level drivers of a device are busy
 * @q : the queue of the device being checked
 *
 * Description:
 *    Check if underlying low-level drivers of a device are busy.
 *    If the drivers want to export their busy state, they must set own
 *    exporting function using blk_queue_lld_busy() first.
 *
 *    Basically, this function is used only by request stacking drivers
 *    to stop dispatching requests to underlying devices when underlying
 *    devices are busy.  This behavior helps more I/O merging on the queue
 *    of the request stacking driver and prevents I/O throughput regression
 *    on burst I/O load.
 *
 * Return:
 *    0 - Not busy (The request stacking driver should dispatch request)
 *    1 - Busy (The request stacking driver should stop dispatching request)
 */
int blk_lld_busy(struct request_queue *q)
{
J
Jens Axboe 已提交
1522
	if (queue_is_mq(q) && q->mq_ops->busy)
J
Jens Axboe 已提交
1523
		return q->mq_ops->busy(q);
1524 1525 1526 1527 1528

	return 0;
}
EXPORT_SYMBOL_GPL(blk_lld_busy);

1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552
/**
 * blk_rq_unprep_clone - Helper function to free all bios in a cloned request
 * @rq: the clone request to be cleaned up
 *
 * Description:
 *     Free all bios in @rq for a cloned request.
 */
void blk_rq_unprep_clone(struct request *rq)
{
	struct bio *bio;

	while ((bio = rq->bio) != NULL) {
		rq->bio = bio->bi_next;

		bio_put(bio);
	}
}
EXPORT_SYMBOL_GPL(blk_rq_unprep_clone);

/*
 * Copy attributes of the original request to the clone request.
 * The actual data parts (e.g. ->cmd, ->sense) are not copied.
 */
static void __blk_rq_prep_clone(struct request *dst, struct request *src)
1553 1554 1555
{
	dst->__sector = blk_rq_pos(src);
	dst->__data_len = blk_rq_bytes(src);
1556 1557 1558 1559
	if (src->rq_flags & RQF_SPECIAL_PAYLOAD) {
		dst->rq_flags |= RQF_SPECIAL_PAYLOAD;
		dst->special_vec = src->special_vec;
	}
1560 1561 1562
	dst->nr_phys_segments = src->nr_phys_segments;
	dst->ioprio = src->ioprio;
	dst->extra_len = src->extra_len;
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591
}

/**
 * blk_rq_prep_clone - Helper function to setup clone request
 * @rq: the request to be setup
 * @rq_src: original request to be cloned
 * @bs: bio_set that bios for clone are allocated from
 * @gfp_mask: memory allocation mask for bio
 * @bio_ctr: setup function to be called for each clone bio.
 *           Returns %0 for success, non %0 for failure.
 * @data: private data to be passed to @bio_ctr
 *
 * Description:
 *     Clones bios in @rq_src to @rq, and copies attributes of @rq_src to @rq.
 *     The actual data parts of @rq_src (e.g. ->cmd, ->sense)
 *     are not copied, and copying such parts is the caller's responsibility.
 *     Also, pages which the original bios are pointing to are not copied
 *     and the cloned bios just point same pages.
 *     So cloned bios must be completed before original bios, which means
 *     the caller must complete @rq before @rq_src.
 */
int blk_rq_prep_clone(struct request *rq, struct request *rq_src,
		      struct bio_set *bs, gfp_t gfp_mask,
		      int (*bio_ctr)(struct bio *, struct bio *, void *),
		      void *data)
{
	struct bio *bio, *bio_src;

	if (!bs)
1592
		bs = &fs_bio_set;
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618

	__rq_for_each_bio(bio_src, rq_src) {
		bio = bio_clone_fast(bio_src, gfp_mask, bs);
		if (!bio)
			goto free_and_out;

		if (bio_ctr && bio_ctr(bio, bio_src, data))
			goto free_and_out;

		if (rq->bio) {
			rq->biotail->bi_next = bio;
			rq->biotail = bio;
		} else
			rq->bio = rq->biotail = bio;
	}

	__blk_rq_prep_clone(rq, rq_src);

	return 0;

free_and_out:
	if (bio)
		bio_put(bio);
	blk_rq_unprep_clone(rq);

	return -ENOMEM;
1619 1620 1621
}
EXPORT_SYMBOL_GPL(blk_rq_prep_clone);

1622
int kblockd_schedule_work(struct work_struct *work)
L
Linus Torvalds 已提交
1623 1624 1625 1626 1627
{
	return queue_work(kblockd_workqueue, work);
}
EXPORT_SYMBOL(kblockd_schedule_work);

1628 1629 1630 1631 1632 1633
int kblockd_schedule_work_on(int cpu, struct work_struct *work)
{
	return queue_work_on(cpu, kblockd_workqueue, work);
}
EXPORT_SYMBOL(kblockd_schedule_work_on);

1634 1635 1636 1637 1638 1639 1640
int kblockd_mod_delayed_work_on(int cpu, struct delayed_work *dwork,
				unsigned long delay)
{
	return mod_delayed_work_on(cpu, kblockd_workqueue, dwork, delay);
}
EXPORT_SYMBOL(kblockd_mod_delayed_work_on);

S
Suresh Jayaraman 已提交
1641 1642 1643 1644 1645
/**
 * blk_start_plug - initialize blk_plug and track it inside the task_struct
 * @plug:	The &struct blk_plug that needs to be initialized
 *
 * Description:
1646 1647 1648 1649 1650 1651 1652 1653 1654
 *   blk_start_plug() indicates to the block layer an intent by the caller
 *   to submit multiple I/O requests in a batch.  The block layer may use
 *   this hint to defer submitting I/Os from the caller until blk_finish_plug()
 *   is called.  However, the block layer may choose to submit requests
 *   before a call to blk_finish_plug() if the number of queued I/Os
 *   exceeds %BLK_MAX_REQUEST_COUNT, or if the size of the I/O is larger than
 *   %BLK_PLUG_FLUSH_SIZE.  The queued I/Os may also be submitted early if
 *   the task schedules (see below).
 *
S
Suresh Jayaraman 已提交
1655 1656 1657 1658 1659 1660 1661 1662 1663
 *   Tracking blk_plug inside the task_struct will help with auto-flushing the
 *   pending I/O should the task end up blocking between blk_start_plug() and
 *   blk_finish_plug(). This is important from a performance perspective, but
 *   also ensures that we don't deadlock. For instance, if the task is blocking
 *   for a memory allocation, memory reclaim could end up wanting to free a
 *   page belonging to that request that is currently residing in our private
 *   plug. By flushing the pending I/O when the process goes to sleep, we avoid
 *   this kind of deadlock.
 */
1664 1665 1666 1667
void blk_start_plug(struct blk_plug *plug)
{
	struct task_struct *tsk = current;

S
Shaohua Li 已提交
1668 1669 1670 1671 1672 1673
	/*
	 * If this is a nested plug, don't actually assign it.
	 */
	if (tsk->plug)
		return;

1674
	INIT_LIST_HEAD(&plug->mq_list);
1675
	INIT_LIST_HEAD(&plug->cb_list);
1676
	plug->rq_count = 0;
1677
	plug->multiple_queues = false;
1678

1679
	/*
S
Shaohua Li 已提交
1680 1681
	 * Store ordering should not be needed here, since a potential
	 * preempt will imply a full memory barrier
1682
	 */
S
Shaohua Li 已提交
1683
	tsk->plug = plug;
1684 1685 1686
}
EXPORT_SYMBOL(blk_start_plug);

1687
static void flush_plug_callbacks(struct blk_plug *plug, bool from_schedule)
1688 1689 1690
{
	LIST_HEAD(callbacks);

S
Shaohua Li 已提交
1691 1692
	while (!list_empty(&plug->cb_list)) {
		list_splice_init(&plug->cb_list, &callbacks);
1693

S
Shaohua Li 已提交
1694 1695
		while (!list_empty(&callbacks)) {
			struct blk_plug_cb *cb = list_first_entry(&callbacks,
1696 1697
							  struct blk_plug_cb,
							  list);
S
Shaohua Li 已提交
1698
			list_del(&cb->list);
1699
			cb->callback(cb, from_schedule);
S
Shaohua Li 已提交
1700
		}
1701 1702 1703
	}
}

1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728
struct blk_plug_cb *blk_check_plugged(blk_plug_cb_fn unplug, void *data,
				      int size)
{
	struct blk_plug *plug = current->plug;
	struct blk_plug_cb *cb;

	if (!plug)
		return NULL;

	list_for_each_entry(cb, &plug->cb_list, list)
		if (cb->callback == unplug && cb->data == data)
			return cb;

	/* Not currently on the callback list */
	BUG_ON(size < sizeof(*cb));
	cb = kzalloc(size, GFP_ATOMIC);
	if (cb) {
		cb->data = data;
		cb->callback = unplug;
		list_add(&cb->list, &plug->cb_list);
	}
	return cb;
}
EXPORT_SYMBOL(blk_check_plugged);

1729
void blk_flush_plug_list(struct blk_plug *plug, bool from_schedule)
1730
{
1731
	flush_plug_callbacks(plug, from_schedule);
1732 1733 1734

	if (!list_empty(&plug->mq_list))
		blk_mq_flush_plug_list(plug, from_schedule);
1735 1736
}

1737 1738 1739 1740 1741 1742 1743 1744 1745 1746
/**
 * blk_finish_plug - mark the end of a batch of submitted I/O
 * @plug:	The &struct blk_plug passed to blk_start_plug()
 *
 * Description:
 * Indicate that a batch of I/O submissions is complete.  This function
 * must be paired with an initial call to blk_start_plug().  The intent
 * is to allow the block layer to optimize I/O submission.  See the
 * documentation for blk_start_plug() for more information.
 */
1747 1748
void blk_finish_plug(struct blk_plug *plug)
{
S
Shaohua Li 已提交
1749 1750
	if (plug != current->plug)
		return;
1751
	blk_flush_plug_list(plug, false);
1752

S
Shaohua Li 已提交
1753
	current->plug = NULL;
1754
}
1755
EXPORT_SYMBOL(blk_finish_plug);
1756

L
Linus Torvalds 已提交
1757 1758
int __init blk_dev_init(void)
{
1759 1760
	BUILD_BUG_ON(REQ_OP_LAST >= (1 << REQ_OP_BITS));
	BUILD_BUG_ON(REQ_OP_BITS + REQ_FLAG_BITS > 8 *
1761
			FIELD_SIZEOF(struct request, cmd_flags));
1762 1763
	BUILD_BUG_ON(REQ_OP_BITS + REQ_FLAG_BITS > 8 *
			FIELD_SIZEOF(struct bio, bi_opf));
1764

1765 1766
	/* used for unplugging and affects IO latency/throughput - HIGHPRI */
	kblockd_workqueue = alloc_workqueue("kblockd",
1767
					    WQ_MEM_RECLAIM | WQ_HIGHPRI, 0);
L
Linus Torvalds 已提交
1768 1769 1770
	if (!kblockd_workqueue)
		panic("Failed to create kblockd\n");

1771
	blk_requestq_cachep = kmem_cache_create("request_queue",
1772
			sizeof(struct request_queue), 0, SLAB_PANIC, NULL);
L
Linus Torvalds 已提交
1773

1774 1775 1776 1777
#ifdef CONFIG_DEBUG_FS
	blk_debugfs_root = debugfs_create_dir("block", NULL);
#endif

1778
	return 0;
L
Linus Torvalds 已提交
1779
}