cfq-iosched.c 94.6 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 471 472
static struct cfq_queue *cfq_get_queue(struct cfq_data *cfqd, bool is_sync,
				       struct io_context *ioc, struct bio *bio,
				       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 io_context *ioc)
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(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 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594
		break;
	case IOPRIO_CLASS_RT:
		cfqq->ioprio = task_ioprio(ioc);
		cfqq->ioprio_class = IOPRIO_CLASS_RT;
		break;
	case IOPRIO_CLASS_BE:
		cfqq->ioprio = task_ioprio(ioc);
		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
		new_cfqq = cfq_get_queue(cfqd, BLK_RW_ASYNC, cic->icq.ioc,
2617
					 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
cfq_find_alloc_queue(struct cfq_data *cfqd, bool is_sync,
2675
		     struct io_context *ioc, struct bio *bio, gfp_t gfp_mask)
2676
{
2677
	struct blkio_cgroup *blkcg;
2678
	struct cfq_queue *cfqq, *new_cfqq = NULL;
2679
	struct cfq_io_cq *cic;
2680
	struct cfq_group *cfqg;
2681 2682

retry:
2683 2684
	rcu_read_lock();

2685
	blkcg = bio_blkio_cgroup(bio);
2686

2687 2688
	cfqg = cfq_lookup_create_cfqg(cfqd, blkcg);

2689
	cic = cfq_cic_lookup(cfqd, ioc);
2690 2691
	/* cic always exists here */
	cfqq = cic_to_cfqq(cic, is_sync);
2692

2693 2694 2695 2696 2697 2698
	/*
	 * 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;
2699 2700 2701 2702
		if (new_cfqq) {
			cfqq = new_cfqq;
			new_cfqq = NULL;
		} else if (gfp_mask & __GFP_WAIT) {
2703
			rcu_read_unlock();
2704
			spin_unlock_irq(cfqd->queue->queue_lock);
2705
			new_cfqq = kmem_cache_alloc_node(cfq_pool,
2706
					gfp_mask | __GFP_ZERO,
2707
					cfqd->queue->node);
2708
			spin_lock_irq(cfqd->queue->queue_lock);
2709 2710
			if (new_cfqq)
				goto retry;
2711
		} else {
2712 2713 2714
			cfqq = kmem_cache_alloc_node(cfq_pool,
					gfp_mask | __GFP_ZERO,
					cfqd->queue->node);
2715 2716
		}

2717 2718 2719
		if (cfqq) {
			cfq_init_cfqq(cfqd, cfqq, current->pid, is_sync);
			cfq_init_prio_data(cfqq, ioc);
2720
			cfq_link_cfqq_cfqg(cfqq, cfqg);
2721 2722 2723
			cfq_log_cfqq(cfqd, cfqq, "alloced");
		} else
			cfqq = &cfqd->oom_cfqq;
2724 2725 2726 2727 2728
	}

	if (new_cfqq)
		kmem_cache_free(cfq_pool, new_cfqq);

2729
	rcu_read_unlock();
2730 2731 2732
	return cfqq;
}

2733 2734 2735
static struct cfq_queue **
cfq_async_queue_prio(struct cfq_data *cfqd, int ioprio_class, int ioprio)
{
2736
	switch (ioprio_class) {
2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747
	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();
	}
}

2748
static struct cfq_queue *
2749
cfq_get_queue(struct cfq_data *cfqd, bool is_sync, struct io_context *ioc,
2750
	      struct bio *bio, gfp_t gfp_mask)
2751
{
2752 2753
	const int ioprio = task_ioprio(ioc);
	const int ioprio_class = task_ioprio_class(ioc);
2754
	struct cfq_queue **async_cfqq = NULL;
2755 2756
	struct cfq_queue *cfqq = NULL;

2757 2758 2759 2760 2761
	if (!is_sync) {
		async_cfqq = cfq_async_queue_prio(cfqd, ioprio_class, ioprio);
		cfqq = *async_cfqq;
	}

2762
	if (!cfqq)
2763
		cfqq = cfq_find_alloc_queue(cfqd, is_sync, ioc, bio, gfp_mask);
2764 2765 2766 2767

	/*
	 * pin the queue now that it's allocated, scheduler exit will prune it
	 */
2768
	if (!is_sync && !(*async_cfqq)) {
2769
		cfqq->ref++;
2770
		*async_cfqq = cfqq;
2771 2772
	}

2773
	cfqq->ref++;
2774 2775 2776
	return cfqq;
}

2777
static void
2778
__cfq_update_io_thinktime(struct cfq_ttime *ttime, unsigned long slice_idle)
L
Linus Torvalds 已提交
2779
{
2780 2781
	unsigned long elapsed = jiffies - ttime->last_end_request;
	elapsed = min(elapsed, 2UL * slice_idle);
2782

2783 2784 2785 2786 2787 2788 2789
	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,
2790
			struct cfq_io_cq *cic)
2791
{
2792
	if (cfq_cfqq_sync(cfqq)) {
2793
		__cfq_update_io_thinktime(&cic->ttime, cfqd->cfq_slice_idle);
2794 2795 2796
		__cfq_update_io_thinktime(&cfqq->service_tree->ttime,
			cfqd->cfq_slice_idle);
	}
S
Shaohua Li 已提交
2797 2798 2799
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	__cfq_update_io_thinktime(&cfqq->cfqg->ttime, cfqd->cfq_group_idle);
#endif
2800
}
L
Linus Torvalds 已提交
2801

2802
static void
2803
cfq_update_io_seektime(struct cfq_data *cfqd, struct cfq_queue *cfqq,
J
Jens Axboe 已提交
2804
		       struct request *rq)
2805
{
2806
	sector_t sdist = 0;
2807
	sector_t n_sec = blk_rq_sectors(rq);
2808 2809 2810 2811 2812 2813
	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);
	}
2814

2815
	cfqq->seek_history <<= 1;
2816 2817 2818 2819
	if (blk_queue_nonrot(cfqd->queue))
		cfqq->seek_history |= (n_sec < CFQQ_SECT_THR_NONROT);
	else
		cfqq->seek_history |= (sdist > CFQQ_SEEK_THR);
2820
}
L
Linus Torvalds 已提交
2821

