summaryrefslogtreecommitdiff
path: root/kernel/smpboot.c
blob: ed1576981801858a9e83185d018d716194d69084 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/*
 * Common SMP CPU bringup/teardown functions
 */
#include <linux/err.h>
#include <linux/smp.h>
#include <linux/init.h>
#include <linux/sched.h>
#include <linux/percpu.h>
#include <linux/workqueue.h>

#include "smpboot.h"

#ifdef CONFIG_GENERIC_SMP_IDLE_THREAD
struct create_idle {
	struct work_struct	work;
	struct task_struct	*idle;
	struct completion	done;
	unsigned int cpu;
};

static void __cpuinit do_fork_idle(struct work_struct *work)
{
	struct create_idle *c = container_of(work, struct create_idle, work);

	c->idle = fork_idle(c->cpu);
	complete(&c->done);
}

static struct task_struct * __cpuinit idle_thread_create(unsigned int cpu)
{
	struct create_idle c_idle = {
		.cpu	= cpu,
		.done	= COMPLETION_INITIALIZER_ONSTACK(c_idle.done),
	};

	INIT_WORK_ONSTACK(&c_idle.work, do_fork_idle);
	schedule_work(&c_idle.work);
	wait_for_completion(&c_idle.done);
	destroy_work_on_stack(&c_idle.work);
	return c_idle.idle;
}

/*
 * For the hotplug case we keep the task structs around and reuse
 * them.
 */
static DEFINE_PER_CPU(struct task_struct *, idle_threads);

static inline struct task_struct *get_idle_for_cpu(unsigned int cpu)
{
	struct task_struct *tsk = per_cpu(idle_threads, cpu);

	if (!tsk)
		return idle_thread_create(cpu);
	init_idle(tsk, cpu);
	return tsk;
}

struct task_struct * __cpuinit idle_thread_get(unsigned int cpu)
{
	return per_cpu(idle_threads, cpu);
}

void __init idle_thread_set_boot_cpu(void)
{
	per_cpu(idle_threads, smp_processor_id()) = current;
}

/**
 * idle_thread_init - Initialize the idle thread for a cpu
 * @cpu:	The cpu for which the idle thread should be initialized
 *
 * Creates the thread if it does not exist.
 */
static int __cpuinit idle_thread_init(unsigned int cpu)
{
	struct task_struct *idle = get_idle_for_cpu(cpu);

	if (IS_ERR(idle)) {
		printk(KERN_ERR "failed fork for CPU %u\n", cpu);
		return PTR_ERR(idle);
	}
	per_cpu(idle_threads, cpu) = idle;
	return 0;
}
#else
static inline int idle_thread_init(unsigned int cpu) { return 0; }
#endif

/**
 * smpboot_prepare - generic smpboot preparation
 */
int __cpuinit smpboot_prepare(unsigned int cpu)
{
	return idle_thread_init(cpu);
}