ef1d2b30af62907e1413255d0bd1d3910d08444f
[projects/modsched/linux.git] / kernel / cpu.c
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/oom.h>
14 #include <linux/rcupdate.h>
15 #include <linux/export.h>
16 #include <linux/bug.h>
17 #include <linux/kthread.h>
18 #include <linux/stop_machine.h>
19 #include <linux/mutex.h>
20 #include <linux/gfp.h>
21 #include <linux/suspend.h>
22 #include <linux/lockdep.h>
23 #include <linux/tick.h>
24 #include <trace/events/power.h>
25
26 #ifdef CONFIG_MOD_SCHED
27     #include <fw_comm.h>
28 #endif
29
30 #include "smpboot.h"
31
32 #ifdef CONFIG_SMP
33 /* Serializes the updates to cpu_online_mask, cpu_present_mask */
34 static DEFINE_MUTEX(cpu_add_remove_lock);
35
36 /*
37  * The following two APIs (cpu_maps_update_begin/done) must be used when
38  * attempting to serialize the updates to cpu_online_mask & cpu_present_mask.
39  * The APIs cpu_notifier_register_begin/done() must be used to protect CPU
40  * hotplug callback (un)registration performed using __register_cpu_notifier()
41  * or __unregister_cpu_notifier().
42  */
43 void cpu_maps_update_begin(void)
44 {
45         mutex_lock(&cpu_add_remove_lock);
46 }
47 EXPORT_SYMBOL(cpu_notifier_register_begin);
48
49 void cpu_maps_update_done(void)
50 {
51         mutex_unlock(&cpu_add_remove_lock);
52 }
53 EXPORT_SYMBOL(cpu_notifier_register_done);
54
55 static RAW_NOTIFIER_HEAD(cpu_chain);
56
57 /* If set, cpu_up and cpu_down will return -EBUSY and do nothing.
58  * Should always be manipulated under cpu_add_remove_lock
59  */
60 static int cpu_hotplug_disabled;
61
62 #ifdef CONFIG_HOTPLUG_CPU
63
64 static struct {
65         struct task_struct *active_writer;
66         /* wait queue to wake up the active_writer */
67         wait_queue_head_t wq;
68         /* verifies that no writer will get active while readers are active */
69         struct mutex lock;
70         /*
71          * Also blocks the new readers during
72          * an ongoing cpu hotplug operation.
73          */
74         atomic_t refcount;
75
76 #ifdef CONFIG_DEBUG_LOCK_ALLOC
77         struct lockdep_map dep_map;
78 #endif
79 } cpu_hotplug = {
80         .active_writer = NULL,
81         .wq = __WAIT_QUEUE_HEAD_INITIALIZER(cpu_hotplug.wq),
82         .lock = __MUTEX_INITIALIZER(cpu_hotplug.lock),
83 #ifdef CONFIG_DEBUG_LOCK_ALLOC
84         .dep_map = {.name = "cpu_hotplug.lock" },
85 #endif
86 };
87
88 /* Lockdep annotations for get/put_online_cpus() and cpu_hotplug_begin/end() */
89 #define cpuhp_lock_acquire_read() lock_map_acquire_read(&cpu_hotplug.dep_map)
90 #define cpuhp_lock_acquire_tryread() \
91                                   lock_map_acquire_tryread(&cpu_hotplug.dep_map)
92 #define cpuhp_lock_acquire()      lock_map_acquire(&cpu_hotplug.dep_map)
93 #define cpuhp_lock_release()      lock_map_release(&cpu_hotplug.dep_map)
94
95
96 void get_online_cpus(void)
97 {
98         might_sleep();
99         if (cpu_hotplug.active_writer == current)
100                 return;
101         cpuhp_lock_acquire_read();
102         mutex_lock(&cpu_hotplug.lock);
103         atomic_inc(&cpu_hotplug.refcount);
104         mutex_unlock(&cpu_hotplug.lock);
105 }
106 EXPORT_SYMBOL_GPL(get_online_cpus);
107
108 bool try_get_online_cpus(void)
109 {
110         if (cpu_hotplug.active_writer == current)
111                 return true;
112         if (!mutex_trylock(&cpu_hotplug.lock))
113                 return false;
114         cpuhp_lock_acquire_tryread();
115         atomic_inc(&cpu_hotplug.refcount);
116         mutex_unlock(&cpu_hotplug.lock);
117         return true;
118 }
119 EXPORT_SYMBOL_GPL(try_get_online_cpus);
120
121 void put_online_cpus(void)
122 {
123         int refcount;
124
125         if (cpu_hotplug.active_writer == current)
126                 return;
127
128         refcount = atomic_dec_return(&cpu_hotplug.refcount);
129         if (WARN_ON(refcount < 0)) /* try to fix things up */
130                 atomic_inc(&cpu_hotplug.refcount);
131
132         if (refcount <= 0 && waitqueue_active(&cpu_hotplug.wq))
133                 wake_up(&cpu_hotplug.wq);
134
135         cpuhp_lock_release();
136
137 }
138 EXPORT_SYMBOL_GPL(put_online_cpus);
139
140 /*
141  * This ensures that the hotplug operation can begin only when the
142  * refcount goes to zero.
143  *
144  * Note that during a cpu-hotplug operation, the new readers, if any,
145  * will be blocked by the cpu_hotplug.lock
146  *
147  * Since cpu_hotplug_begin() is always called after invoking
148  * cpu_maps_update_begin(), we can be sure that only one writer is active.
149  *
150  * Note that theoretically, there is a possibility of a livelock:
151  * - Refcount goes to zero, last reader wakes up the sleeping
152  *   writer.
153  * - Last reader unlocks the cpu_hotplug.lock.
154  * - A new reader arrives at this moment, bumps up the refcount.
155  * - The writer acquires the cpu_hotplug.lock finds the refcount
156  *   non zero and goes to sleep again.
157  *
158  * However, this is very difficult to achieve in practice since
159  * get_online_cpus() not an api which is called all that often.
160  *
161  */
162 void cpu_hotplug_begin(void)
163 {
164         DEFINE_WAIT(wait);
165
166         cpu_hotplug.active_writer = current;
167         cpuhp_lock_acquire();
168
169         for (;;) {
170                 mutex_lock(&cpu_hotplug.lock);
171                 prepare_to_wait(&cpu_hotplug.wq, &wait, TASK_UNINTERRUPTIBLE);
172                 if (likely(!atomic_read(&cpu_hotplug.refcount)))
173                                 break;
174                 mutex_unlock(&cpu_hotplug.lock);
175                 schedule();
176         }
177         finish_wait(&cpu_hotplug.wq, &wait);
178 }
179
180 void cpu_hotplug_done(void)
181 {
182         cpu_hotplug.active_writer = NULL;
183         mutex_unlock(&cpu_hotplug.lock);
184         cpuhp_lock_release();
185 }
186
187 /*
188  * Wait for currently running CPU hotplug operations to complete (if any) and
189  * disable future CPU hotplug (from sysfs). The 'cpu_add_remove_lock' protects
190  * the 'cpu_hotplug_disabled' flag. The same lock is also acquired by the
191  * hotplug path before performing hotplug operations. So acquiring that lock
192  * guarantees mutual exclusion from any currently running hotplug operations.
193  */
194 void cpu_hotplug_disable(void)
195 {
196         cpu_maps_update_begin();
197         cpu_hotplug_disabled = 1;
198         cpu_maps_update_done();
199 }
200
201 void cpu_hotplug_enable(void)
202 {
203         cpu_maps_update_begin();
204         cpu_hotplug_disabled = 0;
205         cpu_maps_update_done();
206 }
207
208 #endif  /* CONFIG_HOTPLUG_CPU */
209
210 /* Need to know about CPUs going up/down? */
211 int __ref register_cpu_notifier(struct notifier_block *nb)
212 {
213         int ret;
214         cpu_maps_update_begin();
215         ret = raw_notifier_chain_register(&cpu_chain, nb);
216         cpu_maps_update_done();
217         return ret;
218 }
219
220 int __ref __register_cpu_notifier(struct notifier_block *nb)
221 {
222         return raw_notifier_chain_register(&cpu_chain, nb);
223 }
224
225 static int __cpu_notify(unsigned long val, void *v, int nr_to_call,
226                         int *nr_calls)
227 {
228         int ret;
229
230         ret = __raw_notifier_call_chain(&cpu_chain, val, v, nr_to_call,
231                                         nr_calls);
232
233         return notifier_to_errno(ret);
234 }
235
236 static int cpu_notify(unsigned long val, void *v)
237 {
238         return __cpu_notify(val, v, -1, NULL);
239 }
240
241 #ifdef CONFIG_HOTPLUG_CPU
242
243 static void cpu_notify_nofail(unsigned long val, void *v)
244 {
245         BUG_ON(cpu_notify(val, v));
246 }
247 EXPORT_SYMBOL(register_cpu_notifier);
248 EXPORT_SYMBOL(__register_cpu_notifier);
249
250 void __ref unregister_cpu_notifier(struct notifier_block *nb)
251 {
252         cpu_maps_update_begin();
253         raw_notifier_chain_unregister(&cpu_chain, nb);
254         cpu_maps_update_done();
255 }
256 EXPORT_SYMBOL(unregister_cpu_notifier);
257
258 void __ref __unregister_cpu_notifier(struct notifier_block *nb)
259 {
260         raw_notifier_chain_unregister(&cpu_chain, nb);
261 }
262 EXPORT_SYMBOL(__unregister_cpu_notifier);
263
264 /**
265  * clear_tasks_mm_cpumask - Safely clear tasks' mm_cpumask for a CPU
266  * @cpu: a CPU id
267  *
268  * This function walks all processes, finds a valid mm struct for each one and
269  * then clears a corresponding bit in mm's cpumask.  While this all sounds
270  * trivial, there are various non-obvious corner cases, which this function
271  * tries to solve in a safe manner.
272  *
273  * Also note that the function uses a somewhat relaxed locking scheme, so it may
274  * be called only for an already offlined CPU.
275  */
276 void clear_tasks_mm_cpumask(int cpu)
277 {
278         struct task_struct *p;
279
280         /*
281          * This function is called after the cpu is taken down and marked
282          * offline, so its not like new tasks will ever get this cpu set in
283          * their mm mask. -- Peter Zijlstra
284          * Thus, we may use rcu_read_lock() here, instead of grabbing
285          * full-fledged tasklist_lock.
286          */
287         WARN_ON(cpu_online(cpu));
288         rcu_read_lock();
289         for_each_process(p) {
290                 struct task_struct *t;
291
292                 /*
293                  * Main thread might exit, but other threads may still have
294                  * a valid mm. Find one.
295                  */
296                 t = find_lock_task_mm(p);
297                 if (!t)
298                         continue;
299                 cpumask_clear_cpu(cpu, mm_cpumask(t->mm));
300                 task_unlock(t);
301         }
302         rcu_read_unlock();
303 }
304
305 static inline void check_for_tasks(int dead_cpu)
306 {
307         struct task_struct *g, *p;
308
309         read_lock_irq(&tasklist_lock);
310         do_each_thread(g, p) {
311                 if (!p->on_rq)
312                         continue;
313                 /*
314                  * We do the check with unlocked task_rq(p)->lock.
315                  * Order the reading to do not warn about a task,
316                  * which was running on this cpu in the past, and
317                  * it's just been woken on another cpu.
318                  */
319                 rmb();
320                 if (task_cpu(p) != dead_cpu)
321                         continue;
322
323                 pr_warn("Task %s (pid=%d) is on cpu %d (state=%ld, flags=%x)\n",
324                         p->comm, task_pid_nr(p), dead_cpu, p->state, p->flags);
325         } while_each_thread(g, p);
326         read_unlock_irq(&tasklist_lock);
327 }
328
329 struct take_cpu_down_param {
330         unsigned long mod;
331         void *hcpu;
332 };
333
334 /* Take this CPU down. */
335 static int __ref take_cpu_down(void *_param)
336 {
337         struct take_cpu_down_param *param = _param;
338         int err;
339
340         /* Ensure this CPU doesn't handle any more interrupts. */
341         err = __cpu_disable();
342         if (err < 0)
343                 return err;
344
345         cpu_notify(CPU_DYING | param->mod, param->hcpu);
346         /* Give up timekeeping duties */
347         tick_handover_do_timer();
348         /* Park the stopper thread */
349         kthread_park(current);
350         return 0;
351 }
352
353 /* Requires cpu_add_remove_lock to be held */
354 static int __ref _cpu_down(unsigned int cpu, int tasks_frozen)
355 {
356         int err, nr_calls = 0;
357         void *hcpu = (void *)(long)cpu;
358         unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
359         struct take_cpu_down_param tcd_param = {
360                 .mod = mod,
361                 .hcpu = hcpu,
362         };
363
364         if (num_online_cpus() == 1)
365                 return -EBUSY;
366
367         if (!cpu_online(cpu))
368                 return -EINVAL;
369
370         cpu_hotplug_begin();
371
372         err = __cpu_notify(CPU_DOWN_PREPARE | mod, hcpu, -1, &nr_calls);
373         if (err) {
374                 nr_calls--;
375                 __cpu_notify(CPU_DOWN_FAILED | mod, hcpu, nr_calls, NULL);
376                 pr_warn("%s: attempt to take down CPU %u failed\n",
377                         __func__, cpu);
378                 goto out_release;
379         }
380
381         /*
382          * By now we've cleared cpu_active_mask, wait for all preempt-disabled
383          * and RCU users of this state to go away such that all new such users
384          * will observe it.
385          *
386          * For CONFIG_PREEMPT we have preemptible RCU and its sync_rcu() might
387          * not imply sync_sched(), so explicitly call both.
388          *
389          * Do sync before park smpboot threads to take care the rcu boost case.
390          */
391 #ifdef CONFIG_PREEMPT
392         synchronize_sched();
393 #endif
394         synchronize_rcu();
395
396         smpboot_park_threads(cpu);
397
398         /*
399          * So now all preempt/rcu users must observe !cpu_active().
400          */
401
402         err = __stop_machine(take_cpu_down, &tcd_param, cpumask_of(cpu));
403         if (err) {
404                 /* CPU didn't die: tell everyone.  Can't complain. */
405                 smpboot_unpark_threads(cpu);
406                 cpu_notify_nofail(CPU_DOWN_FAILED | mod, hcpu);
407                 goto out_release;
408         }
409         BUG_ON(cpu_online(cpu));
410
411         /*
412          * The migration_call() CPU_DYING callback will have removed all
413          * runnable tasks from the cpu, there's only the idle task left now
414          * that the migration thread is done doing the stop_machine thing.
415          *
416          * Wait for the stop thread to go away.
417          */
418         while (!per_cpu(cpu_dead_idle, cpu))
419                 cpu_relax();
420         smp_mb(); /* Read from cpu_dead_idle before __cpu_die(). */
421         per_cpu(cpu_dead_idle, cpu) = false;
422
423         hotplug_cpu__broadcast_tick_pull(cpu);
424         /* This actually kills the CPU. */
425         __cpu_die(cpu);
426
427         /* CPU is completely dead: tell everyone.  Too late to complain. */
428         tick_cleanup_dead_cpu(cpu);
429         cpu_notify_nofail(CPU_DEAD | mod, hcpu);
430
431         check_for_tasks(cpu);
432
433 out_release:
434         cpu_hotplug_done();
435         if (!err)
436                 cpu_notify_nofail(CPU_POST_DEAD | mod, hcpu);
437         return err;
438 }
439
440 int __ref cpu_down(unsigned int cpu)
441 {
442         int err;
443
444         cpu_maps_update_begin();
445
446         if (cpu_hotplug_disabled) {
447                 err = -EBUSY;
448                 goto out;
449         }
450
451         err = _cpu_down(cpu, 0);
452
453 out:
454         cpu_maps_update_done();
455         return err;
456 }
457 EXPORT_SYMBOL(cpu_down);
458 #endif /*CONFIG_HOTPLUG_CPU*/
459
460 /*
461  * Unpark per-CPU smpboot kthreads at CPU-online time.
462  */
463 static int smpboot_thread_call(struct notifier_block *nfb,
464                                unsigned long action, void *hcpu)
465 {
466         int cpu = (long)hcpu;
467
468         switch (action & ~CPU_TASKS_FROZEN) {
469
470         case CPU_ONLINE:
471                 smpboot_unpark_threads(cpu);
472                 break;
473
474         default:
475                 break;
476         }
477
478         return NOTIFY_OK;
479 }
480
481 static struct notifier_block smpboot_thread_notifier = {
482         .notifier_call = smpboot_thread_call,
483         .priority = CPU_PRI_SMPBOOT,
484 };
485
486 void __cpuinit smpboot_thread_init(void)
487 {
488         register_cpu_notifier(&smpboot_thread_notifier);
489 }
490
491 /* Requires cpu_add_remove_lock to be held */
492 static int _cpu_up(unsigned int cpu, int tasks_frozen)
493 {
494         int ret, nr_calls = 0;
495         void *hcpu = (void *)(long)cpu;
496         unsigned long mod = tasks_frozen ? CPU_TASKS_FROZEN : 0;
497         struct task_struct *idle;
498
499         cpu_hotplug_begin();
500
501         if (cpu_online(cpu) || !cpu_present(cpu)) {
502                 ret = -EINVAL;
503                 goto out;
504         }
505
506         idle = idle_thread_get(cpu);
507         if (IS_ERR(idle)) {
508                 ret = PTR_ERR(idle);
509                 goto out;
510         }
511
512         ret = smpboot_create_threads(cpu);
513         if (ret)
514                 goto out;
515
516         ret = __cpu_notify(CPU_UP_PREPARE | mod, hcpu, -1, &nr_calls);
517         if (ret) {
518                 nr_calls--;
519                 pr_warn("%s: attempt to bring up CPU %u failed\n",
520                         __func__, cpu);
521                 goto out_notify;
522         }
523
524         /* Arch-specific enabling code. */
525         ret = __cpu_up(cpu, idle);
526         if (ret != 0)
527                 goto out_notify;
528         BUG_ON(!cpu_online(cpu));
529
530         /* Now call notifier in preparation. */
531         cpu_notify(CPU_ONLINE | mod, hcpu);
532
533 out_notify:
534         if (ret != 0)
535                 __cpu_notify(CPU_UP_CANCELED | mod, hcpu, nr_calls, NULL);
536 out:
537         cpu_hotplug_done();
538
539         return ret;
540 }
541
542 int cpu_up(unsigned int cpu)
543 {
544         int err = 0;
545
546         if (!cpu_possible(cpu)) {
547                 pr_err("can't online cpu %d because it is not configured as may-hotadd at boot time\n",
548                        cpu);
549 #if defined(CONFIG_IA64)
550                 pr_err("please check additional_cpus= boot parameter\n");
551 #endif
552                 return -EINVAL;
553         }
554
555         err = try_online_node(cpu_to_node(cpu));
556         if (err)
557                 return err;
558
559         cpu_maps_update_begin();
560
561         if (cpu_hotplug_disabled) {
562                 err = -EBUSY;
563                 goto out;
564         }
565
566         err = _cpu_up(cpu, 0);
567
568 #ifdef CONFIG_MOD_SCHED
569         if(!err)
570                 fw_notify(FW_CPU_ONLINE,&cpu);
571 #endif
572
573 out:
574         cpu_maps_update_done();
575         return err;
576 }
577 EXPORT_SYMBOL_GPL(cpu_up);
578
579 #ifdef CONFIG_PM_SLEEP_SMP
580 static cpumask_var_t frozen_cpus;
581
582 int disable_nonboot_cpus(void)
583 {
584         int cpu, first_cpu, error = 0;
585
586         cpu_maps_update_begin();
587         first_cpu = cpumask_first(cpu_online_mask);
588         /*
589          * We take down all of the non-boot CPUs in one shot to avoid races
590          * with the userspace trying to use the CPU hotplug at the same time
591          */
592         cpumask_clear(frozen_cpus);
593
594         pr_info("Disabling non-boot CPUs ...\n");
595         for_each_online_cpu(cpu) {
596                 if (cpu == first_cpu)
597                         continue;
598                 trace_suspend_resume(TPS("CPU_OFF"), cpu, true);
599                 error = _cpu_down(cpu, 1);
600                 trace_suspend_resume(TPS("CPU_OFF"), cpu, false);
601                 if (!error)
602                         cpumask_set_cpu(cpu, frozen_cpus);
603                 else {
604                         pr_err("Error taking CPU%d down: %d\n", cpu, error);
605                         break;
606                 }
607         }
608
609         if (!error) {
610                 BUG_ON(num_online_cpus() > 1);
611                 /* Make sure the CPUs won't be enabled by someone else */
612                 cpu_hotplug_disabled = 1;
613         } else {
614                 pr_err("Non-boot CPUs are not disabled\n");
615         }
616         cpu_maps_update_done();
617         return error;
618 }
619
620 void __weak arch_enable_nonboot_cpus_begin(void)
621 {
622 }
623
624 void __weak arch_enable_nonboot_cpus_end(void)
625 {
626 }
627
628 void __ref enable_nonboot_cpus(void)
629 {
630         int cpu, error;
631
632         /* Allow everyone to use the CPU hotplug again */
633         cpu_maps_update_begin();
634         cpu_hotplug_disabled = 0;
635         if (cpumask_empty(frozen_cpus))
636                 goto out;
637
638         pr_info("Enabling non-boot CPUs ...\n");
639
640         arch_enable_nonboot_cpus_begin();
641
642         for_each_cpu(cpu, frozen_cpus) {
643                 trace_suspend_resume(TPS("CPU_ON"), cpu, true);
644                 error = _cpu_up(cpu, 1);
645                 trace_suspend_resume(TPS("CPU_ON"), cpu, false);
646                 if (!error) {
647                         pr_info("CPU%d is up\n", cpu);
648                         continue;
649                 }
650                 pr_warn("Error taking CPU%d up: %d\n", cpu, error);
651         }
652
653         arch_enable_nonboot_cpus_end();
654
655         cpumask_clear(frozen_cpus);
656 out:
657         cpu_maps_update_done();
658 }
659
660 static int __init alloc_frozen_cpus(void)
661 {
662         if (!alloc_cpumask_var(&frozen_cpus, GFP_KERNEL|__GFP_ZERO))
663                 return -ENOMEM;
664         return 0;
665 }
666 core_initcall(alloc_frozen_cpus);
667
668 /*
669  * When callbacks for CPU hotplug notifications are being executed, we must
670  * ensure that the state of the system with respect to the tasks being frozen
671  * or not, as reported by the notification, remains unchanged *throughout the
672  * duration* of the execution of the callbacks.
673  * Hence we need to prevent the freezer from racing with regular CPU hotplug.
674  *
675  * This synchronization is implemented by mutually excluding regular CPU
676  * hotplug and Suspend/Hibernate call paths by hooking onto the Suspend/
677  * Hibernate notifications.
678  */
679 static int
680 cpu_hotplug_pm_callback(struct notifier_block *nb,
681                         unsigned long action, void *ptr)
682 {
683         switch (action) {
684
685         case PM_SUSPEND_PREPARE:
686         case PM_HIBERNATION_PREPARE:
687                 cpu_hotplug_disable();
688                 break;
689
690         case PM_POST_SUSPEND:
691         case PM_POST_HIBERNATION:
692                 cpu_hotplug_enable();
693                 break;
694
695         default:
696                 return NOTIFY_DONE;
697         }
698
699         return NOTIFY_OK;
700 }
701
702
703 static int __init cpu_hotplug_pm_sync_init(void)
704 {
705         /*
706          * cpu_hotplug_pm_callback has higher priority than x86
707          * bsp_pm_callback which depends on cpu_hotplug_pm_callback
708          * to disable cpu hotplug to avoid cpu hotplug race.
709          */
710         pm_notifier(cpu_hotplug_pm_callback, 0);
711         return 0;
712 }
713 core_initcall(cpu_hotplug_pm_sync_init);
714
715 #endif /* CONFIG_PM_SLEEP_SMP */
716
717 /**
718  * notify_cpu_starting(cpu) - call the CPU_STARTING notifiers
719  * @cpu: cpu that just started
720  *
721  * This function calls the cpu_chain notifiers with CPU_STARTING.
722  * It must be called by the arch code on the new cpu, before the new cpu
723  * enables interrupts and before the "boot" cpu returns from __cpu_up().
724  */
725 void notify_cpu_starting(unsigned int cpu)
726 {
727         unsigned long val = CPU_STARTING;
728
729 #ifdef CONFIG_PM_SLEEP_SMP
730         if (frozen_cpus != NULL && cpumask_test_cpu(cpu, frozen_cpus))
731                 val = CPU_STARTING_FROZEN;
732 #endif /* CONFIG_PM_SLEEP_SMP */
733         cpu_notify(val, (void *)(long)cpu);
734 }
735
736 #endif /* CONFIG_SMP */
737
738 /*
739  * cpu_bit_bitmap[] is a special, "compressed" data structure that
740  * represents all NR_CPUS bits binary values of 1<<nr.
741  *
742  * It is used by cpumask_of() to get a constant address to a CPU
743  * mask value that has a single bit set only.
744  */
745
746 /* cpu_bit_bitmap[0] is empty - so we can back into it */
747 #define MASK_DECLARE_1(x)       [x+1][0] = (1UL << (x))
748 #define MASK_DECLARE_2(x)       MASK_DECLARE_1(x), MASK_DECLARE_1(x+1)
749 #define MASK_DECLARE_4(x)       MASK_DECLARE_2(x), MASK_DECLARE_2(x+2)
750 #define MASK_DECLARE_8(x)       MASK_DECLARE_4(x), MASK_DECLARE_4(x+4)
751
752 const unsigned long cpu_bit_bitmap[BITS_PER_LONG+1][BITS_TO_LONGS(NR_CPUS)] = {
753
754         MASK_DECLARE_8(0),      MASK_DECLARE_8(8),
755         MASK_DECLARE_8(16),     MASK_DECLARE_8(24),
756 #if BITS_PER_LONG > 32
757         MASK_DECLARE_8(32),     MASK_DECLARE_8(40),
758         MASK_DECLARE_8(48),     MASK_DECLARE_8(56),
759 #endif
760 };
761 EXPORT_SYMBOL_GPL(cpu_bit_bitmap);
762
763 const DECLARE_BITMAP(cpu_all_bits, NR_CPUS) = CPU_BITS_ALL;
764 EXPORT_SYMBOL(cpu_all_bits);
765
766 #ifdef CONFIG_INIT_ALL_POSSIBLE
767 static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly
768         = CPU_BITS_ALL;
769 #else
770 static DECLARE_BITMAP(cpu_possible_bits, CONFIG_NR_CPUS) __read_mostly;
771 #endif
772 const struct cpumask *const cpu_possible_mask = to_cpumask(cpu_possible_bits);
773 EXPORT_SYMBOL(cpu_possible_mask);
774
775 static DECLARE_BITMAP(cpu_online_bits, CONFIG_NR_CPUS) __read_mostly;
776 const struct cpumask *const cpu_online_mask = to_cpumask(cpu_online_bits);
777 EXPORT_SYMBOL(cpu_online_mask);
778
779 static DECLARE_BITMAP(cpu_present_bits, CONFIG_NR_CPUS) __read_mostly;
780 const struct cpumask *const cpu_present_mask = to_cpumask(cpu_present_bits);
781 EXPORT_SYMBOL(cpu_present_mask);
782
783 static DECLARE_BITMAP(cpu_active_bits, CONFIG_NR_CPUS) __read_mostly;
784 const struct cpumask *const cpu_active_mask = to_cpumask(cpu_active_bits);
785 EXPORT_SYMBOL(cpu_active_mask);
786
787 void set_cpu_possible(unsigned int cpu, bool possible)
788 {
789         if (possible)
790                 cpumask_set_cpu(cpu, to_cpumask(cpu_possible_bits));
791         else
792                 cpumask_clear_cpu(cpu, to_cpumask(cpu_possible_bits));
793 }
794
795 void set_cpu_present(unsigned int cpu, bool present)
796 {
797         if (present)
798                 cpumask_set_cpu(cpu, to_cpumask(cpu_present_bits));
799         else
800                 cpumask_clear_cpu(cpu, to_cpumask(cpu_present_bits));
801 }
802
803 void set_cpu_online(unsigned int cpu, bool online)
804 {
805         if (online) {
806                 cpumask_set_cpu(cpu, to_cpumask(cpu_online_bits));
807                 cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
808         } else {
809                 cpumask_clear_cpu(cpu, to_cpumask(cpu_online_bits));
810         }
811 }
812
813 void set_cpu_active(unsigned int cpu, bool active)
814 {
815         if (active)
816                 cpumask_set_cpu(cpu, to_cpumask(cpu_active_bits));
817         else
818                 cpumask_clear_cpu(cpu, to_cpumask(cpu_active_bits));
819 }
820
821 void init_cpu_present(const struct cpumask *src)
822 {
823         cpumask_copy(to_cpumask(cpu_present_bits), src);
824 }
825
826 void init_cpu_possible(const struct cpumask *src)
827 {
828         cpumask_copy(to_cpumask(cpu_possible_bits), src);
829 }
830
831 void init_cpu_online(const struct cpumask *src)
832 {
833         cpumask_copy(to_cpumask(cpu_online_bits), src);
834 }