2822 2823 2824 2825 2826 2827
/*
 * 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,
2828
		       struct cfq_io_cq *cic)
2829
{
2830
	int old_idle, enable_idle;
2831

2832 2833 2834 2835
	/*
	 * Don't idle for async or idle io prio class
	 */
	if (!cfq_cfqq_sync(cfqq) || cfq_class_idle(cfqq))
2836 2837
		return;

2838
	enable_idle = old_idle = cfq_cfqq_idle_window(cfqq);
L
Linus Torvalds 已提交
2839

2840 2841 2842
	if (cfqq->queued[0] + cfqq->queued[1] >= 4)
		cfq_mark_cfqq_deep(cfqq);

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

2856 2857 2858 2859 2860 2861 2862
	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);
	}
2863
}
L
Linus Torvalds 已提交
2864

2865 2866 2867 2868
/*
 * 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.
 */
2869
static bool
2870
cfq_should_preempt(struct cfq_data *cfqd, struct cfq_queue *new_cfqq,
J
Jens Axboe 已提交
2871
		   struct request *rq)
2872
{
J
Jens Axboe 已提交
2873
	struct cfq_queue *cfqq;
2874

J
Jens Axboe 已提交
2875 2876
	cfqq = cfqd->active_queue;
	if (!cfqq)
2877
		return false;
2878

J
Jens Axboe 已提交
2879
	if (cfq_class_idle(new_cfqq))
2880
		return false;
2881 2882

	if (cfq_class_idle(cfqq))
2883
		return true;
2884

2885 2886 2887 2888 2889 2890
	/*
	 * 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;

2891 2892 2893 2894
	/*
	 * if the new request is sync, but the currently running queue is
	 * not, let the sync request have priority.
	 */
J
Jens Axboe 已提交
2895
	if (rq_is_sync(rq) && !cfq_cfqq_sync(cfqq))
2896
		return true;
2897

2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910
	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;

2911 2912 2913 2914
	/*
	 * 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.
	 */
2915
	if ((rq->cmd_flags & REQ_PRIO) && !cfqq->prio_pending)
2916 2917
		return true;

2918 2919 2920 2921
	/*
	 * Allow an RT request to pre-empt an ongoing non-RT cfqq timeslice.
	 */
	if (cfq_class_rt(new_cfqq) && !cfq_class_rt(cfqq))
2922
		return true;
2923

2924 2925 2926 2927
	/* 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;

2928
	if (!cfqd->active_cic || !cfq_cfqq_wait_request(cfqq))
2929
		return false;
2930 2931 2932 2933 2934

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

2938
	return false;
2939 2940 2941 2942 2943 2944 2945 2946
}

/*
 * 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 已提交
2947 2948
	enum wl_type_t old_type = cfqq_type(cfqd->active_queue);

2949
	cfq_log_cfqq(cfqd, cfqq, "preempt");
S
Shaohua Li 已提交
2950
	cfq_slice_expired(cfqd, 1);
2951

2952 2953 2954 2955
	/*
	 * workload type is changed, don't save slice, otherwise preempt
	 * doesn't happen
	 */
S
Shaohua Li 已提交
2956
	if (old_type != cfqq_type(cfqq))
2957 2958
		cfqq->cfqg->saved_workload_slice = 0;

2959 2960 2961 2962 2963
	/*
	 * 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));
2964 2965

	cfq_service_tree_add(cfqd, cfqq, 1);
2966

2967 2968
	cfqq->slice_end = 0;
	cfq_mark_cfqq_slice_new(cfqq);
2969 2970 2971
}

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

2981
	cfqd->rq_queued++;
2982 2983
	if (rq->cmd_flags & REQ_PRIO)
		cfqq->prio_pending++;
2984

2985
	cfq_update_io_thinktime(cfqd, cfqq, cic);
2986
	cfq_update_io_seektime(cfqd, cfqq, rq);
J
Jens Axboe 已提交
2987 2988
	cfq_update_idle_window(cfqd, cfqq, cic);

2989
	cfqq->last_request_pos = blk_rq_pos(rq) + blk_rq_sectors(rq);
2990 2991 2992

	if (cfqq == cfqd->active_queue) {
		/*
2993 2994 2995
		 * 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
2996 2997
		 * and merging. If the request is already larger than a single
		 * page, let it rip immediately. For that case we assume that
2998 2999 3000
		 * 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.
3001
		 */
3002
		if (cfq_cfqq_wait_request(cfqq)) {
3003 3004
			if (blk_rq_bytes(rq) > PAGE_CACHE_SIZE ||
			    cfqd->busy_queues > 1) {
3005
				cfq_del_timer(cfqd, cfqq);
3006
				cfq_clear_cfqq_wait_request(cfqq);
3007
				__blk_run_queue(cfqd->queue);
3008
			} else {
3009
				cfq_blkiocg_update_idle_time_stats(
3010 3011
						cfqg_to_blkg(cfqq->cfqg),
						&blkio_policy_cfq);
3012
				cfq_mark_cfqq_must_dispatch(cfqq);
3013
			}
3014
		}
J
Jens Axboe 已提交
3015
	} else if (cfq_should_preempt(cfqd, cfqq, rq)) {
3016 3017 3018
		/*
		 * not the active queue - expire current slice if it is
		 * idle and has expired it's mean thinktime or this new queue
3019 3020
		 * has some old slice time left and is of higher priority or
		 * this new queue is RT and the current one is BE
3021 3022
		 */
		cfq_preempt_queue(cfqd, cfqq);
3023
		__blk_run_queue(cfqd->queue);
3024
	}
