xref: /linux-6.15/kernel/cpu.c (revision b2c4623d)
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;
61d221938cSGautham R Shenoy 	struct mutex lock; /* Synchronizes accesses to refcount, */
62d221938cSGautham R Shenoy 	/*
63d221938cSGautham R Shenoy 	 * Also blocks the new readers during
64d221938cSGautham R Shenoy 	 * an ongoing cpu hotplug operation.
65d221938cSGautham R Shenoy 	 */
66d221938cSGautham R Shenoy 	int refcount;
67*b2c4623dSPaul E. McKenney 	/* And allows lockless put_online_cpus(). */
68*b2c4623dSPaul E. McKenney 	atomic_t puts_pending;
69a19423b9SGautham R. Shenoy 
70a19423b9SGautham R. Shenoy #ifdef CONFIG_DEBUG_LOCK_ALLOC
71a19423b9SGautham R. Shenoy 	struct lockdep_map dep_map;
72a19423b9SGautham R. Shenoy #endif
7331950eb6SLinus Torvalds } cpu_hotplug = {
7431950eb6SLinus Torvalds 	.active_writer = NULL,
7531950eb6SLinus Torvalds 	.lock = __MUTEX_INITIALIZER(cpu_hotplug.lock),
7631950eb6SLinus Torvalds 	.refcount = 0,
77a19423b9SGautham R. Shenoy #ifdef CONFIG_DEBUG_LOCK_ALLOC
78a19423b9SGautham R. Shenoy 	.dep_map = {.name = "cpu_hotplug.lock" },
79a19423b9SGautham R. Shenoy #endif
8031950eb6SLinus Torvalds };
81d221938cSGautham R Shenoy 
82a19423b9SGautham R. Shenoy /* Lockdep annotations for get/put_online_cpus() and cpu_hotplug_begin/end() */
83a19423b9SGautham R. Shenoy #define cpuhp_lock_acquire_read() lock_map_acquire_read(&cpu_hotplug.dep_map)
84dd56af42SPaul E. McKenney #define cpuhp_lock_acquire_tryread() \
85dd56af42SPaul E. McKenney 				  lock_map_acquire_tryread(&cpu_hotplug.dep_map)
86a19423b9SGautham R. Shenoy #define cpuhp_lock_acquire()      lock_map_acquire(&cpu_hotplug.dep_map)
87a19423b9SGautham R. Shenoy #define cpuhp_lock_release()      lock_map_release(&cpu_hotplug.dep_map)
88a19423b9SGautham R. Shenoy 
8986ef5c9aSGautham R Shenoy void get_online_cpus(void)
90a9d9baa1SAshok Raj {
91d221938cSGautham R Shenoy 	might_sleep();
92d221938cSGautham R Shenoy 	if (cpu_hotplug.active_writer == current)
93aa953877SLinus Torvalds 		return;
94a19423b9SGautham R. Shenoy 	cpuhp_lock_acquire_read();
95d221938cSGautham R Shenoy 	mutex_lock(&cpu_hotplug.lock);
96d221938cSGautham R Shenoy 	cpu_hotplug.refcount++;
97d221938cSGautham R Shenoy 	mutex_unlock(&cpu_hotplug.lock);
98a9d9baa1SAshok Raj }
9986ef5c9aSGautham R Shenoy EXPORT_SYMBOL_GPL(get_online_cpus);
100a9d9baa1SAshok Raj 
101dd56af42SPaul E. McKenney bool try_get_online_cpus(void)
102dd56af42SPaul E. McKenney {
103dd56af42SPaul E. McKenney 	if (cpu_hotplug.active_writer == current)
104dd56af42SPaul E. McKenney 		return true;
105dd56af42SPaul E. McKenney 	if (!mutex_trylock(&cpu_hotplug.lock))
106dd56af42SPaul E. McKenney 		return false;
107dd56af42SPaul E. McKenney 	cpuhp_lock_acquire_tryread();
108dd56af42SPaul E. McKenney 	cpu_hotplug.refcount++;
109dd56af42SPaul E. McKenney 	mutex_unlock(&cpu_hotplug.lock);
110dd56af42SPaul E. McKenney 	return true;
111dd56af42SPaul E. McKenney }
112dd56af42SPaul E. McKenney EXPORT_SYMBOL_GPL(try_get_online_cpus);
113dd56af42SPaul E. McKenney 
11486ef5c9aSGautham R Shenoy void put_online_cpus(void)
115a9d9baa1SAshok Raj {
116d221938cSGautham R Shenoy 	if (cpu_hotplug.active_writer == current)
117aa953877SLinus Torvalds 		return;
118*b2c4623dSPaul E. McKenney 	if (!mutex_trylock(&cpu_hotplug.lock)) {
119*b2c4623dSPaul E. McKenney 		atomic_inc(&cpu_hotplug.puts_pending);
120*b2c4623dSPaul E. McKenney 		cpuhp_lock_release();
121*b2c4623dSPaul E. McKenney 		return;
122*b2c4623dSPaul E. McKenney 	}
123075663d1SSrivatsa S. Bhat 
124075663d1SSrivatsa S. Bhat 	if (WARN_ON(!cpu_hotplug.refcount))
125075663d1SSrivatsa S. Bhat 		cpu_hotplug.refcount++; /* try to fix things up */
126075663d1SSrivatsa S. Bhat 
127d2ba7e2aSOleg Nesterov 	if (!--cpu_hotplug.refcount && unlikely(cpu_hotplug.active_writer))
128d2ba7e2aSOleg Nesterov 		wake_up_process(cpu_hotplug.active_writer);
129d221938cSGautham R Shenoy 	mutex_unlock(&cpu_hotplug.lock);
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 {
159d221938cSGautham R Shenoy 	cpu_hotplug.active_writer = current;
160d2ba7e2aSOleg Nesterov 
161a19423b9SGautham R. Shenoy 	cpuhp_lock_acquire();
162d2ba7e2aSOleg Nesterov 	for (;;) {
163d2ba7e2aSOleg Nesterov 		mutex_lock(&cpu_hotplug.lock);
164*b2c4623dSPaul E. McKenney 		if (atomic_read(&cpu_hotplug.puts_pending)) {
165*b2c4623dSPaul E. McKenney 			int delta;
166*b2c4623dSPaul E. McKenney 
167*b2c4623dSPaul E. McKenney 			delta = atomic_xchg(&cpu_hotplug.puts_pending, 0);
168*b2c4623dSPaul E. McKenney 			cpu_hotplug.refcount -= delta;
169*b2c4623dSPaul E. McKenney 		}
170d2ba7e2aSOleg Nesterov 		if (likely(!cpu_hotplug.refcount))
171d2ba7e2aSOleg Nesterov 			break;
172d2ba7e2aSOleg Nesterov 		__set_current_state(TASK_UNINTERRUPTIBLE);
173d221938cSGautham R Shenoy 		mutex_unlock(&cpu_hotplug.lock);
174d221938cSGautham R Shenoy 		schedule();
175d221938cSGautham R Shenoy 	}
176d221938cSGautham R Shenoy }
177d221938cSGautham R Shenoy 
178b9d10be7SToshi Kani void cpu_hotplug_done(void)
179d221938cSGautham R Shenoy {
180d221938cSGautham R Shenoy 	cpu_hotplug.active_writer = NULL;
181d221938cSGautham R Shenoy 	mutex_unlock(&cpu_hotplug.lock);
182a19423b9SGautham R. Shenoy 	cpuhp_lock_release();
183d221938cSGautham R Shenoy }
18479a6cdebSLai Jiangshan 
18516e53dbfSSrivatsa S. Bhat /*
18616e53dbfSSrivatsa S. Bhat  * Wait for currently running CPU hotplug operations to complete (if any) and
18716e53dbfSSrivatsa S. Bhat  * disable future CPU hotplug (from sysfs). The 'cpu_add_remove_lock' protects
18816e53dbfSSrivatsa S. Bhat  * the 'cpu_hotplug_disabled' flag. The same lock is also acquired by the
18916e53dbfSSrivatsa S. Bhat  * hotplug path before performing hotplug operations. So acquiring that lock
19016e53dbfSSrivatsa S. Bhat  * guarantees mutual exclusion from any currently running hotplug operations.
19116e53dbfSSrivatsa S. Bhat  */
19216e53dbfSSrivatsa S. Bhat void cpu_hotplug_disable(void)
19316e53dbfSSrivatsa S. Bhat {
19416e53dbfSSrivatsa S. Bhat 	cpu_maps_update_begin();
19516e53dbfSSrivatsa S. Bhat 	cpu_hotplug_disabled = 1;
19616e53dbfSSrivatsa S. Bhat 	cpu_maps_update_done();
19716e53dbfSSrivatsa S. Bhat }
19816e53dbfSSrivatsa S. Bhat 
19916e53dbfSSrivatsa S. Bhat void cpu_hotplug_enable(void)
20016e53dbfSSrivatsa S. Bhat {
20116e53dbfSSrivatsa S. Bhat 	cpu_maps_update_begin();
20216e53dbfSSrivatsa S. Bhat 	cpu_hotplug_disabled = 0;
20316e53dbfSSrivatsa S. Bhat 	cpu_maps_update_done();
20416e53dbfSSrivatsa S. Bhat }
20516e53dbfSSrivatsa S. Bhat 
206b9d10be7SToshi Kani #endif	/* CONFIG_HOTPLUG_CPU */
20779a6cdebSLai Jiangshan 
2081da177e4SLinus Torvalds /* Need to know about CPUs going up/down? */
209f7b16c10SSam Ravnborg int __ref register_cpu_notifier(struct notifier_block *nb)
2101da177e4SLinus Torvalds {
211bd5349cfSNeil Brown 	int ret;
212d221938cSGautham R Shenoy 	cpu_maps_update_begin();
213bd5349cfSNeil Brown 	ret = raw_notifier_chain_register(&cpu_chain, nb);
214d221938cSGautham R Shenoy 	cpu_maps_update_done();
215bd5349cfSNeil Brown 	return ret;
2161da177e4SLinus Torvalds }
21765edc68cSChandra Seetharaman 
21893ae4f97SSrivatsa S. Bhat int __ref __register_cpu_notifier(struct notifier_block *nb)
21993ae4f97SSrivatsa S. Bhat {
22093ae4f97SSrivatsa S. Bhat 	return raw_notifier_chain_register(&cpu_chain, nb);
22193ae4f97SSrivatsa S. Bhat }
22293ae4f97SSrivatsa S. Bhat 
223e9fb7631SAkinobu Mita static int __cpu_notify(unsigned long val, void *v, int nr_to_call,
224e9fb7631SAkinobu Mita 			int *nr_calls)
225e9fb7631SAkinobu Mita {
226e6bde73bSAkinobu Mita 	int ret;
227e6bde73bSAkinobu Mita 
228e6bde73bSAkinobu Mita 	ret = __raw_notifier_call_chain(&cpu_chain, val, v, nr_to_call,
229e9fb7631SAkinobu Mita 					nr_calls);
230e6bde73bSAkinobu Mita 
231e6bde73bSAkinobu Mita 	return notifier_to_errno(ret);
232e9fb7631SAkinobu Mita }
233e9fb7631SAkinobu Mita 
234e9fb7631SAkinobu Mita static int cpu_notify(unsigned long val, void *v)
235e9fb7631SAkinobu Mita {
236e9fb7631SAkinobu Mita 	return __cpu_notify(val, v, -1, NULL);
237e9fb7631SAkinobu Mita }
238e9fb7631SAkinobu Mita 
23900b9b0afSLinus Torvalds #ifdef CONFIG_HOTPLUG_CPU
24000b9b0afSLinus Torvalds 
241e9fb7631SAkinobu Mita static void cpu_notify_nofail(unsigned long val, void *v)
242e9fb7631SAkinobu Mita {
24300b9b0afSLinus Torvalds 	BUG_ON(cpu_notify(val, v));
244e9fb7631SAkinobu Mita }
2451da177e4SLinus Torvalds EXPORT_SYMBOL(register_cpu_notifier);
24693ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(__register_cpu_notifier);
2471da177e4SLinus Torvalds 
2489647155fSSam Ravnborg void __ref unregister_cpu_notifier(struct notifier_block *nb)
2491da177e4SLinus Torvalds {
250d221938cSGautham R Shenoy 	cpu_maps_update_begin();
251bd5349cfSNeil Brown 	raw_notifier_chain_unregister(&cpu_chain, nb);
252d221938cSGautham R Shenoy 	cpu_maps_update_done();
2531da177e4SLinus Torvalds }
2541da177e4SLinus Torvalds EXPORT_SYMBOL(unregister_cpu_notifier);
2551da177e4SLinus Torvalds 
25693ae4f97SSrivatsa S. Bhat void __ref __unregister_cpu_notifier(struct notifier_block *nb)
25793ae4f97SSrivatsa S. Bhat {
25893ae4f97SSrivatsa S. Bhat 	raw_notifier_chain_unregister(&cpu_chain, nb);
25993ae4f97SSrivatsa S. Bhat }
26093ae4f97SSrivatsa S. Bhat EXPORT_SYMBOL(__unregister_cpu_notifier);
26193ae4f97SSrivatsa S. Bhat 
262e4cc2f87SAnton Vorontsov /**
263e4cc2f87SAnton Vorontsov  * clear_tasks_mm_cpumask - Safely clear tasks' mm_cpumask for a CPU
264e4cc2f87SAnton Vorontsov  * @cpu: a CPU id
265e4cc2f87SAnton Vorontsov  *
266e4cc2f87SAnton Vorontsov  * This function walks all processes, finds a valid mm struct for each one and
267e4cc2f87SAnton Vorontsov  * then clears a corresponding bit in mm's cpumask.  While this all sounds
268e4cc2f87SAnton Vorontsov  * trivial, there are various non-obvious corner cases, which this function
269e4cc2f87SAnton Vorontsov  * tries to solve in a safe manner.
270e4cc2f87SAnton Vorontsov  *
271e4cc2f87SAnton Vorontsov  * Also note that the function uses a somewhat relaxed locking scheme, so it may
272e4cc2f87SAnton Vorontsov  * be called only for an already offlined CPU.
273e4cc2f87SAnton Vorontsov  */
274cb79295eSAnton Vorontsov void clear_tasks_mm_cpumask(int cpu)
275cb79295eSAnton Vorontsov {
276cb79295eSAnton Vorontsov 	struct task_struct *p;
277cb79295eSAnton Vorontsov 
278cb79295eSAnton Vorontsov 	/*
279cb79295eSAnton Vorontsov 	 * This function is called after the cpu is taken down and marked
280cb79295eSAnton Vorontsov 	 * offline, so its not like new tasks will ever get this cpu set in
281cb79295eSAnton Vorontsov 	 * their mm mask. -- Peter Zijlstra
282cb79295eSAnton Vorontsov 	 * Thus, we may use rcu_read_lock() here, instead of grabbing
283cb79295eSAnton Vorontsov 	 * full-fledged tasklist_lock.
284cb79295eSAnton Vorontsov 	 */
285e4cc2f87SAnton Vorontsov 	WARN_ON(cpu_online(cpu));
286cb79295eSAnton Vorontsov 	rcu_read_lock();
287cb79295eSAnton Vorontsov 	for_each_process(p) {
288cb79295eSAnton Vorontsov 		struct task_struct *t;
289cb79295eSAnton Vorontsov 
290e4cc2f87SAnton Vorontsov 		/*
291e4cc2f87SAnton Vorontsov 		 * Main thread might exit, but other threads may still have
292e4cc2f87SAnton Vorontsov 		 * a valid mm. Find one.
293e4cc2f87SAnton Vorontsov 		 */
294cb79295eSAnton Vorontsov 		t = find_lock_task_mm(p);
295cb79295eSAnton Vorontsov 		if (!t)
296cb79295eSAnton Vorontsov 			continue;
297cb79295eSAnton Vorontsov 		cpumask_clear_cpu(cpu, mm_cpumask(t->mm));
298cb79295eSAnton Vorontsov 		task_unlock(t);
299cb79295eSAnton Vorontsov 	}
300cb79295eSAnton Vorontsov 	rcu_read_unlock();
301cb79295eSAnton Vorontsov }
302cb79295eSAnton Vorontsov 
303b728ca06SKirill Tkhai static inline void check_for_tasks(int dead_cpu)
3041da177e4SLinus Torvalds {
305b728ca06SKirill Tkhai 	struct task_struct *g, *p;
3061da177e4SLinus Torvalds 
307b728ca06SKirill Tkhai 	read_lock_irq(&tasklist_lock);
308b728ca06SKirill Tkhai 	do_each_thread(g, p) {
309b728ca06SKirill Tkhai 		if (!p->on_rq)
310b728ca06SKirill Tkhai 			continue;
311b728ca06SKirill Tkhai 		/*
312b728ca06SKirill Tkhai 		 * We do the check with unlocked task_rq(p)->lock.
313b728ca06SKirill Tkhai 		 * Order the reading to do not warn about a task,
314b728ca06SKirill Tkhai 		 * which was running on this cpu in the past, and
315b728ca06SKirill Tkhai 		 * it's just been woken on another cpu.
316b728ca06SKirill Tkhai 		 */
317b728ca06SKirill Tkhai 		rmb();
318b728ca06SKirill Tkhai 		if (task_cpu(p) != dead_cpu)
319b728ca06SKirill Tkhai 			continue;
320b728ca06SKirill Tkhai 
32184117da5SFabian Frederick 		pr_warn("Task %s (pid=%d) is on cpu %d (state=%ld, flags=%x)\n",
322b728ca06SKirill Tkhai 			p->comm, task_pid_nr(p), dead_cpu, p->state, p->flags);
323b728ca06SKirill Tkhai 	} while_each_thread(g, p);
324b728ca06SKirill Tkhai 	read_unlock_irq(&tasklist_lock);
3251da177e4SLinus Torvalds }
3261da177e4SLinus Torvalds 
327db912f96SAvi Kivity struct take_cpu_down_param {
328db912f96SAvi Kivity 	unsigned long mod;
329db912f96SAvi Kivity 	void *hcpu;
330db912f96SAvi Kivity };
331db912f96SAvi Kivity 
3321da177e4SLinus Torvalds /* Take this CPU down. */
333514a20a5SSam Ravnborg static int __ref take_cpu_down(void *_param)
3341da177e4SLinus Torvalds {
335db912f96SAvi Kivity 	struct take_cpu_down_param *param = _param;
3361da177e4SLinus Torvalds 	int err;
3371da177e4SLinus Torvalds 
3381da177e4SLinus Torvalds 	/* Ensure this CPU doesn't handle any more interrupts. */
3391da177e4SLinus Torvalds 	err = __cpu_disable();
3401da177e4SLinus Torvalds 	if (err < 0)
341f3705136SZwane Mwaikambo 		return err;
342f3705136SZwane Mwaikambo 
343e9fb7631SAkinobu Mita 	cpu_notify(CPU_DYING | param->mod, param->hcpu);
34414e568e7SThomas Gleixner 	/* Park the stopper thread */
34514e568e7SThomas Gleixner 	kthread_park(current);
346f3705136SZwane Mwaikambo 	return 0;
3471da177e4SLinus Torvalds }
3481da177e4SLinus Torvalds 
349e3920fb4SRafael J. Wysocki /* Requires cpu_add_remove_lock to be held */
350514a20a5SSam Ravnborg static int __ref _cpu_down(unsigned int cpu, int tasks_frozen)
3511da177e4SLinus Torvalds {
352e7407dccSHeiko Carstens 	int err, nr_calls = 0;
353e7407dccSHeiko Carstens 	void *hcpu = (void *)(long)cpu;
3548bb78442SRafael J. Wysocki 	unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
355db912f96SAvi Kivity 	struct take_cpu_down_param tcd_param = {
356db912f96SAvi Kivity 		.mod = mod,
357db912f96SAvi Kivity 		.hcpu = hcpu,
358db912f96SAvi Kivity 	};
3591da177e4SLinus Torvalds 
360e3920fb4SRafael J. Wysocki 	if (num_online_cpus() == 1)
361e3920fb4SRafael J. Wysocki 		return -EBUSY;
3621da177e4SLinus Torvalds 
363e3920fb4SRafael J. Wysocki 	if (!cpu_online(cpu))
364e3920fb4SRafael J. Wysocki 		return -EINVAL;
3651da177e4SLinus Torvalds 
366d221938cSGautham R Shenoy 	cpu_hotplug_begin();
3674d51985eSMichael Rodriguez 
368e9fb7631SAkinobu Mita 	err = __cpu_notify(CPU_DOWN_PREPARE | mod, hcpu, -1, &nr_calls);
369e6bde73bSAkinobu Mita 	if (err) {
370a0d8cdb6SAkinobu Mita 		nr_calls--;
371e9fb7631SAkinobu Mita 		__cpu_notify(CPU_DOWN_FAILED | mod, hcpu, nr_calls, NULL);
37284117da5SFabian Frederick 		pr_warn("%s: attempt to take down CPU %u failed\n",
373af1f16d0SHarvey Harrison 			__func__, cpu);
374baaca49fSGautham R Shenoy 		goto out_release;
3751da177e4SLinus Torvalds 	}
3761da177e4SLinus Torvalds 
3776acce3efSPeter Zijlstra 	/*
3786acce3efSPeter Zijlstra 	 * By now we've cleared cpu_active_mask, wait for all preempt-disabled
3796acce3efSPeter Zijlstra 	 * and RCU users of this state to go away such that all new such users
3806acce3efSPeter Zijlstra 	 * will observe it.
3816acce3efSPeter Zijlstra 	 *
3826acce3efSPeter Zijlstra 	 * For CONFIG_PREEMPT we have preemptible RCU and its sync_rcu() might
3836acce3efSPeter Zijlstra 	 * not imply sync_sched(), so explicitly call both.
384106dd5afSMichael wang 	 *
385106dd5afSMichael wang 	 * Do sync before park smpboot threads to take care the rcu boost case.
3866acce3efSPeter Zijlstra 	 */
3876acce3efSPeter Zijlstra #ifdef CONFIG_PREEMPT
3886acce3efSPeter Zijlstra 	synchronize_sched();
3896acce3efSPeter Zijlstra #endif
3906acce3efSPeter Zijlstra 	synchronize_rcu();
3916acce3efSPeter Zijlstra 
392106dd5afSMichael wang 	smpboot_park_threads(cpu);
393106dd5afSMichael wang 
3946acce3efSPeter Zijlstra 	/*
3956acce3efSPeter Zijlstra 	 * So now all preempt/rcu users must observe !cpu_active().
3966acce3efSPeter Zijlstra 	 */
3976acce3efSPeter Zijlstra 
398e0b582ecSRusty Russell 	err = __stop_machine(take_cpu_down, &tcd_param, cpumask_of(cpu));
39904321587SRusty Russell 	if (err) {
4001da177e4SLinus Torvalds 		/* CPU didn't die: tell everyone.  Can't complain. */
401f97f8f06SThomas Gleixner 		smpboot_unpark_threads(cpu);
402e9fb7631SAkinobu Mita 		cpu_notify_nofail(CPU_DOWN_FAILED | mod, hcpu);
4036a1bdc1bSOleg Nesterov 		goto out_release;
4041da177e4SLinus Torvalds 	}
40504321587SRusty Russell 	BUG_ON(cpu_online(cpu));
4061da177e4SLinus Torvalds 
40748c5ccaeSPeter Zijlstra 	/*
40848c5ccaeSPeter Zijlstra 	 * The migration_call() CPU_DYING callback will have removed all
40948c5ccaeSPeter Zijlstra 	 * runnable tasks from the cpu, there's only the idle task left now
41048c5ccaeSPeter Zijlstra 	 * that the migration thread is done doing the stop_machine thing.
41151a96c77SPeter Zijlstra 	 *
41251a96c77SPeter Zijlstra 	 * Wait for the stop thread to go away.
41348c5ccaeSPeter Zijlstra 	 */
41451a96c77SPeter Zijlstra 	while (!idle_cpu(cpu))
41551a96c77SPeter Zijlstra 		cpu_relax();
4161da177e4SLinus Torvalds 
4171da177e4SLinus Torvalds 	/* This actually kills the CPU. */
4181da177e4SLinus Torvalds 	__cpu_die(cpu);
4191da177e4SLinus Torvalds 
4201da177e4SLinus Torvalds 	/* CPU is completely dead: tell everyone.  Too late to complain. */
421e9fb7631SAkinobu Mita 	cpu_notify_nofail(CPU_DEAD | mod, hcpu);
4221da177e4SLinus Torvalds 
4231da177e4SLinus Torvalds 	check_for_tasks(cpu);
4241da177e4SLinus Torvalds 
425baaca49fSGautham R Shenoy out_release:
426d221938cSGautham R Shenoy 	cpu_hotplug_done();
427e9fb7631SAkinobu Mita 	if (!err)
428e9fb7631SAkinobu Mita 		cpu_notify_nofail(CPU_POST_DEAD | mod, hcpu);
429e3920fb4SRafael J. Wysocki 	return err;
430e3920fb4SRafael J. Wysocki }
431e3920fb4SRafael J. Wysocki 
432514a20a5SSam Ravnborg int __ref cpu_down(unsigned int cpu)
433e3920fb4SRafael J. Wysocki {
4349ea09af3SHeiko Carstens 	int err;
435e3920fb4SRafael J. Wysocki 
436d221938cSGautham R Shenoy 	cpu_maps_update_begin();
437e761b772SMax Krasnyansky 
438e761b772SMax Krasnyansky 	if (cpu_hotplug_disabled) {
439e3920fb4SRafael J. Wysocki 		err = -EBUSY;
440e761b772SMax Krasnyansky 		goto out;
441e761b772SMax Krasnyansky 	}
442e761b772SMax Krasnyansky 
4438bb78442SRafael J. Wysocki 	err = _cpu_down(cpu, 0);
444e3920fb4SRafael J. Wysocki 
445e761b772SMax Krasnyansky out:
446d221938cSGautham R Shenoy 	cpu_maps_update_done();
4471da177e4SLinus Torvalds 	return err;
4481da177e4SLinus Torvalds }
449b62b8ef9SZhang Rui EXPORT_SYMBOL(cpu_down);
4501da177e4SLinus Torvalds #endif /*CONFIG_HOTPLUG_CPU*/
4511da177e4SLinus Torvalds 
452e3920fb4SRafael J. Wysocki /* Requires cpu_add_remove_lock to be held */
4530db0628dSPaul Gortmaker static int _cpu_up(unsigned int cpu, int tasks_frozen)
4541da177e4SLinus Torvalds {
455baaca49fSGautham R Shenoy 	int ret, nr_calls = 0;
4561da177e4SLinus Torvalds 	void *hcpu = (void *)(long)cpu;
4578bb78442SRafael J. Wysocki 	unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
4583bb5d2eeSSuresh Siddha 	struct task_struct *idle;
4591da177e4SLinus Torvalds 
460d221938cSGautham R Shenoy 	cpu_hotplug_begin();
46138498a67SThomas Gleixner 
4625e5041f3SYasuaki Ishimatsu 	if (cpu_online(cpu) || !cpu_present(cpu)) {
4635e5041f3SYasuaki Ishimatsu 		ret = -EINVAL;
4645e5041f3SYasuaki Ishimatsu 		goto out;
4655e5041f3SYasuaki Ishimatsu 	}
4665e5041f3SYasuaki Ishimatsu 
4673bb5d2eeSSuresh Siddha 	idle = idle_thread_get(cpu);
4683bb5d2eeSSuresh Siddha 	if (IS_ERR(idle)) {
4693bb5d2eeSSuresh Siddha 		ret = PTR_ERR(idle);
47038498a67SThomas Gleixner 		goto out;
4713bb5d2eeSSuresh Siddha 	}
47238498a67SThomas Gleixner 
473f97f8f06SThomas Gleixner 	ret = smpboot_create_threads(cpu);
474f97f8f06SThomas Gleixner 	if (ret)
475f97f8f06SThomas Gleixner 		goto out;
476f97f8f06SThomas Gleixner 
477e9fb7631SAkinobu Mita 	ret = __cpu_notify(CPU_UP_PREPARE | mod, hcpu, -1, &nr_calls);
478e6bde73bSAkinobu Mita 	if (ret) {
479a0d8cdb6SAkinobu Mita 		nr_calls--;
48084117da5SFabian Frederick 		pr_warn("%s: attempt to bring up CPU %u failed\n",
481af1f16d0SHarvey Harrison 			__func__, cpu);
4821da177e4SLinus Torvalds 		goto out_notify;
4831da177e4SLinus Torvalds 	}
4841da177e4SLinus Torvalds 
4851da177e4SLinus Torvalds 	/* Arch-specific enabling code. */
4863bb5d2eeSSuresh Siddha 	ret = __cpu_up(cpu, idle);
4871da177e4SLinus Torvalds 	if (ret != 0)
4881da177e4SLinus Torvalds 		goto out_notify;
4896978c705SEric Sesterhenn 	BUG_ON(!cpu_online(cpu));
4901da177e4SLinus Torvalds 
491f97f8f06SThomas Gleixner 	/* Wake the per cpu threads */
492f97f8f06SThomas Gleixner 	smpboot_unpark_threads(cpu);
493f97f8f06SThomas Gleixner 
4941da177e4SLinus Torvalds 	/* Now call notifier in preparation. */
495e9fb7631SAkinobu Mita 	cpu_notify(CPU_ONLINE | mod, hcpu);
4961da177e4SLinus Torvalds 
4971da177e4SLinus Torvalds out_notify:
4981da177e4SLinus Torvalds 	if (ret != 0)
499e9fb7631SAkinobu Mita 		__cpu_notify(CPU_UP_CANCELED | mod, hcpu, nr_calls, NULL);
50038498a67SThomas Gleixner out:
501d221938cSGautham R Shenoy 	cpu_hotplug_done();
502e3920fb4SRafael J. Wysocki 
5031da177e4SLinus Torvalds 	return ret;
5041da177e4SLinus Torvalds }
505e3920fb4SRafael J. Wysocki 
5060db0628dSPaul Gortmaker int cpu_up(unsigned int cpu)
507e3920fb4SRafael J. Wysocki {
508e3920fb4SRafael J. Wysocki 	int err = 0;
509cf23422bSminskey guo 
510e0b582ecSRusty Russell 	if (!cpu_possible(cpu)) {
51184117da5SFabian Frederick 		pr_err("can't online cpu %d because it is not configured as may-hotadd at boot time\n",
51284117da5SFabian Frederick 		       cpu);
51387d5e023SChen Gong #if defined(CONFIG_IA64)
51484117da5SFabian Frederick 		pr_err("please check additional_cpus= boot parameter\n");
51573e753a5SKAMEZAWA Hiroyuki #endif
51673e753a5SKAMEZAWA Hiroyuki 		return -EINVAL;
51773e753a5SKAMEZAWA Hiroyuki 	}
518e3920fb4SRafael J. Wysocki 
51901b0f197SToshi Kani 	err = try_online_node(cpu_to_node(cpu));
520cf23422bSminskey guo 	if (err)
521cf23422bSminskey guo 		return err;
522cf23422bSminskey guo 
523d221938cSGautham R Shenoy 	cpu_maps_update_begin();
524e761b772SMax Krasnyansky 
525e761b772SMax Krasnyansky 	if (cpu_hotplug_disabled) {
526e3920fb4SRafael J. Wysocki 		err = -EBUSY;
527e761b772SMax Krasnyansky 		goto out;
528e761b772SMax Krasnyansky 	}
529e761b772SMax Krasnyansky 
5308bb78442SRafael J. Wysocki 	err = _cpu_up(cpu, 0);
531e3920fb4SRafael J. Wysocki 
532e761b772SMax Krasnyansky out:
533d221938cSGautham R Shenoy 	cpu_maps_update_done();
534e3920fb4SRafael J. Wysocki 	return err;
535e3920fb4SRafael J. Wysocki }
536a513f6baSPaul E. McKenney EXPORT_SYMBOL_GPL(cpu_up);
537e3920fb4SRafael J. Wysocki 
538f3de4be9SRafael J. Wysocki #ifdef CONFIG_PM_SLEEP_SMP
539e0b582ecSRusty Russell static cpumask_var_t frozen_cpus;
540e3920fb4SRafael J. Wysocki 
541e3920fb4SRafael J. Wysocki int disable_nonboot_cpus(void)
542e3920fb4SRafael J. Wysocki {
543e9a5f426SRafael J. Wysocki 	int cpu, first_cpu, error = 0;
544e3920fb4SRafael J. Wysocki 
545d221938cSGautham R Shenoy 	cpu_maps_update_begin();
546e0b582ecSRusty Russell 	first_cpu = cpumask_first(cpu_online_mask);
5479ee349adSXiaotian Feng 	/*
5489ee349adSXiaotian Feng 	 * We take down all of the non-boot CPUs in one shot to avoid races
549e3920fb4SRafael J. Wysocki 	 * with the userspace trying to use the CPU hotplug at the same time
550e3920fb4SRafael J. Wysocki 	 */
551e0b582ecSRusty Russell 	cpumask_clear(frozen_cpus);
5526ad4c188SPeter Zijlstra 
55384117da5SFabian Frederick 	pr_info("Disabling non-boot CPUs ...\n");
554e3920fb4SRafael J. Wysocki 	for_each_online_cpu(cpu) {
555e3920fb4SRafael J. Wysocki 		if (cpu == first_cpu)
556e3920fb4SRafael J. Wysocki 			continue;
557bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_OFF"), cpu, true);
5588bb78442SRafael J. Wysocki 		error = _cpu_down(cpu, 1);
559bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_OFF"), cpu, false);
560feae3203SMike Travis 		if (!error)
561e0b582ecSRusty Russell 			cpumask_set_cpu(cpu, frozen_cpus);
562feae3203SMike Travis 		else {
56384117da5SFabian Frederick 			pr_err("Error taking CPU%d down: %d\n", cpu, error);
564e3920fb4SRafael J. Wysocki 			break;
565e3920fb4SRafael J. Wysocki 		}
566e3920fb4SRafael J. Wysocki 	}
56786886e55SJoseph Cihula 
568e3920fb4SRafael J. Wysocki 	if (!error) {
569e3920fb4SRafael J. Wysocki 		BUG_ON(num_online_cpus() > 1);
570e3920fb4SRafael J. Wysocki 		/* Make sure the CPUs won't be enabled by someone else */
571e3920fb4SRafael J. Wysocki 		cpu_hotplug_disabled = 1;
572e3920fb4SRafael J. Wysocki 	} else {
57384117da5SFabian Frederick 		pr_err("Non-boot CPUs are not disabled\n");
574e3920fb4SRafael J. Wysocki 	}
575d221938cSGautham R Shenoy 	cpu_maps_update_done();
576e3920fb4SRafael J. Wysocki 	return error;
577e3920fb4SRafael J. Wysocki }
578e3920fb4SRafael J. Wysocki 
579d0af9eedSSuresh Siddha void __weak arch_enable_nonboot_cpus_begin(void)
580d0af9eedSSuresh Siddha {
581d0af9eedSSuresh Siddha }
582d0af9eedSSuresh Siddha 
583d0af9eedSSuresh Siddha void __weak arch_enable_nonboot_cpus_end(void)
584d0af9eedSSuresh Siddha {
585d0af9eedSSuresh Siddha }
586d0af9eedSSuresh Siddha 
587fa7303e2SSam Ravnborg void __ref enable_nonboot_cpus(void)
588e3920fb4SRafael J. Wysocki {
589e3920fb4SRafael J. Wysocki 	int cpu, error;
590e3920fb4SRafael J. Wysocki 
591e3920fb4SRafael J. Wysocki 	/* Allow everyone to use the CPU hotplug again */
592d221938cSGautham R Shenoy 	cpu_maps_update_begin();
593e3920fb4SRafael J. Wysocki 	cpu_hotplug_disabled = 0;
594e0b582ecSRusty Russell 	if (cpumask_empty(frozen_cpus))
5951d64b9cbSRafael J. Wysocki 		goto out;
596e3920fb4SRafael J. Wysocki 
59784117da5SFabian Frederick 	pr_info("Enabling non-boot CPUs ...\n");
598d0af9eedSSuresh Siddha 
599d0af9eedSSuresh Siddha 	arch_enable_nonboot_cpus_begin();
600d0af9eedSSuresh Siddha 
601e0b582ecSRusty Russell 	for_each_cpu(cpu, frozen_cpus) {
602bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_ON"), cpu, true);
6038bb78442SRafael J. Wysocki 		error = _cpu_up(cpu, 1);
604bb3632c6STodd E Brandt 		trace_suspend_resume(TPS("CPU_ON"), cpu, false);
605e3920fb4SRafael J. Wysocki 		if (!error) {
60684117da5SFabian Frederick 			pr_info("CPU%d is up\n", cpu);
607e3920fb4SRafael J. Wysocki 			continue;
608e3920fb4SRafael J. Wysocki 		}
60984117da5SFabian Frederick 		pr_warn("Error taking CPU%d up: %d\n", cpu, error);
610e3920fb4SRafael J. Wysocki 	}
611d0af9eedSSuresh Siddha 
612d0af9eedSSuresh Siddha 	arch_enable_nonboot_cpus_end();
613d0af9eedSSuresh Siddha 
614e0b582ecSRusty Russell 	cpumask_clear(frozen_cpus);
6151d64b9cbSRafael J. Wysocki out:
616d221938cSGautham R Shenoy 	cpu_maps_update_done();
617e3920fb4SRafael J. Wysocki }
618e0b582ecSRusty Russell 
619d7268a31SFenghua Yu static int __init alloc_frozen_cpus(void)
620e0b582ecSRusty Russell {
621e0b582ecSRusty Russell 	if (!alloc_cpumask_var(&frozen_cpus, GFP_KERNEL|__GFP_ZERO))
622e0b582ecSRusty Russell 		return -ENOMEM;
623e0b582ecSRusty Russell 	return 0;
624e0b582ecSRusty Russell }
625e0b582ecSRusty Russell core_initcall(alloc_frozen_cpus);
62679cfbdfaSSrivatsa S. Bhat 
62779cfbdfaSSrivatsa S. Bhat /*
62879cfbdfaSSrivatsa S. Bhat  * When callbacks for CPU hotplug notifications are being executed, we must
62979cfbdfaSSrivatsa S. Bhat  * ensure that the state of the system with respect to the tasks being frozen
63079cfbdfaSSrivatsa S. Bhat  * or not, as reported by the notification, remains unchanged *throughout the
63179cfbdfaSSrivatsa S. Bhat  * duration* of the execution of the callbacks.
63279cfbdfaSSrivatsa S. Bhat  * Hence we need to prevent the freezer from racing with regular CPU hotplug.
63379cfbdfaSSrivatsa S. Bhat  *
63479cfbdfaSSrivatsa S. Bhat  * This synchronization is implemented by mutually excluding regular CPU
63579cfbdfaSSrivatsa S. Bhat  * hotplug and Suspend/Hibernate call paths by hooking onto the Suspend/
63679cfbdfaSSrivatsa S. Bhat  * Hibernate notifications.
63779cfbdfaSSrivatsa S. Bhat  */
63879cfbdfaSSrivatsa S. Bhat static int
63979cfbdfaSSrivatsa S. Bhat cpu_hotplug_pm_callback(struct notifier_block *nb,
64079cfbdfaSSrivatsa S. Bhat 			unsigned long action, void *ptr)
64179cfbdfaSSrivatsa S. Bhat {
64279cfbdfaSSrivatsa S. Bhat 	switch (action) {
64379cfbdfaSSrivatsa S. Bhat 
64479cfbdfaSSrivatsa S. Bhat 	case PM_SUSPEND_PREPARE:
64579cfbdfaSSrivatsa S. Bhat 	case PM_HIBERNATION_PREPARE:
64616e53dbfSSrivatsa S. Bhat 		cpu_hotplug_disable();
64779cfbdfaSSrivatsa S. Bhat 		break;
64879cfbdfaSSrivatsa S. Bhat 
64979cfbdfaSSrivatsa S. Bhat 	case PM_POST_SUSPEND:
65079cfbdfaSSrivatsa S. Bhat 	case PM_POST_HIBERNATION:
65116e53dbfSSrivatsa S. Bhat 		cpu_hotplug_enable();
65279cfbdfaSSrivatsa S. Bhat 		break;
65379cfbdfaSSrivatsa S. Bhat 
65479cfbdfaSSrivatsa S. Bhat 	default:
65579cfbdfaSSrivatsa S. Bhat 		return NOTIFY_DONE;
65679cfbdfaSSrivatsa S. Bhat 	}
65779cfbdfaSSrivatsa S. Bhat 
65879cfbdfaSSrivatsa S. Bhat 	return NOTIFY_OK;
65979cfbdfaSSrivatsa S. Bhat }
66079cfbdfaSSrivatsa S. Bhat 
66179cfbdfaSSrivatsa S. Bhat 
662d7268a31SFenghua Yu static int __init cpu_hotplug_pm_sync_init(void)
66379cfbdfaSSrivatsa S. Bhat {
6646e32d479SFenghua Yu 	/*
6656e32d479SFenghua Yu 	 * cpu_hotplug_pm_callback has higher priority than x86
6666e32d479SFenghua Yu 	 * bsp_pm_callback which depends on cpu_hotplug_pm_callback
6676e32d479SFenghua Yu 	 * to disable cpu hotplug to avoid cpu hotplug race.
6686e32d479SFenghua Yu 	 */
66979cfbdfaSSrivatsa S. Bhat 	pm_notifier(cpu_hotplug_pm_callback, 0);
67079cfbdfaSSrivatsa S. Bhat 	return 0;
67179cfbdfaSSrivatsa S. Bhat }
67279cfbdfaSSrivatsa S. Bhat core_initcall(cpu_hotplug_pm_sync_init);
67379cfbdfaSSrivatsa S. Bhat 
674f3de4be9SRafael J. Wysocki #endif /* CONFIG_PM_SLEEP_SMP */
67568f4f1ecSMax Krasnyansky 
676e545a614SManfred Spraul /**
677e545a614SManfred Spraul  * notify_cpu_starting(cpu) - call the CPU_STARTING notifiers
678e545a614SManfred Spraul  * @cpu: cpu that just started
679e545a614SManfred Spraul  *
680e545a614SManfred Spraul  * This function calls the cpu_chain notifiers with CPU_STARTING.
681e545a614SManfred Spraul  * It must be called by the arch code on the new cpu, before the new cpu
682e545a614SManfred Spraul  * enables interrupts and before the "boot" cpu returns from __cpu_up().
683e545a614SManfred Spraul  */
6840db0628dSPaul Gortmaker void notify_cpu_starting(unsigned int cpu)
685e545a614SManfred Spraul {
686e545a614SManfred Spraul 	unsigned long val = CPU_STARTING;
687e545a614SManfred Spraul 
688e545a614SManfred Spraul #ifdef CONFIG_PM_SLEEP_SMP
689e0b582ecSRusty Russell 	if (frozen_cpus != NULL && cpumask_test_cpu(cpu, frozen_cpus))
690e545a614SManfred Spraul 		val = CPU_STARTING_FROZEN;
691e545a614SManfred Spraul #endif /* CONFIG_PM_SLEEP_SMP */
692e9fb7631SAkinobu Mita 	cpu_notify(val, (void *)(long)cpu);
693e545a614SManfred Spraul }
694e545a614SManfred Spraul 
69568f4f1ecSMax Krasnyansky #endif /* CONFIG_SMP */
696b8d317d1SMike Travis 
697e56b3bc7SLinus Torvalds /*
698e56b3bc7SLinus Torvalds  * cpu_bit_bitmap[] is a special, "compressed" data structure that
699e56b3bc7SLinus Torvalds  * represents all NR_CPUS bits binary values of 1<<nr.
700e56b3bc7SLinus Torvalds  *
701e0b582ecSRusty Russell  * It is used by cpumask_of() to get a constant address to a CPU
702e56b3bc7SLinus Torvalds  * mask value that has a single bit set only.
703e56b3bc7SLinus Torvalds  */
704b8d317d1SMike Travis 
705e56b3bc7SLinus Torvalds /* cpu_bit_bitmap[0] is empty - so we can back into it */
7064d51985eSMichael Rodriguez #define MASK_DECLARE_1(x)	[x+1][0] = (1UL << (x))
707e56b3bc7SLinus Torvalds #define MASK_DECLARE_2(x)	MASK_DECLARE_1(x), MASK_DECLARE_1(x+1)
708e56b3bc7SLinus Torvalds #define MASK_DECLARE_4(x)	MASK_DECLARE_2(x), MASK_DECLARE_2(x+2)
709e56b3bc7SLinus Torvalds #define MASK_DECLARE_8(x)	MASK_DECLARE_4(x), MASK_DECLARE_4(x+4)
710b8d317d1SMike Travis 
711e56b3bc7SLinus Torvalds const unsigned long cpu_bit_bitmap[BITS_PER_LONG+1][BITS_TO_LONGS(NR_CPUS)] = {
712b8d317d1SMike Travis 
713e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(0),	MASK_DECLARE_8(8),
714e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(16),	MASK_DECLARE_8(24),
715e56b3bc7SLinus Torvalds #if BITS_PER_LONG > 32
716e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(32),	MASK_DECLARE_8(40),
717e56b3bc7SLinus Torvalds 	MASK_DECLARE_8(48),	MASK_DECLARE_8(56),
718b8d317d1SMike Travis #endif
719b8d317d1SMike Travis };
720e56b3bc7SLinus Torvalds EXPORT_SYMBOL_GPL(cpu_bit_bitmap);
7212d3854a3SRusty Russell 
7222d3854a3SRusty Russell const DECLARE_BITMAP(cpu_all_bits, NR_CPUS) = CPU_BITS_ALL;
7232d3854a3SRusty Russell EXPORT_SYMBOL(cpu_all_bits);
724b3199c02SRusty Russell 
725b3199c02SRusty Russell #ifdef CONFIG_INIT_ALL_POSSIBLE
726b3199c02SRusty Russell static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly
727b3199c02SRusty Russell 	= CPU_BITS_ALL;
728b3199c02SRusty Russell #else
729b3199c02SRusty Russell static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly;
730b3199c02SRusty Russell #endif
731b3199c02SRusty Russell const struct cpumask *const cpu_possible_mask = to_cpumask(cpu_possible_bits);
732b3199c02SRusty Russell EXPORT_SYMBOL(cpu_possible_mask);
733b3199c02SRusty Russell 
734b3199c02SRusty Russell static DECLARE_BITMAP(cpu_online_bits, CONFIG_NR_CPUS) __read_mostly;
735b3199c02SRusty Russell const struct cpumask *const cpu_online_mask = to_cpumask(cpu_online_bits);
736b3199c02SRusty Russell EXPORT_SYMBOL(cpu_online_mask);
737b3199c02SRusty Russell 
738b3199c02SRusty Russell static DECLARE_BITMAP(cpu_present_bits, CONFIG_NR_CPUS) __read_mostly;
739b3199c02SRusty Russell const struct cpumask *const cpu_present_mask = to_cpumask(cpu_present_bits);
740b3199c02SRusty Russell EXPORT_SYMBOL(cpu_present_mask);
741b3199c02SRusty Russell 
742b3199c02SRusty Russell static DECLARE_BITMAP(cpu_active_bits, CONFIG_NR_CPUS) __read_mostly;
743b3199c02SRusty Russell const struct cpumask *const cpu_active_mask = to_cpumask(cpu_active_bits);
744b3199c02SRusty Russell EXPORT_SYMBOL(cpu_active_mask);
7453fa41520SRusty Russell 
7463fa41520SRusty Russell void set_cpu_possible(unsigned int cpu, bool possible)
7473fa41520SRusty Russell {
7483fa41520SRusty Russell 	if (possible)
7493fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_possible_bits));
7503fa41520SRusty Russell 	else
7513fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_possible_bits));
7523fa41520SRusty Russell }
7533fa41520SRusty Russell 
7543fa41520SRusty Russell void set_cpu_present(unsigned int cpu, bool present)
7553fa41520SRusty Russell {
7563fa41520SRusty Russell 	if (present)
7573fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_present_bits));
7583fa41520SRusty Russell 	else
7593fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_present_bits));
7603fa41520SRusty Russell }
7613fa41520SRusty Russell 
7623fa41520SRusty Russell void set_cpu_online(unsigned int cpu, bool online)
7633fa41520SRusty Russell {
7646acbfb96SLai Jiangshan 	if (online) {
7653fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_online_bits));
7666acbfb96SLai Jiangshan 		cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
7676acbfb96SLai Jiangshan 	} else {
7683fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_online_bits));
7693fa41520SRusty Russell 	}
7706acbfb96SLai Jiangshan }
7713fa41520SRusty Russell 
7723fa41520SRusty Russell void set_cpu_active(unsigned int cpu, bool active)
7733fa41520SRusty Russell {
7743fa41520SRusty Russell 	if (active)
7753fa41520SRusty Russell 		cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
7763fa41520SRusty Russell 	else
7773fa41520SRusty Russell 		cpumask_clear_cpu(cpu, to_cpumask(cpu_active_bits));
7783fa41520SRusty Russell }
7793fa41520SRusty Russell 
7803fa41520SRusty Russell void init_cpu_present(const struct cpumask *src)
7813fa41520SRusty Russell {
7823fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_present_bits), src);
7833fa41520SRusty Russell }
7843fa41520SRusty Russell 
7853fa41520SRusty Russell void init_cpu_possible(const struct cpumask *src)
7863fa41520SRusty Russell {
7873fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_possible_bits), src);
7883fa41520SRusty Russell }
7893fa41520SRusty Russell 
7903fa41520SRusty Russell void init_cpu_online(const struct cpumask *src)
7913fa41520SRusty Russell {
7923fa41520SRusty Russell 	cpumask_copy(to_cpumask(cpu_online_bits), src);
7933fa41520SRusty Russell }
794