cfq-iosched.c 94.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6
/*
 *  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.
 *
7
 *  Copyright (C) 2003 Jens Axboe <axboe@kernel.dk>
L
Linus Torvalds 已提交
8 9
 */
#include <linux/module.h>
10
#include <linux/slab.h>
A
Al Viro 已提交
11 12
#include <linux/blkdev.h>
#include <linux/elevator.h>
R
Randy Dunlap 已提交
13
#include <linux/jiffies.h>
L
Linus Torvalds 已提交
14
#include <linux/rbtree.h>
15
#include <linux/ioprio.h>
16
#include <linux/blktrace_api.h>
17
#include "blk.h"
18
#include "cfq.h"
L
Linus Torvalds 已提交
19

20 21
static struct blkio_policy_type blkio_policy_cfq;

L
Linus Torvalds 已提交
22 23 24
/*
 * tunables
 */
25
/* max queue in one round of service */
S
Shaohua Li 已提交
26
static const int cfq_quantum = 8;
27
static const int cfq_fifo_expire[2] = { HZ / 4, HZ / 8 };
28 29 30 31
/* maximum backwards seek, in KiB */
static const int cfq_back_max = 16 * 1024;
/* penalty of a backwards seek */
static const int cfq_back_penalty = 2;
32
static const int cfq_slice_sync = HZ / 10;
J
Jens Axboe 已提交
33
static int cfq_slice_async = HZ / 25;
34
static const int cfq_slice_async_rq = 2;
35
static int cfq_slice_idle = HZ / 125;
36
static int cfq_group_idle = HZ / 125;
37 38
static const int cfq_target_latency = HZ * 3/10; /* 300 ms */
static const int cfq_hist_divisor = 4;
39

40
/*
41
 * offset from end of service tree
42
 */
43
#define CFQ_IDLE_DELAY		(HZ / 5)
44 45 46 47 48 49

/*
 * below this threshold, we consider thinktime immediate
 */
#define CFQ_MIN_TT		(2)

50
#define CFQ_SLICE_SCALE		(5)
51
#define CFQ_HW_QUEUE_MIN	(5)
52
#define CFQ_SERVICE_SHIFT       12
53

54
#define CFQQ_SEEK_THR		(sector_t)(8 * 100)
55
#define CFQQ_CLOSE_THR		(sector_t)(8 * 1024)
56
#define CFQQ_SECT_THR_NONROT	(sector_t)(2 * 32)
57
#define CFQQ_SEEKY(cfqq)	(hweight32(cfqq->seek_history) > 32/8)
58

59 60 61
#define RQ_CIC(rq)		icq_to_cic((rq)->elv.icq)
#define RQ_CFQQ(rq)		(struct cfq_queue *) ((rq)->elv.priv[0])
#define RQ_CFQG(rq)		(struct cfq_group *) ((rq)->elv.priv[1])
L
Linus Torvalds 已提交
62

63
static struct kmem_cache *cfq_pool;
L
Linus Torvalds 已提交
64

65 66 67 68
#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)

69
#define sample_valid(samples)	((samples) > 80)
70
#define rb_entry_cfqg(node)	rb_entry((node), struct cfq_group, rb_node)
71

72 73 74 75 76 77 78 79
struct cfq_ttime {
	unsigned long last_end_request;

	unsigned long ttime_total;
	unsigned long ttime_samples;
	unsigned long ttime_mean;
};

80 81 82 83 84 85 86 87 88
/*
 * Most of our rbtree usage is for sorting with min extraction, so
 * if we cache the leftmost node we don't have to walk down the tree
 * to find it. Idea borrowed from Ingo Molnars CFS scheduler. We should
 * move this into the elevator for the rq sorting as well.
 */
struct cfq_rb_root {
	struct rb_root rb;
	struct rb_node *left;
89
	unsigned count;
90
	unsigned total_weight;
91
	u64 min_vdisktime;
92
	struct cfq_ttime ttime;
93
};
94 95
#define CFQ_RB_ROOT	(struct cfq_rb_root) { .rb = RB_ROOT, \
			.ttime = {.last_end_request = jiffies,},}
96

97 98 99 100 101
/*
 * Per process-grouping structure
 */
struct cfq_queue {
	/* reference count */
102
	int ref;
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
	/* various state flags, see below */
	unsigned int flags;
	/* parent cfq_data */
	struct cfq_data *cfqd;
	/* service_tree member */
	struct rb_node rb_node;
	/* service_tree key */
	unsigned long rb_key;
	/* prio tree member */
	struct rb_node p_node;
	/* prio tree root we belong to, if any */
	struct rb_root *p_root;
	/* sorted list of pending requests */
	struct rb_root sort_list;
	/* if fifo isn't expired, next request to serve */
	struct request *next_rq;
	/* requests queued in sort_list */
	int queued[2];
	/* currently allocated requests */
	int allocated[2];
	/* fifo list of requests in sort_list */
	struct list_head fifo;

126 127
	/* time when queue got scheduled in to dispatch first request. */
	unsigned long dispatch_start;
128
	unsigned int allocated_slice;
129
	unsigned int slice_dispatch;
130 131
	/* time when first request from queue completed and slice started. */
	unsigned long slice_start;
132 133 134
	unsigned long slice_end;
	long slice_resid;

135 136
	/* pending priority requests */
	int prio_pending;
137 138 139 140 141
	/* number of requests that are on the dispatch list or inside driver */
	int dispatched;

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

144 145
	pid_t pid;

146
	u32 seek_history;
147 148
	sector_t last_request_pos;

149
	struct cfq_rb_root *service_tree;
J
Jeff Moyer 已提交
150
	struct cfq_queue *new_cfqq;
151
	struct cfq_group *cfqg;
152 153
	/* Number of sectors dispatched from queue in single dispatch round */
	unsigned long nr_sectors;
154 155
};

156
/*
157
 * First index in the service_trees.
158 159 160 161
 * IDLE is handled separately, so it has negative index
 */
enum wl_prio_t {
	BE_WORKLOAD = 0,
162 163
	RT_WORKLOAD = 1,
	IDLE_WORKLOAD = 2,
164
	CFQ_PRIO_NR,
165 166
};

167 168 169 170 171 172 173 174 175
/*
 * Second index in the service_trees.
 */
enum wl_type_t {
	ASYNC_WORKLOAD = 0,
	SYNC_NOIDLE_WORKLOAD = 1,
	SYNC_WORKLOAD = 2
};

176 177
/* This is per cgroup per device grouping structure */
struct cfq_group {
178 179 180 181 182
	/* group service_tree member */
	struct rb_node rb_node;

	/* group service_tree key */
	u64 vdisktime;
183
	unsigned int weight;
184 185
	unsigned int new_weight;
	bool needs_update;
186 187 188 189

	/* number of cfqq currently on this group */
	int nr_cfqq;

190
	/*
191
	 * Per group busy queues average. Useful for workload slice calc. We
192 193 194 195 196 197 198 199 200 201 202
	 * create the array for each prio class but at run time it is used
	 * only for RT and BE class and slot for IDLE class remains unused.
	 * This is primarily done to avoid confusion and a gcc warning.
	 */
	unsigned int busy_queues_avg[CFQ_PRIO_NR];
	/*
	 * rr lists of queues with requests. We maintain service trees for
	 * RT and BE classes. These trees are subdivided in subclasses
	 * of SYNC, SYNC_NOIDLE and ASYNC based on workload type. For IDLE
	 * class there is no subclassification and all the cfq queues go on
	 * a single tree service_tree_idle.
203 204 205 206
	 * Counts are embedded in the cfq_rb_root
	 */
	struct cfq_rb_root service_trees[2][3];
	struct cfq_rb_root service_tree_idle;
207 208 209 210

	unsigned long saved_workload_slice;
	enum wl_type_t saved_workload;
	enum wl_prio_t saved_serving_prio;
211

212 213
	/* number of requests that are on the dispatch list or inside driver */
	int dispatched;
S
Shaohua Li 已提交
214
	struct cfq_ttime ttime;
215
};
216

217 218 219 220 221 222
struct cfq_io_cq {
	struct io_cq		icq;		/* must be the first member */
	struct cfq_queue	*cfqq[2];
	struct cfq_ttime	ttime;
};

223 224 225
/*
 * Per block device queue structure
 */
L
Linus Torvalds 已提交
226
struct cfq_data {
227
	struct request_queue *queue;
228 229
	/* Root service tree for cfq_groups */
	struct cfq_rb_root grp_service_tree;
230
	struct cfq_group *root_group;
231

232 233
	/*
	 * The priority currently being served
234
	 */
235
	enum wl_prio_t serving_prio;
236 237
	enum wl_type_t serving_type;
	unsigned long workload_expires;
238
	struct cfq_group *serving_group;
239 240 241 242 243 244 245 246

	/*
	 * Each priority tree is sorted by next_request position.  These
	 * trees are used when determining if two or more queues are
	 * interleaving requests (see cfq_close_cooperator).
	 */
	struct rb_root prio_trees[CFQ_PRIO_LISTS];

247
	unsigned int busy_queues;
248
	unsigned int busy_sync_queues;
249

250 251
	int rq_in_driver;
	int rq_in_flight[2];
252 253 254 255 256

	/*
	 * queue-depth detection
	 */
	int rq_queued;
257
	int hw_tag;
258 259 260 261 262 263 264 265
	/*
	 * hw_tag can be
	 * -1 => indeterminate, (cfq will behave as if NCQ is present, to allow better detection)
	 *  1 => NCQ is present (hw_tag_est_depth is the estimated max depth)
	 *  0 => no NCQ
	 */
	int hw_tag_est_depth;
	unsigned int hw_tag_samples;
L
Linus Torvalds 已提交
266

267 268 269 270
	/*
	 * idle window management
	 */
	struct timer_list idle_slice_timer;
271
	struct work_struct unplug_work;
L
Linus Torvalds 已提交
272

273
	struct cfq_queue *active_queue;
274
	struct cfq_io_cq *active_cic;
275

276 277 278 279 280
	/*
	 * async queue for each priority case
	 */
	struct cfq_queue *async_cfqq[2][IOPRIO_BE_NR];
	struct cfq_queue *async_idle_cfqq;
281

J
Jens Axboe 已提交
282
	sector_t last_position;
L
Linus Torvalds 已提交
283 284 285 286 287

	/*
	 * tunables, see top of file
	 */
	unsigned int cfq_quantum;
288
	unsigned int cfq_fifo_expire[2];
L
Linus Torvalds 已提交
289 290
	unsigned int cfq_back_penalty;
	unsigned int cfq_back_max;
291 292 293
	unsigned int cfq_slice[2];
	unsigned int cfq_slice_async_rq;
	unsigned int cfq_slice_idle;
294
	unsigned int cfq_group_idle;
295
	unsigned int cfq_latency;
296

297 298 299 300
	/*
	 * Fallback dummy cfqq for extreme OOM conditions
	 */
	struct cfq_queue oom_cfqq;
301

302
	unsigned long last_delayed_sync;
L
Linus Torvalds 已提交
303 304
};

305 306 307 308 309 310 311 312 313 314
static inline struct cfq_group *blkg_to_cfqg(struct blkio_group *blkg)
{
	return blkg_to_pdata(blkg, &blkio_policy_cfq);
}

static inline struct blkio_group *cfqg_to_blkg(struct cfq_group *cfqg)
{
	return pdata_to_blkg(cfqg, &blkio_policy_cfq);
}

315 316
static struct cfq_group *cfq_get_next_cfqg(struct cfq_data *cfqd);

317 318
static struct cfq_rb_root *service_tree_for(struct cfq_group *cfqg,
					    enum wl_prio_t prio,
319
					    enum wl_type_t type)
320
{
321 322 323
	if (!cfqg)
		return NULL;

324
	if (prio == IDLE_WORKLOAD)
325
		return &cfqg->service_tree_idle;
326

327
	return &cfqg->service_trees[prio][type];
328 329
}

J
Jens Axboe 已提交
330
enum cfqq_state_flags {
331 332
	CFQ_CFQQ_FLAG_on_rr = 0,	/* on round-robin busy list */
	CFQ_CFQQ_FLAG_wait_request,	/* waiting for a request */
333
	CFQ_CFQQ_FLAG_must_dispatch,	/* must be allowed a dispatch */
334 335 336 337
	CFQ_CFQQ_FLAG_must_alloc_slice,	/* per-slice must_alloc flag */
	CFQ_CFQQ_FLAG_fifo_expire,	/* FIFO checked in this slice */
	CFQ_CFQQ_FLAG_idle_window,	/* slice idling enabled */
	CFQ_CFQQ_FLAG_prio_changed,	/* task priority has changed */
338
	CFQ_CFQQ_FLAG_slice_new,	/* no requests dispatched in slice */
339
	CFQ_CFQQ_FLAG_sync,		/* synchronous queue */
340
	CFQ_CFQQ_FLAG_coop,		/* cfqq is shared */
341
	CFQ_CFQQ_FLAG_split_coop,	/* shared cfqq will be splitted */
342
	CFQ_CFQQ_FLAG_deep,		/* sync cfqq experienced large depth */
343
	CFQ_CFQQ_FLAG_wait_busy,	/* Waiting for next request */
J
Jens Axboe 已提交
344 345 346 347 348
};

