blob: a56ead42d65106ea5e07fc622594fe35c5c403b2 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * kernel/sched.c
3 *
4 * Kernel scheduler and related syscalls
5 *
6 * Copyright (C) 1991-2002 Linus Torvalds
7 *
8 * 1996-12-23 Modified by Dave Grothe to fix bugs in semaphores and
9 * make semaphores SMP safe
10 * 1998-11-19 Implemented schedule_timeout() and related stuff
11 * by Andrea Arcangeli
12 * 2002-01-04 New ultra-scalable O(1) scheduler by Ingo Molnar:
13 * hybrid priority-list and round-robin design with
14 * an array-switch method of distributing timeslices
15 * and per-CPU runqueues. Cleanups and useful suggestions
16 * by Davide Libenzi, preemptible kernel bits by Robert Love.
17 * 2003-09-03 Interactivity tuning by Con Kolivas.
18 * 2004-04-02 Scheduler domains code by Nick Piggin
Ingo Molnarc31f2e82007-07-09 18:52:01 +020019 * 2007-04-15 Work begun on replacing all interactivity tuning with a
20 * fair scheduling design by Con Kolivas.
21 * 2007-05-05 Load balancing (smp-nice) and other improvements
22 * by Peter Williams
23 * 2007-05-06 Interactivity improvements to CFS by Mike Galbraith
24 * 2007-07-01 Group scheduling enhancements by Srivatsa Vaddagiri
Ingo Molnarb9131762008-01-25 21:08:19 +010025 * 2007-11-29 RT balancing improvements by Steven Rostedt, Gregory Haskins,
26 * Thomas Gleixner, Mike Kravetz
Linus Torvalds1da177e2005-04-16 15:20:36 -070027 */
28
29#include <linux/mm.h>
30#include <linux/module.h>
31#include <linux/nmi.h>
32#include <linux/init.h>
Ingo Molnardff06c12007-07-09 18:52:00 +020033#include <linux/uaccess.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070034#include <linux/highmem.h>
35#include <linux/smp_lock.h>
36#include <asm/mmu_context.h>
37#include <linux/interrupt.h>
Randy.Dunlapc59ede72006-01-11 12:17:46 -080038#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070039#include <linux/completion.h>
40#include <linux/kernel_stat.h>
Ingo Molnar9a11b49a2006-07-03 00:24:33 -070041#include <linux/debug_locks.h>
Ingo Molnarcdd6c482009-09-21 12:02:48 +020042#include <linux/perf_event.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070043#include <linux/security.h>
44#include <linux/notifier.h>
45#include <linux/profile.h>
Nigel Cunningham7dfb7102006-12-06 20:34:23 -080046#include <linux/freezer.h>
akpm@osdl.org198e2f12006-01-12 01:05:30 -080047#include <linux/vmalloc.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070048#include <linux/blkdev.h>
49#include <linux/delay.h>
Pavel Emelyanovb4888932007-10-18 23:40:14 -070050#include <linux/pid_namespace.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070051#include <linux/smp.h>
52#include <linux/threads.h>
53#include <linux/timer.h>
54#include <linux/rcupdate.h>
55#include <linux/cpu.h>
56#include <linux/cpuset.h>
57#include <linux/percpu.h>
58#include <linux/kthread.h>
Alexey Dobriyanb5aadf72008-10-06 13:23:43 +040059#include <linux/proc_fs.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070060#include <linux/seq_file.h>
Nick Piggine692ab52007-07-26 13:40:43 +020061#include <linux/sysctl.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070062#include <linux/syscalls.h>
63#include <linux/times.h>
Jay Lan8f0ab512006-09-30 23:28:59 -070064#include <linux/tsacct_kern.h>
bibo maoc6fd91f2006-03-26 01:38:20 -080065#include <linux/kprobes.h>
Shailabh Nagar0ff92242006-07-14 00:24:37 -070066#include <linux/delayacct.h>
Ingo Molnardff06c12007-07-09 18:52:00 +020067#include <linux/unistd.h>
Jens Axboef5ff8422007-09-21 09:19:54 +020068#include <linux/pagemap.h>
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +010069#include <linux/hrtimer.h>
Reynes Philippe30914a52008-03-17 16:19:05 -070070#include <linux/tick.h>
Peter Zijlstraf00b45c2008-04-19 19:45:00 +020071#include <linux/debugfs.h>
72#include <linux/ctype.h>
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +020073#include <linux/ftrace.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070074
Eric Dumazet5517d862007-05-08 00:32:57 -070075#include <asm/tlb.h>
Satyam Sharma838225b2007-10-24 18:23:50 +020076#include <asm/irq_regs.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070077
Gregory Haskins6e0534f2008-05-12 21:21:01 +020078#include "sched_cpupri.h"
79
Steven Rostedta8d154b2009-04-10 09:36:00 -040080#define CREATE_TRACE_POINTS
Steven Rostedtad8d75f2009-04-14 19:39:12 -040081#include <trace/events/sched.h>
Steven Rostedta8d154b2009-04-10 09:36:00 -040082
Linus Torvalds1da177e2005-04-16 15:20:36 -070083/*
84 * Convert user-nice values [ -20 ... 0 ... 19 ]
85 * to static priority [ MAX_RT_PRIO..MAX_PRIO-1 ],
86 * and back.
87 */
88#define NICE_TO_PRIO(nice) (MAX_RT_PRIO + (nice) + 20)
89#define PRIO_TO_NICE(prio) ((prio) - MAX_RT_PRIO - 20)
90#define TASK_NICE(p) PRIO_TO_NICE((p)->static_prio)
91
92/*
93 * 'User priority' is the nice value converted to something we
94 * can work with better when scaling various scheduler parameters,
95 * it's a [ 0 ... 39 ] range.
96 */
97#define USER_PRIO(p) ((p)-MAX_RT_PRIO)
98#define TASK_USER_PRIO(p) USER_PRIO((p)->static_prio)
99#define MAX_USER_PRIO (USER_PRIO(MAX_PRIO))
100
101/*
Ingo Molnard7876a02008-01-25 21:08:19 +0100102 * Helpers for converting nanosecond timing to jiffy resolution
Linus Torvalds1da177e2005-04-16 15:20:36 -0700103 */
Eric Dumazetd6322fa2007-11-09 22:39:38 +0100104#define NS_TO_JIFFIES(TIME) ((unsigned long)(TIME) / (NSEC_PER_SEC / HZ))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700105
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200106#define NICE_0_LOAD SCHED_LOAD_SCALE
107#define NICE_0_SHIFT SCHED_LOAD_SHIFT
108
Linus Torvalds1da177e2005-04-16 15:20:36 -0700109/*
110 * These are the 'tuning knobs' of the scheduler:
111 *
Dmitry Adamushkoa4ec24b2007-10-15 17:00:13 +0200112 * default timeslice is 100 msecs (used only for SCHED_RR tasks).
Linus Torvalds1da177e2005-04-16 15:20:36 -0700113 * Timeslices get refilled after they expire.
114 */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700115#define DEF_TIMESLICE (100 * HZ / 1000)
Peter Williams2dd73a42006-06-27 02:54:34 -0700116
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200117/*
118 * single value that denotes runtime == period, ie unlimited time.
119 */
120#define RUNTIME_INF ((u64)~0ULL)
121
Ingo Molnare05606d2007-07-09 18:51:59 +0200122static inline int rt_policy(int policy)
123{
Roel Kluin3f33a7c2008-05-13 23:44:11 +0200124 if (unlikely(policy == SCHED_FIFO || policy == SCHED_RR))
Ingo Molnare05606d2007-07-09 18:51:59 +0200125 return 1;
126 return 0;
127}
128
129static inline int task_has_rt_policy(struct task_struct *p)
130{
131 return rt_policy(p->policy);
132}
133
Linus Torvalds1da177e2005-04-16 15:20:36 -0700134/*
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200135 * This is the priority-queue data structure of the RT scheduling class:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700136 */
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200137struct rt_prio_array {
138 DECLARE_BITMAP(bitmap, MAX_RT_PRIO+1); /* include 1 bit for delimiter */
139 struct list_head queue[MAX_RT_PRIO];
140};
Linus Torvalds1da177e2005-04-16 15:20:36 -0700141
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200142struct rt_bandwidth {
Ingo Molnarea736ed2008-03-25 13:51:45 +0100143 /* nests inside the rq lock: */
Thomas Gleixner0986b112009-11-17 15:32:06 +0100144 raw_spinlock_t rt_runtime_lock;
Ingo Molnarea736ed2008-03-25 13:51:45 +0100145 ktime_t rt_period;
146 u64 rt_runtime;
147 struct hrtimer rt_period_timer;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200148};
149
150static struct rt_bandwidth def_rt_bandwidth;
151
152static int do_sched_rt_period_timer(struct rt_bandwidth *rt_b, int overrun);
153
154static enum hrtimer_restart sched_rt_period_timer(struct hrtimer *timer)
155{
156 struct rt_bandwidth *rt_b =
157 container_of(timer, struct rt_bandwidth, rt_period_timer);
158 ktime_t now;
159 int overrun;
160 int idle = 0;
161
162 for (;;) {
163 now = hrtimer_cb_get_time(timer);
164 overrun = hrtimer_forward(timer, now, rt_b->rt_period);
165
166 if (!overrun)
167 break;
168
169 idle = do_sched_rt_period_timer(rt_b, overrun);
170 }
171
172 return idle ? HRTIMER_NORESTART : HRTIMER_RESTART;
173}
174
175static
176void init_rt_bandwidth(struct rt_bandwidth *rt_b, u64 period, u64 runtime)
177{
178 rt_b->rt_period = ns_to_ktime(period);
179 rt_b->rt_runtime = runtime;
180
Thomas Gleixner0986b112009-11-17 15:32:06 +0100181 raw_spin_lock_init(&rt_b->rt_runtime_lock);
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200182
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200183 hrtimer_init(&rt_b->rt_period_timer,
184 CLOCK_MONOTONIC, HRTIMER_MODE_REL);
185 rt_b->rt_period_timer.function = sched_rt_period_timer;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200186}
187
Krzysztof Heltc8bfff62008-09-05 23:46:19 +0200188static inline int rt_bandwidth_enabled(void)
189{
190 return sysctl_sched_rt_runtime >= 0;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200191}
192
193static void start_rt_bandwidth(struct rt_bandwidth *rt_b)
194{
195 ktime_t now;
196
Hiroshi Shimamotocac64d02009-02-25 09:59:26 -0800197 if (!rt_bandwidth_enabled() || rt_b->rt_runtime == RUNTIME_INF)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200198 return;
199
200 if (hrtimer_active(&rt_b->rt_period_timer))
201 return;
202
Thomas Gleixner0986b112009-11-17 15:32:06 +0100203 raw_spin_lock(&rt_b->rt_runtime_lock);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200204 for (;;) {
Peter Zijlstra7f1e2ca2009-03-13 12:21:27 +0100205 unsigned long delta;
206 ktime_t soft, hard;
207
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200208 if (hrtimer_active(&rt_b->rt_period_timer))
209 break;
210
211 now = hrtimer_cb_get_time(&rt_b->rt_period_timer);
212 hrtimer_forward(&rt_b->rt_period_timer, now, rt_b->rt_period);
Peter Zijlstra7f1e2ca2009-03-13 12:21:27 +0100213
214 soft = hrtimer_get_softexpires(&rt_b->rt_period_timer);
215 hard = hrtimer_get_expires(&rt_b->rt_period_timer);
216 delta = ktime_to_ns(ktime_sub(hard, soft));
217 __hrtimer_start_range_ns(&rt_b->rt_period_timer, soft, delta,
Arun R Bharadwaj5c333862009-04-16 12:14:37 +0530218 HRTIMER_MODE_ABS_PINNED, 0);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200219 }
Thomas Gleixner0986b112009-11-17 15:32:06 +0100220 raw_spin_unlock(&rt_b->rt_runtime_lock);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200221}
222
223#ifdef CONFIG_RT_GROUP_SCHED
224static void destroy_rt_bandwidth(struct rt_bandwidth *rt_b)
225{
226 hrtimer_cancel(&rt_b->rt_period_timer);
227}
228#endif
229
Heiko Carstens712555e2008-04-28 11:33:07 +0200230/*
231 * sched_domains_mutex serializes calls to arch_init_sched_domains,
232 * detach_destroy_domains and partition_sched_domains.
233 */
234static DEFINE_MUTEX(sched_domains_mutex);
235
Dhaval Giani7c941432010-01-20 13:26:18 +0100236#ifdef CONFIG_CGROUP_SCHED
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200237
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -0700238#include <linux/cgroup.h>
239
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200240struct cfs_rq;
241
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100242static LIST_HEAD(task_groups);
243
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200244/* task group related information */
Ingo Molnar4cf86d72007-10-15 17:00:14 +0200245struct task_group {
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -0700246 struct cgroup_subsys_state css;
Arun R Bharadwaj6c415b92008-12-01 20:49:05 +0530247
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100248#ifdef CONFIG_FAIR_GROUP_SCHED
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200249 /* schedulable entities of this group on each cpu */
250 struct sched_entity **se;
251 /* runqueue "owned" by this group on each cpu */
252 struct cfs_rq **cfs_rq;
253 unsigned long shares;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100254#endif
255
256#ifdef CONFIG_RT_GROUP_SCHED
257 struct sched_rt_entity **rt_se;
258 struct rt_rq **rt_rq;
259
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200260 struct rt_bandwidth rt_bandwidth;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100261#endif
Srivatsa Vaddagiri6b2d7702008-01-25 21:08:00 +0100262
Srivatsa Vaddagiriae8393e2007-10-29 21:18:11 +0100263 struct rcu_head rcu;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100264 struct list_head list;
Peter Zijlstraf473aa52008-04-19 19:45:00 +0200265
266 struct task_group *parent;
267 struct list_head siblings;
268 struct list_head children;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200269};
270
Peter Zijlstraeff766a2008-04-19 19:45:00 +0200271#define root_task_group init_task_group
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100272
Peter Zijlstra8ed36992008-02-13 15:45:39 +0100273/* task_group_lock serializes add/remove of task groups and also changes to
Srivatsa Vaddagiriec2c5072008-01-25 21:07:59 +0100274 * a task group's cpu shares.
275 */
Peter Zijlstra8ed36992008-02-13 15:45:39 +0100276static DEFINE_SPINLOCK(task_group_lock);
Srivatsa Vaddagiriec2c5072008-01-25 21:07:59 +0100277
Cyrill Gorcunove9036b32009-10-26 22:24:14 +0300278#ifdef CONFIG_FAIR_GROUP_SCHED
279
Peter Zijlstra57310a92009-03-09 13:56:21 +0100280#ifdef CONFIG_SMP
281static int root_task_group_empty(void)
282{
283 return list_empty(&root_task_group.children);
284}
285#endif
286
Srivatsa Vaddagiri93f992c2008-01-25 21:07:59 +0100287# define INIT_TASK_GROUP_LOAD NICE_0_LOAD
Srivatsa Vaddagiri24e377a2007-10-15 17:00:09 +0200288
Miao Xiecb4ad1f2008-04-28 12:54:56 +0800289/*
Lai Jiangshan2e084782008-06-12 16:42:58 +0800290 * A weight of 0 or 1 can cause arithmetics problems.
291 * A weight of a cfs_rq is the sum of weights of which entities
292 * are queued on this cfs_rq, so a weight of a entity should not be
293 * too large, so as the shares value of a task group.
Miao Xiecb4ad1f2008-04-28 12:54:56 +0800294 * (The default weight is 1024 - so there's no practical
295 * limitation from this.)
296 */
Peter Zijlstra18d95a22008-04-19 19:45:00 +0200297#define MIN_SHARES 2
Lai Jiangshan2e084782008-06-12 16:42:58 +0800298#define MAX_SHARES (1UL << 18)
Peter Zijlstra18d95a22008-04-19 19:45:00 +0200299
Srivatsa Vaddagiri93f992c2008-01-25 21:07:59 +0100300static int init_task_group_load = INIT_TASK_GROUP_LOAD;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100301#endif
302
303/* Default task group.
304 * Every task in system belong to this group at bootup.
305 */
Mike Travis434d53b2008-04-04 18:11:04 -0700306struct task_group init_task_group;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200307
308/* return group to which a task belongs */
Ingo Molnar4cf86d72007-10-15 17:00:14 +0200309static inline struct task_group *task_group(struct task_struct *p)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200310{
Ingo Molnar4cf86d72007-10-15 17:00:14 +0200311 struct task_group *tg;
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +0200312
Dhaval Giani7c941432010-01-20 13:26:18 +0100313#ifdef CONFIG_CGROUP_SCHED
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -0700314 tg = container_of(task_subsys_state(p, cpu_cgroup_subsys_id),
315 struct task_group, css);
Srivatsa Vaddagiri24e377a2007-10-15 17:00:09 +0200316#else
Ingo Molnar41a2d6c2007-12-05 15:46:09 +0100317 tg = &init_task_group;
Srivatsa Vaddagiri24e377a2007-10-15 17:00:09 +0200318#endif
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +0200319 return tg;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200320}
321
322/* Change a task's cfs_rq and parent entity if it moves across CPUs/groups */
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100323static inline void set_task_rq(struct task_struct *p, unsigned int cpu)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200324{
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100325#ifdef CONFIG_FAIR_GROUP_SCHED
Dmitry Adamushkoce96b5a2007-11-15 20:57:40 +0100326 p->se.cfs_rq = task_group(p)->cfs_rq[cpu];
327 p->se.parent = task_group(p)->se[cpu];
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100328#endif
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100329
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100330#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100331 p->rt.rt_rq = task_group(p)->rt_rq[cpu];
332 p->rt.parent = task_group(p)->rt_se[cpu];
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100333#endif
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200334}
335
336#else
337
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100338static inline void set_task_rq(struct task_struct *p, unsigned int cpu) { }
Peter Zijlstra83378262008-06-27 13:41:37 +0200339static inline struct task_group *task_group(struct task_struct *p)
340{
341 return NULL;
342}
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200343
Dhaval Giani7c941432010-01-20 13:26:18 +0100344#endif /* CONFIG_CGROUP_SCHED */
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +0200345
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200346/* CFS-related fields in a runqueue */
347struct cfs_rq {
348 struct load_weight load;
349 unsigned long nr_running;
350
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200351 u64 exec_clock;
Ingo Molnare9acbff2007-10-15 17:00:04 +0200352 u64 min_vruntime;
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200353
354 struct rb_root tasks_timeline;
355 struct rb_node *rb_leftmost;
Peter Zijlstra4a55bd52008-04-19 19:45:00 +0200356
357 struct list_head tasks;
358 struct list_head *balance_iterator;
359
360 /*
361 * 'curr' points to currently running entity on this cfs_rq.
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200362 * It is set to NULL otherwise (i.e when none are currently running).
363 */
Peter Zijlstra47932412008-11-04 21:25:09 +0100364 struct sched_entity *curr, *next, *last;
Peter Zijlstraddc97292007-10-15 17:00:10 +0200365
Peter Zijlstra5ac5c4d2008-11-10 10:46:32 +0100366 unsigned int nr_spread_over;
Peter Zijlstraddc97292007-10-15 17:00:10 +0200367
Ingo Molnar62160e3f2007-10-15 17:00:03 +0200368#ifdef CONFIG_FAIR_GROUP_SCHED
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200369 struct rq *rq; /* cpu runqueue to which this cfs_rq is attached */
370
Ingo Molnar41a2d6c2007-12-05 15:46:09 +0100371 /*
372 * leaf cfs_rqs are those that hold tasks (lowest schedulable entity in
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200373 * a hierarchy). Non-leaf lrqs hold other higher schedulable entities
374 * (like users, containers etc.)
375 *
376 * leaf_cfs_rq_list ties together list of leaf cfs_rq's in a cpu. This
377 * list is used during load balance.
378 */
Ingo Molnar41a2d6c2007-12-05 15:46:09 +0100379 struct list_head leaf_cfs_rq_list;
380 struct task_group *tg; /* group that "owns" this runqueue */
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200381
382#ifdef CONFIG_SMP
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200383 /*
Peter Zijlstrac8cba852008-06-27 13:41:23 +0200384 * the part of load.weight contributed by tasks
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200385 */
Peter Zijlstrac8cba852008-06-27 13:41:23 +0200386 unsigned long task_weight;
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200387
Peter Zijlstrac8cba852008-06-27 13:41:23 +0200388 /*
389 * h_load = weight * f(tg)
390 *
391 * Where f(tg) is the recursive weight fraction assigned to
392 * this group.
393 */
394 unsigned long h_load;
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200395
Peter Zijlstrac8cba852008-06-27 13:41:23 +0200396 /*
397 * this cpu's part of tg->shares
398 */
399 unsigned long shares;
Peter Zijlstraf1d239f2008-06-27 13:41:38 +0200400
401 /*
402 * load.weight at the time we set shares
403 */
404 unsigned long rq_weight;
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200405#endif
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200406#endif
407};
408
409/* Real-Time classes' related field in a runqueue: */
410struct rt_rq {
411 struct rt_prio_array active;
Steven Rostedt63489e42008-01-25 21:08:03 +0100412 unsigned long rt_nr_running;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100413#if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
Gregory Haskinse864c492008-12-29 09:39:49 -0500414 struct {
415 int curr; /* highest queued rt task prio */
Gregory Haskins398a1532009-01-14 09:10:04 -0500416#ifdef CONFIG_SMP
Gregory Haskinse864c492008-12-29 09:39:49 -0500417 int next; /* next highest */
Gregory Haskins398a1532009-01-14 09:10:04 -0500418#endif
Gregory Haskinse864c492008-12-29 09:39:49 -0500419 } highest_prio;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100420#endif
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100421#ifdef CONFIG_SMP
Gregory Haskins73fe6aae2008-01-25 21:08:07 +0100422 unsigned long rt_nr_migratory;
Peter Zijlstraa1ba4d82009-04-01 18:40:15 +0200423 unsigned long rt_nr_total;
Gregory Haskinsa22d7fc12008-01-25 21:08:12 +0100424 int overloaded;
Gregory Haskins917b6272008-12-29 09:39:53 -0500425 struct plist_head pushable_tasks;
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100426#endif
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100427 int rt_throttled;
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100428 u64 rt_time;
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200429 u64 rt_runtime;
Ingo Molnarea736ed2008-03-25 13:51:45 +0100430 /* Nests inside the rq lock: */
Thomas Gleixner0986b112009-11-17 15:32:06 +0100431 raw_spinlock_t rt_runtime_lock;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100432
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100433#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100434 unsigned long rt_nr_boosted;
435
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100436 struct rq *rq;
437 struct list_head leaf_rt_rq_list;
438 struct task_group *tg;
439 struct sched_rt_entity *rt_se;
440#endif
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200441};
442
Gregory Haskins57d885f2008-01-25 21:08:18 +0100443#ifdef CONFIG_SMP
444
445/*
446 * We add the notion of a root-domain which will be used to define per-domain
Ingo Molnar0eab9142008-01-25 21:08:19 +0100447 * variables. Each exclusive cpuset essentially defines an island domain by
448 * fully partitioning the member cpus from any other cpuset. Whenever a new
Gregory Haskins57d885f2008-01-25 21:08:18 +0100449 * exclusive cpuset is created, we also create and attach a new root-domain
450 * object.
451 *
Gregory Haskins57d885f2008-01-25 21:08:18 +0100452 */
453struct root_domain {
454 atomic_t refcount;
Rusty Russellc6c49272008-11-25 02:35:05 +1030455 cpumask_var_t span;
456 cpumask_var_t online;
Gregory Haskins637f5082008-01-25 21:08:18 +0100457
Ingo Molnar0eab9142008-01-25 21:08:19 +0100458 /*
Gregory Haskins637f5082008-01-25 21:08:18 +0100459 * The "RT overload" flag: it gets set if a CPU has more than
460 * one runnable RT task.
461 */
Rusty Russellc6c49272008-11-25 02:35:05 +1030462 cpumask_var_t rto_mask;
Ingo Molnar0eab9142008-01-25 21:08:19 +0100463 atomic_t rto_count;
Gregory Haskins6e0534f2008-05-12 21:21:01 +0200464#ifdef CONFIG_SMP
465 struct cpupri cpupri;
466#endif
Gregory Haskins57d885f2008-01-25 21:08:18 +0100467};
468
Gregory Haskinsdc938522008-01-25 21:08:26 +0100469/*
470 * By default the system creates a single root-domain with all cpus as
471 * members (mimicking the global state we have today).
472 */
Gregory Haskins57d885f2008-01-25 21:08:18 +0100473static struct root_domain def_root_domain;
474
475#endif
476
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200477/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700478 * This is the main, per-CPU runqueue data structure.
479 *
480 * Locking rule: those places that want to lock multiple runqueues
481 * (such as the load balancing or the thread migration code), lock
482 * acquire operations must be ordered by ascending &runqueue.
483 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700484struct rq {
Ingo Molnard8016492007-10-18 21:32:55 +0200485 /* runqueue lock: */
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100486 raw_spinlock_t lock;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700487
488 /*
489 * nr_running and cpu_load should be in the same cacheline because
490 * remote CPUs use both these fields when doing load calculation.
491 */
492 unsigned long nr_running;
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200493 #define CPU_LOAD_IDX_MAX 5
494 unsigned long cpu_load[CPU_LOAD_IDX_MAX];
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -0700495#ifdef CONFIG_NO_HZ
496 unsigned char in_nohz_recently;
497#endif
Ingo Molnard8016492007-10-18 21:32:55 +0200498 /* capture load from *all* tasks on this cpu: */
499 struct load_weight load;
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200500 unsigned long nr_load_updates;
501 u64 nr_switches;
502
503 struct cfs_rq cfs;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100504 struct rt_rq rt;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100505
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200506#ifdef CONFIG_FAIR_GROUP_SCHED
Ingo Molnard8016492007-10-18 21:32:55 +0200507 /* list of leaf cfs_rq on this cpu: */
508 struct list_head leaf_cfs_rq_list;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100509#endif
510#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100511 struct list_head leaf_rt_rq_list;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700512#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700513
514 /*
515 * This is part of a global counter where only the total sum
516 * over all CPUs matters. A task can increase this counter on
517 * one CPU and if it got migrated afterwards it may decrease
518 * it on another CPU. Always updated under the runqueue lock:
519 */
520 unsigned long nr_uninterruptible;
521
Ingo Molnar36c8b582006-07-03 00:25:41 -0700522 struct task_struct *curr, *idle;
Christoph Lameterc9819f42006-12-10 02:20:25 -0800523 unsigned long next_balance;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700524 struct mm_struct *prev_mm;
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200525
Peter Zijlstra3e51f332008-05-03 18:29:28 +0200526 u64 clock;
Ingo Molnar6aa645e2007-07-09 18:51:58 +0200527
Linus Torvalds1da177e2005-04-16 15:20:36 -0700528 atomic_t nr_iowait;
529
530#ifdef CONFIG_SMP
Ingo Molnar0eab9142008-01-25 21:08:19 +0100531 struct root_domain *rd;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700532 struct sched_domain *sd;
533
Henrik Austada0a522c2009-02-13 20:35:45 +0100534 unsigned char idle_at_tick;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700535 /* For active balancing */
Gregory Haskins3f029d32009-07-29 11:08:47 -0400536 int post_schedule;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700537 int active_balance;
538 int push_cpu;
Ingo Molnard8016492007-10-18 21:32:55 +0200539 /* cpu of this runqueue: */
540 int cpu;
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -0400541 int online;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700542
Peter Zijlstraa8a51d52008-06-27 13:41:26 +0200543 unsigned long avg_load_per_task;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700544
Ingo Molnar36c8b582006-07-03 00:25:41 -0700545 struct task_struct *migration_thread;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700546 struct list_head migration_queue;
Peter Zijlstrae9e92502009-09-01 10:34:37 +0200547
548 u64 rt_avg;
549 u64 age_stamp;
Mike Galbraith1b9508f2009-11-04 17:53:50 +0100550 u64 idle_stamp;
551 u64 avg_idle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700552#endif
553
Thomas Gleixnerdce48a82009-04-11 10:43:41 +0200554 /* calc_load related fields */
555 unsigned long calc_load_update;
556 long calc_load_active;
557
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100558#ifdef CONFIG_SCHED_HRTICK
Peter Zijlstra31656512008-07-18 18:01:23 +0200559#ifdef CONFIG_SMP
560 int hrtick_csd_pending;
561 struct call_single_data hrtick_csd;
562#endif
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100563 struct hrtimer hrtick_timer;
564#endif
565
Linus Torvalds1da177e2005-04-16 15:20:36 -0700566#ifdef CONFIG_SCHEDSTATS
567 /* latency stats */
568 struct sched_info rq_sched_info;
Ken Chen9c2c4802008-12-16 23:41:22 -0800569 unsigned long long rq_cpu_time;
570 /* could above be rq->cfs_rq.exec_clock + rq->rt_rq.rt_runtime ? */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700571
572 /* sys_sched_yield() stats */
Ken Chen480b9432007-10-18 21:32:56 +0200573 unsigned int yld_count;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700574
575 /* schedule() stats */
Ken Chen480b9432007-10-18 21:32:56 +0200576 unsigned int sched_switch;
577 unsigned int sched_count;
578 unsigned int sched_goidle;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700579
580 /* try_to_wake_up() stats */
Ken Chen480b9432007-10-18 21:32:56 +0200581 unsigned int ttwu_count;
582 unsigned int ttwu_local;
Ingo Molnarb8efb562007-10-15 17:00:10 +0200583
584 /* BKL stats */
Ken Chen480b9432007-10-18 21:32:56 +0200585 unsigned int bkl_count;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700586#endif
587};
588
Fenghua Yuf34e3b62007-07-19 01:48:13 -0700589static DEFINE_PER_CPU_SHARED_ALIGNED(struct rq, runqueues);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700590
Peter Zijlstra7d478722009-09-14 19:55:44 +0200591static inline
592void check_preempt_curr(struct rq *rq, struct task_struct *p, int flags)
Ingo Molnardd41f592007-07-09 18:51:59 +0200593{
Peter Zijlstra7d478722009-09-14 19:55:44 +0200594 rq->curr->sched_class->check_preempt_curr(rq, p, flags);
Ingo Molnardd41f592007-07-09 18:51:59 +0200595}
596
Christoph Lameter0a2966b2006-09-25 23:30:51 -0700597static inline int cpu_of(struct rq *rq)
598{
599#ifdef CONFIG_SMP
600 return rq->cpu;
601#else
602 return 0;
603#endif
604}
605
Ingo Molnar20d315d2007-07-09 18:51:58 +0200606/*
Nick Piggin674311d2005-06-25 14:57:27 -0700607 * The domain tree (rq->sd) is protected by RCU's quiescent state transition.
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -0700608 * See detach_destroy_domains: synchronize_sched for details.
Nick Piggin674311d2005-06-25 14:57:27 -0700609 *
610 * The domain tree of any CPU may only be accessed from within
611 * preempt-disabled sections.
612 */
Ingo Molnar48f24c42006-07-03 00:25:40 -0700613#define for_each_domain(cpu, __sd) \
614 for (__sd = rcu_dereference(cpu_rq(cpu)->sd); __sd; __sd = __sd->parent)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700615
616#define cpu_rq(cpu) (&per_cpu(runqueues, (cpu)))
617#define this_rq() (&__get_cpu_var(runqueues))
618#define task_rq(p) cpu_rq(task_cpu(p))
619#define cpu_curr(cpu) (cpu_rq(cpu)->curr)
Hitoshi Mitake54d35f22009-06-29 14:44:57 +0900620#define raw_rq() (&__raw_get_cpu_var(runqueues))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700621
Ingo Molnaraa9c4c02008-12-17 14:10:57 +0100622inline void update_rq_clock(struct rq *rq)
Peter Zijlstra3e51f332008-05-03 18:29:28 +0200623{
624 rq->clock = sched_clock_cpu(cpu_of(rq));
625}
626
Ingo Molnare436d802007-07-19 21:28:35 +0200627/*
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200628 * Tunables that become constants when CONFIG_SCHED_DEBUG is off:
629 */
630#ifdef CONFIG_SCHED_DEBUG
631# define const_debug __read_mostly
632#else
633# define const_debug static const
634#endif
635
Ingo Molnar017730c2008-05-12 21:20:52 +0200636/**
637 * runqueue_is_locked
Randy Dunlape17b38b2009-10-11 19:12:00 -0700638 * @cpu: the processor in question.
Ingo Molnar017730c2008-05-12 21:20:52 +0200639 *
640 * Returns true if the current cpu runqueue is locked.
641 * This interface allows printk to be called with the runqueue lock
642 * held and know whether or not it is OK to wake up the klogd.
643 */
Andrew Morton89f19f02009-09-19 11:55:44 -0700644int runqueue_is_locked(int cpu)
Ingo Molnar017730c2008-05-12 21:20:52 +0200645{
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100646 return raw_spin_is_locked(&cpu_rq(cpu)->lock);
Ingo Molnar017730c2008-05-12 21:20:52 +0200647}
648
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200649/*
650 * Debugging: various feature bits
651 */
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200652
653#define SCHED_FEAT(name, enabled) \
654 __SCHED_FEAT_##name ,
655
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200656enum {
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200657#include "sched_features.h"
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200658};
659
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200660#undef SCHED_FEAT
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200661
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200662#define SCHED_FEAT(name, enabled) \
663 (1UL << __SCHED_FEAT_##name) * enabled |
664
665const_debug unsigned int sysctl_sched_features =
666#include "sched_features.h"
667 0;
668
669#undef SCHED_FEAT
670
671#ifdef CONFIG_SCHED_DEBUG
672#define SCHED_FEAT(name, enabled) \
673 #name ,
674
Harvey Harrison983ed7a2008-04-24 18:17:55 -0700675static __read_mostly char *sched_feat_names[] = {
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200676#include "sched_features.h"
677 NULL
678};
679
680#undef SCHED_FEAT
681
Li Zefan34f3a812008-10-30 15:23:32 +0800682static int sched_feat_show(struct seq_file *m, void *v)
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200683{
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200684 int i;
685
686 for (i = 0; sched_feat_names[i]; i++) {
Li Zefan34f3a812008-10-30 15:23:32 +0800687 if (!(sysctl_sched_features & (1UL << i)))
688 seq_puts(m, "NO_");
689 seq_printf(m, "%s ", sched_feat_names[i]);
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200690 }
Li Zefan34f3a812008-10-30 15:23:32 +0800691 seq_puts(m, "\n");
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200692
Li Zefan34f3a812008-10-30 15:23:32 +0800693 return 0;
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200694}
695
696static ssize_t
697sched_feat_write(struct file *filp, const char __user *ubuf,
698 size_t cnt, loff_t *ppos)
699{
700 char buf[64];
701 char *cmp = buf;
702 int neg = 0;
703 int i;
704
705 if (cnt > 63)
706 cnt = 63;
707
708 if (copy_from_user(&buf, ubuf, cnt))
709 return -EFAULT;
710
711 buf[cnt] = 0;
712
Ingo Molnarc24b7c52008-04-18 10:55:34 +0200713 if (strncmp(buf, "NO_", 3) == 0) {
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200714 neg = 1;
715 cmp += 3;
716 }
717
718 for (i = 0; sched_feat_names[i]; i++) {
719 int len = strlen(sched_feat_names[i]);
720
721 if (strncmp(cmp, sched_feat_names[i], len) == 0) {
722 if (neg)
723 sysctl_sched_features &= ~(1UL << i);
724 else
725 sysctl_sched_features |= (1UL << i);
726 break;
727 }
728 }
729
730 if (!sched_feat_names[i])
731 return -EINVAL;
732
Jan Blunck42994722009-11-20 17:40:37 +0100733 *ppos += cnt;
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200734
735 return cnt;
736}
737
Li Zefan34f3a812008-10-30 15:23:32 +0800738static int sched_feat_open(struct inode *inode, struct file *filp)
739{
740 return single_open(filp, sched_feat_show, NULL);
741}
742
Alexey Dobriyan828c0952009-10-01 15:43:56 -0700743static const struct file_operations sched_feat_fops = {
Li Zefan34f3a812008-10-30 15:23:32 +0800744 .open = sched_feat_open,
745 .write = sched_feat_write,
746 .read = seq_read,
747 .llseek = seq_lseek,
748 .release = single_release,
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200749};
750
751static __init int sched_init_debug(void)
752{
Peter Zijlstraf00b45c2008-04-19 19:45:00 +0200753 debugfs_create_file("sched_features", 0644, NULL, NULL,
754 &sched_feat_fops);
755
756 return 0;
757}
758late_initcall(sched_init_debug);
759
760#endif
761
762#define sched_feat(x) (sysctl_sched_features & (1UL << __SCHED_FEAT_##x))
Ingo Molnarbf5c91b2007-10-15 17:00:04 +0200763
764/*
Peter Zijlstrab82d9fd2007-11-09 22:39:39 +0100765 * Number of tasks to iterate in a single balance run.
766 * Limited because this is done with IRQs disabled.
767 */
768const_debug unsigned int sysctl_sched_nr_migrate = 32;
769
770/*
Peter Zijlstra2398f2c2008-06-27 13:41:35 +0200771 * ratelimit for updating the group shares.
Peter Zijlstra55cd5342008-08-04 08:54:26 +0200772 * default: 0.25ms
Peter Zijlstra2398f2c2008-06-27 13:41:35 +0200773 */
Peter Zijlstra55cd5342008-08-04 08:54:26 +0200774unsigned int sysctl_sched_shares_ratelimit = 250000;
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +0100775unsigned int normalized_sysctl_sched_shares_ratelimit = 250000;
Peter Zijlstra2398f2c2008-06-27 13:41:35 +0200776
777/*
Peter Zijlstraffda12a2008-10-17 19:27:02 +0200778 * Inject some fuzzyness into changing the per-cpu group shares
779 * this avoids remote rq-locks at the expense of fairness.
780 * default: 4
781 */
782unsigned int sysctl_sched_shares_thresh = 4;
783
784/*
Peter Zijlstrae9e92502009-09-01 10:34:37 +0200785 * period over which we average the RT time consumption, measured
786 * in ms.
787 *
788 * default: 1s
789 */
790const_debug unsigned int sysctl_sched_time_avg = MSEC_PER_SEC;
791
792/*
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100793 * period over which we measure -rt task cpu usage in us.
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100794 * default: 1s
795 */
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100796unsigned int sysctl_sched_rt_period = 1000000;
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100797
Ingo Molnar6892b752008-02-13 14:02:36 +0100798static __read_mostly int scheduler_running;
799
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100800/*
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100801 * part of the period that we allow rt tasks to run in us.
802 * default: 0.95s
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100803 */
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100804int sysctl_sched_rt_runtime = 950000;
805
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200806static inline u64 global_rt_period(void)
807{
808 return (u64)sysctl_sched_rt_period * NSEC_PER_USEC;
809}
810
811static inline u64 global_rt_runtime(void)
812{
roel kluine26873b2008-07-22 16:51:15 -0400813 if (sysctl_sched_rt_runtime < 0)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200814 return RUNTIME_INF;
815
816 return (u64)sysctl_sched_rt_runtime * NSEC_PER_USEC;
817}
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100818
Linus Torvalds1da177e2005-04-16 15:20:36 -0700819#ifndef prepare_arch_switch
Nick Piggin4866cde2005-06-25 14:57:23 -0700820# define prepare_arch_switch(next) do { } while (0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700821#endif
Nick Piggin4866cde2005-06-25 14:57:23 -0700822#ifndef finish_arch_switch
823# define finish_arch_switch(prev) do { } while (0)
824#endif
825
Dmitry Adamushko051a1d12007-12-18 15:21:13 +0100826static inline int task_current(struct rq *rq, struct task_struct *p)
827{
828 return rq->curr == p;
829}
830
Nick Piggin4866cde2005-06-25 14:57:23 -0700831#ifndef __ARCH_WANT_UNLOCKED_CTXSW
Ingo Molnar70b97a72006-07-03 00:25:42 -0700832static inline int task_running(struct rq *rq, struct task_struct *p)
Nick Piggin4866cde2005-06-25 14:57:23 -0700833{
Dmitry Adamushko051a1d12007-12-18 15:21:13 +0100834 return task_current(rq, p);
Nick Piggin4866cde2005-06-25 14:57:23 -0700835}
836
Ingo Molnar70b97a72006-07-03 00:25:42 -0700837static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -0700838{
839}
840
Ingo Molnar70b97a72006-07-03 00:25:42 -0700841static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
Nick Piggin4866cde2005-06-25 14:57:23 -0700842{
Ingo Molnarda04c032005-09-13 11:17:59 +0200843#ifdef CONFIG_DEBUG_SPINLOCK
844 /* this is a valid case when another task releases the spinlock */
845 rq->lock.owner = current;
846#endif
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700847 /*
848 * If we are tracking spinlock dependencies then we have to
849 * fix up the runqueue lock - which gets 'carried over' from
850 * prev into current:
851 */
852 spin_acquire(&rq->lock.dep_map, 0, 0, _THIS_IP_);
853
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100854 raw_spin_unlock_irq(&rq->lock);
Nick Piggin4866cde2005-06-25 14:57:23 -0700855}
856
857#else /* __ARCH_WANT_UNLOCKED_CTXSW */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700858static inline int task_running(struct rq *rq, struct task_struct *p)
Nick Piggin4866cde2005-06-25 14:57:23 -0700859{
860#ifdef CONFIG_SMP
861 return p->oncpu;
862#else
Dmitry Adamushko051a1d12007-12-18 15:21:13 +0100863 return task_current(rq, p);
Nick Piggin4866cde2005-06-25 14:57:23 -0700864#endif
865}
866
Ingo Molnar70b97a72006-07-03 00:25:42 -0700867static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -0700868{
869#ifdef CONFIG_SMP
870 /*
871 * We can optimise this out completely for !SMP, because the
872 * SMP rebalancing from interrupt is the only thing that cares
873 * here.
874 */
875 next->oncpu = 1;
876#endif
877#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100878 raw_spin_unlock_irq(&rq->lock);
Nick Piggin4866cde2005-06-25 14:57:23 -0700879#else
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100880 raw_spin_unlock(&rq->lock);
Nick Piggin4866cde2005-06-25 14:57:23 -0700881#endif
882}
883
Ingo Molnar70b97a72006-07-03 00:25:42 -0700884static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
Nick Piggin4866cde2005-06-25 14:57:23 -0700885{
886#ifdef CONFIG_SMP
887 /*
888 * After ->oncpu is cleared, the task can be moved to a different CPU.
889 * We must ensure this doesn't happen until the switch is completely
890 * finished.
891 */
892 smp_wmb();
893 prev->oncpu = 0;
894#endif
895#ifndef __ARCH_WANT_INTERRUPTS_ON_CTXSW
896 local_irq_enable();
897#endif
898}
899#endif /* __ARCH_WANT_UNLOCKED_CTXSW */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700900
901/*
Ingo Molnarb29739f2006-06-27 02:54:51 -0700902 * __task_rq_lock - lock the runqueue a given task resides on.
903 * Must be called interrupts disabled.
904 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700905static inline struct rq *__task_rq_lock(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700906 __acquires(rq->lock)
907{
Andi Kleen3a5c3592007-10-15 17:00:14 +0200908 for (;;) {
909 struct rq *rq = task_rq(p);
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100910 raw_spin_lock(&rq->lock);
Andi Kleen3a5c3592007-10-15 17:00:14 +0200911 if (likely(rq == task_rq(p)))
912 return rq;
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100913 raw_spin_unlock(&rq->lock);
Ingo Molnarb29739f2006-06-27 02:54:51 -0700914 }
Ingo Molnarb29739f2006-06-27 02:54:51 -0700915}
916
917/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700918 * task_rq_lock - lock the runqueue a given task resides on and disable
Ingo Molnar41a2d6c2007-12-05 15:46:09 +0100919 * interrupts. Note the ordering: we can safely lookup the task_rq without
Linus Torvalds1da177e2005-04-16 15:20:36 -0700920 * explicitly disabling preemption.
921 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700922static struct rq *task_rq_lock(struct task_struct *p, unsigned long *flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700923 __acquires(rq->lock)
924{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700925 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700926
Andi Kleen3a5c3592007-10-15 17:00:14 +0200927 for (;;) {
928 local_irq_save(*flags);
929 rq = task_rq(p);
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100930 raw_spin_lock(&rq->lock);
Andi Kleen3a5c3592007-10-15 17:00:14 +0200931 if (likely(rq == task_rq(p)))
932 return rq;
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100933 raw_spin_unlock_irqrestore(&rq->lock, *flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700934 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700935}
936
Oleg Nesterovad474ca2008-11-10 15:39:30 +0100937void task_rq_unlock_wait(struct task_struct *p)
938{
939 struct rq *rq = task_rq(p);
940
941 smp_mb(); /* spin-unlock-wait is not a full memory barrier */
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100942 raw_spin_unlock_wait(&rq->lock);
Oleg Nesterovad474ca2008-11-10 15:39:30 +0100943}
944
Alexey Dobriyana9957442007-10-15 17:00:13 +0200945static void __task_rq_unlock(struct rq *rq)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700946 __releases(rq->lock)
947{
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100948 raw_spin_unlock(&rq->lock);
Ingo Molnarb29739f2006-06-27 02:54:51 -0700949}
950
Ingo Molnar70b97a72006-07-03 00:25:42 -0700951static inline void task_rq_unlock(struct rq *rq, unsigned long *flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700952 __releases(rq->lock)
953{
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100954 raw_spin_unlock_irqrestore(&rq->lock, *flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700955}
956
Linus Torvalds1da177e2005-04-16 15:20:36 -0700957/*
Robert P. J. Daycc2a73b2006-12-10 02:20:00 -0800958 * this_rq_lock - lock this runqueue and disable interrupts.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700959 */
Alexey Dobriyana9957442007-10-15 17:00:13 +0200960static struct rq *this_rq_lock(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700961 __acquires(rq->lock)
962{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700963 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700964
965 local_irq_disable();
966 rq = this_rq();
Thomas Gleixner05fa7852009-11-17 14:28:38 +0100967 raw_spin_lock(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700968
969 return rq;
970}
971
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100972#ifdef CONFIG_SCHED_HRTICK
973/*
974 * Use HR-timers to deliver accurate preemption points.
975 *
976 * Its all a bit involved since we cannot program an hrt while holding the
977 * rq->lock. So what we do is store a state in in rq->hrtick_* and ask for a
978 * reschedule event.
979 *
980 * When we get rescheduled we reprogram the hrtick_timer outside of the
981 * rq->lock.
982 */
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100983
984/*
985 * Use hrtick when:
986 * - enabled by features
987 * - hrtimer is actually high res
988 */
989static inline int hrtick_enabled(struct rq *rq)
990{
991 if (!sched_feat(HRTICK))
992 return 0;
Ingo Molnarba420592008-07-20 11:02:06 +0200993 if (!cpu_active(cpu_of(rq)))
Peter Zijlstrab328ca12008-04-29 10:02:46 +0200994 return 0;
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100995 return hrtimer_is_hres_active(&rq->hrtick_timer);
996}
997
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +0100998static void hrtick_clear(struct rq *rq)
999{
1000 if (hrtimer_active(&rq->hrtick_timer))
1001 hrtimer_cancel(&rq->hrtick_timer);
1002}
1003
1004/*
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001005 * High-resolution timer tick.
1006 * Runs from hardirq context with interrupts disabled.
1007 */
1008static enum hrtimer_restart hrtick(struct hrtimer *timer)
1009{
1010 struct rq *rq = container_of(timer, struct rq, hrtick_timer);
1011
1012 WARN_ON_ONCE(cpu_of(rq) != smp_processor_id());
1013
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001014 raw_spin_lock(&rq->lock);
Peter Zijlstra3e51f332008-05-03 18:29:28 +02001015 update_rq_clock(rq);
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001016 rq->curr->sched_class->task_tick(rq, rq->curr, 1);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001017 raw_spin_unlock(&rq->lock);
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001018
1019 return HRTIMER_NORESTART;
1020}
1021
Rabin Vincent95e904c2008-05-11 05:55:33 +05301022#ifdef CONFIG_SMP
Peter Zijlstra31656512008-07-18 18:01:23 +02001023/*
1024 * called from hardirq (IPI) context
1025 */
1026static void __hrtick_start(void *arg)
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001027{
Peter Zijlstra31656512008-07-18 18:01:23 +02001028 struct rq *rq = arg;
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001029
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001030 raw_spin_lock(&rq->lock);
Peter Zijlstra31656512008-07-18 18:01:23 +02001031 hrtimer_restart(&rq->hrtick_timer);
1032 rq->hrtick_csd_pending = 0;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001033 raw_spin_unlock(&rq->lock);
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001034}
1035
Peter Zijlstra31656512008-07-18 18:01:23 +02001036/*
1037 * Called to set the hrtick timer state.
1038 *
1039 * called with rq->lock held and irqs disabled
1040 */
1041static void hrtick_start(struct rq *rq, u64 delay)
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001042{
Peter Zijlstra31656512008-07-18 18:01:23 +02001043 struct hrtimer *timer = &rq->hrtick_timer;
1044 ktime_t time = ktime_add_ns(timer->base->get_time(), delay);
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001045
Arjan van de Vencc584b22008-09-01 15:02:30 -07001046 hrtimer_set_expires(timer, time);
Peter Zijlstra31656512008-07-18 18:01:23 +02001047
1048 if (rq == this_rq()) {
1049 hrtimer_restart(timer);
1050 } else if (!rq->hrtick_csd_pending) {
Peter Zijlstra6e275632009-02-25 13:59:48 +01001051 __smp_call_function_single(cpu_of(rq), &rq->hrtick_csd, 0);
Peter Zijlstra31656512008-07-18 18:01:23 +02001052 rq->hrtick_csd_pending = 1;
1053 }
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001054}
1055
1056static int
1057hotplug_hrtick(struct notifier_block *nfb, unsigned long action, void *hcpu)
1058{
1059 int cpu = (int)(long)hcpu;
1060
1061 switch (action) {
1062 case CPU_UP_CANCELED:
1063 case CPU_UP_CANCELED_FROZEN:
1064 case CPU_DOWN_PREPARE:
1065 case CPU_DOWN_PREPARE_FROZEN:
1066 case CPU_DEAD:
1067 case CPU_DEAD_FROZEN:
Peter Zijlstra31656512008-07-18 18:01:23 +02001068 hrtick_clear(cpu_rq(cpu));
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001069 return NOTIFY_OK;
1070 }
1071
1072 return NOTIFY_DONE;
1073}
1074
Rakib Mullickfa748202008-09-22 14:55:45 -07001075static __init void init_hrtick(void)
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001076{
1077 hotcpu_notifier(hotplug_hrtick, 0);
1078}
Peter Zijlstra31656512008-07-18 18:01:23 +02001079#else
1080/*
1081 * Called to set the hrtick timer state.
1082 *
1083 * called with rq->lock held and irqs disabled
1084 */
1085static void hrtick_start(struct rq *rq, u64 delay)
1086{
Peter Zijlstra7f1e2ca2009-03-13 12:21:27 +01001087 __hrtimer_start_range_ns(&rq->hrtick_timer, ns_to_ktime(delay), 0,
Arun R Bharadwaj5c333862009-04-16 12:14:37 +05301088 HRTIMER_MODE_REL_PINNED, 0);
Peter Zijlstra31656512008-07-18 18:01:23 +02001089}
1090
Andrew Morton006c75f2008-09-22 14:55:46 -07001091static inline void init_hrtick(void)
Peter Zijlstra31656512008-07-18 18:01:23 +02001092{
1093}
Rabin Vincent95e904c2008-05-11 05:55:33 +05301094#endif /* CONFIG_SMP */
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001095
1096static void init_rq_hrtick(struct rq *rq)
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001097{
Peter Zijlstra31656512008-07-18 18:01:23 +02001098#ifdef CONFIG_SMP
1099 rq->hrtick_csd_pending = 0;
1100
1101 rq->hrtick_csd.flags = 0;
1102 rq->hrtick_csd.func = __hrtick_start;
1103 rq->hrtick_csd.info = rq;
1104#endif
1105
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001106 hrtimer_init(&rq->hrtick_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
1107 rq->hrtick_timer.function = hrtick;
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001108}
Andrew Morton006c75f2008-09-22 14:55:46 -07001109#else /* CONFIG_SCHED_HRTICK */
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001110static inline void hrtick_clear(struct rq *rq)
1111{
1112}
1113
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001114static inline void init_rq_hrtick(struct rq *rq)
1115{
1116}
1117
Peter Zijlstrab328ca12008-04-29 10:02:46 +02001118static inline void init_hrtick(void)
1119{
1120}
Andrew Morton006c75f2008-09-22 14:55:46 -07001121#endif /* CONFIG_SCHED_HRTICK */
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001122
Ingo Molnar1b9f19c2007-07-09 18:51:59 +02001123/*
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001124 * resched_task - mark a task 'to be rescheduled now'.
1125 *
1126 * On UP this means the setting of the need_resched flag, on SMP it
1127 * might also involve a cross-CPU call to trigger the scheduler on
1128 * the target CPU.
1129 */
1130#ifdef CONFIG_SMP
1131
1132#ifndef tsk_is_polling
1133#define tsk_is_polling(t) test_tsk_thread_flag(t, TIF_POLLING_NRFLAG)
1134#endif
1135
Peter Zijlstra31656512008-07-18 18:01:23 +02001136static void resched_task(struct task_struct *p)
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001137{
1138 int cpu;
1139
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001140 assert_raw_spin_locked(&task_rq(p)->lock);
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001141
Lai Jiangshan5ed0cec2009-03-06 19:40:20 +08001142 if (test_tsk_need_resched(p))
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001143 return;
1144
Lai Jiangshan5ed0cec2009-03-06 19:40:20 +08001145 set_tsk_need_resched(p);
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001146
1147 cpu = task_cpu(p);
1148 if (cpu == smp_processor_id())
1149 return;
1150
1151 /* NEED_RESCHED must be visible before we test polling */
1152 smp_mb();
1153 if (!tsk_is_polling(p))
1154 smp_send_reschedule(cpu);
1155}
1156
1157static void resched_cpu(int cpu)
1158{
1159 struct rq *rq = cpu_rq(cpu);
1160 unsigned long flags;
1161
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001162 if (!raw_spin_trylock_irqsave(&rq->lock, flags))
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001163 return;
1164 resched_task(cpu_curr(cpu));
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001165 raw_spin_unlock_irqrestore(&rq->lock, flags);
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001166}
Thomas Gleixner06d83082008-03-22 09:20:24 +01001167
1168#ifdef CONFIG_NO_HZ
1169/*
1170 * When add_timer_on() enqueues a timer into the timer wheel of an
1171 * idle CPU then this timer might expire before the next timer event
1172 * which is scheduled to wake up that CPU. In case of a completely
1173 * idle system the next event might even be infinite time into the
1174 * future. wake_up_idle_cpu() ensures that the CPU is woken up and
1175 * leaves the inner idle loop so the newly added timer is taken into
1176 * account when the CPU goes back to idle and evaluates the timer
1177 * wheel for the next timer event.
1178 */
1179void wake_up_idle_cpu(int cpu)
1180{
1181 struct rq *rq = cpu_rq(cpu);
1182
1183 if (cpu == smp_processor_id())
1184 return;
1185
1186 /*
1187 * This is safe, as this function is called with the timer
1188 * wheel base lock of (cpu) held. When the CPU is on the way
1189 * to idle and has not yet set rq->curr to idle then it will
1190 * be serialized on the timer wheel base lock and take the new
1191 * timer into account automatically.
1192 */
1193 if (rq->curr != rq->idle)
1194 return;
1195
1196 /*
1197 * We can set TIF_RESCHED on the idle task of the other CPU
1198 * lockless. The worst case is that the other CPU runs the
1199 * idle task through an additional NOOP schedule()
1200 */
Lai Jiangshan5ed0cec2009-03-06 19:40:20 +08001201 set_tsk_need_resched(rq->idle);
Thomas Gleixner06d83082008-03-22 09:20:24 +01001202
1203 /* NEED_RESCHED must be visible before we test polling */
1204 smp_mb();
1205 if (!tsk_is_polling(rq->idle))
1206 smp_send_reschedule(cpu);
1207}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02001208#endif /* CONFIG_NO_HZ */
Thomas Gleixner06d83082008-03-22 09:20:24 +01001209
Peter Zijlstrae9e92502009-09-01 10:34:37 +02001210static u64 sched_avg_period(void)
1211{
1212 return (u64)sysctl_sched_time_avg * NSEC_PER_MSEC / 2;
1213}
1214
1215static void sched_avg_update(struct rq *rq)
1216{
1217 s64 period = sched_avg_period();
1218
1219 while ((s64)(rq->clock - rq->age_stamp) > period) {
1220 rq->age_stamp += period;
1221 rq->rt_avg /= 2;
1222 }
1223}
1224
1225static void sched_rt_avg_update(struct rq *rq, u64 rt_delta)
1226{
1227 rq->rt_avg += rt_delta;
1228 sched_avg_update(rq);
1229}
1230
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02001231#else /* !CONFIG_SMP */
Peter Zijlstra31656512008-07-18 18:01:23 +02001232static void resched_task(struct task_struct *p)
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001233{
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001234 assert_raw_spin_locked(&task_rq(p)->lock);
Peter Zijlstra31656512008-07-18 18:01:23 +02001235 set_tsk_need_resched(p);
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001236}
Peter Zijlstrae9e92502009-09-01 10:34:37 +02001237
1238static void sched_rt_avg_update(struct rq *rq, u64 rt_delta)
1239{
1240}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02001241#endif /* CONFIG_SMP */
Ingo Molnarc24d20d2007-07-09 18:51:59 +02001242
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001243#if BITS_PER_LONG == 32
1244# define WMULT_CONST (~0UL)
1245#else
1246# define WMULT_CONST (1UL << 32)
1247#endif
1248
1249#define WMULT_SHIFT 32
1250
Ingo Molnar194081e2007-08-09 11:16:51 +02001251/*
1252 * Shift right and round:
1253 */
Ingo Molnarcf2ab462007-09-05 14:32:49 +02001254#define SRR(x, y) (((x) + (1UL << ((y) - 1))) >> (y))
Ingo Molnar194081e2007-08-09 11:16:51 +02001255
Peter Zijlstraa7be37a2008-06-27 13:41:11 +02001256/*
1257 * delta *= weight / lw
1258 */
Ingo Molnarcb1c4fc2007-08-02 17:41:40 +02001259static unsigned long
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001260calc_delta_mine(unsigned long delta_exec, unsigned long weight,
1261 struct load_weight *lw)
1262{
1263 u64 tmp;
1264
Lai Jiangshan7a232e02008-06-12 16:43:07 +08001265 if (!lw->inv_weight) {
1266 if (BITS_PER_LONG > 32 && unlikely(lw->weight >= WMULT_CONST))
1267 lw->inv_weight = 1;
1268 else
1269 lw->inv_weight = 1 + (WMULT_CONST-lw->weight/2)
1270 / (lw->weight+1);
1271 }
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001272
1273 tmp = (u64)delta_exec * weight;
1274 /*
1275 * Check whether we'd overflow the 64-bit multiplication:
1276 */
Ingo Molnar194081e2007-08-09 11:16:51 +02001277 if (unlikely(tmp > WMULT_CONST))
Ingo Molnarcf2ab462007-09-05 14:32:49 +02001278 tmp = SRR(SRR(tmp, WMULT_SHIFT/2) * lw->inv_weight,
Ingo Molnar194081e2007-08-09 11:16:51 +02001279 WMULT_SHIFT/2);
1280 else
Ingo Molnarcf2ab462007-09-05 14:32:49 +02001281 tmp = SRR(tmp * lw->inv_weight, WMULT_SHIFT);
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001282
Ingo Molnarecf691d2007-08-02 17:41:40 +02001283 return (unsigned long)min(tmp, (u64)(unsigned long)LONG_MAX);
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001284}
1285
Ingo Molnar10919852007-10-15 17:00:04 +02001286static inline void update_load_add(struct load_weight *lw, unsigned long inc)
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001287{
1288 lw->weight += inc;
Ingo Molnare89996a2008-03-14 23:48:28 +01001289 lw->inv_weight = 0;
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001290}
1291
Ingo Molnar10919852007-10-15 17:00:04 +02001292static inline void update_load_sub(struct load_weight *lw, unsigned long dec)
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001293{
1294 lw->weight -= dec;
Ingo Molnare89996a2008-03-14 23:48:28 +01001295 lw->inv_weight = 0;
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001296}
1297
Linus Torvalds1da177e2005-04-16 15:20:36 -07001298/*
Peter Williams2dd73a42006-06-27 02:54:34 -07001299 * To aid in avoiding the subversion of "niceness" due to uneven distribution
1300 * of tasks with abnormal "nice" values across CPUs the contribution that
1301 * each task makes to its run queue's load is weighted according to its
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01001302 * scheduling class and "nice" value. For SCHED_NORMAL tasks this is just a
Peter Williams2dd73a42006-06-27 02:54:34 -07001303 * scaled version of the new time slice allocation that they receive on time
1304 * slice expiry etc.
1305 */
1306
Peter Zijlstracce7ade2009-01-15 14:53:37 +01001307#define WEIGHT_IDLEPRIO 3
1308#define WMULT_IDLEPRIO 1431655765
Ingo Molnardd41f592007-07-09 18:51:59 +02001309
1310/*
1311 * Nice levels are multiplicative, with a gentle 10% change for every
1312 * nice level changed. I.e. when a CPU-bound task goes from nice 0 to
1313 * nice 1, it will get ~10% less CPU time than another CPU-bound task
1314 * that remained on nice 0.
1315 *
1316 * The "10% effect" is relative and cumulative: from _any_ nice level,
1317 * if you go up 1 level, it's -10% CPU usage, if you go down 1 level
Ingo Molnarf9153ee62007-07-16 09:46:30 +02001318 * it's +10% CPU usage. (to achieve that we use a multiplier of 1.25.
1319 * If a task goes up by ~10% and another task goes down by ~10% then
1320 * the relative distance between them is ~25%.)
Ingo Molnardd41f592007-07-09 18:51:59 +02001321 */
1322static const int prio_to_weight[40] = {
Ingo Molnar254753d2007-08-09 11:16:51 +02001323 /* -20 */ 88761, 71755, 56483, 46273, 36291,
1324 /* -15 */ 29154, 23254, 18705, 14949, 11916,
1325 /* -10 */ 9548, 7620, 6100, 4904, 3906,
1326 /* -5 */ 3121, 2501, 1991, 1586, 1277,
1327 /* 0 */ 1024, 820, 655, 526, 423,
1328 /* 5 */ 335, 272, 215, 172, 137,
1329 /* 10 */ 110, 87, 70, 56, 45,
1330 /* 15 */ 36, 29, 23, 18, 15,
Ingo Molnardd41f592007-07-09 18:51:59 +02001331};
1332
Ingo Molnar5714d2d2007-07-16 09:46:31 +02001333/*
1334 * Inverse (2^32/x) values of the prio_to_weight[] array, precalculated.
1335 *
1336 * In cases where the weight does not change often, we can use the
1337 * precalculated inverse to speed up arithmetics by turning divisions
1338 * into multiplications:
1339 */
Ingo Molnardd41f592007-07-09 18:51:59 +02001340static const u32 prio_to_wmult[40] = {
Ingo Molnar254753d2007-08-09 11:16:51 +02001341 /* -20 */ 48388, 59856, 76040, 92818, 118348,
1342 /* -15 */ 147320, 184698, 229616, 287308, 360437,
1343 /* -10 */ 449829, 563644, 704093, 875809, 1099582,
1344 /* -5 */ 1376151, 1717300, 2157191, 2708050, 3363326,
1345 /* 0 */ 4194304, 5237765, 6557202, 8165337, 10153587,
1346 /* 5 */ 12820798, 15790321, 19976592, 24970740, 31350126,
1347 /* 10 */ 39045157, 49367440, 61356676, 76695844, 95443717,
1348 /* 15 */ 119304647, 148102320, 186737708, 238609294, 286331153,
Ingo Molnardd41f592007-07-09 18:51:59 +02001349};
Peter Williams2dd73a42006-06-27 02:54:34 -07001350
Bharata B Raoef12fef2009-03-31 10:02:22 +05301351/* Time spent by the tasks of the cpu accounting group executing in ... */
1352enum cpuacct_stat_index {
1353 CPUACCT_STAT_USER, /* ... user mode */
1354 CPUACCT_STAT_SYSTEM, /* ... kernel mode */
1355
1356 CPUACCT_STAT_NSTATS,
1357};
1358
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01001359#ifdef CONFIG_CGROUP_CPUACCT
1360static void cpuacct_charge(struct task_struct *tsk, u64 cputime);
Bharata B Raoef12fef2009-03-31 10:02:22 +05301361static void cpuacct_update_stats(struct task_struct *tsk,
1362 enum cpuacct_stat_index idx, cputime_t val);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01001363#else
1364static inline void cpuacct_charge(struct task_struct *tsk, u64 cputime) {}
Bharata B Raoef12fef2009-03-31 10:02:22 +05301365static inline void cpuacct_update_stats(struct task_struct *tsk,
1366 enum cpuacct_stat_index idx, cputime_t val) {}
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01001367#endif
1368
Peter Zijlstra18d95a22008-04-19 19:45:00 +02001369static inline void inc_cpu_load(struct rq *rq, unsigned long load)
1370{
1371 update_load_add(&rq->load, load);
1372}
1373
1374static inline void dec_cpu_load(struct rq *rq, unsigned long load)
1375{
1376 update_load_sub(&rq->load, load);
1377}
1378
Ingo Molnar7940ca32008-08-19 13:40:47 +02001379#if (defined(CONFIG_SMP) && defined(CONFIG_FAIR_GROUP_SCHED)) || defined(CONFIG_RT_GROUP_SCHED)
Peter Zijlstraeb755802008-08-19 12:33:05 +02001380typedef int (*tg_visitor)(struct task_group *, void *);
1381
1382/*
1383 * Iterate the full tree, calling @down when first entering a node and @up when
1384 * leaving it for the final time.
1385 */
1386static int walk_tg_tree(tg_visitor down, tg_visitor up, void *data)
1387{
1388 struct task_group *parent, *child;
1389 int ret;
1390
1391 rcu_read_lock();
1392 parent = &root_task_group;
1393down:
1394 ret = (*down)(parent, data);
1395 if (ret)
1396 goto out_unlock;
1397 list_for_each_entry_rcu(child, &parent->children, siblings) {
1398 parent = child;
1399 goto down;
1400
1401up:
1402 continue;
1403 }
1404 ret = (*up)(parent, data);
1405 if (ret)
1406 goto out_unlock;
1407
1408 child = parent;
1409 parent = parent->parent;
1410 if (parent)
1411 goto up;
1412out_unlock:
1413 rcu_read_unlock();
1414
1415 return ret;
1416}
1417
1418static int tg_nop(struct task_group *tg, void *data)
1419{
1420 return 0;
1421}
1422#endif
1423
Gregory Haskinse7693a32008-01-25 21:08:09 +01001424#ifdef CONFIG_SMP
Peter Zijlstraf5f08f32009-09-10 13:35:28 +02001425/* Used instead of source_load when we know the type == 0 */
1426static unsigned long weighted_cpuload(const int cpu)
1427{
1428 return cpu_rq(cpu)->load.weight;
1429}
1430
1431/*
1432 * Return a low guess at the load of a migration-source cpu weighted
1433 * according to the scheduling class and "nice" value.
1434 *
1435 * We want to under-estimate the load of migration sources, to
1436 * balance conservatively.
1437 */
1438static unsigned long source_load(int cpu, int type)
1439{
1440 struct rq *rq = cpu_rq(cpu);
1441 unsigned long total = weighted_cpuload(cpu);
1442
1443 if (type == 0 || !sched_feat(LB_BIAS))
1444 return total;
1445
1446 return min(rq->cpu_load[type-1], total);
1447}
1448
1449/*
1450 * Return a high guess at the load of a migration-target cpu weighted
1451 * according to the scheduling class and "nice" value.
1452 */
1453static unsigned long target_load(int cpu, int type)
1454{
1455 struct rq *rq = cpu_rq(cpu);
1456 unsigned long total = weighted_cpuload(cpu);
1457
1458 if (type == 0 || !sched_feat(LB_BIAS))
1459 return total;
1460
1461 return max(rq->cpu_load[type-1], total);
1462}
1463
Peter Zijlstraae154be2009-09-10 14:40:57 +02001464static struct sched_group *group_of(int cpu)
1465{
1466 struct sched_domain *sd = rcu_dereference(cpu_rq(cpu)->sd);
1467
1468 if (!sd)
1469 return NULL;
1470
1471 return sd->groups;
1472}
1473
1474static unsigned long power_of(int cpu)
1475{
1476 struct sched_group *group = group_of(cpu);
1477
1478 if (!group)
1479 return SCHED_LOAD_SCALE;
1480
1481 return group->cpu_power;
1482}
1483
Gregory Haskinse7693a32008-01-25 21:08:09 +01001484static int task_hot(struct task_struct *p, u64 now, struct sched_domain *sd);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001485
Peter Zijlstraa8a51d52008-06-27 13:41:26 +02001486static unsigned long cpu_avg_load_per_task(int cpu)
1487{
1488 struct rq *rq = cpu_rq(cpu);
Ingo Molnaraf6d5962008-11-29 20:45:15 +01001489 unsigned long nr_running = ACCESS_ONCE(rq->nr_running);
Peter Zijlstraa8a51d52008-06-27 13:41:26 +02001490
Steven Rostedt4cd42622008-11-26 21:04:24 -05001491 if (nr_running)
1492 rq->avg_load_per_task = rq->load.weight / nr_running;
Balbir Singha2d47772008-11-12 16:19:00 +05301493 else
1494 rq->avg_load_per_task = 0;
Peter Zijlstraa8a51d52008-06-27 13:41:26 +02001495
1496 return rq->avg_load_per_task;
1497}
1498
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001499#ifdef CONFIG_FAIR_GROUP_SCHED
1500
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001501static __read_mostly unsigned long *update_shares_data;
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001502
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001503static void __set_se_shares(struct sched_entity *se, unsigned long shares);
1504
1505/*
1506 * Calculate and set the cpu's group shares.
1507 */
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001508static void update_group_shares_cpu(struct task_group *tg, int cpu,
1509 unsigned long sd_shares,
1510 unsigned long sd_rq_weight,
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001511 unsigned long *usd_rq_weight)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001512{
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001513 unsigned long shares, rq_weight;
Peter Zijlstraa5004272009-07-27 14:04:49 +02001514 int boost = 0;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001515
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001516 rq_weight = usd_rq_weight[cpu];
Peter Zijlstraa5004272009-07-27 14:04:49 +02001517 if (!rq_weight) {
1518 boost = 1;
1519 rq_weight = NICE_0_LOAD;
1520 }
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001521
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001522 /*
Peter Zijlstraa8af7242009-08-21 13:58:54 +02001523 * \Sum_j shares_j * rq_weight_i
1524 * shares_i = -----------------------------
1525 * \Sum_j rq_weight_j
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001526 */
Ken Chenec4e0e22008-11-18 22:41:57 -08001527 shares = (sd_shares * rq_weight) / sd_rq_weight;
Peter Zijlstraffda12a2008-10-17 19:27:02 +02001528 shares = clamp_t(unsigned long, shares, MIN_SHARES, MAX_SHARES);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001529
Peter Zijlstraffda12a2008-10-17 19:27:02 +02001530 if (abs(shares - tg->se[cpu]->load.weight) >
1531 sysctl_sched_shares_thresh) {
1532 struct rq *rq = cpu_rq(cpu);
1533 unsigned long flags;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001534
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001535 raw_spin_lock_irqsave(&rq->lock, flags);
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001536 tg->cfs_rq[cpu]->rq_weight = boost ? 0 : rq_weight;
Peter Zijlstraa5004272009-07-27 14:04:49 +02001537 tg->cfs_rq[cpu]->shares = boost ? 0 : shares;
Peter Zijlstraffda12a2008-10-17 19:27:02 +02001538 __set_se_shares(tg->se[cpu], shares);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001539 raw_spin_unlock_irqrestore(&rq->lock, flags);
Peter Zijlstraffda12a2008-10-17 19:27:02 +02001540 }
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001541}
1542
1543/*
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001544 * Re-compute the task group their per cpu shares over the given domain.
1545 * This needs to be done in a bottom-up fashion because the rq weight of a
1546 * parent group depends on the shares of its child groups.
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001547 */
Peter Zijlstraeb755802008-08-19 12:33:05 +02001548static int tg_shares_up(struct task_group *tg, void *data)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001549{
Peter Zijlstracd8ad402009-12-03 18:00:07 +01001550 unsigned long weight, rq_weight = 0, sum_weight = 0, shares = 0;
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001551 unsigned long *usd_rq_weight;
Peter Zijlstraeb755802008-08-19 12:33:05 +02001552 struct sched_domain *sd = data;
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001553 unsigned long flags;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001554 int i;
1555
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001556 if (!tg->se[0])
1557 return 0;
1558
1559 local_irq_save(flags);
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001560 usd_rq_weight = per_cpu_ptr(update_shares_data, smp_processor_id());
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001561
Rusty Russell758b2cd2008-11-25 02:35:04 +10301562 for_each_cpu(i, sched_domain_span(sd)) {
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001563 weight = tg->cfs_rq[i]->load.weight;
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001564 usd_rq_weight[i] = weight;
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001565
Peter Zijlstracd8ad402009-12-03 18:00:07 +01001566 rq_weight += weight;
Ken Chenec4e0e22008-11-18 22:41:57 -08001567 /*
1568 * If there are currently no tasks on the cpu pretend there
1569 * is one of average load so that when a new task gets to
1570 * run here it will not get delayed by group starvation.
1571 */
Ken Chenec4e0e22008-11-18 22:41:57 -08001572 if (!weight)
1573 weight = NICE_0_LOAD;
1574
Peter Zijlstracd8ad402009-12-03 18:00:07 +01001575 sum_weight += weight;
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001576 shares += tg->cfs_rq[i]->shares;
1577 }
1578
Peter Zijlstracd8ad402009-12-03 18:00:07 +01001579 if (!rq_weight)
1580 rq_weight = sum_weight;
1581
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001582 if ((!shares && rq_weight) || shares > tg->shares)
1583 shares = tg->shares;
1584
1585 if (!sd->parent || !(sd->parent->flags & SD_LOAD_BALANCE))
1586 shares = tg->shares;
1587
Rusty Russell758b2cd2008-11-25 02:35:04 +10301588 for_each_cpu(i, sched_domain_span(sd))
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09001589 update_group_shares_cpu(tg, i, shares, rq_weight, usd_rq_weight);
Peter Zijlstra34d76c42009-08-27 13:08:56 +02001590
1591 local_irq_restore(flags);
Peter Zijlstraeb755802008-08-19 12:33:05 +02001592
1593 return 0;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001594}
1595
1596/*
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001597 * Compute the cpu's hierarchical load factor for each task group.
1598 * This needs to be done in a top-down fashion because the load of a child
1599 * group is a fraction of its parents load.
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001600 */
Peter Zijlstraeb755802008-08-19 12:33:05 +02001601static int tg_load_down(struct task_group *tg, void *data)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001602{
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001603 unsigned long load;
Peter Zijlstraeb755802008-08-19 12:33:05 +02001604 long cpu = (long)data;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001605
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001606 if (!tg->parent) {
1607 load = cpu_rq(cpu)->load.weight;
1608 } else {
1609 load = tg->parent->cfs_rq[cpu]->h_load;
1610 load *= tg->cfs_rq[cpu]->shares;
1611 load /= tg->parent->cfs_rq[cpu]->load.weight + 1;
1612 }
1613
1614 tg->cfs_rq[cpu]->h_load = load;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001615
Peter Zijlstraeb755802008-08-19 12:33:05 +02001616 return 0;
Peter Zijlstra4d8d5952008-06-27 13:41:19 +02001617}
1618
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001619static void update_shares(struct sched_domain *sd)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001620{
Peter Zijlstrae7097152009-06-03 15:41:20 +02001621 s64 elapsed;
1622 u64 now;
1623
1624 if (root_task_group_empty())
1625 return;
1626
1627 now = cpu_clock(raw_smp_processor_id());
1628 elapsed = now - sd->last_update;
Peter Zijlstra2398f2c2008-06-27 13:41:35 +02001629
1630 if (elapsed >= (s64)(u64)sysctl_sched_shares_ratelimit) {
1631 sd->last_update = now;
Peter Zijlstraeb755802008-08-19 12:33:05 +02001632 walk_tg_tree(tg_nop, tg_shares_up, sd);
Peter Zijlstra2398f2c2008-06-27 13:41:35 +02001633 }
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001634}
1635
Peter Zijlstra3e5459b2008-06-27 13:41:24 +02001636static void update_shares_locked(struct rq *rq, struct sched_domain *sd)
1637{
Peter Zijlstrae7097152009-06-03 15:41:20 +02001638 if (root_task_group_empty())
1639 return;
1640
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001641 raw_spin_unlock(&rq->lock);
Peter Zijlstra3e5459b2008-06-27 13:41:24 +02001642 update_shares(sd);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001643 raw_spin_lock(&rq->lock);
Peter Zijlstra3e5459b2008-06-27 13:41:24 +02001644}
1645
Peter Zijlstraeb755802008-08-19 12:33:05 +02001646static void update_h_load(long cpu)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001647{
Peter Zijlstrae7097152009-06-03 15:41:20 +02001648 if (root_task_group_empty())
1649 return;
1650
Peter Zijlstraeb755802008-08-19 12:33:05 +02001651 walk_tg_tree(tg_load_down, tg_nop, (void *)cpu);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001652}
1653
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001654#else
1655
Peter Zijlstrac8cba852008-06-27 13:41:23 +02001656static inline void update_shares(struct sched_domain *sd)
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001657{
1658}
1659
Peter Zijlstra3e5459b2008-06-27 13:41:24 +02001660static inline void update_shares_locked(struct rq *rq, struct sched_domain *sd)
1661{
1662}
1663
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001664#endif
1665
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001666#ifdef CONFIG_PREEMPT
1667
Peter Zijlstrab78bb862009-09-15 14:23:18 +02001668static void double_rq_lock(struct rq *rq1, struct rq *rq2);
1669
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001670/*
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001671 * fair double_lock_balance: Safely acquires both rq->locks in a fair
1672 * way at the expense of forcing extra atomic operations in all
1673 * invocations. This assures that the double_lock is acquired using the
1674 * same underlying policy as the spinlock_t on this architecture, which
1675 * reduces latency compared to the unfair variant below. However, it
1676 * also adds more overhead and therefore may reduce throughput.
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001677 */
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001678static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
1679 __releases(this_rq->lock)
1680 __acquires(busiest->lock)
1681 __acquires(this_rq->lock)
1682{
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001683 raw_spin_unlock(&this_rq->lock);
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001684 double_rq_lock(this_rq, busiest);
1685
1686 return 1;
1687}
1688
1689#else
1690/*
1691 * Unfair double_lock_balance: Optimizes throughput at the expense of
1692 * latency by eliminating extra atomic operations when the locks are
1693 * already in proper order on entry. This favors lower cpu-ids and will
1694 * grant the double lock to lower cpus over higher ids under contention,
1695 * regardless of entry order into the function.
1696 */
1697static int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001698 __releases(this_rq->lock)
1699 __acquires(busiest->lock)
1700 __acquires(this_rq->lock)
1701{
1702 int ret = 0;
1703
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001704 if (unlikely(!raw_spin_trylock(&busiest->lock))) {
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001705 if (busiest < this_rq) {
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001706 raw_spin_unlock(&this_rq->lock);
1707 raw_spin_lock(&busiest->lock);
1708 raw_spin_lock_nested(&this_rq->lock,
1709 SINGLE_DEPTH_NESTING);
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001710 ret = 1;
1711 } else
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001712 raw_spin_lock_nested(&busiest->lock,
1713 SINGLE_DEPTH_NESTING);
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001714 }
1715 return ret;
1716}
1717
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001718#endif /* CONFIG_PREEMPT */
1719
1720/*
1721 * double_lock_balance - lock the busiest runqueue, this_rq is locked already.
1722 */
1723static int double_lock_balance(struct rq *this_rq, struct rq *busiest)
1724{
1725 if (unlikely(!irqs_disabled())) {
1726 /* printk() doesn't work good under rq->lock */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001727 raw_spin_unlock(&this_rq->lock);
Gregory Haskins8f45e2b2008-12-29 09:39:51 -05001728 BUG_ON(1);
1729 }
1730
1731 return _double_lock_balance(this_rq, busiest);
1732}
1733
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001734static inline void double_unlock_balance(struct rq *this_rq, struct rq *busiest)
1735 __releases(busiest->lock)
1736{
Thomas Gleixner05fa7852009-11-17 14:28:38 +01001737 raw_spin_unlock(&busiest->lock);
Alexey Dobriyan70574a92008-11-28 22:08:00 +03001738 lock_set_subclass(&this_rq->lock.dep_map, 0, _RET_IP_);
1739}
Peter Zijlstra1e3c88b2009-12-17 17:00:43 +01001740
1741/*
1742 * double_rq_lock - safely lock two runqueues
1743 *
1744 * Note this does not disable interrupts like task_rq_lock,
1745 * you need to do so manually before calling.
1746 */
1747static void double_rq_lock(struct rq *rq1, struct rq *rq2)
1748 __acquires(rq1->lock)
1749 __acquires(rq2->lock)
1750{
1751 BUG_ON(!irqs_disabled());
1752 if (rq1 == rq2) {
1753 raw_spin_lock(&rq1->lock);
1754 __acquire(rq2->lock); /* Fake it out ;) */
1755 } else {
1756 if (rq1 < rq2) {
1757 raw_spin_lock(&rq1->lock);
1758 raw_spin_lock_nested(&rq2->lock, SINGLE_DEPTH_NESTING);
1759 } else {
1760 raw_spin_lock(&rq2->lock);
1761 raw_spin_lock_nested(&rq1->lock, SINGLE_DEPTH_NESTING);
1762 }
1763 }
1764 update_rq_clock(rq1);
1765 update_rq_clock(rq2);
1766}
1767
1768/*
1769 * double_rq_unlock - safely unlock two runqueues
1770 *
1771 * Note this does not restore interrupts like task_rq_unlock,
1772 * you need to do so manually after calling.
1773 */
1774static void double_rq_unlock(struct rq *rq1, struct rq *rq2)
1775 __releases(rq1->lock)
1776 __releases(rq2->lock)
1777{
1778 raw_spin_unlock(&rq1->lock);
1779 if (rq1 != rq2)
1780 raw_spin_unlock(&rq2->lock);
1781 else
1782 __release(rq2->lock);
1783}
1784
Peter Zijlstra18d95a22008-04-19 19:45:00 +02001785#endif
Peter Zijlstra18d95a22008-04-19 19:45:00 +02001786
1787#ifdef CONFIG_FAIR_GROUP_SCHED
1788static void cfs_rq_set_shares(struct cfs_rq *cfs_rq, unsigned long shares)
1789{
Vegard Nossum30432092008-06-27 21:35:50 +02001790#ifdef CONFIG_SMP
Ingo Molnar34e83e82008-06-27 15:42:36 +02001791 cfs_rq->shares = shares;
1792#endif
Peter Zijlstra18d95a22008-04-19 19:45:00 +02001793}
1794#endif
1795
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02001796static void calc_load_account_active(struct rq *this_rq);
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +01001797static void update_sysctl(void);
Christian Ehrhardtacb4a842009-11-30 12:16:48 +01001798static int get_update_sysctl_factor(void);
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02001799
Peter Zijlstracd29fe62009-11-27 17:32:46 +01001800static inline void __set_task_cpu(struct task_struct *p, unsigned int cpu)
1801{
1802 set_task_rq(p, cpu);
1803#ifdef CONFIG_SMP
1804 /*
1805 * After ->cpu is set up to a new value, task_rq_lock(p, ...) can be
1806 * successfuly executed on another CPU. We must ensure that updates of
1807 * per-task data have been completed by this moment.
1808 */
1809 smp_wmb();
1810 task_thread_info(p)->cpu = cpu;
1811#endif
1812}
Gregory Haskinse7693a32008-01-25 21:08:09 +01001813
Peter Zijlstra1e3c88b2009-12-17 17:00:43 +01001814static const struct sched_class rt_sched_class;
Ingo Molnardd41f592007-07-09 18:51:59 +02001815
1816#define sched_class_highest (&rt_sched_class)
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04001817#define for_each_class(class) \
1818 for (class = sched_class_highest; class; class = class->next)
Ingo Molnardd41f592007-07-09 18:51:59 +02001819
Peter Zijlstra1e3c88b2009-12-17 17:00:43 +01001820#include "sched_stats.h"
1821
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001822static void inc_nr_running(struct rq *rq)
Ingo Molnar6363ca52008-05-29 11:28:57 +02001823{
1824 rq->nr_running++;
Ingo Molnar6363ca52008-05-29 11:28:57 +02001825}
1826
Peter Zijlstrac09595f2008-06-27 13:41:14 +02001827static void dec_nr_running(struct rq *rq)
Ingo Molnar9c217242007-08-02 17:41:40 +02001828{
1829 rq->nr_running--;
Ingo Molnar9c217242007-08-02 17:41:40 +02001830}
1831
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001832static void set_load_weight(struct task_struct *p)
1833{
1834 if (task_has_rt_policy(p)) {
Ingo Molnardd41f592007-07-09 18:51:59 +02001835 p->se.load.weight = prio_to_weight[0] * 2;
1836 p->se.load.inv_weight = prio_to_wmult[0] >> 1;
1837 return;
1838 }
1839
1840 /*
1841 * SCHED_IDLE tasks get minimal weight:
1842 */
1843 if (p->policy == SCHED_IDLE) {
1844 p->se.load.weight = WEIGHT_IDLEPRIO;
1845 p->se.load.inv_weight = WMULT_IDLEPRIO;
1846 return;
1847 }
1848
1849 p->se.load.weight = prio_to_weight[p->static_prio - MAX_RT_PRIO];
1850 p->se.load.inv_weight = prio_to_wmult[p->static_prio - MAX_RT_PRIO];
Ingo Molnar45bf76d2007-07-09 18:51:59 +02001851}
1852
Gregory Haskins2087a1a2008-06-27 14:30:00 -06001853static void update_avg(u64 *avg, u64 sample)
1854{
1855 s64 diff = sample - *avg;
1856 *avg += diff >> 3;
1857}
1858
Thomas Gleixnerea87bb72010-01-20 20:58:57 +00001859static void
1860enqueue_task(struct rq *rq, struct task_struct *p, int wakeup, bool head)
Ingo Molnar71f8bd42007-07-09 18:51:59 +02001861{
Peter Zijlstra831451a2009-01-14 12:39:18 +01001862 if (wakeup)
1863 p->se.start_runtime = p->se.sum_exec_runtime;
1864
Ingo Molnar71f8bd42007-07-09 18:51:59 +02001865 sched_info_queued(p);
Thomas Gleixnerea87bb72010-01-20 20:58:57 +00001866 p->sched_class->enqueue_task(rq, p, wakeup, head);
Ingo Molnardd41f592007-07-09 18:51:59 +02001867 p->se.on_rq = 1;
1868}
1869
Ingo Molnar69be72c2007-08-09 11:16:49 +02001870static void dequeue_task(struct rq *rq, struct task_struct *p, int sleep)
Ingo Molnardd41f592007-07-09 18:51:59 +02001871{
Peter Zijlstra831451a2009-01-14 12:39:18 +01001872 if (sleep) {
1873 if (p->se.last_wakeup) {
1874 update_avg(&p->se.avg_overlap,
1875 p->se.sum_exec_runtime - p->se.last_wakeup);
1876 p->se.last_wakeup = 0;
1877 } else {
1878 update_avg(&p->se.avg_wakeup,
1879 sysctl_sched_wakeup_granularity);
1880 }
Gregory Haskins2087a1a2008-06-27 14:30:00 -06001881 }
1882
Ankita Garg46ac22b2008-07-01 14:30:06 +05301883 sched_info_dequeued(p);
Ingo Molnarf02231e2007-08-09 11:16:48 +02001884 p->sched_class->dequeue_task(rq, p, sleep);
Ingo Molnardd41f592007-07-09 18:51:59 +02001885 p->se.on_rq = 0;
Ingo Molnar71f8bd42007-07-09 18:51:59 +02001886}
1887
1888/*
Peter Zijlstra1e3c88b2009-12-17 17:00:43 +01001889 * activate_task - move a task to the runqueue.
1890 */
1891static void activate_task(struct rq *rq, struct task_struct *p, int wakeup)
1892{
1893 if (task_contributes_to_load(p))
1894 rq->nr_uninterruptible--;
1895
Thomas Gleixnerea87bb72010-01-20 20:58:57 +00001896 enqueue_task(rq, p, wakeup, false);
Peter Zijlstra1e3c88b2009-12-17 17:00:43 +01001897 inc_nr_running(rq);
1898}
1899
1900/*
1901 * deactivate_task - remove a task from the runqueue.
1902 */
1903static void deactivate_task(struct rq *rq, struct task_struct *p, int sleep)
1904{
1905 if (task_contributes_to_load(p))
1906 rq->nr_uninterruptible++;
1907
1908 dequeue_task(rq, p, sleep);
1909 dec_nr_running(rq);
1910}
1911
1912#include "sched_idletask.c"
1913#include "sched_fair.c"
1914#include "sched_rt.c"
1915#ifdef CONFIG_SCHED_DEBUG
1916# include "sched_debug.c"
1917#endif
1918
1919/*
Ingo Molnardd41f592007-07-09 18:51:59 +02001920 * __normal_prio - return the priority that is based on the static prio
Ingo Molnar71f8bd42007-07-09 18:51:59 +02001921 */
Ingo Molnar14531182007-07-09 18:51:59 +02001922static inline int __normal_prio(struct task_struct *p)
1923{
Ingo Molnardd41f592007-07-09 18:51:59 +02001924 return p->static_prio;
Ingo Molnar14531182007-07-09 18:51:59 +02001925}
1926
1927/*
Ingo Molnarb29739f2006-06-27 02:54:51 -07001928 * Calculate the expected normal priority: i.e. priority
1929 * without taking RT-inheritance into account. Might be
1930 * boosted by interactivity modifiers. Changes upon fork,
1931 * setprio syscalls, and whenever the interactivity
1932 * estimator recalculates.
1933 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001934static inline int normal_prio(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -07001935{
1936 int prio;
1937
Ingo Molnare05606d2007-07-09 18:51:59 +02001938 if (task_has_rt_policy(p))
Ingo Molnarb29739f2006-06-27 02:54:51 -07001939 prio = MAX_RT_PRIO-1 - p->rt_priority;
1940 else
1941 prio = __normal_prio(p);
1942 return prio;
1943}
1944
1945/*
1946 * Calculate the current priority, i.e. the priority
1947 * taken into account by the scheduler. This value might
1948 * be boosted by RT tasks, or might be boosted by
1949 * interactivity modifiers. Will be RT if the task got
1950 * RT-boosted. If not then it returns p->normal_prio.
1951 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001952static int effective_prio(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -07001953{
1954 p->normal_prio = normal_prio(p);
1955 /*
1956 * If we are RT tasks or we were boosted to RT priority,
1957 * keep the priority unchanged. Otherwise, update priority
1958 * to the normal priority:
1959 */
1960 if (!rt_prio(p->prio))
1961 return p->normal_prio;
1962 return p->prio;
1963}
1964
Linus Torvalds1da177e2005-04-16 15:20:36 -07001965/**
1966 * task_curr - is this task currently executing on a CPU?
1967 * @p: the task in question.
1968 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001969inline int task_curr(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001970{
1971 return cpu_curr(task_cpu(p)) == p;
1972}
1973
Steven Rostedtcb469842008-01-25 21:08:22 +01001974static inline void check_class_changed(struct rq *rq, struct task_struct *p,
1975 const struct sched_class *prev_class,
1976 int oldprio, int running)
1977{
1978 if (prev_class != p->sched_class) {
1979 if (prev_class->switched_from)
1980 prev_class->switched_from(rq, p, running);
1981 p->sched_class->switched_to(rq, p, running);
1982 } else
1983 p->sched_class->prio_changed(rq, p, oldprio, running);
1984}
1985
Linus Torvalds1da177e2005-04-16 15:20:36 -07001986#ifdef CONFIG_SMP
Ingo Molnarcc367732007-10-15 17:00:18 +02001987/*
1988 * Is this task likely cache-hot:
1989 */
Gregory Haskinse7693a32008-01-25 21:08:09 +01001990static int
Ingo Molnarcc367732007-10-15 17:00:18 +02001991task_hot(struct task_struct *p, u64 now, struct sched_domain *sd)
1992{
1993 s64 delta;
1994
Peter Zijlstrae6c8fba2009-12-16 18:04:33 +01001995 if (p->sched_class != &fair_sched_class)
1996 return 0;
1997
Ingo Molnarf540a602008-03-15 17:10:34 +01001998 /*
1999 * Buddy candidates are cache hot:
2000 */
Mike Galbraithf685cea2009-10-23 23:09:22 +02002001 if (sched_feat(CACHE_HOT_BUDDY) && this_rq()->nr_running &&
Peter Zijlstra47932412008-11-04 21:25:09 +01002002 (&p->se == cfs_rq_of(&p->se)->next ||
2003 &p->se == cfs_rq_of(&p->se)->last))
Ingo Molnarf540a602008-03-15 17:10:34 +01002004 return 1;
2005
Ingo Molnar6bc16652007-10-15 17:00:18 +02002006 if (sysctl_sched_migration_cost == -1)
2007 return 1;
2008 if (sysctl_sched_migration_cost == 0)
2009 return 0;
2010
Ingo Molnarcc367732007-10-15 17:00:18 +02002011 delta = now - p->se.exec_start;
2012
2013 return delta < (s64)sysctl_sched_migration_cost;
2014}
2015
Ingo Molnardd41f592007-07-09 18:51:59 +02002016void set_task_cpu(struct task_struct *p, unsigned int new_cpu)
Ingo Molnarc65cc872007-07-09 18:51:58 +02002017{
Peter Zijlstrae2912002009-12-16 18:04:36 +01002018#ifdef CONFIG_SCHED_DEBUG
2019 /*
2020 * We should never call set_task_cpu() on a blocked task,
2021 * ttwu() will sort out the placement.
2022 */
Peter Zijlstra077614e2009-12-17 13:16:31 +01002023 WARN_ON_ONCE(p->state != TASK_RUNNING && p->state != TASK_WAKING &&
2024 !(task_thread_info(p)->preempt_count & PREEMPT_ACTIVE));
Peter Zijlstrae2912002009-12-16 18:04:36 +01002025#endif
2026
Mathieu Desnoyersde1d7282009-05-05 16:49:59 +08002027 trace_sched_migrate_task(p, new_cpu);
Peter Zijlstracbc34ed2008-12-10 08:08:22 +01002028
Peter Zijlstra0c697742009-12-22 15:43:19 +01002029 if (task_cpu(p) != new_cpu) {
2030 p->se.nr_migrations++;
2031 perf_sw_event(PERF_COUNT_SW_CPU_MIGRATIONS, 1, 1, NULL, 0);
2032 }
Ingo Molnardd41f592007-07-09 18:51:59 +02002033
2034 __set_task_cpu(p, new_cpu);
Ingo Molnarc65cc872007-07-09 18:51:58 +02002035}
2036
Ingo Molnar70b97a72006-07-03 00:25:42 -07002037struct migration_req {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002038 struct list_head list;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002039
Ingo Molnar36c8b582006-07-03 00:25:41 -07002040 struct task_struct *task;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002041 int dest_cpu;
2042
Linus Torvalds1da177e2005-04-16 15:20:36 -07002043 struct completion done;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002044};
Linus Torvalds1da177e2005-04-16 15:20:36 -07002045
2046/*
2047 * The task's runqueue lock must be held.
2048 * Returns true if you have to wait for migration thread.
2049 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07002050static int
Ingo Molnar70b97a72006-07-03 00:25:42 -07002051migrate_task(struct task_struct *p, int dest_cpu, struct migration_req *req)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002052{
Ingo Molnar70b97a72006-07-03 00:25:42 -07002053 struct rq *rq = task_rq(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002054
2055 /*
2056 * If the task is not on a runqueue (and not running), then
Peter Zijlstrae2912002009-12-16 18:04:36 +01002057 * the next wake-up will properly place the task.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002058 */
Peter Zijlstrae2912002009-12-16 18:04:36 +01002059 if (!p->se.on_rq && !task_running(rq, p))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002060 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002061
2062 init_completion(&req->done);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002063 req->task = p;
2064 req->dest_cpu = dest_cpu;
2065 list_add(&req->list, &rq->migration_queue);
Ingo Molnar48f24c42006-07-03 00:25:40 -07002066
Linus Torvalds1da177e2005-04-16 15:20:36 -07002067 return 1;
2068}
2069
2070/*
Markus Metzgera26b89f2009-04-03 16:43:34 +02002071 * wait_task_context_switch - wait for a thread to complete at least one
2072 * context switch.
2073 *
2074 * @p must not be current.
2075 */
2076void wait_task_context_switch(struct task_struct *p)
2077{
2078 unsigned long nvcsw, nivcsw, flags;
2079 int running;
2080 struct rq *rq;
2081
2082 nvcsw = p->nvcsw;
2083 nivcsw = p->nivcsw;
2084 for (;;) {
2085 /*
2086 * The runqueue is assigned before the actual context
2087 * switch. We need to take the runqueue lock.
2088 *
2089 * We could check initially without the lock but it is
2090 * very likely that we need to take the lock in every
2091 * iteration.
2092 */
2093 rq = task_rq_lock(p, &flags);
2094 running = task_running(rq, p);
2095 task_rq_unlock(rq, &flags);
2096
2097 if (likely(!running))
2098 break;
2099 /*
2100 * The switch count is incremented before the actual
2101 * context switch. We thus wait for two switches to be
2102 * sure at least one completed.
2103 */
2104 if ((p->nvcsw - nvcsw) > 1)
2105 break;
2106 if ((p->nivcsw - nivcsw) > 1)
2107 break;
2108
2109 cpu_relax();
2110 }
2111}
2112
2113/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07002114 * wait_task_inactive - wait for a thread to unschedule.
2115 *
Roland McGrath85ba2d82008-07-25 19:45:58 -07002116 * If @match_state is nonzero, it's the @p->state value just checked and
2117 * not expected to change. If it changes, i.e. @p might have woken up,
2118 * then return zero. When we succeed in waiting for @p to be off its CPU,
2119 * we return a positive number (its total switch count). If a second call
2120 * a short while later returns the same number, the caller can be sure that
2121 * @p has remained unscheduled the whole time.
2122 *
Linus Torvalds1da177e2005-04-16 15:20:36 -07002123 * The caller must ensure that the task *will* unschedule sometime soon,
2124 * else this function might spin for a *long* time. This function can't
2125 * be called with interrupts off, or it may introduce deadlock with
2126 * smp_call_function() if an IPI is sent by the same process we are
2127 * waiting to become inactive.
2128 */
Roland McGrath85ba2d82008-07-25 19:45:58 -07002129unsigned long wait_task_inactive(struct task_struct *p, long match_state)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002130{
2131 unsigned long flags;
Ingo Molnardd41f592007-07-09 18:51:59 +02002132 int running, on_rq;
Roland McGrath85ba2d82008-07-25 19:45:58 -07002133 unsigned long ncsw;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002134 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002135
Andi Kleen3a5c3592007-10-15 17:00:14 +02002136 for (;;) {
2137 /*
2138 * We do the initial early heuristics without holding
2139 * any task-queue locks at all. We'll only try to get
2140 * the runqueue lock when things look like they will
2141 * work out!
2142 */
2143 rq = task_rq(p);
Linus Torvaldsfa490cf2007-06-18 09:34:40 -07002144
Andi Kleen3a5c3592007-10-15 17:00:14 +02002145 /*
2146 * If the task is actively running on another CPU
2147 * still, just relax and busy-wait without holding
2148 * any locks.
2149 *
2150 * NOTE! Since we don't hold any locks, it's not
2151 * even sure that "rq" stays as the right runqueue!
2152 * But we don't care, since "task_running()" will
2153 * return false if the runqueue has changed and p
2154 * is actually now running somewhere else!
2155 */
Roland McGrath85ba2d82008-07-25 19:45:58 -07002156 while (task_running(rq, p)) {
2157 if (match_state && unlikely(p->state != match_state))
2158 return 0;
Andi Kleen3a5c3592007-10-15 17:00:14 +02002159 cpu_relax();
Roland McGrath85ba2d82008-07-25 19:45:58 -07002160 }
Linus Torvaldsfa490cf2007-06-18 09:34:40 -07002161
Andi Kleen3a5c3592007-10-15 17:00:14 +02002162 /*
2163 * Ok, time to look more closely! We need the rq
2164 * lock now, to be *sure*. If we're wrong, we'll
2165 * just go back and repeat.
2166 */
2167 rq = task_rq_lock(p, &flags);
Mathieu Desnoyers0a16b602008-07-18 12:16:17 -04002168 trace_sched_wait_task(rq, p);
Andi Kleen3a5c3592007-10-15 17:00:14 +02002169 running = task_running(rq, p);
2170 on_rq = p->se.on_rq;
Roland McGrath85ba2d82008-07-25 19:45:58 -07002171 ncsw = 0;
Oleg Nesterovf31e11d2008-08-20 16:54:44 -07002172 if (!match_state || p->state == match_state)
Oleg Nesterov93dcf552008-08-20 16:54:44 -07002173 ncsw = p->nvcsw | LONG_MIN; /* sets MSB */
Andi Kleen3a5c3592007-10-15 17:00:14 +02002174 task_rq_unlock(rq, &flags);
Linus Torvaldsfa490cf2007-06-18 09:34:40 -07002175
Andi Kleen3a5c3592007-10-15 17:00:14 +02002176 /*
Roland McGrath85ba2d82008-07-25 19:45:58 -07002177 * If it changed from the expected state, bail out now.
2178 */
2179 if (unlikely(!ncsw))
2180 break;
2181
2182 /*
Andi Kleen3a5c3592007-10-15 17:00:14 +02002183 * Was it really running after all now that we
2184 * checked with the proper locks actually held?
2185 *
2186 * Oops. Go back and try again..
2187 */
2188 if (unlikely(running)) {
2189 cpu_relax();
2190 continue;
2191 }
2192
2193 /*
2194 * It's not enough that it's not actively running,
2195 * it must be off the runqueue _entirely_, and not
2196 * preempted!
2197 *
Luis Henriques80dd99b2009-03-16 19:58:09 +00002198 * So if it was still runnable (but just not actively
Andi Kleen3a5c3592007-10-15 17:00:14 +02002199 * running right now), it's preempted, and we should
2200 * yield - it could be a while.
2201 */
2202 if (unlikely(on_rq)) {
2203 schedule_timeout_uninterruptible(1);
2204 continue;
2205 }
2206
2207 /*
2208 * Ahh, all good. It wasn't running, and it wasn't
2209 * runnable, which means that it will never become
2210 * running in the future either. We're all done!
2211 */
2212 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002213 }
Roland McGrath85ba2d82008-07-25 19:45:58 -07002214
2215 return ncsw;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002216}
2217
2218/***
2219 * kick_process - kick a running thread to enter/exit the kernel
2220 * @p: the to-be-kicked thread
2221 *
2222 * Cause a process which is running on another CPU to enter
2223 * kernel-mode, without any delay. (to get signals handled.)
2224 *
2225 * NOTE: this function doesnt have to take the runqueue lock,
2226 * because all it wants to ensure is that the remote task enters
2227 * the kernel. If the IPI races and the task has been migrated
2228 * to another CPU then no harm is done and the purpose has been
2229 * achieved as well.
2230 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07002231void kick_process(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002232{
2233 int cpu;
2234
2235 preempt_disable();
2236 cpu = task_cpu(p);
2237 if ((cpu != smp_processor_id()) && task_curr(p))
2238 smp_send_reschedule(cpu);
2239 preempt_enable();
2240}
Rusty Russellb43e3522009-06-12 22:27:00 -06002241EXPORT_SYMBOL_GPL(kick_process);
Nick Piggin476d1392005-06-25 14:57:29 -07002242#endif /* CONFIG_SMP */
Linus Torvalds1da177e2005-04-16 15:20:36 -07002243
Thomas Gleixner0793a612008-12-04 20:12:29 +01002244/**
2245 * task_oncpu_function_call - call a function on the cpu on which a task runs
2246 * @p: the task to evaluate
2247 * @func: the function to be called
2248 * @info: the function call argument
2249 *
2250 * Calls the function @func when the task is currently running. This might
2251 * be on the current CPU, which just calls the function directly
2252 */
2253void task_oncpu_function_call(struct task_struct *p,
2254 void (*func) (void *info), void *info)
2255{
2256 int cpu;
2257
2258 preempt_disable();
2259 cpu = task_cpu(p);
2260 if (task_curr(p))
2261 smp_call_function_single(cpu, func, info, 1);
2262 preempt_enable();
2263}
2264
Peter Zijlstra970b13b2009-11-25 13:31:39 +01002265#ifdef CONFIG_SMP
Peter Zijlstra5da9a0f2009-12-16 18:04:38 +01002266static int select_fallback_rq(int cpu, struct task_struct *p)
2267{
2268 int dest_cpu;
2269 const struct cpumask *nodemask = cpumask_of_node(cpu_to_node(cpu));
2270
2271 /* Look for allowed, online CPU in same node. */
2272 for_each_cpu_and(dest_cpu, nodemask, cpu_active_mask)
2273 if (cpumask_test_cpu(dest_cpu, &p->cpus_allowed))
2274 return dest_cpu;
2275
2276 /* Any allowed, online CPU? */
2277 dest_cpu = cpumask_any_and(&p->cpus_allowed, cpu_active_mask);
2278 if (dest_cpu < nr_cpu_ids)
2279 return dest_cpu;
2280
2281 /* No more Mr. Nice Guy. */
2282 if (dest_cpu >= nr_cpu_ids) {
2283 rcu_read_lock();
2284 cpuset_cpus_allowed_locked(p, &p->cpus_allowed);
2285 rcu_read_unlock();
2286 dest_cpu = cpumask_any_and(cpu_active_mask, &p->cpus_allowed);
2287
2288 /*
2289 * Don't tell them about moving exiting tasks or
2290 * kernel threads (both mm NULL), since they never
2291 * leave kernel.
2292 */
2293 if (p->mm && printk_ratelimit()) {
2294 printk(KERN_INFO "process %d (%s) no "
2295 "longer affine to cpu%d\n",
2296 task_pid_nr(p), p->comm, cpu);
2297 }
2298 }
2299
2300 return dest_cpu;
2301}
2302
Peter Zijlstrae2912002009-12-16 18:04:36 +01002303/*
2304 * Called from:
2305 *
2306 * - fork, @p is stable because it isn't on the tasklist yet
2307 *
Peter Zijlstra38022902009-12-16 18:04:37 +01002308 * - exec, @p is unstable, retry loop
Peter Zijlstrae2912002009-12-16 18:04:36 +01002309 *
2310 * - wake-up, we serialize ->cpus_allowed against TASK_WAKING so
2311 * we should be good.
2312 */
Peter Zijlstra970b13b2009-11-25 13:31:39 +01002313static inline
2314int select_task_rq(struct task_struct *p, int sd_flags, int wake_flags)
2315{
Peter Zijlstrae2912002009-12-16 18:04:36 +01002316 int cpu = p->sched_class->select_task_rq(p, sd_flags, wake_flags);
2317
2318 /*
2319 * In order not to call set_task_cpu() on a blocking task we need
2320 * to rely on ttwu() to place the task on a valid ->cpus_allowed
2321 * cpu.
2322 *
2323 * Since this is common to all placement strategies, this lives here.
2324 *
2325 * [ this allows ->select_task() to simply return task_cpu(p) and
2326 * not worry about this generic constraint ]
2327 */
2328 if (unlikely(!cpumask_test_cpu(cpu, &p->cpus_allowed) ||
Peter Zijlstra70f11202009-12-20 17:36:27 +01002329 !cpu_online(cpu)))
Peter Zijlstra5da9a0f2009-12-16 18:04:38 +01002330 cpu = select_fallback_rq(task_cpu(p), p);
Peter Zijlstrae2912002009-12-16 18:04:36 +01002331
2332 return cpu;
Peter Zijlstra970b13b2009-11-25 13:31:39 +01002333}
2334#endif
2335
Linus Torvalds1da177e2005-04-16 15:20:36 -07002336/***
2337 * try_to_wake_up - wake up a thread
2338 * @p: the to-be-woken-up thread
2339 * @state: the mask of task states that can be woken
2340 * @sync: do a synchronous wakeup?
2341 *
2342 * Put it on the run-queue if it's not already there. The "current"
2343 * thread is always on the run-queue (except when the actual
2344 * re-schedule is in progress), and as such you're allowed to do
2345 * the simpler "current->state = TASK_RUNNING" to mark yourself
2346 * runnable without the overhead of this.
2347 *
2348 * returns failure only if the task is already active.
2349 */
Peter Zijlstra7d478722009-09-14 19:55:44 +02002350static int try_to_wake_up(struct task_struct *p, unsigned int state,
2351 int wake_flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002352{
Ingo Molnarcc367732007-10-15 17:00:18 +02002353 int cpu, orig_cpu, this_cpu, success = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002354 unsigned long flags;
Mike Galbraithf5dc3752009-10-09 08:35:03 +02002355 struct rq *rq, *orig_rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002356
Ingo Molnarb85d0662008-03-16 20:03:22 +01002357 if (!sched_feat(SYNC_WAKEUPS))
Peter Zijlstra7d478722009-09-14 19:55:44 +02002358 wake_flags &= ~WF_SYNC;
Ingo Molnarb85d0662008-03-16 20:03:22 +01002359
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002360 this_cpu = get_cpu();
Peter Zijlstra2398f2c2008-06-27 13:41:35 +02002361
Linus Torvalds04e2f172008-02-23 18:05:03 -08002362 smp_wmb();
Mike Galbraithf5dc3752009-10-09 08:35:03 +02002363 rq = orig_rq = task_rq_lock(p, &flags);
Mike Galbraith03e89e42008-12-16 08:45:30 +01002364 update_rq_clock(rq);
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002365 if (!(p->state & state))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002366 goto out;
2367
Ingo Molnardd41f592007-07-09 18:51:59 +02002368 if (p->se.on_rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002369 goto out_running;
2370
2371 cpu = task_cpu(p);
Ingo Molnarcc367732007-10-15 17:00:18 +02002372 orig_cpu = cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002373
2374#ifdef CONFIG_SMP
2375 if (unlikely(task_running(rq, p)))
2376 goto out_activate;
2377
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002378 /*
2379 * In order to handle concurrent wakeups and release the rq->lock
2380 * we put the task in TASK_WAKING state.
Ingo Molnareb240732009-09-16 21:09:13 +02002381 *
2382 * First fix up the nr_uninterruptible count:
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002383 */
Ingo Molnareb240732009-09-16 21:09:13 +02002384 if (task_contributes_to_load(p))
2385 rq->nr_uninterruptible--;
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002386 p->state = TASK_WAKING;
Peter Zijlstraefbbd052009-12-16 18:04:40 +01002387
2388 if (p->sched_class->task_waking)
2389 p->sched_class->task_waking(rq, p);
2390
Peter Zijlstraab19cb22009-11-27 15:44:43 +01002391 __task_rq_unlock(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002392
Peter Zijlstra970b13b2009-11-25 13:31:39 +01002393 cpu = select_task_rq(p, SD_BALANCE_WAKE, wake_flags);
Peter Zijlstraab19cb22009-11-27 15:44:43 +01002394 if (cpu != orig_cpu)
Mike Galbraith055a0082009-11-12 11:07:44 +01002395 set_task_cpu(p, cpu);
Peter Zijlstraab19cb22009-11-27 15:44:43 +01002396
2397 rq = __task_rq_lock(p);
2398 update_rq_clock(rq);
Mike Galbraithf5dc3752009-10-09 08:35:03 +02002399
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002400 WARN_ON(p->state != TASK_WAKING);
2401 cpu = task_cpu(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002402
Gregory Haskinse7693a32008-01-25 21:08:09 +01002403#ifdef CONFIG_SCHEDSTATS
2404 schedstat_inc(rq, ttwu_count);
2405 if (cpu == this_cpu)
2406 schedstat_inc(rq, ttwu_local);
2407 else {
2408 struct sched_domain *sd;
2409 for_each_domain(this_cpu, sd) {
Rusty Russell758b2cd2008-11-25 02:35:04 +10302410 if (cpumask_test_cpu(cpu, sched_domain_span(sd))) {
Gregory Haskinse7693a32008-01-25 21:08:09 +01002411 schedstat_inc(sd, ttwu_wake_remote);
2412 break;
2413 }
2414 }
2415 }
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02002416#endif /* CONFIG_SCHEDSTATS */
Gregory Haskinse7693a32008-01-25 21:08:09 +01002417
Linus Torvalds1da177e2005-04-16 15:20:36 -07002418out_activate:
2419#endif /* CONFIG_SMP */
Ingo Molnarcc367732007-10-15 17:00:18 +02002420 schedstat_inc(p, se.nr_wakeups);
Peter Zijlstra7d478722009-09-14 19:55:44 +02002421 if (wake_flags & WF_SYNC)
Ingo Molnarcc367732007-10-15 17:00:18 +02002422 schedstat_inc(p, se.nr_wakeups_sync);
2423 if (orig_cpu != cpu)
2424 schedstat_inc(p, se.nr_wakeups_migrate);
2425 if (cpu == this_cpu)
2426 schedstat_inc(p, se.nr_wakeups_local);
2427 else
2428 schedstat_inc(p, se.nr_wakeups_remote);
Ingo Molnardd41f592007-07-09 18:51:59 +02002429 activate_task(rq, p, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002430 success = 1;
2431
Peter Zijlstra831451a2009-01-14 12:39:18 +01002432 /*
2433 * Only attribute actual wakeups done by this task.
2434 */
2435 if (!in_interrupt()) {
2436 struct sched_entity *se = &current->se;
2437 u64 sample = se->sum_exec_runtime;
2438
2439 if (se->last_wakeup)
2440 sample -= se->last_wakeup;
2441 else
2442 sample -= se->start_runtime;
2443 update_avg(&se->avg_wakeup, sample);
2444
2445 se->last_wakeup = se->sum_exec_runtime;
2446 }
2447
Linus Torvalds1da177e2005-04-16 15:20:36 -07002448out_running:
Peter Zijlstra468a15b2008-12-16 08:07:03 +01002449 trace_sched_wakeup(rq, p, success);
Peter Zijlstra7d478722009-09-14 19:55:44 +02002450 check_preempt_curr(rq, p, wake_flags);
Ingo Molnar4ae7d5c2008-03-19 01:42:00 +01002451
Linus Torvalds1da177e2005-04-16 15:20:36 -07002452 p->state = TASK_RUNNING;
Steven Rostedt9a897c52008-01-25 21:08:22 +01002453#ifdef CONFIG_SMP
Peter Zijlstraefbbd052009-12-16 18:04:40 +01002454 if (p->sched_class->task_woken)
2455 p->sched_class->task_woken(rq, p);
Mike Galbraitheae0c9d2009-11-10 03:50:02 +01002456
2457 if (unlikely(rq->idle_stamp)) {
2458 u64 delta = rq->clock - rq->idle_stamp;
2459 u64 max = 2*sysctl_sched_migration_cost;
2460
2461 if (delta > max)
2462 rq->avg_idle = max;
2463 else
2464 update_avg(&rq->avg_idle, delta);
2465 rq->idle_stamp = 0;
2466 }
Steven Rostedt9a897c52008-01-25 21:08:22 +01002467#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002468out:
2469 task_rq_unlock(rq, &flags);
Peter Zijlstrae9c84312009-09-15 14:43:03 +02002470 put_cpu();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002471
2472 return success;
2473}
2474
David Howells50fa6102009-04-28 15:01:38 +01002475/**
2476 * wake_up_process - Wake up a specific process
2477 * @p: The process to be woken up.
2478 *
2479 * Attempt to wake up the nominated process and move it to the set of runnable
2480 * processes. Returns 1 if the process was woken up, 0 if it was already
2481 * running.
2482 *
2483 * It may be assumed that this function implies a write memory barrier before
2484 * changing the task state if and only if any tasks are woken up.
2485 */
Harvey Harrison7ad5b3a2008-02-08 04:19:53 -08002486int wake_up_process(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002487{
Matthew Wilcoxd9514f62007-12-06 11:07:07 -05002488 return try_to_wake_up(p, TASK_ALL, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002489}
Linus Torvalds1da177e2005-04-16 15:20:36 -07002490EXPORT_SYMBOL(wake_up_process);
2491
Harvey Harrison7ad5b3a2008-02-08 04:19:53 -08002492int wake_up_state(struct task_struct *p, unsigned int state)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002493{
2494 return try_to_wake_up(p, state, 0);
2495}
2496
Linus Torvalds1da177e2005-04-16 15:20:36 -07002497/*
2498 * Perform scheduler related setup for a newly forked process p.
2499 * p is forked by current.
Ingo Molnardd41f592007-07-09 18:51:59 +02002500 *
2501 * __sched_fork() is basic setup used by init_idle() too:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002502 */
Ingo Molnardd41f592007-07-09 18:51:59 +02002503static void __sched_fork(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002504{
Ingo Molnardd41f592007-07-09 18:51:59 +02002505 p->se.exec_start = 0;
2506 p->se.sum_exec_runtime = 0;
Ingo Molnarf6cf8912007-08-28 12:53:24 +02002507 p->se.prev_sum_exec_runtime = 0;
Ingo Molnar6c594c22008-12-14 12:34:15 +01002508 p->se.nr_migrations = 0;
Ingo Molnar4ae7d5c2008-03-19 01:42:00 +01002509 p->se.last_wakeup = 0;
2510 p->se.avg_overlap = 0;
Peter Zijlstra831451a2009-01-14 12:39:18 +01002511 p->se.start_runtime = 0;
2512 p->se.avg_wakeup = sysctl_sched_wakeup_granularity;
Ingo Molnar6cfb0d52007-08-02 17:41:40 +02002513
2514#ifdef CONFIG_SCHEDSTATS
Lucas De Marchi77935272009-07-09 13:57:20 +02002515 p->se.wait_start = 0;
2516 p->se.wait_max = 0;
2517 p->se.wait_count = 0;
2518 p->se.wait_sum = 0;
2519
2520 p->se.sleep_start = 0;
2521 p->se.sleep_max = 0;
2522 p->se.sum_sleep_runtime = 0;
2523
2524 p->se.block_start = 0;
2525 p->se.block_max = 0;
2526 p->se.exec_max = 0;
2527 p->se.slice_max = 0;
2528
2529 p->se.nr_migrations_cold = 0;
2530 p->se.nr_failed_migrations_affine = 0;
2531 p->se.nr_failed_migrations_running = 0;
2532 p->se.nr_failed_migrations_hot = 0;
2533 p->se.nr_forced_migrations = 0;
Lucas De Marchi77935272009-07-09 13:57:20 +02002534
2535 p->se.nr_wakeups = 0;
2536 p->se.nr_wakeups_sync = 0;
2537 p->se.nr_wakeups_migrate = 0;
2538 p->se.nr_wakeups_local = 0;
2539 p->se.nr_wakeups_remote = 0;
2540 p->se.nr_wakeups_affine = 0;
2541 p->se.nr_wakeups_affine_attempts = 0;
2542 p->se.nr_wakeups_passive = 0;
2543 p->se.nr_wakeups_idle = 0;
2544
Ingo Molnar6cfb0d52007-08-02 17:41:40 +02002545#endif
Nick Piggin476d1392005-06-25 14:57:29 -07002546
Peter Zijlstrafa717062008-01-25 21:08:27 +01002547 INIT_LIST_HEAD(&p->rt.run_list);
Ingo Molnardd41f592007-07-09 18:51:59 +02002548 p->se.on_rq = 0;
Peter Zijlstra4a55bd52008-04-19 19:45:00 +02002549 INIT_LIST_HEAD(&p->se.group_node);
Nick Piggin476d1392005-06-25 14:57:29 -07002550
Avi Kivitye107be32007-07-26 13:40:43 +02002551#ifdef CONFIG_PREEMPT_NOTIFIERS
2552 INIT_HLIST_HEAD(&p->preempt_notifiers);
2553#endif
Ingo Molnardd41f592007-07-09 18:51:59 +02002554}
2555
2556/*
2557 * fork()/clone()-time setup:
2558 */
2559void sched_fork(struct task_struct *p, int clone_flags)
2560{
2561 int cpu = get_cpu();
2562
2563 __sched_fork(p);
Peter Zijlstra06b83b52009-12-16 18:04:35 +01002564 /*
2565 * We mark the process as waking here. This guarantees that
2566 * nobody will actually run it, and a signal or other external
2567 * event cannot wake it up and insert it on the runqueue either.
2568 */
2569 p->state = TASK_WAKING;
Ingo Molnardd41f592007-07-09 18:51:59 +02002570
Ingo Molnarb29739f2006-06-27 02:54:51 -07002571 /*
Mike Galbraithb9dc29e2009-06-17 10:46:01 +02002572 * Revert to default priority/policy on fork if requested.
2573 */
2574 if (unlikely(p->sched_reset_on_fork)) {
Peter Williamsf83f9ac2009-09-24 06:47:10 +00002575 if (p->policy == SCHED_FIFO || p->policy == SCHED_RR) {
Mike Galbraithb9dc29e2009-06-17 10:46:01 +02002576 p->policy = SCHED_NORMAL;
Peter Williamsf83f9ac2009-09-24 06:47:10 +00002577 p->normal_prio = p->static_prio;
2578 }
Mike Galbraithb9dc29e2009-06-17 10:46:01 +02002579
Mike Galbraith6c697bd2009-06-17 10:48:02 +02002580 if (PRIO_TO_NICE(p->static_prio) < 0) {
2581 p->static_prio = NICE_TO_PRIO(0);
Peter Williamsf83f9ac2009-09-24 06:47:10 +00002582 p->normal_prio = p->static_prio;
Mike Galbraith6c697bd2009-06-17 10:48:02 +02002583 set_load_weight(p);
2584 }
2585
Mike Galbraithb9dc29e2009-06-17 10:46:01 +02002586 /*
2587 * We don't need the reset flag anymore after the fork. It has
2588 * fulfilled its duty:
2589 */
2590 p->sched_reset_on_fork = 0;
2591 }
Lennart Poetteringca94c442009-06-15 17:17:47 +02002592
Peter Williamsf83f9ac2009-09-24 06:47:10 +00002593 /*
2594 * Make sure we do not leak PI boosting priority to the child.
2595 */
2596 p->prio = current->normal_prio;
2597
Hiroshi Shimamoto2ddbf952007-10-15 17:00:11 +02002598 if (!rt_prio(p->prio))
2599 p->sched_class = &fair_sched_class;
Ingo Molnarb29739f2006-06-27 02:54:51 -07002600
Peter Zijlstracd29fe62009-11-27 17:32:46 +01002601 if (p->sched_class->task_fork)
2602 p->sched_class->task_fork(p);
2603
Peter Zijlstra5f3edc12009-09-10 13:42:00 +02002604#ifdef CONFIG_SMP
Peter Zijlstra970b13b2009-11-25 13:31:39 +01002605 cpu = select_task_rq(p, SD_BALANCE_FORK, 0);
Peter Zijlstra5f3edc12009-09-10 13:42:00 +02002606#endif
2607 set_task_cpu(p, cpu);
2608
Chandra Seetharaman52f17b62006-07-14 00:24:38 -07002609#if defined(CONFIG_SCHEDSTATS) || defined(CONFIG_TASK_DELAY_ACCT)
Ingo Molnardd41f592007-07-09 18:51:59 +02002610 if (likely(sched_info_on()))
Chandra Seetharaman52f17b62006-07-14 00:24:38 -07002611 memset(&p->sched_info, 0, sizeof(p->sched_info));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002612#endif
Chen, Kenneth Wd6077cb2006-02-14 13:53:10 -08002613#if defined(CONFIG_SMP) && defined(__ARCH_WANT_UNLOCKED_CTXSW)
Nick Piggin4866cde2005-06-25 14:57:23 -07002614 p->oncpu = 0;
2615#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002616#ifdef CONFIG_PREEMPT
Nick Piggin4866cde2005-06-25 14:57:23 -07002617 /* Want to start with kernel preemption disabled. */
Al Viroa1261f542005-11-13 16:06:55 -08002618 task_thread_info(p)->preempt_count = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002619#endif
Gregory Haskins917b6272008-12-29 09:39:53 -05002620 plist_node_init(&p->pushable_tasks, MAX_PRIO);
2621
Nick Piggin476d1392005-06-25 14:57:29 -07002622 put_cpu();
Linus Torvalds1da177e2005-04-16 15:20:36 -07002623}
2624
2625/*
2626 * wake_up_new_task - wake up a newly created task for the first time.
2627 *
2628 * This function will do some initial scheduler statistics housekeeping
2629 * that must be done for every newly created context, then puts the task
2630 * on the runqueue and wakes it.
2631 */
Harvey Harrison7ad5b3a2008-02-08 04:19:53 -08002632void wake_up_new_task(struct task_struct *p, unsigned long clone_flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002633{
2634 unsigned long flags;
Ingo Molnardd41f592007-07-09 18:51:59 +02002635 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002636
2637 rq = task_rq_lock(p, &flags);
Peter Zijlstra06b83b52009-12-16 18:04:35 +01002638 BUG_ON(p->state != TASK_WAKING);
2639 p->state = TASK_RUNNING;
Ingo Molnara8e504d2007-08-09 11:16:47 +02002640 update_rq_clock(rq);
Peter Zijlstracd29fe62009-11-27 17:32:46 +01002641 activate_task(rq, p, 0);
Ingo Molnarc71dd422008-12-19 01:09:51 +01002642 trace_sched_wakeup_new(rq, p, 1);
Peter Zijlstraa7558e02009-09-14 20:02:34 +02002643 check_preempt_curr(rq, p, WF_FORK);
Steven Rostedt9a897c52008-01-25 21:08:22 +01002644#ifdef CONFIG_SMP
Peter Zijlstraefbbd052009-12-16 18:04:40 +01002645 if (p->sched_class->task_woken)
2646 p->sched_class->task_woken(rq, p);
Steven Rostedt9a897c52008-01-25 21:08:22 +01002647#endif
Ingo Molnardd41f592007-07-09 18:51:59 +02002648 task_rq_unlock(rq, &flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002649}
2650
Avi Kivitye107be32007-07-26 13:40:43 +02002651#ifdef CONFIG_PREEMPT_NOTIFIERS
2652
2653/**
Luis Henriques80dd99b2009-03-16 19:58:09 +00002654 * preempt_notifier_register - tell me when current is being preempted & rescheduled
Randy Dunlap421cee22007-07-31 00:37:50 -07002655 * @notifier: notifier struct to register
Avi Kivitye107be32007-07-26 13:40:43 +02002656 */
2657void preempt_notifier_register(struct preempt_notifier *notifier)
2658{
2659 hlist_add_head(&notifier->link, &current->preempt_notifiers);
2660}
2661EXPORT_SYMBOL_GPL(preempt_notifier_register);
2662
2663/**
2664 * preempt_notifier_unregister - no longer interested in preemption notifications
Randy Dunlap421cee22007-07-31 00:37:50 -07002665 * @notifier: notifier struct to unregister
Avi Kivitye107be32007-07-26 13:40:43 +02002666 *
2667 * This is safe to call from within a preemption notifier.
2668 */
2669void preempt_notifier_unregister(struct preempt_notifier *notifier)
2670{
2671 hlist_del(&notifier->link);
2672}
2673EXPORT_SYMBOL_GPL(preempt_notifier_unregister);
2674
2675static void fire_sched_in_preempt_notifiers(struct task_struct *curr)
2676{
2677 struct preempt_notifier *notifier;
2678 struct hlist_node *node;
2679
2680 hlist_for_each_entry(notifier, node, &curr->preempt_notifiers, link)
2681 notifier->ops->sched_in(notifier, raw_smp_processor_id());
2682}
2683
2684static void
2685fire_sched_out_preempt_notifiers(struct task_struct *curr,
2686 struct task_struct *next)
2687{
2688 struct preempt_notifier *notifier;
2689 struct hlist_node *node;
2690
2691 hlist_for_each_entry(notifier, node, &curr->preempt_notifiers, link)
2692 notifier->ops->sched_out(notifier, next);
2693}
2694
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02002695#else /* !CONFIG_PREEMPT_NOTIFIERS */
Avi Kivitye107be32007-07-26 13:40:43 +02002696
2697static void fire_sched_in_preempt_notifiers(struct task_struct *curr)
2698{
2699}
2700
2701static void
2702fire_sched_out_preempt_notifiers(struct task_struct *curr,
2703 struct task_struct *next)
2704{
2705}
2706
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02002707#endif /* CONFIG_PREEMPT_NOTIFIERS */
Avi Kivitye107be32007-07-26 13:40:43 +02002708
Linus Torvalds1da177e2005-04-16 15:20:36 -07002709/**
Nick Piggin4866cde2005-06-25 14:57:23 -07002710 * prepare_task_switch - prepare to switch tasks
2711 * @rq: the runqueue preparing to switch
Randy Dunlap421cee22007-07-31 00:37:50 -07002712 * @prev: the current task that is being switched out
Nick Piggin4866cde2005-06-25 14:57:23 -07002713 * @next: the task we are going to switch to.
2714 *
2715 * This is called with the rq lock held and interrupts off. It must
2716 * be paired with a subsequent finish_task_switch after the context
2717 * switch.
2718 *
2719 * prepare_task_switch sets up locking and calls architecture specific
2720 * hooks.
2721 */
Avi Kivitye107be32007-07-26 13:40:43 +02002722static inline void
2723prepare_task_switch(struct rq *rq, struct task_struct *prev,
2724 struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -07002725{
Avi Kivitye107be32007-07-26 13:40:43 +02002726 fire_sched_out_preempt_notifiers(prev, next);
Nick Piggin4866cde2005-06-25 14:57:23 -07002727 prepare_lock_switch(rq, next);
2728 prepare_arch_switch(next);
2729}
2730
2731/**
Linus Torvalds1da177e2005-04-16 15:20:36 -07002732 * finish_task_switch - clean up after a task-switch
Jeff Garzik344baba2005-09-07 01:15:17 -04002733 * @rq: runqueue associated with task-switch
Linus Torvalds1da177e2005-04-16 15:20:36 -07002734 * @prev: the thread we just switched away from.
2735 *
Nick Piggin4866cde2005-06-25 14:57:23 -07002736 * finish_task_switch must be called after the context switch, paired
2737 * with a prepare_task_switch call before the context switch.
2738 * finish_task_switch will reconcile locking set up by prepare_task_switch,
2739 * and do any other architecture-specific cleanup actions.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002740 *
2741 * Note that we may have delayed dropping an mm in context_switch(). If
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01002742 * so, we finish that here outside of the runqueue lock. (Doing it
Linus Torvalds1da177e2005-04-16 15:20:36 -07002743 * with the lock held can cause deadlocks; see schedule() for
2744 * details.)
2745 */
Alexey Dobriyana9957442007-10-15 17:00:13 +02002746static void finish_task_switch(struct rq *rq, struct task_struct *prev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002747 __releases(rq->lock)
2748{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002749 struct mm_struct *mm = rq->prev_mm;
Oleg Nesterov55a101f2006-09-29 02:01:10 -07002750 long prev_state;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002751
2752 rq->prev_mm = NULL;
2753
2754 /*
2755 * A task struct has one reference for the use as "current".
Oleg Nesterovc394cc92006-09-29 02:01:11 -07002756 * If a task dies, then it sets TASK_DEAD in tsk->state and calls
Oleg Nesterov55a101f2006-09-29 02:01:10 -07002757 * schedule one last time. The schedule call will never return, and
2758 * the scheduled task must drop that reference.
Oleg Nesterovc394cc92006-09-29 02:01:11 -07002759 * The test for TASK_DEAD must occur while the runqueue locks are
Linus Torvalds1da177e2005-04-16 15:20:36 -07002760 * still held, otherwise prev could be scheduled on another cpu, die
2761 * there before we look at prev->state, and then the reference would
2762 * be dropped twice.
2763 * Manfred Spraul <manfred@colorfullife.com>
2764 */
Oleg Nesterov55a101f2006-09-29 02:01:10 -07002765 prev_state = prev->state;
Nick Piggin4866cde2005-06-25 14:57:23 -07002766 finish_arch_switch(prev);
Ingo Molnarcdd6c482009-09-21 12:02:48 +02002767 perf_event_task_sched_in(current, cpu_of(rq));
Nick Piggin4866cde2005-06-25 14:57:23 -07002768 finish_lock_switch(rq, prev);
Steven Rostedte8fa1362008-01-25 21:08:05 +01002769
Avi Kivitye107be32007-07-26 13:40:43 +02002770 fire_sched_in_preempt_notifiers(current);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002771 if (mm)
2772 mmdrop(mm);
Oleg Nesterovc394cc92006-09-29 02:01:11 -07002773 if (unlikely(prev_state == TASK_DEAD)) {
bibo maoc6fd91f2006-03-26 01:38:20 -08002774 /*
2775 * Remove function-return probe instances associated with this
2776 * task and put them back on the free list.
Ingo Molnar9761eea2007-07-09 18:52:00 +02002777 */
bibo maoc6fd91f2006-03-26 01:38:20 -08002778 kprobe_flush_task(prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002779 put_task_struct(prev);
bibo maoc6fd91f2006-03-26 01:38:20 -08002780 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002781}
2782
Gregory Haskins3f029d32009-07-29 11:08:47 -04002783#ifdef CONFIG_SMP
2784
2785/* assumes rq->lock is held */
2786static inline void pre_schedule(struct rq *rq, struct task_struct *prev)
2787{
2788 if (prev->sched_class->pre_schedule)
2789 prev->sched_class->pre_schedule(rq, prev);
2790}
2791
2792/* rq->lock is NOT held, but preemption is disabled */
2793static inline void post_schedule(struct rq *rq)
2794{
2795 if (rq->post_schedule) {
2796 unsigned long flags;
2797
Thomas Gleixner05fa7852009-11-17 14:28:38 +01002798 raw_spin_lock_irqsave(&rq->lock, flags);
Gregory Haskins3f029d32009-07-29 11:08:47 -04002799 if (rq->curr->sched_class->post_schedule)
2800 rq->curr->sched_class->post_schedule(rq);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01002801 raw_spin_unlock_irqrestore(&rq->lock, flags);
Gregory Haskins3f029d32009-07-29 11:08:47 -04002802
2803 rq->post_schedule = 0;
2804 }
2805}
2806
2807#else
2808
2809static inline void pre_schedule(struct rq *rq, struct task_struct *p)
2810{
2811}
2812
2813static inline void post_schedule(struct rq *rq)
2814{
2815}
2816
2817#endif
2818
Linus Torvalds1da177e2005-04-16 15:20:36 -07002819/**
2820 * schedule_tail - first thing a freshly forked thread must call.
2821 * @prev: the thread we just switched away from.
2822 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07002823asmlinkage void schedule_tail(struct task_struct *prev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002824 __releases(rq->lock)
2825{
Ingo Molnar70b97a72006-07-03 00:25:42 -07002826 struct rq *rq = this_rq();
2827
Nick Piggin4866cde2005-06-25 14:57:23 -07002828 finish_task_switch(rq, prev);
Steven Rostedtda19ab52009-07-29 00:21:22 -04002829
Gregory Haskins3f029d32009-07-29 11:08:47 -04002830 /*
2831 * FIXME: do we need to worry about rq being invalidated by the
2832 * task_switch?
2833 */
2834 post_schedule(rq);
Steven Rostedtda19ab52009-07-29 00:21:22 -04002835
Nick Piggin4866cde2005-06-25 14:57:23 -07002836#ifdef __ARCH_WANT_UNLOCKED_CTXSW
2837 /* In this case, finish_task_switch does not reenable preemption */
2838 preempt_enable();
2839#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002840 if (current->set_child_tid)
Pavel Emelyanovb4888932007-10-18 23:40:14 -07002841 put_user(task_pid_vnr(current), current->set_child_tid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002842}
2843
2844/*
2845 * context_switch - switch to the new MM and the new
2846 * thread's register state.
2847 */
Ingo Molnardd41f592007-07-09 18:51:59 +02002848static inline void
Ingo Molnar70b97a72006-07-03 00:25:42 -07002849context_switch(struct rq *rq, struct task_struct *prev,
Ingo Molnar36c8b582006-07-03 00:25:41 -07002850 struct task_struct *next)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002851{
Ingo Molnardd41f592007-07-09 18:51:59 +02002852 struct mm_struct *mm, *oldmm;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002853
Avi Kivitye107be32007-07-26 13:40:43 +02002854 prepare_task_switch(rq, prev, next);
Mathieu Desnoyers0a16b602008-07-18 12:16:17 -04002855 trace_sched_switch(rq, prev, next);
Ingo Molnardd41f592007-07-09 18:51:59 +02002856 mm = next->mm;
2857 oldmm = prev->active_mm;
Zachary Amsden9226d122007-02-13 13:26:21 +01002858 /*
2859 * For paravirt, this is coupled with an exit in switch_to to
2860 * combine the page table reload and the switch backend into
2861 * one hypercall.
2862 */
Jeremy Fitzhardinge224101e2009-02-18 11:18:57 -08002863 arch_start_context_switch(prev);
Zachary Amsden9226d122007-02-13 13:26:21 +01002864
Tim Blechmann710390d2009-11-24 11:55:27 +01002865 if (likely(!mm)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002866 next->active_mm = oldmm;
2867 atomic_inc(&oldmm->mm_count);
2868 enter_lazy_tlb(oldmm, next);
2869 } else
2870 switch_mm(oldmm, mm, next);
2871
Tim Blechmann710390d2009-11-24 11:55:27 +01002872 if (likely(!prev->mm)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002873 prev->active_mm = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002874 rq->prev_mm = oldmm;
2875 }
Ingo Molnar3a5f5e42006-07-14 00:24:27 -07002876 /*
2877 * Since the runqueue lock will be released by the next
2878 * task (which is an invalid locking op but in the case
2879 * of the scheduler it's an obvious special-case), so we
2880 * do an early lockdep release here:
2881 */
2882#ifndef __ARCH_WANT_UNLOCKED_CTXSW
Ingo Molnar8a25d5d2006-07-03 00:24:54 -07002883 spin_release(&rq->lock.dep_map, 1, _THIS_IP_);
Ingo Molnar3a5f5e42006-07-14 00:24:27 -07002884#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002885
2886 /* Here we just switch the register state and the stack. */
2887 switch_to(prev, next, prev);
2888
Ingo Molnardd41f592007-07-09 18:51:59 +02002889 barrier();
2890 /*
2891 * this_rq must be evaluated again because prev may have moved
2892 * CPUs since it called schedule(), thus the 'rq' on its stack
2893 * frame will be invalid.
2894 */
2895 finish_task_switch(this_rq(), prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002896}
2897
2898/*
2899 * nr_running, nr_uninterruptible and nr_context_switches:
2900 *
2901 * externally visible scheduler statistics: current number of runnable
2902 * threads, current number of uninterruptible-sleeping threads, total
2903 * number of context switches performed since bootup.
2904 */
2905unsigned long nr_running(void)
2906{
2907 unsigned long i, sum = 0;
2908
2909 for_each_online_cpu(i)
2910 sum += cpu_rq(i)->nr_running;
2911
2912 return sum;
2913}
2914
2915unsigned long nr_uninterruptible(void)
2916{
2917 unsigned long i, sum = 0;
2918
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08002919 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002920 sum += cpu_rq(i)->nr_uninterruptible;
2921
2922 /*
2923 * Since we read the counters lockless, it might be slightly
2924 * inaccurate. Do not allow it to go below zero though:
2925 */
2926 if (unlikely((long)sum < 0))
2927 sum = 0;
2928
2929 return sum;
2930}
2931
2932unsigned long long nr_context_switches(void)
2933{
Steven Rostedtcc94abf2006-06-27 02:54:31 -07002934 int i;
2935 unsigned long long sum = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002936
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08002937 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002938 sum += cpu_rq(i)->nr_switches;
2939
2940 return sum;
2941}
2942
2943unsigned long nr_iowait(void)
2944{
2945 unsigned long i, sum = 0;
2946
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08002947 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002948 sum += atomic_read(&cpu_rq(i)->nr_iowait);
2949
2950 return sum;
2951}
2952
Arjan van de Ven69d25872009-09-21 17:04:08 -07002953unsigned long nr_iowait_cpu(void)
2954{
2955 struct rq *this = this_rq();
2956 return atomic_read(&this->nr_iowait);
2957}
2958
2959unsigned long this_cpu_load(void)
2960{
2961 struct rq *this = this_rq();
2962 return this->cpu_load[0];
2963}
2964
2965
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02002966/* Variables and functions for calc_load */
2967static atomic_long_t calc_load_tasks;
2968static unsigned long calc_load_update;
2969unsigned long avenrun[3];
2970EXPORT_SYMBOL(avenrun);
2971
Thomas Gleixner2d024942009-05-02 20:08:52 +02002972/**
2973 * get_avenrun - get the load average array
2974 * @loads: pointer to dest load array
2975 * @offset: offset to add
2976 * @shift: shift count to shift the result left
2977 *
2978 * These values are estimates at best, so no need for locking.
2979 */
2980void get_avenrun(unsigned long *loads, unsigned long offset, int shift)
2981{
2982 loads[0] = (avenrun[0] + offset) << shift;
2983 loads[1] = (avenrun[1] + offset) << shift;
2984 loads[2] = (avenrun[2] + offset) << shift;
2985}
2986
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02002987static unsigned long
2988calc_load(unsigned long load, unsigned long exp, unsigned long active)
Jack Steinerdb1b1fe2006-03-31 02:31:21 -08002989{
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02002990 load *= exp;
2991 load += active * (FIXED_1 - exp);
2992 return load >> FSHIFT;
2993}
Jack Steinerdb1b1fe2006-03-31 02:31:21 -08002994
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02002995/*
2996 * calc_load - update the avenrun load estimates 10 ticks after the
2997 * CPUs have updated calc_load_tasks.
2998 */
2999void calc_global_load(void)
3000{
3001 unsigned long upd = calc_load_update + 10;
3002 long active;
3003
3004 if (time_before(jiffies, upd))
3005 return;
3006
3007 active = atomic_long_read(&calc_load_tasks);
3008 active = active > 0 ? active * FIXED_1 : 0;
3009
3010 avenrun[0] = calc_load(avenrun[0], EXP_1, active);
3011 avenrun[1] = calc_load(avenrun[1], EXP_5, active);
3012 avenrun[2] = calc_load(avenrun[2], EXP_15, active);
3013
3014 calc_load_update += LOAD_FREQ;
3015}
3016
3017/*
3018 * Either called from update_cpu_load() or from a cpu going idle
3019 */
3020static void calc_load_account_active(struct rq *this_rq)
3021{
3022 long nr_active, delta;
3023
3024 nr_active = this_rq->nr_running;
3025 nr_active += (long) this_rq->nr_uninterruptible;
3026
3027 if (nr_active != this_rq->calc_load_active) {
3028 delta = nr_active - this_rq->calc_load_active;
3029 this_rq->calc_load_active = nr_active;
3030 atomic_long_add(delta, &calc_load_tasks);
Jack Steinerdb1b1fe2006-03-31 02:31:21 -08003031 }
Jack Steinerdb1b1fe2006-03-31 02:31:21 -08003032}
3033
Linus Torvalds1da177e2005-04-16 15:20:36 -07003034/*
Ingo Molnardd41f592007-07-09 18:51:59 +02003035 * Update rq->cpu_load[] statistics. This function is usually called every
3036 * scheduler tick (TICK_NSEC).
Ingo Molnar48f24c42006-07-03 00:25:40 -07003037 */
Ingo Molnardd41f592007-07-09 18:51:59 +02003038static void update_cpu_load(struct rq *this_rq)
Ingo Molnar48f24c42006-07-03 00:25:40 -07003039{
Dmitry Adamushko495eca42007-10-15 17:00:06 +02003040 unsigned long this_load = this_rq->load.weight;
Ingo Molnardd41f592007-07-09 18:51:59 +02003041 int i, scale;
3042
3043 this_rq->nr_load_updates++;
Ingo Molnardd41f592007-07-09 18:51:59 +02003044
3045 /* Update our load: */
3046 for (i = 0, scale = 1; i < CPU_LOAD_IDX_MAX; i++, scale += scale) {
3047 unsigned long old_load, new_load;
3048
3049 /* scale is effectively 1 << i now, and >> i divides by scale */
3050
3051 old_load = this_rq->cpu_load[i];
3052 new_load = this_load;
Ingo Molnara25707f2007-10-15 17:00:03 +02003053 /*
3054 * Round up the averaging division if load is increasing. This
3055 * prevents us from getting stuck on 9 if the load is 10, for
3056 * example.
3057 */
3058 if (new_load > old_load)
3059 new_load += scale-1;
Ingo Molnardd41f592007-07-09 18:51:59 +02003060 this_rq->cpu_load[i] = (old_load*(scale-1) + new_load) >> i;
3061 }
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02003062
3063 if (time_after_eq(jiffies, this_rq->calc_load_update)) {
3064 this_rq->calc_load_update += LOAD_FREQ;
3065 calc_load_account_active(this_rq);
3066 }
Ingo Molnar48f24c42006-07-03 00:25:40 -07003067}
3068
Ingo Molnardd41f592007-07-09 18:51:59 +02003069#ifdef CONFIG_SMP
3070
Ingo Molnar48f24c42006-07-03 00:25:40 -07003071/*
Peter Zijlstra38022902009-12-16 18:04:37 +01003072 * sched_exec - execve() is a valuable balancing opportunity, because at
3073 * this point the task has the smallest effective memory and cache footprint.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003074 */
Peter Zijlstra38022902009-12-16 18:04:37 +01003075void sched_exec(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003076{
Peter Zijlstra38022902009-12-16 18:04:37 +01003077 struct task_struct *p = current;
Ingo Molnar70b97a72006-07-03 00:25:42 -07003078 struct migration_req req;
Peter Zijlstra38022902009-12-16 18:04:37 +01003079 int dest_cpu, this_cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003080 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07003081 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003082
Peter Zijlstra38022902009-12-16 18:04:37 +01003083again:
3084 this_cpu = get_cpu();
3085 dest_cpu = select_task_rq(p, SD_BALANCE_EXEC, 0);
3086 if (dest_cpu == this_cpu) {
3087 put_cpu();
3088 return;
3089 }
3090
Linus Torvalds1da177e2005-04-16 15:20:36 -07003091 rq = task_rq_lock(p, &flags);
Peter Zijlstra38022902009-12-16 18:04:37 +01003092 put_cpu();
3093
3094 /*
3095 * select_task_rq() can race against ->cpus_allowed
3096 */
Rusty Russell96f874e22008-11-25 02:35:14 +10303097 if (!cpumask_test_cpu(dest_cpu, &p->cpus_allowed)
Peter Zijlstra38022902009-12-16 18:04:37 +01003098 || unlikely(!cpu_active(dest_cpu))) {
3099 task_rq_unlock(rq, &flags);
3100 goto again;
3101 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003102
3103 /* force the process onto the specified CPU */
3104 if (migrate_task(p, dest_cpu, &req)) {
3105 /* Need to wait for migration thread (might exit: take ref). */
3106 struct task_struct *mt = rq->migration_thread;
Ingo Molnar36c8b582006-07-03 00:25:41 -07003107
Linus Torvalds1da177e2005-04-16 15:20:36 -07003108 get_task_struct(mt);
3109 task_rq_unlock(rq, &flags);
3110 wake_up_process(mt);
3111 put_task_struct(mt);
3112 wait_for_completion(&req.done);
Ingo Molnar36c8b582006-07-03 00:25:41 -07003113
Linus Torvalds1da177e2005-04-16 15:20:36 -07003114 return;
3115 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003116 task_rq_unlock(rq, &flags);
3117}
3118
Linus Torvalds1da177e2005-04-16 15:20:36 -07003119#endif
3120
Linus Torvalds1da177e2005-04-16 15:20:36 -07003121DEFINE_PER_CPU(struct kernel_stat, kstat);
3122
3123EXPORT_PER_CPU_SYMBOL(kstat);
3124
3125/*
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003126 * Return any ns on the sched_clock that have not yet been accounted in
Frank Mayharf06febc2008-09-12 09:54:39 -07003127 * @p in case that task is currently running.
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003128 *
3129 * Called with task_rq_lock() held on @rq.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003130 */
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003131static u64 do_task_delta_exec(struct task_struct *p, struct rq *rq)
3132{
3133 u64 ns = 0;
3134
3135 if (task_current(rq, p)) {
3136 update_rq_clock(rq);
3137 ns = rq->clock - p->se.exec_start;
3138 if ((s64)ns < 0)
3139 ns = 0;
3140 }
3141
3142 return ns;
3143}
3144
Frank Mayharbb34d922008-09-12 09:54:39 -07003145unsigned long long task_delta_exec(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003146{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003147 unsigned long flags;
Ingo Molnar41b86e92007-07-09 18:51:58 +02003148 struct rq *rq;
Frank Mayharbb34d922008-09-12 09:54:39 -07003149 u64 ns = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07003150
Ingo Molnar41b86e92007-07-09 18:51:58 +02003151 rq = task_rq_lock(p, &flags);
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003152 ns = do_task_delta_exec(p, rq);
3153 task_rq_unlock(rq, &flags);
Ingo Molnar15084872008-09-30 08:28:17 +02003154
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003155 return ns;
3156}
Frank Mayharf06febc2008-09-12 09:54:39 -07003157
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003158/*
3159 * Return accounted runtime for the task.
3160 * In case the task is currently running, return the runtime plus current's
3161 * pending runtime that have not been accounted yet.
3162 */
3163unsigned long long task_sched_runtime(struct task_struct *p)
3164{
3165 unsigned long flags;
3166 struct rq *rq;
3167 u64 ns = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07003168
Hidetoshi Setoc5f8d992009-03-31 16:56:03 +09003169 rq = task_rq_lock(p, &flags);
3170 ns = p->se.sum_exec_runtime + do_task_delta_exec(p, rq);
3171 task_rq_unlock(rq, &flags);
3172
3173 return ns;
3174}
3175
3176/*
3177 * Return sum_exec_runtime for the thread group.
3178 * In case the task is currently running, return the sum plus current's
3179 * pending runtime that have not been accounted yet.
3180 *
3181 * Note that the thread group might have other running tasks as well,
3182 * so the return value not includes other pending runtime that other
3183 * running tasks might have.
3184 */
3185unsigned long long thread_group_sched_runtime(struct task_struct *p)
3186{
3187 struct task_cputime totals;
3188 unsigned long flags;
3189 struct rq *rq;
3190 u64 ns;
3191
3192 rq = task_rq_lock(p, &flags);
3193 thread_group_cputime(p, &totals);
3194 ns = totals.sum_exec_runtime + do_task_delta_exec(p, rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003195 task_rq_unlock(rq, &flags);
3196
3197 return ns;
3198}
3199
3200/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07003201 * Account user cpu time to a process.
3202 * @p: the process that the cpu time gets accounted to
Linus Torvalds1da177e2005-04-16 15:20:36 -07003203 * @cputime: the cpu time spent in user space since the last update
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003204 * @cputime_scaled: cputime scaled by cpu frequency
Linus Torvalds1da177e2005-04-16 15:20:36 -07003205 */
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003206void account_user_time(struct task_struct *p, cputime_t cputime,
3207 cputime_t cputime_scaled)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003208{
3209 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
3210 cputime64_t tmp;
3211
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003212 /* Add user time to process. */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003213 p->utime = cputime_add(p->utime, cputime);
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003214 p->utimescaled = cputime_add(p->utimescaled, cputime_scaled);
Frank Mayharf06febc2008-09-12 09:54:39 -07003215 account_group_user_time(p, cputime);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003216
3217 /* Add user time to cpustat. */
3218 tmp = cputime_to_cputime64(cputime);
3219 if (TASK_NICE(p) > 0)
3220 cpustat->nice = cputime64_add(cpustat->nice, tmp);
3221 else
3222 cpustat->user = cputime64_add(cpustat->user, tmp);
Bharata B Raoef12fef2009-03-31 10:02:22 +05303223
3224 cpuacct_update_stats(p, CPUACCT_STAT_USER, cputime);
Jonathan Lim49b5cf32008-07-25 01:48:40 -07003225 /* Account for user time used */
3226 acct_update_integrals(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003227}
3228
3229/*
Laurent Vivier94886b82007-10-15 17:00:19 +02003230 * Account guest cpu time to a process.
3231 * @p: the process that the cpu time gets accounted to
3232 * @cputime: the cpu time spent in virtual machine since the last update
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003233 * @cputime_scaled: cputime scaled by cpu frequency
Laurent Vivier94886b82007-10-15 17:00:19 +02003234 */
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003235static void account_guest_time(struct task_struct *p, cputime_t cputime,
3236 cputime_t cputime_scaled)
Laurent Vivier94886b82007-10-15 17:00:19 +02003237{
3238 cputime64_t tmp;
3239 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
3240
3241 tmp = cputime_to_cputime64(cputime);
3242
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003243 /* Add guest time to process. */
Laurent Vivier94886b82007-10-15 17:00:19 +02003244 p->utime = cputime_add(p->utime, cputime);
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003245 p->utimescaled = cputime_add(p->utimescaled, cputime_scaled);
Frank Mayharf06febc2008-09-12 09:54:39 -07003246 account_group_user_time(p, cputime);
Laurent Vivier94886b82007-10-15 17:00:19 +02003247 p->gtime = cputime_add(p->gtime, cputime);
3248
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003249 /* Add guest time to cpustat. */
Ryota Ozakice0e7b22009-10-24 01:20:10 +09003250 if (TASK_NICE(p) > 0) {
3251 cpustat->nice = cputime64_add(cpustat->nice, tmp);
3252 cpustat->guest_nice = cputime64_add(cpustat->guest_nice, tmp);
3253 } else {
3254 cpustat->user = cputime64_add(cpustat->user, tmp);
3255 cpustat->guest = cputime64_add(cpustat->guest, tmp);
3256 }
Laurent Vivier94886b82007-10-15 17:00:19 +02003257}
3258
3259/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07003260 * Account system cpu time to a process.
3261 * @p: the process that the cpu time gets accounted to
3262 * @hardirq_offset: the offset to subtract from hardirq_count()
3263 * @cputime: the cpu time spent in kernel space since the last update
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003264 * @cputime_scaled: cputime scaled by cpu frequency
Linus Torvalds1da177e2005-04-16 15:20:36 -07003265 */
3266void account_system_time(struct task_struct *p, int hardirq_offset,
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003267 cputime_t cputime, cputime_t cputime_scaled)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003268{
3269 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003270 cputime64_t tmp;
3271
Harvey Harrison983ed7a2008-04-24 18:17:55 -07003272 if ((p->flags & PF_VCPU) && (irq_count() - hardirq_offset == 0)) {
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003273 account_guest_time(p, cputime, cputime_scaled);
Harvey Harrison983ed7a2008-04-24 18:17:55 -07003274 return;
3275 }
Laurent Vivier94886b82007-10-15 17:00:19 +02003276
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003277 /* Add system time to process. */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003278 p->stime = cputime_add(p->stime, cputime);
Martin Schwidefsky457533a2008-12-31 15:11:37 +01003279 p->stimescaled = cputime_add(p->stimescaled, cputime_scaled);
Frank Mayharf06febc2008-09-12 09:54:39 -07003280 account_group_system_time(p, cputime);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003281
3282 /* Add system time to cpustat. */
3283 tmp = cputime_to_cputime64(cputime);
3284 if (hardirq_count() - hardirq_offset)
3285 cpustat->irq = cputime64_add(cpustat->irq, tmp);
3286 else if (softirq_count())
3287 cpustat->softirq = cputime64_add(cpustat->softirq, tmp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003288 else
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003289 cpustat->system = cputime64_add(cpustat->system, tmp);
3290
Bharata B Raoef12fef2009-03-31 10:02:22 +05303291 cpuacct_update_stats(p, CPUACCT_STAT_SYSTEM, cputime);
3292
Linus Torvalds1da177e2005-04-16 15:20:36 -07003293 /* Account for system time used */
3294 acct_update_integrals(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003295}
3296
3297/*
3298 * Account for involuntary wait time.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003299 * @steal: the cpu time spent in involuntary wait
Linus Torvalds1da177e2005-04-16 15:20:36 -07003300 */
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003301void account_steal_time(cputime_t cputime)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003302{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003303 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003304 cputime64_t cputime64 = cputime_to_cputime64(cputime);
3305
3306 cpustat->steal = cputime64_add(cpustat->steal, cputime64);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003307}
3308
Christoph Lameter7835b982006-12-10 02:20:22 -08003309/*
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003310 * Account for idle time.
3311 * @cputime: the cpu time spent in idle wait
Linus Torvalds1da177e2005-04-16 15:20:36 -07003312 */
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003313void account_idle_time(cputime_t cputime)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003314{
3315 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003316 cputime64_t cputime64 = cputime_to_cputime64(cputime);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003317 struct rq *rq = this_rq();
3318
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003319 if (atomic_read(&rq->nr_iowait) > 0)
3320 cpustat->iowait = cputime64_add(cpustat->iowait, cputime64);
3321 else
3322 cpustat->idle = cputime64_add(cpustat->idle, cputime64);
Christoph Lameter7835b982006-12-10 02:20:22 -08003323}
3324
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003325#ifndef CONFIG_VIRT_CPU_ACCOUNTING
3326
3327/*
3328 * Account a single tick of cpu time.
3329 * @p: the process that the cpu time gets accounted to
3330 * @user_tick: indicates if the tick is a user or a system tick
3331 */
3332void account_process_tick(struct task_struct *p, int user_tick)
3333{
Stanislaw Gruszkaa42548a2009-07-29 12:15:29 +02003334 cputime_t one_jiffy_scaled = cputime_to_scaled(cputime_one_jiffy);
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003335 struct rq *rq = this_rq();
3336
3337 if (user_tick)
Stanislaw Gruszkaa42548a2009-07-29 12:15:29 +02003338 account_user_time(p, cputime_one_jiffy, one_jiffy_scaled);
Eric Dumazetf5f293a2009-04-29 14:44:49 +02003339 else if ((p != rq->idle) || (irq_count() != HARDIRQ_OFFSET))
Stanislaw Gruszkaa42548a2009-07-29 12:15:29 +02003340 account_system_time(p, HARDIRQ_OFFSET, cputime_one_jiffy,
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003341 one_jiffy_scaled);
3342 else
Stanislaw Gruszkaa42548a2009-07-29 12:15:29 +02003343 account_idle_time(cputime_one_jiffy);
Martin Schwidefsky79741dd2008-12-31 15:11:38 +01003344}
3345
3346/*
3347 * Account multiple ticks of steal time.
3348 * @p: the process from which the cpu time has been stolen
3349 * @ticks: number of stolen ticks
3350 */
3351void account_steal_ticks(unsigned long ticks)
3352{
3353 account_steal_time(jiffies_to_cputime(ticks));
3354}
3355
3356/*
3357 * Account multiple ticks of idle time.
3358 * @ticks: number of stolen ticks
3359 */
3360void account_idle_ticks(unsigned long ticks)
3361{
3362 account_idle_time(jiffies_to_cputime(ticks));
3363}
3364
3365#endif
3366
Christoph Lameter7835b982006-12-10 02:20:22 -08003367/*
Balbir Singh49048622008-09-05 18:12:23 +02003368 * Use precise platform statistics if available:
3369 */
3370#ifdef CONFIG_VIRT_CPU_ACCOUNTING
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003371void task_times(struct task_struct *p, cputime_t *ut, cputime_t *st)
Balbir Singh49048622008-09-05 18:12:23 +02003372{
Hidetoshi Setod99ca3b2009-12-02 17:26:47 +09003373 *ut = p->utime;
3374 *st = p->stime;
Balbir Singh49048622008-09-05 18:12:23 +02003375}
3376
Hidetoshi Seto0cf55e12009-12-02 17:28:07 +09003377void thread_group_times(struct task_struct *p, cputime_t *ut, cputime_t *st)
Balbir Singh49048622008-09-05 18:12:23 +02003378{
Hidetoshi Seto0cf55e12009-12-02 17:28:07 +09003379 struct task_cputime cputime;
3380
3381 thread_group_cputime(p, &cputime);
3382
3383 *ut = cputime.utime;
3384 *st = cputime.stime;
Balbir Singh49048622008-09-05 18:12:23 +02003385}
3386#else
Hidetoshi Seto761b1d22009-11-12 13:33:45 +09003387
3388#ifndef nsecs_to_cputime
Hidetoshi Setob7b20df92009-11-26 14:49:27 +09003389# define nsecs_to_cputime(__nsecs) nsecs_to_jiffies(__nsecs)
Hidetoshi Seto761b1d22009-11-12 13:33:45 +09003390#endif
3391
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003392void task_times(struct task_struct *p, cputime_t *ut, cputime_t *st)
Balbir Singh49048622008-09-05 18:12:23 +02003393{
Hidetoshi Setod99ca3b2009-12-02 17:26:47 +09003394 cputime_t rtime, utime = p->utime, total = cputime_add(utime, p->stime);
Balbir Singh49048622008-09-05 18:12:23 +02003395
3396 /*
3397 * Use CFS's precise accounting:
3398 */
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003399 rtime = nsecs_to_cputime(p->se.sum_exec_runtime);
Balbir Singh49048622008-09-05 18:12:23 +02003400
3401 if (total) {
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003402 u64 temp;
Balbir Singh49048622008-09-05 18:12:23 +02003403
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003404 temp = (u64)(rtime * utime);
Balbir Singh49048622008-09-05 18:12:23 +02003405 do_div(temp, total);
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003406 utime = (cputime_t)temp;
3407 } else
3408 utime = rtime;
Balbir Singh49048622008-09-05 18:12:23 +02003409
3410 /*
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003411 * Compare with previous values, to keep monotonicity:
Balbir Singh49048622008-09-05 18:12:23 +02003412 */
Hidetoshi Seto761b1d22009-11-12 13:33:45 +09003413 p->prev_utime = max(p->prev_utime, utime);
Hidetoshi Setod99ca3b2009-12-02 17:26:47 +09003414 p->prev_stime = max(p->prev_stime, cputime_sub(rtime, p->prev_utime));
Balbir Singh49048622008-09-05 18:12:23 +02003415
Hidetoshi Setod99ca3b2009-12-02 17:26:47 +09003416 *ut = p->prev_utime;
3417 *st = p->prev_stime;
Hidetoshi Setod180c5b2009-11-26 14:48:30 +09003418}
Balbir Singh49048622008-09-05 18:12:23 +02003419
Hidetoshi Seto0cf55e12009-12-02 17:28:07 +09003420/*
3421 * Must be called with siglock held.
3422 */
3423void thread_group_times(struct task_struct *p, cputime_t *ut, cputime_t *st)
3424{
3425 struct signal_struct *sig = p->signal;
3426 struct task_cputime cputime;
3427 cputime_t rtime, utime, total;
3428
3429 thread_group_cputime(p, &cputime);
3430
3431 total = cputime_add(cputime.utime, cputime.stime);
3432 rtime = nsecs_to_cputime(cputime.sum_exec_runtime);
3433
3434 if (total) {
3435 u64 temp;
3436
3437 temp = (u64)(rtime * cputime.utime);
3438 do_div(temp, total);
3439 utime = (cputime_t)temp;
3440 } else
3441 utime = rtime;
3442
3443 sig->prev_utime = max(sig->prev_utime, utime);
3444 sig->prev_stime = max(sig->prev_stime,
3445 cputime_sub(rtime, sig->prev_utime));
3446
3447 *ut = sig->prev_utime;
3448 *st = sig->prev_stime;
Balbir Singh49048622008-09-05 18:12:23 +02003449}
3450#endif
3451
Balbir Singh49048622008-09-05 18:12:23 +02003452/*
Christoph Lameter7835b982006-12-10 02:20:22 -08003453 * This function gets called by the timer code, with HZ frequency.
3454 * We call it with interrupts disabled.
3455 *
3456 * It also gets called by the fork code, when changing the parent's
3457 * timeslices.
3458 */
3459void scheduler_tick(void)
3460{
Christoph Lameter7835b982006-12-10 02:20:22 -08003461 int cpu = smp_processor_id();
3462 struct rq *rq = cpu_rq(cpu);
Ingo Molnardd41f592007-07-09 18:51:59 +02003463 struct task_struct *curr = rq->curr;
Peter Zijlstra3e51f332008-05-03 18:29:28 +02003464
3465 sched_clock_tick();
Christoph Lameter7835b982006-12-10 02:20:22 -08003466
Thomas Gleixner05fa7852009-11-17 14:28:38 +01003467 raw_spin_lock(&rq->lock);
Peter Zijlstra3e51f332008-05-03 18:29:28 +02003468 update_rq_clock(rq);
Ingo Molnarf1a438d2007-08-09 11:16:45 +02003469 update_cpu_load(rq);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01003470 curr->sched_class->task_tick(rq, curr, 0);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01003471 raw_spin_unlock(&rq->lock);
Ingo Molnardd41f592007-07-09 18:51:59 +02003472
Ingo Molnarcdd6c482009-09-21 12:02:48 +02003473 perf_event_task_tick(curr, cpu);
Peter Zijlstrae220d2d2009-05-23 18:28:55 +02003474
Christoph Lametere418e1c2006-12-10 02:20:23 -08003475#ifdef CONFIG_SMP
Ingo Molnardd41f592007-07-09 18:51:59 +02003476 rq->idle_at_tick = idle_cpu(cpu);
3477 trigger_load_balance(rq, cpu);
Christoph Lametere418e1c2006-12-10 02:20:23 -08003478#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07003479}
3480
Lai Jiangshan132380a2009-04-02 14:18:25 +08003481notrace unsigned long get_parent_ip(unsigned long addr)
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003482{
3483 if (in_lock_functions(addr)) {
3484 addr = CALLER_ADDR2;
3485 if (in_lock_functions(addr))
3486 addr = CALLER_ADDR3;
3487 }
3488 return addr;
3489}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003490
Steven Rostedt7e49fcc2009-01-22 19:01:40 -05003491#if defined(CONFIG_PREEMPT) && (defined(CONFIG_DEBUG_PREEMPT) || \
3492 defined(CONFIG_PREEMPT_TRACER))
3493
Srinivasa Ds43627582008-02-23 15:24:04 -08003494void __kprobes add_preempt_count(int val)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003495{
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003496#ifdef CONFIG_DEBUG_PREEMPT
Linus Torvalds1da177e2005-04-16 15:20:36 -07003497 /*
3498 * Underflow?
3499 */
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003500 if (DEBUG_LOCKS_WARN_ON((preempt_count() < 0)))
3501 return;
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003502#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07003503 preempt_count() += val;
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003504#ifdef CONFIG_DEBUG_PREEMPT
Linus Torvalds1da177e2005-04-16 15:20:36 -07003505 /*
3506 * Spinlock count overflowing soon?
3507 */
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08003508 DEBUG_LOCKS_WARN_ON((preempt_count() & PREEMPT_MASK) >=
3509 PREEMPT_MASK - 10);
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003510#endif
3511 if (preempt_count() == val)
3512 trace_preempt_off(CALLER_ADDR0, get_parent_ip(CALLER_ADDR1));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003513}
3514EXPORT_SYMBOL(add_preempt_count);
3515
Srinivasa Ds43627582008-02-23 15:24:04 -08003516void __kprobes sub_preempt_count(int val)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003517{
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003518#ifdef CONFIG_DEBUG_PREEMPT
Linus Torvalds1da177e2005-04-16 15:20:36 -07003519 /*
3520 * Underflow?
3521 */
Ingo Molnar01e3eb82009-01-12 13:00:50 +01003522 if (DEBUG_LOCKS_WARN_ON(val > preempt_count()))
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003523 return;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003524 /*
3525 * Is the spinlock portion underflowing?
3526 */
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003527 if (DEBUG_LOCKS_WARN_ON((val < PREEMPT_MASK) &&
3528 !(preempt_count() & PREEMPT_MASK)))
3529 return;
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003530#endif
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003531
Steven Rostedt6cd8a4b2008-05-12 21:20:42 +02003532 if (preempt_count() == val)
3533 trace_preempt_on(CALLER_ADDR0, get_parent_ip(CALLER_ADDR1));
Linus Torvalds1da177e2005-04-16 15:20:36 -07003534 preempt_count() -= val;
3535}
3536EXPORT_SYMBOL(sub_preempt_count);
3537
3538#endif
3539
3540/*
Ingo Molnardd41f592007-07-09 18:51:59 +02003541 * Print scheduling while atomic bug:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003542 */
Ingo Molnardd41f592007-07-09 18:51:59 +02003543static noinline void __schedule_bug(struct task_struct *prev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003544{
Satyam Sharma838225b2007-10-24 18:23:50 +02003545 struct pt_regs *regs = get_irq_regs();
3546
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01003547 printk(KERN_ERR "BUG: scheduling while atomic: %s/%d/0x%08x\n",
3548 prev->comm, prev->pid, preempt_count());
Satyam Sharma838225b2007-10-24 18:23:50 +02003549
Ingo Molnardd41f592007-07-09 18:51:59 +02003550 debug_show_held_locks(prev);
Arjan van de Vene21f5b12008-05-23 09:05:58 -07003551 print_modules();
Ingo Molnardd41f592007-07-09 18:51:59 +02003552 if (irqs_disabled())
3553 print_irqtrace_events(prev);
Satyam Sharma838225b2007-10-24 18:23:50 +02003554
3555 if (regs)
3556 show_regs(regs);
3557 else
3558 dump_stack();
Ingo Molnardd41f592007-07-09 18:51:59 +02003559}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003560
Ingo Molnardd41f592007-07-09 18:51:59 +02003561/*
3562 * Various schedule()-time debugging checks and statistics:
3563 */
3564static inline void schedule_debug(struct task_struct *prev)
3565{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003566 /*
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01003567 * Test if we are atomic. Since do_exit() needs to call into
Linus Torvalds1da177e2005-04-16 15:20:36 -07003568 * schedule() atomically, we ignore that path for now.
3569 * Otherwise, whine if we are scheduling when we should not be.
3570 */
Roel Kluin3f33a7c2008-05-13 23:44:11 +02003571 if (unlikely(in_atomic_preempt_off() && !prev->exit_state))
Ingo Molnardd41f592007-07-09 18:51:59 +02003572 __schedule_bug(prev);
3573
Linus Torvalds1da177e2005-04-16 15:20:36 -07003574 profile_hit(SCHED_PROFILING, __builtin_return_address(0));
3575
Ingo Molnar2d723762007-10-15 17:00:12 +02003576 schedstat_inc(this_rq(), sched_count);
Ingo Molnarb8efb562007-10-15 17:00:10 +02003577#ifdef CONFIG_SCHEDSTATS
3578 if (unlikely(prev->lock_depth >= 0)) {
Ingo Molnar2d723762007-10-15 17:00:12 +02003579 schedstat_inc(this_rq(), bkl_count);
3580 schedstat_inc(prev, sched_info.bkl_count);
Ingo Molnarb8efb562007-10-15 17:00:10 +02003581 }
3582#endif
Ingo Molnardd41f592007-07-09 18:51:59 +02003583}
3584
Peter Zijlstra6cecd082009-11-30 13:00:37 +01003585static void put_prev_task(struct rq *rq, struct task_struct *prev)
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003586{
Peter Zijlstra6cecd082009-11-30 13:00:37 +01003587 if (prev->state == TASK_RUNNING) {
3588 u64 runtime = prev->se.sum_exec_runtime;
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003589
Peter Zijlstra6cecd082009-11-30 13:00:37 +01003590 runtime -= prev->se.prev_sum_exec_runtime;
3591 runtime = min_t(u64, runtime, 2*sysctl_sched_migration_cost);
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003592
3593 /*
3594 * In order to avoid avg_overlap growing stale when we are
3595 * indeed overlapping and hence not getting put to sleep, grow
3596 * the avg_overlap on preemption.
3597 *
3598 * We use the average preemption runtime because that
3599 * correlates to the amount of cache footprint a task can
3600 * build up.
3601 */
Peter Zijlstra6cecd082009-11-30 13:00:37 +01003602 update_avg(&prev->se.avg_overlap, runtime);
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003603 }
Peter Zijlstra6cecd082009-11-30 13:00:37 +01003604 prev->sched_class->put_prev_task(rq, prev);
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003605}
3606
Ingo Molnardd41f592007-07-09 18:51:59 +02003607/*
3608 * Pick up the highest-prio task:
3609 */
3610static inline struct task_struct *
Wang Chenb67802e2009-03-02 13:55:26 +08003611pick_next_task(struct rq *rq)
Ingo Molnardd41f592007-07-09 18:51:59 +02003612{
Ingo Molnar5522d5d2007-10-15 17:00:12 +02003613 const struct sched_class *class;
Ingo Molnardd41f592007-07-09 18:51:59 +02003614 struct task_struct *p;
3615
3616 /*
3617 * Optimization: we know that if all tasks are in
3618 * the fair class we can call that function directly:
3619 */
3620 if (likely(rq->nr_running == rq->cfs.nr_running)) {
Ingo Molnarfb8d4722007-08-09 11:16:48 +02003621 p = fair_sched_class.pick_next_task(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02003622 if (likely(p))
3623 return p;
3624 }
3625
3626 class = sched_class_highest;
3627 for ( ; ; ) {
Ingo Molnarfb8d4722007-08-09 11:16:48 +02003628 p = class->pick_next_task(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02003629 if (p)
3630 return p;
3631 /*
3632 * Will never be NULL as the idle class always
3633 * returns a non-NULL p:
3634 */
3635 class = class->next;
3636 }
3637}
3638
3639/*
3640 * schedule() is the main scheduler function.
3641 */
Peter Zijlstraff743342009-03-13 12:21:26 +01003642asmlinkage void __sched schedule(void)
Ingo Molnardd41f592007-07-09 18:51:59 +02003643{
3644 struct task_struct *prev, *next;
Harvey Harrison67ca7bd2008-02-15 09:56:36 -08003645 unsigned long *switch_count;
Ingo Molnardd41f592007-07-09 18:51:59 +02003646 struct rq *rq;
Peter Zijlstra31656512008-07-18 18:01:23 +02003647 int cpu;
Ingo Molnardd41f592007-07-09 18:51:59 +02003648
Peter Zijlstraff743342009-03-13 12:21:26 +01003649need_resched:
3650 preempt_disable();
Ingo Molnardd41f592007-07-09 18:51:59 +02003651 cpu = smp_processor_id();
3652 rq = cpu_rq(cpu);
Paul E. McKenneyd6714c22009-08-22 13:56:46 -07003653 rcu_sched_qs(cpu);
Ingo Molnardd41f592007-07-09 18:51:59 +02003654 prev = rq->curr;
3655 switch_count = &prev->nivcsw;
3656
Linus Torvalds1da177e2005-04-16 15:20:36 -07003657 release_kernel_lock(prev);
3658need_resched_nonpreemptible:
Linus Torvalds1da177e2005-04-16 15:20:36 -07003659
Ingo Molnardd41f592007-07-09 18:51:59 +02003660 schedule_debug(prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003661
Peter Zijlstra31656512008-07-18 18:01:23 +02003662 if (sched_feat(HRTICK))
Mike Galbraithf333fdc2008-05-12 21:20:55 +02003663 hrtick_clear(rq);
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01003664
Thomas Gleixner05fa7852009-11-17 14:28:38 +01003665 raw_spin_lock_irq(&rq->lock);
Peter Zijlstra3e51f332008-05-03 18:29:28 +02003666 update_rq_clock(rq);
Ingo Molnar1e819952007-10-15 17:00:13 +02003667 clear_tsk_need_resched(prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003668
Ingo Molnardd41f592007-07-09 18:51:59 +02003669 if (prev->state && !(preempt_count() & PREEMPT_ACTIVE)) {
Oleg Nesterov16882c12008-06-08 21:20:41 +04003670 if (unlikely(signal_pending_state(prev->state, prev)))
Ingo Molnardd41f592007-07-09 18:51:59 +02003671 prev->state = TASK_RUNNING;
Oleg Nesterov16882c12008-06-08 21:20:41 +04003672 else
Ingo Molnar2e1cb742007-08-09 11:16:49 +02003673 deactivate_task(rq, prev, 1);
Ingo Molnardd41f592007-07-09 18:51:59 +02003674 switch_count = &prev->nvcsw;
3675 }
3676
Gregory Haskins3f029d32009-07-29 11:08:47 -04003677 pre_schedule(rq, prev);
Steven Rostedtf65eda42008-01-25 21:08:07 +01003678
Ingo Molnardd41f592007-07-09 18:51:59 +02003679 if (unlikely(!rq->nr_running))
3680 idle_balance(cpu, rq);
3681
Mike Galbraithdf1c99d2009-03-10 19:08:11 +01003682 put_prev_task(rq, prev);
Wang Chenb67802e2009-03-02 13:55:26 +08003683 next = pick_next_task(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003684
Linus Torvalds1da177e2005-04-16 15:20:36 -07003685 if (likely(prev != next)) {
David Simner673a90a2008-04-29 10:08:59 +01003686 sched_info_switch(prev, next);
Ingo Molnarcdd6c482009-09-21 12:02:48 +02003687 perf_event_task_sched_out(prev, next, cpu);
David Simner673a90a2008-04-29 10:08:59 +01003688
Linus Torvalds1da177e2005-04-16 15:20:36 -07003689 rq->nr_switches++;
3690 rq->curr = next;
3691 ++*switch_count;
3692
Ingo Molnardd41f592007-07-09 18:51:59 +02003693 context_switch(rq, prev, next); /* unlocks the rq */
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01003694 /*
3695 * the context switch might have flipped the stack from under
3696 * us, hence refresh the local variables.
3697 */
3698 cpu = smp_processor_id();
3699 rq = cpu_rq(cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003700 } else
Thomas Gleixner05fa7852009-11-17 14:28:38 +01003701 raw_spin_unlock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003702
Gregory Haskins3f029d32009-07-29 11:08:47 -04003703 post_schedule(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003704
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01003705 if (unlikely(reacquire_kernel_lock(current) < 0))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003706 goto need_resched_nonpreemptible;
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01003707
Linus Torvalds1da177e2005-04-16 15:20:36 -07003708 preempt_enable_no_resched();
Peter Zijlstraff743342009-03-13 12:21:26 +01003709 if (need_resched())
Linus Torvalds1da177e2005-04-16 15:20:36 -07003710 goto need_resched;
3711}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003712EXPORT_SYMBOL(schedule);
3713
Frederic Weisbeckerc08f7822009-12-02 20:49:17 +01003714#ifdef CONFIG_MUTEX_SPIN_ON_OWNER
Peter Zijlstra0d66bf62009-01-12 14:01:47 +01003715/*
3716 * Look out! "owner" is an entirely speculative pointer
3717 * access and not reliable.
3718 */
3719int mutex_spin_on_owner(struct mutex *lock, struct thread_info *owner)
3720{
3721 unsigned int cpu;
3722 struct rq *rq;
3723
3724 if (!sched_feat(OWNER_SPIN))
3725 return 0;
3726
3727#ifdef CONFIG_DEBUG_PAGEALLOC
3728 /*
3729 * Need to access the cpu field knowing that
3730 * DEBUG_PAGEALLOC could have unmapped it if
3731 * the mutex owner just released it and exited.
3732 */
3733 if (probe_kernel_address(&owner->cpu, cpu))
3734 goto out;
3735#else
3736 cpu = owner->cpu;
3737#endif
3738
3739 /*
3740 * Even if the access succeeded (likely case),
3741 * the cpu field may no longer be valid.
3742 */
3743 if (cpu >= nr_cpumask_bits)
3744 goto out;
3745
3746 /*
3747 * We need to validate that we can do a
3748 * get_cpu() and that we have the percpu area.
3749 */
3750 if (!cpu_online(cpu))
3751 goto out;
3752
3753 rq = cpu_rq(cpu);
3754
3755 for (;;) {
3756 /*
3757 * Owner changed, break to re-assess state.
3758 */
3759 if (lock->owner != owner)
3760 break;
3761
3762 /*
3763 * Is that owner really running on that cpu?
3764 */
3765 if (task_thread_info(rq->curr) != owner || need_resched())
3766 return 0;
3767
3768 cpu_relax();
3769 }
3770out:
3771 return 1;
3772}
3773#endif
3774
Linus Torvalds1da177e2005-04-16 15:20:36 -07003775#ifdef CONFIG_PREEMPT
3776/*
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003777 * this is the entry point to schedule() from in-kernel preemption
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01003778 * off of preempt_enable. Kernel preemptions off return from interrupt
Linus Torvalds1da177e2005-04-16 15:20:36 -07003779 * occur there and call schedule directly.
3780 */
3781asmlinkage void __sched preempt_schedule(void)
3782{
3783 struct thread_info *ti = current_thread_info();
Ingo Molnar6478d882008-01-25 21:08:33 +01003784
Linus Torvalds1da177e2005-04-16 15:20:36 -07003785 /*
3786 * If there is a non-zero preempt_count or interrupts are disabled,
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01003787 * we do not want to preempt the current task. Just return..
Linus Torvalds1da177e2005-04-16 15:20:36 -07003788 */
Nick Pigginbeed33a2006-10-11 01:21:52 -07003789 if (likely(ti->preempt_count || irqs_disabled()))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003790 return;
3791
Andi Kleen3a5c3592007-10-15 17:00:14 +02003792 do {
3793 add_preempt_count(PREEMPT_ACTIVE);
Andi Kleen3a5c3592007-10-15 17:00:14 +02003794 schedule();
Andi Kleen3a5c3592007-10-15 17:00:14 +02003795 sub_preempt_count(PREEMPT_ACTIVE);
3796
3797 /*
3798 * Check again in case we missed a preemption opportunity
3799 * between schedule and now.
3800 */
3801 barrier();
Lai Jiangshan5ed0cec2009-03-06 19:40:20 +08003802 } while (need_resched());
Linus Torvalds1da177e2005-04-16 15:20:36 -07003803}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003804EXPORT_SYMBOL(preempt_schedule);
3805
3806/*
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003807 * this is the entry point to schedule() from kernel preemption
Linus Torvalds1da177e2005-04-16 15:20:36 -07003808 * off of irq context.
3809 * Note, that this is called and return with irqs disabled. This will
3810 * protect us against recursive calling from irq.
3811 */
3812asmlinkage void __sched preempt_schedule_irq(void)
3813{
3814 struct thread_info *ti = current_thread_info();
Ingo Molnar6478d882008-01-25 21:08:33 +01003815
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003816 /* Catch callers which need to be fixed */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003817 BUG_ON(ti->preempt_count || !irqs_disabled());
3818
Andi Kleen3a5c3592007-10-15 17:00:14 +02003819 do {
3820 add_preempt_count(PREEMPT_ACTIVE);
Andi Kleen3a5c3592007-10-15 17:00:14 +02003821 local_irq_enable();
3822 schedule();
3823 local_irq_disable();
Andi Kleen3a5c3592007-10-15 17:00:14 +02003824 sub_preempt_count(PREEMPT_ACTIVE);
3825
3826 /*
3827 * Check again in case we missed a preemption opportunity
3828 * between schedule and now.
3829 */
3830 barrier();
Lai Jiangshan5ed0cec2009-03-06 19:40:20 +08003831 } while (need_resched());
Linus Torvalds1da177e2005-04-16 15:20:36 -07003832}
3833
3834#endif /* CONFIG_PREEMPT */
3835
Peter Zijlstra63859d42009-09-15 19:14:42 +02003836int default_wake_function(wait_queue_t *curr, unsigned mode, int wake_flags,
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07003837 void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003838{
Peter Zijlstra63859d42009-09-15 19:14:42 +02003839 return try_to_wake_up(curr->private, mode, wake_flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003840}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003841EXPORT_SYMBOL(default_wake_function);
3842
3843/*
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01003844 * The core wakeup function. Non-exclusive wakeups (nr_exclusive == 0) just
3845 * wake everything up. If it's an exclusive wakeup (nr_exclusive == small +ve
Linus Torvalds1da177e2005-04-16 15:20:36 -07003846 * number) then we wake all the non-exclusive tasks and one exclusive task.
3847 *
3848 * There are circumstances in which we can try to wake a task which has already
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01003849 * started to run but is not in state TASK_RUNNING. try_to_wake_up() returns
Linus Torvalds1da177e2005-04-16 15:20:36 -07003850 * zero in this (rare) case, and we handle it by continuing to scan the queue.
3851 */
Johannes Weiner78ddb082009-04-14 16:53:05 +02003852static void __wake_up_common(wait_queue_head_t *q, unsigned int mode,
Peter Zijlstra63859d42009-09-15 19:14:42 +02003853 int nr_exclusive, int wake_flags, void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003854{
Matthias Kaehlcke2e458742007-10-15 17:00:02 +02003855 wait_queue_t *curr, *next;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003856
Matthias Kaehlcke2e458742007-10-15 17:00:02 +02003857 list_for_each_entry_safe(curr, next, &q->task_list, task_list) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07003858 unsigned flags = curr->flags;
3859
Peter Zijlstra63859d42009-09-15 19:14:42 +02003860 if (curr->func(curr, mode, wake_flags, key) &&
Ingo Molnar48f24c42006-07-03 00:25:40 -07003861 (flags & WQ_FLAG_EXCLUSIVE) && !--nr_exclusive)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003862 break;
3863 }
3864}
3865
3866/**
3867 * __wake_up - wake up threads blocked on a waitqueue.
3868 * @q: the waitqueue
3869 * @mode: which threads
3870 * @nr_exclusive: how many wake-one or wake-many threads to wake up
Martin Waitz67be2dd2005-05-01 08:59:26 -07003871 * @key: is directly passed to the wakeup function
David Howells50fa6102009-04-28 15:01:38 +01003872 *
3873 * It may be assumed that this function implies a write memory barrier before
3874 * changing the task state if and only if any tasks are woken up.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003875 */
Harvey Harrison7ad5b3a2008-02-08 04:19:53 -08003876void __wake_up(wait_queue_head_t *q, unsigned int mode,
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07003877 int nr_exclusive, void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003878{
3879 unsigned long flags;
3880
3881 spin_lock_irqsave(&q->lock, flags);
3882 __wake_up_common(q, mode, nr_exclusive, 0, key);
3883 spin_unlock_irqrestore(&q->lock, flags);
3884}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003885EXPORT_SYMBOL(__wake_up);
3886
3887/*
3888 * Same as __wake_up but called with the spinlock in wait_queue_head_t held.
3889 */
Harvey Harrison7ad5b3a2008-02-08 04:19:53 -08003890void __wake_up_locked(wait_queue_head_t *q, unsigned int mode)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003891{
3892 __wake_up_common(q, mode, 1, 0, NULL);
3893}
3894
Davide Libenzi4ede8162009-03-31 15:24:20 -07003895void __wake_up_locked_key(wait_queue_head_t *q, unsigned int mode, void *key)
3896{
3897 __wake_up_common(q, mode, 1, 0, key);
3898}
3899
Linus Torvalds1da177e2005-04-16 15:20:36 -07003900/**
Davide Libenzi4ede8162009-03-31 15:24:20 -07003901 * __wake_up_sync_key - wake up threads blocked on a waitqueue.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003902 * @q: the waitqueue
3903 * @mode: which threads
3904 * @nr_exclusive: how many wake-one or wake-many threads to wake up
Davide Libenzi4ede8162009-03-31 15:24:20 -07003905 * @key: opaque value to be passed to wakeup targets
Linus Torvalds1da177e2005-04-16 15:20:36 -07003906 *
3907 * The sync wakeup differs that the waker knows that it will schedule
3908 * away soon, so while the target thread will be woken up, it will not
3909 * be migrated to another CPU - ie. the two threads are 'synchronized'
3910 * with each other. This can prevent needless bouncing between CPUs.
3911 *
3912 * On UP it can prevent extra preemption.
David Howells50fa6102009-04-28 15:01:38 +01003913 *
3914 * It may be assumed that this function implies a write memory barrier before
3915 * changing the task state if and only if any tasks are woken up.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003916 */
Davide Libenzi4ede8162009-03-31 15:24:20 -07003917void __wake_up_sync_key(wait_queue_head_t *q, unsigned int mode,
3918 int nr_exclusive, void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003919{
3920 unsigned long flags;
Peter Zijlstra7d478722009-09-14 19:55:44 +02003921 int wake_flags = WF_SYNC;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003922
3923 if (unlikely(!q))
3924 return;
3925
3926 if (unlikely(!nr_exclusive))
Peter Zijlstra7d478722009-09-14 19:55:44 +02003927 wake_flags = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003928
3929 spin_lock_irqsave(&q->lock, flags);
Peter Zijlstra7d478722009-09-14 19:55:44 +02003930 __wake_up_common(q, mode, nr_exclusive, wake_flags, key);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003931 spin_unlock_irqrestore(&q->lock, flags);
3932}
Davide Libenzi4ede8162009-03-31 15:24:20 -07003933EXPORT_SYMBOL_GPL(__wake_up_sync_key);
3934
3935/*
3936 * __wake_up_sync - see __wake_up_sync_key()
3937 */
3938void __wake_up_sync(wait_queue_head_t *q, unsigned int mode, int nr_exclusive)
3939{
3940 __wake_up_sync_key(q, mode, nr_exclusive, NULL);
3941}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003942EXPORT_SYMBOL_GPL(__wake_up_sync); /* For internal use only */
3943
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02003944/**
3945 * complete: - signals a single thread waiting on this completion
3946 * @x: holds the state of this particular completion
3947 *
3948 * This will wake up a single thread waiting on this completion. Threads will be
3949 * awakened in the same order in which they were queued.
3950 *
3951 * See also complete_all(), wait_for_completion() and related routines.
David Howells50fa6102009-04-28 15:01:38 +01003952 *
3953 * It may be assumed that this function implies a write memory barrier before
3954 * changing the task state if and only if any tasks are woken up.
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02003955 */
Ingo Molnarb15136e2007-10-24 18:23:48 +02003956void complete(struct completion *x)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003957{
3958 unsigned long flags;
3959
3960 spin_lock_irqsave(&x->wait.lock, flags);
3961 x->done++;
Matthew Wilcoxd9514f62007-12-06 11:07:07 -05003962 __wake_up_common(&x->wait, TASK_NORMAL, 1, 0, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003963 spin_unlock_irqrestore(&x->wait.lock, flags);
3964}
3965EXPORT_SYMBOL(complete);
3966
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02003967/**
3968 * complete_all: - signals all threads waiting on this completion
3969 * @x: holds the state of this particular completion
3970 *
3971 * This will wake up all threads waiting on this particular completion event.
David Howells50fa6102009-04-28 15:01:38 +01003972 *
3973 * It may be assumed that this function implies a write memory barrier before
3974 * changing the task state if and only if any tasks are woken up.
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02003975 */
Ingo Molnarb15136e2007-10-24 18:23:48 +02003976void complete_all(struct completion *x)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003977{
3978 unsigned long flags;
3979
3980 spin_lock_irqsave(&x->wait.lock, flags);
3981 x->done += UINT_MAX/2;
Matthew Wilcoxd9514f62007-12-06 11:07:07 -05003982 __wake_up_common(&x->wait, TASK_NORMAL, 0, 0, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003983 spin_unlock_irqrestore(&x->wait.lock, flags);
3984}
3985EXPORT_SYMBOL(complete_all);
3986
Andi Kleen8cbbe862007-10-15 17:00:14 +02003987static inline long __sched
3988do_wait_for_common(struct completion *x, long timeout, int state)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003989{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003990 if (!x->done) {
3991 DECLARE_WAITQUEUE(wait, current);
3992
3993 wait.flags |= WQ_FLAG_EXCLUSIVE;
3994 __add_wait_queue_tail(&x->wait, &wait);
3995 do {
Oleg Nesterov94d3d822008-08-20 16:54:41 -07003996 if (signal_pending_state(state, current)) {
Oleg Nesterovea71a542008-06-20 18:32:20 +04003997 timeout = -ERESTARTSYS;
3998 break;
Andi Kleen8cbbe862007-10-15 17:00:14 +02003999 }
4000 __set_current_state(state);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004001 spin_unlock_irq(&x->wait.lock);
Andi Kleen8cbbe862007-10-15 17:00:14 +02004002 timeout = schedule_timeout(timeout);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004003 spin_lock_irq(&x->wait.lock);
Oleg Nesterovea71a542008-06-20 18:32:20 +04004004 } while (!x->done && timeout);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004005 __remove_wait_queue(&x->wait, &wait);
Oleg Nesterovea71a542008-06-20 18:32:20 +04004006 if (!x->done)
4007 return timeout;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004008 }
4009 x->done--;
Oleg Nesterovea71a542008-06-20 18:32:20 +04004010 return timeout ?: 1;
Andi Kleen8cbbe862007-10-15 17:00:14 +02004011}
4012
4013static long __sched
4014wait_for_common(struct completion *x, long timeout, int state)
4015{
4016 might_sleep();
4017
4018 spin_lock_irq(&x->wait.lock);
4019 timeout = do_wait_for_common(x, timeout, state);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004020 spin_unlock_irq(&x->wait.lock);
Andi Kleen8cbbe862007-10-15 17:00:14 +02004021 return timeout;
4022}
4023
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02004024/**
4025 * wait_for_completion: - waits for completion of a task
4026 * @x: holds the state of this particular completion
4027 *
4028 * This waits to be signaled for completion of a specific task. It is NOT
4029 * interruptible and there is no timeout.
4030 *
4031 * See also similar routines (i.e. wait_for_completion_timeout()) with timeout
4032 * and interrupt capability. Also see complete().
4033 */
Ingo Molnarb15136e2007-10-24 18:23:48 +02004034void __sched wait_for_completion(struct completion *x)
Andi Kleen8cbbe862007-10-15 17:00:14 +02004035{
4036 wait_for_common(x, MAX_SCHEDULE_TIMEOUT, TASK_UNINTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004037}
4038EXPORT_SYMBOL(wait_for_completion);
4039
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02004040/**
4041 * wait_for_completion_timeout: - waits for completion of a task (w/timeout)
4042 * @x: holds the state of this particular completion
4043 * @timeout: timeout value in jiffies
4044 *
4045 * This waits for either a completion of a specific task to be signaled or for a
4046 * specified timeout to expire. The timeout is in jiffies. It is not
4047 * interruptible.
4048 */
Ingo Molnarb15136e2007-10-24 18:23:48 +02004049unsigned long __sched
Linus Torvalds1da177e2005-04-16 15:20:36 -07004050wait_for_completion_timeout(struct completion *x, unsigned long timeout)
4051{
Andi Kleen8cbbe862007-10-15 17:00:14 +02004052 return wait_for_common(x, timeout, TASK_UNINTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004053}
4054EXPORT_SYMBOL(wait_for_completion_timeout);
4055
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02004056/**
4057 * wait_for_completion_interruptible: - waits for completion of a task (w/intr)
4058 * @x: holds the state of this particular completion
4059 *
4060 * This waits for completion of a specific task to be signaled. It is
4061 * interruptible.
4062 */
Andi Kleen8cbbe862007-10-15 17:00:14 +02004063int __sched wait_for_completion_interruptible(struct completion *x)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004064{
Andi Kleen51e97992007-10-18 21:32:55 +02004065 long t = wait_for_common(x, MAX_SCHEDULE_TIMEOUT, TASK_INTERRUPTIBLE);
4066 if (t == -ERESTARTSYS)
4067 return t;
4068 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004069}
4070EXPORT_SYMBOL(wait_for_completion_interruptible);
4071
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02004072/**
4073 * wait_for_completion_interruptible_timeout: - waits for completion (w/(to,intr))
4074 * @x: holds the state of this particular completion
4075 * @timeout: timeout value in jiffies
4076 *
4077 * This waits for either a completion of a specific task to be signaled or for a
4078 * specified timeout to expire. It is interruptible. The timeout is in jiffies.
4079 */
Ingo Molnarb15136e2007-10-24 18:23:48 +02004080unsigned long __sched
Linus Torvalds1da177e2005-04-16 15:20:36 -07004081wait_for_completion_interruptible_timeout(struct completion *x,
4082 unsigned long timeout)
4083{
Andi Kleen8cbbe862007-10-15 17:00:14 +02004084 return wait_for_common(x, timeout, TASK_INTERRUPTIBLE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004085}
4086EXPORT_SYMBOL(wait_for_completion_interruptible_timeout);
4087
Kevin Diggs65eb3dc2008-08-26 10:26:54 +02004088/**
4089 * wait_for_completion_killable: - waits for completion of a task (killable)
4090 * @x: holds the state of this particular completion
4091 *
4092 * This waits to be signaled for completion of a specific task. It can be
4093 * interrupted by a kill signal.
4094 */
Matthew Wilcox009e5772007-12-06 12:29:54 -05004095int __sched wait_for_completion_killable(struct completion *x)
4096{
4097 long t = wait_for_common(x, MAX_SCHEDULE_TIMEOUT, TASK_KILLABLE);
4098 if (t == -ERESTARTSYS)
4099 return t;
4100 return 0;
4101}
4102EXPORT_SYMBOL(wait_for_completion_killable);
4103
Dave Chinnerbe4de352008-08-15 00:40:44 -07004104/**
4105 * try_wait_for_completion - try to decrement a completion without blocking
4106 * @x: completion structure
4107 *
4108 * Returns: 0 if a decrement cannot be done without blocking
4109 * 1 if a decrement succeeded.
4110 *
4111 * If a completion is being used as a counting completion,
4112 * attempt to decrement the counter without blocking. This
4113 * enables us to avoid waiting if the resource the completion
4114 * is protecting is not available.
4115 */
4116bool try_wait_for_completion(struct completion *x)
4117{
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004118 unsigned long flags;
Dave Chinnerbe4de352008-08-15 00:40:44 -07004119 int ret = 1;
4120
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004121 spin_lock_irqsave(&x->wait.lock, flags);
Dave Chinnerbe4de352008-08-15 00:40:44 -07004122 if (!x->done)
4123 ret = 0;
4124 else
4125 x->done--;
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004126 spin_unlock_irqrestore(&x->wait.lock, flags);
Dave Chinnerbe4de352008-08-15 00:40:44 -07004127 return ret;
4128}
4129EXPORT_SYMBOL(try_wait_for_completion);
4130
4131/**
4132 * completion_done - Test to see if a completion has any waiters
4133 * @x: completion structure
4134 *
4135 * Returns: 0 if there are waiters (wait_for_completion() in progress)
4136 * 1 if there are no waiters.
4137 *
4138 */
4139bool completion_done(struct completion *x)
4140{
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004141 unsigned long flags;
Dave Chinnerbe4de352008-08-15 00:40:44 -07004142 int ret = 1;
4143
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004144 spin_lock_irqsave(&x->wait.lock, flags);
Dave Chinnerbe4de352008-08-15 00:40:44 -07004145 if (!x->done)
4146 ret = 0;
Rafael J. Wysocki7539a3b2009-12-13 00:07:30 +01004147 spin_unlock_irqrestore(&x->wait.lock, flags);
Dave Chinnerbe4de352008-08-15 00:40:44 -07004148 return ret;
4149}
4150EXPORT_SYMBOL(completion_done);
4151
Andi Kleen8cbbe862007-10-15 17:00:14 +02004152static long __sched
4153sleep_on_common(wait_queue_head_t *q, int state, long timeout)
Ingo Molnar0fec1712007-07-09 18:52:01 +02004154{
4155 unsigned long flags;
4156 wait_queue_t wait;
4157
4158 init_waitqueue_entry(&wait, current);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004159
Andi Kleen8cbbe862007-10-15 17:00:14 +02004160 __set_current_state(state);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004161
Andi Kleen8cbbe862007-10-15 17:00:14 +02004162 spin_lock_irqsave(&q->lock, flags);
4163 __add_wait_queue(q, &wait);
4164 spin_unlock(&q->lock);
4165 timeout = schedule_timeout(timeout);
4166 spin_lock_irq(&q->lock);
4167 __remove_wait_queue(q, &wait);
4168 spin_unlock_irqrestore(&q->lock, flags);
4169
4170 return timeout;
4171}
4172
4173void __sched interruptible_sleep_on(wait_queue_head_t *q)
4174{
4175 sleep_on_common(q, TASK_INTERRUPTIBLE, MAX_SCHEDULE_TIMEOUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004176}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004177EXPORT_SYMBOL(interruptible_sleep_on);
4178
Ingo Molnar0fec1712007-07-09 18:52:01 +02004179long __sched
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004180interruptible_sleep_on_timeout(wait_queue_head_t *q, long timeout)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004181{
Andi Kleen8cbbe862007-10-15 17:00:14 +02004182 return sleep_on_common(q, TASK_INTERRUPTIBLE, timeout);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004183}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004184EXPORT_SYMBOL(interruptible_sleep_on_timeout);
4185
Ingo Molnar0fec1712007-07-09 18:52:01 +02004186void __sched sleep_on(wait_queue_head_t *q)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004187{
Andi Kleen8cbbe862007-10-15 17:00:14 +02004188 sleep_on_common(q, TASK_UNINTERRUPTIBLE, MAX_SCHEDULE_TIMEOUT);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004189}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004190EXPORT_SYMBOL(sleep_on);
4191
Ingo Molnar0fec1712007-07-09 18:52:01 +02004192long __sched sleep_on_timeout(wait_queue_head_t *q, long timeout)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004193{
Andi Kleen8cbbe862007-10-15 17:00:14 +02004194 return sleep_on_common(q, TASK_UNINTERRUPTIBLE, timeout);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004195}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004196EXPORT_SYMBOL(sleep_on_timeout);
4197
Ingo Molnarb29739f2006-06-27 02:54:51 -07004198#ifdef CONFIG_RT_MUTEXES
4199
4200/*
4201 * rt_mutex_setprio - set the current priority of a task
4202 * @p: task
4203 * @prio: prio value (kernel-internal form)
4204 *
4205 * This function changes the 'effective' priority of a task. It does
4206 * not touch ->normal_prio like __setscheduler().
4207 *
4208 * Used by the rt_mutex code to implement priority inheritance logic.
4209 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004210void rt_mutex_setprio(struct task_struct *p, int prio)
Ingo Molnarb29739f2006-06-27 02:54:51 -07004211{
4212 unsigned long flags;
Srivatsa Vaddagiri83b699e2007-10-15 17:00:08 +02004213 int oldprio, on_rq, running;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004214 struct rq *rq;
Steven Rostedtcb469842008-01-25 21:08:22 +01004215 const struct sched_class *prev_class = p->sched_class;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004216
4217 BUG_ON(prio < 0 || prio > MAX_PRIO);
4218
4219 rq = task_rq_lock(p, &flags);
Ingo Molnara8e504d2007-08-09 11:16:47 +02004220 update_rq_clock(rq);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004221
Andrew Mortond5f9f942007-05-08 20:27:06 -07004222 oldprio = p->prio;
Ingo Molnardd41f592007-07-09 18:51:59 +02004223 on_rq = p->se.on_rq;
Dmitry Adamushko051a1d12007-12-18 15:21:13 +01004224 running = task_current(rq, p);
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004225 if (on_rq)
Ingo Molnar69be72c2007-08-09 11:16:49 +02004226 dequeue_task(rq, p, 0);
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004227 if (running)
4228 p->sched_class->put_prev_task(rq, p);
Ingo Molnardd41f592007-07-09 18:51:59 +02004229
4230 if (rt_prio(prio))
4231 p->sched_class = &rt_sched_class;
4232 else
4233 p->sched_class = &fair_sched_class;
4234
Ingo Molnarb29739f2006-06-27 02:54:51 -07004235 p->prio = prio;
4236
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004237 if (running)
4238 p->sched_class->set_curr_task(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02004239 if (on_rq) {
Thomas Gleixner60db48c2010-01-20 20:59:06 +00004240 enqueue_task(rq, p, 0, oldprio < prio);
Steven Rostedtcb469842008-01-25 21:08:22 +01004241
4242 check_class_changed(rq, p, prev_class, oldprio, running);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004243 }
4244 task_rq_unlock(rq, &flags);
4245}
4246
4247#endif
4248
Ingo Molnar36c8b582006-07-03 00:25:41 -07004249void set_user_nice(struct task_struct *p, long nice)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004250{
Ingo Molnardd41f592007-07-09 18:51:59 +02004251 int old_prio, delta, on_rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004252 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004253 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004254
4255 if (TASK_NICE(p) == nice || nice < -20 || nice > 19)
4256 return;
4257 /*
4258 * We have to be careful, if called from sys_setpriority(),
4259 * the task might be in the middle of scheduling on another CPU.
4260 */
4261 rq = task_rq_lock(p, &flags);
Ingo Molnara8e504d2007-08-09 11:16:47 +02004262 update_rq_clock(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004263 /*
4264 * The RT priorities are set via sched_setscheduler(), but we still
4265 * allow the 'normal' nice value to be set - but as expected
4266 * it wont have any effect on scheduling until the task is
Ingo Molnardd41f592007-07-09 18:51:59 +02004267 * SCHED_FIFO/SCHED_RR:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004268 */
Ingo Molnare05606d2007-07-09 18:51:59 +02004269 if (task_has_rt_policy(p)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004270 p->static_prio = NICE_TO_PRIO(nice);
4271 goto out_unlock;
4272 }
Ingo Molnardd41f592007-07-09 18:51:59 +02004273 on_rq = p->se.on_rq;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02004274 if (on_rq)
Ingo Molnar69be72c2007-08-09 11:16:49 +02004275 dequeue_task(rq, p, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004276
Linus Torvalds1da177e2005-04-16 15:20:36 -07004277 p->static_prio = NICE_TO_PRIO(nice);
Peter Williams2dd73a42006-06-27 02:54:34 -07004278 set_load_weight(p);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004279 old_prio = p->prio;
4280 p->prio = effective_prio(p);
4281 delta = p->prio - old_prio;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004282
Ingo Molnardd41f592007-07-09 18:51:59 +02004283 if (on_rq) {
Thomas Gleixnerea87bb72010-01-20 20:58:57 +00004284 enqueue_task(rq, p, 0, false);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004285 /*
Andrew Mortond5f9f942007-05-08 20:27:06 -07004286 * If the task increased its priority or is running and
4287 * lowered its priority, then reschedule its CPU:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004288 */
Andrew Mortond5f9f942007-05-08 20:27:06 -07004289 if (delta < 0 || (delta > 0 && task_running(rq, p)))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004290 resched_task(rq->curr);
4291 }
4292out_unlock:
4293 task_rq_unlock(rq, &flags);
4294}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004295EXPORT_SYMBOL(set_user_nice);
4296
Matt Mackalle43379f2005-05-01 08:59:00 -07004297/*
4298 * can_nice - check if a task can reduce its nice value
4299 * @p: task
4300 * @nice: nice value
4301 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004302int can_nice(const struct task_struct *p, const int nice)
Matt Mackalle43379f2005-05-01 08:59:00 -07004303{
Matt Mackall024f4742005-08-18 11:24:19 -07004304 /* convert nice value [19,-20] to rlimit style value [1,40] */
4305 int nice_rlim = 20 - nice;
Ingo Molnar48f24c42006-07-03 00:25:40 -07004306
Matt Mackalle43379f2005-05-01 08:59:00 -07004307 return (nice_rlim <= p->signal->rlim[RLIMIT_NICE].rlim_cur ||
4308 capable(CAP_SYS_NICE));
4309}
4310
Linus Torvalds1da177e2005-04-16 15:20:36 -07004311#ifdef __ARCH_WANT_SYS_NICE
4312
4313/*
4314 * sys_nice - change the priority of the current process.
4315 * @increment: priority increment
4316 *
4317 * sys_setpriority is a more generic, but much slower function that
4318 * does similar things.
4319 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004320SYSCALL_DEFINE1(nice, int, increment)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004321{
Ingo Molnar48f24c42006-07-03 00:25:40 -07004322 long nice, retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004323
4324 /*
4325 * Setpriority might change our priority at the same moment.
4326 * We don't have to worry. Conceptually one call occurs first
4327 * and we have a single winner.
4328 */
Matt Mackalle43379f2005-05-01 08:59:00 -07004329 if (increment < -40)
4330 increment = -40;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004331 if (increment > 40)
4332 increment = 40;
4333
Américo Wang2b8f8362009-02-16 18:54:21 +08004334 nice = TASK_NICE(current) + increment;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004335 if (nice < -20)
4336 nice = -20;
4337 if (nice > 19)
4338 nice = 19;
4339
Matt Mackalle43379f2005-05-01 08:59:00 -07004340 if (increment < 0 && !can_nice(current, nice))
4341 return -EPERM;
4342
Linus Torvalds1da177e2005-04-16 15:20:36 -07004343 retval = security_task_setnice(current, nice);
4344 if (retval)
4345 return retval;
4346
4347 set_user_nice(current, nice);
4348 return 0;
4349}
4350
4351#endif
4352
4353/**
4354 * task_prio - return the priority value of a given task.
4355 * @p: the task in question.
4356 *
4357 * This is the priority value as seen by users in /proc.
4358 * RT tasks are offset by -200. Normal tasks are centered
4359 * around 0, value goes from -16 to +15.
4360 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004361int task_prio(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004362{
4363 return p->prio - MAX_RT_PRIO;
4364}
4365
4366/**
4367 * task_nice - return the nice value of a given task.
4368 * @p: the task in question.
4369 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004370int task_nice(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004371{
4372 return TASK_NICE(p);
4373}
Pavel Roskin150d8be2008-03-05 16:56:37 -05004374EXPORT_SYMBOL(task_nice);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004375
4376/**
4377 * idle_cpu - is a given cpu idle currently?
4378 * @cpu: the processor in question.
4379 */
4380int idle_cpu(int cpu)
4381{
4382 return cpu_curr(cpu) == cpu_rq(cpu)->idle;
4383}
4384
Linus Torvalds1da177e2005-04-16 15:20:36 -07004385/**
4386 * idle_task - return the idle task for a given cpu.
4387 * @cpu: the processor in question.
4388 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004389struct task_struct *idle_task(int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004390{
4391 return cpu_rq(cpu)->idle;
4392}
4393
4394/**
4395 * find_process_by_pid - find a process with a matching PID value.
4396 * @pid: the pid in question.
4397 */
Alexey Dobriyana9957442007-10-15 17:00:13 +02004398static struct task_struct *find_process_by_pid(pid_t pid)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004399{
Pavel Emelyanov228ebcb2007-10-18 23:40:16 -07004400 return pid ? find_task_by_vpid(pid) : current;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004401}
4402
4403/* Actually do priority change: must hold rq lock. */
Ingo Molnardd41f592007-07-09 18:51:59 +02004404static void
4405__setscheduler(struct rq *rq, struct task_struct *p, int policy, int prio)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004406{
Ingo Molnardd41f592007-07-09 18:51:59 +02004407 BUG_ON(p->se.on_rq);
Ingo Molnar48f24c42006-07-03 00:25:40 -07004408
Linus Torvalds1da177e2005-04-16 15:20:36 -07004409 p->policy = policy;
4410 p->rt_priority = prio;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004411 p->normal_prio = normal_prio(p);
4412 /* we are holding p->pi_lock already */
4413 p->prio = rt_mutex_getprio(p);
Peter Zijlstraffd44db2009-11-10 20:12:01 +01004414 if (rt_prio(p->prio))
4415 p->sched_class = &rt_sched_class;
4416 else
4417 p->sched_class = &fair_sched_class;
Peter Williams2dd73a42006-06-27 02:54:34 -07004418 set_load_weight(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004419}
4420
David Howellsc69e8d92008-11-14 10:39:19 +11004421/*
4422 * check the target process has a UID that matches the current process's
4423 */
4424static bool check_same_owner(struct task_struct *p)
4425{
4426 const struct cred *cred = current_cred(), *pcred;
4427 bool match;
4428
4429 rcu_read_lock();
4430 pcred = __task_cred(p);
4431 match = (cred->euid == pcred->euid ||
4432 cred->euid == pcred->uid);
4433 rcu_read_unlock();
4434 return match;
4435}
4436
Rusty Russell961ccdd2008-06-23 13:55:38 +10004437static int __sched_setscheduler(struct task_struct *p, int policy,
4438 struct sched_param *param, bool user)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004439{
Srivatsa Vaddagiri83b699e2007-10-15 17:00:08 +02004440 int retval, oldprio, oldpolicy = -1, on_rq, running;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004441 unsigned long flags;
Steven Rostedtcb469842008-01-25 21:08:22 +01004442 const struct sched_class *prev_class = p->sched_class;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004443 struct rq *rq;
Lennart Poetteringca94c442009-06-15 17:17:47 +02004444 int reset_on_fork;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004445
Steven Rostedt66e53932006-06-27 02:54:44 -07004446 /* may grab non-irq protected spin_locks */
4447 BUG_ON(in_interrupt());
Linus Torvalds1da177e2005-04-16 15:20:36 -07004448recheck:
4449 /* double check policy once rq lock held */
Lennart Poetteringca94c442009-06-15 17:17:47 +02004450 if (policy < 0) {
4451 reset_on_fork = p->sched_reset_on_fork;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004452 policy = oldpolicy = p->policy;
Lennart Poetteringca94c442009-06-15 17:17:47 +02004453 } else {
4454 reset_on_fork = !!(policy & SCHED_RESET_ON_FORK);
4455 policy &= ~SCHED_RESET_ON_FORK;
4456
4457 if (policy != SCHED_FIFO && policy != SCHED_RR &&
4458 policy != SCHED_NORMAL && policy != SCHED_BATCH &&
4459 policy != SCHED_IDLE)
4460 return -EINVAL;
4461 }
4462
Linus Torvalds1da177e2005-04-16 15:20:36 -07004463 /*
4464 * Valid priorities for SCHED_FIFO and SCHED_RR are
Ingo Molnardd41f592007-07-09 18:51:59 +02004465 * 1..MAX_USER_RT_PRIO-1, valid priority for SCHED_NORMAL,
4466 * SCHED_BATCH and SCHED_IDLE is 0.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004467 */
4468 if (param->sched_priority < 0 ||
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004469 (p->mm && param->sched_priority > MAX_USER_RT_PRIO-1) ||
Steven Rostedtd46523e2005-07-25 16:28:39 -04004470 (!p->mm && param->sched_priority > MAX_RT_PRIO-1))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004471 return -EINVAL;
Ingo Molnare05606d2007-07-09 18:51:59 +02004472 if (rt_policy(policy) != (param->sched_priority != 0))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004473 return -EINVAL;
4474
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004475 /*
4476 * Allow unprivileged RT tasks to decrease priority:
4477 */
Rusty Russell961ccdd2008-06-23 13:55:38 +10004478 if (user && !capable(CAP_SYS_NICE)) {
Ingo Molnare05606d2007-07-09 18:51:59 +02004479 if (rt_policy(policy)) {
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004480 unsigned long rlim_rtprio;
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004481
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004482 if (!lock_task_sighand(p, &flags))
4483 return -ESRCH;
4484 rlim_rtprio = p->signal->rlim[RLIMIT_RTPRIO].rlim_cur;
4485 unlock_task_sighand(p, &flags);
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004486
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004487 /* can't set/change the rt policy */
4488 if (policy != p->policy && !rlim_rtprio)
4489 return -EPERM;
4490
4491 /* can't increase priority */
4492 if (param->sched_priority > p->rt_priority &&
4493 param->sched_priority > rlim_rtprio)
4494 return -EPERM;
4495 }
Ingo Molnardd41f592007-07-09 18:51:59 +02004496 /*
4497 * Like positive nice levels, dont allow tasks to
4498 * move out of SCHED_IDLE either:
4499 */
4500 if (p->policy == SCHED_IDLE && policy != SCHED_IDLE)
4501 return -EPERM;
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004502
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004503 /* can't change other user's priorities */
David Howellsc69e8d92008-11-14 10:39:19 +11004504 if (!check_same_owner(p))
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004505 return -EPERM;
Lennart Poetteringca94c442009-06-15 17:17:47 +02004506
4507 /* Normal users shall not reset the sched_reset_on_fork flag */
4508 if (p->sched_reset_on_fork && !reset_on_fork)
4509 return -EPERM;
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004510 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004511
Jeremy Fitzhardinge725aad22008-08-03 09:33:03 -07004512 if (user) {
Peter Zijlstrab68aa232008-02-13 15:45:40 +01004513#ifdef CONFIG_RT_GROUP_SCHED
Jeremy Fitzhardinge725aad22008-08-03 09:33:03 -07004514 /*
4515 * Do not allow realtime tasks into groups that have no runtime
4516 * assigned.
4517 */
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02004518 if (rt_bandwidth_enabled() && rt_policy(policy) &&
4519 task_group(p)->rt_bandwidth.rt_runtime == 0)
Jeremy Fitzhardinge725aad22008-08-03 09:33:03 -07004520 return -EPERM;
Peter Zijlstrab68aa232008-02-13 15:45:40 +01004521#endif
4522
Jeremy Fitzhardinge725aad22008-08-03 09:33:03 -07004523 retval = security_task_setscheduler(p, policy, param);
4524 if (retval)
4525 return retval;
4526 }
4527
Linus Torvalds1da177e2005-04-16 15:20:36 -07004528 /*
Ingo Molnarb29739f2006-06-27 02:54:51 -07004529 * make sure no PI-waiters arrive (or leave) while we are
4530 * changing the priority of the task:
4531 */
Thomas Gleixner1d615482009-11-17 14:54:03 +01004532 raw_spin_lock_irqsave(&p->pi_lock, flags);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004533 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -07004534 * To be able to change p->policy safely, the apropriate
4535 * runqueue lock must be held.
4536 */
Ingo Molnarb29739f2006-06-27 02:54:51 -07004537 rq = __task_rq_lock(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004538 /* recheck policy now with rq lock held */
4539 if (unlikely(oldpolicy != -1 && oldpolicy != p->policy)) {
4540 policy = oldpolicy = -1;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004541 __task_rq_unlock(rq);
Thomas Gleixner1d615482009-11-17 14:54:03 +01004542 raw_spin_unlock_irqrestore(&p->pi_lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004543 goto recheck;
4544 }
Ingo Molnar2daa3572007-08-09 11:16:51 +02004545 update_rq_clock(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02004546 on_rq = p->se.on_rq;
Dmitry Adamushko051a1d12007-12-18 15:21:13 +01004547 running = task_current(rq, p);
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004548 if (on_rq)
Ingo Molnar2e1cb742007-08-09 11:16:49 +02004549 deactivate_task(rq, p, 0);
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004550 if (running)
4551 p->sched_class->put_prev_task(rq, p);
Dmitry Adamushkof6b532052007-10-15 17:00:08 +02004552
Lennart Poetteringca94c442009-06-15 17:17:47 +02004553 p->sched_reset_on_fork = reset_on_fork;
4554
Linus Torvalds1da177e2005-04-16 15:20:36 -07004555 oldprio = p->prio;
Ingo Molnardd41f592007-07-09 18:51:59 +02004556 __setscheduler(rq, p, policy, param->sched_priority);
Dmitry Adamushkof6b532052007-10-15 17:00:08 +02004557
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07004558 if (running)
4559 p->sched_class->set_curr_task(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02004560 if (on_rq) {
4561 activate_task(rq, p, 0);
Steven Rostedtcb469842008-01-25 21:08:22 +01004562
4563 check_class_changed(rq, p, prev_class, oldprio, running);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004564 }
Ingo Molnarb29739f2006-06-27 02:54:51 -07004565 __task_rq_unlock(rq);
Thomas Gleixner1d615482009-11-17 14:54:03 +01004566 raw_spin_unlock_irqrestore(&p->pi_lock, flags);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004567
Thomas Gleixner95e02ca2006-06-27 02:55:02 -07004568 rt_mutex_adjust_pi(p);
4569
Linus Torvalds1da177e2005-04-16 15:20:36 -07004570 return 0;
4571}
Rusty Russell961ccdd2008-06-23 13:55:38 +10004572
4573/**
4574 * sched_setscheduler - change the scheduling policy and/or RT priority of a thread.
4575 * @p: the task in question.
4576 * @policy: new policy.
4577 * @param: structure containing the new RT priority.
4578 *
4579 * NOTE that the task may be already dead.
4580 */
4581int sched_setscheduler(struct task_struct *p, int policy,
4582 struct sched_param *param)
4583{
4584 return __sched_setscheduler(p, policy, param, true);
4585}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004586EXPORT_SYMBOL_GPL(sched_setscheduler);
4587
Rusty Russell961ccdd2008-06-23 13:55:38 +10004588/**
4589 * sched_setscheduler_nocheck - change the scheduling policy and/or RT priority of a thread from kernelspace.
4590 * @p: the task in question.
4591 * @policy: new policy.
4592 * @param: structure containing the new RT priority.
4593 *
4594 * Just like sched_setscheduler, only don't bother checking if the
4595 * current context has permission. For example, this is needed in
4596 * stop_machine(): we create temporary high priority worker threads,
4597 * but our caller might not have that capability.
4598 */
4599int sched_setscheduler_nocheck(struct task_struct *p, int policy,
4600 struct sched_param *param)
4601{
4602 return __sched_setscheduler(p, policy, param, false);
4603}
4604
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004605static int
4606do_sched_setscheduler(pid_t pid, int policy, struct sched_param __user *param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004607{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004608 struct sched_param lparam;
4609 struct task_struct *p;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004610 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004611
4612 if (!param || pid < 0)
4613 return -EINVAL;
4614 if (copy_from_user(&lparam, param, sizeof(struct sched_param)))
4615 return -EFAULT;
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004616
4617 rcu_read_lock();
4618 retval = -ESRCH;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004619 p = find_process_by_pid(pid);
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004620 if (p != NULL)
4621 retval = sched_setscheduler(p, policy, &lparam);
4622 rcu_read_unlock();
Ingo Molnar36c8b582006-07-03 00:25:41 -07004623
Linus Torvalds1da177e2005-04-16 15:20:36 -07004624 return retval;
4625}
4626
4627/**
4628 * sys_sched_setscheduler - set/change the scheduler policy and RT priority
4629 * @pid: the pid in question.
4630 * @policy: new policy.
4631 * @param: structure containing the new RT priority.
4632 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004633SYSCALL_DEFINE3(sched_setscheduler, pid_t, pid, int, policy,
4634 struct sched_param __user *, param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004635{
Jason Baronc21761f2006-01-18 17:43:03 -08004636 /* negative values for policy are not valid */
4637 if (policy < 0)
4638 return -EINVAL;
4639
Linus Torvalds1da177e2005-04-16 15:20:36 -07004640 return do_sched_setscheduler(pid, policy, param);
4641}
4642
4643/**
4644 * sys_sched_setparam - set/change the RT priority of a thread
4645 * @pid: the pid in question.
4646 * @param: structure containing the new RT priority.
4647 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004648SYSCALL_DEFINE2(sched_setparam, pid_t, pid, struct sched_param __user *, param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004649{
4650 return do_sched_setscheduler(pid, -1, param);
4651}
4652
4653/**
4654 * sys_sched_getscheduler - get the policy (scheduling class) of a thread
4655 * @pid: the pid in question.
4656 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004657SYSCALL_DEFINE1(sched_getscheduler, pid_t, pid)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004658{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004659 struct task_struct *p;
Andi Kleen3a5c3592007-10-15 17:00:14 +02004660 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004661
4662 if (pid < 0)
Andi Kleen3a5c3592007-10-15 17:00:14 +02004663 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004664
4665 retval = -ESRCH;
Thomas Gleixner5fe85be2009-12-09 10:14:58 +00004666 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004667 p = find_process_by_pid(pid);
4668 if (p) {
4669 retval = security_task_getscheduler(p);
4670 if (!retval)
Lennart Poetteringca94c442009-06-15 17:17:47 +02004671 retval = p->policy
4672 | (p->sched_reset_on_fork ? SCHED_RESET_ON_FORK : 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004673 }
Thomas Gleixner5fe85be2009-12-09 10:14:58 +00004674 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004675 return retval;
4676}
4677
4678/**
Lennart Poetteringca94c442009-06-15 17:17:47 +02004679 * sys_sched_getparam - get the RT priority of a thread
Linus Torvalds1da177e2005-04-16 15:20:36 -07004680 * @pid: the pid in question.
4681 * @param: structure containing the RT priority.
4682 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004683SYSCALL_DEFINE2(sched_getparam, pid_t, pid, struct sched_param __user *, param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004684{
4685 struct sched_param lp;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004686 struct task_struct *p;
Andi Kleen3a5c3592007-10-15 17:00:14 +02004687 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004688
4689 if (!param || pid < 0)
Andi Kleen3a5c3592007-10-15 17:00:14 +02004690 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004691
Thomas Gleixner5fe85be2009-12-09 10:14:58 +00004692 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004693 p = find_process_by_pid(pid);
4694 retval = -ESRCH;
4695 if (!p)
4696 goto out_unlock;
4697
4698 retval = security_task_getscheduler(p);
4699 if (retval)
4700 goto out_unlock;
4701
4702 lp.sched_priority = p->rt_priority;
Thomas Gleixner5fe85be2009-12-09 10:14:58 +00004703 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004704
4705 /*
4706 * This one might sleep, we cannot do it with a spinlock held ...
4707 */
4708 retval = copy_to_user(param, &lp, sizeof(*param)) ? -EFAULT : 0;
4709
Linus Torvalds1da177e2005-04-16 15:20:36 -07004710 return retval;
4711
4712out_unlock:
Thomas Gleixner5fe85be2009-12-09 10:14:58 +00004713 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004714 return retval;
4715}
4716
Rusty Russell96f874e22008-11-25 02:35:14 +10304717long sched_setaffinity(pid_t pid, const struct cpumask *in_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004718{
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304719 cpumask_var_t cpus_allowed, new_mask;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004720 struct task_struct *p;
4721 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004722
Gautham R Shenoy95402b32008-01-25 21:08:02 +01004723 get_online_cpus();
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004724 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004725
4726 p = find_process_by_pid(pid);
4727 if (!p) {
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004728 rcu_read_unlock();
Gautham R Shenoy95402b32008-01-25 21:08:02 +01004729 put_online_cpus();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004730 return -ESRCH;
4731 }
4732
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004733 /* Prevent p going away */
Linus Torvalds1da177e2005-04-16 15:20:36 -07004734 get_task_struct(p);
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004735 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004736
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304737 if (!alloc_cpumask_var(&cpus_allowed, GFP_KERNEL)) {
4738 retval = -ENOMEM;
4739 goto out_put_task;
4740 }
4741 if (!alloc_cpumask_var(&new_mask, GFP_KERNEL)) {
4742 retval = -ENOMEM;
4743 goto out_free_cpus_allowed;
4744 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004745 retval = -EPERM;
David Howellsc69e8d92008-11-14 10:39:19 +11004746 if (!check_same_owner(p) && !capable(CAP_SYS_NICE))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004747 goto out_unlock;
4748
David Quigleye7834f82006-06-23 02:03:59 -07004749 retval = security_task_setscheduler(p, 0, NULL);
4750 if (retval)
4751 goto out_unlock;
4752
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304753 cpuset_cpus_allowed(p, cpus_allowed);
4754 cpumask_and(new_mask, in_mask, cpus_allowed);
Paul Menage8707d8b2007-10-18 23:40:22 -07004755 again:
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304756 retval = set_cpus_allowed_ptr(p, new_mask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004757
Paul Menage8707d8b2007-10-18 23:40:22 -07004758 if (!retval) {
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304759 cpuset_cpus_allowed(p, cpus_allowed);
4760 if (!cpumask_subset(new_mask, cpus_allowed)) {
Paul Menage8707d8b2007-10-18 23:40:22 -07004761 /*
4762 * We must have raced with a concurrent cpuset
4763 * update. Just reset the cpus_allowed to the
4764 * cpuset's cpus_allowed
4765 */
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304766 cpumask_copy(new_mask, cpus_allowed);
Paul Menage8707d8b2007-10-18 23:40:22 -07004767 goto again;
4768 }
4769 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004770out_unlock:
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304771 free_cpumask_var(new_mask);
4772out_free_cpus_allowed:
4773 free_cpumask_var(cpus_allowed);
4774out_put_task:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004775 put_task_struct(p);
Gautham R Shenoy95402b32008-01-25 21:08:02 +01004776 put_online_cpus();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004777 return retval;
4778}
4779
4780static int get_user_cpu_mask(unsigned long __user *user_mask_ptr, unsigned len,
Rusty Russell96f874e22008-11-25 02:35:14 +10304781 struct cpumask *new_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004782{
Rusty Russell96f874e22008-11-25 02:35:14 +10304783 if (len < cpumask_size())
4784 cpumask_clear(new_mask);
4785 else if (len > cpumask_size())
4786 len = cpumask_size();
4787
Linus Torvalds1da177e2005-04-16 15:20:36 -07004788 return copy_from_user(new_mask, user_mask_ptr, len) ? -EFAULT : 0;
4789}
4790
4791/**
4792 * sys_sched_setaffinity - set the cpu affinity of a process
4793 * @pid: pid of the process
4794 * @len: length in bytes of the bitmask pointed to by user_mask_ptr
4795 * @user_mask_ptr: user-space pointer to the new cpu mask
4796 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004797SYSCALL_DEFINE3(sched_setaffinity, pid_t, pid, unsigned int, len,
4798 unsigned long __user *, user_mask_ptr)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004799{
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304800 cpumask_var_t new_mask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004801 int retval;
4802
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304803 if (!alloc_cpumask_var(&new_mask, GFP_KERNEL))
4804 return -ENOMEM;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004805
Rusty Russell5a16f3d2008-11-25 02:35:11 +10304806 retval = get_user_cpu_mask(user_mask_ptr, len, new_mask);
4807 if (retval == 0)
4808 retval = sched_setaffinity(pid, new_mask);
4809 free_cpumask_var(new_mask);
4810 return retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004811}
4812
Rusty Russell96f874e22008-11-25 02:35:14 +10304813long sched_getaffinity(pid_t pid, struct cpumask *mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004814{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004815 struct task_struct *p;
Thomas Gleixner31605682009-12-08 20:24:16 +00004816 unsigned long flags;
4817 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004818 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004819
Gautham R Shenoy95402b32008-01-25 21:08:02 +01004820 get_online_cpus();
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004821 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004822
4823 retval = -ESRCH;
4824 p = find_process_by_pid(pid);
4825 if (!p)
4826 goto out_unlock;
4827
David Quigleye7834f82006-06-23 02:03:59 -07004828 retval = security_task_getscheduler(p);
4829 if (retval)
4830 goto out_unlock;
4831
Thomas Gleixner31605682009-12-08 20:24:16 +00004832 rq = task_rq_lock(p, &flags);
Rusty Russell96f874e22008-11-25 02:35:14 +10304833 cpumask_and(mask, &p->cpus_allowed, cpu_online_mask);
Thomas Gleixner31605682009-12-08 20:24:16 +00004834 task_rq_unlock(rq, &flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004835
4836out_unlock:
Thomas Gleixner23f5d142009-12-09 10:15:01 +00004837 rcu_read_unlock();
Gautham R Shenoy95402b32008-01-25 21:08:02 +01004838 put_online_cpus();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004839
Ulrich Drepper9531b622007-08-09 11:16:46 +02004840 return retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004841}
4842
4843/**
4844 * sys_sched_getaffinity - get the cpu affinity of a process
4845 * @pid: pid of the process
4846 * @len: length in bytes of the bitmask pointed to by user_mask_ptr
4847 * @user_mask_ptr: user-space pointer to hold the current cpu mask
4848 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004849SYSCALL_DEFINE3(sched_getaffinity, pid_t, pid, unsigned int, len,
4850 unsigned long __user *, user_mask_ptr)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004851{
4852 int ret;
Rusty Russellf17c8602008-11-25 02:35:11 +10304853 cpumask_var_t mask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004854
Rusty Russellf17c8602008-11-25 02:35:11 +10304855 if (len < cpumask_size())
Linus Torvalds1da177e2005-04-16 15:20:36 -07004856 return -EINVAL;
4857
Rusty Russellf17c8602008-11-25 02:35:11 +10304858 if (!alloc_cpumask_var(&mask, GFP_KERNEL))
4859 return -ENOMEM;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004860
Rusty Russellf17c8602008-11-25 02:35:11 +10304861 ret = sched_getaffinity(pid, mask);
4862 if (ret == 0) {
4863 if (copy_to_user(user_mask_ptr, mask, cpumask_size()))
4864 ret = -EFAULT;
4865 else
4866 ret = cpumask_size();
4867 }
4868 free_cpumask_var(mask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004869
Rusty Russellf17c8602008-11-25 02:35:11 +10304870 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004871}
4872
4873/**
4874 * sys_sched_yield - yield the current processor to other threads.
4875 *
Ingo Molnardd41f592007-07-09 18:51:59 +02004876 * This function yields the current CPU to other tasks. If there are no
4877 * other threads running on this CPU then this function will return.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004878 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01004879SYSCALL_DEFINE0(sched_yield)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004880{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004881 struct rq *rq = this_rq_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004882
Ingo Molnar2d723762007-10-15 17:00:12 +02004883 schedstat_inc(rq, yld_count);
Dmitry Adamushko4530d7a2007-10-15 17:00:08 +02004884 current->sched_class->yield_task(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004885
4886 /*
4887 * Since we are going to call schedule() anyway, there's
4888 * no need to preempt or enable interrupts:
4889 */
4890 __release(rq->lock);
Ingo Molnar8a25d5d2006-07-03 00:24:54 -07004891 spin_release(&rq->lock.dep_map, 1, _THIS_IP_);
Thomas Gleixner9828ea92009-12-03 20:55:53 +01004892 do_raw_spin_unlock(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004893 preempt_enable_no_resched();
4894
4895 schedule();
4896
4897 return 0;
4898}
4899
Peter Zijlstrad86ee482009-07-10 14:57:57 +02004900static inline int should_resched(void)
4901{
4902 return need_resched() && !(preempt_count() & PREEMPT_ACTIVE);
4903}
4904
Andrew Mortone7b38402006-06-30 01:56:00 -07004905static void __cond_resched(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004906{
Frederic Weisbeckere7aaaa62009-07-16 15:44:29 +02004907 add_preempt_count(PREEMPT_ACTIVE);
4908 schedule();
4909 sub_preempt_count(PREEMPT_ACTIVE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004910}
4911
Herbert Xu02b67cc32008-01-25 21:08:28 +01004912int __sched _cond_resched(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004913{
Peter Zijlstrad86ee482009-07-10 14:57:57 +02004914 if (should_resched()) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004915 __cond_resched();
4916 return 1;
4917 }
4918 return 0;
4919}
Herbert Xu02b67cc32008-01-25 21:08:28 +01004920EXPORT_SYMBOL(_cond_resched);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004921
4922/*
Frederic Weisbecker613afbf2009-07-16 15:44:29 +02004923 * __cond_resched_lock() - if a reschedule is pending, drop the given lock,
Linus Torvalds1da177e2005-04-16 15:20:36 -07004924 * call schedule, and on return reacquire the lock.
4925 *
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01004926 * This works OK both with and without CONFIG_PREEMPT. We do strange low-level
Linus Torvalds1da177e2005-04-16 15:20:36 -07004927 * operations here to prevent schedule() from being called twice (once via
4928 * spin_unlock(), once by hand).
4929 */
Frederic Weisbecker613afbf2009-07-16 15:44:29 +02004930int __cond_resched_lock(spinlock_t *lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004931{
Peter Zijlstrad86ee482009-07-10 14:57:57 +02004932 int resched = should_resched();
Jan Kara6df3cec2005-06-13 15:52:32 -07004933 int ret = 0;
4934
Peter Zijlstraf607c662009-07-20 19:16:29 +02004935 lockdep_assert_held(lock);
4936
Nick Piggin95c354f2008-01-30 13:31:20 +01004937 if (spin_needbreak(lock) || resched) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004938 spin_unlock(lock);
Peter Zijlstrad86ee482009-07-10 14:57:57 +02004939 if (resched)
Nick Piggin95c354f2008-01-30 13:31:20 +01004940 __cond_resched();
4941 else
4942 cpu_relax();
Jan Kara6df3cec2005-06-13 15:52:32 -07004943 ret = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004944 spin_lock(lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004945 }
Jan Kara6df3cec2005-06-13 15:52:32 -07004946 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004947}
Frederic Weisbecker613afbf2009-07-16 15:44:29 +02004948EXPORT_SYMBOL(__cond_resched_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004949
Frederic Weisbecker613afbf2009-07-16 15:44:29 +02004950int __sched __cond_resched_softirq(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004951{
4952 BUG_ON(!in_softirq());
4953
Peter Zijlstrad86ee482009-07-10 14:57:57 +02004954 if (should_resched()) {
Thomas Gleixner98d825672007-05-23 13:58:18 -07004955 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004956 __cond_resched();
4957 local_bh_disable();
4958 return 1;
4959 }
4960 return 0;
4961}
Frederic Weisbecker613afbf2009-07-16 15:44:29 +02004962EXPORT_SYMBOL(__cond_resched_softirq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004963
Linus Torvalds1da177e2005-04-16 15:20:36 -07004964/**
4965 * yield - yield the current processor to other threads.
4966 *
Robert P. J. Day72fd4a32007-02-10 01:45:59 -08004967 * This is a shortcut for kernel-space yielding - it marks the
Linus Torvalds1da177e2005-04-16 15:20:36 -07004968 * thread runnable and calls sys_sched_yield().
4969 */
4970void __sched yield(void)
4971{
4972 set_current_state(TASK_RUNNING);
4973 sys_sched_yield();
4974}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004975EXPORT_SYMBOL(yield);
4976
4977/*
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01004978 * This task is about to go to sleep on IO. Increment rq->nr_iowait so
Linus Torvalds1da177e2005-04-16 15:20:36 -07004979 * that process accounting knows that this is a task in IO wait state.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004980 */
4981void __sched io_schedule(void)
4982{
Hitoshi Mitake54d35f22009-06-29 14:44:57 +09004983 struct rq *rq = raw_rq();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004984
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004985 delayacct_blkio_start();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004986 atomic_inc(&rq->nr_iowait);
Arjan van de Ven8f0dfc32009-07-20 11:26:58 -07004987 current->in_iowait = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004988 schedule();
Arjan van de Ven8f0dfc32009-07-20 11:26:58 -07004989 current->in_iowait = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004990 atomic_dec(&rq->nr_iowait);
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004991 delayacct_blkio_end();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004992}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004993EXPORT_SYMBOL(io_schedule);
4994
4995long __sched io_schedule_timeout(long timeout)
4996{
Hitoshi Mitake54d35f22009-06-29 14:44:57 +09004997 struct rq *rq = raw_rq();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004998 long ret;
4999
Shailabh Nagar0ff92242006-07-14 00:24:37 -07005000 delayacct_blkio_start();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005001 atomic_inc(&rq->nr_iowait);
Arjan van de Ven8f0dfc32009-07-20 11:26:58 -07005002 current->in_iowait = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005003 ret = schedule_timeout(timeout);
Arjan van de Ven8f0dfc32009-07-20 11:26:58 -07005004 current->in_iowait = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005005 atomic_dec(&rq->nr_iowait);
Shailabh Nagar0ff92242006-07-14 00:24:37 -07005006 delayacct_blkio_end();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005007 return ret;
5008}
5009
5010/**
5011 * sys_sched_get_priority_max - return maximum RT priority.
5012 * @policy: scheduling class.
5013 *
5014 * this syscall returns the maximum rt_priority that can be used
5015 * by a given scheduling class.
5016 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01005017SYSCALL_DEFINE1(sched_get_priority_max, int, policy)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005018{
5019 int ret = -EINVAL;
5020
5021 switch (policy) {
5022 case SCHED_FIFO:
5023 case SCHED_RR:
5024 ret = MAX_USER_RT_PRIO-1;
5025 break;
5026 case SCHED_NORMAL:
Ingo Molnarb0a94992006-01-14 13:20:41 -08005027 case SCHED_BATCH:
Ingo Molnardd41f592007-07-09 18:51:59 +02005028 case SCHED_IDLE:
Linus Torvalds1da177e2005-04-16 15:20:36 -07005029 ret = 0;
5030 break;
5031 }
5032 return ret;
5033}
5034
5035/**
5036 * sys_sched_get_priority_min - return minimum RT priority.
5037 * @policy: scheduling class.
5038 *
5039 * this syscall returns the minimum rt_priority that can be used
5040 * by a given scheduling class.
5041 */
Heiko Carstens5add95d2009-01-14 14:14:08 +01005042SYSCALL_DEFINE1(sched_get_priority_min, int, policy)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005043{
5044 int ret = -EINVAL;
5045
5046 switch (policy) {
5047 case SCHED_FIFO:
5048 case SCHED_RR:
5049 ret = 1;
5050 break;
5051 case SCHED_NORMAL:
Ingo Molnarb0a94992006-01-14 13:20:41 -08005052 case SCHED_BATCH:
Ingo Molnardd41f592007-07-09 18:51:59 +02005053 case SCHED_IDLE:
Linus Torvalds1da177e2005-04-16 15:20:36 -07005054 ret = 0;
5055 }
5056 return ret;
5057}
5058
5059/**
5060 * sys_sched_rr_get_interval - return the default timeslice of a process.
5061 * @pid: pid of the process.
5062 * @interval: userspace pointer to the timeslice value.
5063 *
5064 * this syscall writes the default timeslice value of a given process
5065 * into the user-space timespec buffer. A value of '0' means infinity.
5066 */
Heiko Carstens17da2bd2009-01-14 14:14:10 +01005067SYSCALL_DEFINE2(sched_rr_get_interval, pid_t, pid,
Heiko Carstens754fe8d2009-01-14 14:14:09 +01005068 struct timespec __user *, interval)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005069{
Ingo Molnar36c8b582006-07-03 00:25:41 -07005070 struct task_struct *p;
Dmitry Adamushkoa4ec24b2007-10-15 17:00:13 +02005071 unsigned int time_slice;
Thomas Gleixnerdba091b2009-12-09 09:32:03 +01005072 unsigned long flags;
5073 struct rq *rq;
Andi Kleen3a5c3592007-10-15 17:00:14 +02005074 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005075 struct timespec t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005076
5077 if (pid < 0)
Andi Kleen3a5c3592007-10-15 17:00:14 +02005078 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005079
5080 retval = -ESRCH;
Thomas Gleixner1a551ae2009-12-09 10:15:11 +00005081 rcu_read_lock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005082 p = find_process_by_pid(pid);
5083 if (!p)
5084 goto out_unlock;
5085
5086 retval = security_task_getscheduler(p);
5087 if (retval)
5088 goto out_unlock;
5089
Thomas Gleixnerdba091b2009-12-09 09:32:03 +01005090 rq = task_rq_lock(p, &flags);
5091 time_slice = p->sched_class->get_rr_interval(rq, p);
5092 task_rq_unlock(rq, &flags);
Dmitry Adamushkoa4ec24b2007-10-15 17:00:13 +02005093
Thomas Gleixner1a551ae2009-12-09 10:15:11 +00005094 rcu_read_unlock();
Dmitry Adamushkoa4ec24b2007-10-15 17:00:13 +02005095 jiffies_to_timespec(time_slice, &t);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005096 retval = copy_to_user(interval, &t, sizeof(t)) ? -EFAULT : 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005097 return retval;
Andi Kleen3a5c3592007-10-15 17:00:14 +02005098
Linus Torvalds1da177e2005-04-16 15:20:36 -07005099out_unlock:
Thomas Gleixner1a551ae2009-12-09 10:15:11 +00005100 rcu_read_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005101 return retval;
5102}
5103
Steven Rostedt7c731e02008-05-12 21:20:41 +02005104static const char stat_nam[] = TASK_STATE_TO_CHAR_STR;
Ingo Molnar36c8b582006-07-03 00:25:41 -07005105
Ingo Molnar82a1fcb2008-01-25 21:08:02 +01005106void sched_show_task(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005107{
Linus Torvalds1da177e2005-04-16 15:20:36 -07005108 unsigned long free = 0;
Ingo Molnar36c8b582006-07-03 00:25:41 -07005109 unsigned state;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005110
Linus Torvalds1da177e2005-04-16 15:20:36 -07005111 state = p->state ? __ffs(p->state) + 1 : 0;
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005112 printk(KERN_INFO "%-13.13s %c", p->comm,
Andreas Mohr2ed6e342006-07-10 04:43:52 -07005113 state < sizeof(stat_nam) - 1 ? stat_nam[state] : '?');
Ingo Molnar4bd77322007-07-11 21:21:47 +02005114#if BITS_PER_LONG == 32
Linus Torvalds1da177e2005-04-16 15:20:36 -07005115 if (state == TASK_RUNNING)
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005116 printk(KERN_CONT " running ");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005117 else
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005118 printk(KERN_CONT " %08lx ", thread_saved_pc(p));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005119#else
5120 if (state == TASK_RUNNING)
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005121 printk(KERN_CONT " running task ");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005122 else
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005123 printk(KERN_CONT " %016lx ", thread_saved_pc(p));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005124#endif
5125#ifdef CONFIG_DEBUG_STACK_USAGE
Eric Sandeen7c9f8862008-04-22 16:38:23 -05005126 free = stack_not_used(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005127#endif
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005128 printk(KERN_CONT "%5lu %5d %6d 0x%08lx\n", free,
David Rientjesaa47b7e2009-05-04 01:38:05 -07005129 task_pid_nr(p), task_pid_nr(p->real_parent),
5130 (unsigned long)task_thread_info(p)->flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005131
Nick Piggin5fb5e6d2008-01-25 21:08:34 +01005132 show_stack(p, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005133}
5134
Ingo Molnare59e2ae2006-12-06 20:35:59 -08005135void show_state_filter(unsigned long state_filter)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005136{
Ingo Molnar36c8b582006-07-03 00:25:41 -07005137 struct task_struct *g, *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005138
Ingo Molnar4bd77322007-07-11 21:21:47 +02005139#if BITS_PER_LONG == 32
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005140 printk(KERN_INFO
5141 " task PC stack pid father\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005142#else
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01005143 printk(KERN_INFO
5144 " task PC stack pid father\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005145#endif
5146 read_lock(&tasklist_lock);
5147 do_each_thread(g, p) {
5148 /*
5149 * reset the NMI-timeout, listing all files on a slow
5150 * console might take alot of time:
5151 */
5152 touch_nmi_watchdog();
Ingo Molnar39bc89f2007-04-25 20:50:03 -07005153 if (!state_filter || (p->state & state_filter))
Ingo Molnar82a1fcb2008-01-25 21:08:02 +01005154 sched_show_task(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005155 } while_each_thread(g, p);
5156
Jeremy Fitzhardinge04c91672007-05-08 00:28:05 -07005157 touch_all_softlockup_watchdogs();
5158
Ingo Molnardd41f592007-07-09 18:51:59 +02005159#ifdef CONFIG_SCHED_DEBUG
5160 sysrq_sched_debug_show();
5161#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07005162 read_unlock(&tasklist_lock);
Ingo Molnare59e2ae2006-12-06 20:35:59 -08005163 /*
5164 * Only show locks if all tasks are dumped:
5165 */
Shmulik Ladkani93335a22009-11-25 15:23:41 +02005166 if (!state_filter)
Ingo Molnare59e2ae2006-12-06 20:35:59 -08005167 debug_show_all_locks();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005168}
5169
Ingo Molnar1df21052007-07-09 18:51:58 +02005170void __cpuinit init_idle_bootup_task(struct task_struct *idle)
5171{
Ingo Molnardd41f592007-07-09 18:51:59 +02005172 idle->sched_class = &idle_sched_class;
Ingo Molnar1df21052007-07-09 18:51:58 +02005173}
5174
Ingo Molnarf340c0d2005-06-28 16:40:42 +02005175/**
5176 * init_idle - set up an idle thread for a given CPU
5177 * @idle: task in question
5178 * @cpu: cpu the idle task belongs to
5179 *
5180 * NOTE: this function does not set the idle thread's NEED_RESCHED
5181 * flag, to make booting more robust.
5182 */
Nick Piggin5c1e1762006-10-03 01:14:04 -07005183void __cpuinit init_idle(struct task_struct *idle, int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005184{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005185 struct rq *rq = cpu_rq(cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005186 unsigned long flags;
5187
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005188 raw_spin_lock_irqsave(&rq->lock, flags);
Ingo Molnar5cbd54e2008-11-12 20:05:50 +01005189
Ingo Molnardd41f592007-07-09 18:51:59 +02005190 __sched_fork(idle);
Peter Zijlstra06b83b52009-12-16 18:04:35 +01005191 idle->state = TASK_RUNNING;
Ingo Molnardd41f592007-07-09 18:51:59 +02005192 idle->se.exec_start = sched_clock();
5193
Rusty Russell96f874e22008-11-25 02:35:14 +10305194 cpumask_copy(&idle->cpus_allowed, cpumask_of(cpu));
Ingo Molnardd41f592007-07-09 18:51:59 +02005195 __set_task_cpu(idle, cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005196
Linus Torvalds1da177e2005-04-16 15:20:36 -07005197 rq->curr = rq->idle = idle;
Nick Piggin4866cde2005-06-25 14:57:23 -07005198#if defined(CONFIG_SMP) && defined(__ARCH_WANT_UNLOCKED_CTXSW)
5199 idle->oncpu = 1;
5200#endif
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005201 raw_spin_unlock_irqrestore(&rq->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005202
5203 /* Set the preempt count _outside_ the spinlocks! */
Linus Torvalds8e3e0762008-05-10 20:58:02 -07005204#if defined(CONFIG_PREEMPT)
5205 task_thread_info(idle)->preempt_count = (idle->lock_depth >= 0);
5206#else
Al Viroa1261f542005-11-13 16:06:55 -08005207 task_thread_info(idle)->preempt_count = 0;
Linus Torvalds8e3e0762008-05-10 20:58:02 -07005208#endif
Ingo Molnardd41f592007-07-09 18:51:59 +02005209 /*
5210 * The idle tasks have their own, simple scheduling class:
5211 */
5212 idle->sched_class = &idle_sched_class;
Frederic Weisbeckerfb526072008-11-25 21:07:04 +01005213 ftrace_graph_init_task(idle);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005214}
5215
5216/*
5217 * In a system that switches off the HZ timer nohz_cpu_mask
5218 * indicates which cpus entered this state. This is used
5219 * in the rcu update to wait only for active cpus. For system
5220 * which do not switch off the HZ timer nohz_cpu_mask should
Rusty Russell6a7b3dc2008-11-25 02:35:04 +10305221 * always be CPU_BITS_NONE.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005222 */
Rusty Russell6a7b3dc2008-11-25 02:35:04 +10305223cpumask_var_t nohz_cpu_mask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005224
Ingo Molnar19978ca2007-11-09 22:39:38 +01005225/*
5226 * Increase the granularity value when there are more CPUs,
5227 * because with more CPUs the 'effective latency' as visible
5228 * to users decreases. But the relationship is not linear,
5229 * so pick a second-best guess by going with the log2 of the
5230 * number of CPUs.
5231 *
5232 * This idea comes from the SD scheduler of Con Kolivas:
5233 */
Christian Ehrhardtacb4a842009-11-30 12:16:48 +01005234static int get_update_sysctl_factor(void)
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +01005235{
Mike Galbraith4ca3ef72009-12-10 09:25:53 +01005236 unsigned int cpus = min_t(int, num_online_cpus(), 8);
Christian Ehrhardt1983a922009-11-30 12:16:47 +01005237 unsigned int factor;
5238
5239 switch (sysctl_sched_tunable_scaling) {
5240 case SCHED_TUNABLESCALING_NONE:
5241 factor = 1;
5242 break;
5243 case SCHED_TUNABLESCALING_LINEAR:
5244 factor = cpus;
5245 break;
5246 case SCHED_TUNABLESCALING_LOG:
5247 default:
5248 factor = 1 + ilog2(cpus);
5249 break;
5250 }
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +01005251
Christian Ehrhardtacb4a842009-11-30 12:16:48 +01005252 return factor;
5253}
5254
5255static void update_sysctl(void)
5256{
5257 unsigned int factor = get_update_sysctl_factor();
5258
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +01005259#define SET_SYSCTL(name) \
5260 (sysctl_##name = (factor) * normalized_sysctl_##name)
5261 SET_SYSCTL(sched_min_granularity);
5262 SET_SYSCTL(sched_latency);
5263 SET_SYSCTL(sched_wakeup_granularity);
5264 SET_SYSCTL(sched_shares_ratelimit);
5265#undef SET_SYSCTL
5266}
5267
Ingo Molnar19978ca2007-11-09 22:39:38 +01005268static inline void sched_init_granularity(void)
5269{
Christian Ehrhardt0bcdcf22009-11-30 12:16:46 +01005270 update_sysctl();
Ingo Molnar19978ca2007-11-09 22:39:38 +01005271}
5272
Linus Torvalds1da177e2005-04-16 15:20:36 -07005273#ifdef CONFIG_SMP
5274/*
5275 * This is how migration works:
5276 *
Ingo Molnar70b97a72006-07-03 00:25:42 -07005277 * 1) we queue a struct migration_req structure in the source CPU's
Linus Torvalds1da177e2005-04-16 15:20:36 -07005278 * runqueue and wake up that CPU's migration thread.
5279 * 2) we down() the locked semaphore => thread blocks.
5280 * 3) migration thread wakes up (implicitly it forces the migrated
5281 * thread off the CPU)
5282 * 4) it gets the migration request and checks whether the migrated
5283 * task is still in the wrong runqueue.
5284 * 5) if it's in the wrong runqueue then the migration thread removes
5285 * it and puts it into the right queue.
5286 * 6) migration thread up()s the semaphore.
5287 * 7) we wake up and the migration is done.
5288 */
5289
5290/*
5291 * Change a given task's CPU affinity. Migrate the thread to a
5292 * proper CPU and schedule it away if the CPU it's executing on
5293 * is removed from the allowed bitmask.
5294 *
5295 * NOTE: the caller must have a valid reference to the task, the
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005296 * task must not exit() & deallocate itself prematurely. The
Linus Torvalds1da177e2005-04-16 15:20:36 -07005297 * call is not atomic; no spinlocks may be held.
5298 */
Rusty Russell96f874e22008-11-25 02:35:14 +10305299int set_cpus_allowed_ptr(struct task_struct *p, const struct cpumask *new_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005300{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005301 struct migration_req req;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005302 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005303 struct rq *rq;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005304 int ret = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005305
Peter Zijlstrae2912002009-12-16 18:04:36 +01005306 /*
5307 * Since we rely on wake-ups to migrate sleeping tasks, don't change
5308 * the ->cpus_allowed mask from under waking tasks, which would be
5309 * possible when we change rq->lock in ttwu(), so synchronize against
5310 * TASK_WAKING to avoid that.
5311 */
5312again:
5313 while (p->state == TASK_WAKING)
5314 cpu_relax();
5315
Linus Torvalds1da177e2005-04-16 15:20:36 -07005316 rq = task_rq_lock(p, &flags);
Peter Zijlstrae2912002009-12-16 18:04:36 +01005317
5318 if (p->state == TASK_WAKING) {
5319 task_rq_unlock(rq, &flags);
5320 goto again;
5321 }
5322
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01005323 if (!cpumask_intersects(new_mask, cpu_active_mask)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07005324 ret = -EINVAL;
5325 goto out;
5326 }
5327
David Rientjes9985b0b2008-06-05 12:57:11 -07005328 if (unlikely((p->flags & PF_THREAD_BOUND) && p != current &&
Rusty Russell96f874e22008-11-25 02:35:14 +10305329 !cpumask_equal(&p->cpus_allowed, new_mask))) {
David Rientjes9985b0b2008-06-05 12:57:11 -07005330 ret = -EINVAL;
5331 goto out;
5332 }
5333
Gregory Haskins73fe6aae2008-01-25 21:08:07 +01005334 if (p->sched_class->set_cpus_allowed)
Mike Traviscd8ba7c2008-03-26 14:23:49 -07005335 p->sched_class->set_cpus_allowed(p, new_mask);
Gregory Haskins73fe6aae2008-01-25 21:08:07 +01005336 else {
Rusty Russell96f874e22008-11-25 02:35:14 +10305337 cpumask_copy(&p->cpus_allowed, new_mask);
5338 p->rt.nr_cpus_allowed = cpumask_weight(new_mask);
Gregory Haskins73fe6aae2008-01-25 21:08:07 +01005339 }
5340
Linus Torvalds1da177e2005-04-16 15:20:36 -07005341 /* Can the task run on the task's current CPU? If so, we're done */
Rusty Russell96f874e22008-11-25 02:35:14 +10305342 if (cpumask_test_cpu(task_cpu(p), new_mask))
Linus Torvalds1da177e2005-04-16 15:20:36 -07005343 goto out;
5344
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01005345 if (migrate_task(p, cpumask_any_and(cpu_active_mask, new_mask), &req)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07005346 /* Need help from migration thread: drop lock and wait. */
Peter Zijlstra693525e2009-07-21 13:56:38 +02005347 struct task_struct *mt = rq->migration_thread;
5348
5349 get_task_struct(mt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005350 task_rq_unlock(rq, &flags);
5351 wake_up_process(rq->migration_thread);
Peter Zijlstra693525e2009-07-21 13:56:38 +02005352 put_task_struct(mt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005353 wait_for_completion(&req.done);
5354 tlb_migrate_finish(p->mm);
5355 return 0;
5356 }
5357out:
5358 task_rq_unlock(rq, &flags);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005359
Linus Torvalds1da177e2005-04-16 15:20:36 -07005360 return ret;
5361}
Mike Traviscd8ba7c2008-03-26 14:23:49 -07005362EXPORT_SYMBOL_GPL(set_cpus_allowed_ptr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005363
5364/*
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005365 * Move (not current) task off this cpu, onto dest cpu. We're doing
Linus Torvalds1da177e2005-04-16 15:20:36 -07005366 * this because either it can't run here any more (set_cpus_allowed()
5367 * away from this CPU, or CPU going down), or because we're
5368 * attempting to rebalance this task on exec (sched_exec).
5369 *
5370 * So we race with normal scheduler movements, but that's OK, as long
5371 * as the task is no longer on this CPU.
Kirill Korotaevefc30812006-06-27 02:54:32 -07005372 *
5373 * Returns non-zero if task was successfully migrated.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005374 */
Kirill Korotaevefc30812006-06-27 02:54:32 -07005375static int __migrate_task(struct task_struct *p, int src_cpu, int dest_cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005376{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005377 struct rq *rq_dest, *rq_src;
Peter Zijlstrae2912002009-12-16 18:04:36 +01005378 int ret = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005379
Max Krasnyanskye761b772008-07-15 04:43:49 -07005380 if (unlikely(!cpu_active(dest_cpu)))
Kirill Korotaevefc30812006-06-27 02:54:32 -07005381 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005382
5383 rq_src = cpu_rq(src_cpu);
5384 rq_dest = cpu_rq(dest_cpu);
5385
5386 double_rq_lock(rq_src, rq_dest);
5387 /* Already moved. */
5388 if (task_cpu(p) != src_cpu)
Linus Torvaldsb1e38732008-07-10 11:25:03 -07005389 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005390 /* Affinity changed (again). */
Rusty Russell96f874e22008-11-25 02:35:14 +10305391 if (!cpumask_test_cpu(dest_cpu, &p->cpus_allowed))
Linus Torvaldsb1e38732008-07-10 11:25:03 -07005392 goto fail;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005393
Peter Zijlstrae2912002009-12-16 18:04:36 +01005394 /*
5395 * If we're not on a rq, the next wake-up will ensure we're
5396 * placed properly.
5397 */
5398 if (p->se.on_rq) {
Ingo Molnar2e1cb742007-08-09 11:16:49 +02005399 deactivate_task(rq_src, p, 0);
Peter Zijlstrae2912002009-12-16 18:04:36 +01005400 set_task_cpu(p, dest_cpu);
Ingo Molnardd41f592007-07-09 18:51:59 +02005401 activate_task(rq_dest, p, 0);
Peter Zijlstra15afe092008-09-20 23:38:02 +02005402 check_preempt_curr(rq_dest, p, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005403 }
Linus Torvaldsb1e38732008-07-10 11:25:03 -07005404done:
Kirill Korotaevefc30812006-06-27 02:54:32 -07005405 ret = 1;
Linus Torvaldsb1e38732008-07-10 11:25:03 -07005406fail:
Linus Torvalds1da177e2005-04-16 15:20:36 -07005407 double_rq_unlock(rq_src, rq_dest);
Kirill Korotaevefc30812006-06-27 02:54:32 -07005408 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005409}
5410
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005411#define RCU_MIGRATION_IDLE 0
5412#define RCU_MIGRATION_NEED_QS 1
5413#define RCU_MIGRATION_GOT_QS 2
5414#define RCU_MIGRATION_MUST_SYNC 3
5415
Linus Torvalds1da177e2005-04-16 15:20:36 -07005416/*
5417 * migration_thread - this is a highprio system thread that performs
5418 * thread migration by bumping thread off CPU then 'pushing' onto
5419 * another runqueue.
5420 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07005421static int migration_thread(void *data)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005422{
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005423 int badcpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005424 int cpu = (long)data;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005425 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005426
5427 rq = cpu_rq(cpu);
5428 BUG_ON(rq->migration_thread != current);
5429
5430 set_current_state(TASK_INTERRUPTIBLE);
5431 while (!kthread_should_stop()) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07005432 struct migration_req *req;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005433 struct list_head *head;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005434
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005435 raw_spin_lock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005436
5437 if (cpu_is_offline(cpu)) {
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005438 raw_spin_unlock_irq(&rq->lock);
Oleg Nesterov371cbb32009-06-17 16:27:45 -07005439 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005440 }
5441
5442 if (rq->active_balance) {
5443 active_load_balance(rq, cpu);
5444 rq->active_balance = 0;
5445 }
5446
5447 head = &rq->migration_queue;
5448
5449 if (list_empty(head)) {
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005450 raw_spin_unlock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005451 schedule();
5452 set_current_state(TASK_INTERRUPTIBLE);
5453 continue;
5454 }
Ingo Molnar70b97a72006-07-03 00:25:42 -07005455 req = list_entry(head->next, struct migration_req, list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005456 list_del_init(head->next);
5457
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005458 if (req->task != NULL) {
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005459 raw_spin_unlock(&rq->lock);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005460 __migrate_task(req->task, cpu, req->dest_cpu);
5461 } else if (likely(cpu == (badcpu = smp_processor_id()))) {
5462 req->dest_cpu = RCU_MIGRATION_GOT_QS;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005463 raw_spin_unlock(&rq->lock);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005464 } else {
5465 req->dest_cpu = RCU_MIGRATION_MUST_SYNC;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005466 raw_spin_unlock(&rq->lock);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07005467 WARN_ONCE(1, "migration_thread() on CPU %d, expected %d\n", badcpu, cpu);
5468 }
Nick Piggin674311d2005-06-25 14:57:27 -07005469 local_irq_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005470
5471 complete(&req->done);
5472 }
5473 __set_current_state(TASK_RUNNING);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005474
Linus Torvalds1da177e2005-04-16 15:20:36 -07005475 return 0;
5476}
5477
5478#ifdef CONFIG_HOTPLUG_CPU
Oleg Nesterovf7b4cdd2007-10-16 23:30:56 -07005479
5480static int __migrate_task_irq(struct task_struct *p, int src_cpu, int dest_cpu)
5481{
5482 int ret;
5483
5484 local_irq_disable();
5485 ret = __migrate_task(p, src_cpu, dest_cpu);
5486 local_irq_enable();
5487 return ret;
5488}
5489
Kirill Korotaev054b9102006-12-10 02:20:11 -08005490/*
Robert P. J. Day3a4fa0a2007-10-19 23:10:43 +02005491 * Figure out where task on dead CPU should go, use force if necessary.
Kirill Korotaev054b9102006-12-10 02:20:11 -08005492 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005493static void move_task_off_dead_cpu(int dead_cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005494{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005495 int dest_cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005496
Rusty Russelle76bd8d2008-11-25 02:35:11 +10305497again:
Peter Zijlstra5da9a0f2009-12-16 18:04:38 +01005498 dest_cpu = select_fallback_rq(dead_cpu, p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005499
Rusty Russelle76bd8d2008-11-25 02:35:11 +10305500 /* It can have affinity changed while we were choosing. */
5501 if (unlikely(!__migrate_task_irq(p, dead_cpu, dest_cpu)))
5502 goto again;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005503}
5504
5505/*
5506 * While a dead CPU has no uninterruptible tasks queued at this point,
5507 * it might still have a nonzero ->nr_uninterruptible counter, because
5508 * for performance reasons the counter is not stricly tracking tasks to
5509 * their home CPUs. So we just add the counter to another CPU's counter,
5510 * to keep the global sum constant after CPU-down:
5511 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07005512static void migrate_nr_uninterruptible(struct rq *rq_src)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005513{
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01005514 struct rq *rq_dest = cpu_rq(cpumask_any(cpu_active_mask));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005515 unsigned long flags;
5516
5517 local_irq_save(flags);
5518 double_rq_lock(rq_src, rq_dest);
5519 rq_dest->nr_uninterruptible += rq_src->nr_uninterruptible;
5520 rq_src->nr_uninterruptible = 0;
5521 double_rq_unlock(rq_src, rq_dest);
5522 local_irq_restore(flags);
5523}
5524
5525/* Run through task list and migrate tasks from the dead cpu. */
5526static void migrate_live_tasks(int src_cpu)
5527{
Ingo Molnar48f24c42006-07-03 00:25:40 -07005528 struct task_struct *p, *t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005529
Oleg Nesterovf7b4cdd2007-10-16 23:30:56 -07005530 read_lock(&tasklist_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005531
Ingo Molnar48f24c42006-07-03 00:25:40 -07005532 do_each_thread(t, p) {
5533 if (p == current)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005534 continue;
5535
Ingo Molnar48f24c42006-07-03 00:25:40 -07005536 if (task_cpu(p) == src_cpu)
5537 move_task_off_dead_cpu(src_cpu, p);
5538 } while_each_thread(t, p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005539
Oleg Nesterovf7b4cdd2007-10-16 23:30:56 -07005540 read_unlock(&tasklist_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005541}
5542
Ingo Molnardd41f592007-07-09 18:51:59 +02005543/*
5544 * Schedules idle task to be the next runnable task on current CPU.
Dmitry Adamushko94bc9a72007-11-15 20:57:40 +01005545 * It does so by boosting its priority to highest possible.
5546 * Used by CPU offline code.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005547 */
5548void sched_idle_next(void)
5549{
Ingo Molnar48f24c42006-07-03 00:25:40 -07005550 int this_cpu = smp_processor_id();
Ingo Molnar70b97a72006-07-03 00:25:42 -07005551 struct rq *rq = cpu_rq(this_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005552 struct task_struct *p = rq->idle;
5553 unsigned long flags;
5554
5555 /* cpu has to be offline */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005556 BUG_ON(cpu_online(this_cpu));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005557
Ingo Molnar48f24c42006-07-03 00:25:40 -07005558 /*
5559 * Strictly not necessary since rest of the CPUs are stopped by now
5560 * and interrupts disabled on the current cpu.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005561 */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005562 raw_spin_lock_irqsave(&rq->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005563
Ingo Molnardd41f592007-07-09 18:51:59 +02005564 __setscheduler(rq, p, SCHED_FIFO, MAX_RT_PRIO-1);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005565
Dmitry Adamushko94bc9a72007-11-15 20:57:40 +01005566 update_rq_clock(rq);
5567 activate_task(rq, p, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005568
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005569 raw_spin_unlock_irqrestore(&rq->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005570}
5571
Ingo Molnar48f24c42006-07-03 00:25:40 -07005572/*
5573 * Ensures that the idle task is using init_mm right before its cpu goes
Linus Torvalds1da177e2005-04-16 15:20:36 -07005574 * offline.
5575 */
5576void idle_task_exit(void)
5577{
5578 struct mm_struct *mm = current->active_mm;
5579
5580 BUG_ON(cpu_online(smp_processor_id()));
5581
5582 if (mm != &init_mm)
5583 switch_mm(mm, &init_mm, current);
5584 mmdrop(mm);
5585}
5586
Kirill Korotaev054b9102006-12-10 02:20:11 -08005587/* called under rq->lock with disabled interrupts */
Ingo Molnar36c8b582006-07-03 00:25:41 -07005588static void migrate_dead(unsigned int dead_cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005589{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005590 struct rq *rq = cpu_rq(dead_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005591
5592 /* Must be exiting, otherwise would be on tasklist. */
Eugene Teo270f7222007-10-18 23:40:38 -07005593 BUG_ON(!p->exit_state);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005594
5595 /* Cannot have done final schedule yet: would have vanished. */
Oleg Nesterovc394cc92006-09-29 02:01:11 -07005596 BUG_ON(p->state == TASK_DEAD);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005597
Ingo Molnar48f24c42006-07-03 00:25:40 -07005598 get_task_struct(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005599
5600 /*
5601 * Drop lock around migration; if someone else moves it,
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005602 * that's OK. No task can be added to this CPU, so iteration is
Linus Torvalds1da177e2005-04-16 15:20:36 -07005603 * fine.
5604 */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005605 raw_spin_unlock_irq(&rq->lock);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005606 move_task_off_dead_cpu(dead_cpu, p);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005607 raw_spin_lock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005608
Ingo Molnar48f24c42006-07-03 00:25:40 -07005609 put_task_struct(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005610}
5611
5612/* release_task() removes task from tasklist, so we won't find dead tasks. */
5613static void migrate_dead_tasks(unsigned int dead_cpu)
5614{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005615 struct rq *rq = cpu_rq(dead_cpu);
Ingo Molnardd41f592007-07-09 18:51:59 +02005616 struct task_struct *next;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005617
Ingo Molnardd41f592007-07-09 18:51:59 +02005618 for ( ; ; ) {
5619 if (!rq->nr_running)
5620 break;
Ingo Molnara8e504d2007-08-09 11:16:47 +02005621 update_rq_clock(rq);
Wang Chenb67802e2009-03-02 13:55:26 +08005622 next = pick_next_task(rq);
Ingo Molnardd41f592007-07-09 18:51:59 +02005623 if (!next)
5624 break;
Dmitry Adamushko79c53792008-06-29 00:16:56 +02005625 next->sched_class->put_prev_task(rq, next);
Ingo Molnardd41f592007-07-09 18:51:59 +02005626 migrate_dead(dead_cpu, next);
Nick Piggine692ab52007-07-26 13:40:43 +02005627
Linus Torvalds1da177e2005-04-16 15:20:36 -07005628 }
5629}
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02005630
5631/*
5632 * remove the tasks which were accounted by rq from calc_load_tasks.
5633 */
5634static void calc_global_load_remove(struct rq *rq)
5635{
5636 atomic_long_sub(rq->calc_load_active, &calc_load_tasks);
Thomas Gleixnera468d382009-07-17 14:15:46 +02005637 rq->calc_load_active = 0;
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02005638}
Linus Torvalds1da177e2005-04-16 15:20:36 -07005639#endif /* CONFIG_HOTPLUG_CPU */
5640
Nick Piggine692ab52007-07-26 13:40:43 +02005641#if defined(CONFIG_SCHED_DEBUG) && defined(CONFIG_SYSCTL)
5642
5643static struct ctl_table sd_ctl_dir[] = {
Alexey Dobriyane0361852007-08-09 11:16:46 +02005644 {
5645 .procname = "sched_domain",
Eric W. Biedermanc57baf12007-08-23 15:18:02 +02005646 .mode = 0555,
Alexey Dobriyane0361852007-08-09 11:16:46 +02005647 },
Eric W. Biederman56992302009-11-05 15:38:40 -08005648 {}
Nick Piggine692ab52007-07-26 13:40:43 +02005649};
5650
5651static struct ctl_table sd_ctl_root[] = {
Alexey Dobriyane0361852007-08-09 11:16:46 +02005652 {
5653 .procname = "kernel",
Eric W. Biedermanc57baf12007-08-23 15:18:02 +02005654 .mode = 0555,
Alexey Dobriyane0361852007-08-09 11:16:46 +02005655 .child = sd_ctl_dir,
5656 },
Eric W. Biederman56992302009-11-05 15:38:40 -08005657 {}
Nick Piggine692ab52007-07-26 13:40:43 +02005658};
5659
5660static struct ctl_table *sd_alloc_ctl_entry(int n)
5661{
5662 struct ctl_table *entry =
Milton Miller5cf9f062007-10-15 17:00:19 +02005663 kcalloc(n, sizeof(struct ctl_table), GFP_KERNEL);
Nick Piggine692ab52007-07-26 13:40:43 +02005664
Nick Piggine692ab52007-07-26 13:40:43 +02005665 return entry;
5666}
5667
Milton Miller6382bc92007-10-15 17:00:19 +02005668static void sd_free_ctl_entry(struct ctl_table **tablep)
5669{
Milton Millercd7900762007-10-17 16:55:11 +02005670 struct ctl_table *entry;
Milton Miller6382bc92007-10-15 17:00:19 +02005671
Milton Millercd7900762007-10-17 16:55:11 +02005672 /*
5673 * In the intermediate directories, both the child directory and
5674 * procname are dynamically allocated and could fail but the mode
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005675 * will always be set. In the lowest directory the names are
Milton Millercd7900762007-10-17 16:55:11 +02005676 * static strings and all have proc handlers.
5677 */
5678 for (entry = *tablep; entry->mode; entry++) {
Milton Miller6382bc92007-10-15 17:00:19 +02005679 if (entry->child)
5680 sd_free_ctl_entry(&entry->child);
Milton Millercd7900762007-10-17 16:55:11 +02005681 if (entry->proc_handler == NULL)
5682 kfree(entry->procname);
5683 }
Milton Miller6382bc92007-10-15 17:00:19 +02005684
5685 kfree(*tablep);
5686 *tablep = NULL;
5687}
5688
Nick Piggine692ab52007-07-26 13:40:43 +02005689static void
Alexey Dobriyane0361852007-08-09 11:16:46 +02005690set_table_entry(struct ctl_table *entry,
Nick Piggine692ab52007-07-26 13:40:43 +02005691 const char *procname, void *data, int maxlen,
5692 mode_t mode, proc_handler *proc_handler)
5693{
Nick Piggine692ab52007-07-26 13:40:43 +02005694 entry->procname = procname;
5695 entry->data = data;
5696 entry->maxlen = maxlen;
5697 entry->mode = mode;
5698 entry->proc_handler = proc_handler;
5699}
5700
5701static struct ctl_table *
5702sd_alloc_ctl_domain_table(struct sched_domain *sd)
5703{
Ingo Molnara5d8c342008-10-09 11:35:51 +02005704 struct ctl_table *table = sd_alloc_ctl_entry(13);
Nick Piggine692ab52007-07-26 13:40:43 +02005705
Milton Millerad1cdc12007-10-15 17:00:19 +02005706 if (table == NULL)
5707 return NULL;
5708
Alexey Dobriyane0361852007-08-09 11:16:46 +02005709 set_table_entry(&table[0], "min_interval", &sd->min_interval,
Nick Piggine692ab52007-07-26 13:40:43 +02005710 sizeof(long), 0644, proc_doulongvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005711 set_table_entry(&table[1], "max_interval", &sd->max_interval,
Nick Piggine692ab52007-07-26 13:40:43 +02005712 sizeof(long), 0644, proc_doulongvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005713 set_table_entry(&table[2], "busy_idx", &sd->busy_idx,
Nick Piggine692ab52007-07-26 13:40:43 +02005714 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005715 set_table_entry(&table[3], "idle_idx", &sd->idle_idx,
Nick Piggine692ab52007-07-26 13:40:43 +02005716 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005717 set_table_entry(&table[4], "newidle_idx", &sd->newidle_idx,
Nick Piggine692ab52007-07-26 13:40:43 +02005718 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005719 set_table_entry(&table[5], "wake_idx", &sd->wake_idx,
Nick Piggine692ab52007-07-26 13:40:43 +02005720 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005721 set_table_entry(&table[6], "forkexec_idx", &sd->forkexec_idx,
Nick Piggine692ab52007-07-26 13:40:43 +02005722 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005723 set_table_entry(&table[7], "busy_factor", &sd->busy_factor,
Nick Piggine692ab52007-07-26 13:40:43 +02005724 sizeof(int), 0644, proc_dointvec_minmax);
Alexey Dobriyane0361852007-08-09 11:16:46 +02005725 set_table_entry(&table[8], "imbalance_pct", &sd->imbalance_pct,
Nick Piggine692ab52007-07-26 13:40:43 +02005726 sizeof(int), 0644, proc_dointvec_minmax);
Zou Nan haiace8b3d2007-10-15 17:00:14 +02005727 set_table_entry(&table[9], "cache_nice_tries",
Nick Piggine692ab52007-07-26 13:40:43 +02005728 &sd->cache_nice_tries,
5729 sizeof(int), 0644, proc_dointvec_minmax);
Zou Nan haiace8b3d2007-10-15 17:00:14 +02005730 set_table_entry(&table[10], "flags", &sd->flags,
Nick Piggine692ab52007-07-26 13:40:43 +02005731 sizeof(int), 0644, proc_dointvec_minmax);
Ingo Molnara5d8c342008-10-09 11:35:51 +02005732 set_table_entry(&table[11], "name", sd->name,
5733 CORENAME_MAX_SIZE, 0444, proc_dostring);
5734 /* &table[12] is terminator */
Nick Piggine692ab52007-07-26 13:40:43 +02005735
5736 return table;
5737}
5738
Ingo Molnar9a4e7152007-11-28 15:52:56 +01005739static ctl_table *sd_alloc_ctl_cpu_table(int cpu)
Nick Piggine692ab52007-07-26 13:40:43 +02005740{
5741 struct ctl_table *entry, *table;
5742 struct sched_domain *sd;
5743 int domain_num = 0, i;
5744 char buf[32];
5745
5746 for_each_domain(cpu, sd)
5747 domain_num++;
5748 entry = table = sd_alloc_ctl_entry(domain_num + 1);
Milton Millerad1cdc12007-10-15 17:00:19 +02005749 if (table == NULL)
5750 return NULL;
Nick Piggine692ab52007-07-26 13:40:43 +02005751
5752 i = 0;
5753 for_each_domain(cpu, sd) {
5754 snprintf(buf, 32, "domain%d", i);
Nick Piggine692ab52007-07-26 13:40:43 +02005755 entry->procname = kstrdup(buf, GFP_KERNEL);
Eric W. Biedermanc57baf12007-08-23 15:18:02 +02005756 entry->mode = 0555;
Nick Piggine692ab52007-07-26 13:40:43 +02005757 entry->child = sd_alloc_ctl_domain_table(sd);
5758 entry++;
5759 i++;
5760 }
5761 return table;
5762}
5763
5764static struct ctl_table_header *sd_sysctl_header;
Milton Miller6382bc92007-10-15 17:00:19 +02005765static void register_sched_domain_sysctl(void)
Nick Piggine692ab52007-07-26 13:40:43 +02005766{
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01005767 int i, cpu_num = num_possible_cpus();
Nick Piggine692ab52007-07-26 13:40:43 +02005768 struct ctl_table *entry = sd_alloc_ctl_entry(cpu_num + 1);
5769 char buf[32];
5770
Milton Miller73785472007-10-24 18:23:48 +02005771 WARN_ON(sd_ctl_dir[0].child);
5772 sd_ctl_dir[0].child = entry;
5773
Milton Millerad1cdc12007-10-15 17:00:19 +02005774 if (entry == NULL)
5775 return;
5776
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01005777 for_each_possible_cpu(i) {
Nick Piggine692ab52007-07-26 13:40:43 +02005778 snprintf(buf, 32, "cpu%d", i);
Nick Piggine692ab52007-07-26 13:40:43 +02005779 entry->procname = kstrdup(buf, GFP_KERNEL);
Eric W. Biedermanc57baf12007-08-23 15:18:02 +02005780 entry->mode = 0555;
Nick Piggine692ab52007-07-26 13:40:43 +02005781 entry->child = sd_alloc_ctl_cpu_table(i);
Milton Miller97b6ea72007-10-15 17:00:19 +02005782 entry++;
Nick Piggine692ab52007-07-26 13:40:43 +02005783 }
Milton Miller73785472007-10-24 18:23:48 +02005784
5785 WARN_ON(sd_sysctl_header);
Nick Piggine692ab52007-07-26 13:40:43 +02005786 sd_sysctl_header = register_sysctl_table(sd_ctl_root);
5787}
Milton Miller6382bc92007-10-15 17:00:19 +02005788
Milton Miller73785472007-10-24 18:23:48 +02005789/* may be called multiple times per register */
Milton Miller6382bc92007-10-15 17:00:19 +02005790static void unregister_sched_domain_sysctl(void)
5791{
Milton Miller73785472007-10-24 18:23:48 +02005792 if (sd_sysctl_header)
5793 unregister_sysctl_table(sd_sysctl_header);
Milton Miller6382bc92007-10-15 17:00:19 +02005794 sd_sysctl_header = NULL;
Milton Miller73785472007-10-24 18:23:48 +02005795 if (sd_ctl_dir[0].child)
5796 sd_free_ctl_entry(&sd_ctl_dir[0].child);
Milton Miller6382bc92007-10-15 17:00:19 +02005797}
Nick Piggine692ab52007-07-26 13:40:43 +02005798#else
Milton Miller6382bc92007-10-15 17:00:19 +02005799static void register_sched_domain_sysctl(void)
5800{
5801}
5802static void unregister_sched_domain_sysctl(void)
Nick Piggine692ab52007-07-26 13:40:43 +02005803{
5804}
5805#endif
5806
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04005807static void set_rq_online(struct rq *rq)
5808{
5809 if (!rq->online) {
5810 const struct sched_class *class;
5811
Rusty Russellc6c49272008-11-25 02:35:05 +10305812 cpumask_set_cpu(rq->cpu, rq->rd->online);
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04005813 rq->online = 1;
5814
5815 for_each_class(class) {
5816 if (class->rq_online)
5817 class->rq_online(rq);
5818 }
5819 }
5820}
5821
5822static void set_rq_offline(struct rq *rq)
5823{
5824 if (rq->online) {
5825 const struct sched_class *class;
5826
5827 for_each_class(class) {
5828 if (class->rq_offline)
5829 class->rq_offline(rq);
5830 }
5831
Rusty Russellc6c49272008-11-25 02:35:05 +10305832 cpumask_clear_cpu(rq->cpu, rq->rd->online);
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04005833 rq->online = 0;
5834 }
5835}
5836
Linus Torvalds1da177e2005-04-16 15:20:36 -07005837/*
5838 * migration_call - callback that gets triggered when a CPU is added.
5839 * Here we can start up the necessary migration thread for the new CPU.
5840 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005841static int __cpuinit
5842migration_call(struct notifier_block *nfb, unsigned long action, void *hcpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005843{
Linus Torvalds1da177e2005-04-16 15:20:36 -07005844 struct task_struct *p;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005845 int cpu = (long)hcpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005846 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005847 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005848
5849 switch (action) {
Gautham R Shenoy5be93612007-05-09 02:34:04 -07005850
Linus Torvalds1da177e2005-04-16 15:20:36 -07005851 case CPU_UP_PREPARE:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07005852 case CPU_UP_PREPARE_FROZEN:
Ingo Molnardd41f592007-07-09 18:51:59 +02005853 p = kthread_create(migration_thread, hcpu, "migration/%d", cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005854 if (IS_ERR(p))
5855 return NOTIFY_BAD;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005856 kthread_bind(p, cpu);
5857 /* Must be high prio: stop_machine expects to yield to it. */
5858 rq = task_rq_lock(p, &flags);
Ingo Molnardd41f592007-07-09 18:51:59 +02005859 __setscheduler(rq, p, SCHED_FIFO, MAX_RT_PRIO-1);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005860 task_rq_unlock(rq, &flags);
Oleg Nesterov371cbb32009-06-17 16:27:45 -07005861 get_task_struct(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005862 cpu_rq(cpu)->migration_thread = p;
Thomas Gleixnera468d382009-07-17 14:15:46 +02005863 rq->calc_load_update = calc_load_update;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005864 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005865
Linus Torvalds1da177e2005-04-16 15:20:36 -07005866 case CPU_ONLINE:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07005867 case CPU_ONLINE_FROZEN:
Robert P. J. Day3a4fa0a2007-10-19 23:10:43 +02005868 /* Strictly unnecessary, as first user will wake it. */
Linus Torvalds1da177e2005-04-16 15:20:36 -07005869 wake_up_process(cpu_rq(cpu)->migration_thread);
Gregory Haskins1f94ef52008-03-10 16:52:41 -04005870
5871 /* Update our root-domain */
5872 rq = cpu_rq(cpu);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005873 raw_spin_lock_irqsave(&rq->lock, flags);
Gregory Haskins1f94ef52008-03-10 16:52:41 -04005874 if (rq->rd) {
Rusty Russellc6c49272008-11-25 02:35:05 +10305875 BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04005876
5877 set_rq_online(rq);
Gregory Haskins1f94ef52008-03-10 16:52:41 -04005878 }
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005879 raw_spin_unlock_irqrestore(&rq->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005880 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005881
Linus Torvalds1da177e2005-04-16 15:20:36 -07005882#ifdef CONFIG_HOTPLUG_CPU
5883 case CPU_UP_CANCELED:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07005884 case CPU_UP_CANCELED_FROZEN:
Heiko Carstensfc75cdf2006-06-25 05:49:10 -07005885 if (!cpu_rq(cpu)->migration_thread)
5886 break;
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005887 /* Unbind it from offline cpu so it can run. Fall thru. */
Heiko Carstensa4c4af72005-11-07 00:58:38 -08005888 kthread_bind(cpu_rq(cpu)->migration_thread,
Rusty Russell1e5ce4f2008-11-25 02:35:03 +10305889 cpumask_any(cpu_online_mask));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005890 kthread_stop(cpu_rq(cpu)->migration_thread);
Oleg Nesterov371cbb32009-06-17 16:27:45 -07005891 put_task_struct(cpu_rq(cpu)->migration_thread);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005892 cpu_rq(cpu)->migration_thread = NULL;
5893 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005894
Linus Torvalds1da177e2005-04-16 15:20:36 -07005895 case CPU_DEAD:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07005896 case CPU_DEAD_FROZEN:
Cliff Wickman470fd642007-10-18 23:40:46 -07005897 cpuset_lock(); /* around calls to cpuset_cpus_allowed_lock() */
Linus Torvalds1da177e2005-04-16 15:20:36 -07005898 migrate_live_tasks(cpu);
5899 rq = cpu_rq(cpu);
5900 kthread_stop(rq->migration_thread);
Oleg Nesterov371cbb32009-06-17 16:27:45 -07005901 put_task_struct(rq->migration_thread);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005902 rq->migration_thread = NULL;
5903 /* Idle task back to normal (off runqueue, low prio) */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005904 raw_spin_lock_irq(&rq->lock);
Ingo Molnara8e504d2007-08-09 11:16:47 +02005905 update_rq_clock(rq);
Ingo Molnar2e1cb742007-08-09 11:16:49 +02005906 deactivate_task(rq, rq->idle, 0);
Ingo Molnardd41f592007-07-09 18:51:59 +02005907 __setscheduler(rq, rq->idle, SCHED_NORMAL, 0);
5908 rq->idle->sched_class = &idle_sched_class;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005909 migrate_dead_tasks(cpu);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005910 raw_spin_unlock_irq(&rq->lock);
Cliff Wickman470fd642007-10-18 23:40:46 -07005911 cpuset_unlock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005912 migrate_nr_uninterruptible(rq);
5913 BUG_ON(rq->nr_running != 0);
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02005914 calc_global_load_remove(rq);
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01005915 /*
5916 * No need to migrate the tasks: it was best-effort if
5917 * they didn't take sched_hotcpu_mutex. Just wake up
5918 * the requestors.
5919 */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005920 raw_spin_lock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005921 while (!list_empty(&rq->migration_queue)) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07005922 struct migration_req *req;
5923
Linus Torvalds1da177e2005-04-16 15:20:36 -07005924 req = list_entry(rq->migration_queue.next,
Ingo Molnar70b97a72006-07-03 00:25:42 -07005925 struct migration_req, list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005926 list_del_init(&req->list);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005927 raw_spin_unlock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005928 complete(&req->done);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005929 raw_spin_lock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005930 }
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005931 raw_spin_unlock_irq(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005932 break;
Gregory Haskins57d885f2008-01-25 21:08:18 +01005933
Gregory Haskins08f503b2008-03-10 17:59:11 -04005934 case CPU_DYING:
5935 case CPU_DYING_FROZEN:
Gregory Haskins57d885f2008-01-25 21:08:18 +01005936 /* Update our root-domain */
5937 rq = cpu_rq(cpu);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005938 raw_spin_lock_irqsave(&rq->lock, flags);
Gregory Haskins57d885f2008-01-25 21:08:18 +01005939 if (rq->rd) {
Rusty Russellc6c49272008-11-25 02:35:05 +10305940 BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04005941 set_rq_offline(rq);
Gregory Haskins57d885f2008-01-25 21:08:18 +01005942 }
Thomas Gleixner05fa7852009-11-17 14:28:38 +01005943 raw_spin_unlock_irqrestore(&rq->lock, flags);
Gregory Haskins57d885f2008-01-25 21:08:18 +01005944 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005945#endif
5946 }
5947 return NOTIFY_OK;
5948}
5949
Paul Mackerrasf38b0822009-06-02 21:05:16 +10005950/*
5951 * Register at high priority so that task migration (migrate_all_tasks)
5952 * happens before everything else. This has to be lower priority than
Ingo Molnarcdd6c482009-09-21 12:02:48 +02005953 * the notifier in the perf_event subsystem, though.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005954 */
Chandra Seetharaman26c21432006-06-27 02:54:10 -07005955static struct notifier_block __cpuinitdata migration_notifier = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07005956 .notifier_call = migration_call,
5957 .priority = 10
5958};
5959
Eduard - Gabriel Munteanu7babe8d2008-07-25 19:45:11 -07005960static int __init migration_init(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005961{
5962 void *cpu = (void *)(long)smp_processor_id();
Akinobu Mita07dccf32006-09-29 02:00:22 -07005963 int err;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005964
5965 /* Start one for the boot CPU: */
Akinobu Mita07dccf32006-09-29 02:00:22 -07005966 err = migration_call(&migration_notifier, CPU_UP_PREPARE, cpu);
5967 BUG_ON(err == NOTIFY_BAD);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005968 migration_call(&migration_notifier, CPU_ONLINE, cpu);
5969 register_cpu_notifier(&migration_notifier);
Eduard - Gabriel Munteanu7babe8d2008-07-25 19:45:11 -07005970
Thomas Gleixnera004cd42009-07-21 09:54:05 +02005971 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005972}
Eduard - Gabriel Munteanu7babe8d2008-07-25 19:45:11 -07005973early_initcall(migration_init);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005974#endif
5975
5976#ifdef CONFIG_SMP
Christoph Lameter476f3532007-05-06 14:48:58 -07005977
Ingo Molnar3e9830d2007-10-15 17:00:13 +02005978#ifdef CONFIG_SCHED_DEBUG
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02005979
Mike Travisf6630112009-11-17 18:22:15 -06005980static __read_mostly int sched_domain_debug_enabled;
5981
5982static int __init sched_domain_debug_setup(char *str)
5983{
5984 sched_domain_debug_enabled = 1;
5985
5986 return 0;
5987}
5988early_param("sched_debug", sched_domain_debug_setup);
5989
Mike Travis7c16ec52008-04-04 18:11:11 -07005990static int sched_domain_debug_one(struct sched_domain *sd, int cpu, int level,
Rusty Russell96f874e22008-11-25 02:35:14 +10305991 struct cpumask *groupmask)
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02005992{
5993 struct sched_group *group = sd->groups;
Mike Travis434d53b2008-04-04 18:11:04 -07005994 char str[256];
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02005995
Rusty Russell968ea6d2008-12-13 21:55:51 +10305996 cpulist_scnprintf(str, sizeof(str), sched_domain_span(sd));
Rusty Russell96f874e22008-11-25 02:35:14 +10305997 cpumask_clear(groupmask);
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02005998
5999 printk(KERN_DEBUG "%*s domain %d: ", level, "", level);
6000
6001 if (!(sd->flags & SD_LOAD_BALANCE)) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006002 printk("does not load-balance\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006003 if (sd->parent)
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006004 printk(KERN_ERR "ERROR: !SD_LOAD_BALANCE domain"
6005 " has parent");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006006 return -1;
6007 }
6008
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006009 printk(KERN_CONT "span %s level %s\n", str, sd->name);
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006010
Rusty Russell758b2cd2008-11-25 02:35:04 +10306011 if (!cpumask_test_cpu(cpu, sched_domain_span(sd))) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006012 printk(KERN_ERR "ERROR: domain->span does not contain "
6013 "CPU%d\n", cpu);
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006014 }
Rusty Russell758b2cd2008-11-25 02:35:04 +10306015 if (!cpumask_test_cpu(cpu, sched_group_cpus(group))) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006016 printk(KERN_ERR "ERROR: domain->groups does not contain"
6017 " CPU%d\n", cpu);
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006018 }
6019
6020 printk(KERN_DEBUG "%*s groups:", level + 1, "");
6021 do {
6022 if (!group) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006023 printk("\n");
6024 printk(KERN_ERR "ERROR: group is NULL\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006025 break;
6026 }
6027
Peter Zijlstra18a38852009-09-01 10:34:39 +02006028 if (!group->cpu_power) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006029 printk(KERN_CONT "\n");
6030 printk(KERN_ERR "ERROR: domain->cpu_power not "
6031 "set\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006032 break;
6033 }
6034
Rusty Russell758b2cd2008-11-25 02:35:04 +10306035 if (!cpumask_weight(sched_group_cpus(group))) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006036 printk(KERN_CONT "\n");
6037 printk(KERN_ERR "ERROR: empty group\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006038 break;
6039 }
6040
Rusty Russell758b2cd2008-11-25 02:35:04 +10306041 if (cpumask_intersects(groupmask, sched_group_cpus(group))) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006042 printk(KERN_CONT "\n");
6043 printk(KERN_ERR "ERROR: repeated CPUs\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006044 break;
6045 }
6046
Rusty Russell758b2cd2008-11-25 02:35:04 +10306047 cpumask_or(groupmask, groupmask, sched_group_cpus(group));
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006048
Rusty Russell968ea6d2008-12-13 21:55:51 +10306049 cpulist_scnprintf(str, sizeof(str), sched_group_cpus(group));
Gautham R Shenoy381512c2009-04-14 09:09:36 +05306050
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006051 printk(KERN_CONT " %s", str);
Peter Zijlstra18a38852009-09-01 10:34:39 +02006052 if (group->cpu_power != SCHED_LOAD_SCALE) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006053 printk(KERN_CONT " (cpu_power = %d)",
6054 group->cpu_power);
Gautham R Shenoy381512c2009-04-14 09:09:36 +05306055 }
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006056
6057 group = group->next;
6058 } while (group != sd->groups);
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006059 printk(KERN_CONT "\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006060
Rusty Russell758b2cd2008-11-25 02:35:04 +10306061 if (!cpumask_equal(sched_domain_span(sd), groupmask))
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006062 printk(KERN_ERR "ERROR: groups don't span domain->span\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006063
Rusty Russell758b2cd2008-11-25 02:35:04 +10306064 if (sd->parent &&
6065 !cpumask_subset(groupmask, sched_domain_span(sd->parent)))
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006066 printk(KERN_ERR "ERROR: parent span is not a superset "
6067 "of domain->span\n");
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006068 return 0;
6069}
6070
Linus Torvalds1da177e2005-04-16 15:20:36 -07006071static void sched_domain_debug(struct sched_domain *sd, int cpu)
6072{
Rusty Russelld5dd3db2008-11-25 02:35:12 +10306073 cpumask_var_t groupmask;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006074 int level = 0;
6075
Mike Travisf6630112009-11-17 18:22:15 -06006076 if (!sched_domain_debug_enabled)
6077 return;
6078
Nick Piggin41c7ce92005-06-25 14:57:24 -07006079 if (!sd) {
6080 printk(KERN_DEBUG "CPU%d attaching NULL sched-domain.\n", cpu);
6081 return;
6082 }
6083
Linus Torvalds1da177e2005-04-16 15:20:36 -07006084 printk(KERN_DEBUG "CPU%d attaching sched-domain:\n", cpu);
6085
Rusty Russelld5dd3db2008-11-25 02:35:12 +10306086 if (!alloc_cpumask_var(&groupmask, GFP_KERNEL)) {
Mike Travis7c16ec52008-04-04 18:11:11 -07006087 printk(KERN_DEBUG "Cannot load-balance (out of memory)\n");
6088 return;
6089 }
6090
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006091 for (;;) {
Mike Travis7c16ec52008-04-04 18:11:11 -07006092 if (sched_domain_debug_one(sd, cpu, level, groupmask))
Linus Torvalds1da177e2005-04-16 15:20:36 -07006093 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006094 level++;
6095 sd = sd->parent;
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006096 if (!sd)
Ingo Molnar4dcf6af2007-10-24 18:23:48 +02006097 break;
6098 }
Rusty Russelld5dd3db2008-11-25 02:35:12 +10306099 free_cpumask_var(groupmask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006100}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006101#else /* !CONFIG_SCHED_DEBUG */
Ingo Molnar48f24c42006-07-03 00:25:40 -07006102# define sched_domain_debug(sd, cpu) do { } while (0)
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006103#endif /* CONFIG_SCHED_DEBUG */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006104
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006105static int sd_degenerate(struct sched_domain *sd)
Suresh Siddha245af2c2005-06-25 14:57:25 -07006106{
Rusty Russell758b2cd2008-11-25 02:35:04 +10306107 if (cpumask_weight(sched_domain_span(sd)) == 1)
Suresh Siddha245af2c2005-06-25 14:57:25 -07006108 return 1;
6109
6110 /* Following flags need at least 2 groups */
6111 if (sd->flags & (SD_LOAD_BALANCE |
6112 SD_BALANCE_NEWIDLE |
6113 SD_BALANCE_FORK |
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006114 SD_BALANCE_EXEC |
6115 SD_SHARE_CPUPOWER |
6116 SD_SHARE_PKG_RESOURCES)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07006117 if (sd->groups != sd->groups->next)
6118 return 0;
6119 }
6120
6121 /* Following flags don't use groups */
Peter Zijlstrac88d5912009-09-10 13:50:02 +02006122 if (sd->flags & (SD_WAKE_AFFINE))
Suresh Siddha245af2c2005-06-25 14:57:25 -07006123 return 0;
6124
6125 return 1;
6126}
6127
Ingo Molnar48f24c42006-07-03 00:25:40 -07006128static int
6129sd_parent_degenerate(struct sched_domain *sd, struct sched_domain *parent)
Suresh Siddha245af2c2005-06-25 14:57:25 -07006130{
6131 unsigned long cflags = sd->flags, pflags = parent->flags;
6132
6133 if (sd_degenerate(parent))
6134 return 1;
6135
Rusty Russell758b2cd2008-11-25 02:35:04 +10306136 if (!cpumask_equal(sched_domain_span(sd), sched_domain_span(parent)))
Suresh Siddha245af2c2005-06-25 14:57:25 -07006137 return 0;
6138
Suresh Siddha245af2c2005-06-25 14:57:25 -07006139 /* Flags needing groups don't count if only 1 group in parent */
6140 if (parent->groups == parent->groups->next) {
6141 pflags &= ~(SD_LOAD_BALANCE |
6142 SD_BALANCE_NEWIDLE |
6143 SD_BALANCE_FORK |
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006144 SD_BALANCE_EXEC |
6145 SD_SHARE_CPUPOWER |
6146 SD_SHARE_PKG_RESOURCES);
Ken Chen54364992008-12-07 18:47:37 -08006147 if (nr_node_ids == 1)
6148 pflags &= ~SD_SERIALIZE;
Suresh Siddha245af2c2005-06-25 14:57:25 -07006149 }
6150 if (~cflags & pflags)
6151 return 0;
6152
6153 return 1;
6154}
6155
Rusty Russellc6c49272008-11-25 02:35:05 +10306156static void free_rootdomain(struct root_domain *rd)
6157{
Peter Zijlstra047106a2009-11-16 10:28:09 +01006158 synchronize_sched();
6159
Rusty Russell68e74562008-11-25 02:35:13 +10306160 cpupri_cleanup(&rd->cpupri);
6161
Rusty Russellc6c49272008-11-25 02:35:05 +10306162 free_cpumask_var(rd->rto_mask);
6163 free_cpumask_var(rd->online);
6164 free_cpumask_var(rd->span);
6165 kfree(rd);
6166}
6167
Gregory Haskins57d885f2008-01-25 21:08:18 +01006168static void rq_attach_root(struct rq *rq, struct root_domain *rd)
6169{
Ingo Molnara0490fa2009-02-12 11:35:40 +01006170 struct root_domain *old_rd = NULL;
Gregory Haskins57d885f2008-01-25 21:08:18 +01006171 unsigned long flags;
Gregory Haskins57d885f2008-01-25 21:08:18 +01006172
Thomas Gleixner05fa7852009-11-17 14:28:38 +01006173 raw_spin_lock_irqsave(&rq->lock, flags);
Gregory Haskins57d885f2008-01-25 21:08:18 +01006174
6175 if (rq->rd) {
Ingo Molnara0490fa2009-02-12 11:35:40 +01006176 old_rd = rq->rd;
Gregory Haskins57d885f2008-01-25 21:08:18 +01006177
Rusty Russellc6c49272008-11-25 02:35:05 +10306178 if (cpumask_test_cpu(rq->cpu, old_rd->online))
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04006179 set_rq_offline(rq);
Gregory Haskins57d885f2008-01-25 21:08:18 +01006180
Rusty Russellc6c49272008-11-25 02:35:05 +10306181 cpumask_clear_cpu(rq->cpu, old_rd->span);
Gregory Haskinsdc938522008-01-25 21:08:26 +01006182
Ingo Molnara0490fa2009-02-12 11:35:40 +01006183 /*
6184 * If we dont want to free the old_rt yet then
6185 * set old_rd to NULL to skip the freeing later
6186 * in this function:
6187 */
6188 if (!atomic_dec_and_test(&old_rd->refcount))
6189 old_rd = NULL;
Gregory Haskins57d885f2008-01-25 21:08:18 +01006190 }
6191
6192 atomic_inc(&rd->refcount);
6193 rq->rd = rd;
6194
Rusty Russellc6c49272008-11-25 02:35:05 +10306195 cpumask_set_cpu(rq->cpu, rd->span);
Gregory Haskins00aec932009-07-30 10:57:23 -04006196 if (cpumask_test_cpu(rq->cpu, cpu_active_mask))
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04006197 set_rq_online(rq);
Gregory Haskins57d885f2008-01-25 21:08:18 +01006198
Thomas Gleixner05fa7852009-11-17 14:28:38 +01006199 raw_spin_unlock_irqrestore(&rq->lock, flags);
Ingo Molnara0490fa2009-02-12 11:35:40 +01006200
6201 if (old_rd)
6202 free_rootdomain(old_rd);
Gregory Haskins57d885f2008-01-25 21:08:18 +01006203}
6204
Li Zefanfd5e1b52009-06-15 13:34:19 +08006205static int init_rootdomain(struct root_domain *rd, bool bootmem)
Gregory Haskins57d885f2008-01-25 21:08:18 +01006206{
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03006207 gfp_t gfp = GFP_KERNEL;
6208
Gregory Haskins57d885f2008-01-25 21:08:18 +01006209 memset(rd, 0, sizeof(*rd));
6210
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03006211 if (bootmem)
6212 gfp = GFP_NOWAIT;
Gregory Haskins6e0534f2008-05-12 21:21:01 +02006213
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03006214 if (!alloc_cpumask_var(&rd->span, gfp))
Li Zefan0c910d22009-01-06 17:39:06 +08006215 goto out;
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03006216 if (!alloc_cpumask_var(&rd->online, gfp))
Rusty Russellc6c49272008-11-25 02:35:05 +10306217 goto free_span;
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03006218 if (!alloc_cpumask_var(&rd->rto_mask, gfp))
Rusty Russellc6c49272008-11-25 02:35:05 +10306219 goto free_online;
Gregory Haskins6e0534f2008-05-12 21:21:01 +02006220
Pekka Enberg0fb53022009-06-11 08:41:22 +03006221 if (cpupri_init(&rd->cpupri, bootmem) != 0)
Rusty Russell68e74562008-11-25 02:35:13 +10306222 goto free_rto_mask;
Rusty Russellc6c49272008-11-25 02:35:05 +10306223 return 0;
6224
Rusty Russell68e74562008-11-25 02:35:13 +10306225free_rto_mask:
6226 free_cpumask_var(rd->rto_mask);
Rusty Russellc6c49272008-11-25 02:35:05 +10306227free_online:
6228 free_cpumask_var(rd->online);
6229free_span:
6230 free_cpumask_var(rd->span);
Li Zefan0c910d22009-01-06 17:39:06 +08006231out:
Rusty Russellc6c49272008-11-25 02:35:05 +10306232 return -ENOMEM;
Gregory Haskins57d885f2008-01-25 21:08:18 +01006233}
6234
6235static void init_defrootdomain(void)
6236{
Rusty Russellc6c49272008-11-25 02:35:05 +10306237 init_rootdomain(&def_root_domain, true);
6238
Gregory Haskins57d885f2008-01-25 21:08:18 +01006239 atomic_set(&def_root_domain.refcount, 1);
6240}
6241
Gregory Haskinsdc938522008-01-25 21:08:26 +01006242static struct root_domain *alloc_rootdomain(void)
Gregory Haskins57d885f2008-01-25 21:08:18 +01006243{
6244 struct root_domain *rd;
6245
6246 rd = kmalloc(sizeof(*rd), GFP_KERNEL);
6247 if (!rd)
6248 return NULL;
6249
Rusty Russellc6c49272008-11-25 02:35:05 +10306250 if (init_rootdomain(rd, false) != 0) {
6251 kfree(rd);
6252 return NULL;
6253 }
Gregory Haskins57d885f2008-01-25 21:08:18 +01006254
6255 return rd;
6256}
6257
Linus Torvalds1da177e2005-04-16 15:20:36 -07006258/*
Ingo Molnar0eab9142008-01-25 21:08:19 +01006259 * Attach the domain 'sd' to 'cpu' as its base domain. Callers must
Linus Torvalds1da177e2005-04-16 15:20:36 -07006260 * hold the hotplug lock.
6261 */
Ingo Molnar0eab9142008-01-25 21:08:19 +01006262static void
6263cpu_attach_domain(struct sched_domain *sd, struct root_domain *rd, int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006264{
Ingo Molnar70b97a72006-07-03 00:25:42 -07006265 struct rq *rq = cpu_rq(cpu);
Suresh Siddha245af2c2005-06-25 14:57:25 -07006266 struct sched_domain *tmp;
6267
6268 /* Remove the sched domains which do not contribute to scheduling. */
Li Zefanf29c9b12008-11-06 09:45:16 +08006269 for (tmp = sd; tmp; ) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07006270 struct sched_domain *parent = tmp->parent;
6271 if (!parent)
6272 break;
Li Zefanf29c9b12008-11-06 09:45:16 +08006273
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006274 if (sd_parent_degenerate(tmp, parent)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07006275 tmp->parent = parent->parent;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006276 if (parent->parent)
6277 parent->parent->child = tmp;
Li Zefanf29c9b12008-11-06 09:45:16 +08006278 } else
6279 tmp = tmp->parent;
Suresh Siddha245af2c2005-06-25 14:57:25 -07006280 }
6281
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006282 if (sd && sd_degenerate(sd)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07006283 sd = sd->parent;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006284 if (sd)
6285 sd->child = NULL;
6286 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07006287
6288 sched_domain_debug(sd, cpu);
6289
Gregory Haskins57d885f2008-01-25 21:08:18 +01006290 rq_attach_root(rq, rd);
Nick Piggin674311d2005-06-25 14:57:27 -07006291 rcu_assign_pointer(rq->sd, sd);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006292}
6293
6294/* cpus with isolated domains */
Rusty Russelldcc30a32008-11-25 02:35:12 +10306295static cpumask_var_t cpu_isolated_map;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006296
6297/* Setup the mask of cpus configured for isolated domains */
6298static int __init isolated_cpu_setup(char *str)
6299{
Rusty Russellbdddd292009-12-02 14:09:16 +10306300 alloc_bootmem_cpumask_var(&cpu_isolated_map);
Rusty Russell968ea6d2008-12-13 21:55:51 +10306301 cpulist_parse(str, cpu_isolated_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006302 return 1;
6303}
6304
Ingo Molnar8927f492007-10-15 17:00:13 +02006305__setup("isolcpus=", isolated_cpu_setup);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006306
6307/*
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006308 * init_sched_build_groups takes the cpumask we wish to span, and a pointer
6309 * to a function which identifies what group(along with sched group) a CPU
Rusty Russell96f874e22008-11-25 02:35:14 +10306310 * belongs to. The return value of group_fn must be a >= 0 and < nr_cpu_ids
6311 * (due to the fact that we keep track of groups covered with a struct cpumask).
Linus Torvalds1da177e2005-04-16 15:20:36 -07006312 *
6313 * init_sched_build_groups will build a circular linked list of the groups
6314 * covered by the given span, and will set each group's ->cpumask correctly,
6315 * and ->cpu_power to 0.
6316 */
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006317static void
Rusty Russell96f874e22008-11-25 02:35:14 +10306318init_sched_build_groups(const struct cpumask *span,
6319 const struct cpumask *cpu_map,
6320 int (*group_fn)(int cpu, const struct cpumask *cpu_map,
Mike Travis7c16ec52008-04-04 18:11:11 -07006321 struct sched_group **sg,
Rusty Russell96f874e22008-11-25 02:35:14 +10306322 struct cpumask *tmpmask),
6323 struct cpumask *covered, struct cpumask *tmpmask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006324{
6325 struct sched_group *first = NULL, *last = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006326 int i;
6327
Rusty Russell96f874e22008-11-25 02:35:14 +10306328 cpumask_clear(covered);
Mike Travis7c16ec52008-04-04 18:11:11 -07006329
Rusty Russellabcd0832008-11-25 02:35:02 +10306330 for_each_cpu(i, span) {
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006331 struct sched_group *sg;
Mike Travis7c16ec52008-04-04 18:11:11 -07006332 int group = group_fn(i, cpu_map, &sg, tmpmask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006333 int j;
6334
Rusty Russell758b2cd2008-11-25 02:35:04 +10306335 if (cpumask_test_cpu(i, covered))
Linus Torvalds1da177e2005-04-16 15:20:36 -07006336 continue;
6337
Rusty Russell758b2cd2008-11-25 02:35:04 +10306338 cpumask_clear(sched_group_cpus(sg));
Peter Zijlstra18a38852009-09-01 10:34:39 +02006339 sg->cpu_power = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006340
Rusty Russellabcd0832008-11-25 02:35:02 +10306341 for_each_cpu(j, span) {
Mike Travis7c16ec52008-04-04 18:11:11 -07006342 if (group_fn(j, cpu_map, NULL, tmpmask) != group)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006343 continue;
6344
Rusty Russell96f874e22008-11-25 02:35:14 +10306345 cpumask_set_cpu(j, covered);
Rusty Russell758b2cd2008-11-25 02:35:04 +10306346 cpumask_set_cpu(j, sched_group_cpus(sg));
Linus Torvalds1da177e2005-04-16 15:20:36 -07006347 }
6348 if (!first)
6349 first = sg;
6350 if (last)
6351 last->next = sg;
6352 last = sg;
6353 }
6354 last->next = first;
6355}
6356
John Hawkes9c1cfda2005-09-06 15:18:14 -07006357#define SD_NODES_PER_DOMAIN 16
Linus Torvalds1da177e2005-04-16 15:20:36 -07006358
John Hawkes9c1cfda2005-09-06 15:18:14 -07006359#ifdef CONFIG_NUMA
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006360
John Hawkes9c1cfda2005-09-06 15:18:14 -07006361/**
6362 * find_next_best_node - find the next node to include in a sched_domain
6363 * @node: node whose sched_domain we're building
6364 * @used_nodes: nodes already in the sched_domain
6365 *
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006366 * Find the next node to include in a given scheduling domain. Simply
John Hawkes9c1cfda2005-09-06 15:18:14 -07006367 * finds the closest node not already in the @used_nodes map.
6368 *
6369 * Should use nodemask_t.
6370 */
Mike Travisc5f59f02008-04-04 18:11:10 -07006371static int find_next_best_node(int node, nodemask_t *used_nodes)
John Hawkes9c1cfda2005-09-06 15:18:14 -07006372{
6373 int i, n, val, min_val, best_node = 0;
6374
6375 min_val = INT_MAX;
6376
Mike Travis076ac2a2008-05-12 21:21:12 +02006377 for (i = 0; i < nr_node_ids; i++) {
John Hawkes9c1cfda2005-09-06 15:18:14 -07006378 /* Start at @node */
Mike Travis076ac2a2008-05-12 21:21:12 +02006379 n = (node + i) % nr_node_ids;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006380
6381 if (!nr_cpus_node(n))
6382 continue;
6383
6384 /* Skip already used nodes */
Mike Travisc5f59f02008-04-04 18:11:10 -07006385 if (node_isset(n, *used_nodes))
John Hawkes9c1cfda2005-09-06 15:18:14 -07006386 continue;
6387
6388 /* Simple min distance search */
6389 val = node_distance(node, n);
6390
6391 if (val < min_val) {
6392 min_val = val;
6393 best_node = n;
6394 }
6395 }
6396
Mike Travisc5f59f02008-04-04 18:11:10 -07006397 node_set(best_node, *used_nodes);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006398 return best_node;
6399}
6400
6401/**
6402 * sched_domain_node_span - get a cpumask for a node's sched_domain
6403 * @node: node whose cpumask we're constructing
Randy Dunlap73486722008-04-22 10:07:22 -07006404 * @span: resulting cpumask
John Hawkes9c1cfda2005-09-06 15:18:14 -07006405 *
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006406 * Given a node, construct a good cpumask for its sched_domain to span. It
John Hawkes9c1cfda2005-09-06 15:18:14 -07006407 * should be one that prevents unnecessary balancing, but also spreads tasks
6408 * out optimally.
6409 */
Rusty Russell96f874e22008-11-25 02:35:14 +10306410static void sched_domain_node_span(int node, struct cpumask *span)
John Hawkes9c1cfda2005-09-06 15:18:14 -07006411{
Mike Travisc5f59f02008-04-04 18:11:10 -07006412 nodemask_t used_nodes;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006413 int i;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006414
Mike Travis6ca09df2008-12-31 18:08:45 -08006415 cpumask_clear(span);
Mike Travisc5f59f02008-04-04 18:11:10 -07006416 nodes_clear(used_nodes);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006417
Mike Travis6ca09df2008-12-31 18:08:45 -08006418 cpumask_or(span, span, cpumask_of_node(node));
Mike Travisc5f59f02008-04-04 18:11:10 -07006419 node_set(node, used_nodes);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006420
6421 for (i = 1; i < SD_NODES_PER_DOMAIN; i++) {
Mike Travisc5f59f02008-04-04 18:11:10 -07006422 int next_node = find_next_best_node(node, &used_nodes);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006423
Mike Travis6ca09df2008-12-31 18:08:45 -08006424 cpumask_or(span, span, cpumask_of_node(next_node));
John Hawkes9c1cfda2005-09-06 15:18:14 -07006425 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07006426}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006427#endif /* CONFIG_NUMA */
John Hawkes9c1cfda2005-09-06 15:18:14 -07006428
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006429int sched_smt_power_savings = 0, sched_mc_power_savings = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006430
John Hawkes9c1cfda2005-09-06 15:18:14 -07006431/*
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306432 * The cpus mask in sched_group and sched_domain hangs off the end.
Ingo Molnar4200efd2009-05-19 09:22:19 +02006433 *
6434 * ( See the the comments in include/linux/sched.h:struct sched_group
6435 * and struct sched_domain. )
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306436 */
6437struct static_sched_group {
6438 struct sched_group sg;
6439 DECLARE_BITMAP(cpus, CONFIG_NR_CPUS);
6440};
6441
6442struct static_sched_domain {
6443 struct sched_domain sd;
6444 DECLARE_BITMAP(span, CONFIG_NR_CPUS);
6445};
6446
Andreas Herrmann49a02c52009-08-18 12:51:52 +02006447struct s_data {
6448#ifdef CONFIG_NUMA
6449 int sd_allnodes;
6450 cpumask_var_t domainspan;
6451 cpumask_var_t covered;
6452 cpumask_var_t notcovered;
6453#endif
6454 cpumask_var_t nodemask;
6455 cpumask_var_t this_sibling_map;
6456 cpumask_var_t this_core_map;
6457 cpumask_var_t send_covered;
6458 cpumask_var_t tmpmask;
6459 struct sched_group **sched_group_nodes;
6460 struct root_domain *rd;
6461};
6462
Andreas Herrmann2109b992009-08-18 12:53:00 +02006463enum s_alloc {
6464 sa_sched_groups = 0,
6465 sa_rootdomain,
6466 sa_tmpmask,
6467 sa_send_covered,
6468 sa_this_core_map,
6469 sa_this_sibling_map,
6470 sa_nodemask,
6471 sa_sched_group_nodes,
6472#ifdef CONFIG_NUMA
6473 sa_notcovered,
6474 sa_covered,
6475 sa_domainspan,
6476#endif
6477 sa_none,
6478};
6479
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306480/*
Ingo Molnar48f24c42006-07-03 00:25:40 -07006481 * SMT sched-domains:
John Hawkes9c1cfda2005-09-06 15:18:14 -07006482 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006483#ifdef CONFIG_SCHED_SMT
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306484static DEFINE_PER_CPU(struct static_sched_domain, cpu_domains);
Tejun Heo1871e522009-10-29 22:34:13 +09006485static DEFINE_PER_CPU(struct static_sched_group, sched_groups);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006486
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006487static int
Rusty Russell96f874e22008-11-25 02:35:14 +10306488cpu_to_cpu_group(int cpu, const struct cpumask *cpu_map,
6489 struct sched_group **sg, struct cpumask *unused)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006490{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006491 if (sg)
Tejun Heo1871e522009-10-29 22:34:13 +09006492 *sg = &per_cpu(sched_groups, cpu).sg;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006493 return cpu;
6494}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006495#endif /* CONFIG_SCHED_SMT */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006496
Ingo Molnar48f24c42006-07-03 00:25:40 -07006497/*
6498 * multi-core sched-domains:
6499 */
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006500#ifdef CONFIG_SCHED_MC
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306501static DEFINE_PER_CPU(struct static_sched_domain, core_domains);
6502static DEFINE_PER_CPU(struct static_sched_group, sched_group_core);
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006503#endif /* CONFIG_SCHED_MC */
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006504
6505#if defined(CONFIG_SCHED_MC) && defined(CONFIG_SCHED_SMT)
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006506static int
Rusty Russell96f874e22008-11-25 02:35:14 +10306507cpu_to_core_group(int cpu, const struct cpumask *cpu_map,
6508 struct sched_group **sg, struct cpumask *mask)
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006509{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006510 int group;
Mike Travis7c16ec52008-04-04 18:11:11 -07006511
Rusty Russellc69fc562009-03-13 14:49:46 +10306512 cpumask_and(mask, topology_thread_cpumask(cpu), cpu_map);
Rusty Russell96f874e22008-11-25 02:35:14 +10306513 group = cpumask_first(mask);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006514 if (sg)
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306515 *sg = &per_cpu(sched_group_core, group).sg;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006516 return group;
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006517}
6518#elif defined(CONFIG_SCHED_MC)
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006519static int
Rusty Russell96f874e22008-11-25 02:35:14 +10306520cpu_to_core_group(int cpu, const struct cpumask *cpu_map,
6521 struct sched_group **sg, struct cpumask *unused)
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006522{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006523 if (sg)
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306524 *sg = &per_cpu(sched_group_core, cpu).sg;
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006525 return cpu;
6526}
6527#endif
6528
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306529static DEFINE_PER_CPU(struct static_sched_domain, phys_domains);
6530static DEFINE_PER_CPU(struct static_sched_group, sched_group_phys);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006531
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01006532static int
Rusty Russell96f874e22008-11-25 02:35:14 +10306533cpu_to_phys_group(int cpu, const struct cpumask *cpu_map,
6534 struct sched_group **sg, struct cpumask *mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006535{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006536 int group;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006537#ifdef CONFIG_SCHED_MC
Mike Travis6ca09df2008-12-31 18:08:45 -08006538 cpumask_and(mask, cpu_coregroup_mask(cpu), cpu_map);
Rusty Russell96f874e22008-11-25 02:35:14 +10306539 group = cpumask_first(mask);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006540#elif defined(CONFIG_SCHED_SMT)
Rusty Russellc69fc562009-03-13 14:49:46 +10306541 cpumask_and(mask, topology_thread_cpumask(cpu), cpu_map);
Rusty Russell96f874e22008-11-25 02:35:14 +10306542 group = cpumask_first(mask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006543#else
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006544 group = cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006545#endif
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006546 if (sg)
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306547 *sg = &per_cpu(sched_group_phys, group).sg;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006548 return group;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006549}
6550
6551#ifdef CONFIG_NUMA
John Hawkes9c1cfda2005-09-06 15:18:14 -07006552/*
6553 * The init_sched_build_groups can't handle what we want to do with node
6554 * groups, so roll our own. Now each node has its own list of groups which
6555 * gets dynamically allocated.
6556 */
Rusty Russell62ea9ce2009-01-11 01:04:16 +01006557static DEFINE_PER_CPU(struct static_sched_domain, node_domains);
Mike Travis434d53b2008-04-04 18:11:04 -07006558static struct sched_group ***sched_group_nodes_bycpu;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006559
Rusty Russell62ea9ce2009-01-11 01:04:16 +01006560static DEFINE_PER_CPU(struct static_sched_domain, allnodes_domains);
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306561static DEFINE_PER_CPU(struct static_sched_group, sched_group_allnodes);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006562
Rusty Russell96f874e22008-11-25 02:35:14 +10306563static int cpu_to_allnodes_group(int cpu, const struct cpumask *cpu_map,
6564 struct sched_group **sg,
6565 struct cpumask *nodemask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006566{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006567 int group;
6568
Mike Travis6ca09df2008-12-31 18:08:45 -08006569 cpumask_and(nodemask, cpumask_of_node(cpu_to_node(cpu)), cpu_map);
Rusty Russell96f874e22008-11-25 02:35:14 +10306570 group = cpumask_first(nodemask);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006571
6572 if (sg)
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306573 *sg = &per_cpu(sched_group_allnodes, group).sg;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006574 return group;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006575}
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006576
Siddha, Suresh B08069032006-03-27 01:15:23 -08006577static void init_numa_sched_groups_power(struct sched_group *group_head)
6578{
6579 struct sched_group *sg = group_head;
6580 int j;
6581
6582 if (!sg)
6583 return;
Andi Kleen3a5c3592007-10-15 17:00:14 +02006584 do {
Rusty Russell758b2cd2008-11-25 02:35:04 +10306585 for_each_cpu(j, sched_group_cpus(sg)) {
Andi Kleen3a5c3592007-10-15 17:00:14 +02006586 struct sched_domain *sd;
Siddha, Suresh B08069032006-03-27 01:15:23 -08006587
Rusty Russell6c99e9a2008-11-25 02:35:04 +10306588 sd = &per_cpu(phys_domains, j).sd;
Miao Xie13318a72009-04-15 09:59:10 +08006589 if (j != group_first_cpu(sd->groups)) {
Andi Kleen3a5c3592007-10-15 17:00:14 +02006590 /*
6591 * Only add "power" once for each
6592 * physical package.
6593 */
6594 continue;
6595 }
6596
Peter Zijlstra18a38852009-09-01 10:34:39 +02006597 sg->cpu_power += sd->groups->cpu_power;
Siddha, Suresh B08069032006-03-27 01:15:23 -08006598 }
Andi Kleen3a5c3592007-10-15 17:00:14 +02006599 sg = sg->next;
6600 } while (sg != group_head);
Siddha, Suresh B08069032006-03-27 01:15:23 -08006601}
Andreas Herrmann0601a882009-08-18 13:01:11 +02006602
6603static int build_numa_sched_groups(struct s_data *d,
6604 const struct cpumask *cpu_map, int num)
6605{
6606 struct sched_domain *sd;
6607 struct sched_group *sg, *prev;
6608 int n, j;
6609
6610 cpumask_clear(d->covered);
6611 cpumask_and(d->nodemask, cpumask_of_node(num), cpu_map);
6612 if (cpumask_empty(d->nodemask)) {
6613 d->sched_group_nodes[num] = NULL;
6614 goto out;
6615 }
6616
6617 sched_domain_node_span(num, d->domainspan);
6618 cpumask_and(d->domainspan, d->domainspan, cpu_map);
6619
6620 sg = kmalloc_node(sizeof(struct sched_group) + cpumask_size(),
6621 GFP_KERNEL, num);
6622 if (!sg) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006623 printk(KERN_WARNING "Can not alloc domain group for node %d\n",
6624 num);
Andreas Herrmann0601a882009-08-18 13:01:11 +02006625 return -ENOMEM;
6626 }
6627 d->sched_group_nodes[num] = sg;
6628
6629 for_each_cpu(j, d->nodemask) {
6630 sd = &per_cpu(node_domains, j).sd;
6631 sd->groups = sg;
6632 }
6633
Peter Zijlstra18a38852009-09-01 10:34:39 +02006634 sg->cpu_power = 0;
Andreas Herrmann0601a882009-08-18 13:01:11 +02006635 cpumask_copy(sched_group_cpus(sg), d->nodemask);
6636 sg->next = sg;
6637 cpumask_or(d->covered, d->covered, d->nodemask);
6638
6639 prev = sg;
6640 for (j = 0; j < nr_node_ids; j++) {
6641 n = (num + j) % nr_node_ids;
6642 cpumask_complement(d->notcovered, d->covered);
6643 cpumask_and(d->tmpmask, d->notcovered, cpu_map);
6644 cpumask_and(d->tmpmask, d->tmpmask, d->domainspan);
6645 if (cpumask_empty(d->tmpmask))
6646 break;
6647 cpumask_and(d->tmpmask, d->tmpmask, cpumask_of_node(n));
6648 if (cpumask_empty(d->tmpmask))
6649 continue;
6650 sg = kmalloc_node(sizeof(struct sched_group) + cpumask_size(),
6651 GFP_KERNEL, num);
6652 if (!sg) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006653 printk(KERN_WARNING
6654 "Can not alloc domain group for node %d\n", j);
Andreas Herrmann0601a882009-08-18 13:01:11 +02006655 return -ENOMEM;
6656 }
Peter Zijlstra18a38852009-09-01 10:34:39 +02006657 sg->cpu_power = 0;
Andreas Herrmann0601a882009-08-18 13:01:11 +02006658 cpumask_copy(sched_group_cpus(sg), d->tmpmask);
6659 sg->next = prev->next;
6660 cpumask_or(d->covered, d->covered, d->tmpmask);
6661 prev->next = sg;
6662 prev = sg;
6663 }
6664out:
6665 return 0;
6666}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006667#endif /* CONFIG_NUMA */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006668
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006669#ifdef CONFIG_NUMA
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006670/* Free memory allocated for various sched_group structures */
Rusty Russell96f874e22008-11-25 02:35:14 +10306671static void free_sched_groups(const struct cpumask *cpu_map,
6672 struct cpumask *nodemask)
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006673{
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006674 int cpu, i;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006675
Rusty Russellabcd0832008-11-25 02:35:02 +10306676 for_each_cpu(cpu, cpu_map) {
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006677 struct sched_group **sched_group_nodes
6678 = sched_group_nodes_bycpu[cpu];
6679
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006680 if (!sched_group_nodes)
6681 continue;
6682
Mike Travis076ac2a2008-05-12 21:21:12 +02006683 for (i = 0; i < nr_node_ids; i++) {
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006684 struct sched_group *oldsg, *sg = sched_group_nodes[i];
6685
Mike Travis6ca09df2008-12-31 18:08:45 -08006686 cpumask_and(nodemask, cpumask_of_node(i), cpu_map);
Rusty Russell96f874e22008-11-25 02:35:14 +10306687 if (cpumask_empty(nodemask))
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006688 continue;
6689
6690 if (sg == NULL)
6691 continue;
6692 sg = sg->next;
6693next_sg:
6694 oldsg = sg;
6695 sg = sg->next;
6696 kfree(oldsg);
6697 if (oldsg != sched_group_nodes[i])
6698 goto next_sg;
6699 }
6700 kfree(sched_group_nodes);
6701 sched_group_nodes_bycpu[cpu] = NULL;
6702 }
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006703}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006704#else /* !CONFIG_NUMA */
Rusty Russell96f874e22008-11-25 02:35:14 +10306705static void free_sched_groups(const struct cpumask *cpu_map,
6706 struct cpumask *nodemask)
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006707{
6708}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02006709#endif /* CONFIG_NUMA */
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006710
Linus Torvalds1da177e2005-04-16 15:20:36 -07006711/*
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006712 * Initialize sched groups cpu_power.
6713 *
6714 * cpu_power indicates the capacity of sched group, which is used while
6715 * distributing the load between different sched groups in a sched domain.
6716 * Typically cpu_power for all the groups in a sched domain will be same unless
6717 * there are asymmetries in the topology. If there are asymmetries, group
6718 * having more cpu_power will pickup more load compared to the group having
6719 * less cpu_power.
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006720 */
6721static void init_sched_groups_power(int cpu, struct sched_domain *sd)
6722{
6723 struct sched_domain *child;
6724 struct sched_group *group;
Peter Zijlstraf93e65c2009-09-01 10:34:32 +02006725 long power;
6726 int weight;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006727
6728 WARN_ON(!sd || !sd->groups);
6729
Miao Xie13318a72009-04-15 09:59:10 +08006730 if (cpu != group_first_cpu(sd->groups))
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006731 return;
6732
6733 child = sd->child;
6734
Peter Zijlstra18a38852009-09-01 10:34:39 +02006735 sd->groups->cpu_power = 0;
Eric Dumazet5517d862007-05-08 00:32:57 -07006736
Peter Zijlstraf93e65c2009-09-01 10:34:32 +02006737 if (!child) {
6738 power = SCHED_LOAD_SCALE;
6739 weight = cpumask_weight(sched_domain_span(sd));
6740 /*
6741 * SMT siblings share the power of a single core.
Peter Zijlstraa52bfd732009-09-01 10:34:35 +02006742 * Usually multiple threads get a better yield out of
6743 * that one core than a single thread would have,
6744 * reflect that in sd->smt_gain.
Peter Zijlstraf93e65c2009-09-01 10:34:32 +02006745 */
Peter Zijlstraa52bfd732009-09-01 10:34:35 +02006746 if ((sd->flags & SD_SHARE_CPUPOWER) && weight > 1) {
6747 power *= sd->smt_gain;
Peter Zijlstraf93e65c2009-09-01 10:34:32 +02006748 power /= weight;
Peter Zijlstraa52bfd732009-09-01 10:34:35 +02006749 power >>= SCHED_LOAD_SHIFT;
6750 }
Peter Zijlstra18a38852009-09-01 10:34:39 +02006751 sd->groups->cpu_power += power;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006752 return;
6753 }
6754
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006755 /*
Peter Zijlstraf93e65c2009-09-01 10:34:32 +02006756 * Add cpu_power of each child group to this groups cpu_power.
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006757 */
6758 group = child->groups;
6759 do {
Peter Zijlstra18a38852009-09-01 10:34:39 +02006760 sd->groups->cpu_power += group->cpu_power;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006761 group = group->next;
6762 } while (group != child->groups);
6763}
6764
6765/*
Mike Travis7c16ec52008-04-04 18:11:11 -07006766 * Initializers for schedule domains
6767 * Non-inlined to reduce accumulated stack pressure in build_sched_domains()
6768 */
6769
Ingo Molnara5d8c342008-10-09 11:35:51 +02006770#ifdef CONFIG_SCHED_DEBUG
6771# define SD_INIT_NAME(sd, type) sd->name = #type
6772#else
6773# define SD_INIT_NAME(sd, type) do { } while (0)
6774#endif
6775
Mike Travis7c16ec52008-04-04 18:11:11 -07006776#define SD_INIT(sd, type) sd_init_##type(sd)
Ingo Molnara5d8c342008-10-09 11:35:51 +02006777
Mike Travis7c16ec52008-04-04 18:11:11 -07006778#define SD_INIT_FUNC(type) \
6779static noinline void sd_init_##type(struct sched_domain *sd) \
6780{ \
6781 memset(sd, 0, sizeof(*sd)); \
6782 *sd = SD_##type##_INIT; \
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09006783 sd->level = SD_LV_##type; \
Ingo Molnara5d8c342008-10-09 11:35:51 +02006784 SD_INIT_NAME(sd, type); \
Mike Travis7c16ec52008-04-04 18:11:11 -07006785}
6786
6787SD_INIT_FUNC(CPU)
6788#ifdef CONFIG_NUMA
6789 SD_INIT_FUNC(ALLNODES)
6790 SD_INIT_FUNC(NODE)
6791#endif
6792#ifdef CONFIG_SCHED_SMT
6793 SD_INIT_FUNC(SIBLING)
6794#endif
6795#ifdef CONFIG_SCHED_MC
6796 SD_INIT_FUNC(MC)
6797#endif
6798
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09006799static int default_relax_domain_level = -1;
6800
6801static int __init setup_relax_domain_level(char *str)
6802{
Li Zefan30e0e172008-05-13 10:27:17 +08006803 unsigned long val;
6804
6805 val = simple_strtoul(str, NULL, 0);
6806 if (val < SD_LV_MAX)
6807 default_relax_domain_level = val;
6808
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09006809 return 1;
6810}
6811__setup("relax_domain_level=", setup_relax_domain_level);
6812
6813static void set_domain_attribute(struct sched_domain *sd,
6814 struct sched_domain_attr *attr)
6815{
6816 int request;
6817
6818 if (!attr || attr->relax_domain_level < 0) {
6819 if (default_relax_domain_level < 0)
6820 return;
6821 else
6822 request = default_relax_domain_level;
6823 } else
6824 request = attr->relax_domain_level;
6825 if (request < sd->level) {
6826 /* turn off idle balance on this domain */
Peter Zijlstrac88d5912009-09-10 13:50:02 +02006827 sd->flags &= ~(SD_BALANCE_WAKE|SD_BALANCE_NEWIDLE);
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09006828 } else {
6829 /* turn on idle balance on this domain */
Peter Zijlstrac88d5912009-09-10 13:50:02 +02006830 sd->flags |= (SD_BALANCE_WAKE|SD_BALANCE_NEWIDLE);
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09006831 }
6832}
6833
Andreas Herrmann2109b992009-08-18 12:53:00 +02006834static void __free_domain_allocs(struct s_data *d, enum s_alloc what,
6835 const struct cpumask *cpu_map)
6836{
6837 switch (what) {
6838 case sa_sched_groups:
6839 free_sched_groups(cpu_map, d->tmpmask); /* fall through */
6840 d->sched_group_nodes = NULL;
6841 case sa_rootdomain:
6842 free_rootdomain(d->rd); /* fall through */
6843 case sa_tmpmask:
6844 free_cpumask_var(d->tmpmask); /* fall through */
6845 case sa_send_covered:
6846 free_cpumask_var(d->send_covered); /* fall through */
6847 case sa_this_core_map:
6848 free_cpumask_var(d->this_core_map); /* fall through */
6849 case sa_this_sibling_map:
6850 free_cpumask_var(d->this_sibling_map); /* fall through */
6851 case sa_nodemask:
6852 free_cpumask_var(d->nodemask); /* fall through */
6853 case sa_sched_group_nodes:
6854#ifdef CONFIG_NUMA
6855 kfree(d->sched_group_nodes); /* fall through */
6856 case sa_notcovered:
6857 free_cpumask_var(d->notcovered); /* fall through */
6858 case sa_covered:
6859 free_cpumask_var(d->covered); /* fall through */
6860 case sa_domainspan:
6861 free_cpumask_var(d->domainspan); /* fall through */
6862#endif
6863 case sa_none:
6864 break;
6865 }
6866}
6867
6868static enum s_alloc __visit_domain_allocation_hell(struct s_data *d,
6869 const struct cpumask *cpu_map)
6870{
6871#ifdef CONFIG_NUMA
6872 if (!alloc_cpumask_var(&d->domainspan, GFP_KERNEL))
6873 return sa_none;
6874 if (!alloc_cpumask_var(&d->covered, GFP_KERNEL))
6875 return sa_domainspan;
6876 if (!alloc_cpumask_var(&d->notcovered, GFP_KERNEL))
6877 return sa_covered;
6878 /* Allocate the per-node list of sched groups */
6879 d->sched_group_nodes = kcalloc(nr_node_ids,
6880 sizeof(struct sched_group *), GFP_KERNEL);
6881 if (!d->sched_group_nodes) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006882 printk(KERN_WARNING "Can not alloc sched group node list\n");
Andreas Herrmann2109b992009-08-18 12:53:00 +02006883 return sa_notcovered;
6884 }
6885 sched_group_nodes_bycpu[cpumask_first(cpu_map)] = d->sched_group_nodes;
6886#endif
6887 if (!alloc_cpumask_var(&d->nodemask, GFP_KERNEL))
6888 return sa_sched_group_nodes;
6889 if (!alloc_cpumask_var(&d->this_sibling_map, GFP_KERNEL))
6890 return sa_nodemask;
6891 if (!alloc_cpumask_var(&d->this_core_map, GFP_KERNEL))
6892 return sa_this_sibling_map;
6893 if (!alloc_cpumask_var(&d->send_covered, GFP_KERNEL))
6894 return sa_this_core_map;
6895 if (!alloc_cpumask_var(&d->tmpmask, GFP_KERNEL))
6896 return sa_send_covered;
6897 d->rd = alloc_rootdomain();
6898 if (!d->rd) {
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01006899 printk(KERN_WARNING "Cannot alloc root domain\n");
Andreas Herrmann2109b992009-08-18 12:53:00 +02006900 return sa_tmpmask;
6901 }
6902 return sa_rootdomain;
6903}
6904
Andreas Herrmann7f4588f2009-08-18 12:54:06 +02006905static struct sched_domain *__build_numa_sched_domains(struct s_data *d,
6906 const struct cpumask *cpu_map, struct sched_domain_attr *attr, int i)
6907{
6908 struct sched_domain *sd = NULL;
6909#ifdef CONFIG_NUMA
6910 struct sched_domain *parent;
6911
6912 d->sd_allnodes = 0;
6913 if (cpumask_weight(cpu_map) >
6914 SD_NODES_PER_DOMAIN * cpumask_weight(d->nodemask)) {
6915 sd = &per_cpu(allnodes_domains, i).sd;
6916 SD_INIT(sd, ALLNODES);
6917 set_domain_attribute(sd, attr);
6918 cpumask_copy(sched_domain_span(sd), cpu_map);
6919 cpu_to_allnodes_group(i, cpu_map, &sd->groups, d->tmpmask);
6920 d->sd_allnodes = 1;
6921 }
6922 parent = sd;
6923
6924 sd = &per_cpu(node_domains, i).sd;
6925 SD_INIT(sd, NODE);
6926 set_domain_attribute(sd, attr);
6927 sched_domain_node_span(cpu_to_node(i), sched_domain_span(sd));
6928 sd->parent = parent;
6929 if (parent)
6930 parent->child = sd;
6931 cpumask_and(sched_domain_span(sd), sched_domain_span(sd), cpu_map);
6932#endif
6933 return sd;
6934}
6935
Andreas Herrmann87cce662009-08-18 12:54:55 +02006936static struct sched_domain *__build_cpu_sched_domain(struct s_data *d,
6937 const struct cpumask *cpu_map, struct sched_domain_attr *attr,
6938 struct sched_domain *parent, int i)
6939{
6940 struct sched_domain *sd;
6941 sd = &per_cpu(phys_domains, i).sd;
6942 SD_INIT(sd, CPU);
6943 set_domain_attribute(sd, attr);
6944 cpumask_copy(sched_domain_span(sd), d->nodemask);
6945 sd->parent = parent;
6946 if (parent)
6947 parent->child = sd;
6948 cpu_to_phys_group(i, cpu_map, &sd->groups, d->tmpmask);
6949 return sd;
6950}
6951
Andreas Herrmann410c4082009-08-18 12:56:14 +02006952static struct sched_domain *__build_mc_sched_domain(struct s_data *d,
6953 const struct cpumask *cpu_map, struct sched_domain_attr *attr,
6954 struct sched_domain *parent, int i)
6955{
6956 struct sched_domain *sd = parent;
6957#ifdef CONFIG_SCHED_MC
6958 sd = &per_cpu(core_domains, i).sd;
6959 SD_INIT(sd, MC);
6960 set_domain_attribute(sd, attr);
6961 cpumask_and(sched_domain_span(sd), cpu_map, cpu_coregroup_mask(i));
6962 sd->parent = parent;
6963 parent->child = sd;
6964 cpu_to_core_group(i, cpu_map, &sd->groups, d->tmpmask);
6965#endif
6966 return sd;
6967}
6968
Andreas Herrmannd8173532009-08-18 12:57:03 +02006969static struct sched_domain *__build_smt_sched_domain(struct s_data *d,
6970 const struct cpumask *cpu_map, struct sched_domain_attr *attr,
6971 struct sched_domain *parent, int i)
6972{
6973 struct sched_domain *sd = parent;
6974#ifdef CONFIG_SCHED_SMT
6975 sd = &per_cpu(cpu_domains, i).sd;
6976 SD_INIT(sd, SIBLING);
6977 set_domain_attribute(sd, attr);
6978 cpumask_and(sched_domain_span(sd), cpu_map, topology_thread_cpumask(i));
6979 sd->parent = parent;
6980 parent->child = sd;
6981 cpu_to_cpu_group(i, cpu_map, &sd->groups, d->tmpmask);
6982#endif
6983 return sd;
6984}
6985
Andreas Herrmann0e8e85c2009-08-18 12:57:51 +02006986static void build_sched_groups(struct s_data *d, enum sched_domain_level l,
6987 const struct cpumask *cpu_map, int cpu)
6988{
6989 switch (l) {
6990#ifdef CONFIG_SCHED_SMT
6991 case SD_LV_SIBLING: /* set up CPU (sibling) groups */
6992 cpumask_and(d->this_sibling_map, cpu_map,
6993 topology_thread_cpumask(cpu));
6994 if (cpu == cpumask_first(d->this_sibling_map))
6995 init_sched_build_groups(d->this_sibling_map, cpu_map,
6996 &cpu_to_cpu_group,
6997 d->send_covered, d->tmpmask);
6998 break;
6999#endif
Andreas Herrmanna2af04c2009-08-18 12:58:38 +02007000#ifdef CONFIG_SCHED_MC
7001 case SD_LV_MC: /* set up multi-core groups */
7002 cpumask_and(d->this_core_map, cpu_map, cpu_coregroup_mask(cpu));
7003 if (cpu == cpumask_first(d->this_core_map))
7004 init_sched_build_groups(d->this_core_map, cpu_map,
7005 &cpu_to_core_group,
7006 d->send_covered, d->tmpmask);
7007 break;
7008#endif
Andreas Herrmann86548092009-08-18 12:59:28 +02007009 case SD_LV_CPU: /* set up physical groups */
7010 cpumask_and(d->nodemask, cpumask_of_node(cpu), cpu_map);
7011 if (!cpumask_empty(d->nodemask))
7012 init_sched_build_groups(d->nodemask, cpu_map,
7013 &cpu_to_phys_group,
7014 d->send_covered, d->tmpmask);
7015 break;
Andreas Herrmannde616e32009-08-18 13:00:13 +02007016#ifdef CONFIG_NUMA
7017 case SD_LV_ALLNODES:
7018 init_sched_build_groups(cpu_map, cpu_map, &cpu_to_allnodes_group,
7019 d->send_covered, d->tmpmask);
7020 break;
7021#endif
Andreas Herrmann0e8e85c2009-08-18 12:57:51 +02007022 default:
7023 break;
7024 }
7025}
7026
Mike Travis7c16ec52008-04-04 18:11:11 -07007027/*
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007028 * Build sched domains for a given set of cpus and attach the sched domains
7029 * to the individual cpus
Linus Torvalds1da177e2005-04-16 15:20:36 -07007030 */
Rusty Russell96f874e22008-11-25 02:35:14 +10307031static int __build_sched_domains(const struct cpumask *cpu_map,
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007032 struct sched_domain_attr *attr)
Linus Torvalds1da177e2005-04-16 15:20:36 -07007033{
Andreas Herrmann2109b992009-08-18 12:53:00 +02007034 enum s_alloc alloc_state = sa_none;
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007035 struct s_data d;
Andreas Herrmann294b0c92009-08-18 13:02:29 +02007036 struct sched_domain *sd;
Andreas Herrmann2109b992009-08-18 12:53:00 +02007037 int i;
John Hawkesd1b55132005-09-06 15:18:14 -07007038#ifdef CONFIG_NUMA
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007039 d.sd_allnodes = 0;
Rusty Russell3404c8d2008-11-25 02:35:03 +10307040#endif
7041
Andreas Herrmann2109b992009-08-18 12:53:00 +02007042 alloc_state = __visit_domain_allocation_hell(&d, cpu_map);
7043 if (alloc_state != sa_rootdomain)
7044 goto error;
7045 alloc_state = sa_sched_groups;
Mike Travis7c16ec52008-04-04 18:11:11 -07007046
Linus Torvalds1da177e2005-04-16 15:20:36 -07007047 /*
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007048 * Set up domains for cpus specified by the cpu_map.
Linus Torvalds1da177e2005-04-16 15:20:36 -07007049 */
Rusty Russellabcd0832008-11-25 02:35:02 +10307050 for_each_cpu(i, cpu_map) {
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007051 cpumask_and(d.nodemask, cpumask_of_node(cpu_to_node(i)),
7052 cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007053
Andreas Herrmann7f4588f2009-08-18 12:54:06 +02007054 sd = __build_numa_sched_domains(&d, cpu_map, attr, i);
Andreas Herrmann87cce662009-08-18 12:54:55 +02007055 sd = __build_cpu_sched_domain(&d, cpu_map, attr, sd, i);
Andreas Herrmann410c4082009-08-18 12:56:14 +02007056 sd = __build_mc_sched_domain(&d, cpu_map, attr, sd, i);
Andreas Herrmannd8173532009-08-18 12:57:03 +02007057 sd = __build_smt_sched_domain(&d, cpu_map, attr, sd, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007058 }
7059
Rusty Russellabcd0832008-11-25 02:35:02 +10307060 for_each_cpu(i, cpu_map) {
Andreas Herrmann0e8e85c2009-08-18 12:57:51 +02007061 build_sched_groups(&d, SD_LV_SIBLING, cpu_map, i);
Andreas Herrmanna2af04c2009-08-18 12:58:38 +02007062 build_sched_groups(&d, SD_LV_MC, cpu_map, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007063 }
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08007064
Linus Torvalds1da177e2005-04-16 15:20:36 -07007065 /* Set up physical groups */
Andreas Herrmann86548092009-08-18 12:59:28 +02007066 for (i = 0; i < nr_node_ids; i++)
7067 build_sched_groups(&d, SD_LV_CPU, cpu_map, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007068
7069#ifdef CONFIG_NUMA
7070 /* Set up node groups */
Andreas Herrmannde616e32009-08-18 13:00:13 +02007071 if (d.sd_allnodes)
7072 build_sched_groups(&d, SD_LV_ALLNODES, cpu_map, 0);
John Hawkes9c1cfda2005-09-06 15:18:14 -07007073
Andreas Herrmann0601a882009-08-18 13:01:11 +02007074 for (i = 0; i < nr_node_ids; i++)
7075 if (build_numa_sched_groups(&d, cpu_map, i))
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07007076 goto error;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007077#endif
7078
7079 /* Calculate CPU power for physical packages and nodes */
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007080#ifdef CONFIG_SCHED_SMT
Rusty Russellabcd0832008-11-25 02:35:02 +10307081 for_each_cpu(i, cpu_map) {
Andreas Herrmann294b0c92009-08-18 13:02:29 +02007082 sd = &per_cpu(cpu_domains, i).sd;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07007083 init_sched_groups_power(i, sd);
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007084 }
7085#endif
7086#ifdef CONFIG_SCHED_MC
Rusty Russellabcd0832008-11-25 02:35:02 +10307087 for_each_cpu(i, cpu_map) {
Andreas Herrmann294b0c92009-08-18 13:02:29 +02007088 sd = &per_cpu(core_domains, i).sd;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07007089 init_sched_groups_power(i, sd);
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007090 }
7091#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07007092
Rusty Russellabcd0832008-11-25 02:35:02 +10307093 for_each_cpu(i, cpu_map) {
Andreas Herrmann294b0c92009-08-18 13:02:29 +02007094 sd = &per_cpu(phys_domains, i).sd;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07007095 init_sched_groups_power(i, sd);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007096 }
7097
John Hawkes9c1cfda2005-09-06 15:18:14 -07007098#ifdef CONFIG_NUMA
Mike Travis076ac2a2008-05-12 21:21:12 +02007099 for (i = 0; i < nr_node_ids; i++)
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007100 init_numa_sched_groups_power(d.sched_group_nodes[i]);
John Hawkes9c1cfda2005-09-06 15:18:14 -07007101
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007102 if (d.sd_allnodes) {
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08007103 struct sched_group *sg;
Siddha, Suresh Bf712c0c72006-07-30 03:02:59 -07007104
Rusty Russell96f874e22008-11-25 02:35:14 +10307105 cpu_to_allnodes_group(cpumask_first(cpu_map), cpu_map, &sg,
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007106 d.tmpmask);
Siddha, Suresh Bf712c0c72006-07-30 03:02:59 -07007107 init_numa_sched_groups_power(sg);
7108 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07007109#endif
7110
Linus Torvalds1da177e2005-04-16 15:20:36 -07007111 /* Attach the domains */
Rusty Russellabcd0832008-11-25 02:35:02 +10307112 for_each_cpu(i, cpu_map) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07007113#ifdef CONFIG_SCHED_SMT
Rusty Russell6c99e9a2008-11-25 02:35:04 +10307114 sd = &per_cpu(cpu_domains, i).sd;
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08007115#elif defined(CONFIG_SCHED_MC)
Rusty Russell6c99e9a2008-11-25 02:35:04 +10307116 sd = &per_cpu(core_domains, i).sd;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007117#else
Rusty Russell6c99e9a2008-11-25 02:35:04 +10307118 sd = &per_cpu(phys_domains, i).sd;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007119#endif
Andreas Herrmann49a02c52009-08-18 12:51:52 +02007120 cpu_attach_domain(sd, d.rd, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007121 }
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07007122
Andreas Herrmann2109b992009-08-18 12:53:00 +02007123 d.sched_group_nodes = NULL; /* don't free this we still need it */
7124 __free_domain_allocs(&d, sa_tmpmask, cpu_map);
7125 return 0;
Rusty Russell3404c8d2008-11-25 02:35:03 +10307126
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07007127error:
Andreas Herrmann2109b992009-08-18 12:53:00 +02007128 __free_domain_allocs(&d, alloc_state, cpu_map);
7129 return -ENOMEM;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007130}
Paul Jackson029190c2007-10-18 23:40:20 -07007131
Rusty Russell96f874e22008-11-25 02:35:14 +10307132static int build_sched_domains(const struct cpumask *cpu_map)
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007133{
7134 return __build_sched_domains(cpu_map, NULL);
7135}
7136
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307137static cpumask_var_t *doms_cur; /* current sched domains */
Paul Jackson029190c2007-10-18 23:40:20 -07007138static int ndoms_cur; /* number of sched domains in 'doms_cur' */
Ingo Molnar4285f5942008-05-16 17:47:14 +02007139static struct sched_domain_attr *dattr_cur;
7140 /* attribues of custom domains in 'doms_cur' */
Paul Jackson029190c2007-10-18 23:40:20 -07007141
7142/*
7143 * Special case: If a kmalloc of a doms_cur partition (array of
Rusty Russell42128232008-11-25 02:35:12 +10307144 * cpumask) fails, then fallback to a single sched domain,
7145 * as determined by the single cpumask fallback_doms.
Paul Jackson029190c2007-10-18 23:40:20 -07007146 */
Rusty Russell42128232008-11-25 02:35:12 +10307147static cpumask_var_t fallback_doms;
Paul Jackson029190c2007-10-18 23:40:20 -07007148
Heiko Carstensee79d1b2008-12-09 18:49:50 +01007149/*
7150 * arch_update_cpu_topology lets virtualized architectures update the
7151 * cpu core maps. It is supposed to return 1 if the topology changed
7152 * or 0 if it stayed the same.
7153 */
7154int __attribute__((weak)) arch_update_cpu_topology(void)
Heiko Carstens22e52b02008-03-12 18:31:59 +01007155{
Heiko Carstensee79d1b2008-12-09 18:49:50 +01007156 return 0;
Heiko Carstens22e52b02008-03-12 18:31:59 +01007157}
7158
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307159cpumask_var_t *alloc_sched_domains(unsigned int ndoms)
7160{
7161 int i;
7162 cpumask_var_t *doms;
7163
7164 doms = kmalloc(sizeof(*doms) * ndoms, GFP_KERNEL);
7165 if (!doms)
7166 return NULL;
7167 for (i = 0; i < ndoms; i++) {
7168 if (!alloc_cpumask_var(&doms[i], GFP_KERNEL)) {
7169 free_sched_domains(doms, i);
7170 return NULL;
7171 }
7172 }
7173 return doms;
7174}
7175
7176void free_sched_domains(cpumask_var_t doms[], unsigned int ndoms)
7177{
7178 unsigned int i;
7179 for (i = 0; i < ndoms; i++)
7180 free_cpumask_var(doms[i]);
7181 kfree(doms);
7182}
7183
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007184/*
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01007185 * Set up scheduler domains and groups. Callers must hold the hotplug lock.
Paul Jackson029190c2007-10-18 23:40:20 -07007186 * For now this just excludes isolated cpus, but could be used to
7187 * exclude other special cases in the future.
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007188 */
Rusty Russell96f874e22008-11-25 02:35:14 +10307189static int arch_init_sched_domains(const struct cpumask *cpu_map)
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007190{
Milton Miller73785472007-10-24 18:23:48 +02007191 int err;
7192
Heiko Carstens22e52b02008-03-12 18:31:59 +01007193 arch_update_cpu_topology();
Paul Jackson029190c2007-10-18 23:40:20 -07007194 ndoms_cur = 1;
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307195 doms_cur = alloc_sched_domains(ndoms_cur);
Paul Jackson029190c2007-10-18 23:40:20 -07007196 if (!doms_cur)
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307197 doms_cur = &fallback_doms;
7198 cpumask_andnot(doms_cur[0], cpu_map, cpu_isolated_map);
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007199 dattr_cur = NULL;
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307200 err = build_sched_domains(doms_cur[0]);
Milton Miller6382bc92007-10-15 17:00:19 +02007201 register_sched_domain_sysctl();
Milton Miller73785472007-10-24 18:23:48 +02007202
7203 return err;
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007204}
7205
Rusty Russell96f874e22008-11-25 02:35:14 +10307206static void arch_destroy_sched_domains(const struct cpumask *cpu_map,
7207 struct cpumask *tmpmask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07007208{
Mike Travis7c16ec52008-04-04 18:11:11 -07007209 free_sched_groups(cpu_map, tmpmask);
John Hawkes9c1cfda2005-09-06 15:18:14 -07007210}
Linus Torvalds1da177e2005-04-16 15:20:36 -07007211
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007212/*
7213 * Detach sched domains from a group of cpus specified in cpu_map
7214 * These cpus will now be attached to the NULL domain
7215 */
Rusty Russell96f874e22008-11-25 02:35:14 +10307216static void detach_destroy_domains(const struct cpumask *cpu_map)
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007217{
Rusty Russell96f874e22008-11-25 02:35:14 +10307218 /* Save because hotplug lock held. */
7219 static DECLARE_BITMAP(tmpmask, CONFIG_NR_CPUS);
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007220 int i;
7221
Rusty Russellabcd0832008-11-25 02:35:02 +10307222 for_each_cpu(i, cpu_map)
Gregory Haskins57d885f2008-01-25 21:08:18 +01007223 cpu_attach_domain(NULL, &def_root_domain, i);
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007224 synchronize_sched();
Rusty Russell96f874e22008-11-25 02:35:14 +10307225 arch_destroy_sched_domains(cpu_map, to_cpumask(tmpmask));
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07007226}
7227
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007228/* handle null as "default" */
7229static int dattrs_equal(struct sched_domain_attr *cur, int idx_cur,
7230 struct sched_domain_attr *new, int idx_new)
7231{
7232 struct sched_domain_attr tmp;
7233
7234 /* fast path */
7235 if (!new && !cur)
7236 return 1;
7237
7238 tmp = SD_ATTR_INIT;
7239 return !memcmp(cur ? (cur + idx_cur) : &tmp,
7240 new ? (new + idx_new) : &tmp,
7241 sizeof(struct sched_domain_attr));
7242}
7243
Paul Jackson029190c2007-10-18 23:40:20 -07007244/*
7245 * Partition sched domains as specified by the 'ndoms_new'
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01007246 * cpumasks in the array doms_new[] of cpumasks. This compares
Paul Jackson029190c2007-10-18 23:40:20 -07007247 * doms_new[] to the current sched domain partitioning, doms_cur[].
7248 * It destroys each deleted domain and builds each new domain.
7249 *
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307250 * 'doms_new' is an array of cpumask_var_t's of length 'ndoms_new'.
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01007251 * The masks don't intersect (don't overlap.) We should setup one
7252 * sched domain for each mask. CPUs not in any of the cpumasks will
7253 * not be load balanced. If the same cpumask appears both in the
Paul Jackson029190c2007-10-18 23:40:20 -07007254 * current 'doms_cur' domains and in the new 'doms_new', we can leave
7255 * it as it is.
7256 *
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307257 * The passed in 'doms_new' should be allocated using
7258 * alloc_sched_domains. This routine takes ownership of it and will
7259 * free_sched_domains it when done with it. If the caller failed the
7260 * alloc call, then it can pass in doms_new == NULL && ndoms_new == 1,
7261 * and partition_sched_domains() will fallback to the single partition
7262 * 'fallback_doms', it also forces the domains to be rebuilt.
Paul Jackson029190c2007-10-18 23:40:20 -07007263 *
Rusty Russell96f874e22008-11-25 02:35:14 +10307264 * If doms_new == NULL it will be replaced with cpu_online_mask.
Li Zefan700018e2008-11-18 14:02:03 +08007265 * ndoms_new == 0 is a special case for destroying existing domains,
7266 * and it will not create the default domain.
Max Krasnyanskydfb512e2008-08-29 13:11:41 -07007267 *
Paul Jackson029190c2007-10-18 23:40:20 -07007268 * Call with hotplug lock held
7269 */
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307270void partition_sched_domains(int ndoms_new, cpumask_var_t doms_new[],
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007271 struct sched_domain_attr *dattr_new)
Paul Jackson029190c2007-10-18 23:40:20 -07007272{
Max Krasnyanskydfb512e2008-08-29 13:11:41 -07007273 int i, j, n;
Heiko Carstensd65bd5e2008-12-09 18:49:51 +01007274 int new_topology;
Paul Jackson029190c2007-10-18 23:40:20 -07007275
Heiko Carstens712555e2008-04-28 11:33:07 +02007276 mutex_lock(&sched_domains_mutex);
Srivatsa Vaddagiria1835612008-01-25 21:08:00 +01007277
Milton Miller73785472007-10-24 18:23:48 +02007278 /* always unregister in case we don't destroy any domains */
7279 unregister_sched_domain_sysctl();
7280
Heiko Carstensd65bd5e2008-12-09 18:49:51 +01007281 /* Let architecture update cpu core mappings. */
7282 new_topology = arch_update_cpu_topology();
7283
Max Krasnyanskydfb512e2008-08-29 13:11:41 -07007284 n = doms_new ? ndoms_new : 0;
Paul Jackson029190c2007-10-18 23:40:20 -07007285
7286 /* Destroy deleted domains */
7287 for (i = 0; i < ndoms_cur; i++) {
Heiko Carstensd65bd5e2008-12-09 18:49:51 +01007288 for (j = 0; j < n && !new_topology; j++) {
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307289 if (cpumask_equal(doms_cur[i], doms_new[j])
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007290 && dattrs_equal(dattr_cur, i, dattr_new, j))
Paul Jackson029190c2007-10-18 23:40:20 -07007291 goto match1;
7292 }
7293 /* no match - a current sched domain not in new doms_new[] */
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307294 detach_destroy_domains(doms_cur[i]);
Paul Jackson029190c2007-10-18 23:40:20 -07007295match1:
7296 ;
7297 }
7298
Max Krasnyanskye761b772008-07-15 04:43:49 -07007299 if (doms_new == NULL) {
7300 ndoms_cur = 0;
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307301 doms_new = &fallback_doms;
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01007302 cpumask_andnot(doms_new[0], cpu_active_mask, cpu_isolated_map);
Li Zefanfaa2f982008-11-04 16:20:23 +08007303 WARN_ON_ONCE(dattr_new);
Max Krasnyanskye761b772008-07-15 04:43:49 -07007304 }
7305
Paul Jackson029190c2007-10-18 23:40:20 -07007306 /* Build new domains */
7307 for (i = 0; i < ndoms_new; i++) {
Heiko Carstensd65bd5e2008-12-09 18:49:51 +01007308 for (j = 0; j < ndoms_cur && !new_topology; j++) {
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307309 if (cpumask_equal(doms_new[i], doms_cur[j])
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007310 && dattrs_equal(dattr_new, i, dattr_cur, j))
Paul Jackson029190c2007-10-18 23:40:20 -07007311 goto match2;
7312 }
7313 /* no match - add a new doms_new */
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307314 __build_sched_domains(doms_new[i],
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007315 dattr_new ? dattr_new + i : NULL);
Paul Jackson029190c2007-10-18 23:40:20 -07007316match2:
7317 ;
7318 }
7319
7320 /* Remember the new sched domains */
Rusty Russellacc3f5d2009-11-03 14:53:40 +10307321 if (doms_cur != &fallback_doms)
7322 free_sched_domains(doms_cur, ndoms_cur);
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007323 kfree(dattr_cur); /* kfree(NULL) is safe */
Paul Jackson029190c2007-10-18 23:40:20 -07007324 doms_cur = doms_new;
Hidetoshi Seto1d3504f2008-04-15 14:04:23 +09007325 dattr_cur = dattr_new;
Paul Jackson029190c2007-10-18 23:40:20 -07007326 ndoms_cur = ndoms_new;
Milton Miller73785472007-10-24 18:23:48 +02007327
7328 register_sched_domain_sysctl();
Srivatsa Vaddagiria1835612008-01-25 21:08:00 +01007329
Heiko Carstens712555e2008-04-28 11:33:07 +02007330 mutex_unlock(&sched_domains_mutex);
Paul Jackson029190c2007-10-18 23:40:20 -07007331}
7332
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007333#if defined(CONFIG_SCHED_MC) || defined(CONFIG_SCHED_SMT)
Li Zefanc70f22d2009-01-05 19:07:50 +08007334static void arch_reinit_sched_domains(void)
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007335{
Gautham R Shenoy95402b32008-01-25 21:08:02 +01007336 get_online_cpus();
Max Krasnyanskydfb512e2008-08-29 13:11:41 -07007337
7338 /* Destroy domains first to force the rebuild */
7339 partition_sched_domains(0, NULL, NULL);
7340
Max Krasnyanskye761b772008-07-15 04:43:49 -07007341 rebuild_sched_domains();
Gautham R Shenoy95402b32008-01-25 21:08:02 +01007342 put_online_cpus();
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007343}
7344
7345static ssize_t sched_power_savings_store(const char *buf, size_t count, int smt)
7346{
Gautham R Shenoyafb8a9b2008-12-18 23:26:09 +05307347 unsigned int level = 0;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007348
Gautham R Shenoyafb8a9b2008-12-18 23:26:09 +05307349 if (sscanf(buf, "%u", &level) != 1)
7350 return -EINVAL;
7351
7352 /*
7353 * level is always be positive so don't check for
7354 * level < POWERSAVINGS_BALANCE_NONE which is 0
7355 * What happens on 0 or 1 byte write,
7356 * need to check for count as well?
7357 */
7358
7359 if (level >= MAX_POWERSAVINGS_BALANCE_LEVELS)
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007360 return -EINVAL;
7361
7362 if (smt)
Gautham R Shenoyafb8a9b2008-12-18 23:26:09 +05307363 sched_smt_power_savings = level;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007364 else
Gautham R Shenoyafb8a9b2008-12-18 23:26:09 +05307365 sched_mc_power_savings = level;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007366
Li Zefanc70f22d2009-01-05 19:07:50 +08007367 arch_reinit_sched_domains();
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007368
Li Zefanc70f22d2009-01-05 19:07:50 +08007369 return count;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007370}
7371
Adrian Bunk6707de002007-08-12 18:08:19 +02007372#ifdef CONFIG_SCHED_MC
Andi Kleenf718cd42008-07-29 22:33:52 -07007373static ssize_t sched_mc_power_savings_show(struct sysdev_class *class,
7374 char *page)
Adrian Bunk6707de002007-08-12 18:08:19 +02007375{
7376 return sprintf(page, "%u\n", sched_mc_power_savings);
7377}
Andi Kleenf718cd42008-07-29 22:33:52 -07007378static ssize_t sched_mc_power_savings_store(struct sysdev_class *class,
Adrian Bunk6707de002007-08-12 18:08:19 +02007379 const char *buf, size_t count)
7380{
7381 return sched_power_savings_store(buf, count, 0);
7382}
Andi Kleenf718cd42008-07-29 22:33:52 -07007383static SYSDEV_CLASS_ATTR(sched_mc_power_savings, 0644,
7384 sched_mc_power_savings_show,
7385 sched_mc_power_savings_store);
Adrian Bunk6707de002007-08-12 18:08:19 +02007386#endif
7387
7388#ifdef CONFIG_SCHED_SMT
Andi Kleenf718cd42008-07-29 22:33:52 -07007389static ssize_t sched_smt_power_savings_show(struct sysdev_class *dev,
7390 char *page)
Adrian Bunk6707de002007-08-12 18:08:19 +02007391{
7392 return sprintf(page, "%u\n", sched_smt_power_savings);
7393}
Andi Kleenf718cd42008-07-29 22:33:52 -07007394static ssize_t sched_smt_power_savings_store(struct sysdev_class *dev,
Adrian Bunk6707de002007-08-12 18:08:19 +02007395 const char *buf, size_t count)
7396{
7397 return sched_power_savings_store(buf, count, 1);
7398}
Andi Kleenf718cd42008-07-29 22:33:52 -07007399static SYSDEV_CLASS_ATTR(sched_smt_power_savings, 0644,
7400 sched_smt_power_savings_show,
Adrian Bunk6707de002007-08-12 18:08:19 +02007401 sched_smt_power_savings_store);
7402#endif
7403
Li Zefan39aac642009-01-05 19:18:02 +08007404int __init sched_create_sysfs_power_savings_entries(struct sysdev_class *cls)
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007405{
7406 int err = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07007407
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007408#ifdef CONFIG_SCHED_SMT
7409 if (smt_capable())
7410 err = sysfs_create_file(&cls->kset.kobj,
7411 &attr_sched_smt_power_savings.attr);
7412#endif
7413#ifdef CONFIG_SCHED_MC
7414 if (!err && mc_capable())
7415 err = sysfs_create_file(&cls->kset.kobj,
7416 &attr_sched_mc_power_savings.attr);
7417#endif
7418 return err;
7419}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02007420#endif /* CONFIG_SCHED_MC || CONFIG_SCHED_SMT */
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07007421
Max Krasnyanskye761b772008-07-15 04:43:49 -07007422#ifndef CONFIG_CPUSETS
Linus Torvalds1da177e2005-04-16 15:20:36 -07007423/*
Max Krasnyanskye761b772008-07-15 04:43:49 -07007424 * Add online and remove offline CPUs from the scheduler domains.
7425 * When cpusets are enabled they take over this function.
Linus Torvalds1da177e2005-04-16 15:20:36 -07007426 */
7427static int update_sched_domains(struct notifier_block *nfb,
7428 unsigned long action, void *hcpu)
7429{
Max Krasnyanskye761b772008-07-15 04:43:49 -07007430 switch (action) {
7431 case CPU_ONLINE:
7432 case CPU_ONLINE_FROZEN:
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01007433 case CPU_DOWN_PREPARE:
7434 case CPU_DOWN_PREPARE_FROZEN:
7435 case CPU_DOWN_FAILED:
7436 case CPU_DOWN_FAILED_FROZEN:
Max Krasnyanskydfb512e2008-08-29 13:11:41 -07007437 partition_sched_domains(1, NULL, NULL);
Max Krasnyanskye761b772008-07-15 04:43:49 -07007438 return NOTIFY_OK;
7439
7440 default:
7441 return NOTIFY_DONE;
7442 }
7443}
7444#endif
7445
7446static int update_runtime(struct notifier_block *nfb,
7447 unsigned long action, void *hcpu)
7448{
Peter Zijlstra7def2be2008-06-05 14:49:58 +02007449 int cpu = (int)(long)hcpu;
7450
Linus Torvalds1da177e2005-04-16 15:20:36 -07007451 switch (action) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07007452 case CPU_DOWN_PREPARE:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07007453 case CPU_DOWN_PREPARE_FROZEN:
Peter Zijlstra7def2be2008-06-05 14:49:58 +02007454 disable_runtime(cpu_rq(cpu));
Linus Torvalds1da177e2005-04-16 15:20:36 -07007455 return NOTIFY_OK;
7456
Linus Torvalds1da177e2005-04-16 15:20:36 -07007457 case CPU_DOWN_FAILED:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07007458 case CPU_DOWN_FAILED_FROZEN:
Linus Torvalds1da177e2005-04-16 15:20:36 -07007459 case CPU_ONLINE:
Rafael J. Wysocki8bb78442007-05-09 02:35:10 -07007460 case CPU_ONLINE_FROZEN:
Peter Zijlstra7def2be2008-06-05 14:49:58 +02007461 enable_runtime(cpu_rq(cpu));
Max Krasnyanskye761b772008-07-15 04:43:49 -07007462 return NOTIFY_OK;
7463
Linus Torvalds1da177e2005-04-16 15:20:36 -07007464 default:
7465 return NOTIFY_DONE;
7466 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07007467}
Linus Torvalds1da177e2005-04-16 15:20:36 -07007468
7469void __init sched_init_smp(void)
7470{
Rusty Russelldcc30a32008-11-25 02:35:12 +10307471 cpumask_var_t non_isolated_cpus;
7472
7473 alloc_cpumask_var(&non_isolated_cpus, GFP_KERNEL);
Yong Zhangcb5fd132009-09-14 20:20:16 +08007474 alloc_cpumask_var(&fallback_doms, GFP_KERNEL);
Nick Piggin5c1e1762006-10-03 01:14:04 -07007475
Mike Travis434d53b2008-04-04 18:11:04 -07007476#if defined(CONFIG_NUMA)
7477 sched_group_nodes_bycpu = kzalloc(nr_cpu_ids * sizeof(void **),
7478 GFP_KERNEL);
7479 BUG_ON(sched_group_nodes_bycpu == NULL);
7480#endif
Gautham R Shenoy95402b32008-01-25 21:08:02 +01007481 get_online_cpus();
Heiko Carstens712555e2008-04-28 11:33:07 +02007482 mutex_lock(&sched_domains_mutex);
Peter Zijlstra6ad4c182009-11-25 13:31:39 +01007483 arch_init_sched_domains(cpu_active_mask);
Rusty Russelldcc30a32008-11-25 02:35:12 +10307484 cpumask_andnot(non_isolated_cpus, cpu_possible_mask, cpu_isolated_map);
7485 if (cpumask_empty(non_isolated_cpus))
7486 cpumask_set_cpu(smp_processor_id(), non_isolated_cpus);
Heiko Carstens712555e2008-04-28 11:33:07 +02007487 mutex_unlock(&sched_domains_mutex);
Gautham R Shenoy95402b32008-01-25 21:08:02 +01007488 put_online_cpus();
Max Krasnyanskye761b772008-07-15 04:43:49 -07007489
7490#ifndef CONFIG_CPUSETS
Linus Torvalds1da177e2005-04-16 15:20:36 -07007491 /* XXX: Theoretical race here - CPU may be hotplugged now */
7492 hotcpu_notifier(update_sched_domains, 0);
Max Krasnyanskye761b772008-07-15 04:43:49 -07007493#endif
7494
7495 /* RT runtime code needs to handle some hotplug events */
7496 hotcpu_notifier(update_runtime, 0);
7497
Peter Zijlstrab328ca12008-04-29 10:02:46 +02007498 init_hrtick();
Nick Piggin5c1e1762006-10-03 01:14:04 -07007499
7500 /* Move init over to a non-isolated CPU */
Rusty Russelldcc30a32008-11-25 02:35:12 +10307501 if (set_cpus_allowed_ptr(current, non_isolated_cpus) < 0)
Nick Piggin5c1e1762006-10-03 01:14:04 -07007502 BUG();
Ingo Molnar19978ca2007-11-09 22:39:38 +01007503 sched_init_granularity();
Rusty Russelldcc30a32008-11-25 02:35:12 +10307504 free_cpumask_var(non_isolated_cpus);
Rusty Russell42128232008-11-25 02:35:12 +10307505
Rusty Russell0e3900e2008-11-25 02:35:13 +10307506 init_sched_rt_class();
Linus Torvalds1da177e2005-04-16 15:20:36 -07007507}
7508#else
7509void __init sched_init_smp(void)
7510{
Ingo Molnar19978ca2007-11-09 22:39:38 +01007511 sched_init_granularity();
Linus Torvalds1da177e2005-04-16 15:20:36 -07007512}
7513#endif /* CONFIG_SMP */
7514
Arun R Bharadwajcd1bb942009-04-16 12:15:34 +05307515const_debug unsigned int sysctl_timer_migration = 1;
7516
Linus Torvalds1da177e2005-04-16 15:20:36 -07007517int in_sched_functions(unsigned long addr)
7518{
Linus Torvalds1da177e2005-04-16 15:20:36 -07007519 return in_lock_functions(addr) ||
7520 (addr >= (unsigned long)__sched_text_start
7521 && addr < (unsigned long)__sched_text_end);
7522}
7523
Alexey Dobriyana9957442007-10-15 17:00:13 +02007524static void init_cfs_rq(struct cfs_rq *cfs_rq, struct rq *rq)
Ingo Molnardd41f592007-07-09 18:51:59 +02007525{
7526 cfs_rq->tasks_timeline = RB_ROOT;
Peter Zijlstra4a55bd52008-04-19 19:45:00 +02007527 INIT_LIST_HEAD(&cfs_rq->tasks);
Ingo Molnardd41f592007-07-09 18:51:59 +02007528#ifdef CONFIG_FAIR_GROUP_SCHED
7529 cfs_rq->rq = rq;
7530#endif
Peter Zijlstra67e9fb22007-10-15 17:00:10 +02007531 cfs_rq->min_vruntime = (u64)(-(1LL << 20));
Ingo Molnardd41f592007-07-09 18:51:59 +02007532}
7533
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007534static void init_rt_rq(struct rt_rq *rt_rq, struct rq *rq)
7535{
7536 struct rt_prio_array *array;
7537 int i;
7538
7539 array = &rt_rq->active;
7540 for (i = 0; i < MAX_RT_PRIO; i++) {
7541 INIT_LIST_HEAD(array->queue + i);
7542 __clear_bit(i, array->bitmap);
7543 }
7544 /* delimiter for bitsearch: */
7545 __set_bit(MAX_RT_PRIO, array->bitmap);
7546
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007547#if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
Gregory Haskinse864c492008-12-29 09:39:49 -05007548 rt_rq->highest_prio.curr = MAX_RT_PRIO;
Gregory Haskins398a1532009-01-14 09:10:04 -05007549#ifdef CONFIG_SMP
Gregory Haskinse864c492008-12-29 09:39:49 -05007550 rt_rq->highest_prio.next = MAX_RT_PRIO;
Peter Zijlstra48d5e252008-01-25 21:08:31 +01007551#endif
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007552#endif
7553#ifdef CONFIG_SMP
7554 rt_rq->rt_nr_migratory = 0;
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007555 rt_rq->overloaded = 0;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01007556 plist_head_init_raw(&rt_rq->pushable_tasks, &rq->lock);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007557#endif
7558
7559 rt_rq->rt_time = 0;
7560 rt_rq->rt_throttled = 0;
Peter Zijlstraac086bc2008-04-19 19:44:58 +02007561 rt_rq->rt_runtime = 0;
Thomas Gleixner0986b112009-11-17 15:32:06 +01007562 raw_spin_lock_init(&rt_rq->rt_runtime_lock);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007563
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007564#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +01007565 rt_rq->rt_nr_boosted = 0;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007566 rt_rq->rq = rq;
7567#endif
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007568}
7569
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007570#ifdef CONFIG_FAIR_GROUP_SCHED
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007571static void init_tg_cfs_entry(struct task_group *tg, struct cfs_rq *cfs_rq,
7572 struct sched_entity *se, int cpu, int add,
7573 struct sched_entity *parent)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007574{
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007575 struct rq *rq = cpu_rq(cpu);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007576 tg->cfs_rq[cpu] = cfs_rq;
7577 init_cfs_rq(cfs_rq, rq);
7578 cfs_rq->tg = tg;
7579 if (add)
7580 list_add(&cfs_rq->leaf_cfs_rq_list, &rq->leaf_cfs_rq_list);
7581
7582 tg->se[cpu] = se;
Dhaval Giani354d60c2008-04-19 19:44:59 +02007583 /* se could be NULL for init_task_group */
7584 if (!se)
7585 return;
7586
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007587 if (!parent)
7588 se->cfs_rq = &rq->cfs;
7589 else
7590 se->cfs_rq = parent->my_q;
7591
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007592 se->my_q = cfs_rq;
7593 se->load.weight = tg->shares;
Peter Zijlstrae05510d2008-05-05 23:56:17 +02007594 se->load.inv_weight = 0;
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007595 se->parent = parent;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007596}
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007597#endif
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007598
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007599#ifdef CONFIG_RT_GROUP_SCHED
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007600static void init_tg_rt_entry(struct task_group *tg, struct rt_rq *rt_rq,
7601 struct sched_rt_entity *rt_se, int cpu, int add,
7602 struct sched_rt_entity *parent)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007603{
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007604 struct rq *rq = cpu_rq(cpu);
7605
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007606 tg->rt_rq[cpu] = rt_rq;
7607 init_rt_rq(rt_rq, rq);
7608 rt_rq->tg = tg;
7609 rt_rq->rt_se = rt_se;
Peter Zijlstraac086bc2008-04-19 19:44:58 +02007610 rt_rq->rt_runtime = tg->rt_bandwidth.rt_runtime;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007611 if (add)
7612 list_add(&rt_rq->leaf_rt_rq_list, &rq->leaf_rt_rq_list);
7613
7614 tg->rt_se[cpu] = rt_se;
Dhaval Giani354d60c2008-04-19 19:44:59 +02007615 if (!rt_se)
7616 return;
7617
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007618 if (!parent)
7619 rt_se->rt_rq = &rq->rt;
7620 else
7621 rt_se->rt_rq = parent->my_q;
7622
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007623 rt_se->my_q = rt_rq;
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007624 rt_se->parent = parent;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007625 INIT_LIST_HEAD(&rt_se->run_list);
7626}
7627#endif
7628
Linus Torvalds1da177e2005-04-16 15:20:36 -07007629void __init sched_init(void)
7630{
Ingo Molnardd41f592007-07-09 18:51:59 +02007631 int i, j;
Mike Travis434d53b2008-04-04 18:11:04 -07007632 unsigned long alloc_size = 0, ptr;
7633
7634#ifdef CONFIG_FAIR_GROUP_SCHED
7635 alloc_size += 2 * nr_cpu_ids * sizeof(void **);
7636#endif
7637#ifdef CONFIG_RT_GROUP_SCHED
7638 alloc_size += 2 * nr_cpu_ids * sizeof(void **);
7639#endif
Rusty Russelldf7c8e82009-03-19 15:22:20 +10307640#ifdef CONFIG_CPUMASK_OFFSTACK
Rusty Russell8c083f02009-03-19 15:22:20 +10307641 alloc_size += num_possible_cpus() * cpumask_size();
Rusty Russelldf7c8e82009-03-19 15:22:20 +10307642#endif
Mike Travis434d53b2008-04-04 18:11:04 -07007643 if (alloc_size) {
Pekka Enberg36b7b6d2009-06-10 23:42:36 +03007644 ptr = (unsigned long)kzalloc(alloc_size, GFP_NOWAIT);
Mike Travis434d53b2008-04-04 18:11:04 -07007645
7646#ifdef CONFIG_FAIR_GROUP_SCHED
7647 init_task_group.se = (struct sched_entity **)ptr;
7648 ptr += nr_cpu_ids * sizeof(void **);
7649
7650 init_task_group.cfs_rq = (struct cfs_rq **)ptr;
7651 ptr += nr_cpu_ids * sizeof(void **);
Peter Zijlstraeff766a2008-04-19 19:45:00 +02007652
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02007653#endif /* CONFIG_FAIR_GROUP_SCHED */
Mike Travis434d53b2008-04-04 18:11:04 -07007654#ifdef CONFIG_RT_GROUP_SCHED
7655 init_task_group.rt_se = (struct sched_rt_entity **)ptr;
7656 ptr += nr_cpu_ids * sizeof(void **);
7657
7658 init_task_group.rt_rq = (struct rt_rq **)ptr;
Peter Zijlstraeff766a2008-04-19 19:45:00 +02007659 ptr += nr_cpu_ids * sizeof(void **);
7660
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02007661#endif /* CONFIG_RT_GROUP_SCHED */
Rusty Russelldf7c8e82009-03-19 15:22:20 +10307662#ifdef CONFIG_CPUMASK_OFFSTACK
7663 for_each_possible_cpu(i) {
7664 per_cpu(load_balance_tmpmask, i) = (void *)ptr;
7665 ptr += cpumask_size();
7666 }
7667#endif /* CONFIG_CPUMASK_OFFSTACK */
Mike Travis434d53b2008-04-04 18:11:04 -07007668 }
Ingo Molnardd41f592007-07-09 18:51:59 +02007669
Gregory Haskins57d885f2008-01-25 21:08:18 +01007670#ifdef CONFIG_SMP
7671 init_defrootdomain();
7672#endif
7673
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02007674 init_rt_bandwidth(&def_rt_bandwidth,
7675 global_rt_period(), global_rt_runtime());
7676
7677#ifdef CONFIG_RT_GROUP_SCHED
7678 init_rt_bandwidth(&init_task_group.rt_bandwidth,
7679 global_rt_period(), global_rt_runtime());
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02007680#endif /* CONFIG_RT_GROUP_SCHED */
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02007681
Dhaval Giani7c941432010-01-20 13:26:18 +01007682#ifdef CONFIG_CGROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007683 list_add(&init_task_group.list, &task_groups);
Peter Zijlstraf473aa52008-04-19 19:45:00 +02007684 INIT_LIST_HEAD(&init_task_group.children);
7685
Dhaval Giani7c941432010-01-20 13:26:18 +01007686#endif /* CONFIG_CGROUP_SCHED */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007687
Jiri Kosina4a6cc4b2009-10-29 00:26:00 +09007688#if defined CONFIG_FAIR_GROUP_SCHED && defined CONFIG_SMP
7689 update_shares_data = __alloc_percpu(nr_cpu_ids * sizeof(unsigned long),
7690 __alignof__(unsigned long));
7691#endif
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08007692 for_each_possible_cpu(i) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07007693 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007694
7695 rq = cpu_rq(i);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01007696 raw_spin_lock_init(&rq->lock);
Nick Piggin78979862005-06-25 14:57:13 -07007697 rq->nr_running = 0;
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02007698 rq->calc_load_active = 0;
7699 rq->calc_load_update = jiffies + LOAD_FREQ;
Ingo Molnardd41f592007-07-09 18:51:59 +02007700 init_cfs_rq(&rq->cfs, rq);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +01007701 init_rt_rq(&rq->rt, rq);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007702#ifdef CONFIG_FAIR_GROUP_SCHED
7703 init_task_group.shares = init_task_group_load;
7704 INIT_LIST_HEAD(&rq->leaf_cfs_rq_list);
Dhaval Giani354d60c2008-04-19 19:44:59 +02007705#ifdef CONFIG_CGROUP_SCHED
7706 /*
7707 * How much cpu bandwidth does init_task_group get?
7708 *
7709 * In case of task-groups formed thr' the cgroup filesystem, it
7710 * gets 100% of the cpu resources in the system. This overall
7711 * system cpu resource is divided among the tasks of
7712 * init_task_group and its child task-groups in a fair manner,
7713 * based on each entity's (task or task-group's) weight
7714 * (se->load.weight).
7715 *
7716 * In other words, if init_task_group has 10 tasks of weight
7717 * 1024) and two child groups A0 and A1 (of weight 1024 each),
7718 * then A0's share of the cpu resource is:
7719 *
Ingo Molnar0d905bc2009-05-04 19:13:30 +02007720 * A0's bandwidth = 1024 / (10*1024 + 1024 + 1024) = 8.33%
Dhaval Giani354d60c2008-04-19 19:44:59 +02007721 *
7722 * We achieve this by letting init_task_group's tasks sit
7723 * directly in rq->cfs (i.e init_task_group->se[] = NULL).
7724 */
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007725 init_tg_cfs_entry(&init_task_group, &rq->cfs, NULL, i, 1, NULL);
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007726#endif
Dhaval Giani354d60c2008-04-19 19:44:59 +02007727#endif /* CONFIG_FAIR_GROUP_SCHED */
7728
7729 rq->rt.rt_runtime = def_rt_bandwidth.rt_runtime;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007730#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007731 INIT_LIST_HEAD(&rq->leaf_rt_rq_list);
Dhaval Giani354d60c2008-04-19 19:44:59 +02007732#ifdef CONFIG_CGROUP_SCHED
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007733 init_tg_rt_entry(&init_task_group, &rq->rt, NULL, i, 1, NULL);
Dhaval Giani354d60c2008-04-19 19:44:59 +02007734#endif
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007735#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07007736
Ingo Molnardd41f592007-07-09 18:51:59 +02007737 for (j = 0; j < CPU_LOAD_IDX_MAX; j++)
7738 rq->cpu_load[j] = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007739#ifdef CONFIG_SMP
Nick Piggin41c7ce92005-06-25 14:57:24 -07007740 rq->sd = NULL;
Gregory Haskins57d885f2008-01-25 21:08:18 +01007741 rq->rd = NULL;
Gregory Haskins3f029d32009-07-29 11:08:47 -04007742 rq->post_schedule = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007743 rq->active_balance = 0;
Ingo Molnardd41f592007-07-09 18:51:59 +02007744 rq->next_balance = jiffies;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007745 rq->push_cpu = 0;
Christoph Lameter0a2966b2006-09-25 23:30:51 -07007746 rq->cpu = i;
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04007747 rq->online = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007748 rq->migration_thread = NULL;
Mike Galbraitheae0c9d2009-11-10 03:50:02 +01007749 rq->idle_stamp = 0;
7750 rq->avg_idle = 2*sysctl_sched_migration_cost;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007751 INIT_LIST_HEAD(&rq->migration_queue);
Gregory Haskinsdc938522008-01-25 21:08:26 +01007752 rq_attach_root(rq, &def_root_domain);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007753#endif
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01007754 init_rq_hrtick(rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007755 atomic_set(&rq->nr_iowait, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007756 }
7757
Peter Williams2dd73a42006-06-27 02:54:34 -07007758 set_load_weight(&init_task);
Heiko Carstensb50f60c2006-07-30 03:03:52 -07007759
Avi Kivitye107be32007-07-26 13:40:43 +02007760#ifdef CONFIG_PREEMPT_NOTIFIERS
7761 INIT_HLIST_HEAD(&init_task.preempt_notifiers);
7762#endif
7763
Christoph Lameterc9819f42006-12-10 02:20:25 -08007764#ifdef CONFIG_SMP
Carlos R. Mafra962cf362008-05-15 11:15:37 -03007765 open_softirq(SCHED_SOFTIRQ, run_rebalance_domains);
Christoph Lameterc9819f42006-12-10 02:20:25 -08007766#endif
7767
Heiko Carstensb50f60c2006-07-30 03:03:52 -07007768#ifdef CONFIG_RT_MUTEXES
Thomas Gleixner1d615482009-11-17 14:54:03 +01007769 plist_head_init_raw(&init_task.pi_waiters, &init_task.pi_lock);
Heiko Carstensb50f60c2006-07-30 03:03:52 -07007770#endif
7771
Linus Torvalds1da177e2005-04-16 15:20:36 -07007772 /*
7773 * The boot idle thread does lazy MMU switching as well:
7774 */
7775 atomic_inc(&init_mm.mm_count);
7776 enter_lazy_tlb(&init_mm, current);
7777
7778 /*
7779 * Make us the idle thread. Technically, schedule() should not be
7780 * called from this thread, however somewhere below it might be,
7781 * but because we are the idle thread, we just pick up running again
7782 * when this runqueue becomes "idle".
7783 */
7784 init_idle(current, smp_processor_id());
Thomas Gleixnerdce48a82009-04-11 10:43:41 +02007785
7786 calc_load_update = jiffies + LOAD_FREQ;
7787
Ingo Molnardd41f592007-07-09 18:51:59 +02007788 /*
7789 * During early bootup we pretend to be a normal task:
7790 */
7791 current->sched_class = &fair_sched_class;
Ingo Molnar6892b752008-02-13 14:02:36 +01007792
Rusty Russell6a7b3dc2008-11-25 02:35:04 +10307793 /* Allocate the nohz_cpu_mask if CONFIG_CPUMASK_OFFSTACK */
Rusty Russell49557e62009-11-02 20:37:20 +10307794 zalloc_cpumask_var(&nohz_cpu_mask, GFP_NOWAIT);
Rusty Russellbf4d83f2008-11-25 09:57:51 +10307795#ifdef CONFIG_SMP
Rusty Russell7d1e6a92008-11-25 02:35:09 +10307796#ifdef CONFIG_NO_HZ
Rusty Russell49557e62009-11-02 20:37:20 +10307797 zalloc_cpumask_var(&nohz.cpu_mask, GFP_NOWAIT);
Pekka Enberg4bdddf82009-06-11 08:35:27 +03007798 alloc_cpumask_var(&nohz.ilb_grp_nohz_mask, GFP_NOWAIT);
Rusty Russell7d1e6a92008-11-25 02:35:09 +10307799#endif
Rusty Russellbdddd292009-12-02 14:09:16 +10307800 /* May be allocated at isolcpus cmdline parse time */
7801 if (cpu_isolated_map == NULL)
7802 zalloc_cpumask_var(&cpu_isolated_map, GFP_NOWAIT);
Rusty Russellbf4d83f2008-11-25 09:57:51 +10307803#endif /* SMP */
Rusty Russell6a7b3dc2008-11-25 02:35:04 +10307804
Ingo Molnarcdd6c482009-09-21 12:02:48 +02007805 perf_event_init();
Ingo Molnar0d905bc2009-05-04 19:13:30 +02007806
Ingo Molnar6892b752008-02-13 14:02:36 +01007807 scheduler_running = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007808}
7809
7810#ifdef CONFIG_DEBUG_SPINLOCK_SLEEP
Frederic Weisbeckere4aafea2009-07-16 15:44:29 +02007811static inline int preempt_count_equals(int preempt_offset)
7812{
Frederic Weisbecker234da7b2009-12-16 20:21:05 +01007813 int nested = (preempt_count() & ~PREEMPT_ACTIVE) + rcu_preempt_depth();
Frederic Weisbeckere4aafea2009-07-16 15:44:29 +02007814
7815 return (nested == PREEMPT_INATOMIC_BASE + preempt_offset);
7816}
7817
Simon Kagstromd8948372009-12-23 11:08:18 +01007818void __might_sleep(const char *file, int line, int preempt_offset)
Linus Torvalds1da177e2005-04-16 15:20:36 -07007819{
Ingo Molnar48f24c42006-07-03 00:25:40 -07007820#ifdef in_atomic
Linus Torvalds1da177e2005-04-16 15:20:36 -07007821 static unsigned long prev_jiffy; /* ratelimiting */
7822
Frederic Weisbeckere4aafea2009-07-16 15:44:29 +02007823 if ((preempt_count_equals(preempt_offset) && !irqs_disabled()) ||
7824 system_state != SYSTEM_RUNNING || oops_in_progress)
Ingo Molnaraef745f2008-08-28 11:34:43 +02007825 return;
7826 if (time_before(jiffies, prev_jiffy + HZ) && prev_jiffy)
7827 return;
7828 prev_jiffy = jiffies;
7829
Peter Zijlstra3df0fc52009-12-20 14:23:57 +01007830 printk(KERN_ERR
7831 "BUG: sleeping function called from invalid context at %s:%d\n",
7832 file, line);
7833 printk(KERN_ERR
7834 "in_atomic(): %d, irqs_disabled(): %d, pid: %d, name: %s\n",
7835 in_atomic(), irqs_disabled(),
7836 current->pid, current->comm);
Ingo Molnaraef745f2008-08-28 11:34:43 +02007837
7838 debug_show_held_locks(current);
7839 if (irqs_disabled())
7840 print_irqtrace_events(current);
7841 dump_stack();
Linus Torvalds1da177e2005-04-16 15:20:36 -07007842#endif
7843}
7844EXPORT_SYMBOL(__might_sleep);
7845#endif
7846
7847#ifdef CONFIG_MAGIC_SYSRQ
Andi Kleen3a5e4dc2007-10-15 17:00:15 +02007848static void normalize_task(struct rq *rq, struct task_struct *p)
7849{
7850 int on_rq;
Peter Zijlstra3e51f332008-05-03 18:29:28 +02007851
Andi Kleen3a5e4dc2007-10-15 17:00:15 +02007852 update_rq_clock(rq);
7853 on_rq = p->se.on_rq;
7854 if (on_rq)
7855 deactivate_task(rq, p, 0);
7856 __setscheduler(rq, p, SCHED_NORMAL, 0);
7857 if (on_rq) {
7858 activate_task(rq, p, 0);
7859 resched_task(rq->curr);
7860 }
7861}
7862
Linus Torvalds1da177e2005-04-16 15:20:36 -07007863void normalize_rt_tasks(void)
7864{
Ingo Molnara0f98a12007-06-17 18:37:45 +02007865 struct task_struct *g, *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007866 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07007867 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007868
Peter Zijlstra4cf5d772008-02-13 15:45:39 +01007869 read_lock_irqsave(&tasklist_lock, flags);
Ingo Molnara0f98a12007-06-17 18:37:45 +02007870 do_each_thread(g, p) {
Ingo Molnar178be792007-10-15 17:00:18 +02007871 /*
7872 * Only normalize user tasks:
7873 */
7874 if (!p->mm)
7875 continue;
7876
Ingo Molnardd41f592007-07-09 18:51:59 +02007877 p->se.exec_start = 0;
Ingo Molnar6cfb0d52007-08-02 17:41:40 +02007878#ifdef CONFIG_SCHEDSTATS
7879 p->se.wait_start = 0;
7880 p->se.sleep_start = 0;
Ingo Molnardd41f592007-07-09 18:51:59 +02007881 p->se.block_start = 0;
Ingo Molnar6cfb0d52007-08-02 17:41:40 +02007882#endif
Ingo Molnardd41f592007-07-09 18:51:59 +02007883
7884 if (!rt_task(p)) {
7885 /*
7886 * Renice negative nice level userspace
7887 * tasks back to 0:
7888 */
7889 if (TASK_NICE(p) < 0 && p->mm)
7890 set_user_nice(p, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007891 continue;
Ingo Molnardd41f592007-07-09 18:51:59 +02007892 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07007893
Thomas Gleixner1d615482009-11-17 14:54:03 +01007894 raw_spin_lock(&p->pi_lock);
Ingo Molnarb29739f2006-06-27 02:54:51 -07007895 rq = __task_rq_lock(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007896
Ingo Molnar178be792007-10-15 17:00:18 +02007897 normalize_task(rq, p);
Andi Kleen3a5e4dc2007-10-15 17:00:15 +02007898
Ingo Molnarb29739f2006-06-27 02:54:51 -07007899 __task_rq_unlock(rq);
Thomas Gleixner1d615482009-11-17 14:54:03 +01007900 raw_spin_unlock(&p->pi_lock);
Ingo Molnara0f98a12007-06-17 18:37:45 +02007901 } while_each_thread(g, p);
7902
Peter Zijlstra4cf5d772008-02-13 15:45:39 +01007903 read_unlock_irqrestore(&tasklist_lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007904}
7905
7906#endif /* CONFIG_MAGIC_SYSRQ */
Linus Torvalds1df5c102005-09-12 07:59:21 -07007907
7908#ifdef CONFIG_IA64
7909/*
7910 * These functions are only useful for the IA64 MCA handling.
7911 *
7912 * They can only be called when the whole system has been
7913 * stopped - every CPU needs to be quiescent, and no scheduling
7914 * activity can take place. Using them for anything else would
7915 * be a serious bug, and as a result, they aren't even visible
7916 * under any other configuration.
7917 */
7918
7919/**
7920 * curr_task - return the current task for a given cpu.
7921 * @cpu: the processor in question.
7922 *
7923 * ONLY VALID WHEN THE WHOLE SYSTEM IS STOPPED!
7924 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07007925struct task_struct *curr_task(int cpu)
Linus Torvalds1df5c102005-09-12 07:59:21 -07007926{
7927 return cpu_curr(cpu);
7928}
7929
7930/**
7931 * set_curr_task - set the current task for a given cpu.
7932 * @cpu: the processor in question.
7933 * @p: the task pointer to set.
7934 *
7935 * Description: This function must only be used when non-maskable interrupts
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01007936 * are serviced on a separate stack. It allows the architecture to switch the
7937 * notion of the current task on a cpu in a non-blocking manner. This function
Linus Torvalds1df5c102005-09-12 07:59:21 -07007938 * must be called with all CPU's synchronized, and interrupts disabled, the
7939 * and caller must save the original value of the current task (see
7940 * curr_task() above) and restore that value before reenabling interrupts and
7941 * re-starting the system.
7942 *
7943 * ONLY VALID WHEN THE WHOLE SYSTEM IS STOPPED!
7944 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07007945void set_curr_task(int cpu, struct task_struct *p)
Linus Torvalds1df5c102005-09-12 07:59:21 -07007946{
7947 cpu_curr(cpu) = p;
7948}
7949
7950#endif
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007951
Peter Zijlstrabccbe082008-02-13 15:45:40 +01007952#ifdef CONFIG_FAIR_GROUP_SCHED
7953static void free_fair_sched_group(struct task_group *tg)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007954{
7955 int i;
7956
7957 for_each_possible_cpu(i) {
7958 if (tg->cfs_rq)
7959 kfree(tg->cfs_rq[i]);
7960 if (tg->se)
7961 kfree(tg->se[i]);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007962 }
7963
7964 kfree(tg->cfs_rq);
7965 kfree(tg->se);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01007966}
7967
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02007968static
7969int alloc_fair_sched_group(struct task_group *tg, struct task_group *parent)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007970{
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007971 struct cfs_rq *cfs_rq;
Li Zefaneab17222008-10-29 17:03:22 +08007972 struct sched_entity *se;
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02007973 struct rq *rq;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007974 int i;
7975
Mike Travis434d53b2008-04-04 18:11:04 -07007976 tg->cfs_rq = kzalloc(sizeof(cfs_rq) * nr_cpu_ids, GFP_KERNEL);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007977 if (!tg->cfs_rq)
7978 goto err;
Mike Travis434d53b2008-04-04 18:11:04 -07007979 tg->se = kzalloc(sizeof(se) * nr_cpu_ids, GFP_KERNEL);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007980 if (!tg->se)
7981 goto err;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01007982
7983 tg->shares = NICE_0_LOAD;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007984
7985 for_each_possible_cpu(i) {
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02007986 rq = cpu_rq(i);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007987
Li Zefaneab17222008-10-29 17:03:22 +08007988 cfs_rq = kzalloc_node(sizeof(struct cfs_rq),
7989 GFP_KERNEL, cpu_to_node(i));
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007990 if (!cfs_rq)
7991 goto err;
7992
Li Zefaneab17222008-10-29 17:03:22 +08007993 se = kzalloc_node(sizeof(struct sched_entity),
7994 GFP_KERNEL, cpu_to_node(i));
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007995 if (!se)
Phil Carmodydfc12eb2009-12-10 14:29:37 +02007996 goto err_free_rq;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02007997
Li Zefaneab17222008-10-29 17:03:22 +08007998 init_tg_cfs_entry(tg, cfs_rq, se, i, 0, parent->se[i]);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01007999 }
8000
8001 return 1;
8002
Phil Carmodydfc12eb2009-12-10 14:29:37 +02008003 err_free_rq:
8004 kfree(cfs_rq);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008005 err:
8006 return 0;
8007}
8008
8009static inline void register_fair_sched_group(struct task_group *tg, int cpu)
8010{
8011 list_add_rcu(&tg->cfs_rq[cpu]->leaf_cfs_rq_list,
8012 &cpu_rq(cpu)->leaf_cfs_rq_list);
8013}
8014
8015static inline void unregister_fair_sched_group(struct task_group *tg, int cpu)
8016{
8017 list_del_rcu(&tg->cfs_rq[cpu]->leaf_cfs_rq_list);
8018}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008019#else /* !CONFG_FAIR_GROUP_SCHED */
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008020static inline void free_fair_sched_group(struct task_group *tg)
8021{
8022}
8023
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008024static inline
8025int alloc_fair_sched_group(struct task_group *tg, struct task_group *parent)
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008026{
8027 return 1;
8028}
8029
8030static inline void register_fair_sched_group(struct task_group *tg, int cpu)
8031{
8032}
8033
8034static inline void unregister_fair_sched_group(struct task_group *tg, int cpu)
8035{
8036}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008037#endif /* CONFIG_FAIR_GROUP_SCHED */
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008038
8039#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008040static void free_rt_sched_group(struct task_group *tg)
8041{
8042 int i;
8043
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008044 destroy_rt_bandwidth(&tg->rt_bandwidth);
8045
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008046 for_each_possible_cpu(i) {
8047 if (tg->rt_rq)
8048 kfree(tg->rt_rq[i]);
8049 if (tg->rt_se)
8050 kfree(tg->rt_se[i]);
8051 }
8052
8053 kfree(tg->rt_rq);
8054 kfree(tg->rt_se);
8055}
8056
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008057static
8058int alloc_rt_sched_group(struct task_group *tg, struct task_group *parent)
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008059{
8060 struct rt_rq *rt_rq;
Li Zefaneab17222008-10-29 17:03:22 +08008061 struct sched_rt_entity *rt_se;
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008062 struct rq *rq;
8063 int i;
8064
Mike Travis434d53b2008-04-04 18:11:04 -07008065 tg->rt_rq = kzalloc(sizeof(rt_rq) * nr_cpu_ids, GFP_KERNEL);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008066 if (!tg->rt_rq)
8067 goto err;
Mike Travis434d53b2008-04-04 18:11:04 -07008068 tg->rt_se = kzalloc(sizeof(rt_se) * nr_cpu_ids, GFP_KERNEL);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008069 if (!tg->rt_se)
8070 goto err;
8071
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008072 init_rt_bandwidth(&tg->rt_bandwidth,
8073 ktime_to_ns(def_rt_bandwidth.rt_period), 0);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008074
8075 for_each_possible_cpu(i) {
8076 rq = cpu_rq(i);
8077
Li Zefaneab17222008-10-29 17:03:22 +08008078 rt_rq = kzalloc_node(sizeof(struct rt_rq),
8079 GFP_KERNEL, cpu_to_node(i));
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008080 if (!rt_rq)
8081 goto err;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008082
Li Zefaneab17222008-10-29 17:03:22 +08008083 rt_se = kzalloc_node(sizeof(struct sched_rt_entity),
8084 GFP_KERNEL, cpu_to_node(i));
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008085 if (!rt_se)
Phil Carmodydfc12eb2009-12-10 14:29:37 +02008086 goto err_free_rq;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008087
Li Zefaneab17222008-10-29 17:03:22 +08008088 init_tg_rt_entry(tg, rt_rq, rt_se, i, 0, parent->rt_se[i]);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008089 }
8090
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008091 return 1;
8092
Phil Carmodydfc12eb2009-12-10 14:29:37 +02008093 err_free_rq:
8094 kfree(rt_rq);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008095 err:
8096 return 0;
8097}
8098
8099static inline void register_rt_sched_group(struct task_group *tg, int cpu)
8100{
8101 list_add_rcu(&tg->rt_rq[cpu]->leaf_rt_rq_list,
8102 &cpu_rq(cpu)->leaf_rt_rq_list);
8103}
8104
8105static inline void unregister_rt_sched_group(struct task_group *tg, int cpu)
8106{
8107 list_del_rcu(&tg->rt_rq[cpu]->leaf_rt_rq_list);
8108}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008109#else /* !CONFIG_RT_GROUP_SCHED */
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008110static inline void free_rt_sched_group(struct task_group *tg)
8111{
8112}
8113
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008114static inline
8115int alloc_rt_sched_group(struct task_group *tg, struct task_group *parent)
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008116{
8117 return 1;
8118}
8119
8120static inline void register_rt_sched_group(struct task_group *tg, int cpu)
8121{
8122}
8123
8124static inline void unregister_rt_sched_group(struct task_group *tg, int cpu)
8125{
8126}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008127#endif /* CONFIG_RT_GROUP_SCHED */
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008128
Dhaval Giani7c941432010-01-20 13:26:18 +01008129#ifdef CONFIG_CGROUP_SCHED
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008130static void free_sched_group(struct task_group *tg)
8131{
8132 free_fair_sched_group(tg);
8133 free_rt_sched_group(tg);
8134 kfree(tg);
8135}
8136
8137/* allocate runqueue etc for a new task group */
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008138struct task_group *sched_create_group(struct task_group *parent)
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008139{
8140 struct task_group *tg;
8141 unsigned long flags;
8142 int i;
8143
8144 tg = kzalloc(sizeof(*tg), GFP_KERNEL);
8145 if (!tg)
8146 return ERR_PTR(-ENOMEM);
8147
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008148 if (!alloc_fair_sched_group(tg, parent))
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008149 goto err;
8150
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008151 if (!alloc_rt_sched_group(tg, parent))
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008152 goto err;
8153
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008154 spin_lock_irqsave(&task_group_lock, flags);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008155 for_each_possible_cpu(i) {
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008156 register_fair_sched_group(tg, i);
8157 register_rt_sched_group(tg, i);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008158 }
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008159 list_add_rcu(&tg->list, &task_groups);
Peter Zijlstraf473aa52008-04-19 19:45:00 +02008160
8161 WARN_ON(!parent); /* root should already exist */
8162
8163 tg->parent = parent;
Peter Zijlstraf473aa52008-04-19 19:45:00 +02008164 INIT_LIST_HEAD(&tg->children);
Zhang, Yanmin09f27242030-08-14 15:56:40 +08008165 list_add_rcu(&tg->siblings, &parent->children);
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008166 spin_unlock_irqrestore(&task_group_lock, flags);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008167
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008168 return tg;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008169
8170err:
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008171 free_sched_group(tg);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008172 return ERR_PTR(-ENOMEM);
8173}
8174
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008175/* rcu callback to free various structures associated with a task group */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008176static void free_sched_group_rcu(struct rcu_head *rhp)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008177{
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008178 /* now it should be safe to free those cfs_rqs */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008179 free_sched_group(container_of(rhp, struct task_group, rcu));
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008180}
8181
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008182/* Destroy runqueue etc associated with a task group */
Ingo Molnar4cf86d72007-10-15 17:00:14 +02008183void sched_destroy_group(struct task_group *tg)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008184{
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008185 unsigned long flags;
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008186 int i;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008187
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008188 spin_lock_irqsave(&task_group_lock, flags);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008189 for_each_possible_cpu(i) {
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008190 unregister_fair_sched_group(tg, i);
8191 unregister_rt_sched_group(tg, i);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008192 }
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008193 list_del_rcu(&tg->list);
Peter Zijlstraf473aa52008-04-19 19:45:00 +02008194 list_del_rcu(&tg->siblings);
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008195 spin_unlock_irqrestore(&task_group_lock, flags);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008196
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008197 /* wait for possible concurrent references to cfs_rqs complete */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008198 call_rcu(&tg->rcu, free_sched_group_rcu);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008199}
8200
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008201/* change task's runqueue when it moves between groups.
Ingo Molnar3a252012007-10-15 17:00:12 +02008202 * The caller of this function should have put the task in its new group
8203 * by now. This function just updates tsk->se.cfs_rq and tsk->se.parent to
8204 * reflect its new group.
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008205 */
8206void sched_move_task(struct task_struct *tsk)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008207{
8208 int on_rq, running;
8209 unsigned long flags;
8210 struct rq *rq;
8211
8212 rq = task_rq_lock(tsk, &flags);
8213
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008214 update_rq_clock(rq);
8215
Dmitry Adamushko051a1d12007-12-18 15:21:13 +01008216 running = task_current(rq, tsk);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008217 on_rq = tsk->se.on_rq;
8218
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07008219 if (on_rq)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008220 dequeue_task(rq, tsk, 0);
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07008221 if (unlikely(running))
8222 tsk->sched_class->put_prev_task(rq, tsk);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008223
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008224 set_task_rq(tsk, task_cpu(tsk));
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008225
Peter Zijlstra810b3812008-02-29 15:21:01 -05008226#ifdef CONFIG_FAIR_GROUP_SCHED
8227 if (tsk->sched_class->moved_group)
Peter Zijlstra88ec22d2009-12-16 18:04:41 +01008228 tsk->sched_class->moved_group(tsk, on_rq);
Peter Zijlstra810b3812008-02-29 15:21:01 -05008229#endif
8230
Hiroshi Shimamoto0e1f3482008-03-10 11:01:20 -07008231 if (unlikely(running))
8232 tsk->sched_class->set_curr_task(rq);
8233 if (on_rq)
Thomas Gleixnerea87bb72010-01-20 20:58:57 +00008234 enqueue_task(rq, tsk, 0, false);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008235
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008236 task_rq_unlock(rq, &flags);
8237}
Dhaval Giani7c941432010-01-20 13:26:18 +01008238#endif /* CONFIG_CGROUP_SCHED */
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008239
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008240#ifdef CONFIG_FAIR_GROUP_SCHED
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008241static void __set_se_shares(struct sched_entity *se, unsigned long shares)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008242{
8243 struct cfs_rq *cfs_rq = se->cfs_rq;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008244 int on_rq;
8245
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008246 on_rq = se->on_rq;
Peter Zijlstra62fb1852008-02-25 17:34:02 +01008247 if (on_rq)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008248 dequeue_entity(cfs_rq, se, 0);
8249
8250 se->load.weight = shares;
Peter Zijlstrae05510d2008-05-05 23:56:17 +02008251 se->load.inv_weight = 0;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008252
Peter Zijlstra62fb1852008-02-25 17:34:02 +01008253 if (on_rq)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008254 enqueue_entity(cfs_rq, se, 0);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008255}
Peter Zijlstra62fb1852008-02-25 17:34:02 +01008256
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008257static void set_se_shares(struct sched_entity *se, unsigned long shares)
8258{
8259 struct cfs_rq *cfs_rq = se->cfs_rq;
8260 struct rq *rq = cfs_rq->rq;
8261 unsigned long flags;
8262
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008263 raw_spin_lock_irqsave(&rq->lock, flags);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008264 __set_se_shares(se, shares);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008265 raw_spin_unlock_irqrestore(&rq->lock, flags);
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008266}
8267
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008268static DEFINE_MUTEX(shares_mutex);
8269
Ingo Molnar4cf86d72007-10-15 17:00:14 +02008270int sched_group_set_shares(struct task_group *tg, unsigned long shares)
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008271{
8272 int i;
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008273 unsigned long flags;
Ingo Molnarc61935f2008-01-22 11:24:58 +01008274
Peter Zijlstra62fb1852008-02-25 17:34:02 +01008275 /*
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008276 * We can't change the weight of the root cgroup.
8277 */
8278 if (!tg->se[0])
8279 return -EINVAL;
8280
Peter Zijlstra18d95a22008-04-19 19:45:00 +02008281 if (shares < MIN_SHARES)
8282 shares = MIN_SHARES;
Miao Xiecb4ad1f2008-04-28 12:54:56 +08008283 else if (shares > MAX_SHARES)
8284 shares = MAX_SHARES;
Peter Zijlstra62fb1852008-02-25 17:34:02 +01008285
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008286 mutex_lock(&shares_mutex);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008287 if (tg->shares == shares)
Dhaval Giani5cb350b2007-10-15 17:00:14 +02008288 goto done;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008289
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008290 spin_lock_irqsave(&task_group_lock, flags);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008291 for_each_possible_cpu(i)
8292 unregister_fair_sched_group(tg, i);
Peter Zijlstraf473aa52008-04-19 19:45:00 +02008293 list_del_rcu(&tg->siblings);
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008294 spin_unlock_irqrestore(&task_group_lock, flags);
Srivatsa Vaddagiri6b2d7702008-01-25 21:08:00 +01008295
8296 /* wait for any ongoing reference to this group to finish */
8297 synchronize_sched();
8298
8299 /*
8300 * Now we are free to modify the group's share on each cpu
8301 * w/o tripping rebalance_share or load_balance_fair.
8302 */
8303 tg->shares = shares;
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008304 for_each_possible_cpu(i) {
8305 /*
8306 * force a rebalance
8307 */
8308 cfs_rq_set_shares(tg->cfs_rq[i], 0);
Miao Xiecb4ad1f2008-04-28 12:54:56 +08008309 set_se_shares(tg->se[i], shares);
Peter Zijlstrac09595f2008-06-27 13:41:14 +02008310 }
Srivatsa Vaddagiri6b2d7702008-01-25 21:08:00 +01008311
8312 /*
8313 * Enable load balance activity on this group, by inserting it back on
8314 * each cpu's rq->leaf_cfs_rq_list.
8315 */
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008316 spin_lock_irqsave(&task_group_lock, flags);
Peter Zijlstrabccbe082008-02-13 15:45:40 +01008317 for_each_possible_cpu(i)
8318 register_fair_sched_group(tg, i);
Peter Zijlstraf473aa52008-04-19 19:45:00 +02008319 list_add_rcu(&tg->siblings, &tg->parent->children);
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008320 spin_unlock_irqrestore(&task_group_lock, flags);
Dhaval Giani5cb350b2007-10-15 17:00:14 +02008321done:
Peter Zijlstra8ed36992008-02-13 15:45:39 +01008322 mutex_unlock(&shares_mutex);
Srivatsa Vaddagiri9b5b7752007-10-15 17:00:09 +02008323 return 0;
Srivatsa Vaddagiri29f59db2007-10-15 17:00:07 +02008324}
8325
Dhaval Giani5cb350b2007-10-15 17:00:14 +02008326unsigned long sched_group_shares(struct task_group *tg)
8327{
8328 return tg->shares;
8329}
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008330#endif
Dhaval Giani5cb350b2007-10-15 17:00:14 +02008331
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008332#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008333/*
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008334 * Ensure that the real time constraints are schedulable.
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008335 */
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008336static DEFINE_MUTEX(rt_constraints_mutex);
8337
8338static unsigned long to_ratio(u64 period, u64 runtime)
8339{
8340 if (runtime == RUNTIME_INF)
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008341 return 1ULL << 20;
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008342
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008343 return div64_u64(runtime << 20, period);
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008344}
8345
Dhaval Giani521f1a242008-02-28 15:21:56 +05308346/* Must be called with tasklist_lock held */
8347static inline int tg_has_rt_tasks(struct task_group *tg)
8348{
8349 struct task_struct *g, *p;
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008350
Dhaval Giani521f1a242008-02-28 15:21:56 +05308351 do_each_thread(g, p) {
8352 if (rt_task(p) && rt_rq_of_se(&p->rt)->tg == tg)
8353 return 1;
8354 } while_each_thread(g, p);
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008355
Dhaval Giani521f1a242008-02-28 15:21:56 +05308356 return 0;
8357}
8358
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008359struct rt_schedulable_data {
8360 struct task_group *tg;
8361 u64 rt_period;
8362 u64 rt_runtime;
8363};
8364
8365static int tg_schedulable(struct task_group *tg, void *data)
8366{
8367 struct rt_schedulable_data *d = data;
8368 struct task_group *child;
8369 unsigned long total, sum = 0;
8370 u64 period, runtime;
8371
8372 period = ktime_to_ns(tg->rt_bandwidth.rt_period);
8373 runtime = tg->rt_bandwidth.rt_runtime;
8374
8375 if (tg == d->tg) {
8376 period = d->rt_period;
8377 runtime = d->rt_runtime;
8378 }
8379
Peter Zijlstra4653f802008-09-23 15:33:44 +02008380 /*
8381 * Cannot have more runtime than the period.
8382 */
8383 if (runtime > period && runtime != RUNTIME_INF)
8384 return -EINVAL;
8385
8386 /*
8387 * Ensure we don't starve existing RT tasks.
8388 */
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008389 if (rt_bandwidth_enabled() && !runtime && tg_has_rt_tasks(tg))
8390 return -EBUSY;
8391
8392 total = to_ratio(period, runtime);
8393
Peter Zijlstra4653f802008-09-23 15:33:44 +02008394 /*
8395 * Nobody can have more than the global setting allows.
8396 */
8397 if (total > to_ratio(global_rt_period(), global_rt_runtime()))
8398 return -EINVAL;
8399
8400 /*
8401 * The sum of our children's runtime should not exceed our own.
8402 */
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008403 list_for_each_entry_rcu(child, &tg->children, siblings) {
8404 period = ktime_to_ns(child->rt_bandwidth.rt_period);
8405 runtime = child->rt_bandwidth.rt_runtime;
8406
8407 if (child == d->tg) {
8408 period = d->rt_period;
8409 runtime = d->rt_runtime;
8410 }
8411
8412 sum += to_ratio(period, runtime);
8413 }
8414
8415 if (sum > total)
8416 return -EINVAL;
8417
8418 return 0;
8419}
8420
8421static int __rt_schedulable(struct task_group *tg, u64 period, u64 runtime)
8422{
8423 struct rt_schedulable_data data = {
8424 .tg = tg,
8425 .rt_period = period,
8426 .rt_runtime = runtime,
8427 };
8428
8429 return walk_tg_tree(tg_schedulable, tg_nop, &data);
8430}
8431
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008432static int tg_set_bandwidth(struct task_group *tg,
8433 u64 rt_period, u64 rt_runtime)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008434{
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008435 int i, err = 0;
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008436
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008437 mutex_lock(&rt_constraints_mutex);
Dhaval Giani521f1a242008-02-28 15:21:56 +05308438 read_lock(&tasklist_lock);
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008439 err = __rt_schedulable(tg, rt_period, rt_runtime);
8440 if (err)
Dhaval Giani521f1a242008-02-28 15:21:56 +05308441 goto unlock;
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008442
Thomas Gleixner0986b112009-11-17 15:32:06 +01008443 raw_spin_lock_irq(&tg->rt_bandwidth.rt_runtime_lock);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008444 tg->rt_bandwidth.rt_period = ns_to_ktime(rt_period);
8445 tg->rt_bandwidth.rt_runtime = rt_runtime;
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008446
8447 for_each_possible_cpu(i) {
8448 struct rt_rq *rt_rq = tg->rt_rq[i];
8449
Thomas Gleixner0986b112009-11-17 15:32:06 +01008450 raw_spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008451 rt_rq->rt_runtime = rt_runtime;
Thomas Gleixner0986b112009-11-17 15:32:06 +01008452 raw_spin_unlock(&rt_rq->rt_runtime_lock);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008453 }
Thomas Gleixner0986b112009-11-17 15:32:06 +01008454 raw_spin_unlock_irq(&tg->rt_bandwidth.rt_runtime_lock);
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008455 unlock:
Dhaval Giani521f1a242008-02-28 15:21:56 +05308456 read_unlock(&tasklist_lock);
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008457 mutex_unlock(&rt_constraints_mutex);
8458
8459 return err;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008460}
8461
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008462int sched_group_set_rt_runtime(struct task_group *tg, long rt_runtime_us)
8463{
8464 u64 rt_runtime, rt_period;
8465
8466 rt_period = ktime_to_ns(tg->rt_bandwidth.rt_period);
8467 rt_runtime = (u64)rt_runtime_us * NSEC_PER_USEC;
8468 if (rt_runtime_us < 0)
8469 rt_runtime = RUNTIME_INF;
8470
8471 return tg_set_bandwidth(tg, rt_period, rt_runtime);
8472}
8473
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008474long sched_group_rt_runtime(struct task_group *tg)
8475{
8476 u64 rt_runtime_us;
8477
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008478 if (tg->rt_bandwidth.rt_runtime == RUNTIME_INF)
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008479 return -1;
8480
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008481 rt_runtime_us = tg->rt_bandwidth.rt_runtime;
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008482 do_div(rt_runtime_us, NSEC_PER_USEC);
8483 return rt_runtime_us;
8484}
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008485
8486int sched_group_set_rt_period(struct task_group *tg, long rt_period_us)
8487{
8488 u64 rt_runtime, rt_period;
8489
8490 rt_period = (u64)rt_period_us * NSEC_PER_USEC;
8491 rt_runtime = tg->rt_bandwidth.rt_runtime;
8492
Raistlin619b0482008-06-26 18:54:09 +02008493 if (rt_period == 0)
8494 return -EINVAL;
8495
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008496 return tg_set_bandwidth(tg, rt_period, rt_runtime);
8497}
8498
8499long sched_group_rt_period(struct task_group *tg)
8500{
8501 u64 rt_period_us;
8502
8503 rt_period_us = ktime_to_ns(tg->rt_bandwidth.rt_period);
8504 do_div(rt_period_us, NSEC_PER_USEC);
8505 return rt_period_us;
8506}
8507
8508static int sched_rt_global_constraints(void)
8509{
Peter Zijlstra4653f802008-09-23 15:33:44 +02008510 u64 runtime, period;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008511 int ret = 0;
8512
Hiroshi Shimamotoec5d4982008-09-10 17:00:19 -07008513 if (sysctl_sched_rt_period <= 0)
8514 return -EINVAL;
8515
Peter Zijlstra4653f802008-09-23 15:33:44 +02008516 runtime = global_rt_runtime();
8517 period = global_rt_period();
8518
8519 /*
8520 * Sanity check on the sysctl variables.
8521 */
8522 if (runtime > period && runtime != RUNTIME_INF)
8523 return -EINVAL;
Peter Zijlstra10b612f2008-06-19 14:22:27 +02008524
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008525 mutex_lock(&rt_constraints_mutex);
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008526 read_lock(&tasklist_lock);
Peter Zijlstra4653f802008-09-23 15:33:44 +02008527 ret = __rt_schedulable(NULL, 0, 0);
Peter Zijlstra9a7e0b12008-08-19 12:33:06 +02008528 read_unlock(&tasklist_lock);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008529 mutex_unlock(&rt_constraints_mutex);
8530
8531 return ret;
8532}
Dhaval Giani54e99122009-02-27 15:13:54 +05308533
8534int sched_rt_can_attach(struct task_group *tg, struct task_struct *tsk)
8535{
8536 /* Don't accept realtime tasks when there is no way for them to run */
8537 if (rt_task(tsk) && tg->rt_bandwidth.rt_runtime == 0)
8538 return 0;
8539
8540 return 1;
8541}
8542
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008543#else /* !CONFIG_RT_GROUP_SCHED */
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008544static int sched_rt_global_constraints(void)
8545{
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008546 unsigned long flags;
8547 int i;
8548
Hiroshi Shimamotoec5d4982008-09-10 17:00:19 -07008549 if (sysctl_sched_rt_period <= 0)
8550 return -EINVAL;
8551
Peter Zijlstra60aa6052009-05-05 17:50:21 +02008552 /*
8553 * There's always some RT tasks in the root group
8554 * -- migration, kstopmachine etc..
8555 */
8556 if (sysctl_sched_rt_runtime == 0)
8557 return -EBUSY;
8558
Thomas Gleixner0986b112009-11-17 15:32:06 +01008559 raw_spin_lock_irqsave(&def_rt_bandwidth.rt_runtime_lock, flags);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008560 for_each_possible_cpu(i) {
8561 struct rt_rq *rt_rq = &cpu_rq(i)->rt;
8562
Thomas Gleixner0986b112009-11-17 15:32:06 +01008563 raw_spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008564 rt_rq->rt_runtime = global_rt_runtime();
Thomas Gleixner0986b112009-11-17 15:32:06 +01008565 raw_spin_unlock(&rt_rq->rt_runtime_lock);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008566 }
Thomas Gleixner0986b112009-11-17 15:32:06 +01008567 raw_spin_unlock_irqrestore(&def_rt_bandwidth.rt_runtime_lock, flags);
Peter Zijlstraac086bc2008-04-19 19:44:58 +02008568
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008569 return 0;
8570}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008571#endif /* CONFIG_RT_GROUP_SCHED */
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008572
8573int sched_rt_handler(struct ctl_table *table, int write,
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07008574 void __user *buffer, size_t *lenp,
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008575 loff_t *ppos)
8576{
8577 int ret;
8578 int old_period, old_runtime;
8579 static DEFINE_MUTEX(mutex);
8580
8581 mutex_lock(&mutex);
8582 old_period = sysctl_sched_rt_period;
8583 old_runtime = sysctl_sched_rt_runtime;
8584
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07008585 ret = proc_dointvec(table, write, buffer, lenp, ppos);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008586
8587 if (!ret && write) {
8588 ret = sched_rt_global_constraints();
8589 if (ret) {
8590 sysctl_sched_rt_period = old_period;
8591 sysctl_sched_rt_runtime = old_runtime;
8592 } else {
8593 def_rt_bandwidth.rt_runtime = global_rt_runtime();
8594 def_rt_bandwidth.rt_period =
8595 ns_to_ktime(global_rt_period());
8596 }
8597 }
8598 mutex_unlock(&mutex);
8599
8600 return ret;
8601}
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008602
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008603#ifdef CONFIG_CGROUP_SCHED
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008604
8605/* return corresponding task_group object of a cgroup */
Paul Menage2b01dfe2007-10-24 18:23:50 +02008606static inline struct task_group *cgroup_tg(struct cgroup *cgrp)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008607{
Paul Menage2b01dfe2007-10-24 18:23:50 +02008608 return container_of(cgroup_subsys_state(cgrp, cpu_cgroup_subsys_id),
8609 struct task_group, css);
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008610}
8611
8612static struct cgroup_subsys_state *
Paul Menage2b01dfe2007-10-24 18:23:50 +02008613cpu_cgroup_create(struct cgroup_subsys *ss, struct cgroup *cgrp)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008614{
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008615 struct task_group *tg, *parent;
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008616
Paul Menage2b01dfe2007-10-24 18:23:50 +02008617 if (!cgrp->parent) {
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008618 /* This is early initialization for the top cgroup */
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008619 return &init_task_group.css;
8620 }
8621
Dhaval Gianiec7dc8a2008-04-19 19:44:59 +02008622 parent = cgroup_tg(cgrp->parent);
8623 tg = sched_create_group(parent);
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008624 if (IS_ERR(tg))
8625 return ERR_PTR(-ENOMEM);
8626
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008627 return &tg->css;
8628}
8629
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01008630static void
8631cpu_cgroup_destroy(struct cgroup_subsys *ss, struct cgroup *cgrp)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008632{
Paul Menage2b01dfe2007-10-24 18:23:50 +02008633 struct task_group *tg = cgroup_tg(cgrp);
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008634
8635 sched_destroy_group(tg);
8636}
8637
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01008638static int
Ben Blumbe367d02009-09-23 15:56:31 -07008639cpu_cgroup_can_attach_task(struct cgroup *cgrp, struct task_struct *tsk)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008640{
Peter Zijlstrab68aa232008-02-13 15:45:40 +01008641#ifdef CONFIG_RT_GROUP_SCHED
Dhaval Giani54e99122009-02-27 15:13:54 +05308642 if (!sched_rt_can_attach(cgroup_tg(cgrp), tsk))
Peter Zijlstrab68aa232008-02-13 15:45:40 +01008643 return -EINVAL;
8644#else
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008645 /* We don't support RT-tasks being in separate groups */
8646 if (tsk->sched_class != &fair_sched_class)
8647 return -EINVAL;
Peter Zijlstrab68aa232008-02-13 15:45:40 +01008648#endif
Ben Blumbe367d02009-09-23 15:56:31 -07008649 return 0;
8650}
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008651
Ben Blumbe367d02009-09-23 15:56:31 -07008652static int
8653cpu_cgroup_can_attach(struct cgroup_subsys *ss, struct cgroup *cgrp,
8654 struct task_struct *tsk, bool threadgroup)
8655{
8656 int retval = cpu_cgroup_can_attach_task(cgrp, tsk);
8657 if (retval)
8658 return retval;
8659 if (threadgroup) {
8660 struct task_struct *c;
8661 rcu_read_lock();
8662 list_for_each_entry_rcu(c, &tsk->thread_group, thread_group) {
8663 retval = cpu_cgroup_can_attach_task(cgrp, c);
8664 if (retval) {
8665 rcu_read_unlock();
8666 return retval;
8667 }
8668 }
8669 rcu_read_unlock();
8670 }
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008671 return 0;
8672}
8673
8674static void
Paul Menage2b01dfe2007-10-24 18:23:50 +02008675cpu_cgroup_attach(struct cgroup_subsys *ss, struct cgroup *cgrp,
Ben Blumbe367d02009-09-23 15:56:31 -07008676 struct cgroup *old_cont, struct task_struct *tsk,
8677 bool threadgroup)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008678{
8679 sched_move_task(tsk);
Ben Blumbe367d02009-09-23 15:56:31 -07008680 if (threadgroup) {
8681 struct task_struct *c;
8682 rcu_read_lock();
8683 list_for_each_entry_rcu(c, &tsk->thread_group, thread_group) {
8684 sched_move_task(c);
8685 }
8686 rcu_read_unlock();
8687 }
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008688}
8689
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008690#ifdef CONFIG_FAIR_GROUP_SCHED
Paul Menagef4c753b2008-04-29 00:59:56 -07008691static int cpu_shares_write_u64(struct cgroup *cgrp, struct cftype *cftype,
Paul Menage2b01dfe2007-10-24 18:23:50 +02008692 u64 shareval)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008693{
Paul Menage2b01dfe2007-10-24 18:23:50 +02008694 return sched_group_set_shares(cgroup_tg(cgrp), shareval);
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008695}
8696
Paul Menagef4c753b2008-04-29 00:59:56 -07008697static u64 cpu_shares_read_u64(struct cgroup *cgrp, struct cftype *cft)
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008698{
Paul Menage2b01dfe2007-10-24 18:23:50 +02008699 struct task_group *tg = cgroup_tg(cgrp);
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008700
8701 return (u64) tg->shares;
8702}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008703#endif /* CONFIG_FAIR_GROUP_SCHED */
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008704
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008705#ifdef CONFIG_RT_GROUP_SCHED
Mirco Tischler0c708142008-05-14 16:05:46 -07008706static int cpu_rt_runtime_write(struct cgroup *cgrp, struct cftype *cft,
Paul Menage06ecb272008-04-29 01:00:06 -07008707 s64 val)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008708{
Paul Menage06ecb272008-04-29 01:00:06 -07008709 return sched_group_set_rt_runtime(cgroup_tg(cgrp), val);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008710}
8711
Paul Menage06ecb272008-04-29 01:00:06 -07008712static s64 cpu_rt_runtime_read(struct cgroup *cgrp, struct cftype *cft)
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008713{
Paul Menage06ecb272008-04-29 01:00:06 -07008714 return sched_group_rt_runtime(cgroup_tg(cgrp));
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008715}
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008716
8717static int cpu_rt_period_write_uint(struct cgroup *cgrp, struct cftype *cftype,
8718 u64 rt_period_us)
8719{
8720 return sched_group_set_rt_period(cgroup_tg(cgrp), rt_period_us);
8721}
8722
8723static u64 cpu_rt_period_read_uint(struct cgroup *cgrp, struct cftype *cft)
8724{
8725 return sched_group_rt_period(cgroup_tg(cgrp));
8726}
Dhaval Giani6d6bc0a2008-05-30 14:23:45 +02008727#endif /* CONFIG_RT_GROUP_SCHED */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008728
Paul Menagefe5c7cc2007-10-29 21:18:11 +01008729static struct cftype cpu_files[] = {
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008730#ifdef CONFIG_FAIR_GROUP_SCHED
Paul Menagefe5c7cc2007-10-29 21:18:11 +01008731 {
8732 .name = "shares",
Paul Menagef4c753b2008-04-29 00:59:56 -07008733 .read_u64 = cpu_shares_read_u64,
8734 .write_u64 = cpu_shares_write_u64,
Paul Menagefe5c7cc2007-10-29 21:18:11 +01008735 },
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008736#endif
8737#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008738 {
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +01008739 .name = "rt_runtime_us",
Paul Menage06ecb272008-04-29 01:00:06 -07008740 .read_s64 = cpu_rt_runtime_read,
8741 .write_s64 = cpu_rt_runtime_write,
Peter Zijlstra6f505b12008-01-25 21:08:30 +01008742 },
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008743 {
8744 .name = "rt_period_us",
Paul Menagef4c753b2008-04-29 00:59:56 -07008745 .read_u64 = cpu_rt_period_read_uint,
8746 .write_u64 = cpu_rt_period_write_uint,
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +02008747 },
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008748#endif
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008749};
8750
8751static int cpu_cgroup_populate(struct cgroup_subsys *ss, struct cgroup *cont)
8752{
Paul Menagefe5c7cc2007-10-29 21:18:11 +01008753 return cgroup_add_files(cont, ss, cpu_files, ARRAY_SIZE(cpu_files));
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008754}
8755
8756struct cgroup_subsys cpu_cgroup_subsys = {
Ingo Molnar38605ca2007-10-29 21:18:11 +01008757 .name = "cpu",
8758 .create = cpu_cgroup_create,
8759 .destroy = cpu_cgroup_destroy,
8760 .can_attach = cpu_cgroup_can_attach,
8761 .attach = cpu_cgroup_attach,
8762 .populate = cpu_cgroup_populate,
8763 .subsys_id = cpu_cgroup_subsys_id,
Srivatsa Vaddagiri68318b82007-10-18 23:41:03 -07008764 .early_init = 1,
8765};
8766
Peter Zijlstra052f1dc2008-02-13 15:45:40 +01008767#endif /* CONFIG_CGROUP_SCHED */
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008768
8769#ifdef CONFIG_CGROUP_CPUACCT
8770
8771/*
8772 * CPU accounting code for task groups.
8773 *
8774 * Based on the work by Paul Menage (menage@google.com) and Balbir Singh
8775 * (balbir@in.ibm.com).
8776 */
8777
Bharata B Rao934352f2008-11-10 20:41:13 +05308778/* track cpu usage of a group of tasks and its child groups */
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008779struct cpuacct {
8780 struct cgroup_subsys_state css;
8781 /* cpuusage holds pointer to a u64-type object on every cpu */
8782 u64 *cpuusage;
Bharata B Raoef12fef2009-03-31 10:02:22 +05308783 struct percpu_counter cpustat[CPUACCT_STAT_NSTATS];
Bharata B Rao934352f2008-11-10 20:41:13 +05308784 struct cpuacct *parent;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008785};
8786
8787struct cgroup_subsys cpuacct_subsys;
8788
8789/* return cpu accounting group corresponding to this container */
Dhaval Giani32cd7562008-02-29 10:02:43 +05308790static inline struct cpuacct *cgroup_ca(struct cgroup *cgrp)
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008791{
Dhaval Giani32cd7562008-02-29 10:02:43 +05308792 return container_of(cgroup_subsys_state(cgrp, cpuacct_subsys_id),
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008793 struct cpuacct, css);
8794}
8795
8796/* return cpu accounting group to which this task belongs */
8797static inline struct cpuacct *task_ca(struct task_struct *tsk)
8798{
8799 return container_of(task_subsys_state(tsk, cpuacct_subsys_id),
8800 struct cpuacct, css);
8801}
8802
8803/* create a new cpu accounting group */
8804static struct cgroup_subsys_state *cpuacct_create(
Dhaval Giani32cd7562008-02-29 10:02:43 +05308805 struct cgroup_subsys *ss, struct cgroup *cgrp)
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008806{
8807 struct cpuacct *ca = kzalloc(sizeof(*ca), GFP_KERNEL);
Bharata B Raoef12fef2009-03-31 10:02:22 +05308808 int i;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008809
8810 if (!ca)
Bharata B Raoef12fef2009-03-31 10:02:22 +05308811 goto out;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008812
8813 ca->cpuusage = alloc_percpu(u64);
Bharata B Raoef12fef2009-03-31 10:02:22 +05308814 if (!ca->cpuusage)
8815 goto out_free_ca;
8816
8817 for (i = 0; i < CPUACCT_STAT_NSTATS; i++)
8818 if (percpu_counter_init(&ca->cpustat[i], 0))
8819 goto out_free_counters;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008820
Bharata B Rao934352f2008-11-10 20:41:13 +05308821 if (cgrp->parent)
8822 ca->parent = cgroup_ca(cgrp->parent);
8823
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008824 return &ca->css;
Bharata B Raoef12fef2009-03-31 10:02:22 +05308825
8826out_free_counters:
8827 while (--i >= 0)
8828 percpu_counter_destroy(&ca->cpustat[i]);
8829 free_percpu(ca->cpuusage);
8830out_free_ca:
8831 kfree(ca);
8832out:
8833 return ERR_PTR(-ENOMEM);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008834}
8835
8836/* destroy an existing cpu accounting group */
Ingo Molnar41a2d6c2007-12-05 15:46:09 +01008837static void
Dhaval Giani32cd7562008-02-29 10:02:43 +05308838cpuacct_destroy(struct cgroup_subsys *ss, struct cgroup *cgrp)
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008839{
Dhaval Giani32cd7562008-02-29 10:02:43 +05308840 struct cpuacct *ca = cgroup_ca(cgrp);
Bharata B Raoef12fef2009-03-31 10:02:22 +05308841 int i;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008842
Bharata B Raoef12fef2009-03-31 10:02:22 +05308843 for (i = 0; i < CPUACCT_STAT_NSTATS; i++)
8844 percpu_counter_destroy(&ca->cpustat[i]);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008845 free_percpu(ca->cpuusage);
8846 kfree(ca);
8847}
8848
Ken Chen720f5492008-12-15 22:02:01 -08008849static u64 cpuacct_cpuusage_read(struct cpuacct *ca, int cpu)
8850{
Rusty Russellb36128c2009-02-20 16:29:08 +09008851 u64 *cpuusage = per_cpu_ptr(ca->cpuusage, cpu);
Ken Chen720f5492008-12-15 22:02:01 -08008852 u64 data;
8853
8854#ifndef CONFIG_64BIT
8855 /*
8856 * Take rq->lock to make 64-bit read safe on 32-bit platforms.
8857 */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008858 raw_spin_lock_irq(&cpu_rq(cpu)->lock);
Ken Chen720f5492008-12-15 22:02:01 -08008859 data = *cpuusage;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008860 raw_spin_unlock_irq(&cpu_rq(cpu)->lock);
Ken Chen720f5492008-12-15 22:02:01 -08008861#else
8862 data = *cpuusage;
8863#endif
8864
8865 return data;
8866}
8867
8868static void cpuacct_cpuusage_write(struct cpuacct *ca, int cpu, u64 val)
8869{
Rusty Russellb36128c2009-02-20 16:29:08 +09008870 u64 *cpuusage = per_cpu_ptr(ca->cpuusage, cpu);
Ken Chen720f5492008-12-15 22:02:01 -08008871
8872#ifndef CONFIG_64BIT
8873 /*
8874 * Take rq->lock to make 64-bit write safe on 32-bit platforms.
8875 */
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008876 raw_spin_lock_irq(&cpu_rq(cpu)->lock);
Ken Chen720f5492008-12-15 22:02:01 -08008877 *cpuusage = val;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01008878 raw_spin_unlock_irq(&cpu_rq(cpu)->lock);
Ken Chen720f5492008-12-15 22:02:01 -08008879#else
8880 *cpuusage = val;
8881#endif
8882}
8883
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008884/* return total cpu usage (in nanoseconds) of a group */
Dhaval Giani32cd7562008-02-29 10:02:43 +05308885static u64 cpuusage_read(struct cgroup *cgrp, struct cftype *cft)
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008886{
Dhaval Giani32cd7562008-02-29 10:02:43 +05308887 struct cpuacct *ca = cgroup_ca(cgrp);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008888 u64 totalcpuusage = 0;
8889 int i;
8890
Ken Chen720f5492008-12-15 22:02:01 -08008891 for_each_present_cpu(i)
8892 totalcpuusage += cpuacct_cpuusage_read(ca, i);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008893
8894 return totalcpuusage;
8895}
8896
Dhaval Giani0297b802008-02-29 10:02:44 +05308897static int cpuusage_write(struct cgroup *cgrp, struct cftype *cftype,
8898 u64 reset)
8899{
8900 struct cpuacct *ca = cgroup_ca(cgrp);
8901 int err = 0;
8902 int i;
8903
8904 if (reset) {
8905 err = -EINVAL;
8906 goto out;
8907 }
8908
Ken Chen720f5492008-12-15 22:02:01 -08008909 for_each_present_cpu(i)
8910 cpuacct_cpuusage_write(ca, i, 0);
Dhaval Giani0297b802008-02-29 10:02:44 +05308911
Dhaval Giani0297b802008-02-29 10:02:44 +05308912out:
8913 return err;
8914}
8915
Ken Chene9515c32008-12-15 22:04:15 -08008916static int cpuacct_percpu_seq_read(struct cgroup *cgroup, struct cftype *cft,
8917 struct seq_file *m)
8918{
8919 struct cpuacct *ca = cgroup_ca(cgroup);
8920 u64 percpu;
8921 int i;
8922
8923 for_each_present_cpu(i) {
8924 percpu = cpuacct_cpuusage_read(ca, i);
8925 seq_printf(m, "%llu ", (unsigned long long) percpu);
8926 }
8927 seq_printf(m, "\n");
8928 return 0;
8929}
8930
Bharata B Raoef12fef2009-03-31 10:02:22 +05308931static const char *cpuacct_stat_desc[] = {
8932 [CPUACCT_STAT_USER] = "user",
8933 [CPUACCT_STAT_SYSTEM] = "system",
8934};
8935
8936static int cpuacct_stats_show(struct cgroup *cgrp, struct cftype *cft,
8937 struct cgroup_map_cb *cb)
8938{
8939 struct cpuacct *ca = cgroup_ca(cgrp);
8940 int i;
8941
8942 for (i = 0; i < CPUACCT_STAT_NSTATS; i++) {
8943 s64 val = percpu_counter_read(&ca->cpustat[i]);
8944 val = cputime64_to_clock_t(val);
8945 cb->fill(cb, cpuacct_stat_desc[i], val);
8946 }
8947 return 0;
8948}
8949
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008950static struct cftype files[] = {
8951 {
8952 .name = "usage",
Paul Menagef4c753b2008-04-29 00:59:56 -07008953 .read_u64 = cpuusage_read,
8954 .write_u64 = cpuusage_write,
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008955 },
Ken Chene9515c32008-12-15 22:04:15 -08008956 {
8957 .name = "usage_percpu",
8958 .read_seq_string = cpuacct_percpu_seq_read,
8959 },
Bharata B Raoef12fef2009-03-31 10:02:22 +05308960 {
8961 .name = "stat",
8962 .read_map = cpuacct_stats_show,
8963 },
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008964};
8965
Dhaval Giani32cd7562008-02-29 10:02:43 +05308966static int cpuacct_populate(struct cgroup_subsys *ss, struct cgroup *cgrp)
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008967{
Dhaval Giani32cd7562008-02-29 10:02:43 +05308968 return cgroup_add_files(cgrp, ss, files, ARRAY_SIZE(files));
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008969}
8970
8971/*
8972 * charge this task's execution time to its accounting group.
8973 *
8974 * called with rq->lock held.
8975 */
8976static void cpuacct_charge(struct task_struct *tsk, u64 cputime)
8977{
8978 struct cpuacct *ca;
Bharata B Rao934352f2008-11-10 20:41:13 +05308979 int cpu;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008980
Li Zefanc40c6f82009-02-26 15:40:15 +08008981 if (unlikely(!cpuacct_subsys.active))
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008982 return;
8983
Bharata B Rao934352f2008-11-10 20:41:13 +05308984 cpu = task_cpu(tsk);
Bharata B Raoa18b83b2009-03-23 10:02:53 +05308985
8986 rcu_read_lock();
8987
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008988 ca = task_ca(tsk);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008989
Bharata B Rao934352f2008-11-10 20:41:13 +05308990 for (; ca; ca = ca->parent) {
Rusty Russellb36128c2009-02-20 16:29:08 +09008991 u64 *cpuusage = per_cpu_ptr(ca->cpuusage, cpu);
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008992 *cpuusage += cputime;
8993 }
Bharata B Raoa18b83b2009-03-23 10:02:53 +05308994
8995 rcu_read_unlock();
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01008996}
8997
Bharata B Raoef12fef2009-03-31 10:02:22 +05308998/*
8999 * Charge the system/user time to the task's accounting group.
9000 */
9001static void cpuacct_update_stats(struct task_struct *tsk,
9002 enum cpuacct_stat_index idx, cputime_t val)
9003{
9004 struct cpuacct *ca;
9005
9006 if (unlikely(!cpuacct_subsys.active))
9007 return;
9008
9009 rcu_read_lock();
9010 ca = task_ca(tsk);
9011
9012 do {
9013 percpu_counter_add(&ca->cpustat[idx], val);
9014 ca = ca->parent;
9015 } while (ca);
9016 rcu_read_unlock();
9017}
9018
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +01009019struct cgroup_subsys cpuacct_subsys = {
9020 .name = "cpuacct",
9021 .create = cpuacct_create,
9022 .destroy = cpuacct_destroy,
9023 .populate = cpuacct_populate,
9024 .subsys_id = cpuacct_subsys_id,
9025};
9026#endif /* CONFIG_CGROUP_CPUACCT */
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009027
9028#ifndef CONFIG_SMP
9029
9030int rcu_expedited_torture_stats(char *page)
9031{
9032 return 0;
9033}
9034EXPORT_SYMBOL_GPL(rcu_expedited_torture_stats);
9035
9036void synchronize_sched_expedited(void)
9037{
9038}
9039EXPORT_SYMBOL_GPL(synchronize_sched_expedited);
9040
9041#else /* #ifndef CONFIG_SMP */
9042
9043static DEFINE_PER_CPU(struct migration_req, rcu_migration_req);
9044static DEFINE_MUTEX(rcu_sched_expedited_mutex);
9045
9046#define RCU_EXPEDITED_STATE_POST -2
9047#define RCU_EXPEDITED_STATE_IDLE -1
9048
9049static int rcu_expedited_state = RCU_EXPEDITED_STATE_IDLE;
9050
9051int rcu_expedited_torture_stats(char *page)
9052{
9053 int cnt = 0;
9054 int cpu;
9055
9056 cnt += sprintf(&page[cnt], "state: %d /", rcu_expedited_state);
9057 for_each_online_cpu(cpu) {
9058 cnt += sprintf(&page[cnt], " %d:%d",
9059 cpu, per_cpu(rcu_migration_req, cpu).dest_cpu);
9060 }
9061 cnt += sprintf(&page[cnt], "\n");
9062 return cnt;
9063}
9064EXPORT_SYMBOL_GPL(rcu_expedited_torture_stats);
9065
9066static long synchronize_sched_expedited_count;
9067
9068/*
9069 * Wait for an rcu-sched grace period to elapse, but use "big hammer"
9070 * approach to force grace period to end quickly. This consumes
9071 * significant time on all CPUs, and is thus not recommended for
9072 * any sort of common-case code.
9073 *
9074 * Note that it is illegal to call this function while holding any
9075 * lock that is acquired by a CPU-hotplug notifier. Failing to
9076 * observe this restriction will result in deadlock.
9077 */
9078void synchronize_sched_expedited(void)
9079{
9080 int cpu;
9081 unsigned long flags;
9082 bool need_full_sync = 0;
9083 struct rq *rq;
9084 struct migration_req *req;
9085 long snap;
9086 int trycount = 0;
9087
9088 smp_mb(); /* ensure prior mod happens before capturing snap. */
9089 snap = ACCESS_ONCE(synchronize_sched_expedited_count) + 1;
9090 get_online_cpus();
9091 while (!mutex_trylock(&rcu_sched_expedited_mutex)) {
9092 put_online_cpus();
9093 if (trycount++ < 10)
9094 udelay(trycount * num_online_cpus());
9095 else {
9096 synchronize_sched();
9097 return;
9098 }
9099 if (ACCESS_ONCE(synchronize_sched_expedited_count) - snap > 0) {
9100 smp_mb(); /* ensure test happens before caller kfree */
9101 return;
9102 }
9103 get_online_cpus();
9104 }
9105 rcu_expedited_state = RCU_EXPEDITED_STATE_POST;
9106 for_each_online_cpu(cpu) {
9107 rq = cpu_rq(cpu);
9108 req = &per_cpu(rcu_migration_req, cpu);
9109 init_completion(&req->done);
9110 req->task = NULL;
9111 req->dest_cpu = RCU_MIGRATION_NEED_QS;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01009112 raw_spin_lock_irqsave(&rq->lock, flags);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009113 list_add(&req->list, &rq->migration_queue);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01009114 raw_spin_unlock_irqrestore(&rq->lock, flags);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009115 wake_up_process(rq->migration_thread);
9116 }
9117 for_each_online_cpu(cpu) {
9118 rcu_expedited_state = cpu;
9119 req = &per_cpu(rcu_migration_req, cpu);
9120 rq = cpu_rq(cpu);
9121 wait_for_completion(&req->done);
Thomas Gleixner05fa7852009-11-17 14:28:38 +01009122 raw_spin_lock_irqsave(&rq->lock, flags);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009123 if (unlikely(req->dest_cpu == RCU_MIGRATION_MUST_SYNC))
9124 need_full_sync = 1;
9125 req->dest_cpu = RCU_MIGRATION_IDLE;
Thomas Gleixner05fa7852009-11-17 14:28:38 +01009126 raw_spin_unlock_irqrestore(&rq->lock, flags);
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009127 }
9128 rcu_expedited_state = RCU_EXPEDITED_STATE_IDLE;
Paul E. McKenney956539b2009-11-10 13:37:20 -08009129 synchronize_sched_expedited_count++;
Paul E. McKenney03b042b2009-06-25 09:08:16 -07009130 mutex_unlock(&rcu_sched_expedited_mutex);
9131 put_online_cpus();
9132 if (need_full_sync)
9133 synchronize_sched();
9134}
9135EXPORT_SYMBOL_GPL(synchronize_sched_expedited);
9136
9137#endif /* #else #ifndef CONFIG_SMP */