cfq-iosched.c 50.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9
/*
 *  CFQ, or complete fairness queueing, disk scheduler.
 *
 *  Based on ideas from a previously unfinished io
 *  scheduler (round robin per-process disk scheduling) and Andrea Arcangeli.
 *
 *  Copyright (C) 2003 Jens Axboe <axboe@suse.de>
 */
#include <linux/module.h>
A
Al Viro 已提交
10 11
#include <linux/blkdev.h>
#include <linux/elevator.h>
L
Linus Torvalds 已提交
12 13
#include <linux/hash.h>
#include <linux/rbtree.h>
14
#include <linux/ioprio.h>
L
Linus Torvalds 已提交
15 16 17 18

/*
 * tunables
 */
19 20 21 22
static const int cfq_quantum = 4;		/* max queue in one round of service */
static const int cfq_fifo_expire[2] = { HZ / 4, HZ / 8 };
static const int cfq_back_max = 16 * 1024;	/* maximum backwards seek, in KiB */
static const int cfq_back_penalty = 2;		/* penalty of a backwards seek */
L
Linus Torvalds 已提交
23

24
static const int cfq_slice_sync = HZ / 10;
J
Jens Axboe 已提交
25
static int cfq_slice_async = HZ / 25;
26
static const int cfq_slice_async_rq = 2;
27
static int cfq_slice_idle = HZ / 125;
28 29 30 31 32 33

#define CFQ_IDLE_GRACE		(HZ / 10)
#define CFQ_SLICE_SCALE		(5)

#define CFQ_KEY_ASYNC		(0)

L
Linus Torvalds 已提交
34 35 36 37 38 39 40 41 42
/*
 * for the hash of cfqq inside the cfqd
 */
#define CFQ_QHASH_SHIFT		6
#define CFQ_QHASH_ENTRIES	(1 << CFQ_QHASH_SHIFT)
#define list_entry_qhash(entry)	hlist_entry((entry), struct cfq_queue, cfq_hash)

#define list_entry_cfqq(ptr)	list_entry((ptr), struct cfq_queue, cfq_list)

J
Jens Axboe 已提交
43 44
#define RQ_CIC(rq)		((struct cfq_io_context*)(rq)->elevator_private)
#define RQ_CFQQ(rq)		((rq)->elevator_private2)
L
Linus Torvalds 已提交
45 46 47 48

static kmem_cache_t *cfq_pool;
static kmem_cache_t *cfq_ioc_pool;

49
static DEFINE_PER_CPU(unsigned long, ioc_count);
50 51
static struct completion *ioc_gone;

52 53 54 55
#define CFQ_PRIO_LISTS		IOPRIO_BE_NR
#define cfq_class_idle(cfqq)	((cfqq)->ioprio_class == IOPRIO_CLASS_IDLE)
#define cfq_class_rt(cfqq)	((cfqq)->ioprio_class == IOPRIO_CLASS_RT)

J
Jens Axboe 已提交
56 57 58 59 60 61 62 63 64 65
#define ASYNC			(0)
#define SYNC			(1)

#define cfq_cfqq_dispatched(cfqq)	\
	((cfqq)->on_dispatch[ASYNC] + (cfqq)->on_dispatch[SYNC])

#define cfq_cfqq_class_sync(cfqq)	((cfqq)->key != CFQ_KEY_ASYNC)

#define cfq_cfqq_sync(cfqq)		\
	(cfq_cfqq_class_sync(cfqq) || (cfqq)->on_dispatch[SYNC])
66

67 68
#define sample_valid(samples)	((samples) > 80)

69 70 71
/*
 * Per block device queue structure
 */
L
Linus Torvalds 已提交
72
struct cfq_data {
73 74 75 76 77 78 79 80 81 82 83 84 85 86
	request_queue_t *queue;

	/*
	 * rr list of queues with requests and the count of them
	 */
	struct list_head rr_list[CFQ_PRIO_LISTS];
	struct list_head busy_rr;
	struct list_head cur_rr;
	struct list_head idle_rr;
	unsigned int busy_queues;

	/*
	 * non-ordered list of empty cfqq's
	 */
L
Linus Torvalds 已提交
87 88
	struct list_head empty_list;

89 90 91
	/*
	 * cfqq lookup hash
	 */
L
Linus Torvalds 已提交
92 93
	struct hlist_head *cfq_hash;

94
	int rq_in_driver;
95
	int hw_tag;
L
Linus Torvalds 已提交
96

97 98 99 100 101
	/*
	 * idle window management
	 */
	struct timer_list idle_slice_timer;
	struct work_struct unplug_work;
L
Linus Torvalds 已提交
102

103 104 105 106 107 108
	struct cfq_queue *active_queue;
	struct cfq_io_context *active_cic;
	int cur_prio, cur_end_prio;
	unsigned int dispatch_slice;

	struct timer_list idle_class_timer;
L
Linus Torvalds 已提交
109 110

	sector_t last_sector;
111
	unsigned long last_end_request;
L
Linus Torvalds 已提交
112 113 114 115 116

	/*
	 * tunables, see top of file
	 */
	unsigned int cfq_quantum;
117
	unsigned int cfq_fifo_expire[2];
L
Linus Torvalds 已提交
118 119
	unsigned int cfq_back_penalty;
	unsigned int cfq_back_max;
120 121 122
	unsigned int cfq_slice[2];
	unsigned int cfq_slice_async_rq;
	unsigned int cfq_slice_idle;
123 124

	struct list_head cic_list;
L
Linus Torvalds 已提交
125 126
};

127 128 129
/*
 * Per process-grouping structure
 */
L
Linus Torvalds 已提交
130 131 132 133 134
struct cfq_queue {
	/* reference count */
	atomic_t ref;
	/* parent cfq_data */
	struct cfq_data *cfqd;
135
	/* cfqq lookup hash */
L
Linus Torvalds 已提交
136 137
	struct hlist_node cfq_hash;
	/* hash key */
138
	unsigned int key;
L
Linus Torvalds 已提交
139 140 141 142 143
	/* on either rr or empty list of cfqd */
	struct list_head cfq_list;
	/* sorted list of pending requests */
	struct rb_root sort_list;
	/* if fifo isn't expired, next request to serve */
J
Jens Axboe 已提交
144
	struct request *next_rq;
L
Linus Torvalds 已提交
145 146 147 148 149
	/* requests queued in sort_list */
	int queued[2];
	/* currently allocated requests */
	int allocated[2];
	/* fifo list of requests in sort_list */
150
	struct list_head fifo;
L
Linus Torvalds 已提交
151

152 153 154 155
	unsigned long slice_start;
	unsigned long slice_end;
	unsigned long slice_left;
	unsigned long service_last;
L
Linus Torvalds 已提交
156

J
Jens Axboe 已提交
157 158
	/* number of requests that are on the dispatch list */
	int on_dispatch[2];
159 160 161 162 163

	/* io prio of this group */
	unsigned short ioprio, org_ioprio;
	unsigned short ioprio_class, org_ioprio_class;

J
Jens Axboe 已提交
164 165
	/* various state flags, see below */
	unsigned int flags;
L
Linus Torvalds 已提交
166 167
};

J
Jens Axboe 已提交
168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
enum cfqq_state_flags {
	CFQ_CFQQ_FLAG_on_rr = 0,
	CFQ_CFQQ_FLAG_wait_request,
	CFQ_CFQQ_FLAG_must_alloc,
	CFQ_CFQQ_FLAG_must_alloc_slice,
	CFQ_CFQQ_FLAG_must_dispatch,
	CFQ_CFQQ_FLAG_fifo_expire,
	CFQ_CFQQ_FLAG_idle_window,
	CFQ_CFQQ_FLAG_prio_changed,
};

