elevator.c 25.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5
/*
 *  Block device elevator/IO-scheduler.
 *
 *  Copyright (C) 2000 Andrea Arcangeli <andrea@suse.de> SuSE
 *
6
 * 30042000 Jens Axboe <axboe@kernel.dk> :
L
Linus Torvalds 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
 *
 * Split the elevator a bit so that it is possible to choose a different
 * one or even write a new "plug in". There are three pieces:
 * - elevator_fn, inserts a new request in the queue list
 * - elevator_merge_fn, decides whether a new buffer can be merged with
 *   an existing request
 * - elevator_dequeue_fn, called when a request is taken off the active list
 *
 * 20082000 Dave Jones <davej@suse.de> :
 * Removed tests for max-bomb-segments, which was breaking elvtune
 *  when run without -bN
 *
 * Jens:
 * - Rework again to work with bio instead of buffer_heads
 * - loose bi_dev comparisons, partition handling is right now
 * - completely modularize elevator setup and teardown
 *
 */
#include <linux/kernel.h>
#include <linux/fs.h>
#include <linux/blkdev.h>
#include <linux/elevator.h>
#include <linux/bio.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/init.h>
#include <linux/compiler.h>
T
Tejun Heo 已提交
34
#include <linux/delay.h>
35
#include <linux/blktrace_api.h>
36
#include <linux/hash.h>
L
Linus Torvalds 已提交
37 38 39 40 41 42

#include <asm/uaccess.h>

static DEFINE_SPINLOCK(elv_list_lock);
static LIST_HEAD(elv_list);

43 44 45 46 47
/*
 * Merge hash stuff.
 */
static const int elv_hash_shift = 6;
#define ELV_HASH_BLOCK(sec)	((sec) >> 3)
48 49
#define ELV_HASH_FN(sec)	\
		(hash_long(ELV_HASH_BLOCK((sec)), elv_hash_shift))
50 51 52 53
#define ELV_HASH_ENTRIES	(1 << elv_hash_shift)
#define rq_hash_key(rq)		((rq)->sector + (rq)->nr_sectors)
#define ELV_ON_HASH(rq)		(!hlist_unhashed(&(rq)->hash))

54 55 56 57 58 59
/*
 * Query io scheduler to see if the current process issuing bio may be
 * merged with rq.
 */
static int elv_iosched_allow_merge(struct request *rq, struct bio *bio)
{
60
	struct request_queue *q = rq->q;
61 62 63 64 65 66 67 68
	elevator_t *e = q->elevator;

	if (e->ops->elevator_allow_merge_fn)
		return e->ops->elevator_allow_merge_fn(q, rq, bio);

	return 1;
}

L
Linus Torvalds 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
/*
 * can we safely merge with this request?
 */
inline int elv_rq_merge_ok(struct request *rq, struct bio *bio)
{
	if (!rq_mergeable(rq))
		return 0;

	/*
	 * different data direction or already started, don't merge
	 */
	if (bio_data_dir(bio) != rq_data_dir(rq))
		return 0;

	/*
84
	 * must be same device and not a special request
L
Linus Torvalds 已提交
85
	 */
86
	if (rq->rq_disk != bio->bi_bdev->bd_disk || rq->special)
87 88 89 90
		return 0;

	if (!elv_iosched_allow_merge(rq, bio))
		return 0;
L
Linus Torvalds 已提交
91

92
	return 1;
L
Linus Torvalds 已提交
93 94 95
}
EXPORT_SYMBOL(elv_rq_merge_ok);

96
static inline int elv_try_merge(struct request *__rq, struct bio *bio)
L
Linus Torvalds 已提交
97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
{
	int ret = ELEVATOR_NO_MERGE;

	/*
	 * we can merge and sequence is ok, check if it's possible
	 */
	if (elv_rq_merge_ok(__rq, bio)) {
		if (__rq->sector + __rq->nr_sectors == bio->bi_sector)
			ret = ELEVATOR_BACK_MERGE;
		else if (__rq->sector - bio_sectors(bio) == bio->bi_sector)
			ret = ELEVATOR_FRONT_MERGE;
	}

	return ret;
}

static struct elevator_type *elevator_find(const char *name)
{
115
	struct elevator_type *e;
L
Linus Torvalds 已提交
116

117
	list_for_each_entry(e, &elv_list, list) {
118 119
		if (!strcmp(e->elevator_name, name))
			return e;
L
Linus Torvalds 已提交
120 121
	}

122
	return NULL;
L
Linus Torvalds 已提交
123 124 125 126 127 128 129 130 131
}

static void elevator_put(struct elevator_type *e)
{
	module_put(e->elevator_owner);
}

static struct elevator_type *elevator_get(const char *name)
{
132
	struct elevator_type *e;
L
Linus Torvalds 已提交
133

134
	spin_lock(&elv_list_lock);
135 136 137 138 139

	e = elevator_find(name);
	if (e && !try_module_get(e->elevator_owner))
		e = NULL;

140
	spin_unlock(&elv_list_lock);
L
Linus Torvalds 已提交
141 142 143 144

	return e;
}

145 146
static void *elevator_init_queue(struct request_queue *q,
				 struct elevator_queue *eq)
L
Linus Torvalds 已提交
147
{
148
	return eq->ops->elevator_init_fn(q);
J
Jens Axboe 已提交
149
}
L
Linus Torvalds 已提交
150