L
Linus Torvalds 已提交
3025 3026
}

3027
static void cfq_insert_request(struct request_queue *q, struct request *rq)
L
Linus Torvalds 已提交
3028
{
3029
	struct cfq_data *cfqd = q->elevator->elevator_data;
J
Jens Axboe 已提交
3030
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
3031

3032
	cfq_log_cfqq(cfqd, cfqq, "insert_request");
3033
	cfq_init_prio_data(cfqq, RQ_CIC(rq)->icq.ioc);
L
Linus Torvalds 已提交
3034

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

3045 3046 3047 3048 3049 3050
/*
 * 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 已提交
3051 3052
	struct cfq_queue *cfqq = cfqd->active_queue;

3053 3054
	if (cfqd->rq_in_driver > cfqd->hw_tag_est_depth)
		cfqd->hw_tag_est_depth = cfqd->rq_in_driver;
3055 3056 3057

	if (cfqd->hw_tag == 1)
		return;
3058 3059

	if (cfqd->rq_queued <= CFQ_HW_QUEUE_MIN &&
3060
	    cfqd->rq_in_driver <= CFQ_HW_QUEUE_MIN)
3061 3062
		return;

S
Shaohua Li 已提交
3063 3064 3065 3066 3067 3068 3069
	/*
	 * 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] <
3070
	    CFQ_HW_QUEUE_MIN && cfqd->rq_in_driver < CFQ_HW_QUEUE_MIN)
S
Shaohua Li 已提交
3071 3072
		return;

3073 3074 3075
	if (cfqd->hw_tag_samples++ < 50)
		return;

3076
	if (cfqd->hw_tag_est_depth >= CFQ_HW_QUEUE_MIN)
3077 3078 3079 3080 3081
		cfqd->hw_tag = 1;
	else
		cfqd->hw_tag = 0;
}

3082 3083
static bool cfq_should_wait_busy(struct cfq_data *cfqd, struct cfq_queue *cfqq)
{
3084
	struct cfq_io_cq *cic = cfqd->active_cic;
3085

3086 3087 3088 3089
	/* If the queue already has requests, don't wait */
	if (!RB_EMPTY_ROOT(&cfqq->sort_list))
		return false;

3090 3091 3092 3093
	/* If there are other queues in the group, don't wait */
	if (cfqq->cfqg->nr_cfqq > 1)
		return false;

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

3098 3099 3100 3101
	if (cfq_slice_used(cfqq))
		return true;

	/* if slice left is less than think time, wait busy */
3102 3103
	if (cic && sample_valid(cic->ttime.ttime_samples)
	    && (cfqq->slice_end - jiffies < cic->ttime.ttime_mean))
3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118
		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;
}

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

