Eliminate lock_cpu_hotplug in kernel/schedc
[linux-2.6-block.git] / kernel / cpu.c
CommitLineData
1da177e4
LT
1/* CPU control.
2 * (C) 2001, 2002, 2003, 2004 Rusty Russell
3 *
4 * This code is licenced under the GPL.
5 */
6#include <linux/proc_fs.h>
7#include <linux/smp.h>
8#include <linux/init.h>
9#include <linux/notifier.h>
10#include <linux/sched.h>
11#include <linux/unistd.h>
12#include <linux/cpu.h>
13#include <linux/module.h>
14#include <linux/kthread.h>
15#include <linux/stop_machine.h>
81615b62 16#include <linux/mutex.h>
1da177e4
LT
17
18/* This protects CPUs going up and down... */
aa953877
LT
19static DEFINE_MUTEX(cpu_add_remove_lock);
20static DEFINE_MUTEX(cpu_bitmask_lock);
1da177e4 21
bd5349cf 22static __cpuinitdata RAW_NOTIFIER_HEAD(cpu_chain);
1da177e4 23
e3920fb4
RW
24/* If set, cpu_up and cpu_down will return -EBUSY and do nothing.
25 * Should always be manipulated under cpu_add_remove_lock
26 */
27static int cpu_hotplug_disabled;
28
a9d9baa1 29#ifdef CONFIG_HOTPLUG_CPU
90d45d17 30
aa953877
LT
31/* Crappy recursive lock-takers in cpufreq! Complain loudly about idiots */
32static struct task_struct *recursive;
33static int recursive_depth;
90d45d17 34
a9d9baa1
AR
35void lock_cpu_hotplug(void)
36{
aa953877
LT
37 struct task_struct *tsk = current;
38
39 if (tsk == recursive) {
40 static int warnings = 10;
41 if (warnings) {
42 printk(KERN_ERR "Lukewarm IQ detected in hotplug locking\n");
43 WARN_ON(1);
44 warnings--;
45 }
46 recursive_depth++;
47 return;
48 }
49 mutex_lock(&cpu_bitmask_lock);
50 recursive = tsk;
a9d9baa1
AR
51}
52EXPORT_SYMBOL_GPL(lock_cpu_hotplug);
90d45d17 53
a9d9baa1
AR
54void unlock_cpu_hotplug(void)
55{
aa953877
LT
56 WARN_ON(recursive != current);
57 if (recursive_depth) {
58 recursive_depth--;
59 return;
a9d9baa1 60 }
aa953877 61 recursive = NULL;
4b96b1a1 62 mutex_unlock(&cpu_bitmask_lock);
a9d9baa1
AR
63}
64EXPORT_SYMBOL_GPL(unlock_cpu_hotplug);
65
a9d9baa1 66#endif /* CONFIG_HOTPLUG_CPU */
90d45d17 67
1da177e4 68/* Need to know about CPUs going up/down? */
65edc68c 69int __cpuinit register_cpu_notifier(struct notifier_block *nb)
1da177e4 70{
bd5349cf
NB
71 int ret;
72 mutex_lock(&cpu_add_remove_lock);
73 ret = raw_notifier_chain_register(&cpu_chain, nb);
74 mutex_unlock(&cpu_add_remove_lock);
75 return ret;
1da177e4 76}
65edc68c
CS
77
78#ifdef CONFIG_HOTPLUG_CPU
79
1da177e4
LT
80EXPORT_SYMBOL(register_cpu_notifier);
81
82void unregister_cpu_notifier(struct notifier_block *nb)
83{
bd5349cf
NB
84 mutex_lock(&cpu_add_remove_lock);
85 raw_notifier_chain_unregister(&cpu_chain, nb);
86 mutex_unlock(&cpu_add_remove_lock);
1da177e4
LT
87}
88EXPORT_SYMBOL(unregister_cpu_notifier);
89
1da177e4
LT
90static inline void check_for_tasks(int cpu)
91{
92 struct task_struct *p;
93
94 write_lock_irq(&tasklist_lock);
95 for_each_process(p) {
96 if (task_cpu(p) == cpu &&
97 (!cputime_eq(p->utime, cputime_zero) ||
98 !cputime_eq(p->stime, cputime_zero)))
99 printk(KERN_WARNING "Task %s (pid = %d) is on cpu %d\
100 (state = %ld, flags = %lx) \n",
101 p->comm, p->pid, cpu, p->state, p->flags);
102 }
103 write_unlock_irq(&tasklist_lock);
104}
105
106/* Take this CPU down. */
107static int take_cpu_down(void *unused)
108{
109 int err;
110
1da177e4
LT
111 /* Ensure this CPU doesn't handle any more interrupts. */
112 err = __cpu_disable();
113 if (err < 0)
f3705136 114 return err;
1da177e4 115
f3705136
ZM
116 /* Force idle task to run as soon as we yield: it should
117 immediately notice cpu is offline and die quickly. */
118 sched_idle_next();
119 return 0;
1da177e4
LT
120}
121
e3920fb4
RW
122/* Requires cpu_add_remove_lock to be held */
123static int _cpu_down(unsigned int cpu)
1da177e4
LT
124{
125 int err;
126 struct task_struct *p;
127 cpumask_t old_allowed, tmp;
128
e3920fb4
RW
129 if (num_online_cpus() == 1)
130 return -EBUSY;
1da177e4 131
e3920fb4
RW
132 if (!cpu_online(cpu))
133 return -EINVAL;
1da177e4 134
baaca49f
GS
135 raw_notifier_call_chain(&cpu_chain, CPU_LOCK_ACQUIRE,
136 (void *)(long)cpu);
bd5349cf 137 err = raw_notifier_call_chain(&cpu_chain, CPU_DOWN_PREPARE,
1da177e4
LT
138 (void *)(long)cpu);
139 if (err == NOTIFY_BAD) {
140 printk("%s: attempt to take down CPU %u failed\n",
141 __FUNCTION__, cpu);
baaca49f
GS
142 err = -EINVAL;
143 goto out_release;
1da177e4
LT
144 }
145
146 /* Ensure that we are not runnable on dying cpu */
147 old_allowed = current->cpus_allowed;
148 tmp = CPU_MASK_ALL;
149 cpu_clear(cpu, tmp);
150 set_cpus_allowed(current, tmp);
151
aa953877 152 mutex_lock(&cpu_bitmask_lock);
1da177e4 153 p = __stop_machine_run(take_cpu_down, NULL, cpu);
aa953877
LT
154 mutex_unlock(&cpu_bitmask_lock);
155
8fa1d7d3 156 if (IS_ERR(p) || cpu_online(cpu)) {
1da177e4 157 /* CPU didn't die: tell everyone. Can't complain. */
bd5349cf 158 if (raw_notifier_call_chain(&cpu_chain, CPU_DOWN_FAILED,
1da177e4
LT
159 (void *)(long)cpu) == NOTIFY_BAD)
160 BUG();
161
8fa1d7d3
ST
162 if (IS_ERR(p)) {
163 err = PTR_ERR(p);
164 goto out_allowed;
165 }
1da177e4 166 goto out_thread;
8fa1d7d3 167 }
1da177e4
LT
168
169 /* Wait for it to sleep (leaving idle task). */
170 while (!idle_cpu(cpu))
171 yield();
172
173 /* This actually kills the CPU. */
174 __cpu_die(cpu);
175
176 /* Move it here so it can run. */
177 kthread_bind(p, get_cpu());
178 put_cpu();
179
180 /* CPU is completely dead: tell everyone. Too late to complain. */
bd5349cf 181 if (raw_notifier_call_chain(&cpu_chain, CPU_DEAD,
e041c683 182 (void *)(long)cpu) == NOTIFY_BAD)
1da177e4
LT
183 BUG();
184
185 check_for_tasks(cpu);
186
187out_thread:
188 err = kthread_stop(p);
189out_allowed:
190 set_cpus_allowed(current, old_allowed);
baaca49f
GS
191out_release:
192 raw_notifier_call_chain(&cpu_chain, CPU_LOCK_RELEASE,
193 (void *)(long)cpu);
e3920fb4
RW
194 return err;
195}
196
197int cpu_down(unsigned int cpu)
198{
199 int err = 0;
200
201 mutex_lock(&cpu_add_remove_lock);
202 if (cpu_hotplug_disabled)
203 err = -EBUSY;
204 else
205 err = _cpu_down(cpu);
206
aa953877 207 mutex_unlock(&cpu_add_remove_lock);
1da177e4
LT
208 return err;
209}
210#endif /*CONFIG_HOTPLUG_CPU*/
211
e3920fb4 212/* Requires cpu_add_remove_lock to be held */
b282b6f8 213static int __cpuinit _cpu_up(unsigned int cpu)
1da177e4 214{
baaca49f 215 int ret, nr_calls = 0;
1da177e4
LT
216 void *hcpu = (void *)(long)cpu;
217
e3920fb4
RW
218 if (cpu_online(cpu) || !cpu_present(cpu))
219 return -EINVAL;
90d45d17 220
baaca49f
GS
221 raw_notifier_call_chain(&cpu_chain, CPU_LOCK_ACQUIRE, hcpu);
222 ret = __raw_notifier_call_chain(&cpu_chain, CPU_UP_PREPARE, hcpu,
223 -1, &nr_calls);
1da177e4
LT
224 if (ret == NOTIFY_BAD) {
225 printk("%s: attempt to bring up CPU %u failed\n",
226 __FUNCTION__, cpu);
227 ret = -EINVAL;
228 goto out_notify;
229 }
230
231 /* Arch-specific enabling code. */
aa953877 232 mutex_lock(&cpu_bitmask_lock);
1da177e4 233 ret = __cpu_up(cpu);
aa953877 234 mutex_unlock(&cpu_bitmask_lock);
1da177e4
LT
235 if (ret != 0)
236 goto out_notify;
6978c705 237 BUG_ON(!cpu_online(cpu));
1da177e4
LT
238
239 /* Now call notifier in preparation. */
bd5349cf 240 raw_notifier_call_chain(&cpu_chain, CPU_ONLINE, hcpu);
1da177e4
LT
241
242out_notify:
243 if (ret != 0)
baaca49f
GS
244 __raw_notifier_call_chain(&cpu_chain,
245 CPU_UP_CANCELED, hcpu, nr_calls, NULL);
246 raw_notifier_call_chain(&cpu_chain, CPU_LOCK_RELEASE, hcpu);
e3920fb4
RW
247
248 return ret;
249}
250
b282b6f8 251int __cpuinit cpu_up(unsigned int cpu)
e3920fb4
RW
252{
253 int err = 0;
254
255 mutex_lock(&cpu_add_remove_lock);
256 if (cpu_hotplug_disabled)
257 err = -EBUSY;
258 else
259 err = _cpu_up(cpu);
260
261 mutex_unlock(&cpu_add_remove_lock);
262 return err;
263}
264
265#ifdef CONFIG_SUSPEND_SMP
1d64b9cb
RW
266/* Needed to prevent the microcode driver from requesting firmware in its CPU
267 * hotplug notifier during the suspend/resume.
268 */
269int suspend_cpu_hotplug;
270EXPORT_SYMBOL(suspend_cpu_hotplug);
271
e3920fb4
RW
272static cpumask_t frozen_cpus;
273
274int disable_nonboot_cpus(void)
275{
e1d9fd2e 276 int cpu, first_cpu, error = 0;
e3920fb4
RW
277
278 mutex_lock(&cpu_add_remove_lock);
1d64b9cb
RW
279 suspend_cpu_hotplug = 1;
280 first_cpu = first_cpu(cpu_online_map);
e3920fb4
RW
281 /* We take down all of the non-boot CPUs in one shot to avoid races
282 * with the userspace trying to use the CPU hotplug at the same time
283 */
284 cpus_clear(frozen_cpus);
285 printk("Disabling non-boot CPUs ...\n");
286 for_each_online_cpu(cpu) {
287 if (cpu == first_cpu)
288 continue;
289 error = _cpu_down(cpu);
290 if (!error) {
291 cpu_set(cpu, frozen_cpus);
292 printk("CPU%d is down\n", cpu);
293 } else {
294 printk(KERN_ERR "Error taking CPU%d down: %d\n",
295 cpu, error);
296 break;
297 }
298 }
299 if (!error) {
300 BUG_ON(num_online_cpus() > 1);
301 /* Make sure the CPUs won't be enabled by someone else */
302 cpu_hotplug_disabled = 1;
303 } else {
e1d9fd2e 304 printk(KERN_ERR "Non-boot CPUs are not disabled\n");
e3920fb4 305 }
1d64b9cb 306 suspend_cpu_hotplug = 0;
aa953877 307 mutex_unlock(&cpu_add_remove_lock);
e3920fb4
RW
308 return error;
309}
310
311void enable_nonboot_cpus(void)
312{
313 int cpu, error;
314
315 /* Allow everyone to use the CPU hotplug again */
316 mutex_lock(&cpu_add_remove_lock);
317 cpu_hotplug_disabled = 0;
ed746e3b 318 if (cpus_empty(frozen_cpus))
1d64b9cb 319 goto out;
e3920fb4 320
1d64b9cb 321 suspend_cpu_hotplug = 1;
e3920fb4
RW
322 printk("Enabling non-boot CPUs ...\n");
323 for_each_cpu_mask(cpu, frozen_cpus) {
1d64b9cb 324 error = _cpu_up(cpu);
e3920fb4
RW
325 if (!error) {
326 printk("CPU%d is up\n", cpu);
327 continue;
328 }
1d64b9cb 329 printk(KERN_WARNING "Error taking CPU%d up: %d\n", cpu, error);
e3920fb4
RW
330 }
331 cpus_clear(frozen_cpus);
1d64b9cb
RW
332 suspend_cpu_hotplug = 0;
333out:
334 mutex_unlock(&cpu_add_remove_lock);
1da177e4 335}
e3920fb4 336#endif