151
static void elevator_attach(struct request_queue *q, struct elevator_queue *eq,
J
Jens Axboe 已提交
152 153
			   void *data)
{
L
Linus Torvalds 已提交
154
	q->elevator = eq;
J
Jens Axboe 已提交
155
	eq->elevator_data = data;
L
Linus Torvalds 已提交
156 157 158 159
}

static char chosen_elevator[16];

160
static int __init elevator_setup(char *str)
L
Linus Torvalds 已提交
161
{
162 163 164 165
	/*
	 * Be backwards-compatible with previous kernels, so users
	 * won't get the wrong elevator.
	 */
166
	if (!strcmp(str, "as"))
167
		strcpy(chosen_elevator, "anticipatory");
Z
Zachary Amsden 已提交
168
	else
169
		strncpy(chosen_elevator, str, sizeof(chosen_elevator) - 1);
170
	return 1;
L
Linus Torvalds 已提交
171 172 173 174
}

__setup("elevator=", elevator_setup);

175 176
static struct kobj_type elv_ktype;

177 178
static elevator_t *elevator_alloc(struct request_queue *q,
				  struct elevator_type *e)
179
{
180 181 182
	elevator_t *eq;
	int i;

183
	eq = kmalloc_node(sizeof(elevator_t), GFP_KERNEL | __GFP_ZERO, q->node);
184 185 186 187 188
	if (unlikely(!eq))
		goto err;

	eq->ops = &e->ops;
	eq->elevator_type = e;
189
	kobject_init(&eq->kobj, &elv_ktype);
190 191
	mutex_init(&eq->sysfs_lock);

192 193
	eq->hash = kmalloc_node(sizeof(struct hlist_head) * ELV_HASH_ENTRIES,
					GFP_KERNEL, q->node);
194 195 196 197 198 199
	if (!eq->hash)
		goto err;

	for (i = 0; i < ELV_HASH_ENTRIES; i++)
		INIT_HLIST_HEAD(&eq->hash[i]);

200
	return eq;
201 202 203 204
err:
	kfree(eq);
	elevator_put(e);
	return NULL;
205 206 207 208 209
}

static void elevator_release(struct kobject *kobj)
{
	elevator_t *e = container_of(kobj, elevator_t, kobj);
210

211
	elevator_put(e->elevator_type);
212
	kfree(e->hash);
213 214 215
	kfree(e);
}

216
int elevator_init(struct request_queue *q, char *name)
L
Linus Torvalds 已提交
217 218 219 220
{
	struct elevator_type *e = NULL;
	struct elevator_queue *eq;
	int ret = 0;
J
Jens Axboe 已提交
221
	void *data;
L
Linus Torvalds 已提交
222

T
Tejun Heo 已提交
223 224 225 226 227
	INIT_LIST_HEAD(&q->queue_head);
	q->last_merge = NULL;
	q->end_sector = 0;
	q->boundary_rq = NULL;

228 229 230 231 232
	if (name) {
		e = elevator_get(name);
		if (!e)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
233

234 235 236 237 238 239
	if (!e && *chosen_elevator) {
		e = elevator_get(chosen_elevator);
		if (!e)
			printk(KERN_ERR "I/O scheduler %s not found\n",
							chosen_elevator);
	}
240

241 242 243 244 245 246 247 248
	if (!e) {
		e = elevator_get(CONFIG_DEFAULT_IOSCHED);
		if (!e) {
			printk(KERN_ERR
				"Default I/O scheduler not found. " \
				"Using noop.\n");
			e = elevator_get("noop");
		}
249 250
	}

251
	eq = elevator_alloc(q, e);
252
	if (!eq)
L
Linus Torvalds 已提交
253 254
		return -ENOMEM;

J
Jens Axboe 已提交
255 256
	data = elevator_init_queue(q, eq);
	if (!data) {
257
		kobject_put(&eq->kobj);
J
Jens Axboe 已提交
258 259
		return -ENOMEM;
	}
L
Linus Torvalds 已提交
260

J
Jens Axboe 已提交
261
	elevator_attach(q, eq, data);
L
Linus Torvalds 已提交
262 263
	return ret;
}
264 265
EXPORT_SYMBOL(elevator_init);

L
Linus Torvalds 已提交
266 267
void elevator_exit(elevator_t *e)
{
268
	mutex_lock(&e->sysfs_lock);
L
Linus Torvalds 已提交
269 270
	if (e->ops->elevator_exit_fn)
		e->ops->elevator_exit_fn(e);
271 272
	e->ops = NULL;
	mutex_unlock(&e->sysfs_lock);
L
Linus Torvalds 已提交
273

274
	kobject_put(&e->kobj);
L
Linus Torvalds 已提交
275
}
276 277
EXPORT_SYMBOL(elevator_exit);

278
static void elv_activate_rq(struct request_queue *q, struct request *rq)
279 280 281 282 283 284 285
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_activate_req_fn)
		e->ops->elevator_activate_req_fn(q, rq);
}

286
static void elv_deactivate_rq(struct request_queue *q, struct request *rq)
287 288 289 290 291 292 293
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_deactivate_req_fn)
		e->ops->elevator_deactivate_req_fn(q, rq);
}

294 295 296 297 298
static inline void __elv_rqhash_del(struct request *rq)
{
	hlist_del_init(&rq->hash);
}

299
static void elv_rqhash_del(struct request_queue *q, struct request *rq)
300 301 302 303 304
{
	if (ELV_ON_HASH(rq))
		__elv_rqhash_del(rq);
}

