xref: /linux-6.15/kernel/cpu.c (revision 87af9e7f)
11da177e4SLinus Torvalds /* CPU control.
21da177e4SLinus Torvalds  * (C) 2001, 2002, 2003, 2004 Rusty Russell
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  * This code is licenced under the GPL.
51da177e4SLinus Torvalds  */
61da177e4SLinus Torvalds #include <linux/proc_fs.h>
71da177e4SLinus Torvalds #include <linux/smp.h>
81da177e4SLinus Torvalds #include <linux/init.h>
91da177e4SLinus Torvalds #include <linux/notifier.h>
101da177e4SLinus Torvalds #include <linux/sched.h>
111da177e4SLinus Torvalds #include <linux/unistd.h>
121da177e4SLinus Torvalds #include <linux/cpu.h>
13cb79295eSAnton Vorontsov #include <linux/oom.h>
14cb79295eSAnton Vorontsov #include <linux/rcupdate.h>
159984de1aSPaul Gortmaker #include <linux/export.h>
16e4cc2f87SAnton Vorontsov #include <linux/bug.h>
171da177e4SLinus Torvalds #include <linux/kthread.h>
181da177e4SLinus Torvalds #include <linux/stop_machine.h>
1981615b62SIngo Molnar #include <linux/mutex.h>
205a0e3ad6STejun Heo #include <linux/gfp.h>
2179cfbdfaSSrivatsa S. Bhat #include <linux/suspend.h>
22a19423b9SGautham R. Shenoy #include <linux/lockdep.h>
23bb3632c6STodd E Brandt #include <trace/events/power.h>
241da177e4SLinus Torvalds 
2538498a67SThomas Gleixner #include "smpboot.h"
2638498a67SThomas Gleixner 
2798a79d6aSRusty Russell #ifdef CONFIG_SMP
28b3199c02SRusty Russell /* Serializes the updates to cpu_online_mask, cpu_present_mask */
29aa953877SLinus Torvalds static DEFINE_MUTEX(cpu_add_remove_lock);
301da177e4SLinus Torvalds 
3179a6cdebSLai Jiangshan /*
3293ae4f97SSrivatsa S. Bhat  * The following two APIs (cpu_maps_update_begin/done) must be used when
3393ae4f97SSrivatsa S. Bhat  * attempting to serialize the updates to cpu_online_mask & cpu_present_mask.
3493ae4f97SSrivatsa S. Bhat  * The APIs cpu_notifier_register_begin/done() must be used to protect CPU
3593ae4f97SSrivatsa S. Bhat  * hotplug callback (un)registration performed using __register_cpu_notifier()
3693ae4f97SSrivatsa S. Bhat  * or __unregister_cpu_notifier().
3779a6cdebSLai Jiangshan  */
3879a6cdebSLai Jiangshan void cpu_maps_update_begin(void)
3979a6cdebSLai Jiangshan {
4079a6cdebSLai Jiangshan 	mutex_lock(&cpu_add_remove_lock);
4179a6cdebSLai Jiangshan }
4293ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(cpu_notifier_register_begin);
4379a6cdebSLai Jiangshan 
4479a6cdebSLai Jiangshan void cpu_maps_update_done(void)
4579a6cdebSLai Jiangshan {
4679a6cdebSLai Jiangshan 	mutex_unlock(&cpu_add_remove_lock);
4779a6cdebSLai Jiangshan }
4893ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(cpu_notifier_register_done);
4979a6cdebSLai Jiangshan 
505c113fbeSDaniel J Blueman static RAW_NOTIFIER_HEAD(cpu_chain);
511da177e4SLinus Torvalds 
52e3920fb4SRafael J. Wysocki /* If set, cpu_up and cpu_down will return -EBUSY and do nothing.
53e3920fb4SRafael J. Wysocki  * Should always be manipulated under cpu_add_remove_lock
54e3920fb4SRafael J. Wysocki  */
55e3920fb4SRafael J. Wysocki static int cpu_hotplug_disabled;
56e3920fb4SRafael J. Wysocki 
5779a6cdebSLai Jiangshan #ifdef CONFIG_HOTPLUG_CPU
5879a6cdebSLai Jiangshan 
59d221938cSGautham R Shenoy static struct {
60d221938cSGautham R Shenoy 	struct task_struct *active_writer;
61*87af9e7fSDavid Hildenbrand 	/* wait queue to wake up the active_writer */
62*87af9e7fSDavid Hildenbrand 	wait_queue_head_t wq;
63*87af9e7fSDavid Hildenbrand 	/* verifies that no writer will get active while readers are active */
64*87af9e7fSDavid Hildenbrand 	struct mutex lock;
65d221938cSGautham R Shenoy 	/*
66d221938cSGautham R Shenoy 	 * Also blocks the new readers during
67d221938cSGautham R Shenoy 	 * an ongoing cpu hotplug operation.
68d221938cSGautham R Shenoy 	 */
69*87af9e7fSDavid Hildenbrand 	atomic_t refcount;
70a19423b9SGautham R. Shenoy 
71a19423b9SGautham R. Shenoy #ifdef CONFIG_DEBUG_LOCK_ALLOC
72a19423b9SGautham R. Shenoy 	struct lockdep_map dep_map;
73a19423b9SGautham R. Shenoy #endif
7431950eb6SLinus Torvalds } cpu_hotplug = {
7531950eb6SLinus Torvalds 	.active_writer = NULL,
76*87af9e7fSDavid Hildenbrand 	.wq = __WAIT_QUEUE_HEAD_INITIALIZER(cpu_hotplug.wq),
7731950eb6SLinus Torvalds 	.lock = __MUTEX_INITIALIZER(cpu_hotplug.lock),
78a19423b9SGautham R. Shenoy #ifdef CONFIG_DEBUG_LOCK_ALLOC
79a19423b9SGautham R. Shenoy 	.dep_map = {.name = "cpu_hotplug.lock" },
80a19423b9SGautham R. Shenoy #endif
8131950eb6SLinus Torvalds };
82d221938cSGautham R Shenoy 
83a19423b9SGautham R. Shenoy /* Lockdep annotations for get/put_online_cpus() and cpu_hotplug_begin/end() */
84a19423b9SGautham R. Shenoy #define cpuhp_lock_acquire_read() lock_map_acquire_read(&cpu_hotplug.dep_map)
85dd56af42SPaul E. McKenney #define cpuhp_lock_acquire_tryread() \
86dd56af42SPaul E. McKenney 				  lock_map_acquire_tryread(&cpu_hotplug.dep_map)
87a19423b9SGautham R. Shenoy #define cpuhp_lock_acquire()      lock_map_acquire(&cpu_hotplug.dep_map)
88a19423b9SGautham R. Shenoy #define cpuhp_lock_release()      lock_map_release(&cpu_hotplug.dep_map)
89a19423b9SGautham R. Shenoy 
9062db99f4SPaul E. McKenney 
9186ef5c9aSGautham R Shenoy void get_online_cpus(void)
92a9d9baa1SAshok Raj {
93d221938cSGautham R Shenoy 	might_sleep();
94d221938cSGautham R Shenoy 	if (cpu_hotplug.active_writer == current)
95aa953877SLinus Torvalds 		return;
96a19423b9SGautham R. Shenoy 	cpuhp_lock_acquire_read();
97d221938cSGautham R Shenoy 	mutex_lock(&cpu_hotplug.lock);
98*87af9e7fSDavid Hildenbrand 	atomic_inc(&cpu_hotplug.refcount);
99d221938cSGautham R Shenoy 	mutex_unlock(&cpu_hotplug.lock);
100a9d9baa1SAshok Raj }
10186ef5c9aSGautham R Shenoy EXPORT_SYMBOL_GPL(get_online_cpus);
102a9d9baa1SAshok Raj 
103dd56af42SPaul E. McKenney bool try_get_online_cpus(void)
104dd56af42SPaul E. McKenney {
105dd56af42SPaul E. McKenney 	if (cpu_hotplug.active_writer == current)
106dd56af42SPaul E. McKenney 		return true;
107dd56af42SPaul E. McKenney 	if (!mutex_trylock(&cpu_hotplug.lock))
108dd56af42SPaul E. McKenney 		return false;
109dd56af42SPaul E. McKenney 	cpuhp_lock_acquire_tryread();
110*87af9e7fSDavid Hildenbrand 	atomic_inc(&cpu_hotplug.refcount);
111dd56af42SPaul E. McKenney 	mutex_unlock(&cpu_hotplug.lock);
112dd56af42SPaul E. McKenney 	return true;
113dd56af42SPaul E. McKenney }
114dd56af42SPaul E. McKenney EXPORT_SYMBOL_GPL(try_get_online_cpus);
115dd56af42SPaul E. McKenney 
11686ef5c9aSGautham R Shenoy void put_online_cpus(void)
117a9d9baa1SAshok Raj {
118*87af9e7fSDavid Hildenbrand 	int refcount;
119*87af9e7fSDavid Hildenbrand 
120d221938cSGautham R Shenoy 	if (cpu_hotplug.active_writer == current)
121aa953877SLinus Torvalds 		return;
122075663d1SSrivatsa S. Bhat 
123*87af9e7fSDavid Hildenbrand 	refcount = atomic_dec_return(&cpu_hotplug.refcount);
124*87af9e7fSDavid Hildenbrand 	if (WARN_ON(refcount < 0)) /* try to fix things up */
125*87af9e7fSDavid Hildenbrand 		atomic_inc(&cpu_hotplug.refcount);
126075663d1SSrivatsa S. Bhat 
127*87af9e7fSDavid Hildenbrand 	if (refcount <= 0 && waitqueue_active(&cpu_hotplug.wq))
128*87af9e7fSDavid Hildenbrand 		wake_up(&cpu_hotplug.wq);
129*87af9e7fSDavid Hildenbrand 
130a19423b9SGautham R. Shenoy 	cpuhp_lock_release();
131d221938cSGautham R Shenoy 
132a9d9baa1SAshok Raj }
13386ef5c9aSGautham R Shenoy EXPORT_SYMBOL_GPL(put_online_cpus);
134a9d9baa1SAshok Raj 
135d221938cSGautham R Shenoy /*
136d221938cSGautham R Shenoy  * This ensures that the hotplug operation can begin only when the
137d221938cSGautham R Shenoy  * refcount goes to zero.
138d221938cSGautham R Shenoy  *
139d221938cSGautham R Shenoy  * Note that during a cpu-hotplug operation, the new readers, if any,
140d221938cSGautham R Shenoy  * will be blocked by the cpu_hotplug.lock
141d221938cSGautham R Shenoy  *
142d2ba7e2aSOleg Nesterov  * Since cpu_hotplug_begin() is always called after invoking
143d2ba7e2aSOleg Nesterov  * cpu_maps_update_begin(), we can be sure that only one writer is active.
144d221938cSGautham R Shenoy  *
145d221938cSGautham R Shenoy  * Note that theoretically, there is a possibility of a livelock:
146d221938cSGautham R Shenoy  * - Refcount goes to zero, last reader wakes up the sleeping
147d221938cSGautham R Shenoy  *   writer.
148d221938cSGautham R Shenoy  * - Last reader unlocks the cpu_hotplug.lock.
149d221938cSGautham R Shenoy  * - A new reader arrives at this moment, bumps up the refcount.
150d221938cSGautham R Shenoy  * - The writer acquires the cpu_hotplug.lock finds the refcount
151d221938cSGautham R Shenoy  *   non zero and goes to sleep again.
152d221938cSGautham R Shenoy  *
153d221938cSGautham R Shenoy  * However, this is very difficult to achieve in practice since
15486ef5c9aSGautham R Shenoy  * get_online_cpus() not an api which is called all that often.
155d221938cSGautham R Shenoy  *
156d221938cSGautham R Shenoy  */
157b9d10be7SToshi Kani void cpu_hotplug_begin(void)
158d221938cSGautham R Shenoy {
159*87af9e7fSDavid Hildenbrand 	DEFINE_WAIT(wait);
160d2ba7e2aSOleg Nesterov 
161*87af9e7fSDavid Hildenbrand 	cpu_hotplug.active_writer = current;
162a19423b9SGautham R. Shenoy 	cpuhp_lock_acquire();
163*87af9e7fSDavid Hildenbrand 
164d2ba7e2aSOleg Nesterov 	for (;;) {
165d2ba7e2aSOleg Nesterov 		mutex_lock(&cpu_hotplug.lock);
166*87af9e7fSDavid Hildenbrand 		prepare_to_wait(&cpu_hotplug.wq, &wait, TASK_UNINTERRUPTIBLE);
167*87af9e7fSDavid Hildenbrand 		if (likely(!atomic_read(&cpu_hotplug.refcount)))
168d2ba7e2aSOleg Nesterov 				break;
169d221938cSGautham R Shenoy 		mutex_unlock(&cpu_hotplug.lock);
170d221938cSGautham R Shenoy 		schedule();
171d221938cSGautham R Shenoy 	}
172*87af9e7fSDavid Hildenbrand 	finish_wait(&cpu_hotplug.wq, &wait);
173d221938cSGautham R Shenoy }
174d221938cSGautham R Shenoy 
175b9d10be7SToshi Kani void cpu_hotplug_done(void)
176d221938cSGautham R Shenoy {
177d221938cSGautham R Shenoy 	cpu_hotplug.active_writer = NULL;
178d221938cSGautham R Shenoy 	mutex_unlock(&cpu_hotplug.lock);
179a19423b9SGautham R. Shenoy 	cpuhp_lock_release();
180d221938cSGautham R Shenoy }
18179a6cdebSLai Jiangshan 
18216e53dbfSSrivatsa S. Bhat /*
18316e53dbfSSrivatsa S. Bhat  * Wait for currently running CPU hotplug operations to complete (if any) and
18416e53dbfSSrivatsa S. Bhat  * disable future CPU hotplug (from sysfs). The 'cpu_add_remove_lock' protects
18516e53dbfSSrivatsa S. Bhat  * the 'cpu_hotplug_disabled' flag. The same lock is also acquired by the
18616e53dbfSSrivatsa S. Bhat  * hotplug path before performing hotplug operations. So acquiring that lock
18716e53dbfSSrivatsa S. Bhat  * guarantees mutual exclusion from any currently running hotplug operations.
18816e53dbfSSrivatsa S. Bhat  */
18916e53dbfSSrivatsa S. Bhat void cpu_hotplug_disable(void)
19016e53dbfSSrivatsa S. Bhat {
19116e53dbfSSrivatsa S. Bhat 	cpu_maps_update_begin();
19216e53dbfSSrivatsa S. Bhat 	cpu_hotplug_disabled = 1;
19316e53dbfSSrivatsa S. Bhat 	cpu_maps_update_done();
19416e53dbfSSrivatsa S. Bhat }
19516e53dbfSSrivatsa S. Bhat 
19616e53dbfSSrivatsa S. Bhat void cpu_hotplug_enable(void)
19716e53dbfSSrivatsa S. Bhat {
19816e53dbfSSrivatsa S. Bhat 	cpu_maps_update_begin();
19916e53dbfSSrivatsa S. Bhat 	cpu_hotplug_disabled = 0;
20016e53dbfSSrivatsa S. Bhat 	cpu_maps_update_done();
20116e53dbfSSrivatsa S. Bhat }
20216e53dbfSSrivatsa S. Bhat 
203b9d10be7SToshi Kani #endif	/* CONFIG_HOTPLUG_CPU */
20479a6cdebSLai Jiangshan 
2051da177e4SLinus Torvalds /* Need to know about CPUs going up/down? */
206f7b16c10SSam Ravnborg int __ref register_cpu_notifier(struct notifier_block *nb)
2071da177e4SLinus Torvalds {
208bd5349cfSNeil Brown 	int ret;
209d221938cSGautham R Shenoy 	cpu_maps_update_begin();
210bd5349cfSNeil Brown 	ret = raw_notifier_chain_register(&cpu_chain, nb);
211d221938cSGautham R Shenoy 	cpu_maps_update_done();
212bd5349cfSNeil Brown 	return ret;
2131da177e4SLinus Torvalds }
21465edc68cSChandra Seetharaman 
21593ae4f97SSrivatsa S. Bhat int __ref __register_cpu_notifier(struct notifier_block *nb)
21693ae4f97SSrivatsa S. Bhat {
21793ae4f97SSrivatsa S. Bhat 	return raw_notifier_chain_register(&cpu_chain, nb);
21893ae4f97SSrivatsa S. Bhat }
21993ae4f97SSrivatsa S. Bhat 
220e9fb7631SAkinobu Mita static int __cpu_notify(unsigned long val, void *v, int nr_to_call,
221e9fb7631SAkinobu Mita 			int *nr_calls)
222e9fb7631SAkinobu Mita {
223e6bde73bSAkinobu Mita 	int ret;
224e6bde73bSAkinobu Mita 
225e6bde73bSAkinobu Mita 	ret = __raw_notifier_call_chain(&cpu_chain, val, v, nr_to_call,
226e9fb7631SAkinobu Mita 					nr_calls);
227e6bde73bSAkinobu Mita 
228e6bde73bSAkinobu Mita 	return notifier_to_errno(ret);
229e9fb7631SAkinobu Mita }
230e9fb7631SAkinobu Mita 
231e9fb7631SAkinobu Mita static int cpu_notify(unsigned long val, void *v)
232e9fb7631SAkinobu Mita {
233e9fb7631SAkinobu Mita 	return __cpu_notify(val, v, -1, NULL);
234e9fb7631SAkinobu Mita }
235e9fb7631SAkinobu Mita 
23600b9b0afSLinus Torvalds #ifdef CONFIG_HOTPLUG_CPU
23700b9b0afSLinus Torvalds 
238e9fb7631SAkinobu Mita static void cpu_notify_nofail(unsigned long val, void *v)
239e9fb7631SAkinobu Mita {
24000b9b0afSLinus Torvalds 	BUG_ON(cpu_notify(val, v));
241e9fb7631SAkinobu Mita }
2421da177e4SLinus Torvalds EXPORT_SYMBOL(register_cpu_notifier);
24393ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(__register_cpu_notifier);
2441da177e4SLinus Torvalds 
2459647155fSSam Ravnborg void __ref unregister_cpu_notifier(struct notifier_block *nb)
2461da177e4SLinus Torvalds {
247d221938cSGautham R Shenoy 	cpu_maps_update_begin();
248bd5349cfSNeil Brown 	raw_notifier_chain_unregister(&cpu_chain, nb);
249d221938cSGautham R Shenoy 	cpu_maps_update_done();
2501da177e4SLinus Torvalds }
2511da177e4SLinus Torvalds EXPORT_SYMBOL(unregister_cpu_notifier);
2521da177e4SLinus Torvalds 
25393ae4f97SSrivatsa S. Bhat void __ref __unregister_cpu_notifier(struct notifier_block *nb)
25493ae4f97SSrivatsa S. Bhat {
25593ae4f97SSrivatsa S. Bhat 	raw_notifier_chain_unregister(&cpu_chain, nb);
25693ae4f97SSrivatsa S. Bhat }
25793ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(__unregister_cpu_notifier);
25893ae4f97SSrivatsa S. Bhat 
259e4cc2f87SAnton Vorontsov /**
260e4cc2f87SAnton Vorontsov  * clear_tasks_mm_cpumask - Safely clear tasks' mm_cpumask for a CPU
261e4cc2f87SAnton Vorontsov  * @cpu: a CPU id
262e4cc2f87SAnton Vorontsov  *
263e4cc2f87SAnton Vorontsov  * This function walks all processes, finds a valid mm struct for each one and
264e4cc2f87SAnton Vorontsov  * then clears a corresponding bit in mm's cpumask.  While this all sounds
265e4cc2f87SAnton Vorontsov  * trivial, there are various non-obvious corner cases, which this function
266e4cc2f87SAnton Vorontsov  * tries to solve in a safe manner.
267e4cc2f87SAnton Vorontsov  *
268e4cc2f87SAnton Vorontsov  * Also note that the function uses a somewhat relaxed locking scheme, so it may
269e4cc2f87SAnton Vorontsov  * be called only for an already offlined CPU.
270e4cc2f87SAnton Vorontsov  */
271cb79295eSAnton Vorontsov void clear_tasks_mm_cpumask(int cpu)
272cb79295eSAnton Vorontsov {
273cb79295eSAnton Vorontsov 	struct task_struct *p;
274cb79295eSAnton Vorontsov 
275cb79295eSAnton Vorontsov 	/*
276cb79295eSAnton Vorontsov 	 * This function is called after the cpu is taken down and marked
277cb79295eSAnton Vorontsov 	 * offline, so its not like new tasks will ever get this cpu set in
278cb79295eSAnton Vorontsov 	 * their mm mask. -- Peter Zijlstra
279cb79295eSAnton Vorontsov 	 * Thus, we may use rcu_read_lock() here, instead of grabbing
280cb79295eSAnton Vorontsov 	 * full-fledged tasklist_lock.
281cb79295eSAnton Vorontsov 	 */
282e4cc2f87SAnton Vorontsov 	WARN_ON(cpu_online(cpu));
283cb79295eSAnton Vorontsov 	rcu_read_lock();
284cb79295eSAnton Vorontsov 	for_each_process(p) {
285cb79295eSAnton Vorontsov 		struct task_struct *t;
286cb79295eSAnton Vorontsov 
287e4cc2f87SAnton Vorontsov 		/*
288e4cc2f87SAnton Vorontsov 		 * Main thread might exit, but other threads may still have
289e4cc2f87SAnton Vorontsov 		 * a valid mm. Find one.
290e4cc2f87SAnton Vorontsov 		 */
291cb79295eSAnton Vorontsov 		t = find_lock_task_mm(p);
292cb79295eSAnton Vorontsov 		if (!t)
293cb79295eSAnton Vorontsov 			continue;
294cb79295eSAnton Vorontsov 		cpumask_clear_cpu(cpu, mm_cpumask(t->mm));
295cb79295eSAnton Vorontsov 		task_unlock(t);
296cb79295eSAnton Vorontsov 	}
297cb79295eSAnton Vorontsov 	rcu_read_unlock();
298cb79295eSAnton Vorontsov }
299cb79295eSAnton Vorontsov 
300b728ca06SKirill Tkhai static inline void check_for_tasks(int dead_cpu)
3011da177e4SLinus Torvalds {
302b728ca06SKirill Tkhai 	struct task_struct *g, *p;
3031da177e4SLinus Torvalds 
304b728ca06SKirill Tkhai 	read_lock_irq(&tasklist_lock);
305b728ca06SKirill Tkhai 	do_each_thread(g, p) {
306b728ca06SKirill Tkhai 		if (!p->on_rq)
307b728ca06SKirill Tkhai 			continue;
308b728ca06SKirill Tkhai 		/*
309b728ca06SKirill Tkhai 		 * We do the check with unlocked task_rq(p)->lock.
310b728ca06SKirill Tkhai 		 * Order the reading to do not warn about a task,
311b728ca06SKirill Tkhai 		 * which was running on this cpu in the past, and
312b728ca06SKirill Tkhai 		 * it's just been woken on another cpu.
313b728ca06SKirill Tkhai 		 */
314b728ca06SKirill Tkhai 		rmb();
315b728ca06SKirill Tkhai 		if (task_cpu(p) != dead_cpu)
316b728ca06SKirill Tkhai 			continue;
317b728ca06SKirill Tkhai 
31884117da5SFabian Frederick 		pr_warn("Task %s (pid=%d) is on cpu %d (state=%ld, flags=%x)\n",
319b728ca06SKirill Tkhai 			p->comm, task_pid_nr(p), dead_cpu, p->state, p->flags);
320b728ca06SKirill Tkhai 	} while_each_thread(g, p);
321b728ca06SKirill Tkhai 	read_unlock_irq(&tasklist_lock);
3221da177e4SLinus Torvalds }
3231da177e4SLinus Torvalds 
324db912f96SAvi Kivity struct take_cpu_down_param {
325db912f96SAvi Kivity 	unsigned long mod;
326db912f96SAvi Kivity 	void *hcpu;
327db912f96SAvi Kivity };
328db912f96SAvi Kivity 
3291da177e4SLinus Torvalds /* Take this CPU down. */
330514a20a5SSam Ravnborg static int __ref take_cpu_down(void *_param)
3311da177e4SLinus Torvalds {
332db912f96SAvi Kivity 	struct take_cpu_down_param *param = _param;
3331da177e4SLinus Torvalds 	int err;
3341da177e4SLinus Torvalds 
3351da177e4SLinus Torvalds 	/* Ensure this CPU doesn't handle any more interrupts. */
3361da177e4SLinus Torvalds 	err = __cpu_disable();
3371da177e4SLinus Torvalds 	if (err < 0)
338f3705136SZwane Mwaikambo 		return err;
339f3705136SZwane Mwaikambo 
340e9fb7631SAkinobu Mita 	cpu_notify(CPU_DYING | param->mod, param->hcpu);
34114e568e7SThomas Gleixner 	/* Park the stopper thread */
34214e568e7SThomas Gleixner 	kthread_park(current);
343f3705136SZwane Mwaikambo 	return 0;
3441da177e4SLinus Torvalds }
3451da177e4SLinus Torvalds 
346e3920fb4SRafael J. Wysocki /* Requires cpu_add_remove_lock to be held */
347514a20a5SSam Ravnborg static int __ref _cpu_down(unsigned int cpu, int tasks_frozen)
3481da177e4SLinus Torvalds {
349e7407dccSHeiko Carstens 	int err, nr_calls = 0;
350e7407dccSHeiko Carstens 	void *hcpu = (void *)(long)cpu;
3518bb78442SRafael J. Wysocki 	unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
352db912f96SAvi Kivity 	struct take_cpu_down_param tcd_param = {
353db912f96SAvi Kivity 		.mod = mod,
354db912f96SAvi Kivity 		.hcpu = hcpu,
355db912f96SAvi Kivity 	};
3561da177e4SLinus Torvalds 
357e3920fb4SRafael J. Wysocki 	if (num_online_cpus() == 1)
358e3920fb4SRafael J. Wysocki 		return -EBUSY;
3591da177e4SLinus Torvalds 
360e3920fb4SRafael J. Wysocki 	if (!cpu_online(cpu))
361e3920fb4SRafael J. Wysocki 		return -EINVAL;
3621da177e4SLinus Torvalds 
363d221938cSGautham R Shenoy 	cpu_hotplug_begin();
3644d51985eSMichael Rodriguez 
365e9fb7631SAkinobu Mita 	err = __cpu_notify(CPU_DOWN_PREPARE | mod, hcpu, -1, &nr_calls);
366e6bde73bSAkinobu Mita 	if (err) {
367a0d8cdb6SAkinobu Mita 		nr_calls--;
368e9fb7631SAkinobu Mita 		__cpu_notify(CPU_DOWN_FAILED | mod, hcpu, nr_calls, NULL);
36984117da5SFabian Frederick 		pr_warn("%s: attempt to take down CPU %u failed\n",
370af1f16d0SHarvey Harrison 			__func__, cpu);
371baaca49fSGautham R Shenoy 		goto out_release;
3721da177e4SLinus Torvalds 	}
3731da177e4SLinus Torvalds 
3746acce3efSPeter Zijlstra 	/*
3756acce3efSPeter Zijlstra 	 * By now we've cleared cpu_active_mask, wait for all preempt-disabled
3766acce3efSPeter Zijlstra 	 * and RCU users of this state to go away such that all new such users
3776acce3efSPeter Zijlstra 	 * will observe it.
3786acce3efSPeter Zijlstra 	 *
3796acce3efSPeter Zijlstra 	 * For CONFIG_PREEMPT we have preemptible RCU and its sync_rcu() might
3806acce3efSPeter Zijlstra 	 * not imply sync_sched(), so explicitly call both.
381106dd5afSMichael wang 	 *
382106dd5afSMichael wang 	 * Do sync before park smpboot threads to take care the rcu boost case.
3836acce3efSPeter Zijlstra 	 */
3846acce3efSPeter Zijlstra #ifdef CONFIG_PREEMPT
3856acce3efSPeter Zijlstra 	synchronize_sched();
3866acce3efSPeter Zijlstra #endif
3876acce3efSPeter Zijlstra 	synchronize_rcu();
3886acce3efSPeter Zijlstra 
389106dd5afSMichael wang 	smpboot_park_threads(cpu);
390106dd5afSMichael wang 
3916acce3efSPeter Zijlstra 	/*
3926acce3efSPeter Zijlstra 	 * So now all preempt/rcu users must observe !cpu_active().
3936acce3efSPeter Zijlstra 	 */
3946acce3efSPeter Zijlstra 
395e0b582ecSRusty Russell 	err = __stop_machine(take_cpu_down, &tcd_param, cpumask_of(cpu));
39604321587SRusty Russell 	if (err) {
3971da177e4SLinus Torvalds 		/* CPU didn't die: tell everyone.  Can't complain. */
398f97f8f06SThomas Gleixner 		smpboot_unpark_threads(cpu);
399e9fb7631SAkinobu Mita 		cpu_notify_nofail(CPU_DOWN_FAILED | mod, hcpu);
4006a1bdc1bSOleg Nesterov 		goto out_release;
4011da177e4SLinus Torvalds 	}
40204321587SRusty Russell 	BUG_ON(cpu_online(cpu));
4031da177e4SLinus Torvalds 
40448c5ccaeSPeter Zijlstra 	/*
40548c5ccaeSPeter Zijlstra 	 * The migration_call() CPU_DYING callback will have removed all
40648c5ccaeSPeter Zijlstra 	 * runnable tasks from the cpu, there's only the idle task left now
40748c5ccaeSPeter Zijlstra 	 * that the migration thread is done doing the stop_machine thing.
40851a96c77SPeter Zijlstra 	 *
40951a96c77SPeter Zijlstra 	 * Wait for the stop thread to go away.
41048c5ccaeSPeter Zijlstra 	 */
41151a96c77SPeter Zijlstra 	while (!idle_cpu(cpu))
41251a96c77SPeter Zijlstra 		cpu_relax();
4131da177e4SLinus Torvalds 
4141da177e4SLinus Torvalds 	/* This actually kills the CPU. */
4151da177e4SLinus Torvalds 	__cpu_die(cpu);
4161da177e4SLinus Torvalds 
4171da177e4SLinus Torvalds 	/* CPU is completely dead: tell everyone.  Too late to complain. */
418e9fb7631SAkinobu Mita 	cpu_notify_nofail(CPU_DEAD | mod, hcpu);
4191da177e4SLinus Torvalds 
4201da177e4SLinus Torvalds 	check_for_tasks(cpu);
4211da177e4SLinus Torvalds 
422baaca49fSGautham R Shenoy out_release:
423d221938cSGautham R Shenoy 	cpu_hotplug_done();
424e9fb7631SAkinobu Mita 	if (!err)
425e9fb7631SAkinobu Mita 		cpu_notify_nofail(CPU_POST_DEAD | mod, hcpu);
426e3920fb4SRafael J. Wysocki 	return err;
427e3920fb4SRafael J. Wysocki }
428e3920fb4SRafael J. Wysocki 
429514a20a5SSam Ravnborg int __ref cpu_down(unsigned int cpu)
430e3920fb4SRafael J. Wysocki {
4319ea09af3SHeiko Carstens 	int err;
432e3920fb4SRafael J. Wysocki 
433d221938cSGautham R Shenoy 	cpu_maps_update_begin();
434e761b772SMax Krasnyansky 
435e761b772SMax Krasnyansky 	if (cpu_hotplug_disabled) {
436e3920fb4SRafael J. Wysocki 		err = -EBUSY;
437e761b772SMax Krasnyansky 		goto out;
438e761b772SMax Krasnyansky 	}
439e761b772SMax Krasnyansky 
4408bb78442SRafael J. Wysocki 	err = _cpu_down(cpu, 0);
441e3920fb4SRafael J. Wysocki 
442e761b772SMax Krasnyansky out:
443d221938cSGautham R Shenoy 	cpu_maps_update_done();
4441da177e4SLinus Torvalds 	return err;
4451da177e4SLinus Torvalds }
446b62b8ef9SZhang Rui EXPORT_SYMBOL(cpu_down);
4471da177e4SLinus Torvalds #endif /*CONFIG_HOTPLUG_CPU*/
4481da177e4SLinus Torvalds 
449e3920fb4SRafael J. Wysocki /* Requires cpu_add_remove_lock to be held */
4500db0628dSPaul Gortmaker static int _cpu_up(unsigned int cpu, int tasks_frozen)
4511da177e4SLinus Torvalds {
452baaca49fSGautham R Shenoy 	int ret, nr_calls = 0;
4531da177e4SLinus Torvalds 	void *hcpu = (void *)(long)cpu;
4548bb78442SRafael J. Wysocki 	unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
4553bb5d2eeSSuresh Siddha 	struct task_struct *idle;
4561da177e4SLinus Torvalds 
457d221938cSGautham R Shenoy 	cpu_hotplug_begin();
45838498a67SThomas Gleixner 
4595e5041f3SYasuaki Ishimatsu 	if (cpu_online(cpu) || !cpu_present(cpu)) {
4605e5041f3SYasuaki Ishimatsu 		ret = -EINVAL;
4615e5041f3SYasuaki Ishimatsu 		goto out;
4625e5041f3SYasuaki Ishimatsu 	}
4635e5041f3SYasuaki Ishimatsu 
4643bb5d2eeSSuresh Siddha 	idle = idle_thread_get(cpu);
4653bb5d2eeSSuresh Siddha 	if (IS_ERR(idle)) {
4663bb5d2eeSSuresh Siddha 		ret = PTR_ERR(idle);
46738498a67SThomas Gleixner 		goto out;
4683bb5d2eeSSuresh Siddha 	}
46938498a67SThomas Gleixner 
470f97f8f06SThomas Gleixner 	ret = smpboot_create_threads(cpu);
471f97f8f06SThomas Gleixner 	if (ret)
472f97f8f06SThomas Gleixner 		goto out;
473f97f8f06SThomas Gleixner 
474e9fb7631SAkinobu Mita 	ret = __cpu_notify(CPU_UP_PREPARE | mod, hcpu, -1, &nr_calls);
475e6bde73bSAkinobu Mita 	if (ret) {
476a0d8cdb6SAkinobu Mita 		nr_calls--;
47784117da5SFabian Frederick 		pr_warn("%s: attempt to bring up CPU %u failed\n",
478af1f16d0SHarvey Harrison 			__func__, cpu);
4791da177e4SLinus Torvalds 		goto out_notify;
4801da177e4SLinus Torvalds 	}
4811da177e4SLinus Torvalds 
4821da177e4SLinus Torvalds 	/* Arch-specific enabling code. */
4833bb5d2eeSSuresh Siddha 	ret = __cpu_up(cpu, idle);
4841da177e4SLinus Torvalds 	if (ret != 0)
4851da177e4SLinus Torvalds 		goto out_notify;
4866978c705SEric Sesterhenn 	BUG_ON(!cpu_online(cpu));
4871da177e4SLinus Torvalds 
488f97f8f06SThomas Gleixner 	/* Wake the per cpu threads */
489f97f8f06SThomas Gleixner 	smpboot_unpark_threads(cpu);
490f97f8f06SThomas Gleixner 
4911da177e4SLinus Torvalds 	/* Now call notifier in preparation. */
492e9fb7631SAkinobu Mita 	cpu_notify(CPU_ONLINE | mod, hcpu);
4931da177e4SLinus Torvalds 
4941da177e4SLinus Torvalds out_notify:
4951da177e4SLinus Torvalds 	if (ret != 0)
496e9fb7631SAkinobu Mita 		__cpu_notify(CPU_UP_CANCELED | mod, hcpu, nr_calls, NULL);
49738498a67SThomas Gleixner out:
498d221938cSGautham R Shenoy 	cpu_hotplug_done();
499e3920fb4SRafael J. Wysocki 
5001da177e4SLinus Torvalds 	return ret;
5011da177e4SLinus Torvalds }
502e3920fb4SRafael J. Wysocki 
5030db0628dSPaul Gortmaker int cpu_up(unsigned int cpu)
504e3920fb4SRafael J. Wysocki {
505e3920fb4SRafael J. Wysocki 	int err = 0;
506cf23422bSminskey guo 
507e0b582ecSRusty Russell 	if (!cpu_possible(cpu)) {
50884117da5SFabian Frederick 		pr_err("can't online cpu %d because it is not configured as may-hotadd at boot time\n",
50984117da5SFabian Frederick 		       cpu);
51087d5e023SChen Gong #if defined(CONFIG_IA64)
51184117da5SFabian Frederick 		pr_err("please check additional_cpus= boot parameter\n");
51273e753a5SKAMEZAWA Hiroyuki #endif
51373e753a5SKAMEZAWA Hiroyuki 		return -EINVAL;
51473e753a5SKAMEZAWA Hiroyuki 	}
515e3920fb4SRafael J. Wysocki 
51601b0f197SToshi Kani 	err = try_online_node(cpu_to_node(cpu));
517cf23422bSminskey guo 	if (err)
518cf23422bSminskey guo 		return err;
519cf23422bSminskey guo 
520d221938cSGautham R Shenoy 	cpu_maps_update_begin();
521e761b772SMax Krasnyansky 
522e761b772SMax Krasnyansky 	if (cpu_hotplug_disabled) {
523e3920fb4SRafael J. Wysocki 		err = -EBUSY;
524e761b772SMax Krasnyansky 		goto out;
525e761b772SMax Krasnyansky 	}
526e761b772SMax Krasnyansky 
5278bb78442SRafael J. Wysocki 	err = _cpu_up(cpu, 0);
528e3920fb4SRafael J. Wysocki 
529e761b772SMax Krasnyansky out:
530d221938cSGautham R Shenoy 	cpu_maps_update_done();
531e3920fb4SRafael J. Wysocki 	return err;
532e3920fb4SRafael J. Wysocki }
533a513f6baSPaul E. McKenney EXPORT_SYMBOL_GPL(cpu_up);
534e3920fb4SRafael J. Wysocki 
535f3de4be9SRafael J. Wysocki #ifdef CONFIG_PM_SLEEP_SMP
536e0b582ecSRusty Russell static cpumask_var_t frozen_cpus;
537e3920fb4SRafael J. Wysocki 
538e3920fb4SRafael J. Wysocki int disable_nonboot_cpus(void)
539e3920fb4SRafael J. Wysocki {
540e9a5f426SRafael J. Wysocki 	int cpu, first_cpu, error = 0;
541e3920fb4SRafael J. Wysocki 
542d221938cSGautham R Shenoy 	cpu_maps_update_begin();
543e0b582ecSRusty Russell 	first_cpu = cpumask_first(cpu_online_mask);
5449ee349adSXiaotian Feng 	/*
5459ee349adSXiaotian Feng 	 * We take down all of the non-boot CPUs in one shot to avoid races
546e3920fb4SRafael J. Wysocki 	 * with the userspace trying to use the CPU hotplug at the same time
547e3920fb4SRafael J. Wysocki 	 */
548e0b582ecSRusty Russell 	cpumask_clear(frozen_cpus);
5496ad4c188SPeter Zijlstra 
55084117da5SFabian Frederick 	pr_info("Disabling non-boot CPUs ...\n");
551e3920fb4SRafael J. Wysocki 	for_each_online_cpu(cpu) {
552e3920fb4SRafael J. Wysocki 		if (cpu == first_cpu)
553e3920fb4SRafael J. Wysocki 			continue;
554bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_OFF"), cpu, true);
5558bb78442SRafael J. Wysocki 		error = _cpu_down(cpu, 1);
556bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_OFF"), cpu, false);
557feae3203SMike Travis 		if (!error)
558e0b582ecSRusty Russell 			cpumask_set_cpu(cpu, frozen_cpus);
559feae3203SMike Travis 		else {
56084117da5SFabian Frederick 			pr_err("Error taking CPU%d down: %d\n", cpu, error);
561e3920fb4SRafael J. Wysocki 			break;
562e3920fb4SRafael J. Wysocki 		}
563e3920fb4SRafael J. Wysocki 	}
56486886e55SJoseph Cihula 
565e3920fb4SRafael J. Wysocki 	if (!error) {
566e3920fb4SRafael J. Wysocki 		BUG_ON(num_online_cpus() > 1);
567e3920fb4SRafael J. Wysocki 		/* Make sure the CPUs won't be enabled by someone else */
568e3920fb4SRafael J. Wysocki 		cpu_hotplug_disabled = 1;
569e3920fb4SRafael J. Wysocki 	} else {
57084117da5SFabian Frederick 		pr_err("Non-boot CPUs are not disabled\n");
571e3920fb4SRafael J. Wysocki 	}
572d221938cSGautham R Shenoy 	cpu_maps_update_done();
573e3920fb4SRafael J. Wysocki 	return error;
574e3920fb4SRafael J. Wysocki }
575e3920fb4SRafael J. Wysocki 
576d0af9eedSSuresh Siddha void __weak arch_enable_nonboot_cpus_begin(void)
577d0af9eedSSuresh Siddha {
578d0af9eedSSuresh Siddha }
579d0af9eedSSuresh Siddha 
580d0af9eedSSuresh Siddha void __weak arch_enable_nonboot_cpus_end(void)
581d0af9eedSSuresh Siddha {
582d0af9eedSSuresh Siddha }
583d0af9eedSSuresh Siddha 
584fa7303e2SSam Ravnborg void __ref enable_nonboot_cpus(void)
585e3920fb4SRafael J. Wysocki {
586e3920fb4SRafael J. Wysocki 	int cpu, error;
587e3920fb4SRafael J. Wysocki 
588e3920fb4SRafael J. Wysocki 	/* Allow everyone to use the CPU hotplug again */
589d221938cSGautham R Shenoy 	cpu_maps_update_begin();
590e3920fb4SRafael J. Wysocki 	cpu_hotplug_disabled = 0;
591e0b582ecSRusty Russell 	if (cpumask_empty(frozen_cpus))
5921d64b9cbSRafael J. Wysocki 		goto out;
593e3920fb4SRafael J. Wysocki 
59484117da5SFabian Frederick 	pr_info("Enabling non-boot CPUs ...\n");
595d0af9eedSSuresh Siddha 
596d0af9eedSSuresh Siddha 	arch_enable_nonboot_cpus_begin();
597d0af9eedSSuresh Siddha 
598e0b582ecSRusty Russell 	for_each_cpu(cpu, frozen_cpus) {
599bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_ON"), cpu, true);
6008bb78442SRafael J. Wysocki 		error = _cpu_up(cpu, 1);
601bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_ON"), cpu, false);
602e3920fb4SRafael J. Wysocki 		if (!error) {
60384117da5SFabian Frederick 			pr_info("CPU%d is up\n", cpu);
604e3920fb4SRafael J. Wysocki 			continue;
605e3920fb4SRafael J. Wysocki 		}
60684117da5SFabian Frederick 		pr_warn("Error taking CPU%d up: %d\n", cpu, error);
607e3920fb4SRafael J. Wysocki 	}
608d0af9eedSSuresh Siddha 
609d0af9eedSSuresh Siddha 	arch_enable_nonboot_cpus_end();
610d0af9eedSSuresh Siddha 
611e0b582ecSRusty Russell 	cpumask_clear(frozen_cpus);
6121d64b9cbSRafael J. Wysocki out:
613d221938cSGautham R Shenoy 	cpu_maps_update_done();
614e3920fb4SRafael J. Wysocki }
615e0b582ecSRusty Russell 
616d7268a31SFenghua Yu static int __init alloc_frozen_cpus(void)
617e0b582ecSRusty Russell {
618e0b582ecSRusty Russell 	if (!alloc_cpumask_var(&frozen_cpus, GFP_KERNEL|__GFP_ZERO))
619e0b582ecSRusty Russell 		return -ENOMEM;
620e0b582ecSRusty Russell 	return 0;
621e0b582ecSRusty Russell }
622e0b582ecSRusty Russell core_initcall(alloc_frozen_cpus);
62379cfbdfaSSrivatsa S. Bhat 
62479cfbdfaSSrivatsa S. Bhat /*
62579cfbdfaSSrivatsa S. Bhat  * When callbacks for CPU hotplug notifications are being executed, we must
62679cfbdfaSSrivatsa S. Bhat  * ensure that the state of the system with respect to the tasks being frozen
62779cfbdfaSSrivatsa S. Bhat  * or not, as reported by the notification, remains unchanged *throughout the
62879cfbdfaSSrivatsa S. Bhat  * duration* of the execution of the callbacks.
62979cfbdfaSSrivatsa S. Bhat  * Hence we need to prevent the freezer from racing with regular CPU hotplug.
63079cfbdfaSSrivatsa S. Bhat  *
63179cfbdfaSSrivatsa S. Bhat  * This synchronization is implemented by mutually excluding regular CPU
63279cfbdfaSSrivatsa S. Bhat  * hotplug and Suspend/Hibernate call paths by hooking onto the Suspend/
63379cfbdfaSSrivatsa S. Bhat  * Hibernate notifications.
63479cfbdfaSSrivatsa S. Bhat  */
63579cfbdfaSSrivatsa S. Bhat static int
63679cfbdfaSSrivatsa S. Bhat cpu_hotplug_pm_callback(struct notifier_block *nb,
63779cfbdfaSSrivatsa S. Bhat 			unsigned long action, void *ptr)
63879cfbdfaSSrivatsa S. Bhat {
63979cfbdfaSSrivatsa S. Bhat 	switch (action) {
64079cfbdfaSSrivatsa S. Bhat 
64179cfbdfaSSrivatsa S. Bhat 	case PM_SUSPEND_PREPARE:
64279cfbdfaSSrivatsa S. Bhat 	case PM_HIBERNATION_PREPARE:
64316e53dbfSSrivatsa S. Bhat 		cpu_hotplug_disable();
64479cfbdfaSSrivatsa S. Bhat 		break;
64579cfbdfaSSrivatsa S. Bhat 
64679cfbdfaSSrivatsa S. Bhat 	case PM_POST_SUSPEND:
64779cfbdfaSSrivatsa S. Bhat 	case PM_POST_HIBERNATION:
64816e53dbfSSrivatsa S. Bhat 		cpu_hotplug_enable();
64979cfbdfaSSrivatsa S. Bhat 		break;
65079cfbdfaSSrivatsa S. Bhat 
65179cfbdfaSSrivatsa S. Bhat 	default:
65279cfbdfaSSrivatsa S. Bhat 		return NOTIFY_DONE;
65379cfbdfaSSrivatsa S. Bhat 	}
65479cfbdfaSSrivatsa S. Bhat 
65579cfbdfaSSrivatsa S. Bhat 	return NOTIFY_OK;
65679cfbdfaSSrivatsa S. Bhat }
65779cfbdfaSSrivatsa S. Bhat 
65879cfbdfaSSrivatsa S. Bhat 
659d7268a31SFenghua Yu static int __init cpu_hotplug_pm_sync_init(void)
66079cfbdfaSSrivatsa S. Bhat {
6616e32d479SFenghua Yu 	/*
6626e32d479SFenghua Yu 	 * cpu_hotplug_pm_callback has higher priority than x86
6636e32d479SFenghua Yu 	 * bsp_pm_callback which depends on cpu_hotplug_pm_callback
6646e32d479SFenghua Yu 	 * to disable cpu hotplug to avoid cpu hotplug race.
6656e32d479SFenghua Yu 	 */
66679cfbdfaSSrivatsa S. Bhat 	pm_notifier(cpu_hotplug_pm_callback, 0);
66779cfbdfaSSrivatsa S. Bhat 	return 0;
66879cfbdfaSSrivatsa S. Bhat }
66979cfbdfaSSrivatsa S. Bhat core_initcall(cpu_hotplug_pm_sync_init);
67079cfbdfaSSrivatsa S. Bhat 
671f3de4be9SRafael J. Wysocki #endif /* CONFIG_PM_SLEEP_SMP */
67268f4f1ecSMax Krasnyansky 
673e545a614SManfred Spraul /**
674e545a614SManfred Spraul  * notify_cpu_starting(cpu) - call the CPU_STARTING notifiers
675e545a614SManfred Spraul  * @cpu: cpu that just started
676e545a614SManfred Spraul  *
677e545a614SManfred Spraul  * This function calls the cpu_chain notifiers with CPU_STARTING.
678e545a614SManfred Spraul  * It must be called by the arch code on the new cpu, before the new cpu
679e545a614SManfred Spraul  * enables interrupts and before the "boot" cpu returns from __cpu_up().
680e545a614SManfred Spraul  */
6810db0628dSPaul Gortmaker void notify_cpu_starting(unsigned int cpu)
682e545a614SManfred Spraul {
683e545a614SManfred Spraul 	unsigned long val = CPU_STARTING;
684e545a614SManfred Spraul 
685e545a614SManfred Spraul #ifdef CONFIG_PM_SLEEP_SMP
686e0b582ecSRusty Russell 	if (frozen_cpus != NULL && cpumask_test_cpu(cpu, frozen_cpus))
687e545a614SManfred Spraul 		val = CPU_STARTING_FROZEN;
688e545a614SManfred Spraul #endif /* CONFIG_PM_SLEEP_SMP */
689e9fb7631SAkinobu Mita 	cpu_notify(val, (void *)(long)cpu);
690e545a614SManfred Spraul }
691e545a614SManfred Spraul 
69268f4f1ecSMax Krasnyansky #endif /* CONFIG_SMP */
693b8d317d1SMike Travis 
694e56b3bc7SLinus Torvalds /*
695e56b3bc7SLinus Torvalds  * cpu_bit_bitmap[] is a special, "compressed" data structure that
696e56b3bc7SLinus Torvalds  * represents all NR_CPUS bits binary values of 1<<nr.
697e56b3bc7SLinus Torvalds  *
698e0b582ecSRusty Russell  * It is used by cpumask_of() to get a constant address to a CPU
699e56b3bc7SLinus Torvalds  * mask value that has a single bit set only.
700e56b3bc7SLinus Torvalds  */
701b8d317d1SMike Travis 
702e56b3bc7SLinus Torvalds /* cpu_bit_bitmap[0] is empty - so we can back into it */
7034d51985eSMichael Rodriguez #define MASK_DECLARE_1(x)	[x+1][0] = (1UL << (x))
704e56b3bc7SLinus Torvalds #define MASK_DECLARE_2(x)	MASK_DECLARE_1(x), MASK_DECLARE_1(x+1)
705e56b3bc7SLinus Torvalds #define MASK_DECLARE_4(x)	MASK_DECLARE_2(x), MASK_DECLARE_2(x+2)
706e56b3bc7SLinus Torvalds #define MASK_DECLARE_8(x)	MASK_DECLARE_4(x), MASK_DECLARE_4(x+4)
707b8d317d1SMike Travis 
708e56b3bc7SLinus Torvalds const unsigned long cpu_bit_bitmap[BITS_PER_LONG+1][BITS_TO_LONGS(NR_CPUS)] = {
709b8d317d1SMike Travis 
710e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(0),	MASK_DECLARE_8(8),
711e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(16),	MASK_DECLARE_8(24),
712e56b3bc7SLinus Torvalds #if BITS_PER_LONG > 32
713e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(32),	MASK_DECLARE_8(40),
714e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(48),	MASK_DECLARE_8(56),
715b8d317d1SMike Travis #endif
716b8d317d1SMike Travis };
717e56b3bc7SLinus Torvalds EXPORT_SYMBOL_GPL(cpu_bit_bitmap);
7182d3854a3SRusty Russell 
7192d3854a3SRusty Russell const DECLARE_BITMAP(cpu_all_bits, NR_CPUS) = CPU_BITS_ALL;
7202d3854a3SRusty Russell EXPORT_SYMBOL(cpu_all_bits);
721b3199c02SRusty Russell 
722b3199c02SRusty Russell #ifdef CONFIG_INIT_ALL_POSSIBLE
723b3199c02SRusty Russell static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly
724b3199c02SRusty Russell 	= CPU_BITS_ALL;
725b3199c02SRusty Russell #else
726b3199c02SRusty Russell static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly;
727b3199c02SRusty Russell #endif
728b3199c02SRusty Russell const struct cpumask *const cpu_possible_mask = to_cpumask(cpu_possible_bits);
729b3199c02SRusty Russell EXPORT_SYMBOL(cpu_possible_mask);
730b3199c02SRusty Russell 
731b3199c02SRusty Russell static DECLARE_BITMAP(cpu_online_bits, CONFIG_NR_CPUS) __read_mostly;
732b3199c02SRusty Russell const struct cpumask *const cpu_online_mask = to_cpumask(cpu_online_bits);
733b3199c02SRusty Russell EXPORT_SYMBOL(cpu_online_mask);
734b3199c02SRusty Russell 
735b3199c02SRusty Russell static DECLARE_BITMAP(cpu_present_bits, CONFIG_NR_CPUS) __read_mostly;
736b3199c02SRusty Russell const struct cpumask *const cpu_present_mask = to_cpumask(cpu_present_bits);
737b3199c02SRusty Russell EXPORT_SYMBOL(cpu_present_mask);
738b3199c02SRusty Russell 
739b3199c02SRusty Russell static DECLARE_BITMAP(cpu_active_bits, CONFIG_NR_CPUS) __read_mostly;
740b3199c02SRusty Russell const struct cpumask *const cpu_active_mask = to_cpumask(cpu_active_bits);
741b3199c02SRusty Russell EXPORT_SYMBOL(cpu_active_mask);
7423fa41520SRusty Russell 
7433fa41520SRusty Russell void set_cpu_possible(unsigned int cpu, bool possible)
7443fa41520SRusty Russell {
7453fa41520SRusty Russell 	if (possible)
7463fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_possible_bits));
7473fa41520SRusty Russell 	else
7483fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_possible_bits));
7493fa41520SRusty Russell }
7503fa41520SRusty Russell 
7513fa41520SRusty Russell void set_cpu_present(unsigned int cpu, bool present)
7523fa41520SRusty Russell {
7533fa41520SRusty Russell 	if (present)
7543fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_present_bits));
7553fa41520SRusty Russell 	else
7563fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_present_bits));
7573fa41520SRusty Russell }
7583fa41520SRusty Russell 
7593fa41520SRusty Russell void set_cpu_online(unsigned int cpu, bool online)
7603fa41520SRusty Russell {
7616acbfb96SLai Jiangshan 	if (online) {
7623fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_online_bits));
7636acbfb96SLai Jiangshan 		cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
7646acbfb96SLai Jiangshan 	} else {
7653fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_online_bits));
7663fa41520SRusty Russell 	}
7676acbfb96SLai Jiangshan }
7683fa41520SRusty Russell 
7693fa41520SRusty Russell void set_cpu_active(unsigned int cpu, bool active)
7703fa41520SRusty Russell {
7713fa41520SRusty Russell 	if (active)
7723fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
7733fa41520SRusty Russell 	else
7743fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_active_bits));
7753fa41520SRusty Russell }
7763fa41520SRusty Russell 
7773fa41520SRusty Russell void init_cpu_present(const struct cpumask *src)
7783fa41520SRusty Russell {
7793fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_present_bits), src);
7803fa41520SRusty Russell }
7813fa41520SRusty Russell 
7823fa41520SRusty Russell void init_cpu_possible(const struct cpumask *src)
7833fa41520SRusty Russell {
7843fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_possible_bits), src);
7853fa41520SRusty Russell }
7863fa41520SRusty Russell 
7873fa41520SRusty Russell void init_cpu_online(const struct cpumask *src)
7883fa41520SRusty Russell {
7893fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_online_bits), src);
7903fa41520SRusty Russell }
791