3126
	now = jiffies;
3127 3128
	cfq_log_cfqq(cfqd, cfqq, "complete rqnoidle %d",
		     !!(rq->cmd_flags & REQ_NOIDLE));
L
Linus Torvalds 已提交
3129

3130 3131
	cfq_update_hw_tag(cfqd);

3132
	WARN_ON(!cfqd->rq_in_driver);
J
Jens Axboe 已提交
3133
	WARN_ON(!cfqq->dispatched);
3134
	cfqd->rq_in_driver--;
J
Jens Axboe 已提交
3135
	cfqq->dispatched--;
3136
	(RQ_CFQG(rq))->dispatched--;
3137
	cfq_blkiocg_update_completion_stats(cfqg_to_blkg(cfqq->cfqg),
3138 3139 3140
			&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 已提交
3141

3142
	cfqd->rq_in_flight[cfq_cfqq_sync(cfqq)]--;
3143

3144
	if (sync) {
3145 3146
		struct cfq_rb_root *service_tree;

3147
		RQ_CIC(rq)->ttime.last_end_request = now;
3148 3149 3150 3151 3152 3153 3154

		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;
3155 3156
		if (!time_after(rq->start_time + cfqd->cfq_fifo_expire[1], now))
			cfqd->last_delayed_sync = now;
3157
	}
3158

S
Shaohua Li 已提交
3159 3160 3161 3162
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	cfqq->cfqg->ttime.last_end_request = now;
#endif

3163 3164 3165 3166 3167
	/*
	 * 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) {
3168 3169
		const bool cfqq_empty = RB_EMPTY_ROOT(&cfqq->sort_list);

3170 3171 3172 3173
		if (cfq_cfqq_slice_new(cfqq)) {
			cfq_set_prio_slice(cfqd, cfqq);
			cfq_clear_cfqq_slice_new(cfqq);
		}
3174 3175

		/*
3176 3177
		 * Should we wait for next request to come in before we expire
		 * the queue.
3178
		 */
3179
		if (cfq_should_wait_busy(cfqd, cfqq)) {
3180 3181 3182 3183
			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;
3184
			cfq_mark_cfqq_wait_busy(cfqq);
3185
			cfq_log_cfqq(cfqd, cfqq, "will busy wait");
3186 3187
		}

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

3204
	if (!cfqd->rq_in_driver)
3205
		cfq_schedule_dispatch(cfqd);
L
Linus Torvalds 已提交
3206 3207
}