305
static void elv_rqhash_add(struct request_queue *q, struct request *rq)
306 307 308 309 310 311 312
{
	elevator_t *e = q->elevator;

	BUG_ON(ELV_ON_HASH(rq));
	hlist_add_head(&rq->hash, &e->hash[ELV_HASH_FN(rq_hash_key(rq))]);
}

313
static void elv_rqhash_reposition(struct request_queue *q, struct request *rq)
314 315 316 317 318
{
	__elv_rqhash_del(rq);
	elv_rqhash_add(q, rq);
}

319
static struct request *elv_rqhash_find(struct request_queue *q, sector_t offset)
320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
{
	elevator_t *e = q->elevator;
	struct hlist_head *hash_list = &e->hash[ELV_HASH_FN(offset)];
	struct hlist_node *entry, *next;
	struct request *rq;

	hlist_for_each_entry_safe(rq, entry, next, hash_list, hash) {
		BUG_ON(!ELV_ON_HASH(rq));

		if (unlikely(!rq_mergeable(rq))) {
			__elv_rqhash_del(rq);
			continue;
		}

		if (rq_hash_key(rq) == offset)
			return rq;
	}

	return NULL;
}

341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
/*
 * RB-tree support functions for inserting/lookup/removal of requests
 * in a sorted RB tree.
 */
struct request *elv_rb_add(struct rb_root *root, struct request *rq)
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct request *__rq;

	while (*p) {
		parent = *p;
		__rq = rb_entry(parent, struct request, rb_node);

		if (rq->sector < __rq->sector)
			p = &(*p)->rb_left;
		else if (rq->sector > __rq->sector)
			p = &(*p)->rb_right;
		else
			return __rq;
	}

	rb_link_node(&rq->rb_node, parent, p);
	rb_insert_color(&rq->rb_node, root);
	return NULL;
}
EXPORT_SYMBOL(elv_rb_add);

void elv_rb_del(struct rb_root *root, struct request *rq)
{
	BUG_ON(RB_EMPTY_NODE(&rq->rb_node));
	rb_erase(&rq->rb_node, root);
	RB_CLEAR_NODE(&rq->rb_node);
}
EXPORT_SYMBOL(elv_rb_del);

struct request *elv_rb_find(struct rb_root *root, sector_t sector)
{
	struct rb_node *n = root->rb_node;
	struct request *rq;

	while (n) {
		rq = rb_entry(n, struct request, rb_node);

		if (sector < rq->sector)
			n = n->rb_left;
		else if (sector > rq->sector)
			n = n->rb_right;
		else
			return rq;
	}

	return NULL;
}
EXPORT_SYMBOL(elv_rb_find);

397 398
/*
 * Insert rq into dispatch queue of q.  Queue lock must be held on
U
Uwe Kleine-König 已提交
399
 * entry.  rq is sort instead into the dispatch queue. To be used by
400
 * specific elevators.
401
 */
402
void elv_dispatch_sort(struct request_queue *q, struct request *rq)
403 404 405
{
	sector_t boundary;
	struct list_head *entry;
406
	int stop_flags;
407

408 409
	if (q->last_merge == rq)
		q->last_merge = NULL;
410 411 412

	elv_rqhash_del(q, rq);

413
	q->nr_sorted--;
414

J
Jens Axboe 已提交
415
	boundary = q->end_sector;
416
	stop_flags = REQ_SOFTBARRIER | REQ_HARDBARRIER | REQ_STARTED;
417 418 419
	list_for_each_prev(entry, &q->queue_head) {
		struct request *pos = list_entry_rq(entry);

420 421
		if (rq_data_dir(rq) != rq_data_dir(pos))
			break;
422
		if (pos->cmd_flags & stop_flags)
423 424 425 426 427 428 429 430 431 432 433 434 435 436
			break;
		if (rq->sector >= boundary) {
			if (pos->sector < boundary)
				continue;
		} else {
			if (pos->sector >= boundary)
				break;
		}
		if (rq->sector >= pos->sector)
			break;
	}

	list_add(&rq->queuelist, entry);
}
437 438
EXPORT_SYMBOL(elv_dispatch_sort);

439
/*
440 441 442
 * Insert rq into dispatch queue of q.  Queue lock must be held on
 * entry.  rq is added to the back of the dispatch queue. To be used by
 * specific elevators.
443 444 445 446 447 448 449 450 451 452 453 454 455 456
 */
void elv_dispatch_add_tail(struct request_queue *q, struct request *rq)
{
	if (q->last_merge == rq)
		q->last_merge = NULL;

	elv_rqhash_del(q, rq);

	q->nr_sorted--;

	q->end_sector = rq_end_sector(rq);
	q->boundary_rq = rq;
	list_add_tail(&rq->queuelist, &q->queue_head);
}
457 458
EXPORT_SYMBOL(elv_dispatch_add_tail);