#define CFQ_CFQQ_FNS(name)						\
static inline void cfq_mark_cfqq_##name(struct cfq_queue *cfqq)		\
{									\
349
	(cfqq)->flags |= (1 << CFQ_CFQQ_FLAG_##name);			\
J
Jens Axboe 已提交
350 351 352
}									\
static inline void cfq_clear_cfqq_##name(struct cfq_queue *cfqq)	\
{									\
353
	(cfqq)->flags &= ~(1 << CFQ_CFQQ_FLAG_##name);			\
J
Jens Axboe 已提交
354 355 356
}									\
static inline int cfq_cfqq_##name(const struct cfq_queue *cfqq)		\
{									\
357
	return ((cfqq)->flags & (1 << CFQ_CFQQ_FLAG_##name)) != 0;	\
J
Jens Axboe 已提交
358 359 360 361
}

CFQ_CFQQ_FNS(on_rr);
CFQ_CFQQ_FNS(wait_request);
362
CFQ_CFQQ_FNS(must_dispatch);
J
Jens Axboe 已提交
363 364 365 366
CFQ_CFQQ_FNS(must_alloc_slice);
CFQ_CFQQ_FNS(fifo_expire);
CFQ_CFQQ_FNS(idle_window);
CFQ_CFQQ_FNS(prio_changed);
367
CFQ_CFQQ_FNS(slice_new);
368
CFQ_CFQQ_FNS(sync);
369
CFQ_CFQQ_FNS(coop);
370
CFQ_CFQQ_FNS(split_coop);
371
CFQ_CFQQ_FNS(deep);
372
CFQ_CFQQ_FNS(wait_busy);
J
Jens Axboe 已提交
373 374
#undef CFQ_CFQQ_FNS

375
#ifdef CONFIG_CFQ_GROUP_IOSCHED
V
Vivek Goyal 已提交
376 377 378
#define cfq_log_cfqq(cfqd, cfqq, fmt, args...)	\
	blk_add_trace_msg((cfqd)->queue, "cfq%d%c %s " fmt, (cfqq)->pid, \
			cfq_cfqq_sync((cfqq)) ? 'S' : 'A', \
379
			blkg_path(cfqg_to_blkg((cfqq)->cfqg)), ##args)
V
Vivek Goyal 已提交
380 381 382

#define cfq_log_cfqg(cfqd, cfqg, fmt, args...)				\
	blk_add_trace_msg((cfqd)->queue, "%s " fmt,			\
383
			blkg_path(cfqg_to_blkg((cfqg))), ##args)	\
V
Vivek Goyal 已提交
384 385

#else
386 387
#define cfq_log_cfqq(cfqd, cfqq, fmt, args...)	\
	blk_add_trace_msg((cfqd)->queue, "cfq%d " fmt, (cfqq)->pid, ##args)
388
#define cfq_log_cfqg(cfqd, cfqg, fmt, args...)		do {} while (0)
V
Vivek Goyal 已提交
389
#endif
390 391 392
#define cfq_log(cfqd, fmt, args...)	\
	blk_add_trace_msg((cfqd)->queue, "cfq " fmt, ##args)

393 394 395 396 397 398 399 400 401 402
/* Traverses through cfq group service trees */
#define for_each_cfqg_st(cfqg, i, j, st) \
	for (i = 0; i <= IDLE_WORKLOAD; i++) \
		for (j = 0, st = i < IDLE_WORKLOAD ? &cfqg->service_trees[i][j]\
			: &cfqg->service_tree_idle; \
			(i < IDLE_WORKLOAD && j <= SYNC_WORKLOAD) || \
			(i == IDLE_WORKLOAD && j == 0); \
			j++, st = i < IDLE_WORKLOAD ? \
			&cfqg->service_trees[i][j]: NULL) \

403 404 405 406 407 408 409 410 411 412 413 414
static inline bool cfq_io_thinktime_big(struct cfq_data *cfqd,
	struct cfq_ttime *ttime, bool group_idle)
{
	unsigned long slice;
	if (!sample_valid(ttime->ttime_samples))
		return false;
	if (group_idle)
		slice = cfqd->cfq_group_idle;
	else
		slice = cfqd->cfq_slice_idle;
	return ttime->ttime_mean > slice;
}
415

416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
static inline bool iops_mode(struct cfq_data *cfqd)
{
	/*
	 * If we are not idling on queues and it is a NCQ drive, parallel
	 * execution of requests is on and measuring time is not possible
	 * in most of the cases until and unless we drive shallower queue
	 * depths and that becomes a performance bottleneck. In such cases
	 * switch to start providing fairness in terms of number of IOs.
	 */
	if (!cfqd->cfq_slice_idle && cfqd->hw_tag)
		return true;
	else
		return false;
}

431 432 433 434 435 436 437 438 439
static inline enum wl_prio_t cfqq_prio(struct cfq_queue *cfqq)
{
	if (cfq_class_idle(cfqq))
		return IDLE_WORKLOAD;
	if (cfq_class_rt(cfqq))
		return RT_WORKLOAD;
	return BE_WORKLOAD;
}

440 441 442 443 444 445 446 447 448 449

static enum wl_type_t cfqq_type(struct cfq_queue *cfqq)
{
	if (!cfq_cfqq_sync(cfqq))
		return ASYNC_WORKLOAD;
	if (!cfq_cfqq_idle_window(cfqq))
		return SYNC_NOIDLE_WORKLOAD;
	return SYNC_WORKLOAD;
}

450 451 452
static inline int cfq_group_busy_queues_wl(enum wl_prio_t wl,
					struct cfq_data *cfqd,
					struct cfq_group *cfqg)
453 454
{
	if (wl == IDLE_WORKLOAD)
455
		return cfqg->service_tree_idle.count;
456

457 458 459
	return cfqg->service_trees[wl][ASYNC_WORKLOAD].count
		+ cfqg->service_trees[wl][SYNC_NOIDLE_WORKLOAD].count
		+ cfqg->service_trees[wl][SYNC_WORKLOAD].count;
460 461
}

462 463 464 465 466 467 468
static inline int cfqg_busy_async_queues(struct cfq_data *cfqd,
					struct cfq_group *cfqg)
{
	return cfqg->service_trees[RT_WORKLOAD][ASYNC_WORKLOAD].count
		+ cfqg->service_trees[BE_WORKLOAD][ASYNC_WORKLOAD].count;
}

469
static void cfq_dispatch_insert(struct request_queue *, struct request *);
470
static struct cfq_queue *cfq_get_queue(struct cfq_data *cfqd, bool is_sync,
471
				       struct cfq_io_cq *cic, struct bio *bio,
472
				       gfp_t gfp_mask);
473

474 475 476 477 478 479
static inline struct cfq_io_cq *icq_to_cic(struct io_cq *icq)
{
	/* cic->icq is the first member, %NULL will convert to %NULL */
	return container_of(icq, struct cfq_io_cq, icq);
}

480 481 482 483 484 485 486 487
static inline struct cfq_io_cq *cfq_cic_lookup(struct cfq_data *cfqd,
					       struct io_context *ioc)
{
	if (ioc)
		return icq_to_cic(ioc_lookup_icq(ioc, cfqd->queue));
	return NULL;
}

488
static inline struct cfq_queue *cic_to_cfqq(struct cfq_io_cq *cic, bool is_sync)
489
{
490
	return cic->cfqq[is_sync];
491 492
}

493 494
static inline void cic_set_cfqq(struct cfq_io_cq *cic, struct cfq_queue *cfqq,
				bool is_sync)
495
{
496
	cic->cfqq[is_sync] = cfqq;
497 498
}

499
static inline struct cfq_data *cic_to_cfqd(struct cfq_io_cq *cic)
500
{
501
	return cic->icq.q->elevator->elevator_data;
502 503
}

504 505 506 507
/*
 * We regard a request as SYNC, if it's either a read or has the SYNC bit
 * set (in which case it could also be direct WRITE).
 */
508
static inline bool cfq_bio_sync(struct bio *bio)
509
{
510
	return bio_data_dir(bio) == READ || (bio->bi_rw & REQ_SYNC);
511
}
L
Linus Torvalds 已提交
512

A
Andrew Morton 已提交
513 514 515 516
/*
 * scheduler run of queue, if there are requests pending and no one in the
 * driver that will restart queueing
 */
517
static inline void cfq_schedule_dispatch(struct cfq_data *cfqd)
A
Andrew Morton 已提交
518
{
519 520
	if (cfqd->busy_queues) {
		cfq_log(cfqd, "schedule dispatch");
521
		kblockd_schedule_work(cfqd->queue, &cfqd->unplug_work);
522
	}
A
Andrew Morton 已提交
523 524
}

525 526 527 528 529
/*
 * 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.
 */
530
static inline int cfq_prio_slice(struct cfq_data *cfqd, bool sync,
531
				 unsigned short prio)
532
{
533
	const int base_slice = cfqd->cfq_slice[sync];
534

535 536 537 538
	WARN_ON(prio >= IOPRIO_BE_NR);

	return base_slice + (base_slice/CFQ_SLICE_SCALE * (4 - prio));
}
539

540 541 542 543
static inline int
cfq_prio_to_slice(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	return cfq_prio_slice(cfqd, cfq_cfqq_sync(cfqq), cfqq->ioprio);
544 545
}

546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578
static inline u64 cfq_scale_slice(unsigned long delta, struct cfq_group *cfqg)
{
	u64 d = delta << CFQ_SERVICE_SHIFT;

	d = d * BLKIO_WEIGHT_DEFAULT;
	do_div(d, cfqg->weight);
	return d;
}

static inline u64 max_vdisktime(u64 min_vdisktime, u64 vdisktime)
{
	s64 delta = (s64)(vdisktime - min_vdisktime);
	if (delta > 0)
		min_vdisktime = vdisktime;

	return min_vdisktime;
}

static inline u64 min_vdisktime(u64 min_vdisktime, u64 vdisktime)
{
	s64 delta = (s64)(vdisktime - min_vdisktime);
	if (delta < 0)
		min_vdisktime = vdisktime;

	return min_vdisktime;
}

static void update_min_vdisktime(struct cfq_rb_root *st)
{
	struct cfq_group *cfqg;

	if (st->left) {
		cfqg = rb_entry_cfqg(st->left);
579 580
		st->min_vdisktime = max_vdisktime(st->min_vdisktime,
						  cfqg->vdisktime);
581 582 583
	}
}

584 585 586 587 588 589
/*
 * get averaged number of queues of RT/BE priority.
 * average is updated, with a formula that gives more weight to higher numbers,
 * to quickly follows sudden increases and decrease slowly
 */

590 591
static inline unsigned cfq_group_get_avg_queues(struct cfq_data *cfqd,
					struct cfq_group *cfqg, bool rt)
592
{
593 594 595
	unsigned min_q, max_q;
	unsigned mult  = cfq_hist_divisor - 1;
	unsigned round = cfq_hist_divisor / 2;
596
	unsigned busy = cfq_group_busy_queues_wl(rt, cfqd, cfqg);
597

598 599 600
	min_q = min(cfqg->busy_queues_avg[rt], busy);
	max_q = max(cfqg->busy_queues_avg[rt], busy);
	cfqg->busy_queues_avg[rt] = (mult * max_q + min_q + round) /
601
		cfq_hist_divisor;
602 603 604 605 606 607 608 609 610
	return cfqg->busy_queues_avg[rt];
}

static inline unsigned
cfq_group_slice(struct cfq_data *cfqd, struct cfq_group *cfqg)
{
	struct cfq_rb_root *st = &cfqd->grp_service_tree;

	return cfq_target_latency * cfqg->weight / st->total_weight;
611 612
}

613
static inline unsigned
614
cfq_scaled_cfqq_slice(struct cfq_data *cfqd, struct cfq_queue *cfqq)
615
{
616 617
	unsigned slice = cfq_prio_to_slice(cfqd, cfqq);
	if (cfqd->cfq_latency) {
618 619 620 621 622 623
		/*
		 * interested queues (we consider only the ones with the same
		 * priority class in the cfq group)
		 */
		unsigned iq = cfq_group_get_avg_queues(cfqd, cfqq->cfqg,
						cfq_class_rt(cfqq));
624 625
		unsigned sync_slice = cfqd->cfq_slice[1];
		unsigned expect_latency = sync_slice * iq;
626 627 628
		unsigned group_slice = cfq_group_slice(cfqd, cfqq->cfqg);

		if (expect_latency > group_slice) {
629 630 631 632 633 634 635
			unsigned base_low_slice = 2 * cfqd->cfq_slice_idle;
			/* scale low_slice according to IO priority
			 * and sync vs async */
			unsigned low_slice =
				min(slice, base_low_slice * slice / sync_slice);
			/* the adapted slice value is scaled to fit all iqs
			 * into the target latency */
636
			slice = max(slice * group_slice / expect_latency,
637 638 639
				    low_slice);
		}
	}
640 641 642 643 644 645
	return slice;
}

static inline void
cfq_set_prio_slice(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
646
	unsigned slice = cfq_scaled_cfqq_slice(cfqd, cfqq);
647

648
	cfqq->slice_start = jiffies;
649
	cfqq->slice_end = jiffies + slice;
650
	cfqq->allocated_slice = slice;
651
	cfq_log_cfqq(cfqd, cfqq, "set_slice=%lu", cfqq->slice_end - jiffies);
652 653 654 655 656 657 658
}

/*
 * We need to wrap this check in cfq_cfqq_slice_new(), since ->slice_end
 * isn't valid until the first request from the dispatch is activated
 * and the slice time set.
 */
659
static inline bool cfq_slice_used(struct cfq_queue *cfqq)
660 661
{
	if (cfq_cfqq_slice_new(cfqq))
S
Shaohua Li 已提交
662
		return false;
663
	if (time_before(jiffies, cfqq->slice_end))
S
Shaohua Li 已提交
664
		return false;
665

S
Shaohua Li 已提交
666
	return true;
667 668
}

L
Linus Torvalds 已提交
669
/*
J
Jens Axboe 已提交
670
 * Lifted from AS - choose which of rq1 and rq2 that is best served now.
L
Linus Torvalds 已提交
671
 * We choose the request that is closest to the head right now. Distance
672
 * behind the head is penalized and only allowed to a certain extent.
L
Linus Torvalds 已提交
673
 */
J
Jens Axboe 已提交
674
static struct request *
675
cfq_choose_req(struct cfq_data *cfqd, struct request *rq1, struct request *rq2, sector_t last)
L
Linus Torvalds 已提交
676
{
677
	sector_t s1, s2, d1 = 0, d2 = 0;
L
Linus Torvalds 已提交
678
	unsigned long back_max;
679 680 681
#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 已提交
682

J
Jens Axboe 已提交
683 684 685 686
	if (rq1 == NULL || rq1 == rq2)
		return rq2;
	if (rq2 == NULL)
		return rq1;
J
Jens Axboe 已提交
687

688 689 690
	if (rq_is_sync(rq1) != rq_is_sync(rq2))
		return rq_is_sync(rq1) ? rq1 : rq2;

691 692
	if ((rq1->cmd_flags ^ rq2->cmd_flags) & REQ_PRIO)
		return rq1->cmd_flags & REQ_PRIO ? rq1 : rq2;
693

694 695
	s1 = blk_rq_pos(rq1);
	s2 = blk_rq_pos(rq2);
L
Linus Torvalds 已提交
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711

	/*
	 * 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
712
		wrap |= CFQ_RQ1_WRAP;
L
Linus Torvalds 已提交
713 714 715 716 717 718

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

	/* Found required data */
722 723 724 725 726 727

	/*
	 * By doing switch() on the bit mask "wrap" we avoid having to
	 * check two variables for all permutations: --> faster!
	 */
	switch (wrap) {
J
Jens Axboe 已提交
728
	case 0: /* common case for CFQ: rq1 and rq2 not wrapped */
729
		if (d1 < d2)
J
Jens Axboe 已提交
730
			return rq1;
731
		else if (d2 < d1)
J
Jens Axboe 已提交
732
			return rq2;
733 734
		else {
			if (s1 >= s2)
J
Jens Axboe 已提交
735
				return rq1;
736
			else
J
Jens Axboe 已提交
737
				return rq2;
738
		}
L
Linus Torvalds 已提交
739

740
	case CFQ_RQ2_WRAP:
J
Jens Axboe 已提交
741
		return rq1;
742
	case CFQ_RQ1_WRAP:
J
Jens Axboe 已提交
743 744
		return rq2;
	case (CFQ_RQ1_WRAP|CFQ_RQ2_WRAP): /* both rqs wrapped */
745 746 747 748 749 750 751 752
	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 已提交
753
			return rq1;
L
Linus Torvalds 已提交
754
		else
J
Jens Axboe 已提交
755
			return rq2;
L
Linus Torvalds 已提交
756 757 758
	}
}

759 760 761
/*
 * The below is leftmost cache rbtree addon
 */
762
static struct cfq_queue *cfq_rb_first(struct cfq_rb_root *root)
763
{
764 765 766 767
	/* Service tree is empty */
	if (!root->count)
		return NULL;

768 769 770
	if (!root->left)
		root->left = rb_first(&root->rb);

771 772 773 774
	if (root->left)
		return rb_entry(root->left, struct cfq_queue, rb_node);

	return NULL;
775 776
}

777 778 779 780 781 782 783 784 785 786 787
static struct cfq_group *cfq_rb_first_group(struct cfq_rb_root *root)
{
	if (!root->left)
		root->left = rb_first(&root->rb);

	if (root->left)
		return rb_entry_cfqg(root->left);

	return NULL;
}

788 789 790 791 792 793
static void rb_erase_init(struct rb_node *n, struct rb_root *root)
{
	rb_erase(n, root);
	RB_CLEAR_NODE(n);
}

794 795 796 797
static void cfq_rb_erase(struct rb_node *n, struct cfq_rb_root *root)
{
	if (root->left == n)
		root->left = NULL;
798
	rb_erase_init(n, &root->rb);
799
	--root->count;
800 801
}

L
Linus Torvalds 已提交
802 803 804
/*
 * would be nice to take fifo expire time into account as well
 */
J
Jens Axboe 已提交
805 806 807
static struct request *
cfq_find_next_rq(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		  struct request *last)
L
Linus Torvalds 已提交
808
{
809 810
	struct rb_node *rbnext = rb_next(&last->rb_node);
	struct rb_node *rbprev = rb_prev(&last->rb_node);
J
Jens Axboe 已提交
811
	struct request *next = NULL, *prev = NULL;
L
Linus Torvalds 已提交
812

813
	BUG_ON(RB_EMPTY_NODE(&last->rb_node));
L
Linus Torvalds 已提交
814 815

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

818
	if (rbnext)
J
Jens Axboe 已提交
819
		next = rb_entry_rq(rbnext);
820 821 822
	else {
		rbnext = rb_first(&cfqq->sort_list);
		if (rbnext && rbnext != &last->rb_node)
J
Jens Axboe 已提交
823
			next = rb_entry_rq(rbnext);
824
	}
L
Linus Torvalds 已提交
825

826
	return cfq_choose_req(cfqd, next, prev, blk_rq_pos(last));
L
Linus Torvalds 已提交
827 828
}

829 830
static unsigned long cfq_slice_offset(struct cfq_data *cfqd,
				      struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
831
{
832 833 834
	/*
	 * just an approximation, should be ok.
	 */
835
	return (cfqq->cfqg->nr_cfqq - 1) * (cfq_prio_slice(cfqd, 1, 0) -
836
		       cfq_prio_slice(cfqd, cfq_cfqq_sync(cfqq), cfqq->ioprio));
837 838
}

839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873
static inline s64
cfqg_key(struct cfq_rb_root *st, struct cfq_group *cfqg)
{
	return cfqg->vdisktime - st->min_vdisktime;
}

static void
__cfq_group_service_tree_add(struct cfq_rb_root *st, struct cfq_group *cfqg)
{
	struct rb_node **node = &st->rb.rb_node;
	struct rb_node *parent = NULL;
	struct cfq_group *__cfqg;
	s64 key = cfqg_key(st, cfqg);
	int left = 1;

	while (*node != NULL) {
		parent = *node;
		__cfqg = rb_entry_cfqg(parent);

		if (key < cfqg_key(st, __cfqg))
			node = &parent->rb_left;
		else {
			node = &parent->rb_right;
			left = 0;
		}
	}

	if (left)
		st->left = &cfqg->rb_node;

	rb_link_node(&cfqg->rb_node, parent, node);
	rb_insert_color(&cfqg->rb_node, &st->rb);
}

static void
874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894
cfq_update_group_weight(struct cfq_group *cfqg)
{
	BUG_ON(!RB_EMPTY_NODE(&cfqg->rb_node));
	if (cfqg->needs_update) {
		cfqg->weight = cfqg->new_weight;
		cfqg->needs_update = false;
	}
}

static void
cfq_group_service_tree_add(struct cfq_rb_root *st, struct cfq_group *cfqg)
{
	BUG_ON(!RB_EMPTY_NODE(&cfqg->rb_node));

	cfq_update_group_weight(cfqg);
	__cfq_group_service_tree_add(st, cfqg);
	st->total_weight += cfqg->weight;
}

static void
cfq_group_notify_queue_add(struct cfq_data *cfqd, struct cfq_group *cfqg)
895 896 897 898 899 900
{
	struct cfq_rb_root *st = &cfqd->grp_service_tree;
	struct cfq_group *__cfqg;
	struct rb_node *n;

	cfqg->nr_cfqq++;
G
Gui Jianfeng 已提交
901
	if (!RB_EMPTY_NODE(&cfqg->rb_node))
902 903 904 905 906
		return;

	/*
	 * Currently put the group at the end. Later implement something
	 * so that groups get lesser vtime based on their weights, so that
L
Lucas De Marchi 已提交
907
	 * if group does not loose all if it was not continuously backlogged.
908 909 910 911 912 913 914
	 */
	n = rb_last(&st->rb);
	if (n) {
		__cfqg = rb_entry_cfqg(n);
		cfqg->vdisktime = __cfqg->vdisktime + CFQ_IDLE_DELAY;
	} else
		cfqg->vdisktime = st->min_vdisktime;
915 916
	cfq_group_service_tree_add(st, cfqg);
}
917

918 919 920 921 922 923
static void
cfq_group_service_tree_del(struct cfq_rb_root *st, struct cfq_group *cfqg)
{
	st->total_weight -= cfqg->weight;
	if (!RB_EMPTY_NODE(&cfqg->rb_node))
		cfq_rb_erase(&cfqg->rb_node, st);
924 925 926
}

static void
927
cfq_group_notify_queue_del(struct cfq_data *cfqd, struct cfq_group *cfqg)
928 929 930 931 932
{
	struct cfq_rb_root *st = &cfqd->grp_service_tree;

	BUG_ON(cfqg->nr_cfqq < 1);
	cfqg->nr_cfqq--;
933

934 935 936 937
	/* If there are other cfq queues under this group, don't delete it */
	if (cfqg->nr_cfqq)
		return;

V
Vivek Goyal 已提交
938
	cfq_log_cfqg(cfqd, cfqg, "del_from_rr group");
939
	cfq_group_service_tree_del(st, cfqg);
940
	cfqg->saved_workload_slice = 0;
941 942
	cfq_blkiocg_update_dequeue_stats(cfqg_to_blkg(cfqg),
					 &blkio_policy_cfq, 1);
943 944
}

945 946
static inline unsigned int cfq_cfqq_slice_usage(struct cfq_queue *cfqq,
						unsigned int *unaccounted_time)
947
{
948
	unsigned int slice_used;
949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964

	/*
	 * Queue got expired before even a single request completed or
	 * got expired immediately after first request completion.
	 */
	if (!cfqq->slice_start || cfqq->slice_start == jiffies) {
		/*
		 * Also charge the seek time incurred to the group, otherwise
		 * if there are mutiple queues in the group, each can dispatch
		 * a single request on seeky media and cause lots of seek time
		 * and group will never know it.
		 */
		slice_used = max_t(unsigned, (jiffies - cfqq->dispatch_start),
					1);
	} else {
		slice_used = jiffies - cfqq->slice_start;
965 966
		if (slice_used > cfqq->allocated_slice) {
			*unaccounted_time = slice_used - cfqq->allocated_slice;
967
			slice_used = cfqq->allocated_slice;
968 969 970 971
		}
		if (time_after(cfqq->slice_start, cfqq->dispatch_start))
			*unaccounted_time += cfqq->slice_start -
					cfqq->dispatch_start;
972 973 974 975 976 977
	}

	return slice_used;
}

static void cfq_group_served(struct cfq_data *cfqd, struct cfq_group *cfqg,
978
				struct cfq_queue *cfqq)
979 980
{
	struct cfq_rb_root *st = &cfqd->grp_service_tree;
981
	unsigned int used_sl, charge, unaccounted_sl = 0;
982 983 984 985
	int nr_sync = cfqg->nr_cfqq - cfqg_busy_async_queues(cfqd, cfqg)
			- cfqg->service_tree_idle.count;

	BUG_ON(nr_sync < 0);
986
	used_sl = charge = cfq_cfqq_slice_usage(cfqq, &unaccounted_sl);
987

988 989 990 991
	if (iops_mode(cfqd))
		charge = cfqq->slice_dispatch;
	else if (!cfq_cfqq_sync(cfqq) && !nr_sync)
		charge = cfqq->allocated_slice;
992 993

	/* Can't update vdisktime while group is on service tree */
994
	cfq_group_service_tree_del(st, cfqg);
995
	cfqg->vdisktime += cfq_scale_slice(charge, cfqg);
996 997
	/* If a new weight was requested, update now, off tree */
	cfq_group_service_tree_add(st, cfqg);
998 999 1000 1001 1002 1003 1004 1005 1006

	/* This group is being expired. Save the context */
	if (time_after(cfqd->workload_expires, jiffies)) {
		cfqg->saved_workload_slice = cfqd->workload_expires
						- jiffies;
		cfqg->saved_workload = cfqd->serving_type;
		cfqg->saved_serving_prio = cfqd->serving_prio;
	} else
		cfqg->saved_workload_slice = 0;
V
Vivek Goyal 已提交
1007 1008 1009

	cfq_log_cfqg(cfqd, cfqg, "served: vt=%llu min_vt=%llu", cfqg->vdisktime,
					st->min_vdisktime);
1010 1011 1012 1013
	cfq_log_cfqq(cfqq->cfqd, cfqq,
		     "sl_used=%u disp=%u charge=%u iops=%u sect=%lu",
		     used_sl, cfqq->slice_dispatch, charge,
		     iops_mode(cfqd), cfqq->nr_sectors);
1014 1015 1016
	cfq_blkiocg_update_timeslice_used(cfqg_to_blkg(cfqg), &blkio_policy_cfq,
					  used_sl, unaccounted_sl);
	cfq_blkiocg_set_start_empty_time(cfqg_to_blkg(cfqg), &blkio_policy_cfq);
1017 1018
}

1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
/**
 * cfq_init_cfqg_base - initialize base part of a cfq_group
 * @cfqg: cfq_group to initialize
 *
 * Initialize the base part which is used whether %CONFIG_CFQ_GROUP_IOSCHED
 * is enabled or not.
 */
static void cfq_init_cfqg_base(struct cfq_group *cfqg)
{
	struct cfq_rb_root *st;
	int i, j;

	for_each_cfqg_st(cfqg, i, j, st)
		*st = CFQ_RB_ROOT;
	RB_CLEAR_NODE(&cfqg->rb_node);

	cfqg->ttime.last_end_request = jiffies;
}

1038
#ifdef CONFIG_CFQ_GROUP_IOSCHED
1039 1040
static void cfq_update_blkio_group_weight(struct request_queue *q,
					  struct blkio_group *blkg,
P
Paul Bolle 已提交
1041
					  unsigned int weight)
1042
{
1043 1044
	struct cfq_group *cfqg = blkg_to_cfqg(blkg);

1045 1046
	cfqg->new_weight = weight;
	cfqg->needs_update = true;
1047 1048
}

1049
static void cfq_init_blkio_group(struct blkio_group *blkg)
1050
{
1051
	struct cfq_group *cfqg = blkg_to_cfqg(blkg);
1052

1053
	cfq_init_cfqg_base(cfqg);
1054
	cfqg->weight = blkg->blkcg->weight;
1055 1056 1057
}

/*
1058 1059
 * Search for the cfq group current task belongs to. request_queue lock must
 * be held.
1060
 */
1061 1062
static struct cfq_group *cfq_lookup_create_cfqg(struct cfq_data *cfqd,
						struct blkio_cgroup *blkcg)
1063
{
1064
	struct request_queue *q = cfqd->queue;
1065
	struct cfq_group *cfqg = NULL;
1066

1067 1068 1069 1070 1071
	/* avoid lookup for the common case where there's no blkio cgroup */
	if (blkcg == &blkio_root_cgroup) {
		cfqg = cfqd->root_group;
	} else {
		struct blkio_group *blkg;
1072

1073 1074
		blkg = blkg_lookup_create(blkcg, q, BLKIO_POLICY_PROP, false);
		if (!IS_ERR(blkg))
1075
			cfqg = blkg_to_cfqg(blkg);
1076
	}
1077

1078 1079 1080 1081 1082 1083 1084
	return cfqg;
}

static void cfq_link_cfqq_cfqg(struct cfq_queue *cfqq, struct cfq_group *cfqg)
{
	/* Currently, all async queues are mapped to root group */
	if (!cfq_cfqq_sync(cfqq))
1085
		cfqg = cfqq->cfqd->root_group;
1086 1087

	cfqq->cfqg = cfqg;
1088
	/* cfqq reference on cfqg */
T
Tejun Heo 已提交
1089
	blkg_get(cfqg_to_blkg(cfqg));
1090 1091
}

1092
#else /* GROUP_IOSCHED */
1093 1094
static struct cfq_group *cfq_lookup_create_cfqg(struct cfq_data *cfqd,
						struct blkio_cgroup *blkcg)
1095
{
1096
	return cfqd->root_group;
1097
}
1098

1099 1100 1101 1102 1103 1104 1105
static inline void
cfq_link_cfqq_cfqg(struct cfq_queue *cfqq, struct cfq_group *cfqg) {
	cfqq->cfqg = cfqg;
}

#endif /* GROUP_IOSCHED */

1106
/*
1107
 * The cfqd->service_trees holds all pending cfq_queue's that have
1108 1109 1110
 * requests waiting to be processed. It is sorted in the order that
 * we will service the queues.
 */
1111
static void cfq_service_tree_add(struct cfq_data *cfqd, struct cfq_queue *cfqq,
1112
				 bool add_front)
1113
{
1114 1115
	struct rb_node **p, *parent;
	struct cfq_queue *__cfqq;
1116
	unsigned long rb_key;
1117
	struct cfq_rb_root *service_tree;
1118
	int left;
1119
	int new_cfqq = 1;
1120

1121
	service_tree = service_tree_for(cfqq->cfqg, cfqq_prio(cfqq),
1122
						cfqq_type(cfqq));
1123 1124
	if (cfq_class_idle(cfqq)) {
		rb_key = CFQ_IDLE_DELAY;
1125
		parent = rb_last(&service_tree->rb);
1126 1127 1128 1129 1130 1131
		if (parent && parent != &cfqq->rb_node) {
			__cfqq = rb_entry(parent, struct cfq_queue, rb_node);
			rb_key += __cfqq->rb_key;
		} else
			rb_key += jiffies;
	} else if (!add_front) {
1132 1133 1134 1135 1136 1137
		/*
		 * Get our rb key offset. Subtract any residual slice
		 * value carried from last service. A negative resid
		 * count indicates slice overrun, and this should position
		 * the next service time further away in the tree.
		 */
1138
		rb_key = cfq_slice_offset(cfqd, cfqq) + jiffies;
1139
		rb_key -= cfqq->slice_resid;
1140
		cfqq->slice_resid = 0;
1141 1142
	} else {
		rb_key = -HZ;
1143
		__cfqq = cfq_rb_first(service_tree);
1144 1145
		rb_key += __cfqq ? __cfqq->rb_key : jiffies;
	}
L
Linus Torvalds 已提交
1146

1147
	if (!RB_EMPTY_NODE(&cfqq->rb_node)) {
1148
		new_cfqq = 0;
1149
		/*
1150
		 * same position, nothing more to do
1151
		 */
1152 1153
		if (rb_key == cfqq->rb_key &&
		    cfqq->service_tree == service_tree)
1154
			return;
L
Linus Torvalds 已提交
1155

1156 1157
		cfq_rb_erase(&cfqq->rb_node, cfqq->service_tree);
		cfqq->service_tree = NULL;
L
Linus Torvalds 已提交
1158
	}
1159

1160
	left = 1;
1161
	parent = NULL;
1162 1163
	cfqq->service_tree = service_tree;
	p = &service_tree->rb.rb_node;
1164
	while (*p) {
1165
		struct rb_node **n;
1166

1167 1168 1169
		parent = *p;
		__cfqq = rb_entry(parent, struct cfq_queue, rb_node);

1170
		/*
1171
		 * sort by key, that represents service time.
1172
		 */
1173
		if (time_before(rb_key, __cfqq->rb_key))
1174
			n = &(*p)->rb_left;
1175
		else {
1176
			n = &(*p)->rb_right;
1177
			left = 0;
1178
		}
1179 1180

		p = n;
1181 1182
	}

1183
	if (left)
1184
		service_tree->left = &cfqq->rb_node;
1185

1186 1187
	cfqq->rb_key = rb_key;
	rb_link_node(&cfqq->rb_node, parent, p);
1188 1189
	rb_insert_color(&cfqq->rb_node, &service_tree->rb);
	service_tree->count++;
1190
	if (add_front || !new_cfqq)
1191
		return;
1192
	cfq_group_notify_queue_add(cfqd, cfqq->cfqg);
L
Linus Torvalds 已提交
1193 1194
}

1195
static struct cfq_queue *
1196 1197 1198
cfq_prio_tree_lookup(struct cfq_data *cfqd, struct rb_root *root,
		     sector_t sector, struct rb_node **ret_parent,
		     struct rb_node ***rb_link)
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
{
	struct rb_node **p, *parent;
	struct cfq_queue *cfqq = NULL;

	parent = NULL;
	p = &root->rb_node;
	while (*p) {
		struct rb_node **n;

		parent = *p;
		cfqq = rb_entry(parent, struct cfq_queue, p_node);

		/*
		 * Sort strictly based on sector.  Smallest to the left,
		 * largest to the right.
		 */
1215
		if (sector > blk_rq_pos(cfqq->next_rq))
1216
			n = &(*p)->rb_right;
1217
		else if (sector < blk_rq_pos(cfqq->next_rq))
1218 1219 1220 1221
			n = &(*p)->rb_left;
		else
			break;
		p = n;
1222
		cfqq = NULL;
1223 1224 1225 1226 1227
	}

	*ret_parent = parent;
	if (rb_link)
		*rb_link = p;
1228
	return cfqq;
1229 1230 1231 1232 1233 1234 1235
}

static void cfq_prio_tree_add(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	struct rb_node **p, *parent;
	struct cfq_queue *__cfqq;

1236 1237 1238 1239
	if (cfqq->p_root) {
		rb_erase(&cfqq->p_node, cfqq->p_root);
		cfqq->p_root = NULL;
	}
1240 1241 1242 1243 1244 1245

	if (cfq_class_idle(cfqq))
		return;
	if (!cfqq->next_rq)
		return;

1246
	cfqq->p_root = &cfqd->prio_trees[cfqq->org_ioprio];
1247 1248
	__cfqq = cfq_prio_tree_lookup(cfqd, cfqq->p_root,
				      blk_rq_pos(cfqq->next_rq), &parent, &p);
1249 1250
	if (!__cfqq) {
		rb_link_node(&cfqq->p_node, parent, p);
1251 1252 1253
		rb_insert_color(&cfqq->p_node, cfqq->p_root);
	} else
		cfqq->p_root = NULL;
1254 1255
}

1256 1257 1258
/*
 * Update cfqq's position in the service tree.
 */
1259
static void cfq_resort_rr_list(struct cfq_data *cfqd, struct cfq_queue *cfqq)
J
Jens Axboe 已提交
1260 1261 1262 1263
{
	/*
	 * Resorting requires the cfqq to be on the RR list already.
	 */
1264
	if (cfq_cfqq_on_rr(cfqq)) {
1265
		cfq_service_tree_add(cfqd, cfqq, 0);
1266 1267
		cfq_prio_tree_add(cfqd, cfqq);
	}
J
Jens Axboe 已提交
1268 1269
}

L
Linus Torvalds 已提交
1270 1271
/*
 * add to busy list of queues for service, trying to be fair in ordering
1272
 * the pending list according to last request service
L
Linus Torvalds 已提交
1273
 */
J
Jens Axboe 已提交
1274
static void cfq_add_cfqq_rr(struct cfq_data *cfqd, struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
1275
{
1276
	cfq_log_cfqq(cfqd, cfqq, "add_to_rr");
J
Jens Axboe 已提交
1277 1278
	BUG_ON(cfq_cfqq_on_rr(cfqq));
	cfq_mark_cfqq_on_rr(cfqq);
L
Linus Torvalds 已提交
1279
	cfqd->busy_queues++;
1280 1281
	if (cfq_cfqq_sync(cfqq))
		cfqd->busy_sync_queues++;
L
Linus Torvalds 已提交
1282

1283
	cfq_resort_rr_list(cfqd, cfqq);
L
Linus Torvalds 已提交
1284 1285
}

1286 1287 1288 1289
/*
 * Called when the cfqq no longer has requests pending, remove it from
 * the service tree.
 */
J
Jens Axboe 已提交
1290
static void cfq_del_cfqq_rr(struct cfq_data *cfqd, struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
1291
{
1292
	cfq_log_cfqq(cfqd, cfqq, "del_from_rr");
J
Jens Axboe 已提交
1293 1294
	BUG_ON(!cfq_cfqq_on_rr(cfqq));
	cfq_clear_cfqq_on_rr(cfqq);
L
Linus Torvalds 已提交
1295

1296 1297 1298 1299
	if (!RB_EMPTY_NODE(&cfqq->rb_node)) {
		cfq_rb_erase(&cfqq->rb_node, cfqq->service_tree);
		cfqq->service_tree = NULL;
	}
1300 1301 1302 1303
	if (cfqq->p_root) {
		rb_erase(&cfqq->p_node, cfqq->p_root);
		cfqq->p_root = NULL;
	}
1304

1305
	cfq_group_notify_queue_del(cfqd, cfqq->cfqg);
L
Linus Torvalds 已提交
1306 1307
	BUG_ON(!cfqd->busy_queues);
	cfqd->busy_queues--;
1308 1309
	if (cfq_cfqq_sync(cfqq))
		cfqd->busy_sync_queues--;
L
Linus Torvalds 已提交
1310 1311 1312 1313 1314
}

/*
 * rb tree support functions
 */
J
Jens Axboe 已提交
1315
static void cfq_del_rq_rb(struct request *rq)
L
Linus Torvalds 已提交
1316
{
J
Jens Axboe 已提交
1317 1318
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
	const int sync = rq_is_sync(rq);
L
Linus Torvalds 已提交
1319

1320 1321
	BUG_ON(!cfqq->queued[sync]);
	cfqq->queued[sync]--;
L
Linus Torvalds 已提交
1322

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

1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335
	if (cfq_cfqq_on_rr(cfqq) && RB_EMPTY_ROOT(&cfqq->sort_list)) {
		/*
		 * Queue will be deleted from service tree when we actually
		 * expire it later. Right now just remove it from prio tree
		 * as it is empty.
		 */
		if (cfqq->p_root) {
			rb_erase(&cfqq->p_node, cfqq->p_root);
			cfqq->p_root = NULL;
		}
	}
L
Linus Torvalds 已提交
1336 1337
}

J
Jens Axboe 已提交
1338
static void cfq_add_rq_rb(struct request *rq)
L
Linus Torvalds 已提交
1339
{
J
Jens Axboe 已提交
1340
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
L
Linus Torvalds 已提交
1341
	struct cfq_data *cfqd = cfqq->cfqd;
1342
	struct request *prev;
L
Linus Torvalds 已提交
1343

1344
	cfqq->queued[rq_is_sync(rq)]++;
L
Linus Torvalds 已提交
1345

1346
	elv_rb_add(&cfqq->sort_list, rq);
1347 1348 1349

	if (!cfq_cfqq_on_rr(cfqq))
		cfq_add_cfqq_rr(cfqd, cfqq);
1350 1351 1352 1353

	/*
	 * check if this request is a better next-serve candidate
	 */
1354
	prev = cfqq->next_rq;
1355
	cfqq->next_rq = cfq_choose_req(cfqd, cfqq->next_rq, rq, cfqd->last_position);
1356 1357 1358 1359 1360 1361 1362

	/*
	 * adjust priority tree position, if ->next_rq changes
	 */
	if (prev != cfqq->next_rq)
		cfq_prio_tree_add(cfqd, cfqq);

1363
	BUG_ON(!cfqq->next_rq);
L
Linus Torvalds 已提交
1364 1365
}

J
Jens Axboe 已提交
1366
static void cfq_reposition_rq_rb(struct cfq_queue *cfqq, struct request *rq)
L
Linus Torvalds 已提交
1367
{
1368 1369
	elv_rb_del(&cfqq->sort_list, rq);
	cfqq->queued[rq_is_sync(rq)]--;
1370
	cfq_blkiocg_update_io_remove_stats(cfqg_to_blkg(RQ_CFQG(rq)),
1371 1372
					   &blkio_policy_cfq, rq_data_dir(rq),
					   rq_is_sync(rq));
J
Jens Axboe 已提交
1373
	cfq_add_rq_rb(rq);
1374
	cfq_blkiocg_update_io_add_stats(cfqg_to_blkg(RQ_CFQG(rq)),
1375
					&blkio_policy_cfq,
1376 1377
					cfqg_to_blkg(cfqq->cfqd->serving_group),
					rq_data_dir(rq), rq_is_sync(rq));
L
Linus Torvalds 已提交
1378 1379
}

1380 1381
static struct request *
cfq_find_rq_fmerge(struct cfq_data *cfqd, struct bio *bio)
L
Linus Torvalds 已提交
1382
{
1383
	struct task_struct *tsk = current;
1384
	struct cfq_io_cq *cic;
1385
	struct cfq_queue *cfqq;
L
Linus Torvalds 已提交
1386

1387
	cic = cfq_cic_lookup(cfqd, tsk->io_context);
1388 1389 1390 1391
	if (!cic)
		return NULL;

	cfqq = cic_to_cfqq(cic, cfq_bio_sync(bio));
1392 1393 1394
	if (cfqq) {
		sector_t sector = bio->bi_sector + bio_sectors(bio);

1395
		return elv_rb_find(&cfqq->sort_list, sector);
1396
	}
L
Linus Torvalds 已提交
1397 1398 1399 1400

	return NULL;
}

1401
static void cfq_activate_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
1402
{
1403
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
1404

1405
	cfqd->rq_in_driver++;
1406
	cfq_log_cfqq(cfqd, RQ_CFQQ(rq), "activate rq, drv=%d",
1407
						cfqd->rq_in_driver);
1408

1409
	cfqd->last_position = blk_rq_pos(rq) + blk_rq_sectors(rq);
L
Linus Torvalds 已提交
1410 1411
}

1412
static void cfq_deactivate_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
1413
{
1414 1415
	struct cfq_data *cfqd = q->elevator->elevator_data;

1416 1417
	WARN_ON(!cfqd->rq_in_driver);
	cfqd->rq_in_driver--;
1418
	cfq_log_cfqq(cfqd, RQ_CFQQ(rq), "deactivate rq, drv=%d",
1419
						cfqd->rq_in_driver);
L
Linus Torvalds 已提交
1420 1421
}

1422
static void cfq_remove_request(struct request *rq)
L
Linus Torvalds 已提交
1423
{
J
Jens Axboe 已提交
1424
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
1425

J
Jens Axboe 已提交
1426 1427
	if (cfqq->next_rq == rq)
		cfqq->next_rq = cfq_find_next_rq(cfqq->cfqd, cfqq, rq);
L
Linus Torvalds 已提交
1428

1429
	list_del_init(&rq->queuelist);
J
Jens Axboe 已提交
1430
	cfq_del_rq_rb(rq);
1431

1432
	cfqq->cfqd->rq_queued--;
1433
	cfq_blkiocg_update_io_remove_stats(cfqg_to_blkg(RQ_CFQG(rq)),
1434 1435
					   &blkio_policy_cfq, rq_data_dir(rq),
					   rq_is_sync(rq));
1436 1437 1438
	if (rq->cmd_flags & REQ_PRIO) {
		WARN_ON(!cfqq->prio_pending);
		cfqq->prio_pending--;
1439
	}
L
Linus Torvalds 已提交
1440 1441
}

1442 1443
static int cfq_merge(struct request_queue *q, struct request **req,
		     struct bio *bio)
L
Linus Torvalds 已提交
1444 1445 1446 1447
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
	struct request *__rq;

1448
	__rq = cfq_find_rq_fmerge(cfqd, bio);
1449
	if (__rq && elv_rq_merge_ok(__rq, bio)) {
1450 1451
		*req = __rq;
		return ELEVATOR_FRONT_MERGE;
L
Linus Torvalds 已提交
1452 1453 1454 1455 1456
	}

	return ELEVATOR_NO_MERGE;
}

1457
static void cfq_merged_request(struct request_queue *q, struct request *req,
1458
			       int type)
L
Linus Torvalds 已提交
1459
{
1460
	if (type == ELEVATOR_FRONT_MERGE) {
J
Jens Axboe 已提交
1461
		struct cfq_queue *cfqq = RQ_CFQQ(req);
L
Linus Torvalds 已提交
1462

J
Jens Axboe 已提交
1463
		cfq_reposition_rq_rb(cfqq, req);
L
Linus Torvalds 已提交
1464 1465 1466
	}
}

D
Divyesh Shah 已提交
1467 1468 1469
static void cfq_bio_merged(struct request_queue *q, struct request *req,
				struct bio *bio)
{
1470
	cfq_blkiocg_update_io_merged_stats(cfqg_to_blkg(RQ_CFQG(req)),
1471 1472
					   &blkio_policy_cfq, bio_data_dir(bio),
					   cfq_bio_sync(bio));
D
Divyesh Shah 已提交
1473 1474
}

L
Linus Torvalds 已提交
1475
static void
1476
cfq_merged_requests(struct request_queue *q, struct request *rq,
L
Linus Torvalds 已提交
1477 1478
		    struct request *next)
{
1479
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
1480 1481
	struct cfq_data *cfqd = q->elevator->elevator_data;

1482 1483 1484 1485
	/*
	 * reposition in fifo if next is older than rq
	 */
	if (!list_empty(&rq->queuelist) && !list_empty(&next->queuelist) &&
1486
	    time_before(rq_fifo_time(next), rq_fifo_time(rq))) {
1487
		list_move(&rq->queuelist, &next->queuelist);
1488 1489
		rq_set_fifo_time(rq, rq_fifo_time(next));
	}
1490

1491 1492
	if (cfqq->next_rq == next)
		cfqq->next_rq = rq;
1493
	cfq_remove_request(next);
1494
	cfq_blkiocg_update_io_merged_stats(cfqg_to_blkg(RQ_CFQG(rq)),
1495 1496
					   &blkio_policy_cfq, rq_data_dir(next),
					   rq_is_sync(next));
1497 1498 1499 1500 1501 1502 1503 1504 1505 1506

	cfqq = RQ_CFQQ(next);
	/*
	 * all requests of this queue are merged to other queues, delete it
	 * from the service tree. If it's the active_queue,
	 * cfq_dispatch_requests() will choose to expire it or do idle
	 */
	if (cfq_cfqq_on_rr(cfqq) && RB_EMPTY_ROOT(&cfqq->sort_list) &&
	    cfqq != cfqd->active_queue)
		cfq_del_cfqq_rr(cfqd, cfqq);
1507 1508
}

1509
static int cfq_allow_merge(struct request_queue *q, struct request *rq,
1510 1511 1512
			   struct bio *bio)
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
1513
	struct cfq_io_cq *cic;
1514 1515 1516
	struct cfq_queue *cfqq;

	/*
1517
	 * Disallow merge of a sync bio into an async request.
1518
	 */
1519
	if (cfq_bio_sync(bio) && !rq_is_sync(rq))
1520
		return false;
1521 1522

	/*
T
Tejun Heo 已提交
1523
	 * Lookup the cfqq that this bio will be queued with and allow
1524
	 * merge only if rq is queued there.
T
Tejun Heo 已提交
1525
	 */
1526 1527 1528
	cic = cfq_cic_lookup(cfqd, current->io_context);
	if (!cic)
		return false;
1529

1530
	cfqq = cic_to_cfqq(cic, cfq_bio_sync(bio));
1531
	return cfqq == RQ_CFQQ(rq);
1532 1533
}

1534 1535 1536
static inline void cfq_del_timer(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	del_timer(&cfqd->idle_slice_timer);
1537 1538
	cfq_blkiocg_update_idle_time_stats(cfqg_to_blkg(cfqq->cfqg),
					   &blkio_policy_cfq);
1539 1540
}

J
Jens Axboe 已提交
1541 1542
static void __cfq_set_active_queue(struct cfq_data *cfqd,
				   struct cfq_queue *cfqq)
1543 1544
{
	if (cfqq) {
1545 1546
		cfq_log_cfqq(cfqd, cfqq, "set_active wl_prio:%d wl_type:%d",
				cfqd->serving_prio, cfqd->serving_type);
1547 1548
		cfq_blkiocg_update_avg_queue_size_stats(cfqg_to_blkg(cfqq->cfqg),
							&blkio_policy_cfq);
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
		cfqq->slice_start = 0;
		cfqq->dispatch_start = jiffies;
		cfqq->allocated_slice = 0;
		cfqq->slice_end = 0;
		cfqq->slice_dispatch = 0;
		cfqq->nr_sectors = 0;

		cfq_clear_cfqq_wait_request(cfqq);
		cfq_clear_cfqq_must_dispatch(cfqq);
		cfq_clear_cfqq_must_alloc_slice(cfqq);
		cfq_clear_cfqq_fifo_expire(cfqq);
		cfq_mark_cfqq_slice_new(cfqq);

		cfq_del_timer(cfqd, cfqq);
1563 1564 1565 1566 1567
	}

	cfqd->active_queue = cfqq;
}

1568 1569 1570 1571 1572
/*
 * 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,
1573
		    bool timed_out)
1574
{
1575 1576
	cfq_log_cfqq(cfqd, cfqq, "slice expired t=%d", timed_out);

1577
	if (cfq_cfqq_wait_request(cfqq))
1578
		cfq_del_timer(cfqd, cfqq);
1579 1580

	cfq_clear_cfqq_wait_request(cfqq);
1581
	cfq_clear_cfqq_wait_busy(cfqq);
1582

1583 1584 1585 1586 1587 1588 1589 1590 1591
	/*
	 * If this cfqq is shared between multiple processes, check to
	 * make sure that those processes are still issuing I/Os within
	 * the mean seek distance.  If not, it may be time to break the
	 * queues apart again.
	 */
	if (cfq_cfqq_coop(cfqq) && CFQQ_SEEKY(cfqq))
		cfq_mark_cfqq_split_coop(cfqq);

1592
	/*
1593
	 * store what was left of this slice, if the queue idled/timed out
1594
	 */
1595 1596
	if (timed_out) {
		if (cfq_cfqq_slice_new(cfqq))
1597
			cfqq->slice_resid = cfq_scaled_cfqq_slice(cfqd, cfqq);
1598 1599
		else
			cfqq->slice_resid = cfqq->slice_end - jiffies;
1600 1601
		cfq_log_cfqq(cfqd, cfqq, "resid=%ld", cfqq->slice_resid);
	}
1602

1603
	cfq_group_served(cfqd, cfqq->cfqg, cfqq);
1604

1605 1606 1607
	if (cfq_cfqq_on_rr(cfqq) && RB_EMPTY_ROOT(&cfqq->sort_list))
		cfq_del_cfqq_rr(cfqd, cfqq);

1608
	cfq_resort_rr_list(cfqd, cfqq);
1609 1610 1611 1612 1613

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

	if (cfqd->active_cic) {
1614
		put_io_context(cfqd->active_cic->icq.ioc);
1615 1616 1617 1618
		cfqd->active_cic = NULL;
	}
}

1619
static inline void cfq_slice_expired(struct cfq_data *cfqd, bool timed_out)
1620 1621 1622 1623
{
	struct cfq_queue *cfqq = cfqd->active_queue;

	if (cfqq)
1624
		__cfq_slice_expired(cfqd, cfqq, timed_out);
1625 1626
}

1627 1628 1629 1630
/*
 * Get next queue for service. Unless we have a queue preemption,
 * we'll simply select the first cfqq in the service tree.
 */
J
Jens Axboe 已提交
1631
static struct cfq_queue *cfq_get_next_queue(struct cfq_data *cfqd)
1632
{
1633
	struct cfq_rb_root *service_tree =
1634
		service_tree_for(cfqd->serving_group, cfqd->serving_prio,
1635
					cfqd->serving_type);
1636

1637 1638 1639
	if (!cfqd->rq_queued)
		return NULL;

1640 1641 1642
	/* There is nothing to dispatch */
	if (!service_tree)
		return NULL;
1643 1644 1645
	if (RB_EMPTY_ROOT(&service_tree->rb))
		return NULL;
	return cfq_rb_first(service_tree);
J
Jens Axboe 已提交
1646 1647
}

1648 1649
static struct cfq_queue *cfq_get_next_queue_forced(struct cfq_data *cfqd)
{
1650
	struct cfq_group *cfqg;
1651 1652 1653 1654 1655 1656 1657
	struct cfq_queue *cfqq;
	int i, j;
	struct cfq_rb_root *st;

	if (!cfqd->rq_queued)
		return NULL;

1658 1659 1660 1661
	cfqg = cfq_get_next_cfqg(cfqd);
	if (!cfqg)
		return NULL;

1662 1663 1664 1665 1666 1667
	for_each_cfqg_st(cfqg, i, j, st)
		if ((cfqq = cfq_rb_first(st)) != NULL)
			return cfqq;
	return NULL;
}

1668 1669 1670
/*
 * Get and set a new active queue for service.
 */
1671 1672
static struct cfq_queue *cfq_set_active_queue(struct cfq_data *cfqd,
					      struct cfq_queue *cfqq)
J
Jens Axboe 已提交
1673
{
1674
	if (!cfqq)
1675
		cfqq = cfq_get_next_queue(cfqd);
J
Jens Axboe 已提交
1676

1677
	__cfq_set_active_queue(cfqd, cfqq);
J
Jens Axboe 已提交
1678
	return cfqq;
1679 1680
}

1681 1682 1683
static inline sector_t cfq_dist_from_last(struct cfq_data *cfqd,
					  struct request *rq)
{
1684 1685
	if (blk_rq_pos(rq) >= cfqd->last_position)
		return blk_rq_pos(rq) - cfqd->last_position;
1686
	else
1687
		return cfqd->last_position - blk_rq_pos(rq);
1688 1689
}

1690
static inline int cfq_rq_close(struct cfq_data *cfqd, struct cfq_queue *cfqq,
1691
			       struct request *rq)
J
Jens Axboe 已提交
1692
{
1693
	return cfq_dist_from_last(cfqd, rq) <= CFQQ_CLOSE_THR;
J
Jens Axboe 已提交
1694 1695
}

1696 1697 1698
static struct cfq_queue *cfqq_close(struct cfq_data *cfqd,
				    struct cfq_queue *cur_cfqq)
{
1699
	struct rb_root *root = &cfqd->prio_trees[cur_cfqq->org_ioprio];
1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710
	struct rb_node *parent, *node;
	struct cfq_queue *__cfqq;
	sector_t sector = cfqd->last_position;

	if (RB_EMPTY_ROOT(root))
		return NULL;

	/*
	 * First, if we find a request starting at the end of the last
	 * request, choose it.
	 */
1711
	__cfqq = cfq_prio_tree_lookup(cfqd, root, sector, &parent, NULL);
1712 1713 1714 1715 1716 1717 1718 1719
	if (__cfqq)
		return __cfqq;

	/*
	 * If the exact sector wasn't found, the parent of the NULL leaf
	 * will contain the closest sector.
	 */
	__cfqq = rb_entry(parent, struct cfq_queue, p_node);
1720
	if (cfq_rq_close(cfqd, cur_cfqq, __cfqq->next_rq))
1721 1722
		return __cfqq;

1723
	if (blk_rq_pos(__cfqq->next_rq) < sector)
1724 1725 1726 1727 1728 1729 1730
		node = rb_next(&__cfqq->p_node);
	else
		node = rb_prev(&__cfqq->p_node);
	if (!node)
		return NULL;

	__cfqq = rb_entry(node, struct cfq_queue, p_node);
1731
	if (cfq_rq_close(cfqd, cur_cfqq, __cfqq->next_rq))
1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747
		return __cfqq;

	return NULL;
}

/*
 * cfqd - obvious
 * cur_cfqq - passed in so that we don't decide that the current queue is
 * 	      closely cooperating with itself.
 *
 * So, basically we're assuming that that cur_cfqq has dispatched at least
 * one request, and that cfqd->last_position reflects a position on the disk
 * associated with the I/O issued by cur_cfqq.  I'm not sure this is a valid
 * assumption.
 */
static struct cfq_queue *cfq_close_cooperator(struct cfq_data *cfqd,
1748
					      struct cfq_queue *cur_cfqq)
J
Jens Axboe 已提交
1749
{
1750 1751
	struct cfq_queue *cfqq;

1752 1753
	if (cfq_class_idle(cur_cfqq))
		return NULL;
1754 1755 1756 1757 1758
	if (!cfq_cfqq_sync(cur_cfqq))
		return NULL;
	if (CFQQ_SEEKY(cur_cfqq))
		return NULL;

1759 1760 1761 1762 1763 1764
	/*
	 * Don't search priority tree if it's the only queue in the group.
	 */
	if (cur_cfqq->cfqg->nr_cfqq == 1)
		return NULL;

J
Jens Axboe 已提交
1765
	/*
1766 1767 1768
	 * We should notice if some of the queues are cooperating, eg
	 * working closely on the same area of the disk. In that case,
	 * we can group them together and don't waste time idling.
J
Jens Axboe 已提交
1769
	 */
1770 1771 1772 1773
	cfqq = cfqq_close(cfqd, cur_cfqq);
	if (!cfqq)
		return NULL;

1774 1775 1776 1777
	/* If new queue belongs to different cfq_group, don't choose it */
	if (cur_cfqq->cfqg != cfqq->cfqg)
		return NULL;

J
Jeff Moyer 已提交
1778 1779 1780 1781 1782
	/*
	 * It only makes sense to merge sync queues.
	 */
	if (!cfq_cfqq_sync(cfqq))
		return NULL;
1783 1784
	if (CFQQ_SEEKY(cfqq))
		return NULL;
J
Jeff Moyer 已提交
1785

1786 1787 1788 1789 1790 1791
	/*
	 * Do not merge queues of different priority classes
	 */
	if (cfq_class_rt(cfqq) != cfq_class_rt(cur_cfqq))
		return NULL;

1792
	return cfqq;
J
Jens Axboe 已提交
1793 1794
}

1795 1796 1797 1798 1799 1800 1801
/*
 * Determine whether we should enforce idle window for this queue.
 */

static bool cfq_should_idle(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	enum wl_prio_t prio = cfqq_prio(cfqq);
1802
	struct cfq_rb_root *service_tree = cfqq->service_tree;
1803

1804 1805 1806
	BUG_ON(!service_tree);
	BUG_ON(!service_tree->count);

1807 1808 1809
	if (!cfqd->cfq_slice_idle)
		return false;

1810 1811 1812 1813 1814
	/* We never do for idle class queues. */
	if (prio == IDLE_WORKLOAD)
		return false;

	/* We do for queues that were marked with idle window flag. */
1815 1816
	if (cfq_cfqq_idle_window(cfqq) &&
	   !(blk_queue_nonrot(cfqd->queue) && cfqd->hw_tag))
1817 1818 1819 1820 1821 1822
		return true;

	/*
	 * Otherwise, we do only if they are the last ones
	 * in their service tree.
	 */
1823 1824
	if (service_tree->count == 1 && cfq_cfqq_sync(cfqq) &&
	   !cfq_io_thinktime_big(cfqd, &service_tree->ttime, false))
S
Shaohua Li 已提交
1825
		return true;
1826 1827
	cfq_log_cfqq(cfqd, cfqq, "Not idling. st->count:%d",
			service_tree->count);
S
Shaohua Li 已提交
1828
	return false;
1829 1830
}

J
Jens Axboe 已提交
1831
static void cfq_arm_slice_timer(struct cfq_data *cfqd)
1832
{
1833
	struct cfq_queue *cfqq = cfqd->active_queue;
1834
	struct cfq_io_cq *cic;
1835
	unsigned long sl, group_idle = 0;
1836

1837
	/*
J
Jens Axboe 已提交
1838 1839 1840
	 * SSD device without seek penalty, disable idling. But only do so
	 * for devices that support queuing, otherwise we still have a problem
	 * with sync vs async workloads.
1841
	 */
J
Jens Axboe 已提交
1842
	if (blk_queue_nonrot(cfqd->queue) && cfqd->hw_tag)
1843 1844
		return;

1845
	WARN_ON(!RB_EMPTY_ROOT(&cfqq->sort_list));
J
Jens Axboe 已提交
1846
	WARN_ON(cfq_cfqq_slice_new(cfqq));
1847 1848 1849 1850

	/*
	 * idle is disabled, either manually or by past process history
	 */
1851 1852 1853 1854 1855 1856 1857
	if (!cfq_should_idle(cfqd, cfqq)) {
		/* no queue idling. Check for group idling */
		if (cfqd->cfq_group_idle)
			group_idle = cfqd->cfq_group_idle;
		else
			return;
	}
J
Jens Axboe 已提交
1858

1859
	/*
1860
	 * still active requests from this queue, don't idle
1861
	 */
1862
	if (cfqq->dispatched)
1863 1864
		return;

1865 1866 1867
	/*
	 * task has exited, don't wait
	 */
1868
	cic = cfqd->active_cic;
T
Tejun Heo 已提交
1869
	if (!cic || !atomic_read(&cic->icq.ioc->active_ref))
J
Jens Axboe 已提交
1870 1871
		return;

1872 1873 1874 1875 1876
	/*
	 * If our average think time is larger than the remaining time
	 * slice, then don't idle. This avoids overrunning the allotted
	 * time slice.
	 */
1877 1878
	if (sample_valid(cic->ttime.ttime_samples) &&
	    (cfqq->slice_end - jiffies < cic->ttime.ttime_mean)) {
1879
		cfq_log_cfqq(cfqd, cfqq, "Not idling. think_time:%lu",
1880
			     cic->ttime.ttime_mean);
1881
		return;
1882
	}
1883

1884 1885 1886 1887
	/* There are other queues in the group, don't do group idle */
	if (group_idle && cfqq->cfqg->nr_cfqq > 1)
		return;

J
Jens Axboe 已提交
1888
	cfq_mark_cfqq_wait_request(cfqq);
1889

1890 1891 1892 1893
	if (group_idle)
		sl = cfqd->cfq_group_idle;
	else
		sl = cfqd->cfq_slice_idle;
1894

1895
	mod_timer(&cfqd->idle_slice_timer, jiffies + sl);
1896 1897
	cfq_blkiocg_update_set_idle_time_stats(cfqg_to_blkg(cfqq->cfqg),
					       &blkio_policy_cfq);
1898 1899
	cfq_log_cfqq(cfqd, cfqq, "arm_idle: %lu group_idle: %d", sl,
			group_idle ? 1 : 0);
L
Linus Torvalds 已提交
1900 1901
}

1902 1903 1904
/*
 * Move request from internal lists to the request queue dispatch list.
 */
1905
static void cfq_dispatch_insert(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
1906
{
1907
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
1908
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
1909

1910 1911
	cfq_log_cfqq(cfqd, cfqq, "dispatch_insert");

1912
	cfqq->next_rq = cfq_find_next_rq(cfqd, cfqq, rq);
1913
	cfq_remove_request(rq);
J
Jens Axboe 已提交
1914
	cfqq->dispatched++;
1915
	(RQ_CFQG(rq))->dispatched++;
1916
	elv_dispatch_sort(q, rq);
1917

1918
	cfqd->rq_in_flight[cfq_cfqq_sync(cfqq)]++;
1919
	cfqq->nr_sectors += blk_rq_sectors(rq);
1920
	cfq_blkiocg_update_dispatch_stats(cfqg_to_blkg(cfqq->cfqg),
1921 1922
					  &blkio_policy_cfq, blk_rq_bytes(rq),
					  rq_data_dir(rq), rq_is_sync(rq));
L
Linus Torvalds 已提交
1923 1924 1925 1926 1927
}

/*
 * return expired entry, or NULL to just start from scratch in rbtree
 */
J
Jens Axboe 已提交
1928
static struct request *cfq_check_fifo(struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
1929
{
1930
	struct request *rq = NULL;
L
Linus Torvalds 已提交
1931

J
Jens Axboe 已提交
1932
	if (cfq_cfqq_fifo_expire(cfqq))
L
Linus Torvalds 已提交
1933
		return NULL;
1934 1935 1936

	cfq_mark_cfqq_fifo_expire(cfqq);

1937 1938
	if (list_empty(&cfqq->fifo))
		return NULL;
L
Linus Torvalds 已提交
1939

1940
	rq = rq_entry_fifo(cfqq->fifo.next);
1941
	if (time_before(jiffies, rq_fifo_time(rq)))
1942
		rq = NULL;
L
Linus Torvalds 已提交
1943

1944
	cfq_log_cfqq(cfqq->cfqd, cfqq, "fifo=%p", rq);
J
Jens Axboe 已提交
1945
	return rq;
L
Linus Torvalds 已提交
1946 1947
}

1948 1949 1950 1951
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 已提交
1952

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

1955
	return 2 * base_rq * (IOPRIO_BE_NR - cfqq->ioprio);
L
Linus Torvalds 已提交
1956 1957
}

J
Jeff Moyer 已提交
1958 1959 1960 1961 1962 1963 1964 1965
/*
 * Must be called with the queue_lock held.
 */
static int cfqq_process_refs(struct cfq_queue *cfqq)
{
	int process_refs, io_refs;

	io_refs = cfqq->allocated[READ] + cfqq->allocated[WRITE];
1966
	process_refs = cfqq->ref - io_refs;
J
Jeff Moyer 已提交
1967 1968 1969 1970 1971 1972
	BUG_ON(process_refs < 0);
	return process_refs;
}

static void cfq_setup_merge(struct cfq_queue *cfqq, struct cfq_queue *new_cfqq)
{
1973
	int process_refs, new_process_refs;
J
Jeff Moyer 已提交
1974 1975
	struct cfq_queue *__cfqq;

1976 1977 1978 1979 1980 1981 1982 1983 1984
	/*
	 * If there are no process references on the new_cfqq, then it is
	 * unsafe to follow the ->new_cfqq chain as other cfqq's in the
	 * chain may have dropped their last reference (not just their
	 * last process reference).
	 */
	if (!cfqq_process_refs(new_cfqq))
		return;

J
Jeff Moyer 已提交
1985 1986 1987 1988 1989 1990 1991 1992
	/* Avoid a circular list and skip interim queue merges */
	while ((__cfqq = new_cfqq->new_cfqq)) {
		if (__cfqq == cfqq)
			return;
		new_cfqq = __cfqq;
	}

	process_refs = cfqq_process_refs(cfqq);
1993
	new_process_refs = cfqq_process_refs(new_cfqq);
J
Jeff Moyer 已提交
1994 1995 1996 1997
	/*
	 * If the process for the cfqq has gone away, there is no
	 * sense in merging the queues.
	 */
1998
	if (process_refs == 0 || new_process_refs == 0)
J
Jeff Moyer 已提交
1999 2000
		return;

2001 2002 2003 2004 2005
	/*
	 * Merge in the direction of the lesser amount of work.
	 */
	if (new_process_refs >= process_refs) {
		cfqq->new_cfqq = new_cfqq;
2006
		new_cfqq->ref += process_refs;
2007 2008
	} else {
		new_cfqq->new_cfqq = cfqq;
2009
		cfqq->ref += new_process_refs;
2010
	}
J
Jeff Moyer 已提交
2011 2012
}

2013
static enum wl_type_t cfq_choose_wl(struct cfq_data *cfqd,
2014
				struct cfq_group *cfqg, enum wl_prio_t prio)
2015 2016 2017 2018 2019 2020 2021
{
	struct cfq_queue *queue;
	int i;
	bool key_valid = false;
	unsigned long lowest_key = 0;
	enum wl_type_t cur_best = SYNC_NOIDLE_WORKLOAD;

2022 2023 2024
	for (i = 0; i <= SYNC_WORKLOAD; ++i) {
		/* select the one with lowest rb_key */
		queue = cfq_rb_first(service_tree_for(cfqg, prio, i));
2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
		if (queue &&
		    (!key_valid || time_before(queue->rb_key, lowest_key))) {
			lowest_key = queue->rb_key;
			cur_best = i;
			key_valid = true;
		}
	}

	return cur_best;
}

2036
static void choose_service_tree(struct cfq_data *cfqd, struct cfq_group *cfqg)
2037 2038 2039
{
	unsigned slice;
	unsigned count;
2040
	struct cfq_rb_root *st;
2041
	unsigned group_slice;
2042
	enum wl_prio_t original_prio = cfqd->serving_prio;
2043

2044
	/* Choose next priority. RT > BE > IDLE */
2045
	if (cfq_group_busy_queues_wl(RT_WORKLOAD, cfqd, cfqg))
2046
		cfqd->serving_prio = RT_WORKLOAD;
2047
	else if (cfq_group_busy_queues_wl(BE_WORKLOAD, cfqd, cfqg))
2048 2049 2050 2051 2052 2053 2054
		cfqd->serving_prio = BE_WORKLOAD;
	else {
		cfqd->serving_prio = IDLE_WORKLOAD;
		cfqd->workload_expires = jiffies + 1;
		return;
	}

2055 2056 2057
	if (original_prio != cfqd->serving_prio)
		goto new_workload;

2058 2059 2060 2061 2062
	/*
	 * For RT and BE, we have to choose also the type
	 * (SYNC, SYNC_NOIDLE, ASYNC), and to compute a workload
	 * expiration time
	 */
2063
	st = service_tree_for(cfqg, cfqd->serving_prio, cfqd->serving_type);
2064
	count = st->count;
2065 2066

	/*
2067
	 * check workload expiration, and that we still have other queues ready
2068
	 */
2069
	if (count && !time_after(jiffies, cfqd->workload_expires))
2070 2071
		return;

2072
new_workload:
2073 2074
	/* otherwise select new workload type */
	cfqd->serving_type =
2075 2076
		cfq_choose_wl(cfqd, cfqg, cfqd->serving_prio);
	st = service_tree_for(cfqg, cfqd->serving_prio, cfqd->serving_type);
2077
	count = st->count;
2078 2079 2080 2081 2082 2083

	/*
	 * the workload slice is computed as a fraction of target latency
	 * proportional to the number of queues in that workload, over
	 * all the queues in the same priority class
	 */
2084 2085 2086 2087 2088
	group_slice = cfq_group_slice(cfqd, cfqg);

	slice = group_slice * count /
		max_t(unsigned, cfqg->busy_queues_avg[cfqd->serving_prio],
		      cfq_group_busy_queues_wl(cfqd->serving_prio, cfqd, cfqg));
2089

2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103
	if (cfqd->serving_type == ASYNC_WORKLOAD) {
		unsigned int tmp;

		/*
		 * Async queues are currently system wide. Just taking
		 * proportion of queues with-in same group will lead to higher
		 * async ratio system wide as generally root group is going
		 * to have higher weight. A more accurate thing would be to
		 * calculate system wide asnc/sync ratio.
		 */
		tmp = cfq_target_latency * cfqg_busy_async_queues(cfqd, cfqg);
		tmp = tmp/cfqd->busy_queues;
		slice = min_t(unsigned, slice, tmp);

2104 2105 2106
		/* async workload slice is scaled down according to
		 * the sync/async slice ratio. */
		slice = slice * cfqd->cfq_slice[0] / cfqd->cfq_slice[1];
2107
	} else
2108 2109 2110 2111
		/* sync workload slice is at least 2 * cfq_slice_idle */
		slice = max(slice, 2 * cfqd->cfq_slice_idle);

	slice = max_t(unsigned, slice, CFQ_MIN_TT);
2112
	cfq_log(cfqd, "workload slice:%d", slice);
2113 2114 2115
	cfqd->workload_expires = jiffies + slice;
}

2116 2117 2118
static struct cfq_group *cfq_get_next_cfqg(struct cfq_data *cfqd)
{
	struct cfq_rb_root *st = &cfqd->grp_service_tree;
2119
	struct cfq_group *cfqg;
2120 2121 2122

	if (RB_EMPTY_ROOT(&st->rb))
		return NULL;
2123 2124 2125
	cfqg = cfq_rb_first_group(st);
	update_min_vdisktime(st);
	return cfqg;
2126 2127
}

2128 2129
static void cfq_choose_cfqg(struct cfq_data *cfqd)
{
2130 2131 2132
	struct cfq_group *cfqg = cfq_get_next_cfqg(cfqd);

	cfqd->serving_group = cfqg;
2133 2134 2135 2136 2137 2138

	/* Restore the workload type data */
	if (cfqg->saved_workload_slice) {
		cfqd->workload_expires = jiffies + cfqg->saved_workload_slice;
		cfqd->serving_type = cfqg->saved_workload;
		cfqd->serving_prio = cfqg->saved_serving_prio;
2139 2140 2141
	} else
		cfqd->workload_expires = jiffies - 1;

2142
	choose_service_tree(cfqd, cfqg);
2143 2144
}

2145
/*
2146 2147
 * Select a queue for service. If we have a current active queue,
 * check whether to continue servicing it, or retrieve and set a new one.
2148
 */
2149
static struct cfq_queue *cfq_select_queue(struct cfq_data *cfqd)
L
Linus Torvalds 已提交
2150
{
2151
	struct cfq_queue *cfqq, *new_cfqq = NULL;
L
Linus Torvalds 已提交
2152

2153 2154 2155
	cfqq = cfqd->active_queue;
	if (!cfqq)
		goto new_queue;
L
Linus Torvalds 已提交
2156

2157 2158
	if (!cfqd->rq_queued)
		return NULL;
2159 2160 2161 2162 2163 2164 2165

	/*
	 * We were waiting for group to get backlogged. Expire the queue
	 */
	if (cfq_cfqq_wait_busy(cfqq) && !RB_EMPTY_ROOT(&cfqq->sort_list))
		goto expire;

2166
	/*
J
Jens Axboe 已提交
2167
	 * The active queue has run out of time, expire it and select new.
2168
	 */
2169 2170 2171 2172 2173 2174 2175 2176 2177 2178
	if (cfq_slice_used(cfqq) && !cfq_cfqq_must_dispatch(cfqq)) {
		/*
		 * If slice had not expired at the completion of last request
		 * we might not have turned on wait_busy flag. Don't expire
		 * the queue yet. Allow the group to get backlogged.
		 *
		 * The very fact that we have used the slice, that means we
		 * have been idling all along on this queue and it should be
		 * ok to wait for this request to complete.
		 */
2179 2180 2181
		if (cfqq->cfqg->nr_cfqq == 1 && RB_EMPTY_ROOT(&cfqq->sort_list)
		    && cfqq->dispatched && cfq_should_idle(cfqd, cfqq)) {
			cfqq = NULL;
2182
			goto keep_queue;
2183
		} else
2184
			goto check_group_idle;
2185
	}
L
Linus Torvalds 已提交
2186

2187
	/*
J
Jens Axboe 已提交
2188 2189
	 * The active queue has requests and isn't expired, allow it to
	 * dispatch.
2190
	 */
2191
	if (!RB_EMPTY_ROOT(&cfqq->sort_list))
2192
		goto keep_queue;
J
Jens Axboe 已提交
2193

2194 2195 2196 2197
	/*
	 * If another queue has a request waiting within our mean seek
	 * distance, let it run.  The expire code will check for close
	 * cooperators and put the close queue at the front of the service
J
Jeff Moyer 已提交
2198
	 * tree.  If possible, merge the expiring queue with the new cfqq.
2199
	 */
2200
	new_cfqq = cfq_close_cooperator(cfqd, cfqq);
J
Jeff Moyer 已提交
2201 2202 2203
	if (new_cfqq) {
		if (!cfqq->new_cfqq)
			cfq_setup_merge(cfqq, new_cfqq);
2204
		goto expire;
J
Jeff Moyer 已提交
2205
	}
2206

J
Jens Axboe 已提交
2207 2208 2209 2210 2211
	/*
	 * No requests pending. If the active queue still has requests in
	 * flight or is idling for a new request, allow either of these
	 * conditions to happen (or time out) before selecting a new queue.
	 */
2212 2213 2214 2215 2216
	if (timer_pending(&cfqd->idle_slice_timer)) {
		cfqq = NULL;
		goto keep_queue;
	}

2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227
	/*
	 * This is a deep seek queue, but the device is much faster than
	 * the queue can deliver, don't idle
	 **/
	if (CFQQ_SEEKY(cfqq) && cfq_cfqq_idle_window(cfqq) &&
	    (cfq_cfqq_slice_new(cfqq) ||
	    (cfqq->slice_end - jiffies > jiffies - cfqq->slice_start))) {
		cfq_clear_cfqq_deep(cfqq);
		cfq_clear_cfqq_idle_window(cfqq);
	}

2228 2229 2230 2231 2232 2233 2234 2235 2236 2237
	if (cfqq->dispatched && cfq_should_idle(cfqd, cfqq)) {
		cfqq = NULL;
		goto keep_queue;
	}

	/*
	 * If group idle is enabled and there are requests dispatched from
	 * this group, wait for requests to complete.
	 */
check_group_idle:
S
Shaohua Li 已提交
2238 2239 2240
	if (cfqd->cfq_group_idle && cfqq->cfqg->nr_cfqq == 1 &&
	    cfqq->cfqg->dispatched &&
	    !cfq_io_thinktime_big(cfqd, &cfqq->cfqg->ttime, true)) {
2241 2242
		cfqq = NULL;
		goto keep_queue;
2243 2244
	}

J
Jens Axboe 已提交
2245
expire:
2246
	cfq_slice_expired(cfqd, 0);
J
Jens Axboe 已提交
2247
new_queue:
2248 2249 2250 2251 2252
	/*
	 * Current queue expired. Check if we have to switch to a new
	 * service tree
	 */
	if (!new_cfqq)
2253
		cfq_choose_cfqg(cfqd);
2254

2255
	cfqq = cfq_set_active_queue(cfqd, new_cfqq);
2256
keep_queue:
J
Jens Axboe 已提交
2257
	return cfqq;
2258 2259
}

J
Jens Axboe 已提交
2260
static int __cfq_forced_dispatch_cfqq(struct cfq_queue *cfqq)
2261 2262 2263 2264 2265 2266 2267 2268 2269
{
	int dispatched = 0;

	while (cfqq->next_rq) {
		cfq_dispatch_insert(cfqq->cfqd->queue, cfqq->next_rq);
		dispatched++;
	}

	BUG_ON(!list_empty(&cfqq->fifo));
2270 2271

	/* By default cfqq is not expired if it is empty. Do it explicitly */
2272
	__cfq_slice_expired(cfqq->cfqd, cfqq, 0);
2273 2274 2275
	return dispatched;
}

2276 2277 2278 2279
/*
 * Drain our current requests. Used for barriers and when switching
 * io schedulers on-the-fly.
 */
2280
static int cfq_forced_dispatch(struct cfq_data *cfqd)
2281
{
2282
	struct cfq_queue *cfqq;
2283
	int dispatched = 0;
2284

2285
	/* Expire the timeslice of the current active queue first */
2286
	cfq_slice_expired(cfqd, 0);
2287 2288
	while ((cfqq = cfq_get_next_queue_forced(cfqd)) != NULL) {
		__cfq_set_active_queue(cfqd, cfqq);
2289
		dispatched += __cfq_forced_dispatch_cfqq(cfqq);
2290
	}
2291 2292 2293

	BUG_ON(cfqd->busy_queues);

2294
	cfq_log(cfqd, "forced_dispatch=%d", dispatched);
2295 2296 2297
	return dispatched;
}

S
Shaohua Li 已提交
2298 2299 2300 2301 2302
static inline bool cfq_slice_used_soon(struct cfq_data *cfqd,
	struct cfq_queue *cfqq)
{
	/* the queue hasn't finished any request, can't estimate */
	if (cfq_cfqq_slice_new(cfqq))
S
Shaohua Li 已提交
2303
		return true;
S
Shaohua Li 已提交
2304 2305
	if (time_after(jiffies + cfqd->cfq_slice_idle * cfqq->dispatched,
		cfqq->slice_end))
S
Shaohua Li 已提交
2306
		return true;
S
Shaohua Li 已提交
2307

S
Shaohua Li 已提交
2308
	return false;
S
Shaohua Li 已提交
2309 2310
}

2311
static bool cfq_may_dispatch(struct cfq_data *cfqd, struct cfq_queue *cfqq)
2312 2313
{
	unsigned int max_dispatch;
2314

2315 2316 2317
	/*
	 * Drain async requests before we start sync IO
	 */
2318
	if (cfq_should_idle(cfqd, cfqq) && cfqd->rq_in_flight[BLK_RW_ASYNC])
2319
		return false;
2320

2321 2322 2323
	/*
	 * If this is an async queue and we have sync IO in flight, let it wait
	 */
2324
	if (cfqd->rq_in_flight[BLK_RW_SYNC] && !cfq_cfqq_sync(cfqq))
2325
		return false;
2326

S
Shaohua Li 已提交
2327
	max_dispatch = max_t(unsigned int, cfqd->cfq_quantum / 2, 1);
2328 2329
	if (cfq_class_idle(cfqq))
		max_dispatch = 1;
2330

2331 2332 2333 2334
	/*
	 * Does this cfqq already have too much IO in flight?
	 */
	if (cfqq->dispatched >= max_dispatch) {
2335
		bool promote_sync = false;
2336 2337 2338
		/*
		 * idle queue must always only have a single IO in flight
		 */
2339
		if (cfq_class_idle(cfqq))
2340
			return false;
2341

2342
		/*
2343 2344
		 * If there is only one sync queue
		 * we can ignore async queue here and give the sync
2345 2346 2347 2348
		 * queue no dispatch limit. The reason is a sync queue can
		 * preempt async queue, limiting the sync queue doesn't make
		 * sense. This is useful for aiostress test.
		 */
2349 2350
		if (cfq_cfqq_sync(cfqq) && cfqd->busy_sync_queues == 1)
			promote_sync = true;
2351

2352 2353 2354
		/*
		 * We have other queues, don't allow more IO from this one
		 */
2355 2356
		if (cfqd->busy_queues > 1 && cfq_slice_used_soon(cfqd, cfqq) &&
				!promote_sync)
2357
			return false;
2358

2359
		/*
2360
		 * Sole queue user, no limit
2361
		 */
2362
		if (cfqd->busy_queues == 1 || promote_sync)
S
Shaohua Li 已提交
2363 2364 2365 2366 2367 2368 2369 2370 2371
			max_dispatch = -1;
		else
			/*
			 * Normally we start throttling cfqq when cfq_quantum/2
			 * requests have been dispatched. But we can drive
			 * deeper queue depths at the beginning of slice
			 * subjected to upper limit of cfq_quantum.
			 * */
			max_dispatch = cfqd->cfq_quantum;
2372 2373 2374 2375 2376 2377 2378
	}

	/*
	 * Async queues must wait a bit before being allowed dispatch.
	 * We also ramp up the dispatch depth gradually for async IO,
	 * based on the last sync IO we serviced
	 */
2379
	if (!cfq_cfqq_sync(cfqq) && cfqd->cfq_latency) {
2380
		unsigned long last_sync = jiffies - cfqd->last_delayed_sync;
2381
		unsigned int depth;
2382

2383
		depth = last_sync / cfqd->cfq_slice[1];
2384 2385
		if (!depth && !cfqq->dispatched)
			depth = 1;
2386 2387
		if (depth < max_dispatch)
			max_dispatch = depth;
2388
	}
2389

2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421
	/*
	 * If we're below the current max, allow a dispatch
	 */
	return cfqq->dispatched < max_dispatch;
}

/*
 * Dispatch a request from cfqq, moving them to the request queue
 * dispatch list.
 */
static bool cfq_dispatch_request(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	struct request *rq;

	BUG_ON(RB_EMPTY_ROOT(&cfqq->sort_list));

	if (!cfq_may_dispatch(cfqd, cfqq))
		return false;

	/*
	 * follow expired path, else get first next available
	 */
	rq = cfq_check_fifo(cfqq);
	if (!rq)
		rq = cfqq->next_rq;

	/*
	 * insert request into driver dispatch list
	 */
	cfq_dispatch_insert(cfqd->queue, rq);

	if (!cfqd->active_cic) {
2422
		struct cfq_io_cq *cic = RQ_CIC(rq);
2423

2424
		atomic_long_inc(&cic->icq.ioc->refcount);
2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447
		cfqd->active_cic = cic;
	}

	return true;
}

/*
 * Find the cfqq that we need to service and move a request from that to the
 * dispatch list
 */
static int cfq_dispatch_requests(struct request_queue *q, int force)
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
	struct cfq_queue *cfqq;

	if (!cfqd->busy_queues)
		return 0;

	if (unlikely(force))
		return cfq_forced_dispatch(cfqd);

	cfqq = cfq_select_queue(cfqd);
	if (!cfqq)
2448 2449
		return 0;

2450
	/*
2451
	 * Dispatch a request from this cfqq, if it is allowed
2452
	 */
2453 2454 2455
	if (!cfq_dispatch_request(cfqd, cfqq))
		return 0;

2456
	cfqq->slice_dispatch++;
2457
	cfq_clear_cfqq_must_dispatch(cfqq);
2458

2459 2460 2461 2462 2463 2464 2465 2466
	/*
	 * expire an async queue immediately if it has used up its slice. idle
	 * queue always expire after 1 dispatch round.
	 */
	if (cfqd->busy_queues > 1 && ((!cfq_cfqq_sync(cfqq) &&
	    cfqq->slice_dispatch >= cfq_prio_to_maxrq(cfqd, cfqq)) ||
	    cfq_class_idle(cfqq))) {
		cfqq->slice_end = jiffies + 1;
2467
		cfq_slice_expired(cfqd, 0);
L
Linus Torvalds 已提交
2468 2469
	}

2470
	cfq_log_cfqq(cfqd, cfqq, "dispatched a request");
2471
	return 1;
L
Linus Torvalds 已提交
2472 2473 2474
}

/*
J
Jens Axboe 已提交
2475 2476
 * 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 已提交
2477
 *
2478
 * Each cfq queue took a reference on the parent group. Drop it now.
L
Linus Torvalds 已提交
2479 2480 2481 2482
 * queue lock must be held here.
 */
static void cfq_put_queue(struct cfq_queue *cfqq)
{
2483
	struct cfq_data *cfqd = cfqq->cfqd;
2484
	struct cfq_group *cfqg;
2485

2486
	BUG_ON(cfqq->ref <= 0);
L
Linus Torvalds 已提交
2487

2488 2489
	cfqq->ref--;
	if (cfqq->ref)
L
Linus Torvalds 已提交
2490 2491
		return;

2492
	cfq_log_cfqq(cfqd, cfqq, "put_queue");
L
Linus Torvalds 已提交
2493
	BUG_ON(rb_first(&cfqq->sort_list));
2494
	BUG_ON(cfqq->allocated[READ] + cfqq->allocated[WRITE]);
2495
	cfqg = cfqq->cfqg;
L
Linus Torvalds 已提交
2496

2497
	if (unlikely(cfqd->active_queue == cfqq)) {
2498
		__cfq_slice_expired(cfqd, cfqq, 0);
2499
		cfq_schedule_dispatch(cfqd);
2500
	}
2501

2502
	BUG_ON(cfq_cfqq_on_rr(cfqq));
L
Linus Torvalds 已提交
2503
	kmem_cache_free(cfq_pool, cfqq);
T
Tejun Heo 已提交
2504
	blkg_put(cfqg_to_blkg(cfqg));
L
Linus Torvalds 已提交
2505 2506
}

2507
static void cfq_put_cooperator(struct cfq_queue *cfqq)
L
Linus Torvalds 已提交
2508
{
J
Jeff Moyer 已提交
2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525
	struct cfq_queue *__cfqq, *next;

	/*
	 * If this queue was scheduled to merge with another queue, be
	 * sure to drop the reference taken on that queue (and others in
	 * the merge chain).  See cfq_setup_merge and cfq_merge_cfqqs.
	 */
	__cfqq = cfqq->new_cfqq;
	while (__cfqq) {
		if (__cfqq == cfqq) {
			WARN(1, "cfqq->new_cfqq loop detected\n");
			break;
		}
		next = __cfqq->new_cfqq;
		cfq_put_queue(__cfqq);
		__cfqq = next;
	}
2526 2527 2528 2529 2530 2531 2532 2533 2534 2535
}

static void cfq_exit_cfqq(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
	if (unlikely(cfqq == cfqd->active_queue)) {
		__cfq_slice_expired(cfqd, cfqq, 0);
		cfq_schedule_dispatch(cfqd);
	}

	cfq_put_cooperator(cfqq);
J
Jeff Moyer 已提交
2536

2537 2538
	cfq_put_queue(cfqq);
}
2539

2540 2541 2542 2543 2544 2545 2546
static void cfq_init_icq(struct io_cq *icq)
{
	struct cfq_io_cq *cic = icq_to_cic(icq);

	cic->ttime.last_end_request = jiffies;
}

2547
static void cfq_exit_icq(struct io_cq *icq)
2548
{
2549
	struct cfq_io_cq *cic = icq_to_cic(icq);
2550
	struct cfq_data *cfqd = cic_to_cfqd(cic);
2551

2552 2553 2554
	if (cic->cfqq[BLK_RW_ASYNC]) {
		cfq_exit_cfqq(cfqd, cic->cfqq[BLK_RW_ASYNC]);
		cic->cfqq[BLK_RW_ASYNC] = NULL;
2555 2556
	}

2557 2558 2559
	if (cic->cfqq[BLK_RW_SYNC]) {
		cfq_exit_cfqq(cfqd, cic->cfqq[BLK_RW_SYNC]);
		cic->cfqq[BLK_RW_SYNC] = NULL;
2560
	}
2561 2562
}

2563
static void cfq_init_prio_data(struct cfq_queue *cfqq, struct cfq_io_cq *cic)
2564 2565 2566 2567
{
	struct task_struct *tsk = current;
	int ioprio_class;

J
Jens Axboe 已提交
2568
	if (!cfq_cfqq_prio_changed(cfqq))
2569 2570
		return;

2571
	ioprio_class = IOPRIO_PRIO_CLASS(cic->icq.ioc->ioprio);
2572
	switch (ioprio_class) {
2573 2574 2575 2576
	default:
		printk(KERN_ERR "cfq: bad prio %x\n", ioprio_class);
	case IOPRIO_CLASS_NONE:
		/*
2577
		 * no prio set, inherit CPU scheduling settings
2578 2579
		 */
		cfqq->ioprio = task_nice_ioprio(tsk);
2580
		cfqq->ioprio_class = task_nice_ioclass(tsk);
2581 2582
		break;
	case IOPRIO_CLASS_RT:
2583
		cfqq->ioprio = task_ioprio(cic->icq.ioc);
2584 2585 2586
		cfqq->ioprio_class = IOPRIO_CLASS_RT;
		break;
	case IOPRIO_CLASS_BE:
2587
		cfqq->ioprio = task_ioprio(cic->icq.ioc);
2588 2589 2590 2591 2592 2593 2594
		cfqq->ioprio_class = IOPRIO_CLASS_BE;
		break;
	case IOPRIO_CLASS_IDLE:
		cfqq->ioprio_class = IOPRIO_CLASS_IDLE;
		cfqq->ioprio = 7;
		cfq_clear_cfqq_idle_window(cfqq);
		break;
2595 2596 2597 2598 2599 2600 2601
	}

	/*
	 * keep track of original prio settings in case we have to temporarily
	 * elevate the priority of this queue
	 */
	cfqq->org_ioprio = cfqq->ioprio;
J
Jens Axboe 已提交
2602
	cfq_clear_cfqq_prio_changed(cfqq);
2603 2604
}

2605
static void changed_ioprio(struct cfq_io_cq *cic, struct bio *bio)
2606
{
2607
	struct cfq_data *cfqd = cic_to_cfqd(cic);
2608
	struct cfq_queue *cfqq;
2609

2610 2611 2612
	if (unlikely(!cfqd))
		return;

2613
	cfqq = cic->cfqq[BLK_RW_ASYNC];
2614 2615
	if (cfqq) {
		struct cfq_queue *new_cfqq;
2616 2617
		new_cfqq = cfq_get_queue(cfqd, BLK_RW_ASYNC, cic, bio,
					 GFP_ATOMIC);
2618
		if (new_cfqq) {
2619
			cic->cfqq[BLK_RW_ASYNC] = new_cfqq;
2620 2621
			cfq_put_queue(cfqq);
		}
2622
	}
2623

2624
	cfqq = cic->cfqq[BLK_RW_SYNC];
2625 2626
	if (cfqq)
		cfq_mark_cfqq_prio_changed(cfqq);
2627 2628
}

2629
static void cfq_init_cfqq(struct cfq_data *cfqd, struct cfq_queue *cfqq,
2630
			  pid_t pid, bool is_sync)
2631 2632 2633 2634 2635
{
	RB_CLEAR_NODE(&cfqq->rb_node);
	RB_CLEAR_NODE(&cfqq->p_node);
	INIT_LIST_HEAD(&cfqq->fifo);

2636
	cfqq->ref = 0;
2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648
	cfqq->cfqd = cfqd;

	cfq_mark_cfqq_prio_changed(cfqq);

	if (is_sync) {
		if (!cfq_class_idle(cfqq))
			cfq_mark_cfqq_idle_window(cfqq);
		cfq_mark_cfqq_sync(cfqq);
	}
	cfqq->pid = pid;
}

2649
#ifdef CONFIG_CFQ_GROUP_IOSCHED
2650
static void changed_cgroup(struct cfq_io_cq *cic)
2651 2652
{
	struct cfq_queue *sync_cfqq = cic_to_cfqq(cic, 1);
2653
	struct cfq_data *cfqd = cic_to_cfqd(cic);
2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672
	struct request_queue *q;

	if (unlikely(!cfqd))
		return;

	q = cfqd->queue;

	if (sync_cfqq) {
		/*
		 * Drop reference to sync queue. A new sync queue will be
		 * assigned in new group upon arrival of a fresh request.
		 */
		cfq_log_cfqq(cfqd, sync_cfqq, "changed cgroup");
		cic_set_cfqq(cic, NULL, 1);
		cfq_put_queue(sync_cfqq);
	}
}
#endif  /* CONFIG_CFQ_GROUP_IOSCHED */

2673
static struct cfq_queue *
2674 2675
cfq_find_alloc_queue(struct cfq_data *cfqd, bool is_sync, struct cfq_io_cq *cic,
		     struct bio *bio, gfp_t gfp_mask)
2676
{
2677
	struct blkio_cgroup *blkcg;
2678
	struct cfq_queue *cfqq, *new_cfqq = NULL;
2679
	struct cfq_group *cfqg;
2680 2681

retry:
2682 2683
	rcu_read_lock();

2684
	blkcg = bio_blkio_cgroup(bio);
2685
	cfqg = cfq_lookup_create_cfqg(cfqd, blkcg);
2686
	cfqq = cic_to_cfqq(cic, is_sync);
2687

2688 2689 2690 2691 2692 2693
	/*
	 * Always try a new alloc if we fell back to the OOM cfqq
	 * originally, since it should just be a temporary situation.
	 */
	if (!cfqq || cfqq == &cfqd->oom_cfqq) {
		cfqq = NULL;
2694 2695 2696 2697
		if (new_cfqq) {
			cfqq = new_cfqq;
			new_cfqq = NULL;
		} else if (gfp_mask & __GFP_WAIT) {
2698
			rcu_read_unlock();
2699
			spin_unlock_irq(cfqd->queue->queue_lock);
2700
			new_cfqq = kmem_cache_alloc_node(cfq_pool,
2701
					gfp_mask | __GFP_ZERO,
2702
					cfqd->queue->node);
2703
			spin_lock_irq(cfqd->queue->queue_lock);
2704 2705
			if (new_cfqq)
				goto retry;
2706
		} else {
2707 2708 2709
			cfqq = kmem_cache_alloc_node(cfq_pool,
					gfp_mask | __GFP_ZERO,
					cfqd->queue->node);
2710 2711
		}

2712 2713
		if (cfqq) {
			cfq_init_cfqq(cfqd, cfqq, current->pid, is_sync);
2714
			cfq_init_prio_data(cfqq, cic);
2715
			cfq_link_cfqq_cfqg(cfqq, cfqg);
2716 2717 2718
			cfq_log_cfqq(cfqd, cfqq, "alloced");
		} else
			cfqq = &cfqd->oom_cfqq;
2719 2720 2721 2722 2723
	}

	if (new_cfqq)
		kmem_cache_free(cfq_pool, new_cfqq);

2724
	rcu_read_unlock();
2725 2726 2727
	return cfqq;
}

2728 2729 2730
static struct cfq_queue **
cfq_async_queue_prio(struct cfq_data *cfqd, int ioprio_class, int ioprio)
{
2731
	switch (ioprio_class) {
2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742
	case IOPRIO_CLASS_RT:
		return &cfqd->async_cfqq[0][ioprio];
	case IOPRIO_CLASS_BE:
		return &cfqd->async_cfqq[1][ioprio];
	case IOPRIO_CLASS_IDLE:
		return &cfqd->async_idle_cfqq;
	default:
		BUG();
	}
}

2743
static struct cfq_queue *
2744
cfq_get_queue(struct cfq_data *cfqd, bool is_sync, struct cfq_io_cq *cic,
2745
	      struct bio *bio, gfp_t gfp_mask)
2746
{
2747 2748
	const int ioprio = task_ioprio(cic->icq.ioc);
	const int ioprio_class = task_ioprio_class(cic->icq.ioc);
2749
	struct cfq_queue **async_cfqq = NULL;
2750 2751
	struct cfq_queue *cfqq = NULL;

2752 2753 2754 2755 2756
	if (!is_sync) {
		async_cfqq = cfq_async_queue_prio(cfqd, ioprio_class, ioprio);
		cfqq = *async_cfqq;
	}

2757
	if (!cfqq)
2758
		cfqq = cfq_find_alloc_queue(cfqd, is_sync, cic, bio, gfp_mask);
2759 2760 2761 2762

	/*
	 * pin the queue now that it's allocated, scheduler exit will prune it
	 */
2763
	if (!is_sync && !(*async_cfqq)) {
2764
		cfqq->ref++;
2765
		*async_cfqq = cfqq;
2766 2767
	}

2768
	cfqq->ref++;
2769 2770 2771
	return cfqq;
}

2772
static void
2773
__cfq_update_io_thinktime(struct cfq_ttime *ttime, unsigned long slice_idle)
L
Linus Torvalds 已提交
2774
{
2775 2776
	unsigned long elapsed = jiffies - ttime->last_end_request;
	elapsed = min(elapsed, 2UL * slice_idle);
2777

2778 2779 2780 2781 2782 2783 2784
	ttime->ttime_samples = (7*ttime->ttime_samples + 256) / 8;
	ttime->ttime_total = (7*ttime->ttime_total + 256*elapsed) / 8;
	ttime->ttime_mean = (ttime->ttime_total + 128) / ttime->ttime_samples;
}

static void
cfq_update_io_thinktime(struct cfq_data *cfqd, struct cfq_queue *cfqq,
2785
			struct cfq_io_cq *cic)
2786
{
2787
	if (cfq_cfqq_sync(cfqq)) {
2788
		__cfq_update_io_thinktime(&cic->ttime, cfqd->cfq_slice_idle);
2789 2790 2791
		__cfq_update_io_thinktime(&cfqq->service_tree->ttime,
			cfqd->cfq_slice_idle);
	}
S
Shaohua Li 已提交
2792 2793 2794
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	__cfq_update_io_thinktime(&cfqq->cfqg->ttime, cfqd->cfq_group_idle);
#endif
2795
}
L
Linus Torvalds 已提交
2796

2797
static void
2798
cfq_update_io_seektime(struct cfq_data *cfqd, struct cfq_queue *cfqq,
J
Jens Axboe 已提交
2799
		       struct request *rq)
2800
{
2801
	sector_t sdist = 0;
2802
	sector_t n_sec = blk_rq_sectors(rq);
2803 2804 2805 2806 2807 2808
	if (cfqq->last_request_pos) {
		if (cfqq->last_request_pos < blk_rq_pos(rq))
			sdist = blk_rq_pos(rq) - cfqq->last_request_pos;
		else
			sdist = cfqq->last_request_pos - blk_rq_pos(rq);
	}
2809

2810
	cfqq->seek_history <<= 1;
2811 2812 2813 2814
	if (blk_queue_nonrot(cfqd->queue))
		cfqq->seek_history |= (n_sec < CFQQ_SECT_THR_NONROT);
	else
		cfqq->seek_history |= (sdist > CFQQ_SEEK_THR);
2815
}
L
Linus Torvalds 已提交
2816

2817 2818 2819 2820 2821 2822
/*
 * 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,
2823
		       struct cfq_io_cq *cic)
2824
{
2825
	int old_idle, enable_idle;
2826

2827 2828 2829 2830
	/*
	 * Don't idle for async or idle io prio class
	 */
	if (!cfq_cfqq_sync(cfqq) || cfq_class_idle(cfqq))
2831 2832
		return;

2833
	enable_idle = old_idle = cfq_cfqq_idle_window(cfqq);
L
Linus Torvalds 已提交
2834

2835 2836 2837
	if (cfqq->queued[0] + cfqq->queued[1] >= 4)
		cfq_mark_cfqq_deep(cfqq);

2838 2839
	if (cfqq->next_rq && (cfqq->next_rq->cmd_flags & REQ_NOIDLE))
		enable_idle = 0;
T
Tejun Heo 已提交
2840
	else if (!atomic_read(&cic->icq.ioc->active_ref) ||
2841 2842
		 !cfqd->cfq_slice_idle ||
		 (!cfq_cfqq_deep(cfqq) && CFQQ_SEEKY(cfqq)))
2843
		enable_idle = 0;
2844 2845
	else if (sample_valid(cic->ttime.ttime_samples)) {
		if (cic->ttime.ttime_mean > cfqd->cfq_slice_idle)
2846 2847 2848
			enable_idle = 0;
		else
			enable_idle = 1;
L
Linus Torvalds 已提交
2849 2850
	}

2851 2852 2853 2854 2855 2856 2857
	if (old_idle != enable_idle) {
		cfq_log_cfqq(cfqd, cfqq, "idle=%d", enable_idle);
		if (enable_idle)
			cfq_mark_cfqq_idle_window(cfqq);
		else
			cfq_clear_cfqq_idle_window(cfqq);
	}
2858
}
L
Linus Torvalds 已提交
2859

2860 2861 2862 2863
/*
 * 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.
 */
2864
static bool
2865
cfq_should_preempt(struct cfq_data *cfqd, struct cfq_queue *new_cfqq,
J
Jens Axboe 已提交
2866
		   struct request *rq)
2867
{
J
Jens Axboe 已提交
2868
	struct cfq_queue *cfqq;
2869

J
Jens Axboe 已提交
2870 2871
	cfqq = cfqd->active_queue;
	if (!cfqq)
2872
		return false;
2873

J
Jens Axboe 已提交
2874
	if (cfq_class_idle(new_cfqq))
2875
		return false;
2876 2877

	if (cfq_class_idle(cfqq))
2878
		return true;
2879

2880 2881 2882 2883 2884 2885
	/*
	 * Don't allow a non-RT request to preempt an ongoing RT cfqq timeslice.
	 */
	if (cfq_class_rt(cfqq) && !cfq_class_rt(new_cfqq))
		return false;

2886 2887 2888 2889
	/*
	 * if the new request is sync, but the currently running queue is
	 * not, let the sync request have priority.
	 */
J
Jens Axboe 已提交
2890
	if (rq_is_sync(rq) && !cfq_cfqq_sync(cfqq))
2891
		return true;
2892

2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905
	if (new_cfqq->cfqg != cfqq->cfqg)
		return false;

	if (cfq_slice_used(cfqq))
		return true;

	/* Allow preemption only if we are idling on sync-noidle tree */
	if (cfqd->serving_type == SYNC_NOIDLE_WORKLOAD &&
	    cfqq_type(new_cfqq) == SYNC_NOIDLE_WORKLOAD &&
	    new_cfqq->service_tree->count == 2 &&
	    RB_EMPTY_ROOT(&cfqq->sort_list))
		return true;

2906 2907 2908 2909
	/*
	 * So both queues are sync. Let the new request get disk time if
	 * it's a metadata request and the current queue is doing regular IO.
	 */
2910
	if ((rq->cmd_flags & REQ_PRIO) && !cfqq->prio_pending)
2911 2912
		return true;

2913 2914 2915 2916
	/*
	 * Allow an RT request to pre-empt an ongoing non-RT cfqq timeslice.
	 */
	if (cfq_class_rt(new_cfqq) && !cfq_class_rt(cfqq))
2917
		return true;
2918

2919 2920 2921 2922
	/* An idle queue should not be idle now for some reason */
	if (RB_EMPTY_ROOT(&cfqq->sort_list) && !cfq_should_idle(cfqd, cfqq))
		return true;

2923
	if (!cfqd->active_cic || !cfq_cfqq_wait_request(cfqq))
2924
		return false;
2925 2926 2927 2928 2929

	/*
	 * if this request is as-good as one we would expect from the
	 * current cfqq, let it preempt
	 */
2930
	if (cfq_rq_close(cfqd, cfqq, rq))
2931
		return true;
2932

2933
	return false;
2934 2935 2936 2937 2938 2939 2940 2941
}

/*
 * 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)
{
S
Shaohua Li 已提交
2942 2943
	enum wl_type_t old_type = cfqq_type(cfqd->active_queue);

2944
	cfq_log_cfqq(cfqd, cfqq, "preempt");
S
Shaohua Li 已提交
2945
	cfq_slice_expired(cfqd, 1);
2946

2947 2948 2949 2950
	/*
	 * workload type is changed, don't save slice, otherwise preempt
	 * doesn't happen
	 */
S
Shaohua Li 已提交
2951
	if (old_type != cfqq_type(cfqq))
2952 2953
		cfqq->cfqg->saved_workload_slice = 0;

2954 2955 2956 2957 2958
	/*
	 * Put the new queue at the front of the of the current list,
	 * so we know that it will be selected next.
	 */
	BUG_ON(!cfq_cfqq_on_rr(cfqq));
2959 2960

	cfq_service_tree_add(cfqd, cfqq, 1);
2961

2962 2963
	cfqq->slice_end = 0;
	cfq_mark_cfqq_slice_new(cfqq);
2964 2965 2966
}

/*
J
Jens Axboe 已提交
2967
 * Called when a new fs request (rq) is added (to cfqq). Check if there's
2968 2969 2970
 * something we should do about it
 */
static void
J
Jens Axboe 已提交
2971 2972
cfq_rq_enqueued(struct cfq_data *cfqd, struct cfq_queue *cfqq,
		struct request *rq)
2973
{
2974
	struct cfq_io_cq *cic = RQ_CIC(rq);
2975

2976
	cfqd->rq_queued++;
2977 2978
	if (rq->cmd_flags & REQ_PRIO)
		cfqq->prio_pending++;
2979

2980
	cfq_update_io_thinktime(cfqd, cfqq, cic);
2981
	cfq_update_io_seektime(cfqd, cfqq, rq);
J
Jens Axboe 已提交
2982 2983
	cfq_update_idle_window(cfqd, cfqq, cic);

2984
	cfqq->last_request_pos = blk_rq_pos(rq) + blk_rq_sectors(rq);
2985 2986 2987

	if (cfqq == cfqd->active_queue) {
		/*
2988 2989 2990
		 * Remember that we saw a request from this process, but
		 * don't start queuing just yet. Otherwise we risk seeing lots
		 * of tiny requests, because we disrupt the normal plugging
2991 2992
		 * and merging. If the request is already larger than a single
		 * page, let it rip immediately. For that case we assume that
2993 2994 2995
		 * merging is already done. Ditto for a busy system that
		 * has other work pending, don't risk delaying until the
		 * idle timer unplug to continue working.
2996
		 */
2997
		if (cfq_cfqq_wait_request(cfqq)) {
2998 2999
			if (blk_rq_bytes(rq) > PAGE_CACHE_SIZE ||
			    cfqd->busy_queues > 1) {
3000
				cfq_del_timer(cfqd, cfqq);
3001
				cfq_clear_cfqq_wait_request(cfqq);
3002
				__blk_run_queue(cfqd->queue);
3003
			} else {
3004
				cfq_blkiocg_update_idle_time_stats(
3005 3006
						cfqg_to_blkg(cfqq->cfqg),
						&blkio_policy_cfq);
3007
				cfq_mark_cfqq_must_dispatch(cfqq);
3008
			}
3009
		}
J
Jens Axboe 已提交
3010
	} else if (cfq_should_preempt(cfqd, cfqq, rq)) {
3011 3012 3013
		/*
		 * not the active queue - expire current slice if it is
		 * idle and has expired it's mean thinktime or this new queue
3014 3015
		 * has some old slice time left and is of higher priority or
		 * this new queue is RT and the current one is BE
3016 3017
		 */
		cfq_preempt_queue(cfqd, cfqq);
3018
		__blk_run_queue(cfqd->queue);
3019
	}
L
Linus Torvalds 已提交
3020 3021
}

3022
static void cfq_insert_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
3023
{
3024
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
3025
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
3026

3027
	cfq_log_cfqq(cfqd, cfqq, "insert_request");
3028
	cfq_init_prio_data(cfqq, RQ_CIC(rq));
L
Linus Torvalds 已提交
3029

3030
	rq_set_fifo_time(rq, jiffies + cfqd->cfq_fifo_expire[rq_is_sync(rq)]);
3031
	list_add_tail(&rq->queuelist, &cfqq->fifo);
3032
	cfq_add_rq_rb(rq);
3033
	cfq_blkiocg_update_io_add_stats(cfqg_to_blkg(RQ_CFQG(rq)),
3034
					&blkio_policy_cfq,
3035 3036
					cfqg_to_blkg(cfqd->serving_group),
					rq_data_dir(rq), rq_is_sync(rq));
J
Jens Axboe 已提交
3037
	cfq_rq_enqueued(cfqd, cfqq, rq);
L
Linus Torvalds 已提交
3038 3039
}

3040 3041 3042 3043 3044 3045
/*
 * Update hw_tag based on peak queue depth over 50 samples under
 * sufficient load.
 */
static void cfq_update_hw_tag(struct cfq_data *cfqd)
{
S
Shaohua Li 已提交
3046 3047
	struct cfq_queue *cfqq = cfqd->active_queue;

3048 3049
	if (cfqd->rq_in_driver > cfqd->hw_tag_est_depth)
		cfqd->hw_tag_est_depth = cfqd->rq_in_driver;
3050 3051 3052

	if (cfqd->hw_tag == 1)
		return;
3053 3054

	if (cfqd->rq_queued <= CFQ_HW_QUEUE_MIN &&
3055
	    cfqd->rq_in_driver <= CFQ_HW_QUEUE_MIN)
3056 3057
		return;

S
Shaohua Li 已提交
3058 3059 3060 3061 3062 3063 3064
	/*
	 * If active queue hasn't enough requests and can idle, cfq might not
	 * dispatch sufficient requests to hardware. Don't zero hw_tag in this
	 * case
	 */
	if (cfqq && cfq_cfqq_idle_window(cfqq) &&
	    cfqq->dispatched + cfqq->queued[0] + cfqq->queued[1] <
3065
	    CFQ_HW_QUEUE_MIN && cfqd->rq_in_driver < CFQ_HW_QUEUE_MIN)
S
Shaohua Li 已提交
3066 3067
		return;

3068 3069 3070
	if (cfqd->hw_tag_samples++ < 50)
		return;

3071
	if (cfqd->hw_tag_est_depth >= CFQ_HW_QUEUE_MIN)
3072 3073 3074 3075 3076
		cfqd->hw_tag = 1;
	else
		cfqd->hw_tag = 0;
}

3077 3078
static bool cfq_should_wait_busy(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
3079
	struct cfq_io_cq *cic = cfqd->active_cic;
3080

3081 3082 3083 3084
	/* If the queue already has requests, don't wait */
	if (!RB_EMPTY_ROOT(&cfqq->sort_list))
		return false;

3085 3086 3087 3088
	/* If there are other queues in the group, don't wait */
	if (cfqq->cfqg->nr_cfqq > 1)
		return false;

S
Shaohua Li 已提交
3089 3090 3091 3092
	/* the only queue in the group, but think time is big */
	if (cfq_io_thinktime_big(cfqd, &cfqq->cfqg->ttime, true))
		return false;

3093 3094 3095 3096
	if (cfq_slice_used(cfqq))
		return true;

	/* if slice left is less than think time, wait busy */
3097 3098
	if (cic && sample_valid(cic->ttime.ttime_samples)
	    && (cfqq->slice_end - jiffies < cic->ttime.ttime_mean))
3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113
		return true;

	/*
	 * If think times is less than a jiffy than ttime_mean=0 and above
	 * will not be true. It might happen that slice has not expired yet
	 * but will expire soon (4-5 ns) during select_queue(). To cover the
	 * case where think time is less than a jiffy, mark the queue wait
	 * busy if only 1 jiffy is left in the slice.
	 */
	if (cfqq->slice_end - jiffies == 1)
		return true;

	return false;
}

3114
static void cfq_completed_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
3115
{
J
Jens Axboe 已提交
3116
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
3117
	struct cfq_data *cfqd = cfqq->cfqd;
3118
	const int sync = rq_is_sync(rq);
3119
	unsigned long now;
L
Linus Torvalds 已提交
3120

3121
	now = jiffies;
3122 3123
	cfq_log_cfqq(cfqd, cfqq, "complete rqnoidle %d",
		     !!(rq->cmd_flags & REQ_NOIDLE));
L
Linus Torvalds 已提交
3124

3125 3126
	cfq_update_hw_tag(cfqd);

3127
	WARN_ON(!cfqd->rq_in_driver);
J
Jens Axboe 已提交
3128
	WARN_ON(!cfqq->dispatched);
3129
	cfqd->rq_in_driver--;
J
Jens Axboe 已提交
3130
	cfqq->dispatched--;
3131
	(RQ_CFQG(rq))->dispatched--;
3132
	cfq_blkiocg_update_completion_stats(cfqg_to_blkg(cfqq->cfqg),
3133 3134 3135
			&blkio_policy_cfq, rq_start_time_ns(rq),
			rq_io_start_time_ns(rq), rq_data_dir(rq),
			rq_is_sync(rq));
L
Linus Torvalds 已提交
3136

3137
	cfqd->rq_in_flight[cfq_cfqq_sync(cfqq)]--;
3138

3139
	if (sync) {
3140 3141
		struct cfq_rb_root *service_tree;

3142
		RQ_CIC(rq)->ttime.last_end_request = now;
3143 3144 3145 3146 3147 3148 3149

		if (cfq_cfqq_on_rr(cfqq))
			service_tree = cfqq->service_tree;
		else
			service_tree = service_tree_for(cfqq->cfqg,
				cfqq_prio(cfqq), cfqq_type(cfqq));
		service_tree->ttime.last_end_request = now;
3150 3151
		if (!time_after(rq->start_time + cfqd->cfq_fifo_expire[1], now))
			cfqd->last_delayed_sync = now;
3152
	}
3153

S
Shaohua Li 已提交
3154 3155 3156 3157
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	cfqq->cfqg->ttime.last_end_request = now;
#endif

3158 3159 3160 3161 3162
	/*
	 * 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) {
3163 3164
		const bool cfqq_empty = RB_EMPTY_ROOT(&cfqq->sort_list);

3165 3166 3167 3168
		if (cfq_cfqq_slice_new(cfqq)) {
			cfq_set_prio_slice(cfqd, cfqq);
			cfq_clear_cfqq_slice_new(cfqq);
		}
3169 3170

		/*
3171 3172
		 * Should we wait for next request to come in before we expire
		 * the queue.
3173
		 */
3174
		if (cfq_should_wait_busy(cfqd, cfqq)) {
3175 3176 3177 3178
			unsigned long extend_sl = cfqd->cfq_slice_idle;
			if (!cfqd->cfq_slice_idle)
				extend_sl = cfqd->cfq_group_idle;
			cfqq->slice_end = jiffies + extend_sl;
3179
			cfq_mark_cfqq_wait_busy(cfqq);
3180
			cfq_log_cfqq(cfqd, cfqq, "will busy wait");
3181 3182
		}

3183
		/*
3184 3185 3186 3187 3188 3189
		 * Idling is not enabled on:
		 * - expired queues
		 * - idle-priority queues
		 * - async queues
		 * - queues with still some requests queued
		 * - when there is a close cooperator
3190
		 */
3191
		if (cfq_slice_used(cfqq) || cfq_class_idle(cfqq))
3192
			cfq_slice_expired(cfqd, 1);
3193 3194
		else if (sync && cfqq_empty &&
			 !cfq_close_cooperator(cfqd, cfqq)) {
3195
			cfq_arm_slice_timer(cfqd);
3196
		}
3197
	}
J
Jens Axboe 已提交
3198

3199
	if (!cfqd->rq_in_driver)
3200
		cfq_schedule_dispatch(cfqd);
L
Linus Torvalds 已提交
3201 3202
}

3203
static inline int __cfq_may_queue(struct cfq_queue *cfqq)
3204
{
3205
	if (cfq_cfqq_wait_request(cfqq) && !cfq_cfqq_must_alloc_slice(cfqq)) {
J
Jens Axboe 已提交
3206
		cfq_mark_cfqq_must_alloc_slice(cfqq);
3207
		return ELV_MQUEUE_MUST;
J
Jens Axboe 已提交
3208
	}
L
Linus Torvalds 已提交
3209

3210 3211 3212
	return ELV_MQUEUE_MAY;
}

3213
static int cfq_may_queue(struct request_queue *q, int rw)
3214 3215 3216
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
	struct task_struct *tsk = current;
3217
	struct cfq_io_cq *cic;
3218 3219 3220 3221 3222 3223 3224 3225
	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
	 */
3226
	cic = cfq_cic_lookup(cfqd, tsk->io_context);
3227 3228 3229
	if (!cic)
		return ELV_MQUEUE_MAY;

3230
	cfqq = cic_to_cfqq(cic, rw_is_sync(rw));
3231
	if (cfqq) {
3232
		cfq_init_prio_data(cfqq, cic);
3233

3234
		return __cfq_may_queue(cfqq);
3235 3236 3237
	}

	return ELV_MQUEUE_MAY;
L
Linus Torvalds 已提交
3238 3239 3240 3241 3242
}

/*
 * queue lock held here
 */
3243
static void cfq_put_request(struct request *rq)
L
Linus Torvalds 已提交
3244
{
J
Jens Axboe 已提交
3245
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
L
Linus Torvalds 已提交
3246

J
Jens Axboe 已提交
3247
	if (cfqq) {
3248
		const int rw = rq_data_dir(rq);
L
Linus Torvalds 已提交
3249

3250 3251
		BUG_ON(!cfqq->allocated[rw]);
		cfqq->allocated[rw]--;
L
Linus Torvalds 已提交
3252

3253
		/* Put down rq reference on cfqg */
T
Tejun Heo 已提交
3254
		blkg_put(cfqg_to_blkg(RQ_CFQG(rq)));
3255 3256
		rq->elv.priv[0] = NULL;
		rq->elv.priv[1] = NULL;
3257

L
Linus Torvalds 已提交
3258 3259 3260 3261
		cfq_put_queue(cfqq);
	}
}

J
Jeff Moyer 已提交
3262
static struct cfq_queue *
3263
cfq_merge_cfqqs(struct cfq_data *cfqd, struct cfq_io_cq *cic,
J
Jeff Moyer 已提交
3264 3265 3266 3267
		struct cfq_queue *cfqq)
{
	cfq_log_cfqq(cfqd, cfqq, "merging with queue %p", cfqq->new_cfqq);
	cic_set_cfqq(cic, cfqq->new_cfqq, 1);
3268
	cfq_mark_cfqq_coop(cfqq->new_cfqq);
J
Jeff Moyer 已提交
3269 3270 3271 3272
	cfq_put_queue(cfqq);
	return cic_to_cfqq(cic, 1);
}

3273 3274 3275 3276 3277
/*
 * Returns NULL if a new cfqq should be allocated, or the old cfqq if this
 * was the last process referring to said cfqq.
 */
static struct cfq_queue *
3278
split_cfqq(struct cfq_io_cq *cic, struct cfq_queue *cfqq)
3279 3280 3281 3282
{
	if (cfqq_process_refs(cfqq) == 1) {
		cfqq->pid = current->pid;
		cfq_clear_cfqq_coop(cfqq);
3283
		cfq_clear_cfqq_split_coop(cfqq);
3284 3285 3286 3287
		return cfqq;
	}

	cic_set_cfqq(cic, NULL, 1);
3288 3289 3290

	cfq_put_cooperator(cfqq);

3291 3292 3293
	cfq_put_queue(cfqq);
	return NULL;
}
L
Linus Torvalds 已提交
3294
/*
3295
 * Allocate cfq data structures associated with this request.
L
Linus Torvalds 已提交
3296
 */
3297
static int
3298 3299
cfq_set_request(struct request_queue *q, struct request *rq, struct bio *bio,
		gfp_t gfp_mask)
L
Linus Torvalds 已提交
3300 3301
{
	struct cfq_data *cfqd = q->elevator->elevator_data;
3302
	struct cfq_io_cq *cic = icq_to_cic(rq->elv.icq);
L
Linus Torvalds 已提交
3303
	const int rw = rq_data_dir(rq);
3304
	const bool is_sync = rq_is_sync(rq);
3305
	struct cfq_queue *cfqq;
3306
	unsigned int changed;
L
Linus Torvalds 已提交
3307 3308 3309

	might_sleep_if(gfp_mask & __GFP_WAIT);

3310
	spin_lock_irq(q->queue_lock);
3311 3312

	/* handle changed notifications */
3313 3314
	changed = icq_get_changed(&cic->icq);
	if (unlikely(changed & ICQ_IOPRIO_CHANGED))
3315
		changed_ioprio(cic, bio);
3316
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3317 3318
	if (unlikely(changed & ICQ_CGROUP_CHANGED))
		changed_cgroup(cic);
3319
#endif
3320

3321
new_queue:
3322
	cfqq = cic_to_cfqq(cic, is_sync);
3323
	if (!cfqq || cfqq == &cfqd->oom_cfqq) {
3324
		cfqq = cfq_get_queue(cfqd, is_sync, cic, bio, gfp_mask);
3325
		cic_set_cfqq(cic, cfqq, is_sync);
J
Jeff Moyer 已提交
3326
	} else {
3327 3328 3329
		/*
		 * If the queue was seeky for too long, break it apart.
		 */
3330
		if (cfq_cfqq_coop(cfqq) && cfq_cfqq_split_coop(cfqq)) {
3331 3332 3333 3334 3335 3336
			cfq_log_cfqq(cfqd, cfqq, "breaking apart cfqq");
			cfqq = split_cfqq(cic, cfqq);
			if (!cfqq)
				goto new_queue;
		}

J
Jeff Moyer 已提交
3337 3338 3339 3340 3341 3342 3343 3344
		/*
		 * Check to see if this queue is scheduled to merge with
		 * another, closely cooperating queue.  The merging of
		 * queues happens here as it must be done in process context.
		 * The reference on new_cfqq was taken in merge_cfqqs.
		 */
		if (cfqq->new_cfqq)
			cfqq = cfq_merge_cfqqs(cfqd, cic, cfqq);
3345
	}
L
Linus Torvalds 已提交
3346 3347 3348

	cfqq->allocated[rw]++;

3349
	cfqq->ref++;
T
Tejun Heo 已提交
3350
	blkg_get(cfqg_to_blkg(cfqq->cfqg));
3351
	rq->elv.priv[0] = cfqq;
T
Tejun Heo 已提交
3352
	rq->elv.priv[1] = cfqq->cfqg;
3353
	spin_unlock_irq(q->queue_lock);
J
Jens Axboe 已提交
3354
	return 0;
L
Linus Torvalds 已提交
3355 3356
}

3357
static void cfq_kick_queue(struct work_struct *work)
3358
{
3359
	struct cfq_data *cfqd =
3360
		container_of(work, struct cfq_data, unplug_work);
3361
	struct request_queue *q = cfqd->queue;
3362

3363
	spin_lock_irq(q->queue_lock);
3364
	__blk_run_queue(cfqd->queue);
3365
	spin_unlock_irq(q->queue_lock);
3366 3367 3368 3369 3370 3371 3372 3373 3374 3375
}

/*
 * 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;
3376
	int timed_out = 1;
3377

3378 3379
	cfq_log(cfqd, "idle timer fired");

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

3382 3383
	cfqq = cfqd->active_queue;
	if (cfqq) {
3384 3385
		timed_out = 0;

3386 3387 3388 3389 3390 3391
		/*
		 * We saw a request before the queue expired, let it through
		 */
		if (cfq_cfqq_must_dispatch(cfqq))
			goto out_kick;

3392 3393 3394
		/*
		 * expired
		 */
3395
		if (cfq_slice_used(cfqq))
3396 3397 3398 3399 3400 3401
			goto expire;

		/*
		 * only expire and reinvoke request handler, if there are
		 * other queues with pending requests
		 */
3402
		if (!cfqd->busy_queues)
3403 3404 3405 3406 3407
			goto out_cont;

		/*
		 * not expired and it has a request pending, let it dispatch
		 */
3408
		if (!RB_EMPTY_ROOT(&cfqq->sort_list))
3409
			goto out_kick;
3410 3411 3412 3413 3414

		/*
		 * Queue depth flag is reset only when the idle didn't succeed
		 */
		cfq_clear_cfqq_deep(cfqq);
3415 3416
	}
expire:
3417
	cfq_slice_expired(cfqd, timed_out);
3418
out_kick:
3419
	cfq_schedule_dispatch(cfqd);
3420 3421 3422 3423
out_cont:
	spin_unlock_irqrestore(cfqd->queue->queue_lock, flags);
}

J
Jens Axboe 已提交
3424 3425 3426
static void cfq_shutdown_timer_wq(struct cfq_data *cfqd)
{
	del_timer_sync(&cfqd->idle_slice_timer);
3427
	cancel_work_sync(&cfqd->unplug_work);
J
Jens Axboe 已提交
3428
}
3429

3430 3431 3432 3433 3434 3435 3436 3437 3438 3439
static void cfq_put_async_queues(struct cfq_data *cfqd)
{
	int i;

	for (i = 0; i < IOPRIO_BE_NR; i++) {
		if (cfqd->async_cfqq[0][i])
			cfq_put_queue(cfqd->async_cfqq[0][i]);
		if (cfqd->async_cfqq[1][i])
			cfq_put_queue(cfqd->async_cfqq[1][i]);
	}
3440 3441 3442

	if (cfqd->async_idle_cfqq)
		cfq_put_queue(cfqd->async_idle_cfqq);
3443 3444
}

J
Jens Axboe 已提交
3445
static void cfq_exit_queue(struct elevator_queue *e)
L
Linus Torvalds 已提交
3446
{
3447
	struct cfq_data *cfqd = e->elevator_data;
3448
	struct request_queue *q = cfqd->queue;
3449

J
Jens Axboe 已提交
3450
	cfq_shutdown_timer_wq(cfqd);
3451

3452
	spin_lock_irq(q->queue_lock);
3453

3454
	if (cfqd->active_queue)
3455
		__cfq_slice_expired(cfqd, cfqd->active_queue, 0);
3456

3457
	cfq_put_async_queues(cfqd);
3458 3459 3460

	spin_unlock_irq(q->queue_lock);

3461 3462
	cfq_shutdown_timer_wq(cfqd);

3463 3464
#ifndef CONFIG_CFQ_GROUP_IOSCHED
	kfree(cfqd->root_group);
3465
#endif
3466
	update_root_blkg_pd(q, BLKIO_POLICY_PROP);
3467
	kfree(cfqd);
L
Linus Torvalds 已提交
3468 3469
}

3470
static int cfq_init_queue(struct request_queue *q)
L
Linus Torvalds 已提交
3471 3472
{
	struct cfq_data *cfqd;
3473
	struct blkio_group *blkg __maybe_unused;
3474
	int i;
L
Linus Torvalds 已提交
3475

3476
	cfqd = kmalloc_node(sizeof(*cfqd), GFP_KERNEL | __GFP_ZERO, q->node);
3477
	if (!cfqd)
3478
		return -ENOMEM;
3479

3480 3481 3482
	cfqd->queue = q;
	q->elevator->elevator_data = cfqd;

3483 3484 3485
	/* Init root service tree */
	cfqd->grp_service_tree = CFQ_RB_ROOT;

3486
	/* Init root group and prefer root group over other groups by default */
3487
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3488 3489
	rcu_read_lock();
	spin_lock_irq(q->queue_lock);
3490

3491 3492 3493
	blkg = blkg_lookup_create(&blkio_root_cgroup, q, BLKIO_POLICY_PROP,
				  true);
	if (!IS_ERR(blkg))
3494
		cfqd->root_group = blkg_to_cfqg(blkg);
3495 3496 3497 3498 3499 3500 3501 3502 3503 3504

	spin_unlock_irq(q->queue_lock);
	rcu_read_unlock();
#else
	cfqd->root_group = kzalloc_node(sizeof(*cfqd->root_group),
					GFP_KERNEL, cfqd->queue->node);
	if (cfqd->root_group)
		cfq_init_cfqg_base(cfqd->root_group);
#endif
	if (!cfqd->root_group) {
3505
		kfree(cfqd);
3506
		return -ENOMEM;
3507 3508
	}

3509
	cfqd->root_group->weight = 2*BLKIO_WEIGHT_DEFAULT;
3510

3511 3512 3513 3514 3515 3516 3517 3518
	/*
	 * Not strictly needed (since RB_ROOT just clears the node and we
	 * zeroed cfqd on alloc), but better be safe in case someone decides
	 * to add magic to the rb code
	 */
	for (i = 0; i < CFQ_PRIO_LISTS; i++)
		cfqd->prio_trees[i] = RB_ROOT;

3519 3520 3521
	/*
	 * Our fallback cfqq if cfq_find_alloc_queue() runs into OOM issues.
	 * Grab a permanent reference to it, so that the normal code flow
3522 3523 3524
	 * will not attempt to free it.  oom_cfqq is linked to root_group
	 * but shouldn't hold a reference as it'll never be unlinked.  Lose
	 * the reference from linking right away.
3525 3526
	 */
	cfq_init_cfqq(cfqd, &cfqd->oom_cfqq, 1, 0);
3527
	cfqd->oom_cfqq.ref++;
T
Tejun Heo 已提交
3528 3529

	spin_lock_irq(q->queue_lock);
3530
	cfq_link_cfqq_cfqg(&cfqd->oom_cfqq, cfqd->root_group);
T
Tejun Heo 已提交
3531 3532
	blkg_put(cfqg_to_blkg(cfqd->root_group));
	spin_unlock_irq(q->queue_lock);
L
Linus Torvalds 已提交
3533

3534 3535 3536 3537
	init_timer(&cfqd->idle_slice_timer);
	cfqd->idle_slice_timer.function = cfq_idle_slice_timer;
	cfqd->idle_slice_timer.data = (unsigned long) cfqd;

3538
	INIT_WORK(&cfqd->unplug_work, cfq_kick_queue);
3539

L
Linus Torvalds 已提交
3540
	cfqd->cfq_quantum = cfq_quantum;
3541 3542
	cfqd->cfq_fifo_expire[0] = cfq_fifo_expire[0];
	cfqd->cfq_fifo_expire[1] = cfq_fifo_expire[1];
L
Linus Torvalds 已提交
3543 3544
	cfqd->cfq_back_max = cfq_back_max;
	cfqd->cfq_back_penalty = cfq_back_penalty;
3545 3546 3547 3548
	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;
3549
	cfqd->cfq_group_idle = cfq_group_idle;
3550
	cfqd->cfq_latency = 1;
3551
	cfqd->hw_tag = -1;
3552 3553 3554 3555
	/*
	 * we optimistically start assuming sync ops weren't delayed in last
	 * second, in order to have larger depth for async operations.
	 */
3556
	cfqd->last_delayed_sync = jiffies - HZ;
3557
	return 0;
L
Linus Torvalds 已提交
3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578
}

/*
 * 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)				\
J
Jens Axboe 已提交
3579
static ssize_t __FUNC(struct elevator_queue *e, char *page)		\
L
Linus Torvalds 已提交
3580
{									\
3581
	struct cfq_data *cfqd = e->elevator_data;			\
L
Linus Torvalds 已提交
3582 3583 3584 3585 3586 3587
	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);
3588 3589
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);
3590 3591
SHOW_FUNCTION(cfq_back_seek_max_show, cfqd->cfq_back_max, 0);
SHOW_FUNCTION(cfq_back_seek_penalty_show, cfqd->cfq_back_penalty, 0);
3592
SHOW_FUNCTION(cfq_slice_idle_show, cfqd->cfq_slice_idle, 1);
3593
SHOW_FUNCTION(cfq_group_idle_show, cfqd->cfq_group_idle, 1);
3594 3595 3596
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);
3597
SHOW_FUNCTION(cfq_low_latency_show, cfqd->cfq_latency, 0);
L
Linus Torvalds 已提交
3598 3599 3600
#undef SHOW_FUNCTION

#define STORE_FUNCTION(__FUNC, __PTR, MIN, MAX, __CONV)			\
J
Jens Axboe 已提交
3601
static ssize_t __FUNC(struct elevator_queue *e, const char *page, size_t count)	\
L
Linus Torvalds 已提交
3602
{									\
3603
	struct cfq_data *cfqd = e->elevator_data;			\
L
Linus Torvalds 已提交
3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616
	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);
3617 3618 3619 3620
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);
3621
STORE_FUNCTION(cfq_back_seek_max_store, &cfqd->cfq_back_max, 0, UINT_MAX, 0);
3622 3623
STORE_FUNCTION(cfq_back_seek_penalty_store, &cfqd->cfq_back_penalty, 1,
		UINT_MAX, 0);
3624
STORE_FUNCTION(cfq_slice_idle_store, &cfqd->cfq_slice_idle, 0, UINT_MAX, 1);
3625
STORE_FUNCTION(cfq_group_idle_store, &cfqd->cfq_group_idle, 0, UINT_MAX, 1);
3626 3627
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);
3628 3629
STORE_FUNCTION(cfq_slice_async_rq_store, &cfqd->cfq_slice_async_rq, 1,
		UINT_MAX, 0);
3630
STORE_FUNCTION(cfq_low_latency_store, &cfqd->cfq_latency, 0, 1, 0);
L
Linus Torvalds 已提交
3631 3632
#undef STORE_FUNCTION

3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645
#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),
3646
	CFQ_ATTR(group_idle),
3647
	CFQ_ATTR(low_latency),
3648
	__ATTR_NULL
L
Linus Torvalds 已提交
3649 3650 3651 3652 3653 3654 3655
};

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,
3656
		.elevator_allow_merge_fn =	cfq_allow_merge,
D
Divyesh Shah 已提交
3657
		.elevator_bio_merged_fn =	cfq_bio_merged,
3658
		.elevator_dispatch_fn =		cfq_dispatch_requests,
L
Linus Torvalds 已提交
3659
		.elevator_add_req_fn =		cfq_insert_request,
3660
		.elevator_activate_req_fn =	cfq_activate_request,
L
Linus Torvalds 已提交
3661 3662
		.elevator_deactivate_req_fn =	cfq_deactivate_request,
		.elevator_completed_req_fn =	cfq_completed_request,
3663 3664
		.elevator_former_req_fn =	elv_rb_former_request,
		.elevator_latter_req_fn =	elv_rb_latter_request,
3665
		.elevator_init_icq_fn =		cfq_init_icq,
3666
		.elevator_exit_icq_fn =		cfq_exit_icq,
L
Linus Torvalds 已提交
3667 3668 3669 3670 3671 3672
		.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,
	},
3673 3674
	.icq_size	=	sizeof(struct cfq_io_cq),
	.icq_align	=	__alignof__(struct cfq_io_cq),
3675
	.elevator_attrs =	cfq_attrs,
3676
	.elevator_name	=	"cfq",
L
Linus Torvalds 已提交
3677 3678 3679
	.elevator_owner =	THIS_MODULE,
};

3680 3681 3682
#ifdef CONFIG_CFQ_GROUP_IOSCHED
static struct blkio_policy_type blkio_policy_cfq = {
	.ops = {
3683
		.blkio_init_group_fn =		cfq_init_blkio_group,
3684 3685
		.blkio_update_group_weight_fn =	cfq_update_blkio_group_weight,
	},
3686
	.plid = BLKIO_POLICY_PROP,
3687
	.pdata_size = sizeof(struct cfq_group),
3688 3689 3690
};
#endif

L
Linus Torvalds 已提交
3691 3692
static int __init cfq_init(void)
{
3693 3694
	int ret;

3695 3696 3697 3698 3699 3700 3701 3702
	/*
	 * could be 0 on HZ < 1000 setups
	 */
	if (!cfq_slice_async)
		cfq_slice_async = 1;
	if (!cfq_slice_idle)
		cfq_slice_idle = 1;

3703 3704 3705 3706 3707 3708
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	if (!cfq_group_idle)
		cfq_group_idle = 1;
#else
		cfq_group_idle = 0;
#endif
3709 3710
	cfq_pool = KMEM_CACHE(cfq_queue, 0);
	if (!cfq_pool)
L
Linus Torvalds 已提交
3711 3712
		return -ENOMEM;

3713 3714 3715 3716 3717 3718
	ret = elv_register(&iosched_cfq);
	if (ret) {
		kmem_cache_destroy(cfq_pool);
		return ret;
	}

3719
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3720
	blkio_policy_register(&blkio_policy_cfq);
3721
#endif
3722
	return 0;
L
Linus Torvalds 已提交
3723 3724 3725 3726
}

static void __exit cfq_exit(void)
{
3727
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3728
	blkio_policy_unregister(&blkio_policy_cfq);
3729
#endif
L
Linus Torvalds 已提交
3730
	elv_unregister(&iosched_cfq);
3731
	kmem_cache_destroy(cfq_pool);
L
Linus Torvalds 已提交
3732 3733 3734 3735 3736 3737 3738 3739
}

module_init(cfq_init);
module_exit(cfq_exit);

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