#define CFQ_CFQQ_FNS(name)						\
static inline void cfq_mark_cfqq_##name(struct cfq_queue *cfqq)		\
{									\
	cfqq->flags |= (1 << CFQ_CFQQ_FLAG_##name);			\
}									\
static inline void cfq_clear_cfqq_##name(struct cfq_queue *cfqq)	\
{									\
	cfqq->flags &= ~(1 << CFQ_CFQQ_FLAG_##name);			\
}									\
static inline int cfq_cfqq_##name(const struct cfq_queue *cfqq)		\
{									\
	return (cfqq->flags & (1 << CFQ_CFQQ_FLAG_##name)) != 0;	\
}

CFQ_CFQQ_FNS(on_rr);
CFQ_CFQQ_FNS(wait_request);
CFQ_CFQQ_FNS(must_alloc);
CFQ_CFQQ_FNS(must_alloc_slice);
CFQ_CFQQ_FNS(must_dispatch);
CFQ_CFQQ_FNS(fifo_expire);
CFQ_CFQQ_FNS(idle_window);
CFQ_CFQQ_FNS(prio_changed);
#undef CFQ_CFQQ_FNS

static struct cfq_queue *cfq_find_cfq_hash(struct cfq_data *, unsigned int, unsigned short);
J
Jens Axboe 已提交
204
static void cfq_dispatch_insert(request_queue_t *, struct request *);
205
static struct cfq_queue *cfq_get_queue(struct cfq_data *cfqd, unsigned int key, struct task_struct *tsk, gfp_t gfp_mask);
L
Linus Torvalds 已提交
206

A
Andrew Morton 已提交
207 208 209 210 211 212
/*
 * scheduler run of queue, if there are requests pending and no one in the
 * driver that will restart queueing
 */
static inline void cfq_schedule_dispatch(struct cfq_data *cfqd)
{
213
	if (cfqd->busy_queues)
A
Andrew Morton 已提交
214 215 216 217 218 219 220
		kblockd_schedule_work(&cfqd->unplug_work);
}

static int cfq_queue_empty(request_queue_t *q)
{
	struct cfq_data *cfqd = q->elevator->elevator_data;

221
	return !cfqd->busy_queues;
A
Andrew Morton 已提交
222 223
}

224 225
static inline pid_t cfq_queue_pid(struct task_struct *task, int rw)
{
J
Jens Axboe 已提交
226
	if (rw == READ || rw == WRITE_SYNC)
227 228 229 230 231
		return task->pid;

	return CFQ_KEY_ASYNC;
}

L
Linus Torvalds 已提交
232
/*
J
Jens Axboe 已提交
233
 * Lifted from AS - choose which of rq1 and rq2 that is best served now.
L
Linus Torvalds 已提交
234
 * We choose the request that is closest to the head right now. Distance
235
 * behind the head is penalized and only allowed to a certain extent.
L
Linus Torvalds 已提交
236
 */
J
Jens Axboe 已提交
237 238
static struct request *
cfq_choose_req(struct cfq_data *cfqd, struct request *rq1, struct request *rq2)
L
Linus Torvalds 已提交
239 240 241
{
	sector_t last, s1, s2, d1 = 0, d2 = 0;
	unsigned long back_max;
242 243 244
#define CFQ_RQ1_WRAP	0x01 /* request 1 wraps */
#define CFQ_RQ2_WRAP	0x02 /* request 2 wraps */
	unsigned wrap = 0; /* bit mask: requests behind the disk head? */
L
Linus Torvalds 已提交
245

J
Jens Axboe 已提交
246 247 248 249
	if (rq1 == NULL || rq1 == rq2)
		return rq2;
	if (rq2 == NULL)
		return rq1;
J
Jens Axboe 已提交
250

J
Jens Axboe 已提交
251 252 253 254
	if (rq_is_sync(rq1) && !rq_is_sync(rq2))
		return rq1;
	else if (rq_is_sync(rq2) && !rq_is_sync(rq1))
		return rq2;
L
Linus Torvalds 已提交
255

J
Jens Axboe 已提交
256 257
	s1 = rq1->sector;
	s2 = rq2->sector;
L
Linus Torvalds 已提交
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275

	last = cfqd->last_sector;

	/*
	 * by definition, 1KiB is 2 sectors
	 */
	back_max = cfqd->cfq_back_max * 2;

	/*
	 * Strict one way elevator _except_ in the case where we allow
	 * short backward seeks which are biased as twice the cost of a
	 * similar forward seek.
	 */
	if (s1 >= last)
		d1 = s1 - last;
	else if (s1 + back_max >= last)
		d1 = (last - s1) * cfqd->cfq_back_penalty;
	else
276
		wrap |= CFQ_RQ1_WRAP;
L
Linus Torvalds 已提交
277 278 279 280 281 282

	if (s2 >= last)
		d2 = s2 - last;
	else if (s2 + back_max >= last)
		d2 = (last - s2) * cfqd->cfq_back_penalty;
	else
283
		wrap |= CFQ_RQ2_WRAP;
L
Linus Torvalds 已提交
284 285

	/* Found required data */
286 287 288 289 290 291

	/*
	 * By doing switch() on the bit mask "wrap" we avoid having to
	 * check two variables for all permutations: --> faster!
	 */
	switch (wrap) {
J
Jens Axboe 已提交
292
	case 0: /* common case for CFQ: rq1 and rq2 not wrapped */
293
		if (d1 < d2)
J
Jens Axboe 已提交
294
			return rq1;
295
		else if (d2 < d1)
J
Jens Axboe 已提交
296
			return rq2;
297 298
		else {
			if (s1 >= s2)
J
Jens Axboe 已提交
299
				return rq1;
300
			else
J
Jens Axboe 已提交
301
				return rq2;
302
		}
L
Linus Torvalds 已提交
303

304
	case CFQ_RQ2_WRAP:
J
Jens Axboe 已提交
305
		return rq1;
306
	case CFQ_RQ1_WRAP:
J
Jens Axboe 已提交
307 308
		return rq2;
	case (CFQ_RQ1_WRAP|CFQ_RQ2_WRAP): /* both rqs wrapped */
309 310 311 312 313 314 315 316
	default:
		/*
		 * Since both rqs are wrapped,
		 * start with the one that's further behind head
		 * (--> only *one* back seek required),
		 * since back seek takes more time than forward.
		 */
		if (s1 <= s2)
J
Jens Axboe 已提交
317
			return rq1;
L
Linus Torvalds 已提交
318
		else
J
Jens Axboe 已提交
319
			return rq2;
L
Linus Torvalds 已提交
320 321 322 323 324 325
	}
}

/*
 * would be nice to take fifo expire time into account as well
 */
J
Jens Axboe 已提交
326 327 328
static struct request *
cfq_find_next_rq(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		  struct request *last)
L
Linus Torvalds 已提交
329
{
330 331
	struct rb_node *rbnext = rb_next(&last->rb_node);
	struct rb_node *rbprev = rb_prev(&last->rb_node);
J
Jens Axboe 已提交
332
	struct request *next = NULL, *prev = NULL;
L
Linus Torvalds 已提交
333

334
	BUG_ON(RB_EMPTY_NODE(&last->rb_node));
L
Linus Torvalds 已提交
335 336

	if (rbprev)
J
Jens Axboe 已提交
337
		prev = rb_entry_rq(rbprev);
L
Linus Torvalds 已提交
338

339
	if (rbnext)
J
Jens Axboe 已提交
340
		next = rb_entry_rq(rbnext);
341 342 343
	else {
		rbnext = rb_first(&cfqq->sort_list);
		if (rbnext && rbnext != &last->rb_node)
J
Jens Axboe 已提交
344
			next = rb_entry_rq(rbnext);
345
	}
L
Linus Torvalds 已提交
346

347
	return cfq_choose_req(cfqd, next, prev);
L
Linus Torvalds 已提交
348 349
}

350
static void cfq_resort_rr_list(struct cfq_queue *cfqq, int preempted)
L
Linus Torvalds 已提交
351
{
352 353
	struct cfq_data *cfqd = cfqq->cfqd;
	struct list_head *list, *entry;
L
Linus Torvalds 已提交
354

J
Jens Axboe 已提交
355
	BUG_ON(!cfq_cfqq_on_rr(cfqq));
L
Linus Torvalds 已提交
356

357
	list_del(&cfqq->cfq_list);
L
Linus Torvalds 已提交
358

359 360 361 362 363 364 365 366 367 368 369 370
	if (cfq_class_rt(cfqq))
		list = &cfqd->cur_rr;
	else if (cfq_class_idle(cfqq))
		list = &cfqd->idle_rr;
	else {
		/*
		 * if cfqq has requests in flight, don't allow it to be
		 * found in cfq_set_active_queue before it has finished them.
		 * this is done to increase fairness between a process that
		 * has lots of io pending vs one that only generates one
		 * sporadically or synchronously
		 */
J
Jens Axboe 已提交
371
		if (cfq_cfqq_dispatched(cfqq))
372 373 374
			list = &cfqd->busy_rr;
		else
			list = &cfqd->rr_list[cfqq->ioprio];
L
Linus Torvalds 已提交
375 376
	}

377 378
	/*
	 * if queue was preempted, just add to front to be fair. busy_rr
379
	 * isn't sorted, but insert at the back for fairness.
380 381
	 */
	if (preempted || list == &cfqd->busy_rr) {
382 383 384 385
		if (preempted)
			list = list->prev;

		list_add_tail(&cfqq->cfq_list, list);
L
Linus Torvalds 已提交
386
		return;
387
	}
L
Linus Torvalds 已提交
388 389

	/*
390
	 * sort by when queue was last serviced
L
Linus Torvalds 已提交
391
	 */
392 393
	entry = list;
	while ((entry = entry->prev) != list) {
L
Linus Torvalds 已提交
394 395
		struct cfq_queue *__cfqq = list_entry_cfqq(entry);

396 397 398
		if (!__cfqq->service_last)
			break;
		if (time_before(__cfqq->service_last, cfqq->service_last))
L
Linus Torvalds 已提交
399 400 401 402 403 404 405 406
			break;
	}

	list_add(&cfqq->cfq_list, entry);
}

/*
 * add to busy list of queues for service, trying to be fair in ordering
407
 * the pending list according to last request service
L
Linus Torvalds 已提交
408 409
 */
static inline void
410
cfq_add_cfqq_rr(struct cfq_data *cfqd, struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
411
{
J
Jens Axboe 已提交
412 413
	BUG_ON(cfq_cfqq_on_rr(cfqq));
	cfq_mark_cfqq_on_rr(cfqq);
L
Linus Torvalds 已提交
414 415
	cfqd->busy_queues++;

416
	cfq_resort_rr_list(cfqq, 0);
L
Linus Torvalds 已提交
417 418 419 420 421
}

static inline void
cfq_del_cfqq_rr(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
J
Jens Axboe 已提交
422 423
	BUG_ON(!cfq_cfqq_on_rr(cfqq));
	cfq_clear_cfqq_on_rr(cfqq);
424
	list_move(&cfqq->cfq_list, &cfqd->empty_list);
L
Linus Torvalds 已提交
425 426 427 428 429 430 431 432

	BUG_ON(!cfqd->busy_queues);
	cfqd->busy_queues--;
}

/*
 * rb tree support functions
 */
J
Jens Axboe 已提交
433
static inline void cfq_del_rq_rb(struct request *rq)
L
Linus Torvalds 已提交
434
{
J
Jens Axboe 已提交
435
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
436
	struct cfq_data *cfqd = cfqq->cfqd;
J
Jens Axboe 已提交
437
	const int sync = rq_is_sync(rq);
L
Linus Torvalds 已提交
438

439 440
	BUG_ON(!cfqq->queued[sync]);
	cfqq->queued[sync]--;
L
Linus Torvalds 已提交
441

J
Jens Axboe 已提交
442
	elv_rb_del(&cfqq->sort_list, rq);
L
Linus Torvalds 已提交
443

444
	if (cfq_cfqq_on_rr(cfqq) && RB_EMPTY_ROOT(&cfqq->sort_list))
445
		cfq_del_cfqq_rr(cfqd, cfqq);
L
Linus Torvalds 已提交
446 447
}

J
Jens Axboe 已提交
448
static void cfq_add_rq_rb(struct request *rq)
L
Linus Torvalds 已提交
449
{
J
Jens Axboe 已提交
450
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
L
Linus Torvalds 已提交
451
	struct cfq_data *cfqd = cfqq->cfqd;
452
	struct request *__alias;
L
Linus Torvalds 已提交
453

454
	cfqq->queued[rq_is_sync(rq)]++;
L
Linus Torvalds 已提交
455 456 457 458 459

	/*
	 * looks a little odd, but the first insert might return an alias.
	 * if that happens, put the alias on the dispatch list
	 */
460
	while ((__alias = elv_rb_add(&cfqq->sort_list, rq)) != NULL)
J
Jens Axboe 已提交
461
		cfq_dispatch_insert(cfqd->queue, __alias);
L
Linus Torvalds 已提交
462 463 464
}

static inline void
J
Jens Axboe 已提交
465
cfq_reposition_rq_rb(struct cfq_queue *cfqq, struct request *rq)
L
Linus Torvalds 已提交
466
{
467 468
	elv_rb_del(&cfqq->sort_list, rq);
	cfqq->queued[rq_is_sync(rq)]--;
J
Jens Axboe 已提交
469
	cfq_add_rq_rb(rq);
L
Linus Torvalds 已提交
470 471
}

472 473
static struct request *
cfq_find_rq_fmerge(struct cfq_data *cfqd, struct bio *bio)
L
Linus Torvalds 已提交
474
{
475 476 477
	struct task_struct *tsk = current;
	pid_t key = cfq_queue_pid(tsk, bio_data_dir(bio));
	struct cfq_queue *cfqq;
L
Linus Torvalds 已提交
478

479
	cfqq = cfq_find_cfq_hash(cfqd, key, tsk->ioprio);
480 481 482
	if (cfqq) {
		sector_t sector = bio->bi_sector + bio_sectors(bio);

483
		return elv_rb_find(&cfqq->sort_list, sector);
484
	}
L
Linus Torvalds 已提交
485 486 487 488

	return NULL;
}

489
static void cfq_activate_request(request_queue_t *q, struct request *rq)
L
Linus Torvalds 已提交
490
{
491
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
492

493
	cfqd->rq_in_driver++;
494 495 496 497 498 499 500 501 502

	/*
	 * If the depth is larger 1, it really could be queueing. But lets
	 * make the mark a little higher - idling could still be good for
	 * low queueing, and a low queueing number could also just indicate
	 * a SCSI mid layer like behaviour where limit+1 is often seen.
	 */
	if (!cfqd->hw_tag && cfqd->rq_in_driver > 4)
		cfqd->hw_tag = 1;
L
Linus Torvalds 已提交
503 504
}

505
static void cfq_deactivate_request(request_queue_t *q, struct request *rq)
L
Linus Torvalds 已提交
506
{
507 508 509 510
	struct cfq_data *cfqd = q->elevator->elevator_data;

	WARN_ON(!cfqd->rq_in_driver);
	cfqd->rq_in_driver--;
L
Linus Torvalds 已提交
511 512
}

513
static void cfq_remove_request(struct request *rq)
L
Linus Torvalds 已提交
514
{
J
Jens Axboe 已提交
515
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
516

J
Jens Axboe 已提交
517 518
	if (cfqq->next_rq == rq)
		cfqq->next_rq = cfq_find_next_rq(cfqq->cfqd, cfqq, rq);
L
Linus Torvalds 已提交
519

520
	list_del_init(&rq->queuelist);
J
Jens Axboe 已提交
521
	cfq_del_rq_rb(rq);
L
Linus Torvalds 已提交
522 523 524 525 526 527 528 529
}

static int
cfq_merge(request_queue_t *q, struct request **req, struct bio *bio)
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
	struct request *__rq;

530
	__rq = cfq_find_rq_fmerge(cfqd, bio);
531
	if (__rq && elv_rq_merge_ok(__rq, bio)) {
532 533
		*req = __rq;
		return ELEVATOR_FRONT_MERGE;
L
Linus Torvalds 已提交
534 535 536 537 538
	}

	return ELEVATOR_NO_MERGE;
}

539 540
static void cfq_merged_request(request_queue_t *q, struct request *req,
			       int type)
L
Linus Torvalds 已提交
541
{
542
	if (type == ELEVATOR_FRONT_MERGE) {
J
Jens Axboe 已提交
543
		struct cfq_queue *cfqq = RQ_CFQQ(req);
L
Linus Torvalds 已提交
544

J
Jens Axboe 已提交
545
		cfq_reposition_rq_rb(cfqq, req);
L
Linus Torvalds 已提交
546 547 548 549 550 551 552
	}
}

static void
cfq_merged_requests(request_queue_t *q, struct request *rq,
		    struct request *next)
{
553 554 555 556 557 558 559
	/*
	 * reposition in fifo if next is older than rq
	 */
	if (!list_empty(&rq->queuelist) && !list_empty(&next->queuelist) &&
	    time_before(next->start_time, rq->start_time))
		list_move(&rq->queuelist, &next->queuelist);

560
	cfq_remove_request(next);
561 562 563 564 565 566 567 568 569 570 571 572 573 574
}

static inline void
__cfq_set_active_queue(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	if (cfqq) {
		/*
		 * stop potential idle class queues waiting service
		 */
		del_timer(&cfqd->idle_class_timer);

		cfqq->slice_start = jiffies;
		cfqq->slice_end = 0;
		cfqq->slice_left = 0;
J
Jens Axboe 已提交
575 576
		cfq_clear_cfqq_must_alloc_slice(cfqq);
		cfq_clear_cfqq_fifo_expire(cfqq);
577 578 579 580 581
	}

	cfqd->active_queue = cfqq;
}

582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
/*
 * current cfqq expired its slice (or was too idle), select new one
 */
static void
__cfq_slice_expired(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		    int preempted)
{
	unsigned long now = jiffies;

	if (cfq_cfqq_wait_request(cfqq))
		del_timer(&cfqd->idle_slice_timer);

	if (!preempted && !cfq_cfqq_dispatched(cfqq)) {
		cfqq->service_last = now;
		cfq_schedule_dispatch(cfqd);
	}

	cfq_clear_cfqq_must_dispatch(cfqq);
	cfq_clear_cfqq_wait_request(cfqq);

	/*
	 * store what was left of this slice, if the queue idled out
	 * or was preempted
	 */
	if (time_after(cfqq->slice_end, now))
		cfqq->slice_left = cfqq->slice_end - now;
	else
		cfqq->slice_left = 0;

	if (cfq_cfqq_on_rr(cfqq))
		cfq_resort_rr_list(cfqq, preempted);

	if (cfqq == cfqd->active_queue)
		cfqd->active_queue = NULL;

	if (cfqd->active_cic) {
		put_io_context(cfqd->active_cic->ioc);
		cfqd->active_cic = NULL;
	}

	cfqd->dispatch_slice = 0;
}

static inline void cfq_slice_expired(struct cfq_data *cfqd, int preempted)
{
	struct cfq_queue *cfqq = cfqd->active_queue;

	if (cfqq)
		__cfq_slice_expired(cfqd, cfqq, preempted);
}

633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666
/*
 * 0
 * 0,1
 * 0,1,2
 * 0,1,2,3
 * 0,1,2,3,4
 * 0,1,2,3,4,5
 * 0,1,2,3,4,5,6
 * 0,1,2,3,4,5,6,7
 */
static int cfq_get_next_prio_level(struct cfq_data *cfqd)
{
	int prio, wrap;

	prio = -1;
	wrap = 0;
	do {
		int p;

		for (p = cfqd->cur_prio; p <= cfqd->cur_end_prio; p++) {
			if (!list_empty(&cfqd->rr_list[p])) {
				prio = p;
				break;
			}
		}

		if (prio != -1)
			break;
		cfqd->cur_prio = 0;
		if (++cfqd->cur_end_prio == CFQ_PRIO_LISTS) {
			cfqd->cur_end_prio = 0;
			if (wrap)
				break;
			wrap = 1;
L
Linus Torvalds 已提交
667
		}
668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684
	} while (1);

	if (unlikely(prio == -1))
		return -1;

	BUG_ON(prio >= CFQ_PRIO_LISTS);

	list_splice_init(&cfqd->rr_list[prio], &cfqd->cur_rr);

	cfqd->cur_prio = prio + 1;
	if (cfqd->cur_prio > cfqd->cur_end_prio) {
		cfqd->cur_end_prio = cfqd->cur_prio;
		cfqd->cur_prio = 0;
	}
	if (cfqd->cur_end_prio == CFQ_PRIO_LISTS) {
		cfqd->cur_prio = 0;
		cfqd->cur_end_prio = 0;
L
Linus Torvalds 已提交
685 686
	}

687 688 689
	return prio;
}

J
Jens Axboe 已提交
690
static struct cfq_queue *cfq_set_active_queue(struct cfq_data *cfqd)
691
{
692
	struct cfq_queue *cfqq = NULL;
693

694 695 696 697 698 699
	if (!list_empty(&cfqd->cur_rr) || cfq_get_next_prio_level(cfqd) != -1) {
		/*
		 * if current list is non-empty, grab first entry. if it is
		 * empty, get next prio level and grab first entry then if any
		 * are spliced
		 */
700
		cfqq = list_entry_cfqq(cfqd->cur_rr.next);
701 702 703 704 705
	} else if (!list_empty(&cfqd->busy_rr)) {
		/*
		 * If no new queues are available, check if the busy list has
		 * some before falling back to idle io.
		 */
706
		cfqq = list_entry_cfqq(cfqd->busy_rr.next);
707 708 709 710 711 712
	} else if (!list_empty(&cfqd->idle_rr)) {
		/*
		 * if we have idle queues and no rt or be queues had pending
		 * requests, either allow immediate service if the grace period
		 * has passed or arm the idle grace timer
		 */
713 714 715 716 717 718 719 720 721
		unsigned long end = cfqd->last_end_request + CFQ_IDLE_GRACE;

		if (time_after_eq(jiffies, end))
			cfqq = list_entry_cfqq(cfqd->idle_rr.next);
		else
			mod_timer(&cfqd->idle_class_timer, end);
	}

	__cfq_set_active_queue(cfqd, cfqq);
J
Jens Axboe 已提交
722
	return cfqq;
723 724
}

725 726
#define CIC_SEEKY(cic) ((cic)->seek_mean > (128 * 1024))

727 728 729
static int cfq_arm_slice_timer(struct cfq_data *cfqd, struct cfq_queue *cfqq)

{
730
	struct cfq_io_context *cic;
731 732
	unsigned long sl;

733
	WARN_ON(!RB_EMPTY_ROOT(&cfqq->sort_list));
734 735 736 737 738 739 740
	WARN_ON(cfqq != cfqd->active_queue);

	/*
	 * idle is disabled, either manually or by past process history
	 */
	if (!cfqd->cfq_slice_idle)
		return 0;
J
Jens Axboe 已提交
741
	if (!cfq_cfqq_idle_window(cfqq))
742 743 744 745
		return 0;
	/*
	 * task has exited, don't wait
	 */
746 747
	cic = cfqd->active_cic;
	if (!cic || !cic->ioc->task)
748 749
		return 0;

J
Jens Axboe 已提交
750 751
	cfq_mark_cfqq_must_dispatch(cfqq);
	cfq_mark_cfqq_wait_request(cfqq);
752

753
	sl = min(cfqq->slice_end - 1, (unsigned long) cfqd->cfq_slice_idle);
754 755 756 757 758 759

	/*
	 * we don't want to idle for seeks, but we do want to allow
	 * fair distribution of slice time for a process doing back-to-back
	 * seeks. so allow a little bit of time for him to submit a new rq
	 */
760
	if (sample_valid(cic->seek_samples) && CIC_SEEKY(cic))
761
		sl = min(sl, msecs_to_jiffies(2));
762

763
	mod_timer(&cfqd->idle_slice_timer, jiffies + sl);
764
	return 1;
L
Linus Torvalds 已提交
765 766
}

J
Jens Axboe 已提交
767
static void cfq_dispatch_insert(request_queue_t *q, struct request *rq)
L
Linus Torvalds 已提交
768 769
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
770
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
771

772 773 774
	cfq_remove_request(rq);
	cfqq->on_dispatch[rq_is_sync(rq)]++;
	elv_dispatch_sort(q, rq);
775 776 777

	rq = list_entry(q->queue_head.prev, struct request, queuelist);
	cfqd->last_sector = rq->sector + rq->nr_sectors;
L
Linus Torvalds 已提交
778 779 780 781 782
}

/*
 * return expired entry, or NULL to just start from scratch in rbtree
 */
J
Jens Axboe 已提交
783
static inline struct request *cfq_check_fifo(struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
784 785
{
	struct cfq_data *cfqd = cfqq->cfqd;
786
	struct request *rq;
787
	int fifo;
L
Linus Torvalds 已提交
788

J
Jens Axboe 已提交
789
	if (cfq_cfqq_fifo_expire(cfqq))
L
Linus Torvalds 已提交
790
		return NULL;
791 792
	if (list_empty(&cfqq->fifo))
		return NULL;
L
Linus Torvalds 已提交
793

794 795
	fifo = cfq_cfqq_class_sync(cfqq);
	rq = rq_entry_fifo(cfqq->fifo.next);
L
Linus Torvalds 已提交
796

797 798 799
	if (time_after(jiffies, rq->start_time + cfqd->cfq_fifo_expire[fifo])) {
		cfq_mark_cfqq_fifo_expire(cfqq);
		return rq;
L
Linus Torvalds 已提交
800 801 802 803 804 805
	}

	return NULL;
}

/*
J
Jens Axboe 已提交
806 807 808
 * Scale schedule slice based on io priority. Use the sync time slice only
 * if a queue is marked sync and has sync io queued. A sync queue with async
 * io only, should not get full sync slice length.
L
Linus Torvalds 已提交
809
 */
810 811 812 813 814 815 816 817 818 819
static inline int
cfq_prio_to_slice(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	const int base_slice = cfqd->cfq_slice[cfq_cfqq_sync(cfqq)];

	WARN_ON(cfqq->ioprio >= IOPRIO_BE_NR);

	return base_slice + (base_slice/CFQ_SLICE_SCALE * (4 - cfqq->ioprio));
}

L
Linus Torvalds 已提交
820
static inline void
821
cfq_set_prio_slice(struct cfq_data *cfqd, struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
822
{
823 824
	cfqq->slice_end = cfq_prio_to_slice(cfqd, cfqq) + jiffies;
}
L
Linus Torvalds 已提交
825

826 827 828 829
static inline int
cfq_prio_to_maxrq(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	const int base_rq = cfqd->cfq_slice_async_rq;
L
Linus Torvalds 已提交
830

831
	WARN_ON(cfqq->ioprio >= IOPRIO_BE_NR);
L
Linus Torvalds 已提交
832

833
	return 2 * (base_rq + base_rq * (CFQ_PRIO_LISTS - 1 - cfqq->ioprio));
L
Linus Torvalds 已提交
834 835
}

836 837 838
/*
 * get next queue for service
 */
839
static struct cfq_queue *cfq_select_queue(struct cfq_data *cfqd)
L
Linus Torvalds 已提交
840
{
841
	unsigned long now = jiffies;
L
Linus Torvalds 已提交
842 843
	struct cfq_queue *cfqq;

844 845 846
	cfqq = cfqd->active_queue;
	if (!cfqq)
		goto new_queue;
L
Linus Torvalds 已提交
847

848 849 850
	/*
	 * slice has expired
	 */
J
Jens Axboe 已提交
851 852
	if (!cfq_cfqq_must_dispatch(cfqq) && time_after(now, cfqq->slice_end))
		goto expire;
L
Linus Torvalds 已提交
853

854 855 856 857
	/*
	 * if queue has requests, dispatch one. if not, check if
	 * enough slice is left to wait for one
	 */
858
	if (!RB_EMPTY_ROOT(&cfqq->sort_list))
859
		goto keep_queue;
860 861 862 863
	else if (cfq_cfqq_dispatched(cfqq)) {
		cfqq = NULL;
		goto keep_queue;
	} else if (cfq_cfqq_class_sync(cfqq)) {
864 865 866 867
		if (cfq_arm_slice_timer(cfqd, cfqq))
			return NULL;
	}

J
Jens Axboe 已提交
868
expire:
869
	cfq_slice_expired(cfqd, 0);
J
Jens Axboe 已提交
870 871
new_queue:
	cfqq = cfq_set_active_queue(cfqd);
872
keep_queue:
J
Jens Axboe 已提交
873
	return cfqq;
874 875 876 877 878 879 880 881
}

static int
__cfq_dispatch_requests(struct cfq_data *cfqd, struct cfq_queue *cfqq,
			int max_dispatch)
{
	int dispatched = 0;

882
	BUG_ON(RB_EMPTY_ROOT(&cfqq->sort_list));
883 884

	do {
J
Jens Axboe 已提交
885
		struct request *rq;
L
Linus Torvalds 已提交
886 887

		/*
888
		 * follow expired path, else get first next available
L
Linus Torvalds 已提交
889
		 */
J
Jens Axboe 已提交
890 891
		if ((rq = cfq_check_fifo(cfqq)) == NULL)
			rq = cfqq->next_rq;
892 893 894 895

		/*
		 * finally, insert request into driver dispatch list
		 */
J
Jens Axboe 已提交
896
		cfq_dispatch_insert(cfqd->queue, rq);
L
Linus Torvalds 已提交
897

898 899
		cfqd->dispatch_slice++;
		dispatched++;
L
Linus Torvalds 已提交
900

901
		if (!cfqd->active_cic) {
J
Jens Axboe 已提交
902 903
			atomic_inc(&RQ_CIC(rq)->ioc->refcount);
			cfqd->active_cic = RQ_CIC(rq);
904
		}
L
Linus Torvalds 已提交
905

906
		if (RB_EMPTY_ROOT(&cfqq->sort_list))
907 908 909 910 911
			break;

	} while (dispatched < max_dispatch);

	/*
912
	 * if slice end isn't set yet, set it.
913 914 915 916 917 918 919 920 921 922
	 */
	if (!cfqq->slice_end)
		cfq_set_prio_slice(cfqd, cfqq);

	/*
	 * expire an async queue immediately if it has used up its slice. idle
	 * queue always expire after 1 dispatch round.
	 */
	if ((!cfq_cfqq_sync(cfqq) &&
	    cfqd->dispatch_slice >= cfq_prio_to_maxrq(cfqd, cfqq)) ||
923 924
	    cfq_class_idle(cfqq) ||
	    !cfq_cfqq_idle_window(cfqq))
925 926 927 928 929
		cfq_slice_expired(cfqd, 0);

	return dispatched;
}

930 931 932 933
static int
cfq_forced_dispatch_cfqqs(struct list_head *list)
{
	struct cfq_queue *cfqq, *next;
934
	int dispatched;
935

936
	dispatched = 0;
937
	list_for_each_entry_safe(cfqq, next, list, cfq_list) {
J
Jens Axboe 已提交
938 939
		while (cfqq->next_rq) {
			cfq_dispatch_insert(cfqq->cfqd->queue, cfqq->next_rq);
940 941 942 943
			dispatched++;
		}
		BUG_ON(!list_empty(&cfqq->fifo));
	}
944

945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966
	return dispatched;
}

static int
cfq_forced_dispatch(struct cfq_data *cfqd)
{
	int i, dispatched = 0;

	for (i = 0; i < CFQ_PRIO_LISTS; i++)
		dispatched += cfq_forced_dispatch_cfqqs(&cfqd->rr_list[i]);

	dispatched += cfq_forced_dispatch_cfqqs(&cfqd->busy_rr);
	dispatched += cfq_forced_dispatch_cfqqs(&cfqd->cur_rr);
	dispatched += cfq_forced_dispatch_cfqqs(&cfqd->idle_rr);

	cfq_slice_expired(cfqd, 0);

	BUG_ON(cfqd->busy_queues);

	return dispatched;
}

967
static int
968
cfq_dispatch_requests(request_queue_t *q, int force)
969 970
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
971 972
	struct cfq_queue *cfqq, *prev_cfqq;
	int dispatched;
973 974 975 976

	if (!cfqd->busy_queues)
		return 0;

977 978 979
	if (unlikely(force))
		return cfq_forced_dispatch(cfqd);

980 981 982
	dispatched = 0;
	prev_cfqq = NULL;
	while ((cfqq = cfq_select_queue(cfqd)) != NULL) {
983 984
		int max_dispatch;

985 986 987 988 989 990
		/*
		 * Don't repeat dispatch from the previous queue.
		 */
		if (prev_cfqq == cfqq)
			break;

J
Jens Axboe 已提交
991 992
		cfq_clear_cfqq_must_dispatch(cfqq);
		cfq_clear_cfqq_wait_request(cfqq);
993 994
		del_timer(&cfqd->idle_slice_timer);

995 996 997
		max_dispatch = cfqd->cfq_quantum;
		if (cfq_class_idle(cfqq))
			max_dispatch = 1;
L
Linus Torvalds 已提交
998

999 1000 1001 1002 1003 1004 1005 1006 1007 1008
		dispatched += __cfq_dispatch_requests(cfqd, cfqq, max_dispatch);

		/*
		 * If the dispatch cfqq has idling enabled and is still
		 * the active queue, break out.
		 */
		if (cfq_cfqq_idle_window(cfqq) && cfqd->active_queue)
			break;

		prev_cfqq = cfqq;
L
Linus Torvalds 已提交
1009 1010
	}

1011
	return dispatched;
L
Linus Torvalds 已提交
1012 1013 1014
}

/*
J
Jens Axboe 已提交
1015 1016
 * task holds one reference to the queue, dropped when task exits. each rq
 * in-flight on this queue also holds a reference, dropped when rq is freed.
L
Linus Torvalds 已提交
1017 1018 1019 1020 1021
 *
 * queue lock must be held here.
 */
static void cfq_put_queue(struct cfq_queue *cfqq)
{
1022 1023 1024
	struct cfq_data *cfqd = cfqq->cfqd;

	BUG_ON(atomic_read(&cfqq->ref) <= 0);
L
Linus Torvalds 已提交
1025 1026 1027 1028 1029

	if (!atomic_dec_and_test(&cfqq->ref))
		return;

	BUG_ON(rb_first(&cfqq->sort_list));
1030
	BUG_ON(cfqq->allocated[READ] + cfqq->allocated[WRITE]);
J
Jens Axboe 已提交
1031
	BUG_ON(cfq_cfqq_on_rr(cfqq));
L
Linus Torvalds 已提交
1032

1033
	if (unlikely(cfqd->active_queue == cfqq))
J
Jens Axboe 已提交
1034
		__cfq_slice_expired(cfqd, cfqq, 0);
1035

L
Linus Torvalds 已提交
1036 1037 1038 1039 1040 1041 1042 1043
	/*
	 * it's on the empty list and still hashed
	 */
	list_del(&cfqq->cfq_list);
	hlist_del(&cfqq->cfq_hash);
	kmem_cache_free(cfq_pool, cfqq);
}

J
Jens Axboe 已提交
1044
static struct cfq_queue *
J
Jens Axboe 已提交
1045 1046
__cfq_find_cfq_hash(struct cfq_data *cfqd, unsigned int key, unsigned int prio,
		    const int hashval)
L
Linus Torvalds 已提交
1047 1048
{
	struct hlist_head *hash_list = &cfqd->cfq_hash[hashval];
1049 1050
	struct hlist_node *entry;
	struct cfq_queue *__cfqq;
L
Linus Torvalds 已提交
1051

1052
	hlist_for_each_entry(__cfqq, entry, hash_list, cfq_hash) {
A
Al Viro 已提交
1053
		const unsigned short __p = IOPRIO_PRIO_VALUE(__cfqq->org_ioprio_class, __cfqq->org_ioprio);
L
Linus Torvalds 已提交
1054

1055
		if (__cfqq->key == key && (__p == prio || !prio))
L
Linus Torvalds 已提交
1056 1057 1058 1059 1060 1061 1062
			return __cfqq;
	}

	return NULL;
}

static struct cfq_queue *
J
Jens Axboe 已提交
1063
cfq_find_cfq_hash(struct cfq_data *cfqd, unsigned int key, unsigned short prio)
L
Linus Torvalds 已提交
1064
{
J
Jens Axboe 已提交
1065
	return __cfq_find_cfq_hash(cfqd, key, prio, hash_long(key, CFQ_QHASH_SHIFT));
L
Linus Torvalds 已提交
1066 1067
}

1068
static void cfq_free_io_context(struct io_context *ioc)
L
Linus Torvalds 已提交
1069
{
1070
	struct cfq_io_context *__cic;
1071 1072
	struct rb_node *n;
	int freed = 0;
L
Linus Torvalds 已提交
1073

1074 1075 1076
	while ((n = rb_first(&ioc->cic_root)) != NULL) {
		__cic = rb_entry(n, struct cfq_io_context, rb_node);
		rb_erase(&__cic->rb_node, &ioc->cic_root);
1077
		kmem_cache_free(cfq_ioc_pool, __cic);
1078
		freed++;
L
Linus Torvalds 已提交
1079 1080
	}

1081 1082 1083
	elv_ioc_count_mod(ioc_count, -freed);

	if (ioc_gone && !elv_ioc_count_read(ioc_count))
1084
		complete(ioc_gone);
L
Linus Torvalds 已提交
1085 1086
}

1087
static void cfq_exit_cfqq(struct cfq_data *cfqd, struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
1088
{
1089 1090
	if (unlikely(cfqq == cfqd->active_queue))
		__cfq_slice_expired(cfqd, cfqq, 0);
1091

1092 1093
	cfq_put_queue(cfqq);
}
1094

1095 1096 1097
static void __cfq_exit_single_io_context(struct cfq_data *cfqd,
					 struct cfq_io_context *cic)
{
1098 1099 1100 1101
	list_del_init(&cic->queue_list);
	smp_wmb();
	cic->key = NULL;

1102
	if (cic->cfqq[ASYNC]) {
1103
		cfq_exit_cfqq(cfqd, cic->cfqq[ASYNC]);
1104 1105 1106 1107
		cic->cfqq[ASYNC] = NULL;
	}

	if (cic->cfqq[SYNC]) {
1108
		cfq_exit_cfqq(cfqd, cic->cfqq[SYNC]);
1109 1110
		cic->cfqq[SYNC] = NULL;
	}
1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123
}


/*
 * Called with interrupts disabled
 */
static void cfq_exit_single_io_context(struct cfq_io_context *cic)
{
	struct cfq_data *cfqd = cic->key;

	if (cfqd) {
		request_queue_t *q = cfqd->queue;

1124
		spin_lock_irq(q->queue_lock);
1125
		__cfq_exit_single_io_context(cfqd, cic);
1126
		spin_unlock_irq(q->queue_lock);
1127
	}
L
Linus Torvalds 已提交
1128 1129
}

1130
static void cfq_exit_io_context(struct io_context *ioc)
L
Linus Torvalds 已提交
1131
{
1132
	struct cfq_io_context *__cic;
1133
	struct rb_node *n;
1134

L
Linus Torvalds 已提交
1135 1136 1137
	/*
	 * put the reference this task is holding to the various queues
	 */
1138 1139 1140 1141 1142

	n = rb_first(&ioc->cic_root);
	while (n != NULL) {
		__cic = rb_entry(n, struct cfq_io_context, rb_node);

1143
		cfq_exit_single_io_context(__cic);
1144
		n = rb_next(n);
L
Linus Torvalds 已提交
1145 1146 1147
	}
}

1148
static struct cfq_io_context *
A
Al Viro 已提交
1149
cfq_alloc_io_context(struct cfq_data *cfqd, gfp_t gfp_mask)
L
Linus Torvalds 已提交
1150
{
1151
	struct cfq_io_context *cic;
L
Linus Torvalds 已提交
1152

1153
	cic = kmem_cache_alloc_node(cfq_ioc_pool, gfp_mask, cfqd->queue->node);
L
Linus Torvalds 已提交
1154
	if (cic) {
1155
		memset(cic, 0, sizeof(*cic));
1156
		cic->last_end_request = jiffies;
1157
		INIT_LIST_HEAD(&cic->queue_list);
1158 1159
		cic->dtor = cfq_free_io_context;
		cic->exit = cfq_exit_io_context;
1160
		elv_ioc_count_inc(ioc_count);
L
Linus Torvalds 已提交
1161 1162 1163 1164 1165
	}

	return cic;
}

1166 1167 1168 1169 1170
static void cfq_init_prio_data(struct cfq_queue *cfqq)
{
	struct task_struct *tsk = current;
	int ioprio_class;

J
Jens Axboe 已提交
1171
	if (!cfq_cfqq_prio_changed(cfqq))
1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195
		return;

	ioprio_class = IOPRIO_PRIO_CLASS(tsk->ioprio);
	switch (ioprio_class) {
		default:
			printk(KERN_ERR "cfq: bad prio %x\n", ioprio_class);
		case IOPRIO_CLASS_NONE:
			/*
			 * no prio set, place us in the middle of the BE classes
			 */
			cfqq->ioprio = task_nice_ioprio(tsk);
			cfqq->ioprio_class = IOPRIO_CLASS_BE;
			break;
		case IOPRIO_CLASS_RT:
			cfqq->ioprio = task_ioprio(tsk);
			cfqq->ioprio_class = IOPRIO_CLASS_RT;
			break;
		case IOPRIO_CLASS_BE:
			cfqq->ioprio = task_ioprio(tsk);
			cfqq->ioprio_class = IOPRIO_CLASS_BE;
			break;
		case IOPRIO_CLASS_IDLE:
			cfqq->ioprio_class = IOPRIO_CLASS_IDLE;
			cfqq->ioprio = 7;
J
Jens Axboe 已提交
1196
			cfq_clear_cfqq_idle_window(cfqq);
1197 1198 1199 1200 1201 1202 1203 1204 1205 1206
			break;
	}

	/*
	 * keep track of original prio settings in case we have to temporarily
	 * elevate the priority of this queue
	 */
	cfqq->org_ioprio = cfqq->ioprio;
	cfqq->org_ioprio_class = cfqq->ioprio_class;

J
Jens Axboe 已提交
1207
	if (cfq_cfqq_on_rr(cfqq))
1208 1209
		cfq_resort_rr_list(cfqq, 0);

J
Jens Axboe 已提交
1210
	cfq_clear_cfqq_prio_changed(cfqq);
1211 1212
}

1213
static inline void changed_ioprio(struct cfq_io_context *cic)
1214
{
1215 1216
	struct cfq_data *cfqd = cic->key;
	struct cfq_queue *cfqq;
1217

1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
	if (unlikely(!cfqd))
		return;

	spin_lock(cfqd->queue->queue_lock);

	cfqq = cic->cfqq[ASYNC];
	if (cfqq) {
		struct cfq_queue *new_cfqq;
		new_cfqq = cfq_get_queue(cfqd, CFQ_KEY_ASYNC, cic->ioc->task,
					 GFP_ATOMIC);
		if (new_cfqq) {
			cic->cfqq[ASYNC] = new_cfqq;
			cfq_put_queue(cfqq);
		}
1232
	}
1233 1234 1235 1236 1237 1238

	cfqq = cic->cfqq[SYNC];
	if (cfqq)
		cfq_mark_cfqq_prio_changed(cfqq);

	spin_unlock(cfqd->queue->queue_lock);
1239 1240
}

1241
static void cfq_ioc_set_ioprio(struct io_context *ioc)
1242
{
1243
	struct cfq_io_context *cic;
1244
	struct rb_node *n;
1245

1246
	ioc->ioprio_changed = 0;
1247

1248 1249 1250
	n = rb_first(&ioc->cic_root);
	while (n != NULL) {
		cic = rb_entry(n, struct cfq_io_context, rb_node);
1251

1252
		changed_ioprio(cic);
1253 1254
		n = rb_next(n);
	}
1255 1256 1257
}

static struct cfq_queue *
1258
cfq_get_queue(struct cfq_data *cfqd, unsigned int key, struct task_struct *tsk,
A
Al Viro 已提交
1259
	      gfp_t gfp_mask)
1260 1261 1262
{
	const int hashval = hash_long(key, CFQ_QHASH_SHIFT);
	struct cfq_queue *cfqq, *new_cfqq = NULL;
1263
	unsigned short ioprio;
1264 1265

retry:
1266
	ioprio = tsk->ioprio;
J
Jens Axboe 已提交
1267
	cfqq = __cfq_find_cfq_hash(cfqd, key, ioprio, hashval);
1268 1269 1270 1271 1272 1273

	if (!cfqq) {
		if (new_cfqq) {
			cfqq = new_cfqq;
			new_cfqq = NULL;
		} else if (gfp_mask & __GFP_WAIT) {
1274 1275 1276 1277 1278 1279
			/*
			 * Inform the allocator of the fact that we will
			 * just repeat this allocation if it fails, to allow
			 * the allocator to do whatever it needs to attempt to
			 * free memory.
			 */
1280
			spin_unlock_irq(cfqd->queue->queue_lock);
1281
			new_cfqq = kmem_cache_alloc_node(cfq_pool, gfp_mask|__GFP_NOFAIL, cfqd->queue->node);
1282 1283 1284
			spin_lock_irq(cfqd->queue->queue_lock);
			goto retry;
		} else {
1285
			cfqq = kmem_cache_alloc_node(cfq_pool, gfp_mask, cfqd->queue->node);
1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
			if (!cfqq)
				goto out;
		}

		memset(cfqq, 0, sizeof(*cfqq));

		INIT_HLIST_NODE(&cfqq->cfq_hash);
		INIT_LIST_HEAD(&cfqq->cfq_list);
		INIT_LIST_HEAD(&cfqq->fifo);

		cfqq->key = key;
		hlist_add_head(&cfqq->cfq_hash, &cfqd->cfq_hash[hashval]);
		atomic_set(&cfqq->ref, 0);
		cfqq->cfqd = cfqd;
		cfqq->service_last = 0;
		/*
		 * set ->slice_left to allow preemption for a new process
		 */
		cfqq->slice_left = 2 * cfqd->cfq_slice_idle;
1305
		cfq_mark_cfqq_idle_window(cfqq);
J
Jens Axboe 已提交
1306 1307
		cfq_mark_cfqq_prio_changed(cfqq);
		cfq_init_prio_data(cfqq);
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318
	}

	if (new_cfqq)
		kmem_cache_free(cfq_pool, new_cfqq);

	atomic_inc(&cfqq->ref);
out:
	WARN_ON((gfp_mask & __GFP_WAIT) && !cfqq);
	return cfqq;
}

1319 1320 1321
static void
cfq_drop_dead_cic(struct io_context *ioc, struct cfq_io_context *cic)
{
1322
	WARN_ON(!list_empty(&cic->queue_list));
1323 1324
	rb_erase(&cic->rb_node, &ioc->cic_root);
	kmem_cache_free(cfq_ioc_pool, cic);
1325
	elv_ioc_count_dec(ioc_count);
1326 1327
}

1328 1329 1330
static struct cfq_io_context *
cfq_cic_rb_lookup(struct cfq_data *cfqd, struct io_context *ioc)
{
1331
	struct rb_node *n;
1332
	struct cfq_io_context *cic;
1333
	void *k, *key = cfqd;
1334

1335 1336
restart:
	n = ioc->cic_root.rb_node;
1337 1338
	while (n) {
		cic = rb_entry(n, struct cfq_io_context, rb_node);
1339 1340 1341
		/* ->key must be copied to avoid race with cfq_exit_queue() */
		k = cic->key;
		if (unlikely(!k)) {
1342 1343 1344
			cfq_drop_dead_cic(ioc, cic);
			goto restart;
		}
1345

1346
		if (key < k)
1347
			n = n->rb_left;
1348
		else if (key > k)
1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
			n = n->rb_right;
		else
			return cic;
	}

	return NULL;
}

static inline void
cfq_cic_link(struct cfq_data *cfqd, struct io_context *ioc,
	     struct cfq_io_context *cic)
{
1361 1362
	struct rb_node **p;
	struct rb_node *parent;
1363
	struct cfq_io_context *__cic;
1364
	void *k;
1365 1366 1367 1368

	cic->ioc = ioc;
	cic->key = cfqd;

1369 1370 1371
restart:
	parent = NULL;
	p = &ioc->cic_root.rb_node;
1372 1373 1374
	while (*p) {
		parent = *p;
		__cic = rb_entry(parent, struct cfq_io_context, rb_node);
1375 1376 1377
		/* ->key must be copied to avoid race with cfq_exit_queue() */
		k = __cic->key;
		if (unlikely(!k)) {
1378
			cfq_drop_dead_cic(ioc, __cic);
1379 1380
			goto restart;
		}
1381

1382
		if (cic->key < k)
1383
			p = &(*p)->rb_left;
1384
		else if (cic->key > k)
1385 1386 1387 1388 1389 1390 1391
			p = &(*p)->rb_right;
		else
			BUG();
	}

	rb_link_node(&cic->rb_node, parent, p);
	rb_insert_color(&cic->rb_node, &ioc->cic_root);
1392 1393

	spin_lock_irq(cfqd->queue->queue_lock);
1394
	list_add(&cic->queue_list, &cfqd->cic_list);
1395
	spin_unlock_irq(cfqd->queue->queue_lock);
1396 1397
}

L
Linus Torvalds 已提交
1398 1399 1400
/*
 * Setup general io context and cfq io context. There can be several cfq
 * io contexts per general io context, if this process is doing io to more
1401
 * than one device managed by cfq.
L
Linus Torvalds 已提交
1402 1403
 */
static struct cfq_io_context *
1404
cfq_get_io_context(struct cfq_data *cfqd, gfp_t gfp_mask)
L
Linus Torvalds 已提交
1405
{
1406
	struct io_context *ioc = NULL;
L
Linus Torvalds 已提交
1407 1408
	struct cfq_io_context *cic;

1409
	might_sleep_if(gfp_mask & __GFP_WAIT);
L
Linus Torvalds 已提交
1410

1411
	ioc = get_io_context(gfp_mask, cfqd->queue->node);
L
Linus Torvalds 已提交
1412 1413 1414
	if (!ioc)
		return NULL;

1415 1416 1417
	cic = cfq_cic_rb_lookup(cfqd, ioc);
	if (cic)
		goto out;
L
Linus Torvalds 已提交
1418

1419 1420 1421
	cic = cfq_alloc_io_context(cfqd, gfp_mask);
	if (cic == NULL)
		goto err;
L
Linus Torvalds 已提交
1422

1423
	cfq_cic_link(cfqd, ioc, cic);
L
Linus Torvalds 已提交
1424
out:
1425 1426 1427 1428
	smp_read_barrier_depends();
	if (unlikely(ioc->ioprio_changed))
		cfq_ioc_set_ioprio(ioc);

L
Linus Torvalds 已提交
1429 1430 1431 1432 1433 1434
	return cic;
err:
	put_io_context(ioc);
	return NULL;
}

1435 1436
static void
cfq_update_io_thinktime(struct cfq_data *cfqd, struct cfq_io_context *cic)
L
Linus Torvalds 已提交
1437
{
1438
	unsigned long elapsed, ttime;
L
Linus Torvalds 已提交
1439

1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
	/*
	 * if this context already has stuff queued, thinktime is from
	 * last queue not last end
	 */
#if 0
	if (time_after(cic->last_end_request, cic->last_queue))
		elapsed = jiffies - cic->last_end_request;
	else
		elapsed = jiffies - cic->last_queue;
#else
		elapsed = jiffies - cic->last_end_request;
#endif
L
Linus Torvalds 已提交
1452

1453
	ttime = min(elapsed, 2UL * cfqd->cfq_slice_idle);
1454

1455 1456 1457 1458
	cic->ttime_samples = (7*cic->ttime_samples + 256) / 8;
	cic->ttime_total = (7*cic->ttime_total + 256*ttime) / 8;
	cic->ttime_mean = (cic->ttime_total + 128) / cic->ttime_samples;
}
L
Linus Torvalds 已提交
1459

1460 1461
static void
cfq_update_io_seektime(struct cfq_data *cfqd, struct cfq_io_context *cic,
J
Jens Axboe 已提交
1462
		       struct request *rq)
1463 1464 1465 1466
{
	sector_t sdist;
	u64 total;

J
Jens Axboe 已提交
1467 1468
	if (cic->last_request_pos < rq->sector)
		sdist = rq->sector - cic->last_request_pos;
1469
	else
J
Jens Axboe 已提交
1470
		sdist = cic->last_request_pos - rq->sector;
1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486

	/*
	 * Don't allow the seek distance to get too large from the
	 * odd fragment, pagein, etc
	 */
	if (cic->seek_samples <= 60) /* second&third seek */
		sdist = min(sdist, (cic->seek_mean * 4) + 2*1024*1024);
	else
		sdist = min(sdist, (cic->seek_mean * 4)	+ 2*1024*64);

	cic->seek_samples = (7*cic->seek_samples + 256) / 8;
	cic->seek_total = (7*cic->seek_total + (u64)256*sdist) / 8;
	total = cic->seek_total + (cic->seek_samples/2);
	do_div(total, cic->seek_samples);
	cic->seek_mean = (sector_t)total;
}
L
Linus Torvalds 已提交
1487

1488 1489 1490 1491 1492 1493 1494 1495
/*
 * Disable idle window if the process thinks too long or seeks so much that
 * it doesn't matter
 */
static void
cfq_update_idle_window(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		       struct cfq_io_context *cic)
{
J
Jens Axboe 已提交
1496
	int enable_idle = cfq_cfqq_idle_window(cfqq);
L
Linus Torvalds 已提交
1497

1498 1499
	if (!cic->ioc->task || !cfqd->cfq_slice_idle ||
	    (cfqd->hw_tag && CIC_SEEKY(cic)))
1500 1501 1502 1503 1504 1505
		enable_idle = 0;
	else if (sample_valid(cic->ttime_samples)) {
		if (cic->ttime_mean > cfqd->cfq_slice_idle)
			enable_idle = 0;
		else
			enable_idle = 1;
L
Linus Torvalds 已提交
1506 1507
	}

J
Jens Axboe 已提交
1508 1509 1510 1511
	if (enable_idle)
		cfq_mark_cfqq_idle_window(cfqq);
	else
		cfq_clear_cfqq_idle_window(cfqq);
1512
}
L
Linus Torvalds 已提交
1513

1514 1515 1516 1517 1518 1519 1520

/*
 * Check if new_cfqq should preempt the currently active queue. Return 0 for
 * no or if we aren't sure, a 1 will cause a preempt.
 */
static int
cfq_should_preempt(struct cfq_data *cfqd, struct cfq_queue *new_cfqq,
J
Jens Axboe 已提交
1521
		   struct request *rq)
1522 1523 1524 1525 1526 1527 1528
{
	struct cfq_queue *cfqq = cfqd->active_queue;

	if (cfq_class_idle(new_cfqq))
		return 0;

	if (!cfqq)
1529
		return 0;
1530 1531 1532

	if (cfq_class_idle(cfqq))
		return 1;
J
Jens Axboe 已提交
1533
	if (!cfq_cfqq_wait_request(new_cfqq))
1534 1535 1536 1537 1538 1539
		return 0;
	/*
	 * if it doesn't have slice left, forget it
	 */
	if (new_cfqq->slice_left < cfqd->cfq_slice_idle)
		return 0;
J
Jens Axboe 已提交
1540
	if (rq_is_sync(rq) && !cfq_cfqq_sync(cfqq))
1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560
		return 1;

	return 0;
}

/*
 * cfqq preempts the active queue. if we allowed preempt with no slice left,
 * let it have half of its nominal slice.
 */
static void cfq_preempt_queue(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	struct cfq_queue *__cfqq, *next;

	list_for_each_entry_safe(__cfqq, next, &cfqd->cur_rr, cfq_list)
		cfq_resort_rr_list(__cfqq, 1);

	if (!cfqq->slice_left)
		cfqq->slice_left = cfq_prio_to_slice(cfqd, cfqq) / 2;

	cfqq->slice_end = cfqq->slice_left + jiffies;
1561
	cfq_slice_expired(cfqd, 1);
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
	__cfq_set_active_queue(cfqd, cfqq);
}

/*
 * should really be a ll_rw_blk.c helper
 */
static void cfq_start_queueing(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	request_queue_t *q = cfqd->queue;

	if (!blk_queue_plugged(q))
		q->request_fn(q);
	else
		__generic_unplug_device(q);
}

/*
J
Jens Axboe 已提交
1579
 * Called when a new fs request (rq) is added (to cfqq). Check if there's
1580 1581 1582
 * something we should do about it
 */
static void
J
Jens Axboe 已提交
1583 1584
cfq_rq_enqueued(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		struct request *rq)
1585
{
J
Jens Axboe 已提交
1586
	struct cfq_io_context *cic = RQ_CIC(rq);
1587

1588
	/*
1589
	 * check if this request is a better next-serve candidate)) {
1590
	 */
J
Jens Axboe 已提交
1591 1592
	cfqq->next_rq = cfq_choose_req(cfqd, cfqq->next_rq, rq);
	BUG_ON(!cfqq->next_rq);
1593

J
Jens Axboe 已提交
1594 1595 1596 1597
	/*
	 * we never wait for an async request and we don't allow preemption
	 * of an async request. so just return early
	 */
J
Jens Axboe 已提交
1598
	if (!rq_is_sync(rq)) {
1599 1600 1601 1602 1603 1604 1605 1606 1607
		/*
		 * sync process issued an async request, if it's waiting
		 * then expire it and kick rq handling.
		 */
		if (cic == cfqd->active_cic &&
		    del_timer(&cfqd->idle_slice_timer)) {
			cfq_slice_expired(cfqd, 0);
			cfq_start_queueing(cfqd, cfqq);
		}
J
Jens Axboe 已提交
1608
		return;
1609
	}
1610

J
Jens Axboe 已提交
1611
	cfq_update_io_thinktime(cfqd, cic);
J
Jens Axboe 已提交
1612
	cfq_update_io_seektime(cfqd, cic, rq);
J
Jens Axboe 已提交
1613 1614 1615
	cfq_update_idle_window(cfqd, cfqq, cic);

	cic->last_queue = jiffies;
J
Jens Axboe 已提交
1616
	cic->last_request_pos = rq->sector + rq->nr_sectors;
1617 1618 1619 1620 1621 1622 1623

	if (cfqq == cfqd->active_queue) {
		/*
		 * if we are waiting for a request for this queue, let it rip
		 * immediately and flag that we must not expire this queue
		 * just now
		 */
J
Jens Axboe 已提交
1624 1625
		if (cfq_cfqq_wait_request(cfqq)) {
			cfq_mark_cfqq_must_dispatch(cfqq);
1626 1627 1628
			del_timer(&cfqd->idle_slice_timer);
			cfq_start_queueing(cfqd, cfqq);
		}
J
Jens Axboe 已提交
1629
	} else if (cfq_should_preempt(cfqd, cfqq, rq)) {
1630 1631 1632 1633 1634 1635
		/*
		 * not the active queue - expire current slice if it is
		 * idle and has expired it's mean thinktime or this new queue
		 * has some old slice time left and is of higher priority
		 */
		cfq_preempt_queue(cfqd, cfqq);
J
Jens Axboe 已提交
1636
		cfq_mark_cfqq_must_dispatch(cfqq);
1637 1638
		cfq_start_queueing(cfqd, cfqq);
	}
L
Linus Torvalds 已提交
1639 1640
}

1641
static void cfq_insert_request(request_queue_t *q, struct request *rq)
L
Linus Torvalds 已提交
1642
{
1643
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
1644
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
1645 1646

	cfq_init_prio_data(cfqq);
L
Linus Torvalds 已提交
1647

J
Jens Axboe 已提交
1648
	cfq_add_rq_rb(rq);
L
Linus Torvalds 已提交
1649

1650 1651 1652
	if (!cfq_cfqq_on_rr(cfqq))
		cfq_add_cfqq_rr(cfqd, cfqq);

1653 1654
	list_add_tail(&rq->queuelist, &cfqq->fifo);

J
Jens Axboe 已提交
1655
	cfq_rq_enqueued(cfqd, cfqq, rq);
L
Linus Torvalds 已提交
1656 1657 1658 1659
}

static void cfq_completed_request(request_queue_t *q, struct request *rq)
{
J
Jens Axboe 已提交
1660
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
1661
	struct cfq_data *cfqd = cfqq->cfqd;
1662
	const int sync = rq_is_sync(rq);
1663
	unsigned long now;
L
Linus Torvalds 已提交
1664

1665
	now = jiffies;
L
Linus Torvalds 已提交
1666

1667 1668 1669 1670
	WARN_ON(!cfqd->rq_in_driver);
	WARN_ON(!cfqq->on_dispatch[sync]);
	cfqd->rq_in_driver--;
	cfqq->on_dispatch[sync]--;
L
Linus Torvalds 已提交
1671

1672 1673
	if (!cfq_class_idle(cfqq))
		cfqd->last_end_request = now;
J
Jens Axboe 已提交
1674

1675 1676 1677 1678 1679
	if (!cfq_cfqq_dispatched(cfqq)) {
		if (cfq_cfqq_on_rr(cfqq)) {
			cfqq->service_last = now;
			cfq_resort_rr_list(cfqq, 0);
		}
L
Linus Torvalds 已提交
1680 1681
	}

1682
	if (sync)
J
Jens Axboe 已提交
1683
		RQ_CIC(rq)->last_end_request = now;
1684 1685 1686 1687 1688 1689 1690 1691

	/*
	 * If this is the active queue, check if it needs to be expired,
	 * or if we want to idle in case it has no pending requests.
	 */
	if (cfqd->active_queue == cfqq) {
		if (time_after(now, cfqq->slice_end))
			cfq_slice_expired(cfqd, 0);
1692
		else if (sync && RB_EMPTY_ROOT(&cfqq->sort_list)) {
1693 1694 1695 1696
			if (!cfq_arm_slice_timer(cfqd, cfqq))
				cfq_schedule_dispatch(cfqd);
		}
	}
L
Linus Torvalds 已提交
1697 1698
}

1699 1700 1701 1702 1703
/*
 * we temporarily boost lower priority queues if they are holding fs exclusive
 * resources. they are boosted to normal prio (CLASS_BE/4)
 */
static void cfq_prio_boost(struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
1704
{
1705 1706
	const int ioprio_class = cfqq->ioprio_class;
	const int ioprio = cfqq->ioprio;
L
Linus Torvalds 已提交
1707

1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725
	if (has_fs_excl()) {
		/*
		 * boost idle prio on transactions that would lock out other
		 * users of the filesystem
		 */
		if (cfq_class_idle(cfqq))
			cfqq->ioprio_class = IOPRIO_CLASS_BE;
		if (cfqq->ioprio > IOPRIO_NORM)
			cfqq->ioprio = IOPRIO_NORM;
	} else {
		/*
		 * check if we need to unboost the queue
		 */
		if (cfqq->ioprio_class != cfqq->org_ioprio_class)
			cfqq->ioprio_class = cfqq->org_ioprio_class;
		if (cfqq->ioprio != cfqq->org_ioprio)
			cfqq->ioprio = cfqq->org_ioprio;
	}
L
Linus Torvalds 已提交
1726

1727 1728 1729 1730
	/*
	 * refile between round-robin lists if we moved the priority class
	 */
	if ((ioprio_class != cfqq->ioprio_class || ioprio != cfqq->ioprio) &&
J
Jens Axboe 已提交
1731
	    cfq_cfqq_on_rr(cfqq))
1732 1733
		cfq_resort_rr_list(cfqq, 0);
}
L
Linus Torvalds 已提交
1734

1735
static inline int __cfq_may_queue(struct cfq_queue *cfqq)
1736
{
J
Jens Axboe 已提交
1737
	if ((cfq_cfqq_wait_request(cfqq) || cfq_cfqq_must_alloc(cfqq)) &&
A
Andrew Morton 已提交
1738
	    !cfq_cfqq_must_alloc_slice(cfqq)) {
J
Jens Axboe 已提交
1739
		cfq_mark_cfqq_must_alloc_slice(cfqq);
1740
		return ELV_MQUEUE_MUST;
J
Jens Axboe 已提交
1741
	}
L
Linus Torvalds 已提交
1742

1743 1744 1745
	return ELV_MQUEUE_MAY;
}

1746
static int cfq_may_queue(request_queue_t *q, int rw)
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
	struct task_struct *tsk = current;
	struct cfq_queue *cfqq;

	/*
	 * don't force setup of a queue from here, as a call to may_queue
	 * does not necessarily imply that a request actually will be queued.
	 * so just lookup a possibly existing queue, or return 'may queue'
	 * if that fails
	 */
J
Jens Axboe 已提交
1758
	cfqq = cfq_find_cfq_hash(cfqd, cfq_queue_pid(tsk, rw), tsk->ioprio);
1759 1760 1761 1762
	if (cfqq) {
		cfq_init_prio_data(cfqq);
		cfq_prio_boost(cfqq);

1763
		return __cfq_may_queue(cfqq);
1764 1765 1766
	}

	return ELV_MQUEUE_MAY;
L
Linus Torvalds 已提交
1767 1768 1769 1770 1771 1772 1773
}

/*
 * queue lock held here
 */
static void cfq_put_request(request_queue_t *q, struct request *rq)
{
J
Jens Axboe 已提交
1774
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
L
Linus Torvalds 已提交
1775

J
Jens Axboe 已提交
1776
	if (cfqq) {
1777
		const int rw = rq_data_dir(rq);
L
Linus Torvalds 已提交
1778

1779 1780
		BUG_ON(!cfqq->allocated[rw]);
		cfqq->allocated[rw]--;
L
Linus Torvalds 已提交
1781

J
Jens Axboe 已提交
1782
		put_io_context(RQ_CIC(rq)->ioc);
L
Linus Torvalds 已提交
1783 1784

		rq->elevator_private = NULL;
J
Jens Axboe 已提交
1785
		rq->elevator_private2 = NULL;
L
Linus Torvalds 已提交
1786 1787 1788 1789 1790 1791

		cfq_put_queue(cfqq);
	}
}

/*
1792
 * Allocate cfq data structures associated with this request.
L
Linus Torvalds 已提交
1793
 */
1794
static int
1795
cfq_set_request(request_queue_t *q, struct request *rq, gfp_t gfp_mask)
L
Linus Torvalds 已提交
1796 1797
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
1798
	struct task_struct *tsk = current;
L
Linus Torvalds 已提交
1799 1800
	struct cfq_io_context *cic;
	const int rw = rq_data_dir(rq);
J
Jens Axboe 已提交
1801
	pid_t key = cfq_queue_pid(tsk, rw);
1802
	struct cfq_queue *cfqq;
L
Linus Torvalds 已提交
1803
	unsigned long flags;
1804
	int is_sync = key != CFQ_KEY_ASYNC;
L
Linus Torvalds 已提交
1805 1806 1807

	might_sleep_if(gfp_mask & __GFP_WAIT);

1808
	cic = cfq_get_io_context(cfqd, gfp_mask);
1809

L
Linus Torvalds 已提交
1810 1811
	spin_lock_irqsave(q->queue_lock, flags);

1812 1813 1814
	if (!cic)
		goto queue_fail;

1815
	if (!cic->cfqq[is_sync]) {
1816
		cfqq = cfq_get_queue(cfqd, key, tsk, gfp_mask);
1817 1818
		if (!cfqq)
			goto queue_fail;
L
Linus Torvalds 已提交
1819

1820
		cic->cfqq[is_sync] = cfqq;
1821
	} else
1822
		cfqq = cic->cfqq[is_sync];
L
Linus Torvalds 已提交
1823 1824

	cfqq->allocated[rw]++;
J
Jens Axboe 已提交
1825
	cfq_clear_cfqq_must_alloc(cfqq);
1826
	atomic_inc(&cfqq->ref);
L
Linus Torvalds 已提交
1827

J
Jens Axboe 已提交
1828
	spin_unlock_irqrestore(q->queue_lock, flags);
J
Jens Axboe 已提交
1829

J
Jens Axboe 已提交
1830 1831 1832
	rq->elevator_private = cic;
	rq->elevator_private2 = cfqq;
	return 0;
L
Linus Torvalds 已提交
1833

1834 1835 1836
queue_fail:
	if (cic)
		put_io_context(cic->ioc);
1837

J
Jens Axboe 已提交
1838
	cfq_schedule_dispatch(cfqd);
L
Linus Torvalds 已提交
1839 1840 1841 1842
	spin_unlock_irqrestore(q->queue_lock, flags);
	return 1;
}

1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
static void cfq_kick_queue(void *data)
{
	request_queue_t *q = data;
	unsigned long flags;

	spin_lock_irqsave(q->queue_lock, flags);
	blk_remove_plug(q);
	q->request_fn(q);
	spin_unlock_irqrestore(q->queue_lock, flags);
}

/*
 * Timer running if the active_queue is currently idling inside its time slice
 */
static void cfq_idle_slice_timer(unsigned long data)
{
	struct cfq_data *cfqd = (struct cfq_data *) data;
	struct cfq_queue *cfqq;
	unsigned long flags;

	spin_lock_irqsave(cfqd->queue->queue_lock, flags);

	if ((cfqq = cfqd->active_queue) != NULL) {
		unsigned long now = jiffies;

		/*
		 * expired
		 */
		if (time_after(now, cfqq->slice_end))
			goto expire;

		/*
		 * only expire and reinvoke request handler, if there are
		 * other queues with pending requests
		 */
1878
		if (!cfqd->busy_queues)
1879 1880 1881 1882 1883
			goto out_cont;

		/*
		 * not expired and it has a request pending, let it dispatch
		 */
1884
		if (!RB_EMPTY_ROOT(&cfqq->sort_list)) {
J
Jens Axboe 已提交
1885
			cfq_mark_cfqq_must_dispatch(cfqq);
1886 1887 1888 1889 1890 1891
			goto out_kick;
		}
	}
expire:
	cfq_slice_expired(cfqd, 0);
out_kick:
J
Jens Axboe 已提交
1892
	cfq_schedule_dispatch(cfqd);
1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910
out_cont:
	spin_unlock_irqrestore(cfqd->queue->queue_lock, flags);
}

/*
 * Timer running if an idle class queue is waiting for service
 */
static void cfq_idle_class_timer(unsigned long data)
{
	struct cfq_data *cfqd = (struct cfq_data *) data;
	unsigned long flags, end;

	spin_lock_irqsave(cfqd->queue->queue_lock, flags);

	/*
	 * race with a non-idle queue, reset timer
	 */
	end = cfqd->last_end_request + CFQ_IDLE_GRACE;
1911 1912 1913
	if (!time_after_eq(jiffies, end))
		mod_timer(&cfqd->idle_class_timer, end);
	else
J
Jens Axboe 已提交
1914
		cfq_schedule_dispatch(cfqd);
1915 1916 1917 1918

	spin_unlock_irqrestore(cfqd->queue->queue_lock, flags);
}

J
Jens Axboe 已提交
1919 1920 1921 1922 1923 1924
static void cfq_shutdown_timer_wq(struct cfq_data *cfqd)
{
	del_timer_sync(&cfqd->idle_slice_timer);
	del_timer_sync(&cfqd->idle_class_timer);
	blk_sync_queue(cfqd->queue);
}
1925

L
Linus Torvalds 已提交
1926 1927
static void cfq_exit_queue(elevator_t *e)
{
1928
	struct cfq_data *cfqd = e->elevator_data;
1929
	request_queue_t *q = cfqd->queue;
1930

J
Jens Axboe 已提交
1931
	cfq_shutdown_timer_wq(cfqd);
1932

1933
	spin_lock_irq(q->queue_lock);
1934

1935 1936
	if (cfqd->active_queue)
		__cfq_slice_expired(cfqd, cfqd->active_queue, 0);
1937 1938

	while (!list_empty(&cfqd->cic_list)) {
1939 1940 1941
		struct cfq_io_context *cic = list_entry(cfqd->cic_list.next,
							struct cfq_io_context,
							queue_list);
1942 1943

		__cfq_exit_single_io_context(cfqd, cic);
1944
	}
1945

1946
	spin_unlock_irq(q->queue_lock);
1947 1948 1949 1950 1951

	cfq_shutdown_timer_wq(cfqd);

	kfree(cfqd->cfq_hash);
	kfree(cfqd);
L
Linus Torvalds 已提交
1952 1953
}

J
Jens Axboe 已提交
1954
static void *cfq_init_queue(request_queue_t *q, elevator_t *e)
L
Linus Torvalds 已提交
1955 1956 1957 1958
{
	struct cfq_data *cfqd;
	int i;

1959
	cfqd = kmalloc_node(sizeof(*cfqd), GFP_KERNEL, q->node);
L
Linus Torvalds 已提交
1960
	if (!cfqd)
J
Jens Axboe 已提交
1961
		return NULL;
L
Linus Torvalds 已提交
1962 1963

	memset(cfqd, 0, sizeof(*cfqd));
1964 1965 1966 1967 1968 1969 1970

	for (i = 0; i < CFQ_PRIO_LISTS; i++)
		INIT_LIST_HEAD(&cfqd->rr_list[i]);

	INIT_LIST_HEAD(&cfqd->busy_rr);
	INIT_LIST_HEAD(&cfqd->cur_rr);
	INIT_LIST_HEAD(&cfqd->idle_rr);
L
Linus Torvalds 已提交
1971
	INIT_LIST_HEAD(&cfqd->empty_list);
1972
	INIT_LIST_HEAD(&cfqd->cic_list);
L
Linus Torvalds 已提交
1973

1974
	cfqd->cfq_hash = kmalloc_node(sizeof(struct hlist_head) * CFQ_QHASH_ENTRIES, GFP_KERNEL, q->node);
L
Linus Torvalds 已提交
1975
	if (!cfqd->cfq_hash)
J
Jens Axboe 已提交
1976
		goto out_free;
L
Linus Torvalds 已提交
1977 1978 1979 1980 1981 1982

	for (i = 0; i < CFQ_QHASH_ENTRIES; i++)
		INIT_HLIST_HEAD(&cfqd->cfq_hash[i]);

	cfqd->queue = q;

1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
	init_timer(&cfqd->idle_slice_timer);
	cfqd->idle_slice_timer.function = cfq_idle_slice_timer;
	cfqd->idle_slice_timer.data = (unsigned long) cfqd;

	init_timer(&cfqd->idle_class_timer);
	cfqd->idle_class_timer.function = cfq_idle_class_timer;
	cfqd->idle_class_timer.data = (unsigned long) cfqd;

	INIT_WORK(&cfqd->unplug_work, cfq_kick_queue, q);

L
Linus Torvalds 已提交
1993
	cfqd->cfq_quantum = cfq_quantum;
1994 1995
	cfqd->cfq_fifo_expire[0] = cfq_fifo_expire[0];
	cfqd->cfq_fifo_expire[1] = cfq_fifo_expire[1];
L
Linus Torvalds 已提交
1996 1997
	cfqd->cfq_back_max = cfq_back_max;
	cfqd->cfq_back_penalty = cfq_back_penalty;
1998 1999 2000 2001
	cfqd->cfq_slice[0] = cfq_slice_async;
	cfqd->cfq_slice[1] = cfq_slice_sync;
	cfqd->cfq_slice_async_rq = cfq_slice_async_rq;
	cfqd->cfq_slice_idle = cfq_slice_idle;
J
Jens Axboe 已提交
2002

J
Jens Axboe 已提交
2003
	return cfqd;
J
Jens Axboe 已提交
2004
out_free:
L
Linus Torvalds 已提交
2005
	kfree(cfqd);
J
Jens Axboe 已提交
2006
	return NULL;
L
Linus Torvalds 已提交
2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054
}

static void cfq_slab_kill(void)
{
	if (cfq_pool)
		kmem_cache_destroy(cfq_pool);
	if (cfq_ioc_pool)
		kmem_cache_destroy(cfq_ioc_pool);
}

static int __init cfq_slab_setup(void)
{
	cfq_pool = kmem_cache_create("cfq_pool", sizeof(struct cfq_queue), 0, 0,
					NULL, NULL);
	if (!cfq_pool)
		goto fail;

	cfq_ioc_pool = kmem_cache_create("cfq_ioc_pool",
			sizeof(struct cfq_io_context), 0, 0, NULL, NULL);
	if (!cfq_ioc_pool)
		goto fail;

	return 0;
fail:
	cfq_slab_kill();
	return -ENOMEM;
}

/*
 * sysfs parts below -->
 */

static ssize_t
cfq_var_show(unsigned int var, char *page)
{
	return sprintf(page, "%d\n", var);
}

static ssize_t
cfq_var_store(unsigned int *var, const char *page, size_t count)
{
	char *p = (char *) page;

	*var = simple_strtoul(p, &p, 10);
	return count;
}

#define SHOW_FUNCTION(__FUNC, __VAR, __CONV)				\
2055
static ssize_t __FUNC(elevator_t *e, char *page)			\
L
Linus Torvalds 已提交
2056
{									\
2057
	struct cfq_data *cfqd = e->elevator_data;			\
L
Linus Torvalds 已提交
2058 2059 2060 2061 2062 2063
	unsigned int __data = __VAR;					\
	if (__CONV)							\
		__data = jiffies_to_msecs(__data);			\
	return cfq_var_show(__data, (page));				\
}
SHOW_FUNCTION(cfq_quantum_show, cfqd->cfq_quantum, 0);
2064 2065
SHOW_FUNCTION(cfq_fifo_expire_sync_show, cfqd->cfq_fifo_expire[1], 1);
SHOW_FUNCTION(cfq_fifo_expire_async_show, cfqd->cfq_fifo_expire[0], 1);
2066 2067
SHOW_FUNCTION(cfq_back_seek_max_show, cfqd->cfq_back_max, 0);
SHOW_FUNCTION(cfq_back_seek_penalty_show, cfqd->cfq_back_penalty, 0);
2068 2069 2070 2071
SHOW_FUNCTION(cfq_slice_idle_show, cfqd->cfq_slice_idle, 1);
SHOW_FUNCTION(cfq_slice_sync_show, cfqd->cfq_slice[1], 1);
SHOW_FUNCTION(cfq_slice_async_show, cfqd->cfq_slice[0], 1);
SHOW_FUNCTION(cfq_slice_async_rq_show, cfqd->cfq_slice_async_rq, 0);
L
Linus Torvalds 已提交
2072 2073 2074
#undef SHOW_FUNCTION

#define STORE_FUNCTION(__FUNC, __PTR, MIN, MAX, __CONV)			\
2075
static ssize_t __FUNC(elevator_t *e, const char *page, size_t count)	\
L
Linus Torvalds 已提交
2076
{									\
2077
	struct cfq_data *cfqd = e->elevator_data;			\
L
Linus Torvalds 已提交
2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090
	unsigned int __data;						\
	int ret = cfq_var_store(&__data, (page), count);		\
	if (__data < (MIN))						\
		__data = (MIN);						\
	else if (__data > (MAX))					\
		__data = (MAX);						\
	if (__CONV)							\
		*(__PTR) = msecs_to_jiffies(__data);			\
	else								\
		*(__PTR) = __data;					\
	return ret;							\
}
STORE_FUNCTION(cfq_quantum_store, &cfqd->cfq_quantum, 1, UINT_MAX, 0);
2091 2092
STORE_FUNCTION(cfq_fifo_expire_sync_store, &cfqd->cfq_fifo_expire[1], 1, UINT_MAX, 1);
STORE_FUNCTION(cfq_fifo_expire_async_store, &cfqd->cfq_fifo_expire[0], 1, UINT_MAX, 1);
2093 2094
STORE_FUNCTION(cfq_back_seek_max_store, &cfqd->cfq_back_max, 0, UINT_MAX, 0);
STORE_FUNCTION(cfq_back_seek_penalty_store, &cfqd->cfq_back_penalty, 1, UINT_MAX, 0);
2095 2096 2097 2098
STORE_FUNCTION(cfq_slice_idle_store, &cfqd->cfq_slice_idle, 0, UINT_MAX, 1);
STORE_FUNCTION(cfq_slice_sync_store, &cfqd->cfq_slice[1], 1, UINT_MAX, 1);
STORE_FUNCTION(cfq_slice_async_store, &cfqd->cfq_slice[0], 1, UINT_MAX, 1);
STORE_FUNCTION(cfq_slice_async_rq_store, &cfqd->cfq_slice_async_rq, 1, UINT_MAX, 0);
L
Linus Torvalds 已提交
2099 2100
#undef STORE_FUNCTION

2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114
#define CFQ_ATTR(name) \
	__ATTR(name, S_IRUGO|S_IWUSR, cfq_##name##_show, cfq_##name##_store)

static struct elv_fs_entry cfq_attrs[] = {
	CFQ_ATTR(quantum),
	CFQ_ATTR(fifo_expire_sync),
	CFQ_ATTR(fifo_expire_async),
	CFQ_ATTR(back_seek_max),
	CFQ_ATTR(back_seek_penalty),
	CFQ_ATTR(slice_sync),
	CFQ_ATTR(slice_async),
	CFQ_ATTR(slice_async_rq),
	CFQ_ATTR(slice_idle),
	__ATTR_NULL
L
Linus Torvalds 已提交
2115 2116 2117 2118 2119 2120 2121
};

static struct elevator_type iosched_cfq = {
	.ops = {
		.elevator_merge_fn = 		cfq_merge,
		.elevator_merged_fn =		cfq_merged_request,
		.elevator_merge_req_fn =	cfq_merged_requests,
2122
		.elevator_dispatch_fn =		cfq_dispatch_requests,
L
Linus Torvalds 已提交
2123
		.elevator_add_req_fn =		cfq_insert_request,
2124
		.elevator_activate_req_fn =	cfq_activate_request,
L
Linus Torvalds 已提交
2125 2126 2127
		.elevator_deactivate_req_fn =	cfq_deactivate_request,
		.elevator_queue_empty_fn =	cfq_queue_empty,
		.elevator_completed_req_fn =	cfq_completed_request,
2128 2129
		.elevator_former_req_fn =	elv_rb_former_request,
		.elevator_latter_req_fn =	elv_rb_latter_request,
L
Linus Torvalds 已提交
2130 2131 2132 2133 2134
		.elevator_set_req_fn =		cfq_set_request,
		.elevator_put_req_fn =		cfq_put_request,
		.elevator_may_queue_fn =	cfq_may_queue,
		.elevator_init_fn =		cfq_init_queue,
		.elevator_exit_fn =		cfq_exit_queue,
2135
		.trim =				cfq_free_io_context,
L
Linus Torvalds 已提交
2136
	},
2137
	.elevator_attrs =	cfq_attrs,
L
Linus Torvalds 已提交
2138 2139 2140 2141 2142 2143 2144 2145
	.elevator_name =	"cfq",
	.elevator_owner =	THIS_MODULE,
};

static int __init cfq_init(void)
{
	int ret;

2146 2147 2148 2149 2150 2151 2152 2153
	/*
	 * could be 0 on HZ < 1000 setups
	 */
	if (!cfq_slice_async)
		cfq_slice_async = 1;
	if (!cfq_slice_idle)
		cfq_slice_idle = 1;

L
Linus Torvalds 已提交
2154 2155 2156 2157
	if (cfq_slab_setup())
		return -ENOMEM;

	ret = elv_register(&iosched_cfq);
2158 2159
	if (ret)
		cfq_slab_kill();
L
Linus Torvalds 已提交
2160 2161 2162 2163 2164 2165

	return ret;
}

static void __exit cfq_exit(void)
{
2166
	DECLARE_COMPLETION(all_gone);
L
Linus Torvalds 已提交
2167
	elv_unregister(&iosched_cfq);
2168
	ioc_gone = &all_gone;
2169 2170
	/* ioc_gone's update must be visible before reading ioc_count */
	smp_wmb();
2171
	if (elv_ioc_count_read(ioc_count))
2172
		wait_for_completion(ioc_gone);
2173
	synchronize_rcu();
2174
	cfq_slab_kill();
L
Linus Torvalds 已提交
2175 2176 2177 2178 2179 2180 2181 2182
}

module_init(cfq_init);
module_exit(cfq_exit);

MODULE_AUTHOR("Jens Axboe");
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Completely Fair Queueing IO scheduler");