459
int elv_merge(struct request_queue *q, struct request **req, struct bio *bio)
L
Linus Torvalds 已提交
460 461
{
	elevator_t *e = q->elevator;
462
	struct request *__rq;
463 464
	int ret;

465 466 467
	/*
	 * First try one-hit cache.
	 */
468 469 470 471 472 473 474
	if (q->last_merge) {
		ret = elv_try_merge(q->last_merge, bio);
		if (ret != ELEVATOR_NO_MERGE) {
			*req = q->last_merge;
			return ret;
		}
	}
L
Linus Torvalds 已提交
475

476 477 478 479 480 481 482 483 484
	/*
	 * See if our hash lookup can find a potential backmerge.
	 */
	__rq = elv_rqhash_find(q, bio->bi_sector);
	if (__rq && elv_rq_merge_ok(__rq, bio)) {
		*req = __rq;
		return ELEVATOR_BACK_MERGE;
	}

L
Linus Torvalds 已提交
485 486 487 488 489 490
	if (e->ops->elevator_merge_fn)
		return e->ops->elevator_merge_fn(q, req, bio);

	return ELEVATOR_NO_MERGE;
}

491
void elv_merged_request(struct request_queue *q, struct request *rq, int type)
L
Linus Torvalds 已提交
492 493 494 495
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_merged_fn)
496
		e->ops->elevator_merged_fn(q, rq, type);
497

498 499
	if (type == ELEVATOR_BACK_MERGE)
		elv_rqhash_reposition(q, rq);
500

501
	q->last_merge = rq;
L
Linus Torvalds 已提交
502 503
}

504
void elv_merge_requests(struct request_queue *q, struct request *rq,
L
Linus Torvalds 已提交
505 506 507 508 509 510
			     struct request *next)
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_merge_req_fn)
		e->ops->elevator_merge_req_fn(q, rq, next);
511

512 513 514 515
	elv_rqhash_reposition(q, rq);
	elv_rqhash_del(q, next);

	q->nr_sorted--;
516
	q->last_merge = rq;
L
Linus Torvalds 已提交
517 518
}

519
void elv_requeue_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
520 521 522 523 524
{
	/*
	 * it already went through dequeue, we need to decrement the
	 * in_flight count again
	 */
525
	if (blk_account_rq(rq)) {
L
Linus Torvalds 已提交
526
		q->in_flight--;
527 528
		if (blk_sorted_rq(rq))
			elv_deactivate_rq(q, rq);
529
	}
L
Linus Torvalds 已提交
530

531
	rq->cmd_flags &= ~REQ_STARTED;
L
Linus Torvalds 已提交
532

533
	elv_insert(q, rq, ELEVATOR_INSERT_REQUEUE);
L
Linus Torvalds 已提交
534 535
}

536
static void elv_drain_elevator(struct request_queue *q)
537 538 539 540 541 542 543 544 545 546 547 548 549
{
	static int printed;
	while (q->elevator->ops->elevator_dispatch_fn(q, 1))
		;
	if (q->nr_sorted == 0)
		return;
	if (printed++ < 10) {
		printk(KERN_ERR "%s: forced dispatching is broken "
		       "(nr_sorted=%u), please report this\n",
		       q->elevator->elevator_type->elevator_name, q->nr_sorted);
	}
}

550
void elv_insert(struct request_queue *q, struct request *rq, int where)
L
Linus Torvalds 已提交
551
{
552 553
	struct list_head *pos;
	unsigned ordseq;
J
Jens Axboe 已提交
554
	int unplug_it = 1;
555

556 557
	blk_add_trace_rq(q, rq, BLK_TA_INSERT);

L
Linus Torvalds 已提交
558 559
	rq->q = q;

560 561
	switch (where) {
	case ELEVATOR_INSERT_FRONT:
562
		rq->cmd_flags |= REQ_SOFTBARRIER;
563 564 565 566 567

		list_add(&rq->queuelist, &q->queue_head);
		break;

	case ELEVATOR_INSERT_BACK:
568
		rq->cmd_flags |= REQ_SOFTBARRIER;
569
		elv_drain_elevator(q);
570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
		list_add_tail(&rq->queuelist, &q->queue_head);
		/*
		 * We kick the queue here for the following reasons.
		 * - The elevator might have returned NULL previously
		 *   to delay requests and returned them now.  As the
		 *   queue wasn't empty before this request, ll_rw_blk
		 *   won't run the queue on return, resulting in hang.
		 * - Usually, back inserted requests won't be merged
		 *   with anything.  There's no point in delaying queue
		 *   processing.
		 */
		blk_remove_plug(q);
		q->request_fn(q);
		break;

	case ELEVATOR_INSERT_SORT:
		BUG_ON(!blk_fs_request(rq));
587
		rq->cmd_flags |= REQ_SORTED;
588
		q->nr_sorted++;
589 590 591 592 593 594
		if (rq_mergeable(rq)) {
			elv_rqhash_add(q, rq);
			if (!q->last_merge)
				q->last_merge = rq;
		}

595 596 597 598 599 600
		/*
		 * Some ioscheds (cfq) run q->request_fn directly, so
		 * rq cannot be accessed after calling
		 * elevator_add_req_fn.
		 */
		q->elevator->ops->elevator_add_req_fn(q, rq);
601 602
		break;

603 604 605 606 607 608
	case ELEVATOR_INSERT_REQUEUE:
		/*
		 * If ordered flush isn't in progress, we do front
		 * insertion; otherwise, requests should be requeued
		 * in ordseq order.
		 */
609
		rq->cmd_flags |= REQ_SOFTBARRIER;
610

611 612 613 614 615 616
		/*
		 * Most requeues happen because of a busy condition,
		 * don't force unplug of the queue for that case.
		 */
		unplug_it = 0;

617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
		if (q->ordseq == 0) {
			list_add(&rq->queuelist, &q->queue_head);
			break;
		}

		ordseq = blk_ordered_req_seq(rq);

		list_for_each(pos, &q->queue_head) {
			struct request *pos_rq = list_entry_rq(pos);
			if (ordseq <= blk_ordered_req_seq(pos_rq))
				break;
		}

		list_add_tail(&rq->queuelist, pos);
		break;

633 634 635 636 637 638
	default:
		printk(KERN_ERR "%s: bad insertion point %d\n",
		       __FUNCTION__, where);
		BUG();
	}

J
Jens Axboe 已提交
639
	if (unplug_it && blk_queue_plugged(q)) {
640 641 642 643 644 645
		int nrq = q->rq.count[READ] + q->rq.count[WRITE]
			- q->in_flight;

		if (nrq >= q->unplug_thresh)
			__generic_unplug_device(q);
	}
L
Linus Torvalds 已提交
646 647
}