3208
static inline int __cfq_may_queue(struct cfq_queue *cfqq)
3209
{
3210
	if (cfq_cfqq_wait_request(cfqq) && !cfq_cfqq_must_alloc_slice(cfqq)) {
J
Jens Axboe 已提交
3211
		cfq_mark_cfqq_must_alloc_slice(cfqq);
3212
		return ELV_MQUEUE_MUST;
J
Jens Axboe 已提交
3213
	}
L
Linus Torvalds 已提交
3214

3215 3216 3217
	return ELV_MQUEUE_MAY;
}

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

3235
	cfqq = cic_to_cfqq(cic, rw_is_sync(rw));
3236
	if (cfqq) {
3237
		cfq_init_prio_data(cfqq, cic->icq.ioc);
3238

3239
		return __cfq_may_queue(cfqq);
3240 3241 3242
	}

	return ELV_MQUEUE_MAY;
L
Linus Torvalds 已提交
3243 3244 3245 3246 3247
}

/*
 * queue lock held here
 */
3248
static void cfq_put_request(struct request *rq)
L
Linus Torvalds 已提交
3249
{
J
Jens Axboe 已提交
3250
	struct cfq_queue *cfqq = RQ_CFQQ(rq);
L
Linus Torvalds 已提交
3251

J
Jens Axboe 已提交
3252
	if (cfqq) {
3253
		const int rw = rq_data_dir(rq);
L
Linus Torvalds 已提交
3254

3255 3256
		BUG_ON(!cfqq->allocated[rw]);
		cfqq->allocated[rw]--;
L
Linus Torvalds 已提交
3257

3258
		/* Put down rq reference on cfqg */
T
Tejun Heo 已提交
3259
		blkg_put(cfqg_to_blkg(RQ_CFQG(rq)));
3260 3261
		rq->elv.priv[0] = NULL;
		rq->elv.priv[1] = NULL;
3262

L
Linus Torvalds 已提交
3263 3264 3265 3266
		cfq_put_queue(cfqq);
	}
}

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

3278 3279 3280 3281 3282
/*
 * 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 *
3283
split_cfqq(struct cfq_io_cq *cic, struct cfq_queue *cfqq)
3284 3285 3286 3287
{
	if (cfqq_process_refs(cfqq) == 1) {
		cfqq->pid = current->pid;
		cfq_clear_cfqq_coop(cfqq);
3288
		cfq_clear_cfqq_split_coop(cfqq);
3289 3290 3291 3292
		return cfqq;
	}

	cic_set_cfqq(cic, NULL, 1);
3293 3294 3295

	cfq_put_cooperator(cfqq);

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

	might_sleep_if(gfp_mask & __GFP_WAIT);

3315
	spin_lock_irq(q->queue_lock);
3316 3317

	/* handle changed notifications */
3318 3319
	changed = icq_get_changed(&cic->icq);
	if (unlikely(changed & ICQ_IOPRIO_CHANGED))
3320
		changed_ioprio(cic, bio);
3321
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3322 3323
	if (unlikely(changed & ICQ_CGROUP_CHANGED))
		changed_cgroup(cic);
3324
#endif
3325

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