648
void __elv_add_request(struct request_queue *q, struct request *rq, int where,
649 650 651
		       int plug)
{
	if (q->ordcolor)
652
		rq->cmd_flags |= REQ_ORDERED_COLOR;
653

654
	if (rq->cmd_flags & (REQ_SOFTBARRIER | REQ_HARDBARRIER)) {
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674
		/*
		 * toggle ordered color
		 */
		if (blk_barrier_rq(rq))
			q->ordcolor ^= 1;

		/*
		 * barriers implicitly indicate back insertion
		 */
		if (where == ELEVATOR_INSERT_SORT)
			where = ELEVATOR_INSERT_BACK;

		/*
		 * this request is scheduling boundary, update
		 * end_sector
		 */
		if (blk_fs_request(rq)) {
			q->end_sector = rq_end_sector(rq);
			q->boundary_rq = rq;
		}
675 676
	} else if (!(rq->cmd_flags & REQ_ELVPRIV) &&
		    where == ELEVATOR_INSERT_SORT)
677 678 679 680 681 682 683
		where = ELEVATOR_INSERT_BACK;

	if (plug)
		blk_plug_device(q);

	elv_insert(q, rq, where);
}
684 685
EXPORT_SYMBOL(__elv_add_request);

686
void elv_add_request(struct request_queue *q, struct request *rq, int where,
L
Linus Torvalds 已提交
687 688 689 690 691 692 693 694
		     int plug)
{
	unsigned long flags;

	spin_lock_irqsave(q->queue_lock, flags);
	__elv_add_request(q, rq, where, plug);
	spin_unlock_irqrestore(q->queue_lock, flags);
}
695 696
EXPORT_SYMBOL(elv_add_request);

697
static inline struct request *__elv_next_request(struct request_queue *q)
L
Linus Torvalds 已提交
698
{
699 700
	struct request *rq;

701 702 703 704 705 706
	while (1) {
		while (!list_empty(&q->queue_head)) {
			rq = list_entry_rq(q->queue_head.next);
			if (blk_do_ordered(q, &rq))
				return rq;
		}
L
Linus Torvalds 已提交
707

708 709
		if (!q->elevator->ops->elevator_dispatch_fn(q, 0))
			return NULL;
L
Linus Torvalds 已提交
710 711 712
	}
}

713
struct request *elv_next_request(struct request_queue *q)
L
Linus Torvalds 已提交
714 715 716 717 718
{
	struct request *rq;
	int ret;

	while ((rq = __elv_next_request(q)) != NULL) {
719 720 721 722 723 724 725 726
		/*
		 * Kill the empty barrier place holder, the driver must
		 * not ever see it.
		 */
		if (blk_empty_barrier(rq)) {
			end_queued_request(rq, 1);
			continue;
		}
727
		if (!(rq->cmd_flags & REQ_STARTED)) {
728 729 730 731 732
			/*
			 * This is the first time the device driver
			 * sees this request (possibly after
			 * requeueing).  Notify IO scheduler.
			 */
733 734
			if (blk_sorted_rq(rq))
				elv_activate_rq(q, rq);
L
Linus Torvalds 已提交
735

736 737 738 739 740
			/*
			 * just mark as started even if we don't start
			 * it, a request that has been delayed should
			 * not be passed by new incoming requests
			 */
741
			rq->cmd_flags |= REQ_STARTED;
742
			blk_add_trace_rq(q, rq, BLK_TA_ISSUE);
743
		}
L
Linus Torvalds 已提交
744

745
		if (!q->boundary_rq || q->boundary_rq == rq) {
J
Jens Axboe 已提交
746
			q->end_sector = rq_end_sector(rq);
747 748
			q->boundary_rq = NULL;
		}
L
Linus Torvalds 已提交
749

J
James Bottomley 已提交
750 751 752 753 754 755 756 757 758 759 760 761 762 763 764
		if (rq->cmd_flags & REQ_DONTPREP)
			break;

		if (q->dma_drain_size && rq->data_len) {
			/*
			 * make sure space for the drain appears we
			 * know we can do this because max_hw_segments
			 * has been adjusted to be one fewer than the
			 * device can handle
			 */
			rq->nr_phys_segments++;
			rq->nr_hw_segments++;
		}

		if (!q->prep_rq_fn)
L
Linus Torvalds 已提交
765 766 767 768 769 770
			break;

		ret = q->prep_rq_fn(q, rq);
		if (ret == BLKPREP_OK) {
			break;
		} else if (ret == BLKPREP_DEFER) {
771 772 773
			/*
			 * the request may have been (partially) prepped.
			 * we need to keep this request in the front to
774 775
			 * avoid resource deadlock.  REQ_STARTED will
			 * prevent other fs requests from passing this one.
776
			 */
J
James Bottomley 已提交
777 778 779 780 781 782 783 784 785 786
			if (q->dma_drain_size && rq->data_len &&
			    !(rq->cmd_flags & REQ_DONTPREP)) {
				/*
				 * remove the space for the drain we added
				 * so that we don't add it again
				 */
				--rq->nr_phys_segments;
				--rq->nr_hw_segments;
			}

L
Linus Torvalds 已提交
787 788 789
			rq = NULL;
			break;
		} else if (ret == BLKPREP_KILL) {
790
			rq->cmd_flags |= REQ_QUIET;
791
			end_queued_request(rq, 0);
L
Linus Torvalds 已提交
792 793 794 795 796 797 798 799 800
		} else {
			printk(KERN_ERR "%s: bad return=%d\n", __FUNCTION__,
								ret);
			break;
		}
	}

	return rq;
}
801 802
EXPORT_SYMBOL(elv_next_request);

803
void elv_dequeue_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
804
{
805
	BUG_ON(list_empty(&rq->queuelist));
806
	BUG_ON(ELV_ON_HASH(rq));
807 808

	list_del_init(&rq->queuelist);
L
Linus Torvalds 已提交
809 810 811 812

	/*
	 * the time frame between a request being removed from the lists
	 * and to it is freed is accounted as io that is in progress at
813
	 * the driver side.
L
Linus Torvalds 已提交
814 815 816 817
	 */
	if (blk_account_rq(rq))
		q->in_flight++;
}
818 819
EXPORT_SYMBOL(elv_dequeue_request);

820
int elv_queue_empty(struct request_queue *q)
L
Linus Torvalds 已提交
821 822 823
{
	elevator_t *e = q->elevator;

824 825 826
	if (!list_empty(&q->queue_head))
		return 0;

L
Linus Torvalds 已提交
827 828 829
	if (e->ops->elevator_queue_empty_fn)
		return e->ops->elevator_queue_empty_fn(q);

830
	return 1;
L
Linus Torvalds 已提交
831
}
832 833
EXPORT_SYMBOL(elv_queue_empty);

834
struct request *elv_latter_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
835 836 837 838 839 840 841 842
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_latter_req_fn)
		return e->ops->elevator_latter_req_fn(q, rq);
	return NULL;
}

843
struct request *elv_former_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
844 845 846 847 848 849 850 851
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_former_req_fn)
		return e->ops->elevator_former_req_fn(q, rq);
	return NULL;
}

852
int elv_set_request(struct request_queue *q, struct request *rq, gfp_t gfp_mask)
L
Linus Torvalds 已提交
853 854 855 856
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_set_req_fn)
857
		return e->ops->elevator_set_req_fn(q, rq, gfp_mask);
L
Linus Torvalds 已提交
858 859 860 861 862

	rq->elevator_private = NULL;
	return 0;
}

863
void elv_put_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
864 865 866 867
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_put_req_fn)
868
		e->ops->elevator_put_req_fn(rq);
L
Linus Torvalds 已提交
869 870
}

871
int elv_may_queue(struct request_queue *q, int rw)
L
Linus Torvalds 已提交
872 873 874 875
{
	elevator_t *e = q->elevator;

	if (e->ops->elevator_may_queue_fn)
876
		return e->ops->elevator_may_queue_fn(q, rw);
L
Linus Torvalds 已提交
877 878 879 880

	return ELV_MQUEUE_MAY;
}

881
void elv_completed_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
882 883 884 885 886 887
{
	elevator_t *e = q->elevator;

	/*
	 * request is released from the driver, io must be done
	 */
888
	if (blk_account_rq(rq)) {
L
Linus Torvalds 已提交
889
		q->in_flight--;
890 891 892
		if (blk_sorted_rq(rq) && e->ops->elevator_completed_req_fn)
			e->ops->elevator_completed_req_fn(q, rq);
	}
893

894 895 896 897 898 899 900
	/*
	 * Check if the queue is waiting for fs requests to be
	 * drained for flush sequence.
	 */
	if (unlikely(q->ordseq)) {
		struct request *first_rq = list_entry_rq(q->queue_head.next);
		if (q->in_flight == 0 &&
901 902 903 904 905
		    blk_ordered_cur_seq(q) == QUEUE_ORDSEQ_DRAIN &&
		    blk_ordered_req_seq(first_rq) > QUEUE_ORDSEQ_DRAIN) {
			blk_ordered_complete_seq(q, QUEUE_ORDSEQ_DRAIN, 0);
			q->request_fn(q);
		}
906
	}
L
Linus Torvalds 已提交
907 908
}

909 910 911 912
#define to_elv(atr) container_of((atr), struct elv_fs_entry, attr)

static ssize_t
elv_attr_show(struct kobject *kobj, struct attribute *attr, char *page)
L
Linus Torvalds 已提交
913
{
914 915 916 917 918 919 920 921 922 923 924 925
	elevator_t *e = container_of(kobj, elevator_t, kobj);
	struct elv_fs_entry *entry = to_elv(attr);
	ssize_t error;

	if (!entry->show)
		return -EIO;

	mutex_lock(&e->sysfs_lock);
	error = e->ops ? entry->show(e, page) : -ENOENT;
	mutex_unlock(&e->sysfs_lock);
	return error;
}
L
Linus Torvalds 已提交
926