J
Jeff Moyer 已提交
3342 3343 3344 3345 3346 3347 3348 3349
		/*
		 * 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);
3350
	}
L
Linus Torvalds 已提交
3351 3352 3353

	cfqq->allocated[rw]++;

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

3362
static void cfq_kick_queue(struct work_struct *work)
3363
{
3364
	struct cfq_data *cfqd =
3365
		container_of(work, struct cfq_data, unplug_work);
3366
	struct request_queue *q = cfqd->queue;
3367

3368
	spin_lock_irq(q->queue_lock);
3369
	__blk_run_queue(cfqd->queue);
3370
	spin_unlock_irq(q->queue_lock);
3371 3372 3373 3374 3375 3376 3377 3378 3379 3380
}

/*
 * 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;
3381
	int timed_out = 1;
3382

3383 3384
	cfq_log(cfqd, "idle timer fired");

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

3387 3388
	cfqq = cfqd->active_queue;
	if (cfqq) {
3389 3390
		timed_out = 0;

3391 3392 3393 3394 3395 3396
		/*
		 * We saw a request before the queue expired, let it through
		 */
		if (cfq_cfqq_must_dispatch(cfqq))
			goto out_kick;

3397 3398 3399
		/*
		 * expired
		 */
3400
		if (cfq_slice_used(cfqq))
3401 3402 3403 3404 3405 3406
			goto expire;

		/*
		 * only expire and reinvoke request handler, if there are
		 * other queues with pending requests
		 */
3407
		if (!cfqd->busy_queues)
3408 3409 3410 3411 3412
			goto out_cont;

		/*
		 * not expired and it has a request pending, let it dispatch
		 */
3413
		if (!RB_EMPTY_ROOT(&cfqq->sort_list))
3414
			goto out_kick;
3415 3416 3417 3418 3419

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

J
Jens Axboe 已提交
3429 3430 3431
static void cfq_shutdown_timer_wq(struct cfq_data *cfqd)
{
	del_timer_sync(&cfqd->idle_slice_timer);
3432
	cancel_work_sync(&cfqd->unplug_work);
J
Jens Axboe 已提交
3433
}
3434

3435 3436 3437 3438 3439 3440 3441 3442 3443 3444
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]);
	}
3445 3446 3447

	if (cfqd->async_idle_cfqq)
		cfq_put_queue(cfqd->async_idle_cfqq);
3448 3449
}

J
Jens Axboe 已提交
3450
static void cfq_exit_queue(struct elevator_queue *e)
L
Linus Torvalds 已提交
3451
{
3452
	struct cfq_data *cfqd = e->elevator_data;
3453
	struct request_queue *q = cfqd->queue;
3454

J
Jens Axboe 已提交
3455
	cfq_shutdown_timer_wq(cfqd);
3456

3457
	spin_lock_irq(q->queue_lock);
3458

3459
	if (cfqd->active_queue)
3460
		__cfq_slice_expired(cfqd, cfqd->active_queue, 0);
3461

3462
	cfq_put_async_queues(cfqd);
3463 3464 3465

	spin_unlock_irq(q->queue_lock);

3466 3467
	cfq_shutdown_timer_wq(cfqd);

3468 3469
#ifndef CONFIG_CFQ_GROUP_IOSCHED
	kfree(cfqd->root_group);
3470
#endif
3471
	update_root_blkg_pd(q, BLKIO_POLICY_PROP);
3472
	kfree(cfqd);
L
Linus Torvalds 已提交
3473 3474
}

3475
static int cfq_init_queue(struct request_queue *q)
L
Linus Torvalds 已提交
3476 3477
{
	struct cfq_data *cfqd;
3478
	struct blkio_group *blkg __maybe_unused;
3479
	int i;
L
Linus Torvalds 已提交
3480

3481
	cfqd = kmalloc_node(sizeof(*cfqd), GFP_KERNEL | __GFP_ZERO, q->node);
3482
	if (!cfqd)
3483
		return -ENOMEM;
3484

3485 3486 3487
	cfqd->queue = q;
	q->elevator->elevator_data = cfqd;

3488 3489 3490
	/* Init root service tree */
	cfqd->grp_service_tree = CFQ_RB_ROOT;

3491
	/* Init root group and prefer root group over other groups by default */
3492
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3493 3494
	rcu_read_lock();
	spin_lock_irq(q->queue_lock);
3495

3496 3497 3498
	blkg = blkg_lookup_create(&blkio_root_cgroup, q, BLKIO_POLICY_PROP,
				  true);
	if (!IS_ERR(blkg))
3499
		cfqd->root_group = blkg_to_cfqg(blkg);
3500 3501 3502 3503 3504 3505 3506 3507 3508 3509

	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) {
3510
		kfree(cfqd);
3511
		return -ENOMEM;
3512 3513
	}