927 928 929 930 931 932 933
static ssize_t
elv_attr_store(struct kobject *kobj, struct attribute *attr,
	       const char *page, size_t length)
{
	elevator_t *e = container_of(kobj, elevator_t, kobj);
	struct elv_fs_entry *entry = to_elv(attr);
	ssize_t error;
L
Linus Torvalds 已提交
934

935 936
	if (!entry->store)
		return -EIO;
L
Linus Torvalds 已提交
937

938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958
	mutex_lock(&e->sysfs_lock);
	error = e->ops ? entry->store(e, page, length) : -ENOENT;
	mutex_unlock(&e->sysfs_lock);
	return error;
}

static struct sysfs_ops elv_sysfs_ops = {
	.show	= elv_attr_show,
	.store	= elv_attr_store,
};

static struct kobj_type elv_ktype = {
	.sysfs_ops	= &elv_sysfs_ops,
	.release	= elevator_release,
};

int elv_register_queue(struct request_queue *q)
{
	elevator_t *e = q->elevator;
	int error;

959
	error = kobject_add(&e->kobj, &q->kobj, "%s", "iosched");
960
	if (!error) {
961
		struct elv_fs_entry *attr = e->elevator_type->elevator_attrs;
962
		if (attr) {
963 964
			while (attr->attr.name) {
				if (sysfs_create_file(&e->kobj, &attr->attr))
965
					break;
966
				attr++;
967 968 969 970 971
			}
		}
		kobject_uevent(&e->kobj, KOBJ_ADD);
	}
	return error;
L
Linus Torvalds 已提交
972 973
}

J
Jens Axboe 已提交
974 975 976 977 978 979
static void __elv_unregister_queue(elevator_t *e)
{
	kobject_uevent(&e->kobj, KOBJ_REMOVE);
	kobject_del(&e->kobj);
}

L
Linus Torvalds 已提交
980 981
void elv_unregister_queue(struct request_queue *q)
{
J
Jens Axboe 已提交
982 983
	if (q)
		__elv_unregister_queue(q->elevator);
L
Linus Torvalds 已提交
984 985
}

986
void elv_register(struct elevator_type *e)
L
Linus Torvalds 已提交
987
{
988
	char *def = "";
989 990

	spin_lock(&elv_list_lock);
991
	BUG_ON(elevator_find(e->elevator_name));
L
Linus Torvalds 已提交
992
	list_add_tail(&e->list, &elv_list);
993
	spin_unlock(&elv_list_lock);
L
Linus Torvalds 已提交
994

995 996 997
	if (!strcmp(e->elevator_name, chosen_elevator) ||
			(!*chosen_elevator &&
			 !strcmp(e->elevator_name, CONFIG_DEFAULT_IOSCHED)))
998 999
				def = " (default)";

1000 1001
	printk(KERN_INFO "io scheduler %s registered%s\n", e->elevator_name,
								def);
L
Linus Torvalds 已提交
1002 1003 1004 1005 1006
}
EXPORT_SYMBOL_GPL(elv_register);

void elv_unregister(struct elevator_type *e)
{
1007 1008 1009 1010 1011
	struct task_struct *g, *p;

	/*
	 * Iterate every thread in the process to remove the io contexts.
	 */
1012 1013 1014 1015
	if (e->ops.trim) {
		read_lock(&tasklist_lock);
		do_each_thread(g, p) {
			task_lock(p);
1016 1017
			if (p->io_context)
				e->ops.trim(p->io_context);
1018 1019 1020 1021
			task_unlock(p);
		} while_each_thread(g, p);
		read_unlock(&tasklist_lock);
	}
1022

1023
	spin_lock(&elv_list_lock);
L
Linus Torvalds 已提交
1024
	list_del_init(&e->list);
1025
	spin_unlock(&elv_list_lock);
L
Linus Torvalds 已提交
1026 1027 1028 1029 1030 1031 1032
}
EXPORT_SYMBOL_GPL(elv_unregister);

/*
 * switch to new_e io scheduler. be careful not to introduce deadlocks -
 * we don't free the old io scheduler, before we have allocated what we
 * need for the new one. this way we have a chance of going back to the old
T
Tejun Heo 已提交
1033
 * one, if the new one fails init for some reason.
L
Linus Torvalds 已提交
1034
 */
1035
static int elevator_switch(struct request_queue *q, struct elevator_type *new_e)
L
Linus Torvalds 已提交
1036
{
T
Tejun Heo 已提交
1037
	elevator_t *old_elevator, *e;
J
Jens Axboe 已提交
1038
	void *data;
L
Linus Torvalds 已提交
1039

T
Tejun Heo 已提交
1040 1041 1042
	/*
	 * Allocate new elevator
	 */
1043
	e = elevator_alloc(q, new_e);
L
Linus Torvalds 已提交
1044
	if (!e)
1045
		return 0;
L
Linus Torvalds 已提交
1046

J
Jens Axboe 已提交
1047 1048 1049 1050 1051 1052
	data = elevator_init_queue(q, e);
	if (!data) {
		kobject_put(&e->kobj);
		return 0;
	}

L
Linus Torvalds 已提交
1053
	/*
T
Tejun Heo 已提交
1054
	 * Turn on BYPASS and drain all requests w/ elevator private data
L
Linus Torvalds 已提交
1055
	 */
T
Tejun Heo 已提交
1056 1057
	spin_lock_irq(q->queue_lock);

J
Jens Axboe 已提交
1058
	set_bit(QUEUE_FLAG_ELVSWITCH, &q->queue_flags);
T
Tejun Heo 已提交
1059

1060
	elv_drain_elevator(q);
T
Tejun Heo 已提交
1061 1062

	while (q->rq.elvpriv) {
1063 1064
		blk_remove_plug(q);
		q->request_fn(q);
T
Tejun Heo 已提交
1065
		spin_unlock_irq(q->queue_lock);
J
Jens Axboe 已提交
1066
		msleep(10);
T
Tejun Heo 已提交
1067
		spin_lock_irq(q->queue_lock);
1068
		elv_drain_elevator(q);
T
Tejun Heo 已提交
1069 1070
	}

L
Linus Torvalds 已提交
1071
	/*
J
Jens Axboe 已提交
1072
	 * Remember old elevator.
L
Linus Torvalds 已提交
1073 1074 1075 1076 1077 1078
	 */
	old_elevator = q->elevator;

	/*
	 * attach and start new elevator
	 */
J
Jens Axboe 已提交
1079 1080 1081 1082 1083
	elevator_attach(q, e, data);

	spin_unlock_irq(q->queue_lock);

	__elv_unregister_queue(old_elevator);
L
Linus Torvalds 已提交
1084 1085 1086 1087 1088

	if (elv_register_queue(q))
		goto fail_register;

	/*
T
Tejun Heo 已提交
1089
	 * finally exit old elevator and turn off BYPASS.
L
Linus Torvalds 已提交
1090 1091
	 */
	elevator_exit(old_elevator);
J
Jens Axboe 已提交
1092
	clear_bit(QUEUE_FLAG_ELVSWITCH, &q->queue_flags);
1093
	return 1;
L
Linus Torvalds 已提交
1094 1095 1096 1097 1098 1099 1100 1101 1102

fail_register:
	/*
	 * switch failed, exit the new io scheduler and reattach the old
	 * one again (along with re-adding the sysfs dir)
	 */
	elevator_exit(e);
	q->elevator = old_elevator;
	elv_register_queue(q);
J
Jens Axboe 已提交
1103
	clear_bit(QUEUE_FLAG_ELVSWITCH, &q->queue_flags);
1104
	return 0;
L
Linus Torvalds 已提交
1105 1106
}

1107 1108
ssize_t elv_iosched_store(struct request_queue *q, const char *name,
			  size_t count)
L
Linus Torvalds 已提交
1109 1110
{
	char elevator_name[ELV_NAME_MAX];
1111
	size_t len;
L
Linus Torvalds 已提交
1112 1113
	struct elevator_type *e;

1114 1115 1116
	elevator_name[sizeof(elevator_name) - 1] = '\0';
	strncpy(elevator_name, name, sizeof(elevator_name) - 1);
	len = strlen(elevator_name);
L
Linus Torvalds 已提交
1117

1118 1119
	if (len && elevator_name[len - 1] == '\n')
		elevator_name[len - 1] = '\0';
L
Linus Torvalds 已提交
1120 1121 1122 1123 1124 1125 1126

	e = elevator_get(elevator_name);
	if (!e) {
		printk(KERN_ERR "elevator: type %s not found\n", elevator_name);
		return -EINVAL;
	}

1127 1128
	if (!strcmp(elevator_name, q->elevator->elevator_type->elevator_name)) {
		elevator_put(e);
L
Linus Torvalds 已提交
1129
		return count;
1130
	}
L
Linus Torvalds 已提交
1131

1132
	if (!elevator_switch(q, e))
1133 1134
		printk(KERN_ERR "elevator: switch to %s failed\n",
							elevator_name);
L
Linus Torvalds 已提交
1135 1136 1137
	return count;
}

1138
ssize_t elv_iosched_show(struct request_queue *q, char *name)
L
Linus Torvalds 已提交
1139 1140 1141
{
	elevator_t *e = q->elevator;
	struct elevator_type *elv = e->elevator_type;
1142
	struct elevator_type *__e;
L
Linus Torvalds 已提交
1143 1144
	int len = 0;

1145
	spin_lock(&elv_list_lock);
1146
	list_for_each_entry(__e, &elv_list, list) {
L
Linus Torvalds 已提交
1147 1148 1149 1150 1151
		if (!strcmp(elv->elevator_name, __e->elevator_name))
			len += sprintf(name+len, "[%s] ", elv->elevator_name);
		else
			len += sprintf(name+len, "%s ", __e->elevator_name);
	}
1152
	spin_unlock(&elv_list_lock);
L
Linus Torvalds 已提交
1153 1154 1155 1156 1157

	len += sprintf(len+name, "\n");
	return len;
}

1158 1159
struct request *elv_rb_former_request(struct request_queue *q,
				      struct request *rq)
1160 1161 1162 1163 1164 1165 1166 1167 1168 1169
{
	struct rb_node *rbprev = rb_prev(&rq->rb_node);

	if (rbprev)
		return rb_entry_rq(rbprev);

	return NULL;
}
EXPORT_SYMBOL(elv_rb_former_request);

1170 1171
struct request *elv_rb_latter_request(struct request_queue *q,
				      struct request *rq)
1172 1173 1174 1175 1176 1177 1178 1179 1180
{
	struct rb_node *rbnext = rb_next(&rq->rb_node);

	if (rbnext)
		return rb_entry_rq(rbnext);

	return NULL;
}
EXPORT_SYMBOL(elv_rb_latter_request);