3514
	cfqd->root_group->weight = 2*BLKIO_WEIGHT_DEFAULT;
3515

3516 3517 3518 3519 3520 3521 3522 3523
	/*
	 * 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;

3524 3525 3526
	/*
	 * Our fallback cfqq if cfq_find_alloc_queue() runs into OOM issues.
	 * Grab a permanent reference to it, so that the normal code flow
3527 3528 3529
	 * 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.
3530 3531
	 */
	cfq_init_cfqq(cfqd, &cfqd->oom_cfqq, 1, 0);
3532
	cfqd->oom_cfqq.ref++;
T
Tejun Heo 已提交
3533 3534

	spin_lock_irq(q->queue_lock);
3535
	cfq_link_cfqq_cfqg(&cfqd->oom_cfqq, cfqd->root_group);
T
Tejun Heo 已提交
3536 3537
	blkg_put(cfqg_to_blkg(cfqd->root_group));
	spin_unlock_irq(q->queue_lock);
L
Linus Torvalds 已提交
3538

3539 3540 3541 3542
	init_timer(&cfqd->idle_slice_timer);
	cfqd->idle_slice_timer.function = cfq_idle_slice_timer;
	cfqd->idle_slice_timer.data = (unsigned long) cfqd;

3543
	INIT_WORK(&cfqd->unplug_work, cfq_kick_queue);
3544

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

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

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

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

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

3685 3686 3687
#ifdef CONFIG_CFQ_GROUP_IOSCHED
static struct blkio_policy_type blkio_policy_cfq = {
	.ops = {
3688
		.blkio_init_group_fn =		cfq_init_blkio_group,
3689 3690
		.blkio_update_group_weight_fn =	cfq_update_blkio_group_weight,
	},
3691
	.plid = BLKIO_POLICY_PROP,
3692
	.pdata_size = sizeof(struct cfq_group),
3693 3694 3695
};
#endif

L
Linus Torvalds 已提交
3696 3697
static int __init cfq_init(void)
{
3698 3699
	int ret;

3700 3701 3702 3703 3704 3705 3706 3707
	/*
	 * could be 0 on HZ < 1000 setups
	 */
	if (!cfq_slice_async)
		cfq_slice_async = 1;
	if (!cfq_slice_idle)
		cfq_slice_idle = 1;

3708 3709 3710 3711 3712 3713
#ifdef CONFIG_CFQ_GROUP_IOSCHED
	if (!cfq_group_idle)
		cfq_group_idle = 1;
#else
		cfq_group_idle = 0;
#endif
3714 3715
	cfq_pool = KMEM_CACHE(cfq_queue, 0);
	if (!cfq_pool)
L
Linus Torvalds 已提交
3716 3717
		return -ENOMEM;

3718 3719 3720 3721 3722 3723
	ret = elv_register(&iosched_cfq);
	if (ret) {
		kmem_cache_destroy(cfq_pool);
		return ret;
	}

3724
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3725
	blkio_policy_register(&blkio_policy_cfq);
3726
#endif
3727
	return 0;
L
Linus Torvalds 已提交
3728 3729 3730 3731
}

static void __exit cfq_exit(void)
{
3732
#ifdef CONFIG_CFQ_GROUP_IOSCHED
3733
	blkio_policy_unregister(&blkio_policy_cfq);
3734
#endif
L
Linus Torvalds 已提交
3735
	elv_unregister(&iosched_cfq);
3736
	kmem_cache_destroy(cfq_pool);
L
Linus Torvalds 已提交
3737 3738 3739 3740 3741 3742 3743 3744
}

module_init(cfq_init);
module_exit(cfq_exit);

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